Home | History | Annotate | Download | only in dex2oat
      1 /*
      2  * Copyright (C) 2011 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 
     17 #include <stdio.h>
     18 #include <stdlib.h>
     19 #include <sys/stat.h>
     20 #include <valgrind.h>
     21 
     22 #include <fstream>
     23 #include <iostream>
     24 #include <sstream>
     25 #include <string>
     26 #include <vector>
     27 
     28 #include "base/stl_util.h"
     29 #include "base/stringpiece.h"
     30 #include "base/timing_logger.h"
     31 #include "base/unix_file/fd_file.h"
     32 #include "class_linker.h"
     33 #include "dex_file-inl.h"
     34 #include "driver/compiler_driver.h"
     35 #include "elf_fixup.h"
     36 #include "elf_stripper.h"
     37 #include "gc/space/image_space.h"
     38 #include "gc/space/space-inl.h"
     39 #include "image_writer.h"
     40 #include "leb128.h"
     41 #include "mirror/art_method-inl.h"
     42 #include "mirror/class-inl.h"
     43 #include "mirror/class_loader.h"
     44 #include "mirror/object-inl.h"
     45 #include "mirror/object_array-inl.h"
     46 #include "oat_writer.h"
     47 #include "object_utils.h"
     48 #include "os.h"
     49 #include "runtime.h"
     50 #include "ScopedLocalRef.h"
     51 #include "scoped_thread_state_change.h"
     52 #include "sirt_ref.h"
     53 #include "vector_output_stream.h"
     54 #include "well_known_classes.h"
     55 #include "zip_archive.h"
     56 
     57 namespace art {
     58 
     59 static void UsageErrorV(const char* fmt, va_list ap) {
     60   std::string error;
     61   StringAppendV(&error, fmt, ap);
     62   LOG(ERROR) << error;
     63 }
     64 
     65 static void UsageError(const char* fmt, ...) {
     66   va_list ap;
     67   va_start(ap, fmt);
     68   UsageErrorV(fmt, ap);
     69   va_end(ap);
     70 }
     71 
     72 static void Usage(const char* fmt, ...) {
     73   va_list ap;
     74   va_start(ap, fmt);
     75   UsageErrorV(fmt, ap);
     76   va_end(ap);
     77 
     78   UsageError("Usage: dex2oat [options]...");
     79   UsageError("");
     80   UsageError("  --dex-file=<dex-file>: specifies a .dex file to compile.");
     81   UsageError("      Example: --dex-file=/system/framework/core.jar");
     82   UsageError("");
     83   UsageError("  --zip-fd=<file-descriptor>: specifies a file descriptor of a zip file");
     84   UsageError("      containing a classes.dex file to compile.");
     85   UsageError("      Example: --zip-fd=5");
     86   UsageError("");
     87   UsageError("  --zip-location=<zip-location>: specifies a symbolic name for the file");
     88   UsageError("      corresponding to the file descriptor specified by --zip-fd.");
     89   UsageError("      Example: --zip-location=/system/app/Calculator.apk");
     90   UsageError("");
     91   UsageError("  --oat-file=<file.oat>: specifies the oat output destination via a filename.");
     92   UsageError("      Example: --oat-file=/system/framework/boot.oat");
     93   UsageError("");
     94   UsageError("  --oat-fd=<number>: specifies the oat output destination via a file descriptor.");
     95   UsageError("      Example: --oat-file=/system/framework/boot.oat");
     96   UsageError("");
     97   UsageError("  --oat-location=<oat-name>: specifies a symbolic name for the file corresponding");
     98   UsageError("      to the file descriptor specified by --oat-fd.");
     99   UsageError("      Example: --oat-location=/data/dalvik-cache/system@app (at) Calculator.apk.oat");
    100   UsageError("");
    101   UsageError("  --oat-symbols=<file.oat>: specifies the oat output destination with full symbols.");
    102   UsageError("      Example: --oat-symbols=/symbols/system/framework/boot.oat");
    103   UsageError("");
    104   UsageError("  --bitcode=<file.bc>: specifies the optional bitcode filename.");
    105   UsageError("      Example: --bitcode=/system/framework/boot.bc");
    106   UsageError("");
    107   UsageError("  --image=<file.art>: specifies the output image filename.");
    108   UsageError("      Example: --image=/system/framework/boot.art");
    109   UsageError("");
    110   UsageError("  --image-classes=<classname-file>: specifies classes to include in an image.");
    111   UsageError("      Example: --image=frameworks/base/preloaded-classes");
    112   UsageError("");
    113   UsageError("  --base=<hex-address>: specifies the base address when creating a boot image.");
    114   UsageError("      Example: --base=0x50000000");
    115   UsageError("");
    116   UsageError("  --boot-image=<file.art>: provide the image file for the boot class path.");
    117   UsageError("      Example: --boot-image=/system/framework/boot.art");
    118   UsageError("      Default: <host-prefix>/system/framework/boot.art");
    119   UsageError("");
    120   UsageError("  --host-prefix=<path>: used to translate host paths to target paths during");
    121   UsageError("      cross compilation.");
    122   UsageError("      Example: --host-prefix=out/target/product/crespo");
    123   UsageError("      Default: $ANDROID_PRODUCT_OUT");
    124   UsageError("");
    125   UsageError("  --android-root=<path>: used to locate libraries for portable linking.");
    126   UsageError("      Example: --android-root=out/host/linux-x86");
    127   UsageError("      Default: $ANDROID_ROOT");
    128   UsageError("");
    129   UsageError("  --instruction-set=(arm|mips|x86): compile for a particular instruction");
    130   UsageError("      set.");
    131   UsageError("      Example: --instruction-set=x86");
    132   UsageError("      Default: arm");
    133   UsageError("");
    134   UsageError("  --compiler-backend=(Quick|QuickGBC|Portable): select compiler backend");
    135   UsageError("      set.");
    136   UsageError("      Example: --instruction-set=Portable");
    137   UsageError("      Default: Quick");
    138   UsageError("");
    139   UsageError("  --host: used with Portable backend to link against host runtime libraries");
    140   UsageError("");
    141   UsageError("  --dump-timing: display a breakdown of where time was spent");
    142   UsageError("");
    143   UsageError("  --runtime-arg <argument>: used to specify various arguments for the runtime,");
    144   UsageError("      such as initial heap size, maximum heap size, and verbose output.");
    145   UsageError("      Use a separate --runtime-arg switch for each argument.");
    146   UsageError("      Example: --runtime-arg -Xms256m");
    147   UsageError("");
    148   std::cerr << "See log for usage error information\n";
    149   exit(EXIT_FAILURE);
    150 }
    151 
    152 class Dex2Oat {
    153  public:
    154   static bool Create(Dex2Oat** p_dex2oat,
    155                      Runtime::Options& options,
    156                      CompilerBackend compiler_backend,
    157                      InstructionSet instruction_set,
    158                      size_t thread_count)
    159       SHARED_TRYLOCK_FUNCTION(true, Locks::mutator_lock_) {
    160     if (!CreateRuntime(options, instruction_set)) {
    161       *p_dex2oat = NULL;
    162       return false;
    163     }
    164     *p_dex2oat = new Dex2Oat(Runtime::Current(), compiler_backend, instruction_set, thread_count);
    165     return true;
    166   }
    167 
    168   ~Dex2Oat() {
    169     delete runtime_;
    170     VLOG(compiler) << "dex2oat took " << PrettyDuration(NanoTime() - start_ns_)
    171               << " (threads: " << thread_count_ << ")";
    172   }
    173 
    174 
    175   // Reads the class names (java.lang.Object) and returns a set of descriptors (Ljava/lang/Object;)
    176   CompilerDriver::DescriptorSet* ReadImageClassesFromFile(const char* image_classes_filename) {
    177     UniquePtr<std::ifstream> image_classes_file(new std::ifstream(image_classes_filename,
    178                                                                   std::ifstream::in));
    179     if (image_classes_file.get() == NULL) {
    180       LOG(ERROR) << "Failed to open image classes file " << image_classes_filename;
    181       return NULL;
    182     }
    183     UniquePtr<CompilerDriver::DescriptorSet> result(ReadImageClasses(*image_classes_file.get()));
    184     image_classes_file->close();
    185     return result.release();
    186   }
    187 
    188   CompilerDriver::DescriptorSet* ReadImageClasses(std::istream& image_classes_stream) {
    189     UniquePtr<CompilerDriver::DescriptorSet> image_classes(new CompilerDriver::DescriptorSet);
    190     while (image_classes_stream.good()) {
    191       std::string dot;
    192       std::getline(image_classes_stream, dot);
    193       if (StartsWith(dot, "#") || dot.empty()) {
    194         continue;
    195       }
    196       std::string descriptor(DotToDescriptor(dot.c_str()));
    197       image_classes->insert(descriptor);
    198     }
    199     return image_classes.release();
    200   }
    201 
    202   // Reads the class names (java.lang.Object) and returns a set of descriptors (Ljava/lang/Object;)
    203   CompilerDriver::DescriptorSet* ReadImageClassesFromZip(const std::string& zip_filename,
    204                                                          const char* image_classes_filename) {
    205     UniquePtr<ZipArchive> zip_archive(ZipArchive::Open(zip_filename));
    206     if (zip_archive.get() == NULL) {
    207       LOG(ERROR) << "Failed to open zip file " << zip_filename;
    208       return NULL;
    209     }
    210     UniquePtr<ZipEntry> zip_entry(zip_archive->Find(image_classes_filename));
    211     if (zip_entry.get() == NULL) {
    212       LOG(ERROR) << "Failed to find " << image_classes_filename << " within " << zip_filename;
    213       return NULL;
    214     }
    215     UniquePtr<MemMap> image_classes_file(zip_entry->ExtractToMemMap(image_classes_filename));
    216     if (image_classes_file.get() == NULL) {
    217       LOG(ERROR) << "Failed to extract " << image_classes_filename << " from " << zip_filename;
    218       return NULL;
    219     }
    220     const std::string image_classes_string(reinterpret_cast<char*>(image_classes_file->Begin()),
    221                                            image_classes_file->Size());
    222     std::istringstream image_classes_stream(image_classes_string);
    223     return ReadImageClasses(image_classes_stream);
    224   }
    225 
    226   const CompilerDriver* CreateOatFile(const std::string& boot_image_option,
    227                                       const std::string* host_prefix,
    228                                       const std::string& android_root,
    229                                       bool is_host,
    230                                       const std::vector<const DexFile*>& dex_files,
    231                                       File* oat_file,
    232                                       const std::string& bitcode_filename,
    233                                       bool image,
    234                                       UniquePtr<CompilerDriver::DescriptorSet>& image_classes,
    235                                       bool dump_stats,
    236                                       base::TimingLogger& timings) {
    237     // SirtRef and ClassLoader creation needs to come after Runtime::Create
    238     jobject class_loader = NULL;
    239     Thread* self = Thread::Current();
    240     if (!boot_image_option.empty()) {
    241       ClassLinker* class_linker = Runtime::Current()->GetClassLinker();
    242       std::vector<const DexFile*> class_path_files(dex_files);
    243       OpenClassPathFiles(runtime_->GetClassPathString(), class_path_files);
    244       ScopedObjectAccess soa(self);
    245       for (size_t i = 0; i < class_path_files.size(); i++) {
    246         class_linker->RegisterDexFile(*class_path_files[i]);
    247       }
    248       soa.Env()->AllocObject(WellKnownClasses::dalvik_system_PathClassLoader);
    249       ScopedLocalRef<jobject> class_loader_local(soa.Env(),
    250           soa.Env()->AllocObject(WellKnownClasses::dalvik_system_PathClassLoader));
    251       class_loader = soa.Env()->NewGlobalRef(class_loader_local.get());
    252       Runtime::Current()->SetCompileTimeClassPath(class_loader, class_path_files);
    253     }
    254 
    255     UniquePtr<CompilerDriver> driver(new CompilerDriver(compiler_backend_,
    256                                                         instruction_set_,
    257                                                         image,
    258                                                         image_classes.release(),
    259                                                         thread_count_,
    260                                                         dump_stats));
    261 
    262     if (compiler_backend_ == kPortable) {
    263       driver->SetBitcodeFileName(bitcode_filename);
    264     }
    265 
    266     driver->CompileAll(class_loader, dex_files, timings);
    267 
    268     timings.NewSplit("dex2oat OatWriter");
    269     std::string image_file_location;
    270     uint32_t image_file_location_oat_checksum = 0;
    271     uint32_t image_file_location_oat_data_begin = 0;
    272     if (!driver->IsImage()) {
    273       gc::space::ImageSpace* image_space = Runtime::Current()->GetHeap()->GetImageSpace();
    274       image_file_location_oat_checksum = image_space->GetImageHeader().GetOatChecksum();
    275       image_file_location_oat_data_begin =
    276           reinterpret_cast<uint32_t>(image_space->GetImageHeader().GetOatDataBegin());
    277       image_file_location = image_space->GetImageFilename();
    278       if (host_prefix != NULL && StartsWith(image_file_location, host_prefix->c_str())) {
    279         image_file_location = image_file_location.substr(host_prefix->size());
    280       }
    281     }
    282 
    283     OatWriter oat_writer(dex_files,
    284                          image_file_location_oat_checksum,
    285                          image_file_location_oat_data_begin,
    286                          image_file_location,
    287                          driver.get());
    288 
    289     if (!driver->WriteElf(android_root, is_host, dex_files, oat_writer, oat_file)) {
    290       LOG(ERROR) << "Failed to write ELF file " << oat_file->GetPath();
    291       return NULL;
    292     }
    293 
    294     return driver.release();
    295   }
    296 
    297   bool CreateImageFile(const std::string& image_filename,
    298                        uintptr_t image_base,
    299                        const std::string& oat_filename,
    300                        const std::string& oat_location,
    301                        const CompilerDriver& compiler)
    302       LOCKS_EXCLUDED(Locks::mutator_lock_) {
    303     uintptr_t oat_data_begin;
    304     {
    305       // ImageWriter is scoped so it can free memory before doing FixupElf
    306       ImageWriter image_writer(compiler);
    307       if (!image_writer.Write(image_filename, image_base, oat_filename, oat_location)) {
    308         LOG(ERROR) << "Failed to create image file " << image_filename;
    309         return false;
    310       }
    311       oat_data_begin = image_writer.GetOatDataBegin();
    312     }
    313 
    314     UniquePtr<File> oat_file(OS::OpenFileReadWrite(oat_filename.c_str()));
    315     if (oat_file.get() == NULL) {
    316       PLOG(ERROR) << "Failed to open ELF file: " << oat_filename;
    317       return false;
    318     }
    319     if (!ElfFixup::Fixup(oat_file.get(), oat_data_begin)) {
    320       LOG(ERROR) << "Failed to fixup ELF file " << oat_file->GetPath();
    321       return false;
    322     }
    323     return true;
    324   }
    325 
    326  private:
    327   explicit Dex2Oat(Runtime* runtime,
    328                    CompilerBackend compiler_backend,
    329                    InstructionSet instruction_set,
    330                    size_t thread_count)
    331       : compiler_backend_(compiler_backend),
    332         instruction_set_(instruction_set),
    333         runtime_(runtime),
    334         thread_count_(thread_count),
    335         start_ns_(NanoTime()) {
    336   }
    337 
    338   static bool CreateRuntime(Runtime::Options& options, InstructionSet instruction_set)
    339       SHARED_TRYLOCK_FUNCTION(true, Locks::mutator_lock_) {
    340     if (!Runtime::Create(options, false)) {
    341       LOG(ERROR) << "Failed to create runtime";
    342       return false;
    343     }
    344     Runtime* runtime = Runtime::Current();
    345     // if we loaded an existing image, we will reuse values from the image roots.
    346     if (!runtime->HasResolutionMethod()) {
    347       runtime->SetResolutionMethod(runtime->CreateResolutionMethod());
    348     }
    349     for (int i = 0; i < Runtime::kLastCalleeSaveType; i++) {
    350       Runtime::CalleeSaveType type = Runtime::CalleeSaveType(i);
    351       if (!runtime->HasCalleeSaveMethod(type)) {
    352         runtime->SetCalleeSaveMethod(runtime->CreateCalleeSaveMethod(instruction_set, type), type);
    353       }
    354     }
    355     runtime->GetClassLinker()->FixupDexCaches(runtime->GetResolutionMethod());
    356     return true;
    357   }
    358 
    359   // Appends to dex_files any elements of class_path that it doesn't already
    360   // contain. This will open those dex files as necessary.
    361   static void OpenClassPathFiles(const std::string& class_path,
    362                                  std::vector<const DexFile*>& dex_files) {
    363     std::vector<std::string> parsed;
    364     Split(class_path, ':', parsed);
    365     // Take Locks::mutator_lock_ so that lock ordering on the ClassLinker::dex_lock_ is maintained.
    366     ScopedObjectAccess soa(Thread::Current());
    367     for (size_t i = 0; i < parsed.size(); ++i) {
    368       if (DexFilesContains(dex_files, parsed[i])) {
    369         continue;
    370       }
    371       const DexFile* dex_file = DexFile::Open(parsed[i], parsed[i]);
    372       if (dex_file == NULL) {
    373         LOG(WARNING) << "Failed to open dex file " << parsed[i];
    374       } else {
    375         dex_files.push_back(dex_file);
    376       }
    377     }
    378   }
    379 
    380   // Returns true if dex_files has a dex with the named location.
    381   static bool DexFilesContains(const std::vector<const DexFile*>& dex_files,
    382                                const std::string& location) {
    383     for (size_t i = 0; i < dex_files.size(); ++i) {
    384       if (dex_files[i]->GetLocation() == location) {
    385         return true;
    386       }
    387     }
    388     return false;
    389   }
    390 
    391   const CompilerBackend compiler_backend_;
    392 
    393   const InstructionSet instruction_set_;
    394 
    395   Runtime* runtime_;
    396   size_t thread_count_;
    397   uint64_t start_ns_;
    398 
    399   DISALLOW_IMPLICIT_CONSTRUCTORS(Dex2Oat);
    400 };
    401 
    402 static bool ParseInt(const char* in, int* out) {
    403   char* end;
    404   int result = strtol(in, &end, 10);
    405   if (in == end || *end != '\0') {
    406     return false;
    407   }
    408   *out = result;
    409   return true;
    410 }
    411 
    412 static size_t OpenDexFiles(const std::vector<const char*>& dex_filenames,
    413                            const std::vector<const char*>& dex_locations,
    414                            std::vector<const DexFile*>& dex_files) {
    415   size_t failure_count = 0;
    416   for (size_t i = 0; i < dex_filenames.size(); i++) {
    417     const char* dex_filename = dex_filenames[i];
    418     const char* dex_location = dex_locations[i];
    419     const DexFile* dex_file = DexFile::Open(dex_filename, dex_location);
    420     if (dex_file == NULL) {
    421       LOG(WARNING) << "Failed to open .dex from file '" << dex_filename << "'\n";
    422       ++failure_count;
    423     } else {
    424       dex_files.push_back(dex_file);
    425     }
    426   }
    427   return failure_count;
    428 }
    429 
    430 // The primary goal of the watchdog is to prevent stuck build servers
    431 // during development when fatal aborts lead to a cascade of failures
    432 // that result in a deadlock.
    433 class WatchDog {
    434 // WatchDog defines its own CHECK_PTHREAD_CALL to avoid using Log which uses locks
    435 #undef CHECK_PTHREAD_CALL
    436 #define CHECK_WATCH_DOG_PTHREAD_CALL(call, args, what) \
    437   do { \
    438     int rc = call args; \
    439     if (rc != 0) { \
    440       errno = rc; \
    441       std::string message(# call); \
    442       message += " failed for "; \
    443       message += reason; \
    444       Fatal(message); \
    445     } \
    446   } while (false)
    447 
    448  public:
    449   explicit WatchDog(bool is_watch_dog_enabled) {
    450     is_watch_dog_enabled_ = is_watch_dog_enabled;
    451     if (!is_watch_dog_enabled_) {
    452       return;
    453     }
    454     shutting_down_ = false;
    455     const char* reason = "dex2oat watch dog thread startup";
    456     CHECK_WATCH_DOG_PTHREAD_CALL(pthread_mutex_init, (&mutex_, NULL), reason);
    457     CHECK_WATCH_DOG_PTHREAD_CALL(pthread_cond_init, (&cond_, NULL), reason);
    458     CHECK_WATCH_DOG_PTHREAD_CALL(pthread_attr_init, (&attr_), reason);
    459     CHECK_WATCH_DOG_PTHREAD_CALL(pthread_create, (&pthread_, &attr_, &CallBack, this), reason);
    460     CHECK_WATCH_DOG_PTHREAD_CALL(pthread_attr_destroy, (&attr_), reason);
    461   }
    462   ~WatchDog() {
    463     if (!is_watch_dog_enabled_) {
    464       return;
    465     }
    466     const char* reason = "dex2oat watch dog thread shutdown";
    467     CHECK_WATCH_DOG_PTHREAD_CALL(pthread_mutex_lock, (&mutex_), reason);
    468     shutting_down_ = true;
    469     CHECK_WATCH_DOG_PTHREAD_CALL(pthread_cond_signal, (&cond_), reason);
    470     CHECK_WATCH_DOG_PTHREAD_CALL(pthread_mutex_unlock, (&mutex_), reason);
    471 
    472     CHECK_WATCH_DOG_PTHREAD_CALL(pthread_join, (pthread_, NULL), reason);
    473 
    474     CHECK_WATCH_DOG_PTHREAD_CALL(pthread_cond_destroy, (&cond_), reason);
    475     CHECK_WATCH_DOG_PTHREAD_CALL(pthread_mutex_destroy, (&mutex_), reason);
    476   }
    477 
    478  private:
    479   static void* CallBack(void* arg) {
    480     WatchDog* self = reinterpret_cast<WatchDog*>(arg);
    481     ::art::SetThreadName("dex2oat watch dog");
    482     self->Wait();
    483     return NULL;
    484   }
    485 
    486   static void Message(char severity, const std::string& message) {
    487     // TODO: Remove when we switch to LOG when we can guarantee it won't prevent shutdown in error
    488     //       cases.
    489     fprintf(stderr, "dex2oat%s %c %d %d %s\n",
    490             kIsDebugBuild ? "d" : "",
    491             severity,
    492             getpid(),
    493             GetTid(),
    494             message.c_str());
    495   }
    496 
    497   static void Warn(const std::string& message) {
    498     Message('W', message);
    499   }
    500 
    501   static void Fatal(const std::string& message) {
    502     Message('F', message);
    503     exit(1);
    504   }
    505 
    506   void Wait() {
    507     bool warning = true;
    508     CHECK_GT(kWatchDogTimeoutSeconds, kWatchDogWarningSeconds);
    509     // TODO: tune the multiplier for GC verification, the following is just to make the timeout
    510     //       large.
    511     int64_t multiplier = gc::kDesiredHeapVerification > gc::kVerifyAllFast ? 100 : 1;
    512     timespec warning_ts;
    513     InitTimeSpec(true, CLOCK_REALTIME, multiplier * kWatchDogWarningSeconds * 1000, 0, &warning_ts);
    514     timespec timeout_ts;
    515     InitTimeSpec(true, CLOCK_REALTIME, multiplier * kWatchDogTimeoutSeconds * 1000, 0, &timeout_ts);
    516     const char* reason = "dex2oat watch dog thread waiting";
    517     CHECK_WATCH_DOG_PTHREAD_CALL(pthread_mutex_lock, (&mutex_), reason);
    518     while (!shutting_down_) {
    519       int rc = TEMP_FAILURE_RETRY(pthread_cond_timedwait(&cond_, &mutex_,
    520                                                          warning ? &warning_ts
    521                                                                  : &timeout_ts));
    522       if (rc == ETIMEDOUT) {
    523         std::string message(StringPrintf("dex2oat did not finish after %d seconds",
    524                                          warning ? kWatchDogWarningSeconds
    525                                                  : kWatchDogTimeoutSeconds));
    526         if (warning) {
    527           Warn(message.c_str());
    528           warning = false;
    529         } else {
    530           Fatal(message.c_str());
    531         }
    532       } else if (rc != 0) {
    533         std::string message(StringPrintf("pthread_cond_timedwait failed: %s",
    534                                          strerror(errno)));
    535         Fatal(message.c_str());
    536       }
    537     }
    538     CHECK_WATCH_DOG_PTHREAD_CALL(pthread_mutex_unlock, (&mutex_), reason);
    539   }
    540 
    541   // When setting timeouts, keep in mind that the build server may not be as fast as your desktop.
    542 #if ART_USE_PORTABLE_COMPILER
    543   static const unsigned int kWatchDogWarningSeconds =  2 * 60;  // 2 minutes.
    544   static const unsigned int kWatchDogTimeoutSeconds = 30 * 60;  // 25 minutes + buffer.
    545 #else
    546   static const unsigned int kWatchDogWarningSeconds =  1 * 60;  // 1 minute.
    547   static const unsigned int kWatchDogTimeoutSeconds =  6 * 60;  // 5 minutes + buffer.
    548 #endif
    549 
    550   bool is_watch_dog_enabled_;
    551   bool shutting_down_;
    552   // TODO: Switch to Mutex when we can guarantee it won't prevent shutdown in error cases.
    553   pthread_mutex_t mutex_;
    554   pthread_cond_t cond_;
    555   pthread_attr_t attr_;
    556   pthread_t pthread_;
    557 };
    558 const unsigned int WatchDog::kWatchDogWarningSeconds;
    559 const unsigned int WatchDog::kWatchDogTimeoutSeconds;
    560 
    561 static int dex2oat(int argc, char** argv) {
    562   base::TimingLogger timings("compiler", false, false);
    563 
    564   InitLogging(argv);
    565 
    566   // Skip over argv[0].
    567   argv++;
    568   argc--;
    569 
    570   if (argc == 0) {
    571     Usage("No arguments specified");
    572   }
    573 
    574   std::vector<const char*> dex_filenames;
    575   std::vector<const char*> dex_locations;
    576   int zip_fd = -1;
    577   std::string zip_location;
    578   std::string oat_filename;
    579   std::string oat_symbols;
    580   std::string oat_location;
    581   int oat_fd = -1;
    582   std::string bitcode_filename;
    583   const char* image_classes_zip_filename = NULL;
    584   const char* image_classes_filename = NULL;
    585   std::string image_filename;
    586   std::string boot_image_filename;
    587   uintptr_t image_base = 0;
    588   UniquePtr<std::string> host_prefix;
    589   std::string android_root;
    590   std::vector<const char*> runtime_args;
    591   int thread_count = sysconf(_SC_NPROCESSORS_CONF);
    592 #if defined(ART_USE_PORTABLE_COMPILER)
    593   CompilerBackend compiler_backend = kPortable;
    594 #else
    595   CompilerBackend compiler_backend = kQuick;
    596 #endif
    597 #if defined(__arm__)
    598   InstructionSet instruction_set = kThumb2;
    599 #elif defined(__i386__)
    600   InstructionSet instruction_set = kX86;
    601 #elif defined(__mips__)
    602   InstructionSet instruction_set = kMips;
    603 #else
    604 #error "Unsupported architecture"
    605 #endif
    606   bool is_host = false;
    607   bool dump_stats = kIsDebugBuild;
    608   bool dump_timing = false;
    609   bool dump_slow_timing = kIsDebugBuild;
    610   bool watch_dog_enabled = !kIsTargetBuild;
    611 
    612 
    613   for (int i = 0; i < argc; i++) {
    614     const StringPiece option(argv[i]);
    615     bool log_options = false;
    616     if (log_options) {
    617       LOG(INFO) << "dex2oat: option[" << i << "]=" << argv[i];
    618     }
    619     if (option.starts_with("--dex-file=")) {
    620       dex_filenames.push_back(option.substr(strlen("--dex-file=")).data());
    621     } else if (option.starts_with("--dex-location=")) {
    622       dex_locations.push_back(option.substr(strlen("--dex-location=")).data());
    623     } else if (option.starts_with("--zip-fd=")) {
    624       const char* zip_fd_str = option.substr(strlen("--zip-fd=")).data();
    625       if (!ParseInt(zip_fd_str, &zip_fd)) {
    626         Usage("Failed to parse --zip-fd argument '%s' as an integer", zip_fd_str);
    627       }
    628     } else if (option.starts_with("--zip-location=")) {
    629       zip_location = option.substr(strlen("--zip-location=")).data();
    630     } else if (option.starts_with("--oat-file=")) {
    631       oat_filename = option.substr(strlen("--oat-file=")).data();
    632     } else if (option.starts_with("--oat-symbols=")) {
    633       oat_symbols = option.substr(strlen("--oat-symbols=")).data();
    634     } else if (option.starts_with("--oat-fd=")) {
    635       const char* oat_fd_str = option.substr(strlen("--oat-fd=")).data();
    636       if (!ParseInt(oat_fd_str, &oat_fd)) {
    637         Usage("Failed to parse --oat-fd argument '%s' as an integer", oat_fd_str);
    638       }
    639     } else if (option == "--watch-dog") {
    640       watch_dog_enabled = true;
    641     } else if (option == "--no-watch-dog") {
    642       watch_dog_enabled = false;
    643     } else if (option.starts_with("-j")) {
    644       const char* thread_count_str = option.substr(strlen("-j")).data();
    645       if (!ParseInt(thread_count_str, &thread_count)) {
    646         Usage("Failed to parse -j argument '%s' as an integer", thread_count_str);
    647       }
    648     } else if (option.starts_with("--oat-location=")) {
    649       oat_location = option.substr(strlen("--oat-location=")).data();
    650     } else if (option.starts_with("--bitcode=")) {
    651       bitcode_filename = option.substr(strlen("--bitcode=")).data();
    652     } else if (option.starts_with("--image=")) {
    653       image_filename = option.substr(strlen("--image=")).data();
    654     } else if (option.starts_with("--image-classes=")) {
    655       image_classes_filename = option.substr(strlen("--image-classes=")).data();
    656     } else if (option.starts_with("--image-classes-zip=")) {
    657       image_classes_zip_filename = option.substr(strlen("--image-classes-zip=")).data();
    658     } else if (option.starts_with("--base=")) {
    659       const char* image_base_str = option.substr(strlen("--base=")).data();
    660       char* end;
    661       image_base = strtoul(image_base_str, &end, 16);
    662       if (end == image_base_str || *end != '\0') {
    663         Usage("Failed to parse hexadecimal value for option %s", option.data());
    664       }
    665     } else if (option.starts_with("--boot-image=")) {
    666       boot_image_filename = option.substr(strlen("--boot-image=")).data();
    667     } else if (option.starts_with("--host-prefix=")) {
    668       host_prefix.reset(new std::string(option.substr(strlen("--host-prefix=")).data()));
    669     } else if (option.starts_with("--android-root=")) {
    670       android_root = option.substr(strlen("--android-root=")).data();
    671     } else if (option.starts_with("--instruction-set=")) {
    672       StringPiece instruction_set_str = option.substr(strlen("--instruction-set=")).data();
    673       if (instruction_set_str == "arm") {
    674         instruction_set = kThumb2;
    675       } else if (instruction_set_str == "mips") {
    676         instruction_set = kMips;
    677       } else if (instruction_set_str == "x86") {
    678         instruction_set = kX86;
    679       }
    680     } else if (option.starts_with("--compiler-backend=")) {
    681       StringPiece backend_str = option.substr(strlen("--compiler-backend=")).data();
    682       if (backend_str == "Quick") {
    683         compiler_backend = kQuick;
    684       } else if (backend_str == "Portable") {
    685         compiler_backend = kPortable;
    686       }
    687     } else if (option == "--host") {
    688       is_host = true;
    689     } else if (option == "--runtime-arg") {
    690       if (++i >= argc) {
    691         Usage("Missing required argument for --runtime-arg");
    692       }
    693       if (log_options) {
    694         LOG(INFO) << "dex2oat: option[" << i << "]=" << argv[i];
    695       }
    696       runtime_args.push_back(argv[i]);
    697     } else if (option == "--dump-timing") {
    698       dump_timing = true;
    699     } else {
    700       Usage("Unknown argument %s", option.data());
    701     }
    702   }
    703 
    704   if (oat_filename.empty() && oat_fd == -1) {
    705     Usage("Output must be supplied with either --oat-file or --oat-fd");
    706   }
    707 
    708   if (!oat_filename.empty() && oat_fd != -1) {
    709     Usage("--oat-file should not be used with --oat-fd");
    710   }
    711 
    712   if (!oat_symbols.empty() && oat_fd != -1) {
    713     Usage("--oat-symbols should not be used with --oat-fd");
    714   }
    715 
    716   if (!oat_symbols.empty() && is_host) {
    717     Usage("--oat-symbols should not be used with --host");
    718   }
    719 
    720   if (oat_fd != -1 && !image_filename.empty()) {
    721     Usage("--oat-fd should not be used with --image");
    722   }
    723 
    724   if (host_prefix.get() == NULL) {
    725     const char* android_product_out = getenv("ANDROID_PRODUCT_OUT");
    726     if (android_product_out != NULL) {
    727         host_prefix.reset(new std::string(android_product_out));
    728     }
    729   }
    730 
    731   if (android_root.empty()) {
    732     const char* android_root_env_var = getenv("ANDROID_ROOT");
    733     if (android_root_env_var == NULL) {
    734       Usage("--android-root unspecified and ANDROID_ROOT not set");
    735     }
    736     android_root += android_root_env_var;
    737   }
    738 
    739   bool image = (!image_filename.empty());
    740   if (!image && boot_image_filename.empty()) {
    741     if (host_prefix.get() == NULL) {
    742       boot_image_filename += GetAndroidRoot();
    743     } else {
    744       boot_image_filename += *host_prefix.get();
    745       boot_image_filename += "/system";
    746     }
    747     boot_image_filename += "/framework/boot.art";
    748   }
    749   std::string boot_image_option;
    750   if (!boot_image_filename.empty()) {
    751     boot_image_option += "-Ximage:";
    752     boot_image_option += boot_image_filename;
    753   }
    754 
    755   if (image_classes_filename != NULL && !image) {
    756     Usage("--image-classes should only be used with --image");
    757   }
    758 
    759   if (image_classes_filename != NULL && !boot_image_option.empty()) {
    760     Usage("--image-classes should not be used with --boot-image");
    761   }
    762 
    763   if (image_classes_zip_filename != NULL && image_classes_filename == NULL) {
    764     Usage("--image-classes-zip should be used with --image-classes");
    765   }
    766 
    767   if (dex_filenames.empty() && zip_fd == -1) {
    768     Usage("Input must be supplied with either --dex-file or --zip-fd");
    769   }
    770 
    771   if (!dex_filenames.empty() && zip_fd != -1) {
    772     Usage("--dex-file should not be used with --zip-fd");
    773   }
    774 
    775   if (!dex_filenames.empty() && !zip_location.empty()) {
    776     Usage("--dex-file should not be used with --zip-location");
    777   }
    778 
    779   if (dex_locations.empty()) {
    780     for (size_t i = 0; i < dex_filenames.size(); i++) {
    781       dex_locations.push_back(dex_filenames[i]);
    782     }
    783   } else if (dex_locations.size() != dex_filenames.size()) {
    784     Usage("--dex-location arguments do not match --dex-file arguments");
    785   }
    786 
    787   if (zip_fd != -1 && zip_location.empty()) {
    788     Usage("--zip-location should be supplied with --zip-fd");
    789   }
    790 
    791   if (boot_image_option.empty()) {
    792     if (image_base == 0) {
    793       Usage("Non-zero --base not specified");
    794     }
    795   }
    796 
    797   std::string oat_stripped(oat_filename);
    798   std::string oat_unstripped;
    799   if (!oat_symbols.empty()) {
    800     oat_unstripped += oat_symbols;
    801   } else {
    802     oat_unstripped += oat_filename;
    803   }
    804 
    805   // Done with usage checks, enable watchdog if requested
    806   WatchDog watch_dog(watch_dog_enabled);
    807 
    808   // Check early that the result of compilation can be written
    809   UniquePtr<File> oat_file;
    810   bool create_file = !oat_unstripped.empty();  // as opposed to using open file descriptor
    811   if (create_file) {
    812     oat_file.reset(OS::CreateEmptyFile(oat_unstripped.c_str()));
    813     if (oat_location.empty()) {
    814       oat_location = oat_filename;
    815     }
    816   } else {
    817     oat_file.reset(new File(oat_fd, oat_location));
    818     oat_file->DisableAutoClose();
    819   }
    820   if (oat_file.get() == NULL) {
    821     PLOG(ERROR) << "Failed to create oat file: " << oat_location;
    822     return EXIT_FAILURE;
    823   }
    824   if (create_file && fchmod(oat_file->Fd(), 0644) != 0) {
    825     PLOG(ERROR) << "Failed to make oat file world readable: " << oat_location;
    826     return EXIT_FAILURE;
    827   }
    828 
    829   timings.StartSplit("dex2oat Setup");
    830   LOG(INFO) << "dex2oat: " << oat_location;
    831 
    832   if (image) {
    833     bool has_compiler_filter = false;
    834     for (const char* r : runtime_args) {
    835       if (strncmp(r, "-compiler-filter:", 17) == 0) {
    836         has_compiler_filter = true;
    837         break;
    838       }
    839     }
    840     if (!has_compiler_filter) {
    841       runtime_args.push_back("-compiler-filter:everything");
    842     }
    843   }
    844 
    845   Runtime::Options options;
    846   options.push_back(std::make_pair("compiler", reinterpret_cast<void*>(NULL)));
    847   std::vector<const DexFile*> boot_class_path;
    848   if (boot_image_option.empty()) {
    849     size_t failure_count = OpenDexFiles(dex_filenames, dex_locations, boot_class_path);
    850     if (failure_count > 0) {
    851       LOG(ERROR) << "Failed to open some dex files: " << failure_count;
    852       return EXIT_FAILURE;
    853     }
    854     options.push_back(std::make_pair("bootclasspath", &boot_class_path));
    855   } else {
    856     options.push_back(std::make_pair(boot_image_option.c_str(), reinterpret_cast<void*>(NULL)));
    857   }
    858   if (host_prefix.get() != NULL) {
    859     options.push_back(std::make_pair("host-prefix", host_prefix->c_str()));
    860   }
    861   for (size_t i = 0; i < runtime_args.size(); i++) {
    862     options.push_back(std::make_pair(runtime_args[i], reinterpret_cast<void*>(NULL)));
    863   }
    864 
    865 #ifdef ART_SEA_IR_MODE
    866   options.push_back(std::make_pair("-sea_ir", reinterpret_cast<void*>(NULL)));
    867 #endif
    868 
    869   Dex2Oat* p_dex2oat;
    870   if (!Dex2Oat::Create(&p_dex2oat, options, compiler_backend, instruction_set, thread_count)) {
    871     LOG(ERROR) << "Failed to create dex2oat";
    872     return EXIT_FAILURE;
    873   }
    874   UniquePtr<Dex2Oat> dex2oat(p_dex2oat);
    875   // Runtime::Create acquired the mutator_lock_ that is normally given away when we Runtime::Start,
    876   // give it away now so that we don't starve GC.
    877   Thread* self = Thread::Current();
    878   self->TransitionFromRunnableToSuspended(kNative);
    879   // If we're doing the image, override the compiler filter to force full compilation. Must be
    880   // done ahead of WellKnownClasses::Init that causes verification.  Note: doesn't force
    881   // compilation of class initializers.
    882   // Whilst we're in native take the opportunity to initialize well known classes.
    883   WellKnownClasses::Init(self->GetJniEnv());
    884 
    885   // If --image-classes was specified, calculate the full list of classes to include in the image
    886   UniquePtr<CompilerDriver::DescriptorSet> image_classes(NULL);
    887   if (image_classes_filename != NULL) {
    888     if (image_classes_zip_filename != NULL) {
    889       image_classes.reset(dex2oat->ReadImageClassesFromZip(image_classes_zip_filename,
    890                                                            image_classes_filename));
    891     } else {
    892       image_classes.reset(dex2oat->ReadImageClassesFromFile(image_classes_filename));
    893     }
    894     if (image_classes.get() == NULL) {
    895       LOG(ERROR) << "Failed to create list of image classes from " << image_classes_filename;
    896       return EXIT_FAILURE;
    897     }
    898   }
    899 
    900   std::vector<const DexFile*> dex_files;
    901   if (boot_image_option.empty()) {
    902     dex_files = Runtime::Current()->GetClassLinker()->GetBootClassPath();
    903   } else {
    904     if (dex_filenames.empty()) {
    905       UniquePtr<ZipArchive> zip_archive(ZipArchive::OpenFromFd(zip_fd));
    906       if (zip_archive.get() == NULL) {
    907         LOG(ERROR) << "Failed to open zip from file descriptor for " << zip_location;
    908         return EXIT_FAILURE;
    909       }
    910       const DexFile* dex_file = DexFile::Open(*zip_archive.get(), zip_location);
    911       if (dex_file == NULL) {
    912         LOG(ERROR) << "Failed to open dex from file descriptor for zip file: " << zip_location;
    913         return EXIT_FAILURE;
    914       }
    915       dex_files.push_back(dex_file);
    916     } else {
    917       size_t failure_count = OpenDexFiles(dex_filenames, dex_locations, dex_files);
    918       if (failure_count > 0) {
    919         LOG(ERROR) << "Failed to open some dex files: " << failure_count;
    920         return EXIT_FAILURE;
    921       }
    922     }
    923 
    924     // Ensure opened dex files are writable for dex-to-dex transformations.
    925     for (const auto& dex_file : dex_files) {
    926       if (!dex_file->EnableWrite()) {
    927         PLOG(ERROR) << "Failed to make .dex file writeable '" << dex_file->GetLocation() << "'\n";
    928       }
    929     }
    930   }
    931 
    932   /*
    933    * If we're not in interpret-only mode, go ahead and compile small applications. Don't
    934    * bother to check if we're doing the image.
    935    */
    936   if (!image && (Runtime::Current()->GetCompilerFilter() != Runtime::kInterpretOnly)) {
    937     size_t num_methods = 0;
    938     for (size_t i = 0; i != dex_files.size(); ++i) {
    939       const DexFile* dex_file = dex_files[i];
    940       CHECK(dex_file != NULL);
    941       num_methods += dex_file->NumMethodIds();
    942     }
    943     if (num_methods <= Runtime::Current()->GetNumDexMethodsThreshold()) {
    944       Runtime::Current()->SetCompilerFilter(Runtime::kSpeed);
    945       VLOG(compiler) << "Below method threshold, compiling anyways";
    946     }
    947   }
    948 
    949   UniquePtr<const CompilerDriver> compiler(dex2oat->CreateOatFile(boot_image_option,
    950                                                                   host_prefix.get(),
    951                                                                   android_root,
    952                                                                   is_host,
    953                                                                   dex_files,
    954                                                                   oat_file.get(),
    955                                                                   bitcode_filename,
    956                                                                   image,
    957                                                                   image_classes,
    958                                                                   dump_stats,
    959                                                                   timings));
    960 
    961   if (compiler.get() == NULL) {
    962     LOG(ERROR) << "Failed to create oat file: " << oat_location;
    963     return EXIT_FAILURE;
    964   }
    965 
    966   VLOG(compiler) << "Oat file written successfully (unstripped): " << oat_location;
    967 
    968   // Notes on the interleaving of creating the image and oat file to
    969   // ensure the references between the two are correct.
    970   //
    971   // Currently we have a memory layout that looks something like this:
    972   //
    973   // +--------------+
    974   // | image        |
    975   // +--------------+
    976   // | boot oat     |
    977   // +--------------+
    978   // | alloc spaces |
    979   // +--------------+
    980   //
    981   // There are several constraints on the loading of the image and boot.oat.
    982   //
    983   // 1. The image is expected to be loaded at an absolute address and
    984   // contains Objects with absolute pointers within the image.
    985   //
    986   // 2. There are absolute pointers from Methods in the image to their
    987   // code in the oat.
    988   //
    989   // 3. There are absolute pointers from the code in the oat to Methods
    990   // in the image.
    991   //
    992   // 4. There are absolute pointers from code in the oat to other code
    993   // in the oat.
    994   //
    995   // To get this all correct, we go through several steps.
    996   //
    997   // 1. We have already created that oat file above with
    998   // CreateOatFile. Originally this was just our own proprietary file
    999   // but now it is contained within an ELF dynamic object (aka an .so
   1000   // file). The Compiler returned by CreateOatFile provides
   1001   // PatchInformation for references to oat code and Methods that need
   1002   // to be update once we know where the oat file will be located
   1003   // after the image.
   1004   //
   1005   // 2. We create the image file. It needs to know where the oat file
   1006   // will be loaded after itself. Originally when oat file was simply
   1007   // memory mapped so we could predict where its contents were based
   1008   // on the file size. Now that it is an ELF file, we need to inspect
   1009   // the ELF file to understand the in memory segment layout including
   1010   // where the oat header is located within. ImageWriter's
   1011   // PatchOatCodeAndMethods uses the PatchInformation from the
   1012   // Compiler to touch up absolute references in the oat file.
   1013   //
   1014   // 3. We fixup the ELF program headers so that dlopen will try to
   1015   // load the .so at the desired location at runtime by offsetting the
   1016   // Elf32_Phdr.p_vaddr values by the desired base address.
   1017   //
   1018   if (image) {
   1019     timings.NewSplit("dex2oat ImageWriter");
   1020     bool image_creation_success = dex2oat->CreateImageFile(image_filename,
   1021                                                            image_base,
   1022                                                            oat_unstripped,
   1023                                                            oat_location,
   1024                                                            *compiler.get());
   1025     if (!image_creation_success) {
   1026       return EXIT_FAILURE;
   1027     }
   1028     VLOG(compiler) << "Image written successfully: " << image_filename;
   1029   }
   1030 
   1031   if (is_host) {
   1032     if (dump_timing || (dump_slow_timing && timings.GetTotalNs() > MsToNs(1000))) {
   1033       LOG(INFO) << Dumpable<base::TimingLogger>(timings);
   1034     }
   1035     return EXIT_SUCCESS;
   1036   }
   1037 
   1038   // If we don't want to strip in place, copy from unstripped location to stripped location.
   1039   // We need to strip after image creation because FixupElf needs to use .strtab.
   1040   if (oat_unstripped != oat_stripped) {
   1041     timings.NewSplit("dex2oat OatFile copy");
   1042     oat_file.reset();
   1043      UniquePtr<File> in(OS::OpenFileForReading(oat_unstripped.c_str()));
   1044     UniquePtr<File> out(OS::CreateEmptyFile(oat_stripped.c_str()));
   1045     size_t buffer_size = 8192;
   1046     UniquePtr<uint8_t> buffer(new uint8_t[buffer_size]);
   1047     while (true) {
   1048       int bytes_read = TEMP_FAILURE_RETRY(read(in->Fd(), buffer.get(), buffer_size));
   1049       if (bytes_read <= 0) {
   1050         break;
   1051       }
   1052       bool write_ok = out->WriteFully(buffer.get(), bytes_read);
   1053       CHECK(write_ok);
   1054     }
   1055     oat_file.reset(out.release());
   1056     VLOG(compiler) << "Oat file copied successfully (stripped): " << oat_stripped;
   1057   }
   1058 
   1059 #if ART_USE_PORTABLE_COMPILER  // We currently only generate symbols on Portable
   1060   timings.NewSplit("dex2oat ElfStripper");
   1061   // Strip unneeded sections for target
   1062   off_t seek_actual = lseek(oat_file->Fd(), 0, SEEK_SET);
   1063   CHECK_EQ(0, seek_actual);
   1064   ElfStripper::Strip(oat_file.get());
   1065 
   1066 
   1067   // We wrote the oat file successfully, and want to keep it.
   1068   VLOG(compiler) << "Oat file written successfully (stripped): " << oat_location;
   1069 #endif  // ART_USE_PORTABLE_COMPILER
   1070 
   1071   timings.EndSplit();
   1072 
   1073   if (dump_timing || (dump_slow_timing && timings.GetTotalNs() > MsToNs(1000))) {
   1074     LOG(INFO) << Dumpable<base::TimingLogger>(timings);
   1075   }
   1076 
   1077   // Everything was successfully written, do an explicit exit here to avoid running Runtime
   1078   // destructors that take time (bug 10645725) unless we're a debug build or running on valgrind.
   1079   if (!kIsDebugBuild || (RUNNING_ON_VALGRIND == 0)) {
   1080     exit(EXIT_SUCCESS);
   1081   }
   1082 
   1083   return EXIT_SUCCESS;
   1084 }
   1085 
   1086 
   1087 }  // namespace art
   1088 
   1089 int main(int argc, char** argv) {
   1090   return art::dex2oat(argc, argv);
   1091 }
   1092