+/* Copyright (c) 2014. The SimGrid Team.
+ * All rights reserved. */
+
+/* This program is free software; you can redistribute it and/or modify it
+ * under the terms of the license (GNU LGPL) which comes with this package. */
+
package org.simgrid;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.File;
+import java.nio.file.Files;
+import java.nio.file.Path;
public final class NativeLib {
+ /* Statically load the library which contains all native functions used in here */
+ static private boolean isNativeInited = false;
+ public static void nativeInit() {
+ if (isNativeInited)
+ return;
+
+ if (System.getProperty("os.name").toLowerCase().startsWith("win"))
+ NativeLib.nativeInit("winpthread-1");
- public static String getPath() {
- String prefix = "NATIVE";
- String os = System.getProperty("os.name");
- String arch = System.getProperty("os.arch");
-
- if (os.toLowerCase().startsWith("^win"))
- os = "Windows";
- else if (os.contains("OS X"))
- os = "Darwin";
-
- if (arch.matches("^i[3-6]86$"))
- arch = "x86";
- else if (arch.equalsIgnoreCase("amd64"))
- arch = "x86_64";
-
- os = os.replace(' ', '_');
- arch = arch.replace(' ', '_');
-
- return prefix + "/" + os + "/" + arch + "/";
- }
- public static void nativeInit(String name) {
- try {
- /* prefer the version on disk, if existing */
- System.loadLibrary(name);
- } catch (UnsatisfiedLinkError e) {
- /* If not found, unpack the one bundled into the jar file and use it */
- loadLib(name);
- }
- }
-
- private static void loadLib (String name) {
- String Path = NativeLib.getPath();
-
- String filename=name;
- InputStream in = NativeLib.class.getClassLoader().getResourceAsStream(Path+filename);
-
- if (in == null) {
- filename = "lib"+name+".so";
- in = NativeLib.class.getClassLoader().getResourceAsStream(Path+filename);
- }
- if (in == null) {
- filename = name+".dll";
- in = NativeLib.class.getClassLoader().getResourceAsStream(Path+filename);
- }
- if (in == null) {
- filename = "lib"+name+".dll";
- in = NativeLib.class.getClassLoader().getResourceAsStream(Path+filename);
- }
- if (in == null) {
- filename = "lib"+name+".dylib";
- in = NativeLib.class.getClassLoader().getResourceAsStream(Path+filename);
- }
- if (in == null) {
- throw new RuntimeException("Cannot find library "+name+" in path "+Path+". Sorry, but this jar does not seem to be usable on your machine.");
+ try {
+ NativeLib.nativeInit("boost_context");
+ } catch (Exception e) {/* Dont care */}
+ NativeLib.nativeInit("simgrid");
+ NativeLib.nativeInit("simgrid-java");
+ isNativeInited = true;
}
- try {
- // We must write the lib onto the disk before loading it -- stupid operating systems
- File fileOut = new File(filename);
- fileOut = File.createTempFile(name+"-", ".tmp");
- // don't leak the file on disk, but remove it on JVM shutdown
- Runtime.getRuntime().addShutdownHook(new Thread(new FileCleaner(fileOut.getAbsolutePath())));
- OutputStream out = new FileOutputStream(fileOut);
-
- /* copy the library in position */
- byte[] buffer = new byte[4096];
- int bytes_read;
- while ((bytes_read = in.read(buffer)) != -1) // Read until EOF
- out.write(buffer, 0, bytes_read);
-
- /* close all file descriptors, and load that shit */
- in.close();
- out.close();
- System.load(fileOut.getAbsolutePath());
-
- } catch (Exception e) {
- System.err.println("Cannot load the bindings to the "+name+" library: ");
- e.printStackTrace();
- System.err.println("This jar file does not seem to fit your system, sorry");
- System.exit(1);
+
+ public static void nativeInit(String name) {
+ try {
+ /* Prefer the version of the library bundled into the jar file and use it */
+ loadLib(name);
+ } catch (SimGridLibNotFoundException e) {
+ /* If not found, try to see if we can find a version on disk */
+ try {
+ System.loadLibrary(name);
+ } catch (UnsatisfiedLinkError e2) {
+ if (! name.equals("boost_context")) {
+ System.err.println("Cannot load the bindings to the "+name+" library in path "+getPath());
+ e.printStackTrace();
+ System.err.println("This jar file does not seem to fit your system, and I cannot find an installation of SimGrid.");
+ System.exit(1);
+ }
+ }
+ }
+ }
+
+ public static String getPath() {
+ // Inspiration: https://github.com/xerial/snappy-java/blob/develop/src/main/java/org/xerial/snappy/OSInfo.java
+ String prefix = "NATIVE";
+ String os = System.getProperty("os.name");
+ String arch = System.getProperty("os.arch");
+
+ if (arch.matches("^i[3-6]86$"))
+ arch = "x86";
+ else if (arch.equalsIgnoreCase("x86_64"))
+ arch = "amd64";
+ else if (arch.equalsIgnoreCase("AMD64"))
+ arch = "amd64";
+
+ if (os.toLowerCase().startsWith("win")){
+ os = "Windows";
+ } else if (os.contains("OS X"))
+ os = "Darwin";
+
+ os = os.replace(' ', '_');
+ arch = arch.replace(' ', '_');
+
+ return prefix + "/" + os + "/" + arch + "/";
}
- }
-
+ static Path tempDir = null;
+ private static void loadLib (String name) throws SimGridLibNotFoundException {
+ String Path = NativeLib.getPath();
+
+ String filename=name;
+ InputStream in = NativeLib.class.getClassLoader().getResourceAsStream(Path+filename);
+
+ if (in == null) {
+ filename = "lib"+name+".so";
+ in = NativeLib.class.getClassLoader().getResourceAsStream(Path+filename);
+ }
+ if (in == null) {
+ filename = name+".dll";
+ in = NativeLib.class.getClassLoader().getResourceAsStream(Path+filename);
+ }
+ if (in == null) {
+ filename = "lib"+name+".dll";
+ in = NativeLib.class.getClassLoader().getResourceAsStream(Path+filename);
+ }
+ if (in == null) {
+ filename = "lib"+name+".dylib";
+ in = NativeLib.class.getClassLoader().getResourceAsStream(Path+filename);
+ }
+ if (in == null) {
+ throw new SimGridLibNotFoundException("Cannot find library "+name+" in path "+Path+". Sorry, but this jar does not seem to be usable on your machine.");
+ }
+ try {
+ // We must write the lib onto the disk before loading it -- stupid operating systems
+ if (tempDir == null) {
+ tempDir = Files.createTempDirectory("simgrid-java");
+ // don't leak the files on disk, but remove it on JVM shutdown
+ Runtime.getRuntime().addShutdownHook(new Thread(new FileCleaner(tempDir.toFile())));
+ }
+ File fileOut = new File(tempDir.toFile().getAbsolutePath() + File.separator + filename);
+
+ /* copy the library in position */
+ OutputStream out = new FileOutputStream(fileOut);
+ byte[] buffer = new byte[4096];
+ int bytes_read;
+ while ((bytes_read = in.read(buffer)) != -1) // Read until EOF
+ out.write(buffer, 0, bytes_read);
+
+ /* close all file descriptors, and load that shit */
+ in.close();
+ out.close();
+ System.load(fileOut.getAbsolutePath());
+ } catch (Exception e) {
+ System.err.println("Error while extracting the native library from the jar: ");
+ e.printStackTrace();
+ throw new SimGridLibNotFoundException("Cannot load the bindings to the "+name+" library in path "+getPath(), e);
+ }
+ }
+
/* A hackish mechanism used to remove the file containing our library when the JVM shuts down */
private static class FileCleaner implements Runnable {
- private String target;
- public FileCleaner(String name) {
- target = name;
+ private File dir;
+ public FileCleaner(File dir) {
+ this.dir = dir;
}
- public void run() {
- try {
- new File(target).delete();
- } catch(Exception e) {
- System.out.println("Unable to clean temporary file "+target+" during shutdown.");
- e.printStackTrace();
- }
- }
+ public void run() {
+ try {
+ for (File f : dir.listFiles())
+ f.delete();
+ dir.delete();
+ } catch(Exception e) {
+ System.out.println("Unable to clean temporary file "+dir.getAbsolutePath()+" during shutdown.");
+ e.printStackTrace();
+ }
+ }
}
- public static void main(String[] args) {
- System.out.println(getPath());
- }
+ public static void main(String[] args) {
+ System.out.println("This jarfile searches the native code under: " +getPath());
+ }
}
+
+class SimGridLibNotFoundException extends Exception {
+ private static final long serialVersionUID = 1L;
+ public SimGridLibNotFoundException(String msg) {
+ super(msg);
+ }
+
+ public SimGridLibNotFoundException(String msg, Exception e) {
+ super(msg,e);
+ }
+}
\ No newline at end of file
