src/bindings/java/jmsg_vm.cpp

   1 /* Functions related to the Virtual Machines.                               */
   2
   3 /* Copyright (c) 2012-2019. The SimGrid Team. All rights reserved.          */
   4
   5 /* This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the license (GNU LGPL) which comes with this package. */
   7
   8 #include "jmsg_vm.h"
   9 #include "jmsg_host.h"
  10 #include "jxbt_utilities.hpp"
  11 #include "simgrid/Exception.hpp"
  12 #include "simgrid/plugins/live_migration.h"
  13 #include "src/kernel/context/Context.hpp"
  14 #include "src/plugins/vm/VirtualMachineImpl.hpp"
  15
  16 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(java);
  17
  18 extern int JAVA_HOST_LEVEL;
  19 static jfieldID jvm_field_bind;
  20
  21 void jvm_bind(JNIEnv *env, jobject jvm, msg_vm_t vm)
  22 {
  23   env->SetLongField(jvm, jvm_field_bind, (intptr_t)vm);
  24 }
  25
  26 msg_vm_t jvm_get_native(JNIEnv *env, jobject jvm)
  27 {
  28   return (msg_vm_t)(intptr_t)env->GetLongField(jvm, jvm_field_bind);
  29 }
  30
  31 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_nativeInit(JNIEnv *env, jclass cls)
  32 {
  33   jclass jprocess_class_VM = env->FindClass("org/simgrid/msg/VM");
  34   jvm_field_bind = jxbt_get_jfield(env, jprocess_class_VM, "bind", "J");
  35   xbt_assert(jvm_field_bind, "Native initialization of msg/VM failed. Please report that bug");
  36 }
  37
  38 JNIEXPORT jint JNICALL Java_org_simgrid_msg_VM_isCreated(JNIEnv * env, jobject jvm)
  39 {
  40   msg_vm_t vm = jvm_get_native(env,jvm);
  41   return MSG_vm_is_created(vm);
  42 }
  43
  44 JNIEXPORT jint JNICALL Java_org_simgrid_msg_VM_isRunning(JNIEnv * env, jobject jvm)
  45 {
  46   msg_vm_t vm = jvm_get_native(env,jvm);
  47   return (jint) MSG_vm_is_running(vm);
  48 }
  49
  50 JNIEXPORT jint JNICALL Java_org_simgrid_msg_VM_isMigrating(JNIEnv * env, jobject jvm)
  51 {
  52   msg_vm_t vm = jvm_get_native(env,jvm);
  53   return (jint) MSG_vm_is_migrating(vm);
  54 }
  55
  56 JNIEXPORT jint JNICALL Java_org_simgrid_msg_VM_isSuspended(JNIEnv * env, jobject jvm)
  57 {
  58   msg_vm_t vm = jvm_get_native(env,jvm);
  59   return (jint) MSG_vm_is_suspended(vm);
  60 }
  61
  62 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_setBound(JNIEnv *env, jobject jvm, jdouble bound)
  63 {
  64   msg_vm_t vm = jvm_get_native(env,jvm);
  65   MSG_vm_set_bound(vm, bound);
  66 }
  67
  68 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_create(JNIEnv* env, jobject jVm, jobject jHost, jstring jname,
  69                                                       jint coreAmount, jint jramsize, jint jmig_netspeed,
  70                                                       jint jdp_intensity)
  71 {
  72   msg_host_t host = jhost_get_native(env, jHost);
  73
  74   const char* name = env->GetStringUTFChars(jname, 0);
  75   msg_vm_t vm      = MSG_vm_create_migratable(host, name, static_cast<int>(coreAmount), static_cast<int>(jramsize),
  76                                          static_cast<int>(jmig_netspeed), static_cast<int>(jdp_intensity));
  77   env->ReleaseStringUTFChars(jname, name);
  78
  79   jvm_bind(env, jVm, vm);
  80   jVm = env->NewGlobalRef(jVm);
  81   // We use the extension level of the host, even if that's somehow disturbing
  82   vm->extension_set(JAVA_HOST_LEVEL, (void*)jVm);
  83 }
  84
  85 JNIEXPORT jobjectArray JNICALL Java_org_simgrid_msg_VM_all(JNIEnv* env, jclass cls_arg)
  86 {
  87   xbt_dynar_t hosts = MSG_hosts_as_dynar();
  88   std::vector<jobject> vms;
  89
  90   unsigned int it;
  91   msg_host_t h;
  92   xbt_dynar_foreach (hosts, it, h) {
  93     simgrid::s4u::VirtualMachine* vm = dynamic_cast<simgrid::s4u::VirtualMachine*>(h);
  94     if (vm != nullptr && vm->get_state() != simgrid::s4u::VirtualMachine::state::DESTROYED) {
  95       jobject jvm = static_cast<jobject>(vm->extension(JAVA_HOST_LEVEL));
  96       vms.push_back(jvm);
  97     }
  98   }
  99   xbt_dynar_free(&hosts);
 100
 101   vms.shrink_to_fit();
 102   int count = vms.size();
 103
 104   jclass cls = jxbt_get_class(env, "org/simgrid/msg/VM");
 105   if (not cls)
 106     return nullptr;
 107
 108   jobjectArray jtable = env->NewObjectArray((jsize)count, cls, nullptr);
 109   if (not jtable) {
 110     jxbt_throw_jni(env, "Hosts table allocation failed");
 111     return nullptr;
 112   }
 113
 114   for (int index = 0; index < count; index++) {
 115     jobject jhost = vms.at(index);
 116     env->SetObjectArrayElement(jtable, index, jhost);
 117   }
 118   return jtable;
 119 }
 120
 121 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_nativeFinalize(JNIEnv *env, jobject jvm)
 122 {
 123   msg_vm_t vm = jvm_get_native(env,jvm);
 124   MSG_vm_destroy(vm);
 125 }
 126
 127 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_start(JNIEnv *env, jobject jvm)
 128 {
 129   msg_vm_t vm = jvm_get_native(env,jvm);
 130   MSG_vm_start(vm);
 131 }
 132
 133 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_shutdown(JNIEnv *env, jobject jvm)
 134 {
 135   msg_vm_t vm = jvm_get_native(env,jvm);
 136   if (vm)
 137     MSG_vm_shutdown(vm);
 138 }
 139
 140 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_destroy(JNIEnv* env, jobject jvm)
 141 {
 142   msg_vm_t vm = jvm_get_native(env, jvm);
 143   if (vm) {
 144     MSG_vm_destroy(vm);
 145     auto* vmList = &simgrid::vm::VirtualMachineImpl::allVms_;
 146     vmList->erase(std::remove(vmList->begin(), vmList->end(), vm), vmList->end());
 147   }
 148 }
 149
 150 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_nativeMigration(JNIEnv* env, jobject jvm, jobject jhost)
 151 {
 152   msg_vm_t vm = jvm_get_native(env,jvm);
 153   msg_host_t host = jhost_get_native(env, jhost);
 154   if (not simgrid::ForcefulKillException::try_n_catch([&vm, &host]() { MSG_vm_migrate(vm, host); })) {
 155     XBT_VERB("Caught exception during migration");
 156     jxbt_throw_host_failure(env, "during migration");
 157   }
 158 }
 159
 160 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_suspend(JNIEnv *env, jobject jvm)
 161 {
 162   msg_vm_t vm = jvm_get_native(env,jvm);
 163   MSG_vm_suspend(vm);
 164 }
 165
 166 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_resume(JNIEnv *env, jobject jvm)
 167 {
 168   msg_vm_t vm = jvm_get_native(env,jvm);
 169   MSG_vm_resume(vm);
 170 }
 171
 172 JNIEXPORT jobject JNICALL Java_org_simgrid_msg_VM_getVMByName(JNIEnv* env, jclass cls, jstring jname)
 173 {
 174
 175   /* get the C string from the java string */
 176   if (jname == nullptr) {
 177     jxbt_throw_null(env, "No VM can have a null name");
 178     return nullptr;
 179   }
 180   const char* name = env->GetStringUTFChars(jname, 0);
 181   /* get the VM by name   (VMs are just special hosts, unfortunately) */
 182   msg_host_t host = MSG_host_by_name(name);
 183
 184   if (not host) { /* invalid name */
 185     jxbt_throw_host_not_found(env, name);
 186     env->ReleaseStringUTFChars(jname, name);
 187     return nullptr;
 188   }
 189   env->ReleaseStringUTFChars(jname, name);
 190
 191   return static_cast<jobject>(host->extension(JAVA_HOST_LEVEL));
 192 }