int main(int argc, char *argv[])
{
struct sigaction act, old;
struct addrinfo ask, *res;
struct sockaddr_storage ss;
int c, opt, rc, ss_len;
char host[INET6_ADDRSTRLEN + 1];
char serv[16];
const char options[] = "hd" "p:";
memset(&ask, 0, sizeof(ask));
/* parse options */
while(1) {
if (-1 == (c = getopt(argc, argv, options))) {
break;
}
switch (c) {
case 'h':
usage(argv[0]);
break;
case 'p':
listen_port = optarg;
break;
default:
exit(1);
}
}
/* bind to socket */
memset(&ask, 0, sizeof(ask));
ask.ai_flags = AI_PASSIVE;
if (listen_ip) {
ask.ai_flags |= AI_CANONNAME;
}
ask.ai_socktype = SOCK_STREAM;
ask.ai_family = PF_INET;
if (0 != (rc = getaddrinfo(listen_ip, listen_port, &ask, &res))) {
fprintf(stderr, "getaddrinfo (ipv4): %s\n", gai_strerror(rc));
exit(1);
}
if (-1 == (slisten = socket(res->ai_family, res->ai_socktype, res->ai_protocol))) {
exit(1);
}
if (-1 == slisten) {
exit(1);
}
close_on_exec(slisten);
memcpy(&ss, res->ai_addr, res->ai_addrlen);
ss_len = res->ai_addrlen;
if (0 != (rc = getnameinfo((struct sockaddr *)&ss, ss_len,
host, INET6_ADDRSTRLEN, serv, 15,
NI_NUMERICHOST | NI_NUMERICSERV))) {
exit(1);
}
tcp_port = atoi(serv);
opt = 1;
setsockopt(slisten, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
fcntl(slisten, F_SETFL, O_NONBLOCK);
if (-1 == bind(slisten, (struct sockaddr *) &ss, ss_len)) {
exit(1);
}
if (-1 == listen(slisten, 2 * max_conn)) {
exit(1);
}
init_quote();
printf("gx start!\n\n");
#if defined(linux)
printf("###############################\n");
printf("HOST INFO:\n");
host_info();
#endif
printf("###############################\n");
printf("SERVER INFO:\n");
/* setup signal handler */
memset(&act, 0, sizeof(act));
sigemptyset(&act.sa_mask);
act.sa_handler = SIG_IGN;
sigaction(SIGPIPE, &act, &old);
sigaction(SIGCHLD, &act, &old);
act.sa_handler = catchsig;
sigaction(SIGHUP, &act, &old);
sigaction(SIGTERM, &act, &old);
sigaction(SIGINT, &act, &old);
/* go! */
if (nthreads > 1) {
int i;
threads = malloc(sizeof(pthread_t) * nthreads);
for (i = 1; i < nthreads; i++) {
pthread_create(threads + i, NULL, mainloop, threads + i);
pthread_detach(threads[i]);
}
}
mainloop(NULL);
fprintf(stderr, "bye...\n");
exit(0);
}
/* return file descriptor if success, <0 otherwise */
int httpd_init(int sport, char *sname, int use_ssl, const char *certificate,
const char *password, int use_v6)
{
struct addrinfo ask,*res = NULL;
struct sockaddr_storage ss;
int opt, rc, ss_len, v4 = 1, v6 = 0;
char host[INET6_ADDRSTRLEN+1];
char serv[16];
char listen_port[6];
char *listen_ip = NULL;
/* set config */
sprintf(listen_port, "%d", sport);
server_name = sname;
if (use_v6) {
v6 = 1;
}
#ifdef USE_SSL
with_ssl = use_ssl;
#endif
/* FIXME nthreads */
/* get server name */
gethostname(server_host,255);
memset(&ask,0,sizeof(ask));
ask.ai_flags = AI_CANONNAME;
if ((rc = getaddrinfo(server_host, NULL, &ask, &res)) == 0) {
if (res->ai_canonname) {
strcpy(server_host,res->ai_canonname);
}
if (res != NULL) {
freeaddrinfo(res);
res = NULL;
}
}
/* bind to socket */
slisten = -1;
memset(&ask, 0, sizeof(ask));
ask.ai_flags = AI_PASSIVE;
if (listen_ip) {
ask.ai_flags |= AI_CANONNAME;
}
ask.ai_socktype = SOCK_STREAM;
/* try ipv6 first ... */
if (slisten == -1 && v6) {
ask.ai_family = PF_INET6;
g_timeout = 0;
alarm(2);
rc = getaddrinfo(listen_ip, listen_port, &ask, &res);
if (g_timeout) {
log_error_func(1, LOG_ERR,"getaddrinfo (ipv6): DNS timeout",NULL);
}
alarm(0);
if (rc == 0) {
if ((slisten = socket(res->ai_family, res->ai_socktype,
res->ai_protocol)) == -1) {
log_error_func(1, LOG_ERR,"socket (ipv6)",NULL);
}
} else {
log_error_func(1, LOG_ERR, "getaddrinfo (ipv6)", NULL);
}
}
g_timeout = 0;
alarm(2);
/* ... failing that try ipv4 */
if (slisten == -1 && v4) {
ask.ai_family = PF_INET;
g_timeout = 0;
alarm(1);
rc = getaddrinfo(listen_ip, listen_port, &ask, &res);
if (g_timeout) {
log_error_func(1, LOG_ERR,"getaddrinfo (ipv4): DNS timeout",NULL);
return -1;
}
alarm(0);
if (rc == 0) {
if ((slisten = socket(res->ai_family, res->ai_socktype,
res->ai_protocol)) == -1) {
log_error_func(1, LOG_ERR,"socket (ipv4)",NULL);
return -1;
}
} else {
log_error_func(1, LOG_ERR, "getaddrinfo (ipv4)", NULL);
return -1;
}
}
if (slisten == -1) {
return -1;
}
memcpy(&ss,res->ai_addr,res->ai_addrlen);
ss_len = res->ai_addrlen;
if (res->ai_canonname) {
strcpy(server_host,res->ai_canonname);
}
if ((rc = getnameinfo((struct sockaddr*)&ss,ss_len,
host,INET6_ADDRSTRLEN,serv,15,
NI_NUMERICHOST | NI_NUMERICSERV)) != 0) {
log_error_func(1, LOG_ERR, "getnameinfo", NULL);
return -1;
}
tcp_port = atoi(serv);
opt = 1;
setsockopt(slisten,SOL_SOCKET,SO_REUSEADDR,&opt,sizeof(opt));
fcntl(slisten,F_SETFL,O_NONBLOCK);
/* Use accept filtering, if available. */
#ifdef SO_ACCEPTFILTER
{
struct accept_filter_arg af;
memset(&af,0,sizeof(af));
strcpy(af.af_name,"httpready");
setsockopt(slisten, SOL_SOCKET, SO_ACCEPTFILTER, (char*)&af, sizeof(af));
}
#endif /* SO_ACCEPTFILTER */
if (bind(slisten, (struct sockaddr*) &ss, ss_len) == -1) {
log_error_func(1, LOG_ERR,"bind",NULL);
return -1;
}
if (listen(slisten, 2*max_conn) == -1) {
log_error_func(1, LOG_ERR,"listen",NULL);
return -1;
}
/* init misc stuff */
init_mime(mimetypes,"text/plain");
init_quote();
#ifdef USE_SSL
if (with_ssl) {
init_ssl(certificate, password);
}
#endif
#ifdef DEBUG
fprintf(stderr,
"http server started\n"
" ipv6 : %s\n"
#ifdef USE_SSL
" ssl : %s\n"
#endif
" node : %s\n"
" ipaddr: %s\n"
" port : %d\n"
,
res->ai_family == PF_INET6 ? "yes" : "no",
#ifdef USE_SSL
with_ssl ? "yes" : "no",
#endif
server_host,host,tcp_port);
#endif
if (res != NULL) {
freeaddrinfo(res);
}
/* go! */
#ifdef HTTPD_USE_THREADS
if (nthreads > 1) {
int i;
threads = malloc(sizeof(pthread_t) * nthreads);
for (i = 1; i < nthreads; i++) {
pthread_create(threads+i,NULL,mainloop,threads+i);
pthread_detach(threads[i]);
}
}
#endif
return slisten;
}
