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
|
#! /usr/bin/python3
import os
import re
import struct
script_dir = os.path.dirname(os.path.realpath(__file__))
root_dir = script_dir + "/../"
data_dir = root_dir + "data/"
asm_dir = root_dir + "asm/"
def try_text(text_bytes):
bad_bytes = 0
for byte in text_bytes:
if byte < 32:
bad_bytes += 1
# Arbitrary string detection heuristic
#if bad_bytes / len(text_bytes) >= 0.3:
# return None
try:
text = text_bytes.decode("EUC-JP")
except UnicodeDecodeError:
return None
text = text.strip(" \0")
if len(text) > 0 and bool(re.search('[a-zA-Z]', text)):
text = text.replace("\\x00", "")
text = text.replace("\n", "\\n")
text = text.replace("\"", "\\\"")
return text
def is_zeros(stuff):
for piece in stuff:
if piece.strip() != "0x00000000":
return False
return True
def try_float(word):
if (word[:3] == "0x3") or (word[:3] == "0x4") or \
(word[:3] == "0xB") or (word[:3] == "0xC"):
return struct.unpack('!f', bytes.fromhex(word[2:10]))[0]
def quick_convert(words_string):
words = words_string.split(",")
byte_array = b""
for word in words:
data = word.strip()[2:]
byte_array += bytearray.fromhex(data)
return byte_array.decode("EUC-JP")
def word_convert(byte_string):
try:
words = byte_string.split(",")
byte_array = b""
for word in words:
data = word.strip()[2:]
byte_array += bytearray.fromhex(data)
except ValueError:
return byte_string
if byte_array[0] == 63 and byte_array[-1] == 45:
return byte_string
if data == '0A0A0000':
return "\\n\\n"
if len(words) > 1 and not is_zeros(words[1:]):
res = try_text(byte_array)
if res is not None:
return " .asciz \"" + res + "\"\n .balign 4\n"
if len(words) == 1 or is_zeros(words[1:]):
res = str(try_float(words[0].strip()))
if res is not None:
return " .float " + res + "\n"
return byte_string
def handle_match(match):
in_str = match.group()[6:]
ret = word_convert(in_str)
if ret == in_str:
return match.group()
return ret
def process_data_file(file_path):
with open(file_path) as f:
file_text = f.read()
new_file_text = re.sub(" \\.word 0x.*", handle_match, file_text)
if new_file_text != file_text:
with open(file_path, "w") as f:
f.write(new_file_text)
return True
return False
def main():
i = 0
for root, dirs, files in os.walk(asm_dir):
for file in files:
#if i == 10:
# return
if file.endswith(".s"):
path = os.path.join(root, file)
if process_data_file(path):
print("Processed " + path)
i += 1
main()
|
