blob: 15abf5c5179775e05a20054d297a437580e8b24f [file] [log] [blame]
kjellander@webrtc.org7951e812011-10-13 12:24:41 +00001/*
andrew@webrtc.org7a281a52012-06-27 03:22:37 +00002 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
kjellander@webrtc.org7951e812011-10-13 12:24:41 +00003 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
pbos@webrtc.org34741c82013-05-27 08:02:22 +000011#include "webrtc/test/testsupport/fileutils.h"
kjellander@webrtc.org7951e812011-10-13 12:24:41 +000012
kjellander@webrtc.org7de47bc2014-04-16 08:04:26 +000013#include <assert.h>
henrike@webrtc.orgf2aafe42014-04-29 17:54:17 +000014
15#ifdef WIN32
kjellander@webrtc.org7951e812011-10-13 12:24:41 +000016#include <direct.h>
kjellander@webrtc.org7de47bc2014-04-16 08:04:26 +000017#include <tchar.h>
18#include <windows.h>
kjellander@webrtc.orgde499662013-08-29 11:26:41 +000019#include <algorithm>
kjellander@webrtc.org7de47bc2014-04-16 08:04:26 +000020
Henrik Kjellander98f53512015-10-28 18:17:40 +010021#include "webrtc/system_wrappers/include/utf_util_win.h"
kjellander@webrtc.org7951e812011-10-13 12:24:41 +000022#define GET_CURRENT_DIR _getcwd
kjellander@webrtc.org7951e812011-10-13 12:24:41 +000023#else
24#include <unistd.h>
kjellander@webrtc.org7de47bc2014-04-16 08:04:26 +000025
kwiberg@webrtc.org00b8f6b2015-02-26 14:34:55 +000026#include "webrtc/base/scoped_ptr.h"
kjellander@webrtc.org7951e812011-10-13 12:24:41 +000027#define GET_CURRENT_DIR getcwd
kjellander@webrtc.org4d8cd9d2011-11-09 11:24:14 +000028#endif
29
30#include <sys/stat.h> // To check for directory existence.
31#ifndef S_ISDIR // Not defined in stat.h on Windows.
32#define S_ISDIR(mode) (((mode) & S_IFMT) == S_IFDIR)
kjellander@webrtc.org7951e812011-10-13 12:24:41 +000033#endif
34
pbos@webrtc.org12dc1a32013-08-05 16:22:53 +000035#include <stdio.h>
kjellander@webrtc.org7de47bc2014-04-16 08:04:26 +000036#include <stdlib.h>
pbos@webrtc.org12dc1a32013-08-05 16:22:53 +000037#include <string.h>
kjellander@webrtc.org7951e812011-10-13 12:24:41 +000038
pbos@webrtc.org34741c82013-05-27 08:02:22 +000039#include "webrtc/typedefs.h" // For architecture defines
kjellander@webrtc.org4ed4f242011-12-05 16:31:12 +000040
kjellander@webrtc.org7951e812011-10-13 12:24:41 +000041namespace webrtc {
42namespace test {
43
henrika1d34fe92015-06-16 10:04:20 +020044#if defined(WEBRTC_IOS)
45// Defined in iosfileutils.mm. No header file to discourage use elsewhere.
46std::string IOSResourcePath(std::string name, std::string extension);
47#endif
48
henrike@webrtc.org34773d92013-07-08 14:55:23 +000049namespace {
50
kjellander@webrtc.org4d8cd9d2011-11-09 11:24:14 +000051#ifdef WIN32
henrike@webrtc.org34773d92013-07-08 14:55:23 +000052const char* kPathDelimiter = "\\";
kjellander@webrtc.org4d8cd9d2011-11-09 11:24:14 +000053#else
henrike@webrtc.org34773d92013-07-08 14:55:23 +000054const char* kPathDelimiter = "/";
kjellander@webrtc.org4d8cd9d2011-11-09 11:24:14 +000055#endif
leozwang@webrtc.org363efef2012-10-16 04:31:20 +000056
57#ifdef WEBRTC_ANDROID
kjellander@webrtc.org72fd3392014-11-05 06:28:50 +000058const char* kRootDirName = "/sdcard/";
leozwang@webrtc.org363efef2012-10-16 04:31:20 +000059#else
andrew@webrtc.org1e10bb32011-10-31 20:22:02 +000060// The file we're looking for to identify the project root dir.
henrike@webrtc.org34773d92013-07-08 14:55:23 +000061const char* kProjectRootFileName = "DEPS";
henrike@webrtc.org34773d92013-07-08 14:55:23 +000062const char* kOutputDirName = "out";
kjellander@webrtc.org72fd3392014-11-05 06:28:50 +000063const char* kFallbackPath = "./";
64#endif
henrika1d34fe92015-06-16 10:04:20 +020065#if !defined(WEBRTC_IOS)
kjellander@webrtc.org72fd3392014-11-05 06:28:50 +000066const char* kResourcesDirName = "resources";
henrika1d34fe92015-06-16 10:04:20 +020067#endif
kjellander@webrtc.org72fd3392014-11-05 06:28:50 +000068
pbos@webrtc.orgdb7d82f2013-07-05 08:49:09 +000069char relative_dir_path[FILENAME_MAX];
70bool relative_dir_path_set = false;
henrike@webrtc.org34773d92013-07-08 14:55:23 +000071
72} // namespace
73
74const char* kCannotFindProjectRootDir = "ERROR_CANNOT_FIND_PROJECT_ROOT_DIR";
75
kjellander@webrtc.org193600b2012-10-17 04:39:44 +000076void SetExecutablePath(const std::string& path) {
77 std::string working_dir = WorkingDir();
78 std::string temp_path = path;
79
80 // Handle absolute paths; convert them to relative paths to the working dir.
81 if (path.find(working_dir) != std::string::npos) {
82 temp_path = path.substr(working_dir.length() + 1);
83 }
kjellander@webrtc.orgde499662013-08-29 11:26:41 +000084 // On Windows, when tests are run under memory tools like DrMemory and TSan,
85 // slashes occur in the path as directory separators. Make sure we replace
86 // such cases with backslashes in order for the paths to be correct.
87#ifdef WIN32
88 std::replace(temp_path.begin(), temp_path.end(), '/', '\\');
89#endif
90
kjellander@webrtc.org193600b2012-10-17 04:39:44 +000091 // Trim away the executable name; only store the relative dir path.
92 temp_path = temp_path.substr(0, temp_path.find_last_of(kPathDelimiter));
kjellander@webrtc.org83b767b2012-10-15 18:14:12 +000093 strncpy(relative_dir_path, temp_path.c_str(), FILENAME_MAX);
94 relative_dir_path_set = true;
95}
96
97bool FileExists(std::string& file_name) {
98 struct stat file_info = {0};
99 return stat(file_name.c_str(), &file_info) == 0;
100}
101
leozwang@webrtc.org363efef2012-10-16 04:31:20 +0000102#ifdef WEBRTC_ANDROID
103
104std::string ProjectRootPath() {
kjellander@webrtc.org72fd3392014-11-05 06:28:50 +0000105 return kRootDirName;
leozwang@webrtc.org363efef2012-10-16 04:31:20 +0000106}
107
108std::string OutputPath() {
kjellander@webrtc.org72fd3392014-11-05 06:28:50 +0000109 return kRootDirName;
leozwang@webrtc.org363efef2012-10-16 04:31:20 +0000110}
111
112std::string WorkingDir() {
kjellander@webrtc.org72fd3392014-11-05 06:28:50 +0000113 return kRootDirName;
leozwang@webrtc.org363efef2012-10-16 04:31:20 +0000114}
115
116#else // WEBRTC_ANDROID
117
andrew@webrtc.org0db7dc62011-11-13 01:34:05 +0000118std::string ProjectRootPath() {
kjellander@webrtc.org83b767b2012-10-15 18:14:12 +0000119 std::string path = WorkingDir();
120 if (path == kFallbackPath) {
kjellander@webrtc.org7951e812011-10-13 12:24:41 +0000121 return kCannotFindProjectRootDir;
122 }
kjellander@webrtc.org83b767b2012-10-15 18:14:12 +0000123 if (relative_dir_path_set) {
124 path = path + kPathDelimiter + relative_dir_path;
125 }
kjellander@webrtc.org7951e812011-10-13 12:24:41 +0000126 // Check for our file that verifies the root dir.
kjellander@webrtc.org193600b2012-10-17 04:39:44 +0000127 size_t path_delimiter_index = path.find_last_of(kPathDelimiter);
128 while (path_delimiter_index != std::string::npos) {
kjellander@webrtc.org83b767b2012-10-15 18:14:12 +0000129 std::string root_filename = path + kPathDelimiter + kProjectRootFileName;
130 if (FileExists(root_filename)) {
131 return path + kPathDelimiter;
kjellander@webrtc.org7951e812011-10-13 12:24:41 +0000132 }
kjellander@webrtc.org7951e812011-10-13 12:24:41 +0000133 // Move up one directory in the directory tree.
kjellander@webrtc.org83b767b2012-10-15 18:14:12 +0000134 path = path.substr(0, path_delimiter_index);
135 path_delimiter_index = path.find_last_of(kPathDelimiter);
kjellander@webrtc.org7951e812011-10-13 12:24:41 +0000136 }
kjellander@webrtc.org7951e812011-10-13 12:24:41 +0000137 // Reached the root directory.
138 fprintf(stderr, "Cannot find project root directory!\n");
139 return kCannotFindProjectRootDir;
140}
141
andrew@webrtc.org0db7dc62011-11-13 01:34:05 +0000142std::string OutputPath() {
kjellander@webrtc.org72fd3392014-11-05 06:28:50 +0000143 std::string path = ProjectRootPath();
144 if (path == kCannotFindProjectRootDir) {
145 return kFallbackPath;
146 }
147 path += kOutputDirName;
148 if (!CreateDir(path)) {
149 return kFallbackPath;
150 }
151 return path + kPathDelimiter;
kjellander@webrtc.org4d8cd9d2011-11-09 11:24:14 +0000152}
kjellander@webrtc.org4ed4f242011-12-05 16:31:12 +0000153
154std::string WorkingDir() {
155 char path_buffer[FILENAME_MAX];
156 if (!GET_CURRENT_DIR(path_buffer, sizeof(path_buffer))) {
157 fprintf(stderr, "Cannot get current directory!\n");
158 return kFallbackPath;
kjellander@webrtc.org5b97b122011-12-08 07:42:18 +0000159 } else {
kjellander@webrtc.org4ed4f242011-12-05 16:31:12 +0000160 return std::string(path_buffer);
161 }
162}
163
leozwang@webrtc.org363efef2012-10-16 04:31:20 +0000164#endif // !WEBRTC_ANDROID
165
kjellander@webrtc.org7de47bc2014-04-16 08:04:26 +0000166// Generate a temporary filename in a safe way.
167// Largely copied from talk/base/{unixfilesystem,win32filesystem}.cc.
168std::string TempFilename(const std::string &dir, const std::string &prefix) {
169#ifdef WIN32
170 wchar_t filename[MAX_PATH];
171 if (::GetTempFileName(ToUtf16(dir).c_str(),
172 ToUtf16(prefix).c_str(), 0, filename) != 0)
173 return ToUtf8(filename);
174 assert(false);
175 return "";
176#else
177 int len = dir.size() + prefix.size() + 2 + 6;
kwiberg@webrtc.org00b8f6b2015-02-26 14:34:55 +0000178 rtc::scoped_ptr<char[]> tempname(new char[len]);
kjellander@webrtc.org7de47bc2014-04-16 08:04:26 +0000179
180 snprintf(tempname.get(), len, "%s/%sXXXXXX", dir.c_str(),
181 prefix.c_str());
182 int fd = ::mkstemp(tempname.get());
183 if (fd == -1) {
184 assert(false);
185 return "";
186 } else {
187 ::close(fd);
188 }
189 std::string ret(tempname.get());
190 return ret;
191#endif
192}
193
194bool CreateDir(std::string directory_name) {
kjellander@webrtc.org4ed4f242011-12-05 16:31:12 +0000195 struct stat path_info = {0};
196 // Check if the path exists already:
197 if (stat(directory_name.c_str(), &path_info) == 0) {
198 if (!S_ISDIR(path_info.st_mode)) {
199 fprintf(stderr, "Path %s exists but is not a directory! Remove this "
200 "file and re-run to create the directory.\n",
201 directory_name.c_str());
202 return false;
203 }
204 } else {
205#ifdef WIN32
206 return _mkdir(directory_name.c_str()) == 0;
207#else
208 return mkdir(directory_name.c_str(), S_IRWXU | S_IRWXG | S_IRWXO) == 0;
209#endif
210 }
211 return true;
212}
213
kjellander@webrtc.org4ed4f242011-12-05 16:31:12 +0000214std::string ResourcePath(std::string name, std::string extension) {
henrika1d34fe92015-06-16 10:04:20 +0200215#if defined(WEBRTC_IOS)
216 return IOSResourcePath(name, extension);
217#else
kjellander@webrtc.org4ed4f242011-12-05 16:31:12 +0000218 std::string platform = "win";
219#ifdef WEBRTC_LINUX
220 platform = "linux";
221#endif // WEBRTC_LINUX
222#ifdef WEBRTC_MAC
223 platform = "mac";
224#endif // WEBRTC_MAC
225
226#ifdef WEBRTC_ARCH_64_BITS
227 std::string architecture = "64";
228#else
229 std::string architecture = "32";
230#endif // WEBRTC_ARCH_64_BITS
231
kjellander@webrtc.org5b97b122011-12-08 07:42:18 +0000232 std::string resources_path = ProjectRootPath() + kResourcesDirName +
233 kPathDelimiter;
kjellander@webrtc.org4ed4f242011-12-05 16:31:12 +0000234 std::string resource_file = resources_path + name + "_" + platform + "_" +
235 architecture + "." + extension;
kjellander@webrtc.org80b26612011-12-07 18:50:17 +0000236 if (FileExists(resource_file)) {
kjellander@webrtc.org4ed4f242011-12-05 16:31:12 +0000237 return resource_file;
238 }
239 // Try without architecture.
240 resource_file = resources_path + name + "_" + platform + "." + extension;
241 if (FileExists(resource_file)) {
242 return resource_file;
243 }
244 // Try without platform.
245 resource_file = resources_path + name + "_" + architecture + "." + extension;
246 if (FileExists(resource_file)) {
247 return resource_file;
248 }
leozwang@webrtc.org363efef2012-10-16 04:31:20 +0000249
kjellander@webrtc.org4ed4f242011-12-05 16:31:12 +0000250 // Fall back on name without architecture or platform.
251 return resources_path + name + "." + extension;
henrika1d34fe92015-06-16 10:04:20 +0200252#endif // defined (WEBRTC_IOS)
kjellander@webrtc.org4ed4f242011-12-05 16:31:12 +0000253}
254
kjellander@webrtc.org5b97b122011-12-08 07:42:18 +0000255size_t GetFileSize(std::string filename) {
256 FILE* f = fopen(filename.c_str(), "rb");
257 size_t size = 0;
258 if (f != NULL) {
259 if (fseek(f, 0, SEEK_END) == 0) {
260 size = ftell(f);
261 }
262 fclose(f);
263 }
264 return size;
265}
266
kjellander@webrtc.org7951e812011-10-13 12:24:41 +0000267} // namespace test
kjellander@webrtc.org4d8cd9d2011-11-09 11:24:14 +0000268} // namespace webrtc