src/bindings/java/jmsg_vm.cpp

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