1 /*
   2  * Copyright (c) 2010, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 /*
  25  * @test
  26  * @bug  6911256 6964740
  27  * @summary Test that the resource variable kind is appropriately set
  28  * @author  Joseph D. Darcy
  29  * @build TestResourceVariable
  30  * @compile/fail -processor TestResourceVariable -proc:only TestResourceVariable.java
  31  */
  32 
  33 // Bug should be filed for this misbehavior
  34 
  35 import java.io.*;
  36 import javax.annotation.processing.*;
  37 import javax.lang.model.*;
  38 import javax.lang.model.element.*;
  39 import javax.lang.model.type.*;
  40 import javax.lang.model.util.*;
  41 import java.util.*;
  42 import com.sun.source.tree.*;
  43 import com.sun.source.util.*;
  44 import static javax.tools.Diagnostic.Kind.*;
  45 
  46 /**
  47  * Using the tree API, retrieve element representations of the
  48  * resource of an ARM block and verify their kind tags are set
  49  * appropriately.
  50  */
  51 @SupportedAnnotationTypes("*")
  52 public class TestResourceVariable extends AbstractProcessor implements AutoCloseable {
  53     int resourceVariableCount = 0;
  54 
  55     public boolean process(Set<? extends TypeElement> annotations,
  56                           RoundEnvironment roundEnv) {
  57        if (!roundEnv.processingOver()) {
  58            Trees trees = Trees.instance(processingEnv);
  59 
  60            for(Element rootElement : roundEnv.getRootElements()) {
  61                TreePath treePath = trees.getPath(rootElement);
  62 
  63                (new ResourceVariableScanner(trees)).
  64                    scan(trees.getTree(rootElement),
  65                         treePath.getCompilationUnit());
  66            }
  67            if (resourceVariableCount != 3)
  68                throw new RuntimeException("Bad resource variable count " +
  69                                           resourceVariableCount);
  70        }
  71        return true;
  72     }
  73 
  74     @Override
  75     public void close() {}
  76 
  77     private void test1() {
  78         try(TestResourceVariable trv = this) {}
  79     }
  80 
  81     private void test2() {
  82         try(TestResourceVariable trv1 = this; TestResourceVariable trv2 = trv1) {}
  83     }
  84 
  85     class ResourceVariableScanner extends TreeScanner<Void, CompilationUnitTree> {
  86        private Trees trees;
  87 
  88        public ResourceVariableScanner(Trees trees) {
  89            super();
  90            this.trees = trees;
  91        }
  92        @Override
  93        public Void visitVariable(VariableTree node, CompilationUnitTree cu) {
  94            Element element = trees.getElement(trees.getPath(cu, node));
  95            if (element == null) {
  96                System.out.println("Null variable element: " + node);
  97            } else {
  98                System.out.println("Name: " + element.getSimpleName() +
  99                                   "\tKind: " + element.getKind());
 100            }
 101            if (element != null &&
 102                element.getKind() == ElementKind.RESOURCE_VARIABLE) {
 103                resourceVariableCount++;
 104            }
 105            return super.visitVariable(node, cu);
 106        }
 107    }
 108 
 109    @Override
 110    public SourceVersion getSupportedSourceVersion() {
 111        return SourceVersion.latest();
 112    }
 113 }