2009-07-17 11:10:31 +02:00
## @file
# generate flash image
#
2010-05-18 07:04:32 +02:00
# Copyright (c) 2007 - 2010, Intel Corporation. All rights reserved.<BR>
2009-07-17 11:10:31 +02:00
#
2010-05-18 07:04:32 +02:00
# This program and the accompanying materials
2009-07-17 11:10:31 +02:00
# 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
#
from optparse import OptionParser
import sys
import os
import linecache
import FdfParser
2009-09-11 05:14:43 +02:00
import Common . BuildToolError as BuildToolError
2009-07-17 11:10:31 +02:00
from GenFdsGlobalVariable import GenFdsGlobalVariable
from Workspace . WorkspaceDatabase import WorkspaceDatabase
from Workspace . BuildClassObject import PcdClassObject
from Workspace . BuildClassObject import ModuleBuildClassObject
import RuleComplexFile
from EfiSection import EfiSection
import StringIO
import Common . TargetTxtClassObject as TargetTxtClassObject
import Common . ToolDefClassObject as ToolDefClassObject
import Common . DataType
import Common . GlobalData as GlobalData
from Common import EdkLogger
from Common . String import *
from Common . Misc import DirCache , PathClass
2010-05-18 07:04:32 +02:00
from Common . Misc import SaveFileOnChange
2011-09-18 14:17:25 +02:00
from Common . BuildVersion import gBUILD_VERSION
2009-07-17 11:10:31 +02:00
## Version and Copyright
2011-09-18 14:17:25 +02:00
versionNumber = " 1.0 " + ' ' + gBUILD_VERSION
2009-07-17 11:10:31 +02:00
__version__ = " % prog Version " + versionNumber
2010-03-01 00:39:39 +01:00
__copyright__ = " Copyright (c) 2007 - 2010, Intel Corporation All rights reserved. "
2009-07-17 11:10:31 +02:00
## Tool entrance method
#
# This method mainly dispatch specific methods per the command line options.
# If no error found, return zero value so the caller of this tool can know
# if it's executed successfully or not.
#
# @retval 0 Tool was successful
# @retval 1 Tool failed
#
def main ( ) :
global Options
Options = myOptionParser ( )
global Workspace
Workspace = " "
ArchList = None
ReturnCode = 0
EdkLogger . Initialize ( )
try :
if Options . verbose != None :
EdkLogger . SetLevel ( EdkLogger . VERBOSE )
GenFdsGlobalVariable . VerboseMode = True
if Options . FixedAddress != None :
GenFdsGlobalVariable . FixedLoadAddress = True
if Options . quiet != None :
EdkLogger . SetLevel ( EdkLogger . QUIET )
if Options . debug != None :
EdkLogger . SetLevel ( Options . debug + 1 )
GenFdsGlobalVariable . DebugLevel = Options . debug
else :
EdkLogger . SetLevel ( EdkLogger . INFO )
if ( Options . Workspace == None ) :
2009-09-11 05:14:43 +02:00
EdkLogger . error ( " GenFds " , OPTION_MISSING , " WORKSPACE not defined " ,
2009-07-17 11:10:31 +02:00
ExtraData = " Please use ' -w ' switch to pass it or set the WORKSPACE environment variable. " )
elif not os . path . exists ( Options . Workspace ) :
2009-09-11 05:14:43 +02:00
EdkLogger . error ( " GenFds " , PARAMETER_INVALID , " WORKSPACE is invalid " ,
2009-07-17 11:10:31 +02:00
ExtraData = " Please use ' -w ' switch to pass it or set the WORKSPACE environment variable. " )
else :
Workspace = os . path . normcase ( Options . Workspace )
GenFdsGlobalVariable . WorkSpaceDir = Workspace
if ' EDK_SOURCE ' in os . environ . keys ( ) :
GenFdsGlobalVariable . EdkSourceDir = os . path . normcase ( os . environ [ ' EDK_SOURCE ' ] )
if ( Options . debug ) :
GenFdsGlobalVariable . VerboseLogger ( " Using Workspace: " + Workspace )
os . chdir ( GenFdsGlobalVariable . WorkSpaceDir )
if ( Options . filename ) :
FdfFilename = Options . filename
FdfFilename = GenFdsGlobalVariable . ReplaceWorkspaceMacro ( FdfFilename )
2010-03-01 00:39:39 +01:00
if FdfFilename [ 0 : 2 ] == ' .. ' :
FdfFilename = os . path . realpath ( FdfFilename )
if not os . path . isabs ( FdfFilename ) :
FdfFilename = os . path . join ( GenFdsGlobalVariable . WorkSpaceDir , FdfFilename )
if not os . path . exists ( FdfFilename ) :
EdkLogger . error ( " GenFds " , FILE_NOT_FOUND , ExtraData = FdfFilename )
if os . path . normcase ( FdfFilename ) . find ( Workspace ) != 0 :
EdkLogger . error ( " GenFds " , FILE_NOT_FOUND , " FdfFile doesn ' t exist in Workspace! " )
GenFdsGlobalVariable . FdfFile = FdfFilename
GenFdsGlobalVariable . FdfFileTimeStamp = os . path . getmtime ( FdfFilename )
2009-07-17 11:10:31 +02:00
else :
2009-09-11 05:14:43 +02:00
EdkLogger . error ( " GenFds " , OPTION_MISSING , " Missing FDF filename " )
2009-07-17 11:10:31 +02:00
if ( Options . BuildTarget ) :
GenFdsGlobalVariable . TargetName = Options . BuildTarget
else :
2009-09-11 05:14:43 +02:00
EdkLogger . error ( " GenFds " , OPTION_MISSING , " Missing build target " )
2009-07-17 11:10:31 +02:00
if ( Options . ToolChain ) :
GenFdsGlobalVariable . ToolChainTag = Options . ToolChain
else :
2009-09-11 05:14:43 +02:00
EdkLogger . error ( " GenFds " , OPTION_MISSING , " Missing tool chain tag " )
2009-07-17 11:10:31 +02:00
if ( Options . activePlatform ) :
ActivePlatform = Options . activePlatform
ActivePlatform = GenFdsGlobalVariable . ReplaceWorkspaceMacro ( ActivePlatform )
if ActivePlatform [ 0 : 2 ] == ' .. ' :
ActivePlatform = os . path . realpath ( ActivePlatform )
2010-03-01 00:39:39 +01:00
if not os . path . isabs ( ActivePlatform ) :
2009-07-17 11:10:31 +02:00
ActivePlatform = os . path . join ( GenFdsGlobalVariable . WorkSpaceDir , ActivePlatform )
if not os . path . exists ( ActivePlatform ) :
2009-09-11 05:14:43 +02:00
EdkLogger . error ( " GenFds " , FILE_NOT_FOUND , " ActivePlatform doesn ' t exist! " )
2009-07-17 11:10:31 +02:00
2010-03-01 00:39:39 +01:00
if os . path . normcase ( ActivePlatform ) . find ( Workspace ) != 0 :
2009-09-11 05:14:43 +02:00
EdkLogger . error ( " GenFds " , FILE_NOT_FOUND , " ActivePlatform doesn ' t exist in Workspace! " )
2009-07-17 11:10:31 +02:00
2010-03-01 00:39:39 +01:00
ActivePlatform = ActivePlatform [ len ( Workspace ) : ]
2009-07-17 11:10:31 +02:00
if len ( ActivePlatform ) > 0 :
if ActivePlatform [ 0 ] == ' \\ ' or ActivePlatform [ 0 ] == ' / ' :
ActivePlatform = ActivePlatform [ 1 : ]
else :
2009-09-11 05:14:43 +02:00
EdkLogger . error ( " GenFds " , FILE_NOT_FOUND , " ActivePlatform doesn ' t exist! " )
2010-03-01 00:39:39 +01:00
else :
2009-09-11 05:14:43 +02:00
EdkLogger . error ( " GenFds " , OPTION_MISSING , " Missing active platform " )
2009-07-17 11:10:31 +02:00
GenFdsGlobalVariable . ActivePlatform = PathClass ( NormPath ( ActivePlatform ) , Workspace )
BuildConfigurationFile = os . path . normpath ( os . path . join ( GenFdsGlobalVariable . WorkSpaceDir , " Conf/target.txt " ) )
if os . path . isfile ( BuildConfigurationFile ) == True :
TargetTxtClassObject . TargetTxtClassObject ( BuildConfigurationFile )
else :
2009-09-11 05:14:43 +02:00
EdkLogger . error ( " GenFds " , FILE_NOT_FOUND , ExtraData = BuildConfigurationFile )
2009-07-17 11:10:31 +02:00
if Options . Macros :
for Pair in Options . Macros :
Pair . strip ( ' " ' )
List = Pair . split ( ' = ' )
if len ( List ) == 2 :
if List [ 0 ] . strip ( ) == " EFI_SOURCE " :
GlobalData . gEfiSource = List [ 1 ] . strip ( )
2009-09-11 05:14:43 +02:00
continue
2009-07-17 11:10:31 +02:00
elif List [ 0 ] . strip ( ) == " EDK_SOURCE " :
GlobalData . gEdkSource = List [ 1 ] . strip ( )
2009-09-11 05:14:43 +02:00
continue
2009-07-17 11:10:31 +02:00
else :
GlobalData . gEdkGlobal [ List [ 0 ] . strip ( ) ] = List [ 1 ] . strip ( )
2009-09-11 05:14:43 +02:00
FdfParser . InputMacroDict [ List [ 0 ] . strip ( ) ] = List [ 1 ] . strip ( )
2009-07-17 11:10:31 +02:00
else :
2009-09-11 05:14:43 +02:00
FdfParser . InputMacroDict [ List [ 0 ] . strip ( ) ] = " "
2009-07-17 11:10:31 +02:00
""" call Workspace build create database """
os . environ [ " WORKSPACE " ] = Workspace
2010-09-06 03:58:00 +02:00
FdfParser . InputMacroDict [ " WORKSPACE " ] = Workspace
2009-09-11 05:14:43 +02:00
BuildWorkSpace = WorkspaceDatabase ( ' :memory: ' , FdfParser . InputMacroDict )
2009-07-17 11:10:31 +02:00
BuildWorkSpace . InitDatabase ( )
#
# Get files real name in workspace dir
#
GlobalData . gAllFiles = DirCache ( Workspace )
GlobalData . gWorkspace = Workspace
if ( Options . archList ) :
ArchList = Options . archList . split ( ' , ' )
else :
2009-09-11 05:14:43 +02:00
# EdkLogger.error("GenFds", OPTION_MISSING, "Missing build ARCH")
2009-07-17 11:10:31 +02:00
ArchList = BuildWorkSpace . BuildObject [ GenFdsGlobalVariable . ActivePlatform , ' COMMON ' ] . SupArchList
TargetArchList = set ( BuildWorkSpace . BuildObject [ GenFdsGlobalVariable . ActivePlatform , ' COMMON ' ] . SupArchList ) & set ( ArchList )
if len ( TargetArchList ) == 0 :
EdkLogger . error ( " GenFds " , GENFDS_ERROR , " Target ARCH %s not in platform supported ARCH %s " % ( str ( ArchList ) , str ( BuildWorkSpace . BuildObject [ GenFdsGlobalVariable . ActivePlatform , ' COMMON ' ] . SupArchList ) ) )
for Arch in ArchList :
GenFdsGlobalVariable . OutputDirFromDscDict [ Arch ] = NormPath ( BuildWorkSpace . BuildObject [ GenFdsGlobalVariable . ActivePlatform , Arch ] . OutputDirectory )
2010-03-01 00:39:39 +01:00
GenFdsGlobalVariable . PlatformName = BuildWorkSpace . BuildObject [ GenFdsGlobalVariable . ActivePlatform , Arch ] . PlatformName
2009-07-17 11:10:31 +02:00
if ( Options . outputDir ) :
OutputDirFromCommandLine = GenFdsGlobalVariable . ReplaceWorkspaceMacro ( Options . outputDir )
2010-03-04 06:29:52 +01:00
if not os . path . isabs ( OutputDirFromCommandLine ) :
OutputDirFromCommandLine = os . path . join ( GenFdsGlobalVariable . WorkSpaceDir , OutputDirFromCommandLine )
2009-07-17 11:10:31 +02:00
for Arch in ArchList :
GenFdsGlobalVariable . OutputDirDict [ Arch ] = OutputDirFromCommandLine
else :
for Arch in ArchList :
GenFdsGlobalVariable . OutputDirDict [ Arch ] = os . path . join ( GenFdsGlobalVariable . OutputDirFromDscDict [ Arch ] , GenFdsGlobalVariable . TargetName + ' _ ' + GenFdsGlobalVariable . ToolChainTag )
for Key in GenFdsGlobalVariable . OutputDirDict :
OutputDir = GenFdsGlobalVariable . OutputDirDict [ Key ]
if OutputDir [ 0 : 2 ] == ' .. ' :
OutputDir = os . path . realpath ( OutputDir )
if OutputDir [ 1 ] != ' : ' :
OutputDir = os . path . join ( GenFdsGlobalVariable . WorkSpaceDir , OutputDir )
if not os . path . exists ( OutputDir ) :
2009-09-11 05:14:43 +02:00
EdkLogger . error ( " GenFds " , FILE_NOT_FOUND , ExtraData = OutputDir )
2009-07-17 11:10:31 +02:00
GenFdsGlobalVariable . OutputDirDict [ Key ] = OutputDir
""" Parse Fdf file, has to place after build Workspace as FDF may contain macros from DSC file """
FdfParserObj = FdfParser . FdfParser ( FdfFilename )
FdfParserObj . ParseFile ( )
if FdfParserObj . CycleReferenceCheck ( ) :
2009-09-11 05:14:43 +02:00
EdkLogger . error ( " GenFds " , FORMAT_NOT_SUPPORTED , " Cycle Reference Detected in FDF file " )
2009-07-17 11:10:31 +02:00
if ( Options . uiFdName ) :
if Options . uiFdName . upper ( ) in FdfParserObj . Profile . FdDict . keys ( ) :
GenFds . OnlyGenerateThisFd = Options . uiFdName
else :
2009-09-11 05:14:43 +02:00
EdkLogger . error ( " GenFds " , OPTION_VALUE_INVALID ,
2009-07-17 11:10:31 +02:00
" No such an FD in FDF file: %s " % Options . uiFdName )
if ( Options . uiFvName ) :
if Options . uiFvName . upper ( ) in FdfParserObj . Profile . FvDict . keys ( ) :
GenFds . OnlyGenerateThisFv = Options . uiFvName
else :
2009-09-11 05:14:43 +02:00
EdkLogger . error ( " GenFds " , OPTION_VALUE_INVALID ,
2009-07-17 11:10:31 +02:00
" No such an FV in FDF file: %s " % Options . uiFvName )
2011-08-26 09:46:26 +02:00
if ( Options . uiCapName ) :
if Options . uiCapName . upper ( ) in FdfParserObj . Profile . CapsuleDict . keys ( ) :
GenFds . OnlyGenerateThisCap = Options . uiCapName
else :
EdkLogger . error ( " GenFds " , OPTION_VALUE_INVALID ,
" No such a Capsule in FDF file: %s " % Options . uiCapName )
2009-07-17 11:10:31 +02:00
""" Modify images from build output if the feature of loading driver at fixed address is on. """
if GenFdsGlobalVariable . FixedLoadAddress :
GenFds . PreprocessImage ( BuildWorkSpace , GenFdsGlobalVariable . ActivePlatform )
""" Call GenFds """
GenFds . GenFd ( ' ' , FdfParserObj , BuildWorkSpace , ArchList )
2010-03-01 00:39:39 +01:00
""" Generate GUID cross reference file """
GenFds . GenerateGuidXRefFile ( BuildWorkSpace , ArchList )
2009-07-17 11:10:31 +02:00
""" Display FV space info. """
GenFds . DisplayFvSpaceInfo ( FdfParserObj )
2010-03-01 00:39:39 +01:00
2009-07-17 11:10:31 +02:00
except FdfParser . Warning , X :
2009-09-11 05:14:43 +02:00
EdkLogger . error ( X . ToolName , FORMAT_INVALID , File = X . FileName , Line = X . LineNumber , ExtraData = X . Message , RaiseError = False )
ReturnCode = FORMAT_INVALID
2009-07-17 11:10:31 +02:00
except FatalError , X :
if Options . debug != None :
import traceback
EdkLogger . quiet ( traceback . format_exc ( ) )
ReturnCode = X . args [ 0 ]
except :
import traceback
EdkLogger . error (
" \n Python " ,
CODE_ERROR ,
" Tools code failure " ,
2010-03-19 07:55:07 +01:00
ExtraData = " Please send email to edk2-buildtools-devel@lists.sourceforge.net for help, attaching following call stack trace! \n " ,
2009-07-17 11:10:31 +02:00
RaiseError = False
)
EdkLogger . quiet ( traceback . format_exc ( ) )
ReturnCode = CODE_ERROR
return ReturnCode
gParamCheck = [ ]
def SingleCheckCallback ( option , opt_str , value , parser ) :
if option not in gParamCheck :
setattr ( parser . values , option . dest , value )
gParamCheck . append ( option )
else :
parser . error ( " Option %s only allows one instance in command line! " % option )
## Parse command line options
#
# Using standard Python module optparse to parse command line option of this tool.
#
# @retval Opt A optparse.Values object containing the parsed options
# @retval Args Target of build command
#
def myOptionParser ( ) :
usage = " % prog [options] -f input_file -a arch_list -b build_target -p active_platform -t tool_chain_tag -D \" MacroName [= MacroValue] \" "
Parser = OptionParser ( usage = usage , description = __copyright__ , version = " % prog " + str ( versionNumber ) )
Parser . add_option ( " -f " , " --file " , dest = " filename " , type = " string " , help = " Name of FDF file to convert " , action = " callback " , callback = SingleCheckCallback )
Parser . add_option ( " -a " , " --arch " , dest = " archList " , help = " comma separated list containing one or more of: IA32, X64, IPF, ARM or EBC which should be built, overrides target.txt?s TARGET_ARCH " )
Parser . add_option ( " -q " , " --quiet " , action = " store_true " , type = None , help = " Disable all messages except FATAL ERRORS. " )
Parser . add_option ( " -v " , " --verbose " , action = " store_true " , type = None , help = " Turn on verbose output with informational messages printed. " )
Parser . add_option ( " -d " , " --debug " , action = " store " , type = " int " , help = " Enable debug messages at specified level. " )
Parser . add_option ( " -p " , " --platform " , type = " string " , dest = " activePlatform " , help = " Set the ACTIVE_PLATFORM, overrides target.txt ACTIVE_PLATFORM setting. " ,
action = " callback " , callback = SingleCheckCallback )
Parser . add_option ( " -w " , " --workspace " , type = " string " , dest = " Workspace " , default = os . environ . get ( ' WORKSPACE ' ) , help = " Set the WORKSPACE " ,
action = " callback " , callback = SingleCheckCallback )
Parser . add_option ( " -o " , " --outputDir " , type = " string " , dest = " outputDir " , help = " Name of Build Output directory " ,
action = " callback " , callback = SingleCheckCallback )
Parser . add_option ( " -r " , " --rom_image " , dest = " uiFdName " , help = " Build the image using the [FD] section named by FdUiName. " )
2011-08-26 09:46:26 +02:00
Parser . add_option ( " -i " , " --FvImage " , dest = " uiFvName " , help = " Build the FV image using the [FV] section named by UiFvName " )
Parser . add_option ( " -C " , " --CapsuleImage " , dest = " uiCapName " , help = " Build the Capsule image using the [Capsule] section named by UiCapName " )
2011-09-18 14:17:25 +02:00
Parser . add_option ( " -b " , " --buildtarget " , type = " choice " , choices = [ ' DEBUG ' , ' RELEASE ' , ' NOOPT ' ] , dest = " BuildTarget " , help = " Build TARGET is one of list: DEBUG, RELEASE, NOOPT. " ,
2009-07-17 11:10:31 +02:00
action = " callback " , callback = SingleCheckCallback )
Parser . add_option ( " -t " , " --tagname " , type = " string " , dest = " ToolChain " , help = " Using the tools: TOOL_CHAIN_TAG name to build the platform. " ,
action = " callback " , callback = SingleCheckCallback )
Parser . add_option ( " -D " , " --define " , action = " append " , type = " string " , dest = " Macros " , help = " Macro: \" Name [= Value] \" . " )
Parser . add_option ( " -s " , " --specifyaddress " , dest = " FixedAddress " , action = " store_true " , type = None , help = " Specify driver load address. " )
( Options , args ) = Parser . parse_args ( )
return Options
## The class implementing the EDK2 flash image generation process
#
# This process includes:
# 1. Collect workspace information, includes platform and module information
# 2. Call methods of Fd class to generate FD
# 3. Call methods of Fv class to generate FV that not belong to FD
#
class GenFds :
FdfParsef = None
2009-09-11 05:14:43 +02:00
# FvName, FdName, CapName in FDF, Image file name
ImageBinDict = { }
2009-07-17 11:10:31 +02:00
OnlyGenerateThisFd = None
OnlyGenerateThisFv = None
2011-08-26 09:46:26 +02:00
OnlyGenerateThisCap = None
2009-07-17 11:10:31 +02:00
## GenFd()
#
# @param OutputDir Output directory
# @param FdfParser FDF contents parser
# @param Workspace The directory of workspace
# @param ArchList The Arch list of platform
#
def GenFd ( OutputDir , FdfParser , WorkSpace , ArchList ) :
GenFdsGlobalVariable . SetDir ( ' ' , FdfParser , WorkSpace , ArchList )
2009-09-11 05:14:43 +02:00
GenFdsGlobalVariable . VerboseLogger ( " Generate all Fd images and their required FV and Capsule images! " )
2011-08-26 09:46:26 +02:00
if GenFds . OnlyGenerateThisCap != None and GenFds . OnlyGenerateThisCap . upper ( ) in GenFdsGlobalVariable . FdfParser . Profile . CapsuleDict . keys ( ) :
CapsuleObj = GenFdsGlobalVariable . FdfParser . Profile . CapsuleDict . get ( GenFds . OnlyGenerateThisCap . upper ( ) )
if CapsuleObj != None :
CapsuleObj . GenCapsule ( )
return
2009-07-17 11:10:31 +02:00
if GenFds . OnlyGenerateThisFd != None and GenFds . OnlyGenerateThisFd . upper ( ) in GenFdsGlobalVariable . FdfParser . Profile . FdDict . keys ( ) :
FdObj = GenFdsGlobalVariable . FdfParser . Profile . FdDict . get ( GenFds . OnlyGenerateThisFd . upper ( ) )
if FdObj != None :
2009-09-11 05:14:43 +02:00
FdObj . GenFd ( )
2011-08-26 09:46:26 +02:00
return
elif GenFds . OnlyGenerateThisFd == None and GenFds . OnlyGenerateThisFv == None :
2009-07-17 11:10:31 +02:00
for FdName in GenFdsGlobalVariable . FdfParser . Profile . FdDict . keys ( ) :
FdObj = GenFdsGlobalVariable . FdfParser . Profile . FdDict [ FdName ]
2009-09-11 05:14:43 +02:00
FdObj . GenFd ( )
2009-07-17 11:10:31 +02:00
2009-09-11 05:14:43 +02:00
GenFdsGlobalVariable . VerboseLogger ( " \n Generate other FV images! " )
2009-07-17 11:10:31 +02:00
if GenFds . OnlyGenerateThisFv != None and GenFds . OnlyGenerateThisFv . upper ( ) in GenFdsGlobalVariable . FdfParser . Profile . FvDict . keys ( ) :
FvObj = GenFdsGlobalVariable . FdfParser . Profile . FvDict . get ( GenFds . OnlyGenerateThisFv . upper ( ) )
if FvObj != None :
Buffer = StringIO . StringIO ( )
2010-11-15 03:51:34 +01:00
FvObj . AddToBuffer ( Buffer )
2009-07-17 11:10:31 +02:00
Buffer . close ( )
return
2009-09-11 05:14:43 +02:00
elif GenFds . OnlyGenerateThisFv == None :
2009-07-17 11:10:31 +02:00
for FvName in GenFdsGlobalVariable . FdfParser . Profile . FvDict . keys ( ) :
Buffer = StringIO . StringIO ( ' ' )
FvObj = GenFdsGlobalVariable . FdfParser . Profile . FvDict [ FvName ]
2010-11-15 03:51:34 +01:00
FvObj . AddToBuffer ( Buffer )
2009-07-17 11:10:31 +02:00
Buffer . close ( )
2010-03-01 00:39:39 +01:00
2011-08-26 09:46:26 +02:00
if GenFds . OnlyGenerateThisFv == None and GenFds . OnlyGenerateThisFd == None and GenFds . OnlyGenerateThisCap == None :
2009-09-11 05:14:43 +02:00
if GenFdsGlobalVariable . FdfParser . Profile . CapsuleDict != { } :
GenFdsGlobalVariable . VerboseLogger ( " \n Generate other Capsule images! " )
for CapsuleName in GenFdsGlobalVariable . FdfParser . Profile . CapsuleDict . keys ( ) :
CapsuleObj = GenFdsGlobalVariable . FdfParser . Profile . CapsuleDict [ CapsuleName ]
CapsuleObj . GenCapsule ( )
2009-07-17 11:10:31 +02:00
if GenFdsGlobalVariable . FdfParser . Profile . OptRomDict != { } :
2009-09-11 05:14:43 +02:00
GenFdsGlobalVariable . VerboseLogger ( " \n Generate all Option ROM! " )
2009-07-17 11:10:31 +02:00
for DriverName in GenFdsGlobalVariable . FdfParser . Profile . OptRomDict . keys ( ) :
OptRomObj = GenFdsGlobalVariable . FdfParser . Profile . OptRomDict [ DriverName ]
OptRomObj . AddToBuffer ( None )
## GetFvBlockSize()
#
# @param FvObj Whose block size to get
# @retval int Block size value
#
def GetFvBlockSize ( FvObj ) :
2010-03-01 00:39:39 +01:00
DefaultBlockSize = 0x1
2009-07-17 11:10:31 +02:00
FdObj = None
if GenFds . OnlyGenerateThisFd != None and GenFds . OnlyGenerateThisFd . upper ( ) in GenFdsGlobalVariable . FdfParser . Profile . FdDict . keys ( ) :
FdObj = GenFdsGlobalVariable . FdfParser . Profile . FdDict [ GenFds . OnlyGenerateThisFd . upper ( ) ]
if FdObj == None :
for ElementFd in GenFdsGlobalVariable . FdfParser . Profile . FdDict . values ( ) :
for ElementRegion in ElementFd . RegionList :
if ElementRegion . RegionType == ' FV ' :
for ElementRegionData in ElementRegion . RegionDataList :
if ElementRegionData != None and ElementRegionData . upper ( ) == FvObj . UiFvName :
if FvObj . BlockSizeList != [ ] :
return FvObj . BlockSizeList [ 0 ] [ 0 ]
else :
return ElementRegion . BlockSizeOfRegion ( ElementFd . BlockSizeList )
if FvObj . BlockSizeList != [ ] :
return FvObj . BlockSizeList [ 0 ] [ 0 ]
return DefaultBlockSize
else :
for ElementRegion in FdObj . RegionList :
if ElementRegion . RegionType == ' FV ' :
for ElementRegionData in ElementRegion . RegionDataList :
if ElementRegionData != None and ElementRegionData . upper ( ) == FvObj . UiFvName :
if FvObj . BlockSizeList != [ ] :
return FvObj . BlockSizeList [ 0 ] [ 0 ]
else :
return ElementRegion . BlockSizeOfRegion ( ElementFd . BlockSizeList )
return DefaultBlockSize
## DisplayFvSpaceInfo()
#
# @param FvObj Whose block size to get
# @retval None
#
def DisplayFvSpaceInfo ( FdfParser ) :
FvSpaceInfoList = [ ]
MaxFvNameLength = 0
for FvName in FdfParser . Profile . FvDict :
if len ( FvName ) > MaxFvNameLength :
MaxFvNameLength = len ( FvName )
FvSpaceInfoFileName = os . path . join ( GenFdsGlobalVariable . FvDir , FvName . upper ( ) + ' .Fv.map ' )
if os . path . exists ( FvSpaceInfoFileName ) :
FileLinesList = linecache . getlines ( FvSpaceInfoFileName )
TotalFound = False
Total = ' '
UsedFound = False
Used = ' '
FreeFound = False
Free = ' '
for Line in FileLinesList :
NameValue = Line . split ( ' = ' )
if len ( NameValue ) == 2 :
if NameValue [ 0 ] . strip ( ) == ' EFI_FV_TOTAL_SIZE ' :
TotalFound = True
Total = NameValue [ 1 ] . strip ( )
if NameValue [ 0 ] . strip ( ) == ' EFI_FV_TAKEN_SIZE ' :
UsedFound = True
Used = NameValue [ 1 ] . strip ( )
if NameValue [ 0 ] . strip ( ) == ' EFI_FV_SPACE_SIZE ' :
FreeFound = True
Free = NameValue [ 1 ] . strip ( )
if TotalFound and UsedFound and FreeFound :
FvSpaceInfoList . append ( ( FvName , Total , Used , Free ) )
GenFdsGlobalVariable . InfLogger ( ' \n FV Space Information ' )
for FvSpaceInfo in FvSpaceInfoList :
Name = FvSpaceInfo [ 0 ]
TotalSizeValue = long ( FvSpaceInfo [ 1 ] , 0 )
UsedSizeValue = long ( FvSpaceInfo [ 2 ] , 0 )
FreeSizeValue = long ( FvSpaceInfo [ 3 ] , 0 )
2010-11-15 03:51:34 +01:00
if UsedSizeValue == TotalSizeValue :
Percentage = ' 100 '
else :
Percentage = str ( ( UsedSizeValue + 0.0 ) / TotalSizeValue ) [ 0 : 4 ] . lstrip ( ' 0. ' )
GenFdsGlobalVariable . InfLogger ( Name + ' ' + ' [ ' + Percentage + ' %F ull] ' + str ( TotalSizeValue ) + ' total, ' + str ( UsedSizeValue ) + ' used, ' + str ( FreeSizeValue ) + ' free ' )
2009-07-17 11:10:31 +02:00
## PreprocessImage()
#
# @param BuildDb Database from build meta data files
# @param DscFile modules from dsc file will be preprocessed
# @retval None
#
def PreprocessImage ( BuildDb , DscFile ) :
PcdDict = BuildDb . BuildObject [ DscFile , ' COMMON ' ] . Pcds
PcdValue = ' '
for Key in PcdDict :
PcdObj = PcdDict [ Key ]
if PcdObj . TokenCName == ' PcdBsBaseAddress ' :
PcdValue = PcdObj . DefaultValue
break
if PcdValue == ' ' :
return
Int64PcdValue = long ( PcdValue , 0 )
if Int64PcdValue == 0 or Int64PcdValue < - 1 :
return
TopAddress = 0
if Int64PcdValue > 0 :
TopAddress = Int64PcdValue
ModuleDict = BuildDb . BuildObject [ DscFile , ' COMMON ' ] . Modules
for Key in ModuleDict :
ModuleObj = BuildDb . BuildObject [ Key , ' COMMON ' ]
print ModuleObj . BaseName + ' ' + ModuleObj . ModuleType
2010-03-01 00:39:39 +01:00
def GenerateGuidXRefFile ( BuildDb , ArchList ) :
GuidXRefFileName = os . path . join ( GenFdsGlobalVariable . FvDir , " Guid.xref " )
2010-05-18 07:04:32 +02:00
GuidXRefFile = StringIO . StringIO ( ' ' )
2010-03-01 00:39:39 +01:00
for Arch in ArchList :
PlatformDataBase = BuildDb . BuildObject [ GenFdsGlobalVariable . ActivePlatform , Arch ]
for ModuleFile in PlatformDataBase . Modules :
Module = BuildDb . BuildObject [ ModuleFile , Arch ]
GuidXRefFile . write ( " %s %s \n " % ( Module . Guid , Module . BaseName ) )
2010-05-18 07:04:32 +02:00
SaveFileOnChange ( GuidXRefFileName , GuidXRefFile . getvalue ( ) , False )
2010-03-01 00:39:39 +01:00
GuidXRefFile . close ( )
2010-05-18 07:04:32 +02:00
GenFdsGlobalVariable . InfLogger ( " \n GUID cross reference file can be found at %s " % GuidXRefFileName )
2010-03-01 00:39:39 +01:00
2009-07-17 11:10:31 +02:00
##Define GenFd as static function
GenFd = staticmethod ( GenFd )
GetFvBlockSize = staticmethod ( GetFvBlockSize )
DisplayFvSpaceInfo = staticmethod ( DisplayFvSpaceInfo )
PreprocessImage = staticmethod ( PreprocessImage )
2010-03-01 00:39:39 +01:00
GenerateGuidXRefFile = staticmethod ( GenerateGuidXRefFile )
2009-07-17 11:10:31 +02:00
if __name__ == ' __main__ ' :
r = main ( )
## 0-127 is a safe return range, and 1 is a standard default error
if r < 0 or r > 127 : r = 1
sys . exit ( r )