Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
New function Process.getCount()
[simgrid.git] / src / jmsg_process.c
index 520241d..aaa73e7 100644 (file)
@@ -1,6 +1,6 @@
 /* Functions related to the java process instances.                         */
 
-/* Copyright (c) 2007, 2008, 2009, 2010. The SimGrid Team.
+/* Copyright (c) 2007-2012. The SimGrid Team.
  * All rights reserved.                                                     */
 
 /* This program is free software; you can redistribute it and/or modify it
 #include "jmsg_host.h"
 #include "jxbt_utilities.h"
 #include "smx_context_java.h"
+#include "smx_context_cojava.h"
 
 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(jmsg);
 
-static jfieldID jprocess_field_Process_host;
-static jfieldID jprocess_field_Process_pid;
-static jfieldID jprocess_field_Process_ppid;
+JNIEXPORT void JNICALL
+Java_org_simgrid_msg_Process_exit(JNIEnv *env, jobject jprocess) {
+  if (smx_factory_initializer_to_use == SIMIX_ctx_cojava_factory_init) {
+    msg_process_t process = jprocess_to_native_process(jprocess, env);
+    smx_context_t context = MSG_process_get_smx_ctx(process);
+    smx_ctx_cojava_stop(context);
+  }
+}
 
-jobject native_to_java_process(m_process_t process)
+jobject native_to_java_process(msg_process_t process)
 {
   return ((smx_ctx_java_t)MSG_process_get_smx_ctx(process))->jprocess;
 }
@@ -35,77 +41,29 @@ void jprocess_delete_global_ref(jobject jprocess, JNIEnv * env)
 
 void jprocess_join(jobject jprocess, JNIEnv * env)
 {
-       m_process_t process = jprocess_to_native_process(jprocess,env);
+       msg_process_t process = jprocess_to_native_process(jprocess,env);
        smx_ctx_java_t context = (smx_ctx_java_t)MSG_process_get_smx_ctx(process);
        xbt_os_thread_join(context->thread,NULL);
 }
 
-void jprocess_exit(jobject jprocess, JNIEnv * env)
+msg_process_t jprocess_to_native_process(jobject jprocess, JNIEnv * env)
 {
-  jmethodID id =
-      jxbt_get_smethod(env, "org/simgrid/msg/Process", "interrupt", "()V");
-
-  if (!id)
-    return;
-
-  (*env)->CallVoidMethod(env, jprocess, id);
+  return (msg_process_t) (long) (*env)->GetLongField(env, jprocess, jprocess_field_Process_bind);
 }
 
-void jprocess_start(jobject jprocess, JNIEnv * env)
+void jprocess_bind(jobject jprocess, msg_process_t process, JNIEnv * env)
 {
-  jmethodID id =
-      jxbt_get_smethod(env, "org/simgrid/msg/Process", "start", "()V");
-
-  if (!id)
-    return;
-
-  XBT_DEBUG("jprocess_start(jproc=%p,env=%p)", jprocess, env);
-  (*env)->CallVoidMethod(env, jprocess, id);
-  XBT_DEBUG("jprocess started");
-}
-
-m_process_t jprocess_to_native_process(jobject jprocess, JNIEnv * env)
-{
-  jfieldID id = jxbt_get_sfield(env, "org/simgrid/msg/Process", "bind", "J");
-
-  if (!id) {
-       XBT_INFO("Can't find bind field in Process");
-    return NULL;
-  }
-
-  return (m_process_t) (long) (*env)->GetLongField(env, jprocess, id);
-}
-
-void jprocess_bind(jobject jprocess, m_process_t process, JNIEnv * env)
-{
-  jfieldID id = jxbt_get_sfield(env, "org/simgrid/msg/Process", "bind", "J");
-  if (!id)
-    return;
-
-  (*env)->SetLongField(env, jprocess, id, (jlong) (long) (process));
+  (*env)->SetLongField(env, jprocess, jprocess_field_Process_bind, (jlong)(process));
 }
 
 jlong jprocess_get_id(jobject jprocess, JNIEnv * env)
 {
-  jfieldID id = jxbt_get_sfield(env, "org/simgrid/msg/Process", "id", "J");
-
-  if (!id)
-    return 0;
-
-  return (*env)->GetLongField(env, jprocess, id);
+  return (*env)->GetLongField(env, jprocess, jprocess_field_Process_id);
 }
 
 jstring jprocess_get_name(jobject jprocess, JNIEnv * env)
 {
-  jfieldID id = jxbt_get_sfield(env, "org/simgrid/msg/Process", "name",
-                                "Ljava/lang/String;");
-  jobject jname;
-
-  if (!id)
-    return NULL;
-
-  jname = (jstring) (*env)->GetObjectField(env, jprocess, id);
-
+  jstring jname = (jstring) (*env)->GetObjectField(env, jprocess, jprocess_field_Process_name);
   return (*env)->NewGlobalRef(env, jname);
 
 }
@@ -121,16 +79,19 @@ jboolean jprocess_is_valid(jobject jprocess, JNIEnv * env)
 }
 JNIEXPORT void JNICALL
 Java_org_simgrid_msg_Process_nativeInit(JNIEnv *env, jclass cls) {
-       jclass jprocess_class_Process = (*env)->FindClass(env, "org/simgrid/msg/Process");
-
-       jprocess_field_Process_pid = jxbt_get_sfield(env, "org/simgrid/msg/Process", "pid", "I");
-       jprocess_field_Process_ppid = jxbt_get_sfield(env, "org/simgrid/msg/Process", "ppid", "I");
-       jprocess_field_Process_host = jxbt_get_jfield(env, jprocess_class_Process, "host", "Lorg/simgrid/msg/Host;");
-
-       if (!jprocess_class_Process || !jprocess_field_Process_pid ||
-                       !jprocess_field_Process_ppid || !jprocess_field_Process_host) {
-       jxbt_throw_native(env,bprintf("Can't find some fields in Java class. You should report this bug."));
-       }
+  jclass jprocess_class_Process = (*env)->FindClass(env, "org/simgrid/msg/Process");
+
+  jprocess_field_Process_name = jxbt_get_jfield(env, jprocess_class_Process, "name", "Ljava/lang/String;");
+  jprocess_field_Process_bind = jxbt_get_jfield(env, jprocess_class_Process, "bind", "J");
+  jprocess_field_Process_id = jxbt_get_jfield(env, jprocess_class_Process, "id", "J");
+  jprocess_field_Process_pid = jxbt_get_jfield(env, jprocess_class_Process, "pid", "I");
+  jprocess_field_Process_ppid = jxbt_get_jfield(env, jprocess_class_Process, "ppid", "I");
+  jprocess_field_Process_host = jxbt_get_jfield(env, jprocess_class_Process, "host", "Lorg/simgrid/msg/Host;");
+  jprocess_field_Process_killTime = jxbt_get_jfield(env, jprocess_class_Process, "killTime", "D");
+  if (!jprocess_class_Process || !jprocess_field_Process_id || !jprocess_field_Process_name || !jprocess_field_Process_pid ||
+      !jprocess_field_Process_ppid || !jprocess_field_Process_host) {
+    jxbt_throw_native(env,bprintf("Can't find some fields in Java class. You should report this bug."));
+  }
 }
 JNIEXPORT void JNICALL
 Java_org_simgrid_msg_Process_create(JNIEnv * env,
@@ -143,15 +104,11 @@ Java_org_simgrid_msg_Process_create(JNIEnv * env,
   jstring jname;                /* the name of the java process instance                */
   const char *name;             /* the C name of the process                            */
   const char *hostname;
