在c中捕获ctrl-c并继续执行

时间:2018-10-07 23:19:36

标签: c shell signals

我正在用c语言编写一个简单的shell程序,需要处理ctrl-c。

如果前台进程正在运行,我需要终止它并继续主shell循环。如果没有,我什么都不做,只需要打印出信号已被捕获即可。

以下是我的代码,基于以下线程:Catch Ctrl-C in C

void inthandler(int dummy){
    signal(dummy, SIG_IGN);
    printf("ctrl-c caught\n");
}

然后我在进入主循环之前立即调用signal()

int main(int argc, char*argv[]){
    signal(SIGINT, inthandler)
    while(true){
        //main loop
    }
}

到目前为止,我已经能够拦截ctrl-c并打印我想要的消息,但是任何进一步的输入都将导致段错误。

输入inthandler后如何返回主循环执行?

1 个答案:

答案 0 :(得分:0)

  1. 使用sigaction(),而不是signal(),除非将处置方式设置为SIG_DFLSIG_IGN。尽管signal()是在C中指定的,而sigaction()是POSIX.1的内容,但是您仍然需要POSIX才能对信号进行有意义的处理。

  2. 仅在信号处理程序中使用async-signal safe函数。除了标准I / O(在<stdio.h>中声明),您可以对基础文件描述符使用POSIX低级I / O(read()write())。不过,您确实需要避免对使用那些相同的基础描述符的流使用标准I / O,否则输出可能会由于在标准I / O中进行缓冲而出现乱码。

  3. 如果将信号处理程序中的信号设置更改为忽略,则只会捕获(每种捕获类型的)第一个信号。

请考虑以下示例程序:

#define _POSIX_C_SOURCE 200809L
#include <stdlib.h>
#include <unistd.h>
#include <signal.h>
#include <string.h>
#include <stdio.h>
#include <errno.h>

/* Helper function: Write a string to a descriptor, keeping errno unchanged. 
   Returns 0 if success, errno error code if an error occurs. */
static inline int  wrs(const int fd, const char *s)
{
    /* Invalid descriptor? */
    if (fd == -1)
        return EBADF;

    /* Anything to actually write? */
    if (s && *s) {
        const int   saved_errno = errno;
        const char *z = s;
        ssize_t     n;
        int         err = 0;

        /* Use a loop to find end of s, since strlen() is not async-signal safe. */
        while (*z)
            z++;

        /* Write the string, ignoring EINTR, and allowing short writes. */
        while (s < z) {
            n = write(fd, s, (size_t)(z - s));
            if (n > 0)
                s += n;
            else
            if (n != -1) {
                /* This should never occur, so it's an I/O error. */
                err = EIO;
                break;
            } else
            if (errno != EINTR) {
                /* An actual error occurred. */
                err = errno;
                break;
            }
        }

        errno = saved_errno;
        return err;

    } else {
        /* Nothing to write. NULL s is silently ignored without error. */
        return 0;
    }
}

/* Signal handler. Just outputs a line to standard error. */
void catcher(int signum)
{
    switch (signum) {
    case SIGINT:
        wrs(STDERR_FILENO, "Caught INT signal.\n");
        return;
    default:
        wrs(STDERR_FILENO, "Caught a signal.\n");
        return;
    }
}

/* Helper function to install the signal handler. */
int install_catcher(const int signum)
{
    struct sigaction  act;

    memset(&act, 0, sizeof act);
    sigemptyset(&act.sa_mask);

    act.sa_handler = catcher;
    act.sa_flags = SA_RESTART;  /* Do not interrupt "slow" functions */
    if (sigaction(signum, &act, NULL) == -1)
        return -1;  /* Failed */

    return 0;
}

/* Helper function to remove embedded NUL characters and CRs and LFs,
   as well as leading and trailing whitespace.  Assumes data[size] is writable. */
size_t clean(char *data, size_t size)
{
    char *const  end = data + size;
    char        *src = data;
    char        *dst = data;

    /* Skip leading ASCII whitespace. */
    while (src < end && (*src == '\t' || *src == '\n' || *src == '\v' ||
                         *src == '\f' || *src == '\r' || *src == ' '))
        src++;

    /* Copy all but NUL, CR, and LF chars. */
    while (src < end)
        if (*src != '\0' && *src != '\n' && *src != '\r')
            *(dst++) = *(src++);
        else
            src++;

    /* Backtrack trailing ASCII whitespace. */
    while (dst > data && (dst[-1] == '\t' || dst[-1] == '\n' || dst[-1] == '\v' ||
                          dst[-1] == '\n' || dst[-1] == '\r' || dst[-1] == ' '))
        dst--;

    /* Mark end of string. */
    *dst = '\0';

    /* Return new length. */
    return (size_t)(dst - data);
}


int main(void)
{
    char   *line = NULL;
    size_t  size = 0;
    ssize_t len;

    if (install_catcher(SIGINT)) {
        fprintf(stderr, "Cannot install signal handler: %s.\n", strerror(errno));
        return EXIT_FAILURE;
    }

    printf("Type Ctrl+D, 'exit', or 'quit' on an empty line to exit.\n");

    while (1) {
        len = getline(&line, &size, stdin);
        if (len < 0)
            break;

        clean(line, len);
        printf("Read %zd chars: %s\n", len, line);

        if (!strcmp(line, "exit") || !strcmp(line, "quit"))
            break;
    }

    return EXIT_SUCCESS;
}

在大多数POSIXy系统上,伪终端中的 Ctrl + C 也会清除当前行缓冲区,因此在以交互方式提供行的中间按下它会丢弃该行(数据未发送到进程)。

请注意,当您按下 Ctrl + C 时,通常会在伪终端中看到的^C是受ECHO { {3}}设置。该设置还控制一般是否在终端上回显按键。

如果将signal(SIGINT, SIG_IGN)添加到catcher(),则在wrs()行之后,只有第一个 Ctrl + C 会打印“捕获到的INT信号“;以下任何 Ctrl + C 都将丢弃不完整的行。

因此,如果您输入Foo Ctrl + C Bar Enter ,要么看

Foo^CCaught INT signal.
Bar
Read 4 chars: Bar

Foo^CBar
Read 4 chars: Bar

取决于处理程序是捕获还是忽略了 Ctrl + C 生成的INT信号。

要退出,请在行首或在 Ctrl + C 之后立即输入exitquit

这里没有分段错误,因此,如果您的代码确实生成了一个分段错误,则一定是由于程序中的错误所致。


  

输入inthandler后如何返回主循环执行?

信号传送中断在执行信号处理程序的过程中中断执行;然后,被中断的代码继续执行。因此,对该问题的严格正确答案是从信号处理程序返回

如果在安装了SA_RESTART标志的情况下安装了信号处理程序,则被中断的代码应继续进行,好像什么都没发生一样。如果安装的信号处理程序没有该标志,则中断“慢速”系统调用可能会返回EINTR错误。

errno必须在信号处理程序中保持不变的原因-这是许多很多程序员忽略的错误-原因是,如果信号处理程序中的操作更改errno,并且信号处理程序被调用在系统或C库调用失败并显示错误代码后,代码所看到的errno将是错误的!当然,这是一种罕见的情况(对于每个系统或C库调用而言,发生这种情况的时间窗口很小),但这是可能发生的实际错误。当确实发生这种情况时,正是这种Heisenbug导致开发人员扯开头发,赤身裸体地奔跑,并且通常变得比现在更疯狂。

还要注意,stderr仅在安装信号处理程序失败的代码路径中使用,因为我想确保不要将I / O和POSIX低级I / O混合到标准错误中。 / p>