123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978 |
- /*
- Minetest
- Copyright (C) 2013 celeron55, Perttu Ahola <celeron55@gmail.com>
- This program is free software; you can redistribute it and/or modify
- it under the terms of the GNU Lesser General Public License as published by
- the Free Software Foundation; either version 2.1 of the License, or
- (at your option) any later version.
- This program is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- GNU Lesser General Public License for more details.
- You should have received a copy of the GNU Lesser General Public License along
- with this program; if not, write to the Free Software Foundation, Inc.,
- 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
- */
- #include "filesys.h"
- #include "util/string.h"
- #include <iostream>
- #include <cstdio>
- #include <cstdlib>
- #include <cstring>
- #include <cerrno>
- #include <fstream>
- #include "log.h"
- #include "config.h"
- #include "porting.h"
- #ifndef SERVER
- #include "irr_ptr.h"
- #include <IFileArchive.h>
- #include <IFileSystem.h>
- #endif
- #ifdef __linux__
- #include <fcntl.h>
- #include <sys/ioctl.h>
- #ifndef FICLONE
- #define FICLONE _IOW(0x94, 9, int)
- #endif
- #endif
- namespace fs
- {
- #ifdef _WIN32
- /***********
- * Windows *
- ***********/
- #include <windows.h>
- #include <shlwapi.h>
- #include <io.h>
- #include <direct.h>
- std::vector<DirListNode> GetDirListing(const std::string &pathstring)
- {
- std::vector<DirListNode> listing;
- WIN32_FIND_DATA FindFileData;
- HANDLE hFind = INVALID_HANDLE_VALUE;
- DWORD dwError;
- std::string dirSpec = pathstring + "\\*";
- // Find the first file in the directory.
- hFind = FindFirstFile(dirSpec.c_str(), &FindFileData);
- if (hFind == INVALID_HANDLE_VALUE) {
- dwError = GetLastError();
- if (dwError != ERROR_FILE_NOT_FOUND && dwError != ERROR_PATH_NOT_FOUND) {
- errorstream << "GetDirListing: FindFirstFile error."
- << " Error is " << dwError << std::endl;
- }
- } else {
- // NOTE:
- // Be very sure to not include '..' in the results, it will
- // result in an epic failure when deleting stuff.
- DirListNode node;
- node.name = FindFileData.cFileName;
- node.dir = FindFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
- if (node.name != "." && node.name != "..")
- listing.push_back(node);
- // List all the other files in the directory.
- while (FindNextFile(hFind, &FindFileData) != 0) {
- DirListNode node;
- node.name = FindFileData.cFileName;
- node.dir = FindFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
- if(node.name != "." && node.name != "..")
- listing.push_back(node);
- }
- dwError = GetLastError();
- FindClose(hFind);
- if (dwError != ERROR_NO_MORE_FILES) {
- errorstream << "GetDirListing: FindNextFile error."
- << " Error is " << dwError << std::endl;
- listing.clear();
- return listing;
- }
- }
- return listing;
- }
- bool CreateDir(const std::string &path)
- {
- bool r = CreateDirectory(path.c_str(), NULL);
- if(r == true)
- return true;
- if(GetLastError() == ERROR_ALREADY_EXISTS)
- return true;
- return false;
- }
- bool PathExists(const std::string &path)
- {
- return (GetFileAttributes(path.c_str()) != INVALID_FILE_ATTRIBUTES);
- }
- bool IsPathAbsolute(const std::string &path)
- {
- return !PathIsRelative(path.c_str());
- }
- bool IsDir(const std::string &path)
- {
- DWORD attr = GetFileAttributes(path.c_str());
- return (attr != INVALID_FILE_ATTRIBUTES &&
- (attr & FILE_ATTRIBUTE_DIRECTORY));
- }
- bool IsExecutable(const std::string &path)
- {
- DWORD type;
- return GetBinaryType(path.c_str(), &type) != 0;
- }
- bool IsDirDelimiter(char c)
- {
- return c == '/' || c == '\\';
- }
- bool RecursiveDelete(const std::string &path)
- {
- infostream << "Recursively deleting \"" << path << "\"" << std::endl;
- if (!IsDir(path)) {
- infostream << "RecursiveDelete: Deleting file " << path << std::endl;
- if (!DeleteFile(path.c_str())) {
- errorstream << "RecursiveDelete: Failed to delete file "
- << path << std::endl;
- return false;
- }
- return true;
- }
- infostream << "RecursiveDelete: Deleting content of directory "
- << path << std::endl;
- std::vector<DirListNode> content = GetDirListing(path);
- for (const DirListNode &n: content) {
- std::string fullpath = path + DIR_DELIM + n.name;
- if (!RecursiveDelete(fullpath)) {
- errorstream << "RecursiveDelete: Failed to recurse to "
- << fullpath << std::endl;
- return false;
- }
- }
- infostream << "RecursiveDelete: Deleting directory " << path << std::endl;
- if (!RemoveDirectory(path.c_str())) {
- errorstream << "Failed to recursively delete directory "
- << path << std::endl;
- return false;
- }
- return true;
- }
- bool DeleteSingleFileOrEmptyDirectory(const std::string &path)
- {
- DWORD attr = GetFileAttributes(path.c_str());
- bool is_directory = (attr != INVALID_FILE_ATTRIBUTES &&
- (attr & FILE_ATTRIBUTE_DIRECTORY));
- if(!is_directory)
- {
- bool did = DeleteFile(path.c_str());
- return did;
- }
- else
- {
- bool did = RemoveDirectory(path.c_str());
- return did;
- }
- }
- std::string TempPath()
- {
- DWORD bufsize = GetTempPath(0, NULL);
- if(bufsize == 0){
- errorstream<<"GetTempPath failed, error = "<<GetLastError()<<std::endl;
- return "";
- }
- std::string buf;
- buf.resize(bufsize);
- DWORD len = GetTempPath(bufsize, &buf[0]);
- if(len == 0 || len > bufsize){
- errorstream<<"GetTempPath failed, error = "<<GetLastError()<<std::endl;
- return "";
- }
- buf.resize(len);
- return buf;
- }
- std::string CreateTempFile()
- {
- std::string path = TempPath() + DIR_DELIM "MT_XXXXXX";
- _mktemp_s(&path[0], path.size() + 1); // modifies path
- HANDLE file = CreateFile(path.c_str(), GENERIC_WRITE, 0, nullptr,
- CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, nullptr);
- if (file == INVALID_HANDLE_VALUE)
- return "";
- CloseHandle(file);
- return path;
- }
- bool CopyFileContents(const std::string &source, const std::string &target)
- {
- BOOL ok = CopyFileEx(source.c_str(), target.c_str(), nullptr, nullptr,
- nullptr, COPY_FILE_ALLOW_DECRYPTED_DESTINATION);
- if (!ok) {
- errorstream << "copying " << source << " to " << target
- << " failed: " << GetLastError() << std::endl;
- return false;
- }
- // docs: "File attributes for the existing file are copied to the new file."
- // This is not our intention so get rid of unwanted attributes:
- DWORD attr = GetFileAttributes(target.c_str());
- if (attr == INVALID_FILE_ATTRIBUTES) {
- errorstream << target << ": file disappeared after copy" << std::endl;
- return false;
- }
- attr &= ~(FILE_ATTRIBUTE_READONLY | FILE_ATTRIBUTE_HIDDEN);
- SetFileAttributes(target.c_str(), attr);
- tracestream << "copied " << source << " to " << target
- << " using CopyFileEx" << std::endl;
- return true;
- }
- #else
- /*********
- * POSIX *
- *********/
- #include <sys/types.h>
- #include <dirent.h>
- #include <sys/stat.h>
- #include <sys/wait.h>
- #include <unistd.h>
- std::vector<DirListNode> GetDirListing(const std::string &pathstring)
- {
- std::vector<DirListNode> listing;
- DIR *dp;
- struct dirent *dirp;
- if((dp = opendir(pathstring.c_str())) == NULL) {
- //infostream<<"Error("<<errno<<") opening "<<pathstring<<std::endl;
- return listing;
- }
- while ((dirp = readdir(dp)) != NULL) {
- // NOTE:
- // Be very sure to not include '..' in the results, it will
- // result in an epic failure when deleting stuff.
- if(strcmp(dirp->d_name, ".") == 0 || strcmp(dirp->d_name, "..") == 0)
- continue;
- DirListNode node;
- node.name = dirp->d_name;
- int isdir = -1; // -1 means unknown
- /*
- POSIX doesn't define d_type member of struct dirent and
- certain filesystems on glibc/Linux will only return
- DT_UNKNOWN for the d_type member.
- Also we don't know whether symlinks are directories or not.
- */
- #ifdef _DIRENT_HAVE_D_TYPE
- if(dirp->d_type != DT_UNKNOWN && dirp->d_type != DT_LNK)
- isdir = (dirp->d_type == DT_DIR);
- #endif /* _DIRENT_HAVE_D_TYPE */
- /*
- Was d_type DT_UNKNOWN, DT_LNK or nonexistent?
- If so, try stat().
- */
- if(isdir == -1) {
- struct stat statbuf{};
- if (stat((pathstring + "/" + node.name).c_str(), &statbuf))
- continue;
- isdir = ((statbuf.st_mode & S_IFDIR) == S_IFDIR);
- }
- node.dir = isdir;
- listing.push_back(node);
- }
- closedir(dp);
- return listing;
- }
- bool CreateDir(const std::string &path)
- {
- int r = mkdir(path.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
- if (r == 0) {
- return true;
- }
- // If already exists, return true
- if (errno == EEXIST)
- return true;
- return false;
- }
- bool PathExists(const std::string &path)
- {
- struct stat st{};
- return (stat(path.c_str(),&st) == 0);
- }
- bool IsPathAbsolute(const std::string &path)
- {
- return path[0] == '/';
- }
- bool IsDir(const std::string &path)
- {
- struct stat statbuf{};
- if(stat(path.c_str(), &statbuf))
- return false; // Actually error; but certainly not a directory
- return ((statbuf.st_mode & S_IFDIR) == S_IFDIR);
- }
- bool IsExecutable(const std::string &path)
- {
- return access(path.c_str(), X_OK) == 0;
- }
- bool IsDirDelimiter(char c)
- {
- return c == '/';
- }
- bool RecursiveDelete(const std::string &path)
- {
- /*
- Execute the 'rm' command directly, by fork() and execve()
- */
- infostream<<"Removing \""<<path<<"\""<<std::endl;
- pid_t child_pid = fork();
- if(child_pid == 0)
- {
- // Child
- const char *argv[4] = {
- #ifdef __ANDROID__
- "/system/bin/rm",
- #else
- "/bin/rm",
- #endif
- "-rf",
- path.c_str(),
- NULL
- };
- verbosestream<<"Executing '"<<argv[0]<<"' '"<<argv[1]<<"' '"
- <<argv[2]<<"'"<<std::endl;
- execv(argv[0], const_cast<char**>(argv));
- // Execv shouldn't return. Failed.
- _exit(1);
- }
- else
- {
- // Parent
- int child_status;
- pid_t tpid;
- do{
- tpid = wait(&child_status);
- }while(tpid != child_pid);
- return (child_status == 0);
- }
- }
- bool DeleteSingleFileOrEmptyDirectory(const std::string &path)
- {
- if (IsDir(path)) {
- bool did = (rmdir(path.c_str()) == 0);
- if (!did)
- errorstream << "rmdir errno: " << errno << ": " << strerror(errno)
- << std::endl;
- return did;
- }
- bool did = (unlink(path.c_str()) == 0);
- if (!did)
- errorstream << "unlink errno: " << errno << ": " << strerror(errno)
- << std::endl;
- return did;
- }
- std::string TempPath()
- {
- /*
- Should the environment variables TMPDIR, TMP and TEMP
- and the macro P_tmpdir (if defined by stdio.h) be checked
- before falling back on /tmp?
- Probably not, because this function is intended to be
- compatible with lua's os.tmpname which under the default
- configuration hardcodes mkstemp("/tmp/lua_XXXXXX").
- */
- #ifdef __ANDROID__
- return porting::path_cache;
- #else
- return DIR_DELIM "tmp";
- #endif
- }
- std::string CreateTempFile()
- {
- std::string path = TempPath() + DIR_DELIM "MT_XXXXXX";
- int fd = mkstemp(&path[0]); // modifies path
- if (fd == -1)
- return "";
- close(fd);
- return path;
- }
- namespace {
- struct FileDeleter {
- void operator()(FILE *stream) {
- fclose(stream);
- }
- };
- typedef std::unique_ptr<FILE, FileDeleter> FileUniquePtr;
- }
- bool CopyFileContents(const std::string &source, const std::string &target)
- {
- FileUniquePtr sourcefile, targetfile;
- #ifdef __linux__
- // Try to clone using Copy-on-Write (CoW). This is instant but supported
- // only by some filesystems.
- int srcfd, tgtfd;
- srcfd = open(source.c_str(), O_RDONLY);
- if (srcfd == -1) {
- errorstream << source << ": can't open for reading: "
- << strerror(errno) << std::endl;
- return false;
- }
- tgtfd = open(target.c_str(), O_WRONLY | O_CREAT | O_TRUNC, 0644);
- if (tgtfd == -1) {
- errorstream << target << ": can't open for writing: "
- << strerror(errno) << std::endl;
- close(srcfd);
- return false;
- }
- if (ioctl(tgtfd, FICLONE, srcfd) == 0) {
- tracestream << "copied " << source << " to " << target
- << " using FICLONE" << std::endl;
- close(srcfd);
- close(tgtfd);
- return true;
- }
- // fallback to normal copy, but no need to reopen the files
- sourcefile.reset(fdopen(srcfd, "rb"));
- targetfile.reset(fdopen(tgtfd, "wb"));
- goto fallback;
- #endif
- sourcefile.reset(fopen(source.c_str(), "rb"));
- targetfile.reset(fopen(target.c_str(), "wb"));
- fallback:
- if (!sourcefile) {
- errorstream << source << ": can't open for reading: "
- << strerror(errno) << std::endl;
- return false;
- }
- if (!targetfile) {
- errorstream << target << ": can't open for writing: "
- << strerror(errno) << std::endl;
- return false;
- }
- size_t total = 0;
- bool done = false;
- char readbuffer[BUFSIZ];
- while (!done) {
- size_t readbytes = fread(readbuffer, 1,
- sizeof(readbuffer), sourcefile.get());
- total += readbytes;
- if (ferror(sourcefile.get())) {
- errorstream << source << ": IO error: "
- << strerror(errno) << std::endl;
- return false;
- }
- if (readbytes > 0)
- fwrite(readbuffer, 1, readbytes, targetfile.get());
- if (feof(sourcefile.get())) {
- // flush destination file to catch write errors (e.g. disk full)
- fflush(targetfile.get());
- done = true;
- }
- if (ferror(targetfile.get())) {
- errorstream << target << ": IO error: "
- << strerror(errno) << std::endl;
- return false;
- }
- }
- tracestream << "copied " << total << " bytes from "
- << source << " to " << target << std::endl;
- return true;
- }
- #endif
- /****************************
- * portable implementations *
- ****************************/
- void GetRecursiveDirs(std::vector<std::string> &dirs, const std::string &dir)
- {
- static const std::set<char> chars_to_ignore = { '_', '.' };
- if (dir.empty() || !IsDir(dir))
- return;
- dirs.push_back(dir);
- fs::GetRecursiveSubPaths(dir, dirs, false, chars_to_ignore);
- }
- std::vector<std::string> GetRecursiveDirs(const std::string &dir)
- {
- std::vector<std::string> result;
- GetRecursiveDirs(result, dir);
- return result;
- }
- void GetRecursiveSubPaths(const std::string &path,
- std::vector<std::string> &dst,
- bool list_files,
- const std::set<char> &ignore)
- {
- std::vector<DirListNode> content = GetDirListing(path);
- for (const auto &n : content) {
- std::string fullpath = path + DIR_DELIM + n.name;
- if (ignore.count(n.name[0]))
- continue;
- if (list_files || n.dir)
- dst.push_back(fullpath);
- if (n.dir)
- GetRecursiveSubPaths(fullpath, dst, list_files, ignore);
- }
- }
- bool RecursiveDeleteContent(const std::string &path)
- {
- infostream<<"Removing content of \""<<path<<"\""<<std::endl;
- std::vector<DirListNode> list = GetDirListing(path);
- for (const DirListNode &dln : list) {
- if(trim(dln.name) == "." || trim(dln.name) == "..")
- continue;
- std::string childpath = path + DIR_DELIM + dln.name;
- bool r = RecursiveDelete(childpath);
- if(!r) {
- errorstream << "Removing \"" << childpath << "\" failed" << std::endl;
- return false;
- }
- }
- return true;
- }
- bool CreateAllDirs(const std::string &path)
- {
- std::vector<std::string> tocreate;
- std::string basepath = path;
- while(!PathExists(basepath))
- {
- tocreate.push_back(basepath);
- basepath = RemoveLastPathComponent(basepath);
- if(basepath.empty())
- break;
- }
- for(int i=tocreate.size()-1;i>=0;i--)
- if(!CreateDir(tocreate[i]))
- return false;
- return true;
- }
- bool CopyDir(const std::string &source, const std::string &target)
- {
- if(PathExists(source)){
- if(!PathExists(target)){
- fs::CreateAllDirs(target);
- }
- bool retval = true;
- std::vector<DirListNode> content = fs::GetDirListing(source);
- for (const auto &dln : content) {
- std::string sourcechild = source + DIR_DELIM + dln.name;
- std::string targetchild = target + DIR_DELIM + dln.name;
- if(dln.dir){
- if(!fs::CopyDir(sourcechild, targetchild)){
- retval = false;
- }
- }
- else {
- if(!fs::CopyFileContents(sourcechild, targetchild)){
- retval = false;
- }
- }
- }
- return retval;
- }
- return false;
- }
- bool MoveDir(const std::string &source, const std::string &target)
- {
- infostream << "Moving \"" << source << "\" to \"" << target << "\"" << std::endl;
- // If target exists as empty folder delete, otherwise error
- if (fs::PathExists(target)) {
- if (rmdir(target.c_str()) != 0) {
- errorstream << "MoveDir: target \"" << target
- << "\" exists as file or non-empty folder" << std::endl;
- return false;
- }
- }
- // Try renaming first which is instant
- if (fs::Rename(source, target))
- return true;
- infostream << "MoveDir: rename not possible, will copy instead" << std::endl;
- bool retval = fs::CopyDir(source, target);
- if (retval)
- retval &= fs::RecursiveDelete(source);
- return retval;
- }
- bool PathStartsWith(const std::string &path, const std::string &prefix)
- {
- size_t pathsize = path.size();
- size_t pathpos = 0;
- size_t prefixsize = prefix.size();
- size_t prefixpos = 0;
- for(;;){
- bool delim1 = pathpos == pathsize
- || IsDirDelimiter(path[pathpos]);
- bool delim2 = prefixpos == prefixsize
- || IsDirDelimiter(prefix[prefixpos]);
- if(delim1 != delim2)
- return false;
- if(delim1){
- while(pathpos < pathsize &&
- IsDirDelimiter(path[pathpos]))
- ++pathpos;
- while(prefixpos < prefixsize &&
- IsDirDelimiter(prefix[prefixpos]))
- ++prefixpos;
- if(prefixpos == prefixsize)
- return true;
- if(pathpos == pathsize)
- return false;
- }
- else{
- size_t len = 0;
- do{
- char pathchar = path[pathpos+len];
- char prefixchar = prefix[prefixpos+len];
- if(FILESYS_CASE_INSENSITIVE){
- pathchar = tolower(pathchar);
- prefixchar = tolower(prefixchar);
- }
- if(pathchar != prefixchar)
- return false;
- ++len;
- } while(pathpos+len < pathsize
- && !IsDirDelimiter(path[pathpos+len])
- && prefixpos+len < prefixsize
- && !IsDirDelimiter(
- prefix[prefixpos+len]));
- pathpos += len;
- prefixpos += len;
- }
- }
- }
- std::string RemoveLastPathComponent(const std::string &path,
- std::string *removed, int count)
- {
- if(removed)
- removed->clear();
- size_t remaining = path.size();
- for(int i = 0; i < count; ++i){
- // strip a dir delimiter
- while(remaining != 0 && IsDirDelimiter(path[remaining-1]))
- remaining--;
- // strip a path component
- size_t component_end = remaining;
- while(remaining != 0 && !IsDirDelimiter(path[remaining-1]))
- remaining--;
- size_t component_start = remaining;
- // strip a dir delimiter
- while(remaining != 0 && IsDirDelimiter(path[remaining-1]))
- remaining--;
- if(removed){
- std::string component = path.substr(component_start,
- component_end - component_start);
- if(i)
- *removed = component + DIR_DELIM + *removed;
- else
- *removed = component;
- }
- }
- return path.substr(0, remaining);
- }
- std::string RemoveRelativePathComponents(std::string path)
- {
- size_t pos = path.size();
- size_t dotdot_count = 0;
- while (pos != 0) {
- size_t component_with_delim_end = pos;
- // skip a dir delimiter
- while (pos != 0 && IsDirDelimiter(path[pos-1]))
- pos--;
- // strip a path component
- size_t component_end = pos;
- while (pos != 0 && !IsDirDelimiter(path[pos-1]))
- pos--;
- size_t component_start = pos;
- std::string component = path.substr(component_start,
- component_end - component_start);
- bool remove_this_component = false;
- if (component == ".") {
- remove_this_component = true;
- } else if (component == "..") {
- remove_this_component = true;
- dotdot_count += 1;
- } else if (dotdot_count != 0) {
- remove_this_component = true;
- dotdot_count -= 1;
- }
- if (remove_this_component) {
- while (pos != 0 && IsDirDelimiter(path[pos-1]))
- pos--;
- if (component_start == 0) {
- // We need to remove the delemiter too
- path = path.substr(component_with_delim_end, std::string::npos);
- } else {
- path = path.substr(0, pos) + DIR_DELIM +
- path.substr(component_with_delim_end, std::string::npos);
- }
- if (pos > 0)
- pos++;
- }
- }
- if (dotdot_count > 0)
- return "";
- // remove trailing dir delimiters
- pos = path.size();
- while (pos != 0 && IsDirDelimiter(path[pos-1]))
- pos--;
- return path.substr(0, pos);
- }
- std::string AbsolutePath(const std::string &path)
- {
- #ifdef _WIN32
- char *abs_path = _fullpath(NULL, path.c_str(), MAX_PATH);
- #else
- char *abs_path = realpath(path.c_str(), NULL);
- #endif
- if (!abs_path) return "";
- std::string abs_path_str(abs_path);
- free(abs_path);
- return abs_path_str;
- }
- const char *GetFilenameFromPath(const char *path)
- {
- const char *filename = strrchr(path, DIR_DELIM_CHAR);
- // Consistent with IsDirDelimiter this function handles '/' too
- if (DIR_DELIM_CHAR != '/') {
- const char *tmp = strrchr(path, '/');
- if (tmp && tmp > filename)
- filename = tmp;
- }
- return filename ? filename + 1 : path;
- }
- bool safeWriteToFile(const std::string &path, std::string_view content)
- {
- std::string tmp_file = path + ".~mt";
- // Write to a tmp file
- bool tmp_success = false;
- #ifdef _WIN32
- // We've observed behavior suggesting that the MSVC implementation of std::ofstream::flush doesn't
- // actually flush, so we use win32 APIs.
- HANDLE tmp_handle = CreateFile(
- tmp_file.c_str(), GENERIC_WRITE, 0, nullptr, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, nullptr);
- if (tmp_handle == INVALID_HANDLE_VALUE) {
- return false;
- }
- DWORD bytes_written;
- tmp_success = (WriteFile(tmp_handle, content.data(), content.size(), &bytes_written, nullptr) &&
- FlushFileBuffers(tmp_handle));
- CloseHandle(tmp_handle);
- #else
- std::ofstream os(tmp_file.c_str(), std::ios::binary);
- if (!os.good()) {
- return false;
- }
- os << content;
- os.flush();
- os.close();
- tmp_success = !os.fail();
- #endif
- if (!tmp_success) {
- remove(tmp_file.c_str());
- return false;
- }
- bool rename_success = false;
- // Move the finished temporary file over the real file
- #ifdef _WIN32
- // When creating the file, it can cause Windows Search indexer, virus scanners and other apps
- // to query the file. This can make the move file call below fail.
- // We retry up to 5 times, with a 1ms sleep between, before we consider the whole operation failed
- for (int attempt = 0; attempt < 5; attempt++) {
- rename_success = MoveFileEx(tmp_file.c_str(), path.c_str(),
- MOVEFILE_REPLACE_EXISTING | MOVEFILE_WRITE_THROUGH);
- if (rename_success)
- break;
- sleep_ms(1);
- }
- #else
- // On POSIX compliant systems rename() is specified to be able to swap the
- // file in place of the destination file, making this a truly error-proof
- // transaction.
- rename_success = rename(tmp_file.c_str(), path.c_str()) == 0;
- #endif
- if (!rename_success) {
- warningstream << "Failed to write to file: " << path.c_str() << std::endl;
- // Remove the temporary file because moving it over the target file
- // failed.
- remove(tmp_file.c_str());
- return false;
- }
- return true;
- }
- #ifndef SERVER
- bool extractZipFile(io::IFileSystem *fs, const char *filename, const std::string &destination)
- {
- // Be careful here not to touch the global file hierarchy in Irrlicht
- // since this function needs to be thread-safe!
- io::IArchiveLoader *zip_loader = nullptr;
- for (u32 i = 0; i < fs->getArchiveLoaderCount(); i++) {
- if (fs->getArchiveLoader(i)->isALoadableFileFormat(io::EFAT_ZIP)) {
- zip_loader = fs->getArchiveLoader(i);
- break;
- }
- }
- if (!zip_loader) {
- warningstream << "fs::extractZipFile(): Irrlicht said it doesn't support ZIPs." << std::endl;
- return false;
- }
- irr_ptr<io::IFileArchive> opened_zip(zip_loader->createArchive(filename, false, false));
- const io::IFileList* files_in_zip = opened_zip->getFileList();
- for (u32 i = 0; i < files_in_zip->getFileCount(); i++) {
- std::string fullpath = destination + DIR_DELIM;
- fullpath += files_in_zip->getFullFileName(i).c_str();
- std::string fullpath_dir = fs::RemoveLastPathComponent(fullpath);
- if (files_in_zip->isDirectory(i))
- continue; // ignore, we create dirs as necessary
- if (!fs::PathExists(fullpath_dir) && !fs::CreateAllDirs(fullpath_dir))
- return false;
- irr_ptr<io::IReadFile> toread(opened_zip->createAndOpenFile(i));
- std::ofstream os(fullpath.c_str(), std::ios::binary);
- if (!os.good())
- return false;
- char buffer[4096];
- long total_read = 0;
- while (total_read < toread->getSize()) {
- long bytes_read = toread->read(buffer, sizeof(buffer));
- bool error = true;
- if (bytes_read != 0) {
- os.write(buffer, bytes_read);
- error = os.fail();
- }
- if (error) {
- os.close();
- remove(fullpath.c_str());
- return false;
- }
- total_read += bytes_read;
- }
- }
- return true;
- }
- #endif
- bool ReadFile(const std::string &path, std::string &out)
- {
- std::ifstream is(path, std::ios::binary | std::ios::ate);
- if (!is.good()) {
- return false;
- }
- auto size = is.tellg();
- out.resize(size);
- is.seekg(0);
- is.read(&out[0], size);
- return !is.fail();
- }
- bool Rename(const std::string &from, const std::string &to)
- {
- return rename(from.c_str(), to.c_str()) == 0;
- }
- } // namespace fs
|