simgrid-java/src/jmsg_process.c

   1 /* Functions related to the java process instances.                         */
   2
   3 /* Copyright (c) 2007-2012. 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 #include "jmsg_process.h"
   9
  10 #include "jmsg.h"
  11 #include "jmsg_host.h"
  12 #include "jxbt_utilities.h"
  13 #include "smx_context_java.h"
  14 #include "smx_context_cojava.h"
  15
  16 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(jmsg);
  17
  18 JNIEXPORT void JNICALL
  19 Java_org_simgrid_msg_Process_exit(JNIEnv *env, jobject jprocess) {
  20   if (smx_factory_initializer_to_use == SIMIX_ctx_cojava_factory_init) {
  21     msg_process_t process = jprocess_to_native_process(jprocess, env);
  22     smx_context_t context = MSG_process_get_smx_ctx(process);
  23     smx_ctx_cojava_stop(context);
  24   }
  25 }
  26
  27 jobject native_to_java_process(msg_process_t process)
  28 {
  29   return ((smx_ctx_java_t)MSG_process_get_smx_ctx(process))->jprocess;
  30 }
  31
  32 jobject jprocess_new_global_ref(jobject jprocess, JNIEnv * env)
  33 {
  34   return (*env)->NewGlobalRef(env, jprocess);
  35 }
  36
  37 void jprocess_delete_global_ref(jobject jprocess, JNIEnv * env)
  38 {
  39   (*env)->DeleteGlobalRef(env, jprocess);
  40 }
  41
  42 void jprocess_join(jobject jprocess, JNIEnv * env)
  43 {
  44         msg_process_t process = jprocess_to_native_process(jprocess,env);
  45         smx_ctx_java_t context = (smx_ctx_java_t)MSG_process_get_smx_ctx(process);
  46         xbt_os_thread_join(context->thread,NULL);
  47 }
  48
  49 msg_process_t jprocess_to_native_process(jobject jprocess, JNIEnv * env)
  50 {
  51   return (msg_process_t) (long) (*env)->GetLongField(env, jprocess, jprocess_field_Process_bind);
  52 }
  53
  54 void jprocess_bind(jobject jprocess, msg_process_t process, JNIEnv * env)
  55 {
  56   (*env)->SetLongField(env, jprocess, jprocess_field_Process_bind, (jlong)(process));
  57 }
  58
  59 jlong jprocess_get_id(jobject jprocess, JNIEnv * env)
  60 {
  61   return (*env)->GetLongField(env, jprocess, jprocess_field_Process_id);
  62 }
  63
  64 jstring jprocess_get_name(jobject jprocess, JNIEnv * env)
  65 {
  66   jstring jname = (jstring) (*env)->GetObjectField(env, jprocess, jprocess_field_Process_name);
  67   return (*env)->NewGlobalRef(env, jname);
  68
  69 }
  70
  71 jboolean jprocess_is_valid(jobject jprocess, JNIEnv * env)
  72 {
  73   jfieldID id = jxbt_get_sfield(env, "org/simgrid/msg/Process", "bind", "J");
  74
  75   if (!id)
  76     return JNI_FALSE;
  77
  78   return (*env)->GetLongField(env, jprocess, id) ? JNI_TRUE : JNI_FALSE;
  79 }
  80 JNIEXPORT void JNICALL
  81 Java_org_simgrid_msg_Process_nativeInit(JNIEnv *env, jclass cls) {
  82   jclass jprocess_class_Process = (*env)->FindClass(env, "org/simgrid/msg/Process");
  83
  84   jprocess_field_Process_name = jxbt_get_jfield(env, jprocess_class_Process, "name", "Ljava/lang/String;");
  85   jprocess_field_Process_bind = jxbt_get_jfield(env, jprocess_class_Process, "bind", "J");
  86   jprocess_field_Process_id = jxbt_get_jfield(env, jprocess_class_Process, "id", "J");
  87   jprocess_field_Process_pid = jxbt_get_jfield(env, jprocess_class_Process, "pid", "I");
  88   jprocess_field_Process_ppid = jxbt_get_jfield(env, jprocess_class_Process, "ppid", "I");
  89   jprocess_field_Process_host = jxbt_get_jfield(env, jprocess_class_Process, "host", "Lorg/simgrid/msg/Host;");
  90   jprocess_field_Process_killTime = jxbt_get_jfield(env, jprocess_class_Process, "killTime", "D");
  91   if (!jprocess_class_Process || !jprocess_field_Process_id || !jprocess_field_Process_name || !jprocess_field_Process_pid ||
  92       !jprocess_field_Process_ppid || !jprocess_field_Process_host) {
  93     jxbt_throw_native(env,bprintf("Can't find some fields in Java class. You should report this bug."));
  94   }
  95 }
  96 JNIEXPORT void JNICALL
  97 Java_org_simgrid_msg_Process_create(JNIEnv * env,
  98                                     jobject jprocess_arg,
  99                                     jobject jhostname)
 100 {
 101
 102
 103   jobject jprocess;             /* the global reference to the java process instance    */
 104   jstring jname;                /* the name of the java process instance                */
 105   const char *name;             /* the C name of the process                            */
 106   const char *hostname;
 107   msg_process_t process;          /* the native process to create                         */
 108   msg_host_t host;                /* Where that process lives */
 109
 110   hostname = (*env)->GetStringUTFChars(env, jhostname, 0);
 111
 112   /* get the name of the java process */
 113   jname = jprocess_get_name(jprocess_arg, env);
 114   if (!jname) {
 115     jxbt_throw_null(env,
 116             xbt_strdup("Internal error: Process name cannot be NULL"));
 117     return;
 118   }
 119
 120   /* bind/retrieve the msg host */
 121   host = MSG_get_host_by_name(hostname);
 122
 123   if (!(host)) {    /* not binded */
 124     jxbt_throw_host_not_found(env, hostname);
 125     return;
 126   }
 127
 128   /* create a global java process instance */
 129   jprocess = jprocess_new_global_ref(jprocess_arg, env);
 130   if (!jprocess) {
 131     jxbt_throw_jni(env, "Can't get a global ref to the java process");
 132     return;
 133   }
 134
 135   /* build the C name of the process */
 136   name = (*env)->GetStringUTFChars(env, jname, 0);
 137   name = xbt_strdup(name);
 138
 139   /* Retrieve the kill time from the process */
 140   jdouble jkill = (*env)->GetDoubleField(env, jprocess, jprocess_field_Process_killTime);
 141   /* Actually build the MSG process */
 142   process = MSG_process_create_with_environment(name,
 143                                                 (xbt_main_func_t) jprocess,
 144                                                 /*data*/ jprocess,
 145                                                 host,
 146                                                 /*argc, argv, properties*/
 147                                                 0,NULL,NULL);
 148   MSG_process_set_kill_time(process, (double)jkill);
 149   /* bind the java process instance to the native process */
 150   jprocess_bind(jprocess, process, env);
 151
 152   /* release our reference to the process name (variable name becomes invalid) */
 153   //FIXME : This line should be uncommented but with mac it doesn't work. BIG WARNING
 154   //(*env)->ReleaseStringUTFChars(env, jname, name);
 155   (*env)->ReleaseStringUTFChars(env, jhostname, hostname);
 156
 157   /* sets the PID and the PPID of the process */
 158   (*env)->SetIntField(env, jprocess, jprocess_field_Process_pid,(jint) MSG_process_get_PID(process));
 159   (*env)->SetIntField(env, jprocess, jprocess_field_Process_ppid, (jint) MSG_process_get_PPID(process));
 160   /* sets the Host of the process */
 161   jobject jhost = Java_org_simgrid_msg_Host_getByName(env,NULL,jhostname);
 162
 163   (*env)->SetObjectField(env, jprocess, jprocess_field_Process_host, jhost);
 164 }
 165
 166 JNIEXPORT jint JNICALL
 167 Java_org_simgrid_msg_Process_killAll(JNIEnv * env, jclass cls,
 168                                      jint jresetPID)
 169 {
 170   return (jint) MSG_process_killall((int) jresetPID);
 171 }
 172
 173 JNIEXPORT jobject JNICALL
 174 Java_org_simgrid_msg_Process_fromPID(JNIEnv * env, jclass cls,
 175                                      jint PID)
 176 {
 177   msg_process_t process = MSG_process_from_PID(PID);
 178
 179   if (!process) {
 180     jxbt_throw_process_not_found(env, bprintf("PID = %d",(int) PID));
 181     return NULL;
 182   }
 183
 184   jobject jprocess = native_to_java_process(process);
 185
 186   if (!jprocess) {
 187     jxbt_throw_jni(env, "get process failed");
 188     return NULL;
 189   }
 190
 191   return jprocess;
 192 }
 193 JNIEXPORT jobject JNICALL
 194 Java_org_simgrid_msg_Process_getProperty(JNIEnv *env, jobject jprocess, jobject jname) {
 195   msg_process_t process = jprocess_to_native_process(jprocess, env);
 196
 197   if (!process) {
 198     jxbt_throw_notbound(env, "process", jprocess);
 199     return NULL;
 200   }
 201   const char *name = (*env)->GetStringUTFChars(env, jname, 0);
 202
 203   const char *property = MSG_process_get_property_value(process, name);
 204   if (!property) {
 205     return NULL;
 206   }
 207
 208   jobject jproperty = (*env)->NewStringUTF(env, property);
 209
 210   (*env)->ReleaseStringUTFChars(env, jname, name);
 211
 212   return jproperty;
 213 }
 214 JNIEXPORT jobject JNICALL
 215 Java_org_simgrid_msg_Process_currentProcess(JNIEnv * env, jclass cls)
 216 {
 217   msg_process_t process = MSG_process_self();
 218   jobject jprocess;
 219
 220   if (!process) {
 221     jxbt_throw_jni(env, xbt_strdup("MSG_process_self() failed"));
 222     return NULL;
 223   }
 224
 225   jprocess = native_to_java_process(process);
 226
 227   if (!jprocess)
 228     jxbt_throw_jni(env, xbt_strdup("SIMIX_process_get_jprocess() failed"));
 229
 230   return jprocess;
 231 }
 232
 233 JNIEXPORT void JNICALL
 234 Java_org_simgrid_msg_Process_suspend(JNIEnv * env,
 235                                    jobject jprocess)
 236 {
 237   msg_process_t process = jprocess_to_native_process(jprocess, env);
 238
 239   if (!process) {
 240     jxbt_throw_notbound(env, "process", jprocess);
 241     return;
 242   }
 243
 244   /* try to suspend the process */
 245   msg_error_t rv = MSG_process_suspend(process);
 246
 247   jxbt_check_res("MSG_process_suspend()", rv, MSG_OK,
 248                  bprintf("unexpected error , please report this bug"));
 249
 250 }
 251
 252 JNIEXPORT void JNICALL
 253 Java_org_simgrid_msg_Process_resume(JNIEnv * env,
 254                                      jobject jprocess)
 255 {
 256   msg_process_t process = jprocess_to_native_process(jprocess, env);
 257
 258   if (!process) {
 259     jxbt_throw_notbound(env, "process", jprocess);
 260     return;
 261   }
 262
 263   /* try to resume the process */
 264   msg_error_t rv = MSG_process_resume(process);
 265
 266   jxbt_check_res("MSG_process_resume()", rv, MSG_OK,
 267                  bprintf("unexpected error , please report this bug"));
 268 }
 269 JNIEXPORT void JNICALL
 270 Java_org_simgrid_msg_Process_setAutoRestart
 271     (JNIEnv *env, jobject jprocess, jboolean jauto_restart) {
 272   msg_process_t process = jprocess_to_native_process(jprocess, env);
 273   xbt_ex_t e;
 274
 275   int auto_restart = jauto_restart == JNI_TRUE ? 1 : 0;
 276
 277   if (!process) {
 278     jxbt_throw_notbound(env, "process", jprocess);
 279     return;
 280   }
 281
 282   TRY {
 283     MSG_process_auto_restart_set(process,auto_restart);
 284   }
 285   CATCH (e) {
 286     xbt_ex_free(e);
 287   }
 288 }
 289 JNIEXPORT void JNICALL
 290 Java_org_simgrid_msg_Process_restart
 291     (JNIEnv *env, jobject jprocess) {
 292   msg_process_t process = jprocess_to_native_process(jprocess, env);
 293   xbt_ex_t e;
 294
 295   if (!process) {
 296     jxbt_throw_notbound(env, "process", jprocess);
 297     return;
 298   }
 299
 300   TRY {
 301     MSG_process_restart(process);
 302   }
 303   CATCH (e) {
 304     xbt_ex_free(e);
 305   }
 306
 307 }
 308 JNIEXPORT jboolean JNICALL
 309 Java_org_simgrid_msg_Process_isSuspended(JNIEnv * env,
 310                                          jobject jprocess)
 311 {
 312   msg_process_t process = jprocess_to_native_process(jprocess, env);
 313
 314   if (!process) {
 315     jxbt_throw_notbound(env, "process", jprocess);
 316     return 0;
 317   }
 318
 319   /* true is the process is suspended, false otherwise */
 320   return (jboolean) MSG_process_is_suspended(process);
 321 }
 322
 323 JNIEXPORT void JNICALL
 324 Java_org_simgrid_msg_Process_sleep
 325         (JNIEnv *env, jclass cls, jlong jmillis, jint jnanos) {
 326
 327         double time =  jmillis / 1000 + jnanos / 1000;
 328         msg_error_t rv;
 329         xbt_ex_t e;
 330         TRY {
 331                 rv = MSG_process_sleep(time);
 332         }
 333         CATCH(e) {
 334             xbt_ex_free(e);
 335                 return;
 336         }
 337   if (rv != MSG_OK) {
 338         jmsg_throw_status(env,rv);
 339   }
 340 }
 341 JNIEXPORT void JNICALL
 342 Java_org_simgrid_msg_Process_waitFor(JNIEnv * env, jobject jprocess,
 343                                      jdouble jseconds)
 344 {
 345   msg_error_t rv;
 346   xbt_ex_t e;
 347   TRY {
 348    rv = MSG_process_sleep((double)jseconds);
 349   }
 350   CATCH(e) {
 351     xbt_ex_free(e);
 352         return;
 353   }
 354   if (rv != MSG_OK) {
 355         XBT_INFO("Status NOK");
 356         jmsg_throw_status(env,rv);
 357   }
 358 }
 359
 360 JNIEXPORT void JNICALL
 361 Java_org_simgrid_msg_Process_kill(JNIEnv * env,
 362                                   jobject jprocess)
 363 {
 364         /* get the native instances from the java ones */
 365   msg_process_t process = jprocess_to_native_process(jprocess, env);
 366   if (!process) {
 367     jxbt_throw_notbound(env, "process", jprocess);
 368     return;
 369   }
 370
 371         MSG_process_kill(process);
 372 }
 373 JNIEXPORT void JNICALL
 374 Java_org_simgrid_msg_Process_migrate(JNIEnv * env,
 375                                      jobject jprocess, jobject jhost)
 376 {
 377   msg_process_t process = jprocess_to_native_process(jprocess, env);
 378
 379   if (!process) {
 380     jxbt_throw_notbound(env, "process", jprocess);
 381     return;
 382   }
 383
 384   msg_host_t host = jhost_get_native(env, jhost);
 385
 386   if (!host) {
 387     jxbt_throw_notbound(env, "host", jhost);
 388     return;
 389   }
 390
 391   /* try to change the host of the process */
 392   msg_error_t rv = MSG_process_migrate(process, host);
 393   if (rv != MSG_OK) {
 394     jmsg_throw_status(env,rv);
 395   }
 396   /* change the host java side */
 397   (*env)->SetObjectField(env, jprocess, jprocess_field_Process_host, jhost);
 398 }
 399 JNIEXPORT void JNICALL
 400 Java_org_simgrid_msg_Process_setKillTime (JNIEnv *env , jobject jprocess, jdouble jkilltime) {
 401         msg_process_t process = jprocess_to_native_process(jprocess, env);
 402         MSG_process_set_kill_time(process, (double)jkilltime);
 403 }
 404
 405 JNIEXPORT jint JNICALL
 406 Java_org_simgrid_msg_Process_getCount(JNIEnv * env, jclass cls) {
 407         /* FIXME: the next test on SimGrid version is to ensure that this still compiles with SG 3.8 while the C function were added in SG 3.9 only.
 408          * This kind of pimple becomes mandatory when you get so slow to release the java version that it begins evolving further after the C release date.
 409          */
 410 #if SIMGRID_VERSION >= 30900
 411   return (jint) MSG_process_get_number();
 412 #else
 413   return (jint) -1;
 414 #endif
 415 }