Commit | Line | Data |
---|---|---|
7dfb7103 NC |
1 | /* Freezer declarations */ |
2 | ||
83144186 RW |
3 | #ifndef FREEZER_H_INCLUDED |
4 | #define FREEZER_H_INCLUDED | |
5 | ||
5c543eff | 6 | #include <linux/sched.h> |
e42837bc | 7 | #include <linux/wait.h> |
5c543eff | 8 | |
8174f150 | 9 | #ifdef CONFIG_FREEZER |
7dfb7103 NC |
10 | /* |
11 | * Check if a process has been frozen | |
12 | */ | |
13 | static inline int frozen(struct task_struct *p) | |
14 | { | |
15 | return p->flags & PF_FROZEN; | |
16 | } | |
17 | ||
18 | /* | |
19 | * Check if there is a request to freeze a process | |
20 | */ | |
21 | static inline int freezing(struct task_struct *p) | |
22 | { | |
8a102eed | 23 | return test_tsk_thread_flag(p, TIF_FREEZE); |
7dfb7103 NC |
24 | } |
25 | ||
26 | /* | |
27 | * Request that a process be frozen | |
7dfb7103 | 28 | */ |
0c1eecfb | 29 | static inline void set_freeze_flag(struct task_struct *p) |
7dfb7103 | 30 | { |
8a102eed | 31 | set_tsk_thread_flag(p, TIF_FREEZE); |
7dfb7103 NC |
32 | } |
33 | ||
34 | /* | |
35 | * Sometimes we may need to cancel the previous 'freeze' request | |
36 | */ | |
0c1eecfb | 37 | static inline void clear_freeze_flag(struct task_struct *p) |
7dfb7103 | 38 | { |
8a102eed | 39 | clear_tsk_thread_flag(p, TIF_FREEZE); |
7dfb7103 NC |
40 | } |
41 | ||
8174f150 MH |
42 | static inline bool should_send_signal(struct task_struct *p) |
43 | { | |
44 | return !(p->flags & PF_FREEZER_NOSIG); | |
45 | } | |
46 | ||
dc52ddc0 MH |
47 | /* Takes and releases task alloc lock using task_lock() */ |
48 | extern int thaw_process(struct task_struct *p); | |
7dfb7103 | 49 | |
8a32c441 | 50 | extern bool __refrigerator(bool check_kthr_stop); |
7dfb7103 | 51 | extern int freeze_processes(void); |
2aede851 | 52 | extern int freeze_kernel_threads(void); |
a9b6f562 | 53 | extern void thaw_processes(void); |
7dfb7103 | 54 | |
a0acae0e | 55 | static inline bool try_to_freeze(void) |
7dfb7103 | 56 | { |
a0acae0e TH |
57 | might_sleep(); |
58 | if (likely(!freezing(current))) | |
59 | return false; | |
8a32c441 | 60 | return __refrigerator(false); |
7dfb7103 | 61 | } |
ff39593a | 62 | |
8174f150 MH |
63 | extern bool freeze_task(struct task_struct *p, bool sig_only); |
64 | extern void cancel_freezing(struct task_struct *p); | |
65 | ||
dc52ddc0 | 66 | #ifdef CONFIG_CGROUP_FREEZER |
5a7aadfe | 67 | extern int cgroup_freezing_or_frozen(struct task_struct *task); |
dc52ddc0 | 68 | #else /* !CONFIG_CGROUP_FREEZER */ |
5a7aadfe MH |
69 | static inline int cgroup_freezing_or_frozen(struct task_struct *task) |
70 | { | |
71 | return 0; | |
72 | } | |
dc52ddc0 MH |
73 | #endif /* !CONFIG_CGROUP_FREEZER */ |
74 | ||
ba96a0c8 RW |
75 | /* |
76 | * The PF_FREEZER_SKIP flag should be set by a vfork parent right before it | |
77 | * calls wait_for_completion(&vfork) and reset right after it returns from this | |
78 | * function. Next, the parent should call try_to_freeze() to freeze itself | |
79 | * appropriately in case the child has exited before the freezing of tasks is | |
80 | * complete. However, we don't want kernel threads to be frozen in unexpected | |
81 | * places, so we allow them to block freeze_processes() instead or to set | |
82 | * PF_NOFREEZE if needed and PF_FREEZER_SKIP is only set for userland vfork | |
83 | * parents. Fortunately, in the ____call_usermodehelper() case the parent won't | |
84 | * really block freeze_processes(), since ____call_usermodehelper() (the child) | |
85 | * does a little before exec/exit and it can't be frozen before waking up the | |
86 | * parent. | |
87 | */ | |
88 | ||
89 | /* | |
90 | * If the current task is a user space one, tell the freezer not to count it as | |
91 | * freezable. | |
92 | */ | |
93 | static inline void freezer_do_not_count(void) | |
94 | { | |
95 | if (current->mm) | |
96 | current->flags |= PF_FREEZER_SKIP; | |
97 | } | |
98 | ||
99 | /* | |
100 | * If the current task is a user space one, tell the freezer to count it as | |
101 | * freezable again and try to freeze it. | |
102 | */ | |
103 | static inline void freezer_count(void) | |
104 | { | |
105 | if (current->mm) { | |
106 | current->flags &= ~PF_FREEZER_SKIP; | |
107 | try_to_freeze(); | |
108 | } | |
109 | } | |
110 | ||
111 | /* | |
58a69cb4 | 112 | * Check if the task should be counted as freezable by the freezer |
ba96a0c8 RW |
113 | */ |
114 | static inline int freezer_should_skip(struct task_struct *p) | |
115 | { | |
116 | return !!(p->flags & PF_FREEZER_SKIP); | |
117 | } | |
ff39593a | 118 | |
83144186 RW |
119 | /* |
120 | * Tell the freezer that the current task should be frozen by it | |
121 | */ | |
122 | static inline void set_freezable(void) | |
123 | { | |
124 | current->flags &= ~PF_NOFREEZE; | |
125 | } | |
126 | ||
ebb12db5 RW |
127 | /* |
128 | * Tell the freezer that the current task should be frozen by it and that it | |
129 | * should send a fake signal to the task to freeze it. | |
130 | */ | |
131 | static inline void set_freezable_with_signal(void) | |
132 | { | |
133 | current->flags &= ~(PF_NOFREEZE | PF_FREEZER_NOSIG); | |
134 | } | |
135 | ||
e42837bc | 136 | /* |
f06ac72e JL |
137 | * Freezer-friendly wrappers around wait_event_interruptible(), |
138 | * wait_event_killable() and wait_event_interruptible_timeout(), originally | |
139 | * defined in <linux/wait.h> | |
e42837bc RW |
140 | */ |
141 | ||
f06ac72e JL |
142 | #define wait_event_freezekillable(wq, condition) \ |
143 | ({ \ | |
144 | int __retval; \ | |
6f35c4ab ON |
145 | freezer_do_not_count(); \ |
146 | __retval = wait_event_killable(wq, (condition)); \ | |
147 | freezer_count(); \ | |
f06ac72e JL |
148 | __retval; \ |
149 | }) | |
150 | ||
e42837bc RW |
151 | #define wait_event_freezable(wq, condition) \ |
152 | ({ \ | |
153 | int __retval; \ | |
154 | do { \ | |
155 | __retval = wait_event_interruptible(wq, \ | |
156 | (condition) || freezing(current)); \ | |
157 | if (__retval && !freezing(current)) \ | |
158 | break; \ | |
159 | else if (!(condition)) \ | |
160 | __retval = -ERESTARTSYS; \ | |
161 | } while (try_to_freeze()); \ | |
162 | __retval; \ | |
163 | }) | |
164 | ||
165 | ||
166 | #define wait_event_freezable_timeout(wq, condition, timeout) \ | |
167 | ({ \ | |
168 | long __retval = timeout; \ | |
169 | do { \ | |
170 | __retval = wait_event_interruptible_timeout(wq, \ | |
171 | (condition) || freezing(current), \ | |
172 | __retval); \ | |
173 | } while (try_to_freeze()); \ | |
174 | __retval; \ | |
175 | }) | |
8174f150 | 176 | #else /* !CONFIG_FREEZER */ |
7dfb7103 NC |
177 | static inline int frozen(struct task_struct *p) { return 0; } |
178 | static inline int freezing(struct task_struct *p) { return 0; } | |
0c1eecfb RW |
179 | static inline void set_freeze_flag(struct task_struct *p) {} |
180 | static inline void clear_freeze_flag(struct task_struct *p) {} | |
7dfb7103 | 181 | static inline int thaw_process(struct task_struct *p) { return 1; } |
7dfb7103 | 182 | |
8a32c441 | 183 | static inline bool __refrigerator(bool check_kthr_stop) { return false; } |
2aede851 RW |
184 | static inline int freeze_processes(void) { return -ENOSYS; } |
185 | static inline int freeze_kernel_threads(void) { return -ENOSYS; } | |
7dfb7103 NC |
186 | static inline void thaw_processes(void) {} |
187 | ||
a0acae0e | 188 | static inline bool try_to_freeze(void) { return false; } |
7dfb7103 | 189 | |
ba96a0c8 RW |
190 | static inline void freezer_do_not_count(void) {} |
191 | static inline void freezer_count(void) {} | |
192 | static inline int freezer_should_skip(struct task_struct *p) { return 0; } | |
83144186 | 193 | static inline void set_freezable(void) {} |
ebb12db5 | 194 | static inline void set_freezable_with_signal(void) {} |
e42837bc RW |
195 | |
196 | #define wait_event_freezable(wq, condition) \ | |
197 | wait_event_interruptible(wq, condition) | |
198 | ||
199 | #define wait_event_freezable_timeout(wq, condition, timeout) \ | |
200 | wait_event_interruptible_timeout(wq, condition, timeout) | |
201 | ||
e0c8ea1a SF |
202 | #define wait_event_freezekillable(wq, condition) \ |
203 | wait_event_killable(wq, condition) | |
204 | ||
8174f150 | 205 | #endif /* !CONFIG_FREEZER */ |
83144186 RW |
206 | |
207 | #endif /* FREEZER_H_INCLUDED */ |