1 /*
2  * Copyright (c) 2010, 2015, 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 6855236
27  * @summary Compiler Tree API TreePath class generates NullPointerException from Iterator
28  * @modules jdk.compiler
29  * @compile T6855236.java
30  * @compile -processor T6855236 -proc:only T6855236.java
31  */
32 
33 import java.util.*;
34 
35 import javax.annotation.processing.*;
36 import javax.lang.model.*;
37 import javax.lang.model.element.*;
38 
39 import com.sun.source.tree.*;
40 import com.sun.source.util.*;
41 
42 @SupportedAnnotationTypes("*")
43 public class T6855236 extends AbstractProcessor {
44 
45     private Trees trees;
46 
47     @Override
init(ProcessingEnvironment pe)48     public void init(ProcessingEnvironment pe) {
49         super.init(pe);
50         trees = Trees.instance(pe);
51     }
52 
53     @Override
process(Set<? extends TypeElement> arg0, RoundEnvironment roundEnvironment)54     public boolean process(Set<? extends TypeElement> arg0, RoundEnvironment roundEnvironment) {
55         // Scanner class to scan through various component elements
56         CodeVisitor visitor = new CodeVisitor();
57 
58         for (Element e : roundEnvironment.getRootElements()) {
59             TreePath tp = trees.getPath(e);
60             visitor.scan(tp, trees);
61         }
62 
63         return true;
64     }
65 
66     @Override
getSupportedSourceVersion()67     public SourceVersion getSupportedSourceVersion() {
68         return SourceVersion.latest();
69     }
70 
71     class CodeVisitor extends TreePathScanner<Object, Trees> {
72 
73         @Override
visitMethodInvocation(MethodInvocationTree node, Trees p)74         public Object visitMethodInvocation(MethodInvocationTree node, Trees p) {
75             System.out.println("current path: ");
76             for (Tree t : getCurrentPath()) {
77                 System.out.println("    " + t.getKind() + ": " + trim(t, 64));
78             }
79             System.out.println("parent path: " + getCurrentPath().getParentPath());
80             System.out.println("method select: " + node.getMethodSelect().toString());
81             for (ExpressionTree arg : node.getArguments()) {
82                 System.out.println("argument: " + arg.toString());
83             }
84             return super.visitMethodInvocation(node, p);
85         }
86 
87         @Override
visitExpressionStatement(ExpressionStatementTree node, Trees p)88         public Object visitExpressionStatement(ExpressionStatementTree node, Trees p) {
89             ExpressionTree t = node.getExpression();
90             System.out.println();
91             System.out.println("expression statement: " + trim(t, 64));
92             return super.visitExpressionStatement(node, p);
93         }
94 
95     }
96 
trim(Tree t, int len)97     private String trim(Tree t, int len) {
98         String s = t.toString().trim().replaceAll("\\s+", " ");
99         if (s.length() > len)
100             s = s.substring(0, len) + "...";
101         return s;
102     }
103 
104 }
105 
106 
107