OSDN Git Service

KVM: x86: Protect MSR-based index computations in pmu.h from Spectre-v1/L1TF attacks
[sagit-ice-cold/kernel_xiaomi_msm8998.git] / arch / x86 / kvm / pmu.h
1 #ifndef __KVM_X86_PMU_H
2 #define __KVM_X86_PMU_H
3
4 #include <linux/nospec.h>
5
6 #define vcpu_to_pmu(vcpu) (&(vcpu)->arch.pmu)
7 #define pmu_to_vcpu(pmu)  (container_of((pmu), struct kvm_vcpu, arch.pmu))
8 #define pmc_to_pmu(pmc)   (&(pmc)->vcpu->arch.pmu)
9
10 /* retrieve the 4 bits for EN and PMI out of IA32_FIXED_CTR_CTRL */
11 #define fixed_ctrl_field(ctrl_reg, idx) (((ctrl_reg) >> ((idx)*4)) & 0xf)
12
13 struct kvm_event_hw_type_mapping {
14         u8 eventsel;
15         u8 unit_mask;
16         unsigned event_type;
17 };
18
19 struct kvm_pmu_ops {
20         unsigned (*find_arch_event)(struct kvm_pmu *pmu, u8 event_select,
21                                     u8 unit_mask);
22         unsigned (*find_fixed_event)(int idx);
23         bool (*pmc_is_enabled)(struct kvm_pmc *pmc);
24         struct kvm_pmc *(*pmc_idx_to_pmc)(struct kvm_pmu *pmu, int pmc_idx);
25         struct kvm_pmc *(*msr_idx_to_pmc)(struct kvm_vcpu *vcpu, unsigned idx);
26         int (*is_valid_msr_idx)(struct kvm_vcpu *vcpu, unsigned idx);
27         bool (*is_valid_msr)(struct kvm_vcpu *vcpu, u32 msr);
28         int (*get_msr)(struct kvm_vcpu *vcpu, u32 msr, u64 *data);
29         int (*set_msr)(struct kvm_vcpu *vcpu, struct msr_data *msr_info);
30         void (*refresh)(struct kvm_vcpu *vcpu);
31         void (*init)(struct kvm_vcpu *vcpu);
32         void (*reset)(struct kvm_vcpu *vcpu);
33 };
34
35 static inline u64 pmc_bitmask(struct kvm_pmc *pmc)
36 {
37         struct kvm_pmu *pmu = pmc_to_pmu(pmc);
38
39         return pmu->counter_bitmask[pmc->type];
40 }
41
42 static inline u64 pmc_read_counter(struct kvm_pmc *pmc)
43 {
44         u64 counter, enabled, running;
45
46         counter = pmc->counter;
47         if (pmc->perf_event)
48                 counter += perf_event_read_value(pmc->perf_event,
49                                                  &enabled, &running);
50         /* FIXME: Scaling needed? */
51         return counter & pmc_bitmask(pmc);
52 }
53
54 static inline void pmc_stop_counter(struct kvm_pmc *pmc)
55 {
56         if (pmc->perf_event) {
57                 pmc->counter = pmc_read_counter(pmc);
58                 perf_event_release_kernel(pmc->perf_event);
59                 pmc->perf_event = NULL;
60         }
61 }
62
63 static inline bool pmc_is_gp(struct kvm_pmc *pmc)
64 {
65         return pmc->type == KVM_PMC_GP;
66 }
67
68 static inline bool pmc_is_fixed(struct kvm_pmc *pmc)
69 {
70         return pmc->type == KVM_PMC_FIXED;
71 }
72
73 static inline bool pmc_is_enabled(struct kvm_pmc *pmc)
74 {
75         return kvm_x86_ops->pmu_ops->pmc_is_enabled(pmc);
76 }
77
78 /* returns general purpose PMC with the specified MSR. Note that it can be
79  * used for both PERFCTRn and EVNTSELn; that is why it accepts base as a
80  * paramenter to tell them apart.
81  */
82 static inline struct kvm_pmc *get_gp_pmc(struct kvm_pmu *pmu, u32 msr,
83                                          u32 base)
84 {
85         if (msr >= base && msr < base + pmu->nr_arch_gp_counters) {
86                 u32 index = array_index_nospec(msr - base,
87                                                pmu->nr_arch_gp_counters);
88
89                 return &pmu->gp_counters[index];
90         }
91
92         return NULL;
93 }
94
95 /* returns fixed PMC with the specified MSR */
96 static inline struct kvm_pmc *get_fixed_pmc(struct kvm_pmu *pmu, u32 msr)
97 {
98         int base = MSR_CORE_PERF_FIXED_CTR0;
99
100         if (msr >= base && msr < base + pmu->nr_arch_fixed_counters) {
101                 u32 index = array_index_nospec(msr - base,
102                                                pmu->nr_arch_fixed_counters);
103
104                 return &pmu->fixed_counters[index];
105         }
106
107         return NULL;
108 }
109
110 void reprogram_gp_counter(struct kvm_pmc *pmc, u64 eventsel);
111 void reprogram_fixed_counter(struct kvm_pmc *pmc, u8 ctrl, int fixed_idx);
112 void reprogram_counter(struct kvm_pmu *pmu, int pmc_idx);
113
114 void kvm_pmu_deliver_pmi(struct kvm_vcpu *vcpu);
115 void kvm_pmu_handle_event(struct kvm_vcpu *vcpu);
116 int kvm_pmu_rdpmc(struct kvm_vcpu *vcpu, unsigned pmc, u64 *data);
117 int kvm_pmu_is_valid_msr_idx(struct kvm_vcpu *vcpu, unsigned idx);
118 bool kvm_pmu_is_valid_msr(struct kvm_vcpu *vcpu, u32 msr);
119 int kvm_pmu_get_msr(struct kvm_vcpu *vcpu, u32 msr, u64 *data);
120 int kvm_pmu_set_msr(struct kvm_vcpu *vcpu, struct msr_data *msr_info);
121 void kvm_pmu_refresh(struct kvm_vcpu *vcpu);
122 void kvm_pmu_reset(struct kvm_vcpu *vcpu);
123 void kvm_pmu_init(struct kvm_vcpu *vcpu);
124 void kvm_pmu_destroy(struct kvm_vcpu *vcpu);
125
126 extern struct kvm_pmu_ops intel_pmu_ops;
127 extern struct kvm_pmu_ops amd_pmu_ops;
128 #endif /* __KVM_X86_PMU_H */