aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--Library/UefiShellUfmCommandLib/UefiShellUfmCommandLib.c4
1 files changed, 4 insertions, 0 deletions
diff --git a/Library/UefiShellUfmCommandLib/UefiShellUfmCommandLib.c b/Library/UefiShellUfmCommandLib/UefiShellUfmCommandLib.c
index d08298a..b5d4f38 100644
--- a/Library/UefiShellUfmCommandLib/UefiShellUfmCommandLib.c
+++ b/Library/UefiShellUfmCommandLib/UefiShellUfmCommandLib.c
@@ -112,6 +112,10 @@ SHELL_STATUS EFIAPI ShellCommandRunUFM(
SHELL_STATUS status = SHELL_LOAD_ERROR;
BOOLEAN res;
+ // Check for UEFI Shell 2.0 protocol
+ if(!gEfiShellProtocol)
+ return SHELL_UNSUPPORTED;
+
res = prepare_context();
if(res) {
do_ec();