[ashd.git] / src / plaintcp.c

/*
    ashd - A Sane HTTP Daemon
    Copyright (C) 2008  Fredrik Tolf <fredrik@dolda2000.com>

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#include <stdlib.h>
#include <unistd.h>
#include <stdio.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <errno.h>
#include <string.h>

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <utils.h>
#include <req.h>
#include <mt.h>
#include <mtio.h>
#include <log.h>

#include "htparser.h"

struct tcpport {
    int fd;
    int sport;
};

struct tcpconn {
    struct sockaddr_storage name;
    struct tcpport *port;
    int fd;
};

int listensock4(int port)
{
    struct sockaddr_in name;
    int fd;
    int valbuf;
    
    memset(&name, 0, sizeof(name));
    name.sin_family = AF_INET;
    name.sin_port = htons(port);
    if((fd = socket(PF_INET, SOCK_STREAM, 0)) < 0)
	return(-1);
    valbuf = 1;
    setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &valbuf, sizeof(valbuf));
    if(bind(fd, (struct sockaddr *)&name, sizeof(name))) {
	close(fd);
	return(-1);
    }
    if(listen(fd, 128) < 0) {
	close(fd);
	return(-1);
    }
    fcntl(fd, F_SETFD, FD_CLOEXEC);
    return(fd);
}

int listensock6(int port)
{
    struct sockaddr_in6 name;
    int fd;
    int valbuf;
    
    memset(&name, 0, sizeof(name));
    name.sin6_family = AF_INET6;
    name.sin6_port = htons(port);
    if((fd = socket(PF_INET6, SOCK_STREAM, 0)) < 0)
	return(-1);
    valbuf = 1;
    setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &valbuf, sizeof(valbuf));
    if(bind(fd, (struct sockaddr *)&name, sizeof(name))) {
	close(fd);
	return(-1);
    }
    if(listen(fd, 128) < 0) {
	close(fd);
	return(-1);
    }
    fcntl(fd, F_SETFD, FD_CLOEXEC);
    return(fd);
}

char *formathaddress(struct sockaddr *name, socklen_t namelen)
{
    static char buf[128];
    struct sockaddr_in *v4;
    struct sockaddr_in6 *v6;

    switch(name->sa_family) {
    case AF_INET:
	v4 = (struct sockaddr_in *)name;
	if(!inet_ntop(AF_INET, &v4->sin_addr, buf, sizeof(buf)))
	    return(NULL);
	return(buf);
    case AF_INET6:
	v6 = (struct sockaddr_in6 *)name;
	if(IN6_IS_ADDR_V4MAPPED(&v6->sin6_addr)) {
	    if(!inet_ntop(AF_INET, ((char *)&v6->sin6_addr) + 12, buf, sizeof(buf)))
		return(NULL);
	} else {
	    if(!inet_ntop(AF_INET6, &v6->sin6_addr, buf, sizeof(buf)))
		return(NULL);
	}
	return(buf);
    default:
	errno = EPFNOSUPPORT;
	return(NULL);
    }
}

static int initreq(struct conn *conn, struct hthead *req)
{
    struct tcpconn *tcp = conn->pdata;
    struct sockaddr_storage sa;
    socklen_t salen;
    
    headappheader(req, "X-Ash-Address", formathaddress((struct sockaddr *)&tcp->name, sizeof(sa)));
    if(tcp->name.ss_family == AF_INET)
	headappheader(req, "X-Ash-Port", sprintf3("%i", ntohs(((struct sockaddr_in *)&tcp->name)->sin_port)));
    else if(tcp->name.ss_family == AF_INET6)
	headappheader(req, "X-Ash-Port", sprintf3("%i", ntohs(((struct sockaddr_in6 *)&tcp->name)->sin6_port)));
    salen = sizeof(sa);
    if(!getsockname(tcp->fd, (struct sockaddr *)&sa, &salen))
	headappheader(req, "X-Ash-Server-Address", formathaddress((struct sockaddr *)&sa, sizeof(sa)));
    headappheader(req, "X-Ash-Server-Port", sprintf3("%i", tcp->port->sport));
    headappheader(req, "X-Ash-Protocol", "http");
    return(0);
}

void servetcp(struct muth *muth, va_list args)
{
    vavar(int, fd);
    vavar(struct sockaddr_storage, name);
    vavar(struct tcpport *, stcp);
    struct bufio *in;
    struct conn conn;
    struct tcpconn tcp;
    
    memset(&conn, 0, sizeof(conn));
    memset(&tcp, 0, sizeof(tcp));
    in = mtbioopen(fd, 1, 60, "r+", NULL);
    conn.pdata = &tcp;
    conn.initreq = initreq;
    tcp.fd = fd;
    tcp.name = name;
    tcp.port = stcp;
    serve(in, fd, &conn);
}

static void listenloop(struct muth *muth, va_list args)
{
    vavar(struct tcpport *, tcp);
    int i, ns, n;
    struct sockaddr_storage name;
    socklen_t namelen;
    
    fcntl(tcp->fd, F_SETFL, fcntl(tcp->fd, F_GETFL) | O_NONBLOCK);
    while(1) {
	namelen = sizeof(name);
	if(block(tcp->fd, EV_READ, 0) == 0)
	    goto out;
	n = 0;
	while(1) {
	    ns = accept(tcp->fd, (struct sockaddr *)&name, &namelen);
	    if(ns < 0) {
		if(errno == EAGAIN)
		    break;
		if(errno == ECONNABORTED)
		    continue;
		flog(LOG_ERR, "accept: %s", strerror(errno));
		goto out;
	    }
	    mustart(servetcp, ns, name, tcp);
	    if(++n >= 100)
		break;
	}
    }
    
out:
    close(tcp->fd);
    free(tcp);
    for(i = 0; i < listeners.d; i++) {
	if(listeners.b[i] == muth)
	    bufdel(listeners, i);
    }
}

void handleplain(int argc, char **argp, char **argv)
{
    int port, fd;
    int i;
    struct tcpport *tcp;
    
    port = 80;
    for(i = 0; i < argc; i++) {
	if(!strcmp(argp[i], "help")) {
	    printf("plain handler parameters:\n");
	    printf("\tport=TCP-PORT   [80]\n");
	    printf("\t\tThe TCP port to listen on.\n");
	    exit(0);
	} else if(!strcmp(argp[i], "port")) {
	    port = atoi(argv[i]);
	} else {
	    flog(LOG_ERR, "unknown parameter `%s' to plain handler", argp[i]);
	    exit(1);
	}
    }
    if((fd = listensock6(port)) < 0) {
	flog(LOG_ERR, "could not listen on IPv6 (port %i): %s", port, strerror(errno));
	exit(1);
    }
    omalloc(tcp);
    tcp->fd = fd;
    tcp->sport = port;
    bufadd(listeners, mustart(listenloop, tcp));
    if((fd = listensock4(port)) < 0) {
	if(errno != EADDRINUSE) {
	    flog(LOG_ERR, "could not listen on IPv4 (port %i): %s", port, strerror(errno));
	    exit(1);
	}
    } else {
	omalloc(tcp);
	tcp->fd = fd;
	tcp->sport = port;
	bufadd(listeners, mustart(listenloop, tcp));
    }
}
Commit	Line	Data
8774c31b FT	1	/*
	2	ashd - A Sane HTTP Daemon
	3	Copyright (C) 2008 Fredrik Tolf <fredrik@dolda2000.com>
	4
	5	This program is free software: you can redistribute it and/or modify
	6	it under the terms of the GNU General Public License as published by
	7	the Free Software Foundation, either version 3 of the License, or
	8	(at your option) any later version.
	9
	10	This program is distributed in the hope that it will be useful,
	11	but WITHOUT ANY WARRANTY; without even the implied warranty of
	12	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	13	GNU General Public License for more details.
	14
	15	You should have received a copy of the GNU General Public License
	16	along with this program. If not, see <http://www.gnu.org/licenses/>.
	17	*/
	18
	19	#include <stdlib.h>
	20	#include <unistd.h>
	21	#include <stdio.h>
	22	#include <sys/socket.h>
	23	#include <netinet/in.h>
	24	#include <arpa/inet.h>
