Better handling of malformed pdf

f6381680 · jvoisin · b84f73c5 · f6381680
Commit f6381680 authored Nov 06, 2020 by jvoisin
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

libmat2/pdf.py libmat2/pdf.py +3 -0

No files found.
--- a/libmat2/pdf.py
+++ b/libmat2/pdf.py
@@ -84,6 +84,9 @@ class PDFParser(abstract.AbstractParser):

        for pagenum in range(pages_count):
            page = document.get_page(pagenum)
+            if page is None:
+                logging.error("Unable to get PDF pages")
+                return False
            page_width, page_height = page.get_size()
            logging.info("Rendering page %d/%d", pagenum + 1, pages_count)