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("