[PATCH] taskstats_tgid_alloc: optimization
[safe/jmp/linux-2.6] / include / linux / taskstats_kern.h
1 /* taskstats_kern.h - kernel header for per-task statistics interface
2  *
3  * Copyright (C) Shailabh Nagar, IBM Corp. 2006
4  *           (C) Balbir Singh,   IBM Corp. 2006
5  */
6
7 #ifndef _LINUX_TASKSTATS_KERN_H
8 #define _LINUX_TASKSTATS_KERN_H
9
10 #include <linux/taskstats.h>
11 #include <linux/sched.h>
12 #include <net/genetlink.h>
13
14 #ifdef CONFIG_TASKSTATS
15 extern kmem_cache_t *taskstats_cache;
16 extern struct mutex taskstats_exit_mutex;
17
18 static inline void taskstats_exit_free(struct taskstats *tidstats)
19 {
20         if (tidstats)
21                 kmem_cache_free(taskstats_cache, tidstats);
22 }
23
24 static inline void taskstats_tgid_init(struct signal_struct *sig)
25 {
26         spin_lock_init(&sig->stats_lock);
27         sig->stats = NULL;
28 }
29
30 static inline void taskstats_tgid_alloc(struct signal_struct *sig)
31 {
32         struct taskstats *stats;
33         unsigned long flags;
34
35         if (sig->stats != NULL)
36                 return;
37
38         stats = kmem_cache_zalloc(taskstats_cache, SLAB_KERNEL);
39         if (!stats)
40                 return;
41
42         spin_lock_irqsave(&sig->stats_lock, flags);
43         if (!sig->stats) {
44                 sig->stats = stats;
45                 stats = NULL;
46         }
47         spin_unlock_irqrestore(&sig->stats_lock, flags);
48
49         if (stats)
50                 kmem_cache_free(taskstats_cache, stats);
51 }
52
53 static inline void taskstats_tgid_free(struct signal_struct *sig)
54 {
55         if (sig->stats)
56                 kmem_cache_free(taskstats_cache, sig->stats);
57 }
58
59 extern void taskstats_exit_alloc(struct taskstats **, unsigned int *);
60 extern void taskstats_exit_send(struct task_struct *, struct taskstats *, int, unsigned int);
61 extern void taskstats_init_early(void);
62 extern void taskstats_tgid_alloc(struct signal_struct *);
63 #else
64 static inline void taskstats_exit_alloc(struct taskstats **ptidstats, unsigned int *mycpu)
65 {}
66 static inline void taskstats_exit_free(struct taskstats *ptidstats)
67 {}
68 static inline void taskstats_exit_send(struct task_struct *tsk,
69                                        struct taskstats *tidstats,
70                                        int group_dead, unsigned int cpu)
71 {}
72 static inline void taskstats_tgid_init(struct signal_struct *sig)
73 {}
74 static inline void taskstats_tgid_alloc(struct signal_struct *sig)
75 {}
76 static inline void taskstats_tgid_free(struct signal_struct *sig)
77 {}
78 static inline void taskstats_init_early(void)
79 {}
80 #endif /* CONFIG_TASKSTATS */
81
82 #endif
83