gp_ini: Parse .ini files in SYSVOL

These are fdeploy, scripts + psscripts as well as the GPT.ini at the top
level. Note that GPT.ini has a different character encoding and we
specify it here.

Signed-off-by: Garming Sam <garming@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>

diff --git a/python/samba/gp_parse/gp_ini.py b/python/samba/gp_parse/gp_ini.py
new file mode 100644 (file)
index 0000000..6c9b616
--- /dev/null
+++ b/python/samba/gp_parse/gp_ini.py
@@ -0,0 +1,83 @@
+# GPO Parser for extensions with ini files
+#
+# Copyright (C) Andrew Bartlett <abartlet@samba.org> 2018
+# Written by Garming Sam <garming@catalyst.net.nz>
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+
+import codecs
+import collections
+
+from ConfigParser import ConfigParser
+from xml.etree.ElementTree import Element, SubElement
+from StringIO import StringIO
+
+from samba.gp_parse import GPParser
+
+# [MS-GPFR] Group Policy Folder Redirection
+# [MS-GPSCR] Scripts Extension
+class GPIniParser(GPParser):
+    ini_conf = None
+
+    def parse(self, contents):
+        # Required dict_type in Python 2.7
+        self.ini_conf = ConfigParser(dict_type=collections.OrderedDict)
+        self.ini_conf.optionxform = str
+
+        self.ini_conf.readfp(StringIO(contents.decode(self.encoding)))
+
+    def write_xml(self, filename):
+        with file(filename, 'w') as f:
+            root = Element('IniFile')
+
+            for sec_ini in self.ini_conf.sections():
+                section = SubElement(root, 'Section')
+                section.attrib['name'] = sec_ini
+                for key_ini, val_ini in self.ini_conf.items(sec_ini, raw=True):
+                    child = SubElement(section, 'Parameter')
+                    key = SubElement(child, 'Key')
+                    value = SubElement(child, 'Value')
+                    key.text = key_ini
+                    value.text = val_ini
+
+            self.write_pretty_xml(root, f)
+
+        # from xml.etree.ElementTree import fromstring
+        # contents = codecs.open(filename, encoding='utf-8').read()
+        # self.load_xml(fromstring(contents))
+
+    def load_xml(self, root):
+        # Required dict_type in Python 2.7
+        self.ini_conf = ConfigParser(dict_type=collections.OrderedDict)
+        self.ini_conf.optionxform = str
+
+        for s in root.findall('Section'):
+            section_name = s.attrib['name']
+            self.ini_conf.add_section(section_name)
+
+            for param in s.findall('Parameter'):
+                key = param.find('Key').text
+                value = param.find('Value').text
+                if value is None:
+                    value = ''
+                self.ini_conf.set(section_name, key, value)
+
+    def write_binary(self, filename):
+        with codecs.open(filename, 'wb+', self.encoding) as f:
+            self.ini_conf.write(f)
+
+
+class GPTIniParser(GPIniParser):
+    encoding = 'utf-8'