// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef MEDIA_CDM_LIBRARY_CDM_CDM_HOST_PROXY_IMPL_H_
#define MEDIA_CDM_LIBRARY_CDM_CDM_HOST_PROXY_IMPL_H_

#include "base/memory/raw_ptr.h"
#include "base/notimplemented.h"
#include "media/cdm/library_cdm/cdm_host_proxy.h"

namespace media {

namespace {

cdm::KeyStatus ToKeyStatus(const cdm::KeyStatus_2& key_status) {
  switch (key_status) {
    case cdm::KeyStatus_2::kUsable:
      return cdm::KeyStatus::kUsable;
    case cdm::KeyStatus_2::kInternalError:
    // Return UsableInFuture as error for KeyStatus as it should not be handled.
    case cdm::KeyStatus_2::kUsableInFuture:
      return cdm::KeyStatus::kInternalError;
    case cdm::KeyStatus_2::kExpired:
      return cdm::KeyStatus::kExpired;
    case cdm::KeyStatus_2::kOutputRestricted:
      return cdm::KeyStatus::kOutputRestricted;
    case cdm::KeyStatus_2::kOutputDownscaled:
      return cdm::KeyStatus::kOutputDownscaled;
    case cdm::KeyStatus_2::kStatusPending:
      return cdm::KeyStatus::kStatusPending;
    case cdm::KeyStatus_2::kReleased:
      return cdm::KeyStatus::kReleased;
    default:
      return cdm::KeyStatus::kInternalError;
  }
}

std::vector<cdm::KeyInformation> ConvertKeyInformation(
    const cdm::KeyInformation_2* keys_info,
    uint32_t keys_info_count) {
  std::vector<cdm::KeyInformation> keys_vector;
  keys_vector.reserve(keys_info_count);

  UNSAFE_BUFFERS({
    // SAFETY: The loop iterates exactly keys_info_count times, which is the
    // size of the buffer passed in.
    for (uint32_t i = 0; i < keys_info_count; ++i) {
      cdm::KeyInformation key = {};
      key.key_id = keys_info[i].key_id;
      key.key_id_size = keys_info[i].key_id_size;
      key.status = ToKeyStatus(keys_info[i].status);
      key.system_code = keys_info[i].system_code;
      keys_vector.push_back(key);
    }
  });

  return keys_vector;
}

}  // namespace

// A templated implementation of CdmHostProxy to forward Host calls to the
// correct CDM Host.
template <typename HostInterface>
class CdmHostProxyImpl : public CdmHostProxy {
 public:
  explicit CdmHostProxyImpl(HostInterface* host) : host_(host) {}

  CdmHostProxyImpl(const CdmHostProxyImpl&) = delete;
  CdmHostProxyImpl& operator=(const CdmHostProxyImpl&) = delete;

  ~CdmHostProxyImpl() override {}

  void OnInitialized(bool success) final {
    return host_->OnInitialized(success);
  }

  cdm::Buffer* Allocate(uint32_t capacity) final {
    return host_->Allocate(capacity);
  }

  void SetTimer(int64_t delay_ms, void* context) final {
    host_->SetTimer(delay_ms, context);
  }

  cdm::Time GetCurrentWallTime() final { return host_->GetCurrentWallTime(); }

  void OnResolveKeyStatusPromise(uint32_t promise_id,
                                 cdm::KeyStatus key_status) final {
    // Use cdm::KeyStatus_2 function instead.
    NOTIMPLEMENTED();
  }

  void OnResolveKeyStatusPromise(uint32_t promise_id,
                                 cdm::KeyStatus_2 key_status) final {
    host_->OnResolveKeyStatusPromise(promise_id, key_status);
  }

  void OnResolveNewSessionPromise(uint32_t promise_id,
                                  const char* session_id,
                                  uint32_t session_id_size) final {
    host_->OnResolveNewSessionPromise(promise_id, session_id, session_id_size);
  }

  void OnResolvePromise(uint32_t promise_id) final {
    host_->OnResolvePromise(promise_id);
  }

  void OnRejectPromise(uint32_t promise_id,
                       cdm::Exception exception,
                       uint32_t system_code,
                       const char* error_message,
                       uint32_t error_message_size) final {
    host_->OnRejectPromise(promise_id, exception, system_code, error_message,
                           error_message_size);
  }

  void OnSessionMessage(const char* session_id,
                        uint32_t session_id_size,
                        cdm::MessageType message_type,
                        const char* message,
                        uint32_t message_size) final {
    host_->OnSessionMessage(session_id, session_id_size, message_type, message,
                            message_size);
  }

