jcfa/src/jp/igapyon/jcfa/JavaClassFileAnalyzer.java

   1 package jp.igapyon.jcfa;\r
   2 \r
   3 import java.io.File;\r
   4 import java.io.IOException;\r
   5 \r
   6 import jp.igapyon.jcfa.util.JcfaUtil;\r
   7 import jp.igapyon.jcfa.util.JcfaWriteUtil;\r
   8 import jp.igapyon.jcfa.vo.JcfaClass;\r
   9 import jp.igapyon.jcfa.vo.JcfaField;\r
  10 import jp.igapyon.jcfa.vo.JcfaMethod;\r
  11 import jp.igapyon.jcfa.vo.JcfaUnit;\r
  12 \r
  13 import org.apache.bcel.Constants;\r
  14 import org.apache.bcel.classfile.ClassFormatException;\r
  15 import org.apache.bcel.classfile.ClassParser;\r
  16 import org.apache.bcel.classfile.Code;\r
  17 import org.apache.bcel.classfile.ConstantValue;\r
  18 import org.apache.bcel.classfile.Field;\r
  19 import org.apache.bcel.classfile.JavaClass;\r
  20 import org.apache.bcel.classfile.Method;\r
  21 \r
  22 public class JavaClassFileAnalyzer {\r
  23         protected JcfaUnit jcfaUnit = new JcfaUnit();\r
  24 \r
  25         public static final void main(final String[] args) {\r
  26                 new JavaClassFileAnalyzer().parseDir(new File("./bin/test"));\r
  27         }\r
  28 \r
  29         private void parseDir(final File dir) {\r
  30                 final File[] files = dir.listFiles();\r
  31                 if (files == null) {\r
  32                         return;\r
  33                 }\r
  34                 for (File file : files) {\r
  35                         if (file.isDirectory()) {\r
  36                                 parseDir(file);\r
  37                         }\r
  38                         if (file.isFile()) {\r
  39                                 if (file.getName().endsWith(".class")) {\r
  40                                         new JavaClassFileAnalyzer().process(file, new File(\r
  41                                                         "./testJavaClass/output"));\r
  42                                 }\r
  43                         }\r
  44                 }\r
  45         }\r
  46 \r
  47         private final void process(final File inputFile, final File outputDir) {\r
  48                 try {\r
  49                         final JavaClass jc = new ClassParser(inputFile.getCanonicalPath())\r
  50                                         .parse();\r
  51                         final JcfaClass jcfaClass = new JcfaClass();\r
  52                         jcfaUnit.getClassList().add(jcfaClass);\r
  53 \r
  54                         jcfaClass.setName(jc.getClassName());\r
  55                         jcfaClass.setExtendsName(jc.getSuperclassName());\r
  56 \r
  57                         jcfaClass.getComment().getCommentList()\r
  58                                         .add("TODO import func. is missing.");\r
  59 \r
  60                         final String[] split = jc.getClassName().split("\\.");\r
  61                         File actualyTargetDir = outputDir;\r
  62                         if (split.length > 1) {\r
  63                                 for (int index = 0; index < split.length - 1; index++) {\r
  64                                         actualyTargetDir = new File(outputDir, split[index]);\r
  65                                         actualyTargetDir.mkdirs();\r
  66                                 }\r
  67                         }\r
  68 \r
  69                         analyzeFields(jc, jcfaClass);\r
  70                         analyzeMethods(jc, jcfaClass);\r
  71 \r
  72                         jcfaUnit.setTargetFile(new File(actualyTargetDir,\r
  73                                         split[split.length - 1] + ".jcfa"));\r
  74 \r
  75                         JcfaWriteUtil.writeToFile(jcfaUnit);\r
  76                 } catch (ClassFormatException e) {\r
  77                         e.printStackTrace();\r
  78                 } catch (IOException e) {\r
  79                         e.printStackTrace();\r
  80                 }\r
  81         }\r
  82 \r
  83         private void analyzeFields(final JavaClass jc, final JcfaClass jcfaClass) {\r
  84                 final org.apache.bcel.classfile.Field[] fields = jc.getFields();\r
  85                 for (int indexField = 0; indexField < fields.length; indexField++) {\r
  86                         final Field field = fields[indexField];\r
  87                         analyzeField(jc, field, jcfaClass);\r
  88                 }\r
  89         }\r
  90 \r
  91         private void analyzeField(final JavaClass jc, final Field field,\r
  92                         final JcfaClass jcfaClass) {\r
  93                 final JcfaField jcfaField = new JcfaField();\r
  94                 jcfaField.setName(field.getName());\r
  95                 jcfaClass.getFieldList().add(jcfaField);\r
  96 \r
  97                 // TODO type should be more collect.\r
  98                 jcfaField.setType(field.getType().toString());\r
  99 \r
 100                 String access = "";\r
 101                 access += field.isPublic() ? "public " : "";\r
 102                 access += field.isProtected() ? "protected " : "";\r
 103                 access += field.isPrivate() ? "private " : "";\r
 104                 access += field.isAbstract() ? "abstract " : "";\r
 105                 access += field.isStatic() ? "static " : "";\r
 106                 access += field.isVolatile() ? "volatile " : "";\r
 107                 access += field.isFinal() ? "final " : "";\r
 108                 jcfaField.setAccess(access);\r
 109 \r
 110                 String constValue = null;\r
 111                 final ConstantValue cv = field.getConstantValue();\r
 112                 if (cv != null) {\r
 113                         jcfaField.setConstantValue("\""\r
 114                                         + jc.getConstantPool().getConstantString(\r
 115                                                         cv.getConstantValueIndex(),\r
 116                                                         Constants.CONSTANT_String) + "\"");\r
 117 \r
 118                         jcfaField\r
 119                                         .getComment()\r
 120                                         .getCommentList()\r
 121                                         .add("FIXME other type support is missing. <br />Now only String.");\r
 122                 }\r
 123         }\r
 124 \r
 125         private void analyzeMethods(final JavaClass jc, final JcfaClass jcfaClass)\r
 126                         throws IOException {\r
 127                 final org.apache.bcel.classfile.Method[] methods = jc.getMethods();\r
 128                 for (int indexMethod = 0; indexMethod < methods.length; indexMethod++) {\r
 129                         final Method method = methods[indexMethod];\r
 130                         analyzeMethod(jc, method, jcfaClass);\r
 131 \r
 132                 }\r
 133         }\r
 134 \r
 135         private void analyzeMethod(final JavaClass jc, final Method method,\r
 136                         final JcfaClass jcfaClass) throws IOException {\r
 137                 final JcfaMethod jcfaMethod = new JcfaMethod();\r
 138                 jcfaClass.getMethodList().add(jcfaMethod);\r
 139 \r
 140                 jcfaMethod.setName(method.getName());\r
 141 \r
 142                 final Code code = method.getCode();\r
 143                 if (code == null) {\r
 144                         return;\r
 145                 }\r
 146 \r
 147                 final byte[] codes = code.getCode();\r
 148                 for (int pc = 0; pc < codes.length; pc++) {\r
 149                         final short opcode = JcfaUtil.byte2UnsignedByte(codes[pc]);\r
 150                         short operands = Constants.NO_OF_OPERANDS[opcode];\r
 151                         if (operands < 0) {\r
 152                                 System.err.println("  TODO negative value:"\r
 153                                                 + Constants.OPCODE_NAMES[opcode] + ": " + operands);\r
 154                                 // break;\r
 155                         }\r
 156 \r
 157                         switch (opcode) {\r
 158                         case Constants.RETURN: {\r
 159                                 System.out.println("    " + pc + ": "\r
 160                                                 + Constants.OPCODE_NAMES[opcode]);\r
 161                                 break;\r
 162                         }\r
 163                         case Constants.GETSTATIC: {\r
 164                                 System.out.println("    "\r
 165                                                 + pc\r
 166                                                 + ": "\r
 167                                                 + Constants.OPCODE_NAMES[opcode]\r
 168                                                 + ": "\r
 169                                                 + JcfaUtil.getConstantFieldrefString(jc, codes[pc + 1],\r
 170                                                                 codes[pc + 2]));\r
 171                                 break;\r
 172                         }\r
 173                         case Constants.LDC: {\r
 174                                 System.out.println("    " + pc + ": "\r
 175                                                 + Constants.OPCODE_NAMES[opcode] + ": "\r
 176                                                 + JcfaUtil.getConstantString(jc, codes[pc + 1]));\r
 177                         }\r
 178                                 break;\r
 179                         case Constants.INVOKEVIRTUAL:\r
 180                         case Constants.INVOKESPECIAL: {\r
 181                                 final int operand = JcfaUtil.byte2UnsignedShort(codes[pc + 1],\r
 182                                                 codes[pc + 2]);\r
 183                                 System.out.println("    " + pc + ": "\r
 184                                                 + Constants.OPCODE_NAMES[opcode] + ": "\r
 185                                                 + JcfaUtil.getConstantMethodRefString(jc, operand));\r
 186                         }\r
 187                                 break;\r
 188                         case Constants.LOOKUPSWITCH:\r
 189                                 int result = JcfaUtil.byte2Int(codes[pc + 1], codes[pc + 2],\r
 190                                                 codes[pc + 3], codes[pc + 4]);\r
 191                                 System.out.println("  TODO skipping bytes: " + (result));\r
 192 \r
 193                                 int lookupOp = pc + 5;\r
 194 \r
 195                                 short diff = JcfaUtil.byte2UnsignedByte(codes[lookupOp++]);\r
 196                                 System.out.println("  TODO skipping bytes: " + (diff));\r
 197 \r
 198                                 int loopCount = JcfaUtil\r
 199                                                 .byte2Int(codes[lookupOp++], codes[lookupOp++],\r
 200                                                                 codes[lookupOp++], codes[lookupOp++]);\r
 201                                 for (int index = 0; index < loopCount; index++) {\r
 202                                         System.out.println("      "\r
 203                                                         + JcfaUtil.byte2Int(codes[lookupOp++],\r
 204                                                                         codes[lookupOp++], codes[lookupOp++],\r
 205                                                                         codes[lookupOp++])\r
 206                                                         + ":"\r
 207                                                         + (JcfaUtil.byte2Int(codes[lookupOp++],\r
 208                                                                         codes[lookupOp++], codes[lookupOp++],\r
 209                                                                         codes[lookupOp++]) + pc));\r
 210                                 }\r
 211 \r
 212                                 short diff2 = JcfaUtil.byte2UnsignedByte(codes[lookupOp++]);\r
 213                                 System.out.println("  TODO skipping bytes: " + (diff2));\r
 214 \r
 215                                 operands += (lookupOp - pc);\r
 216 \r
 217                                 break;\r
 218                         default: {\r
 219                                 System.out.println("    " + pc + ": "\r
 220                                                 + Constants.OPCODE_NAMES[opcode] + " (" + operands\r
 221                                                 + ")");\r
 222                         }\r
 223                                 break;\r
 224                         }\r
 225                         pc += operands;\r
 226                 }\r
 227         }\r
 228 }