////////////////////////////////////////////////////////////////////////////////
// Copyright 2019 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.
////////////////////////////////////////////////////////////////////////////////
//
// Description:
//   Declaration for fake EC key source used for testing.

#ifndef COMMON_FAKE_EC_KEY_SOURCE_H_
#define COMMON_FAKE_EC_KEY_SOURCE_H_

#include <map>
#include <string>

#include "common/ec_key_source.h"
#include "common/ec_test_keys.h"

namespace widevine {

// This is a fake implementation of the ECKeySource.  It returns the values
// specified in ec_test_keys.h.
class FakeECKeySource : public ECKeySource {
 public:
  FakeECKeySource();
  FakeECKeySource(const FakeECKeySource&) = delete;
  FakeECKeySource& operator=(const FakeECKeySource&) = delete;

  // This method allows a caller to specify which keys to return from GetKeys.
  // This overrides the default test keys.
  bool SetKey(ECPrivateKey::EllipticCurve curve, const std::string& private_key,
              const std::string& public_key);

  // Gets precomputed test keys based on |curve|.
  // Parameter |curve| is a standard curve which defines the parameters of the
  // key generation.
  // Parameter |private_key| is the serialized EC private key.
  // Parameter |public_key| is the serialized EC public key.
  // Caller retains ownership of all pointers and they cannot be nullptr.
  // Returns true if successful, false otherwise.
  bool GetECKey(ECPrivateKey::EllipticCurve curve, std::string* private_key,
                std::string* public_key) override;

 private:
  std::map<ECPrivateKey::EllipticCurve, std::pair<std::string, std::string>>
      test_keys_;
};

}  // namespace widevine

#endif  // COMMON_FAKE_EC_KEY_SOURCE_H_