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