You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.7 KiB
52 lines
1.7 KiB
import sys
|
|
import os
|
|
#import pefile
|
|
import io
|
|
import string
|
|
|
|
def main():
|
|
path = os.path.dirname(os.path.realpath(sys.argv[0]))
|
|
file = sys.argv[1]
|
|
|
|
proxies = io.StringIO()
|
|
proxy_add = io.StringIO()
|
|
proxy_def = io.StringIO()
|
|
proxy_def_file = io.StringIO()
|
|
|
|
count = 0
|
|
|
|
""" for dll in dlls:
|
|
pe = pefile.PE(dll, fast_load=True)
|
|
pe.parse_data_directories()
|
|
for exp in pe.DIRECTORY_ENTRY_EXPORT.symbols:
|
|
name = exp.name.decode()
|
|
|
|
proxies.write("%s\n" % name)
|
|
proxy_add.write("ADD_ORIGINAL(%d, %s);\n" % (count, name))
|
|
proxy_def.write("PROXY(%d, %s);\n" % (count, name))
|
|
count = count + 1 """
|
|
|
|
with open(file, "r") as includes:
|
|
names = includes.readlines()
|
|
for name in names:
|
|
name = name.strip()
|
|
proxies.write(f"{name}\n")
|
|
proxy_add.write(f"ADD_ORIGINAL({count}, {name});\n")
|
|
proxy_def.write(f"PROXY({count}, {name});\n")
|
|
proxy_def_file.write(f" {name} @{count+1}\n")
|
|
count = count + 1
|
|
|
|
with open(f"{path}\\templates\\proxy_template.c", "r") as proxy_file:
|
|
new_proxy = string.Template(proxy_file.read()).safe_substitute(proxy_count=count, proxy_add=proxy_add.getvalue(), proxy_def=proxy_def.getvalue())
|
|
|
|
with open(f"{path}\\..\\Proxy\\proxy.c", "w") as proxy_c_file:
|
|
proxy_c_file.write(new_proxy)
|
|
|
|
with open(f"{path}\\templates\\proxy_template.def", "r") as proxy_file:
|
|
new_proxy_def = string.Template(proxy_file.read()).safe_substitute(proxy_exports=proxy_def_file.getvalue())
|
|
|
|
with open(f"{path}\\..\\Proxy\\proxy.def", "w") as proxy_def_file:
|
|
proxy_def_file.write(new_proxy_def)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|