mirror of
https://github.com/CloverHackyColor/CloverBootloader.git
synced 2024-12-11 14:28:08 +01:00
278 lines
10 KiB
C
278 lines
10 KiB
C
|
/** @file
|
||
|
|
Main file for attrib shell level 2 function.
|
||
|
|
|
||
|
|
(C) Copyright 2014-2015 Hewlett-Packard Development Company, L.P.<BR>
|
||
|
|
Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>
|
||
|
|
This program and the accompanying materials
|
||
|
|
are licensed and made available under the terms and conditions of the BSD License
|
||
|
|
which accompanies this distribution. The full text of the license may be found at
|
||
|
|
http://opensource.org/licenses/bsd-license.php
|
||
|
|
|
||
|
|
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
|
||
|
|
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
|
||
|
|
|
||
|
|
**/
|
||
|
|
|
||
|
|
#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);
|
||
|
|
}
|