[mirror_edk2.git] / ShellPkg / Library / UefiShellLevel2CommandsLib / Vol.c

/** @file\r
  Main file for vol shell level 2 function.\r
\r
  (C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>\r
  Copyright (c) 2011 - 2018, Intel Corporation. All rights reserved.<BR>\r
  This program and the accompanying materials\r
  are licensed and made available under the terms and conditions of the BSD License\r
  which accompanies this distribution.  The full text of the license may be found at\r
  http://opensource.org/licenses/bsd-license.php\r
\r
  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
**/\r
\r
#include "UefiShellLevel2CommandsLib.h"\r
#include <Guid/FileSystemInfo.h>\r
#include <Guid/FileSystemVolumeLabelInfo.h>\r
\r
/**\r
  Print the info or change the volume info.\r
\r
  @param[in] Path           String with starting path.\r
  @param[in] Delete         TRUE to delete the volume label. FALSE otherwise.\r
  @param[in] Name           New name to set to the volume label.\r
\r
  @retval SHELL_SUCCESS     The operation was sucessful.\r
**/\r
SHELL_STATUS\r
HandleVol(\r
  IN CONST CHAR16  *Path,\r
  IN CONST BOOLEAN Delete,\r
  IN CONST CHAR16  *Name OPTIONAL\r
  )\r
{\r
  EFI_STATUS            Status;\r
  SHELL_STATUS          ShellStatus;\r
  EFI_FILE_SYSTEM_INFO  *SysInfo;\r
  UINTN                 SysInfoSize;\r
  SHELL_FILE_HANDLE     ShellFileHandle;\r
  EFI_FILE_PROTOCOL     *EfiFpHandle;\r
  UINTN                 Size1;\r
  UINTN                 Size2;\r
\r
  ShellStatus   = SHELL_SUCCESS;\r
\r
  if (\r
      Name != NULL && (\r
      StrStr(Name, L"%") != NULL ||\r
      StrStr(Name, L"^") != NULL ||\r
      StrStr(Name, L"*") != NULL ||\r
      StrStr(Name, L"+") != NULL ||\r
      StrStr(Name, L"=") != NULL ||\r
      StrStr(Name, L"[") != NULL ||\r
      StrStr(Name, L"]") != NULL ||\r
      StrStr(Name, L"|") != NULL ||\r
      StrStr(Name, L":") != NULL ||\r
      StrStr(Name, L";") != NULL ||\r
      StrStr(Name, L"\"") != NULL ||\r
      StrStr(Name, L"<") != NULL ||\r
      StrStr(Name, L">") != NULL ||\r
      StrStr(Name, L"?") != NULL ||\r
      StrStr(Name, L"/") != NULL ||\r
      StrStr(Name, L" ") != NULL )\r
      ){\r
    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellLevel2HiiHandle, L"vol", Name);\r
    return (SHELL_INVALID_PARAMETER);\r
  }\r
\r
  Status = gEfiShellProtocol->OpenFileByName(\r
    Path,\r
    &ShellFileHandle,\r
    Name != NULL?EFI_FILE_MODE_READ|EFI_FILE_MODE_WRITE:EFI_FILE_MODE_READ);\r
\r
  if (EFI_ERROR(Status) || ShellFileHandle == NULL) {\r
    ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_OPEN_FAIL), gShellLevel2HiiHandle, L"vol", Path);\r
    return (SHELL_ACCESS_DENIED);\r
  }\r
\r
  //\r
  // Get the Volume Info from ShellFileHandle\r
  //\r
  SysInfo     = NULL;\r
  SysInfoSize = 0;\r
  EfiFpHandle = ConvertShellHandleToEfiFileProtocol(ShellFileHandle);\r
  Status = EfiFpHandle->GetInfo(\r
    EfiFpHandle,\r
    &gEfiFileSystemInfoGuid,\r
    &SysInfoSize,\r
    SysInfo);\r
