Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
allow java actors to kill themselves
[simgrid.git] / src / bindings / java / jmsg.cpp
index aea5885..f8fff39 100644 (file)
@@ -6,25 +6,28 @@
 /* This program is free software; you can redistribute it and/or modify it
  * under the terms of the license (GNU LGPL) which comes with this package. */
 
+#include <locale.h>
+
 #include <simgrid/msg.h>
 #include <simgrid/simix.h>
-#include <surf/surfxml_parse.h>
-#include <locale.h>
+#include <simgrid/plugins/energy.h>
+
+#include <simgrid/s4u/host.hpp>
+
 #include <src/simix/smx_private.h>
 
 #include "jmsg_process.h"
-
 #include "jmsg_as.h"
-
 #include "jmsg_host.h"
 #include "jmsg_storage.h"
 #include "jmsg_task.h"
 #include "jxbt_utilities.h"
-
 #include "jmsg.h"
 
 #include "JavaContext.hpp"
 
+#include <xbt/ex.hpp>
+
 /* Shut up some errors in eclipse online compiler. I wish such a pimple wouldn't be needed */
 #ifndef JNIEXPORT
 #define JNIEXPORT
 
 SG_BEGIN_DECL()
 
-int JAVA_HOST_LEVEL;
-int JAVA_STORAGE_LEVEL;
+int JAVA_HOST_LEVEL = -1;
+int JAVA_STORAGE_LEVEL = -1;
 
 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(jmsg);
 
-JavaVM *__java_vm = NULL;
+JavaVM *__java_vm = nullptr;
 
-JavaVM *get_java_VM(void)
+JavaVM *get_java_VM()
 {
   return __java_vm;
 }
 
-JNIEnv *get_current_thread_env(void)
+JNIEnv *get_current_thread_env()
 {
   JNIEnv *env;
 
-  __java_vm->AttachCurrentThread((void **) &env, NULL);
+  __java_vm->AttachCurrentThread((void **) &env, nullptr);
   return env;
 }
 
 void jmsg_throw_status(JNIEnv *env, msg_error_t status) {
   switch (status) {
     case MSG_TIMEOUT:
-        jxbt_throw_time_out_failure(env,NULL);
+        jxbt_throw_time_out_failure(env,nullptr);
     break;
     case MSG_TRANSFER_FAILURE:
-        jxbt_throw_transfer_failure(env,NULL);
+        jxbt_throw_transfer_failure(env,nullptr);
     break;
     case MSG_HOST_FAILURE:
-        jxbt_throw_host_failure(env,NULL);
+        jxbt_throw_host_failure(env,nullptr);
     break;
     case MSG_TASK_CANCELED:
-        jxbt_throw_task_cancelled(env,NULL);
+        jxbt_throw_task_cancelled(env,nullptr);
     break;
     default:
-        jxbt_throw_native(env,xbt_strdup("undefined message failed (please see jmsg_throw_status function in jmsg.c)"));
+        jxbt_throw_native(env,xbt_strdup("undefined message failed "
+          "(please see jmsg_throw_status function in jmsg.cpp)"));
   }
 }
 
-
 /***************************************************************************************
  * Unsortable functions                                                        *
  ***************************************************************************************/
 
-JNIEXPORT jdouble JNICALL
-Java_org_simgrid_msg_Msg_getClock(JNIEnv * env, jclass cls)
+JNIEXPORT jdouble JNICALL Java_org_simgrid_msg_Msg_getClock(JNIEnv * env, jclass cls)
 {
   return (jdouble) MSG_get_clock();
 }
@@ -94,10 +96,9 @@ static void __JAVA_storage_priv_free(void *storage)
 {
 }
 
