////////////////////////////////////////////////////////////////////////////////
// Copyright 2018 Google LLC.
//
// This software is licensed under the terms defined in the Widevine Master
// License Agreement. For a copy of this agreement, please contact
// widevine-licensing@google.com.
////////////////////////////////////////////////////////////////////////////////

// Example of how to use the wv_cas_ecm library.

#include <iostream>
#include <string>

#include "media_cas_packager_sdk/public/wv_cas_ecm.h"
#include "media_cas_packager_sdk/public/wv_cas_types.h"

const char kCsaEvenKey[] = "even_key";                    // 8 bytes
const char kEvenContentIv8Bytes[] = "even_iv.";           // 8 bytes
const char kEvenEntitlementKeyId[] = "fake_key_id1....";  // 16 bytes
const char kEvenEntitlementKey[] =
    "fakefakefakefakefakefakefake1...";                  // 32 bytes
const char kCsaOddKey[] = "odd_key.";                    // 8 bytes
const char kOddContentIv8Bytes[] = "odd_iv..";           // 8 bytes
const char kOddEntitlementKeyId[] = "fake_key_id2....";  // 16 bytes
const char kOddEntitlementKey[] =
    "fakefakefakefakefakefakefake2...";  // 32 bytes

int main(int argc, char **argv) {
  widevine::cas::WvCasEcm wv_cas_ecm;
  widevine::cas::WvCasStatus status = wv_cas_ecm.Initialize(
      /* content_iv_size= */ 8, /* key_rotation_enabled= */ true,
      widevine::cas::CryptoMode::kDvbCsa2);
  if (status != widevine::cas::OK) {
    std::cerr << "Failed to initialize WV CAS ECM, error: "
              << widevine::cas::GetWvCasStatusMessage(status)
              << std::endl;
  }
  std::string ecm;
  status = wv_cas_ecm.GenerateEcm(
      kCsaEvenKey, kEvenContentIv8Bytes, kEvenEntitlementKeyId,
      kEvenEntitlementKey, kCsaOddKey, kOddContentIv8Bytes,
      kOddEntitlementKeyId, kOddEntitlementKey, &ecm);
  if (status != widevine::cas::OK) {
    std::cerr << "Failed to generate WV CAS ECM, error: "
              << widevine::cas::GetWvCasStatusMessage(status)
              << std::endl;
  } else {
    std::cout << "ECM size: " << ecm.size() << std::endl;
    std::cout << "ECM bytes: ";
    for (size_t i = 0; i < ecm.size(); i++) {
      printf("'\\x%02x', ", static_cast<uint16_t>(ecm.at(i)));
    }
    std::cout << std::endl;
  }

  return 0;
}