[mirror_edk2.git] / BaseTools / Source / Python / GenFds / FfsFileStatement.py

## @file\r
# process FFS generation from FILE statement\r
#\r
#  Copyright (c) 2007 - 2016, Intel Corporation. All rights reserved.<BR>\r
#\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
#\r
\r
##\r
# Import Modules\r
#\r
import Ffs\r
import Rule\r
import Common.LongFilePathOs as os\r
import StringIO\r
import subprocess\r
\r
from GenFdsGlobalVariable import GenFdsGlobalVariable\r
from CommonDataClass.FdfClass import FileStatementClassObject\r
from Common import EdkLogger\r
from Common.BuildToolError import *\r
from Common.Misc import GuidStructureByteArrayToGuidString\r
from GuidSection import GuidSection\r
from FvImageSection import FvImageSection\r
from Common.Misc import SaveFileOnChange\r
from struct import *\r
\r
## generate FFS from FILE\r
#\r
#\r
class FileStatement (FileStatementClassObject) :\r
    ## The constructor\r
    #\r
    #   @param  self        The object pointer\r
    #\r
    def __init__(self):\r
        FileStatementClassObject.__init__(self)\r
        self.CurrentLineNum = None\r
        self.CurrentLineContent = None\r
        self.FileName = None\r
        self.InfFileName = None\r
        self.SubAlignment = None\r
\r
    ## GenFfs() method\r
    #\r
    #   Generate FFS\r
    #\r
    #   @param  self         The object pointer\r
    #   @param  Dict         dictionary contains macro and value pair\r
    #   @param  FvChildAddr  Array of the inside FvImage base address\r
    #   @param  FvParentAddr Parent Fv base address\r
    #   @retval string       Generated FFS file name\r
    #\r
    def GenFfs(self, Dict = {}, FvChildAddr=[], FvParentAddr=None):\r
        \r
        if self.NameGuid != None and self.NameGuid.startswith('PCD('):\r
            PcdValue = GenFdsGlobalVariable.GetPcdValue(self.NameGuid)\r
            if len(PcdValue) == 0:\r
                EdkLogger.error("GenFds", GENFDS_ERROR, '%s NOT defined.' \\r
                            % (self.NameGuid))\r
            if PcdValue.startswith('{'):\r
                PcdValue = GuidStructureByteArrayToGuidString(PcdValue)\r
            RegistryGuidStr = PcdValue\r
            if len(RegistryGuidStr) == 0:\r
                EdkLogger.error("GenFds", GENFDS_ERROR, 'GUID value for %s in wrong format.' \\r
                            % (self.NameGuid))\r
            self.NameGuid = RegistryGuidStr\r
        \r
        OutputDir = os.path.join(GenFdsGlobalVariable.FfsDir, self.NameGuid)\r
        if not os.path.exists(OutputDir):\r
            os.makedirs(OutputDir)\r
\r
        Dict.update(self.DefineVarDict)\r
        SectionAlignments = None\r
        if self.FvName != None :\r
            Buffer = StringIO.StringIO('')\r
            if self.FvName.upper() not in GenFdsGlobalVariable.FdfParser.Profile.FvDict.keys():\r
                EdkLogger.error("GenFds", GENFDS_ERROR, "FV (%s) is NOT described in FDF file!" % (self.FvName))\r
            Fv = GenFdsGlobalVariable.FdfParser.Profile.FvDict.get(self.FvName.upper())\r
            FileName = Fv.AddToBuffer(Buffer)\r
            SectionFiles = [FileName]\r
\r
        elif self.FdName != None:\r
            if self.FdName.upper() not in GenFdsGlobalVariable.FdfParser.Profile.FdDict.keys():\r
                EdkLogger.error("GenFds", GENFDS_ERROR, "FD (%s) is NOT described in FDF file!" % (self.FdName))\r
            Fd = GenFdsGlobalVariable.FdfParser.Profile.FdDict.get(self.FdName.upper())\r
            FileName = Fd.GenFd()\r
            SectionFiles = [FileName]\r
