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 "Sign_fp.h"
9 
Sign_Out_Marshal(Sign_Out * source,TPMI_ST_COMMAND_TAG tag,BYTE ** buffer,INT32 * size)10 UINT16 Sign_Out_Marshal(Sign_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 += TPMT_SIGNATURE_Marshal(&source->signature, 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 
Sign_In_Unmarshal(Sign_In * target,TPM_HANDLE request_handles[],BYTE ** buffer,INT32 * size)37 TPM_RC Sign_In_Unmarshal(Sign_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_DIGEST_Unmarshal(&target->digest, buffer, size);
46   if (result != TPM_RC_SUCCESS) {
47     return result;
48   }
49   result = TPMT_SIG_SCHEME_Unmarshal(&target->inScheme, buffer, size);
50   if (result != TPM_RC_SUCCESS) {
51     return result;
52   }
53   result = TPMT_TK_HASHCHECK_Unmarshal(&target->validation, 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_Sign(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_Sign(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   Sign_In in;
71   Sign_Out out;
72 #ifdef TPM_CC_Sign
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 = Sign_In_Unmarshal(&in, request_handles, request_parameter_buffer,
82                              request_parameter_buffer_size);
83   if (result != TPM_RC_SUCCESS) {
84     return result;
85   }
86   // Execute command.
87   result = TPM2_Sign(&in, &out);
88   if (result != TPM_RC_SUCCESS) {
89     return result;
90   }
91 // Marshal output structure to global response buffer.
92 #ifdef TPM_CC_Sign
93   response_buffer = MemoryGetResponseBuffer(TPM_CC_Sign) + 10;
94   response_buffer_size = MAX_RESPONSE_SIZE - 10;
95   bytes_marshalled =
96       Sign_Out_Marshal(&out, tag, &response_buffer, &response_buffer_size);
97   *response_handle_buffer_size = num_response_handles * sizeof(TPM_HANDLE);
98   *response_parameter_buffer_size =
99       bytes_marshalled - *response_handle_buffer_size;
100   return TPM_RC_SUCCESS;
101 #endif
102   return TPM_RC_COMMAND_CODE;
103 }
104