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

## @file\r
# generate capsule\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
from GenFdsGlobalVariable import GenFdsGlobalVariable\r
from CommonDataClass.FdfClass import CapsuleClassObject\r
import Common.LongFilePathOs as os\r
import subprocess\r
import StringIO\r
from Common.Misc import SaveFileOnChange\r
from GenFds import GenFds\r
from Common.Misc import PackRegistryFormatGuid\r
import uuid\r
from struct import pack\r
from GenFds import FindExtendTool\r
from Common import EdkLogger\r
from Common.BuildToolError import *\r
\r
\r
T_CHAR_LF = '\n'\r
WIN_CERT_REVISION      = 0x0200\r
WIN_CERT_TYPE_EFI_GUID = 0x0EF1\r
EFI_CERT_TYPE_PKCS7_GUID = uuid.UUID('{4aafd29d-68df-49ee-8aa9-347d375665a7}')\r
EFI_CERT_TYPE_RSA2048_SHA256_GUID = uuid.UUID('{a7717414-c616-4977-9420-844712a735bf}')\r
\r
## create inf file describes what goes into capsule and call GenFv to generate capsule\r
#\r
#\r
class Capsule (CapsuleClassObject) :\r
    ## The constructor\r
    #\r
    #   @param  self        The object pointer\r
    #\r
    def __init__(self):\r
        CapsuleClassObject.__init__(self)\r
        # For GenFv\r
        self.BlockSize = None\r
        # For GenFv\r
        self.BlockNum = None\r
        self.CapsuleName = None\r
\r
    ## Generate FMP capsule\r
    #\r
    #   @retval string      Generated Capsule file path\r
    #\r
    def GenFmpCapsule(self):\r
        #\r
        # Generate capsule header\r
        # typedef struct {\r
        #     EFI_GUID          CapsuleGuid;\r
        #     UINT32            HeaderSize;\r
        #     UINT32            Flags;\r
        #     UINT32            CapsuleImageSize;\r
        # } EFI_CAPSULE_HEADER;\r
        #\r
        Header = StringIO.StringIO()\r
        #\r
        # Use FMP capsule GUID: 6DCBD5ED-E82D-4C44-BDA1-7194199AD92A\r
        #\r
        Header.write(PackRegistryFormatGuid('6DCBD5ED-E82D-4C44-BDA1-7194199AD92A'))\r
        HdrSize = 0\r
        if 'CAPSULE_HEADER_SIZE' in self.TokensDict:\r
            Header.write(pack('=I', int(self.TokensDict['CAPSULE_HEADER_SIZE'], 16)))\r
            HdrSize = int(self.TokensDict['CAPSULE_HEADER_SIZE'], 16)\r
        else:\r
            Header.write(pack('=I', 0x20))\r
            HdrSize = 0x20\r
        Flags = 0\r
        if 'CAPSULE_FLAGS' in self.TokensDict:\r
            for flag in self.TokensDict['CAPSULE_FLAGS'].split(','):\r
                flag = flag.strip()\r
                if flag == 'PopulateSystemTable':\r
                    Flags |= 0x00010000 | 0x00020000\r
                elif flag == 'PersistAcrossReset':\r
                    Flags |= 0x00010000\r
                elif flag == 'InitiateReset':\r
                    Flags |= 0x00040000\r
        Header.write(pack('=I', Flags))\r
        #\r
        # typedef struct {\r
        #     UINT32 Version;\r
        #     UINT16 EmbeddedDriverCount;\r
        #     UINT16 PayloadItemCount;\r
        #     // UINT64 ItemOffsetList[];\r
        # } EFI_FIRMWARE_MANAGEMENT_CAPSULE_HEADER;\r
        #\r
        FwMgrHdr = StringIO.StringIO()\r
        if 'CAPSULE_HEADER_INIT_VERSION' in self.TokensDict:\r
            FwMgrHdr.write(pack('=I', int(self.TokensDict['CAPSULE_HEADER_INIT_VERSION'], 16)))\r
        else:\r
            FwMgrHdr.write(pack('=I', 0x00000001))\r
        FwMgrHdr.write(pack('=HH', len(self.CapsuleDataList), len(self.FmpPayloadList)))\r
        FwMgrHdrSize = 4+2+2+8*(len(self.CapsuleDataList)+len(self.FmpPayloadList))\r