\r
        elif self.FileName != None:\r
            if hasattr(self, 'FvFileType') and self.FvFileType == 'RAW':\r
                if isinstance(self.FileName, list) and isinstance(self.SubAlignment, list) and len(self.FileName) == len(self.SubAlignment):\r
                    FileContent = ''\r
                    MaxAlignIndex = 0\r
                    MaxAlignValue = 1\r
                    for Index, File in enumerate(self.FileName):\r
                        try:\r
                            f = open(File, 'rb')\r
                        except:\r
                            GenFdsGlobalVariable.ErrorLogger("Error opening RAW file %s." % (File))\r
                        Content = f.read()\r
                        f.close()\r
                        AlignValue = 1\r
                        if self.SubAlignment[Index] != None:\r
                            AlignValue = GenFdsGlobalVariable.GetAlignment(self.SubAlignment[Index])\r
                        if AlignValue > MaxAlignValue:\r
                            MaxAlignIndex = Index\r
                            MaxAlignValue = AlignValue\r
                        FileContent += Content\r
                        if len(FileContent) % AlignValue != 0:\r
                            Size = AlignValue - len(FileContent) % AlignValue\r
                            for i in range(0, Size):\r
                                FileContent += pack('B', 0xFF)\r
\r
                    if FileContent:\r
                        OutputRAWFile = os.path.join(GenFdsGlobalVariable.FfsDir, self.NameGuid, self.NameGuid + '.raw')\r
                        SaveFileOnChange(OutputRAWFile, FileContent, True)\r
                        self.FileName = OutputRAWFile\r
                        self.SubAlignment = self.SubAlignment[MaxAlignIndex]\r
\r
                if self.Alignment and self.SubAlignment:\r
                    if GenFdsGlobalVariable.GetAlignment (self.Alignment) < GenFdsGlobalVariable.GetAlignment (self.SubAlignment):\r
                        self.Alignment = self.SubAlignment\r
                elif self.SubAlignment:\r
                    self.Alignment = self.SubAlignment\r
\r
            self.FileName = GenFdsGlobalVariable.ReplaceWorkspaceMacro(self.FileName)\r
            #Replace $(SAPCE) with real space\r
            self.FileName = self.FileName.replace('$(SPACE)', ' ')\r
            SectionFiles = [GenFdsGlobalVariable.MacroExtend(self.FileName, Dict)]\r
\r
        else:\r
            SectionFiles = []\r
            Index = 0\r
            SectionAlignments = []\r
            for section in self.SectionList :\r
                Index = Index + 1\r
                SecIndex = '%d' %Index\r
                # process the inside FvImage from FvSection or GuidSection\r
                if FvChildAddr != []:\r
                    if isinstance(section, FvImageSection):\r
                        section.FvAddr = FvChildAddr.pop(0)\r
                    elif isinstance(section, GuidSection):\r
                        section.FvAddr = FvChildAddr\r
                if FvParentAddr != None and isinstance(section, GuidSection):\r
                    section.FvParentAddr = FvParentAddr\r
\r
                if self.KeepReloc == False:\r
                    section.KeepReloc = False\r
                sectList, align = section.GenSection(OutputDir, self.NameGuid, SecIndex, self.KeyStringList, None, Dict)\r
                if sectList != []:\r
                    for sect in sectList:\r
                        SectionFiles.append(sect)\r
                        SectionAlignments.append(align)\r
\r
        #\r
        # Prepare the parameter\r
        #\r
        FfsFileOutput = os.path.join(OutputDir, self.NameGuid + '.ffs')\r
        GenFdsGlobalVariable.GenerateFfs(FfsFileOutput, SectionFiles,\r
                                         Ffs.Ffs.FdfFvFileTypeToFileType.get(self.FvFileType),\r
                                         self.NameGuid,\r
                                         Fixed=self.Fixed,\r
                                         CheckSum=self.CheckSum,\r
                                         Align=self.Alignment,\r
                                         SectionAlign=SectionAlignments\r
                                        )\r
\r
        return FfsFileOutput\r
