/** @file | |
Main file for attrib shell level 2 function. | |
(C) Copyright 2014-2015 Hewlett-Packard Development Company, L.P.<BR> | |
Copyright (c) 2009 - 2018, Intel Corporation. All rights reserved.<BR> | |
SPDX-License-Identifier: BSD-2-Clause-Patent | |
**/ | |
#include "UefiShellLevel2CommandsLib.h" | |
STATIC CONST CHAR16 AllFiles[] = L"*"; | |
STATIC CONST SHELL_PARAM_ITEM AttribParamList[] = { | |
{ L"-a", TypeFlag }, | |
{ L"+a", TypeFlag }, | |
{ L"-s", TypeFlag }, | |
{ L"+s", TypeFlag }, | |
{ L"-h", TypeFlag }, | |
{ L"+h", TypeFlag }, | |
{ L"-r", TypeFlag }, | |
{ L"+r", TypeFlag }, | |
{ NULL, TypeMax } | |
}; | |
/** | |
Function for 'attrib' command. | |
@param[in] ImageHandle Handle to the Image (NULL if Internal). | |
@param[in] SystemTable Pointer to the System Table (NULL if Internal). | |
**/ | |
SHELL_STATUS | |
EFIAPI | |
ShellCommandRunAttrib ( | |
IN EFI_HANDLE ImageHandle, | |
IN EFI_SYSTEM_TABLE *SystemTable | |
) | |
{ | |
UINT64 FileAttributesToAdd; | |
UINT64 FileAttributesToRemove; | |
EFI_STATUS Status; | |
LIST_ENTRY *Package; | |
CHAR16 *ProblemParam; | |
SHELL_STATUS ShellStatus; | |
UINTN ParamNumberCount; | |
CONST CHAR16 *FileName; | |
EFI_SHELL_FILE_INFO *ListOfFiles; | |
EFI_SHELL_FILE_INFO *FileNode; | |
EFI_FILE_INFO *FileInfo; | |
ListOfFiles = NULL; | |
ShellStatus = SHELL_SUCCESS; | |
ProblemParam = NULL; | |
// | |
// initialize the shell lib (we must be in non-auto-init...) | |
// | |
Status = ShellInitialize (); | |
ASSERT_EFI_ERROR (Status); | |
// | |
// parse the command line | |
// | |
Status = ShellCommandLineParse (AttribParamList, &Package, &ProblemParam, TRUE); | |
if (EFI_ERROR (Status)) { | |
if ((Status == EFI_VOLUME_CORRUPTED) && (ProblemParam != NULL)) { | |
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, L"attrib", ProblemParam); | |
FreePool (ProblemParam); | |
ShellStatus = SHELL_INVALID_PARAMETER; | |
} else { | |
ASSERT (FALSE); | |
} | |
} else { | |
// | |
// check for "-?" | |
// | |
if (ShellCommandLineGetFlag (Package, L"-?")) { | |
ASSERT (FALSE); | |
} else { | |
FileAttributesToAdd = 0; | |
FileAttributesToRemove = 0; | |
// | |
// apply or remove each flag | |
// | |
if (ShellCommandLineGetFlag (Package, L"+a")) { | |
FileAttributesToAdd |= EFI_FILE_ARCHIVE; | |
} | |
if (ShellCommandLineGetFlag (Package, L"-a")) { | |
FileAttributesToRemove |= EFI_FILE_ARCHIVE; | |
} | |
if (ShellCommandLineGetFlag (Package, L"+s")) { | |
FileAttributesToAdd |= EFI_FILE_SYSTEM; | |
} | |
if (ShellCommandLineGetFlag (Package, L"-s")) { | |
FileAttributesToRemove |= EFI_FILE_SYSTEM; | |
} | |
if (ShellCommandLineGetFlag (Package, L"+h")) { | |
FileAttributesToAdd |= EFI_FILE_HIDDEN; | |
} | |
if (ShellCommandLineGetFlag (Package, L"-h")) { | |
FileAttributesToRemove |= EFI_FILE_HIDDEN; | |
} | |
if (ShellCommandLineGetFlag (Package, L"+r")) { | |
FileAttributesToAdd |= EFI_FILE_READ_ONLY; | |
} | |
if (ShellCommandLineGetFlag (Package, L"-r")) { | |
FileAttributesToRemove |= EFI_FILE_READ_ONLY; | |
} | |
if ((FileAttributesToRemove == 0) && (FileAttributesToAdd == 0)) { | |
// | |
// Do display as we have no attributes to change | |
// | |
for ( ParamNumberCount = 1 | |
; | |
; ParamNumberCount++ | |
) | |
{ | |
FileName = ShellCommandLineGetRawValue (Package, ParamNumberCount); | |
// if we dont have anything left, move on... | |
if ((FileName == NULL) && (ParamNumberCount == 1)) { | |
FileName = (CHAR16 *)AllFiles; | |
} else if (FileName == NULL) { | |
break; | |
} | |
ASSERT (ListOfFiles == NULL); | |
Status = ShellOpenFileMetaArg ((CHAR16 *)FileName, EFI_FILE_MODE_READ, &ListOfFiles); | |
if (EFI_ERROR (Status)) { | |
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_OPEN_FAIL), gShellLevel2HiiHandle, L"attrib", ShellCommandLineGetRawValue (Package, ParamNumberCount)); | |
ShellStatus = SHELL_NOT_FOUND; | |
} else { | |
for (FileNode = (EFI_SHELL_FILE_INFO *)GetFirstNode (&ListOfFiles->Link) | |
; !IsNull (&ListOfFiles->Link, &FileNode->Link) | |
; FileNode = (EFI_SHELL_FILE_INFO *)GetNextNode (&ListOfFiles->Link, &FileNode->Link) | |
) | |
{ | |
ShellPrintHiiEx ( | |
-1, | |
-1, | |
NULL, | |
STRING_TOKEN (STR_ATTRIB_OUTPUT_LINE), | |
gShellLevel2HiiHandle, | |
FileNode->Info->Attribute&EFI_FILE_DIRECTORY ? L'D' : L' ', | |
FileNode->Info->Attribute&EFI_FILE_ARCHIVE ? L'A' : L' ', | |
FileNode->Info->Attribute&EFI_FILE_SYSTEM ? L'S' : L' ', | |
FileNode->Info->Attribute&EFI_FILE_HIDDEN ? L'H' : L' ', | |
FileNode->Info->Attribute&EFI_FILE_READ_ONLY ? L'R' : L' ', | |
FileNode->FileName | |
); | |
if (ShellGetExecutionBreakFlag ()) { | |
ShellStatus = SHELL_ABORTED; | |
break; | |
} | |
} | |
Status = ShellCloseFileMetaArg (&ListOfFiles); | |
ListOfFiles = NULL; | |
if (EFI_ERROR (Status)) { | |
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_CLOSE_FAIL), gShellLevel2HiiHandle, L"attrib", ShellCommandLineGetRawValue (Package, ParamNumberCount)); | |
ShellStatus = SHELL_NOT_FOUND; | |
} | |
} // for loop for handling wildcard filenames | |
} // for loop for printing out the info | |
} else if ((FileAttributesToRemove & FileAttributesToAdd) != 0) { | |
// | |
// fail as we have conflcting params. | |
// | |
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CON), gShellLevel2HiiHandle, L"attrib"); | |
ShellStatus = SHELL_INVALID_PARAMETER; | |
} else { | |
// | |
// enumerate through all the files/directories and apply the attributes | |
// | |
for ( ParamNumberCount = 1 | |
; | |
; ParamNumberCount++ | |
) | |
{ | |
FileName = ShellCommandLineGetRawValue (Package, ParamNumberCount); | |
// if we dont have anything left, move on... | |
if (FileName == NULL) { | |
// | |
// make sure we are not failing on the first one we do... if yes that's an error... | |
// | |
if (ParamNumberCount == 1) { | |
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellLevel2HiiHandle, L"attrib"); | |
ShellStatus = SHELL_INVALID_PARAMETER; | |
} | |
break; | |
} | |
// | |
// OpenFileByName / GetFileInfo / Change attributes / SetFileInfo / CloseFile / free memory | |
// for each file or directory on the line. | |
// | |
// | |
// Open the file(s) | |
// | |
ASSERT (ListOfFiles == NULL); | |
Status = ShellOpenFileMetaArg ((CHAR16 *)FileName, EFI_FILE_MODE_READ, &ListOfFiles); | |
if (EFI_ERROR (Status)) { | |
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_OPEN_FAIL), gShellLevel2HiiHandle, L"attrib", ShellCommandLineGetRawValue (Package, ParamNumberCount)); | |
ShellStatus = SHELL_NOT_FOUND; | |
} else { | |
for (FileNode = (EFI_SHELL_FILE_INFO *)GetFirstNode (&ListOfFiles->Link) | |
; !IsNull (&ListOfFiles->Link, &FileNode->Link) | |
; FileNode = (EFI_SHELL_FILE_INFO *)GetNextNode (&ListOfFiles->Link, &FileNode->Link) | |
) | |
{ | |
// | |
// skip the directory traversing stuff... | |
// | |
if ((StrCmp (FileNode->FileName, L".") == 0) || (StrCmp (FileNode->FileName, L"..") == 0)) { | |
continue; | |
} | |
FileInfo = gEfiShellProtocol->GetFileInfo (FileNode->Handle); | |
// | |
// if we are removing Read-Only we need to do that alone | |
// | |
if ((FileAttributesToRemove & EFI_FILE_READ_ONLY) == EFI_FILE_READ_ONLY) { | |
FileInfo->Attribute &= ~EFI_FILE_READ_ONLY; | |
// | |
// SetFileInfo | |
// | |
Status = ShellSetFileInfo (FileNode->Handle, FileInfo); | |
if (EFI_ERROR (Status)) { | |
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_AD), gShellLevel2HiiHandle, L"attrib", ShellCommandLineGetRawValue (Package, ParamNumberCount)); | |
ShellStatus = SHELL_ACCESS_DENIED; | |
} | |
} | |
// | |
// change the attribute | |
// | |
FileInfo->Attribute &= ~FileAttributesToRemove; | |
FileInfo->Attribute |= FileAttributesToAdd; | |
// | |
// SetFileInfo | |
// | |
Status = ShellSetFileInfo (FileNode->Handle, FileInfo); | |
if (EFI_ERROR (Status)) { | |
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_AD), gShellLevel2HiiHandle, L"attrib", ShellCommandLineGetRawValue (Package, ParamNumberCount)); | |
ShellStatus = SHELL_ACCESS_DENIED; | |
} | |
SHELL_FREE_NON_NULL (FileInfo); | |
} | |
Status = ShellCloseFileMetaArg (&ListOfFiles); | |
ListOfFiles = NULL; | |
if (EFI_ERROR (Status)) { | |
ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_CLOSE_FAIL), gShellLevel2HiiHandle, L"attrib", ShellCommandLineGetRawValue (Package, ParamNumberCount)); | |
ShellStatus = SHELL_NOT_FOUND; | |
} | |
} // for loop for handling wildcard filenames | |
} | |
} | |
} | |
} | |
// | |
// free the command line package | |
// | |
ShellCommandLineFreeVarList (Package); | |
// | |
// return the status | |
// | |
return (ShellStatus); | |
} |