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 "Rewrap_fp.h"
9
10UINT16 Rewrap_Out_Marshal(Rewrap_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_PRIVATE_Marshal(&source->outDuplicate, buffer, size);
28  total_size +=
29      TPM2B_ENCRYPTED_SECRET_Marshal(&source->outSymSeed, buffer, size);
30  // Compute actual parameter_size. Don't add result to total_size.
31  if (tag == TPM_ST_SESSIONS) {
32    parameter_size = total_size - num_response_handles * sizeof(TPM_HANDLE);
33    UINT32_Marshal(&parameter_size, &parameter_size_location,
34                   &parameter_size_size);
35  }
36  return total_size;
37}
38
39TPM_RC Rewrap_In_Unmarshal(Rewrap_In* target,
40                           TPM_HANDLE request_handles[],
41                           BYTE** buffer,
42                           INT32* size) {
43  TPM_RC result = TPM_RC_SUCCESS;
44  // Get request handles from request_handles array.
45  target->oldParent = request_handles[0];
46  target->newParent = request_handles[1];
47  // Unmarshal request parameters.
48  result = TPM2B_PRIVATE_Unmarshal(&target->inDuplicate, buffer, size);
49  if (result != TPM_RC_SUCCESS) {
50    return result;
51  }
52  result = TPM2B_NAME_Unmarshal(&target->name, buffer, size);
53  if (result != TPM_RC_SUCCESS) {
54    return result;
55  }
56  result = TPM2B_ENCRYPTED_SECRET_Unmarshal(&target->inSymSeed, buffer, size);
57  if (result != TPM_RC_SUCCESS) {
58    return result;
59  }
60  if ((result == TPM_RC_SUCCESS) && *size) {
61    result = TPM_RC_SIZE;
62  }
63  return result;
64}
65
66TPM_RC Exec_Rewrap(TPMI_ST_COMMAND_TAG tag,
67                   BYTE** request_parameter_buffer,
68                   INT32* request_parameter_buffer_size,
69                   TPM_HANDLE request_handles[],
70                   UINT32* response_handle_buffer_size,
71                   UINT32* response_parameter_buffer_size) {
72  TPM_RC result = TPM_RC_SUCCESS;
73  Rewrap_In in;
74  Rewrap_Out out;
75#ifdef TPM_CC_Rewrap
76  BYTE* response_buffer;
77  INT32 response_buffer_size;
78  UINT16 bytes_marshalled;
79  UINT16 num_response_handles = 0;
80#endif
81  *response_handle_buffer_size = 0;
82  *response_parameter_buffer_size = 0;
83  // Unmarshal request parameters to input structure.
84  result = Rewrap_In_Unmarshal(&in, request_handles, request_parameter_buffer,
85                               request_parameter_buffer_size);
86  if (result != TPM_RC_SUCCESS) {
87    return result;
88  }
89  // Execute command.
90  result = TPM2_Rewrap(&in, &out);
91  if (result != TPM_RC_SUCCESS) {
92    return result;
93  }
94// Marshal output structure to global response buffer.
95#ifdef TPM_CC_Rewrap
96  response_buffer = MemoryGetResponseBuffer(TPM_CC_Rewrap) + 10;
97  response_buffer_size = MAX_RESPONSE_SIZE - 10;
98  bytes_marshalled =
99      Rewrap_Out_Marshal(&out, tag, &response_buffer, &response_buffer_size);
100  *response_handle_buffer_size = num_response_handles * sizeof(TPM_HANDLE);
101  *response_parameter_buffer_size =
102      bytes_marshalled - *response_handle_buffer_size;
103  return TPM_RC_SUCCESS;
104#endif
105  return TPM_RC_COMMAND_CODE;
106}
107