blob: 5fbce664126732d8eaebafd5b64d12952fe1589c [file] [log] [blame]
Calin Juravle2e2db782016-02-23 12:00:03 +00001/*
2 * Copyright (C) 2016 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
Andreas Gampe8cf9cb32017-07-19 09:28:38 -070017#include <errno.h>
Calin Juravle2e2db782016-02-23 12:00:03 +000018#include <stdio.h>
19#include <stdlib.h>
20#include <sys/file.h>
Calin Juravlee10c1e22018-01-26 20:10:15 -080021#include <sys/param.h>
Calin Juravle2e2db782016-02-23 12:00:03 +000022#include <unistd.h>
23
David Sehr7c80f2d2017-02-07 16:47:58 -080024#include <fstream>
Calin Juravle876f3502016-03-24 16:16:34 +000025#include <iostream>
David Sehr7c80f2d2017-02-07 16:47:58 -080026#include <set>
Calin Juravle2e2db782016-02-23 12:00:03 +000027#include <string>
David Sehr7c80f2d2017-02-07 16:47:58 -080028#include <unordered_set>
Calin Juravle2e2db782016-02-23 12:00:03 +000029#include <vector>
30
Andreas Gampe46ee31b2016-12-14 10:11:49 -080031#include "android-base/stringprintf.h"
Andreas Gampe9186ced2016-12-12 14:28:21 -080032#include "android-base/strings.h"
33
Calin Juravle2e2db782016-02-23 12:00:03 +000034#include "base/dumpable.h"
Andreas Gampe57943812017-12-06 21:39:13 -080035#include "base/logging.h" // For InitLogging.
David Sehr671af6c2018-05-17 11:00:35 -070036#include "base/mem_map.h"
Calin Juravle2e2db782016-02-23 12:00:03 +000037#include "base/scoped_flock.h"
38#include "base/stringpiece.h"
Calin Juravle2e2db782016-02-23 12:00:03 +000039#include "base/time_utils.h"
40#include "base/unix_file/fd_file.h"
David Sehrc431b9d2018-03-02 12:01:51 -080041#include "base/utils.h"
David Sehr79e26072018-04-06 17:58:50 -070042#include "base/zip_archive.h"
Mathieu Chartier2f794552017-06-19 10:58:08 -070043#include "boot_image_profile.h"
Mathieu Chartier818cb802018-05-11 05:30:16 +000044#include "dex/art_dex_file_loader.h"
David Sehr312f3b22018-03-19 08:39:26 -070045#include "dex/bytecode_utils.h"
Mathieu Chartier20f49922018-05-24 16:04:17 -070046#include "dex/class_accessor-inl.h"
David Sehr9e734c72018-01-04 17:56:19 -080047#include "dex/code_item_accessors-inl.h"
48#include "dex/dex_file.h"
49#include "dex/dex_file_loader.h"
50#include "dex/dex_file_types.h"
David Sehr312f3b22018-03-19 08:39:26 -070051#include "dex/type_reference.h"
David Sehr82d046e2018-04-23 08:14:19 -070052#include "profile/profile_compilation_info.h"
Mathieu Chartierdbddc222017-05-24 12:04:13 -070053#include "profile_assistant.h"
Calin Juravle2e2db782016-02-23 12:00:03 +000054
55namespace art {
56
57static int original_argc;
58static char** original_argv;
59
60static std::string CommandLine() {
61 std::vector<std::string> command;
62 for (int i = 0; i < original_argc; ++i) {
63 command.push_back(original_argv[i]);
64 }
Andreas Gampe9186ced2016-12-12 14:28:21 -080065 return android::base::Join(command, ' ');
Calin Juravle2e2db782016-02-23 12:00:03 +000066}
67
David Sehr4fcdd6d2016-05-24 14:52:31 -070068static constexpr int kInvalidFd = -1;
69
70static bool FdIsValid(int fd) {
71 return fd != kInvalidFd;
72}
73
Calin Juravle2e2db782016-02-23 12:00:03 +000074static void UsageErrorV(const char* fmt, va_list ap) {
75 std::string error;
Andreas Gampe46ee31b2016-12-14 10:11:49 -080076 android::base::StringAppendV(&error, fmt, ap);
Calin Juravle2e2db782016-02-23 12:00:03 +000077 LOG(ERROR) << error;
78}
79
80static void UsageError(const char* fmt, ...) {
81 va_list ap;
82 va_start(ap, fmt);
83 UsageErrorV(fmt, ap);
84 va_end(ap);
85}
86
87NO_RETURN static void Usage(const char *fmt, ...) {
88 va_list ap;
89 va_start(ap, fmt);
90 UsageErrorV(fmt, ap);
91 va_end(ap);
92
93 UsageError("Command: %s", CommandLine().c_str());
94 UsageError("Usage: profman [options]...");
95 UsageError("");
David Sehr4fcdd6d2016-05-24 14:52:31 -070096 UsageError(" --dump-only: dumps the content of the specified profile files");
97 UsageError(" to standard output (default) in a human readable form.");
98 UsageError("");
David Sehr7c80f2d2017-02-07 16:47:58 -080099 UsageError(" --dump-output-to-fd=<number>: redirects --dump-only output to a file descriptor.");
100 UsageError("");
Mathieu Chartier34067262017-04-06 13:55:46 -0700101 UsageError(" --dump-classes-and-methods: dumps a sorted list of classes and methods that are");
102 UsageError(" in the specified profile file to standard output (default) in a human");
103 UsageError(" readable form. The output is valid input for --create-profile-from");
Calin Juravle876f3502016-03-24 16:16:34 +0000104 UsageError("");
Calin Juravle2e2db782016-02-23 12:00:03 +0000105 UsageError(" --profile-file=<filename>: specify profiler output file to use for compilation.");
106 UsageError(" Can be specified multiple time, in which case the data from the different");
107 UsageError(" profiles will be aggregated.");
108 UsageError("");
109 UsageError(" --profile-file-fd=<number>: same as --profile-file but accepts a file descriptor.");
110 UsageError(" Cannot be used together with --profile-file.");
111 UsageError("");
112 UsageError(" --reference-profile-file=<filename>: specify a reference profile.");
113 UsageError(" The data in this file will be compared with the data obtained by merging");
114 UsageError(" all the files specified with --profile-file or --profile-file-fd.");
115 UsageError(" If the exit code is EXIT_COMPILE then all --profile-file will be merged into");
116 UsageError(" --reference-profile-file. ");
117 UsageError("");
118 UsageError(" --reference-profile-file-fd=<number>: same as --reference-profile-file but");
119 UsageError(" accepts a file descriptor. Cannot be used together with");
120 UsageError(" --reference-profile-file.");
David Sehr7c80f2d2017-02-07 16:47:58 -0800121 UsageError("");
Calin Juravle7bcdb532016-06-07 16:14:47 +0100122 UsageError(" --generate-test-profile=<filename>: generates a random profile file for testing.");
123 UsageError(" --generate-test-profile-num-dex=<number>: number of dex files that should be");
124 UsageError(" included in the generated profile. Defaults to 20.");
Shubham Ajmerad704f0b2017-07-26 16:33:35 -0700125 UsageError(" --generate-test-profile-method-percentage=<number>: the percentage from the maximum");
Calin Juravle7bcdb532016-06-07 16:14:47 +0100126 UsageError(" number of methods that should be generated. Defaults to 5.");
Shubham Ajmerad704f0b2017-07-26 16:33:35 -0700127 UsageError(" --generate-test-profile-class-percentage=<number>: the percentage from the maximum");
Calin Juravle7bcdb532016-06-07 16:14:47 +0100128 UsageError(" number of classes that should be generated. Defaults to 5.");
Jeff Haof0a31f82017-03-27 15:50:37 -0700129 UsageError(" --generate-test-profile-seed=<number>: seed for random number generator used when");
130 UsageError(" generating random test profiles. Defaults to using NanoTime.");
Calin Juravle7bcdb532016-06-07 16:14:47 +0100131 UsageError("");
Mathieu Chartier34067262017-04-06 13:55:46 -0700132 UsageError(" --create-profile-from=<filename>: creates a profile from a list of classes and");
133 UsageError(" methods.");
David Sehr7c80f2d2017-02-07 16:47:58 -0800134 UsageError("");
David Sehr546d24f2016-06-02 10:46:19 -0700135 UsageError(" --dex-location=<string>: location string to use with corresponding");
136 UsageError(" apk-fd to find dex files");
David Sehr4fcdd6d2016-05-24 14:52:31 -0700137 UsageError("");
David Sehr546d24f2016-06-02 10:46:19 -0700138 UsageError(" --apk-fd=<number>: file descriptor containing an open APK to");
David Sehr4fcdd6d2016-05-24 14:52:31 -0700139 UsageError(" search for dex files");
David Sehr7c80f2d2017-02-07 16:47:58 -0800140 UsageError(" --apk-=<filename>: an APK to search for dex files");
David Brazdilf13ac7c2018-01-30 10:09:08 +0000141 UsageError(" --skip-apk-verification: do not attempt to verify APKs");
David Sehr4fcdd6d2016-05-24 14:52:31 -0700142 UsageError("");
Mathieu Chartier2f794552017-06-19 10:58:08 -0700143 UsageError(" --generate-boot-image-profile: Generate a boot image profile based on input");
144 UsageError(" profiles. Requires passing in dex files to inspect properties of classes.");
145 UsageError(" --boot-image-class-threshold=<value>: specify minimum number of class occurrences");
146 UsageError(" to include a class in the boot image profile. Default is 10.");
147 UsageError(" --boot-image-clean-class-threshold=<value>: specify minimum number of clean class");
148 UsageError(" occurrences to include a class in the boot image profile. A clean class is a");
149 UsageError(" class that doesn't have any static fields or native methods and is likely to");
150 UsageError(" remain clean in the image. Default is 3.");
Mathieu Chartier8eecddf2017-07-12 16:05:54 -0700151 UsageError(" --boot-image-sampled-method-threshold=<value>: minimum number of profiles a");
152 UsageError(" non-hot method needs to be in order to be hot in the output profile. The");
153 UsageError(" default is max int.");
Calin Juravle2dba0ab2018-01-22 19:22:24 -0800154 UsageError(" --copy-and-update-profile-key: if present, profman will copy the profile from");
155 UsageError(" the file passed with --profile-fd(file) to the profile passed with");
156 UsageError(" --reference-profile-fd(file) and update at the same time the profile-key");
157 UsageError(" of entries corresponding to the apks passed with --apk(-fd).");
Mathieu Chartier2f794552017-06-19 10:58:08 -0700158 UsageError("");
Calin Juravle2e2db782016-02-23 12:00:03 +0000159
160 exit(EXIT_FAILURE);
161}
162
Calin Juravle7bcdb532016-06-07 16:14:47 +0100163// Note: make sure you update the Usage if you change these values.
164static constexpr uint16_t kDefaultTestProfileNumDex = 20;
Shubham Ajmerad704f0b2017-07-26 16:33:35 -0700165static constexpr uint16_t kDefaultTestProfileMethodPercentage = 5;
166static constexpr uint16_t kDefaultTestProfileClassPercentage = 5;
Calin Juravle7bcdb532016-06-07 16:14:47 +0100167
Calin Juravlee0ac1152017-02-13 19:03:47 -0800168// Separators used when parsing human friendly representation of profiles.
Igor Murashkin2ffb7032017-11-08 13:35:21 -0800169static const std::string kMethodSep = "->"; // NOLINT [runtime/string] [4]
170static const std::string kMissingTypesMarker = "missing_types"; // NOLINT [runtime/string] [4]
171static const std::string kInvalidClassDescriptor = "invalid_class"; // NOLINT [runtime/string] [4]
172static const std::string kInvalidMethod = "invalid_method"; // NOLINT [runtime/string] [4]
173static const std::string kClassAllMethods = "*"; // NOLINT [runtime/string] [4]
Calin Juravlee0ac1152017-02-13 19:03:47 -0800174static constexpr char kProfileParsingInlineChacheSep = '+';
175static constexpr char kProfileParsingTypeSep = ',';
176static constexpr char kProfileParsingFirstCharInSignature = '(';
Mathieu Chartierea650f32017-05-24 12:04:13 -0700177static constexpr char kMethodFlagStringHot = 'H';
178static constexpr char kMethodFlagStringStartup = 'S';
179static constexpr char kMethodFlagStringPostStartup = 'P';
Calin Juravlee0ac1152017-02-13 19:03:47 -0800180
David Sehr671af6c2018-05-17 11:00:35 -0700181NO_RETURN static void Abort(const char* msg) {
182 LOG(ERROR) << msg;
183 exit(1);
184}
185
Calin Juravlee0ac1152017-02-13 19:03:47 -0800186// TODO(calin): This class has grown too much from its initial design. Split the functionality
187// into smaller, more contained pieces.
Calin Juravle2e2db782016-02-23 12:00:03 +0000188class ProfMan FINAL {
189 public:
190 ProfMan() :
David Sehr4fcdd6d2016-05-24 14:52:31 -0700191 reference_profile_file_fd_(kInvalidFd),
192 dump_only_(false),
Mathieu Chartier34067262017-04-06 13:55:46 -0700193 dump_classes_and_methods_(false),
Mathieu Chartier2f794552017-06-19 10:58:08 -0700194 generate_boot_image_profile_(false),
David Sehr4fcdd6d2016-05-24 14:52:31 -0700195 dump_output_to_fd_(kInvalidFd),
Calin Juravle7bcdb532016-06-07 16:14:47 +0100196 test_profile_num_dex_(kDefaultTestProfileNumDex),
Shubham Ajmerad704f0b2017-07-26 16:33:35 -0700197 test_profile_method_percerntage_(kDefaultTestProfileMethodPercentage),
198 test_profile_class_percentage_(kDefaultTestProfileClassPercentage),
Jeff Haof0a31f82017-03-27 15:50:37 -0700199 test_profile_seed_(NanoTime()),
Calin Juravle2dba0ab2018-01-22 19:22:24 -0800200 start_ns_(NanoTime()),
201 copy_and_update_profile_key_(false) {}
Calin Juravle2e2db782016-02-23 12:00:03 +0000202
203 ~ProfMan() {
204 LogCompletionTime();
205 }
206
207 void ParseArgs(int argc, char **argv) {
208 original_argc = argc;
209 original_argv = argv;
210
David Sehr671af6c2018-05-17 11:00:35 -0700211 MemMap::Init();
212 InitLogging(argv, Abort);
Calin Juravle2e2db782016-02-23 12:00:03 +0000213
214 // Skip over the command name.
215 argv++;
216 argc--;
217
218 if (argc == 0) {
219 Usage("No arguments specified");
220 }
221
222 for (int i = 0; i < argc; ++i) {
223 const StringPiece option(argv[i]);
224 const bool log_options = false;
225 if (log_options) {
Calin Juravle876f3502016-03-24 16:16:34 +0000226 LOG(INFO) << "profman: option[" << i << "]=" << argv[i];
Calin Juravle2e2db782016-02-23 12:00:03 +0000227 }
David Sehr4fcdd6d2016-05-24 14:52:31 -0700228 if (option == "--dump-only") {
229 dump_only_ = true;
Mathieu Chartier34067262017-04-06 13:55:46 -0700230 } else if (option == "--dump-classes-and-methods") {
231 dump_classes_and_methods_ = true;
David Sehr7c80f2d2017-02-07 16:47:58 -0800232 } else if (option.starts_with("--create-profile-from=")) {
233 create_profile_from_file_ = option.substr(strlen("--create-profile-from=")).ToString();
David Sehr4fcdd6d2016-05-24 14:52:31 -0700234 } else if (option.starts_with("--dump-output-to-fd=")) {
235 ParseUintOption(option, "--dump-output-to-fd", &dump_output_to_fd_, Usage);
Mathieu Chartier2f794552017-06-19 10:58:08 -0700236 } else if (option == "--generate-boot-image-profile") {
237 generate_boot_image_profile_ = true;
238 } else if (option.starts_with("--boot-image-class-threshold=")) {
239 ParseUintOption(option,
240 "--boot-image-class-threshold",
241 &boot_image_options_.image_class_theshold,
242 Usage);
243 } else if (option.starts_with("--boot-image-clean-class-threshold=")) {
244 ParseUintOption(option,
245 "--boot-image-clean-class-threshold",
246 &boot_image_options_.image_class_clean_theshold,
247 Usage);
Mathieu Chartier8eecddf2017-07-12 16:05:54 -0700248 } else if (option.starts_with("--boot-image-sampled-method-threshold=")) {
249 ParseUintOption(option,
250 "--boot-image-sampled-method-threshold",
251 &boot_image_options_.compiled_method_threshold,
252 Usage);
Calin Juravle876f3502016-03-24 16:16:34 +0000253 } else if (option.starts_with("--profile-file=")) {
Calin Juravle2e2db782016-02-23 12:00:03 +0000254 profile_files_.push_back(option.substr(strlen("--profile-file=")).ToString());
255 } else if (option.starts_with("--profile-file-fd=")) {
256 ParseFdForCollection(option, "--profile-file-fd", &profile_files_fd_);
257 } else if (option.starts_with("--reference-profile-file=")) {
258 reference_profile_file_ = option.substr(strlen("--reference-profile-file=")).ToString();
259 } else if (option.starts_with("--reference-profile-file-fd=")) {
260 ParseUintOption(option, "--reference-profile-file-fd", &reference_profile_file_fd_, Usage);
David Sehr546d24f2016-06-02 10:46:19 -0700261 } else if (option.starts_with("--dex-location=")) {
262 dex_locations_.push_back(option.substr(strlen("--dex-location=")).ToString());
263 } else if (option.starts_with("--apk-fd=")) {
264 ParseFdForCollection(option, "--apk-fd", &apks_fd_);
David Sehr7c80f2d2017-02-07 16:47:58 -0800265 } else if (option.starts_with("--apk=")) {
266 apk_files_.push_back(option.substr(strlen("--apk=")).ToString());
Calin Juravle7bcdb532016-06-07 16:14:47 +0100267 } else if (option.starts_with("--generate-test-profile=")) {
268 test_profile_ = option.substr(strlen("--generate-test-profile=")).ToString();
269 } else if (option.starts_with("--generate-test-profile-num-dex=")) {
270 ParseUintOption(option,
271 "--generate-test-profile-num-dex",
272 &test_profile_num_dex_,
273 Usage);
Shubham Ajmerad704f0b2017-07-26 16:33:35 -0700274 } else if (option.starts_with("--generate-test-profile-method-percentage")) {
Calin Juravle7bcdb532016-06-07 16:14:47 +0100275 ParseUintOption(option,
Shubham Ajmerad704f0b2017-07-26 16:33:35 -0700276 "--generate-test-profile-method-percentage",
277 &test_profile_method_percerntage_,
Calin Juravle7bcdb532016-06-07 16:14:47 +0100278 Usage);
Shubham Ajmerad704f0b2017-07-26 16:33:35 -0700279 } else if (option.starts_with("--generate-test-profile-class-percentage")) {
Calin Juravle7bcdb532016-06-07 16:14:47 +0100280 ParseUintOption(option,
Shubham Ajmerad704f0b2017-07-26 16:33:35 -0700281 "--generate-test-profile-class-percentage",
282 &test_profile_class_percentage_,
Calin Juravle7bcdb532016-06-07 16:14:47 +0100283 Usage);
Jeff Haof0a31f82017-03-27 15:50:37 -0700284 } else if (option.starts_with("--generate-test-profile-seed=")) {
285 ParseUintOption(option, "--generate-test-profile-seed", &test_profile_seed_, Usage);
Calin Juravle02c08792018-02-15 19:40:48 -0800286 } else if (option.starts_with("--copy-and-update-profile-key")) {
287 copy_and_update_profile_key_ = true;
Calin Juravle2e2db782016-02-23 12:00:03 +0000288 } else {
David Sehr4fcdd6d2016-05-24 14:52:31 -0700289 Usage("Unknown argument '%s'", option.data());
Calin Juravle2e2db782016-02-23 12:00:03 +0000290 }
291 }
292
David Sehr153da0e2017-02-15 08:54:51 -0800293 // Validate global consistency between file/fd options.
Calin Juravle2e2db782016-02-23 12:00:03 +0000294 if (!profile_files_.empty() && !profile_files_fd_.empty()) {
295 Usage("Profile files should not be specified with both --profile-file-fd and --profile-file");
296 }
David Sehr4fcdd6d2016-05-24 14:52:31 -0700297 if (!reference_profile_file_.empty() && FdIsValid(reference_profile_file_fd_)) {
298 Usage("Reference profile should not be specified with both "
299 "--reference-profile-file-fd and --reference-profile-file");
300 }
David Sehr153da0e2017-02-15 08:54:51 -0800301 if (!apk_files_.empty() && !apks_fd_.empty()) {
302 Usage("APK files should not be specified with both --apk-fd and --apk");
Calin Juravle2e2db782016-02-23 12:00:03 +0000303 }
304 }
305
Calin Juravlee10c1e22018-01-26 20:10:15 -0800306 struct ProfileFilterKey {
307 ProfileFilterKey(const std::string& dex_location, uint32_t checksum)
308 : dex_location_(dex_location), checksum_(checksum) {}
309 const std::string dex_location_;
310 uint32_t checksum_;
311
312 bool operator==(const ProfileFilterKey& other) const {
313 return checksum_ == other.checksum_ && dex_location_ == other.dex_location_;
314 }
315 bool operator<(const ProfileFilterKey& other) const {
316 return checksum_ == other.checksum_
317 ? dex_location_ < other.dex_location_
318 : checksum_ < other.checksum_;
319 }
320 };
321
Calin Juravle2e2db782016-02-23 12:00:03 +0000322 ProfileAssistant::ProcessingResult ProcessProfiles() {
David Sehr153da0e2017-02-15 08:54:51 -0800323 // Validate that at least one profile file was passed, as well as a reference profile.
324 if (profile_files_.empty() && profile_files_fd_.empty()) {
325 Usage("No profile files specified.");
326 }
327 if (reference_profile_file_.empty() && !FdIsValid(reference_profile_file_fd_)) {
328 Usage("No reference profile file specified.");
329 }
330 if ((!profile_files_.empty() && FdIsValid(reference_profile_file_fd_)) ||
331 (!profile_files_fd_.empty() && !FdIsValid(reference_profile_file_fd_))) {
332 Usage("Options --profile-file-fd and --reference-profile-file-fd "
333 "should only be used together");
334 }
Calin Juravlee10c1e22018-01-26 20:10:15 -0800335
336 // Check if we have any apks which we should use to filter the profile data.
337 std::set<ProfileFilterKey> profile_filter_keys;
338 if (!GetProfileFilterKeyFromApks(&profile_filter_keys)) {
339 return ProfileAssistant::kErrorIO;
340 }
341
342 // Build the profile filter function. If the set of keys is empty it means we
343 // don't have any apks; as such we do not filter anything.
344 const ProfileCompilationInfo::ProfileLoadFilterFn& filter_fn =
345 [profile_filter_keys](const std::string& dex_location, uint32_t checksum) {
346 if (profile_filter_keys.empty()) {
347 // No --apk was specified. Accept all dex files.
348 return true;
349 } else {
350 bool res = profile_filter_keys.find(
351 ProfileFilterKey(dex_location, checksum)) != profile_filter_keys.end();
352 return res;
353 }
354 };
355
Calin Juravle2e2db782016-02-23 12:00:03 +0000356 ProfileAssistant::ProcessingResult result;
Calin Juravle2dba0ab2018-01-22 19:22:24 -0800357
Calin Juravle2e2db782016-02-23 12:00:03 +0000358 if (profile_files_.empty()) {
359 // The file doesn't need to be flushed here (ProcessProfiles will do it)
360 // so don't check the usage.
361 File file(reference_profile_file_fd_, false);
Calin Juravle2dba0ab2018-01-22 19:22:24 -0800362 result = ProfileAssistant::ProcessProfiles(profile_files_fd_,
Calin Juravlee10c1e22018-01-26 20:10:15 -0800363 reference_profile_file_fd_,
364 filter_fn);
Calin Juravle2e2db782016-02-23 12:00:03 +0000365 CloseAllFds(profile_files_fd_, "profile_files_fd_");
366 } else {
Calin Juravlee10c1e22018-01-26 20:10:15 -0800367 result = ProfileAssistant::ProcessProfiles(profile_files_,
368 reference_profile_file_,
369 filter_fn);
Calin Juravle2e2db782016-02-23 12:00:03 +0000370 }
371 return result;
372 }
373
Calin Juravlee10c1e22018-01-26 20:10:15 -0800374 bool GetProfileFilterKeyFromApks(std::set<ProfileFilterKey>* profile_filter_keys) {
375 auto process_fn = [profile_filter_keys](std::unique_ptr<const DexFile>&& dex_file) {
376 // Store the profile key of the location instead of the location itself.
377 // This will make the matching in the profile filter method much easier.
378 profile_filter_keys->emplace(ProfileCompilationInfo::GetProfileDexFileKey(
379 dex_file->GetLocation()), dex_file->GetLocationChecksum());
380 };
381 return OpenApkFilesFromLocations(process_fn);
382 }
383
384 bool OpenApkFilesFromLocations(std::vector<std::unique_ptr<const DexFile>>* dex_files) {
385 auto process_fn = [dex_files](std::unique_ptr<const DexFile>&& dex_file) {
386 dex_files->emplace_back(std::move(dex_file));
387 };
388 return OpenApkFilesFromLocations(process_fn);
389 }
390
391 bool OpenApkFilesFromLocations(
Andreas Gampe68562142018-06-20 21:49:11 +0000392 std::function<void(std::unique_ptr<const DexFile>&&)> process_fn) {
David Sehr7c80f2d2017-02-07 16:47:58 -0800393 bool use_apk_fd_list = !apks_fd_.empty();
394 if (use_apk_fd_list) {
David Sehr153da0e2017-02-15 08:54:51 -0800395 // Get the APKs from the collection of FDs.
Calin Juravlee10c1e22018-01-26 20:10:15 -0800396 if (dex_locations_.empty()) {
397 // Try to compute the dex locations from the file paths of the descriptions.
398 // This will make it easier to invoke profman with --apk-fd and without
399 // being force to pass --dex-location when the location would be the apk path.
400 if (!ComputeDexLocationsFromApkFds()) {
401 return false;
402 }
403 } else {
404 if (dex_locations_.size() != apks_fd_.size()) {
405 Usage("The number of apk-fds must match the number of dex-locations.");
406 }
407 }
David Sehr153da0e2017-02-15 08:54:51 -0800408 } else if (!apk_files_.empty()) {
Calin Juravle02c08792018-02-15 19:40:48 -0800409 if (dex_locations_.empty()) {
410 // If no dex locations are specified use the apk names as locations.
411 dex_locations_ = apk_files_;
412 } else if (dex_locations_.size() != apk_files_.size()) {
413 Usage("The number of apk-fds must match the number of dex-locations.");
414 }
David Sehr153da0e2017-02-15 08:54:51 -0800415 } else {
416 // No APKs were specified.
417 CHECK(dex_locations_.empty());
Calin Juravlee10c1e22018-01-26 20:10:15 -0800418 return true;
David Sehr7c80f2d2017-02-07 16:47:58 -0800419 }
420 static constexpr bool kVerifyChecksum = true;
421 for (size_t i = 0; i < dex_locations_.size(); ++i) {
Mathieu Chartier818cb802018-05-11 05:30:16 +0000422 std::string error_msg;
423 const ArtDexFileLoader dex_file_loader;
424 std::vector<std::unique_ptr<const DexFile>> dex_files_for_location;
Calin Juravle1bfe4bd2018-04-26 16:00:11 -0700425 // We do not need to verify the apk for processing profiles.
David Sehr7c80f2d2017-02-07 16:47:58 -0800426 if (use_apk_fd_list) {
Mathieu Chartier818cb802018-05-11 05:30:16 +0000427 if (dex_file_loader.OpenZip(apks_fd_[i],
428 dex_locations_[i],
429 /* verify */ false,
430 kVerifyChecksum,
431 &error_msg,
432 &dex_files_for_location)) {
433 } else {
434 LOG(ERROR) << "OpenZip failed for '" << dex_locations_[i] << "' " << error_msg;
Calin Juravlee10c1e22018-01-26 20:10:15 -0800435 return false;
David Sehr7c80f2d2017-02-07 16:47:58 -0800436 }
Mathieu Chartier818cb802018-05-11 05:30:16 +0000437 } else {
438 if (dex_file_loader.Open(apk_files_[i].c_str(),
439 dex_locations_[i],
440 /* verify */ false,
441 kVerifyChecksum,
442 &error_msg,
443 &dex_files_for_location)) {
444 } else {
445 LOG(ERROR) << "Open failed for '" << dex_locations_[i] << "' " << error_msg;
446 return false;
447 }
David Sehr2b80ed42018-05-08 08:58:15 -0700448 }
David Sehr7c80f2d2017-02-07 16:47:58 -0800449 for (std::unique_ptr<const DexFile>& dex_file : dex_files_for_location) {
Calin Juravlee10c1e22018-01-26 20:10:15 -0800450 process_fn(std::move(dex_file));
David Sehr7c80f2d2017-02-07 16:47:58 -0800451 }
452 }
Calin Juravlee10c1e22018-01-26 20:10:15 -0800453 return true;
454 }
455
456 // Get the dex locations from the apk fds.
457 // The methods reads the links from /proc/self/fd/ to find the original apk paths
458 // and puts them in the dex_locations_ vector.
459 bool ComputeDexLocationsFromApkFds() {
460 // We can't use a char array of PATH_MAX size without exceeding the frame size.
461 // So we use a vector as the buffer for the path.
462 std::vector<char> buffer(PATH_MAX, 0);
463 for (size_t i = 0; i < apks_fd_.size(); ++i) {
464 std::string fd_path = "/proc/self/fd/" + std::to_string(apks_fd_[i]);
465 ssize_t len = readlink(fd_path.c_str(), buffer.data(), buffer.size() - 1);
466 if (len == -1) {
467 PLOG(ERROR) << "Could not open path from fd";
468 return false;
469 }
470
471 buffer[len] = '\0';
472 dex_locations_.push_back(buffer.data());
473 }
474 return true;
David Sehr7c80f2d2017-02-07 16:47:58 -0800475 }
476
Mathieu Chartier2f794552017-06-19 10:58:08 -0700477 std::unique_ptr<const ProfileCompilationInfo> LoadProfile(const std::string& filename, int fd) {
478 if (!filename.empty()) {
479 fd = open(filename.c_str(), O_RDWR);
480 if (fd < 0) {
481 LOG(ERROR) << "Cannot open " << filename << strerror(errno);
482 return nullptr;
483 }
484 }
485 std::unique_ptr<ProfileCompilationInfo> info(new ProfileCompilationInfo);
486 if (!info->Load(fd)) {
487 LOG(ERROR) << "Cannot load profile info from fd=" << fd << "\n";
488 return nullptr;
489 }
490 return info;
491 }
492
David Sehrb18991b2017-02-08 20:58:10 -0800493 int DumpOneProfile(const std::string& banner,
494 const std::string& filename,
495 int fd,
496 const std::vector<std::unique_ptr<const DexFile>>* dex_files,
497 std::string* dump) {
Mathieu Chartier2f794552017-06-19 10:58:08 -0700498 std::unique_ptr<const ProfileCompilationInfo> info(LoadProfile(filename, fd));
499 if (info == nullptr) {
500 LOG(ERROR) << "Cannot load profile info from filename=" << filename << " fd=" << fd;
Calin Juravle876f3502016-03-24 16:16:34 +0000501 return -1;
502 }
Mathieu Chartier2f794552017-06-19 10:58:08 -0700503 *dump += banner + "\n" + info->DumpInfo(dex_files) + "\n";
David Sehr4fcdd6d2016-05-24 14:52:31 -0700504 return 0;
505 }
506
507 int DumpProfileInfo() {
David Sehr153da0e2017-02-15 08:54:51 -0800508 // Validate that at least one profile file or reference was specified.
509 if (profile_files_.empty() && profile_files_fd_.empty() &&
510 reference_profile_file_.empty() && !FdIsValid(reference_profile_file_fd_)) {
511 Usage("No profile files or reference profile specified.");
512 }
David Sehr4fcdd6d2016-05-24 14:52:31 -0700513 static const char* kEmptyString = "";
David Sehr546d24f2016-06-02 10:46:19 -0700514 static const char* kOrdinaryProfile = "=== profile ===";
515 static const char* kReferenceProfile = "=== reference profile ===";
516
David Sehrb18991b2017-02-08 20:58:10 -0800517 std::vector<std::unique_ptr<const DexFile>> dex_files;
David Sehr7c80f2d2017-02-07 16:47:58 -0800518 OpenApkFilesFromLocations(&dex_files);
David Sehr4fcdd6d2016-05-24 14:52:31 -0700519 std::string dump;
David Sehr4fcdd6d2016-05-24 14:52:31 -0700520 // Dump individual profile files.
521 if (!profile_files_fd_.empty()) {
522 for (int profile_file_fd : profile_files_fd_) {
David Sehr546d24f2016-06-02 10:46:19 -0700523 int ret = DumpOneProfile(kOrdinaryProfile,
524 kEmptyString,
525 profile_file_fd,
526 &dex_files,
527 &dump);
David Sehr4fcdd6d2016-05-24 14:52:31 -0700528 if (ret != 0) {
529 return ret;
530 }
531 }
532 }
533 if (!profile_files_.empty()) {
534 for (const std::string& profile_file : profile_files_) {
David Sehr546d24f2016-06-02 10:46:19 -0700535 int ret = DumpOneProfile(kOrdinaryProfile, profile_file, kInvalidFd, &dex_files, &dump);
David Sehr4fcdd6d2016-05-24 14:52:31 -0700536 if (ret != 0) {
537 return ret;
538 }
539 }
540 }
541 // Dump reference profile file.
542 if (FdIsValid(reference_profile_file_fd_)) {
David Sehr546d24f2016-06-02 10:46:19 -0700543 int ret = DumpOneProfile(kReferenceProfile,
544 kEmptyString,
545 reference_profile_file_fd_,
546 &dex_files,
547 &dump);
David Sehr4fcdd6d2016-05-24 14:52:31 -0700548 if (ret != 0) {
549 return ret;
550 }
551 }
552 if (!reference_profile_file_.empty()) {
David Sehr546d24f2016-06-02 10:46:19 -0700553 int ret = DumpOneProfile(kReferenceProfile,
554 reference_profile_file_,
555 kInvalidFd,
556 &dex_files,
David Sehr4fcdd6d2016-05-24 14:52:31 -0700557 &dump);
558 if (ret != 0) {
559 return ret;
560 }
561 }
562 if (!FdIsValid(dump_output_to_fd_)) {
563 std::cout << dump;
564 } else {
565 unix_file::FdFile out_fd(dump_output_to_fd_, false /*check_usage*/);
566 if (!out_fd.WriteFully(dump.c_str(), dump.length())) {
567 return -1;
568 }
569 }
Calin Juravle876f3502016-03-24 16:16:34 +0000570 return 0;
571 }
572
573 bool ShouldOnlyDumpProfile() {
David Sehr4fcdd6d2016-05-24 14:52:31 -0700574 return dump_only_;
Calin Juravle876f3502016-03-24 16:16:34 +0000575 }
576
Mathieu Chartier34067262017-04-06 13:55:46 -0700577 bool GetClassNamesAndMethods(int fd,
578 std::vector<std::unique_ptr<const DexFile>>* dex_files,
579 std::set<std::string>* out_lines) {
David Sehr7c80f2d2017-02-07 16:47:58 -0800580 ProfileCompilationInfo profile_info;
581 if (!profile_info.Load(fd)) {
582 LOG(ERROR) << "Cannot load profile info";
583 return false;
584 }
Mathieu Chartier34067262017-04-06 13:55:46 -0700585 for (const std::unique_ptr<const DexFile>& dex_file : *dex_files) {
586 std::set<dex::TypeIndex> class_types;
Mathieu Chartierea650f32017-05-24 12:04:13 -0700587 std::set<uint16_t> hot_methods;
588 std::set<uint16_t> startup_methods;
589 std::set<uint16_t> post_startup_methods;
590 std::set<uint16_t> combined_methods;
591 if (profile_info.GetClassesAndMethods(*dex_file.get(),
592 &class_types,
593 &hot_methods,
594 &startup_methods,
595 &post_startup_methods)) {
Mathieu Chartier34067262017-04-06 13:55:46 -0700596 for (const dex::TypeIndex& type_index : class_types) {
597 const DexFile::TypeId& type_id = dex_file->GetTypeId(type_index);
598 out_lines->insert(std::string(dex_file->GetTypeDescriptor(type_id)));
599 }
Mathieu Chartierea650f32017-05-24 12:04:13 -0700600 combined_methods = hot_methods;
601 combined_methods.insert(startup_methods.begin(), startup_methods.end());
602 combined_methods.insert(post_startup_methods.begin(), post_startup_methods.end());
603 for (uint16_t dex_method_idx : combined_methods) {
Mathieu Chartier34067262017-04-06 13:55:46 -0700604 const DexFile::MethodId& id = dex_file->GetMethodId(dex_method_idx);
605 std::string signature_string(dex_file->GetMethodSignature(id).ToString());
606 std::string type_string(dex_file->GetTypeDescriptor(dex_file->GetTypeId(id.class_idx_)));
607 std::string method_name(dex_file->GetMethodName(id));
Mathieu Chartierea650f32017-05-24 12:04:13 -0700608 std::string flags_string;
609 if (hot_methods.find(dex_method_idx) != hot_methods.end()) {
610 flags_string += kMethodFlagStringHot;
611 }
612 if (startup_methods.find(dex_method_idx) != startup_methods.end()) {
613 flags_string += kMethodFlagStringStartup;
614 }
615 if (post_startup_methods.find(dex_method_idx) != post_startup_methods.end()) {
616 flags_string += kMethodFlagStringPostStartup;
617 }
618 out_lines->insert(flags_string +
619 type_string +
620 kMethodSep +
621 method_name +
622 signature_string);
Mathieu Chartier34067262017-04-06 13:55:46 -0700623 }
624 }
625 }
David Sehr7c80f2d2017-02-07 16:47:58 -0800626 return true;
627 }
628
Mathieu Chartier34067262017-04-06 13:55:46 -0700629 bool GetClassNamesAndMethods(const std::string& profile_file,
630 std::vector<std::unique_ptr<const DexFile>>* dex_files,
631 std::set<std::string>* out_lines) {
David Sehr7c80f2d2017-02-07 16:47:58 -0800632 int fd = open(profile_file.c_str(), O_RDONLY);
633 if (!FdIsValid(fd)) {
634 LOG(ERROR) << "Cannot open " << profile_file << strerror(errno);
635 return false;
636 }
Mathieu Chartier34067262017-04-06 13:55:46 -0700637 if (!GetClassNamesAndMethods(fd, dex_files, out_lines)) {
David Sehr7c80f2d2017-02-07 16:47:58 -0800638 return false;
639 }
640 if (close(fd) < 0) {
641 PLOG(WARNING) << "Failed to close descriptor";
642 }
643 return true;
644 }
645
Mathieu Chartierea650f32017-05-24 12:04:13 -0700646 int DumpClassesAndMethods() {
David Sehr153da0e2017-02-15 08:54:51 -0800647 // Validate that at least one profile file or reference was specified.
648 if (profile_files_.empty() && profile_files_fd_.empty() &&
649 reference_profile_file_.empty() && !FdIsValid(reference_profile_file_fd_)) {
650 Usage("No profile files or reference profile specified.");
651 }
Calin Juravlee10c1e22018-01-26 20:10:15 -0800652
David Sehr7c80f2d2017-02-07 16:47:58 -0800653 // Open the dex files to get the names for classes.
654 std::vector<std::unique_ptr<const DexFile>> dex_files;
655 OpenApkFilesFromLocations(&dex_files);
656 // Build a vector of class names from individual profile files.
657 std::set<std::string> class_names;
658 if (!profile_files_fd_.empty()) {
659 for (int profile_file_fd : profile_files_fd_) {
Mathieu Chartier34067262017-04-06 13:55:46 -0700660 if (!GetClassNamesAndMethods(profile_file_fd, &dex_files, &class_names)) {
David Sehr7c80f2d2017-02-07 16:47:58 -0800661 return -1;
662 }
663 }
664 }
665 if (!profile_files_.empty()) {
666 for (const std::string& profile_file : profile_files_) {
Mathieu Chartier34067262017-04-06 13:55:46 -0700667 if (!GetClassNamesAndMethods(profile_file, &dex_files, &class_names)) {
David Sehr7c80f2d2017-02-07 16:47:58 -0800668 return -1;
669 }
670 }
671 }
672 // Concatenate class names from reference profile file.
673 if (FdIsValid(reference_profile_file_fd_)) {
Mathieu Chartier34067262017-04-06 13:55:46 -0700674 if (!GetClassNamesAndMethods(reference_profile_file_fd_, &dex_files, &class_names)) {
David Sehr7c80f2d2017-02-07 16:47:58 -0800675 return -1;
676 }
677 }
678 if (!reference_profile_file_.empty()) {
Mathieu Chartier34067262017-04-06 13:55:46 -0700679 if (!GetClassNamesAndMethods(reference_profile_file_, &dex_files, &class_names)) {
David Sehr7c80f2d2017-02-07 16:47:58 -0800680 return -1;
681 }
682 }
683 // Dump the class names.
684 std::string dump;
685 for (const std::string& class_name : class_names) {
686 dump += class_name + std::string("\n");
687 }
688 if (!FdIsValid(dump_output_to_fd_)) {
689 std::cout << dump;
690 } else {
691 unix_file::FdFile out_fd(dump_output_to_fd_, false /*check_usage*/);
692 if (!out_fd.WriteFully(dump.c_str(), dump.length())) {
693 return -1;
694 }
695 }
696 return 0;
697 }
698
Mathieu Chartier34067262017-04-06 13:55:46 -0700699 bool ShouldOnlyDumpClassesAndMethods() {
700 return dump_classes_and_methods_;
David Sehr7c80f2d2017-02-07 16:47:58 -0800701 }
702
703 // Read lines from the given file, dropping comments and empty lines. Post-process each line with
704 // the given function.
705 template <typename T>
706 static T* ReadCommentedInputFromFile(
707 const char* input_filename, std::function<std::string(const char*)>* process) {
708 std::unique_ptr<std::ifstream> input_file(new std::ifstream(input_filename, std::ifstream::in));
709 if (input_file.get() == nullptr) {
710 LOG(ERROR) << "Failed to open input file " << input_filename;
711 return nullptr;
712 }
713 std::unique_ptr<T> result(
714 ReadCommentedInputStream<T>(*input_file, process));
715 input_file->close();
716 return result.release();
717 }
718
719 // Read lines from the given stream, dropping comments and empty lines. Post-process each line
720 // with the given function.
721 template <typename T>
722 static T* ReadCommentedInputStream(
723 std::istream& in_stream,
724 std::function<std::string(const char*)>* process) {
725 std::unique_ptr<T> output(new T());
726 while (in_stream.good()) {
727 std::string dot;
728 std::getline(in_stream, dot);
729 if (android::base::StartsWith(dot, "#") || dot.empty()) {
730 continue;
731 }
732 if (process != nullptr) {
733 std::string descriptor((*process)(dot.c_str()));
734 output->insert(output->end(), descriptor);
735 } else {
736 output->insert(output->end(), dot);
737 }
738 }
739 return output.release();
740 }
741
Calin Juravlee0ac1152017-02-13 19:03:47 -0800742 // Find class klass_descriptor in the given dex_files and store its reference
743 // in the out parameter class_ref.
744 // Return true if the definition of the class was found in any of the dex_files.
745 bool FindClass(const std::vector<std::unique_ptr<const DexFile>>& dex_files,
746 const std::string& klass_descriptor,
Mathieu Chartierdbddc222017-05-24 12:04:13 -0700747 /*out*/TypeReference* class_ref) {
Calin Juravle08556882017-05-26 16:40:45 -0700748 constexpr uint16_t kInvalidTypeIndex = std::numeric_limits<uint16_t>::max() - 1;
Calin Juravlee0ac1152017-02-13 19:03:47 -0800749 for (const std::unique_ptr<const DexFile>& dex_file_ptr : dex_files) {
750 const DexFile* dex_file = dex_file_ptr.get();
Calin Juravle08556882017-05-26 16:40:45 -0700751 if (klass_descriptor == kInvalidClassDescriptor) {
752 if (kInvalidTypeIndex >= dex_file->NumTypeIds()) {
753 // The dex file does not contain all possible type ids which leaves us room
754 // to add an "invalid" type id.
Mathieu Chartierfc8b4222017-09-17 13:44:24 -0700755 *class_ref = TypeReference(dex_file, dex::TypeIndex(kInvalidTypeIndex));
Calin Juravle08556882017-05-26 16:40:45 -0700756 return true;
757 } else {
758 // The dex file contains all possible type ids. We don't have any free type id
759 // that we can use as invalid.
760 continue;
761 }
762 }
763
Calin Juravlee0ac1152017-02-13 19:03:47 -0800764 const DexFile::TypeId* type_id = dex_file->FindTypeId(klass_descriptor.c_str());
765 if (type_id == nullptr) {
766 continue;
767 }
768 dex::TypeIndex type_index = dex_file->GetIndexForTypeId(*type_id);
769 if (dex_file->FindClassDef(type_index) == nullptr) {
770 // Class is only referenced in the current dex file but not defined in it.
771 continue;
772 }
Mathieu Chartierfc8b4222017-09-17 13:44:24 -0700773 *class_ref = TypeReference(dex_file, type_index);
Calin Juravlee0ac1152017-02-13 19:03:47 -0800774 return true;
775 }
776 return false;
777 }
778
Mathieu Chartier34067262017-04-06 13:55:46 -0700779 // Find the method specified by method_spec in the class class_ref.
Calin Juravle08556882017-05-26 16:40:45 -0700780 uint32_t FindMethodIndex(const TypeReference& class_ref,
781 const std::string& method_spec) {
782 const DexFile* dex_file = class_ref.dex_file;
783 if (method_spec == kInvalidMethod) {
784 constexpr uint16_t kInvalidMethodIndex = std::numeric_limits<uint16_t>::max() - 1;
785 return kInvalidMethodIndex >= dex_file->NumMethodIds()
786 ? kInvalidMethodIndex
Andreas Gampee2abbc62017-09-15 11:59:26 -0700787 : dex::kDexNoIndex;
Calin Juravle08556882017-05-26 16:40:45 -0700788 }
789
Calin Juravlee0ac1152017-02-13 19:03:47 -0800790 std::vector<std::string> name_and_signature;
791 Split(method_spec, kProfileParsingFirstCharInSignature, &name_and_signature);
792 if (name_and_signature.size() != 2) {
793 LOG(ERROR) << "Invalid method name and signature " << method_spec;
Andreas Gampee2abbc62017-09-15 11:59:26 -0700794 return dex::kDexNoIndex;
Calin Juravlee0ac1152017-02-13 19:03:47 -0800795 }
Calin Juravle08556882017-05-26 16:40:45 -0700796
Calin Juravlee0ac1152017-02-13 19:03:47 -0800797 const std::string& name = name_and_signature[0];
798 const std::string& signature = kProfileParsingFirstCharInSignature + name_and_signature[1];
Calin Juravlee0ac1152017-02-13 19:03:47 -0800799
800 const DexFile::StringId* name_id = dex_file->FindStringId(name.c_str());
801 if (name_id == nullptr) {
Mathieu Chartier66aae3b2017-05-18 10:38:28 -0700802 LOG(WARNING) << "Could not find name: " << name;
Andreas Gampee2abbc62017-09-15 11:59:26 -0700803 return dex::kDexNoIndex;
Calin Juravlee0ac1152017-02-13 19:03:47 -0800804 }
805 dex::TypeIndex return_type_idx;
806 std::vector<dex::TypeIndex> param_type_idxs;
807 if (!dex_file->CreateTypeList(signature, &return_type_idx, &param_type_idxs)) {
Mathieu Chartier66aae3b2017-05-18 10:38:28 -0700808 LOG(WARNING) << "Could not create type list" << signature;
Andreas Gampee2abbc62017-09-15 11:59:26 -0700809 return dex::kDexNoIndex;
Calin Juravlee0ac1152017-02-13 19:03:47 -0800810 }
811 const DexFile::ProtoId* proto_id = dex_file->FindProtoId(return_type_idx, param_type_idxs);
812 if (proto_id == nullptr) {
Mathieu Chartier66aae3b2017-05-18 10:38:28 -0700813 LOG(WARNING) << "Could not find proto_id: " << name;
Andreas Gampee2abbc62017-09-15 11:59:26 -0700814 return dex::kDexNoIndex;
Calin Juravlee0ac1152017-02-13 19:03:47 -0800815 }
816 const DexFile::MethodId* method_id = dex_file->FindMethodId(
Mathieu Chartierfc8b4222017-09-17 13:44:24 -0700817 dex_file->GetTypeId(class_ref.TypeIndex()), *name_id, *proto_id);
Calin Juravlee0ac1152017-02-13 19:03:47 -0800818 if (method_id == nullptr) {
Mathieu Chartier66aae3b2017-05-18 10:38:28 -0700819 LOG(WARNING) << "Could not find method_id: " << name;
Andreas Gampee2abbc62017-09-15 11:59:26 -0700820 return dex::kDexNoIndex;
Calin Juravlee0ac1152017-02-13 19:03:47 -0800821 }
822
Mathieu Chartier34067262017-04-06 13:55:46 -0700823 return dex_file->GetIndexForMethodId(*method_id);
824 }
Calin Juravlee0ac1152017-02-13 19:03:47 -0800825
Mathieu Chartier34067262017-04-06 13:55:46 -0700826 // Given a method, return true if the method has a single INVOKE_VIRTUAL in its byte code.
827 // Upon success it returns true and stores the method index and the invoke dex pc
828 // in the output parameters.
829 // The format of the method spec is "inlinePolymorphic(LSuper;)I+LSubA;,LSubB;,LSubC;".
830 //
831 // TODO(calin): support INVOKE_INTERFACE and the range variants.
Mathieu Chartierdbddc222017-05-24 12:04:13 -0700832 bool HasSingleInvoke(const TypeReference& class_ref,
Mathieu Chartier34067262017-04-06 13:55:46 -0700833 uint16_t method_index,
834 /*out*/uint32_t* dex_pc) {
835 const DexFile* dex_file = class_ref.dex_file;
Calin Juravlee0ac1152017-02-13 19:03:47 -0800836 uint32_t offset = dex_file->FindCodeItemOffset(
Mathieu Chartierfc8b4222017-09-17 13:44:24 -0700837 *dex_file->FindClassDef(class_ref.TypeIndex()),
Mathieu Chartier34067262017-04-06 13:55:46 -0700838 method_index);
Calin Juravlee0ac1152017-02-13 19:03:47 -0800839 const DexFile::CodeItem* code_item = dex_file->GetCodeItem(offset);
840
841 bool found_invoke = false;
Mathieu Chartier698ebbc2018-01-05 11:00:42 -0800842 for (const DexInstructionPcPair& inst : CodeItemInstructionAccessor(*dex_file, code_item)) {
Rico Windc24fa5d2018-04-25 12:44:58 +0200843 if (inst->Opcode() == Instruction::INVOKE_VIRTUAL ||
844 inst->Opcode() == Instruction::INVOKE_VIRTUAL_RANGE) {
Calin Juravlee0ac1152017-02-13 19:03:47 -0800845 if (found_invoke) {
Mathieu Chartier34067262017-04-06 13:55:46 -0700846 LOG(ERROR) << "Multiple invoke INVOKE_VIRTUAL found: "
847 << dex_file->PrettyMethod(method_index);
Calin Juravlee0ac1152017-02-13 19:03:47 -0800848 return false;
849 }
850 found_invoke = true;
Mathieu Chartier0021feb2017-11-07 00:08:52 -0800851 *dex_pc = inst.DexPc();
Calin Juravlee0ac1152017-02-13 19:03:47 -0800852 }
853 }
854 if (!found_invoke) {
Mathieu Chartier34067262017-04-06 13:55:46 -0700855 LOG(ERROR) << "Could not find any INVOKE_VIRTUAL: " << dex_file->PrettyMethod(method_index);
Calin Juravlee0ac1152017-02-13 19:03:47 -0800856 }
857 return found_invoke;
858 }
859
860 // Process a line defining a class or a method and its inline caches.
861 // Upon success return true and add the class or the method info to profile.
Calin Juravle589e71e2017-03-03 16:05:05 -0800862 // The possible line formats are:
863 // "LJustTheCass;".
Calin Juravlee0ac1152017-02-13 19:03:47 -0800864 // "LTestInline;->inlinePolymorphic(LSuper;)I+LSubA;,LSubB;,LSubC;".
Calin Juravle08556882017-05-26 16:40:45 -0700865 // "LTestInline;->inlinePolymorphic(LSuper;)I+LSubA;,LSubB;,invalid_class".
Calin Juravle589e71e2017-03-03 16:05:05 -0800866 // "LTestInline;->inlineMissingTypes(LSuper;)I+missing_types".
867 // "LTestInline;->inlineNoInlineCaches(LSuper;)I".
Mathieu Chartierd808e8b2017-03-21 13:37:41 -0700868 // "LTestInline;->*".
Calin Juravle08556882017-05-26 16:40:45 -0700869 // "invalid_class".
870 // "LTestInline;->invalid_method".
Calin Juravlee0ac1152017-02-13 19:03:47 -0800871 // The method and classes are searched only in the given dex files.
872 bool ProcessLine(const std::vector<std::unique_ptr<const DexFile>>& dex_files,
873 const std::string& line,
874 /*out*/ProfileCompilationInfo* profile) {
875 std::string klass;
876 std::string method_str;
Mathieu Chartierea650f32017-05-24 12:04:13 -0700877 bool is_hot = false;
878 bool is_startup = false;
879 bool is_post_startup = false;
880 const size_t method_sep_index = line.find(kMethodSep, 0);
Calin Juravlee0ac1152017-02-13 19:03:47 -0800881 if (method_sep_index == std::string::npos) {
Mathieu Chartierea650f32017-05-24 12:04:13 -0700882 klass = line.substr(0);
Calin Juravlee0ac1152017-02-13 19:03:47 -0800883 } else {
Mathieu Chartierea650f32017-05-24 12:04:13 -0700884 // The method prefix flags are only valid for method strings.
885 size_t start_index = 0;
886 while (start_index < line.size() && line[start_index] != 'L') {
887 const char c = line[start_index];
888 if (c == kMethodFlagStringHot) {
889 is_hot = true;
890 } else if (c == kMethodFlagStringStartup) {
891 is_startup = true;
892 } else if (c == kMethodFlagStringPostStartup) {
893 is_post_startup = true;
894 } else {
895 LOG(WARNING) << "Invalid flag " << c;
896 return false;
897 }
898 ++start_index;
899 }
900 klass = line.substr(start_index, method_sep_index - start_index);
Calin Juravlee0ac1152017-02-13 19:03:47 -0800901 method_str = line.substr(method_sep_index + kMethodSep.size());
902 }
903
Calin Juravleee9cb412018-02-13 20:32:35 -0800904 uint32_t flags = 0;
905 if (is_hot) {
906 flags |= ProfileCompilationInfo::MethodHotness::kFlagHot;
907 }
908 if (is_startup) {
909 flags |= ProfileCompilationInfo::MethodHotness::kFlagStartup;
910 }
911 if (is_post_startup) {
912 flags |= ProfileCompilationInfo::MethodHotness::kFlagPostStartup;
913 }
914
Vladimir Markof3c52b42017-11-17 17:32:12 +0000915 TypeReference class_ref(/* dex_file */ nullptr, dex::TypeIndex());
Calin Juravlee0ac1152017-02-13 19:03:47 -0800916 if (!FindClass(dex_files, klass, &class_ref)) {
Mathieu Chartier3f121342017-03-06 11:16:20 -0800917 LOG(WARNING) << "Could not find class: " << klass;
Calin Juravlee0ac1152017-02-13 19:03:47 -0800918 return false;
919 }
920
Mathieu Chartierd808e8b2017-03-21 13:37:41 -0700921 if (method_str.empty() || method_str == kClassAllMethods) {
922 // Start by adding the class.
Calin Juravlee0ac1152017-02-13 19:03:47 -0800923 std::set<DexCacheResolvedClasses> resolved_class_set;
924 const DexFile* dex_file = class_ref.dex_file;
925 const auto& dex_resolved_classes = resolved_class_set.emplace(
926 dex_file->GetLocation(),
Mathieu Chartier79c87da2017-10-10 11:54:29 -0700927 DexFileLoader::GetBaseLocation(dex_file->GetLocation()),
Mathieu Chartierea650f32017-05-24 12:04:13 -0700928 dex_file->GetLocationChecksum(),
929 dex_file->NumMethodIds());
Mathieu Chartierfc8b4222017-09-17 13:44:24 -0700930 dex_resolved_classes.first->AddClass(class_ref.TypeIndex());
Mathieu Chartierd808e8b2017-03-21 13:37:41 -0700931 std::vector<ProfileMethodInfo> methods;
932 if (method_str == kClassAllMethods) {
Mathieu Chartier18e26872018-06-04 17:19:02 -0700933 ClassAccessor accessor(
934 *dex_file,
935 dex_file->GetIndexForClassDef(*dex_file->FindClassDef(class_ref.TypeIndex())));
Mathieu Chartier0d896bd2018-05-25 00:20:27 -0700936 for (const ClassAccessor::Method& method : accessor.GetMethods()) {
Mathieu Chartier20f49922018-05-24 16:04:17 -0700937 if (method.GetCodeItemOffset() != 0) {
938 // Add all of the methods that have code to the profile.
939 methods.push_back(ProfileMethodInfo(method.GetReference()));
Mathieu Chartierd808e8b2017-03-21 13:37:41 -0700940 }
Mathieu Chartier0d896bd2018-05-25 00:20:27 -0700941 }
Mathieu Chartierd808e8b2017-03-21 13:37:41 -0700942 }
Mathieu Chartierbbe3a5e2017-06-13 16:36:17 -0700943 // TODO: Check return values?
Calin Juravleee9cb412018-02-13 20:32:35 -0800944 profile->AddMethods(methods, static_cast<ProfileCompilationInfo::MethodHotness::Flag>(flags));
Mathieu Chartierbbe3a5e2017-06-13 16:36:17 -0700945 profile->AddClasses(resolved_class_set);
Calin Juravlee0ac1152017-02-13 19:03:47 -0800946 return true;
947 }
948
949 // Process the method.
950 std::string method_spec;
951 std::vector<std::string> inline_cache_elems;
952
Mathieu Chartierea650f32017-05-24 12:04:13 -0700953 // If none of the flags are set, default to hot.
954 is_hot = is_hot || (!is_hot && !is_startup && !is_post_startup);
955
Calin Juravlee0ac1152017-02-13 19:03:47 -0800956 std::vector<std::string> method_elems;
Calin Juravle589e71e2017-03-03 16:05:05 -0800957 bool is_missing_types = false;
Calin Juravlee0ac1152017-02-13 19:03:47 -0800958 Split(method_str, kProfileParsingInlineChacheSep, &method_elems);
959 if (method_elems.size() == 2) {
960 method_spec = method_elems[0];
Calin Juravle589e71e2017-03-03 16:05:05 -0800961 is_missing_types = method_elems[1] == kMissingTypesMarker;
962 if (!is_missing_types) {
963 Split(method_elems[1], kProfileParsingTypeSep, &inline_cache_elems);
964 }
Calin Juravlee0ac1152017-02-13 19:03:47 -0800965 } else if (method_elems.size() == 1) {
966 method_spec = method_elems[0];
967 } else {
968 LOG(ERROR) << "Invalid method line: " << line;
969 return false;
970 }
971
Mathieu Chartier34067262017-04-06 13:55:46 -0700972 const uint32_t method_index = FindMethodIndex(class_ref, method_spec);
Andreas Gampee2abbc62017-09-15 11:59:26 -0700973 if (method_index == dex::kDexNoIndex) {
Calin Juravlee0ac1152017-02-13 19:03:47 -0800974 return false;
975 }
Mathieu Chartier34067262017-04-06 13:55:46 -0700976
Mathieu Chartier34067262017-04-06 13:55:46 -0700977 std::vector<ProfileMethodInfo::ProfileInlineCache> inline_caches;
978 if (is_missing_types || !inline_cache_elems.empty()) {
979 uint32_t dex_pc;
980 if (!HasSingleInvoke(class_ref, method_index, &dex_pc)) {
Calin Juravlee0ac1152017-02-13 19:03:47 -0800981 return false;
982 }
Vladimir Markof3c52b42017-11-17 17:32:12 +0000983 std::vector<TypeReference> classes(inline_cache_elems.size(),
984 TypeReference(/* dex_file */ nullptr, dex::TypeIndex()));
Mathieu Chartier34067262017-04-06 13:55:46 -0700985 size_t class_it = 0;
986 for (const std::string& ic_class : inline_cache_elems) {
987 if (!FindClass(dex_files, ic_class, &(classes[class_it++]))) {
988 LOG(ERROR) << "Could not find class: " << ic_class;
989 return false;
990 }
991 }
992 inline_caches.emplace_back(dex_pc, is_missing_types, classes);
Calin Juravlee0ac1152017-02-13 19:03:47 -0800993 }
Mathieu Chartierbbe3a5e2017-06-13 16:36:17 -0700994 MethodReference ref(class_ref.dex_file, method_index);
Mathieu Chartierea650f32017-05-24 12:04:13 -0700995 if (is_hot) {
Calin Juravleee9cb412018-02-13 20:32:35 -0800996 profile->AddMethod(ProfileMethodInfo(ref, inline_caches),
997 static_cast<ProfileCompilationInfo::MethodHotness::Flag>(flags));
Mathieu Chartierbbe3a5e2017-06-13 16:36:17 -0700998 }
999 if (flags != 0) {
Calin Juravleee9cb412018-02-13 20:32:35 -08001000 if (!profile->AddMethodIndex(
1001 static_cast<ProfileCompilationInfo::MethodHotness::Flag>(flags), ref)) {
Mathieu Chartierea650f32017-05-24 12:04:13 -07001002 return false;
1003 }
Mathieu Chartiere46f3a82017-06-19 19:54:12 -07001004 DCHECK(profile->GetMethodHotness(ref).IsInProfile());
Mathieu Chartierea650f32017-05-24 12:04:13 -07001005 }
Calin Juravlee0ac1152017-02-13 19:03:47 -08001006 return true;
1007 }
1008
Mathieu Chartier2f794552017-06-19 10:58:08 -07001009 int OpenReferenceProfile() const {
1010 int fd = reference_profile_file_fd_;
1011 if (!FdIsValid(fd)) {
1012 CHECK(!reference_profile_file_.empty());
1013 fd = open(reference_profile_file_.c_str(), O_CREAT | O_TRUNC | O_WRONLY, 0644);
1014 if (fd < 0) {
1015 LOG(ERROR) << "Cannot open " << reference_profile_file_ << strerror(errno);
1016 return kInvalidFd;
1017 }
1018 }
1019 return fd;
1020 }
1021
Calin Juravlee0ac1152017-02-13 19:03:47 -08001022 // Creates a profile from a human friendly textual representation.
1023 // The expected input format is:
1024 // # Classes
1025 // Ljava/lang/Comparable;
1026 // Ljava/lang/Math;
1027 // # Methods with inline caches
1028 // LTestInline;->inlinePolymorphic(LSuper;)I+LSubA;,LSubB;,LSubC;
1029 // LTestInline;->noInlineCache(LSuper;)I
David Sehr7c80f2d2017-02-07 16:47:58 -08001030 int CreateProfile() {
David Sehr153da0e2017-02-15 08:54:51 -08001031 // Validate parameters for this command.
1032 if (apk_files_.empty() && apks_fd_.empty()) {
1033 Usage("APK files must be specified");
1034 }
1035 if (dex_locations_.empty()) {
1036 Usage("DEX locations must be specified");
1037 }
1038 if (reference_profile_file_.empty() && !FdIsValid(reference_profile_file_fd_)) {
1039 Usage("Reference profile must be specified with --reference-profile-file or "
1040 "--reference-profile-file-fd");
1041 }
1042 if (!profile_files_.empty() || !profile_files_fd_.empty()) {
1043 Usage("Profile must be specified with --reference-profile-file or "
1044 "--reference-profile-file-fd");
1045 }
David Sehr7c80f2d2017-02-07 16:47:58 -08001046 // Open the profile output file if needed.
Mathieu Chartier2f794552017-06-19 10:58:08 -07001047 int fd = OpenReferenceProfile();
David Sehr7c80f2d2017-02-07 16:47:58 -08001048 if (!FdIsValid(fd)) {
David Sehr7c80f2d2017-02-07 16:47:58 -08001049 return -1;
David Sehr7c80f2d2017-02-07 16:47:58 -08001050 }
Calin Juravlee0ac1152017-02-13 19:03:47 -08001051 // Read the user-specified list of classes and methods.
David Sehr7c80f2d2017-02-07 16:47:58 -08001052 std::unique_ptr<std::unordered_set<std::string>>
Calin Juravlee0ac1152017-02-13 19:03:47 -08001053 user_lines(ReadCommentedInputFromFile<std::unordered_set<std::string>>(
David Sehr7c80f2d2017-02-07 16:47:58 -08001054 create_profile_from_file_.c_str(), nullptr)); // No post-processing.
Calin Juravlee0ac1152017-02-13 19:03:47 -08001055
1056 // Open the dex files to look up classes and methods.
David Sehr7c80f2d2017-02-07 16:47:58 -08001057 std::vector<std::unique_ptr<const DexFile>> dex_files;
1058 OpenApkFilesFromLocations(&dex_files);
Calin Juravlee0ac1152017-02-13 19:03:47 -08001059
1060 // Process the lines one by one and add the successful ones to the profile.
David Sehr7c80f2d2017-02-07 16:47:58 -08001061 ProfileCompilationInfo info;
Calin Juravlee0ac1152017-02-13 19:03:47 -08001062
1063 for (const auto& line : *user_lines) {
1064 ProcessLine(dex_files, line, &info);
1065 }
1066
David Sehr7c80f2d2017-02-07 16:47:58 -08001067 // Write the profile file.
1068 CHECK(info.Save(fd));
1069 if (close(fd) < 0) {
1070 PLOG(WARNING) << "Failed to close descriptor";
1071 }
1072 return 0;
1073 }
1074
Mathieu Chartier2f794552017-06-19 10:58:08 -07001075 bool ShouldCreateBootProfile() const {
1076 return generate_boot_image_profile_;
1077 }
1078
1079 int CreateBootProfile() {
Mathieu Chartier2f794552017-06-19 10:58:08 -07001080 // Open the profile output file.
1081 const int reference_fd = OpenReferenceProfile();
1082 if (!FdIsValid(reference_fd)) {
1083 PLOG(ERROR) << "Error opening reference profile";
1084 return -1;
1085 }
1086 // Open the dex files.
1087 std::vector<std::unique_ptr<const DexFile>> dex_files;
1088 OpenApkFilesFromLocations(&dex_files);
1089 if (dex_files.empty()) {
1090 PLOG(ERROR) << "Expected dex files for creating boot profile";
1091 return -2;
1092 }
1093 // Open the input profiles.
1094 std::vector<std::unique_ptr<const ProfileCompilationInfo>> profiles;
1095 if (!profile_files_fd_.empty()) {
1096 for (int profile_file_fd : profile_files_fd_) {
1097 std::unique_ptr<const ProfileCompilationInfo> profile(LoadProfile("", profile_file_fd));
1098 if (profile == nullptr) {
1099 return -3;
1100 }
1101 profiles.emplace_back(std::move(profile));
1102 }
1103 }
1104 if (!profile_files_.empty()) {
1105 for (const std::string& profile_file : profile_files_) {
1106 std::unique_ptr<const ProfileCompilationInfo> profile(LoadProfile(profile_file, kInvalidFd));
1107 if (profile == nullptr) {
1108 return -4;
1109 }
1110 profiles.emplace_back(std::move(profile));
1111 }
1112 }
1113 ProfileCompilationInfo out_profile;
1114 GenerateBootImageProfile(dex_files,
1115 profiles,
1116 boot_image_options_,
1117 VLOG_IS_ON(profiler),
1118 &out_profile);
1119 out_profile.Save(reference_fd);
1120 close(reference_fd);
1121 return 0;
1122 }
1123
David Sehr7c80f2d2017-02-07 16:47:58 -08001124 bool ShouldCreateProfile() {
1125 return !create_profile_from_file_.empty();
1126 }
1127
Calin Juravle7bcdb532016-06-07 16:14:47 +01001128 int GenerateTestProfile() {
David Sehr153da0e2017-02-15 08:54:51 -08001129 // Validate parameters for this command.
Shubham Ajmerad704f0b2017-07-26 16:33:35 -07001130 if (test_profile_method_percerntage_ > 100) {
1131 Usage("Invalid percentage for --generate-test-profile-method-percentage");
David Sehr153da0e2017-02-15 08:54:51 -08001132 }
Shubham Ajmerad704f0b2017-07-26 16:33:35 -07001133 if (test_profile_class_percentage_ > 100) {
1134 Usage("Invalid percentage for --generate-test-profile-class-percentage");
David Sehr153da0e2017-02-15 08:54:51 -08001135 }
Jeff Haof0a31f82017-03-27 15:50:37 -07001136 // If given APK files or DEX locations, check that they're ok.
1137 if (!apk_files_.empty() || !apks_fd_.empty() || !dex_locations_.empty()) {
1138 if (apk_files_.empty() && apks_fd_.empty()) {
1139 Usage("APK files must be specified when passing DEX locations to --generate-test-profile");
1140 }
1141 if (dex_locations_.empty()) {
1142 Usage("DEX locations must be specified when passing APK files to --generate-test-profile");
1143 }
1144 }
David Sehr153da0e2017-02-15 08:54:51 -08001145 // ShouldGenerateTestProfile confirms !test_profile_.empty().
George Burgess IV71f77262016-10-25 13:08:17 -07001146 int profile_test_fd = open(test_profile_.c_str(), O_CREAT | O_TRUNC | O_WRONLY, 0644);
Calin Juravle7bcdb532016-06-07 16:14:47 +01001147 if (profile_test_fd < 0) {
David Sehr7c80f2d2017-02-07 16:47:58 -08001148 LOG(ERROR) << "Cannot open " << test_profile_ << strerror(errno);
Calin Juravle7bcdb532016-06-07 16:14:47 +01001149 return -1;
1150 }
Jeff Haof0a31f82017-03-27 15:50:37 -07001151 bool result;
1152 if (apk_files_.empty() && apks_fd_.empty() && dex_locations_.empty()) {
1153 result = ProfileCompilationInfo::GenerateTestProfile(profile_test_fd,
1154 test_profile_num_dex_,
Shubham Ajmerad704f0b2017-07-26 16:33:35 -07001155 test_profile_method_percerntage_,
1156 test_profile_class_percentage_,
Jeff Haof0a31f82017-03-27 15:50:37 -07001157 test_profile_seed_);
1158 } else {
Jeff Haof0a31f82017-03-27 15:50:37 -07001159 // Open the dex files to look up classes and methods.
1160 std::vector<std::unique_ptr<const DexFile>> dex_files;
1161 OpenApkFilesFromLocations(&dex_files);
1162 // Create a random profile file based on the set of dex files.
1163 result = ProfileCompilationInfo::GenerateTestProfile(profile_test_fd,
1164 dex_files,
Shubham Ajmerad704f0b2017-07-26 16:33:35 -07001165 test_profile_method_percerntage_,
1166 test_profile_class_percentage_,
Jeff Haof0a31f82017-03-27 15:50:37 -07001167 test_profile_seed_);
1168 }
Calin Juravle7bcdb532016-06-07 16:14:47 +01001169 close(profile_test_fd); // ignore close result.
1170 return result ? 0 : -1;
1171 }
1172
1173 bool ShouldGenerateTestProfile() {
1174 return !test_profile_.empty();
1175 }
1176
Calin Juravle2dba0ab2018-01-22 19:22:24 -08001177 bool ShouldCopyAndUpdateProfileKey() const {
1178 return copy_and_update_profile_key_;
1179 }
1180
Calin Juravle02c08792018-02-15 19:40:48 -08001181 int32_t CopyAndUpdateProfileKey() {
Calin Juravle2dba0ab2018-01-22 19:22:24 -08001182 // Validate that at least one profile file was passed, as well as a reference profile.
1183 if (!(profile_files_.size() == 1 ^ profile_files_fd_.size() == 1)) {
1184 Usage("Only one profile file should be specified.");
1185 }
1186 if (reference_profile_file_.empty() && !FdIsValid(reference_profile_file_fd_)) {
1187 Usage("No reference profile file specified.");
1188 }
1189
1190 if (apk_files_.empty() && apks_fd_.empty()) {
1191 Usage("No apk files specified");
1192 }
1193
Calin Juravle02c08792018-02-15 19:40:48 -08001194 static constexpr int32_t kErrorFailedToUpdateProfile = -1;
1195 static constexpr int32_t kErrorFailedToSaveProfile = -2;
1196 static constexpr int32_t kErrorFailedToLoadProfile = -3;
1197
Calin Juravle2dba0ab2018-01-22 19:22:24 -08001198 bool use_fds = profile_files_fd_.size() == 1;
1199
1200 ProfileCompilationInfo profile;
1201 // Do not clear if invalid. The input might be an archive.
Calin Juravle02c08792018-02-15 19:40:48 -08001202 bool load_ok = use_fds
1203 ? profile.Load(profile_files_fd_[0])
1204 : profile.Load(profile_files_[0], /*clear_if_invalid*/ false);
1205 if (load_ok) {
Calin Juravle2dba0ab2018-01-22 19:22:24 -08001206 // Open the dex files to look up classes and methods.
1207 std::vector<std::unique_ptr<const DexFile>> dex_files;
1208 OpenApkFilesFromLocations(&dex_files);
1209 if (!profile.UpdateProfileKeys(dex_files)) {
Calin Juravle02c08792018-02-15 19:40:48 -08001210 return kErrorFailedToUpdateProfile;
Calin Juravle2dba0ab2018-01-22 19:22:24 -08001211 }
Calin Juravle02c08792018-02-15 19:40:48 -08001212 bool result = use_fds
1213 ? profile.Save(reference_profile_file_fd_)
1214 : profile.Save(reference_profile_file_, /*bytes_written*/ nullptr);
1215 return result ? 0 : kErrorFailedToSaveProfile;
Calin Juravle2dba0ab2018-01-22 19:22:24 -08001216 } else {
Calin Juravle02c08792018-02-15 19:40:48 -08001217 return kErrorFailedToLoadProfile;
Calin Juravle2dba0ab2018-01-22 19:22:24 -08001218 }
1219 }
1220
Calin Juravle2e2db782016-02-23 12:00:03 +00001221 private:
1222 static void ParseFdForCollection(const StringPiece& option,
1223 const char* arg_name,
1224 std::vector<int>* fds) {
1225 int fd;
1226 ParseUintOption(option, arg_name, &fd, Usage);
1227 fds->push_back(fd);
1228 }
1229
1230 static void CloseAllFds(const std::vector<int>& fds, const char* descriptor) {
1231 for (size_t i = 0; i < fds.size(); i++) {
1232 if (close(fds[i]) < 0) {
Calin Juravlee10c1e22018-01-26 20:10:15 -08001233 PLOG(WARNING) << "Failed to close descriptor for "
1234 << descriptor << " at index " << i << ": " << fds[i];
Calin Juravle2e2db782016-02-23 12:00:03 +00001235 }
1236 }
1237 }
1238
1239 void LogCompletionTime() {
David Sehr52683cf2016-05-05 09:02:38 -07001240 static constexpr uint64_t kLogThresholdTime = MsToNs(100); // 100ms
1241 uint64_t time_taken = NanoTime() - start_ns_;
David Sehred793012016-05-05 13:39:43 -07001242 if (time_taken > kLogThresholdTime) {
David Sehrd8557182016-05-06 12:29:35 -07001243 LOG(WARNING) << "profman took " << PrettyDuration(time_taken);
David Sehred793012016-05-05 13:39:43 -07001244 }
Calin Juravle2e2db782016-02-23 12:00:03 +00001245 }
1246
1247 std::vector<std::string> profile_files_;
1248 std::vector<int> profile_files_fd_;
David Sehr546d24f2016-06-02 10:46:19 -07001249 std::vector<std::string> dex_locations_;
David Sehr7c80f2d2017-02-07 16:47:58 -08001250 std::vector<std::string> apk_files_;
David Sehr546d24f2016-06-02 10:46:19 -07001251 std::vector<int> apks_fd_;
Calin Juravle2e2db782016-02-23 12:00:03 +00001252 std::string reference_profile_file_;
1253 int reference_profile_file_fd_;
David Sehr4fcdd6d2016-05-24 14:52:31 -07001254 bool dump_only_;
Mathieu Chartier34067262017-04-06 13:55:46 -07001255 bool dump_classes_and_methods_;
Mathieu Chartier2f794552017-06-19 10:58:08 -07001256 bool generate_boot_image_profile_;
David Sehr4fcdd6d2016-05-24 14:52:31 -07001257 int dump_output_to_fd_;
Mathieu Chartier2f794552017-06-19 10:58:08 -07001258 BootImageOptions boot_image_options_;
Calin Juravle7bcdb532016-06-07 16:14:47 +01001259 std::string test_profile_;
David Sehr7c80f2d2017-02-07 16:47:58 -08001260 std::string create_profile_from_file_;
Calin Juravle7bcdb532016-06-07 16:14:47 +01001261 uint16_t test_profile_num_dex_;
Shubham Ajmerad704f0b2017-07-26 16:33:35 -07001262 uint16_t test_profile_method_percerntage_;
1263 uint16_t test_profile_class_percentage_;
Jeff Haof0a31f82017-03-27 15:50:37 -07001264 uint32_t test_profile_seed_;
Calin Juravle2e2db782016-02-23 12:00:03 +00001265 uint64_t start_ns_;
Calin Juravle2dba0ab2018-01-22 19:22:24 -08001266 bool copy_and_update_profile_key_;
Calin Juravle2e2db782016-02-23 12:00:03 +00001267};
1268
1269// See ProfileAssistant::ProcessingResult for return codes.
1270static int profman(int argc, char** argv) {
1271 ProfMan profman;
1272
1273 // Parse arguments. Argument mistakes will lead to exit(EXIT_FAILURE) in UsageError.
1274 profman.ParseArgs(argc, argv);
1275
Calin Juravlee10c1e22018-01-26 20:10:15 -08001276 // Initialize MemMap for ZipArchive::OpenFromFd.
1277 MemMap::Init();
1278
Calin Juravle7bcdb532016-06-07 16:14:47 +01001279 if (profman.ShouldGenerateTestProfile()) {
1280 return profman.GenerateTestProfile();
1281 }
Calin Juravle876f3502016-03-24 16:16:34 +00001282 if (profman.ShouldOnlyDumpProfile()) {
1283 return profman.DumpProfileInfo();
1284 }
Mathieu Chartier34067262017-04-06 13:55:46 -07001285 if (profman.ShouldOnlyDumpClassesAndMethods()) {
Mathieu Chartierea650f32017-05-24 12:04:13 -07001286 return profman.DumpClassesAndMethods();
David Sehr7c80f2d2017-02-07 16:47:58 -08001287 }
1288 if (profman.ShouldCreateProfile()) {
1289 return profman.CreateProfile();
1290 }
Mathieu Chartier2f794552017-06-19 10:58:08 -07001291
1292 if (profman.ShouldCreateBootProfile()) {
1293 return profman.CreateBootProfile();
1294 }
Calin Juravle02c08792018-02-15 19:40:48 -08001295
1296 if (profman.ShouldCopyAndUpdateProfileKey()) {
1297 return profman.CopyAndUpdateProfileKey();
1298 }
1299
Calin Juravle2e2db782016-02-23 12:00:03 +00001300 // Process profile information and assess if we need to do a profile guided compilation.
1301 // This operation involves I/O.
1302 return profman.ProcessProfiles();
1303}
1304
1305} // namespace art
1306
1307int main(int argc, char **argv) {
1308 return art::profman(argc, argv);
1309}