2005-04-16 22:20:36 +00:00
|
|
|
/*
|
|
|
|
* ip_vs_est.c: simple rate estimator for IPVS
|
|
|
|
*
|
|
|
|
* Authors: Wensong Zhang <wensong@linuxvirtualserver.org>
|
|
|
|
*
|
|
|
|
* 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.
|
|
|
|
*
|
|
|
|
* Changes:
|
|
|
|
*
|
|
|
|
*/
|
2009-07-30 21:29:44 +00:00
|
|
|
|
|
|
|
#define KMSG_COMPONENT "IPVS"
|
|
|
|
#define pr_fmt(fmt) KMSG_COMPONENT ": " fmt
|
|
|
|
|
2005-04-16 22:20:36 +00:00
|
|
|
#include <linux/kernel.h>
|
2005-12-27 04:43:12 +00:00
|
|
|
#include <linux/jiffies.h>
|
2005-04-16 22:20:36 +00:00
|
|
|
#include <linux/types.h>
|
2006-01-05 20:14:43 +00:00
|
|
|
#include <linux/interrupt.h>
|
2008-01-12 10:33:50 +00:00
|
|
|
#include <linux/sysctl.h>
|
2008-08-10 18:24:41 +00:00
|
|
|
#include <linux/list.h>
|
2005-04-16 22:20:36 +00:00
|
|
|
|
|
|
|
#include <net/ip_vs.h>
|
|
|
|
|
|
|
|
/*
|
|
|
|
This code is to estimate rate in a shorter interval (such as 8
|
|
|
|
seconds) for virtual services and real servers. For measure rate in a
|
|
|
|
long interval, it is easy to implement a user level daemon which
|
|
|
|
periodically reads those statistical counters and measure rate.
|
|
|
|
|
|
|
|
Currently, the measurement is activated by slow timer handler. Hope
|
|
|
|
this measurement will not introduce too much load.
|
|
|
|
|
|
|
|
We measure rate during the last 8 seconds every 2 seconds:
|
|
|
|
|
|
|
|
avgrate = avgrate*(1-W) + rate*W
|
|
|
|
|
|
|
|
where W = 2^(-2)
|
|
|
|
|
|
|
|
NOTES.
|
|
|
|
|
|
|
|
* The stored value for average bps is scaled by 2^5, so that maximal
|
|
|
|
rate is ~2.15Gbits/s, average pps and cps are scaled by 2^10.
|
|
|
|
|
|
|
|
* A lot code is taken from net/sched/estimator.c
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
2008-08-10 18:24:41 +00:00
|
|
|
static void estimation_timer(unsigned long arg);
|
2005-04-16 22:20:36 +00:00
|
|
|
|
2008-08-10 18:24:41 +00:00
|
|
|
static LIST_HEAD(est_list);
|
|
|
|
static DEFINE_SPINLOCK(est_lock);
|
|
|
|
static DEFINE_TIMER(est_timer, estimation_timer, 0, 0);
|
2005-04-16 22:20:36 +00:00
|
|
|
|
|
|
|
static void estimation_timer(unsigned long arg)
|
|
|
|
{
|
|
|
|
struct ip_vs_estimator *e;
|
|
|
|
struct ip_vs_stats *s;
|
|
|
|
u32 n_conns;
|
|
|
|
u32 n_inpkts, n_outpkts;
|
|
|
|
u64 n_inbytes, n_outbytes;
|
|
|
|
u32 rate;
|
|
|
|
|
2008-08-10 18:24:41 +00:00
|
|
|
spin_lock(&est_lock);
|
|
|
|
list_for_each_entry(e, &est_list, list) {
|
|
|
|
s = container_of(e, struct ip_vs_stats, est);
|
2005-04-16 22:20:36 +00:00
|
|
|
|
|
|
|
spin_lock(&s->lock);
|
2008-09-08 11:39:04 +00:00
|
|
|
n_conns = s->ustats.conns;
|
|
|
|
n_inpkts = s->ustats.inpkts;
|
|
|
|
n_outpkts = s->ustats.outpkts;
|
|
|
|
n_inbytes = s->ustats.inbytes;
|
|
|
|
n_outbytes = s->ustats.outbytes;
|
2005-04-16 22:20:36 +00:00
|
|
|
|
|
|
|
/* scaled by 2^10, but divided 2 seconds */
|
|
|
|
rate = (n_conns - e->last_conns)<<9;
|
|
|
|
e->last_conns = n_conns;
|
|
|
|
e->cps += ((long)rate - (long)e->cps)>>2;
|
2008-09-08 11:39:04 +00:00
|
|
|
s->ustats.cps = (e->cps+0x1FF)>>10;
|
2005-04-16 22:20:36 +00:00
|
|
|
|
|
|
|
rate = (n_inpkts - e->last_inpkts)<<9;
|
|
|
|
e->last_inpkts = n_inpkts;
|
|
|
|
e->inpps += ((long)rate - (long)e->inpps)>>2;
|
2008-09-08 11:39:04 +00:00
|
|
|
s->ustats.inpps = (e->inpps+0x1FF)>>10;
|
2005-04-16 22:20:36 +00:00
|
|
|
|
|
|
|
rate = (n_outpkts - e->last_outpkts)<<9;
|
|
|
|
e->last_outpkts = n_outpkts;
|
|
|
|
e->outpps += ((long)rate - (long)e->outpps)>>2;
|
2008-09-08 11:39:04 +00:00
|
|
|
s->ustats.outpps = (e->outpps+0x1FF)>>10;
|
2005-04-16 22:20:36 +00:00
|
|
|
|
|
|
|
rate = (n_inbytes - e->last_inbytes)<<4;
|
|
|
|
e->last_inbytes = n_inbytes;
|
|
|
|
e->inbps += ((long)rate - (long)e->inbps)>>2;
|
2008-09-08 11:39:04 +00:00
|
|
|
s->ustats.inbps = (e->inbps+0xF)>>5;
|
2005-04-16 22:20:36 +00:00
|
|
|
|
|
|
|
rate = (n_outbytes - e->last_outbytes)<<4;
|
|
|
|
e->last_outbytes = n_outbytes;
|
|
|
|
e->outbps += ((long)rate - (long)e->outbps)>>2;
|
2008-09-08 11:39:04 +00:00
|
|
|
s->ustats.outbps = (e->outbps+0xF)>>5;
|
2005-04-16 22:20:36 +00:00
|
|
|
spin_unlock(&s->lock);
|
|
|
|
}
|
2008-08-10 18:24:41 +00:00
|
|
|
spin_unlock(&est_lock);
|
2005-04-16 22:20:36 +00:00
|
|
|
mod_timer(&est_timer, jiffies + 2*HZ);
|
|
|
|
}
|
|
|
|
|
2008-08-10 18:24:41 +00:00
|
|
|
void ip_vs_new_estimator(struct ip_vs_stats *stats)
|
2005-04-16 22:20:36 +00:00
|
|
|
{
|
2008-08-10 18:24:41 +00:00
|
|
|
struct ip_vs_estimator *est = &stats->est;
|
2005-04-16 22:20:36 +00:00
|
|
|
|
2008-08-10 18:24:41 +00:00
|
|
|
INIT_LIST_HEAD(&est->list);
|
2005-04-16 22:20:36 +00:00
|
|
|
|
2008-09-08 11:39:04 +00:00
|
|
|
est->last_conns = stats->ustats.conns;
|
|
|
|
est->cps = stats->ustats.cps<<10;
|
2005-04-16 22:20:36 +00:00
|
|
|
|
2008-09-08 11:39:04 +00:00
|
|
|
est->last_inpkts = stats->ustats.inpkts;
|
|
|
|
est->inpps = stats->ustats.inpps<<10;
|
2005-04-16 22:20:36 +00:00
|
|
|
|
2008-09-08 11:39:04 +00:00
|
|
|
est->last_outpkts = stats->ustats.outpkts;
|
|
|
|
est->outpps = stats->ustats.outpps<<10;
|
2005-04-16 22:20:36 +00:00
|
|
|
|
2008-09-08 11:39:04 +00:00
|
|
|
est->last_inbytes = stats->ustats.inbytes;
|
|
|
|
est->inbps = stats->ustats.inbps<<5;
|
2005-04-16 22:20:36 +00:00
|
|
|
|
2008-09-08 11:39:04 +00:00
|
|
|
est->last_outbytes = stats->ustats.outbytes;
|
|
|
|
est->outbps = stats->ustats.outbps<<5;
|
2005-04-16 22:20:36 +00:00
|
|
|
|
2008-08-10 18:24:41 +00:00
|
|
|
spin_lock_bh(&est_lock);
|
|
|
|
list_add(&est->list, &est_list);
|
|
|
|
spin_unlock_bh(&est_lock);
|
2005-04-16 22:20:36 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void ip_vs_kill_estimator(struct ip_vs_stats *stats)
|
|
|
|
{
|
2008-08-10 18:24:41 +00:00
|
|
|
struct ip_vs_estimator *est = &stats->est;
|
|
|
|
|
|
|
|
spin_lock_bh(&est_lock);
|
|
|
|
list_del(&est->list);
|
|
|
|
spin_unlock_bh(&est_lock);
|
2005-04-16 22:20:36 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void ip_vs_zero_estimator(struct ip_vs_stats *stats)
|
|
|
|
{
|
2008-08-10 18:24:41 +00:00
|
|
|
struct ip_vs_estimator *est = &stats->est;
|
|
|
|
|
|
|
|
/* set counters zero, caller must hold the stats->lock lock */
|
|
|
|
est->last_inbytes = 0;
|
|
|
|
est->last_outbytes = 0;
|
|
|
|
est->last_conns = 0;
|
|
|
|
est->last_inpkts = 0;
|
|
|
|
est->last_outpkts = 0;
|
|
|
|
est->cps = 0;
|
|
|
|
est->inpps = 0;
|
|
|
|
est->outpps = 0;
|
|
|
|
est->inbps = 0;
|
|
|
|
est->outbps = 0;
|
2005-04-16 22:20:36 +00:00
|
|
|
}
|
2008-08-13 22:47:16 +00:00
|
|
|
|
|
|
|
int __init ip_vs_estimator_init(void)
|
|
|
|
{
|
|
|
|
mod_timer(&est_timer, jiffies + 2 * HZ);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
void ip_vs_estimator_cleanup(void)
|
|
|
|
{
|
|
|
|
del_timer_sync(&est_timer);
|
|
|
|
}
|