1 // Copyright 2015 The Chromium OS Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4 
5 // THIS CODE IS GENERATED - DO NOT MODIFY!
6 
7 #include "MemoryLib_fp.h"
8 #include "ECDH_KeyGen_fp.h"
9 
ECDH_KeyGen_Out_Marshal(ECDH_KeyGen_Out * source,TPMI_ST_COMMAND_TAG tag,BYTE ** buffer,INT32 * size)10 UINT16 ECDH_KeyGen_Out_Marshal(ECDH_KeyGen_Out* source,
11                                TPMI_ST_COMMAND_TAG tag,
12                                BYTE** buffer,
13                                INT32* size) {
14   UINT16 total_size = 0;
15   UINT32 parameter_size = 0;
16   BYTE* parameter_size_location;
17   INT32 parameter_size_size = sizeof(UINT32);
18   UINT32 num_response_handles = 0;
19   // Add parameter_size=0 to indicate size of the parameter area. Will be
20   // replaced later by computed parameter_size.
21   if (tag == TPM_ST_SESSIONS) {
22     parameter_size_location = *buffer;
23     // Don't add to total_size, but increment *buffer and decrement *size.
24     UINT32_Marshal(&parameter_size, buffer, size);
25   }
26   // Marshal response parameters.
27   total_size += TPM2B_ECC_POINT_Marshal(&source->zPoint, buffer, size);
28   total_size += TPM2B_ECC_POINT_Marshal(&source->pubPoint, buffer, size);
29   // Compute actual parameter_size. Don't add result to total_size.
30   if (tag == TPM_ST_SESSIONS) {
31     parameter_size = total_size - num_response_handles * sizeof(TPM_HANDLE);
32     UINT32_Marshal(&parameter_size, &parameter_size_location,
33                    &parameter_size_size);
34   }
35   return total_size;
36 }
37 
ECDH_KeyGen_In_Unmarshal(ECDH_KeyGen_In * target,TPM_HANDLE request_handles[],BYTE ** buffer,INT32 * size)38 TPM_RC ECDH_KeyGen_In_Unmarshal(ECDH_KeyGen_In* target,
39                                 TPM_HANDLE request_handles[],
40                                 BYTE** buffer,
41                                 INT32* size) {
42   TPM_RC result = TPM_RC_SUCCESS;
43   // Get request handles from request_handles array.
44   target->keyHandle = request_handles[0];
45   if ((result == TPM_RC_SUCCESS) && *size) {
46     result = TPM_RC_SIZE;
47   }
48   return result;
49 }
50 
Exec_ECDH_KeyGen(TPMI_ST_COMMAND_TAG tag,BYTE ** request_parameter_buffer,INT32 * request_parameter_buffer_size,TPM_HANDLE request_handles[],UINT32 * response_handle_buffer_size,UINT32 * response_parameter_buffer_size)51 TPM_RC Exec_ECDH_KeyGen(TPMI_ST_COMMAND_TAG tag,
52                         BYTE** request_parameter_buffer,
53                         INT32* request_parameter_buffer_size,
54                         TPM_HANDLE request_handles[],
55                         UINT32* response_handle_buffer_size,
56                         UINT32* response_parameter_buffer_size) {
57   TPM_RC result = TPM_RC_SUCCESS;
58   ECDH_KeyGen_In in;
59   ECDH_KeyGen_Out out;
60 #ifdef TPM_CC_ECDH_KeyGen
61   BYTE* response_buffer;
62   INT32 response_buffer_size;
63   UINT16 bytes_marshalled;
64   UINT16 num_response_handles = 0;
65 #endif
66   *response_handle_buffer_size = 0;
67   *response_parameter_buffer_size = 0;
68   // Unmarshal request parameters to input structure.
69   result =
70       ECDH_KeyGen_In_Unmarshal(&in, request_handles, request_parameter_buffer,
71                                request_parameter_buffer_size);
72   if (result != TPM_RC_SUCCESS) {
73     return result;
74   }
75   // Execute command.
76   result = TPM2_ECDH_KeyGen(&in, &out);
77   if (result != TPM_RC_SUCCESS) {
78     return result;
79   }
80 // Marshal output structure to global response buffer.
81 #ifdef TPM_CC_ECDH_KeyGen
82   response_buffer = MemoryGetResponseBuffer(TPM_CC_ECDH_KeyGen) + 10;
83   response_buffer_size = MAX_RESPONSE_SIZE - 10;
84   bytes_marshalled = ECDH_KeyGen_Out_Marshal(&out, tag, &response_buffer,
85                                              &response_buffer_size);
86   *response_handle_buffer_size = num_response_handles * sizeof(TPM_HANDLE);
87   *response_parameter_buffer_size =
88       bytes_marshalled - *response_handle_buffer_size;
89   return TPM_RC_SUCCESS;
90 #endif
91   return TPM_RC_COMMAND_CODE;
92 }
93