Merge tag 'v4.4.214' into 10

[sagit-ice-cold/kernel_xiaomi_msm8998.git] / arch / x86 / kvm / lapic.c

diff --git a/arch/x86/kvm/lapic.c b/arch/x86/kvm/lapic.c
index 3c70f6c..ce8c4ae 100644 (file)
--- a/arch/x86/kvm/lapic.c
+++ b/arch/x86/kvm/lapic.c
@@ -36,6 +36,7 @@
 #include <asm/delay.h>
 #include <linux/atomic.h>
 #include <linux/jump_label.h>
+#include <linux/nospec.h>
 #include "kvm_cache_regs.h"
 #include "irq.h"
 #include "trace.h"
@@ -1432,15 +1433,21 @@ static int apic_reg_write(struct kvm_lapic *apic, u32 reg, u32 val)
        case APIC_LVTTHMR:
        case APIC_LVTPC:
        case APIC_LVT1:
-       case APIC_LVTERR:
+       case APIC_LVTERR: {
                /* TODO: Check vector */
+               size_t size;
+               u32 index;
+
                if (!kvm_apic_sw_enabled(apic))
                        val |= APIC_LVT_MASKED;
 
-               val &= apic_lvt_mask[(reg - APIC_LVTT) >> 4];
+               size = ARRAY_SIZE(apic_lvt_mask);
+               index = array_index_nospec(
+                               (reg - APIC_LVTT) >> 4, size);
+               val &= apic_lvt_mask[index];
                apic_set_reg(apic, reg, val);
-
                break;
+       }
 
        case APIC_LVTT:
                if (!kvm_apic_sw_enabled(apic))