[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 - 2014, 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
EFIAPI\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
      StrnCpy ((CHAR16 *) SysInfo->VolumeLabel, Name, (Size1 > Size2?Size1/sizeof(CHAR16):Size2/sizeof(CHAR16))-1);\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
cea5e3b9	5	Copyright (c) 2011 - 2014, 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
	30	EFIAPI\r
	31	HandleVol(\r
	32	IN CONST CHAR16 *Path,\r
	33	IN CONST BOOLEAN Delete,\r
ae724571	34	IN CONST CHAR16 *Name OPTIONAL\r
b54fd049	35	)\r
	36	{\r
	37	EFI_STATUS Status;\r
	38	SHELL_STATUS ShellStatus;\r
	39	EFI_FILE_SYSTEM_INFO *SysInfo;\r
	40	UINTN SysInfoSize;\r
	41	SHELL_FILE_HANDLE ShellFileHandle;\r
	42	EFI_FILE_PROTOCOL *EfiFpHandle;\r
	43	UINTN Size1;\r
	44	UINTN Size2;\r
	45	\r
	46	ShellStatus = SHELL_SUCCESS;\r
	47	\r
6b825919	48	if (\r
ae724571	49	Name != NULL && (\r
6b825919	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
	64	StrStr(Name, L"/") != NULL \|\|\r
ae724571	65	StrStr(Name, L" ") != NULL )\r
6b825919	66	){\r
099e8ff5	67	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellLevel2HiiHandle, L"vol", Name); \r
6b825919	68	return (SHELL_INVALID_PARAMETER);\r
	69	}\r
	70	\r
b54fd049	71	Status = gEfiShellProtocol->OpenFileByName(\r
	72	Path,\r
	73	&ShellFileHandle,\r
	74	Name != NULL?EFI_FILE_MODE_READ\|EFI_FILE_MODE_WRITE:EFI_FILE_MODE_READ);\r
	75	\r
	76	if (EFI_ERROR(Status) \|\| ShellFileHandle == NULL) {\r
099e8ff5	77	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_OPEN_FAIL), gShellLevel2HiiHandle, L"vol", Path); \r
6b825919	78	return (SHELL_ACCESS_DENIED);\r
b54fd049	79	}\r
	80	\r
	81	//\r
	82	// Get the Volume Info from ShellFileHandle\r
	83	//\r
	84	SysInfo = NULL;\r
	85	SysInfoSize = 0;\r
	86	EfiFpHandle = ConvertShellHandleToEfiFileProtocol(ShellFileHandle);\r
	87	Status = EfiFpHandle->GetInfo(\r
	88	EfiFpHandle,\r
	89	&gEfiFileSystemInfoGuid,\r
	90	&SysInfoSize,\r
	91	SysInfo);\r
	92	\r
	93	if (Status == EFI_BUFFER_TOO_SMALL) {\r
	94	SysInfo = AllocateZeroPool(SysInfoSize);\r
	95	Status = EfiFpHandle->GetInfo(\r
	96	EfiFpHandle,\r
	97	&gEfiFileSystemInfoGuid,\r
	98	&SysInfoSize,\r
	99	SysInfo);\r
	100	}\r
	101	\r
33c031ee	102	ASSERT(SysInfo != NULL);\r
33c031ee	103	\r
b54fd049	104	if (Delete) {\r
cea5e3b9	105	((CHAR16 ) SysInfo->VolumeLabel) = CHAR_NULL;\r
b54fd049	106	SysInfo->Size = SIZE_OF_EFI_FILE_SYSTEM_INFO + StrSize(SysInfo->VolumeLabel);\r
	107	Status = EfiFpHandle->SetInfo(\r
	108	EfiFpHandle,\r
	109	&gEfiFileSystemInfoGuid,\r
	110	(UINTN)SysInfo->Size,\r
	111	SysInfo);\r
	112	} else if (Name != NULL) {\r
	113	Size1 = StrSize(Name);\r
	114	Size2 = StrSize(SysInfo->VolumeLabel);\r
	115	if (Size1 > Size2) {\r
	116	SysInfo = ReallocatePool((UINTN)SysInfo->Size, (UINTN)SysInfo->Size + Size1 - Size2, SysInfo);\r
cbdd109b	117	if (SysInfo == NULL) {\r
099e8ff5	118	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_OUT_MEM), gShellLevel2HiiHandle, L"vol"); \r
cbdd109b	119	ShellStatus = SHELL_OUT_OF_RESOURCES;\r
	120	} \r
	121	}\r
	122	if (SysInfo != NULL) {\r
cea5e3b9	123	StrnCpy ((CHAR16 *) SysInfo->VolumeLabel, Name, (Size1 > Size2?Size1/sizeof(CHAR16):Size2/sizeof(CHAR16))-1);\r
cbdd109b	124	SysInfo->Size = SIZE_OF_EFI_FILE_SYSTEM_INFO + Size1;\r
	125	Status = EfiFpHandle->SetInfo(\r
	126	EfiFpHandle,\r
	127	&gEfiFileSystemInfoGuid,\r
	128	(UINTN)SysInfo->Size,\r
	129	SysInfo);\r
b54fd049	130	}\r
b54fd049	131	} \r
	132	\r
	133	FreePool(SysInfo);\r
	134	\r
	135	if (Delete \|\| Name != NULL) {\r
	136	if (EFI_ERROR(Status)) {\r
099e8ff5	137	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_AD), gShellLevel2HiiHandle, L"vol", Path); \r
b54fd049	138	ShellStatus = SHELL_ACCESS_DENIED;\r
	139	}\r
	140	}\r
	141	\r