470938bd	25	#include <fcntl.h>
8774c31b FT	26	#include <errno.h>
	27	#include <string.h>
	28
	29	#ifdef HAVE_CONFIG_H
	30	#include <config.h>
	31	#endif
	32	#include <utils.h>
	33	#include <req.h>
	34	#include <mt.h>
	35	#include <mtio.h>
	36	#include <log.h>
	37
	38	#include "htparser.h"
	39
c063c3b8 FT	40	struct tcpport {
	41	int fd;
	42	int sport;
	43	};
	44
8774c31b FT	45	struct tcpconn {
8774c31b FT	46	struct sockaddr_storage name;
c063c3b8	47	struct tcpport *port;
6b84641a	48	int fd;
8774c31b FT	49	};
8774c31b FT	50
6ca53b2e	51	int listensock4(int port)
8774c31b FT	52	{
	53	struct sockaddr_in name;
	54	int fd;
	55	int valbuf;
	56
	57	memset(&name, 0, sizeof(name));
	58	name.sin_family = AF_INET;
	59	name.sin_port = htons(port);
	60	if((fd = socket(PF_INET, SOCK_STREAM, 0)) < 0)
	61	return(-1);
	62	valbuf = 1;
	63	setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &valbuf, sizeof(valbuf));
	64	if(bind(fd, (struct sockaddr *)&name, sizeof(name))) {
	65	close(fd);
	66	return(-1);
	67	}
