FunctionsRename.java

/*
 * Copyright (c) 2024 Abc Xyz — All Rights Reserved
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
//Rename functions from file
//@author saruman9
//@category Functions
//@keybinding
//@menupath
//@toolbar

import ghidra.program.model.address.Address;
import ghidra.program.model.listing.Function;
import ghidra.program.model.listing.FunctionManager;
import ghidra.program.model.symbol.SourceType;

import java.util.Scanner;

import ghidra.app.script.GhidraScript;;

public class FunctionsRename extends GhidraScript {

    @Override
    protected void run() throws Exception {
        FunctionManager functionManager = currentProgram.getFunctionManager();
        Scanner scanner = new Scanner(askFile("Functions file", "Open"));
        while (scanner.hasNextLine()) {
            String line = scanner.nextLine();
            String[] address_name = line.split(" : ");
            Address address = toAddr(address_name[0]);
            String name = address_name[1];
            Function function = functionManager.getFunctionContaining(address);
            if (function != null) {
                String old_name = function.getName();
                if (function.getSymbol().getSource() == SourceType.DEFAULT) {
                    if (old_name.contains("FUN") && !old_name.contains("thunk")) {
                        println(String.format("%s -> %s (%x)", function.getName(), name,
                            address.getOffset()));
                        function.setName(name, SourceType.USER_DEFINED);
                    }
                }
            }
            else {
                printerr(String.format("NULL: %s (%x)\n", name, address.getOffset()));
            }
        }
        scanner.close();
    }
}