Add comments for added function in utils.h
[platform/core/dotnet/launcher.git] / NativeLauncher / util / utils.cc
index d4142eb..5e15b43 100644 (file)
  * limitations under the License.
  */
 
-#include <stdio.h>
 #include <dirent.h>
+#include <fcntl.h>
 #include <sys/stat.h>
 #include <unistd.h>
 #include <limits.h>
 #include <strings.h>
-#include <pthread.h>
 
 #include <cstdlib>
 #include <cstring>
 #include <vector>
 #include <iterator>
 #include <sstream>
+#include <map>
 
-#include "utils.h"
 #include "log.h"
+#include "utils.h"
+#include "path_manager.h"
 
-static int pfd[2];
-static pthread_t loggingThread;
-
-bool iCompare(const std::string& a, const std::string& b)
-{
-       return a.length() == b.length() &&
-               std::equal(b.begin(), b.end(), a.begin(),
-                       [](unsigned char a, unsigned char b)
-                       { return std::tolower(a) == std::tolower(b); });
-}
-
-bool iCompare(const std::string& a, int aOffset, const std::string& b, int bOffset, int length)
+static bool iCompare(const std::string& a, int aOffset, const std::string& b, int bOffset, int length)
 {
        return static_cast<int>(a.length()) - length >= aOffset &&
                static_cast<int>(b.length()) - length >= bOffset &&
@@ -55,8 +45,9 @@ bool iCompare(const std::string& a, int aOffset, const std::string& b, int bOffs
 
 bool isManagedAssembly(const std::string& fileName)
 {
-       return iCompare(fileName, fileName.size()-4, ".dll", 0, 4) ||
-               iCompare(fileName, fileName.size()-4, ".exe", 0, 4);
+       return (iCompare(fileName, fileName.size()-4, ".dll", 0, 4) ||
+                       iCompare(fileName, fileName.size()-4, ".exe", 0, 4)) &&
+                       !isNativeImage(fileName);
 }
 
 bool isNativeImage(const std::string& fileName)
@@ -89,13 +80,13 @@ std::string concatPath(const std::string& path1, const std::string& path2)
        return path;
 }
 
-void appendPath(std::string& path1, const std::string& path2)
+void splitPath(const std::string& path, std::vector<std::string>& out)
 {
-       if (path1.back() == PATH_SEPARATOR) {
-               path1.append(path2);
-       } else {
-               path1 += PATH_SEPARATOR;
-               path1.append(path2);
+       std::istringstream ss(path);
+       std::string token;
+
+       while (std::getline(ss, token, ':')) {
+               out.push_back(token);
        }
 }
 
@@ -119,55 +110,11 @@ std::string baseName(const std::string& path)
        return path;
 }
 
-bool endWithIgnoreCase(const std::string& str1, const std::string& str2, std::string& fileName)
-{
-       std::string::size_type len1 = str1.length();
-       std::string::size_type len2 = str2.length();
-       if (len2 > len1)
-               return false;
-
-       int i = 0;
-       bool result = std::all_of(str1.cend() - len2, str1.end(),
-                               [&i, &str2] (char x) {
-                                       return std::tolower(x) == std::tolower(str2[i++]);
-                               });
-       if (result)
-               fileName = str1.substr(0, len1 - len2);
-
-       return result;
-}
-
-bool fileNotExist(const std::string& path)
+bool isFileExist(const std::string& path)
 {
        struct stat sb;
-       return stat(path.c_str(), &sb) != 0;
-}
-
-#ifdef NOT_USE_FUNCTION
-static bool extCheckAndGetFileNameIfExist(const std::string& dir, const std::string& ext, struct dirent* entry, std::string& fileName)
-{
-       std::string fName(entry->d_name);
-       if (fName.length() < ext.length() ||
-                       fHame.compare(fName.length() - ext.length(), ext.length(), ext) != 0) {
-               return false;
-       }
-
-       std::string fullName = concatPath(dir, entry->d_name);
-       switch (entry->d_type) {
-               case DT_REG: break;
-               case DT_LNK:
-               case DT_UNKNOWN:
-                       if (fileNotExist(fullName))
-                               return false;
-               default:
-                       return false;
-       }
-
-       fileName = fullName;
-
-       return true;
+       return stat(path.c_str(), &sb) == 0;
 }
-#endif
 
 std::string stripNiDLL(const std::string& path)
 {
@@ -184,57 +131,19 @@ std::string stripNiDLL(const std::string& path)
        return niPath;
 }
 
-std::string joinStrings(const std::vector<std::string>& strings, const char* const delimeter)
-{
-       switch (strings.size()) {
-               case 0:
-                       return "";
-               case 1:
-                       return strings[0];
-               default:
-                       std::ostringstream os;
-                       std::copy(strings.begin(), strings.end()-1, std::ostream_iterator<std::string>(os, delimeter));
-                       os << *strings.rbegin();
-                       return os.str();
-       }
-}
-
-struct AssemblyFile {
-       std::string noExt;
-       std::string ext;
-};
-
-bool operator == (const AssemblyFile& lhs, const AssemblyFile& rhs)
-{
-       return lhs.noExt == rhs.noExt && lhs.ext == rhs.ext;
-}
-
-namespace std {
-       template<>
-       struct hash<AssemblyFile> {
-               std::size_t operator () (const AssemblyFile& f) const {
-                       const std::size_t h1 = std::hash<std::string>{}(f.noExt);
-                       const std::size_t h2 = std::hash<std::string>{}(f.ext);
-
-                       return h1 ^ (h2 << 1);
-               }
-       };
-}
-
 void assembliesInDirectory(const std::vector<std::string>& directories, std::string& tpaList)
 {
        std::map<std::string, std::string> assemblyList;
        std::map<std::string, std::string> tmpList;
 
-       auto reader = [&assemblyList, &tmpList] (const char* path, const char* name) {
-               std::string pathStr(path);
-               if (isManagedAssembly(pathStr)) {
+       auto reader = [&assemblyList, &tmpList] (const std::string& path, const char* name) {
+               if (isManagedAssembly(path) || isNativeImage(path)) {
                        std::string dllName = stripNiDLL(name);
                        std::pair<std::map<std::string, std::string>::iterator, bool> ret;
-                       ret = tmpList.insert(std::pair<std::string, std::string>(dllName, pathStr));
+                       ret = tmpList.insert(std::pair<std::string, std::string>(dllName, path));
                        if (ret.second == false) {
-                               if (isNativeImage(pathStr))
-                                       tmpList[dllName] = pathStr;
+                               if (isNativeImage(path))
+                                       tmpList[dllName] = path;
                        }
                }
        };
@@ -254,13 +163,16 @@ void assembliesInDirectory(const std::vector<std::string>& directories, std::str
                tpaList.pop_back();
 }
 
-void scanFilesInDir(const char* directory, FileReader reader, unsigned int depth)
+void scanFilesInDir(const std::string& directory, FileReader reader, unsigned int depth)
 {
        DIR *dir;
        struct dirent* entry;
        bool isDir;
 
-       dir = opendir(directory);
+       if (strstr(directory.c_str(), "TAC.Release") != NULL)
+               return; // skip nitool --regen-all-app (--r2r)
+
+       dir = opendir(directory.c_str());
 
        if (dir == nullptr)
                return;
@@ -287,74 +199,220 @@ void scanFilesInDir(const char* directory, FileReader reader, unsigned int depth
                                continue;
                }
                if (!isDir)
-                       reader(path.c_str(), entry->d_name);
+                       reader(path, entry->d_name);
                else if (depth > 1 && strcmp(entry->d_name, ".") && strcmp(entry->d_name, ".."))
                        innerDirectories.push_back(path);
        }
 
        if (depth != 0)
                for (auto& d : innerDirectories)
-                       scanFilesInDir(d.c_str(), reader, depth-1);
+                       scanFilesInDir(d.c_str(), reader, depth - 1);
 
        closedir(dir);
 }
 
-static void *stdlog(void*)
-{
-    ssize_t readSize;
-    char buf[1024];
-
-    while ((readSize = read(pfd[0], buf, sizeof buf - 1)) > 0) {
-        if (buf[readSize - 1] == '\n') {
-            --readSize;
-        }
-
-        buf[readSize] = 0;
-
-        _INFO("%s", buf);
-    }
+static bool setOwnership(const bf::path& path, uid_t uid, gid_t gid) {
+       int fd = open(path.c_str(), O_RDONLY);
+       if (fd < 0) {
+               _ERR("Can't open directory: %s", path.c_str());
+               return false;
+       }
+       int ret = fchown(fd, uid, gid);
+       close(fd);
+       if (ret != 0) {
+               _ERR("Failed to change owner of: %s", path.c_str());
+               return false;
+       }
+       return true;
+}
 
-    return 0;
+static bool setDirPermissions(const bf::path& path, bf::perms permissions) {
+       bs::error_code error;
+       bf::permissions(path, permissions, error);
+       if (error) {
+               _ERR("Failed to set permissions for directory: %s, %s", path.c_str(), error.message().c_str());
+               return false;
+       }
+       return true;
 }
 
-int runLoggingThread() { // run this function to redirect your output to android log
-    if (setvbuf(stdout, NULL, _IOLBF, 0) < 0) {
-               _DBG("fail to make stdout line-buffered");
-               return -1;
-    }
+static bool setDirOwnershipAndPermissions(const bf::path& path, bf::perms permissions, uid_t uid, gid_t gid) {
+       if (!setOwnership(path, uid, gid)) {
+               _ERR("Failed to change owner: %s, (uid: %d, gid: %d)", path.c_str(), uid, gid);
+               return false;
+       }
+       if (!setDirPermissions(path, permissions)) {
+               _ERR("Failed to change permission: %s, (%d)", path.c_str(), permissions);
+               return false;
+       }
+       return true;
+}
 
-    if (setvbuf(stderr, NULL, _IONBF, 0) < 0) {
-               _DBG("make stderr unbuffered");
-               return -1;
+static bool copyOwnershipAndPermissions(const bf::path& path, const bf::path& path2) {
+       if (!bf::exists(path)) {
+               _ERR("Failed to copy ownership and permissions from %s to %s", path.c_str(), path2.c_str());
+               return false;
+       }
+       bf::perms permissions = bf::status(path).permissions();
+       struct stat stats;
+       if (stat(path.c_str(), &stats) != 0) {
+               return false;
+       }
+       if (!setDirOwnershipAndPermissions(path2, permissions, stats.st_uid, stats.st_gid)) {
+               _ERR("Failed to copy ownership and permissions from %s to %s", path.c_str(), path2.c_str());
+               return false;
        }
+       return true;
+}
 
-    /* create the pipe and redirect stdout and stderr */
-    if (pipe(pfd) < 0) {
-               _DBG("fail to create pipe for logging");
-               return -1;
-    }
+bool createDir(const bf::path& path) {
+       if (bf::exists(path)) {
+               return true;
+       }
+       bs::error_code error;
+       bf::create_directories(path, error);
+       if (error) {
+               _ERR("Failed to create directory: %s", error.message().c_str());
+               return false;
+       }
+       return true;
+}
 
-    if (dup2(pfd[1], fileno(stdout)) == -1) {
-               _DBG("fail to duplicate fd to stdout");
-               return -1;
-    }
+bool copyDir(const bf::path& path1, const bf::path& path2, FSFlag flags) {
+       try {
+               // Check whether the function call is valid
+               if (!bf::exists(path1) || !bf::is_directory(path1)) {
+                       _ERR("Source directory %s does not exist or is not a directory", path1.c_str());
+                       return false;
+               }
+               if (!bf::exists(path2)) {
+                       // Create the destination directory
+                       if (!createDir(path2)) {
+                               _ERR("Unable to create destination directory %s", path2.c_str());
+                               return false;
+                       }
+                       if (flags & FS_PRESERVE_OWNERSHIP_AND_PERMISSIONS) {
+                               copyOwnershipAndPermissions(path1, path2);
+                       }
+               } else {
+                       if (!(flags & (FS_MERGE_SKIP | FS_MERGE_OVERWRITE))) {
+                               _ERR("Destination directory %s already exists", path2.c_str());
+                               return false;
+                       }
+                       if (flags & (FS_MERGE_OVERWRITE | FS_PRESERVE_OWNERSHIP_AND_PERMISSIONS)) {
+                               copyOwnershipAndPermissions(path1, path2);
+                       }
+               }
+       } catch (const bf::filesystem_error& error) {
+               _ERR("Failed to copy directory: %s", error.what());
+               return false;
+       }
 
-    if (dup2(pfd[1], fileno(stderr)) == -1) {
-               _DBG("fail to duplicate fd to stderr");
-               return -1;
+       // Iterate through the source directory
+       for (bf::directory_iterator file(path1); file != bf::directory_iterator(); ++file) {
+               try {
+                       bf::path current(file->path());
+                       bf::path target = path2 / current.filename();
+                       if (bf::is_symlink(symlink_status(current))) {
+                               if ((flags & (FS_MERGE_SKIP | FS_MERGE_OVERWRITE)) && bf::exists(target)) {
+                                       continue;
+                               }
+                               bs::error_code error;
+                               bf::copy_symlink(current, target, error);
+                               if (error) {
+                                       _ERR("Failed to copy symlink: %s, %s", current.c_str(), error.message().c_str());
+                                       return false;
+                               }
+                       } else if (bf::is_directory(current)) {
+                               // Found directory: Recursion
+                               if (!copyDir(current, target, flags)) {
+                                       return false;
+                               }
+                       } else {
+                               if ((flags & FS_MERGE_SKIP) && bf::exists(target)) {
+                                       continue;
+                               }
+                               bf::path destination = target;
+                               if (flags & FS_COMMIT_COPY_FILE) {
+                                       destination = bf::unique_path(target.parent_path() / "%%%%-%%%%-%%%%-%%%%");
+                               }
+                               if (flags & FS_MERGE_OVERWRITE) {
+                                       bf::copy_file(current, destination, bf::copy_option::overwrite_if_exists);
+                               } else {
+                                       bf::copy_file(current, destination);
+                               }
+                               if (flags & FS_PRESERVE_OWNERSHIP_AND_PERMISSIONS) {
+                                       copyOwnershipAndPermissions(current, destination);
+                               }
+                               if (flags & FS_COMMIT_COPY_FILE) {
+                                       if (flags & FS_MERGE_OVERWRITE) {
+                                               bf::remove(target);
+                                       }
+                                       bf::rename(destination, target);
+                               }
+                       }
+               } catch (const bf::filesystem_error& error) {
+                       _ERR("Failed to copy directory: %s", error.what());
+                       return false;
+               }
        }
+       return true;
+}
 
-    /* spawn the logging thread */
-    if (pthread_create(&loggingThread, 0, stdlog, 0) != 0) {
-               _DBG("fail to create pthread");
-        return -1;
-    }
+bool copyFile(const bf::path& path1, const bf::path& path2) {
+       bs::error_code error;
+       bf::copy_file(path1, path2, bf::copy_option::overwrite_if_exists, error);
+       if (error) {
+               _ERR("copy file %s due to error [%s]", path1.c_str(), error.message().c_str());
+               return false;
+       }
+       return true;
+}
 
-    if (pthread_detach(loggingThread) != 0) {
-               _DBG("fail to detach pthread");
-               return -1;
+bool moveFile(const bf::path& path1, const bf::path& path2) {
+       if (bf::exists(path2)) {
+               return false;
        }
+       bs::error_code error;
+       bf::rename(path1, path2, error);
+       if (error) {
+               _ERR("Cannot move file: %s. Will copy/remove... with error [%s]", path1.c_str(), error.message().c_str());
+               bf::copy_file(path1, path2, bf::copy_option::overwrite_if_exists, error);
+               if (error) {
+                       _ERR("Cannot copy file %s due to error [%s]", path1.c_str(), error.message().c_str());
+                       return false;
+               }
+               bf::remove_all(path1, error);
+               if (error) {
+                       _ERR("Cannot remove old file when coping: %s with error [%s]", path1.c_str(), error.message().c_str());
+                       return false;
+               }
+       }
+       return true;
+}
 
-    return 0;
+bool removeFile(const bf::path& path) {
+       if (!bf::exists(path)) {
+               return true;
+       }
+       bs::error_code error;
+       bf::remove(path, error);
+       if (error) {
+               _ERR("Cannot remove: %s, %s", path.c_str(), error.message().c_str());
+               return false;
+       }
+       return true;
 }
 
+bool removeAll(const bf::path& path) {
+       if (!exists(path)) {
+               return true;
+       }
+       bs::error_code error;
+       bf::remove_all(path, error);
+       if (error) {
+               _ERR("Cannot remove: %s, %s", path.c_str(), error.message().c_str());
+               return false;
+       }
+       return true;
+}
\ No newline at end of file