tinywebd

view src/main.c @ 8:121b991ccc1d

find changesets by author, revision, files, or words in the commit message
fixed the SIGPIPE bug
author John Tsiombikas <nuclear@member.fsf.org>
date Fri, 17 Apr 2015 12:02:59 +0300
parents 5ec50ca0d071
children 0244b08cc9d3
line source
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <string.h>
4 #include <errno.h>
5 #include <signal.h>
6 #include "tinyweb.h"
7
8 int parse_args(int argc, char **argv);
9 void sighandler(int s);
10
11
12 int main(int argc, char **argv)
13 {
14 int *sockets, num_sockets, sockets_arr_size = 0;
15
16 if(parse_args(argc, argv) == -1) {
17 return 1;
18 }
19
20 signal(SIGINT, sighandler);
21 signal(SIGTERM, sighandler);
22 signal(SIGQUIT, sighandler);
23
24 tw_start();
25
26 for(;;) {
27 int i;
28 fd_set rdset;
29
30 num_sockets = tw_get_sockets(0);
31 if(num_sockets > sockets_arr_size) {
32 int newsz = sockets_arr_size ? sockets_arr_size * 2 : 16;
33 int *newarr = realloc(sockets, newsz * sizeof *sockets);
34 if(!newarr) {
35 fprintf(stderr, "failed to allocate sockets array\n");
36 tw_stop();
37 return 1;
38 }
39 sockets = newarr;
40 sockets_arr_size = newsz;
41 }
42 tw_get_sockets(sockets);
43
44 FD_ZERO(&rdset);
45 for(i=0; i<num_sockets; i++) {
46 FD_SET(sockets[i], &rdset);
47 }
48
49 while(select(tw_get_maxfd() + 1, &rdset, 0, 0, 0) == -1 && errno == EINTR);
50
51 for(i=0; i<num_sockets; i++) {
52 if(FD_ISSET(sockets[i], &rdset)) {
53 tw_handle_socket(sockets[i]);
54 }
55 }
56 }
57
58 return 0; /* unreachable */
59 }
60
61 void sighandler(int s)
62 {
63 if(s == SIGINT || s == SIGTERM || s == SIGQUIT) {
64 tw_stop();
65 printf("bye!\n");
66 exit(0);
67 }
68 }
69
70
71 static void print_help(const char *argv0)
72 {
73 printf("Usage: %s [options]\n", argv0);
74 printf("Options:\n");
75 printf(" -p <port> set the TCP/IP port number to use\n");
76 printf(" -h print usage help and exit\n");
77 }
78
79 int parse_args(int argc, char **argv)
80 {
81 int i;
82
83 for(i=1; i<argc; i++) {
84 if(argv[i][0] == '-' && argv[i][2] == 0) {
85 switch(argv[i][1]) {
86 case 'p':
87 {
88 int port = atoi(argv[++i]);
89 if(!port) {
90 fprintf(stderr, "-p must be followed by a valid port number\n");
91 return -1;
92 }
93 tw_set_port(port);
94 }
95 break;
96
97 case 'c':
98 if(tw_set_root(argv[++i]) == -1) {
99 return -1;
100 }
101 break;
102
103 case 'h':
104 print_help(argv[0]);
105 exit(0);
106
107 default:
108 fprintf(stderr, "unrecognized option: %s\n", argv[i]);
109 return -1;
110 }
111 } else {
112 fprintf(stderr, "unexpected argument: %s\n", argv[i]);
113 return -1;
114 }
115 }
116 return 0;
117 }