e35b5317	142	SysInfoSize = 0;\r
	143	SysInfo = NULL;\r
	144	\r
b54fd049	145	Status = EfiFpHandle->GetInfo(\r
	146	EfiFpHandle,\r
	147	&gEfiFileSystemInfoGuid,\r
	148	&SysInfoSize,\r
	149	SysInfo);\r
	150	\r
e35b5317	151	if (Status == EFI_BUFFER_TOO_SMALL) {\r
	152	SysInfo = AllocateZeroPool(SysInfoSize);\r
	153	Status = EfiFpHandle->GetInfo(\r
	154	EfiFpHandle,\r
	155	&gEfiFileSystemInfoGuid,\r
	156	&SysInfoSize,\r
	157	SysInfo);\r
	158	}\r
	159	\r
b54fd049	160	gEfiShellProtocol->CloseFile(ShellFileHandle);\r
ae724571	161	\r
ae724571	162	ASSERT(SysInfo != NULL);\r
b54fd049	163	\r
cbdd109b	164	if (SysInfo != NULL) {\r
	165	//\r
	166	// print VolumeInfo table\r
	167	//\r
	168	ShellPrintHiiEx (\r
	169	0,\r
	170	gST->ConOut->Mode->CursorRow,\r
	171	NULL,\r
	172	STRING_TOKEN (STR_VOL_VOLINFO),\r
	173	gShellLevel2HiiHandle,\r
	174	SysInfo->VolumeLabel,\r
	175	SysInfo->ReadOnly?L"r":L"rw",\r
	176	SysInfo->VolumeSize,\r
	177	SysInfo->FreeSpace,\r
	178	SysInfo->BlockSize\r
	179	);\r
	180	SHELL_FREE_NON_NULL(SysInfo);\r
	181	}\r
