test/hotspot/jtreg/vmTestbase/PropertyResolvingWrapper.java

/*
 * Copyright (c) 2017, 2018, 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.
 *
 * 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.
 */

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.lang.reflect.InvocationTargetException;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;

/**
 * Replaces all {@code ${<X>}} with value of corresponding property({@code X}),
 * resulting string is handled similarly to {@code @run main} in jtreg.
 * In other words, {@code main} of first token will be executed with the rest
 * tokens as arguments.
 *
 * If one of properties can't be resolved, {@link Error} will be thrown.
 */
public class PropertyResolvingWrapper {
    private static final Properties properties;
    static {
        Properties p = System.getProperties();
        String name = p.getProperty("os.name");
        String arch = p.getProperty("os.arch");
        String family;
        String simple_arch;

        // copy from jtreg/src/share/classes/com/sun/javatest/regtest/config/OS.java
        if (name.startsWith("AIX"))
            family = "aix";
        else if (name.startsWith("Linux"))
            family = "linux";
        else if (name.startsWith("Mac") || name.startsWith("Darwin"))
            family = "mac";
        else if (name.startsWith("OS400") || name.startsWith("OS/400") )
            family = "os400";
        else if (name.startsWith("SunOS") || name.startsWith("Solaris"))
            family = "solaris";
        else if (name.startsWith("Windows"))
            family = "windows";
        else
            family = name.replaceFirst("^([^ ]+).*", "$1"); // use first word of name

        if (arch.contains("64")
                 && !arch.equals("ia64")
                 && !arch.equals("ppc64")
                 && !arch.equals("ppc64le")
                 && !arch.equals("zArch_64")
                 && !arch.equals("aarch64"))
             simple_arch = "x64";
        else if (arch.contains("86"))
            simple_arch = "i586";
        else if (arch.equals("ppc") || arch.equals("powerpc"))
            simple_arch = "ppc";
        else if (arch.equals("s390x") || arch.equals("zArch_64"))
            simple_arch = "s390x";
        else
            simple_arch = arch;

        p.setProperty("os.family", family);
        p.setProperty("os.simpleArch", simple_arch);
        properties = p;
    }

    public static void main(String[] args) throws Throwable {
        List<String> command = new ArrayList<>(args.length);
        for (int i = 0; i < args.length; ++i) {
            StringBuilder arg = new StringBuilder(args[i]);
            while (i < args.length - 1
                    && (arg.chars()
                       .filter(c -> c == '"')
                       .count() % 2) != 0) {
                arg.append(" ")
                   .append(args[++i]);
            }
            command.add(eval(arg.toString()));
        }
        System.out.println("run " + command);
        try {
            Class.forName(command.remove(0))
                 .getMethod("main", String[].class)
                 .invoke(null, new Object[]{command.toArray(new String[0])});
        } catch (InvocationTargetException e) {
           Throwable t = e.getCause();
           t = t != null ? t : e;
           throw t;
        }
    }

    private static String eval(String string) {
        int index;
        int current = 0;
        StringBuilder result = new StringBuilder();
        while (current < string.length() && (index = string.indexOf("${", current)) >= 0) {
            result.append(string.substring(current, index));
            int endName = string.indexOf('}', index);
            current = endName + 1;
            String name = string.substring(index + 2, endName);
            String value = properties.getProperty(name);
            if (value == null) {
                throw new Error("can't find property " + name);
            }
            result.append(value);
        }
        if (current < string.length()) {
            result.append(string.substring(current));
        }
        int length = result.length();

        if (length > 1 && result.charAt(0) == '"' && result.charAt(length - 1) == '"') {
            result.deleteCharAt(length - 1);
            result.deleteCharAt(0);
        }
        return result.toString();
    }
}