Add cookie usage

1 files changed, 20 insertions, 10 deletions

diff --git a/getgbook.c b/getgbook.c
index a11bf24..08cee6c 100644
--- a/getgbook.c
+++ b/getgbook.c
@@ -20,7 +20,7 @@ typedef struct {
 
 char pagecodes[][3] = { "PP", "PR", "PA", "PT", "\0" };
 
-Page *getpagedetail(char *bookid, char *pg)
+Page *getpagedetail(char *bookid, char *pg, char *cookie)
 {
 	char url[URLMAX];
 	char *buf, *c, *d, m[80], *p;
@@ -28,7 +28,7 @@ Page *getpagedetail(char *bookid, char *pg)
 
 	snprintf(url, URLMAX, "/books?id=%s&pg=%s&jscmd=click3", bookid, pg);
 
-	if(!get("books.google.com", url, &buf))
+	if(!get("books.google.com", url, cookie, NULL, &buf))
 		return NULL;
 
 	snprintf(m, 80, "\"pid\":\"%s\"", pg);
@@ -67,8 +67,8 @@ Page *getpagedetail(char *bookid, char *pg)
 
 int main(int argc, char *argv[])
 {
-	char *bookid, pg[16], buf[1024], n[80], code[3];
-	int i, c;
+	char *bookid, *tmp, pg[16], buf[1024], n[80], code[3], cookie[COOKIEMAX], u[1024];
+	int i, c, retry;
 	Page *page;
 
 	if(argc < 2 || argc > 3 ||
@@ -81,10 +81,10 @@ int main(int argc, char *argv[])
 
 	if(argv[1][0] == '-') {
 		strncpy(code, pagecodes[0], 3);
-		c = i = 0;
+		c = i = retry = 0;
 		while(++i) {
 			snprintf(pg, 15, "%s%d", code, i);
-			if(!(page = getpagedetail(bookid, pg))) {
+			if(!(page = getpagedetail(bookid, pg, cookie))) {
 				/* no more pages with that code */
 				strncpy(code, pagecodes[++c], 3);
 				if(code[0] == '\0') break;
@@ -92,13 +92,23 @@ int main(int argc, char *argv[])
 				continue;
 			}
 			if(!page->url[0]) {
-				fprintf(stderr, "%s not available\n", pg);
 				free(page);
+				/* try with fresh cookie */
+				if(!retry) {
+					snprintf(u, URLMAX, "/books?id=%s", bookid);
+					get("books.google.com", u, NULL, cookie, &tmp);
+					free(tmp);
+					retry=1;
+					i--;
+				} else {
+					fprintf(stderr, "%s not available\n", pg);
+					retry=0;
+				}
 				continue;
 			}
 			if(argv[1][1] == 'a') {
 				snprintf(n, 80, "%05d.png", page->num);
-				gettofile("books.google.com", page->url, n);
+				gettofile("books.google.com", page->url, cookie, NULL, n);
 				printf("Downloaded page %d\n", page->num);
 			} else if(page->num != -1)
 				printf("%s %d\n", page->name, page->num);
@@ -107,13 +117,13 @@ int main(int argc, char *argv[])
 	} else {
 		while(fgets(buf, 1024, stdin)) {
 			sscanf(buf, "%15s", pg);
-			if(!(page = getpagedetail(bookid, pg)) || !page->url[0]) {
+			if(!(page = getpagedetail(bookid, pg, cookie)) || !page->url[0]) {
 				fprintf(stderr, "%s failed\n", pg);
 				free(page);
 				continue;
 			}
 			snprintf(n, 80, "%05d.png", page->num);
-			gettofile("books.google.com", page->url, n);
+			gettofile("books.google.com", page->url, cookie, NULL, n);
 			printf("Downloaded page %d\n", page->num);
 			free(page);
 		}