Merge pull request #470 from marco-c/count_opcode_usage

Implement Python tool that counts how many times a bytecode is used in a...

tests/opcodeCoverage.py

@@ -0,0 +1,79 @@
+#!/usr/bin/env python
+
+import sys
+import os
+import re
+
+from os.path import walk
+
+cache = {}
+
+table = ["nop", "aconst_null", "iconst_m1", "iconst_0", "iconst_1", "iconst_2", "iconst_3", "iconst_4", "iconst_5",
+         "lconst_0", "lconst_1", "fconst_0", "fconst_1", "fconst_2", "dconst_0", "dconst_1", "bipush", "sipush",
+         "ldc", "ldc_w", "ldc2_w", "iload", "lload", "fload", "dload", "aload", "iload_0", "iload_1", "iload_2",
+         "iload_3", "lload_0", "lload_1", "lload_2", "lload_3", "fload_0", "fload_1", "fload_2", "fload_3", "dload_0",
+         "dload_1", "dload_2", "dload_3", "aload_0", "aload_1", "aload_2", "aload_3", "iaload", "laload", "faload",
+         "daload", "aaload", "baload", "caload", "saload", "istore", "lstore", "fstore", "dstore", "astore",
+         "istore_0", "istore_1", "istore_2", "istore_3", "lstore_0", "lstore_1", "lstore_2", "lstore_3", "fstore_0",
+         "fstore_1", "fstore_2", "fstore_3", "dstore_0", "dstore_1", "dstore_2", "dstore_3", "astore_0", "astore_1",
+         "astore_2", "astore_3", "iastore", "lastore", "fastore", "dastore", "aastore", "bastore", "castore",
+         "sastore", "pop", "pop2", "dup", "dup_x1", "dup_x2", "dup2", "dup2_x1", "dup2_x2", "swap", "iadd", "ladd",
+         "fadd", "dadd", "isub", "lsub", "fsub", "dsub", "imul", "lmul", "fmul", "dmul", "idiv", "ldiv", "fdiv",
+         "ddiv", "irem", "lrem", "frem", "drem", "ineg", "lneg", "fneg", "dneg", "ishl", "lshl", "ishr", "lshr",
+         "iushr", "lushr", "iand", "land", "ior", "lor", "ixor", "lxor", "iinc", "i2l", "i2f", "i2d", "l2i", "l2f",
+         "l2d", "f2i", "f2l", "f2d", "d2i", "d2l", "d2f", "i2b", "i2c", "i2s", "lcmp", "fcmpl", "fcmpg", "dcmpl",
+         "dcmpg", "ifeq", "ifne", "iflt", "ifge", "ifgt", "ifle", "if_icmpeq", "if_icmpne", "if_icmplt", "if_icmpge",
+         "if_icmpgt", "if_icmple", "if_acmpeq", "if_acmpne", "goto", "jsr", "ret", "tableswitch", "lookupswitch",
+         "ireturn", "lreturn", "freturn", "dreturn", "areturn", "return", "getstatic", "putstatic", "getfield",
+         "putfield", "invokevirtual", "invokespecial", "invokestatic", "invokeinterface", "new", "newarray",
+         "anewarray", "arraylength", "athrow", "checkcast", "instanceof", "monitorenter", "monitorexit", "wide",
+         "multianewarray", "ifnull", "ifnonnull", "goto_w", "jsr_w"]
+
+def decompile(jar, path):
+  os.system("unzip " + jar + " -d " + path)
+
+  for root, dirs, files in os.walk(path):
+    for name in files:
+      if name.endswith(".class"):
+        print("Decompiling " + name + "...")
+        filePath = root + "/" + name[:-6]
+        os.system("javap -verbose -l -c -s -private " + filePath + " > " + filePath+".jbc")
+
+  for root, dirs, files in os.walk(path):
+    for name in files:
+      if not name.endswith(".jbc"):
+        os.remove(root + "/" + name)
+
+def readAll(path):
+  for root, dirs, files in os.walk(path):
+    for name in files:
+      if name.endswith(".jbc") and len(name) > 4:
+        cache[name[:-4]] = open(root + "/" + name).read()
+
+def countOpcodeUsage():
+  opcodes = {}
+  for opcode in table:
+    opcodes[opcode] = 0
+
+  for opcode in table:
+    for elem in cache:
+      opcodes[opcode] += cache[elem].split().count(opcode)
+
+  for opcode in sorted(opcodes, key=opcodes.get, reverse=True):
+    print(opcode + ": " + str(opcodes[opcode]))
+
+def main(argv):
+  if len(sys.argv) == 4 and sys.argv[1] == "dejar":
+    jar = sys.argv[2]
+    destPath = sys.argv[3]
+    decompile(jar, destPath)
+    return
+
+  path = sys.argv[1]
+
+  readAll(path)
+
+  countOpcodeUsage()
+
+if __name__ == "__main__":
+  main(sys.argv)