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