[ashd.git] / lib / req.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 <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <errno.h>
#include <ctype.h>
#include <stdio.h>

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

struct hthead *mkreq(char *method, char *url, char *ver)
{
    struct hthead *req;
    
    omalloc(req);
    req->method = sstrdup(method);
    req->url = sstrdup(url);
    req->ver = sstrdup(ver);
    req->rest = sstrdup(url);
    return(req);
}

struct hthead *mkresp(int code, char *msg, char *ver)
{
    struct hthead *resp;
    
    omalloc(resp);
    resp->code = code;
    resp->msg = sstrdup(msg);
    resp->ver = sstrdup(ver);
    return(resp);
}

void freehthead(struct hthead *head)
{
    int i;
    
    if(head->method != NULL)
	free(head->method);
    if(head->url != NULL)
	free(head->url);
    if(head->msg != NULL)
	free(head->msg);
    if(head->ver != NULL)
	free(head->ver);
    if(head->rest != NULL)
	free(head->rest);
    if(head->headers) {
	for(i = 0; i < head->noheaders; i++) {
	    free(head->headers[i][0]);
	    free(head->headers[i][1]);
	    free(head->headers[i]);
	}
	free(head->headers);
    }
    free(head);
}

char *getheader(struct hthead *head, char *name)
{
    int i;
    
    for(i = 0; i < head->noheaders; i++) {
	if(!strcasecmp(head->headers[i][0], name))
	    return(head->headers[i][1]);
    }
    return(NULL);
}

static void trim(struct charbuf *buf)
{
    char *p;
    
    for(p = buf->b; (p - buf->b < buf->d) && isspace(*p); p++);
    memmove(buf->b, p, buf->d -= (p - buf->b));
    if(buf->d > 0)
	for(p = buf->b + buf->d - 1; (p > buf->b) && isspace(*p); p--, buf->d--);
}

int parseheaders(struct hthead *head, FILE *in)
{
    int c, state;
    struct charbuf name, val;
    
    bufinit(name);
    bufinit(val);
    state = 0;
    while(1) {
	c = fgetc(in);
    again:
	if(state == 0) {
	    if(c == '\r') {
	    } else if(c == '\n') {
		break;
	    } else if(c == EOF) {
		goto fail;
	    } else {
		state = 1;
		goto again;
	    }
	} else if(state == 1) {
	    if(c == ':') {
		trim(&name);
		bufadd(name, 0);
		state = 2;
	    } else if(c == '\r') {
	    } else if(c == '\n') {
		goto fail;
	    } else if(c == EOF) {
		goto fail;
	    } else {
		bufadd(name, c);
	    }
	} else if(state == 2) {
	    if(c == '\r') {
	    } else if(c == '\n') {
		trim(&val);
		bufadd(val, 0);
		headappheader(head, name.b, val.b);
		buffree(name);
		buffree(val);
		state = 0;
	    } else if(c == EOF) {
		goto fail;
	    } else {
		bufadd(val, c);
	    }
	}
    }
    return(0);
    
fail:
    buffree(name);
    buffree(val);
    return(-1);
}

void replrest(struct hthead *head, char *rest)
{
    char *tmp;
    
    /* Do not free the current rest string yet, so that the new one
     * can be taken from a subpart of the old one. */
    tmp = head->rest;
    head->rest = sstrdup(rest);
    free(tmp);
}

void headpreheader(struct hthead *head, const char *name, const char *val)
{
    head->headers = srealloc(head->headers, sizeof(*head->headers) * (head->noheaders + 1));
    memmove(head->headers + 1, head->headers, sizeof(*head->headers) * head->noheaders);
    head->noheaders++;
    head->headers[0] = smalloc(sizeof(*head->headers[0]) * 2);
    head->headers[0][0] = sstrdup(name);
    head->headers[0][1] = sstrdup(val);
}

void headappheader(struct hthead *head, const char *name, const char *val)
{
    int i;

    i = head->noheaders++;
    head->headers = srealloc(head->headers, sizeof(*head->headers) * head->noheaders);
    head->headers[i] = smalloc(sizeof(*head->headers[i]) * 2);
    head->headers[i][0] = sstrdup(name);
    head->headers[i][1] = sstrdup(val);
}

int writeresp(FILE *out, struct hthead *resp)
{
    int i;
    
    if(fprintf(out, "%s %i %s\r\n", resp->ver, resp->code, resp->msg) < 0)
	return(-1);
    for(i = 0; i < resp->noheaders; i++) {
	if(fprintf(out, "%s: %s\r\n", resp->headers[i][0], resp->headers[i][1]) < 0)
	    return(-1);
    }
    return(0);
}

