## | |
# UefiBuild Plugin that supports Window Capsule files based on the | |
# Windows Firmware Update Platform spec. | |
# Creates INF, Cat, and then signs it | |
# | |
# To install run pip install --upgrade edk2-pytool-library | |
# edk2-pytool-library-0.9.1 is required. | |
# | |
# Copyright (c) Microsoft Corporation. All rights reserved. | |
# Copyright (c) 2019, Intel Corporation. All rights reserved.<BR> | |
# SPDX-License-Identifier: BSD-2-Clause-Patent | |
## | |
import sys | |
import re | |
import datetime | |
import os | |
import logging | |
from edk2toollib.windows.capsule.cat_generator import CatGenerator | |
from edk2toollib.windows.capsule.inf_generator import InfGenerator | |
from edk2toollib.utility_functions import CatalogSignWithSignTool | |
from edk2toollib.windows.locate_tools import FindToolInWinSdk | |
class WindowsCapsuleSupportHelper(object): | |
def RegisterHelpers(self, obj): | |
fp = os.path.abspath(__file__) | |
obj.Register("PackageWindowsCapsuleFiles", WindowsCapsuleSupportHelper.PackageWindowsCapsuleFiles, fp) | |
@staticmethod | |
def PackageWindowsCapsuleFiles(OutputFolder, ProductName, ProductFmpGuid, CapsuleVersion_DotString, | |
CapsuleVersion_HexString, ProductFwProvider, ProductFwMfgName, ProductFwDesc, CapsuleFileName, PfxFile=None, PfxPass=None, | |
Rollback=False, Arch='amd64', OperatingSystem_String='Win10'): | |
logging.debug("CapsulePackage: Create Windows Capsule Files") | |
#Make INF | |
InfFilePath = os.path.join(OutputFolder, ProductName + ".inf") | |
InfTool = InfGenerator(ProductName, ProductFwProvider, ProductFmpGuid, Arch, ProductFwDesc, CapsuleVersion_DotString, CapsuleVersion_HexString) | |
InfTool.Manufacturer = ProductFwMfgName #optional | |
ret = InfTool.MakeInf(InfFilePath, CapsuleFileName, Rollback) | |
if(ret != 0): | |
raise Exception("CreateWindowsInf Failed with errorcode %d" % ret) | |
#Make CAT | |
CatFilePath = os.path.realpath(os.path.join(OutputFolder, ProductName + ".cat")) | |
CatTool = CatGenerator(Arch, OperatingSystem_String) | |
ret = CatTool.MakeCat(CatFilePath) | |
if(ret != 0): | |
raise Exception("Creating Cat file Failed with errorcode %d" % ret) | |
if(PfxFile is not None): | |
#Find Signtool | |
SignToolPath = FindToolInWinSdk("signtool.exe") | |
if not os.path.exists(SignToolPath): | |
raise Exception("Can't find signtool on this machine.") | |
#dev sign the cat file | |
ret = CatalogSignWithSignTool(SignToolPath, CatFilePath, PfxFile, PfxPass) | |
if(ret != 0): | |
raise Exception("Signing Cat file Failed with errorcode %d" % ret) | |
return ret |