|
|
|
@ -37,14 +37,24 @@
|
|
|
|
|
#define S_ISDIR(mode) (((mode)&S_IFMT) == S_IFDIR)
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
static void* dlsym(void* handle, const char* symbol_name) {
|
|
|
|
|
static void *dlsym(void *handle, const char *symbol_name) {
|
|
|
|
|
FARPROC found_symbol;
|
|
|
|
|
found_symbol = GetProcAddress((HMODULE)handle, symbol_name);
|
|
|
|
|
|
|
|
|
|
if (found_symbol == NULL) {
|
|
|
|
|
throw std::runtime_error(std::string(symbol_name) + " not found.");
|
|
|
|
|
}
|
|
|
|
|
return reinterpret_cast<void*>(found_symbol);
|
|
|
|
|
return reinterpret_cast<void *>(found_symbol);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static void *dlopen(const char *filename, int flag) {
|
|
|
|
|
std::string file_name(filename);
|
|
|
|
|
std::replace(file_name.begin(), file_name.end(), '/', '\\');
|
|
|
|
|
HMODULE hModule = LoadLibrary(file_name);
|
|
|
|
|
if (!hModule) {
|
|
|
|
|
throw std::runtime_error(file_name + " not found.");
|
|
|
|
|
}
|
|
|
|
|
return reinterpret_cast<void *>(hModule);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#endif // !_WIN32
|
|
|
|
@ -85,3 +95,49 @@ static bool PathExists(const std::string &path) {
|
|
|
|
|
#endif // !_WIN32
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// TODO(yuyang18): If the functions below are needed by other files, move them
|
|
|
|
|
// to paddle::filesystem namespace.
|
|
|
|
|
#if !defined(_WIN32)
|
|
|
|
|
constexpr char kSEP = '/';
|
|
|
|
|
#else
|
|
|
|
|
constexpr char kSEP = '\\';
|
|
|
|
|
#endif // _WIN32
|
|
|
|
|
|
|
|
|
|
static bool FileExists(const std::string &filepath) {
|
|
|
|
|
#if !defined(_WIN32)
|
|
|
|
|
struct stat buffer;
|
|
|
|
|
return (stat(filepath.c_str(), &buffer) == 0);
|
|
|
|
|
#else
|
|
|
|
|
struct _stat buffer;
|
|
|
|
|
return (_stat(filepath.c_str(), &buffer) == 0);
|
|
|
|
|
#endif // !_WIN32
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static std::string DirName(const std::string &filepath) {
|
|
|
|
|
auto pos = filepath.rfind(kSEP);
|
|
|
|
|
if (pos == std::string::npos) {
|
|
|
|
|
return "";
|
|
|
|
|
}
|
|
|
|
|
return filepath.substr(0, pos);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static void MkDir(const char *path) {
|
|
|
|
|
#if !defined(_WIN32)
|
|
|
|
|
if (mkdir(path, 0755)) {
|
|
|
|
|
PADDLE_ENFORCE_EQ(errno, EEXIST, "%s mkdir failed!", path);
|
|
|
|
|
}
|
|
|
|
|
#else
|
|
|
|
|
CreateDirectory(path, NULL);
|
|
|
|
|
auto errorno = GetLastError();
|
|
|
|
|
PADDLE_ENFORCE_EQ(errorno, ERROR_ALREADY_EXISTS, "%s mkdir failed!", path);
|
|
|
|
|
#endif // !_WIN32
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static void MkDirRecursively(const char *fullpath) {
|
|
|
|
|
if (*fullpath == '\0') return; // empty string
|
|
|
|
|
if (FileExists(fullpath)) return;
|
|
|
|
|
|
|
|
|
|
MkDirRecursively(DirName(fullpath).c_str());
|
|
|
|
|
MkDir(fullpath);
|
|
|
|
|
}
|
|
|
|
|