Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Stop using JNI_GetCreatedJavaVMs, cache the java_vm object
[simgrid.git] / src / smx_context_java.c
1 /* context_java - implementation of context switching for java threads */
2
3 /* Copyright (c) 2009, 2010. The SimGrid Team.
4  * All rights reserved.                                                     */
5
6 /* This program is free software; you can redistribute it and/or modify it
7  * under the terms of the license (GNU LGPL) which comes with this package. */
8
9
10 #include <xbt/function_types.h>
11 #include <simgrid/simix.h>
12 #include <xbt/ex.h>
13 #include "smx_context_java.h"
14 #include "jxbt_utilities.h"
15 #include "xbt/dynar.h"
16 extern JavaVM *__java_vm;
17
18 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(jmsg, bindings, "MSG for Java(TM)");
19
20 static smx_context_t
21 smx_ctx_java_factory_create_context(xbt_main_func_t code, int argc,
22                                     char **argv,
23                                     void_pfn_smxprocess_t cleanup_func,
24                                     void *data);
25
26 static void smx_ctx_java_free(smx_context_t context);
27 static void smx_ctx_java_start(smx_context_t context);
28 static void smx_ctx_java_suspend(smx_context_t context);
29 static void smx_ctx_java_resume(smx_context_t new_context);
30 static void smx_ctx_java_runall(void);
31 static void* smx_ctx_java_thread_run(void *data);
32 void SIMIX_ctx_java_factory_init(smx_context_factory_t * factory)
33 {
34   /* instantiate the context factory */
35   smx_ctx_base_factory_init(factory);
36
37   (*factory)->create_context = smx_ctx_java_factory_create_context;
38   /* Leave default behavior of (*factory)->finalize */
39   (*factory)->free = smx_ctx_java_free;
40   (*factory)->stop = smx_ctx_java_stop;
41   (*factory)->suspend = smx_ctx_java_suspend;
42   (*factory)->runall = smx_ctx_java_runall;
43   (*factory)->name = "ctx_java_factory";
44   //(*factory)->finalize = smx_ctx_base_factory_finalize;
45   (*factory)->self = smx_ctx_java_self;
46   (*factory)->get_data = smx_ctx_base_get_data;
47 }
48 smx_context_t smx_ctx_java_self(void)
49 {
50         return (smx_context_t)xbt_os_thread_get_extra_data();
51 }
52
53 static smx_context_t
54 smx_ctx_java_factory_create_context(xbt_main_func_t code, int argc,
55                                     char **argv,
56                                     void_pfn_smxprocess_t cleanup_func,
57                                     void* data)
58 {
59         smx_ctx_java_t context = xbt_new0(s_smx_ctx_java_t, 1);
60   /* If the user provided a function for the process then use it
61      otherwise is the context for maestro */
62   if (code) {
63                 if (argc == 0) {
64                         context->jprocess = (jobject) code;
65                 }
66                 else {
67                         context->jprocess = NULL;
68                 }
69                 context->super.cleanup_func = cleanup_func;
70                 context->begin = xbt_os_sem_init(0);
71                 context->end = xbt_os_sem_init(0);
72
73                 context->super.argc = argc;
74                 context->super.argv = argv;
75                 context->super.code = code;
76
77                 context->thread = xbt_os_thread_create(NULL,smx_ctx_java_thread_run,context,NULL);
78   }
79   else {
80         context->thread = NULL;
81     xbt_os_thread_set_extra_data(context);
82   }
83   context->super.data = data;
84   
85   return (smx_context_t) context;
86 }
87
88 static void* smx_ctx_java_thread_run(void *data) {
89         smx_ctx_java_t context = (smx_ctx_java_t)data;
90         xbt_os_thread_set_extra_data(context);
91         //Attach the thread to the JVM
92         JNIEnv *env;
93   jint error = (*__java_vm)->AttachCurrentThread(__java_vm, (void **) &env, NULL);
94   xbt_assert((error == JNI_OK), "The thread could not be attached to the JVM");
95   context->jenv = get_current_thread_env();
96   //Wait for the first scheduling round to happen.
97   xbt_os_sem_acquire(context->begin);
98   //Create the "Process" object if needed.
99         if (context->super.argc > 0) {
100                 (*(context->super.code))(context->super.argc, context->super.argv);
101         }
102         xbt_assert((context->jprocess != NULL), "Process not created...");
103   //wait for the process to be able to begin
104   //TODO: Cache it
105         jfieldID jprocess_field_Process_startTime = jxbt_get_sfield(env, "org/simgrid/msg/Process", "startTime", "D");
106   jdouble startTime =  (*env)->GetDoubleField(env, context->jprocess, jprocess_field_Process_startTime);
107   if (startTime > MSG_get_clock()) {
108         MSG_process_sleep(startTime - MSG_get_clock());
109   }
110   //Execution of the "run" method.
111   jmethodID id = jxbt_get_smethod(env, "org/simgrid/msg/Process", "run", "()V");
112   xbt_assert( (id != NULL), "Method not found...");
113   (*env)->CallVoidMethod(env, context->jprocess, id);
114   smx_ctx_java_stop((smx_context_t)context);
115
116   return NULL;
117 }
118
119 static void smx_ctx_java_free(smx_context_t context)
120 {
121         if (context) {
122                 smx_ctx_java_t ctx_java = (smx_ctx_java_t) context;
123                 if (ctx_java->thread) { /* We are not in maestro context */
124                         xbt_os_thread_join(ctx_java->thread, NULL);
125                         xbt_os_sem_destroy(ctx_java->begin);
126                         xbt_os_sem_destroy(ctx_java->end);
127                 }
128   }
129   smx_ctx_base_free(context);
130 }
131
132
133 void smx_ctx_java_stop(smx_context_t context)
134 {
135         smx_ctx_java_t ctx_java = (smx_ctx_java_t)context;
136   /* I am the current process and I am dying */
137         if (context->iwannadie == -1) {
138         context->iwannadie = 0;
139         JNIEnv *env = get_current_thread_env();
140         jxbt_throw_by_name(env, "org/simgrid/msg/ProcessKilledError", bprintf("Process killed :)"));
141         THROWF(cancel_error, 0, "process cancelled");
142   }
143   else {
144     smx_ctx_base_stop(context);
145                 /* detach the thread and kills it */
146                 JNIEnv *env = ctx_java->jenv;
147                 (*env)->DeleteGlobalRef(env,ctx_java->jprocess);
148                 jint error = (*__java_vm)->DetachCurrentThread(__java_vm);
149                 xbt_assert((error == JNI_OK), "The thread couldn't be detached.");
150                 xbt_os_sem_release(((smx_ctx_java_t)context)->end);
151                 xbt_os_thread_exit(NULL);
152
153   }
154 }
155
156 static void smx_ctx_java_suspend(smx_context_t context)
157 {
158         smx_ctx_java_t ctx_java = (smx_ctx_java_t) context;
159         xbt_os_sem_release(ctx_java->end);
160         xbt_os_sem_acquire(ctx_java->begin);
161 }
162
163 // FIXME: inline those functions
164 static void smx_ctx_java_resume(smx_context_t new_context)
165 {
166         smx_ctx_java_t ctx_java = (smx_ctx_java_t) new_context;
167         xbt_os_sem_release(ctx_java->begin);
168         xbt_os_sem_acquire(ctx_java->end);
169 }
170
171 static void smx_ctx_java_runall(void)
172 {
173   xbt_dynar_t processes = SIMIX_process_get_runnable();
174   smx_process_t process;
175   unsigned int cursor;
176   xbt_dynar_foreach(processes, cursor, process) {
177     smx_ctx_java_resume(SIMIX_process_get_context(process));
178   }
179 }