added ir/opt include
[libfirm] / ir / be / bejavacoal.c
index 5830cf6..fde61e6 100644 (file)
 #include <libcore/lc_opts_enum.h>
 #endif /* WITH_LIBCORE */
 
+#ifdef _WIN32
+#include <windows.h>
+#else
+#include <dlfcn.h>
+#endif
+
+#include <signal.h>
 #include <stdlib.h>
+#include <assert.h>
 #include <stdio.h>
-#include <jni.h>
 
 #include "bejavacoal.h"
 
-static char jar_file[256] = "/ben/hack/public/coal.jar";
+/* Path to the jar file. A little OS dependent convenience. */
+#ifdef _WIN32
+static char jar_file[512] = "y:\\user\\hack\\public\\coal.jar";
+#else
+static char jar_file[512] = "/ben/hack/public/coal.jar";
+#endif
+
 static char cls_name[256] = "coalescing/mst/safe/Algo";
 
+/* Name of teh JVM dll/so */
+static char jvm_lib[512] = { 0 };
+
 #ifdef WITH_LIBCORE
 static const lc_opt_table_entry_t options[] = {
+       LC_OPT_ENT_STR      ("jvm",  "absolute path to jvm dll",                    jvm_lib, sizeof(jvm_lib)),
        LC_OPT_ENT_STR      ("jar",  "jar file of the coalescer",                   jar_file, sizeof(jar_file)),
        LC_OPT_ENT_STR      ("cls",  "name of the class providing the factory",     cls_name, sizeof(cls_name)),
        { NULL }
 };
 
-void java_coal_register_options(lc_opt_entry_t *grp)
+void be_java_coal_register_options(lc_opt_entry_t *grp)
 {
        lc_opt_entry_t *jc_grp = lc_opt_get_grp(grp, "jc");
        lc_opt_add_table(jc_grp, options);
 }
 #endif
 
+#ifdef WITH_JVM
+#include <jni.h>
+
 typedef struct _jni_env_t {
        JavaVM *jvm;
        JNIEnv *jni;
 } jni_env_t;
 
+/*
+
+       Ugly code to retrieve the JVM dll/so file.
+
+*/
+
+#ifdef _WIN32
+/* Win32 version */
+static void *find_jvm_symbol(const char *vmlibpath, const char *sym)
+{
+       HINSTANCE hVM = LoadLibrary(vmlibpath);
+       return hVM ? GetProcAddress(hVM, sym) : NULL;
+}
+
+#define JRE_KEY "SOFTWARE\\JavaSoft\\Java Development Kit"
+
+static char *locate_jvm_lib(char *path, size_t path_len)
+{
+       char version[32];
+       char buf[256];
+       DWORD version_len = sizeof(version);
+       DWORD dwPathLen = path_len;
+       HKEY hKey;
+
+       RegOpenKeyEx(HKEY_LOCAL_MACHINE, JRE_KEY, 0, KEY_QUERY_VALUE, &hKey);
+       RegQueryValueEx(hKey, "CurrentVersion", NULL, NULL, (LPBYTE) version, &version_len);
+       RegCloseKey(hKey);
+
+       _snprintf(buf, sizeof(buf), JRE_KEY "\\%s", version);
+       RegOpenKeyEx(HKEY_LOCAL_MACHINE, buf, 0, KEY_QUERY_VALUE, &hKey);
+       RegQueryValueEx(hKey, "JavaHome", NULL, NULL, (LPBYTE) path, &dwPathLen);
+       RegCloseKey(hKey);
+
+       strncat(path, "\\jre\\bin\\server\\jvm.dll", path_len);
+       return path;
+}
+
+#else
+/* Unix version */
+static void *find_jvm_symbol(const char *vmlibpath, const char *sym)
+{
+       void *libVM = dlopen(vmlibpath, RTLD_LAZY);
+       return libVM ? dlsym(libVM, sym) : NULL;
+}
+
+static char *locate_jvm_lib(char *path, size_t n)
+{
+       return NULL;
+}
+#endif
+
 static int start_vm(jni_env_t *env, int argc, char *argv[])
 {
        int i;
@@ -43,6 +114,13 @@ static int start_vm(jni_env_t *env, int argc, char *argv[])
        JavaVMInitArgs args;
        JavaVMOption *opts;
 
+       long (JNICALL * create_func)(JavaVM **, void **, void *) = find_jvm_symbol(jvm_lib, "JNI_CreateJavaVM");
+
+       if(!create_func) {
+               fprintf(stderr, "could not find JVM creation function\n");
+               exit(1);
+       }
+
        memset(&args, 0, sizeof(args));
        opts = malloc(argc * sizeof(opts[0]));
        for(i = 0; i < argc; ++i) {
@@ -55,10 +133,12 @@ static int start_vm(jni_env_t *env, int argc, char *argv[])
        args.options  = opts;
        args.ignoreUnrecognized = JNI_FALSE;
 
-       ret = JNI_CreateJavaVM(&env->jvm, (void **) &env->jni, &args);
+       ret = create_func(&env->jvm, (void **) &env->jni, &args);
        free(opts);
-       if(ret == JNI_ERR)
+       if(ret != JNI_OK) {
+               fprintf(stderr, "JNI_CreateJavaVM returned errrocode %ld\n" , ret);
                return 0;
+       }
 
        return 1;
 }
@@ -70,34 +150,62 @@ static void stop_vm(jni_env_t *env)
        (*jvm)->DestroyJavaVM(jvm);
 }
 
