]> git.proxmox.com Git - mirror_edk2.git/blobdiff - SourceLevelDebugPkg/Library/DebugAgent/DebugAgentCommon/DebugAgent.c
Refine code to make it more safely.
[mirror_edk2.git] / SourceLevelDebugPkg / Library / DebugAgent / DebugAgentCommon / DebugAgent.c
index 7ee4cca43b6303faa450a76135def5b7f45c9f55..48bc38ca547cbb6fa1eb47477c23962899bde3b1 100644 (file)
@@ -4,7 +4,7 @@
   read/write debug packet to communication with HOST based on transfer\r
   protocol.\r
 \r
-  Copyright (c) 2010 - 2012, Intel Corporation. All rights reserved.<BR>\r
+  Copyright (c) 2010 - 2014, Intel Corporation. All rights reserved.<BR>\r
   This program and the accompanying materials\r
   are licensed and made available under the terms and conditions of the BSD License\r
   which accompanies this distribution.  The full text of the license may be found at\r
 #include "DebugAgent.h"\r
 #include "Ia32/DebugException.h"\r
 \r
-#define INIT_BREAK_ACK_TIMEOUT  (200 * 1000)\r
+CHAR8 mErrorMsgVersionAlert[]       = "\rThe SourceLevelDebugPkg you are using requires a newer version of the Intel(R) UDK Debugger Tool.\r\n";\r
+CHAR8 mErrorMsgSendInitPacket[]     = "\rSend INIT break packet and try to connect the HOST (Intel(R) UDK Debugger Tool v1.4) ...\r\n";\r
+CHAR8 mErrorMsgConnectOK[]          = "HOST connection is successful!\r\n";\r
+CHAR8 mErrorMsgConnectFail[]        = "HOST connection is failed!\r\n";\r
+CHAR8 mWarningMsgIngoreBreakpoint[] = "Ignore break point in SMM for SMI issued during DXE debugging!\r\n";\r
+\r
+//\r
+// Vector Handoff Info list used by Debug Agent for persist\r
+//\r
+EFI_VECTOR_HANDOFF_INFO mVectorHandoffInfoDebugAgent[] = {\r
+  {\r
+    DEBUG_EXCEPT_DIVIDE_ERROR,         // Vector 0\r
+    EFI_VECTOR_HANDOFF_HOOK_BEFORE,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_EXCEPT_DEBUG,                // Vector 1\r
+    EFI_VECTOR_HANDOFF_DO_NOT_HOOK,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_EXCEPT_NMI,                  // Vector 2\r
+    EFI_VECTOR_HANDOFF_HOOK_BEFORE,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_EXCEPT_BREAKPOINT,           // Vector 3\r
+    EFI_VECTOR_HANDOFF_DO_NOT_HOOK,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_EXCEPT_OVERFLOW,             // Vector 4\r
+    EFI_VECTOR_HANDOFF_HOOK_BEFORE,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_EXCEPT_BOUND,                // Vector 5\r
+    EFI_VECTOR_HANDOFF_HOOK_BEFORE,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_EXCEPT_INVALID_OPCODE,       // Vector 6\r
+    EFI_VECTOR_HANDOFF_HOOK_BEFORE,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_EXCEPT_DOUBLE_FAULT,         // Vector 8\r
+    EFI_VECTOR_HANDOFF_HOOK_BEFORE,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_EXCEPT_INVALID_TSS,          // Vector 10\r
+    EFI_VECTOR_HANDOFF_HOOK_BEFORE,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_EXCEPT_SEG_NOT_PRESENT,      // Vector 11\r
+    EFI_VECTOR_HANDOFF_HOOK_BEFORE,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_EXCEPT_STACK_FAULT,          // Vector 12\r
+    EFI_VECTOR_HANDOFF_HOOK_BEFORE,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_EXCEPT_GP_FAULT,             // Vector 13\r
+    EFI_VECTOR_HANDOFF_HOOK_BEFORE,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+    {\r
+    DEBUG_EXCEPT_PAGE_FAULT,           // Vector 14\r
+    EFI_VECTOR_HANDOFF_HOOK_BEFORE,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_EXCEPT_FP_ERROR,             // Vector 16\r
+    EFI_VECTOR_HANDOFF_HOOK_BEFORE,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_EXCEPT_ALIGNMENT_CHECK,      // Vector 17\r
+    EFI_VECTOR_HANDOFF_HOOK_BEFORE,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_EXCEPT_MACHINE_CHECK,        // Vector 18\r
+    EFI_VECTOR_HANDOFF_HOOK_BEFORE,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_EXCEPT_SIMD,                 // Vector 19\r
+    EFI_VECTOR_HANDOFF_HOOK_BEFORE,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_TIMER_VECTOR,                // Vector 32\r
+    EFI_VECTOR_HANDOFF_DO_NOT_HOOK,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    DEBUG_MAILBOX_VECTOR,              // Vector 33\r
+    EFI_VECTOR_HANDOFF_DO_NOT_HOOK,\r
+    EFI_DEBUG_AGENT_GUID\r
+  },\r
+  {\r
+    0,\r
+    EFI_VECTOR_HANDOFF_LAST_ENTRY,\r
+    { 0 }\r
+  }\r
+};\r
+\r
+UINTN mVectorHandoffInfoCount = sizeof (mVectorHandoffInfoDebugAgent) / sizeof (EFI_VECTOR_HANDOFF_INFO);\r
+\r
+/**\r
+  Calculate CRC16 for target data.\r
+\r
+  @param[in]  Data              The target data.\r
+  @param[in]  DataSize          The target data size.\r
+  @param[in]  Crc               Initial CRC.\r
+\r
+  @return UINT16     The CRC16 value.\r
+\r
+**/\r
+UINT16\r
+CalculateCrc16 (\r
+  IN UINT8   *Data,\r
+  IN UINTN   DataSize,\r
+  IN UINT16  Crc\r
+  )\r
+{\r
+  UINTN  Index;\r
+  UINTN  BitIndex;\r
+\r
+  for (Index = 0; Index < DataSize; Index++) {\r
+    Crc ^= Data[Index];\r
+    for (BitIndex = 0; BitIndex < 8; BitIndex++) {\r
+      if ((Crc & 0x8000) != 0) {\r
+        Crc <<= 1;\r
+        Crc ^= 0x1021;\r
+      } else {\r
+        Crc <<= 1;\r
+      }\r
+    }\r
+  }\r
+  return Crc;\r
+}\r
+\r
+\r
+/**\r
+  Read IDT entry to check if IDT entries are setup by Debug Agent.\r
+\r
+  @retval  TRUE     IDT entries were setup by Debug Agent.\r
+  @retval  FALSE    IDT entries were not setup by Debug Agent.\r
+\r
+**/\r
+BOOLEAN \r
+IsDebugAgentInitialzed (\r
+  VOID\r
+  )\r
+{\r
+  UINTN                      InterruptHandler;\r
+\r
+  InterruptHandler = (UINTN) GetExceptionHandlerInIdtEntry (0);\r
+  if (InterruptHandler >= 4 &&  *(UINT32 *)(InterruptHandler - 4) == AGENT_HANDLER_SIGNATURE) {\r
+    return TRUE;\r
+  } else {\r
+    return FALSE;\r
+  }\r
+}\r
+\r
+/**\r
+  Find and report module image info to HOST.\r
+  \r
+  @param[in] AlignSize      Image aligned size.\r
+  \r
+**/\r
+VOID \r
+FindAndReportModuleImageInfo (\r
+  IN UINTN          AlignSize                   \r
+  )\r
+{\r
+  UINTN                                Pe32Data;\r
+  EFI_IMAGE_DOS_HEADER                 *DosHdr;\r
+  EFI_IMAGE_OPTIONAL_HEADER_PTR_UNION  Hdr;\r
+  PE_COFF_LOADER_IMAGE_CONTEXT         ImageContext;\r
+\r
+  //\r
+  // Find Image Base\r
+  //\r
+  Pe32Data = ((UINTN)mErrorMsgVersionAlert) & ~(AlignSize - 1);\r
+  while (Pe32Data != 0) {\r
+    DosHdr = (EFI_IMAGE_DOS_HEADER *) Pe32Data;\r
+    if (DosHdr->e_magic == EFI_IMAGE_DOS_SIGNATURE) {\r
+      //\r
+      // DOS image header is present, so read the PE header after the DOS image header.\r
+      // Check if address overflow firstly.\r
+      //\r
+      if ((MAX_ADDRESS - (UINTN)DosHdr->e_lfanew) > Pe32Data) {\r
+        Hdr.Pe32 = (EFI_IMAGE_NT_HEADERS32 *)(Pe32Data + (UINTN)(DosHdr->e_lfanew));\r
+        if (Hdr.Pe32->Signature == EFI_IMAGE_NT_SIGNATURE) {\r
+          //\r
+          // It's PE image.\r
+          //\r
+          break;\r
+        }\r
+      }\r
+    } else {\r
+      //\r
+      // DOS image header is not present, TE header is at the image base.\r
+      //\r
+      Hdr.Pe32 = (EFI_IMAGE_NT_HEADERS32 *)Pe32Data;\r
+      if ((Hdr.Te->Signature == EFI_TE_IMAGE_HEADER_SIGNATURE) &&\r
+          ((Hdr.Te->Machine == IMAGE_FILE_MACHINE_I386) || Hdr.Te->Machine == IMAGE_FILE_MACHINE_X64)) {\r
+        //\r
+        // It's TE image, it TE header and Machine type match\r
+        //\r
+        break;\r
+      }\r
+    }\r
+\r
+    //\r
+    // Not found the image base, check the previous aligned address\r
+    // \r
+    Pe32Data -= AlignSize;\r
+  }\r
+\r
+  ImageContext.ImageAddress = Pe32Data;\r
+  ImageContext.PdbPointer = PeCoffLoaderGetPdbPointer ((VOID*) (UINTN) ImageContext.ImageAddress);\r
+  PeCoffLoaderRelocateImageExtraAction (&ImageContext);\r
+}\r
+\r
+/**\r
+  Trigger one software interrupt to debug agent to handle it.\r
+\r
+  @param[in] Signature       Software interrupt signature.\r
+\r
+**/\r
+VOID\r
+TriggerSoftInterrupt (\r
+  IN UINT32                 Signature\r
+  )\r
+{\r
+  UINTN                  Dr0;\r
+  UINTN                  Dr1;\r
+\r
+  //\r
+  // Save Debug Register State\r
+  //\r
+  Dr0 = AsmReadDr0 ();\r
+  Dr1 = AsmReadDr1 ();\r
+\r
+  //\r
+  // DR0 = Signature\r
+  //\r
+  AsmWriteDr0 (SOFT_INTERRUPT_SIGNATURE);\r
+  AsmWriteDr1 (Signature);\r
+\r
+  //\r
+  // Do INT3 to communicate with HOST side\r
+  //\r
+  CpuBreakpoint ();\r
+\r
+  //\r
+  // Restore Debug Register State only when Host didn't change it inside exception handler.\r
+  //   Dr registers can only be changed by setting the HW breakpoint.\r
+  //\r
+  AsmWriteDr0 (Dr0);\r
+  AsmWriteDr1 (Dr1);\r
+\r
+}\r
+\r
+/**\r
+  Caculate Mailbox checksum and update the checksum field.\r
+\r
+  @param[in]  Mailbox  Debug Agent Mailbox pointer.\r
+\r
+**/\r
+VOID\r
+UpdateMailboxChecksum (\r
+  IN DEBUG_AGENT_MAILBOX    *Mailbox\r
+  )\r
+{\r
+  Mailbox->CheckSum = CalculateCheckSum8 ((UINT8 *)Mailbox, sizeof (DEBUG_AGENT_MAILBOX) - 2);    \r
+}\r
+\r
+/**\r
+  Verify Mailbox checksum.\r
+\r
+  If checksum error, print debug message and run init dead loop.\r
+\r
+  @param[in]  Mailbox  Debug Agent Mailbox pointer.\r
+\r
+**/\r
+VOID \r
+VerifyMailboxChecksum (\r
+  IN DEBUG_AGENT_MAILBOX    *Mailbox\r
+  )\r
+{\r
+  UINT8                     CheckSum;\r
+  \r
+  CheckSum = CalculateCheckSum8 ((UINT8 *) Mailbox, sizeof (DEBUG_AGENT_MAILBOX) - 2);\r
+  //\r
+  // The checksum updating process may be disturbed by hardware SMI, we need to check CheckSum field \r
+  // and ToBeCheckSum field to validate the mail box.\r
+  //\r
+  if (CheckSum != Mailbox->CheckSum && CheckSum != Mailbox->ToBeCheckSum) {\r
+    DEBUG ((EFI_D_ERROR, "DebugAgent: Mailbox checksum error, stack or heap crashed!\n"));\r
+    DEBUG ((EFI_D_ERROR, "DebugAgent: CheckSum = %x, Mailbox->CheckSum = %x, Mailbox->ToBeCheckSum = %x\n", CheckSum, Mailbox->CheckSum, Mailbox->ToBeCheckSum));\r
+    CpuDeadLoop ();\r
+  }\r
+}\r
+\r
+/**\r
+  Update Mailbox content by index.\r
 \r
-CHAR8 mErrorMsgVersionAlert[]   = "\rThe SourceLevelDebugPkg you are using requires a newer version of the Intel(R) UDK Debugger Tool.\r\n";\r
-CHAR8 mErrorMsgSendInitPacket[] = "\rSend INIT break packet to HOST ...\r\n";\r
-CHAR8 mErrorMsgConnectOK[]      = "HOST connection is successful!\r\n";\r
-CHAR8 mErrorMsgConnectFail[]    = "HOST connection is failed!\r\n";\r
+  @param[in]  Mailbox  Debug Agent Mailbox pointer.\r
+  @param[in]  Index    Mailbox content index.\r
+  @param[in]  Value    Value to be set into Mailbox.\r
+  \r
+**/\r
+VOID\r
+UpdateMailboxContent ( \r
+  IN DEBUG_AGENT_MAILBOX    *Mailbox,\r
+  IN UINTN                  Index,\r
+  IN UINT64                 Value\r
+  )\r
+{\r
+  AcquireMpSpinLock (&mDebugMpContext.MailboxSpinLock);\r
+  switch (Index) {\r
+  case DEBUG_MAILBOX_DEBUG_FLAG_INDEX:\r
+    Mailbox->ToBeCheckSum = Mailbox->CheckSum + CalculateSum8 ((UINT8 *)&Mailbox->DebugFlag.Uint64, sizeof(UINT64)) \r
+                                              - CalculateSum8 ((UINT8 *)&Value, sizeof(UINT64));\r
+    Mailbox->DebugFlag.Uint64 = Value;\r
+    break;\r
+  case DEBUG_MAILBOX_DEBUG_PORT_HANDLE_INDEX:\r
+    Mailbox->ToBeCheckSum = Mailbox->CheckSum + CalculateSum8 ((UINT8 *)&Mailbox->DebugPortHandle, sizeof(UINTN))\r
+                                              - CalculateSum8 ((UINT8 *)&Value, sizeof(UINTN));\r
+    Mailbox->DebugPortHandle = (UINTN) Value;\r
+    break;\r
+  case DEBUG_MAILBOX_EXCEPTION_BUFFER_POINTER_INDEX:\r
+    Mailbox->ToBeCheckSum = Mailbox->CheckSum + CalculateSum8 ((UINT8 *)&Mailbox->ExceptionBufferPointer, sizeof(UINTN))\r
+                                              - CalculateSum8 ((UINT8 *)&Value, sizeof(UINTN));\r
+    Mailbox->ExceptionBufferPointer = (UINTN) Value;\r
+    break;\r
+  case DEBUG_MAILBOX_LAST_ACK:\r
+    Mailbox->ToBeCheckSum = Mailbox->CheckSum + CalculateSum8 ((UINT8 *)&Mailbox->LastAck, sizeof(UINT8))\r
+                                              - CalculateSum8 ((UINT8 *)&Value, sizeof(UINT8));\r
+    Mailbox->LastAck = (UINT8) Value;\r
+    break;\r
+  case DEBUG_MAILBOX_SEQUENCE_NO_INDEX:\r
+    Mailbox->ToBeCheckSum = Mailbox->CheckSum + CalculateSum8 ((UINT8 *)&Mailbox->SequenceNo, sizeof(UINT8))\r
+                                              - CalculateSum8 ((UINT8 *)&Value, sizeof(UINT8));\r
+    Mailbox->SequenceNo = (UINT8) Value;\r
+    break;\r
+  case DEBUG_MAILBOX_HOST_SEQUENCE_NO_INDEX:\r
+    Mailbox->ToBeCheckSum = Mailbox->CheckSum + CalculateSum8 ((UINT8 *)&Mailbox->HostSequenceNo, sizeof(UINT8))\r
+                                              - CalculateSum8 ((UINT8 *)&Value, sizeof(UINT8));\r
+    Mailbox->HostSequenceNo = (UINT8) Value;\r
+    break;\r
+  }\r
+  UpdateMailboxChecksum (Mailbox);\r
+  ReleaseMpSpinLock (&mDebugMpContext.MailboxSpinLock);\r
+}\r
+/**\r
+  Set debug flag in mailbox.\r
+\r
+  @param[in]  FlagMask      Debug flag mask value.\r
+  @param[in]  FlagValue     Debug flag value.\r
+\r
+**/\r
+VOID \r
+SetDebugFlag (\r
+  IN UINT64                 FlagMask,\r
+  IN UINT32                 FlagValue                          \r
+  )\r
+{\r
+  DEBUG_AGENT_MAILBOX    *Mailbox;\r
+  UINT64                 Data64;\r
+\r
+  Mailbox = GetMailboxPointer ();\r
+  Data64 = (Mailbox->DebugFlag.Uint64 & ~FlagMask) |\r
+           (LShiftU64 ((UINT64)FlagValue, LowBitSet64 (FlagMask)) & FlagMask);\r
+  UpdateMailboxContent (Mailbox, DEBUG_MAILBOX_DEBUG_FLAG_INDEX, Data64);\r
+}\r
+\r
+/**\r
+  Get debug flag in mailbox.\r
+\r
+  @param[in]  FlagMask      Debug flag mask value.\r
+  \r
+  @return Debug flag value.\r
+\r
+**/\r
+UINT32\r
+GetDebugFlag (\r
+  IN UINT64                 FlagMask\r
+  )\r
+{\r
+  DEBUG_AGENT_MAILBOX    *Mailbox;\r
+  UINT32                 DebugFlag;\r
+\r
+  Mailbox = GetMailboxPointer ();\r
+  DebugFlag = (UINT32) RShiftU64 (Mailbox->DebugFlag.Uint64 & FlagMask, LowBitSet64 (FlagMask));\r
+\r
+  return DebugFlag;\r
+}\r
 \r
 /**\r
   Send a debug message packet to the debug port.\r
@@ -46,8 +447,12 @@ SendDebugMsgPacket (
   DebugHeader.StartSymbol = DEBUG_STARTING_SYMBOL_NORMAL;\r
   DebugHeader.Command     = DEBUG_COMMAND_PRINT_MESSAGE;\r
   DebugHeader.Length      = sizeof (DEBUG_PACKET_HEADER) + (UINT8) Length;\r
-  DebugHeader.CheckSum    = 0;\r
-  DebugHeader.CheckSum    = CalculateCheckSum8 ((UINT8 *)&DebugHeader, sizeof (DEBUG_PACKET_HEADER));\r
+  DebugHeader.SequenceNo  = 0xEE;\r
+  DebugHeader.Crc         = 0;\r
+  DebugHeader.Crc         = CalculateCrc16 (\r
+                              (UINT8 *)Buffer, Length,\r
+                              CalculateCrc16 ((UINT8 *)&DebugHeader, sizeof (DEBUG_PACKET_HEADER), 0)\r
+                              );\r
 \r
   DebugPortWriteBuffer (Handle, (UINT8 *)&DebugHeader, sizeof (DEBUG_PACKET_HEADER));\r
   DebugPortWriteBuffer (Handle, (UINT8 *)Buffer, Length);\r
@@ -73,15 +478,13 @@ DebugAgentMsgPrint (
   ...\r
   )\r
 {\r
-  DEBUG_AGENT_MAILBOX  *Mailbox;\r
   CHAR8                Buffer[DEBUG_DATA_MAXIMUM_REAL_DATA];\r
   VA_LIST              Marker;\r
 \r
-  Mailbox = GetMailboxPointer ();\r
   //\r
   // Check driver debug mask value and global mask\r
   //\r
-  if ((ErrorLevel & Mailbox->DebugFlag.PrintErrorLevel) == 0) {\r
+  if ((ErrorLevel & GetDebugFlag (DEBUG_AGENT_FLAG_PRINT_ERROR_LEVEL)) == 0) {\r
     return;\r
   }\r
 \r
@@ -119,16 +522,14 @@ DebugAgentDataMsgPrint (
   IN UINT8             Length  \r
   )\r
 {\r
-  DEBUG_AGENT_MAILBOX  *Mailbox;\r
   CHAR8                Buffer[DEBUG_DATA_MAXIMUM_REAL_DATA];\r
   CHAR8                *DestBuffer;\r
   UINTN                Index;\r
 \r
-  Mailbox = GetMailboxPointer ();\r
   //\r
   // Check driver debug mask value and global mask\r
   //\r
-  if ((ErrorLevel & Mailbox->DebugFlag.PrintErrorLevel) == 0) {\r
+  if ((ErrorLevel & GetDebugFlag (DEBUG_AGENT_FLAG_PRINT_ERROR_LEVEL)) == 0) {\r
     return;\r
   }\r
 \r
@@ -152,7 +553,7 @@ DebugAgentDataMsgPrint (
     DestBuffer += AsciiSPrint (DestBuffer, DEBUG_DATA_MAXIMUM_REAL_DATA - (DestBuffer - Buffer), "%02x ", Data[Index]);\r
     Index ++;\r
     if (Index >= Length) {\r
-      //s\r
+      //\r
       // The last character of debug message has been foramtted in buffer\r
       //\r
       DestBuffer += AsciiSPrint(DestBuffer, DEBUG_DATA_MAXIMUM_REAL_DATA - (DestBuffer - Buffer), "]\n");\r
@@ -162,6 +563,60 @@ DebugAgentDataMsgPrint (
   }\r
 }\r
 \r
+/**\r
+  Read remaing debug packet except for the start symbol\r
+\r
+  @param[in]      Handle        Pointer to Debug Port handle.\r
+  @param[in, out] DebugHeader   Debug header buffer including start symbol.\r
+\r
+  @retval EFI_SUCCESS        Read the symbol in BreakSymbol.\r
+  @retval EFI_CRC_ERROR      CRC check fail.\r
+  @retval EFI_TIMEOUT        Timeout occurs when reading debug packet.\r
+  @retval EFI_DEVICE_ERROR   Receive the old or responsed packet.\r
+\r
+**/\r
+EFI_STATUS\r
+ReadRemainingBreakPacket (\r
+  IN     DEBUG_PORT_HANDLE      Handle,\r
+  IN OUT DEBUG_PACKET_HEADER    *DebugHeader\r
+  )\r
+{\r
+  UINT16                     Crc;\r
+  DEBUG_AGENT_MAILBOX        *Mailbox;\r
+\r
+  //\r
+  // Has received start symbol, try to read the rest part\r
+  //\r
+  if (DebugPortReadBuffer (Handle, (UINT8 *)DebugHeader + OFFSET_OF (DEBUG_PACKET_HEADER, Command), sizeof (DEBUG_PACKET_HEADER) - OFFSET_OF (DEBUG_PACKET_HEADER, Command), READ_PACKET_TIMEOUT) == 0) {\r
+    //\r
+    // Timeout occur, exit\r
+    //\r
+    DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "Timeout in Debug Timer interrupt\n");\r
+    return EFI_TIMEOUT;\r
+  }\r
+\r
+  Crc = DebugHeader->Crc;\r
+  DebugHeader->Crc = 0;\r
+  if (CalculateCrc16 ((UINT8 *)DebugHeader, DebugHeader->Length, 0) != Crc) {\r
+    DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "Debug Timer CRC (%x) against (%x)\n", Crc, CalculateCrc16 ((UINT8 *) &DebugHeader, DebugHeader->Length, 0));\r
+    DebugAgentDataMsgPrint (DEBUG_AGENT_VERBOSE, FALSE, (UINT8 *)DebugHeader, DebugHeader->Length);\r
+    return EFI_CRC_ERROR;\r
+  }\r
+  Mailbox = GetMailboxPointer();\r
+  if (((DebugHeader->Command & DEBUG_COMMAND_RESPONSE) == 0) &&\r
+       (DebugHeader->SequenceNo == (UINT8) (Mailbox->HostSequenceNo + 1))) {\r
+    //\r
+    // Only updagte HostSequenceNo for new command packet \r
+    //\r
+    UpdateMailboxContent (Mailbox, DEBUG_MAILBOX_HOST_SEQUENCE_NO_INDEX, DebugHeader->SequenceNo);\r
+    return EFI_SUCCESS;\r
+  } else {\r
+    //\r
+    // If one old command or response packet received, skip it\r
+    //\r
+    return EFI_DEVICE_ERROR;\r
+  }\r
+}\r
 \r
 /**\r
   Check if HOST is attached based on Mailbox.\r
@@ -175,7 +630,7 @@ IsHostAttached (
   VOID\r
   )\r
 {\r
-  return (BOOLEAN) (GetMailboxPointer ()->DebugFlag.HostAttached == 1);\r
+  return (BOOLEAN) (GetDebugFlag (DEBUG_AGENT_FLAG_HOST_ATTACHED) == 1);\r
 }\r
 \r
 /**\r
@@ -190,7 +645,7 @@ SetHostAttached (
   )\r
 {\r
   DebugAgentMsgPrint (DEBUG_AGENT_INFO, "Attach status is %d\n", Attached);\r
-  GetMailboxPointer ()->DebugFlag.HostAttached = Attached;\r
+  SetDebugFlag (DEBUG_AGENT_FLAG_HOST_ATTACHED, (UINT32)Attached);\r
 }\r
 \r
 /**\r
@@ -208,17 +663,17 @@ SetDebugSetting (
   )\r
 {\r
   RETURN_STATUS                Status;\r
-  DEBUG_AGENT_MAILBOX          *Mailbox;\r
-\r
-  Mailbox = GetMailboxPointer ();\r
 \r
   Status = RETURN_SUCCESS;\r
   switch (DebugSetting->Key) {\r
   case DEBUG_AGENT_SETTING_SMM_ENTRY_BREAK:\r
-    Mailbox->DebugFlag.BreakOnNextSmi = DebugSetting->Value;\r
+    SetDebugFlag (DEBUG_AGENT_FLAG_BREAK_ON_NEXT_SMI, DebugSetting->Value);\r
     break;\r
   case DEBUG_AGENT_SETTING_PRINT_ERROR_LEVEL:\r
-    Mailbox->DebugFlag.PrintErrorLevel = DebugSetting->Value;\r
+    SetDebugFlag (DEBUG_AGENT_FLAG_PRINT_ERROR_LEVEL, DebugSetting->Value);\r
+    break;\r
+  case DEBUG_AGENT_SETTING_BOOT_SCRIPT_ENTRY_BREAK:\r
+    SetDebugFlag (DEBUG_AGENT_FLAG_BREAK_BOOT_SCRIPT, DebugSetting->Value);\r
     break;\r
   default:\r
     Status = RETURN_UNSUPPORTED;\r
@@ -446,11 +901,13 @@ ArchReadRegisterBuffer (
   Send the packet without data to HOST.\r
 \r
   @param[in] CommandType    Type of Command.\r
+  @param[in] SequenceNo     Sequence number.\r
 \r
 **/\r
 VOID\r
 SendPacketWithoutData (\r
-  IN UINT8                  CommandType\r
+  IN UINT8                  CommandType,\r
+  IN UINT8                  SequenceNo\r
   )\r
 {\r
   DEBUG_PACKET_HEADER       DebugHeader;\r
@@ -461,8 +918,9 @@ SendPacketWithoutData (
   DebugHeader.StartSymbol = DEBUG_STARTING_SYMBOL_NORMAL;\r
   DebugHeader.Command     = CommandType;\r
   DebugHeader.Length      = sizeof (DEBUG_PACKET_HEADER);\r
-  DebugHeader.CheckSum    = 0;\r
-  DebugHeader.CheckSum    = CalculateCheckSum8 ((UINT8 *)&DebugHeader, sizeof (DEBUG_PACKET_HEADER));\r
+  DebugHeader.SequenceNo  = SequenceNo;\r
+  DebugHeader.Crc         = 0;\r
+  DebugHeader.Crc         = CalculateCrc16 ((UINT8 *)&DebugHeader, sizeof (DEBUG_PACKET_HEADER), 0);\r
 \r
   DebugAgentDataMsgPrint (DEBUG_AGENT_VERBOSE, TRUE, (UINT8 *) &DebugHeader, DebugHeader.Length);\r
   DebugPortWriteBuffer (Handle, (UINT8 *) &DebugHeader, DebugHeader.Length);\r
@@ -479,104 +937,142 @@ SendAckPacket (
   IN UINT8                AckCommand\r
   )\r
 {\r
+  UINT8                   SequenceNo;\r
+  DEBUG_AGENT_MAILBOX     *Mailbox;\r
+\r
   if (AckCommand != DEBUG_COMMAND_OK) {\r
+    //\r
+    // This is not ACK OK packet\r
+    //\r
     DebugAgentMsgPrint (DEBUG_AGENT_ERROR, "Send ACK(%d)\n", AckCommand);\r
   }\r
-  SendPacketWithoutData (AckCommand);\r
+  Mailbox = GetMailboxPointer();\r
+  SequenceNo = Mailbox->HostSequenceNo;\r
+  DebugAgentMsgPrint (DEBUG_AGENT_INFO, "SendAckPacket: SequenceNo = %x\n", SequenceNo);\r
+  SendPacketWithoutData (AckCommand, SequenceNo);\r
+  UpdateMailboxContent (Mailbox, DEBUG_MAILBOX_LAST_ACK, AckCommand);\r
 }\r
 \r
 /**\r
-  Receive acknowledge packet from HOST in specified time.\r
+  Receive valid packet from HOST.\r
 \r
-  @param[out] Ack            Returned acknowlege type from HOST.\r
-  @param[in]  Timeout        Time out value to wait for acknowlege from HOST.\r
-                             The unit is microsecond.\r
-  @param[out] BreakReceived  If BreakReceived is not NULL,\r
-                             TRUE is retured if break-in symbol received.\r
-                             FALSE is retured if break-in symbol not received.\r
-  @param[out] CheckSumStatus If CheckSumStatus is not NULL,\r
-                             RETURN_SUCCESS   CheckSum is OK.\r
-                             RETURN_NOT_FOUND Not find the CheckSum field.\r
+  @param[out] InputPacket         Buffer to receive packet.\r
+  @param[out] BreakReceived       TRUE means break-in symbol received.\r
+                                  FALSE means break-in symbol not received.\r
+  @param[out] IncompatibilityFlag If IncompatibilityFlag is not NULL, return\r
+                                  TRUE:  Compatible packet received.\r
+                                  FALSE: Incompatible packet received.\r
+  @param[in]  Timeout             Time out value to wait for acknowlege from HOST.\r
+                                  The unit is microsecond.\r
+  @param[in]  SkipStartSymbol     TRUE:  Skip time out when reading start symbol.\r
+                                  FALSE: Does not Skip time out when reading start symbol.\r
 \r
-  @retval  RETRUEN_SUCCESS  Succeed to receive acknowlege packet from HOST,\r
-                            the type of acknowlege packet saved in Ack.\r
-  @retval  RETURN_TIMEOUT   Specified timeout value was up.\r
+  @retval RETURN_SUCCESS   A valid package was reveived in InputPacket.\r
+  @retval RETURN_TIMEOUT   Timeout occurs.\r
 \r
 **/\r
 RETURN_STATUS\r
-ReceiveAckPacket (\r
-  OUT UINT8                     *Ack,\r
-  IN  UINTN                     Timeout,\r
-  OUT BOOLEAN                   *BreakReceived, OPTIONAL\r
-  OUT RETURN_STATUS             *CheckSumStatus OPTIONAL\r
+ReceivePacket (\r
+  OUT UINT8             *InputPacket,\r
+  OUT BOOLEAN           *BreakReceived,\r
+  OUT BOOLEAN           *IncompatibilityFlag, OPTIONAL\r
+  IN  UINTN             Timeout,\r
+  IN  BOOLEAN           SkipStartSymbol\r
   )\r
 {\r
-  DEBUG_PACKET_HEADER       DebugHeader;\r
-  DEBUG_PORT_HANDLE         Handle;\r
+  DEBUG_PACKET_HEADER   *DebugHeader;\r
+  UINTN                 Received;\r
+  DEBUG_PORT_HANDLE     Handle;\r
+  UINT16                Crc;\r
+  UINTN                 TimeoutForStartSymbol;\r
 \r
   Handle = GetDebugPortHandle();\r
-\r
+  if (SkipStartSymbol) {\r
+    TimeoutForStartSymbol = 0;\r
+  } else {\r
+    TimeoutForStartSymbol = Timeout;\r
+  }\r
+  \r
+  DebugHeader = (DEBUG_PACKET_HEADER *) InputPacket;\r
   while (TRUE) {\r
-    if (DebugPortReadBuffer (Handle, (UINT8 *) &DebugHeader.StartSymbol, sizeof (DebugHeader.StartSymbol), Timeout) == 0) {\r
+    //\r
+    // Find the valid start symbol\r
+    //\r
+    Received = DebugPortReadBuffer (Handle, &DebugHeader->StartSymbol, sizeof (DebugHeader->StartSymbol), TimeoutForStartSymbol);\r
+    if (Received < sizeof (DebugHeader->StartSymbol)) {\r
+      DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "DebugPortReadBuffer(StartSymbol) timeout\n");\r
       return RETURN_TIMEOUT;\r
     }\r
-    if (DebugHeader.StartSymbol == DEBUG_STARTING_SYMBOL_BREAK) {\r
-      if (BreakReceived != NULL) {\r
-        SendAckPacket (DEBUG_COMMAND_HALT_DEFERRED);\r
-        *BreakReceived = TRUE;\r
-      }\r
-    }\r
-    if (DebugHeader.StartSymbol == DEBUG_STARTING_SYMBOL_NORMAL) {\r
-      break;\r
+\r
+    if (DebugHeader->StartSymbol != DEBUG_STARTING_SYMBOL_NORMAL) {\r
+      DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "Invalid start symbol received [%02x]\n", DebugHeader->StartSymbol);\r
+      continue;\r
     }\r
-    DebugAgentMsgPrint (DEBUG_AGENT_ERROR, "Invalid start symbol received [%02x]\n", DebugHeader.StartSymbol);\r
-  }\r
-  //\r
-  // Read ACK packet header till field Length (not including StartSymbol and CheckSum)\r
-  //\r
-  DebugHeader.Length = 0;\r
-  if (DebugPortReadBuffer (\r
-        Handle,\r
-        (UINT8 *)&DebugHeader.Command,\r
-        OFFSET_OF (DEBUG_PACKET_HEADER, Length) + sizeof (DebugHeader.Length) - sizeof (DebugHeader.StartSymbol),\r
-        Timeout\r
-        ) == 0) {\r
-    return RETURN_TIMEOUT;\r
-  }\r
 \r
-  if (DebugHeader.Length == 0) {\r
     //\r
-    // The CheckSum field does not exist\r
+    // Read Package header till field Length\r
     //\r
-    if (CheckSumStatus != NULL) {\r
-      *CheckSumStatus = RETURN_NOT_FOUND;\r
-    }\r
-  } else {\r
-    if (CheckSumStatus != NULL) {\r
-      *CheckSumStatus = RETURN_SUCCESS;\r
-    }\r
-    if (DebugPortReadBuffer (Handle, &DebugHeader.CheckSum, sizeof (DebugHeader.CheckSum), Timeout) == 0) {\r
+    Received = DebugPortReadBuffer (\r
+                 Handle,\r
+                 (UINT8 *)DebugHeader + OFFSET_OF (DEBUG_PACKET_HEADER, Command),\r
+                 OFFSET_OF (DEBUG_PACKET_HEADER, Length) + sizeof (DebugHeader->Length) - sizeof (DebugHeader->StartSymbol),\r
+                 Timeout\r
+                 );\r
+    if (Received == 0) {\r
+      DebugAgentMsgPrint (DEBUG_AGENT_ERROR, "DebugPortReadBuffer(Command) timeout\n");\r
       return RETURN_TIMEOUT;\r
     }\r
+    if (DebugHeader->Length < sizeof (DEBUG_PACKET_HEADER)) {\r
+      if (IncompatibilityFlag != NULL) {\r
+        //\r
+        // This is one old version debug packet format, set Incompatibility flag\r
+        //\r
+        *IncompatibilityFlag = TRUE;\r
+      } else {\r
+        //\r
+        // Skip the bad small packet\r
+        //\r
+        continue;\r
+      }\r
+    } else {\r
+      //\r
+      // Read the payload data include the CRC field\r
+      //\r
+      Received = DebugPortReadBuffer (Handle, &DebugHeader->SequenceNo, (UINT8) (DebugHeader->Length - OFFSET_OF (DEBUG_PACKET_HEADER, SequenceNo)), Timeout);\r
+      if (Received == 0) {\r
+        DebugAgentMsgPrint (DEBUG_AGENT_ERROR, "DebugPortReadBuffer(SequenceNo) timeout\n");\r
+        return RETURN_TIMEOUT;\r
+      }\r
+      //\r
+      // Calculate the CRC of Debug Packet\r
+      //\r
+      Crc = DebugHeader->Crc;\r
+      DebugHeader->Crc = 0;\r
+      if (Crc == CalculateCrc16 ((UINT8 *) DebugHeader, DebugHeader->Length, 0)) {\r
+        break;\r
+      }\r
+      DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "CRC Error (received CRC is %x)\n", Crc);\r
+      DebugAgentDataMsgPrint (DEBUG_AGENT_VERBOSE, FALSE, (UINT8 *) DebugHeader, DebugHeader->Length);\r
+    }\r
   }\r
 \r
-  DebugAgentDataMsgPrint (DEBUG_AGENT_VERBOSE, FALSE, (UINT8 *)&DebugHeader, DebugHeader.Length);\r
-  *Ack = DebugHeader.Command;\r
+  DebugAgentDataMsgPrint (DEBUG_AGENT_VERBOSE, FALSE, (UINT8 *) DebugHeader, DebugHeader->Length);\r
+\r
   return RETURN_SUCCESS;\r
 }\r
 \r
 /**\r
   Receive acknowledge packet OK from HOST in specified time.\r
 \r
-  @param[in]  Command        The command type issued by TARGET.\r
-  @param[in]  Timeout        Time out value to wait for acknowlege from HOST.\r
-                             The unit is microsecond.\r
-  @param[out] BreakReceived  If BreakReceived is not NULL,\r
-                             TRUE is retured if break-in symbol received.\r
-                             FALSE is retured if break-in symbol not received.\r
-  @param[out] CheckSumStatus If CheckSumStatus is not NULL,\r
-                             RETURN_SUCCESS   CheckSum is OK.\r
-                             RETURN_NOT_FOUND Not find the CheckSum field.\r
+  @param[in]  Command             The command type issued by TARGET.\r
+  @param[in]  Timeout             Time out value to wait for acknowlege from HOST.\r
+                                  The unit is microsecond.\r
+  @param[out] BreakReceived       If BreakReceived is not NULL,\r
+                                  TRUE is retured if break-in symbol received.\r
+                                  FALSE is retured if break-in symbol not received.\r
+  @param[out] IncompatibilityFlag If IncompatibilityFlag is not NULL, return\r
+                                  TRUE:  Compatible packet received.\r
+                                  FALSE: Incompatible packet received.\r
 \r
   @retval  RETRUEN_SUCCESS  Succeed to receive acknowlege packet from HOST,\r
                             the type of acknowlege packet saved in Ack.\r
@@ -588,105 +1084,56 @@ SendCommandAndWaitForAckOK (
   IN  UINT8               Command,\r
   IN  UINTN               Timeout,\r
   OUT BOOLEAN             *BreakReceived, OPTIONAL\r
-  OUT RETURN_STATUS       *CheckSumStatus OPTIONAL\r
+  OUT BOOLEAN             *IncompatibilityFlag OPTIONAL\r
   )\r
 {\r
   RETURN_STATUS           Status;\r
-  UINT8                   Ack;\r
-  \r
-  while (TRUE) {\r
-    SendPacketWithoutData (Command);\r
-    while (TRUE) {\r
-      Status = ReceiveAckPacket (&Ack, Timeout, BreakReceived, CheckSumStatus);\r
-      if (Status == RETURN_SUCCESS && Ack == DEBUG_COMMAND_OK) {\r
-        //\r
-        // Received Ack OK or timeout\r
-        //\r
-        return Status;\r
-      }\r
-      if (Status == RETURN_TIMEOUT) {\r
-        break;\r
+  UINT8                   InputPacketBuffer[DEBUG_DATA_UPPER_LIMIT];\r
+  DEBUG_PACKET_HEADER     *DebugHeader;\r
+  UINT8                   SequenceNo;\r
+  UINT8                   HostSequenceNo;\r
+  UINT8                   RetryCount;\r
+\r
+  RetryCount  = 3;\r
+  DebugHeader = (DEBUG_PACKET_HEADER *) InputPacketBuffer;\r
+  Status      = RETURN_TIMEOUT;\r
+  while (RetryCount > 0) {\r
+    SequenceNo = GetMailboxPointer()->SequenceNo;\r
+    HostSequenceNo = GetMailboxPointer()->HostSequenceNo;\r
+    SendPacketWithoutData (Command, SequenceNo);\r
+    Status = ReceivePacket ((UINT8 *) DebugHeader, BreakReceived, IncompatibilityFlag, Timeout, FALSE);\r
+    if (Status == RETURN_TIMEOUT) {\r
+      if (Command == DEBUG_COMMAND_INIT_BREAK) {\r
+        RetryCount--;\r
+      } else {\r
+        DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "TARGET: Timeout when waiting for ACK packet.\n");\r
       }\r
-    }\r
-  }\r
-}\r
-\r
-/**\r
-  Receive valid packet from HOST.\r
-\r
-  @param[out] InputPacket    Buffer to receive packet.\r
-  @param[out] BreakReceived  TRUE means break-in symbol received.\r
-                             FALSE means break-in symbol not received.\r
-\r
-  @retval RETURN_SUCCESS   A valid package was reveived in InputPacket.\r
-  @retval RETURN_TIMEOUT   Timeout occurs.\r
-\r
-**/\r
-RETURN_STATUS\r
-ReceivePacket (\r
-  OUT UINT8             *InputPacket,\r
-  OUT BOOLEAN           *BreakReceived\r
-  )\r
-{\r
-  DEBUG_PACKET_HEADER   *DebugHeader;\r
-  UINTN                 Received;\r
-  DEBUG_PORT_HANDLE     Handle;\r
-  UINT8                 CheckSum;\r
-\r
-  Handle = GetDebugPortHandle();\r
-  \r
-  DebugHeader = (DEBUG_PACKET_HEADER *) InputPacket;\r
-  while (TRUE) {\r
-    //\r
-    // Find the valid start symbol\r
-    //\r
-    DebugPortReadBuffer (Handle, &DebugHeader->StartSymbol, sizeof (DebugHeader->StartSymbol), 0);\r
-\r
-    if (DebugHeader->StartSymbol == DEBUG_STARTING_SYMBOL_BREAK) {\r
-      *BreakReceived = TRUE;\r
-      SendAckPacket (DEBUG_COMMAND_HALT_DEFERRED);\r
-    }\r
-\r
-    if (DebugHeader->StartSymbol != DEBUG_STARTING_SYMBOL_NORMAL) {\r
-      DebugAgentMsgPrint (DEBUG_AGENT_ERROR, "Invalid start symbol received [%02x]\n", DebugHeader->StartSymbol);\r
       continue;\r
     }\r
-\r
-    //\r
-    // Read Package header except for checksum\r
-    //\r
-    Received = DebugPortReadBuffer (\r
-                 Handle,\r
-                 &DebugHeader->Command,\r
-                 OFFSET_OF (DEBUG_PACKET_HEADER, Length) + sizeof (DebugHeader->Length) - sizeof (DebugHeader->StartSymbol),\r
-                 0\r
-                 );\r
-    if (Received == 0) {\r
-      return RETURN_TIMEOUT;\r
-    }\r
-\r
+    ASSERT_EFI_ERROR (Status);\r
     //\r
-    // Read the payload data include the checksum\r
+    // Status == RETURN_SUCCESS\r
     //\r
-    Received = DebugPortReadBuffer (Handle, &DebugHeader->CheckSum, DebugHeader->Length - OFFSET_OF (DEBUG_PACKET_HEADER, CheckSum), 0);\r
-    if (Received == 0) {\r
-      return RETURN_TIMEOUT;\r
+    if (DebugHeader->Command == DEBUG_COMMAND_OK && DebugHeader->SequenceNo == SequenceNo) {\r
+      //\r
+      // Received Ack OK\r
+      //\r
+      UpdateMailboxContent (GetMailboxPointer(), DEBUG_MAILBOX_SEQUENCE_NO_INDEX, ++SequenceNo);\r
+      return Status;\r
     }\r
-    //\r
-    // Calculate the checksum of Debug Packet\r
-    //\r
-    CheckSum = CalculateCheckSum8 ((UINT8 *) DebugHeader, DebugHeader->Length);\r
-    if (CheckSum == 0) {\r
-      break;\r
+    if (DebugHeader->Command == DEBUG_COMMAND_GO && (DebugHeader->SequenceNo == HostSequenceNo || Command == DEBUG_COMMAND_INIT_BREAK)) {\r
+      //\r
+      // Received Old GO\r
+      //\r
+      if (Command == DEBUG_COMMAND_INIT_BREAK) {\r
+        DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "TARGET: Receive GO() in last boot\n");\r
+      }\r
+      SendPacketWithoutData (DEBUG_COMMAND_OK, DebugHeader->SequenceNo);\r
     }\r
-    DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "CheckSum Error (Caculated checksum is %x, received checksum is %x\n", CheckSum, DebugHeader->CheckSum);\r
-    DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "Send DEBUG_COMMAND_RESEND command.\n");\r
-    SendAckPacket (DEBUG_COMMAND_RESEND);\r
   }\r
 \r
-  DebugAgentDataMsgPrint (DEBUG_AGENT_VERBOSE, FALSE, (UINT8 *) DebugHeader, DebugHeader->Length);\r
-\r
-  return RETURN_SUCCESS;\r
+  ASSERT (Command == DEBUG_COMMAND_INIT_BREAK);\r
+  return Status;\r
 }\r
 \r
 /**\r
@@ -769,7 +1216,7 @@ GetBreakCause (
 \r
   default:\r
     if (Vector < 20) {\r
-      if (GetMailboxPointer()->DebugFlag.SteppingFlag == 1) {\r
+      if (GetDebugFlag (DEBUG_AGENT_FLAG_STEPPING) == 1) {\r
         //\r
         // If stepping command is executing\r
         //\r
@@ -785,89 +1232,163 @@ GetBreakCause (
 }\r
 \r
 /**\r
-  Send command packet with data to HOST.\r
+  Copy memory from source to destination with specified width.\r
+\r
+  @param[out] Dest        A pointer to the destination buffer of the memory copy.\r
+  @param[in]  Src         A pointer to the source buffer of the memory copy.\r
+  @param[in]  Count       The number of data with specified width to copy from source to destination.\r
+  @param[in]  Width       Data width in byte.\r
+\r
+**/\r
+VOID\r
+CopyMemByWidth (\r
+  OUT UINT8               *Dest,\r
+  IN  UINT8               *Src,\r
+  IN  UINT16              Count,\r
+  IN  UINT8               Width\r
+  )\r
+{\r
+  UINT8                   *Destination;\r
+  UINT8                   *Source;\r
+  INT8                    Step;\r
+\r
+  if (Src > Dest) {\r
+    Destination = Dest;\r
+    Source      = Src;\r
+    Step        = Width;\r
+  } else {\r
+    //\r
+    // Copy memory from tail to avoid memory overlap\r
+    //\r
+    Destination = Dest + (Count - 1) * Width;\r
+    Source      = Src  + (Count - 1) * Width;\r
+    Step        = -Width;\r
+  }\r
+\r
+  while (Count-- != 0) {\r
+    switch (Width) {\r
+    case 1:\r
+      *(UINT8 *) Destination = MmioRead8 ((UINTN) Source);\r
+      break;\r
+    case 2:\r
+      *(UINT16 *) Destination = MmioRead16 ((UINTN) Source);\r
+      break;\r
+    case 4:\r
+      *(UINT32 *) Destination = MmioRead32 ((UINTN) Source);\r
+      break;\r
+    case 8:\r
+      *(UINT64 *) Destination = MmioRead64 ((UINTN) Source);\r
+      break;\r
+    default:\r
+      ASSERT (FALSE);\r
+    }\r
+    Source      += Step;\r
+    Destination += Step;\r
+  }\r
+}\r
+\r
+/**\r
+  Read memory with speicifed width and send packet with response data to HOST.\r
 \r
-  @param[in] Command     Command type.\r
   @param[in] Data        Pointer to response data buffer.\r
-  @param[in] DataSize    Size of response data in byte.\r
+  @param[in] Count       The number of data with specified Width.\r
+  @param[in] Width       Data width in byte.\r
 \r
   @retval RETURN_SUCCESS      Response data was sent successfully.\r
-  @retval RETURN_DEVICE_ERROR Cannot receive DEBUG_COMMAND_OK from HOST.\r
 \r
 **/\r
 RETURN_STATUS\r
-SendCommandWithDataPacket (\r
-  IN UINT8                Command,\r
+ReadMemoryAndSendResponsePacket (\r
   IN UINT8                *Data,\r
-  IN UINT16               DataSize\r
+  IN UINT16               Count,\r
+  IN UINT8                Width\r
   )\r
 {\r
+  RETURN_STATUS        Status;\r
   DEBUG_PACKET_HEADER  *DebugHeader;\r
   BOOLEAN              LastPacket;\r
-  UINT8                Ack;\r
-  UINT8                DebugPacket[DEBUG_DATA_UPPER_LIMIT];\r
+  DEBUG_PACKET_HEADER  *AckDebugHeader;\r
+  UINT8                DebugPacket[DEBUG_DATA_UPPER_LIMIT + sizeof (UINT64) - 1];\r
+  UINT8                InputPacketBuffer[DEBUG_DATA_UPPER_LIMIT];\r
   DEBUG_PORT_HANDLE    Handle;\r
+  UINT8                SequenceNo;\r
+  UINTN                RemainingDataSize;\r
+  UINTN                CurrentDataSize;\r
 \r
   Handle = GetDebugPortHandle();\r
 \r
-  DebugHeader = (DEBUG_PACKET_HEADER *) &DebugPacket;\r
+  //\r
+  // Data is appended end of Debug Packet header,  make sure data address\r
+  // in Debug Packet 8-byte alignment always\r
+  //\r
+  DebugHeader = (DEBUG_PACKET_HEADER *) (ALIGN_VALUE ((UINTN)&DebugPacket + sizeof (DEBUG_PACKET_HEADER), sizeof (UINT64))\r
+                                         - sizeof (DEBUG_PACKET_HEADER));\r
   DebugHeader->StartSymbol = DEBUG_STARTING_SYMBOL_NORMAL;\r
 \r
+  RemainingDataSize = Count * Width;\r
   while (TRUE) {\r
-    if (DataSize <= DEBUG_DATA_MAXIMUM_REAL_DATA) {\r
+    SequenceNo = GetMailboxPointer()->HostSequenceNo;\r
+    if (RemainingDataSize <= DEBUG_DATA_MAXIMUM_REAL_DATA) {\r
+      //\r
+      // If the remaining data is less one real packet size, this is the last data packet\r
+      //\r
+      CurrentDataSize = RemainingDataSize;\r
       LastPacket = TRUE;\r
-      DebugHeader->Command  = Command;\r
-      DebugHeader->Length   = (UINT8) (DataSize + sizeof (DEBUG_PACKET_HEADER));\r
-      DebugHeader->CheckSum = 0;\r
-      CopyMem (DebugHeader + 1, Data, DataSize);\r
-\r
+      DebugHeader->Command = DEBUG_COMMAND_OK;\r
     } else {\r
+      //\r
+      // Data is too larger to be sent in one packet, calculate the actual data size could\r
+      // be sent in one Maximum data packet\r
+      //\r
+      CurrentDataSize = (DEBUG_DATA_MAXIMUM_REAL_DATA / Width) * Width;\r
       LastPacket = FALSE;\r
-      DebugHeader->Command  = DEBUG_COMMAND_IN_PROGRESS;\r
-      DebugHeader->Length   = DEBUG_DATA_MAXIMUM_REAL_DATA + sizeof (DEBUG_PACKET_HEADER);\r
-      DebugHeader->CheckSum = 0;\r
-      CopyMem (DebugHeader + 1, Data, DEBUG_DATA_MAXIMUM_REAL_DATA);\r
+      DebugHeader->Command = DEBUG_COMMAND_IN_PROGRESS;\r
     }\r
-\r
     //\r
-    // Calculate and fill the checksum\r
+    // Construct the rest Debug header\r
+    //\r
+    DebugHeader->Length     = (UINT8)(CurrentDataSize + sizeof (DEBUG_PACKET_HEADER));\r
+    DebugHeader->SequenceNo = SequenceNo;\r
+    DebugHeader->Crc        = 0;\r
+    CopyMemByWidth ((UINT8 *)(DebugHeader + 1), Data, (UINT16) CurrentDataSize / Width, Width);\r
+    //\r
+    // Calculate and fill the checksum, DebugHeader->Crc should be 0 before invoking CalculateCrc16 ()\r
     //\r
-    DebugHeader->CheckSum = CalculateCheckSum8 ((UINT8 *) DebugHeader, DebugHeader->Length);\r
+    DebugHeader->Crc = CalculateCrc16 ((UINT8 *) DebugHeader, DebugHeader->Length, 0);\r
 \r
     DebugAgentDataMsgPrint (DEBUG_AGENT_VERBOSE, TRUE, (UINT8 *) DebugHeader, DebugHeader->Length);\r
     \r
     DebugPortWriteBuffer (Handle, (UINT8 *) DebugHeader, DebugHeader->Length);\r
 \r
-    ReceiveAckPacket(&Ack, 0, NULL, NULL);\r
-    switch (Ack) {\r
-    case DEBUG_COMMAND_RESEND:\r
-      //\r
-      // Send the packet again\r
-      //\r
-      DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "Received DEBUG_COMMAND_RESEND.\n");\r
-      break;\r
-\r
-    case DEBUG_COMMAND_CONTINUE:\r
-      //\r
-      // Send the rest packet\r
-      //\r
-      Data     += DEBUG_DATA_MAXIMUM_REAL_DATA;\r
-      DataSize -= DEBUG_DATA_MAXIMUM_REAL_DATA;\r
-      break;\r
-\r
-    case DEBUG_COMMAND_OK:\r
-      if (LastPacket) {\r
+    while (TRUE) {\r
+      Status = ReceivePacket (InputPacketBuffer, NULL, NULL, READ_PACKET_TIMEOUT, FALSE);\r
+      if (Status == RETURN_TIMEOUT) {\r
+        DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "TARGET: Timeout in SendDataResponsePacket()\n");\r
+        break;\r
+      }\r
+      AckDebugHeader = (DEBUG_PACKET_HEADER *) InputPacketBuffer;\r
+      SequenceNo = AckDebugHeader->SequenceNo;\r
+      if (AckDebugHeader->Command == DEBUG_COMMAND_OK &&\r
+          SequenceNo == DebugHeader->SequenceNo &&\r
+          LastPacket) {\r
         //\r
         // If this is the last packet, return RETURN_SUCCESS.\r
         //\r
         return RETURN_SUCCESS;\r
-      } else {\r
-        return RETURN_DEVICE_ERROR;\r
       }\r
-\r
-    default:\r
-      return RETURN_DEVICE_ERROR;\r
-\r
+      if ((SequenceNo == (UINT8) (DebugHeader->SequenceNo + 1)) && (AckDebugHeader->Command == DEBUG_COMMAND_CONTINUE)) {\r
+        //\r
+        // Calculate the rest data size\r
+        //\r
+        Data              += CurrentDataSize;\r
+        RemainingDataSize -= CurrentDataSize;\r
+        UpdateMailboxContent (GetMailboxPointer(), DEBUG_MAILBOX_HOST_SEQUENCE_NO_INDEX, (UINT8) SequenceNo);\r
+        break;\r
+      }\r
+      if (SequenceNo >= DebugHeader->SequenceNo) {\r
+        DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "TARGET: Received one old or new command(SequenceNo is %x, last SequenceNo is %x)\n", SequenceNo, DebugHeader->SequenceNo);\r
+        break;\r
+      }\r
     }\r
   }\r
 }\r
@@ -879,7 +1400,6 @@ SendCommandWithDataPacket (
   @param[in] DataSize    Size of response data in byte.\r
 \r
   @retval RETURN_SUCCESS      Response data was sent successfully.\r
-  @retval RETURN_DEVICE_ERROR Cannot receive DEBUG_COMMAND_OK from HOST.\r
 \r
 **/\r
 RETURN_STATUS\r
@@ -888,7 +1408,7 @@ SendDataResponsePacket (
   IN UINT16               DataSize\r
   )\r
 {\r
-  return SendCommandWithDataPacket (DEBUG_COMMAND_OK, Data, DataSize);\r
+  return ReadMemoryAndSendResponsePacket (Data, DataSize, 1);\r
 }\r
 \r
 /**\r
@@ -923,6 +1443,7 @@ SendBreakCausePacket (
   If received acknowlege, check the revision of HOST. \r
   Set Attach Flag if attach successfully.  \r
   \r
+  @param[in]  BreakCause     Break cause of this break event.\r
   @param[in]  Timeout        Time out value to wait for acknowlege from HOST.\r
                              The unit is microsecond.\r
   @param[out] BreakReceived  If BreakReceived is not NULL,\r
@@ -931,42 +1452,45 @@ SendBreakCausePacket (
 **/\r
 RETURN_STATUS\r
 AttachHost (\r
+  IN  UINT8                BreakCause,\r
   IN  UINTN                Timeout,\r
   OUT BOOLEAN              *BreakReceived\r
   )\r
 {\r
   RETURN_STATUS                    Status;\r
   DEBUG_PORT_HANDLE                Handle;\r
-  RETURN_STATUS                    CheckSumStatus;\r
+  BOOLEAN                          IncompatibilityFlag;\r
 \r
+  IncompatibilityFlag = FALSE;\r
   Handle = GetDebugPortHandle();\r
     \r
   //\r
   // Send init break and wait ack in Timeout\r
   //\r
   DebugPortWriteBuffer (Handle, (UINT8 *) mErrorMsgSendInitPacket, AsciiStrLen (mErrorMsgSendInitPacket));\r
-  Status = SendCommandAndWaitForAckOK (DEBUG_COMMAND_INIT_BREAK, Timeout, BreakReceived, &CheckSumStatus);\r
-  if (RETURN_ERROR (Status)) {\r
-    DebugPortWriteBuffer (Handle, (UINT8 *) mErrorMsgConnectFail, AsciiStrLen (mErrorMsgConnectFail));\r
-    return Status;\r
+  if (BreakCause == DEBUG_DATA_BREAK_CAUSE_SYSTEM_RESET) {\r
+    Status = SendCommandAndWaitForAckOK (DEBUG_COMMAND_INIT_BREAK, Timeout, BreakReceived, &IncompatibilityFlag);\r
+  } else {\r
+    Status = SendCommandAndWaitForAckOK (DEBUG_COMMAND_ATTACH_BREAK, Timeout, BreakReceived, &IncompatibilityFlag);\r
   }\r
-  \r
-  if (CheckSumStatus == RETURN_NOT_FOUND) {\r
+  if (IncompatibilityFlag) {\r
     //\r
-    // If the CheckSum field does not exist in Debug Packet,\r
-    // the HOST should be running with 0.1 transfer protocol.\r
-    // It could be UDK Debugger for Windows v1.1 or for Linux v0.8.\r
+    // If the incompatible Debug Packet received, the HOST should be running transfer protocol before DEBUG_AGENT_REVISION.\r
+    // It could be UDK Debugger for Windows v1.1/v1.2 or for Linux v0.8/v1.2.\r
     //\r
     DebugPortWriteBuffer (Handle, (UINT8 *) mErrorMsgVersionAlert, AsciiStrLen (mErrorMsgVersionAlert));\r
     CpuDeadLoop ();\r
   }\r
-\r
-  DebugPortWriteBuffer (Handle, (UINT8 *) mErrorMsgConnectOK, AsciiStrLen (mErrorMsgConnectOK));\r
-  //\r
-  // Set Attach flag\r
-  //\r
-  SetHostAttached (TRUE);\r
-\r
+  \r
+  if (RETURN_ERROR (Status)) {\r
+    DebugPortWriteBuffer (Handle, (UINT8 *) mErrorMsgConnectFail, AsciiStrLen (mErrorMsgConnectFail));\r
+  } else {\r
+    DebugPortWriteBuffer (Handle, (UINT8 *) mErrorMsgConnectOK, AsciiStrLen (mErrorMsgConnectOK));\r
+    //\r
+    // Set Attach flag\r
+    //\r
+    SetHostAttached (TRUE);\r
+  }\r
   return Status;\r
 }\r
 \r
@@ -974,7 +1498,8 @@ AttachHost (
   Send Break point packet to HOST. \r
   \r
   Only the first breaking processor could sent BREAK_POINT packet.\r
\r
+\r
+  @param[in]  BreakCause     Break cause of this break event.\r
   @param[in]  ProcessorIndex Processor index value.\r
   @param[out] BreakReceived  If BreakReceived is not NULL,\r
                              TRUE is retured if break-in symbol received.\r
@@ -983,6 +1508,7 @@ AttachHost (
 **/\r
 VOID\r
 SendBreakPacketToHost (\r
+  IN  UINT8                BreakCause,\r
   IN  UINT32               ProcessorIndex,\r
   OUT BOOLEAN              *BreakReceived\r
   )\r
@@ -994,7 +1520,7 @@ SendBreakPacketToHost (
   \r
   if (IsHostAttached ()) {\r
     DebugAgentMsgPrint (DEBUG_AGENT_INFO, "processor[%x]:Send Break Packet to HOST.\n", ProcessorIndex);\r
-    SendCommandAndWaitForAckOK (DEBUG_COMMAND_BREAK_POINT, 0, BreakReceived, NULL);\r
+    SendCommandAndWaitForAckOK (DEBUG_COMMAND_BREAK_POINT, READ_PACKET_TIMEOUT, BreakReceived, NULL);\r
   } else {\r
     DebugAgentMsgPrint (DEBUG_AGENT_INFO, "processor[%x]:Try to attach HOST.\n", ProcessorIndex);\r
     //\r
@@ -1011,7 +1537,7 @@ SendBreakPacketToHost (
     //\r
     // Try to attach HOST\r
     //\r
-    while (AttachHost (0, NULL) != RETURN_SUCCESS);\r
+    while (AttachHost (BreakCause, 0, NULL) != RETURN_SUCCESS);\r
    \r
   }\r
 }\r
@@ -1035,7 +1561,7 @@ CommandCommunication (
   )\r
 {\r
   RETURN_STATUS                     Status;\r
-  UINT8                             InputPacketBuffer[DEBUG_DATA_UPPER_LIMIT];\r
+  UINT8                             InputPacketBuffer[DEBUG_DATA_UPPER_LIMIT + sizeof (UINT64) - 1];\r
   DEBUG_PACKET_HEADER               *DebugHeader;\r
   UINT8                             Width;\r
   UINT8                             Data8;\r
@@ -1061,12 +1587,14 @@ CommandCommunication (
   DEBUG_PORT_HANDLE                 Handle;\r
   DEBUG_AGENT_EXCEPTION_BUFFER      AgentExceptionBuffer;\r
   UINT32                            IssuedViewPoint;\r
+  DEBUG_AGENT_MAILBOX               *Mailbox;\r
+  UINT8                             *AlignedDataPtr;\r
 \r
   ProcessorIndex  = 0;\r
   IssuedViewPoint = 0;\r
   HaltDeferred    = BreakReceived;\r
 \r
-  if (MultiProcessorDebugSupport) {\r
+  if (MultiProcessorDebugSupport()) {\r
     ProcessorIndex = GetProcessorIndex ();\r
     SetCpuStopFlagByIndex (ProcessorIndex, TRUE);\r
     if (mDebugMpContext.ViewPointIndex == ProcessorIndex) {\r
@@ -1081,14 +1609,14 @@ CommandCommunication (
     //\r
     // Set AgentInProgress Flag.\r
     //\r
-    GetMailboxPointer()->DebugFlag.AgentInProgress = 1;\r
+    SetDebugFlag (DEBUG_AGENT_FLAG_AGENT_IN_PROGRESS, 1);\r
   }  \r
 \r
   Handle = GetDebugPortHandle();\r
 \r
   while (TRUE) {\r
 \r
-    if (MultiProcessorDebugSupport) {\r
+    if (MultiProcessorDebugSupport()) {\r
       //\r
       // Check if the current processor is HOST view point\r
       //\r
@@ -1110,35 +1638,44 @@ CommandCommunication (
       }\r
     }\r
 \r
-    AcquireDebugPortControl ();\r
+    AcquireMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
 \r
-    Status = ReceivePacket (InputPacketBuffer, &BreakReceived);\r
+    DebugHeader =(DEBUG_PACKET_HEADER *) InputPacketBuffer;\r
 \r
-    if (BreakReceived) {\r
-      HaltDeferred = TRUE;\r
-      BreakReceived = FALSE;\r
+    DebugAgentMsgPrint (DEBUG_AGENT_INFO, "TARGET: Try to get command from HOST...\n");\r
+    Status = ReceivePacket ((UINT8 *)DebugHeader, &BreakReceived, NULL, READ_PACKET_TIMEOUT, TRUE);\r
+    if (Status != RETURN_SUCCESS || (DebugHeader->Command & DEBUG_COMMAND_RESPONSE) != 0) {\r
+      DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "TARGET: Get command[%x] sequenceno[%x] returned status is [%x] \n", DebugHeader->Command, DebugHeader->SequenceNo, Status);\r
+      DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "TARGET: Get command failed or it's response packet not expected! \n");\r
+      ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
+      continue;\r
     }\r
 \r
-    if (Status != RETURN_SUCCESS) {\r
-      ReleaseDebugPortControl ();\r
+    Mailbox = GetMailboxPointer ();\r
+    if (DebugHeader->SequenceNo == Mailbox->HostSequenceNo) {\r
+      DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "TARGET: Receive one old command[%x] agaist command[%x]\n", DebugHeader->SequenceNo, Mailbox->HostSequenceNo);\r
+      SendAckPacket (Mailbox->LastAck);\r
+      ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
+      continue;\r
+    } else if (DebugHeader->SequenceNo == (UINT8) (Mailbox->HostSequenceNo + 1)) {\r
+      UpdateMailboxContent (Mailbox, DEBUG_MAILBOX_HOST_SEQUENCE_NO_INDEX, (UINT8) DebugHeader->SequenceNo);\r
+    } else {\r
+      DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "Receive one invalid comamnd[%x] agaist command[%x]\n", DebugHeader->SequenceNo, Mailbox->HostSequenceNo);\r
+      ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
       continue;\r
     }\r
-\r
-    Data8 = 1;\r
-\r
-    DebugHeader =(DEBUG_PACKET_HEADER *) InputPacketBuffer;\r
-\r
-    GetMailboxPointer()->ExceptionBufferPointer = (UINT64)(UINTN) &AgentExceptionBuffer.JumpBuffer;\r
+    \r
     //\r
     // Save CPU content before executing HOST commond\r
     //\r
+    UpdateMailboxContent (Mailbox, DEBUG_MAILBOX_EXCEPTION_BUFFER_POINTER_INDEX, (UINT64)(UINTN) &AgentExceptionBuffer.JumpBuffer);\r
     if (SetJump (&AgentExceptionBuffer.JumpBuffer) != 0) {\r
       //\r
       // If HOST command failed, continue to wait for HOST's next command\r
       // If needed, agent could send exception info to HOST.\r
       //\r
       SendAckPacket (DEBUG_COMMAND_ABORT);\r
-      ReleaseDebugPortControl ();\r
+      ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
       continue;\r
     }\r
 \r
@@ -1146,9 +1683,18 @@ CommandCommunication (
     \r
     switch (DebugHeader->Command) {\r
 \r
+    case DEBUG_COMMAND_HALT:\r
+      SendAckPacket (DEBUG_COMMAND_HALT_DEFERRED);\r
+      HaltDeferred = TRUE;\r
+      BreakReceived = FALSE;\r
+      Status = RETURN_SUCCESS;\r
+      break;\r
+\r
     case DEBUG_COMMAND_RESET:\r
       SendAckPacket (DEBUG_COMMAND_OK);\r
-      ReleaseDebugPortControl ();\r
+      SendAckPacket (DEBUG_COMMAND_OK);\r
+      SendAckPacket (DEBUG_COMMAND_OK);\r
+      ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
 \r
       ResetCold ();\r
       //\r
@@ -1156,6 +1702,9 @@ CommandCommunication (
       // PCD can be used if 2 seconds isn't long enough for some platforms.\r
       //\r
       MicroSecondDelay (2000000);\r
+      UpdateMailboxContent (Mailbox, DEBUG_MAILBOX_HOST_SEQUENCE_NO_INDEX, Mailbox->HostSequenceNo + 1);\r
+      SendAckPacket (DEBUG_COMMAND_TIMEOUT);\r
+      SendAckPacket (DEBUG_COMMAND_TIMEOUT);\r
       SendAckPacket (DEBUG_COMMAND_TIMEOUT);\r
       break;\r
 \r
@@ -1172,13 +1721,13 @@ CommandCommunication (
       //\r
       // Clear Stepping Flag\r
       //\r
-      GetMailboxPointer()->DebugFlag.SteppingFlag = 0;\r
+      SetDebugFlag (DEBUG_AGENT_FLAG_STEPPING, 0);\r
       \r
       if (!HaltDeferred) {\r
         //\r
         // If no HALT command received when being in-active mode\r
         //\r
-        if (MultiProcessorDebugSupport) {\r
+        if (MultiProcessorDebugSupport()) {\r
           Data32 = FindNextPendingBreakCpu ();\r
           if (Data32 != -1) {\r
             //\r
@@ -1196,11 +1745,11 @@ CommandCommunication (
             //\r
             // Send break packet to HOST to let HOST break again\r
             //\r
-            SendBreakPacketToHost (0, &BreakReceived);\r
+            SendBreakPacketToHost (DEBUG_DATA_BREAK_CAUSE_UNKNOWN, mDebugMpContext.BreakAtCpuIndex, &BreakReceived);\r
             //\r
             // Continue to run into loop to read command packet from HOST\r
             //\r
-            ReleaseDebugPortControl (); \r
+            ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
             break;\r
           }\r
 \r
@@ -1236,8 +1785,12 @@ CommandCommunication (
         //\r
         SendAckPacket (DEBUG_COMMAND_OK);\r
 \r
-        ReleaseDebugPortControl ();\r
+        ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
 \r
+        if (!IsHostAttached()) {\r
+          UpdateMailboxContent (Mailbox, DEBUG_MAILBOX_SEQUENCE_NO_INDEX, 0);\r
+          UpdateMailboxContent (Mailbox, DEBUG_MAILBOX_HOST_SEQUENCE_NO_INDEX, 0);\r
+        }\r
         return;\r
 \r
       } else {\r
@@ -1253,7 +1806,7 @@ CommandCommunication (
 \r
     case DEBUG_COMMAND_BREAK_CAUSE:\r
 \r
-      if (MultiProcessorDebugSupport && ProcessorIndex != mDebugMpContext.BreakAtCpuIndex) {\r
+      if (MultiProcessorDebugSupport() && ProcessorIndex != mDebugMpContext.BreakAtCpuIndex) {\r
         Status = SendBreakCausePacket (DEBUG_TIMER_VECTOR, CpuContext);\r
 \r
       } else {\r
@@ -1287,8 +1840,8 @@ CommandCommunication (
       //\r
       // Set Stepping Flag\r
       //\r
-      GetMailboxPointer()->DebugFlag.SteppingFlag = 1;\r
-      ReleaseDebugPortControl ();\r
+      SetDebugFlag (DEBUG_AGENT_FLAG_STEPPING, 1);\r
+      ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
       //\r
       // Executing stepping command directly without sending ACK packet,\r
       // ACK packet will be sent after stepping done.\r
@@ -1304,12 +1857,19 @@ CommandCommunication (
 \r
     case DEBUG_COMMAND_READ_MEMORY:\r
       MemoryRead = (DEBUG_DATA_READ_MEMORY *) (DebugHeader + 1);\r
-      Status = SendDataResponsePacket ((UINT8 *) (UINTN) MemoryRead->Address, (UINT16) (MemoryRead->Count * MemoryRead->Width));\r
+      Status = ReadMemoryAndSendResponsePacket ((UINT8 *) (UINTN) MemoryRead->Address, MemoryRead->Count, MemoryRead->Width);\r
       break;\r
 \r
     case DEBUG_COMMAND_WRITE_MEMORY:\r
       MemoryWrite = (DEBUG_DATA_WRITE_MEMORY *) (DebugHeader + 1);\r
-      CopyMem ((VOID *) (UINTN) MemoryWrite->Address, &MemoryWrite->Data, MemoryWrite->Count * MemoryWrite->Width);\r
+      //\r
+      // Copy data into one memory with 8-byte alignment address\r
+      //\r
+      AlignedDataPtr = ALIGN_POINTER ((UINT8 *) &MemoryWrite->Data, sizeof (UINT64));\r
+      if (AlignedDataPtr != (UINT8 *) &MemoryWrite->Data) {\r
+        CopyMem (AlignedDataPtr, (UINT8 *) &MemoryWrite->Data, MemoryWrite->Count * MemoryWrite->Width);\r
+      }\r
+      CopyMemByWidth ((UINT8 *) (UINTN) MemoryWrite->Address, AlignedDataPtr, MemoryWrite->Count, MemoryWrite->Width);\r
       SendAckPacket (DEBUG_COMMAND_OK);\r
       break;\r
 \r
@@ -1420,7 +1980,7 @@ CommandCommunication (
 \r
     case DEBUG_COMMAND_SET_VIEWPOINT:\r
       SetViewPoint = (DEBUG_DATA_SET_VIEWPOINT *) (DebugHeader + 1);\r
-      if (MultiProcessorDebugSupport) {\r
+      if (MultiProcessorDebugSupport()) {\r
         if (IsCpuStopped (SetViewPoint->ViewPoint)) {\r
           SetDebugViewPoint (SetViewPoint->ViewPoint);\r
           SendAckPacket (DEBUG_COMMAND_OK);\r
@@ -1445,7 +2005,7 @@ CommandCommunication (
       break;\r
 \r
     case DEBUG_COMMAND_MEMORY_READY:\r
-      Data8 = (UINT8) GetMailboxPointer ()->DebugFlag.MemoryReady;\r
+      Data8 = (UINT8) GetDebugFlag (DEBUG_AGENT_FLAG_MEMORY_READY);\r
       SendDataResponsePacket (&Data8, (UINT16) sizeof (UINT8));\r
       break;\r
 \r
@@ -1513,7 +2073,7 @@ CommandCommunication (
       SendAckPacket (DEBUG_COMMAND_ABORT);\r
     }\r
 \r
-    ReleaseDebugPortControl ();\r
+    ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
     CpuPause ();\r
   }\r
 }\r
@@ -1544,11 +2104,22 @@ InterruptProcess (
   UINT32                           IssuedViewPoint;\r
   DEBUG_AGENT_EXCEPTION_BUFFER     *ExceptionBuffer;\r
 \r
+  InputCharacter  = 0;\r
   ProcessorIndex  = 0;\r
   IssuedViewPoint = 0;\r
   BreakReceived   = FALSE;\r
 \r
-  if (MultiProcessorDebugSupport) {\r
+  if (mSkipBreakpoint) {\r
+    //\r
+    // If Skip Breakpoint flag is set, means communication is disturbed by hardware SMI, we need to ignore the break points in SMM\r
+    //\r
+    if ((Vector == DEBUG_INT1_VECTOR) || (Vector == DEBUG_INT3_VECTOR)) {\r
+      DebugPortWriteBuffer (GetDebugPortHandle(), (UINT8 *) mWarningMsgIngoreBreakpoint, AsciiStrLen (mWarningMsgIngoreBreakpoint));\r
+      return;\r
+    }\r
+  }\r
+\r
+  if (MultiProcessorDebugSupport()) {\r
     ProcessorIndex = GetProcessorIndex ();\r
     //\r
     // If this processor has alreay halted before, need to check it later\r
@@ -1558,22 +2129,22 @@ InterruptProcess (
     }\r
   }\r
 \r
-  if (IssuedViewPoint == ProcessorIndex) {\r
+  if (IssuedViewPoint == ProcessorIndex && GetDebugFlag (DEBUG_AGENT_FLAG_STEPPING) != 1) {\r
     //\r
     // Check if this exception is issued by Debug Agent itself\r
     // If yes, fill the debug agent exception buffer and LongJump() back to\r
     // the saved CPU content in CommandCommunication()\r
     //\r
-    if (GetMailboxPointer()->DebugFlag.AgentInProgress == 1) {\r
-      DebugAgentMsgPrint (DEBUG_AGENT_ERROR, "Debug agent meet one Exception, ExceptionNum is %d.\n", Vector);\r
+    if (GetDebugFlag (DEBUG_AGENT_FLAG_AGENT_IN_PROGRESS) == 1) {\r
+      DebugAgentMsgPrint (DEBUG_AGENT_ERROR, "Debug agent meet one Exception, ExceptionNum is %d, EIP = 0x%x.\n", Vector, (UINTN)CpuContext->Eip);\r
       ExceptionBuffer = (DEBUG_AGENT_EXCEPTION_BUFFER *) (UINTN) GetMailboxPointer()->ExceptionBufferPointer;\r
       ExceptionBuffer->ExceptionContent.ExceptionNum  = (UINT8) Vector;\r
       ExceptionBuffer->ExceptionContent.ExceptionData = (UINT32) CpuContext->ExceptionData;\r
-      LongJump ((BASE_LIBRARY_JUMP_BUFFER *)(UINTN)(GetMailboxPointer()->ExceptionBufferPointer), 1);\r
+      LongJump ((BASE_LIBRARY_JUMP_BUFFER *)(UINTN)(ExceptionBuffer), 1);\r
     }\r
   }\r
 \r
-  if (MultiProcessorDebugSupport) {\r
+  if (MultiProcessorDebugSupport()) {\r
     //\r
     // If RUN commmand is executing, wait for it done.  \r
     //\r
@@ -1582,16 +2153,14 @@ InterruptProcess (
     }\r
   }\r
 \r
-  Handle = GetDebugPortHandle();\r
-\r
+  Handle     = GetDebugPortHandle();\r
+  BreakCause = GetBreakCause (Vector, CpuContext);\r
   switch (Vector) {\r
   case DEBUG_INT1_VECTOR:\r
   case DEBUG_INT3_VECTOR:\r
-    BreakCause = GetBreakCause (Vector, CpuContext);\r
-\r
     switch (BreakCause) {\r
     case DEBUG_DATA_BREAK_CAUSE_SYSTEM_RESET:\r
-      if (AttachHost (INIT_BREAK_ACK_TIMEOUT, &BreakReceived) != RETURN_SUCCESS) {\r
+      if (AttachHost (BreakCause, READ_PACKET_TIMEOUT, &BreakReceived) != RETURN_SUCCESS) {\r
         //\r
         // Try to connect HOST, return if fails\r
         //\r
@@ -1604,7 +2173,7 @@ InterruptProcess (
       //\r
       // Stepping is finished, send Ack package.\r
       //\r
-      if (MultiProcessorDebugSupport) {\r
+      if (MultiProcessorDebugSupport()) {\r
         mDebugMpContext.BreakAtCpuIndex = ProcessorIndex;\r
       }\r
       SendAckPacket (DEBUG_COMMAND_OK);\r
@@ -1615,7 +2184,7 @@ InterruptProcess (
       //\r
       // Memory is ready\r
       //\r
-      SendCommandAndWaitForAckOK (DEBUG_COMMAND_MEMORY_READY, 0, &BreakReceived, NULL);\r
+      SendCommandAndWaitForAckOK (DEBUG_COMMAND_MEMORY_READY, READ_PACKET_TIMEOUT, &BreakReceived, NULL);\r
       CommandCommunication (Vector, CpuContext, BreakReceived);\r
       break;\r
 \r
@@ -1643,14 +2212,14 @@ InterruptProcess (
       //\r
       // Send Break packet to HOST\r
       //\r
-      AcquireDebugPortControl ();\r
+      AcquireMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
       //\r
       // Only the first breaking processor could send BREAK_POINT to HOST\r
       // \r
       if (IsFirstBreakProcessor (ProcessorIndex)) {\r
-        SendBreakPacketToHost (ProcessorIndex, &BreakReceived);\r
+        SendBreakPacketToHost (BreakCause, ProcessorIndex, &BreakReceived);\r
       }\r
-      ReleaseDebugPortControl ();\r
+      ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
       \r
       if (Vector == DEBUG_INT3_VECTOR) {\r
         //\r
@@ -1677,9 +2246,9 @@ InterruptProcess (
 \r
   case DEBUG_TIMER_VECTOR:\r
 \r
-    AcquireDebugPortControl ();\r
+    AcquireMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
 \r
-    if (MultiProcessorDebugSupport) {\r
+    if (MultiProcessorDebugSupport()) {\r
       if (IsBsp (ProcessorIndex)) {\r
         //\r
         // If current processor is BSP, check Apic timer's init count if changed,\r
@@ -1693,7 +2262,7 @@ InterruptProcess (
       }\r
 \r
       if (!IsBsp (ProcessorIndex) || mDebugMpContext.IpiSentByAp) {\r
-        ReleaseDebugPortControl ();\r
+        ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
         //\r
         // If current processor is not BSP or this is one IPI sent by AP\r
         //\r
@@ -1721,8 +2290,10 @@ InterruptProcess (
       if (DebugReadBreakSymbol (Handle, &InputCharacter) == EFI_NOT_FOUND) {\r
         break;\r
       }\r
+\r
       if ((!IsHostAttached () && (InputCharacter == DEBUG_STARTING_SYMBOL_ATTACH)) ||\r
-          (IsHostAttached () && (InputCharacter == DEBUG_STARTING_SYMBOL_BREAK))\r
+          (IsHostAttached () && (InputCharacter == DEBUG_COMMAND_HALT)) ||\r
+          (IsHostAttached () && (InputCharacter == DEBUG_COMMAND_GO))\r
          ) {\r
         DebugAgentMsgPrint (DEBUG_AGENT_VERBOSE, "Received data [%02x]\n", InputCharacter);\r
         //\r
@@ -1730,25 +2301,32 @@ InterruptProcess (
         //\r
         SendAckPacket (DEBUG_COMMAND_OK);\r
 \r
+        //\r
+        // If receive GO command in Debug Timer, means HOST may lost ACK packet before.\r
+        //\r
+        if (InputCharacter == DEBUG_COMMAND_GO) {\r
+          break;\r
+        }\r
+\r
         if (!IsHostAttached ()) {\r
           //\r
           // Try to attach HOST, if no ack received after 200ms, return\r
           //\r
-          if (AttachHost (INIT_BREAK_ACK_TIMEOUT, &BreakReceived) != RETURN_SUCCESS) {\r
+          if (AttachHost (BreakCause, READ_PACKET_TIMEOUT, &BreakReceived) != RETURN_SUCCESS) {\r
             break;\r
           }\r
         }\r
 \r
-        if (MultiProcessorDebugSupport) {\r
+        if (MultiProcessorDebugSupport()) {\r
           if(FindNextPendingBreakCpu  () != -1) {\r
             SetCpuBreakFlagByIndex (ProcessorIndex, TRUE);\r
           } else {\r
             HaltOtherProcessors (ProcessorIndex);\r
           }\r
         }\r
-        ReleaseDebugPortControl ();\r
+        ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
         CommandCommunication (Vector, CpuContext, BreakReceived);\r
-        AcquireDebugPortControl ();\r
+        AcquireMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
         break;\r
       }\r
     }\r
@@ -1758,19 +2336,17 @@ InterruptProcess (
     //\r
     SendApicEoi ();\r
 \r
-    ReleaseDebugPortControl ();\r
+    ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
 \r
     break;\r
 \r
   default:\r
-\r
     if (Vector <= DEBUG_EXCEPT_SIMD) {\r
-      BreakCause = GetBreakCause (Vector, CpuContext);\r
       if (BreakCause == DEBUG_DATA_BREAK_CAUSE_STEPPING) {\r
         //\r
         // Stepping is finished, send Ack package.\r
         //\r
-        if (MultiProcessorDebugSupport) {\r
+        if (MultiProcessorDebugSupport()) {\r
           mDebugMpContext.BreakAtCpuIndex = ProcessorIndex;\r
         }\r
         SendAckPacket (DEBUG_COMMAND_OK);\r
@@ -1778,14 +2354,14 @@ InterruptProcess (
         //\r
         // Exception occurs, send Break packet to HOST\r
         //\r
-        AcquireDebugPortControl ();\r
+        AcquireMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
         //\r
         // Only the first breaking processor could send BREAK_POINT to HOST\r
         // \r
         if (IsFirstBreakProcessor (ProcessorIndex)) {\r
-          SendBreakPacketToHost (ProcessorIndex, &BreakReceived);\r
+          SendBreakPacketToHost (BreakCause, ProcessorIndex, &BreakReceived);\r
         }\r
-        ReleaseDebugPortControl ();\r
+        ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
       }\r
       \r
       CommandCommunication (Vector, CpuContext, BreakReceived);\r
@@ -1793,7 +2369,7 @@ InterruptProcess (
     break;\r
   }\r
 \r
-  if (MultiProcessorDebugSupport) {\r
+  if (MultiProcessorDebugSupport()) {\r
     //\r
     // Clear flag and wait for all processors run here\r
     //\r
@@ -1810,11 +2386,11 @@ InterruptProcess (
     }\r
   }\r
 \r
-  if (IssuedViewPoint == ProcessorIndex) {\r
+  if (IssuedViewPoint == ProcessorIndex && GetDebugFlag (DEBUG_AGENT_FLAG_STEPPING) != 1) {\r
     //\r
-    // Clean up AgentInProgress flag\r
+    // If the command is not stepping, clean up AgentInProgress flag\r
     //\r
-    GetMailboxPointer()->DebugFlag.AgentInProgress = 0;\r
+    SetDebugFlag (DEBUG_AGENT_FLAG_AGENT_IN_PROGRESS, 0);\r
   }\r
 \r
   return;\r