/*
 * Copyright (C) 2018 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_LIBARTBASE_BASE_HIDDENAPI_FLAGS_H_
#define ART_LIBARTBASE_BASE_HIDDENAPI_FLAGS_H_

#include <android-base/logging.h>

#include <vector>

#include "base/bit_utils.h"
#include "base/dumpable.h"
#include "base/hiddenapi_stubs.h"
#include "base/macros.h"
#include "sdk_version.h"

namespace art {
namespace hiddenapi {

// Helper methods used inside ApiList. These were moved outside of the ApiList
// class so that they can be used in static_asserts. If they were inside, they
// would be part of an unfinished type.
namespace helper {
  // Casts enum value to uint32_t.
  template<typename T>
  constexpr uint32_t ToUint(T val) { return static_cast<uint32_t>(val); }

  // Returns uint32_t with one bit set at an index given by an enum value.
  template<typename T>
  constexpr uint32_t ToBit(T val) { return 1u << ToUint(val); }

  // Returns a bit mask with `size` least significant bits set.
  constexpr uint32_t BitMask(uint32_t size) { return (1u << size) - 1; }

  // Returns a bit mask formed from an enum defining kMin and kMax. The values
  // are assumed to be indices of min/max bits and the resulting bitmask has
  // bits [kMin, kMax] set.
  template<typename T>
  constexpr uint32_t BitMask() {
    return BitMask(ToUint(T::kMax) + 1) & (~BitMask(ToUint(T::kMin)));
  }

  // Returns true if `val` is a bitwise subset of `mask`.
  constexpr bool MatchesBitMask(uint32_t val, uint32_t mask) { return (val & mask) == val; }

  // Returns true if the uint32_t value of `val` is a bitwise subset of `mask`.
  template<typename T>
  constexpr bool MatchesBitMask(T val, uint32_t mask) { return MatchesBitMask(ToUint(val), mask); }

  // Returns the number of values defined in an enum, assuming the enum defines
  // kMin and kMax and no integer values are skipped between them.
  template<typename T>
  constexpr uint32_t NumValues() { return ToUint(T::kMax) - ToUint(T::kMin) + 1; }

  // Returns enum value at position i from enum list.
  template <typename T>
  constexpr T GetEnumAt(uint32_t i) {
    return static_cast<T>(ToUint(T::kMin) + i);
  }

}  // namespace helper

/*
 * This class represents the information whether a field/method is in
 * public API (SDK) or if it isn't, apps targeting which SDK
 * versions are allowed to access it.
 */
class ApiList {
 private:
  // The representation in dex_flags_ is a combination of a Value in the lowest
  // kValueBitSize bits, and bit flags corresponding to DomainApi in bits above
  // that.
  uint32_t dex_flags_;

  // Number of bits reserved for Value in dex flags, and the corresponding bit mask.
  static constexpr uint32_t kValueBitSize = 4;
  static constexpr uint32_t kValueBitMask = helper::BitMask(kValueBitSize);

  enum class Value : uint32_t {
    // Values independent of target SDK version of app
    kSdk = 0,
    kUnsupported = 1,  // @UnsupportedAppUsage
    kBlocked = 2,

    // Values dependent on target SDK version of app. Put these last as
    // their list will be extended in future releases.
    // The max release code implicitly includes all maintenance releases,
    // e.g. MaxTargetO is accessible to targetSdkVersion <= 27 (O_MR1).
    kMaxTargetO = 3,
    kMaxTargetP = 4,
    kMaxTargetQ = 5,
    kMaxTargetR = 6,
    kMaxTargetS = 7,

    // Invalid value. Does not imply the DomainApi is invalid.
    kInvalid = (static_cast<uint32_t>(-1) & kValueBitMask),

    kMin = kSdk,
    kMax = kMaxTargetS,
    kFuture = kMax + 1,  // Only for testing
  };

  // Additional bit flags after the first kValueBitSize bits in dex flags. These
  // are used for domain-specific APIs. The app domain is the default when no
  // bits are set.
  enum class DomainApi : uint32_t {
    kCorePlatformApi = kValueBitSize,
    kTestApi = kValueBitSize + 1,

    // Special values
    kMin = kCorePlatformApi,
    kMax = kTestApi,
  };

  // Bit mask of all domain API flags.
  static constexpr uint32_t kDomainApiBitMask = helper::BitMask<DomainApi>();

