使用信号中断主线程

Question

我正在用C语言编写服务器/客户端TCP并发程序。每次客户端连接时，服务器的主线程都会创建一个新线程来处理该客户端。之后，主线程返回到accept（），并保持阻塞状态，直到另一个客户端连接为止。

但是，我希望客户端能够关闭服务器。为此，我假设我必须使用信号。我该如何处理。我对信号很陌生。我不知道如何向特定线程发送信号，或者在这里如何使用它们。

我希望客户端写'..'并关闭服务器。

有什么想法吗？

谢谢

Answer 1

尽管我理解了您的问题，但令我感到困惑的是，您在哪里使用这样的服务器？无论如何，我希望您能充分利用此服务器。

#include <signal.h>
#include <sys/socket.h>
#include <pthread.h>
#include <netinet/in.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <stdio.h>

void sigfunc(int signo)
{
    _exit(0); // safe to call in signal handler since it is async-signal-safe func
}

struct client_data {
    int clifd;
    pthread_t ptid;
};

void* client_func(void* _cli_data)
{
    struct client_data* cli_data = (struct client_data*) _cli_data;

    char buf[100];

    for( ; ; ) {
        int rd_status = read(cli_data->clifd, buf, sizeof(buf) - 1);

        if (rd_status <= 0)
            break;

        buf[rd_status] = '\0';
        printf("%s", buf);
        fflush(stdout);
    }

    close(cli_data->clifd);

    pthread_kill(cli_data->ptid, SIGRTMIN);
    return NULL;
}

int main()
{
    /* Setup a Listening server */

    struct sockaddr_in servaddr;
    memset(&servaddr, 0, sizeof(struct sockaddr_in));

    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port = htons(8008);

    int sockfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

    if (bind(sockfd, (struct sockaddr*) &servaddr, sizeof(struct sockaddr_in)) != 0)
        exit(1);

    listen(sockfd, 5);

    /* Setup signal handler, use SIGRTMIN as sync signal */

    struct sigaction act;
    sigemptyset(&act.sa_mask);
    act.sa_flags = 0;

#ifdef SA_INTERRUPT
    act.sa_flags |= SA_INTERRUPT;
#endif

    act.sa_handler = sigfunc;

    sigaction(SIGRTMIN, &act, NULL);

    /* Loop and listen for connections */

    int addr_len;

    for ( ; ; ) {
        struct client_data cli_data;

        cli_data.ptid = pthread_self();
        addr_len = sizeof(struct sockaddr_in);
        cli_data.clifd = accept(sockfd, (struct sockaddr*) &servaddr, &addr_len);

        if (cli_data.clifd < 0)
            continue;

        pthread_t tid;
        pthread_create(&tid, NULL, client_func, &cli_data);
    }
}

终端会议：

服务器

$ gcc SO.c -lpthread
$ ./a.out 
Hi to the server
Bye to the server
$

客户

$ netstat -l | grep 8008 
tcp        0      0 0.0.0.0:8008            0.0.0.0:*               LISTEN     
$ nc 127.0.0.1 8008
Hi to the server
Bye to the server
^C
$

该代码已注释，您可以进行遍历。正如@MartinJames所说的那样，您不必在绝对不必要的地方终止线程，而最好依靠OS例程。由于我们以异步方式处理信号，因此只能在信号处理程序中调用几个函数（异步信号安全函数）； _exit()是其中之一。