From 9f86b203c1a411dd24b3219c104db16b6ecedc95 Mon Sep 17 00:00:00 2001
From: lvtao-sec <lvtao@iie.ac.cn>
Date: Sat, 22 Aug 2020 18:50:00 +0800
Subject: check return value of xmlDocGetRootElement call if is NULL

---
 backend/epub/epub-document.c | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

(limited to 'backend')

diff --git a/backend/epub/epub-document.c b/backend/epub/epub-document.c
index 552a18d3..8f97aa9f 100644
--- a/backend/epub/epub-document.c
+++ b/backend/epub/epub-document.c
@@ -181,8 +181,18 @@ epub_document_check_hits(EvDocumentFind *document_find,
 {
 	gchar *filepath = g_filename_from_uri((gchar*)page->backend_page,NULL,NULL);
 	htmlDocPtr htmldoc =  xmlParseFile(filepath);
-	htmlNodePtr htmltag = xmlDocGetRootElement(htmldoc);
-	int count=0;
+    if (htmldoc == NULL) {
+        g_free(filepath);
+        return 0;
+    }
+    htmlNodePtr htmltag = xmlDocGetRootElement(htmldoc);
+    if (htmltag == NULL) {
+        g_free(filepath);
+        xmlFreeDoc (htmldoc);
+        return 0;
+    }
+
+    int count=0;
 	htmlNodePtr bodytag = htmltag->xmlChildrenNode;
 
 	while ( xmlStrcmp(bodytag->name,(xmlChar*)"body") ) {
-- 
cgit v1.2.1