deejayeye-modder/defog_JEB_export.py at master · dwallersv/deejayeye-modder · GitHub

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
#!/usr/bin/env python
#
# https://regex101.com/r/7i9dhC/1 - '\s*const-string.*\s*v[0-9]+\s*,\s*\"([^\"]*)\"\s*' [Match Smalli Files]
# https://regex101.com/r/EFWkGp/3 - '.*?a\.a\("([A-Za-z0-9+\/=]*)".*'                   [Match Java files]
#
# Updated version of https://github.com/Bin4ry/deejayeye-modder/blob/master/decrypt_strings.py
# This version handles JEB exports.
# https://www.pnfsoftware.com/jeb2/manual/misc/#exporting-output
#
# Mass replace functions from https://stackoverflow.com/a/1597755
#
# Code mashup by Hostile
# usage:
# $ python decrypt_JEB_export.py ~/Desktop/DJI418

import os
import re
import sys
import fnmatch
import base64
import binascii

# We are only worrying about the current key

key = 'Y9*PI8B#gD^6Yhd1'
klen=8

def decrypt(s,l):
        s = base64.decodestring(s)
        decr = ''.join([chr(ord(c) ^ ord(key[i%l*2])) for i,c in enumerate(s)])
        decr = decr.replace('\r', '').replace('\n', '').replace('"', '').replace('\\','')
        return decr

# list of extensions to replace
DEFAULT_REPLACE_EXTENSIONS = (".java")

def try_to_replace(fname, replace_extensions=DEFAULT_REPLACE_EXTENSIONS):
    if replace_extensions:
        return fname.lower().endswith(replace_extensions)
    return True

def file_replace(fname, pat):
    # first, see if the pattern is even in the file.
    with open(fname) as f:
        if not any(re.search(pat, line) for line in f):
            return # pattern does not occur in file so we are done.

    # pattern is in the file, so perform replace operation.
    with open(fname) as f:
        out_fname = fname + ".tmp"
        out = open(out_fname, "w")
        for line in f:
            match = re.search(pat, line)

	    # Grab the rest of the line here!!! don't just replace with the decrypted match
	    if match is not None:
		match = match.group(1)
		s_after = ""
		try:
		        s_after = decrypt(match,klen)
		except binascii.Error as err:
			out.write(line)
			pass

		line_after = line.replace(match,s_after)
		line_after = line_after.replace("a.a(",'(')
                out.write(re.sub(pat, line_after, line)[:-1])
	    else:
            	out.write(line)
        out.close()
	print "wrote " + fname
        os.rename(out_fname, fname)

# https://gist.github.com/whophil/2a999bcaf0ebfbd6e5c0d213fb38f489
def recursive_glob(rootdir='.', pattern='*'):
        matches = []
        for root, dirnames, filenames in os.walk(rootdir):
          for filename in fnmatch.filter(filenames, pattern):
                  matches.append(os.path.join(root, filename))
        return matches

def mass_replace(dir_name, replace_extensions=DEFAULT_REPLACE_EXTENSIONS):
    pat = re.compile('.*?a\.a\("([A-Za-z0-9+\/=]*)".*')
    filenames = recursive_glob(dir_name, "*.java")
    for fname in filenames:
    	if try_to_replace(fname, replace_extensions):
		file_replace(fname, pat)

if len(sys.argv) != 2:
    u = "Usage: mass_replace <dir_name>\n"
    sys.stderr.write(u)
    sys.exit(1)

mass_replace(sys.argv[1])