X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/b7233718def37e1bc1e2a469ac45c738fab0c582..50c886de0c60029fb70b0441f3f021ceb4528424:/src/jmsg_comm.c

diff --git a/src/jmsg_comm.c b/src/jmsg_comm.c
index 3ebc04e449..976cf7e10b 100644
--- a/src/jmsg_comm.c
+++ b/src/jmsg_comm.c
@@ -3,26 +3,28 @@
 /* Copyright (c) 2012. The SimGrid Team. All rights reserved.                   */
 #include "jmsg_comm.h"
 #include "jxbt_utilities.h"
+#include "jmsg.h"
+
 #include <msg/msg.h>
 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(jmsg);
-void jcomm_bind_task(JNIEnv *env, jobject jcomm) {
-	jfieldID id_receiving = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "receiving", "Z");
-	jclass jclass = jxbt_get_class(env,"org/simgrid/msg/Comm");
-	jfieldID id_task = jxbt_get_jfield(env, jclass, "task", "Lorg/simgrid/msg/Task;");
-	jfieldID id_comm = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "bind", "J");
 
-	msg_comm_t comm = (msg_comm_t) (long) (*env)->GetLongField(env, jcomm, id_comm);
+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, id_receiving);
+	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)
-		{
+		if (jtask_global == NULL) {
 			return;
 		}
 
@@ -30,105 +32,105 @@ void jcomm_bind_task(JNIEnv *env, jobject jcomm) {
 		jobject jtask_local = (*env)->NewLocalRef(env, jtask_global);
 		(*env)->DeleteGlobalRef(env, jtask_global);
 
-		(*env)->SetObjectField(env, jcomm, id_task, jtask_local);
+		(*env)->SetObjectField(env, jcomm, jtask_field_Comm_task, jtask_local);
 
 		MSG_task_set_data(task, NULL);
 	}
 
 }
-void jcomm_throw(JNIEnv *env, MSG_error_t status) {
-	switch (status) {
-		case MSG_TIMEOUT:
-			jxbt_throw_time_out_failure(env,NULL);
-		break;
-		case MSG_TRANSFER_FAILURE:
-			jxbt_throw_transfer_failure(env,NULL);
-		break;
-		case MSG_HOST_FAILURE:
-			jxbt_throw_host_failure(env,NULL);
-		break;
-		default:
-			jxbt_throw_native(env,bprintf("communication failed"));
+
+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."));
 	}
 }
+
 JNIEXPORT void JNICALL
-Java_org_simgrid_msg_Comm_unbind(JNIEnv *env, jobject jcomm) {
-	jfieldID id = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "bind", "J");
-	jfieldID id_task = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "bindTask", "J");
+Java_org_simgrid_msg_Comm_destroy(JNIEnv *env, jobject jcomm) {
 	msg_comm_t comm;
 	m_task_t *task_received;
-	if (!id || !id_task)
-		return;
 
-	task_received = (m_task_t*)  (long) (*env)->GetLongField(env, jcomm, id_task);
-	if (task_received != NULL) {
-		xbt_free(task_received);
-	}
+	task_received = (m_task_t*)  (long) (*env)->GetLongField(env, jcomm, jcomm_field_Comm_taskBind);
+	xbt_free(task_received);
 
-	comm = (msg_comm_t) (long) (*env)->GetLongField(env, jcomm, id);
+	comm = (msg_comm_t) (long) (*env)->GetLongField(env, jcomm, jcomm_field_Comm_bind);
 	MSG_comm_destroy(comm);
 }
 
 JNIEXPORT jboolean JNICALL
 Java_org_simgrid_msg_Comm_test(JNIEnv *env, jobject jcomm) {
 	msg_comm_t comm;
+	comm = (msg_comm_t) (long) (*env)->GetLongField(env, jcomm, jcomm_field_Comm_bind);
 
-	jfieldID idComm = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "bind", "J");
-	jclass jclass = jxbt_get_class(env,"org/simgrid/msg/Comm");
-	jfieldID idTask = jxbt_get_jfield(env, jclass, "task", "Lorg/simgrid/msg/Task;");
-
-
-	if (!jclass || !idComm || !idTask) {
-		jxbt_throw_native(env,bprintf("idTask/idComm/jclass/idComm is null"));
-		return JNI_FALSE;
+	jboolean finished = (*env)->GetBooleanField(env, jcomm, jcomm_field_Comm_finished);
+	if (finished == JNI_TRUE) {
+		return JNI_TRUE;
 	}
 
-	comm = (msg_comm_t) (long) (*env)->GetLongField(env, jcomm, idComm);
 	if (!comm) {
 		jxbt_throw_native(env,bprintf("comm is null"));
 		return JNI_FALSE;
 	}
-	if (MSG_comm_test(comm)) {
-		MSG_error_t status = MSG_comm_get_status(comm);
-		if (status == MSG_OK) {
-			jcomm_bind_task(env,jcomm);
-			return JNI_TRUE;
+	TRY {
+		if (MSG_comm_test(comm)) {
+			MSG_error_t status = MSG_comm_get_status(comm);
+			if (status == MSG_OK) {
+				jcomm_bind_task(env,jcomm);
+				return JNI_TRUE;
+			}
+			else {
+				//send the correct exception
+				jmsg_throw_status(env,status);
+				return JNI_FALSE;
+			}
 		}
 		else {
-			//send the correct exception
-			jcomm_throw(env,status);
 			return JNI_FALSE;
 		}
 	}
-	else {
-		return JNI_FALSE;
+	CATCH_ANONYMOUS {
+
 	}
+	return JNI_FALSE;
 }
 JNIEXPORT void JNICALL
 Java_org_simgrid_msg_Comm_waitCompletion(JNIEnv *env, jobject jcomm, jdouble timeout) {
-	jclass jclass = jxbt_get_class(env,"org/simgrid/msg/Comm");
-	jfieldID id_comm = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "bind", "J");
-	jfieldID id_task = jxbt_get_jfield(env, jclass, "task", "Lorg/simgrid/msg/Task;");
-	jfieldID id_receiving = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "receiving", "Z");
-
-	if (!jclass || !id_comm || !id_task || !id_receiving) {
-		jxbt_throw_native(env,bprintf("idTask/idComm/jclass/idComm is null"));
+	msg_comm_t comm = (msg_comm_t) (long) (*env)->GetLongField(env, jcomm, jcomm_field_Comm_bind);
+	if (!comm) {
+		jxbt_throw_native(env,bprintf("comm is null"));
 		return;
 	}
 
-	msg_comm_t comm = (msg_comm_t) (long) (*env)->GetLongField(env, jcomm, id_comm);
-	if (!comm) {
-		jxbt_throw_native(env,bprintf("comm is null"));
+	jboolean finished = (*env)->GetBooleanField(env, jcomm, jcomm_field_Comm_finished);
+	if (finished == JNI_TRUE) {
 		return;
 	}
 
-	MSG_error_t status = MSG_comm_wait(comm,(double)timeout);
+	MSG_error_t status;
+	TRY {
+		status = MSG_comm_wait(comm,(double)timeout);
+	}
+	CATCH_ANONYMOUS {
+		return;
+	}
+	(*env)->SetBooleanField(env, jcomm, jcomm_field_Comm_finished, JNI_TRUE);
 	if (status == MSG_OK) {
 		jcomm_bind_task(env,jcomm);
 		return;
 	}
 	else {
-		jcomm_throw(env,status);
+		jmsg_throw_status(env,status);
 	}