KVM: Check the allocation of pv cpu mask (0f990222) · Commits · 戴 / test

arch/x86/kernel/kvm.c

+19 −3

Original line number	Diff line number	Diff line
		@@ -654,7 +654,6 @@ static void __init kvm_guest_init(void)
		}

		if (pv_tlb_flush_supported()) {
		pv_ops.mmu.flush_tlb_others = kvm_flush_tlb_others;
		pv_ops.mmu.tlb_remove_table = tlb_remove_table;
		pr_info("KVM setup pv remote TLB flush\n");
		}
		@@ -767,6 +766,14 @@ static __init int activate_jump_labels(void)
		}
		arch_initcall(activate_jump_labels);

		static void kvm_free_pv_cpu_mask(void)
		{
		unsigned int cpu;

		for_each_possible_cpu(cpu)
		free_cpumask_var(per_cpu(__pv_cpu_mask, cpu));
		}

		static __init int kvm_alloc_cpumask(void)
		{
		int cpu;
		@@ -785,11 +792,20 @@ static __init int kvm_alloc_cpumask(void)

		if (alloc)
		for_each_possible_cpu(cpu) {
		zalloc_cpumask_var_node(per_cpu_ptr(&__pv_cpu_mask, cpu),
		GFP_KERNEL, cpu_to_node(cpu));
		if (!zalloc_cpumask_var_node(
		per_cpu_ptr(&__pv_cpu_mask, cpu),
		GFP_KERNEL, cpu_to_node(cpu))) {
		goto zalloc_cpumask_fail;
		}
		}

		apic->send_IPI_mask_allbutself = kvm_send_ipi_mask_allbutself;
		pv_ops.mmu.flush_tlb_others = kvm_flush_tlb_others;
		return 0;

		zalloc_cpumask_fail:
		kvm_free_pv_cpu_mask();
		return -ENOMEM;
		}
		arch_initcall(kvm_alloc_cpumask);