processrom.py

import idc
import idaapi
import re
import codecs

def processrom(min, max):
    if min > 0:
        min = min - 1
    curaddr = idc.find_unknown(min, idc.SEARCH_DOWN)
    while curaddr < max:
        if idc.add_func(curaddr) != True:
            idc.create_insn(curaddr)
        curaddr = idc.find_unknown(curaddr, idc.SEARCH_DOWN)

    return

def defineoffset(min, max):
    for addr in range(min, max, 4):
        idc.op_plain_offset(addr,1,0)

def defineasstruct(name, min, max):
    struct_id = idaapi.get_struc_id(name)
    if struct_id == idaapi.BADADDR:
        print("Structure {} does not exist".format(name))
        return
    print("Struct_id with name {} = {}".format(name, struct_id))
    struct_size = idc.get_struc_size(struct_id)
    print("size = {}".format(struct_size))
    for addr in range(min, max, struct_size):
        print("Set addr = {:02X} to struct".format(addr))
        idc.SetType(addr, name)

def undefcode(min, max):
    if min > 0:
        min = min - 1
    curaddr = idc.find_code(min, idc.SEARCH_DOWN)
    while curaddr < max:
        idc.del_items(curaddr)
        curaddr = idc.find_code(curaddr, idc.SEARCH_DOWN)

    return

def makedata(min, max, flags, size, tid):
    curaddr = min
    while curaddr < max:
        idc.del_items(curaddr, size, idaapi.DELIT_SIMPLE)
        idc.create_data(curaddr, flags, size, tid)
        curaddr += size

    return

def a2l(filename):
    lastvarname = ""
    lastaddress = ""
    with codecs.open(filename, 'r', encoding='ascii', errors='ignore') as fp:
        f = fp.read()
        measurements = f.split("/begin MEASUREMENT")
        measurements.pop(0)
        print("Found: %d measurement(s)" % len(measurements))
        for m in measurements:
            namefound = 0
            addrfound = 0
            name = ""
            addr = ""
            is_enum = False
            bit_mask = 0
            for l in m.split("\n"):
                l = l.strip()
                if (len(l) > 0):
                    if (namefound == 0):
                        name = l
                        namefound = 1
                    elif (l.startswith("BIT_MASK")):
                        is_enum = True
                        splitted = l.split(" ")
                        if len(splitted) < 2:
                            splitted = l.split("\t")
                        if len(splitted) > 1:
                            bit_mask = int(splitted[1], 16)
                    elif (l.startswith("ECU_ADDRESS")):
                        addr = l[12:]
                        addrfound = 1
                        break
                    elif (l.startswith("IF_DATA DIM_X")):
                        splitted = list(filter(None, l.split(" ")))
                        print(splitted)
                        if (splitted[2] == 'DIRECT'):
                            addr = splitted[3]
                        else:
                            addr = splitted[2]
                        addrfound = 1
                        break
            if (addrfound != 1):
                print("ERROR")
            elif (is_enum):
                enum_name = "enum_" + addr[2:]
                enum_id = idc.get_enum(enum_name)
                if (enum_id == 4294967295):
                    enum_id = idc.add_enum(-1, enum_name, 0)
                    idc.op_enum(int(addr, 0), -1, enum_id, 0)
                    idc.set_name(int(addr, 0), "", 0)
                if (not idc.is_bf(enum_id)):
                    idc.set_enum_bf(enum_id, 1)
                idc.add_enum_member(enum_id, name, bit_mask, bit_mask)
            else:
                idc.del_items(int(addr, 0), 1, idaapi.DELIT_SIMPLE)
                idc.set_name(int(addr, 0), name, 1)
        characteristics = f.split("/begin CHARACTERISTIC")
        characteristics.pop(0)
        print("Found: %d characteristics(s)" % len(characteristics))
        for m in characteristics:
            namefound = 0
            addrfound = 0
            name = ""
            addr = ""
            is_enum = False
            bit_mask = 0
            for l in m.split("\n"):
                l = l.strip()
                if (len(l) > 0):
                    if (namefound == 0):
                        name = l
                        namefound = 1
                    elif (l.startswith("/begin IF_DATA ASAP1B_CCP")):
                        print(l)
                        addr = l.split(" ")[5]
                        print(addr)
                        #addr = l[28:7]
                        addrfound = 1
                        break
            if (addrfound != 1):
                print("ERROR", name)
            else:
                print("addr", addr)
                idc.del_items(int(addr, 0), 1, idaapi.DELIT_SIMPLE)
                idc.set_name(int(addr, 0), name, 1)
    return

