Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
add Process.daemonize to the Java bindings
[simgrid.git] / src / bindings / java / jmsg_process.cpp
1 /* Functions related to the java process instances.                         */
2
3 /* Copyright (c) 2007-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 <xbt/ex.hpp>
9
10 #include "jmsg_process.h"
11
12 #include "JavaContext.hpp"
13 #include "jmsg.hpp"
14 #include "jmsg_host.h"
15 #include "jxbt_utilities.hpp"
16
17 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(java);
18
19 extern "C" {
20
21 jfieldID jprocess_field_Process_bind;
22 jfieldID jprocess_field_Process_host;
23 jfieldID jprocess_field_Process_killTime;
24 jfieldID jprocess_field_Process_name;
25 jfieldID jprocess_field_Process_pid;
26 jfieldID jprocess_field_Process_ppid;
27
28 jobject jprocess_from_native(msg_process_t process)
29 {
30   simgrid::kernel::context::JavaContext* context = (simgrid::kernel::context::JavaContext*) MSG_process_get_smx_ctx(process);
31   return context->jprocess;
32 }
33
34 jobject jprocess_ref(jobject jprocess, JNIEnv* env)
35 {
36   return env->NewGlobalRef(jprocess);
37 }
38
39 void jprocess_unref(jobject jprocess, JNIEnv* env)
40 {
41   env->DeleteGlobalRef(jprocess);
42 }
43
44 msg_process_t jprocess_to_native(jobject jprocess, JNIEnv* env)
45 {
46   return (msg_process_t)(intptr_t)env->GetLongField(jprocess, jprocess_field_Process_bind);
47 }
48
49 void jprocess_bind(jobject jprocess, msg_process_t process, JNIEnv * env)
50 {
51   env->SetLongField(jprocess, jprocess_field_Process_bind, (intptr_t)process);
52 }
53
54 JNIEXPORT void JNICALL Java_org_simgrid_msg_Process_nativeInit(JNIEnv *env, jclass cls) {
55   jclass jprocess_class_Process = env->FindClass("org/simgrid/msg/Process");
56   xbt_assert(jprocess_class_Process, "Native initialization of msg/Process failed. Please report that bug");
57
58   jprocess_field_Process_name = jxbt_get_jfield(env, jprocess_class_Process, "name", "Ljava/lang/String;");
59   jprocess_field_Process_bind = jxbt_get_jfield(env, jprocess_class_Process, "bind", "J");
60   jprocess_field_Process_pid = jxbt_get_jfield(env, jprocess_class_Process, "pid", "I");
61   jprocess_field_Process_ppid = jxbt_get_jfield(env, jprocess_class_Process, "ppid", "I");
62   jprocess_field_Process_host = jxbt_get_jfield(env, jprocess_class_Process, "host", "Lorg/simgrid/msg/Host;");
63   jprocess_field_Process_killTime = jxbt_get_jfield(env, jprocess_class_Process, "killTime", "D");
64   xbt_assert(jprocess_field_Process_name && jprocess_field_Process_pid && jprocess_field_Process_ppid &&
65                  jprocess_field_Process_host,
66              "Native initialization of msg/Process failed. Please report that bug");
67 }
68
69 JNIEXPORT void JNICALL Java_org_simgrid_msg_Process_create(JNIEnv* env, jobject jprocess_arg, jobject jhost)
70 {
71   /* create a global java process instance */
72   jobject jprocess = jprocess_ref(jprocess_arg, env);
73
74   /* Actually build the MSG process */
75   jstring jname         = (jstring)env->GetObjectField(jprocess, jprocess_field_Process_name);
76   const char* name      = env->GetStringUTFChars(jname, 0);
77   msg_process_t process =
78       MSG_process_create_from_stdfunc(name, [jprocess]() { simgrid::kernel::context::java_main_jprocess(jprocess); },
79                                       /*data*/ nullptr, jhost_get_native(env, jhost), /* properties*/ nullptr);
80   env->ReleaseStringUTFChars(jname, name);
81
82   /* bind the java process instance to the native process */
83   jprocess_bind(jprocess, process, env);
84
85   /* Retrieve the kill time from the process */
86   jdouble jkill = env->GetDoubleField(jprocess, jprocess_field_Process_killTime);
87   MSG_process_set_kill_time(process, (double)jkill);
88
89   /* sets the PID and the PPID of the process */
90   env->SetIntField(jprocess, jprocess_field_Process_pid,(jint) MSG_process_get_PID(process));
91   env->SetIntField(jprocess, jprocess_field_Process_ppid, (jint) MSG_process_get_PPID(process));
92 }
93
94 JNIEXPORT void JNICALL Java_org_simgrid_msg_Process_daemonize(JNIEnv* env, jobject jprocess)
95 {
96   msg_process_t process = jprocess_to_native(jprocess, env);
97
98   if (not process) {
99     jxbt_throw_notbound(env, "process", jprocess);
100     return;
101   }
102
103   MSG_process_daemonize(process);
104 }
105
106 JNIEXPORT jint JNICALL Java_org_simgrid_msg_Process_killAll(JNIEnv * env, jclass cls, jint jresetPID)
107 {
108   return (jint) MSG_process_killall((int) jresetPID);
109 }
110
111 JNIEXPORT jobject JNICALL Java_org_simgrid_msg_Process_fromPID(JNIEnv * env, jclass cls, jint pid)
112 {
113   msg_process_t process = MSG_process_from_PID(pid);
114
115   if (not process) {
116     jxbt_throw_process_not_found(env, std::string("PID = ") + std::to_string(static_cast<int>(pid)));
117     return nullptr;
118   }
119
120   jobject jprocess = jprocess_from_native(process);
121
122   if (not jprocess) {
123     jxbt_throw_jni(env, "get process failed");
124     return nullptr;
125   }
126
127   return jprocess;
128 }
129
130 JNIEXPORT jobject JNICALL Java_org_simgrid_msg_Process_getProperty(JNIEnv *env, jobject jprocess, jobject jname) {
131   msg_process_t process = jprocess_to_native(jprocess, env);
132
133   if (not process) {
134     jxbt_throw_notbound(env, "process", jprocess);
135     return nullptr;
136   }
137   const char *name = env->GetStringUTFChars((jstring)jname, 0);
138
139   const char *property = MSG_process_get_property_value(process, name);
140   if (not property)
141     return nullptr;
142
143   jobject jproperty = env->NewStringUTF(property);
144
145   env->ReleaseStringUTFChars((jstring)jname, name);
146
147   return jproperty;
148 }
149
150 JNIEXPORT jobject JNICALL Java_org_simgrid_msg_Process_getCurrentProcess(JNIEnv * env, jclass cls)
151 {
152   jobject jprocess = jprocess_from_native(MSG_process_self());
153   if (not jprocess)
154     jxbt_throw_jni(env, xbt_strdup("SIMIX_process_get_jprocess() failed"));
155
156   return jprocess;
157 }
158
159 JNIEXPORT void JNICALL Java_org_simgrid_msg_Process_suspend(JNIEnv * env, jobject jprocess)
160 {
161   msg_process_t process = jprocess_to_native(jprocess, env);
162
163   if (not process) {
164     jxbt_throw_notbound(env, "process", jprocess);
165     return;
166   }
167
168   /* try to suspend the process */
169   msg_error_t rv = MSG_process_suspend(process);
170
171   jxbt_check_res("MSG_process_suspend()", rv, MSG_OK, "unexpected error , please report this bug");
172 }
173
174 JNIEXPORT void JNICALL Java_org_simgrid_msg_Process_resume(JNIEnv * env, jobject jprocess)
175 {
176   msg_process_t process = jprocess_to_native(jprocess, env);
177
178   if (not process) {
179     jxbt_throw_notbound(env, "process", jprocess);
180     return;
181   }
182
183   /* try to resume the process */
184   msg_error_t res = MSG_process_resume(process);
185   jxbt_check_res("MSG_process_resume()", res, MSG_OK, "unexpected error , please report this bug");
186 }
187
188 JNIEXPORT void
189 JNICALL Java_org_simgrid_msg_Process_setAutoRestart (JNIEnv *env, jobject jprocess, jboolean jauto_restart) {
190
191   msg_process_t process = jprocess_to_native(jprocess, env);
192   if (not process) {
193     jxbt_throw_notbound(env, "process", jprocess);
194     return;
195   }
196
197   MSG_process_auto_restart_set(process, (jauto_restart == JNI_TRUE));
198 }
199
200 JNIEXPORT void JNICALL Java_org_simgrid_msg_Process_restart (JNIEnv *env, jobject jprocess) {
201   msg_process_t process = jprocess_to_native(jprocess, env);
202
203   if (not process) {
204     jxbt_throw_notbound(env, "process", jprocess);
205     return;
206   }
207
208   MSG_process_restart(process);
209 }
210
211 JNIEXPORT jboolean JNICALL Java_org_simgrid_msg_Process_isSuspended(JNIEnv * env, jobject jprocess)
212 {
213   msg_process_t process = jprocess_to_native(jprocess, env);
214
215   if (not process) {
216     jxbt_throw_notbound(env, "process", jprocess);
217     return 0;
218   }
219
220   /* true is the process is suspended, false otherwise */
221   return (jboolean) MSG_process_is_suspended(process);
222 }
223
224 JNIEXPORT void JNICALL Java_org_simgrid_msg_Process_sleep(JNIEnv *env, jclass cls, jlong jmillis, jint jnanos)
225  {
226   double time =  ((double)jmillis) / 1000 + ((double)jnanos) / 1000000000;
227   msg_error_t rv;
228   rv = MSG_process_sleep(time);
229   if (rv != MSG_OK) {
230     XBT_DEBUG("Something during the MSG_process_sleep invocation was wrong, trigger a HostFailureException");
231
232     jxbt_throw_host_failure(env, "");
233   }
234 }
235
236 JNIEXPORT void JNICALL Java_org_simgrid_msg_Process_waitFor(JNIEnv * env, jobject jprocess, jdouble jseconds)
237 {
238   msg_error_t rv;
239   rv = MSG_process_sleep((double)jseconds);
240   if (env->ExceptionOccurred())
241     return;
242   if (rv != MSG_OK) {
243     XBT_DEBUG("Status NOK");
244     jmsg_throw_status(env,rv);
245   }
246 }
247
248 JNIEXPORT void JNICALL Java_org_simgrid_msg_Process_kill(JNIEnv * env, jobject jprocess)
249 {
250   /* get the native instances from the java ones */
251   msg_process_t process = jprocess_to_native(jprocess, env);
252   if (not process) {
253     jxbt_throw_notbound(env, "process", jprocess);
254     return;
255   }
256   try {
257     MSG_process_kill(process);
258   } catch (xbt_ex& ex) {
259     XBT_VERB("Process %s just committed a suicide", MSG_process_get_name(process));
260     xbt_assert(process == MSG_process_self(),
261                "Killing a process should not raise an exception if it's not a suicide. Please report that bug.");
262   }
263 }
264
265 JNIEXPORT void JNICALL Java_org_simgrid_msg_Process_migrate(JNIEnv * env, jobject jprocess, jobject jhost)
266 {
267   msg_process_t process = jprocess_to_native(jprocess, env);
268
269   if (not process) {
270     jxbt_throw_notbound(env, "process", jprocess);
271     return;
272   }
273
274   msg_host_t host = jhost_get_native(env, jhost);
275
276   if (not host) {
277     jxbt_throw_notbound(env, "host", jhost);
278     return;
279   }
280
281   /* try to change the host of the process */
282   msg_error_t rv = MSG_process_migrate(process, host);
283   if (rv != MSG_OK) {
284     jmsg_throw_status(env,rv);
285     return;
286   }
287   /* change the host java side */
288   env->SetObjectField(jprocess, jprocess_field_Process_host, jhost);
289 }
290
291 JNIEXPORT void JNICALL Java_org_simgrid_msg_Process_yield(JNIEnv* env, jclass cls)
292 {
293   MSG_process_yield();
294 }
295
296 JNIEXPORT void JNICALL Java_org_simgrid_msg_Process_setKillTime (JNIEnv *env , jobject jprocess, jdouble jkilltime) {
297   msg_process_t process = jprocess_to_native(jprocess, env);
298   MSG_process_set_kill_time(process, (double)jkilltime);
299 }
300
301 JNIEXPORT jint JNICALL Java_org_simgrid_msg_Process_getCount(JNIEnv * env, jclass cls) {
302   return (jint) MSG_process_get_number();
303 }
304 }