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
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
|
""" A tool to handle creating users and groups with useradd/mod/del
and groupadd/mod/del """
import sys
import pwd
import grp
import Bcfg2.Client.XML
import subprocess
import Bcfg2.Client.Tools
class ExecutionError(Exception):
""" Raised when running an external command fails """
def __init__(self, msg, retval=None):
Exception.__init__(self, msg)
self.retval = retval
def __str__(self):
return "%s (rv: %s)" % (Exception.__str__(self),
self.retval)
class Executor(object):
""" A better version of Bcfg2.Client.Tool.Executor, which captures
stderr, raises exceptions on error, and doesn't use the shell to
execute by default """
def __init__(self, logger):
self.logger = logger
self.stdout = None
self.stderr = None
self.retval = None
def run(self, command, inputdata=None, shell=False):
""" Run a command, given as a list, optionally giving it the
specified input data """
self.logger.debug("Running: %s" % " ".join(command))
proc = subprocess.Popen(command, shell=shell, bufsize=16384,
stdin=subprocess.PIPE, stdout=subprocess.PIPE,
stderr=subprocess.PIPE, close_fds=True)
if inputdata:
for line in inputdata.splitlines():
self.logger.debug('> %s' % line)
(self.stdout, self.stderr) = proc.communicate(inputdata)
else:
(self.stdout, self.stderr) = proc.communicate()
for line in self.stdout.splitlines(): # pylint: disable=E1103
self.logger.debug('< %s' % line)
self.retval = proc.wait()
if self.retval == 0:
for line in self.stderr.splitlines(): # pylint: disable=E1103
self.logger.warning(line)
return True
else:
raise ExecutionError(self.stderr, self.retval)
class POSIXUsers(Bcfg2.Client.Tools.Tool):
""" A tool to handle creating users and groups with
useradd/mod/del and groupadd/mod/del """
__execs__ = ['/usr/sbin/useradd', '/usr/sbin/usermod', '/usr/sbin/userdel',
'/usr/sbin/groupadd', '/usr/sbin/groupmod',
'/usr/sbin/groupdel']
__handles__ = [('POSIXUser', None),
('POSIXGroup', None)]
__req__ = dict(POSIXUser=['name'],
POSIXGroup=['name'])
experimental = True
# A mapping of XML entry attributes to the indexes of
# corresponding values in the get*ent data structures
attr_mapping = dict(POSIXUser=dict(name=0, uid=2, gecos=4, home=5,
shell=6),
POSIXGroup=dict(name=0, gid=2))
def __init__(self, logger, setup, config):
Bcfg2.Client.Tools.Tool.__init__(self, logger, setup, config)
self.set_defaults = dict(POSIXUser=self.populate_user_entry,
POSIXGroup=lambda g: g)
self.cmd = Executor(logger)
self._existing = None
@property
def existing(self):
""" Get a dict of existing users and groups """
if self._existing is None:
self._existing = dict(POSIXUser=dict([(u[0], u)
for u in pwd.getpwall()]),
POSIXGroup=dict([(g[0], g)
for g in grp.getgrall()]))
return self._existing
def Inventory(self, states, structures=None):
if not structures:
structures = self.config.getchildren()
# we calculate a list of all POSIXUser and POSIXGroup entries,
# and then add POSIXGroup entries that are required to create
# the primary group for each user to the structures. this is
# sneaky and possibly evil, but it works great.
groups = []
for struct in structures:
groups.extend([e.get("name")
for e in struct.findall("POSIXGroup")])
for struct in structures:
for entry in struct.findall("POSIXUser"):
group = self.set_defaults[entry.tag](entry).get('group')
if group and group not in groups:
self.logger.debug("POSIXUsers: Adding POSIXGroup entry "
"'%s' for user '%s'" %
(group, entry.get("name")))
struct.append(Bcfg2.Client.XML.Element("POSIXGroup",
name=group))
return Bcfg2.Client.Tools.Tool.Inventory(self, states, structures)
def FindExtra(self):
extra = []
for handles in self.__handles__:
tag = handles[0]
specified = []
for entry in self.getSupportedEntries():
if entry.tag == tag:
specified.append(entry.get("name"))
extra.extend([Bcfg2.Client.XML.Element(tag, name=e)
for e in self.existing[tag].keys()
if e not in specified])
return extra
def populate_user_entry(self, entry):
""" Given a POSIXUser entry, set all of the 'missing' attributes
with their defaults """
defaults = dict(group=entry.get('name'),
gecos=entry.get('name'),
shell='/bin/bash')
if entry.get('name') == 'root':
defaults['home'] = '/root'
else:
defaults['home'] = '/home/%s' % entry.get('name')
for key, val in defaults.items():
if entry.get(key) is None:
entry.set(key, val)
if entry.get('group') in self.existing['POSIXGroup']:
entry.set('gid',
str(self.existing['POSIXGroup'][entry.get('group')][2]))
return entry
def user_supplementary_groups(self, entry):
""" Get a list of supplmentary groups that the user in the
given entry is a member of """
return [g for g in self.existing['POSIXGroup'].values()
if entry.get("name") in g[3] and g[0] != entry.get("group")]
def VerifyPOSIXUser(self, entry, _):
""" Verify a POSIXUser entry """
rv = self._verify(self.populate_user_entry(entry))
if entry.get("current_exists", "true") == "true":
# verify supplemental groups
actual = [g[0] for g in self.user_supplementary_groups(entry)]
expected = [e.text for e in entry.findall("MemberOf")]
if set(expected) != set(actual):
entry.set('qtext',
"\n".join([entry.get('qtext', '')] +
["%s %s has incorrect supplemental group "
"membership. Currently: %s. Should be: %s"
% (entry.tag, entry.get("name"),
actual, expected)]))
rv = False
if self.setup['interactive'] and not rv:
entry.set('qtext',
'%s\nInstall %s %s: (y/N) ' %
(entry.get('qtext', ''), entry.tag, entry.get('name')))
return rv
def VerifyPOSIXGroup(self, entry, _):
""" Verify a POSIXGroup entry """
rv = self._verify(entry)
if self.setup['interactive'] and not rv:
entry.set('qtext',
'%s\nInstall %s %s: (y/N) ' %
(entry.get('qtext', ''), entry.tag, entry.get('name')))
return rv
def _verify(self, entry):
""" Perform most of the actual work of verification """
errors = []
if entry.get("name") not in self.existing[entry.tag]:
entry.set('current_exists', 'false')
errors.append("%s %s does not exist" % (entry.tag,
entry.get("name")))
else:
for attr, idx in self.attr_mapping[entry.tag].items():
val = str(self.existing[entry.tag][entry.get("name")][idx])
entry.set("current_%s" % attr, val)
if attr in ["uid", "gid"]:
if entry.get(attr) is None:
# no uid/gid specified, so we let the tool
# automatically determine one -- i.e., it always
# verifies
continue
if val != entry.get(attr):
errors.append("%s for %s %s is incorrect. Current %s is "
"%s, but should be %s" %
(attr.title(), entry.tag, entry.get("name"),
attr, entry.get(attr), val))
if errors:
for error in errors:
self.logger.debug("%s: %s" % (self.name, error))
entry.set('qtext', "\n".join([entry.get('qtext', '')] + errors))
return len(errors) == 0
def Install(self, entries, states):
for entry in entries:
# install groups first, so that all groups exist for
# users that might need them
if entry.tag == 'POSIXGroup':
states[entry] = self._install(entry)
for entry in entries:
if entry.tag == 'POSIXUser':
states[entry] = self._install(entry)
self._existing = None
def _install(self, entry):
""" add or modify a user or group using the appropriate command """
if entry.get("name") not in self.existing[entry.tag]:
action = "add"
else:
action = "mod"
try:
self.cmd.run(self._get_cmd(action,
self.set_defaults[entry.tag](entry)))
self.modified.append(entry)
return True
except ExecutionError:
self.logger.error("POSIXUsers: Error creating %s %s: %s" %
(entry.tag, entry.get("name"),
sys.exc_info()[1]))
return False
def _get_cmd(self, action, entry):
""" Get a command to perform the appropriate action (add, mod,
del) on the given entry. The command is always the same; we
set all attributes on a given user or group when modifying it
rather than checking which ones need to be changed. This
makes things fail as a unit (e.g., if a user is logged in, you
can't change its home dir, but you could change its GECOS, but
the whole operation fails), but it also makes this function a
lot, lot easier and simpler."""
cmd = ["/usr/sbin/%s%s" % (entry.tag[5:].lower(), action)]
if action != 'del':
if entry.tag == 'POSIXGroup':
if entry.get('gid'):
cmd.extend(['-g', entry.get('gid')])
elif entry.tag == 'POSIXUser':
cmd.append('-m')
if entry.get('uid'):
cmd.extend(['-u', entry.get('uid')])
cmd.extend(['-g', entry.get('group')])
extras = [e.text for e in entry.findall("MemberOf")]
if extras:
cmd.extend(['-G', ",".join(extras)])
cmd.extend(['-d', entry.get('home')])
cmd.extend(['-s', entry.get('shell')])
cmd.extend(['-c', entry.get('gecos')])
cmd.append(entry.get('name'))
return cmd
def Remove(self, entries):
for entry in entries:
# remove users first, so that all users have been removed
# from groups before we remove them
if entry.tag == 'POSIXUser':
self._remove(entry)
for entry in entries:
if entry.tag == 'POSIXGroup':
try:
grp.getgrnam(entry.get("name"))
self._remove(entry)
except KeyError:
# at least some versions of userdel automatically
# remove the primary group for a user if the group
# name is the same as the username, and no other
# users are in the group
self.logger.info("POSIXUsers: Group %s does not exist. "
"It may have already been removed when "
"its users were deleted" %
entry.get("name"))
self._existing = None
self.extra = self.FindExtra()
def _remove(self, entry):
""" Remove an entry """
try:
self.cmd.run(self._get_cmd("del", entry))
return True
except ExecutionError:
self.logger.error("POSIXUsers: Error deleting %s %s: %s" %
(entry.tag, entry.get("name"),
sys.exc_info()[1]))
return False
|
