To get this branch, use:
bzr branch
http://bzr.recompile.se/loggerhead/mandos/trunk
« back to all changes in this revision
Viewing changes to mandos
- browse files at revision 329
- compare with another revision
- download diff
- download tarball
- view history from revision 329
- Committer: Teddy Hogeborn
- Date: 2009-04-01 03:37:45 UTC
- Revision ID: teddy@fukt.bsnet.se-20090401033745-c89k6bij5opdm1rk
* mandos (ClientDBus.__del__): Bug fix: Correct mispasted code, and do
not try to call
dbus.service.Object.__del__() if it
does not exist.
(ClientDBus.start_checker): Simplify logic.
not try to call
dbus.service.Object.__del__() if it
does not exist.
(ClientDBus.start_checker): Simplify logic.
- files added:
- .bzr-builddeb
- debian
- debian/po
- files removed:
- network-protocol.txt
- files renamed:
- mandos-client.c => plugin-runner.c
- mandos-client.xml => plugin-runner.xml
- plugins.d/password-request.c => plugins.d/mandos-client.c
- plugins.d/password-request.xml => plugins.d/mandos-client.xml
- files modified:
- Makefile
added
removed
mandos
11
11
# and some lines in "main".
12
12
#
13
13
# Everything else is
14
# Copyright © 2007-2008 Teddy Hogeborn & Björn Påhlsson
14
# Copyright © 2008,2009 Teddy Hogeborn
15
# Copyright © 2008,2009 Björn Påhlsson
15
16
#
16
17
# This program is free software: you can redistribute it and/or modify
17
18
# it under the terms of the GNU General Public License as published by
24
25
# GNU General Public License for more details.
25
26
#
26
27
# You should have received a copy of the GNU General Public License
27
# along with this program. If not, see <http://www.gnu.org/licenses/>.
28
# along with this program. If not, see
29
# <http://www.gnu.org/licenses/>.
28
30
#
29
31
# Contact the authors at <mandos@fukt.bsnet.se>.
30
32
#
31
33
32
from __future__ import division
34
from __future__ import division, with_statement, absolute_import
33
35
34
36
import SocketServer
35
37
import socket
36
import select
37
from optparse import OptionParser
38
import optparse
38
39
import datetime
39
40
import errno
40
41
import gnutls.crypto
54
55
import stat
55
56
import logging
56
57
import logging.handlers
58
import pwd
59
from contextlib import closing
57
60
58
61
import dbus
62
import dbus.service
59
63
import gobject
60
64
import avahi
61
65
from dbus.mainloop.glib import DBusGMainLoop
62
66
import ctypes
67
import ctypes.util
63
68
64
version = "1.0"
69
version = "1.0.8"
65
70
66
71
logger = logging.Logger('mandos')
67
syslogger = logging.handlers.SysLogHandler\
68
(facility = logging.handlers.SysLogHandler.LOG_DAEMON,
69
address = "/dev/log")
70
syslogger.setFormatter(logging.Formatter\
71
('Mandos: %(levelname)s: %(message)s'))
72
syslogger = (logging.handlers.SysLogHandler
73
(facility = logging.handlers.SysLogHandler.LOG_DAEMON,
74
address = "/dev/log"))
75
syslogger.setFormatter(logging.Formatter
76
('Mandos [%(process)d]: %(levelname)s:'
77
' %(message)s'))
72
78
logger.addHandler(syslogger)
73
79
74
80
console = logging.StreamHandler()
75
console.setFormatter(logging.Formatter('%(name)s: %(levelname)s:'
76
' %(message)s'))
81
console.setFormatter(logging.Formatter('%(name)s [%(process)d]:'
82
' %(levelname)s: %(message)s'))
77
83
logger.addHandler(console)
78
84
79
85
class AvahiError(Exception):
80
def __init__(self, value):
86
def __init__(self, value, *args, **kwargs):
81
87
self.value = value
82
def __str__(self):
83
return repr(self.value)
88
super(AvahiError, self).__init__(value, *args, **kwargs)
89
def __unicode__(self):
90
return unicode(repr(self.value))
84
91
85
92
class AvahiServiceError(AvahiError):
86
93
pass
106
113
a sensible number of times
107
114
"""
108
115
def __init__(self, interface = avahi.IF_UNSPEC, name = None,
109
type = None, port = None, TXT = None, domain = "",
110
host = "", max_renames = 32768):
116
servicetype = None, port = None, TXT = None,
117
domain = "", host = "", max_renames = 32768,
118
protocol = avahi.PROTO_UNSPEC):
111
119
self.interface = interface
112
120
self.name = name
113
self.type = type
121
self.type = servicetype
114
122
self.port = port
115
if TXT is None:
116
self.TXT = []
117
else:
118
self.TXT = TXT
123
self.TXT = TXT if TXT is not None else []
119
124
self.domain = domain
120
125
self.host = host
121
126
self.rename_count = 0
127
self.max_renames = max_renames
128
self.protocol = protocol
122
129
def rename(self):
123
130
"""Derived from the Avahi example code"""
124
131
if self.rename_count >= self.max_renames:
125
logger.critical(u"No suitable service name found after %i"
126
u" retries, exiting.", rename_count)
127
raise AvahiServiceError("Too many renames")
128
name = server.GetAlternativeServiceName(name)
129
logger.error(u"Changing name to %r ...", name)
130
syslogger.setFormatter(logging.Formatter\
131
('Mandos (%s): %%(levelname)s:'
132
' %%(message)s' % name))
132
logger.critical(u"No suitable Zeroconf service name found"
133
u" after %i retries, exiting.",
134
self.rename_count)
135
raise AvahiServiceError(u"Too many renames")
136
self.name = server.GetAlternativeServiceName(self.name)
137
logger.info(u"Changing Zeroconf service name to %r ...",
138
str(self.name))
139
syslogger.setFormatter(logging.Formatter
140
('Mandos (%s) [%%(process)d]:'
141
' %%(levelname)s: %%(message)s'
142
% self.name))
133
143
self.remove()
134
144
self.add()
135
145
self.rename_count += 1
141
151
"""Derived from the Avahi example code"""
142
152
global group
143
153
if group is None:
144
group = dbus.Interface\
145
(bus.get_object(avahi.DBUS_NAME,
154
group = dbus.Interface(bus.get_object
155
(avahi.DBUS_NAME,
146
156
server.EntryGroupNew()),
147
avahi.DBUS_INTERFACE_ENTRY_GROUP)
157
avahi.DBUS_INTERFACE_ENTRY_GROUP)
148
158
group.connect_to_signal('StateChanged',
149
159
entry_group_state_changed)
150
logger.debug(u"Adding service '%s' of type '%s' ...",
160
logger.debug(u"Adding Zeroconf service '%s' of type '%s' ...",
151
161
service.name, service.type)
152
162
group.AddService(
153
163
self.interface, # interface
154
avahi.PROTO_INET6, # protocol
164
self.protocol, # protocol
155
165
dbus.UInt32(0), # flags
156
166
self.name, self.type,
157
167
self.domain, self.host,
164
174
# End of Avahi example code
165
175
166
176
177
def _datetime_to_dbus(dt, variant_level=0):
178
"""Convert a UTC datetime.datetime() to a D-Bus type."""
179
return dbus.String(dt.isoformat(), variant_level=variant_level)
180
181
167
182
class Client(object):
168
183
"""A representation of a client host served by this server.
169
184
Attributes:
170
name: string; from the config file, used in log messages
185
name: string; from the config file, used in log messages and
186
D-Bus identifiers
171
187
fingerprint: string (40 or 32 hexadecimal digits); used to
172
188
uniquely identify the client
173
secret: bytestring; sent verbatim (over TLS) to client
174
host: string; available for use by the checker command
175
created: datetime.datetime(); object creation, not client host
176
last_checked_ok: datetime.datetime() or None if not yet checked OK
177
timeout: datetime.timedelta(); How long from last_checked_ok
178
until this client is invalid
179
interval: datetime.timedelta(); How often to start a new checker
180
stop_hook: If set, called by stop() as stop_hook(self)
181
checker: subprocess.Popen(); a running checker process used
182
to see if the client lives.
183
'None' if no process is running.
189
secret: bytestring; sent verbatim (over TLS) to client
190
host: string; available for use by the checker command
191
created: datetime.datetime(); (UTC) object creation
192
last_enabled: datetime.datetime(); (UTC)
193
enabled: bool()
194
last_checked_ok: datetime.datetime(); (UTC) or None
195
timeout: datetime.timedelta(); How long from last_checked_ok
196
until this client is invalid
197
interval: datetime.timedelta(); How often to start a new checker
198
disable_hook: If set, called by disable() as disable_hook(self)
199
checker: subprocess.Popen(); a running checker process used
200
to see if the client lives.
201
'None' if no process is running.
184
202
checker_initiator_tag: a gobject event source tag, or None
185
stop_initiator_tag: - '' -
203
disable_initiator_tag: - '' -
186
204
checker_callback_tag: - '' -
187
205
checker_command: string; External command which is run to check if
188
206
client lives. %() expansions are done at
189
207
runtime with vars(self) as dict, so that for
190
208
instance %(name)s can be used in the command.
191
Private attibutes:
192
_timeout: Real variable for 'timeout'
193
_interval: Real variable for 'interval'
194
_timeout_milliseconds: Used when calling gobject.timeout_add()
195
_interval_milliseconds: - '' -
209
current_checker_command: string; current running checker_command
196
210
"""
197
def _set_timeout(self, timeout):
198
"Setter function for 'timeout' attribute"
199
self._timeout = timeout
200
self._timeout_milliseconds = ((self.timeout.days
201
* 24 * 60 * 60 * 1000)
202
+ (self.timeout.seconds * 1000)
203
+ (self.timeout.microseconds
204
// 1000))
205
timeout = property(lambda self: self._timeout,
206
_set_timeout)
207
del _set_timeout
208
def _set_interval(self, interval):
209
"Setter function for 'interval' attribute"
210
self._interval = interval
211
self._interval_milliseconds = ((self.interval.days
212
* 24 * 60 * 60 * 1000)
213
+ (self.interval.seconds
214
* 1000)
215
+ (self.interval.microseconds
216
// 1000))
217
interval = property(lambda self: self._interval,
218
_set_interval)
219
del _set_interval
220
def __init__(self, name = None, stop_hook=None, config={}):
211
def timeout_milliseconds(self):
212
"Return the 'timeout' attribute in milliseconds"
213
return ((self.timeout.days * 24 * 60 * 60 * 1000)
214
+ (self.timeout.seconds * 1000)
215
+ (self.timeout.microseconds // 1000))
216
217
def interval_milliseconds(self):
218
"Return the 'interval' attribute in milliseconds"
219
return ((self.interval.days * 24 * 60 * 60 * 1000)
220
+ (self.interval.seconds * 1000)
221
+ (self.interval.microseconds // 1000))
222
223
def __init__(self, name = None, disable_hook=None, config=None):
221
224
"""Note: the 'checker' key in 'config' sets the
222
225
'checker_command' attribute and *not* the 'checker'
223
226
attribute."""
224
227
self.name = name
228
if config is None:
229
config = {}
225
230
logger.debug(u"Creating client %r", self.name)
226
231
# Uppercase and remove spaces from fingerprint for later
227
232
# comparison purposes with return value from the fingerprint()
228
233
# function
229
self.fingerprint = config["fingerprint"].upper()\
230
.replace(u" ", u"")
234
self.fingerprint = (config["fingerprint"].upper()
235
.replace(u" ", u""))
231
236
logger.debug(u" Fingerprint: %s", self.fingerprint)
232
237
if "secret" in config:
233
238
self.secret = config["secret"].decode(u"base64")
234
239
elif "secfile" in config:
235
sf = open(config["secfile"])
236
self.secret = sf.read()
237
sf.close()
240
with closing(open(os.path.expanduser
241
(os.path.expandvars
242
(config["secfile"])))) as secfile:
243
self.secret = secfile.read()
238
244
else:
239
245
raise TypeError(u"No secret or secfile for client %s"
240
246
% self.name)
241
247
self.host = config.get("host", "")
242
self.created = datetime.datetime.now()
248
self.created = datetime.datetime.utcnow()
249
self.enabled = False
250
self.last_enabled = None
243
251
self.last_checked_ok = None
244
252
self.timeout = string_to_delta(config["timeout"])
245
253
self.interval = string_to_delta(config["interval"])
246
self.stop_hook = stop_hook
254
self.disable_hook = disable_hook
247
255
self.checker = None
248
256
self.checker_initiator_tag = None
249
self.stop_initiator_tag = None
257
self.disable_initiator_tag = None
250
258
self.checker_callback_tag = None
251
self.check_command = config["checker"]
252
def start(self):
259
self.checker_command = config["checker"]
260
self.current_checker_command = None
261
self.last_connect = None
262
263
def enable(self):
253
264
"""Start this client's checker and timeout hooks"""
265
self.last_enabled = datetime.datetime.utcnow()
254
266
# Schedule a new checker to be started an 'interval' from now,
255
267
# and every interval from then on.
256
self.checker_initiator_tag = gobject.timeout_add\
257
(self._interval_milliseconds,
258
self.start_checker)
268
self.checker_initiator_tag = (gobject.timeout_add
269
(self.interval_milliseconds(),
270
self.start_checker))
259
271
# Also start a new checker *right now*.
260
272
self.start_checker()
261
# Schedule a stop() when 'timeout' has passed
262
self.stop_initiator_tag = gobject.timeout_add\
263
(self._timeout_milliseconds,
264
self.stop)
265
def stop(self):
266
"""Stop this client.
267
The possibility that a client might be restarted is left open,
268
but not currently used."""
269
# If this client doesn't have a secret, it is already stopped.
270
if hasattr(self, "secret") and self.secret:
271
logger.info(u"Stopping client %s", self.name)
272
self.secret = None
273
else:
273
# Schedule a disable() when 'timeout' has passed
274
self.disable_initiator_tag = (gobject.timeout_add
275
(self.timeout_milliseconds(),
276
self.disable))
277
self.enabled = True
278
279
def disable(self):
280
"""Disable this client."""
281
if not getattr(self, "enabled", False):
274
282
return False
275
if getattr(self, "stop_initiator_tag", False):
276
gobject.source_remove(self.stop_initiator_tag)
277
self.stop_initiator_tag = None
283
logger.info(u"Disabling client %s", self.name)
284
if getattr(self, "disable_initiator_tag", False):
285
gobject.source_remove(self.disable_initiator_tag)
286
self.disable_initiator_tag = None
278
287
if getattr(self, "checker_initiator_tag", False):
279
288
gobject.source_remove(self.checker_initiator_tag)
280
289
self.checker_initiator_tag = None
281
290
self.stop_checker()
282
if self.stop_hook:
283
self.stop_hook(self)
291
if self.disable_hook:
292
self.disable_hook(self)
293
self.enabled = False
284
294
# Do not run this again if called by a gobject.timeout_add
285
295
return False
296
286
297
def __del__(self):
287
self.stop_hook = None
288
self.stop()
289
def checker_callback(self, pid, condition):
298
self.disable_hook = None
299
self.disable()
300
301
def checker_callback(self, pid, condition, command):
290
302
"""The checker has completed, so take appropriate actions."""
291
now = datetime.datetime.now()
292
303
self.checker_callback_tag = None
293
304
self.checker = None
294
if os.WIFEXITED(condition) \
295
and (os.WEXITSTATUS(condition) == 0):
296
logger.info(u"Checker for %(name)s succeeded",
297
vars(self))
298
self.last_checked_ok = now
299
gobject.source_remove(self.stop_initiator_tag)
300
self.stop_initiator_tag = gobject.timeout_add\
301
(self._timeout_milliseconds,
302
self.stop)
303
elif not os.WIFEXITED(condition):
305
if os.WIFEXITED(condition):
306
exitstatus = os.WEXITSTATUS(condition)
307
if exitstatus == 0:
308
logger.info(u"Checker for %(name)s succeeded",
309
vars(self))
310
self.checked_ok()
311
else:
312
logger.info(u"Checker for %(name)s failed",
313
vars(self))
314
else:
304
315
logger.warning(u"Checker for %(name)s crashed?",
305
316
vars(self))
306
else:
307
logger.info(u"Checker for %(name)s failed",
308
vars(self))
317
318
def checked_ok(self):
319
"""Bump up the timeout for this client.
320
This should only be called when the client has been seen,
321
alive and well.
322
"""
323
self.last_checked_ok = datetime.datetime.utcnow()
324
gobject.source_remove(self.disable_initiator_tag)
325
self.disable_initiator_tag = (gobject.timeout_add
326
(self.timeout_milliseconds(),
327
self.disable))
328
309
329
def start_checker(self):
310
330
"""Start a new checker subprocess if one is not running.
311
331
If a checker already exists, leave it running and do
318
338
# checkers alone, the checker would have to take more time
319
339
# than 'timeout' for the client to be declared invalid, which
320
340
# is as it should be.
341
342
# If a checker exists, make sure it is not a zombie
343
if self.checker is not None:
344
pid, status = os.waitpid(self.checker.pid, os.WNOHANG)
345
if pid:
346
logger.warning("Checker was a zombie")
347
gobject.source_remove(self.checker_callback_tag)
348
self.checker_callback(pid, status,
349
self.current_checker_command)
350
# Start a new checker if needed
321
351
if self.checker is None:
322
352
try:
323
# In case check_command has exactly one % operator
324
command = self.check_command % self.host
353
# In case checker_command has exactly one % operator
354
command = self.checker_command % self.host
325
355
except TypeError:
326
356
# Escape attributes for the shell
327
357
escaped_attrs = dict((key, re.escape(str(val)))
328
358
for key, val in
329
359
vars(self).iteritems())
330
360
try:
331
command = self.check_command % escaped_attrs
361
command = self.checker_command % escaped_attrs
332
362
except TypeError, error:
333
363
logger.error(u'Could not format string "%s":'
334
u' %s', self.check_command, error)
364
u' %s', self.checker_command, error)
335
365
return True # Try again later
366
self.current_checker_command = command
336
367
try:
337
368
logger.info(u"Starting checker %r for %s",
338
369
command, self.name)
370
# We don't need to redirect stdout and stderr, since
371
# in normal mode, that is already done by daemon(),
372
# and in debug mode we don't want to. (Stdin is
373
# always replaced by /dev/null.)
339
374
self.checker = subprocess.Popen(command,
340
375
close_fds=True,
341
376
shell=True, cwd="/")
342
self.checker_callback_tag = gobject.child_watch_add\
343
(self.checker.pid,
344
self.checker_callback)
345
except subprocess.OSError, error:
377
self.checker_callback_tag = (gobject.child_watch_add
378
(self.checker.pid,
379
self.checker_callback,
380
data=command))
381
# The checker may have completed before the gobject
382
# watch was added. Check for this.
383
pid, status = os.waitpid(self.checker.pid, os.WNOHANG)
384
if pid:
385
gobject.source_remove(self.checker_callback_tag)
386
self.checker_callback(pid, status, command)
387
except OSError, error:
346
388
logger.error(u"Failed to start subprocess: %s",
347
389
error)
348
390
# Re-run this periodically if run by gobject.timeout_add
349
391
return True
392
350
393
def stop_checker(self):
351
394
"""Force the checker process, if any, to stop."""
352
395
if self.checker_callback_tag:
364
407
if error.errno != errno.ESRCH: # No such process
365
408
raise
366
409
self.checker = None
410
367
411
def still_valid(self):
368
412
"""Has the timeout not yet passed for this client?"""
369
now = datetime.datetime.now()
413
if not getattr(self, "enabled", False):
414
return False
415
now = datetime.datetime.utcnow()
370
416
if self.last_checked_ok is None:
371
417
return now < (self.created + self.timeout)
372
418
else:
373
419
return now < (self.last_checked_ok + self.timeout)
374
420
375
421
422
class ClientDBus(Client, dbus.service.Object):
423
"""A Client class using D-Bus
424
Attributes:
425
dbus_object_path: dbus.ObjectPath ; only set if self.use_dbus
426
"""
427
# dbus.service.Object doesn't use super(), so we can't either.
428
429
def __init__(self, *args, **kwargs):
430
Client.__init__(self, *args, **kwargs)
431
# Only now, when this client is initialized, can it show up on
432
# the D-Bus
433
self.dbus_object_path = (dbus.ObjectPath
434
("/clients/"
435
+ self.name.replace(".", "_")))
436
dbus.service.Object.__init__(self, bus,
437
self.dbus_object_path)
438
def enable(self):
439
oldstate = getattr(self, "enabled", False)
440
r = Client.enable(self)
441
if oldstate != self.enabled:
442
# Emit D-Bus signals
443
self.PropertyChanged(dbus.String(u"enabled"),
444
dbus.Boolean(True, variant_level=1))
445
self.PropertyChanged(dbus.String(u"last_enabled"),
446
(_datetime_to_dbus(self.last_enabled,
447
variant_level=1)))
448
return r
449
450
def disable(self, signal = True):
451
oldstate = getattr(self, "enabled", False)
452
r = Client.disable(self)
453
if signal and oldstate != self.enabled:
454
# Emit D-Bus signal
455
self.PropertyChanged(dbus.String(u"enabled"),
456
dbus.Boolean(False, variant_level=1))
457
return r
458
459
def __del__(self, *args, **kwargs):
460
try:
461
self.remove_from_connection()
462
except LookupError:
463
pass
464
if hasattr(dbus.service.Object, "__del__"):
465
dbus.service.Object.__del__(self, *args, **kwargs)
466
Client.__del__(self, *args, **kwargs)
467
468
def checker_callback(self, pid, condition, command,
469
*args, **kwargs):
470
self.checker_callback_tag = None
471
self.checker = None
472
# Emit D-Bus signal
473
self.PropertyChanged(dbus.String(u"checker_running"),
474
dbus.Boolean(False, variant_level=1))
475
if os.WIFEXITED(condition):
476
exitstatus = os.WEXITSTATUS(condition)
477
# Emit D-Bus signal
478
self.CheckerCompleted(dbus.Int16(exitstatus),
479
dbus.Int64(condition),
480
dbus.String(command))
481
else:
482
# Emit D-Bus signal
483
self.CheckerCompleted(dbus.Int16(-1),
484
dbus.Int64(condition),
485
dbus.String(command))
486
487
return Client.checker_callback(self, pid, condition, command,
488
*args, **kwargs)
489
490
def checked_ok(self, *args, **kwargs):
491
r = Client.checked_ok(self, *args, **kwargs)
492
# Emit D-Bus signal
493
self.PropertyChanged(
494
dbus.String(u"last_checked_ok"),
495
(_datetime_to_dbus(self.last_checked_ok,
496
variant_level=1)))
497
return r
498
499
def start_checker(self, *args, **kwargs):
500
old_checker = self.checker
501
if self.checker is not None:
502
old_checker_pid = self.checker.pid
503
else:
504
old_checker_pid = None
505
r = Client.start_checker(self, *args, **kwargs)
506
# Only if new checker process was started
507
if (self.checker is not None
508
and old_checker_pid != self.checker.pid):
509
# Emit D-Bus signal
510
self.CheckerStarted(self.current_checker_command)
511
self.PropertyChanged(
512
dbus.String("checker_running"),
513
dbus.Boolean(True, variant_level=1))
514
return r
515
516
def stop_checker(self, *args, **kwargs):
517
old_checker = getattr(self, "checker", None)
518
r = Client.stop_checker(self, *args, **kwargs)
519
if (old_checker is not None
520
and getattr(self, "checker", None) is None):
521
self.PropertyChanged(dbus.String(u"checker_running"),
522
dbus.Boolean(False, variant_level=1))
523
return r
524
525
## D-Bus methods & signals
526
_interface = u"se.bsnet.fukt.Mandos.Client"
527
528
# CheckedOK - method
529
CheckedOK = dbus.service.method(_interface)(checked_ok)
530
CheckedOK.__name__ = "CheckedOK"
531
532
# CheckerCompleted - signal
533
@dbus.service.signal(_interface, signature="nxs")
534
def CheckerCompleted(self, exitcode, waitstatus, command):
535
"D-Bus signal"
536
pass
537
538
# CheckerStarted - signal
539
@dbus.service.signal(_interface, signature="s")
540
def CheckerStarted(self, command):
541
"D-Bus signal"
542
pass
543
544
# GetAllProperties - method
545
@dbus.service.method(_interface, out_signature="a{sv}")
546
def GetAllProperties(self):
547
"D-Bus method"
548
return dbus.Dictionary({
549
dbus.String("name"):
550
dbus.String(self.name, variant_level=1),
551
dbus.String("fingerprint"):
552
dbus.String(self.fingerprint, variant_level=1),
553
dbus.String("host"):
554
dbus.String(self.host, variant_level=1),
555
dbus.String("created"):
556
_datetime_to_dbus(self.created, variant_level=1),
557
dbus.String("last_enabled"):
558
(_datetime_to_dbus(self.last_enabled,
559
variant_level=1)
560
if self.last_enabled is not None
561
else dbus.Boolean(False, variant_level=1)),
562
dbus.String("enabled"):
563
dbus.Boolean(self.enabled, variant_level=1),
564
dbus.String("last_checked_ok"):
565
(_datetime_to_dbus(self.last_checked_ok,
566
variant_level=1)
567
if self.last_checked_ok is not None
568
else dbus.Boolean (False, variant_level=1)),
569
dbus.String("timeout"):
570
dbus.UInt64(self.timeout_milliseconds(),
571
variant_level=1),
572
dbus.String("interval"):
573
dbus.UInt64(self.interval_milliseconds(),
574
variant_level=1),
575
dbus.String("checker"):
576
dbus.String(self.checker_command,
577
variant_level=1),
578
dbus.String("checker_running"):
579
dbus.Boolean(self.checker is not None,
580
variant_level=1),
581
dbus.String("object_path"):
582
dbus.ObjectPath(self.dbus_object_path,
583
variant_level=1)
584
}, signature="sv")
585
586
# IsStillValid - method
587
@dbus.service.method(_interface, out_signature="b")
588
def IsStillValid(self):
589
return self.still_valid()
590
591
# PropertyChanged - signal
592
@dbus.service.signal(_interface, signature="sv")
593
def PropertyChanged(self, property, value):
594
"D-Bus signal"
595
pass
596
597
# ReceivedSecret - signal
598
@dbus.service.signal(_interface)
599
def ReceivedSecret(self):
600
"D-Bus signal"
601
pass
602
603
# Rejected - signal
604
@dbus.service.signal(_interface)
605
def Rejected(self):
606
"D-Bus signal"
607
pass
608
609
# SetChecker - method
610
@dbus.service.method(_interface, in_signature="s")
611
def SetChecker(self, checker):
612
"D-Bus setter method"
613
self.checker_command = checker
614
# Emit D-Bus signal
615
self.PropertyChanged(dbus.String(u"checker"),
616
dbus.String(self.checker_command,
617
variant_level=1))
618
619
# SetHost - method
620
@dbus.service.method(_interface, in_signature="s")
621
def SetHost(self, host):
622
"D-Bus setter method"
623
self.host = host
624
# Emit D-Bus signal
625
self.PropertyChanged(dbus.String(u"host"),
626
dbus.String(self.host, variant_level=1))
627
628
# SetInterval - method
629
@dbus.service.method(_interface, in_signature="t")
630
def SetInterval(self, milliseconds):
631
self.interval = datetime.timedelta(0, 0, 0, milliseconds)
632
# Emit D-Bus signal
633
self.PropertyChanged(dbus.String(u"interval"),
634
(dbus.UInt64(self.interval_milliseconds(),
635
variant_level=1)))
636
637
# SetSecret - method
638
@dbus.service.method(_interface, in_signature="ay",
639
byte_arrays=True)
640
def SetSecret(self, secret):
641
"D-Bus setter method"
642
self.secret = str(secret)
643
644
# SetTimeout - method
645
@dbus.service.method(_interface, in_signature="t")
646
def SetTimeout(self, milliseconds):
647
self.timeout = datetime.timedelta(0, 0, 0, milliseconds)
648
# Emit D-Bus signal
649
self.PropertyChanged(dbus.String(u"timeout"),
650
(dbus.UInt64(self.timeout_milliseconds(),
651
variant_level=1)))
652
653
# Enable - method
654
Enable = dbus.service.method(_interface)(enable)
655
Enable.__name__ = "Enable"
656
657
# StartChecker - method
658
@dbus.service.method(_interface)
659
def StartChecker(self):
660
"D-Bus method"
661
self.start_checker()
662
663
# Disable - method
664
@dbus.service.method(_interface)
665
def Disable(self):
666
"D-Bus method"
667
self.disable()
668
669
# StopChecker - method
670
StopChecker = dbus.service.method(_interface)(stop_checker)
671
StopChecker.__name__ = "StopChecker"
672
673
del _interface
674
675
376
676
def peer_certificate(session):
377
677
"Return the peer's OpenPGP certificate as a bytestring"
378
678
# If not an OpenPGP certificate...
379
if gnutls.library.functions.gnutls_certificate_type_get\
380
(session._c_object) \
381
!= gnutls.library.constants.GNUTLS_CRT_OPENPGP:
679
if (gnutls.library.functions
680
.gnutls_certificate_type_get(session._c_object)
681
!= gnutls.library.constants.GNUTLS_CRT_OPENPGP):
382
682
# ...do the normal thing
383
683
return session.peer_certificate
384
list_size = ctypes.c_uint()
385
cert_list = gnutls.library.functions.gnutls_certificate_get_peers\
386
(session._c_object, ctypes.byref(list_size))
684
list_size = ctypes.c_uint(1)
685
cert_list = (gnutls.library.functions
686
.gnutls_certificate_get_peers
687
(session._c_object, ctypes.byref(list_size)))
688
if not bool(cert_list) and list_size.value != 0:
689
raise gnutls.errors.GNUTLSError("error getting peer"
690
" certificate")
387
691
if list_size.value == 0:
388
692
return None
389
693
cert = cert_list[0]
393
697
def fingerprint(openpgp):
394
698
"Convert an OpenPGP bytestring to a hexdigit fingerprint string"
395
699
# New GnuTLS "datum" with the OpenPGP public key
396
datum = gnutls.library.types.gnutls_datum_t\
397
(ctypes.cast(ctypes.c_char_p(openpgp),
398
ctypes.POINTER(ctypes.c_ubyte)),
399
ctypes.c_uint(len(openpgp)))
700
datum = (gnutls.library.types
701
.gnutls_datum_t(ctypes.cast(ctypes.c_char_p(openpgp),
702
ctypes.POINTER
703
(ctypes.c_ubyte)),
704
ctypes.c_uint(len(openpgp))))
400
705
# New empty GnuTLS certificate
401
706
crt = gnutls.library.types.gnutls_openpgp_crt_t()
402
gnutls.library.functions.gnutls_openpgp_crt_init\
403
(ctypes.byref(crt))
707
(gnutls.library.functions
708
.gnutls_openpgp_crt_init(ctypes.byref(crt)))
404
709
# Import the OpenPGP public key into the certificate
405
gnutls.library.functions.gnutls_openpgp_crt_import\
406
(crt, ctypes.byref(datum),
407
gnutls.library.constants.GNUTLS_OPENPGP_FMT_RAW)
710
(gnutls.library.functions
711
.gnutls_openpgp_crt_import(crt, ctypes.byref(datum),
712
gnutls.library.constants
713
.GNUTLS_OPENPGP_FMT_RAW))
714
# Verify the self signature in the key
715
crtverify = ctypes.c_uint()
716
(gnutls.library.functions
717
.gnutls_openpgp_crt_verify_self(crt, 0, ctypes.byref(crtverify)))
718
if crtverify.value != 0:
719
gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
720
raise gnutls.errors.CertificateSecurityError("Verify failed")
408
721
# New buffer for the fingerprint
409
buffer = ctypes.create_string_buffer(20)
410
buffer_length = ctypes.c_size_t()
722
buf = ctypes.create_string_buffer(20)
723
buf_len = ctypes.c_size_t()
411
724
# Get the fingerprint from the certificate into the buffer
412
gnutls.library.functions.gnutls_openpgp_crt_get_fingerprint\
413
(crt, ctypes.byref(buffer), ctypes.byref(buffer_length))
725
(gnutls.library.functions
726
.gnutls_openpgp_crt_get_fingerprint(crt, ctypes.byref(buf),
727
ctypes.byref(buf_len)))
414
728
# Deinit the certificate
415
729
gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
416
730
# Convert the buffer to a Python bytestring
417
fpr = ctypes.string_at(buffer, buffer_length.value)
731
fpr = ctypes.string_at(buf, buf_len.value)
418
732
# Convert the bytestring to hexadecimal notation
419
733
hex_fpr = u''.join(u"%02X" % ord(char) for char in fpr)
420
734
return hex_fpr
421
735
422
736
423
class tcp_handler(SocketServer.BaseRequestHandler, object):
737
class TCP_handler(SocketServer.BaseRequestHandler, object):
424
738
"""A TCP request handler class.
425
739
Instantiated by IPv6_TCPServer for each request to handle it.
426
740
Note: This will run in its own forked process."""
427
741
428
742
def handle(self):
429
743
logger.info(u"TCP connection from: %s",
430
unicode(self.client_address))
431
session = gnutls.connection.ClientSession\
432
(self.request, gnutls.connection.X509Credentials())
433
434
line = self.request.makefile().readline()
435
logger.debug(u"Protocol version: %r", line)
436
try:
437
if int(line.strip().split()[0]) > 1:
438
raise RuntimeError
439
except (ValueError, IndexError, RuntimeError), error:
440
logger.error(u"Unknown protocol version: %s", error)
441
return
442
443
# Note: gnutls.connection.X509Credentials is really a generic
444
# GnuTLS certificate credentials object so long as no X.509
445
# keys are added to it. Therefore, we can use it here despite
446
# using OpenPGP certificates.
447
448
#priority = ':'.join(("NONE", "+VERS-TLS1.1", "+AES-256-CBC",
449
# "+SHA1", "+COMP-NULL", "+CTYPE-OPENPGP",
450
# "+DHE-DSS"))
451
priority = "NORMAL" # Fallback default, since this
452
# MUST be set.
453
if self.server.settings["priority"]:
454
priority = self.server.settings["priority"]
455
gnutls.library.functions.gnutls_priority_set_direct\
456
(session._c_object, priority, None);
457
458
try:
459
session.handshake()
460
except gnutls.errors.GNUTLSError, error:
461
logger.warning(u"Handshake failed: %s", error)
462
# Do not run session.bye() here: the session is not
463
# established. Just abandon the request.
464
return
465
try:
466
fpr = fingerprint(peer_certificate(session))
467
except (TypeError, gnutls.errors.GNUTLSError), error:
468
logger.warning(u"Bad certificate: %s", error)
469
session.bye()
470
return
471
logger.debug(u"Fingerprint: %s", fpr)
472
client = None
473
for c in self.server.clients:
474
if c.fingerprint == fpr:
475
client = c
476
break
477
if not client:
478
logger.warning(u"Client not found for fingerprint: %s",
479
fpr)
480
session.bye()
481
return
482
# Have to check if client.still_valid(), since it is possible
483
# that the client timed out while establishing the GnuTLS
484
# session.
485
if not client.still_valid():
486
logger.warning(u"Client %(name)s is invalid",
487
vars(client))
488
session.bye()
489
return
490
sent_size = 0
491
while sent_size < len(client.secret):
492
sent = session.send(client.secret[sent_size:])
493
logger.debug(u"Sent: %d, remaining: %d",
494
sent, len(client.secret)
495
- (sent_size + sent))
496
sent_size += sent
497
session.bye()
498
499
500
class IPv6_TCPServer(SocketServer.ForkingTCPServer, object):
501
"""IPv6 TCP server. Accepts 'None' as address and/or port.
744
unicode(self.client_address))
745
logger.debug(u"IPC Pipe FD: %d", self.server.pipe[1])
746
# Open IPC pipe to parent process
747
with closing(os.fdopen(self.server.pipe[1], "w", 1)) as ipc:
748
session = (gnutls.connection
749
.ClientSession(self.request,
750
gnutls.connection
751
.X509Credentials()))
752
753
line = self.request.makefile().readline()
754
logger.debug(u"Protocol version: %r", line)
755
try:
756
if int(line.strip().split()[0]) > 1:
757
raise RuntimeError
758
except (ValueError, IndexError, RuntimeError), error:
759
logger.error(u"Unknown protocol version: %s", error)
760
return
761
762
# Note: gnutls.connection.X509Credentials is really a
763
# generic GnuTLS certificate credentials object so long as
764
# no X.509 keys are added to it. Therefore, we can use it
765
# here despite using OpenPGP certificates.
766
767
#priority = ':'.join(("NONE", "+VERS-TLS1.1",
768
# "+AES-256-CBC", "+SHA1",
769
# "+COMP-NULL", "+CTYPE-OPENPGP",
770
# "+DHE-DSS"))
771
# Use a fallback default, since this MUST be set.
772
priority = self.server.settings.get("priority", "NORMAL")
773
(gnutls.library.functions
774
.gnutls_priority_set_direct(session._c_object,
775
priority, None))
776
777
try:
778
session.handshake()
779
except gnutls.errors.GNUTLSError, error:
780
logger.warning(u"Handshake failed: %s", error)
781
# Do not run session.bye() here: the session is not
782
# established. Just abandon the request.
783
return
784
logger.debug(u"Handshake succeeded")
785
try:
786
fpr = fingerprint(peer_certificate(session))
787
except (TypeError, gnutls.errors.GNUTLSError), error:
788
logger.warning(u"Bad certificate: %s", error)
789
session.bye()
790
return
791
logger.debug(u"Fingerprint: %s", fpr)
792
793
for c in self.server.clients:
794
if c.fingerprint == fpr:
795
client = c
796
break
797
else:
798
logger.warning(u"Client not found for fingerprint: %s",
799
fpr)
800
ipc.write("NOTFOUND %s\n" % fpr)
801
session.bye()
802
return
803
# Have to check if client.still_valid(), since it is
804
# possible that the client timed out while establishing
805
# the GnuTLS session.
806
if not client.still_valid():
807
logger.warning(u"Client %(name)s is invalid",
808
vars(client))
809
ipc.write("INVALID %s\n" % client.name)
810
session.bye()
811
return
812
ipc.write("SENDING %s\n" % client.name)
813
sent_size = 0
814
while sent_size < len(client.secret):
815
sent = session.send(client.secret[sent_size:])
816
logger.debug(u"Sent: %d, remaining: %d",
817
sent, len(client.secret)
818
- (sent_size + sent))
819
sent_size += sent
820
session.bye()
821
822
823
class ForkingMixInWithPipe(SocketServer.ForkingMixIn, object):
824
"""Like SocketServer.ForkingMixIn, but also pass a pipe.
825
Assumes a gobject.MainLoop event loop.
826
"""
827
def process_request(self, request, client_address):
828
"""This overrides and wraps the original process_request().
829
This function creates a new pipe in self.pipe
830
"""
831
self.pipe = os.pipe()
832
super(ForkingMixInWithPipe,
833
self).process_request(request, client_address)
834
os.close(self.pipe[1]) # close write end
835
# Call "handle_ipc" for both data and EOF events
836
gobject.io_add_watch(self.pipe[0],
837
gobject.IO_IN | gobject.IO_HUP,
838
self.handle_ipc)
839
def handle_ipc(source, condition):
840
"""Dummy function; override as necessary"""
841
os.close(source)
842
return False
843
844
845
class IPv6_TCPServer(ForkingMixInWithPipe,
846
SocketServer.TCPServer, object):
847
"""IPv6-capable TCP server. Accepts 'None' as address and/or port
502
848
Attributes:
503
849
settings: Server settings
504
850
clients: Set() of Client objects
851
enabled: Boolean; whether this server is activated yet
505
852
"""
506
853
address_family = socket.AF_INET6
507
854
def __init__(self, *args, **kwargs):
511
858
if "clients" in kwargs:
512
859
self.clients = kwargs["clients"]
513
860
del kwargs["clients"]
514
return super(type(self), self).__init__(*args, **kwargs)
861
if "use_ipv6" in kwargs:
862
if not kwargs["use_ipv6"]:
863
self.address_family = socket.AF_INET
864
del kwargs["use_ipv6"]
865
self.enabled = False
866
super(IPv6_TCPServer, self).__init__(*args, **kwargs)
515
867
def server_bind(self):
516
868
"""This overrides the normal server_bind() function
517
869
to bind to an interface if one was specified, and also NOT to
529
881
u" bind to interface %s",
530
882
self.settings["interface"])
531
883
else:
532
raise error
884
raise
533
885
# Only bind(2) the socket if we really need to.
534
886
if self.server_address[0] or self.server_address[1]:
535
887
if not self.server_address[0]:
536
in6addr_any = "::"
537
self.server_address = (in6addr_any,
888
if self.address_family == socket.AF_INET6:
889
any_address = "::" # in6addr_any
890
else:
891
any_address = socket.INADDR_ANY
892
self.server_address = (any_address,
538
893
self.server_address[1])
539
894
elif not self.server_address[1]:
540
895
self.server_address = (self.server_address[0],
546
901
# if_nametoindex
547
902
# (self.settings
548
903
# ["interface"]))
549
return super(type(self), self).server_bind()
904
return super(IPv6_TCPServer, self).server_bind()
905
def server_activate(self):
906
if self.enabled:
907
return super(IPv6_TCPServer, self).server_activate()
908
def enable(self):
909
self.enabled = True
910
def handle_ipc(self, source, condition, file_objects={}):
911
condition_names = {
912
gobject.IO_IN: "IN", # There is data to read.
913
gobject.IO_OUT: "OUT", # Data can be written (without
914
# blocking).
915
gobject.IO_PRI: "PRI", # There is urgent data to read.
916
gobject.IO_ERR: "ERR", # Error condition.
917
gobject.IO_HUP: "HUP" # Hung up (the connection has been
918
# broken, usually for pipes and
919
# sockets).
920
}
921
conditions_string = ' | '.join(name
922
for cond, name in
923
condition_names.iteritems()
924
if cond & condition)
925
logger.debug("Handling IPC: FD = %d, condition = %s", source,
926
conditions_string)
927
928
# Turn the pipe file descriptor into a Python file object
929
if source not in file_objects:
930
file_objects[source] = os.fdopen(source, "r", 1)
931
932
# Read a line from the file object
933
cmdline = file_objects[source].readline()
934
if not cmdline: # Empty line means end of file
935
# close the IPC pipe
936
file_objects[source].close()
937
del file_objects[source]
938
939
# Stop calling this function
940
return False
941
942
logger.debug("IPC command: %r\n" % cmdline)
943
944
# Parse and act on command
945
cmd, args = cmdline.split(None, 1)
946
if cmd == "NOTFOUND":
947
if self.settings["use_dbus"]:
948
# Emit D-Bus signal
949
mandos_dbus_service.ClientNotFound(args)
950
elif cmd == "INVALID":
951
if self.settings["use_dbus"]:
952
for client in self.clients:
953
if client.name == args:
954
# Emit D-Bus signal
955
client.Rejected()
956
break
957
elif cmd == "SENDING":
958
for client in self.clients:
959
if client.name == args:
960
client.checked_ok()
961
if self.settings["use_dbus"]:
962
# Emit D-Bus signal
963
client.ReceivedSecret()
964
break
965
else:
966
logger.error("Unknown IPC command: %r", cmdline)
967
968
# Keep calling this function
969
return True
550
970
551
971
552
972
def string_to_delta(interval):
553
973
"""Parse a string and return a datetime.timedelta
554
974
555
975
>>> string_to_delta('7d')
556
976
datetime.timedelta(7)
557
977
>>> string_to_delta('60s')
562
982
datetime.timedelta(1)
563
983
>>> string_to_delta(u'1w')
564
984
datetime.timedelta(7)
985
>>> string_to_delta('5m 30s')
986
datetime.timedelta(0, 330)
565
987
"""
566
try:
567
suffix=unicode(interval[-1])
568
value=int(interval[:-1])
569
if suffix == u"d":
570
delta = datetime.timedelta(value)
571
elif suffix == u"s":
572
delta = datetime.timedelta(0, value)
573
elif suffix == u"m":
574
delta = datetime.timedelta(0, 0, 0, 0, value)
575
elif suffix == u"h":
576
delta = datetime.timedelta(0, 0, 0, 0, 0, value)
577
elif suffix == u"w":
578
delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
579
else:
988
timevalue = datetime.timedelta(0)
989
for s in interval.split():
990
try:
991
suffix = unicode(s[-1])
992
value = int(s[:-1])
993
if suffix == u"d":
994
delta = datetime.timedelta(value)
995
elif suffix == u"s":
996
delta = datetime.timedelta(0, value)
997
elif suffix == u"m":
998
delta = datetime.timedelta(0, 0, 0, 0, value)
999
elif suffix == u"h":
1000
delta = datetime.timedelta(0, 0, 0, 0, 0, value)
1001
elif suffix == u"w":
1002
delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
1003
else:
1004
raise ValueError
1005
except (ValueError, IndexError):
580
1006
raise ValueError
581
except (ValueError, IndexError):
582
raise ValueError
583
return delta
1007
timevalue += delta
1008
return timevalue
584
1009
585
1010
586
1011
def server_state_changed(state):
587
1012
"""Derived from the Avahi example code"""
588
1013
if state == avahi.SERVER_COLLISION:
589
logger.error(u"Server name collision")
1014
logger.error(u"Zeroconf server name collision")
590
1015
service.remove()
591
1016
elif state == avahi.SERVER_RUNNING:
592
1017
service.add()
594
1019
595
1020
def entry_group_state_changed(state, error):
596
1021
"""Derived from the Avahi example code"""
597
logger.debug(u"state change: %i", state)
1022
logger.debug(u"Avahi state change: %i", state)
598
1023
599
1024
if state == avahi.ENTRY_GROUP_ESTABLISHED:
600
logger.debug(u"Service established.")
1025
logger.debug(u"Zeroconf service established.")
601
1026
elif state == avahi.ENTRY_GROUP_COLLISION:
602
logger.warning(u"Service name collision.")
1027
logger.warning(u"Zeroconf service name collision.")
603
1028
service.rename()
604
1029
elif state == avahi.ENTRY_GROUP_FAILURE:
605
logger.critical(u"Error in group state changed %s",
1030
logger.critical(u"Avahi: Error in group state changed %s",
606
1031
unicode(error))
607
raise AvahiGroupError("State changed: %s", str(error))
1032
raise AvahiGroupError(u"State changed: %s" % unicode(error))
608
1033
609
1034
def if_nametoindex(interface):
610
1035
"""Call the C function if_nametoindex(), or equivalent"""
611
1036
global if_nametoindex
612
1037
try:
613
if "ctypes.util" not in sys.modules:
614
import ctypes.util
615
if_nametoindex = ctypes.cdll.LoadLibrary\
616
(ctypes.util.find_library("c")).if_nametoindex
1038
if_nametoindex = (ctypes.cdll.LoadLibrary
1039
(ctypes.util.find_library("c"))
1040
.if_nametoindex)
617
1041
except (OSError, AttributeError):
618
1042
if "struct" not in sys.modules:
619
1043
import struct
622
1046
def if_nametoindex(interface):
623
1047
"Get an interface index the hard way, i.e. using fcntl()"
624
1048
SIOCGIFINDEX = 0x8933 # From /usr/include/linux/sockios.h
625
s = socket.socket()
626
ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
627
struct.pack("16s16x", interface))
628
s.close()
1049
with closing(socket.socket()) as s:
1050
ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
1051
struct.pack("16s16x", interface))
629
1052
interface_index = struct.unpack("I", ifreq[16:20])[0]
630
1053
return interface_index
631
1054
return if_nametoindex(interface)
655
1078
656
1079
657
1080
def main():
658
global main_loop_started
659
main_loop_started = False
660
661
parser = OptionParser(version = "%%prog %s" % version)
1081
1082
######################################################################
1083
# Parsing of options, both command line and config file
1084
1085
parser = optparse.OptionParser(version = "%%prog %s" % version)
662
1086
parser.add_option("-i", "--interface", type="string",
663
1087
metavar="IF", help="Bind to interface IF")
664
1088
parser.add_option("-a", "--address", type="string",
665
1089
help="Address to listen for requests on")
666
1090
parser.add_option("-p", "--port", type="int",
667
1091
help="Port number to receive requests on")
668
parser.add_option("--check", action="store_true", default=False,
1092
parser.add_option("--check", action="store_true",
669
1093
help="Run self-test")
670
1094
parser.add_option("--debug", action="store_true",
671
1095
help="Debug mode; run in foreground and log to"
678
1102
default="/etc/mandos", metavar="DIR",
679
1103
help="Directory to search for configuration"
680
1104
" files")
681
(options, args) = parser.parse_args()
1105
parser.add_option("--no-dbus", action="store_false",
1106
dest="use_dbus",
1107
help="Do not provide D-Bus system bus"
1108
" interface")
1109
parser.add_option("--no-ipv6", action="store_false",
1110
dest="use_ipv6", help="Do not use IPv6")
1111
options = parser.parse_args()[0]
682
1112
683
1113
if options.check:
684
1114
import doctest
693
1123
"priority":
694
1124
"SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
695
1125
"servicename": "Mandos",
1126
"use_dbus": "True",
1127
"use_ipv6": "True",
696
1128
}
697
1129
698
1130
# Parse config file for server-global settings
699
1131
server_config = ConfigParser.SafeConfigParser(server_defaults)
700
1132
del server_defaults
701
1133
server_config.read(os.path.join(options.configdir, "mandos.conf"))
702
server_section = "server"
703
1134
# Convert the SafeConfigParser object to a dict
704
server_settings = dict(server_config.items(server_section))
705
# Use getboolean on the boolean config option
706
server_settings["debug"] = server_config.getboolean\
707
(server_section, "debug")
1135
server_settings = server_config.defaults()
1136
# Use the appropriate methods on the non-string config options
1137
server_settings["debug"] = server_config.getboolean("DEFAULT",
1138
"debug")
1139
server_settings["use_dbus"] = server_config.getboolean("DEFAULT",
1140
"use_dbus")
1141
server_settings["use_ipv6"] = server_config.getboolean("DEFAULT",
1142
"use_ipv6")
1143
if server_settings["port"]:
1144
server_settings["port"] = server_config.getint("DEFAULT",
1145
"port")
708
1146
del server_config
709
1147
710
1148
# Override the settings from the config file with command line
711
1149
# options, if set.
712
1150
for option in ("interface", "address", "port", "debug",
713
"priority", "servicename", "configdir"):
1151
"priority", "servicename", "configdir",
1152
"use_dbus", "use_ipv6"):
714
1153
value = getattr(options, option)
715
1154
if value is not None:
716
1155
server_settings[option] = value
717
1156
del options
718
1157
# Now we have our good server settings in "server_settings"
719
1158
1159
##################################################################
1160
1161
# For convenience
720
1162
debug = server_settings["debug"]
1163
use_dbus = server_settings["use_dbus"]
1164
use_ipv6 = server_settings["use_ipv6"]
721
1165
722
1166
if not debug:
723
1167
syslogger.setLevel(logging.WARNING)
724
1168
console.setLevel(logging.WARNING)
725
1169
726
1170
if server_settings["servicename"] != "Mandos":
727
syslogger.setFormatter(logging.Formatter\
728
('Mandos (%s): %%(levelname)s:'
729
' %%(message)s'
1171
syslogger.setFormatter(logging.Formatter
1172
('Mandos (%s) [%%(process)d]:'
1173
' %%(levelname)s: %%(message)s'
730
1174
% server_settings["servicename"]))
731
1175
732
1176
# Parse config file with clients
733
1177
client_defaults = { "timeout": "1h",
734
1178
"interval": "5m",
735
1179
"checker": "fping -q -- %%(host)s",
1180
"host": "",
736
1181
}
737
1182
client_config = ConfigParser.SafeConfigParser(client_defaults)
738
1183
client_config.read(os.path.join(server_settings["configdir"],
739
1184
"clients.conf"))
1185
1186
global mandos_dbus_service
1187
mandos_dbus_service = None
1188
1189
clients = Set()
1190
tcp_server = IPv6_TCPServer((server_settings["address"],
1191
server_settings["port"]),
1192
TCP_handler,
1193
settings=server_settings,
1194
clients=clients, use_ipv6=use_ipv6)
1195
pidfilename = "/var/run/mandos.pid"
1196
try:
1197
pidfile = open(pidfilename, "w")
1198
except IOError:
1199
logger.error("Could not open file %r", pidfilename)
1200
1201
try:
1202
uid = pwd.getpwnam("_mandos").pw_uid
1203
gid = pwd.getpwnam("_mandos").pw_gid
1204
except KeyError:
1205
try:
1206
uid = pwd.getpwnam("mandos").pw_uid
1207
gid = pwd.getpwnam("mandos").pw_gid
1208
except KeyError:
1209
try:
1210
uid = pwd.getpwnam("nobody").pw_uid
1211
gid = pwd.getpwnam("nogroup").pw_gid
1212
except KeyError:
1213
uid = 65534
1214
gid = 65534
1215
try:
1216
os.setgid(gid)
1217
os.setuid(uid)
1218
except OSError, error:
1219
if error[0] != errno.EPERM:
1220
raise error
1221
1222
# Enable all possible GnuTLS debugging
1223
if debug:
1224
# "Use a log level over 10 to enable all debugging options."
1225
# - GnuTLS manual
1226
gnutls.library.functions.gnutls_global_set_log_level(11)
1227
1228
@gnutls.library.types.gnutls_log_func
1229
def debug_gnutls(level, string):
1230
logger.debug("GnuTLS: %s", string[:-1])
1231
1232
(gnutls.library.functions
1233
.gnutls_global_set_log_function(debug_gnutls))
740
1234
741
1235
global service
1236
protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
742
1237
service = AvahiService(name = server_settings["servicename"],
743
type = "_mandos._tcp", );
1238
servicetype = "_mandos._tcp",
1239
protocol = protocol)
744
1240
if server_settings["interface"]:
745
service.interface = if_nametoindex(server_settings["interface"])
1241
service.interface = (if_nametoindex
1242
(server_settings["interface"]))
746
1243
747
1244
global main_loop
748
1245
global bus
751
1248
DBusGMainLoop(set_as_default=True )
752
1249
main_loop = gobject.MainLoop()
753
1250
bus = dbus.SystemBus()
754
server = dbus.Interface(
755
bus.get_object( avahi.DBUS_NAME, avahi.DBUS_PATH_SERVER ),
756
avahi.DBUS_INTERFACE_SERVER )
1251
server = dbus.Interface(bus.get_object(avahi.DBUS_NAME,
1252
avahi.DBUS_PATH_SERVER),
1253
avahi.DBUS_INTERFACE_SERVER)
757
1254
# End of Avahi example code
758
759
clients = Set()
760
def remove_from_clients(client):
761
clients.remove(client)
762
if not clients:
763
logger.critical(u"No clients left, exiting")
764
sys.exit()
765
766
clients.update(Set(Client(name = section,
767
stop_hook = remove_from_clients,
768
config
769
= dict(client_config.items(section)))
770
for section in client_config.sections()))
1255
if use_dbus:
1256
bus_name = dbus.service.BusName(u"se.bsnet.fukt.Mandos", bus)
1257
1258
client_class = Client
1259
if use_dbus:
1260
client_class = ClientDBus
1261
clients.update(Set(
1262
client_class(name = section,
1263
config= dict(client_config.items(section)))
1264
for section in client_config.sections()))
771
1265
if not clients:
772
logger.critical(u"No clients defined")
773
sys.exit(1)
1266
logger.warning(u"No clients defined")
774
1267
775
if not debug:
1268
if debug:
1269
# Redirect stdin so all checkers get /dev/null
1270
null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
1271
os.dup2(null, sys.stdin.fileno())
1272
if null > 2:
1273
os.close(null)
1274
else:
1275
# No console logging
776
1276
logger.removeHandler(console)
1277
# Close all input and output, do double fork, etc.
777
1278
daemon()
778
1279
779
pidfilename = "/var/run/mandos/mandos.pid"
780
pid = os.getpid()
781
1280
try:
782
pidfile = open(pidfilename, "w")
783
pidfile.write(str(pid) + "\n")
784
pidfile.close()
1281
with closing(pidfile):
1282
pid = os.getpid()
1283
pidfile.write(str(pid) + "\n")
785
1284
del pidfile
786
except IOError, err:
787
logger.error(u"Could not write %s file with PID %d",
788
pidfilename, os.getpid())
1285
except IOError:
1286
logger.error(u"Could not write to file %r with PID %d",
1287
pidfilename, pid)
1288
except NameError:
1289
# "pidfile" was never created
1290
pass
1291
del pidfilename
789
1292
790
1293
def cleanup():
791
1294
"Cleanup function; run on exit"
798
1301
799
1302
while clients:
800
1303
client = clients.pop()
801
client.stop_hook = None
802
client.stop()
1304
client.disable_hook = None
1305
client.disable()
803
1306
804
1307
atexit.register(cleanup)
805
1308
808
1311
signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
809
1312
signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
810
1313
1314
if use_dbus:
1315
class MandosDBusService(dbus.service.Object):
1316
"""A D-Bus proxy object"""
1317
def __init__(self):
1318
dbus.service.Object.__init__(self, bus, "/")
1319
_interface = u"se.bsnet.fukt.Mandos"
1320
1321
@dbus.service.signal(_interface, signature="oa{sv}")
1322
def ClientAdded(self, objpath, properties):
1323
"D-Bus signal"
1324
pass
1325
1326
@dbus.service.signal(_interface, signature="s")
1327
def ClientNotFound(self, fingerprint):
1328
"D-Bus signal"
1329
pass
1330
1331
@dbus.service.signal(_interface, signature="os")
1332
def ClientRemoved(self, objpath, name):
1333
"D-Bus signal"
1334
pass
1335
1336
@dbus.service.method(_interface, out_signature="ao")
1337
def GetAllClients(self):
1338
"D-Bus method"
1339
return dbus.Array(c.dbus_object_path for c in clients)
1340
1341
@dbus.service.method(_interface, out_signature="a{oa{sv}}")
1342
def GetAllClientsWithProperties(self):
1343
"D-Bus method"
1344
return dbus.Dictionary(
1345
((c.dbus_object_path, c.GetAllProperties())
1346
for c in clients),
1347
signature="oa{sv}")
1348
1349
@dbus.service.method(_interface, in_signature="o")
1350
def RemoveClient(self, object_path):
1351
"D-Bus method"
1352
for c in clients:
1353
if c.dbus_object_path == object_path:
1354
clients.remove(c)
1355
c.remove_from_connection()
1356
# Don't signal anything except ClientRemoved
1357
c.disable(signal=False)
1358
# Emit D-Bus signal
1359
self.ClientRemoved(object_path, c.name)
1360
return
1361
raise KeyError
1362
1363
del _interface
1364
1365
mandos_dbus_service = MandosDBusService()
1366
811
1367
for client in clients:
812
client.start()
813
814
tcp_server = IPv6_TCPServer((server_settings["address"],
815
server_settings["port"]),
816
tcp_handler,
817
settings=server_settings,
818
clients=clients)
1368
if use_dbus:
1369
# Emit D-Bus signal
1370
mandos_dbus_service.ClientAdded(client.dbus_object_path,
1371
client.GetAllProperties())
1372
client.enable()
1373
1374
tcp_server.enable()
1375
tcp_server.server_activate()
1376
819
1377
# Find out what port we got
820
1378
service.port = tcp_server.socket.getsockname()[1]
821
logger.info(u"Now listening on address %r, port %d, flowinfo %d,"
822
u" scope_id %d" % tcp_server.socket.getsockname())
1379
if use_ipv6:
1380
logger.info(u"Now listening on address %r, port %d,"
1381
" flowinfo %d, scope_id %d"
1382
% tcp_server.socket.getsockname())
1383
else: # IPv4
1384
logger.info(u"Now listening on address %r, port %d"
1385
% tcp_server.socket.getsockname())
823
1386
824
1387
#service.interface = tcp_server.socket.getsockname()[3]
825
1388
835
1398
836
1399
gobject.io_add_watch(tcp_server.fileno(), gobject.IO_IN,
837
1400
lambda *args, **kwargs:
838
tcp_server.handle_request\
839
(*args[2:], **kwargs) or True)
1401
(tcp_server.handle_request
1402
(*args[2:], **kwargs) or True))
840
1403
841
1404
logger.debug(u"Starting main loop")
842
main_loop_started = True
843
1405
main_loop.run()
844
1406
except AvahiError, error:
845
logger.critical(u"AvahiError: %s" + unicode(error))
1407
logger.critical(u"AvahiError: %s", error)
846
1408
sys.exit(1)
847
1409
except KeyboardInterrupt:
848
1410
if debug:
849
print
1411
print >> sys.stderr
1412
logger.debug("Server received KeyboardInterrupt")
1413
logger.debug("Server exiting")
850
1414
851
1415
if __name__ == '__main__':
852
1416
main()
Loggerhead is a web-based interface for Breezy
Version: 2.0.2.dev0