Linux C 程序 【03】线程栈空间

1.开发背景

        上一个篇章创建了线程，参考 FreeRTOS，每个线程都是有自己的内存空间，Linux上面也是一样的，这个篇章主要描述线程栈空间的设置。

2.开发需求

设计实验：

        1）创建线程，并配置线程内存大小

3.开发环境

        ubuntu20.04 + RK3568 + Linux4.19.232

4.实现步骤

4.1 实现代码

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include <pthread.h>

#include "com_save.h"
#include "app_log.h"

#define FILE_PATH_TEST   ("/home/reetoo/log/test")

typedef struct 
{
    /* 线程 */
    pthread_t thread;

}ctrl_t;

static ctrl_t s_ctrl = {0};
static ctrl_t *p = &s_ctrl;

/* 递归函数 */
void recursive_function(int depth) 
{
    char buffer[1024 * 1024]; // 使用一个较大的局部变量来加速栈的消耗
    (void)buffer;
    alog_info("%s Depth: %d\n", __func__, depth);
    recursive_function(depth + 1);
}

/* 线程函数 */
void* pthread_process(void* arg)
{
    /* 打印日志 */
    alog_info("i am pthread_process, pid: %d, pthread_self: %ld\r\n", getpid(), pthread_self());

    /* 递归函数调用 */
    recursive_function(0);

    return NULL;
}

/* 主函数 */
int main(int argc, char* argv[])
{
    /* 文件保存初始化 */
    csave_init(FILE_PATH_TEST);

    /* 日志模块初始化 */
    alog_init();

    /* 打印日志 */
    alog_info("i am main, pid: %d, pthread_self: %ld\r\n", getpid(), pthread_self());

    /* 创建线程 */
    pthread_attr_t attr;

    /* 设置线程属性 */
    size_t stacksize;
    pthread_attr_init(&attr);
    pthread_attr_setstacksize(&attr, 1024 * 128); // 设置线程栈大小
    pthread_attr_getstacksize(&attr, &stacksize); // 获取线程栈大小
    alog_info("128K stacksize: %ldBytes, %ldKBytes\n", stacksize, stacksize / 1024);

    pthread_attr_init(&attr);
    pthread_attr_setstacksize(&attr, 1024 * 127); // 设置线程栈大小
    pthread_attr_getstacksize(&attr, &stacksize); // 获取线程栈大小
    alog_info("127K stacksize: %ldBytes, %ldKBytes\n", stacksize, stacksize / 1024);

    pthread_create(&p->thread, &attr, (void*)pthread_process, NULL);
    pthread_attr_destroy(&attr);

    /* 等待线程结束 */
    pthread_join(p->thread, NULL);

    return 0;
}

4.2 测试结果

        由测试结果可知，线程的内存堆栈是可以设置的，而且有最小值，这里的最小内存是 128KB，小于128KB的按照默认大小 8192KB 配置。

        代码里递归函数 recursive_function 每递归一次消耗 1024KB 内存，完成了 0 - 6 共 7 次递归，第 8 次递归直接越界，产生段错误，内存溢出。