b54fd049	182	\r
	183	return (ShellStatus);\r
	184	}\r
	185	\r
	186	STATIC CONST SHELL_PARAM_ITEM ParamList[] = {\r
	187	{L"-d", TypeFlag},\r
	188	{L"-n", TypeValue},\r
	189	{NULL, TypeMax}\r
	190	};\r
	191	\r
	192	/**\r
	193	Function for 'Vol' command.\r
	194	\r
	195	@param[in] ImageHandle Handle to the Image (NULL if Internal).\r
	196	@param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
	197	**/\r
	198	SHELL_STATUS\r
	199	EFIAPI\r
	200	ShellCommandRunVol (\r
	201	IN EFI_HANDLE ImageHandle,\r
	202	IN EFI_SYSTEM_TABLE *SystemTable\r
	203	)\r
	204	{\r
	205	EFI_STATUS Status;\r
	206	LIST_ENTRY *Package;\r
	207	CHAR16 *ProblemParam;\r
	208	SHELL_STATUS ShellStatus;\r
	209	CONST CHAR16 *PathName;\r
	210	CONST CHAR16 *CurDir;\r
	211	BOOLEAN DeleteMode;\r
	212	CHAR16 *FullPath;\r
e35b5317	213	CHAR16 *TempSpot;\r
b54fd049	214	UINTN Length;\r
ae724571	215	CONST CHAR16 *NewName;\r
b54fd049	216	\r
	217	Length = 0;\r
	218	ProblemParam = NULL;\r
	219	ShellStatus = SHELL_SUCCESS;\r
	220	PathName = NULL;\r
	221	CurDir = NULL;\r
	222	FullPath = NULL;\r
	223	\r
	224	//\r
	225	// initialize the shell lib (we must be in non-auto-init...)\r
	226	//\r
	227	Status = ShellInitialize();\r
	228	ASSERT_EFI_ERROR(Status);\r
	229	\r
	230	//\r
	231	// Fix local copies of the protocol pointers\r
	232	//\r
	233	Status = CommandInit();\r
	234	ASSERT_EFI_ERROR(Status);\r
	235	\r
	236	//\r
	237	// parse the command line\r
	238	//\r
	239	Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);\r
	240	if (EFI_ERROR(Status)) {\r
	241	if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
099e8ff5	242	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, L"vol", ProblemParam); \r
b54fd049	243	FreePool(ProblemParam);\r
	244	ShellStatus = SHELL_INVALID_PARAMETER;\r
	245	} else {\r
	246	ASSERT(FALSE);\r
	247	}\r
	248	} else {\r
	249	//\r
	250	// check for "-?"\r
	251	//\r
	252	if (ShellCommandLineGetFlag(Package, L"-?")) {\r
	253	ASSERT(FALSE);\r
	254	}\r
	255	\r
	256	if (ShellCommandLineGetCount(Package) > 2) {\r
099e8ff5	257	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellLevel2HiiHandle, L"vol"); \r
b54fd049	258	ShellStatus = SHELL_INVALID_PARAMETER;\r
	259	} else {\r
	260	PathName = ShellCommandLineGetRawValue(Package, 1);\r
	261	if (PathName == NULL) {\r
	262	CurDir = gEfiShellProtocol->GetCurDir(NULL);\r
	263	if (CurDir == NULL) {\r
	264	ShellStatus = SHELL_NOT_FOUND;\r
099e8ff5	265	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellLevel2HiiHandle, L"vol"); \r
b54fd049	266	} else {\r
	267	PathName = CurDir;\r
	268	}\r
	269	}\r
	270	if (PathName != NULL) {\r
e35b5317	271	TempSpot = StrStr(PathName, L":");\r
	272	if (TempSpot != NULL) {\r
	273	*TempSpot = CHAR_NULL;\r
	274	}\r
	275	TempSpot = StrStr(PathName, L"\\");\r
	276	if (TempSpot != NULL) {\r
	277	*TempSpot = CHAR_NULL;\r
b54fd049	278	}\r
e35b5317	279	StrnCatGrow(&FullPath, &Length, PathName, 0);\r
e35b5317	280	StrnCatGrow(&FullPath, &Length, L":\\", 0);\r
b54fd049	281	DeleteMode = ShellCommandLineGetFlag(Package, L"-d");\r
ae724571	282	NewName = ShellCommandLineGetValue(Package, L"-n");\r
b54fd049	283	if (DeleteMode && ShellCommandLineGetFlag(Package, L"-n")) {\r
099e8ff5	284	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_CONFLICT), gShellLevel2HiiHandle, L"vol", L"-d", L"-n"); \r
b54fd049	285	ShellStatus = SHELL_INVALID_PARAMETER;\r
ae724571	286	} else if (ShellCommandLineGetFlag(Package, L"-n") && NewName == NULL) {\r
099e8ff5	287	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_VALUE), gShellLevel2HiiHandle, L"vol", L"-n"); \r
b54fd049	288	ShellStatus = SHELL_INVALID_PARAMETER;\r
ae724571	289	} else if (NewName != NULL && StrLen(NewName) > 11) {\r
099e8ff5	290	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM_VAL), gShellLevel2HiiHandle, L"vol", NewName, L"-n"); \r
b54fd049	291	ShellStatus = SHELL_INVALID_PARAMETER;\r
	292	} else if (ShellStatus == SHELL_SUCCESS) {\r
	293	ShellStatus = HandleVol(\r
	294	FullPath,\r
	295	DeleteMode,\r
ae724571	296	NewName\r
b54fd049	297	);\r
	298	}\r
	299	}\r
	300	}\r
	301	}\r
	302	\r
	303	SHELL_FREE_NON_NULL(FullPath);\r
	304	\r
	305	//\r
	306	// free the command line package\r
	307	//\r
	308	ShellCommandLineFreeVarList (Package);\r
	309	\r
	310	return (ShellStatus);\r
	311	}\r