From 40199b38e08b550bdc27062c78e8cbe38a5e7c80 Mon Sep 17 00:00:00 2001 From: Nick White Date: Fri, 15 Jul 2011 22:50:16 +0100 Subject: Separate into util.c again, so can use common functions in different book getters --- util.c | 69 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 69 insertions(+) create mode 100644 util.c (limited to 'util.c') diff --git a/util.c b/util.c new file mode 100644 index 0000000..427833b --- /dev/null +++ b/util.c @@ -0,0 +1,69 @@ +/* See COPYING file for copyright, license and warranty details. */ + +#include +#include +#include + +void die(char *msg) { + fputs(msg, stderr); + exit(EXIT_FAILURE); +} + +/* plundered from suckless' sic */ +static int dial(char *host, char *port) { + static struct addrinfo hints; + int srv; + struct addrinfo *res, *r; + + memset(&hints, 0, sizeof hints); + hints.ai_family = AF_UNSPEC; + hints.ai_socktype = SOCK_STREAM; + if(getaddrinfo(host, port, &hints, &res) != 0) + die("error: cannot resolve hostname\n"); + for(r = res; r; r = r->ai_next) { + if((srv = socket(r->ai_family, r->ai_socktype, r->ai_protocol)) == -1) + continue; + if(connect(srv, r->ai_addr, r->ai_addrlen) == 0) + break; + close(srv); + } + freeaddrinfo(res); + if(!r) + die("error: cannot connect to host\n"); + return srv; +} + +char *get(FILE *srv, char *host, char *path) { + size_t l, res; + int fd, i; + char *buf, *c, *p; + + fprintf(srv, "GET %s HTTP/1.0\r\nUser-Agent: getgbook-"VERSION" (not mozilla)\r\nHost: %s\r\n\r\n", path, host); + + fflush(srv); + + l=0; + fd = fileno(srv); + + buf = malloc(sizeof(char *) * 4096); + for(i=0; (res = read(fd, buf+l, 4096)) > 0; l+=res, i++) + buf = realloc(buf, sizeof(char *) * (l+4096)); + + /* check that it's a 200 */ + if(strncmp(buf+9, "200 ", 4)) { + free(buf); + return NULL; + } + + /* exclude header */ + for(p = buf; *p && *(p+1) && *(p+2) && *(p+3); p++) + if(!strncmp(p, "\r\n\r\n", 4)) break; + p+=4; + + i = l - (p - buf); + c = malloc(i); + memcpy(c, p, i); + free(buf); + + return c; +} -- cgit v1.2.3