rrd based system stats
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

212 lines
5.2 KiB

/***************************************************************************
* Copyright (C) 12/2012 by Olaf Rempel *
* razzor@kopf-tisch.de *
* *
* 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; either version 2 of the License, or *
* (at your option) any later version. *
* *
* 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. *
* *
* 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., *
* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
***************************************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/un.h>
#include "configfile.h"
#include "helper.h"
#include "list.h"
#include "logging.h"
#include "plugins.h"
#include "probe.h"
#include "sockaddr.h"
#define RESP_SIZE 1024
struct sammler_plugin plugin;
struct server_entry {
struct list_head list;
char *name;
int errors;
struct sockaddr_in sa;
};
static LIST_HEAD(server_list);
static char *resp_buf;
static const char *ds_def = {
"DS:temperature:GAUGE:90:0:100"
};
static const char * get_ds(int ds_id)
{
return ds_def;
}
static int probe(void)
{
struct server_entry *entry;
list_for_each_entry(entry, &server_list, list) {
int sock = socket(AF_INET, SOCK_STREAM, 0);
if (sock == -1) {
log_print(LOG_WARN, "%s: socket()", plugin.name);
continue;
}
int ret = connect(sock, (struct sockaddr *)&entry->sa, sizeof(entry->sa));
if (ret != 0) {
if (entry->errors++ == 0)
log_print(LOG_ERROR, "%s: connect(%s)", plugin.name, get_sockaddr_buf(&entry->sa));
close(sock);
continue;
}
int pos = 0;
int len;
do {
len = read(sock, resp_buf + pos, RESP_SIZE - pos);
if ((pos == 0) && (len <= 0)) {
if (entry->errors++ == 0)
log_print(LOG_ERROR, "%s: read()", plugin.name);
break;
}
pos += len;
} while (len > 0);
close(sock);
if (pos == 0)
continue;
resp_buf[pos] = '\0';
/* '|/dev/sda|OCZ-VERTEX3|128|C||/dev/sdb|HDT722525DLA380|SLP|*||/dev/sdc|Generic STORAGE DEVICE|NA|*|' */
char *tok, *tmp = NULL, *resp = resp_buf;
int i = 0;
char *device = NULL;
int temperature = 0;
while ((tok = strtok_r(resp, "|", &tmp)) != NULL) {
if ((i % 4) == 0) {
device = strrchr(tok, '/');
if (device != NULL && *(device +1) == '\0')
device = NULL;
else
device++;
} else if ((i % 4) == 2) {
char *tmp2;
temperature = strtol(tok, &tmp2, 10);
if (*tmp2 != '\0')
device = NULL;
} else if ((i % 4) == 3) {
if ((tok[0] == 'C') && (device != NULL)) {
char filename[32];
len = snprintf(filename, sizeof(filename), "%s-%s.rrd", plugin.name, device);
if (len < 0 || len >= sizeof(filename))
continue;
probe_submit(&plugin, filename, 0, "%d", temperature);
}
device = NULL;
temperature = 0;
}
i++;
resp = NULL;
}
if (entry->errors > 0) {
log_print(LOG_ERROR, "%s: success (%s) after %d errors",
plugin.name, entry->name, entry->errors);
entry->errors = 0;
}
}
return 0;
}
static int init_cb(struct strtoken *tokens, void *privdata)
{
if (tokens->count != 2) {
log_print(LOG_ERROR, "%s: parse error", plugin.name);
return -1;
}
struct server_entry *entry = malloc(sizeof(struct server_entry));
if (entry == NULL) {
log_print(LOG_ERROR, "%s: out of memory", plugin.name);
return -1;
}
entry->name = strdup(tokens->field[0]);
entry->errors = 0;
if (parse_sockaddr(tokens->field[1], &entry->sa) < 0) {
log_print(LOG_ERROR, "%s: invalid address: <%s>", plugin.name, tokens->field[1]);
free(entry->name);
free(entry);
return -1;
}
log_print(LOG_INFO, "%s: added server '%s'", plugin.name, entry->name);
list_add_tail(&entry->list, &server_list);
return 0;
}
static int init(void)
{
resp_buf = malloc(RESP_SIZE);
if (resp_buf == NULL) {
log_print(LOG_ERROR, "%s: out of memory", plugin.name);
return -1;
}
config_get_strtokens("p_hddtemp", "server", ",", 2, init_cb, NULL);
return 0;
}
static int fini(void)
{
struct server_entry *entry, *tmp;
list_for_each_entry_safe(entry, tmp, &server_list, list) {
free(entry->name);
free(entry);
}
free(resp_buf);
return 0;
}
struct sammler_plugin plugin = {
.name = "hddtemp",
.interval = 60,
.init = &init,
.fini = &fini,
.probe = &probe,
.get_ds = &get_ds,
};