src/share/classes/com/sun/tools/javac/processing/JavacProcessingEnvironment.java

Mon, 10 Dec 2012 16:21:26 +0000

author
vromero
date
Mon, 10 Dec 2012 16:21:26 +0000
changeset 1442
fcf89720ae71
parent 1416
c0f0c41cafa0
child 1521
71f35e4b93a5
permissions
-rw-r--r--

8003967: detect and remove all mutable implicit static enum fields in langtools
Reviewed-by: jjg

duke@1 1 /*
jjg@1210 2 * Copyright (c) 2005, 2012, Oracle and/or its affiliates. All rights reserved.
duke@1 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
duke@1 4 *
duke@1 5 * This code is free software; you can redistribute it and/or modify it
duke@1 6 * under the terms of the GNU General Public License version 2 only, as
ohair@554 7 * published by the Free Software Foundation. Oracle designates this
duke@1 8 * particular file as subject to the "Classpath" exception as provided
ohair@554 9 * by Oracle in the LICENSE file that accompanied this code.
duke@1 10 *
duke@1 11 * This code is distributed in the hope that it will be useful, but WITHOUT
duke@1 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
duke@1 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
duke@1 14 * version 2 for more details (a copy is included in the LICENSE file that
duke@1 15 * accompanied this code).
duke@1 16 *
duke@1 17 * You should have received a copy of the GNU General Public License version
duke@1 18 * 2 along with this work; if not, write to the Free Software Foundation,
duke@1 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
duke@1 20 *
ohair@554 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
ohair@554 22 * or visit www.oracle.com if you need additional information or have any
ohair@554 23 * questions.
duke@1 24 */
duke@1 25
duke@1 26 package com.sun.tools.javac.processing;
duke@1 27
jjg@50 28 import java.io.Closeable;
jjg@50 29 import java.io.File;
jjg@50 30 import java.io.PrintWriter;
jjg@944 31 import java.io.StringWriter;
jjg@50 32 import java.net.MalformedURLException;
jjg@1210 33 import java.net.URL;
jjg@1210 34 import java.util.*;
jjg@1210 35 import java.util.regex.*;
duke@1 36
duke@1 37 import javax.annotation.processing.*;
duke@1 38 import javax.lang.model.SourceVersion;
duke@1 39 import javax.lang.model.element.AnnotationMirror;
duke@1 40 import javax.lang.model.element.Element;
jjg@1210 41 import javax.lang.model.element.PackageElement;
duke@1 42 import javax.lang.model.element.TypeElement;
duke@1 43 import javax.lang.model.util.*;
jjg@1210 44 import javax.tools.DiagnosticListener;
duke@1 45 import javax.tools.JavaFileManager;
jjg@1210 46 import javax.tools.JavaFileObject;
duke@1 47 import javax.tools.StandardJavaFileManager;
jjg@1210 48 import static javax.tools.StandardLocation.*;
jjg@50 49
jjg@1210 50 import com.sun.source.util.JavacTask;
jjg@50 51 import com.sun.source.util.TaskEvent;
jjg@1334 52 import com.sun.tools.javac.api.BasicJavacTask;
jjg@722 53 import com.sun.tools.javac.api.JavacTrees;
jjg@1210 54 import com.sun.tools.javac.api.MultiTaskListener;
jjg@50 55 import com.sun.tools.javac.code.*;
jjg@50 56 import com.sun.tools.javac.code.Symbol.*;
jjg@872 57 import com.sun.tools.javac.file.FSInfo;
jjg@50 58 import com.sun.tools.javac.file.JavacFileManager;
jjg@50 59 import com.sun.tools.javac.jvm.*;
jjg@952 60 import com.sun.tools.javac.jvm.ClassReader.BadClassFile;
jjg@50 61 import com.sun.tools.javac.main.JavaCompiler;
jjg@310 62 import com.sun.tools.javac.main.JavaCompiler.CompileState;
jjg@50 63 import com.sun.tools.javac.model.JavacElements;
jjg@50 64 import com.sun.tools.javac.model.JavacTypes;
jjg@50 65 import com.sun.tools.javac.parser.*;
jjg@50 66 import com.sun.tools.javac.tree.*;
jjg@50 67 import com.sun.tools.javac.tree.JCTree.*;
jjg@50 68 import com.sun.tools.javac.util.Abort;
jjg@816 69 import com.sun.tools.javac.util.Assert;
jjg@946 70 import com.sun.tools.javac.util.ClientCodeException;
jjg@50 71 import com.sun.tools.javac.util.Context;
jjg@483 72 import com.sun.tools.javac.util.Convert;
jjg@663 73 import com.sun.tools.javac.util.JCDiagnostic;
jjg@1210 74 import com.sun.tools.javac.util.JavacMessages;
jjg@50 75 import com.sun.tools.javac.util.List;
jjg@50 76 import com.sun.tools.javac.util.Log;
jjg@50 77 import com.sun.tools.javac.util.Name;
jjg@113 78 import com.sun.tools.javac.util.Names;
jjg@50 79 import com.sun.tools.javac.util.Options;
jjg@1210 80 import static com.sun.tools.javac.code.Lint.LintCategory.PROCESSING;
jjg@1210 81 import static com.sun.tools.javac.main.Option.*;
jjg@664 82 import static com.sun.tools.javac.util.JCDiagnostic.DiagnosticFlag.*;
duke@1 83
duke@1 84 /**
duke@1 85 * Objects of this class hold and manage the state needed to support
duke@1 86 * annotation processing.
duke@1 87 *
jjg@581 88 * <p><b>This is NOT part of any supported API.
duke@1 89 * If you write code that depends on this, you do so at your own risk.
duke@1 90 * This code and its internal interfaces are subject to change or
duke@1 91 * deletion without notice.</b>
duke@1 92 */
duke@1 93 public class JavacProcessingEnvironment implements ProcessingEnvironment, Closeable {
duke@1 94 Options options;
duke@1 95
duke@1 96 private final boolean printProcessorInfo;
duke@1 97 private final boolean printRounds;
duke@1 98 private final boolean verbose;
duke@1 99 private final boolean lint;
duke@1 100 private final boolean fatalErrors;
jjg@614 101 private final boolean werror;
jjg@664 102 private final boolean showResolveErrors;
duke@1 103
duke@1 104 private final JavacFiler filer;
duke@1 105 private final JavacMessager messager;
duke@1 106 private final JavacElements elementUtils;
duke@1 107 private final JavacTypes typeUtils;
duke@1 108
duke@1 109 /**
duke@1 110 * Holds relevant state history of which processors have been
duke@1 111 * used.
duke@1 112 */
duke@1 113 private DiscoveredProcessors discoveredProcs;
duke@1 114
duke@1 115 /**
duke@1 116 * Map of processor-specific options.
duke@1 117 */
duke@1 118 private final Map<String, String> processorOptions;
duke@1 119
duke@1 120 /**
duke@1 121 */
duke@1 122 private final Set<String> unmatchedProcessorOptions;
duke@1 123
duke@1 124 /**
duke@1 125 * Annotations implicitly processed and claimed by javac.
duke@1 126 */
duke@1 127 private final Set<String> platformAnnotations;
duke@1 128
duke@1 129 /**
duke@1 130 * Set of packages given on command line.
duke@1 131 */
duke@1 132 private Set<PackageSymbol> specifiedPackages = Collections.emptySet();
duke@1 133
duke@1 134 /** The log to be used for error reporting.
duke@1 135 */
duke@1 136 Log log;
duke@1 137
jjg@663 138 /** Diagnostic factory.
jjg@663 139 */
jjg@663 140 JCDiagnostic.Factory diags;
jjg@663 141
duke@1 142 /**
duke@1 143 * Source level of the compile.
duke@1 144 */
duke@1 145 Source source;
duke@1 146
jjg@372 147 private ClassLoader processorClassLoader;
jjg@1416 148 private SecurityException processorClassLoaderException;
jjg@372 149
mcimadamore@136 150 /**
mcimadamore@136 151 * JavacMessages object used for localization
mcimadamore@136 152 */
mcimadamore@136 153 private JavacMessages messages;
mcimadamore@136 154
jjg@1210 155 private MultiTaskListener taskListener;
jjg@1210 156
duke@1 157 private Context context;
duke@1 158
jjg@1416 159 /** Get the JavacProcessingEnvironment instance for this context. */
jjg@1416 160 public static JavacProcessingEnvironment instance(Context context) {
jjg@1416 161 JavacProcessingEnvironment instance = context.get(JavacProcessingEnvironment.class);
jjg@1416 162 if (instance == null)
jjg@1416 163 instance = new JavacProcessingEnvironment(context);
jjg@1416 164 return instance;
jjg@1416 165 }
jjg@1416 166
jjg@1416 167 protected JavacProcessingEnvironment(Context context) {
duke@1 168 this.context = context;
duke@1 169 log = Log.instance(context);
duke@1 170 source = Source.instance(context);
jjg@663 171 diags = JCDiagnostic.Factory.instance(context);
jjg@663 172 options = Options.instance(context);
jjg@700 173 printProcessorInfo = options.isSet(XPRINTPROCESSORINFO);
jjg@700 174 printRounds = options.isSet(XPRINTROUNDS);
jjg@700 175 verbose = options.isSet(VERBOSE);
jjg@700 176 lint = Lint.instance(context).isEnabled(PROCESSING);
jjg@1340 177 if (options.isSet(PROC, "only") || options.isSet(XPRINT)) {
jjg@1340 178 JavaCompiler compiler = JavaCompiler.instance(context);
jjg@1340 179 compiler.shouldStopPolicyIfNoError = CompileState.PROCESS;
jjg@1340 180 }
jjg@700 181 fatalErrors = options.isSet("fatalEnterError");
jjg@700 182 showResolveErrors = options.isSet("showResolveErrors");
jjg@700 183 werror = options.isSet(WERROR);
duke@1 184 platformAnnotations = initPlatformAnnotations();
duke@1 185
jjg@706 186 // Initialize services before any processors are initialized
duke@1 187 // in case processors use them.
duke@1 188 filer = new JavacFiler(context);
duke@1 189 messager = new JavacMessager(context, this);
jjg@706 190 elementUtils = JavacElements.instance(context);
jjg@706 191 typeUtils = JavacTypes.instance(context);
duke@1 192 processorOptions = initProcessorOptions(context);
duke@1 193 unmatchedProcessorOptions = initUnmatchedProcessorOptions();
mcimadamore@136 194 messages = JavacMessages.instance(context);
jjg@1210 195 taskListener = MultiTaskListener.instance(context);
jjg@1416 196 initProcessorClassLoader();
jjg@1416 197 }
jjg@1416 198
jjg@1416 199 public void setProcessors(Iterable<? extends Processor> processors) {
jjg@1416 200 Assert.checkNull(discoveredProcs);
duke@1 201 initProcessorIterator(context, processors);
duke@1 202 }
duke@1 203
duke@1 204 private Set<String> initPlatformAnnotations() {
duke@1 205 Set<String> platformAnnotations = new HashSet<String>();
duke@1 206 platformAnnotations.add("java.lang.Deprecated");
duke@1 207 platformAnnotations.add("java.lang.Override");
duke@1 208 platformAnnotations.add("java.lang.SuppressWarnings");
duke@1 209 platformAnnotations.add("java.lang.annotation.Documented");
duke@1 210 platformAnnotations.add("java.lang.annotation.Inherited");
duke@1 211 platformAnnotations.add("java.lang.annotation.Retention");
duke@1 212 platformAnnotations.add("java.lang.annotation.Target");
duke@1 213 return Collections.unmodifiableSet(platformAnnotations);
duke@1 214 }
duke@1 215
jjg@1416 216 private void initProcessorClassLoader() {
jjg@1416 217 JavaFileManager fileManager = context.get(JavaFileManager.class);
jjg@1416 218 try {
jjg@1416 219 // If processorpath is not explicitly set, use the classpath.
jjg@1416 220 processorClassLoader = fileManager.hasLocation(ANNOTATION_PROCESSOR_PATH)
jjg@1416 221 ? fileManager.getClassLoader(ANNOTATION_PROCESSOR_PATH)
jjg@1416 222 : fileManager.getClassLoader(CLASS_PATH);
jjg@1416 223
jjg@1416 224 if (processorClassLoader != null && processorClassLoader instanceof Closeable) {
jjg@1416 225 JavaCompiler compiler = JavaCompiler.instance(context);
jjg@1416 226 compiler.closeables = compiler.closeables.prepend((Closeable) processorClassLoader);
jjg@1416 227 }
jjg@1416 228 } catch (SecurityException e) {
jjg@1416 229 processorClassLoaderException = e;
jjg@1416 230 }
jjg@1416 231 }
jjg@1416 232
duke@1 233 private void initProcessorIterator(Context context, Iterable<? extends Processor> processors) {
duke@1 234 Log log = Log.instance(context);
duke@1 235 Iterator<? extends Processor> processorIterator;
duke@1 236
jjg@700 237 if (options.isSet(XPRINT)) {
duke@1 238 try {
duke@1 239 Processor processor = PrintingProcessor.class.newInstance();
duke@1 240 processorIterator = List.of(processor).iterator();
duke@1 241 } catch (Throwable t) {
duke@1 242 AssertionError assertError =
duke@1 243 new AssertionError("Problem instantiating PrintingProcessor.");
duke@1 244 assertError.initCause(t);
duke@1 245 throw assertError;
duke@1 246 }
duke@1 247 } else if (processors != null) {
duke@1 248 processorIterator = processors.iterator();
duke@1 249 } else {
jjg@700 250 String processorNames = options.get(PROCESSOR);
jjg@1416 251 if (processorClassLoaderException == null) {
duke@1 252 /*
duke@1 253 * If the "-processor" option is used, search the appropriate
duke@1 254 * path for the named class. Otherwise, use a service
duke@1 255 * provider mechanism to create the processor iterator.
duke@1 256 */
duke@1 257 if (processorNames != null) {
jjg@372 258 processorIterator = new NameProcessIterator(processorNames, processorClassLoader, log);
duke@1 259 } else {
jjg@372 260 processorIterator = new ServiceIterator(processorClassLoader, log);
duke@1 261 }
jjg@1416 262 } else {
duke@1 263 /*
duke@1 264 * A security exception will occur if we can't create a classloader.
duke@1 265 * Ignore the exception if, with hindsight, we didn't need it anyway
duke@1 266 * (i.e. no processor was specified either explicitly, or implicitly,
duke@1 267 * in service configuration file.) Otherwise, we cannot continue.
duke@1 268 */
jjg@1416 269 processorIterator = handleServiceLoaderUnavailability("proc.cant.create.loader",
jjg@1416 270 processorClassLoaderException);
duke@1 271 }
duke@1 272 }
duke@1 273 discoveredProcs = new DiscoveredProcessors(processorIterator);
duke@1 274 }
duke@1 275
duke@1 276 /**
duke@1 277 * Returns an empty processor iterator if no processors are on the
duke@1 278 * relevant path, otherwise if processors are present, logs an
duke@1 279 * error. Called when a service loader is unavailable for some
duke@1 280 * reason, either because a service loader class cannot be found
duke@1 281 * or because a security policy prevents class loaders from being
duke@1 282 * created.
duke@1 283 *
duke@1 284 * @param key The resource key to use to log an error message
duke@1 285 * @param e If non-null, pass this exception to Abort
duke@1 286 */
duke@1 287 private Iterator<Processor> handleServiceLoaderUnavailability(String key, Exception e) {
duke@1 288 JavaFileManager fileManager = context.get(JavaFileManager.class);
duke@1 289
duke@1 290 if (fileManager instanceof JavacFileManager) {
duke@1 291 StandardJavaFileManager standardFileManager = (JavacFileManager) fileManager;
duke@1 292 Iterable<? extends File> workingPath = fileManager.hasLocation(ANNOTATION_PROCESSOR_PATH)
duke@1 293 ? standardFileManager.getLocation(ANNOTATION_PROCESSOR_PATH)
duke@1 294 : standardFileManager.getLocation(CLASS_PATH);
duke@1 295
jjg@700 296 if (needClassLoader(options.get(PROCESSOR), workingPath) )
duke@1 297 handleException(key, e);
duke@1 298
duke@1 299 } else {
duke@1 300 handleException(key, e);
duke@1 301 }
duke@1 302
duke@1 303 java.util.List<Processor> pl = Collections.emptyList();
duke@1 304 return pl.iterator();
duke@1 305 }
duke@1 306
duke@1 307 /**
duke@1 308 * Handle a security exception thrown during initializing the
duke@1 309 * Processor iterator.
duke@1 310 */
duke@1 311 private void handleException(String key, Exception e) {
duke@1 312 if (e != null) {
duke@1 313 log.error(key, e.getLocalizedMessage());
duke@1 314 throw new Abort(e);
duke@1 315 } else {
duke@1 316 log.error(key);
duke@1 317 throw new Abort();
duke@1 318 }
duke@1 319 }
duke@1 320
duke@1 321 /**
duke@1 322 * Use a service loader appropriate for the platform to provide an
darcy@1048 323 * iterator over annotations processors; fails if a loader is
darcy@1048 324 * needed but unavailable.
duke@1 325 */
duke@1 326 private class ServiceIterator implements Iterator<Processor> {
darcy@1048 327 private Iterator<Processor> iterator;
duke@1 328 private Log log;
darcy@1048 329 private ServiceLoader<Processor> loader;
duke@1 330
duke@1 331 ServiceIterator(ClassLoader classLoader, Log log) {
duke@1 332 this.log = log;
duke@1 333 try {
duke@1 334 try {
darcy@1048 335 loader = ServiceLoader.load(Processor.class, classLoader);
darcy@1048 336 this.iterator = loader.iterator();
darcy@1048 337 } catch (Exception e) {
darcy@1048 338 // Fail softly if a loader is not actually needed.
darcy@1048 339 this.iterator = handleServiceLoaderUnavailability("proc.no.service", null);
duke@1 340 }
duke@1 341 } catch (Throwable t) {
duke@1 342 log.error("proc.service.problem");
duke@1 343 throw new Abort(t);
duke@1 344 }
duke@1 345 }
duke@1 346
duke@1 347 public boolean hasNext() {
duke@1 348 try {
duke@1 349 return iterator.hasNext();
darcy@1048 350 } catch(ServiceConfigurationError sce) {
darcy@1048 351 log.error("proc.bad.config.file", sce.getLocalizedMessage());
darcy@1048 352 throw new Abort(sce);
duke@1 353 } catch (Throwable t) {
duke@1 354 throw new Abort(t);
duke@1 355 }
duke@1 356 }
duke@1 357
duke@1 358 public Processor next() {
duke@1 359 try {
darcy@1048 360 return iterator.next();
darcy@1048 361 } catch (ServiceConfigurationError sce) {
darcy@1048 362 log.error("proc.bad.config.file", sce.getLocalizedMessage());
darcy@1048 363 throw new Abort(sce);
duke@1 364 } catch (Throwable t) {
duke@1 365 throw new Abort(t);
duke@1 366 }
duke@1 367 }
duke@1 368
duke@1 369 public void remove() {
duke@1 370 throw new UnsupportedOperationException();
duke@1 371 }
darcy@382 372
darcy@382 373 public void close() {
darcy@1048 374 if (loader != null) {
darcy@382 375 try {
darcy@1048 376 loader.reload();
darcy@382 377 } catch(Exception e) {
darcy@382 378 ; // Ignore problems during a call to reload.
darcy@382 379 }
darcy@382 380 }
darcy@382 381 }
duke@1 382 }
duke@1 383
duke@1 384
duke@1 385 private static class NameProcessIterator implements Iterator<Processor> {
duke@1 386 Processor nextProc = null;
duke@1 387 Iterator<String> names;
duke@1 388 ClassLoader processorCL;
duke@1 389 Log log;
duke@1 390
duke@1 391 NameProcessIterator(String names, ClassLoader processorCL, Log log) {
duke@1 392 this.names = Arrays.asList(names.split(",")).iterator();
duke@1 393 this.processorCL = processorCL;
duke@1 394 this.log = log;
duke@1 395 }
duke@1 396
duke@1 397 public boolean hasNext() {
duke@1 398 if (nextProc != null)
duke@1 399 return true;
duke@1 400 else {
duke@1 401 if (!names.hasNext())
duke@1 402 return false;
duke@1 403 else {
duke@1 404 String processorName = names.next();
duke@1 405
duke@1 406 Processor processor;
duke@1 407 try {
duke@1 408 try {
duke@1 409 processor =
duke@1 410 (Processor) (processorCL.loadClass(processorName).newInstance());
duke@1 411 } catch (ClassNotFoundException cnfe) {
duke@1 412 log.error("proc.processor.not.found", processorName);
duke@1 413 return false;
duke@1 414 } catch (ClassCastException cce) {
duke@1 415 log.error("proc.processor.wrong.type", processorName);
duke@1 416 return false;
duke@1 417 } catch (Exception e ) {
duke@1 418 log.error("proc.processor.cant.instantiate", processorName);
duke@1 419 return false;
duke@1 420 }
jjg@946 421 } catch(ClientCodeException e) {
jjg@946 422 throw e;
duke@1 423 } catch(Throwable t) {
duke@1 424 throw new AnnotationProcessingError(t);
duke@1 425 }
duke@1 426 nextProc = processor;
duke@1 427 return true;
duke@1 428 }
duke@1 429
duke@1 430 }
duke@1 431 }
duke@1 432
duke@1 433 public Processor next() {
duke@1 434 if (hasNext()) {
duke@1 435 Processor p = nextProc;
duke@1 436 nextProc = null;
duke@1 437 return p;
duke@1 438 } else
duke@1 439 throw new NoSuchElementException();
duke@1 440 }
duke@1 441
duke@1 442 public void remove () {
duke@1 443 throw new UnsupportedOperationException();
duke@1 444 }
duke@1 445 }
duke@1 446
duke@1 447 public boolean atLeastOneProcessor() {
duke@1 448 return discoveredProcs.iterator().hasNext();
duke@1 449 }
duke@1 450
duke@1 451 private Map<String, String> initProcessorOptions(Context context) {
duke@1 452 Options options = Options.instance(context);
duke@1 453 Set<String> keySet = options.keySet();
duke@1 454 Map<String, String> tempOptions = new LinkedHashMap<String, String>();
duke@1 455
duke@1 456 for(String key : keySet) {
duke@1 457 if (key.startsWith("-A") && key.length() > 2) {
duke@1 458 int sepIndex = key.indexOf('=');
duke@1 459 String candidateKey = null;
duke@1 460 String candidateValue = null;
duke@1 461
duke@1 462 if (sepIndex == -1)
duke@1 463 candidateKey = key.substring(2);
duke@1 464 else if (sepIndex >= 3) {
duke@1 465 candidateKey = key.substring(2, sepIndex);
duke@1 466 candidateValue = (sepIndex < key.length()-1)?
duke@1 467 key.substring(sepIndex+1) : null;
duke@1 468 }
duke@1 469 tempOptions.put(candidateKey, candidateValue);
duke@1 470 }
duke@1 471 }
duke@1 472
duke@1 473 return Collections.unmodifiableMap(tempOptions);
duke@1 474 }
duke@1 475
duke@1 476 private Set<String> initUnmatchedProcessorOptions() {
duke@1 477 Set<String> unmatchedProcessorOptions = new HashSet<String>();
duke@1 478 unmatchedProcessorOptions.addAll(processorOptions.keySet());
duke@1 479 return unmatchedProcessorOptions;
duke@1 480 }
duke@1 481
duke@1 482 /**
duke@1 483 * State about how a processor has been used by the tool. If a
duke@1 484 * processor has been used on a prior round, its process method is
duke@1 485 * called on all subsequent rounds, perhaps with an empty set of
jjg@1340 486 * annotations to process. The {@code annotationSupported} method
duke@1 487 * caches the supported annotation information from the first (and
duke@1 488 * only) getSupportedAnnotationTypes call to the processor.
duke@1 489 */
duke@1 490 static class ProcessorState {
duke@1 491 public Processor processor;
duke@1 492 public boolean contributed;
duke@1 493 private ArrayList<Pattern> supportedAnnotationPatterns;
duke@1 494 private ArrayList<String> supportedOptionNames;
duke@1 495
duke@1 496 ProcessorState(Processor p, Log log, Source source, ProcessingEnvironment env) {
duke@1 497 processor = p;
duke@1 498 contributed = false;
duke@1 499
duke@1 500 try {
duke@1 501 processor.init(env);
duke@1 502
duke@1 503 checkSourceVersionCompatibility(source, log);
duke@1 504
duke@1 505 supportedAnnotationPatterns = new ArrayList<Pattern>();
duke@1 506 for (String importString : processor.getSupportedAnnotationTypes()) {
duke@1 507 supportedAnnotationPatterns.add(importStringToPattern(importString,
duke@1 508 processor,
duke@1 509 log));
duke@1 510 }
duke@1 511
duke@1 512 supportedOptionNames = new ArrayList<String>();
duke@1 513 for (String optionName : processor.getSupportedOptions() ) {
duke@1 514 if (checkOptionName(optionName, log))
duke@1 515 supportedOptionNames.add(optionName);
duke@1 516 }
duke@1 517
jjg@946 518 } catch (ClientCodeException e) {
jjg@946 519 throw e;
duke@1 520 } catch (Throwable t) {
duke@1 521 throw new AnnotationProcessingError(t);
duke@1 522 }
duke@1 523 }
duke@1 524
duke@1 525 /**
duke@1 526 * Checks whether or not a processor's source version is
duke@1 527 * compatible with the compilation source version. The
duke@1 528 * processor's source version needs to be greater than or
duke@1 529 * equal to the source version of the compile.
duke@1 530 */
duke@1 531 private void checkSourceVersionCompatibility(Source source, Log log) {
duke@1 532 SourceVersion procSourceVersion = processor.getSupportedSourceVersion();
duke@1 533
duke@1 534 if (procSourceVersion.compareTo(Source.toSourceVersion(source)) < 0 ) {
duke@1 535 log.warning("proc.processor.incompatible.source.version",
duke@1 536 procSourceVersion,
duke@1 537 processor.getClass().getName(),
duke@1 538 source.name);
duke@1 539 }
duke@1 540 }
duke@1 541
duke@1 542 private boolean checkOptionName(String optionName, Log log) {
duke@1 543 boolean valid = isValidOptionName(optionName);
duke@1 544 if (!valid)
duke@1 545 log.error("proc.processor.bad.option.name",
duke@1 546 optionName,
duke@1 547 processor.getClass().getName());
duke@1 548 return valid;
duke@1 549 }
duke@1 550
duke@1 551 public boolean annotationSupported(String annotationName) {
duke@1 552 for(Pattern p: supportedAnnotationPatterns) {
duke@1 553 if (p.matcher(annotationName).matches())
duke@1 554 return true;
duke@1 555 }
duke@1 556 return false;
duke@1 557 }
duke@1 558
duke@1 559 /**
duke@1 560 * Remove options that are matched by this processor.
duke@1 561 */
duke@1 562 public void removeSupportedOptions(Set<String> unmatchedProcessorOptions) {
duke@1 563 unmatchedProcessorOptions.removeAll(supportedOptionNames);
duke@1 564 }
duke@1 565 }
duke@1 566
duke@1 567 // TODO: These two classes can probably be rewritten better...
duke@1 568 /**
duke@1 569 * This class holds information about the processors that have
duke@1 570 * been discoverd so far as well as the means to discover more, if
duke@1 571 * necessary. A single iterator should be used per round of
duke@1 572 * annotation processing. The iterator first visits already
darcy@382 573 * discovered processors then fails over to the service provider
duke@1 574 * mechanism if additional queries are made.
duke@1 575 */
duke@1 576 class DiscoveredProcessors implements Iterable<ProcessorState> {
duke@1 577
duke@1 578 class ProcessorStateIterator implements Iterator<ProcessorState> {
duke@1 579 DiscoveredProcessors psi;
duke@1 580 Iterator<ProcessorState> innerIter;
duke@1 581 boolean onProcInterator;
duke@1 582
duke@1 583 ProcessorStateIterator(DiscoveredProcessors psi) {
duke@1 584 this.psi = psi;
duke@1 585 this.innerIter = psi.procStateList.iterator();
duke@1 586 this.onProcInterator = false;
duke@1 587 }
duke@1 588
duke@1 589 public ProcessorState next() {
duke@1 590 if (!onProcInterator) {
duke@1 591 if (innerIter.hasNext())
duke@1 592 return innerIter.next();
duke@1 593 else
duke@1 594 onProcInterator = true;
duke@1 595 }
duke@1 596
duke@1 597 if (psi.processorIterator.hasNext()) {
duke@1 598 ProcessorState ps = new ProcessorState(psi.processorIterator.next(),
duke@1 599 log, source, JavacProcessingEnvironment.this);
duke@1 600 psi.procStateList.add(ps);
duke@1 601 return ps;
duke@1 602 } else
duke@1 603 throw new NoSuchElementException();
duke@1 604 }
duke@1 605
duke@1 606 public boolean hasNext() {
duke@1 607 if (onProcInterator)
duke@1 608 return psi.processorIterator.hasNext();
duke@1 609 else
duke@1 610 return innerIter.hasNext() || psi.processorIterator.hasNext();
duke@1 611 }
duke@1 612
duke@1 613 public void remove () {
duke@1 614 throw new UnsupportedOperationException();
duke@1 615 }
duke@1 616
duke@1 617 /**
duke@1 618 * Run all remaining processors on the procStateList that
duke@1 619 * have not already run this round with an empty set of
duke@1 620 * annotations.
duke@1 621 */
duke@1 622 public void runContributingProcs(RoundEnvironment re) {
duke@1 623 if (!onProcInterator) {
duke@1 624 Set<TypeElement> emptyTypeElements = Collections.emptySet();
duke@1 625 while(innerIter.hasNext()) {
duke@1 626 ProcessorState ps = innerIter.next();
duke@1 627 if (ps.contributed)
duke@1 628 callProcessor(ps.processor, emptyTypeElements, re);
duke@1 629 }
duke@1 630 }
duke@1 631 }
duke@1 632 }
duke@1 633
duke@1 634 Iterator<? extends Processor> processorIterator;
duke@1 635 ArrayList<ProcessorState> procStateList;
duke@1 636
duke@1 637 public ProcessorStateIterator iterator() {
duke@1 638 return new ProcessorStateIterator(this);
duke@1 639 }
duke@1 640
duke@1 641 DiscoveredProcessors(Iterator<? extends Processor> processorIterator) {
duke@1 642 this.processorIterator = processorIterator;
duke@1 643 this.procStateList = new ArrayList<ProcessorState>();
duke@1 644 }
darcy@382 645
darcy@382 646 /**
darcy@382 647 * Free jar files, etc. if using a service loader.
darcy@382 648 */
darcy@382 649 public void close() {
darcy@382 650 if (processorIterator != null &&
darcy@382 651 processorIterator instanceof ServiceIterator) {
darcy@382 652 ((ServiceIterator) processorIterator).close();
darcy@382 653 }
darcy@382 654 }
duke@1 655 }
duke@1 656
duke@1 657 private void discoverAndRunProcs(Context context,
duke@1 658 Set<TypeElement> annotationsPresent,
duke@1 659 List<ClassSymbol> topLevelClasses,
duke@1 660 List<PackageSymbol> packageInfoFiles) {
duke@1 661 Map<String, TypeElement> unmatchedAnnotations =
duke@1 662 new HashMap<String, TypeElement>(annotationsPresent.size());
duke@1 663
duke@1 664 for(TypeElement a : annotationsPresent) {
duke@1 665 unmatchedAnnotations.put(a.getQualifiedName().toString(),
duke@1 666 a);
duke@1 667 }
duke@1 668
duke@1 669 // Give "*" processors a chance to match
duke@1 670 if (unmatchedAnnotations.size() == 0)
duke@1 671 unmatchedAnnotations.put("", null);
duke@1 672
duke@1 673 DiscoveredProcessors.ProcessorStateIterator psi = discoveredProcs.iterator();
duke@1 674 // TODO: Create proper argument values; need past round
duke@1 675 // information to fill in this constructor. Note that the 1
duke@1 676 // st round of processing could be the last round if there
duke@1 677 // were parse errors on the initial source files; however, we
duke@1 678 // are not doing processing in that case.
duke@1 679
duke@1 680 Set<Element> rootElements = new LinkedHashSet<Element>();
duke@1 681 rootElements.addAll(topLevelClasses);
duke@1 682 rootElements.addAll(packageInfoFiles);
duke@1 683 rootElements = Collections.unmodifiableSet(rootElements);
duke@1 684
duke@1 685 RoundEnvironment renv = new JavacRoundEnvironment(false,
duke@1 686 false,
duke@1 687 rootElements,
duke@1 688 JavacProcessingEnvironment.this);
duke@1 689
duke@1 690 while(unmatchedAnnotations.size() > 0 && psi.hasNext() ) {
duke@1 691 ProcessorState ps = psi.next();
duke@1 692 Set<String> matchedNames = new HashSet<String>();
duke@1 693 Set<TypeElement> typeElements = new LinkedHashSet<TypeElement>();
darcy@506 694
darcy@506 695 for (Map.Entry<String, TypeElement> entry: unmatchedAnnotations.entrySet()) {
darcy@506 696 String unmatchedAnnotationName = entry.getKey();
duke@1 697 if (ps.annotationSupported(unmatchedAnnotationName) ) {
duke@1 698 matchedNames.add(unmatchedAnnotationName);
darcy@506 699 TypeElement te = entry.getValue();
duke@1 700 if (te != null)
duke@1 701 typeElements.add(te);
duke@1 702 }
duke@1 703 }
duke@1 704
duke@1 705 if (matchedNames.size() > 0 || ps.contributed) {
duke@1 706 boolean processingResult = callProcessor(ps.processor, typeElements, renv);
duke@1 707 ps.contributed = true;
duke@1 708 ps.removeSupportedOptions(unmatchedProcessorOptions);
duke@1 709
duke@1 710 if (printProcessorInfo || verbose) {
jjg@1136 711 log.printLines("x.print.processor.info",
jjg@604 712 ps.processor.getClass().getName(),
jjg@604 713 matchedNames.toString(),
jjg@604 714 processingResult);
duke@1 715 }
duke@1 716
duke@1 717 if (processingResult) {
duke@1 718 unmatchedAnnotations.keySet().removeAll(matchedNames);
duke@1 719 }
duke@1 720
duke@1 721 }
duke@1 722 }
duke@1 723 unmatchedAnnotations.remove("");
duke@1 724
duke@1 725 if (lint && unmatchedAnnotations.size() > 0) {
duke@1 726 // Remove annotations processed by javac
duke@1 727 unmatchedAnnotations.keySet().removeAll(platformAnnotations);
duke@1 728 if (unmatchedAnnotations.size() > 0) {
duke@1 729 log = Log.instance(context);
duke@1 730 log.warning("proc.annotations.without.processors",
duke@1 731 unmatchedAnnotations.keySet());
duke@1 732 }
duke@1 733 }
duke@1 734
duke@1 735 // Run contributing processors that haven't run yet
duke@1 736 psi.runContributingProcs(renv);
duke@1 737
duke@1 738 // Debugging
jjg@700 739 if (options.isSet("displayFilerState"))
duke@1 740 filer.displayState();
duke@1 741 }
duke@1 742
duke@1 743 /**
duke@1 744 * Computes the set of annotations on the symbol in question.
duke@1 745 * Leave class public for external testing purposes.
duke@1 746 */
duke@1 747 public static class ComputeAnnotationSet extends
darcy@1054 748 ElementScanner8<Set<TypeElement>, Set<TypeElement>> {
duke@1 749 final Elements elements;
duke@1 750
duke@1 751 public ComputeAnnotationSet(Elements elements) {
duke@1 752 super();
duke@1 753 this.elements = elements;
duke@1 754 }
duke@1 755
duke@1 756 @Override
duke@1 757 public Set<TypeElement> visitPackage(PackageElement e, Set<TypeElement> p) {
duke@1 758 // Don't scan enclosed elements of a package
duke@1 759 return p;
duke@1 760 }
duke@1 761
duke@1 762 @Override
jjg@620 763 public Set<TypeElement> scan(Element e, Set<TypeElement> p) {
duke@1 764 for (AnnotationMirror annotationMirror :
duke@1 765 elements.getAllAnnotationMirrors(e) ) {
duke@1 766 Element e2 = annotationMirror.getAnnotationType().asElement();
duke@1 767 p.add((TypeElement) e2);
duke@1 768 }
duke@1 769 return super.scan(e, p);
duke@1 770 }
duke@1 771 }
duke@1 772
duke@1 773 private boolean callProcessor(Processor proc,
duke@1 774 Set<? extends TypeElement> tes,
duke@1 775 RoundEnvironment renv) {
duke@1 776 try {
duke@1 777 return proc.process(tes, renv);
jjg@952 778 } catch (BadClassFile ex) {
jjg@952 779 log.error("proc.cant.access.1", ex.sym, ex.getDetailValue());
jjg@952 780 return false;
duke@1 781 } catch (CompletionFailure ex) {
duke@1 782 StringWriter out = new StringWriter();
duke@1 783 ex.printStackTrace(new PrintWriter(out));
jjg@12 784 log.error("proc.cant.access", ex.sym, ex.getDetailValue(), out.toString());
duke@1 785 return false;
jjg@946 786 } catch (ClientCodeException e) {
jjg@946 787 throw e;
duke@1 788 } catch (Throwable t) {
duke@1 789 throw new AnnotationProcessingError(t);
duke@1 790 }
duke@1 791 }
duke@1 792
jjg@620 793 /**
jjg@620 794 * Helper object for a single round of annotation processing.
jjg@620 795 */
jjg@620 796 class Round {
jjg@620 797 /** The round number. */
jjg@620 798 final int number;
jjg@620 799 /** The context for the round. */
jjg@620 800 final Context context;
jjg@620 801 /** The compiler for the round. */
jjg@620 802 final JavaCompiler compiler;
jjg@620 803 /** The log for the round. */
jjg@620 804 final Log log;
jjg@1406 805 /** The diagnostic handler for the round. */
jjg@1406 806 final Log.DeferredDiagnosticHandler deferredDiagnosticHandler;
jjg@642 807
jjg@642 808 /** The ASTs to be compiled. */
jjg@642 809 List<JCCompilationUnit> roots;
jjg@642 810 /** The classes to be compiler that have were generated. */
jjg@642 811 Map<String, JavaFileObject> genClassFiles;
jjg@620 812
jjg@620 813 /** The set of annotations to be processed this round. */
jjg@620 814 Set<TypeElement> annotationsPresent;
jjg@620 815 /** The set of top level classes to be processed this round. */
jjg@620 816 List<ClassSymbol> topLevelClasses;
jjg@620 817 /** The set of package-info files to be processed this round. */
jjg@620 818 List<PackageSymbol> packageInfoFiles;
jjg@620 819
jjg@933 820 /** The number of Messager errors generated in this round. */
jjg@933 821 int nMessagerErrors;
jjg@933 822
jjg@642 823 /** Create a round (common code). */
jjg@1406 824 private Round(Context context, int number, int priorErrors, int priorWarnings,
jjg@1406 825 Log.DeferredDiagnosticHandler deferredDiagnosticHandler) {
jjg@620 826 this.context = context;
jjg@620 827 this.number = number;
jjg@642 828
jjg@620 829 compiler = JavaCompiler.instance(context);
jjg@620 830 log = Log.instance(context);
jjg@933 831 log.nerrors = priorErrors;
jjg@898 832 log.nwarnings += priorWarnings;
jjg@1406 833 if (number == 1) {
jjg@1406 834 Assert.checkNonNull(deferredDiagnosticHandler);
jjg@1406 835 this.deferredDiagnosticHandler = deferredDiagnosticHandler;
jjg@1406 836 } else {
jjg@1406 837 this.deferredDiagnosticHandler = new Log.DeferredDiagnosticHandler(log);
jjg@1406 838 }
jjg@620 839
jjg@620 840 // the following is for the benefit of JavacProcessingEnvironment.getContext()
jjg@620 841 JavacProcessingEnvironment.this.context = context;
jjg@620 842
jjg@620 843 // the following will be populated as needed
jjg@620 844 topLevelClasses = List.nil();
jjg@620 845 packageInfoFiles = List.nil();
jjg@620 846 }
jjg@620 847
jjg@642 848 /** Create the first round. */
jjg@1406 849 Round(Context context, List<JCCompilationUnit> roots, List<ClassSymbol> classSymbols,
jjg@1406 850 Log.DeferredDiagnosticHandler deferredDiagnosticHandler) {
jjg@1406 851 this(context, 1, 0, 0, deferredDiagnosticHandler);
jjg@642 852 this.roots = roots;
jjg@642 853 genClassFiles = new HashMap<String,JavaFileObject>();
jjg@642 854
jjg@642 855 compiler.todo.clear(); // free the compiler's resources
jjg@642 856
jjg@642 857 // The reverse() in the following line is to maintain behavioural
jjg@642 858 // compatibility with the previous revision of the code. Strictly speaking,
jjg@642 859 // it should not be necessary, but a javah golden file test fails without it.
jjg@642 860 topLevelClasses =
jjg@642 861 getTopLevelClasses(roots).prependList(classSymbols.reverse());
jjg@642 862
jjg@642 863 packageInfoFiles = getPackageInfoFiles(roots);
jjg@642 864
jjg@642 865 findAnnotationsPresent();
jjg@642 866 }
jjg@642 867
jjg@642 868 /** Create a new round. */
jjg@642 869 private Round(Round prev,
jjg@663 870 Set<JavaFileObject> newSourceFiles, Map<String,JavaFileObject> newClassFiles) {
jjg@933 871 this(prev.nextContext(),
jjg@933 872 prev.number+1,
jjg@933 873 prev.nMessagerErrors,
jjg@1406 874 prev.compiler.log.nwarnings,
jjg@1406 875 null);
jjg@642 876 this.genClassFiles = prev.genClassFiles;
jjg@642 877
jjg@642 878 List<JCCompilationUnit> parsedFiles = compiler.parseFiles(newSourceFiles);
jjg@642 879 roots = cleanTrees(prev.roots).appendList(parsedFiles);
jjg@642 880
jjg@642 881 // Check for errors after parsing
jjg@642 882 if (unrecoverableError())
jjg@642 883 return;
jjg@642 884
jjg@642 885 enterClassFiles(genClassFiles);
jjg@642 886 List<ClassSymbol> newClasses = enterClassFiles(newClassFiles);
jjg@642 887 genClassFiles.putAll(newClassFiles);
jjg@642 888 enterTrees(roots);
jjg@642 889
jjg@642 890 if (unrecoverableError())
jjg@642 891 return;
jjg@642 892
jjg@642 893 topLevelClasses = join(
jjg@642 894 getTopLevelClasses(parsedFiles),
jjg@642 895 getTopLevelClassesFromClasses(newClasses));
jjg@642 896
jjg@642 897 packageInfoFiles = join(
jjg@642 898 getPackageInfoFiles(parsedFiles),
jjg@642 899 getPackageInfoFilesFromClasses(newClasses));
jjg@642 900
jjg@642 901 findAnnotationsPresent();
jjg@642 902 }
jjg@642 903
jjg@620 904 /** Create the next round to be used. */
jjg@663 905 Round next(Set<JavaFileObject> newSourceFiles, Map<String, JavaFileObject> newClassFiles) {
jjg@642 906 try {
jjg@642 907 return new Round(this, newSourceFiles, newClassFiles);
jjg@642 908 } finally {
jjg@642 909 compiler.close(false);
jjg@642 910 }
jjg@642 911 }
jjg@642 912
jjg@642 913 /** Create the compiler to be used for the final compilation. */
jjg@642 914 JavaCompiler finalCompiler(boolean errorStatus) {
jjg@642 915 try {
jjg@1340 916 Context nextCtx = nextContext();
jjg@1340 917 JavacProcessingEnvironment.this.context = nextCtx;
jjg@1340 918 JavaCompiler c = JavaCompiler.instance(nextCtx);
jjg@898 919 c.log.nwarnings += compiler.log.nwarnings;
jjg@642 920 if (errorStatus) {
jjg@642 921 c.log.nerrors += compiler.log.nerrors;
jjg@642 922 }
jjg@642 923 return c;
jjg@642 924 } finally {
jjg@642 925 compiler.close(false);
jjg@642 926 }
jjg@620 927 }
jjg@620 928
jjg@620 929 /** Return the number of errors found so far in this round.
jjg@620 930 * This may include uncoverable errors, such as parse errors,
jjg@620 931 * and transient errors, such as missing symbols. */
jjg@620 932 int errorCount() {
jjg@620 933 return compiler.errorCount();
jjg@620 934 }
jjg@620 935
jjg@620 936 /** Return the number of warnings found so far in this round. */
jjg@620 937 int warningCount() {
jjg@620 938 return compiler.warningCount();
jjg@620 939 }
jjg@620 940
jjg@620 941 /** Return whether or not an unrecoverable error has occurred. */
jjg@620 942 boolean unrecoverableError() {
jjg@664 943 if (messager.errorRaised())
jjg@664 944 return true;
jjg@664 945
jjg@1406 946 for (JCDiagnostic d: deferredDiagnosticHandler.getDiagnostics()) {
jjg@664 947 switch (d.getKind()) {
jjg@664 948 case WARNING:
jjg@664 949 if (werror)
jjg@664 950 return true;
jjg@664 951 break;
jjg@664 952
jjg@664 953 case ERROR:
jjg@726 954 if (fatalErrors || !d.isFlagSet(RECOVERABLE))
jjg@664 955 return true;
jjg@664 956 break;
jjg@664 957 }
jjg@664 958 }
jjg@664 959
jjg@664 960 return false;
jjg@620 961 }
jjg@620 962
jjg@620 963 /** Find the set of annotations present in the set of top level
jjg@642 964 * classes and package info files to be processed this round. */
jjg@642 965 void findAnnotationsPresent() {
jjg@642 966 ComputeAnnotationSet annotationComputer = new ComputeAnnotationSet(elementUtils);
jjg@620 967 // Use annotation processing to compute the set of annotations present
jjg@620 968 annotationsPresent = new LinkedHashSet<TypeElement>();
jjg@620 969 for (ClassSymbol classSym : topLevelClasses)
jjg@620 970 annotationComputer.scan(classSym, annotationsPresent);
jjg@620 971 for (PackageSymbol pkgSym : packageInfoFiles)
jjg@620 972 annotationComputer.scan(pkgSym, annotationsPresent);
jjg@620 973 }
jjg@620 974
jjg@642 975 /** Enter a set of generated class files. */
jjg@664 976 private List<ClassSymbol> enterClassFiles(Map<String, JavaFileObject> classFiles) {
jjg@620 977 ClassReader reader = ClassReader.instance(context);
jjg@620 978 Names names = Names.instance(context);
jjg@620 979 List<ClassSymbol> list = List.nil();
jjg@620 980
jjg@642 981 for (Map.Entry<String,JavaFileObject> entry : classFiles.entrySet()) {
jjg@620 982 Name name = names.fromString(entry.getKey());
jjg@620 983 JavaFileObject file = entry.getValue();
jjg@620 984 if (file.getKind() != JavaFileObject.Kind.CLASS)
jjg@620 985 throw new AssertionError(file);
jjg@620 986 ClassSymbol cs;
jjg@620 987 if (isPkgInfo(file, JavaFileObject.Kind.CLASS)) {
jjg@620 988 Name packageName = Convert.packagePart(name);
jjg@620 989 PackageSymbol p = reader.enterPackage(packageName);
jjg@620 990 if (p.package_info == null)
jjg@620 991 p.package_info = reader.enterClass(Convert.shortName(name), p);
jjg@620 992 cs = p.package_info;
jjg@620 993 if (cs.classfile == null)
jjg@620 994 cs.classfile = file;
jjg@620 995 } else
jjg@620 996 cs = reader.enterClass(name, file);
jjg@620 997 list = list.prepend(cs);
jjg@620 998 }
jjg@620 999 return list.reverse();
jjg@620 1000 }
jjg@620 1001
jjg@620 1002 /** Enter a set of syntax trees. */
jjg@664 1003 private void enterTrees(List<JCCompilationUnit> roots) {
jjg@620 1004 compiler.enterTrees(roots);
jjg@620 1005 }
jjg@620 1006
jjg@620 1007 /** Run a processing round. */
jjg@620 1008 void run(boolean lastRound, boolean errorStatus) {
jjg@642 1009 printRoundInfo(lastRound);
jjg@620 1010
jjg@1210 1011 if (!taskListener.isEmpty())
jjg@620 1012 taskListener.started(new TaskEvent(TaskEvent.Kind.ANNOTATION_PROCESSING_ROUND));
jjg@620 1013
jjg@620 1014 try {
jjg@620 1015 if (lastRound) {
jjg@620 1016 filer.setLastRound(true);
jjg@620 1017 Set<Element> emptyRootElements = Collections.emptySet(); // immutable
jjg@620 1018 RoundEnvironment renv = new JavacRoundEnvironment(true,
jjg@620 1019 errorStatus,
jjg@620 1020 emptyRootElements,
jjg@620 1021 JavacProcessingEnvironment.this);
jjg@620 1022 discoveredProcs.iterator().runContributingProcs(renv);
jjg@620 1023 } else {
jjg@620 1024 discoverAndRunProcs(context, annotationsPresent, topLevelClasses, packageInfoFiles);
jjg@620 1025 }
jjg@620 1026 } finally {
jjg@1210 1027 if (!taskListener.isEmpty())
jjg@620 1028 taskListener.finished(new TaskEvent(TaskEvent.Kind.ANNOTATION_PROCESSING_ROUND));
jjg@620 1029 }
jjg@933 1030
jjg@933 1031 nMessagerErrors = messager.errorCount();
jjg@620 1032 }
jjg@620 1033
jjg@664 1034 void showDiagnostics(boolean showAll) {
jjg@664 1035 Set<JCDiagnostic.Kind> kinds = EnumSet.allOf(JCDiagnostic.Kind.class);
jjg@664 1036 if (!showAll) {
jjg@664 1037 // suppress errors, which are all presumed to be transient resolve errors
jjg@664 1038 kinds.remove(JCDiagnostic.Kind.ERROR);
jjg@664 1039 }
jjg@1406 1040 deferredDiagnosticHandler.reportDeferredDiagnostics(kinds);
jjg@1406 1041 log.popDiagnosticHandler(deferredDiagnosticHandler);
jjg@664 1042 }
jjg@664 1043
jjg@620 1044 /** Print info about this round. */
jjg@642 1045 private void printRoundInfo(boolean lastRound) {
jjg@620 1046 if (printRounds || verbose) {
jjg@642 1047 List<ClassSymbol> tlc = lastRound ? List.<ClassSymbol>nil() : topLevelClasses;
jjg@642 1048 Set<TypeElement> ap = lastRound ? Collections.<TypeElement>emptySet() : annotationsPresent;
jjg@1136 1049 log.printLines("x.print.rounds",
jjg@642 1050 number,
jjg@642 1051 "{" + tlc.toString(", ") + "}",
jjg@642 1052 ap,
jjg@620 1053 lastRound);
jjg@620 1054 }
jjg@620 1055 }
jjg@620 1056
jjg@620 1057 /** Get the context for the next round of processing.
jjg@1340 1058 * Important values are propagated from round to round;
jjg@620 1059 * other values are implicitly reset.
jjg@620 1060 */
jjg@642 1061 private Context nextContext() {
jjg@893 1062 Context next = new Context(context);
jjg@620 1063
jjg@620 1064 Options options = Options.instance(context);
jjg@816 1065 Assert.checkNonNull(options);
jjg@620 1066 next.put(Options.optionsKey, options);
jjg@620 1067
jjg@944 1068 Locale locale = context.get(Locale.class);
jjg@944 1069 if (locale != null)
jjg@944 1070 next.put(Locale.class, locale);
jjg@1159 1071
jjg@944 1072 Assert.checkNonNull(messages);
jjg@944 1073 next.put(JavacMessages.messagesKey, messages);
jjg@620 1074
jjg@620 1075 final boolean shareNames = true;
jjg@620 1076 if (shareNames) {
jjg@620 1077 Names names = Names.instance(context);
jjg@816 1078 Assert.checkNonNull(names);
jjg@620 1079 next.put(Names.namesKey, names);
jjg@620 1080 }
jjg@620 1081
jjg@620 1082 DiagnosticListener<?> dl = context.get(DiagnosticListener.class);
jjg@620 1083 if (dl != null)
jjg@620 1084 next.put(DiagnosticListener.class, dl);
jjg@620 1085
jjg@1210 1086 MultiTaskListener mtl = context.get(MultiTaskListener.taskListenerKey);
jjg@1210 1087 if (mtl != null)
jjg@1210 1088 next.put(MultiTaskListener.taskListenerKey, mtl);
jjg@620 1089
jjg@872 1090 FSInfo fsInfo = context.get(FSInfo.class);
jjg@872 1091 if (fsInfo != null)
jjg@872 1092 next.put(FSInfo.class, fsInfo);
jjg@872 1093
jjg@620 1094 JavaFileManager jfm = context.get(JavaFileManager.class);
jjg@816 1095 Assert.checkNonNull(jfm);
jjg@620 1096 next.put(JavaFileManager.class, jfm);
jjg@620 1097 if (jfm instanceof JavacFileManager) {
jjg@620 1098 ((JavacFileManager)jfm).setContext(next);
jjg@620 1099 }
jjg@620 1100
jjg@620 1101 Names names = Names.instance(context);
jjg@816 1102 Assert.checkNonNull(names);
jjg@620 1103 next.put(Names.namesKey, names);
jjg@620 1104
mcimadamore@1113 1105 Tokens tokens = Tokens.instance(context);
mcimadamore@1113 1106 Assert.checkNonNull(tokens);
mcimadamore@1113 1107 next.put(Tokens.tokensKey, tokens);
jjg@620 1108
jjg@1323 1109 Log nextLog = Log.instance(next);
jjg@1159 1110 // propogate the log's writers directly, instead of going through context
jjg@1323 1111 nextLog.setWriters(log);
jjg@1323 1112 nextLog.setSourceMap(log);
jjg@1159 1113
jjg@620 1114 JavaCompiler oldCompiler = JavaCompiler.instance(context);
jjg@620 1115 JavaCompiler nextCompiler = JavaCompiler.instance(next);
jjg@620 1116 nextCompiler.initRound(oldCompiler);
jjg@620 1117
jjg@706 1118 filer.newRound(next);
jjg@706 1119 messager.newRound(next);
jjg@706 1120 elementUtils.setContext(next);
jjg@706 1121 typeUtils.setContext(next);
jjg@706 1122
jjg@1334 1123 JavacTask task = context.get(JavacTask.class);
jjg@620 1124 if (task != null) {
jjg@1210 1125 next.put(JavacTask.class, task);
jjg@1334 1126 if (task instanceof BasicJavacTask)
jjg@1334 1127 ((BasicJavacTask) task).updateContext(next);
jjg@620 1128 }
jjg@620 1129
jjg@696 1130 JavacTrees trees = context.get(JavacTrees.class);
jjg@696 1131 if (trees != null) {
jjg@696 1132 next.put(JavacTrees.class, trees);
jjg@696 1133 trees.updateContext(next);
jjg@696 1134 }
jjg@696 1135
jjg@620 1136 context.clear();
jjg@620 1137 return next;
jjg@620 1138 }
jjg@620 1139 }
jjg@620 1140
duke@1 1141
duke@1 1142 // TODO: internal catch clauses?; catch and rethrow an annotation
duke@1 1143 // processing error
duke@1 1144 public JavaCompiler doProcessing(Context context,
duke@1 1145 List<JCCompilationUnit> roots,
duke@1 1146 List<ClassSymbol> classSymbols,
jjg@1406 1147 Iterable<? extends PackageSymbol> pckSymbols,
jjg@1406 1148 Log.DeferredDiagnosticHandler deferredDiagnosticHandler) {
duke@1 1149 log = Log.instance(context);
duke@1 1150
duke@1 1151 Set<PackageSymbol> specifiedPackages = new LinkedHashSet<PackageSymbol>();
duke@1 1152 for (PackageSymbol psym : pckSymbols)
duke@1 1153 specifiedPackages.add(psym);
duke@1 1154 this.specifiedPackages = Collections.unmodifiableSet(specifiedPackages);
duke@1 1155
jjg@1406 1156 Round round = new Round(context, roots, classSymbols, deferredDiagnosticHandler);
duke@1 1157
jjg@642 1158 boolean errorStatus;
jjg@642 1159 boolean moreToDo;
jjg@642 1160 do {
jjg@642 1161 // Run processors for round n
jjg@620 1162 round.run(false, false);
duke@1 1163
jjg@642 1164 // Processors for round n have run to completion.
jjg@642 1165 // Check for errors and whether there is more work to do.
jjg@642 1166 errorStatus = round.unrecoverableError();
jjg@642 1167 moreToDo = moreToDo();
jjg@642 1168
jjg@664 1169 round.showDiagnostics(errorStatus || showResolveErrors);
jjg@664 1170
jjg@642 1171 // Set up next round.
jjg@642 1172 // Copy mutable collections returned from filer.
jjg@642 1173 round = round.next(
jjg@642 1174 new LinkedHashSet<JavaFileObject>(filer.getGeneratedSourceFileObjects()),
jjg@642 1175 new LinkedHashMap<String,JavaFileObject>(filer.getGeneratedClasses()));
jjg@642 1176
jjg@642 1177 // Check for errors during setup.
jjg@642 1178 if (round.unrecoverableError())
duke@1 1179 errorStatus = true;
duke@1 1180
jjg@642 1181 } while (moreToDo && !errorStatus);
jjg@620 1182
jjg@620 1183 // run last round
jjg@620 1184 round.run(true, errorStatus);
jjg@664 1185 round.showDiagnostics(true);
jjg@620 1186
duke@1 1187 filer.warnIfUnclosedFiles();
duke@1 1188 warnIfUnmatchedOptions();
duke@1 1189
jjg@642 1190 /*
jjg@642 1191 * If an annotation processor raises an error in a round,
jjg@642 1192 * that round runs to completion and one last round occurs.
jjg@642 1193 * The last round may also occur because no more source or
jjg@642 1194 * class files have been generated. Therefore, if an error
jjg@642 1195 * was raised on either of the last *two* rounds, the compile
jjg@642 1196 * should exit with a nonzero exit code. The current value of
jjg@642 1197 * errorStatus holds whether or not an error was raised on the
jjg@642 1198 * second to last round; errorRaised() gives the error status
jjg@642 1199 * of the last round.
jjg@642 1200 */
jjg@642 1201 if (messager.errorRaised()
jjg@642 1202 || werror && round.warningCount() > 0 && round.errorCount() > 0)
jjg@642 1203 errorStatus = true;
jjg@642 1204
jjg@642 1205 Set<JavaFileObject> newSourceFiles =
jjg@642 1206 new LinkedHashSet<JavaFileObject>(filer.getGeneratedSourceFileObjects());
jjg@642 1207 roots = cleanTrees(round.roots);
jjg@642 1208
jjg@642 1209 JavaCompiler compiler = round.finalCompiler(errorStatus);
jjg@642 1210
jjg@642 1211 if (newSourceFiles.size() > 0)
jjg@642 1212 roots = roots.appendList(compiler.parseFiles(newSourceFiles));
jjg@642 1213
jjg@642 1214 errorStatus = errorStatus || (compiler.errorCount() > 0);
duke@1 1215
duke@1 1216 // Free resources
duke@1 1217 this.close();
duke@1 1218
jjg@1210 1219 if (!taskListener.isEmpty())
duke@1 1220 taskListener.finished(new TaskEvent(TaskEvent.Kind.ANNOTATION_PROCESSING));
duke@1 1221
duke@1 1222 if (errorStatus) {
duke@1 1223 if (compiler.errorCount() == 0)
duke@1 1224 compiler.log.nerrors++;
jjg@642 1225 return compiler;
jjg@642 1226 }
jjg@642 1227
jjg@1340 1228 compiler.enterTreesIfNeeded(roots);
duke@1 1229
duke@1 1230 return compiler;
duke@1 1231 }
duke@1 1232
duke@1 1233 private void warnIfUnmatchedOptions() {
duke@1 1234 if (!unmatchedProcessorOptions.isEmpty()) {
duke@1 1235 log.warning("proc.unmatched.processor.options", unmatchedProcessorOptions.toString());
duke@1 1236 }
duke@1 1237 }
duke@1 1238
duke@1 1239 /**
duke@1 1240 * Free resources related to annotation processing.
duke@1 1241 */
jjg@663 1242 public void close() {
duke@1 1243 filer.close();
darcy@382 1244 if (discoveredProcs != null) // Make calling close idempotent
darcy@382 1245 discoveredProcs.close();
duke@1 1246 discoveredProcs = null;
duke@1 1247 }
duke@1 1248
duke@1 1249 private List<ClassSymbol> getTopLevelClasses(List<? extends JCCompilationUnit> units) {
duke@1 1250 List<ClassSymbol> classes = List.nil();
duke@1 1251 for (JCCompilationUnit unit : units) {
duke@1 1252 for (JCTree node : unit.defs) {
jjg@1127 1253 if (node.hasTag(JCTree.Tag.CLASSDEF)) {
jjg@642 1254 ClassSymbol sym = ((JCClassDecl) node).sym;
jjg@816 1255 Assert.checkNonNull(sym);
jjg@642 1256 classes = classes.prepend(sym);
duke@1 1257 }
duke@1 1258 }
duke@1 1259 }
duke@1 1260 return classes.reverse();
duke@1 1261 }
duke@1 1262
jjg@483 1263 private List<ClassSymbol> getTopLevelClassesFromClasses(List<? extends ClassSymbol> syms) {
jjg@483 1264 List<ClassSymbol> classes = List.nil();
jjg@483 1265 for (ClassSymbol sym : syms) {
jjg@483 1266 if (!isPkgInfo(sym)) {
jjg@483 1267 classes = classes.prepend(sym);
jjg@483 1268 }
jjg@483 1269 }
jjg@483 1270 return classes.reverse();
jjg@483 1271 }
jjg@483 1272
duke@1 1273 private List<PackageSymbol> getPackageInfoFiles(List<? extends JCCompilationUnit> units) {
duke@1 1274 List<PackageSymbol> packages = List.nil();
duke@1 1275 for (JCCompilationUnit unit : units) {
jjg@483 1276 if (isPkgInfo(unit.sourcefile, JavaFileObject.Kind.SOURCE)) {
duke@1 1277 packages = packages.prepend(unit.packge);
duke@1 1278 }
duke@1 1279 }
duke@1 1280 return packages.reverse();
duke@1 1281 }
duke@1 1282
jjg@483 1283 private List<PackageSymbol> getPackageInfoFilesFromClasses(List<? extends ClassSymbol> syms) {
jjg@483 1284 List<PackageSymbol> packages = List.nil();
jjg@483 1285 for (ClassSymbol sym : syms) {
jjg@483 1286 if (isPkgInfo(sym)) {
jjg@483 1287 packages = packages.prepend((PackageSymbol) sym.owner);
jjg@483 1288 }
jjg@483 1289 }
jjg@483 1290 return packages.reverse();
jjg@483 1291 }
jjg@483 1292
jjg@620 1293 // avoid unchecked warning from use of varargs
jjg@620 1294 private static <T> List<T> join(List<T> list1, List<T> list2) {
jjg@620 1295 return list1.appendList(list2);
jjg@620 1296 }
jjg@620 1297
jjg@483 1298 private boolean isPkgInfo(JavaFileObject fo, JavaFileObject.Kind kind) {
jjg@483 1299 return fo.isNameCompatible("package-info", kind);
jjg@483 1300 }
jjg@483 1301
jjg@483 1302 private boolean isPkgInfo(ClassSymbol sym) {
jjg@483 1303 return isPkgInfo(sym.classfile, JavaFileObject.Kind.CLASS) && (sym.packge().package_info == sym);
jjg@483 1304 }
jjg@483 1305
duke@1 1306 /*
duke@1 1307 * Called retroactively to determine if a class loader was required,
duke@1 1308 * after we have failed to create one.
duke@1 1309 */
duke@1 1310 private boolean needClassLoader(String procNames, Iterable<? extends File> workingpath) {
duke@1 1311 if (procNames != null)
duke@1 1312 return true;
duke@1 1313
duke@1 1314 String procPath;
duke@1 1315 URL[] urls = new URL[1];
duke@1 1316 for(File pathElement : workingpath) {
duke@1 1317 try {
duke@1 1318 urls[0] = pathElement.toURI().toURL();
duke@1 1319 if (ServiceProxy.hasService(Processor.class, urls))
duke@1 1320 return true;
duke@1 1321 } catch (MalformedURLException ex) {
duke@1 1322 throw new AssertionError(ex);
duke@1 1323 }
duke@1 1324 catch (ServiceProxy.ServiceConfigurationError e) {
duke@1 1325 log.error("proc.bad.config.file", e.getLocalizedMessage());
duke@1 1326 return true;
duke@1 1327 }
duke@1 1328 }
duke@1 1329
duke@1 1330 return false;
duke@1 1331 }
duke@1 1332
duke@1 1333 private static <T extends JCTree> List<T> cleanTrees(List<T> nodes) {
duke@1 1334 for (T node : nodes)
duke@1 1335 treeCleaner.scan(node);
duke@1 1336 return nodes;
duke@1 1337 }
duke@1 1338
vromero@1442 1339 private static final TreeScanner treeCleaner = new TreeScanner() {
duke@1 1340 public void scan(JCTree node) {
duke@1 1341 super.scan(node);
duke@1 1342 if (node != null)
duke@1 1343 node.type = null;
duke@1 1344 }
duke@1 1345 public void visitTopLevel(JCCompilationUnit node) {
duke@1 1346 node.packge = null;
duke@1 1347 super.visitTopLevel(node);
duke@1 1348 }
duke@1 1349 public void visitClassDef(JCClassDecl node) {
duke@1 1350 node.sym = null;
duke@1 1351 super.visitClassDef(node);
duke@1 1352 }
duke@1 1353 public void visitMethodDef(JCMethodDecl node) {
duke@1 1354 node.sym = null;
duke@1 1355 super.visitMethodDef(node);
duke@1 1356 }
duke@1 1357 public void visitVarDef(JCVariableDecl node) {
duke@1 1358 node.sym = null;
duke@1 1359 super.visitVarDef(node);
duke@1 1360 }
duke@1 1361 public void visitNewClass(JCNewClass node) {
duke@1 1362 node.constructor = null;
duke@1 1363 super.visitNewClass(node);
duke@1 1364 }
duke@1 1365 public void visitAssignop(JCAssignOp node) {
duke@1 1366 node.operator = null;
duke@1 1367 super.visitAssignop(node);
duke@1 1368 }
duke@1 1369 public void visitUnary(JCUnary node) {
duke@1 1370 node.operator = null;
duke@1 1371 super.visitUnary(node);
duke@1 1372 }
duke@1 1373 public void visitBinary(JCBinary node) {
duke@1 1374 node.operator = null;
duke@1 1375 super.visitBinary(node);
duke@1 1376 }
duke@1 1377 public void visitSelect(JCFieldAccess node) {
duke@1 1378 node.sym = null;
duke@1 1379 super.visitSelect(node);
duke@1 1380 }
duke@1 1381 public void visitIdent(JCIdent node) {
duke@1 1382 node.sym = null;
duke@1 1383 super.visitIdent(node);
duke@1 1384 }
duke@1 1385 };
duke@1 1386
duke@1 1387
duke@1 1388 private boolean moreToDo() {
duke@1 1389 return filer.newFiles();
duke@1 1390 }
duke@1 1391
duke@1 1392 /**
duke@1 1393 * {@inheritdoc}
duke@1 1394 *
duke@1 1395 * Command line options suitable for presenting to annotation
jjg@1326 1396 * processors.
jjg@1326 1397 * {@literal "-Afoo=bar"} should be {@literal "-Afoo" => "bar"}.
duke@1 1398 */
duke@1 1399 public Map<String,String> getOptions() {
duke@1 1400 return processorOptions;
duke@1 1401 }
duke@1 1402
duke@1 1403 public Messager getMessager() {
duke@1 1404 return messager;
duke@1 1405 }
duke@1 1406
duke@1 1407 public Filer getFiler() {
duke@1 1408 return filer;
duke@1 1409 }
duke@1 1410
duke@1 1411 public JavacElements getElementUtils() {
duke@1 1412 return elementUtils;
duke@1 1413 }
duke@1 1414
duke@1 1415 public JavacTypes getTypeUtils() {
duke@1 1416 return typeUtils;
duke@1 1417 }
duke@1 1418
duke@1 1419 public SourceVersion getSourceVersion() {
duke@1 1420 return Source.toSourceVersion(source);
duke@1 1421 }
duke@1 1422
duke@1 1423 public Locale getLocale() {
mcimadamore@136 1424 return messages.getCurrentLocale();
duke@1 1425 }
duke@1 1426
duke@1 1427 public Set<Symbol.PackageSymbol> getSpecifiedPackages() {
duke@1 1428 return specifiedPackages;
duke@1 1429 }
duke@1 1430
darcy@497 1431 private static final Pattern allMatches = Pattern.compile(".*");
darcy@497 1432 public static final Pattern noMatches = Pattern.compile("(\\P{all})+");
darcy@497 1433
duke@1 1434 /**
darcy@497 1435 * Convert import-style string for supported annotations into a
darcy@497 1436 * regex matching that string. If the string is a valid
darcy@497 1437 * import-style string, return a regex that won't match anything.
duke@1 1438 */
darcy@497 1439 private static Pattern importStringToPattern(String s, Processor p, Log log) {
darcy@497 1440 if (isValidImportString(s)) {
darcy@497 1441 return validImportStringToPattern(s);
darcy@497 1442 } else {
darcy@497 1443 log.warning("proc.malformed.supported.string", s, p.getClass().getName());
darcy@497 1444 return noMatches; // won't match any valid identifier
duke@1 1445 }
duke@1 1446 }
duke@1 1447
duke@1 1448 /**
darcy@497 1449 * Return true if the argument string is a valid import-style
darcy@497 1450 * string specifying claimed annotations; return false otherwise.
duke@1 1451 */
darcy@497 1452 public static boolean isValidImportString(String s) {
darcy@497 1453 if (s.equals("*"))
darcy@497 1454 return true;
darcy@497 1455
darcy@497 1456 boolean valid = true;
darcy@497 1457 String t = s;
darcy@497 1458 int index = t.indexOf('*');
darcy@497 1459
darcy@497 1460 if (index != -1) {
darcy@497 1461 // '*' must be last character...
darcy@497 1462 if (index == t.length() -1) {
darcy@497 1463 // ... any and preceding character must be '.'
darcy@497 1464 if ( index-1 >= 0 ) {
darcy@497 1465 valid = t.charAt(index-1) == '.';
darcy@497 1466 // Strip off ".*$" for identifier checks
darcy@497 1467 t = t.substring(0, t.length()-2);
darcy@497 1468 }
darcy@497 1469 } else
darcy@497 1470 return false;
duke@1 1471 }
darcy@497 1472
darcy@497 1473 // Verify string is off the form (javaId \.)+ or javaId
darcy@497 1474 if (valid) {
darcy@497 1475 String[] javaIds = t.split("\\.", t.length()+2);
darcy@497 1476 for(String javaId: javaIds)
darcy@497 1477 valid &= SourceVersion.isIdentifier(javaId);
duke@1 1478 }
darcy@497 1479 return valid;
duke@1 1480 }
duke@1 1481
darcy@497 1482 public static Pattern validImportStringToPattern(String s) {
duke@1 1483 if (s.equals("*")) {
duke@1 1484 return allMatches;
duke@1 1485 } else {
darcy@497 1486 String s_prime = s.replace(".", "\\.");
duke@1 1487
duke@1 1488 if (s_prime.endsWith("*")) {
duke@1 1489 s_prime = s_prime.substring(0, s_prime.length() - 1) + ".+";
duke@1 1490 }
duke@1 1491
duke@1 1492 return Pattern.compile(s_prime);
duke@1 1493 }
duke@1 1494 }
duke@1 1495
duke@1 1496 /**
jjg@1416 1497 * For internal use only. This method may be removed without warning.
duke@1 1498 */
duke@1 1499 public Context getContext() {
duke@1 1500 return context;
duke@1 1501 }
duke@1 1502
jjg@1416 1503 /**
jjg@1416 1504 * For internal use only. This method may be removed without warning.
jjg@1416 1505 */
jjg@1416 1506 public ClassLoader getProcessorClassLoader() {
jjg@1416 1507 return processorClassLoader;
jjg@1416 1508 }
jjg@1416 1509
duke@1 1510 public String toString() {
duke@1 1511 return "javac ProcessingEnvironment";
duke@1 1512 }
duke@1 1513
duke@1 1514 public static boolean isValidOptionName(String optionName) {
duke@1 1515 for(String s : optionName.split("\\.", -1)) {
duke@1 1516 if (!SourceVersion.isIdentifier(s))
duke@1 1517 return false;
duke@1 1518 }
duke@1 1519 return true;
duke@1 1520 }
duke@1 1521 }

mercurial