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

/** @file\r
  Main file for ls shell level 2 function.\r
\r
  Copyright (c) 2009 - 2010, 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
\r
/**\r
  print out the list of files and directories from the LS command\r
\r
  @param[in] Rec            TRUE to automatically recurse into each found directory\r
                            FALSE to only list the specified directory.\r
  @param[in] Attribs        List of required Attribute for display.\r
                            If 0 then all non-system and non-hidden files will be printed.\r
  @param[in] Sfo            TRUE to use Standard Format Output, FALSE otherwise\r
  @param[in] Path           String with starting path.\r
  @param[in] First          TRUE for the original and FALSE for any recursion spawned instances.\r
  @param[in] Count          The count of bits enabled in Attribs.\r
  @param[in] TimeZone       The current time zone offset.\r
\r
  @retval SHELL_SUCCESS     the printing was sucessful.\r
**/\r
SHELL_STATUS\r
EFIAPI\r
PrintLsOutput(\r
  IN CONST BOOLEAN Rec,\r
  IN CONST UINT64  Attribs,\r
  IN CONST BOOLEAN Sfo,\r
  IN CONST CHAR16  *Path,\r
  IN CONST BOOLEAN First,\r
  IN CONST UINTN   Count,\r
  IN CONST INT16   TimeZone\r
  )\r
{\r
  EFI_STATUS            Status;\r
  EFI_SHELL_FILE_INFO   *ListHead;\r
  EFI_SHELL_FILE_INFO   *Node;\r
  SHELL_STATUS          ShellStatus;\r
  UINT64                FileCount;\r
  UINT64                DirCount;\r
  UINT64                FileSize;\r
  CHAR16                *DirectoryName;\r
  UINTN                 LongestPath;\r
  EFI_FILE_SYSTEM_INFO  *SysInfo;\r
  UINTN                 SysInfoSize;\r
  SHELL_FILE_HANDLE     ShellFileHandle;\r
  CHAR16                *CorrectedPath;\r
  EFI_FILE_PROTOCOL     *EfiFpHandle;\r
\r
  FileCount     = 0;\r
  DirCount      = 0;\r
  FileSize      = 0;\r
  ListHead      = NULL;\r
  ShellStatus   = SHELL_SUCCESS;\r
  LongestPath   = 0;\r
  CorrectedPath = NULL;\r
\r
  CorrectedPath = StrnCatGrow(&CorrectedPath, NULL, Path, 0);\r
  ASSERT(CorrectedPath != NULL);\r
  ShellCommandCleanPath(CorrectedPath);\r
\r
  Status = ShellOpenFileMetaArg((CHAR16*)CorrectedPath, EFI_FILE_MODE_READ, &ListHead);\r
  if (EFI_ERROR(Status)) {\r
    return (SHELL_DEVICE_ERROR);\r
  }\r
  if (ListHead == NULL || IsListEmpty(&ListHead->Link)) {\r
    //\r
    // On the first one only we expect to find something...\r
    // do we find the . and .. directories otherwise?\r
    //\r
    if (First) {\r
      return (SHELL_NOT_FOUND);\r
    }\r
    return (SHELL_SUCCESS);\r
  }\r
\r
  if (Sfo && First) {\r
    //\r
    // Get the first valid handle (directories)\r
    //\r
    for ( Node = (EFI_SHELL_FILE_INFO *)GetFirstNode(&ListHead->Link)\r
        ; !IsNull(&ListHead->Link, &Node->Link) && Node->Handle == NULL\r
        ; Node = (EFI_SHELL_FILE_INFO *)GetNextNode(&ListHead->Link, &Node->Link)\r
       );\r
\r
    if (Node->Handle == NULL) {\r
      DirectoryName = GetFullyQualifiedPath(((EFI_SHELL_FILE_INFO *)GetFirstNode(&ListHead->Link))->FullName);\r
\r
      //\r
      // We need to open something up to get system information\r
      //\r
      Status = gEfiShellProtocol->OpenFileByName(\r
        DirectoryName,\r
        &ShellFileHandle,\r
        EFI_FILE_MODE_READ);\r
\r
      ASSERT_EFI_ERROR(Status);\r
      FreePool(DirectoryName);\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_EFI_ERROR(Status);\r
\r
      gEfiShellProtocol->CloseFile(ShellFileHandle);\r
    } else {\r
      //\r
      // Get the Volume Info from Node->Handle\r
      //\r
      SysInfo = NULL;\r
      SysInfoSize = 0;\r
      EfiFpHandle = ConvertShellHandleToEfiFileProtocol(Node->Handle);\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_EFI_ERROR(Status);\r
    }\r
\r
    ShellPrintHiiEx (\r
      -1,\r
      -1,\r
      NULL,\r
      STRING_TOKEN (STR_GEN_SFO_HEADER),\r
      gShellLevel2HiiHandle,\r
      L"ls");\r
    //\r
    // print VolumeInfo table\r
    //\r
    ASSERT(SysInfo != NULL);\r
    ShellPrintHiiEx (\r
      0,\r
      gST->ConOut->Mode->CursorRow,\r
      NULL,\r
      STRING_TOKEN (STR_LS_SFO_VOLINFO),\r
      gShellLevel2HiiHandle,\r
      SysInfo->VolumeLabel,\r
      SysInfo->VolumeSize,\r
      SysInfo->ReadOnly?L"TRUE":L"FALSE",\r
      SysInfo->FreeSpace,\r
      SysInfo->BlockSize\r
     );\r
    if (SysInfo != NULL) {\r
      FreePool(SysInfo);\r
    }\r
  }\r
\r
  if (!Sfo) {\r
    //\r
    // get directory name from path...\r
    //\r
    DirectoryName = GetFullyQualifiedPath(CorrectedPath);\r
\r
    //\r
    // print header\r
    //\r
    ShellPrintHiiEx (\r
      0,\r
      gST->ConOut->Mode->CursorRow,\r
      NULL,\r
      STRING_TOKEN (STR_LS_HEADER_LINE1),\r
      gShellLevel2HiiHandle,\r
      DirectoryName\r
     );\r
    FreePool(DirectoryName);\r
  }\r
  for ( Node = (EFI_SHELL_FILE_INFO *)GetFirstNode(&ListHead->Link)\r
      ; !IsNull(&ListHead->Link, &Node->Link)\r
      ; Node = (EFI_SHELL_FILE_INFO *)GetNextNode(&ListHead->Link, &Node->Link)\r
     ){\r
    ASSERT(Node != NULL);\r
    if (LongestPath < StrSize(Node->FullName)) {\r
      LongestPath = StrSize(Node->FullName);\r
    }\r
    ASSERT(Node->Info != NULL);\r
    ASSERT((Node->Info->Attribute & EFI_FILE_VALID_ATTR) == Node->Info->Attribute);\r
    if (Attribs == 0) {\r
      //\r
      // NOT system & NOT hidden\r
      //\r
      if ( (Node->Info->Attribute & EFI_FILE_SYSTEM)\r
        || (Node->Info->Attribute & EFI_FILE_HIDDEN)\r
       ){\r
        continue;\r
      }\r
    } else if (Attribs != EFI_FILE_VALID_ATTR) {\r
      if (Count == 1) {\r
        //\r
        // the bit must match\r
        //\r
        if ( (Node->Info->Attribute & Attribs) != Attribs) {\r
          continue;\r
        }\r
      } else {\r
        //\r
        // exact match on all bits\r
        //\r
        if ( Node->Info->Attribute != Attribs) {\r
          continue;\r
        }\r
      }\r
    }\r
\r
    if (Sfo) {\r
      //\r
      // Print the FileInfo Table\r
      //\r
    ShellPrintHiiEx (\r
      0,\r
      gST->ConOut->Mode->CursorRow,\r
      NULL,\r
      STRING_TOKEN (STR_LS_SFO_FILEINFO),\r
      gShellLevel2HiiHandle,\r
      Node->FullName,\r
      Node->Info->FileSize,\r
      Node->Info->PhysicalSize,\r
      (Node->Info->Attribute & EFI_FILE_ARCHIVE)   != 0?L"a":L"",\r
      (Node->Info->Attribute & EFI_FILE_DIRECTORY) != 0?L"d":L"",\r
      (Node->Info->Attribute & EFI_FILE_HIDDEN)    != 0?L"h":L"",\r
      (Node->Info->Attribute & EFI_FILE_READ_ONLY) != 0?L"r":L"",\r
      (Node->Info->Attribute & EFI_FILE_SYSTEM)    != 0?L"s":L"",\r
      Node->Info->CreateTime.Hour,\r
      Node->Info->CreateTime.Minute,\r
      Node->Info->CreateTime.Second,\r
      Node->Info->CreateTime.Day,\r
      Node->Info->CreateTime.Month,\r
      Node->Info->CreateTime.Year,\r
      Node->Info->LastAccessTime.Hour,\r
      Node->Info->LastAccessTime.Minute,\r
      Node->Info->LastAccessTime.Second,\r
      Node->Info->LastAccessTime.Day,\r
      Node->Info->LastAccessTime.Month,\r
      Node->Info->LastAccessTime.Year,\r
      Node->Info->ModificationTime.Hour,\r
      Node->Info->ModificationTime.Minute,\r
      Node->Info->ModificationTime.Second,\r
      Node->Info->ModificationTime.Day,\r
      Node->Info->ModificationTime.Month,\r
      Node->Info->ModificationTime.Year\r
     );\r
    } else {\r
      //\r
      // print this one out...\r
      // first print the universal start, next print the type specific name format, last print the CRLF\r
      //\r
      ShellPrintHiiEx (\r
        -1,\r
        -1,\r
        NULL,\r
        STRING_TOKEN (STR_LS_LINE_START_ALL),\r
        gShellLevel2HiiHandle,\r
        &Node->Info->ModificationTime,\r
        (Node->Info->Attribute & EFI_FILE_DIRECTORY) != 0?L"<DIR>":L"",\r
        (Node->Info->Attribute & EFI_FILE_READ_ONLY) != 0?L'r':L' ',\r
        Node->Info->FileSize\r
       );\r
      if (Node->Info->Attribute & EFI_FILE_DIRECTORY) {\r
        DirCount++;\r
        ShellPrintHiiEx (\r
          -1,\r
          -1,\r
          NULL,\r
          STRING_TOKEN (STR_LS_LINE_END_DIR),\r
          gShellLevel2HiiHandle,\r
          Node->FileName\r
         );\r
      } else {\r
        FileCount++;\r
        FileSize += Node->Info->FileSize;\r
        if ( (gUnicodeCollation->StriColl(gUnicodeCollation, (CHAR16*)L".nsh", (CHAR16*)&(Node->FileName[StrLen (Node->FileName) - 4])) == 0)\r
          || (gUnicodeCollation->StriColl(gUnicodeCollation, (CHAR16*)L".efi", (CHAR16*)&(Node->FileName[StrLen (Node->FileName) - 4])) == 0)\r
         ){\r
          ShellPrintHiiEx (\r
            -1,\r
            -1,\r
            NULL,\r
            STRING_TOKEN (STR_LS_LINE_END_EXE),\r
            gShellLevel2HiiHandle,\r
            Node->FileName\r
           );\r
        } else {\r
          ShellPrintHiiEx (\r
            -1,\r
            -1,\r
            NULL,\r
            STRING_TOKEN (STR_LS_LINE_END_FILE),\r
            gShellLevel2HiiHandle,\r
            Node->FileName\r
           );\r
        }\r
      }\r
    }\r
  }\r
\r
  if (!Sfo) {\r
    //\r
    // print footer\r
    //\r
    ShellPrintHiiEx (\r
      -1,\r
      -1,\r
      NULL,\r
      STRING_TOKEN (STR_LS_FOOTER_LINE),\r
      gShellLevel2HiiHandle,\r
      FileCount,\r
      FileSize,\r
      DirCount\r
     );\r
  }\r
\r
  if (Rec){\r
    DirectoryName = AllocatePool(LongestPath + 2*sizeof(CHAR16));\r
    for ( Node = (EFI_SHELL_FILE_INFO *)GetFirstNode(&ListHead->Link)\r
        ; !IsNull(&ListHead->Link, &Node->Link)\r
        ; Node = (EFI_SHELL_FILE_INFO *)GetNextNode(&ListHead->Link, &Node->Link)\r
       ){\r
      //\r
      // recurse on any directory except the traversing ones...\r
      //\r
      if (((Node->Info->Attribute & EFI_FILE_DIRECTORY) == EFI_FILE_DIRECTORY)\r
        && StrCmp(Node->FileName, L".") != 0\r
        && StrCmp(Node->FileName, L"..") != 0\r
       ){\r
        StrCpy(DirectoryName, Node->FullName);\r
        StrCat(DirectoryName, L"\\*");\r
        PrintLsOutput(\r
          Rec,\r
          Attribs,\r
          Sfo,\r
          DirectoryName,\r
          FALSE,\r
          Count,\r
          TimeZone);\r
      }\r
    }\r
    FreePool(DirectoryName);\r
  }\r
\r
  FreePool(CorrectedPath);\r
  ShellCloseFileMetaArg(&ListHead);\r
  FreePool(ListHead);\r
  return (ShellStatus);\r
}\r
\r
STATIC CONST SHELL_PARAM_ITEM LsParamList[] = {\r
  {L"-r", TypeFlag},\r
  {L"-a", TypeStart},\r
  {L"-sfo", TypeFlag},\r
  {NULL, TypeMax}\r
  };\r
