Refine code to make it more safely.

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Eric Dong <eric.dong@intel.com>
Reviewed-by: Jeff Fan <jeff.fan@intel.com>




git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@15591 6f19259b-4bc3-4df7-8a09-765794883524

3 files changed, 8 insertions, 4 deletions

diff --git a/SourceLevelDebugPkg/DebugAgentDxe/DebugAgentDxe.c b/SourceLevelDebugPkg/DebugAgentDxe/DebugAgentDxe.c
index 99bb2fd..80d997b 100644
--- a/SourceLevelDebugPkg/DebugAgentDxe/DebugAgentDxe.c
+++ b/SourceLevelDebugPkg/DebugAgentDxe/DebugAgentDxe.c
@@ -1,7 +1,7 @@
 /** @file

   Initialize Debug Agent in DXE by invoking Debug Agent Library.

 

-Copyright (c) 2013, Intel Corporation. All rights reserved.<BR>

+Copyright (c) 2013 - 2014, 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        

@@ -58,6 +58,7 @@ DebugAgentDxeInitialize(
 {

   EFI_STATUS      Status;

 

+  Status = EFI_UNSUPPORTED;

   InitializeDebugAgent (DEBUG_AGENT_INIT_DXE_LOAD, &Status, NULL);

   if (EFI_ERROR (Status)) {

     return Status;

@@ -95,6 +96,7 @@ DebugAgentDxeUnload (
 {

   EFI_STATUS          Status;

 

+  Status = EFI_UNSUPPORTED;

   InitializeDebugAgent (DEBUG_AGENT_INIT_DXE_UNLOAD, &Status, NULL);

 

   return Status;

diff --git a/SourceLevelDebugPkg/DebugAgentPei/DebugAgentPei.c b/SourceLevelDebugPkg/DebugAgentPei/DebugAgentPei.c
index df0ac87..2dcc7a7 100644
--- a/SourceLevelDebugPkg/DebugAgentPei/DebugAgentPei.c
+++ b/SourceLevelDebugPkg/DebugAgentPei/DebugAgentPei.c
@@ -1,7 +1,7 @@
 /** @file

   Initialize Debug Agent in PEI by invoking Debug Agent Library.

 

-Copyright (c) 2013, Intel Corporation. All rights reserved.<BR>

+Copyright (c) 2013 - 2014, 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        

@@ -39,6 +39,7 @@ DebugAgentPeiInitialize (
 {

   EFI_STATUS                 Status;

 

+  Status = EFI_UNSUPPORTED;

   InitializeDebugAgent (DEBUG_AGENT_INIT_PEI, &Status, NULL);

 

   return Status;

diff --git a/SourceLevelDebugPkg/Library/DebugAgent/DebugAgentCommon/DebugAgent.c b/SourceLevelDebugPkg/Library/DebugAgent/DebugAgentCommon/DebugAgent.c
index 95306e5..48bc38c 100644
--- a/SourceLevelDebugPkg/Library/DebugAgent/DebugAgentCommon/DebugAgent.c
+++ b/SourceLevelDebugPkg/Library/DebugAgent/DebugAgentCommon/DebugAgent.c
@@ -587,7 +587,7 @@ ReadRemainingBreakPacket (
   //

   // Has received start symbol, try to read the rest part

   //

-  if (DebugPortReadBuffer (Handle, &DebugHeader->Command, sizeof (DEBUG_PACKET_HEADER) - 1, READ_PACKET_TIMEOUT) == 0) {

+  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) {

     //

     // Timeout occur, exit

     //

@@ -1014,7 +1014,7 @@ ReceivePacket (
     //

     Received = DebugPortReadBuffer (

                  Handle,

-                 &DebugHeader->Command,

+                 (UINT8 *)DebugHeader + OFFSET_OF (DEBUG_PACKET_HEADER, Command),

                  OFFSET_OF (DEBUG_PACKET_HEADER, Length) + sizeof (DebugHeader->Length) - sizeof (DebugHeader->StartSymbol),

                  Timeout

                  );

@@ -2104,6 +2104,7 @@ InterruptProcess (
   UINT32                           IssuedViewPoint;

   DEBUG_AGENT_EXCEPTION_BUFFER     *ExceptionBuffer;

 

+  InputCharacter  = 0;

   ProcessorIndex  = 0;

   IssuedViewPoint = 0;

   BreakReceived   = FALSE;