cpix_key_source.h

// Copyright 2026 Google LLC. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd
 
#ifndef PACKAGER_MEDIA_BASE_CPIX_KEY_SOURCE_H_
#define PACKAGER_MEDIA_BASE_CPIX_KEY_SOURCE_H_
 
#include <cstdint>
#include <memory>
#include <string>
#include <vector>
 
#include <packager/crypto_params.h>
#include <packager/media/base/key_source.h>
 
namespace shaka {
namespace media {
 
class CpixFetcher {
 public:
  virtual ~CpixFetcher() = default;
 
  virtual Status Fetch(const std::string& url,
                       const std::string& request_body,
                       const std::vector<std::string>& headers,
                       std::string* response) = 0;
};
 
class CpixKeySource : public KeySource {
 public:
  ~CpixKeySource() override;
 
  Status FetchKeys(EmeInitDataType init_data_type,
                   const std::vector<uint8_t>& init_data) override;
  Status GetKey(const std::string& stream_label, EncryptionKey* key) override;
  Status GetKey(const std::vector<uint8_t>& key_id,
                EncryptionKey* key) override;
  Status GetCryptoPeriodKey(uint32_t crypto_period_index,
                            int32_t crypto_period_duration_in_seconds,
                            const std::string& stream_label,
                            EncryptionKey* key) override;
 
  static std::unique_ptr<CpixKeySource> Create(
      const CpixEncryptionParams& cpix_params);
 
  static std::unique_ptr<CpixKeySource> CreateWithFetcher(
      const CpixEncryptionParams& cpix_params,
      CpixFetcher* fetcher);
 
  static std::unique_ptr<CpixKeySource> CreateForDecryption(
      const CpixEncryptionParams& cpix_params);
 
 private:
  static std::unique_ptr<CpixKeySource> CreateInternal(
      const CpixEncryptionParams& cpix_params,
      CpixFetcher* fetcher,
      bool for_decryption);
 
  explicit CpixKeySource(EncryptionKeyMap&& encryption_key_map);
  CpixKeySource(const CpixKeySource&) = delete;
  CpixKeySource& operator=(const CpixKeySource&) = delete;
 
  EncryptionKeyMap encryption_key_map_;
};
 
}  // namespace media
}  // namespace shaka
 
#endif  // PACKAGER_MEDIA_BASE_CPIX_KEY_SOURCE_H_