blob: 1739510d89943df23f193d5fb04332f47cdaf958 [file] [log] [blame]
Amir Vadai122733a2012-07-18 22:33:50 +00001#ifndef __LINUX_CPU_RMAP_H
2#define __LINUX_CPU_RMAP_H
3
Ben Hutchingsc39649c2011-01-19 11:03:25 +00004/*
5 * cpu_rmap.c: CPU affinity reverse-map support
6 * Copyright 2011 Solarflare Communications Inc.
7 *
8 * This program is free software; you can redistribute it and/or modify it
9 * under the terms of the GNU General Public License version 2 as published
10 * by the Free Software Foundation, incorporated herein by reference.
11 */
12
13#include <linux/cpumask.h>
14#include <linux/gfp.h>
15#include <linux/slab.h>
David Decotigny896f97e2013-01-11 14:31:36 -080016#include <linux/kref.h>
Ben Hutchingsc39649c2011-01-19 11:03:25 +000017
18/**
19 * struct cpu_rmap - CPU affinity reverse-map
David Decotigny896f97e2013-01-11 14:31:36 -080020 * @refcount: kref for object
Ben Hutchingsc39649c2011-01-19 11:03:25 +000021 * @size: Number of objects to be reverse-mapped
22 * @used: Number of objects added
23 * @obj: Pointer to array of object pointers
24 * @near: For each CPU, the index and distance to the nearest object,
25 * based on affinity masks
26 */
27struct cpu_rmap {
David Decotigny896f97e2013-01-11 14:31:36 -080028 struct kref refcount;
Ben Hutchingsc39649c2011-01-19 11:03:25 +000029 u16 size, used;
30 void **obj;
31 struct {
32 u16 index;
33 u16 dist;
34 } near[0];
35};
36#define CPU_RMAP_DIST_INF 0xffff
37
38extern struct cpu_rmap *alloc_cpu_rmap(unsigned int size, gfp_t flags);
David Decotigny896f97e2013-01-11 14:31:36 -080039extern int cpu_rmap_put(struct cpu_rmap *rmap);
Ben Hutchingsc39649c2011-01-19 11:03:25 +000040
41extern int cpu_rmap_add(struct cpu_rmap *rmap, void *obj);
42extern int cpu_rmap_update(struct cpu_rmap *rmap, u16 index,
43 const struct cpumask *affinity);
44
45static inline u16 cpu_rmap_lookup_index(struct cpu_rmap *rmap, unsigned int cpu)
46{
47 return rmap->near[cpu].index;
48}
49
50static inline void *cpu_rmap_lookup_obj(struct cpu_rmap *rmap, unsigned int cpu)
51{
52 return rmap->obj[rmap->near[cpu].index];
53}
54
55#ifdef CONFIG_GENERIC_HARDIRQS
56
57/**
58 * alloc_irq_cpu_rmap - allocate CPU affinity reverse-map for IRQs
59 * @size: Number of objects to be mapped
60 *
61 * Must be called in process context.
62 */
63static inline struct cpu_rmap *alloc_irq_cpu_rmap(unsigned int size)
64{
65 return alloc_cpu_rmap(size, GFP_KERNEL);
66}
67extern void free_irq_cpu_rmap(struct cpu_rmap *rmap);
68
69extern int irq_cpu_rmap_add(struct cpu_rmap *rmap, int irq);
70
71#endif
Amir Vadai122733a2012-07-18 22:33:50 +000072#endif /* __LINUX_CPU_RMAP_H */