// Copyright 2012 Google Inc. All Rights Reserved.
//
// Crypto - wrapper classes for OEMCrypto interface
//

#include "crypto_session.h"

#include <arpa/inet.h>  // needed for ntoh()
#include <iostream>
#include <string.h>

#include "crypto_key.h"
#include "log.h"
#include "properties.h"
#include "string_conversions.h"
#include "wv_cdm_constants.h"

namespace {
// Encode unsigned integer into a big endian formatted string
std::string EncodeUint32(unsigned int u) {
  std::string s;
  s.append(1, (u >> 24) & 0xFF);
  s.append(1, (u >> 16) & 0xFF);
  s.append(1, (u >> 8) & 0xFF);
  s.append(1, (u >> 0) & 0xFF);
  return s;
}
const uint32_t kRsaSignatureLength = 256;
}

namespace wvcdm {

Lock CryptoSession::crypto_lock_;
bool CryptoSession::initialized_ = false;
int CryptoSession::session_count_ = 0;
uint64_t CryptoSession::request_id_index_ = 0;

CryptoSession::CryptoSession()
    : open_(false),
      update_usage_table_after_close_session_(false),
      is_destination_buffer_type_valid_(false),
      requested_security_level_(kLevelDefault),
      request_id_base_(0) {
  Init();
}

CryptoSession::~CryptoSession() {
  if (open_) {
    Close();
  }
  Terminate();
}

void CryptoSession::Init() {
  LOGV("CryptoSession::Init");
  AutoLock auto_lock(crypto_lock_);
  session_count_ += 1;
  if (initialized_) return;
  OEMCryptoResult sts = OEMCrypto_Initialize();
  if (OEMCrypto_SUCCESS != sts) {
    LOGE("OEMCrypto_Initialize failed: %d", sts);
    return;
  }
  initialized_ = true;
}

void CryptoSession::Terminate() {
  LOGV("CryptoSession::Terminate");
  AutoLock auto_lock(crypto_lock_);
  if (session_count_ > 0) {
    session_count_ -= 1;
  } else {
    LOGE("CryptoSession::Terminate error, session count: %d", session_count_);
  }
  if (session_count_ > 0 || !initialized_) return;
  OEMCryptoResult sts = OEMCrypto_Terminate();
  if (OEMCrypto_SUCCESS != sts) {
    LOGE("OEMCrypto_Terminate failed: %d", sts);
  }
  initialized_ = false;
}

bool CryptoSession::ValidateKeybox() {
  LOGV("CryptoSession::ValidateKeybox: Lock");
  AutoLock auto_lock(crypto_lock_);
  if (!initialized_) {
    return false;
  }
  OEMCryptoResult result = OEMCrypto_IsKeyboxValid(requested_security_level_);
  return (OEMCrypto_SUCCESS == result);
}

bool CryptoSession::GetToken(std::string* token) {
  if (!token) {
    LOGE("CryptoSession::GetToken : No token passed to method.");
    return false;
  }
  uint8_t buf[KEYBOX_KEY_DATA_SIZE];
  size_t bufSize = sizeof(buf);
  LOGV("CryptoSession::GetToken: Lock");
  AutoLock auto_lock(crypto_lock_);
  if (!initialized_) {
    return false;
  }
  OEMCryptoResult sts =
      OEMCrypto_GetKeyData(buf, &bufSize, requested_security_level_);
  if (OEMCrypto_SUCCESS != sts) {
    return false;
  }
  token->assign((const char*)buf, (size_t)bufSize);
  return true;
}

CdmSecurityLevel CryptoSession::GetSecurityLevel() {
  LOGV("CryptoSession::GetSecurityLevel: Lock");
  AutoLock auto_lock(crypto_lock_);
  if (!initialized_) {
    return kSecurityLevelUninitialized;
  }

  std::string security_level =
      OEMCrypto_SecurityLevel(requested_security_level_);

  if ((security_level.size() != 2) || (security_level.at(0) != 'L')) {
    return kSecurityLevelUnknown;
  }

  switch (security_level.at(1)) {
    case '1':
      return kSecurityLevelL1;
    case '2':
      return kSecurityLevelL2;
    case '3':
      return kSecurityLevelL3;
    default:
      return kSecurityLevelUnknown;
  }

  return kSecurityLevelUnknown;
}

bool CryptoSession::GetDeviceUniqueId(std::string* device_id) {
  if (!device_id) {
    LOGE("CryptoSession::GetDeviceUniqueId : No buffer passed to method.");
    return false;
  }

  std::vector<uint8_t> id;
  size_t id_length = 32;

  id.resize(id_length);

  LOGV("CryptoSession::GetDeviceUniqueId: Lock");
  AutoLock auto_lock(crypto_lock_);
  if (!initialized_) {
    return false;
  }
  OEMCryptoResult sts =
      OEMCrypto_GetDeviceID(&id[0], &id_length, requested_security_level_);

  if (OEMCrypto_SUCCESS != sts) {
    return false;
  }

  device_id->assign(reinterpret_cast<char*>(&id[0]), id_length);
  return true;
}

bool CryptoSession::GetApiVersion(uint32_t* version) {
  if (!version) {
    LOGE("CryptoSession::GetApiVersion: No buffer passed to method.");
    return false;
  }

  if (!initialized_) {
    return false;
  }
  *version = OEMCrypto_APIVersion(requested_security_level_);
  return true;
}

bool CryptoSession::GetSystemId(uint32_t* system_id) {
  if (!system_id) {
    LOGE("CryptoSession::GetSystemId : No buffer passed to method.");
    return false;
  }

  uint8_t buf[KEYBOX_KEY_DATA_SIZE];
  size_t buf_size = sizeof(buf);

  LOGV("CryptoSession::GetSystemId: Lock");
  AutoLock auto_lock(crypto_lock_);
  if (!initialized_) {
    return false;
  }
  OEMCryptoResult sts =
      OEMCrypto_GetKeyData(buf, &buf_size, requested_security_level_);

  if (OEMCrypto_SUCCESS != sts) {
    return false;
  }

  // Decode 32-bit int encoded as network-byte-order byte array starting at
  // index 4.
  uint32_t* id = reinterpret_cast<uint32_t*>(&buf[4]);

  *system_id = ntohl(*id);
  return true;
}

bool CryptoSession::GetProvisioningId(std::string* provisioning_id) {
  if (!provisioning_id) {
    LOGE("CryptoSession::GetProvisioningId : No buffer passed to method.");
    return false;
  }

  uint8_t buf[KEYBOX_KEY_DATA_SIZE];
  size_t buf_size = sizeof(buf);

  LOGV("CryptoSession::GetProvisioningId: Lock");
  AutoLock auto_lock(crypto_lock_);
  if (!initialized_) {
    return false;
  }
  OEMCryptoResult sts =
      OEMCrypto_GetKeyData(buf, &buf_size, requested_security_level_);

  if (OEMCrypto_SUCCESS != sts) {
    return false;
  }

  provisioning_id->assign(reinterpret_cast<char*>(&buf[8]), 16);
  return true;
}

CdmResponseType CryptoSession::Open(SecurityLevel requested_security_level) {
  LOGV("CryptoSession::Open: Lock");
  AutoLock auto_lock(crypto_lock_);
  if (!initialized_) return UNKNOWN_ERROR;
  if (open_) return NO_ERROR;

  OEMCrypto_SESSION sid;
  requested_security_level_ = requested_security_level;
  OEMCryptoResult sts = OEMCrypto_OpenSession(&sid, requested_security_level);
  if (OEMCrypto_SUCCESS == sts) {
    oec_session_id_ = static_cast<CryptoSessionId>(sid);
    LOGV("OpenSession: id= %ld", (uint32_t)oec_session_id_);
    open_ = true;
  } else if (OEMCrypto_ERROR_TOO_MANY_SESSIONS == sts) {
    LOGE("OEMCrypto_Open failed: %d, open sessions: %ld, initialized: %d", sts,
         session_count_, (int)initialized_);
    return INSUFFICIENT_CRYPTO_RESOURCES;
  }
  if (!open_) {
    LOGE("OEMCrypto_Open failed: %d, open sessions: %ld, initialized: %d", sts,
         session_count_, (int)initialized_);
    return UNKNOWN_ERROR;
  }

  OEMCrypto_GetRandom(reinterpret_cast<uint8_t*>(&request_id_base_),
                      sizeof(request_id_base_));
  ++request_id_index_;
  return NO_ERROR;
}

void CryptoSession::Close() {
  LOGV("CloseSession: id=%ld open=%s", (uint32_t)oec_session_id_,
       open_ ? "true" : "false");
  AutoLock auto_lock(crypto_lock_);
  if (!open_) return;
  if (OEMCrypto_SUCCESS == OEMCrypto_CloseSession(oec_session_id_)) {
    open_ = false;
    if (update_usage_table_after_close_session_) {
      OEMCryptoResult sts = OEMCrypto_UpdateUsageTable();
      if (sts != OEMCrypto_SUCCESS)
        LOGW("CryptoSession::Close: OEMCrypto_UpdateUsageTable error=%ld", sts);
    }
  }
}

bool CryptoSession::GenerateRequestId(std::string* req_id_str) {
  LOGV("CryptoSession::GenerateRequestId: Lock");
  AutoLock auto_lock(crypto_lock_);
  if (!req_id_str) {
    LOGE("CryptoSession::GenerateRequestId: No output destination provided.");
    return false;
  }

  *req_id_str = HexEncode(reinterpret_cast<uint8_t*>(&request_id_base_),
                          sizeof(request_id_base_)) +
                HexEncode(reinterpret_cast<uint8_t*>(&request_id_index_),
                          sizeof(request_id_index_));
  return true;
}

bool CryptoSession::PrepareRequest(const std::string& message,
                                   bool is_provisioning,
                                   std::string* signature) {
  LOGV("CryptoSession::PrepareRequest: Lock");
  AutoLock auto_lock(crypto_lock_);

  if (!signature) {
    LOGE("CryptoSession::PrepareRequest : No output destination provided.");
    return false;
  }

  if (!Properties::use_certificates_as_identification() || is_provisioning) {
    if (!GenerateDerivedKeys(message)) return false;

    if (!GenerateSignature(message, signature)) return false;
  } else {
    if (!GenerateRsaSignature(message, signature)) return false;
  }

  return true;
}

bool CryptoSession::PrepareRenewalRequest(const std::string& message,
                                          std::string* signature) {
  LOGV("CryptoSession::PrepareRenewalRequest: Lock");
  AutoLock auto_lock(crypto_lock_);

  if (!signature) {
    LOGE(
        "CryptoSession::PrepareRenewalRequest : No output destination "
        "provided.");
    return false;
  }

  if (!GenerateSignature(message, signature)) {
    return false;
  }

  return true;
}

void CryptoSession::GenerateMacContext(const std::string& input_context,
                                       std::string* deriv_context) {
  if (!deriv_context) {
    LOGE("CryptoSession::GenerateMacContext : No output destination provided.");
    return;
  }

  const std::string kSigningKeyLabel = "AUTHENTICATION";
  const size_t kSigningKeySizeBits = MAC_KEY_SIZE * 8;

  deriv_context->assign(kSigningKeyLabel);
  deriv_context->append(1, '\0');
  deriv_context->append(input_context);
  deriv_context->append(EncodeUint32(kSigningKeySizeBits * 2));
}

void CryptoSession::GenerateEncryptContext(const std::string& input_context,
                                           std::string* deriv_context) {
  if (!deriv_context) {
    LOGE(
        "CryptoSession::GenerateEncryptContext : No output destination "
        "provided.");
    return;
  }

  const std::string kEncryptionKeyLabel = "ENCRYPTION";
  const size_t kEncryptionKeySizeBits = KEY_SIZE * 8;

  deriv_context->assign(kEncryptionKeyLabel);
  deriv_context->append(1, '\0');
  deriv_context->append(input_context);
  deriv_context->append(EncodeUint32(kEncryptionKeySizeBits));
}

size_t CryptoSession::GetOffset(std::string message, std::string field) {
  size_t pos = message.find(field);
  if (pos == std::string::npos) {
    LOGE("CryptoSession::GetOffset : Cannot find offset for %s", field.c_str());
    pos = 0;
  }
  return pos;
}

CdmResponseType CryptoSession::LoadKeys(
    const std::string& message, const std::string& signature,
    const std::string& mac_key_iv, const std::string& mac_key,
    const std::vector<CryptoKey>& keys,
    const std::string& provider_session_token) {
  LOGV("CryptoSession::LoadKeys: Lock");
  AutoLock auto_lock(crypto_lock_);

  const uint8_t* msg = reinterpret_cast<const uint8_t*>(message.data());
  const uint8_t* enc_mac_key = NULL;
  const uint8_t* enc_mac_key_iv = NULL;
  if (mac_key.size() >= MAC_KEY_SIZE && mac_key_iv.size() >= KEY_IV_SIZE) {
    enc_mac_key = msg + GetOffset(message, mac_key);
    enc_mac_key_iv = msg + GetOffset(message, mac_key_iv);
  } else {
    LOGV("CryptoSession::LoadKeys: enc_mac_key not set");
  }
  std::vector<OEMCrypto_KeyObject> load_keys(keys.size());
  for (size_t i = 0; i < keys.size(); ++i) {
    const CryptoKey* ki = &keys[i];
    OEMCrypto_KeyObject* ko = &load_keys[i];
    ko->key_id = msg + GetOffset(message, ki->key_id());
    ko->key_id_length = ki->key_id().length();
    ko->key_data_iv = msg + GetOffset(message, ki->key_data_iv());
    ko->key_data = msg + GetOffset(message, ki->key_data());
    ko->key_data_length = ki->key_data().length();
    if (ki->HasKeyControl()) {
      ko->key_control_iv = msg + GetOffset(message, ki->key_control_iv());
      ko->key_control = msg + GetOffset(message, ki->key_control());
    } else {
      LOGE("For key %d: XXX key has no control block. size=%d", i,
           ki->key_control().size());
      ko->key_control_iv = NULL;
      ko->key_control = NULL;
    }
  }
  uint8_t* pst = NULL;
  if (!provider_session_token.empty()) {
    pst =
        const_cast<uint8_t*>(msg) + GetOffset(message, provider_session_token);
  }
  LOGV("LoadKeys: id=%ld", (uint32_t)oec_session_id_);
  OEMCryptoResult sts = OEMCrypto_LoadKeys(
      oec_session_id_, msg, message.size(),
      reinterpret_cast<const uint8_t*>(signature.data()), signature.size(),
      enc_mac_key_iv, enc_mac_key, keys.size(), &load_keys[0], pst,
      provider_session_token.length());

  if (OEMCrypto_SUCCESS == sts) {
    if (!provider_session_token.empty()) {
      update_usage_table_after_close_session_ = true;
      sts = OEMCrypto_UpdateUsageTable();
      if (sts != OEMCrypto_SUCCESS) {
        LOGW("CryptoSession::LoadKeys: OEMCrypto_UpdateUsageTable error=%ld",
             sts);
      }
    }
    return KEY_ADDED;
  } else if (OEMCrypto_ERROR_TOO_MANY_KEYS == sts) {
    LOGE("CryptoSession::LoadKeys: OEMCrypto_LoadKeys error=%d", sts);
    return INSUFFICIENT_CRYPTO_RESOURCES;
  } else {
    LOGE("CryptoSession::LoadKeys: OEMCrypto_LoadKeys error=%d", sts);
    return KEY_ERROR;
  }
}

bool CryptoSession::LoadCertificatePrivateKey(std::string& wrapped_key) {
  LOGV("CryptoSession::LoadCertificatePrivateKey: Lock");
  AutoLock auto_lock(crypto_lock_);

  LOGV("LoadDeviceRSAKey: id=%ld", (uint32_t)oec_session_id_);
  OEMCryptoResult sts = OEMCrypto_LoadDeviceRSAKey(
      oec_session_id_, reinterpret_cast<const uint8_t*>(wrapped_key.data()),
      wrapped_key.size());

  if (OEMCrypto_SUCCESS != sts) {
    LOGE("LoadCertificatePrivateKey: OEMCrypto_LoadDeviceRSAKey error=%d", sts);
    return false;
  }

  return true;
}

bool CryptoSession::RefreshKeys(const std::string& message,
                                const std::string& signature, int num_keys,
                                const CryptoKey* key_array) {
  LOGV("CryptoSession::RefreshKeys: Lock");
  AutoLock auto_lock(crypto_lock_);

  const uint8_t* msg = reinterpret_cast<const uint8_t*>(message.data());
  std::vector<OEMCrypto_KeyRefreshObject> load_key_array(num_keys);
  for (int i = 0; i < num_keys; ++i) {
    const CryptoKey* ki = &key_array[i];
    OEMCrypto_KeyRefreshObject* ko = &load_key_array[i];
    if (ki->key_id().empty()) {
      ko->key_id = NULL;
    } else {
      ko->key_id = msg + GetOffset(message, ki->key_id());
    }
    if (ki->HasKeyControl()) {
      if (ki->key_control_iv().empty()) {
        ko->key_control_iv = NULL;
      } else {
        ko->key_control_iv = msg + GetOffset(message, ki->key_control_iv());
      }
      ko->key_control = msg + GetOffset(message, ki->key_control());
    } else {
      ko->key_control_iv = NULL;
      ko->key_control = NULL;
    }
  }
  LOGV("RefreshKeys: id=%ld", static_cast<uint32_t>(oec_session_id_));
  return (
      OEMCrypto_SUCCESS ==
      OEMCrypto_RefreshKeys(oec_session_id_, msg, message.size(),
                            reinterpret_cast<const uint8_t*>(signature.data()),
                            signature.size(), num_keys, &load_key_array[0]));
}

bool CryptoSession::SelectKey(const std::string& key_id) {
  const uint8_t* key_id_string =
      reinterpret_cast<const uint8_t*>(key_id.data());

  OEMCryptoResult sts =
      OEMCrypto_SelectKey(oec_session_id_, key_id_string, key_id.size());
  if (OEMCrypto_SUCCESS != sts) {
    return false;
  }
  return true;
}

bool CryptoSession::GenerateDerivedKeys(const std::string& message) {
  std::string mac_deriv_message;
  std::string enc_deriv_message;
  GenerateMacContext(message, &mac_deriv_message);
  GenerateEncryptContext(message, &enc_deriv_message);

  LOGV("GenerateDerivedKeys: id=%ld", (uint32_t)oec_session_id_);
  OEMCryptoResult sts = OEMCrypto_GenerateDerivedKeys(
      oec_session_id_,
      reinterpret_cast<const uint8_t*>(mac_deriv_message.data()),
      mac_deriv_message.size(),
      reinterpret_cast<const uint8_t*>(enc_deriv_message.data()),
      enc_deriv_message.size());

  if (OEMCrypto_SUCCESS != sts) {
    LOGE("GenerateDerivedKeys: OEMCrypto_GenerateDerivedKeys error=%d", sts);
    return false;
  }

  return true;
}

bool CryptoSession::GenerateDerivedKeys(const std::string& message,
                                        const std::string& session_key) {
  std::string mac_deriv_message;
  std::string enc_deriv_message;
  GenerateMacContext(message, &mac_deriv_message);
  GenerateEncryptContext(message, &enc_deriv_message);

  LOGV("GenerateDerivedKeys: id=%ld", (uint32_t)oec_session_id_);
  OEMCryptoResult sts = OEMCrypto_DeriveKeysFromSessionKey(
      oec_session_id_, reinterpret_cast<const uint8_t*>(session_key.data()),
      session_key.size(),
      reinterpret_cast<const uint8_t*>(mac_deriv_message.data()),
      mac_deriv_message.size(),
      reinterpret_cast<const uint8_t*>(enc_deriv_message.data()),
      enc_deriv_message.size());

  if (OEMCrypto_SUCCESS != sts) {
    LOGE("GenerateDerivedKeys: OEMCrypto_DeriveKeysFromSessionKey err=%d", sts);
    return false;
  }

  return true;
}

bool CryptoSession::GenerateSignature(const std::string& message,
                                      std::string* signature) {
  LOGV("GenerateSignature: id=%ld", (uint32_t)oec_session_id_);
  if (!signature) return false;

  size_t length = signature->size();
  OEMCryptoResult sts = OEMCrypto_GenerateSignature(
      oec_session_id_, reinterpret_cast<const uint8_t*>(message.data()),
      message.size(),
      reinterpret_cast<uint8_t*>(const_cast<char*>(signature->data())),
      &length);

  if (OEMCrypto_SUCCESS != sts) {
    if (OEMCrypto_ERROR_SHORT_BUFFER != sts) {
      LOGE("GenerateSignature: OEMCrypto_GenerateSignature err=%d", sts);
      return false;
    }

    // Retry with proper-sized signature buffer
    signature->resize(length);
    sts = OEMCrypto_GenerateSignature(
        oec_session_id_, reinterpret_cast<const uint8_t*>(message.data()),
        message.size(),
        reinterpret_cast<uint8_t*>(const_cast<char*>(signature->data())),
        &length);

    if (OEMCrypto_SUCCESS != sts) {
      LOGE("GenerateSignature: OEMCrypto_GenerateSignature err=%d", sts);
      return false;
    }
  }

  // Trim signature buffer
  signature->resize(length);

  return true;
}

bool CryptoSession::GenerateRsaSignature(const std::string& message,
                                         std::string* signature) {
  LOGV("GenerateRsaSignature: id=%ld", (uint32_t)oec_session_id_);
  if (!signature) return false;

  signature->resize(kRsaSignatureLength);
  size_t length = signature->size();
  OEMCryptoResult sts = OEMCrypto_GenerateRSASignature(
      oec_session_id_, reinterpret_cast<const uint8_t*>(message.data()),
      message.size(),
      reinterpret_cast<uint8_t*>(const_cast<char*>(signature->data())), &length,
      kSign_RSASSA_PSS);

  if (OEMCrypto_SUCCESS != sts) {
    if (OEMCrypto_ERROR_SHORT_BUFFER != sts) {
      LOGE("GenerateRsaSignature: OEMCrypto_GenerateRSASignature err=%d", sts);
      return false;
    }

    // Retry with proper-sized signature buffer
    signature->resize(length);
    sts = OEMCrypto_GenerateRSASignature(
        oec_session_id_, reinterpret_cast<const uint8_t*>(message.data()),
        message.size(),
        reinterpret_cast<uint8_t*>(const_cast<char*>(signature->data())),
        &length, kSign_RSASSA_PSS);

    if (OEMCrypto_SUCCESS != sts) {
      LOGE("GenerateRsaSignature: OEMCrypto_GenerateRSASignature err=%d", sts);
      return false;
    }
  }

  // Trim signature buffer
  signature->resize(length);

  return true;
}

CdmResponseType CryptoSession::Decrypt(const CdmDecryptionParameters& params) {
  if (!is_destination_buffer_type_valid_) {
    if (!SetDestinationBufferType()) return UNKNOWN_ERROR;
  }

  AutoLock auto_lock(crypto_lock_);
  // Check if key needs to be selected
  if (params.is_encrypted) {
    if (key_id_ != *params.key_id) {
      if (SelectKey(*params.key_id)) {
        key_id_ = *params.key_id;
      } else {
        return NEED_KEY;
      }
    }
  }

  OEMCrypto_DestBufferDesc buffer_descriptor;
  buffer_descriptor.type =
      params.is_secure ? destination_buffer_type_ : OEMCrypto_BufferType_Clear;

  switch (buffer_descriptor.type) {
    case OEMCrypto_BufferType_Clear:
      buffer_descriptor.buffer.clear.address =
          static_cast<uint8_t*>(params.decrypt_buffer) +
          params.decrypt_buffer_offset;
      buffer_descriptor.buffer.clear.max_length =
          params.decrypt_buffer_length - params.decrypt_buffer_offset;
      break;
    case OEMCrypto_BufferType_Secure:
      buffer_descriptor.buffer.secure.handle = params.decrypt_buffer;
      buffer_descriptor.buffer.secure.offset = params.decrypt_buffer_offset;
      buffer_descriptor.buffer.secure.max_length = params.decrypt_buffer_length;
      break;
    case OEMCrypto_BufferType_Direct:
      buffer_descriptor.type = OEMCrypto_BufferType_Direct;
      buffer_descriptor.buffer.direct.is_video = params.is_video;
      break;
  }

  OEMCryptoResult sts = OEMCrypto_DecryptCTR(
      oec_session_id_, params.encrypt_buffer, params.encrypt_length,
      params.is_encrypted, &(*params.iv).front(), params.block_offset,
      &buffer_descriptor, params.subsample_flags);

  switch (sts) {
    case OEMCrypto_SUCCESS:
      return NO_ERROR;
    case OEMCrypto_ERROR_INSUFFICIENT_RESOURCES:
      return INSUFFICIENT_CRYPTO_RESOURCES;
    case OEMCrypto_ERROR_KEY_EXPIRED:
      return NEED_KEY;
    default:
      return UNKNOWN_ERROR;
  }
}

bool CryptoSession::UsageInformationSupport(bool* has_support) {
  LOGV("UsageInformationSupport: id=%ld", (uint32_t)oec_session_id_);
  if (!initialized_) return false;

  *has_support = OEMCrypto_SupportsUsageTable(requested_security_level_);
  return true;
}

CdmResponseType CryptoSession::UpdateUsageInformation() {
  LOGV("UpdateUsageInformation: id=%ld", (uint32_t)oec_session_id_);
  AutoLock auto_lock(crypto_lock_);
  if (!initialized_) return UNKNOWN_ERROR;

  OEMCryptoResult status = OEMCrypto_UpdateUsageTable();
  if (status != OEMCrypto_SUCCESS) {
    LOGE("CryptoSession::UsageUsageInformation: error=%ld", status);
    return UNKNOWN_ERROR;
  }
  return NO_ERROR;
}

CdmResponseType CryptoSession::DeactivateUsageInformation(
    const std::string& provider_session_token) {
  LOGV("DeactivateUsageInformation: id=%ld", (uint32_t)oec_session_id_);

  AutoLock auto_lock(crypto_lock_);
  uint8_t* pst = reinterpret_cast<uint8_t*>(
      const_cast<char*>(provider_session_token.data()));

  OEMCryptoResult status =
      OEMCrypto_DeactivateUsageEntry(pst, provider_session_token.length());

  switch (status) {
    case OEMCrypto_SUCCESS:
      return NO_ERROR;
    case OEMCrypto_ERROR_INVALID_CONTEXT:
      LOGE("CryptoSession::DeactivateUsageInformation: invalid context error");
      return KEY_CANCELED;
    default:
      LOGE("CryptoSession::DeactivateUsageInformation: error=%ld", status);
      return UNKNOWN_ERROR;
  }
}

CdmResponseType CryptoSession::GenerateUsageReport(
    const std::string& provider_session_token, std::string* usage_report,
    UsageDurationStatus* usage_duration_status, int64_t* seconds_since_started,
    int64_t* seconds_since_last_played) {
  LOGV("GenerateUsageReport: id=%ld", (uint32_t)oec_session_id_);

  if (NULL == usage_report) {
    LOGE("CryptoSession::GenerateUsageReport: usage_report parameter is null");
    return UNKNOWN_ERROR;
  }

  AutoLock auto_lock(crypto_lock_);
  uint8_t* pst = reinterpret_cast<uint8_t*>(
      const_cast<char*>(provider_session_token.data()));

  size_t usage_length = 0;
  OEMCryptoResult status = OEMCrypto_ReportUsage(
      oec_session_id_, pst, provider_session_token.length(), NULL,
      &usage_length);

  if (OEMCrypto_SUCCESS != status) {
    if (OEMCrypto_ERROR_SHORT_BUFFER != status) {
      LOGE("CryptoSession::GenerateUsageReport: Report Usage error=%ld",
           status);
      return UNKNOWN_ERROR;
    }
  }

  usage_report->resize(usage_length);
  OEMCrypto_PST_Report* report = reinterpret_cast<OEMCrypto_PST_Report*>(
      const_cast<char*>(usage_report->data()));
  status = OEMCrypto_ReportUsage(oec_session_id_, pst,
                                 provider_session_token.length(), report,
                                 &usage_length);

  if (OEMCrypto_SUCCESS != status) {
    LOGE("CryptoSession::GenerateUsageReport: Report Usage error=%ld", status);
    return UNKNOWN_ERROR;
  }

  if (usage_length != usage_report->length()) {
    usage_report->resize(usage_length);
  }

  OEMCrypto_PST_Report pst_report;
  *usage_duration_status = kUsageDurationsInvalid;
  if (usage_length < sizeof(pst_report)) {
    LOGE("CryptoSession::GenerateUsageReport: usage report too small=%ld",
         usage_length);
    return NO_ERROR;  // usage report available but no duration information
  }

  memcpy(&pst_report, usage_report->data(), sizeof(pst_report));
  if (kUnused == pst_report.status) {
    *usage_duration_status = kUsageDurationPlaybackNotBegun;
    return NO_ERROR;
  }
  LOGV("OEMCrypto_PST_Report.status: %d\n", pst_report.status);
  LOGV("OEMCrypto_PST_Report.clock_security_level: %d\n",
       pst_report.clock_security_level);
  LOGV("OEMCrypto_PST_Report.pst_length: %d\n", pst_report.pst_length);
  LOGV("OEMCrypto_PST_Report.padding: %d\n", pst_report.padding);
  LOGV("OEMCrypto_PST_Report.seconds_since_license_received: %lld\n",
       ntohll64(pst_report.seconds_since_license_received));
  LOGV("OEMCrypto_PST_Report.seconds_since_first_decrypt: %lld\n",
       ntohll64(pst_report.seconds_since_first_decrypt));
  LOGV("OEMCrypto_PST_Report.seconds_since_last_decrypt: %lld\n",
       ntohll64(pst_report.seconds_since_last_decrypt));
  LOGV("OEMCrypto_PST_Report: %s\n", b2a_hex(*usage_report).c_str());

  // When usage report state is inactive, we have to deduce whether the
  // license was ever used.
  if (kInactive == pst_report.status &&
      (0 > ntohll64(pst_report.seconds_since_first_decrypt) ||
       ntohll64(pst_report.seconds_since_license_received) <
           ntohll64(pst_report.seconds_since_first_decrypt))) {
    *usage_duration_status = kUsageDurationPlaybackNotBegun;
    return NO_ERROR;
  }

  *usage_duration_status = kUsageDurationsValid;
  *seconds_since_started = ntohll64(pst_report.seconds_since_first_decrypt);
  *seconds_since_last_played = ntohll64(pst_report.seconds_since_last_decrypt);
  return NO_ERROR;
}

CdmResponseType CryptoSession::ReleaseUsageInformation(
    const std::string& message, const std::string& signature,
    const std::string& provider_session_token) {
  LOGV("ReleaseUsageInformation: id=%ld", (uint32_t)oec_session_id_);
  AutoLock auto_lock(crypto_lock_);
  const uint8_t* msg = reinterpret_cast<const uint8_t*>(message.data());
  const uint8_t* sig = reinterpret_cast<const uint8_t*>(signature.data());
  const uint8_t* pst = msg + GetOffset(message, provider_session_token);

  OEMCryptoResult status = OEMCrypto_DeleteUsageEntry(
      oec_session_id_, pst, provider_session_token.length(), msg,
      message.length(), sig, signature.length());

  if (OEMCrypto_SUCCESS != status) {
    LOGE("CryptoSession::ReleaseUsageInformation: Report Usage error=%ld",
         status);
    return UNKNOWN_ERROR;
  }
  status = OEMCrypto_UpdateUsageTable();
  if (status != OEMCrypto_SUCCESS) {
    LOGW("CryptoSession::ReleaseUsageInformation: update table error=%ld",
         status);
  }

  return NO_ERROR;
}

CdmResponseType CryptoSession::DeleteAllUsageReports() {
  LOGV("DeleteAllUsageReports");
  OEMCryptoResult status = OEMCrypto_DeleteUsageTable();

  if (OEMCrypto_SUCCESS != status) {
    LOGE("CryptoSession::DeleteAllUsageReports: Delete Usage Table error =%ld",
         status);
  }

  status = OEMCrypto_UpdateUsageTable();
  if (status != OEMCrypto_SUCCESS) {
    LOGE("CryptoSession::DeletaAllUsageReports: update table error=%ld",
         status);
  }

  return NO_ERROR;
}

bool CryptoSession::GenerateNonce(uint32_t* nonce) {
  if (!nonce) {
    LOGE("input parameter is null");
    return false;
  }

  LOGV("CryptoSession::GenerateNonce: Lock");
  AutoLock auto_lock(crypto_lock_);

  return (OEMCrypto_SUCCESS == OEMCrypto_GenerateNonce(oec_session_id_, nonce));
}

bool CryptoSession::SetDestinationBufferType() {
  if (Properties::oem_crypto_use_secure_buffers()) {
    if (GetSecurityLevel() == kSecurityLevelL1) {
      destination_buffer_type_ = OEMCrypto_BufferType_Secure;
    } else {
      destination_buffer_type_ = OEMCrypto_BufferType_Clear;
    }
  } else if (Properties::oem_crypto_use_fifo()) {
    destination_buffer_type_ = OEMCrypto_BufferType_Direct;
  } else if (Properties::oem_crypto_use_userspace_buffers()) {
    destination_buffer_type_ = OEMCrypto_BufferType_Clear;
  } else {
    return false;
  }

  is_destination_buffer_type_valid_ = true;
  return true;
}

bool CryptoSession::RewrapDeviceRSAKey(const std::string& message,
                                       const std::string& signature,
                                       const std::string& nonce,
                                       const std::string& enc_rsa_key,
                                       const std::string& rsa_key_iv,
                                       std::string* wrapped_rsa_key) {
  LOGV("CryptoSession::RewrapDeviceRSAKey, session id=%ld",
       static_cast<uint32_t>(oec_session_id_));

  const uint8_t* signed_msg = reinterpret_cast<const uint8_t*>(message.data());
  const uint8_t* msg_rsa_key = NULL;
  const uint8_t* msg_rsa_key_iv = NULL;
  const uint32_t* msg_nonce = NULL;
  if (enc_rsa_key.size() >= MAC_KEY_SIZE && rsa_key_iv.size() >= KEY_IV_SIZE) {
    msg_rsa_key = signed_msg + GetOffset(message, enc_rsa_key);
    msg_rsa_key_iv = signed_msg + GetOffset(message, rsa_key_iv);
    msg_nonce = reinterpret_cast<const uint32_t*>(signed_msg +
                                                  GetOffset(message, nonce));
  }

  // Gets wrapped_rsa_key_length by passing NULL as uint8_t* wrapped_rsa_key
  // and 0 as wrapped_rsa_key_length.
  size_t wrapped_rsa_key_length = 0;
  OEMCryptoResult status = OEMCrypto_RewrapDeviceRSAKey(
      oec_session_id_, signed_msg, message.size(),
      reinterpret_cast<const uint8_t*>(signature.data()), signature.size(),
      msg_nonce, msg_rsa_key, enc_rsa_key.size(), msg_rsa_key_iv, NULL,
      &wrapped_rsa_key_length);
  if (status != OEMCrypto_ERROR_SHORT_BUFFER) {
    LOGE("OEMCrypto_RewrapDeviceRSAKey fails to get wrapped_rsa_key_length");
    return false;
  }

  wrapped_rsa_key->resize(wrapped_rsa_key_length);
  status = OEMCrypto_RewrapDeviceRSAKey(
      oec_session_id_, signed_msg, message.size(),
      reinterpret_cast<const uint8_t*>(signature.data()), signature.size(),
      msg_nonce, msg_rsa_key, enc_rsa_key.size(), msg_rsa_key_iv,
      reinterpret_cast<uint8_t*>(&(*wrapped_rsa_key)[0]),
      &wrapped_rsa_key_length);

  wrapped_rsa_key->resize(wrapped_rsa_key_length);

  if (OEMCrypto_SUCCESS != status) {
    LOGE("OEMCrypto_RewrapDeviceRSAKey fails with %d", status);
    return false;
  }

  return true;
}

bool CryptoSession::GetHdcpCapabilities(OemCryptoHdcpVersion* current_version,
                                        OemCryptoHdcpVersion* max_version) {
  LOGV("GetHdcpCapabilities: id=%ld", (uint32_t)oec_session_id_);
  if (!initialized_) return false;
  if (current_version == NULL || max_version == NULL) {
    LOGE(
        "CryptoSession::GetHdcpCapabilities: |current_version|, "
        "|max_version| cannot be NULL");
    return false;
  }
  OEMCrypto_HDCP_Capability current, max;
  OEMCryptoResult status = OEMCrypto_GetHDCPCapability(&current, &max);

  if (OEMCrypto_SUCCESS != status) {
    LOGW("OEMCrypto_GetHDCPCapability fails with %d", status);
    return false;
  }
  *current_version = static_cast<OemCryptoHdcpVersion>(current);
  *max_version = static_cast<OemCryptoHdcpVersion>(max);

  return true;
}

bool CryptoSession::GetRandom(size_t data_length, uint8_t* random_data) {
  if (random_data == NULL) {
    LOGE("CryptoSession::GetRandom: random data destination not provided");
    return false;
  }
  OEMCryptoResult sts = OEMCrypto_GetRandom(random_data, data_length);

  if (sts != OEMCrypto_SUCCESS) {
    LOGE("OEMCrypto_GetRandom fails with %d", sts);
    return false;
  }

  return true;
}

bool CryptoSession::GetMaxNumberOfSessions(size_t* max) {
  LOGV("GetMaxNumberOfSessions");
  if (!initialized_) return false;
  if (max == NULL) {
    LOGE("CryptoSession::GetMaxNumberOfSessions: |max| cannot be NULL");
    return false;
  }

  size_t max_sessions;
  OEMCryptoResult status = OEMCrypto_GetMaxNumberOfSessions(&max_sessions);
  if (OEMCrypto_SUCCESS != status) {
    LOGW("OEMCrypto_GetMaxNumberOfSessions fails with %d", status);
    return false;
  }
  *max = max_sessions;
  return true;
}

}  // namespace wvcdm