All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • org.wildfly.galleon.plugin.server.ForkedEmbeddedUtil Maven / Gradle / Ivy

    There is a newer version: 7.3.1.Final
    Show newest version
    /*
 * Copyright 2016-2018 Red Hat, Inc. and/or its affiliates
 * and other contributors as indicated by the @author tags.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.wildfly.galleon.plugin.server;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.URL;
import java.net.URLClassLoader;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.jboss.galleon.Errors;
import org.jboss.galleon.ProvisioningException;
import org.jboss.galleon.util.CollectionUtils;
import org.jboss.galleon.util.IoUtils;


/**
 *
 * @author Alexey Loubyansky
 */
public class ForkedEmbeddedUtil {

    public static final String FORKED_EMBEDDED_ERROR_START = "Forked embedded process has failed with the following error:";

    public interface ForkCallback {

        void forkedForEmbedded(String... args) throws ProvisioningException;

        default void forkedEmbeddedMessage(String msg) {}

        default void forkedEmbeddedDone(String... args) throws ProvisioningException {}
    }

    private static int javaVersion = -1;
    private static String javaHome;
    private static String javaCmd;

    private static int getJavaVersion() {
        if (javaVersion < 0) {
            try {
                String vmVersionStr = System.getProperty("java.specification.version", null);
                Matcher matcher = Pattern.compile("^(?:1\\.)?(\\d+)$").matcher(vmVersionStr); // match 1. or 
                if (matcher.find()) {
                    javaVersion = Integer.valueOf(matcher.group(1));
                } else {
                    throw new RuntimeException("Unknown version of jvm " + vmVersionStr);
                }
            } catch (Exception e) {
                javaVersion = 8;
            }
        }
        return javaVersion;
    }

    private static String getJavaHome() {
        return javaHome == null ? javaHome = System.getProperty("java.home") : javaHome;
    }

    private static String getJavaCmd() {
        return javaCmd == null ? javaCmd = Paths.get(getJavaHome()).resolve("bin").resolve("java").toString() : javaCmd;
    }

    public static void fork(ForkCallback callback, String... args) throws ProvisioningException {
        final Path props = storeSystemProps();
        try {
            fork(callback, props, args);
            callback.forkedEmbeddedDone(args);
        } finally {
            IoUtils.recursiveDelete(props);
        }
    }

