54 lines
1.9 KiB
C++
54 lines
1.9 KiB
C++
////////////////////////////////////////////////////////////////////////////////
|
|
// 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_
|