Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
distinguish shutdown and destroy in Java VMs
[simgrid.git] / 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 }
140
141 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_destroy(JNIEnv* env, jobject jvm)
142 {
143   msg_vm_t vm = jvm_get_native(env, jvm);
144   if (vm) {
145     MSG_vm_destroy(vm);
146     auto vmList = &simgrid::vm::VirtualMachineImpl::allVms_;
147     vmList->erase(std::remove(vmList->begin(), vmList->end(), vm), vmList->end());
148   }
149 }
150
151 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_nativeMigration(JNIEnv* env, jobject jvm, jobject jhost)
152 {
153   msg_vm_t vm = jvm_get_native(env,jvm);
154   msg_host_t host = jhost_get_native(env, jhost);
155   try {
156     MSG_vm_migrate(vm,host);
157   }
158   catch(xbt_ex& e){
159      XBT_VERB("CATCH EXCEPTION MIGRATION %s",e.what());
160      jxbt_throw_host_failure(env, (char*)"during migration");
161   }
162 }
163
164 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_suspend(JNIEnv *env, jobject jvm)
165 {
166   msg_vm_t vm = jvm_get_native(env,jvm);
167   MSG_vm_suspend(vm);
168 }
169
170 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_resume(JNIEnv *env, jobject jvm)
171 {
172   msg_vm_t vm = jvm_get_native(env,jvm);
173   MSG_vm_resume(vm);
174 }
175
176 JNIEXPORT jobject JNICALL Java_org_simgrid_msg_VM_getVMByName(JNIEnv* env, jclass cls, jstring jname)
177 {
178
179   /* get the C string from the java string */
180   if (jname == nullptr) {
181     jxbt_throw_null(env, "No VM can have a null name");
182     return nullptr;
183   }
184   const char* name = env->GetStringUTFChars(jname, 0);
185   /* get the VM by name   (VMs are just special hosts, unfortunately) */
186   msg_host_t host = MSG_host_by_name(name);
187
188   if (not host) { /* invalid name */
189     jxbt_throw_host_not_found(env, name);
190     env->ReleaseStringUTFChars(jname, name);
191     return nullptr;
192   }
193   env->ReleaseStringUTFChars(jname, name);
194
195   return static_cast<jobject>(host->extension(JAVA_HOST_LEVEL));
196 }
197 }