# #\r
# Import Modules\r
#\r
-from struct import pack,unpack\r
+from struct import pack, unpack\r
import collections\r
import copy\r
from Common.VariableAttributes import VariableAttributes\r
NvStorageHeaderSize = 28\r
VariableHeaderSize = 32\r
\r
-def StringArrayToList(StringArray):\r
- StringArray = StringArray[1:-1]\r
- StringArray = '[' + StringArray + ']'\r
- return eval(StringArray)\r
-\r
-def PackGUID(Guid):\r
- GuidBuffer = pack('=LHHBBBBBBBB',\r
- int(Guid[0], 16),\r
- int(Guid[1], 16),\r
- int(Guid[2], 16),\r
- int(Guid[3][-4:-2], 16),\r
- int(Guid[3][-2:], 16),\r
- int(Guid[4][-12:-10], 16),\r
- int(Guid[4][-10:-8], 16),\r
- int(Guid[4][-8:-6], 16),\r
- int(Guid[4][-6:-4], 16),\r
- int(Guid[4][-4:-2], 16),\r
- int(Guid[4][-2:], 16)\r
- )\r
- return GuidBuffer\r
-\r
class VariableMgr(object):\r
- def __init__(self, DefaultStoreMap,SkuIdMap):\r
+ def __init__(self, DefaultStoreMap, SkuIdMap):\r
self.VarInfo = []\r
self.DefaultStoreMap = DefaultStoreMap\r
self.SkuIdMap = SkuIdMap\r
self.VarDefaultBuff = None\r
self.VarDeltaBuff = None\r
\r
- def append_variable(self,uefi_var):\r
+ def append_variable(self, uefi_var):\r
self.VarInfo.append(uefi_var)\r
\r
- def SetVpdRegionMaxSize(self,maxsize):\r
+ def SetVpdRegionMaxSize(self, maxsize):\r
self.VpdRegionSize = maxsize\r
\r
- def SetVpdRegionOffset(self,vpdoffset):\r
+ def SetVpdRegionOffset(self, vpdoffset):\r
self.VpdRegionOffset = vpdoffset\r
\r
- def PatchNVStoreDefaultMaxSize(self,maxsize):\r
+ def PatchNVStoreDefaultMaxSize(self, maxsize):\r
if not self.NVHeaderBuff:\r
return ""\r
- self.NVHeaderBuff = self.NVHeaderBuff[:8] + pack("=Q",maxsize)\r
+ self.NVHeaderBuff = self.NVHeaderBuff[:8] + pack("=Q", maxsize)\r
default_var_bin = VariableMgr.format_data(self.NVHeaderBuff + self.VarDefaultBuff + self.VarDeltaBuff)\r
value_str = "{"\r
default_var_bin_strip = [ data.strip("""'""") for data in default_var_bin]\r
def combine_variable(self):\r
indexedvarinfo = collections.OrderedDict()\r
for item in self.VarInfo:\r
- if (item.skuname,item.defaultstoragename, item.var_name,item.var_guid) not in indexedvarinfo:\r
- indexedvarinfo[(item.skuname,item.defaultstoragename, item.var_name,item.var_guid) ] = []\r
- indexedvarinfo[(item.skuname,item.defaultstoragename, item.var_name,item.var_guid)].append(item)\r
+ if (item.skuname, item.defaultstoragename, item.var_name, item.var_guid) not in indexedvarinfo:\r
+ indexedvarinfo[(item.skuname, item.defaultstoragename, item.var_name, item.var_guid) ] = []\r
+ indexedvarinfo[(item.skuname, item.defaultstoragename, item.var_name, item.var_guid)].append(item)\r
for key in indexedvarinfo:\r
sku_var_info_offset_list = indexedvarinfo[key]\r
if len(sku_var_info_offset_list) == 1:\r
data_type = item.data_type\r
value_list = item.default_value.strip("{").strip("}").split(",")\r
if data_type in DataType.TAB_PCD_NUMERIC_TYPES:\r
- if data_type == ["BOOLEAN", DataType.TAB_UINT8]:\r
- data_flag = "=B"\r
- elif data_type == DataType.TAB_UINT16:\r
- data_flag = "=H"\r
- elif data_type == DataType.TAB_UINT32:\r
- data_flag = "=L"\r
- elif data_type == DataType.TAB_UINT64:\r
- data_flag = "=Q"\r
+ data_flag = DataType.PACK_CODE_BY_SIZE[MAX_SIZE_TYPE[data_type]]\r
data = value_list[0]\r
value_list = []\r
- for data_byte in pack(data_flag,int(data,16) if data.upper().startswith('0X') else int(data)):\r
- value_list += [hex(unpack("B",data_byte)[0])]\r
- newvalue[int(item.var_offset,16) if item.var_offset.upper().startswith("0X") else int(item.var_offset)] = value_list\r
+ for data_byte in pack(data_flag, int(data, 16) if data.upper().startswith('0X') else int(data)):\r
+ value_list.append(hex(unpack("B", data_byte)[0]))\r
+ newvalue[int(item.var_offset, 16) if item.var_offset.upper().startswith("0X") else int(item.var_offset)] = value_list\r
try:\r
newvaluestr = "{" + ",".join(VariableMgr.assemble_variable(newvalue)) +"}"\r
except:\r
- EdkLogger.error("build", AUTOGEN_ERROR, "Variable offset conflict in PCDs: %s \n" % (" and ".join([item.pcdname for item in sku_var_info_offset_list])))\r
+ EdkLogger.error("build", AUTOGEN_ERROR, "Variable offset conflict in PCDs: %s \n" % (" and ".join(item.pcdname for item in sku_var_info_offset_list)))\r
n = sku_var_info_offset_list[0]\r
- indexedvarinfo[key] = [var_info(n.pcdindex,n.pcdname,n.defaultstoragename,n.skuname,n.var_name, n.var_guid, "0x00",n.var_attribute,newvaluestr , newvaluestr , DataType.TAB_VOID)]\r
+ indexedvarinfo[key] = [var_info(n.pcdindex, n.pcdname, n.defaultstoragename, n.skuname, n.var_name, n.var_guid, "0x00", n.var_attribute, newvaluestr, newvaluestr, DataType.TAB_VOID)]\r
self.VarInfo = [item[0] for item in indexedvarinfo.values()]\r
\r
@staticmethod\r
def assemble_variable(valuedict):\r
- ordered_offset = sorted(valuedict.keys())\r
- ordered_value = [valuedict[k] for k in ordered_offset]\r
+ ordered_valuedict_keys = sorted(valuedict.keys())\r
var_value = []\r
- num = 0\r
- for offset in ordered_offset:\r
- if offset < len(var_value):\r
+ for current_valuedict_key in ordered_valuedict_keys:\r
+ if current_valuedict_key < len(var_value):\r
raise\r
- for _ in xrange(offset - len(var_value)):\r
+ for _ in xrange(current_valuedict_key - len(var_value)):\r
var_value.append('0x00')\r
- var_value += ordered_value[num]\r
- num +=1\r
+ var_value += valuedict[current_valuedict_key]\r
return var_value\r
\r
def process_variable_data(self):\r
for item in self.VarInfo:\r
if item.pcdindex not in indexedvarinfo:\r
indexedvarinfo[item.pcdindex] = dict()\r
- indexedvarinfo[item.pcdindex][(item.skuname,item.defaultstoragename)] = item\r
+ indexedvarinfo[item.pcdindex][(item.skuname, item.defaultstoragename)] = item\r
\r
for index in indexedvarinfo:\r
sku_var_info = indexedvarinfo[index]\r
default_data_buffer = ""\r
others_data_buffer = ""\r
tail = None\r
- default_sku_default = indexedvarinfo[index].get((DataType.TAB_DEFAULT,DataType.TAB_DEFAULT_STORES_DEFAULT))\r
+ default_sku_default = indexedvarinfo[index].get((DataType.TAB_DEFAULT, DataType.TAB_DEFAULT_STORES_DEFAULT))\r
\r
if default_sku_default.data_type not in DataType.TAB_PCD_NUMERIC_TYPES:\r
- var_max_len = max([len(var_item.default_value.split(",")) for var_item in sku_var_info.values()])\r
+ var_max_len = max(len(var_item.default_value.split(",")) for var_item in sku_var_info.values())\r
if len(default_sku_default.default_value.split(",")) < var_max_len:\r
- tail = ",".join([ "0x00" for i in range(var_max_len-len(default_sku_default.default_value.split(",")))])\r
+ tail = ",".join("0x00" for i in range(var_max_len-len(default_sku_default.default_value.split(","))))\r
\r
- default_data_buffer = VariableMgr.PACK_VARIABLES_DATA(default_sku_default.default_value,default_sku_default.data_type,tail)\r
+ default_data_buffer = VariableMgr.PACK_VARIABLES_DATA(default_sku_default.default_value, default_sku_default.data_type, tail)\r
\r
default_data_array = ()\r
for item in default_data_buffer:\r
- default_data_array += unpack("B",item)\r
+ default_data_array += unpack("B", item)\r
\r
- var_data[(DataType.TAB_DEFAULT,DataType.TAB_DEFAULT_STORES_DEFAULT)][index] = (default_data_buffer,sku_var_info[(DataType.TAB_DEFAULT,DataType.TAB_DEFAULT_STORES_DEFAULT)])\r
+ var_data[(DataType.TAB_DEFAULT, DataType.TAB_DEFAULT_STORES_DEFAULT)][index] = (default_data_buffer, sku_var_info[(DataType.TAB_DEFAULT, DataType.TAB_DEFAULT_STORES_DEFAULT)])\r
\r
- for (skuid,defaultstoragename) in indexedvarinfo[index]:\r
+ for (skuid, defaultstoragename) in indexedvarinfo[index]:\r
tail = None\r
- if (skuid,defaultstoragename) == (DataType.TAB_DEFAULT,DataType.TAB_DEFAULT_STORES_DEFAULT):\r
+ if (skuid, defaultstoragename) == (DataType.TAB_DEFAULT, DataType.TAB_DEFAULT_STORES_DEFAULT):\r
continue\r
- other_sku_other = indexedvarinfo[index][(skuid,defaultstoragename)]\r
+ other_sku_other = indexedvarinfo[index][(skuid, defaultstoragename)]\r
\r
if default_sku_default.data_type not in DataType.TAB_PCD_NUMERIC_TYPES:\r
if len(other_sku_other.default_value.split(",")) < var_max_len:\r
- tail = ",".join([ "0x00" for i in range(var_max_len-len(other_sku_other.default_value.split(",")))])\r
+ tail = ",".join("0x00" for i in range(var_max_len-len(other_sku_other.default_value.split(","))))\r
\r
- others_data_buffer = VariableMgr.PACK_VARIABLES_DATA(other_sku_other.default_value,other_sku_other.data_type,tail)\r
+ others_data_buffer = VariableMgr.PACK_VARIABLES_DATA(other_sku_other.default_value, other_sku_other.data_type, tail)\r
\r
others_data_array = ()\r
for item in others_data_buffer:\r
- others_data_array += unpack("B",item)\r
+ others_data_array += unpack("B", item)\r
\r
data_delta = VariableMgr.calculate_delta(default_data_array, others_data_array)\r
\r
- var_data[(skuid,defaultstoragename)][index] = (data_delta,sku_var_info[(skuid,defaultstoragename)])\r
+ var_data[(skuid, defaultstoragename)][index] = (data_delta, sku_var_info[(skuid, defaultstoragename)])\r
return var_data\r
\r
def new_process_varinfo(self):\r
if not var_data:\r
return []\r
\r
- pcds_default_data = var_data.get((DataType.TAB_DEFAULT,DataType.TAB_DEFAULT_STORES_DEFAULT),{})\r
+ pcds_default_data = var_data.get((DataType.TAB_DEFAULT, DataType.TAB_DEFAULT_STORES_DEFAULT), {})\r
NvStoreDataBuffer = ""\r
var_data_offset = collections.OrderedDict()\r
offset = NvStorageHeaderSize\r
- for default_data,default_info in pcds_default_data.values():\r
+ for default_data, default_info in pcds_default_data.values():\r
var_name_buffer = VariableMgr.PACK_VARIABLE_NAME(default_info.var_name)\r
\r
vendorguid = default_info.var_guid.split('-')\r
\r
if default_info.var_attribute:\r
- var_attr_value,_ = VariableAttributes.GetVarAttributes(default_info.var_attribute)\r
+ var_attr_value, _ = VariableAttributes.GetVarAttributes(default_info.var_attribute)\r
else:\r
var_attr_value = 0x07\r
\r
nv_default_part = VariableMgr.AlignData(VariableMgr.PACK_DEFAULT_DATA(0, 0, VariableMgr.unpack_data(variable_storage_header_buffer+NvStoreDataBuffer)), 8)\r
\r
data_delta_structure_buffer = ""\r
- for skuname,defaultstore in var_data:\r
- if (skuname,defaultstore) == (DataType.TAB_DEFAULT,DataType.TAB_DEFAULT_STORES_DEFAULT):\r
+ for skuname, defaultstore in var_data:\r
+ if (skuname, defaultstore) == (DataType.TAB_DEFAULT, DataType.TAB_DEFAULT_STORES_DEFAULT):\r
continue\r
- pcds_sku_data = var_data[(skuname,defaultstore)]\r
+ pcds_sku_data = var_data[(skuname, defaultstore)]\r
delta_data_set = []\r
for pcdindex in pcds_sku_data:\r
offset = var_data_offset[pcdindex]\r
- delta_data,_ = pcds_sku_data[pcdindex]\r
+ delta_data, _ = pcds_sku_data[pcdindex]\r
delta_data = [(item[0] + offset, item[1]) for item in delta_data]\r
delta_data_set.extend(delta_data)\r
\r
- data_delta_structure_buffer += VariableMgr.AlignData(self.PACK_DELTA_DATA(skuname,defaultstore,delta_data_set), 8)\r
+ data_delta_structure_buffer += VariableMgr.AlignData(self.PACK_DELTA_DATA(skuname, defaultstore, delta_data_set), 8)\r
\r
size = len(nv_default_part + data_delta_structure_buffer) + 16\r
maxsize = self.VpdRegionSize if self.VpdRegionSize else size\r
- NV_Store_Default_Header = VariableMgr.PACK_NV_STORE_DEFAULT_HEADER(size,maxsize)\r
+ NV_Store_Default_Header = VariableMgr.PACK_NV_STORE_DEFAULT_HEADER(size, maxsize)\r
\r
self.NVHeaderBuff = NV_Store_Default_Header\r
self.VarDefaultBuff =nv_default_part\r
def unpack_data(data):\r
final_data = ()\r
for item in data:\r
- final_data += unpack("B",item)\r
+ final_data += unpack("B", item)\r
return final_data\r
\r
@staticmethod\r
data_delta = []\r
for i in range(len(default)):\r
if default[i] != theother[i]:\r
- data_delta.append((i,theother[i]))\r
+ data_delta.append((i, theother[i]))\r
return data_delta\r
\r
def dump(self):\r
Guid = GuidStructureStringToGuidString(Guid)\r
GuidBuffer = PackGUID(Guid.split('-'))\r
\r
- SizeBuffer = pack('=L',size)\r
- FormatBuffer = pack('=B',0x5A)\r
- StateBuffer = pack('=B',0xFE)\r
- reservedBuffer = pack('=H',0)\r
- reservedBuffer += pack('=L',0)\r
+ SizeBuffer = pack('=L', size)\r
+ FormatBuffer = pack('=B', 0x5A)\r
+ StateBuffer = pack('=B', 0xFE)\r
+ reservedBuffer = pack('=H', 0)\r
+ reservedBuffer += pack('=L', 0)\r
\r
return GuidBuffer + SizeBuffer + FormatBuffer + StateBuffer + reservedBuffer\r
\r
@staticmethod\r
- def PACK_NV_STORE_DEFAULT_HEADER(size,maxsize):\r
- Signature = pack('=B',ord('N'))\r
- Signature += pack("=B",ord('S'))\r
- Signature += pack("=B",ord('D'))\r
- Signature += pack("=B",ord('B'))\r
+ def PACK_NV_STORE_DEFAULT_HEADER(size, maxsize):\r
+ Signature = pack('=B', ord('N'))\r
+ Signature += pack("=B", ord('S'))\r
+ Signature += pack("=B", ord('D'))\r
+ Signature += pack("=B", ord('B'))\r
\r
- SizeBuffer = pack("=L",size)\r
- MaxSizeBuffer = pack("=Q",maxsize)\r
+ SizeBuffer = pack("=L", size)\r
+ MaxSizeBuffer = pack("=Q", maxsize)\r
\r
return Signature + SizeBuffer + MaxSizeBuffer\r
\r
@staticmethod\r
- def PACK_VARIABLE_HEADER(attribute,namesize,datasize,vendorguid):\r
+ def PACK_VARIABLE_HEADER(attribute, namesize, datasize, vendorguid):\r
\r
- Buffer = pack('=H',0x55AA) # pack StartID\r
- Buffer += pack('=B',0x3F) # pack State\r
- Buffer += pack('=B',0) # pack reserved\r
+ Buffer = pack('=H', 0x55AA) # pack StartID\r
+ Buffer += pack('=B', 0x3F) # pack State\r
+ Buffer += pack('=B', 0) # pack reserved\r
\r
- Buffer += pack('=L',attribute)\r
- Buffer += pack('=L',namesize)\r
- Buffer += pack('=L',datasize)\r
+ Buffer += pack('=L', attribute)\r
+ Buffer += pack('=L', namesize)\r
+ Buffer += pack('=L', datasize)\r
\r
Buffer += PackGUID(vendorguid)\r
\r
data_len = 0\r
if data_type == DataType.TAB_VOID:\r
for value_char in var_value.strip("{").strip("}").split(","):\r
- Buffer += pack("=B",int(value_char,16))\r
+ Buffer += pack("=B", int(value_char, 16))\r
data_len += len(var_value.split(","))\r
if tail:\r
for value_char in tail.split(","):\r
- Buffer += pack("=B",int(value_char,16))\r
+ Buffer += pack("=B", int(value_char, 16))\r
data_len += len(tail.split(","))\r
elif data_type == "BOOLEAN":\r
- Buffer += pack("=B",True) if var_value.upper() == "TRUE" else pack("=B",False)\r
+ Buffer += pack("=B", True) if var_value.upper() == "TRUE" else pack("=B", False)\r
data_len += 1\r
elif data_type == DataType.TAB_UINT8:\r
- Buffer += pack("=B",GetIntegerValue(var_value))\r
+ Buffer += pack("=B", GetIntegerValue(var_value))\r
data_len += 1\r
elif data_type == DataType.TAB_UINT16:\r
- Buffer += pack("=H",GetIntegerValue(var_value))\r
+ Buffer += pack("=H", GetIntegerValue(var_value))\r
data_len += 2\r
elif data_type == DataType.TAB_UINT32:\r
- Buffer += pack("=L",GetIntegerValue(var_value))\r
+ Buffer += pack("=L", GetIntegerValue(var_value))\r
data_len += 4\r
elif data_type == DataType.TAB_UINT64:\r
- Buffer += pack("=Q",GetIntegerValue(var_value))\r
+ Buffer += pack("=Q", GetIntegerValue(var_value))\r
data_len += 8\r
\r
return Buffer\r
\r
@staticmethod\r
- def PACK_DEFAULT_DATA(defaultstoragename,skuid,var_value):\r
+ def PACK_DEFAULT_DATA(defaultstoragename, skuid, var_value):\r
Buffer = ""\r
- Buffer += pack("=L",4+8+8)\r
- Buffer += pack("=Q",int(skuid))\r
- Buffer += pack("=Q",int(defaultstoragename))\r
+ Buffer += pack("=L", 4+8+8)\r
+ Buffer += pack("=Q", int(skuid))\r
+ Buffer += pack("=Q", int(defaultstoragename))\r
\r
for item in var_value:\r
- Buffer += pack("=B",item)\r
+ Buffer += pack("=B", item)\r
\r
- Buffer = pack("=L",len(Buffer)+4) + Buffer\r
+ Buffer = pack("=L", len(Buffer)+4) + Buffer\r
\r
return Buffer\r
\r
- def GetSkuId(self,skuname):\r
+ def GetSkuId(self, skuname):\r
if skuname not in self.SkuIdMap:\r
return None\r
return self.SkuIdMap.get(skuname)[0]\r
\r
- def GetDefaultStoreId(self,dname):\r
+ def GetDefaultStoreId(self, dname):\r
if dname not in self.DefaultStoreMap:\r
return None\r
return self.DefaultStoreMap.get(dname)[0]\r
\r
- def PACK_DELTA_DATA(self,skuname,defaultstoragename,delta_list):\r
+ def PACK_DELTA_DATA(self, skuname, defaultstoragename, delta_list):\r
skuid = self.GetSkuId(skuname)\r
defaultstorageid = self.GetDefaultStoreId(defaultstoragename)\r
Buffer = ""\r
- Buffer += pack("=L",4+8+8)\r
- Buffer += pack("=Q",int(skuid))\r
- Buffer += pack("=Q",int(defaultstorageid))\r
- for (delta_offset,value) in delta_list:\r
- Buffer += pack("=L",delta_offset)\r
- Buffer = Buffer[:-1] + pack("=B",value)\r
+ Buffer += pack("=L", 4+8+8)\r
+ Buffer += pack("=Q", int(skuid))\r
+ Buffer += pack("=Q", int(defaultstorageid))\r
+ for (delta_offset, value) in delta_list:\r
+ Buffer += pack("=L", delta_offset)\r
+ Buffer = Buffer[:-1] + pack("=B", value)\r
\r
- Buffer = pack("=L",len(Buffer) + 4) + Buffer\r
+ Buffer = pack("=L", len(Buffer) + 4) + Buffer\r
\r
return Buffer\r
\r
mybuffer = data\r
if (len(data) % align) > 0:\r
for i in range(align - (len(data) % align)):\r
- mybuffer += pack("=B",0)\r
+ mybuffer += pack("=B", 0)\r
\r
return mybuffer\r
\r
def PACK_VARIABLE_NAME(var_name):\r
Buffer = ""\r
for name_char in var_name.strip("{").strip("}").split(","):\r
- Buffer += pack("=B",int(name_char,16))\r
+ Buffer += pack("=B", int(name_char, 16))\r
\r
return Buffer\r