1 /*
2 * Copyright (c) 2008, 2013, 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 NewArray
29 * @bug 6843077
30 * @summary Test type annotations on local array are in method's code attribute.
31 */
32
33 public class NewArray {
34
35 public static void main(String[] args) throws Exception {
36 new NewArray().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 Attribute attr = null;
64 Code_attribute cAttr = null;
65 RuntimeTypeAnnotations_attribute tAttr = null;
66
67 int index = m.attributes.getIndex(cf.constant_pool, Attribute.Code);
68 if(index!= -1) {
69 attr = m.attributes.get(index);
70 assert attr instanceof Code_attribute;
71 cAttr = (Code_attribute)attr;
72 index = cAttr.attributes.getIndex(cf.constant_pool, name);
73 if(index!= -1) {
74 attr = cAttr.attributes.get(index);
75 assert attr instanceof RuntimeTypeAnnotations_attribute;
76 tAttr = (RuntimeTypeAnnotations_attribute)attr;
77 all += tAttr.annotations.length;
78 if (visible)
79 visibles += tAttr.annotations.length;
80 else
81 invisibles += tAttr.annotations.length;
82 }
83 }
84 }
85
86 File writeTestFile() throws IOException {
87 File f = new File("Test.java");
88 PrintWriter out = new PrintWriter(new BufferedWriter(new FileWriter(f)));
89 out.println("import java.lang.annotation.*;");
90 out.println("import java.util.*;");
91 out.println("class Test { ");
92 out.println(" @Target(ElementType.TYPE_USE) @interface A { }");
93 out.println(" void test() {");
94 out.println(" Object a = new @A String @A [5] @A [];");
95 out.println(" Object b = new @A String @A [5] @A [3];");
96 out.println(" Object c = new @A String @A [] @A [] {};");
97 out.println(" }");
98 out.println("}");
99
100 out.close();
101 return f;
102 }
103
104 File compileTestFile(File f) {
105 int rc = com.sun.tools.javac.Main.compile(new String[] { "-source", "1.8", "-g", f.getPath() });
106 if (rc != 0)
107 throw new Error("compilation failed. rc=" + rc);
108 String path = f.getPath();
109 return new File(path.substring(0, path.length() - 5) + ".class");
110 }
111
112 void countAnnotations() {
113 int expected_visibles = 0, expected_invisibles = 9;
114 int expected_all = expected_visibles + expected_invisibles;
115
116 if (expected_all != all) {
117 errors++;
118 System.err.println("expected " + expected_all
119 + " annotations but found " + all);
120 }
121
122 if (expected_visibles != visibles) {
123 errors++;
124 System.err.println("expected " + expected_visibles
125 + " visibles annotations but found " + visibles);
126 }
127
128 if (expected_invisibles != invisibles) {
129 errors++;
130 System.err.println("expected " + expected_invisibles
131 + " invisibles annotations but found " + invisibles);
132 }
133
134 }
135
136 int errors;
137 int all;
138 int visibles;
139 int invisibles;
140 }