栈大小可以怎么改?

简介: 栈大小可以怎么改?

缘起

前一段日子,遇到了好几个栈溢出异常,相关文章可以点击 这里这里这里。解决栈溢出,一般可以从两个方面进行处理。简而言之就是:开源节流!

开源 就是增大栈尺寸。节流 就是减少不必要的内存占用。解决栈溢出,正统的方法是减少内存占用(比如,尽量按引用传递参数)。但是有些情况下了解如何增大线程栈尺寸也非常有必要,可以应急!本文介绍几种增大栈尺寸的方法。

1. 修改 API 参数

可以在创建线程的时候指定栈保留大小。创建线程常用的 APICreateThread,函数原型如下:

HANDLE CreateThread(
  LPSECURITY_ATTRIBUTES   lpThreadAttributes,
  SIZE_T                  dwStackSize,
  LPTHREAD_START_ROUTINE  lpStartAddress,
  __drv_aliasesMem LPVOID lpParameter,
  DWORD                   dwCreationFlags,
  LPDWORD                 lpThreadId
);

可以通过参数 dwStackSize 指定栈大小,单位是字节。如果该参数是 0 的话,会使用 PE 文件头中的值。

2. 修改工程配置

可以在 vs 中设置默认的栈保留大小和栈提交大小,相应的值会记录在生成的 PE 头中。

在工程上右键 -> 属性 -> 配置属性 -> 链接器 -> 系统 -> 堆栈保留大小堆栈提交大小。即可设置。具体位置如下图:

stack-reverse-setting.png

说明:这里设置的单位是字节。而且是 10 进制的。

3. 手动修改PE文件头

如有由于某些原因,未能在开发的时候调整默认值,可以通过二进制编辑工具直接修改已经生成好的 PE 文件头中的字段来修改栈保留大小和栈提交大小。

stack-reverse-size-in-image-header.png

SizeOfStackReserve 指定了栈保留大小,默认值是 1MB,也就是这里的 00100000

SizeOfStackCommit 指定了栈提交大小,默认值是 4KB,也就是这里的 00001000

说明:这里的值是用 16 进制表示的。

深入思考

线程栈默认大小是由哪个 PE 头决定的?主程序还是动态库?还是在主程序中创建的线程由主程序决定,在动态库中创建的线程由动态库决定?

为了明确这个问题,特意逆向了一下 CreateThread,内部会调用 CreateRemoteThread。在逆向的过程中突然想到大名鼎鼎的 ReactOS,于是赶紧翻看 CreateRemoteThread 的源码,CreateRemoteThread 内部会调用 BaseCreateStack 创建线程栈。我摘录了一段关键代码(为了突出重点,代码做了调整删减):

NTSTATUS WINAPI BaseCreateStack(_In_ HANDLE hProcess,
        _In_opt_ SIZE_T StackCommit,
        _In_opt_ SIZE_T StackReserve,
        _Out_ PINITIAL_TEB InitialTeb 
)     
{
     /* Get the Image Headers */
     PIMAGE_NT_HEADERS Headers = RtlImageNtHeader(NtCurrentPeb()->ImageBaseAddress);
     if (!Headers) return STATUS_INVALID_IMAGE_FORMAT;

     if (StackReserve == 0)
         StackReserve = Headers->OptionalHeader.SizeOfStackReserve;

     if (StackCommit == 0)
     {
         StackCommit = Headers->OptionalHeader.SizeOfStackCommit;
     }
     /* Check if the commit is higher than the reserve */
     else if (StackCommit >= StackReserve)
     {
         /* Grow the reserve beyond the commit, up to 1MB alignment */
         StackReserve = ROUND_UP(StackCommit, 1024 * 1024);
     }

    /*省略N行*/
 }

从上述代码可知,如果传入的 StackReserveStackCommit )是 0 的话,那么会取 NtCurrentPeb()->ImageBaseAddres 对应的文件头中的相关字段。而 NtCurrentPeb() 一个进程只有一份,对应着主程序。由此可以得到结论:线程栈保留(提交)大小是根据主程序文件头中的相关字段来决定的,而不是根据动态链接库中的字段。

总结

  • 可以在创建线程的时候指定栈大小,也可以在 vs 中设置默认的线程大小,还可以通过修改 PE 文件头中的对应字段来改变栈大小。
  • 不论是在 vs 中设置,还是手动修改 PE 文件头中的对应字段,都需要修改主程序对应的字段,修改动态链接库中的字段并不起作用。

参考资料

相关文章
|
15天前
|
C语言
【数据结构】栈和队列(c语言实现)(附源码)
本文介绍了栈和队列两种数据结构。栈是一种只能在一端进行插入和删除操作的线性表,遵循“先进后出”原则;队列则在一端插入、另一端删除,遵循“先进先出”原则。文章详细讲解了栈和队列的结构定义、方法声明及实现,并提供了完整的代码示例。栈和队列在实际应用中非常广泛,如二叉树的层序遍历和快速排序的非递归实现等。
90 9
|
6天前
|
存储 算法
非递归实现后序遍历时,如何避免栈溢出?
后序遍历的递归实现和非递归实现各有优缺点,在实际应用中需要根据具体的问题需求、二叉树的特点以及性能和空间的限制等因素来选择合适的实现方式。
15 1
|
9天前
|
存储 算法 Java
数据结构的栈
栈作为一种简单而高效的数据结构,在计算机科学和软件开发中有着广泛的应用。通过合理地使用栈,可以有效地解决许多与数据存储和操作相关的问题。
|
12天前
|
存储 JavaScript 前端开发
执行上下文和执行栈
执行上下文是JavaScript运行代码时的环境,每个执行上下文都有自己的变量对象、作用域链和this值。执行栈用于管理函数调用,每当调用一个函数,就会在栈中添加一个新的执行上下文。
|
14天前
|
存储
系统调用处理程序在内核栈中保存了哪些上下文信息?
【10月更文挑战第29天】系统调用处理程序在内核栈中保存的这些上下文信息对于保证系统调用的正确执行和用户程序的正常恢复至关重要。通过准确地保存和恢复这些信息,操作系统能够实现用户模式和内核模式之间的无缝切换,为用户程序提供稳定、可靠的系统服务。
41 4
|
1月前
|
算法 程序员 索引
数据结构与算法学习七:栈、数组模拟栈、单链表模拟栈、栈应用实例 实现 综合计算器
栈的基本概念、应用场景以及如何使用数组和单链表模拟栈,并展示了如何利用栈和中缀表达式实现一个综合计算器。
30 1
数据结构与算法学习七:栈、数组模拟栈、单链表模拟栈、栈应用实例 实现 综合计算器
|
18天前
|
算法 安全 NoSQL
2024重生之回溯数据结构与算法系列学习之栈和队列精题汇总(10)【无论是王道考研人还是IKUN都能包会的;不然别给我家鸽鸽丢脸好嘛?】
数据结构王道第3章之IKUN和I原达人之数据结构与算法系列学习栈与队列精题详解、数据结构、C++、排序算法、java、动态规划你个小黑子;这都学不会;能不能不要给我家鸽鸽丢脸啊~除了会黑我家鸽鸽还会干嘛?!!!
|
1月前
初步认识栈和队列
初步认识栈和队列
58 10
|
1月前
数据结构(栈与列队)
数据结构(栈与列队)
17 1
|
1月前
|
算法
数据结构与算法二:栈、前缀、中缀、后缀表达式、中缀表达式转换为后缀表达式
这篇文章讲解了栈的基本概念及其应用,并详细介绍了中缀表达式转换为后缀表达式的算法和实现步骤。
44 3

热门文章

最新文章