/* * Copyright (c) 2018, 2019, 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. */ package jdk.tools.jaotc; import java.io.BufferedReader; import java.io.File; import java.io.InputStream; import java.io.InputStreamReader; import java.nio.file.Files; import java.nio.file.Path; import java.nio.file.Paths; import java.util.List; import java.util.stream.Stream; final class Linker { private final Options options; private String objectFileName; private String libraryFileName; private String linkerCmd; String objFile() { return objectFileName; } String libFile() { return libraryFileName; } private static Stream getLines(InputStream stream) { return new BufferedReader(new InputStreamReader(stream)).lines(); } private static String getString(InputStream stream) { Stream lines = getLines(stream); StringBuilder sb = new StringBuilder(); lines.iterator().forEachRemaining(e -> sb.append(e)); return sb.toString(); } Linker(Main main) throws Exception { this.options = main.options; String name = options.outputName; objectFileName = name; libraryFileName = name; if (options.linkerpath != null && !(new File(options.linkerpath).exists())) { throw new InternalError("Invalid linker path: " + options.linkerpath); } String linkerPath; String linkerCheck; switch (options.osName) { case "Linux": if (name.endsWith(".so")) { objectFileName = name.substring(0, name.length() - ".so".length()); } objectFileName = objectFileName + ".o"; linkerPath = (options.linkerpath != null) ? options.linkerpath : "ld"; linkerCmd = linkerPath + " -shared -z noexecstack -o " + libraryFileName + " " + objectFileName; linkerCheck = linkerPath + " -v"; break; case "Mac OS X": if (name.endsWith(".dylib")) { objectFileName = name.substring(0, name.length() - ".dylib".length()); } objectFileName = objectFileName + ".o"; linkerPath = (options.linkerpath != null) ? options.linkerpath : "ld"; linkerCmd = linkerPath + " -dylib -o " + libraryFileName + " " + objectFileName; linkerCheck = linkerPath + " -v"; break; default: if (options.osName.startsWith("Windows")) { if (name.endsWith(".dll")) { objectFileName = name.substring(0, name.length() - ".dll".length()); } objectFileName = objectFileName + ".obj"; linkerPath = (options.linkerpath != null) ? options.linkerpath : getWindowsLinkPath(); if (linkerPath == null) { throw new InternalError("Can't locate Microsoft Visual Studio amd64 link.exe"); } linkerCmd = linkerPath + " /DLL /OPT:NOREF /NOLOGO /NOENTRY" + " /OUT:" + libraryFileName + " " + objectFileName; linkerCheck = null; // link.exe presence is verified already break; } else { throw new InternalError("Unsupported platform: " + options.osName); } } // Check linker presence on platforms by printing its version if (linkerCheck != null) { Process p = Runtime.getRuntime().exec(linkerCheck); final int exitCode = p.waitFor(); if (exitCode != 0) { throw new InternalError(getString(p.getErrorStream())); } } } void link() throws Exception { Process p = Runtime.getRuntime().exec(linkerCmd); final int exitCode = p.waitFor(); if (exitCode != 0) { String errorMessage = getString(p.getErrorStream()); if (errorMessage.isEmpty()) { errorMessage = getString(p.getInputStream()); } throw new InternalError(errorMessage); } File objFile = new File(objectFileName); boolean keepObjFile = Boolean.parseBoolean(System.getProperty("aot.keep.objFile", "false")); if (objFile.exists() && !keepObjFile) { if (!objFile.delete()) { throw new InternalError("Failed to delete " + objectFileName + " file"); } } // Make non-executable for all. File libFile = new File(libraryFileName); if (libFile.exists() && !options.osName.startsWith("Windows")) { if (!libFile.setExecutable(false, false)) { throw new InternalError("Failed to change attribute for " + libraryFileName + " file"); } } } /** * Search for Visual Studio link.exe Search Order is: VS2017+, VS2013, VS2015, VS2012. */ private static String getWindowsLinkPath() throws Exception { try { Path vc141NewerLinker = getVC141AndNewerLinker(); if (vc141NewerLinker != null) { return vc141NewerLinker.toString(); } } catch (Exception e) { e.printStackTrace(); } String link = "\\VC\\bin\\amd64\\link.exe"; /** * First try searching the paths pointed to by the VS environment variables. */ for (VSVERSIONS vs : VSVERSIONS.values()) { String vspath = System.getenv(vs.getEnvVariable()); if (vspath != null) { File commonTools = new File(vspath); File vsRoot = commonTools.getParentFile().getParentFile(); File linkPath = new File(vsRoot, link); if (linkPath.exists()) { return linkPath.getPath(); } } } /** * If we didn't find via the VS environment variables, try the well known paths */ for (VSVERSIONS vs : VSVERSIONS.values()) { String wkp = vs.getWellKnownPath(); if (new File(wkp).exists()) { return wkp; } } return null; } private static Path getVC141AndNewerLinker() throws Exception { String programFilesX86 = System.getenv("ProgramFiles(x86)"); if (programFilesX86 == null) { throw new InternalError("Could not read the ProgramFiles(x86) environment variable"); } Path vswhere = Paths.get(programFilesX86 + "\\Microsoft Visual Studio\\Installer\\vswhere.exe"); if (!Files.exists(vswhere)) { return null; } ProcessBuilder processBuilder = new ProcessBuilder(vswhere.toString(), "-requires", "Microsoft.VisualStudio.Component.VC.Tools.x86.x64", "-property", "installationPath", "-latest"); processBuilder.redirectOutput(ProcessBuilder.Redirect.PIPE); processBuilder.redirectError(ProcessBuilder.Redirect.PIPE); Process process = processBuilder.start(); final int exitCode = process.waitFor(); if (exitCode != 0) { String errorMessage = getString(process.getErrorStream()); if (errorMessage.isEmpty()) { errorMessage = getString(process.getInputStream()); } throw new InternalError(errorMessage); } String installationPath = getLines(process.getInputStream()).findFirst().orElseThrow(() -> new InternalError("Unexpected empty output from vswhere")); Path vcToolsVersionFilePath = Paths.get(installationPath, "VC\\Auxiliary\\Build\\Microsoft.VCToolsVersion.default.txt"); List vcToolsVersionFileLines = Files.readAllLines(vcToolsVersionFilePath); if (vcToolsVersionFileLines.isEmpty()) { throw new InternalError(vcToolsVersionFilePath.toString() + " is empty"); } String vcToolsVersion = vcToolsVersionFileLines.get(0); Path linkPath = Paths.get(installationPath, "VC\\Tools\\MSVC", vcToolsVersion, "bin\\Hostx64\\x64\\link.exe"); if (!Files.exists(linkPath)) { throw new InternalError("Linker at path " + linkPath.toString() + " does not exist"); } return linkPath; } // @formatter:off (workaround for Eclipse formatting bug) enum VSVERSIONS { VS2013("VS120COMNTOOLS", "C:\\Program Files (x86)\\Microsoft Visual Studio 12.0\\VC\\bin\\amd64\\link.exe"), VS2015("VS140COMNTOOLS", "C:\\Program Files (x86)\\Microsoft Visual Studio 14.0\\VC\\bin\\amd64\\link.exe"), VS2012("VS110COMNTOOLS", "C:\\Program Files (x86)\\Microsoft Visual Studio 11.0\\VC\\bin\\amd64\\link.exe"); private final String envvariable; private final String wkp; VSVERSIONS(String envvariable, String wellknownpath) { this.envvariable = envvariable; this.wkp = wellknownpath; } String getEnvVariable() { return envvariable; } String getWellKnownPath() { return wkp; } } // @formatter:on }