|
@@ -259,14 +259,15 @@ static int __init msr_init(void)
|
|
goto out_chrdev;
|
|
goto out_chrdev;
|
|
}
|
|
}
|
|
msr_class->devnode = msr_devnode;
|
|
msr_class->devnode = msr_devnode;
|
|
- get_online_cpus();
|
|
|
|
|
|
+
|
|
|
|
+ cpu_notifier_register_begin();
|
|
for_each_online_cpu(i) {
|
|
for_each_online_cpu(i) {
|
|
err = msr_device_create(i);
|
|
err = msr_device_create(i);
|
|
if (err != 0)
|
|
if (err != 0)
|
|
goto out_class;
|
|
goto out_class;
|
|
}
|
|
}
|
|
- register_hotcpu_notifier(&msr_class_cpu_notifier);
|
|
|
|
- put_online_cpus();
|
|
|
|
|
|
+ __register_hotcpu_notifier(&msr_class_cpu_notifier);
|
|
|
|
+ cpu_notifier_register_done();
|
|
|
|
|
|
err = 0;
|
|
err = 0;
|
|
goto out;
|
|
goto out;
|
|
@@ -275,7 +276,7 @@ out_class:
|
|
i = 0;
|
|
i = 0;
|
|
for_each_online_cpu(i)
|
|
for_each_online_cpu(i)
|
|
msr_device_destroy(i);
|
|
msr_device_destroy(i);
|
|
- put_online_cpus();
|
|
|
|
|
|
+ cpu_notifier_register_done();
|
|
class_destroy(msr_class);
|
|
class_destroy(msr_class);
|
|
out_chrdev:
|
|
out_chrdev:
|
|
__unregister_chrdev(MSR_MAJOR, 0, NR_CPUS, "cpu/msr");
|
|
__unregister_chrdev(MSR_MAJOR, 0, NR_CPUS, "cpu/msr");
|
|
@@ -286,13 +287,14 @@ out:
|
|
static void __exit msr_exit(void)
|
|
static void __exit msr_exit(void)
|
|
{
|
|
{
|
|
int cpu = 0;
|
|
int cpu = 0;
|
|
- get_online_cpus();
|
|
|
|
|
|
+
|
|
|
|
+ cpu_notifier_register_begin();
|
|
for_each_online_cpu(cpu)
|
|
for_each_online_cpu(cpu)
|
|
msr_device_destroy(cpu);
|
|
msr_device_destroy(cpu);
|
|
class_destroy(msr_class);
|
|
class_destroy(msr_class);
|
|
__unregister_chrdev(MSR_MAJOR, 0, NR_CPUS, "cpu/msr");
|
|
__unregister_chrdev(MSR_MAJOR, 0, NR_CPUS, "cpu/msr");
|
|
- unregister_hotcpu_notifier(&msr_class_cpu_notifier);
|
|
|
|
- put_online_cpus();
|
|
|
|
|
|
+ __unregister_hotcpu_notifier(&msr_class_cpu_notifier);
|
|
|
|
+ cpu_notifier_register_done();
|
|
}
|
|
}
|
|
|
|
|
|
module_init(msr_init);
|
|
module_init(msr_init);
|