src/jdk.aot/share/classes/jdk.tools.jaotc/src/jdk/tools/jaotc/Linker.java - platform/libcore - Git at Google

 /*
  * 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<String> getLines(InputStream stream) {
         return new BufferedReader(new InputStreamReader(stream)).lines();
     }

     private static String getString(InputStream stream) {
         Stream<String> 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 "SunOS":
                 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 -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(main);
                     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()));
             }
         }

         main.printer.printlnVerbose("Found linker: " + linkerPath);
     }

     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(Main main) throws Exception {
         try {
             Path vc141NewerLinker = getVC141AndNewerLinker();
             if (vc141NewerLinker != null) {
                 return vc141NewerLinker.toString();
             }
         } catch (Exception e) {
             main.printer.printlnVerbose("Could not find VC14 or newer version of linker: " + e.getMessage());
             if (main.options.debug) {
                 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 IllegalStateException("Could not read the ProgramFiles(x86) environment variable");
         }
         String vswherePath = programFilesX86 + "\\Microsoft Visual Studio\\Installer\\vswhere.exe";
         Path vswhere = Paths.get(vswherePath);
         if (!Files.exists(vswhere)) {
             throw new IllegalStateException("Could not find " + vswherePath);
         }

         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 IllegalStateException("vswhere error: " + errorMessage);
         }

         String installationPath = getLines(process.getInputStream()).findFirst().orElseThrow(() -> new IllegalStateException("Unexpected empty output from vswhere"));
         Path vcToolsVersionFilePath = Paths.get(installationPath, "VC\\Auxiliary\\Build\\Microsoft.VCToolsVersion.default.txt");
         List<String> vcToolsVersionFileLines = Files.readAllLines(vcToolsVersionFilePath);
         if (vcToolsVersionFileLines.isEmpty()) {
             throw new IllegalStateException(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 IllegalStateException("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
 }
	/*
	* 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<String> getLines(InputStream stream) {
	return new BufferedReader(new InputStreamReader(stream)).lines();
	}

	private static String getString(InputStream stream) {
	Stream<String> 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 "SunOS":
	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 -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(main);
	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()));
	}
	}

	main.printer.printlnVerbose("Found linker: " + linkerPath);
	}

	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(Main main) throws Exception {
	try {
	Path vc141NewerLinker = getVC141AndNewerLinker();
	if (vc141NewerLinker != null) {
	return vc141NewerLinker.toString();
	}
	} catch (Exception e) {
	main.printer.printlnVerbose("Could not find VC14 or newer version of linker: " + e.getMessage());
	if (main.options.debug) {
	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 IllegalStateException("Could not read the ProgramFiles(x86) environment variable");
	}
	String vswherePath = programFilesX86 + "\\Microsoft Visual Studio\\Installer\\vswhere.exe";
	Path vswhere = Paths.get(vswherePath);
	if (!Files.exists(vswhere)) {
	throw new IllegalStateException("Could not find " + vswherePath);
	}

	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 IllegalStateException("vswhere error: " + errorMessage);
	}

	String installationPath = getLines(process.getInputStream()).findFirst().orElseThrow(() -> new IllegalStateException("Unexpected empty output from vswhere"));
	Path vcToolsVersionFilePath = Paths.get(installationPath, "VC\\Auxiliary\\Build\\Microsoft.VCToolsVersion.default.txt");
	List<String> vcToolsVersionFileLines = Files.readAllLines(vcToolsVersionFilePath);
	if (vcToolsVersionFileLines.isEmpty()) {
	throw new IllegalStateException(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 IllegalStateException("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
	}