75 lines
1.5 KiB
Python
75 lines
1.5 KiB
Python
#
|
|
#
|
|
#
|
|
class JpgBinWrite:
|
|
|
|
soi_marker = b'\xff\xd8'
|
|
eof_marker = b'\xff\xd9'
|
|
picseal_marker = b'\xff\xe0'
|
|
|
|
#
|
|
def __init__(self, jpg_in, jpg_out):
|
|
self.fhr = jpg_in
|
|
self.fhw = jpg_out
|
|
|
|
#
|
|
# input is the Crypto Sig class
|
|
#
|
|
def writeJpgPicSealPub(self, crypto_sig, fp):
|
|
self.__writeJpgHeader()
|
|
|
|
# write picseal serialize data
|
|
|
|
self.__writeJpgImg(fp)
|
|
|
|
#
|
|
# input is the Crypto Sig class
|
|
#
|
|
def writeJpgPicSealPvt(self, crypto_sig, fp):
|
|
self.__writeJpgHeader()
|
|
|
|
# write picseal serialize data
|
|
|
|
self.__writeJpgImg(fp)
|
|
|
|
|
|
#
|
|
def __writeJpgImg(self, fp):
|
|
self.__writeJpgMetadata(fp.markers_meta)
|
|
self.__writeJpgImgData(fp.markers_img)
|
|
self.__writeJpgFooter()
|
|
self.fhw.flush()
|
|
self.fhw.close()
|
|
|
|
#
|
|
# array of the marker information
|
|
#
|
|
def __writeJpgMetadata(self, markers):
|
|
for marker in markers:
|
|
self.fhr.seek(marker.fpos)
|
|
data = self.fhr.read(marker.len)
|
|
self.fhw.write(data)
|
|
self.fhw.flush()
|
|
|
|
#
|
|
# array of the marker information
|
|
#
|
|
def __writeJpgImgData(self, markers):
|
|
for marker in markers:
|
|
# most of the image data is sequential
|
|
cpos = self.fhr.tell()
|
|
if (marker.fpos != cpos):
|
|
self.fhr.seek(marker.fpos)
|
|
|
|
data = self.fhr.read(marker.len)
|
|
self.fhw.write(data)
|
|
self.fhw.flush()
|
|
|
|
#
|
|
def __writeJpgHeader(self):
|
|
self.fhw.write(bytes(JpgBinWrite.soi_marker))
|
|
|
|
#
|
|
def __writeJpgFooter(self):
|
|
self.fhw.write(bytes(JpgBinWrite.eof_marker))
|