\r
  if (Status == EFI_BUFFER_TOO_SMALL) {\r
    SysInfo = AllocateZeroPool(SysInfoSize);\r
    Status = EfiFpHandle->GetInfo(\r
      EfiFpHandle,\r
      &gEfiFileSystemInfoGuid,\r
      &SysInfoSize,\r
      SysInfo);\r
  }\r
\r
  ASSERT(SysInfo != NULL);\r
\r
  if (Delete) {\r
    *((CHAR16 *) SysInfo->VolumeLabel) = CHAR_NULL;\r
    SysInfo->Size = SIZE_OF_EFI_FILE_SYSTEM_INFO + StrSize(SysInfo->VolumeLabel);\r
    Status = EfiFpHandle->SetInfo(\r
      EfiFpHandle,\r
      &gEfiFileSystemInfoGuid,\r
      (UINTN)SysInfo->Size,\r
      SysInfo);\r
  } else if (Name != NULL) {\r
    Size1 = StrSize(Name);\r
    Size2 = StrSize(SysInfo->VolumeLabel);\r
    if (Size1 > Size2) {\r
      SysInfo = ReallocatePool((UINTN)SysInfo->Size, (UINTN)SysInfo->Size + Size1 - Size2, SysInfo);\r
      if (SysInfo == NULL) {\r
        ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellLevel2HiiHandle, L"vol");\r
        ShellStatus = SHELL_OUT_OF_RESOURCES;\r
      }\r
    }\r
    if (SysInfo != NULL) {\r
      StrCpyS ( (CHAR16 *) SysInfo->VolumeLabel,\r
                  (Size1>Size2? Size1/sizeof(CHAR16) : Size2/sizeof(CHAR16)),\r
                  Name\r
                  );\r
      SysInfo->Size = SIZE_OF_EFI_FILE_SYSTEM_INFO + Size1;\r
      Status = EfiFpHandle->SetInfo(\r
        EfiFpHandle,\r
        &gEfiFileSystemInfoGuid,\r
        (UINTN)SysInfo->Size,\r
        SysInfo);\r
    }\r
  }\r
\r
  FreePool(SysInfo);\r
\r
  if (Delete || Name != NULL) {\r
    if (EFI_ERROR(Status)) {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_AD), gShellLevel2HiiHandle, L"vol", Path);\r
      ShellStatus = SHELL_ACCESS_DENIED;\r
    }\r
  }\r
\r
  SysInfoSize = 0;\r
  SysInfo = NULL;\r
\r
  Status = EfiFpHandle->GetInfo(\r
    EfiFpHandle,\r
    &gEfiFileSystemInfoGuid,\r
    &SysInfoSize,\r
    SysInfo);\r
\r
  if (Status == EFI_BUFFER_TOO_SMALL) {\r
    SysInfo = AllocateZeroPool(SysInfoSize);\r
    Status = EfiFpHandle->GetInfo(\r
      EfiFpHandle,\r
      &gEfiFileSystemInfoGuid,\r
      &SysInfoSize,\r
      SysInfo);\r
  }\r
\r
  gEfiShellProtocol->CloseFile(ShellFileHandle);\r
\r
  ASSERT(SysInfo != NULL);\r
\r
  if (SysInfo != NULL) {\r
    //\r
    // print VolumeInfo table\r
    //\r
    ShellPrintHiiEx (\r
      0,\r
      gST->ConOut->Mode->CursorRow,\r
      NULL,\r
      STRING_TOKEN (STR_VOL_VOLINFO),\r
      gShellLevel2HiiHandle,\r
      SysInfo->VolumeLabel,\r
      SysInfo->ReadOnly?L"r":L"rw",\r
      SysInfo->VolumeSize,\r
      SysInfo->FreeSpace,\r
      SysInfo->BlockSize\r
     );\r
    SHELL_FREE_NON_NULL(SysInfo);\r
  }\r
