]> git.proxmox.com Git - mirror_edk2.git/blobdiff - SourceLevelDebugPkg/Library/DebugAgent/DebugAgentCommon/DebugAgent.c
PeCoffGetEntryPointLib: Fix spelling issue
[mirror_edk2.git] / SourceLevelDebugPkg / Library / DebugAgent / DebugAgentCommon / DebugAgent.c
index 95306e5759e767d0d62980013af24c4f80555f42..f156fe24db2d3578e2766d96cc147b84a01ef80e 100644 (file)
@@ -4,7 +4,7 @@
   read/write debug packet to communication with HOST based on transfer\r
   protocol.\r
 \r
-  Copyright (c) 2010 - 2014, Intel Corporation. All rights reserved.<BR>\r
+  Copyright (c) 2010 - 2017, 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
-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
+GLOBAL_REMOVE_IF_UNREFERENCED CHAR8 mErrorMsgVersionAlert[]       = "\rThe SourceLevelDebugPkg you are using requires a newer version of the Intel(R) UDK Debugger Tool.\r\n";\r
+GLOBAL_REMOVE_IF_UNREFERENCED CHAR8 mErrorMsgSendInitPacket[]     = "\rSend INIT break packet and try to connect the HOST (Intel(R) UDK Debugger Tool v1.5) ...\r\n";\r
+GLOBAL_REMOVE_IF_UNREFERENCED CHAR8 mErrorMsgConnectOK[]          = "HOST connection is successful!\r\n";\r
+GLOBAL_REMOVE_IF_UNREFERENCED CHAR8 mErrorMsgConnectFail[]        = "HOST connection is failed!\r\n";\r
+GLOBAL_REMOVE_IF_UNREFERENCED 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
+GLOBAL_REMOVE_IF_UNREFERENCED EFI_VECTOR_HANDOFF_INFO mVectorHandoffInfoDebugAgent[] = {\r
   {\r
     DEBUG_EXCEPT_DIVIDE_ERROR,         // Vector 0\r
     EFI_VECTOR_HANDOFF_HOOK_BEFORE,\r
@@ -130,7 +130,7 @@ EFI_VECTOR_HANDOFF_INFO mVectorHandoffInfoDebugAgent[] = {
   }\r
 };\r
 \r
-UINTN mVectorHandoffInfoCount = sizeof (mVectorHandoffInfoDebugAgent) / sizeof (EFI_VECTOR_HANDOFF_INFO);\r
+GLOBAL_REMOVE_IF_UNREFERENCED UINTN mVectorHandoffInfoCount = sizeof (mVectorHandoffInfoDebugAgent) / sizeof (EFI_VECTOR_HANDOFF_INFO);\r
 \r
 /**\r
   Calculate CRC16 for target data.\r
@@ -153,7 +153,7 @@ CalculateCrc16 (
   UINTN  BitIndex;\r
 \r
   for (Index = 0; Index < DataSize; Index++) {\r
-    Crc ^= Data[Index];\r
+    Crc ^= (UINT16)Data[Index];\r
     for (BitIndex = 0; BitIndex < 8; BitIndex++) {\r
       if ((Crc & 0x8000) != 0) {\r
         Crc <<= 1;\r
@@ -174,7 +174,7 @@ CalculateCrc16 (
   @retval  FALSE    IDT entries were not setup by Debug Agent.\r
 \r
 **/\r
-BOOLEAN \r
+BOOLEAN\r
 IsDebugAgentInitialzed (\r
   VOID\r
   )\r
@@ -191,63 +191,27 @@ IsDebugAgentInitialzed (
 \r
 /**\r
   Find and report module image info to HOST.\r
-  \r
+\r
   @param[in] AlignSize      Image aligned size.\r
-  \r
+\r
 **/\r
-VOID \r
+VOID\r
 FindAndReportModuleImageInfo (\r
-  IN UINTN          AlignSize                   \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
+  Pe32Data = PeCoffSearchImageBase ((UINTN) mErrorMsgVersionAlert);\r
+  if (Pe32Data != 0) {\r
+    ImageContext.ImageAddress = Pe32Data;\r
+    ImageContext.PdbPointer = PeCoffLoaderGetPdbPointer ((VOID*) (UINTN) ImageContext.ImageAddress);\r
+    PeCoffLoaderRelocateImageExtraAction (&ImageContext);\r
   }\r
-\r
-  ImageContext.ImageAddress = Pe32Data;\r
-  ImageContext.PdbPointer = PeCoffLoaderGetPdbPointer ((VOID*) (UINTN) ImageContext.ImageAddress);\r
-  PeCoffLoaderRelocateImageExtraAction (&ImageContext);\r
 }\r
 \r
 /**\r
@@ -291,7 +255,7 @@ TriggerSoftInterrupt (
 }\r
 \r
 /**\r
-  Caculate Mailbox checksum and update the checksum field.\r
+  Calculate Mailbox checksum and update the checksum field.\r
 \r
   @param[in]  Mailbox  Debug Agent Mailbox pointer.\r
 \r
@@ -301,7 +265,7 @@ UpdateMailboxChecksum (
   IN DEBUG_AGENT_MAILBOX    *Mailbox\r
   )\r
 {\r
-  Mailbox->CheckSum = CalculateCheckSum8 ((UINT8 *)Mailbox, sizeof (DEBUG_AGENT_MAILBOX) - 2);    \r
+  Mailbox->CheckSum = CalculateCheckSum8 ((UINT8 *)Mailbox, sizeof (DEBUG_AGENT_MAILBOX) - 2);\r
 }\r
 \r
 /**\r
@@ -312,16 +276,16 @@ UpdateMailboxChecksum (
   @param[in]  Mailbox  Debug Agent Mailbox pointer.\r
 \r
 **/\r
-VOID \r
+VOID\r
 VerifyMailboxChecksum (\r
   IN DEBUG_AGENT_MAILBOX    *Mailbox\r
   )\r
 {\r
   UINT8                     CheckSum;\r
-  \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
+  // 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
@@ -337,10 +301,10 @@ VerifyMailboxChecksum (
   @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
 **/\r
 VOID\r
-UpdateMailboxContent ( \r
+UpdateMailboxContent (\r
   IN DEBUG_AGENT_MAILBOX    *Mailbox,\r
   IN UINTN                  Index,\r
   IN UINT64                 Value\r
@@ -349,7 +313,7 @@ UpdateMailboxContent (
   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
+    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
@@ -378,10 +342,92 @@ UpdateMailboxContent (
                                               - CalculateSum8 ((UINT8 *)&Value, sizeof(UINT8));\r
     Mailbox->HostSequenceNo = (UINT8) Value;\r
     break;\r
+  case DEBUG_MAILBOX_DEBUG_TIMER_FREQUENCY:\r
+    Mailbox->ToBeCheckSum = Mailbox->CheckSum + CalculateSum8 ((UINT8 *)&Mailbox->DebugTimerFrequency, sizeof(UINT32))\r
+                                              - CalculateSum8 ((UINT8 *)&Value, sizeof(UINT32));\r
+    Mailbox->DebugTimerFrequency = (UINT32) Value;\r
+    break;\r
   }\r
   UpdateMailboxChecksum (Mailbox);\r
   ReleaseMpSpinLock (&mDebugMpContext.MailboxSpinLock);\r
 }\r
+\r
+/**\r
+  Read data from debug device and save the data in buffer.\r
+\r
+  Reads NumberOfBytes data bytes from a debug device into the buffer\r
+  specified by Buffer. The number of bytes actually read is returned.\r
+  If the return value is less than NumberOfBytes, then the rest operation failed.\r
+  If NumberOfBytes is zero, then return 0.\r
+\r
+  @param  Handle           Debug port handle.\r
+  @param  Buffer           Pointer to the data buffer to store the data read from the debug device.\r
+  @param  NumberOfBytes    Number of bytes which will be read.\r
+  @param  Timeout          Timeout value for reading from debug device. It unit is Microsecond.\r
+\r
+  @retval 0                Read data failed, no data is to be read.\r
+  @retval >0               Actual number of bytes read from debug device.\r
+\r
+**/\r
+UINTN\r
+DebugAgentReadBuffer (\r
+  IN DEBUG_PORT_HANDLE     Handle,\r
+  IN UINT8                 *Buffer,\r
+  IN UINTN                 NumberOfBytes,\r
+  IN UINTN                 Timeout\r
+  )\r
+{\r
+  UINTN                    Index;\r
+  UINT32                   Begin;\r
+  UINT32                   TimeoutTicker;\r
+  UINT32                   TimerRound;\r
+  UINT32                   TimerFrequency;\r
+  UINT32                   TimerCycle;\r
+  \r
+  Begin         = 0;\r
+  TimeoutTicker = 0;  \r
+  TimerRound    = 0;\r
+  TimerFrequency = GetMailboxPointer()->DebugTimerFrequency;\r
+  TimerCycle = GetApicTimerInitCount ();\r
+\r
+  if (Timeout != 0) {\r
+    Begin = GetApicTimerCurrentCount ();\r
+    TimeoutTicker = (UINT32) DivU64x32 (\r
+                      MultU64x64 (\r
+                        TimerFrequency,\r
+                        Timeout\r
+                        ),\r
+                      1000000u\r
+                      );\r
+    TimerRound = (UINT32) DivU64x32Remainder (TimeoutTicker,  TimerCycle / 2, &TimeoutTicker);\r
+  }\r
+  Index = 0;\r
+  while (Index < NumberOfBytes) {\r
+    if (DebugPortPollBuffer (Handle)) {\r
+      DebugPortReadBuffer (Handle, Buffer + Index, 1, 0);\r
+      Index ++; \r
+      continue;\r
+    }\r
+    if (Timeout != 0) {\r
+      if (TimerRound == 0) {\r
+        if (IsDebugTimerTimeout (TimerCycle, Begin, TimeoutTicker)) {\r
+          //\r
+          // If time out occurs.\r
+          //\r
+          return 0;\r
+        }\r
+      } else {\r
+        if (IsDebugTimerTimeout (TimerCycle, Begin, TimerCycle / 2)) {\r
+          TimerRound --;\r
+          Begin = GetApicTimerCurrentCount ();\r
+        }\r
+      }\r
+    }\r
+  }\r
+\r
+  return Index;\r
+}\r
+\r
 /**\r
   Set debug flag in mailbox.\r
 \r
@@ -389,10 +435,10 @@ UpdateMailboxContent (
   @param[in]  FlagValue     Debug flag value.\r
 \r
 **/\r
-VOID \r
+VOID\r
 SetDebugFlag (\r
   IN UINT64                 FlagMask,\r
-  IN UINT32                 FlagValue                          \r
+  IN UINT32                 FlagValue\r
   )\r
 {\r
   DEBUG_AGENT_MAILBOX    *Mailbox;\r
@@ -408,7 +454,7 @@ SetDebugFlag (
   Get debug flag in mailbox.\r
 \r
   @param[in]  FlagMask      Debug flag mask value.\r
-  \r
+\r
   @return Debug flag value.\r
 \r
 **/\r
@@ -436,12 +482,12 @@ GetDebugFlag (
 VOID\r
 SendDebugMsgPacket (\r
   IN CHAR8         *Buffer,\r
-  IN UINTN         Length         \r
+  IN UINTN         Length\r
   )\r
 {\r
   DEBUG_PACKET_HEADER  DebugHeader;\r
   DEBUG_PORT_HANDLE    Handle;\r
-  \r
+\r
   Handle = GetDebugPortHandle();\r
 \r
   DebugHeader.StartSymbol = DEBUG_STARTING_SYMBOL_NORMAL;\r
@@ -466,7 +512,7 @@ SendDebugMsgPacket (
 \r
   @param[in] ErrorLevel  The error level of the debug message.\r
   @param[in] Format      Format string for the debug message to print.\r
-  @param[in] ...         Variable argument list whose contents are accessed \r
+  @param[in] ...         Variable argument list whose contents are accessed\r
                          based on the format string specified by Format.\r
 \r
 **/\r
@@ -501,15 +547,15 @@ DebugAgentMsgPrint (
 /**\r
   Prints a debug message to the debug output device if the specified error level is enabled.\r
 \r
-  If any bit in ErrorLevel is also set in DebugPrintErrorLevelLib function \r
-  GetDebugPrintErrorLevel (), then print the message specified by Format and the \r
+  If any bit in ErrorLevel is also set in DebugPrintErrorLevelLib function\r
+  GetDebugPrintErrorLevel (), then print the message specified by Format and the\r
   associated variable argument list to the debug output device.\r
 \r
   If Format is NULL, then ASSERT().\r
 \r
   @param[in] ErrorLevel  The error level of the debug message.\r
   @param[in] IsSend      Flag of debug message to declare that the data is being sent or being received.\r
-  @param[in] Data        Variable argument list whose contents are accessed \r
+  @param[in] Data        Variable argument list whose contents are accessed\r
   @param[in] Length      based on the format string specified by Format.\r
 \r
 **/\r
@@ -519,7 +565,7 @@ DebugAgentDataMsgPrint (
   IN UINT8             ErrorLevel,\r
   IN BOOLEAN           IsSend,\r
   IN UINT8             *Data,\r
-  IN UINT8             Length  \r
+  IN UINT8             Length\r
   )\r
 {\r
   CHAR8                Buffer[DEBUG_DATA_MAXIMUM_REAL_DATA];\r
@@ -544,7 +590,7 @@ DebugAgentDataMsgPrint (
   while (TRUE) {\r
     if (DestBuffer - Buffer > DEBUG_DATA_MAXIMUM_REAL_DATA - 6) {\r
       //\r
-      // If there was no enough space in buffer, send out the debug message, \r
+      // If there was no enough space in buffer, send out the debug message,\r
       // reserving 6 bytes is for the last data and end characters "]\n".\r
       //\r
       SendDebugMsgPacket (Buffer, DestBuffer - Buffer);\r
@@ -587,7 +633,7 @@ ReadRemainingBreakPacket (
   //\r
   // Has received start symbol, try to read the rest part\r
   //\r
-  if (DebugPortReadBuffer (Handle, &DebugHeader->Command, sizeof (DEBUG_PACKET_HEADER) - 1, READ_PACKET_TIMEOUT) == 0) {\r
+  if (DebugAgentReadBuffer (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
@@ -603,19 +649,20 @@ ReadRemainingBreakPacket (
     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
+  if (IS_REQUEST (DebugHeader)) {\r
+    if (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
+    }\r
+    if (DebugHeader->SequenceNo == Mailbox->HostSequenceNo) {\r
+      return EFI_SUCCESS;\r
+    }\r
   }\r
+\r
+  return EFI_DEVICE_ERROR;\r
 }\r
 \r
 /**\r
@@ -637,7 +684,7 @@ IsHostAttached (
   Set HOST connect flag in Mailbox.\r
 \r
   @param[in] Attached        Attach status.\r
-  \r
+\r
 **/\r
 VOID\r
 SetHostAttached (\r
@@ -652,14 +699,14 @@ SetHostAttached (
   Set debug setting of Debug Agent in Mailbox.\r
 \r
   @param DebugSetting         Pointer to Debug Setting defined by transfer protocol.\r
-  \r
+\r
   @retval RETURN_SUCCESS      The setting is set successfully.\r
   @retval RETURN_UNSUPPORTED  The Key value is not supported.\r
 \r
 **/\r
 RETURN_STATUS\r
 SetDebugSetting (\r
-  IN DEBUG_DATA_SET_DEBUG_SETTING  *DebugSetting               \r
+  IN DEBUG_DATA_SET_DEBUG_SETTING  *DebugSetting\r
   )\r
 {\r
   RETURN_STATUS                Status;\r
@@ -700,7 +747,7 @@ CommandGo (
 }\r
 \r
 /**\r
-  Exectue Stepping command.\r
+  Execute Stepping command.\r
 \r
   @param[in] CpuContext        Pointer to saved CPU context.\r
 \r
@@ -715,6 +762,39 @@ CommandStepping (
   Eflags = (IA32_EFLAGS32 *) &CpuContext->Eflags;\r
   Eflags->Bits.TF = 1;\r
   Eflags->Bits.RF = 1;\r
+  //\r
+  // Save and clear EFLAGS.IF to avoid interrupt happen when executing Stepping\r
+  //\r
+  SetDebugFlag (DEBUG_AGENT_FLAG_INTERRUPT_FLAG, Eflags->Bits.IF);\r
+  Eflags->Bits.IF = 0;\r
+  //\r
+  // Set Stepping Flag\r
+  //\r
+  SetDebugFlag (DEBUG_AGENT_FLAG_STEPPING, 1);\r
+}\r
+\r
+/**\r
+  Do some cleanup after Stepping command done.\r
+\r
+  @param[in] CpuContext        Pointer to saved CPU context.\r
+\r
+**/\r
+VOID\r
+CommandSteppingCleanup (\r
+  IN DEBUG_CPU_CONTEXT          *CpuContext\r
+  )\r
+{\r
+  IA32_EFLAGS32                *Eflags;\r
+\r
+  Eflags = (IA32_EFLAGS32 *) &CpuContext->Eflags;\r
+  //\r
+  // Restore EFLAGS.IF\r
+  //\r
+  Eflags->Bits.IF = GetDebugFlag (DEBUG_AGENT_FLAG_INTERRUPT_FLAG);\r
+  //\r
+  // Clear Stepping flag\r
+  //\r
+  SetDebugFlag (DEBUG_AGENT_FLAG_STEPPING, 0);\r
 }\r
 \r
 /**\r
@@ -745,11 +825,11 @@ SetDebugRegister (
   //\r
   // Enable Gx, Lx\r
   //\r
-  Dr7Value |= 0x3 << (RegisterIndex * 2);\r
+  Dr7Value |= (UINTN) (0x3 << (RegisterIndex * 2));\r
   //\r
   // Set RWx and Lenx\r
   //\r
-  Dr7Value &= ~(0xf << (16 + RegisterIndex * 4));\r
+  Dr7Value &= (UINTN) (~(0xf << (16 + RegisterIndex * 4)));\r
   Dr7Value |= (UINTN) ((SetHwBreakpoint->Type.Length << 2) | SetHwBreakpoint->Type.Access) << (16 + RegisterIndex * 4);\r
   //\r
   // Enable GE, LE\r
@@ -774,19 +854,19 @@ ClearDebugRegister (
 {\r
   if ((ClearHwBreakpoint->IndexMask & BIT0) != 0) {\r
     CpuContext->Dr0 = 0;\r
-    CpuContext->Dr7 &= ~(0x3 << 0);\r
+    CpuContext->Dr7 &= (UINTN)(~(0x3 << 0));\r
   }\r
   if ((ClearHwBreakpoint->IndexMask & BIT1) != 0) {\r
     CpuContext->Dr1 = 0;\r
-    CpuContext->Dr7 &= ~(0x3 << 2);\r
+    CpuContext->Dr7 &= (UINTN)(~(0x3 << 2));\r
   }\r
   if ((ClearHwBreakpoint->IndexMask & BIT2) != 0) {\r
     CpuContext->Dr2 = 0;\r
-    CpuContext->Dr7 &= ~(0x3 << 4);\r
+    CpuContext->Dr7 &= (UINTN)(~(0x3 << 4));\r
   }\r
   if ((ClearHwBreakpoint->IndexMask & BIT3) != 0) {\r
     CpuContext->Dr3 = 0;\r
-    CpuContext->Dr7 &= ~(0x3 << 6);\r
+    CpuContext->Dr7 &= (UINTN)(~(0x3 << 6));\r
   }\r
 }\r
 \r
@@ -953,6 +1033,51 @@ SendAckPacket (
   UpdateMailboxContent (Mailbox, DEBUG_MAILBOX_LAST_ACK, AckCommand);\r
 }\r
 \r
+/**\r
+  Decompress the Data in place.\r
+\r
+  @param[in, out] Data   The compressed data buffer.\r
+                         The buffer is assumed large enough to hold the uncompressed data.\r
+  @param[in]      Length The length of the compressed data buffer.\r
+\r
+  @return   The length of the uncompressed data buffer.\r
+**/\r
+UINT8\r
+DecompressDataInPlace (\r
+  IN OUT UINT8   *Data,\r
+  IN UINTN       Length\r
+  )\r
+{\r
+  UINTN  Index;\r
+  UINT16 LastChar;\r
+  UINTN  LastCharCount;\r
+  UINT8  CurrentChar;\r
+\r
+  LastChar = (UINT16) -1;\r
+  LastCharCount = 0;\r
+  for (Index = 0; Index < Length; Index++) {\r
+    CurrentChar = Data[Index];\r
+    if (LastCharCount == 2) {\r
+      LastCharCount = 0;\r
+      CopyMem (&Data[Index + CurrentChar], &Data[Index + 1], Length - Index - 1);\r
+      SetMem (&Data[Index], CurrentChar, (UINT8) LastChar);\r
+      LastChar = (UINT16) -1;\r
+      Index += CurrentChar - 1;\r
+      Length += CurrentChar - 1;\r
+    } else {\r
+      if (LastChar != CurrentChar) {\r
+        LastCharCount = 0;\r
+      }\r
+      LastCharCount++;\r
+      LastChar = CurrentChar;\r
+    }\r
+  }\r
+\r
+  ASSERT (Length <= DEBUG_DATA_MAXIMUM_REAL_DATA);\r
+\r
+  return (UINT8) Length;\r
+}\r
+\r
 /**\r
   Receive valid packet from HOST.\r
 \r
@@ -992,19 +1117,19 @@ ReceivePacket (
   } else {\r
     TimeoutForStartSymbol = Timeout;\r
   }\r
-  \r
+\r
   DebugHeader = (DEBUG_PACKET_HEADER *) InputPacket;\r
   while (TRUE) {\r
     //\r
     // Find the valid start symbol\r
     //\r
-    Received = DebugPortReadBuffer (Handle, &DebugHeader->StartSymbol, sizeof (DebugHeader->StartSymbol), TimeoutForStartSymbol);\r
+    Received = DebugAgentReadBuffer (Handle, &DebugHeader->StartSymbol, sizeof (DebugHeader->StartSymbol), TimeoutForStartSymbol);\r
     if (Received < sizeof (DebugHeader->StartSymbol)) {\r
-      DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "DebugPortReadBuffer(StartSymbol) timeout\n");\r
+      DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "DebugAgentReadBuffer(StartSymbol) timeout\n");\r
       return RETURN_TIMEOUT;\r
     }\r
 \r
-    if (DebugHeader->StartSymbol != DEBUG_STARTING_SYMBOL_NORMAL) {\r
+    if ((DebugHeader->StartSymbol != DEBUG_STARTING_SYMBOL_NORMAL) && (DebugHeader->StartSymbol != DEBUG_STARTING_SYMBOL_COMPRESS)) {\r
       DebugAgentMsgPrint (DEBUG_AGENT_WARNING, "Invalid start symbol received [%02x]\n", DebugHeader->StartSymbol);\r
       continue;\r
     }\r
@@ -1012,14 +1137,14 @@ ReceivePacket (
     //\r
     // Read Package header till field Length\r
     //\r
-    Received = DebugPortReadBuffer (\r
+    Received = DebugAgentReadBuffer (\r
                  Handle,\r
-                 &DebugHeader->Command,\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
+      DebugAgentMsgPrint (DEBUG_AGENT_ERROR, "DebugAgentReadBuffer(Command) timeout\n");\r
       return RETURN_TIMEOUT;\r
     }\r
     if (DebugHeader->Length < sizeof (DEBUG_PACKET_HEADER)) {\r
@@ -1038,9 +1163,9 @@ ReceivePacket (
       //\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
+      Received = DebugAgentReadBuffer (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
+        DebugAgentMsgPrint (DEBUG_AGENT_ERROR, "DebugAgentReadBuffer(SequenceNo) timeout\n");\r
         return RETURN_TIMEOUT;\r
       }\r
       //\r
@@ -1058,6 +1183,12 @@ ReceivePacket (
 \r
   DebugAgentDataMsgPrint (DEBUG_AGENT_VERBOSE, FALSE, (UINT8 *) DebugHeader, DebugHeader->Length);\r
 \r
+  if (DebugHeader->StartSymbol == DEBUG_STARTING_SYMBOL_COMPRESS) {\r
+    DebugHeader->StartSymbol = DEBUG_STARTING_SYMBOL_NORMAL;\r
+    DebugHeader->Length      = DecompressDataInPlace (\r
+                                 (UINT8 *) (DebugHeader + 1), DebugHeader->Length - sizeof (DEBUG_PACKET_HEADER)\r
+                                 ) + sizeof (DEBUG_PACKET_HEADER);\r
+  }\r
   return RETURN_SUCCESS;\r
 }\r
 \r
@@ -1166,8 +1297,12 @@ GetBreakCause (
       if ((CpuContext->Dr6 & BIT14) != 0) {\r
         Cause = DEBUG_DATA_BREAK_CAUSE_STEPPING;\r
         //\r
-        // If it's single step, no need to check DR0, to ensure single step work in PeCoffExtraActionLib\r
-        // (right after triggering a breakpoint to report image load/unload).\r
+        // DR6.BIT14 Indicates (when set) that the debug exception was\r
+        // triggered by the single step execution mode.\r
+        // The single-step mode is the highest priority debug exception.\r
+        // This is single step, no need to check DR0, to ensure single step\r
+        // work in PeCoffExtraActionLib (right after triggering a breakpoint\r
+        // to report image load/unload).\r
         //\r
         return Cause;\r
 \r
@@ -1187,13 +1322,13 @@ GetBreakCause (
 \r
       if (CpuContext->Dr3 == IO_PORT_BREAKPOINT_ADDRESS) {\r
 \r
-        Cause = (UINT8) ((CpuContext->Dr0 == IMAGE_LOAD_SIGNATURE) ? \r
+        Cause = (UINT8) ((CpuContext->Dr0 == IMAGE_LOAD_SIGNATURE) ?\r
           DEBUG_DATA_BREAK_CAUSE_IMAGE_LOAD : DEBUG_DATA_BREAK_CAUSE_IMAGE_UNLOAD);\r
       }\r
       break;\r
 \r
     case SOFT_INTERRUPT_SIGNATURE:\r
-   \r
+\r
       if (CpuContext->Dr1 == MEMORY_READY_SIGNATURE) {\r
         Cause = DEBUG_DATA_BREAK_CAUSE_MEMORY_READY;\r
         CpuContext->Dr0 = 0;\r
@@ -1287,44 +1422,116 @@ CopyMemByWidth (
   }\r
 }\r
 \r
+/**\r
+  Compress the data buffer but do not modify the original buffer.\r
+\r
+  The compressed data is directly send to the debug channel.\r
+  Compressing in place doesn't work because the data may become larger\r
+  during compressing phase. ("3 3 ..." --> "3 3 0 ...")\r
+  The routine is expected to be called three times:\r
+  1. Compute the length of the compressed data buffer;\r
+  2. Compute the CRC of the compressed data buffer;\r
+  3. Compress the data and send to the debug channel.\r
+\r
+  @param[in]  Handle           The debug channel handle to send the compressed data buffer.\r
+  @param[in]  Data             The data buffer.\r
+  @param[in]  Length           The length of the data buffer.\r
+  @param[in]  Send             TRUE to send the compressed data buffer.\r
+  @param[out] CompressedLength Return the length of the compressed data buffer.\r
+                               It may be larger than the Length in some cases.\r
+  @param[out] CompressedCrc    Return the CRC of the compressed data buffer.\r
+**/\r
+VOID\r
+CompressData (\r
+  IN  DEBUG_PORT_HANDLE Handle,\r
+  IN  UINT8             *Data,\r
+  IN  UINT8             Length,\r
+  IN  BOOLEAN           Send,\r
+  OUT UINTN             *CompressedLength,  OPTIONAL\r
+  OUT UINT16            *CompressedCrc      OPTIONAL\r
+  )\r
+{\r
+  UINTN                 Index;\r
+  UINT8                 LastChar;\r
+  UINT8                 LastCharCount;\r
+  UINT8                 CurrentChar;\r
+  UINTN                 CompressedIndex;\r
+\r
+  ASSERT (Length > 0);\r
+  LastChar      = Data[0] + 1; // Just ensure it's different from the first byte.\r
+  LastCharCount = 0;\r
+\r
+  for (Index = 0, CompressedIndex = 0; Index <= Length; Index++) {\r
+    if (Index < Length) {\r
+      CurrentChar = Data[Index];\r
+    } else {\r
+      CurrentChar = (UINT8) LastChar + 1; // just ensure it's different from LastChar\r
+    }\r
+    if (LastChar != CurrentChar) {\r
+      if (LastCharCount == 1) {\r
+        CompressedIndex++;\r
+        if (CompressedCrc != NULL) {\r
+          *CompressedCrc = CalculateCrc16 (&LastChar, 1, *CompressedCrc);\r
+        }\r
+        if (Send) {\r
+          DebugPortWriteBuffer (Handle, &LastChar, 1);\r
+        }\r
+        \r
+      } else if (LastCharCount >= 2) {\r
+        CompressedIndex += 3;\r
+        LastCharCount -= 2;\r
+        if (CompressedCrc != NULL) {\r
+          *CompressedCrc = CalculateCrc16 (&LastChar, 1, *CompressedCrc);\r
+          *CompressedCrc = CalculateCrc16 (&LastChar, 1, *CompressedCrc);\r
+          *CompressedCrc = CalculateCrc16 (&LastCharCount, 1, *CompressedCrc);\r
+        }\r
+        if (Send) {\r
+          DebugPortWriteBuffer (Handle, &LastChar, 1);\r
+          DebugPortWriteBuffer (Handle, &LastChar, 1);\r
+          DebugPortWriteBuffer (Handle, &LastCharCount, 1);\r
+        }\r
+      }\r
+      LastCharCount = 0;\r
+    }\r
+    LastCharCount++;\r
+    LastChar = CurrentChar;\r
+  }\r
+\r
+  if (CompressedLength != NULL) {\r
+    *CompressedLength = CompressedIndex;\r
+  }\r
+}\r
+\r
 /**\r
   Read memory with speicifed width and send packet with response data to HOST.\r
 \r
   @param[in] Data        Pointer to response data buffer.\r
   @param[in] Count       The number of data with specified Width.\r
   @param[in] Width       Data width in byte.\r
+  @param[in] DebugHeader Pointer to a buffer for creating response packet and receiving ACK packet,\r
+                         to minimize the stack usage.\r
 \r
   @retval RETURN_SUCCESS      Response data was sent successfully.\r
 \r
 **/\r
 RETURN_STATUS\r
 ReadMemoryAndSendResponsePacket (\r
-  IN UINT8                *Data,\r
-  IN UINT16               Count,\r
-  IN UINT8                Width\r
+  IN UINT8                   *Data,\r
+  IN UINT16                  Count,\r
+  IN UINT8                   Width,\r
+  IN DEBUG_PACKET_HEADER     *DebugHeader\r
   )\r
 {\r
   RETURN_STATUS        Status;\r
-  DEBUG_PACKET_HEADER  *DebugHeader;\r
   BOOLEAN              LastPacket;\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
+  UINT8                CurrentDataSize;\r
+  UINTN                CompressedDataSize;\r
 \r
   Handle = GetDebugPortHandle();\r
 \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
     SequenceNo = GetMailboxPointer()->HostSequenceNo;\r
@@ -1332,7 +1539,7 @@ ReadMemoryAndSendResponsePacket (
       //\r
       // If the remaining data is less one real packet size, this is the last data packet\r
       //\r
-      CurrentDataSize = RemainingDataSize;\r
+      CurrentDataSize = (UINT8) RemainingDataSize;\r
       LastPacket = TRUE;\r
       DebugHeader->Command = DEBUG_COMMAND_OK;\r
     } else {\r
@@ -1347,45 +1554,95 @@ ReadMemoryAndSendResponsePacket (
     //\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
+    DebugHeader->StartSymbol = DEBUG_STARTING_SYMBOL_NORMAL;\r
+    DebugHeader->Length      = CurrentDataSize + sizeof (DEBUG_PACKET_HEADER);\r
+    DebugHeader->SequenceNo  = SequenceNo;\r
+    DebugHeader->Crc         = 0;\r
+    CopyMemByWidth ((UINT8 *) (DebugHeader + 1), Data, CurrentDataSize / Width, Width);\r
+\r
     //\r
-    // Calculate and fill the checksum, DebugHeader->Crc should be 0 before invoking CalculateCrc16 ()\r
+    // Compression/decompression support was added since revision 0.4.\r
+    // Revision 0.3 shouldn't compress the packet.\r
     //\r
-    DebugHeader->Crc = CalculateCrc16 ((UINT8 *) DebugHeader, DebugHeader->Length, 0);\r
+    if (DEBUG_AGENT_REVISION >= DEBUG_AGENT_REVISION_04) {\r
+      //\r
+      // Get the compressed data size without modifying the packet.\r
+      //\r
+      CompressData (\r
+        Handle,\r
+        (UINT8 *) (DebugHeader + 1),\r
+        CurrentDataSize,\r
+        FALSE,\r
+        &CompressedDataSize,\r
+        NULL\r
+        );\r
+    } else {\r
+      CompressedDataSize = CurrentDataSize;\r
+    }\r
+    if (CompressedDataSize < CurrentDataSize) {\r
+      DebugHeader->Length = (UINT8) CompressedDataSize + sizeof (DEBUG_PACKET_HEADER);\r
+      DebugHeader->StartSymbol = DEBUG_STARTING_SYMBOL_COMPRESS;\r
+      //\r
+      // Compute the CRC of the packet head without modifying the packet.\r
+      //\r
+      DebugHeader->Crc = CalculateCrc16 ((UINT8 *) DebugHeader, sizeof (DEBUG_PACKET_HEADER), 0);\r
+      CompressData (\r
+        Handle,\r
+        (UINT8 *) (DebugHeader + 1),\r
+        CurrentDataSize,\r
+        FALSE,\r
+        NULL,\r
+        &DebugHeader->Crc\r
+        );\r
+      //\r
+      // Send out the packet head.\r
+      //\r
+      DebugPortWriteBuffer (Handle, (UINT8 *) DebugHeader, sizeof (DEBUG_PACKET_HEADER));\r
+      //\r
+      // Compress and send out the packet data.\r
+      //\r
+      CompressData (\r
+        Handle,\r
+        (UINT8 *) (DebugHeader + 1),\r
+        CurrentDataSize,\r
+        TRUE,\r
+        NULL,\r
+        NULL\r
+        );\r
+    } else {\r
 \r
-    DebugAgentDataMsgPrint (DEBUG_AGENT_VERBOSE, TRUE, (UINT8 *) DebugHeader, DebugHeader->Length);\r
-    \r
-    DebugPortWriteBuffer (Handle, (UINT8 *) DebugHeader, DebugHeader->Length);\r
+      //\r
+      // Calculate and fill the checksum, DebugHeader->Crc should be 0 before invoking CalculateCrc16 ()\r
+      //\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
 \r
     while (TRUE) {\r
-      Status = ReceivePacket (InputPacketBuffer, NULL, NULL, READ_PACKET_TIMEOUT, FALSE);\r
+      Status = ReceivePacket ((UINT8 *) DebugHeader, 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
+      if ((DebugHeader->Command == DEBUG_COMMAND_OK) && (DebugHeader->SequenceNo == SequenceNo) && LastPacket) {\r
         //\r
         // If this is the last packet, return RETURN_SUCCESS.\r
         //\r
         return RETURN_SUCCESS;\r
       }\r
-      if ((SequenceNo == (UINT8) (DebugHeader->SequenceNo + 1)) && (AckDebugHeader->Command == DEBUG_COMMAND_CONTINUE)) {\r
+      if ((DebugHeader->Command == DEBUG_COMMAND_CONTINUE) && (DebugHeader->SequenceNo == (UINT8) (SequenceNo + 1))) {\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
+        UpdateMailboxContent (GetMailboxPointer(), DEBUG_MAILBOX_HOST_SEQUENCE_NO_INDEX, DebugHeader->SequenceNo);\r
         break;\r
       }\r
-      if (SequenceNo >= DebugHeader->SequenceNo) {\r
+      if (DebugHeader->SequenceNo >= 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
@@ -1396,53 +1653,32 @@ ReadMemoryAndSendResponsePacket (
 /**\r
   Send packet with response data to HOST.\r
 \r
-  @param[in] Data        Pointer to response data buffer.\r
-  @param[in] DataSize    Size of response data in byte.\r
+  @param[in]      Data        Pointer to response data buffer.\r
+  @param[in]      DataSize    Size of response data in byte.\r
+  @param[in, out] DebugHeader Pointer to a buffer for creating response packet and receiving ACK packet,\r
+                              to minimize the stack usage.\r
 \r
   @retval RETURN_SUCCESS      Response data was sent successfully.\r
 \r
 **/\r
 RETURN_STATUS\r
 SendDataResponsePacket (\r
-  IN UINT8                *Data,\r
-  IN UINT16               DataSize\r
-  )\r
-{\r
-  return ReadMemoryAndSendResponsePacket (Data, DataSize, 1);\r
-}\r
-\r
-/**\r
-  Send break cause packet to HOST.\r
-\r
-  @param[in] Vector      Vector value of exception or interrutp.\r
-  @param[in] CpuContext  Pointer to save CPU context.\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
-SendBreakCausePacket (\r
-  IN UINTN                    Vector,\r
-  IN DEBUG_CPU_CONTEXT        *CpuContext\r
+  IN UINT8                   *Data,\r
+  IN UINT16                  DataSize,\r
+  IN OUT DEBUG_PACKET_HEADER *DebugHeader\r
   )\r
 {\r
-  DEBUG_DATA_RESPONSE_BREAK_CAUSE    DebugDataBreakCause;\r
-\r
-  DebugDataBreakCause.StopAddress = CpuContext->Eip;\r
-  DebugDataBreakCause.Cause       = GetBreakCause (Vector, CpuContext);\r
-\r
-  return SendDataResponsePacket ((UINT8 *) &DebugDataBreakCause, (UINT16) sizeof (DEBUG_DATA_RESPONSE_BREAK_CAUSE));\r
+  return ReadMemoryAndSendResponsePacket (Data, DataSize, 1, DebugHeader);\r
 }\r
 \r
 /**\r
   Try to attach the HOST.\r
-  \r
+\r
   Send init break packet to HOST:\r
-  If no acknowlege received in specified Timeout, return RETURN_TIMEOUT. \r
-  If received acknowlege, check the revision of HOST. \r
-  Set Attach Flag if attach successfully.  \r
-  \r
+  If no acknowlege received in specified Timeout, return RETURN_TIMEOUT.\r
+  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
@@ -1463,7 +1699,7 @@ AttachHost (
 \r
   IncompatibilityFlag = FALSE;\r
   Handle = GetDebugPortHandle();\r
-    \r
+\r
   //\r
   // Send init break and wait ack in Timeout\r
   //\r
@@ -1481,7 +1717,7 @@ AttachHost (
     DebugPortWriteBuffer (Handle, (UINT8 *) mErrorMsgVersionAlert, AsciiStrLen (mErrorMsgVersionAlert));\r
     CpuDeadLoop ();\r
   }\r
-  \r
+\r
   if (RETURN_ERROR (Status)) {\r
     DebugPortWriteBuffer (Handle, (UINT8 *) mErrorMsgConnectFail, AsciiStrLen (mErrorMsgConnectFail));\r
   } else {\r
@@ -1495,8 +1731,8 @@ AttachHost (
 }\r
 \r
 /**\r
-  Send Break point packet to HOST. \r
-  \r
+  Send Break point packet to HOST.\r
+\r
   Only the first breaking processor could sent BREAK_POINT packet.\r
 \r
   @param[in]  BreakCause     Break cause of this break event.\r
@@ -1504,7 +1740,7 @@ AttachHost (
   @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
-                            \r
+\r
 **/\r
 VOID\r
 SendBreakPacketToHost (\r
@@ -1515,9 +1751,9 @@ SendBreakPacketToHost (
 {\r
   UINT8                 InputCharacter;\r
   DEBUG_PORT_HANDLE     Handle;\r
-  \r
+\r
   Handle = GetDebugPortHandle();\r
-  \r
+\r
   if (IsHostAttached ()) {\r
     DebugAgentMsgPrint (DEBUG_AGENT_INFO, "processor[%x]:Send Break Packet to HOST.\n", ProcessorIndex);\r
     SendCommandAndWaitForAckOK (DEBUG_COMMAND_BREAK_POINT, READ_PACKET_TIMEOUT, BreakReceived, NULL);\r
@@ -1528,17 +1764,17 @@ SendBreakPacketToHost (
     //\r
     //\r
     // Poll Attach symbols from HOST and ack OK\r
-    //  \r
+    //\r
     do {\r
-      DebugPortReadBuffer (Handle, &InputCharacter, 1, 0);\r
+      DebugAgentReadBuffer (Handle, &InputCharacter, 1, 0);\r
     } while (InputCharacter != DEBUG_STARTING_SYMBOL_ATTACH);\r
     SendAckPacket (DEBUG_COMMAND_OK);\r
-    \r
+\r
     //\r
     // Try to attach HOST\r
     //\r
     while (AttachHost (BreakCause, 0, NULL) != RETURN_SUCCESS);\r
-   \r
+\r
   }\r
 }\r
 \r
@@ -1577,6 +1813,7 @@ CommandCommunication (
   DEBUG_DATA_READ_MSR               *MsrRegisterRead;\r
   DEBUG_DATA_WRITE_MSR              *MsrRegisterWrite;\r
   DEBUG_DATA_CPUID                  *Cpuid;\r
+  DEBUG_DATA_RESPONSE_BREAK_CAUSE   BreakCause;\r
   DEBUG_DATA_RESPONSE_CPUID         CpuidResponse;\r
   DEBUG_DATA_SEARCH_SIGNATURE       *SearchSignature;\r
   DEBUG_DATA_RESPONSE_GET_EXCEPTION Exception;\r
@@ -1584,7 +1821,6 @@ CommandCommunication (
   DEBUG_DATA_SET_VIEWPOINT          *SetViewPoint;\r
   BOOLEAN                           HaltDeferred;\r
   UINT32                            ProcessorIndex;\r
-  DEBUG_PORT_HANDLE                 Handle;\r
   DEBUG_AGENT_EXCEPTION_BUFFER      AgentExceptionBuffer;\r
   UINT32                            IssuedViewPoint;\r
   DEBUG_AGENT_MAILBOX               *Mailbox;\r
@@ -1599,7 +1835,7 @@ CommandCommunication (
     SetCpuStopFlagByIndex (ProcessorIndex, TRUE);\r
     if (mDebugMpContext.ViewPointIndex == ProcessorIndex) {\r
       //\r
-      // Only the current view processor could set AgentInProgress Flag. \r
+      // Only the current view processor could set AgentInProgress Flag.\r
       //\r
       IssuedViewPoint = ProcessorIndex;\r
     }\r
@@ -1610,9 +1846,7 @@ CommandCommunication (
     // Set AgentInProgress Flag.\r
     //\r
     SetDebugFlag (DEBUG_AGENT_FLAG_AGENT_IN_PROGRESS, 1);\r
-  }  \r
-\r
-  Handle = GetDebugPortHandle();\r
+  }\r
 \r
   while (TRUE) {\r
 \r
@@ -1643,8 +1877,8 @@ CommandCommunication (
     DebugHeader =(DEBUG_PACKET_HEADER *) InputPacketBuffer;\r
 \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
+    Status = ReceivePacket ((UINT8 *) DebugHeader, &BreakReceived, NULL, READ_PACKET_TIMEOUT, TRUE);\r
+    if (Status != RETURN_SUCCESS || !IS_REQUEST (DebugHeader)) {\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
@@ -1664,7 +1898,7 @@ CommandCommunication (
       ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
       continue;\r
     }\r
-    \r
+\r
     //\r
     // Save CPU content before executing HOST commond\r
     //\r
@@ -1680,7 +1914,7 @@ CommandCommunication (
     }\r
 \r
     DebugAgentMsgPrint (DEBUG_AGENT_INFO, "Processor[%x]:Received one command(%x)\n", mDebugMpContext.ViewPointIndex, DebugHeader->Command);\r
-    \r
+\r
     switch (DebugHeader->Command) {\r
 \r
     case DEBUG_COMMAND_HALT:\r
@@ -1718,11 +1952,7 @@ CommandCommunication (
       if (Data8 == DEBUG_DATA_BREAK_CAUSE_IMAGE_LOAD || Data8 == DEBUG_DATA_BREAK_CAUSE_IMAGE_UNLOAD) {\r
         CpuContext->Dr0 = 0;\r
       }\r
-      //\r
-      // Clear Stepping Flag\r
-      //\r
-      SetDebugFlag (DEBUG_AGENT_FLAG_STEPPING, 0);\r
-      \r
+\r
       if (!HaltDeferred) {\r
         //\r
         // If no HALT command received when being in-active mode\r
@@ -1731,7 +1961,7 @@ CommandCommunication (
           Data32 = FindNextPendingBreakCpu ();\r
           if (Data32 != -1) {\r
             //\r
-            // If there are still others processors being in break state,          \r
+            // If there are still others processors being in break state,\r
             // send OK packet to HOST to finish this go command\r
             //\r
             SendAckPacket (DEBUG_COMMAND_OK);\r
@@ -1761,7 +1991,7 @@ CommandCommunication (
           SetCpuRunningFlag (TRUE);\r
           CpuPause ();\r
           //\r
-          // Wait for all processors are in running state \r
+          // Wait for all processors are in running state\r
           //\r
           while (TRUE) {\r
             if (IsAllCpuRunning ()) {\r
@@ -1805,14 +2035,13 @@ CommandCommunication (
       break;\r
 \r
     case DEBUG_COMMAND_BREAK_CAUSE:\r
-\r
+      BreakCause.StopAddress = CpuContext->Eip;\r
       if (MultiProcessorDebugSupport() && ProcessorIndex != mDebugMpContext.BreakAtCpuIndex) {\r
-        Status = SendBreakCausePacket (DEBUG_TIMER_VECTOR, CpuContext);\r
-\r
+        BreakCause.Cause       = GetBreakCause (DEBUG_TIMER_VECTOR, CpuContext);\r
       } else {\r
-        Status = SendBreakCausePacket (Vector, CpuContext);\r
+        BreakCause.Cause       = GetBreakCause (Vector, CpuContext);\r
       }\r
-\r
+      SendDataResponsePacket ((UINT8 *) &BreakCause, (UINT16) sizeof (DEBUG_DATA_RESPONSE_BREAK_CAUSE), DebugHeader);\r
       break;\r
 \r
     case DEBUG_COMMAND_SET_HW_BREAKPOINT:\r
@@ -1837,10 +2066,6 @@ CommandCommunication (
       }\r
 \r
       mDebugMpContext.BreakAtCpuIndex = (UINT32) (-1);\r
-      //\r
-      // Set Stepping Flag\r
-      //\r
-      SetDebugFlag (DEBUG_AGENT_FLAG_STEPPING, 1);\r
       ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
       //\r
       // Executing stepping command directly without sending ACK packet,\r
@@ -1852,12 +2077,12 @@ CommandCommunication (
       Data64 = (UINTN) (((DEBUG_DATA_SET_SW_BREAKPOINT *) (DebugHeader + 1))->Address);\r
       Data8 = *(UINT8 *) (UINTN) Data64;\r
       *(UINT8 *) (UINTN) Data64 = DEBUG_SW_BREAKPOINT_SYMBOL;\r
-      Status = SendDataResponsePacket ((UINT8 *) &Data8, (UINT16) sizeof (UINT8));\r
+      Status = SendDataResponsePacket ((UINT8 *) &Data8, (UINT16) sizeof (UINT8), DebugHeader);\r
       break;\r
 \r
     case DEBUG_COMMAND_READ_MEMORY:\r
       MemoryRead = (DEBUG_DATA_READ_MEMORY *) (DebugHeader + 1);\r
-      Status = ReadMemoryAndSendResponsePacket ((UINT8 *) (UINTN) MemoryRead->Address, MemoryRead->Count, MemoryRead->Width);\r
+      Status = ReadMemoryAndSendResponsePacket ((UINT8 *) (UINTN) MemoryRead->Address, MemoryRead->Count, MemoryRead->Width, DebugHeader);\r
       break;\r
 \r
     case DEBUG_COMMAND_WRITE_MEMORY:\r
@@ -1891,7 +2116,7 @@ CommandCommunication (
       default:\r
         Data64  = (UINT64) -1;\r
       }\r
-      Status = SendDataResponsePacket ((UINT8 *) &Data64, IoRead->Width);\r
+      Status = SendDataResponsePacket ((UINT8 *) &Data64, IoRead->Width, DebugHeader);\r
       break;\r
 \r
     case DEBUG_COMMAND_WRITE_IO:\r
@@ -1916,7 +2141,7 @@ CommandCommunication (
       break;\r
 \r
     case DEBUG_COMMAND_READ_ALL_REGISTERS:\r
-      Status = SendDataResponsePacket ((UINT8 *) CpuContext, sizeof (*CpuContext));\r
+      Status = SendDataResponsePacket ((UINT8 *) CpuContext, sizeof (*CpuContext), DebugHeader);\r
       break;\r
 \r
     case DEBUG_COMMAND_READ_REGISTER:\r
@@ -1924,7 +2149,7 @@ CommandCommunication (
 \r
       if (RegisterRead->Index <= SOFT_DEBUGGER_REGISTER_MAX) {\r
         RegisterBuffer = ArchReadRegisterBuffer (CpuContext, RegisterRead->Index, &Width);\r
-        Status = SendDataResponsePacket (RegisterBuffer, Width);\r
+        Status = SendDataResponsePacket (RegisterBuffer, Width, DebugHeader);\r
       } else {\r
         Status = RETURN_UNSUPPORTED;\r
       }\r
@@ -1944,13 +2169,13 @@ CommandCommunication (
 \r
     case DEBUG_COMMAND_ARCH_MODE:\r
       Data8 = DEBUG_ARCH_SYMBOL;\r
-      Status = SendDataResponsePacket ((UINT8 *) &Data8, (UINT16) sizeof (UINT8));\r
+      Status = SendDataResponsePacket ((UINT8 *) &Data8, (UINT16) sizeof (UINT8), DebugHeader);\r
       break;\r
 \r
     case DEBUG_COMMAND_READ_MSR:\r
       MsrRegisterRead = (DEBUG_DATA_READ_MSR *) (DebugHeader + 1);\r
       Data64 = AsmReadMsr64 (MsrRegisterRead->Index);\r
-      Status = SendDataResponsePacket ((UINT8 *) &Data64, (UINT16) sizeof (UINT64));\r
+      Status = SendDataResponsePacket ((UINT8 *) &Data64, (UINT16) sizeof (UINT64), DebugHeader);\r
       break;\r
 \r
     case DEBUG_COMMAND_WRITE_MSR:\r
@@ -1969,13 +2194,13 @@ CommandCommunication (
     case DEBUG_COMMAND_GET_REVISION:\r
       DebugAgentRevision.Revision = DEBUG_AGENT_REVISION;\r
       DebugAgentRevision.Capabilities = DEBUG_AGENT_CAPABILITIES;\r
-      Status = SendDataResponsePacket ((UINT8 *) &DebugAgentRevision, (UINT16) sizeof (DEBUG_DATA_RESPONSE_GET_REVISION));\r
+      Status = SendDataResponsePacket ((UINT8 *) &DebugAgentRevision, (UINT16) sizeof (DEBUG_DATA_RESPONSE_GET_REVISION), DebugHeader);\r
       break;\r
 \r
     case DEBUG_COMMAND_GET_EXCEPTION:\r
       Exception.ExceptionNum  = (UINT8) Vector;\r
       Exception.ExceptionData = (UINT32) CpuContext->ExceptionData;\r
-      Status = SendDataResponsePacket ((UINT8 *) &Exception, (UINT16) sizeof (DEBUG_DATA_RESPONSE_GET_EXCEPTION));\r
+      Status = SendDataResponsePacket ((UINT8 *) &Exception, (UINT16) sizeof (DEBUG_DATA_RESPONSE_GET_EXCEPTION), DebugHeader);\r
       break;\r
 \r
     case DEBUG_COMMAND_SET_VIEWPOINT:\r
@@ -2001,12 +2226,12 @@ CommandCommunication (
 \r
     case DEBUG_COMMAND_GET_VIEWPOINT:\r
       Data32 = mDebugMpContext.ViewPointIndex;\r
-      SendDataResponsePacket((UINT8 *) &Data32, (UINT16) sizeof (UINT32));\r
+      SendDataResponsePacket((UINT8 *) &Data32, (UINT16) sizeof (UINT32), DebugHeader);\r
       break;\r
 \r
     case DEBUG_COMMAND_MEMORY_READY:\r
       Data8 = (UINT8) GetDebugFlag (DEBUG_AGENT_FLAG_MEMORY_READY);\r
-      SendDataResponsePacket (&Data8, (UINT16) sizeof (UINT8));\r
+      SendDataResponsePacket (&Data8, (UINT16) sizeof (UINT8), DebugHeader);\r
       break;\r
 \r
     case DEBUG_COMMAND_DETACH:\r
@@ -2017,16 +2242,16 @@ CommandCommunication (
     case DEBUG_COMMAND_CPUID:\r
       Cpuid = (DEBUG_DATA_CPUID *) (DebugHeader + 1);\r
       AsmCpuidEx (\r
-        Cpuid->Eax, Cpuid->Ecx, \r
+        Cpuid->Eax, Cpuid->Ecx,\r
         &CpuidResponse.Eax, &CpuidResponse.Ebx,\r
         &CpuidResponse.Ecx, &CpuidResponse.Edx\r
         );\r
-      SendDataResponsePacket ((UINT8 *) &CpuidResponse, (UINT16) sizeof (CpuidResponse));\r
+      SendDataResponsePacket ((UINT8 *) &CpuidResponse, (UINT16) sizeof (CpuidResponse), DebugHeader);\r
       break;\r
 \r
    case DEBUG_COMMAND_SEARCH_SIGNATURE:\r
       SearchSignature = (DEBUG_DATA_SEARCH_SIGNATURE *) (DebugHeader + 1);\r
-      if ((SearchSignature->Alignment != 0) && \r
+      if ((SearchSignature->Alignment != 0) &&\r
           (SearchSignature->Alignment == GetPowerOfTwo32 (SearchSignature->Alignment))\r
          ) {\r
         if (SearchSignature->Positive) {\r
@@ -2056,7 +2281,7 @@ CommandCommunication (
             Data64 = (UINT64) -1;\r
           }\r
         }\r
-        SendDataResponsePacket ((UINT8 *) &Data64, (UINT16) sizeof (Data64));\r
+        SendDataResponsePacket ((UINT8 *) &Data64, (UINT16) sizeof (Data64), DebugHeader);\r
       } else {\r
         Status = RETURN_UNSUPPORTED;\r
       }\r
@@ -2104,6 +2329,7 @@ InterruptProcess (
   UINT32                           IssuedViewPoint;\r
   DEBUG_AGENT_EXCEPTION_BUFFER     *ExceptionBuffer;\r
 \r
+  InputCharacter  = 0;\r
   ProcessorIndex  = 0;\r
   IssuedViewPoint = 0;\r
   BreakReceived   = FALSE;\r
@@ -2133,9 +2359,16 @@ InterruptProcess (
     // 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
+    // If exception is issued when executing Stepping, will be handled in\r
+    // exception handle procedure.\r
     //\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
+      DebugAgentMsgPrint (\r
+        DEBUG_AGENT_ERROR,\r
+        "Debug agent meet one Exception, ExceptionNum is %d, EIP = 0x%x.\n",\r
+        Vector,\r
+        (UINTN)CpuContext->Eip\r
+        );\r
       ExceptionBuffer = (DEBUG_AGENT_EXCEPTION_BUFFER *) (UINTN) GetMailboxPointer()->ExceptionBufferPointer;\r
       ExceptionBuffer->ExceptionContent.ExceptionNum  = (UINT8) Vector;\r
       ExceptionBuffer->ExceptionContent.ExceptionData = (UINT32) CpuContext->ExceptionData;\r
@@ -2145,7 +2378,7 @@ InterruptProcess (
 \r
   if (MultiProcessorDebugSupport()) {\r
     //\r
-    // If RUN commmand is executing, wait for it done.  \r
+    // If RUN commmand is executing, wait for it done.\r
     //\r
     while (mDebugMpContext.RunCommandSet) {\r
       CpuPause ();\r
@@ -2175,6 +2408,10 @@ InterruptProcess (
       if (MultiProcessorDebugSupport()) {\r
         mDebugMpContext.BreakAtCpuIndex = ProcessorIndex;\r
       }\r
+      //\r
+      // Clear Stepping Flag and restore EFLAGS.IF\r
+      //\r
+      CommandSteppingCleanup (CpuContext);\r
       SendAckPacket (DEBUG_COMMAND_OK);\r
       CommandCommunication (Vector, CpuContext, BreakReceived);\r
       break;\r
@@ -2204,7 +2441,7 @@ InterruptProcess (
       //\r
       // Continue to run the following common code\r
       //\r
-      \r
+\r
     case DEBUG_DATA_BREAK_CAUSE_HW_BREAKPOINT:\r
     case DEBUG_DATA_BREAK_CAUSE_SW_BREAKPOINT:\r
     default:\r
@@ -2214,12 +2451,12 @@ InterruptProcess (
       AcquireMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
       //\r
       // Only the first breaking processor could send BREAK_POINT to HOST\r
-      // \r
+      //\r
       if (IsFirstBreakProcessor (ProcessorIndex)) {\r
         SendBreakPacketToHost (BreakCause, ProcessorIndex, &BreakReceived);\r
       }\r
       ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
-      \r
+\r
       if (Vector == DEBUG_INT3_VECTOR) {\r
         //\r
         // go back address located "0xCC"\r
@@ -2256,7 +2493,8 @@ InterruptProcess (
         //\r
         CurrentDebugTimerInitCount = GetApicTimerInitCount ();\r
         if (mDebugMpContext.DebugTimerInitCount != CurrentDebugTimerInitCount) {\r
-          InitializeDebugTimer ();\r
+          InitializeDebugTimer (NULL, FALSE);\r
+          SaveAndSetDebugTimerInterrupt (TRUE);\r
         }\r
       }\r
 \r
@@ -2341,13 +2579,24 @@ InterruptProcess (
 \r
   default:\r
     if (Vector <= DEBUG_EXCEPT_SIMD) {\r
+      DebugAgentMsgPrint (\r
+        DEBUG_AGENT_ERROR,\r
+        "Exception happened, ExceptionNum is %d, EIP = 0x%x.\n",\r
+        Vector,\r
+        (UINTN) CpuContext->Eip\r
+        );\r
       if (BreakCause == DEBUG_DATA_BREAK_CAUSE_STEPPING) {\r
         //\r
-        // Stepping is finished, send Ack package.\r
+        // If exception happened when executing Stepping, send Ack package.\r
+        // HOST consider Stepping command was finished.\r
         //\r
         if (MultiProcessorDebugSupport()) {\r
           mDebugMpContext.BreakAtCpuIndex = ProcessorIndex;\r
         }\r
+        //\r
+        // Clear Stepping flag and restore EFLAGS.IF\r
+        //\r
+        CommandSteppingCleanup (CpuContext);\r
         SendAckPacket (DEBUG_COMMAND_OK);\r
       } else {\r
         //\r
@@ -2356,13 +2605,13 @@ InterruptProcess (
         AcquireMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
         //\r
         // Only the first breaking processor could send BREAK_POINT to HOST\r
-        // \r
+        //\r
         if (IsFirstBreakProcessor (ProcessorIndex)) {\r
           SendBreakPacketToHost (BreakCause, ProcessorIndex, &BreakReceived);\r
         }\r
         ReleaseMpSpinLock (&mDebugMpContext.DebugPortSpinLock);\r
       }\r
-      \r
+\r
       CommandCommunication (Vector, CpuContext, BreakReceived);\r
     }\r
     break;\r