001: /***
002: * ASM Guide
003: * Copyright (c) 2007 Eric Bruneton
004: * All rights reserved.
005: *
006: * Redistribution and use in source and binary forms, with or without
007: * modification, are permitted provided that the following conditions
008: * are met:
009: * 1. Redistributions of source code must retain the above copyright
010: * notice, this list of conditions and the following disclaimer.
011: * 2. Redistributions in binary form must reproduce the above copyright
012: * notice, this list of conditions and the following disclaimer in the
013: * documentation and/or other materials provided with the distribution.
014: * 3. Neither the name of the copyright holders nor the names of its
015: * contributors may be used to endorse or promote products derived from
016: * this software without specific prior written permission.
017: *
018: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
019: * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
020: * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
021: * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
022: * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
023: * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
024: * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
025: * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
026: * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
027: * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
028: * THE POSSIBILITY OF SUCH DAMAGE.
029: */package ch6.sec1;
030:
031: import static org.objectweb.asm.Opcodes.ALOAD;
032: import static org.objectweb.asm.Opcodes.ATHROW;
033: import static org.objectweb.asm.Opcodes.DUP;
034: import static org.objectweb.asm.Opcodes.F_SAME;
035: import static org.objectweb.asm.Opcodes.GOTO;
036: import static org.objectweb.asm.Opcodes.IFLT;
037: import static org.objectweb.asm.Opcodes.ILOAD;
038: import static org.objectweb.asm.Opcodes.INVOKESPECIAL;
039: import static org.objectweb.asm.Opcodes.NEW;
040: import static org.objectweb.asm.Opcodes.PUTFIELD;
041: import static org.objectweb.asm.Opcodes.RETURN;
042:
043: import java.util.List;
044:
045: import org.objectweb.asm.ClassVisitor;
046: import org.objectweb.asm.Label;
047: import org.objectweb.asm.tree.ClassNode;
048: import org.objectweb.asm.tree.MethodNode;
049: import org.objectweb.asm.util.TraceMethodVisitor;
050:
051: import util.AbstractTestCase;
052:
053: /**
054: * ASM Guide example test class.
055: *
056: * @author Eric Bruneton
057: */
058: public class OptimizeJumpTransformerTest extends AbstractTestCase {
059:
060: public void test() {
061: TraceMethodVisitor tmv = new TraceMethodVisitor(null);
062: MethodNode mn = new MethodNode(0, null, null, null, null);
063: mn.visitCode();
064: mn.visitVarInsn(ILOAD, 1);
065: Label label = new Label();
066: mn.visitJumpInsn(IFLT, label);
067: mn.visitVarInsn(ALOAD, 0);
068: mn.visitVarInsn(ILOAD, 1);
069: mn.visitFieldInsn(PUTFIELD, "pkg/Bean", "f", "I");
070: Label end = new Label();
071: mn.visitJumpInsn(GOTO, end);
072: mn.visitLabel(label);
073: mn.visitFrame(F_SAME, 0, null, 0, null);
074: mn.visitTypeInsn(NEW, "java/lang/IllegalArgumentException");
075: mn.visitInsn(DUP);
076: mn.visitMethodInsn(INVOKESPECIAL,
077: "java/lang/IllegalArgumentException", "<init>", "()V");
078: mn.visitInsn(ATHROW);
079: mn.visitLabel(end);
080: mn.visitFrame(F_SAME, 0, null, 0, null);
081: mn.visitInsn(RETURN);
082: mn.visitMaxs(0, 0);
083: mn.visitEnd();
084: new OptimizeJumpTransformer(null).transform(mn);
085: mn.accept(tmv);
086: checkMethod(tmv);
087: }
088:
089: protected void checkMethod(TraceMethodVisitor tmv) {
090: TraceMethodVisitor mv = new TraceMethodVisitor(null);
091: mv.visitCode();
092: mv.visitVarInsn(ILOAD, 1);
093: Label label = new Label();
094: mv.visitJumpInsn(IFLT, label);
095: mv.visitVarInsn(ALOAD, 0);
096: mv.visitVarInsn(ILOAD, 1);
097: mv.visitFieldInsn(PUTFIELD, "pkg/Bean", "f", "I");
098: Label end = new Label();
099: mv.visitInsn(RETURN);
100: mv.visitLabel(label);
101: mv.visitFrame(F_SAME, 0, null, 0, null);
102: mv.visitTypeInsn(NEW, "java/lang/IllegalArgumentException");
103: mv.visitInsn(DUP);
104: mv.visitMethodInsn(INVOKESPECIAL,
105: "java/lang/IllegalArgumentException", "<init>", "()V");
106: mv.visitInsn(ATHROW);
107: mv.visitLabel(end);
108: mv.visitFrame(F_SAME, 0, null, 0, null);
109: mv.visitInsn(RETURN);
110: mv.visitMaxs(0, 0);
111: mv.visitEnd();
112: assertEquals(mv, tmv);
113: }
114:
115: @Override
116: protected ClassVisitor getClassAdapter(final ClassVisitor cv) {
117: return new ClassNode() {
118: @Override
119: public void visitEnd() {
120: for (MethodNode mn : (List<MethodNode>) methods) {
121: new OptimizeJumpTransformer(null).transform(mn);
122: }
123: accept(cv);
124: }
125: };
126: }
127: }
|