X-Git-Url: https://git.proxmox.com/?a=blobdiff_plain;f=BaseTools%2FSource%2FPython%2FGenFds%2FCapsule.py;fp=BaseTools%2FSource%2FPython%2FGenFds%2FCapsule.py;h=baa6e1dfa2ab8bc31f61aaaca0b8c0bc501ef82f;hb=1ccc4d895dd8d659d016efcd6ef8a48749aba1d0;hp=1c673949fa8009b8b6cd621aac6f290a1e898764;hpb=678f85131238622e576705117e299d81cff755c9;p=mirror_edk2.git

diff --git a/BaseTools/Source/Python/GenFds/Capsule.py b/BaseTools/Source/Python/GenFds/Capsule.py
index 1c673949fa..baa6e1dfa2 100644
--- a/BaseTools/Source/Python/GenFds/Capsule.py
+++ b/BaseTools/Source/Python/GenFds/Capsule.py
@@ -15,13 +15,13 @@
 ##
 # Import Modules
 #
+from __future__ import absolute_import
 from .GenFdsGlobalVariable import GenFdsGlobalVariable
 from .GenFdsGlobalVariable import FindExtendTool
 from CommonDataClass.FdfClass import CapsuleClassObject
 import Common.LongFilePathOs as os
 import subprocess
 from io import BytesIO
-from io import StringIO
 from Common.Misc import SaveFileOnChange
 from Common.Misc import PackRegistryFormatGuid
 import uuid
@@ -185,7 +185,7 @@ class Capsule (CapsuleClassObject) :
         #
         # The real capsule header structure is 28 bytes
         #
-        Header.write(b'\x00'*(HdrSize-28))
+        Header.write('\x00'*(HdrSize-28))
         Header.write(FwMgrHdr.getvalue())
         Header.write(Content.getvalue())
         #
@@ -247,7 +247,7 @@ class Capsule (CapsuleClassObject) :
     def GenCapInf(self):
         self.CapInfFileName = os.path.join(GenFdsGlobalVariable.FvDir,
                                    self.UiCapsuleName +  "_Cap" + '.inf')
-        CapInfFile = StringIO() #open (self.CapInfFileName , 'w+')
+        CapInfFile = BytesIO() #open (self.CapInfFileName , 'w+')
 
         CapInfFile.writelines("[options]" + T_CHAR_LF)