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_ZGen_fp.h"
9 
ECDH_ZGen_Out_Marshal(ECDH_ZGen_Out * source,TPMI_ST_COMMAND_TAG tag,BYTE ** buffer,INT32 * size)10 UINT16 ECDH_ZGen_Out_Marshal(ECDH_ZGen_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->outPoint, buffer, size);
28   // Compute actual parameter_size. Don't add result to total_size.
29   if (tag == TPM_ST_SESSIONS) {
30     parameter_size = total_size - num_response_handles * sizeof(TPM_HANDLE);
31     UINT32_Marshal(&parameter_size, &parameter_size_location,
32                    &parameter_size_size);
33   }
34   return total_size;
35 }
36 
ECDH_ZGen_In_Unmarshal(ECDH_ZGen_In * target,TPM_HANDLE request_handles[],BYTE ** buffer,INT32 * size)37 TPM_RC ECDH_ZGen_In_Unmarshal(ECDH_ZGen_In* target,
38                               TPM_HANDLE request_handles[],
39                               BYTE** buffer,
40                               INT32* size) {
41   TPM_RC result = TPM_RC_SUCCESS;
42   // Get request handles from request_handles array.
43   target->keyHandle = request_handles[0];
44   // Unmarshal request parameters.
45   result = TPM2B_ECC_POINT_Unmarshal(&target->inPoint, buffer, size);
46   if (result != TPM_RC_SUCCESS) {
47     return result;
48   }
49   if ((result == TPM_RC_SUCCESS) && *size) {
50     result = TPM_RC_SIZE;
51   }
52   return result;
53 }
54 
Exec_ECDH_ZGen(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)55 TPM_RC Exec_ECDH_ZGen(TPMI_ST_COMMAND_TAG tag,
56                       BYTE** request_parameter_buffer,
57                       INT32* request_parameter_buffer_size,
58                       TPM_HANDLE request_handles[],
59                       UINT32* response_handle_buffer_size,
60                       UINT32* response_parameter_buffer_size) {
61   TPM_RC result = TPM_RC_SUCCESS;
62   ECDH_ZGen_In in;
63   ECDH_ZGen_Out out;
64 #ifdef TPM_CC_ECDH_ZGen
65   BYTE* response_buffer;
66   INT32 response_buffer_size;
67   UINT16 bytes_marshalled;
68   UINT16 num_response_handles = 0;
69 #endif
70   *response_handle_buffer_size = 0;
71   *response_parameter_buffer_size = 0;
72   // Unmarshal request parameters to input structure.
73   result =
74       ECDH_ZGen_In_Unmarshal(&in, request_handles, request_parameter_buffer,
75                              request_parameter_buffer_size);
76   if (result != TPM_RC_SUCCESS) {
77     return result;
78   }
79   // Execute command.
80   result = TPM2_ECDH_ZGen(&in, &out);
81   if (result != TPM_RC_SUCCESS) {
82     return result;
83   }
84 // Marshal output structure to global response buffer.
85 #ifdef TPM_CC_ECDH_ZGen
86   response_buffer = MemoryGetResponseBuffer(TPM_CC_ECDH_ZGen) + 10;
87   response_buffer_size = MAX_RESPONSE_SIZE - 10;
88   bytes_marshalled =
89       ECDH_ZGen_Out_Marshal(&out, tag, &response_buffer, &response_buffer_size);
90   *response_handle_buffer_size = num_response_handles * sizeof(TPM_HANDLE);
91   *response_parameter_buffer_size =
92       bytes_marshalled - *response_handle_buffer_size;
93   return TPM_RC_SUCCESS;
94 #endif
95   return TPM_RC_COMMAND_CODE;
96 }
97