1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
|
# 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
import re
from xml.etree.ElementTree import Element, SubElement
from samba.compat import ConfigParser
from samba.compat import StringIO
from samba.gp_parse import GPParser, ENTITY_USER_ID
# [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 build_xml_parameter(self, section_xml, section, key_ini, val_ini):
child = SubElement(section_xml, 'Parameter')
key = SubElement(child, 'Key')
value = SubElement(child, 'Value')
key.text = key_ini
value.text = val_ini
return child
def load_xml_parameter(self, param_xml, section):
key = param_xml.find('Key').text
value = param_xml.find('Value').text
if value is None:
value = ''
self.ini_conf.set(section, key, value)
return (key, value)
def build_xml_section(self, root_xml, sec_ini):
section = SubElement(root_xml, 'Section')
section.attrib['name'] = sec_ini
return section
def load_xml_section(self, section_xml):
section_name = section_xml.attrib['name']
self.ini_conf.add_section(section_name)
return section_name
def write_xml(self, filename):
with open(filename, 'wb') as f:
root = Element('IniFile')
for sec_ini in self.ini_conf.sections():
section = self.build_xml_section(root, sec_ini)
for key_ini, val_ini in self.ini_conf.items(sec_ini, raw=True):
self.build_xml_parameter(section, sec_ini, key_ini,
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 = self.load_xml_section(s)
for param in s.findall('Parameter'):
self.load_xml_parameter(param, section_name)
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'
def parse(self, contents):
try:
super(GPTIniParser, self).parse(contents)
except UnicodeDecodeError:
# Required dict_type in Python 2.7
self.ini_conf = ConfigParser(dict_type=collections.OrderedDict)
self.ini_conf.optionxform = str
# Fallback to Latin-1 which RSAT appears to use
self.ini_conf.readfp(StringIO(contents.decode('iso-8859-1')))
class GPScriptsIniParser(GPIniParser):
def build_xml_parameter(self, section_xml, section, key_ini, val_ini):
parent_return = super(GPScriptsIniParser,
self).build_xml_parameter(section_xml, section,
key_ini, val_ini)
cmdline = re.match('\\d+CmdLine$', key_ini)
if cmdline is not None:
value = parent_return.find('Value')
value.attrib['network_path'] = 'TRUE'
return parent_return
class GPFDeploy1IniParser(GPIniParser):
def build_xml_parameter(self, section_xml, section, key_ini, val_ini):
parent_return = super(GPFDeploy1IniParser,
self).build_xml_parameter(section_xml, section,
key_ini, val_ini)
# Add generalization metadata and parse out SID list
if section.lower() == 'folder_redirection':
# Process the header section
# {GUID} = S-1-1-0;S-1-1-0
# Remove the un-split SID values
key = parent_return.find('Value')
parent_return.remove(key)
sid_list = val_ini.strip().strip(';').split(';')
for sid in sid_list:
value = SubElement(parent_return, 'Value')
value.text = sid
value.attrib['user_id'] = 'TRUE'
else:
# Process redirection sections
# Only FullPath should be a network path
if key_ini == 'FullPath':
key = parent_return.find('Value')
key.attrib['network_path'] = 'TRUE'
return parent_return
def load_xml_parameter(self, param_xml, section):
# Re-join the SID list before entering ConfigParser
if section.lower() == 'folder_redirection':
key = param_xml.find('Key').text
values = param_xml.findall('Value')
if len(values) == 1:
# There appears to be a convention of a trailing semi-colon
# with only one value in the SID list.
value = values[0].text + ';'
else:
value = ';'.join([x.text for x in values])
self.ini_conf.set(section, key, value)
return (key, value)
# Do the normal ini code for other sections
return super(GPFDeploy1IniParser,
self).load_xml_parameter(param_xml, section)
def build_xml_section(self, root_xml, sec_ini):
section = SubElement(root_xml, 'Section')
if (sec_ini.lower() != 'folder_redirection' and
sec_ini.lower() != 'version'):
guid, sid = sec_ini.split('_')
section.attrib['fdeploy_GUID'] = guid
section.attrib['fdeploy_SID'] = sid
else:
section.attrib['name'] = sec_ini
return section
def load_xml_section(self, section_xml):
# Construct the name from GUID + SID if no name exists
if 'name' in section_xml.attrib:
section_name = section_xml.attrib['name']
else:
guid = section_xml.attrib['fdeploy_GUID']
sid = section_xml.attrib['fdeploy_SID']
section_name = guid + '_' + sid
self.ini_conf.add_section(section_name)
return section_name
def custom_entities(self, root, global_entities):
entities = []
fdeploy_sids = root.findall('.//Section[@fdeploy_SID]')
fdeploy_sids.sort(key = lambda x: x.tag)
for sid in fdeploy_sids:
old_attrib = sid.attrib['fdeploy_SID']
if old_attrib in global_entities:
new_attrib = global_entities[old_attrib]
else:
new_attrib = self.new_xml_entity(old_attrib, ENTITY_USER_ID)
entities.append((new_attrib, old_attrib))
global_entities.update([(old_attrib, new_attrib)])
sid.attrib['fdeploy_SID'] = new_attrib
return entities
|
