Loader bug fix

This commit is contained in:
Jack Andersen 2018-03-11 21:49:46 -10:00
parent 2fba10ed1b
commit c22fd5e30e
1 changed files with 17 additions and 15 deletions

View File

@ -1,46 +1,48 @@
doc = Document.getCurrentDocument()
file_path = Document.askFile('Select Object File', None, False)
file_path = Document.askFile('Select Objects File', None, False)
if file_path is not None:
with open(file_path, 'r') as file:
while True:
l = file.readline()
if l is None:
if not l:
break
parts = l.split(' ', 3)
address = int(parts[0], 16)
segment = doc.getSegmentAtAddress(address)
if segment is None:
continue
arr_count = int(parts[2], 10)
if parts[1] == 'FUNC':
if segment.getTypeAtAddress(address) != Segment.TYPE_PROCEDURE:
segment.markAsProcedure(address)
elif parts[1] == 'STR':
segment.setTypeAtAddress(address, parts[2], Segment.TYPE_ASCII)
segment.setTypeAtAddress(address, arr_count, Segment.TYPE_ASCII)
elif parts[1] == 'WSTR':
segment.setTypeAtAddress(address, parts[2], Segment.TYPE_UNICODE)
segment.setTypeAtAddress(address, arr_count, Segment.TYPE_UNICODE)
elif parts[1] == 'BYTE':
if parts[2]:
segment.setTypeAtAddress(address, parts[2], Segment.TYPE_INT8)
if arr_count:
segment.setTypeAtAddress(address, arr_count, Segment.TYPE_INT8)
else:
segment.setTypeAtAddress(address, 1, Segment.TYPE_INT8)
elif parts[1] == 'WORD':
if parts[2]:
segment.setTypeAtAddress(address, parts[2], Segment.TYPE_INT16)
if arr_count:
segment.setTypeAtAddress(address, arr_count, Segment.TYPE_INT16)
else:
segment.setTypeAtAddress(address, 2, Segment.TYPE_INT16)
elif parts[1] == 'DWORD':
if parts[2]:
segment.setTypeAtAddress(address, parts[2], Segment.TYPE_INT32)
if arr_count:
segment.setTypeAtAddress(address, arr_count, Segment.TYPE_INT32)
else:
segment.setTypeAtAddress(address, 4, Segment.TYPE_INT32)
elif parts[1] == 'FLOAT':
if parts[2]:
segment.setTypeAtAddress(address, parts[2], Segment.TYPE_INT32)
if arr_count:
segment.setTypeAtAddress(address, arr_count, Segment.TYPE_INT32)
else:
segment.setTypeAtAddress(address, 4, Segment.TYPE_INT32)
doc.setOperandFormat(address, 0, Document.FORMAT_FLOAT)
elif parts[1] == 'DOUBLE':
if parts[2]:
segment.setTypeAtAddress(address, parts[2], Segment.TYPE_INT64)
if arr_count:
segment.setTypeAtAddress(address, arr_count, Segment.TYPE_INT64)
else:
segment.setTypeAtAddress(address, 8, Segment.TYPE_INT64)
doc.setOperandFormat(address, 0, Document.FORMAT_FLOAT)