  // Check that Values fit in the designated number of bits.
  static_assert(kValueBitSize >= MinimumBitsToStore(helper::ToUint(Value::kMax)),
                "Not enough bits to store all ApiList values");

  // Check that all Values are covered by kValueBitMask.
  static_assert(helper::MatchesBitMask(Value::kMin, kValueBitMask));
  static_assert(helper::MatchesBitMask(Value::kMax, kValueBitMask));
  static_assert(helper::MatchesBitMask(Value::kFuture, kValueBitMask));
  static_assert(helper::MatchesBitMask(Value::kInvalid, kValueBitMask));

  // Check that there's no offset between Values and the corresponding uint32
  // dex flags, so they can be converted between each other without any change.
  static_assert(helper::ToUint(Value::kMin) == 0);

  // Check that Value::kInvalid is larger than kFuture (which is larger than kMax).
  static_assert(helper::ToUint(Value::kFuture) < helper::ToUint(Value::kInvalid));

  // Check that no DomainApi bit flag is covered by kValueBitMask.
  static_assert((helper::ToBit(DomainApi::kMin) & kValueBitMask) == 0);
  static_assert((helper::ToBit(DomainApi::kMax) & kValueBitMask) == 0);

  // Names corresponding to Values.
  static constexpr const char* kValueNames[] = {
    "sdk",
    "unsupported",
    "blocked",
    "max-target-o",
    "max-target-p",
    "max-target-q",
    "max-target-r",
    "max-target-s",
  };

  // A magic marker used by tests to mimic a hiddenapi list which doesn't exist
  // yet.
  static constexpr const char* kFutureValueName = "max-target-future";

  // Names corresponding to DomainApis.
  static constexpr const char* kDomainApiNames[] {
    "core-platform-api",
    "test-api",
  };

  // Maximum SDK versions allowed to access ApiList of given Value.
  static constexpr SdkVersion kMaxSdkVersions[] {
    /* sdk */ SdkVersion::kMax,
    /* unsupported */ SdkVersion::kMax,
    /* blocklist */ SdkVersion::kMin,
    /* max-target-o */ SdkVersion::kO_MR1,
    /* max-target-p */ SdkVersion::kP,
    /* max-target-q */ SdkVersion::kQ,
    /* max-target-r */ SdkVersion::kR,
    /* max-target-s */ SdkVersion::kS,
  };

  explicit ApiList(uint32_t dex_flags) : dex_flags_(dex_flags) {
    DCHECK_EQ(dex_flags_, (dex_flags_ & kValueBitMask) | (dex_flags_ & kDomainApiBitMask));
  }

  static ApiList FromValue(Value val) {
    ApiList api_list(helper::ToUint(val));
    DCHECK(api_list.GetValue() == val);
    DCHECK_EQ(api_list.GetDomainApis(), 0u);
    return api_list;
  }

  // Returns an ApiList with only a DomainApi bit set - the Value is invalid. It
  // can be Combine'd with another ApiList with a Value to produce a valid combination.
  static ApiList FromDomainApi(DomainApi domain_api) {
    ApiList api_list(helper::ToUint(Value::kInvalid) | helper::ToBit(domain_api));
    DCHECK(api_list.GetValue() == Value::kInvalid);
    DCHECK_EQ(api_list.GetDomainApis(), helper::ToBit(domain_api));
    return api_list;
  }

  static ApiList FromValueAndDomainApis(Value val, uint32_t domain_apis) {
    ApiList api_list(helper::ToUint(val) | domain_apis);
    DCHECK(api_list.GetValue() == val);
    DCHECK_EQ(api_list.GetDomainApis(), domain_apis);
    return api_list;
  }

  Value GetValue() const {
    uint32_t value = (dex_flags_ & kValueBitMask);

    // Treat all ones as invalid value
    if (value == helper::ToUint(Value::kInvalid)) {
      return Value::kInvalid;
    } else if (value > helper::ToUint(Value::kMax)) {
      // For future unknown flag values, return unsupported.
      return Value::kUnsupported;
    } else {
      DCHECK_GE(value, helper::ToUint(Value::kMin));
      return static_cast<Value>(value);
    }
  }

  uint32_t GetDomainApis() const { return (dex_flags_ & kDomainApiBitMask); }

