From 217d6114fd903fa670b8c67bcf7983b653fe518f Mon Sep 17 00:00:00 2001 From: Christian Herdtweck Date: Fri, 8 Dec 2017 17:53:56 +0100 Subject: [PATCH] record_base: ensure streams are closed in iter_streams --- oletools/record_base.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/oletools/record_base.py b/oletools/record_base.py index 11208ac..b15cd26 100644 --- a/oletools/record_base.py +++ b/oletools/record_base.py @@ -145,10 +145,12 @@ class OleRecordFile(olefile.OleFileIO): clz = OleSummaryInformationStream else: clz = self.stream_class_for_name(direntry.name) - yield clz(self._open(direntry.isectStart, direntry.size), - direntry.size, - None if is_orphan else direntry.name, - direntry.entry_type) + stream = clz(self._open(direntry.isectStart, direntry.size), + direntry.size, + None if is_orphan else direntry.name, + direntry.entry_type) + yield stream + stream.close() class OleRecordStream(object): -- libgit2 0.21.4