\r
        #\r
        # typedef struct _WIN_CERTIFICATE {\r
        #   UINT32 dwLength;\r
        #   UINT16 wRevision;\r
        #   UINT16 wCertificateType;\r
        # //UINT8 bCertificate[ANYSIZE_ARRAY];\r
        # } WIN_CERTIFICATE;\r
        #\r
        # typedef struct _WIN_CERTIFICATE_UEFI_GUID {\r
        #   WIN_CERTIFICATE Hdr;\r
        #   EFI_GUID        CertType;\r
        # //UINT8 CertData[ANYSIZE_ARRAY];\r
        # } WIN_CERTIFICATE_UEFI_GUID;\r
        #\r
        # typedef struct {\r
        #   UINT64                    MonotonicCount;\r
        #   WIN_CERTIFICATE_UEFI_GUID AuthInfo;\r
        # } EFI_FIRMWARE_IMAGE_AUTHENTICATION;\r
        #\r
        # typedef struct _EFI_CERT_BLOCK_RSA_2048_SHA256 {\r
        #   EFI_GUID HashType;\r
        #   UINT8 PublicKey[256];\r
        #   UINT8 Signature[256];\r
        # } EFI_CERT_BLOCK_RSA_2048_SHA256;\r
        #\r
\r
        PreSize = FwMgrHdrSize\r
        Content = StringIO.StringIO()\r
        for driver in self.CapsuleDataList:\r
            FileName = driver.GenCapsuleSubItem()\r
            FwMgrHdr.write(pack('=Q', PreSize))\r
            PreSize += os.path.getsize(FileName)\r
            File = open(FileName, 'rb')\r
            Content.write(File.read())\r
            File.close()\r
        for fmp in self.FmpPayloadList:\r
            if fmp.Certificate_Guid:\r
                ExternalTool, ExternalOption = FindExtendTool([], GenFdsGlobalVariable.ArchList, fmp.Certificate_Guid)\r
                CmdOption = ''\r
                CapInputFile = fmp.ImageFile\r
                if not os.path.isabs(fmp.ImageFile):\r
                    CapInputFile = os.path.join(GenFdsGlobalVariable.WorkSpaceDir, fmp.ImageFile)\r
                CapOutputTmp = os.path.join(GenFdsGlobalVariable.FvDir, self.UiCapsuleName) + '.tmp'\r
                if ExternalTool == None:\r
                    EdkLogger.error("GenFds", GENFDS_ERROR, "No tool found with GUID %s" % fmp.Certificate_Guid)\r
                else:\r
                    CmdOption += ExternalTool\r
                if ExternalOption:\r
                    CmdOption = CmdOption + ' ' + ExternalOption\r
                CmdOption += ' -e ' + ' --monotonic-count ' + str(fmp.MonotonicCount) + ' -o ' + CapOutputTmp + ' ' + CapInputFile\r
                CmdList = CmdOption.split()\r
                GenFdsGlobalVariable.CallExternalTool(CmdList, "Failed to generate FMP auth capsule")\r
                if uuid.UUID(fmp.Certificate_Guid) == EFI_CERT_TYPE_PKCS7_GUID:\r
                    dwLength = 4 + 2 + 2 + 16 + os.path.getsize(CapOutputTmp) - os.path.getsize(CapInputFile)\r
                else:\r
                    dwLength = 4 + 2 + 2 + 16 + 16 + 256 + 256\r
                fmp.ImageFile = CapOutputTmp\r
                AuthData = [fmp.MonotonicCount, dwLength, WIN_CERT_REVISION, WIN_CERT_TYPE_EFI_GUID, fmp.Certificate_Guid]\r
                Buffer = fmp.GenCapsuleSubItem(AuthData)\r
            else:\r
                Buffer = fmp.GenCapsuleSubItem()\r
            FwMgrHdr.write(pack('=Q', PreSize))\r
            PreSize += len(Buffer)\r
            Content.write(Buffer)\r
        BodySize = len(FwMgrHdr.getvalue()) + len(Content.getvalue())\r
        Header.write(pack('=I', HdrSize + BodySize))\r
        #\r
        # The real capsule header structure is 28 bytes\r
        #\r
        Header.write('\x00'*(HdrSize-28))\r
        Header.write(FwMgrHdr.getvalue())\r
        Header.write(Content.getvalue())\r
        #\r
        # Generate FMP capsule file\r
        #\r
        CapOutputFile = os.path.join(GenFdsGlobalVariable.FvDir, self.UiCapsuleName) + '.Cap'\r
        SaveFileOnChange(CapOutputFile, Header.getvalue(), True)\r
        return CapOutputFile\r
