Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
fix a segfault when Host.getByName(null) is called
[simgrid.git] / src / bindings / java / jmsg_host.c
1 /* Functions related to the java host instances.                            */
2
3 /* Copyright (c) 2007-2014. The SimGrid Team.
4  * All rights reserved.                                                     */
5
6 /* This program is free software; you can redistribute it and/or modify it
7  * under the terms of the license (GNU LGPL) which comes with this package. */
8
9 #include "xbt/str.h"
10 #include "msg/msg.h"
11 #include "jmsg.h"
12 #include "jmsg_host.h"
13 #include "jxbt_utilities.h"
14
15 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(jmsg);
16
17 static jmethodID jhost_method_Host_constructor;
18 static jfieldID jhost_field_Host_bind;
19 static jfieldID jhost_field_Host_name;
20
21 jobject jhost_new_instance(JNIEnv * env) {
22   jclass cls = jxbt_get_class(env, "org/simgrid/msg/Host");
23   return (*env)->NewObject(env, cls, jhost_method_Host_constructor);
24 }
25
26 jobject jhost_ref(JNIEnv * env, jobject jhost) {
27   return (*env)->NewGlobalRef(env, jhost);
28 }
29
30 void jhost_unref(JNIEnv * env, jobject jhost) {
31   (*env)->DeleteGlobalRef(env, jhost);
32 }
33
34 void jhost_bind(jobject jhost, msg_host_t host, JNIEnv * env) {
35   (*env)->SetLongField(env, jhost, jhost_field_Host_bind, (jlong) (long) (host));
36 }
37
38 msg_host_t jhost_get_native(JNIEnv * env, jobject jhost) {
39   return (msg_host_t) (long) (*env)->GetLongField(env, jhost, jhost_field_Host_bind);
40 }
41
42 const char *jhost_get_name(jobject jhost, JNIEnv * env) {
43   msg_host_t host = jhost_get_native(env, jhost);
44   return MSG_host_get_name(host);
45 }
46
47 jboolean jhost_is_valid(jobject jhost, JNIEnv * env) {
48   if ((*env)->GetLongField(env, jhost, jhost_field_Host_bind)) {
49     return JNI_TRUE;
50   } else {
51     return JNI_FALSE;
52   }
53 }
54
55 JNIEXPORT void JNICALL
56 Java_org_simgrid_msg_Host_nativeInit(JNIEnv *env, jclass cls) {
57   jclass class_Host = (*env)->FindClass(env, "org/simgrid/msg/Host");
58   jhost_method_Host_constructor = (*env)->GetMethodID(env, class_Host, "<init>", "()V");
59   jhost_field_Host_bind = jxbt_get_jfield(env,class_Host, "bind", "J");
60   jhost_field_Host_name = jxbt_get_jfield(env, class_Host, "name", "Ljava/lang/String;");
61   if (!class_Host || !jhost_field_Host_name || !jhost_method_Host_constructor || !jhost_field_Host_bind) {
62     jxbt_throw_native(env,bprintf("Can't find some fields in Java class. You should report this bug."));
63   }
64 }
65 JNIEXPORT jobject JNICALL
66 Java_org_simgrid_msg_Host_getByName(JNIEnv * env, jclass cls,
67                                          jstring jname) {
68   msg_host_t host;                /* native host                                          */
69   jobject jhost;                /* global reference to the java host instance returned  */
70
71   /* get the C string from the java string */
72   if (jname == NULL) {
73         jxbt_throw_null(env,bprintf("No host can have a null name"));
74         return NULL;
75   }
76   const char *name = (*env)->GetStringUTFChars(env, jname, 0);
77   /* get the host by name       (the hosts are created during the grid resolution) */
78   host = MSG_get_host_by_name(name);
79
80   if (!host) {                  /* invalid name */
81     jxbt_throw_host_not_found(env, name);
82     (*env)->ReleaseStringUTFChars(env, jname, name);
83     return NULL;
84   }
85   (*env)->ReleaseStringUTFChars(env, jname, name);
86
87   if (!xbt_lib_get_level(host, JAVA_HOST_LEVEL)) {       /* native host not associated yet with java host */
88
89     /* Instantiate a new java host */
90     jhost = jhost_new_instance(env);
91
92     if (!jhost) {
93       jxbt_throw_jni(env, "java host instantiation failed");
94       return NULL;
95     }
96
97     /* get a global reference to the newly created host */
98     jhost = jhost_ref(env, jhost);
99
100     if (!jhost) {
101       jxbt_throw_jni(env, "new global ref allocation failed");
102       return NULL;
103     }
104     /* Sets the java host name */
105     (*env)->SetObjectField(env, jhost, jhost_field_Host_name, jname);
106     /* bind the java host and the native host */
107     jhost_bind(jhost, host, env);
108
109     /* the native host data field is set with the global reference to the
110      * java host returned by this function
111      */
112     xbt_lib_set(host_lib, host->key, JAVA_HOST_LEVEL, (void *) jhost);
113   }
114
115   /* return the global reference to the java host instance */
116   return (jobject) xbt_lib_get_level(host, JAVA_HOST_LEVEL);
117 }
118
119 JNIEXPORT jobject JNICALL
120 Java_org_simgrid_msg_Host_currentHost(JNIEnv * env, jclass cls) {
121   jobject jhost;
122
123   msg_host_t host = MSG_host_self();
124
125   if (!xbt_lib_get_level(host, JAVA_HOST_LEVEL)) {
126     /* the native host not yet associated with the java host instance */
127
128     /* instanciate a new java host instance */
129     jhost = jhost_new_instance(env);
130
131     if (!jhost) {
132       jxbt_throw_jni(env, "java host instantiation failed");
133       return NULL;
134     }
135
136     /* get a global reference to the newly created host */
137     jhost = jhost_ref(env, jhost);
138
139     if (!jhost) {
140       jxbt_throw_jni(env, "global ref allocation failed");
141       return NULL;
142     }
143     /* Sets the host name */
144     const char *name = MSG_host_get_name(host);
145     jobject jname = (*env)->NewStringUTF(env,name);
146     (*env)->SetObjectField(env, jhost, jhost_field_Host_name, jname);
147     /* Bind & store it */
148     jhost_bind(jhost, host, env);
149     xbt_lib_set(host_lib, host->key, JAVA_HOST_LEVEL, (void *) jhost);
150   } else {
151     jhost = (jobject) xbt_lib_get_level(host, JAVA_HOST_LEVEL);
152   }
153
154   return jhost;
155 }
156
157 JNIEXPORT void JNICALL
158 Java_org_simgrid_msg_Host_on(JNIEnv *env, jobject jhost) {
159   msg_host_t host = jhost_get_native(env, jhost);
160   MSG_host_on(host);
161 }
162
163 JNIEXPORT void JNICALL
164 Java_org_simgrid_msg_Host_off(JNIEnv *env, jobject jhost) {
165   msg_host_t host = jhost_get_native(env, jhost);
166   MSG_host_off(host);
167 }
168
169 JNIEXPORT jint JNICALL
170 Java_org_simgrid_msg_Host_getCount(JNIEnv * env, jclass cls) {
171   xbt_dynar_t hosts =  MSG_hosts_as_dynar();
172   int nb_host = xbt_dynar_length(hosts);
173   xbt_dynar_free(&hosts);
174   return (jint) nb_host;
175 }
176
177 JNIEXPORT jdouble JNICALL
178 Java_org_simgrid_msg_Host_getSpeed(JNIEnv * env,
179                                         jobject jhost) {
180   msg_host_t host = jhost_get_native(env, jhost);
181
182   if (!host) {
183     jxbt_throw_notbound(env, "host", jhost);
184     return -1;
185   }
186
187   return (jdouble) MSG_get_host_speed(host);
188 }
189
190 JNIEXPORT jdouble JNICALL
191 Java_org_simgrid_msg_Host_getCoreNumber(JNIEnv * env,
192                                         jobject jhost) {
193   msg_host_t host = jhost_get_native(env, jhost);
194
195   if (!host) {
196     jxbt_throw_notbound(env, "host", jhost);
197     return -1;
198   }
199
200   return (jdouble) MSG_host_get_core_number(host);
201 }
202
203 JNIEXPORT jint JNICALL
204 Java_org_simgrid_msg_Host_getLoad(JNIEnv * env, jobject jhost) {
205   msg_host_t host = jhost_get_native(env, jhost);
206
207   if (!host) {
208     jxbt_throw_notbound(env, "host", jhost);
209     return -1;
210   }
211
212   return (jint) MSG_get_host_msgload(host);
213 }
214 JNIEXPORT jobject JNICALL
215 Java_org_simgrid_msg_Host_getProperty(JNIEnv *env, jobject jhost, jobject jname) {
216   msg_host_t host = jhost_get_native(env, jhost);
217
218   if (!host) {
219     jxbt_throw_notbound(env, "host", jhost);
220     return NULL;
221   }
222   const char *name = (*env)->GetStringUTFChars(env, jname, 0);
223
224   const char *property = MSG_host_get_property_value(host, name);
225   if (!property) {
226     return NULL;
227   }
228
229   jobject jproperty = (*env)->NewStringUTF(env, property);
230
231   (*env)->ReleaseStringUTFChars(env, jname, name);
232
233   return jproperty;
234 }
235
236 JNIEXPORT void JNICALL
237 Java_org_simgrid_msg_Host_setProperty(JNIEnv *env, jobject jhost, jobject jname, jobject jvalue) {
238   msg_host_t host = jhost_get_native(env, jhost);
239
240   if (!host) {
241     jxbt_throw_notbound(env, "host", jhost);
242     return;
243   }
244   const char *name = (*env)->GetStringUTFChars(env, jname, 0);
245   const char *value_java = (*env)->GetStringUTFChars(env, jvalue, 0);
246   char *value = strdup(value_java);
247
248   MSG_host_set_property_value(host,name,value,xbt_free);
249
250   (*env)->ReleaseStringUTFChars(env, jvalue, value);
251   (*env)->ReleaseStringUTFChars(env, jname, name);
252
253 }
254 JNIEXPORT jboolean JNICALL
255 Java_org_simgrid_msg_Host_isAvail(JNIEnv * env, jobject jhost) {
256   msg_host_t host = jhost_get_native(env, jhost);
257
258   if (!host) {
259     jxbt_throw_notbound(env, "host", jhost);
260     return 0;
261   }
262
263   return (jboolean) MSG_host_is_avail(host);
264 }
265
266 JNIEXPORT jobjectArray JNICALL
267 Java_org_simgrid_msg_Host_all(JNIEnv * env, jclass cls_arg)
268 {
269   int index;
270   jobjectArray jtable;
271   jobject jhost;
272   jstring jname;
273   msg_host_t host;
274
275   xbt_dynar_t table =  MSG_hosts_as_dynar();
276   int count = xbt_dynar_length(table);
277
278   jclass cls = jxbt_get_class(env, "org/simgrid/msg/Host");
279
280   if (!cls) {
281     return NULL;
282   }
283
284   jtable = (*env)->NewObjectArray(env, (jsize) count, cls, NULL);
285
286   if (!jtable) {
287     jxbt_throw_jni(env, "Hosts table allocation failed");
288     return NULL;
289   }
290
291   for (index = 0; index < count; index++) {
292     host = xbt_dynar_get_as(table,index,msg_host_t);
293     jhost = (jobject) (xbt_lib_get_level(host, JAVA_HOST_LEVEL));
294
295     if (!jhost) {
296       jname = (*env)->NewStringUTF(env, MSG_host_get_name(host));
297
298       jhost =
299                 Java_org_simgrid_msg_Host_getByName(env, cls_arg, jname);
300       /* FIXME: leak of jname ? */
301     }
302
303     (*env)->SetObjectArrayElement(env, jtable, index, jhost);
304   }
305   xbt_dynar_free(&table);
306   return jtable;
307 }
308
309 JNIEXPORT void JNICALL 
310 Java_org_simgrid_msg_Host_setAsyncMailbox(JNIEnv * env, jclass cls_arg, jobject jname){
311
312   const char *name = (*env)->GetStringUTFChars(env, jname, 0);
313   MSG_mailbox_set_async(name);
314   (*env)->ReleaseStringUTFChars(env, jname, name);
315
316 }