-  m_process_t process;          /* the native process to create                         */
-  m_host_t host;                /* Where that process lives */
+  msg_process_t process;          /* the native process to create                         */
+  msg_host_t host;                /* Where that process lives */
 
   hostname = (*env)->GetStringUTFChars(env, jhostname, 0);
 
-  XBT_DEBUG("Java_org_simgrid_msg_MsgNative_processCreate(env=%p,jproc=%p,host=%s)",
-        env, jprocess_arg, hostname);
-
-
   /* get the name of the java process */
   jname = jprocess_get_name(jprocess_arg, env);
   if (!jname) {
@@ -179,25 +136,24 @@ Java_org_simgrid_msg_Process_create(JNIEnv * env,
   name = (*env)->GetStringUTFChars(env, jname, 0);
   name = xbt_strdup(name);
 
+  /* Retrieve the kill time from the process */
+  jdouble jkill = (*env)->GetDoubleField(env, jprocess, jprocess_field_Process_killTime);
   /* Actually build the MSG process */
   process = MSG_process_create_with_environment(name,
                                                (xbt_main_func_t) jprocess,
-                                               /*data*/ NULL,
+                                               /*data*/ jprocess,
                                                host,
-                                               /* kill_time */-1,
                                                /*argc, argv, properties*/
                                                0,NULL,NULL);
-
-  MSG_process_set_data(process,&process);
+  MSG_process_set_kill_time(process, (double)jkill);
+  /* bind the java process instance to the native process */
+  jprocess_bind(jprocess, process, env);
 
   /* release our reference to the process name (variable name becomes invalid) */
   //FIXME : This line should be uncommented but with mac it doesn't work. BIG WARNING
   //(*env)->ReleaseStringUTFChars(env, jname, name);
   (*env)->ReleaseStringUTFChars(env, jhostname, hostname);
 
-  /* bind the java process instance to the native process */
-  jprocess_bind(jprocess, process, env);
-
   /* sets the PID and the PPID of the process */
   (*env)->SetIntField(env, jprocess, jprocess_field_Process_pid,(jint) MSG_process_get_PID(process));
   (*env)->SetIntField(env, jprocess, jprocess_field_Process_ppid, (jint) MSG_process_get_PPID(process));
@@ -218,7 +174,7 @@ JNIEXPORT jobject JNICALL
 Java_org_simgrid_msg_Process_fromPID(JNIEnv * env, jclass cls,
                                      jint PID)
 {
-  m_process_t process = MSG_process_from_PID(PID);
+  msg_process_t process = MSG_process_from_PID(PID);
 
   if (!process) {
     jxbt_throw_process_not_found(env, bprintf("PID = %d",(int) PID));
@@ -228,17 +184,37 @@ Java_org_simgrid_msg_Process_fromPID(JNIEnv * env, jclass cls,
   jobject jprocess = native_to_java_process(process);
 
   if (!jprocess) {
-    jxbt_throw_jni(env, "SIMIX_process_get_jprocess() failed");
+    jxbt_throw_jni(env, "get process failed");
     return NULL;
   }
 
   return jprocess;
 }
+JNIEXPORT jobject JNICALL
+Java_org_simgrid_msg_Process_getProperty(JNIEnv *env, jobject jprocess, jobject jname) {
+  msg_process_t process = jprocess_to_native_process(jprocess, env);
+
+  if (!process) {
+    jxbt_throw_notbound(env, "process", jprocess);
+    return NULL;
+  }
+  const char *name = (*env)->GetStringUTFChars(env, jname, 0);
+
+  const char *property = MSG_process_get_property_value(process, name);
+  if (!property) {
+    return NULL;
+  }
+
+  jobject jproperty = (*env)->NewStringUTF(env, property);
+
+  (*env)->ReleaseStringUTFChars(env, jname, name);
 
+  return jproperty;
+}
 JNIEXPORT jobject JNICALL
 Java_org_simgrid_msg_Process_currentProcess(JNIEnv * env, jclass cls)
 {
-  m_process_t process = MSG_process_self();
+  msg_process_t process = MSG_process_self();
   jobject jprocess;
 
   if (!process) {
@@ -255,10 +231,10 @@ Java_org_simgrid_msg_Process_currentProcess(JNIEnv * env, jclass cls)
 }
 
 JNIEXPORT void JNICALL
-Java_org_simgrid_msg_Process_pause(JNIEnv * env,
+Java_org_simgrid_msg_Process_suspend(JNIEnv * env,
                                    jobject jprocess)
 {
-  m_process_t process = jprocess_to_native_process(jprocess, env);
+  msg_process_t process = jprocess_to_native_process(jprocess, env);
 
   if (!process) {
     jxbt_throw_notbound(env, "process", jprocess);
@@ -266,17 +242,18 @@ Java_org_simgrid_msg_Process_pause(JNIEnv * env,
   }
 
   /* try to suspend the process */
-  MSG_error_t rv = MSG_process_suspend(process);
+  msg_error_t rv = MSG_process_suspend(process);
 
   jxbt_check_res("MSG_process_suspend()", rv, MSG_OK,
                  bprintf("unexpected error , please report this bug"));
 
 }
+
 JNIEXPORT void JNICALL
-Java_org_simgrid_msg_Process_restart(JNIEnv * env,
+Java_org_simgrid_msg_Process_resume(JNIEnv * env,
                                      jobject jprocess)
 {
-  m_process_t process = jprocess_to_native_process(jprocess, env);
+  msg_process_t process = jprocess_to_native_process(jprocess, env);
 
   if (!process) {
     jxbt_throw_notbound(env, "process", jprocess);
@@ -284,16 +261,55 @@ Java_org_simgrid_msg_Process_restart(JNIEnv * env,
   }
 
   /* try to resume the process */
-  MSG_error_t rv = MSG_process_resume(process);
+  msg_error_t rv = MSG_process_resume(process);
 
   jxbt_check_res("MSG_process_resume()", rv, MSG_OK,
                  bprintf("unexpected error , please report this bug"));
 }
+JNIEXPORT void JNICALL
+Java_org_simgrid_msg_Process_setAutoRestart
+    (JNIEnv *env, jobject jprocess, jboolean jauto_restart) {
+  msg_process_t process = jprocess_to_native_process(jprocess, env);
+  xbt_ex_t e;
+
+  int auto_restart = jauto_restart == JNI_TRUE ? 1 : 0;
+
+  if (!process) {
+    jxbt_throw_notbound(env, "process", jprocess);
+    return;
+  }
+
+  TRY {
+    MSG_process_auto_restart_set(process,auto_restart);
+  }
+  CATCH (e) {
+    xbt_ex_free(e);
+  }
+}
+JNIEXPORT void JNICALL
+Java_org_simgrid_msg_Process_restart
+    (JNIEnv *env, jobject jprocess) {
+  msg_process_t process = jprocess_to_native_process(jprocess, env);
+  xbt_ex_t e;
+
+  if (!process) {
+    jxbt_throw_notbound(env, "process", jprocess);
+    return;
+  }
+
+  TRY {
+    MSG_process_restart(process);
+  }
+  CATCH (e) {
+    xbt_ex_free(e);
+  }
+
+}
 JNIEXPORT jboolean JNICALL
 Java_org_simgrid_msg_Process_isSuspended(JNIEnv * env,
                                          jobject jprocess)
 {
-  m_process_t process = jprocess_to_native_process(jprocess, env);
+  msg_process_t process = jprocess_to_native_process(jprocess, env);
 
   if (!process) {
     jxbt_throw_notbound(env, "process", jprocess);
@@ -304,76 +320,68 @@ Java_org_simgrid_msg_Process_isSuspended(JNIEnv * env,
   return (jboolean) MSG_process_is_suspended(process);
 }
 
-JNIEXPORT void JNICALL Java_org_simgrid_msg_Process_sleep
+JNIEXPORT void JNICALL
+Java_org_simgrid_msg_Process_sleep
        (JNIEnv *env, jclass cls, jlong jmillis, jint jnanos) {
 
        double time =  jmillis / 1000 + jnanos / 1000;
-
-       MSG_error_t rv = MSG_process_sleep(time);
-  jxbt_check_res("MSG_process_sleep()", rv, MSG_OK,
-                 bprintf("unexpected error , please report this bug"));
+       msg_error_t rv;
+       xbt_ex_t e;
+       TRY {
+               rv = MSG_process_sleep(time);
+       }
+       CATCH(e) {
+           xbt_ex_free(e);
+               return;
+       }
+  if (rv != MSG_OK) {
+       jmsg_throw_status(env,rv);
+  }
 }
 JNIEXPORT void JNICALL
 Java_org_simgrid_msg_Process_waitFor(JNIEnv * env, jobject jprocess,
                                      jdouble jseconds)
 {
-  m_process_t process = jprocess_to_native_process(jprocess, env);
-
-  if (!process) {
-    jxbt_throw_notbound(env, "process", jprocess);
-    return;
+  msg_error_t rv;
+  xbt_ex_t e;
+  TRY {
+   rv = MSG_process_sleep((double)jseconds);
   }
-  MSG_error_t rv = MSG_process_sleep((double)jseconds);
-  if (rv != MSG_OK) {
-//     smx_ctx_java_stop(smx_ctx_java_self());
+  CATCH(e) {
+    xbt_ex_free(e);
+       return;
   }
-}
-
-JNIEXPORT void JNICALL
-Java_org_simgrid_msg_Process_exit(JNIEnv * env,
-                                  jobject jprocess)
-{
-
-  m_process_t process = jprocess_to_native_process(jprocess, env);
-
-  if (!process) {
-    jxbt_throw_notbound(env, "process", jprocess);
-    return;
+  if (rv != MSG_OK) {
+       XBT_INFO("Status NOK");
+       jmsg_throw_status(env,rv);
   }
-       smx_ctx_java_t context = (smx_ctx_java_t)MSG_process_get_smx_ctx(process);
-       smx_ctx_java_stop(MSG_process_get_smx_ctx(process));
-  xbt_os_sem_release(context->end);
 }
 
 JNIEXPORT void JNICALL
-Java_org_simgrid_msg_Process_kill(JNIEnv * env, jclass cls,
+Java_org_simgrid_msg_Process_kill(JNIEnv * env,
                                   jobject jprocess)
 {
-  /* get the native instances from the java ones */
-  m_process_t process = jprocess_to_native_process(jprocess, env);
-
+       /* get the native instances from the java ones */
+  msg_process_t process = jprocess_to_native_process(jprocess, env);
   if (!process) {
     jxbt_throw_notbound(env, "process", jprocess);
     return;
   }
 
-  /* kill the native process (this wrapper is call by the destructor of the java
-   * process instance)
-   */
-  MSG_process_kill(process);
+       MSG_process_kill(process);
 }
 JNIEXPORT void JNICALL
 Java_org_simgrid_msg_Process_migrate(JNIEnv * env,
                                      jobject jprocess, jobject jhost)
 {
-  m_process_t process = jprocess_to_native_process(jprocess, env);
+  msg_process_t process = jprocess_to_native_process(jprocess, env);
 
   if (!process) {
     jxbt_throw_notbound(env, "process", jprocess);
     return;
   }
 
-  m_host_t host = jhost_get_native(env, jhost);
+  msg_host_t host = jhost_get_native(env, jhost);
 
   if (!host) {
     jxbt_throw_notbound(env, "host", jhost);
@@ -381,9 +389,27 @@ Java_org_simgrid_msg_Process_migrate(JNIEnv * env,
   }
 
   /* try to change the host of the process */
-  MSG_error_t rv = MSG_process_migrate(process, host);
-  jxbt_check_res("MSG_process_migrate()", rv, MSG_OK,
-                 bprintf("unexpected error , please report this bug"));
+  msg_error_t rv = MSG_process_migrate(process, host);
+  if (rv != MSG_OK) {
+    jmsg_throw_status(env,rv);
+  }
   /* change the host java side */
   (*env)->SetObjectField(env, jprocess, jprocess_field_Process_host, jhost);
 }
+JNIEXPORT void JNICALL
+Java_org_simgrid_msg_Process_setKillTime (JNIEnv *env , jobject jprocess, jdouble jkilltime) {
+       msg_process_t process = jprocess_to_native_process(jprocess, env);
+       MSG_process_set_kill_time(process, (double)jkilltime);
+}
+
+JNIEXPORT jint JNICALL
+Java_org_simgrid_msg_Process_getCount(JNIEnv * env, jclass cls) {
+       /* 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.
+        * 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.
+        */
+#if SIMGRID_VERSION >= 30900
+  return (jint) MSG_process_get_number();
+#else
+  return (jint) -1;
+#endif
+}