public inbox for devel@edk2.groups.io
 help / color / mirror / Atom feed
* [Patch] BaseTools: add capsule image header for auth FMP capsule file
@ 2016-08-22  7:23 Yonghong Zhu
  2016-08-22  8:47 ` Gao, Liming
  0 siblings, 1 reply; 2+ messages in thread
From: Yonghong Zhu @ 2016-08-22  7:23 UTC (permalink / raw)
  To: edk2-devel; +Cc: Liming Gao

in last commit 91ae29, it missed to add the
EFI_FIRMWARE_MANAGEMENT_CAPSULE_IMAGE_HEADER for the auth FMP capsule.

Cc: Liming Gao <liming.gao@intel.com>
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Yonghong Zhu <yonghong.zhu@intel.com>
---
 BaseTools/Source/Python/GenFds/Capsule.py     | 15 +++++++++++----
 BaseTools/Source/Python/GenFds/CapsuleData.py | 10 ----------
 2 files changed, 11 insertions(+), 14 deletions(-)

diff --git a/BaseTools/Source/Python/GenFds/Capsule.py b/BaseTools/Source/Python/GenFds/Capsule.py
index f8af12a..93ecee1 100644
--- a/BaseTools/Source/Python/GenFds/Capsule.py
+++ b/BaseTools/Source/Python/GenFds/Capsule.py
@@ -139,10 +139,11 @@ class Capsule (CapsuleClassObject) :
             PreSize += os.path.getsize(FileName)
             File = open(FileName, 'rb')
             Content.write(File.read())
             File.close()
         for fmp in self.FmpPayloadList:
+            Buffer = fmp.GenCapsuleSubItem()
             if fmp.Certificate_Guid:
                 ExternalTool, ExternalOption = FindExtendTool([], GenFdsGlobalVariable.ArchList, fmp.Certificate_Guid)
                 CmdOption = ''
                 CapInputFile = fmp.ImageFile
                 if not os.path.isabs(fmp.ImageFile):
@@ -159,11 +160,11 @@ class Capsule (CapsuleClassObject) :
                 GenFdsGlobalVariable.CallExternalTool(CmdList, "Failed to generate FMP auth capsule")
                 if uuid.UUID(fmp.Certificate_Guid) == EFI_CERT_TYPE_PKCS7_GUID:
                     dwLength = 4 + 2 + 2 + 16 + os.path.getsize(CapOutputTmp) - os.path.getsize(CapInputFile)
                 else:
                     dwLength = 4 + 2 + 2 + 16 + 16 + 256 + 256
-                Buffer  = pack('Q', fmp.MonotonicCount)
+                Buffer += pack('Q', fmp.MonotonicCount)
                 Buffer += pack('I', dwLength)
                 Buffer += pack('H', WIN_CERT_REVISION)
                 Buffer += pack('H', WIN_CERT_TYPE_EFI_GUID)
                 Buffer += uuid.UUID(fmp.Certificate_Guid).get_bytes_le()
                 if os.path.exists(CapOutputTmp):
@@ -176,14 +177,20 @@ class Capsule (CapsuleClassObject) :
                         VendorFile.close()
                     FwMgrHdr.write(pack('=Q', PreSize))
                     PreSize += len(Buffer)
                     Content.write(Buffer)
             else:
-                payload = fmp.GenCapsuleSubItem()
+                ImageFile = open(fmp.ImageFile, 'rb')
+                Buffer += ImageFile.read()
+                ImageFile.close()
+                if fmp.VendorCodeFile:
+                    VendorFile = open(fmp.VendorCodeFile, 'rb')
+                    Buffer += VendorFile.read()
+                    VendorFile.close()
                 FwMgrHdr.write(pack('=Q', PreSize))
-                PreSize += len(payload)
-                Content.write(payload)
+                PreSize += len(Buffer)
+                Content.write(Buffer)
         BodySize = len(FwMgrHdr.getvalue()) + len(Content.getvalue())
         Header.write(pack('=I', HdrSize + BodySize))
         #
         # The real capsule header structure is 28 bytes
         #
diff --git a/BaseTools/Source/Python/GenFds/CapsuleData.py b/BaseTools/Source/Python/GenFds/CapsuleData.py
index 2a5c454..5d5a1e4 100644
--- a/BaseTools/Source/Python/GenFds/CapsuleData.py
+++ b/BaseTools/Source/Python/GenFds/CapsuleData.py
@@ -214,16 +214,6 @@ class CapsulePayload(CapsuleData):
                        0,
                        ImageFileSize,
                        VendorFileSize,
                        int(self.HardwareInstance, 16)
                        )
-        #
-        # Append file content to the structure
-        #
-        ImageFile = open(self.ImageFile, 'rb')
-        Buffer += ImageFile.read()
-        ImageFile.close()
-        if self.VendorCodeFile:
-            VendorFile = open(self.VendorCodeFile, 'rb')
-            Buffer += VendorFile.read()
-            VendorFile.close()
         return Buffer
-- 
2.6.1.windows.1



^ permalink raw reply related	[flat|nested] 2+ messages in thread

* Re: [Patch] BaseTools: add capsule image header for auth FMP capsule file
  2016-08-22  7:23 [Patch] BaseTools: add capsule image header for auth FMP capsule file Yonghong Zhu
@ 2016-08-22  8:47 ` Gao, Liming
  0 siblings, 0 replies; 2+ messages in thread
From: Gao, Liming @ 2016-08-22  8:47 UTC (permalink / raw)
  To: Zhu, Yonghong, edk2-devel@lists.01.org

Reviewed-by: Liming Gao <liming.gao@intel.com>

