kernel_samsung_a34x-permissive/drivers/misc/mediatek/lpm/mtk_lpm_registry.c
2024-04-28 15:49:01 +02:00

89 lines
1.7 KiB
C
Executable file

// SPDX-License-Identifier: GPL-2.0
/*
* Copyright (c) 2019 MediaTek Inc.
*/
#include <linux/kernel.h>
#include <linux/cpuidle.h>
#include <linux/spinlock.h>
#include <linux/workqueue.h>
#include "mtk_lpm_registry.h"
#include "mtk_lpm_internal.h"
#define MTK_LPM_REGISTRY_MAGIC 0xCDAC9131
struct mtk_lpm_registry_wk {
unsigned int magic;
int type;
void *priv;
blockcall cb;
struct cpumask cpus;
};
#define INIT_MTK_LPM_REG_WK(wk, _type, _cb, _priv) ({\
(wk)->magic = MTK_LPM_REGISTRY_MAGIC;\
(wk)->type = _type;\
(wk)->cb = _cb;\
(wk)->priv = _priv; })
#define IS_MTK_LPM_REG_WK(wk)\
((wk) && ((wk)->magic == MTK_LPM_REGISTRY_MAGIC))
#define IS_MTK_LPM_REG_WAKEALL(cpumask)\
(cpumask_weight(cpumask) == num_online_cpus())
static long mtk_lpm_registry_work(void *pData)
{
struct mtk_lpm_registry_wk *lpm_wk =
(struct mtk_lpm_registry_wk *)pData;
unsigned long flags;
if (!IS_MTK_LPM_REG_WK(lpm_wk))
return 0;
switch (lpm_wk->type) {
case MTK_LPM_REG_PER_CPU:
mtk_lpm_system_lock(flags);
lpm_wk->cb(smp_processor_id(), lpm_wk->priv);
mtk_lpm_system_unlock(flags);
synchronize_rcu();
break;
case MTK_LPM_REG_ALL_ONLINE:
mtk_lpm_system_lock(flags);
cpumask_set_cpu(smp_processor_id(), &lpm_wk->cpus);
if (IS_MTK_LPM_REG_WAKEALL(&lpm_wk->cpus))
lpm_wk->cb(smp_processor_id(), lpm_wk->priv);
mtk_lpm_system_unlock(flags);
synchronize_rcu();
break;
default:
break;
}
return 0;
}
int mtk_lpm_do_work(int type, blockcall call, void *priv)
{
int cpu;
struct mtk_lpm_registry_wk lpm_wk;
INIT_MTK_LPM_REG_WK(&lpm_wk, type, call, priv);
cpuidle_pause_and_lock();
cpumask_clear(&lpm_wk.cpus);
for_each_online_cpu(cpu) {
work_on_cpu(cpu, mtk_lpm_registry_work, &lpm_wk);
}
cpuidle_resume_and_unlock();
return 0;
}