OSDN Git Service

save
[jcfa/jcfa.git] / jcfa / src / jp / igapyon / jcfa / JavaClassFileAnalyzer.java
index c2fd3b1..d5c9c11 100644 (file)
 package jp.igapyon.jcfa;\r
 \r
+import java.io.ByteArrayOutputStream;\r
+import java.io.File;\r
 import java.io.IOException;\r
 \r
 import jp.igapyon.jcfa.util.JcfaUtil;\r
+import jp.igapyon.jcfa.util.JcfaWriteUtil;\r
+import jp.igapyon.jcfa.vo.JcfaClass;\r
+import jp.igapyon.jcfa.vo.JcfaCode;\r
+import jp.igapyon.jcfa.vo.JcfaField;\r
+import jp.igapyon.jcfa.vo.JcfaMethod;\r
+import jp.igapyon.jcfa.vo.JcfaUnit;\r
 \r
 import org.apache.bcel.Constants;\r
-/**\r
- * Apache Jakarta BCELを用いたクラスファイル(バイトコード)解析サンプル\r
- */\r
 import org.apache.bcel.classfile.ClassFormatException;\r
 import org.apache.bcel.classfile.ClassParser;\r
 import org.apache.bcel.classfile.Code;\r
+import org.apache.bcel.classfile.ConstantValue;\r
+import org.apache.bcel.classfile.Field;\r
 import org.apache.bcel.classfile.JavaClass;\r
 import org.apache.bcel.classfile.Method;\r