\r
  return (ShellStatus);\r
}\r
\r
STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
  {L"-d", TypeFlag},\r
  {L"-n", TypeValue},\r
  {NULL, TypeMax}\r
  };\r
\r
/**\r
  Function for 'Vol' command.\r
\r
  @param[in] ImageHandle  Handle to the Image (NULL if Internal).\r
  @param[in] SystemTable  Pointer to the System Table (NULL if Internal).\r
**/\r
SHELL_STATUS\r
EFIAPI\r
ShellCommandRunVol (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  EFI_STATUS    Status;\r
  LIST_ENTRY    *Package;\r
  CHAR16        *ProblemParam;\r
  SHELL_STATUS  ShellStatus;\r
  CONST CHAR16  *PathName;\r
  CONST CHAR16  *CurDir;\r
  BOOLEAN       DeleteMode;\r
  CHAR16        *FullPath;\r
  CHAR16        *TempSpot;\r
  UINTN         Length;\r
  CONST CHAR16  *NewName;\r
\r
  Length              = 0;\r
  ProblemParam        = NULL;\r
  ShellStatus         = SHELL_SUCCESS;\r
  PathName            = NULL;\r
  CurDir              = NULL;\r
  FullPath            = NULL;\r
\r
  //\r
  // initialize the shell lib (we must be in non-auto-init...)\r
  //\r
  Status = ShellInitialize();\r
  ASSERT_EFI_ERROR(Status);\r
\r
  //\r
  // Fix local copies of the protocol pointers\r
  //\r
  Status = CommandInit();\r
  ASSERT_EFI_ERROR(Status);\r
\r
  //\r
  // parse the command line\r
  //\r
  Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
  if (EFI_ERROR(Status)) {\r
    if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, L"vol", ProblemParam);\r
      FreePool(ProblemParam);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
    } else {\r
      ASSERT(FALSE);\r
    }\r
  } else {\r
    //\r
    // check for "-?"\r
    //\r
    if (ShellCommandLineGetFlag(Package, L"-?")) {\r
      ASSERT(FALSE);\r
    }\r
\r
    if (ShellCommandLineGetCount(Package) > 2) {\r
      ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellLevel2HiiHandle, L"vol");\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
    } else {\r
      PathName = ShellCommandLineGetRawValue(Package, 1);\r
      if (PathName == NULL) {\r
        CurDir = gEfiShellProtocol->GetCurDir(NULL);\r
        if (CurDir == NULL) {\r
          ShellStatus = SHELL_NOT_FOUND;\r
          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellLevel2HiiHandle, L"vol");\r
        } else {\r
          PathName = CurDir;\r
        }\r
      }\r
      if (PathName != NULL) {\r
        TempSpot = StrStr(PathName, L":");\r
        if (TempSpot != NULL) {\r
          *TempSpot = CHAR_NULL;\r
        }\r
        TempSpot = StrStr(PathName, L"\\");\r
        if (TempSpot != NULL) {\r
          *TempSpot = CHAR_NULL;\r
        }\r
        StrnCatGrow(&FullPath, &Length, PathName, 0);\r
        StrnCatGrow(&FullPath, &Length, L":\\", 0);\r
        DeleteMode = ShellCommandLineGetFlag(Package, L"-d");\r
        NewName    = ShellCommandLineGetValue(Package, L"-n");\r
        if (DeleteMode && ShellCommandLineGetFlag(Package, L"-n")) {\r
          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CONFLICT), gShellLevel2HiiHandle, L"vol", L"-d", L"-n");\r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
        } else if (ShellCommandLineGetFlag(Package, L"-n") && NewName == NULL) {\r
          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_VALUE), gShellLevel2HiiHandle, L"vol", L"-n");\r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
        } else if (NewName != NULL && StrLen(NewName) > 11) {\r
          ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM_VAL), gShellLevel2HiiHandle, L"vol", NewName, L"-n");\r
          ShellStatus = SHELL_INVALID_PARAMETER;\r
        } else if (ShellStatus == SHELL_SUCCESS) {\r
          ShellStatus = HandleVol(\r
            FullPath,\r
            DeleteMode,\r
            NewName\r
           );\r
        }\r
      }\r
    }\r
  }\r
