线程不同于进程,每个线程的栈是独立的,不共享的。每个线程栈保存了返回地址、旧的栈指针、局部变量、函数调用的参数。如果栈太大会浪费内存,开启多线程又影响程序性能,如果栈太小可能会发生溢出,溢出会触发异常,操作系统会调用异常处理程序解决溢出,异常处理程序会进行栈扩展,也会影响程序性能。所以根据具体业务场景设置线程栈的大小就起到了作用。想要改变默认线程栈的大小,但又不想自己手动处理线程栈的分配问题,使用pthread_attr_getstacksize函数和pthread_attr_setstacksize函数就非常方便,短短几行代码就能实现。
在 Linux 中,栈大小在线程属性对象中设置,也就是说,将类型为pthread_attr_t的参数threadAttr传递给函数 pthread_create()。在设置属性之前,需要通过调用 pthread_attr_init() 函数来初始化这个对象,然后使用pthread_attr_setstacksize()函数设置线程栈空间大小,单位为字节;接着调用pthread_create()创建线程,该线程打印自己线程栈的大小,完毕后调用 pthread_attr_destroy()来销毁属性对象。
#include<stdio.h>
#include<pthread.h>
#include<unistd.h>
#include<errno.h>
#include<stdlib.h>
#include<fcntl.h>
#include<sys/types.h>
pthread_attr_t attr;
size_t stacksize;
void *thread()
{
pthread_attr_getstacksize(&attr,&stacksize);
printf("修改之后线程栈的大小是%zuMB\n",stacksize/(1<<20));
}
int main(int argc,char **argv)
{
int errno,ret;
pthread_t tid;
errno=pthread_attr_init(&attr);
if(errno!=0)
printf("init error");
if(pthread_attr_getstacksize(&attr,&stacksize)==0)
printf("修改之前线程栈的大小是%zuMB\n",stacksize/(1<<20));
else
printf("pthread_attr_getstacksize error");
if(pthread_attr_setstacksize(&attr,1<<24)==0)
pthread_create(&tid,&attr,thread,NULL);
pthread_attr_destroy(&attr);
pthread_exit(NULL);
exit(0);
}
运行结果:
这样就完成了线程栈大小的修改。