Add comments and local variables

This commit is contained in:
Jack Andersen 2018-03-12 01:06:11 -10:00
parent c22fd5e30e
commit 1e949d4575
4 changed files with 14077 additions and 3862 deletions

File diff suppressed because it is too large Load Diff

File diff suppressed because it is too large Load Diff

File diff suppressed because it is too large Load Diff

View File

@ -1,49 +0,0 @@
doc = Document.getCurrentDocument()
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 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, arr_count, Segment.TYPE_ASCII)
elif parts[1] == 'WSTR':
segment.setTypeAtAddress(address, arr_count, Segment.TYPE_UNICODE)
elif parts[1] == 'BYTE':
if arr_count:
segment.setTypeAtAddress(address, arr_count, Segment.TYPE_INT8)
else:
segment.setTypeAtAddress(address, 1, Segment.TYPE_INT8)
elif parts[1] == 'WORD':
if arr_count:
segment.setTypeAtAddress(address, arr_count, Segment.TYPE_INT16)
else:
segment.setTypeAtAddress(address, 2, Segment.TYPE_INT16)
elif parts[1] == 'DWORD':
if arr_count:
segment.setTypeAtAddress(address, arr_count, Segment.TYPE_INT32)
else:
segment.setTypeAtAddress(address, 4, Segment.TYPE_INT32)
elif parts[1] == 'FLOAT':
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 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)
segment.setNameAtAddress(address, parts[3])