summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--Makefile2
-rw-r--r--TODO4
-rw-r--r--getbnbook.c152
3 files changed, 154 insertions, 4 deletions
diff --git a/Makefile b/Makefile
index 733a7e1..65ea185 100644
--- a/Makefile
+++ b/Makefile
@@ -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
diff --git a/TODO b/TODO
index 7133837..15cf2ef 100644
--- a/TODO
+++ b/TODO
@@ -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;
+}