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