> -----Original Message-----
> From: Zhu, Yonghong
> Sent: Monday, August 22, 2016 3:24 PM
> To: edk2-devel@lists.01.org
> Cc: Gao, Liming <liming.gao@intel.com>
> Subject: [Patch] BaseTools: add capsule image header for auth FMP capsule
> file
> 
> in last commit 91ae29, it missed to add the
> EFI_FIRMWARE_MANAGEMENT_CAPSULE_IMAGE_HEADER for the auth
> FMP capsule.
> 
> Cc: Liming Gao <liming.gao@intel.com>
> Contributed-under: TianoCore Contribution Agreement 1.0
> Signed-off-by: Yonghong Zhu <yonghong.zhu@intel.com>
> ---
>  BaseTools/Source/Python/GenFds/Capsule.py     | 15 +++++++++++----
>  BaseTools/Source/Python/GenFds/CapsuleData.py | 10 ----------
>  2 files changed, 11 insertions(+), 14 deletions(-)
> 
> diff --git a/BaseTools/Source/Python/GenFds/Capsule.py
> b/BaseTools/Source/Python/GenFds/Capsule.py
> index f8af12a..93ecee1 100644
> --- a/BaseTools/Source/Python/GenFds/Capsule.py
> +++ b/BaseTools/Source/Python/GenFds/Capsule.py
> @@ -139,10 +139,11 @@ class Capsule (CapsuleClassObject) :
>              PreSize += os.path.getsize(FileName)
>              File = open(FileName, 'rb')
>              Content.write(File.read())
>              File.close()
>          for fmp in self.FmpPayloadList:
> +            Buffer = fmp.GenCapsuleSubItem()
>              if fmp.Certificate_Guid:
>                  ExternalTool, ExternalOption = FindExtendTool([],
> GenFdsGlobalVariable.ArchList, fmp.Certificate_Guid)
>                  CmdOption = ''
>                  CapInputFile = fmp.ImageFile
>                  if not os.path.isabs(fmp.ImageFile):
> @@ -159,11 +160,11 @@ class Capsule (CapsuleClassObject) :
>                  GenFdsGlobalVariable.CallExternalTool(CmdList, "Failed to generate
> FMP auth capsule")
>                  if uuid.UUID(fmp.Certificate_Guid) == EFI_CERT_TYPE_PKCS7_GUID:
>                      dwLength = 4 + 2 + 2 + 16 + os.path.getsize(CapOutputTmp) -
> os.path.getsize(CapInputFile)
>                  else:
>                      dwLength = 4 + 2 + 2 + 16 + 16 + 256 + 256
> -                Buffer  = pack('Q', fmp.MonotonicCount)
> +                Buffer += pack('Q', fmp.MonotonicCount)
>                  Buffer += pack('I', dwLength)
>                  Buffer += pack('H', WIN_CERT_REVISION)
>                  Buffer += pack('H', WIN_CERT_TYPE_EFI_GUID)
>                  Buffer += uuid.UUID(fmp.Certificate_Guid).get_bytes_le()
>                  if os.path.exists(CapOutputTmp):
> @@ -176,14 +177,20 @@ class Capsule (CapsuleClassObject) :
>                          VendorFile.close()
>                      FwMgrHdr.write(pack('=Q', PreSize))
>                      PreSize += len(Buffer)
>                      Content.write(Buffer)
>              else:
> -                payload = fmp.GenCapsuleSubItem()
> +                ImageFile = open(fmp.ImageFile, 'rb')
> +                Buffer += ImageFile.read()
> +                ImageFile.close()
> +                if fmp.VendorCodeFile:
> +                    VendorFile = open(fmp.VendorCodeFile, 'rb')
> +                    Buffer += VendorFile.read()
> +                    VendorFile.close()
>                  FwMgrHdr.write(pack('=Q', PreSize))
> -                PreSize += len(payload)
> -                Content.write(payload)
> +                PreSize += len(Buffer)
> +                Content.write(Buffer)
>          BodySize = len(FwMgrHdr.getvalue()) + len(Content.getvalue())
>          Header.write(pack('=I', HdrSize + BodySize))
>          #
>          # The real capsule header structure is 28 bytes
>          #
> diff --git a/BaseTools/Source/Python/GenFds/CapsuleData.py
> b/BaseTools/Source/Python/GenFds/CapsuleData.py
> index 2a5c454..5d5a1e4 100644
> --- a/BaseTools/Source/Python/GenFds/CapsuleData.py
> +++ b/BaseTools/Source/Python/GenFds/CapsuleData.py
> @@ -214,16 +214,6 @@ class CapsulePayload(CapsuleData):
>                         0,
>                         ImageFileSize,
>                         VendorFileSize,
>                         int(self.HardwareInstance, 16)
>                         )
> -        #
> -        # Append file content to the structure
> -        #
> -        ImageFile = open(self.ImageFile, 'rb')
> -        Buffer += ImageFile.read()
> -        ImageFile.close()
> -        if self.VendorCodeFile:
> -            VendorFile = open(self.VendorCodeFile, 'rb')
> -            Buffer += VendorFile.read()
> -            VendorFile.close()
>          return Buffer
> --
> 2.6.1.windows.1



^ permalink raw reply	[flat|nested] 2+ messages in thread

end of thread, other threads:[~2016-08-22  8:47 UTC | newest]

Thread overview: 2+ messages (download: mbox.gz follow: Atom feed
-- links below jump to the message on this page --
2016-08-22  7:23 [Patch] BaseTools: add capsule image header for auth FMP capsule file Yonghong Zhu
2016-08-22  8:47 ` Gao, Liming

This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox