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