\r
    ## Generate capsule\r
    #\r
    #   @param  self        The object pointer\r
    #   @retval string      Generated Capsule file path\r
    #\r
    def GenCapsule(self):\r
        if self.UiCapsuleName.upper() + 'cap' in GenFds.ImageBinDict.keys():\r
            return GenFds.ImageBinDict[self.UiCapsuleName.upper() + 'cap']\r
\r
        GenFdsGlobalVariable.InfLogger( "\nGenerate %s Capsule" %self.UiCapsuleName)\r
        if ('CAPSULE_GUID' in self.TokensDict and \r
            uuid.UUID(self.TokensDict['CAPSULE_GUID']) == uuid.UUID('6DCBD5ED-E82D-4C44-BDA1-7194199AD92A')):\r
            return self.GenFmpCapsule()\r
\r
        CapInfFile = self.GenCapInf()\r
        CapInfFile.writelines("[files]" + T_CHAR_LF)\r
        CapFileList = []\r
        for CapsuleDataObj in self.CapsuleDataList :\r
            CapsuleDataObj.CapsuleName = self.CapsuleName\r
            FileName = CapsuleDataObj.GenCapsuleSubItem()\r
            CapsuleDataObj.CapsuleName = None\r
            CapFileList.append(FileName)\r
            CapInfFile.writelines("EFI_FILE_NAME = " + \\r
                                   FileName      + \\r
                                   T_CHAR_LF)\r
        SaveFileOnChange(self.CapInfFileName, CapInfFile.getvalue(), False)\r
        CapInfFile.close()\r
        #\r
        # Call GenFv tool to generate capsule\r
        #\r
        CapOutputFile = os.path.join(GenFdsGlobalVariable.FvDir, self.UiCapsuleName)\r
        CapOutputFile = CapOutputFile + '.Cap'\r
        GenFdsGlobalVariable.GenerateFirmwareVolume(\r
                                CapOutputFile,\r
                                [self.CapInfFileName],\r
                                Capsule=True,\r
                                FfsList=CapFileList\r
                                )\r
\r
        GenFdsGlobalVariable.VerboseLogger( "\nGenerate %s Capsule Successfully" %self.UiCapsuleName)\r
        GenFdsGlobalVariable.SharpCounter = 0\r
        GenFds.ImageBinDict[self.UiCapsuleName.upper() + 'cap'] = CapOutputFile\r
        return CapOutputFile\r
\r
    ## Generate inf file for capsule\r
    #\r
    #   @param  self        The object pointer\r
    #   @retval file        inf file object\r
    #\r
    def GenCapInf(self):\r
        self.CapInfFileName = os.path.join(GenFdsGlobalVariable.FvDir,\r
                                   self.UiCapsuleName +  "_Cap" + '.inf')\r
        CapInfFile = StringIO.StringIO() #open (self.CapInfFileName , 'w+')\r
\r
        CapInfFile.writelines("[options]" + T_CHAR_LF)\r
\r
        for Item in self.TokensDict.keys():\r
            CapInfFile.writelines("EFI_"                    + \\r
                                  Item                      + \\r
                                  ' = '                     + \\r
                                  self.TokensDict.get(Item) + \\r
                                  T_CHAR_LF)\r
\r
        return CapInfFile\r
Commit	Line	Data
30fdf114 LG	1	## @file\r
	2	# generate capsule\r
	3	#\r
91ae2988	4	# Copyright (c) 2007 - 2016, Intel Corporation. All rights reserved.<BR>\r
30fdf114	5	#\r
40d841f6	6	# This program and the accompanying materials\r
30fdf114 LG	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	from GenFdsGlobalVariable import GenFdsGlobalVariable\r
	19	from CommonDataClass.FdfClass import CapsuleClassObject\r
1be2ed90	20	import Common.LongFilePathOs as os\r
30fdf114 LG	21	import subprocess\r
	22	import StringIO\r
	23	from Common.Misc import SaveFileOnChange\r
fd171542	24	from GenFds import GenFds\r
a3251d84 YL	25	from Common.Misc import PackRegistryFormatGuid\r
	26	import uuid\r
	27	from struct import pack\r
91ae2988 YZ	28	from GenFds import FindExtendTool\r
	29	from Common import EdkLogger\r
	30	from Common.BuildToolError import *\r
