RTOS 系统线程栈溢出检查实现方式【对比 FreeRTOS 和 RT-Thread】

murphyzhao · 发表于 2019-6-27 09:09:24

本帖最后由 murphyzhao 于 2019-6-28 09:20 编辑

RTOS 系统线程栈溢出检查实现方式

在实现的时候主要考量哪些因素？我只想到下面两个方向，说说我自己的理解：

线程栈溢出检查方式
什么情况、什么时候检查线程栈

1. RT-Thread 线程栈的检查

1.1 线程栈溢出检查方式分析
线程栈溢出检查函数：

static void _rt_scheduler_stack_check(struct rt_thread *thread)
{
RT_ASSERT(thread != RT_NULL);
#if defined(ARCH_CPU_STACK_GROWS_UPWARD)
if (*((rt_uint8_t *)((rt_ubase_t)thread->stack_addr + thread->stack_size - 1)) != '#' ||
#else
if (*((rt_uint8_t *)thread->stack_addr) != '#' ||
#endif
(rt_ubase_t)thread->sp <= (rt_ubase_t)thread->stack_addr ||
(rt_ubase_t)thread->sp >
(rt_ubase_t)thread->stack_addr + (rt_ubase_t)thread->stack_size)
{
rt_ubase_t level;
rt_kprintf("thread:%s stack overflow\n", thread->name);
#ifdef RT_USING_FINSH
{
extern long list_thread(void);
list_thread();
}
#endif
level = rt_hw_interrupt_disable();
while (level);
}
#if defined(ARCH_CPU_STACK_GROWS_UPWARD)
else if ((rt_ubase_t)thread->sp > ((rt_ubase_t)thread->stack_addr + thread->stack_size))
{
rt_kprintf("warning: %s stack is close to the top of stack address.\n",
thread->name);
}
#else
else if ((rt_ubase_t)thread->sp <= ((rt_ubase_t)thread->stack_addr + 32))
{
rt_kprintf("warning: %s stack is close to end of stack address.\n",
thread->name);
}
#endif
}

复制代码

从上面的代码中可以直观地看到，RT-Thread 线程栈的检查方式分为两种情况：

栈向下增长
栈向上增长

判断栈的增长方向这个是必须的，因为系统要适配不同的 CPU 架构，而有些 CPU 架构的栈增长方向是可配置的。

虽然根据栈的增长方向分为两种检查方式，但核心的检查功能确是一样的，通过判断栈尾地址内容是否为 "#" 来进行初步溢出判断，如果栈尾地址内容不是 "#"，那么栈溢出。另外，代码还配合检查了线程栈帧地址是否超出栈的地址空间范围来辅助判断。

【这里感觉不严谨，如果线程栈刚好用到这个末尾，那就会误判；另外，无法检查其它任务改写栈的问题】

1.2. 什么时候检查

通常的做法就是在调度器切换线程的时候检查，调度器简化代码如下：

void rt_schedule(void)
{
rt_base_t level;
struct rt_thread *to_thread;
struct rt_thread *from_thread;
/* disable interrupt */
level = rt_hw_interrupt_disable();
to_thread = _get_highest_priority_thread(&highest_ready_priority);
from_thread = rt_current_thread;
rt_current_thread = to_thread;
_rt_scheduler_stack_check(to_thread);
rt_hw_interrupt_enable(level);
}

复制代码

从上面的代码可以看到 RT-Thread 在调度器中调用 _rt_scheduler_stack_check(to_thread); 函数检查 to_thread 线程的线程栈。
【有个疑问：为什么仅仅检查 to_thread，不同时检查 from_thread？】

2. FreeRTOS 线程栈检查
1.1 线程栈溢出检查方式分析
线程栈溢出检查函数：

FreeRTOS 的栈溢出检查就有了一个特点，首先使用了 taskFIRST_CHECK_FOR_STACK_OVERFLOW 函数，如下所示：（这里粘贴的代码实在看不下去，放图吧）

carbon (2).png

然后，使用了 taskSECOND_CHECK_FOR_STACK_OVERFLOW 函数：

carbon (3).png

从上面的两段代码，可以看到，FreeRTOS 的栈溢出检查分为第一次和第二次，但同 RT-Thread 类似，也分了栈的增长方向。
第一次检查，仅仅判断栈顶指针是否超出了栈尾，具体的栈溢出操作实现交给了 vApplicationStackOverflowHook 回调函数。
第二次检查，通过内存比较函数，比较 pcEndOfStack 栈尾 20 个字节大小的空间是否与栈的填充值 tskSTACK_FILL_BYTE 匹配（填充值为 0xa5）。

当然，FreeRTOS 关键起作用的是第二次检查，看着相对还是比较靠谱的（相对 RT-Thread 仅检查一个字节）。FreeRTOS 这种检查方式肯定是与其线程栈的分配方式有关系的，因为它检查的是栈尾部的 20 字节，也就是说，FreeRTOS 为每一个线程栈都预留了 20 字节的空档，造成一部分空间的浪费。

1.2. 什么时候检查

当然，FreeRTOS 也是在调度器切换任务的时候检查。直接上代码了，如下：

void vTaskSwitchContext( void )
{
/* Check for stack overflow, if configured. */
taskFIRST_CHECK_FOR_STACK_OVERFLOW();
taskSECOND_CHECK_FOR_STACK_OVERFLOW();
/* Select a new task to run using either the generic C or port
optimised asm code. */
taskSELECT_HIGHEST_PRIORITY_TASK();
traceTASK_SWITCHED_IN();
}

复制代码

如上，注意到一个问题，FreeRTOS 的栈检查，貌似是检查的 from_thread，而不是像 RT-Thread 一样检查的 to_thread？

总结

不知道设计栈溢出检查的时候有哪些考究，RT-Thread 和 FreeRTOS 都没有对同时检查 from_thread 和 to_thread。这两个系统的栈检查方式差不多，相对来说，感觉 FreeRTOS 的更严谨些，但都不能完全检查到，也没法检查线程栈被其它任务错误修改的情况。

----

如上分析，如有问题，请拍砖