blob: b96bd299966f04b374e0b1e17dbc793e3efa784d [file] [log] [blame]
Frederic Weisbecker91d1aa432012-11-27 19:33:25 +01001#ifndef _LINUX_CONTEXT_TRACKING_H
2#define _LINUX_CONTEXT_TRACKING_H
3
Frederic Weisbecker91d1aa432012-11-27 19:33:25 +01004#include <linux/sched.h>
Frederic Weisbecker521921b2013-05-16 01:21:38 +02005#include <linux/vtime.h>
Frederic Weisbeckere7358b32013-07-12 02:15:49 +02006#include <linux/context_tracking_state.h>
Frederic Weisbecker56dd9472013-02-24 00:23:25 +01007#include <asm/ptrace.h>
Frederic Weisbecker95a79fd2013-01-07 18:12:14 +01008
Frederic Weisbecker521921b2013-05-16 01:21:38 +02009
Frederic Weisbecker6c1e0252013-02-24 01:19:14 +010010#ifdef CONFIG_CONTEXT_TRACKING
Frederic Weisbecker2e709332013-07-10 00:55:25 +020011extern void context_tracking_cpu_set(int cpu);
12
Rik van Riel3aab4f52015-02-10 15:27:50 -050013extern void context_tracking_enter(enum ctx_state state);
14extern void context_tracking_exit(enum ctx_state state);
Frederic Weisbeckerad657822013-07-10 02:44:35 +020015extern void context_tracking_user_enter(void);
16extern void context_tracking_user_exit(void);
17
18static inline void user_enter(void)
19{
Frederic Weisbecker58135f52013-11-06 14:45:57 +010020 if (context_tracking_is_enabled())
Frederic Weisbeckerad657822013-07-10 02:44:35 +020021 context_tracking_user_enter();
22
23}
24static inline void user_exit(void)
25{
Frederic Weisbecker58135f52013-11-06 14:45:57 +010026 if (context_tracking_is_enabled())
Frederic Weisbeckerad657822013-07-10 02:44:35 +020027 context_tracking_user_exit();
28}
Frederic Weisbecker56dd9472013-02-24 00:23:25 +010029
Frederic Weisbecker6c1e0252013-02-24 01:19:14 +010030static inline enum ctx_state exception_enter(void)
Frederic Weisbecker56dd9472013-02-24 00:23:25 +010031{
Frederic Weisbecker6c1e0252013-02-24 01:19:14 +010032 enum ctx_state prev_ctx;
33
Frederic Weisbecker58135f52013-11-06 14:45:57 +010034 if (!context_tracking_is_enabled())
Frederic Weisbeckerad657822013-07-10 02:44:35 +020035 return 0;
36
Frederic Weisbecker6c1e0252013-02-24 01:19:14 +010037 prev_ctx = this_cpu_read(context_tracking.state);
Rik van Riel3aab4f52015-02-10 15:27:50 -050038 if (prev_ctx != CONTEXT_KERNEL)
39 context_tracking_exit(prev_ctx);
Frederic Weisbecker6c1e0252013-02-24 01:19:14 +010040
41 return prev_ctx;
Frederic Weisbecker56dd9472013-02-24 00:23:25 +010042}
43
Frederic Weisbecker6c1e0252013-02-24 01:19:14 +010044static inline void exception_exit(enum ctx_state prev_ctx)
Frederic Weisbecker56dd9472013-02-24 00:23:25 +010045{
Frederic Weisbecker58135f52013-11-06 14:45:57 +010046 if (context_tracking_is_enabled()) {
Rik van Riel3aab4f52015-02-10 15:27:50 -050047 if (prev_ctx != CONTEXT_KERNEL)
48 context_tracking_enter(prev_ctx);
Frederic Weisbeckerad657822013-07-10 02:44:35 +020049 }
Frederic Weisbecker56dd9472013-02-24 00:23:25 +010050}
51
Frederic Weisbecker91d1aa432012-11-27 19:33:25 +010052#else
Frederic Weisbecker91d1aa432012-11-27 19:33:25 +010053static inline void user_enter(void) { }
54static inline void user_exit(void) { }
Frederic Weisbecker6c1e0252013-02-24 01:19:14 +010055static inline enum ctx_state exception_enter(void) { return 0; }
56static inline void exception_exit(enum ctx_state prev_ctx) { }
Frederic Weisbecker91d1aa432012-11-27 19:33:25 +010057#endif /* !CONFIG_CONTEXT_TRACKING */
58
Frederic Weisbecker65f382f2013-07-11 19:12:32 +020059
60#ifdef CONFIG_CONTEXT_TRACKING_FORCE
61extern void context_tracking_init(void);
62#else
63static inline void context_tracking_init(void) { }
64#endif /* CONFIG_CONTEXT_TRACKING_FORCE */
65
66
Frederic Weisbecker2d854e52013-07-12 19:02:30 +020067#ifdef CONFIG_VIRT_CPU_ACCOUNTING_GEN
Frederic Weisbecker48d6a812013-07-10 02:44:35 +020068static inline void guest_enter(void)
69{
Frederic Weisbeckerb0493402013-07-12 03:10:15 +020070 if (vtime_accounting_enabled())
Frederic Weisbecker48d6a812013-07-10 02:44:35 +020071 vtime_guest_enter(current);
72 else
73 current->flags |= PF_VCPU;
Rik van Riel126a6a52015-02-10 15:27:54 -050074
75 if (context_tracking_is_enabled())
76 context_tracking_enter(CONTEXT_GUEST);
Frederic Weisbecker48d6a812013-07-10 02:44:35 +020077}
78
79static inline void guest_exit(void)
80{
Rik van Riel126a6a52015-02-10 15:27:54 -050081 if (context_tracking_is_enabled())
82 context_tracking_exit(CONTEXT_GUEST);
83
Frederic Weisbeckerb0493402013-07-12 03:10:15 +020084 if (vtime_accounting_enabled())
Frederic Weisbecker48d6a812013-07-10 02:44:35 +020085 vtime_guest_exit(current);
86 else
87 current->flags &= ~PF_VCPU;
88}
Frederic Weisbecker73d424f2013-07-11 19:42:13 +020089
Frederic Weisbecker2d854e52013-07-12 19:02:30 +020090#else
91static inline void guest_enter(void)
92{
93 /*
Frederic Weisbecker5b206d42013-07-12 19:05:14 +020094 * This is running in ioctl context so its safe
95 * to assume that it's the stime pending cputime
96 * to flush.
Frederic Weisbecker2d854e52013-07-12 19:02:30 +020097 */
98 vtime_account_system(current);
99 current->flags |= PF_VCPU;
100}
101
102static inline void guest_exit(void)
103{
Frederic Weisbecker5b206d42013-07-12 19:05:14 +0200104 /* Flush the guest cputime we spent on the guest */
Frederic Weisbecker2d854e52013-07-12 19:02:30 +0200105 vtime_account_system(current);
106 current->flags &= ~PF_VCPU;
107}
108#endif /* CONFIG_VIRT_CPU_ACCOUNTING_GEN */
109
Frederic Weisbecker91d1aa432012-11-27 19:33:25 +0100110#endif