From 0a16169be8fcade376d291fd70371e262b2592ee Mon Sep 17 00:00:00 2001
From: pgao2 <pgao2@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Mon, 13 Sep 2010 02:42:14 +0000
Subject: [PATCH] Fix the issue that might prevent single step in
 PeCoffExtraActionLib (right after triggering breakpoint to notify image
 load/unload) work.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10870 6f19259b-4bc3-4df7-8a09-765794883524
---
 .../Library/DebugAgent/DebugAgentCommon/DebugAgent.c         | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/SourceLevelDebugPkg/Library/DebugAgent/DebugAgentCommon/DebugAgent.c b/SourceLevelDebugPkg/Library/DebugAgent/DebugAgentCommon/DebugAgent.c
index 39b24c39c6..7834f885cc 100644
--- a/SourceLevelDebugPkg/Library/DebugAgent/DebugAgentCommon/DebugAgent.c
+++ b/SourceLevelDebugPkg/Library/DebugAgent/DebugAgentCommon/DebugAgent.c
@@ -384,6 +384,11 @@ GetBreakCause (
       //
       if ((CpuContext->Dr6 & BIT14) != 0) {
         Cause = DEBUG_DATA_BREAK_CAUSE_STEPPING;
+        //
+        // If it's single step, no need to check DR0, to ensure single step work in PeCoffExtraActionLib
+        // (right after triggering a breakpoint to report image load/unload).
+        //
+        return Cause;
 
       } else {
         Cause = DEBUG_DATA_BREAK_CAUSE_HW_BREAKPOINT;
-- 
2.39.2