X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/6683debe0efeb6371e0cbafb4f0d461325dd59a5..77d095374d3f966ce0f54f3837116c73bc3e52e4:/src/bindings/java/org/simgrid/NativeLib.java diff --git a/src/bindings/java/org/simgrid/NativeLib.java b/src/bindings/java/org/simgrid/NativeLib.java index 130c904f6c..7c08bc267b 100644 --- a/src/bindings/java/org/simgrid/NativeLib.java +++ b/src/bindings/java/org/simgrid/NativeLib.java @@ -1,112 +1,162 @@ +/* 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