From eb6808111cd29294c8a7b8dc501e98f836015983 Mon Sep 17 00:00:00 2001
From: Christian Herdtweck <christian.herdtweck@intra2net.com>
Date: Wed, 27 Apr 2016 14:39:01 +0200
Subject: [PATCH] create exception MsoExtractionError, raise from mso_file_extract

---
 oletools/olevba.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/oletools/olevba.py b/oletools/olevba.py
index 5a06b08..24eff4a 100755
--- a/oletools/olevba.py
+++ b/oletools/olevba.py
@@ -304,6 +304,14 @@ class FileOpenError(Exception):
         self.filename = filename
 
 
+class MsoExtractionError(RuntimeError):
+    """ raised by mso_file_extract if parsing MSO/ActiveMIME data failed """
+
+    def __init__(self, msg):
+        super(MsoExtractionError, self).__init__(msg)
+        self.msg = msg
+
+
 #--- CONSTANTS ----------------------------------------------------------------
 
 # URL and message to report issues:
@@ -859,7 +867,7 @@ def mso_file_extract(data):
     :param data: bytes string, MSO/ActiveMime file content
     :return: bytes string, extracted data (uncompressed)
 
-    raise a RuntimeError if the data cannot be extracted
+    raise a MsoExtractionError if the data cannot be extracted
     """
     # check the magic:
     assert is_mso_file(data)
@@ -894,7 +902,7 @@ def mso_file_extract(data):
             return extracted_data
         except Exception:
             log.exception('zlib decompression failed')
-    raise RuntimeError('Unable to decompress data from a MSO/ActiveMime file')
+    raise MsoExtractionError('Unable to decompress data from a MSO/ActiveMime file')
 
 
 #--- FUNCTIONS ----------------------------------------------------------------
--
libgit2 0.21.4