summaryrefslogtreecommitdiffstats
path: root/src/timer.c
blob: b85c286313e9c67ffc194f4b7a3ec446669957c6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
/*
 * (C) 2006-2007 by Pablo Neira Ayuso <pablo@netfilter.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.
 *
 * 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., 675 Mass Ave, Cambridge, MA 02139, USA.
 */
#include <stdlib.h>
#include <assert.h>
#include <string.h>
#include "conntrackd.h"
#include "timer.h"

#define TIMESLICE_CREDIT (1000000 / STEPS_PER_SECONDS) /* 200 ms timeslice */

void timer_init(struct timer *timer)
{
	memset(timer, 0, sizeof(struct timer));
	timer->credits = TIMESLICE_CREDIT;
}

void timer_start(struct timer *timer)
{
	gettimeofday(&timer->start, NULL);
}

static int timeval_subtract(struct timeval *diff, 
			    struct timeval *start, 
			    struct timeval *stop)
{
	diff->tv_sec = stop->tv_sec - start->tv_sec;
	diff->tv_usec = stop->tv_usec - start->tv_usec;

	if (diff->tv_usec < 0) {
		diff->tv_usec += 1000000;
		diff->tv_sec--;
	}

	/* Return 1 if result is negative. */
	return diff->tv_sec < 0;
}

void timer_stop(struct timer *timer)
{
	gettimeofday(&timer->stop, NULL);
	timeval_subtract(&timer->diff, &timer->start, &timer->stop);
}

int timer_adjust_credit(struct timer *timer)
{
	if (timer->diff.tv_sec != 0) {
		timer->credits = TIMESLICE_CREDIT;
		return 1;
	}

	timer->credits -= timer->diff.tv_usec;

	if (timer->credits < 0) {
		timer->credits += TIMESLICE_CREDIT;
		if (timer->credits < 0)
			timer->credits = TIMESLICE_CREDIT;
		return 1;
	}
	return 0;
}