forked from DidierStevens/DidierStevensSuite
-
Notifications
You must be signed in to change notification settings - Fork 0
/
mPDF.py
194 lines (171 loc) · 6.55 KB
/
mPDF.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
#!/usr/bin/python
# module with simple class to build PDF documents with basic PDF elements
# Source code put in public domain by Didier Stevens, no Copyright
# https://DidierStevens.com
# Use at your own risk
#
# History:
#
# 2008/05/18: continue
# 2008/05/19: continue
# 2008/05/28: stream2
# 2008/11/09: cleanup for release
# 2008/11/21: Added support for other OSes than Windows
# 2009/05/04: Added support for abbreviated filters (/AHx and /Fl), thanks Binjo
# 2011/03/03: Added support for info in trailer and xrefAndTrailer
# 2011/07/01: V0.1.4: Added support for filters i and I; added support for Python 3
# 2012/02/25: fixed printing \n for filters i and I
# 2015/04/15: updated header and added comment
# Todo:
# - add support for extra filters to stream2
__author__ = 'Didier Stevens'
__version__ = '0.1.5'
__date__ = '2015/04/15'
import sys
import zlib
import platform
def SplitByLength(input, length):
result = []
while len(input) > length:
result.append(input[0:length] + '\n')
input = input[length:]
result.append(input + '>')
return result
class cPDF:
def __init__(self, filename):
self.filename = filename
self.indirectObjects = {}
def appendString(self, str):
fPDF = open(self.filename, 'a')
fPDF.write(str)
fPDF.close()
def appendBinary(self, str):
fPDF = open(self.filename, 'ab')
if sys.version_info[0] == 2:
fPDF.write(str)
else:
fPDF.write(bytes(str, 'ascii'))
fPDF.close()
def filesize(self):
fPDF = open(self.filename, 'rb')
fPDF.seek(0, 2)
size = fPDF.tell()
fPDF.close()
return size
def IsWindows(self):
return platform.system() in ('Windows', 'Microsoft')
def header(self, version='1.1'):
fPDF = open(self.filename, 'w')
fPDF.write('%%PDF-%s\n' % version)
fPDF.close()
def binary(self):
self.appendString("%\xD0\xD0\xD0\xD0\n")
def comment(self, comment):
self.appendString('%' + comment + '\n')
def indirectobject(self, index, version, io):
self.appendString("\n")
self.indirectObjects[index] = self.filesize()
self.appendString("%d %d obj\n%s\nendobj\n" % (index, version, io))
def stream(self, index, version, streamdata, dictionary="<< /Length %d >>"):
self.appendString("\n")
self.indirectObjects[index] = self.filesize()
self.appendString(("%d %d obj\n" + dictionary + "\nstream\n") % (index, version, len(streamdata)))
self.appendBinary(streamdata)
self.appendString("\nendstream\nendobj\n")
def Data2HexStr(self, data):
hex = ''
if sys.version_info[0] == 2:
for b in data:
hex += "%02x" % ord(b)
else:
for b in data:
hex += "%02x" % b
return hex
def stream2(self, index, version, streamdata, entries="", filters=""):
"""
* h ASCIIHexDecode
* H AHx
* i like ASCIIHexDecode but with 512 long lines
* I like AHx but with 512 long lines
* ASCII85Decode
* LZWDecode
* f FlateDecode
* F Fl
* RunLengthDecode
* CCITTFaxDecode
* JBIG2Decode
* DCTDecode
* JPXDecode
* Crypt
"""
encodeddata = streamdata
filter = []
for i in filters:
if i.lower() == "h":
encodeddata = self.Data2HexStr(encodeddata) + '>'
if i == "h":
filter.insert(0, "/ASCIIHexDecode")
else:
filter.insert(0, "/AHx")
elif i.lower() == "i":
encodeddata = ''.join(SplitByLength(self.Data2HexStr(encodeddata), 512))
if i == "i":
filter.insert(0, "/ASCIIHexDecode")
else:
filter.insert(0, "/AHx")
elif i.lower() == "f":
encodeddata = zlib.compress(encodeddata)
if i == "f":
filter.insert(0, "/FlateDecode")
else:
filter.insert(0, "/Fl")
else:
print("Error")
return
self.appendString("\n")
self.indirectObjects[index] = self.filesize()
self.appendString("%d %d obj\n<<\n /Length %d\n" % (index, version, len(encodeddata)))
if len(filter) == 1:
self.appendString(" /Filter %s\n" % filter[0])
if len(filter) > 1:
self.appendString(" /Filter [%s]\n" % ' '.join(filter))
if entries != "":
self.appendString(" %s\n" % entries)
self.appendString(">>\nstream\n")
if filters[-1].lower() == 'i':
self.appendString(encodeddata)
else:
self.appendBinary(encodeddata)
self.appendString("\nendstream\nendobj\n")
def xref(self):
self.appendString("\n")
startxref = self.filesize()
max = 0
for i in self.indirectObjects.keys():
if i > max:
max = i
self.appendString("xref\n0 %d\n" % (max+1))
if self.IsWindows():
eol = '\n'
else:
eol = ' \n'
for i in range(0, max+1):
if i in self.indirectObjects:
self.appendString("%010d %05d n%s" % (self.indirectObjects[i], 0, eol))
else:
self.appendString("0000000000 65535 f%s" % eol)
return (startxref, (max+1))
def trailer(self, startxref, size, root, info=None):
if info == None:
self.appendString("trailer\n<<\n /Size %d\n /Root %s\n>>\nstartxref\n%d\n%%%%EOF\n" % (size, root, startxref))
else:
self.appendString("trailer\n<<\n /Size %d\n /Root %s\n /Info %s\n>>\nstartxref\n%d\n%%%%EOF\n" % (size, root, info, startxref))
def xrefAndTrailer(self, root, info=None):
xrefdata = self.xref()
self.trailer(xrefdata[0], xrefdata[1], root, info)
def template1(self):
self.indirectobject(1, 0, "<<\n /Type /Catalog\n /Outlines 2 0 R\n /Pages 3 0 R\n>>")
self.indirectobject(2, 0, "<<\n /Type /Outlines\n /Count 0\n>>")
self.indirectobject(3, 0, "<<\n /Type /Pages\n /Kids [4 0 R]\n /Count 1\n>>")
self.indirectobject(4, 0, "<<\n /Type /Page\n /Parent 3 0 R\n /MediaBox [0 0 612 792]\n /Contents 5 0 R\n /Resources <<\n /ProcSet [/PDF /Text]\n /Font << /F1 6 0 R >>\n >>\n>>")
self.indirectobject(6, 0, "<<\n /Type /Font\n /Subtype /Type1\n /Name /F1\n /BaseFont /Helvetica\n /Encoding /MacRomanEncoding\n>>")