First public commit

PolyglotFile/__init__.py

@@ -0,0 +1,7 @@
+# -*- coding: utf-8 -*-
+# -*- coding: utf-8 -*-
+
+from .polyglotpdfzip import PolyglotPdfZip
+from .polyglotzippdf import PolyglotZipPdf
+from .polyglotszippdf import PolyglotSZipPdf
+

PolyglotFile/polyglotpdfzip.py

@@ -0,0 +1,55 @@
+# -*- coding: utf-8 -*-
+
+import logging
+
+'''
+    |-------------------------------|    -
+    |--------- PDF Header ----------K1   | J1
+    |-------------------------------|    -
+    |----- PDF OBJ 1 = ZIP Data ----K2   |
+    |-------------------------------|    -
+    |---- Original PDF Ojbects -----K3   | J2
+    |-------------------------------|    -
+    |--- Last OBJ = End Zip Data ---K4   |
+    |-------------------------------|    |
+    |---------- Xref Table ---------|    |
+    |-------------------------------K5   |
+    |----------- Trailer -----------|    |
+    |-------------------------------|    |
+'''
+
+
+class PolyglotPdfZip():
+    from PdfFileTransformer import Pdf
+    from ZipFileTransformer import Zip
+
+    def __init__(self, Pdf, Zip):
+        self.buffer = bytearray()
+        self.pdf = Pdf
+        self.zip = Zip
+        self.buffer = bytearray()
+
+    def generate(self):
+        k2_stream = self.zip.buffer[:self.zip.end_of_data]
+        size_k2_stream = len(k2_stream)
+        self.pdf.insert_new_obj_stream_at_start(k2_stream)
+        offset_k2_stream = self.pdf.get_first_stream_offset()
+
+        k4_stream = self.zip.buffer[self.zip.central_dir_file_header:]
+        size_k4_stream = len(k4_stream)
+        self.pdf.insert_new_obj_stream_at_end(k4_stream)
+        offset_k4_stream = self.pdf.get_last_stream_offset()
+
+        pdf_buffer = self.pdf.get_build_buffer()
+
+        j1 = pdf_buffer[0:offset_k2_stream]
+        j2 = pdf_buffer[offset_k2_stream + size_k2_stream:offset_k4_stream]
+        self.zip.add_data_to_file(j1, j2, True)
+
+        k5 = pdf_buffer[offset_k4_stream + size_k4_stream:]
+        self.buffer = self.zip.buffer + k5
+
+    def write(self, filename):
+        fd = open(filename, "wb")
+        fd.write(self.buffer)
+        fd.close()

PolyglotFile/polyglotszippdf.py

