#include <pkgmgr_installer_info.h>
#include <sys/smack.h>
#include <sys/prctl.h>
+#include <openssl/sha.h>
#include <cstdlib>
#include <cstring>
#include <fstream>
#include <sstream>
#include <map>
+#include <iomanip>
#include "log.h"
#include "utils.h"
return iCompare(fileName, fileName.size()-7, ".ni", 0, 3);
}
-bool cmdOptionExists(char** begin, char** end, const std::string& option)
-{
- return std::find(begin, end, option) != end;
-}
-
-std::string readSelfPath()
-{
- char buff[PATH_MAX];
- ssize_t len = ::readlink("/proc/self/exe", buff, sizeof(buff)-1);
- if (len != -1) {
- buff[len] = '\0';
- return std::string(buff);
- }
-
- return "";
-}
-
std::string concatPath(const std::string& path1, const std::string& path2)
{
std::string path(path1);
}
}
-std::string absolutePath(const std::string& path)
+std::string getAbsolutePath(const std::string& path)
{
std::string absPath;
char *realPath = realpath(path.c_str(), NULL);
return absPath;
}
-int getRootPath(std::string pkgId, std::string& rootPath)
+std::string getRootPath(const std::string& pkgId)
{
int ret = 0;
char *path = 0;
uid_t uid = 0;
+ std::string rootPath;
if (pkgmgr_installer_info_get_target_uid(&uid) < 0) {
_ERR("Failed to get UID");
- return -1;
+ return rootPath;
}
pkgmgrinfo_pkginfo_h handle;
if (uid == 0) {
ret = pkgmgrinfo_pkginfo_get_pkginfo(pkgId.c_str(), &handle);
if (ret != PMINFO_R_OK) {
- return -1;
+ return rootPath;
}
} else {
ret = pkgmgrinfo_pkginfo_get_usr_pkginfo(pkgId.c_str(), uid, &handle);
if (ret != PMINFO_R_OK) {
- return -1;
+ return rootPath;
}
}
ret = pkgmgrinfo_pkginfo_get_root_path(handle, &path);
if (ret != PMINFO_R_OK) {
pkgmgrinfo_pkginfo_destroy_pkginfo(handle);
- return -1;
+ return rootPath;
}
rootPath = path;
pkgmgrinfo_pkginfo_destroy_pkginfo(handle);
- return 0;
+
+ return rootPath;
}
-int getExecName(std::string pkgId, std::string& execName)
+std::string getExecName(const std::string& pkgId)
{
char *exec = NULL;
char *appId = 0;
+ std::string execName;
pkgmgrinfo_pkginfo_h pkg_handle;
int ret = pkgmgrinfo_pkginfo_get_pkginfo(pkgId.c_str(), &pkg_handle);
if (ret != PMINFO_R_OK) {
- return -1;
+ return execName;
}
ret = pkgmgrinfo_pkginfo_get_mainappid(pkg_handle, &appId);
if (ret != PMINFO_R_OK) {
pkgmgrinfo_pkginfo_destroy_pkginfo(pkg_handle);
- return -1;
+ return execName;
}
pkgmgrinfo_appinfo_h app_handle;
ret = pkgmgrinfo_appinfo_get_appinfo(appId, &app_handle);
if (ret != PMINFO_R_OK) {
pkgmgrinfo_pkginfo_destroy_pkginfo(pkg_handle);
- return -1;
+ return execName;
}
ret = pkgmgrinfo_appinfo_get_exec(app_handle, &exec);
if (ret != PMINFO_R_OK) {
pkgmgrinfo_appinfo_destroy_appinfo(app_handle);
pkgmgrinfo_pkginfo_destroy_pkginfo(pkg_handle);
- return -1;
+ return execName;
}
execName = std::string(exec).substr(std::string(exec).rfind('/') + 1);
pkgmgrinfo_appinfo_destroy_appinfo(app_handle);
pkgmgrinfo_pkginfo_destroy_pkginfo(pkg_handle);
- return 0;
+
+ return execName;
+}
+
+std::string getAppType(const std::string& pkgId)
+{
+ char *appId = 0;
+ char *type = 0;
+ std::string appType;
+
+ pkgmgrinfo_pkginfo_h pkg_handle;
+ int ret = pkgmgrinfo_pkginfo_get_pkginfo(pkgId.c_str(), &pkg_handle);
+ if (ret != PMINFO_R_OK) {
+ return appType;
+ }
+ ret = pkgmgrinfo_pkginfo_get_mainappid(pkg_handle, &appId);
+ if (ret != PMINFO_R_OK) {
+ pkgmgrinfo_pkginfo_destroy_pkginfo(pkg_handle);
+ return appType;
+ }
+
+ pkgmgrinfo_appinfo_h app_handle;
+ ret = pkgmgrinfo_appinfo_get_appinfo(appId, &app_handle);
+ if (ret != PMINFO_R_OK) {
+ pkgmgrinfo_pkginfo_destroy_pkginfo(pkg_handle);
+ return appType;
+ }
+ ret = pkgmgrinfo_appinfo_get_apptype(app_handle, &type);
+ if (ret != PMINFO_R_OK) {
+ pkgmgrinfo_appinfo_destroy_appinfo(app_handle);
+ pkgmgrinfo_pkginfo_destroy_pkginfo(pkg_handle);
+ return appType;
+ }
+ appType = type;
+
+ pkgmgrinfo_appinfo_destroy_appinfo(app_handle);
+ pkgmgrinfo_pkginfo_destroy_pkginfo(pkg_handle);
+
+ return appType;
}
-int getMetadataValue(std::string pkgId, std::string metadataKey, std::string& metadataValue)
+std::string getMetadataValue(const std::string& pkgId, const std::string& key)
{
char *value = NULL;
char *appId = 0;
+ std::string metadataValue;
pkgmgrinfo_pkginfo_h pkg_handle;
int ret = pkgmgrinfo_pkginfo_get_pkginfo(pkgId.c_str(), &pkg_handle);
if (ret != PMINFO_R_OK) {
- return -1;
+ return metadataValue;
}
ret = pkgmgrinfo_pkginfo_get_mainappid(pkg_handle, &appId);
if (ret != PMINFO_R_OK) {
pkgmgrinfo_pkginfo_destroy_pkginfo(pkg_handle);
- return -1;
+ return metadataValue;
}
pkgmgrinfo_appinfo_h app_handle;
ret = pkgmgrinfo_appinfo_get_appinfo(appId, &app_handle);
if (ret != PMINFO_R_OK) {
pkgmgrinfo_pkginfo_destroy_pkginfo(pkg_handle);
- return -1;
+ return metadataValue;
}
- ret = pkgmgrinfo_appinfo_get_metadata_value(app_handle, metadataKey.c_str(), &value);
+ ret = pkgmgrinfo_appinfo_get_metadata_value(app_handle, key.c_str(), &value);
if (ret != PMINFO_R_OK) {
pkgmgrinfo_appinfo_destroy_appinfo(app_handle);
pkgmgrinfo_pkginfo_destroy_pkginfo(pkg_handle);
- //Does not return error because the metadata key may not exist.
- return 0;
+ return metadataValue;
}
metadataValue = std::string(value);
pkgmgrinfo_appinfo_destroy_appinfo(app_handle);
pkgmgrinfo_pkginfo_destroy_pkginfo(pkg_handle);
- return 0;
+
+ return metadataValue;
}
-std::string baseName(const std::string& path)
+std::string getBaseName(const std::string& path)
{
auto pos = path.find_last_of(PATH_SEPARATOR);
if (pos != std::string::npos)
return path;
}
-std::string replaceAll(const std::string &str, const std::string &pattern, const std::string &replace)
+std::string replaceAll(const std::string& str, const std::string& pattern, const std::string& replace)
{
std::string result = str;
std::string::size_type pos = 0;
return result;
}
-bool isFileExist(const std::string& path)
+bool isFile(const std::string& path)
{
struct stat sb;
- return stat(path.c_str(), &sb) == 0;
+ return lstat(path.c_str(), &sb) == 0;
}
-uintptr_t getFileSize(const std::string& path)
+bool isDirectory(const std::string& path)
{
struct stat sb;
-
- if (stat(path.c_str(), &sb) == 0) {
- return sb.st_size;
+ if (stat(path.c_str(), &sb) != 0) {
+ return false;
+ }
+ if (sb.st_mode & S_IFDIR) {
+ return true;
+ } else {
+ return false;
}
-
- return 0;
}
-std::string stripNiDLL(const std::string& path)
+std::string getAssemblyNameFromPath(const std::string& path)
{
- std::string niPath(path);
- if (path.size() < 5) return niPath;
- if (!strncasecmp(path.c_str() + path.size() - 4, ".dll", 4))
- niPath = path.substr(0, path.size()-4);
- else if (!strncasecmp(path.c_str() + path.size() - 4, ".exe", 4))
- niPath = path.substr(0, path.size()-4);
+ std::string ret(getFileName(path));
+
+ if (ret.find_last_of(".") == std::string::npos)
+ return ret;
+ ret.erase(ret.find_last_of("."));
- if (!strncasecmp(niPath.c_str() + niPath.size() - 3, ".ni", 3))
- return niPath.substr(0, niPath.size()-3);
+ if (ret.size() > 3 && std::equal(ret.begin() + ret.size() - 3, ret.end(), ".ni"))
+ ret.erase(ret.size() - 3);
- return niPath;
+ return ret;
}
-void assembliesInDirectory(const std::vector<std::string>& directories, std::string& tpaList)
+void addAssembliesFromDirectories(const std::vector<std::string>& directories, std::string& list)
{
- std::map<std::string, std::string> assemblyList;
- std::map<std::string, std::string> tmpList;
-
- 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, path));
- if (ret.second == false) {
- if (isNativeImage(path))
- tmpList[dllName] = path;
+ std::vector<std::string> assems;
+ std::unordered_map<std::string, std::string> assemPaths;
+
+ auto reader = [&assems, &assemPaths](const std::string& path, const std::string& filename) {
+ if (isManagedAssembly(filename) || isNativeImage(filename)) {
+ std::string assem = getAssemblyNameFromPath(filename);
+
+ if (assemPaths.count(assem) == 0) {
+ assems.push_back(assem);
+ assemPaths[assem] = path;
+ } else if (isManagedAssembly(assemPaths[assem]) && isNativeImage(filename)) {
+ // Update only if a native image is found in the same directory.
+ // For example, if we have two directories = { X, Y } where X contains A.dll and
+ // Y contains both A.dll and A.ni.dll, always A.dll in X will be used.
+ if (getBaseName(assemPaths[assem]).compare(getBaseName(path)) == 0)
+ assemPaths[assem] = path;
}
}
};
+ for (auto& directory : directories)
+ scanFilesInDirectory(directory, reader, 0);
- for (auto directory : directories) {
- scanFilesInDir(directory.c_str(), reader, 1);
- // merge scaned dll list to tpa list.
- // if the dll is already exist in the list, that is skipped.
- assemblyList.insert(tmpList.begin(), tmpList.end());
- }
+ if (!list.empty() && list.back() != ':')
+ list.push_back(':');
- std::map<std::string, std::string>::iterator it;
- for (it = assemblyList.begin(); it != assemblyList.end(); it++)
- tpaList += it->second + ':';
+ for (auto& assem : assems)
+ list += assemPaths[assem] + ":";
- if (tpaList.back() == ':')
- tpaList.pop_back();
+ if (list.back() == ':')
+ list.pop_back();
}
-void scanFilesInDir(const std::string& directory, FileReader reader, unsigned int depth)
+void scanFilesInDirectory(const std::string& directory, FileReader reader, unsigned int depth)
{
DIR *dir;
struct dirent* entry;
bool isDir;
- if (strstr(directory.c_str(), TAC_SYMLINK_SUB_DIR) != NULL)
- return;
-
dir = opendir(directory.c_str());
if (dir == nullptr)
case DT_DIR:
isDir = true;
break;
+ // symlink is added to the list even if there is no original file.
+ // It used to remove broken symlinks related to TAC
case DT_LNK:
+ break;
case DT_UNKNOWN:
- struct stat sb;
- if (stat(path.c_str(), &sb) == -1)
- continue;
-
- if (S_ISREG(sb.st_mode) || S_ISDIR(sb.st_mode))
- break;
+ continue;
default:
continue;
}
if (!isDir)
reader(path, entry->d_name);
- else if (depth > 1 && strcmp(entry->d_name, ".") && strcmp(entry->d_name, ".."))
+ else if (depth > 0 && strcmp(entry->d_name, ".") && strcmp(entry->d_name, ".."))
innerDirectories.push_back(path);
}
- if (depth != 0)
+ if (depth > 0)
for (auto& d : innerDirectories)
- scanFilesInDir(d.c_str(), reader, depth - 1);
+ scanFilesInDirectory(d, reader, depth - 1);
closedir(dir);
}
}
// change owner and groups for symbolic link.
- if (!stat(fromPath.c_str(), &info)) {
+ if (!lstat(fromPath.c_str(), &info)) {
if (lchown(toPath.c_str(), info.st_uid, info.st_gid) == -1)
fprintf(stderr, "Failed to change owner and group name\n");
}
}
}
-static bool setOwnership(const bf::path& path, uid_t uid, gid_t gid) {
+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 true;
}
-static bool setDirPermissions(const bf::path& path, bf::perms permissions) {
+static bool setDirPermissions(const bf::path& path, bf::perms permissions)
+{
bs::error_code error;
bf::permissions(path, permissions, error);
if (error) {
return true;
}
-static bool setDirOwnershipAndPermissions(const bf::path& path, bf::perms permissions, uid_t uid, gid_t gid) {
+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;
return true;
}
-static bool copyOwnershipAndPermissions(const bf::path& path, const bf::path& path2) {
+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;
return true;
}
-bool createDir(const bf::path& path) {
+bool createDir(const bf::path& path)
+{
if (bf::exists(path)) {
return true;
}
return true;
}
-bool copyDir(const bf::path& path1, const bf::path& path2, FSFlag flags) {
+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)) {
return true;
}
-bool copyFile(const bf::path& path1, const bf::path& path2) {
+bool copyFile(const bf::path& path1, const bf::path& path2)
+{
bs::error_code error;
if (!bf::exists(path1)) {
return false;
return true;
}
-bool moveFile(const bf::path& path1, const bf::path& path2) {
+bool moveFile(const bf::path& path1, const bf::path& path2)
+{
if (!bf::exists(path1) || bf::exists(path2)) {
return false;
}
return true;
}
-bool removeFile(const bf::path& path) {
+bool removeFile(const bf::path& path)
+{
if (!bf::exists(path)) {
return true;
}
return true;
}
-bool removeAll(const bf::path& path) {
+bool removeAll(const bf::path& path)
+{
if (!exists(path)) {
return true;
}
return true;
}
-void setCmdName(const char* name)
+void setCmdName(const std::string& name)
{
#define PRC_NAME_LENGTH 16
char processName[PRC_NAME_LENGTH] = {0, };
- if (name == NULL || *name == '\0') {
+ if (name.empty())
return;
- }
memset(processName, '\0', PRC_NAME_LENGTH);
- snprintf(processName, PRC_NAME_LENGTH, "%s", name);
- prctl(PR_SET_NAME, processName);
+ snprintf(processName, PRC_NAME_LENGTH, "%s", name.c_str());
+ prctl(PR_SET_NAME, processName);
+}
+
+std::string getFileName(const std::string& path)
+{
+ std::string ret(path);
+ size_t index = ret.find_last_of(PATH_SEPARATOR);
+ return index == std::string::npos ? ret : ret.substr(index + 1);
}
-char* getFileNameFromPath(char* path)
+std::string SHA256(const std::string& path)
{
- char* fileName = strrchr(path, PATH_SEPARATOR);
- if (fileName != NULL && *fileName != '\0') {
- return ++fileName;
+ std::string output = "";
+ FILE *file = fopen(path.c_str(), "rb");
+ if (!file) {
+ return output;
}
- return NULL;
-}
+ unsigned char hash[SHA256_DIGEST_LENGTH];
+ SHA256_CTX sha256;
+ SHA256_Init(&sha256);
+ int bytesRead = 0;
+ const int bufSize = 32768;
+ char *buffer = (char*)malloc(bufSize);
+ if (!buffer) {
+ fclose(file);
+ return output;
+ }
+
+ while ((bytesRead = fread(buffer, 1, bufSize, file))) {
+ SHA256_Update(&sha256, buffer, bytesRead);
+ }
+ SHA256_Final(hash, &sha256);
+
+ std::stringstream ss;
+ for (int i = 0; i < SHA256_DIGEST_LENGTH; i++) {
+ ss << std::hex << std::setw(2) << std::setfill('0') << (int)hash[i];
+ }
+ output = ss.str();
+
+ fclose(file);
+ free(buffer);
+ return output;
+}
\ No newline at end of file