From: Joseph Sutton <josephsutton@catalyst.net.nz>
Date: Wed, 1 Sep 2021 03:42:28 +0000 (+1200)
Subject: python: Don't leak file handles
X-Git-Tag: ldb-2.5.0~599
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=cde38d36b98f1d40e7b58cd4c4b4bedfab76c390;p=thirdparty%2Fsamba.git

python: Don't leak file handles

Signed-off-by: Joseph Sutton <josephsutton@catalyst.net.nz>
Reviewed-by: Noel Power <npower@samba.org>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
---

diff --git a/python/samba/__init__.py b/python/samba/__init__.py
index 449e4826ffb..0e6a33322f8 100644
--- a/python/samba/__init__.py
+++ b/python/samba/__init__.py
@@ -217,7 +217,8 @@ class Ldb(_Ldb):
 
         :param ldif_path: Path to LDIF file.
         """
-        self.add_ldif(open(ldif_path, 'r').read())
+        with open(ldif_path, 'r') as ldif_file:
+            self.add_ldif(ldif_file.read())
 
     def add_ldif(self, ldif, controls=None):
         """Add data based on a LDIF string.
@@ -279,10 +280,11 @@ def read_and_sub_file(file_name, subst_vars):
     :param file_name: File to be read (typically from setup directory)
      param subst_vars: Optional variables to subsitute in the file.
     """
-    data = open(file_name, 'r', encoding="utf-8").read()
-    if subst_vars is not None:
-        data = substitute_var(data, subst_vars)
-        check_all_substituted(data)
+    with open(file_name, 'r', encoding="utf-8") as data_file:
+        data = data_file.read()
+        if subst_vars is not None:
+            data = substitute_var(data, subst_vars)
+            check_all_substituted(data)
     return data
 
 
diff --git a/python/samba/ms_schema.py b/python/samba/ms_schema.py
index b9ca3c61b72..2250fb55e3b 100644
--- a/python/samba/ms_schema.py
+++ b/python/samba/ms_schema.py
@@ -294,9 +294,9 @@ def __parse_schema_file(filename, objectClass):
     out = []
 
     from io import open
-    f = open(filename, "r", encoding='latin-1')
-    for entry in __read_raw_entries(f):
-        out.append(__write_ldif_one(__transform_entry(entry, objectClass)))
+    with open(filename, "r", encoding='latin-1') as f:
+        for entry in __read_raw_entries(f):
+            out.append(__write_ldif_one(__transform_entry(entry, objectClass)))
 
     return "\n\n".join(out)
 
diff --git a/python/samba/schema.py b/python/samba/schema.py
index 54fc9fc3125..a3adc162fa3 100644
--- a/python/samba/schema.py
+++ b/python/samba/schema.py
@@ -110,8 +110,13 @@ class Schema(object):
             setup_path('ad-schema/%s' % Schema.base_schemas[base_schema][0]),
             setup_path('ad-schema/%s' % Schema.base_schemas[base_schema][1]))
 
+        def read_file(file):
+            with open(file, 'rb') as data_file:
+                return data_file.read()
+
         if files is not None:
-            self.schema_data = "".join(get_string(open(file, 'rb').read()) for file in files)
+            self.schema_data = "".join(get_string(read_file(file))
+                                       for file in files)
 
         self.schema_data = substitute_var(self.schema_data,
                                           {"SCHEMADN": schemadn})
@@ -130,7 +135,7 @@ class Schema(object):
         if override_prefixmap is not None:
             self.prefixmap_data = override_prefixmap
         else:
-            self.prefixmap_data = open(setup_path("prefixMap.txt"), 'rb').read()
+            self.prefixmap_data = read_file(setup_path("prefixMap.txt"))
 
         if additional_prefixmap is not None:
             self.prefixmap_data += "".join("%s\n" % map for map in additional_prefixmap)