summaryrefslogtreecommitdiff
path: root/lib/pcounter.c
blob: 9b56807da93b483bc91f6e0bfdcedc9854485ec6 (plain) (blame)
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
/*
 * Define default pcounter functions
 * Note that often used pcounters use dedicated functions to get a speed increase.
 * (see DEFINE_PCOUNTER/REF_PCOUNTER_MEMBER)
 */

#include <linux/module.h>
#include <linux/pcounter.h>
#include <linux/smp.h>
#include <linux/cpumask.h>

static void pcounter_dyn_add(struct pcounter *self, int inc)
{
	per_cpu_ptr(self->per_cpu_values, smp_processor_id())[0] += inc;
}

static int pcounter_dyn_getval(const struct pcounter *self, int cpu)
{
	return per_cpu_ptr(self->per_cpu_values, cpu)[0];
}

int pcounter_getval(const struct pcounter *self)
{
	int res = 0, cpu;

	for_each_possible_cpu(cpu)
		res += self->getval(self, cpu);

	return res;
}
EXPORT_SYMBOL_GPL(pcounter_getval);

int pcounter_alloc(struct pcounter *self)
{
	int rc = 0;
	if (self->add == NULL) {
		self->per_cpu_values = alloc_percpu(int);
		if (self->per_cpu_values != NULL) {
			self->add    = pcounter_dyn_add;
			self->getval = pcounter_dyn_getval;
		} else
			rc = 1;
	}
	return rc;
}
EXPORT_SYMBOL_GPL(pcounter_alloc);

void pcounter_free(struct pcounter *self)
{
	if (self->per_cpu_values != NULL) {
		free_percpu(self->per_cpu_values);
		self->per_cpu_values = NULL;
		self->getval = NULL;
		self->add = NULL;
	}
}
EXPORT_SYMBOL_GPL(pcounter_free);