// 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_