/*
 * Copyright (C) 2014 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef ART_RUNTIME_OAT_OAT_FILE_ASSISTANT_H_
#define ART_RUNTIME_OAT_OAT_FILE_ASSISTANT_H_

#include <cstdint>
#include <memory>
#include <optional>
#include <sstream>
#include <string>
#include <string_view>
#include <utility>
#include <variant>
#include <vector>

#include "arch/instruction_set.h"
#include "base/compiler_filter.h"
#include "base/macros.h"
#include "base/os.h"
#include "base/scoped_flock.h"
#include "base/unix_file/fd_file.h"
#include "class_loader_context.h"
#include "oat_file.h"
#include "oat_file_assistant_context.h"

namespace art HIDDEN {

namespace gc {
namespace space {
class ImageSpace;
}  // namespace space
}  // namespace gc

// Class for assisting with oat file management.
//
// This class collects common utilities for determining the status of an oat
// file on the device, updating the oat file, and loading the oat file.
//
// The oat file assistant is intended to be used with dex locations not on the
// boot class path. See the IsInBootClassPath method for a way to check if the
// dex location is in the boot class path.
class OatFileAssistant {
 public:
  enum DexOptNeeded {
    // No dexopt should (or can) be done to update the apk/jar.
    // Matches Java: dalvik.system.DexFile.NO_DEXOPT_NEEDED = 0
    kNoDexOptNeeded = 0,

    // dex2oat should be run to update the apk/jar from scratch.
    // Matches Java: dalvik.system.DexFile.DEX2OAT_FROM_SCRATCH = 1
    kDex2OatFromScratch = 1,

    // dex2oat should be run to update the apk/jar because the existing code
    // is out of date with respect to the boot image.
    // Matches Java: dalvik.system.DexFile.DEX2OAT_FOR_BOOT_IMAGE
    kDex2OatForBootImage = 2,

    // dex2oat should be run to update the apk/jar because the existing code
    // is out of date with respect to the target compiler filter.
    // Matches Java: dalvik.system.DexFile.DEX2OAT_FOR_FILTER
    kDex2OatForFilter = 3,
  };

  enum OatStatus {
    // kOatCannotOpen - The oat file cannot be opened, because it does not
    // exist, is unreadable, or otherwise corrupted.
    kOatCannotOpen,

    // kOatDexOutOfDate - The oat file is out of date with respect to the dex file.
    kOatDexOutOfDate,

    // kOatBootImageOutOfDate - The oat file is up to date with respect to the
    // dex file, but is out of date with respect to the boot image.
    kOatBootImageOutOfDate,

    // kOatContextOutOfDate - The context in the oat file is out of date with
    // respect to the class loader context.
    kOatContextOutOfDate,

    // kOatAssumedValuesOutOfDate - The assumed compiler values in the oat file
    // are of date with respect to the current runtime.
    kOatAssumedValuesOutOfDate,

    // kOatUpToDate - The oat file is completely up to date with respect to
    // the dex file and boot image.
    kOatUpToDate,
  };

  // A bit field to represent the conditions where dexopt should be performed.
  struct DexOptTrigger {
    // Dexopt should be performed if the target compiler filter is better than the current compiler
    // filter. See `CompilerFilter::IsBetter`.
    bool targetFilterIsBetter : 1;
    // Dexopt should be performed if the target compiler filter is the same as the current compiler
    // filter.
    bool targetFilterIsSame : 1;
    // Dexopt should be performed if the target compiler filter is worse than the current compiler
    // filter. See `CompilerFilter::IsBetter`.
    bool targetFilterIsWorse : 1;
    // Dexopt should be performed if the current oat file was compiled without a primary image,
    // and the runtime is now running with a primary image loaded from disk.
    bool primaryBootImageBecomesUsable : 1;
    // Dexopt should be performed if the APK is compressed and the current oat/vdex file doesn't
    // contain dex code.
    bool needExtraction : 1;
  };

  // Represents the location of the current oat file and/or vdex file.
  enum Location {
    // Does not exist, or an error occurs.
    kLocationNoneOrError = 0,
    // In the global "dalvik-cache" folder.
    kLocationOat = 1,
    // In the "oat" folder next to the dex file.
    kLocationOdex = 2,
    // In the dm file. This means the only usable file is the vdex file.
    kLocationDm = 3,
    // The oat and art files are in the sdm file next to the dex file. The vdex file is in the dm
    // file next to the dex file. The sdc file is in the global "dalvik-cache" folder.
    kLocationSdmOat = 4,
    // The oat and art files are in the sdm file next to the dex file. The vdex file is in the dm
    // file next to the dex file. The sdc file is next to the dex file.
    kLocationSdmOdex = 5,
  };

  // Represents the status of the current oat file and/or vdex file.
  class DexOptStatus {
   public:
    Location GetLocation() { return location_; }
    bool IsVdexUsable() { return location_ != kLocationNoneOrError; }

   private:
    Location location_ = kLocationNoneOrError;
    friend class OatFileAssistant;
  };

  // Constructs an OatFileAssistant object to assist the oat file
  // corresponding to the given dex location with the target instruction set.
  //
  // The dex_location must not be null and should remain available and
  // unchanged for the duration of the lifetime of the OatFileAssistant object.
  // Typically the dex_location is the absolute path to the original,
  // un-optimized dex file.
  //
  // Note: Currently the dex_location must have an extension.
  // TODO: Relax this restriction?
  //
  // The isa should be either the 32 bit or 64 bit variant for the current
  // device. For example, on an arm device, use arm or arm64. An oat file can
  // be loaded executable only if the ISA matches the current runtime.
  //
  // context should be the class loader context to check against, or null to skip the check.
  //
  // load_executable should be true if the caller intends to try and load
  // executable code for this dex location.
  //
  // only_load_trusted_executable should be true if the caller intends to have
  // only oat files from trusted locations loaded executable. See IsTrustedLocation() for
  // details on trusted locations.
  //
  // runtime_options should be provided with all the required fields filled if the caller intends to
  // use OatFileAssistant without a runtime.
  EXPORT OatFileAssistant(const char* dex_location,
                          const InstructionSet isa,
                          ClassLoaderContext* context,
                          bool load_executable,
                          bool only_load_trusted_executable = false,
                          OatFileAssistantContext* ofa_context = nullptr);

  // Similar to this(const char*, const InstructionSet, bool), however, if a valid zip_fd is
  // provided, vdex, oat, and zip files will be read from vdex_fd, oat_fd and zip_fd respectively.
  // Otherwise, dex_location will be used to construct necessary filenames.
  EXPORT OatFileAssistant(const char* dex_location,
                          const InstructionSet isa,
                          ClassLoaderContext* context,
                          bool load_executable,
                          bool only_load_trusted_executable,
                          OatFileAssistantContext* ofa_context,
                          int vdex_fd,
                          int oat_fd,
                          int zip_fd);

  EXPORT ~OatFileAssistant();

  // A convenient factory function that accepts ISA, class loader context, and compiler filter in
  // strings. Returns the created instance and ClassLoaderContext on success, or returns nullptr and
  // outputs an error message if it fails to parse the input strings.
  // The returned ClassLoaderContext must live at least as long as the OatFileAssistant.
  EXPORT static std::unique_ptr<OatFileAssistant> Create(
      const std::string& filename,
      const std::string& isa_str,
      const std::optional<std::string>& context_str,
      bool load_executable,
      bool only_load_trusted_executable,
      OatFileAssistantContext* ofa_context,
      /*out*/ std::unique_ptr<ClassLoaderContext>* context,
      /*out*/ std::string* error_msg);

  // Returns true if the dex location refers to an element of the boot class
  // path.
  EXPORT bool IsInBootClassPath();

  // Return what action needs to be taken to produce up-to-date code for this
  // dex location. If "downgrade" is set to false, it verifies if the current
  // compiler filter is at least as good as an oat file generated with the
  // given compiler filter otherwise, if its set to true, it checks whether
  // the oat file generated with the target filter will be downgraded as
  // compared to the current state. For example, if the current compiler filter is verify and the
  // target filter is speed profile it will recommend to keep it in its current state.
  // profile_changed should be true to indicate the profile has recently changed
  // for this dex location.
  // If the purpose of the dexopt is to downgrade the compiler filter,
  // set downgrade to true.
  // Returns a positive status code if the status refers to the oat file in
  // the oat location. Returns a negative status code if the status refers to
  // the oat file in the odex location.
  //
  // Deprecated. Use the other overload.
  EXPORT int GetDexOptNeeded(CompilerFilter::Filter target_compiler_filter,
                             bool profile_changed = false,
                             bool downgrade = false);

  // Returns true if dexopt needs to be performed with respect to the given target compilation
  // filter and dexopt trigger. Also returns the status of the current oat file and/or vdex file.
  EXPORT bool GetDexOptNeeded(CompilerFilter::Filter target_compiler_filter,
                              const DexOptTrigger dexopt_trigger,
                              /*out*/ DexOptStatus* dexopt_status);

  // Returns true if there is up-to-date code for this dex location,
  // irrespective of the compiler filter of the up-to-date code.
  bool IsUpToDate();

  // Returns an oat file that can be used for loading dex files.
  // Returns null if no suitable oat file was found.
  //
  // After this call, no other methods of the OatFileAssistant should be
  // called, because access to the loaded oat file has been taken away from
  // the OatFileAssistant object.
  std::unique_ptr<OatFile> GetBestOatFile();

  // Computes the optimization status of the given dex file. The result is returned via output
  // parameters.
  //   - odex_location: the location of the best oat file that will be used for loading. See
  //        GetBestInfo().
  //   - compilation_filter: the level of optimizations (compiler filter)
  //   - compilation_reason: the optimization reason. The reason might be "unknown" if the compiler
  //        artifacts were not annotated during optimizations.
  //   - odex_status: a human readable refined status of the validity of the odex file. Possible
  //        values are: "up-to-date", "apk-more-recent", and "io-error-no-oat".
  //   - location: an enum indicating the location of the best oat file.
  //   - is_backed_by_vdex_only: true if the best oat file is backed by vdex file only.
  //
  // This method will try to mimic the runtime effect of loading the dex file. For example, if there
  // is no usable oat file, the compiler filter will be set to "run-from-apk".
  EXPORT void GetOptimizationStatus(/*out*/ std::string* odex_location,
                                    /*out*/ std::string* compilation_filter,
                                    /*out*/ std::string* compilation_reason,
                                    /*out*/ std::string* odex_status,
                                    /*out*/ Location* location,
                                    /*out*/ bool* is_backed_by_vdex_only);

  static void GetOptimizationStatus(const std::string& filename,
                                    InstructionSet isa,
                                    std::string* out_compilation_filter,
                                    std::string* out_compilation_reason,
                                    OatFileAssistantContext* ofa_context = nullptr);

  // Open and returns an image space associated with the oat file.
  static std::unique_ptr<gc::space::ImageSpace> OpenImageSpace(const OatFile* oat_file);

  // Loads the dex files in the given oat file for the given dex location.
  // The oat file should be up to date for the given dex location.
  // This loads multiple dex files in the case of multidex.
  // Returns an empty vector if no dex files for that location could be loaded
  // from the oat file.
  //
  // The caller is responsible for freeing the dex_files returned, if any. The
  // dex_files will only remain valid as long as the oat_file is valid.
  static std::vector<std::unique_ptr<const DexFile>> LoadDexFiles(
      const OatFile& oat_file, const char* dex_location);

  // Same as `std::vector<std::unique_ptr<const DexFile>> LoadDexFiles(...)` with the difference:
  //   - puts the dex files in the given vector
  //   - returns whether or not all dex files were successfully opened
  static bool LoadDexFiles(const OatFile& oat_file,
                           const std::string& dex_location,
                           std::vector<std::unique_ptr<const DexFile>>* out_dex_files);

  // Returns whether this is an apk/zip wit a classes.dex entry, or nullopt if an error occurred.
  EXPORT std::optional<bool> HasDexFiles(std::string* error_msg);

  // If the dex file has been installed with a compiled oat file alongside
  // it, the compiled oat file will have the extension .odex, and is referred
  // to as the odex file. It is called odex for legacy reasons; the file is
  // really an oat file. The odex file will often, but not always, have a
  // patch delta of 0 and need to be relocated before use for the purposes of
  // ASLR. The odex file is treated as if it were read-only.
  //
  // Returns the status of the odex file for the dex location.
  //
  // For testing purposes only.
  OatStatus OdexFileStatus();

  // When the dex files is compiled on the target device, the oat file is the
  // result. The oat file will have been relocated to some
  // (possibly-out-of-date) offset for ASLR.
  //
  // Returns the status of the oat file for the dex location.
  //
  // For testing purposes only.
  OatStatus OatFileStatus();

  OatStatus GetBestStatus() {
    return GetBestInfo().Status();
  }

  // Constructs the odex file name for the given dex location.
  // Returns true on success, in which case odex_filename is set to the odex
  // file name.
  // Returns false on error, in which case error_msg describes the error and
  // odex_filename is not changed.
  // Neither odex_filename nor error_msg may be null.
  EXPORT static bool DexLocationToOdexFilename(const std::string& location,
                                               InstructionSet isa,
                                               std::string* odex_filename,
                                               std::string* error_msg);

  // Constructs the oat file name for the given dex location.
  // Returns true on success, in which case oat_filename is set to the oat
  // file name.
  // Returns false on error, in which case error_msg describes the error and
  // oat_filename is not changed.
  // Neither oat_filename nor error_msg may be null.
  //
  // Calling this function requires an active runtime.
  static bool DexLocationToOatFilename(const std::string& location,
                                       InstructionSet isa,
                                       std::string* oat_filename,
                                       std::string* error_msg);

  // Same as above, but also takes `deny_art_apex_data_files` from input.
  //
  // Calling this function does not require an active runtime.
  EXPORT static bool DexLocationToOatFilename(const std::string& location,
                                              InstructionSet isa,
                                              bool deny_art_apex_data_files,
                                              std::string* oat_filename,
                                              std::string* error_msg);

  // Computes the dex location and vdex filename. If the data directory of the process
  // is known, creates an absolute path in that directory and tries to infer path
  // of a corresponding vdex file. Otherwise only creates a basename dex_location
  // from the combined checksums. Returns true if all out-arguments have been set.
  //
  // Calling this function requires an active runtime.
  static bool AnonymousDexVdexLocation(const std::vector<const DexFile::Header*>& dex_headers,
                                       InstructionSet isa,
                                       /* out */ std::string* dex_location,
                                       /* out */ std::string* vdex_filename);

  // Returns true if a filename (given as basename) is a name of a vdex for
  // anonymous dex file(s) created by AnonymousDexVdexLocation.
  EXPORT static bool IsAnonymousVdexBasename(const std::string& basename);

  bool ClassLoaderContextIsOkay(const OatFile& oat_file, /*out*/ std::string* error_msg) const;

  // Validates the boot class path checksum of an OatFile.
  EXPORT bool ValidateBootClassPathChecksums(const OatFile& oat_file,
                                             /*out*/ std::string* error_msg);

  // Validates the given bootclasspath and bootclasspath checksums found in an oat header.
  static bool ValidateBootClassPathChecksums(OatFileAssistantContext* ofa_context,
                                             InstructionSet isa,
                                             std::string_view oat_checksums,
                                             std::string_view oat_boot_class_path,
                                             /*out*/ std::string* error_msg);

 private:
  enum class OatFileType {
    kNone,
    kOat,
    kSdm,
    kVdex,
    kDm,
  };

  class OatFileInfo {
   public:
    // Empty info. Treated as kOatCannotOpen.
    // Use constructors in subclasses to construct a real instance.
    explicit OatFileInfo(OatFileAssistant* oat_file_assistant)
        : oat_file_assistant_(oat_file_assistant), filename_(""), is_oat_location_(false) {}

    virtual ~OatFileInfo() = default;

    // ART code is compiled with `-fno-rtti`, so we need a virtual function to return type
    // information.
    virtual OatFileType GetType() { return OatFileType::kNone; }

    // Returns a string indicating the location of the oat file, for debugging purposes only.
    virtual const char* GetLocationDebugString() { return "none"; }

    bool IsOatLocation() const;

    const std::string* Filename() const;

    const char* DisplayFilename() const;

    // Returns true if this oat file can be used for running code. The oat
    // file can be used for running code as long as it is not out of date with
    // respect to the dex code or boot image. An oat file that is out of date
    // with respect to relocation is considered useable, because it's possible
    // to interpret the dex code rather than run the unrelocated compiled
    // code.
    bool IsUseable();

    // Returns the status of this oat file.
    // Optionally, returns `error_msg` showing why the status is not `kOatUpToDate`.
    OatStatus Status(/*out*/ std::string* error_msg = nullptr);

    // Return the DexOptNeeded value for this oat file with respect to the given target compilation
    // filter and dexopt trigger.
    DexOptNeeded GetDexOptNeeded(CompilerFilter::Filter target_compiler_filter,
                                 const DexOptTrigger dexopt_trigger);

    // Returns true if the file exists.
    virtual bool FileExists() const;

    // Returns the loaded file.
    // Loads the file if needed. Returns null if the file failed to load.
    // The caller shouldn't clean up or free the returned pointer.
    // Optionally, returns `error_msg` showing why the file failed to load.
    const OatFile* GetFile(/*out*/ std::string* error_msg = nullptr);

    // Returns true if the file is opened executable.
    bool IsExecutable();

    // Release the loaded oat file for runtime use.
    // Returns null if the oat file hasn't been loaded or is out of date.
    // Ensures the returned file is not loaded executable if it has unuseable
    // compiled code.
    //
    // After this call, no other methods of the OatFileInfo should be
    // called, because access to the loaded oat file has been taken away from
    // the OatFileInfo object.
    std::unique_ptr<OatFile> ReleaseFileForUse();

   protected:
    // Constructs a real instance.
    // Pass true for is_oat_location if the information associated with this
    // OatFileInfo is for the oat location, as opposed to the odex location.
    OatFileInfo(OatFileAssistant* oat_file_assistant,
                const std::string& filename,
                bool is_oat_location)
        : oat_file_assistant_(oat_file_assistant),
          filename_(filename),
          is_oat_location_(is_oat_location) {}

    // Loads the file.
    virtual std::unique_ptr<OatFile> LoadFile(std::string* error_msg) const {
      *error_msg = "Not implemented";
      return nullptr;
    }

    OatFileAssistant* const oat_file_assistant_;
    const std::string filename_;

   private:
    // Returns true if the oat file is usable but at least one dexopt trigger is matched. This
    // function should only be called if the oat file is usable.
    bool ShouldRecompileForFilter(CompilerFilter::Filter target,
                                  const DexOptTrigger dexopt_trigger);

    // Release the loaded oat file.
    // Returns null if the oat file hasn't been loaded.
    //
    // After this call, no other methods of the OatFileInfo should be
    // called, because access to the loaded oat file has been taken away from
    // the OatFileInfo object.
    std::unique_ptr<OatFile> ReleaseFile();

    const bool is_oat_location_;

    // A pair of the loaded file and the error message, if `GetFile` has been attempted.
    // `std::nullopt` if `GetFile` has not been attempted.
    std::optional<std::pair<std::unique_ptr<OatFile>, std::string>> file_ = std::nullopt;

    // A pair of the oat status and the error message, if `Status` has been attempted.
    // `std::nullopt` if `Status` has not been attempted.
    std::optional<std::pair<OatStatus, std::string>> status_ = std::nullopt;

    // For debugging only.
    // If this flag is set, the file has been released to the user and the
    // OatFileInfo object is in a bad state and should no longer be used.
    bool file_released_ = false;
  };

  class OatFileInfoBackedByOat : public OatFileInfo {
   public:
    OatFileInfoBackedByOat(OatFileAssistant* oat_file_assistant,
                           const std::string& filename,
                           bool is_oat_location,
                           bool use_fd,
                           int zip_fd = -1,
                           int vdex_fd = -1,
                           int oat_fd = -1)
        : OatFileInfo(oat_file_assistant, filename, is_oat_location),
          use_fd_(use_fd),
          zip_fd_(zip_fd),
          vdex_fd_(vdex_fd),
          oat_fd_(oat_fd) {}

    OatFileType GetType() override { return OatFileType::kOat; }

    const char* GetLocationDebugString() override {
      return IsOatLocation() ? "odex in dalvik-cache" : "odex next to the dex file";
    }

    bool FileExists() const override;

   protected:
    std::unique_ptr<OatFile> LoadFile(std::string* error_msg) const override;

   private:
    const bool use_fd_;
    const int zip_fd_;
    const int vdex_fd_;
    const int oat_fd_;
  };

  class OatFileInfoBackedBySdm : public OatFileInfo {
   public:
    OatFileInfoBackedBySdm(OatFileAssistant* oat_file_assistant,
                           const std::string& sdm_filename,
                           bool is_oat_location,
                           const std::string& dm_filename,
                           const std::string& sdc_filename)
        : OatFileInfo(oat_file_assistant, sdm_filename, is_oat_location),
          dm_filename_(dm_filename),
          sdc_filename_(sdc_filename) {}

    OatFileType GetType() override { return OatFileType::kSdm; }

    const char* GetLocationDebugString() override {
      return IsOatLocation() ? "sdm with sdc in dalvik-cache" : "sdm with sdc next to the dex file";
    }

    bool FileExists() const override;

   protected:
    std::unique_ptr<OatFile> LoadFile(std::string* error_msg) const override;

   private:
    const std::string dm_filename_;
    const std::string sdc_filename_;
  };

  class OatFileInfoBackedByVdex : public OatFileInfo {
   public:
    OatFileInfoBackedByVdex(OatFileAssistant* oat_file_assistant,
                            const std::string& filename,
                            bool is_oat_location,
                            bool use_fd,
                            int zip_fd = -1,
                            int vdex_fd = -1)
        : OatFileInfo(oat_file_assistant, filename, is_oat_location),
          use_fd_(use_fd),
          zip_fd_(zip_fd),
          vdex_fd_(vdex_fd) {}

    OatFileType GetType() override { return OatFileType::kVdex; }

    const char* GetLocationDebugString() override {
      return IsOatLocation() ? "vdex in dalvik-cache" : "vdex next to the dex file";
    }

    bool FileExists() const override;

   protected:
    std::unique_ptr<OatFile> LoadFile(std::string* error_msg) const override;

   private:
    const bool use_fd_;
    const int zip_fd_;
    const int vdex_fd_;
  };

  class OatFileInfoBackedByDm : public OatFileInfo {
   public:
    OatFileInfoBackedByDm(OatFileAssistant* oat_file_assistant, const std::string& filename)
        : OatFileInfo(oat_file_assistant, filename, /*is_oat_location=*/false) {}

    OatFileType GetType() override { return OatFileType::kDm; }

    const char* GetLocationDebugString() override { return "dm"; }

   protected:
    std::unique_ptr<OatFile> LoadFile(std::string* error_msg) const override;
  };

  // Return info for the best oat file.
  OatFileInfo& GetBestInfo();

  // Returns true when vdex/oat/odex files should be read from file descriptors.
  // The method checks the value of zip_fd_, and if the value is valid, returns
  // true. This is required to have a deterministic behavior around how different
  // files are being read.
  bool UseFdToReadFiles();

  // Returns true if the dex checksums in the given oat file are up to date
  // with respect to the dex location. If the dex checksums are not up to
  // date, error_msg is updated with a message describing the problem.
  bool DexChecksumUpToDate(const OatFile& file, std::string* error_msg);

  // Return the status for a given opened oat file with respect to the dex
  // location.
  OatStatus GivenOatFileStatus(const OatFile& file, /*out*/ std::string* error_msg);

  // Gets the dex checksum required for an up-to-date oat file.
  // Returns cached result from GetMultiDexChecksum.
  bool GetRequiredDexChecksum(std::optional<uint32_t>* checksum, std::string* error);

  // Returns whether there is at least one boot image usable.
  bool IsPrimaryBootImageUsable();

  // Returns the trigger for the deprecated overload of `GetDexOptNeeded`.
  //
  // Deprecated. Do not use in new code.
  DexOptTrigger GetDexOptTrigger(CompilerFilter::Filter target_compiler_filter,
                                 bool profile_changed,
                                 bool downgrade);

  // Returns the pointer to the owned or unowned instance of OatFileAssistantContext.
  OatFileAssistantContext* GetOatFileAssistantContext() {
    if (std::holds_alternative<OatFileAssistantContext*>(ofa_context_)) {
      return std::get<OatFileAssistantContext*>(ofa_context_);
    } else {
      return std::get<std::unique_ptr<OatFileAssistantContext>>(ofa_context_).get();
    }
  }

  // The runtime options taken from the active runtime or the input.
  //
  // All member functions should get runtime options from this variable rather than referencing the
  // active runtime. This is to allow OatFileAssistant to function without an active runtime.
  const OatFileAssistantContext::RuntimeOptions& GetRuntimeOptions() {
    return GetOatFileAssistantContext()->GetRuntimeOptions();
  }

  // Returns whether the zip file only contains uncompressed dex.
  bool ZipFileOnlyContainsUncompressedDex();

  // Returns the location of the given oat file.
  Location GetLocation(OatFileInfo& info);

  std::string dex_location_;

  // The class loader context to check against, or null representing that the check should be
  // skipped.
  ClassLoaderContext* context_;

  // In a properly constructed OatFileAssistant object, isa_ should be either
  // the 32 or 64 bit variant for the current device.
  const InstructionSet isa_ = InstructionSet::kNone;

  // Whether we will attempt to load oat files executable.
  bool load_executable_ = false;

  // Whether only oat files from trusted locations are loaded executable.
  const bool only_load_trusted_executable_ = false;

  // Cached value of whether the potential zip file only contains uncompressed dex.
  // This should be accessed only by the ZipFileOnlyContainsUncompressedDex() method.
  bool zip_file_only_contains_uncompressed_dex_ = true;

  // Cached value of the required dex checksums.
  // This should be accessed only by the GetRequiredDexChecksums() method.
  std::optional<uint32_t> cached_required_dex_checksums_;
  std::optional<std::string> cached_required_dex_checksums_error_;
  bool required_dex_checksums_attempted_ = false;

  // Empty oat file info, used as a placeholder.
  OatFileInfo empty_info_ = OatFileInfo(this);

  // Oat file info candidates, ordered by precedence.
  std::vector<std::unique_ptr<OatFileInfo>> info_list_;

  // File descriptor corresponding to apk, dex file, or zip.
  int zip_fd_;

  // Owned or unowned instance of OatFileAssistantContext.
  std::variant<std::unique_ptr<OatFileAssistantContext>, OatFileAssistantContext*> ofa_context_;

  friend class OatFileAssistantTest;

  DISALLOW_COPY_AND_ASSIGN(OatFileAssistant);
};

std::ostream& operator << (std::ostream& stream, const OatFileAssistant::OatStatus status);

}  // namespace art

#endif  // ART_RUNTIME_OAT_OAT_FILE_ASSISTANT_H_