]> git.proxmox.com Git - mirror_edk2.git/blobdiff - ShellPkg/Library/UefiShellLevel1CommandsLib/If.c
Updating level 1 commands.
[mirror_edk2.git] / ShellPkg / Library / UefiShellLevel1CommandsLib / If.c
index f3fc6b06a83d7d300aab8d31b4de31252fe6f82a..59069e568f42d09565d005f73c85c046977bc595 100644 (file)
@@ -1,7 +1,7 @@
 /** @file\r
   Main file for If and else shell level 1 function.\r
 \r
-  Copyright (c) 2009 - 2010, Intel Corporation. All rights reserved.<BR>\r
+  Copyright (c) 2009 - 2011, 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
 #include "UefiShellLevel1CommandsLib.h"\r
 #include <Library/PrintLib.h>\r
 \r
-\r
-\r
-\r
 typedef enum {\r
-  END_TAG_OR,\r
-  END_TAG_AND,\r
-  END_TAG_THEN,\r
-  END_TAG_MAX\r
+  EndTagOr,\r
+  EndTagAnd,\r
+  EndTagThen,\r
+  EndTagMax\r
 } END_TAG_TYPE;\r
 \r
 typedef enum {\r
-  OPERATOR_GT,\r
-  OPERATOR_LT,\r
-  OPERATOR_EQ,\r
-  OPERATOR_NE,\r
-  OPERATOR_GE,\r
-  OPERATOR_LE,\r
-  OPERATOR_UGT,\r
-  OPERATOR_ULT,\r
-  OPERATOR_UGE,\r
-  OPERATOR_ULE,\r
-  OPERATOR_MAX\r
+  OperatorGreaterThan,\r
+  OperatorLessThan,\r
+  OperatorEqual,\r
+  OperatorNotEqual,\r
+  OperatorGreatorOrEqual,\r
+  OperatorLessOrEqual,\r
+  OperatorUnisgnedGreaterThan,\r
+  OperatorUnsignedLessThan,\r
+  OperatorUnsignedGreaterOrEqual,\r
+  OperatorUnsignedLessOrEqual,\r
+  OperatorMax\r
 } BIN_OPERATOR_TYPE;\r
 \r
+/**\r
+  Extract the next fragment, if there is one.\r
+\r
+  @param[in,out] Statement    The current remaining statement.\r
+  @param[in] Fragment         The current fragment.\r
+\r
+  @retval FALSE   There is not another fragment.\r
+  @retval TRUE    There is another fragment.\r
+**/\r
 BOOLEAN\r
 EFIAPI\r
 IsNextFragment (\r
-  IN CONST CHAR16         **Statement,\r
+  IN OUT CONST CHAR16     **Statement,\r
   IN CONST CHAR16         *Fragment\r
   )\r
 {\r
@@ -71,6 +77,14 @@ IsNextFragment (
   return (FALSE);\r
 }\r
 \r
+/**\r
+  Determine if String represents a valid profile.\r
+\r
+  @param[in] String     The pointer to the string to test.\r
+\r
+  @retval TRUE    String is a valid profile.\r
+  @retval FALSE   String is not a valid profile.\r
+**/\r
 BOOLEAN\r
 EFIAPI\r
 IsValidProfile (\r
@@ -88,6 +102,17 @@ IsValidProfile (
   return (FALSE);\r
 }\r
 \r
+/**\r
+  Do a comparison between 2 things.\r
+\r
+  @param[in] Compare1           The first item to compare.\r
+  @param[in] Compare2           The second item to compare.\r
+  @param[in] BinOp              The type of comparison to perform.\r
+  @param[in] CaseInsensitive    TRUE to do non-case comparison, FALSE otherwise.\r
+  @param[in] ForceStringCompare TRUE to force string comparison, FALSE otherwise.\r
+\r
+  @return     The result of the comparison.\r
+**/\r
 BOOLEAN\r
 EFIAPI\r
 TestOperation (\r
@@ -105,8 +130,8 @@ TestOperation (
   // "Compare1 BinOp Compare2"\r
   //\r
   switch (BinOp) {\r
-  case OPERATOR_UGT:\r
-  case OPERATOR_GT:\r
+  case OperatorUnisgnedGreaterThan:\r
+  case OperatorGreaterThan:\r
     if (ForceStringCompare || !ShellIsHexOrDecimalNumber(Compare1, FALSE, FALSE) || !ShellIsHexOrDecimalNumber(Compare2, FALSE, FALSE)) {\r
       //\r
       // string compare\r
@@ -119,16 +144,16 @@ TestOperation (
       // numeric compare\r
       //\r
       if (Compare1[0] == L'-') {\r
-        Cmp1 = 0 - (INTN)StrDecimalToUintn(Compare1+1);\r
+        Cmp1 = 0 - (INTN)ShellStrToUintn(Compare1+1);\r
       } else {\r
-        Cmp1 = (INTN)StrDecimalToUintn(Compare1);\r
+        Cmp1 = (INTN)ShellStrToUintn(Compare1);\r
       }\r
       if (Compare2[0] == L'-') {\r
-        Cmp2 = 0 - (INTN)StrDecimalToUintn(Compare2+1);\r
+        Cmp2 = 0 - (INTN)ShellStrToUintn(Compare2+1);\r
       } else {\r
-        Cmp2 = (INTN)StrDecimalToUintn(Compare2);\r
+        Cmp2 = (INTN)ShellStrToUintn(Compare2);\r
       }\r
-      if (BinOp == OPERATOR_GT) {\r
+      if (BinOp == OperatorGreaterThan) {\r
         if (Cmp1 > Cmp2) {\r
           return (TRUE);\r
         }\r
@@ -140,8 +165,8 @@ TestOperation (
     }\r
     return (FALSE);\r
     break;\r
-  case OPERATOR_ULT:\r
-  case OPERATOR_LT:\r
+  case OperatorUnsignedLessThan:\r
+  case OperatorLessThan:\r
     if (ForceStringCompare || !ShellIsHexOrDecimalNumber(Compare1, FALSE, FALSE) || !ShellIsHexOrDecimalNumber(Compare2, FALSE, FALSE)) {\r
       //\r
       // string compare\r
@@ -154,16 +179,16 @@ TestOperation (
       // numeric compare\r
       //\r
       if (Compare1[0] == L'-') {\r
-        Cmp1 = 0 - (INTN)StrDecimalToUintn(Compare1+1);\r
+        Cmp1 = 0 - (INTN)ShellStrToUintn(Compare1+1);\r
       } else {\r
-        Cmp1 = (INTN)StrDecimalToUintn(Compare1);\r
+        Cmp1 = (INTN)ShellStrToUintn(Compare1);\r
       }\r
       if (Compare2[0] == L'-') {\r
-        Cmp2 = 0 - (INTN)StrDecimalToUintn(Compare2+1);\r
+        Cmp2 = 0 - (INTN)ShellStrToUintn(Compare2+1);\r
       } else {\r
-        Cmp2 = (INTN)StrDecimalToUintn(Compare2);\r
+        Cmp2 = (INTN)ShellStrToUintn(Compare2);\r
       }\r
-      if (BinOp == OPERATOR_LT) {\r
+      if (BinOp == OperatorLessThan) {\r
         if (Cmp1 < Cmp2) {\r
           return (TRUE);\r
         }\r
@@ -176,7 +201,7 @@ TestOperation (
     }\r
     return (FALSE);\r
     break;\r
-  case OPERATOR_EQ:\r
+  case OperatorEqual:\r
     if (ForceStringCompare || !ShellIsHexOrDecimalNumber(Compare1, FALSE, FALSE) || !ShellIsHexOrDecimalNumber(Compare2, FALSE, FALSE)) {\r
       //\r
       // string compare\r
@@ -204,7 +229,7 @@ TestOperation (
     }\r
     return (FALSE);\r
     break;\r
-  case OPERATOR_NE:\r
+  case OperatorNotEqual:\r
     if (ForceStringCompare || !ShellIsHexOrDecimalNumber(Compare1, FALSE, FALSE) || !ShellIsHexOrDecimalNumber(Compare2, FALSE, FALSE)) {\r
       //\r
       // string compare\r
@@ -217,14 +242,14 @@ TestOperation (
       // numeric compare\r
       //\r
       if (Compare1[0] == L'-') {\r
-        Cmp1 = 0 - (INTN)StrDecimalToUintn(Compare1+1);\r
+        Cmp1 = 0 - (INTN)ShellStrToUintn(Compare1+1);\r
       } else {\r
-        Cmp1 = (INTN)StrDecimalToUintn(Compare1);\r
+        Cmp1 = (INTN)ShellStrToUintn(Compare1);\r
       }\r
       if (Compare2[0] == L'-') {\r
-        Cmp2 = 0 - (INTN)StrDecimalToUintn(Compare2+1);\r
+        Cmp2 = 0 - (INTN)ShellStrToUintn(Compare2+1);\r
       } else {\r
-        Cmp2 = (INTN)StrDecimalToUintn(Compare2);\r
+        Cmp2 = (INTN)ShellStrToUintn(Compare2);\r
       }\r
       if (Cmp1 != Cmp2) {\r
         return (TRUE);\r
@@ -232,8 +257,8 @@ TestOperation (
     }\r
     return (FALSE);\r
     break;\r
-  case OPERATOR_UGE:\r
-  case OPERATOR_GE:\r
+  case OperatorUnsignedGreaterOrEqual:\r
+  case OperatorGreatorOrEqual:\r
     if (ForceStringCompare || !ShellIsHexOrDecimalNumber(Compare1, FALSE, FALSE) || !ShellIsHexOrDecimalNumber(Compare2, FALSE, FALSE)) {\r
       //\r
       // string compare\r
@@ -246,16 +271,16 @@ TestOperation (
       // numeric compare\r
       //\r
       if (Compare1[0] == L'-') {\r
-        Cmp1 = 0 - (INTN)StrDecimalToUintn(Compare1+1);\r
+        Cmp1 = 0 - (INTN)ShellStrToUintn(Compare1+1);\r
       } else {\r
-        Cmp1 = (INTN)StrDecimalToUintn(Compare1);\r
+        Cmp1 = (INTN)ShellStrToUintn(Compare1);\r
       }\r
       if (Compare2[0] == L'-') {\r
-        Cmp2 = 0 - (INTN)StrDecimalToUintn(Compare2+1);\r
+        Cmp2 = 0 - (INTN)ShellStrToUintn(Compare2+1);\r
       } else {\r
-        Cmp2 = (INTN)StrDecimalToUintn(Compare2);\r
+        Cmp2 = (INTN)ShellStrToUintn(Compare2);\r
       }\r
-      if (BinOp == OPERATOR_GE) {\r
+      if (BinOp == OperatorGreatorOrEqual) {\r
         if (Cmp1 >= Cmp2) {\r
           return (TRUE);\r
         }\r
@@ -267,8 +292,8 @@ TestOperation (
     }\r
     return (FALSE);\r
     break;\r
-  case OPERATOR_LE:\r
-  case OPERATOR_ULE:\r
+  case OperatorLessOrEqual:\r
+  case OperatorUnsignedLessOrEqual:\r
     if (ForceStringCompare || !ShellIsHexOrDecimalNumber(Compare1, FALSE, FALSE) || !ShellIsHexOrDecimalNumber(Compare2, FALSE, FALSE)) {\r
       //\r
       // string compare\r
@@ -281,16 +306,16 @@ TestOperation (
       // numeric compare\r
       //\r
       if (Compare1[0] == L'-') {\r
-        Cmp1 = 0 - (INTN)StrDecimalToUintn(Compare1+1);\r
+        Cmp1 = 0 - (INTN)ShellStrToUintn(Compare1+1);\r
       } else {\r
-        Cmp1 = (INTN)StrDecimalToUintn(Compare1);\r
+        Cmp1 = (INTN)ShellStrToUintn(Compare1);\r
       }\r
       if (Compare2[0] == L'-') {\r
-        Cmp2 = 0 - (INTN)StrDecimalToUintn(Compare2+1);\r
+        Cmp2 = 0 - (INTN)ShellStrToUintn(Compare2+1);\r
       } else {\r
-        Cmp2 = (INTN)StrDecimalToUintn(Compare2);\r
+        Cmp2 = (INTN)ShellStrToUintn(Compare2);\r
       }\r
-      if (BinOp == OPERATOR_LE) {\r
+      if (BinOp == OperatorLessOrEqual) {\r
         if (Cmp1 <= Cmp2) {\r
           return (TRUE);\r
         }\r
@@ -308,6 +333,22 @@ TestOperation (
   }\r
 }\r
 \r
+/**\r
+  Process an if statement and determine if its is valid or not.\r
+\r
+  @param[in,out] PassingState     Opon entry, the current state.  Upon exit, \r
+                                  the new state.\r
+  @param[in] StartParameterNumber The number of the first parameter of\r
+                                  this statement.\r
+  @param[in] EndParameterNumber   The number of the final parameter of\r
+                                  this statement.\r
+  @param[in] OperatorToUse        The type of termination operator.\r
+  @param[in] CaseInsensitive      TRUE for case insensitive, FALSE otherwise.\r
+  @param[in] ForceStringCompare   TRUE for all string based, FALSE otherwise.\r
+\r
+  @retval EFI_INVALID_PARAMETER   A parameter was invalid.\r
+  @retval EFI_SUCCESS             The operation was successful.                                  \r
+**/\r
 EFI_STATUS\r
 EFIAPI\r
 ProcessStatement (\r
@@ -329,10 +370,10 @@ ProcessStatement (
   CHAR16                  HexString[20];\r
   CHAR16                  *TempSpot;\r
 \r
-  ASSERT((END_TAG_TYPE)OperatorToUse != END_TAG_THEN);\r
+  ASSERT((END_TAG_TYPE)OperatorToUse != EndTagThen);\r
 \r
   Status          = EFI_SUCCESS;\r
-  BinOp           = OPERATOR_MAX;\r
+  BinOp           = OperatorMax;\r
   OperationResult = FALSE;\r
   StatementWalker = gEfiShellParametersProtocol->Argv[StartParameterNumber];\r
   if (IsNextFragment((CONST CHAR16**)(&StatementWalker), L"not")) {\r
@@ -397,7 +438,7 @@ ProcessStatement (
     //\r
     Compare1 = NULL;\r
     Compare2 = NULL;\r
-    BinOp    = OPERATOR_MAX;\r
+    BinOp    = OperatorMax;\r
 \r
     //\r
     // get the first item\r
@@ -473,27 +514,27 @@ ProcessStatement (
     ASSERT(StartParameterNumber+1<EndParameterNumber);\r
     StatementWalker = gEfiShellParametersProtocol->Argv[StartParameterNumber+1];\r
     if (IsNextFragment((CONST CHAR16**)(&StatementWalker), L"gt")) {\r
-      BinOp = OPERATOR_GT;\r
+      BinOp = OperatorGreaterThan;\r
     } else if (IsNextFragment((CONST CHAR16**)(&StatementWalker), L"lt")) {\r
-      BinOp = OPERATOR_LT;\r
+      BinOp = OperatorLessThan;\r
     } else if (IsNextFragment((CONST CHAR16**)(&StatementWalker), L"eq")) {\r
-      BinOp = OPERATOR_EQ;\r
+      BinOp = OperatorEqual;\r
     } else if (IsNextFragment((CONST CHAR16**)(&StatementWalker), L"ne")) {\r
-      BinOp = OPERATOR_NE;\r
+      BinOp = OperatorNotEqual;\r
     } else if (IsNextFragment((CONST CHAR16**)(&StatementWalker), L"ge")) {\r
-      BinOp = OPERATOR_GE;\r
+      BinOp = OperatorGreatorOrEqual;\r
     } else if (IsNextFragment((CONST CHAR16**)(&StatementWalker), L"le")) {\r
-      BinOp = OPERATOR_LE;\r
+      BinOp = OperatorLessOrEqual;\r
     } else if (IsNextFragment((CONST CHAR16**)(&StatementWalker), L"==")) {\r
-      BinOp = OPERATOR_EQ;\r
+      BinOp = OperatorEqual;\r
     } else if (IsNextFragment((CONST CHAR16**)(&StatementWalker), L"ugt")) {\r
-      BinOp = OPERATOR_UGT;\r
+      BinOp = OperatorUnisgnedGreaterThan;\r
     } else if (IsNextFragment((CONST CHAR16**)(&StatementWalker), L"ult")) {\r
-      BinOp = OPERATOR_ULT;\r
+      BinOp = OperatorUnsignedLessThan;\r
     } else if (IsNextFragment((CONST CHAR16**)(&StatementWalker), L"uge")) {\r
-      BinOp = OPERATOR_UGE;\r
+      BinOp = OperatorUnsignedGreaterOrEqual;\r
     } else if (IsNextFragment((CONST CHAR16**)(&StatementWalker), L"ule")) {\r
-      BinOp = OPERATOR_ULE;\r
+      BinOp = OperatorUnsignedLessOrEqual;\r
     } else {\r
       ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_INVALID_BINOP), gShellLevel1HiiHandle, StatementWalker);\r
       Status = EFI_INVALID_PARAMETER;\r
@@ -566,7 +607,7 @@ ProcessStatement (
       Compare2 = StrnCatGrow(&Compare2, NULL, StatementWalker, 0);\r
     }\r
 \r
-    if (Compare1 != NULL && Compare2 != NULL && BinOp != OPERATOR_MAX) {\r
+    if (Compare1 != NULL && Compare2 != NULL && BinOp != OperatorMax) {\r
       OperationResult = TestOperation(Compare1, Compare2, BinOp, CaseInsensitive, ForceStringCompare);\r
     }\r
 \r
@@ -583,13 +624,13 @@ ProcessStatement (
       OperationResult = (BOOLEAN)(!OperationResult);\r
     }\r
     switch(OperatorToUse) {\r
-      case END_TAG_OR:\r
+      case EndTagOr:\r
         *PassingState = (BOOLEAN)(*PassingState || OperationResult);\r
         break;\r
-      case END_TAG_AND:\r
+      case EndTagAnd:\r
         *PassingState = (BOOLEAN)(*PassingState && OperationResult);\r
         break;\r
-      case END_TAG_MAX:\r
+      case EndTagMax:\r
         *PassingState = (BOOLEAN)(OperationResult);\r
         break;\r
       default:\r
@@ -599,6 +640,18 @@ ProcessStatement (
   return (Status);\r
 }\r
 \r
+/**\r
+  Break up the next part of the if statement (until the next 'and', 'or', or 'then').\r
+\r
+  @param[in] ParameterNumber      The current parameter number.\r
+  @param[out] EndParameter        Upon successful return, will point to the \r
+                                  parameter to start the next iteration with.\r
+  @param[out] EndTag              Upon successful return, will point to the \r
+                                  type that was found at the end of this statement.\r
+\r
+  @retval TRUE    A valid statement was found.\r
+  @retval FALSE   A valid statement was not found.\r
+**/\r
 BOOLEAN\r
 EFIAPI\r
 BuildNextStatement (\r
@@ -610,7 +663,7 @@ BuildNextStatement (
   CHAR16    *Buffer;\r
   UINTN     BufferSize;\r
 \r
-  *EndTag = END_TAG_MAX;\r
+  *EndTag = EndTagMax;\r
 \r
   for(Buffer = NULL, BufferSize = 0\r
     ; ParameterNumber < gEfiShellParametersProtocol->Argc\r
@@ -621,30 +674,39 @@ BuildNextStatement (
           gEfiShellParametersProtocol->Argv[ParameterNumber],\r
           L"or") == 0) {\r
       *EndParameter = ParameterNumber - 1;\r
-      *EndTag = END_TAG_OR;\r
+      *EndTag = EndTagOr;\r
       break;\r
     } else if (gUnicodeCollation->StriColl(\r
           gUnicodeCollation,\r
           gEfiShellParametersProtocol->Argv[ParameterNumber],\r
           L"and") == 0) {\r
       *EndParameter = ParameterNumber - 1;\r
-      *EndTag = END_TAG_AND;\r
+      *EndTag = EndTagAnd;\r
       break;\r
     } else if (gUnicodeCollation->StriColl(\r
           gUnicodeCollation,\r
           gEfiShellParametersProtocol->Argv[ParameterNumber],\r
           L"then") == 0) {\r
       *EndParameter = ParameterNumber - 1;\r
-      *EndTag = END_TAG_THEN;\r
+      *EndTag = EndTagThen;\r
       break;\r
     }\r
   }\r
-  if (*EndTag == END_TAG_MAX) {\r
+  if (*EndTag == EndTagMax) {\r
     return (FALSE);\r
   }\r
   return (TRUE);\r
 }\r
 \r
+/**\r
+  Move the script file pointer to a different place in the script file.\r
+  This one is special since it handles the if/else/endif syntax.\r
+\r
+  @param[in] ScriptFile     The script file from GetCurrnetScriptFile().\r
+\r
+  @retval TRUE     The move target was found and the move was successful.\r
+  @retval FALSE    Something went wrong.\r
+**/\r
 BOOLEAN\r
 EFIAPI\r
 MoveToTagSpecial (\r
@@ -721,6 +783,14 @@ MoveToTagSpecial (
   return (Found);\r
 }\r
 \r
+/**\r
+  Deal with the result of the if operation.\r
+\r
+  @param[in] Result     The result of the if.\r
+\r
+  @retval EFI_SUCCESS       The operation was successful.\r
+  @retval EFI_NOT_FOUND     The ending tag could not be found.\r
+**/\r
 EFI_STATUS\r
 EFIAPI\r
 PerformResultOperation (\r
@@ -822,7 +892,7 @@ ShellCommandRunIf (
     ForceString     = FALSE;\r
   }\r
 \r
-  for ( ShellStatus = SHELL_SUCCESS, CurrentValue = FALSE, Ending = END_TAG_MAX\r
+  for ( ShellStatus = SHELL_SUCCESS, CurrentValue = FALSE, Ending = EndTagMax\r
       ; CurrentParameter < gEfiShellParametersProtocol->Argc && ShellStatus == SHELL_SUCCESS\r
       ; CurrentParameter++) {\r
     if (gUnicodeCollation->StriColl(\r
@@ -862,7 +932,7 @@ ShellCommandRunIf (
           //\r
           // Optomize to get out of the loop early...\r
           //\r
-          if ((Ending == END_TAG_OR && CurrentValue) || (Ending == END_TAG_AND && !CurrentValue)) {\r
+          if ((Ending == EndTagOr && CurrentValue) || (Ending == EndTagAnd && !CurrentValue)) {\r
             Status = PerformResultOperation(CurrentValue);\r
             if (EFI_ERROR(Status)) {\r
               ShellPrintHiiEx(-1, -1, NULL, STRING_TOKEN (STR_SYNTAX_AFTER_BAD), gShellLevel1HiiHandle, gEfiShellParametersProtocol->Argv[CurrentParameter]);\r
@@ -877,7 +947,7 @@ ShellCommandRunIf (
         //\r
         // Skip over the or or and parameter.\r
         //\r
-        if (Ending == END_TAG_OR || Ending == END_TAG_AND) {\r
+        if (Ending == EndTagOr || Ending == EndTagAnd) {\r
           CurrentParameter++;\r
         }\r
       }\r