From 6d506b87575ded3a59c9fc4f7b28d4160d9e9c43 Mon Sep 17 00:00:00 2001
From: jvoisin <julien.voisin@dustri.org>
Date: Sat, 31 Mar 2018 23:09:54 +0200
Subject: [PATCH] Add a deep check for office/libreoffice files
---
tests/test_libmat2.py | 54 ++++++++++++++++++++++++++++++++++++++++++-
1 file changed, 53 insertions(+), 1 deletion(-)
diff --git a/tests/test_libmat2.py b/tests/test_libmat2.py
index 743a845..c065237 100644
--- a/tests/test_libmat2.py
+++ b/tests/test_libmat2.py
@@ -3,8 +3,10 @@
import unittest
import shutil
import os
+import zipfile
+import tempfile
-from src import pdf, png, jpg, audio, office, libreoffice
+from src import pdf, png, jpg, audio, office, libreoffice, parser_factory
class TestGetMeta(unittest.TestCase):
def test_pdf(self):
@@ -54,6 +56,56 @@ class TestGetMeta(unittest.TestCase):
self.assertEqual(meta['meta:generator'], 'LibreOffice/3.3$Unix LibreOffice_project/330m19$Build-202')
+class TestDeepCleaning(unittest.TestCase):
+ def __check_zip_clean(self, p):
+ tempdir = tempfile.mkdtemp()
+ zipin = zipfile.ZipFile(p.filename)
+ zipin.extractall(tempdir)
+
+ for subdir, dirs, files in os.walk(tempdir):
+ for f in files:
+ complete_path = os.path.join(subdir, f)
+ inside_p = parser_factory.get_parser(complete_path)
+ if inside_p is None:
+ continue
+ print('[+] %s is clean inside %s' %(complete_path, p.filename))
+ self.assertEqual(inside_p.get_meta(), {})
+ shutil.rmtree(tempdir)
+
+ def test_office(self):
+ shutil.copy('./tests/data/dirty.docx', './tests/data/clean.docx')
+ p = office.OfficeParser('./tests/data/clean.docx')
+
+ meta = p.get_meta()
+ self.assertIsNotNone(meta)
+
+ ret = p.remove_all()
+ self.assertTrue(ret)
+
+ p = office.OfficeParser('./tests/data/clean.docx.cleaned')
+ self.assertEqual(p.get_meta(), {})
+
+ self.__check_zip_clean(p)
+
+ os.remove('./tests/data/clean.docx')
+
+
+ def test_libreoffice(self):
+ shutil.copy('./tests/data/dirty.odt', './tests/data/clean.odt')
+ p = libreoffice.LibreOfficeParser('./tests/data/clean.odt')
+
+ meta = p.get_meta()
+ self.assertIsNotNone(meta)
+
+ ret = p.remove_all()
+ self.assertTrue(ret)
+
+ p = libreoffice.LibreOfficeParser('./tests/data/clean.odt.cleaned')
+ self.assertEqual(p.get_meta(), {})
+
+ self.__check_zip_clean(p)
+
+ os.remove('./tests/data/clean.odt')
class TestCleaning(unittest.TestCase):
def test_pdf(self):
--
GitLab