Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Merge branch 'master' into actor-yield
[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 "src/plugins/vm/VirtualMachineImpl.hpp"
12 #include "xbt/ex.hpp"
13
14 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(java);
15
16 extern "C" {
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(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->getState() != SURF_VM_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     auto vmList = &simgrid::vm::VirtualMachineImpl::allVms_;
139     vmList->erase(std::remove(vmList->begin(), vmList->end(), vm), vmList->end());
140   }
141 }
142
143 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_nativeMigration(JNIEnv* env, jobject jvm, jobject jhost)
144 {
145   msg_vm_t vm = jvm_get_native(env,jvm);
146   msg_host_t host = jhost_get_native(env, jhost);
147   try {
148     MSG_vm_migrate(vm,host);
149   }
150   catch(xbt_ex& e){
151      XBT_VERB("CATCH EXCEPTION MIGRATION %s",e.what());
152      jxbt_throw_host_failure(env, (char*)"during migration");
153   }
154 }
155
156 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_suspend(JNIEnv *env, jobject jvm)
157 {
158   msg_vm_t vm = jvm_get_native(env,jvm);
159   MSG_vm_suspend(vm);
160 }
161
162 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_resume(JNIEnv *env, jobject jvm)
163 {
164   msg_vm_t vm = jvm_get_native(env,jvm);
165   MSG_vm_resume(vm);
166 }
167
168 JNIEXPORT jobject JNICALL Java_org_simgrid_msg_VM_getVMByName(JNIEnv* env, jclass cls, jstring jname)
169 {
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   const char* name = env->GetStringUTFChars(jname, 0);
177   /* get the VM by name   (VMs are just special hosts, unfortunately) */
178   msg_host_t host = MSG_host_by_name(name);
179
180   if (not host) { /* invalid name */
181     jxbt_throw_host_not_found(env, name);
182     env->ReleaseStringUTFChars(jname, name);
183     return nullptr;
184   }
185   env->ReleaseStringUTFChars(jname, name);
186
187   return static_cast<jobject>(host->extension(JAVA_HOST_LEVEL));
188 }
189 }