X-Git-Url: https://git.verplant.org/?a=blobdiff_plain;f=src%2Fnut.c;h=edc48c61510ab6e716abe17b52d222136ee7ceba;hb=1ebf2f31bd2e080e6f42de640f0a3899a61501c0;hp=d839ba49a460b9bcde35e41f23008e961d25d432;hpb=896ac2332bbbb8f7aa09c977adc101ed81dfd126;p=collectd.git diff --git a/src/nut.c b/src/nut.c index d839ba49..9540d7c4 100644 --- a/src/nut.c +++ b/src/nut.c @@ -1,125 +1,49 @@ /** * 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 + * Florian octo Forster **/ #include "collectd.h" #include "common.h" #include "plugin.h" -#if HAVE_PTHREAD_H -# include -#endif +#include +#include -#if HAVE_UPSCLIENT_H -# include -# 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) @@ -152,7 +80,7 @@ static int nut_add_ups (const char *name) DEBUG ("nut plugin: nut_add_ups (name = %s);", name); - ups = (nut_ups_t *) malloc (sizeof (nut_ups_t)); + ups = malloc (sizeof (*ups)); if (ups == NULL) { ERROR ("nut plugin: nut_add_ups: malloc failed."); @@ -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 = malloc (sizeof (*ups->conn)); + 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; @@ -346,30 +286,12 @@ static int nut_shutdown (void) return (0); } /* int nut_shutdown */ -#endif /* NUT_HAVE_READ */ -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 : */