    public static void fork(ForkCallback callback, Path props, String... args) throws ProvisioningException {
        // prepare the classpath
        final StringBuilder cp = new StringBuilder();
        collectCpUrls(getJavaHome(), Thread.currentThread().getContextClassLoader(), cp);

        final List argsList = new ArrayList<>(8 + args.length);
        argsList.add(getJavaCmd());
        argsList.add("-server");
        if (getJavaVersion() >= 11) {
            argsList.add("--add-modules=java.se");
        }
        argsList.add("-cp");
        argsList.add(cp.toString());
        argsList.add(ForkedEmbeddedUtil.class.getName());
        argsList.add(props.toString());
        argsList.add(callback.getClass().getName());
        for(String arg : args) {
            argsList.add(arg);
        }

        final Process p;
        try {
            p = new ProcessBuilder(argsList).redirectErrorStream(true).start();
        } catch (IOException e) {
            throw new ProvisioningException("Failed to start a feature spec reading process", e);
        }

        List trace = null;
        try (BufferedReader reader = new BufferedReader(new InputStreamReader(p.getInputStream(), StandardCharsets.UTF_8))) {
            String line = reader.readLine();
            while (line != null) {
                callback.forkedEmbeddedMessage(line);
                if(trace != null) {
                    trace.add(line);
                } else if(FORKED_EMBEDDED_ERROR_START.equals(line)) {
                    trace = new ArrayList<>();
                }
                line = reader.readLine();
            }
            if (p.isAlive()) {
                try {
                    p.waitFor();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }

            int exitCode = p.exitValue();
            if (exitCode != 0) {
                Throwable t = null;
                if(trace != null) {
                    t = parseException(trace, 0);
                    if(t == null) {
                        System.out.println(FORKED_EMBEDDED_ERROR_START);
                        for(String l : trace) {
                            System.out.println(l);
                        }
                    }
                }
                throw new ProvisioningException("Forked embedded process has failed", t);
            }
        } catch (IOException e) {
            throw new ProvisioningException("Forked embedded process has failed", e);
        }
    }

    public static Path storeSystemProps() throws ProvisioningException {
        final Path props;
        try {
            props = Files.createTempFile("wfgp", "sysprops");
        } catch (IOException e) {
            throw new ProvisioningException("Failed to create a tmp file", e);
        }
        try (BufferedWriter writer = Files.newBufferedWriter(props)) {
            System.getProperties().store(writer, "");
        } catch (IOException e) {
            throw new ProvisioningException(Errors.writeFile(props), e);
        }
        return props;
    }

    public static void main(String... args) {
        try{
            if(args.length < 2) {
                throw new IllegalStateException("Expected at least two arguments but got " + Arrays.asList(args));
            }

            // set system properties
            setSystemProps(args[0]);

            Class cls;
            try {
                cls = Thread.currentThread().getContextClassLoader().loadClass(args[1]);
            } catch (ClassNotFoundException e) {
                throw new ProvisioningException("Failed to locate the target class " + args[1], e);
            }
        /*if(!ForkCallback.class.isAssignableFrom(cls)) {
            throw new ProvisioningException(args[1] + " does not implement " + ForkCallback.class.getName());
        }*/

            Object o;
            try {
                o = cls.newInstance();
            } catch (Exception e) {
                throw new ProvisioningException("Failed to instantiate " + args[1], e);
            }
            ((ForkCallback)o).forkedForEmbedded(args.length == 2 ? new String[0] : Arrays.copyOfRange(args, 2, args.length));

        } catch (Throwable t) {
            System.err.println(FORKED_EMBEDDED_ERROR_START);
            final StringBuilder buf = new StringBuilder();
            while(t != null) {
                buf.setLength(0);
                buf.append(t.getClass().getName());
                if(t.getMessage() != null) {
                    buf.append(": ").append(t.getMessage());
                }
                System.err.println(buf.toString());
                for(StackTraceElement e : t.getStackTrace()) {
                    buf.setLength(0);
                    buf.append("\tat ").append(e.toString());
                    System.err.println(buf.toString());
                }
                t = t.getCause();
            }
            System.exit(1);
        }
    }

    private static void setSystemProps(String path) throws ProvisioningException {
        final Path props = Paths.get(path);
        if(!Files.exists(props)) {
            throw new ProvisioningException(Errors.pathDoesNotExist(props));
        }
        final Properties tmp = new Properties();
        try(BufferedReader reader = Files.newBufferedReader(props)) {
            tmp.load(reader);
        } catch (IOException e) {
            throw new ProvisioningException(Errors.readFile(props));
        }
        for(Map.Entry prop : tmp.entrySet()) {
            final String current = System.getProperty(prop.getKey().toString());
            if(current != null) {
                // do not override the default properties
                continue;
            }
            System.setProperty(prop.getKey().toString(), prop.getValue().toString());
        }
    }

    private static void collectCpUrls(String javaHome, ClassLoader cl, StringBuilder buf) {
        final ClassLoader parentCl = cl.getParent();
        if(parentCl != null) {
            collectCpUrls(javaHome, cl.getParent(), buf);
        }
        if (cl instanceof URLClassLoader) {
            for (URL url : ((URLClassLoader)cl).getURLs()) {
                final String file = url.getFile();
                if(file.startsWith(javaHome)) {
                    continue;
                }
                if (buf.length() > 0) {
                    buf.append(File.pathSeparatorChar);
                }
                buf.append(file);
            }
        }
    }

    private static Throwable parseException(List trace, int offset) {
        final String classAndMsg = trace.get(offset);
        String className = null;
        List> ctorArgTypes = Collections.emptyList();
        List ctorArgs = Collections.emptyList();
        int i = classAndMsg.indexOf(':');
        if(i < 0) {
            className = classAndMsg;
        } else {
            className = classAndMsg.substring(0, i);
            ctorArgTypes = CollectionUtils.add(ctorArgTypes, String.class);
            ctorArgs = CollectionUtils.add(ctorArgs, classAndMsg.substring(i + 1).trim());
        }

        final List stack = new ArrayList<>(trace.size() - offset - 1);
        for(i = offset + 1; i < trace.size(); ++i) {
            final StackTraceElement ste = stackTraceElementFromString(trace.get(i));
            if(ste == null) {
                final Throwable cause = parseException(trace, i);
                if(cause == null) {
                    return null;
                }
                ctorArgTypes = CollectionUtils.add(ctorArgTypes, Throwable.class);
                ctorArgs = CollectionUtils.add(ctorArgs, cause);
                break;
            }
            stack.add(ste);
        }

        final Throwable t;
        try {
            final Class excClass = Thread.currentThread().getContextClassLoader().loadClass(className);
            if(ctorArgTypes.isEmpty()) {
                t = (Throwable) excClass.newInstance();
            } else {
                t = (Throwable) excClass.getConstructor(ctorArgTypes.toArray(new Class[ctorArgTypes.size()]))
                        .newInstance(ctorArgs.toArray(new Object[ctorArgs.size()]));
            }
        } catch (Throwable e) {
            return null;
        }
        t.setStackTrace(stack.toArray(new StackTraceElement[stack.size()]));
        return t;
    }

    private static StackTraceElement stackTraceElementFromString(String line) {
        int i = line.length() - 1;
        if(i < 0) {
            return null;
        }
        if(line.charAt(i) != ')') {
            return null;
        }
        int pos = line.lastIndexOf(':');
        if(pos < 0) {
            return null;
        }
        final int lineNumber = Integer.parseInt(line.substring(pos + 1, i));
        i = pos;
        pos = line.lastIndexOf('(');
        if(pos < 0) {
            return null;
        }
        final String file = line.substring(pos + 1, i);
        i = pos;
        while(--pos >= 0) {
            if(line.charAt(pos) == '.') {
                break;
            }
        }
        if(pos < 0) {
            return null;
        }
        final String method = line.substring(pos + 1, i);
        i = pos;
        while(--pos >= 0) {
            if(Character.isWhitespace(line.charAt(pos))) {
                break;
            }
        }
        if(pos < 0) {
            return null;
        }
        return new StackTraceElement(line.substring(pos + 1, i), method, file, lineNumber);
    }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy