// Copyright 2018 Google LLC. All Rights Reserved. This file and proprietary
// source code may only be used and distributed under the Widevine License
// Agreement.

#include "properties.h"
#include "properties_configuration.h"

#include <unistd.h>
#include <sstream>
#include <string>

#include <android-base/properties.h>
#include <android/sysprop/WidevineProperties.sysprop.h>

#include <com_android_mediadrm_widevine_flags.h>

#include "log.h"
#include "wv_android_constants.h"
#include "widevine_apex_info.h"

using namespace android::sysprop;

namespace mediadrm_flags = com::android::mediadrm::widevine::flags;
namespace {

const char kBasePathPrefix[] = "/data/vendor/mediadrm/IDM";
const char kL1Dir[] = "/L1/";
const char kL2Dir[] = "/L2/";
const char kL3Dir[] = "/L3/";
const char kFactoryKeyboxPath[] = "/factory/wv.keys";
const char kChromeOsHardware[] = "bertha";

bool GetAndroidProperty(const char* key, std::string* value) {
  if (!key) {
    LOGW("GetAndroidProperty: Invalid property key parameter");
    return false;
  }

  if (!value) {
    LOGW("GetAndroidProperty: Invalid property value parameter");
    return false;
  }

  *value = android::base::GetProperty(key, "");

  return value->size() > 0;
}

}  // namespace

namespace wvcdm {

void Properties::InitOnce() {
  oem_crypto_use_secure_buffers_ = kPropertyOemCryptoUseSecureBuffers;
  oem_crypto_use_fifo_ = kPropertyOemCryptoUseFifo;
  oem_crypto_use_userspace_buffers_ = kPropertyOemCryptoUseUserSpaceBuffers;
  provisioning_messages_are_binary_ = kPropertyProvisioningMessagesAreBinary;
  allow_service_certificate_requests_ = kAllowServiceCertificateRequests;
  device_files_is_a_real_filesystem_ = kDeviceFilesIsARealFileSystem;
  allow_restore_of_offline_licenses_with_release_ =
      kAllowRestoreOfflineLicenseWithRelease;
  delay_oem_crypto_termination_ = kPropertyDelayOemCryptoTermination;
  {
    std::unique_lock<std::mutex> lock(session_mutex_);
    session_property_set_.reset(new CdmClientPropertySetMap());
  }
}

bool Properties::GetCompanyName(std::string* company_name) {
  if (!company_name) {
    LOGW("Properties::GetCompanyName: Invalid parameter");
    return false;
  }
  std::string hardware;
  if (GetAndroidProperty("ro.hardware", &hardware) &&
      hardware == kChromeOsHardware &&
      GetAndroidProperty("ro.vendor.cdm.manufacturer", company_name)) {
    return true;
  }
  return GetAndroidProperty("ro.product.manufacturer", company_name);
}

bool Properties::GetModelName(std::string* model_name) {
  if (!model_name) {
    LOGW("Properties::GetModelName: Invalid parameter");
    return false;
  }
  std::string hardware;
  if (GetAndroidProperty("ro.hardware", &hardware) &&
      hardware == kChromeOsHardware &&
      GetAndroidProperty("ro.vendor.cdm.model", model_name)) {
    return true;
  }
  return GetAndroidProperty("ro.product.model", model_name);
}

bool Properties::GetModelYear(std::string* /*model_year*/) {
  return false;  // Not implemented on Android
}

bool Properties::GetArchitectureName(std::string* arch_name) {
  if (!arch_name) {
    LOGW("Properties::GetArchitectureName: Invalid parameter");
    return false;
  }
  return GetAndroidProperty("ro.product.cpu.abi", arch_name);
}

bool Properties::GetDeviceName(std::string* device_name) {
  if (!device_name) {
    LOGW("Properties::GetDeviceName: Invalid parameter");
    return false;
  }
  std::string hardware;
  if (GetAndroidProperty("ro.hardware", &hardware) &&
      hardware == kChromeOsHardware &&
      GetAndroidProperty("ro.vendor.cdm.device", device_name)) {
    return true;
  }
  return GetAndroidProperty("ro.product.device", device_name);
}

bool Properties::GetProductName(std::string* product_name) {
  if (!product_name) {
    LOGW("Properties::GetProductName: Invalid parameter");
    return false;
  }
  return GetAndroidProperty("ro.product.name", product_name);
}

bool Properties::GetBuildInfo(std::string* build_info) {
  if (!build_info) {
    LOGW("Properties::GetBuildInfo: Invalid parameter");
    return false;
  }
  return GetAndroidProperty("ro.build.fingerprint", build_info);
}

bool Properties::GetWVCdmVersion(std::string* version) {
  if (!version) {
    LOGW("Properties::GetWVCdmVersion: Invalid parameter");
    return false;
  }

  std::string apex_version = "";
#ifdef __ANDROID_APEX__
{
    auto info = widevine::apex::GetApexInfo();
    apex_version = "@" + std::to_string(info->version);
}
#endif

  if (mediadrm_flags::update_cdm_version_to_18_0_2()) {
    *version = kWVAndroidCdmVersion_18_0_2 + apex_version;
  } else {
    *version = kWVAndroidCdmVersion_18_0_1 + apex_version;
  }
  return true;
}

bool Properties::GetDeviceFilesBasePath(CdmSecurityLevel security_level,
                                        std::string* base_path) {
  if (!base_path) {
    LOGW("Properties::GetDeviceFilesBasePath: Invalid parameter");
    return false;
  }
  std::stringstream ss;
  ss << kBasePathPrefix << getuid();
  switch (security_level) {
    case kSecurityLevelL1:
      ss << kL1Dir;
      break;
    case kSecurityLevelL2:
      ss << kL2Dir;
      break;
    case kSecurityLevelL3:
      ss << kL3Dir;
      break;
    default:
      LOGW("Properties::GetDeviceFilesBasePath: Unknown security level: %d",
           security_level);
      return false;
  }

  *base_path = ss.str();
  return true;
}

bool Properties::GetFactoryKeyboxPath(std::string* keybox) {
  if (!keybox) {
    LOGW("Properties::GetFactoryKeyboxPath: Invalid parameter");
    return false;
  }
  *keybox = kFactoryKeyboxPath;
  return true;
}

bool Properties::GetOEMCryptoPaths(std::vector<std::string>* library_names) {
  if (!library_names) {
    LOGW("Properties::GetOEMCryptoPath: Invalid parameter");
    return false;
  }

  library_names->push_back("liboemcrypto.so");

  return true;
}

bool Properties::GetSandboxId(std::string* /* sandbox_id */) {
  // TODO(fredgc): If needed, we could support android running on a VM by
  // reading the sandbox ID from the file system.  If the file system
  // does not have a sandbox id, we would generate a random
  // one. Another option is to have sandbox id be a system property.
  // However, that is enough work not to do it pre-emptively.  This
  // TODO is just to let future coders know that the framework is in
  // place, and should be pretty easy to plumb.
  return false;
}

bool Properties::AlwaysUseKeySetIds() { return false; }

bool Properties::UseProviderIdInProvisioningRequest() { return false; }

bool Properties::ForceL3() {
  return WidevineProperties::forcel3_enabled().value_or(false);
}

}  // namespace wvcdm