org.robovm.llvm.NativeLibrary Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of robovm-llvm Show documentation
There is a newer version: 1.12.0
/*
 * Copyright (C) 2013 RoboVM AB
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see .
 */
package org.robovm.llvm;

import java.io.BufferedOutputStream;
import java.io.Closeable;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

import org.robovm.llvm.binding.LLVM;

/**
 * 
 */
public class NativeLibrary {
    private static boolean loaded = false;
    private static final String os;
    private static final String arch;
    private static final String libName;

    static {
        String osProp = System.getProperty("os.name").toLowerCase();
        String archProp = System.getProperty("os.arch").toLowerCase();
        String ext = null;
        if (osProp.startsWith("mac") || osProp.startsWith("darwin")) {
            os = "macosx";
            ext = "dylib";
        } else if (osProp.startsWith("linux")) {
            os = "linux";
            ext = "so";
        } else if (osProp.startsWith("windows")) {
            os = "windows";
            ext = "dll";
        } else {
            throw new Error("Unsupported OS: " + System.getProperty("os.name"));
        }
        if (archProp.matches("amd64|x86[-_]64")) {
            arch = "x86_64";
        } else if (archProp.matches("i386|x86")) {
            arch = "x86";
        } else {
            throw new Error("Unsupported arch: " + System.getProperty("os.arch"));
        }
        
        libName = "librobovm-llvm." + ext;
    }
    
    public static synchronized void load() {
        if (loaded) {
            return;
        }
        
        String prefix = libName.substring(0, libName.lastIndexOf('.'));
        String ext = libName.substring(libName.lastIndexOf('.'));
        
        InputStream in = NativeLibrary.class.getResourceAsStream("binding/" + os + "/" + arch + "/" + libName);
        if (in == null) {
            throw new UnsatisfiedLinkError("Native library for " + os + "-" + arch + " not found");
        }
        OutputStream out = null;
        File tmpLibFile = null;
        try {
            tmpLibFile = File.createTempFile(prefix, ext);
            tmpLibFile.deleteOnExit();
            out = new BufferedOutputStream(new FileOutputStream(tmpLibFile));
            copy(in, out);
        } catch (IOException e) {
            throw (Error) new UnsatisfiedLinkError(e.getMessage()).initCause(e);
        } finally {
            closeQuietly(in);
            closeQuietly(out);
        }
        
        Runtime.getRuntime().load(tmpLibFile.getAbsolutePath());
        if (!LLVM.StartMultithreaded()) {
            throw new UnsatisfiedLinkError("LLVMStartMultithreaded failed");
        }
        
        LLVM.InitializeAllTargets();
        LLVM.InitializeAllTargetInfos();
        LLVM.InitializeAllTargetMCs();
        LLVM.InitializeAllAsmPrinters();
        LLVM.InitializeAllAsmParsers();
    }
    
    private static void copy(InputStream in, OutputStream out) throws IOException {
        byte[] buffer = new byte[4096];
        int n = 0;
        while ((n = in.read(buffer)) != -1) {
            out.write(buffer, 0, n);
        }
    }
    
    private static void closeQuietly(Closeable in) {
        try {
            if (in != null) {
                in.close();
            }
        } catch (IOException ioe) {
        }
    }
}