IDAPython plugin - ARMv7-A CPSR_C enum

ida-armv7a-cpsr_c.py

from idautils import *
from idaapi import *
from idc import *

Message("%s starting...\n" % __file__)

CPSR_C_enum_id = GetEnum("CPSR_C_enum")
if CPSR_C_enum_id == BADADDR:
	CPSR_C_enum_id = AddEnum(-1, "CPSR_C_enum", hexflag())
	SetEnumBf(CPSR_C_enum_id,1);
	AddConstEx(CPSR_C_enum_id,"CPSR_MODE_USR", 0b10000, 0x1F)
	AddConstEx(CPSR_C_enum_id,"CPSR_MODE_FIQ", 0b10001, 0x1F)
	AddConstEx(CPSR_C_enum_id,"CPSR_MODE_IRQ", 0b10010, 0x1F)
	AddConstEx(CPSR_C_enum_id,"CPSR_MODE_SVC", 0b10011, 0x1F)
	AddConstEx(CPSR_C_enum_id,"CPSR_MODE_MON", 0b10110, 0x1F)
	AddConstEx(CPSR_C_enum_id,"CPSR_MODE_ABT", 0b10111, 0x1F)
	AddConstEx(CPSR_C_enum_id,"CPSR_MODE_HYP", 0b11010, 0x1F)
	AddConstEx(CPSR_C_enum_id,"CPSR_MODE_UND", 0b11011, 0x1F)
	AddConstEx(CPSR_C_enum_id,"CPSR_MODE_SYS", 0b11111, 0x1F)
	AddConstEx(CPSR_C_enum_id,"CPSR_THUMB", 0x20, 0x20)
	AddConstEx(CPSR_C_enum_id,"CPSR_MASK_FIQ", 0x40, 0x40)
	AddConstEx(CPSR_C_enum_id,"CPSR_MASK_IRQ", 0x80, 0x80)

ea = MinEA()
while True:
	ea= FindText(ea, SEARCH_DOWN|SEARCH_NEXT|SEARCH_REGEX, 0, 0, " (CPS|MSR) ")
	if ea == BADADDR:
		break
	mnem = GetMnem(ea)
	if mnem == "MSR":
		if GetOpnd(ea, 0) == "CPSR_c" and GetOpType(ea, 1) == o_imm:
			OpEnumEx(ea, 1, CPSR_C_enum_id, 0)
			Message("%s MSR CPSR_c, %s\n" % (hex(ea),GetOpnd(ea, 1)))
	elif mnem == "CPS":
		OpEnumEx(ea, 0, CPSR_C_enum_id, 0)
		Message("%s CPS %s\n" % (hex(ea),GetOpnd(ea, 0)))
	else:
		Message("Unrecognized instruction @ %s : %s\n" % (hex(ea), GetDisasm(ea)))
	ea = NextHead(ea)
Message("Done!\n")