* 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 &&
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)
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);
}
}
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)
{
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;
}
}
};
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;
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