import csv

fileName = ida_nalt.get_root_filename()

addrStart = idc.find_binary(0x10000, SEARCH_DOWN,"26 f4 00 10 46 f4")
A6_max = ida_ua.get_immvals(addrStart + 0x4,-1)[0]
addSeg = (addrStart//0x10000)*0x10000
addr1 = idc.find_binary(addrStart, SEARCH_DOWN,"06 f4")

if addrStart == 0xffffffff or not(addr1 < addrStart + 0x14):
    print("search error")
else:
    if ida_ua.get_immvals(addr1,-1)[0]> 0xc000:
        addrMap = 0x0000 + ida_ua.get_immvals(addr1,-1)[0]
    else:
        addrMap = 0x10000 + ida_ua.get_immvals(addr1,-1)[0]


    with open("A6_addrFunc_addrRAM_bmask.csv", mode="w", encoding='ANSI') as w_file:
        file_writer = csv.writer(w_file, delimiter = ",", lineterminator = '\r')
    
        addrVal = []
        for i in range(A6_max+0x1):
            
            ida_bytes.del_items(addrMap+2*i,2)
            ida_bytes.create_word(addrMap+2*i,2,True)
            adddrFunc = addSeg+ida_bytes.get_word(addrMap+2*i)
            comA6 = 0x1000+i+0xA60000
            com_NotA6 = 0x1000+i
            ida_name.set_name(addrMap+2*i, '{:x}'.format(comA6))

            if ida_ua.create_insn(adddrFunc) == 0:
                ida_bytes.del_items(adddrFunc,6)
                ida_ua.create_insn(adddrFunc)
            
            insn_mnem = ida_ua.print_insn_mnem(adddrFunc)
            bmask = '0x0000'

            while insn_mnem != 'rets':
                addrRAM = ida_xref.get_first_dref_from(adddrFunc)
                if addrRAM != 0xffffffff:
                    break
                else:
                    adddrFunc = ida_search.find_code(adddrFunc,SEARCH_DOWN)
                    insn_mnem = ida_ua.print_insn_mnem(adddrFunc)
                    if insn_mnem == 'bmov':
                        bmask = '0x{:x}'.format(2**(idc.get_bytes(adddrFunc,4)[3]>>4))
                    else:
                        bmask = '0x0000'
           
            file_writer.writerow(['{:x}'.format(com_NotA6),'0x{:x}'.format(addSeg+ida_bytes.get_word(addrMap+2*i)), '0x{:x}'.format(addrRAM),bmask])