parse_one_r.py

import re, arm_translate, utils

def getAllSpLinesForLow(lines, before_reg_count, after_reg_count, real_reg_count):
    to_write = []

    # ищем строки sub (add) sp, #a => sub (add) sp, #a+new_regs_count*4 => to_write
    sub_add_sp_lines = list(filter(None,[re.search('.*(add|sub)(.w)?\s*sp, #[0-9]+', line) for line in lines]))

    # если строки нет, выходим (потом подумать,как сделать) todo
    if len(sub_add_sp_lines) !=2: #не нашли sub и add
        return []

    a = int(sub_add_sp_lines[0].group().split('#')[-1])

    #ищем строки вида [sp, #b]
    use_sp_lines = list(filter(None,[re.search('.*(ldr|str)(b|h|sb|sh)?(.w)?.*\[sp, #[0-9]+\].*', line) for line in lines]))

    #todo
    for i in use_sp_lines:
        print(i.group())
    if len([s for s in use_sp_lines if '!' in str(s.group())])>0:
        return -1


    for l in use_sp_lines:
        instr = re.search('v?(ldr|str)(b|h|sb|sh)?(.w)?',l.group()).group()
        b = int(re.search('#[0-9]+', l.group()).group().replace('#',''))
        # если b-a >= -new_regs_count*4 => [sp, #b + new_regs_count*4] => to_write - иначе ничего
        if b-a >= 0:
            rx = re.search('(\s+r10|r11|r12|sp|lr|pc|r[0-9]|((d|s)(([1-2][0-9])|3[0-1]|[0-9]))),', l.group()).group().strip().replace(',','')
            code, is_thumb = utils.getCodeFromLine(l.group())
            if b-a < real_reg_count * 4:
                offset = b + before_reg_count*4
            if b - a >= real_reg_count * 4:
                offset = b + before_reg_count*4 + after_reg_count*4
            new_instr_code = arm_translate.makeLdrOrStr(instr,code , rx, 'sp', offset, is_thumb)
            # to_write ... [sp, #b + new_regs_count*4]
            to_write.append((utils.getAddressFromLine(l.group()), len(code) // 2, utils.toLittleEndian(new_instr_code)))

    #ищем строки вида add rx, sp, (#c) - должна быть одна ? todo
    add_sp_to_reg = list(filter(None, [re.search('.*(add|mov)\s*(r[1-9]|r10|r11|r12), sp(, #[1-9]+)?.*', line) for line in lines]))

    #todo
    for i in add_sp_to_reg:
        print(i.group())
    if len([s for s in add_sp_to_reg if '!' in str(s.group())])>0:
        return -1

    if len(add_sp_to_reg) > 0:
        if len(add_sp_to_reg) > 1:
            print ('Больше одной строки add rx, sp, (#c)')
        line = add_sp_to_reg[0].group()
        c = re.search('#[1-9]+', line)
        c = 0 if c is None else int(c.group()[1:])
        reg = re.search('r11|r10|r12|r[1-9]',line).group()

        # Ищем строки вида [reg, #d]
        use_reg = list(filter(None, [re.search('.*(ldr|str)(b|h|sb|sh|sw)?(.w)?.*\\[{0}, #\\-?[0-9]+\\]'.format(reg), line) for line in lines]))
        for l in use_reg:
            instr = re.search('v?(ldr|str)(b|h|sb|sh)?(.w)?',l.group()).group()
            d = int(re.search('#-?[0-9]+', l.group()).group().replace('#',''))
            # если d < 0 => если c-d<0 =>[reg, #d-new_regs_count*4]
            if c + d >= 0: #todo а что если будет sub rx, sp?
                rx = re.search('(\s+r10|r11|r12|sp|lr|pc|r[0-9]|((d|s)(([1-2][0-9])|3[0-1]|[0-9]))),',
                               l.group()).group().strip().replace(',', '')
                code, is_thumb = utils.getCodeFromLine(l.group())
                if c+d < real_reg_count * 4:
                    offset = d + before_reg_count * 4
                if c+d >= real_reg_count * 4:
                    offset = d + before_reg_count * 4 + after_reg_count * 4
                new_instr_code = arm_translate.makeLdrOrStr(instr, code, rx, reg, offset, is_thumb)
                #to_write ... [reg, #d-new_regs_count*4]
                to_write.append((utils.getAddressFromLine(l.group()), len(code) // 2, utils.toLittleEndian(new_instr_code)))

    #for i in to_write:
        #print(i)
    return to_write