RedMetaClassAnalyzer/Z_createBiosParserServicesVtable.py at main · NyanCatTW1/RedMetaClassAnalyzer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
# A helper script to populate vtable in BiosParserServices
# @author Nyan Cat
# @category A_Red
# @keybinding
# @menupath
# @toolbar

# pyright: reportMissingImports=false
# pyright: reportUndefinedVariable=false

from java.util import ArrayList

from ghidra.program.model.data import CategoryPath, FunctionDefinitionDataType, PointerDataType
from ghidra.program.model.data import DataTypeConflictHandler


def getDataType(typeName, typeManager):
    matches = ArrayList()
    typeManager.findDataTypes(typeName, matches)

    if len(matches) > 1:
        print("Warning: Using the first " + typeName)
    elif len(matches) == 0:
        return None

    return matches[0]


funcManager = currentProgram.getFunctionManager()
typeManager = currentProgram.getDataTypeManager()

funcNames = ["allocateMemory", "deallocateMemory", "readReg32", "writeReg32", "readPll32",
             "writePll32", "readMc32", "writeMc32", "readPci8", "writePci8",
             "readPci16", "writePci16", "readPci32", "writePci32", "delayMicro",
             "delayMilli", "printOut"]
funcNames = ["AtomBiosProxy::" + x for x in funcNames]
done = [False] * len(funcNames)
ptrs = [None] * len(funcNames)

for func in funcManager.getFunctions(True):
    funcName = str(func)
    if funcName not in funcNames:
        continue

    funcIndex = funcNames.index(funcName)
    funcSign = func.getSignature(False)

    funcType = FunctionDefinitionDataType(CategoryPath("/AMDGen/FuncSigns"), funcName + "_sign", funcSign)
    funcType = typeManager.addDataType(funcType, DataTypeConflictHandler.REPLACE_HANDLER)
    ptrType = PointerDataType(funcType, typeManager)
    ptrType = typeManager.addDataType(ptrType, DataTypeConflictHandler.REPLACE_HANDLER)
    done[funcIndex] = True
    ptrs[funcIndex] = ptrType

    print(funcName + "_sign *")

assert all(done)

biosParser = getDataType("BiosParserServices", typeManager)
for i in range(len(funcNames)):
    biosParser.replaceAtOffset(32 + 8 * i, ptrs[i], 8, funcNames[i], "Generated by Z_createBiosParserServicesVtable.py")