Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Implement VM::all() as a native, and use it in tests
[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.h"
11 #include "src/plugins/vm/VirtualMachineImpl.hpp"
12 #include "xbt/ex.hpp"
13
14 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(java);
15
16 SG_BEGIN_DECL()
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 jramsize, jint jmig_netspeed, 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(host, name, static_cast<int>(jramsize), static_cast<int>(jmig_netspeed),
75                               static_cast<int>(jdp_intensity));
76   env->ReleaseStringUTFChars(jname, name);
77
78   jvm_bind(env, jvm, vm);
79   jvm = env->NewWeakGlobalRef(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
99   vms.shrink_to_fit();
100   int count = vms.size();
101
102   jclass cls = jxbt_get_class(env, "org/simgrid/msg/VM");
103   if (!cls)
104     return nullptr;
105
106   jobjectArray jtable = env->NewObjectArray((jsize)count, cls, nullptr);
107   if (!jtable) {
108     jxbt_throw_jni(env, "Hosts table allocation failed");
109     return nullptr;
110   }
111
112   for (int index = 0; index < count; index++) {
113     jobject jhost = vms.at(index);
114     env->SetObjectArrayElement(jtable, index, jhost);
115   }
116   return jtable;
117 }
118
119 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_nativeFinalize(JNIEnv *env, jobject jvm)
120 {
121   msg_vm_t vm = jvm_get_native(env,jvm);
122   MSG_vm_destroy(vm);
123 }
124
125 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_start(JNIEnv *env, jobject jvm)
126 {
127   msg_vm_t vm = jvm_get_native(env,jvm);
128   MSG_vm_start(vm);
129 }
130
131 JNIEXPORT void JNICALL Java_org_simgrid_msg_VM_shutdown(JNIEnv *env, jobject jvm)
132 {
133   msg_vm_t vm = jvm_get_native(env,jvm);
134   if (vm) {
135     MSG_vm_shutdown(vm);
136     auto vmList = &simgrid::vm::VirtualMachineImpl::allVms_;
137     vmList->erase(
138         std::remove_if(vmList->begin(), vmList->end(), [vm](simgrid::s4u::VirtualMachine* it) { return vm == it; }),
139         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 SG_END_DECL()