some small fixes

CTFs_and_WarGames/2014-CSAW-CTF/reverse-engineering/eggshells-100/eggshells-master/capstone.py

@@ -0,0 +1,46 @@
+import utils
+from capstone import *
+
+def arch(arch):
+    if arch == 'arm':
+        return CS_ARCH_ARM
+    elif arch == 'arm64':
+        return CS_ARCH_ARM64
+    elif arch == 'mips':
+        return CS_ARCH_MIPS
+    elif arch == 'x86':
+        return CS_ARCH_X86
+
+
+def mode(mode):
+    if mode == 'arm':
+        return CS_MODE_ARM
+    elif mode == 'thumb':
+        return CS_MODE_THUMB
+    elif mode == '16' or mode == 16:
+        return CS_MODE_16
+    elif mode == '32' or mode == 32:
+        return CS_MODE_32
+    elif mode == '64' or mode == 64:
+        return CS_MODE_32
+
+
+def disassemble(code, _arch, _mode):
+    _arch = arch(_arch)
+    _mode = mode(_mode)
+
+    md = Cs(_arch, _mode)
+
+    disassembly = []
+
+    for i in md.disasm(CODE, 0x0000000):
+        disassembly.append(
+            (i.address, len(str(i.bytes).encode('hex')) / 2, str(i.bytes).encode('hex'), i.mnemonic, i.op_str))
+
+    return disassembly
+
+
+CODE = "\x90\x90\x90"
+
+for x in disassemble(CODE, 'x86', '64'):
+    print "0x%08x (%02x) %-20s %s %s" % (x[0], x[1], x[2], x[3], x[4])