src/jmsg_comm.c

   1 /* Functions related to the java comm instances                                                                                                                                 */
   2
   3 /* Copyright (c) 2012. The SimGrid Team. All rights reserved.                   */
   4 #include "jmsg_comm.h"
   5 #include "jxbt_utilities.h"
   6 #include <msg/msg.h>
   7 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(jmsg);
   8 void jcomm_bind_task(JNIEnv *env, jobject jcomm) {
   9         jfieldID id_receiving = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "receiving", "Z");
  10         jclass jclass = jxbt_get_class(env,"org/simgrid/msg/Comm");
  11         jfieldID id_task = jxbt_get_jfield(env, jclass, "task", "Lorg/simgrid/msg/Task;");
  12         jfieldID id_comm = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "bind", "J");
  13
  14         msg_comm_t comm = (msg_comm_t) (long) (*env)->GetLongField(env, jcomm, id_comm);
  15
  16         //test if we are receiving or sending a task.
  17         jboolean jreceiving = (*env)->GetBooleanField(env, jcomm, id_receiving);
  18         if (jreceiving == JNI_TRUE) {
  19                 //bind the task object.
  20                 m_task_t task = MSG_comm_get_task(comm);
  21                 xbt_assert(task != NULL, "Task is NULL");
  22                 jobject jtask_global = MSG_task_get_data(task);
  23                 //case where the data has already been retrieved
  24                 if (jtask_global == NULL)
  25                 {
  26                         return;
  27                 }
  28
  29                 //Make sure the data will be correctly gc.
  30                 jobject jtask_local = (*env)->NewLocalRef(env, jtask_global);
  31                 (*env)->DeleteGlobalRef(env, jtask_global);
  32
  33                 (*env)->SetObjectField(env, jcomm, id_task, jtask_local);
  34
  35                 MSG_task_set_data(task, NULL);
  36         }
  37
  38 }
  39 void jcomm_throw(JNIEnv *env, MSG_error_t status) {
  40         switch (status) {
  41                 case MSG_TIMEOUT:
  42                         jxbt_throw_time_out_failure(env,NULL);
  43                 break;
  44                 case MSG_TRANSFER_FAILURE:
  45                         jxbt_throw_transfer_failure(env,NULL);
  46                 break;
  47                 case MSG_HOST_FAILURE:
  48                         jxbt_throw_host_failure(env,NULL);
  49                 break;
  50                 default:
  51                         jxbt_throw_native(env,bprintf("communication failed"));
  52         }
  53 }
  54 JNIEXPORT void JNICALL
  55 Java_org_simgrid_msg_Comm_unbind(JNIEnv *env, jobject jcomm) {
  56         jfieldID id = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "bind", "J");
  57         jfieldID id_task = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "bindTask", "J");
  58         msg_comm_t comm;
  59         m_task_t *task_received;
  60         if (!id || !id_task)
  61                 return;
  62
  63         task_received = (m_task_t*)  (long) (*env)->GetLongField(env, jcomm, id_task);
  64         if (task_received != NULL) {
  65                 xbt_free(task_received);
  66         }
  67
  68         comm = (msg_comm_t) (long) (*env)->GetLongField(env, jcomm, id);
  69         MSG_comm_destroy(comm);
  70 }
  71
  72 JNIEXPORT jboolean JNICALL
  73 Java_org_simgrid_msg_Comm_test(JNIEnv *env, jobject jcomm) {
  74         msg_comm_t comm;
  75
  76         jfieldID idComm = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "bind", "J");
  77         jclass jclass = jxbt_get_class(env,"org/simgrid/msg/Comm");
  78         jfieldID idTask = jxbt_get_jfield(env, jclass, "task", "Lorg/simgrid/msg/Task;");
  79
  80
  81         if (!jclass || !idComm || !idTask) {
  82                 jxbt_throw_native(env,bprintf("idTask/idComm/jclass/idComm is null"));
  83                 return JNI_FALSE;
  84         }
  85
  86         comm = (msg_comm_t) (long) (*env)->GetLongField(env, jcomm, idComm);
  87         if (!comm) {
  88                 jxbt_throw_native(env,bprintf("comm is null"));
  89                 return JNI_FALSE;
  90         }
  91         if (MSG_comm_test(comm)) {
  92                 MSG_error_t status = MSG_comm_get_status(comm);
  93                 if (status == MSG_OK) {
  94                         jcomm_bind_task(env,jcomm);
  95                         return JNI_TRUE;
  96                 }
  97                 else {
  98                         //send the correct exception
  99                         jcomm_throw(env,status);
 100                         return JNI_FALSE;
 101                 }
 102         }
 103         else {
 104                 return JNI_FALSE;
 105         }
 106 }
 107 JNIEXPORT void JNICALL
 108 Java_org_simgrid_msg_Comm_waitCompletion(JNIEnv *env, jobject jcomm, jdouble timeout) {
 109         jclass jclass = jxbt_get_class(env,"org/simgrid/msg/Comm");
 110         jfieldID id_comm = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "bind", "J");
 111         jfieldID id_task = jxbt_get_jfield(env, jclass, "task", "Lorg/simgrid/msg/Task;");
 112         jfieldID id_receiving = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "receiving", "Z");
 113
 114         if (!jclass || !id_comm || !id_task || !id_receiving) {
 115                 jxbt_throw_native(env,bprintf("idTask/idComm/jclass/idComm is null"));
 116                 return;
 117         }
 118
 119         msg_comm_t comm = (msg_comm_t) (long) (*env)->GetLongField(env, jcomm, id_comm);
 120         if (!comm) {
 121                 jxbt_throw_native(env,bprintf("comm is null"));
 122                 return;
 123         }
 124
 125         MSG_error_t status = MSG_comm_wait(comm,(double)timeout);
 126         if (status == MSG_OK) {
 127                 jcomm_bind_task(env,jcomm);
 128                 return;
 129         }
 130         else {
 131                 jcomm_throw(env,status);
 132         }
 133
 134
 135 }