Modify to enable TAC when restoring a database of TAC
[platform/core/dotnet/launcher.git] / NativeLauncher / tool / dotnettool.cc
1 /*
2  * Copyright (c) 2019 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 "utils.h"
18 #include "ni_common.h"
19 #include "tac_common.h"
20
21 #include <algorithm>
22 #include <cstdio>
23 #include <cstring>
24 #include <vector>
25
26 std::vector<std::string> getCmdArgs(char** begin, char** end)
27 {
28         std::vector<std::string> list;
29         for (char** itr = begin+1; itr != end; itr++) {
30                 if (strncmp(*itr, "--", 2) != 0) {
31                         list.push_back(*itr);
32                 }
33         }
34         return list;
35 }
36
37 static void help(const char *argv0)
38 {
39         const char* helpDesc =
40                 "Usage: %s [OPTIONS] COMMAND <paths or pkg name>\n"
41                 "\n"
42                 "Commands:\n"
43                 "       --help                 - Display this screen\n"
44                 "       --ni-system            - Create NI under System DLLs\n"
45                 "       --ni-dll               - Create NI for DLL\n"
46                 "       --ni-pkg               - Create NI for package\n"
47                 "       --ni-pkg-dll           - Create NI for DLL in package\n"
48                 "       --ni-dir               - Create NI for directory\n"
49                 "       --ni-reset-system      - Remove System NI files\n"
50                 "       --ni-reset-pkg         - Remove App NI files\n"
51                 "       --ni-regen-all-app     - Re-generate All App NI files\n"
52                 "       --tac-regen-all        - Re-generate All TAC files\n"
53                 "       --tac-restore-db       - Restore TAC Database\n"
54                 "       --tac-disable-pkg      - Disable TAC for package\n"
55                 "       --tac-enable-pkg       - Enable TAC for package\n"
56                 "\n"
57                 "Options:\n"
58                 "       --r2r                  - Generate a Ready-To-Run image (disables /FragileNonVersionable)\n"
59                 "       --compatibility        - Compatibility mode for older versions of crossgen\n"
60                 "                                (replaces /r with /Trusted_Platform_Assemblies)\n"
61                 "       --verbose              - Display verbose information\n"
62                 "       --instrument           - Generate an instrumented image for profiling (enables /Tuning)\n"
63                 "\n"
64                 "Example:\n"
65                 "1. Create native image for dlls and exes under platform directories\n"
66                 "   # %s --ni-system\n"
67                 "2. Create native image for dll\n"
68                 "   # %s --ni-dll /usr/bin/Tizen.Runtime.dll\n"
69                 "3. Create native image under the package's bin and lib directory\n"
70                 "   # %s --ni-pkg org.tizen.FormsGallery\n"
71                 "4. Regenerate native images for all installed .net packages with ready-to-run option\n"
72                 "   # %s --r2r --ni-regen-all-app\n\n";
73         printf(helpDesc, argv0, argv0, argv0, argv0, argv0);
74 }
75
76 int main(int argc, char* argv[])
77 {
78         DWORD flags = 0;
79         bool pkgMode = false;
80         bool dllMode = false;
81         bool dirMode = false;
82         bool rmPkgMode = false;
83         bool pkgDllMode = false;
84         bool disableTacMode = false;
85         bool enableTacMode = false;
86
87         NiCommonOption option = {std::string(), std::string(), std::string()};
88         if (initNICommon(&option) != NI_ERROR_NONE) {
89                 return -1;
90         }
91
92         // Parse optional switches first.
93         if (cmdOptionExists(argv, argv+argc, "--r2r")) {
94                 flags |= NI_FLAGS_ENABLER2R;
95         }
96         if (cmdOptionExists(argv, argv+argc, "--compatibility")) {
97                 flags |= NI_FLAGS_COMPATIBILITY;
98         }
99         if (cmdOptionExists(argv, argv+argc, "--instrument")) {
100                 flags |= NI_FLAGS_INSTRUMENT;
101         }
102         if (cmdOptionExists(argv, argv+argc, "--verbose")) {
103                 flags |= NI_FLAGS_VERBOSE;
104         }
105
106         // The following commands are mutually exclusive.
107         if (cmdOptionExists(argv, argv+argc, "--help")) {
108                 help(argv[0]);
109                 return 0;
110         } else if (cmdOptionExists(argv, argv+argc, "--ni-system")) {
111                 createNiPlatform(flags);
112                 return 0;
113         } else if (cmdOptionExists(argv, argv+argc, "--ni-dll")) {
114                 dllMode = true;
115         } else if (cmdOptionExists(argv, argv+argc, "--ni-pkg")) {
116                 pkgMode = true;
117         } else if (cmdOptionExists(argv, argv+argc, "--ni-dir")) {
118                 dirMode = true;
119         } else if (cmdOptionExists(argv, argv+argc, "--ni-reset-system")) {
120                 removeNiPlatform();
121                 return 0;
122         } else if (cmdOptionExists(argv, argv+argc, "--ni-reset-pkg")) {
123                 rmPkgMode = true;
124         } else if (cmdOptionExists(argv, argv+argc, "--ni-pkg-dll")) {
125                 pkgDllMode = true;
126         } else if (cmdOptionExists(argv, argv+argc, "--ni-regen-all-app")) {
127                 regenerateAppNI(flags);
128                 return 0;
129         } else if (cmdOptionExists(argv, argv+argc, "--tac-regen-all")) {
130                 regenerateTACNI(flags);
131                 return 0;
132         } else if (cmdOptionExists(argv, argv+argc, "--tac-restore-db")) {
133                 restoreTACDB();
134                 return 0;
135         } else if (cmdOptionExists(argv, argv+argc, "--tac-disable-pkg")) {
136                 disableTacMode = true;
137         } else if (cmdOptionExists(argv, argv+argc, "--tac-enable-pkg")) {
138                 enableTacMode = true;
139         } else {
140                 help(argv[0]);
141                 return 0;
142         }
143
144         std::vector<std::string> args = getCmdArgs(argv, argv+argc);
145
146         if (args.size() < 1) {
147                 if (pkgMode || rmPkgMode || disableTacMode || enableTacMode) {
148                         fprintf(stderr, "Package name is missed\n");
149                 } else if (dllMode) {
150                         fprintf(stderr, "DLL path is missing.\n");
151                 }
152                 help(argv[0]);
153                 return -1;
154         } else if (args.size() < 2) {
155                 if (pkgDllMode) {
156                         fprintf(stderr, "Package name or DLL path is missing.\n");
157                         help(argv[0]);
158                         return -1;
159                 }
160         }
161
162         if (pkgMode) {
163                 for (const std::string pkg : args) {
164                         // if there is AOTed dlls under package root, that is skiped.
165                         int ret = createNiUnderPkgRoot(pkg, flags);
166                         if (ret == NI_ERROR_INVALID_PACKAGE) {
167                                 fprintf(stderr, "Failed to get root path from [%s]\n", pkg.c_str());
168                                 return -1;
169                         } else if (ret != NI_ERROR_NONE) {
170                                 fprintf(stderr, "Failed to generate NI file [%s]\n", pkg.c_str());
171                                 return -1;
172                         }
173                         ret = createTACPkgRoot(pkg, flags);
174                         if (ret == TAC_ERROR_INVALID_PACKAGE) {
175                                 fprintf(stderr, "Failed to get root path from [%s]\n", pkg.c_str());
176                                 return -1;
177                         } else if (ret != TAC_ERROR_NONE) {
178                                 fprintf(stderr, "Failed to generate symbolic link file [%s]\n", pkg.c_str());
179                                 return -1;
180                         }
181                 }
182         } else if (pkgDllMode) {
183                 int ret = createNiDllUnderPkgRoot(args[0], args[1], flags);
184                 if (ret == NI_ERROR_INVALID_PACKAGE) {
185                         fprintf(stderr, "Failed to get root path from [%s]\n", args[0].c_str());
186                         return -1;
187                 } else if (ret == NI_ERROR_ALREADY_EXIST) {
188                         // skip for already exist case
189                         return -1;
190                 } else if (ret != NI_ERROR_NONE) {
191                         fprintf(stderr, "Failed to generate NI file [%s]\n", args[1].c_str());
192                         return -1;
193                 }
194         } else if (rmPkgMode) {
195                 for (const std::string pkg : args) {
196                         int ret = removeNiUnderPkgRoot(pkg);
197                         if (ret == NI_ERROR_INVALID_PACKAGE) {
198                                 fprintf(stderr, "Failed to get root path from [%s]\n", pkg.c_str());
199                                 return -1;
200                         } else if (ret != NI_ERROR_NONE) {
201                                 fprintf(stderr, "Failed to remove dlls for given package [%s]\n", pkg.c_str());
202                                 return -1;
203                         }
204                         ret = resetTACPackage(pkg);
205                         if (ret == TAC_ERROR_INVALID_PACKAGE) {
206                                 fprintf(stderr, "Failed to get root path from [%s]\n", pkg.c_str());
207                                 return -1;
208                         } else if (ret != TAC_ERROR_NONE) {
209                                 fprintf(stderr, "Failed to remove symlink for given package [%s]\n", pkg.c_str());
210                                 return -1;
211                         }
212                 }
213         } else if (dllMode) {
214                 // donot return error code for generation failure.
215                 // we have to run crossgen for all input dlls.
216                 for (const std::string dll : args) {
217                         int ret = createNiDll(dll, flags);
218                         if (ret == NI_ERROR_ALREADY_EXIST) {
219                                 // skip for already exist case
220                         } else if (ret != NI_ERROR_NONE) {
221                                 fprintf(stderr, "Failed to generate NI file [%s]\n", dll.c_str());
222                         }
223                 }
224         } else if (dirMode) {
225                 createNiUnderDirs(args.data(), args.size(), flags);
226         } else if (disableTacMode) {
227                 for (const std::string pkg : args) {
228                         int ret = disableTACPackage(pkg);
229                         if (ret == TAC_ERROR_INVALID_PACKAGE) {
230                                 fprintf(stderr, "Failed to get root path from [%s]\n", pkg.c_str());
231                                 return -1;
232                         } else if (ret != TAC_ERROR_NONE) {
233                                 fprintf(stderr, "Failed to disable tac [%s]\n", pkg.c_str());
234                                 return -1;
235                         }
236                 }
237         } else if (enableTacMode) {
238                 for (const std::string pkg : args) {
239                         int ret = enableTACPackage(pkg);
240                         if (ret == TAC_ERROR_INVALID_PACKAGE) {
241                                 fprintf(stderr, "Failed to get root path from [%s]\n", pkg.c_str());
242                                 return -1;
243                         } else if (ret != TAC_ERROR_NONE) {
244                                 fprintf(stderr, "Failed to enable tac [%s]\n", pkg.c_str());
245                                 return -1;
246                         }
247                 }
248         }
249         return 0;
250 }