\r
/**\r
  Function for 'ls' 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
ShellCommandRunLs (\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
  CONST CHAR16  *Attribs;\r
  SHELL_STATUS  ShellStatus;\r
  UINT64        RequiredAttributes;\r
  CONST CHAR16  *PathName;\r
  CONST CHAR16  *CurDir;\r
  UINTN         Count;\r
  CHAR16        *FullPath;\r
  UINTN         Size;\r
  EFI_TIME      theTime;\r
  BOOLEAN       SfoMode;\r
\r
  Size                = 0;\r
  FullPath            = NULL;\r
  ProblemParam        = NULL;\r
  Attribs             = NULL;\r
  ShellStatus         = SHELL_SUCCESS;\r
  RequiredAttributes  = 0;\r
  PathName            = NULL;\r
  CurDir              = NULL;\r
  Count               = 0;\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 (LsParamList, &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, 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);\r
      ShellStatus = SHELL_INVALID_PARAMETER;\r
    } else {\r
      //\r
      // check for -a\r
      //\r
      if (ShellCommandLineGetFlag(Package, L"-a")) {\r
        for ( Attribs = ShellCommandLineGetValue(Package, L"-a")\r
            ; Attribs != NULL && *Attribs != CHAR_NULL && ShellStatus == SHELL_SUCCESS\r
            ; Attribs++\r
           ){\r
          switch (*Attribs) {\r
            case L'a':\r
            case L'A':\r
              RequiredAttributes |= EFI_FILE_ARCHIVE;\r
              Count++;\r
              continue;\r
            case L's':\r
            case L'S':\r
              RequiredAttributes |= EFI_FILE_SYSTEM;\r
              Count++;\r
              continue;\r
            case L'h':\r
            case L'H':\r
              RequiredAttributes |= EFI_FILE_HIDDEN;\r
              Count++;\r
              continue;\r
            case L'r':\r
            case L'R':\r
              RequiredAttributes |= EFI_FILE_READ_ONLY;\r
              Count++;\r
              continue;\r
            case L'd':\r
            case L'D':\r
              RequiredAttributes |= EFI_FILE_DIRECTORY;\r
              Count++;\r
              continue;\r
            default:\r
              ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ATTRIBUTE), gShellLevel2HiiHandle, ShellCommandLineGetValue(Package, L"-a"));\r
              ShellStatus = SHELL_INVALID_PARAMETER;\r
              break;\r
          } // switch\r
        } // for loop\r
        //\r
        // if nothing is specified all are specified\r
        //\r
        if (RequiredAttributes == 0) {\r
          RequiredAttributes = EFI_FILE_VALID_ATTR;\r
        }\r
      } // if -a present\r
      if (ShellStatus == SHELL_SUCCESS) {\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);\r
          }\r
        }\r
        if (PathName != NULL) {\r
          ASSERT((FullPath == NULL && Size == 0) || (FullPath != NULL));\r
          StrnCatGrow(&FullPath, &Size, PathName, 0);\r
          if  (ShellIsDirectory(PathName) == EFI_SUCCESS) {\r
            StrnCatGrow(&FullPath, &Size, L"\\*", 0);\r
          }\r
        } else {\r
          ASSERT(FullPath == NULL);\r
          StrnCatGrow(&FullPath, NULL, L"*", 0);\r
        }\r
        Status = gRT->GetTime(&theTime, NULL);\r
        ASSERT_EFI_ERROR(Status);\r
        SfoMode = ShellCommandLineGetFlag(Package, L"-sfo");\r
        if (ShellStatus == SHELL_SUCCESS) {\r
          ShellStatus = PrintLsOutput(\r
            ShellCommandLineGetFlag(Package, L"-r"),\r
            RequiredAttributes,\r
            SfoMode,\r
            FullPath,\r
            TRUE,\r
            Count,\r
            (INT16)(theTime.TimeZone==2047?0:theTime.TimeZone)\r
           );\r
          if (ShellStatus == SHELL_NOT_FOUND) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_FILES), gShellLevel2HiiHandle);\r
          } else if (ShellStatus == SHELL_INVALID_PARAMETER) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellLevel2HiiHandle);\r
          } else if (ShellStatus != SHELL_SUCCESS) {\r
            ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellLevel2HiiHandle);\r
          }\r
        }\r
      }\r
    }\r
  }\r
\r
  if (FullPath != NULL) {\r
    FreePool(FullPath);\r
  }\r
  //\r
  // free the command line package\r
  //\r
  ShellCommandLineFreeVarList (Package);\r
\r
  return (ShellStatus);\r
}\r
Commit	Line	Data
a405b86d	1	/** @file\r
	2	Main file for ls shell level 2 function.\r
	3	\r
	4	Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
	5	This program and the accompanying materials\r
	6	are licensed and made available under the terms and conditions of the BSD License\r
	7	which accompanies this distribution. The full text of the license may be found at\r
	8	http://opensource.org/licenses/bsd-license.php\r
	9	\r
	10	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	11	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	12	\r
	13	**/\r
	14	\r
	15	#include "UefiShellLevel2CommandsLib.h"\r
	16	#include <Guid/FileSystemInfo.h>\r
	17	\r
	18	/**\r
	19	print out the list of files and directories from the LS command\r
	20	\r
	21	@param[in] Rec TRUE to automatically recurse into each found directory\r
	22	FALSE to only list the specified directory.\r
	23	@param[in] Attribs List of required Attribute for display.\r
	24	If 0 then all non-system and non-hidden files will be printed.\r
	25	@param[in] Sfo TRUE to use Standard Format Output, FALSE otherwise\r
	26	@param[in] Path String with starting path.\r
	27	@param[in] First TRUE for the original and FALSE for any recursion spawned instances.\r
	28	@param[in] Count The count of bits enabled in Attribs.\r
	29	@param[in] TimeZone The current time zone offset.\r
	30	\r
	31	@retval SHELL_SUCCESS the printing was sucessful.\r
	32	**/\r
	33	SHELL_STATUS\r
	34	EFIAPI\r
	35	PrintLsOutput(\r
	36	IN CONST BOOLEAN Rec,\r
	37	IN CONST UINT64 Attribs,\r
	38	IN CONST BOOLEAN Sfo,\r
	39	IN CONST CHAR16 *Path,\r
	40	IN CONST BOOLEAN First,\r
	41	IN CONST UINTN Count,\r
	42	IN CONST INT16 TimeZone\r
	43	)\r
	44	{\r
	45	EFI_STATUS Status;\r
	46	EFI_SHELL_FILE_INFO *ListHead;\r
	47	EFI_SHELL_FILE_INFO *Node;\r
	48	SHELL_STATUS ShellStatus;\r
	49	UINT64 FileCount;\r
	50	UINT64 DirCount;\r
	51	UINT64 FileSize;\r
	52	CHAR16 *DirectoryName;\r
	53	UINTN LongestPath;\r
	54	EFI_FILE_SYSTEM_INFO *SysInfo;\r
	55	UINTN SysInfoSize;\r
	56	SHELL_FILE_HANDLE ShellFileHandle;\r
	57	CHAR16 *CorrectedPath;\r
	58	EFI_FILE_PROTOCOL *EfiFpHandle;\r
	59	\r
	60	FileCount = 0;\r
	61	DirCount = 0;\r
	62	FileSize = 0;\r
	63	ListHead = NULL;\r
	64	ShellStatus = SHELL_SUCCESS;\r
65	LongestPath = 0;\r
66	CorrectedPath = NULL;\r
67	\r
68	CorrectedPath = StrnCatGrow(&CorrectedPath, NULL, Path, 0);\r
69	ASSERT(CorrectedPath != NULL);\r
70	ShellCommandCleanPath(CorrectedPath);\r
71	\r
72	Status = ShellOpenFileMetaArg((CHAR16*)CorrectedPath, EFI_FILE_MODE_READ, &ListHead);\r
73	if (EFI_ERROR(Status)) {\r
74	return (SHELL_DEVICE_ERROR);\r
75	}\r
76	if (ListHead == NULL \|\| IsListEmpty(&ListHead->Link)) {\r
77	//\r
78	// On the first one only we expect to find something...\r
79	// do we find the . and .. directories otherwise?\r
80	//\r
81	if (First) {\r
82	return (SHELL_NOT_FOUND);\r
83	}\r
84	return (SHELL_SUCCESS);\r
85	}\r
86	\r
87	if (Sfo && First) {\r
88	//\r
89	// Get the first valid handle (directories)\r
90	//\r
91	for ( Node = (EFI_SHELL_FILE_INFO *)GetFirstNode(&ListHead->Link)\r
92	; !IsNull(&ListHead->Link, &Node->Link) && Node->Handle == NULL\r
93	; Node = (EFI_SHELL_FILE_INFO *)GetNextNode(&ListHead->Link, &Node->Link)\r
94	);\r
95	\r
96	if (Node->Handle == NULL) {\r
97	DirectoryName = GetFullyQualifiedPath(((EFI_SHELL_FILE_INFO *)GetFirstNode(&ListHead->Link))->FullName);\r
98	\r
99	//\r
100	// We need to open something up to get system information\r
101	//\r
102	Status = gEfiShellProtocol->OpenFileByName(\r
103	DirectoryName,\r
104	&ShellFileHandle,\r
105	EFI_FILE_MODE_READ);\r
106	\r
107	ASSERT_EFI_ERROR(Status);\r
108	FreePool(DirectoryName);\r
109	\r
110	//\r
111	// Get the Volume Info from ShellFileHandle\r
112	//\r
113	SysInfo = NULL;\r
114	SysInfoSize = 0;\r
115	EfiFpHandle = ConvertShellHandleToEfiFileProtocol(ShellFileHandle);\r
116	Status = EfiFpHandle->GetInfo(\r
117	EfiFpHandle,\r
118	&gEfiFileSystemInfoGuid,\r
119	&SysInfoSize,\r
120	SysInfo);\r
121	\r
122	if (Status == EFI_BUFFER_TOO_SMALL) {\r
123	SysInfo = AllocateZeroPool(SysInfoSize);\r
124	Status = EfiFpHandle->GetInfo(\r
125	EfiFpHandle,\r
126	&gEfiFileSystemInfoGuid,\r
127	&SysInfoSize,\r
128	SysInfo);\r
129	}\r
130	\r
131	ASSERT_EFI_ERROR(Status);\r
132	\r
133	gEfiShellProtocol->CloseFile(ShellFileHandle);\r
134	} else {\r
135	//\r
136	// Get the Volume Info from Node->Handle\r
137	//\r
138	SysInfo = NULL;\r
139	SysInfoSize = 0;\r
140	EfiFpHandle = ConvertShellHandleToEfiFileProtocol(Node->Handle);\r
141	Status = EfiFpHandle->GetInfo(\r
142	EfiFpHandle,\r
143	&gEfiFileSystemInfoGuid,\r
144	&SysInfoSize,\r
145	SysInfo);\r
146	\r
147	if (Status == EFI_BUFFER_TOO_SMALL) {\r
148	SysInfo = AllocateZeroPool(SysInfoSize);\r
149	Status = EfiFpHandle->GetInfo(\r
150	EfiFpHandle,\r
151	&gEfiFileSystemInfoGuid,\r
152	&SysInfoSize,\r
153	SysInfo);\r
154	}\r
155	\r
156	ASSERT_EFI_ERROR(Status);\r
157	}\r
158	\r
159	ShellPrintHiiEx (\r
160	-1,\r
161	-1,\r
162	NULL,\r
163	STRING_TOKEN (STR_GEN_SFO_HEADER),\r
164	gShellLevel2HiiHandle,\r
165	L"ls");\r
166	//\r
167	// print VolumeInfo table\r
168	//\r
169	ASSERT(SysInfo != NULL);\r
170	ShellPrintHiiEx (\r
171	0,\r
172	gST->ConOut->Mode->CursorRow,\r
173	NULL,\r
174	STRING_TOKEN (STR_LS_SFO_VOLINFO),\r
175	gShellLevel2HiiHandle,\r
176	SysInfo->VolumeLabel,\r
177	SysInfo->VolumeSize,\r
178	SysInfo->ReadOnly?L"TRUE":L"FALSE",\r
179	SysInfo->FreeSpace,\r
180	SysInfo->BlockSize\r
181	);\r
182	if (SysInfo != NULL) {\r
183	FreePool(SysInfo);\r
184	}\r
185	}\r
186	\r
187	if (!Sfo) {\r
188	//\r
189	// get directory name from path...\r
190	//\r
191	DirectoryName = GetFullyQualifiedPath(CorrectedPath);\r
192	\r
193	//\r
194	// print header\r
195	//\r
196	ShellPrintHiiEx (\r
197	0,\r
198	gST->ConOut->Mode->CursorRow,\r
199	NULL,\r
200	STRING_TOKEN (STR_LS_HEADER_LINE1),\r
201	gShellLevel2HiiHandle,\r
202	DirectoryName\r
203	);\r
204	FreePool(DirectoryName);\r
205	}\r
206	for ( Node = (EFI_SHELL_FILE_INFO *)GetFirstNode(&ListHead->Link)\r
207	; !IsNull(&ListHead->Link, &Node->Link)\r
208	; Node = (EFI_SHELL_FILE_INFO *)GetNextNode(&ListHead->Link, &Node->Link)\r
209	){\r
210	ASSERT(Node != NULL);\r
211	if (LongestPath < StrSize(Node->FullName)) {\r
212	LongestPath = StrSize(Node->FullName);\r
213	}\r
214	ASSERT(Node->Info != NULL);\r
215	ASSERT((Node->Info->Attribute & EFI_FILE_VALID_ATTR) == Node->Info->Attribute);\r
216	if (Attribs == 0) {\r
217	//\r
218	// NOT system & NOT hidden\r
219	//\r
220	if ( (Node->Info->Attribute & EFI_FILE_SYSTEM)\r
221	\|\| (Node->Info->Attribute & EFI_FILE_HIDDEN)\r
222	){\r
223	continue;\r
224	}\r
225	} else if (Attribs != EFI_FILE_VALID_ATTR) {\r
226	if (Count == 1) {\r
227	//\r
228	// the bit must match\r
229	//\r
230	if ( (Node->Info->Attribute & Attribs) != Attribs) {\r
231	continue;\r
232	}\r
233	} else {\r
234	//\r
235	// exact match on all bits\r
236	//\r
237	if ( Node->Info->Attribute != Attribs) {\r
238	continue;\r
239	}\r
240	}\r
241	}\r
242	\r
243	if (Sfo) {\r
244	//\r
245	// Print the FileInfo Table\r
246	//\r
247	ShellPrintHiiEx (\r
248	0,\r
249	gST->ConOut->Mode->CursorRow,\r
250	NULL,\r
251	STRING_TOKEN (STR_LS_SFO_FILEINFO),\r
252	gShellLevel2HiiHandle,\r
253	Node->FullName,\r
254	Node->Info->FileSize,\r
255	Node->Info->PhysicalSize,\r
256	(Node->Info->Attribute & EFI_FILE_ARCHIVE) != 0?L"a":L"",\r
257	(Node->Info->Attribute & EFI_FILE_DIRECTORY) != 0?L"d":L"",\r
258	(Node->Info->Attribute & EFI_FILE_HIDDEN) != 0?L"h":L"",\r
259	(Node->Info->Attribute & EFI_FILE_READ_ONLY) != 0?L"r":L"",\r
260	(Node->Info->Attribute & EFI_FILE_SYSTEM) != 0?L"s":L"",\r
261	Node->Info->CreateTime.Hour,\r
262	Node->Info->CreateTime.Minute,\r
263	Node->Info->CreateTime.Second,\r
264	Node->Info->CreateTime.Day,\r
265	Node->Info->CreateTime.Month,\r
266	Node->Info->CreateTime.Year,\r
267	Node->Info->LastAccessTime.Hour,\r
268	Node->Info->LastAccessTime.Minute,\r
269	Node->Info->LastAccessTime.Second,\r
270	Node->Info->LastAccessTime.Day,\r
271	Node->Info->LastAccessTime.Month,\r
272	Node->Info->LastAccessTime.Year,\r
273	Node->Info->ModificationTime.Hour,\r
274	Node->Info->ModificationTime.Minute,\r
275	Node->Info->ModificationTime.Second,\r
276	Node->Info->ModificationTime.Day,\r
277	Node->Info->ModificationTime.Month,\r
278	Node->Info->ModificationTime.Year\r
279	);\r
280	} else {\r
281	//\r
282	// print this one out...\r
283	// first print the universal start, next print the type specific name format, last print the CRLF\r
284	//\r
285	ShellPrintHiiEx (\r
286	-1,\r
287	-1,\r
288	NULL,\r
289	STRING_TOKEN (STR_LS_LINE_START_ALL),\r
290	gShellLevel2HiiHandle,\r
291	&Node->Info->ModificationTime,\r
292	(Node->Info->Attribute & EFI_FILE_DIRECTORY) != 0?L"<DIR>":L"",\r
293	(Node->Info->Attribute & EFI_FILE_READ_ONLY) != 0?L'r':L' ',\r
294	Node->Info->FileSize\r
295	);\r
296	if (Node->Info->Attribute & EFI_FILE_DIRECTORY) {\r
297	DirCount++;\r
298	ShellPrintHiiEx (\r
299	-1,\r
300	-1,\r
301	NULL,\r
302	STRING_TOKEN (STR_LS_LINE_END_DIR),\r
303	gShellLevel2HiiHandle,\r
304	Node->FileName\r
305	);\r
306	} else {\r
307	FileCount++;\r
308	FileSize += Node->Info->FileSize;\r
309	if ( (gUnicodeCollation->StriColl(gUnicodeCollation, (CHAR16)L".nsh", (CHAR16)&(Node->FileName[StrLen (Node->FileName) - 4])) == 0)\r
310	\|\| (gUnicodeCollation->StriColl(gUnicodeCollation, (CHAR16)L".efi", (CHAR16)&(Node->FileName[StrLen (Node->FileName) - 4])) == 0)\r
311	){\r
312	ShellPrintHiiEx (\r
313	-1,\r
314	-1,\r
315	NULL,\r
316	STRING_TOKEN (STR_LS_LINE_END_EXE),\r
317	gShellLevel2HiiHandle,\r
318	Node->FileName\r
319	);\r
320	} else {\r
321	ShellPrintHiiEx (\r
322	-1,\r
323	-1,\r
324	NULL,\r
325	STRING_TOKEN (STR_LS_LINE_END_FILE),\r
326	gShellLevel2HiiHandle,\r
327	Node->FileName\r
328	);\r
329	}\r
330	}\r
331	}\r
332	}\r
333	\r
334	if (!Sfo) {\r
335	//\r
336	// print footer\r
337	//\r
338	ShellPrintHiiEx (\r
339	-1,\r
340	-1,\r
341	NULL,\r
342	STRING_TOKEN (STR_LS_FOOTER_LINE),\r
343	gShellLevel2HiiHandle,\r
344	FileCount,\r
345	FileSize,\r
346	DirCount\r
347	);\r
348	}\r
349	\r
350	if (Rec){\r
351	DirectoryName = AllocatePool(LongestPath + 2*sizeof(CHAR16));\r
352	for ( Node = (EFI_SHELL_FILE_INFO *)GetFirstNode(&ListHead->Link)\r
353	; !IsNull(&ListHead->Link, &Node->Link)\r
354	; Node = (EFI_SHELL_FILE_INFO *)GetNextNode(&ListHead->Link, &Node->Link)\r
355	){\r
356	//\r
357	// recurse on any directory except the traversing ones...\r
358	//\r
359	if (((Node->Info->Attribute & EFI_FILE_DIRECTORY) == EFI_FILE_DIRECTORY)\r
360	&& StrCmp(Node->FileName, L".") != 0\r
361	&& StrCmp(Node->FileName, L"..") != 0\r
362	){\r
363	StrCpy(DirectoryName, Node->FullName);\r
364	StrCat(DirectoryName, L"\\*");\r
365	PrintLsOutput(\r
366	Rec,\r
367	Attribs,\r
368	Sfo,\r
369	DirectoryName,\r
370	FALSE,\r
371	Count,\r
372	TimeZone);\r
373	}\r
374	}\r
375	FreePool(DirectoryName);\r
376	}\r
377	\r
378	FreePool(CorrectedPath);\r
379	ShellCloseFileMetaArg(&ListHead);\r
380	FreePool(ListHead);\r
381	return (ShellStatus);\r
382	}\r
383	\r
384	STATIC CONST SHELL_PARAM_ITEM LsParamList[] = {\r
385	{L"-r", TypeFlag},\r
386	{L"-a", TypeStart},\r
387	{L"-sfo", TypeFlag},\r
388	{NULL, TypeMax}\r
389	};\r
390	\r
391	/**\r
392	Function for 'ls' command.\r
393	\r
394	@param[in] ImageHandle Handle to the Image (NULL if Internal).\r
395	@param[in] SystemTable Pointer to the System Table (NULL if Internal).\r
396	**/\r
397	SHELL_STATUS\r
398	EFIAPI\r
399	ShellCommandRunLs (\r
400	IN EFI_HANDLE ImageHandle,\r
401	IN EFI_SYSTEM_TABLE *SystemTable\r
402	)\r
403	{\r
404	EFI_STATUS Status;\r
405	LIST_ENTRY *Package;\r
406	CHAR16 *ProblemParam;\r
407	CONST CHAR16 *Attribs;\r
408	SHELL_STATUS ShellStatus;\r
409	UINT64 RequiredAttributes;\r
410	CONST CHAR16 *PathName;\r
411	CONST CHAR16 *CurDir;\r
412	UINTN Count;\r
413	CHAR16 *FullPath;\r
414	UINTN Size;\r
415	EFI_TIME theTime;\r
416	BOOLEAN SfoMode;\r
417	\r
418	Size = 0;\r
419	FullPath = NULL;\r
420	ProblemParam = NULL;\r
421	Attribs = NULL;\r
422	ShellStatus = SHELL_SUCCESS;\r
423	RequiredAttributes = 0;\r
424	PathName = NULL;\r
425	CurDir = NULL;\r
426	Count = 0;\r
427	\r
428	//\r
429	// initialize the shell lib (we must be in non-auto-init...)\r
430	//\r
431	Status = ShellInitialize();\r
432	ASSERT_EFI_ERROR(Status);\r
433	\r
434	//\r
435	// Fix local copies of the protocol pointers\r
436	//\r
437	Status = CommandInit();\r
438	ASSERT_EFI_ERROR(Status);\r
439	\r
440	//\r
441	// parse the command line\r
442	//\r
443	Status = ShellCommandLineParse (LsParamList, &Package, &ProblemParam, TRUE);\r
444	if (EFI_ERROR(Status)) {\r
445	if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {\r
446	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, ProblemParam);\r
447	FreePool(ProblemParam);\r
448	ShellStatus = SHELL_INVALID_PARAMETER;\r
449	} else {\r
450	ASSERT(FALSE);\r
451	}\r
452	} else {\r
453	//\r
454	// check for "-?"\r
455	//\r
456	if (ShellCommandLineGetFlag(Package, L"-?")) {\r
457	ASSERT(FALSE);\r
458	}\r
459	\r
460	if (ShellCommandLineGetCount(Package) > 2) {\r
461	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellLevel2HiiHandle);\r
462	ShellStatus = SHELL_INVALID_PARAMETER;\r
463	} else {\r
464	//\r
465	// check for -a\r
466	//\r
467	if (ShellCommandLineGetFlag(Package, L"-a")) {\r
468	for ( Attribs = ShellCommandLineGetValue(Package, L"-a")\r
469	; Attribs != NULL && *Attribs != CHAR_NULL && ShellStatus == SHELL_SUCCESS\r
470	; Attribs++\r
471	){\r
472	switch (*Attribs) {\r
473	case L'a':\r
474	case L'A':\r
475	RequiredAttributes \|= EFI_FILE_ARCHIVE;\r
476	Count++;\r
477	continue;\r
478	case L's':\r
479	case L'S':\r
480	RequiredAttributes \|= EFI_FILE_SYSTEM;\r
481	Count++;\r
482	continue;\r
483	case L'h':\r
484	case L'H':\r
485	RequiredAttributes \|= EFI_FILE_HIDDEN;\r
486	Count++;\r
487	continue;\r
488	case L'r':\r
489	case L'R':\r
490	RequiredAttributes \|= EFI_FILE_READ_ONLY;\r
491	Count++;\r
492	continue;\r
493	case L'd':\r
494	case L'D':\r
495	RequiredAttributes \|= EFI_FILE_DIRECTORY;\r
496	Count++;\r
497	continue;\r
498	default:\r
499	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_ATTRIBUTE), gShellLevel2HiiHandle, ShellCommandLineGetValue(Package, L"-a"));\r
500	ShellStatus = SHELL_INVALID_PARAMETER;\r
501	break;\r
502	} // switch\r
503	} // for loop\r
504	//\r
505	// if nothing is specified all are specified\r
506	//\r
507	if (RequiredAttributes == 0) {\r
508	RequiredAttributes = EFI_FILE_VALID_ATTR;\r
509	}\r
510	} // if -a present\r
511	if (ShellStatus == SHELL_SUCCESS) {\r
512	PathName = ShellCommandLineGetRawValue(Package, 1);\r
513	if (PathName == NULL) {\r
514	CurDir = gEfiShellProtocol->GetCurDir(NULL);\r
515	if (CurDir == NULL) {\r
516	ShellStatus = SHELL_NOT_FOUND;\r
517	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_CWD), gShellLevel2HiiHandle);\r
518	}\r
519	}\r
520	if (PathName != NULL) {\r
521	ASSERT((FullPath == NULL && Size == 0) \|\| (FullPath != NULL));\r
522	StrnCatGrow(&FullPath, &Size, PathName, 0);\r
523	if (ShellIsDirectory(PathName) == EFI_SUCCESS) {\r
524	StrnCatGrow(&FullPath, &Size, L"\\*", 0);\r
525	}\r
526	} else {\r
527	ASSERT(FullPath == NULL);\r
528	StrnCatGrow(&FullPath, NULL, L"*", 0);\r
529	}\r
530	Status = gRT->GetTime(&theTime, NULL);\r
531	ASSERT_EFI_ERROR(Status);\r
532	SfoMode = ShellCommandLineGetFlag(Package, L"-sfo");\r
533	if (ShellStatus == SHELL_SUCCESS) {\r
534	ShellStatus = PrintLsOutput(\r
535	ShellCommandLineGetFlag(Package, L"-r"),\r
536	RequiredAttributes,\r
537	SfoMode,\r
538	FullPath,\r
539	TRUE,\r
540	Count,\r
541	(INT16)(theTime.TimeZone==2047?0:theTime.TimeZone)\r
542	);\r
543	if (ShellStatus == SHELL_NOT_FOUND) {\r
544	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_NO_FILES), gShellLevel2HiiHandle);\r
545	} else if (ShellStatus == SHELL_INVALID_PARAMETER) {\r
546	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellLevel2HiiHandle);\r
547	} else if (ShellStatus != SHELL_SUCCESS) {\r
548	ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PARAM_INV), gShellLevel2HiiHandle);\r
549	}\r
550	}\r
551	}\r
552	}\r
553	}\r
554	\r
555	if (FullPath != NULL) {\r
556	FreePool(FullPath);\r
557	}\r
558	//\r
559	// free the command line package\r
560	//\r
561	ShellCommandLineFreeVarList (Package);\r
562	\r
563	return (ShellStatus);\r
564	}\r