5a6a2075f5
Merge of http://go/wvgerrit/95666 Mostly fixing coding styles and a few vulnerability check. Updating tests according to the fix. Bug: 150614088 Bug: 150881959 Test: Ran cdm and odk unit tests Change-Id: I109a96ee8ded089d59ab49c2f94b6833c932fd1e
100 lines
3.0 KiB
C++
100 lines
3.0 KiB
C++
/* Copyright 2019 Google LLC. All rights reserved. This file and proprietary */
|
|
/* source code may only be used and distributed under the Widevine Master */
|
|
/* License Agreement. */
|
|
|
|
#ifndef WIDEVINE_ODK_TEST_ODK_TEST_HELPER_H_
|
|
#define WIDEVINE_ODK_TEST_ODK_TEST_HELPER_H_
|
|
|
|
#include <cstdint>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "odk_structs.h"
|
|
#include "odk_structs_priv.h"
|
|
|
|
namespace wvodk_test {
|
|
|
|
enum ODK_FieldType {
|
|
ODK_UINT16,
|
|
ODK_UINT32,
|
|
ODK_UINT64,
|
|
ODK_SUBSTRING,
|
|
ODK_DEVICEID,
|
|
ODK_HASH,
|
|
ODK_NUMTYPES,
|
|
};
|
|
|
|
enum ODK_FieldMode {
|
|
ODK_READ,
|
|
ODK_WRITE,
|
|
ODK_DUMP,
|
|
};
|
|
|
|
struct ODK_Field {
|
|
ODK_FieldType type;
|
|
void* value;
|
|
std::string name;
|
|
};
|
|
|
|
struct ODK_LicenseResponseParams {
|
|
ODK_CoreMessage core_message;
|
|
bool initial_license_load;
|
|
bool usage_entry_present;
|
|
uint8_t request_hash[ODK_SHA256_HASH_SIZE];
|
|
ODK_TimerLimits timer_limits;
|
|
ODK_ClockValues clock_values;
|
|
ODK_ParsedLicense parsed_license;
|
|
std::vector<ODK_Field> extra_fields;
|
|
};
|
|
|
|
struct ODK_RenewalResponseParams {
|
|
ODK_CoreMessage core_message;
|
|
uint64_t system_time;
|
|
uint64_t playback_clock;
|
|
uint64_t renewal_duration;
|
|
ODK_TimerLimits timer_limits;
|
|
ODK_ClockValues clock_values;
|
|
uint64_t playback_timer;
|
|
std::vector<ODK_Field> extra_fields;
|
|
};
|
|
|
|
struct ODK_ProvisioningResponseParams {
|
|
ODK_CoreMessage core_message;
|
|
uint8_t device_id[ODK_DEVICE_ID_LEN_MAX];
|
|
uint32_t device_id_length;
|
|
ODK_ParsedProvisioning parsed_provisioning;
|
|
std::vector<ODK_Field> extra_fields;
|
|
};
|
|
|
|
/* Default values in core_message for testing */
|
|
void ODK_SetDefaultCoreFields(ODK_CoreMessage* core_message,
|
|
uint32_t message_type);
|
|
void ODK_SetDefaultLicenseResponseParams(ODK_LicenseResponseParams* params);
|
|
void ODK_SetDefaultRenewalResponseParams(ODK_RenewalResponseParams* params);
|
|
void ODK_SetDefaultProvisioningResponseParams(
|
|
ODK_ProvisioningResponseParams* params);
|
|
|
|
size_t ODK_FieldLength(ODK_FieldType type);
|
|
size_t ODK_AllocSize(ODK_FieldType type);
|
|
|
|
/* Copy ODK_Field to buf */
|
|
OEMCryptoResult ODK_WriteSingleField(uint8_t* buf, const ODK_Field* field);
|
|
/* Load buf to ODK_Field */
|
|
OEMCryptoResult ODK_ReadSingleField(const uint8_t* buf, const ODK_Field* field);
|
|
OEMCryptoResult ODK_DumpSingleField(const uint8_t* buf, const ODK_Field* field);
|
|
OEMCryptoResult ODK_IterFields(ODK_FieldMode mode, uint8_t* buf,
|
|
const size_t size_in, size_t* size_out,
|
|
const std::vector<ODK_Field>& fields);
|
|
void ODK_ExpectEqualBuf(const void* s1, const void* s2, size_t n,
|
|
const std::vector<ODK_Field>& fields);
|
|
void ODK_ResetOdkFields(std::vector<ODK_Field>* fields);
|
|
|
|
/* Serialize core_message and extra_fields into buf */
|
|
void ODK_BuildMessageBuffer(ODK_CoreMessage* core_message,
|
|
const std::vector<ODK_Field>& extra_fields,
|
|
uint8_t** buf, uint32_t* buf_size);
|
|
|
|
} /* namespace wvodk_test */
|
|
|
|
#endif /* WIDEVINE_ODK_TEST_ODK_TEST_HELPER_H_ */
|