This module handles the various stats relating to tinyproxy's
funcationality.
This commit is contained in:
Robert James Kaes
parent
392c97e3d1
commit
bb32293415
126
src/stats.c
Normal file
126
src/stats.c
Normal file
@ -0,0 +1,126 @@
|
||||
/* $Id: stats.c,v 1.1 2000-09-12 00:06:09 rjkaes Exp $
|
||||
*
|
||||
* This module handles the statistics for tinyproxy. There are only two
|
||||
* public API functions. The reason for the functions, rather than just a
|
||||
* external structure is that tinyproxy is now multi-threaded and we can
|
||||
* not allow more than one thread to access the statistics at the same
|
||||
* 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().
|
||||
*
|
||||
* Copyright (C) 2000 Robert James Kaes (rjkaes@flarenet.com)
|
||||
*
|
||||
* 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, 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.
|
||||
*/
|
||||
|
||||
#include "tinyproxy.h"
|
||||
|
||||
#include "log.h"
|
||||
#include "stats.h"
|
||||
|
||||
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;
|
||||
};
|
||||
|
||||
static struct stat_s stats;
|
||||
|
||||
/*
|
||||
* Locking when we're accessing the statistics, since I don't want multiple
|
||||
* threads changing the information at the same time.
|
||||
*/
|
||||
pthread_mutex_t stats_mutex = PTHREAD_MUTEX_INITIALIZER;
|
||||
#define LOCK() pthread_mutex_lock(&stats_mutex)
|
||||
#define UNLOCK() pthread_mutex_unlock(&stats_mutex)
|
||||
|
||||
/*
|
||||
* Initialise the statistics information to zero.
|
||||
*/
|
||||
void init_stats(void)
|
||||
{
|
||||
LOCK();
|
||||
memset(&stats, 0, sizeof(stats));
|
||||
UNLOCK();
|
||||
}
|
||||
|
||||
/*
|
||||
* Display the statics of the tinyproxy server.
|
||||
*/
|
||||
int showstats(struct conn_s *connptr)
|
||||
{
|
||||
static char *msg = "HTTP/1.0 200 OK\r\n" \
|
||||
"Content-type: text/html\r\n\r\n" \
|
||||
"<html><head><title>%s (%s) stats</title></head>\r\n" \
|
||||
"<body>\r\n" \
|
||||
"<center><h2>%s (%s) run-time statistics</h2></center><hr>\r\n" \
|
||||
"<blockquote>\r\n" \
|
||||
"Number of open connections: %lu<br>\r\n" \
|
||||
"Number of requests: %lu<br>\r\n" \
|
||||
"Number of bad connections: %lu<br>\r\n" \
|
||||
"Number of denied connections: %lu<br>\r\n" \
|
||||
"Number of refused connections due to high load: %lu\r\n" \
|
||||
"</blockquote>\r\n</body></html>\r\n";
|
||||
|
||||
connptr->output_message = malloc(MAXBUFFSIZE);
|
||||
if (!connptr->output_message) {
|
||||
log(LOG_CRIT, "Out of memory!");
|
||||
return -1;
|
||||
}
|
||||
|
||||
LOCK();
|
||||
sprintf(connptr->output_message, msg,
|
||||
PACKAGE, VERSION, PACKAGE, VERSION,
|
||||
stats.num_open,
|
||||
stats.num_reqs,
|
||||
stats.num_badcons,
|
||||
stats.num_denied,
|
||||
stats.num_refused);
|
||||
UNLOCK();
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
/*
|
||||
* Update the value of the statistics. The update_level is defined in
|
||||
* stats.h
|
||||
*/
|
||||
int update_stats(status_t update_level)
|
||||
{
|
||||
LOCK();
|
||||
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:
|
||||
DEBUG2("An invalid STATUS type (%d)", update_level);
|
||||
UNLOCK();
|
||||
return -1;
|
||||
}
|
||||
UNLOCK();
|
||||
|
||||
return 0;
|
||||
}
|
||||
39
src/stats.h
Normal file
39
src/stats.h
Normal file
@ -0,0 +1,39 @@
|
||||
/* $Id: stats.h,v 1.1 2000-09-12 00:06:09 rjkaes Exp $
|
||||
*
|
||||
* See 'stats.h' for a detailed description.
|
||||
*
|
||||
* Copyright (C) 2000 Robert James Kaes (rjkaes@flarenet.com)
|
||||
*
|
||||
* 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, 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.
|
||||
*/
|
||||
|
||||
#ifndef _TINYPROXY_STATS_H_
|
||||
#define _TINYPROXY_STATS_H_
|
||||
|
||||
/*
|
||||
* Various logable statistics
|
||||
*/
|
||||
typedef enum {
|
||||
STAT_BADCONN, /* bad connection, for unknonw reason */
|
||||
STAT_OPEN, /* connection opened */
|
||||
STAT_CLOSE, /* connection closed */
|
||||
STAT_REFUSE, /* connection refused (to outside world) */
|
||||
STAT_DENIED /* connection denied to tinyproxy itself */
|
||||
} status_t;
|
||||
|
||||
/*
|
||||
* Public API to the statistics for tinyproxy
|
||||
*/
|
||||
extern void init_stats(void);
|
||||
extern int showstats(struct conn_s *connptr);
|
||||
extern int update_stats(status_t update_level);
|
||||
|
||||
#endif
|
||||
Loading…
Reference in New Issue
Block a user