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 "GetCapability_fp.h"
9 
GetCapability_Out_Marshal(GetCapability_Out * source,TPMI_ST_COMMAND_TAG tag,BYTE ** buffer,INT32 * size)10 UINT16 GetCapability_Out_Marshal(GetCapability_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 += TPMI_YES_NO_Marshal(&source->moreData, buffer, size);
28   total_size +=
29       TPMS_CAPABILITY_DATA_Marshal(&source->capabilityData, 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 
GetCapability_In_Unmarshal(GetCapability_In * target,TPM_HANDLE request_handles[],BYTE ** buffer,INT32 * size)39 TPM_RC GetCapability_In_Unmarshal(GetCapability_In* target,
40                                   TPM_HANDLE request_handles[],
41                                   BYTE** buffer,
42                                   INT32* size) {
43   TPM_RC result = TPM_RC_SUCCESS;
44   // Unmarshal request parameters.
45   result = TPM_CAP_Unmarshal(&target->capability, buffer, size);
46   if (result != TPM_RC_SUCCESS) {
47     return result;
48   }
49   result = UINT32_Unmarshal(&target->property, buffer, size);
50   if (result != TPM_RC_SUCCESS) {
51     return result;
52   }
53   result = UINT32_Unmarshal(&target->propertyCount, 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 
Exec_GetCapability(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)63 TPM_RC Exec_GetCapability(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   GetCapability_In in;
71   GetCapability_Out out;
72 #ifdef TPM_CC_GetCapability
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       GetCapability_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_GetCapability(&in, &out);
89   if (result != TPM_RC_SUCCESS) {
90     return result;
91   }
92 // Marshal output structure to global response buffer.
93 #ifdef TPM_CC_GetCapability
94   response_buffer = MemoryGetResponseBuffer(TPM_CC_GetCapability) + 10;
95   response_buffer_size = MAX_RESPONSE_SIZE - 10;
96   bytes_marshalled = GetCapability_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