\r
\r
\r
Commit	Line	Data
	1	## @file\r
	2	# process FFS generation from FILE statement\r
	3	#\r
	4	# Copyright (c) 2007 - 2016, Intel Corporation. All rights reserved.<BR>\r
	5	#\r
	6	# This program and the accompanying materials\r
	7	# are licensed and made available under the terms and conditions of the BSD License\r
	8	# which accompanies this distribution. The full text of the license may be found at\r
	9	# http://opensource.org/licenses/bsd-license.php\r
	10	#\r
	11	# THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	12	# WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	13	#\r
	14	\r
	15	##\r
	16	# Import Modules\r
	17	#\r
	18	import Ffs\r
	19	import Rule\r
	20	import Common.LongFilePathOs as os\r
	21	import StringIO\r
	22	import subprocess\r
	23	\r
	24	from GenFdsGlobalVariable import GenFdsGlobalVariable\r
	25	from CommonDataClass.FdfClass import FileStatementClassObject\r
	26	from Common import EdkLogger\r
	27	from Common.BuildToolError import *\r
	28	from Common.Misc import GuidStructureByteArrayToGuidString\r
	29	from GuidSection import GuidSection\r
	30	from FvImageSection import FvImageSection\r
	31	from Common.Misc import SaveFileOnChange\r
	32	from struct import *\r
	33	\r
	34	## generate FFS from FILE\r
	35	#\r
	36	#\r
	37	class FileStatement (FileStatementClassObject) :\r
	38	## The constructor\r
	39	#\r
	40	# @param self The object pointer\r
	41	#\r
	42	def __init__(self):\r
	43	FileStatementClassObject.__init__(self)\r
	44	self.CurrentLineNum = None\r
	45	self.CurrentLineContent = None\r
	46	self.FileName = None\r
	47	self.InfFileName = None\r
	48	self.SubAlignment = None\r
	49	\r
	50	## GenFfs() method\r
	51	#\r
	52	# Generate FFS\r
	53	#\r
	54	# @param self The object pointer\r
	55	# @param Dict dictionary contains macro and value pair\r
	56	# @param FvChildAddr Array of the inside FvImage base address\r
	57	# @param FvParentAddr Parent Fv base address\r
	58	# @retval string Generated FFS file name\r
	59	#\r
	60	def GenFfs(self, Dict = {}, FvChildAddr=[], FvParentAddr=None):\r
	61	\r
	62	if self.NameGuid != None and self.NameGuid.startswith('PCD('):\r
	63	PcdValue = GenFdsGlobalVariable.GetPcdValue(self.NameGuid)\r
	64	if len(PcdValue) == 0:\r
	65	EdkLogger.error("GenFds", GENFDS_ERROR, '%s NOT defined.' \\r
	66	% (self.NameGuid))\r
	67	if PcdValue.startswith('{'):\r
	68	PcdValue = GuidStructureByteArrayToGuidString(PcdValue)\r
	69	RegistryGuidStr = PcdValue\r
	70	if len(RegistryGuidStr) == 0:\r
	71	EdkLogger.error("GenFds", GENFDS_ERROR, 'GUID value for %s in wrong format.' \\r
	72	% (self.NameGuid))\r
	73	self.NameGuid = RegistryGuidStr\r
	74	\r
	75	OutputDir = os.path.join(GenFdsGlobalVariable.FfsDir, self.NameGuid)\r
	76	if not os.path.exists(OutputDir):\r
	77	os.makedirs(OutputDir)\r
	78	\r
	79	Dict.update(self.DefineVarDict)\r
	80	SectionAlignments = None\r
	81	if self.FvName != None :\r
	82	Buffer = StringIO.StringIO('')\r
	83	if self.FvName.upper() not in GenFdsGlobalVariable.FdfParser.Profile.FvDict.keys():\r
	84	EdkLogger.error("GenFds", GENFDS_ERROR, "FV (%s) is NOT described in FDF file!" % (self.FvName))\r
	85	Fv = GenFdsGlobalVariable.FdfParser.Profile.FvDict.get(self.FvName.upper())\r
	86	FileName = Fv.AddToBuffer(Buffer)\r
	87	SectionFiles = [FileName]\r
	88	\r
	89	elif self.FdName != None:\r
	90	if self.FdName.upper() not in GenFdsGlobalVariable.FdfParser.Profile.FdDict.keys():\r
	91	EdkLogger.error("GenFds", GENFDS_ERROR, "FD (%s) is NOT described in FDF file!" % (self.FdName))\r
	92	Fd = GenFdsGlobalVariable.FdfParser.Profile.FdDict.get(self.FdName.upper())\r
	93	FileName = Fd.GenFd()\r
	94	SectionFiles = [FileName]\r
	95	\r
	96	elif self.FileName != None:\r
	97	if hasattr(self, 'FvFileType') and self.FvFileType == 'RAW':\r
	98	if isinstance(self.FileName, list) and isinstance(self.SubAlignment, list) and len(self.FileName) == len(self.SubAlignment):\r
	99	FileContent = ''\r
	100	MaxAlignIndex = 0\r
	101	MaxAlignValue = 1\r
	102	for Index, File in enumerate(self.FileName):\r
	103	try:\r
	104	f = open(File, 'rb')\r
	105	except:\r
	106	GenFdsGlobalVariable.ErrorLogger("Error opening RAW file %s." % (File))\r
	107	Content = f.read()\r
	108	f.close()\r
	109	AlignValue = 1\r
	110	if self.SubAlignment[Index] != None:\r
	111	AlignValue = GenFdsGlobalVariable.GetAlignment(self.SubAlignment[Index])\r
	112	if AlignValue > MaxAlignValue:\r
	113	MaxAlignIndex = Index\r
	114	MaxAlignValue = AlignValue\r
	115	FileContent += Content\r
	116	if len(FileContent) % AlignValue != 0:\r
	117	Size = AlignValue - len(FileContent) % AlignValue\r
	118	for i in range(0, Size):\r
	119	FileContent += pack('B', 0xFF)\r
	120	\r
	121	if FileContent:\r
	122	OutputRAWFile = os.path.join(GenFdsGlobalVariable.FfsDir, self.NameGuid, self.NameGuid + '.raw')\r
	123	SaveFileOnChange(OutputRAWFile, FileContent, True)\r
	124	self.FileName = OutputRAWFile\r
	125	self.SubAlignment = self.SubAlignment[MaxAlignIndex]\r
	126	\r
	127	if self.Alignment and self.SubAlignment:\r
	128	if GenFdsGlobalVariable.GetAlignment (self.Alignment) < GenFdsGlobalVariable.GetAlignment (self.SubAlignment):\r
	129	self.Alignment = self.SubAlignment\r
	130	elif self.SubAlignment:\r
	131	self.Alignment = self.SubAlignment\r
	132	\r
	133	self.FileName = GenFdsGlobalVariable.ReplaceWorkspaceMacro(self.FileName)\r
	134	#Replace $(SAPCE) with real space\r
	135	self.FileName = self.FileName.replace('$(SPACE)', ' ')\r
	136	SectionFiles = [GenFdsGlobalVariable.MacroExtend(self.FileName, Dict)]\r
	137	\r
	138	else:\r
	139	SectionFiles = []\r
	140	Index = 0\r
	141	SectionAlignments = []\r
	142	for section in self.SectionList :\r
	143	Index = Index + 1\r
	144	SecIndex = '%d' %Index\r
	145	# process the inside FvImage from FvSection or GuidSection\r
	146	if FvChildAddr != []:\r
	147	if isinstance(section, FvImageSection):\r
	148	section.FvAddr = FvChildAddr.pop(0)\r
	149	elif isinstance(section, GuidSection):\r
	150	section.FvAddr = FvChildAddr\r
	151	if FvParentAddr != None and isinstance(section, GuidSection):\r
	152	section.FvParentAddr = FvParentAddr\r
	153	\r
	154	if self.KeepReloc == False:\r
	155	section.KeepReloc = False\r
	156	sectList, align = section.GenSection(OutputDir, self.NameGuid, SecIndex, self.KeyStringList, None, Dict)\r
	157	if sectList != []:\r
	158	for sect in sectList:\r
	159	SectionFiles.append(sect)\r
	160	SectionAlignments.append(align)\r
	161	\r
	162	#\r
	163	# Prepare the parameter\r
	164	#\r
	165	FfsFileOutput = os.path.join(OutputDir, self.NameGuid + '.ffs')\r
	166	GenFdsGlobalVariable.GenerateFfs(FfsFileOutput, SectionFiles,\r
	167	Ffs.Ffs.FdfFvFileTypeToFileType.get(self.FvFileType),\r
	168	self.NameGuid,\r
	169	Fixed=self.Fixed,\r
	170	CheckSum=self.CheckSum,\r
	171	Align=self.Alignment,\r
	172	SectionAlign=SectionAlignments\r
	173	)\r
	174	\r
	175	return FfsFileOutput\r
	176	\r
	177	\r
	178	\r