3131import sun .reflect .annotation .AnnotationTypeMismatchExceptionProxy ;
3232import sun .reflect .annotation .EnumValue ;
3333import sun .reflect .annotation .EnumValueArray ;
34+ import sun .reflect .annotation .TypeAnnotation ;
35+ import sun .reflect .annotation .TypeAnnotationParser ;
3436import sun .reflect .annotation .TypeNotPresentExceptionProxy ;
3537
3638import java .io .ByteArrayInputStream ;
@@ -184,14 +186,17 @@ public static int encodeThrowable(Throwable throwable, long buffer, int bufferSi
184186 }
185187
186188 /**
187- * Parses {@code rawAnnotations} into a list of {@link Annotation}s and then
188- * serializes them to a byte array with {@link #encodeAnnotations(Collection)}.
189+ * Parses {@code rawAnnotations} into a list of {@link Annotation}s (isTypeAnnotations == false)
190+ * or {@link TypeAnnotation}s (isTypeAnnotations == true) and then
191+ * serializes them to a byte array with {@link #encodeAnnotations(Collection)} or
192+ * {@link #encodeTypeAnnotations(TypeAnnotation[])}.
189193 */
190194 public static byte [] encodeAnnotations (byte [] rawAnnotations ,
195+ boolean isTypeAnnotations ,
191196 Class <?> declaringClass ,
192197 ConstantPool cp ,
193198 Class <? extends Annotation >[] selectAnnotationClasses ) {
194- if (selectAnnotationClasses != null ) {
199+ if (! isTypeAnnotations && selectAnnotationClasses != null ) {
195200 if (selectAnnotationClasses .length == 0 ) {
196201 throw new IllegalArgumentException ("annotation class selection must be null or non-empty" );
197202 }
@@ -201,7 +206,11 @@ public static byte[] encodeAnnotations(byte[] rawAnnotations,
201206 }
202207 }
203208 }
204- return encodeAnnotations (AnnotationParser .parseSelectAnnotations (rawAnnotations , cp , declaringClass , false , selectAnnotationClasses ).values ());
209+ if (isTypeAnnotations ) {
210+ return encodeTypeAnnotations (TypeAnnotationParser .parseTypeAnnotations (rawAnnotations , cp , null , false , declaringClass ));
211+ } else {
212+ return encodeAnnotations (AnnotationParser .parseSelectAnnotations (rawAnnotations , cp , declaringClass , false , selectAnnotationClasses ).values ());
213+ }
205214 }
206215
207216 /**
@@ -399,18 +408,54 @@ private static void encodeAnnotation(DataOutputStream dos, Annotation a) throws
399408 }
400409 }
401410
411+ /**
412+ * Encodes typeAnnotations to a byte array. The byte array can be decoded with {@link #decodeAnnotations(byte[], AnnotationDecoder)}.
413+ */
414+ public static byte [] encodeTypeAnnotations (TypeAnnotation [] typeAnnotations ) {
415+ try {
416+ ByteArrayOutputStream baos = new ByteArrayOutputStream (128 );
417+ try (DataOutputStream dos = new DataOutputStream (baos )) {
418+ writeLength (dos , typeAnnotations .length );
419+ for (TypeAnnotation ta : typeAnnotations ) {
420+ encodeTypeAnnotation (dos , ta );
421+ }
422+ }
423+ return baos .toByteArray ();
424+ } catch (Exception e ) {
425+ throw new InternalError (e );
426+ }
427+ }
428+
429+ private static void encodeTypeAnnotation (DataOutputStream dos , TypeAnnotation ta ) throws Exception {
430+ TypeAnnotation .TypeAnnotationTargetInfo ti = ta .getTargetInfo ();
431+ TypeAnnotation .TypeAnnotationTarget target = ti .getTarget ();
432+ dos .writeUTF (target .name ());
433+ dos .writeInt (ti .getCount ());
434+ dos .writeInt (ti .getSecondaryIndex ());
435+ TypeAnnotation .LocationInfo li = ta .getLocationInfo ();
436+ int depth = li .getDepth ();
437+ dos .writeInt (depth );
438+ for (int i = 0 ; i < depth ; i ++) {
439+ TypeAnnotation .LocationInfo .Location loc = li .getLocationAt (i );
440+ dos .write (loc .tag );
441+ dos .writeShort (loc .index );
442+ }
443+ encodeAnnotation (dos , ta .getAnnotation ());
444+ }
445+
402446 /**
403447 * Helper for {@link #decodeAnnotations(byte[], AnnotationDecoder)} to convert a byte
404448 * array (ostensibly produced by {@link VMSupport#encodeAnnotations}) into objects.
405449 *
406450 * @param <T> type to which a type name is {@linkplain #resolveType(String) resolved}
407451 * @param <A> type of the object representing a decoded annotation
452+ * @param <TA> type of the object representing a decoded type annotation
408453 * @param <E> type of the object representing a decoded enum constant
409454 * @param <EA> type of the object representing a decoded array of enum constants
410455 * @param <MT> type of the object representing a missing type
411456 * @param <ETM> type of the object representing a decoded element type mismatch
412457 */
413- public interface AnnotationDecoder <T , A , E , EA , MT , ETM > {
458+ public interface AnnotationDecoder <T , A , TA , E , EA , MT , ETM > {
414459 /**
415460 * Resolves a name in {@link Class#getName()} format to an object of type {@code T}.
416461 */
@@ -424,6 +469,11 @@ public interface AnnotationDecoder<T, A, E, EA, MT, ETM> {
424469 */
425470 A newAnnotation (T type , Map .Entry <String , Object >[] elements );
426471
472+ /**
473+ * Creates an object representing a decoded type annotation.
474+ */
475+ TA newTypeAnnotation (TypeAnnotation .TypeAnnotationTargetInfo targetInfo , TypeAnnotation .LocationInfo locationInfo , A annotation );
476+
427477 /**
428478 * Creates an object representing a decoded enum.
429479 *
@@ -468,7 +518,7 @@ public interface AnnotationDecoder<T, A, E, EA, MT, ETM> {
468518 * @return an immutable list of {@code A} objects
469519 */
470520 @ SuppressWarnings ("unchecked" )
471- public static <T , A , E , EA , MT , ETM > List <A > decodeAnnotations (byte [] encoded , AnnotationDecoder <T , A , E , EA , MT , ETM > decoder ) {
521+ public static <T , A , TA , E , EA , MT , ETM > List <A > decodeAnnotations (byte [] encoded , AnnotationDecoder <T , A , TA , E , EA , MT , ETM > decoder ) {
472522 try {
473523 ByteArrayInputStream bais = new ByteArrayInputStream (encoded );
474524 DataInputStream dis = new DataInputStream (bais );
@@ -479,7 +529,7 @@ public static <T, A, E, EA, MT, ETM> List<A> decodeAnnotations(byte[] encoded, A
479529 }
480530
481531 @ SuppressWarnings ({"rawtypes" , "unchecked" })
482- private static <T , A , E , EA , MT , ETM > A decodeAnnotation (DataInputStream dis , AnnotationDecoder <T , A , E , EA , MT , ETM > decoder ) throws IOException {
532+ private static <T , A , TA , E , EA , MT , ETM > A decodeAnnotation (DataInputStream dis , AnnotationDecoder <T , A , TA , E , EA , MT , ETM > decoder ) throws IOException {
483533 String typeName = dis .readUTF ();
484534 T type = decoder .resolveType (typeName );
485535 int n = readLength (dis );
@@ -508,12 +558,13 @@ private static <T, A, E, EA, MT, ETM> A decodeAnnotation(DataInputStream dis, An
508558 }
509559 return decoder .newAnnotation (type , (Map .Entry <String , Object >[]) elements );
510560 }
561+
511562 @ FunctionalInterface
512563 interface IOReader {
513564 Object read () throws IOException ;
514565 }
515566
516- private static <T , A , E , EA , MT , ETM > Object decodeArray (DataInputStream dis , AnnotationDecoder <T , A , E , EA , MT , ETM > decoder ) throws IOException {
567+ private static <T , A , TA , E , EA , MT , ETM > Object decodeArray (DataInputStream dis , AnnotationDecoder <T , A , TA , E , EA , MT , ETM > decoder ) throws IOException {
517568 byte componentTag = dis .readByte ();
518569 return switch (componentTag ) {
519570 case 'B' -> readArray (dis , dis ::readByte );
@@ -540,7 +591,7 @@ private static <T, A, E, EA, MT, ETM> Object decodeArray(DataInputStream dis, An
540591 * returns it as an object of type {@code E}.
541592 */
542593 @ SuppressWarnings ("unchecked" )
543- private static <T , A , E , EA , MT , ETM > EA readEnumArray (DataInputStream dis , AnnotationDecoder <T , A , E , EA , MT , ETM > decoder , T enumType ) throws IOException {
594+ private static <T , A , TA , E , EA , MT , ETM > EA readEnumArray (DataInputStream dis , AnnotationDecoder <T , A , TA , E , EA , MT , ETM > decoder , T enumType ) throws IOException {
544595 List <String > names = (List <String >) readArray (dis , dis ::readUTF );
545596 return decoder .newEnumArray (enumType , names );
546597 }
@@ -549,7 +600,7 @@ private static <T, A, E, EA, MT, ETM> EA readEnumArray(DataInputStream dis, Anno
549600 * Reads a class encoded at the current read position of {@code dis} and
550601 * returns it as an object of type {@code T}.
551602 */
552- private static <T , A , E , EA , MT , ETM > T readClass (DataInputStream dis , AnnotationDecoder <T , A , E , EA , MT , ETM > decoder ) throws IOException {
603+ private static <T , A , TA , E , EA , MT , ETM > T readClass (DataInputStream dis , AnnotationDecoder <T , A , TA , E , EA , MT , ETM > decoder ) throws IOException {
553604 return decoder .resolveType (dis .readUTF ());
554605 }
555606
@@ -594,4 +645,45 @@ private static int readLength(DataInputStream dis) throws IOException {
594645 }
595646 return length ;
596647 }
648+
649+ /**
650+ * Decodes type annotations serialized in {@code encoded} to objects.
651+ *
652+ * @param <T> type to which a type name is resolved
653+ * @param <A> type of the object representing a decoded annotation
654+ * @param <E> type of the object representing a decoded enum constant
655+ * @param <MT> type of the object representing a missing type
656+ * @param <ETM> type of the object representing a decoded element type mismatch
657+ * @return an immutable list of {@code A} objects
658+ */
659+ @ SuppressWarnings ("unchecked" )
660+ public static <T , A , TA , E , EA , MT , ETM > List <TA > decodeTypeAnnotations (byte [] encoded , AnnotationDecoder <T , A , TA , E , EA , MT , ETM > decoder ) {
661+ try {
662+ ByteArrayInputStream bais = new ByteArrayInputStream (encoded );
663+ DataInputStream dis = new DataInputStream (bais );
664+ return (List <TA >) readArray (dis , () -> decodeTypeAnnotation (dis , decoder ));
665+ } catch (Exception e ) {
666+ throw new InternalError (e );
667+ }
668+ }
669+
670+ private static <T , A , TA , E , EA , MT , ETM > TA decodeTypeAnnotation (DataInputStream dis , AnnotationDecoder <T , A , TA , E , EA , MT , ETM > decoder ) throws IOException {
671+ TypeAnnotation .TypeAnnotationTarget target = TypeAnnotation .TypeAnnotationTarget .valueOf (dis .readUTF ());
672+ int count = dis .readInt ();
673+ int secondaryIndex = dis .readInt ();
674+ TypeAnnotation .TypeAnnotationTargetInfo ti = new TypeAnnotation .TypeAnnotationTargetInfo (target , count , secondaryIndex );
675+ int depth = dis .readInt ();
676+ TypeAnnotation .LocationInfo li ;
677+ if (depth == 0 ) {
678+ li = TypeAnnotation .LocationInfo .BASE_LOCATION ;
679+ } else {
680+ TypeAnnotation .LocationInfo .Location [] locs = new TypeAnnotation .LocationInfo .Location [depth ];
681+ for (int i = 0 ; i != depth ; i ++) {
682+ locs [i ] = new TypeAnnotation .LocationInfo .Location (dis .readByte (), dis .readShort ());
683+ }
684+ li = new TypeAnnotation .LocationInfo (depth , locs );
685+ }
686+ A annotation = decodeAnnotation (dis , decoder );
687+ return decoder .newTypeAnnotation (ti , li , annotation );
688+ }
597689}
0 commit comments