让程序跑在特定的cpu上在kernel space的实现

如果需要将程序跑在特定的cpu上，可以参考下面的code实现in_interrupt
int my_func_on_cpu(int cpu)
{
int err;
get_online_cpus();
if (!cpu_online(cpu))
err = -EINVAL;
else
if（in_interrupt（））
err = work_on_cpu(cpu, __my_func_on_cpu, NULL);
else
smp_call_function_single(cpu, __my_func_on_cpu, &err,
true);
endif
put_online_cpus();
return err;
}
如果确定你的code 是在interrupt context 就用smp_call_function_single。如果跑在thread context就用work_on_cpu，因为work_on_cpu
可能会sleep，所以不能用在interrupt context中.
如果两种环境都有可能跑到就用上面这种方式就可以，即用in_interrupt（）函数来确定是否在interrupt context 中.
long work_on_cpu(int cpu, long (*fn)(void *), void *arg)
{
struct work_for_cpu wfc = { .fn = fn, .arg = arg };


INIT_WORK_ONSTACK(&wfc.work, work_for_cpu_fn);
schedule_work_on(cpu, &wfc.work);
flush_work(&wfc.work);
destroy_work_on_stack(&wfc.work);
return wfc.ret;
}


work_on_cpu 首先在stack上建议一个work_struct。
#define INIT_WORK_ONSTACK(_work, _func) \
__INIT_WORK((_work), (_func), 1)
我们没有定义CONFIG_LOCKDEP。所以我们走的是else的case
#ifdef CONFIG_LOCKDEP
#define __INIT_WORK(_work, _func, _onstack) \
do { \
static struct lock_class_key __key;\
\
__init_work((_work), _onstack);\
(_work)->data = (atomic_long_t) WORK_DATA_INIT();\
lockdep_init_map(&(_work)->lockdep_map, #_work, &__key, 0); \
INIT_LIST_HEAD(&(_work)->entry);\
(_work)->func = (_func);\
} while (0)
#else
#define __INIT_WORK(_work, _func, _onstack) \
do { \
__init_work((_work), _onstack);\
(_work)->data = (atomic_long_t) WORK_DATA_INIT();\
INIT_LIST_HEAD(&(_work)->entry);\
(_work)->func = (_func);\
} while (0)
#endif


这个workqueue的回调函数是
static void work_for_cpu_fn(struct work_struct *work)
{
struct work_for_cpu *wfc = container_of(work, struct work_for_cpu, work);


wfc->ret = wfc->fn(wfc->arg);
}


也就是直接调用用户注册的函数.
其次调用schedule_work_on(cpu, &wfc.work);来运行workqueue
static inline bool schedule_work_on(int cpu, struct work_struct *work)
{
return queue_work_on(cpu, system_wq, work);
}
schedule_work_on 是将我们前面在stack上建立的work_struct 放到system_wq 这个系统已经定义好的workqueue_struct 中来运行.
最后调用flush_work(&wfc.work);来等待wfc.work 执行完成.
由于我们的work_struct是建立在stack上的，因此work_on_cpu 函数返回的时候work_struct 就会被自动销毁
因此在没有定义CONFIG_DEBUG_OBJECTS_WORK来debug的时候destroy_work_on_stack是一个空函数