Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
input sanitization continued.
[simgrid.git] / src / bindings / java / jmsg_storage.cpp
1 /* Java bindings of the Storage API.                                        */
2
3 /* Copyright (c) 2012-2019. 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 "simgrid/msg.h"
9 #include "simgrid/plugins/file_system.h"
10
11 #include "include/xbt/signal.hpp"
12 #include "jmsg.hpp"
13 #include "jmsg_storage.h"
14 #include "jxbt_utilities.hpp"
15 #include "simgrid/s4u/Storage.hpp"
16
17 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(java);
18
19 static jmethodID jstorage_method_Storage_constructor;
20 static jfieldID jstorage_field_Storage_bind;
21 static jfieldID jstorage_field_Storage_name;
22
23 jobject jstorage_new_instance(JNIEnv * env) {
24   jclass cls = jxbt_get_class(env, "org/simgrid/msg/Storage");
25   return env->NewObject(cls, jstorage_method_Storage_constructor);
26 }
27
28 msg_storage_t jstorage_get_native(JNIEnv * env, jobject jstorage) {
29   return (msg_storage_t) (uintptr_t) env->GetLongField(jstorage, jstorage_field_Storage_bind);
30 }
31
32 JNIEXPORT void JNICALL Java_org_simgrid_msg_Storage_nativeInit(JNIEnv *env, jclass cls) {
33   jclass class_Storage = env->FindClass("org/simgrid/msg/Storage");
34   xbt_assert(class_Storage, "Native initialization of msg/Storage failed. Please report that bug");
35   jstorage_method_Storage_constructor = env->GetMethodID(class_Storage, "<init>", "()V");
36   jstorage_field_Storage_bind = jxbt_get_jfield(env,class_Storage, "bind", "J");
37   jstorage_field_Storage_name = jxbt_get_jfield(env, class_Storage, "name", "Ljava/lang/String;");
38   xbt_assert(jstorage_field_Storage_name && jstorage_method_Storage_constructor && jstorage_field_Storage_bind,
39              "Native initialization of msg/Storage failed. Please report that bug");
40 }
41
42 void jstorage_bind(jobject jstorage, msg_storage_t storage, JNIEnv * env) {
43   env->SetLongField(jstorage, jstorage_field_Storage_bind, (jlong) (uintptr_t) (storage));
44 }
45
46 jobject jstorage_ref(JNIEnv * env, jobject jstorage) {
47   return env->NewGlobalRef(jstorage);
48 }
49
50 void jstorage_unref(JNIEnv * env, jobject jstorage) {
51   env->DeleteGlobalRef(jstorage);
52 }
53
54 JNIEXPORT jobject JNICALL Java_org_simgrid_msg_Storage_getByName(JNIEnv * env, jclass cls, jstring jname) {
55   msg_storage_t storage;
56   jobject jstorage = nullptr;
57
58   /* get the C string from the java string */
59   if (jname == nullptr) {
60     jxbt_throw_null(env, "No host can have a null name");
61     return nullptr;
62   }
63   const char *name = env->GetStringUTFChars(jname, 0);
64   storage = MSG_storage_get_by_name(name);
65
66   if (not storage) { /* invalid name */
67     jxbt_throw_storage_not_found(env, name);
68     env->ReleaseStringUTFChars(jname, name);
69     return nullptr;
70   }
71   env->ReleaseStringUTFChars(jname, name);
72
73   if (java_storage_map.find(storage) == java_storage_map.end()) {
74     /* Instantiate a new java storage */
75     jstorage = jstorage_new_instance(env);
76
77     if (not jstorage) {
78       jxbt_throw_jni(env, "java storage instantiation failed");
79       return nullptr;
80     }
81
82     /* get a global reference to the newly created storage */
83     jstorage = jstorage_ref(env, jstorage);
84
85     if (not jstorage) {
86       jxbt_throw_jni(env, "new global ref allocation failed");
87       return nullptr;
88     }
89     /* Sets the java storage name */
90     env->SetObjectField(jstorage, jstorage_field_Storage_name, jname);
91     /* bind the java storage and the native storage */
92     jstorage_bind(jstorage, storage, env);
93
94     /* the native storage data field is set with the global reference to the
95      * java storage returned by this function
96      */
97     java_storage_map.insert({storage, jstorage});
98   } else
99     jstorage = java_storage_map.at(storage);
100
101   /* return the global reference to the java storage instance */
102   return (jobject)jstorage;
103 }
104
105 JNIEXPORT jlong JNICALL Java_org_simgrid_msg_Storage_getSize(JNIEnv * env,jobject jstorage) {
106   msg_storage_t storage = jstorage_get_native(env, jstorage);
107
108   if (not storage) {
109     jxbt_throw_notbound(env, "storage", jstorage);
110     return -1;
111   }
112
113   return (jlong) MSG_storage_get_size(storage);
114 }
115
116 JNIEXPORT jlong JNICALL Java_org_simgrid_msg_Storage_getFreeSize(JNIEnv * env,jobject jstorage) {
117   msg_storage_t storage = jstorage_get_native(env, jstorage);
118
119   if (not storage) {
120     jxbt_throw_notbound(env, "storage", jstorage);
121     return -1;
122   }
123
124   return (jlong) MSG_storage_get_free_size(storage);
125 }
126
127 JNIEXPORT jlong JNICALL Java_org_simgrid_msg_Storage_getUsedSize(JNIEnv * env,jobject jstorage) {
128   msg_storage_t storage = jstorage_get_native(env, jstorage);
129
130   if (not storage) {
131     jxbt_throw_notbound(env, "storage", jstorage);
132     return -1;
133   }
134
135   return (jlong) MSG_storage_get_used_size(storage);
136 }
137
138 JNIEXPORT jobject JNICALL Java_org_simgrid_msg_Storage_getProperty(JNIEnv *env, jobject jstorage, jobject jname) {
139   msg_storage_t storage = jstorage_get_native(env, jstorage);
140
141   if (not storage) {
142     jxbt_throw_notbound(env, "storage", jstorage);
143     return nullptr;
144   }
145   const char *name = env->GetStringUTFChars((jstring) jname, 0);
146
147   const char *property = MSG_storage_get_property_value(storage, name);
148   if (not property) {
149     return nullptr;
150   }
151   jobject jproperty = env->NewStringUTF(property);
152
153   env->ReleaseStringUTFChars((jstring) jname, name);
154
155   return jproperty;
156 }
157
158 JNIEXPORT void JNICALL
159 Java_org_simgrid_msg_Storage_setProperty(JNIEnv *env, jobject jstorage, jobject jname, jobject jvalue) {
160   msg_storage_t storage = jstorage_get_native(env, jstorage);
161
162   if (not storage) {
163     jxbt_throw_notbound(env, "storage", jstorage);
164     return;
165   }
166   const char *name = env->GetStringUTFChars((jstring) jname, 0);
167   const char *value_java = env->GetStringUTFChars((jstring) jvalue, 0);
168
169   storage->set_property(name, std::string(value_java));
170
171   env->ReleaseStringUTFChars((jstring) jvalue, value_java);
172   env->ReleaseStringUTFChars((jstring) jname, name);
173
174 }
175
176 JNIEXPORT jobject JNICALL Java_org_simgrid_msg_Storage_getHost(JNIEnv * env,jobject jstorage) {
177   msg_storage_t storage = jstorage_get_native(env, jstorage);
178
179   if (not storage) {
180     jxbt_throw_notbound(env, "storage", jstorage);
181     return nullptr;
182   }
183   const char *host_name = MSG_storage_get_host(storage);
184   if (not host_name) {
185     return nullptr;
186   }
187   jobject jhost_name = env->NewStringUTF(host_name);
188
189   return jhost_name;
190 }
191
192 JNIEXPORT jobjectArray JNICALL Java_org_simgrid_msg_Storage_all(JNIEnv * env, jclass cls_arg)
193 {
194   int index;
195   jobjectArray jtable;
196   jobject jstorage;
197   jstring jname;
198   msg_storage_t storage;
199
200   xbt_dynar_t table =  MSG_storages_as_dynar();
201   int count = xbt_dynar_length(table);
202
203   jclass cls = jxbt_get_class(env, "org/simgrid/msg/Storage");
204
205   if (not cls) {
206     return nullptr;
207   }
208
209   jtable = env->NewObjectArray((jsize) count, cls, nullptr);
210
211   if (not jtable) {
212     jxbt_throw_jni(env, "Storages table allocation failed");
213     return nullptr;
214   }
215
216   for (index = 0; index < count; index++) {
217     storage = xbt_dynar_get_as(table,index,msg_storage_t);
218     if (java_storage_map.find(storage) != java_storage_map.end()) {
219       jstorage = java_storage_map.at(storage);
220     } else {
221       jname = env->NewStringUTF(MSG_storage_get_name(storage));
222       jstorage = Java_org_simgrid_msg_Storage_getByName(env, cls_arg, jname);
223     }
224
225     env->SetObjectArrayElement(jtable, index, jstorage);
226   }
227   xbt_dynar_free(&table);
228   return jtable;
229 }