bb1c3f5f60c81d3969e0c84e269f3f4883dd0467
[sfrench/cifs-2.6.git] / arch / x86 / kernel / cpu / resctrl / core.c
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Resource Director Technology(RDT)
4  * - Cache Allocation code.
5  *
6  * Copyright (C) 2016 Intel Corporation
7  *
8  * Authors:
9  *    Fenghua Yu <fenghua.yu@intel.com>
10  *    Tony Luck <tony.luck@intel.com>
11  *    Vikas Shivappa <vikas.shivappa@intel.com>
12  *
13  * More information about RDT be found in the Intel (R) x86 Architecture
14  * Software Developer Manual June 2016, volume 3, section 17.17.
15  */
16
17 #define pr_fmt(fmt)     "resctrl: " fmt
18
19 #include <linux/slab.h>
20 #include <linux/err.h>
21 #include <linux/cacheinfo.h>
22 #include <linux/cpuhotplug.h>
23
24 #include <asm/intel-family.h>
25 #include <asm/resctrl.h>
26 #include "internal.h"
27
28 /* Mutex to protect rdtgroup access. */
29 DEFINE_MUTEX(rdtgroup_mutex);
30
31 /*
32  * The cached resctrl_pqr_state is strictly per CPU and can never be
33  * updated from a remote CPU. Functions which modify the state
34  * are called with interrupts disabled and no preemption, which
35  * is sufficient for the protection.
36  */
37 DEFINE_PER_CPU(struct resctrl_pqr_state, pqr_state);
38
39 /*
40  * Used to store the max resource name width and max resource data width
41  * to display the schemata in a tabular format
42  */
43 int max_name_width, max_data_width;
44
45 /*
46  * Global boolean for rdt_alloc which is true if any
47  * resource allocation is enabled.
48  */
49 bool rdt_alloc_capable;
50
51 static void
52 mba_wrmsr_intel(struct rdt_domain *d, struct msr_param *m,
53                 struct rdt_resource *r);
54 static void
55 cat_wrmsr(struct rdt_domain *d, struct msr_param *m, struct rdt_resource *r);
56 static void
57 mba_wrmsr_amd(struct rdt_domain *d, struct msr_param *m,
58               struct rdt_resource *r);
59
60 #define domain_init(id) LIST_HEAD_INIT(rdt_resources_all[id].r_resctrl.domains)
61
62 struct rdt_hw_resource rdt_resources_all[] = {
63         [RDT_RESOURCE_L3] =
64         {
65                 .r_resctrl = {
66                         .rid                    = RDT_RESOURCE_L3,
67                         .name                   = "L3",
68                         .cache_level            = 3,
69                         .cache = {
70                                 .min_cbm_bits   = 1,
71                         },
72                         .domains                = domain_init(RDT_RESOURCE_L3),
73                         .parse_ctrlval          = parse_cbm,
74                         .format_str             = "%d=%0*x",
75                         .fflags                 = RFTYPE_RES_CACHE,
76                 },
77                 .msr_base               = MSR_IA32_L3_CBM_BASE,
78                 .msr_update             = cat_wrmsr,
79         },
80         [RDT_RESOURCE_L2] =
81         {
82                 .r_resctrl = {
83                         .rid                    = RDT_RESOURCE_L2,
84                         .name                   = "L2",
85                         .cache_level            = 2,
86                         .cache = {
87                                 .min_cbm_bits   = 1,
88                         },
89                         .domains                = domain_init(RDT_RESOURCE_L2),
90                         .parse_ctrlval          = parse_cbm,
91                         .format_str             = "%d=%0*x",
92                         .fflags                 = RFTYPE_RES_CACHE,
93                 },
94                 .msr_base               = MSR_IA32_L2_CBM_BASE,
95                 .msr_update             = cat_wrmsr,
96         },
97         [RDT_RESOURCE_MBA] =
98         {
99                 .r_resctrl = {
100                         .rid                    = RDT_RESOURCE_MBA,
101                         .name                   = "MB",
102                         .cache_level            = 3,
103                         .domains                = domain_init(RDT_RESOURCE_MBA),
104                         .parse_ctrlval          = parse_bw,
105                         .format_str             = "%d=%*u",
106                         .fflags                 = RFTYPE_RES_MB,
107                 },
108         },
109 };
110
111 /*
112  * cache_alloc_hsw_probe() - Have to probe for Intel haswell server CPUs
113  * as they do not have CPUID enumeration support for Cache allocation.
114  * The check for Vendor/Family/Model is not enough to guarantee that
115  * the MSRs won't #GP fault because only the following SKUs support
116  * CAT:
117  *      Intel(R) Xeon(R)  CPU E5-2658  v3  @  2.20GHz
118  *      Intel(R) Xeon(R)  CPU E5-2648L v3  @  1.80GHz
119  *      Intel(R) Xeon(R)  CPU E5-2628L v3  @  2.00GHz
120  *      Intel(R) Xeon(R)  CPU E5-2618L v3  @  2.30GHz
121  *      Intel(R) Xeon(R)  CPU E5-2608L v3  @  2.00GHz
122  *      Intel(R) Xeon(R)  CPU E5-2658A v3  @  2.20GHz
123  *
124  * Probe by trying to write the first of the L3 cache mask registers
125  * and checking that the bits stick. Max CLOSids is always 4 and max cbm length
126  * is always 20 on hsw server parts. The minimum cache bitmask length
127  * allowed for HSW server is always 2 bits. Hardcode all of them.
128  */
129 static inline void cache_alloc_hsw_probe(void)
130 {
131         struct rdt_hw_resource *hw_res = &rdt_resources_all[RDT_RESOURCE_L3];
132         struct rdt_resource *r  = &hw_res->r_resctrl;
133         u32 l, h, max_cbm = BIT_MASK(20) - 1;
134
135         if (wrmsr_safe(MSR_IA32_L3_CBM_BASE, max_cbm, 0))
136                 return;
137
138         rdmsr(MSR_IA32_L3_CBM_BASE, l, h);
139
140         /* If all the bits were set in MSR, return success */
141         if (l != max_cbm)
142                 return;
143
144         hw_res->num_closid = 4;
145         r->default_ctrl = max_cbm;
146         r->cache.cbm_len = 20;
147         r->cache.shareable_bits = 0xc0000;
148         r->cache.min_cbm_bits = 2;
149         r->alloc_capable = true;
150         r->alloc_enabled = true;
151
152         rdt_alloc_capable = true;
153 }
154
155 bool is_mba_sc(struct rdt_resource *r)
156 {
157         if (!r)
158                 return rdt_resources_all[RDT_RESOURCE_MBA].r_resctrl.membw.mba_sc;
159
160         return r->membw.mba_sc;
161 }
162
163 /*
164  * rdt_get_mb_table() - get a mapping of bandwidth(b/w) percentage values
165  * exposed to user interface and the h/w understandable delay values.
166  *
167  * The non-linear delay values have the granularity of power of two
168  * and also the h/w does not guarantee a curve for configured delay
169  * values vs. actual b/w enforced.
170  * Hence we need a mapping that is pre calibrated so the user can
171  * express the memory b/w as a percentage value.
172  */
173 static inline bool rdt_get_mb_table(struct rdt_resource *r)
174 {
175         /*
176          * There are no Intel SKUs as of now to support non-linear delay.
177          */
178         pr_info("MBA b/w map not implemented for cpu:%d, model:%d",
179                 boot_cpu_data.x86, boot_cpu_data.x86_model);
180
181         return false;
182 }
183
184 static bool __get_mem_config_intel(struct rdt_resource *r)
185 {
186         struct rdt_hw_resource *hw_res = resctrl_to_arch_res(r);
187         union cpuid_0x10_3_eax eax;
188         union cpuid_0x10_x_edx edx;
189         u32 ebx, ecx, max_delay;
190
191         cpuid_count(0x00000010, 3, &eax.full, &ebx, &ecx, &edx.full);
192         hw_res->num_closid = edx.split.cos_max + 1;
193         max_delay = eax.split.max_delay + 1;
194         r->default_ctrl = MAX_MBA_BW;
195         r->membw.arch_needs_linear = true;
196         if (ecx & MBA_IS_LINEAR) {
197                 r->membw.delay_linear = true;
198                 r->membw.min_bw = MAX_MBA_BW - max_delay;
199                 r->membw.bw_gran = MAX_MBA_BW - max_delay;
200         } else {
201                 if (!rdt_get_mb_table(r))
202                         return false;
203                 r->membw.arch_needs_linear = false;
204         }
205         r->data_width = 3;
206
207         if (boot_cpu_has(X86_FEATURE_PER_THREAD_MBA))
208                 r->membw.throttle_mode = THREAD_THROTTLE_PER_THREAD;
209         else
210                 r->membw.throttle_mode = THREAD_THROTTLE_MAX;
211         thread_throttle_mode_init();
212
213         r->alloc_capable = true;
214         r->alloc_enabled = true;
215
216         return true;
217 }
218
219 static bool __rdt_get_mem_config_amd(struct rdt_resource *r)
220 {
221         struct rdt_hw_resource *hw_res = resctrl_to_arch_res(r);
222         union cpuid_0x10_3_eax eax;
223         union cpuid_0x10_x_edx edx;
224         u32 ebx, ecx;
225
226         cpuid_count(0x80000020, 1, &eax.full, &ebx, &ecx, &edx.full);
227         hw_res->num_closid = edx.split.cos_max + 1;
228         r->default_ctrl = MAX_MBA_BW_AMD;
229
230         /* AMD does not use delay */
231         r->membw.delay_linear = false;
232         r->membw.arch_needs_linear = false;
233
234         /*
235          * AMD does not use memory delay throttle model to control
236          * the allocation like Intel does.
237          */
238         r->membw.throttle_mode = THREAD_THROTTLE_UNDEFINED;
239         r->membw.min_bw = 0;
240         r->membw.bw_gran = 1;
241         /* Max value is 2048, Data width should be 4 in decimal */
242         r->data_width = 4;
243
244         r->alloc_capable = true;
245         r->alloc_enabled = true;
246
247         return true;
248 }
249
250 static void rdt_get_cache_alloc_cfg(int idx, struct rdt_resource *r)
251 {
252         struct rdt_hw_resource *hw_res = resctrl_to_arch_res(r);
253         union cpuid_0x10_1_eax eax;
254         union cpuid_0x10_x_edx edx;
255         u32 ebx, ecx;
256
257         cpuid_count(0x00000010, idx, &eax.full, &ebx, &ecx, &edx.full);
258         hw_res->num_closid = edx.split.cos_max + 1;
259         r->cache.cbm_len = eax.split.cbm_len + 1;
260         r->default_ctrl = BIT_MASK(eax.split.cbm_len + 1) - 1;
261         r->cache.shareable_bits = ebx & r->default_ctrl;
262         r->data_width = (r->cache.cbm_len + 3) / 4;
263         r->alloc_capable = true;
264         r->alloc_enabled = true;
265 }
266
267 static void rdt_get_cdp_config(int level)
268 {
269         /*
270          * By default, CDP is disabled. CDP can be enabled by mount parameter
271          * "cdp" during resctrl file system mount time.
272          */
273         rdt_resources_all[level].cdp_enabled = false;
274         rdt_resources_all[level].r_resctrl.cdp_capable = true;
275 }
276
277 static void rdt_get_cdp_l3_config(void)
278 {
279         rdt_get_cdp_config(RDT_RESOURCE_L3);
280 }
281
282 static void rdt_get_cdp_l2_config(void)
283 {
284         rdt_get_cdp_config(RDT_RESOURCE_L2);
285 }
286
287 static void
288 mba_wrmsr_amd(struct rdt_domain *d, struct msr_param *m, struct rdt_resource *r)
289 {
290         unsigned int i;
291         struct rdt_hw_domain *hw_dom = resctrl_to_arch_dom(d);
292         struct rdt_hw_resource *hw_res = resctrl_to_arch_res(r);
293
294         for (i = m->low; i < m->high; i++)
295                 wrmsrl(hw_res->msr_base + i, hw_dom->ctrl_val[i]);
296 }
297
298 /*
299  * Map the memory b/w percentage value to delay values
300  * that can be written to QOS_MSRs.
301  * There are currently no SKUs which support non linear delay values.
302  */
303 u32 delay_bw_map(unsigned long bw, struct rdt_resource *r)
304 {
305         if (r->membw.delay_linear)
306                 return MAX_MBA_BW - bw;
307
308         pr_warn_once("Non Linear delay-bw map not supported but queried\n");
309         return r->default_ctrl;
310 }
311
312 static void
313 mba_wrmsr_intel(struct rdt_domain *d, struct msr_param *m,
314                 struct rdt_resource *r)
315 {
316         unsigned int i;
317         struct rdt_hw_domain *hw_dom = resctrl_to_arch_dom(d);
318         struct rdt_hw_resource *hw_res = resctrl_to_arch_res(r);
319
320         /*  Write the delay values for mba. */
321         for (i = m->low; i < m->high; i++)
322                 wrmsrl(hw_res->msr_base + i, delay_bw_map(hw_dom->ctrl_val[i], r));
323 }
324
325 static void
326 cat_wrmsr(struct rdt_domain *d, struct msr_param *m, struct rdt_resource *r)
327 {
328         unsigned int i;
329         struct rdt_hw_domain *hw_dom = resctrl_to_arch_dom(d);
330         struct rdt_hw_resource *hw_res = resctrl_to_arch_res(r);
331
332         for (i = m->low; i < m->high; i++)
333                 wrmsrl(hw_res->msr_base + i, hw_dom->ctrl_val[i]);
334 }
335
336 struct rdt_domain *get_domain_from_cpu(int cpu, struct rdt_resource *r)
337 {
338         struct rdt_domain *d;
339
340         list_for_each_entry(d, &r->domains, list) {
341                 /* Find the domain that contains this CPU */
342                 if (cpumask_test_cpu(cpu, &d->cpu_mask))
343                         return d;
344         }
345
346         return NULL;
347 }
348
349 u32 resctrl_arch_get_num_closid(struct rdt_resource *r)
350 {
351         return resctrl_to_arch_res(r)->num_closid;
352 }
353
354 void rdt_ctrl_update(void *arg)
355 {
356         struct msr_param *m = arg;
357         struct rdt_hw_resource *hw_res = resctrl_to_arch_res(m->res);
358         struct rdt_resource *r = m->res;
359         int cpu = smp_processor_id();
360         struct rdt_domain *d;
361
362         d = get_domain_from_cpu(cpu, r);
363         if (d) {
364                 hw_res->msr_update(d, m, r);
365                 return;
366         }
367         pr_warn_once("cpu %d not found in any domain for resource %s\n",
368                      cpu, r->name);
369 }
370
371 /*
372  * rdt_find_domain - Find a domain in a resource that matches input resource id
373  *
374  * Search resource r's domain list to find the resource id. If the resource
375  * id is found in a domain, return the domain. Otherwise, if requested by
376  * caller, return the first domain whose id is bigger than the input id.
377  * The domain list is sorted by id in ascending order.
378  */
379 struct rdt_domain *rdt_find_domain(struct rdt_resource *r, int id,
380                                    struct list_head **pos)
381 {
382         struct rdt_domain *d;
383         struct list_head *l;
384
385         if (id < 0)
386                 return ERR_PTR(-ENODEV);
387
388         list_for_each(l, &r->domains) {
389                 d = list_entry(l, struct rdt_domain, list);
390                 /* When id is found, return its domain. */
391                 if (id == d->id)
392                         return d;
393                 /* Stop searching when finding id's position in sorted list. */
394                 if (id < d->id)
395                         break;
396         }
397
398         if (pos)
399                 *pos = l;
400
401         return NULL;
402 }
403
404 void setup_default_ctrlval(struct rdt_resource *r, u32 *dc, u32 *dm)
405 {
406         struct rdt_hw_resource *hw_res = resctrl_to_arch_res(r);
407         int i;
408
409         /*
410          * Initialize the Control MSRs to having no control.
411          * For Cache Allocation: Set all bits in cbm
412          * For Memory Allocation: Set b/w requested to 100%
413          * and the bandwidth in MBps to U32_MAX
414          */
415         for (i = 0; i < hw_res->num_closid; i++, dc++, dm++) {
416                 *dc = r->default_ctrl;
417                 *dm = MBA_MAX_MBPS;
418         }
419 }
420
421 static int domain_setup_ctrlval(struct rdt_resource *r, struct rdt_domain *d)
422 {
423         struct rdt_hw_resource *hw_res = resctrl_to_arch_res(r);
424         struct rdt_hw_domain *hw_dom = resctrl_to_arch_dom(d);
425         struct msr_param m;
426         u32 *dc, *dm;
427
428         dc = kmalloc_array(hw_res->num_closid, sizeof(*hw_dom->ctrl_val),
429                            GFP_KERNEL);
430         if (!dc)
431                 return -ENOMEM;
432
433         dm = kmalloc_array(hw_res->num_closid, sizeof(*hw_dom->mbps_val),
434                            GFP_KERNEL);
435         if (!dm) {
436                 kfree(dc);
437                 return -ENOMEM;
438         }
439
440         hw_dom->ctrl_val = dc;
441         hw_dom->mbps_val = dm;
442         setup_default_ctrlval(r, dc, dm);
443
444         m.low = 0;
445         m.high = hw_res->num_closid;
446         hw_res->msr_update(d, &m, r);
447         return 0;
448 }
449
450 static int domain_setup_mon_state(struct rdt_resource *r, struct rdt_domain *d)
451 {
452         size_t tsize;
453
454         if (is_llc_occupancy_enabled()) {
455                 d->rmid_busy_llc = bitmap_zalloc(r->num_rmid, GFP_KERNEL);
456                 if (!d->rmid_busy_llc)
457                         return -ENOMEM;
458                 INIT_DELAYED_WORK(&d->cqm_limbo, cqm_handle_limbo);
459         }
460         if (is_mbm_total_enabled()) {
461                 tsize = sizeof(*d->mbm_total);
462                 d->mbm_total = kcalloc(r->num_rmid, tsize, GFP_KERNEL);
463                 if (!d->mbm_total) {
464                         bitmap_free(d->rmid_busy_llc);
465                         return -ENOMEM;
466                 }
467         }
468         if (is_mbm_local_enabled()) {
469                 tsize = sizeof(*d->mbm_local);
470                 d->mbm_local = kcalloc(r->num_rmid, tsize, GFP_KERNEL);
471                 if (!d->mbm_local) {
472                         bitmap_free(d->rmid_busy_llc);
473                         kfree(d->mbm_total);
474                         return -ENOMEM;
475                 }
476         }
477
478         if (is_mbm_enabled()) {
479                 INIT_DELAYED_WORK(&d->mbm_over, mbm_handle_overflow);
480                 mbm_setup_overflow_handler(d, MBM_OVERFLOW_INTERVAL);
481         }
482
483         return 0;
484 }
485
486 /*
487  * domain_add_cpu - Add a cpu to a resource's domain list.
488  *
489  * If an existing domain in the resource r's domain list matches the cpu's
490  * resource id, add the cpu in the domain.
491  *
492  * Otherwise, a new domain is allocated and inserted into the right position
493  * in the domain list sorted by id in ascending order.
494  *
495  * The order in the domain list is visible to users when we print entries
496  * in the schemata file and schemata input is validated to have the same order
497  * as this list.
498  */
499 static void domain_add_cpu(int cpu, struct rdt_resource *r)
500 {
501         int id = get_cpu_cacheinfo_id(cpu, r->cache_level);
502         struct list_head *add_pos = NULL;
503         struct rdt_hw_domain *hw_dom;
504         struct rdt_domain *d;
505
506         d = rdt_find_domain(r, id, &add_pos);
507         if (IS_ERR(d)) {
508                 pr_warn("Couldn't find cache id for CPU %d\n", cpu);
509                 return;
510         }
511
512         if (d) {
513                 cpumask_set_cpu(cpu, &d->cpu_mask);
514                 if (r->cache.arch_has_per_cpu_cfg)
515                         rdt_domain_reconfigure_cdp(r);
516                 return;
517         }
518
519         hw_dom = kzalloc_node(sizeof(*hw_dom), GFP_KERNEL, cpu_to_node(cpu));
520         if (!hw_dom)
521                 return;
522
523         d = &hw_dom->d_resctrl;
524         d->id = id;
525         cpumask_set_cpu(cpu, &d->cpu_mask);
526
527         rdt_domain_reconfigure_cdp(r);
528
529         if (r->alloc_capable && domain_setup_ctrlval(r, d)) {
530                 kfree(hw_dom);
531                 return;
532         }
533
534         if (r->mon_capable && domain_setup_mon_state(r, d)) {
535                 kfree(hw_dom->ctrl_val);
536                 kfree(hw_dom->mbps_val);
537                 kfree(hw_dom);
538                 return;
539         }
540
541         list_add_tail(&d->list, add_pos);
542
543         /*
544          * If resctrl is mounted, add
545          * per domain monitor data directories.
546          */
547         if (static_branch_unlikely(&rdt_mon_enable_key))
548                 mkdir_mondata_subdir_allrdtgrp(r, d);
549 }
550
551 static void domain_remove_cpu(int cpu, struct rdt_resource *r)
552 {
553         int id = get_cpu_cacheinfo_id(cpu, r->cache_level);
554         struct rdt_hw_domain *hw_dom;
555         struct rdt_domain *d;
556
557         d = rdt_find_domain(r, id, NULL);
558         if (IS_ERR_OR_NULL(d)) {
559                 pr_warn("Couldn't find cache id for CPU %d\n", cpu);
560                 return;
561         }
562         hw_dom = resctrl_to_arch_dom(d);
563
564         cpumask_clear_cpu(cpu, &d->cpu_mask);
565         if (cpumask_empty(&d->cpu_mask)) {
566                 /*
567                  * If resctrl is mounted, remove all the
568                  * per domain monitor data directories.
569                  */
570                 if (static_branch_unlikely(&rdt_mon_enable_key))
571                         rmdir_mondata_subdir_allrdtgrp(r, d->id);
572                 list_del(&d->list);
573                 if (r->mon_capable && is_mbm_enabled())
574                         cancel_delayed_work(&d->mbm_over);
575                 if (is_llc_occupancy_enabled() &&  has_busy_rmid(r, d)) {
576                         /*
577                          * When a package is going down, forcefully
578                          * decrement rmid->ebusy. There is no way to know
579                          * that the L3 was flushed and hence may lead to
580                          * incorrect counts in rare scenarios, but leaving
581                          * the RMID as busy creates RMID leaks if the
582                          * package never comes back.
583                          */
584                         __check_limbo(d, true);
585                         cancel_delayed_work(&d->cqm_limbo);
586                 }
587
588                 /*
589                  * rdt_domain "d" is going to be freed below, so clear
590                  * its pointer from pseudo_lock_region struct.
591                  */
592                 if (d->plr)
593                         d->plr->d = NULL;
594
595                 kfree(hw_dom->ctrl_val);
596                 kfree(hw_dom->mbps_val);
597                 bitmap_free(d->rmid_busy_llc);
598                 kfree(d->mbm_total);
599                 kfree(d->mbm_local);
600                 kfree(hw_dom);
601                 return;
602         }
603
604         if (r == &rdt_resources_all[RDT_RESOURCE_L3].r_resctrl) {
605                 if (is_mbm_enabled() && cpu == d->mbm_work_cpu) {
606                         cancel_delayed_work(&d->mbm_over);
607                         mbm_setup_overflow_handler(d, 0);
608                 }
609                 if (is_llc_occupancy_enabled() && cpu == d->cqm_work_cpu &&
610                     has_busy_rmid(r, d)) {
611                         cancel_delayed_work(&d->cqm_limbo);
612                         cqm_setup_limbo_handler(d, 0);
613                 }
614         }
615 }
616
617 static void clear_closid_rmid(int cpu)
618 {
619         struct resctrl_pqr_state *state = this_cpu_ptr(&pqr_state);
620
621         state->default_closid = 0;
622         state->default_rmid = 0;
623         state->cur_closid = 0;
624         state->cur_rmid = 0;
625         wrmsr(IA32_PQR_ASSOC, 0, 0);
626 }
627
628 static int resctrl_online_cpu(unsigned int cpu)
629 {
630         struct rdt_resource *r;
631
632         mutex_lock(&rdtgroup_mutex);
633         for_each_capable_rdt_resource(r)
634                 domain_add_cpu(cpu, r);
635         /* The cpu is set in default rdtgroup after online. */
636         cpumask_set_cpu(cpu, &rdtgroup_default.cpu_mask);
637         clear_closid_rmid(cpu);
638         mutex_unlock(&rdtgroup_mutex);
639
640         return 0;
641 }
642
643 static void clear_childcpus(struct rdtgroup *r, unsigned int cpu)
644 {
645         struct rdtgroup *cr;
646
647         list_for_each_entry(cr, &r->mon.crdtgrp_list, mon.crdtgrp_list) {
648                 if (cpumask_test_and_clear_cpu(cpu, &cr->cpu_mask)) {
649                         break;
650                 }
651         }
652 }
653
654 static int resctrl_offline_cpu(unsigned int cpu)
655 {
656         struct rdtgroup *rdtgrp;
657         struct rdt_resource *r;
658
659         mutex_lock(&rdtgroup_mutex);
660         for_each_capable_rdt_resource(r)
661                 domain_remove_cpu(cpu, r);
662         list_for_each_entry(rdtgrp, &rdt_all_groups, rdtgroup_list) {
663                 if (cpumask_test_and_clear_cpu(cpu, &rdtgrp->cpu_mask)) {
664                         clear_childcpus(rdtgrp, cpu);
665                         break;
666                 }
667         }
668         clear_closid_rmid(cpu);
669         mutex_unlock(&rdtgroup_mutex);
670
671         return 0;
672 }
673
674 /*
675  * Choose a width for the resource name and resource data based on the
676  * resource that has widest name and cbm.
677  */
678 static __init void rdt_init_padding(void)
679 {
680         struct rdt_resource *r;
681
682         for_each_alloc_capable_rdt_resource(r) {
683                 if (r->data_width > max_data_width)
684                         max_data_width = r->data_width;
685         }
686 }
687
688 enum {
689         RDT_FLAG_CMT,
690         RDT_FLAG_MBM_TOTAL,
691         RDT_FLAG_MBM_LOCAL,
692         RDT_FLAG_L3_CAT,
693         RDT_FLAG_L3_CDP,
694         RDT_FLAG_L2_CAT,
695         RDT_FLAG_L2_CDP,
696         RDT_FLAG_MBA,
697 };
698
699 #define RDT_OPT(idx, n, f)      \
700 [idx] = {                       \
701         .name = n,              \
702         .flag = f               \
703 }
704
705 struct rdt_options {
706         char    *name;
707         int     flag;
708         bool    force_off, force_on;
709 };
710
711 static struct rdt_options rdt_options[]  __initdata = {
712         RDT_OPT(RDT_FLAG_CMT,       "cmt",      X86_FEATURE_CQM_OCCUP_LLC),
713         RDT_OPT(RDT_FLAG_MBM_TOTAL, "mbmtotal", X86_FEATURE_CQM_MBM_TOTAL),
714         RDT_OPT(RDT_FLAG_MBM_LOCAL, "mbmlocal", X86_FEATURE_CQM_MBM_LOCAL),
715         RDT_OPT(RDT_FLAG_L3_CAT,    "l3cat",    X86_FEATURE_CAT_L3),
716         RDT_OPT(RDT_FLAG_L3_CDP,    "l3cdp",    X86_FEATURE_CDP_L3),
717         RDT_OPT(RDT_FLAG_L2_CAT,    "l2cat",    X86_FEATURE_CAT_L2),
718         RDT_OPT(RDT_FLAG_L2_CDP,    "l2cdp",    X86_FEATURE_CDP_L2),
719         RDT_OPT(RDT_FLAG_MBA,       "mba",      X86_FEATURE_MBA),
720 };
721 #define NUM_RDT_OPTIONS ARRAY_SIZE(rdt_options)
722
723 static int __init set_rdt_options(char *str)
724 {
725         struct rdt_options *o;
726         bool force_off;
727         char *tok;
728
729         if (*str == '=')
730                 str++;
731         while ((tok = strsep(&str, ",")) != NULL) {
732                 force_off = *tok == '!';
733                 if (force_off)
734                         tok++;
735                 for (o = rdt_options; o < &rdt_options[NUM_RDT_OPTIONS]; o++) {
736                         if (strcmp(tok, o->name) == 0) {
737                                 if (force_off)
738                                         o->force_off = true;
739                                 else
740                                         o->force_on = true;
741                                 break;
742                         }
743                 }
744         }
745         return 1;
746 }
747 __setup("rdt", set_rdt_options);
748
749 static bool __init rdt_cpu_has(int flag)
750 {
751         bool ret = boot_cpu_has(flag);
752         struct rdt_options *o;
753
754         if (!ret)
755                 return ret;
756
757         for (o = rdt_options; o < &rdt_options[NUM_RDT_OPTIONS]; o++) {
758                 if (flag == o->flag) {
759                         if (o->force_off)
760                                 ret = false;
761                         if (o->force_on)
762                                 ret = true;
763                         break;
764                 }
765         }
766         return ret;
767 }
768
769 static __init bool get_mem_config(void)
770 {
771         struct rdt_hw_resource *hw_res = &rdt_resources_all[RDT_RESOURCE_MBA];
772
773         if (!rdt_cpu_has(X86_FEATURE_MBA))
774                 return false;
775
776         if (boot_cpu_data.x86_vendor == X86_VENDOR_INTEL)
777                 return __get_mem_config_intel(&hw_res->r_resctrl);
778         else if (boot_cpu_data.x86_vendor == X86_VENDOR_AMD)
779                 return __rdt_get_mem_config_amd(&hw_res->r_resctrl);
780
781         return false;
782 }
783
784 static __init bool get_rdt_alloc_resources(void)
785 {
786         struct rdt_resource *r;
787         bool ret = false;
788
789         if (rdt_alloc_capable)
790                 return true;
791
792         if (!boot_cpu_has(X86_FEATURE_RDT_A))
793                 return false;
794
795         if (rdt_cpu_has(X86_FEATURE_CAT_L3)) {
796                 r = &rdt_resources_all[RDT_RESOURCE_L3].r_resctrl;
797                 rdt_get_cache_alloc_cfg(1, r);
798                 if (rdt_cpu_has(X86_FEATURE_CDP_L3))
799                         rdt_get_cdp_l3_config();
800                 ret = true;
801         }
802         if (rdt_cpu_has(X86_FEATURE_CAT_L2)) {
803                 /* CPUID 0x10.2 fields are same format at 0x10.1 */
804                 r = &rdt_resources_all[RDT_RESOURCE_L2].r_resctrl;
805                 rdt_get_cache_alloc_cfg(2, r);
806                 if (rdt_cpu_has(X86_FEATURE_CDP_L2))
807                         rdt_get_cdp_l2_config();
808                 ret = true;
809         }
810
811         if (get_mem_config())
812                 ret = true;
813
814         return ret;
815 }
816
817 static __init bool get_rdt_mon_resources(void)
818 {
819         struct rdt_resource *r = &rdt_resources_all[RDT_RESOURCE_L3].r_resctrl;
820
821         if (rdt_cpu_has(X86_FEATURE_CQM_OCCUP_LLC))
822                 rdt_mon_features |= (1 << QOS_L3_OCCUP_EVENT_ID);
823         if (rdt_cpu_has(X86_FEATURE_CQM_MBM_TOTAL))
824                 rdt_mon_features |= (1 << QOS_L3_MBM_TOTAL_EVENT_ID);
825         if (rdt_cpu_has(X86_FEATURE_CQM_MBM_LOCAL))
826                 rdt_mon_features |= (1 << QOS_L3_MBM_LOCAL_EVENT_ID);
827
828         if (!rdt_mon_features)
829                 return false;
830
831         return !rdt_get_mon_l3_config(r);
832 }
833
834 static __init void __check_quirks_intel(void)
835 {
836         switch (boot_cpu_data.x86_model) {
837         case INTEL_FAM6_HASWELL_X:
838                 if (!rdt_options[RDT_FLAG_L3_CAT].force_off)
839                         cache_alloc_hsw_probe();
840                 break;
841         case INTEL_FAM6_SKYLAKE_X:
842                 if (boot_cpu_data.x86_stepping <= 4)
843                         set_rdt_options("!cmt,!mbmtotal,!mbmlocal,!l3cat");
844                 else
845                         set_rdt_options("!l3cat");
846                 fallthrough;
847         case INTEL_FAM6_BROADWELL_X:
848                 intel_rdt_mbm_apply_quirk();
849                 break;
850         }
851 }
852
853 static __init void check_quirks(void)
854 {
855         if (boot_cpu_data.x86_vendor == X86_VENDOR_INTEL)
856                 __check_quirks_intel();
857 }
858
859 static __init bool get_rdt_resources(void)
860 {
861         rdt_alloc_capable = get_rdt_alloc_resources();
862         rdt_mon_capable = get_rdt_mon_resources();
863
864         return (rdt_mon_capable || rdt_alloc_capable);
865 }
866
867 static __init void rdt_init_res_defs_intel(void)
868 {
869         struct rdt_hw_resource *hw_res;
870         struct rdt_resource *r;
871
872         for_each_rdt_resource(r) {
873                 hw_res = resctrl_to_arch_res(r);
874
875                 if (r->rid == RDT_RESOURCE_L3 ||
876                     r->rid == RDT_RESOURCE_L2) {
877                         r->cache.arch_has_sparse_bitmaps = false;
878                         r->cache.arch_has_empty_bitmaps = false;
879                         r->cache.arch_has_per_cpu_cfg = false;
880                 } else if (r->rid == RDT_RESOURCE_MBA) {
881                         hw_res->msr_base = MSR_IA32_MBA_THRTL_BASE;
882                         hw_res->msr_update = mba_wrmsr_intel;
883                 }
884         }
885 }
886
887 static __init void rdt_init_res_defs_amd(void)
888 {
889         struct rdt_hw_resource *hw_res;
890         struct rdt_resource *r;
891
892         for_each_rdt_resource(r) {
893                 hw_res = resctrl_to_arch_res(r);
894
895                 if (r->rid == RDT_RESOURCE_L3 ||
896                     r->rid == RDT_RESOURCE_L2) {
897                         r->cache.arch_has_sparse_bitmaps = true;
898                         r->cache.arch_has_empty_bitmaps = true;
899                         r->cache.arch_has_per_cpu_cfg = true;
900                 } else if (r->rid == RDT_RESOURCE_MBA) {
901                         hw_res->msr_base = MSR_IA32_MBA_BW_BASE;
902                         hw_res->msr_update = mba_wrmsr_amd;
903                 }
904         }
905 }
906
907 static __init void rdt_init_res_defs(void)
908 {
909         if (boot_cpu_data.x86_vendor == X86_VENDOR_INTEL)
910                 rdt_init_res_defs_intel();
911         else if (boot_cpu_data.x86_vendor == X86_VENDOR_AMD)
912                 rdt_init_res_defs_amd();
913 }
914
915 static enum cpuhp_state rdt_online;
916
917 /* Runs once on the BSP during boot. */
918 void resctrl_cpu_detect(struct cpuinfo_x86 *c)
919 {
920         if (!cpu_has(c, X86_FEATURE_CQM_LLC)) {
921                 c->x86_cache_max_rmid  = -1;
922                 c->x86_cache_occ_scale = -1;
923                 c->x86_cache_mbm_width_offset = -1;
924                 return;
925         }
926
927         /* will be overridden if occupancy monitoring exists */
928         c->x86_cache_max_rmid = cpuid_ebx(0xf);
929
930         if (cpu_has(c, X86_FEATURE_CQM_OCCUP_LLC) ||
931             cpu_has(c, X86_FEATURE_CQM_MBM_TOTAL) ||
932             cpu_has(c, X86_FEATURE_CQM_MBM_LOCAL)) {
933                 u32 eax, ebx, ecx, edx;
934
935                 /* QoS sub-leaf, EAX=0Fh, ECX=1 */
936                 cpuid_count(0xf, 1, &eax, &ebx, &ecx, &edx);
937
938                 c->x86_cache_max_rmid  = ecx;
939                 c->x86_cache_occ_scale = ebx;
940                 c->x86_cache_mbm_width_offset = eax & 0xff;
941
942                 if (c->x86_vendor == X86_VENDOR_AMD && !c->x86_cache_mbm_width_offset)
943                         c->x86_cache_mbm_width_offset = MBM_CNTR_WIDTH_OFFSET_AMD;
944         }
945 }
946
947 static int __init resctrl_late_init(void)
948 {
949         struct rdt_resource *r;
950         int state, ret;
951
952         /*
953          * Initialize functions(or definitions) that are different
954          * between vendors here.
955          */
956         rdt_init_res_defs();
957
958         check_quirks();
959
960         if (!get_rdt_resources())
961                 return -ENODEV;
962
963         rdt_init_padding();
964
965         state = cpuhp_setup_state(CPUHP_AP_ONLINE_DYN,
966                                   "x86/resctrl/cat:online:",
967                                   resctrl_online_cpu, resctrl_offline_cpu);
968         if (state < 0)
969                 return state;
970
971         ret = rdtgroup_init();
972         if (ret) {
973                 cpuhp_remove_state(state);
974                 return ret;
975         }
976         rdt_online = state;
977
978         for_each_alloc_capable_rdt_resource(r)
979                 pr_info("%s allocation detected\n", r->name);
980
981         for_each_mon_capable_rdt_resource(r)
982                 pr_info("%s monitoring detected\n", r->name);
983
984         return 0;
985 }
986
987 late_initcall(resctrl_late_init);
988
989 static void __exit resctrl_exit(void)
990 {
991         cpuhp_remove_state(rdt_online);
992         rdtgroup_exit();
993 }
994
995 __exitcall(resctrl_exit);