Modified to add 'NI_FLAGS_APPNI' flag and appPaths when the app generates ni
[platform/core/dotnet/launcher.git] / 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* __TAC_DIR = __STR(TAC_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(stderr, "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         if (absNiPath.empty()) {
242                 fprintf(stderr, "Fail to get ni file name\n");
243                 return NI_ERROR_UNKNOWN;
244         }
245
246         bool isAppNI = flags & NI_FLAGS_APPNI;
247         if (isAppNI && strstr(absNiPath.c_str(), __TAC_DIR) == NULL) {
248                 absNiPath = getAppNIPath(absNiPath);
249         }
250
251 #ifdef UNIQUE_DEFAULT_BASE_ADDR_SUPPORT
252         uintptr_t baseAddr = 0;
253
254         if (isTPADll(dllPath)) {
255                 baseAddr = getNextBaseAddr();
256         }
257 #endif
258
259         pid_t pid = fork();
260         if (pid == -1)
261                 return NI_ERROR_UNKNOWN;
262
263         if (pid > 0) {
264                 int status;
265                 waitpid(pid, &status, 0);
266                 if (WIFEXITED(status)) {
267                         // Do not use niExist() function to check whether ni file created or not.
268                         // niEixst() return false for System.Private.Corelib.dll
269                         if (isFileExist(absNiPath)) {
270                                 copySmackAndOwnership(absDllPath, absNiPath);
271 #ifdef UNIQUE_DEFAULT_BASE_ADDR_SUPPORT
272                                 if (baseAddr != 0) {
273                                         updateBaseAddrFile(absNiPath, baseAddr);
274                                 }
275 #endif
276                                 return NI_ERROR_NONE;
277                         } else {
278                                 fprintf(stderr, "Fail to create native image for %s\n", dllPath.c_str());
279                                 return NI_ERROR_NO_SUCH_FILE;
280                         }
281                 }
282         } else {
283                 std::string jitPath = getRuntimeDir() + "/libclrjit.so";
284                 std::vector<const char*> argv = {
285                         __CROSSGEN_PATH,
286                         "/nologo",
287                         "/JITPath", jitPath.c_str()
288                 };
289
290                 bool compat = flags & NI_FLAGS_COMPATIBILITY;
291                 argv.push_back(compat ? "/Trusted_Platform_Assemblies" : "/r");
292                 argv.push_back(__tpa.c_str());
293
294                 bool enableR2R = flags & NI_FLAGS_ENABLER2R;
295                 if (!enableR2R) {
296                         argv.push_back("/FragileNonVersionable");
297                 }
298
299                 if (flags & NI_FLAGS_VERBOSE) {
300                         argv.push_back("/verbose");
301                 }
302
303                 if (flags & NI_FLAGS_INSTRUMENT) {
304                         argv.push_back("/Tuning");
305                 }
306
307 #ifdef UNIQUE_DEFAULT_BASE_ADDR_SUPPORT
308                 if (baseAddr != 0) {
309                         argv.push_back("/BaseAddress");
310                         argv.push_back(std::to_string(baseAddr).c_str());
311                 }
312 #endif
313
314                 argv.push_back("/App_Paths");
315                 std::string absAppPath;
316                 if (!appPath.empty()) {
317                         absAppPath = appPath;
318                 } else {
319                         absAppPath = baseName(absDllPath);
320                 }
321                 argv.push_back(absAppPath.c_str());
322
323                 argv.push_back("/out");
324                 argv.push_back(absNiPath.c_str());
325
326                 argv.push_back(absDllPath.c_str());
327                 argv.push_back(nullptr);
328
329                 fprintf(stderr, "+ %s (%s)\n", absDllPath.c_str(), enableR2R ? "R2R" : "FNV");
330
331                 execv(__CROSSGEN_PATH, const_cast<char* const*>(argv.data()));
332                 exit(0);
333         }
334
335         return NI_ERROR_NONE;
336 }
337
338 // callback function of "pkgmgrinfo_appinfo_metadata_filter_foreach"
339 static int appAotCb(pkgmgrinfo_appinfo_h handle, void *userData)
340 {
341         char *pkgId = NULL;
342         int ret = 0;
343         DWORD *pFlags = (DWORD*)userData;
344
345         ret = pkgmgrinfo_appinfo_get_pkgid(handle, &pkgId);
346         if (ret != PMINFO_R_OK) {
347                 fprintf(stderr, "Failed to get pkgid\n");
348                 return -1;
349         }
350
351         if (removeNiUnderPkgRoot(pkgId) != NI_ERROR_NONE) {
352                 fprintf(stderr, "Failed to remove previous dlls from [%s]\n", pkgId);
353                 return -1;
354         }
355
356         if (resetTACPackage(pkgId) != TAC_ERROR_NONE) {
357                 fprintf(stderr, "Failed to remove symlink for given package [%s]\n", pkgId);
358                 return -1;
359         }
360
361         // Regenerate ni files with R2R mode forcibiliy. (there is no way to now which option is used)
362         if (createNiUnderPkgRoot(pkgId, *pFlags) != NI_ERROR_NONE) {
363                 fprintf(stderr, "Failed to generate NI file [%s]\n", pkgId);
364                 return -1;
365         } else {
366                 fprintf(stderr, "Complete make application to native image\n");
367         }
368
369         if (createTACPkgRoot(pkgId, *pFlags) != NI_ERROR_NONE) {
370                 fprintf(stderr, "Failed to generate symbolic link file [%s]\n", pkgId);
371                 return -1;
372         }else {
373                 fprintf(stderr, "Complete make symbolic link file to tac\n");
374         }
375
376         return 0;
377 }
378
379 static void createCoreLibNI(DWORD flags)
380 {
381         std::string coreLib = concatPath(getRuntimeDir(), "System.Private.CoreLib.dll");
382         std::string niCoreLib = concatPath(getRuntimeDir(), "System.Private.CoreLib.ni.dll");
383         std::string coreLibBackup = concatPath(getRuntimeDir(), "System.Private.CoreLib.dll.Backup");
384
385         if (!isFileExist(coreLibBackup)) {
386
387                 if (!crossgen(coreLib, std::string(), flags)) {
388                         if (rename(coreLib.c_str(), coreLibBackup.c_str())) {
389                                 fprintf(stderr, "Failed to rename System.Private.CoreLib.dll\n");
390                         }
391                         if (rename(niCoreLib.c_str(), coreLib.c_str())) {
392                                 fprintf(stderr, "Failed to rename System.Private.CoreLib.ni.dll\n");
393                         }
394                 } else {
395                         fprintf(stderr, "Failed to create native image for %s\n", coreLib.c_str());
396                 }
397         }
398 }
399
400 ni_error_e initNICommon(NiCommonOption* option)
401 {
402 #if defined(__arm__)
403         // get interval value
404         const static std::string intervalFile = concatPath(__NATIVE_LIB_DIR, "crossgen_interval.txt");
405         std::ifstream inFile(intervalFile);
406         if (inFile) {
407                 fprintf(stderr, "crossgen_interval.txt is found\n");
408                 inFile >> __interval;
409         }
410
411         if (initializePluginManager("normal")) {
412                 fprintf(stderr, "Fail to initialize PluginManager\n");
413                 return NI_ERROR_UNKNOWN;
414         }
415         if (initializePathManager(option->runtimeDir, option->tizenFXDir, option->extraDirs)) {
416                 fprintf(stderr, "Fail to initialize PathManager\n");
417                 return NI_ERROR_UNKNOWN;
418         }
419
420         __tpa = getTPA();
421
422         return NI_ERROR_NONE;
423 #else
424         fprintf(stderr, "crossgen supports arm architecture only. skip ni file generation\n");
425         return NI_ERROR_NOT_SUPPORTED;
426 #endif
427 }
428
429 void finalizeNICommon()
430 {
431         __interval = 0;
432
433         finalizePluginManager();
434         finalizePathManager();
435
436         __tpa.clear();
437 }
438
439
440 void createNiPlatform(DWORD flags)
441 {
442         const std::string platformDirs[] = {getRuntimeDir(), getTizenFXDir()};
443         createNiUnderDirs(platformDirs, 2, flags);
444 }
445
446 ni_error_e createNiDll(const std::string& dllPath, DWORD flags)
447 {
448         createCoreLibNI(flags);
449         // System.Private.CoreLib.dll is generated in the createCoreLibNI function.
450         // Skip if input dll is System.Private.CoreLib.dll
451         if (dllPath.find("System.Private.CoreLib.dll") != std::string::npos) {
452                 return NI_ERROR_NONE;
453         }
454
455         return crossgen(dllPath, std::string(), flags);
456 }
457
458 void createNiUnderTAC(std::vector<std::string> nugets, DWORD flags)
459 {
460         std::string appPaths;
461         for (auto& nuget : nugets) {
462                 appPaths += concatPath(__TAC_DIR, nuget);
463                 appPaths += ':';
464         }
465         if (appPaths.back() == ':') {
466                 appPaths.pop_back();
467         }
468
469         auto convert = [&appPaths, flags](const std::string& path, const char* name) {
470                 if (strstr(path.c_str(), TAC_SHA_256_INFO) != NULL)
471                         return;
472                 if (!crossgen(path, appPaths.c_str(), flags)) {
473                         waitInterval();
474                 }
475         };
476         for (auto& nuget : nugets) {
477                 scanFilesInDir(concatPath(__TAC_DIR, nuget), convert, -1);
478         }
479 }
480
481 ni_error_e createTACPkgRoot(const std::string& pkgId, DWORD flags)
482 {
483         std::string pkgRoot;
484         if (getRootPath(pkgId, pkgRoot) < 0) {
485                 return NI_ERROR_INVALID_PACKAGE;
486         }
487
488         std::string binDir = concatPath(pkgRoot, "bin");
489         std::string libDir = concatPath(pkgRoot, "lib");
490         std::string tacDir = concatPath(binDir, TAC_SYMLINK_SUB_DIR);
491         std::string binNIDir = concatPath(binDir, APP_NI_SUB_DIR);
492         std::string paths = binDir + ":" + libDir + ":" + tacDir;
493         if (bf::exists(tacDir)) {
494                 try {
495                         for (auto& symlinkAssembly : bf::recursive_directory_iterator(tacDir)) {
496                                 std::string symPath = symlinkAssembly.path().string();
497                                 if (bf::is_symlink(symPath)) {
498                                         if (!isNativeImage(symPath)) {
499                                                 std::string originPath = bf::read_symlink(symPath).string();
500                                                 std::string originNiPath = originPath.substr(0, originPath.rfind(".dll")) + ".ni.dll";
501                                                 if (!bf::exists(originNiPath)) {
502                                                         flags |= NI_FLAGS_APPNI;
503                                                         if(crossgen(originPath, paths, flags) != NI_ERROR_NONE) {
504                                                                 fprintf(stderr, "Failed to create NI file [%s]\n", originPath.c_str());
505                                                                 return NI_ERROR_UNKNOWN;
506                                                         }
507                                                 }
508                                                 std::string symNIPath = symPath.substr(0, symPath.rfind(".dll")) + ".ni.dll";
509                                                 if (!bf::exists(symNIPath)) {
510                                                         bf::create_symlink(originNiPath, symNIPath);
511                                                         fprintf(stderr, "%s symbolic link file generated successfully.\n", symNIPath.c_str());
512                                                         copySmackAndOwnership(tacDir.c_str(), symNIPath.c_str(), true);
513
514                                                         std::string niFileName = symNIPath.substr(symNIPath.rfind('/') + 1);
515                                                         if (!removeFile(concatPath(binNIDir, niFileName))) {
516                                                                 fprintf(stderr, "Failed to remove of %s\n", concatPath(binNIDir, niFileName).c_str());
517                                                         }
518                                                 }
519                                         }
520                                 }
521                         }
522                 } catch (const bf::filesystem_error& error) {
523                         fprintf(stderr, "Failed to recursive directory: %s\n", error.what());
524                         return NI_ERROR_UNKNOWN;
525                 }
526         }
527         return NI_ERROR_NONE;
528 }
529
530 void createNiUnderDirs(const std::string rootPaths[], int count, DWORD flags)
531 {
532         createCoreLibNI(flags);
533
534         std::string appPaths;
535         for (int i = 0; i < count; i++) {
536                 appPaths += rootPaths[i];
537                 appPaths += ':';
538         }
539
540         if (appPaths.back() == ':')
541                 appPaths.pop_back();
542
543         std::vector<std::string> tpaAssemblies;
544         splitPath(__tpa, tpaAssemblies);
545
546         auto convert = [&appPaths, flags, tpaAssemblies](const std::string& path, const char* name) {
547                 bool isAppNI = flags & NI_FLAGS_APPNI;
548                 if (isAppNI) {
549                         for (auto& tpa : tpaAssemblies) {
550                                 if (!strcmp(replaceAll(tpa.substr(tpa.rfind('/') + 1), ".ni.dll", ".dll").c_str(), name)) {
551                                         fprintf(stderr, "%s present in the TPA list skips generation of NI file.\n", name);
552                                         return;
553                                 }
554                         }
555                 }
556                 if (!crossgen(path, appPaths.c_str(), flags)) {
557                         waitInterval();
558                 }
559         };
560
561         for (int i = 0; i < count; i++) {
562                 scanFilesInDir(rootPaths[i], convert, 1);
563         }
564
565         tpaAssemblies.clear();
566 }
567
568 ni_error_e createNiUnderPkgRoot(const std::string& pkgId, DWORD flags)
569 {
570         std::string pkgRoot;
571         if (getRootPath(pkgId, pkgRoot) < 0) {
572                 return NI_ERROR_INVALID_PACKAGE;
573         }
574
575         std::string binDir = concatPath(pkgRoot, "bin");
576         std::string libDir = concatPath(pkgRoot, "lib");
577         std::string tacDir = concatPath(binDir, TAC_SYMLINK_SUB_DIR);
578         std::string paths[] = {binDir, libDir, tacDir};
579
580         flags |= NI_FLAGS_APPNI;
581         createNiUnderDirs(paths, 3, flags);
582
583         return NI_ERROR_NONE;
584 }
585
586 ni_error_e createNiDllUnderPkgRoot(const std::string& pkgId, const std::string& dllPath, DWORD flags)
587 {
588         std::string pkgRoot;
589         if (getRootPath(pkgId, pkgRoot) < 0) {
590                 return NI_ERROR_INVALID_PACKAGE;
591         }
592
593         std::string binDir = concatPath(pkgRoot, "bin");
594         std::string libDir = concatPath(pkgRoot, "lib");
595         std::string tacDir = concatPath(binDir, TAC_SYMLINK_SUB_DIR);
596         std::string binNIDir = concatPath(binDir, APP_NI_SUB_DIR);
597         std::string paths = binDir + ":" + libDir + ":" + tacDir;
598
599         if (bf::is_symlink(dllPath)) {
600                 if (bf::exists(tacDir)) {
601                         if (!isNativeImage(dllPath)) {
602                                 std::string originPath = bf::read_symlink(dllPath).string();
603                                 std::string originNiPath = originPath.substr(0, originPath.rfind(".dll")) + ".ni.dll";
604                                 if (!bf::exists(originNiPath)) {
605                                         flags |= NI_FLAGS_APPNI;
606                                         if(crossgen(originPath, paths, flags) != NI_ERROR_NONE) {
607                                                 fprintf(stderr, "Failed to create NI file [%s]\n", originPath.c_str());
608                                                 return NI_ERROR_UNKNOWN;
609                                         }
610                                 }
611                                 std::string setNiPath = dllPath.substr(0, dllPath.rfind(".dll")) + ".ni.dll";
612                                 if (!bf::exists(setNiPath)) {
613                                         bf::create_symlink(originNiPath, setNiPath);
614                                         fprintf(stderr, "%s symbolic link file generated successfully.\n", setNiPath.c_str());
615                                         copySmackAndOwnership(tacDir.c_str(), setNiPath.c_str(), true);
616                                 }
617                                 std::string niFileName = setNiPath.substr(setNiPath.rfind('/') + 1);
618                                 if (!removeFile(concatPath(binNIDir, niFileName))) {
619                                         fprintf(stderr, "Failed to remove of %s\n", concatPath(binNIDir, niFileName).c_str());
620                                 }
621                         }
622                 }
623                 return NI_ERROR_NONE;
624         } else {
625                 std::string assembly = dllPath.substr(dllPath.rfind('/') + 1);
626                 std::vector<std::string> tpaAssemblies;
627                 splitPath(__tpa, tpaAssemblies);
628                 for (auto& tpa : tpaAssemblies) {
629                         if (!strcmp(replaceAll(tpa.substr(tpa.rfind('/') + 1), ".ni.dll", ".dll").c_str(), assembly.c_str())) {
630                                 fprintf(stderr, "%s present in the TPA list skips generation of NI file.\n", assembly.c_str());
631                                 return NI_ERROR_NONE;
632                         }
633                 }
634                 tpaAssemblies.clear();
635                 flags |= NI_FLAGS_APPNI;
636                 return crossgen(dllPath, paths, flags);
637         }
638 }
639
640 void removeNiPlatform()
641 {
642         std::string coreLib = concatPath(getRuntimeDir(), "System.Private.CoreLib.dll");
643         std::string coreLibBackup = concatPath(getRuntimeDir(), "System.Private.CoreLib.dll.Backup");
644
645         if (!isFileExist(coreLibBackup)) {
646                 return;
647         }
648
649 #ifdef UNIQUE_DEFAULT_BASE_ADDR_SUPPORT
650         if (isFileExist(__SYSTEM_BASE_FILE)) {
651                 if (remove(__SYSTEM_BASE_FILE)) {
652                         fprintf(stderr, "Failed to remove %s\n", __SYSTEM_BASE_FILE);
653                 }
654         }
655 #endif
656
657         if (remove(coreLib.c_str())) {
658                 fprintf(stderr, "Failed to remove System.Private.CoreLib native image file\n");
659         }
660
661         if (rename(coreLibBackup.c_str(), coreLib.c_str())) {
662                 fprintf(stderr, "Failed to rename System.Private.CoreLib.Backup to origin\n");
663         }
664
665         const std::string platformDirs[] = {getRuntimeDir(), getTizenFXDir()};
666
667         removeNiUnderDirs(platformDirs, 2);
668 }
669
670 void removeNiUnderDirs(const std::string rootPaths[], int count)
671 {
672         auto convert = [](const std::string& path, std::string name) {
673                 if (isNativeImage(path)) {
674                         if (remove(path.c_str())) {
675                                 fprintf(stderr, "Failed to remove %s\n", path.c_str());
676                         }
677                 }
678         };
679
680         for (int i = 0; i < count; i++) {
681                 scanFilesInDir(rootPaths[i], convert, -1);
682         }
683 }
684
685 ni_error_e removeNiUnderPkgRoot(const std::string& pkgId)
686 {
687         std::string pkgRoot;
688         if (getRootPath(pkgId, pkgRoot) < 0) {
689                 return NI_ERROR_INVALID_PACKAGE;
690         }
691
692         std::string binDir = concatPath(pkgRoot, "bin");
693         std::string libDir = concatPath(pkgRoot, "lib");
694         std::string paths[] = {binDir, libDir};
695
696         removeNiUnderDirs(paths, 2);
697
698         std::string binNIDir = concatPath(binDir, APP_NI_SUB_DIR);
699         if (isFileExist(binNIDir)) {
700                 if (rmdir(binNIDir.c_str()) != 0) {
701                         fprintf(stderr, "Failed to remove app ni dir [%s]\n", binNIDir.c_str());
702                 }
703         }
704
705         std::string libNIDir = concatPath(libDir, APP_NI_SUB_DIR);
706         if (isFileExist(libNIDir)) {
707                 if (rmdir(libNIDir.c_str()) != 0) {
708                         fprintf(stderr, "Failed to remove app ni dir [%s]\n", libNIDir.c_str());
709                 }
710         }
711
712         return NI_ERROR_NONE;
713 }
714
715 ni_error_e regenerateAppNI(DWORD flags)
716 {
717         int ret = 0;
718         pkgmgrinfo_appinfo_metadata_filter_h handle;
719
720         ret = pkgmgrinfo_appinfo_metadata_filter_create(&handle);
721         if (ret != PMINFO_R_OK)
722                 return NI_ERROR_UNKNOWN;
723
724         ret = pkgmgrinfo_appinfo_metadata_filter_add(handle, AOT_METADATA_KEY, METADATA_VALUE);
725         if (ret != PMINFO_R_OK) {
726                 pkgmgrinfo_appinfo_metadata_filter_destroy(handle);
727                 return NI_ERROR_UNKNOWN;
728         }
729
730         ret = pkgmgrinfo_appinfo_metadata_filter_foreach(handle, appAotCb, &flags);
731         if (ret != PMINFO_R_OK) {
732                 fprintf(stderr, "Failed pkgmgrinfo_appinfo_metadata_filter_foreach\n");
733                 pkgmgrinfo_appinfo_metadata_filter_destroy(handle);
734                 return NI_ERROR_UNKNOWN;
735         }
736
737         fprintf(stderr, "Success pkgmgrinfo_appinfo_metadata_filter_foreach\n");
738
739         pkgmgrinfo_appinfo_metadata_filter_destroy(handle);
740         return NI_ERROR_NONE;
741 }
742
743 // callback function of "pkgmgrinfo_appinfo_metadata_filter_foreach"
744 static int regenTacCb(pkgmgrinfo_appinfo_h handle, void *userData)
745 {
746         char *pkgId = NULL;
747         DWORD *pFlags = (DWORD*)userData;
748
749         int ret = pkgmgrinfo_appinfo_get_pkgid(handle, &pkgId);
750         if (ret != PMINFO_R_OK) {
751                 fprintf(stderr, "Failed to get pkgid\n");
752                 return -1;
753         }
754
755         sqlite3 *tac_db = dbOpen(TAC_APP_LIST_DB);
756         if (!tac_db) {
757                 fprintf(stderr, "Sqlite open error\n");
758                 return -1;
759         }
760
761         char *sql = sqlite3_mprintf("SELECT * FROM TAC WHERE PKGID = %Q;", pkgId);
762         std::vector<std::string> nugets = dbSelect(tac_db, TAC_APP_LIST_DB, sql);
763         sqlite3_free(sql);
764
765         if (tac_db) {
766                 dbClose(tac_db);
767                 tac_db = NULL;
768         }
769
770         createNiUnderTAC(nugets, *pFlags);
771
772         return 0;
773 }
774
775 ni_error_e regenerateTACNI(DWORD flags)
776 {
777         const std::string tacDir[] = {__TAC_DIR};
778         removeNiUnderDirs(tacDir, 1);
779
780         pkgmgrinfo_appinfo_metadata_filter_h handle;
781         int ret = pkgmgrinfo_appinfo_metadata_filter_create(&handle);
782         if (ret != PMINFO_R_OK) {
783                 return NI_ERROR_UNKNOWN;
784         }
785
786         ret = pkgmgrinfo_appinfo_metadata_filter_add(handle, TAC_METADATA_KEY, METADATA_VALUE);
787         if (ret != PMINFO_R_OK) {
788                 pkgmgrinfo_appinfo_metadata_filter_destroy(handle);
789                 return NI_ERROR_UNKNOWN;
790         }
791
792         ret = pkgmgrinfo_appinfo_metadata_filter_foreach(handle, regenTacCb, &flags);
793         if (ret != PMINFO_R_OK) {
794                 fprintf(stderr, "Failed pkgmgrinfo_appinfo_metadata_filter_foreach\n");
795                 pkgmgrinfo_appinfo_metadata_filter_destroy(handle);
796                 return NI_ERROR_UNKNOWN;
797         }
798         fprintf(stderr, "Success pkgmgrinfo_appinfo_metadata_filter_foreach\n");
799
800         pkgmgrinfo_appinfo_metadata_filter_destroy(handle);
801
802         return NI_ERROR_NONE;
803 }