1234567891011121314151617181920212223242526272829303132333435363738394041424344 |
- #include "protobufLAInitiator.h"
- namespace protobufLAInitiator
- {
- 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_wrapper(own_enclave_id, &ret_status, &dh_msg1, session_id);
- if(ret_status != SGX_SUCCESS)
- return 0xffffffff;
- protobufSgxTransformsInitiator::encode_msg1_to_protobuf(protobuf_msg1, &dh_msg1);
- printf("Generated the following Msg1: ------------- \n"); fflush(stdout);
- protobufSgxTransformsInitiator::print_initialized_msg1(protobuf_msg1, &dh_msg1);
- 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(protobufSgxTransformsInitiator::decode_msg2_from_protobuf(protobuf_msg2, &dh_msg2)!=0)
- return -1;
- // process msg2 and generate msg3
- Decryptor_exchange_report_wrapper(own_enclave_id, &ret_status, &dh_msg2, &dh_msg3, session_id);
- if(ret_status!=SGX_SUCCESS)
- return ret_status;
- // convert msg3 sgx_dh_msg3_t object to a protobuf msg3 object.
- protobufSgxTransformsInitiator::encode_msg3_to_protobuf(protobuf_msg3, &dh_msg3);
- return 0;
- }
- };
|