unageanu-javaclass 0.2.1 → 0.4.0

data/sample/caller.rb

@@ -0,0 +1,95 @@
+#!/usr/bin/ruby
+
+require "rubygems"
+require "zip/zip"
+require "javaclass"
+require "kconv"
+
+# クラスプール
+class ClassPool
+  def initialize
+    @classes = {}
+    @classes_r = {}
+  end
+  def <<( jc )
+    jc.methods.each {|m|
+      key = jc.name + "#" + m.name + m.descriptor
+      @classes[key] ||= []
+      next unless m.attributes.key? "Code"
+      m.attributes["Code"].codes.each {|code|
+        case code.opcode
+          when 0xB6,0xB7,0xB8,0xB9
+            m = jc.get_constant( code.operands[0].value )
+            add_caller( key,
+              m.class_name.name + "#" +
+              m.name_and_type.name + m.name_and_type.descriptor
+            )
+          when 0xB2,0xB3,0xB4,0xB5
+            m = jc.get_constant( code.operands[0].value )
+            add_caller( key,
+              m.class_name.name + "." + m.name_and_type.name
+            )
+        end
+      }
+    }
+  end
+  attr_reader :classes
+  attr_reader :classes_r
+private
+  def add_caller( from, to )
+    @classes[from] ||= []
+    @classes[from] << to
+    @classes_r[to] ||= []
+    @classes_r[to] << from
+  end
+end
+
+#ノード
+class Node
+  def initialize( name  )
+    @name = name
+    @children = []
+  end
+  def <<(child)
+    @children << child
+  end
+  def to_s( indent="" )
+    str = indent.dup
+    str << name.to_s << "\n"
+    child_indent = indent.gsub(/├/, "│").gsub(/└/, "　")
+    @children.each_index {|i|
+       next if children[i] == nil
+       tmp = child_indent + ( i >= children.length-1 ? "└" : "├" )
+       str << children[i].to_s( tmp )
+    }
+    return str
+  end
+  attr :name, true
+  attr :children, true
+end
+
+#呼び出し元を収集する。
+def collect( pool, start, stop=[], node=nil, checked=[] )
+  node ||= Node.new(start.to_s)
+  return node unless pool.classes_r.key? start
+  pool.classes_r[start].each {|caller|
+    next if checked.find{|i| i==caller }
+    next if stop && stop.find{|i| caller =~ i }
+    checked.push start.to_s
+    node << collect( pool, caller, stop, Node.new( caller ), checked ) 
+    checked.pop
+  }
+  return node
+end
+
+# クラス情報を収集
+pool = ClassPool.new
+JavaClass::Utils.each_class( *ARGV[0].split(";") ){|jc|
+  pool << jc
+}
+# 呼び出し元をツリー表示
+puts collect(pool,  ARGV[1]).to_s
+
+ 
+
+

data/sample/java_src/com/example/caller/A.java

@@ -0,0 +1,19 @@
+package com.example.caller;
+
+public class A {
+    public void aaa() {
+        aaa2();
+        new B().bbb();
+        C.ccc();
+    }
+    void aaa2() {
+        aaa3();
+    }
+    private void aaa3() {
+        new D().ddd();
+    }
+    void aaa4() {
+        aaa3();
+        aaa4();
+    }
+}

data/sample/java_src/com/example/caller/B.java

@@ -0,0 +1,7 @@
+package com.example.caller;
+
+public class B {
+    public void bbb() {
+        new D().ddd();
+    }
+}

data/sample/java_src/com/example/caller/C.java

@@ -0,0 +1,7 @@
+package com.example.caller;
+
+public class C {
+    public static void ccc() {
+        new D().ddd();
+    }
+}

data/sample/java_src/com/example/caller/CallSample.java

@@ -0,0 +1,23 @@
+package com.example.caller;
+
+public class CallSample {
+    static class A {
+        void aaa() {
+            new B().bbb();
+        }
+        void aaa2() {
+            new C().ccc();
+        }
+        void aaa3() {
+            aaa2();
+        }
+    }
+    static class B {
+        void bbb() {
+            new C().ccc();
+        }
+    }
+    static class C {
+        void ccc() {}
+    }
+}

data/sample/java_src/com/example/caller/D.java

