53 lines
1.8 KiB
Java
53 lines
1.8 KiB
Java
//TODO write a description for this script
|
|
//@author
|
|
//@category _NEW_
|
|
//@keybinding
|
|
//@menupath
|
|
//@toolbar bomb
|
|
|
|
|
|
import java.util.List;
|
|
import java.util.regex.Pattern;
|
|
|
|
import ghidra.app.script.GhidraScript;
|
|
import ghidra.program.model.listing.Function;
|
|
|
|
public class SetCallingConventionBulk extends GhidraScript {
|
|
|
|
public void run() throws Exception {
|
|
|
|
String callingConvention = BulkUtils.askForCallingConvention(this);
|
|
|
|
Pattern pattern = BulkUtils.askForRegex(this, "entire function name");
|
|
List<Function> matchingFunctions = BulkUtils.getFunctionsByRegex(this, pattern);
|
|
int skipped = BulkUtils.removeFunctionsByCallingConvention(matchingFunctions, callingConvention);
|
|
int numOfMatches = matchingFunctions.size();
|
|
// only abort if there are no matches AND no matches have been skipped, so the users understands why there are no matches
|
|
if(numOfMatches == 0 && skipped == 0){
|
|
println("Abort: no matches found");
|
|
return;
|
|
}
|
|
|
|
String title = "Set calling convention of " + matchingFunctions.size() + " function(s) to " + callingConvention;
|
|
boolean confirmed = BulkUtils.showFunctionConfirmationDialog(matchingFunctions, title, skipped, "Number of matches skipped (calling convention already matches): %d");
|
|
if (!confirmed) {
|
|
println("Abort: negative confirmation");
|
|
return;
|
|
}
|
|
|
|
int success = 0;
|
|
int fail = 0;
|
|
for (Function func : matchingFunctions) {
|
|
try {
|
|
func.setCallingConvention(callingConvention);
|
|
success++;
|
|
} catch (Exception e) {
|
|
println("Failed to update function " + func.getName() + " at " + func.getEntryPoint() + " - " + e.getMessage());
|
|
fail++;
|
|
}
|
|
}
|
|
BulkUtils.printDoneMessage(this, success, fail);
|
|
}
|
|
|
|
}
|