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

#ifndef WVCDM_CORE_LICENSE_KEY_STATUS_H_
#define WVCDM_CORE_LICENSE_KEY_STATUS_H_

#include <map>

#include "crypto_session.h"
#include "disallow_copy_and_assign.h"
#include "license_protocol.pb.h"
#include "wv_cdm_types.h"

namespace wvcdm {

class LicenseKeyStatus;

using video_widevine::WidevinePsshData_EntitledKey;

// Holds all content and operator session keys for a session.
class LicenseKeys {
 public:
  LicenseKeys(CdmSecurityLevel security_level)
      : security_level_(security_level) {}
  virtual ~LicenseKeys() { Clear(); }

  virtual bool Empty() { return key_statuses_.empty(); }

  // Returns true if the key is a content key (not an operator session key)
  virtual bool IsContentKey(const KeyId& key_id);

  // Returns true if the key is currently usable for content decryption.
  virtual bool CanDecryptContent(const KeyId& key_id);

  // Returns the allowed usages for a key.
  virtual bool GetAllowedUsage(const KeyId& key_id,
                               CdmKeyAllowedUsage* allowed_usage);

  // Applies a new status to each content key.
  // Returns true if any statuses changed, and sets new_usable_keys to
  // true if the status changes resulted in keys becoming usable.
  virtual bool ApplyStatusChange(CdmKeyStatus new_status,
                                 bool* new_usable_keys);

  // Returns current CdmKeyStatus for the given key.
  // Returns kKeyStatusKeyUnknown if key_id not found.
  virtual CdmKeyStatus GetKeyStatus(const KeyId& key_id);

  // Populates a CdmKeyStatusMap with the current content keys.
  virtual void ExtractKeyStatuses(CdmKeyStatusMap* content_keys);

  // Determines whether the specified key can be used under the current
  // resolution and/or hdcp constraints.  If no constraints have been applied
  // to the key, returns true.
  virtual bool MeetsConstraints(const KeyId& key_id);

  // Indicates whether specified key can be used for the sessions security level
  virtual bool MeetsSecurityLevelConstraints(const KeyId& key_id);

  // Applies a resolution and/or hdcp change to each key, updating their
  // useability under their constraints.
  virtual void ApplyConstraints(uint32_t new_resolution,
                                CryptoSession::HdcpCapability new_hdcp_level);

  // Extracts the keys from a license and makes them available for
  // querying usage and constraint settings.
  virtual void SetFromLicense(const video_widevine::License& license);

  // Sets the keys from the input entitled key data.
  virtual void SetEntitledKeys(
      const std::vector<WidevinePsshData_EntitledKey>& keys);

  // For test use: Sets the OEMCrypto security level
  virtual void SetSecurityLevelForTest(CdmSecurityLevel security_level) {
    security_level_ = security_level;
  }

 private:
  using KeyContainer = ::video_widevine::License::KeyContainer;
  using LicenseKeyStatusIterator =
      std::map<wvcdm::KeyId, LicenseKeyStatus*>::const_iterator;

  void Clear();

  bool is_initialized_;
  // |key_statuses_| can hold either content key statuses, or entitlement key
  // statuses.
  std::map<KeyId, LicenseKeyStatus*> key_statuses_;
  // |content_keyid_to_entitlement_key_id_| maps a content key id to an
  // entitlement_key_id. The resulting key id can be used to obtain the current
  // key status from |key_statuses_| when using entitlement key licensing.
  std::map<KeyId, KeyId> content_keyid_to_entitlement_key_id_;

  CdmSecurityLevel security_level_;

  CORE_DISALLOW_COPY_AND_ASSIGN(LicenseKeys);
};

// Holds the current license status of a key.
class LicenseKeyStatus {
  friend class LicenseKeys;

 public:
  // Returns true if the key is a content key (not an operator session key)
  virtual bool IsContentKey() { return is_content_key_; }

  // Returns true if the key is currently usable for content decryption
  virtual bool CanDecryptContent();

  // Returns the usages allowed for this key.
  virtual bool GetAllowedUsage(CdmKeyAllowedUsage* allowed_usage);

  // Returns the current status of the key.
  virtual CdmKeyStatus GetKeyStatus() const { return key_status_; }

  // Applies a new status to this key.
  // Returns true if the status changed, and sets new_usable_keys to
  // true if the status changes resulted in the key becoming usable.
  virtual bool ApplyStatusChange(CdmKeyStatus new_status,
                                 bool* new_usable_keys);

  // Returns the current constraint status of this key.  The result
  // may change due to calls to ApplyConstraints().
  // Note: this will return true until the first call to ApplyConstraints().
  virtual bool MeetsConstraints() const { return meets_constraints_; }

  // Indicates whether a key can be used for a given security level
  virtual bool MeetsSecurityLevelConstraints() const {
    return meets_security_level_constraints_;
  }

  // Applies the given changes in resolution or HDCP settings.
  virtual void ApplyConstraints(uint32_t new_resolution,
                                CryptoSession::HdcpCapability new_hdcp_level);

 protected:
  using KeyContainer = ::video_widevine::License::KeyContainer;
  using OperatorSessionKeyPermissions =
      KeyContainer::OperatorSessionKeyPermissions;
  using OutputProtection = KeyContainer::OutputProtection;
  using VideoResolutionConstraint = KeyContainer::VideoResolutionConstraint;
  using ConstraintList =
      ::google::protobuf::RepeatedPtrField<VideoResolutionConstraint>;

  LicenseKeyStatus(const KeyContainer& key, const CdmSecurityLevel level);

  virtual ~LicenseKeyStatus() {}

 private:
  void ParseContentKey(const KeyContainer& key, CdmSecurityLevel level);
  void ParseOperatorSessionKey(const KeyContainer& key);

  bool HasConstraints() { return is_content_key_ && constraints_.size() != 0; }

  void SetConstraints(const ConstraintList& constraints);

  bool is_content_key_;
  CdmKeyStatus key_status_;
  bool meets_constraints_;
  bool meets_security_level_constraints_;
  CdmKeyAllowedUsage allowed_usage_;
  CryptoSession::HdcpCapability default_hdcp_level_;
  ConstraintList constraints_;

  CORE_DISALLOW_COPY_AND_ASSIGN(LicenseKeyStatus);
};

}  // namespace wvcdm

#endif  // WVCDM_CORE_LICENSE_KEY_STATUS_H_