# Copyright (c) 2015 - 2018, 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
+# SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
#\r
# This file is used to collect the Variable checking information\r
import os\r
from Common.RangeExpression import RangeExpression\r
from Common.Misc import *\r
-from StringIO import StringIO\r
+from io import BytesIO\r
from struct import pack\r
from Common.DataType import *\r
\r
class VAR_CHECK_PCD_VARIABLE_TAB_CONTAINER(object):\r
def __init__(self):\r
self.var_check_info = []\r
- \r
+\r
def push_back(self, var_check_tab):\r
for tab in self.var_check_info:\r
if tab.equal(var_check_tab):\r
break\r
else:\r
self.var_check_info.append(var_check_tab)\r
- \r
+\r
def dump(self, dest, Phase):\r
- \r
- FormatMap = {}\r
- FormatMap[1] = "=B"\r
- FormatMap[2] = "=H"\r
- FormatMap[4] = "=L"\r
- FormatMap[8] = "=Q"\r
- \r
+\r
if not os.path.isabs(dest):\r
return\r
if not os.path.exists(dest):\r
os.mkdir(dest)\r
BinFileName = "PcdVarCheck.bin"\r
BinFilePath = os.path.join(dest, BinFileName)\r
- Buffer = ''\r
+ Buffer = bytearray()\r
index = 0\r
for var_check_tab in self.var_check_info:\r
index += 1\r
itemIndex += 1\r
realLength += 5\r
for v_data in item.data:\r
- if type(v_data) in (int, long):\r
+ if isinstance(v_data, int):\r
realLength += item.StorageWidth\r
else:\r
realLength += item.StorageWidth\r
realLength += 4\r
\r
Guid = var_check_tab.Guid\r
- b = pack('=LHHBBBBBBBB',\r
- Guid[0],\r
- Guid[1],\r
- Guid[2],\r
- Guid[3],\r
- Guid[4],\r
- Guid[5],\r
- Guid[6],\r
- Guid[7],\r
- Guid[8],\r
- Guid[9],\r
- Guid[10],\r
- )\r
+ b = PackByteFormatGUID(Guid)\r
Buffer += b\r
realLength += 16\r
\r
Buffer += b\r
realLength += 1\r
for v_data in item.data:\r
- if type(v_data) in (int, long):\r
- b = pack(FormatMap[item.StorageWidth], v_data)\r
+ if isinstance(v_data, int):\r
+ b = pack(PACK_CODE_BY_SIZE[item.StorageWidth], v_data)\r
Buffer += b\r
realLength += item.StorageWidth\r
else:\r
- b = pack(FormatMap[item.StorageWidth], v_data[0])\r
+ b = pack(PACK_CODE_BY_SIZE[item.StorageWidth], v_data[0])\r
Buffer += b\r
realLength += item.StorageWidth\r
- b = pack(FormatMap[item.StorageWidth], v_data[1])\r
+ b = pack(PACK_CODE_BY_SIZE[item.StorageWidth], v_data[1])\r
Buffer += b\r
realLength += item.StorageWidth\r
\r
b = pack("=B", var_check_tab.pad)\r
Buffer += b\r
realLength += 1\r
- \r
- DbFile = StringIO()\r
+\r
+ DbFile = BytesIO()\r
if Phase == 'DXE' and os.path.exists(BinFilePath):\r
BinFile = open(BinFilePath, "rb")\r
BinBuffer = BinFile.read()\r
Buffer = BinBuffer + Buffer\r
DbFile.write(Buffer)\r
SaveFileOnChange(BinFilePath, DbFile.getvalue(), True)\r
- \r
+\r
\r
class VAR_CHECK_PCD_VARIABLE_TAB(object):\r
pad = 0xDA\r
def UpdateSize(self):\r
self.HeaderLength = 32 + len(self.Name.split(","))\r
self.Length = 32 + len(self.Name.split(",")) + self.GetValidTabLen()\r
- \r
+\r
def GetValidTabLen(self):\r
validtablen = 0\r
for item in self.validtab:\r
- validtablen += item.Length \r
- return validtablen \r
- \r
+ validtablen += item.Length\r
+ return validtablen\r
+\r
def SetAttributes(self, attributes):\r
self.Attributes = attributes\r
- \r
+\r
def push_back(self, valid_obj):\r
if valid_obj is not None:\r
self.validtab.append(valid_obj)\r
- \r
+\r
def equal(self, varchecktab):\r
if self.Guid == varchecktab.Guid and self.Name == varchecktab.Name:\r
return True\r
else:\r
return False\r
- \r
+\r
def merge(self, varchecktab):\r
for validobj in varchecktab.validtab:\r
if validobj in self.validtab:\r
self.Type = 1\r
self.Length = 0 # Length include this header\r
self.VarOffset = VarOffset\r
- self.StorageWidth = 0\r
self.PcdDataType = PcdDataType.strip()\r
self.rawdata = data\r
self.data = set()\r
- self.ValidData = True\r
- self.updateStorageWidth()\r
-\r
- def updateStorageWidth(self):\r
try:\r
self.StorageWidth = MAX_SIZE_TYPE[self.PcdDataType]\r
+ self.ValidData = True\r
except:\r
self.StorageWidth = 0\r
self.ValidData = False\r
- \r
- def __eq__(self, validObj): \r
+\r
+ def __eq__(self, validObj):\r
return validObj and self.VarOffset == validObj.VarOffset\r
- \r
+\r
class VAR_CHECK_PCD_VALID_LIST(VAR_CHECK_PCD_VALID_OBJ):\r
def __init__(self, VarOffset, validlist, PcdDataType):\r
super(VAR_CHECK_PCD_VALID_LIST, self).__init__(VarOffset, validlist, PcdDataType)\r
self.Type = 1\r
- self.update_data()\r
- self.update_size()\r
- def update_data(self):\r
valid_num_list = []\r
- data_list = []\r
for item in self.rawdata:\r
valid_num_list.extend(item.split(','))\r
- \r
+\r
for valid_num in valid_num_list:\r
valid_num = valid_num.strip()\r
\r
if valid_num.startswith('0x') or valid_num.startswith('0X'):\r
- data_list.append(int(valid_num, 16))\r
+ self.data.add(int(valid_num, 16))\r
else:\r
- data_list.append(int(valid_num))\r
+ self.data.add(int(valid_num))\r
+\r
\r
- \r
- self.data = set(data_list)\r
- \r
- def update_size(self):\r
self.Length = 5 + len(self.data) * self.StorageWidth\r
- \r
- \r
+\r
+\r
class VAR_CHECK_PCD_VALID_RANGE(VAR_CHECK_PCD_VALID_OBJ):\r
def __init__(self, VarOffset, validrange, PcdDataType):\r
super(VAR_CHECK_PCD_VALID_RANGE, self).__init__(VarOffset, validrange, PcdDataType)\r
self.Type = 2\r
- self.update_data()\r
- self.update_size()\r
- def update_data(self):\r
RangeExpr = ""\r
- data_list = []\r
i = 0\r
for item in self.rawdata:\r
if i == 0:\r
range_result = RangeExpression(RangeExpr, self.PcdDataType)(True)\r
for rangelist in range_result:\r
for obj in rangelist.pop():\r
- data_list.append((obj.start, obj.end))\r
- self.data = set(data_list)\r
- \r
- def update_size(self):\r
+ self.data.add((obj.start, obj.end))\r
self.Length = 5 + len(self.data) * 2 * self.StorageWidth\r
- \r
\r
-class VAR_VALID_OBJECT_FACTORY(object):\r
- def __init__(self):\r
- pass\r
- @staticmethod\r
- def Get_valid_object(PcdClass, VarOffset):\r
- if PcdClass.validateranges:\r
- return VAR_CHECK_PCD_VALID_RANGE(VarOffset, PcdClass.validateranges, PcdClass.DatumType)\r
- if PcdClass.validlists:\r
- return VAR_CHECK_PCD_VALID_LIST(VarOffset, PcdClass.validlists, PcdClass.DatumType)\r
- else:\r
- return None\r
+\r
+def GetValidationObject(PcdClass, VarOffset):\r
+ if PcdClass.validateranges:\r
+ return VAR_CHECK_PCD_VALID_RANGE(VarOffset, PcdClass.validateranges, PcdClass.DatumType)\r
+ if PcdClass.validlists:\r
+ return VAR_CHECK_PCD_VALID_LIST(VarOffset, PcdClass.validlists, PcdClass.DatumType)\r
+ else:\r
+ return None\r
projects / mirror_edk2.git / blobdiff