30fdf114 LG	31	\r
	32	\r
	33	T_CHAR_LF = '\n'\r
91ae2988 YZ	34	WIN_CERT_REVISION = 0x0200\r
	35	WIN_CERT_TYPE_EFI_GUID = 0x0EF1\r
	36	EFI_CERT_TYPE_PKCS7_GUID = uuid.UUID('{4aafd29d-68df-49ee-8aa9-347d375665a7}')\r
	37	EFI_CERT_TYPE_RSA2048_SHA256_GUID = uuid.UUID('{a7717414-c616-4977-9420-844712a735bf}')\r
30fdf114 LG	38	\r
	39	## create inf file describes what goes into capsule and call GenFv to generate capsule\r
	40	#\r
	41	#\r
	42	class Capsule (CapsuleClassObject) :\r
	43	## The constructor\r
	44	#\r
	45	# @param self The object pointer\r
	46	#\r
	47	def __init__(self):\r
	48	CapsuleClassObject.__init__(self)\r
	49	# For GenFv\r
	50	self.BlockSize = None\r
	51	# For GenFv\r
	52	self.BlockNum = None\r
fd171542	53	self.CapsuleName = None\r
30fdf114	54	\r
a3251d84 YL	55	## Generate FMP capsule\r
	56	#\r
	57	# @retval string Generated Capsule file path\r
	58	#\r
	59	def GenFmpCapsule(self):\r
	60	#\r
	61	# Generate capsule header\r
	62	# typedef struct {\r
	63	# EFI_GUID CapsuleGuid;\r
	64	# UINT32 HeaderSize;\r
	65	# UINT32 Flags;\r
	66	# UINT32 CapsuleImageSize;\r
	67	# } EFI_CAPSULE_HEADER;\r
	68	#\r
	69	Header = StringIO.StringIO()\r
	70	#\r
	71	# Use FMP capsule GUID: 6DCBD5ED-E82D-4C44-BDA1-7194199AD92A\r
	72	#\r
	73	Header.write(PackRegistryFormatGuid('6DCBD5ED-E82D-4C44-BDA1-7194199AD92A'))\r
	74	HdrSize = 0\r
	75	if 'CAPSULE_HEADER_SIZE' in self.TokensDict:\r
	76	Header.write(pack('=I', int(self.TokensDict['CAPSULE_HEADER_SIZE'], 16)))\r
	77	HdrSize = int(self.TokensDict['CAPSULE_HEADER_SIZE'], 16)\r
	78	else:\r
	79	Header.write(pack('=I', 0x20))\r
	80	HdrSize = 0x20\r
	81	Flags = 0\r
	82	if 'CAPSULE_FLAGS' in self.TokensDict:\r
	83	for flag in self.TokensDict['CAPSULE_FLAGS'].split(','):\r
	84	flag = flag.strip()\r
	85	if flag == 'PopulateSystemTable':\r
	86	Flags \|= 0x00010000 \| 0x00020000\r
	87	elif flag == 'PersistAcrossReset':\r
	88	Flags \|= 0x00010000\r
	89	elif flag == 'InitiateReset':\r
	90	Flags \|= 0x00040000\r
	91	Header.write(pack('=I', Flags))\r
	92	#\r
	93	# typedef struct {\r
	94	# UINT32 Version;\r
	95	# UINT16 EmbeddedDriverCount;\r
	96	# UINT16 PayloadItemCount;\r
	97	# // UINT64 ItemOffsetList[];\r
	98	# } EFI_FIRMWARE_MANAGEMENT_CAPSULE_HEADER;\r
	99	#\r
	100	FwMgrHdr = StringIO.StringIO()\r
	101	if 'CAPSULE_HEADER_INIT_VERSION' in self.TokensDict:\r
	102	FwMgrHdr.write(pack('=I', int(self.TokensDict['CAPSULE_HEADER_INIT_VERSION'], 16)))\r
	103	else:\r
	104	FwMgrHdr.write(pack('=I', 0x00000001))\r
	105	FwMgrHdr.write(pack('=HH', len(self.CapsuleDataList), len(self.FmpPayloadList)))\r
	106	FwMgrHdrSize = 4+2+2+8*(len(self.CapsuleDataList)+len(self.FmpPayloadList))\r
	107	\r
