Merge pull request nihilus#6 from nologic/master

pull req
Tomasuh · Nov 20, 2015 · 731803d · 731803d
2 parents c2c429c + 8c08daa
commit 731803d
Showing 1 changed file with 15 additions and 14 deletions.
diff --git a/idaref.py b/idaref.py
@@ -162,20 +162,21 @@ def OnClose(self):
         time.sleep(1)
 
     def cleanInstruction(self, inst):
-        inst = inst.upper()
-        # hacks for x86
-        if(inst[0:1] == 'J' and inst != 'JMP'):
-            inst = "Jcc"
-        elif(inst[0:4] == "LOOP"):
-            inst = "LOOP"
-        elif(inst[0:3] == "INT"):
-            inst = "INT n"
-        elif(inst[0:5] == "FCMOV"):
-            inst = "FCMOVcc"
-        elif(inst[0:4] == "CMOV"):
-            inst = "CMOVcc"
-        elif(inst[0:3] == "SET"):
-            inst = "SETcc"
+        if(self.arch == "x86-64"):
+            inst = inst.upper()
+            # hacks for x86
+            if(inst[0:1] == 'J' and inst != 'JMP'):
+                inst = "Jcc"
+            elif(inst[0:4] == "LOOP"):
+                inst = "LOOP"
+            elif(inst[0:3] == "INT"):
+                inst = "INT n"
+            elif(inst[0:5] == "FCMOV"):
+                inst = "FCMOVcc"
+            elif(inst[0:4] == "CMOV"):
+                inst = "CMOVcc"
+            elif(inst[0:3] == "SET"):
+                inst = "SETcc"
 
         return inst