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
|
# gp_scripts_ext samba gpo policy
# Copyright (C) David Mulder <dmulder@suse.com> 2020
#
# 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, re
from subprocess import Popen, PIPE
from samba.gp.gpclass import gp_pol_ext, drop_privileges, gp_file_applier, \
gp_misc_applier
from tempfile import NamedTemporaryFile
from samba.gp.util.logging import log
intro = '''
### autogenerated by samba
#
# This file is generated by the gp_scripts_ext Group Policy
# Client Side Extension. To modify the contents of this file,
# modify the appropriate Group Policy objects which apply
# to this machine. DO NOT MODIFY THIS FILE DIRECTLY.
#
'''
end = '''
### autogenerated by samba ###
'''
class gp_scripts_ext(gp_pol_ext, gp_file_applier):
def __str__(self):
return 'Unix Settings/Scripts'
def process_group_policy(self, deleted_gpo_list, changed_gpo_list, cdir=None):
for guid, settings in deleted_gpo_list:
if str(self) in settings:
for attribute, script in settings[str(self)].items():
self.unapply(guid, attribute, script)
for gpo in changed_gpo_list:
if gpo.file_sys_path:
reg_key = 'Software\\Policies\\Samba\\Unix Settings'
sections = { '%s\\Daily Scripts' % reg_key : '/etc/cron.daily',
'%s\\Monthly Scripts' % reg_key : '/etc/cron.monthly',
'%s\\Weekly Scripts' % reg_key : '/etc/cron.weekly',
'%s\\Hourly Scripts' % reg_key : '/etc/cron.hourly' }
pol_file = 'MACHINE/Registry.pol'
path = os.path.join(gpo.file_sys_path, pol_file)
pol_conf = self.parse(path)
if not pol_conf:
continue
policies = {}
for e in pol_conf.entries:
if e.keyname in sections.keys() and e.data.strip():
if e.keyname not in policies:
policies[e.keyname] = []
policies[e.keyname].append(e.data)
def applier_func(keyname, entries):
ret = []
cron_dir = sections[keyname] if not cdir else cdir
for data in entries:
with NamedTemporaryFile(prefix='gp_', mode="w+",
delete=False, dir=cron_dir) as f:
contents = '#!/bin/sh\n%s' % intro
contents += '%s\n' % data
f.write(contents)
os.chmod(f.name, 0o700)
ret.append(f.name)
return ret
for keyname, entries in policies.items():
# Each GPO applies only one set of each type of script, so
# so the attribute matches the keyname.
attribute = keyname
# The value hash is generated from the script entries,
# ensuring any changes to this GPO will cause the scripts
# to be rewritten.
value_hash = self.generate_value_hash(*entries)
self.apply(gpo.name, attribute, value_hash, applier_func,
keyname, entries)
# Cleanup any old scripts that are no longer part of the policy
self.clean(gpo.name, keep=policies.keys())
def rsop(self, gpo, target='MACHINE'):
output = {}
pol_file = '%s/Registry.pol' % target
if gpo.file_sys_path:
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:
key = e.keyname.split('\\')[-1]
if key.endswith('Scripts') and e.data.strip():
if key not in output.keys():
output[key] = []
output[key].append(e.data)
return output
def fetch_crontab(username):
p = Popen(['crontab', '-l', '-u', username], stdout=PIPE, stderr=PIPE)
out, err = p.communicate()
if p.returncode != 0:
log.warning('Failed to read the crontab: %s' % err)
m = re.findall('%s(.*)%s' % (intro, end), out.decode(), re.DOTALL)
if len(m) == 1:
entries = m[0].strip().split('\n')
else:
entries = []
m = re.findall('(.*)%s.*%s(.*)' % (intro, end), out.decode(), re.DOTALL)
if len(m) == 1:
others = '\n'.join([l.strip() for l in m[0]])
else:
others = out.decode()
return others, entries
def install_crontab(fname, username):
p = Popen(['crontab', fname, '-u', username], stdout=PIPE, stderr=PIPE)
_, err = p.communicate()
if p.returncode != 0:
raise RuntimeError('Failed to install crontab: %s' % err)
def install_user_crontab(username, others, entries):
with NamedTemporaryFile() as f:
if len(entries) > 0:
f.write('\n'.join([others, intro,
'\n'.join(entries), end]).encode())
else:
f.write(others.encode())
f.flush()
install_crontab(f.name, username)
class gp_user_scripts_ext(gp_scripts_ext, gp_misc_applier):
def unapply(self, guid, attribute, entry):
others, entries = fetch_crontab(self.username)
if entry in entries:
entries.remove(entry)
install_user_crontab(self.username, others, entries)
self.cache_remove_attribute(guid, attribute)
def apply(self, guid, attribute, entry):
old_val = self.cache_get_attribute_value(guid, attribute)
others, entries = fetch_crontab(self.username)
if not old_val or entry not in entries:
entries.append(entry)
install_user_crontab(self.username, others, entries)
self.cache_add_attribute(guid, attribute, entry)
def process_group_policy(self, deleted_gpo_list, changed_gpo_list):
for guid, settings in deleted_gpo_list:
if str(self) in settings:
for attribute, entry in settings[str(self)].items():
self.unapply(guid, attribute, entry)
for gpo in changed_gpo_list:
if gpo.file_sys_path:
reg_key = 'Software\\Policies\\Samba\\Unix Settings'
sections = { '%s\\Daily Scripts' % reg_key : '@daily',
'%s\\Monthly Scripts' % reg_key : '@monthly',
'%s\\Weekly Scripts' % reg_key : '@weekly',
'%s\\Hourly Scripts' % reg_key : '@hourly' }
pol_file = 'USER/Registry.pol'
path = os.path.join(gpo.file_sys_path, pol_file)
pol_conf = drop_privileges('root', self.parse, path)
if not pol_conf:
continue
attrs = []
for e in pol_conf.entries:
if e.keyname in sections.keys() and e.data.strip():
cron_freq = sections[e.keyname]
attribute = '%s:%s' % (e.keyname,
self.generate_attribute(e.data))
attrs.append(attribute)
entry = '%s %s' % (cron_freq, e.data)
self.apply(gpo.name, attribute, entry)
self.clean(gpo.name, keep=attrs)
def rsop(self, gpo):
return super().rsop(gpo, target='USER')
|
