+static jfieldID jcomm_field_Comm_bind;
+static jfieldID jcomm_field_Comm_finished;
+static jfieldID jcomm_field_Comm_receiving;
+static jfieldID jtask_field_Comm_task;
+static jfieldID jcomm_field_Comm_taskBind;
+
+void jcomm_bind_task(JNIEnv *env, jobject jcomm) {
+ msg_comm_t comm = (msg_comm_t) (long) (*env)->GetLongField(env, jcomm, jcomm_field_Comm_bind);
+ //test if we are receiving or sending a task.
+ jboolean jreceiving = (*env)->GetBooleanField(env, jcomm, jcomm_field_Comm_receiving);
+ if (jreceiving == JNI_TRUE) {
+ //bind the task object.
+ m_task_t task = MSG_comm_get_task(comm);
+ xbt_assert(task != NULL, "Task is NULL");
+ jobject jtask_global = MSG_task_get_data(task);
+ //case where the data has already been retrieved
+ if (jtask_global == NULL) {
+ return;
+ }
+
+ //Make sure the data will be correctly gc.
+ jobject jtask_local = (*env)->NewLocalRef(env, jtask_global);
+ (*env)->DeleteGlobalRef(env, jtask_global);
+
+ (*env)->SetObjectField(env, jcomm, jtask_field_Comm_task, jtask_local);
+
+ MSG_task_set_data(task, NULL);
+ }
+
+}
+
+JNIEXPORT void JNICALL
+Java_org_simgrid_msg_Comm_nativeInit(JNIEnv *env, jclass cls) {
+ jclass jfield_class_Comm = (*env)->FindClass(env, "org/simgrid/msg/Comm");
+ if (!jfield_class_Comm) {
+ jxbt_throw_native(env,bprintf("Can't find the org/simgrid/msg/Comm class."));
+ return;
+ }
+ jcomm_field_Comm_bind = jxbt_get_jfield(env, jfield_class_Comm, "bind", "J");
+ jcomm_field_Comm_taskBind = jxbt_get_jfield(env, jfield_class_Comm, "taskBind", "J");
+ jcomm_field_Comm_receiving = jxbt_get_jfield(env, jfield_class_Comm, "receiving", "Z");
+ jtask_field_Comm_task = jxbt_get_jfield(env, jfield_class_Comm, "task", "Lorg/simgrid/msg/Task;");
+ jcomm_field_Comm_finished = jxbt_get_jfield(env, jfield_class_Comm, "finished", "Z");
+ if (!jcomm_field_Comm_bind || !jcomm_field_Comm_taskBind || !jcomm_field_Comm_receiving || !jtask_field_Comm_task || !jcomm_field_Comm_finished) {
+ jxbt_throw_native(env,bprintf("Can't find some fields in Java class."));
+ }
+}
+