tinywebd

view src/main.c @ 9:0244b08cc9d3

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