Logo AND Algorithmique Numérique Distribuée

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