// Copyright 2015 The Chromium OS Authors. All rights reserved.

// Use of this source code is governed by a BSD-style license that can be

// found in the LICENSE file.

// THIS CODE IS GENERATED - DO NOT MODIFY!

#include "MemoryLib_fp.h"

#include "Import_fp.h"

#if IS_CC_ENABLED(Import)

static UINT16 Import_Out_Marshal(Import_Out* source,

                                 TPMI_ST_COMMAND_TAG tag,

                                 BYTE** buffer,

                                 INT32* size) {

  UINT16 total_size = 0;

  UINT32 parameter_size = 0;

  BYTE* parameter_size_location;

  INT32 parameter_size_size = sizeof(UINT32);

  UINT32 num_response_handles = 0;

  // Add parameter_size=0 to indicate size of the parameter area. Will be

  // replaced later by computed parameter_size.

  if (tag == TPM_ST_SESSIONS) {

    parameter_size_location = *buffer;

    // Don't add to total_size, but increment *buffer and decrement *size.

    UINT32_Marshal(&parameter_size, buffer, size);

  }

  // Marshal response parameters.

  total_size += TPM2B_PRIVATE_Marshal(&source->outPrivate, buffer, size);

  // Compute actual parameter_size. Don't add result to total_size.

  if (tag == TPM_ST_SESSIONS) {

    parameter_size = total_size - num_response_handles * sizeof(TPM_HANDLE);

    UINT32_Marshal(&parameter_size, &parameter_size_location,

                   &parameter_size_size);

  }

  return total_size;

}

#endif

static TPM_RC Import_In_Unmarshal(Import_In* target,

                                  TPM_HANDLE request_handles[],

                                  BYTE** buffer,

                                  INT32* size) {

  TPM_RC result = TPM_RC_SUCCESS;

  // Get request handles from request_handles array.

  target->parentHandle = request_handles[0];

  // Unmarshal request parameters.

  result = TPM2B_DATA_Unmarshal(&target->encryptionKey, buffer, size);

  if (result != TPM_RC_SUCCESS) {

    return result;

  }

  result = TPM2B_PUBLIC_Unmarshal(&target->objectPublic, buffer, size);

  if (result != TPM_RC_SUCCESS) {

    return result;

  }

  result = TPM2B_PRIVATE_Unmarshal(&target->duplicate, buffer, size);

  if (result != TPM_RC_SUCCESS) {

    return result;

  }

  result = TPM2B_ENCRYPTED_SECRET_Unmarshal(&target->inSymSeed, buffer, size);

  if (result != TPM_RC_SUCCESS) {

    return result;

  }

  result = TPMT_SYM_DEF_OBJECT_Unmarshal(&target->symmetricAlg, buffer, size);

  if (result != TPM_RC_SUCCESS) {

    return result;

  }

  if ((result == TPM_RC_SUCCESS) && *size) {

    result = TPM_RC_SIZE;

  }

  return result;

}

TPM_RC Exec_Import(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) {

  TPM_RC result = TPM_RC_SUCCESS;

  Import_In in;

  Import_Out out;

#if IS_CC_ENABLED(Import)

  BYTE* response_buffer;

  INT32 response_buffer_size;

  UINT16 bytes_marshalled;

  UINT16 num_response_handles = 0;

#endif

  *response_handle_buffer_size = 0;

  *response_parameter_buffer_size = 0;

  // Unmarshal request parameters to input structure.

  result = Import_In_Unmarshal(&in, request_handles, request_parameter_buffer,

                               request_parameter_buffer_size);

  if (result != TPM_RC_SUCCESS) {

    return result;

  }

  // Execute command.

  result = TPM2_Import(&in, &out);

  if (result != TPM_RC_SUCCESS) {

    return result;

  }

// Marshal output structure to global response buffer.

#if IS_CC_ENABLED(Import)

  response_buffer = MemoryGetResponseBuffer(TPM_CC_Import) + 10;

  response_buffer_size = MAX_RESPONSE_SIZE - 10;

  bytes_marshalled =

      Import_Out_Marshal(&out, tag, &response_buffer, &response_buffer_size);

  *response_handle_buffer_size = num_response_handles * sizeof(TPM_HANDLE);

  *response_parameter_buffer_size =

      bytes_marshalled - *response_handle_buffer_size;

  return TPM_RC_SUCCESS;

#endif

  return TPM_RC_COMMAND_CODE;

}