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
|
# -*- coding: utf-8 -*-
"""
@author: cleartonic
"""
import os, binascii, sys
# Variables
# buffer_size: how many bytes to check before/after when matching
# on the first of list_of_items below, to scan for the others
buffer_size = 512
# Redefine input path
if len(sys.argv) >= 2:
input_file = sys.argv[1]
else:
input_file = os.path.join("Pokemon Black.nds")
# below list corresponds to Castelia bottom shop first few items
list_of_items = [
'0E00', # heal ball
'0600', # net ball
'0800', # nest ball
'9000', # bridgemail s
'8900', # greet mail
'8A00', # favored mail
]
################################
with open(input_file, 'rb') as f:
data_raw = binascii.hexlify(f.read())
data_str = str(data_raw).upper()[2:-1] # get rid of Python "b" designation for byte strings
def find_all(data_str, sub):
start = 0
matches = []
while start * 2 < len(data_str):
data_slice = data_str[start:start+len(sub)] # len of sub should be 4, the length of the item we're checking
# print(data_slice)
finder = data_slice.find(sub)
if finder == -1:
pass
else:
matches.append(start)
start += 2
return matches
list_of_matches = find_all(data_str,list_of_items[0])
#list_of_matches = [x.start() for x in re.finditer(list_of_items[0], data_str)]
# then iterate over list_of_matches, and extend data by bytes (arbitrary range) to search for similar data areas
# score each "area" if others appear
match_scores = {}
for match in list_of_matches:
match_data = data_str[int(match-buffer_size/2):int(match+buffer_size/2)]
match_score = 0
for item in list_of_items:
if item in match_data:
match_score += 1
# then add entry to match_scores
match_scores[hex(int(match/2)).upper().replace("0X","").zfill(8)] = match_score
# Finally, filter for max scores, based on length of list_of_items
top_scores = [x for x in match_scores if match_scores[x] == len(list_of_items)]
|
