K8: add new variable PasswordSize to remove immeidate constant.

[mirror_edk2.git] / MdeModulePkg / Universal / DriverSampleDxe / DriverSample.c

diff --git a/MdeModulePkg/Universal/DriverSampleDxe/DriverSample.c b/MdeModulePkg/Universal/DriverSampleDxe/DriverSample.c
index 64805ff83df10dc4b7880ceba95288d9487cdd0a..e470a82cc7e5b29d86c802bb1c8dfbfa02961362 100644 (file)
--- a/MdeModulePkg/Universal/DriverSampleDxe/DriverSample.c
+++ b/MdeModulePkg/Universal/DriverSampleDxe/DriverSample.c
@@ -33,7 +33,7 @@ CHAR16     VariableName[] = L"MyIfrNVData";
 VOID\r
 EncodePassword (\r
   IN  CHAR16                      *Password,\r
-  IN  UINT8                       MaxSize\r
+  IN  UINTN                       MaxSize\r
   )\r
 {\r
   UINTN   Index;\r
@@ -167,6 +167,7 @@ SetPassword (
   EFI_STATUS                      Status;\r
   UINTN                           BufferSize;\r
   CHAR16                          *Password;\r
+  UINTN                           PasswordSize;\r
   DRIVER_SAMPLE_CONFIGURATION     *Configuration;\r
 \r
   //\r
@@ -188,7 +189,9 @@ SetPassword (
   // Get user input password\r
   //\r
   Password = &PrivateData->Configuration.WhatIsThePassword2[0];\r
-  ZeroMem (Password, 20 * sizeof (CHAR16));\r
+  PasswordSize = sizeof (PrivateData->Configuration.WhatIsThePassword2);\r
+  \r
+  ZeroMem (Password, PasswordSize);\r
   Status = HiiLibGetString (PrivateData->HiiHandle[0], StringId, Password, &BufferSize);\r
   if (EFI_ERROR (Status)) {\r
     return Status;\r
@@ -205,7 +208,7 @@ SetPassword (
     //\r
     // Update password's clear text in the screen\r
     //\r
-    CopyMem (Configuration->PasswordClearText, Password, 20 * sizeof (CHAR16));\r
+    CopyMem (Configuration->PasswordClearText, Password, PasswordSize);\r
 \r
     //\r
     // Update uncommitted data of Browser\r
@@ -224,7 +227,7 @@ SetPassword (
   //\r
   // Set password\r
   //\r
-  EncodePassword (Password, 20 * sizeof (CHAR16));\r
+  EncodePassword (Password, PasswordSize);\r
   Status = gRT->SetVariable(\r
                   VariableName,\r
                   &mFormSetGuid,\r