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 "RSA_Decrypt_fp.h"
9
10UINT16 RSA_Decrypt_Out_Marshal(RSA_Decrypt_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_PUBLIC_KEY_RSA_Marshal(&source->message, 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
37TPM_RC RSA_Decrypt_In_Unmarshal(RSA_Decrypt_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_PUBLIC_KEY_RSA_Unmarshal(&target->cipherText, buffer, size);
46  if (result != TPM_RC_SUCCESS) {
47    return result;
48  }
49  result = TPMT_RSA_DECRYPT_Unmarshal(&target->inScheme, buffer, size);
50  if (result != TPM_RC_SUCCESS) {
51    return result;
52  }
53  result = TPM2B_DATA_Unmarshal(&target->label, buffer, size);
54  if (result != TPM_RC_SUCCESS) {
55    return result;
56  }
57  if ((result == TPM_RC_SUCCESS) && *size) {
58    result = TPM_RC_SIZE;
59  }
60  return result;
61}
62
63TPM_RC Exec_RSA_Decrypt(TPMI_ST_COMMAND_TAG tag,
64                        BYTE** request_parameter_buffer,
65                        INT32* request_parameter_buffer_size,
66                        TPM_HANDLE request_handles[],
67                        UINT32* response_handle_buffer_size,
68                        UINT32* response_parameter_buffer_size) {
69  TPM_RC result = TPM_RC_SUCCESS;
70  RSA_Decrypt_In in;
71  RSA_Decrypt_Out out;
72#ifdef TPM_CC_RSA_Decrypt
73  BYTE* response_buffer;
74  INT32 response_buffer_size;
75  UINT16 bytes_marshalled;
76  UINT16 num_response_handles = 0;
77#endif
78  *response_handle_buffer_size = 0;
79  *response_parameter_buffer_size = 0;
80  // Unmarshal request parameters to input structure.
81  result =
82      RSA_Decrypt_In_Unmarshal(&in, request_handles, request_parameter_buffer,
83                               request_parameter_buffer_size);
84  if (result != TPM_RC_SUCCESS) {
85    return result;
86  }
87  // Execute command.
88  result = TPM2_RSA_Decrypt(&in, &out);
89  if (result != TPM_RC_SUCCESS) {
90    return result;
91  }
92// Marshal output structure to global response buffer.
93#ifdef TPM_CC_RSA_Decrypt
94  response_buffer = MemoryGetResponseBuffer(TPM_CC_RSA_Decrypt) + 10;
95  response_buffer_size = MAX_RESPONSE_SIZE - 10;
96  bytes_marshalled = RSA_Decrypt_Out_Marshal(&out, tag, &response_buffer,
97                                             &response_buffer_size);
98  *response_handle_buffer_size = num_response_handles * sizeof(TPM_HANDLE);
99  *response_parameter_buffer_size =
100      bytes_marshalled - *response_handle_buffer_size;
101  return TPM_RC_SUCCESS;
102#endif
103  return TPM_RC_COMMAND_CODE;
104}
105