about summary refs log tree commit diff stats
path: root/example/extract_pe_ressources.py
blob: 71d8e2262339d5955fd951a59ac0b15961921e45 (plain) (blame)
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
import sys
import struct
from elfesteem import *
import os
import sys

# example for extracting all pe ressources

def extract_res(res, name_o = "", num = 0, lvl=-1):
    lvl +=1
    if not res:
        return num
    for x in res.resentries:
        print "\t"*lvl, repr(x)
        num += 1

        if x.name_s:
            name = name_o[:]+repr(x.name_s.value[::2])
        else:
            name = name_o[:]

        if x.data:
            print "\t"*lvl, 'data', len(x.data.s)
            open('out/%.3d_%s.bin'%(num, name), 'w').write(str(x.data.s))
        else:
            print "\t"*lvl, None
        if x.offsettosubdir:
            num = extract_res(x.subdir, name, num, lvl+1)
    return num



try:
    os.stat('out')
except:
    os.mkdir('out')

fname = sys.argv[1]
e = pe_init.PE(open(fname, 'rb').read())
res = e.DirRes.resdesc



extract_res(res)