@@ -0,0 +1,5 @@
+package com.example.caller;
+
+public class D {
+    public void ddd() {}
+}

data/sample/java_src/com/example/code/Statement.java

@@ -0,0 +1,41 @@
+package com.example.code;
+
+import java.net.URI;
+import java.net.URISyntaxException;
+
+public class Statement {
+    
+    public void basic( int i ) {
+        if ( i < 10 ) System.out.println( "a" );
+        if ( i > 10 ) {
+            System.out.println( "b" );
+        } else {
+            System.out.println( "c" );
+        }
+        for ( int j=0;j<10;j++ ) {
+            System.out.println( j );
+        }
+        for ( String str : new String[]{"a","b","c"} ) {
+            System.out.println( str );
+        }
+        int j = 9;
+        while ( j-- > 10 ) {
+            System.out.println( j );
+        }
+        switch (i) {
+            case 1 : System.out.println( "1" ); break;
+            case 2 : System.out.println( "2" );
+            default : System.out.println( "3" );
+        }
+        try {
+            new URI( "http://foo" + i );
+        } catch ( RuntimeException e ) {
+            e.printStackTrace();
+        } catch (URISyntaxException e) {
+            e.printStackTrace();
+        } finally {
+            System.out.println( "x" );
+        }
+    }
+    
+}

data/sample/sample.rb

@@ -17,11 +17,11 @@ require "kconv"
   "./java_class/com/example/Deprecated.class",
   "./java_class/com/example/annotation/Annotated.class",
   "./java_class/com/example/annotation/HasDefaultValue.class",
-  "./java_class/com/example/annotation/AnnotatedMethod.class"
+  "./java_class/com/example/annotation/AnnotatedMethod.class",
+  "./java_class/com/example/code/Statement.class"
 ].each { |c|
   open( c, "r+b" ) {|io|
     jc = JavaClass.from io
-    puts jc.to_s.tosjis
-    puts ""
+    puts jc.to_s
   }
 }

data/test/all_tests.rb

@@ -0,0 +1,9 @@
+require 'rubygems'
+require 'runit/testcase'
+require 'runit/cui/testrunner'
+
+require 'test_access_flag'
+require 'test_attribute'
+require 'test_class'
+require 'test_constant'
+require 'test_member'

data/test/java_src/com/example/TestClass1.java

@@ -3,6 +3,8 @@ package com.example;
 import java.io.Closeable;
 import java.io.IOException;
 import java.io.Serializable;
+import java.net.URI;
+import java.net.URISyntaxException;
 import java.util.ArrayList;
 import java.util.List;
 
@@ -25,4 +27,37 @@ implements Serializable, Closeable{
     static boolean booleanConstant  = false;
     
     public void close() throws IOException {}
+    
+    public void statement( int i ) {
+        if ( i < 10 ) System.out.println( "a" );
+        if ( i > 10 ) {
+            System.out.println( "b" );
+        } else {
+            System.out.println( "c" );
+        }
+        for ( int j=0;j<10;j++ ) {
+            System.out.println( j );
+        }
+        for ( String str : new String[]{"a","b","c"} ) {
+            System.out.println( str );
+        }
+        int j = 9;
+        while ( j-- > 10 ) {
+            System.out.println( j );
+        }
+        switch (i) {
+            case 1 : System.out.println( "1" ); break;
+            case 2 : System.out.println( "2" );
+            default : System.out.println( "3" );
+        }
+        try {
+            new URI( "http://foo" + i );
+        } catch ( RuntimeException e ) {
+            e.printStackTrace();
+        } catch (URISyntaxException e) {
+            e.printStackTrace();
+        } finally {
+            System.out.println( "x" );
+        }
+    }
 }

data/test/test_attribute.rb

@@ -36,6 +36,7 @@ module JavaClass
       @java_class.constant_pool[14]  = UTF8Constant.new( @java_class, Constant::CONSTANT_Utf8, "LineNumberTable" )
       @java_class.constant_pool[15]  = UTF8Constant.new( @java_class, Constant::CONSTANT_Utf8, "LocalVariableTable" )
       @java_class.constant_pool[16]  = UTF8Constant.new( @java_class, Constant::CONSTANT_Utf8, "LocalVariableTypeTable" )
