2008-05-24 16:05:49 +08:00
|
|
|
/* tinyproxy - A fast light-weight HTTP proxy
|
|
|
|
|
* Copyright (C) 2000 Robert James Kaes <rjkaes@users.sourceforge.net>
|
2000-09-12 08:06:09 +08:00
|
|
|
*
|
2008-05-24 16:05:49 +08:00
|
|
|
* 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 Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
/* This module handles the statistics for tinyproxy. There are only two
|
2000-09-12 08:06:09 +08:00
|
|
|
* public API functions. The reason for the functions, rather than just a
|
|
|
|
|
* external structure is that tinyproxy is now multi-threaded and we can
|
2002-05-27 02:53:14 +08:00
|
|
|
* not allow more than one child to access the statistics at the same
|
2000-09-12 08:06:09 +08:00
|
|
|
* time. This is prevented by a mutex. If there is a need for more
|
|
|
|
|
* statistics in the future, just add to the structure, enum (in the header),
|
|
|
|
|
* and the switch statement in update_stats().
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
#include "tinyproxy.h"
|
|
|
|
|
|
|
|
|
|
#include "log.h"
|
2002-05-24 02:20:27 +08:00
|
|
|
#include "heap.h"
|
2008-05-24 16:17:14 +08:00
|
|
|
#include "html-error.h"
|
2000-09-12 08:06:09 +08:00
|
|
|
#include "stats.h"
|
2001-09-09 02:58:37 +08:00
|
|
|
#include "utils.h"
|
2000-09-12 08:06:09 +08:00
|
|
|
|
2008-12-01 23:01:11 +08:00
|
|
|
struct stat_s
|
|
|
|
|
{
|
|
|
|
|
unsigned long int num_reqs;
|
|
|
|
|
unsigned long int num_badcons;
|
|
|
|
|
unsigned long int num_open;
|
|
|
|
|
unsigned long int num_refused;
|
|
|
|
|
unsigned long int num_denied;
|
2000-09-12 08:06:09 +08:00
|
|
|
};
|
|
|
|
|
|
2002-05-27 02:53:14 +08:00
|
|
|
static struct stat_s *stats;
|
2000-09-12 08:06:09 +08:00
|
|
|
|
|
|
|
|
/*
|
2002-05-24 12:45:32 +08:00
|
|
|
* Initialize the statistics information to zero.
|
2000-09-12 08:06:09 +08:00
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
void
|
2008-12-01 23:01:11 +08:00
|
|
|
init_stats (void)
|
2000-09-12 08:06:09 +08:00
|
|
|
{
|
2008-12-01 23:01:11 +08:00
|
|
|
stats = malloc_shared_memory (sizeof (struct stat_s));
|
|
|
|
|
if (stats == MAP_FAILED)
|
|
|
|
|
return;
|
2002-05-27 02:53:14 +08:00
|
|
|
|
2008-12-01 23:01:11 +08:00
|
|
|
memset (stats, 0, sizeof (struct stat_s));
|
2000-09-12 08:06:09 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
* Display the statics of the tinyproxy server.
|
|
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
int
|
2008-12-01 23:01:11 +08:00
|
|
|
showstats (struct conn_s *connptr)
|
2000-09-12 08:06:09 +08:00
|
|
|
{
|
2008-12-01 23:01:11 +08:00
|
|
|
static char *msg =
|
|
|
|
|
"<?xml version=\"1.0\" encoding=\"UTF-8\" ?>\n"
|
2008-12-08 20:26:08 +08:00
|
|
|
"<!DOCTYPE html PUBLIC \"-//W3C//DTD XHTML 1.1//EN\" "
|
|
|
|
|
"\"http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd\">\n"
|
2008-12-01 23:01:11 +08:00
|
|
|
"<html>\n"
|
|
|
|
|
"<head><title>%s version %s run-time statistics</title></head>\n"
|
|
|
|
|
"<body>\n"
|
|
|
|
|
"<h1>%s version %s run-time statistics</h1>\n"
|
|
|
|
|
"<p>\n"
|
|
|
|
|
"Number of open connections: %lu<br />\n"
|
|
|
|
|
"Number of requests: %lu<br />\n"
|
|
|
|
|
"Number of bad connections: %lu<br />\n"
|
|
|
|
|
"Number of denied connections: %lu<br />\n"
|
|
|
|
|
"Number of refused connections due to high load: %lu\n"
|
|
|
|
|
"</p>\n"
|
|
|
|
|
"<hr />\n"
|
|
|
|
|
"<p><em>Generated by %s version %s.</em></p>\n" "</body>\n" "</html>\n";
|
|
|
|
|
|
|
|
|
|
char *message_buffer;
|
|
|
|
|
char opens[16], reqs[16], badconns[16], denied[16], refused[16];
|
|
|
|
|
FILE *statfile;
|
|
|
|
|
|
|
|
|
|
snprintf (opens, sizeof (opens), "%lu", stats->num_open);
|
|
|
|
|
snprintf (reqs, sizeof (reqs), "%lu", stats->num_reqs);
|
|
|
|
|
snprintf (badconns, sizeof (badconns), "%lu", stats->num_badcons);
|
|
|
|
|
snprintf (denied, sizeof (denied), "%lu", stats->num_denied);
|
|
|
|
|
snprintf (refused, sizeof (refused), "%lu", stats->num_refused);
|
|
|
|
|
|
|
|
|
|
if (!config.statpage || (!(statfile = fopen (config.statpage, "r"))))
|
|
|
|
|
{
|
|
|
|
|
message_buffer = safemalloc (MAXBUFFSIZE);
|
|
|
|
|
if (!message_buffer)
|
2008-12-08 21:39:44 +08:00
|
|
|
return -1;
|
2008-12-01 23:01:11 +08:00
|
|
|
|
|
|
|
|
snprintf (message_buffer, MAXBUFFSIZE, msg,
|
2008-12-08 21:39:44 +08:00
|
|
|
PACKAGE, VERSION, PACKAGE, VERSION,
|
|
|
|
|
stats->num_open,
|
|
|
|
|
stats->num_reqs,
|
|
|
|
|
stats->num_badcons, stats->num_denied,
|
|
|
|
|
stats->num_refused, PACKAGE, VERSION);
|
2008-12-01 23:01:11 +08:00
|
|
|
|
|
|
|
|
if (send_http_message (connptr, 200, "OK", message_buffer) < 0)
|
2008-12-08 21:39:44 +08:00
|
|
|
{
|
|
|
|
|
safefree (message_buffer);
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
2008-12-01 23:01:11 +08:00
|
|
|
|
|
|
|
|
safefree (message_buffer);
|
|
|
|
|
return 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
add_error_variable (connptr, "opens", opens);
|
|
|
|
|
add_error_variable (connptr, "reqs", reqs);
|
|
|
|
|
add_error_variable (connptr, "badconns", badconns);
|
|
|
|
|
add_error_variable (connptr, "deniedconns", denied);
|
|
|
|
|
add_error_variable (connptr, "refusedconns", refused);
|
|
|
|
|
add_standard_vars (connptr);
|
|
|
|
|
send_http_headers (connptr, 200, "Statistic requested");
|
|
|
|
|
send_html_file (statfile, connptr);
|
|
|
|
|
fclose (statfile);
|
|
|
|
|
|
|
|
|
|
return 0;
|
2000-09-12 08:06:09 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
* Update the value of the statistics. The update_level is defined in
|
|
|
|
|
* stats.h
|
|
|
|
|
*/
|
2001-11-22 08:31:10 +08:00
|
|
|
int
|
2008-12-01 23:01:11 +08:00
|
|
|
update_stats (status_t update_level)
|
2000-09-12 08:06:09 +08:00
|
|
|
{
|
2008-12-01 23:01:11 +08:00
|
|
|
switch (update_level)
|
|
|
|
|
{
|
|
|
|
|
case STAT_BADCONN:
|
|
|
|
|
++stats->num_badcons;
|
|
|
|
|
break;
|
|
|
|
|
case STAT_OPEN:
|
|
|
|
|
++stats->num_open;
|
|
|
|
|
++stats->num_reqs;
|
|
|
|
|
break;
|
|
|
|
|
case STAT_CLOSE:
|
|
|
|
|
--stats->num_open;
|
|
|
|
|
break;
|
|
|
|
|
case STAT_REFUSE:
|
|
|
|
|
++stats->num_refused;
|
|
|
|
|
break;
|
|
|
|
|
case STAT_DENIED:
|
|
|
|
|
++stats->num_denied;
|
|
|
|
|
break;
|
|
|
|
|
default:
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return 0;
|
2000-09-12 08:06:09 +08:00
|
|
|
}
|
