4 changed files with 233 additions and 3 deletions
Split View
Diff Options
-
+1 -1Makefile
-
+215 -0plugins/ts2.c
-
+5 -1sammler.conf
-
+12 -1sammler.php
+ 1
- 1
Makefile
View File
+ 215
- 0
plugins/ts2.c
View File
| @ -0,0 +1,215 @@ | |||
| /*************************************************************************** | |||
| * Copyright (C) 02/2010 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 TS2_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 *ts2_resp_buf; | |||
| static const char *ds_def = { | |||
| "DS:users:GAUGE:90:0:U " | |||
| "DS:channels:GAUGE:90:0:U " | |||
| }; | |||
| 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, "plugin ts2: socket()"); | |||
| continue; | |||
| } | |||
| int ret = connect(sock, (struct sockaddr *)&entry->sa, sizeof(entry->sa)); | |||
| if (ret != 0) { | |||
| if (entry->errors++ == 0) | |||
| log_print(LOG_ERROR, "plugin ts2: connect(%s)", get_sockaddr_buf(&entry->sa)); | |||
| close(sock); | |||
| continue; | |||
| } | |||
| /* read prompt "[TS]" */ | |||
| int len = read(sock, ts2_resp_buf, TS2_RESP_SIZE); | |||
| if (len <= 0) { | |||
| if (entry->errors++ == 0) | |||
| log_print(LOG_ERROR, "ts2: read(1)"); | |||
| close(sock); | |||
| continue; | |||
| } | |||
| if (strncmp(ts2_resp_buf, "[TS]", 4) != 0) { | |||
| if (entry->errors++ == 0) | |||
| log_print(LOG_ERROR, "ts2: invalid prompt from server"); | |||
| close(sock); | |||
| continue; | |||
| } | |||
| /* query "general informations" */ | |||
| char cmd1[] = "gi\n"; | |||
| write(sock, cmd1, sizeof(cmd1)); | |||
| len = read(sock, ts2_resp_buf, TS2_RESP_SIZE); | |||
| if (len <= 0) { | |||
| if (entry->errors++ == 0) | |||
| log_print(LOG_ERROR, "ts2: read(2)"); | |||
| close(sock); | |||
| continue; | |||
| } | |||
| /* send quit (but we close the socket anyway..) */ | |||
| char cmd2[] = "quit"; | |||
| write(sock, cmd2, sizeof(cmd2)); | |||
| close(sock); | |||
| /* terminate response */ | |||
| ts2_resp_buf[len] = '\0'; | |||
| char *tok, *tmp, *resp = ts2_resp_buf; | |||
| int users = 0, channels = 0; | |||
| while ((tok = strtok_r(resp, "\r\n", &tmp)) != NULL) { | |||
| /* total_users_online=14 */ | |||
| if (strncmp(tok, "total_users_online", 18) == 0) { | |||
| users = atoi(tok +19); | |||
| /* total_channels=3 */ | |||
| } else if (strncmp(tok, "total_channels", 14) == 0) { | |||
| channels = atoi(tok +15); | |||
| } | |||
| resp = NULL; | |||
| } | |||
| char filename[32]; | |||
| len = snprintf(filename, sizeof(filename), "ts2-%s.rrd", entry->name); | |||
| if (len < 0 || len >= sizeof(filename)) | |||
| continue; | |||
| probe_submit(&plugin, filename, 0, "%d:%d", users, channels); | |||
| if (entry->errors > 0) { | |||
| log_print(LOG_ERROR, "plugin ts2: success (%s) after %d errors", | |||
| entry->name, entry->errors); | |||
| entry->errors = 0; | |||
| } | |||
| } | |||
| return 0; | |||
| } | |||
| static int init_cb(const char *parameter, void *privdata) | |||
| { | |||
| char *part[2]; | |||
| int ret = strsplit((char *)parameter, ",", part, 2); | |||
| if (ret < 2) { | |||
| log_print(LOG_ERROR, "p_ts2: parse error"); | |||
| return -1; | |||
| } | |||
| struct server_entry *entry = malloc(sizeof(struct server_entry)); | |||
| if (entry == NULL) { | |||
| log_print(LOG_ERROR, "p_ts2: out of memory"); | |||
| return -1; | |||
| } | |||
| entry->name = strdup(part[0]); | |||
| entry->errors = 0; | |||
| if (parse_sockaddr((char *)part[1], &entry->sa) < 0) { | |||
| log_print(LOG_ERROR, "p_ts2: invalid address: <%s>", part[1]); | |||
| free(entry->name); | |||
| free(entry); | |||
| return -1; | |||
| } | |||
| log_print(LOG_INFO, "p_ts2: added server '%s'", entry->name); | |||
| list_add_tail(&entry->list, &server_list); | |||
| return 0; | |||
| } | |||
| static int init(void) | |||
| { | |||
| ts2_resp_buf = malloc(TS2_RESP_SIZE); | |||
| if (ts2_resp_buf == NULL) { | |||
| log_print(LOG_ERROR, "p_ts2: out of memory"); | |||
| return -1; | |||
| } | |||
| config_get_strings("p_ts2", "server", 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(ts2_resp_buf); | |||
| return 0; | |||
| } | |||
| struct sammler_plugin plugin = { | |||
| .name = "ts2", | |||
| .interval = 60, | |||
| .init = &init, | |||
| .fini = &fini, | |||
| .probe = &probe, | |||
| .get_ds = &get_ds, | |||
| }; | |||