Home | History | Annotate | Download | only in util 
      1 /***
      2  * ASM: a very small and fast Java bytecode manipulation framework
      3  * Copyright (c) 2000-2007 INRIA, France Telecom
      4  * All rights reserved.
      5  *
      6  * Redistribution and use in source and binary forms, with or without
      7  * modification, are permitted provided that the following conditions
      8  * are met:
      9  * 1. Redistributions of source code must retain the above copyright
     10  *    notice, this list of conditions and the following disclaimer.
     11  * 2. Redistributions in binary form must reproduce the above copyright
     12  *    notice, this list of conditions and the following disclaimer in the
     13  *    documentation and/or other materials provided with the distribution.
     14  * 3. Neither the name of the copyright holders nor the names of its
     15  *    contributors may be used to endorse or promote products derived from
     16  *    this software without specific prior written permission.
     17  *
     18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
     19  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     20  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     21  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
     22  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     23  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     24  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     25  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     26  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     27  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
     28  * THE POSSIBILITY OF SUCH DAMAGE.
     29  */
     30 package org.mockito.asm.util;
     31 
     32 import org.mockito.asm.AnnotationVisitor;
     33 import org.mockito.asm.Attribute;
     34 import org.mockito.asm.FieldVisitor;
     35 
     36 /**
     37  * A {@link FieldVisitor} that checks that its methods are properly used.
     38  */
     39 public class CheckFieldAdapter implements FieldVisitor {
     40 
     41     private final FieldVisitor fv;
     42 
     43     private boolean end;
     44 
     45     public CheckFieldAdapter(final FieldVisitor fv) {
     46         this.fv = fv;
     47     }
     48 
     49     public AnnotationVisitor visitAnnotation(
     50         final String desc,
     51         final boolean visible)
     52     {
     53         checkEnd();
     54         CheckMethodAdapter.checkDesc(desc, false);
     55         return new CheckAnnotationAdapter(fv.visitAnnotation(desc, visible));
     56     }
     57 
     58     public void visitAttribute(final Attribute attr) {
     59         checkEnd();
     60         if (attr == null) {
     61             throw new IllegalArgumentException("Invalid attribute (must not be null)");
     62         }
     63         fv.visitAttribute(attr);
     64     }
     65 
     66     public void visitEnd() {
     67         checkEnd();
     68         end = true;
     69         fv.visitEnd();
     70     }
     71 
     72     private void checkEnd() {
     73         if (end) {
     74             throw new IllegalStateException("Cannot call a visit method after visitEnd has been called");
     75         }
     76     }
     77 }
     78