-
-
 static int jvm_inited = 0;
 static jni_env_t env;
+void (*old_int_handler)(int);
+void (*old_abrt_handler)(int);
 
-static void jvm_destroy_at_exit(void)
+static void sig_jvm_destroy_at_exit(int signal)
 {
        if(jvm_inited)
                stop_vm(&env);
+
+       switch(signal) {
+       case SIGABRT:
+               old_abrt_handler(signal);
+               break;
+       case SIGINT:
+               old_int_handler(signal);
+               break;
+       default:;
+       }
 }
 
-static jni_env_t *get_jvm(void)
+static void jvm_destroy_at_exit(void)
 {
+       sig_jvm_destroy_at_exit(0);
+}
 
+static jni_env_t *get_jvm(void)
+{
        char cp_param[512];
        char *args[1];
 
        if(!jvm_inited) {
+               /* Find the dll */
+               if(strlen(jvm_lib) == 0) {
+                       if(!locate_jvm_lib(jvm_lib, sizeof(jvm_lib))) {
+                               fprintf(stderr, "could not find jvm library\n");
+                               exit(1);
+                       }
+               }
+
                snprintf(cp_param, sizeof(cp_param), "-Djava.class.path=%s", jar_file);
                args[0] = cp_param;
-               start_vm(&env, sizeof(args) / sizeof(args[0]), args);
+               if(!start_vm(&env, sizeof(args) / sizeof(args[0]), args)) {
+                       fprintf(stderr, "Couldn't initialize java VM\n");
+                       abort();
+               }
                jvm_inited = 1;
+               old_int_handler  = signal(SIGINT,  sig_jvm_destroy_at_exit);
+               old_abrt_handler = signal(SIGABRT, sig_jvm_destroy_at_exit);
                atexit(jvm_destroy_at_exit);
        }
 
        return &env;
 }
 
+
 static void check(jni_env_t *env, const char *file, int line)
 {
        JNIEnv *jni = env->jni;
@@ -107,25 +215,22 @@ static void check(jni_env_t *env, const char *file, int line)
                (*jni)->ExceptionDescribe(jni);
                (*jni)->ExceptionClear(jni);
                stop_vm(env);
-               exit(1);
+               abort();
        }
 }
 
 #define CHECK(env) check(env, __FILE__, __LINE__)
 