-JNIEXPORT void JNICALL
-Java_org_simgrid_msg_Msg_init(JNIEnv * env, jclass cls, jobjectArray jargs)
+JNIEXPORT void JNICALL Java_org_simgrid_msg_Msg_init(JNIEnv * env, jclass cls, jobjectArray jargs)
 {
-  char **argv = NULL;
+  char **argv = nullptr;
   int index;
   int argc = 0;
   jstring jval;
@@ -108,7 +109,7 @@ Java_org_simgrid_msg_Msg_init(JNIEnv * env, jclass cls, jobjectArray jargs)
 
   env->GetJavaVM(&__java_vm);
 
-  simgrid::simix::factory_initializer = simgrid::java::java_factory;
+  simgrid::kernel::context::factory_initializer = &simgrid::kernel::context::java_factory;
   jthrowable exc = env->ExceptionOccurred();
   if (exc) {
     env->ExceptionClear();
@@ -129,12 +130,12 @@ Java_org_simgrid_msg_Msg_init(JNIEnv * env, jclass cls, jobjectArray jargs)
     argv[index + 1] = xbt_strdup(tmp);
     env->ReleaseStringUTFChars(jval, tmp);
   }
-  argv[argc] = NULL;
+  argv[argc] = nullptr;
 
   MSG_init(&argc, argv);
 
-  JAVA_HOST_LEVEL = xbt_lib_add_level(host_lib, (void_f_pvoid_t) __JAVA_host_priv_free);
-  JAVA_STORAGE_LEVEL = xbt_lib_add_level(storage_lib, (void_f_pvoid_t) __JAVA_storage_priv_free);
+  JAVA_HOST_LEVEL = simgrid::s4u::Host::extension_create(__JAVA_host_priv_free);
+  JAVA_STORAGE_LEVEL = xbt_lib_add_level(storage_lib, __JAVA_storage_priv_free);
 
   for (index = 0; index < argc; index++)
     free(argv[index]);
@@ -142,8 +143,7 @@ Java_org_simgrid_msg_Msg_init(JNIEnv * env, jclass cls, jobjectArray jargs)
   free(argv);
 }
 
-JNIEXPORT void JNICALL
-    JNICALL Java_org_simgrid_msg_Msg_run(JNIEnv * env, jclass cls)
+JNIEXPORT void JNICALL JNICALL Java_org_simgrid_msg_Msg_run(JNIEnv * env, jclass cls)
 {
   /* Run everything */
   XBT_DEBUG("Ready to run MSG_MAIN");
@@ -156,7 +156,8 @@ JNIEXPORT void JNICALL
   /* Cleanup java hosts */
   xbt_dynar_t hosts = MSG_hosts_as_dynar();
   for (unsigned long index = 0; index < xbt_dynar_length(hosts) - 1; index++) {
-    jobject jhost = (jobject) xbt_lib_get_level(xbt_dynar_get_as(hosts,index,msg_host_t), JAVA_HOST_LEVEL);
+    msg_host_t msg_host = xbt_dynar_get_as(hosts,index,msg_host_t);
+    jobject jhost = (jobject) msg_host->extension(JAVA_HOST_LEVEL);
     if (jhost)
       jhost_unref(env, jhost);
 
@@ -173,181 +174,169 @@ JNIEXPORT void JNICALL
     }
   }
   xbt_dynar_free(&storages);
-
 }
 
-JNIEXPORT void JNICALL
-Java_org_simgrid_msg_Msg_createEnvironment(JNIEnv * env, jclass cls,
-                                       jstring jplatformFile)
+JNIEXPORT void JNICALL Java_org_simgrid_msg_Msg_createEnvironment(JNIEnv * env, jclass cls, jstring jplatformFile)
 {
-
-  const char *platformFile =
-      env->GetStringUTFChars(jplatformFile, 0);
+  const char *platformFile = env->GetStringUTFChars(jplatformFile, 0);
 
   MSG_create_environment(platformFile);
 
   env->ReleaseStringUTFChars(jplatformFile, platformFile);
 }
 
