162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * arch/arm/common/bL_switcher.c -- big.LITTLE cluster switcher core driver
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Created by:	Nicolas Pitre, March 2012
662306a36Sopenharmony_ci * Copyright:	(C) 2012-2013  Linaro Limited
762306a36Sopenharmony_ci */
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci#include <linux/atomic.h>
1062306a36Sopenharmony_ci#include <linux/init.h>
1162306a36Sopenharmony_ci#include <linux/kernel.h>
1262306a36Sopenharmony_ci#include <linux/module.h>
1362306a36Sopenharmony_ci#include <linux/sched/signal.h>
1462306a36Sopenharmony_ci#include <uapi/linux/sched/types.h>
1562306a36Sopenharmony_ci#include <linux/interrupt.h>
1662306a36Sopenharmony_ci#include <linux/cpu_pm.h>
1762306a36Sopenharmony_ci#include <linux/cpu.h>
1862306a36Sopenharmony_ci#include <linux/cpumask.h>
1962306a36Sopenharmony_ci#include <linux/kthread.h>
2062306a36Sopenharmony_ci#include <linux/wait.h>
2162306a36Sopenharmony_ci#include <linux/time.h>
2262306a36Sopenharmony_ci#include <linux/clockchips.h>
2362306a36Sopenharmony_ci#include <linux/hrtimer.h>
2462306a36Sopenharmony_ci#include <linux/tick.h>
2562306a36Sopenharmony_ci#include <linux/notifier.h>
2662306a36Sopenharmony_ci#include <linux/mm.h>
2762306a36Sopenharmony_ci#include <linux/mutex.h>
2862306a36Sopenharmony_ci#include <linux/smp.h>
2962306a36Sopenharmony_ci#include <linux/spinlock.h>
3062306a36Sopenharmony_ci#include <linux/string.h>
3162306a36Sopenharmony_ci#include <linux/sysfs.h>
3262306a36Sopenharmony_ci#include <linux/irqchip/arm-gic.h>
3362306a36Sopenharmony_ci#include <linux/moduleparam.h>
3462306a36Sopenharmony_ci
3562306a36Sopenharmony_ci#include <asm/smp_plat.h>
3662306a36Sopenharmony_ci#include <asm/cputype.h>
3762306a36Sopenharmony_ci#include <asm/suspend.h>
3862306a36Sopenharmony_ci#include <asm/mcpm.h>
3962306a36Sopenharmony_ci#include <asm/bL_switcher.h>
4062306a36Sopenharmony_ci
4162306a36Sopenharmony_ci#define CREATE_TRACE_POINTS
4262306a36Sopenharmony_ci#include <trace/events/power_cpu_migrate.h>
4362306a36Sopenharmony_ci
4462306a36Sopenharmony_ci
4562306a36Sopenharmony_ci/*
4662306a36Sopenharmony_ci * Use our own MPIDR accessors as the generic ones in asm/cputype.h have
4762306a36Sopenharmony_ci * __attribute_const__ and we don't want the compiler to assume any
4862306a36Sopenharmony_ci * constness here as the value _does_ change along some code paths.
4962306a36Sopenharmony_ci */
5062306a36Sopenharmony_ci
5162306a36Sopenharmony_cistatic int read_mpidr(void)
5262306a36Sopenharmony_ci{
5362306a36Sopenharmony_ci	unsigned int id;
5462306a36Sopenharmony_ci	asm volatile ("mrc p15, 0, %0, c0, c0, 5" : "=r" (id));
5562306a36Sopenharmony_ci	return id & MPIDR_HWID_BITMASK;
5662306a36Sopenharmony_ci}
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_ci/*
5962306a36Sopenharmony_ci * bL switcher core code.
6062306a36Sopenharmony_ci */
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_cistatic void bL_do_switch(void *_arg)
6362306a36Sopenharmony_ci{
6462306a36Sopenharmony_ci	unsigned ib_mpidr, ib_cpu, ib_cluster;
6562306a36Sopenharmony_ci	long volatile handshake, **handshake_ptr = _arg;
6662306a36Sopenharmony_ci
6762306a36Sopenharmony_ci	pr_debug("%s\n", __func__);
6862306a36Sopenharmony_ci
6962306a36Sopenharmony_ci	ib_mpidr = cpu_logical_map(smp_processor_id());
7062306a36Sopenharmony_ci	ib_cpu = MPIDR_AFFINITY_LEVEL(ib_mpidr, 0);
7162306a36Sopenharmony_ci	ib_cluster = MPIDR_AFFINITY_LEVEL(ib_mpidr, 1);
7262306a36Sopenharmony_ci
7362306a36Sopenharmony_ci	/* Advertise our handshake location */
7462306a36Sopenharmony_ci	if (handshake_ptr) {
7562306a36Sopenharmony_ci		handshake = 0;
7662306a36Sopenharmony_ci		*handshake_ptr = &handshake;
7762306a36Sopenharmony_ci	} else
7862306a36Sopenharmony_ci		handshake = -1;
7962306a36Sopenharmony_ci
8062306a36Sopenharmony_ci	/*
8162306a36Sopenharmony_ci	 * Our state has been saved at this point.  Let's release our
8262306a36Sopenharmony_ci	 * inbound CPU.
8362306a36Sopenharmony_ci	 */
8462306a36Sopenharmony_ci	mcpm_set_entry_vector(ib_cpu, ib_cluster, cpu_resume);
8562306a36Sopenharmony_ci	sev();
8662306a36Sopenharmony_ci
8762306a36Sopenharmony_ci	/*
8862306a36Sopenharmony_ci	 * From this point, we must assume that our counterpart CPU might
8962306a36Sopenharmony_ci	 * have taken over in its parallel world already, as if execution
9062306a36Sopenharmony_ci	 * just returned from cpu_suspend().  It is therefore important to
9162306a36Sopenharmony_ci	 * be very careful not to make any change the other guy is not
9262306a36Sopenharmony_ci	 * expecting.  This is why we need stack isolation.
9362306a36Sopenharmony_ci	 *
9462306a36Sopenharmony_ci	 * Fancy under cover tasks could be performed here.  For now
9562306a36Sopenharmony_ci	 * we have none.
9662306a36Sopenharmony_ci	 */
9762306a36Sopenharmony_ci
9862306a36Sopenharmony_ci	/*
9962306a36Sopenharmony_ci	 * Let's wait until our inbound is alive.
10062306a36Sopenharmony_ci	 */
10162306a36Sopenharmony_ci	while (!handshake) {
10262306a36Sopenharmony_ci		wfe();
10362306a36Sopenharmony_ci		smp_mb();
10462306a36Sopenharmony_ci	}
10562306a36Sopenharmony_ci
10662306a36Sopenharmony_ci	/* Let's put ourself down. */
10762306a36Sopenharmony_ci	mcpm_cpu_power_down();
10862306a36Sopenharmony_ci
10962306a36Sopenharmony_ci	/* should never get here */
11062306a36Sopenharmony_ci	BUG();
11162306a36Sopenharmony_ci}
11262306a36Sopenharmony_ci
11362306a36Sopenharmony_ci/*
11462306a36Sopenharmony_ci * Stack isolation.  To ensure 'current' remains valid, we just use another
11562306a36Sopenharmony_ci * piece of our thread's stack space which should be fairly lightly used.
11662306a36Sopenharmony_ci * The selected area starts just above the thread_info structure located
11762306a36Sopenharmony_ci * at the very bottom of the stack, aligned to a cache line, and indexed
11862306a36Sopenharmony_ci * with the cluster number.
11962306a36Sopenharmony_ci */
12062306a36Sopenharmony_ci#define STACK_SIZE 512
12162306a36Sopenharmony_ciextern void call_with_stack(void (*fn)(void *), void *arg, void *sp);
12262306a36Sopenharmony_cistatic int bL_switchpoint(unsigned long _arg)
12362306a36Sopenharmony_ci{
12462306a36Sopenharmony_ci	unsigned int mpidr = read_mpidr();
12562306a36Sopenharmony_ci	unsigned int clusterid = MPIDR_AFFINITY_LEVEL(mpidr, 1);
12662306a36Sopenharmony_ci	void *stack = current_thread_info() + 1;
12762306a36Sopenharmony_ci	stack = PTR_ALIGN(stack, L1_CACHE_BYTES);
12862306a36Sopenharmony_ci	stack += clusterid * STACK_SIZE + STACK_SIZE;
12962306a36Sopenharmony_ci	call_with_stack(bL_do_switch, (void *)_arg, stack);
13062306a36Sopenharmony_ci	BUG();
13162306a36Sopenharmony_ci}
13262306a36Sopenharmony_ci
13362306a36Sopenharmony_ci/*
13462306a36Sopenharmony_ci * Generic switcher interface
13562306a36Sopenharmony_ci */
13662306a36Sopenharmony_ci
13762306a36Sopenharmony_cistatic unsigned int bL_gic_id[MAX_CPUS_PER_CLUSTER][MAX_NR_CLUSTERS];
13862306a36Sopenharmony_cistatic int bL_switcher_cpu_pairing[NR_CPUS];
13962306a36Sopenharmony_ci
14062306a36Sopenharmony_ci/*
14162306a36Sopenharmony_ci * bL_switch_to - Switch to a specific cluster for the current CPU
14262306a36Sopenharmony_ci * @new_cluster_id: the ID of the cluster to switch to.
14362306a36Sopenharmony_ci *
14462306a36Sopenharmony_ci * This function must be called on the CPU to be switched.
14562306a36Sopenharmony_ci * Returns 0 on success, else a negative status code.
14662306a36Sopenharmony_ci */
14762306a36Sopenharmony_cistatic int bL_switch_to(unsigned int new_cluster_id)
14862306a36Sopenharmony_ci{
14962306a36Sopenharmony_ci	unsigned int mpidr, this_cpu, that_cpu;
15062306a36Sopenharmony_ci	unsigned int ob_mpidr, ob_cpu, ob_cluster, ib_mpidr, ib_cpu, ib_cluster;
15162306a36Sopenharmony_ci	struct completion inbound_alive;
15262306a36Sopenharmony_ci	long volatile *handshake_ptr;
15362306a36Sopenharmony_ci	int ipi_nr, ret;
15462306a36Sopenharmony_ci
15562306a36Sopenharmony_ci	this_cpu = smp_processor_id();
15662306a36Sopenharmony_ci	ob_mpidr = read_mpidr();
15762306a36Sopenharmony_ci	ob_cpu = MPIDR_AFFINITY_LEVEL(ob_mpidr, 0);
15862306a36Sopenharmony_ci	ob_cluster = MPIDR_AFFINITY_LEVEL(ob_mpidr, 1);
15962306a36Sopenharmony_ci	BUG_ON(cpu_logical_map(this_cpu) != ob_mpidr);
16062306a36Sopenharmony_ci
16162306a36Sopenharmony_ci	if (new_cluster_id == ob_cluster)
16262306a36Sopenharmony_ci		return 0;
16362306a36Sopenharmony_ci
16462306a36Sopenharmony_ci	that_cpu = bL_switcher_cpu_pairing[this_cpu];
16562306a36Sopenharmony_ci	ib_mpidr = cpu_logical_map(that_cpu);
16662306a36Sopenharmony_ci	ib_cpu = MPIDR_AFFINITY_LEVEL(ib_mpidr, 0);
16762306a36Sopenharmony_ci	ib_cluster = MPIDR_AFFINITY_LEVEL(ib_mpidr, 1);
16862306a36Sopenharmony_ci
16962306a36Sopenharmony_ci	pr_debug("before switch: CPU %d MPIDR %#x -> %#x\n",
17062306a36Sopenharmony_ci		 this_cpu, ob_mpidr, ib_mpidr);
17162306a36Sopenharmony_ci
17262306a36Sopenharmony_ci	this_cpu = smp_processor_id();
17362306a36Sopenharmony_ci
17462306a36Sopenharmony_ci	/* Close the gate for our entry vectors */
17562306a36Sopenharmony_ci	mcpm_set_entry_vector(ob_cpu, ob_cluster, NULL);
17662306a36Sopenharmony_ci	mcpm_set_entry_vector(ib_cpu, ib_cluster, NULL);
17762306a36Sopenharmony_ci
17862306a36Sopenharmony_ci	/* Install our "inbound alive" notifier. */
17962306a36Sopenharmony_ci	init_completion(&inbound_alive);
18062306a36Sopenharmony_ci	ipi_nr = register_ipi_completion(&inbound_alive, this_cpu);
18162306a36Sopenharmony_ci	ipi_nr |= ((1 << 16) << bL_gic_id[ob_cpu][ob_cluster]);
18262306a36Sopenharmony_ci	mcpm_set_early_poke(ib_cpu, ib_cluster, gic_get_sgir_physaddr(), ipi_nr);
18362306a36Sopenharmony_ci
18462306a36Sopenharmony_ci	/*
18562306a36Sopenharmony_ci	 * Let's wake up the inbound CPU now in case it requires some delay
18662306a36Sopenharmony_ci	 * to come online, but leave it gated in our entry vector code.
18762306a36Sopenharmony_ci	 */
18862306a36Sopenharmony_ci	ret = mcpm_cpu_power_up(ib_cpu, ib_cluster);
18962306a36Sopenharmony_ci	if (ret) {
19062306a36Sopenharmony_ci		pr_err("%s: mcpm_cpu_power_up() returned %d\n", __func__, ret);
19162306a36Sopenharmony_ci		return ret;
19262306a36Sopenharmony_ci	}
19362306a36Sopenharmony_ci
19462306a36Sopenharmony_ci	/*
19562306a36Sopenharmony_ci	 * Raise a SGI on the inbound CPU to make sure it doesn't stall
19662306a36Sopenharmony_ci	 * in a possible WFI, such as in bL_power_down().
19762306a36Sopenharmony_ci	 */
19862306a36Sopenharmony_ci	gic_send_sgi(bL_gic_id[ib_cpu][ib_cluster], 0);
19962306a36Sopenharmony_ci
20062306a36Sopenharmony_ci	/*
20162306a36Sopenharmony_ci	 * Wait for the inbound to come up.  This allows for other
20262306a36Sopenharmony_ci	 * tasks to be scheduled in the mean time.
20362306a36Sopenharmony_ci	 */
20462306a36Sopenharmony_ci	wait_for_completion(&inbound_alive);
20562306a36Sopenharmony_ci	mcpm_set_early_poke(ib_cpu, ib_cluster, 0, 0);
20662306a36Sopenharmony_ci
20762306a36Sopenharmony_ci	/*
20862306a36Sopenharmony_ci	 * From this point we are entering the switch critical zone
20962306a36Sopenharmony_ci	 * and can't take any interrupts anymore.
21062306a36Sopenharmony_ci	 */
21162306a36Sopenharmony_ci	local_irq_disable();
21262306a36Sopenharmony_ci	local_fiq_disable();
21362306a36Sopenharmony_ci	trace_cpu_migrate_begin(ktime_get_real_ns(), ob_mpidr);
21462306a36Sopenharmony_ci
21562306a36Sopenharmony_ci	/* redirect GIC's SGIs to our counterpart */
21662306a36Sopenharmony_ci	gic_migrate_target(bL_gic_id[ib_cpu][ib_cluster]);
21762306a36Sopenharmony_ci
21862306a36Sopenharmony_ci	tick_suspend_local();
21962306a36Sopenharmony_ci
22062306a36Sopenharmony_ci	ret = cpu_pm_enter();
22162306a36Sopenharmony_ci
22262306a36Sopenharmony_ci	/* we can not tolerate errors at this point */
22362306a36Sopenharmony_ci	if (ret)
22462306a36Sopenharmony_ci		panic("%s: cpu_pm_enter() returned %d\n", __func__, ret);
22562306a36Sopenharmony_ci
22662306a36Sopenharmony_ci	/* Swap the physical CPUs in the logical map for this logical CPU. */
22762306a36Sopenharmony_ci	cpu_logical_map(this_cpu) = ib_mpidr;
22862306a36Sopenharmony_ci	cpu_logical_map(that_cpu) = ob_mpidr;
22962306a36Sopenharmony_ci
23062306a36Sopenharmony_ci	/* Let's do the actual CPU switch. */
23162306a36Sopenharmony_ci	ret = cpu_suspend((unsigned long)&handshake_ptr, bL_switchpoint);
23262306a36Sopenharmony_ci	if (ret > 0)
23362306a36Sopenharmony_ci		panic("%s: cpu_suspend() returned %d\n", __func__, ret);
23462306a36Sopenharmony_ci
23562306a36Sopenharmony_ci	/* We are executing on the inbound CPU at this point */
23662306a36Sopenharmony_ci	mpidr = read_mpidr();
23762306a36Sopenharmony_ci	pr_debug("after switch: CPU %d MPIDR %#x\n", this_cpu, mpidr);
23862306a36Sopenharmony_ci	BUG_ON(mpidr != ib_mpidr);
23962306a36Sopenharmony_ci
24062306a36Sopenharmony_ci	mcpm_cpu_powered_up();
24162306a36Sopenharmony_ci
24262306a36Sopenharmony_ci	ret = cpu_pm_exit();
24362306a36Sopenharmony_ci
24462306a36Sopenharmony_ci	tick_resume_local();
24562306a36Sopenharmony_ci
24662306a36Sopenharmony_ci	trace_cpu_migrate_finish(ktime_get_real_ns(), ib_mpidr);
24762306a36Sopenharmony_ci	local_fiq_enable();
24862306a36Sopenharmony_ci	local_irq_enable();
24962306a36Sopenharmony_ci
25062306a36Sopenharmony_ci	*handshake_ptr = 1;
25162306a36Sopenharmony_ci	dsb_sev();
25262306a36Sopenharmony_ci
25362306a36Sopenharmony_ci	if (ret)
25462306a36Sopenharmony_ci		pr_err("%s exiting with error %d\n", __func__, ret);
25562306a36Sopenharmony_ci	return ret;
25662306a36Sopenharmony_ci}
25762306a36Sopenharmony_ci
25862306a36Sopenharmony_cistruct bL_thread {
25962306a36Sopenharmony_ci	spinlock_t lock;
26062306a36Sopenharmony_ci	struct task_struct *task;
26162306a36Sopenharmony_ci	wait_queue_head_t wq;
26262306a36Sopenharmony_ci	int wanted_cluster;
26362306a36Sopenharmony_ci	struct completion started;
26462306a36Sopenharmony_ci	bL_switch_completion_handler completer;
26562306a36Sopenharmony_ci	void *completer_cookie;
26662306a36Sopenharmony_ci};
26762306a36Sopenharmony_ci
26862306a36Sopenharmony_cistatic struct bL_thread bL_threads[NR_CPUS];
26962306a36Sopenharmony_ci
27062306a36Sopenharmony_cistatic int bL_switcher_thread(void *arg)
27162306a36Sopenharmony_ci{
27262306a36Sopenharmony_ci	struct bL_thread *t = arg;
27362306a36Sopenharmony_ci	int cluster;
27462306a36Sopenharmony_ci	bL_switch_completion_handler completer;
27562306a36Sopenharmony_ci	void *completer_cookie;
27662306a36Sopenharmony_ci
27762306a36Sopenharmony_ci	sched_set_fifo_low(current);
27862306a36Sopenharmony_ci	complete(&t->started);
27962306a36Sopenharmony_ci
28062306a36Sopenharmony_ci	do {
28162306a36Sopenharmony_ci		if (signal_pending(current))
28262306a36Sopenharmony_ci			flush_signals(current);
28362306a36Sopenharmony_ci		wait_event_interruptible(t->wq,
28462306a36Sopenharmony_ci				t->wanted_cluster != -1 ||
28562306a36Sopenharmony_ci				kthread_should_stop());
28662306a36Sopenharmony_ci
28762306a36Sopenharmony_ci		spin_lock(&t->lock);
28862306a36Sopenharmony_ci		cluster = t->wanted_cluster;
28962306a36Sopenharmony_ci		completer = t->completer;
29062306a36Sopenharmony_ci		completer_cookie = t->completer_cookie;
29162306a36Sopenharmony_ci		t->wanted_cluster = -1;
29262306a36Sopenharmony_ci		t->completer = NULL;
29362306a36Sopenharmony_ci		spin_unlock(&t->lock);
29462306a36Sopenharmony_ci
29562306a36Sopenharmony_ci		if (cluster != -1) {
29662306a36Sopenharmony_ci			bL_switch_to(cluster);
29762306a36Sopenharmony_ci
29862306a36Sopenharmony_ci			if (completer)
29962306a36Sopenharmony_ci				completer(completer_cookie);
30062306a36Sopenharmony_ci		}
30162306a36Sopenharmony_ci	} while (!kthread_should_stop());
30262306a36Sopenharmony_ci
30362306a36Sopenharmony_ci	return 0;
30462306a36Sopenharmony_ci}
30562306a36Sopenharmony_ci
30662306a36Sopenharmony_cistatic struct task_struct *bL_switcher_thread_create(int cpu, void *arg)
30762306a36Sopenharmony_ci{
30862306a36Sopenharmony_ci	struct task_struct *task;
30962306a36Sopenharmony_ci
31062306a36Sopenharmony_ci	task = kthread_create_on_node(bL_switcher_thread, arg,
31162306a36Sopenharmony_ci				      cpu_to_node(cpu), "kswitcher_%d", cpu);
31262306a36Sopenharmony_ci	if (!IS_ERR(task)) {
31362306a36Sopenharmony_ci		kthread_bind(task, cpu);
31462306a36Sopenharmony_ci		wake_up_process(task);
31562306a36Sopenharmony_ci	} else
31662306a36Sopenharmony_ci		pr_err("%s failed for CPU %d\n", __func__, cpu);
31762306a36Sopenharmony_ci	return task;
31862306a36Sopenharmony_ci}
31962306a36Sopenharmony_ci
32062306a36Sopenharmony_ci/*
32162306a36Sopenharmony_ci * bL_switch_request_cb - Switch to a specific cluster for the given CPU,
32262306a36Sopenharmony_ci *      with completion notification via a callback
32362306a36Sopenharmony_ci *
32462306a36Sopenharmony_ci * @cpu: the CPU to switch
32562306a36Sopenharmony_ci * @new_cluster_id: the ID of the cluster to switch to.
32662306a36Sopenharmony_ci * @completer: switch completion callback.  if non-NULL,
32762306a36Sopenharmony_ci *	@completer(@completer_cookie) will be called on completion of
32862306a36Sopenharmony_ci *	the switch, in non-atomic context.
32962306a36Sopenharmony_ci * @completer_cookie: opaque context argument for @completer.
33062306a36Sopenharmony_ci *
33162306a36Sopenharmony_ci * This function causes a cluster switch on the given CPU by waking up
33262306a36Sopenharmony_ci * the appropriate switcher thread.  This function may or may not return
33362306a36Sopenharmony_ci * before the switch has occurred.
33462306a36Sopenharmony_ci *
33562306a36Sopenharmony_ci * If a @completer callback function is supplied, it will be called when
33662306a36Sopenharmony_ci * the switch is complete.  This can be used to determine asynchronously
33762306a36Sopenharmony_ci * when the switch is complete, regardless of when bL_switch_request()
33862306a36Sopenharmony_ci * returns.  When @completer is supplied, no new switch request is permitted
33962306a36Sopenharmony_ci * for the affected CPU until after the switch is complete, and @completer
34062306a36Sopenharmony_ci * has returned.
34162306a36Sopenharmony_ci */
34262306a36Sopenharmony_ciint bL_switch_request_cb(unsigned int cpu, unsigned int new_cluster_id,
34362306a36Sopenharmony_ci			 bL_switch_completion_handler completer,
34462306a36Sopenharmony_ci			 void *completer_cookie)
34562306a36Sopenharmony_ci{
34662306a36Sopenharmony_ci	struct bL_thread *t;
34762306a36Sopenharmony_ci
34862306a36Sopenharmony_ci	if (cpu >= ARRAY_SIZE(bL_threads)) {
34962306a36Sopenharmony_ci		pr_err("%s: cpu %d out of bounds\n", __func__, cpu);
35062306a36Sopenharmony_ci		return -EINVAL;
35162306a36Sopenharmony_ci	}
35262306a36Sopenharmony_ci
35362306a36Sopenharmony_ci	t = &bL_threads[cpu];
35462306a36Sopenharmony_ci
35562306a36Sopenharmony_ci	if (IS_ERR(t->task))
35662306a36Sopenharmony_ci		return PTR_ERR(t->task);
35762306a36Sopenharmony_ci	if (!t->task)
35862306a36Sopenharmony_ci		return -ESRCH;
35962306a36Sopenharmony_ci
36062306a36Sopenharmony_ci	spin_lock(&t->lock);
36162306a36Sopenharmony_ci	if (t->completer) {
36262306a36Sopenharmony_ci		spin_unlock(&t->lock);
36362306a36Sopenharmony_ci		return -EBUSY;
36462306a36Sopenharmony_ci	}
36562306a36Sopenharmony_ci	t->completer = completer;
36662306a36Sopenharmony_ci	t->completer_cookie = completer_cookie;
36762306a36Sopenharmony_ci	t->wanted_cluster = new_cluster_id;
36862306a36Sopenharmony_ci	spin_unlock(&t->lock);
36962306a36Sopenharmony_ci	wake_up(&t->wq);
37062306a36Sopenharmony_ci	return 0;
37162306a36Sopenharmony_ci}
37262306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(bL_switch_request_cb);
37362306a36Sopenharmony_ci
37462306a36Sopenharmony_ci/*
37562306a36Sopenharmony_ci * Activation and configuration code.
37662306a36Sopenharmony_ci */
37762306a36Sopenharmony_ci
37862306a36Sopenharmony_cistatic DEFINE_MUTEX(bL_switcher_activation_lock);
37962306a36Sopenharmony_cistatic BLOCKING_NOTIFIER_HEAD(bL_activation_notifier);
38062306a36Sopenharmony_cistatic unsigned int bL_switcher_active;
38162306a36Sopenharmony_cistatic unsigned int bL_switcher_cpu_original_cluster[NR_CPUS];
38262306a36Sopenharmony_cistatic cpumask_t bL_switcher_removed_logical_cpus;
38362306a36Sopenharmony_ci
38462306a36Sopenharmony_ciint bL_switcher_register_notifier(struct notifier_block *nb)
38562306a36Sopenharmony_ci{
38662306a36Sopenharmony_ci	return blocking_notifier_chain_register(&bL_activation_notifier, nb);
38762306a36Sopenharmony_ci}
38862306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(bL_switcher_register_notifier);
38962306a36Sopenharmony_ci
39062306a36Sopenharmony_ciint bL_switcher_unregister_notifier(struct notifier_block *nb)
39162306a36Sopenharmony_ci{
39262306a36Sopenharmony_ci	return blocking_notifier_chain_unregister(&bL_activation_notifier, nb);
39362306a36Sopenharmony_ci}
39462306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(bL_switcher_unregister_notifier);
39562306a36Sopenharmony_ci
39662306a36Sopenharmony_cistatic int bL_activation_notify(unsigned long val)
39762306a36Sopenharmony_ci{
39862306a36Sopenharmony_ci	int ret;
39962306a36Sopenharmony_ci
40062306a36Sopenharmony_ci	ret = blocking_notifier_call_chain(&bL_activation_notifier, val, NULL);
40162306a36Sopenharmony_ci	if (ret & NOTIFY_STOP_MASK)
40262306a36Sopenharmony_ci		pr_err("%s: notifier chain failed with status 0x%x\n",
40362306a36Sopenharmony_ci			__func__, ret);
40462306a36Sopenharmony_ci	return notifier_to_errno(ret);
40562306a36Sopenharmony_ci}
40662306a36Sopenharmony_ci
40762306a36Sopenharmony_cistatic void bL_switcher_restore_cpus(void)
40862306a36Sopenharmony_ci{
40962306a36Sopenharmony_ci	int i;
41062306a36Sopenharmony_ci
41162306a36Sopenharmony_ci	for_each_cpu(i, &bL_switcher_removed_logical_cpus) {
41262306a36Sopenharmony_ci		struct device *cpu_dev = get_cpu_device(i);
41362306a36Sopenharmony_ci		int ret = device_online(cpu_dev);
41462306a36Sopenharmony_ci		if (ret)
41562306a36Sopenharmony_ci			dev_err(cpu_dev, "switcher: unable to restore CPU\n");
41662306a36Sopenharmony_ci	}
41762306a36Sopenharmony_ci}
41862306a36Sopenharmony_ci
41962306a36Sopenharmony_cistatic int bL_switcher_halve_cpus(void)
42062306a36Sopenharmony_ci{
42162306a36Sopenharmony_ci	int i, j, cluster_0, gic_id, ret;
42262306a36Sopenharmony_ci	unsigned int cpu, cluster, mask;
42362306a36Sopenharmony_ci	cpumask_t available_cpus;
42462306a36Sopenharmony_ci
42562306a36Sopenharmony_ci	/* First pass to validate what we have */
42662306a36Sopenharmony_ci	mask = 0;
42762306a36Sopenharmony_ci	for_each_online_cpu(i) {
42862306a36Sopenharmony_ci		cpu = MPIDR_AFFINITY_LEVEL(cpu_logical_map(i), 0);
42962306a36Sopenharmony_ci		cluster = MPIDR_AFFINITY_LEVEL(cpu_logical_map(i), 1);
43062306a36Sopenharmony_ci		if (cluster >= 2) {
43162306a36Sopenharmony_ci			pr_err("%s: only dual cluster systems are supported\n", __func__);
43262306a36Sopenharmony_ci			return -EINVAL;
43362306a36Sopenharmony_ci		}
43462306a36Sopenharmony_ci		if (WARN_ON(cpu >= MAX_CPUS_PER_CLUSTER))
43562306a36Sopenharmony_ci			return -EINVAL;
43662306a36Sopenharmony_ci		mask |= (1 << cluster);
43762306a36Sopenharmony_ci	}
43862306a36Sopenharmony_ci	if (mask != 3) {
43962306a36Sopenharmony_ci		pr_err("%s: no CPU pairing possible\n", __func__);
44062306a36Sopenharmony_ci		return -EINVAL;
44162306a36Sopenharmony_ci	}
44262306a36Sopenharmony_ci
44362306a36Sopenharmony_ci	/*
44462306a36Sopenharmony_ci	 * Now let's do the pairing.  We match each CPU with another CPU
44562306a36Sopenharmony_ci	 * from a different cluster.  To get a uniform scheduling behavior
44662306a36Sopenharmony_ci	 * without fiddling with CPU topology and compute capacity data,
44762306a36Sopenharmony_ci	 * we'll use logical CPUs initially belonging to the same cluster.
44862306a36Sopenharmony_ci	 */
44962306a36Sopenharmony_ci	memset(bL_switcher_cpu_pairing, -1, sizeof(bL_switcher_cpu_pairing));
45062306a36Sopenharmony_ci	cpumask_copy(&available_cpus, cpu_online_mask);
45162306a36Sopenharmony_ci	cluster_0 = -1;
45262306a36Sopenharmony_ci	for_each_cpu(i, &available_cpus) {
45362306a36Sopenharmony_ci		int match = -1;
45462306a36Sopenharmony_ci		cluster = MPIDR_AFFINITY_LEVEL(cpu_logical_map(i), 1);
45562306a36Sopenharmony_ci		if (cluster_0 == -1)
45662306a36Sopenharmony_ci			cluster_0 = cluster;
45762306a36Sopenharmony_ci		if (cluster != cluster_0)
45862306a36Sopenharmony_ci			continue;
45962306a36Sopenharmony_ci		cpumask_clear_cpu(i, &available_cpus);
46062306a36Sopenharmony_ci		for_each_cpu(j, &available_cpus) {
46162306a36Sopenharmony_ci			cluster = MPIDR_AFFINITY_LEVEL(cpu_logical_map(j), 1);
46262306a36Sopenharmony_ci			/*
46362306a36Sopenharmony_ci			 * Let's remember the last match to create "odd"
46462306a36Sopenharmony_ci			 * pairings on purpose in order for other code not
46562306a36Sopenharmony_ci			 * to assume any relation between physical and
46662306a36Sopenharmony_ci			 * logical CPU numbers.
46762306a36Sopenharmony_ci			 */
46862306a36Sopenharmony_ci			if (cluster != cluster_0)
46962306a36Sopenharmony_ci				match = j;
47062306a36Sopenharmony_ci		}
47162306a36Sopenharmony_ci		if (match != -1) {
47262306a36Sopenharmony_ci			bL_switcher_cpu_pairing[i] = match;
47362306a36Sopenharmony_ci			cpumask_clear_cpu(match, &available_cpus);
47462306a36Sopenharmony_ci			pr_info("CPU%d paired with CPU%d\n", i, match);
47562306a36Sopenharmony_ci		}
47662306a36Sopenharmony_ci	}
47762306a36Sopenharmony_ci
47862306a36Sopenharmony_ci	/*
47962306a36Sopenharmony_ci	 * Now we disable the unwanted CPUs i.e. everything that has no
48062306a36Sopenharmony_ci	 * pairing information (that includes the pairing counterparts).
48162306a36Sopenharmony_ci	 */
48262306a36Sopenharmony_ci	cpumask_clear(&bL_switcher_removed_logical_cpus);
48362306a36Sopenharmony_ci	for_each_online_cpu(i) {
48462306a36Sopenharmony_ci		cpu = MPIDR_AFFINITY_LEVEL(cpu_logical_map(i), 0);
48562306a36Sopenharmony_ci		cluster = MPIDR_AFFINITY_LEVEL(cpu_logical_map(i), 1);
48662306a36Sopenharmony_ci
48762306a36Sopenharmony_ci		/* Let's take note of the GIC ID for this CPU */
48862306a36Sopenharmony_ci		gic_id = gic_get_cpu_id(i);
48962306a36Sopenharmony_ci		if (gic_id < 0) {
49062306a36Sopenharmony_ci			pr_err("%s: bad GIC ID for CPU %d\n", __func__, i);
49162306a36Sopenharmony_ci			bL_switcher_restore_cpus();
49262306a36Sopenharmony_ci			return -EINVAL;
49362306a36Sopenharmony_ci		}
49462306a36Sopenharmony_ci		bL_gic_id[cpu][cluster] = gic_id;
49562306a36Sopenharmony_ci		pr_info("GIC ID for CPU %u cluster %u is %u\n",
49662306a36Sopenharmony_ci			cpu, cluster, gic_id);
49762306a36Sopenharmony_ci
49862306a36Sopenharmony_ci		if (bL_switcher_cpu_pairing[i] != -1) {
49962306a36Sopenharmony_ci			bL_switcher_cpu_original_cluster[i] = cluster;
50062306a36Sopenharmony_ci			continue;
50162306a36Sopenharmony_ci		}
50262306a36Sopenharmony_ci
50362306a36Sopenharmony_ci		ret = device_offline(get_cpu_device(i));
50462306a36Sopenharmony_ci		if (ret) {
50562306a36Sopenharmony_ci			bL_switcher_restore_cpus();
50662306a36Sopenharmony_ci			return ret;
50762306a36Sopenharmony_ci		}
50862306a36Sopenharmony_ci		cpumask_set_cpu(i, &bL_switcher_removed_logical_cpus);
50962306a36Sopenharmony_ci	}
51062306a36Sopenharmony_ci
51162306a36Sopenharmony_ci	return 0;
51262306a36Sopenharmony_ci}
51362306a36Sopenharmony_ci
51462306a36Sopenharmony_ci/* Determine the logical CPU a given physical CPU is grouped on. */
51562306a36Sopenharmony_ciint bL_switcher_get_logical_index(u32 mpidr)
51662306a36Sopenharmony_ci{
51762306a36Sopenharmony_ci	int cpu;
51862306a36Sopenharmony_ci
51962306a36Sopenharmony_ci	if (!bL_switcher_active)
52062306a36Sopenharmony_ci		return -EUNATCH;
52162306a36Sopenharmony_ci
52262306a36Sopenharmony_ci	mpidr &= MPIDR_HWID_BITMASK;
52362306a36Sopenharmony_ci	for_each_online_cpu(cpu) {
52462306a36Sopenharmony_ci		int pairing = bL_switcher_cpu_pairing[cpu];
52562306a36Sopenharmony_ci		if (pairing == -1)
52662306a36Sopenharmony_ci			continue;
52762306a36Sopenharmony_ci		if ((mpidr == cpu_logical_map(cpu)) ||
52862306a36Sopenharmony_ci		    (mpidr == cpu_logical_map(pairing)))
52962306a36Sopenharmony_ci			return cpu;
53062306a36Sopenharmony_ci	}
53162306a36Sopenharmony_ci	return -EINVAL;
53262306a36Sopenharmony_ci}
53362306a36Sopenharmony_ci
53462306a36Sopenharmony_cistatic void bL_switcher_trace_trigger_cpu(void *__always_unused info)
53562306a36Sopenharmony_ci{
53662306a36Sopenharmony_ci	trace_cpu_migrate_current(ktime_get_real_ns(), read_mpidr());
53762306a36Sopenharmony_ci}
53862306a36Sopenharmony_ci
53962306a36Sopenharmony_ciint bL_switcher_trace_trigger(void)
54062306a36Sopenharmony_ci{
54162306a36Sopenharmony_ci	preempt_disable();
54262306a36Sopenharmony_ci
54362306a36Sopenharmony_ci	bL_switcher_trace_trigger_cpu(NULL);
54462306a36Sopenharmony_ci	smp_call_function(bL_switcher_trace_trigger_cpu, NULL, true);
54562306a36Sopenharmony_ci
54662306a36Sopenharmony_ci	preempt_enable();
54762306a36Sopenharmony_ci
54862306a36Sopenharmony_ci	return 0;
54962306a36Sopenharmony_ci}
55062306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(bL_switcher_trace_trigger);
55162306a36Sopenharmony_ci
55262306a36Sopenharmony_cistatic int bL_switcher_enable(void)
55362306a36Sopenharmony_ci{
55462306a36Sopenharmony_ci	int cpu, ret;
55562306a36Sopenharmony_ci
55662306a36Sopenharmony_ci	mutex_lock(&bL_switcher_activation_lock);
55762306a36Sopenharmony_ci	lock_device_hotplug();
55862306a36Sopenharmony_ci	if (bL_switcher_active) {
55962306a36Sopenharmony_ci		unlock_device_hotplug();
56062306a36Sopenharmony_ci		mutex_unlock(&bL_switcher_activation_lock);
56162306a36Sopenharmony_ci		return 0;
56262306a36Sopenharmony_ci	}
56362306a36Sopenharmony_ci
56462306a36Sopenharmony_ci	pr_info("big.LITTLE switcher initializing\n");
56562306a36Sopenharmony_ci
56662306a36Sopenharmony_ci	ret = bL_activation_notify(BL_NOTIFY_PRE_ENABLE);
56762306a36Sopenharmony_ci	if (ret)
56862306a36Sopenharmony_ci		goto error;
56962306a36Sopenharmony_ci
57062306a36Sopenharmony_ci	ret = bL_switcher_halve_cpus();
57162306a36Sopenharmony_ci	if (ret)
57262306a36Sopenharmony_ci		goto error;
57362306a36Sopenharmony_ci
57462306a36Sopenharmony_ci	bL_switcher_trace_trigger();
57562306a36Sopenharmony_ci
57662306a36Sopenharmony_ci	for_each_online_cpu(cpu) {
57762306a36Sopenharmony_ci		struct bL_thread *t = &bL_threads[cpu];
57862306a36Sopenharmony_ci		spin_lock_init(&t->lock);
57962306a36Sopenharmony_ci		init_waitqueue_head(&t->wq);
58062306a36Sopenharmony_ci		init_completion(&t->started);
58162306a36Sopenharmony_ci		t->wanted_cluster = -1;
58262306a36Sopenharmony_ci		t->task = bL_switcher_thread_create(cpu, t);
58362306a36Sopenharmony_ci	}
58462306a36Sopenharmony_ci
58562306a36Sopenharmony_ci	bL_switcher_active = 1;
58662306a36Sopenharmony_ci	bL_activation_notify(BL_NOTIFY_POST_ENABLE);
58762306a36Sopenharmony_ci	pr_info("big.LITTLE switcher initialized\n");
58862306a36Sopenharmony_ci	goto out;
58962306a36Sopenharmony_ci
59062306a36Sopenharmony_cierror:
59162306a36Sopenharmony_ci	pr_warn("big.LITTLE switcher initialization failed\n");
59262306a36Sopenharmony_ci	bL_activation_notify(BL_NOTIFY_POST_DISABLE);
59362306a36Sopenharmony_ci
59462306a36Sopenharmony_ciout:
59562306a36Sopenharmony_ci	unlock_device_hotplug();
59662306a36Sopenharmony_ci	mutex_unlock(&bL_switcher_activation_lock);
59762306a36Sopenharmony_ci	return ret;
59862306a36Sopenharmony_ci}
59962306a36Sopenharmony_ci
60062306a36Sopenharmony_ci#ifdef CONFIG_SYSFS
60162306a36Sopenharmony_ci
60262306a36Sopenharmony_cistatic void bL_switcher_disable(void)
60362306a36Sopenharmony_ci{
60462306a36Sopenharmony_ci	unsigned int cpu, cluster;
60562306a36Sopenharmony_ci	struct bL_thread *t;
60662306a36Sopenharmony_ci	struct task_struct *task;
60762306a36Sopenharmony_ci
60862306a36Sopenharmony_ci	mutex_lock(&bL_switcher_activation_lock);
60962306a36Sopenharmony_ci	lock_device_hotplug();
61062306a36Sopenharmony_ci
61162306a36Sopenharmony_ci	if (!bL_switcher_active)
61262306a36Sopenharmony_ci		goto out;
61362306a36Sopenharmony_ci
61462306a36Sopenharmony_ci	if (bL_activation_notify(BL_NOTIFY_PRE_DISABLE) != 0) {
61562306a36Sopenharmony_ci		bL_activation_notify(BL_NOTIFY_POST_ENABLE);
61662306a36Sopenharmony_ci		goto out;
61762306a36Sopenharmony_ci	}
61862306a36Sopenharmony_ci
61962306a36Sopenharmony_ci	bL_switcher_active = 0;
62062306a36Sopenharmony_ci
62162306a36Sopenharmony_ci	/*
62262306a36Sopenharmony_ci	 * To deactivate the switcher, we must shut down the switcher
62362306a36Sopenharmony_ci	 * threads to prevent any other requests from being accepted.
62462306a36Sopenharmony_ci	 * Then, if the final cluster for given logical CPU is not the
62562306a36Sopenharmony_ci	 * same as the original one, we'll recreate a switcher thread
62662306a36Sopenharmony_ci	 * just for the purpose of switching the CPU back without any
62762306a36Sopenharmony_ci	 * possibility for interference from external requests.
62862306a36Sopenharmony_ci	 */
62962306a36Sopenharmony_ci	for_each_online_cpu(cpu) {
63062306a36Sopenharmony_ci		t = &bL_threads[cpu];
63162306a36Sopenharmony_ci		task = t->task;
63262306a36Sopenharmony_ci		t->task = NULL;
63362306a36Sopenharmony_ci		if (!task || IS_ERR(task))
63462306a36Sopenharmony_ci			continue;
63562306a36Sopenharmony_ci		kthread_stop(task);
63662306a36Sopenharmony_ci		/* no more switch may happen on this CPU at this point */
63762306a36Sopenharmony_ci		cluster = MPIDR_AFFINITY_LEVEL(cpu_logical_map(cpu), 1);
63862306a36Sopenharmony_ci		if (cluster == bL_switcher_cpu_original_cluster[cpu])
63962306a36Sopenharmony_ci			continue;
64062306a36Sopenharmony_ci		init_completion(&t->started);
64162306a36Sopenharmony_ci		t->wanted_cluster = bL_switcher_cpu_original_cluster[cpu];
64262306a36Sopenharmony_ci		task = bL_switcher_thread_create(cpu, t);
64362306a36Sopenharmony_ci		if (!IS_ERR(task)) {
64462306a36Sopenharmony_ci			wait_for_completion(&t->started);
64562306a36Sopenharmony_ci			kthread_stop(task);
64662306a36Sopenharmony_ci			cluster = MPIDR_AFFINITY_LEVEL(cpu_logical_map(cpu), 1);
64762306a36Sopenharmony_ci			if (cluster == bL_switcher_cpu_original_cluster[cpu])
64862306a36Sopenharmony_ci				continue;
64962306a36Sopenharmony_ci		}
65062306a36Sopenharmony_ci		/* If execution gets here, we're in trouble. */
65162306a36Sopenharmony_ci		pr_crit("%s: unable to restore original cluster for CPU %d\n",
65262306a36Sopenharmony_ci			__func__, cpu);
65362306a36Sopenharmony_ci		pr_crit("%s: CPU %d can't be restored\n",
65462306a36Sopenharmony_ci			__func__, bL_switcher_cpu_pairing[cpu]);
65562306a36Sopenharmony_ci		cpumask_clear_cpu(bL_switcher_cpu_pairing[cpu],
65662306a36Sopenharmony_ci				  &bL_switcher_removed_logical_cpus);
65762306a36Sopenharmony_ci	}
65862306a36Sopenharmony_ci
65962306a36Sopenharmony_ci	bL_switcher_restore_cpus();
66062306a36Sopenharmony_ci	bL_switcher_trace_trigger();
66162306a36Sopenharmony_ci
66262306a36Sopenharmony_ci	bL_activation_notify(BL_NOTIFY_POST_DISABLE);
66362306a36Sopenharmony_ci
66462306a36Sopenharmony_ciout:
66562306a36Sopenharmony_ci	unlock_device_hotplug();
66662306a36Sopenharmony_ci	mutex_unlock(&bL_switcher_activation_lock);
66762306a36Sopenharmony_ci}
66862306a36Sopenharmony_ci
66962306a36Sopenharmony_cistatic ssize_t bL_switcher_active_show(struct kobject *kobj,
67062306a36Sopenharmony_ci		struct kobj_attribute *attr, char *buf)
67162306a36Sopenharmony_ci{
67262306a36Sopenharmony_ci	return sprintf(buf, "%u\n", bL_switcher_active);
67362306a36Sopenharmony_ci}
67462306a36Sopenharmony_ci
67562306a36Sopenharmony_cistatic ssize_t bL_switcher_active_store(struct kobject *kobj,
67662306a36Sopenharmony_ci		struct kobj_attribute *attr, const char *buf, size_t count)
67762306a36Sopenharmony_ci{
67862306a36Sopenharmony_ci	int ret;
67962306a36Sopenharmony_ci
68062306a36Sopenharmony_ci	switch (buf[0]) {
68162306a36Sopenharmony_ci	case '0':
68262306a36Sopenharmony_ci		bL_switcher_disable();
68362306a36Sopenharmony_ci		ret = 0;
68462306a36Sopenharmony_ci		break;
68562306a36Sopenharmony_ci	case '1':
68662306a36Sopenharmony_ci		ret = bL_switcher_enable();
68762306a36Sopenharmony_ci		break;
68862306a36Sopenharmony_ci	default:
68962306a36Sopenharmony_ci		ret = -EINVAL;
69062306a36Sopenharmony_ci	}
69162306a36Sopenharmony_ci
69262306a36Sopenharmony_ci	return (ret >= 0) ? count : ret;
69362306a36Sopenharmony_ci}
69462306a36Sopenharmony_ci
69562306a36Sopenharmony_cistatic ssize_t bL_switcher_trace_trigger_store(struct kobject *kobj,
69662306a36Sopenharmony_ci		struct kobj_attribute *attr, const char *buf, size_t count)
69762306a36Sopenharmony_ci{
69862306a36Sopenharmony_ci	int ret = bL_switcher_trace_trigger();
69962306a36Sopenharmony_ci
70062306a36Sopenharmony_ci	return ret ? ret : count;
70162306a36Sopenharmony_ci}
70262306a36Sopenharmony_ci
70362306a36Sopenharmony_cistatic struct kobj_attribute bL_switcher_active_attr =
70462306a36Sopenharmony_ci	__ATTR(active, 0644, bL_switcher_active_show, bL_switcher_active_store);
70562306a36Sopenharmony_ci
70662306a36Sopenharmony_cistatic struct kobj_attribute bL_switcher_trace_trigger_attr =
70762306a36Sopenharmony_ci	__ATTR(trace_trigger, 0200, NULL, bL_switcher_trace_trigger_store);
70862306a36Sopenharmony_ci
70962306a36Sopenharmony_cistatic struct attribute *bL_switcher_attrs[] = {
71062306a36Sopenharmony_ci	&bL_switcher_active_attr.attr,
71162306a36Sopenharmony_ci	&bL_switcher_trace_trigger_attr.attr,
71262306a36Sopenharmony_ci	NULL,
71362306a36Sopenharmony_ci};
71462306a36Sopenharmony_ci
71562306a36Sopenharmony_cistatic struct attribute_group bL_switcher_attr_group = {
71662306a36Sopenharmony_ci	.attrs = bL_switcher_attrs,
71762306a36Sopenharmony_ci};
71862306a36Sopenharmony_ci
71962306a36Sopenharmony_cistatic struct kobject *bL_switcher_kobj;
72062306a36Sopenharmony_ci
72162306a36Sopenharmony_cistatic int __init bL_switcher_sysfs_init(void)
72262306a36Sopenharmony_ci{
72362306a36Sopenharmony_ci	int ret;
72462306a36Sopenharmony_ci
72562306a36Sopenharmony_ci	bL_switcher_kobj = kobject_create_and_add("bL_switcher", kernel_kobj);
72662306a36Sopenharmony_ci	if (!bL_switcher_kobj)
72762306a36Sopenharmony_ci		return -ENOMEM;
72862306a36Sopenharmony_ci	ret = sysfs_create_group(bL_switcher_kobj, &bL_switcher_attr_group);
72962306a36Sopenharmony_ci	if (ret)
73062306a36Sopenharmony_ci		kobject_put(bL_switcher_kobj);
73162306a36Sopenharmony_ci	return ret;
73262306a36Sopenharmony_ci}
73362306a36Sopenharmony_ci
73462306a36Sopenharmony_ci#endif  /* CONFIG_SYSFS */
73562306a36Sopenharmony_ci
73662306a36Sopenharmony_cibool bL_switcher_get_enabled(void)
73762306a36Sopenharmony_ci{
73862306a36Sopenharmony_ci	mutex_lock(&bL_switcher_activation_lock);
73962306a36Sopenharmony_ci
74062306a36Sopenharmony_ci	return bL_switcher_active;
74162306a36Sopenharmony_ci}
74262306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(bL_switcher_get_enabled);
74362306a36Sopenharmony_ci
74462306a36Sopenharmony_civoid bL_switcher_put_enabled(void)
74562306a36Sopenharmony_ci{
74662306a36Sopenharmony_ci	mutex_unlock(&bL_switcher_activation_lock);
74762306a36Sopenharmony_ci}
74862306a36Sopenharmony_ciEXPORT_SYMBOL_GPL(bL_switcher_put_enabled);
74962306a36Sopenharmony_ci
75062306a36Sopenharmony_ci/*
75162306a36Sopenharmony_ci * Veto any CPU hotplug operation on those CPUs we've removed
75262306a36Sopenharmony_ci * while the switcher is active.
75362306a36Sopenharmony_ci * We're just not ready to deal with that given the trickery involved.
75462306a36Sopenharmony_ci */
75562306a36Sopenharmony_cistatic int bL_switcher_cpu_pre(unsigned int cpu)
75662306a36Sopenharmony_ci{
75762306a36Sopenharmony_ci	int pairing;
75862306a36Sopenharmony_ci
75962306a36Sopenharmony_ci	if (!bL_switcher_active)
76062306a36Sopenharmony_ci		return 0;
76162306a36Sopenharmony_ci
76262306a36Sopenharmony_ci	pairing = bL_switcher_cpu_pairing[cpu];
76362306a36Sopenharmony_ci
76462306a36Sopenharmony_ci	if (pairing == -1)
76562306a36Sopenharmony_ci		return -EINVAL;
76662306a36Sopenharmony_ci	return 0;
76762306a36Sopenharmony_ci}
76862306a36Sopenharmony_ci
76962306a36Sopenharmony_cistatic bool no_bL_switcher;
77062306a36Sopenharmony_cicore_param(no_bL_switcher, no_bL_switcher, bool, 0644);
77162306a36Sopenharmony_ci
77262306a36Sopenharmony_cistatic int __init bL_switcher_init(void)
77362306a36Sopenharmony_ci{
77462306a36Sopenharmony_ci	int ret;
77562306a36Sopenharmony_ci
77662306a36Sopenharmony_ci	if (!mcpm_is_available())
77762306a36Sopenharmony_ci		return -ENODEV;
77862306a36Sopenharmony_ci
77962306a36Sopenharmony_ci	cpuhp_setup_state_nocalls(CPUHP_ARM_BL_PREPARE, "arm/bl:prepare",
78062306a36Sopenharmony_ci				  bL_switcher_cpu_pre, NULL);
78162306a36Sopenharmony_ci	ret = cpuhp_setup_state_nocalls(CPUHP_AP_ONLINE_DYN, "arm/bl:predown",
78262306a36Sopenharmony_ci					NULL, bL_switcher_cpu_pre);
78362306a36Sopenharmony_ci	if (ret < 0) {
78462306a36Sopenharmony_ci		cpuhp_remove_state_nocalls(CPUHP_ARM_BL_PREPARE);
78562306a36Sopenharmony_ci		pr_err("bL_switcher: Failed to allocate a hotplug state\n");
78662306a36Sopenharmony_ci		return ret;
78762306a36Sopenharmony_ci	}
78862306a36Sopenharmony_ci	if (!no_bL_switcher) {
78962306a36Sopenharmony_ci		ret = bL_switcher_enable();
79062306a36Sopenharmony_ci		if (ret)
79162306a36Sopenharmony_ci			return ret;
79262306a36Sopenharmony_ci	}
79362306a36Sopenharmony_ci
79462306a36Sopenharmony_ci#ifdef CONFIG_SYSFS
79562306a36Sopenharmony_ci	ret = bL_switcher_sysfs_init();
79662306a36Sopenharmony_ci	if (ret)
79762306a36Sopenharmony_ci		pr_err("%s: unable to create sysfs entry\n", __func__);
79862306a36Sopenharmony_ci#endif
79962306a36Sopenharmony_ci
80062306a36Sopenharmony_ci	return 0;
80162306a36Sopenharmony_ci}
80262306a36Sopenharmony_ci
80362306a36Sopenharmony_cilate_initcall(bL_switcher_init);
804