+      @java_class.constant_pool[17]  = UTF8Constant.new( @java_class, Constant::CONSTANT_Utf8, "StackMapTable" )
 
       @java_class.constant_pool[30]  = UTF8Constant.new( @java_class, Constant::CONSTANT_Utf8, "aaa" )
       @java_class.constant_pool[31]  = UTF8Constant.new( @java_class, Constant::CONSTANT_Utf8, "bbb" )
@@ -274,12 +275,13 @@ module JavaClass
     def test_CodeAttribute
       
       exceptions = [Excpetion.new( @java_class, 1, 10, 16, 103 )]
-      attr = CodeAttribute.new( @java_class, 13, 10, 8, [0x05, 0x06], exceptions, {} )
+      codes = [Code.new(@java_class, 0, 0x05), Code.new(@java_class, 1, 0x06)]
+      attr = CodeAttribute.new( @java_class, 13, 10, 8, codes, exceptions, {} )
       assert_attribute( attr ) {|a|
         assert_equal a.name, "Code"
         assert_equal a.max_stack, 10
         assert_equal a.max_locals, 8
-        assert_equal a.codes, [0x05, 0x06]
+        assert_equal a.codes, codes
         assert_equal a.exception_table, exceptions
         assert_equal a.attributes, {}
         assert_equal a.dump, "000D0000 0016000A 00080000 00020506\n00010001 000A0010 00670000"
@@ -896,6 +898,139 @@ module JavaClass
         assert_equal a.to_bytes, [0x00, 0x66, 0x00, 0x65, 0x00, 0x1F, 0x02, 0x01]
       }
     end
