1 /* 2 * Copyright (c) 2006, 2016, 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 6412669 6997958 27 * @summary Should be able to get SourcePositions from 269 world 28 * @modules jdk.compiler/com.sun.tools.javac.api 29 * jdk.compiler/com.sun.tools.javac.file 30 */ 31 32 import java.io.*; 33 import java.net.*; 34 import java.util.*; 35 import javax.annotation.*; 36 import javax.annotation.processing.*; 37 import javax.lang.model.*; 38 import javax.lang.model.element.*; 39 import javax.tools.*; 40 import com.sun.source.util.*; 41 import com.sun.tools.javac.api.*; 42 43 @SupportedAnnotationTypes("*") 44 public class T6412669 extends AbstractProcessor { 45 public static void main(String... args) throws Exception { 46 File testSrc = new File(System.getProperty("test.src", ".")); 47 File testClasses = new File(System.getProperty("test.classes", ".")); 48 49 JavacTool tool = JavacTool.create(); 50 try (StandardJavaFileManager fm = tool.getStandardFileManager(null, null, null)) { 51 fm.setLocation(StandardLocation.CLASS_PATH, Arrays.asList(testClasses)); 52 Iterable<? extends JavaFileObject> files = 53 fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrc, T6412669.class.getName()+".java"))); 54 String[] opts = { 55 "-proc:only", 56 "-processor", T6412669.class.getName(), 57 "--add-exports", "jdk.compiler/com.sun.tools.javac.api=ALL-UNNAMED" 58 }; 59 StringWriter sw = new StringWriter(); 60 JavacTask task = tool.getTask(sw, fm, null, Arrays.asList(opts), null, files); 61 boolean ok = task.call(); 62 String out = sw.toString(); 63 if (!out.isEmpty()) 64 System.err.println(out); 65 if (!ok) 66 throw new AssertionError("compilation of test program failed"); 67 // verify we found an annotated element to exercise the SourcePositions API 68 if (!out.contains("processing element")) 69 throw new AssertionError("expected text not found in compilation output"); 70 } 71 } 72 73 public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) { 74 Trees trees = Trees.instance(processingEnv); 75 SourcePositions sp = trees.getSourcePositions(); 76 Messager m = processingEnv.getMessager(); 77 m.printMessage(Diagnostic.Kind.NOTE, "processing annotations"); 78 int count = 0; 79 for (TypeElement anno: annotations) { 80 count++; 81 m.printMessage(Diagnostic.Kind.NOTE, " processing annotation " + anno); 82 for (Element e: roundEnv.getElementsAnnotatedWith(anno)) { 83 m.printMessage(Diagnostic.Kind.NOTE, " processing element " + e); 84 TreePath p = trees.getPath(e); 85 long start = sp.getStartPosition(p.getCompilationUnit(), p.getLeaf()); 86 long end = sp.getEndPosition(p.getCompilationUnit(), p.getLeaf()); 87 Diagnostic.Kind k = (start > 0 && end > 0 && start < end 88 ? Diagnostic.Kind.NOTE : Diagnostic.Kind.ERROR); 89 m.printMessage(k, "test [" + start + "," + end + "]", e); 90 } 91 } 92 if (count == 0) 93 m.printMessage(Diagnostic.Kind.NOTE, "no annotations found"); 94 return true; 95 } 96 97 @Override 98 public SourceVersion getSupportedSourceVersion() { 99 return SourceVersion.latest(); 100 } 101 }