summaryrefslogtreecommitdiffstats
path: root/bin/nserver.c
blob: d2452be64b3f01ad1bf880fe464264e570e3e9e0 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
/* SPDX-License-Identifier: BSD-3-Clause */
/*
 * Copyright © 2010, Zed A. Shaw.
 * Copyright © 2020 rsiddharth <s@ricketyspace.net>
 */

#include <nserve.h>
#include <nsocket.h>
#include <signal.h>
#include <sys/wait.h>

#define BACKLOG 10

void handle_sigchild(int sig)
{
    sig = 0; // ignore it
    while(waitpid(-1, NULL, WNOHANG) > 0) {
    }
}

int setup_sigaction()
{
    struct sigaction sa = {
         .sa_handler = handle_sigchild,
         .sa_flags = SA_RESTART | SA_NOCLDSTOP
    };
    sigemptyset(&sa.sa_mask);
    int rc = sigaction(SIGCHLD, &sa, 0);
    check(rc != -1, "signal handler setup failed.");

    return 0;
 error:
    return -1;
}

int main(void)
{
    int rc = 0;
    int sockfd_s = 0, sockfd_c = 0;
    struct sockaddr sockaddr_c;
    socklen_t sockaddr_c_len;
    pid_t pidc;

    // create a sigaction that handles SIGCHLD
    rc = setup_sigaction();
    check(rc != -1, "sigaction setup failed");

    sockfd_s = get_socket();
    check(sockfd_s > 0, "nserver: unable to get socket");

    rc = listen(sockfd_s, BACKLOG);
    check(rc == 0, "nserver: listen failed");

    do {
        printf("Waiting for connection...\n");
        sockfd_c = accept(sockfd_s, &sockaddr_c, &sockaddr_c_len);

        pidc = fork();
        check(pidc != -1, "nserver: fork failed");

        if (pidc == 0) {
            close(sockfd_s);
            nserve(sockfd_c);
        }

        rc = close(sockfd_c);
        check(rc == 0, "nserver: close failed");
    } while(1);

    rc = close(sockfd_s);
    check(rc == 0, "nserver: close failed");

    return 0;

 error:
    if (sockfd_s) {
        rc = close(sockfd_s);
    }
    if (sockfd_c) {
        rc = close(sockfd_c);
    }

    return -1;
}