src/bindings/java/jmsg_vm.cpp

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