EmulatorPkg: Add -D DISABLE_NEW_DEPRECATED_INTERFACES

[mirror_edk2.git] / EmulatorPkg / Library / ThunkProtocolList / ThunkProtocolList.c

diff --git a/EmulatorPkg/Library/ThunkProtocolList/ThunkProtocolList.c b/EmulatorPkg/Library/ThunkProtocolList/ThunkProtocolList.c
index b7aacc851c365b4ac38474ad48d9b599ef99fb39..3a7b6d1cebb4526184c8a18292442add5db84d7a 100644 (file)
--- a/EmulatorPkg/Library/ThunkProtocolList/ThunkProtocolList.c
+++ b/EmulatorPkg/Library/ThunkProtocolList/ThunkProtocolList.c
@@ -2,7 +2,7 @@
   Emulator Thunk to abstract OS services from pure EFI code\r
 \r
   Copyright (c) 2008 - 2011, Apple Inc. All rights reserved.<BR>\r
-  Copyright (c) 2011 - 2018, Intel Corporation. All rights reserved.<BR>\r
+  Copyright (c) 2011 - 2019, Intel Corporation. All rights reserved.<BR>\r
 \r
   SPDX-License-Identifier: BSD-2-Clause-Patent\r
 \r
@@ -37,6 +37,7 @@ AddThunkProtocol (
   IN  BOOLEAN                 EmuBusDriver\r
   )\r
 {\r
+  UINTN                       Size;\r
   CHAR16                      *StartString;\r
   CHAR16                      *SubString;\r
   UINTN                       Instance;\r
@@ -47,8 +48,12 @@ AddThunkProtocol (
   }\r
 \r
   Instance = 0;\r
-  StartString = AllocatePool (StrSize (ConfigString));\r
-  StrCpy (StartString, ConfigString);\r
+  Size = StrSize (ConfigString);\r
+  StartString = AllocatePool (Size);\r
+  if (StartString == NULL) {\r
+    return EFI_OUT_OF_RESOURCES;\r
+  }\r
+  StrCpyS (StartString, Size / sizeof (CHAR16), ConfigString);\r
   while (*StartString != '\0') {\r
 \r
     //\r