2 Main file for SetSize shell Debug1 function.
4 Copyright (c) 2010 - 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 "UefiShellDebug1CommandsLib.h"
18 Function for 'setsize' command.
20 @param[in] ImageHandle Handle to the Image (NULL if Internal).
21 @param[in] SystemTable Pointer to the System Table (NULL if Internal).
25 ShellCommandRunSetSize (
26 IN EFI_HANDLE ImageHandle
,
27 IN EFI_SYSTEM_TABLE
*SystemTable
33 SHELL_STATUS ShellStatus
;
37 SHELL_FILE_HANDLE FileHandle
;
39 ShellStatus
= SHELL_SUCCESS
;
43 // initialize the shell lib (we must be in non-auto-init...)
45 Status
= ShellInitialize();
46 ASSERT_EFI_ERROR(Status
);
48 Status
= CommandInit();
49 ASSERT_EFI_ERROR(Status
);
52 // parse the command line
54 Status
= ShellCommandLineParse (EmptyParamList
, &Package
, &ProblemParam
, TRUE
);
55 if (EFI_ERROR(Status
)) {
56 if (Status
== EFI_VOLUME_CORRUPTED
&& ProblemParam
!= NULL
) {
57 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_PROBLEM
), gShellDebug1HiiHandle
, ProblemParam
);
58 FreePool(ProblemParam
);
59 ShellStatus
= SHELL_INVALID_PARAMETER
;
64 if (ShellCommandLineGetCount(Package
) < 3) {
65 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_GEN_TOO_FEW
), gShellDebug1HiiHandle
);
66 ShellStatus
= SHELL_INVALID_PARAMETER
;
69 Temp1
= ShellCommandLineGetRawValue(Package
, 1);
70 if (!ShellIsHexOrDecimalNumber(Temp1
, FALSE
, FALSE
)) {
71 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_SIZE_NOT_SPEC
), gShellDebug1HiiHandle
);
72 ShellStatus
= SHELL_INVALID_PARAMETER
;
75 NewSize
= ShellStrToUintn(Temp1
);
78 for (LoopVar
= 2 ; LoopVar
< ShellCommandLineGetCount(Package
) && ShellStatus
== SHELL_SUCCESS
; LoopVar
++) {
79 Status
= ShellOpenFileByName(ShellCommandLineGetRawValue(Package
, LoopVar
), &FileHandle
, EFI_FILE_MODE_READ
|EFI_FILE_MODE_WRITE
, 0);
80 if (EFI_ERROR(Status
)) {
81 Status
= ShellOpenFileByName(ShellCommandLineGetRawValue(Package
, LoopVar
), &FileHandle
, EFI_FILE_MODE_READ
|EFI_FILE_MODE_WRITE
|EFI_FILE_MODE_CREATE
, 0);
83 if (EFI_ERROR(Status
) && LoopVar
== 2) {
84 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_FILE_NOT_SPEC
), gShellDebug1HiiHandle
);
85 ShellStatus
= SHELL_INVALID_PARAMETER
;
86 } else if (EFI_ERROR(Status
)) {
87 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_FILE_OPEN_FAIL
), gShellDebug1HiiHandle
, ShellCommandLineGetRawValue(Package
, LoopVar
), Status
);
88 ShellStatus
= SHELL_INVALID_PARAMETER
;
91 Status
= FileHandleSetSize(FileHandle
, NewSize
);
92 if (Status
== EFI_VOLUME_FULL
) {
93 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_VOLUME_FULL
), gShellDebug1HiiHandle
);
94 ShellStatus
= SHELL_VOLUME_FULL
;
95 } else if (EFI_ERROR(Status
)) {
96 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_SET_SIZE_FAIL
), gShellDebug1HiiHandle
, ShellCommandLineGetRawValue(Package
, LoopVar
), Status
);
97 ShellStatus
= SHELL_INVALID_PARAMETER
;
99 ShellPrintHiiEx(-1, -1, NULL
, STRING_TOKEN (STR_SET_SIZE_DONE
), gShellDebug1HiiHandle
, ShellCommandLineGetRawValue(Package
, LoopVar
));
101 ShellCloseFile(&FileHandle
);
105 ShellCommandLineFreeVarList (Package
);
108 return (ShellStatus
);