001 // --- BEGIN LICENSE BLOCK --- 002 /* 003 * Copyright (c) 2009, Mikio L. Braun 004 * All rights reserved. 005 * 006 * Redistribution and use in source and binary forms, with or without 007 * modification, are permitted provided that the following conditions are 008 * met: 009 * 010 * * Redistributions of source code must retain the above copyright 011 * notice, this list of conditions and the following disclaimer. 012 * 013 * * Redistributions in binary form must reproduce the above 014 * copyright notice, this list of conditions and the following 015 * disclaimer in the documentation and/or other materials provided 016 * with the distribution. 017 * 018 * * Neither the name of the Technische Universit??t Berlin nor the 019 * names of its contributors may be used to endorse or promote 020 * products derived from this software without specific prior 021 * written permission. 022 * 023 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 024 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 025 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 026 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 027 * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 028 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 029 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 030 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 031 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 032 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 033 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 034 */ 035 // --- END LICENSE BLOCK --- 036 037 package org.jblas.util; 038 039 import java.io.*; 040 041 /** 042 * Class which allows to load a dynamic file as resource (for example, from a 043 * jar-file) 044 */ 045 public class LibraryLoader { 046 /** Find the library <tt>libname</tt> as a resource, copy it to a tempfile 047 * and load it using System.load(). The name of the library has to be the 048 * base name, it is mapped to the corresponding system name using 049 * System.mapLibraryName(). For example, the library "foo" is called "libfoo.so" 050 * under Linux and "foo.dll" under Windows, but you just have to pass "foo" 051 * the loadLibrary(). 052 * 053 * I'm not quite sure if this doesn't open all kinds of security holes. Any ideas? 054 * 055 * @param libname basename of the library 056 */ 057 public void loadLibrary(String libname) { 058 libname = System.mapLibraryName(libname); 059 060 // We're in a static initializer and need a class. What shall we do? 061 Class cl = getClass(); 062 063 InputStream is = cl.getResourceAsStream("/" + libname); 064 065 // Hm, let's see if we can find it in the build directory 066 if (is == null) 067 is = cl.getResourceAsStream("/bin/" + libname); 068 069 if (is == null) { 070 System.err.println("Couldn't find the resource " + libname + "."); 071 System.exit(0); 072 } 073 074 try { 075 File tempfile = File.createTempFile("jblas", libname); 076 tempfile.deleteOnExit(); 077 OutputStream os = new FileOutputStream(tempfile); 078 079 System.out.println("tempfile.getPath() = " + tempfile.getPath()); 080 081 long savedTime = System.currentTimeMillis(); 082 083 byte buf[] = new byte[1024]; 084 int len; 085 while ((len = is.read(buf)) > 0) { 086 os.write(buf, 0, len); 087 } 088 089 double seconds = (double) (System.currentTimeMillis() - savedTime) / 1e3; 090 System.out.println("Copying took " + seconds + " seconds."); 091 092 os.close(); 093 094 System.load(tempfile.getPath()); 095 } catch (IOException io) { 096 System.err.println("Could not create the temp file: " + io.toString() + ".\n"); 097 } catch (UnsatisfiedLinkError ule) { 098 System.err.println("Couldn't load copied link file: " + ule.toString() + ".\n"); 099 } 100 } 101 102 /** Compute the path to the library. The path is basically 103 "/" + os.name + "/" + os.arch + "/" + libname. */ 104 static public String libraryPath(String libname) { 105 return "/" + System.getProperty("os.name") + "/" + System.getProperty("os.arch") + "/" + libname; 106 } 107 108 static public void main(String[] args) { 109 System.out.println(libraryPath("")); 110 } 111 }