+import org.apache.bcel.generic.Type;\r
 \r
 public class JavaClassFileAnalyzer {\r
-       private static final String CLASS_MODULE = "./bin/test/TestJavaClass001.class";\r
+       protected JcfaUnit jcfaUnit = new JcfaUnit();\r
 \r
        public static final void main(final String[] args) {\r
-               new JavaClassFileAnalyzer().process();\r
+               new JavaClassFileAnalyzer().parseDir(new File("./bin/test"));\r
+               new JavaClassFileAnalyzer().parseDir(new File(\r
+                               "./bin/jp/igapyon/jcfa/vo"));\r
        }\r
 \r
-       private final void process() {\r
-               try {\r
-                       final JavaClass jc = new ClassParser(CLASS_MODULE).parse();\r
-                       System.out.println("Classname   : " + jc.getClassName());\r
-                       System.out.println("Parent class: " + jc.getSuperclassName());\r
-\r
-                       final org.apache.bcel.classfile.Method[] methods = jc.getMethods();\r
-                       for (int indexMethod = 0; indexMethod < methods.length; indexMethod++) {\r
-                               final Method method = methods[indexMethod];\r
-                               System.out.println("  Method:" + method.getName());\r
-                               final Code code = method.getCode();\r
-                               if (code == null) {\r
-                                       continue;\r
+       private void parseDir(final File dir) {\r
+               final File[] files = dir.listFiles();\r
+               if (files == null) {\r
+                       return;\r
+               }\r
+               for (File file : files) {\r
+                       if (file.isDirectory()) {\r
+                               parseDir(file);\r
+                       }\r
+                       if (file.isFile()) {\r
+                               if (file.getName().endsWith(".class")) {\r
+                                       new JavaClassFileAnalyzer().process(file, new File(\r
+                                                       "./testJavaClass/output"));\r
                                }\r
+                       }\r
+               }\r
+       }\r
+\r
+       private final void process(final File inputFile, final File outputDir) {\r
+               parseUnit(inputFile, outputDir);\r
+       }\r
 \r
-                               final byte[] codes = code.getCode();\r
-                               for (int pc = 0; pc < codes.length; pc++) {\r
-                                       final short opcode = JcfaUtil.byte2UnsignedByte(codes[pc]);\r
-                                       short operands = Constants.NO_OF_OPERANDS[opcode];\r
-                                       if (operands < 0) {\r
-                                               System.out.println("  TODO negative value:"\r
-                                                               + Constants.OPCODE_NAMES[opcode] + ": "\r
-                                                               + operands);\r
-                                               // break;\r
-                                       }\r
-\r
-                                       switch (opcode) {\r
-                                       case Constants.RETURN: {\r
-                                               System.out.println("    " + pc + ": "\r
-                                                               + Constants.OPCODE_NAMES[opcode]);\r
-                                               break;\r
-                                       }\r
-                                       case Constants.GETSTATIC: {\r
-                                               System.out.println("    "\r
-                                                               + pc\r
-                                                               + ": "\r
-                                                               + Constants.OPCODE_NAMES[opcode]\r
-                                                               + ": "\r
-                                                               + JcfaUtil.getConstantFieldrefString(jc,\r
-                                                                               codes[pc + 1], codes[pc + 2]));\r
-                                               break;\r
-                                       }\r
-                                       case Constants.LDC: {\r
-                                               System.out\r
-                                                               .println("    "\r
-                                                                               + pc\r
-                                                                               + ": "\r
-                                                                               + Constants.OPCODE_NAMES[opcode]\r
-                                                                               + ": "\r
-                                                                               + JcfaUtil.getConstantString(jc,\r
-                                                                                               codes[pc + 1]));\r
-                                       }\r
-                                               break;\r
-                                       case Constants.INVOKEVIRTUAL:\r
-                                       case Constants.INVOKESPECIAL: {\r
-                                               final int operand = JcfaUtil.byte2UnsignedShort(\r
-                                                               codes[pc + 1], codes[pc + 2]);\r
-                                               System.out.println("    "\r
-                                                               + pc\r
-                                                               + ": "\r
-                                                               + Constants.OPCODE_NAMES[opcode]\r
-                                                               + ": "\r
-                                                               + JcfaUtil.getConstantMethodRefString(jc,\r
-                                                                               operand));\r
-                                       }\r
-                                               break;\r
-                                       case Constants.LOOKUPSWITCH:\r
-                                               int result = JcfaUtil.byte2Int(codes[pc + 1],\r
-                                                               codes[pc + 2], codes[pc + 3], codes[pc + 4]);\r
-                                               System.out\r
-                                                               .println("  TODO skipping bytes: " + (result));\r
-\r
-                                               int lookupOp = pc + 5;\r
-\r
-                                               short diff = JcfaUtil\r
-                                                               .byte2UnsignedByte(codes[lookupOp++]);\r
-                                               System.out.println("  TODO skipping bytes: " + (diff));\r
-\r
-                                               int loopCount = JcfaUtil.byte2Int(codes[lookupOp++],\r
-                                                               codes[lookupOp++], codes[lookupOp++],\r
-                                                               codes[lookupOp++]);\r
-                                               for (int index = 0; index < loopCount; index++) {\r
-                                                       System.out.println("      "\r
-                                                                       + JcfaUtil.byte2Int(codes[lookupOp++],\r
-                                                                                       codes[lookupOp++],\r
-                                                                                       codes[lookupOp++],\r
-                                                                                       codes[lookupOp++])\r
-                                                                       + ":"\r
-                                                                       + (JcfaUtil.byte2Int(codes[lookupOp++],\r
-                                                                                       codes[lookupOp++],\r
-                                                                                       codes[lookupOp++],\r
-                                                                                       codes[lookupOp++]) + pc));\r
-                                               }\r
-\r
-                                               short diff2 = JcfaUtil\r
-                                                               .byte2UnsignedByte(codes[lookupOp++]);\r
-                                               System.out.println("  TODO skipping bytes: " + (diff2));\r
-\r
-                                               operands += (lookupOp - pc);\r
-\r
-                                               break;\r
-                                       default: {\r
-                                               System.out.println("    " + pc + ": "\r
-                                                               + Constants.OPCODE_NAMES[opcode] + " ("\r
-                                                               + operands + ")");\r
-                                       }\r
-                                               break;\r
-                                       }\r
-                                       pc += operands;\r
+       private void parseUnit(final File inputFile, final File outputDir) {\r
+               try {\r
+                       final JavaClass jc = new ClassParser(inputFile.getCanonicalPath())\r
+                                       .parse();\r
+                       final JcfaClass jcfaClass = new JcfaClass();\r
+                       jcfaUnit.getClassList().add(jcfaClass);\r
+\r
+                       jcfaClass.setName(jc.getClassName());\r
+                       jcfaClass.setExtendsName(jc.getSuperclassName());\r
+\r
+                       jcfaClass.getComment().getCommentList()\r
+                                       .add("TODO import func. is missing.");\r
+                       jcfaClass.getComment().setJavaDoc(true);\r
+\r
+                       final String[] split = jc.getClassName().split("\\.");\r
+                       File actualyTargetDir = outputDir;\r
+                       if (split.length > 1) {\r
+                               for (int index = 0; index < split.length - 1; index++) {\r
+                                       actualyTargetDir = new File(actualyTargetDir, split[index]);\r
+                                       actualyTargetDir.mkdirs();\r
                                }\r
                        }\r
+\r
+                       parseFields(jc, jcfaClass);\r
+                       parseMethods(jc, jcfaClass);\r
+\r
+                       jcfaUnit.setTargetFile(new File(actualyTargetDir,\r
+                                       split[split.length - 1] + ".jcfa"));\r
+\r
+                       JcfaWriteUtil.writeToFile(jcfaUnit);\r
                } catch (ClassFormatException e) {\r
                        e.printStackTrace();\r
                } catch (IOException e) {\r
                        e.printStackTrace();\r
                }\r
        }\r
+\r
+       private void parseFields(final JavaClass jc, final JcfaClass jcfaClass) {\r
+               final org.apache.bcel.classfile.Field[] fields = jc.getFields();\r
+               for (int indexField = 0; indexField < fields.length; indexField++) {\r
+                       final Field field = fields[indexField];\r
+                       parseField(jc, field, jcfaClass);\r
+               }\r
+       }\r
+\r
+       private void parseField(final JavaClass jc, final Field field,\r
+                       final JcfaClass jcfaClass) {\r
+               final JcfaField jcfaField = new JcfaField();\r
+               jcfaField.setName(field.getName());\r
+               jcfaClass.getFieldList().add(jcfaField);\r
+\r
+               jcfaField.getComment().setJavaDoc(true);\r
+\r
+               // TODO type should be more collect.\r
+               jcfaField.setType(field.getType().toString());\r
+\r
+               String access = "";\r
+               access += field.isPublic() ? "public " : "";\r
+               access += field.isProtected() ? "protected " : "";\r
+               access += field.isPrivate() ? "private " : "";\r
+               access += field.isAbstract() ? "abstract " : "";\r
+               access += field.isStatic() ? "static " : "";\r
+               access += field.isVolatile() ? "volatile " : "";\r
+               access += field.isFinal() ? "final " : "";\r
+               jcfaField.setAccess(access);\r
+\r
+               final ConstantValue cv = field.getConstantValue();\r
+               if (cv != null) {\r
+                       jcfaField.setConstantValue("\""\r
+                                       + jc.getConstantPool().getConstantString(\r
+                                                       cv.getConstantValueIndex(),\r
+                                                       Constants.CONSTANT_String) + "\"");\r
+\r
+                       jcfaField\r
+                                       .getComment()\r
+                                       .getCommentList()\r
+                                       .add("FIXME other type support is missing. <br />Now only String.");\r
+               }\r
+       }\r
+\r
+       private void parseMethods(final JavaClass jc, final JcfaClass jcfaClass)\r
+                       throws IOException {\r
+               final org.apache.bcel.classfile.Method[] methods = jc.getMethods();\r
+               for (int indexMethod = 0; indexMethod < methods.length; indexMethod++) {\r
+                       final Method method = methods[indexMethod];\r
+                       parseMethod(jc, method, jcfaClass);\r
+\r
+               }\r
+       }\r
+\r
+       /**\r
+        * Analyze method.\r
+        * \r
+        * @param jc\r
+        * @param method\r
+        * @param jcfaClass\r
+        * @throws IOException\r
+        */\r
+       private void parseMethod(final JavaClass jc, final Method method,\r
+                       final JcfaClass jcfaClass) throws IOException {\r
+               final JcfaMethod jcfaMethod = new JcfaMethod();\r
+               jcfaClass.getMethodList().add(jcfaMethod);\r
+\r
+               jcfaMethod.setName(method.getName());\r
+               jcfaMethod.getComment().setJavaDoc(true);\r
+               if (jcfaMethod.getName().equals("<init>")) {\r
+                       jcfaMethod.getComment().getCommentList()\r
+                                       .add("Default constructor.");\r
+               } else {\r
+                       jcfaMethod.getComment().getCommentList().add("Method.");\r
+               }\r
+\r
+               for (Type type : method.getArgumentTypes()) {\r
+                       jcfaMethod.getComment().getCommentList().add(type.toString());\r
+                       jcfaMethod.getArugumentTypeList().add(type.toString());\r
+               }\r
+               jcfaMethod.setType(method.getReturnType().toString());\r
+\r
+               final Code code = method.getCode();\r
+               if (code == null) {\r
+                       return;\r
+               }\r
+\r
+               final byte[] codes = code.getCode();\r
+               for (int pc = 0; pc < codes.length; pc++) {\r
+                       final int operands = parseCodes(jc, method, jcfaClass, jcfaMethod,\r
+                                       pc, codes);\r
+                       if (operands < 0) {\r
+                               break;\r
+                       }\r
+\r
+                       pc += operands;\r
+               }\r
+       }\r
+\r
+       private int parseCodes(final JavaClass jc, final Method method,\r
+                       final JcfaClass jcfaClass, final JcfaMethod jcfaMethod,\r
+                       final int pc, final byte[] codes) throws IOException {\r
+               final JcfaCode jcfaCode = new JcfaCode();\r
+               jcfaMethod.getCodeList().add(jcfaCode);\r
+               jcfaCode.setJavaClass(jc);\r
+\r
+               jcfaCode.setOpcode(JcfaUtil.byte2UnsignedByte(codes[pc]));\r
+               jcfaCode.getComment()\r
+                               .getCommentList()\r
+                               .add("" + pc + ": "\r
+                                               + Constants.OPCODE_NAMES[jcfaCode.getOpcode()]);\r
+\r
+               short operands = Constants.NO_OF_OPERANDS[jcfaCode.getOpcode()];\r
+               if (operands < 0) {\r
+                       jcfaCode.getComment()\r
+                                       .getCommentList()\r
+                                       .add("FIXME NO_OF_OPERANDS has negative value:"\r
+                                                       + Constants.OPCODE_NAMES[jcfaCode.getOpcode()]\r
+                                                       + ": " + operands);\r
+                       return operands;\r
+               }\r
+\r
+               switch (jcfaCode.getOpcode()) {\r
+               case Constants.RETURN:\r
+                       break;\r
+               case Constants.GETSTATIC:\r
+                       break;\r
+               case Constants.LDC:\r
+                       break;\r
+               case Constants.INVOKEVIRTUAL:\r
+               case Constants.INVOKESPECIAL:\r
+                       break;\r
+               case Constants.LOOKUPSWITCH: {\r
+                       int result = JcfaUtil.byte2Int(codes[pc + 1], codes[pc + 2],\r
+                                       codes[pc + 3], codes[pc + 4]);\r
+                       System.out.println("  TODO skipping bytes: " + (result));\r
+\r
+                       int lookupOp = pc + 5;\r
+\r
+                       short diff = JcfaUtil.byte2UnsignedByte(codes[lookupOp++]);\r
+                       System.out.println("  TODO skipping bytes: " + (diff));\r
+\r
+                       int loopCount = JcfaUtil.byte2Int(codes[lookupOp++],\r
+                                       codes[lookupOp++], codes[lookupOp++], codes[lookupOp++]);\r
+\r
+                       short diff2 = JcfaUtil.byte2UnsignedByte(codes[lookupOp++]);\r
+                       jcfaCode.getComment().getCommentList()\r
+                                       .add("  TODO skipping bytes: " + (diff2));\r
+\r
+                       operands += (lookupOp - pc);\r
+               }\r
+                       break;\r
+               default:\r
+                       break;\r
+               }\r
+\r
+               {\r
+                       final ByteArrayOutputStream outStream = new ByteArrayOutputStream();\r
+                       outStream.write(codes, pc, operands + 1);\r
+                       outStream.flush();\r
+                       jcfaCode.setCodes(outStream.toByteArray());\r
+               }\r
+               return operands;\r
+       }\r
 }
\ No newline at end of file