1 /*

     2  * Copyright 2005-2007 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

    20  * CA 95054 USA or visit www.sun.com if you need additional information or

    21  * have any 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 }