]> git.proxmox.com Git - mirror_edk2.git/blobdiff - BaseTools/Source/Python/Common/Expression.py
BaseTools: resolve initialization order errors in VfrFormPkg.h
[mirror_edk2.git] / BaseTools / Source / Python / Common / Expression.py
index ba83e02f75d3bfc00ef2993f80f7a64e221a01cc..55fa06d414ea1029293ee1c92bc2ceb314e4d26f 100644 (file)
@@ -15,7 +15,9 @@
 from Common.GlobalData import *\r
 from CommonDataClass.Exceptions import BadExpression\r
 from CommonDataClass.Exceptions import WrnExpression\r
-from Misc import GuidStringToGuidStructureString\r
+from Misc import GuidStringToGuidStructureString, ParseFieldValue\r
+import Common.EdkLogger as EdkLogger\r
+import copy\r
 \r
 ERR_STRING_EXPR         = 'This operator cannot be used in string expression: [%s].'\r
 ERR_SNYTAX              = 'Syntax error, the rest of expression cannot be evaluated: [%s].'\r
@@ -114,6 +116,15 @@ def ReplaceExprMacro(String, Macros, ExceptionList = None):
         StrList[i] = RetStr\r
     return ''.join(StrList)\r
 \r
+# transfer int to string for in/not in expression\r
+def IntToStr(Value):\r
+    StrList = []\r
+    while Value > 0:\r
+        StrList.append(chr(Value & 0xff))\r
+        Value = Value >> 8\r
+    Value = '"' + ''.join(StrList) + '"'\r
+    return Value\r
+\r
 SupportedInMacroList = ['TARGET', 'TOOL_CHAIN_TAG', 'ARCH', 'FAMILY']\r
 \r
 class ValueExpression(object):\r
@@ -146,10 +157,24 @@ class ValueExpression(object):
     def Eval(Operator, Oprand1, Oprand2 = None):\r
         WrnExp = None\r
 \r