-
-
-typedef struct _jni_env_t jni_env_t;
-
-
 enum {
        mth_add_int_edge,
        mth_add_aff_edge,
        mth_set_color,
+       //mth_set_debug,
        mth_get_color,
        mth_forbid_color,
        mth_coalesce,
        mth_dump,
+       mth_finish,
        mth_last
 };
 
@@ -138,10 +243,12 @@ static const struct _mth_info_t mthis[mth_last] = {
        { "addIntEdge",  "(II)V"                   }, /* public void addIntEdge(int, int); */
        { "addAffEdge",  "(III)V"                  }, /* public void addAffEdge(int, int, int); */
        { "setColor",    "(II)V"                   }, /* public void setColor(int, int); */
-       { "forbidColor", "(II)V"                   }, /* public void forbidColor(int, int); */
+       //{ "setDebug",    "(ILjava/lang/String;)V"  }, /* public void setDebug(int, String); */
        { "getColor",    "(I)I"                    }, /* public int getColor(int); */
+       { "forbidColor", "(II)V"                   }, /* public void forbidColor(int, int); */
        { "coalesce",    "()V"                     }, /* public void coalesce(); */
-       { "dump",        "(Ljava/lang/String;)V"   }  /* public void dump(String); */
+       { "dump",        "(Ljava/lang/String;)V"   }, /* public void dump(String); */
+       { "finish",      "()V"                     }  /* public void finish(); */
 };
 
 /* public static coalescing.Extern createExtern(java.lang.String, int, int, int); */
@@ -149,7 +256,7 @@ static const struct _mth_info_t mthi_factory = {
        "createExtern", "(Ljava/lang/String;III)Lcoalescing/Extern;"
 };
 
