NativeLauncher/tool/nitool.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 "utils.h"
  18 #include "ni_common.h"
  19
  20 #include <cstdio>
  21 #include <cstring>
  22
  23 #include <algorithm>
  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         return list;
  34 }
  35
  36 static bool cmdOptionExists(char** begin, char** end, const std::string& option)
  37 {
  38         return std::find(begin, end, option) != end;
  39 }
  40
  41 static void help(const char *argv0)
  42 {
  43         const char* helpDesc =
  44                 "%s is deprecated. Please use dotnettool instead.\n"
  45                 "Example:\n"
  46                 "   # dotnettool --help\n\n";
  47         printf(helpDesc, argv0);
  48 }
  49
  50 int main(int argc, char* argv[])
  51 {
  52         DWORD flags = 0;
  53         bool pkgMode = false;
  54         bool dllMode = false;
  55         bool dirMode = false;
  56         bool rmPkgMode = false;
  57         bool pkgDllMode = false;
  58
  59         NICommonOption option = {std::string(), std::string(), std::string()};
  60         if (initNICommon(&option) != NI_ERROR_NONE) {
  61                 return -1;
  62         }
  63
  64         // Parse optional switches first.
  65         if (cmdOptionExists(argv, argv+argc, "--r2r")) {
  66                 flags |= NI_FLAGS_ENABLER2R;
  67         }
  68         if (cmdOptionExists(argv, argv+argc, "--compatibility")) {
  69                 flags |= NI_FLAGS_COMPATIBILITY;
  70         }
  71         if (cmdOptionExists(argv, argv+argc, "--instrument")) {
  72                 flags |= NI_FLAGS_INSTRUMENT;
  73         }
  74         if (cmdOptionExists(argv, argv+argc, "--verbose")) {
  75                 flags |= NI_FLAGS_VERBOSE;
  76         }
  77
  78         // The following commands are mutually exclusive.
  79         if (cmdOptionExists(argv, argv+argc, "--help")) {
  80                 help(argv[0]);
  81                 return 0;
  82         } else if (cmdOptionExists(argv, argv+argc, "--system")) {
  83                 createNIPlatform(flags);
  84                 return 0;
  85         } else if (cmdOptionExists(argv, argv+argc, "--dll")) {
  86                 dllMode = true;
  87         } else if (cmdOptionExists(argv, argv+argc, "--pkg")) {
  88                 pkgMode = true;
  89         } else if (cmdOptionExists(argv, argv+argc, "--dir")) {
  90                 dirMode = true;
  91         } else if (cmdOptionExists(argv, argv+argc, "--reset-system")) {
  92                 removeNIPlatform();
  93                 return 0;
  94         } else if (cmdOptionExists(argv, argv+argc, "--reset-pkg")) {
  95                 rmPkgMode = true;
  96         } else if (cmdOptionExists(argv, argv+argc, "--regen-all-app")) {
  97                 regenerateAppNI(flags);
  98                 return 0;
  99         } else if (cmdOptionExists(argv, argv+argc, "--pkg-dll")) {
 100                 pkgDllMode = true;
 101         } else {
 102                 help(argv[0]);
 103                 return 0;
 104         }
 105
 106         std::vector<std::string> args = getCmdArgs(argv, argv+argc);
 107
 108         if (args.size() < 1) {
 109                 if (pkgMode)
 110                         fprintf(stderr, "Package name is missing.\n");
 111                 else if (dllMode)
 112                         fprintf(stderr, "DLL path is missing.\n");
 113                 help(argv[0]);
 114                 return -1;
 115         }
 116
 117         if (pkgMode) {
 118                 for (const std::string pkg : args) {
 119                         // if there is AOTed dlls under package root, that is skiped.
 120                         int ret = createNIUnderPkgRoot(pkg, flags);
 121                         if (ret == NI_ERROR_INVALID_PACKAGE) {
 122                                 fprintf(stderr, "Failed to get root path from [%s]\n", pkg.c_str());
 123                                 return -1;
 124                         } else if (ret != NI_ERROR_NONE) {
 125                                 fprintf(stderr, "Failed to generate NI file [%s]\n", pkg.c_str());
 126                                 return -1;
 127                         }
 128                 }
 129         } else if (pkgDllMode) {
 130                 int ret = createNIDllUnderPkgRoot(args[0], args[1], flags);
 131                 if (ret == NI_ERROR_INVALID_PACKAGE) {
 132                         fprintf(stderr, "Failed to get root path from [%s]\n", args[0].c_str());
 133                         return -1;
 134                 } else if (ret == NI_ERROR_ALREADY_EXIST) {
 135                         // skip for already exist case
 136                         return -1;
 137                 } else if (ret != NI_ERROR_NONE) {
 138                         fprintf(stderr, "Failed to generate NI file [%s]\n", args[1].c_str());
 139                         return -1;
 140                 }
 141         } else if (rmPkgMode) {
 142                 for (const std::string pkg : args) {
 143                         int ret = removeNIUnderPkgRoot(pkg);
 144                         if (ret == NI_ERROR_INVALID_PACKAGE) {
 145                                 fprintf(stderr, "Failed to get root path from [%s]\n", pkg.c_str());
 146                                 return -1;
 147                         } else if (ret != NI_ERROR_NONE) {
 148                                 fprintf(stderr, "Failed to remove dlls for given package [%s]\n", pkg.c_str());
 149                                 return -1;
 150                         }
 151                 }
 152         } else if (dllMode) {
 153                 // donot return error code for generation failure.
 154                 // we have to run crossgen for all input dlls.
 155                 for (const std::string dll : args) {
 156                         int ret = createNIDll(dll, flags);
 157                         if (ret == NI_ERROR_ALREADY_EXIST) {
 158                                 // skip for already exist case
 159                         } else if (ret != NI_ERROR_NONE) {
 160                                 fprintf(stderr, "Failed to generate NI file [%s]\n", dll.c_str());
 161                         }
 162                 }
 163         } else if (dirMode) {
 164                 createNIUnderDirs(args.data(), args.size(), flags);
 165         }
 166
 167         return 0;
 168 }