]> git.proxmox.com Git - mirror_edk2.git/blame - ShellPkg/Library/UefiShellLevel2CommandsLib/Rm.c
udk2010.up2.shell initial release.
[mirror_edk2.git] / ShellPkg / Library / UefiShellLevel2CommandsLib / Rm.c
CommitLineData
a405b86d 1/** @file\r
2 Main file for attrib shell level 2 function.\r
3\r
4 Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
5 This program and the accompanying materials\r
6 are licensed and made available under the terms and conditions of the BSD License\r
7 which accompanies this distribution. The full text of the license may be found at\r
8 http://opensource.org/licenses/bsd-license.php\r
9\r
10 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
11 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
12\r
13**/\r
14\r
15#include "UefiShellLevel2CommandsLib.h"\r
16\r
17STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
18 {L"-q", TypeFlag},\r
19 {NULL, TypeMax}\r
20 };\r
21\r
22BOOLEAN\r
23EFIAPI\r
24IsDirectoryEmpty (\r
25 IN EFI_HANDLE FileHandle\r
26 )\r
27{\r
28 EFI_STATUS Status;\r
29 EFI_FILE_INFO *FileInfo;\r
30 BOOLEAN NoFile;\r
31 BOOLEAN RetVal;\r
32\r
33 RetVal = TRUE;\r
34 NoFile = FALSE;\r
35\r
36 for (Status = FileHandleFindFirstFile(FileHandle, &FileInfo)\r
37 ; !NoFile\r
38 ; Status = FileHandleFindNextFile(FileHandle, FileInfo, &NoFile)\r
39 ){\r
40 if (StrStr(FileInfo->FileName, L".") != FileInfo->FileName\r
41 &&StrStr(FileInfo->FileName, L"..") != FileInfo->FileName) {\r
42 RetVal = FALSE;\r
43 }\r
44 }\r
45 return (RetVal);\r
46}\r
47\r
48SHELL_STATUS\r
49EFIAPI\r
50CascadeDelete(\r
51 IN EFI_SHELL_FILE_INFO *Node,\r
52 IN CONST BOOLEAN Quiet\r
53 )\r
54{\r
55 SHELL_STATUS ShellStatus;\r
56 EFI_SHELL_FILE_INFO *List;\r
57 EFI_SHELL_FILE_INFO *Node2;\r
58 EFI_STATUS Status;\r
59 SHELL_PROMPT_RESPONSE *Resp;\r
60\r
61 Resp = NULL;\r
62 ShellStatus = SHELL_SUCCESS;\r
63 List = NULL;\r
64 Status = EFI_SUCCESS;\r
65\r
66 if ((Node->Info->Attribute & EFI_FILE_READ_ONLY) == EFI_FILE_READ_ONLY) {\r
67 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_RM_LOG_DETELE_RO), gShellLevel2HiiHandle, Node->FullName);\r
68 return (SHELL_ACCESS_DENIED);\r
69 }\r
70\r
71 if ((Node->Info->Attribute & EFI_FILE_DIRECTORY) == EFI_FILE_DIRECTORY) {\r
72 if (!IsDirectoryEmpty(Node->Handle)) {\r
73 if (!Quiet) {\r
74 Status = ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN(STR_RM_LOG_DELETE_CONF), gShellLevel2HiiHandle, Node->FullName);\r
75 Status = ShellPromptForResponse(ShellPromptResponseTypeYesNo, NULL, (VOID**)&Resp);\r
76 ASSERT_EFI_ERROR(Status);\r
77 ASSERT(Resp != NULL);\r
78 if (EFI_ERROR(Status) || *Resp != ShellPromptResponseYes) {\r
79 SHELL_FREE_NON_NULL(Resp);\r
80 return (SHELL_ABORTED);\r
81 }\r
82 SHELL_FREE_NON_NULL(Resp);\r
83 }\r
84 //\r
85 // empty out the directory\r
86 //\r
87 Status = gEfiShellProtocol->FindFilesInDir(Node->Handle, &List);\r
88 if (EFI_ERROR(Status)) {\r
89 if (List!=NULL) {\r
90 gEfiShellProtocol->FreeFileList(&List);\r
91 }\r
92 return (SHELL_DEVICE_ERROR);\r
93 }\r
94 for (Node2 = (EFI_SHELL_FILE_INFO *)GetFirstNode(&List->Link)\r
95 ; !IsNull(&List->Link, &Node2->Link)\r
96 ; Node2 = (EFI_SHELL_FILE_INFO *)GetNextNode(&List->Link, &Node2->Link)\r
97 ){\r
98 //\r
99 // skip the directory traversing stuff...\r
100 //\r
101 if (StrCmp(Node2->FileName, L".") == 0 || StrCmp(Node2->FileName, L"..") == 0) {\r
102 continue;\r
103 }\r
104 Node2->Status = gEfiShellProtocol->OpenFileByName (Node2->FullName, &Node2->Handle, EFI_FILE_MODE_READ|EFI_FILE_MODE_WRITE);\r
105 ShellStatus = CascadeDelete(Node2, Quiet);\r
106 if (ShellStatus != SHELL_SUCCESS) {\r
107 if (List!=NULL) {\r
108 gEfiShellProtocol->FreeFileList(&List);\r
109 }\r
110 return (ShellStatus);\r
111 }\r
112 }\r
113 if (List!=NULL) {\r
114 gEfiShellProtocol->FreeFileList(&List);\r
115 }\r
116 }\r
117 }\r
118\r
119 if (!(StrCmp(Node->FileName, L".") == 0 || StrCmp(Node->FileName, L"..") == 0)) {\r
120 //\r
121 // now delete the current node...\r
122 //\r
123 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_RM_LOG_DELETE), gShellLevel2HiiHandle, Node->FullName);\r
124 Status = gEfiShellProtocol->DeleteFile(Node->Handle);\r
125 Node->Handle = NULL;\r
126 }\r
127\r
128 //\r
129 // We cant allow for the warning here!\r
130 //\r
131 if (Status != EFI_SUCCESS){\r
132 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_RM_LOG_DELETE_ERR), gShellLevel2HiiHandle, Status);\r
133 return (SHELL_ACCESS_DENIED);\r
134 } else {\r
135 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_RM_LOG_DELETE_COMP), gShellLevel2HiiHandle);\r
136 return (SHELL_SUCCESS);\r
137 }\r
138}\r
139\r
140BOOLEAN\r
141EFIAPI\r
142IsValidDeleteTarget(\r
143 IN CONST EFI_SHELL_FILE_INFO *List,\r
144 IN CONST EFI_SHELL_FILE_INFO *Node,\r
145 IN CONST LIST_ENTRY *Package\r
146 )\r
147{\r
148 CONST CHAR16 *TempLocation;\r
149 CHAR16 *Temp2;\r
150 UINTN Size;\r
151\r
152 TempLocation = StrStr(Node->FullName, L":");\r
153 if (StrLen(TempLocation) == 2) {\r
154 //\r
155 // Deleting the root directory is invalid.\r
156 //\r
157 return (FALSE);\r
158 }\r
159 TempLocation = ShellGetCurrentDir(NULL);\r
160 Size = 0;\r
161 Temp2 = NULL;\r
162 StrnCatGrow(&Temp2, &Size, TempLocation, 0);\r
163 if (StrStr(Temp2, Node->FullName) != NULL) {\r
164 FreePool(Temp2);\r
165 return (FALSE);\r
166 }\r
167 FreePool(Temp2);\r
168\r
169 return (TRUE);\r
170}\r
171\r
172/**\r
173 Function for 'rm' command.\r
174\r
175 @param[in] ImageHandle Handle to the Image (NULL if Internal).\r
176 @param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
177**/\r
178SHELL_STATUS\r
179EFIAPI\r
180ShellCommandRunRm (\r
181 IN EFI_HANDLE ImageHandle,\r
182 IN EFI_SYSTEM_TABLE *SystemTable\r
183 )\r
184{\r
185 EFI_STATUS Status;\r
186 LIST_ENTRY *Package;\r
187 CHAR16 *ProblemParam;\r
188 CONST CHAR16 *Param;\r
189 SHELL_STATUS ShellStatus;\r
190 UINTN ParamCount;\r
191 EFI_SHELL_FILE_INFO *FileList;\r
192 EFI_SHELL_FILE_INFO *Node;\r
193\r
194 ProblemParam = NULL;\r
195 ShellStatus = SHELL_SUCCESS;\r
196 ParamCount = 0;\r
197 FileList = NULL;\r
198\r
199 //\r
200 // initialize the shell lib (we must be in non-auto-init...)\r
201 //\r
202 Status = ShellInitialize();\r
203 ASSERT_EFI_ERROR(Status);\r
204\r
205 //\r
206 // parse the command line\r
207 //\r
208 Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
209 if (EFI_ERROR(Status)) {\r
210 if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
211 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, ProblemParam);\r
212 FreePool(ProblemParam);\r
213 ShellStatus = SHELL_INVALID_PARAMETER;\r
214 } else {\r
215 ASSERT(FALSE);\r
216 }\r
217 } else {\r
218 //\r
219 // check for "-?"\r
220 //\r
221 if (ShellCommandLineGetFlag(Package, L"-?")) {\r
222 ASSERT(FALSE);\r
223 }\r
224 if (ShellCommandLineGetRawValue(Package, 1) == NULL) {\r
225 //\r
226 // we insufficient parameters\r
227 //\r
228 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellLevel2HiiHandle);\r
229 ShellStatus = SHELL_INVALID_PARAMETER;\r
230 } else {\r
231 //\r
232 // get a list with each file specified by parameters\r
233 // if parameter is a directory then add all the files below it to the list\r
234 //\r
235 for ( ParamCount = 1, Param = ShellCommandLineGetRawValue(Package, ParamCount)\r
236 ; Param != NULL\r
237 ; ParamCount++, Param = ShellCommandLineGetRawValue(Package, ParamCount)\r
238 ){\r
239 Status = ShellOpenFileMetaArg((CHAR16*)Param, EFI_FILE_MODE_WRITE|EFI_FILE_MODE_READ, &FileList);\r
240 if (EFI_ERROR(Status) || FileList == NULL || IsListEmpty(&FileList->Link)) {\r
241 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_NF), gShellLevel2HiiHandle, (CHAR16*)Param);\r
242 ShellStatus = SHELL_NOT_FOUND;\r
243 break;\r
244 }\r
245 }\r
246\r
247 if (ShellStatus == SHELL_SUCCESS){\r
248 //\r
249 // loop through the list and make sure we are not aborting...\r
250 //\r
251 for ( Node = (EFI_SHELL_FILE_INFO*)GetFirstNode(&FileList->Link)\r
252 ; !IsNull(&FileList->Link, &Node->Link) && !ShellGetExecutionBreakFlag()\r
253 ; Node = (EFI_SHELL_FILE_INFO*)GetNextNode(&FileList->Link, &Node->Link)\r
254 ){\r
255 //\r
256 // skip the directory traversing stuff...\r
257 //\r
258 if (StrCmp(Node->FileName, L".") == 0 || StrCmp(Node->FileName, L"..") == 0) {\r
259 continue;\r
260 }\r
261\r
262 //\r
263 // do the deleting of nodes\r
264 //\r
265 if (EFI_ERROR(Node->Status)){\r
266 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_RM_LOG_DELETE_ERR2), gShellLevel2HiiHandle, Node->Status);\r
267 ShellStatus = SHELL_ACCESS_DENIED;\r
268 break;\r
269 }\r
270 if (!IsValidDeleteTarget(FileList, Node, Package)) {\r
271 ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_RM_LOG_DELETE_ERR3), gShellLevel2HiiHandle, Node->FullName);\r
272 ShellStatus = SHELL_INVALID_PARAMETER;\r
273 break;\r
274 }\r
275\r
276 ShellStatus = CascadeDelete(Node, ShellCommandLineGetFlag(Package, L"-q"));\r
277 }\r
278 }\r
279 //\r
280 // Free the fileList\r
281 //\r
282 if (FileList != NULL) {\r
283 Status = ShellCloseFileMetaArg(&FileList);\r
284 }\r
285 FileList = NULL;\r
286 }\r
287\r
288 //\r
289 // free the command line package\r
290 //\r
291 ShellCommandLineFreeVarList (Package);\r
292 }\r
293\r
294 return (ShellStatus);\r
295}\r
296\r