@@ -0,0 +1,110 @@
+# -*- coding: utf-8 -*-
+
+from .polyglotpdfzip import PolyglotPdfZip
+import logging
+import tempfile
+from ZipFileTransformer import ZipFile
+from ZipFileTransformer import Zip
+from PdfFileTransformer import Pdf
+
+'''
+    |-----------------------------------|  -
+    |--------- ZIP Data[0] = -----------|  |
+    |- PDF Header + PDF Obj[0] Header --|  |
+    |-----------------------------------|  | K2
+    |------- PDF Obj[0] stream =  ------|  |
+    |--------- ZIP Data LF [1:] --------|  |
+    |-----------------------------------|  -
+    |------ Original PDF Ojbects -------|  |
+    |-----------------------------------|  |
+    |------------ Xref Table -----------|  |
+    |-----------------------------------|  | J2
+    |------------- Trailer -------------|  |
+    |-----------------------------------|  -
+    |---------- End Zip Data -----------|
+    |-----------------------------------|
+'''
+
+
+class PolyglotSZipPdf(PolyglotPdfZip):
+
+    def __init__(self, Pdf, Zip):
+        super().__init__(Pdf, Zip)
+
+    def get_rebuild_zip_first_part_size(self):
+
+        zo_path = tempfile.mkstemp()[1]
+        logging.info("use tmp file zip: " + zo_path)
+        zo = ZipFile(zo_path, 'a')
+        zi = ZipFile(self.zip.filename, 'r')
+        for zipinfo in zi.infolist():
+            zo.writestr(zipinfo, zi.read(zipinfo))
+        zi.close()
+        zo.close()
+
+        rebuild_zip = Zip(zo_path)
+
+        p = rebuild_zip.end_of_data
+        k2_stream = rebuild_zip.buffer[:p]
+
+        size_k2_stream = len(k2_stream)
+
+        return size_k2_stream
+
+    def get_pdf_header(self):
+        return self.pdf.get_file_header()
+
+    def generate_zip_with_pdf_part(self, filename, pdf_data):
+
+        zo = ZipFile(filename, 'a')
+        zi = ZipFile(self.zip.filename, 'r')
+        zo.writestr(' ', pdf_data, 0)
+        for zipinfo in zi.infolist():
+            zo.writestr(zipinfo, zi.read(zipinfo))
+        zi.close()
+        zo.close()
+
+    def get_rebuild_pdf(self, zo_path, offset):
+        '''
+            Generate polyglot with final zip.
+        '''
+        new_zip = Zip(zo_path)
+        new_pdf = Pdf(self.pdf.filename)
+
+        p1 = new_zip.end_of_first_local_file_header
+        p2 = new_zip.end_of_data
+        k2_stream = new_zip.buffer[p1:p2]
+
+        size_k2_stream = len(k2_stream)
+        new_pdf.insert_new_obj_stream_at_start(k2_stream)
+        k2_stream_offset = new_pdf.get_first_stream_offset()
+
+        new_pdf.file_offset = offset
+        pdf_buffer = new_pdf.get_build_buffer()
+        j2 = pdf_buffer[k2_stream_offset + size_k2_stream:]
+        new_zip.add_data_to_file(b'', j2, True)
+
+        return new_zip.buffer
+
+    def get_pdf_offset(self, zipfile):
+
+        f = open(zipfile, "rb")
+        data = f.read()
+        return data.find(b"%PDF")
+
+    def generate(self):
+
+        zip_stream_size = self.get_rebuild_zip_first_part_size()
+        pdf_header = self.get_pdf_header()
+        pdf_header = (pdf_header +
+                      b'1 0 obj\n<<\n/Filter /FlateDecode\n/Length ' +
+                      str(zip_stream_size).encode("utf-8") +
+                      b'\n>>\nstream\n')
+
+        filename = tempfile.mkstemp()[1]
+        logging.info("use tmp file for new zip: " + filename)
+        self.generate_zip_with_pdf_part(filename, pdf_header)
+
+        pdf_offset = self.get_pdf_offset(filename)
+
+        self.buffer = self.get_rebuild_pdf(filename, pdf_offset)

PolyglotFile/polyglotzippdf.py

@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+
+from .polyglotpdfzip import PolyglotPdfZip
+
+
+'''
+    |-------------------------------|    -
+    |--------- PDF Header ----------K1   | J1
+    |-------------------------------|    -
+    |----- PDF OBJ 1 = ZIP Data ----K2   |
+    |-------------------------------|    -
+    |---- Original PDF Ojbects -----K3   |
+    |-------------------------------|    |
+    |---------- Xref Table ---------|    |
+    |-------------------------------K4   | J2
+    |----------- Trailer -----------|    |
+    |-------------------------------|    -
+    |-------- End Zip Data ---------|    |
+    |-------------------------------|    |
+'''
+
+
+class PolyglotZipPdf(PolyglotPdfZip):
+
+    def generate(self):
+        k2_stream = self.zip.buffer[:self.zip.end_of_data]
+        size_k2_stream = len(k2_stream)
+        self.pdf.insert_new_obj_stream_at_start(k2_stream)
+        offset_k2_stream = self.pdf.get_first_stream_offset()
+
+        pdf_buffer = self.pdf.get_build_buffer()
+
+        j1 = pdf_buffer[0:offset_k2_stream]
+        j2 = pdf_buffer[offset_k2_stream + size_k2_stream:]
+
+        self.zip.add_data_to_file(j1, j2, True)
+        self.buffer = self.zip.buffer