From 390817c0535690516ae146a8d384108dcd87ea99 Mon Sep 17 00:00:00 2001 From: octo Date: Mon, 20 Mar 2006 16:49:59 +0000 Subject: [PATCH] Added first version of an `apache' plugin. --- src/apache.c | 326 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 326 insertions(+) create mode 100644 src/apache.c diff --git a/src/apache.c b/src/apache.c new file mode 100644 index 00000000..7bb0a2ac --- /dev/null +++ b/src/apache.c @@ -0,0 +1,326 @@ +/** + * collectd - src/apache.c + * Copyright (C) 2006 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; 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., + * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA + * + * Authors: + * Florian octo Forster + **/ + +#include "collectd.h" +#include "common.h" +#include "plugin.h" +#include "configfile.h" + +#define MODULE_NAME "apache" + +#if HAVE_LIBCURL && HAVE_CURL_CURL_H +# define APACHE_HAVE_READ 1 +# include +#else +# define APACHE_HAVE_READ 0 +#endif + +static char *url = NULL; +static char *user = NULL; +static char *pass = NULL; + +#if APACHE_HAVE_READ +static CURL *curl = NULL; + +static char apache_buffer[4096]; +static int apache_buffer_len = 0; +static char apache_curl_error[CURL_ERROR_SIZE]; +#endif + +static char *bytes_file = "apache/apache_bytes.rrd"; +static char *bytes_ds_def[] = +{ + "DS:count:COUNTER:25:0:U", + NULL +}; +static int bytes_ds_num = 1; + +static char *requests_file = "apache/apache_requests.rrd"; +static char *requests_ds_def[] = +{ + "DS:count:COUNTER:25:0:U", + NULL +}; +static int requests_ds_num = 1; + +static char *scoreboard_file = "apache/apache_scoreboard-%s.rrd"; +static char *scoreboard_ds_def[] = +{ + "DS:count:GAUGE:25:0:U", + NULL +}; +static int scoreboard_ds_num = 1; + +static char *config_keys[] = +{ + "URI", + "User", + "Password", + NULL +}; +static int config_keys_num = 3; + + +static size_t apache_curl_callback (void *buf, size_t size, size_t nmemb, void *stream) +{ + size_t len = size * nmemb; + + if ((apache_buffer_len + len) >= 4096) + { + len = 4095 - apache_buffer_len; + } + + if (len <= 0) + return (len); + + memcpy (apache_buffer + apache_buffer_len, (char *) buf, len); + apache_buffer_len += len; + apache_buffer[apache_buffer_len] = '\0'; + + return (len); +} + +static void config_set (char **var, char *value) +{ + if (*var != NULL) + { + free (*var); + *var = NULL; + } + + if ((*var = strdup (value)) == NULL) + return (1); + else + return (0); +} + +static int config (char *key, char *value) +{ + if (strcasecmp (key, "uri") == 0) + return (config_set (&key, value)); + else if (strcasecmp (key, "user") == 0) + return (config_set (&user, value)); + else if (strcasecmp (key, "password") == 0) + return (config_set (&pass, value)); + else + return (-1); +} + +static void init (void) +{ +#if APACHE_HAVE_READ + static credentials[1024]; + + if (curl != NULL) + { + curl_easy_cleanup (curl); + } + + if ((curl = curl_easy_init (CURL_GLOBAL_NOTHING)) == NULL) + { + syslog (LOG_ERR, "apache: `curl_easy_init' failed."); + return; + } + + curl_easy_setopt (curl, CURLOPT_WRITEFUNCTION, apache_curl_callback); + curl_easy_setopt (curl, CURLOPT_USERAGENT, PACKAGE_NAME"/"PACKAGE_VERSION); + curl_easy_setopt (curl, CURLOPT_ERRORBUFFER, apache_curl_error); + + if (user != NULL) + { + if (snprintf (credentials, 1024, "%s:%s", user, pass == NULL ? "" : pass) >= 1024) + { + syslog (LOG_ERR, "apache: Credentials would have been truncated."); + return; + } + + curl_easy_setopt (curl, CURLOPT_USERPWD, credentials); + } + + if (url != NULL) + { + curl_easy_setopt (curl, CURLOPT_URL, url); + } +#endif /* APACHE_HAVE_READ */ +} + +static void bytes_write (char *host, char *inst, char *val) +{ + rrd_update_file (host, bytes_file, val, bytes_ds_def, bytes_ds_num); +} + +static void requests_write (char *host, char *inst, char *val) +{ + rrd_update_file (host, requests_file, val, requests_ds_def, requests_ds_num); +} + +static void scoreboard_write (char *host, char *inst, char *val) +{ + char buf[1024]; + + if (snprintf (buf, 1024, scoreboard_file, inst) >= 1024) + return; + + rrd_update_file (host, buf, val, scoreboard_ds_def, scoreboard_ds_num); +} + +#if APACHE_HAVE_READ +static void submit (char *type, char *inst, long long value) +{ + char buf[1024]; + int status; + + status = snprintf (buf, 1024, "%u:%lli", (unsigned int) curtime, value); + if (status < 0) + { + syslog (LOG_ERR, "apache: bytes_submit: snprintf failed"); + return; + } + else if (status >= 1024) + { + syslog (LOG_WARNING, "apache: bytes_submit: snprintf was truncated"); + return; + } + + plugin_submit (type, inst, buf); +} + +static void submit_scoreboard (char *buf) +{ + /* + * Scoreboard Key: + * "_" Waiting for Connection, "S" Starting up, "R" Reading Request, + * "W" Sending Reply, "K" Keepalive (read), "D" DNS Lookup, + * "C" Closing connection, "L" Logging, "G" Gracefully finishing, + * "I" Idle cleanup of worker, "." Open slot with no current process + */ + long long open = 0LL; + long long waiting = 0LL; + long long starting = 0LL; + long long reading = 0LL; + long long sending = 0LL; + long long keepalive = 0LL; + long long dnslookup = 0LL; + long long closing = 0LL; + long long logging = 0LL; + long long finishing = 0LL; + long long idle_cleanup = 0LL; + + int i; + + for (i = 0; buf[i] != '\0'; i++) + { + if (buf[i] == '.') open++; + else if (buf[i] == '_') waiting++; + else if (buf[i] == 'S') starting++; + else if (buf[i] == 'R') reading++; + else if (buf[i] == 'W') sending++; + else if (buf[i] == 'K') keepalive++; + else if (buf[i] == 'D') dnslookup++; + else if (buf[i] == 'C') closing++; + else if (buf[i] == 'L') logging++; + else if (buf[i] == 'G') finishing++; + else if (buf[i] == 'I') idle_cleanup++; + } + + submit ("apache_scoreboard", "open" , open); + submit ("apache_scoreboard", "waiting" , waiting); + submit ("apache_scoreboard", "starting" , starting); + submit ("apache_scoreboard", "reading" , reading); + submit ("apache_scoreboard", "sending" , sending); + submit ("apache_scoreboard", "keepalive", keepalive); + submit ("apache_scoreboard", "dnslookup", dnslookup); + submit ("apache_scoreboard", "closing" , closing); + submit ("apache_scoreboard", "logging" , logging); + submit ("apache_scoreboard", "finishing", finishing); + submit ("apache_scoreboard", "idle_cleanup", idle_cleanup); +} + +static void read (void) +{ + CURLcode status; + int i; + + char *ptr; + char *lines[16]; + int lines_num = 0; + + char *fields[4]; + int fields_num; + + if (curl == NULL) + return; + if (url == NULL) + return; + + if (curl_easy_perform (curl) != 0) + { + syslog (LOG_WARN, "apache: curl_easy_perform failed: %s", apache_curl_error); + return; + } + + ptr = apache_buffer; + while ((lines[lines_num] = strtok (ptr, "\n\r")) != NULL) + { + ptr = NULL; + lines_num++; + + if (lines_num >= 16) + break; + } + + for (i = 0; i < lines_num; i++) + { + fields_num = strsplit (lines[i], fields, 4); + + if (fields_num == 3) + { + if ((strcmp (fields[0], "Total") == 0) + && (strcmp (fields[1], "Accesses:") == 0)) + submit ("apache_requests", NULL, atoll (fields[2])); + else if ((strcmp (fields[0], "Total") == 0) + && (strcmp (fields[1], "kBytes:") == 0)) + submit ("apache_bytes", NULL, 1024LL * atoll (fields[2])); + } + else if (fields_num == 2) + { + if (strcmp (fields[0], "Scoreboard:") == 0) + submit_scoreboard (fields[1]); + } + } + + apache_buffer_len = 0; +} +#else +# define read NULL +#endif /* APACHE_HAVE_READ */ + +void module_register (void) +{ + plugin_register (MODULE_NAME, init, read, NULL); + plugin_register ("apache_requests", NULL, NULL, requests_write); + plugin_register ("apache_bytes", NULL, NULL, bytes_write); + plugin_register ("apache_scoreboard", NULL, NULL, scoreboard_write); + cf_register (MODULE_NAME, config, config_keys, config_keys_num); +} + +#undef MODULE_NAME -- 2.11.0