ShellPkg/Library/UefiShellLevel2CommandsLib/Parse.c

   1 /** @file
   2   Main file for Parse shell level 2 function.
   3
   4   Copyright (c) 2009 - 2010, 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
   9
  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.
  12
  13 **/
  14
  15 #include "UefiShellLevel2CommandsLib.h"
  16
  17 SHELL_STATUS
  18 EFIAPI
  19 PerformParsing(
  20   IN CONST CHAR16 *FileName,
  21   IN CONST CHAR16 *TableName,
  22   IN CONST UINTN  ColumnIndex,
  23   IN CONST UINTN  TableNameInstance,
  24   IN CONST UINTN  ShellCommandInstance
  25   )
  26 {
  27   SHELL_FILE_HANDLE FileHandle;
  28   EFI_STATUS        Status;
  29   BOOLEAN           Ascii;
  30   UINTN             LoopVariable;
  31   UINTN             ColumnLoop;
  32   CHAR16            *TempLine;
  33   CHAR16            *ColumnPointer;
  34   SHELL_STATUS      ShellStatus;
  35   CHAR16            *TempSpot;
  36
  37   ASSERT(FileName   != NULL);
  38   ASSERT(TableName  != NULL);
  39
  40   ShellStatus       = SHELL_SUCCESS;
  41
  42   Status = ShellOpenFileByName(FileName, &FileHandle, EFI_FILE_MODE_READ, 0);
  43   if (EFI_ERROR(Status)) {
  44     ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_FILE_OPEN_FAIL), gShellLevel2HiiHandle, FileName);
  45     ShellStatus = SHELL_NOT_FOUND;
  46   } else {
  47     for (LoopVariable = 0 ; LoopVariable < ShellCommandInstance && !ShellFileHandleEof(FileHandle);) {
  48       TempLine = ShellFileHandleReturnLine(FileHandle, &Ascii);
  49       if (TempLine == NULL) {
  50         break;
  51       }
  52       if (StrStr(TempLine, L"ShellCommand, \"") == TempLine) {
  53         LoopVariable++;
  54       }
  55       SHELL_FREE_NON_NULL(TempLine);
  56     }
  57     if (LoopVariable == ShellCommandInstance) {
  58       LoopVariable = 0;
  59       while(1) {
  60         TempLine = ShellFileHandleReturnLine(FileHandle, &Ascii);
  61         if ( TempLine == NULL
  62           || *TempLine == CHAR_NULL
  63           || StrStr(TempLine, L"ShellCommand, \"") == TempLine
  64          ){
  65           SHELL_FREE_NON_NULL(TempLine);
  66           break;
  67         }
  68         if (StrStr(TempLine, TableName) == TempLine) {
  69           LoopVariable++;
  70         }
  71         if ( LoopVariable == TableNameInstance
  72           || (TableNameInstance == (UINTN)-1 && StrStr(TempLine, TableName) == TempLine)
  73          ){
  74           for (ColumnLoop = 1, ColumnPointer = TempLine; ColumnLoop < ColumnIndex && ColumnPointer != NULL && *ColumnPointer != CHAR_NULL; ColumnLoop++) {
  75             ColumnPointer = StrStr(ColumnPointer, L",");
  76             if (ColumnPointer != NULL && *ColumnPointer != CHAR_NULL){
  77               ColumnPointer++;
  78             }
  79           }
  80           if (ColumnLoop == ColumnIndex) {
  81             ASSERT(ColumnPointer != NULL);
  82             TempSpot = StrStr(ColumnPointer, L",");
  83             if (TempSpot != NULL) {
  84               *TempSpot = CHAR_NULL;
  85             }
  86             while (ColumnPointer != NULL && *ColumnPointer != CHAR_NULL && ColumnPointer[0] == L' '){
  87               ColumnPointer++;
  88             }
  89             if (ColumnPointer != NULL && *ColumnPointer != CHAR_NULL && ColumnPointer[0] == L'\"'){
  90               ColumnPointer++;
  91             }
  92             if (ColumnPointer != NULL && *ColumnPointer != CHAR_NULL && ColumnPointer[StrLen(ColumnPointer)-1] == L'\"'){
  93               ColumnPointer[StrLen(ColumnPointer)-1] = CHAR_NULL;
  94             }
  95
  96             ShellPrintEx(-1, -1, L"%s\r\n", ColumnPointer);
  97           }
  98         }
  99         SHELL_FREE_NON_NULL(TempLine);
 100       }
 101     }
 102   }
 103   return (ShellStatus);
 104 }
 105
 106 STATIC CONST SHELL_PARAM_ITEM ParamList[] = {
 107   {L"-i", TypeValue},
 108   {L"-s", TypeValue},
 109   {NULL, TypeMax}
 110   };
 111
 112 /**
 113   Function for 'parse' command.
 114
 115   @param[in] ImageHandle  Handle to the Image (NULL if Internal).
 116   @param[in] SystemTable  Pointer to the System Table (NULL if Internal).
 117 **/
 118 SHELL_STATUS
 119 EFIAPI
 120 ShellCommandRunParse (
 121   IN EFI_HANDLE        ImageHandle,
 122   IN EFI_SYSTEM_TABLE  *SystemTable
 123   )
 124 {
 125   EFI_STATUS          Status;
 126   LIST_ENTRY          *Package;
 127   CHAR16              *ProblemParam;
 128   CONST CHAR16        *FileName;
 129   CONST CHAR16        *TableName;
 130   CONST CHAR16        *ColumnString;
 131   SHELL_STATUS        ShellStatus;
 132   UINTN               ShellCommandInstance;
 133   UINTN               TableNameInstance;
 134
 135   ShellStatus   = SHELL_SUCCESS;
 136   ProblemParam  = NULL;
 137
 138   //
 139   // initialize the shell lib (we must be in non-auto-init...)
 140   //
 141   Status = ShellInitialize();
 142   ASSERT_EFI_ERROR(Status);
 143
 144   //
 145   // parse the command line
 146   //
 147   Status = ShellCommandLineParse (ParamList, &Package, &ProblemParam, TRUE);
 148   if (EFI_ERROR(Status)) {
 149     if (Status == EFI_VOLUME_CORRUPTED && ProblemParam != NULL) {
 150       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_PROBLEM), gShellLevel2HiiHandle, ProblemParam);
 151       FreePool(ProblemParam);
 152       ShellStatus = SHELL_INVALID_PARAMETER;
 153     } else {
 154       ASSERT(FALSE);
 155     }
 156   } else {
 157     if (ShellCommandLineGetCount(Package) < 4) {
 158       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_FEW), gShellLevel2HiiHandle);
 159       ShellStatus = SHELL_INVALID_PARAMETER;
 160     } else if (ShellCommandLineGetCount(Package) > 4) {
 161       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_GEN_TOO_MANY), gShellLevel2HiiHandle);
 162       ShellStatus = SHELL_INVALID_PARAMETER;
 163     } else {
 164       FileName      = ShellCommandLineGetRawValue(Package, 1);
 165       TableName     = ShellCommandLineGetRawValue(Package, 2);
 166       ColumnString  = ShellCommandLineGetRawValue(Package, 3);
 167
 168       if (ShellCommandLineGetValue(Package, L"-i") == NULL) {
 169         TableNameInstance = (UINTN)-1;
 170       } else {
 171         TableNameInstance = ShellStrToUintn(ShellCommandLineGetValue(Package, L"-i"));
 172       }
 173       if (ShellCommandLineGetValue(Package, L"-s") == NULL) {
 174         ShellCommandInstance = 1;
 175       } else {
 176         ShellCommandInstance = ShellStrToUintn(ShellCommandLineGetValue(Package, L"-s"));
 177       }
 178
 179       ShellStatus = PerformParsing(FileName, TableName, ShellStrToUintn(ColumnString), TableNameInstance, ShellCommandInstance);
 180     }
 181   }
 182
 183   //
 184   // free the command line package
 185   //
 186   ShellCommandLineFreeVarList (Package);
 187
 188   return (ShellStatus);
 189 }
 190