{GPL, other}: Relicense to MIT license.
[collectd.git] / src / nut.c
index 73a3318..d5ecc98 100644 (file)
--- a/src/nut.c
+++ b/src/nut.c
 /**
  * collectd - src/nut.c
- * Copyright (C) 2007  Florian octo Forster
+ * Copyright (C) 2007       Florian octo Forster
  *
- * This program is free software; you can redistribute it and/or modify it
- * under the terms of the GNU General Public License as published by the
- * Free Software Foundation; only version 2 of the License is applicable.
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
  *
- * This program is distributed in the hope that it will be useful, but
- * WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
- * General Public License for more details.
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
  *
- * You should have received a copy of the GNU General Public License along
- * with this program; if not, write to the Free Software Foundation, Inc.,
- * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+ * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+ * DEALINGS IN THE SOFTWARE.
  *
  * Authors:
- *   Florian octo Forster <octo at verplant.org>
+ *   Florian octo Forster <octo at collectd.org>
  **/
 
 #include "collectd.h"
 #include "common.h"
 #include "plugin.h"
 
-#if HAVE_PTHREAD_H
-# include <pthread.h>
-#endif
+#include <pthread.h>
+#include <upsclient.h>
 
-#if HAVE_UPSCLIENT_H
-# include <upsclient.h>
-# define NUT_HAVE_READ 1
+#if HAVE_UPSCONN_T
+typedef UPSCONN_t collectd_upsconn_t;
+#elif HAVE_UPSCONN
+typedef UPSCONN collectd_upsconn_t;
 #else
-# define NUT_HAVE_READ 0
+# error "Unable to determine the UPS connection type."
 #endif
 
-static data_source_t data_source_current[1] =
-{
-  {"value", DS_TYPE_GAUGE, NAN, NAN}
-};
-
-static data_set_t ds_current =
-{
-  "current", 1, data_source_current
-};
-
-static data_source_t data_source_humidity[1] =
-{
-  {"value", DS_TYPE_GAUGE, 0.0, 100.0}
-};
-
-static data_set_t ds_humidity =
-{
-  "humidity", 1, data_source_humidity
-};
-
-static data_source_t data_source_power[1] =
-{
-  {"value", DS_TYPE_GAUGE, 0.0, NAN}
-};
-
-static data_set_t ds_power =
-{
-  "power", 1, data_source_power
-};
-
-static data_source_t data_source_voltage[1] =
-{
-  {"value", DS_TYPE_GAUGE, NAN, NAN}
-};
-
-static data_set_t ds_voltage =
-{
-  "voltage", 1, data_source_voltage
-};
-
-static data_source_t data_source_percent[1] =
-{
-  {"percent", DS_TYPE_GAUGE, 0, 100.1}
-};
-
-static data_set_t ds_percent =
-{
-  "percent", 1, data_source_percent
-};
-
-static data_source_t data_source_timeleft[1] =
-{
-  {"timeleft", DS_TYPE_GAUGE, 0, 100.0}
-};
-
-static data_set_t ds_timeleft =
-{
-  "timeleft", 1, data_source_timeleft
-};
-
-static data_source_t data_source_temperature[1] =
-{
-  {"value", DS_TYPE_GAUGE, -273.15, NAN}
-};
-
-static data_set_t ds_temperature =
-{
-  "temperature", 1, data_source_temperature
-};
-
-static data_source_t data_source_frequency[1] =
-{
-  {"frequency", DS_TYPE_GAUGE, 0, NAN}
-};
-
-static data_set_t ds_frequency =
-{
-  "frequency", 1, data_source_frequency
-};
-
-#if NUT_HAVE_READ
 struct nut_ups_s;
 typedef struct nut_ups_s nut_ups_t;
 struct nut_ups_s
 {
-  UPSCONN    conn;
+  collectd_upsconn_t *conn;
   char      *upsname;
   char      *hostname;
   int        port;
@@ -139,10 +63,14 @@ static int config_keys_num = STATIC_ARRAY_SIZE(config_keys);
 
 static void free_nut_ups_t (nut_ups_t *ups)
 {
-    upscli_disconnect (&ups->conn);
-    sfree (ups->hostname);
-    sfree (ups->upsname);
-    sfree (ups);
+  if (ups->conn != NULL)
+  {
+    upscli_disconnect (ups->conn);
+    sfree (ups->conn);
+  }
+  sfree (ups->hostname);
+  sfree (ups->upsname);
+  sfree (ups);
 } /* void free_nut_ups_t */
 
 static int nut_add_ups (const char *name)
@@ -169,16 +97,6 @@ static int nut_add_ups (const char *name)
     return (1);
   }
 
-  status = upscli_connect (&ups->conn, ups->hostname, ups->port,
-      UPSCLI_CONN_TRYSSL);
-  if (status != 0)
-  {
-    ERROR ("nut plugin: nut_add_ups: upscli_connect (%s, %i) failed: %s",
-       ups->hostname, ups->port, upscli_strerror (&ups->conn));
-    free_nut_ups_t (ups);
-    return (1);
-  }
-
   if (upslist_head == NULL)
     upslist_head = ups;
   else
