/*
* Copyright (c) 2008, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation. Oracle designates this
* particular file as subject to the "Classpath" exception as provided
* by Oracle in the LICENSE file that accompanied this code.
*
* This code is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
package sun.tracing.dtrace;
import java.lang.reflect.Method;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Modifier;
import java.lang.reflect.Constructor;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.InvocationTargetException;
import java.lang.annotation.Annotation;
import java.util.HashMap;
import sun.tracing.ProviderSkeleton;
import sun.tracing.ProbeSkeleton;
import com.sun.tracing.Provider;
import com.sun.tracing.ProviderName;
import com.sun.tracing.ProbeName;
import com.sun.tracing.dtrace.Attributes;
import com.sun.tracing.dtrace.ModuleName;
import com.sun.tracing.dtrace.FunctionName;
import com.sun.tracing.dtrace.StabilityLevel;
import com.sun.tracing.dtrace.DependencyClass;
import sun.misc.ProxyGenerator;
class DTraceProvider extends ProviderSkeleton {
private Activation activation;
private Object proxy;
// For proxy generation
private final static Class[] constructorParams = { InvocationHandler.class };
private final String proxyClassNamePrefix = "$DTraceTracingProxy";
static final String DEFAULT_MODULE = "java_tracing";
static final String DEFAULT_FUNCTION = "unspecified";
private static long nextUniqueNumber = 0;
private static synchronized long getUniqueNumber() {
return nextUniqueNumber++;
}
protected ProbeSkeleton createProbe(Method m) {
return new DTraceProbe(proxy, m);
}
DTraceProvider(Class<? extends Provider> type) {
super(type);
}
void setProxy(Object p) {
proxy = p;
}
void setActivation(Activation a) {
this.activation = a;
}
public void dispose() {
if (activation != null) {
activation.disposeProvider(this);
activation = null;
}
super.dispose();
}
/**
* Magic routine which creates an implementation of the user's interface.
*
* This method uses the ProxyGenerator directly to bypass the
* java.lang.reflect.proxy cache so that we get a unique class each
* time it's called and can't accidently reuse a $Proxy class.
*
* @return an implementation of the user's interface
*/
@SuppressWarnings("unchecked")
public <T extends Provider> T newProxyInstance() {
/*
* Choose a name for the proxy class to generate.
*/
long num = getUniqueNumber();
String proxyPkg = "";
if (!Modifier.isPublic(providerType.getModifiers())) {
String name = providerType.getName();
int n = name.lastIndexOf('.');
proxyPkg = ((n == -1) ? "" : name.substring(0, n + 1));
}
String proxyName = proxyPkg + proxyClassNamePrefix + num;
/*
* Generate the specified proxy class.
*/
Class<?> proxyClass = null;
byte[] proxyClassFile = ProxyGenerator.generateProxyClass(
proxyName, new Class<?>[] { providerType });
try {
proxyClass = JVM.defineClass(
providerType.getClassLoader(), proxyName,
proxyClassFile, 0, proxyClassFile.length);
} catch (ClassFormatError e) {
/*
* A ClassFormatError here means that (barring bugs in the
* proxy class generation code) there was some other
* invalid aspect of the arguments supplied to the proxy
* class creation (such as virtual machine limitations
* exceeded).
*/
throw new IllegalArgumentException(e.toString());
}
/*
* Invoke its constructor with the designated invocation handler.
*/
try {
Constructor cons = proxyClass.getConstructor(constructorParams);
return (T)cons.newInstance(new Object[] { this });
} catch (NoSuchMethodException e) {
throw new InternalError(e.toString());
} catch (IllegalAccessException e) {
throw new InternalError(e.toString());
} catch (InstantiationException e) {
throw new InternalError(e.toString());
} catch (InvocationTargetException e) {
throw new InternalError(e.toString());
}
}
// In the normal case, the proxy object's method implementations will call
// this method (it usually calls the ProviderSkeleton's version). That
// method uses the passed 'method' object to lookup the associated
// 'ProbeSkeleton' and calls uncheckedTrigger() on that probe to cause the
// probe to fire. DTrace probes are different in that the proxy class's
// methods are immediately overridden with native code to fire the probe
// directly. So this method should never get invoked. We also wire up the
// DTraceProbe.uncheckedTrigger() method to call the proxy method instead
// of doing the work itself.
protected void triggerProbe(Method method, Object[] args) {
assert false : "This method should have been overridden by the JVM";
}
public String getProviderName() {
return super.getProviderName();
}
String getModuleName() {
return getAnnotationString(
providerType, ModuleName.class, DEFAULT_MODULE);
}
static String getProbeName(Method method) {
return getAnnotationString(
method, ProbeName.class, method.getName());
}
static String getFunctionName(Method method) {
return getAnnotationString(
method, FunctionName.class, DEFAULT_FUNCTION);
}
DTraceProbe[] getProbes() {
return probes.values().toArray(new DTraceProbe[0]);
}
StabilityLevel getNameStabilityFor(Class<? extends Annotation> type) {
Attributes attrs = (Attributes)getAnnotationValue(
providerType, type, "value", null);
if (attrs == null) {
return StabilityLevel.PRIVATE;
} else {
return attrs.name();
}
}
StabilityLevel getDataStabilityFor(Class<? extends Annotation> type) {
Attributes attrs = (Attributes)getAnnotationValue(
providerType, type, "value", null);
if (attrs == null) {
return StabilityLevel.PRIVATE;
} else {
return attrs.data();
}
}
DependencyClass getDependencyClassFor(Class<? extends Annotation> type) {
Attributes attrs = (Attributes)getAnnotationValue(
providerType, type, "value", null);
if (attrs == null) {
return DependencyClass.UNKNOWN;
} else {
return attrs.dependency();
}
}
}