Adam Lesinski | 6f6ceb7 | 2014-11-14 14:48:12 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2015 The Android Open Source Project |
| 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 | |
Adam Lesinski | efeb7af | 2017-08-02 14:57:43 -0700 | [diff] [blame] | 17 | #ifdef _WIN32 |
| 18 | // clang-format off |
| 19 | #include <windows.h> |
| 20 | #include <shellapi.h> |
| 21 | // clang-format on |
| 22 | #endif |
| 23 | |
Adam Lesinski | 6f6ceb7 | 2014-11-14 14:48:12 -0800 | [diff] [blame] | 24 | #include <iostream> |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 25 | #include <vector> |
Adam Lesinski | 6f6ceb7 | 2014-11-14 14:48:12 -0800 | [diff] [blame] | 26 | |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 27 | #include "android-base/stringprintf.h" |
Adam Lesinski | efeb7af | 2017-08-02 14:57:43 -0700 | [diff] [blame] | 28 | #include "android-base/utf8.h" |
Adam Lesinski | d5083f6 | 2017-01-16 15:07:21 -0800 | [diff] [blame] | 29 | #include "androidfw/StringPiece.h" |
Adam Lesinski | ce5e56e | 2016-10-21 17:56:45 -0700 | [diff] [blame] | 30 | |
Chris Warrington | 820d72a | 2017-04-27 15:27:01 +0100 | [diff] [blame] | 31 | #include "Diagnostics.h" |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 32 | #include "util/Files.h" |
| 33 | #include "util/Util.h" |
| 34 | |
| 35 | using ::android::StringPiece; |
| 36 | using ::android::base::StringPrintf; |
Chris Warrington | 820d72a | 2017-04-27 15:27:01 +0100 | [diff] [blame] | 37 | |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 38 | namespace aapt { |
Adam Lesinski | 5886a92 | 2015-04-15 20:29:22 -0700 | [diff] [blame] | 39 | |
Adam Lesinski | 0368ebf | 2016-07-26 12:55:51 -0700 | [diff] [blame] | 40 | // DO NOT UPDATE, this is more of a marketing version. |
| 41 | static const char* sMajorVersion = "2"; |
| 42 | |
| 43 | // Update minor version whenever a feature or flag is added. |
Adam Lesinski | 3420eed | 2017-09-13 14:46:00 -0700 | [diff] [blame^] | 44 | static const char* sMinorVersion = "19"; |
Adam Lesinski | 0368ebf | 2016-07-26 12:55:51 -0700 | [diff] [blame] | 45 | |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 46 | static void PrintVersion() { |
| 47 | std::cerr << StringPrintf("Android Asset Packaging Tool (aapt) %s:%s", sMajorVersion, |
| 48 | sMinorVersion) |
| 49 | << std::endl; |
Adam Lesinski | 0368ebf | 2016-07-26 12:55:51 -0700 | [diff] [blame] | 50 | } |
| 51 | |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 52 | static void PrintUsage() { |
| 53 | std::cerr << "\nusage: aapt2 [compile|link|dump|diff|optimize|version] ..." << std::endl; |
| 54 | } |
| 55 | |
| 56 | extern int Compile(const std::vector<StringPiece>& args, IDiagnostics* diagnostics); |
| 57 | extern int Link(const std::vector<StringPiece>& args, IDiagnostics* diagnostics); |
| 58 | extern int Dump(const std::vector<StringPiece>& args); |
| 59 | extern int Diff(const std::vector<StringPiece>& args); |
| 60 | extern int Optimize(const std::vector<StringPiece>& args); |
| 61 | |
| 62 | static int ExecuteCommand(const StringPiece& command, const std::vector<StringPiece>& args, |
| 63 | IDiagnostics* diagnostics) { |
| 64 | if (command == "compile" || command == "c") { |
| 65 | return Compile(args, diagnostics); |
| 66 | } else if (command == "link" || command == "l") { |
| 67 | return Link(args, diagnostics); |
| 68 | } else if (command == "dump" || command == "d") { |
| 69 | return Dump(args); |
| 70 | } else if (command == "diff") { |
| 71 | return Diff(args); |
| 72 | } else if (command == "optimize") { |
| 73 | return Optimize(args); |
| 74 | } else if (command == "version") { |
| 75 | PrintVersion(); |
| 76 | return 0; |
| 77 | } |
| 78 | diagnostics->Error(DiagMessage() << "unknown command '" << command << "'"); |
| 79 | return -1; |
| 80 | } |
| 81 | |
| 82 | static void RunDaemon(IDiagnostics* diagnostics) { |
| 83 | std::cout << "Ready" << std::endl; |
| 84 | |
Adam Lesinski | 43f6f77 | 2017-08-24 15:17:05 -0700 | [diff] [blame] | 85 | // Run in daemon mode. The first line of input is the command. This can be 'quit' which ends |
| 86 | // the daemon mode. Each subsequent line is a single parameter to the command. The end of a |
| 87 | // invocation is signaled by providing an empty line. At any point, an EOF signal or the |
| 88 | // command 'quit' will end the daemon mode. |
| 89 | while (true) { |
| 90 | std::vector<std::string> raw_args; |
| 91 | for (std::string line; std::getline(std::cin, line) && !line.empty();) { |
| 92 | raw_args.push_back(line); |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 93 | } |
| 94 | |
Adam Lesinski | 43f6f77 | 2017-08-24 15:17:05 -0700 | [diff] [blame] | 95 | if (!std::cin) { |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 96 | break; |
| 97 | } |
| 98 | |
Adam Lesinski | 43f6f77 | 2017-08-24 15:17:05 -0700 | [diff] [blame] | 99 | // An empty command does nothing. |
| 100 | if (raw_args.empty()) { |
| 101 | continue; |
| 102 | } |
| 103 | |
| 104 | if (raw_args[0] == "quit") { |
| 105 | break; |
| 106 | } |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 107 | |
| 108 | std::vector<StringPiece> args; |
Adam Lesinski | 43f6f77 | 2017-08-24 15:17:05 -0700 | [diff] [blame] | 109 | args.insert(args.end(), ++raw_args.begin(), raw_args.end()); |
| 110 | int ret = ExecuteCommand(raw_args[0], args, diagnostics); |
Izabela Orlowska | 4a4ebe2 | 2017-08-24 16:19:45 +0100 | [diff] [blame] | 111 | if (ret != 0) { |
| 112 | std::cerr << "Error" << std::endl; |
| 113 | } |
| 114 | std::cerr << "Done" << std::endl; |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 115 | } |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 116 | std::cout << "Exiting daemon" << std::endl; |
| 117 | } |
Adam Lesinski | 6f6ceb7 | 2014-11-14 14:48:12 -0800 | [diff] [blame] | 118 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 119 | } // namespace aapt |
Adam Lesinski | 330edcd | 2015-05-04 17:40:56 -0700 | [diff] [blame] | 120 | |
Adam Lesinski | efeb7af | 2017-08-02 14:57:43 -0700 | [diff] [blame] | 121 | int MainImpl(int argc, char** argv) { |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 122 | if (argc < 2) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 123 | std::cerr << "no command specified\n"; |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 124 | aapt::PrintUsage(); |
| 125 | return -1; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 126 | } |
| 127 | |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 128 | argv += 1; |
| 129 | argc -= 1; |
| 130 | |
| 131 | aapt::StdErrDiagnostics diagnostics; |
| 132 | |
| 133 | // Collect the arguments starting after the program name and command name. |
| 134 | std::vector<StringPiece> args; |
| 135 | for (int i = 1; i < argc; i++) { |
| 136 | args.push_back(argv[i]); |
| 137 | } |
| 138 | |
| 139 | const StringPiece command(argv[0]); |
| 140 | if (command != "daemon" && command != "m") { |
| 141 | // Single execution. |
| 142 | const int result = aapt::ExecuteCommand(command, args, &diagnostics); |
| 143 | if (result < 0) { |
| 144 | aapt::PrintUsage(); |
| 145 | } |
| 146 | return result; |
| 147 | } |
| 148 | |
| 149 | aapt::RunDaemon(&diagnostics); |
| 150 | return 0; |
Adam Lesinski | 6f6ceb7 | 2014-11-14 14:48:12 -0800 | [diff] [blame] | 151 | } |
Adam Lesinski | efeb7af | 2017-08-02 14:57:43 -0700 | [diff] [blame] | 152 | |
| 153 | int main(int argc, char** argv) { |
| 154 | #ifdef _WIN32 |
| 155 | LPWSTR* wide_argv = CommandLineToArgvW(GetCommandLineW(), &argc); |
| 156 | CHECK(wide_argv != nullptr) << "invalid command line parameters passed to process"; |
| 157 | |
| 158 | std::vector<std::string> utf8_args; |
| 159 | for (int i = 0; i < argc; i++) { |
| 160 | std::string utf8_arg; |
| 161 | if (!::android::base::WideToUTF8(wide_argv[i], &utf8_arg)) { |
| 162 | std::cerr << "error converting input arguments to UTF-8" << std::endl; |
| 163 | return 1; |
| 164 | } |
| 165 | utf8_args.push_back(std::move(utf8_arg)); |
| 166 | } |
| 167 | LocalFree(wide_argv); |
| 168 | |
| 169 | std::unique_ptr<char* []> utf8_argv(new char*[utf8_args.size()]); |
| 170 | for (int i = 0; i < argc; i++) { |
| 171 | utf8_argv[i] = const_cast<char*>(utf8_args[i].c_str()); |
| 172 | } |
| 173 | argv = utf8_argv.get(); |
| 174 | #endif |
| 175 | return MainImpl(argc, argv); |
| 176 | } |