-        if Operator not in ["==", "!=", ">=", "<=", ">", "<", "in", "not in"] and \\r
-            (type(Oprand1) == type('') or type(Oprand2) == type('')):\r
-            raise BadExpression(ERR_STRING_EXPR % Operator)\r
-\r
+        if Operator not in ["in", "not in"] and (type(Oprand1) == type('') or type(Oprand2) == type('')):\r
+            if type(Oprand1) == type(''):\r
+                if Oprand1[0] in ['"', "'"] or Oprand1.startswith('L"') or Oprand1.startswith("L'")or Oprand1.startswith('UINT'):\r
+                    Oprand1, Size = ParseFieldValue(Oprand1)\r
+                else:\r
+                    Oprand1,Size = ParseFieldValue('"' + Oprand1 + '"')\r
+            if type(Oprand2) == type(''):\r
+                if Oprand2[0] in ['"', "'"] or Oprand2.startswith('L"') or Oprand2.startswith("L'") or Oprand2.startswith('UINT'):\r
+                    Oprand2, Size = ParseFieldValue(Oprand2)\r
+                else:\r
+                    Oprand2, Size = ParseFieldValue('"' + Oprand2 + '"')\r
+            if type(Oprand1) == type('') or type(Oprand2) == type(''):\r
+                raise BadExpression(ERR_STRING_EXPR % Operator)\r
+        if Operator in ['in', 'not in']:\r
+            if type(Oprand1) != type(''):\r
+                Oprand1 = IntToStr(Oprand1)\r
+            if type(Oprand2) != type(''):\r
+                Oprand2 = IntToStr(Oprand2)\r
         TypeDict = {\r
             type(0)  : 0,\r
             type(0L) : 0,\r
@@ -226,6 +251,9 @@ class ValueExpression(object):
             self._Expr = Expression\r
             self._NoProcess = True\r
             return\r
+        if Expression.strip().startswith('{') and Expression.strip().endswith('}'):\r
+            self._Expr = Expression\r
+            self._NoProcess = True\r
 \r
         self._Expr = ReplaceExprMacro(Expression.strip(),\r
                                   SymbolTable,\r
@@ -237,7 +265,7 @@ class ValueExpression(object):
         #\r
         # The symbol table including PCD and macro mapping\r
         #\r
-        self._Symb = SymbolTable\r
+        self._Symb = copy.deepcopy(SymbolTable)\r
         self._Symb.update(self.LogicalOperators)\r
         self._Idx = 0\r
         self._Len = len(self._Expr)\r
@@ -284,7 +312,7 @@ class ValueExpression(object):
             elif not Val:\r
                 Val = False\r
                 RealVal = '""'\r
-            elif not Val.startswith('L"') and not Val.startswith('{'):\r
+            elif not Val.startswith('L"') and not Val.startswith('{') and not Val.startswith("L'"):\r
                 Val = True\r
                 RealVal = '"' + RealVal + '"'\r
 \r
@@ -426,6 +454,15 @@ class ValueExpression(object):
         Radix = 10\r
         if self._Token.lower()[0:2] == '0x' and len(self._Token) > 2:\r
             Radix = 16\r
+        if self._Token.startswith('"') or self._Token.startswith("'")\\r
+            or self._Token.startswith("L'") or self._Token.startswith('L"'):\r
+            Flag = 0\r
+            for Index in range(len(self._Token)):\r
+                if self._Token[Index] in ['"', "'"]:\r
+                    Flag += 1\r
+            if Flag == 2:\r
+                self._Token = ParseFieldValue(self._Token)[0]\r
+                return True\r
         try:\r
             self._Token = int(self._Token, Radix)\r
             return True\r
@@ -473,10 +510,12 @@ class ValueExpression(object):
         Expr = self._Expr[self._Idx:].replace('\\\\', '//').replace('\\\"', '\\\'')\r
         for Ch in Expr:\r
             self._Idx += 1\r
-            if Ch == '"':\r
+            if Ch == '"' or Ch == "'":\r
                 break\r
         self._Token = self._LiteralToken = self._Expr[Idx:self._Idx]\r
-        if not self._Token.endswith('"'):\r
+        if self._Token.startswith('"') and not self._Token.endswith('"'):\r
+            raise BadExpression(ERR_STRING_TOKEN % self._Token)\r
+        if self._Token.startswith("'") and not self._Token.endswith("'"):\r
             raise BadExpression(ERR_STRING_TOKEN % self._Token)\r
         self._Token = self._Token[1:-1]\r
         return self._Token\r
@@ -578,7 +617,44 @@ class ValueExpression(object):
             self._Idx += 1\r
             UStr = self.__GetString()\r
             self._Token = 'L"' + UStr + '"'\r
+            self._Token, Size = ParseFieldValue(self._Token)\r
+            return self._Token\r
+        elif Expr.startswith("L'"):\r
+            # Skip L\r
+            self._Idx += 1\r
+            UStr = self.__GetString()\r
+            self._Token = "L'" + UStr + "'"\r
+            self._Token, Size = ParseFieldValue(self._Token)\r
+            return self._Token\r
+        elif Expr.startswith('"'):\r
+            UStr = self.__GetString()\r
+            self._Token = '"' + UStr + '"'\r
+            self._Token, Size = ParseFieldValue(self._Token)\r
             return self._Token\r
+        elif Expr.startswith("'"):\r
+            UStr = self.__GetString()\r
+            self._Token = "'" + UStr + "'"\r
+            self._Token, Size = ParseFieldValue(self._Token)\r
+            return self._Token\r
+        elif Expr.startswith('UINT'):\r
+            Re = re.compile('(?:UINT8|UINT16|UINT32|UINT64)\((.+)\)')\r
+            try:\r
+                RetValue = Re.search(Expr).group(1)\r
+            except:\r
+                 raise BadExpression('Invalid Expression %s' % Expr)\r
+            Idx = self._Idx\r
+            for Ch in Expr:\r
+                self._Idx += 1\r
+                if Ch == '(':\r
+                    Prefix = self._Expr[Idx:self._Idx - 1]\r
+                    Idx = self._Idx\r
+                if Ch == ')':\r
+                    TmpValue = self._Expr[Idx :self._Idx - 1]\r
+                    TmpValue = ValueExpression(TmpValue)(True)\r
+                    TmpValue = '0x%x' % int(TmpValue) if type(TmpValue) != type('') else TmpValue\r
+                    break\r
+            self._Token, Size = ParseFieldValue(Prefix + '(' + TmpValue + ')')\r
+            return  self._Token\r
 \r
         self._Token = ''\r
         if Expr:\r
@@ -648,6 +724,158 @@ class ValueExpression(object):
         self._Idx = Idx\r
         return False\r
 \r
+class ValueExpressionEx(ValueExpression):\r
+    def __init__(self, PcdValue, PcdType, SymbolTable={}):\r
+        ValueExpression.__init__(self, PcdValue, SymbolTable)\r
+        self.PcdValue = PcdValue\r
+        self.PcdType = PcdType\r
+\r
+    def __call__(self, RealValue=False, Depth=0):\r
+        PcdValue = self.PcdValue\r
+        try:\r
+            PcdValue = ValueExpression.__call__(self, RealValue, Depth)\r
+        except WrnExpression, Value:\r
+            PcdValue = Value.result\r
+\r
+        if PcdValue == 'True':\r
+            PcdValue = '1'\r
+        if PcdValue == 'False':\r
+            PcdValue = '0'\r
+        if self.PcdType in ['UINT8', 'UINT16', 'UINT32', 'UINT64', 'BOOLEAN']:\r
+            PcdValue = PcdValue.strip()\r
+            if type(PcdValue) == type('') and PcdValue.startswith('{') and PcdValue.endswith('}'):\r
+                PcdValue = PcdValue[1:-1].split(',')\r
+            if type(PcdValue) == type([]):\r
+                TmpValue = 0\r
+                Size = 0\r
+                for Item in PcdValue:\r
+                    if Item.startswith('UINT16'):\r
+                        ItemSize = 2\r
+                    elif Item.startswith('UINT32'):\r
+                        ItemSize = 4\r
+                    elif Item.startswith('UINT64'):\r
+                        ItemSize = 8\r
+                    else:\r
+                        ItemSize = 0\r
+                    Item = ValueExpressionEx(Item, self.PcdType, self._Symb)(True)\r
+\r
+                    if ItemSize == 0:\r
+                        ItemValue, ItemSize = ParseFieldValue(Item)\r
+                    else:\r
+                        ItemValue = ParseFieldValue(Item)[0]\r
+\r
+                    if type(ItemValue) == type(''):\r
+                        ItemValue = int(ItemValue, 16) if ItemValue.startswith('0x') else int(ItemValue)\r
+\r
+                    TmpValue = (ItemValue << (Size * 8)) | TmpValue\r
+                    Size = Size + ItemSize\r
+            else:\r
+                TmpValue, Size = ParseFieldValue(PcdValue)\r
+            if type(TmpValue) == type(''):\r
+                TmpValue = int(TmpValue)\r
+            else:\r
+                PcdValue = '0x%0{}X'.format(Size) % (TmpValue)\r
+            if TmpValue < 0:\r
+                raise  BadExpression('Type %s PCD Value is negative' % self.PcdType)\r
+            if self.PcdType == 'UINT8' and Size > 1:\r
+                raise BadExpression('Type %s PCD Value Size is Larger than 1 byte' % self.PcdType)\r
+            if self.PcdType == 'UINT16' and Size > 2:\r
+                raise BadExpression('Type %s PCD Value Size is Larger than 2 byte' % self.PcdType)\r
+            if self.PcdType == 'UINT32' and Size > 4:\r
+                raise BadExpression('Type %s PCD Value Size is Larger than 4 byte' % self.PcdType)\r
+            if self.PcdType == 'UINT64' and Size > 8:\r
+                raise BadExpression('Type %s PCD Value Size is Larger than 8 byte' % self.PcdType)\r
+        if self.PcdType in ['VOID*']:\r
+            try:\r
+                TmpValue = long(PcdValue)\r
+                TmpList = []\r
+                if TmpValue.bit_length() == 0:\r
+                    PcdValue = '{0x00}'\r
+                else:\r
+                    for I in range((TmpValue.bit_length() + 7) / 8):\r
+                        TmpList.append('0x%02x' % ((TmpValue >> I * 8) & 0xff))\r
+                    PcdValue = '{' + ', '.join(TmpList) + '}'\r
+            except:\r
+                if PcdValue.strip().startswith('{'):\r
+                    PcdValue = PcdValue.strip()[1:-1].strip()\r
+                    Size = 0\r
+                    ValueStr = ''\r
+                    TokenSpaceGuidName = ''\r
+                    if PcdValue.startswith('GUID') and PcdValue.endswith(')'):\r
+                        try:\r
+                            TokenSpaceGuidName = re.search('GUID\((\w+)\)', PcdValue).group(1)\r
+                        except:\r
+                            pass\r
+                        if TokenSpaceGuidName and TokenSpaceGuidName in self._Symb:\r
+                            PcdValue = 'GUID(' + self._Symb[TokenSpaceGuidName] + ')'\r
+                        elif TokenSpaceGuidName:\r
+                            raise BadExpression('%s not found in DEC file' % TokenSpaceGuidName)\r
+\r
+                        ListItem, Size = ParseFieldValue(PcdValue)\r
+                    elif PcdValue.startswith('DEVICE_PATH') and PcdValue.endswith(')'):\r
+                        ListItem, Size = ParseFieldValue(PcdValue)\r
+                    else:\r
+                        ListItem = PcdValue.split(',')\r
+\r
+                    if type(ListItem) == type(0) or type(ListItem) == type(0L):\r
+                        for Index in range(0, Size):\r
+                            ValueStr += '0x%02X' % (int(ListItem) & 255)\r
+                            ListItem >>= 8\r
+                            ValueStr += ', '\r
+                            PcdValue = '{' + ValueStr[:-2] + '}'\r
+                    elif type(ListItem) == type(''):\r
+                        if ListItem.startswith('{') and ListItem.endswith('}'):\r
+                            PcdValue = ListItem\r
+                    else:\r
+                        LabelDict = {}\r
+                        ReLabel = re.compile('LABEL\((\w+)\)')\r
+                        ReOffset = re.compile('OFFSET_OF\((\w+)\)')\r
+                        for Index, Item in enumerate(ListItem):\r
+                            # for LABEL parse\r
+                            Item = Item.strip()\r
+                            try:\r
+                                LabelList = ReLabel.findall(Item)\r
+                                for Label in LabelList:\r
+                                    if Label not in LabelDict.keys():\r
+                                        LabelDict[Label] = str(Index)\r
+                                Item = ReLabel.sub('', Item)\r
+                            except:\r
+                                pass\r
+                            try:\r
+                                OffsetList = ReOffset.findall(Item)\r
+                            except:\r
+                                pass\r
+                            for Offset in OffsetList:\r
+                                if Offset in LabelDict.keys():\r
+                                    Re = re.compile('OFFSET_OF\(%s\)'% Offset)\r
+                                    Item = Re.sub(LabelDict[Offset], Item)\r
+                                else:\r
+                                    raise BadExpression('%s not defined before use' % Offset)\r
+                            if Item.startswith('UINT16'):\r
+                                ItemSize = 2\r
+                            elif Item.startswith('UINT32'):\r
+                                ItemSize = 4\r
+                            elif Item.startswith('UINT64'):\r
+                                ItemSize = 8\r
+                            else:\r
+                                ItemSize = 0\r
+                            TmpValue = ValueExpressionEx(Item, self.PcdType, self._Symb)(True)\r
+                            Item = '0x%x' % TmpValue if type(TmpValue) != type('') else TmpValue\r
+                            if ItemSize == 0:\r
+                                ItemValue, ItemSize = ParseFieldValue(Item)\r
+                            else:\r
+                                ItemValue = ParseFieldValue(Item)[0]\r
+                            for I in range(0, ItemSize):\r
+                                ValueStr += '0x%02X' % (int(ItemValue) & 255)\r
+                                ItemValue >>= 8\r
+                                ValueStr += ', '\r
+                            Size += ItemSize\r
+\r
+                        if Size > 0:\r
+                            PcdValue = '{' + ValueStr[:-2] + '}'\r
+        if RealValue:\r
+            return PcdValue\r
+\r
 if __name__ == '__main__':\r
     pass\r
     while True:\r