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