148 lines
4.2 KiB
C
148 lines
4.2 KiB
C
/*
|
|
* Copyright (c) 2006-2024 RT-Thread Development Team
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*
|
|
* Change Logs:
|
|
* Date Author Notes
|
|
* 2024/9/12 zhujiale the first version
|
|
*/
|
|
|
|
#include "smp.h"
|
|
|
|
#define DBG_TAG "SMP"
|
|
#define DBG_LVL DBG_INFO
|
|
#include <rtdbg.h>
|
|
|
|
static struct rt_smp_call rt_smp_work[RT_CPUS_NR];
|
|
static rt_atomic_t rt_smp_wait;
|
|
|
|
static rt_err_t smp_call_handler(struct rt_smp_event *event)
|
|
{
|
|
switch (event->event_id)
|
|
{
|
|
case SMP_CALL_EVENT_FUNC:
|
|
event->func(event->data);
|
|
rt_atomic_add(&rt_smp_wait, 1);
|
|
break;
|
|
default:
|
|
LOG_E("error event id\n");
|
|
return -RT_ERROR;
|
|
}
|
|
return RT_EOK;
|
|
}
|
|
void rt_smp_call_ipi_handler(int vector, void *param)
|
|
{
|
|
int err;
|
|
int cur_cpu = rt_hw_cpu_id();
|
|
rt_spin_lock(&rt_smp_work[cur_cpu].lock);
|
|
|
|
if (rt_smp_work[cur_cpu].event.event_id)
|
|
{
|
|
err = smp_call_handler(&rt_smp_work[cur_cpu].event);
|
|
if (err)
|
|
{
|
|
LOG_E("Have no event\n");
|
|
rt_memset(&rt_smp_work[cur_cpu].event, 0, sizeof(struct rt_smp_event));
|
|
rt_spin_unlock(&rt_smp_work[cur_cpu].lock);
|
|
}
|
|
rt_memset(&rt_smp_work[cur_cpu].event, 0, sizeof(struct rt_smp_event));
|
|
}
|
|
rt_spin_unlock(&rt_smp_work[cur_cpu].lock);
|
|
}
|
|
|
|
/**
|
|
* @brief call function on specified CPU ,
|
|
*
|
|
* @param cpu_mask cpu mask for call
|
|
* @param func the function pointer
|
|
* @param data the data pointer
|
|
* @param flag call flag if you set SMP_CALL_WAIT_ALL
|
|
* then it will wait all cpu call finish and return
|
|
* else it will call function on specified CPU and return immediately
|
|
* @param cond the condition function pointer,if you set it then it will call function only when cond return true
|
|
*/
|
|
void rt_smp_call_func_cond(int cpu_mask, rt_smp_call_func_back func, void *data, rt_uint8_t flag, rt_smp_cond cond)
|
|
{
|
|
RT_DEBUG_NOT_IN_INTERRUPT;
|
|
struct rt_smp_event event;
|
|
rt_bool_t need_call = RT_TRUE, need_wait = RT_FALSE;
|
|
int cur_cpu = rt_hw_cpu_id();
|
|
int cpuid = 1 << cur_cpu;
|
|
int tmp_id = 0, cpu_nr = 0;
|
|
int tmp_mask;
|
|
int irq_flag;
|
|
|
|
if (flag == SMP_CALL_WAIT_ALL)
|
|
{
|
|
need_wait = RT_TRUE;
|
|
rt_atomic_store(&rt_smp_wait, 0);
|
|
}
|
|
|
|
if (cpuid & cpu_mask)
|
|
{
|
|
func(data);
|
|
cpu_mask = cpu_mask & (~cpuid);
|
|
}
|
|
|
|
if (!cpu_mask)
|
|
need_call = RT_FALSE;
|
|
|
|
tmp_mask = cpu_mask;
|
|
if (need_call)
|
|
{
|
|
while (tmp_mask)
|
|
{
|
|
if ((tmp_mask & 1) && (tmp_id < RT_CPUS_NR))
|
|
{
|
|
if (cond && !cond(tmp_id, data))
|
|
continue;
|
|
cpu_nr++;
|
|
event.event_id = SMP_CALL_EVENT_FUNC;
|
|
event.func = func;
|
|
event.data = data;
|
|
event.cpu_mask = cpu_mask;
|
|
irq_flag = rt_spin_lock_irqsave(&rt_smp_work[tmp_id].lock);
|
|
rt_smp_work[tmp_id].event = event;
|
|
rt_spin_unlock_irqrestore(&rt_smp_work[tmp_id].lock,irq_flag);
|
|
}
|
|
tmp_id++;
|
|
tmp_mask = tmp_mask >> 1;
|
|
}
|
|
rt_hw_ipi_send(RT_FUNC_IPI, cpu_mask);
|
|
}
|
|
|
|
if (need_wait)
|
|
{
|
|
while (rt_atomic_load(&rt_smp_wait) != cpu_nr);
|
|
}
|
|
}
|
|
|
|
void rt_smp_call_each_cpu(rt_smp_call_func_back func, void *data, rt_uint8_t flag)
|
|
{
|
|
rt_smp_call_func_cond(RT_ALL_CPU, func, data, flag, RT_NULL);
|
|
}
|
|
|
|
void rt_smp_call_each_cpu_cond(rt_smp_call_func_back func, void *data, rt_uint8_t flag, rt_smp_cond cond_func)
|
|
{
|
|
rt_smp_call_func_cond(RT_ALL_CPU, func, data, flag, cond_func);
|
|
}
|
|
void rt_smp_call_any_cpu(int cpu_mask, rt_smp_call_func_back func, void *data, rt_uint8_t flag)
|
|
{
|
|
rt_smp_call_func_cond(cpu_mask, func, data, flag, RT_NULL);
|
|
}
|
|
|
|
void rt_smp_call_any_cpu_cond(int cpu_mask, rt_smp_call_func_back func, void *data, rt_uint8_t flag, rt_smp_cond cond_func)
|
|
{
|
|
rt_smp_call_func_cond(cpu_mask, func, data, flag, cond_func);
|
|
}
|
|
|
|
void rt_smp_init(void)
|
|
{
|
|
for (int i = 0; i < RT_CPUS_NR; i++)
|
|
{
|
|
rt_memset(&rt_smp_work[i], 0, sizeof(struct rt_smp_call));
|
|
rt_spin_lock_init(&rt_smp_work[i].lock);
|
|
}
|
|
}
|