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