ohair@286: /*
alanb@368:  * Copyright (c) 2008, 2012, Oracle and/or its affiliates. All rights reserved.
ohair@286:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
ohair@286:  *
ohair@286:  * This code is free software; you can redistribute it and/or modify it
ohair@286:  * under the terms of the GNU General Public License version 2 only, as
ohair@286:  * published by the Free Software Foundation.  Oracle designates this
ohair@286:  * particular file as subject to the "Classpath" exception as provided
ohair@286:  * by Oracle in the LICENSE file that accompanied this code.
ohair@286:  *
ohair@286:  * This code is distributed in the hope that it will be useful, but WITHOUT
ohair@286:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
ohair@286:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
ohair@286:  * version 2 for more details (a copy is included in the LICENSE file that
ohair@286:  * accompanied this code).
ohair@286:  *
ohair@286:  * You should have received a copy of the GNU General Public License version
ohair@286:  * 2 along with this work; if not, write to the Free Software Foundation,
ohair@286:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
ohair@286:  *
ohair@286:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
ohair@286:  * or visit www.oracle.com if you need additional information or have any
ohair@286:  * questions.
ohair@286:  */
ohair@286: 
ohair@286: package com.sun.xml.internal.ws.model;
ohair@286: 
ohair@286: import javax.xml.ws.WebServiceException;
ohair@286: import java.lang.reflect.InvocationTargetException;
ohair@286: import java.lang.reflect.Method;
ohair@286: import java.net.URL;
ohair@286: import java.security.AccessController;
ohair@286: import java.security.PrivilegedAction;
ohair@286: import java.util.logging.Level;
ohair@286: import java.util.logging.Logger;
ohair@286: 
ohair@286: /**
ohair@286:  * A {@link ClassLoader} used to "inject" wrapper and exception bean classes
ohair@286:  * into the VM.
ohair@286:  *
ohair@286:  * @author Jitendra kotamraju
ohair@286:  */
ohair@286: final class Injector {
ohair@286: 
ohair@286:     private static final Logger LOGGER = Logger.getLogger(Injector.class.getName());
ohair@286: 
ohair@286:     private static final Method defineClass;
ohair@286:     private static final Method resolveClass;
ohair@286:     private static final Method getPackage;
ohair@286:     private static final Method definePackage;
ohair@286: 
ohair@286:     static {
ohair@286:         try {
ohair@286:             defineClass = ClassLoader.class.getDeclaredMethod("defineClass",String.class,byte[].class,Integer.TYPE,Integer.TYPE);
ohair@286:             resolveClass = ClassLoader.class.getDeclaredMethod("resolveClass",Class.class);
ohair@286:             getPackage = ClassLoader.class.getDeclaredMethod("getPackage", String.class);
ohair@286:             definePackage = ClassLoader.class.getDeclaredMethod("definePackage",
ohair@286:                     String.class, String.class, String.class, String.class,
ohair@286:                     String.class, String.class, String.class, URL.class);
ohair@286:         } catch (NoSuchMethodException e) {
ohair@286:             // impossible
ohair@286:             throw new NoSuchMethodError(e.getMessage());
ohair@286:         }
ohair@286:         AccessController.doPrivileged(new PrivilegedAction<Void>() {
ohair@286:             public Void run() {
ohair@286:                 // TODO: check security implication
ohair@286:                 // do these setAccessible allow anyone to call these methods freely?s
ohair@286:                 defineClass.setAccessible(true);
ohair@286:                 resolveClass.setAccessible(true);
ohair@286:                 getPackage.setAccessible(true);
ohair@286:                 definePackage.setAccessible(true);
ohair@286:                 return null;
ohair@286:             }
ohair@286:         });
ohair@286:     }
ohair@286: 
ohair@286:     static synchronized Class inject(ClassLoader cl, String className, byte[] image) {
ohair@286:         // To avoid race conditions let us check if the classloader
ohair@286:         // already contains the class
ohair@286:         try {
ohair@286:             return cl.loadClass(className);
ohair@286:         } catch (ClassNotFoundException e) {
ohair@286:             // nothing to do
ohair@286:         }
ohair@286:         try {
ohair@286:             int packIndex = className.lastIndexOf('.');
ohair@286:             if (packIndex != -1) {
ohair@286:                 String pkgname = className.substring(0, packIndex);
ohair@286:                 // Check if package already loaded.
ohair@286:                 Package pkg = (Package)getPackage.invoke(cl, pkgname);
ohair@286:                 if (pkg == null) {
ohair@286:                     definePackage.invoke(cl, pkgname, null, null, null, null, null, null, null);
ohair@286:                 }
ohair@286:             }
ohair@286: 
ohair@286:             Class c = (Class)defineClass.invoke(cl,className.replace('/','.'),image,0,image.length);
ohair@286:             resolveClass.invoke(cl, c);
ohair@286:             return c;
ohair@286:         } catch (IllegalAccessException e) {
ohair@286:             LOGGER.log(Level.FINE,"Unable to inject "+className,e);
ohair@286:             throw new WebServiceException(e);
ohair@286:         } catch (InvocationTargetException e) {
ohair@286:             LOGGER.log(Level.FINE,"Unable to inject "+className,e);
ohair@286:             throw new WebServiceException(e);
ohair@286:         }
ohair@286:     }
ohair@286: 
ohair@286: }