\r
  SHELL_FREE_NON_NULL(FullPath);\r
\r
  //\r
  // free the command line package\r
  //\r
  ShellCommandLineFreeVarList (Package);\r
\r
  return (ShellStatus);\r
}\r
Commit	Line	Data
b54fd049	1	/** @file\r
	2	Main file for vol shell level 2 function.\r
	3	\r
c011b6c9	4	(C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>\r
ba0014b9	5	Copyright (c) 2011 - 2018, Intel Corporation. All rights reserved.<BR>\r
b54fd049	6	This program and the accompanying materials\r
	7	are licensed and made available under the terms and conditions of the BSD License\r
	8	which accompanies this distribution. The full text of the license may be found at\r
	9	http://opensource.org/licenses/bsd-license.php\r
	10	\r
	11	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	12	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	13	\r
	14	**/\r
	15	\r
	16	#include "UefiShellLevel2CommandsLib.h"\r
	17	#include <Guid/FileSystemInfo.h>\r
	18	#include <Guid/FileSystemVolumeLabelInfo.h>\r
	19	\r
	20	/**\r
	21	Print the info or change the volume info.\r
	22	\r
	23	@param[in] Path String with starting path.\r
	24	@param[in] Delete TRUE to delete the volume label. FALSE otherwise.\r
	25	@param[in] Name New name to set to the volume label.\r
	26	\r
	27	@retval SHELL_SUCCESS The operation was sucessful.\r
	28	**/\r
	29	SHELL_STATUS\r
b54fd049	30	HandleVol(\r
	31	IN CONST CHAR16 *Path,\r
	32	IN CONST BOOLEAN Delete,\r
ae724571	33	IN CONST CHAR16 *Name OPTIONAL\r
b54fd049	34	)\r
	35	{\r
	36	EFI_STATUS Status;\r
	37	SHELL_STATUS ShellStatus;\r
	38	EFI_FILE_SYSTEM_INFO *SysInfo;\r
	39	UINTN SysInfoSize;\r
	40	SHELL_FILE_HANDLE ShellFileHandle;\r
	41	EFI_FILE_PROTOCOL *EfiFpHandle;\r
	42	UINTN Size1;\r
	43	UINTN Size2;\r
	44	\r
	45	ShellStatus = SHELL_SUCCESS;\r
	46	\r
6b825919	47	if (\r
ae724571	48	Name != NULL && (\r
6b825919	49	StrStr(Name, L"%") != NULL \|\|\r
	50	StrStr(Name, L"^") != NULL \|\|\r
	51	StrStr(Name, L"*") != NULL \|\|\r
	52	StrStr(Name, L"+") != NULL \|\|\r
	53	StrStr(Name, L"=") != NULL \|\|\r
	54	StrStr(Name, L"[") != NULL \|\|\r
	55	StrStr(Name, L"]") != NULL \|\|\r
	56	StrStr(Name, L"\|") != NULL \|\|\r
	57	StrStr(Name, L":") != NULL \|\|\r
	58	StrStr(Name, L";") != NULL \|\|\r
	59	StrStr(Name, L"\"") != NULL \|\|\r
	60	StrStr(Name, L"<") != NULL \|\|\r
	61	StrStr(Name, L">") != NULL \|\|\r
	62	StrStr(Name, L"?") != NULL \|\|\r
	63	StrStr(Name, L"/") != NULL \|\|\r
ae724571	64	StrStr(Name, L" ") != NULL )\r
6b825919	65	){\r
ba0014b9	66	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellLevel2HiiHandle, L"vol", Name);\r
6b825919	67	return (SHELL_INVALID_PARAMETER);\r
	68	}\r
	69	\r
