Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Adapt to the latest changes in MSG API
[simgrid.git] / src / jmsg.c
1 /* Java Wrappers to the MSG API.                                            */
2
3 /* Copyright (c) 2007, 2008, 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 #include <msg/msg.h>
10 #include <simgrid/simix.h>
11 #include <surf/surfxml_parse.h>
12 #include <locale.h>
13
14 #include "smx_context_java.h"
15 #include "smx_context_cojava.h"
16
17 #include "jmsg_process.h"
18
19 #include "jmsg_host.h"
20 #include "jmsg_task.h"
21 #include "jxbt_utilities.h"
22
23 #include "jmsg.h"
24
25 /* Shut up some errors in eclipse online compiler. I wish such a pimple wouldn't be needed */
26 #ifndef JNIEXPORT
27 #define JNIEXPORT
28 #endif
29 #ifndef JNICALL
30 #define JNICALL
31 #endif
32 /* end of eclipse-mandated pimple */
33
34 static int create_jprocess(int argc, char *argv[]);
35
36 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(jmsg);
37
38 JavaVM *__java_vm = NULL;
39
40 JavaVM *get_java_VM(void)
41 {
42   return __java_vm;
43 }
44
45 JNIEnv *get_current_thread_env(void)
46 {
47   JNIEnv *env;
48
49   (*__java_vm)->AttachCurrentThread(__java_vm, (void **) &env, NULL);
50   return env;
51 }
52
53 void jmsg_throw_status(JNIEnv *env, msg_error_t status) {
54   switch (status) {
55     case MSG_TIMEOUT:
56         jxbt_throw_time_out_failure(env,NULL);
57     break;
58     case MSG_TRANSFER_FAILURE:
59         jxbt_throw_transfer_failure(env,NULL);
60     break;
61     case MSG_HOST_FAILURE:
62         jxbt_throw_host_failure(env,NULL);
63     break;
64     default:
65         jxbt_throw_native(env,bprintf("communication failed"));
66   }
67 }
68
69
70 /***************************************************************************************
71  * Unsortable functions                                                        *
72  ***************************************************************************************/
73
74 JNIEXPORT jdouble JNICALL
75 Java_org_simgrid_msg_Msg_getClock(JNIEnv * env, jclass cls)
76 {
77   return (jdouble) MSG_get_clock();
78 }
79
80 JNIEXPORT void JNICALL
81 Java_org_simgrid_msg_Msg_init(JNIEnv * env, jclass cls, jobjectArray jargs)
82 {
83   char **argv = NULL;
84   int index;
85   int argc = 0;
86   jstring jval;
87   const char *tmp;
88
89   (*env)->GetJavaVM(env, &__java_vm);
90
91   if ((*env)->FindClass(env, "java/dyn/Coroutine")) {
92         XBT_VERB("Using Coroutines");
93         smx_factory_initializer_to_use = SIMIX_ctx_cojava_factory_init;
94   }
95   else {
96         XBT_VERB("Using java threads");
97         smx_factory_initializer_to_use = SIMIX_ctx_java_factory_init;
98   }
99   jthrowable exc = (*env)->ExceptionOccurred(env);
100   if (exc) {
101     (*env)->ExceptionClear(env);
102   }
103
104   setlocale(LC_NUMERIC,"C");
105
106   if (jargs)
107     argc = (int) (*env)->GetArrayLength(env, jargs);
108
109   argc++;
110   argv = xbt_new(char *, argc + 1);
111   argv[0] = strdup("java");
112
113   for (index = 0; index < argc - 1; index++) {
114     jval = (jstring) (*env)->GetObjectArrayElement(env, jargs, index);
115     tmp = (*env)->GetStringUTFChars(env, jval, 0);
116     argv[index + 1] = strdup(tmp);
117     (*env)->ReleaseStringUTFChars(env, jval, tmp);
118   }
119   argv[argc] = NULL;
120
121   MSG_init(&argc, argv);
122
123   for (index = 0; index < argc; index++)
124     free(argv[index]);
125
126   free(argv);
127 }
128
129 JNIEXPORT void JNICALL
130     JNICALL Java_org_simgrid_msg_Msg_run(JNIEnv * env, jclass cls)
131 {
132   msg_error_t rv;
133   int index;
134   xbt_dynar_t hosts;
135   jobject jhost;
136
137   /* Run everything */
138   XBT_INFO("Ready to run MSG_MAIN");
139   rv = MSG_main();
140   XBT_INFO("Done running MSG_MAIN");
141   jxbt_check_res("MSG_main()", rv, MSG_OK,
142                  bprintf
143                  ("unexpected error : MSG_main() failed .. please report this bug "));
144
145   XBT_INFO("MSG_main finished");
146
147   XBT_INFO("Clean java world");
148   /* Cleanup java hosts */
149   hosts = MSG_hosts_as_dynar();
150   for (index = 0; index < xbt_dynar_length(hosts) - 1; index++) {
151     jhost = (jobject) MSG_host_get_data(xbt_dynar_get_as(hosts,index,msg_host_t));
152     if (jhost)
153       jhost_unref(env, jhost);
154
155   }
156   xbt_dynar_free(&hosts);
157   XBT_INFO("Clean native world");
158 }
159 JNIEXPORT void JNICALL
160     JNICALL Java_org_simgrid_msg_Msg_clean(JNIEnv * env, jclass cls)
161 {
162   /* cleanup native stuff. Calling it is ... useless since leaking memory at the end of the simulation is a non-issue */
163   msg_error_t rv = MSG_OK != MSG_clean();
164   jxbt_check_res("MSG_clean()", rv, MSG_OK,
165                  bprintf
166                  ("unexpected error : MSG_clean() failed .. please report this bug "));
167 }
168
169 JNIEXPORT void JNICALL
170 Java_org_simgrid_msg_Msg_createEnvironment(JNIEnv * env, jclass cls,
171                                        jstring jplatformFile)
172 {
173
174   const char *platformFile =
175       (*env)->GetStringUTFChars(env, jplatformFile, 0);
176
177   MSG_create_environment(platformFile);
178
179   (*env)->ReleaseStringUTFChars(env, jplatformFile, platformFile);
180 }
181 JNIEXPORT void JNICALL
182 Java_org_simgrid_msg_Msg_debug(JNIEnv * env, jclass cls, jstring js)
183 {
184   const char *s = (*env)->GetStringUTFChars(env, js, 0);
185   XBT_DEBUG("%s", s);
186   (*env)->ReleaseStringUTFChars(env, js, s);
187 }
188 JNIEXPORT void JNICALL
189 Java_org_simgrid_msg_Msg_info(JNIEnv * env, jclass cls, jstring js)
190 {
191   const char *s = (*env)->GetStringUTFChars(env, js, 0);
192   XBT_INFO("%s", s);
193   (*env)->ReleaseStringUTFChars(env, js, s);
194 }
195
196 JNIEXPORT void JNICALL
197 Java_org_simgrid_msg_Msg_deployApplication(JNIEnv * env, jclass cls,
198                                        jstring jdeploymentFile)
199 {
200
201   const char *deploymentFile =
202       (*env)->GetStringUTFChars(env, jdeploymentFile, 0);
203
204   SIMIX_function_register_default(create_jprocess);
205   MSG_launch_application(deploymentFile);
206 }
207 /**
208  * Function called when there is the need to create the java Process object
209  * (when we are using deployement files).
210  * it HAS to be executed on the process context, else really bad things will happen.
211  */
212 static int create_jprocess(int argc, char *argv[]) {
213   JNIEnv *env = get_current_thread_env();
214   //Change the "." in class name for "/".
215   xbt_str_subst(argv[0],'.','/',0);
216   jclass class_Process = (*env)->FindClass(env, argv[0]);
217   xbt_str_subst(argv[0],'/','.',0);
218   //Retrieve the methodID for the constructor
219   xbt_assert((class_Process != NULL), "Class not found.");
220   jmethodID constructor_Process = (*env)->GetMethodID(env, class_Process, "<init>", "(Lorg/simgrid/msg/Host;Ljava/lang/String;[Ljava/lang/String;)V");
221   xbt_assert((constructor_Process != NULL), "Constructor not found for class %s. Is there a (Host, String ,String[]) constructor in your class ?", argv[0]);
222
223   //Retrieve the name of the process.
224   jstring jname = (*env)->NewStringUTF(env, argv[0]);
225   //Build the arguments
226   jobjectArray args = (jobjectArray)((*env)->NewObjectArray(env,argc - 1,
227   (*env)->FindClass(env,"java/lang/String"),
228   (*env)->NewStringUTF(env,"")));
229   int i;
230   for (i = 1; i < argc; i++)
231       (*env)->SetObjectArrayElement(env,args,i - 1,(*env)->NewStringUTF(env, argv[i]));
232   //Retrieve the host for the process.
233   jstring jhostName = (*env)->NewStringUTF(env, MSG_host_get_name(MSG_host_self()));
234   jobject jhost = Java_org_simgrid_msg_Host_getByName(env, NULL, jhostName);
235   //creates the process
236   jobject jprocess = (*env)->NewObject(env, class_Process, constructor_Process, jhost, jname, args);
237   xbt_assert((jprocess != NULL), "Process allocation failed.");
238   jprocess = (*env)->NewGlobalRef(env, jprocess);
239   //bind the process to the context
240   msg_process_t process = MSG_process_self();
241   smx_ctx_java_t context = (smx_ctx_java_t)MSG_process_get_smx_ctx(process);
242   context->jprocess = jprocess;
243 /* sets the PID and the PPID of the process */
244 (*env)->SetIntField(env, jprocess, jprocess_field_Process_pid,(jint) MSG_process_get_PID(process));
245 (*env)->SetIntField(env, jprocess, jprocess_field_Process_ppid, (jint) MSG_process_get_PPID(process));
246   jprocess_bind(jprocess, process, env);
247
248   return 0;
249 }
250