Loading...
1// SPDX-License-Identifier: GPL-2.0
2
3#include <linux/cpuhotplug.h>
4#include <linux/cpumask.h>
5#include <linux/slab.h>
6#include <linux/mm.h>
7
8#include <asm/apic.h>
9
10#include "local.h"
11
12#define apic_cluster(apicid) ((apicid) >> 4)
13
14/*
15 * __x2apic_send_IPI_mask() possibly needs to read
16 * x86_cpu_to_logical_apicid for all online cpus in a sequential way.
17 * Using per cpu variable would cost one cache line per cpu.
18 */
19static u32 *x86_cpu_to_logical_apicid __read_mostly;
20
21static DEFINE_PER_CPU(cpumask_var_t, ipi_mask);
22static DEFINE_PER_CPU_READ_MOSTLY(struct cpumask *, cluster_masks);
23
24static int x2apic_acpi_madt_oem_check(char *oem_id, char *oem_table_id)
25{
26 return x2apic_enabled();
27}
28
29static void x2apic_send_IPI(int cpu, int vector)
30{
31 u32 dest = x86_cpu_to_logical_apicid[cpu];
32
33 /* x2apic MSRs are special and need a special fence: */
34 weak_wrmsr_fence();
35 __x2apic_send_IPI_dest(dest, vector, APIC_DEST_LOGICAL);
36}
37
38static void
39__x2apic_send_IPI_mask(const struct cpumask *mask, int vector, int apic_dest)
40{
41 unsigned int cpu, clustercpu;
42 struct cpumask *tmpmsk;
43 unsigned long flags;
44 u32 dest;
45
46 /* x2apic MSRs are special and need a special fence: */
47 weak_wrmsr_fence();
48 local_irq_save(flags);
49
50 tmpmsk = this_cpu_cpumask_var_ptr(ipi_mask);
51 cpumask_copy(tmpmsk, mask);
52 /* If IPI should not be sent to self, clear current CPU */
53 if (apic_dest != APIC_DEST_ALLINC)
54 __cpumask_clear_cpu(smp_processor_id(), tmpmsk);
55
56 /* Collapse cpus in a cluster so a single IPI per cluster is sent */
57 for_each_cpu(cpu, tmpmsk) {
58 struct cpumask *cmsk = per_cpu(cluster_masks, cpu);
59
60 dest = 0;
61 for_each_cpu_and(clustercpu, tmpmsk, cmsk)
62 dest |= x86_cpu_to_logical_apicid[clustercpu];
63
64 if (!dest)
65 continue;
66
67 __x2apic_send_IPI_dest(dest, vector, APIC_DEST_LOGICAL);
68 /* Remove cluster CPUs from tmpmask */
69 cpumask_andnot(tmpmsk, tmpmsk, cmsk);
70 }
71
72 local_irq_restore(flags);
73}
74
75static void x2apic_send_IPI_mask(const struct cpumask *mask, int vector)
76{
77 __x2apic_send_IPI_mask(mask, vector, APIC_DEST_ALLINC);
78}
79
80static void
81x2apic_send_IPI_mask_allbutself(const struct cpumask *mask, int vector)
82{
83 __x2apic_send_IPI_mask(mask, vector, APIC_DEST_ALLBUT);
84}
85
86static u32 x2apic_calc_apicid(unsigned int cpu)
87{
88 return x86_cpu_to_logical_apicid[cpu];
89}
90
91static void init_x2apic_ldr(void)
92{
93 struct cpumask *cmsk = this_cpu_read(cluster_masks);
94
95 BUG_ON(!cmsk);
96
97 cpumask_set_cpu(smp_processor_id(), cmsk);
98}
99
100/*
101 * As an optimisation during boot, set the cluster_mask for all present
102 * CPUs at once, to prevent each of them having to iterate over the others
103 * to find the existing cluster_mask.
104 */
105static void prefill_clustermask(struct cpumask *cmsk, unsigned int cpu, u32 cluster)
106{
107 int cpu_i;
108
109 for_each_present_cpu(cpu_i) {
110 struct cpumask **cpu_cmsk = &per_cpu(cluster_masks, cpu_i);
111 u32 apicid = apic->cpu_present_to_apicid(cpu_i);
112
113 if (apicid == BAD_APICID || cpu_i == cpu || apic_cluster(apicid) != cluster)
114 continue;
115
116 if (WARN_ON_ONCE(*cpu_cmsk == cmsk))
117 continue;
118
119 BUG_ON(*cpu_cmsk);
120 *cpu_cmsk = cmsk;
121 }
122}
123
124static int alloc_clustermask(unsigned int cpu, u32 cluster, int node)
125{
126 struct cpumask *cmsk = NULL;
127 unsigned int cpu_i;
128
129 /*
130 * At boot time, the CPU present mask is stable. The cluster mask is
131 * allocated for the first CPU in the cluster and propagated to all
132 * present siblings in the cluster. If the cluster mask is already set
133 * on entry to this function for a given CPU, there is nothing to do.
134 */
135 if (per_cpu(cluster_masks, cpu))
136 return 0;
137
138 if (system_state < SYSTEM_RUNNING)
139 goto alloc;
140
141 /*
142 * On post boot hotplug for a CPU which was not present at boot time,
143 * iterate over all possible CPUs (even those which are not present
144 * any more) to find any existing cluster mask.
145 */
146 for_each_possible_cpu(cpu_i) {
147 u32 apicid = apic->cpu_present_to_apicid(cpu_i);
148
149 if (apicid != BAD_APICID && apic_cluster(apicid) == cluster) {
150 cmsk = per_cpu(cluster_masks, cpu_i);
151 /*
152 * If the cluster is already initialized, just store
153 * the mask and return. There's no need to propagate.
154 */
155 if (cmsk) {
156 per_cpu(cluster_masks, cpu) = cmsk;
157 return 0;
158 }
159 }
160 }
161 /*
162 * No CPU in the cluster has ever been initialized, so fall through to
163 * the boot time code which will also populate the cluster mask for any
164 * other CPU in the cluster which is (now) present.
165 */
166alloc:
167 cmsk = kzalloc_node(sizeof(*cmsk), GFP_KERNEL, node);
168 if (!cmsk)
169 return -ENOMEM;
170 per_cpu(cluster_masks, cpu) = cmsk;
171 prefill_clustermask(cmsk, cpu, cluster);
172
173 return 0;
174}
175
176static int x2apic_prepare_cpu(unsigned int cpu)
177{
178 u32 phys_apicid = apic->cpu_present_to_apicid(cpu);
179 u32 cluster = apic_cluster(phys_apicid);
180 u32 logical_apicid = (cluster << 16) | (1 << (phys_apicid & 0xf));
181 int node = cpu_to_node(cpu);
182
183 x86_cpu_to_logical_apicid[cpu] = logical_apicid;
184
185 if (alloc_clustermask(cpu, cluster, node) < 0)
186 return -ENOMEM;
187
188 if (!zalloc_cpumask_var_node(&per_cpu(ipi_mask, cpu), GFP_KERNEL, node))
189 return -ENOMEM;
190
191 return 0;
192}
193
194static int x2apic_dead_cpu(unsigned int dead_cpu)
195{
196 struct cpumask *cmsk = per_cpu(cluster_masks, dead_cpu);
197
198 if (cmsk)
199 cpumask_clear_cpu(dead_cpu, cmsk);
200 free_cpumask_var(per_cpu(ipi_mask, dead_cpu));
201 return 0;
202}
203
204static int x2apic_cluster_probe(void)
205{
206 u32 slots;
207
208 if (!x2apic_mode)
209 return 0;
210
211 slots = max_t(u32, L1_CACHE_BYTES/sizeof(u32), nr_cpu_ids);
212 x86_cpu_to_logical_apicid = kcalloc(slots, sizeof(u32), GFP_KERNEL);
213 if (!x86_cpu_to_logical_apicid)
214 return 0;
215
216 if (cpuhp_setup_state(CPUHP_X2APIC_PREPARE, "x86/x2apic:prepare",
217 x2apic_prepare_cpu, x2apic_dead_cpu) < 0) {
218 pr_err("Failed to register X2APIC_PREPARE\n");
219 kfree(x86_cpu_to_logical_apicid);
220 x86_cpu_to_logical_apicid = NULL;
221 return 0;
222 }
223 init_x2apic_ldr();
224 return 1;
225}
226
227static struct apic apic_x2apic_cluster __ro_after_init = {
228
229 .name = "cluster x2apic",
230 .probe = x2apic_cluster_probe,
231 .acpi_madt_oem_check = x2apic_acpi_madt_oem_check,
232
233 .dest_mode_logical = true,
234
235 .disable_esr = 0,
236
237 .init_apic_ldr = init_x2apic_ldr,
238 .cpu_present_to_apicid = default_cpu_present_to_apicid,
239
240 .max_apic_id = UINT_MAX,
241 .x2apic_set_max_apicid = true,
242 .get_apic_id = x2apic_get_apic_id,
243
244 .calc_dest_apicid = x2apic_calc_apicid,
245
246 .send_IPI = x2apic_send_IPI,
247 .send_IPI_mask = x2apic_send_IPI_mask,
248 .send_IPI_mask_allbutself = x2apic_send_IPI_mask_allbutself,
249 .send_IPI_allbutself = x2apic_send_IPI_allbutself,
250 .send_IPI_all = x2apic_send_IPI_all,
251 .send_IPI_self = x2apic_send_IPI_self,
252 .nmi_to_offline_cpu = true,
253
254 .read = native_apic_msr_read,
255 .write = native_apic_msr_write,
256 .eoi = native_apic_msr_eoi,
257 .icr_read = native_x2apic_icr_read,
258 .icr_write = native_x2apic_icr_write,
259};
260
261apic_driver(apic_x2apic_cluster);
1// SPDX-License-Identifier: GPL-2.0
2
3#include <linux/cpuhotplug.h>
4#include <linux/cpumask.h>
5#include <linux/slab.h>
6#include <linux/mm.h>
7
8#include <asm/apic.h>
9
10#include "local.h"
11
12struct cluster_mask {
13 unsigned int clusterid;
14 int node;
15 struct cpumask mask;
16};
17
18static DEFINE_PER_CPU(u32, x86_cpu_to_logical_apicid);
19static DEFINE_PER_CPU(cpumask_var_t, ipi_mask);
20static DEFINE_PER_CPU(struct cluster_mask *, cluster_masks);
21static struct cluster_mask *cluster_hotplug_mask;
22
23static int x2apic_acpi_madt_oem_check(char *oem_id, char *oem_table_id)
24{
25 return x2apic_enabled();
26}
27
28static void x2apic_send_IPI(int cpu, int vector)
29{
30 u32 dest = per_cpu(x86_cpu_to_logical_apicid, cpu);
31
32 /* x2apic MSRs are special and need a special fence: */
33 weak_wrmsr_fence();
34 __x2apic_send_IPI_dest(dest, vector, APIC_DEST_LOGICAL);
35}
36
37static void
38__x2apic_send_IPI_mask(const struct cpumask *mask, int vector, int apic_dest)
39{
40 unsigned int cpu, clustercpu;
41 struct cpumask *tmpmsk;
42 unsigned long flags;
43 u32 dest;
44
45 /* x2apic MSRs are special and need a special fence: */
46 weak_wrmsr_fence();
47 local_irq_save(flags);
48
49 tmpmsk = this_cpu_cpumask_var_ptr(ipi_mask);
50 cpumask_copy(tmpmsk, mask);
51 /* If IPI should not be sent to self, clear current CPU */
52 if (apic_dest != APIC_DEST_ALLINC)
53 __cpumask_clear_cpu(smp_processor_id(), tmpmsk);
54
55 /* Collapse cpus in a cluster so a single IPI per cluster is sent */
56 for_each_cpu(cpu, tmpmsk) {
57 struct cluster_mask *cmsk = per_cpu(cluster_masks, cpu);
58
59 dest = 0;
60 for_each_cpu_and(clustercpu, tmpmsk, &cmsk->mask)
61 dest |= per_cpu(x86_cpu_to_logical_apicid, clustercpu);
62
63 if (!dest)
64 continue;
65
66 __x2apic_send_IPI_dest(dest, vector, APIC_DEST_LOGICAL);
67 /* Remove cluster CPUs from tmpmask */
68 cpumask_andnot(tmpmsk, tmpmsk, &cmsk->mask);
69 }
70
71 local_irq_restore(flags);
72}
73
74static void x2apic_send_IPI_mask(const struct cpumask *mask, int vector)
75{
76 __x2apic_send_IPI_mask(mask, vector, APIC_DEST_ALLINC);
77}
78
79static void
80x2apic_send_IPI_mask_allbutself(const struct cpumask *mask, int vector)
81{
82 __x2apic_send_IPI_mask(mask, vector, APIC_DEST_ALLBUT);
83}
84
85static void x2apic_send_IPI_allbutself(int vector)
86{
87 __x2apic_send_IPI_shorthand(vector, APIC_DEST_ALLBUT);
88}
89
90static void x2apic_send_IPI_all(int vector)
91{
92 __x2apic_send_IPI_shorthand(vector, APIC_DEST_ALLINC);
93}
94
95static u32 x2apic_calc_apicid(unsigned int cpu)
96{
97 return per_cpu(x86_cpu_to_logical_apicid, cpu);
98}
99
100static void init_x2apic_ldr(void)
101{
102 struct cluster_mask *cmsk = this_cpu_read(cluster_masks);
103 u32 cluster, apicid = apic_read(APIC_LDR);
104 unsigned int cpu;
105
106 this_cpu_write(x86_cpu_to_logical_apicid, apicid);
107
108 if (cmsk)
109 goto update;
110
111 cluster = apicid >> 16;
112 for_each_online_cpu(cpu) {
113 cmsk = per_cpu(cluster_masks, cpu);
114 /* Matching cluster found. Link and update it. */
115 if (cmsk && cmsk->clusterid == cluster)
116 goto update;
117 }
118 cmsk = cluster_hotplug_mask;
119 cmsk->clusterid = cluster;
120 cluster_hotplug_mask = NULL;
121update:
122 this_cpu_write(cluster_masks, cmsk);
123 cpumask_set_cpu(smp_processor_id(), &cmsk->mask);
124}
125
126static int alloc_clustermask(unsigned int cpu, int node)
127{
128 if (per_cpu(cluster_masks, cpu))
129 return 0;
130 /*
131 * If a hotplug spare mask exists, check whether it's on the right
132 * node. If not, free it and allocate a new one.
133 */
134 if (cluster_hotplug_mask) {
135 if (cluster_hotplug_mask->node == node)
136 return 0;
137 kfree(cluster_hotplug_mask);
138 }
139
140 cluster_hotplug_mask = kzalloc_node(sizeof(*cluster_hotplug_mask),
141 GFP_KERNEL, node);
142 if (!cluster_hotplug_mask)
143 return -ENOMEM;
144 cluster_hotplug_mask->node = node;
145 return 0;
146}
147
148static int x2apic_prepare_cpu(unsigned int cpu)
149{
150 if (alloc_clustermask(cpu, cpu_to_node(cpu)) < 0)
151 return -ENOMEM;
152 if (!zalloc_cpumask_var(&per_cpu(ipi_mask, cpu), GFP_KERNEL))
153 return -ENOMEM;
154 return 0;
155}
156
157static int x2apic_dead_cpu(unsigned int dead_cpu)
158{
159 struct cluster_mask *cmsk = per_cpu(cluster_masks, dead_cpu);
160
161 if (cmsk)
162 cpumask_clear_cpu(dead_cpu, &cmsk->mask);
163 free_cpumask_var(per_cpu(ipi_mask, dead_cpu));
164 return 0;
165}
166
167static int x2apic_cluster_probe(void)
168{
169 if (!x2apic_mode)
170 return 0;
171
172 if (cpuhp_setup_state(CPUHP_X2APIC_PREPARE, "x86/x2apic:prepare",
173 x2apic_prepare_cpu, x2apic_dead_cpu) < 0) {
174 pr_err("Failed to register X2APIC_PREPARE\n");
175 return 0;
176 }
177 init_x2apic_ldr();
178 return 1;
179}
180
181static struct apic apic_x2apic_cluster __ro_after_init = {
182
183 .name = "cluster x2apic",
184 .probe = x2apic_cluster_probe,
185 .acpi_madt_oem_check = x2apic_acpi_madt_oem_check,
186 .apic_id_valid = x2apic_apic_id_valid,
187 .apic_id_registered = x2apic_apic_id_registered,
188
189 .delivery_mode = APIC_DELIVERY_MODE_FIXED,
190 .dest_mode_logical = true,
191
192 .disable_esr = 0,
193
194 .check_apicid_used = NULL,
195 .init_apic_ldr = init_x2apic_ldr,
196 .ioapic_phys_id_map = NULL,
197 .setup_apic_routing = NULL,
198 .cpu_present_to_apicid = default_cpu_present_to_apicid,
199 .apicid_to_cpu_present = NULL,
200 .check_phys_apicid_present = default_check_phys_apicid_present,
201 .phys_pkg_id = x2apic_phys_pkg_id,
202
203 .get_apic_id = x2apic_get_apic_id,
204 .set_apic_id = x2apic_set_apic_id,
205
206 .calc_dest_apicid = x2apic_calc_apicid,
207
208 .send_IPI = x2apic_send_IPI,
209 .send_IPI_mask = x2apic_send_IPI_mask,
210 .send_IPI_mask_allbutself = x2apic_send_IPI_mask_allbutself,
211 .send_IPI_allbutself = x2apic_send_IPI_allbutself,
212 .send_IPI_all = x2apic_send_IPI_all,
213 .send_IPI_self = x2apic_send_IPI_self,
214
215 .inquire_remote_apic = NULL,
216
217 .read = native_apic_msr_read,
218 .write = native_apic_msr_write,
219 .eoi_write = native_apic_msr_eoi_write,
220 .icr_read = native_x2apic_icr_read,
221 .icr_write = native_x2apic_icr_write,
222 .wait_icr_idle = native_x2apic_wait_icr_idle,
223 .safe_wait_icr_idle = native_safe_x2apic_wait_icr_idle,
224};
225
226apic_driver(apic_x2apic_cluster);