Linux应用程序获取线程栈的信息

一. 参考网址

　　1. Linux应用程序设计：用一种讨巧方式，来获取线程栈的使用信息

　　2. chatGPT的输出

二. 代码实战

　　1. 适用于ARM的汇编代码

#define _GNU_SOURCE    //没有此句话，则编译会出错

#include <stdio.h>
#include <string.h>
#include <pthread.h>
#include <unistd.h>
#include <sys/resource.h>

typedef char s8;
typedef unsigned char u8;
typedef int s32;

s32 PrintThreadStackInfo(s8 *comment)
{
    size_t used, avail;
    pthread_attr_t attr;
    void *stack_addr;
    size_t stack_size;
    size_t esp_val;        // 获取栈寄存器 ESP 的当前值
    
#if 0
    asm("movl %%esp, %0" : "=m"(esp_val) :);        //X86汇编
#else
    asm volatile ("mov %0, sp\n" : "=r" (esp_val));    //Arm汇编
#endif
    // 通过线程属性，获取栈区的起始地址和空间总大小
    memset(&attr, 0, sizeof(pthread_attr_t));
    pthread_getattr_np(pthread_self(), &attr);
    pthread_attr_getstack(&attr, &stack_addr, &stack_size);
    pthread_attr_destroy(&attr);
    avail = esp_val - (size_t)stack_addr;
    used = stack_size - avail;

    if (comment)
        printf(comment);
    printf("espVal = 0x%x \n", esp_val);
    printf("statck top = %p \n", stack_addr);
    printf("stack bottom = %p \n", stack_addr + stack_size);
    printf("***stack info: used = %d, avail = %d, total = %d***\n", 
            used, avail, stack_size);
    
    return SUCCESS;
}

　　2. 编译注意事项

　　1）函数：pthread_getattr_np

 　　3. chatGPT输出

对不起，我的错误。你可以使用`pthread_getattr_np`函数和`pthread_attr_getstack`函数来获取当前线程的栈信息。这里是一个修改后的示例代码：

```c
#include <stdio.h>
#include <pthread.h>
#include <unistd.h>

// 获取栈信息的函数
void print_stack_info() {
    pthread_attr_t attr;
    size_t stack_size;
    void *stack_addr;

    // 初始化attr对象
    pthread_getattr_np(pthread_self(), &attr);

    // 获取栈大小和栈地址
    pthread_attr_getstack(&attr, &stack_addr, &stack_size);

    // 计算已使用的栈空间大小
    unsigned long current_sp;
    asm volatile ("mov %0, sp\n" : "=r" (current_sp));
    long used_stack_size = (char*)stack_addr + stack_size - (char*)current_sp;

    // 计算剩余的栈空间大小
    long free_stack_size = stack_size - used_stack_size;

    // 打印栈信息
    printf("Stack Address: %p\n", stack_addr);
    printf("Stack Size: %lu bytes\n", stack_size);
    printf("Used Stack Size: %ld bytes\n", used_stack_size);
    printf("Free Stack Size: %ld bytes\n", free_stack_size);

    // 销毁attr对象
    pthread_attr_destroy(&attr);
}

int main() {
    // 创建一个线程，并在其中调用print_stack_info函数
    pthread_t thread;
    pthread_create(&thread, NULL, (void *)print_stack_info, NULL);
    pthread_join(thread, NULL);

    return 0;
}
```

在这个修改后的示例代码中，我们使用`pthread_getattr_np`函数获取当前线程的属性，并使用`pthread_attr_getstack`函数获取栈的大小和地址。然后，我们使用汇编语言的方式获取当前栈指针位置，从而计算已使用的栈空间大小。最后，我们计算剩余的栈空间大小并打印所有的栈信息。