ShellPkg/DP: Add more check for input parameters

[mirror_edk2.git] / ShellPkg / DynamicCommand / TftpDynamicCommand / TftpDynamicCommand.c

diff --git a/ShellPkg/DynamicCommand/TftpDynamicCommand/TftpDynamicCommand.c b/ShellPkg/DynamicCommand/TftpDynamicCommand/TftpDynamicCommand.c
index 928ef08468e150b56dacca8da4df70657f0b3ced..9e6489dd6fc5abedb6c549b47a666abdddc97274 100644 (file)
--- a/ShellPkg/DynamicCommand/TftpDynamicCommand/TftpDynamicCommand.c
+++ b/ShellPkg/DynamicCommand/TftpDynamicCommand/TftpDynamicCommand.c
@@ -39,6 +39,7 @@ TftpCommandHandler (
   )\r
 {\r
   gEfiShellParametersProtocol = ShellParameters;\r
+  gEfiShellProtocol           = Shell;\r
   return RunTftp (gImageHandle, SystemTable);\r
 }\r
 \r