-JNIEXPORT jobject JNICALL
-Java_org_simgrid_msg_Msg_environmentGetRoutingRoot(JNIEnv * env, jclass cls)
+JNIEXPORT jobject JNICALL Java_org_simgrid_msg_Msg_environmentGetRoutingRoot(JNIEnv * env, jclass cls)
 {
-  msg_as_t as = MSG_environment_get_routing_root();
+  msg_netzone_t as = MSG_environment_get_routing_root();
   jobject jas = jas_new_instance(env);
   if (!jas) {
     jxbt_throw_jni(env, "java As instantiation failed");
-    return NULL;
+    return nullptr;
   }
   jas = jas_ref(env, jas);
   if (!jas) {
     jxbt_throw_jni(env, "new global ref allocation failed");
-    return NULL;
+    return nullptr;
   }
   jas_bind(jas, as, env);
 
   return (jobject) jas;
 }
 
-JNIEXPORT void JNICALL
-Java_org_simgrid_msg_Msg_debug(JNIEnv * env, jclass cls, jstring js)
+JNIEXPORT void JNICALL Java_org_simgrid_msg_Msg_debug(JNIEnv * env, jclass cls, jstring js)
 {
   const char *s = env->GetStringUTFChars(js, 0);
   XBT_DEBUG("%s", s);
   env->ReleaseStringUTFChars(js, s);
 }
-JNIEXPORT void JNICALL
-Java_org_simgrid_msg_Msg_verb(JNIEnv * env, jclass cls, jstring js)
+
+JNIEXPORT void JNICALL Java_org_simgrid_msg_Msg_verb(JNIEnv * env, jclass cls, jstring js)
 {
   const char *s = env->GetStringUTFChars(js, 0);
   XBT_VERB("%s", s);
   env->ReleaseStringUTFChars(js, s);
 }
-JNIEXPORT void JNICALL
-Java_org_simgrid_msg_Msg_info(JNIEnv * env, jclass cls, jstring js)
+
+JNIEXPORT void JNICALL Java_org_simgrid_msg_Msg_info(JNIEnv * env, jclass cls, jstring js)
 {
   const char *s = env->GetStringUTFChars(js, 0);
   XBT_INFO("%s", s);
   env->ReleaseStringUTFChars(js, s);
 }
-JNIEXPORT void JNICALL
-Java_org_simgrid_msg_Msg_warn(JNIEnv * env, jclass cls, jstring js)
+
+JNIEXPORT void JNICALL Java_org_simgrid_msg_Msg_warn(JNIEnv * env, jclass cls, jstring js)
 {
   const char *s = env->GetStringUTFChars(js, 0);
   XBT_WARN("%s", s);
   env->ReleaseStringUTFChars(js, s);
 }
-JNIEXPORT void JNICALL
-Java_org_simgrid_msg_Msg_error(JNIEnv * env, jclass cls, jstring js)
+
+JNIEXPORT void JNICALL Java_org_simgrid_msg_Msg_error(JNIEnv * env, jclass cls, jstring js)
 {
   const char *s = env->GetStringUTFChars(js, 0);
   XBT_ERROR("%s", s);
   env->ReleaseStringUTFChars(js, s);
 }
-JNIEXPORT void JNICALL
-Java_org_simgrid_msg_Msg_critical(JNIEnv * env, jclass cls, jstring js)
+
+JNIEXPORT void JNICALL Java_org_simgrid_msg_Msg_critical(JNIEnv * env, jclass cls, jstring js)
 {
   const char *s = env->GetStringUTFChars(js, 0);
   XBT_CRITICAL("%s", s);
   env->ReleaseStringUTFChars(js, s);
 }
+
+static int java_main(int argc, char *argv[]);
+
 JNIEXPORT void JNICALL
