ShellPkg: allow for mm command to properly write PCIe in non-interactive mode.

[mirror_edk2.git] / ShellPkg / Library / UefiShellDebug1CommandsLib / Mm.c

diff --git a/ShellPkg/Library/UefiShellDebug1CommandsLib/Mm.c b/ShellPkg/Library/UefiShellDebug1CommandsLib/Mm.c
index 5d9915cec8f2f7722e8993b5c6e5a5f1dfdbb86d..6a21ea4268047b54ef858acc5d81a80ab10bfff6 100644 (file)
--- a/ShellPkg/Library/UefiShellDebug1CommandsLib/Mm.c
+++ b/ShellPkg/Library/UefiShellDebug1CommandsLib/Mm.c
@@ -464,25 +464,25 @@ ShellCommandRunMm (
       GetPciEAddressFromInputAddress (Address, &PciEAddress);\r
     }\r
 \r
-//    //\r
-//    // Set value\r
-//    //\r
-//    if (ValueStr != NULL) {\r
-//      if (AccessType == EFIMemoryMappedIo) {\r
-//        IoDev->Mem.Write (IoDev, Width, Address, 1, &Value);\r
-//      } else if (AccessType == EfiIo) {\r
-//        IoDev->Io.Write (IoDev, Width, Address, 1, &Value);\r
-//      } else if (AccessType == EfiPciConfig) {\r
-//        IoDev->Pci.Write (IoDev, Width, Address, 1, &Value);\r
-//      } else if (AccessType == EfiPciEConfig) {\r
-//        IoDev->Pci.Write (IoDev, Width, PciEAddress, 1, &Buffer);\r
-//      } else {\r
-//        WriteMem (Width, Address, 1, &Value);\r
-//      }\r
-//\r
-//      ASSERT(ShellStatus == SHELL_SUCCESS);\r
-//      goto Done;\r
-//    }\r
+    //\r
+    // Set value\r
+    //\r
+    if (ShellCommandLineGetRawValue(Package, 2) != NULL) {\r
+      if (AccessType == EFIMemoryMappedIo) {\r
+        IoDev->Mem.Write (IoDev, Width, Address, 1, &Value);\r
+      } else if (AccessType == EfiIo) {\r
+        IoDev->Io.Write (IoDev, Width, Address, 1, &Value);\r
+      } else if (AccessType == EfiPciConfig) {\r
+        IoDev->Pci.Write (IoDev, Width, Address, 1, &Value);\r
+      } else if (AccessType == EfiPciEConfig) {\r
+        IoDev->Pci.Write (IoDev, Width, PciEAddress, 1, &Value);\r
+      } else {\r
+        WriteMem (Width, Address, 1, &Value);\r
+      }\r
+\r
+      ASSERT(ShellStatus == SHELL_SUCCESS);\r
+      goto Done;\r
+    }\r
 \r
 \r
     //\r