检测内核的堆栈溢出

最新推荐文章于 2023-07-15 19:00:25 发布

肉冬瓜

最新推荐文章于 2023-07-15 19:00:25 发布

阅读量3.9k

点赞数 1

分类专栏： linux

linux 专栏收录该内容

23 篇文章 0 订阅

订阅专栏

内核堆栈溢出通常有两种情况。一种是函数调用栈超出了内核栈THREAD_SIZE的大小，这是栈底越界，另一种是栈上缓冲越界访问，这是栈顶越界。

检测栈底越界

以arm平台为例，内核栈THREAD_SIZE为8K,当调用栈层次过多或某调用栈上分配过大的空间，就会导致它越界。越界后struct thread_info结构可能被破坏，轻则内核 panic，重则内核数据被覆盖仍继续运行。

检测这类栈溢出较通用的办法是在每次中断到来的时候检查当前的栈指针sp是否超过了某个阈值STACK_WARN。以下是在arm支持上DEBUG_STACKOVERFLOW的 arm-debug_stackoverflow.patch.

--- a/arch/arm/include/asm/thread_info.h
+++ b/arch/arm/include/asm/thread_info.h
@@ -19,6 +19,9 @@
 #define THREAD_SIZE            8192
 #define THREAD_START_SP                (THREAD_SIZE - 8)

+#define THREAD_MASK (THREAD_SIZE - 1UL)
+#define STACK_WARN     (THREAD_SIZE / 8)
+
 #ifndef __ASSEMBLY__

 struct task_struct;
diff --git a/arch/arm/kernel/irq.c b/arch/arm/kernel/irq.c
--- a/arch/arm/kernel/irq.c
+++ b/arch/arm/kernel/irq.c
@@ -56,6 +56,24 @@ int arch_show_interrupts(struct seq_file *p, int prec)
        return 0;
 }

+static inline void check_stack_overflow(void)
+{
+
+       register unsigned long sp asm ("sp");
+
+       sp &= THREAD_MASK;
+
+       /*
+        * Check for stack overflow: is there less than STACK_WARN free?
+        * STACK_WARN is defined as 1/8 of THREAD_SIZE by default.
+        */
+       if (unlikely(sp < (sizeof(struct thread_info) + STACK_WARN))) {
+               printk("do_IRQ: stack overflow: %ld\n",
+                      sp - sizeof(struct thread_info));
+               dump_stack();
+       }
+}
+
 /*
  * handle_IRQ handles all hardware IRQ's.  Decoded IRQs should
  * not come via this function.  Instead, they should provide their
@@ -68,6 +86,8 @@ void handle_IRQ(unsigned int irq, struct pt_regs *regs)

        irq_enter();

+       check_stack_overflow();

        /*
         * Some hardware gives randomly wrong interrupts.  Rather
         * than crashing, do something sensible.

以上的方法基于内核的中断栈在当前的cpu线程栈上和内核栈8K对齐两个前提。这种方法有两个缺点：

栈已经溢出了,但中断还没来得及发生。使用lkdtm验证如下：
```
# mount -t debugfs none /d
# echo OVERFLOW > /d//provoke-crash/DIRECT
```
在中断中没有得到警告信息。
arm的中断很可能只发生在某个core上，并不是均等的，这样也不能检测其它的core上内核栈溢出。这可能也是内核没有为arm提供DEBUG_STACK_OVERFLOW的原因。

内核的STACK_TRACER是另一种检测栈底溢出更可靠的方法，选上内核配置CONFIG_STACK_TRACER后，使用方法如下：

  # echo 10 > /sys/module/lkdtm/parameters/recur_count
  # echo 1 > /proc/sys/kernel/stack_tracer_enabled
  # cat stack_trace

打印出当前系统内核栈占用最多的函数。如果此时内核panic了，那也可以通过 Kdump得到RAMDUMP后，用crash工具查看是否由于栈底溢出导致的。

发现是由于内核栈溢出导致的问题后，不必抱怨内核栈太小。在堆上分配空间减小当前栈空间的占用，或通过workqueue下半部的方式减小调用栈层次等可以解决这个问题。不过内核栈真的很小，所以2.6的内核才将task_struct结构没有放置内核栈中，见这里。

检测栈顶越界

对于栈顶越界，gcc提供了支持。打开内核配置CONFIG_CC_STACKPROTECTOR后，会打开编译选项-fstack-protector.使用lkdtm验证如下：

  # echo CORRUPT_STACK > /d/provoke-crash/DIRECT 
  [ 1320.195246] lkdtm: Performing direct entry CORRUPT_STACK
  [ 1320.195681] Kernel panic - not syncing: stack-protector: Kernel stack is corrupted in: c03383dc

内核默认抛出panic.

内核的规模已经从小湖膨胀到大海一样，让身处其中的水手们晕头转向。如果用好内核这些配备的雷达声纳等工具，是不是觉得世界稍美好了一点呢？