b54fd049	70	Status = gEfiShellProtocol->OpenFileByName(\r
	71	Path,\r
	72	&ShellFileHandle,\r
	73	Name != NULL?EFI_FILE_MODE_READ\|EFI_FILE_MODE_WRITE:EFI_FILE_MODE_READ);\r
	74	\r
	75	if (EFI_ERROR(Status) \|\| ShellFileHandle == NULL) {\r
ba0014b9	76	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_OPEN_FAIL), gShellLevel2HiiHandle, L"vol", Path);\r
6b825919	77	return (SHELL_ACCESS_DENIED);\r
b54fd049	78	}\r
	79	\r
	80	//\r
	81	// Get the Volume Info from ShellFileHandle\r
	82	//\r
	83	SysInfo = NULL;\r
	84	SysInfoSize = 0;\r
	85	EfiFpHandle = ConvertShellHandleToEfiFileProtocol(ShellFileHandle);\r
	86	Status = EfiFpHandle->GetInfo(\r
	87	EfiFpHandle,\r
	88	&gEfiFileSystemInfoGuid,\r
	89	&SysInfoSize,\r
	90	SysInfo);\r
	91	\r
	92	if (Status == EFI_BUFFER_TOO_SMALL) {\r
	93	SysInfo = AllocateZeroPool(SysInfoSize);\r
	94	Status = EfiFpHandle->GetInfo(\r
	95	EfiFpHandle,\r
	96	&gEfiFileSystemInfoGuid,\r
	97	&SysInfoSize,\r
	98	SysInfo);\r
	99	}\r
	100	\r
33c031ee	101	ASSERT(SysInfo != NULL);\r
33c031ee	102	\r
b54fd049	103	if (Delete) {\r
cea5e3b9	104	((CHAR16 ) SysInfo->VolumeLabel) = CHAR_NULL;\r
b54fd049	105	SysInfo->Size = SIZE_OF_EFI_FILE_SYSTEM_INFO + StrSize(SysInfo->VolumeLabel);\r
	106	Status = EfiFpHandle->SetInfo(\r
	107	EfiFpHandle,\r
	108	&gEfiFileSystemInfoGuid,\r
	109	(UINTN)SysInfo->Size,\r
	110	SysInfo);\r
	111	} else if (Name != NULL) {\r
	112	Size1 = StrSize(Name);\r
	113	Size2 = StrSize(SysInfo->VolumeLabel);\r
	114	if (Size1 > Size2) {\r
	115	SysInfo = ReallocatePool((UINTN)SysInfo->Size, (UINTN)SysInfo->Size + Size1 - Size2, SysInfo);\r
cbdd109b	116	if (SysInfo == NULL) {\r
ba0014b9	117	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellLevel2HiiHandle, L"vol");\r
cbdd109b	118	ShellStatus = SHELL_OUT_OF_RESOURCES;\r
ba0014b9	119	}\r
cbdd109b	120	}\r
cbdd109b	121	if (SysInfo != NULL) {\r
ba0014b9 LG	122	StrCpyS ( (CHAR16 *) SysInfo->VolumeLabel,\r
ba0014b9 LG	123	(Size1>Size2? Size1/sizeof(CHAR16) : Size2/sizeof(CHAR16)),\r
e75390f0 QS	124	Name\r
e75390f0 QS	125	);\r
cbdd109b	126	SysInfo->Size = SIZE_OF_EFI_FILE_SYSTEM_INFO + Size1;\r
	127	Status = EfiFpHandle->SetInfo(\r
	128	EfiFpHandle,\r
	129	&gEfiFileSystemInfoGuid,\r
	130	(UINTN)SysInfo->Size,\r
	131	SysInfo);\r
b54fd049	132	}\r
ba0014b9	133	}\r
b54fd049	134	\r
	135	FreePool(SysInfo);\r
	136	\r
	137	if (Delete \|\| Name != NULL) {\r
	138	if (EFI_ERROR(Status)) {\r
ba0014b9	139	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_AD), gShellLevel2HiiHandle, L"vol", Path);\r
b54fd049	140	ShellStatus = SHELL_ACCESS_DENIED;\r
	141	}\r
	142	}\r
	143	\r