  // In order to correctly handle flagged changes from Unsupported to the Sdk, where both will be
  // set when the flag is enabled, consider Sdk to take precedence over any form of unsupported.
  // Note, this is not necessary in the inverse direction, because API flagging does not currently
  // support API removal. Moving from the blocklist to unsupported is also a case we don't have to
  // consider.
  // If this is true, the conflict resolves to Value::kSdk.
  static bool IsConflictingFlagsAcceptable(Value x, Value y) {
    const auto predicate_non_symmetric = [](auto l, auto r) {
      if (l != Value::kSdk) {
        return false;
      }
      switch (r) {
        case Value::kSdk:
        case Value::kUnsupported:
        case Value::kMaxTargetO:
        case Value::kMaxTargetP:
        case Value::kMaxTargetQ:
        case Value::kMaxTargetR:
        case Value::kMaxTargetS:
          return true;
        default:
          return false;
      }
    };
    return predicate_non_symmetric(x, y) || predicate_non_symmetric(y, x);
  }

  // Returns true if combining this ApiList with `other` will succeed.
  bool CanCombineWith(const ApiList& other) const {
    const Value val1 = GetValue();
    const Value val2 = other.GetValue();
    return (val1 == val2) || (val1 == Value::kInvalid) || (val2 == Value::kInvalid) ||
           IsConflictingFlagsAcceptable(val1, val2);
  }

 public:
  // Helpers for conveniently constructing ApiList instances.
  static ApiList Sdk() { return FromValue(Value::kSdk); }
  static ApiList Unsupported() { return FromValue(Value::kUnsupported); }
  static ApiList Blocked() { return FromValue(Value::kBlocked); }
  static ApiList MaxTargetO() { return FromValue(Value::kMaxTargetO); }
  static ApiList MaxTargetP() { return FromValue(Value::kMaxTargetP); }
  static ApiList MaxTargetQ() { return FromValue(Value::kMaxTargetQ); }
  static ApiList MaxTargetR() { return FromValue(Value::kMaxTargetR); }
  static ApiList MaxTargetS() { return FromValue(Value::kMaxTargetS); }
  static ApiList Invalid() { return FromValue(Value::kInvalid); }
  static ApiList CorePlatformApi() { return FromDomainApi(DomainApi::kCorePlatformApi); }
  static ApiList TestApi() { return FromDomainApi(DomainApi::kTestApi); }

  uint32_t GetDexFlags() const { return dex_flags_; }
  uint32_t GetIntValue() const { return helper::ToUint(GetValue()); }

  static ApiList FromDexFlags(uint32_t dex_flags) { return ApiList(dex_flags); }

  static ApiList FromIntValue(uint32_t int_val) {
    return FromValue(helper::GetEnumAt<Value>(int_val));
  }

  // Returns the ApiList with a flag of a given name, or an empty ApiList if not matched.
  static ApiList FromName(const std::string& str) {
    for (uint32_t i = 0; i < kValueCount; ++i) {
      if (str == kValueNames[i]) {
        return FromIntValue(i);
      }
    }
    for (uint32_t i = 0; i < kDomainApiCount; ++i) {
      if (str == kDomainApiNames[i]) {
        return FromDomainApi(helper::GetEnumAt<DomainApi>(i));
      }
    }
    if (str == kFutureValueName) {
      return FromValue(Value::kFuture);
    }
    return Invalid();
  }

  // Parses a vector of flag names into a single ApiList value. If successful,
  // returns true and assigns the new ApiList to `out_api_list`.
  static bool FromNames(std::vector<std::string>::iterator begin,
                        std::vector<std::string>::iterator end,
                        /* out */ ApiList* out_api_list) {
    ApiList api_list = Invalid();
    for (std::vector<std::string>::iterator it = begin; it != end; it++) {
      ApiList current = FromName(*it);
      if (current.IsEmpty() || !api_list.CanCombineWith(current)) {
        if (ApiStubs::IsStubsFlag(*it)) {
        // Ignore flags which correspond to the stubs from where the api
        // originates (i.e. system-api, test-api, public-api), as they are not
        // relevant at runtime
          continue;
        }
        return false;
      }
      api_list = Combine(api_list, current);
    }
    if (out_api_list != nullptr) {
      *out_api_list = api_list;
    }
    return true;
  }

  bool operator==(const ApiList& other) const { return dex_flags_ == other.dex_flags_; }
  bool operator!=(const ApiList& other) const { return !(*this == other); }

