113 lines
3.3 KiB
Python
113 lines
3.3 KiB
Python
from elftools.elf.elffile import ELFFile
|
||
import re,os
|
||
|
||
hdr_str=["#include \"opensdk.h\"\n"]
|
||
symbol_table_org=[]
|
||
symbol_table = []
|
||
gen_table=[]
|
||
|
||
def get_export_func(f):
|
||
file = open(f, 'r')
|
||
try:
|
||
while True:
|
||
line = file.readline()
|
||
if line:
|
||
# print ("line=",line)
|
||
tmp = re.findall(r"EXPORT_FUNC\((.+)\)",line)
|
||
if tmp:
|
||
print(tmp[0])
|
||
symbol_table_org.append(line)
|
||
symbol_table.append(tmp[0])
|
||
print("#define app_"+tmp[0]+"_addr ")
|
||
else:
|
||
break
|
||
finally:
|
||
file.close()
|
||
print(symbol_table)
|
||
print("get_export_func")
|
||
|
||
def get_info_from_elf(e_file): # e_file为需要读取的elf文件
|
||
tmp_dict = {} # 这个字典用于存放变量名和变量地址的对应关系
|
||
file = open(e_file, 'rb')
|
||
elf_file_obj = ELFFile(file)
|
||
|
||
data_section = elf_file_obj.get_section_by_name('.data')
|
||
data_address = data_section.header["sh_addr"]
|
||
print(data_section.data())
|
||
print("data addr:0x%X"%data_address)
|
||
dfile = open("test.data.bin", 'wb')
|
||
dfile.write(data_section.data());
|
||
dfile.close()
|
||
|
||
text_section = elf_file_obj.get_section_by_name(".text")
|
||
base_address = text_section.header["sh_addr"]
|
||
print(text_section.data())
|
||
print("text addr:0x%X"%base_address)
|
||
tfile = open("test.text.bin", 'wb')
|
||
tfile.write(text_section.data());
|
||
tfile.close()
|
||
|
||
# got_section = elf_file_obj.get_section_by_name('.got')
|
||
# got_address = got_section.header["sh_addr"]
|
||
# print(got_section.data())
|
||
# print("got addr:0x%X"%got_address)
|
||
# gfile = open("test.got.bin", 'wb')
|
||
# gfile.write(got_section.data());
|
||
# gfile.close()
|
||
|
||
# plt_section = elf_file_obj.get_section_by_name('.got.plt')
|
||
# plt_address = plt_section.header["sh_addr"]
|
||
# print(plt_section.data())
|
||
# print("plt addr:0x%X"%plt_address)
|
||
# pfile = open("test.plt.bin", 'wb')
|
||
# pfile.write(plt_section.data());
|
||
# pfile.close()
|
||
|
||
for section in elf_file_obj.iter_sections():
|
||
# 此处把地址以16进制的字符串形式存入字典，可选择自己需要的格式。
|
||
print(section)
|
||
# tmp_dict[s.name] = str(hex(s.entry["st_value"]))section
|
||
file.close()
|
||
return tmp_dict
|
||
|
||
def gen_hdr(e_file):
|
||
print("gen header file")
|
||
tmp_dict = {}
|
||
file = open(e_file, 'rb')
|
||
elf_file_obj = ELFFile(file)
|
||
|
||
symbol_table_name = elf_file_obj.get_section_by_name(".symtab")
|
||
for sym in symbol_table_name.iter_symbols():
|
||
print(sym.name)
|
||
print(sym['st_value'])
|
||
if sym.name in symbol_table:
|
||
gen_table.append("#define app_"+sym.name+"_addr "+str(hex(sym['st_value']))+"\n")
|
||
print("insert one sym")
|
||
print(gen_table)
|
||
|
||
f = open("export_new.h", 'w')
|
||
for line in hdr_str:
|
||
f.writelines(line)
|
||
|
||
f.writelines("\n")
|
||
for line in symbol_table_org:
|
||
f.writelines(line)
|
||
|
||
f.writelines("\n")
|
||
|
||
for line in gen_table:
|
||
f.writelines(line)
|
||
|
||
f.close()
|
||
|
||
try:
|
||
os.remove("inc/export.h")
|
||
except:
|
||
print("no inc/export.h")
|
||
os.rename("export_new.h","inc/export.h")
|
||
|
||
get_info_from_elf("test.elf")
|
||
get_export_func("inc/export.h")
|
||
gen_hdr("test.elf")
|
||
|