Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Remove the ability to write internal plugins in Java
[simgrid.git] / src / bindings / java / org / simgrid / NativeLib.java
index 226c3fb..7c08bc2 100644 (file)
@@ -10,23 +10,62 @@ 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");
+
+               try {
+                       NativeLib.nativeInit("boost_context");
+               } catch (Exception e) {/* Dont care */}
+               NativeLib.nativeInit("simgrid");
+               NativeLib.nativeInit("simgrid-java");      
+               isNativeInited = true;
+       }
+
+       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("amd64"))
-                       arch = "x86_64";
+               else if (arch.equalsIgnoreCase("x86_64"))
+                       arch = "amd64";
+               else if (arch.equalsIgnoreCase("AMD64"))
+                       arch = "amd64";
 
                if (os.toLowerCase().startsWith("win")){
                        os = "Windows";
-                       arch = "x86";
-               }else if (os.contains("OS X"))
+               } else if (os.contains("OS X"))
                        os = "Darwin";
 
                os = os.replace(' ', '_');
@@ -34,23 +73,7 @@ public final class NativeLib {
 
                return prefix + "/" + os + "/" + arch + "/";
        }
-       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) {
-                               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);
-                       }
-               }
-       }
-
+       static Path tempDir = null;
        private static void loadLib (String name) throws SimGridLibNotFoundException {
                String Path = NativeLib.getPath();
 
@@ -60,31 +83,33 @@ public final class NativeLib {
                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
-                       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);
+                       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
@@ -94,23 +119,26 @@ public final class NativeLib {
                        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();
+                           for (File f : dir.listFiles())
+                               f.delete();
+                               dir.delete();
                        } catch(Exception e) {
-                               System.out.println("Unable to clean temporary file "+target+" during shutdown.");
+                               System.out.println("Unable to clean temporary file "+dir.getAbsolutePath()+" during shutdown.");
                                e.printStackTrace();
                        }
                }    
@@ -118,11 +146,7 @@ public final class NativeLib {
 
 
        public static void main(String[] args) {
-               if (args.length >= 1 && args[0].equals("--quiet"))
-                       /* be careful, this execution path is used in tools/cmake/scripts/java_bundle.sh to determine where to put the libs */
-                       System.out.println(getPath());
-               else 
-                       System.out.println("This java library will try to load the native code under the following name:\n" +getPath());
+               System.out.println("This jarfile searches the native code under: " +getPath());
        }
 }