Contributed-under: TianoCore Contribution Agreement 1.0 Signed-off-by: Hess Chen <hesheng.chen@intel.com> Reviewed-by: Yingke Liu <yingke.d.liu@intel.com> git-svn-id: https://svn.code.sf.net/p/edk2/code/trunk/edk2@15809 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			139 lines
		
	
	
		
			5.6 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			139 lines
		
	
	
		
			5.6 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
## @file
 | 
						|
# process OptionROM generation
 | 
						|
#
 | 
						|
#  Copyright (c) 2007 - 2014, Intel Corporation. All rights reserved.<BR>
 | 
						|
#
 | 
						|
#  This program and the accompanying materials
 | 
						|
#  are licensed and made available under the terms and conditions of the BSD License
 | 
						|
#  which accompanies this distribution.  The full text of the license may be found at
 | 
						|
#  http://opensource.org/licenses/bsd-license.php
 | 
						|
#
 | 
						|
#  THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
 | 
						|
#  WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 | 
						|
#
 | 
						|
 | 
						|
##
 | 
						|
# Import Modules
 | 
						|
#
 | 
						|
import Common.LongFilePathOs as os
 | 
						|
import subprocess
 | 
						|
import StringIO
 | 
						|
 | 
						|
import OptRomInfStatement
 | 
						|
from GenFdsGlobalVariable import GenFdsGlobalVariable
 | 
						|
from GenFds import GenFds
 | 
						|
from CommonDataClass.FdfClass import OptionRomClassObject
 | 
						|
from Common.Misc import SaveFileOnChange
 | 
						|
from Common import EdkLogger
 | 
						|
from Common.BuildToolError import *
 | 
						|
 | 
						|
T_CHAR_LF = '\n'
 | 
						|
 | 
						|
## 
 | 
						|
#
 | 
						|
#
 | 
						|
