1 /*
 2  * Copyright (c) 2005, 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 /**
25  * @test
26  * @bug 6331574
27  * @summary test isModifiableClass
28  * @author Robert Field, Sun Microsystems
29  *
30  * @run build IsModifiableClassApp IsModifiableClassAgent
31  * @run shell MakeJAR3.sh IsModifiableClassAgent 'Can-Retransform-Classes: true'
32  * @run main/othervm -javaagent:IsModifiableClassAgent.jar IsModifiableClassApp
33  */
34 import java.lang.instrument.*;
35 
36 public class IsModifiableClassAgent
37 {
38     public static boolean fail = false;
39     public static boolean completed = false;
40 
41     public static void
42     premain(    String agentArgs,
43                 Instrumentation instrumentation)
44         {
45             System.out.println("IsModifiableClassAgent started");
46 
47             Class[] allClasses = instrumentation.getAllLoadedClasses();
48             int modCount = 0;
49             int unmodCount = 0;
50 
51             for (int i = 0; i < allClasses.length; i++)
52                 {
53                     Class klass = allClasses[i];
54                     boolean isMod = instrumentation.isModifiableClass(klass);
55                     if (isMod && klass.isArray()) {
56                         System.err.println("Error: array class returned as modifiable: " + klass);
57                         fail = true;
58                     }
59                     if (isMod && klass.isPrimitive()) {
60                         System.err.println("Error: primitive class returned as modifiable: " + klass);
61                         fail = true;
62                     }
63                     try {
64                         instrumentation.retransformClasses(klass);
65                         if (!isMod) {
66                             System.err.println("Error: unmodifiable class retransformable: " + klass);
67                             fail = true;
68                         }
69                     } catch (UnmodifiableClassException e) {
70                         if (isMod) {
71                             System.err.println("Error: modifiable class not retransformable: " + klass);
72                             System.err.println("  exception: " + e);
73                             fail = true;
74                         }
75                     } catch (Throwable e) {
76                         System.err.println("Error: bad return from retransform: " + klass);
77                         System.err.println("  ERROR: " + e);
78                         fail = true;
79                     }
80                     if (isMod) {
81                         ++modCount;
82                     } else {
83                         ++unmodCount;
84                     }
85                 }
86             System.out.println("modifiable: " + modCount + ". unmodifiable: " + unmodCount);
87             completed = true;
88         }
89 }