-
Notifications
You must be signed in to change notification settings - Fork 0
/
generate_script.py
58 lines (45 loc) · 1.78 KB
/
generate_script.py
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
import os
import subprocess
def contains(pattern, content):
return (len(content.split(pattern)) != 1)
# changing our template to fit the context of the challenge
def change(old, new, content):
file_to_change = content.split(old)
new_file = ""
for i in range(len(file_to_change)):
new_file += file_to_change[i]
if i != len(file_to_change) - 1:
new_file += new
return new_file
def iter_changes(olds, news, content):
for (old, new) in [(olds[i], news[i]) for i in range(len(olds))]:
content = change(old, new, content)
return content
def contains_libc(content, contains = False):
if not contains:
new_content = change('libc = ELF(b\'./libc.so.6\')', '', content)
return new_content
return content
def modify_template(binary_name, ip, port, contains):
new_file = ''
with open('solve.py', 'r') as f:
content = f.read()
new_file = iter_changes(['binary', '\'ip\'', 'porto'], [binary_name, '\'' + ip + '\'', port], content)
new_file = contains_libc(new_file, contains)
with open('solve_' + binary_name + '.py', 'w') as f:
f.write(new_file)
def main():
print("[*] Collecting data for the template generation")
binary_name = input("select binary name : ")
ip = input("select ip for the challenge : ")
port = input("select port for the challenge : ")
ls_res = subprocess.check_output("ls", shell = True).decode()
contains_libc = bool(contains('libc.so.6', ls_res))
print("[+] Data acquired!")
print("[*] Generating the solve script")
os.system('touch solve_' + binary_name + '.py')
modify_template(binary_name, ip, port, contains_libc)
os.system('rm solve.py')
print("[+] Solve script generated!")
if __name__ == '__main__':
main()