Rusty Russell | e5582ca | 2006-09-29 02:01:35 -0700 | [diff] [blame] | 1 | /* Copyright 2005 Rusty Russell rusty@rustcorp.com.au IBM Corporation. |
| 2 | * GPL v2 and any later version. |
| 3 | */ |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 4 | #include <linux/cpu.h> |
| 5 | #include <linux/err.h> |
Prarit Bhargava | ee527cd | 2007-05-08 00:25:08 -0700 | [diff] [blame] | 6 | #include <linux/kthread.h> |
| 7 | #include <linux/module.h> |
| 8 | #include <linux/sched.h> |
| 9 | #include <linux/stop_machine.h> |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 10 | #include <linux/syscalls.h> |
Benjamin Herrenschmidt | a12bb44 | 2007-05-10 22:22:47 -0700 | [diff] [blame] | 11 | #include <linux/interrupt.h> |
| 12 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 13 | #include <asm/atomic.h> |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 14 | #include <asm/uaccess.h> |
| 15 | |
| 16 | /* Since we effect priority and affinity (both of which are visible |
| 17 | * to, and settable by outside processes) we do indirection via a |
| 18 | * kthread. */ |
| 19 | |
| 20 | /* Thread to stop each CPU in user context. */ |
| 21 | enum stopmachine_state { |
| 22 | STOPMACHINE_WAIT, |
| 23 | STOPMACHINE_PREPARE, |
| 24 | STOPMACHINE_DISABLE_IRQ, |
| 25 | STOPMACHINE_EXIT, |
| 26 | }; |
| 27 | |
| 28 | static enum stopmachine_state stopmachine_state; |
| 29 | static unsigned int stopmachine_num_threads; |
| 30 | static atomic_t stopmachine_thread_ack; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 31 | |
Andrew Morton | d8cb7c1 | 2006-07-03 17:32:22 -0700 | [diff] [blame] | 32 | static int stopmachine(void *cpu) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 33 | { |
| 34 | int irqs_disabled = 0; |
| 35 | int prepared = 0; |
Mike Travis | 65c0118 | 2008-07-15 14:14:30 -0700 | [diff] [blame] | 36 | cpumask_of_cpu_ptr(cpumask, (int)(long)cpu); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 37 | |
Mike Travis | 65c0118 | 2008-07-15 14:14:30 -0700 | [diff] [blame] | 38 | set_cpus_allowed_ptr(current, cpumask); |
Andrew Morton | d8cb7c1 | 2006-07-03 17:32:22 -0700 | [diff] [blame] | 39 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 40 | /* Ack: we are alive */ |
akpm@osdl.org | d59dd46 | 2005-05-01 08:58:47 -0700 | [diff] [blame] | 41 | smp_mb(); /* Theoretically the ack = 0 might not be on this CPU yet. */ |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 42 | atomic_inc(&stopmachine_thread_ack); |
| 43 | |
| 44 | /* Simple state machine */ |
| 45 | while (stopmachine_state != STOPMACHINE_EXIT) { |
| 46 | if (stopmachine_state == STOPMACHINE_DISABLE_IRQ |
| 47 | && !irqs_disabled) { |
| 48 | local_irq_disable(); |
Benjamin Herrenschmidt | a12bb44 | 2007-05-10 22:22:47 -0700 | [diff] [blame] | 49 | hard_irq_disable(); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 50 | irqs_disabled = 1; |
| 51 | /* Ack: irqs disabled. */ |
akpm@osdl.org | d59dd46 | 2005-05-01 08:58:47 -0700 | [diff] [blame] | 52 | smp_mb(); /* Must read state first. */ |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 53 | atomic_inc(&stopmachine_thread_ack); |
| 54 | } else if (stopmachine_state == STOPMACHINE_PREPARE |
| 55 | && !prepared) { |
| 56 | /* Everyone is in place, hold CPU. */ |
| 57 | preempt_disable(); |
| 58 | prepared = 1; |
akpm@osdl.org | d59dd46 | 2005-05-01 08:58:47 -0700 | [diff] [blame] | 59 | smp_mb(); /* Must read state first. */ |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 60 | atomic_inc(&stopmachine_thread_ack); |
| 61 | } |
| 62 | /* Yield in first stage: migration threads need to |
| 63 | * help our sisters onto their CPUs. */ |
| 64 | if (!prepared && !irqs_disabled) |
| 65 | yield(); |
Christian Borntraeger | 3401a61e | 2008-05-08 15:20:38 +0200 | [diff] [blame] | 66 | cpu_relax(); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 67 | } |
| 68 | |
| 69 | /* Ack: we are exiting. */ |
akpm@osdl.org | d59dd46 | 2005-05-01 08:58:47 -0700 | [diff] [blame] | 70 | smp_mb(); /* Must read state first. */ |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 71 | atomic_inc(&stopmachine_thread_ack); |
| 72 | |
| 73 | if (irqs_disabled) |
| 74 | local_irq_enable(); |
| 75 | if (prepared) |
| 76 | preempt_enable(); |
| 77 | |
| 78 | return 0; |
| 79 | } |
| 80 | |
| 81 | /* Change the thread state */ |
| 82 | static void stopmachine_set_state(enum stopmachine_state state) |
| 83 | { |
| 84 | atomic_set(&stopmachine_thread_ack, 0); |
akpm@osdl.org | d59dd46 | 2005-05-01 08:58:47 -0700 | [diff] [blame] | 85 | smp_wmb(); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 86 | stopmachine_state = state; |
| 87 | while (atomic_read(&stopmachine_thread_ack) != stopmachine_num_threads) |
| 88 | cpu_relax(); |
| 89 | } |
| 90 | |
| 91 | static int stop_machine(void) |
| 92 | { |
Andrew Morton | d8cb7c1 | 2006-07-03 17:32:22 -0700 | [diff] [blame] | 93 | int i, ret = 0; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 94 | |
| 95 | atomic_set(&stopmachine_thread_ack, 0); |
| 96 | stopmachine_num_threads = 0; |
| 97 | stopmachine_state = STOPMACHINE_WAIT; |
| 98 | |
| 99 | for_each_online_cpu(i) { |
Ingo Molnar | 39c715b | 2005-06-21 17:14:34 -0700 | [diff] [blame] | 100 | if (i == raw_smp_processor_id()) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 101 | continue; |
Andrew Morton | d8cb7c1 | 2006-07-03 17:32:22 -0700 | [diff] [blame] | 102 | ret = kernel_thread(stopmachine, (void *)(long)i,CLONE_KERNEL); |
| 103 | if (ret < 0) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 104 | break; |
| 105 | stopmachine_num_threads++; |
| 106 | } |
| 107 | |
| 108 | /* Wait for them all to come to life. */ |
Christian Borntraeger | 3401a61e | 2008-05-08 15:20:38 +0200 | [diff] [blame] | 109 | while (atomic_read(&stopmachine_thread_ack) != stopmachine_num_threads) { |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 110 | yield(); |
Christian Borntraeger | 3401a61e | 2008-05-08 15:20:38 +0200 | [diff] [blame] | 111 | cpu_relax(); |
| 112 | } |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 113 | |
| 114 | /* If some failed, kill them all. */ |
| 115 | if (ret < 0) { |
| 116 | stopmachine_set_state(STOPMACHINE_EXIT); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 117 | return ret; |
| 118 | } |
| 119 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 120 | /* Now they are all started, make them hold the CPUs, ready. */ |
Kirill Korotaev | 4557398 | 2005-11-13 16:07:30 -0800 | [diff] [blame] | 121 | preempt_disable(); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 122 | stopmachine_set_state(STOPMACHINE_PREPARE); |
| 123 | |
| 124 | /* Make them disable irqs. */ |
Kirill Korotaev | 4557398 | 2005-11-13 16:07:30 -0800 | [diff] [blame] | 125 | local_irq_disable(); |
Benjamin Herrenschmidt | a12bb44 | 2007-05-10 22:22:47 -0700 | [diff] [blame] | 126 | hard_irq_disable(); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 127 | stopmachine_set_state(STOPMACHINE_DISABLE_IRQ); |
| 128 | |
| 129 | return 0; |
| 130 | } |
| 131 | |
| 132 | static void restart_machine(void) |
| 133 | { |
| 134 | stopmachine_set_state(STOPMACHINE_EXIT); |
| 135 | local_irq_enable(); |
Kirill Korotaev | 4557398 | 2005-11-13 16:07:30 -0800 | [diff] [blame] | 136 | preempt_enable_no_resched(); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 137 | } |
| 138 | |
Pavel Machek | f526448 | 2008-04-21 22:15:06 +0000 | [diff] [blame] | 139 | struct stop_machine_data { |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 140 | int (*fn)(void *); |
| 141 | void *data; |
| 142 | struct completion done; |
| 143 | }; |
| 144 | |
| 145 | static int do_stop(void *_smdata) |
| 146 | { |
| 147 | struct stop_machine_data *smdata = _smdata; |
| 148 | int ret; |
| 149 | |
| 150 | ret = stop_machine(); |
| 151 | if (ret == 0) { |
| 152 | ret = smdata->fn(smdata->data); |
| 153 | restart_machine(); |
| 154 | } |
| 155 | |
| 156 | /* We're done: you can kthread_stop us now */ |
| 157 | complete(&smdata->done); |
| 158 | |
| 159 | /* Wait for kthread_stop */ |
| 160 | set_current_state(TASK_INTERRUPTIBLE); |
| 161 | while (!kthread_should_stop()) { |
| 162 | schedule(); |
| 163 | set_current_state(TASK_INTERRUPTIBLE); |
| 164 | } |
| 165 | __set_current_state(TASK_RUNNING); |
| 166 | return ret; |
| 167 | } |
| 168 | |
| 169 | struct task_struct *__stop_machine_run(int (*fn)(void *), void *data, |
| 170 | unsigned int cpu) |
| 171 | { |
Daniel Walker | 6c6080f | 2008-02-06 01:37:41 -0800 | [diff] [blame] | 172 | static DEFINE_MUTEX(stopmachine_mutex); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 173 | struct stop_machine_data smdata; |
| 174 | struct task_struct *p; |
| 175 | |
| 176 | smdata.fn = fn; |
| 177 | smdata.data = data; |
| 178 | init_completion(&smdata.done); |
| 179 | |
Daniel Walker | 6c6080f | 2008-02-06 01:37:41 -0800 | [diff] [blame] | 180 | mutex_lock(&stopmachine_mutex); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 181 | |
| 182 | /* If they don't care which CPU fn runs on, bind to any online one. */ |
| 183 | if (cpu == NR_CPUS) |
Ingo Molnar | 39c715b | 2005-06-21 17:14:34 -0700 | [diff] [blame] | 184 | cpu = raw_smp_processor_id(); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 185 | |
| 186 | p = kthread_create(do_stop, &smdata, "kstopmachine"); |
| 187 | if (!IS_ERR(p)) { |
Satoru Takeuchi | 85653af | 2007-07-15 23:39:47 -0700 | [diff] [blame] | 188 | struct sched_param param = { .sched_priority = MAX_RT_PRIO-1 }; |
| 189 | |
| 190 | /* One high-prio thread per cpu. We'll do this one. */ |
Rusty Russell | 961ccdd | 2008-06-23 13:55:38 +1000 | [diff] [blame] | 191 | sched_setscheduler_nocheck(p, SCHED_FIFO, ¶m); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 192 | kthread_bind(p, cpu); |
| 193 | wake_up_process(p); |
| 194 | wait_for_completion(&smdata.done); |
| 195 | } |
Daniel Walker | 6c6080f | 2008-02-06 01:37:41 -0800 | [diff] [blame] | 196 | mutex_unlock(&stopmachine_mutex); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 197 | return p; |
| 198 | } |
| 199 | |
| 200 | int stop_machine_run(int (*fn)(void *), void *data, unsigned int cpu) |
| 201 | { |
| 202 | struct task_struct *p; |
| 203 | int ret; |
| 204 | |
| 205 | /* No CPUs can come up or down during this. */ |
Gautham R Shenoy | 86ef5c9 | 2008-01-25 21:08:02 +0100 | [diff] [blame] | 206 | get_online_cpus(); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 207 | p = __stop_machine_run(fn, data, cpu); |
| 208 | if (!IS_ERR(p)) |
| 209 | ret = kthread_stop(p); |
| 210 | else |
| 211 | ret = PTR_ERR(p); |
Gautham R Shenoy | 86ef5c9 | 2008-01-25 21:08:02 +0100 | [diff] [blame] | 212 | put_online_cpus(); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 213 | |
| 214 | return ret; |
| 215 | } |
Prarit Bhargava | ee527cd | 2007-05-08 00:25:08 -0700 | [diff] [blame] | 216 | EXPORT_SYMBOL_GPL(stop_machine_run); |