From 9318b08eb51fbd780a16206009284efc63f3d201 Mon Sep 17 00:00:00 2001
From: jyao1 <jyao1>
Date: Mon, 21 Oct 2013 05:56:07 +0000
Subject: [PATCH] Add TPM1.2 SaveState API.

Signed off by: Yao, Jiewen <Jiewen.yao@intel.com>
Reviewed by: Dong Guo <Gui.dong@intel.com>


git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@14791 6f19259b-4bc3-4df7-8a09-765794883524
---
 SecurityPkg/Include/Library/Tpm12CommandLib.h | 12 +++++
 .../Library/Tpm12CommandLib/Tpm12Startup.c    | 51 ++++++++++++++++++-
 2 files changed, 62 insertions(+), 1 deletion(-)

diff --git a/SecurityPkg/Include/Library/Tpm12CommandLib.h b/SecurityPkg/Include/Library/Tpm12CommandLib.h
index 8b62823716..1dbc7d50c3 100644
--- a/SecurityPkg/Include/Library/Tpm12CommandLib.h
+++ b/SecurityPkg/Include/Library/Tpm12CommandLib.h
@@ -31,6 +31,18 @@ Tpm12Startup (
   IN TPM_STARTUP_TYPE          TpmSt
   );
 
+/**
+  Send SaveState command to TPM1.2.
+
+  @retval EFI_SUCCESS      Operation completed successfully.
+  @retval EFI_DEVICE_ERROR Unexpected device behavior.
+**/
+EFI_STATUS
+EFIAPI
+Tpm12SaveState (
+  VOID
+  );
+
 /**
   Send ForceClear command to TPM1.2.
 
diff --git a/SecurityPkg/Library/Tpm12CommandLib/Tpm12Startup.c b/SecurityPkg/Library/Tpm12CommandLib/Tpm12Startup.c
index 684c9b523e..afbe02e382 100644
--- a/SecurityPkg/Library/Tpm12CommandLib/Tpm12Startup.c
+++ b/SecurityPkg/Library/Tpm12CommandLib/Tpm12Startup.c
@@ -29,6 +29,14 @@ typedef struct {
   TPM_RSP_COMMAND_HDR   Hdr;
 } TPM_RSP_START_UP;
 
+typedef struct {
+  TPM_RQU_COMMAND_HDR   Hdr;
+} TPM_CMD_SAVE_STATE;
+
+typedef struct {
+  TPM_RSP_COMMAND_HDR   Hdr;
+} TPM_RSP_SAVE_STATE;
+
 #pragma pack()
 
 /**
@@ -75,4 +83,45 @@ Tpm12Startup (
   default:
     return EFI_DEVICE_ERROR;
   }
-}
\ No newline at end of file
+}
+
+/**
+  Send SaveState command to TPM1.2.
+
+  @retval EFI_SUCCESS      Operation completed successfully.
+  @retval EFI_DEVICE_ERROR Unexpected device behavior.
+**/
+EFI_STATUS
+EFIAPI
+Tpm12SaveState (
+  VOID
+  )
+{
+  EFI_STATUS                        Status;
+  UINT32                            TpmRecvSize;
+  UINT32                            TpmSendSize;
+  TPM_CMD_SAVE_STATE                SendBuffer;
+  TPM_RSP_SAVE_STATE                RecvBuffer;
+  UINT32                            ReturnCode;
+
+  //
+  // send Tpm command TPM_ORD_SaveState
+  //
+  TpmRecvSize               = sizeof (TPM_RSP_SAVE_STATE);
+  TpmSendSize               = sizeof (TPM_CMD_SAVE_STATE);
+  SendBuffer.Hdr.tag        = SwapBytes16 (TPM_TAG_RQU_COMMAND);
+  SendBuffer.Hdr.paramSize  = SwapBytes32 (TpmSendSize);
+  SendBuffer.Hdr.ordinal    = SwapBytes32 (TPM_ORD_SaveState);
+
+  Status = Tpm12SubmitCommand (TpmSendSize, (UINT8 *)&SendBuffer, &TpmRecvSize, (UINT8 *)&RecvBuffer);
+  if (EFI_ERROR (Status)) {
+    return Status;
+  }
+  ReturnCode = SwapBytes32(RecvBuffer.Hdr.returnCode);
+  switch (ReturnCode) {
+  case TPM_SUCCESS:
+    return EFI_SUCCESS;
+  default:
+    return EFI_DEVICE_ERROR;
+  }
+}