1674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen/*** 2674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * ASM: a very small and fast Java bytecode manipulation framework 3674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * Copyright (c) 2000-2007 INRIA, France Telecom 4674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * All rights reserved. 5674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * 6674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * Redistribution and use in source and binary forms, with or without 7674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * modification, are permitted provided that the following conditions 8674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * are met: 9674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * 1. Redistributions of source code must retain the above copyright 10674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * notice, this list of conditions and the following disclaimer. 11674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * 2. Redistributions in binary form must reproduce the above copyright 12674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * notice, this list of conditions and the following disclaimer in the 13674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * documentation and/or other materials provided with the distribution. 14674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * 3. Neither the name of the copyright holders nor the names of its 15674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * contributors may be used to endorse or promote products derived from 16674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * this software without specific prior written permission. 17674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * 18674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 19674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 20674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 21674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE 22674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 23674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 24674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 25674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 26674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 27674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF 28674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * THE POSSIBILITY OF SUCH DAMAGE. 29674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen */ 30674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogenpackage org.mockito.asm.util; 31674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen 32674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogenimport org.mockito.asm.AnnotationVisitor; 33674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogenimport org.mockito.asm.Attribute; 34674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogenimport org.mockito.asm.FieldVisitor; 35674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen 36674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen/** 37674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen * A {@link FieldVisitor} that checks that its methods are properly used. 38674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen */ 39674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogenpublic class CheckFieldAdapter implements FieldVisitor { 40674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen 41674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen private final FieldVisitor fv; 42674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen 43674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen private boolean end; 44674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen 45674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen public CheckFieldAdapter(final FieldVisitor fv) { 46674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen this.fv = fv; 47674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen } 48674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen 49674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen public AnnotationVisitor visitAnnotation( 50674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen final String desc, 51674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen final boolean visible) 52674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen { 53674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen checkEnd(); 54674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen CheckMethodAdapter.checkDesc(desc, false); 55674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen return new CheckAnnotationAdapter(fv.visitAnnotation(desc, visible)); 56674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen } 57674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen 58674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen public void visitAttribute(final Attribute attr) { 59674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen checkEnd(); 60674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen if (attr == null) { 61674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen throw new IllegalArgumentException("Invalid attribute (must not be null)"); 62674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen } 63674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen fv.visitAttribute(attr); 64674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen } 65674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen 66674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen public void visitEnd() { 67674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen checkEnd(); 68674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen end = true; 69674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen fv.visitEnd(); 70674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen } 71674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen 72674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen private void checkEnd() { 73674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen if (end) { 74674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen throw new IllegalStateException("Cannot call a visit method after visitEnd has been called"); 75674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen } 76674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen } 77674060f01e9090cd21b3c5656cc3204912ad17a6Jon Boekenoogen} 78