4ab547c3	68	if(listen(fd, 128) < 0) {
8774c31b FT	69	close(fd);
	70	return(-1);
	71	}
470938bd	72	fcntl(fd, F_SETFD, FD_CLOEXEC);
8774c31b FT	73	return(fd);
	74	}
	75
6ca53b2e	76	int listensock6(int port)
8774c31b FT	77	{
	78	struct sockaddr_in6 name;
	79	int fd;
	80	int valbuf;
	81
	82	memset(&name, 0, sizeof(name));
	83	name.sin6_family = AF_INET6;
	84	name.sin6_port = htons(port);
	85	if((fd = socket(PF_INET6, SOCK_STREAM, 0)) < 0)
	86	return(-1);
	87	valbuf = 1;
	88	setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &valbuf, sizeof(valbuf));
	89	if(bind(fd, (struct sockaddr *)&name, sizeof(name))) {
	90	close(fd);
	91	return(-1);
	92	}
4ab547c3	93	if(listen(fd, 128) < 0) {
8774c31b FT	94	close(fd);
	95	return(-1);
	96	}
470938bd	97	fcntl(fd, F_SETFD, FD_CLOEXEC);
8774c31b FT	98	return(fd);
	99	}
	100
7595e3a4 FT	101	char formathaddress(struct sockaddr name, socklen_t namelen)
	102	{
	103	static char buf[128];
	104	struct sockaddr_in *v4;
	105	struct sockaddr_in6 *v6;
	106
	107	switch(name->sa_family) {
	108	case AF_INET:
	109	v4 = (struct sockaddr_in *)name;
	110	if(!inet_ntop(AF_INET, &v4->sin_addr, buf, sizeof(buf)))
	111	return(NULL);
	112	return(buf);
	113	case AF_INET6:
	114	v6 = (struct sockaddr_in6 *)name;
	115	if(IN6_IS_ADDR_V4MAPPED(&v6->sin6_addr)) {
	116	if(!inet_ntop(AF_INET, ((char *)&v6->sin6_addr) + 12, buf, sizeof(buf)))
	117	return(NULL);
	118	} else {
	119	if(!inet_ntop(AF_INET6, &v6->sin6_addr, buf, sizeof(buf)))
	120	return(NULL);
	121	}
	122	return(buf);
	123	default:
	124	errno = EPFNOSUPPORT;
	125	return(NULL);
	126	}
	127	}
	128
