mirror of https://github.com/acidanthera/audk.git
65 lines
2.6 KiB
Python
65 lines
2.6 KiB
Python
##
|
|
# 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
|