如何在所有 CPU 上执行一段内核代码？

Question

我正在尝试制作一个内核模块来为 x87 FPU 启用 FOP 兼容模式。这是通过设置IA32_MISC_ENABLE MSR 中的第 2 位来完成的。代码如下：

#include <linux/init.h>
#include <linux/module.h>
#include <linux/kernel.h>
#include <asm/msr-index.h>
#include <asm/msr.h>

MODULE_LICENSE("GPL");
MODULE_AUTHOR("10110111");
MODULE_DESCRIPTION("Module to enable FOPcode compatibility mode");
MODULE_VERSION("0.1");

static int __init fopCompat_init(void)
{
   unsigned long long misc_enable=native_read_msr(MSR_IA32_MISC_ENABLE);
   printk(KERN_INFO "Before trying to set FOP_COMPAT, IA32_MISC_ENABLE=%llx,"
                    " i.e. FOP_COMPAT is %senabled\n"
                    ,misc_enable,misc_enable&MSR_IA32_MISC_ENABLE_X87_COMPAT?"":"NOT ");

   wrmsrl(MSR_IA32_MISC_ENABLE,misc_enable|MSR_IA32_MISC_ENABLE_X87_COMPAT);
   misc_enable=native_read_msr(MSR_IA32_MISC_ENABLE);

   printk(KERN_INFO "Tried to set FOP_COMPAT. Result: IA32_MISC_ENABLE=%llx,"
                    " i.e. FOP_COMPAT is now %senabled\n"
                    ,misc_enable,misc_enable&MSR_IA32_MISC_ENABLE_X87_COMPAT?"":"NOT ");
   return 0;
}

static void __exit fopCompat_exit(void)
{
   const unsigned long long misc_enable=native_read_msr(MSR_IA32_MISC_ENABLE);
   printk(KERN_INFO "Quitting FOP-compat with IA32_MISC_ENABLE=%llx\n",misc_enable);
   if(!(misc_enable & MSR_IA32_MISC_ENABLE_X87_COMPAT))
       printk(KERN_INFO "NOTE: seems some CPUs still have to be set up, "
                        "or compatibility mode will work inconsistently\n");
   printk(KERN_INFO "\n");
}

module_init(fopCompat_init);
module_exit(fopCompat_exit);

它似乎有效，但在多个insmod/rmmod 循环中，我有时会得到dmesg 输出，表明兼容模式仍未启用，尽管它是在执行wrmsr 之后立即启用的。经过一番思考，我意识到这是因为模块代码是在不同的逻辑 CPU 上执行的（我有 Core i7 和 4 核 *HT=8 逻辑 CPU），所以我有 1/8 的机会在做 @ 时获得“启用”打印987654326@。在重复循环大约 20 次后，我得到了一致的“启用”打印，并且我的用户空间应用程序很高兴地使用它。

所以现在我的问题是：如何让我的代码在系统上存在的所有逻辑 CPU 上执行，以便为所有这些 CPU 启用兼容模式？

Answer 1

要在每个 CPU 上执行代码，请使用 on_each_cpu 函数。

签名：

int on_each_cpu(void (*func) (void *info), void *info, int wait)

说明：

在所有处理器上调用一个函数。

如果wait 参数不为零，则在所有 CPU 上等待函数完成。

函数func 不应该休眠，但整个on_each_cpu() 调用不应该在原子上下文中完成。