91ae2988 YZ	108	#\r
	109	# typedef struct _WIN_CERTIFICATE {\r
	110	# UINT32 dwLength;\r
	111	# UINT16 wRevision;\r
	112	# UINT16 wCertificateType;\r
	113	# //UINT8 bCertificate[ANYSIZE_ARRAY];\r
	114	# } WIN_CERTIFICATE;\r
	115	#\r
	116	# typedef struct _WIN_CERTIFICATE_UEFI_GUID {\r
	117	# WIN_CERTIFICATE Hdr;\r
	118	# EFI_GUID CertType;\r
	119	# //UINT8 CertData[ANYSIZE_ARRAY];\r
	120	# } WIN_CERTIFICATE_UEFI_GUID;\r
	121	#\r
	122	# typedef struct {\r
	123	# UINT64 MonotonicCount;\r
	124	# WIN_CERTIFICATE_UEFI_GUID AuthInfo;\r
	125	# } EFI_FIRMWARE_IMAGE_AUTHENTICATION;\r
	126	#\r
	127	# typedef struct _EFI_CERT_BLOCK_RSA_2048_SHA256 {\r
	128	# EFI_GUID HashType;\r
	129	# UINT8 PublicKey[256];\r
	130	# UINT8 Signature[256];\r
	131	# } EFI_CERT_BLOCK_RSA_2048_SHA256;\r
	132	#\r
	133	\r
a3251d84 YL	134	PreSize = FwMgrHdrSize\r
	135	Content = StringIO.StringIO()\r
	136	for driver in self.CapsuleDataList:\r
	137	FileName = driver.GenCapsuleSubItem()\r
	138	FwMgrHdr.write(pack('=Q', PreSize))\r
	139	PreSize += os.path.getsize(FileName)\r
	140	File = open(FileName, 'rb')\r
	141	Content.write(File.read())\r
	142	File.close()\r
	143	for fmp in self.FmpPayloadList:\r
91ae2988 YZ	144	if fmp.Certificate_Guid:\r
	145	ExternalTool, ExternalOption = FindExtendTool([], GenFdsGlobalVariable.ArchList, fmp.Certificate_Guid)\r
	146	CmdOption = ''\r
	147	CapInputFile = fmp.ImageFile\r
	148	if not os.path.isabs(fmp.ImageFile):\r
	149	CapInputFile = os.path.join(GenFdsGlobalVariable.WorkSpaceDir, fmp.ImageFile)\r
	150	CapOutputTmp = os.path.join(GenFdsGlobalVariable.FvDir, self.UiCapsuleName) + '.tmp'\r
	151	if ExternalTool == None:\r
	152	EdkLogger.error("GenFds", GENFDS_ERROR, "No tool found with GUID %s" % fmp.Certificate_Guid)\r
	153	else:\r
	154	CmdOption += ExternalTool\r
	155	if ExternalOption:\r
	156	CmdOption = CmdOption + ' ' + ExternalOption\r
	157	CmdOption += ' -e ' + ' --monotonic-count ' + str(fmp.MonotonicCount) + ' -o ' + CapOutputTmp + ' ' + CapInputFile\r
	158	CmdList = CmdOption.split()\r
	159	GenFdsGlobalVariable.CallExternalTool(CmdList, "Failed to generate FMP auth capsule")\r
	160	if uuid.UUID(fmp.Certificate_Guid) == EFI_CERT_TYPE_PKCS7_GUID:\r
	161	dwLength = 4 + 2 + 2 + 16 + os.path.getsize(CapOutputTmp) - os.path.getsize(CapInputFile)\r
	162	else:\r
	163	dwLength = 4 + 2 + 2 + 16 + 16 + 256 + 256\r
5f53a7aa YZ	164	fmp.ImageFile = CapOutputTmp\r
	165	AuthData = [fmp.MonotonicCount, dwLength, WIN_CERT_REVISION, WIN_CERT_TYPE_EFI_GUID, fmp.Certificate_Guid]\r
	166	Buffer = fmp.GenCapsuleSubItem(AuthData)\r
91ae2988	167	else:\r
5f53a7aa YZ	168	Buffer = fmp.GenCapsuleSubItem()\r
	169	FwMgrHdr.write(pack('=Q', PreSize))\r
	170	PreSize += len(Buffer)\r
	171	Content.write(Buffer)\r
