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
10UINT16 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
38TPM_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
51TPM_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