  // The order doesn't have any significance - only for ordering in containers.
  bool operator<(const ApiList& other) const { return dex_flags_ < other.dex_flags_; }

  // Combine two ApiList instances. The returned value has the union of the API
  // domains. Values are mutually exclusive, so they either have to be identical
  // or one of them can be safely ignored, which includes being kInvalid.
  static ApiList Combine(const ApiList& api1, const ApiList& api2) {
    // DomainApis are not mutually exclusive. Simply OR them.
    // TODO: This is suspect since the app domain doesn't have any bit and hence
    // implicitly disappears if OR'ed with any other domain.
    const uint32_t domain_apis = api1.GetDomainApis() | api2.GetDomainApis();

    const Value val1 = api1.GetValue();
    const Value val2 = api2.GetValue();
    if (val1 == val2) {
      return FromValueAndDomainApis(val1, domain_apis);
    } else if (val1 == Value::kInvalid) {
      return FromValueAndDomainApis(val2, domain_apis);
    } else if (val2 == Value::kInvalid) {
      return FromValueAndDomainApis(val1, domain_apis);
    } else if (IsConflictingFlagsAcceptable(val1, val2)) {
      return FromValueAndDomainApis(Value::kSdk, domain_apis);
    } else {
      LOG(FATAL) << "Invalid combination of values " << Dumpable(FromValue(val1)) << " and "
                 << Dumpable(FromValue(val2));
      UNREACHABLE();
    }
  }

  // Returns true if all flags set in `other` are also set in `this`.
  bool Contains(const ApiList& other) const {
    return ((other.GetValue() == Value::kInvalid) || (GetValue() == other.GetValue())) &&
           helper::MatchesBitMask(other.GetDomainApis(), GetDomainApis());
  }

  // Returns true whether the configuration is valid for runtime use.
  bool IsValid() const { return GetValue() != Value::kInvalid; }

  // Returns true when no ApiList is specified and no domain_api flags either.
  bool IsEmpty() const { return (GetValue() == Value::kInvalid) && (GetDomainApis() == 0); }

  // Returns true if the ApiList is on blocklist.
  bool IsBlocked() const { return GetValue() == Value::kBlocked; }

  bool IsSdkApi() const { return GetValue() == Value::kSdk; }

  // Returns true if the ApiList is a test API.
  bool IsTestApi() const {
    return helper::MatchesBitMask(helper::ToBit(DomainApi::kTestApi), dex_flags_);
  }

  // Returns the maximum target SDK version allowed to access this ApiList.
  SdkVersion GetMaxAllowedSdkVersion() const { return kMaxSdkVersions[GetIntValue()]; }

  void Dump(std::ostream& os) const {
    bool is_first = true;

    if (IsEmpty()) {
      os << "invalid";
      return;
    }

    if (GetValue() != Value::kInvalid) {
      os << kValueNames[GetIntValue()];
      is_first = false;
    }

    const uint32_t domain_apis = GetDomainApis();
    for (uint32_t i = 0; i < kDomainApiCount; i++) {
      if (helper::MatchesBitMask(helper::ToBit(helper::GetEnumAt<DomainApi>(i)), domain_apis)) {
        if (is_first) {
          is_first = false;
        } else {
          os << ",";
        }
        os << kDomainApiNames[i];
      }
    }

    DCHECK_EQ(IsEmpty(), is_first);
  }

  // Number of valid enum values in Value.
  static constexpr uint32_t kValueCount = helper::NumValues<Value>();
  // Number of valid enum values in DomainApi.
  static constexpr uint32_t kDomainApiCount = helper::NumValues<DomainApi>();
  // Total number of possible enum values, including invalid, in Value.
  static constexpr uint32_t kValueSize = (1u << kValueBitSize) + 1;

  // Check min and max values are calculated correctly.
  static_assert(Value::kMin == helper::GetEnumAt<Value>(0));
  static_assert(Value::kMax == helper::GetEnumAt<Value>(kValueCount - 1));

  static_assert(DomainApi::kMin == helper::GetEnumAt<DomainApi>(0));
  static_assert(DomainApi::kMax == helper::GetEnumAt<DomainApi>(kDomainApiCount - 1));
};

inline std::ostream& operator<<(std::ostream& os, ApiList value) {
  value.Dump(os);
  return os;
}

}  // namespace hiddenapi
}  // namespace art


#endif  // ART_LIBARTBASE_BASE_HIDDENAPI_FLAGS_H_