1 files changed, 90 insertions, 0 deletions
diff --git a/apps/pdftool.c b/apps/pdftool.c
new file mode 100644
index 00000000..97a95043
--- /dev/null
+++ b/apps/pdftool.c
@@ -0,0 +1,90 @@
+#include "pdftool.h"
+
+char *basename = nil;
+pdf_xref *xref = nil;
+int pagecount = 0;
+static void (*cleanup)(void) = nil;
+
+void closexref(void);
+
+void die(fz_error error)
+{
+	fz_catch(error, "aborting");
+	if (cleanup)
+		cleanup();
+	closexref();
+	exit(1);
+}
+
+void setcleanup(void (*func)(void))
+{
+	cleanup = func;
+}
+
+void openxref(char *filename, char *password, int dieonbadpass)
+{
+	fz_error error;
+	int okay;
+
+	basename = strrchr(filename, '/');
+	if (!basename)
+		basename = filename;
+	else
+		basename++;
+
+	xref = pdf_newxref();
+	error = pdf_loadxref(xref, filename);
+	if (error)
+	{
+		fz_catch(error, "trying to repair");
+		error = pdf_repairxref(xref, filename);
+		if (error)
+			die(error);
+	}
+
+	error = pdf_decryptxref(xref);
+	if (error)
+		die(error);
+
+	if (pdf_needspassword(xref))
+	{
+		okay = pdf_authenticatepassword(xref, password);
+		if (!okay && !dieonbadpass)
+			fz_warn("invalid password, attempting to continue.");
+		else if (!okay && dieonbadpass)
+			die(fz_throw("invalid password"));
+	}
+
+	xref->root = fz_dictgets(xref->trailer, "Root");
+	if (xref->root)
+		fz_keepobj(xref->root);
+
+	xref->info = fz_dictgets(xref->trailer, "Info");
+	if (xref->info)
+		fz_keepobj(xref->info);
+
+	pagecount = pdf_getpagecount(xref);
+}
+
+void flushxref(void)
+{
+	if (xref)
+	{
+		pdf_flushxref(xref, 0);
+	}
+}
+
+void closexref(void)
+{
+	if (cleanup)
+		cleanup();
+
+	if (xref)
+	{
+		pdf_closexref(xref);
+		xref = nil;
+	}
+
+	basename = nil;
+}
+