@@ -210,21 +128,17 @@ static void nut_submit (nut_ups_t *ups, const char *type,
 
   vl.values = values;
   vl.values_len = STATIC_ARRAY_SIZE (values);
-  vl.time = time (NULL);
-  strncpy (vl.host,
+  sstrncpy (vl.host,
       (strcasecmp (ups->hostname, "localhost") == 0)
       ? hostname_g
       : ups->hostname,
       sizeof (vl.host));
-  strcpy (vl.plugin, "nut");
-  strncpy (vl.plugin_instance, ups->upsname, sizeof (vl.plugin_instance));
-  strncpy (vl.type_instance, type_instance, sizeof (vl.type_instance));
+  sstrncpy (vl.plugin, "nut", sizeof (vl.plugin));
+  sstrncpy (vl.plugin_instance, ups->upsname, sizeof (vl.plugin_instance));
+  sstrncpy (vl.type, type, sizeof (vl.type));
+  sstrncpy (vl.type_instance, type_instance, sizeof (vl.type_instance));
 
-  vl.host[sizeof (vl.host) - 1] = '\0';
-  vl.plugin_instance[sizeof (vl.plugin_instance) - 1] = '\0';
-  vl.type_instance[sizeof (vl.type_instance) - 1] = '\0';
-
-  plugin_dispatch_values (type, &vl);
+  plugin_dispatch_values (&vl);
 } /* void nut_submit */
 
 static int nut_read_one (nut_ups_t *ups)
@@ -235,17 +149,43 @@ static int nut_read_one (nut_ups_t *ups)
   unsigned int answer_num;
   int status;
 
+  /* (Re-)Connect if we have no connection */
+  if (ups->conn == NULL)
+  {
+    ups->conn = (collectd_upsconn_t *) malloc (sizeof (collectd_upsconn_t));
+    if (ups->conn == NULL)
+    {
+      ERROR ("nut plugin: malloc failed.");
+      return (-1);
+    }
+
+    status = upscli_connect (ups->conn, ups->hostname, ups->port,
+       UPSCLI_CONN_TRYSSL);
+    if (status != 0)
+    {
+      ERROR ("nut plugin: nut_read_one: upscli_connect (%s, %i) failed: %s",
+         ups->hostname, ups->port, upscli_strerror (ups->conn));
+      sfree (ups->conn);
+      return (-1);
+    }
+
+    INFO ("nut plugin: Connection to (%s, %i) established.",
+       ups->hostname, ups->port);
+  } /* if (ups->conn == NULL) */
+
   /* nut plugin: nut_read_one: upscli_list_start (adpos) failed: Protocol
    * error */
-  status = upscli_list_start (&ups->conn, query_num, query);
+  status = upscli_list_start (ups->conn, query_num, query);
   if (status != 0)
   {
     ERROR ("nut plugin: nut_read_one: upscli_list_start (%s) failed: %s",
-       ups->upsname, upscli_strerror (&ups->conn));
+       ups->upsname, upscli_strerror (ups->conn));
+    upscli_disconnect (ups->conn);
+    sfree (ups->conn);
     return (-1);
   }
 
-  while ((status = upscli_list_next (&ups->conn, query_num, query,
+  while ((status = upscli_list_next (ups->conn, query_num, query,
          &answer_num, &answer)) == 1)
   {
     char  *key;
@@ -330,7 +270,6 @@ static int nut_read (void)
 
   return ((success != 0) ? 0 : -1);
 } /* int nut_read */
-#endif /* NUT_HAVE_READ */
 
 static int nut_shutdown (void)
 {
@@ -348,28 +287,11 @@ static int nut_shutdown (void)
   return (0);
 } /* int nut_shutdown */
 
-void module_register (modreg_e load)
+void module_register (void)
 {
-  if (load & MR_DATASETS)
-  {
-    plugin_register_data_set (&ds_current);
-    plugin_register_data_set (&ds_humidity);
-    plugin_register_data_set (&ds_power);
-    plugin_register_data_set (&ds_voltage);
-    plugin_register_data_set (&ds_percent);
-    plugin_register_data_set (&ds_timeleft);
-    plugin_register_data_set (&ds_temperature);
-    plugin_register_data_set (&ds_frequency);
-  }
-
-#if NUT_HAVE_READ
-  if (load & MR_READ)
-  {
-    plugin_register_config ("nut", nut_config, config_keys, config_keys_num);
-    plugin_register_read ("nut", nut_read);
-    plugin_register_shutdown ("nut", nut_shutdown);
-  }
-#endif
+  plugin_register_config ("nut", nut_config, config_keys, config_keys_num);
+  plugin_register_read ("nut", nut_read);
+  plugin_register_shutdown ("nut", nut_shutdown);
 } /* void module_register */
 
 /* vim: set sw=2 ts=8 sts=2 tw=78 : */