  void OnSessionKeysChange(const char* session_id,
                           uint32_t session_id_size,
                           bool has_additional_usable_key,
                           const cdm::KeyInformation* keys_info,
                           uint32_t keys_info_count) final {
    // Use cdm::KeyInformation_2 function instead.
    NOTIMPLEMENTED();
  }

  void OnSessionKeysChange(const char* session_id,
                           uint32_t session_id_size,
                           bool has_additional_usable_key,
                           const cdm::KeyInformation_2* keys_info,
                           uint32_t keys_info_count) final {
    host_->OnSessionKeysChange(session_id, session_id_size,
                               has_additional_usable_key, keys_info,
                               keys_info_count);
  }

  void OnExpirationChange(const char* session_id,
                          uint32_t session_id_size,
                          cdm::Time new_expiry_time) final {
    host_->OnExpirationChange(session_id, session_id_size, new_expiry_time);
  }

  void OnSessionClosed(const char* session_id, uint32_t session_id_size) final {
    host_->OnSessionClosed(session_id, session_id_size);
  }

  void SendPlatformChallenge(const char* service_id,
                             uint32_t service_id_size,
                             const char* challenge,
                             uint32_t challenge_size) final {
    host_->SendPlatformChallenge(service_id, service_id_size, challenge,
                                 challenge_size);
  }

  void EnableOutputProtection(uint32_t desired_protection_mask) final {
    host_->EnableOutputProtection(desired_protection_mask);
  }

  void QueryOutputProtectionStatus() final {
    host_->QueryOutputProtectionStatus();
  }

  void OnDeferredInitializationDone(cdm::StreamType stream_type,
                                    cdm::Status decoder_status) final {}

  cdm::FileIO* CreateFileIO(cdm::FileIOClient* client) final {
    return host_->CreateFileIO(client);
  }

  void RequestStorageId(uint32_t version) final {
    host_->RequestStorageId(version);
  }

  void ReportMetrics(cdm::MetricName metric_name, uint64_t value) final {
    host_->ReportMetrics(metric_name, value);
  }

 private:
  const raw_ptr<HostInterface> host_ = nullptr;
};

template <>
void CdmHostProxyImpl<cdm::Host_10>::ReportMetrics(cdm::MetricName metric_name,
                                                   uint64_t value) {
  // cdm::ContentDecryptionModule_10 CDM should never call this.
  NOTIMPLEMENTED();
}

// Specialization for Host_10's OnResolveKeyStatusPromise with KeyStatus_2
template <>
void CdmHostProxyImpl<cdm::Host_10>::OnResolveKeyStatusPromise(
    uint32_t promise_id,
    cdm::KeyStatus_2 key_status) {
  host_->OnResolveKeyStatusPromise(promise_id, ToKeyStatus(key_status));
}

// Specialization for Host_11's OnResolveKeyStatusPromise with KeyStatus_2
template <>
void CdmHostProxyImpl<cdm::Host_11>::OnResolveKeyStatusPromise(
    uint32_t promise_id,
    cdm::KeyStatus_2 key_status) {
  host_->OnResolveKeyStatusPromise(promise_id, ToKeyStatus(key_status));
}

// Specialization for Host_10's OnSessionKeysChange with KeyInformation_2
template <>
void CdmHostProxyImpl<cdm::Host_10>::OnSessionKeysChange(
    const char* session_id,
    uint32_t session_id_size,
    bool has_additional_usable_key,
    const cdm::KeyInformation_2* keys_info,
    uint32_t keys_info_count) {
  std::vector<cdm::KeyInformation> keys_vector =
      ConvertKeyInformation(keys_info, keys_info_count);
  host_->OnSessionKeysChange(session_id, session_id_size,
                             has_additional_usable_key, keys_vector.data(),
                             keys_vector.size());
}

// Specialization for Host_11's OnSessionKeysChange with KeyInformation_2
template <>
void CdmHostProxyImpl<cdm::Host_11>::OnSessionKeysChange(
    const char* session_id,
    uint32_t session_id_size,
    bool has_additional_usable_key,
    const cdm::KeyInformation_2* keys_info,
    uint32_t keys_info_count) {
  std::vector<cdm::KeyInformation> keys_vector =
      ConvertKeyInformation(keys_info, keys_info_count);
  host_->OnSessionKeysChange(session_id, session_id_size,
                             has_additional_usable_key, keys_vector.data(),
                             keys_vector.size());
}

}  // namespace media

#endif  // MEDIA_CDM_LIBRARY_CDM_CDM_HOST_PROXY_IMPL_H_