src/bindings/java/jmsg_host.cpp

   1 /* Functions related to the java host instances.                            */
   2
   3 /* Copyright (c) 2007-2020. 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/Exception.hpp"
   9 #include "simgrid/plugins/energy.h"
  10 #include "simgrid/plugins/load.h"
  11 #include "simgrid/s4u/Host.hpp"
  12 #include "simgrid/s4u/Storage.hpp"
  13
  14 #include "JavaContext.hpp"
  15 #include "jmsg.hpp"
  16 #include "jmsg_host.h"
  17 #include "jmsg_storage.h"
  18 #include "jxbt_utilities.hpp"
  19
  20 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(java);
  21
  22 static jmethodID jhost_method_Host_constructor;
  23 static jfieldID jhost_field_Host_bind;
  24 static jfieldID jhost_field_Host_name;
  25
  26 jobject jhost_new_instance(JNIEnv * env) {
  27   jclass cls = jxbt_get_class(env, "org/simgrid/msg/Host");
  28   return env->NewObject(cls, jhost_method_Host_constructor);
  29 }
  30
  31 jobject jhost_ref(JNIEnv * env, jobject jhost) {
  32   return env->NewGlobalRef(jhost);
  33 }
  34
  35 void jhost_unref(JNIEnv * env, jobject jhost) {
  36   env->DeleteGlobalRef(jhost);
  37 }
  38
  39 void jhost_bind(jobject jhost, msg_host_t host, JNIEnv * env) {
  40   env->SetLongField(jhost, jhost_field_Host_bind, (jlong) (uintptr_t) (host));
  41 }
  42
  43 msg_host_t jhost_get_native(JNIEnv * env, jobject jhost) {
  44   return (msg_host_t) (uintptr_t) env->GetLongField(jhost, jhost_field_Host_bind);
  45 }
  46
  47 JNIEXPORT void JNICALL Java_org_simgrid_msg_Host_nativeInit(JNIEnv *env, jclass cls) {
  48   jclass class_Host = env->FindClass("org/simgrid/msg/Host");
  49   jhost_method_Host_constructor = env->GetMethodID(class_Host, "<init>", "()V");
  50   jhost_field_Host_bind = jxbt_get_jfield(env,class_Host, "bind", "J");
  51   jhost_field_Host_name = jxbt_get_jfield(env, class_Host, "name", "Ljava/lang/String;");
  52   xbt_assert(class_Host && jhost_field_Host_name && jhost_method_Host_constructor && jhost_field_Host_bind,
  53              "Native initialization of msg/Host failed. Please report that bug");
  54 }
  55
  56 JNIEXPORT jobject JNICALL Java_org_simgrid_msg_Host_getByName(JNIEnv* env, jclass cls, jstring jname)
  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   /* get the host by name       (the hosts are created during the grid resolution) */
  65   sg_host_t host = sg_host_by_name(name);
  66
  67   if (not host) { /* invalid name */
  68     jxbt_throw_host_not_found(env, name);
  69     env->ReleaseStringUTFChars(jname, name);
  70     return nullptr;
  71   }
  72   env->ReleaseStringUTFChars(jname, name);
  73
  74   if (not host->extension(JAVA_HOST_LEVEL)) { /* native host not associated yet with java host */
  75     /* Instantiate a new java host */
  76     jobject jhost = jhost_new_instance(env);
  77
  78     if (not jhost) {
  79       jxbt_throw_jni(env, "java host instantiation failed");
  80       return nullptr;
  81     }
  82
  83     /* get a global reference to the newly created host */
  84     jhost = jhost_ref(env, jhost);
  85
  86     if (not jhost) {
  87       jxbt_throw_jni(env, "new global ref allocation failed");
  88       return nullptr;
  89     }
  90     /* Sets the java host name */
  91     env->SetObjectField(jhost, jhost_field_Host_name, jname);
  92     /* bind the java host and the native host */
  93     jhost_bind(jhost, host, env);
  94
  95     /* the native host data field is set with the global reference to the java host returned by this function */
  96     host->extension_set(JAVA_HOST_LEVEL, (void *)jhost);
  97   }
  98
  99   /* return the global reference to the java host instance */
 100   return (jobject) host->extension(JAVA_HOST_LEVEL);
 101 }
 102
 103 JNIEXPORT jobject JNICALL Java_org_simgrid_msg_Host_currentHost(JNIEnv * env, jclass cls) {
 104   jobject jhost;
 105
 106   sg_host_t host = sg_host_self();
 107
 108   if (not host->extension(JAVA_HOST_LEVEL)) {
 109     /* the native host not yet associated with the java host instance */
 110
 111     /* instantiate a new java host instance */
 112     jhost = jhost_new_instance(env);
 113
 114     if (not jhost) {
 115       jxbt_throw_jni(env, "java host instantiation failed");
 116       return nullptr;
 117     }
 118
 119     /* get a global reference to the newly created host */
 120     jhost = jhost_ref(env, jhost);
 121
 122     if (not jhost) {
 123       jxbt_throw_jni(env, "global ref allocation failed");
 124       return nullptr;
 125     }
 126     /* Sets the host name */
 127     jobject jname = env->NewStringUTF(host->get_cname());
 128     env->SetObjectField(jhost, jhost_field_Host_name, jname);
 129     /* Bind & store it */
 130     jhost_bind(jhost, host, env);
 131     host->extension_set(JAVA_HOST_LEVEL, (void *) jhost);
 132   } else {
 133     jhost = (jobject) host->extension(JAVA_HOST_LEVEL);
 134   }
 135
 136   return jhost;
 137 }
 138
 139 JNIEXPORT void JNICALL Java_org_simgrid_msg_Host_on(JNIEnv *env, jobject jhost) {
 140   sg_host_t host = jhost_get_native(env, jhost);
 141   sg_host_turn_on(host);
 142 }
 143
 144 JNIEXPORT void JNICALL Java_org_simgrid_msg_Host_off(JNIEnv *env, jobject jhost) {
 145   sg_host_t host = jhost_get_native(env, jhost);
 146   if (not simgrid::ForcefulKillException::try_n_catch([host]() { sg_host_turn_off(host); }))
 147     jxbt_throw_by_name(env, "org/simgrid/msg/ProcessKilledError", "Host turned off");
 148 }
 149
 150 JNIEXPORT jint JNICALL Java_org_simgrid_msg_Host_getCount(JNIEnv * env, jclass cls) {
 151   return (jint)sg_host_count();
 152 }
 153
 154 JNIEXPORT jdouble JNICALL Java_org_simgrid_msg_Host_getSpeed(JNIEnv * env, jobject jhost) {
 155   const_sg_host_t host = jhost_get_native(env, jhost);
 156
 157   if (not host) {
 158     jxbt_throw_notbound(env, "host", jhost);
 159     return -1;
 160   }
 161
 162   return (jdouble)sg_host_speed(host);
 163 }
 164
 165 JNIEXPORT jdouble JNICALL Java_org_simgrid_msg_Host_getCoreNumber(JNIEnv * env, jobject jhost) {
 166   const_sg_host_t host = jhost_get_native(env, jhost);
 167
 168   if (not host) {
 169     jxbt_throw_notbound(env, "host", jhost);
 170     return -1;
 171   }
 172
 173   return (jdouble)sg_host_core_count(host);
 174 }
 175
 176 JNIEXPORT jobject JNICALL Java_org_simgrid_msg_Host_getProperty(JNIEnv *env, jobject jhost, jobject jname) {
 177   const_sg_host_t host = jhost_get_native(env, jhost);
 178
 179   if (not host) {
 180     jxbt_throw_notbound(env, "host", jhost);
 181     return nullptr;
 182   }
 183   const char *name = env->GetStringUTFChars((jstring) jname, 0);
 184
 185   const char* property = sg_host_get_property_value(host, name);
 186   if (not property) {
 187     return nullptr;
 188   }
 189
 190   jobject jproperty = env->NewStringUTF(property);
 191
 192   env->ReleaseStringUTFChars((jstring) jname, name);
 193
 194   return jproperty;
 195 }
 196
 197 JNIEXPORT void JNICALL
 198 Java_org_simgrid_msg_Host_setProperty(JNIEnv *env, jobject jhost, jobject jname, jobject jvalue) {
 199   sg_host_t host = jhost_get_native(env, jhost);
 200
 201   if (not host) {
 202     jxbt_throw_notbound(env, "host", jhost);
 203     return;
 204   }
 205   const char *name = env->GetStringUTFChars((jstring) jname, 0);
 206   const char *value_java = env->GetStringUTFChars((jstring) jvalue, 0);
 207   const char* value      = xbt_strdup(value_java);
 208
 209   sg_host_set_property_value(host, name, value);
 210
 211   env->ReleaseStringUTFChars((jstring) jvalue, value_java);
 212   env->ReleaseStringUTFChars((jstring) jname, name);
 213 }
 214
 215 JNIEXPORT jboolean JNICALL Java_org_simgrid_msg_Host_isOn(JNIEnv * env, jobject jhost)
 216 {
 217   const_sg_host_t host = jhost_get_native(env, jhost);
 218
 219   if (not host) {
 220     jxbt_throw_notbound(env, "host", jhost);
 221     return 0;
 222   }
 223
 224   return (jboolean)sg_host_is_on(host);
 225 }
 226
 227 JNIEXPORT jobjectArray JNICALL Java_org_simgrid_msg_Host_getMountedStorage(JNIEnv * env, jobject jhost)
 228 {
 229   msg_host_t host = jhost_get_native(env, jhost);
 230   jobject jstorage;
 231   jstring jname;
 232
 233   if (not host) {
 234     jxbt_throw_notbound(env, "host", jhost);
 235     return 0;
 236   }
 237
 238   int index = 0;
 239   jobjectArray jtable;
 240   std::unordered_map<std::string, sg_storage_t> mounted_storages = host->get_mounted_storages();
 241   int count  = mounted_storages.size();
 242   jclass cls = env->FindClass("org/simgrid/msg/Storage");
 243
 244   jtable = env->NewObjectArray((jsize) count, cls, nullptr);
 245
 246   if (not jtable) {
 247     jxbt_throw_jni(env, "Storages table allocation failed");
 248     return nullptr;
 249   }
 250
 251   for (auto const& elm : mounted_storages) {
 252     jname    = env->NewStringUTF(elm.second->get_cname());
 253     jstorage = Java_org_simgrid_msg_Storage_getByName(env,cls,jname);
 254     env->SetObjectArrayElement(jtable, index, jstorage);
 255     index++;
 256   }
 257   return jtable;
 258 }
 259
 260 JNIEXPORT jobjectArray JNICALL Java_org_simgrid_msg_Host_getAttachedStorage(JNIEnv * env, jobject jhost)
 261 {
 262   const_sg_host_t host = jhost_get_native(env, jhost);
 263
 264   if (not host) {
 265     jxbt_throw_notbound(env, "host", jhost);
 266     return 0;
 267   }
 268   jobjectArray jtable;
 269
 270   xbt_dynar_t dyn = sg_host_get_attached_storage_list(host);
 271   jclass cls      = jxbt_get_class(env, "java/lang/String");
 272   jtable          = env->NewObjectArray(static_cast<jsize>(xbt_dynar_length(dyn)), cls, nullptr);
 273   unsigned int index;
 274   const char* storage_name;
 275   jstring jstorage_name;
 276   xbt_dynar_foreach (dyn, index, storage_name) {
 277     jstorage_name = env->NewStringUTF(storage_name);
 278     env->SetObjectArrayElement(jtable, index, jstorage_name);
 279   }
 280   xbt_dynar_free_container(&dyn);
 281   return jtable;
 282 }
 283
 284 JNIEXPORT jobjectArray JNICALL Java_org_simgrid_msg_Host_getStorageContent(JNIEnv * env, jobject jhost)
 285 {
 286   msg_host_t host = jhost_get_native(env, jhost);
 287
 288   if (not host) {
 289     jxbt_throw_notbound(env, "host", jhost);
 290     return 0;
 291   }
 292   return (jobjectArray)sg_host_get_storage_content(host);
 293 }
 294
 295 JNIEXPORT jobjectArray JNICALL Java_org_simgrid_msg_Host_all(JNIEnv * env, jclass cls_arg)
 296 {
 297   sg_host_t* table = sg_host_list();
 298   int count        = sg_host_count();
 299
 300   jclass cls = jxbt_get_class(env, "org/simgrid/msg/Host");
 301   if (not cls)
 302     return nullptr;
 303
 304   jobjectArray jtable = env->NewObjectArray((jsize)count, cls, nullptr);
 305
 306   if (not jtable) {
 307     jxbt_throw_jni(env, "Hosts table allocation failed");
 308     return nullptr;
 309   }
 310
 311   for (int index = 0; index < count; index++) {
 312     jobject jhost = static_cast<jobject>(table[index]->extension(JAVA_HOST_LEVEL));
 313
 314     if (not jhost) {
 315       jstring jname = env->NewStringUTF(table[index]->get_cname());
 316       jhost         = Java_org_simgrid_msg_Host_getByName(env, cls_arg, jname);
 317     }
 318
 319     env->SetObjectArrayElement(jtable, index, jhost);
 320   }
 321   xbt_free(table);
 322   return jtable;
 323 }
 324
 325 JNIEXPORT void JNICALL Java_org_simgrid_msg_Host_setAsyncMailbox(JNIEnv * env, jclass cls_arg, jobject jname)
 326 {
 327   const char *name = env->GetStringUTFChars((jstring) jname, 0);
 328   sg_mailbox_set_receiver(name);
 329   env->ReleaseStringUTFChars((jstring) jname, name);
 330 }
 331
 332 JNIEXPORT void JNICALL Java_org_simgrid_msg_Host_updateAllEnergyConsumptions(JNIEnv* env, jclass cls)
 333 {
 334   sg_host_energy_update_all();
 335 }
 336
 337 JNIEXPORT jdouble JNICALL Java_org_simgrid_msg_Host_getConsumedEnergy (JNIEnv *env, jobject jhost)
 338 {
 339   const_sg_host_t host = jhost_get_native(env, jhost);
 340
 341   if (not host) {
 342     jxbt_throw_notbound(env, "host", jhost);
 343     return 0;
 344   }
 345
 346   return sg_host_get_consumed_energy(host);
 347 }
 348
 349 JNIEXPORT void JNICALL Java_org_simgrid_msg_Host_setPstate(JNIEnv* env, jobject jhost, jint pstate)
 350 {
 351   sg_host_t host = jhost_get_native(env, jhost);
 352   sg_host_set_pstate(host, pstate);
 353 }
 354 JNIEXPORT jint JNICALL Java_org_simgrid_msg_Host_getPstate(JNIEnv* env, jobject jhost)
 355 {
 356   const_sg_host_t host = jhost_get_native(env, jhost);
 357   return sg_host_get_pstate(host);
 358 }
 359 JNIEXPORT jint JNICALL Java_org_simgrid_msg_Host_getPstatesCount(JNIEnv* env, jobject jhost)
 360 {
 361   const_sg_host_t host = jhost_get_native(env, jhost);
 362   return sg_host_get_nb_pstates(host);
 363 }
 364 JNIEXPORT jdouble JNICALL Java_org_simgrid_msg_Host_getCurrentPowerPeak(JNIEnv* env, jobject jhost)
 365 {
 366   const_sg_host_t host = jhost_get_native(env, jhost);
 367   return sg_host_speed(host);
 368 }
 369 JNIEXPORT jdouble JNICALL Java_org_simgrid_msg_Host_getPowerPeakAt(JNIEnv* env, jobject jhost, jint pstate)
 370 {
 371   const_sg_host_t host = jhost_get_native(env, jhost);
 372   return sg_host_get_pstate_speed(host, pstate);
 373 }
 374
 375 JNIEXPORT jdouble JNICALL Java_org_simgrid_msg_Host_getLoad(JNIEnv* env, jobject jhost)
 376 {
 377   const_sg_host_t host = jhost_get_native(env, jhost);
 378   return sg_host_load(host);
 379 }
 380
 381 JNIEXPORT jdouble JNICALL Java_org_simgrid_msg_Host_getCurrentLoad (JNIEnv *env, jobject jhost)
 382 {
 383   const_sg_host_t host = jhost_get_native(env, jhost);
 384
 385   if (not host) {
 386     jxbt_throw_notbound(env, "host", jhost);
 387     return 0;
 388   }
 389
 390   return sg_host_get_current_load(host);
 391 }
 392
 393 JNIEXPORT jdouble JNICALL Java_org_simgrid_msg_Host_getComputedFlops (JNIEnv *env, jobject jhost)
 394 {
 395   const_sg_host_t host = jhost_get_native(env, jhost);
 396
 397   if (not host) {
 398     jxbt_throw_notbound(env, "host", jhost);
 399     return 0;
 400   }
 401
 402   return sg_host_get_computed_flops(host);
 403 }
 404
 405 JNIEXPORT jdouble JNICALL Java_org_simgrid_msg_Host_getAvgLoad (JNIEnv *env, jobject jhost)
 406 {
 407   const_sg_host_t host = jhost_get_native(env, jhost);
 408
 409   if (not host) {
 410     jxbt_throw_notbound(env, "host", jhost);
 411     return 0;
 412   }
 413
 414   return sg_host_get_avg_load(host);
 415 }