Frederic Weisbecker | 91d1aa43 | 2012-11-27 19:33:25 +0100 | [diff] [blame] | 1 | #ifndef _LINUX_CONTEXT_TRACKING_H |
| 2 | #define _LINUX_CONTEXT_TRACKING_H |
| 3 | |
Frederic Weisbecker | 91d1aa43 | 2012-11-27 19:33:25 +0100 | [diff] [blame] | 4 | #include <linux/sched.h> |
Frederic Weisbecker | 521921b | 2013-05-16 01:21:38 +0200 | [diff] [blame] | 5 | #include <linux/vtime.h> |
Frederic Weisbecker | e7358b3 | 2013-07-12 02:15:49 +0200 | [diff] [blame] | 6 | #include <linux/context_tracking_state.h> |
Frederic Weisbecker | 56dd947 | 2013-02-24 00:23:25 +0100 | [diff] [blame] | 7 | #include <asm/ptrace.h> |
Frederic Weisbecker | 95a79fd | 2013-01-07 18:12:14 +0100 | [diff] [blame] | 8 | |
Frederic Weisbecker | 521921b | 2013-05-16 01:21:38 +0200 | [diff] [blame] | 9 | |
Frederic Weisbecker | 6c1e025 | 2013-02-24 01:19:14 +0100 | [diff] [blame] | 10 | #ifdef CONFIG_CONTEXT_TRACKING |
Frederic Weisbecker | 2e70933 | 2013-07-10 00:55:25 +0200 | [diff] [blame] | 11 | extern void context_tracking_cpu_set(int cpu); |
| 12 | |
Rik van Riel | 3aab4f5 | 2015-02-10 15:27:50 -0500 | [diff] [blame] | 13 | extern void context_tracking_enter(enum ctx_state state); |
| 14 | extern void context_tracking_exit(enum ctx_state state); |
Frederic Weisbecker | ad65782 | 2013-07-10 02:44:35 +0200 | [diff] [blame] | 15 | extern void context_tracking_user_enter(void); |
| 16 | extern void context_tracking_user_exit(void); |
| 17 | |
| 18 | static inline void user_enter(void) |
| 19 | { |
Frederic Weisbecker | 58135f5 | 2013-11-06 14:45:57 +0100 | [diff] [blame] | 20 | if (context_tracking_is_enabled()) |
Frederic Weisbecker | ad65782 | 2013-07-10 02:44:35 +0200 | [diff] [blame] | 21 | context_tracking_user_enter(); |
| 22 | |
| 23 | } |
| 24 | static inline void user_exit(void) |
| 25 | { |
Frederic Weisbecker | 58135f5 | 2013-11-06 14:45:57 +0100 | [diff] [blame] | 26 | if (context_tracking_is_enabled()) |
Frederic Weisbecker | ad65782 | 2013-07-10 02:44:35 +0200 | [diff] [blame] | 27 | context_tracking_user_exit(); |
| 28 | } |
Frederic Weisbecker | 56dd947 | 2013-02-24 00:23:25 +0100 | [diff] [blame] | 29 | |
Frederic Weisbecker | 6c1e025 | 2013-02-24 01:19:14 +0100 | [diff] [blame] | 30 | static inline enum ctx_state exception_enter(void) |
Frederic Weisbecker | 56dd947 | 2013-02-24 00:23:25 +0100 | [diff] [blame] | 31 | { |
Frederic Weisbecker | 6c1e025 | 2013-02-24 01:19:14 +0100 | [diff] [blame] | 32 | enum ctx_state prev_ctx; |
| 33 | |
Frederic Weisbecker | 58135f5 | 2013-11-06 14:45:57 +0100 | [diff] [blame] | 34 | if (!context_tracking_is_enabled()) |
Frederic Weisbecker | ad65782 | 2013-07-10 02:44:35 +0200 | [diff] [blame] | 35 | return 0; |
| 36 | |
Frederic Weisbecker | 6c1e025 | 2013-02-24 01:19:14 +0100 | [diff] [blame] | 37 | prev_ctx = this_cpu_read(context_tracking.state); |
Rik van Riel | 3aab4f5 | 2015-02-10 15:27:50 -0500 | [diff] [blame] | 38 | if (prev_ctx != CONTEXT_KERNEL) |
| 39 | context_tracking_exit(prev_ctx); |
Frederic Weisbecker | 6c1e025 | 2013-02-24 01:19:14 +0100 | [diff] [blame] | 40 | |
| 41 | return prev_ctx; |
Frederic Weisbecker | 56dd947 | 2013-02-24 00:23:25 +0100 | [diff] [blame] | 42 | } |
| 43 | |
Frederic Weisbecker | 6c1e025 | 2013-02-24 01:19:14 +0100 | [diff] [blame] | 44 | static inline void exception_exit(enum ctx_state prev_ctx) |
Frederic Weisbecker | 56dd947 | 2013-02-24 00:23:25 +0100 | [diff] [blame] | 45 | { |
Frederic Weisbecker | 58135f5 | 2013-11-06 14:45:57 +0100 | [diff] [blame] | 46 | if (context_tracking_is_enabled()) { |
Rik van Riel | 3aab4f5 | 2015-02-10 15:27:50 -0500 | [diff] [blame] | 47 | if (prev_ctx != CONTEXT_KERNEL) |
| 48 | context_tracking_enter(prev_ctx); |
Frederic Weisbecker | ad65782 | 2013-07-10 02:44:35 +0200 | [diff] [blame] | 49 | } |
Frederic Weisbecker | 56dd947 | 2013-02-24 00:23:25 +0100 | [diff] [blame] | 50 | } |
| 51 | |
Frederic Weisbecker | 91d1aa43 | 2012-11-27 19:33:25 +0100 | [diff] [blame] | 52 | #else |
Frederic Weisbecker | 91d1aa43 | 2012-11-27 19:33:25 +0100 | [diff] [blame] | 53 | static inline void user_enter(void) { } |
| 54 | static inline void user_exit(void) { } |
Frederic Weisbecker | 6c1e025 | 2013-02-24 01:19:14 +0100 | [diff] [blame] | 55 | static inline enum ctx_state exception_enter(void) { return 0; } |
| 56 | static inline void exception_exit(enum ctx_state prev_ctx) { } |
Frederic Weisbecker | 91d1aa43 | 2012-11-27 19:33:25 +0100 | [diff] [blame] | 57 | #endif /* !CONFIG_CONTEXT_TRACKING */ |
| 58 | |
Frederic Weisbecker | 65f382f | 2013-07-11 19:12:32 +0200 | [diff] [blame] | 59 | |
| 60 | #ifdef CONFIG_CONTEXT_TRACKING_FORCE |
| 61 | extern void context_tracking_init(void); |
| 62 | #else |
| 63 | static inline void context_tracking_init(void) { } |
| 64 | #endif /* CONFIG_CONTEXT_TRACKING_FORCE */ |
| 65 | |
| 66 | |
Frederic Weisbecker | 2d854e5 | 2013-07-12 19:02:30 +0200 | [diff] [blame] | 67 | #ifdef CONFIG_VIRT_CPU_ACCOUNTING_GEN |
Frederic Weisbecker | 48d6a81 | 2013-07-10 02:44:35 +0200 | [diff] [blame] | 68 | static inline void guest_enter(void) |
| 69 | { |
Frederic Weisbecker | b049340 | 2013-07-12 03:10:15 +0200 | [diff] [blame] | 70 | if (vtime_accounting_enabled()) |
Frederic Weisbecker | 48d6a81 | 2013-07-10 02:44:35 +0200 | [diff] [blame] | 71 | vtime_guest_enter(current); |
| 72 | else |
| 73 | current->flags |= PF_VCPU; |
Rik van Riel | 126a6a5 | 2015-02-10 15:27:54 -0500 | [diff] [blame] | 74 | |
| 75 | if (context_tracking_is_enabled()) |
| 76 | context_tracking_enter(CONTEXT_GUEST); |
Frederic Weisbecker | 48d6a81 | 2013-07-10 02:44:35 +0200 | [diff] [blame] | 77 | } |
| 78 | |
| 79 | static inline void guest_exit(void) |
| 80 | { |
Rik van Riel | 126a6a5 | 2015-02-10 15:27:54 -0500 | [diff] [blame] | 81 | if (context_tracking_is_enabled()) |
| 82 | context_tracking_exit(CONTEXT_GUEST); |
| 83 | |
Frederic Weisbecker | b049340 | 2013-07-12 03:10:15 +0200 | [diff] [blame] | 84 | if (vtime_accounting_enabled()) |
Frederic Weisbecker | 48d6a81 | 2013-07-10 02:44:35 +0200 | [diff] [blame] | 85 | vtime_guest_exit(current); |
| 86 | else |
| 87 | current->flags &= ~PF_VCPU; |
| 88 | } |
Frederic Weisbecker | 73d424f | 2013-07-11 19:42:13 +0200 | [diff] [blame] | 89 | |
Frederic Weisbecker | 2d854e5 | 2013-07-12 19:02:30 +0200 | [diff] [blame] | 90 | #else |
| 91 | static inline void guest_enter(void) |
| 92 | { |
| 93 | /* |
Frederic Weisbecker | 5b206d4 | 2013-07-12 19:05:14 +0200 | [diff] [blame] | 94 | * This is running in ioctl context so its safe |
| 95 | * to assume that it's the stime pending cputime |
| 96 | * to flush. |
Frederic Weisbecker | 2d854e5 | 2013-07-12 19:02:30 +0200 | [diff] [blame] | 97 | */ |
| 98 | vtime_account_system(current); |
| 99 | current->flags |= PF_VCPU; |
| 100 | } |
| 101 | |
| 102 | static inline void guest_exit(void) |
| 103 | { |
Frederic Weisbecker | 5b206d4 | 2013-07-12 19:05:14 +0200 | [diff] [blame] | 104 | /* Flush the guest cputime we spent on the guest */ |
Frederic Weisbecker | 2d854e5 | 2013-07-12 19:02:30 +0200 | [diff] [blame] | 105 | vtime_account_system(current); |
| 106 | current->flags &= ~PF_VCPU; |
| 107 | } |
| 108 | #endif /* CONFIG_VIRT_CPU_ACCOUNTING_GEN */ |
| 109 | |
Frederic Weisbecker | 91d1aa43 | 2012-11-27 19:33:25 +0100 | [diff] [blame] | 110 | #endif |