e35b5317	144	SysInfoSize = 0;\r
	145	SysInfo = NULL;\r
	146	\r
b54fd049	147	Status = EfiFpHandle->GetInfo(\r
	148	EfiFpHandle,\r
	149	&gEfiFileSystemInfoGuid,\r
	150	&SysInfoSize,\r
	151	SysInfo);\r
	152	\r
e35b5317	153	if (Status == EFI_BUFFER_TOO_SMALL) {\r
	154	SysInfo = AllocateZeroPool(SysInfoSize);\r
	155	Status = EfiFpHandle->GetInfo(\r
	156	EfiFpHandle,\r
	157	&gEfiFileSystemInfoGuid,\r
	158	&SysInfoSize,\r
	159	SysInfo);\r
	160	}\r
	161	\r
b54fd049	162	gEfiShellProtocol->CloseFile(ShellFileHandle);\r
ba0014b9	163	\r
ae724571	164	ASSERT(SysInfo != NULL);\r
b54fd049	165	\r
cbdd109b	166	if (SysInfo != NULL) {\r
	167	//\r
	168	// print VolumeInfo table\r
	169	//\r
	170	ShellPrintHiiEx (\r
	171	0,\r
	172	gST->ConOut->Mode->CursorRow,\r
	173	NULL,\r
	174	STRING_TOKEN (STR_VOL_VOLINFO),\r
	175	gShellLevel2HiiHandle,\r
	176	SysInfo->VolumeLabel,\r
	177	SysInfo->ReadOnly?L"r":L"rw",\r
	178	SysInfo->VolumeSize,\r
	179	SysInfo->FreeSpace,\r
	180	SysInfo->BlockSize\r
	181	);\r
	182	SHELL_FREE_NON_NULL(SysInfo);\r
	183	}\r
