/* * Generic helpers for smp ipi calls * * (C) Jens Axboe <jens.axboe@oracle.com> 2008 * */ #include <linux/init.h> #include <linux/module.h> #include <linux/percpu.h> #include <linux/rcupdate.h> #include <linux/rculist.h> #include <linux/smp.h> static DEFINE_PER_CPU(struct call_single_queue, call_single_queue); static LIST_HEAD(call_function_queue); __cacheline_aligned_in_smp DEFINE_SPINLOCK(call_function_lock); enum { CSD_FLAG_WAIT = 0x01, CSD_FLAG_ALLOC = 0x02, }; struct call_function_data { struct call_single_data csd; spinlock_t lock; unsigned int refs; cpumask_t cpumask; struct rcu_head rcu_head; }; struct call_single_queue { struct list_head list; spinlock_t lock; }; void __cpuinit init_call_single_data(void) { int i; for_each_possible_cpu(i) { struct call_single_queue *q = &per_cpu(call_single_queue, i); spin_lock_init(&q->lock); INIT_LIST_HEAD(&q->list); } } static void csd_flag_wait(struct call_single_data *data) { /* Wait for response */ do { /* * We need to see the flags store in the IPI handler */ smp_mb(); if (!(data->flags & CSD_FLAG_WAIT)) break; cpu_relax(); } while (1); } /* * Insert a previously allocated call_single_data element for execution * on the given CPU. data must already have ->func, ->info, and ->flags set. */ static void generic_exec_single(int cpu, struct call_single_data *data) { struct call_single_queue *dst = &per_cpu(call_single_queue, cpu); int wait = data->flags & CSD_FLAG_WAIT, ipi; unsigned long flags; spin_lock_irqsave(&dst->lock, flags); ipi = list_empty(&dst->list); list_add_tail(&data->list, &dst->list); spin_unlock_irqrestore(&dst->lock, flags); if (ipi) arch_send_call_function_single_ipi(cpu); if (wait) csd_flag_wait(data); } static void rcu_free_call_data(struct rcu_head *head) { struct call_function_data *data; data = container_of(head, struct call_function_data, rcu_head); kfree(data); } /* * Invoked by arch to handle an IPI for call function. Must be called with * interrupts disabled. */ void generic_smp_call_function_interrupt(void) { struct call_function_data *data; int cpu = get_cpu(); /* * It's ok to use list_for_each_rcu() here even though we may delete * 'pos', since list_del_rcu() doesn't clear ->next */ rcu_read_lock(); list_for_each_entry_rcu(data, &call_function_queue, csd.list) { int refs; if (!cpu_isset(cpu, data->cpumask)) continue; data->csd.func(data->csd.info); spin_lock(&data->lock); cpu_clear(cpu, data->cpumask); WARN_ON(data->refs == 0); data->refs--; refs = data->refs; spin_unlock(&data->lock); if (refs) continue; spin_lock(&call_function_lock); list_del_rcu(&data->csd.list); spin_unlock(&call_function_lock); if (data->csd.flags & CSD_FLAG_WAIT) { /* * serialize stores to data with the flag clear * and wakeup */ smp_wmb(); data->csd.flags &= ~CSD_FLAG_WAIT; } else call_rcu(&data->rcu_head, rcu_free_call_data); } rcu_read_unlock(); put_cpu(); } /* * Invoked by arch to handle an IPI for call function single. Must be called * from the arch with interrupts disabled. */ void generic_smp_call_function_single_interrupt(void) { struct call_single_queue *q = &__get_cpu_var(call_single_queue); LIST_HEAD(list); /* * Need to see other stores to list head for checking whether * list is empty without holding q->lock */ smp_mb(); while (!list_empty(&q->list)) { unsigned int data_flags; spin_lock(&q->lock); list_replace_init(&q->list, &list); spin_unlock(&q->lock); while (!list_empty(&list)) { struct call_single_data *data; data = list_entry(list.next, struct call_single_data, list); list_del(&data->list); /* * 'data' can be invalid after this call if * flags == 0 (when called through * generic_exec_single(), so save them away before * making the call. */ data_flags = data->flags; data->func(data->info); if (data_flags & CSD_FLAG_WAIT) { smp_wmb(); data->flags &= ~CSD_FLAG_WAIT; } else if (data_flags & CSD_FLAG_ALLOC) kfree(data); } /* * See comment on outer loop */ smp_mb(); } } /* * smp_call_function_single - Run a function on a specific CPU * @func: The function to run. This must be fast and non-blocking. * @info: An arbitrary pointer to pass to the function. * @wait: If true, wait until function has completed on other CPUs. * * Returns 0 on success, else a negative status code. Note that @wait * will be implicitly turned on in case of allocation failures, since * we fall back to on-stack allocation. */ int smp_call_function_single(int cpu, void (*func) (void *info), void *info, int wait) { struct call_single_data d; unsigned long flags; /* prevent preemption and reschedule on another processor */ int me = get_cpu(); /* Can deadlock when called with interrupts disabled */ WARN_ON(irqs_disabled()); if (cpu == me) { local_irq_save(flags); func(info); local_irq_restore(flags); } else { struct call_single_data *data = NULL; if (!wait) { data = kmalloc(sizeof(*data), GFP_ATOMIC); if (data) data->flags = CSD_FLAG_ALLOC; } if (!data) { data = &d; data->flags = CSD_FLAG_WAIT; } data->func = func; data->info = info; generic_exec_single(cpu, data); } put_cpu(); return 0; } EXPORT_SYMBOL(smp_call_function_single); /** * __smp_call_function_single(): Run a function on another CPU * @cpu: The CPU to run on. * @data: Pre-allocated and setup data structure * * Like smp_call_function_single(), but allow caller to pass in a pre-allocated * data structure. Useful for embedding @data inside other structures, for * instance. * */ void __smp_call_function_single(int cpu, struct call_single_data *data) { /* Can deadlock when called with interrupts disabled */ WARN_ON((data->flags & CSD_FLAG_WAIT) && irqs_disabled()); generic_exec_single(cpu, data); } /** * smp_call_function_mask(): Run a function on a set of other CPUs. * @mask: The set of cpus to run on. * @func: The function to run. This must be fast and non-blocking. * @info: An arbitrary pointer to pass to the function. * @wait: If true, wait (atomically) until function has completed on other CPUs. * * Returns 0 on success, else a negative status code. * * If @wait is true, then returns once @func has returned. Note that @wait * will be implicitly turned on in case of allocation failures, since * we fall back to on-stack allocation. * * You must not call this function with disabled interrupts or from a * hardware interrupt handler or from a bottom half handler. Preemption * must be disabled when calling this function. */ int smp_call_function_mask(cpumask_t mask, void (*func)(void *), void *info, int wait) { struct call_function_data d; struct call_function_data *data = NULL; cpumask_t allbutself; unsigned long flags; int cpu, num_cpus; /* Can deadlock when called with interrupts disabled */ WARN_ON(irqs_disabled()); cpu = smp_processor_id(); allbutself = cpu_online_map; cpu_clear(cpu, allbutself); cpus_and(mask, mask, allbutself); num_cpus = cpus_weight(mask); /* * If zero CPUs, return. If just a single CPU, turn this request * into a targetted single call instead since it's faster. */ if (!num_cpus) return 0; else if (num_cpus == 1) { cpu = first_cpu(mask); return smp_call_function_single(cpu, func, info, wait); } if (!wait) { data = kmalloc(sizeof(*data), GFP_ATOMIC); if (data) data->csd.flags = CSD_FLAG_ALLOC; } if (!data) { data = &d; data->csd.flags = CSD_FLAG_WAIT; wait = 1; } spin_lock_init(&data->lock); data->csd.func = func; data->csd.info = info; data->refs = num_cpus; data->cpumask = mask; spin_lock_irqsave(&call_function_lock, flags); list_add_tail_rcu(&data->csd.list, &call_function_queue); spin_unlock_irqrestore(&call_function_lock, flags); /* Send a message to all CPUs in the map */ arch_send_call_function_ipi(mask); /* optionally wait for the CPUs to complete */ if (wait) csd_flag_wait(&data->csd); return 0; } EXPORT_SYMBOL(smp_call_function_mask); /** * smp_call_function(): Run a function on all other CPUs. * @func: The function to run. This must be fast and non-blocking. * @info: An arbitrary pointer to pass to the function. * @wait: If true, wait (atomically) until function has completed on other CPUs. * * Returns 0 on success, else a negative status code. * * If @wait is true, then returns once @func has returned; otherwise * it returns just before the target cpu calls @func. In case of allocation * failure, @wait will be implicitly turned on. * * You must not call this function with disabled interrupts or from a * hardware interrupt handler or from a bottom half handler. */ int smp_call_function(void (*func)(void *), void *info, int wait) { int ret; preempt_disable(); ret = smp_call_function_mask(cpu_online_map, func, info, wait); preempt_enable(); return ret; } EXPORT_SYMBOL(smp_call_function); void ipi_call_lock(void) { spin_lock(&call_function_lock); } void ipi_call_unlock(void) { spin_unlock(&call_function_lock); } void ipi_call_lock_irq(void) { spin_lock_irq(&call_function_lock); } void ipi_call_unlock_irq(void) { spin_unlock_irq(&call_function_lock); }