NativeLauncher/tool/ni_common.cc

   1 /*
   2  * Copyright (c) 2016 Samsung Electronics Co., Ltd All Rights Reserved
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the License);
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an AS IS BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include <pkgmgr-info.h>
  18 #include <pkgmgr_installer_info.h>
  19 #include <aul.h>
  20
  21 #include "log.h"
  22 #include "utils.h"
  23 #include "pkgmgr_parser_plugin_interface.h"
  24
  25 #include <wait.h>
  26 #include <dirent.h>
  27 #include <sys/stat.h>
  28
  29 #include <algorithm>
  30 #include <string>
  31 #include <fstream>
  32
  33 #include <pwd.h>
  34 #include <grp.h>
  35 #include <unistd.h>
  36 #include <string.h>
  37 #include <sqlite3.h>
  38
  39 #include "ni_common.h"
  40 #include "db_manager.h"
  41 #include "tac_common.h"
  42 #include "path_manager.h"
  43 #include "plugin_manager.h"
  44
  45 #ifdef  LOG_TAG
  46 #undef  LOG_TAG
  47 #endif
  48 #define LOG_TAG "DOTNET_INSTALLER_PLUGIN"
  49
  50 #ifndef CROSSGEN_PATH
  51 #error "CROSSGEN_PATH is missed"
  52 #endif
  53
  54 #define __XSTR(x) #x
  55 #define __STR(x) __XSTR(x)
  56 static const char* __NATIVE_LIB_DIR = __STR(NATIVE_LIB_DIR);
  57 static const char* __CROSSGEN_PATH = __STR(CROSSGEN_PATH);
  58 static const char* __DOTNET_DIR = __STR(DOTNET_DIR);
  59
  60 #ifdef UNIQUE_DEFAULT_BASE_ADDR_SUPPORT
  61 static const char* __SYSTEM_BASE_FILE = __STR(SYSTEM_BASE_FILE);
  62 #endif
  63
  64 #undef __STR
  65 #undef __XSTR
  66
  67 static int __interval = 0;
  68 static std::string __tpa;
  69
  70 static void waitInterval()
  71 {
  72         // by the recommand, ignore small value for performance.
  73         if (__interval > 10000) {
  74                 fprintf(stdout, "sleep %d usec\n", __interval);
  75                 usleep(__interval);
  76         }
  77 }
  78
  79 static std::string getNiFilePath(const std::string& dllPath)
  80 {
  81         size_t index = dllPath.find_last_of(".");
  82         if (index == std::string::npos) {
  83                 fprintf(stderr, "File doesnot contain extension. fail to get NI file name\n");
  84                 return "";
  85         }
  86         std::string fName = dllPath.substr(0, index);
  87         std::string fExt = dllPath.substr(index, dllPath.length());
  88
  89         // crossgen generate file with lower case extension only
  90         std::transform(fExt.begin(), fExt.end(), fExt.begin(), ::tolower);
  91         std::string niPath = fName + ".ni" + fExt;
  92
  93         return niPath;
  94 }
  95
  96 static std::string getAppNIPath(const std::string& niPath)
  97 {
  98         std::string fileName;
  99         std::string niDirPath;
 100         std::string prevPath;
 101
 102         size_t index = niPath.find_last_of("/");
 103         if (index != std::string::npos) {
 104                 prevPath = niPath.substr(0, index);
 105                 fileName = niPath.substr(index + 1, niPath.length());
 106         } else {
 107                 prevPath = ".";
 108                 fileName = niPath;
 109         }
 110
 111         niDirPath = concatPath(prevPath, APP_NI_SUB_DIR);
 112
 113         if (!isFileExist(niDirPath)) {
 114                 if (mkdir(niDirPath.c_str(), 0755) == 0) {
 115                         copySmackAndOwnership(prevPath, niDirPath);
 116                 } else {
 117                         fprintf(stderr, "Fail to create app ni directory (%s)\n", niDirPath.c_str());
 118                 }
 119         }
 120
 121         return concatPath(niDirPath, fileName);
 122 }
 123
 124 static bool niExist(const std::string& path)
 125 {
 126         std::string f = getNiFilePath(path);
 127         if (f.empty()) {
 128                 return false;
 129         }
 130
 131         if (isFileExist(f)) {
 132                 return true;
 133         }
 134
 135         // native image of System.Private.CoreLib.dll should have to overwrite
 136         // original file to support new coreclr
 137         if (path.find("System.Private.CoreLib.dll") != std::string::npos) {
 138                 std::string coreLibBackup = path + ".Backup";
 139                 if (isFileExist(coreLibBackup)) {
 140                         return true;
 141                 }
 142         }
 143
 144         return false;
 145 }
 146
 147 #ifdef UNIQUE_DEFAULT_BASE_ADDR_SUPPORT
 148 // Get next base address to be used for system ni image from file
 149 // __SYSTEM_BASE_FILE should be checked for existance before calling this function
 150 static uintptr_t getNextBaseAddrFromFile()
 151 {
 152         FILE *pFile = fopen(__SYSTEM_BASE_FILE, "r");
 153         if (pFile == NULL) {
 154                 fprintf(stderr, "Failed to open %s\n", __SYSTEM_BASE_FILE);
 155                 return 0;
 156         }
 157
 158         uintptr_t addr = 0;
 159         uintptr_t size = 0;
 160
 161         while (fscanf(pFile, "%u %u", &addr, &size) != EOF) {
 162         }
 163
 164         fclose(pFile);
 165
 166         return addr + size;
 167 }
 168
 169 // Get next base address to be used for system ni image
 170 static uintptr_t getNextBaseAddr()
 171 {
 172         uintptr_t baseAddr = 0;
 173
 174         if (!isFileExist(__SYSTEM_BASE_FILE)) {
 175                 // This is the starting address for all default base addresses
 176                 baseAddr = DEFAULT_BASE_ADDR_START;
 177         } else {
 178                 baseAddr = getNextBaseAddrFromFile();
 179
 180                 // Round to a multple of 64K (see ZapImage::CalculateZapBaseAddress in CoreCLR)
 181                 uintptr_t BASE_ADDRESS_ALIGNMENT = 0xffff;
 182                 baseAddr = (baseAddr + BASE_ADDRESS_ALIGNMENT) & ~BASE_ADDRESS_ALIGNMENT;
 183         }
 184
 185         return baseAddr;
 186 }
 187
 188 // Save base address of system ni image to file
 189 static void updateBaseAddrFile(const std::string &absNiPath, uintptr_t baseAddr)
 190 {
 191         uintptr_t niSize = getFileSize(absNiPath);
 192         if (niSize == 0) {
 193                 fprintf(stderr, "File %s doesn't exist\n", absNiPath.c_str());
 194                 return;
 195         }
 196
 197         // Write new entry to the file
 198         FILE *pFile = fopen(__SYSTEM_BASE_FILE, "a");
 199         if (pFile == NULL) {
 200                 fprintf(stderr, "Failed to open %s\n", __SYSTEM_BASE_FILE);
 201                 return;
 202         }
 203
 204         fprintf(pFile, "%u %u\n", baseAddr, niSize);
 205         fclose(pFile);
 206 }
 207
 208 // check if dll is listed in TPA
 209 static bool isTPADll(const std::string &dllPath)
 210 {
 211         std::string absDllPath = absolutePath(dllPath);
 212
 213         if (__tpa.find(absDllPath) != std::string::npos) {
 214                 return true;
 215         }
 216
 217         return false;
 218 }
 219 #endif
 220
 221 // baseAddr should be checked in file before getting here
 222 static ni_error_e crossgen(const std::string& dllPath, const std::string& appPath, DWORD flags)
 223 {
 224         if (!isFileExist(dllPath)) {
 225                 fprintf(stderr, "dll file is not exist : %s\n", dllPath.c_str());
 226                 return NI_ERROR_NO_SUCH_FILE;
 227         }
 228
 229         if (!isManagedAssembly(dllPath)) {
 230                 //fprintf(stderr, "Input file is not a dll file : %s\n", dllPath.c_str());
 231                 return NI_ERROR_INVALID_PARAMETER;
 232         }
 233
 234         if (niExist(dllPath)) {
 235                 fprintf(stderr, "Already ni file is exist for %s\n", dllPath.c_str());
 236                 return NI_ERROR_ALREADY_EXIST;
 237         }
 238
 239         std::string absDllPath = absolutePath(dllPath);
 240         std::string absNiPath = getNiFilePath(dllPath);
 241
 242
 243         if (absNiPath.empty()) {
 244                 fprintf(stderr, "Fail to get ni file name\n");
 245                 return NI_ERROR_UNKNOWN;
 246         }
 247
 248         bool isAppNI = flags & NI_FLAGS_APPNI;
 249         if (isAppNI && strstr(absNiPath.c_str(), __DOTNET_DIR) == NULL) {
 250                 absNiPath = getAppNIPath(absNiPath);
 251         }
 252
 253 #ifdef UNIQUE_DEFAULT_BASE_ADDR_SUPPORT
 254         uintptr_t baseAddr = 0;
 255
 256         if (isTPADll(dllPath)) {
 257                 baseAddr = getNextBaseAddr();
 258         }
 259 #endif
 260
 261         pid_t pid = fork();
 262         if (pid == -1)
 263                 return NI_ERROR_UNKNOWN;
 264
 265         if (pid > 0) {
 266                 int status;
 267                 waitpid(pid, &status, 0);
 268                 if (WIFEXITED(status)) {
 269                         // Do not use niExist() function to check whether ni file created or not.
 270                         // niEixst() return false for System.Private.Corelib.dll
 271                         if (isFileExist(absNiPath)) {
 272                                 copySmackAndOwnership(absDllPath, absNiPath);
 273                                 std::string absPdbPath = replaceAll(absDllPath, ".dll", ".pdb");
 274                                 std::string pdbFilePath = replaceAll(absNiPath, ".ni.dll", ".pdb");
 275                                 if (isFileExist(absPdbPath) && (absPdbPath != pdbFilePath)) {
 276                                         if (!copyFile(absPdbPath, pdbFilePath)) {
 277                                                 fprintf(stderr, "Failed to copy a .pdb file\n");
 278                                         } else {
 279                                                 copySmackAndOwnership(absPdbPath, pdbFilePath);
 280                                         }
 281                                 }
 282 #ifdef UNIQUE_DEFAULT_BASE_ADDR_SUPPORT
 283                                 if (baseAddr != 0) {
 284                                         updateBaseAddrFile(absNiPath, baseAddr);
 285                                 }
 286 #endif
 287                                 return NI_ERROR_NONE;
 288                         } else {
 289                                 fprintf(stderr, "Fail to create native image for %s\n", dllPath.c_str());
 290                                 return NI_ERROR_NO_SUCH_FILE;
 291                         }
 292                 }
 293         } else {
 294                 std::string jitPath = getRuntimeDir() + "/libclrjit.so";
 295                 std::vector<const char*> argv = {
 296                         __CROSSGEN_PATH,
 297                         "/nologo",
 298                         "/JITPath", jitPath.c_str()
 299                 };
 300
 301                 bool compat = flags & NI_FLAGS_COMPATIBILITY;
 302                 argv.push_back(compat ? "/Trusted_Platform_Assemblies" : "/r");
 303                 argv.push_back(__tpa.c_str());
 304
 305                 bool enableR2R = flags & NI_FLAGS_ENABLER2R;
 306                 if (!enableR2R) {
 307                         argv.push_back("/FragileNonVersionable");
 308                 }
 309
 310                 if (flags & NI_FLAGS_VERBOSE) {
 311                         argv.push_back("/verbose");
 312                 }
 313
 314                 if (flags & NI_FLAGS_INSTRUMENT) {
 315                         argv.push_back("/Tuning");
 316                 }
 317
 318 #ifdef UNIQUE_DEFAULT_BASE_ADDR_SUPPORT
 319                 if (baseAddr != 0) {
 320                         argv.push_back("/BaseAddress");
 321                         argv.push_back(std::to_string(baseAddr).c_str());
 322                 }
 323 #endif
 324
 325                 argv.push_back("/App_Paths");
 326                 std::string absAppPath;
 327                 if (!appPath.empty()) {
 328                         absAppPath = appPath;
 329                 } else {
 330                         absAppPath = baseName(absDllPath);
 331                 }
 332                 argv.push_back(absAppPath.c_str());
 333
 334                 argv.push_back("/out");
 335                 argv.push_back(absNiPath.c_str());
 336
 337                 argv.push_back(absDllPath.c_str());
 338                 argv.push_back(nullptr);
 339
 340                 fprintf(stdout, "+ %s (%s)\n", absDllPath.c_str(), enableR2R ? "R2R" : "FNV");
 341
 342                 execv(__CROSSGEN_PATH, const_cast<char* const*>(argv.data()));
 343                 exit(0);
 344         }
 345
 346         return NI_ERROR_NONE;
 347 }
 348
 349 // callback function of "pkgmgrinfo_appinfo_metadata_filter_foreach"
 350 static int appAotCb(pkgmgrinfo_appinfo_h handle, void *userData)
 351 {
 352         char *pkgId = NULL;
 353         int ret = 0;
 354         DWORD *pFlags = (DWORD*)userData;
 355
 356         ret = pkgmgrinfo_appinfo_get_pkgid(handle, &pkgId);
 357         if (ret != PMINFO_R_OK) {
 358                 fprintf(stderr, "Failed to get pkgid\n");
 359                 return -1;
 360         }
 361
 362         if (removeNiUnderPkgRoot(pkgId) != NI_ERROR_NONE) {
 363                 fprintf(stderr, "Failed to remove previous dlls from [%s]\n", pkgId);
 364                 return -1;
 365         }
 366
 367         if (resetTACPackage(pkgId) != TAC_ERROR_NONE) {
 368                 fprintf(stderr, "Failed to remove symlink for given package [%s]\n", pkgId);
 369                 return -1;
 370         }
 371
 372         // Regenerate ni files with R2R mode forcibiliy. (there is no way to now which option is used)
 373         if (createNiUnderPkgRoot(pkgId, *pFlags) != NI_ERROR_NONE) {
 374                 fprintf(stderr, "Failed to generate NI file [%s]\n", pkgId);
 375                 return -1;
 376         } else {
 377                 fprintf(stdout, "Complete make application to native image\n");
 378         }
 379
 380         if (createTACPkgRoot(pkgId, *pFlags) != NI_ERROR_NONE) {
 381                 fprintf(stderr, "Failed to generate symbolic link file [%s]\n", pkgId);
 382                 return -1;
 383         }else {
 384                 fprintf(stdout, "Complete make symbolic link file to tac\n");
 385         }
 386
 387         return 0;
 388 }
 389
 390 static bool isCoreLibPrepared(DWORD flags)
 391 {
 392         if (flags & NI_FLAGS_ENABLER2R) {
 393                 return true;
 394         }
 395
 396         std::string coreLibBackup = concatPath(getRuntimeDir(), "System.Private.CoreLib.dll.Backup");
 397         if (isFileExist(coreLibBackup)) {
 398                 return true;
 399         } else {
 400                 fprintf(stderr, "The native image of System.Private.CoreLib does not exist\n"
 401                                         "Run the command to create the native image\n"
 402                                         "# dotnettool --ni-dll /usr/share/dotnet.tizen/netcoreapp/System.Private.CoreLib.dll\n\n");
 403                 return false;
 404         }
 405 }
 406
 407 static void createCoreLibNI(DWORD flags)
 408 {
 409         std::string coreLib = concatPath(getRuntimeDir(), "System.Private.CoreLib.dll");
 410         std::string niCoreLib = concatPath(getRuntimeDir(), "System.Private.CoreLib.ni.dll");
 411         std::string coreLibBackup = concatPath(getRuntimeDir(), "System.Private.CoreLib.dll.Backup");
 412
 413         if (!isFileExist(coreLibBackup)) {
 414                 if (!crossgen(coreLib, std::string(), flags)) {
 415                         if (rename(coreLib.c_str(), coreLibBackup.c_str())) {
 416                                 fprintf(stderr, "Failed to rename System.Private.CoreLib.dll\n");
 417                         }
 418                         if (rename(niCoreLib.c_str(), coreLib.c_str())) {
 419                                 fprintf(stderr, "Failed to rename System.Private.CoreLib.ni.dll\n");
 420                         }
 421                 } else {
 422                         fprintf(stderr, "Failed to create native image for %s\n", coreLib.c_str());
 423                 }
 424         }
 425 }
 426
 427 ni_error_e initNICommon(NiCommonOption* option)
 428 {
 429 #if defined(__arm__)
 430         // get interval value
 431         const static std::string intervalFile = concatPath(__NATIVE_LIB_DIR, "crossgen_interval.txt");
 432         std::ifstream inFile(intervalFile);
 433         if (inFile) {
 434                 fprintf(stdout, "crossgen_interval.txt is found\n");
 435                 inFile >> __interval;
 436         }
 437
 438         if (initializePluginManager("normal")) {
 439                 fprintf(stderr, "Fail to initialize PluginManager\n");
 440                 return NI_ERROR_UNKNOWN;
 441         }
 442         if (initializePathManager(option->runtimeDir, option->tizenFXDir, option->extraDirs)) {
 443                 fprintf(stderr, "Fail to initialize PathManager\n");
 444                 return NI_ERROR_UNKNOWN;
 445         }
 446
 447         __tpa = getTPA();
 448
 449         return NI_ERROR_NONE;
 450 #else
 451         fprintf(stderr, "crossgen supports arm architecture only. skip ni file generation\n");
 452         return NI_ERROR_NOT_SUPPORTED;
 453 #endif
 454 }
 455
 456 void finalizeNICommon()
 457 {
 458         __interval = 0;
 459
 460         finalizePluginManager();
 461         finalizePathManager();
 462
 463         __tpa.clear();
 464 }
 465
 466
 467 ni_error_e createNiPlatform(DWORD flags)
 468 {
 469         createCoreLibNI(flags);
 470
 471         const std::string platformDirs[] = {getRuntimeDir(), getTizenFXDir()};
 472         return createNiUnderDirs(platformDirs, 2, flags);
 473 }
 474
 475 ni_error_e createNiDll(const std::string& dllPath, DWORD flags)
 476 {
 477         if (!isCoreLibPrepared(flags)) {
 478                 return NI_ERROR_CORE_NI_FILE;
 479         }
 480
 481         return crossgen(dllPath, std::string(), flags);
 482 }
 483
 484 void createNiUnderTAC(std::vector<std::string> nugets, DWORD flags)
 485 {
 486         std::string appPaths;
 487         for (auto& nuget : nugets) {
 488                 appPaths += concatPath(__DOTNET_DIR, nuget);
 489                 appPaths += ':';
 490         }
 491         if (appPaths.back() == ':') {
 492                 appPaths.pop_back();
 493         }
 494
 495         auto convert = [&appPaths, flags](const std::string& path, const std::string& filename) {
 496                 if (strstr(path.c_str(), TAC_SHA_256_INFO) != NULL)
 497                         return;
 498                 if (!crossgen(path, appPaths.c_str(), flags)) {
 499                         waitInterval();
 500                 }
 501         };
 502         for (auto& nuget : nugets) {
 503                 scanFilesInDirectory(concatPath(__DOTNET_DIR, nuget), convert, -1);
 504         }
 505 }
 506
 507 ni_error_e createTACPkgRoot(const std::string& pkgId, DWORD flags)
 508 {
 509         if (!isCoreLibPrepared(flags)) {
 510                 return NI_ERROR_CORE_NI_FILE;
 511         }
 512
 513         std::string pkgRoot;
 514         if (getRootPath(pkgId, pkgRoot) < 0) {
 515                 return NI_ERROR_INVALID_PACKAGE;
 516         }
 517
 518         std::string binDir = concatPath(pkgRoot, "bin");
 519         std::string libDir = concatPath(pkgRoot, "lib");
 520         std::string tacDir = concatPath(binDir, TAC_SYMLINK_SUB_DIR);
 521         std::string binNIDir = concatPath(binDir, APP_NI_SUB_DIR);
 522         std::string paths = binDir + ":" + libDir + ":" + tacDir;
 523         if (bf::exists(tacDir)) {
 524                 try {
 525                         for (auto& symlinkAssembly : bf::recursive_directory_iterator(tacDir)) {
 526                                 std::string symPath = symlinkAssembly.path().string();
 527                                 if (bf::is_symlink(symPath)) {
 528                                         if (!isNativeImage(symPath)) {
 529                                                 std::string originPath = bf::read_symlink(symPath).string();
 530                                                 std::string originNiPath = originPath.substr(0, originPath.rfind(".dll")) + ".ni.dll";
 531                                                 if (!bf::exists(originNiPath)) {
 532                                                         flags |= NI_FLAGS_APPNI;
 533                                                         if(crossgen(originPath, paths, flags) != NI_ERROR_NONE) {
 534                                                                 fprintf(stderr, "Failed to create NI file [%s]\n", originPath.c_str());
 535                                                                 return NI_ERROR_UNKNOWN;
 536                                                         }
 537                                                 }
 538                                                 std::string symNIPath = symPath.substr(0, symPath.rfind(".dll")) + ".ni.dll";
 539                                                 if (!bf::exists(symNIPath)) {
 540                                                         bf::create_symlink(originNiPath, symNIPath);
 541                                                         fprintf(stdout, "%s symbolic link file generated successfully.\n", symNIPath.c_str());
 542                                                         copySmackAndOwnership(tacDir.c_str(), symNIPath.c_str(), true);
 543
 544                                                         std::string niFileName = symNIPath.substr(symNIPath.rfind('/') + 1);
 545                                                         if (!removeFile(concatPath(binNIDir, niFileName))) {
 546                                                                 fprintf(stderr, "Failed to remove of %s\n", concatPath(binNIDir, niFileName).c_str());
 547                                                         }
 548                                                 }
 549                                         }
 550                                 }
 551                         }
 552                 } catch (const bf::filesystem_error& error) {
 553                         fprintf(stderr, "Failed to recursive directory: %s\n", error.what());
 554                         return NI_ERROR_UNKNOWN;
 555                 }
 556         }
 557         return NI_ERROR_NONE;
 558 }
 559
 560 ni_error_e createNiUnderDirs(const std::string rootPaths[], int count, DWORD flags)
 561 {
 562         if (!isCoreLibPrepared(flags)) {
 563                 return NI_ERROR_CORE_NI_FILE;
 564         }
 565
 566         std::string appPaths;
 567         for (int i = 0; i < count; i++) {
 568                 appPaths += rootPaths[i];
 569                 appPaths += ':';
 570         }
 571
 572         if (appPaths.back() == ':')
 573                 appPaths.pop_back();
 574
 575         std::vector<std::string> tpaAssemblies;
 576         splitPath(__tpa, tpaAssemblies);
 577
 578         auto convert = [&appPaths, flags, tpaAssemblies](const std::string& path, const std::string& filename) {
 579                 bool isAppNI = flags & NI_FLAGS_APPNI;
 580                 if (isAppNI) {
 581                         for (auto& tpa : tpaAssemblies) {
 582                                 if (!strcmp(replaceAll(tpa.substr(tpa.rfind('/') + 1), ".ni.dll", ".dll").c_str(), filename.c_str())) {
 583                                         fprintf(stderr, "%s present in the TPA list skips generation of NI file.\n", filename.c_str());
 584                                         return;
 585                                 }
 586                         }
 587                 }
 588                 if (!crossgen(path, appPaths.c_str(), flags)) {
 589                         waitInterval();
 590                 }
 591         };
 592
 593         for (int i = 0; i < count; i++) {
 594                 scanFilesInDirectory(rootPaths[i], convert, 0);
 595         }
 596
 597         tpaAssemblies.clear();
 598         return NI_ERROR_NONE;
 599 }
 600
 601 ni_error_e createNiUnderPkgRoot(const std::string& pkgId, DWORD flags)
 602 {
 603         std::string pkgRoot;
 604         if (getRootPath(pkgId, pkgRoot) < 0) {
 605                 return NI_ERROR_INVALID_PACKAGE;
 606         }
 607
 608         std::string binDir = concatPath(pkgRoot, "bin");
 609         std::string libDir = concatPath(pkgRoot, "lib");
 610         std::string tacDir = concatPath(binDir, TAC_SYMLINK_SUB_DIR);
 611         std::string paths[] = {binDir, libDir, tacDir};
 612
 613         flags |= NI_FLAGS_APPNI;
 614         return createNiUnderDirs(paths, 3, flags);
 615 }
 616
 617 ni_error_e createNiDllUnderPkgRoot(const std::string& pkgId, const std::string& dllPath, DWORD flags)
 618 {
 619         if (!isCoreLibPrepared(flags)) {
 620                 return NI_ERROR_CORE_NI_FILE;
 621         }
 622
 623         std::string pkgRoot;
 624         if (getRootPath(pkgId, pkgRoot) < 0) {
 625                 return NI_ERROR_INVALID_PACKAGE;
 626         }
 627
 628         std::string binDir = concatPath(pkgRoot, "bin");
 629         std::string libDir = concatPath(pkgRoot, "lib");
 630         std::string tacDir = concatPath(binDir, TAC_SYMLINK_SUB_DIR);
 631         std::string binNIDir = concatPath(binDir, APP_NI_SUB_DIR);
 632         std::string paths = binDir + ":" + libDir + ":" + tacDir;
 633
 634         if (bf::is_symlink(dllPath)) {
 635                 if (bf::exists(tacDir)) {
 636                         if (!isNativeImage(dllPath)) {
 637                                 std::string originPath = bf::read_symlink(dllPath).string();
 638                                 std::string originNiPath = originPath.substr(0, originPath.rfind(".dll")) + ".ni.dll";
 639                                 if (!bf::exists(originNiPath)) {
 640                                         flags |= NI_FLAGS_APPNI;
 641                                         if(crossgen(originPath, paths, flags) != NI_ERROR_NONE) {
 642                                                 fprintf(stderr, "Failed to create NI file [%s]\n", originPath.c_str());
 643                                                 return NI_ERROR_UNKNOWN;
 644                                         }
 645                                 }
 646                                 std::string setNiPath = dllPath.substr(0, dllPath.rfind(".dll")) + ".ni.dll";
 647                                 if (!bf::exists(setNiPath)) {
 648                                         bf::create_symlink(originNiPath, setNiPath);
 649                                         fprintf(stdout, "%s symbolic link file generated successfully.\n", setNiPath.c_str());
 650                                         copySmackAndOwnership(tacDir.c_str(), setNiPath.c_str(), true);
 651                                 }
 652                                 std::string niFileName = setNiPath.substr(setNiPath.rfind('/') + 1);
 653                                 if (!removeFile(concatPath(binNIDir, niFileName))) {
 654                                         fprintf(stderr, "Failed to remove of %s\n", concatPath(binNIDir, niFileName).c_str());
 655                                 }
 656                         }
 657                 }
 658                 return NI_ERROR_NONE;
 659         } else {
 660                 std::string assembly = dllPath.substr(dllPath.rfind('/') + 1);
 661                 std::vector<std::string> tpaAssemblies;
 662                 splitPath(__tpa, tpaAssemblies);
 663                 for (auto& tpa : tpaAssemblies) {
 664                         if (!strcmp(replaceAll(tpa.substr(tpa.rfind('/') + 1), ".ni.dll", ".dll").c_str(), assembly.c_str())) {
 665                                 fprintf(stderr, "%s present in the TPA list skips generation of NI file.\n", assembly.c_str());
 666                                 return NI_ERROR_NONE;
 667                         }
 668                 }
 669                 tpaAssemblies.clear();
 670                 flags |= NI_FLAGS_APPNI;
 671                 return crossgen(dllPath, paths, flags);
 672         }
 673 }
 674
 675 void removeNiPlatform()
 676 {
 677         std::string coreLib = concatPath(getRuntimeDir(), "System.Private.CoreLib.dll");
 678         std::string coreLibBackup = concatPath(getRuntimeDir(), "System.Private.CoreLib.dll.Backup");
 679
 680         if (!isFileExist(coreLibBackup)) {
 681                 return;
 682         }
 683
 684 #ifdef UNIQUE_DEFAULT_BASE_ADDR_SUPPORT
 685         if (isFileExist(__SYSTEM_BASE_FILE)) {
 686                 if (remove(__SYSTEM_BASE_FILE)) {
 687                         fprintf(stderr, "Failed to remove %s\n", __SYSTEM_BASE_FILE);
 688                 }
 689         }
 690 #endif
 691
 692         if (remove(coreLib.c_str())) {
 693                 fprintf(stderr, "Failed to remove System.Private.CoreLib native image file\n");
 694         }
 695
 696         if (rename(coreLibBackup.c_str(), coreLib.c_str())) {
 697                 fprintf(stderr, "Failed to rename System.Private.CoreLib.Backup to origin\n");
 698         }
 699
 700         const std::string platformDirs[] = {getRuntimeDir(), getTizenFXDir()};
 701
 702         removeNiUnderDirs(platformDirs, 2);
 703 }
 704
 705 void removeNiUnderDirs(const std::string rootPaths[], int count)
 706 {
 707         auto convert = [](const std::string& path, const std::string& filename) {
 708                 if (isNativeImage(path)) {
 709                         if (remove(path.c_str())) {
 710                                 fprintf(stderr, "Failed to remove %s\n", path.c_str());
 711                         }
 712                 }
 713         };
 714
 715         for (int i = 0; i < count; i++) {
 716                 scanFilesInDirectory(rootPaths[i], convert, -1);
 717         }
 718 }
 719
 720 ni_error_e removeNiUnderPkgRoot(const std::string& pkgId)
 721 {
 722         std::string pkgRoot;
 723         if (getRootPath(pkgId, pkgRoot) < 0) {
 724                 return NI_ERROR_INVALID_PACKAGE;
 725         }
 726
 727         std::string binDir = concatPath(pkgRoot, "bin");
 728         std::string libDir = concatPath(pkgRoot, "lib");
 729         std::string paths[] = {binDir, libDir};
 730
 731         removeNiUnderDirs(paths, 2);
 732
 733         std::string binNIDir = concatPath(binDir, APP_NI_SUB_DIR);
 734         if (isFileExist(binNIDir)) {
 735                 if (!removeAll(binNIDir.c_str())) {
 736                         fprintf(stderr, "Failed to remove app ni dir [%s]\n", binNIDir.c_str());
 737                 }
 738         }
 739
 740         std::string libNIDir = concatPath(libDir, APP_NI_SUB_DIR);
 741         if (isFileExist(libNIDir)) {
 742                 if (!removeAll(libNIDir.c_str())) {
 743                         fprintf(stderr, "Failed to remove app ni dir [%s]\n", libNIDir.c_str());
 744                 }
 745         }
 746
 747         return NI_ERROR_NONE;
 748 }
 749
 750 ni_error_e regenerateAppNI(DWORD flags)
 751 {
 752         if (!isCoreLibPrepared(flags)) {
 753                 return NI_ERROR_CORE_NI_FILE;
 754         }
 755
 756         int ret = 0;
 757         pkgmgrinfo_appinfo_metadata_filter_h handle;
 758
 759         ret = pkgmgrinfo_appinfo_metadata_filter_create(&handle);
 760         if (ret != PMINFO_R_OK)
 761                 return NI_ERROR_UNKNOWN;
 762
 763         ret = pkgmgrinfo_appinfo_metadata_filter_add(handle, AOT_METADATA_KEY, METADATA_VALUE);
 764         if (ret != PMINFO_R_OK) {
 765                 pkgmgrinfo_appinfo_metadata_filter_destroy(handle);
 766                 return NI_ERROR_UNKNOWN;
 767         }
 768
 769         ret = pkgmgrinfo_appinfo_metadata_filter_foreach(handle, appAotCb, &flags);
 770         if (ret != PMINFO_R_OK) {
 771                 fprintf(stderr, "Failed pkgmgrinfo_appinfo_metadata_filter_foreach\n");
 772                 pkgmgrinfo_appinfo_metadata_filter_destroy(handle);
 773                 return NI_ERROR_UNKNOWN;
 774         }
 775
 776         fprintf(stdout, "Success pkgmgrinfo_appinfo_metadata_filter_foreach\n");
 777
 778         pkgmgrinfo_appinfo_metadata_filter_destroy(handle);
 779         return NI_ERROR_NONE;
 780 }
 781
 782 // callback function of "pkgmgrinfo_appinfo_metadata_filter_foreach"
 783 static int regenTacCb(pkgmgrinfo_appinfo_h handle, void *userData)
 784 {
 785         char *pkgId = NULL;
 786         DWORD *pFlags = (DWORD*)userData;
 787
 788         int ret = pkgmgrinfo_appinfo_get_pkgid(handle, &pkgId);
 789         if (ret != PMINFO_R_OK) {
 790                 fprintf(stderr, "Failed to get pkgid\n");
 791                 return -1;
 792         }
 793
 794         sqlite3 *tac_db = dbOpen(TAC_APP_LIST_DB);
 795         if (!tac_db) {
 796                 fprintf(stderr, "Sqlite open error\n");
 797                 return -1;
 798         }
 799
 800         char *sql = sqlite3_mprintf("SELECT * FROM TAC WHERE PKGID = %Q;", pkgId);
 801         std::vector<std::string> nugets = dbSelect(tac_db, TAC_APP_LIST_DB, sql);
 802         sqlite3_free(sql);
 803
 804         if (tac_db) {
 805                 dbClose(tac_db);
 806                 tac_db = NULL;
 807         }
 808
 809         createNiUnderTAC(nugets, *pFlags);
 810
 811         return 0;
 812 }
 813
 814 ni_error_e regenerateTACNI(DWORD flags)
 815 {
 816         if (!isCoreLibPrepared(flags)) {
 817                 return NI_ERROR_CORE_NI_FILE;
 818         }
 819
 820         const std::string tacDir[] = {__DOTNET_DIR};
 821         removeNiUnderDirs(tacDir, 1);
 822
 823         pkgmgrinfo_appinfo_metadata_filter_h handle;
 824         int ret = pkgmgrinfo_appinfo_metadata_filter_create(&handle);
 825         if (ret != PMINFO_R_OK) {
 826                 return NI_ERROR_UNKNOWN;
 827         }
 828
 829         ret = pkgmgrinfo_appinfo_metadata_filter_add(handle, TAC_METADATA_KEY, METADATA_VALUE);
 830         if (ret != PMINFO_R_OK) {
 831                 pkgmgrinfo_appinfo_metadata_filter_destroy(handle);
 832                 return NI_ERROR_UNKNOWN;
 833         }
 834
 835         ret = pkgmgrinfo_appinfo_metadata_filter_foreach(handle, regenTacCb, &flags);
 836         if (ret != PMINFO_R_OK) {
 837                 fprintf(stderr, "Failed pkgmgrinfo_appinfo_metadata_filter_foreach\n");
 838                 pkgmgrinfo_appinfo_metadata_filter_destroy(handle);
 839                 return NI_ERROR_UNKNOWN;
 840         }
 841         fprintf(stdout, "Success pkgmgrinfo_appinfo_metadata_filter_foreach\n");
 842
 843         pkgmgrinfo_appinfo_metadata_filter_destroy(handle);
 844
 845         return NI_ERROR_NONE;
 846 }