1 /*
2 * Copyright (c) 2008, 2015, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24 import java.io.*;
25 import com.sun.tools.classfile.*;
26
27 /*
28 * @test Visibility
29 * @bug 6843077
30 * @summary test that type annotations are recorded in the classfile
31 * @modules jdk.jdeps/com.sun.tools.classfile
32 */
33
34 public class Visibility {
35 public static void main(String[] args) throws Exception {
36 new Visibility().run();
37 }
38
39 public void run() throws Exception {
40 File javaFile = writeTestFile();
41 File classFile = compileTestFile(javaFile);
42
43 ClassFile cf = ClassFile.read(classFile);
44 for (Method m: cf.methods) {
45 test(cf, m);
46 }
47
48 countAnnotations();
49
50 if (errors > 0)
51 throw new Exception(errors + " errors found");
52 System.out.println("PASSED");
53 }
54
55 void test(ClassFile cf, Method m) {
56 test(cf, m, Attribute.RuntimeVisibleTypeAnnotations, true);
57 test(cf, m, Attribute.RuntimeInvisibleTypeAnnotations, false);
58 }
59
60 // test the result of Attributes.getIndex according to expectations
61 // encoded in the method's name
62 void test(ClassFile cf, Method m, String name, boolean visible) {
63 int index = m.attributes.getIndex(cf.constant_pool, name);
64 if (index != -1) {
65 Attribute attr = m.attributes.get(index);
66 assert attr instanceof RuntimeTypeAnnotations_attribute;
67 RuntimeTypeAnnotations_attribute tAttr = (RuntimeTypeAnnotations_attribute)attr;
68 all += tAttr.annotations.length;
69 if (visible)
70 visibles += tAttr.annotations.length;
71 else
72 invisibles += tAttr.annotations.length;
73 }
74 }
75
76 File writeTestFile() throws IOException {
77 File f = new File("Test.java");
78 PrintWriter out = new PrintWriter(new BufferedWriter(new FileWriter(f)));
79 out.println("import java.lang.annotation.ElementType;");
80 out.println("import java.lang.annotation.Retention;");
81 out.println("import java.lang.annotation.RetentionPolicy;");
82 out.println("import java.lang.annotation.Target;");
83 out.println("abstract class Test { ");
84 // visible annotations: RUNTIME
85 out.println(" @Retention(RetentionPolicy.RUNTIME)");
86 out.println(" @Target(ElementType.TYPE_USE)");
87 out.println(" @interface A { }");
88 out.println(" void visible(@A Test this) { }");
89
90 // invisible annotations: CLASS
91 out.println(" @Retention(RetentionPolicy.CLASS)");
92 out.println(" @Target(ElementType.TYPE_USE)");
93 out.println(" @interface B { }");
94 out.println(" void invisible(@B Test this) { }");
95
96 // source annotations
97 out.println(" @Retention(RetentionPolicy.SOURCE)");
98 out.println(" @Target(ElementType.TYPE_USE)");
99 out.println(" @interface C { }");
100 out.println(" void source(@C Test this) { }");
101
102 // default visibility: CLASS
103 out.println(" @Target(ElementType.TYPE_USE)");
104 out.println(" @interface D { }");
105 out.println(" void def(@D Test this) { }");
106 out.println("}");
107 out.close();
108 return f;
109 }
110
111 File compileTestFile(File f) {
112 int rc = com.sun.tools.javac.Main.compile(new String[] {"-g", f.getPath() });
113 if (rc != 0)
114 throw new Error("compilation failed. rc=" + rc);
115 String path = f.getPath();
116 return new File(path.substring(0, path.length() - 5) + ".class");
117 }
118
119 void countAnnotations() {
120 int expected_all = 3, expected_visibles = 1, expected_invisibles = 2;
121
122 if (expected_all != all) {
123 errors++;
124 System.err.println("expected " + expected_all
125 + " annotations but found " + all);
126 }
127
128 if (expected_visibles != visibles) {
129 errors++;
130 System.err.println("expected " + expected_visibles
131 + " visibles annotations but found " + visibles);
132 }
133
134 if (expected_invisibles != invisibles) {
135 errors++;
136 System.err.println("expected " + expected_invisibles
137 + " invisibles annotations but found " + invisibles);
138 }
139
140 }
141
142 int errors;
143 int all;
144 int visibles;
145 int invisibles;
146 }