From 67a8a4047bac9e618da102e3194b6812c8f9ce85 Mon Sep 17 00:00:00 2001
From: neonknight <github@neonknight.ch>
Date: Wed, 3 Oct 2018 16:20:35 +0200
Subject: [PATCH] fix py3 incompatibility

---
 oletools/olevba.py |  9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/oletools/olevba.py b/oletools/olevba.py
index 531dde5..7e58d75 100644
--- a/oletools/olevba.py
+++ b/oletools/olevba.py
@@ -246,7 +246,10 @@ import sys
 import os
 import logging
 import struct
-import cStringIO
+try:
+    from cStringIO import StringIO
+except ImportError:
+    from io import StringIO
 import math
 import zipfile
 import re
@@ -1352,7 +1355,7 @@ def _extract_vba(ole, vba_root, project_path, dir_path, relaxed=False):
             else:
                 raise UnexpectedDataError(dir_path, name, expected, value)
 
-    dir_stream = cStringIO.StringIO(decompress_stream(dir_compressed))
+    dir_stream = StringIO(decompress_stream(dir_compressed))
 
     # PROJECTSYSKIND Record
     projectsyskind_id = struct.unpack("<H", dir_stream.read(2))[0]
@@ -2309,7 +2312,7 @@ class VBA_Parser(object):
             _file = filename
         else:
             # file already read in memory, make it a file-like object for zipfile:
-            _file = cStringIO.StringIO(data)
+            _file = StringIO(data)
         #self.file = _file
         self.ole_file = None
         self.ole_subfiles = []
--
libgit2 0.21.4