int sendreq(int sock, struct hthead *req, int fd)
{
    int ret, i;
    struct charbuf buf;
    
    bufinit(buf);
    bufcatstr2(buf, req->method);
    bufcatstr2(buf, req->url);
    bufcatstr2(buf, req->ver);
    bufcatstr2(buf, req->rest);
    for(i = 0; i < req->noheaders; i++) {
	bufcatstr2(buf, req->headers[i][0]);
	bufcatstr2(buf, req->headers[i][1]);
    }
    bufcatstr2(buf, "");
    ret = sendfd(sock, fd, buf.b, buf.d);
    buffree(buf);
    if(ret < 0)
	return(-1);
    else
	return(0);
}

int recvreq(int sock, struct hthead **reqp)
{
    int fd;
    struct charbuf buf;
    char *p;
    size_t l;
    char *name, *val;
    struct hthead *req;
    
    if((fd = recvfd(sock, &buf.b, &buf.d)) < 0) {
	return(-1);
    }
    buf.s = buf.d;
    p = buf.b;
    l = buf.d;
    
    *reqp = omalloc(req);
    if((req->method = sstrdup(decstr(&p, &l))) == NULL)
	goto fail;
    if((req->url = sstrdup(decstr(&p, &l))) == NULL)
	goto fail;
    if((req->ver = sstrdup(decstr(&p, &l))) == NULL)
	goto fail;
    if((req->rest = sstrdup(decstr(&p, &l))) == NULL)
	goto fail;
    
    while(1) {
	if(!*(name = decstr(&p, &l)))
	    break;
	val = decstr(&p, &l);
	headappheader(req, name, val);
    }
    
    buffree(buf);
    return(fd);
    
fail:
    close(fd);
    freehthead(req);
    errno = EPROTO;
    return(-1);
}

char *unquoteurl(char *in)
{
    struct charbuf buf;
    char *p;
    int c;
    
    bufinit(buf);
    p = in;
    while(*p) {
	if(*p == '%') {
	    if(!p[1] || !p[2])
		goto fail;
	    c = 0;
	    if((p[1] >= '0') && (p[1] <= '9'))          c |= (p[1] - '0') << 4;
	    else if((p[1] >= 'a') && (p[1] <= 'f'))     c |= (p[1] - 'a' + 10) << 4;
	    else if((p[1] >= 'A') && (p[1] <= 'F'))     c |= (p[1] - 'A' + 10) << 4;
	    else                                        goto fail;
	    if((p[2] >= '0') && (p[2] <= '9'))          c |= (p[2] - '0');
	    else if((p[2] >= 'a') && (p[2] <= 'f'))     c |= (p[2] - 'a' + 10);
	    else if((p[2] >= 'A') && (p[2] <= 'F'))     c |= (p[2] - 'A' + 10);
	    else                                        goto fail;
	    bufadd(buf, c);
	    p += 3;
	} else {
	    bufadd(buf, *(p++));
	}
    }
    bufadd(buf, 0);
    return(buf.b);
fail:
    buffree(buf);
    return(NULL);
}
Commit	Line	Data
33733396 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 <string.h>
41213112 FT	21	#include <unistd.h>
	22	#include <sys/socket.h>
	23	#include <errno.h>
5fc1bf9f FT	24	#include <ctype.h>
5fc1bf9f FT	25	#include <stdio.h>
33733396 FT	26
	27	#ifdef HAVE_CONFIG_H
	28	#include <config.h>
	29	#endif
	30	#include <utils.h>
41213112	31	#include <log.h>
33733396	32	#include <req.h>
41213112	33	#include <proc.h>
33733396	34
41213112	35	struct hthead mkreq(char method, char url, char ver)
33733396	36	{
41213112	37	struct hthead *req;
33733396 FT	38
	39	omalloc(req);
	40	req->method = sstrdup(method);
	41	req->url = sstrdup(url);
	42	req->ver = sstrdup(ver);
41213112	43	req->rest = sstrdup(url);
33733396 FT	44	return(req);
	45	}
	46
41213112 FT	47	struct hthead mkresp(int code, char msg, char *ver)
	48	{
	49	struct hthead *resp;
	50
	51	omalloc(resp);
	52	resp->code = code;
	53	resp->msg = sstrdup(msg);
	54	resp->ver = sstrdup(ver);
	55	return(resp);
	56	}
	57
	58	void freehthead(struct hthead *head)