def a2l2(filename):
    lastvarname = ""
    lastaddress = ""
    with open(filename) as fp:
        f = fp.read()
        characteristics = f.split("/begin CHARACTERISTIC")
        characteristics.pop(0)
        print("Found: %d characteristics(s)" % len(characteristics))
        for m in characteristics:
            namefound = 0
            addrfound = 0
            name = ""
            addr = ""
            is_enum = False
            bit_mask = 0
            addr_next = False
            for l in m.split("\n"):
                l = l.strip()
                if (len(l) > 0):
                    if (namefound == 0):
                        l = l.split()
                        name = l[0]
                        namefound = 1
                    elif addr_next:
                        l = l.split()
                        addr = l[0]
                        addrfound = 1
                        addr_next = False
                        print(addr)
                        break
                    elif (l.startswith("VALUE") or l.startswith("CURVE") or l.startswith("MAP")):
                        addr_next = True
            if (addrfound != 1):
                print(name, "addr not found")
            else:
                print(name, "addr", addr)
                idc.del_items(int(addr, 0), 1, idaapi.DELIT_SIMPLE)
                idc.set_name(int(addr, 0), name, 1)
    return

def get_addr_from_disasm(line):
    match = re.search("_+[0-9a-fA-F]+", line)
    if match:
        return int(match.group(0)[1:],16)


def set_memory_dword():
    curaddr = get_screen_ea()
    curline = idc.GetDisasm(get_screen_ea())
    addr_to_set = get_addr_from_disasm(curline)
    nextline = idc.GetDisasm(get_screen_ea() + 4)
    addr_from_set = get_addr_from_disasm(nextline)
    print(addr_to_set, addr_from_set)
    if addr_to_set and addr_from_set:
        patch_dword(addr_to_set, addr_from_set)
    else:
        nextline = idc.GetDisasm(get_screen_ea() + 8)
        addr_from_set = get_addr_from_disasm(nextline)
        if addr_to_set and addr_from_set:
            patch_dword(addr_to_set, addr_from_set)

def set_memory_dword2():
    curaddr = get_screen_ea()
    curline = idc.GetDisasm(get_screen_ea())
    addr_to_set = get_addr_from_disasm(curline)
    nextline = idc.GetDisasm(get_screen_ea() + 8)
    addr_from_set = get_addr_from_disasm(nextline)
    print(addr_to_set, addr_from_set)
    if addr_to_set and addr_from_set:
        patch_dword(addr_to_set, addr_from_set)

def get_can_id_from_can_packet_struct(curaddr):
    can_id = 0
    uar = idc.get_wide_word(curaddr + 0x0E)
    lar = idc.get_wide_word(curaddr + 0x10)
    can_id |= (uar & 0x00FF) << 21          # ID28..21
    can_id |= ((uar >> 13) & 0x07) << 18    # ID20..18
    can_id |= ((uar >> 8)  & 0x1F) << 13    # ID17..13

    can_id |= (lar & 0x00FF) << 5           # ID12..5
    can_id |= (lar >> 11) & 0x1F           # ID4..0
    return can_id

def comment_can_id_c167():
    curaddr = get_screen_ea()
    can_id = get_can_id_from_can_packet_struct(curaddr)
    idc.set_cmt(curaddr, "can_id: %X" % can_id, False)