Added changes ot asm_helper

dev/feature_extraction/cpc/asm_helper.py

@@ -342,6 +342,8 @@
                 arg_reg_xmm1 + arg_reg_xmm2 + arg_reg_xmm3 + arg_reg_xmm4 +\
                 arg_reg_xmm5 + arg_reg_xmm6 + arg_reg_xmm7
 
+reg_a = ['rax', 'eax', 'ax', 'ah', 'al']
+
 #-------------------------------------
 #Basic arg finding
 #-------------------------------------
@@ -356,6 +358,11 @@ def remove_prefixes(mnemonic):
     new = re.sub('lock', '', new)
     return new.strip()
 
+def is_reg_a(opnd):
+    if opnd in reg_a:
+        return True
+    return False
+
 def is_mov(mnemonic):
     if mov_prefix in mnemonic:
         return True
@@ -390,6 +397,9 @@ def is_jmp(mnemonic):
         return True
     return False
 
+def is_jump(mnemonic):
+    return is_jcc(mnemonic) or is_jmp(mnemonic)
+
 def is_ret(mnemonic):
     mnemonic = remove_prefixes(mnemonic)
     if mnemonic in ret_insts: