/*
    MODULE -- statistics

    Copyright (C) Alberto Ornaghi

    $Id: statistics.c 790 2009-08-03 14:34:04Z alor $
*/

#include <main.h>
#include <threads.h>

static long long heartbeat;
pthread_mutex_t hbeat_mux = PTHREAD_MUTEX_INITIALIZER;
#define HEARTBEAT_LOCK do { pthread_mutex_lock(&hbeat_mux); } while (0)
#define HEARTBEAT_UNLOCK do { pthread_mutex_unlock(&hbeat_mux); } while (0)

long long stat_get_heartbeat(void);
void stat_heartbeat(void);

// get current heartbeat
long long stat_get_heartbeat(void)
{
	long long current;

	HEARTBEAT_LOCK;
	current = heartbeat;
	HEARTBEAT_UNLOCK;

	return current;
}

// do heartbeat
// to be placed in most recurrent code blocks
void stat_heartbeat(void)
{
	HEARTBEAT_LOCK;
	heartbeat++;
	HEARTBEAT_UNLOCK;
}

struct stat_t {
	long long transferred;
	long long received;
	long long sent;
	long long throughput;
	time_t start_time;
	u_int32 packetrate;
};

static struct stat_t statistics;

void stat_init(void);
void stat_update_sent(u_int32 sent);
void stat_update_received(u_int32 received);
void stat_get(u_int32 *transferred, u_int32 *throughput, u_int32 *packetrate);
void stat_log_statistics(void);
long long stat_compute_avg_throughput(void);

/******** STATISTICS *********/

void stat_init(void)
{
	statistics.transferred = 0;
	statistics.received = 0;
	statistics.sent = 0;
	statistics.throughput = 0;
	statistics.packetrate = 0;

	statistics.start_time = time(NULL);
}

void stat_log_statistics(void)
{
	u_int32 xferred;
	u_int32 tput;

	stat_get(&xferred, &tput, NULL);

	DEBUG_MSG(D_INFO, "%s statistics [transferred %d][throughtput %d]", __func__, xferred, tput);
}

void stat_get(u_int32 *transferred, u_int32 *throughput, u_int32 *packetrate)
{

	stat_compute_avg_throughput();

	if (transferred)
		*transferred = statistics.transferred;

	if (throughput)
		*throughput = statistics.throughput;
	
	if (packetrate)
		*packetrate = statistics.packetrate;
}

void stat_update_sent(u_int32 sent)
{
	statistics.sent += sent;
	statistics.transferred += sent;
}

void stat_update_received(u_int32 received)
{
	statistics.received += received;
	statistics.transferred += received;
}

long long stat_compute_avg_throughput(void)
{
	time_t now = time(NULL);	

	// tput in bytes per second
	statistics.throughput = statistics.transferred / (u_int32)(now - statistics.start_time);

	return statistics.throughput;
}

/* EOF */

// vim:ts=3:expandtab