3 files changed, 157 insertions, 0 deletions
diff --git a/SourceLevelDebugPkg/Library/PeCoffExtraActionLibDebug/Ia32/IntHandler.S b/SourceLevelDebugPkg/Library/PeCoffExtraActionLibDebug/Ia32/IntHandler.S
new file mode 100644
index 0000000000..69c903788c
--- /dev/null
+++ b/SourceLevelDebugPkg/Library/PeCoffExtraActionLibDebug/Ia32/IntHandler.S
@@ -0,0 +1,28 @@
+#------------------------------------------------------------------------------
+#
+# Copyright (c) 2013, Intel Corporation. All rights reserved.<BR>
+# This program and the accompanying materials
+# are licensed and made available under the terms and conditions of the BSD License
+# which accompanies this distribution.  The full text of the license may be found at
+# http://opensource.org/licenses/bsd-license.php.
+#
+# THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
+# WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
+#
+# Module Name:
+#
+#   IntHandler.S
+#
+# Abstract:
+#
+#   Assembly interrupt handler function.
+#
+#------------------------------------------------------------------------------
+
+ASM_GLOBAL ASM_PFX(AsmInterruptHandle)
+
+.text
+ASM_PFX(AsmInterruptHandle):
+   cli
+   movb   $1, %al
+   iretl
diff --git a/SourceLevelDebugPkg/Library/PeCoffExtraActionLibDebug/Ia32/IntHandler.asm b/SourceLevelDebugPkg/Library/PeCoffExtraActionLibDebug/Ia32/IntHandler.asm
new file mode 100644
index 0000000000..ce9c03b490
--- /dev/null
+++ b/SourceLevelDebugPkg/Library/PeCoffExtraActionLibDebug/Ia32/IntHandler.asm
@@ -0,0 +1,33 @@
+;------------------------------------------------------------------------------
+;
+; Copyright (c) 2013, Intel Corporation. All rights reserved.<BR>
+; This program and the accompanying materials
+; are licensed and made available under the terms and conditions of the BSD License
+; which accompanies this distribution.  The full text of the license may be found at
+; http://opensource.org/licenses/bsd-license.php.
+;
+; THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
+; WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
+;
+; Module Name:
+;
+;   IntHandler.asm
+;
+; Abstract:
+;
+;   Assembly interrupt handler function.
+;
+;------------------------------------------------------------------------------
+
+
+.686p
+.model  flat,c
+
+public AsmInterruptHandle
+
+.code
+AsmInterruptHandle:
+    cli
+    mov   al, 1
+    iretd
+END
diff --git a/SourceLevelDebugPkg/Library/PeCoffExtraActionLibDebug/Ia32/IntHandlerFuncs.c b/SourceLevelDebugPkg/Library/PeCoffExtraActionLibDebug/Ia32/IntHandlerFuncs.c
new file mode 100644
index 0000000000..b2d4c37a9d
--- /dev/null
+++ b/SourceLevelDebugPkg/Library/PeCoffExtraActionLibDebug/Ia32/IntHandlerFuncs.c
@@ -0,0 +1,96 @@
+/** @file
+  Ia32 arch functions to access IDT vector.
+
+  Copyright (c) 2013, Intel Corporation. All rights reserved.<BR>
+  This program and the accompanying materials
+  are licensed and made available under the terms and conditions of the BSD License
+  which accompanies this distribution.  The full text of the license may be found at
+  http://opensource.org/licenses/bsd-license.php.
+
+  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
+  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
+
+**/
+
+#include <PeCoffExtraActionLib.h>
+
+/**
+  Read IDT entry to check if IDT entries are setup by Debug Agent.
+
+  @param[in]  IdtDescriptor      Pointer to IDT Descriptor.
+
+  @retval  TRUE     IDT entries were setup by Debug Agent.
+  @retval  FALSE    IDT entries were not setuo by Debug Agent.
+
+**/
+BOOLEAN 
+CheckDebugAgentHandler (
+  IN  IA32_DESCRIPTOR            *IdtDescriptor
+  )
+{
+  IA32_IDT_GATE_DESCRIPTOR   *IdtEntry;
+  UINTN                      InterruptHandler;
+
+  IdtEntry = (IA32_IDT_GATE_DESCRIPTOR *) IdtDescriptor->Base;
+  if (IdtEntry == NULL) {
+    return FALSE;
+  }
+
+  InterruptHandler = IdtEntry[0].Bits.OffsetLow + (IdtEntry[0].Bits.OffsetHigh << 16);
+  if (InterruptHandler >= 4 &&  *(UINT32 *)(InterruptHandler - 4) == AGENT_HANDLER_SIGNATURE) {
+    return TRUE;
+  } else {
+    return FALSE;
+  }
+}
+
+/**
+  Save IDT entry for INT1 and update it. 
+
+  @param[in]  IdtDescriptor      Pointer to IDT Descriptor.
+  @param[out] SavedIdtEntry      Original IDT entry returned.
+
+**/
+VOID
+SaveAndUpdateIdtEntry1 (
+  IN  IA32_DESCRIPTOR            *IdtDescriptor,
+  OUT IA32_IDT_GATE_DESCRIPTOR   *SavedIdtEntry
+  )
+{
+  IA32_IDT_GATE_DESCRIPTOR   *IdtEntry;
+  UINT16                     CodeSegment;
+  UINTN                      InterruptHandler;
+  
+  IdtEntry = (IA32_IDT_GATE_DESCRIPTOR *) IdtDescriptor->Base;
+  CopyMem (SavedIdtEntry, &IdtEntry[1], sizeof (IA32_IDT_GATE_DESCRIPTOR));
+
+    //
+  // Use current CS as the segment selector of interrupt gate in IDT
+  //
+  CodeSegment = AsmReadCs ();
+
+  InterruptHandler = (UINTN) &AsmInterruptHandle;
+  IdtEntry[1].Bits.OffsetLow  = (UINT16)(UINTN)InterruptHandler;
+  IdtEntry[1].Bits.OffsetHigh = (UINT16)((UINTN)InterruptHandler >> 16);
+  IdtEntry[1].Bits.Selector   = CodeSegment;
+  IdtEntry[1].Bits.GateType   = IA32_IDT_GATE_TYPE_INTERRUPT_32;
+}
+
+/**
+  Restore IDT entry for INT1. 
+
+  @param[in]  IdtDescriptor      Pointer to IDT Descriptor.
+  @param[in]  RestoredIdtEntry   IDT entry to be restored.
+
+**/
+VOID
+RestoreIdtEntry1 (
+  IN  IA32_DESCRIPTOR            *IdtDescriptor,
+  IN  IA32_IDT_GATE_DESCRIPTOR   *RestoredIdtEntry
+  )
+{
+  IA32_IDT_GATE_DESCRIPTOR   *IdtEntry;
+  
+  IdtEntry = (IA32_IDT_GATE_DESCRIPTOR *) IdtDescriptor->Base;
+  CopyMem (&IdtEntry[1], RestoredIdtEntry, sizeof (IA32_IDT_GATE_DESCRIPTOR));
+}