class OPTIONROM (OptionRomClassObject):
 | 
						|
    ## The constructor
 | 
						|
    #
 | 
						|
    #   @param  self        The object pointer
 | 
						|
    #
 | 
						|
    def __init__(self):
 | 
						|
        OptionRomClassObject.__init__(self)
 | 
						|
 | 
						|
 | 
						|
    ## AddToBuffer()
 | 
						|
    #
 | 
						|
    #   Generate Option ROM
 | 
						|
    #
 | 
						|
    #   @param  self        The object pointer
 | 
						|
    #   @param  Buffer      The buffer generated OptROM data will be put
 | 
						|
    #   @retval string      Generated OptROM file path
 | 
						|
    #
 | 
						|
    def AddToBuffer (self, Buffer) :
 | 
						|
 | 
						|
        GenFdsGlobalVariable.InfLogger( "\nGenerating %s Option ROM ..." %self.DriverName)
 | 
						|
 | 
						|
        EfiFileList = []
 | 
						|
        BinFileList = []
 | 
						|
 | 
						|
        # Process Modules in FfsList
 | 
						|
        for FfsFile in self.FfsList :
 | 
						|
            
 | 
						|
            if isinstance(FfsFile, OptRomInfStatement.OptRomInfStatement):
 | 
						|
                FilePathNameList = FfsFile.GenFfs()
 | 
						|
                if len(FilePathNameList) == 0:
 | 
						|
                    EdkLogger.error("GenFds", GENFDS_ERROR, "Module %s not produce .efi files, so NO file could be put into option ROM." % (FfsFile.InfFileName))
 | 
						|
                if FfsFile.OverrideAttribs == None:
 | 
						|
                    EfiFileList.extend(FilePathNameList)
 | 
						|
                else:
 | 
						|
                    FileName = os.path.basename(FilePathNameList[0])
 | 
						|
                    TmpOutputDir = os.path.join(GenFdsGlobalVariable.FvDir, self.DriverName)
 | 
						|
                    if not os.path.exists(TmpOutputDir) :
 | 
						|
                        os.makedirs(TmpOutputDir)
 | 
						|
                    TmpOutputFile = os.path.join(TmpOutputDir, FileName+'.tmp')
 | 
						|
                    
 | 
						|
                    GenFdsGlobalVariable.GenerateOptionRom(TmpOutputFile, 
 | 
						|
                                                           FilePathNameList, 
 | 
						|
                                                           [], 
 | 
						|
                                                           FfsFile.OverrideAttribs.NeedCompress, 
 | 
						|
                                                           FfsFile.OverrideAttribs.PciClassCode, 
 | 
						|
                                                           FfsFile.OverrideAttribs.PciRevision, 
 | 
						|
                                                           FfsFile.OverrideAttribs.PciDeviceId, 
 | 
						|
                                                           FfsFile.OverrideAttribs.PciVendorId)
 | 
						|
                    BinFileList.append(TmpOutputFile)
 | 
						|
            else:
 | 
						|
                FilePathName = FfsFile.GenFfs()
 | 
						|
                if FfsFile.OverrideAttribs != None:
 | 
						|
                    FileName = os.path.basename(FilePathName)
 | 
						|
                    TmpOutputDir = os.path.join(GenFdsGlobalVariable.FvDir, self.DriverName)
 | 
						|
                    if not os.path.exists(TmpOutputDir) :
 | 
						|
                        os.makedirs(TmpOutputDir)
 | 
						|
                    TmpOutputFile = os.path.join(TmpOutputDir, FileName+'.tmp')
 | 
						|
                    
 | 
						|
                    GenFdsGlobalVariable.GenerateOptionRom(TmpOutputFile, 
 | 
						|
                                                           [FilePathName], 
 | 
						|
                                                           [], 
 | 
						|
                                                           FfsFile.OverrideAttribs.NeedCompress, 
 | 
						|
                                                           FfsFile.OverrideAttribs.PciClassCode, 
 | 
						|
                                                           FfsFile.OverrideAttribs.PciRevision, 
 | 
						|
                                                           FfsFile.OverrideAttribs.PciDeviceId, 
 | 
						|
                                                           FfsFile.OverrideAttribs.PciVendorId)
 | 
						|
                    BinFileList.append(TmpOutputFile)
 | 
						|
                else:
 | 
						|
                    if FfsFile.FileType == 'EFI':
 | 
						|
                        EfiFileList.append(FilePathName)
 | 
						|
                    else:
 | 
						|
                        BinFileList.append(FilePathName)
 | 
						|
                        
 | 
						|
        #
 | 
						|
        # Call EfiRom tool
 | 
						|
        #
 | 
						|
        OutputFile = os.path.join(GenFdsGlobalVariable.FvDir, self.DriverName)
 | 
						|
        OutputFile = OutputFile + '.rom'
 | 
						|
        
 | 
						|
        GenFdsGlobalVariable.GenerateOptionRom(
 | 
						|
                                OutputFile,
 | 
						|
                                EfiFileList,
 | 
						|
                                BinFileList
 | 
						|
                                )
 | 
						|
 | 
						|
        GenFdsGlobalVariable.InfLogger( "\nGenerate %s Option ROM Successfully" %self.DriverName)
 | 
						|
        GenFdsGlobalVariable.SharpCounter = 0
 | 
						|
        
 | 
						|
        return OutputFile
 | 
						|
 | 
						|
class OverrideAttribs:
 | 
						|
        
 | 
						|
    ## The constructor
 | 
						|
    #
 | 
						|
    #   @param  self        The object pointer
 | 
						|
    #
 | 
						|
    def __init__(self):
 | 
						|
        
 | 
						|
        self.PciVendorId = None
 | 
						|
        self.PciClassCode = None
 | 
						|
        self.PciDeviceId = None
 | 
						|
        self.PciRevision = None
 | 
						|
        self.NeedCompress = None
 | 
						|
        
 | 
						|
         |