2 Main file for attrib shell level 2 function.
4 Copyright (c) 2009 - 2011, Intel Corporation. All rights reserved.<BR>
5 This program and the accompanying materials
6 are licensed and made available under the terms and conditions of the BSD License
7 which accompanies this distribution. The full text of the license may be found at
8 http://opensource.org/licenses/bsd-license.php
10 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
11 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
15 #include "UefiShellLevel2CommandsLib.h"
17 STATIC CONST SHELL_PARAM_ITEM ParamList
[] = {
23 Determine if a directory has no files in it.
25 @param[in] FileHandle The EFI_HANDLE to the directory.
27 @retval TRUE The directory has no files (or directories).
28 @retval FALSE The directory has at least 1 file or directory in it.
33 IN EFI_HANDLE FileHandle
37 EFI_FILE_INFO
*FileInfo
;
44 for (Status
= FileHandleFindFirstFile(FileHandle
, &FileInfo
)
46 ; Status
= FileHandleFindNextFile(FileHandle
, FileInfo
, &NoFile
)
48 if (StrStr(FileInfo
->FileName
, L
".") != FileInfo
->FileName
49 &&StrStr(FileInfo
->FileName
, L
"..") != FileInfo
->FileName
) {
57 Delete a node and all nodes under it (including sub directories).
59 @param[in] Node The node to start deleting with.
60 @param[in] Quiet TRUE to print no messages.
62 @retval SHELL_SUCCESS The operation was successful.
63 @retval SHELL_ACCESS_DENIED A file was read only.
64 @retval SHELL_ABORTED The abort message was received.
65 @retval SHELL_DEVICE_ERROR A device error occured reading this Node.
70 IN EFI_SHELL_FILE_INFO
*Node
,
71 IN CONST BOOLEAN Quiet
74 SHELL_STATUS ShellStatus
;
75 EFI_SHELL_FILE_INFO
*List
;
76 EFI_SHELL_FILE_INFO
*Node2
;
78 SHELL_PROMPT_RESPONSE
*Resp
;
81 ShellStatus
= SHELL_SUCCESS
;
85 if ((Node
->Info
->Attribute
& EFI_FILE_READ_ONLY
) == EFI_FILE_READ_ONLY
) {
86 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_RM_LOG_DETELE_RO
), gShellLevel2HiiHandle
, Node
->FullName
);
87 return (SHELL_ACCESS_DENIED
);
90 if ((Node
->Info
->Attribute
& EFI_FILE_DIRECTORY
) == EFI_FILE_DIRECTORY
) {
91 if (!IsDirectoryEmpty(Node
->Handle
)) {
93 Status
= ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN(STR_RM_LOG_DELETE_CONF
), gShellLevel2HiiHandle
, Node
->FullName
);
94 Status
= ShellPromptForResponse(ShellPromptResponseTypeYesNo
, NULL
, (VOID
**)&Resp
);
95 ASSERT_EFI_ERROR(Status
);
97 if (EFI_ERROR(Status
) || *Resp
!= ShellPromptResponseYes
) {
98 SHELL_FREE_NON_NULL(Resp
);
99 return (SHELL_ABORTED
);
101 SHELL_FREE_NON_NULL(Resp
);
104 // empty out the directory
106 Status
= gEfiShellProtocol
->FindFilesInDir(Node
->Handle
, &List
);
107 if (EFI_ERROR(Status
)) {
109 gEfiShellProtocol
->FreeFileList(&List
);
111 return (SHELL_DEVICE_ERROR
);
113 for (Node2
= (EFI_SHELL_FILE_INFO
*)GetFirstNode(&List
->Link
)
114 ; !IsNull(&List
->Link
, &Node2
->Link
)
115 ; Node2
= (EFI_SHELL_FILE_INFO
*)GetNextNode(&List
->Link
, &Node2
->Link
)
118 // skip the directory traversing stuff...
120 if (StrCmp(Node2
->FileName
, L
".") == 0 || StrCmp(Node2
->FileName
, L
"..") == 0) {
123 Node2
->Status
= gEfiShellProtocol
->OpenFileByName (Node2
->FullName
, &Node2
->Handle
, EFI_FILE_MODE_READ
|EFI_FILE_MODE_WRITE
);
124 ShellStatus
= CascadeDelete(Node2
, Quiet
);
125 if (ShellStatus
!= SHELL_SUCCESS
) {
127 gEfiShellProtocol
->FreeFileList(&List
);
129 return (ShellStatus
);
133 gEfiShellProtocol
->FreeFileList(&List
);
138 if (!(StrCmp(Node
->FileName
, L
".") == 0 || StrCmp(Node
->FileName
, L
"..") == 0)) {
140 // now delete the current node...
142 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_RM_LOG_DELETE
), gShellLevel2HiiHandle
, Node
->FullName
);
143 Status
= gEfiShellProtocol
->DeleteFile(Node
->Handle
);
148 // We cant allow for the warning here! (Dont use EFI_ERROR Macro).
150 if (Status
!= EFI_SUCCESS
){
151 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_RM_LOG_DELETE_ERR
), gShellLevel2HiiHandle
, Status
);
152 return (SHELL_ACCESS_DENIED
);
154 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_RM_LOG_DELETE_COMP
), gShellLevel2HiiHandle
);
155 return (SHELL_SUCCESS
);
160 Determins if a Node is a valid delete target. Will prevent deleting the root directory.
162 @param[in] List RESERVED. Not used.
163 @param[in] Node The node to analyze.
164 @param[in] Package RESERVED. Not used.
169 IN CONST EFI_SHELL_FILE_INFO
*List
,
170 IN CONST EFI_SHELL_FILE_INFO
*Node
,
171 IN CONST LIST_ENTRY
*Package
174 CONST CHAR16
*TempLocation
;
178 TempLocation
= StrStr(Node
->FullName
, L
":");
179 if (StrLen(TempLocation
) == 2) {
181 // Deleting the root directory is invalid.
185 TempLocation
= ShellGetCurrentDir(NULL
);
188 StrnCatGrow(&Temp2
, &Size
, TempLocation
, 0);
189 if (StrStr(Temp2
, Node
->FullName
) != NULL
) {
199 Function for 'rm' command.
201 @param[in] ImageHandle Handle to the Image (NULL if Internal).
202 @param[in] SystemTable Pointer to the System Table (NULL if Internal).
207 IN EFI_HANDLE ImageHandle
,
208 IN EFI_SYSTEM_TABLE
*SystemTable
213 CHAR16
*ProblemParam
;
215 SHELL_STATUS ShellStatus
;
217 EFI_SHELL_FILE_INFO
*FileList
;
218 EFI_SHELL_FILE_INFO
*Node
;
221 ShellStatus
= SHELL_SUCCESS
;
226 // initialize the shell lib (we must be in non-auto-init...)
228 Status
= ShellInitialize();
229 ASSERT_EFI_ERROR(Status
);
232 // parse the command line
234 Status
= ShellCommandLineParse (ParamList
, &Package
, &ProblemParam
, TRUE
);
235 if (EFI_ERROR(Status
)) {
236 if (Status
== EFI_VOLUME_CORRUPTED
&& ProblemParam
!= NULL
) {
237 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PROBLEM
), gShellLevel2HiiHandle
, ProblemParam
);
238 FreePool(ProblemParam
);
239 ShellStatus
= SHELL_INVALID_PARAMETER
;
247 if (ShellCommandLineGetFlag(Package
, L
"-?")) {
250 if (ShellCommandLineGetRawValue(Package
, 1) == NULL
) {
252 // we insufficient parameters
254 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_TOO_FEW
), gShellLevel2HiiHandle
);
255 ShellStatus
= SHELL_INVALID_PARAMETER
;
258 // get a list with each file specified by parameters
259 // if parameter is a directory then add all the files below it to the list
261 for ( ParamCount
= 1, Param
= ShellCommandLineGetRawValue(Package
, ParamCount
)
263 ; ParamCount
++, Param
= ShellCommandLineGetRawValue(Package
, ParamCount
)
265 Status
= ShellOpenFileMetaArg((CHAR16
*)Param
, EFI_FILE_MODE_WRITE
|EFI_FILE_MODE_READ
, &FileList
);
266 if (EFI_ERROR(Status
) || FileList
== NULL
|| IsListEmpty(&FileList
->Link
)) {
267 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_FILE_NF
), gShellLevel2HiiHandle
, (CHAR16
*)Param
);
268 ShellStatus
= SHELL_NOT_FOUND
;
273 if (ShellStatus
== SHELL_SUCCESS
){
275 // loop through the list and make sure we are not aborting...
277 for ( Node
= (EFI_SHELL_FILE_INFO
*)GetFirstNode(&FileList
->Link
)
278 ; !IsNull(&FileList
->Link
, &Node
->Link
) && !ShellGetExecutionBreakFlag()
279 ; Node
= (EFI_SHELL_FILE_INFO
*)GetNextNode(&FileList
->Link
, &Node
->Link
)
282 // skip the directory traversing stuff...
284 if (StrCmp(Node
->FileName
, L
".") == 0 || StrCmp(Node
->FileName
, L
"..") == 0) {
289 // do the deleting of nodes
291 if (EFI_ERROR(Node
->Status
)){
292 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_RM_LOG_DELETE_ERR2
), gShellLevel2HiiHandle
, Node
->Status
);
293 ShellStatus
= SHELL_ACCESS_DENIED
;
296 if (!IsValidDeleteTarget(FileList
, Node
, Package
)) {
297 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_RM_LOG_DELETE_ERR3
), gShellLevel2HiiHandle
, Node
->FullName
);
298 ShellStatus
= SHELL_INVALID_PARAMETER
;
302 ShellStatus
= CascadeDelete(Node
, ShellCommandLineGetFlag(Package
, L
"-q"));
308 if (FileList
!= NULL
) {
309 Status
= ShellCloseFileMetaArg(&FileList
);
315 // free the command line package
317 ShellCommandLineFreeVarList (Package
);
320 return (ShellStatus
);