33733396 FT	59	{
	60	int i;
	61
41213112 FT	62	if(head->method != NULL)
	63	free(head->method);
	64	if(head->url != NULL)
	65	free(head->url);
	66	if(head->msg != NULL)
	67	free(head->msg);
	68	if(head->ver != NULL)
	69	free(head->ver);
	70	if(head->rest != NULL)
	71	free(head->rest);
	72	if(head->headers) {
	73	for(i = 0; i < head->noheaders; i++) {
	74	free(head->headers[i][0]);
	75	free(head->headers[i][1]);
	76	free(head->headers[i]);
33733396	77	}
41213112	78	free(head->headers);
33733396	79	}
41213112	80	free(head);
33733396 FT	81	}
33733396 FT	82
41213112	83	char getheader(struct hthead head, char *name)
66987955 FT	84	{
	85	int i;
	86
41213112 FT	87	for(i = 0; i < head->noheaders; i++) {
	88	if(!strcasecmp(head->headers[i][0], name))
	89	return(head->headers[i][1]);
66987955 FT	90	}
	91	return(NULL);
	92	}
	93
5fc1bf9f FT	94	static void trim(struct charbuf *buf)
	95	{
	96	char *p;
	97
a39d9e16 FT	98	for(p = buf->b; (p - buf->b < buf->d) && isspace(*p); p++);
	99	memmove(buf->b, p, buf->d -= (p - buf->b));
	100	if(buf->d > 0)
c7b1ef35	101	for(p = buf->b + buf->d - 1; (p > buf->b) && isspace(*p); p--, buf->d--);
5fc1bf9f FT	102	}
	103
	104	int parseheaders(struct hthead head, FILE in)
	105	{
	106	int c, state;
	107	struct charbuf name, val;
	108
	109	bufinit(name);
	110	bufinit(val);
	111	state = 0;
	112	while(1) {
	113	c = fgetc(in);
	114	again:
	115	if(state == 0) {
	116	if(c == '\r') {
	117	} else if(c == '\n') {
	118	break;
	119	} else if(c == EOF) {
	120	goto fail;
	121	} else {
	122	state = 1;
	123	goto again;
	124	}
	125	} else if(state == 1) {
	126	if(c == ':') {
	127	trim(&name);
	128	bufadd(name, 0);
	129	state = 2;
	130	} else if(c == '\r') {
	131	} else if(c == '\n') {
	132	goto fail;
	133	} else if(c == EOF) {
	134	goto fail;
	135	} else {
	136	bufadd(name, c);
	137	}
	138	} else if(state == 2) {
	139	if(c == '\r') {
	140	} else if(c == '\n') {
	141	trim(&val);
	142	bufadd(val, 0);
	143	headappheader(head, name.b, val.b);
	144	buffree(name);
	145	buffree(val);
	146	state = 0;
	147	} else if(c == EOF) {
	148	goto fail;
	149	} else {
	150	bufadd(val, c);
	151	}
	152	}
	153	}
	154	return(0);
	155
	156	fail:
	157	buffree(name);
	158	buffree(val);
	159	return(-1);
	160	}
	161
9e9eca79 FT	162	void replrest(struct hthead head, char rest)
	163	{
	164	char *tmp;
	165
	166	/* Do not free the current rest string yet, so that the new one
5fc1bf9f	167	* can be taken from a subpart of the old one. */
9e9eca79 FT	168	tmp = head->rest;
	169	head->rest = sstrdup(rest);
	170	free(tmp);
	171	}
	172
41213112	173	void headpreheader(struct hthead head, const char name, const char *val)
33733396	174	{
41213112 FT	175	head->headers = srealloc(head->headers, sizeof(head->headers) (head->noheaders + 1));
	176	memmove(head->headers + 1, head->headers, sizeof(head->headers) head->noheaders);
	177	head->noheaders++;
	178	head->headers[0] = smalloc(sizeof(head->headers[0]) 2);
	179	head->headers[0][0] = sstrdup(name);
	180	head->headers[0][1] = sstrdup(val);
33733396 FT	181	}
33733396 FT	182
41213112	183	void headappheader(struct hthead head, const char name, const char *val)
33733396 FT	184	{
	185	int i;
	186
41213112 FT	187	i = head->noheaders++;
	188	head->headers = srealloc(head->headers, sizeof(head->headers) head->noheaders);
	189	head->headers[i] = smalloc(sizeof(head->headers[i]) 2);
	190	head->headers[i][0] = sstrdup(name);
	191	head->headers[i][1] = sstrdup(val);
	192	}
	193
