From aaf779a7ec7890520d57fd5bb4cc40c826dd42e5 Mon Sep 17 00:00:00 2001
From: hhuan13 <hhuan13@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Wed, 8 Dec 2010 04:42:40 +0000
Subject: [PATCH] Fixed a bug: when underlayer receives much more data, return
 EFI_BUFFER_TOO_SMALL.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@11132 6f19259b-4bc3-4df7-8a09-765794883524
---
 Nt32Pkg/SnpNt32Dxe/SnpNt32.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/Nt32Pkg/SnpNt32Dxe/SnpNt32.c b/Nt32Pkg/SnpNt32Dxe/SnpNt32.c
index 46bc8768f0..86414696c4 100644
--- a/Nt32Pkg/SnpNt32Dxe/SnpNt32.c
+++ b/Nt32Pkg/SnpNt32Dxe/SnpNt32.c
@@ -995,6 +995,9 @@ SnpNt32Receive (
   SNPNT32_INSTANCE_DATA *Instance;
   SNPNT32_GLOBAL_DATA   *GlobalData;
   INT32                 ReturnValue;
+  UINTN                 BufSize;
+  
+  BufSize     = *BuffSize;
 
   Instance    = SNP_NT32_INSTANCE_DATA_FROM_SNP_THIS (This);
 
@@ -1042,7 +1045,7 @@ SnpNt32Receive (
     *Protocol = NTOHS (*((UINT16 *) (((UINT8 *) Buffer) + 12)));
   }
 
-  return EFI_SUCCESS;
+  return (*BuffSize <= BufSize) ? EFI_SUCCESS : EFI_BUFFER_TOO_SMALL; 
 }
 
 /**
-- 
2.39.2