src/smx_context_java.c

   1 /* context_java - implementation of context switching for java threads */
   2
   3 /* Copyright (c) 2009, 2010, 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
   9
  10 #include <xbt/function_types.h>
  11 #include <simgrid/simix.h>
  12 #include <xbt/ex.h>
  13 #include "smx_context_java.h"
  14 #include "jxbt_utilities.h"
  15 #include "xbt/dynar.h"
  16 extern JavaVM *__java_vm;
  17
  18 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(jmsg, bindings, "MSG for Java(TM)");
  19
  20 static smx_context_t
  21 smx_ctx_java_factory_create_context(xbt_main_func_t code, int argc,
  22                                     char **argv,
  23                                     void_pfn_smxprocess_t cleanup_func,
  24                                     void *data);
  25
  26 static void smx_ctx_java_free(smx_context_t context);
  27 static void smx_ctx_java_start(smx_context_t context);
  28 static void smx_ctx_java_suspend(smx_context_t context);
  29 static void smx_ctx_java_resume(smx_context_t new_context);
  30 static void smx_ctx_java_runall(void);
  31 static void* smx_ctx_java_thread_run(void *data);
  32 void SIMIX_ctx_java_factory_init(smx_context_factory_t * factory)
  33 {
  34   /* instantiate the context factory */
  35   smx_ctx_base_factory_init(factory);
  36
  37   (*factory)->create_context = smx_ctx_java_factory_create_context;
  38   /* Leave default behavior of (*factory)->finalize */
  39   (*factory)->free = smx_ctx_java_free;
  40   (*factory)->stop = smx_ctx_java_stop;
  41   (*factory)->suspend = smx_ctx_java_suspend;
  42   (*factory)->runall = smx_ctx_java_runall;
  43   (*factory)->name = "ctx_java_factory";
  44   //(*factory)->finalize = smx_ctx_base_factory_finalize;
  45   (*factory)->self = smx_ctx_java_self;
  46   (*factory)->get_data = smx_ctx_base_get_data;
  47 }
  48 smx_context_t smx_ctx_java_self(void)
  49 {
  50         return (smx_context_t)xbt_os_thread_get_extra_data();
  51 }
  52
  53 static smx_context_t
  54 smx_ctx_java_factory_create_context(xbt_main_func_t code, int argc,
  55                                     char **argv,
  56                                     void_pfn_smxprocess_t cleanup_func,
  57                                     void* data)
  58 {
  59         xbt_ex_t e;
  60         static int thread_amount=0;
  61         smx_ctx_java_t context = xbt_new0(s_smx_ctx_java_t, 1);
  62         thread_amount++;
  63   /* If the user provided a function for the process then use it
  64      otherwise is the context for maestro */
  65   if (code) {
  66     if (argc == 0) {
  67       context->jprocess = (jobject) code;
  68     }
  69     else {
  70       context->jprocess = NULL;
  71     }
  72     context->super.cleanup_func = cleanup_func;
  73     context->begin = xbt_os_sem_init(0);
  74     context->end = xbt_os_sem_init(0);
  75
  76     context->super.argc = argc;
  77     context->super.argv = argv;
  78     context->super.code = code;
  79
  80     TRY {
  81        context->thread = xbt_os_thread_create(NULL,smx_ctx_java_thread_run,context,NULL);
  82     } CATCH(e) {
  83         RETHROWF("Impossible to create context #%d. You may want to switch to Java continuations to increase your limits (error: %s)",
  84                         thread_amount);
  85     }
  86   }
  87   else {
  88         context->thread = NULL;
  89     xbt_os_thread_set_extra_data(context);
  90   }
  91   context->super.data = data;
  92
  93   return (smx_context_t) context;
  94 }
  95
  96 static void* smx_ctx_java_thread_run(void *data) {
  97   smx_ctx_java_t context = (smx_ctx_java_t)data;
  98   xbt_os_thread_set_extra_data(context);
  99   //Attach the thread to the JVM
 100   JNIEnv *env;
 101   jint error = (*__java_vm)->AttachCurrentThread(__java_vm, (void **) &env, NULL);
 102   xbt_assert((error == JNI_OK), "The thread could not be attached to the JVM");
 103   context->jenv = get_current_thread_env();
 104   //Wait for the first scheduling round to happen.
 105   xbt_os_sem_acquire(context->begin);
 106   //Create the "Process" object if needed.
 107   if (context->super.argc > 0) {
 108     (*(context->super.code))(context->super.argc, context->super.argv);
 109   }
 110   else {
 111     smx_process_t process = SIMIX_process_self();
 112     (*env)->SetLongField(env, context->jprocess, jprocess_field_Process_bind, (jlong)process);
 113   }
 114   xbt_assert((context->jprocess != NULL), "Process not created...");
 115   //wait for the process to be able to begin
 116   //TODO: Cache it
 117         jfieldID jprocess_field_Process_startTime = jxbt_get_sfield(env, "org/simgrid/msg/Process", "startTime", "D");
 118   jdouble startTime =  (*env)->GetDoubleField(env, context->jprocess, jprocess_field_Process_startTime);
 119   if (startTime > MSG_get_clock()) {
 120         MSG_process_sleep(startTime - MSG_get_clock());
 121   }
 122   //Execution of the "run" method.
 123   jmethodID id = jxbt_get_smethod(env, "org/simgrid/msg/Process", "run", "()V");
 124   xbt_assert( (id != NULL), "Method not found...");
 125   (*env)->CallVoidMethod(env, context->jprocess, id);
 126   smx_ctx_java_stop((smx_context_t)context);
 127
 128   return NULL;
 129 }
 130
 131 static void smx_ctx_java_free(smx_context_t context)
 132 {
 133   if (context) {
 134     smx_ctx_java_t ctx_java = (smx_ctx_java_t) context;
 135     if (ctx_java->thread) { /* We are not in maestro context */
 136       xbt_os_thread_join(ctx_java->thread, NULL);
 137       xbt_os_sem_destroy(ctx_java->begin);
 138       xbt_os_sem_destroy(ctx_java->end);
 139     }
 140   }
 141   smx_ctx_base_free(context);
 142 }
 143
 144
 145 void smx_ctx_java_stop(smx_context_t context)
 146 {
 147   smx_ctx_java_t ctx_java = (smx_ctx_java_t)context;
 148   /* I am the current process and I am dying */
 149   if (context->iwannadie) {
 150         context->iwannadie = 0;
 151         JNIEnv *env = get_current_thread_env();
 152         jxbt_throw_by_name(env, "org/simgrid/msg/ProcessKilledError", bprintf("Process killed :)"));
 153         THROWF(cancel_error, 0, "process cancelled");
 154   }
 155   else {
 156     smx_ctx_base_stop(context);
 157     /* detach the thread and kills it */
 158     JNIEnv *env = ctx_java->jenv;
 159     (*env)->DeleteGlobalRef(env,ctx_java->jprocess);
 160     jint error = (*__java_vm)->DetachCurrentThread(__java_vm);
 161     xbt_assert((error == JNI_OK), "The thread couldn't be detached.");
 162     xbt_os_sem_release(((smx_ctx_java_t)context)->end);
 163     xbt_os_thread_exit(NULL);
 164   }
 165 }
 166
 167 static void smx_ctx_java_suspend(smx_context_t context)
 168 {
 169   smx_ctx_java_t ctx_java = (smx_ctx_java_t) context;
 170   xbt_os_sem_release(ctx_java->end);
 171   xbt_os_sem_acquire(ctx_java->begin);
 172 }
 173
 174 // FIXME: inline those functions
 175 static void smx_ctx_java_resume(smx_context_t new_context)
 176 {
 177   smx_ctx_java_t ctx_java = (smx_ctx_java_t) new_context;
 178   xbt_os_sem_release(ctx_java->begin);
 179   xbt_os_sem_acquire(ctx_java->end);
 180 }
 181
 182 static void smx_ctx_java_runall(void)
 183 {
 184   xbt_dynar_t processes = SIMIX_process_get_runnable();
 185   smx_process_t process;
 186   unsigned int cursor;
 187   xbt_dynar_foreach(processes, cursor, process) {
 188     smx_ctx_java_resume(SIMIX_process_get_context(process));
 189   }
 190 }