src/java.c

   1 /**
   2  * collectd - src/java.c
   3  * Copyright (C) 2009  Florian octo Forster
   4  * Copyright (C) 2008  Justo Alonso Achaques
   5  *
   6  * This program is free software; you can redistribute it and/or modify it
   7  * under the terms of the GNU General Public License as published by the
   8  * Free Software Foundation; only version 2 of the License is applicable.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with this program; if not, write to the Free Software Foundation, Inc.,
  17  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  18  *
  19  * Authors:
  20  *   Florian octo Forster <octo at verplant.org>
  21  *   Justo Alonso Achaques <justo.alonso at gmail.com>
  22  **/
  23
  24 #include "collectd.h"
  25 #include "plugin.h"
  26 #include "common.h"
  27
  28 #include <pthread.h>
  29 #include <jni.h>
  30
  31 #if !defined(JNI_VERSION_1_2)
  32 # error "Need JNI 1.2 compatible interface!"
  33 #endif
  34
  35 /*
  36  * Types
  37  */
  38 struct cjni_jvm_env_s /* {{{ */
  39 {
  40   JNIEnv *jvm_env;
  41   int reference_counter;
  42 };
  43 typedef struct cjni_jvm_env_s cjni_jvm_env_t;
  44 /* }}} */
  45
  46 struct java_plugin_class_s /* {{{ */
  47 {
  48   char     *name;
  49   jclass    class;
  50   jobject   object;
  51 };
  52 typedef struct java_plugin_class_s java_plugin_class_t;
  53 /* }}} */
  54
  55 struct java_plugin_config_s /* {{{ */
  56 {
  57   char *name;
  58   oconfig_item_t *ci;
  59 };
  60 typedef struct java_plugin_config_s java_plugin_config_t;
  61 /* }}} */
  62
  63 #define CB_TYPE_CONFIG   1
  64 #define CB_TYPE_INIT     2
  65 #define CB_TYPE_READ     3
  66 #define CB_TYPE_WRITE    4
  67 #define CB_TYPE_FLUSH    5
  68 #define CB_TYPE_SHUTDOWN 6
  69 struct cjni_callback_info_s /* {{{ */
  70 {
  71   char     *name;
  72   int       type;
  73   jclass    class;
  74   jobject   object;
  75   jmethodID method;
  76 };
  77 typedef struct cjni_callback_info_s cjni_callback_info_t;
  78 /* }}} */
  79
  80 /*
  81  * Global variables
  82  */
  83 static JavaVM *jvm = NULL;
  84 static pthread_key_t jvm_env_key;
  85
  86 /* Configuration options for the JVM. */
  87 static char **jvm_argv = NULL;
  88 static size_t jvm_argc = 0;
  89
  90 /* List of class names to load */
  91 static java_plugin_class_t  *java_classes_list = NULL;
  92 static size_t                java_classes_list_len;
  93
  94 /* List of `config_item_t's for Java plugins */
  95 static java_plugin_config_t *java_plugin_configs     = NULL;
  96 static size_t                java_plugin_configs_num = 0;
  97
  98 /* List of config, init, and shutdown callbacks. */
  99 static cjni_callback_info_t *java_callbacks      = NULL;
 100 static size_t                java_callbacks_num  = 0;
 101 static pthread_mutex_t       java_callbacks_lock = PTHREAD_MUTEX_INITIALIZER;
 102
 103 /*
 104  * Prototypes
 105  *
 106  * Mostly functions that are needed by the Java interface (``native'')
 107  * functions.
 108  */
 109 static void cjni_callback_info_destroy (void *arg);
 110 static cjni_callback_info_t *cjni_callback_info_create (JNIEnv *jvm_env,
 111     jobject o_name, jobject o_callback, int type);
 112 static int cjni_callback_register (JNIEnv *jvm_env, jobject o_name,
 113     jobject o_callback, int type);
 114 static int cjni_read (user_data_t *user_data);
 115 static int cjni_write (const data_set_t *ds, const value_list_t *vl,
 116     user_data_t *ud);
 117 static int cjni_flush (int timeout, const char *identifier, user_data_t *ud);
 118
 119 /*
 120  * C to Java conversion functions
 121  */
 122 static int ctoj_string (JNIEnv *jvm_env, /* {{{ */
 123     const char *string,
 124     jclass class_ptr, jobject object_ptr, const char *method_name)
 125 {
 126   jmethodID m_set;
 127   jstring o_string;
 128
 129   /* Create a java.lang.String */
 130   o_string = (*jvm_env)->NewStringUTF (jvm_env,
 131       (string != NULL) ? string : "");
 132   if (o_string == NULL)
 133   {
 134     ERROR ("java plugin: ctoj_string: NewStringUTF failed.");
 135     return (-1);
 136   }
 137
 138   /* Search for the `void setFoo (String s)' method. */
 139   m_set = (*jvm_env)->GetMethodID (jvm_env, class_ptr,
 140       method_name, "(Ljava/lang/String;)V");
 141   if (m_set == NULL)
 142   {
 143     ERROR ("java plugin: ctoj_string: Cannot find method `void %s (String)'.",
 144         method_name);
 145     (*jvm_env)->DeleteLocalRef (jvm_env, o_string);
 146     return (-1);
 147   }
 148
 149   /* Call the method. */
 150   (*jvm_env)->CallVoidMethod (jvm_env, object_ptr, m_set, o_string);
 151
 152   /* Decrease reference counter on the java.lang.String object. */
 153   (*jvm_env)->DeleteLocalRef (jvm_env, o_string);
 154
 155   return (0);
 156 } /* }}} int ctoj_string */
 157
 158 static int ctoj_int (JNIEnv *jvm_env, /* {{{ */
 159     jint value,
 160     jclass class_ptr, jobject object_ptr, const char *method_name)
 161 {
 162   jmethodID m_set;
 163
 164   /* Search for the `void setFoo (int i)' method. */
 165   m_set = (*jvm_env)->GetMethodID (jvm_env, class_ptr,
 166       method_name, "(I)V");
 167   if (m_set == NULL)
 168   {
 169     ERROR ("java plugin: ctoj_int: Cannot find method `void %s (int)'.",
 170         method_name);
 171     return (-1);
 172   }
 173
 174   (*jvm_env)->CallVoidMethod (jvm_env, object_ptr, m_set, value);
 175
 176   return (0);
 177 } /* }}} int ctoj_int */
 178
 179 static int ctoj_long (JNIEnv *jvm_env, /* {{{ */
 180     jlong value,
 181     jclass class_ptr, jobject object_ptr, const char *method_name)
 182 {
 183   jmethodID m_set;
 184
 185   /* Search for the `void setFoo (long l)' method. */
 186   m_set = (*jvm_env)->GetMethodID (jvm_env, class_ptr,
 187       method_name, "(J)V");
 188   if (m_set == NULL)
 189   {
 190     ERROR ("java plugin: ctoj_long: Cannot find method `void %s (long)'.",
 191         method_name);
 192     return (-1);
 193   }
 194
 195   (*jvm_env)->CallVoidMethod (jvm_env, object_ptr, m_set, value);
 196
 197   return (0);
 198 } /* }}} int ctoj_long */
 199
 200 static int ctoj_double (JNIEnv *jvm_env, /* {{{ */
 201     jdouble value,
 202     jclass class_ptr, jobject object_ptr, const char *method_name)
 203 {
 204   jmethodID m_set;
 205
 206   /* Search for the `void setFoo (double d)' method. */
 207   m_set = (*jvm_env)->GetMethodID (jvm_env, class_ptr,
 208       method_name, "(D)V");
 209   if (m_set == NULL)
 210   {
 211     ERROR ("java plugin: ctoj_double: Cannot find method `void %s (double)'.",
 212         method_name);
 213     return (-1);
 214   }
 215
 216   (*jvm_env)->CallVoidMethod (jvm_env, object_ptr, m_set, value);
 217
 218   return (0);
 219 } /* }}} int ctoj_double */
 220
 221 /* Convert a jlong to a java.lang.Number */
 222 static jobject ctoj_jlong_to_number (JNIEnv *jvm_env, jlong value) /* {{{ */
 223 {
 224   jclass c_long;
 225   jmethodID m_long_constructor;
 226
 227   /* Look up the java.lang.Long class */
 228   c_long = (*jvm_env)->FindClass (jvm_env, "java.lang.Long");
 229   if (c_long == NULL)
 230   {
 231     ERROR ("java plugin: ctoj_jlong_to_number: Looking up the "
 232         "java.lang.Long class failed.");
 233     return (NULL);
 234   }
 235
 236   m_long_constructor = (*jvm_env)->GetMethodID (jvm_env,
 237       c_long, "<init>", "(J)V");
 238   if (m_long_constructor == NULL)
 239   {
 240     ERROR ("java plugin: ctoj_jlong_to_number: Looking up the "
 241         "`Long (long)' constructor failed.");
 242     return (NULL);
 243   }
 244
 245   return ((*jvm_env)->NewObject (jvm_env,
 246         c_long, m_long_constructor, value));
 247 } /* }}} jobject ctoj_jlong_to_number */
 248
 249 /* Convert a jdouble to a java.lang.Number */
 250 static jobject ctoj_jdouble_to_number (JNIEnv *jvm_env, jdouble value) /* {{{ */
 251 {
 252   jclass c_double;
 253   jmethodID m_double_constructor;
 254
 255   /* Look up the java.lang.Long class */
 256   c_double = (*jvm_env)->FindClass (jvm_env, "java.lang.Double");
 257   if (c_double == NULL)
 258   {
 259     ERROR ("java plugin: ctoj_jdouble_to_number: Looking up the "
 260         "java.lang.Double class failed.");
 261     return (NULL);
 262   }
 263
 264   m_double_constructor = (*jvm_env)->GetMethodID (jvm_env,
 265       c_double, "<init>", "(D)V");
 266   if (m_double_constructor == NULL)
 267   {
 268     ERROR ("java plugin: ctoj_jdouble_to_number: Looking up the "
 269         "`Double (double)' constructor failed.");
 270     return (NULL);
 271   }
 272
 273   return ((*jvm_env)->NewObject (jvm_env,
 274         c_double, m_double_constructor, value));
 275 } /* }}} jobject ctoj_jdouble_to_number */
 276
 277 /* Convert a value_t to a java.lang.Number */
 278 static jobject ctoj_value_to_number (JNIEnv *jvm_env, /* {{{ */
 279     value_t value, int ds_type)
 280 {
 281   if (ds_type == DS_TYPE_COUNTER)
 282     return (ctoj_jlong_to_number (jvm_env, (jlong) value.counter));
 283   else if (ds_type == DS_TYPE_GAUGE)
 284     return (ctoj_jdouble_to_number (jvm_env, (jdouble) value.gauge));
 285   else
 286     return (NULL);
 287 } /* }}} jobject ctoj_value_to_number */
 288
 289 /* Convert a data_source_t to a org.collectd.api.DataSource */
 290 static jobject ctoj_data_source (JNIEnv *jvm_env, /* {{{ */
 291     const data_source_t *dsrc)
 292 {
 293   jclass c_datasource;
 294   jmethodID m_datasource_constructor;
 295   jobject o_datasource;
 296   int status;
 297
 298   /* Look up the DataSource class */
 299   c_datasource = (*jvm_env)->FindClass (jvm_env,
 300       "org.collectd.api.DataSource");
 301   if (c_datasource == NULL)
 302   {
 303     ERROR ("java plugin: ctoj_data_source: "
 304         "FindClass (org.collectd.api.DataSource) failed.");
 305     return (NULL);
 306   }
 307
 308   /* Lookup the `ValueList ()' constructor. */
 309   m_datasource_constructor = (*jvm_env)->GetMethodID (jvm_env, c_datasource,
 310       "<init>", "()V");
 311   if (m_datasource_constructor == NULL)
 312   {
 313     ERROR ("java plugin: ctoj_data_source: Cannot find the "
 314         "`DataSource ()' constructor.");
 315     return (NULL);
 316   }
 317
 318   /* Create a new instance. */
 319   o_datasource = (*jvm_env)->NewObject (jvm_env, c_datasource,
 320       m_datasource_constructor);
 321   if (o_datasource == NULL)
 322   {
 323     ERROR ("java plugin: ctoj_data_source: "
 324         "Creating a new DataSource instance failed.");
 325     return (NULL);
 326   }
 327
 328   /* Set name via `void setName (String name)' */
 329   status = ctoj_string (jvm_env, dsrc->name,
 330       c_datasource, o_datasource, "setName");
 331   if (status != 0)
 332   {
 333     ERROR ("java plugin: ctoj_data_source: "
 334         "ctoj_string (setName) failed.");
 335     (*jvm_env)->DeleteLocalRef (jvm_env, o_datasource);
 336     return (NULL);
 337   }
 338
 339   /* Set type via `void setType (int type)' */
 340   status = ctoj_int (jvm_env, dsrc->type,
 341       c_datasource, o_datasource, "setType");
 342   if (status != 0)
 343   {
 344     ERROR ("java plugin: ctoj_data_source: "
 345         "ctoj_int (setType) failed.");
 346     (*jvm_env)->DeleteLocalRef (jvm_env, o_datasource);
 347     return (NULL);
 348   }
 349
 350   /* Set min via `void setMin (double min)' */
 351   status = ctoj_double (jvm_env, dsrc->min,
 352       c_datasource, o_datasource, "setMin");
 353   if (status != 0)
 354   {
 355     ERROR ("java plugin: ctoj_data_source: "
 356         "ctoj_double (setMin) failed.");
 357     (*jvm_env)->DeleteLocalRef (jvm_env, o_datasource);
 358     return (NULL);
 359   }
 360
 361   /* Set max via `void setMax (double max)' */
 362   status = ctoj_double (jvm_env, dsrc->max,
 363       c_datasource, o_datasource, "setMax");
 364   if (status != 0)
 365   {
 366     ERROR ("java plugin: ctoj_data_source: "
 367         "ctoj_double (setMax) failed.");
 368     (*jvm_env)->DeleteLocalRef (jvm_env, o_datasource);
 369     return (NULL);
 370   }
 371
 372   return (o_datasource);
 373 } /* }}} jobject ctoj_data_source */
 374
 375 /* Convert a oconfig_value_t to a org.collectd.api.OConfigValue */
 376 static jobject ctoj_oconfig_value (JNIEnv *jvm_env, /* {{{ */
 377     oconfig_value_t ocvalue)
 378 {
 379   jclass c_ocvalue;
 380   jmethodID m_ocvalue_constructor;
 381   jobject o_argument;
 382   jobject o_ocvalue;
 383
 384   m_ocvalue_constructor = NULL;
 385   o_argument = NULL;
 386
 387   c_ocvalue = (*jvm_env)->FindClass (jvm_env,
 388       "org.collectd.api.OConfigValue");
 389   if (c_ocvalue == NULL)
 390   {
 391     ERROR ("java plugin: ctoj_oconfig_value: "
 392         "FindClass (org.collectd.api.OConfigValue) failed.");
 393     return (NULL);
 394   }
 395
 396   if (ocvalue.type == OCONFIG_TYPE_BOOLEAN)
 397   {
 398     jboolean tmp_boolean;
 399
 400     tmp_boolean = (ocvalue.value.boolean == 0) ? JNI_FALSE : JNI_TRUE;
 401
 402     m_ocvalue_constructor = (*jvm_env)->GetMethodID (jvm_env, c_ocvalue,
 403         "<init>", "(Z)V");
 404     if (m_ocvalue_constructor == NULL)
 405     {
 406       ERROR ("java plugin: ctoj_oconfig_value: Cannot find the "
 407           "`OConfigValue (boolean)' constructor.");
 408       return (NULL);
 409     }
 410
 411     return ((*jvm_env)->NewObject (jvm_env,
 412           c_ocvalue, m_ocvalue_constructor, tmp_boolean));
 413   } /* if (ocvalue.type == OCONFIG_TYPE_BOOLEAN) */
 414   else if (ocvalue.type == OCONFIG_TYPE_STRING)
 415   {
 416     m_ocvalue_constructor = (*jvm_env)->GetMethodID (jvm_env, c_ocvalue,
 417         "<init>", "(Ljava/lang/String;)V");
 418     if (m_ocvalue_constructor == NULL)
 419     {
 420       ERROR ("java plugin: ctoj_oconfig_value: Cannot find the "
 421           "`OConfigValue (String)' constructor.");
 422       return (NULL);
 423     }
 424
 425     o_argument = (*jvm_env)->NewStringUTF (jvm_env, ocvalue.value.string);
 426     if (o_argument == NULL)
 427     {
 428       ERROR ("java plugin: ctoj_oconfig_value: "
 429           "Creating a String object failed.");
 430       return (NULL);
 431     }
 432   }
 433   else if (ocvalue.type == OCONFIG_TYPE_NUMBER)
 434   {
 435     m_ocvalue_constructor = (*jvm_env)->GetMethodID (jvm_env, c_ocvalue,
 436         "<init>", "(Ljava/lang/Number;)V");
 437     if (m_ocvalue_constructor == NULL)
 438     {
 439       ERROR ("java plugin: ctoj_oconfig_value: Cannot find the "
 440           "`OConfigValue (Number)' constructor.");
 441       return (NULL);
 442     }
 443
 444     o_argument = ctoj_jdouble_to_number (jvm_env,
 445         (jdouble) ocvalue.value.number);
 446     if (o_argument == NULL)
 447     {
 448       ERROR ("java plugin: ctoj_oconfig_value: "
 449           "Creating a Number object failed.");
 450       return (NULL);
 451     }
 452   }
 453   else
 454   {
 455     return (NULL);
 456   }
 457
 458   assert (m_ocvalue_constructor != NULL);
 459   assert (o_argument != NULL);
 460
 461   o_ocvalue = (*jvm_env)->NewObject (jvm_env,
 462       c_ocvalue, m_ocvalue_constructor, o_argument);
 463   if (o_ocvalue == NULL)
 464   {
 465     ERROR ("java plugin: ctoj_oconfig_value: "
 466         "Creating an OConfigValue object failed.");
 467     (*jvm_env)->DeleteLocalRef (jvm_env, o_argument);
 468     return (NULL);
 469   }
 470
 471   (*jvm_env)->DeleteLocalRef (jvm_env, o_argument);
 472   return (o_ocvalue);
 473 } /* }}} jobject ctoj_oconfig_value */
 474
 475 /* Convert a oconfig_item_t to a org.collectd.api.OConfigItem */
 476 static jobject ctoj_oconfig_item (JNIEnv *jvm_env, /* {{{ */
 477     const oconfig_item_t *ci)
 478 {
 479   jclass c_ocitem;
 480   jmethodID m_ocitem_constructor;
 481   jmethodID m_addvalue;
 482   jmethodID m_addchild;
 483   jobject o_key;
 484   jobject o_ocitem;
 485   int i;
 486
 487   c_ocitem = (*jvm_env)->FindClass (jvm_env, "org.collectd.api.OConfigItem");
 488   if (c_ocitem == NULL)
 489   {
 490     ERROR ("java plugin: ctoj_oconfig_item: "
 491         "FindClass (org.collectd.api.OConfigItem) failed.");
 492     return (NULL);
 493   }
 494
 495   /* Get the required methods: m_ocitem_constructor, m_addvalue, and m_addchild
 496    * {{{ */
 497   m_ocitem_constructor = (*jvm_env)->GetMethodID (jvm_env, c_ocitem,
 498       "<init>", "(Ljava/lang/String;)V");
 499   if (m_ocitem_constructor == NULL)
 500   {
 501     ERROR ("java plugin: ctoj_oconfig_item: Cannot find the "
 502         "`OConfigItem (String)' constructor.");
 503     return (NULL);
 504   }
 505
 506   m_addvalue = (*jvm_env)->GetMethodID (jvm_env, c_ocitem,
 507       "addValue", "(Lorg/collectd/api/OConfigValue;)V");
 508   if (m_addvalue == NULL)
 509   {
 510     ERROR ("java plugin: ctoj_oconfig_item: Cannot find the "
 511         "`addValue (OConfigValue)' method.");
 512     return (NULL);
 513   }
 514
 515   m_addchild = (*jvm_env)->GetMethodID (jvm_env, c_ocitem,
 516       "addChild", "(Lorg/collectd/api/OConfigItem;)V");
 517   if (m_addchild == NULL)
 518   {
 519     ERROR ("java plugin: ctoj_oconfig_item: Cannot find the "
 520         "`addChild (OConfigItem)' method.");
 521     return (NULL);
 522   }
 523   /* }}} */
 524
 525   /* Create a String object with the key.
 526    * Needed for calling the constructor. */
 527   o_key = (*jvm_env)->NewStringUTF (jvm_env, ci->key);
 528   if (o_key == NULL)
 529   {
 530     ERROR ("java plugin: ctoj_oconfig_item: "
 531         "Creating String object failed.");
 532     return (NULL);
 533   }
 534
 535   /* Create an OConfigItem object */
 536   o_ocitem = (*jvm_env)->NewObject (jvm_env,
 537       c_ocitem, m_ocitem_constructor, o_key);
 538   if (o_ocitem == NULL)
 539   {
 540     ERROR ("java plugin: ctoj_oconfig_item: "
 541         "Creating an OConfigItem object failed.");
 542     (*jvm_env)->DeleteLocalRef (jvm_env, o_key);
 543     return (NULL);
 544   }
 545
 546   /* We don't need the String object any longer.. */
 547   (*jvm_env)->DeleteLocalRef (jvm_env, o_key);
 548
 549   /* Call OConfigItem.addValue for each value */
 550   for (i = 0; i < ci->values_num; i++) /* {{{ */
 551   {
 552     jobject o_value;
 553
 554     o_value = ctoj_oconfig_value (jvm_env, ci->values[i]);
 555     if (o_value == NULL)
 556     {
 557       ERROR ("java plugin: ctoj_oconfig_item: "
 558           "Creating an OConfigValue object failed.");
 559       (*jvm_env)->DeleteLocalRef (jvm_env, o_ocitem);
 560       return (NULL);
 561     }
 562
 563     (*jvm_env)->CallVoidMethod (jvm_env, o_ocitem, m_addvalue, o_value);
 564     (*jvm_env)->DeleteLocalRef (jvm_env, o_value);
 565   } /* }}} for (i = 0; i < ci->values_num; i++) */
 566
 567   /* Call OConfigItem.addChild for each child */
 568   for (i = 0; i < ci->children_num; i++) /* {{{ */
 569   {
 570     jobject o_child;
 571
 572     o_child = ctoj_oconfig_item (jvm_env, ci->children + i);
 573     if (o_child == NULL)
 574     {
 575       ERROR ("java plugin: ctoj_oconfig_item: "
 576           "Creating an OConfigItem object failed.");
 577       (*jvm_env)->DeleteLocalRef (jvm_env, o_ocitem);
 578       return (NULL);
 579     }
 580
 581     (*jvm_env)->CallVoidMethod (jvm_env, o_ocitem, m_addvalue, o_child);
 582     (*jvm_env)->DeleteLocalRef (jvm_env, o_child);
 583   } /* }}} for (i = 0; i < ci->children_num; i++) */
 584
 585   return (o_ocitem);
 586 } /* }}} jobject ctoj_oconfig_item */
 587
 588 /* Convert a data_set_t to a org.collectd.api.DataSet */
 589 static jobject ctoj_data_set (JNIEnv *jvm_env, const data_set_t *ds) /* {{{ */
 590 {
 591   jclass c_dataset;
 592   jmethodID m_constructor;
 593   jmethodID m_add;
 594   jobject o_type;
 595   jobject o_dataset;
 596   int i;
 597
 598   /* Look up the org.collectd.api.DataSet class */
 599   c_dataset = (*jvm_env)->FindClass (jvm_env, "org.collectd.api.DataSet");
 600   if (c_dataset == NULL)
 601   {
 602     ERROR ("java plugin: ctoj_data_set: Looking up the "
 603         "org.collectd.api.DataSet class failed.");
 604     return (NULL);
 605   }
 606
 607   /* Search for the `DataSet (String type)' constructor. */
 608   m_constructor = (*jvm_env)->GetMethodID (jvm_env,
 609       c_dataset, "<init>", "(Ljava.lang.String;)V");
 610   if (m_constructor == NULL)
 611   {
 612     ERROR ("java plugin: ctoj_data_set: Looking up the "
 613         "`DataSet (String)' constructor failed.");
 614     return (NULL);
 615   }
 616
 617   /* Search for the `void addDataSource (DataSource)' method. */
 618   m_add = (*jvm_env)->GetMethodID (jvm_env,
 619       c_dataset, "addDataSource", "(Lorg.collectd.api.DataSource;)V");
 620   if (m_add == NULL)
 621   {
 622     ERROR ("java plugin: ctoj_data_set: Looking up the "
 623         "`addDataSource (DataSource)' method failed.");
 624     return (NULL);
 625   }
 626
 627   o_type = (*jvm_env)->NewStringUTF (jvm_env, ds->type);
 628   if (o_type == NULL)
 629   {
 630     ERROR ("java plugin: ctoj_data_set: Creating a String object failed.");
 631     return (NULL);
 632   }
 633
 634   o_dataset = (*jvm_env)->NewObject (jvm_env,
 635       c_dataset, m_constructor, o_type);
 636   if (o_dataset == NULL)
 637   {
 638     ERROR ("java plugin: ctoj_data_set: Creating a DataSet object failed.");
 639     (*jvm_env)->DeleteLocalRef (jvm_env, o_type);
 640     return (NULL);
 641   }
 642
 643   /* Decrease reference counter on the java.lang.String object. */
 644   (*jvm_env)->DeleteLocalRef (jvm_env, o_type);
 645
 646   for (i = 0; i < ds->ds_num; i++)
 647   {
 648     jobject o_datasource;
 649
 650     o_datasource = ctoj_data_source (jvm_env, ds->ds + i);
 651     if (o_datasource == NULL)
 652     {
 653       ERROR ("java plugin: ctoj_data_set: ctoj_data_source (%s.%s) failed",
 654           ds->type, ds->ds[i].name);
 655       (*jvm_env)->DeleteLocalRef (jvm_env, o_dataset);
 656       return (NULL);
 657     }
 658
 659     (*jvm_env)->CallVoidMethod (jvm_env, o_dataset, m_add, o_datasource);
 660
 661     (*jvm_env)->DeleteLocalRef (jvm_env, o_datasource);
 662   } /* for (i = 0; i < ds->ds_num; i++) */
 663
 664   return (o_dataset);
 665 } /* }}} jobject ctoj_data_set */
 666
 667 static int ctoj_value_list_add_value (JNIEnv *jvm_env, /* {{{ */
 668     value_t value, int ds_type,
 669     jclass class_ptr, jobject object_ptr)
 670 {
 671   jmethodID m_addvalue;
 672   jobject o_number;
 673
 674   m_addvalue = (*jvm_env)->GetMethodID (jvm_env, class_ptr,
 675       "addValue", "(Ljava/lang/Number;)V");
 676   if (m_addvalue == NULL)
 677   {
 678     ERROR ("java plugin: ctoj_value_list_add_value: "
 679         "Cannot find method `void addValue (Number)'.");
 680     return (-1);
 681   }
 682
 683   o_number = ctoj_value_to_number (jvm_env, value, ds_type);
 684   if (o_number == NULL)
 685   {
 686     ERROR ("java plugin: ctoj_value_list_add_value: "
 687         "ctoj_value_to_number failed.");
 688     return (-1);
 689   }
 690
 691   (*jvm_env)->CallVoidMethod (jvm_env, object_ptr, m_addvalue, o_number);
 692
 693   (*jvm_env)->DeleteLocalRef (jvm_env, o_number);
 694
 695   return (0);
 696 } /* }}} int ctoj_value_list_add_value */
 697
 698 static int ctoj_value_list_add_data_set (JNIEnv *jvm_env, /* {{{ */
 699     jclass c_valuelist, jobject o_valuelist, const data_set_t *ds)
 700 {
 701   jmethodID m_setdataset;
 702   jobject o_dataset;
 703
 704   /* Look for the `void setDataSource (List<DataSource> ds)' method. */
 705   m_setdataset = (*jvm_env)->GetMethodID (jvm_env, c_valuelist,
 706       "setDataSet", "(Lorg.collectd.api.DataSet;)V");
 707   if (m_setdataset == NULL)
 708   {
 709     ERROR ("java plugin: ctoj_value_list_add_data_set: "
 710         "Cannot find the `void setDataSet (DataSet)' method.");
 711     return (-1);
 712   }
 713
 714   /* Create a DataSet object. */
 715   o_dataset = ctoj_data_set (jvm_env, ds);
 716   if (o_dataset == NULL)
 717   {
 718     ERROR ("java plugin: ctoj_value_list_add_data_set: "
 719         "ctoj_data_set (%s) failed.", ds->type);
 720     return (-1);
 721   }
 722
 723   /* Actually call the method. */
 724   (*jvm_env)->CallVoidMethod (jvm_env,
 725       o_valuelist, m_setdataset, o_dataset);
 726
 727   /* Decrease reference counter on the List<DataSource> object. */
 728   (*jvm_env)->DeleteLocalRef (jvm_env, o_dataset);
 729
 730   return (0);
 731 } /* }}} int ctoj_value_list_add_data_set */
 732
 733 /* Convert a value_list_t (and data_set_t) to a org.collectd.api.ValueList */
 734 static jobject ctoj_value_list (JNIEnv *jvm_env, /* {{{ */
 735     const data_set_t *ds, const value_list_t *vl)
 736 {
 737   jclass c_valuelist;
 738   jmethodID m_valuelist_constructor;
 739   jobject o_valuelist;
 740   int status;
 741   int i;
 742
 743   /* First, create a new ValueList instance..
 744    * Look up the class.. */
 745   c_valuelist = (*jvm_env)->FindClass (jvm_env,
 746       "org.collectd.api.ValueList");
 747   if (c_valuelist == NULL)
 748   {
 749     ERROR ("java plugin: ctoj_value_list: "
 750         "FindClass (org.collectd.api.ValueList) failed.");
 751     return (NULL);
 752   }
 753
 754   /* Lookup the `ValueList ()' constructor. */
 755   m_valuelist_constructor = (*jvm_env)->GetMethodID (jvm_env, c_valuelist,
 756       "<init>", "()V");
 757   if (m_valuelist_constructor == NULL)
 758   {
 759     ERROR ("java plugin: ctoj_value_list: Cannot find the "
 760         "`ValueList ()' constructor.");
 761     return (NULL);
 762   }
 763
 764   /* Create a new instance. */
 765   o_valuelist = (*jvm_env)->NewObject (jvm_env, c_valuelist,
 766       m_valuelist_constructor);
 767   if (o_valuelist == NULL)
 768   {
 769     ERROR ("java plugin: ctoj_value_list: Creating a new ValueList instance "
 770         "failed.");
 771     return (NULL);
 772   }
 773
 774   status = ctoj_value_list_add_data_set (jvm_env,
 775       c_valuelist, o_valuelist, ds);
 776   if (status != 0)
 777   {
 778     ERROR ("java plugin: ctoj_value_list: "
 779         "ctoj_value_list_add_data_set failed.");
 780     (*jvm_env)->DeleteLocalRef (jvm_env, o_valuelist);
 781     return (NULL);
 782   }
 783
 784   /* Set the strings.. */
 785 #define SET_STRING(str,method_name) do { \
 786   status = ctoj_string (jvm_env, str, \
 787       c_valuelist, o_valuelist, method_name); \
 788   if (status != 0) { \
 789     ERROR ("java plugin: ctoj_value_list: jtoc_string (%s) failed.", \
 790         method_name); \
 791     (*jvm_env)->DeleteLocalRef (jvm_env, o_valuelist); \
 792     return (NULL); \
 793   } } while (0)
 794
 795   SET_STRING (vl->host,            "setHost");
 796   SET_STRING (vl->plugin,          "setPlugin");
 797   SET_STRING (vl->plugin_instance, "setPluginInstance");
 798   SET_STRING (vl->type,            "setType");
 799   SET_STRING (vl->type_instance,   "setTypeInstance");
 800
 801 #undef SET_STRING
 802
 803   /* Set the `time' member. Java stores time in milliseconds. */
 804   status = ctoj_long (jvm_env, ((jlong) vl->time) * ((jlong) 1000),
 805       c_valuelist, o_valuelist, "setTime");
 806   if (status != 0)
 807   {
 808     ERROR ("java plugin: ctoj_value_list: ctoj_long (setTime) failed.");
 809     (*jvm_env)->DeleteLocalRef (jvm_env, o_valuelist);
 810     return (NULL);
 811   }
 812
 813   /* Set the `interval' member.. */
 814   status = ctoj_long (jvm_env, (jlong) vl->interval,
 815       c_valuelist, o_valuelist, "setInterval");
 816   if (status != 0)
 817   {
 818     ERROR ("java plugin: ctoj_value_list: ctoj_long (setInterval) failed.");
 819     (*jvm_env)->DeleteLocalRef (jvm_env, o_valuelist);
 820     return (NULL);
 821   }
 822
 823   for (i = 0; i < vl->values_len; i++)
 824   {
 825     status = ctoj_value_list_add_value (jvm_env, vl->values[i], ds->ds[i].type,
 826         c_valuelist, o_valuelist);
 827     if (status != 0)
 828     {
 829       ERROR ("java plugin: ctoj_value_list: "
 830           "ctoj_value_list_add_value failed.");
 831       (*jvm_env)->DeleteLocalRef (jvm_env, o_valuelist);
 832       return (NULL);
 833     }
 834   }
 835
 836   return (o_valuelist);
 837 } /* }}} int ctoj_value_list */
 838
 839 /*
 840  * Java to C conversion functions
 841  */
 842 /* Call a `String <method> ()' method. */
 843 static int jtoc_string (JNIEnv *jvm_env, /* {{{ */
 844     char *buffer, size_t buffer_size,
 845     jclass class_ptr, jobject object_ptr, const char *method_name)
 846 {
 847   jmethodID method_id;
 848   jobject string_obj;
 849   const char *c_str;
 850
 851   method_id = (*jvm_env)->GetMethodID (jvm_env, class_ptr,
 852       method_name, "()Ljava/lang/String;");
 853   if (method_id == NULL)
 854   {
 855     ERROR ("java plugin: jtoc_string: Cannot find method `String %s ()'.",
 856         method_name);
 857     return (-1);
 858   }
 859
 860   string_obj = (*jvm_env)->CallObjectMethod (jvm_env, object_ptr, method_id);
 861   if (string_obj == NULL)
 862   {
 863     ERROR ("java plugin: jtoc_string: CallObjectMethod (%s) failed.",
 864         method_name);
 865     return (-1);
 866   }
 867
 868   c_str = (*jvm_env)->GetStringUTFChars (jvm_env, string_obj, 0);
 869   if (c_str == NULL)
 870   {
 871     ERROR ("java plugin: jtoc_string: GetStringUTFChars failed.");
 872     (*jvm_env)->DeleteLocalRef (jvm_env, string_obj);
 873     return (-1);
 874   }
 875
 876   sstrncpy (buffer, c_str, buffer_size);
 877
 878   (*jvm_env)->ReleaseStringUTFChars (jvm_env, string_obj, c_str);
 879   (*jvm_env)->DeleteLocalRef (jvm_env, string_obj);
 880
 881   return (0);
 882 } /* }}} int jtoc_string */
 883
 884 /* Call a `long <method> ()' method. */
 885 static int jtoc_long (JNIEnv *jvm_env, /* {{{ */
 886     jlong *ret_value,
 887     jclass class_ptr, jobject object_ptr, const char *method_name)
 888 {
 889   jmethodID method_id;
 890
 891   method_id = (*jvm_env)->GetMethodID (jvm_env, class_ptr,
 892       method_name, "()J");
 893   if (method_id == NULL)
 894   {
 895     ERROR ("java plugin: jtoc_long: Cannot find method `long %s ()'.",
 896         method_name);
 897     return (-1);
 898   }
 899
 900   *ret_value = (*jvm_env)->CallLongMethod (jvm_env, object_ptr, method_id);
 901
 902   return (0);
 903 } /* }}} int jtoc_long */
 904
 905 /* Call a `double <method> ()' method. */
 906 static int jtoc_double (JNIEnv *jvm_env, /* {{{ */
 907     jdouble *ret_value,
 908     jclass class_ptr, jobject object_ptr, const char *method_name)
 909 {
 910   jmethodID method_id;
 911
 912   method_id = (*jvm_env)->GetMethodID (jvm_env, class_ptr,
 913       method_name, "()D");
 914   if (method_id == NULL)
 915   {
 916     ERROR ("java plugin: jtoc_string: Cannot find method `double %s ()'.",
 917         method_name);
 918     return (-1);
 919   }
 920
 921   *ret_value = (*jvm_env)->CallDoubleMethod (jvm_env, object_ptr, method_id);
 922
 923   return (0);
 924 } /* }}} int jtoc_double */
 925
 926 static int jtoc_value (JNIEnv *jvm_env, /* {{{ */
 927     value_t *ret_value, int ds_type, jobject object_ptr)
 928 {
 929   jclass class_ptr;
 930   int status;
 931
 932   class_ptr = (*jvm_env)->GetObjectClass (jvm_env, object_ptr);
 933
 934   if (ds_type == DS_TYPE_COUNTER)
 935   {
 936     jlong tmp_long;
 937
 938     status = jtoc_long (jvm_env, &tmp_long,
 939         class_ptr, object_ptr, "longValue");
 940     if (status != 0)
 941     {
 942       ERROR ("java plugin: jtoc_value: "
 943           "jtoc_long failed.");
 944       return (-1);
 945     }
 946     (*ret_value).counter = (counter_t) tmp_long;
 947   }
 948   else
 949   {
 950     jdouble tmp_double;
 951
 952     status = jtoc_double (jvm_env, &tmp_double,
 953         class_ptr, object_ptr, "doubleValue");
 954     if (status != 0)
 955     {
 956       ERROR ("java plugin: jtoc_value: "
 957           "jtoc_double failed.");
 958       return (-1);
 959     }
 960     (*ret_value).gauge = (gauge_t) tmp_double;
 961   }
 962
 963   return (0);
 964 } /* }}} int jtoc_value */
 965
 966 /* Read a List<Number>, convert it to `value_t' and add it to the given
 967  * `value_list_t'. */
 968 static int jtoc_values_array (JNIEnv *jvm_env, /* {{{ */
 969     const data_set_t *ds, value_list_t *vl,
 970     jclass class_ptr, jobject object_ptr)
 971 {
 972   jmethodID m_getvalues;
 973   jmethodID m_toarray;
 974   jobject o_list;
 975   jobjectArray o_number_array;
 976
 977   value_t *values;
 978   int values_num;
 979   int i;
 980
 981   values_num = ds->ds_num;
 982
 983   values = NULL;
 984   o_number_array = NULL;
 985   o_list = NULL;
 986
 987 #define BAIL_OUT(status) \
 988   free (values); \
 989   if (o_number_array != NULL) \
 990     (*jvm_env)->DeleteLocalRef (jvm_env, o_number_array); \
 991   if (o_list != NULL) \
 992     (*jvm_env)->DeleteLocalRef (jvm_env, o_list); \
 993   return (status);
 994
 995   /* Call: List<Number> ValueList.getValues () */
 996   m_getvalues = (*jvm_env)->GetMethodID (jvm_env, class_ptr,
 997       "getValues", "()Ljava/util/List;");
 998   if (m_getvalues == NULL)
 999   {
1000     ERROR ("java plugin: jtoc_values_array: "
1001         "Cannot find method `List getValues ()'.");
1002     BAIL_OUT (-1);
1003   }
1004
1005   o_list = (*jvm_env)->CallObjectMethod (jvm_env, object_ptr, m_getvalues);
1006   if (o_list == NULL)
1007   {
1008     ERROR ("java plugin: jtoc_values_array: "
1009         "CallObjectMethod (getValues) failed.");
1010     BAIL_OUT (-1);
1011   }
1012
1013   /* Call: Number[] List.toArray () */
1014   m_toarray = (*jvm_env)->GetMethodID (jvm_env,
1015       (*jvm_env)->GetObjectClass (jvm_env, o_list),
1016       "toArray", "()[Ljava/lang/Object;");
1017   if (m_toarray == NULL)
1018   {
1019     ERROR ("java plugin: jtoc_values_array: "
1020         "Cannot find method `Object[] toArray ()'.");
1021     BAIL_OUT (-1);
1022   }
1023
1024   o_number_array = (*jvm_env)->CallObjectMethod (jvm_env, o_list, m_toarray);
1025   if (o_number_array == NULL)
1026   {
1027     ERROR ("java plugin: jtoc_values_array: "
1028         "CallObjectMethod (toArray) failed.");
1029     BAIL_OUT (-1);
1030   }
1031
1032   values = (value_t *) calloc (values_num, sizeof (value_t));
1033   if (values == NULL)
1034   {
1035     ERROR ("java plugin: jtoc_values_array: calloc failed.");
1036     BAIL_OUT (-1);
1037   }
1038
1039   for (i = 0; i < values_num; i++)
1040   {
1041     jobject o_number;
1042     int status;
1043
1044     o_number = (*jvm_env)->GetObjectArrayElement (jvm_env,
1045         o_number_array, (jsize) i);
1046     if (o_number == NULL)
1047     {
1048       ERROR ("java plugin: jtoc_values_array: "
1049           "GetObjectArrayElement (%i) failed.", i);
1050       BAIL_OUT (-1);
1051     }
1052
1053     status = jtoc_value (jvm_env, values + i, ds->ds[i].type, o_number);
1054     if (status != 0)
1055     {
1056       ERROR ("java plugin: jtoc_values_array: "
1057           "jtoc_value (%i) failed.", i);
1058       BAIL_OUT (-1);
1059     }
1060   } /* for (i = 0; i < values_num; i++) */
1061
1062   vl->values = values;
1063   vl->values_len = values_num;
1064
1065 #undef BAIL_OUT
1066   (*jvm_env)->DeleteLocalRef (jvm_env, o_number_array);
1067   (*jvm_env)->DeleteLocalRef (jvm_env, o_list);
1068   return (0);
1069 } /* }}} int jtoc_values_array */
1070
1071 /* Convert a org.collectd.api.ValueList to a value_list_t. */
1072 static int jtoc_value_list (JNIEnv *jvm_env, value_list_t *vl, /* {{{ */
1073     jobject object_ptr)
1074 {
1075   jclass class_ptr;
1076   int status;
1077   jlong tmp_long;
1078   const data_set_t *ds;
1079
1080   class_ptr = (*jvm_env)->GetObjectClass (jvm_env, object_ptr);
1081   if (class_ptr == NULL)
1082   {
1083     ERROR ("java plugin: jtoc_value_list: GetObjectClass failed.");
1084     return (-1);
1085   }
1086
1087 #define SET_STRING(buffer,method) do { \
1088   status = jtoc_string (jvm_env, buffer, sizeof (buffer), \
1089       class_ptr, object_ptr, method); \
1090   if (status != 0) { \
1091     ERROR ("java plugin: jtoc_value_list: jtoc_string (%s) failed.", \
1092         method); \
1093     return (-1); \
1094   } } while (0)
1095
1096   SET_STRING(vl->type, "getType");
1097
1098   ds = plugin_get_ds (vl->type);
1099   if (ds == NULL)
1100   {
1101     ERROR ("java plugin: jtoc_value_list: Data-set `%s' is not defined. "
1102         "Please consult the types.db(5) manpage for mor information.",
1103         vl->type);
1104     return (-1);
1105   }
1106
1107   SET_STRING(vl->host, "getHost");
1108   SET_STRING(vl->plugin, "getPlugin");
1109   SET_STRING(vl->plugin_instance, "getPluginInstance");
1110   SET_STRING(vl->type_instance, "getTypeInstance");
1111
1112 #undef SET_STRING
1113
1114   status = jtoc_long (jvm_env, &tmp_long, class_ptr, object_ptr, "getTime");
1115   if (status != 0)
1116   {
1117     ERROR ("java plugin: jtoc_value_list: jtoc_long (getTime) failed.");
1118     return (-1);
1119   }
1120   /* Java measures time in milliseconds. */
1121   vl->time = (time_t) (tmp_long / ((jlong) 1000));
1122
1123   status = jtoc_long (jvm_env, &tmp_long,
1124       class_ptr, object_ptr, "getInterval");
1125   if (status != 0)
1126   {
1127     ERROR ("java plugin: jtoc_value_list: jtoc_long (getInterval) failed.");
1128     return (-1);
1129   }
1130   vl->interval = (int) tmp_long;
1131
1132   status = jtoc_values_array (jvm_env, ds, vl, class_ptr, object_ptr);
1133   if (status != 0)
1134   {
1135     ERROR ("java plugin: jtoc_value_list: jtoc_values_array failed.");
1136     return (-1);
1137   }
1138
1139   return (0);
1140 } /* }}} int jtoc_value_list */
1141
1142 /*
1143  * Functions accessible from Java
1144  */
1145 static jint JNICALL cjni_api_dispatch_values (JNIEnv *jvm_env, /* {{{ */
1146     jobject this, jobject java_vl)
1147 {
1148   value_list_t vl = VALUE_LIST_INIT;
1149   int status;
1150
1151   DEBUG ("cjni_api_dispatch_values: java_vl = %p;", (void *) java_vl);
1152
1153   status = jtoc_value_list (jvm_env, &vl, java_vl);
1154   if (status != 0)
1155   {
1156     ERROR ("java plugin: cjni_api_dispatch_values: jtoc_value_list failed.");
1157     return (-1);
1158   }
1159
1160   status = plugin_dispatch_values (&vl);
1161
1162   sfree (vl.values);
1163
1164   return (status);
1165 } /* }}} jint cjni_api_dispatch_values */
1166
1167 static jobject JNICALL cjni_api_get_ds (JNIEnv *jvm_env, /* {{{ */
1168     jobject this, jobject o_string_type)
1169 {
1170   const char *ds_name;
1171   const data_set_t *ds;
1172   jobject o_dataset;
1173
1174   ds_name = (*jvm_env)->GetStringUTFChars (jvm_env, o_string_type, 0);
1175   if (ds_name == NULL)
1176   {
1177     ERROR ("java plugin: cjni_api_get_ds: GetStringUTFChars failed.");
1178     return (NULL);
1179   }
1180
1181   ds = plugin_get_ds (ds_name);
1182   DEBUG ("java plugin: cjni_api_get_ds: "
1183       "plugin_get_ds (%s) = %p;", ds_name, (void *) ds);
1184
1185   (*jvm_env)->ReleaseStringUTFChars (jvm_env, o_string_type, ds_name);
1186
1187   if (ds == NULL)
1188     return (NULL);
1189
1190   o_dataset = ctoj_data_set (jvm_env, ds);
1191   return (o_dataset);
1192 } /* }}} jint cjni_api_get_ds */
1193
1194 static jint JNICALL cjni_api_register_config (JNIEnv *jvm_env, /* {{{ */
1195     jobject this, jobject o_name, jobject o_config)
1196 {
1197   return (cjni_callback_register (jvm_env, o_name, o_config, CB_TYPE_CONFIG));
1198 } /* }}} jint cjni_api_register_config */
1199
1200 static jint JNICALL cjni_api_register_init (JNIEnv *jvm_env, /* {{{ */
1201     jobject this, jobject o_name, jobject o_config)
1202 {
1203   return (cjni_callback_register (jvm_env, o_name, o_config, CB_TYPE_INIT));
1204 } /* }}} jint cjni_api_register_init */
1205
1206 static jint JNICALL cjni_api_register_read (JNIEnv *jvm_env, /* {{{ */
1207     jobject this, jobject o_name, jobject o_read)
1208 {
1209   user_data_t ud;
1210   cjni_callback_info_t *cbi;
1211
1212   cbi = cjni_callback_info_create (jvm_env, o_name, o_read, CB_TYPE_READ);
1213   if (cbi == NULL)
1214     return (-1);
1215
1216   DEBUG ("java plugin: Registering new read callback: %s", cbi->name);
1217
1218   memset (&ud, 0, sizeof (ud));
1219   ud.data = (void *) cbi;
1220   ud.free_func = cjni_callback_info_destroy;
1221
1222   plugin_register_complex_read (cbi->name, cjni_read, &ud);
1223
1224   (*jvm_env)->DeleteLocalRef (jvm_env, o_read);
1225
1226   return (0);
1227 } /* }}} jint cjni_api_register_read */
1228
1229 static jint JNICALL cjni_api_register_write (JNIEnv *jvm_env, /* {{{ */
1230     jobject this, jobject o_name, jobject o_write)
1231 {
1232   user_data_t ud;
1233   cjni_callback_info_t *cbi;
1234
1235   cbi = cjni_callback_info_create (jvm_env, o_name, o_write, CB_TYPE_WRITE);
1236   if (cbi == NULL)
1237     return (-1);
1238
1239   DEBUG ("java plugin: Registering new write callback: %s", cbi->name);
1240
1241   memset (&ud, 0, sizeof (ud));
1242   ud.data = (void *) cbi;
1243   ud.free_func = cjni_callback_info_destroy;
1244
1245   plugin_register_write (cbi->name, cjni_write, &ud);
1246
1247   (*jvm_env)->DeleteLocalRef (jvm_env, o_write);
1248
1249   return (0);
1250 } /* }}} jint cjni_api_register_write */
1251
1252 static jint JNICALL cjni_api_register_flush (JNIEnv *jvm_env, /* {{{ */
1253     jobject this, jobject o_name, jobject o_flush)
1254 {
1255   user_data_t ud;
1256   cjni_callback_info_t *cbi;
1257
1258   cbi = cjni_callback_info_create (jvm_env, o_name, o_flush, CB_TYPE_FLUSH);
1259   if (cbi == NULL)
1260     return (-1);
1261
1262   DEBUG ("java plugin: Registering new flush callback: %s", cbi->name);
1263
1264   memset (&ud, 0, sizeof (ud));
1265   ud.data = (void *) cbi;
1266   ud.free_func = cjni_callback_info_destroy;
1267
1268   plugin_register_flush (cbi->name, cjni_flush, &ud);
1269
1270   (*jvm_env)->DeleteLocalRef (jvm_env, o_flush);
1271
1272   return (0);
1273 } /* }}} jint cjni_api_register_flush */
1274
1275 static jint JNICALL cjni_api_register_shutdown (JNIEnv *jvm_env, /* {{{ */
1276     jobject this, jobject o_name, jobject o_shutdown)
1277 {
1278   return (cjni_callback_register (jvm_env, o_name, o_shutdown,
1279         CB_TYPE_SHUTDOWN));
1280 } /* }}} jint cjni_api_register_shutdown */
1281
1282 static void JNICALL cjni_api_log (JNIEnv *jvm_env, /* {{{ */
1283     jobject this, jint severity, jobject o_message)
1284 {
1285   const char *c_str;
1286
1287   c_str = (*jvm_env)->GetStringUTFChars (jvm_env, o_message, 0);
1288   if (c_str == NULL)
1289   {
1290     ERROR ("java plugin: cjni_api_log: GetStringUTFChars failed.");
1291     return;
1292   }
1293
1294   if (severity < LOG_ERR)
1295     severity = LOG_ERR;
1296   if (severity > LOG_DEBUG)
1297     severity = LOG_DEBUG;
1298
1299   plugin_log (severity, "%s", c_str);
1300
1301   (*jvm_env)->ReleaseStringUTFChars (jvm_env, o_message, c_str);
1302 } /* }}} void cjni_api_log */
1303
1304 /* List of ``native'' functions, i. e. C-functions that can be called from
1305  * Java. */
1306 static JNINativeMethod jni_api_functions[] = /* {{{ */
1307 {
1308   { "dispatchValues",
1309     "(Lorg/collectd/api/ValueList;)I",
1310     cjni_api_dispatch_values },
1311
1312   { "getDS",
1313     "(Ljava/lang/String;)Lorg/collectd/api/DataSet;",
1314     cjni_api_get_ds },
1315
1316   { "registerConfig",
1317     "(Ljava/lang/String;Lorg/collectd/api/CollectdConfigInterface;)I",
1318     cjni_api_register_config },
1319
1320   { "registerInit",
1321     "(Ljava/lang/String;Lorg/collectd/api/CollectdInitInterface;)I",
1322     cjni_api_register_init },
1323
1324   { "registerRead",
1325     "(Ljava/lang/String;Lorg/collectd/api/CollectdReadInterface;)I",
1326     cjni_api_register_read },
1327
1328   { "registerWrite",
1329     "(Ljava/lang/String;Lorg/collectd/api/CollectdWriteInterface;)I",
1330     cjni_api_register_write },
1331
1332   { "registerFlush",
1333     "(Ljava/lang/String;Lorg/collectd/api/CollectdFlushInterface;)I",
1334     cjni_api_register_flush },
1335
1336   { "registerShutdown",
1337     "(Ljava/lang/String;Lorg/collectd/api/CollectdShutdownInterface;)I",
1338     cjni_api_register_shutdown },
1339
1340   { "log",
1341     "(ILjava/lang/String;)V",
1342     cjni_api_log },
1343 };
1344 static size_t jni_api_functions_num = sizeof (jni_api_functions)
1345   / sizeof (jni_api_functions[0]);
1346 /* }}} */
1347
1348 /*
1349  * Functions
1350  */
1351 /* Allocate a `cjni_callback_info_t' given the type and objects necessary for
1352  * all registration functions. */
1353 static cjni_callback_info_t *cjni_callback_info_create (JNIEnv *jvm_env, /* {{{ */
1354     jobject o_name, jobject o_callback, int type)
1355 {
1356   const char *c_name;
1357   cjni_callback_info_t *cbi;
1358   const char *method_name;
1359   const char *method_signature;
1360
1361   switch (type)
1362   {
1363     case CB_TYPE_CONFIG:
1364       method_name = "config";
1365       method_signature = "(Lorg/collectd/api/OConfigItem;)I";
1366       break;
1367
1368     case CB_TYPE_INIT:
1369       method_name = "init";
1370       method_signature = "()I";
1371       break;
1372
1373     case CB_TYPE_READ:
1374       method_name = "read";
1375       method_signature = "()I";
1376       break;
1377
1378     case CB_TYPE_WRITE:
1379       method_name = "write";
1380       method_signature = "(Lorg/collectd/api/ValueList;)I";
1381       break;
1382
1383     case CB_TYPE_FLUSH:
1384       method_name = "flush";
1385       method_signature = "(ILjava/lang/String;)I";
1386       break;
1387
1388     case CB_TYPE_SHUTDOWN:
1389       method_name = "shutdown";
1390       method_signature = "()I";
1391       break;
1392
1393     default:
1394       ERROR ("java plugin: cjni_callback_info_create: Unknown type: %#x",
1395           type);
1396       return (NULL);
1397   }
1398
1399   c_name = (*jvm_env)->GetStringUTFChars (jvm_env, o_name, 0);
1400   if (c_name == NULL)
1401   {
1402     ERROR ("java plugin: cjni_callback_info_create: "
1403         "GetStringUTFChars failed.");
1404     return (NULL);
1405   }
1406
1407   cbi = (cjni_callback_info_t *) malloc (sizeof (*cbi));
1408   if (cbi == NULL)
1409   {
1410     ERROR ("java plugin: cjni_callback_info_create: malloc failed.");
1411     (*jvm_env)->ReleaseStringUTFChars (jvm_env, o_name, c_name);
1412     return (NULL);
1413   }
1414   memset (cbi, 0, sizeof (*cbi));
1415   cbi->type = type;
1416
1417   cbi->name = strdup (c_name);
1418   if (cbi->name == NULL)
1419   {
1420     pthread_mutex_unlock (&java_callbacks_lock);
1421     ERROR ("java plugin: cjni_callback_info_create: strdup failed.");
1422     (*jvm_env)->ReleaseStringUTFChars (jvm_env, o_name, c_name);
1423     return (NULL);
1424   }
1425
1426   (*jvm_env)->ReleaseStringUTFChars (jvm_env, o_name, c_name);
1427
1428   cbi->class  = (*jvm_env)->GetObjectClass (jvm_env, o_callback);
1429   if (cbi->class == NULL)
1430   {
1431     ERROR ("java plugin: cjni_callback_info_create: GetObjectClass failed.");
1432     free (cbi);
1433     return (NULL);
1434   }
1435
1436   cbi->object = o_callback;
1437
1438   cbi->method = (*jvm_env)->GetMethodID (jvm_env, cbi->class,
1439       method_name, method_signature);
1440   if (cbi->method == NULL)
1441   {
1442     ERROR ("java plugin: cjni_callback_info_create: "
1443         "Cannot find the `%s' method with signature `%s'.",
1444         method_name, method_signature);
1445     free (cbi);
1446     return (NULL);
1447   }
1448
1449   (*jvm_env)->NewGlobalRef (jvm_env, o_callback);
1450
1451   return (cbi);
1452 } /* }}} cjni_callback_info_t cjni_callback_info_create */
1453
1454 /* Allocate a `cjni_callback_info_t' via `cjni_callback_info_create' and add it
1455  * to the global `java_callbacks' variable. This is used for `config', `init',
1456  * and `shutdown' callbacks. */
1457 static int cjni_callback_register (JNIEnv *jvm_env, /* {{{ */
1458     jobject o_name, jobject o_callback, int type)
1459 {
1460   cjni_callback_info_t *cbi;
1461   cjni_callback_info_t *tmp;
1462 #if COLLECT_DEBUG
1463   const char *type_str;
1464 #endif
1465
1466   cbi = cjni_callback_info_create (jvm_env, o_name, o_callback, type);
1467   if (cbi == NULL)
1468     return (-1);
1469
1470 #if COLLECT_DEBUG
1471   switch (type)
1472   {
1473     case CB_TYPE_CONFIG:
1474       type_str = "config";
1475       break;
1476
1477     case CB_TYPE_INIT:
1478       type_str = "init";
1479       break;
1480
1481     case CB_TYPE_SHUTDOWN:
1482       type_str = "shutdown";
1483       break;
1484
1485     default:
1486       type_str = "<unknown>";
1487   }
1488   DEBUG ("java plugin: Registering new %s callback: %s",
1489       type_str, cbi->name);
1490 #endif
1491
1492   pthread_mutex_lock (&java_callbacks_lock);
1493
1494   tmp = (cjni_callback_info_t *) realloc (java_callbacks,
1495       (java_callbacks_num + 1) * sizeof (*java_callbacks));
1496   if (tmp == NULL)
1497   {
1498     pthread_mutex_unlock (&java_callbacks_lock);
1499     ERROR ("java plugin: cjni_callback_register: realloc failed.");
1500
1501     (*jvm_env)->DeleteGlobalRef (jvm_env, cbi->object);
1502     free (cbi);
1503
1504     return (-1);
1505   }
1506   java_callbacks = tmp;
1507   java_callbacks[java_callbacks_num] = *cbi;
1508   java_callbacks_num++;
1509
1510   pthread_mutex_unlock (&java_callbacks_lock);
1511
1512   free (cbi);
1513   return (0);
1514 } /* }}} int cjni_callback_register */
1515
1516 /* Increase the reference counter to the JVM for this thread. If it was zero,
1517  * attach the JVM first. */
1518 static JNIEnv *cjni_thread_attach (void) /* {{{ */
1519 {
1520   cjni_jvm_env_t *cjni_env;
1521   JNIEnv *jvm_env;
1522
1523   cjni_env = pthread_getspecific (jvm_env_key);
1524   if (cjni_env == NULL)
1525   {
1526     /* This pointer is free'd in `cjni_jvm_env_destroy'. */
1527     cjni_env = (cjni_jvm_env_t *) malloc (sizeof (*cjni_env));
1528     if (cjni_env == NULL)
1529     {
1530       ERROR ("java plugin: cjni_thread_attach: malloc failed.");
1531       return (NULL);
1532     }
1533     memset (cjni_env, 0, sizeof (*cjni_env));
1534     cjni_env->reference_counter = 0;
1535     cjni_env->jvm_env = NULL;
1536
1537     pthread_setspecific (jvm_env_key, cjni_env);
1538   }
1539
1540   if (cjni_env->reference_counter > 0)
1541   {
1542     cjni_env->reference_counter++;
1543     jvm_env = cjni_env->jvm_env;
1544   }
1545   else
1546   {
1547     int status;
1548     JavaVMAttachArgs args;
1549
1550     assert (cjni_env->jvm_env == NULL);
1551
1552     memset (&args, 0, sizeof (args));
1553     args.version = JNI_VERSION_1_2;
1554
1555     status = (*jvm)->AttachCurrentThread (jvm, (void *) &jvm_env, (void *) &args);
1556     if (status != 0)
1557     {
1558       ERROR ("java plugin: cjni_thread_attach: AttachCurrentThread failed "
1559           "with status %i.", status);
1560       return (NULL);
1561     }
1562
1563     cjni_env->reference_counter = 1;
1564     cjni_env->jvm_env = jvm_env;
1565   }
1566
1567   DEBUG ("java plugin: cjni_thread_attach: cjni_env->reference_counter = %i",
1568       cjni_env->reference_counter);
1569   assert (jvm_env != NULL);
1570   return (jvm_env);
1571 } /* }}} JNIEnv *cjni_thread_attach */
1572
1573 /* Decrease the reference counter of this thread. If it reaches zero, detach
1574  * from the JVM. */
1575 static int cjni_thread_detach (void) /* {{{ */
1576 {
1577   cjni_jvm_env_t *cjni_env;
1578   int status;
1579
1580   cjni_env = pthread_getspecific (jvm_env_key);
1581   if (cjni_env == NULL)
1582   {
1583     ERROR ("java plugin: cjni_thread_detach: pthread_getspecific failed.");
1584     return (-1);
1585   }
1586
1587   assert (cjni_env->reference_counter > 0);
1588   assert (cjni_env->jvm_env != NULL);
1589
1590   cjni_env->reference_counter--;
1591   DEBUG ("java plugin: cjni_thread_detach: cjni_env->reference_counter = %i",
1592       cjni_env->reference_counter);
1593
1594   if (cjni_env->reference_counter > 0)
1595     return (0);
1596
1597   status = (*jvm)->DetachCurrentThread (jvm);
1598   if (status != 0)
1599   {
1600     ERROR ("java plugin: cjni_thread_detach: DetachCurrentThread failed "
1601         "with status %i.", status);
1602   }
1603
1604   cjni_env->reference_counter = 0;
1605   cjni_env->jvm_env = NULL;
1606
1607   return (0);
1608 } /* }}} JNIEnv *cjni_thread_attach */
1609
1610 /* Callback for `pthread_key_create'. It frees the data contained in
1611  * `jvm_env_key' and prints a warning if the reference counter is not zero. */
1612 static void cjni_jvm_env_destroy (void *args) /* {{{ */
1613 {
1614   cjni_jvm_env_t *cjni_env;
1615
1616   if (args == NULL)
1617     return;
1618
1619   cjni_env = (cjni_jvm_env_t *) args;
1620
1621   if (cjni_env->reference_counter > 0)
1622   {
1623     ERROR ("java plugin: cjni_jvm_env_destroy: "
1624         "cjni_env->reference_counter = %i;", cjni_env->reference_counter);
1625   }
1626
1627   if (cjni_env->jvm_env != NULL)
1628   {
1629     ERROR ("java plugin: cjni_jvm_env_destroy: cjni_env->jvm_env = %p;",
1630         (void *) cjni_env->jvm_env);
1631   }
1632
1633   /* The pointer is allocated in `cjni_thread_attach' */
1634   free (cjni_env);
1635 } /* }}} void cjni_jvm_env_destroy */
1636
1637 /* Boring configuration functions.. {{{ */
1638 static int cjni_config_add_jvm_arg (oconfig_item_t *ci) /* {{{ */
1639 {
1640   char **tmp;
1641
1642   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING))
1643   {
1644     WARNING ("java plugin: `JVMArg' needs exactly one string argument.");
1645     return (-1);
1646   }
1647
1648   tmp = (char **) realloc (jvm_argv, sizeof (char *) * (jvm_argc + 1));
1649   if (tmp == NULL)
1650   {
1651     ERROR ("java plugin: realloc failed.");
1652     return (-1);
1653   }
1654   jvm_argv = tmp;
1655
1656   jvm_argv[jvm_argc] = strdup (ci->values[0].value.string);
1657   if (jvm_argv[jvm_argc] == NULL)
1658   {
1659     ERROR ("java plugin: strdup failed.");
1660     return (-1);
1661   }
1662   jvm_argc++;
1663
1664   return (0);
1665 } /* }}} int cjni_config_add_jvm_arg */
1666
1667 static int cjni_config_load_plugin (oconfig_item_t *ci) /* {{{ */
1668 {
1669   java_plugin_class_t *tmp;
1670
1671   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING))
1672   {
1673     WARNING ("java plugin: `LoadPlugin' needs exactly one string argument.");
1674     return (-1);
1675   }
1676
1677   tmp = (java_plugin_class_t *) realloc (java_classes_list,
1678       (java_classes_list_len + 1) * sizeof (*java_classes_list));
1679   if (tmp == NULL)
1680   {
1681     ERROR ("java plugin: realloc failed.");
1682     return (-1);
1683   }
1684   java_classes_list = tmp;
1685   tmp = java_classes_list + java_classes_list_len;
1686
1687   memset (tmp, 0, sizeof (*tmp));
1688   tmp->name = strdup (ci->values[0].value.string);
1689   if (tmp->name == NULL)
1690   {
1691     ERROR ("java plugin: strdup failed.");
1692     return (-1);
1693   }
1694   tmp->class = NULL;
1695   tmp->object = NULL;
1696
1697   java_classes_list_len++;
1698
1699   return (0);
1700 } /* }}} int cjni_config_load_plugin */
1701
1702 static int cjni_config_plugin_block (oconfig_item_t *ci) /* {{{ */
1703 {
1704   java_plugin_config_t *tmp;
1705   char *name;
1706   size_t i;
1707
1708   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING))
1709   {
1710     WARNING ("java plugin: `Plugin' blocks "
1711         "need exactly one string argument.");
1712     return (-1);
1713   }
1714
1715   name = strdup (ci->values[0].value.string);
1716   if (name == NULL)
1717   {
1718     ERROR ("java plugin: cjni_config_plugin_block: strdup faiiled.");
1719     return (-1);
1720   }
1721
1722   for (i = 0; i < java_plugin_configs_num; i++)
1723   {
1724     if (strcmp (java_plugin_configs[i].name, name) == 0)
1725     {
1726       WARNING ("java plugin: There is more than one <Plugin \"%s\"> block. "
1727           "This is currently not supported - "
1728           "only the first block will be used!",
1729           name);
1730       free (name);
1731       return (0);
1732     }
1733   }
1734
1735   tmp = (java_plugin_config_t *) realloc (java_plugin_configs,
1736       (java_plugin_configs_num + 1) * sizeof (*java_plugin_configs));
1737   if (tmp == NULL)
1738   {
1739     ERROR ("java plugin: cjni_config_plugin_block: realloc failed.");
1740     free (name);
1741     return (-1);
1742   }
1743   java_plugin_configs = tmp;
1744   tmp = java_plugin_configs + java_plugin_configs_num;
1745
1746   tmp->name = name;
1747   tmp->ci = oconfig_clone (ci);
1748   if (tmp->ci == NULL)
1749   {
1750     ERROR ("java plugin: cjni_config_plugin_block: "
1751         "oconfig_clone failed for `%s'.",
1752         name);
1753     free (name);
1754     return (-1);
1755   }
1756
1757   DEBUG ("java plugin: cjni_config_plugin_block: "
1758       "Successfully copied config for `%s'.",
1759       name);
1760
1761   java_plugin_configs_num++;
1762   return (0);
1763 } /* }}} int cjni_config_plugin_block */
1764
1765 static int cjni_config (oconfig_item_t *ci) /* {{{ */
1766 {
1767   int success;
1768   int errors;
1769   int status;
1770   int i;
1771
1772   success = 0;
1773   errors = 0;
1774
1775   for (i = 0; i < ci->children_num; i++)
1776   {
1777     oconfig_item_t *child = ci->children + i;
1778
1779     if (strcasecmp ("JVMArg", child->key) == 0)
1780     {
1781       status = cjni_config_add_jvm_arg (child);
1782       if (status == 0)
1783         success++;
1784       else
1785         errors++;
1786     }
1787     else if (strcasecmp ("LoadPlugin", child->key) == 0)
1788     {
1789       status = cjni_config_load_plugin (child);
1790       if (status == 0)
1791         success++;
1792       else
1793         errors++;
1794     }
1795     else if (strcasecmp ("Plugin", child->key) == 0)
1796     {
1797       status = cjni_config_plugin_block (child);
1798       if (status == 0)
1799         success++;
1800       else
1801         errors++;
1802     }
1803     else
1804     {
1805       WARNING ("java plugin: Option `%s' not allowed here.", child->key);
1806       errors++;
1807     }
1808   }
1809
1810   DEBUG ("java plugin: jvm_argc = %zu;", jvm_argc);
1811   DEBUG ("java plugin: java_classes_list_len = %zu;", java_classes_list_len);
1812   DEBUG ("java plugin: java_plugin_configs_num = %zu;",
1813       java_plugin_configs_num);
1814
1815   if ((success == 0) && (errors > 0))
1816   {
1817     ERROR ("java plugin: All statements failed.");
1818     return (-1);
1819   }
1820
1821   return (0);
1822 } /* }}} int cjni_config */
1823 /* }}} */
1824
1825 /* Free the data contained in the `user_data_t' pointer passed to `cjni_read'
1826  * and `cjni_write'. In particular, delete the global reference to the Java
1827  * object. */
1828 static void cjni_callback_info_destroy (void *arg) /* {{{ */
1829 {
1830   JNIEnv *jvm_env;
1831   cjni_callback_info_t *cbi;
1832
1833   DEBUG ("java plugin: cjni_callback_info_destroy (arg = %p);", arg);
1834
1835   if (arg == NULL)
1836     return;
1837
1838   jvm_env = cjni_thread_attach ();
1839   if (jvm_env == NULL)
1840   {
1841     ERROR ("java plugin: cjni_callback_info_destroy: cjni_thread_attach failed.");
1842     return;
1843   }
1844
1845   cbi = (cjni_callback_info_t *) arg;
1846
1847   (*jvm_env)->DeleteGlobalRef (jvm_env, cbi->object);
1848
1849   cbi->method = NULL;
1850   cbi->object = NULL;
1851   cbi->class  = NULL;
1852   free (cbi);
1853
1854   cjni_thread_detach ();
1855 } /* }}} void cjni_callback_info_destroy */
1856
1857 /* Call the CB_TYPE_READ callback pointed to by the `user_data_t' pointer. */
1858 static int cjni_read (user_data_t *ud) /* {{{ */
1859 {
1860   JNIEnv *jvm_env;
1861   cjni_callback_info_t *cbi;
1862   int status;
1863
1864   if (jvm == NULL)
1865   {
1866     ERROR ("java plugin: cjni_read: jvm == NULL");
1867     return (-1);
1868   }
1869
1870   if ((ud == NULL) || (ud->data == NULL))
1871   {
1872     ERROR ("java plugin: cjni_read: Invalid user data.");
1873     return (-1);
1874   }
1875
1876   jvm_env = cjni_thread_attach ();
1877   if (jvm_env == NULL)
1878     return (-1);
1879
1880   cbi = (cjni_callback_info_t *) ud->data;
1881
1882   status = (*jvm_env)->CallIntMethod (jvm_env, cbi->object,
1883       cbi->method);
1884
1885   status = cjni_thread_detach ();
1886   if (status != 0)
1887   {
1888     ERROR ("java plugin: cjni_read: cjni_thread_detach failed.");
1889     return (-1);
1890   }
1891
1892   return (status);
1893 } /* }}} int cjni_read */
1894
1895 /* Call the CB_TYPE_WRITE callback pointed to by the `user_data_t' pointer. */
1896 static int cjni_write (const data_set_t *ds, const value_list_t *vl, /* {{{ */
1897     user_data_t *ud)
1898 {
1899   JNIEnv *jvm_env;
1900   cjni_callback_info_t *cbi;
1901   jobject vl_java;
1902   int status;
1903
1904   if (jvm == NULL)
1905   {
1906     ERROR ("java plugin: cjni_write: jvm == NULL");
1907     return (-1);
1908   }
1909
1910   if ((ud == NULL) || (ud->data == NULL))
1911   {
1912     ERROR ("java plugin: cjni_write: Invalid user data.");
1913     return (-1);
1914   }
1915
1916   jvm_env = cjni_thread_attach ();
1917   if (jvm_env == NULL)
1918     return (-1);
1919
1920   cbi = (cjni_callback_info_t *) ud->data;
1921
1922   vl_java = ctoj_value_list (jvm_env, ds, vl);
1923   if (vl_java == NULL)
1924   {
1925     ERROR ("java plugin: cjni_write: ctoj_value_list failed.");
1926     return (-1);
1927   }
1928
1929   status = (*jvm_env)->CallIntMethod (jvm_env,
1930       cbi->object, cbi->method, vl_java);
1931
1932   (*jvm_env)->DeleteLocalRef (jvm_env, vl_java);
1933
1934   status = cjni_thread_detach ();
1935   if (status != 0)
1936   {
1937     ERROR ("java plugin: cjni_write: cjni_thread_detach failed.");
1938     return (-1);
1939   }
1940
1941   return (status);
1942 } /* }}} int cjni_write */
1943
1944 /* Call the CB_TYPE_WRITE callback pointed to by the `user_data_t' pointer. */
1945 static int cjni_flush (int timeout, const char *identifier, /* {{{ */
1946     user_data_t *ud)
1947 {
1948   JNIEnv *jvm_env;
1949   cjni_callback_info_t *cbi;
1950   jobject o_identifier;
1951   int status;
1952
1953   if (jvm == NULL)
1954   {
1955     ERROR ("java plugin: cjni_flush: jvm == NULL");
1956     return (-1);
1957   }
1958
1959   if ((ud == NULL) || (ud->data == NULL))
1960   {
1961     ERROR ("java plugin: cjni_flush: Invalid user data.");
1962     return (-1);
1963   }
1964
1965   jvm_env = cjni_thread_attach ();
1966   if (jvm_env == NULL)
1967     return (-1);
1968
1969   cbi = (cjni_callback_info_t *) ud->data;
1970
1971   o_identifier = NULL;
1972   if (identifier != NULL)
1973   {
1974     o_identifier = (*jvm_env)->NewStringUTF (jvm_env, identifier);
1975     if (o_identifier == NULL)
1976     {
1977       ERROR ("java plugin: cjni_flush: NewStringUTF failed.");
1978       return (-1);
1979     }
1980   }
1981
1982   status = (*jvm_env)->CallIntMethod (jvm_env,
1983       cbi->object, cbi->method, (jint) timeout, o_identifier);
1984
1985   (*jvm_env)->DeleteLocalRef (jvm_env, o_identifier);
1986
1987   status = cjni_thread_detach ();
1988   if (status != 0)
1989   {
1990     ERROR ("java plugin: cjni_flush: cjni_thread_detach failed.");
1991     return (-1);
1992   }
1993
1994   return (status);
1995 } /* }}} int cjni_flush */
1996
1997 /* Iterate over `java_classes_list' and create one object of each class. This
1998  * will trigger the object's constructors, to the objects can register callback
1999  * methods. */
2000 static int cjni_load_plugins (JNIEnv *jvm_env) /* {{{ */
2001 {
2002   size_t i;
2003
2004   for (i = 0; i < java_classes_list_len; i++)
2005   {
2006     java_plugin_class_t *class;
2007     jmethodID constructor_id;
2008
2009     class = java_classes_list + i;
2010
2011     DEBUG ("java plugin: Loading class %s", class->name);
2012
2013     class->class = (*jvm_env)->FindClass (jvm_env, class->name);
2014     if (class->class == NULL)
2015     {
2016       ERROR ("java plugin: cjni_load_plugins: FindClass (%s) failed.",
2017           class->name);
2018       continue;
2019     }
2020
2021     constructor_id = (*jvm_env)->GetMethodID (jvm_env, class->class,
2022         "<init>", "()V");
2023     if (constructor_id == NULL)
2024     {
2025       ERROR ("java plugin: cjni_load_plugins: Could not find the constructor for `%s'.",
2026           class->name);
2027       continue;
2028     }
2029
2030     class->object = (*jvm_env)->NewObject (jvm_env, class->class,
2031         constructor_id);
2032     if (class->object == NULL)
2033     {
2034       ERROR ("java plugin: cjni_load_plugins: Could create a new `%s' object.",
2035           class->name);
2036       continue;
2037     }
2038
2039     (*jvm_env)->NewGlobalRef (jvm_env, class->object);
2040   } /* for (i = 0; i < java_classes_list_len; i++) */
2041
2042   return (0);
2043 } /* }}} int cjni_load_plugins */
2044
2045 /* Iterate over `java_plugin_configs' and `java_callbacks' and call all
2046  * `config' callback methods for which a configuration is available. */
2047 static int cjni_config_plugins (JNIEnv *jvm_env) /* {{{ */
2048 {
2049   int status;
2050   size_t i;
2051   size_t j;
2052
2053   for (i = 0; i < java_plugin_configs_num; i++)
2054   {
2055     jobject o_ocitem;
2056
2057     if (java_plugin_configs[i].ci == NULL)
2058       continue;
2059
2060     for (j = 0; j < java_callbacks_num; j++)
2061     {
2062       if (java_callbacks[j].type != CB_TYPE_CONFIG)
2063         continue;
2064
2065       if (strcmp (java_plugin_configs[i].name, java_callbacks[j].name) == 0)
2066         break;
2067     }
2068
2069     if (j >= java_callbacks_num)
2070     {
2071       NOTICE ("java plugin: Configuration for `%s' is present, but no such "
2072           "configuration callback has been registered.",
2073           java_plugin_configs[i].name);
2074       continue;
2075     }
2076
2077     DEBUG ("java plugin: Configuring %s", java_plugin_configs[i].name);
2078
2079     o_ocitem = ctoj_oconfig_item (jvm_env, java_plugin_configs[i].ci);
2080     if (o_ocitem == NULL)
2081     {
2082       ERROR ("java plugin: cjni_config_plugins: ctoj_oconfig_item failed.");
2083       continue;
2084     }
2085
2086     status = (*jvm_env)->CallIntMethod (jvm_env,
2087         java_callbacks[j].object, java_callbacks[j].method, o_ocitem);
2088     WARNING ("java plugin: Config callback for `%s' returned status %i.",
2089         java_plugin_configs[i].name, status);
2090   } /* for (i = 0; i < java_plugin_configs; i++) */
2091
2092   return (0);
2093 } /* }}} int cjni_config_plugins */
2094
2095 /* Iterate over `java_callbacks' and call all CB_TYPE_INIT callbacks. */
2096 static int cjni_init_plugins (JNIEnv *jvm_env) /* {{{ */
2097 {
2098   int status;
2099   size_t i;
2100
2101   for (i = 0; i < java_callbacks_num; i++)
2102   {
2103     if (java_callbacks[i].type != CB_TYPE_INIT)
2104       continue;
2105
2106     DEBUG ("java plugin: Initializing %s", java_callbacks[i].name);
2107
2108     status = (*jvm_env)->CallIntMethod (jvm_env,
2109         java_callbacks[i].object, java_callbacks[i].method);
2110     if (status != 0)
2111     {
2112       ERROR ("java plugin: Initializing `%s' failed with status %i. "
2113           "Removing read function.",
2114           java_callbacks[i].name, status);
2115       plugin_unregister_read (java_callbacks[i].name);
2116     }
2117   }
2118
2119   return (0);
2120 } /* }}} int cjni_init_plugins */
2121
2122 /* Iterate over `java_callbacks' and call all CB_TYPE_SHUTDOWN callbacks. */
2123 static int cjni_shutdown_plugins (JNIEnv *jvm_env) /* {{{ */
2124 {
2125   int status;
2126   size_t i;
2127
2128   for (i = 0; i < java_callbacks_num; i++)
2129   {
2130     if (java_callbacks[i].type != CB_TYPE_SHUTDOWN)
2131       continue;
2132
2133     DEBUG ("java plugin: Shutting down %s", java_callbacks[i].name);
2134
2135     status = (*jvm_env)->CallIntMethod (jvm_env,
2136         java_callbacks[i].object, java_callbacks[i].method);
2137     if (status != 0)
2138     {
2139       ERROR ("java plugin: Shutting down `%s' failed with status %i. ",
2140           java_callbacks[i].name, status);
2141     }
2142   }
2143
2144   return (0);
2145 } /* }}} int cjni_shutdown_plugins */
2146
2147
2148 static int cjni_shutdown (void) /* {{{ */
2149 {
2150   JNIEnv *jvm_env;
2151   JavaVMAttachArgs args;
2152   int status;
2153   size_t i;
2154
2155   if (jvm == NULL)
2156     return (0);
2157
2158   jvm_env = NULL;
2159   memset (&args, 0, sizeof (args));
2160   args.version = JNI_VERSION_1_2;
2161
2162   status = (*jvm)->AttachCurrentThread (jvm, (void **) &jvm_env, &args);
2163   if (status != 0)
2164   {
2165     ERROR ("java plugin: cjni_shutdown: AttachCurrentThread failed with status %i.",
2166         status);
2167     return (-1);
2168   }
2169
2170   /* Execute all the shutdown functions registered by plugins. */
2171   cjni_shutdown_plugins (jvm_env);
2172
2173   /* Release all the global references to callback functions */
2174   for (i = 0; i < java_callbacks_num; i++)
2175   {
2176     if (java_callbacks[i].object != NULL)
2177     {
2178       (*jvm_env)->DeleteGlobalRef (jvm_env, java_callbacks[i].object);
2179       java_callbacks[i].object = NULL;
2180     }
2181     sfree (java_callbacks[i].name);
2182   }
2183   java_callbacks_num = 0;
2184   sfree (java_callbacks);
2185
2186   /* Release all the global references to directly loaded classes. */
2187   for (i = 0; i < java_classes_list_len; i++)
2188   {
2189     if (java_classes_list[i].object != NULL)
2190     {
2191       (*jvm_env)->DeleteGlobalRef (jvm_env, java_classes_list[i].object);
2192       java_classes_list[i].object = NULL;
2193     }
2194     sfree (java_classes_list[i].name);
2195   }
2196   java_classes_list_len = 0;
2197   sfree (java_classes_list);
2198
2199   /* Destroy the JVM */
2200   (*jvm)->DestroyJavaVM (jvm);
2201   jvm = NULL;
2202   jvm_env = NULL;
2203
2204   pthread_key_delete (jvm_env_key);
2205
2206   /* Free the JVM argument list */
2207   for (i = 0; i < jvm_argc; i++)
2208     sfree (jvm_argv[i]);
2209   jvm_argc = 0;
2210   sfree (jvm_argv);
2211
2212   /* Free the copied configuration */
2213   for (i = 0; i < java_plugin_configs_num; i++)
2214   {
2215     sfree (java_plugin_configs[i].name);
2216     oconfig_free (java_plugin_configs[i].ci);
2217   }
2218   java_plugin_configs_num = 0;
2219   sfree (java_plugin_configs);
2220
2221   return (0);
2222 } /* }}} int cjni_shutdown */
2223
2224 /* Register ``native'' functions with the JVM. Native functions are C-functions
2225  * that can be called by Java code. */
2226 static int cjni_init_native (JNIEnv *jvm_env) /* {{{ */
2227 {
2228   jclass api_class_ptr;
2229   int status;
2230
2231   api_class_ptr = (*jvm_env)->FindClass (jvm_env, "org.collectd.api.Collectd");
2232   if (api_class_ptr == NULL)
2233   {
2234     ERROR ("cjni_init_native: Cannot find API class `org.collectd.api.Collectd'.");
2235     return (-1);
2236   }
2237
2238   status = (*jvm_env)->RegisterNatives (jvm_env, api_class_ptr,
2239       jni_api_functions, (jint) jni_api_functions_num);
2240   if (status != 0)
2241   {
2242     ERROR ("cjni_init_native: RegisterNatives failed with status %i.", status);
2243     return (-1);
2244   }
2245
2246   return (0);
2247 } /* }}} int cjni_init_native */
2248
2249 /* Initialization: Create a JVM, load all configured classes and call their
2250  * `config' and `init' callback methods. */
2251 static int cjni_init (void) /* {{{ */
2252 {
2253   JNIEnv *jvm_env;
2254   JavaVMInitArgs vm_args;
2255   JavaVMOption vm_options[jvm_argc];
2256
2257   int status;
2258   size_t i;
2259
2260   if (jvm != NULL)
2261     return (0);
2262
2263   status = pthread_key_create (&jvm_env_key, cjni_jvm_env_destroy);
2264   if (status != 0)
2265   {
2266     ERROR ("java plugin: cjni_init: pthread_key_create failed "
2267         "with status %i.", status);
2268     return (-1);
2269   }
2270
2271   jvm_env = NULL;
2272
2273   memset (&vm_args, 0, sizeof (vm_args));
2274   vm_args.version = JNI_VERSION_1_2;
2275   vm_args.options = vm_options;
2276   vm_args.nOptions = (jint) jvm_argc;
2277
2278   for (i = 0; i < jvm_argc; i++)
2279   {
2280     DEBUG ("java plugin: cjni_init: jvm_argv[%zu] = %s", i, jvm_argv[i]);
2281     vm_args.options[i].optionString = jvm_argv[i];
2282   }
2283   /*
2284   vm_args.options[0].optionString = "-verbose:jni";
2285   vm_args.options[1].optionString = "-Djava.class.path=/home/octo/collectd/bindings/java";
2286   */
2287
2288   status = JNI_CreateJavaVM (&jvm, (void **) &jvm_env, (void **) &vm_args);
2289   if (status != 0)
2290   {
2291     ERROR ("cjni_init: JNI_CreateJavaVM failed with status %i.",
2292         status);
2293     return (-1);
2294   }
2295   assert (jvm != NULL);
2296   assert (jvm_env != NULL);
2297
2298   /* Call RegisterNatives */
2299   status = cjni_init_native (jvm_env);
2300   if (status != 0)
2301   {
2302     ERROR ("cjni_init: cjni_init_native failed.");
2303     return (-1);
2304   }
2305
2306   cjni_load_plugins (jvm_env);
2307   cjni_config_plugins (jvm_env);
2308   cjni_init_plugins (jvm_env);
2309
2310   return (0);
2311 } /* }}} int cjni_init */
2312
2313 void module_register (void)
2314 {
2315   plugin_register_complex_config ("java", cjni_config);
2316   plugin_register_init ("java", cjni_init);
2317   plugin_register_shutdown ("java", cjni_shutdown);
2318 } /* void module_register (void) */
2319
2320 /* vim: set sw=2 sts=2 et fdm=marker : */