From bd2fd4f1007508298660d1b617b9368280ed9f51 Mon Sep 17 00:00:00 2001 From: Nick White Date: Sun, 24 Jul 2011 12:14:40 +0100 Subject: Add cookie usage --- util.c | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) (limited to 'util.c') diff --git a/util.c b/util.c index ab6fdf8..21bc598 100644 --- a/util.c +++ b/util.c @@ -6,9 +6,10 @@ #include #include #include +#include "util.h" /* plundered from suckless' sic */ -static int dial(char *host, char *port) { +int dial(char *host, char *port) { static struct addrinfo hints; int srv; struct addrinfo *res, *r; @@ -35,23 +36,29 @@ static int dial(char *host, char *port) { return srv; } -int get(char *host, char *path, char **buf) { +int get(char *host, char *path, char *sendcookie, char *savecookie, char **buf) { size_t l, res; int fd, i, p; char h[1024] = "\0"; + char c[1024] = ""; FILE *srv; + if(savecookie) savecookie[0] = 0; /* TEMP TO PLEASE GCC */ if((fd = dial(host, "80")) == -1) return 0; srv = fdopen(fd, "r+"); + if(sendcookie) + snprintf(c, COOKIEMAX-1, "\r\nCookie: %s", sendcookie); fprintf(srv, "GET %s HTTP/1.0\r\nUser-Agent: getgbook-"VERSION \ - " (not mozilla)\r\nHost: %s\r\n\r\n", path, host); + " (not mozilla)\r\nHost: %s%s\r\n\r\n", path, host, c); fflush(srv); while(h[0] != '\r') { fgets(h, 1024, srv); if(sscanf(h, "HTTP/%d.%d %d", &i, &i, &p) == 3 && p != 200) return 0; + if(savecookie != NULL && sscanf(h, "Set-Cookie: %s;", c)) + strncat(savecookie, c, COOKIEMAX-1); } *buf = malloc(sizeof(char *) * 4096); @@ -62,12 +69,12 @@ int get(char *host, char *path, char **buf) { return l; } -int gettofile(char *host, char *url, char *savepath) { +int gettofile(char *host, char *url, char *sendcookie, char *savecookie, char *savepath) { char *buf = 0; FILE *f; size_t i, l; - if(!(l = get(host, url, &buf))) { + if(!(l = get(host, url, sendcookie, savecookie, &buf))) { fprintf(stderr, "Could not download %s\n", url); return 1; } -- cgit v1.2.3