diff options
| -rw-r--r-- | TODO | 6 | ||||
| -rw-r--r-- | getgbook.c | 74 | ||||
| -rw-r--r-- | util.c | 69 |
3 files changed, 80 insertions, 69 deletions
@@ -0,0 +1,6 @@ +usage: getgbook [-p|-a] isbn + -p print all available pages + -a get all available pages + otherwise, get pages from stdin + + @@ -1,79 +1,15 @@ -#define VERSION "prealpha" - -#define usage "getgbook bookid" +/* See COPYING file for copyright, license and warranty details. */ -#define hostname "books.google.com" +#define VERSION "prealpha" #include <stdio.h> #include <stdlib.h> #include <string.h> -#include <netdb.h> -#include <netinet/in.h> -#include <sys/socket.h> - -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; -} +#include "util.c" -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); +#define usage "getgbook bookid" - return c; -} +#define hostname "books.google.com" int main(int argc, char *argv[]) { @@ -0,0 +1,69 @@ +/* See COPYING file for copyright, license and warranty details. */ + +#include <netdb.h> +#include <netinet/in.h> +#include <sys/socket.h> + +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; +} |