Fixed one bug of WriteUnaligned24()

[mirror_edk2.git] / MdePkg / Library / BaseLib / Unaligned.c

diff --git a/MdePkg/Library/BaseLib/Unaligned.c b/MdePkg/Library/BaseLib/Unaligned.c
index 9d0521e595b350b902fe75a705c06b996b102bbc..2ce8fc69924debfe81343f1d3666d2d6244b0846 100644 (file)
--- a/MdePkg/Library/BaseLib/Unaligned.c
+++ b/MdePkg/Library/BaseLib/Unaligned.c
@@ -33,6 +33,8 @@ ReadUnaligned16 (
   IN      CONST UINT16              *Buffer\r
   )\r
 {\r
+  ASSERT (Buffer != NULL);\r
+\r
   return *Buffer;\r
 }\r
 \r
@@ -58,6 +60,8 @@ WriteUnaligned16 (
   IN      UINT16                    Value\r
   )\r
 {\r
+  ASSERT (Buffer != NULL);\r
+\r
   return *Buffer = Value;\r
 }\r
 \r
@@ -80,6 +84,8 @@ ReadUnaligned24 (
   IN      CONST UINT32              *Buffer\r
   )\r
 {\r
+  ASSERT (Buffer != NULL);\r
+\r
   return *Buffer & 0xffffff;\r
 }\r
 \r
@@ -105,7 +111,10 @@ WriteUnaligned24 (
   IN      UINT32                    Value\r
   )\r
 {\r
-  return *Buffer = BitFieldWrite32 (*Buffer, 0, 23, Value);\r
+  ASSERT (Buffer != NULL);\r
+\r
+  BitFieldWrite32 (*Buffer, 0, 23, Value);\r
+  return Value;\r
 }\r
 \r
 /**\r
@@ -127,6 +136,8 @@ ReadUnaligned32 (
   IN      CONST UINT32              *Buffer\r
   )\r
 {\r
+  ASSERT (Buffer != NULL);\r
+\r
   return *Buffer;\r
 }\r
 \r
@@ -152,6 +163,8 @@ WriteUnaligned32 (
   IN      UINT32                    Value\r
   )\r
 {\r
+  ASSERT (Buffer != NULL);\r
+\r
   return *Buffer = Value;\r
 }\r
 \r
@@ -174,6 +187,8 @@ ReadUnaligned64 (
   IN      CONST UINT64              *Buffer\r
   )\r
 {\r
+  ASSERT (Buffer != NULL);\r
+\r
   return *Buffer;\r
 }\r
 \r
@@ -199,5 +214,7 @@ WriteUnaligned64 (
   IN      UINT64                    Value\r
   )\r
 {\r
+  ASSERT (Buffer != NULL);\r
+\r
   return *Buffer = Value;\r
 }\r