-struct _java_coal_t {
+struct _be_java_coal_t {
        jni_env_t *env;
        jclass    cls;
        jobject   obj;
@@ -157,9 +264,7 @@ struct _java_coal_t {
        jmethodID mth_ids[mth_last];
 };
 
-typedef struct _java_coal_t java_coal_t;
-
-static void jc_call_void(java_coal_t *c, int mth_index, ...)
+static void jc_call_void(be_java_coal_t *c, int mth_index, ...)
 {
        JNIEnv *jni   = c->env->jni;
        jmethodID mid = c->mth_ids[mth_index];
@@ -172,7 +277,7 @@ static void jc_call_void(java_coal_t *c, int mth_index, ...)
        va_end(args);
 }
 
-static int jc_call_int(java_coal_t *c, int mth_index, ...)
+static int jc_call_int(be_java_coal_t *c, int mth_index, ...)
 {
        JNIEnv *jni   = c->env->jni;
        jmethodID mid = c->mth_ids[mth_index];
@@ -188,9 +293,9 @@ static int jc_call_int(java_coal_t *c, int mth_index, ...)
        return res;
 }
 
-java_coal_t *java_coal_init(const char *graph_name, int n_nodes, int n_regs, int dbg_level)
+be_java_coal_t *be_java_coal_init(const char *graph_name, int n_nodes, int n_regs, int dbg_level)
 {
-       java_coal_t *c;
+       be_java_coal_t *c;
        jni_env_t *env = get_jvm();
        JNIEnv *jni = env->jni;
        jmethodID fact;
@@ -230,38 +335,53 @@ java_coal_t *java_coal_init(const char *graph_name, int n_nodes, int n_regs, int
        return c;
 }
 
-void java_coal_destroy(java_coal_t *c) {
+void be_java_coal_destroy(be_java_coal_t *c) {
        JNIEnv *jni = c->env->jni;
+       jc_call_void(c, mth_finish);
        (*jni)->DeleteGlobalRef(jni, c->obj);
        free(c);
 }
 
-void java_coal_add_int_edge(java_coal_t *c, int n, int m)
+void be_java_coal_add_int_edge(be_java_coal_t *c, int n, int m)
 {
        jc_call_void(c, mth_add_int_edge, (jint) n, (jint) m);
 }
 
-void java_coal_add_aff_edge(java_coal_t *c, int n, int m, int weight)
+void be_java_coal_add_aff_edge(be_java_coal_t *c, int n, int m, int weight)
 {
        jc_call_void(c, mth_add_aff_edge, (jint) n, (jint) m, (jint) weight);
 }
 
-void java_coal_set_color(java_coal_t *c, int n, int col)
+void be_java_coal_set_color(be_java_coal_t *c, int n, int col)
 {
        jc_call_void(c, mth_set_color, (jint) n, (jint) col);
 }
 
-void java_coal_forbid_color(java_coal_t *c, int n, int col)
+void be_java_coal_set_debug(be_java_coal_t *c, int n, const char *dbg)
+{
+#if 0
+       JNIEnv *jni   = c->env->jni;
+       jmethodID mid = c->mth_ids[mth_set_debug];
+       jstring str;
+
+       str = (*jni)->NewStringUTF(jni, dbg);
+       CHECK(c->env);
+       (*jni)->CallVoidMethod(jni, c->obj, mid, (jint) n, str);
+       CHECK(c->env);
+#endif
+}
+
+void be_java_coal_forbid_color(be_java_coal_t *c, int n, int col)
 {
        jc_call_void(c, mth_forbid_color, (jint) n, (jint) col);
 }
 
-void java_coal_coalesce(java_coal_t *c)
+void be_java_coal_coalesce(be_java_coal_t *c)
 {
        jc_call_void(c, mth_coalesce);
 }
 
-void java_coal_dump(java_coal_t *c, const char *fn)
+void be_java_coal_dump(be_java_coal_t *c, const char *fn)
 {
        JNIEnv *jni   = c->env->jni;
        jmethodID mid = c->mth_ids[mth_dump];
@@ -273,7 +393,74 @@ void java_coal_dump(java_coal_t *c, const char *fn)
        CHECK(c->env);
 }
 
-int java_coal_get_color(java_coal_t *c, int n)
+int be_java_coal_get_color(be_java_coal_t *c, int n)
 {
        return jc_call_int(c, mth_get_color, (jint) n);
 }
+
+void be_java_coal_start_jvm(void)
+{
+       get_jvm();
+}
+
+#else
+
+be_java_coal_t *be_java_coal_init(const char *graph_name, int n_nodes, int n_regs, int dbg_level)
+{
+       assert(0 && "use --enable-jvm");
+       return NULL;
+}
+
+void be_java_coal_destroy(be_java_coal_t *c)
+{
+       assert(0 && "use --enable-jvm");
+}
+
+
+void be_java_coal_add_int_edge(be_java_coal_t *c, int n, int m)
+{
+       assert(0 && "use --enable-jvm");
+}
+
+void be_java_coal_add_aff_edge(be_java_coal_t *c, int n, int m, int weight)
+{
+       assert(0 && "use --enable-jvm");
+}
+
+void be_java_coal_set_color(be_java_coal_t *c, int n, int col)
+{
+       assert(0 && "use --enable-jvm");
+}
+
+void be_java_coal_set_debug(be_java_coal_t *c, int n, const char *dbg)
+{
+       assert(0 && "use --enable-jvm");
+}
+
+void be_java_coal_forbid_color(be_java_coal_t *c, int n, int col)
+{
+       assert(0 && "use --enable-jvm");
+}
+
+void be_java_coal_coalesce(be_java_coal_t *c)
+{
+       assert(0 && "use --enable-jvm");
+}
+
+void be_java_coal_dump(be_java_coal_t *c, const char *fn)
+{
+       assert(0 && "use --enable-jvm");
+}
+
+int be_java_coal_get_color(be_java_coal_t *c, int n)
+{
+       assert(0 && "use --enable-jvm");
+       return -1;
+}
+
+void be_java_coal_start_jvm(void)
+{
+}
+
+
+#endif /* WITH_JVM */