1 /** @file
2 Main file for attrib shell level 2 function.
3
4 (C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>
5 Copyright (c) 2009 - 2015, Intel Corporation. All rights reserved.<BR>
6 This program and the accompanying materials
7 are licensed and made available under the terms and conditions of the BSD License
8 which accompanies this distribution. The full text of the license may be found at
9 http://opensource.org/licenses/bsd-license.php
10
11 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
12 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
13
14 **/
15
16 #include "UefiShellLevel2CommandsLib.h"
17
18 /**
19 Print out each environment variable registered with the Shell 2.0 GUID.
20
21 If you spawn a pre 2.0 shell from the Shell 2.0 the environment variable may not carry through.
22
23 @retval STATUS_SUCCESS the printout was sucessful
24 @return any return code from GetNextVariableName except EFI_NOT_FOUND
25 **/
26 SHELL_STATUS
27 EFIAPI
PrintAllShellEnvVars(VOID)28 PrintAllShellEnvVars(
29 VOID
30 )
31 {
32 CONST CHAR16 *Value;
33 CONST CHAR16 *ConstEnvNameList;
34
35 ConstEnvNameList = gEfiShellProtocol->GetEnv(NULL);
36 if (ConstEnvNameList == NULL) {
37 return (SHELL_SUCCESS);
38 }
39 while (*ConstEnvNameList != CHAR_NULL){
40 Value = gEfiShellProtocol->GetEnv(ConstEnvNameList);
41 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SET_DISP), gShellLevel2HiiHandle, ConstEnvNameList, Value);
42 ConstEnvNameList += StrLen(ConstEnvNameList)+1;
43 }
44
45 return (SHELL_SUCCESS);
46 }
47
48 STATIC CONST SHELL_PARAM_ITEM SetParamList[] = {
49 {L"-d", TypeValue},
50 {L"-v", TypeFlag},
51 {NULL, TypeMax}
52 };
53
54 /**
55 Function for 'set' command.
56
57 @param[in] ImageHandle Handle to the Image (NULL if Internal).
58 @param[in] SystemTable Pointer to the System Table (NULL if Internal).
59 **/
60 SHELL_STATUS
61 EFIAPI
ShellCommandRunSet(IN EFI_HANDLE ImageHandle,IN EFI_SYSTEM_TABLE * SystemTable)62 ShellCommandRunSet (
63 IN EFI_HANDLE ImageHandle,
64 IN EFI_SYSTEM_TABLE *SystemTable
65 )
66 {
67 EFI_STATUS Status;
68 LIST_ENTRY *Package;
69 CONST CHAR16 *KeyName;
70 CONST CHAR16 *Value;
71 CHAR16 *ProblemParam;
72 SHELL_STATUS ShellStatus;
73
74 ProblemParam = NULL;
75 ShellStatus = SHELL_SUCCESS;
76
77 //
78 // initialize the shell lib (we must be in non-auto-init...)
79 //
80 Status = ShellInitialize();
81 ASSERT_EFI_ERROR(Status);
82
83 //
84 // Make sure globals are good...
85 //
86 Status = CommandInit();
87 ASSERT_EFI_ERROR(Status);
88
89 //
90 // parse the command line
91 //
92 Status = ShellCommandLineParse (SetParamList, &Package, &ProblemParam, TRUE);
93 if (EFI_ERROR(Status)) {
94 if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {
95 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, L"set", ProblemParam);
96 FreePool(ProblemParam);
97 return (SHELL_INVALID_PARAMETER);
98 } else {
99 ASSERT(FALSE);
100 }
101 } else {
102 //
103 // check for "-?"
104 //
105 if (ShellCommandLineGetFlag(Package, L"-?")) {
106 ASSERT(FALSE);
107 } else if (ShellCommandLineGetRawValue(Package, 3) != NULL) {
108 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellLevel2HiiHandle, L"set");
109 ShellStatus = SHELL_INVALID_PARAMETER;
110 } else if (ShellCommandLineGetRawValue(Package, 1) != NULL && ShellCommandLineGetFlag(Package, L"-d")) {
111 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellLevel2HiiHandle, L"set");
112 ShellStatus = SHELL_INVALID_PARAMETER;
113 } else if (ShellCommandLineGetFlag(Package, L"-d")) {
114 //
115 // delete a environment variable
116 //
117 KeyName = ShellCommandLineGetValue(Package, L"-d");
118 if (KeyName == NULL) {
119 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_VALUE), gShellLevel2HiiHandle, L"set", L"-d");
120 ShellStatus = SHELL_INVALID_PARAMETER;
121 } else {
122 Status = ShellSetEnvironmentVariable(KeyName, L"", ShellCommandLineGetFlag(Package, L"-v"));
123 if (EFI_ERROR(Status)) {
124 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SET_ND), gShellLevel2HiiHandle, L"set", KeyName);
125 ShellStatus = SHELL_DEVICE_ERROR;
126 }
127 }
128 } else if (ShellCommandLineGetRawValue(Package, 1) == NULL) {
129 //
130 // print out all current environment variables
131 //
132 return(PrintAllShellEnvVars());
133 } else {
134 //
135 // we are either printing one or assigning one
136 //
137 KeyName = ShellCommandLineGetRawValue(Package, 1);
138 Value = ShellCommandLineGetRawValue(Package, 2);
139 if (KeyName != NULL && Value != NULL) {
140 //
141 // assigning one
142 //
143 Status = ShellSetEnvironmentVariable(KeyName, Value, ShellCommandLineGetFlag(Package, L"-v"));
144 if (EFI_ERROR(Status)) {
145 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SET_ERROR_SET), gShellLevel2HiiHandle, L"set", KeyName);
146 ShellStatus = (SHELL_STATUS) (Status & (~MAX_BIT));
147 }
148
149 } else {
150 if (KeyName != NULL) {
151 //
152 // print out value for this one only.
153 //
154 Value = ShellGetEnvironmentVariable(KeyName);
155 if (Value == NULL) {
156 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SET_NF), gShellLevel2HiiHandle, L"set", KeyName);
157 ShellStatus = SHELL_SUCCESS;
158 } else {
159 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SET_DISP), gShellLevel2HiiHandle, KeyName, Value);
160 ShellStatus = SHELL_SUCCESS;
161 }
162 } else {
163 ASSERT(FALSE);
164 }
165 }
166 }
167 }
168
169 //
170 // free the command line package
171 //
172 ShellCommandLineFreeVarList (Package);
173
174 return (ShellStatus);
175 }
176