diff options
Diffstat (limited to 'BaseTools/Source/Python/Capsule/WindowsCapsuleSupportHelper.py')
| -rw-r--r-- | BaseTools/Source/Python/Capsule/WindowsCapsuleSupportHelper.py | 128 |
1 files changed, 64 insertions, 64 deletions
diff --git a/BaseTools/Source/Python/Capsule/WindowsCapsuleSupportHelper.py b/BaseTools/Source/Python/Capsule/WindowsCapsuleSupportHelper.py index a29ac21ae8..9b049bddba 100644 --- a/BaseTools/Source/Python/Capsule/WindowsCapsuleSupportHelper.py +++ b/BaseTools/Source/Python/Capsule/WindowsCapsuleSupportHelper.py @@ -1,64 +1,64 @@ -##
-# 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
+## +# 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 |
