From a93786ae6c4f8a92e216899c05366ba7aacf72e7 Mon Sep 17 00:00:00 2001
From: Gary Lin <glin@suse.com>
Date: Mon, 9 May 2016 12:32:55 +0800
Subject: NetworkPkg: Make HttpBootGetBootFile return EFI_BUFFER_TOO_SMALL

Per the description of HttpBootGetBootFile, the function should return
EFI_BUFFER_TOO_SMALL if the given buffer is smaller than the remote image.

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Gary Lin <glin@suse.com>
Reviewed-by: Fu Siyuan <siyuan.fu@intel.com>
---
 NetworkPkg/HttpBootDxe/HttpBootClient.c | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'NetworkPkg/HttpBootDxe/HttpBootClient.c')

diff --git a/NetworkPkg/HttpBootDxe/HttpBootClient.c b/NetworkPkg/HttpBootDxe/HttpBootClient.c
index 9d445e3c8e..46cf9ca6fb 100644
--- a/NetworkPkg/HttpBootDxe/HttpBootClient.c
+++ b/NetworkPkg/HttpBootDxe/HttpBootClient.c
@@ -1074,6 +1074,8 @@ HttpBootGetBootFile (
 
   if (*BufferSize < ContentLength) {
     Status = EFI_BUFFER_TOO_SMALL;
+  } else {
+    Status = EFI_SUCCESS;
   }
   *BufferSize = ContentLength;
 
@@ -1089,7 +1091,7 @@ HttpBootGetBootFile (
     HttpFreeMsgParser (Parser);
   }
 
-  return EFI_SUCCESS;
+  return Status;
   
 ERROR_6:
   if (Parser != NULL) {
-- 
cgit v1.2.3