a3251d84 YL	172	BodySize = len(FwMgrHdr.getvalue()) + len(Content.getvalue())\r
	173	Header.write(pack('=I', HdrSize + BodySize))\r
	174	#\r
	175	# The real capsule header structure is 28 bytes\r
	176	#\r
	177	Header.write('\x00'*(HdrSize-28))\r
	178	Header.write(FwMgrHdr.getvalue())\r
	179	Header.write(Content.getvalue())\r
	180	#\r
	181	# Generate FMP capsule file\r
	182	#\r
	183	CapOutputFile = os.path.join(GenFdsGlobalVariable.FvDir, self.UiCapsuleName) + '.Cap'\r
	184	SaveFileOnChange(CapOutputFile, Header.getvalue(), True)\r
	185	return CapOutputFile\r
	186	\r
30fdf114 LG	187	## Generate capsule\r
	188	#\r
	189	# @param self The object pointer\r
fd171542	190	# @retval string Generated Capsule file path\r
30fdf114 LG	191	#\r
30fdf114 LG	192	def GenCapsule(self):\r
fd171542	193	if self.UiCapsuleName.upper() + 'cap' in GenFds.ImageBinDict.keys():\r
	194	return GenFds.ImageBinDict[self.UiCapsuleName.upper() + 'cap']\r
	195	\r
	196	GenFdsGlobalVariable.InfLogger( "\nGenerate %s Capsule" %self.UiCapsuleName)\r
a3251d84 YL	197	if ('CAPSULE_GUID' in self.TokensDict and \r
	198	uuid.UUID(self.TokensDict['CAPSULE_GUID']) == uuid.UUID('6DCBD5ED-E82D-4C44-BDA1-7194199AD92A')):\r
	199	return self.GenFmpCapsule()\r
	200	\r
30fdf114 LG	201	CapInfFile = self.GenCapInf()\r
30fdf114 LG	202	CapInfFile.writelines("[files]" + T_CHAR_LF)\r
fd171542	203	CapFileList = []\r
30fdf114	204	for CapsuleDataObj in self.CapsuleDataList :\r
fd171542	205	CapsuleDataObj.CapsuleName = self.CapsuleName\r
30fdf114	206	FileName = CapsuleDataObj.GenCapsuleSubItem()\r
fd171542	207	CapsuleDataObj.CapsuleName = None\r
fd171542	208	CapFileList.append(FileName)\r
30fdf114 LG	209	CapInfFile.writelines("EFI_FILE_NAME = " + \\r
	210	FileName + \\r
	211	T_CHAR_LF)\r
	212	SaveFileOnChange(self.CapInfFileName, CapInfFile.getvalue(), False)\r
	213	CapInfFile.close()\r
	214	#\r
	215	# Call GenFv tool to generate capsule\r
	216	#\r
	217	CapOutputFile = os.path.join(GenFdsGlobalVariable.FvDir, self.UiCapsuleName)\r
	218	CapOutputFile = CapOutputFile + '.Cap'\r
	219	GenFdsGlobalVariable.GenerateFirmwareVolume(\r
	220	CapOutputFile,\r
	221	[self.CapInfFileName],\r
fd171542	222	Capsule=True,\r
fd171542	223	FfsList=CapFileList\r
30fdf114	224	)\r
fd171542	225	\r
fd171542	226	GenFdsGlobalVariable.VerboseLogger( "\nGenerate %s Capsule Successfully" %self.UiCapsuleName)\r
30fdf114	227	GenFdsGlobalVariable.SharpCounter = 0\r
fd171542	228	GenFds.ImageBinDict[self.UiCapsuleName.upper() + 'cap'] = CapOutputFile\r
fd171542	229	return CapOutputFile\r
30fdf114 LG	230	\r
	231	## Generate inf file for capsule\r
	232	#\r
	233	# @param self The object pointer\r
	234	# @retval file inf file object\r
	235	#\r
	236	def GenCapInf(self):\r
	237	self.CapInfFileName = os.path.join(GenFdsGlobalVariable.FvDir,\r
	238	self.UiCapsuleName + "_Cap" + '.inf')\r
	239	CapInfFile = StringIO.StringIO() #open (self.CapInfFileName , 'w+')\r
	240	\r
	241	CapInfFile.writelines("[options]" + T_CHAR_LF)\r
	242	\r
	243	for Item in self.TokensDict.keys():\r
	244	CapInfFile.writelines("EFI_" + \\r
	245	Item + \\r
	246	' = ' + \\r
	247	self.TokensDict.get(Item) + \\r
	248	T_CHAR_LF)\r
	249	\r
	250	return CapInfFile\r