b54fd049	184	\r
	185	return (ShellStatus);\r
	186	}\r
	187	\r
	188	STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
	189	{L"-d", TypeFlag},\r
	190	{L"-n", TypeValue},\r
	191	{NULL, TypeMax}\r
	192	};\r
	193	\r
	194	/**\r
	195	Function for 'Vol' command.\r
	196	\r
	197	@param[in] ImageHandle Handle to the Image (NULL if Internal).\r
	198	@param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
	199	**/\r
	200	SHELL_STATUS\r
	201	EFIAPI\r
	202	ShellCommandRunVol (\r
	203	IN EFI_HANDLE ImageHandle,\r
	204	IN EFI_SYSTEM_TABLE *SystemTable\r
	205	)\r
	206	{\r
	207	EFI_STATUS Status;\r
	208	LIST_ENTRY *Package;\r
	209	CHAR16 *ProblemParam;\r
	210	SHELL_STATUS ShellStatus;\r
	211	CONST CHAR16 *PathName;\r
	212	CONST CHAR16 *CurDir;\r
	213	BOOLEAN DeleteMode;\r
	214	CHAR16 *FullPath;\r
e35b5317	215	CHAR16 *TempSpot;\r
b54fd049	216	UINTN Length;\r
ae724571	217	CONST CHAR16 *NewName;\r
b54fd049	218	\r
	219	Length = 0;\r
	220	ProblemParam = NULL;\r
	221	ShellStatus = SHELL_SUCCESS;\r
	222	PathName = NULL;\r
	223	CurDir = NULL;\r
	224	FullPath = NULL;\r
	225	\r
	226	//\r
	227	// initialize the shell lib (we must be in non-auto-init...)\r
	228	//\r
	229	Status = ShellInitialize();\r
	230	ASSERT_EFI_ERROR(Status);\r
	231	\r
	232	//\r
	233	// Fix local copies of the protocol pointers\r
	234	//\r
	235	Status = CommandInit();\r
	236	ASSERT_EFI_ERROR(Status);\r
	237	\r
	238	//\r
	239	// parse the command line\r
	240	//\r
	241	Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
	242	if (EFI_ERROR(Status)) {\r
	243	if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
ba0014b9	244	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, L"vol", ProblemParam);\r
b54fd049	245	FreePool(ProblemParam);\r
	246	ShellStatus = SHELL_INVALID_PARAMETER;\r
	247	} else {\r
	248	ASSERT(FALSE);\r
	249	}\r
	250	} else {\r
	251	//\r
	252	// check for "-?"\r
	253	//\r
	254	if (ShellCommandLineGetFlag(Package, L"-?")) {\r
	255	ASSERT(FALSE);\r
	256	}\r
	257	\r
	258	if (ShellCommandLineGetCount(Package) > 2) {\r
ba0014b9	259	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellLevel2HiiHandle, L"vol");\r
b54fd049	260	ShellStatus = SHELL_INVALID_PARAMETER;\r
	261	} else {\r
	262	PathName = ShellCommandLineGetRawValue(Package, 1);\r
	263	if (PathName == NULL) {\r
	264	CurDir = gEfiShellProtocol->GetCurDir(NULL);\r
	265	if (CurDir == NULL) {\r
	266	ShellStatus = SHELL_NOT_FOUND;\r
ba0014b9	267	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellLevel2HiiHandle, L"vol");\r
b54fd049	268	} else {\r
	269	PathName = CurDir;\r
	270	}\r
	271	}\r
	272	if (PathName != NULL) {\r
e35b5317	273	TempSpot = StrStr(PathName, L":");\r
	274	if (TempSpot != NULL) {\r
	275	*TempSpot = CHAR_NULL;\r
	276	}\r
	277	TempSpot = StrStr(PathName, L"\\");\r
	278	if (TempSpot != NULL) {\r
	279	*TempSpot = CHAR_NULL;\r
b54fd049	280	}\r
e35b5317	281	StrnCatGrow(&FullPath, &Length, PathName, 0);\r
e35b5317	282	StrnCatGrow(&FullPath, &Length, L":\\", 0);\r
b54fd049	283	DeleteMode = ShellCommandLineGetFlag(Package, L"-d");\r
ae724571	284	NewName = ShellCommandLineGetValue(Package, L"-n");\r
b54fd049	285	if (DeleteMode && ShellCommandLineGetFlag(Package, L"-n")) {\r
ba0014b9	286	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CONFLICT), gShellLevel2HiiHandle, L"vol", L"-d", L"-n");\r
b54fd049	287	ShellStatus = SHELL_INVALID_PARAMETER;\r
ae724571	288	} else if (ShellCommandLineGetFlag(Package, L"-n") && NewName == NULL) {\r
ba0014b9	289	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_VALUE), gShellLevel2HiiHandle, L"vol", L"-n");\r
b54fd049	290	ShellStatus = SHELL_INVALID_PARAMETER;\r
ae724571	291	} else if (NewName != NULL && StrLen(NewName) > 11) {\r
ba0014b9	292	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM_VAL), gShellLevel2HiiHandle, L"vol", NewName, L"-n");\r
b54fd049	293	ShellStatus = SHELL_INVALID_PARAMETER;\r
	294	} else if (ShellStatus == SHELL_SUCCESS) {\r
	295	ShellStatus = HandleVol(\r
	296	FullPath,\r
	297	DeleteMode,\r
ae724571	298	NewName\r
b54fd049	299	);\r
	300	}\r
	301	}\r
	302	}\r
	303	}\r
	304	\r
	305	SHELL_FREE_NON_NULL(FullPath);\r
	306	\r
	307	//\r
	308	// free the command line package\r
	309	//\r
	310	ShellCommandLineFreeVarList (Package);\r
	311	\r
	312	return (ShellStatus);\r
	313	}\r