1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
/*
* Copyright (c) 2021, Konstantin Aladyshev <aladyshev22@gmail.com>
*
* SPDX-License-Identifier: MIT
*/
#include <Library/UefiBootServicesTableLib.h>
#include <Library/UefiRuntimeServicesTableLib.h>
#include <Library/UefiLib.h>
VOID Usage()
{
Print(L"Delete variable\n");
Print(L" SetVariableExample <variable name>\n");
Print(L"\n");
Print(L"Set variable\n");
Print(L" SetVariableExample <variable name> <attributes> <value>\n");
Print(L"\n");
Print(L"<attributes> can be <n|b|r>\n");
Print(L"n - NON_VOLATILE\n");
Print(L"b - BOOTSERVICE_ACCESS\n");
Print(L"r - RUNTIME_ACCESS\n");
}
INTN EFIAPI ShellAppMain(IN UINTN Argc, IN CHAR16 **Argv)
{
EFI_STATUS Status;
if (Argc==2) {
CHAR16* VariableName = Argv[1];
Status = gRT->SetVariable (
VariableName,
&gEfiCallerIdGuid,
0,
0,
NULL
);
if (EFI_ERROR (Status)) {
Print(L"%r\n", Status);
} else {
Print(L"Variable %s was successfully deleted\n", VariableName);
}
return Status;
} else if (Argc==4) {
CHAR16* VariableName = Argv[1];
CHAR16* AttributesStr = Argv[2];
CHAR16* Value = Argv[3];
UINT32 Attributes = 0;
for (UINTN i=0; i<StrLen(AttributesStr); i++) {
switch(AttributesStr[i]) {
case L'n':
Attributes |= EFI_VARIABLE_NON_VOLATILE;
break;
case L'b':
Attributes |= EFI_VARIABLE_BOOTSERVICE_ACCESS;
break;
case L'r':
Attributes |= EFI_VARIABLE_RUNTIME_ACCESS;
break;
default:
Print(L"Error! Unknown attribute!");
return EFI_INVALID_PARAMETER;
}
}
Status = gRT->SetVariable (
VariableName,
&gEfiCallerIdGuid,
Attributes,
StrSize(Value),
Value
);
if (EFI_ERROR (Status)) {
Print(L"%r\n", Status);
} else {
Print(L"Variable %s was successfully changed\n", VariableName);
}
return Status;
} else {
Usage();
}
return EFI_SUCCESS;
}
|
