xref: /openbmc/linux/arch/arm/common/bL_switcher.c (revision feac8c8b)
1 /*
2  * arch/arm/common/bL_switcher.c -- big.LITTLE cluster switcher core driver
3  *
4  * Created by:	Nicolas Pitre, March 2012
5  * Copyright:	(C) 2012-2013  Linaro Limited
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License version 2 as
9  * published by the Free Software Foundation.
10  */
11 
12 #include <linux/atomic.h>
13 #include <linux/init.h>
14 #include <linux/kernel.h>
15 #include <linux/module.h>
16 #include <linux/sched/signal.h>
17 #include <uapi/linux/sched/types.h>
18 #include <linux/interrupt.h>
19 #include <linux/cpu_pm.h>
20 #include <linux/cpu.h>
21 #include <linux/cpumask.h>
22 #include <linux/kthread.h>
23 #include <linux/wait.h>
24 #include <linux/time.h>
25 #include <linux/clockchips.h>
26 #include <linux/hrtimer.h>
27 #include <linux/tick.h>
28 #include <linux/notifier.h>
29 #include <linux/mm.h>
30 #include <linux/mutex.h>
31 #include <linux/smp.h>
32 #include <linux/spinlock.h>
33 #include <linux/string.h>
34 #include <linux/sysfs.h>
35 #include <linux/irqchip/arm-gic.h>
36 #include <linux/moduleparam.h>
37 
38 #include <asm/smp_plat.h>
39 #include <asm/cputype.h>
40 #include <asm/suspend.h>
41 #include <asm/mcpm.h>
42 #include <asm/bL_switcher.h>
43 
44 #define CREATE_TRACE_POINTS
45 #include <trace/events/power_cpu_migrate.h>
46 
47 
48 /*
49  * Use our own MPIDR accessors as the generic ones in asm/cputype.h have
50  * __attribute_const__ and we don't want the compiler to assume any
51  * constness here as the value _does_ change along some code paths.
52  */
53 
54 static int read_mpidr(void)
55 {
56 	unsigned int id;
57 	asm volatile ("mrc p15, 0, %0, c0, c0, 5" : "=r" (id));
58 	return id & MPIDR_HWID_BITMASK;
59 }
60 
61 /*
62  * bL switcher core code.
63  */
64 
65 static void bL_do_switch(void *_arg)
66 {
67 	unsigned ib_mpidr, ib_cpu, ib_cluster;
68 	long volatile handshake, **handshake_ptr = _arg;
69 
70 	pr_debug("%s\n", __func__);
71 
72 	ib_mpidr = cpu_logical_map(smp_processor_id());
73 	ib_cpu = MPIDR_AFFINITY_LEVEL(ib_mpidr, 0);
74 	ib_cluster = MPIDR_AFFINITY_LEVEL(ib_mpidr, 1);
75 
76 	/* Advertise our handshake location */
77 	if (handshake_ptr) {
78 		handshake = 0;
79 		*handshake_ptr = &handshake;
80 	} else
81 		handshake = -1;
82 
83 	/*
84 	 * Our state has been saved at this point.  Let's release our
85 	 * inbound CPU.
86 	 */
87 	mcpm_set_entry_vector(ib_cpu, ib_cluster, cpu_resume);
88 	sev();
89 
90 	/*
91 	 * From this point, we must assume that our counterpart CPU might
92 	 * have taken over in its parallel world already, as if execution
93 	 * just returned from cpu_suspend().  It is therefore important to
94 	 * be very careful not to make any change the other guy is not
95 	 * expecting.  This is why we need stack isolation.
96 	 *
97 	 * Fancy under cover tasks could be performed here.  For now
98 	 * we have none.
99 	 */
100 
101 	/*
102 	 * Let's wait until our inbound is alive.
103 	 */
104 	while (!handshake) {
105 		wfe();
106 		smp_mb();
107 	}
108 
109 	/* Let's put ourself down. */
110 	mcpm_cpu_power_down();
111 
112 	/* should never get here */
113 	BUG();
114 }
115 
116 /*
117  * Stack isolation.  To ensure 'current' remains valid, we just use another
118  * piece of our thread's stack space which should be fairly lightly used.
119  * The selected area starts just above the thread_info structure located
120  * at the very bottom of the stack, aligned to a cache line, and indexed
121  * with the cluster number.
122  */
123 #define STACK_SIZE 512
124 extern void call_with_stack(void (*fn)(void *), void *arg, void *sp);
125 static int bL_switchpoint(unsigned long _arg)
126 {
127 	unsigned int mpidr = read_mpidr();
128 	unsigned int clusterid = MPIDR_AFFINITY_LEVEL(mpidr, 1);
129 	void *stack = current_thread_info() + 1;
130 	stack = PTR_ALIGN(stack, L1_CACHE_BYTES);
131 	stack += clusterid * STACK_SIZE + STACK_SIZE;
132 	call_with_stack(bL_do_switch, (void *)_arg, stack);
133 	BUG();
134 }
135 
136 /*
137  * Generic switcher interface
138  */
139 
140 static unsigned int bL_gic_id[MAX_CPUS_PER_CLUSTER][MAX_NR_CLUSTERS];
141 static int bL_switcher_cpu_pairing[NR_CPUS];
142 
143 /*
144  * bL_switch_to - Switch to a specific cluster for the current CPU
145  * @new_cluster_id: the ID of the cluster to switch to.
146  *
147  * This function must be called on the CPU to be switched.
148  * Returns 0 on success, else a negative status code.
149  */
150 static int bL_switch_to(unsigned int new_cluster_id)
151 {
152 	unsigned int mpidr, this_cpu, that_cpu;
153 	unsigned int ob_mpidr, ob_cpu, ob_cluster, ib_mpidr, ib_cpu, ib_cluster;
154 	struct completion inbound_alive;
155 	long volatile *handshake_ptr;
156 	int ipi_nr, ret;
157 
158 	this_cpu = smp_processor_id();
159 	ob_mpidr = read_mpidr();
160 	ob_cpu = MPIDR_AFFINITY_LEVEL(ob_mpidr, 0);
161 	ob_cluster = MPIDR_AFFINITY_LEVEL(ob_mpidr, 1);
162 	BUG_ON(cpu_logical_map(this_cpu) != ob_mpidr);
163 
164 	if (new_cluster_id == ob_cluster)
165 		return 0;
166 
167 	that_cpu = bL_switcher_cpu_pairing[this_cpu];
168 	ib_mpidr = cpu_logical_map(that_cpu);
169 	ib_cpu = MPIDR_AFFINITY_LEVEL(ib_mpidr, 0);
170 	ib_cluster = MPIDR_AFFINITY_LEVEL(ib_mpidr, 1);
171 
172 	pr_debug("before switch: CPU %d MPIDR %#x -> %#x\n",
173 		 this_cpu, ob_mpidr, ib_mpidr);
174 
175 	this_cpu = smp_processor_id();
176 
177 	/* Close the gate for our entry vectors */
178 	mcpm_set_entry_vector(ob_cpu, ob_cluster, NULL);
179 	mcpm_set_entry_vector(ib_cpu, ib_cluster, NULL);
180 
181 	/* Install our "inbound alive" notifier. */
182 	init_completion(&inbound_alive);
183 	ipi_nr = register_ipi_completion(&inbound_alive, this_cpu);
184 	ipi_nr |= ((1 << 16) << bL_gic_id[ob_cpu][ob_cluster]);
185 	mcpm_set_early_poke(ib_cpu, ib_cluster, gic_get_sgir_physaddr(), ipi_nr);
186 
187 	/*
188 	 * Let's wake up the inbound CPU now in case it requires some delay
189 	 * to come online, but leave it gated in our entry vector code.
190 	 */
191 	ret = mcpm_cpu_power_up(ib_cpu, ib_cluster);
192 	if (ret) {
193 		pr_err("%s: mcpm_cpu_power_up() returned %d\n", __func__, ret);
194 		return ret;
195 	}
196 
197 	/*
198 	 * Raise a SGI on the inbound CPU to make sure it doesn't stall
199 	 * in a possible WFI, such as in bL_power_down().
200 	 */
201 	gic_send_sgi(bL_gic_id[ib_cpu][ib_cluster], 0);
202 
203 	/*
204 	 * Wait for the inbound to come up.  This allows for other
205 	 * tasks to be scheduled in the mean time.
206 	 */
207 	wait_for_completion(&inbound_alive);
208 	mcpm_set_early_poke(ib_cpu, ib_cluster, 0, 0);
209 
210 	/*
211 	 * From this point we are entering the switch critical zone
212 	 * and can't take any interrupts anymore.
213 	 */
214 	local_irq_disable();
215 	local_fiq_disable();
216 	trace_cpu_migrate_begin(ktime_get_real_ns(), ob_mpidr);
217 
218 	/* redirect GIC's SGIs to our counterpart */
219 	gic_migrate_target(bL_gic_id[ib_cpu][ib_cluster]);
220 
221 	tick_suspend_local();
222 
223 	ret = cpu_pm_enter();
224 
225 	/* we can not tolerate errors at this point */
226 	if (ret)
227 		panic("%s: cpu_pm_enter() returned %d\n", __func__, ret);
228 
229 	/* Swap the physical CPUs in the logical map for this logical CPU. */
230 	cpu_logical_map(this_cpu) = ib_mpidr;
231 	cpu_logical_map(that_cpu) = ob_mpidr;
232 
233 	/* Let's do the actual CPU switch. */
234 	ret = cpu_suspend((unsigned long)&handshake_ptr, bL_switchpoint);
235 	if (ret > 0)
236 		panic("%s: cpu_suspend() returned %d\n", __func__, ret);
237 
238 	/* We are executing on the inbound CPU at this point */
239 	mpidr = read_mpidr();
240 	pr_debug("after switch: CPU %d MPIDR %#x\n", this_cpu, mpidr);
241 	BUG_ON(mpidr != ib_mpidr);
242 
243 	mcpm_cpu_powered_up();
244 
245 	ret = cpu_pm_exit();
246 
247 	tick_resume_local();
248 
249 	trace_cpu_migrate_finish(ktime_get_real_ns(), ib_mpidr);
250 	local_fiq_enable();
251 	local_irq_enable();
252 
253 	*handshake_ptr = 1;
254 	dsb_sev();
255 
256 	if (ret)
257 		pr_err("%s exiting with error %d\n", __func__, ret);
258 	return ret;
259 }
260 
261 struct bL_thread {
262 	spinlock_t lock;
263 	struct task_struct *task;
264 	wait_queue_head_t wq;
265 	int wanted_cluster;
266 	struct completion started;
267 	bL_switch_completion_handler completer;
268 	void *completer_cookie;
269 };
270 
271 static struct bL_thread bL_threads[NR_CPUS];
272 
273 static int bL_switcher_thread(void *arg)
274 {
275 	struct bL_thread *t = arg;
276 	struct sched_param param = { .sched_priority = 1 };
277 	int cluster;
278 	bL_switch_completion_handler completer;
279 	void *completer_cookie;
280 
281 	sched_setscheduler_nocheck(current, SCHED_FIFO, &param);
282 	complete(&t->started);
283 
284 	do {
285 		if (signal_pending(current))
286 			flush_signals(current);
287 		wait_event_interruptible(t->wq,
288 				t->wanted_cluster != -1 ||
289 				kthread_should_stop());
290 
291 		spin_lock(&t->lock);
292 		cluster = t->wanted_cluster;
293 		completer = t->completer;
294 		completer_cookie = t->completer_cookie;
295 		t->wanted_cluster = -1;
296 		t->completer = NULL;
297 		spin_unlock(&t->lock);
298 
299 		if (cluster != -1) {
300 			bL_switch_to(cluster);
301 
302 			if (completer)
303 				completer(completer_cookie);
304 		}
305 	} while (!kthread_should_stop());
306 
307 	return 0;
308 }
309 
310 static struct task_struct *bL_switcher_thread_create(int cpu, void *arg)
311 {
312 	struct task_struct *task;
313 
314 	task = kthread_create_on_node(bL_switcher_thread, arg,
315 				      cpu_to_node(cpu), "kswitcher_%d", cpu);
316 	if (!IS_ERR(task)) {
317 		kthread_bind(task, cpu);
318 		wake_up_process(task);
319 	} else
320 		pr_err("%s failed for CPU %d\n", __func__, cpu);
321 	return task;
322 }
323 
324 /*
325  * bL_switch_request_cb - Switch to a specific cluster for the given CPU,
326  *      with completion notification via a callback
327  *
328  * @cpu: the CPU to switch
329  * @new_cluster_id: the ID of the cluster to switch to.
330  * @completer: switch completion callback.  if non-NULL,
331  *	@completer(@completer_cookie) will be called on completion of
332  *	the switch, in non-atomic context.
333  * @completer_cookie: opaque context argument for @completer.
334  *
335  * This function causes a cluster switch on the given CPU by waking up
336  * the appropriate switcher thread.  This function may or may not return
337  * before the switch has occurred.
338  *
339  * If a @completer callback function is supplied, it will be called when
340  * the switch is complete.  This can be used to determine asynchronously
341  * when the switch is complete, regardless of when bL_switch_request()
342  * returns.  When @completer is supplied, no new switch request is permitted
343  * for the affected CPU until after the switch is complete, and @completer
344  * has returned.
345  */
346 int bL_switch_request_cb(unsigned int cpu, unsigned int new_cluster_id,
347 			 bL_switch_completion_handler completer,
348 			 void *completer_cookie)
349 {
350 	struct bL_thread *t;
351 
352 	if (cpu >= ARRAY_SIZE(bL_threads)) {
353 		pr_err("%s: cpu %d out of bounds\n", __func__, cpu);
354 		return -EINVAL;
355 	}
356 
357 	t = &bL_threads[cpu];
358 
359 	if (IS_ERR(t->task))
360 		return PTR_ERR(t->task);
361 	if (!t->task)
362 		return -ESRCH;
363 
364 	spin_lock(&t->lock);
365 	if (t->completer) {
366 		spin_unlock(&t->lock);
367 		return -EBUSY;
368 	}
369 	t->completer = completer;
370 	t->completer_cookie = completer_cookie;
371 	t->wanted_cluster = new_cluster_id;
372 	spin_unlock(&t->lock);
373 	wake_up(&t->wq);
374 	return 0;
375 }
376 EXPORT_SYMBOL_GPL(bL_switch_request_cb);
377 
378 /*
379  * Activation and configuration code.
380  */
381 
382 static DEFINE_MUTEX(bL_switcher_activation_lock);
383 static BLOCKING_NOTIFIER_HEAD(bL_activation_notifier);
384 static unsigned int bL_switcher_active;
385 static unsigned int bL_switcher_cpu_original_cluster[NR_CPUS];
386 static cpumask_t bL_switcher_removed_logical_cpus;
387 
388 int bL_switcher_register_notifier(struct notifier_block *nb)
389 {
390 	return blocking_notifier_chain_register(&bL_activation_notifier, nb);
391 }
392 EXPORT_SYMBOL_GPL(bL_switcher_register_notifier);
393 
394 int bL_switcher_unregister_notifier(struct notifier_block *nb)
395 {
396 	return blocking_notifier_chain_unregister(&bL_activation_notifier, nb);
397 }
398 EXPORT_SYMBOL_GPL(bL_switcher_unregister_notifier);
399 
400 static int bL_activation_notify(unsigned long val)
401 {
402 	int ret;
403 
404 	ret = blocking_notifier_call_chain(&bL_activation_notifier, val, NULL);
405 	if (ret & NOTIFY_STOP_MASK)
406 		pr_err("%s: notifier chain failed with status 0x%x\n",
407 			__func__, ret);
408 	return notifier_to_errno(ret);
409 }
410 
411 static void bL_switcher_restore_cpus(void)
412 {
413 	int i;
414 
415 	for_each_cpu(i, &bL_switcher_removed_logical_cpus) {
416 		struct device *cpu_dev = get_cpu_device(i);
417 		int ret = device_online(cpu_dev);
418 		if (ret)
419 			dev_err(cpu_dev, "switcher: unable to restore CPU\n");
420 	}
421 }
422 
423 static int bL_switcher_halve_cpus(void)
424 {
425 	int i, j, cluster_0, gic_id, ret;
426 	unsigned int cpu, cluster, mask;
427 	cpumask_t available_cpus;
428 
429 	/* First pass to validate what we have */
430 	mask = 0;
431 	for_each_online_cpu(i) {
432 		cpu = MPIDR_AFFINITY_LEVEL(cpu_logical_map(i), 0);
433 		cluster = MPIDR_AFFINITY_LEVEL(cpu_logical_map(i), 1);
434 		if (cluster >= 2) {
435 			pr_err("%s: only dual cluster systems are supported\n", __func__);
436 			return -EINVAL;
437 		}
438 		if (WARN_ON(cpu >= MAX_CPUS_PER_CLUSTER))
439 			return -EINVAL;
440 		mask |= (1 << cluster);
441 	}
442 	if (mask != 3) {
443 		pr_err("%s: no CPU pairing possible\n", __func__);
444 		return -EINVAL;
445 	}
446 
447 	/*
448 	 * Now let's do the pairing.  We match each CPU with another CPU
449 	 * from a different cluster.  To get a uniform scheduling behavior
450 	 * without fiddling with CPU topology and compute capacity data,
451 	 * we'll use logical CPUs initially belonging to the same cluster.
452 	 */
453 	memset(bL_switcher_cpu_pairing, -1, sizeof(bL_switcher_cpu_pairing));
454 	cpumask_copy(&available_cpus, cpu_online_mask);
455 	cluster_0 = -1;
456 	for_each_cpu(i, &available_cpus) {
457 		int match = -1;
458 		cluster = MPIDR_AFFINITY_LEVEL(cpu_logical_map(i), 1);
459 		if (cluster_0 == -1)
460 			cluster_0 = cluster;
461 		if (cluster != cluster_0)
462 			continue;
463 		cpumask_clear_cpu(i, &available_cpus);
464 		for_each_cpu(j, &available_cpus) {
465 			cluster = MPIDR_AFFINITY_LEVEL(cpu_logical_map(j), 1);
466 			/*
467 			 * Let's remember the last match to create "odd"
468 			 * pairings on purpose in order for other code not
469 			 * to assume any relation between physical and
470 			 * logical CPU numbers.
471 			 */
472 			if (cluster != cluster_0)
473 				match = j;
474 		}
475 		if (match != -1) {
476 			bL_switcher_cpu_pairing[i] = match;
477 			cpumask_clear_cpu(match, &available_cpus);
478 			pr_info("CPU%d paired with CPU%d\n", i, match);
479 		}
480 	}
481 
482 	/*
483 	 * Now we disable the unwanted CPUs i.e. everything that has no
484 	 * pairing information (that includes the pairing counterparts).
485 	 */
486 	cpumask_clear(&bL_switcher_removed_logical_cpus);
487 	for_each_online_cpu(i) {
488 		cpu = MPIDR_AFFINITY_LEVEL(cpu_logical_map(i), 0);
489 		cluster = MPIDR_AFFINITY_LEVEL(cpu_logical_map(i), 1);
490 
491 		/* Let's take note of the GIC ID for this CPU */
492 		gic_id = gic_get_cpu_id(i);
493 		if (gic_id < 0) {
494 			pr_err("%s: bad GIC ID for CPU %d\n", __func__, i);
495 			bL_switcher_restore_cpus();
496 			return -EINVAL;
497 		}
498 		bL_gic_id[cpu][cluster] = gic_id;
499 		pr_info("GIC ID for CPU %u cluster %u is %u\n",
500 			cpu, cluster, gic_id);
501 
502 		if (bL_switcher_cpu_pairing[i] != -1) {
503 			bL_switcher_cpu_original_cluster[i] = cluster;
504 			continue;
505 		}
506 
507 		ret = device_offline(get_cpu_device(i));
508 		if (ret) {
509 			bL_switcher_restore_cpus();
510 			return ret;
511 		}
512 		cpumask_set_cpu(i, &bL_switcher_removed_logical_cpus);
513 	}
514 
515 	return 0;
516 }
517 
518 /* Determine the logical CPU a given physical CPU is grouped on. */
519 int bL_switcher_get_logical_index(u32 mpidr)
520 {
521 	int cpu;
522 
523 	if (!bL_switcher_active)
524 		return -EUNATCH;
525 
526 	mpidr &= MPIDR_HWID_BITMASK;
527 	for_each_online_cpu(cpu) {
528 		int pairing = bL_switcher_cpu_pairing[cpu];
529 		if (pairing == -1)
530 			continue;
531 		if ((mpidr == cpu_logical_map(cpu)) ||
532 		    (mpidr == cpu_logical_map(pairing)))
533 			return cpu;
534 	}
535 	return -EINVAL;
536 }
537 
538 static void bL_switcher_trace_trigger_cpu(void *__always_unused info)
539 {
540 	trace_cpu_migrate_current(ktime_get_real_ns(), read_mpidr());
541 }
542 
543 int bL_switcher_trace_trigger(void)
544 {
545 	int ret;
546 
547 	preempt_disable();
548 
549 	bL_switcher_trace_trigger_cpu(NULL);
550 	ret = smp_call_function(bL_switcher_trace_trigger_cpu, NULL, true);
551 
552 	preempt_enable();
553 
554 	return ret;
555 }
556 EXPORT_SYMBOL_GPL(bL_switcher_trace_trigger);
557 
558 static int bL_switcher_enable(void)
559 {
560 	int cpu, ret;
561 
562 	mutex_lock(&bL_switcher_activation_lock);
563 	lock_device_hotplug();
564 	if (bL_switcher_active) {
565 		unlock_device_hotplug();
566 		mutex_unlock(&bL_switcher_activation_lock);
567 		return 0;
568 	}
569 
570 	pr_info("big.LITTLE switcher initializing\n");
571 
572 	ret = bL_activation_notify(BL_NOTIFY_PRE_ENABLE);
573 	if (ret)
574 		goto error;
575 
576 	ret = bL_switcher_halve_cpus();
577 	if (ret)
578 		goto error;
579 
580 	bL_switcher_trace_trigger();
581 
582 	for_each_online_cpu(cpu) {
583 		struct bL_thread *t = &bL_threads[cpu];
584 		spin_lock_init(&t->lock);
585 		init_waitqueue_head(&t->wq);
586 		init_completion(&t->started);
587 		t->wanted_cluster = -1;
588 		t->task = bL_switcher_thread_create(cpu, t);
589 	}
590 
591 	bL_switcher_active = 1;
592 	bL_activation_notify(BL_NOTIFY_POST_ENABLE);
593 	pr_info("big.LITTLE switcher initialized\n");
594 	goto out;
595 
596 error:
597 	pr_warn("big.LITTLE switcher initialization failed\n");
598 	bL_activation_notify(BL_NOTIFY_POST_DISABLE);
599 
600 out:
601 	unlock_device_hotplug();
602 	mutex_unlock(&bL_switcher_activation_lock);
603 	return ret;
604 }
605 
606 #ifdef CONFIG_SYSFS
607 
608 static void bL_switcher_disable(void)
609 {
610 	unsigned int cpu, cluster;
611 	struct bL_thread *t;
612 	struct task_struct *task;
613 
614 	mutex_lock(&bL_switcher_activation_lock);
615 	lock_device_hotplug();
616 
617 	if (!bL_switcher_active)
618 		goto out;
619 
620 	if (bL_activation_notify(BL_NOTIFY_PRE_DISABLE) != 0) {
621 		bL_activation_notify(BL_NOTIFY_POST_ENABLE);
622 		goto out;
623 	}
624 
625 	bL_switcher_active = 0;
626 
627 	/*
628 	 * To deactivate the switcher, we must shut down the switcher
629 	 * threads to prevent any other requests from being accepted.
630 	 * Then, if the final cluster for given logical CPU is not the
631 	 * same as the original one, we'll recreate a switcher thread
632 	 * just for the purpose of switching the CPU back without any
633 	 * possibility for interference from external requests.
634 	 */
635 	for_each_online_cpu(cpu) {
636 		t = &bL_threads[cpu];
637 		task = t->task;
638 		t->task = NULL;
639 		if (!task || IS_ERR(task))
640 			continue;
641 		kthread_stop(task);
642 		/* no more switch may happen on this CPU at this point */
643 		cluster = MPIDR_AFFINITY_LEVEL(cpu_logical_map(cpu), 1);
644 		if (cluster == bL_switcher_cpu_original_cluster[cpu])
645 			continue;
646 		init_completion(&t->started);
647 		t->wanted_cluster = bL_switcher_cpu_original_cluster[cpu];
648 		task = bL_switcher_thread_create(cpu, t);
649 		if (!IS_ERR(task)) {
650 			wait_for_completion(&t->started);
651 			kthread_stop(task);
652 			cluster = MPIDR_AFFINITY_LEVEL(cpu_logical_map(cpu), 1);
653 			if (cluster == bL_switcher_cpu_original_cluster[cpu])
654 				continue;
655 		}
656 		/* If execution gets here, we're in trouble. */
657 		pr_crit("%s: unable to restore original cluster for CPU %d\n",
658 			__func__, cpu);
659 		pr_crit("%s: CPU %d can't be restored\n",
660 			__func__, bL_switcher_cpu_pairing[cpu]);
661 		cpumask_clear_cpu(bL_switcher_cpu_pairing[cpu],
662 				  &bL_switcher_removed_logical_cpus);
663 	}
664 
665 	bL_switcher_restore_cpus();
666 	bL_switcher_trace_trigger();
667 
668 	bL_activation_notify(BL_NOTIFY_POST_DISABLE);
669 
670 out:
671 	unlock_device_hotplug();
672 	mutex_unlock(&bL_switcher_activation_lock);
673 }
674 
675 static ssize_t bL_switcher_active_show(struct kobject *kobj,
676 		struct kobj_attribute *attr, char *buf)
677 {
678 	return sprintf(buf, "%u\n", bL_switcher_active);
679 }
680 
681 static ssize_t bL_switcher_active_store(struct kobject *kobj,
682 		struct kobj_attribute *attr, const char *buf, size_t count)
683 {
684 	int ret;
685 
686 	switch (buf[0]) {
687 	case '0':
688 		bL_switcher_disable();
689 		ret = 0;
690 		break;
691 	case '1':
692 		ret = bL_switcher_enable();
693 		break;
694 	default:
695 		ret = -EINVAL;
696 	}
697 
698 	return (ret >= 0) ? count : ret;
699 }
700 
701 static ssize_t bL_switcher_trace_trigger_store(struct kobject *kobj,
702 		struct kobj_attribute *attr, const char *buf, size_t count)
703 {
704 	int ret = bL_switcher_trace_trigger();
705 
706 	return ret ? ret : count;
707 }
708 
709 static struct kobj_attribute bL_switcher_active_attr =
710 	__ATTR(active, 0644, bL_switcher_active_show, bL_switcher_active_store);
711 
712 static struct kobj_attribute bL_switcher_trace_trigger_attr =
713 	__ATTR(trace_trigger, 0200, NULL, bL_switcher_trace_trigger_store);
714 
715 static struct attribute *bL_switcher_attrs[] = {
716 	&bL_switcher_active_attr.attr,
717 	&bL_switcher_trace_trigger_attr.attr,
718 	NULL,
719 };
720 
721 static struct attribute_group bL_switcher_attr_group = {
722 	.attrs = bL_switcher_attrs,
723 };
724 
725 static struct kobject *bL_switcher_kobj;
726 
727 static int __init bL_switcher_sysfs_init(void)
728 {
729 	int ret;
730 
731 	bL_switcher_kobj = kobject_create_and_add("bL_switcher", kernel_kobj);
732 	if (!bL_switcher_kobj)
733 		return -ENOMEM;
734 	ret = sysfs_create_group(bL_switcher_kobj, &bL_switcher_attr_group);
735 	if (ret)
736 		kobject_put(bL_switcher_kobj);
737 	return ret;
738 }
739 
740 #endif  /* CONFIG_SYSFS */
741 
742 bool bL_switcher_get_enabled(void)
743 {
744 	mutex_lock(&bL_switcher_activation_lock);
745 
746 	return bL_switcher_active;
747 }
748 EXPORT_SYMBOL_GPL(bL_switcher_get_enabled);
749 
750 void bL_switcher_put_enabled(void)
751 {
752 	mutex_unlock(&bL_switcher_activation_lock);
753 }
754 EXPORT_SYMBOL_GPL(bL_switcher_put_enabled);
755 
756 /*
757  * Veto any CPU hotplug operation on those CPUs we've removed
758  * while the switcher is active.
759  * We're just not ready to deal with that given the trickery involved.
760  */
761 static int bL_switcher_cpu_pre(unsigned int cpu)
762 {
763 	int pairing;
764 
765 	if (!bL_switcher_active)
766 		return 0;
767 
768 	pairing = bL_switcher_cpu_pairing[cpu];
769 
770 	if (pairing == -1)
771 		return -EINVAL;
772 	return 0;
773 }
774 
775 static bool no_bL_switcher;
776 core_param(no_bL_switcher, no_bL_switcher, bool, 0644);
777 
778 static int __init bL_switcher_init(void)
779 {
780 	int ret;
781 
782 	if (!mcpm_is_available())
783 		return -ENODEV;
784 
785 	cpuhp_setup_state_nocalls(CPUHP_ARM_BL_PREPARE, "arm/bl:prepare",
786 				  bL_switcher_cpu_pre, NULL);
787 	ret = cpuhp_setup_state_nocalls(CPUHP_AP_ONLINE_DYN, "arm/bl:predown",
788 					NULL, bL_switcher_cpu_pre);
789 	if (ret < 0) {
790 		cpuhp_remove_state_nocalls(CPUHP_ARM_BL_PREPARE);
791 		pr_err("bL_switcher: Failed to allocate a hotplug state\n");
792 		return ret;
793 	}
794 	if (!no_bL_switcher) {
795 		ret = bL_switcher_enable();
796 		if (ret)
797 			return ret;
798 	}
799 
800 #ifdef CONFIG_SYSFS
801 	ret = bL_switcher_sysfs_init();
802 	if (ret)
803 		pr_err("%s: unable to create sysfs entry\n", __func__);
804 #endif
805 
806 	return 0;
807 }
808 
809 late_initcall(bL_switcher_init);
810