/*
* Copyright (c) 2007, 2011, 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 com.sun.btrace.spi;
import com.sun.btrace.api.BTraceCompiler;
import com.sun.btrace.api.BTraceTask;
import com.sun.tools.attach.AttachNotSupportedException;
import com.sun.tools.attach.VirtualMachine;
import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.Writer;
import java.net.MalformedURLException;
import java.net.URISyntaxException;
import java.net.URL;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
/**
*
* @author Jaroslav Bachorik
*/
public class BaseBTraceCompiler extends BTraceCompiler {
final private static Logger LOGGER = Logger.getLogger(BTraceCompiler.class.getName());
final private static Pattern classNamePattern = Pattern.compile("@BTrace\\s*.+?\\s*class\\s*(.*?)\\s+\\{", Pattern.MULTILINE | Pattern.DOTALL | Pattern.UNIX_LINES);
final private com.sun.btrace.compiler.Compiler c;
final private BTraceTask task;
public BaseBTraceCompiler(BTraceTask task) {
c = new com.sun.btrace.compiler.Compiler(null);
this.task = task;
}
@Override
@SuppressWarnings("DefaultCharset")
public byte[] compile(String source, String classPath, Writer errorWriter) {
try {
Matcher matcher = classNamePattern.matcher(source);
if (matcher.find()) {
if (errorWriter == null) {
errorWriter = new PrintWriter(System.out);
}
String fileName = matcher.group(1) + ".java";
String completeCP = getToolsJarPath() + File.pathSeparator + getClientJarPath() + File.pathSeparator + classPath;
Map<String, byte[]> compilationMap =c.compile(fileName, source, errorWriter, ".", completeCP);
if (compilationMap != null) {
return compilationMap.values().iterator().next();
}
}
} catch (Exception e) {
LOGGER.log(Level.SEVERE, null, e);
}
return new byte[0];
}
@Override
public String getAgentJarPath() {
return getJarBaseDir() + File.separatorChar + "btrace-agent.jar";
}
@Override
public String getClientJarPath() {
return getJarBaseDir() + File.separatorChar + "btrace-client.jar";
}
final private Object toolsJarLock = new Object();
private String toolsJar = null;
@Override
public String getToolsJarPath() {
synchronized(toolsJarLock) {
if (toolsJar == null) {
String javaHome = getJavaHome(task);
// no java home found for the target app; fallback to tools.jar provided from VisualVM
if (javaHome == null) {
File toolsJarFile = getContainingJar("com/sun/tools/javac/Main.class"); // NOI18N
if (toolsJarFile != null) {
toolsJar = toolsJarFile.getAbsolutePath();
}
} else {
File toolsJarFile = new File(javaHome + "/lib/tools.jar"); // NOI18N
if (toolsJarFile.exists()) {
toolsJar = toolsJarFile.getAbsolutePath();
} else {
toolsJarFile = new File(javaHome + "/../lib/tools.jar"); // NOI18N
if (toolsJarFile.exists()) {
toolsJar = toolsJarFile.getAbsolutePath();
} else {
toolsJarFile = new File(javaHome + "/../Classes/classes.jar"); // NOI18N
if (toolsJarFile.exists()) {
toolsJar = toolsJarFile.getAbsolutePath();
} else {
toolsJar="";
}
}
}
}
}
return toolsJar;
}
}
final private Object jarBaseDirLock = new Object();
private String jarBaseDir = null;
private String getJarBaseDir() {
synchronized(jarBaseDirLock) {
if (jarBaseDir == null) {
File f = getContainingJar("com/sun/btrace/BTraceRuntime.class");
while (f != null && (!f.exists() || !f.isDirectory())) {
f = f.getParentFile();
}
if (f != null) {
jarBaseDir = f.getAbsolutePath();
}
}
return jarBaseDir;
}
}
private File getContainingJar(String clz) {
File jarFile;
URL url = getClass().getClassLoader().getResource(clz);
if ("jar".equals(url.getProtocol())) { //NOI18N
String path = url.getPath();
int index = path.indexOf("!/"); //NOI18N
if (index >= 0) {
try {
String jarPath = path.substring(0, index);
if (jarPath.contains("file://") && !jarPath.contains("file:////")) { //NOI18N
/* Replace because JDK application classloader wrongly recognizes UNC paths. */
jarPath = jarPath.replaceFirst("file://", "file:////"); //NOI18N
}
url = new URL(jarPath);
} catch (MalformedURLException mue) {
throw new RuntimeException(mue);
}
}
}
try {
jarFile = new File(url.toURI());
} catch (URISyntaxException ex) {
throw new RuntimeException(ex);
}
assert jarFile.exists();
return jarFile;
}
private static String getJavaHome(BTraceTask btt) {
try {
VirtualMachine vm = VirtualMachine.attach(String.valueOf(btt.getPid()));
return vm.getSystemProperties().getProperty("java.home");
} catch (AttachNotSupportedException e) {
LOGGER.log(Level.SEVERE, null, e);
} catch (IOException e) {
LOGGER.log(Level.SEVERE, null, e);
}
return null;
}
}