X-Git-Url: http://nsz.repo.hu/git/?a=blobdiff_plain;f=ir%2Fbe%2Fbejavacoal.c;h=3ef97dbb5a7e000d67a5244edac91c57d9d965ac;hb=80a6158fdd766f42ee6c508a773bc114ff1b61f3;hp=5830cf6757c44e25b4fce4ce70d940e15fe74420;hpb=6ae2d84d2d3e245249b5a9baec13895838d22b64;p=libfirm diff --git a/ir/be/bejavacoal.c b/ir/be/bejavacoal.c index 5830cf675..3ef97dbb5 100644 --- a/ir/be/bejavacoal.c +++ b/ir/be/bejavacoal.c @@ -8,34 +8,113 @@ #include #endif /* WITH_LIBCORE */ +#ifdef _WIN32 +#include +#else +#include +#endif + +#include #include +#include #include -#include #include "bejavacoal.h" +#include "irtools.h" +#include "bemodule.h" + +#ifdef WITH_JVM + +/* 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 jar_file[256] = "/ben/hack/public/coal.jar"; static char cls_name[256] = "coalescing/mst/safe/Algo"; +/* Name of the 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_init_javacoal(void) { - lc_opt_entry_t *jc_grp = lc_opt_get_grp(grp, "jc"); + lc_opt_entry_t *be_grp = lc_opt_get_grp(firm_opt_get_root(), "be"); + lc_opt_entry_t *ra_grp = lc_opt_get_grp(be_grp, "ra"); + lc_opt_entry_t *chordal_grp = lc_opt_get_grp(ra_grp, "chordal"); + lc_opt_entry_t *jc_grp = lc_opt_get_grp(chordal_grp, "jc"); lc_opt_add_table(jc_grp, options); } -#endif + +BE_REGISTER_MODULE_CONSTRUCTOR(be_init_javacoal); +#endif /* WITH_LIBCORE */ + +#include 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 +122,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 +141,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 +158,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 +223,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 +251,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 +264,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 +272,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 +285,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 +301,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 +343,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 +401,73 @@ 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 */