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" |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 32 | #include "cmd/Command.h" |
| 33 | #include "cmd/Compile.h" |
| 34 | #include "cmd/Convert.h" |
| 35 | #include "cmd/Diff.h" |
| 36 | #include "cmd/Dump.h" |
| 37 | #include "cmd/Link.h" |
| 38 | #include "cmd/Optimize.h" |
Ryan Mitchell | 214846d | 2018-09-19 16:57:01 -0700 | [diff] [blame] | 39 | #include "io/FileStream.h" |
Fabien Sanglard | 2d34e76 | 2019-02-21 15:13:29 -0800 | [diff] [blame] | 40 | #include "trace/TraceBuffer.h" |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 41 | #include "util/Files.h" |
| 42 | #include "util/Util.h" |
| 43 | |
| 44 | using ::android::StringPiece; |
| 45 | using ::android::base::StringPrintf; |
Chris Warrington | 820d72a | 2017-04-27 15:27:01 +0100 | [diff] [blame] | 46 | |
Adam Lesinski | 1ab598f | 2015-08-14 14:26:04 -0700 | [diff] [blame] | 47 | namespace aapt { |
Adam Lesinski | 5886a92 | 2015-04-15 20:29:22 -0700 | [diff] [blame] | 48 | |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 49 | /** Prints the version information of AAPT2. */ |
| 50 | class VersionCommand : public Command { |
| 51 | public: |
| 52 | explicit VersionCommand() : Command("version") { |
| 53 | SetDescription("Prints the version of aapt."); |
| 54 | } |
Adam Lesinski | 0368ebf | 2016-07-26 12:55:51 -0700 | [diff] [blame] | 55 | |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 56 | int Action(const std::vector<std::string>& /* args */) override { |
Ryan Mitchell | 34039b2 | 2019-03-18 08:57:47 -0700 | [diff] [blame] | 57 | std::cerr << StringPrintf("%s %s", util::GetToolName(), util::GetToolFingerprint().c_str()) |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 58 | << std::endl; |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 59 | return 0; |
| 60 | } |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 61 | }; |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 62 | |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 63 | /** The main entry point of AAPT. */ |
| 64 | class MainCommand : public Command { |
| 65 | public: |
Ryan Mitchell | 214846d | 2018-09-19 16:57:01 -0700 | [diff] [blame] | 66 | explicit MainCommand(text::Printer* printer, IDiagnostics* diagnostics) |
| 67 | : Command("aapt2"), diagnostics_(diagnostics) { |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 68 | AddOptionalSubcommand(util::make_unique<CompileCommand>(diagnostics)); |
| 69 | AddOptionalSubcommand(util::make_unique<LinkCommand>(diagnostics)); |
Ryan Mitchell | 214846d | 2018-09-19 16:57:01 -0700 | [diff] [blame] | 70 | AddOptionalSubcommand(util::make_unique<DumpCommand>(printer, diagnostics)); |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 71 | AddOptionalSubcommand(util::make_unique<DiffCommand>()); |
| 72 | AddOptionalSubcommand(util::make_unique<OptimizeCommand>()); |
| 73 | AddOptionalSubcommand(util::make_unique<ConvertCommand>()); |
| 74 | AddOptionalSubcommand(util::make_unique<VersionCommand>()); |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 75 | } |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 76 | |
| 77 | int Action(const std::vector<std::string>& args) override { |
| 78 | if (args.size() == 0) { |
| 79 | diagnostics_->Error(DiagMessage() << "no subcommand specified"); |
| 80 | } else { |
| 81 | diagnostics_->Error(DiagMessage() << "unknown subcommand '" << args[0] << "'"); |
| 82 | } |
| 83 | |
| 84 | Usage(&std::cerr); |
| 85 | return -1; |
| 86 | } |
| 87 | |
| 88 | private: |
| 89 | IDiagnostics* diagnostics_; |
| 90 | }; |
| 91 | |
| 92 | /* |
| 93 | * Run in daemon mode. The first line of input is the command. This can be 'quit' which ends |
| 94 | * the daemon mode. Each subsequent line is a single parameter to the command. The end of a |
| 95 | * invocation is signaled by providing an empty line. At any point, an EOF signal or the |
| 96 | * command 'quit' will end the daemon mode. |
| 97 | */ |
| 98 | class DaemonCommand : public Command { |
| 99 | public: |
Ryan Mitchell | 214846d | 2018-09-19 16:57:01 -0700 | [diff] [blame] | 100 | explicit DaemonCommand(io::FileOutputStream* out, IDiagnostics* diagnostics) |
| 101 | : Command("daemon", "m"), out_(out), diagnostics_(diagnostics) { |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 102 | SetDescription("Runs aapt in daemon mode. Each subsequent line is a single parameter to the\n" |
| 103 | "command. The end of an invocation is signaled by providing an empty line."); |
Fabien Sanglard | 2d34e76 | 2019-02-21 15:13:29 -0800 | [diff] [blame] | 104 | AddOptionalFlag("--trace_folder", "Generate systrace json trace fragment to specified folder.", |
| 105 | &trace_folder_); |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 106 | } |
| 107 | |
Fabien Sanglard | 2d34e76 | 2019-02-21 15:13:29 -0800 | [diff] [blame] | 108 | int Action(const std::vector<std::string>& arguments) override { |
| 109 | TRACE_FLUSH_ARGS(trace_folder_ ? trace_folder_.value() : "", "daemon", arguments); |
Ryan Mitchell | 214846d | 2018-09-19 16:57:01 -0700 | [diff] [blame] | 110 | text::Printer printer(out_); |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 111 | std::cout << "Ready" << std::endl; |
| 112 | |
| 113 | while (true) { |
| 114 | std::vector<std::string> raw_args; |
| 115 | for (std::string line; std::getline(std::cin, line) && !line.empty();) { |
| 116 | raw_args.push_back(line); |
| 117 | } |
| 118 | |
| 119 | if (!std::cin) { |
| 120 | break; |
| 121 | } |
| 122 | |
| 123 | // An empty command does nothing. |
| 124 | if (raw_args.empty()) { |
| 125 | continue; |
| 126 | } |
| 127 | |
| 128 | // End the dameon |
| 129 | if (raw_args[0] == "quit") { |
| 130 | break; |
| 131 | } |
| 132 | |
| 133 | std::vector<StringPiece> args; |
| 134 | args.insert(args.end(), raw_args.begin(), raw_args.end()); |
Ryan Mitchell | 214846d | 2018-09-19 16:57:01 -0700 | [diff] [blame] | 135 | int result = MainCommand(&printer, diagnostics_).Execute(args, &std::cerr); |
| 136 | out_->Flush(); |
| 137 | if (result != 0) { |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 138 | std::cerr << "Error" << std::endl; |
| 139 | } |
| 140 | std::cerr << "Done" << std::endl; |
| 141 | } |
| 142 | std::cout << "Exiting daemon" << std::endl; |
| 143 | |
| 144 | return 0; |
| 145 | } |
| 146 | |
| 147 | private: |
Ryan Mitchell | 214846d | 2018-09-19 16:57:01 -0700 | [diff] [blame] | 148 | io::FileOutputStream* out_; |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 149 | IDiagnostics* diagnostics_; |
Fabien Sanglard | 2d34e76 | 2019-02-21 15:13:29 -0800 | [diff] [blame] | 150 | Maybe<std::string> trace_folder_; |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 151 | }; |
Adam Lesinski | 6f6ceb7 | 2014-11-14 14:48:12 -0800 | [diff] [blame] | 152 | |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 153 | } // namespace aapt |
Adam Lesinski | 330edcd | 2015-05-04 17:40:56 -0700 | [diff] [blame] | 154 | |
Adam Lesinski | efeb7af | 2017-08-02 14:57:43 -0700 | [diff] [blame] | 155 | int MainImpl(int argc, char** argv) { |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 156 | if (argc < 1) { |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 157 | return -1; |
Adam Lesinski | cacb28f | 2016-10-19 12:18:14 -0700 | [diff] [blame] | 158 | } |
| 159 | |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 160 | // Collect the arguments starting after the program name and command name. |
| 161 | std::vector<StringPiece> args; |
| 162 | for (int i = 1; i < argc; i++) { |
| 163 | args.push_back(argv[i]); |
| 164 | } |
| 165 | |
Ryan Mitchell | 214846d | 2018-09-19 16:57:01 -0700 | [diff] [blame] | 166 | // Use a smaller buffer so that there is less latency for printing to stdout. |
| 167 | constexpr size_t kStdOutBufferSize = 1024u; |
| 168 | aapt::io::FileOutputStream fout(STDOUT_FILENO, kStdOutBufferSize); |
| 169 | aapt::text::Printer printer(&fout); |
Adam Lesinski | 448a15c | 2017-07-25 10:59:26 -0700 | [diff] [blame] | 170 | |
Ryan Mitchell | 214846d | 2018-09-19 16:57:01 -0700 | [diff] [blame] | 171 | aapt::StdErrDiagnostics diagnostics; |
| 172 | auto main_command = new aapt::MainCommand(&printer, &diagnostics); |
| 173 | |
| 174 | // Add the daemon subcommand here so it cannot be called while executing the daemon |
| 175 | main_command->AddOptionalSubcommand( |
| 176 | aapt::util::make_unique<aapt::DaemonCommand>(&fout, &diagnostics)); |
Ryan Mitchell | 833a1a6 | 2018-07-10 13:51:36 -0700 | [diff] [blame] | 177 | return main_command->Execute(args, &std::cerr); |
Adam Lesinski | 6f6ceb7 | 2014-11-14 14:48:12 -0800 | [diff] [blame] | 178 | } |
Adam Lesinski | efeb7af | 2017-08-02 14:57:43 -0700 | [diff] [blame] | 179 | |
| 180 | int main(int argc, char** argv) { |
| 181 | #ifdef _WIN32 |
| 182 | LPWSTR* wide_argv = CommandLineToArgvW(GetCommandLineW(), &argc); |
| 183 | CHECK(wide_argv != nullptr) << "invalid command line parameters passed to process"; |
| 184 | |
| 185 | std::vector<std::string> utf8_args; |
| 186 | for (int i = 0; i < argc; i++) { |
| 187 | std::string utf8_arg; |
| 188 | if (!::android::base::WideToUTF8(wide_argv[i], &utf8_arg)) { |
| 189 | std::cerr << "error converting input arguments to UTF-8" << std::endl; |
| 190 | return 1; |
| 191 | } |
| 192 | utf8_args.push_back(std::move(utf8_arg)); |
| 193 | } |
| 194 | LocalFree(wide_argv); |
| 195 | |
| 196 | std::unique_ptr<char* []> utf8_argv(new char*[utf8_args.size()]); |
| 197 | for (int i = 0; i < argc; i++) { |
| 198 | utf8_argv[i] = const_cast<char*>(utf8_args[i].c_str()); |
| 199 | } |
| 200 | argv = utf8_argv.get(); |
| 201 | #endif |
| 202 | return MainImpl(argc, argv); |
| 203 | } |