mirror of
				https://github.com/acidanthera/audk.git
				synced 2025-10-30 18:53:45 +01:00 
			
		
		
		
	git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10607 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			119 lines
		
	
	
		
			5.0 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			119 lines
		
	
	
		
			5.0 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| ## @file
 | |
| # process data section generation
 | |
| #
 | |
| #  Copyright (c) 2007, 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 Section
 | |
| from GenFdsGlobalVariable import GenFdsGlobalVariable
 | |
| import subprocess
 | |
| from Ffs import Ffs
 | |
| import os
 | |
| from CommonDataClass.FdfClass import DataSectionClassObject
 | |
| from Common.Misc import PeImageClass
 | |
| import shutil
 | |
| 
 | |
| ## generate data section
 | |
| #
 | |
| #
 | |
| class DataSection (DataSectionClassObject):
 | |
|     ## The constructor
 | |
|     #
 | |
|     #   @param  self        The object pointer
 | |
|     #
 | |
|     def __init__(self):
 | |
|         DataSectionClassObject.__init__(self)
 | |
| 
 | |
|     ## GenSection() method
 | |
|     #
 | |
|     #   Generate compressed section
 | |
|     #
 | |
|     #   @param  self        The object pointer
 | |
|     #   @param  OutputPath  Where to place output file
 | |
|     #   @param  ModuleName  Which module this section belongs to
 | |
|     #   @param  SecNum      Index of section
 | |
|     #   @param  KeyStringList  Filter for inputs of section generation
 | |
|     #   @param  FfsInf      FfsInfStatement object that contains this section data
 | |
|     #   @param  Dict        dictionary contains macro and its value
 | |
|     #   @retval tuple       (Generated file name list, section alignment)
 | |
|     #
 | |
|     def GenSection(self, OutputPath, ModuleName, SecNum, keyStringList, FfsFile = None, Dict = {}):
 | |
|         #
 | |
|         # Prepare the parameter of GenSection
 | |
|         #
 | |
|         if FfsFile != None:
 | |
|             self.SectFileName = GenFdsGlobalVariable.ReplaceWorkspaceMacro(self.SectFileName)
 | |
|             self.SectFileName = GenFdsGlobalVariable.MacroExtend(self.SectFileName, Dict, FfsFile.CurrentArch)
 | |
|         else:
 | |
|             self.SectFileName = GenFdsGlobalVariable.ReplaceWorkspaceMacro(self.SectFileName)
 | |
|             self.SectFileName = GenFdsGlobalVariable.MacroExtend(self.SectFileName, Dict)
 | |
| 
 | |
|         """Check Section file exist or not !"""
 | |
| 
 | |
|         if not os.path.exists(self.SectFileName):
 | |
|             self.SectFileName = os.path.join (GenFdsGlobalVariable.WorkSpaceDir,
 | |
|                                               self.SectFileName)
 | |
| 
 | |
|         """Copy Map file to Ffs output"""
 | |
|         Filename = GenFdsGlobalVariable.MacroExtend(self.SectFileName)
 | |
|         if Filename[(len(Filename)-4):] == '.efi':
 | |
|             MapFile = Filename.replace('.efi', '.map')
 | |
|             if os.path.exists(MapFile):
 | |
|                 CopyMapFile = os.path.join(OutputPath, ModuleName + '.map')
 | |
|                 if not os.path.exists(CopyMapFile) or \
 | |
|                     (os.path.getmtime(MapFile) > os.path.getmtime(CopyMapFile)):
 | |
|                     shutil.copyfile(MapFile, CopyMapFile)
 | |
| 
 | |
|         #Get PE Section alignment when align is set to AUTO
 | |
|         if self.Alignment == 'Auto' and self.SecType in ('TE', 'PE32'):
 | |
|             ImageObj = PeImageClass (Filename)
 | |
|             if ImageObj.SectionAlignment < 0x400:
 | |
|                 self.Alignment = str (ImageObj.SectionAlignment)
 | |
|             else:
 | |
|                 self.Alignment = str (ImageObj.SectionAlignment / 0x400) + 'K'
 | |
| 
 | |
|         NoStrip = True
 | |
|         if self.SecType in ('TE', 'PE32'):
 | |
|             if self.KeepReloc != None:
 | |
|                 NoStrip = self.KeepReloc
 | |
| 
 | |
|         if not NoStrip:
 | |
|             FileBeforeStrip = os.path.join(OutputPath, ModuleName + '.efi')
 | |
|             if not os.path.exists(FileBeforeStrip) or \
 | |
|                 (os.path.getmtime(self.SectFileName) > os.path.getmtime(FileBeforeStrip)):
 | |
|                 shutil.copyfile(self.SectFileName, FileBeforeStrip)
 | |
|             StrippedFile = os.path.join(OutputPath, ModuleName + '.stripped')
 | |
|             GenFdsGlobalVariable.GenerateFirmwareImage(
 | |
|                                     StrippedFile,
 | |
|                                     [GenFdsGlobalVariable.MacroExtend(self.SectFileName, Dict)],
 | |
|                                     Strip=True
 | |
|                                     )
 | |
|             self.SectFileName = StrippedFile
 | |
| 
 | |
|         if self.SecType == 'TE':
 | |
|             TeFile = os.path.join( OutputPath, ModuleName + 'Te.raw')
 | |
|             GenFdsGlobalVariable.GenerateFirmwareImage(
 | |
|                                     TeFile,
 | |
|                                     [GenFdsGlobalVariable.MacroExtend(self.SectFileName, Dict)],
 | |
|                                     Type='te'
 | |
|                                     )
 | |
|             self.SectFileName = TeFile
 | |
| 
 | |
|         OutputFile = os.path.join (OutputPath, ModuleName + 'SEC' + SecNum + Ffs.SectionSuffix.get(self.SecType))
 | |
|         OutputFile = os.path.normpath(OutputFile)
 | |
| 
 | |
|         GenFdsGlobalVariable.GenerateSection(OutputFile, [self.SectFileName], Section.Section.SectionType.get(self.SecType))
 | |
|         FileList = [OutputFile]
 | |
|         return FileList, self.Alignment
 |