Chromium Code Reviews| Index: webrtc/test/testsupport/fileutils.cc |
| diff --git a/webrtc/test/testsupport/fileutils.cc b/webrtc/test/testsupport/fileutils.cc |
| index 02ef06580bbcec4203d46ac1c23f42d4e279ebd6..56e2f84a787542970880cb28f33b965c428b824f 100644 |
| --- a/webrtc/test/testsupport/fileutils.cc |
| +++ b/webrtc/test/testsupport/fileutils.cc |
| @@ -24,6 +24,7 @@ |
| #include "webrtc/base/win32.h" |
| #define GET_CURRENT_DIR _getcwd |
| #else |
| +#include <dirent.h> |
| #include <unistd.h> |
| #define GET_CURRENT_DIR getcwd |
| @@ -40,6 +41,7 @@ |
| #include <memory> |
| +#include "webrtc/base/checks.h" |
| #include "webrtc/typedefs.h" // For architecture defines |
| namespace webrtc { |
| @@ -126,7 +128,7 @@ std::string WorkingDir() { |
| return kRootDirName; |
| } |
| -#else // WEBRTC_ANDROID |
| +#else // WEBRTC_ANDROID |
| std::string ProjectRootPath() { |
| #if defined(WEBRTC_IOS) |
| @@ -212,6 +214,48 @@ std::string TempFilename(const std::string &dir, const std::string &prefix) { |
| #endif |
| } |
| +bool ReadDirectory(std::string path, std::vector<std::string>* output) { |
| + RTC_DCHECK(output); |
| + if (path.length() == 0 || !DirExists(path)) return false; |
|
nisse-webrtc
2017/05/29 10:56:39
On second look, I think the DirExists check here i
AleBzk
2017/05/29 11:09:29
Done.
|
| + output->clear(); |
|
nisse-webrtc
2017/05/29 10:56:39
The comment in the header file just says that stri
AleBzk
2017/05/29 11:09:29
Done.
|
| +#if defined(WEBRTC_WIN) |
| + // Append separator character if needed. |
| + if (path.back() != '\\') path += '\\'; |
| + |
| + // Init. |
| + WIN32_FIND_DATA data; |
| + HANDLE handle = ::FindFirstFile(rtc::ToUtf16(path + '*').c_str(), &data); |
| + if (handle == INVALID_HANDLE_VALUE) return false; |
| + |
| + // Populate output. |
| + do { |
| + const std::string name = rtc::ToUtf8(data.cFileName); |
| + if (name != "." && name != "..") output->emplace_back(path + name); |
| + } while (::FindNextFile(handle, &data) == TRUE); |
| + |
| + // Release resources. |
| + if (handle != INVALID_HANDLE_VALUE) ::FindClose(handle); |
| +#else |
| + // Append separator character if needed. |
| + if (path.back() != '/') path += '/'; |
| + |
| + // Init. |
| + DIR *dir = ::opendir(path.c_str()); |
| + if (dir == nullptr) return false; |
| + |
| + // Populate output. |
| + while (struct dirent* dirent = readdir(dir)) { |
| + RTC_DCHECK(dirent); |
|
nisse-webrtc
2017/05/29 10:56:39
A DCHECK on the loop condition right above seems u
AleBzk
2017/05/29 11:09:29
Done.
|
| + const std::string& name = dirent->d_name; |
| + if (name != "." && name != "..") output->emplace_back(path + name); |
| + } |
| + |
| + // Release resources. |
| + closedir(dir); |
| +#endif |
| + return true; |
| +} |
| + |
| bool CreateDir(const std::string& directory_name) { |
| struct stat path_info = {0}; |
| // Check if the path exists already: |