#include "../Decryptor/Decryptor_u.h"
#include "sgx_eid.h"
#include "sgx_urts.h"
#define __STDC_FORMAT_MACROS
#include <inttypes.h>
#include "dhmsgs.pb.h"
#include "protobufSgxLATransformsInitiator.h"
#include <stdio.h>

int generate_protobuf_dh_msg1(uint32_t own_enclave_id, protobuf_sgx_dh_msg1_t& protobuf_msg1, uint32_t* session_id)
{
  sgx_dh_msg1_t dh_msg1;            //Diffie-Hellman Message 1
  memset(&dh_msg1, 0, sizeof(sgx_dh_msg1_t));
  uint32_t ret_status;
  Decryptor_session_request(own_enclave_id, &ret_status, &dh_msg1, session_id); // TODO: Check Return status
  if(ret_status != SGX_SUCCESS)
  {
    printf("session request returned failure\n");  fflush(stdout); return 0xffffffff;
  }
  encode_msg1_to_protobuf(protobuf_msg1, &dh_msg1);
  print_initialized_msg1(protobuf_msg1, &dh_msg1);
  printf("Done initialization with session id 0x%x", *session_id);
  fflush(stdout);
  return 0;
}

int process_protobuf_dh_msg2_generate_protobuf_dh_msg3(uint32_t own_enclave_id, protobuf_sgx_dh_msg2_t& protobuf_msg2, protobuf_sgx_dh_msg3_t& protobuf_msg3, uint32_t* session_id)
{
  uint32_t ret_status;
  sgx_dh_msg2_t dh_msg2;            //Diffie-Hellman Message 2
  sgx_dh_msg3_t dh_msg3;            //Diffie-Hellman Message 3
  sgx_key_128bit_t dh_aek;        // Session Key
  memset(&dh_aek,0, sizeof(sgx_key_128bit_t));
  memset(&dh_msg2, 0, sizeof(sgx_dh_msg2_t));
  memset(&dh_msg3, 0, sizeof(sgx_dh_msg3_t));

  if(decode_msg2_from_protobuf(protobuf_msg2, &dh_msg2)!=0)
    return -1;
  printf("Done reading and decoding msg2\n");
  // process msg2 and generate msg3
  Decryptor_exchange_report(own_enclave_id, &ret_status, &dh_msg2, &dh_msg3, session_id);
  if(ret_status!=SGX_SUCCESS)
  {
    fflush(stdout);
    printf("exchange report failed:%x\n", ret_status);
    fflush(stdout);
    return -1;
  }
  // convert msg3 sgx_dh_msg3_t object to a protobuf msg3 object.
  encode_msg3_to_protobuf(protobuf_msg3, &dh_msg3);
  printf("about to write msg3\n");
  fflush(stdout);
  return 0;
}