diff --git a/Ghidra/Features/Python/ghidra_scripts/ImportSymbolsScript.py b/Ghidra/Features/Python/ghidra_scripts/ImportSymbolsScript.py index cff4052e49..f0e952f62f 100644 --- a/Ghidra/Features/Python/ghidra_scripts/ImportSymbolsScript.py +++ b/Ghidra/Features/Python/ghidra_scripts/ImportSymbolsScript.py @@ -1,11 +1,38 @@ -#Imports a file with lines in the form "symbolName 0xADDRESS" -#@category Data -#@author - +# Imports a file with lines in the form "symbolName 0xADDRESS function_or_label" where "f" indicates a function and "l" a label +# @author unkown; edited by matedealer +# @category Data +# + +from ghidra.program.model.symbol.SourceType import * +import string + +functionManager = currentProgram.getFunctionManager() + f = askFile("Give me a file to open", "Go baby go!") for line in file(f.absolutePath): # note, cannot use open(), since that is in GhidraScript pieces = line.split() + + name = pieces[0] address = toAddr(long(pieces[1], 16)) - print "creating symbol", pieces[0], "at address", address - createLabel(address, pieces[0], False) + + try: + function_or_label = pieces[2] + except IndexError: + function_or_label = "l" + + + if function_or_label == "f": + func = functionManager.getFunctionAt(address) + + if func is not None: + old_name = func.getName() + func.setName(name, USER_DEFINED) + print("Renamed function {} to {} at address {}".format(old_name, name, address)) + else: + func = createFunction(address, name) + print("Created function {} at address {}".format(name, address)) + + else: + print("Created label {} at address {}".format(name, address)) + createLabel(address, name, False)