5fc1bf9f FT	194	int writeresp(FILE out, struct hthead resp)
	195	{
	196	int i;
	197
	198	if(fprintf(out, "%s %i %s\r\n", resp->ver, resp->code, resp->msg) < 0)
	199	return(-1);
	200	for(i = 0; i < resp->noheaders; i++) {
	201	if(fprintf(out, "%s: %s\r\n", resp->headers[i][0], resp->headers[i][1]) < 0)
	202	return(-1);
	203	}
	204	return(0);
	205	}
	206
af34331c	207	int sendreq(int sock, struct hthead *req, int fd)
41213112 FT	208	{
41213112 FT	209	int ret, i;
41213112 FT	210	struct charbuf buf;
41213112 FT	211
41213112 FT	212	bufinit(buf);
	213	bufcatstr2(buf, req->method);
	214	bufcatstr2(buf, req->url);
	215	bufcatstr2(buf, req->ver);
	216	bufcatstr2(buf, req->rest);
	217	for(i = 0; i < req->noheaders; i++) {
	218	bufcatstr2(buf, req->headers[i][0]);
	219	bufcatstr2(buf, req->headers[i][1]);
	220	}
	221	bufcatstr2(buf, "");
af34331c	222	ret = sendfd(sock, fd, buf.b, buf.d);
41213112	223	buffree(buf);
af34331c	224	if(ret < 0)
41213112	225	return(-1);
af34331c FT	226	else
af34331c FT	227	return(0);
41213112 FT	228	}
	229
	230	int recvreq(int sock, struct hthead **reqp)
	231	{
	232	int fd;
	233	struct charbuf buf;
	234	char *p;
	235	size_t l;
	236	char name, val;
	237	struct hthead *req;
	238
	239	if((fd = recvfd(sock, &buf.b, &buf.d)) < 0) {
	240	return(-1);
	241	}
	242	buf.s = buf.d;
	243	p = buf.b;
	244	l = buf.d;
	245
	246	*reqp = omalloc(req);
	247	if((req->method = sstrdup(decstr(&p, &l))) == NULL)
	248	goto fail;
	249	if((req->url = sstrdup(decstr(&p, &l))) == NULL)
	250	goto fail;
	251	if((req->ver = sstrdup(decstr(&p, &l))) == NULL)
	252	goto fail;
	253	if((req->rest = sstrdup(decstr(&p, &l))) == NULL)
	254	goto fail;
	255
	256	while(1) {
	257	if(!*(name = decstr(&p, &l)))
	258	break;
	259	val = decstr(&p, &l);
	260	headappheader(req, name, val);
	261	}
	262
	263	buffree(buf);
	264	return(fd);
	265
	266	fail:
	267	close(fd);
	268	freehthead(req);
	269	errno = EPROTO;
	270	return(-1);
33733396	271	}
1604c096 FT	272
	273	char unquoteurl(char in)
	274	{
	275	struct charbuf buf;
	276	char *p;
	277	int c;
	278
	279	bufinit(buf);
	280	p = in;
	281	while(*p) {
	282	if(*p == '%') {
	283	if(!p[1] \|\| !p[2])
	284	goto fail;
	285	c = 0;
	286	if((p[1] >= '0') && (p[1] <= '9')) c \|= (p[1] - '0') << 4;
	287	else if((p[1] >= 'a') && (p[1] <= 'f')) c \|= (p[1] - 'a' + 10) << 4;
	288	else if((p[1] >= 'A') && (p[1] <= 'F')) c \|= (p[1] - 'A' + 10) << 4;
	289	else goto fail;
	290	if((p[2] >= '0') && (p[2] <= '9')) c \|= (p[2] - '0');
	291	else if((p[2] >= 'a') && (p[2] <= 'f')) c \|= (p[2] - 'a' + 10);
	292	else if((p[2] >= 'A') && (p[2] <= 'F')) c \|= (p[2] - 'A' + 10);
	293	else goto fail;
	294	bufadd(buf, c);
	295	p += 3;
	296	} else {
	297	bufadd(buf, *(p++));
	298	}
	299	}
	300	bufadd(buf, 0);
	301	return(buf.b);
	302	fail:
	303	buffree(buf);
	304	return(NULL);
	305	}