在 SIGSEGV 信号后打破无限循环答案

【问题标题】：Break Infinite Loop After SIGSEVG Signal在 SIGSEGV 信号后打破无限循环
【发布时间】：2020-04-24 12:51:07
【问题描述】：

我正在尝试使用我的程序找到我的最大堆栈大小。我知道，在我的操作系统上，堆栈的最大大小是 8MB。所以我试图进行无限循环，直到它导致分段错误。然后我需要打破这个循环，这样我就可以计算大小。我怎样才能做到这一点？ SIGSEGV 信号后有没有办法打破无限循环？

 static void handler(int signo)
    {
        if(signo == SIGSEGV)
        {
            printf("Waoh, caught signal\n");
        }
    }

    int main()
    {
        int counter = 8964;
        stack_t sigstack;
        sigstack.ss_sp = malloc(SIGSTKSZ);
        if( sigstack.ss_sp == NULL)
        {
            printf("Err: malloc error\n");
            exit(EXIT_FAILURE);
        }
        sigstack.ss_size = SIGSTKSZ;
        sigstack.ss_flags = 0;

        if(sigaltstack(&sigstack, NULL) == -1)
        {
            printf("Err: sigaltstack error\n");
            exit(EXIT_FAILURE);
        }
        struct sigaction act;
        act.sa_flags = SA_ONSTACK;
        sigemptyset(&act.sa_mask);
        act.sa_handler = handler;
        sigaction(SIGSEGV, &act, NULL);

        do
        {
            char a[counter];
            counter += 8964;
        } while (1);

        return 0;
    }

【问题讨论】：

您可以在信号处理程序中_exit() 打破循环
然后我需要打破这个循环，这样我才能计算大小。我不想终止我的程序。之后我需要恢复。这就是为什么我需要打破这个循环。
在循环之前分叉，然后通过管道或类似的东西将结果传输到主程序。您也可以尝试使用setjmp/longjmp 进行恢复，但我会尽量避免这种情况......
天哪，我怎么没想到。谢谢。

标签： c signals

【解决方案1】：

我已经修改了你的程序来计算堆栈大小的估计值，然后使用 siglongjmp 从信号处理程序返回到main 中可能安全的位置，就在分配变长数组。（请注意，分配可变长度数组的方式取决于编译器，并且在 main 溢出堆栈后可能没有安全的方式来恢复执行）。

信号处理程序中的 printf 仅用于调试；它们可以被删除而不影响程序的其余部分。

#include <stdio.h>
#include <stddef.h>
#include <stdlib.h>
#include <signal.h>
#include <setjmp.h>

ptrdiff_t s_size_computed;
void *s_base_estimated;
sigjmp_buf env;

void handler (int signo, siginfo_t *info, void *ucontext)
{
        printf("Caught signal %d\n", signo);
        if (signo == SIGSEGV)
        {
                printf("faulting address %p\n", info->si_addr);
                s_size_computed = (char *)s_base_estimated - (char *)info->si_addr;
                siglongjmp(env, 1);
        }
}

int main()
{
        int counter = 8964;
        stack_t sigstack;
        sigstack.ss_sp = malloc(SIGSTKSZ);
        if (sigstack.ss_sp == NULL)
        {
                printf("Err: malloc error\n");
                exit(EXIT_FAILURE);
        }
        sigstack.ss_size = SIGSTKSZ;
        sigstack.ss_flags = 0;

        if (sigaltstack(&sigstack, NULL) == -1)
        {
                printf("Err: sigaltstack error\n");
                exit(EXIT_FAILURE);
        }
        s_base_estimated = &counter;
        printf("&locals %p\n", s_base_estimated);
        struct sigaction act;
        act.sa_flags = SA_ONSTACK|SA_SIGINFO;
        sigemptyset(&act.sa_mask);
        act.sa_sigaction = handler;
        sigaction(SIGSEGV, &act, NULL);
        if (sigsetjmp(env, 0) == 1)
        {
                printf("computed stack size %td\n", s_size_computed);
                exit(1);
        }
        do
        {
                char a[counter];
                counter += 8964;
                a[0] = 0;         /* access element closest to top of stack (x86) */
        } while (1);

        return 0;
}

这是我的 Ubuntu 系统上的输出：

mp:~$ ./stack
&locals 0x7fffeb60e0dc
Caught signal 11
faulting address 0x7fffeae0d960
computed stack size 8390524
mp:~$ ulimit -s
8192

【讨论】：