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
|
# gp_smb_conf_ext smb.conf gpo policy
# Copyright (C) David Mulder <dmulder@suse.com> 2018
#
# 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 os, numbers
from samba.gp.gpclass import gp_pol_ext, gp_misc_applier
from tempfile import NamedTemporaryFile
from samba.gp.util.logging import log
def is_number(x):
return isinstance(x, numbers.Number) and \
type(x) != bool
class gp_smb_conf_ext(gp_pol_ext, gp_misc_applier):
def unapply(self, guid, attribute, val):
current = self.lp.get(attribute)
data = self.parse_value(val)
# Only overwrite the smb.conf setting if it hasn't been modified. It
# may have been modified by another GPO.
if 'new_val' not in data or \
self.lptype_to_string(current) == data['new_val']:
self.lp.set(attribute, self.regtype_to_lptype(data['old_val'],
current))
self.store_lp_smb_conf(self.lp)
log.info('smb.conf [global] was changed',
{ attribute : str(data['old_val']) })
self.cache_remove_attribute(guid, attribute)
def apply(self, guid, attribute, val):
old_val = self.lp.get(attribute)
val = self.regtype_to_lptype(val, old_val)
self.lp.set(attribute, val)
self.store_lp_smb_conf(self.lp)
log.info('smb.conf [global] was changed', { attribute : str(val) })
data = self.generate_value(old_val=self.lptype_to_string(old_val),
new_val=self.lptype_to_string(val))
self.cache_add_attribute(guid, attribute, data)
def process_group_policy(self, deleted_gpo_list, changed_gpo_list):
pol_file = 'MACHINE/Registry.pol'
for guid, settings in deleted_gpo_list:
smb_conf = settings.get('smb.conf')
if smb_conf is None:
continue
for key, value in smb_conf.items():
self.unapply(guid, key, value)
for gpo in changed_gpo_list:
if gpo.file_sys_path:
section_name = 'Software\\Policies\\Samba\\smb_conf'
path = os.path.join(gpo.file_sys_path, pol_file)
pol_conf = self.parse(path)
if not pol_conf:
continue
attrs = []
for e in pol_conf.entries:
if not e.keyname.startswith(section_name):
continue
attrs.append(e.valuename)
self.apply(gpo.name, e.valuename, e.data)
# Cleanup settings which were removed from the policy
self.clean(gpo.name, keep=attrs)
def regtype_to_lptype(self, val, old_val):
if type(val) == bytes:
val = val.decode()
if is_number(val) and is_number(old_val):
val = str(val)
elif is_number(val) and type(old_val) == bool:
val = bool(val)
if type(val) == bool:
val = 'yes' if val else 'no'
return val
def store_lp_smb_conf(self, lp):
with NamedTemporaryFile(delete=False,
dir=os.path.dirname(lp.configfile)) as f:
lp.dump(False, f.name)
mode = os.stat(lp.configfile).st_mode
os.chmod(f.name, mode)
os.rename(f.name, lp.configfile)
def lptype_to_string(self, val):
if is_number(val):
val = str(val)
elif type(val) == bool:
val = 'yes' if val else 'no'
elif type(val) == list:
val = ' '.join(val)
return val
def __str__(self):
return "smb.conf"
def rsop(self, gpo):
output = {}
if gpo.file_sys_path:
section_name = 'Software\\Policies\\Samba\\smb_conf'
pol_file = 'MACHINE/Registry.pol'
path = os.path.join(gpo.file_sys_path, pol_file)
pol_conf = self.parse(path)
if not pol_conf:
return output
for e in pol_conf.entries:
if not e.keyname.startswith(section_name):
continue
if 'smb.conf' not in output.keys():
output['smb.conf'] = {}
output['smb.conf'][e.valuename] = e.data
return output
|
