2 Main file for vol shell level 2 function.
4 Copyright (c) 2015, Hewlett-Packard Development Company, L.P.<BR>
5 Copyright (c) 2011 - 2014, 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
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.
16 #include "UefiShellLevel2CommandsLib.h"
17 #include <Guid/FileSystemInfo.h>
18 #include <Guid/FileSystemVolumeLabelInfo.h>
21 Print the info or change the volume info.
23 @param[in] Path String with starting path.
24 @param[in] Delete TRUE to delete the volume label. FALSE otherwise.
25 @param[in] Name New name to set to the volume label.
27 @retval SHELL_SUCCESS The operation was sucessful.
32 IN CONST CHAR16
*Path
,
33 IN CONST BOOLEAN Delete
,
34 IN CONST CHAR16
*Name OPTIONAL
38 SHELL_STATUS ShellStatus
;
39 EFI_FILE_SYSTEM_INFO
*SysInfo
;
41 SHELL_FILE_HANDLE ShellFileHandle
;
42 EFI_FILE_PROTOCOL
*EfiFpHandle
;
46 ShellStatus
= SHELL_SUCCESS
;
50 StrStr(Name
, L
"%") != NULL
||
51 StrStr(Name
, L
"^") != NULL
||
52 StrStr(Name
, L
"*") != NULL
||
53 StrStr(Name
, L
"+") != NULL
||
54 StrStr(Name
, L
"=") != NULL
||
55 StrStr(Name
, L
"[") != NULL
||
56 StrStr(Name
, L
"]") != NULL
||
57 StrStr(Name
, L
"|") != NULL
||
58 StrStr(Name
, L
":") != NULL
||
59 StrStr(Name
, L
";") != NULL
||
60 StrStr(Name
, L
"\"") != NULL
||
61 StrStr(Name
, L
"<") != NULL
||
62 StrStr(Name
, L
">") != NULL
||
63 StrStr(Name
, L
"?") != NULL
||
64 StrStr(Name
, L
"/") != NULL
||
65 StrStr(Name
, L
" ") != NULL
)
67 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PARAM_INV
), gShellLevel2HiiHandle
, L
"vol", Name
);
68 return (SHELL_INVALID_PARAMETER
);
71 Status
= gEfiShellProtocol
->OpenFileByName(
74 Name
!= NULL
?EFI_FILE_MODE_READ
|EFI_FILE_MODE_WRITE
:EFI_FILE_MODE_READ
);
76 if (EFI_ERROR(Status
) || ShellFileHandle
== NULL
) {
77 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_FILE_OPEN_FAIL
), gShellLevel2HiiHandle
, L
"vol", Path
);
78 return (SHELL_ACCESS_DENIED
);
82 // Get the Volume Info from ShellFileHandle
86 EfiFpHandle
= ConvertShellHandleToEfiFileProtocol(ShellFileHandle
);
87 Status
= EfiFpHandle
->GetInfo(
89 &gEfiFileSystemInfoGuid
,
93 if (Status
== EFI_BUFFER_TOO_SMALL
) {
94 SysInfo
= AllocateZeroPool(SysInfoSize
);
95 Status
= EfiFpHandle
->GetInfo(
97 &gEfiFileSystemInfoGuid
,
102 ASSERT(SysInfo
!= NULL
);
105 *((CHAR16
*) SysInfo
->VolumeLabel
) = CHAR_NULL
;
106 SysInfo
->Size
= SIZE_OF_EFI_FILE_SYSTEM_INFO
+ StrSize(SysInfo
->VolumeLabel
);
107 Status
= EfiFpHandle
->SetInfo(
109 &gEfiFileSystemInfoGuid
,
110 (UINTN
)SysInfo
->Size
,
112 } else if (Name
!= NULL
) {
113 Size1
= StrSize(Name
);
114 Size2
= StrSize(SysInfo
->VolumeLabel
);
116 SysInfo
= ReallocatePool((UINTN
)SysInfo
->Size
, (UINTN
)SysInfo
->Size
+ Size1
- Size2
, SysInfo
);
117 if (SysInfo
== NULL
) {
118 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_OUT_MEM
), gShellLevel2HiiHandle
, L
"vol");
119 ShellStatus
= SHELL_OUT_OF_RESOURCES
;
122 if (SysInfo
!= NULL
) {
123 StrnCpy ((CHAR16
*) SysInfo
->VolumeLabel
, Name
, (Size1
> Size2
?Size1
/sizeof(CHAR16
):Size2
/sizeof(CHAR16
))-1);
124 SysInfo
->Size
= SIZE_OF_EFI_FILE_SYSTEM_INFO
+ Size1
;
125 Status
= EfiFpHandle
->SetInfo(
127 &gEfiFileSystemInfoGuid
,
128 (UINTN
)SysInfo
->Size
,
135 if (Delete
|| Name
!= NULL
) {
136 if (EFI_ERROR(Status
)) {
137 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_FILE_AD
), gShellLevel2HiiHandle
, L
"vol", Path
);
138 ShellStatus
= SHELL_ACCESS_DENIED
;
145 Status
= EfiFpHandle
->GetInfo(
147 &gEfiFileSystemInfoGuid
,
151 if (Status
== EFI_BUFFER_TOO_SMALL
) {
152 SysInfo
= AllocateZeroPool(SysInfoSize
);
153 Status
= EfiFpHandle
->GetInfo(
155 &gEfiFileSystemInfoGuid
,
160 gEfiShellProtocol
->CloseFile(ShellFileHandle
);
162 ASSERT(SysInfo
!= NULL
);
164 if (SysInfo
!= NULL
) {
166 // print VolumeInfo table
170 gST
->ConOut
->Mode
->CursorRow
,
172 STRING_TOKEN (STR_VOL_VOLINFO
),
173 gShellLevel2HiiHandle
,
174 SysInfo
->VolumeLabel
,
175 SysInfo
->ReadOnly
?L
"r":L
"rw",
180 SHELL_FREE_NON_NULL(SysInfo
);
183 return (ShellStatus
);
186 STATIC CONST SHELL_PARAM_ITEM ParamList
[] = {
193 Function for 'Vol' command.
195 @param[in] ImageHandle Handle to the Image (NULL if Internal).
196 @param[in] SystemTable Pointer to the System Table (NULL if Internal).
201 IN EFI_HANDLE ImageHandle
,
202 IN EFI_SYSTEM_TABLE
*SystemTable
207 CHAR16
*ProblemParam
;
208 SHELL_STATUS ShellStatus
;
209 CONST CHAR16
*PathName
;
210 CONST CHAR16
*CurDir
;
215 CONST CHAR16
*NewName
;
219 ShellStatus
= SHELL_SUCCESS
;
225 // initialize the shell lib (we must be in non-auto-init...)
227 Status
= ShellInitialize();
228 ASSERT_EFI_ERROR(Status
);
231 // Fix local copies of the protocol pointers
233 Status
= CommandInit();
234 ASSERT_EFI_ERROR(Status
);
237 // parse the command line
239 Status
= ShellCommandLineParse (ParamList
, &Package
, &ProblemParam
, TRUE
);
240 if (EFI_ERROR(Status
)) {
241 if (Status
== EFI_VOLUME_CORRUPTED
&& ProblemParam
!= NULL
) {
242 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PROBLEM
), gShellLevel2HiiHandle
, L
"vol", ProblemParam
);
243 FreePool(ProblemParam
);
244 ShellStatus
= SHELL_INVALID_PARAMETER
;
252 if (ShellCommandLineGetFlag(Package
, L
"-?")) {
256 if (ShellCommandLineGetCount(Package
) > 2) {
257 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_TOO_MANY
), gShellLevel2HiiHandle
, L
"vol");
258 ShellStatus
= SHELL_INVALID_PARAMETER
;
260 PathName
= ShellCommandLineGetRawValue(Package
, 1);
261 if (PathName
== NULL
) {
262 CurDir
= gEfiShellProtocol
->GetCurDir(NULL
);
263 if (CurDir
== NULL
) {
264 ShellStatus
= SHELL_NOT_FOUND
;
265 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_NO_CWD
), gShellLevel2HiiHandle
, L
"vol");
270 if (PathName
!= NULL
) {
271 TempSpot
= StrStr(PathName
, L
":");
272 if (TempSpot
!= NULL
) {
273 *TempSpot
= CHAR_NULL
;
275 TempSpot
= StrStr(PathName
, L
"\\");
276 if (TempSpot
!= NULL
) {
277 *TempSpot
= CHAR_NULL
;
279 StrnCatGrow(&FullPath
, &Length
, PathName
, 0);
280 StrnCatGrow(&FullPath
, &Length
, L
":\\", 0);
281 DeleteMode
= ShellCommandLineGetFlag(Package
, L
"-d");
282 NewName
= ShellCommandLineGetValue(Package
, L
"-n");
283 if (DeleteMode
&& ShellCommandLineGetFlag(Package
, L
"-n")) {
284 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PARAM_CONFLICT
), gShellLevel2HiiHandle
, L
"vol", L
"-d", L
"-n");
285 ShellStatus
= SHELL_INVALID_PARAMETER
;
286 } else if (ShellCommandLineGetFlag(Package
, L
"-n") && NewName
== NULL
) {
287 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_NO_VALUE
), gShellLevel2HiiHandle
, L
"vol", L
"-n");
288 ShellStatus
= SHELL_INVALID_PARAMETER
;
289 } else if (NewName
!= NULL
&& StrLen(NewName
) > 11) {
290 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PROBLEM_VAL
), gShellLevel2HiiHandle
, L
"vol", NewName
, L
"-n");
291 ShellStatus
= SHELL_INVALID_PARAMETER
;
292 } else if (ShellStatus
== SHELL_SUCCESS
) {
293 ShellStatus
= HandleVol(
303 SHELL_FREE_NON_NULL(FullPath
);
306 // free the command line package
308 ShellCommandLineFreeVarList (Package
);
310 return (ShellStatus
);