diff options
| -rw-r--r-- | Makefile | 2 | ||||
| -rw-r--r-- | TODO | 4 | ||||
| -rw-r--r-- | getbnbook.c | 152 |
3 files changed, 154 insertions, 4 deletions
@@ -3,7 +3,7 @@ include config.mk NAME = getxbook -SRC = getgbook.c getabook.c +SRC = getgbook.c getabook.c getbnbook.c LIB = util.o GUI = getxbookgui.tcl DOC = README COPYING INSTALL LEGAL @@ -1,6 +1,4 @@ -before 1.0: create bn tool, fix http bugs, package for osx & windows - -# getbnbook +before 1.0: fix http bugs, package for osx & windows # other todos diff --git a/getbnbook.c b/getbnbook.c new file mode 100644 index 0000000..806b2e8 --- /dev/null +++ b/getbnbook.c @@ -0,0 +1,152 @@ +/* See COPYING file for copyright and license details. */ +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <sys/stat.h> +#include <sys/types.h> +#include <dirent.h> +#include "util.h" + +#define usage "getbnbook " VERSION " - a barnes and noble book downloader\n" \ + "usage: getbnbook [-n] isbn13\n" \ + " -n download pages from numbers in stdin\n" \ + " otherwise, all available pages will be downloaded\n" + +#define URLMAX 1024 +#define STRMAX 1024 +#define MAXPAGES 9999 + +int pages[MAXPAGES]; +int numpages; +char urlpath[STRMAX]; +char bookid[STRMAX]; +char *bookdir; +char cookies[COOKIEMAX]; + +int getpagelist() +{ + char url[URLMAX]; + char *buf = NULL; + char *s, *l; + int i, num; + char avail[STRMAX]; + char dummy1[STRMAX], dummy2[STRMAX]; + + numpages = 0; + + snprintf(url, URLMAX, "/DigBooks/viewer/bookviewmanager.aspx?op=getbookinfo&ean=%s", bookid); + + if(!get("search2.barnesandnoble.com", url, NULL, cookies, &buf)) + return 1; + + /* find page url structure */ + if((s=strstr(buf, "<imagesize name=\"med\"><param name=\"path\">")) == NULL) { + free(buf); + return 1; + } + s+=strlen("<imagesize name=\"med\"><param name=\"path\">"); + l = strchr(s, '<'); + for(i=0; s<l && i<STRMAX; s++,i++) + urlpath[i] = *s; + urlpath[i] = '\0'; + + /* find available pages */ + for(i=0, s=buf;*s && i<MAXPAGES; s++) { + if((s = strstr(s, "<page ")) == NULL) + break; + sscanf(s, "<page sequence=\"%d\" type=\"bitmap\" toc=\"%[^\"]\" folio=\"%[^\"]\" freevendstatus=\"%[^\"]\" />", &num, dummy1, dummy2, avail); + + if(strncmp(avail, "true", STRMAX) == 0) + pages[i++] = num; + } + + numpages = i; + + free(buf); + return 0; +} + +int getpage(int pagenum) +{ + char path[STRMAX], pageurl[STRMAX]; + char *s; + + snprintf(path, STRMAX, "%s/%04d.png", bookdir, pagenum); + + s=strchr(urlpath+7, '/'); + snprintf(pageurl, STRMAX, s, pagenum); + + if(gettofile("search2.barnesandnoble.com", pageurl, cookies, NULL, path)) { + fprintf(stderr, "%d failed\n", pagenum); + return 1; + } + + printf("%d downloaded\n", pagenum); + fflush(stdout); + return 0; +} + +int main(int argc, char *argv[]) +{ + char *tmp; + char buf[STRMAX], pgpath[STRMAX]; + char in[16]; + int a, i, n; + FILE *f; + DIR *d; + + if(argc < 2 || argc > 3 || + (argc == 3 && (argv[1][0]!='-' || argv[1][1] != 'n')) + || (argc >= 2 && argv[1][0] == '-' && argv[1][1] == 'h')) { + fputs(usage, stdout); + return 1; + } + + strncpy(bookid, argv[argc-1], STRMAX); + bookdir = argv[argc-1]; + + /* get cookie */ + if(get("www.barnesandnoble.com", "/", NULL, cookies, &tmp)) + free(tmp); + + if(getpagelist(bookid, pages)) { + fprintf(stderr, "Could not find any pages for %s\n", bookid); + return 1; + } + + if(!((d = opendir(bookdir)) || !mkdir(bookdir, S_IRWXU))) { + fprintf(stderr, "Could not create directory %s\n", bookdir); + return 1; + } + if(d) closedir(d); + + if(argc == 2) { + for(i=0; i<numpages; i++) { + snprintf(pgpath, STRMAX, "%s/%04d.png", bookdir, pages[i]); + if((f = fopen(pgpath, "r")) != NULL) { + fclose(f); + continue; + } + getpage(pages[i]); + } + } else if(argv[1][0] == '-' && argv[1][1] == 'n') { + while(fgets(buf, BUFSIZ, stdin)) { + sscanf(buf, "%15s", in); + i = -1; + sscanf(in, "%d", &n); + for(a=0; a<numpages; a++) { + if(pages[a] == n) { + i = a; + break; + } + } + if(i == -1) { + fprintf(stderr, "%s not found\n", in); + continue; + } + getpage(pages[i]); + } + } + + return EXIT_SUCCESS; +} |