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 | f903d5f | 2017-07-10 04:19:28 -0700 | [diff] [blame] | 44 | static const char* sMinorVersion = "18"; |
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 | |
| 85 | // Run in daemon mode. Each line of input from stdin is treated as a command line argument |
| 86 | // invocation. This means we need to split the line into a vector of args. |
| 87 | for (std::string line; std::getline(std::cin, line);) { |
| 88 | const util::Tokenizer tokenizer = util::Tokenize(line, file::sPathSep); |
| 89 | auto token_iter = tokenizer.begin(); |
| 90 | if (token_iter == tokenizer.end()) { |
| 91 | diagnostics->Error(DiagMessage() << "no command"); |
| 92 | continue; |
| 93 | } |
| 94 | |
| 95 | const StringPiece command(*token_iter); |
| 96 | if (command == "quit") { |
| 97 | break; |
| 98 | } |
| 99 | |
| 100 | ++token_iter; |
| 101 | |
| 102 | std::vector<StringPiece> args; |
| 103 | args.insert(args.end(), token_iter, tokenizer.end()); |
| 104 | ExecuteCommand(command, args, diagnostics); |
| 105 | std::cout << "Done" << std::endl; |
| 106 | } |
| 107 | |
| 108 | std::cout << "Exiting daemon" << std::endl; |
| 109 | } |
Adam Lesinski | 6f6ceb7 | 2014-11-14 14:48:12 -0800 | [diff] [blame] | 110 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 111 | } // namespace aapt |
Adam Lesinski | 330edcd | 2015-05-04 17:40:56 -0700 | [diff] [blame] | 112 | |
Adam Lesinski | efeb7af | 2017-08-02 14:57:43 -0700 | [diff] [blame] | 113 | int MainImpl(int argc, char** argv) { |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame^] | 114 | if (argc < 2) { |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 115 | std::cerr << "no command specified\n"; |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame^] | 116 | aapt::PrintUsage(); |
| 117 | return -1; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 118 | } |
| 119 | |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame^] | 120 | argv += 1; |
| 121 | argc -= 1; |
| 122 | |
| 123 | aapt::StdErrDiagnostics diagnostics; |
| 124 | |
| 125 | // Collect the arguments starting after the program name and command name. |
| 126 | std::vector<StringPiece> args; |
| 127 | for (int i = 1; i < argc; i++) { |
| 128 | args.push_back(argv[i]); |
| 129 | } |
| 130 | |
| 131 | const StringPiece command(argv[0]); |
| 132 | if (command != "daemon" && command != "m") { |
| 133 | // Single execution. |
| 134 | const int result = aapt::ExecuteCommand(command, args, &diagnostics); |
| 135 | if (result < 0) { |
| 136 | aapt::PrintUsage(); |
| 137 | } |
| 138 | return result; |
| 139 | } |
| 140 | |
| 141 | aapt::RunDaemon(&diagnostics); |
| 142 | return 0; |
Adam Lesinski | 6f6ceb7 | 2014-11-14 14:48:12 -0800 | [diff] [blame] | 143 | } |
Adam Lesinski | efeb7af | 2017-08-02 14:57:43 -0700 | [diff] [blame] | 144 | |
| 145 | int main(int argc, char** argv) { |
| 146 | #ifdef _WIN32 |
| 147 | LPWSTR* wide_argv = CommandLineToArgvW(GetCommandLineW(), &argc); |
| 148 | CHECK(wide_argv != nullptr) << "invalid command line parameters passed to process"; |
| 149 | |
| 150 | std::vector<std::string> utf8_args; |
| 151 | for (int i = 0; i < argc; i++) { |
| 152 | std::string utf8_arg; |
| 153 | if (!::android::base::WideToUTF8(wide_argv[i], &utf8_arg)) { |
| 154 | std::cerr << "error converting input arguments to UTF-8" << std::endl; |
| 155 | return 1; |
| 156 | } |
| 157 | utf8_args.push_back(std::move(utf8_arg)); |
| 158 | } |
| 159 | LocalFree(wide_argv); |
| 160 | |
| 161 | std::unique_ptr<char* []> utf8_argv(new char*[utf8_args.size()]); |
| 162 | for (int i = 0; i < argc; i++) { |
| 163 | utf8_argv[i] = const_cast<char*>(utf8_args[i].c_str()); |
| 164 | } |
| 165 | argv = utf8_argv.get(); |
| 166 | #endif |
| 167 | return MainImpl(argc, argv); |
| 168 | } |