src/bindings/java/org/simgrid/NativeLib.java

   1 /* Copyright (c) 2014. The SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the license (GNU LGPL) which comes with this package. */
   6
   7 package org.simgrid;
   8
   9 import java.io.FileOutputStream;
  10 import java.io.InputStream;
  11 import java.io.OutputStream;
  12 import java.io.File;
  13
  14 public final class NativeLib {
  15
  16     public static String getPath() {
  17         String prefix = "NATIVE";
  18         String os = System.getProperty("os.name");
  19         String arch = System.getProperty("os.arch");
  20
  21         if (arch.matches("^i[3-6]86$"))
  22             arch = "x86";
  23         else if (arch.equalsIgnoreCase("amd64"))
  24             arch = "x86_64";
  25
  26         if (os.toLowerCase().startsWith("^win")){
  27             os = "Windows";
  28             arch = "x86";
  29         }else if (os.contains("OS X"))
  30             os = "Darwin";
  31
  32         os = os.replace(' ', '_');
  33         arch = arch.replace(' ', '_');
  34
  35         return prefix + "/" + os + "/" + arch + "/";
  36     }
  37     public static void nativeInit(String name) {
  38         try {
  39             /* prefer the version on disk, if existing */
  40             System.loadLibrary(name);
  41         } catch (UnsatisfiedLinkError e) {
  42             /* If not found, unpack the one bundled into the jar file and use it */
  43             loadLib(name);
  44         }
  45     }
  46
  47     private static void loadLib (String name) {
  48         String Path = NativeLib.getPath();
  49
  50         String filename=name;
  51         InputStream in = NativeLib.class.getClassLoader().getResourceAsStream(Path+filename);
  52
  53         if (in == null) {
  54             filename = "lib"+name+".so";
  55             in = NativeLib.class.getClassLoader().getResourceAsStream(Path+filename);
  56         }
  57         if (in == null) {
  58             filename = name+".dll";
  59             in =  NativeLib.class.getClassLoader().getResourceAsStream(Path+filename);
  60         }
  61         if (in == null) {
  62             filename = "lib"+name+".dll";
  63             in =  NativeLib.class.getClassLoader().getResourceAsStream(Path+filename);
  64         }
  65         if (in == null) {
  66             filename = "lib"+name+".dylib";
  67             in =  NativeLib.class.getClassLoader().getResourceAsStream(Path+filename);
  68         }
  69         if (in == null) {
  70             throw new RuntimeException("Cannot find library "+name+" in path "+Path+". Sorry, but this jar does not seem to be usable on your machine.");
  71         }
  72         try {
  73             // We must write the lib onto the disk before loading it -- stupid operating systems
  74             File fileOut = new File(filename);
  75             fileOut = File.createTempFile(name+"-", ".tmp");
  76             // don't leak the file on disk, but remove it on JVM shutdown
  77             Runtime.getRuntime().addShutdownHook(new Thread(new FileCleaner(fileOut.getAbsolutePath())));
  78             OutputStream out = new FileOutputStream(fileOut);
  79
  80             /* copy the library in position */
  81             byte[] buffer = new byte[4096];
  82             int bytes_read;
  83             while ((bytes_read = in.read(buffer)) != -1)     // Read until EOF
  84                     out.write(buffer, 0, bytes_read);
  85
  86             /* close all file descriptors, and load that shit */
  87             in.close();
  88             out.close();
  89             System.load(fileOut.getAbsolutePath());
  90
  91         } catch (Exception e) {
  92             System.err.println("Cannot load the bindings to the "+name+" library: ");
  93             e.printStackTrace();
  94             System.err.println("This jar file does not seem to fit your system, sorry");
  95             System.exit(1);
  96         }
  97     }
  98
  99         /* A hackish mechanism used to remove the file containing our library when the JVM shuts down */
 100         private static class FileCleaner implements Runnable {
 101                 private String target;
 102                 public FileCleaner(String name) {
 103                         target = name;
 104                 }
 105         public void run() {
 106             try {
 107                 new File(target).delete();
 108             } catch(Exception e) {
 109                 System.out.println("Unable to clean temporary file "+target+" during shutdown.");
 110                 e.printStackTrace();
 111             }
 112         }
 113         }
 114
 115
 116     public static void main(String[] args) {
 117         System.out.println(getPath());
 118     }
 119 }