现在的位置: 首页 > 综合 > 正文

VC中捕捉StackOverflow异常

2013年10月20日 ⁄ 综合 ⁄ 共 2296字 ⁄ 字号 评论关闭

    程序中栈溢出通常会导致进程直接关闭。本文分析Windows中栈增长和溢出过程,并提供一种捕捉栈溢出的方法,作为调试程序的参考。

栈大小

    VC中,栈大小默认为1M,但可以在编译的时候或在创建线程的时候指定其他值。

栈增长和溢出

   栈底地址在线程生命期内是常量,栈顶地址保存在SP(ESP、RSP)寄存器。栈地址从高向低增长,因此栈增长,对应着SP寄存器减小。

   1M空间只是预留进程地址空间(Reserve的过程),默认没有分配物理内存。随着程序对栈的使用,操作系统逐步为地址分配物理内存(commit的过程)。x86平台下,每次commit一个页面,也就是4K大小的物理内存。1M的地址空间可以对应256个页面。

   正常情况下,假设已经为栈分配了N个物理页面。只要N<254,第1到N个页面就是可读写的。第N+1个页面也会分配物理内存,但是其保护位是PAGE_GUARD,不可读写。现在栈继续增长,需要N+1个页面的内存,当程序试图读写第N+1个页面的时候,由于PAGE_GUARD属性,发生STATUS_GUARD_PAGE_VIOLATION。操作系统捕捉该异常,把第N+1个页面的保护属性修改为可读写。同时为第N+2个页面地址分配物理内存,设置保护属性为PAGE_GUARD。

   在N=254的时候,如果栈继续增长,则会触发第255个页面的STATUS_GUARD_PAGE_VIOLATION异常,操作系统捕捉该异常,把第255个页面设置为可读写,然后触发栈溢出异常(EXCEPTION_STACK_OVERFLOW)。

   N=254与N<254不同点是

  1. N=254时,不为第256个页面分配物理内存。实际上系统永远不会为第256个页面地址分配物理内存,第256个页面永远只是Reserve状态。
  2. N=254时,触发EXCEPTION_STACK_OVERFLOW。

栈溢出捕捉和恢复

   EXCEPTION_STACK_OVERFLOW不是C++异常,只能用操作系统提供的结构化异常(Structured Exception Handling,SEH)措施来处理。如下所示:

__try{
		// 导致栈溢出的操作
	}__except((GetExceptionCode() == EXCEPTION_STACK_OVERFLOW) ? 
				EXCEPTION_EXECUTE_HANDLER : EXCEPTION_CONTINUE_SEARCH)
	{
		int reset = _resetstkoflw();
		if(reset)
			std::cout << "栈恢复成功。递归次数=" << count << std::endl;
		else
			std::cout << "栈恢复失败。" << count << std::endl;
	}

   注意到代码中,在捕捉到栈溢出异常后,调用了_resetstkoflw函数。_resetstkoflw是VC运行库提供的函数,作用是恢复栈内存中相应页的PAGE_GUARD属性。如果不执行_resetstkoflw,本次溢出异常看似捕捉到了,程序也可以继续运行。但是由于栈中的页面不存在PAGE_GUARD属性,如果栈下一次溢出,访问第256个页面的时候,由于第256个页面没有分配物理内存, 直接抛出ACCESS_VIOLATION异常,而不是EXCEPTION_STACK_OVERFLOW。

   _resetstkoflw是有返回值的。如果_resetstkoflw执行的时候,栈又不够用,则返回0。此时无法恢复栈到正常状态。SetThreadStackGuarantee函数可以要求系统在抛出EXCEPTION_STACK_OVERFLOW异常的时候,保证仍然有指定大小的空闲区可以用。SetThreadStackGuarantee实际上减少了程序实际可用栈的大小。

 

完整代码

#include"stdafx.h"
#include<iostream>
#include<Windows.h>

int count = 0;

void recursive(){
     char dummy[1024];
     ++count;
     recursive();
}

 

void testRecursive(){

     __try{
         recursive();
     }__except((GetExceptionCode() == EXCEPTION_STACK_OVERFLOW) ?
                   EXCEPTION_EXECUTE_HANDLER : EXCEPTION_CONTINUE_SEARCH)
     {
         int reset = _resetstkoflw();
         if(reset)
              std::cout << "栈恢复成功。递归次数=" << count << std::endl;
         else
              std::cout << "栈恢复失败。" << count << std::endl;
     }

}

 
int_tmain(intargc, _TCHAR* argv[])
{
     std::cout << GetProcessId(GetCurrentProcess()) << std::endl;
 
     ULONG stkLimit = 1024*4;// 4K
     if(!SetThreadStackGuarantee(&stkLimit)){
         std::cout << "SetThreadStackGuarantee失败" << std::endl;
         return -1;
     }


     // 第一次栈溢出
     testRecursive();

     // 第二次栈溢出
     testRecursive();

     std::cout << "测试结束" << std::endl;

     bool tmp = std::cin >> tmp;
     return 0;
}

 

 

抱歉!评论已关闭.