+ ## 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
+ 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
+ payload = fmp.GenCapsuleSubItem()\r
+ FwMgrHdr.write(pack('=Q', PreSize))\r
+ PreSize += len(payload)\r
+ Content.write(payload)\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