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
24import java.io.*;
25import 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 * @modules jdk.jdeps/com.sun.tools.classfile
32 */
33
34public class NewArray {
35
36    public static void main(String[] args) throws Exception {
37        new NewArray().run();
38    }
39
40    public void run() throws Exception {
41        File javaFile = writeTestFile();
42        File classFile = compileTestFile(javaFile);
43
44        ClassFile cf = ClassFile.read(classFile);
45        for (Method m: cf.methods) {
46            test(cf, m);
47        }
48
49        countAnnotations();
50
51        if (errors > 0)
52            throw new Exception(errors + " errors found");
53        System.out.println("PASSED");
54    }
55
56    void test(ClassFile cf, Method m) {
57        test(cf, m, Attribute.RuntimeVisibleTypeAnnotations, true);
58        test(cf, m, Attribute.RuntimeInvisibleTypeAnnotations, false);
59    }
60
61    // test the result of Attributes.getIndex according to expectations
62    // encoded in the method's name
63    void test(ClassFile cf, Method m, String name, boolean visible) {
64        Attribute attr = null;
65        Code_attribute cAttr = null;
66        RuntimeTypeAnnotations_attribute tAttr = null;
67
68        int index = m.attributes.getIndex(cf.constant_pool, Attribute.Code);
69        if(index!= -1) {
70            attr = m.attributes.get(index);
71            assert attr instanceof Code_attribute;
72            cAttr = (Code_attribute)attr;
73            index = cAttr.attributes.getIndex(cf.constant_pool, name);
74            if(index!= -1) {
75                attr = cAttr.attributes.get(index);
76                assert attr instanceof RuntimeTypeAnnotations_attribute;
77                tAttr = (RuntimeTypeAnnotations_attribute)attr;
78                all += tAttr.annotations.length;
79                if (visible)
80                    visibles += tAttr.annotations.length;
81                else
82                    invisibles += tAttr.annotations.length;
83               }
84        }
85    }
86
87    File writeTestFile() throws IOException {
88      File f = new File("Test.java");
89        PrintWriter out = new PrintWriter(new BufferedWriter(new FileWriter(f)));
90        out.println("import java.lang.annotation.*;");
91        out.println("import java.util.*;");
92        out.println("class Test { ");
93        out.println("  @Target(ElementType.TYPE_USE) @interface A { }");
94        out.println("  void test() {");
95        out.println("    Object a = new @A String @A [5] @A  [];");
96        out.println("    Object b = new @A String @A [5] @A [3];");
97        out.println("    Object c = new @A String @A [] @A [] {};");
98        out.println("  }");
99        out.println("}");
100
101        out.close();
102        return f;
103    }
104
105    File compileTestFile(File f) {
106        int rc = com.sun.tools.javac.Main.compile(new String[] {"-g", f.getPath() });
107        if (rc != 0)
108            throw new Error("compilation failed. rc=" + rc);
109        String path = f.getPath();
110        return new File(path.substring(0, path.length() - 5) + ".class");
111    }
112
113    void countAnnotations() {
114        int expected_visibles = 0, expected_invisibles = 9;
115        int expected_all = expected_visibles + expected_invisibles;
116
117        if (expected_all != all) {
118            errors++;
119            System.err.println("expected " + expected_all
120                    + " annotations but found " + all);
121        }
122
123        if (expected_visibles != visibles) {
124            errors++;
125            System.err.println("expected " + expected_visibles
126                    + " visibles annotations but found " + visibles);
127        }
128
129        if (expected_invisibles != invisibles) {
130            errors++;
131            System.err.println("expected " + expected_invisibles
132                    + " invisibles annotations but found " + invisibles);
133        }
134
135    }
136
137    int errors;
138    int all;
139    int visibles;
140    int invisibles;
141}
142