X-Git-Url: https://git.tokkee.org/?a=blobdiff_plain;f=src%2Fapcups.c;h=b9a500d00f64769b08d66d77f8e467330daf84d8;hb=d82759c8f5904fc793c42b6637b90ddaa194a65c;hp=f481d0eb1ea36fcec5542702f10e15f1c5a9cdef;hpb=bfec4e6fa12d3ff5f3a4f36cc35b296abe2c04ad;p=collectd.git diff --git a/src/apcups.c b/src/apcups.c index f481d0eb..b9a500d0 100644 --- a/src/apcups.c +++ b/src/apcups.c @@ -22,112 +22,90 @@ * Anthony Gialluca **/ +/* + * FIXME: Don't know why but without this here atof() was not returning + * correct values for me. This is behavior that I don't understand and + * should be examined in closer detail. + */ +#include + #include "collectd.h" -#include "common.h" /* rrd_update_file */ -#include "plugin.h" /* plugin_register, plugin_submit */ -#include "configfile.h" /* cf_register */ - -/* FIXME: Check defines before including anything! */ -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include +#include "common.h" /* rrd_update_file */ +#include "plugin.h" /* plugin_register, plugin_submit */ +#include "configfile.h" /* cf_register */ +#include "utils_debug.h" #include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include -#include + +#if HAVE_SYS_TYPES_H +# include +#endif +#if HAVE_SYS_SOCKET_H +# include +#endif +#if HAVE_NETDB_H +# include +#endif + +#if HAVE_NETINET_IN_H +# include +#endif + +#ifndef APCMAIN +# define APCMAIN 0 +#endif #define NISPORT 3551 -#define _(String) (String) -#define N_(String) (String) #define MAXSTRING 256 -#define Error_abort0(fmd) generic_error_out(__FILE__, __LINE__, fmd) #define MODULE_NAME "apcups" - -/* Prototypes */ -static void generic_error_out(const char *, int , const char *, ...); - /* Default values for contacting daemon */ -static char *host = "localhost"; -static int port = NISPORT; - -static struct sockaddr_in tcp_serv_addr; /* socket information */ -static int net_errno = 0; /* error number -- not yet implemented */ -static char *net_errmsg = NULL; /* pointer to error message */ -static char net_errbuf[256]; /* error message buffer for messages */ +static char *global_host = "localhost"; +static int global_port = NISPORT; /* * The following are only if not compiled to test the module with its own main. */ -/* FIXME: Rename DSes to be more generic and follow established conventions. */ -#ifndef APCMAIN -static char *volt_file_template = "apcups_volt-%s.rrd"; -static char *volt_ds_def[] = -{ - "DS:linev:GAUGE:"COLLECTD_HEARTBEAT":0:250", - "DS:outputv:GAUGE:"COLLECTD_HEARTBEAT":0:250", - NULL -}; -static int volt_ds_num = 2; - -static char *bvolt_file_template = "apcups_bvolt-%s.rrd"; +#if !APCMAIN +static char *bvolt_file_template = "apcups/voltage-%s.rrd"; static char *bvolt_ds_def[] = { - "DS:battv:GAUGE:"COLLECTD_HEARTBEAT":0:100", + "DS:voltage:GAUGE:"COLLECTD_HEARTBEAT":0:U", }; static int bvolt_ds_num = 1; -static char *load_file_template = "apcups_load-%s.rrd"; +static char *load_file_template = "apcups/charge_percent.rrd"; static char *load_ds_def[] = { - "DS:loadpct:GAUGE:"COLLECTD_HEARTBEAT":0:120", + "DS:percent:GAUGE:"COLLECTD_HEARTBEAT":0:110", }; static int load_ds_num = 1; -static char *charge_file_template = "apcups_charge-%s.rrd"; +static char *charge_file_template = "apcups/charge.rrd"; static char *charge_ds_def[] = { - "DS:bcharge:GAUGE:"COLLECTD_HEARTBEAT":0:100", + "DS:charge:GAUGE:"COLLECTD_HEARTBEAT":0:U", }; static int charge_ds_num = 1; -static char *time_file_template = "apcups_time-%s.rrd"; +static char *time_file_template = "apcups/time.rrd"; static char *time_ds_def[] = { "DS:timeleft:GAUGE:"COLLECTD_HEARTBEAT":0:100", }; static int time_ds_num = 1; -static char *temp_file_template = "apcups_temp-%s.rrd"; +static char *temp_file_template = "apcups/temperature.rrd"; static char *temp_ds_def[] = { - "DS:itemp:GAUGE:"COLLECTD_HEARTBEAT":0:100", + /* -273.15 is absolute zero */ + "DS:temperature:GAUGE:"COLLECTD_HEARTBEAT":-274:U", }; static int temp_ds_num = 1; -static char *freq_file_template = "apcups_freq-%s.rrd"; +static char *freq_file_template = "apcups/frequency-%s.rrd"; static char *freq_ds_def[] = { - "DS:linefreq:GAUGE:"COLLECTD_HEARTBEAT":0:65", + "DS:frequency:GAUGE:"COLLECTD_HEARTBEAT":0:U", }; static int freq_ds_num = 1; @@ -139,102 +117,108 @@ static char *config_keys[] = }; static int config_keys_num = 2; -#endif /* ifndef APCMAIN */ +#endif /* if APCMAIN */ struct apc_detail_s { - float linev; - float loadpct; - float bcharge; - float timeleft; - float outputv; - float itemp; - float battv; - float linefreq; + double linev; + double loadpct; + double bcharge; + double timeleft; + double outputv; + double itemp; + double battv; + double linefreq; }; #define BIG_BUF 4096 -/* - * Subroutine normally called by macro error_abort() to print - * FATAL ERROR message and supplied error message - */ -static void generic_error_out(const char *file, int line, const char *fmt, ...) -{ - char buf[256]; - va_list arg_ptr; - int i; - - snprintf(buf, sizeof(buf), _("FATAL ERROR in %s at line %d\n"), file, line); - i = strlen(buf); - va_start(arg_ptr, fmt); - vsnprintf((char *)&buf[i], sizeof(buf) - i, (char *)fmt, arg_ptr); - va_end(arg_ptr); - fprintf(stdout, "%s", buf); - - exit(1); -} - /* * Read nbytes from the network. * It is possible that the total bytes require in several * read requests */ -static int read_nbytes(int fd, char *ptr, int nbytes) +static int read_nbytes (int *fd, char *ptr, int nbytes) { - int nleft, nread; + int nleft; + int nread; nleft = nbytes; + nread = -1; + + assert (*fd >= 0); + + while ((nleft > 0) && (nread != 0)) + { + nread = read (*fd, ptr, nleft); - while (nleft > 0) { - do { - nread = read(fd, ptr, nleft); - } while (nread == -1 && (errno == EINTR || errno == EAGAIN)); + if (nread == -1 && (errno == EINTR || errno == EAGAIN)) + continue; - if (nread <= 0) { - net_errno = errno; - return (nread); /* error, or EOF */ + if (nread == -1) + { + *fd = -1; + syslog (LOG_ERR, "apcups plugin: write failed: %s", strerror (errno)); + return (-1); } nleft -= nread; ptr += nread; } - return (nbytes - nleft); /* return >= 0 */ + return (nbytes - nleft); } /* * Write nbytes to the network. * It may require several writes. */ -static int write_nbytes(int fd, char *ptr, int nbytes) +static int write_nbytes (int *fd, void *buf, int buflen) { - int nleft, nwritten; + int nleft; + int nwritten; + char *ptr; - nleft = nbytes; - while (nleft > 0) { - nwritten = write(fd, ptr, nleft); + assert (buflen > 0); + assert (*fd >= 0); + + ptr = (char *) buf; + + nleft = buflen; + while (nleft > 0) + { + nwritten = write (*fd, ptr, nleft); + + if ((nwritten == -1) && ((errno == EAGAIN) || (errno == EINTR))) + continue; - if (nwritten <= 0) { - net_errno = errno; - return (nwritten); /* error */ + if (nwritten == -1) + { + syslog (LOG_ERR, "Writing to socket failed: %s", strerror (errno)); + *fd = -1; + return (-1); } nleft -= nwritten; ptr += nwritten; } - return (nbytes - nleft); + /* If we get here, (nleft <= 0) is true */ + return (buflen); } /* Close the network connection */ -void net_close (int sockfd) +static void net_close (int *fd) { short pktsiz = 0; + assert (*fd >= 0); + /* send EOF sentinel */ - write_nbytes(sockfd, (char *) &pktsiz, sizeof(short)); - close (sockfd); + write_nbytes (fd, &pktsiz, sizeof (short)); + + close (*fd); + *fd = -1; } @@ -243,63 +227,62 @@ void net_close (int sockfd) * Returns -1 on error * Returns socket file descriptor otherwise */ -int net_open(char *host, char *service, int port) -{ - int sockfd; - struct hostent *hp; - unsigned int inaddr; /* Careful here to use unsigned int for */ - /* compatibility with Alpha */ - - /* - * Fill in the structure serv_addr with the address of the server that - * we want to connect with. - */ - memset((char *)&tcp_serv_addr, 0, sizeof(tcp_serv_addr)); - tcp_serv_addr.sin_family = AF_INET; - tcp_serv_addr.sin_port = htons(port); - - if ((inaddr = inet_addr(host)) != INADDR_NONE) { - tcp_serv_addr.sin_addr.s_addr = inaddr; - } else { - if ((hp = gethostbyname(host)) == NULL) { - net_errmsg = "tcp_open: hostname error\n"; - return -1; - } - - if (hp->h_length != sizeof(inaddr) || hp->h_addrtype != AF_INET) { - net_errmsg = "tcp_open: funny gethostbyname value\n"; - return -1; - } +static int net_open (char *host, char *service, int port) +{ + int sd; + int status; + char port_str[8]; + struct addrinfo ai_hints; + struct addrinfo *ai_return; + struct addrinfo *ai_list; + + assert ((port > 0x00000000) && (port <= 0x0000FFFF)); + + /* Convert the port to a string */ + snprintf (port_str, 8, "%i", port); + port_str[7] = '\0'; + + /* Resolve name */ + memset ((void *) &ai_hints, '\0', sizeof (ai_hints)); + ai_hints.ai_family = AF_INET; /* XXX: Change this to `AF_UNSPEC' if apcupsd can handle IPv6 */ + ai_hints.ai_socktype = SOCK_STREAM; + + status = getaddrinfo (host, port_str, &ai_hints, &ai_return); + if (status != 0) + { + DBG ("getaddrinfo failed: %s", status == EAI_SYSTEM ? strerror (errno) : gai_strerror (status)); + return (-1); + } - tcp_serv_addr.sin_addr.s_addr = *(unsigned int *)hp->h_addr; + /* Create socket */ + sd = -1; + for (ai_list = ai_return; ai_list != NULL; ai_list = ai_list->ai_next) + { + sd = socket (ai_list->ai_family, ai_list->ai_socktype, ai_list->ai_protocol); + if (sd >= 0) + break; } + /* `ai_list' still holds the current description of the socket.. */ - /* Open a TCP socket */ - if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) { - net_errmsg = "tcp_open: cannot open stream socket\n"; - return -1; + if (sd < 0) + { + DBG ("Unable to open a socket"); + freeaddrinfo (ai_return); + return (-1); } - /* connect to server */ -#if defined HAVE_OPENBSD_OS || defined HAVE_FREEBSD_OS - /* - * Work around a bug in OpenBSD & FreeBSD userspace pthreads - * implementations. Rationale is the same as described above. - */ - fcntl(sockfd, F_SETFL, fcntl(sockfd, F_GETFL)); -#endif + status = connect (sd, ai_list->ai_addr, ai_list->ai_addrlen); - if (connect(sockfd, (struct sockaddr *)&tcp_serv_addr, sizeof(tcp_serv_addr)) < 0) { - snprintf(net_errbuf, sizeof(net_errbuf), - _("tcp_open: cannot connect to server %s on port %d.\n" - "ERR=%s\n"), host, port, strerror(errno)); - net_errmsg = net_errbuf; - close(sockfd); - return -1; + freeaddrinfo (ai_return); + + if (status != 0) /* `connect(2)' failed */ + { + DBG ("connect failed: %s", strerror (errno)); + return (-1); } - return sockfd; -} /* int net_open(char *host, char *service, int port) */ + return (sd); +} /* int net_open (char *host, char *service, int port) */ /* * Receive a message from the other end. Each message consists of @@ -310,433 +293,331 @@ int net_open(char *host, char *service, int port) * Returns -1 on hard end of file (i.e. network connection close) * Returns -2 on error */ -int net_recv(int sockfd, char *buff, int maxlen) +static int net_recv (int *sockfd, char *buf, int buflen) { - int nbytes; - short pktsiz; - - /* get data size -- in short */ - if ((nbytes = read_nbytes(sockfd, (char *)&pktsiz, sizeof(short))) <= 0) { - /* probably pipe broken because client died */ - return -1; /* assume hard EOF received */ - } - if (nbytes != sizeof(short)) - return -2; - - pktsiz = ntohs(pktsiz); /* decode no. of bytes that follow */ - if (pktsiz > maxlen) { - net_errmsg = "net_recv: record length too large\n"; - return -2; - } - if (pktsiz == 0) - return 0; /* soft EOF */ - - /* now read the actual data */ - if ((nbytes = read_nbytes(sockfd, buff, pktsiz)) <= 0) { - net_errmsg = "net_recv: read_nbytes error\n"; - return -2; - } - if (nbytes != pktsiz) { - net_errmsg = "net_recv: error in read_nbytes\n"; - return -2; - } - - return (nbytes); /* return actual length of message */ -} + int nbytes; + short pktsiz; + + /* get data size -- in short */ + if ((nbytes = read_nbytes (sockfd, (char *) &pktsiz, sizeof (short))) <= 0) + return (-1); + + if (nbytes != sizeof (short)) + return (-2); + + pktsiz = ntohs (pktsiz); + if (pktsiz > buflen) + { + DBG ("record length too large"); + return (-2); + } + + if (pktsiz == 0) + return (0); + + /* now read the actual data */ + if ((nbytes = read_nbytes (sockfd, buf, pktsiz)) <= 0) + return (-2); + + if (nbytes != pktsiz) + return (-2); + + return (nbytes); +} /* static int net_recv (int sockfd, char *buf, int buflen) */ /* * Send a message over the network. The send consists of * two network packets. The first is sends a short containing * the length of the data packet which follows. - * Returns number of bytes sent - * Returns -1 on error + * Returns zero on success + * Returns non-zero on error */ -int net_send(int sockfd, char *buff, int len) +static int net_send (int *sockfd, char *buff, int len) { - int rc; - short pktsiz; - - /* send short containing size of data packet */ - pktsiz = htons((short)len); - rc = write_nbytes(sockfd, (char *)&pktsiz, sizeof(short)); - if (rc != sizeof(short)) { - net_errmsg = "net_send: write_nbytes error of length prefix\n"; - return -1; - } - - /* send data packet */ - rc = write_nbytes(sockfd, buff, len); - if (rc != len) { - net_errmsg = "net_send: write_nbytes error\n"; - return -1; - } - - return rc; -} + int rc; + short packet_size; + assert (len > 0); -/* Get and print status from apcupsd NIS server */ -static void do_pthreads_status(char *host, int port, struct apc_detail_s *apcups_detail) -{ - int sockfd, n; - char recvline[MAXSTRING + 1]; - char *tokptr; + /* send short containing size of data packet */ + packet_size = htons ((short) len); - if ((sockfd = net_open(host, NULL, port)) < 0) - Error_abort0(net_errmsg); - - net_send(sockfd, "status", 6); - - while ((n = net_recv(sockfd, recvline, sizeof(recvline))) > 0) { - recvline[n] = 0; -#ifdef APCMAIN - fputs(recvline, stdout); - int printit = 1; -#else - int printit = 0; -#endif /* ifdef APCMAIN */ - - tokptr = strtok(recvline,":"); - while(tokptr != NULL) { - /* Look for Limit_Add */ - if(strncmp("LINEV",tokptr,5) == 0) { - if(printit) fprintf(stdout," Found LINEV.\n"); - tokptr = strtok(NULL," \t"); - if(tokptr == NULL) continue; - if(printit) fprintf(stdout," Value %s.\n",tokptr); - apcups_detail->linev = atof (tokptr); - }else if(strncmp("BATTV",tokptr,5) == 0) { - if(printit) fprintf(stdout," Found BATTV.\n"); - tokptr = strtok(NULL," \t"); - if(tokptr == NULL) continue; - if(printit) fprintf(stdout," Value %s.\n",tokptr); - apcups_detail->battv = atof (tokptr); - }else if(strncmp("ITEMP",tokptr,5) == 0) { - if(printit) fprintf(stdout," Found ITEMP.\n"); - tokptr = strtok(NULL," \t"); - if(tokptr == NULL) continue; - if(printit) fprintf(stdout," Value %s.\n",tokptr); - apcups_detail->itemp = atof (tokptr); - }else if(strncmp("LOADPCT",tokptr,7) == 0) { - if(printit) fprintf(stdout," Found LOADPCT.\n"); - tokptr = strtok(NULL," \t"); - if(tokptr == NULL) continue; - if(printit) fprintf(stdout," Value %s.\n",tokptr); - apcups_detail->loadpct = atof (tokptr); - }else if(strncmp("BCHARGE",tokptr,7) == 0) { - if(printit) fprintf(stdout," Found BCHARGE.\n"); - tokptr = strtok(NULL," \t"); - if(tokptr == NULL) continue; - if(printit) fprintf(stdout," Value %s.\n",tokptr); - apcups_detail->bcharge = atof (tokptr); - }else if(strncmp("OUTPUTV",tokptr,7) == 0) { - if(printit) fprintf(stdout," Found OUTPUTV.\n"); - tokptr = strtok(NULL," \t"); - if(tokptr == NULL) continue; - if(printit) fprintf(stdout," Value %s.\n",tokptr); - apcups_detail->outputv = atof (tokptr); - }else if(strncmp("LINEFREQ",tokptr,8) == 0) { - if(printit) fprintf(stdout," Found LINEFREQ.\n"); - tokptr = strtok(NULL," \t"); - if(tokptr == NULL) continue; - if(printit) fprintf(stdout," Value %s.\n",tokptr); - apcups_detail->linefreq = atof (tokptr); - }else if(strncmp("TIMELEFT",tokptr,8) == 0) { - if(printit) fprintf(stdout," Found TIMELEFT.\n"); - tokptr = strtok(NULL," \t"); - if(tokptr == NULL) continue; - if(printit) fprintf(stdout," Value %s.\n",tokptr); - apcups_detail->timeleft = atof (tokptr); - } /* */ - tokptr = strtok(NULL,":"); - } - } - - if (n < 0) - Error_abort0(net_errmsg); - - net_close(sockfd); + rc = write_nbytes (sockfd, &packet_size, sizeof (packet_size)); + if (rc != sizeof (packet_size)) + return (-1); + + /* send data packet */ + rc = write_nbytes (sockfd, buff, len); + if (rc != len) + return (-1); + + return (0); } -#ifdef APCMAIN -int main(int argc, char **argv) +/* Get and print status from apcupsd NIS server */ +static int apc_query_server (char *host, int port, + struct apc_detail_s *apcups_detail) { - /* we are not really going to use this */ - struct apc_detail_s apcups_detail; + int n; + char recvline[1024]; + char *tokptr; + char *key; + double value; - if (!*host || strcmp(host, "0.0.0.0") == 0) - host = "localhost"; - - do_pthreads_status(host, port, &apcups_detail); - - return 0; -} + static int sockfd = -1; + static unsigned int complain = 0; + +#if APCMAIN +# define PRINT_VALUE(name, val) printf(" Found property: name = %s; value = %f;\n", name, val) #else -static void apcups_init (void) -{ - return; -} +# define PRINT_VALUE(name, val) /**/ +#endif -static int apcups_config (char *key, char *value) -{ - static char lhost[126]; - - if (strcasecmp (key, "host") == 0) - { - lhost[0] = '\0'; - strcpy(lhost,key); - host = lhost; - } - else if (strcasecmp (key, "Port") == 0) - { - int port_tmp = atoi (value); - if(port_tmp < 1 || port_tmp > 65535) { - syslog (LOG_WARNING, "apcups: `port' failed: %s", - value); - return (1); - } else { - port = port_tmp; - } - } - else - { - return (-1); - } - return(0); -} + if (sockfd < 0) + { + if ((sockfd = net_open (host, NULL, port)) < 0) + { + /* Complain first time and once every six hours. */ + int complain_step = 21600 / atoi (COLLECTD_STEP); + + if (complain == 0 || (complain % complain_step) == 0) + syslog (LOG_ERR, "apcups plugin: Connecting to the apcupsd failed."); + complain++; + + return (-1); + } else { + if(complain > 1) + syslog (LOG_ERR, "apcups plugin: Connection re-established to the apcupsd."); + } + complain = 0; + } -#define BUFSIZE 256 -static void apcups_submit (char *host, - struct apc_detail_s *apcups_detail) -{ - char buf[BUFSIZE]; + if (net_send (&sockfd, "status", 6) < 0) + { + syslog (LOG_ERR, "apcups plugin: Writing to the socket failed."); + return (-1); + } - if (snprintf (buf, BUFSIZE, "%u:%f:%f", - (unsigned int) curtime, - apcups_detail->linev, - apcups_detail->outputv) >= BUFSIZE) - return; + /* XXX: Do we read `n' or `n-1' bytes? */ + while ((n = net_recv (&sockfd, recvline, sizeof (recvline) - 1)) > 0) + { + assert (n < sizeof (recvline)); + recvline[n] = '\0'; +#if APCMAIN + printf ("net_recv = `%s';\n", recvline); +#endif /* if APCMAIN */ + + tokptr = strtok (recvline, ":"); + while (tokptr != NULL) + { + key = tokptr; + if ((tokptr = strtok (NULL, " \t")) == NULL) + continue; + value = atof (tokptr); + PRINT_VALUE (key, value); + + if (strcmp ("LINEV", key) == 0) + apcups_detail->linev = value; + else if (strcmp ("BATTV", tokptr) == 0) + apcups_detail->battv = value; + else if (strcmp ("ITEMP", tokptr) == 0) + apcups_detail->itemp = value; + else if (strcmp ("LOADPCT", tokptr) == 0) + apcups_detail->loadpct = value; + else if (strcmp ("BCHARGE", tokptr) == 0) + apcups_detail->bcharge = value; + else if (strcmp ("OUTPUTV", tokptr) == 0) + apcups_detail->outputv = value; + else if (strcmp ("LINEFREQ", tokptr) == 0) + apcups_detail->linefreq = value; + else if (strcmp ("TIMELEFT", tokptr) == 0) + apcups_detail->timeleft = value; + + tokptr = strtok (NULL, ":"); + } /* while (tokptr != NULL) */ + } - plugin_submit (MODULE_NAME, host, buf); -} + if (n < 0) + { + syslog (LOG_WARNING, "apcups plugin: Error reading from socket"); + return (-1); + } else { + /* close the opened socket */ + net_close(&sockfd); + } -static void apc_bvolt_submit (char *host, - struct apc_detail_s *apcups_detail) -{ - char buf[BUFSIZE]; - if (snprintf (buf, BUFSIZE, "%u:%f", - (unsigned int) curtime, - apcups_detail->battv) >= BUFSIZE) - return; - - plugin_submit ("apcups_bvolt", host, buf); + return (0); } -static void apc_load_submit (char *host, - struct apc_detail_s *apcups_detail) +#if APCMAIN +/* + * This is used for testing apcups in a standalone mode. + * Usefull for debugging. + */ +int main (int argc, char **argv) { - char buf[BUFSIZE]; + /* we are not really going to use this */ + struct apc_detail_s apcups_detail; - if (snprintf (buf, BUFSIZE, "%u:%f", - (unsigned int) curtime, - apcups_detail->loadpct) >= BUFSIZE) - return; - - plugin_submit ("apcups_load", host, buf); -} + openlog ("apcups", LOG_PID | LOG_NDELAY | LOG_LOCAL1, LOG_USER); -static void apc_charge_submit (char *host, - struct apc_detail_s *apcups_detail) -{ - char buf[BUFSIZE]; + if (global_host == NULL || strcmp (global_host, "0.0.0.0") == 0) + global_host = "localhost"; - if (snprintf (buf, BUFSIZE, "%u:%f", - (unsigned int) curtime, - apcups_detail->bcharge) >= BUFSIZE) - return; - - plugin_submit ("apcups_charge", host, buf); -} + if(apc_query_server (global_host, global_port, &apcups_detail) < 0) + { + printf("apcups: Failed...\n"); + return(-1); + } -static void apc_temp_submit (char *host, - struct apc_detail_s *apcups_detail) + return 0; +} +#else +static int apcups_config (char *key, char *value) { - char buf[BUFSIZE]; - - if (snprintf (buf, BUFSIZE, "%u:%f", - (unsigned int) curtime, - apcups_detail->itemp) >= BUFSIZE) - return; - - plugin_submit ("apcups_temp", host, buf); + if (strcasecmp (key, "host") == 0) + { + if (global_host != NULL) + { + free (global_host); + global_host = NULL; + } + if ((global_host = strdup (value)) == NULL) + return (1); + } + else if (strcasecmp (key, "Port") == 0) + { + int port_tmp = atoi (value); + if (port_tmp < 1 || port_tmp > 65535) + { + syslog (LOG_WARNING, "apcups plugin: Invalid port: %i", port_tmp); + return (1); + } + global_port = port_tmp; + } + else + { + return (-1); + } + return (0); } -static void apc_time_submit (char *host, - struct apc_detail_s *apcups_detail) +static void apcups_init (void) { - char buf[BUFSIZE]; - - if (snprintf (buf, BUFSIZE, "%u:%f", - (unsigned int) curtime, - apcups_detail->timeleft) >= BUFSIZE) - return; - - plugin_submit ("apcups_time", host, buf); + return; } -static void apc_freq_submit (char *host, - struct apc_detail_s *apcups_detail) +static void apc_write_voltage (char *host, char *inst, char *val) { - char buf[BUFSIZE]; + char file[512]; + int status; - if (snprintf (buf, BUFSIZE, "%u:%f", - (unsigned int) curtime, - apcups_detail->linefreq) >= BUFSIZE) - return; - - plugin_submit ("apcups_freq", host, buf); + status = snprintf (file, 512, bvolt_file_template, inst); + if ((status < 1) || (status >= 512)) + return; + + rrd_update_file (host, file, val, bvolt_ds_def, bvolt_ds_num); } -#undef BUFSIZE -static void apcups_read (void) +static void apc_write_charge (char *host, char *inst, char *val) { - struct apc_detail_s apcups_detail; - - apcups_detail.linev = 0.0; - apcups_detail.loadpct = 0.0; - apcups_detail.bcharge = 0.0; - apcups_detail.timeleft = 0.0; - apcups_detail.outputv = 0.0; - apcups_detail.itemp = 0.0; - apcups_detail.battv = 0.0; - apcups_detail.linefreq = 0.0; - - - if (!*host || strcmp(host, "0.0.0.0") == 0) - host = "localhost"; - - do_pthreads_status(host, port, &apcups_detail); - - apcups_submit (host, &apcups_detail); - apc_bvolt_submit (host, &apcups_detail); - apc_load_submit (host, &apcups_detail); - apc_charge_submit (host, &apcups_detail); - apc_temp_submit (host, &apcups_detail); - apc_time_submit (host, &apcups_detail); - apc_freq_submit (host, &apcups_detail); + rrd_update_file (host, charge_file_template, val, charge_ds_def, charge_ds_num); } - -static void apcups_write (char *host, char *inst, char *val) +static void apc_write_percent (char *host, char *inst, char *val) { - char file[512]; - int status; - - status = snprintf (file, 512, volt_file_template, inst); - if (status < 1) - return; - else if (status >= 512) - return; - - rrd_update_file (host, file, val, volt_ds_def, volt_ds_num); + rrd_update_file (host, load_file_template, val, load_ds_def, load_ds_num); } -static void apc_bvolt_write (char *host, char *inst, char *val) +static void apc_write_timeleft (char *host, char *inst, char *val) { - char file[512]; - int status; - - status = snprintf (file, 512, bvolt_file_template, inst); - if (status < 1) - return; - else if (status >= 512) - return; - - rrd_update_file (host, file, val, bvolt_ds_def, bvolt_ds_num); + rrd_update_file (host, time_file_template, val, time_ds_def, time_ds_num); } -static void apc_load_write (char *host, char *inst, char *val) +static void apc_write_temperature (char *host, char *inst, char *val) { - char file[512]; - int status; - - status = snprintf (file, 512, load_file_template, inst); - if (status < 1) - return; - else if (status >= 512) - return; - - rrd_update_file (host, file, val, load_ds_def, load_ds_num); + rrd_update_file (host, temp_file_template, val, temp_ds_def, temp_ds_num); } -static void apc_charge_write (char *host, char *inst, char *val) +static void apc_write_frequency (char *host, char *inst, char *val) { - char file[512]; - int status; - - status = snprintf (file, 512, charge_file_template, inst); - if (status < 1) - return; - else if (status >= 512) - return; - - rrd_update_file (host, file, val, charge_ds_def, charge_ds_num); + char file[512]; + int status; + + status = snprintf (file, 512, freq_file_template, inst); + if ((status < 1) || (status >= 512)) + return; + + rrd_update_file (host, file, val, freq_ds_def, freq_ds_num); } -static void apc_temp_write (char *host, char *inst, char *val) +static void apc_submit_generic (char *type, char *inst, + double value) { - char file[512]; - int status; - - status = snprintf (file, 512, temp_file_template, inst); - if (status < 1) - return; - else if (status >= 512) - return; - - rrd_update_file (host, file, val, temp_ds_def, temp_ds_num); + char buf[512]; + int status; + + status = snprintf (buf, 512, "%u:%f", + (unsigned int) curtime, value); + if ((status < 1) || (status >= 512)) + return; + + plugin_submit (type, inst, buf); } -static void apc_time_write (char *host, char *inst, char *val) +static void apc_submit (struct apc_detail_s *apcups_detail) { - char file[512]; - int status; - - status = snprintf (file, 512, time_file_template, inst); - if (status < 1) - return; - else if (status >= 512) - return; - - rrd_update_file (host, file, val, time_ds_def, time_ds_num); + apc_submit_generic ("apcups_voltage", "input", apcups_detail->linev); + apc_submit_generic ("apcups_voltage", "output", apcups_detail->outputv); + apc_submit_generic ("apcups_voltage", "battery", apcups_detail->battv); + apc_submit_generic ("apcups_charge", "-", apcups_detail->bcharge); + apc_submit_generic ("apcups_charge_pct", "-", apcups_detail->loadpct); + apc_submit_generic ("apcups_timeleft", "-", apcups_detail->timeleft); + apc_submit_generic ("apcups_temp", "-", apcups_detail->itemp); + apc_submit_generic ("apcups_frequency", "input", apcups_detail->linefreq); } -static void apc_freq_write (char *host, char *inst, char *val) +static void apcups_read (void) { - char file[512]; - int status; - - status = snprintf (file, 512, freq_file_template, inst); - if (status < 1) - return; - else if (status >= 512) - return; + struct apc_detail_s apcups_detail; + int status; + + if (global_host == NULL) + return; + + apcups_detail.linev = -1.0; + apcups_detail.outputv = -1.0; + apcups_detail.battv = -1.0; + apcups_detail.loadpct = -1.0; + apcups_detail.bcharge = -1.0; + apcups_detail.timeleft = -1.0; + apcups_detail.itemp = -300.0; + apcups_detail.linefreq = -1.0; - rrd_update_file (host, file, val, freq_ds_def, freq_ds_num); -} + status = apc_query_server (global_host, global_port, &apcups_detail); + + /* + * if we did not connect then do not bother submitting + * zeros. We want rrd files to have NAN. + */ + if (status != 0) + return; + + apc_submit (&apcups_detail); +} /* apcups_read */ void module_register (void) { - plugin_register (MODULE_NAME, apcups_init, apcups_read, apcups_write); - plugin_register ("apcups_bvolt", NULL, NULL, apc_bvolt_write); - plugin_register ("apcups_load", NULL, NULL, apc_load_write); - plugin_register ("apcups_charge", NULL, NULL, apc_charge_write); - plugin_register ("apcups_temp", NULL, NULL, apc_temp_write); - plugin_register ("apcups_time", NULL, NULL, apc_time_write); - plugin_register ("apcups_freq", NULL, NULL, apc_freq_write); + plugin_register (MODULE_NAME, apcups_init, apcups_read, NULL); + plugin_register ("apcups_voltage", NULL, NULL, apc_write_voltage); + plugin_register ("apcups_charge", NULL, NULL, apc_write_charge); + plugin_register ("apcups_charge_pct", NULL, NULL, apc_write_percent); + plugin_register ("apcups_timeleft", NULL, NULL, apc_write_timeleft); + plugin_register ("apcups_temp", NULL, NULL, apc_write_temperature); + plugin_register ("apcups_frequency", NULL, NULL, apc_write_frequency); cf_register (MODULE_NAME, apcups_config, config_keys, config_keys_num); } -#endif /* ifdef APCMAIN */ +#endif /* if APCMAIN */ #undef MODULE_NAME