src/jmsg_task.c

   1 /* Functions related to the java task instances.                            */
   2
   3 /* Copyright (c) 2007, 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 "jmsg.h"
  10
  11 #include "smx_context_java.h"
  12
  13 #include "jmsg_host.h"
  14 #include "jmsg_task.h"
  15
  16 #include "jxbt_utilities.h"
  17
  18 #include <msg/msg.h>
  19 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(jmsg);
  20
  21 static jmethodID jtask_field_Comm_constructor;
  22
  23 static jfieldID jtask_field_Task_bind;
  24 static jfieldID jtask_field_Comm_bind;
  25 static jfieldID jtask_field_Comm_taskBind;
  26 static jfieldID jtask_field_Comm_receiving;
  27
  28 void jtask_bind(jobject jtask, m_task_t task, JNIEnv * env)
  29 {
  30   (*env)->SetLongField(env, jtask, jtask_field_Task_bind, (jlong) (long) (task));
  31 }
  32
  33 m_task_t jtask_to_native_task(jobject jtask, JNIEnv * env)
  34 {
  35   return (m_task_t) (long) (*env)->GetLongField(env, jtask, jtask_field_Task_bind);
  36 }
  37
  38 jboolean jtask_is_valid(jobject jtask, JNIEnv * env)
  39 {
  40   return (*env)->GetLongField(env, jtask, jtask_field_Task_bind) ? JNI_TRUE : JNI_FALSE;
  41 }
  42
  43 JNIEXPORT void JNICALL
  44 Java_org_simgrid_msg_Task_nativeInit(JNIEnv *env, jclass cls) {
  45         jclass jtask_class_Comm = (*env)->FindClass(env, "org/simgrid/msg/Comm");
  46
  47         jtask_field_Comm_constructor = (*env)->GetMethodID(env, jtask_class_Comm, "<init>", "()V");
  48         //FIXME: Don't use jxbt_get_sfield directly.
  49         jtask_field_Task_bind = jxbt_get_sfield(env, "org/simgrid/msg/Task", "bind", "J");
  50         jtask_field_Comm_bind = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "bind", "J");
  51         jtask_field_Comm_taskBind = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "taskBind", "J");
  52         jtask_field_Comm_receiving = jxbt_get_sfield(env, "org/simgrid/msg/Comm", "receiving", "Z");
  53 }
  54
  55 JNIEXPORT void JNICALL
  56 Java_org_simgrid_msg_Task_create(JNIEnv * env,
  57                                       jobject jtask, jstring jname,
  58                                       jdouble jcomputeDuration,
  59                                       jdouble jmessageSize)
  60 {
  61   m_task_t task;                /* the native task to create                            */
  62   const char *name = NULL;      /* the name of the task                                 */
  63
  64   if (jcomputeDuration < 0) {
  65     jxbt_throw_illegal(env,
  66                        bprintf
  67                        ("Task ComputeDuration (%f) cannot be negative",
  68                         (double) jcomputeDuration));
  69     return;
  70   }
  71
  72   if (jmessageSize < 0) {
  73     jxbt_throw_illegal(env,
  74                        bprintf("Task MessageSize (%f) cannot be negative",
  75                                (double) jmessageSize));
  76     return;
  77   }
  78
  79   if (jname) {
  80     /* get the C string from the java string */
  81     name = (*env)->GetStringUTFChars(env, jname, 0);
  82   }
  83
  84
  85   /* create the task */
  86   task =
  87       MSG_task_create(name, (double) jcomputeDuration,
  88                       (double) jmessageSize, NULL);
  89   if (jname)
  90     (*env)->ReleaseStringUTFChars(env, jname, name);
  91
  92   /* bind & store the task */
  93   jtask_bind(jtask, task, env);
  94   MSG_task_set_data(task, jtask);
  95 }
  96
  97 JNIEXPORT void JNICALL
  98 Java_org_simgrid_msg_Task_parallelCreate(JNIEnv * env,
  99                                                jobject jtask,
 100                                                jstring jname,
 101                                                jobjectArray jhosts,
 102                                                jdoubleArray
 103                                                jcomputeDurations_arg,
 104                                                jdoubleArray
 105                                                jmessageSizes_arg) {
 106
 107   m_task_t task;                /* the native parallel task to create           */
 108   const char *name;             /* the name of the task                         */
 109   int host_count;
 110   m_host_t *hosts;
 111   double *computeDurations;
 112   double *messageSizes;
 113   jdouble *jcomputeDurations;
 114   jdouble *jmessageSizes;
 115
 116   jobject jhost;
 117   int index;
 118
 119
 120   if (!jcomputeDurations_arg) {
 121     jxbt_throw_null(env,
 122                     xbt_strdup
 123                     ("Parallel task compute durations cannot be null"));
 124     return;
 125   }
 126
 127   if (!jmessageSizes_arg) {
 128     jxbt_throw_null(env,
 129                     xbt_strdup
 130                     ("Parallel task message sizes cannot be null"));
 131     return;
 132   }
 133
 134   if (!jname) {
 135     jxbt_throw_null(env, xbt_strdup("Parallel task name cannot be null"));
 136     return;
 137   }
 138
 139   host_count = (int) (*env)->GetArrayLength(env, jhosts);
 140
 141
 142   hosts = xbt_new0(m_host_t, host_count);
 143   computeDurations = xbt_new0(double, host_count);
 144   messageSizes = xbt_new0(double, host_count * host_count);
 145
 146   jcomputeDurations =
 147       (*env)->GetDoubleArrayElements(env, jcomputeDurations_arg, 0);
 148   jmessageSizes =
 149       (*env)->GetDoubleArrayElements(env, jmessageSizes_arg, 0);
 150
 151   for (index = 0; index < host_count; index++) {
 152     jhost = (*env)->GetObjectArrayElement(env, jhosts, index);
 153     hosts[index] = jhost_get_native(env, jhost);
 154     computeDurations[index] = jcomputeDurations[index];
 155   }
 156   for (index = 0; index < host_count * host_count; index++) {
 157     messageSizes[index] = jmessageSizes[index];
 158   }
 159
 160   (*env)->ReleaseDoubleArrayElements(env, jcomputeDurations_arg,
 161                                      jcomputeDurations, 0);
 162   (*env)->ReleaseDoubleArrayElements(env, jmessageSizes_arg, jmessageSizes,
 163                                      0);
 164
 165
 166   /* get the C string from the java string */
 167   name = (*env)->GetStringUTFChars(env, jname, 0);
 168
 169   task =
 170       MSG_parallel_task_create(name, host_count, hosts, computeDurations,
 171                                messageSizes, NULL);
 172
 173   (*env)->ReleaseStringUTFChars(env, jname, name);
 174
 175   /* associate the java task object and the native task */
 176   jtask_bind(jtask, task, env);
 177
 178   MSG_task_set_data(task, (void *) jtask);
 179
 180   if (!MSG_task_get_data(task))
 181     jxbt_throw_jni(env, "global ref allocation failed");
 182 }
 183
 184 JNIEXPORT void JNICALL
 185 Java_org_simgrid_msg_Task_cancel(JNIEnv * env,
 186                                       jobject jtask)
 187 {
 188   m_task_t ptask = jtask_to_native_task(jtask, env);
 189
 190   if (!ptask) {
 191     jxbt_throw_notbound(env, "task", jtask);
 192     return;
 193   }
 194
 195   MSG_error_t rv = MSG_task_cancel(ptask);
 196
 197   jxbt_check_res("MSG_task_cancel()", rv, MSG_OK,
 198                  bprintf("unexpected error , please report this bug"));
 199 }
 200
 201 JNIEXPORT void JNICALL
 202 Java_org_simgrid_msg_Task_execute(JNIEnv * env,
 203                                        jobject jtask)
 204 {
 205   m_task_t task = jtask_to_native_task(jtask, env);
 206
 207   if (!task) {
 208     jxbt_throw_notbound(env, "task", jtask);
 209     return;
 210   }
 211
 212   MSG_error_t rv = MSG_task_execute(task);
 213
 214   jxbt_check_res("MSG_task_execute()", rv,
 215                  MSG_HOST_FAILURE | MSG_TASK_CANCELED,
 216                  bprintf("while executing task %s",
 217                          MSG_task_get_name(task)));
 218 }
 219
 220 JNIEXPORT jstring JNICALL
 221 Java_org_simgrid_msg_Task_getName(JNIEnv * env,
 222                                        jobject jtask) {
 223   m_task_t task = jtask_to_native_task(jtask, env);
 224
 225   if (!task) {
 226     jxbt_throw_notbound(env, "task", jtask);
 227     return NULL;
 228   }
 229
 230   return (*env)->NewStringUTF(env, MSG_task_get_name(task));
 231 }
 232
 233 JNIEXPORT jobject JNICALL
 234 Java_org_simgrid_msg_Task_getSender(JNIEnv * env,
 235                                          jobject jtask) {
 236   m_process_t process;
 237
 238   m_task_t task = jtask_to_native_task(jtask, env);
 239
 240   if (!task) {
 241     jxbt_throw_notbound(env, "task", jtask);
 242     return NULL;
 243   }
 244
 245   process = MSG_task_get_sender(task);
 246   return (jobject) native_to_java_process(process);
 247 }
 248
 249 JNIEXPORT jobject JNICALL
 250 Java_org_simgrid_msg_Task_getSource(JNIEnv * env,
 251                                          jobject jtask)
 252 {
 253   m_host_t host;
 254   m_task_t task = jtask_to_native_task(jtask, env);
 255
 256   if (!task) {
 257     jxbt_throw_notbound(env, "task", jtask);
 258     return NULL;
 259   }
 260
 261   host = MSG_task_get_source(task);
 262
 263   if (!MSG_host_get_data(host)) {
 264     jxbt_throw_jni(env, "MSG_task_get_source() failed");
 265     return NULL;
 266   }
 267
 268   return (jobject) MSG_host_get_data(host);
 269 }
 270
 271 JNIEXPORT jdouble JNICALL
 272 Java_org_simgrid_msg_Task_getComputeDuration(JNIEnv * env,
 273                                                   jobject jtask)
 274 {
 275   m_task_t ptask = jtask_to_native_task(jtask, env);
 276
 277   if (!ptask) {
 278     jxbt_throw_notbound(env, "task", jtask);
 279     return -1;
 280   }
 281   return (jdouble) MSG_task_get_compute_duration(ptask);
 282 }
 283
 284 JNIEXPORT jdouble JNICALL
 285 Java_org_simgrid_msg_Task_getRemainingDuration(JNIEnv * env, jobject jtask)
 286 {
 287   m_task_t ptask = jtask_to_native_task(jtask, env);
 288
 289   if (!ptask) {
 290     jxbt_throw_notbound(env, "task", jtask);
 291     return -1;
 292   }
 293   return (jdouble) MSG_task_get_remaining_computation(ptask);
 294 }
 295
 296 JNIEXPORT void JNICALL
 297 Java_org_simgrid_msg_Task_setPriority(JNIEnv * env,
 298                                            jobject jtask, jdouble priority)
 299 {
 300   m_task_t task = jtask_to_native_task(jtask, env);
 301
 302   if (!task) {
 303     jxbt_throw_notbound(env, "task", jtask);
 304     return;
 305   }
 306   MSG_task_set_priority(task, (double) priority);
 307 }
 308 JNIEXPORT void JNICALL
 309 Java_org_simgrid_msg_Task_send(JNIEnv * env,jobject jtask,
 310                                     jstring jalias,
 311                                     jdouble jtimeout)
 312 {
 313   MSG_error_t rv;
 314   const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
 315
 316   m_task_t task = jtask_to_native_task(jtask, env);
 317
 318
 319   if (!task) {
 320     (*env)->ReleaseStringUTFChars(env, jalias, alias);
 321     jxbt_throw_notbound(env, "task", jtask);
 322     return;
 323   }
 324
 325   /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
 326   MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
 327   rv = MSG_task_send_with_timeout(task, alias, (double) jtimeout);
 328
 329   (*env)->ReleaseStringUTFChars(env, jalias, alias);
 330
 331   jxbt_check_res("MSG_task_send_with_timeout()", rv,
 332                  MSG_HOST_FAILURE | MSG_TRANSFER_FAILURE | MSG_TIMEOUT,
 333                  bprintf("while sending task %s to mailbox %s",
 334                          MSG_task_get_name(task), alias));
 335 }
 336
 337 JNIEXPORT void JNICALL
 338 Java_org_simgrid_msg_Task_sendBounded(JNIEnv * env, jobject jtask,
 339                                            jstring jalias,
 340                                            jdouble jmaxRate)
 341 {
 342   m_task_t task = jtask_to_native_task(jtask, env);
 343   MSG_error_t rv;
 344   const char *alias;
 345
 346   if (!task) {
 347     jxbt_throw_notbound(env, "task", jtask);
 348     return;
 349   }
 350
 351   alias = (*env)->GetStringUTFChars(env, jalias, 0);
 352
 353   /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
 354   MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
 355   rv = MSG_task_send_bounded(task, alias, (double) jmaxRate);
 356
 357   (*env)->ReleaseStringUTFChars(env, jalias, alias);
 358
 359   jxbt_check_res("MSG_task_send_bounded()", rv,
 360                  MSG_HOST_FAILURE | MSG_TRANSFER_FAILURE | MSG_TIMEOUT,
 361                  bprintf
 362                  ("while sending task %s to mailbox %s with max rate %f",
 363                   MSG_task_get_name(task), alias, (double) jmaxRate));
 364
 365 }
 366
 367
 368 JNIEXPORT jobject JNICALL
 369 Java_org_simgrid_msg_Task_receive(JNIEnv * env, jclass cls,
 370                                        jstring jalias, jdouble jtimeout,
 371                                        jobject jhost)
 372 {
 373   MSG_error_t rv;
 374   m_task_t task = NULL;
 375   m_host_t host = NULL;
 376   jobject jtask_global, jtask_local;
 377   const char *alias;
 378
 379   if (jhost) {
 380     host = jhost_get_native(env, jhost);
 381
 382     if (!host) {
 383       jxbt_throw_notbound(env, "host", jhost);
 384       return NULL;
 385     }
 386   }
 387
 388   alias = (*env)->GetStringUTFChars(env, jalias, 0);
 389
 390   rv = MSG_task_receive_ext(&task, alias, (double) jtimeout, host);
 391   if (rv != MSG_OK) {
 392         switch (rv) {
 393                 case MSG_TIMEOUT:
 394                         jxbt_throw_time_out_failure(env,NULL);
 395                 break;
 396                 case MSG_TRANSFER_FAILURE:
 397                         jxbt_throw_transfer_failure(env,NULL);
 398                 break;
 399                 case MSG_HOST_FAILURE:
 400                         jxbt_throw_host_failure(env,NULL);
 401                 break;
 402                 default:
 403                         jxbt_throw_native(env,bprintf("receive failed"));
 404         }
 405         return NULL;
 406   }
 407   jtask_global = MSG_task_get_data(task);
 408
 409   /* Convert the global ref into a local ref so that the JVM can free the stuff */
 410   jtask_local = (*env)->NewLocalRef(env, jtask_global);
 411   (*env)->DeleteGlobalRef(env, jtask_global);
 412   MSG_task_set_data(task, NULL);
 413
 414   (*env)->ReleaseStringUTFChars(env, jalias, alias);
 415
 416   jxbt_check_res("MSG_task_receive_ext()", rv,
 417                  MSG_HOST_FAILURE | MSG_TRANSFER_FAILURE | MSG_TIMEOUT,
 418                  bprintf("while receiving from mailbox %s", alias));
 419
 420   return (jobject) jtask_local;
 421 }
 422
 423
 424 JNIEXPORT jobject JNICALL
 425 Java_org_simgrid_msg_Task_irecv(JNIEnv * env, jclass cls, jstring jmailbox) {
 426         msg_comm_t comm;
 427         const char *mailbox;
 428         jclass comm_class;
 429         //pointer to store the task object pointer.
 430         m_task_t *task = xbt_new(m_task_t,1);
 431         *task = NULL;
 432         /* There should be a cache here */
 433         comm_class = (*env)->FindClass(env, "org/simgrid/msg/Comm");
 434
 435         if (!comm_class) {
 436                 jxbt_throw_native(env,bprintf("fieldID or methodID or class not found."));
 437                 return NULL;
 438         }
 439
 440         jobject jcomm = (*env)->NewObject(env, comm_class, jtask_field_Comm_constructor);
 441         if (!jcomm) {
 442                 jxbt_throw_native(env,bprintf("Can't create a Comm object."));
 443                 return NULL;
 444         }
 445
 446         mailbox = (*env)->GetStringUTFChars(env, jmailbox, 0);
 447
 448         comm = MSG_task_irecv(task,mailbox);
 449
 450         (*env)->SetLongField(env, jcomm, jtask_field_Comm_bind, (jlong) (long)(comm));
 451         (*env)->SetLongField(env, jcomm, jtask_field_Comm_taskBind, (jlong) (long)(task));
 452         (*env)->SetBooleanField(env, jcomm, jtask_field_Comm_receiving, JNI_TRUE);
 453
 454         (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
 455
 456         return jcomm;
 457 }
 458
 459 JNIEXPORT jobject JNICALL
 460 Java_org_simgrid_msg_Task_isend(JNIEnv *env, jobject jtask, jstring jmailbox) {
 461         jclass comm_class;
 462
 463         const char *mailbox;
 464
 465         m_task_t task;
 466
 467         jobject jcomm;
 468         msg_comm_t comm;
 469
 470         comm_class = (*env)->FindClass(env, "org/simgrid/msg/Comm");
 471
 472         if (!comm_class) return NULL;
 473
 474         jcomm = (*env)->NewObject(env, comm_class, jtask_field_Comm_constructor);
 475         mailbox = (*env)->GetStringUTFChars(env, jmailbox, 0);
 476
 477         task = jtask_to_native_task(jtask, env);
 478
 479         if (!task) {
 480     (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
 481     (*env)->DeleteLocalRef(env, jcomm);
 482     jxbt_throw_notbound(env, "task", jtask);
 483                 return NULL;
 484         }
 485
 486   MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
 487         comm = MSG_task_isend(task,mailbox);
 488
 489         (*env)->SetLongField(env, jcomm, jtask_field_Comm_bind, (jlong) (long)(comm));
 490         (*env)->SetLongField(env, jcomm, jtask_field_Comm_taskBind, (jlong) (long)(NULL));
 491         (*env)->SetBooleanField(env, jcomm, jtask_field_Comm_receiving, JNI_FALSE);
 492
 493         (*env)->ReleaseStringUTFChars(env, jmailbox, mailbox);
 494
 495         return jcomm;
 496 }
 497
 498 static void msg_task_cancel_on_failed_dsend(void*t) {
 499         m_task_t task = t;
 500         JNIEnv *env =get_current_thread_env();
 501         jobject jtask_global = MSG_task_get_data(task);
 502
 503         /* Destroy the global ref so that the JVM can free the stuff */
 504         (*env)->DeleteGlobalRef(env, jtask_global);
 505         MSG_task_set_data(task, NULL);
 506         MSG_task_destroy(task);
 507 }
 508
 509 JNIEXPORT void JNICALL
 510 Java_org_simgrid_msg_Task_dsend(JNIEnv * env, jobject jtask,
 511                                 jstring jalias) {
 512
 513   const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
 514
 515   m_task_t task = jtask_to_native_task(jtask, env);
 516
 517
 518   if (!task) {
 519     (*env)->ReleaseStringUTFChars(env, jalias, alias);
 520     jxbt_throw_notbound(env, "task", jtask);
 521     return;
 522   }
 523
 524   /* Pass a global ref to the Jtask into the Ctask so that the receiver can use it */
 525   MSG_task_set_data(task, (void *) (*env)->NewGlobalRef(env, jtask));
 526   MSG_task_dsend(task, alias, msg_task_cancel_on_failed_dsend);
 527
 528   (*env)->ReleaseStringUTFChars(env, jalias, alias);
 529 }
 530
 531 JNIEXPORT jboolean JNICALL
 532 Java_org_simgrid_msg_Task_listen(JNIEnv * env, jclass cls,
 533                                                                                                            jstring jalias) {
 534
 535   const char *alias;
 536   int rv;
 537
 538   alias = (*env)->GetStringUTFChars(env, jalias, 0);
 539
 540   rv = MSG_task_listen(alias);
 541
 542   (*env)->ReleaseStringUTFChars(env, jalias, alias);
 543
 544   return (jboolean) rv;
 545 }
 546
 547 JNIEXPORT jint JNICALL
 548 Java_org_simgrid_msg_Task_listenFromHost(JNIEnv * env, jclass cls,
 549                                                                                                                                            jstring jalias,
 550                                                                                                                                            jobject jhost) {
 551   int rv;
 552   const char *alias;
 553
 554   m_host_t host = jhost_get_native(env, jhost);
 555
 556   if (!host) {
 557     jxbt_throw_notbound(env, "host", jhost);
 558     return -1;
 559   }
 560   alias = (*env)->GetStringUTFChars(env, jalias, 0);
 561
 562   rv = MSG_task_listen_from_host(alias, host);
 563
 564   (*env)->ReleaseStringUTFChars(env, jalias, alias);
 565
 566   return (jint) rv;
 567 }
 568
 569
 570 JNIEXPORT jint JNICALL
 571 Java_org_simgrid_msg_Task_listenFrom(JNIEnv * env, jclass cls,
 572                                                         jstring jalias) {
 573
 574   int rv;
 575   const char *alias = (*env)->GetStringUTFChars(env, jalias, 0);
 576
 577   rv = MSG_task_listen_from(alias);
 578
 579   (*env)->ReleaseStringUTFChars(env, jalias, alias);
 580
 581   return (jint) rv;
 582 }