mirror of
				https://github.com/acidanthera/audk.git
				synced 2025-11-04 05:25:45 +01:00 
			
		
		
		
	2. Update PeCoffExtraActionLib to detect if debug agent initialized or not by checking each IDT entry instead of whole IDT table. Signed-off-by: Jeff Fan <jeff.fan@intel.com> Reviewed-by: Ruiyu Ni <ruiyu.ni@intel.com> git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@14886 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			100 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			100 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/** @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.
 | 
						|
  @param[in]  InterruptType      Interrupt type.
 | 
						|
 | 
						|
  @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,
 | 
						|
  IN  UINTN                      InterruptType
 | 
						|
  )
 | 
						|
{
 | 
						|
  IA32_IDT_GATE_DESCRIPTOR   *IdtEntry;
 | 
						|
  UINTN                      InterruptHandler;
 | 
						|
 | 
						|
  IdtEntry = (IA32_IDT_GATE_DESCRIPTOR *) IdtDescriptor->Base;
 | 
						|
  if (IdtEntry == NULL) {
 | 
						|
    return FALSE;
 | 
						|
  }
 | 
						|
 | 
						|
  InterruptHandler = IdtEntry[InterruptType].Bits.OffsetLow +
 | 
						|
                    (IdtEntry[InterruptType].Bits.OffsetHigh << 16);
 | 
						|
  if (InterruptHandler >= sizeof (UINT32) &&  *(UINT32 *)(InterruptHandler - sizeof (UINT32)) == 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));
 | 
						|
}
 |