public void testMethodInInner2() throws Exception { CtClass inner = sloader.get("test2.Nested3$Inner"); CtClass outer = sloader.get("test2.Nested3"); String src = "public int f() {" + " int k = 0;" + " test2.Nested3 n = new test2.Nested3(3);" + " k += n.geti();" + " n = new test2.Nested3();" + " k += n.geti();" + " n = new test2.Nested3(\"foo\");" + " k += n.geti();" + " return k; }"; outer.stopPruning(true); outer.writeFile(); try { CtMethod m = CtNewMethod.make(src, inner); fail(); } catch (RuntimeException e) { } outer.defrost(); CtMethod m = CtNewMethod.make(src, inner); inner.addMethod(m); inner.writeFile(); outer.writeFile(); Object iobj = make(inner.getName()); assertEquals(6, invoke(iobj, "f")); }
public void testCodeGen2() throws Exception { CtClass cc = sloader.makeClass("test2.CodeGen2"); CtMethod m1 = CtNewMethod.make( "public int test() {" + " int len;" + " String s = \"foo\" + \"bar\" + 3;" + " System.out.println(s); len = s.length();" + " len = -3 + len; len = len - (7 - 2 + -1);" + " int k = 3; len += ~k - ~3;" + " return len; }", cc); cc.addMethod(m1); CtMethod m2 = CtNewMethod.make( "public int test2() {" + " double d = 0.2 - -0.1;" + " d += (0.2 + 0.3) * 1.0;" + " return (int)(d * 10); }", cc); cc.addMethod(m2); cc.writeFile(); Object obj = make(cc.getName()); assertEquals(0, invoke(obj, "test")); assertEquals(8, invoke(obj, "test2")); }
public void testNullArg() throws Exception { CtClass cc = sloader.makeClass("test2.NullArgTest"); CtMethod m1 = CtNewMethod.make( "public Object foo(Object[] obj, int idx) throws Throwable {" + " return null; }", cc); cc.addMethod(m1); CtMethod m2 = CtNewMethod.make("public void bar() { this.foo(null, 0); }", cc); cc.addMethod(m2); CtMethod m3 = CtNewMethod.make("public void bar2() { this.foo((Object[])null, 0); }", cc); cc.addMethod(m3); cc.writeFile(); }
public void testCodeGen() throws Exception { CtClass cc = sloader.get("test2.CodeGen"); CtMethod m1 = cc.getDeclaredMethod("run"); m1.insertBefore( "{ double d = true ? 1 : 0.1; " + " d = d > 0.5 ? 0.0 : - 1.0; " + " System.out.println(d); " + " String s = \"foo\"; " + " s = 1 + 2 + s + \"bar\"; " + " s += \"poi\" + 3 + seven() + seven(\":\" + ' '); " + " s += .14; " + " msg = s; " + " System.out.println(s); }"); // recursive type check is done if $proceed is used. CtMethod m2 = CtNewMethod.make( "public int test() {" + " String s = $proceed(\"int\" + (3 + 0.14)) + '.'; " + " System.out.println(s); return s.length(); }", cc, "this", "seven"); cc.addMethod(m2); cc.writeFile(); Object obj = make(cc.getName()); assertEquals(19, invoke(obj, "run")); assertEquals(9, invoke(obj, "test")); }
public void testArrayLen() throws Exception { CtClass cc = sloader.get("test2.ArrayLenTest"); cc.addMethod(CtNewMethod.make("public int foo(){ return this.length; }", cc)); cc.writeFile(); Object obj = make(cc.getName()); assertEquals(1, invoke(obj, "foo")); }
public void testAddMethod() throws Exception { CtClass cc = sloader.get("test2.AddMethod"); CtMethod m = CtNewMethod.make("public void f() { return 1; }", cc); try { cc.addMethod(m); fail(); } catch (CannotCompileException e) { } CtMethod m2 = CtNewMethod.make("public void f(int i, int j) { return 1; }", cc); cc.addMethod(m2); try { cc.addField(new CtField(CtClass.longType, "f", cc)); fail(); } catch (CannotCompileException e) { } }
public void testPrivateMethod() throws Exception { CtClass cc = sloader.get("test2.PrivateMethod"); try { CtMethod m = CtNewMethod.make("public int f(test2.PrivateMethod2 p) { return p.f(); }", cc); fail(); } catch (CannotCompileException e) { } }
public void testObjectSuper() throws Exception { CtClass cc = sloader.get("java.lang.Object"); try { cc.addMethod(CtNewMethod.make("public int foo(){ return super.hashCode(); }", cc)); fail("could access the super of java.lang.Object"); } catch (CannotCompileException e) { } }
public void testMakeStaticMethod() throws Exception { CtClass cc = sloader.makeClass("test2.MakeStaticMethod"); CtMethod m = CtNewMethod.make( Modifier.PUBLIC | Modifier.STATIC, CtClass.intType, "create", new CtClass[] {CtClass.intType}, null, "{ return $1; }", cc); cc.addMethod(m); cc.addMethod(CtNewMethod.make("public int test() { return create(13); }", cc)); cc.writeFile(); Object obj = make(cc.getName()); assertEquals(13, invoke(obj, "test")); }
public void testSuperInterface() throws Exception { CtClass cc = sloader.makeClass("test2.SuperInterface3"); CtClass cc2 = sloader.get("test2.SuperInterface2"); cc.addInterface(cc2); cc.addField(new CtField(cc2, "inner", cc)); CtMethod m = CtNewMethod.make("public int getAge() { return inner.getAge(); }", cc); cc.addMethod(m); cc.writeFile(); }
public void testArrayLength() throws Exception { CtClass cc = sloader.makeClass("test2.ArrayLength"); CtMethod m2 = CtNewMethod.make( "public int f() { String[] s = new String[3]; " + "return s.length; }", cc); cc.addMethod(m2); cc.writeFile(); Object obj = make(cc.getName()); assertEquals(3, invoke(obj, "f")); }
public void testMethodInInner() throws Exception { CtClass inner = sloader.get("test2.Nested2$Inner"); CtClass outer = sloader.get("test2.Nested2"); String src = "public int f(test2.Nested2 n) {" + " n.i = 1; n.i++; n.i += 2; return n.i; }"; outer.writeFile(); try { CtMethod m = CtNewMethod.make(src, inner); fail(); } catch (RuntimeException e) { } outer.defrost(); CtMethod m = CtNewMethod.make(src, inner); inner.addMethod(m); src = "public int g(test2.Nested2 n) {" + " n.d = 1.0; n.d++; n.d += 2.0;" + " return n.d == 4.0 ? 7 : 8; }"; m = CtNewMethod.make(src, inner); inner.addMethod(m); src = "public int h(test2.Nested2 n) {" + " n.s = \"poi\";" + "return n.s.length() + f(n) + g(n); }"; m = CtNewMethod.make(src, inner); inner.addMethod(m); inner.writeFile(); outer.writeFile(); Object nobj = make(outer.getName()); Object iobj = make(inner.getName()); Method mth = iobj.getClass().getMethod("h", new Class[] {nobj.getClass()}); Object resobj = mth.invoke(iobj, new Object[] {nobj}); int res = ((Integer) resobj).intValue(); assertEquals(14, res); }
public void testStaticMember2() throws Exception { CtClass cc = sloader.get("test2.StaticMember2"); cc.addMethod( CtNewMethod.make( "public int run() {" + " return test2.StaticMember2.k + test2.StaticMember2.seven()" + " + (test2.StaticMember2.f + f)" + " + test2.StaticMember2.f + f; }", cc)); cc.addMethod( CtNewMethod.make( "public int run1() {" + " long j = 1L;" + " return (int)(j + (test2.StaticMember2.fj + fj)" + " + test2.StaticMember2.fj + fj); }", cc)); cc.addMethod( CtNewMethod.make( "public int run2() {" + " double x = 1.0;" + " double d = x + test2.StaticMember2.fd + fd" + " + (test2.StaticMember2.fd + fd);" + " return (int)(d * 10); }", cc)); cc.addMethod( CtNewMethod.make( "public int run3() {" + " return (test2.StaticMember2.fb & fb) ? 1 : 0; }", cc)); cc.writeFile(); Object obj = make(cc.getName()); assertEquals(54, invoke(obj, "run")); assertEquals(53, invoke(obj, "run1")); assertEquals(958, invoke(obj, "run2")); assertEquals(0, invoke(obj, "run3")); }
public void testBrennan() throws Exception { CtClass cc = sloader.get("test2.Brennan"); cc.addMethod( CtNewMethod.make( "public int foo(){" + " java.text.SimpleDateFormat df;" + " if((df = (java.text.SimpleDateFormat)format) == null)" + " df = new java.text.SimpleDateFormat(\"yyyyMMdd\");" + " return 1;}", cc)); cc.writeFile(); Object obj = make(cc.getName()); assertEquals(1, invoke(obj, "foo")); }
public void testArrayInit() throws Exception { CtClass cc = sloader.makeClass("test2.ArrayInit"); cc.addMethod( CtNewMethod.make( "public int foo(){ " + " int[] i = new int[] { 1, 2 };" + " double[] d = new double[] { 3.0, 4.0 };" + " String[] s = new String[] { \"foo\", \"12345\" };" + " return i[0] + (int)d[0] + s[1].length(); }", cc)); cc.addMethod( CtNewMethod.make( "public int bar(){ " + " int[] i = { 1, 2.0 };" + " double[] d = { 3.0, 4 };" + " String[] s = { \"foo\", \"12345\" };" + " return i[0] + (int)d[0] + s[1].length(); }", cc)); cc.writeFile(); Object obj = make(cc.getName()); assertEquals(9, invoke(obj, "foo")); assertEquals(9, invoke(obj, "bar")); }
public void testInheritance() throws Exception { ClassPool pool = ClassPool.getDefault(); String classname = "test2.Inherit"; CtClass target = pool.get(classname); String src = "public void sampleMethod() {" + " test2.Inherit i = new test2.Inherit();" + " test2.Inherit2 i2 = i;" + " test2.Inherit3 i3 = i;" + " i3.foo2(); i3.foo2(); i2.foo1(); }"; CtMethod newmethod = CtNewMethod.make(src, target); target.addMethod(newmethod); target.writeFile(); }
public void testInner() throws Exception { ClassPool pool = ClassPool.getDefault(); String classname = "test2.Inner"; CtClass target = pool.get(classname); String src = "public void sampleMethod() throws Exception {" + "java.util.Properties props = new java.util.Properties();" + "java.rmi.activation.ActivationGroupDesc.CommandEnvironment ace " + " = null;" + "java.rmi.activation.ActivationGroupDesc agd " + " = new java.rmi.activation.ActivationGroupDesc(props,ace);}"; CtMethod newmethod = CtNewMethod.make(src, target); target.addMethod(newmethod); String src2 = "public java.lang.Character.Subset sampleMethod2() {" + " java.lang.Character.Subset s " + " = Character.UnicodeBlock.HIRAGANA; " + " return s; }"; CtMethod newmethod2 = CtNewMethod.make(src2, target); target.addMethod(newmethod2); target.writeFile(); }
public void testDeclaringClass() throws Exception { try { CtClass cc = sloader.get("test2.NotExistingClass"); } catch (NotFoundException e) { System.out.println(e); } CtClass inner = sloader.get("test2.Nested$Inner"); CtClass outer = sloader.get("test2.Nested"); assertEquals(outer, inner.getDeclaringClass()); assertEquals(null, outer.getDeclaringClass()); assertEquals(null, CtClass.intType.getDeclaringClass()); CtClass inner3 = sloader.get("test2.Nested$Inner3"); outer.writeFile(); try { CtMethod m = CtNewMethod.make("public void f(test2.Nested n) { return n.geti(); }", inner3); fail(); } catch (RuntimeException e) { } outer.defrost(); CtMethod m = CtNewMethod.make( "public int f(test2.Nested n) { " + "return n.geti() + test2.Nested.getj(1) + f() + g(); } ", inner3); inner3.addMethod(m); inner3.writeFile(); outer.writeFile(); Object nobj = make(outer.getName()); Object iobj = make(inner3.getName()); Method mth = iobj.getClass().getMethod("f", new Class[] {nobj.getClass()}); Object resobj = mth.invoke(iobj, new Object[] {nobj}); int res = ((Integer) resobj).intValue(); assertEquals(6, res); }
private void testDotClass(String cname, boolean java5) throws Exception { CtClass cc = sloader.makeClass(cname); if (java5) cc.getClassFile2().setVersionToJava5(); CtMethod m = CtNewMethod.make( "public String getclass() {" + " return int.class.getName() + int[].class.getName()" + " + String.class.getName() + String[].class.getName()" + " + java.lang.Object.class.getName()" + " + java.util.Vector.class.getName(); }", cc); cc.addMethod(m); CtMethod m2 = CtNewMethod.make( "public int test() {" + " String s = getclass(); System.out.println(s);" + " return s.length(); }", cc); cc.addMethod(m2); cc.writeFile(); Object obj = make(cc.getName()); assertEquals(72, invoke(obj, "test")); }
public void testIncOp() throws Exception { CtClass target = sloader.makeClass("test2.IncOp"); String src = "public int sample() {" + " int ia[] = new int[50];" + " ia[0] = 1;" + " int v = ++(ia[0]);" + " return v; }"; CtMethod newmethod = CtNewMethod.make(src, target); target.addMethod(newmethod); target.writeFile(); Object obj = make(target.getName()); assertEquals(2, invoke(obj, "sample")); }
public void testImportPackage() throws Exception { CtClass cc2 = sloader.makeClass("test2.Imported"); cc2.writeFile(); CtClass cc = sloader.makeClass("test2.Importer"); sloader.importPackage("test2"); cc.addMethod( CtNewMethod.make( "public int foo(){ " + " Imported obj = new Imported();" + " return obj.getClass().getName().length(); }", cc)); sloader.clearImportedPackages(); cc.writeFile(); Object obj = make(cc.getName()); assertEquals(14, invoke(obj, "foo")); }
private void testDotClass4(String cname, boolean java5) throws Exception { CtClass cc = sloader.makeClass(cname); if (java5) cc.getClassFile2().setVersionToJava5(); CtMethod m = CtNewMethod.make( "public int test() {" + " String s = Object.class.getName()" + " + Object[].class.getName();" + " return s.length(); }", cc); cc.addMethod(m); cc.writeFile(); Object obj = make(cc.getName()); assertEquals(35, invoke(obj, "test")); }
public void testKmatcha() throws Exception { CtClass cc = sloader.makeClass("test2.Kmatcha"); cc.addMethod( CtNewMethod.make( "public void display(String [] params){" + " if(params == null){" + " System.out.println(\"Nothing to display\");" + " }else{" + " int k = params.length - 1;" + " if(k >= 0)" + " do " + " System.out.println(params[k]);" + " while(--k >= 0);" + " }}", cc)); }
public void testStaticFinal() throws Exception { CtClass cc = sloader.makeClass("test2.StaticFinal"); CtField f = new CtField(CtClass.intType, "sff1", cc); f.setModifiers(Modifier.STATIC | Modifier.FINAL); cc.addField(f, "5"); assertEquals(new Integer(5), f.getConstantValue()); f = new CtField(CtClass.longType, "sff2", cc); f.setModifiers(Modifier.STATIC | Modifier.FINAL); cc.addField(f, "6"); assertEquals(new Long(6), f.getConstantValue()); f = new CtField(CtClass.floatType, "sff3", cc); f.setModifiers(Modifier.STATIC | Modifier.FINAL); cc.addField(f, "7"); assertEquals(new Float(7.0F), f.getConstantValue()); f = new CtField(CtClass.floatType, "sff4", cc); f.setModifiers(Modifier.STATIC | Modifier.FINAL); cc.addField(f, "8.0"); assertEquals(new Float(8.0F), f.getConstantValue()); f = new CtField(CtClass.doubleType, "sff5", cc); f.setModifiers(Modifier.STATIC | Modifier.FINAL); cc.addField(f, "9"); assertEquals(new Double(9.0), f.getConstantValue()); f = new CtField(CtClass.doubleType, "sff6", cc); f.setModifiers(Modifier.STATIC | Modifier.FINAL); cc.addField(f, "10.0"); assertEquals(new Double(10.0), f.getConstantValue()); f = new CtField(sloader.get("java.lang.String"), "sff7", cc); f.setModifiers(Modifier.STATIC | Modifier.FINAL); cc.addField(f, "\"test\""); assertEquals("test", f.getConstantValue()); f = new CtField(sloader.get("java.lang.String"), "sff8", cc); f.setModifiers(Modifier.STATIC); cc.addField(f, "\"static\""); assertEquals(null, f.getConstantValue()); cc.addMethod(CtNewMethod.make("public int foo(){ return sff1 + sff7.length(); }", cc)); cc.writeFile(); Object obj = make(cc.getName()); assertEquals(9, invoke(obj, "foo")); }
private void testDotClass2(String cname, boolean java5) throws Exception { CtClass cc = sloader.makeClass(cname); CtClass cc3 = sloader.makeClass("test2.DotClass3"); if (java5) cc.getClassFile2().setVersionToJava5(); CtMethod m = CtNewMethod.make( "public int test() {" + " return test2.DotClass3.class.getName().length(); }", cc); cc.addMethod(m); cc.writeFile(); // don't execute cc3.writeFile(); Object obj = make(cc.getName()); try { assertEquals(15, invoke(obj, "test")); } catch (java.lang.reflect.InvocationTargetException e) { Throwable t = e.getCause(); assertTrue(t instanceof java.lang.NoClassDefFoundError); } }
public void testMakeNestedClass() throws Exception { CtClass outer = sloader.get("test2.Nested4"); try { CtClass inner = outer.makeNestedClass("Inner", false); fail(); } catch (RuntimeException e) { print(e.getMessage()); } CtClass nested = outer.makeNestedClass("Inner", true); outer.stopPruning(true); outer.writeFile(); outer.defrost(); String src = "public int f() { return test2.Nested4.value; }"; CtMethod m = CtNewMethod.make(src, nested); nested.addMethod(m); nested.writeFile(); outer.writeFile(); Object iobj = make(nested.getName()); assertEquals(6, invoke(iobj, "f")); }
public void testNewOp() throws Exception { CtClass cc = sloader.get("test2.NewOp"); CtMethod add = CtNewMethod.make( "public test2.NewOp2 " + " addMonitoringRemoteEventListener(" + " test2.NewOp2 listener)" + " throws java.rmi.RemoteException {" + " $0.listenerList.addElement(listener);" + " return new test2.NewOp2(0L, this, null, 0L);" + "}", cc); add.setModifiers(Modifier.PUBLIC); cc.addMethod(add); /* CtMethod type= CtNewMethod.make( "public test2.Where getNewType() { return new test2.Where(); }", cc); cc.addMethod(type); */ cc.writeFile(); }
public void testStaticMember() throws Exception { CtClass cc = sloader.get("test2.StaticMember"); CtMethod m = CtNewMethod.make( "public int run() {" + "return test2.StaticMember#k + test2.StaticMember#foo(); }", cc); cc.addMethod(m); m = CtNewMethod.make("public int run2() {" + "return k + foo(); }", cc); cc.addMethod(m); m = CtNewMethod.make( "public int run3() {" + " test2.StaticMember sm = this;" + " return sm.k + sm.foo(); }", cc); cc.addMethod(m); m = CtNewMethod.make("public int run4() {" + " return this.k + this.foo(); }", cc); cc.addMethod(m); m = CtNewMethod.make("public static int run5() {" + " return k + foo(); }", cc); cc.addMethod(m); m = CtNewMethod.make( "public int run6() {" + " test2.IStaticMember i = this; return i.bar(); }", cc); cc.addMethod(m); cc.writeFile(); Object obj = make(cc.getName()); assertEquals(10, invoke(obj, "run")); assertEquals(10, invoke(obj, "run2")); assertEquals(10, invoke(obj, "run3")); assertEquals(10, invoke(obj, "run4")); assertEquals(10, invoke(obj, "run5")); assertEquals(3, invoke(obj, "run6")); }
public void testSwitch() throws Exception { CtClass cc = sloader.makeClass("test2.Switch"); cc.addMethod( CtNewMethod.make( "public int test1() {" + " int i = 1;" + " int j;" + " switch (i) {" + " case 0: j = i; break;" + " case 1: j = -i; break;" + " default: j = 0; break;" + " }" + " return j; }", cc)); cc.addMethod( CtNewMethod.make( "public int test2() {" + " int i = 2;" + " int j = 7;" + " switch (i) {" + " case 0: j = i; break;" + " case 1: j = -i; break;" + " }" + " return j; }", cc)); cc.addMethod( CtNewMethod.make( "public int test3() {" + " int i = Byte.MAX_VALUE;" + " int j;" + " switch (i) {" + " case Byte.MAX_VALUE: j = i; break;" + " case Byte.MIN_VALUE: j = -i; break;" + " default: j = 0; break;" + " }" + " return j; }", cc)); try { cc.addMethod( CtNewMethod.make( "public int test4() {" + " int i = Byte.MAX_VALUE;" + " int j;" + " switch (i) {" + " case Byte.MAX_VALUE: j = i; return j;" + " case Byte.MIN_VALUE: j = -i; return j;" + " default: j = 0;" + " }" + "}", cc)); fail("does not report an error (no return)"); } catch (CannotCompileException e) { System.out.println(e); } try { cc.addMethod( CtNewMethod.make( "public int test5() {" + " int i = Byte.MAX_VALUE;" + " int j;" + " switch (i) {" + " case Byte.MAX_VALUE: j = i; return j;" + " case Byte.MIN_VALUE: j = -i; return j;" + " }" + "}", cc)); fail("does not report an error (not default)"); } catch (CannotCompileException e) { System.out.println(e); } try { cc.addMethod( CtNewMethod.make( "public int test6() {" + " int i = Byte.MAX_VALUE;" + " int j;" + " switch (i) {" + " case Byte.MAX_VALUE: j = i; break;" + " default: j = -i; return j;" + " }" + " }", cc)); fail("does not report an error (break)"); } catch (CannotCompileException e) { System.out.println(e); } cc.addField(CtField.make("public static int k;", cc)); cc.addMethod( CtNewMethod.make( "public void foo() {" + " int i = 0;" + " k = 3;" + " switch (i) {" + " case Byte.MAX_VALUE: k = 1;" + " case Byte.MIN_VALUE: k = 2;" + " }" + "}", cc)); cc.addMethod( CtNewMethod.make( "public int test7() {" + " int i = Byte.MAX_VALUE;" + " int j = 3; foo();" + " System.out.println(k);" + " switch (i) {" + " case Byte.MAX_VALUE: return k;" + " case Byte.MIN_VALUE: return j;" + " default: return 0;" + " }" + "}", cc)); cc.writeFile(); Object obj = make(cc.getName()); assertEquals(-1, invoke(obj, "test1")); assertEquals(7, invoke(obj, "test2")); assertEquals(Byte.MAX_VALUE, invoke(obj, "test3")); assertEquals(3, invoke(obj, "test7")); }
public void testUnicodeIdentifier() throws Exception { CtClass cc = sloader.makeClass("test2.UnicodeIdentifier"); String src = "public int foo(){ int \u5206 = 0; return \u5206; }"; cc.addMethod(CtNewMethod.make(src, cc)); }