1 /*
   2  * Copyright (c) 2006, 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 6439826 6411930 6380018 6392177
  27  * @summary Exception issuing Diagnostic while processing generated errant code
  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.util.*;
  34 import javax.annotation.processing.*;
  35 import javax.lang.model.*;
  36 import javax.lang.model.element.*;
  37 import javax.tools.*;
  38 import com.sun.source.util.*;
  39 import com.sun.tools.javac.api.*;
  40 import static javax.lang.model.util.ElementFilter.*;
  41 
  42 
  43 @SupportedAnnotationTypes("*")
  44 public class T6439826 extends AbstractProcessor {
  45     public static void main(String... args) throws IOException {
  46         String testSrc = System.getProperty("test.src", ".");
  47         String testClasses = System.getProperty("test.classes");
  48         JavacTool tool = JavacTool.create();
  49         MyDiagListener dl = new MyDiagListener();
  50         try (StandardJavaFileManager fm = tool.getStandardFileManager(dl, null, null)) {
  51             Iterable<? extends JavaFileObject> files =
  52                 fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrc, T6439826.class.getName()+".java")));
  53             Iterable<String> opts = Arrays.asList("-proc:only",
  54                                                   "-processor", "T6439826",
  55                                                   "-processorpath", testClasses);
  56             StringWriter out = new StringWriter();
  57             JavacTask task = tool.getTask(out, fm, dl, opts, null, files);
  58             task.call();
  59             String s = out.toString();
  60             System.err.print(s);
  61             // Expect the following 2 diagnostics, and no output to log
  62             //   Foo.java:1: illegal character: \35
  63             //   Foo.java:1: reached end of file while parsing
  64             System.err.println(dl.count + " diagnostics; " + s.length() + " characters");
  65             if (dl.count != 2 || s.length() != 0)
  66                 throw new AssertionError("unexpected output from compiler");
  67         }
  68     }
  69 
  70     public boolean process(Set<? extends TypeElement> annotations,
  71                            RoundEnvironment roundEnv) {
  72         Set<? extends TypeElement> elems = typesIn(roundEnv.getRootElements());
  73         for (TypeElement e: elems) {
  74             if (e.getSimpleName().toString().equals(T6439826.class.getName()))
  75                 writeBadFile();
  76         }
  77         return false;
  78     }
  79 
  80     @Override
  81     public SourceVersion getSupportedSourceVersion() {
  82         return SourceVersion.latest();
  83     }
  84 
  85     private void writeBadFile() {
  86         Filer filer = processingEnv.getFiler();
  87         Messager messager = processingEnv.getMessager();
  88         try {
  89             Writer out = filer.createSourceFile("Foo").openWriter();
  90             out.write("class Foo #"); // write a file that generates a scanner error
  91             out.close();
  92         } catch (IOException e) {
  93             messager.printMessage(Diagnostic.Kind.ERROR, e.toString());
  94         }
  95     }
  96 
  97     static class MyDiagListener implements DiagnosticListener {
  98         public void report(Diagnostic d) {
  99             System.err.println(d);
 100             count++;
 101         }
 102 
 103         public int count;
 104     }
 105 }