+
+    #
+    #=== StackMapTableAttributeのテスト
+    #
+    def test_StackMapTableAttribute
+      
+       vals = [
+        VariableInfo.new( 3 ),
+        ObjectVariableInfo.new( 7, 10 ),
+        UninitializedVariableInfo .new( 8, 2 )
+      ]
+      entries = [
+        SameFrame.new( 3 ),
+        SameLocals1StackItemFrame.new( 70, [UninitializedVariableInfo .new( 8, 2 )] ),
+        FullFrame.new( 255, 3, vals, vals[0..1] )
+      ]
+      attr = StackMapTableAttribute.new( @java_class, 17, entries )
+      assert_attribute( attr ) {|a|
+        assert_equal a.name, "StackMapTable"
+        assert_equal a.stack_map_frame_entries, entries
+        assert_equal a.dump, "00110000 00190003 03460800 02FF0003\n"+
+                                          "00030307 000A0800 02000203 07000A"
+        assert_equal a.to_bytes, [
+          0x00, 0x11, 0x00, 0x00, 0x00, 0x19, 0x00, 0x03, 0x03, 0x46, 0x08, 0x00, 
+          0x02, 0xFF, 0x00, 0x03, 0x00, 0x03, 0x03, 0x07, 0x00, 0x0A, 0x08, 0x00,
+          0x02, 0x00, 0x02, 0x03, 0x07, 0x00, 0x0A
+        ]
+      }
+
+      attr.stack_map_frame_entries = []
+      assert_attribute( attr ) {|a|
+        assert_equal a.name, "StackMapTable"
+        assert_equal a.stack_map_frame_entries, []
+        assert_equal a.dump, "00110000 00020000"
+        assert_equal a.to_bytes, [0x00, 0x11, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00]
+      }
+    end
+    
+    #
+    #=== StackMapFrameのテスト
+    #
+    def test_StackMapFrame
+      
+      vals = [
+        VariableInfo.new( 3 ),
+        ObjectVariableInfo.new( 7, 10 ),
+        UninitializedVariableInfo .new( 8, 2 )
+      ]
+      
+      value = SameFrame.new( 3 )
+      assert_stack_map_frame( value ) {|a|
+        assert_equal a.frame_type, 3
+        assert_equal a.dump, "03"
+        assert_equal a.to_bytes, [0x03]
+      }
+      value = SameLocals1StackItemFrame.new( 70, vals[0..0] )
+      assert_stack_map_frame( value ) {|a|
+        assert_equal a.frame_type, 70
+        assert_equal a.verification_type_info, vals[0..0]
+        assert_equal a.dump, "4603"
+        assert_equal a.to_bytes, [0x46, 0x03]
+      }
+      value = SameLocals1StackItemFrameExtended.new( 247, 3, vals[1..1] )
+      assert_stack_map_frame( value ) {|a|
+        assert_equal a.frame_type, 247
+        assert_equal a.offset_delta, 3
+        assert_equal a.verification_type_info, vals[1..1]
+        assert_equal a.dump, "F7000307 000A"
+        assert_equal a.to_bytes, [0xF7, 0x00, 0x03, 0x07, 0x00, 0x0A]
+      }
+      value = ChopFrame.new( 248, 5 )
+      assert_stack_map_frame( value ) {|a|
+        assert_equal a.frame_type, 248
+        assert_equal a.offset_delta, 5
+        assert_equal a.dump, "F80005"
+        assert_equal a.to_bytes, [0xF8, 0x00, 0x05]
+      }
+      value = SameFrameExtended.new( 251, 16 )
+      assert_stack_map_frame( value ) {|a|
+        assert_equal a.frame_type, 251
+        assert_equal a.offset_delta, 16
+        assert_equal a.dump, "FB0010"
+        assert_equal a.to_bytes, [0xFB, 0x00, 0x10]
+      }
+      value = AppendFrame.new( 254, 10, vals )
+      assert_stack_map_frame( value ) {|a|
+        assert_equal a.frame_type, 254
+        assert_equal a.offset_delta, 10
+        assert_equal a.verification_type_info, vals
+        assert_equal a.dump, "FE000A03 07000A08 0002"
+        assert_equal a.to_bytes, [0xFE, 0x00, 0x0A, 0x03, 0x07, 0x00, 0x0A, 0x08, 0x00, 0x02]
+      }
+      value = FullFrame.new( 255, 3, vals, vals[0..1] )
+      assert_stack_map_frame( value ) {|a|
+        assert_equal a.frame_type, 255
+        assert_equal a.offset_delta, 3
+        assert_equal a.verification_type_info_local, vals
+        assert_equal a.verification_type_info_stack, vals[0..1]
+        assert_equal a.dump, "FF000300 03030700 0A080002 00020307\n" +
+                                          "000A"
+        assert_equal a.to_bytes, [
+          0xFF, 0x00, 0x03, 0x00, 0x03, 0x03, 0x07, 0x00, 0x0A, 0x08, 0x00, 0x02,
+          0x00, 0x02, 0x03, 0x07, 0x00, 0x0A
+        ]
+      }
+    end
+
+    #
+    #=== VariableInfoのテスト
+    #
+    def test_VariableInfo
+      
+      value = VariableInfo.new( 3 )
+      assert_variable_info( value ) {|a|
+        assert_equal a.tag, 3
+        assert_equal a.dump, "03"
+        assert_equal a.to_bytes, [0x03]
+      }
+      value = ObjectVariableInfo.new( 7, 10 )
+      assert_variable_info( value ) {|a|
+        assert_equal a.tag, 7
+        assert_equal a.cpool_index, 10
+        assert_equal a.dump, "07000A"
+        assert_equal a.to_bytes, [0x07, 0x00, 0x0A]
+      }
+      value = UninitializedVariableInfo .new( 8, 2 )
+      assert_variable_info( value ) {|a|
+        assert_equal a.tag, 8
+        assert_equal a.offset, 2
+        assert_equal a.dump, "080002"
+        assert_equal a.to_bytes, [0x08, 0x00, 0x02]
+      }
+    end
     
     def assert_attribute( a, &block )
       assert_to_byte_and_read a, block, proc {|io|  
@@ -950,6 +1085,19 @@ module JavaClass
         JavaClass.read_local_variable_type( io, @java_class )
       }
     end
+    
+    def assert_stack_map_frame( a, &block )
+      assert_to_byte_and_read a, block, proc {|io|  
+        JavaClass.read_stack_map_frame_entry( io, @java_class )
+      }
+    end
+    
+    def assert_variable_info( a, &block )
+      assert_to_byte_and_read a, block, proc {|io|  
+        JavaClass.read_variable_info( io, @java_class )
+      }
+    end
+    
   end
 
 end