1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192 |
- uint32_t process_protobuf_dh_msg1_generate_protobuf_dh_msg2(protobuf_sgx_dh_msg1_t& protobuf_msg1, protobuf_sgx_dh_msg2_t& protobuf_msg2)
- {
- sgx_dh_msg1_t dh_msg1; //Diffie-Hellman Message 1
- sgx_dh_msg2_t dh_msg2;
- memset(&dh_msg1, 0, sizeof(sgx_dh_msg1_t));
- uint32_t ret_status;
- if(Initiator_Transforms::decode_msg1_from_protobuf(protobuf_msg1, &dh_msg1)!=0)
- return -1;
- //Intialize the session as a session initiator
- ret_status = sgx_dh_init_session(SGX_DH_SESSION_INITIATOR, &sgx_dh_session);
- if(ret_status != SGX_SUCCESS)
- return ret_status;
- //Process the message 1 obtained from desination enclave and generate message 2
- ret_status = sgx_dh_initiator_proc_msg1(&dh_msg1, &dh_msg2, &sgx_dh_session);
- if(ret_status != SGX_SUCCESS)
- return ret_status;
- Initiator_Transforms::encode_msg2_to_protobuf(protobuf_msg2, &dh_msg2);
- return 0;
- }
- uint32_t process_protobuf_dh_msg3(protobuf_sgx_dh_msg3_t& protobuf_msg3) {
- uint32_t ret_status;
- sgx_dh_msg3_t dh_msg3;
- sgx_key_128bit_t dh_aek; // Session Key
- sgx_dh_session_enclave_identity_t responder_identity;
- memset(&dh_aek,0, sizeof(sgx_key_128bit_t));
- if(Initiator_Transforms::decode_msg3_from_protobuf(protobuf_msg3, &dh_msg3)!=0)
- return -1;
- //Process Message 3 obtained from the destination enclave
- ret_status = sgx_dh_initiator_proc_msg3(&dh_msg3, &sgx_dh_session, &dh_aek, &responder_identity);
- if(SGX_SUCCESS != ret_status)
- return ret_status;
- memcpy(global_session_info.active.AEK, &dh_aek, sizeof(sgx_key_128bit_t));
- global_session_info.session_id = 1; // TODO: session_id;
- global_session_info.active.counter = 0;
- global_session_info.status = ACTIVE;
- memset(&dh_aek,0, sizeof(sgx_key_128bit_t));
- return 0;
- }
- uint32_t conduct_la(int decryptor_fd) {
- // declare msg1, msg2, msg3 protobuf objects
- protobuf_sgx_dh_msg1_t protobuf_msg1;
- protobuf_sgx_dh_msg2_t protobuf_msg2;
- protobuf_sgx_dh_msg3_t protobuf_msg3;
- uint32_t protobuf_sgx_ret;
- setbuf(stdout,NULL);
- protobufReaderWriter.set_fd(decryptor_fd);
- if(protobufReaderWriter.read_msg(protobuf_msg1)!=0)
- return -1;
- protobuf_sgx_ret = process_protobuf_dh_msg1_generate_protobuf_dh_msg2(protobuf_msg1, protobuf_msg2);
- if(protobuf_sgx_ret != 0)
- {
- printf("Error in process_protobuf_dh_msg1_generate_protobuf_dh_msg2: 0x%x", protobuf_sgx_ret); fflush(stdout); return protobuf_sgx_ret;
- }
- if(protobufReaderWriter.write_msg(protobuf_msg2)!=0)
- return -1;
- if(protobufReaderWriter.read_msg(protobuf_msg3)!=0)
- return -1;
- protobuf_sgx_ret = process_protobuf_dh_msg3(protobuf_msg3);
- if(protobuf_sgx_ret != 0)
- {
- printf("Error in process_protobuf_dh_msg3: 0x%x", protobuf_sgx_ret); fflush(stdout); return protobuf_sgx_ret;
- }
- return 0;
- }
- uint32_t get_la_symmetric_key(uint8_t* op_key)
- {
- uint32_t counter;
- for(counter=0; counter<32; counter++)
- op_key[counter] = key[counter];
- // TODO: Fix this code - there is no attribute called key - it's in one of the existing attributes.
- }
|