1 /*
   2  * Copyright (c) 2003, 2022, 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 
  26 package com.sun.tools.javac.comp;
  27 
  28 import com.sun.tools.javac.code.*;
  29 import com.sun.tools.javac.code.Attribute.Compound;
  30 import com.sun.tools.javac.code.Attribute.TypeCompound;
  31 import com.sun.tools.javac.code.Kinds.KindSelector;
  32 import com.sun.tools.javac.code.Scope.WriteableScope;
  33 import com.sun.tools.javac.code.Source.Feature;
  34 import com.sun.tools.javac.code.Symbol.*;
  35 import com.sun.tools.javac.code.TypeMetadata.Annotations;
  36 import com.sun.tools.javac.comp.Check.CheckContext;
  37 import com.sun.tools.javac.resources.CompilerProperties.Errors;
  38 import com.sun.tools.javac.resources.CompilerProperties.Fragments;
  39 import com.sun.tools.javac.tree.JCTree;
  40 import com.sun.tools.javac.tree.JCTree.*;
  41 import com.sun.tools.javac.tree.TreeInfo;
  42 import com.sun.tools.javac.tree.TreeMaker;
  43 import com.sun.tools.javac.tree.TreeScanner;
  44 import com.sun.tools.javac.util.*;
  45 import com.sun.tools.javac.util.JCDiagnostic.DiagnosticPosition;
  46 import com.sun.tools.javac.util.List;
  47 
  48 import javax.tools.JavaFileObject;
  49 
  50 import java.util.*;
  51 
  52 import static com.sun.tools.javac.code.Flags.SYNTHETIC;
  53 import static com.sun.tools.javac.code.Kinds.Kind.MDL;
  54 import static com.sun.tools.javac.code.Kinds.Kind.MTH;
  55 import static com.sun.tools.javac.code.Kinds.Kind.PCK;
  56 import static com.sun.tools.javac.code.Kinds.Kind.TYP;
  57 import static com.sun.tools.javac.code.Kinds.Kind.VAR;
  58 import static com.sun.tools.javac.code.Scope.LookupKind.NON_RECURSIVE;
  59 import static com.sun.tools.javac.code.TypeTag.ARRAY;
  60 import static com.sun.tools.javac.code.TypeTag.CLASS;
  61 import static com.sun.tools.javac.tree.JCTree.Tag.ANNOTATION;
  62 import static com.sun.tools.javac.tree.JCTree.Tag.ASSIGN;
  63 import static com.sun.tools.javac.tree.JCTree.Tag.IDENT;
  64 import static com.sun.tools.javac.tree.JCTree.Tag.NEWARRAY;
  65 
  66 import com.sun.tools.javac.util.JCDiagnostic.DiagnosticFlag;
  67 
  68 
  69 /** Enter annotations onto symbols and types (and trees).
  70  *
  71  *  This is also a pseudo stage in the compiler taking care of scheduling when annotations are
  72  *  entered.
  73  *
  74  *  <p><b>This is NOT part of any supported API.
  75  *  If you write code that depends on this, you do so at your own risk.
  76  *  This code and its internal interfaces are subject to change or
  77  *  deletion without notice.</b>
  78  */
  79 public class Annotate {
  80     protected static final Context.Key<Annotate> annotateKey = new Context.Key<>();
  81 
  82     public static Annotate instance(Context context) {
  83         Annotate instance = context.get(annotateKey);
  84         if (instance == null)
  85             instance = new Annotate(context);
  86         return instance;
  87     }
  88 
  89     private final Attr attr;
  90     private final Check chk;
  91     private final ConstFold cfolder;
  92     private final DeferredLintHandler deferredLintHandler;
  93     private final Enter enter;
  94     private final Lint lint;
  95     private final Log log;
  96     private final Names names;
  97     private final Resolve resolve;
  98     private final TreeMaker make;
  99     private final Symtab syms;
 100     private final TypeEnvs typeEnvs;
 101     private final Types types;
 102 
 103     private final Attribute theUnfinishedDefaultValue;
 104     private final String sourceName;
 105 
 106     @SuppressWarnings("this-escape")
 107     protected Annotate(Context context) {
 108         context.put(annotateKey, this);
 109 
 110         attr = Attr.instance(context);
 111         chk = Check.instance(context);
 112         cfolder = ConstFold.instance(context);
 113         deferredLintHandler = DeferredLintHandler.instance(context);
 114         enter = Enter.instance(context);
 115         log = Log.instance(context);
 116         lint = Lint.instance(context);
 117         make = TreeMaker.instance(context);
 118         names = Names.instance(context);
 119         resolve = Resolve.instance(context);
 120         syms = Symtab.instance(context);
 121         typeEnvs = TypeEnvs.instance(context);
 122         types = Types.instance(context);
 123 
 124         theUnfinishedDefaultValue =  new Attribute.Error(syms.errType);
 125 
 126         Source source = Source.instance(context);
 127         sourceName = source.name;
 128 
 129         blockCount = 1;
 130     }
 131 
 132     /** Semaphore to delay annotation processing */
 133     private int blockCount = 0;
 134 
 135     /** Called when annotations processing needs to be postponed. */
 136     public void blockAnnotations() {
 137         blockCount++;
 138     }
 139 
 140     /** Called when annotation processing can be resumed. */
 141     public void unblockAnnotations() {
 142         blockCount--;
 143         if (blockCount == 0)
 144             flush();
 145     }
 146 
 147     /** Variant which allows for a delayed flush of annotations.
 148      * Needed by ClassReader */
 149     public void unblockAnnotationsNoFlush() {
 150         blockCount--;
 151     }
 152 
 153     /** are we blocking annotation processing? */
 154     public boolean annotationsBlocked() {return blockCount > 0; }
 155 
 156     public void enterDone() {
 157         unblockAnnotations();
 158     }
 159 
 160     public List<TypeCompound> fromAnnotations(List<JCAnnotation> annotations) {
 161         if (annotations.isEmpty()) {
 162             return List.nil();
 163         }
 164 
 165         ListBuffer<TypeCompound> buf = new ListBuffer<>();
 166         for (JCAnnotation anno : annotations) {
 167             Assert.checkNonNull(anno.attribute);
 168             buf.append((TypeCompound) anno.attribute);
 169         }
 170         return buf.toList();
 171     }
 172 
 173     /** Annotate (used for everything else) */
 174     public void normal(Runnable r) {
 175         q.append(r);
 176     }
 177 
 178     /** Validate, triggers after 'normal' */
 179     public void validate(Runnable a) {
 180         validateQ.append(a);
 181     }
 182 
 183     /** Flush all annotation queues */
 184     public void flush() {
 185         if (annotationsBlocked()) return;
 186         if (isFlushing()) return;
 187 
 188         startFlushing();
 189         try {
 190             while (q.nonEmpty()) {
 191                 q.next().run();
 192             }
 193             while (typesQ.nonEmpty()) {
 194                 typesQ.next().run();
 195             }
 196             while (afterTypesQ.nonEmpty()) {
 197                 afterTypesQ.next().run();
 198             }
 199             while (validateQ.nonEmpty()) {
 200                 validateQ.next().run();
 201             }
 202         } finally {
 203             doneFlushing();
 204         }
 205     }
 206 
 207     private ListBuffer<Runnable> q = new ListBuffer<>();
 208     private ListBuffer<Runnable> validateQ = new ListBuffer<>();
 209 
 210     private int flushCount = 0;
 211     private boolean isFlushing() { return flushCount > 0; }
 212     private void startFlushing() { flushCount++; }
 213     private void doneFlushing() { flushCount--; }
 214 
 215     ListBuffer<Runnable> typesQ = new ListBuffer<>();
 216     ListBuffer<Runnable> afterTypesQ = new ListBuffer<>();
 217 
 218 
 219     public void typeAnnotation(Runnable a) {
 220         typesQ.append(a);
 221     }
 222 
 223     public void afterTypes(Runnable a) {
 224         afterTypesQ.append(a);
 225     }
 226 
 227     /**
 228      * Queue annotations for later attribution and entering. This is probably the method you are looking for.
 229      *
 230      * @param annotations the list of JCAnnotations to attribute and enter
 231      * @param localEnv    the enclosing env
 232      * @param s           the Symbol on which to enter the annotations
 233      * @param deferPos    report errors here
 234      */
 235     public void annotateLater(List<JCAnnotation> annotations, Env<AttrContext> localEnv,
 236             Symbol s, DiagnosticPosition deferPos)
 237     {
 238         if (annotations.isEmpty()) {
 239             return;
 240         }
 241 
 242         s.resetAnnotations(); // mark Annotations as incomplete for now
 243 
 244         normal(() -> {
 245             // Packages are unusual, in that they are the only type of declaration that can legally appear
 246             // more than once in a compilation, and in all cases refer to the same underlying symbol.
 247             // This means they are the only kind of declaration that syntactically may have multiple sets
 248             // of annotations, each on a different package declaration, even though that is ultimately
 249             // forbidden by JLS 8 section 7.4.
 250             // The corollary here is that all of the annotations on a package symbol may have already
 251             // been handled, meaning that the set of annotations pending completion is now empty.
 252             Assert.check(s.kind == PCK || s.annotationsPendingCompletion());
 253             JavaFileObject prev = log.useSource(localEnv.toplevel.sourcefile);
 254             DiagnosticPosition prevLintPos =
 255                     deferPos != null
 256                             ? deferredLintHandler.setPos(deferPos)
 257                             : deferredLintHandler.immediate();
 258             Lint prevLint = deferPos != null ? null : chk.setLint(lint);
 259             try {
 260                 if (s.hasAnnotations() && annotations.nonEmpty())
 261                     log.error(annotations.head.pos, Errors.AlreadyAnnotated(Kinds.kindName(s), s));
 262 
 263                 Assert.checkNonNull(s, "Symbol argument to actualEnterAnnotations is null");
 264 
 265                 // false is passed as fifth parameter since annotateLater is
 266                 // never called for a type parameter
 267                 annotateNow(s, annotations, localEnv, false, false);
 268             } finally {
 269                 if (prevLint != null)
 270                     chk.setLint(prevLint);
 271                 deferredLintHandler.setPos(prevLintPos);
 272                 log.useSource(prev);
 273             }
 274         });
 275 
 276         validate(() -> { //validate annotations
 277             JavaFileObject prev = log.useSource(localEnv.toplevel.sourcefile);
 278             try {
 279                 chk.validateAnnotations(annotations, TreeInfo.declarationFor(s, localEnv.tree), s);
 280             } finally {
 281                 log.useSource(prev);
 282             }
 283         });
 284     }
 285 
 286 
 287     /** Queue processing of an attribute default value. */
 288     public void annotateDefaultValueLater(JCExpression defaultValue, Env<AttrContext> localEnv,
 289             MethodSymbol m, DiagnosticPosition deferPos)
 290     {
 291         normal(() -> {
 292             JavaFileObject prev = log.useSource(localEnv.toplevel.sourcefile);
 293             DiagnosticPosition prevLintPos = deferredLintHandler.setPos(deferPos);
 294             try {
 295                 enterDefaultValue(defaultValue, localEnv, m);
 296             } finally {
 297                 deferredLintHandler.setPos(prevLintPos);
 298                 log.useSource(prev);
 299             }
 300         });
 301 
 302         validate(() -> { //validate annotations
 303             JavaFileObject prev = log.useSource(localEnv.toplevel.sourcefile);
 304             try {
 305                 // if default value is an annotation, check it is a well-formed
 306                 // annotation value (e.g. no duplicate values, no missing values, etc.)
 307                 chk.validateAnnotationTree(defaultValue);
 308             } finally {
 309                 log.useSource(prev);
 310             }
 311         });
 312     }
 313 
 314     /** Enter a default value for an annotation element. */
 315     private void enterDefaultValue(JCExpression defaultValue,
 316             Env<AttrContext> localEnv, MethodSymbol m) {
 317         m.defaultValue = attributeAnnotationValue(m.type.getReturnType(), defaultValue, localEnv);
 318     }
 319 
 320     /**
 321      * Gather up annotations into a map from type symbols to lists of Compound attributes,
 322      * then continue on with repeating annotations processing.
 323      */
 324     private <T extends Attribute.Compound> void annotateNow(Symbol toAnnotate,
 325             List<JCAnnotation> withAnnotations, Env<AttrContext> env, boolean typeAnnotations,
 326             boolean isTypeParam)
 327     {
 328         Map<TypeSymbol, ListBuffer<T>> annotated = new LinkedHashMap<>();
 329         Map<T, DiagnosticPosition> pos = new HashMap<>();
 330 
 331         for (List<JCAnnotation> al = withAnnotations; !al.isEmpty(); al = al.tail) {
 332             JCAnnotation a = al.head;
 333 
 334             T c;
 335             if (typeAnnotations) {
 336                 @SuppressWarnings("unchecked")
 337                 T tmp = (T)attributeTypeAnnotation(a, syms.annotationType, env);
 338                 c = tmp;
 339             } else {
 340                 @SuppressWarnings("unchecked")
 341                 T tmp = (T)attributeAnnotation(a, syms.annotationType, env);
 342                 c = tmp;
 343             }
 344 
 345             Assert.checkNonNull(c, "Failed to create annotation");
 346 
 347             if (a.type.isErroneous() || a.type.tsym.isAnnotationType()) {
 348                 if (annotated.containsKey(a.type.tsym)) {
 349                     ListBuffer<T> l = annotated.get(a.type.tsym);
 350                     l = l.append(c);
 351                     annotated.put(a.type.tsym, l);
 352                     pos.put(c, a.pos());
 353                 } else {
 354                     annotated.put(a.type.tsym, ListBuffer.of(c));
 355                     pos.put(c, a.pos());
 356                 }
 357             }
 358 
 359             // Note: @Deprecated has no effect on local variables and parameters
 360             if (!c.type.isErroneous()
 361                     && (toAnnotate.kind == MDL || toAnnotate.owner.kind != MTH)
 362                     && types.isSameType(c.type, syms.deprecatedType)) {
 363                 toAnnotate.flags_field |= (Flags.DEPRECATED | Flags.DEPRECATED_ANNOTATION);
 364                 if (isAttributeTrue(c.member(names.forRemoval))) {
 365                     toAnnotate.flags_field |= Flags.DEPRECATED_REMOVAL;
 366                 }
 367             }
 368 
 369             if (!c.type.isErroneous()
 370                     && types.isSameType(c.type, syms.previewFeatureType)) {
 371                 toAnnotate.flags_field |= Flags.PREVIEW_API;
 372                 if (isAttributeTrue(c.member(names.reflective))) {
 373                     toAnnotate.flags_field |= Flags.PREVIEW_REFLECTIVE;
 374                 }
 375             }
 376 
 377             if (!c.type.isErroneous()
 378                     && toAnnotate.kind == TYP
 379                     && types.isSameType(c.type, syms.valueBasedType)) {
 380                 toAnnotate.flags_field |= Flags.VALUE_BASED;
 381             }
 382 
 383             if (!c.type.isErroneous()
 384                     && types.isSameType(c.type, syms.restrictedType)) {
 385                 toAnnotate.flags_field |= Flags.RESTRICTED;
 386             }
 387         }
 388 
 389         List<T> buf = List.nil();
 390         for (ListBuffer<T> lb : annotated.values()) {
 391             if (lb.size() == 1) {
 392                 buf = buf.prepend(lb.first());
 393             } else {
 394                 AnnotationContext<T> ctx = new AnnotationContext<>(env, annotated, pos, typeAnnotations);
 395                 T res = makeContainerAnnotation(lb.toList(), ctx, toAnnotate, isTypeParam);
 396                 if (res != null)
 397                     buf = buf.prepend(res);
 398             }
 399         }
 400 
 401         if (typeAnnotations) {
 402             @SuppressWarnings("unchecked")
 403             List<TypeCompound> attrs = (List<TypeCompound>)buf.reverse();
 404             toAnnotate.appendUniqueTypeAttributes(attrs);
 405         } else {
 406             @SuppressWarnings("unchecked")
 407             List<Attribute.Compound> attrs =  (List<Attribute.Compound>)buf.reverse();
 408             toAnnotate.resetAnnotations();
 409             toAnnotate.setDeclarationAttributes(attrs);
 410         }
 411     }
 412     //where:
 413         private boolean isAttributeTrue(Attribute attr) {
 414             return (attr instanceof Attribute.Constant constant)
 415                     && constant.type == syms.booleanType
 416                     && ((Integer) constant.value) != 0;
 417         }
 418 
 419     /**
 420      * Attribute and store a semantic representation of the annotation tree {@code tree} into the
 421      * tree.attribute field.
 422      *
 423      * @param tree the tree representing an annotation
 424      * @param expectedAnnotationType the expected (super)type of the annotation
 425      * @param env the current env in where the annotation instance is found
 426      */
 427     public Attribute.Compound attributeAnnotation(JCAnnotation tree, Type expectedAnnotationType,
 428                                                   Env<AttrContext> env)
 429     {
 430         // The attribute might have been entered if it is Target or Repeatable
 431         // Because TreeCopier does not copy type, redo this if type is null
 432         if (tree.attribute != null && tree.type != null)
 433             return tree.attribute;
 434 
 435         List<Pair<MethodSymbol, Attribute>> elems = attributeAnnotationValues(tree, expectedAnnotationType, env);
 436         Attribute.Compound ac = new Attribute.Compound(tree.type, elems);
 437 
 438         return tree.attribute = ac;
 439     }
 440 
 441     /** Attribute and store a semantic representation of the type annotation tree {@code tree} into
 442      * the tree.attribute field.
 443      *
 444      * @param a the tree representing an annotation
 445      * @param expectedAnnotationType the expected (super)type of the annotation
 446      * @param env the current env in where the annotation instance is found
 447      */
 448     public Attribute.TypeCompound attributeTypeAnnotation(JCAnnotation a, Type expectedAnnotationType,
 449                                                           Env<AttrContext> env)
 450     {
 451         // The attribute might have been entered if it is Target or Repeatable
 452         // Because TreeCopier does not copy type, redo this if type is null
 453         if (a.attribute == null || a.type == null || !(a.attribute instanceof Attribute.TypeCompound typeCompound)) {
 454             // Create a new TypeCompound
 455             List<Pair<MethodSymbol,Attribute>> elems =
 456                     attributeAnnotationValues(a, expectedAnnotationType, env);
 457 
 458             Attribute.TypeCompound tc =
 459                     new Attribute.TypeCompound(a.type, elems, TypeAnnotationPosition.unknown);
 460             a.attribute = tc;
 461             return tc;
 462         } else {
 463             // Use an existing TypeCompound
 464             return typeCompound;
 465         }
 466     }
 467 
 468     /**
 469      *  Attribute annotation elements creating a list of pairs of the Symbol representing that
 470      *  element and the value of that element as an Attribute. */
 471     private List<Pair<MethodSymbol, Attribute>> attributeAnnotationValues(JCAnnotation a,
 472             Type expected, Env<AttrContext> env)
 473     {
 474         // The annotation might have had its type attributed (but not
 475         // checked) by attr.attribAnnotationTypes during MemberEnter,
 476         // in which case we do not need to do it again.
 477         Type at = (a.annotationType.type != null ?
 478                 a.annotationType.type : attr.attribType(a.annotationType, env));
 479         a.type = chk.checkType(a.annotationType.pos(), at, expected);
 480 
 481         boolean isError = a.type.isErroneous();
 482         if (!a.type.tsym.isAnnotationType() && !isError) {
 483             log.error(a.annotationType.pos(), Errors.NotAnnotationType(a.type));
 484             isError = true;
 485         }
 486 
 487         // List of name=value pairs (or implicit "value=" if size 1)
 488         List<JCExpression> args = a.args;
 489 
 490         boolean elidedValue = false;
 491         // special case: elided "value=" assumed
 492         if (args.length() == 1 && !args.head.hasTag(ASSIGN)) {
 493             args.head = make.at(args.head.pos).
 494                     Assign(make.Ident(names.value), args.head);
 495             elidedValue = true;
 496         }
 497 
 498         ListBuffer<Pair<MethodSymbol,Attribute>> buf = new ListBuffer<>();
 499         for (List<JCExpression> tl = args; tl.nonEmpty(); tl = tl.tail) {
 500             Pair<MethodSymbol, Attribute> p = attributeAnnotationNameValuePair(tl.head, a.type, isError, env, elidedValue);
 501             if (p != null && !p.fst.type.isErroneous())
 502                 buf.append(p);
 503         }
 504         return buf.toList();
 505     }
 506 
 507     // where
 508     private Pair<MethodSymbol, Attribute> attributeAnnotationNameValuePair(JCExpression nameValuePair,
 509             Type thisAnnotationType, boolean badAnnotation, Env<AttrContext> env, boolean elidedValue)
 510     {
 511         if (!nameValuePair.hasTag(ASSIGN)) {
 512             log.error(nameValuePair.pos(), Errors.AnnotationValueMustBeNameValue);
 513             attributeAnnotationValue(nameValuePair.type = syms.errType, nameValuePair, env);
 514             return null;
 515         }
 516         JCAssign assign = (JCAssign)nameValuePair;
 517         if (!assign.lhs.hasTag(IDENT)) {
 518             log.error(nameValuePair.pos(), Errors.AnnotationValueMustBeNameValue);
 519             attributeAnnotationValue(nameValuePair.type = syms.errType, nameValuePair, env);
 520             return null;
 521         }
 522 
 523         // Resolve element to MethodSym
 524         JCIdent left = (JCIdent)assign.lhs;
 525         Symbol method = resolve.resolveQualifiedMethod(elidedValue ? assign.rhs.pos() : left.pos(),
 526                 env, thisAnnotationType,
 527                 left.name, List.nil(), null);
 528         left.sym = method;
 529         left.type = method.type;
 530         chk.checkDeprecated(left, env.info.scope.owner, method);
 531         if (method.owner != thisAnnotationType.tsym && !badAnnotation)
 532             log.error(left.pos(), Errors.NoAnnotationMember(left.name, thisAnnotationType));
 533         Type resultType = method.type.getReturnType();
 534 
 535         // Compute value part
 536         Attribute value = attributeAnnotationValue(resultType, assign.rhs, env);
 537         nameValuePair.type = resultType;
 538 
 539         return method.type.isErroneous() ? null : new Pair<>((MethodSymbol)method, value);
 540 
 541     }
 542 
 543     /** Attribute an annotation element value */
 544     private Attribute attributeAnnotationValue(Type expectedElementType, JCExpression tree,
 545             Env<AttrContext> env)
 546     {
 547         //first, try completing the symbol for the annotation value - if a completion
 548         //error is thrown, we should recover gracefully, and display an
 549         //ordinary resolution diagnostic.
 550         try {
 551             expectedElementType.tsym.complete();
 552         } catch(CompletionFailure e) {
 553             log.error(tree.pos(), Errors.CantResolve(Kinds.kindName(e.sym), e.sym.getQualifiedName(), null, null));
 554             expectedElementType = syms.errType;
 555         }
 556 
 557         if (expectedElementType.hasTag(ARRAY)) {
 558             return getAnnotationArrayValue(expectedElementType, tree, env);
 559         }
 560 
 561         //error recovery
 562         if (tree.hasTag(NEWARRAY)) {
 563             if (!expectedElementType.isErroneous())
 564                 log.error(tree.pos(), Errors.AnnotationValueNotAllowableType);
 565             JCNewArray na = (JCNewArray)tree;
 566             if (na.elemtype != null) {
 567                 log.error(na.elemtype.pos(), Errors.NewNotAllowedInAnnotation);
 568             }
 569             for (List<JCExpression> l = na.elems; l.nonEmpty(); l=l.tail) {
 570                 attributeAnnotationValue(syms.errType,
 571                         l.head,
 572                         env);
 573             }
 574             return new Attribute.Error(syms.errType);
 575         }
 576 
 577         if (expectedElementType.tsym.isAnnotationType()) {
 578             if (tree.hasTag(ANNOTATION)) {
 579                 return attributeAnnotation((JCAnnotation)tree, expectedElementType, env);
 580             } else {
 581                 log.error(tree.pos(), Errors.AnnotationValueMustBeAnnotation);
 582                 expectedElementType = syms.errType;
 583             }
 584         }
 585 
 586         //error recovery
 587         if (tree.hasTag(ANNOTATION)) {
 588             if (!expectedElementType.isErroneous())
 589                 log.error(tree.pos(), Errors.AnnotationNotValidForType(expectedElementType));
 590             attributeAnnotation((JCAnnotation)tree, syms.errType, env);
 591             return new Attribute.Error(((JCAnnotation)tree).annotationType.type);
 592         }
 593 
 594         MemberEnter.InitTreeVisitor initTreeVisitor = new MemberEnter.InitTreeVisitor() {
 595             // the methods below are added to allow class literals on top of constant expressions
 596             @Override
 597             public void visitTypeIdent(JCPrimitiveTypeTree that) {}
 598 
 599             @Override
 600             public void visitTypeArray(JCArrayTypeTree that) {}
 601         };
 602         tree.accept(initTreeVisitor);
 603         if (!initTreeVisitor.result) {
 604             log.error(tree.pos(), Errors.ExpressionNotAllowableAsAnnotationValue);
 605             return new Attribute.Error(syms.errType);
 606         }
 607 
 608         if (expectedElementType.isPrimitive() ||
 609                 (types.isSameType(expectedElementType, syms.stringType) && !expectedElementType.hasTag(TypeTag.ERROR))) {
 610             return getAnnotationPrimitiveValue(expectedElementType, tree, env);
 611         }
 612 
 613         if (expectedElementType.tsym == syms.classType.tsym) {
 614             return getAnnotationClassValue(expectedElementType, tree, env);
 615         }
 616 
 617         if (expectedElementType.hasTag(CLASS) &&
 618                 (expectedElementType.tsym.flags() & Flags.ENUM) != 0) {
 619             return getAnnotationEnumValue(expectedElementType, tree, env);
 620         }
 621 
 622         //error recovery:
 623         if (!expectedElementType.isErroneous())
 624             log.error(tree.pos(), Errors.AnnotationValueNotAllowableType);
 625         return new Attribute.Error(attr.attribExpr(tree, env, expectedElementType));
 626     }
 627 
 628     private Attribute getAnnotationEnumValue(Type expectedElementType, JCExpression tree, Env<AttrContext> env) {
 629         Type result = attr.attribTree(tree, env, annotationValueInfo(expectedElementType));
 630         Symbol sym = TreeInfo.symbol(tree);
 631         if (sym == null ||
 632                 TreeInfo.nonstaticSelect(tree) ||
 633                 sym.kind != VAR ||
 634                 (sym.flags() & Flags.ENUM) == 0) {
 635             log.error(tree.pos(), Errors.EnumAnnotationMustBeEnumConstant);
 636             return new Attribute.Error(result.getOriginalType());
 637         }
 638         VarSymbol enumerator = (VarSymbol) sym;
 639         return new Attribute.Enum(expectedElementType, enumerator);
 640     }
 641 
 642     private Attribute getAnnotationClassValue(Type expectedElementType, JCExpression tree, Env<AttrContext> env) {
 643         Type result = attr.attribTree(tree, env, annotationValueInfo(expectedElementType));
 644         if (result.isErroneous()) {
 645             // Does it look like an unresolved class literal?
 646             if (TreeInfo.name(tree) == names._class &&
 647                     ((JCFieldAccess) tree).selected.type.isErroneous()) {
 648                 Name n = (((JCFieldAccess) tree).selected).type.tsym.flatName();
 649                 return new Attribute.UnresolvedClass(expectedElementType,
 650                         types.createErrorType(n,
 651                                 syms.unknownSymbol, syms.classType));
 652             } else {
 653                 return new Attribute.Error(result.getOriginalType());
 654             }
 655         }
 656 
 657         // Class literals look like field accesses of a field named class
 658         // at the tree level
 659         if (TreeInfo.name(tree) != names._class) {
 660             log.error(tree.pos(), Errors.AnnotationValueMustBeClassLiteral);
 661             return new Attribute.Error(syms.errType);
 662         }
 663 
 664         return new Attribute.Class(types,
 665                 (((JCFieldAccess) tree).selected).type);
 666     }
 667 
 668     private Attribute getAnnotationPrimitiveValue(Type expectedElementType, JCExpression tree, Env<AttrContext> env) {
 669         Type result = attr.attribTree(tree, env, annotationValueInfo(expectedElementType));
 670         if (result.isErroneous())
 671             return new Attribute.Error(result.getOriginalType());
 672         if (result.constValue() == null) {
 673             log.error(tree.pos(), Errors.AttributeValueMustBeConstant);
 674             return new Attribute.Error(expectedElementType);
 675         }
 676 
 677         // Scan the annotation element value and then attribute nested annotations if present
 678         if (tree.type != null && tree.type.tsym != null) {
 679             queueScanTreeAndTypeAnnotate(tree, env, tree.type.tsym, tree.pos());
 680         }
 681 
 682         result = cfolder.coerce(result, expectedElementType);
 683         return new Attribute.Constant(expectedElementType, result.constValue());
 684     }
 685 
 686     private Attr.ResultInfo annotationValueInfo(Type pt) {
 687         return attr.unknownExprInfo.dup(pt, new AnnotationValueContext(attr.unknownExprInfo.checkContext));
 688     }
 689 
 690     class AnnotationValueContext extends Check.NestedCheckContext {
 691         AnnotationValueContext(CheckContext enclosingContext) {
 692             super(enclosingContext);
 693         }
 694 
 695         @Override
 696         public boolean compatible(Type found, Type req, Warner warn) {
 697             //handle non-final implicitly-typed vars (will be rejected later on)
 698             return found.hasTag(TypeTag.NONE) || super.compatible(found, req, warn);
 699         }
 700     }
 701 
 702     private Attribute getAnnotationArrayValue(Type expectedElementType, JCExpression tree, Env<AttrContext> env) {
 703         // Special case, implicit array
 704         if (!tree.hasTag(NEWARRAY)) {
 705             tree = make.at(tree.pos).
 706                     NewArray(null, List.nil(), List.of(tree));
 707         }
 708 
 709         JCNewArray na = (JCNewArray)tree;
 710         List<JCExpression> elems = na.elems;
 711         if (na.elemtype != null) {
 712             log.error(na.elemtype.pos(), Errors.NewNotAllowedInAnnotation);
 713             if (elems == null) {
 714                 elems = List.nil();
 715             }
 716         }
 717         ListBuffer<Attribute> buf = new ListBuffer<>();
 718         for (List<JCExpression> l = elems; l.nonEmpty(); l = l.tail) {
 719             buf.append(attributeAnnotationValue(types.elemtype(expectedElementType),
 720                     l.head,
 721                     env));
 722         }
 723         na.type = expectedElementType;
 724         return new Attribute.
 725                 Array(expectedElementType, buf.toArray(new Attribute[buf.length()]));
 726     }
 727 
 728     /* *********************************
 729      * Support for repeating annotations
 730      ***********************************/
 731 
 732     /**
 733      * This context contains all the information needed to synthesize new
 734      * annotations trees for repeating annotations.
 735      */
 736     private class AnnotationContext<T extends Attribute.Compound> {
 737         public final Env<AttrContext> env;
 738         public final Map<Symbol.TypeSymbol, ListBuffer<T>> annotated;
 739         public final Map<T, JCDiagnostic.DiagnosticPosition> pos;
 740         public final boolean isTypeCompound;
 741 
 742         public AnnotationContext(Env<AttrContext> env,
 743                                  Map<Symbol.TypeSymbol, ListBuffer<T>> annotated,
 744                                  Map<T, JCDiagnostic.DiagnosticPosition> pos,
 745                                  boolean isTypeCompound) {
 746             Assert.checkNonNull(env);
 747             Assert.checkNonNull(annotated);
 748             Assert.checkNonNull(pos);
 749 
 750             this.env = env;
 751             this.annotated = annotated;
 752             this.pos = pos;
 753             this.isTypeCompound = isTypeCompound;
 754         }
 755     }
 756 
 757     /* Process repeated annotations. This method returns the
 758      * synthesized container annotation or null IFF all repeating
 759      * annotation are invalid.  This method reports errors/warnings.
 760      */
 761     private <T extends Attribute.Compound> T processRepeatedAnnotations(List<T> annotations,
 762             AnnotationContext<T> ctx, Symbol on, boolean isTypeParam)
 763     {
 764         T firstOccurrence = annotations.head;
 765         List<Attribute> repeated = List.nil();
 766         Type origAnnoType = null;
 767         Type arrayOfOrigAnnoType = null;
 768         Type targetContainerType = null;
 769         MethodSymbol containerValueSymbol = null;
 770 
 771         Assert.check(!annotations.isEmpty() && !annotations.tail.isEmpty()); // i.e. size() > 1
 772 
 773         int count = 0;
 774         for (List<T> al = annotations; !al.isEmpty(); al = al.tail) {
 775             count++;
 776 
 777             // There must be more than a single anno in the annotation list
 778             Assert.check(count > 1 || !al.tail.isEmpty());
 779 
 780             T currentAnno = al.head;
 781 
 782             origAnnoType = currentAnno.type;
 783             if (arrayOfOrigAnnoType == null) {
 784                 arrayOfOrigAnnoType = types.makeArrayType(origAnnoType);
 785             }
 786 
 787             // Only report errors if this isn't the first occurrence I.E. count > 1
 788             boolean reportError = count > 1;
 789             Type currentContainerType = getContainingType(currentAnno, ctx.pos.get(currentAnno), reportError);
 790             if (currentContainerType == null) {
 791                 continue;
 792             }
 793             // Assert that the target Container is == for all repeated
 794             // annos of the same annotation type, the types should
 795             // come from the same Symbol, i.e. be '=='
 796             Assert.check(targetContainerType == null || currentContainerType == targetContainerType);
 797             targetContainerType = currentContainerType;
 798 
 799             containerValueSymbol = validateContainer(targetContainerType, origAnnoType, ctx.pos.get(currentAnno));
 800 
 801             if (containerValueSymbol == null) { // Check of CA type failed
 802                 // errors are already reported
 803                 continue;
 804             }
 805 
 806             repeated = repeated.prepend(currentAnno);
 807         }
 808 
 809         if (!repeated.isEmpty() && targetContainerType == null) {
 810             log.error(ctx.pos.get(annotations.head), Errors.DuplicateAnnotationInvalidRepeated(origAnnoType));
 811             return null;
 812         }
 813 
 814         if (!repeated.isEmpty()) {
 815             repeated = repeated.reverse();
 816             DiagnosticPosition pos = ctx.pos.get(firstOccurrence);
 817             TreeMaker m = make.at(pos);
 818             Pair<MethodSymbol, Attribute> p =
 819                     new Pair<MethodSymbol, Attribute>(containerValueSymbol,
 820                             new Attribute.Array(arrayOfOrigAnnoType, repeated));
 821             if (ctx.isTypeCompound) {
 822                 /* TODO: the following code would be cleaner:
 823                 Attribute.TypeCompound at = new Attribute.TypeCompound(targetContainerType, List.of(p),
 824                         ((Attribute.TypeCompound)annotations.head).position);
 825                 JCTypeAnnotation annoTree = m.TypeAnnotation(at);
 826                 at = attributeTypeAnnotation(annoTree, targetContainerType, ctx.env);
 827                 */
 828                 // However, we directly construct the TypeCompound to keep the
 829                 // direct relation to the contained TypeCompounds.
 830                 Attribute.TypeCompound at = new Attribute.TypeCompound(targetContainerType, List.of(p),
 831                         ((Attribute.TypeCompound)annotations.head).position);
 832 
 833                 JCAnnotation annoTree = m.TypeAnnotation(at);
 834                 if (!chk.validateAnnotationDeferErrors(annoTree))
 835                     log.error(annoTree.pos(), Errors.DuplicateAnnotationInvalidRepeated(origAnnoType));
 836 
 837                 if (!chk.isTypeAnnotation(annoTree, isTypeParam)) {
 838                     log.error(pos, isTypeParam ? Errors.InvalidRepeatableAnnotationNotApplicable(targetContainerType, on)
 839                                                : Errors.InvalidRepeatableAnnotationNotApplicableInContext(targetContainerType));
 840                 }
 841 
 842                 at.setSynthesized(true);
 843 
 844                 @SuppressWarnings("unchecked")
 845                 T x = (T) at;
 846                 return x;
 847             } else {
 848                 Attribute.Compound c = new Attribute.Compound(targetContainerType, List.of(p));
 849                 JCAnnotation annoTree = m.Annotation(c);
 850 
 851                 boolean isRecordMember = (on.flags_field & Flags.RECORD) != 0 || on.enclClass() != null && on.enclClass().isRecord();
 852                 /* if it is a record member we will not issue the error now and wait until annotations on records are
 853                  * checked at Check::validateAnnotation, which will issue it
 854                  */
 855                 if (!chk.annotationApplicable(annoTree, on) && (!isRecordMember || isRecordMember && (on.flags_field & Flags.GENERATED_MEMBER) == 0)) {
 856                     log.error(annoTree.pos(),
 857                               Errors.InvalidRepeatableAnnotationNotApplicable(targetContainerType, on));
 858                 }
 859 
 860                 if (!chk.validateAnnotationDeferErrors(annoTree))
 861                     log.error(annoTree.pos(), Errors.DuplicateAnnotationInvalidRepeated(origAnnoType));
 862 
 863                 c = attributeAnnotation(annoTree, targetContainerType, ctx.env);
 864                 c.setSynthesized(true);
 865 
 866                 @SuppressWarnings("unchecked")
 867                 T x = (T) c;
 868                 return x;
 869             }
 870         } else {
 871             return null; // errors should have been reported elsewhere
 872         }
 873     }
 874 
 875     /**
 876      * Fetches the actual Type that should be the containing annotation.
 877      */
 878     private Type getContainingType(Attribute.Compound currentAnno,
 879                                    DiagnosticPosition pos,
 880                                    boolean reportError)
 881     {
 882         Type origAnnoType = currentAnno.type;
 883         TypeSymbol origAnnoDecl = origAnnoType.tsym;
 884 
 885         // Fetch the Repeatable annotation from the current
 886         // annotation's declaration, or null if it has none
 887         Attribute.Compound ca = origAnnoDecl.getAnnotationTypeMetadata().getRepeatable();
 888         if (ca == null) { // has no Repeatable annotation
 889             if (reportError)
 890                 log.error(pos, Errors.DuplicateAnnotationMissingContainer(origAnnoType));
 891             return null;
 892         }
 893 
 894         return filterSame(extractContainingType(ca, pos, origAnnoDecl),
 895                 origAnnoType);
 896     }
 897 
 898     // returns null if t is same as 's', returns 't' otherwise
 899     private Type filterSame(Type t, Type s) {
 900         if (t == null || s == null) {
 901             return t;
 902         }
 903 
 904         return types.isSameType(t, s) ? null : t;
 905     }
 906 
 907     /** Extract the actual Type to be used for a containing annotation. */
 908     private Type extractContainingType(Attribute.Compound ca,
 909                                        DiagnosticPosition pos,
 910                                        TypeSymbol annoDecl)
 911     {
 912         // The next three checks check that the Repeatable annotation
 913         // on the declaration of the annotation type that is repeating is
 914         // valid.
 915 
 916         // Repeatable must have at least one element
 917         if (ca.values.isEmpty()) {
 918             log.error(pos, Errors.InvalidRepeatableAnnotation(annoDecl));
 919             return null;
 920         }
 921         Pair<MethodSymbol,Attribute> p = ca.values.head;
 922         Name name = p.fst.name;
 923         if (name != names.value) { // should contain only one element, named "value"
 924             log.error(pos, Errors.InvalidRepeatableAnnotation(annoDecl));
 925             return null;
 926         }
 927         if (!(p.snd instanceof Attribute.Class attributeClass)) { // check that the value of "value" is an Attribute.Class
 928             log.error(pos, Errors.InvalidRepeatableAnnotation(annoDecl));
 929             return null;
 930         }
 931 
 932         return attributeClass.getValue();
 933     }
 934 
 935     /* Validate that the suggested targetContainerType Type is a valid
 936      * container type for repeated instances of originalAnnoType
 937      * annotations. Return null and report errors if this is not the
 938      * case, return the MethodSymbol of the value element in
 939      * targetContainerType if it is suitable (this is needed to
 940      * synthesize the container). */
 941     private MethodSymbol validateContainer(Type targetContainerType,
 942                                            Type originalAnnoType,
 943                                            DiagnosticPosition pos) {
 944         MethodSymbol containerValueSymbol = null;
 945         boolean fatalError = false;
 946 
 947         // Validate that there is a (and only 1) value method
 948         Scope scope = targetContainerType.tsym.members();
 949         int nr_value_elems = 0;
 950         boolean error = false;
 951         for(Symbol elm : scope.getSymbolsByName(names.value)) {
 952             nr_value_elems++;
 953 
 954             if (nr_value_elems == 1 &&
 955                     elm.kind == MTH) {
 956                 containerValueSymbol = (MethodSymbol)elm;
 957             } else {
 958                 error = true;
 959             }
 960         }
 961         if (error) {
 962             log.error(pos,
 963                       Errors.InvalidRepeatableAnnotationMultipleValues(targetContainerType,
 964                                                                        nr_value_elems));
 965             return null;
 966         } else if (nr_value_elems == 0) {
 967             log.error(pos,
 968                       Errors.InvalidRepeatableAnnotationNoValue(targetContainerType));
 969             return null;
 970         }
 971 
 972         // validate that the 'value' element is a method
 973         // probably "impossible" to fail this
 974         if (containerValueSymbol.kind != MTH) {
 975             log.error(pos,
 976                     Errors.InvalidRepeatableAnnotationInvalidValue(targetContainerType));
 977             fatalError = true;
 978         }
 979 
 980         // validate that the 'value' element has the correct return type
 981         // i.e. array of original anno
 982         Type valueRetType = containerValueSymbol.type.getReturnType();
 983         Type expectedType = types.makeArrayType(originalAnnoType);
 984         if (!(types.isArray(valueRetType) &&
 985                 types.isSameType(expectedType, valueRetType))) {
 986             log.error(pos,
 987                       Errors.InvalidRepeatableAnnotationValueReturn(targetContainerType,
 988                                                                     valueRetType,
 989                                                                     expectedType));
 990             fatalError = true;
 991         }
 992 
 993         return fatalError ? null : containerValueSymbol;
 994     }
 995 
 996     private <T extends Attribute.Compound> T makeContainerAnnotation(List<T> toBeReplaced,
 997             AnnotationContext<T> ctx, Symbol sym, boolean isTypeParam)
 998     {
 999         // Process repeated annotations
1000         T validRepeated =
1001                 processRepeatedAnnotations(toBeReplaced, ctx, sym, isTypeParam);
1002 
1003         if (validRepeated != null) {
1004             // Check that the container isn't manually
1005             // present along with repeated instances of
1006             // its contained annotation.
1007             ListBuffer<T> manualContainer = ctx.annotated.get(validRepeated.type.tsym);
1008             if (manualContainer != null) {
1009                 log.error(ctx.pos.get(manualContainer.first()),
1010                           Errors.InvalidRepeatableAnnotationRepeatedAndContainerPresent(manualContainer.first().type.tsym));
1011             }
1012         }
1013 
1014         // A null return will delete the Placeholder
1015         return validRepeated;
1016     }
1017 
1018     /********************
1019      * Type annotations *
1020      ********************/
1021 
1022     /**
1023      * Attribute the list of annotations and enter them onto s.
1024      */
1025     public void enterTypeAnnotations(List<JCAnnotation> annotations, Env<AttrContext> env,
1026             Symbol s, DiagnosticPosition deferPos, boolean isTypeParam)
1027     {
1028         Assert.checkNonNull(s, "Symbol argument to actualEnterTypeAnnotations is nul/");
1029         JavaFileObject prev = log.useSource(env.toplevel.sourcefile);
1030         DiagnosticPosition prevLintPos = null;
1031 
1032         if (deferPos != null) {
1033             prevLintPos = deferredLintHandler.setPos(deferPos);
1034         }
1035         try {
1036             annotateNow(s, annotations, env, true, isTypeParam);
1037         } finally {
1038             if (prevLintPos != null)
1039                 deferredLintHandler.setPos(prevLintPos);
1040             log.useSource(prev);
1041         }
1042     }
1043 
1044     /**
1045      * Enqueue tree for scanning of type annotations, attaching to the Symbol sym.
1046      */
1047     public void queueScanTreeAndTypeAnnotate(JCTree tree, Env<AttrContext> env, Symbol sym,
1048             DiagnosticPosition deferPos)
1049     {
1050         Assert.checkNonNull(sym);
1051         normal(() -> tree.accept(new TypeAnnotate(env, sym, deferPos)));
1052     }
1053 
1054     /**
1055      * Apply the annotations to the particular type.
1056      */
1057     public void annotateTypeSecondStage(JCTree tree, List<JCAnnotation> annotations, Type storeAt) {
1058         typeAnnotation(() -> {
1059             List<Attribute.TypeCompound> compounds = fromAnnotations(annotations);
1060             Assert.check(annotations.size() == compounds.size());
1061             // the type already has annotation metadata, but it's empty
1062             Annotations metadata = storeAt.getMetadata(Annotations.class);
1063             Assert.checkNonNull(metadata);
1064             Assert.check(metadata.annotationBuffer().isEmpty());
1065             metadata.annotationBuffer().appendList(compounds);
1066         });
1067     }
1068 
1069     /**
1070      * Apply the annotations to the particular type.
1071      */
1072     public void annotateTypeParameterSecondStage(JCTree tree, List<JCAnnotation> annotations) {
1073         typeAnnotation(() -> {
1074             List<Attribute.TypeCompound> compounds = fromAnnotations(annotations);
1075             Assert.check(annotations.size() == compounds.size());
1076         });
1077     }
1078 
1079     /**
1080      * We need to use a TreeScanner, because it is not enough to visit the top-level
1081      * annotations. We also need to visit type arguments, etc.
1082      */
1083     private class TypeAnnotate extends TreeScanner {
1084         private final Env<AttrContext> env;
1085         private final Symbol sym;
1086         private DiagnosticPosition deferPos;
1087 
1088         public TypeAnnotate(Env<AttrContext> env, Symbol sym, DiagnosticPosition deferPos) {
1089 
1090             this.env = env;
1091             this.sym = sym;
1092             this.deferPos = deferPos;
1093         }
1094 
1095         @Override
1096         public void visitAnnotatedType(JCAnnotatedType tree) {
1097             enterTypeAnnotations(tree.annotations, env, sym, deferPos, false);
1098             scan(tree.underlyingType);
1099         }
1100 
1101         @Override
1102         public void visitTypeParameter(JCTypeParameter tree) {
1103             enterTypeAnnotations(tree.annotations, env, sym, deferPos, true);
1104             scan(tree.bounds);
1105         }
1106 
1107         @Override
1108         public void visitNewArray(JCNewArray tree) {
1109             enterTypeAnnotations(tree.annotations, env, sym, deferPos, false);
1110             for (List<JCAnnotation> dimAnnos : tree.dimAnnotations)
1111                 enterTypeAnnotations(dimAnnos, env, sym, deferPos, false);
1112             scan(tree.elemtype);
1113             scan(tree.elems);
1114         }
1115 
1116         @Override
1117         public void visitMethodDef(JCMethodDecl tree) {
1118             scan(tree.mods);
1119             scan(tree.restype);
1120             scan(tree.typarams);
1121             scan(tree.recvparam);
1122             scan(tree.params);
1123             scan(tree.thrown);
1124             scan(tree.defaultValue);
1125             // Do not annotate the body, just the signature.
1126         }
1127 
1128         @Override
1129         public void visitVarDef(JCVariableDecl tree) {
1130             DiagnosticPosition prevPos = deferPos;
1131             deferPos = tree.pos();
1132             try {
1133                 if (sym != null && sym.kind == VAR) {
1134                     // Don't visit a parameter once when the sym is the method
1135                     // and once when the sym is the parameter.
1136                     scan(tree.mods);
1137                     scan(tree.vartype);
1138                 }
1139                 scan(tree.init);
1140             } finally {
1141                 deferPos = prevPos;
1142             }
1143         }
1144 
1145         @Override
1146         public void visitBindingPattern(JCTree.JCBindingPattern tree) {
1147             //type binding pattern's type will be annotated separately, avoid
1148             //adding its annotations into the owning method here (would clash
1149             //with repeatable annotations).
1150         }
1151 
1152         @Override
1153         public void visitClassDef(JCClassDecl tree) {
1154             // We can only hit a classdef if it is declared within
1155             // a method. Ignore it - the class will be visited
1156             // separately later.
1157         }
1158 
1159         @Override
1160         public void visitNewClass(JCNewClass tree) {
1161             scan(tree.encl);
1162             scan(tree.typeargs);
1163             if (tree.def == null) {
1164                 scan(tree.clazz);
1165             }
1166             scan(tree.args);
1167             // the anonymous class instantiation if any will be visited separately.
1168         }
1169     }
1170 
1171     /*********************
1172      * Completer support *
1173      *********************/
1174 
1175     private AnnotationTypeCompleter theSourceCompleter = new AnnotationTypeCompleter() {
1176         @Override
1177         public void complete(ClassSymbol sym) throws CompletionFailure {
1178             Env<AttrContext> context = typeEnvs.get(sym);
1179             Annotate.this.attributeAnnotationType(context);
1180         }
1181     };
1182 
1183     /* Last stage completer to enter just enough annotations to have a prototype annotation type.
1184      * This currently means entering @Target and @Repeatable.
1185      */
1186     public AnnotationTypeCompleter annotationTypeSourceCompleter() {
1187         return theSourceCompleter;
1188     }
1189 
1190     private void attributeAnnotationType(Env<AttrContext> env) {
1191         Assert.check(((JCClassDecl)env.tree).sym.isAnnotationType(),
1192                 "Trying to annotation type complete a non-annotation type");
1193 
1194         JavaFileObject prev = log.useSource(env.toplevel.sourcefile);
1195         try {
1196             JCClassDecl tree = (JCClassDecl)env.tree;
1197             AnnotationTypeVisitor v = new AnnotationTypeVisitor(attr, chk, syms, typeEnvs);
1198             v.scanAnnotationType(tree);
1199             tree.sym.getAnnotationTypeMetadata().setRepeatable(v.repeatable);
1200             tree.sym.getAnnotationTypeMetadata().setTarget(v.target);
1201         } finally {
1202             log.useSource(prev);
1203         }
1204     }
1205 
1206     public Attribute unfinishedDefaultValue() {
1207         return theUnfinishedDefaultValue;
1208     }
1209 
1210     public static interface AnnotationTypeCompleter {
1211         void complete(ClassSymbol sym) throws CompletionFailure;
1212     }
1213 
1214     /** Visitor to determine a prototype annotation type for a class declaring an annotation type.
1215      *
1216      *  <p><b>This is NOT part of any supported API.
1217      *  If you write code that depends on this, you do so at your own risk.
1218      *  This code and its internal interfaces are subject to change or
1219      *  deletion without notice.</b>
1220      */
1221     public class AnnotationTypeVisitor extends TreeScanner {
1222         private Env<AttrContext> env;
1223 
1224         private final Attr attr;
1225         private final Check check;
1226         private final Symtab tab;
1227         private final TypeEnvs typeEnvs;
1228 
1229         private Compound target;
1230         private Compound repeatable;
1231 
1232         public AnnotationTypeVisitor(Attr attr, Check check, Symtab tab, TypeEnvs typeEnvs) {
1233             this.attr = attr;
1234             this.check = check;
1235             this.tab = tab;
1236             this.typeEnvs = typeEnvs;
1237         }
1238 
1239         public Compound getRepeatable() {
1240             return repeatable;
1241         }
1242 
1243         public Compound getTarget() {
1244             return target;
1245         }
1246 
1247         public void scanAnnotationType(JCClassDecl decl) {
1248             visitClassDef(decl);
1249         }
1250 
1251         @Override
1252         public void visitClassDef(JCClassDecl tree) {
1253             Env<AttrContext> prevEnv = env;
1254             env = typeEnvs.get(tree.sym);
1255             try {
1256                 scan(tree.mods); // look for repeatable and target
1257                 // don't descend into body
1258             } finally {
1259                 env = prevEnv;
1260             }
1261         }
1262 
1263         @Override
1264         public void visitAnnotation(JCAnnotation tree) {
1265             Type t = tree.annotationType.type;
1266             if (t == null) {
1267                 t = attr.attribType(tree.annotationType, env);
1268                 tree.annotationType.type = t = check.checkType(tree.annotationType.pos(), t, tab.annotationType);
1269             }
1270 
1271             if (t == tab.annotationTargetType) {
1272                 target = Annotate.this.attributeAnnotation(tree, tab.annotationTargetType, env);
1273             } else if (t == tab.repeatableType) {
1274                 repeatable = Annotate.this.attributeAnnotation(tree, tab.repeatableType, env);
1275             }
1276         }
1277     }
1278 
1279     /** Represents the semantics of an Annotation Type.
1280      *
1281      *  <p><b>This is NOT part of any supported API.
1282      *  If you write code that depends on this, you do so at your own risk.
1283      *  This code and its internal interfaces are subject to change or
1284      *  deletion without notice.</b>
1285      */
1286     public static class AnnotationTypeMetadata {
1287         final ClassSymbol metaDataFor;
1288         private Compound target;
1289         private Compound repeatable;
1290         private AnnotationTypeCompleter annotationTypeCompleter;
1291 
1292         public AnnotationTypeMetadata(ClassSymbol metaDataFor, AnnotationTypeCompleter annotationTypeCompleter) {
1293             this.metaDataFor = metaDataFor;
1294             this.annotationTypeCompleter = annotationTypeCompleter;
1295         }
1296 
1297         private void init() {
1298             // Make sure metaDataFor is member entered
1299             while (!metaDataFor.isCompleted())
1300                 metaDataFor.complete();
1301 
1302             if (annotationTypeCompleter != null) {
1303                 AnnotationTypeCompleter c = annotationTypeCompleter;
1304                 annotationTypeCompleter = null;
1305                 c.complete(metaDataFor);
1306             }
1307         }
1308 
1309         public void complete() {
1310             init();
1311         }
1312 
1313         public Compound getRepeatable() {
1314             init();
1315             return repeatable;
1316         }
1317 
1318         public void setRepeatable(Compound repeatable) {
1319             Assert.checkNull(this.repeatable);
1320             this.repeatable = repeatable;
1321         }
1322 
1323         public Compound getTarget() {
1324             init();
1325             return target;
1326         }
1327 
1328         public void setTarget(Compound target) {
1329             Assert.checkNull(this.target);
1330                 this.target = target;
1331         }
1332 
1333         public Set<MethodSymbol> getAnnotationElements() {
1334             init();
1335             Set<MethodSymbol> members = new LinkedHashSet<>();
1336             WriteableScope s = metaDataFor.members();
1337             Iterable<Symbol> ss = s.getSymbols(NON_RECURSIVE);
1338             for (Symbol sym : ss)
1339                 if (sym.kind == MTH &&
1340                         sym.name != sym.name.table.names.clinit &&
1341                         (sym.flags() & SYNTHETIC) == 0)
1342                     members.add((MethodSymbol)sym);
1343             return members;
1344         }
1345 
1346         public Set<MethodSymbol> getAnnotationElementsWithDefault() {
1347             init();
1348             Set<MethodSymbol> members = getAnnotationElements();
1349             Set<MethodSymbol> res = new LinkedHashSet<>();
1350             for (MethodSymbol m : members)
1351                 if (m.defaultValue != null)
1352                     res.add(m);
1353             return res;
1354         }
1355 
1356         @Override
1357         public String toString() {
1358             return "Annotation type for: " + metaDataFor;
1359         }
1360 
1361         public boolean isMetadataForAnnotationType() { return true; }
1362 
1363         public static AnnotationTypeMetadata notAnAnnotationType() {
1364             return NOT_AN_ANNOTATION_TYPE;
1365         }
1366 
1367         private static final AnnotationTypeMetadata NOT_AN_ANNOTATION_TYPE =
1368                 new AnnotationTypeMetadata(null, null) {
1369                     @Override
1370                     public void complete() {
1371                     } // do nothing
1372 
1373                     @Override
1374                     public String toString() {
1375                         return "Not an annotation type";
1376                     }
1377 
1378                     @Override
1379                     public Set<MethodSymbol> getAnnotationElements() {
1380                         return new LinkedHashSet<>(0);
1381                     }
1382 
1383                     @Override
1384                     public Set<MethodSymbol> getAnnotationElementsWithDefault() {
1385                         return new LinkedHashSet<>(0);
1386                     }
1387 
1388                     @Override
1389                     public boolean isMetadataForAnnotationType() {
1390                         return false;
1391                     }
1392 
1393                     @Override
1394                     public Compound getTarget() {
1395                         return null;
1396                     }
1397 
1398                     @Override
1399                     public Compound getRepeatable() {
1400                         return null;
1401                     }
1402                 };
1403     }
1404 
1405     public void newRound() {
1406         blockCount = 1;
1407     }
1408 
1409     public Queues setQueues(Queues nue) {
1410         Queues stored = new Queues(q, validateQ, typesQ, afterTypesQ);
1411         this.q = nue.q;
1412         this.typesQ = nue.typesQ;
1413         this.afterTypesQ = nue.afterTypesQ;
1414         this.validateQ = nue.validateQ;
1415         return stored;
1416     }
1417 
1418     static class Queues {
1419         private final ListBuffer<Runnable> q;
1420         private final ListBuffer<Runnable> validateQ;
1421         private final ListBuffer<Runnable> typesQ;
1422         private final ListBuffer<Runnable> afterTypesQ;
1423 
1424         public Queues() {
1425             this(new ListBuffer<Runnable>(), new ListBuffer<Runnable>(), new ListBuffer<Runnable>(), new ListBuffer<Runnable>());
1426         }
1427 
1428         public Queues(ListBuffer<Runnable> q, ListBuffer<Runnable> validateQ, ListBuffer<Runnable> typesQ, ListBuffer<Runnable> afterTypesQ) {
1429             this.q = q;
1430             this.validateQ = validateQ;
1431             this.typesQ = typesQ;
1432             this.afterTypesQ = afterTypesQ;
1433         }
1434     }
1435 }