-Java_org_simgrid_msg_Msg_deployApplication(JNIEnv * env, jclass cls,
-                                       jstring jdeploymentFile)
+Java_org_simgrid_msg_Msg_deployApplication(JNIEnv * env, jclass cls, jstring jdeploymentFile)
 {
+  const char *deploymentFile = env->GetStringUTFChars(jdeploymentFile, 0);
 
-  const char *deploymentFile =
-      env->GetStringUTFChars(jdeploymentFile, 0);
-
-  SIMIX_function_register_default(simgrid::java::java_main);
+  SIMIX_function_register_default(java_main);
   MSG_launch_application(deploymentFile);
 }
-/**
- * Function called when there is the need to create the java Process object
- * (when we are using deployement files).
- * it HAS to be executed on the process context, else really bad things will happen.
+
+SG_END_DECL()
+
+JNIEXPORT void JNICALL Java_org_simgrid_msg_Msg_energyInit() {
+  sg_energy_plugin_init();
+}
+
+/** Run a Java org.simgrid.msg.Process
+ *
+ *  If needed, this waits for the process starting time.
+ *  Then it calls the Process.run() method.
  */
-static int create_jprocess(int argc, char *argv[]) {
+static void run_jprocess(JNIEnv *env, jobject jprocess)
+{
+  // wait for the process's start time
+  jfieldID jprocess_field_Process_startTime = jxbt_get_sfield(env, "org/simgrid/msg/Process", "startTime", "D");
+  jdouble startTime = env->GetDoubleField(jprocess, jprocess_field_Process_startTime);
+  if (startTime > MSG_get_clock())
+    MSG_process_sleep(startTime - MSG_get_clock());
+  //Execution of the "run" method.
+  jmethodID id = jxbt_get_smethod(env, "org/simgrid/msg/Process", "run", "()V");
+  xbt_assert((id != nullptr), "Method run() not found...");
+  env->CallVoidMethod(jprocess, id);
+}
+
+/** Create a Java org.simgrid.msg.Process with the arguments and run it */
+static int java_main(int argc, char *argv[])
+{
   JNIEnv *env = get_current_thread_env();
+  simgrid::kernel::context::JavaContext* context = static_cast<simgrid::kernel::context::JavaContext*>(SIMIX_context_self());
+
   //Change the "." in class name for "/".
   xbt_str_subst(argv[0],'.','/',0);
   jclass class_Process = env->FindClass(argv[0]);
   xbt_str_subst(argv[0],'/','.',0);
   //Retrieve the methodID for the constructor
-  xbt_assert((class_Process != NULL), "Class not found (%s). The deployment file must use the fully qualified class name, including the package. The case is important.", argv[0]);
+  xbt_assert((class_Process != nullptr), "Class not found (%s). The deployment file must use the fully qualified class name, including the package. The case is important.", argv[0]);
   jmethodID constructor_Process = env->GetMethodID(class_Process, "<init>", "(Lorg/simgrid/msg/Host;Ljava/lang/String;[Ljava/lang/String;)V");
-  xbt_assert((constructor_Process != NULL), "Constructor not found for class %s. Is there a (Host, String ,String[]) constructor in your class ?", argv[0]);
+  xbt_assert((constructor_Process != nullptr), "Constructor not found for class %s. Is there a (Host, String ,String[]) constructor in your class ?", argv[0]);
 
   //Retrieve the name of the process.
   jstring jname = env->NewStringUTF(argv[0]);
   //Build the arguments
-  jobjectArray args = (jobjectArray)env->NewObjectArray(argc - 1,
-    env->FindClass("java/lang/String"),
-    env->NewStringUTF(""));
-  int i;
-  for (i = 1; i < argc; i++)
-      env->SetObjectArrayElement(args,i - 1,
-        env->NewStringUTF(argv[i]));
+  jobjectArray args = static_cast<jobjectArray>(env->NewObjectArray(argc - 1, env->FindClass("java/lang/String"),
+                                                                    env->NewStringUTF("")));
+  for (int i = 1; i < argc; i++)
+      env->SetObjectArrayElement(args,i - 1, env->NewStringUTF(argv[i]));
   //Retrieve the host for the process.
-  jstring jhostName = env->NewStringUTF(MSG_host_get_name(MSG_host_self()));
-  jobject jhost = Java_org_simgrid_msg_Host_getByName(env, NULL, jhostName);
+  jstring jhostName = env->NewStringUTF(MSG_host_self()->cname());
+  jobject jhost = Java_org_simgrid_msg_Host_getByName(env, nullptr, jhostName);
   //creates the process
   jobject jprocess = env->NewObject(class_Process, constructor_Process, jhost, jname, args);
-  xbt_assert((jprocess != NULL), "Process allocation failed.");
+  xbt_assert((jprocess != nullptr), "Process allocation failed.");
   jprocess = env->NewGlobalRef(jprocess);
   //bind the process to the context
   msg_process_t process = MSG_process_self();
-  simgrid::java::JavaContext* context =
-    (simgrid::java::JavaContext*) MSG_process_get_smx_ctx(process);
+
   context->jprocess = jprocess;
   /* sets the PID and the PPID of the process */
-  env->SetIntField(jprocess, jprocess_field_Process_pid,(jint) MSG_process_get_PID(process));
-  env->SetIntField(jprocess, jprocess_field_Process_ppid, (jint) MSG_process_get_PPID(process));
+  env->SetIntField(jprocess, jprocess_field_Process_pid, static_cast<jint>(MSG_process_get_PID(process)));
+  env->SetIntField(jprocess, jprocess_field_Process_ppid, static_cast<jint>(MSG_process_get_PPID(process)));
   jprocess_bind(jprocess, process, env);
 
+  run_jprocess(env, context->jprocess);
   return 0;
 }
 
-SG_END_DECL()
-
-
-
 namespace simgrid {
-namespace java {
+namespace kernel {
+namespace context {
 
-int java_main(int argc, char *argv[])
+/** Run the Java org.simgrid.msg.Process */
+void java_main_jprocess(jobject jprocess)
 {
   JNIEnv *env = get_current_thread_env();
-  simgrid::java::JavaContext* context =
-    (simgrid::java::JavaContext*) SIMIX_context_self();
-
-  if (argc > 0)
-    create_jprocess(argc, argv);
-  else {
-    smx_process_t process = SIMIX_process_self();
-    context->jprocess = (jobject) MSG_process_get_data(process);
-    jprocess_bind(context->jprocess, process, env);
-  }
-
-  // Adrien, ugly path, just to bypass creation of context at low levels
-  // (i.e such as for the VM migration for instance)
-  if (context->jprocess != nullptr){
-    xbt_assert((context->jprocess != nullptr), "Process not created...");
-    //wait for the process to be able to begin
-    //TODO: Cache it
-    jfieldID jprocess_field_Process_startTime = jxbt_get_sfield(env, "org/simgrid/msg/Process", "startTime", "D");
-    jdouble startTime =  env->GetDoubleField(context->jprocess, jprocess_field_Process_startTime);
-    if (startTime > MSG_get_clock()) {
-      MSG_process_sleep(startTime - MSG_get_clock());
-    }
-    //Execution of the "run" method.
-    jmethodID id = jxbt_get_smethod(env, "org/simgrid/msg/Process", "run", "()V");
-    xbt_assert( (id != nullptr), "Method not found...");
-    env->CallVoidMethod(context->jprocess, id);
-  }
-
-  return 0;
-}
+  simgrid::kernel::context::JavaContext* context = static_cast<simgrid::kernel::context::JavaContext*>(SIMIX_context_self());
+  context->jprocess = jprocess;
+  smx_actor_t process = SIMIX_process_self();
+  jprocess_bind(context->jprocess, process, env);
 
+  // Adrien, ugly path, just to bypass creation of context at low levels (i.e such as for the VM migration for instance)
+  if (context->jprocess != nullptr)
+    run_jprocess(env, context->jprocess);
 }
-}
-
+}}}