nuclear@12: /* tinyweb - tiny web server library and daemon
nuclear@12:  * Author: John Tsiombikas <nuclear@member.fsf.org>
nuclear@12:  *
nuclear@12:  * This program is placed in the public domain. Feel free to use it any
nuclear@12:  * way you like. Mentions and retaining this attribution header will be
nuclear@12:  * appreciated, but not required.
nuclear@12:  */
nuclear@0: #include <stdio.h>
nuclear@0: #include <stdlib.h>
nuclear@0: #include <string.h>
nuclear@7: #include <errno.h>
nuclear@3: #include <signal.h>
nuclear@7: #include "tinyweb.h"
nuclear@1: 
nuclear@7: int parse_args(int argc, char **argv);
nuclear@3: void sighandler(int s);
nuclear@5: 
nuclear@5: 
nuclear@0: int main(int argc, char **argv)
nuclear@0: {
nuclear@7: 	int *sockets, num_sockets, sockets_arr_size = 0;
nuclear@7: 
nuclear@1: 	if(parse_args(argc, argv) == -1) {
nuclear@1: 		return 1;
nuclear@1: 	}
nuclear@1: 
nuclear@3: 	signal(SIGINT, sighandler);
nuclear@3: 	signal(SIGTERM, sighandler);
nuclear@3: 	signal(SIGQUIT, sighandler);
nuclear@3: 
nuclear@9: 	if(tw_start() == -1) {
nuclear@9: 		return 1;
nuclear@9: 	}
nuclear@0: 
nuclear@0: 	for(;;) {
nuclear@7: 		int i;
nuclear@0: 		fd_set rdset;
nuclear@0: 
nuclear@7: 		num_sockets = tw_get_sockets(0);
nuclear@7: 		if(num_sockets > sockets_arr_size) {
nuclear@7: 			int newsz = sockets_arr_size ? sockets_arr_size * 2 : 16;
nuclear@7: 			int *newarr = realloc(sockets, newsz * sizeof *sockets);
nuclear@7: 			if(!newarr) {
nuclear@7: 				fprintf(stderr, "failed to allocate sockets array\n");
nuclear@7: 				tw_stop();
nuclear@7: 				return 1;
nuclear@7: 			}
nuclear@7: 			sockets = newarr;
nuclear@7: 			sockets_arr_size = newsz;
nuclear@7: 		}
nuclear@7: 		tw_get_sockets(sockets);
nuclear@7: 
nuclear@0: 		FD_ZERO(&rdset);
nuclear@7: 		for(i=0; i<num_sockets; i++) {
nuclear@7: 			FD_SET(sockets[i], &rdset);
nuclear@0: 		}
nuclear@0: 
nuclear@7: 		while(select(tw_get_maxfd() + 1, &rdset, 0, 0, 0) == -1 && errno == EINTR);
nuclear@0: 
nuclear@7: 		for(i=0; i<num_sockets; i++) {
nuclear@7: 			if(FD_ISSET(sockets[i], &rdset)) {
nuclear@7: 				tw_handle_socket(sockets[i]);
nuclear@0: 			}
nuclear@0: 		}
nuclear@0: 	}
nuclear@3: 
nuclear@3: 	return 0;	/* unreachable */
nuclear@0: }
nuclear@0: 
nuclear@3: void sighandler(int s)
nuclear@3: {
nuclear@3: 	if(s == SIGINT || s == SIGTERM || s == SIGQUIT) {
nuclear@7: 		tw_stop();
nuclear@3: 		printf("bye!\n");
nuclear@3: 		exit(0);
nuclear@3: 	}
nuclear@3: }
nuclear@3: 
nuclear@1: 
nuclear@1: static void print_help(const char *argv0)
nuclear@1: {
nuclear@1: 	printf("Usage: %s [options]\n", argv0);
nuclear@1: 	printf("Options:\n");
nuclear@1: 	printf(" -p <port>  set the TCP/IP port number to use\n");
nuclear@1: 	printf(" -h         print usage help and exit\n");
nuclear@1: }
nuclear@1: 
nuclear@1: int parse_args(int argc, char **argv)
nuclear@1: {
nuclear@1: 	int i;
nuclear@1: 
nuclear@1: 	for(i=1; i<argc; i++) {
nuclear@1: 		if(argv[i][0] == '-' && argv[i][2] == 0) {
nuclear@1: 			switch(argv[i][1]) {
nuclear@1: 			case 'p':
nuclear@7: 				{
nuclear@7: 					int port = atoi(argv[++i]);
nuclear@7: 					if(!port) {
nuclear@7: 						fprintf(stderr, "-p must be followed by a valid port number\n");
nuclear@7: 						return -1;
nuclear@7: 					}
nuclear@7: 					tw_set_port(port);
nuclear@1: 				}
nuclear@1: 				break;
nuclear@1: 
nuclear@8: 			case 'c':
nuclear@8: 				if(tw_set_root(argv[++i]) == -1) {
nuclear@8: 					return -1;
nuclear@8: 				}
nuclear@8: 				break;
nuclear@8: 
nuclear@1: 			case 'h':
nuclear@1: 				print_help(argv[0]);
nuclear@1: 				exit(0);
nuclear@1: 
nuclear@1: 			default:
nuclear@1: 				fprintf(stderr, "unrecognized option: %s\n", argv[i]);
nuclear@1: 				return -1;
nuclear@1: 			}
nuclear@1: 		} else {
nuclear@1: 			fprintf(stderr, "unexpected argument: %s\n", argv[i]);
nuclear@1: 			return -1;
nuclear@1: 		}
nuclear@0: 	}
nuclear@0: 	return 0;
nuclear@0: }