8774c31b FT	129	static int initreq(struct conn conn, struct hthead req)
	130	{
	131	struct tcpconn *tcp = conn->pdata;
6b84641a FT	132	struct sockaddr_storage sa;
6b84641a FT	133	socklen_t salen;
8774c31b	134
7595e3a4 FT	135	headappheader(req, "X-Ash-Address", formathaddress((struct sockaddr *)&tcp->name, sizeof(sa)));
7595e3a4 FT	136	if(tcp->name.ss_family == AF_INET)
8774c31b	137	headappheader(req, "X-Ash-Port", sprintf3("%i", ntohs(((struct sockaddr_in *)&tcp->name)->sin_port)));
7595e3a4	138	else if(tcp->name.ss_family == AF_INET6)
8774c31b	139	headappheader(req, "X-Ash-Port", sprintf3("%i", ntohs(((struct sockaddr_in6 *)&tcp->name)->sin6_port)));
6b84641a	140	salen = sizeof(sa);
7595e3a4 FT	141	if(!getsockname(tcp->fd, (struct sockaddr *)&sa, &salen))
7595e3a4 FT	142	headappheader(req, "X-Ash-Server-Address", formathaddress((struct sockaddr *)&sa, sizeof(sa)));
c063c3b8	143	headappheader(req, "X-Ash-Server-Port", sprintf3("%i", tcp->port->sport));
46e66302	144	headappheader(req, "X-Ash-Protocol", "http");
8774c31b FT	145	return(0);
	146	}
	147
	148	void servetcp(struct muth *muth, va_list args)
	149	{
	150	vavar(int, fd);
	151	vavar(struct sockaddr_storage, name);
c063c3b8	152	vavar(struct tcpport *, stcp);
fc253e2f	153	struct bufio *in;
c063c3b8 FT	154	struct conn conn;
c063c3b8 FT	155	struct tcpconn tcp;
8774c31b	156
c063c3b8 FT	157	memset(&conn, 0, sizeof(conn));
c063c3b8 FT	158	memset(&tcp, 0, sizeof(tcp));
fc253e2f	159	in = mtbioopen(fd, 1, 60, "r+", NULL);
c063c3b8 FT	160	conn.pdata = &tcp;
c063c3b8 FT	161	conn.initreq = initreq;
6b84641a	162	tcp.fd = fd;
c063c3b8 FT	163	tcp.name = name;
c063c3b8 FT	164	tcp.port = stcp;
a68db17d	165	serve(in, fd, &conn);
8774c31b FT	166	}
	167
	168	static void listenloop(struct muth *muth, va_list args)
	169	{
c063c3b8	170	vavar(struct tcpport *, tcp);
f24b7bb5	171	int i, ns, n;
8774c31b FT	172	struct sockaddr_storage name;
	173	socklen_t namelen;
	174
f24b7bb5	175	fcntl(tcp->fd, F_SETFL, fcntl(tcp->fd, F_GETFL) \| O_NONBLOCK);
8774c31b FT	176	while(1) {
8774c31b FT	177	namelen = sizeof(name);
cac13158 FT	178	if(block(tcp->fd, EV_READ, 0) == 0)
cac13158 FT	179	goto out;
f24b7bb5 FT	180	n = 0;
	181	while(1) {
	182	ns = accept(tcp->fd, (struct sockaddr *)&name, &namelen);
	183	if(ns < 0) {
	184	if(errno == EAGAIN)
	185	break;
51893006 FT	186	if(errno == ECONNABORTED)
51893006 FT	187	continue;
f24b7bb5 FT	188	flog(LOG_ERR, "accept: %s", strerror(errno));
	189	goto out;
	190	}
	191	mustart(servetcp, ns, name, tcp);
	192	if(++n >= 100)
	193	break;
8774c31b	194	}
8774c31b FT	195	}
	196
	197	out:
c063c3b8 FT	198	close(tcp->fd);
c063c3b8 FT	199	free(tcp);
8e9ec020 FT	200	for(i = 0; i < listeners.d; i++) {
	201	if(listeners.b[i] == muth)
	202	bufdel(listeners, i);
	203	}
8774c31b FT	204	}
	205
	206	void handleplain(int argc, char argp, char argv)
	207	{
	208	int port, fd;
	209	int i;
c063c3b8	210	struct tcpport *tcp;
8774c31b FT	211
	212	port = 80;
	213	for(i = 0; i < argc; i++) {
	214	if(!strcmp(argp[i], "help")) {
	215	printf("plain handler parameters:\n");
6ca53b2e FT	216	printf("\tport=TCP-PORT [80]\n");
6ca53b2e FT	217	printf("\t\tThe TCP port to listen on.\n");
8774c31b FT	218	exit(0);
	219	} else if(!strcmp(argp[i], "port")) {
	220	port = atoi(argv[i]);
	221	} else {
	222	flog(LOG_ERR, "unknown parameter `%s' to plain handler", argp[i]);
	223	exit(1);
	224	}
	225	}
	226	if((fd = listensock6(port)) < 0) {
	227	flog(LOG_ERR, "could not listen on IPv6 (port %i): %s", port, strerror(errno));
	228	exit(1);
	229	}
c063c3b8 FT	230	omalloc(tcp);
	231	tcp->fd = fd;
	232	tcp->sport = port;
cac13158	233	bufadd(listeners, mustart(listenloop, tcp));
8774c31b FT	234	if((fd = listensock4(port)) < 0) {
	235	if(errno != EADDRINUSE) {
	236	flog(LOG_ERR, "could not listen on IPv4 (port %i): %s", port, strerror(errno));
	237	exit(1);
	238	}
	239	} else {
41fe6c39 FT	240	omalloc(tcp);
	241	tcp->fd = fd;
	242	tcp->sport = port;
cac13158	243	bufadd(listeners, mustart(listenloop, tcp));
8774c31b FT	244	}
8774c31b FT	245	}