com.sun.btrace.spi.BaseBTraceCompiler Maven / Gradle / Ivy

Go to download
/*
 * 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, task.isUnsafe());
        this.task = task;
    }
    
    @Override
    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 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.indexOf("file://") > -1 && jarPath.indexOf("file:////") == -1) {  //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;
    }
}