51
22
from sets import Set
56
import logging.handlers
58
from contextlib import closing
64
28
from dbus.mainloop.glib import DBusGMainLoop
32
import logging.handlers
34
# logghandler.setFormatter(logging.Formatter('%(levelname)s %(message)s')
70
36
logger = logging.Logger('mandos')
71
syslogger = logging.handlers.SysLogHandler\
72
(facility = logging.handlers.SysLogHandler.LOG_DAEMON,
74
syslogger.setFormatter(logging.Formatter\
75
('Mandos: %(levelname)s: %(message)s'))
76
logger.addHandler(syslogger)
78
console = logging.StreamHandler()
79
console.setFormatter(logging.Formatter('%(name)s: %(levelname)s:'
81
logger.addHandler(console)
83
class AvahiError(Exception):
84
def __init__(self, value):
86
super(AvahiError, self).__init__()
88
return repr(self.value)
90
class AvahiServiceError(AvahiError):
93
class AvahiGroupError(AvahiError):
97
class AvahiService(object):
98
"""An Avahi (Zeroconf) service.
100
interface: integer; avahi.IF_UNSPEC or an interface index.
101
Used to optionally bind to the specified interface.
102
name: string; Example: 'Mandos'
103
type: string; Example: '_mandos._tcp'.
104
See <http://www.dns-sd.org/ServiceTypes.html>
105
port: integer; what port to announce
106
TXT: list of strings; TXT record for the service
107
domain: string; Domain to publish on, default to .local if empty.
108
host: string; Host to publish records for, default is localhost
109
max_renames: integer; maximum number of renames
110
rename_count: integer; counter so we only rename after collisions
111
a sensible number of times
113
def __init__(self, interface = avahi.IF_UNSPEC, name = None,
114
servicetype = None, port = None, TXT = None, domain = "",
115
host = "", max_renames = 32768):
116
self.interface = interface
118
self.type = servicetype
126
self.rename_count = 0
127
self.max_renames = max_renames
129
"""Derived from the Avahi example code"""
130
if self.rename_count >= self.max_renames:
131
logger.critical(u"No suitable Zeroconf service name found"
132
u" after %i retries, exiting.",
134
raise AvahiServiceError("Too many renames")
135
self.name = server.GetAlternativeServiceName(self.name)
136
logger.info(u"Changing Zeroconf service name to %r ...",
138
syslogger.setFormatter(logging.Formatter\
139
('Mandos (%s): %%(levelname)s:'
140
' %%(message)s' % self.name))
143
self.rename_count += 1
145
"""Derived from the Avahi example code"""
146
if group is not None:
149
"""Derived from the Avahi example code"""
152
group = dbus.Interface\
153
(bus.get_object(avahi.DBUS_NAME,
154
server.EntryGroupNew()),
155
avahi.DBUS_INTERFACE_ENTRY_GROUP)
156
group.connect_to_signal('StateChanged',
157
entry_group_state_changed)
158
logger.debug(u"Adding Zeroconf service '%s' of type '%s' ...",
159
service.name, service.type)
161
self.interface, # interface
162
avahi.PROTO_INET6, # protocol
163
dbus.UInt32(0), # flags
164
self.name, self.type,
165
self.domain, self.host,
166
dbus.UInt16(self.port),
167
avahi.string_array_to_txt_array(self.TXT))
170
# From the Avahi example code:
171
group = None # our entry group
37
logger.addHandler(logging.handlers.SysLogHandler(facility = logging.handlers.SysLogHandler.LOG_DAEMON))
39
# This variable is used to optionally bind to a specified interface.
40
# It is a global variable to fit in with the other variables from the
41
# Avahi server example code.
42
serviceInterface = avahi.IF_UNSPEC
43
# From the Avahi server example code:
44
serviceName = "Mandos"
45
serviceType = "_mandos._tcp" # http://www.dns-sd.org/ServiceTypes.html
46
servicePort = None # Not known at startup
47
serviceTXT = [] # TXT record for the service
48
domain = "" # Domain to publish on, default to .local
49
host = "" # Host to publish records for, default to localhost
50
group = None #our entry group
51
rename_count = 12 # Counter so we only rename after collisions a
52
# sensible number of times
172
53
# End of Avahi example code
175
class Client(dbus.service.Object):
176
57
"""A representation of a client host served by this server.
178
59
name: string; from the config file, used in log messages
179
60
fingerprint: string (40 or 32 hexadecimal digits); used to
180
61
uniquely identify the client
181
62
secret: bytestring; sent verbatim (over TLS) to client
182
host: string; available for use by the checker command
183
created: datetime.datetime(); object creation, not client host
185
last_checked_ok: datetime.datetime() or None if not yet checked OK
186
timeout: datetime.timedelta(); How long from last_checked_ok
187
until this client is invalid
63
fqdn: string (FQDN); available for use by the checker command
64
created: datetime.datetime()
65
last_seen: datetime.datetime() or None if not yet seen
66
timeout: datetime.timedelta(); How long from last_seen until
67
this client is invalid
188
68
interval: datetime.timedelta(); How often to start a new checker
189
69
stop_hook: If set, called by stop() as stop_hook(self)
190
70
checker: subprocess.Popen(); a running checker process used
191
71
to see if the client lives.
192
'None' if no process is running.
72
Is None if no process is running.
193
73
checker_initiator_tag: a gobject event source tag, or None
194
74
stop_initiator_tag: - '' -
195
75
checker_callback_tag: - '' -
196
76
checker_command: string; External command which is run to check if
197
client lives. %() expansions are done at
77
client lives. %()s expansions are done at
198
78
runtime with vars(self) as dict, so that for
199
79
instance %(name)s can be used in the command.
200
80
Private attibutes:
201
81
_timeout: Real variable for 'timeout'
202
82
_interval: Real variable for 'interval'
203
_timeout_milliseconds: Used when calling gobject.timeout_add()
83
_timeout_milliseconds: Used by gobject.timeout_add()
204
84
_interval_milliseconds: - '' -
206
interface = u"org.mandos_system.Mandos.Clients"
208
@dbus.service.method(interface, out_signature="s")
210
"D-Bus getter method"
213
@dbus.service.method(interface, out_signature="s")
214
def getFingerprint(self):
215
"D-Bus getter method"
216
return self.fingerprint
218
@dbus.service.method(interface, in_signature="ay",
220
def setSecret(self, secret):
221
"D-Bus setter method"
224
86
def _set_timeout(self, timeout):
225
"Setter function for the 'timeout' attribute"
87
"Setter function for 'timeout' attribute"
226
88
self._timeout = timeout
227
89
self._timeout_milliseconds = ((self.timeout.days
228
90
* 24 * 60 * 60 * 1000)
229
91
+ (self.timeout.seconds * 1000)
230
92
+ (self.timeout.microseconds
233
self.TimeoutChanged(self._timeout_milliseconds)
234
timeout = property(lambda self: self._timeout, _set_timeout)
94
timeout = property(lambda self: self._timeout,
237
@dbus.service.method(interface, out_signature="t")
238
def getTimeout(self):
239
"D-Bus getter method"
240
return self._timeout_milliseconds
242
@dbus.service.signal(interface, signature="t")
243
def TimeoutChanged(self, t):
247
97
def _set_interval(self, interval):
248
"Setter function for the 'interval' attribute"
98
"Setter function for 'interval' attribute"
249
99
self._interval = interval
250
100
self._interval_milliseconds = ((self.interval.days
251
101
* 24 * 60 * 60 * 1000)
254
104
+ (self.interval.microseconds
257
self.IntervalChanged(self._interval_milliseconds)
258
interval = property(lambda self: self._interval, _set_interval)
106
interval = property(lambda self: self._interval,
259
108
del _set_interval
261
@dbus.service.method(interface, out_signature="t")
262
def getInterval(self):
263
"D-Bus getter method"
264
return self._interval_milliseconds
266
@dbus.service.signal(interface, signature="t")
267
def IntervalChanged(self, t):
271
def __init__(self, name = None, stop_hook=None, config=None):
272
"""Note: the 'checker' key in 'config' sets the
273
'checker_command' attribute and *not* the 'checker'
275
dbus.service.Object.__init__(self, bus,
277
% name.replace(".", "_"))
109
def __init__(self, name=None, options=None, stop_hook=None,
110
fingerprint=None, secret=None, secfile=None, fqdn=None,
111
timeout=None, interval=-1, checker=None):
281
logger.debug(u"Creating client %r", self.name)
282
# Uppercase and remove spaces from fingerprint for later
283
# comparison purposes with return value from the fingerprint()
285
self.fingerprint = config["fingerprint"].upper()\
287
logger.debug(u" Fingerprint: %s", self.fingerprint)
288
if "secret" in config:
289
self.secret = config["secret"].decode(u"base64")
290
elif "secfile" in config:
291
with closing(open(os.path.expanduser
293
(config["secfile"])))) \
295
self.secret = secfile.read()
113
# Uppercase and remove spaces from fingerprint
114
# for later comparison purposes with return value of
115
# the fingerprint() function
116
self.fingerprint = fingerprint.upper().replace(u" ", u"")
118
self.secret = secret.decode(u"base64")
121
self.secret = sf.read()
297
raise TypeError(u"No secret or secfile for client %s"
299
self.host = config.get("host", "")
124
raise RuntimeError(u"No secret or secfile for client %s"
126
self.fqdn = fqdn # string
300
127
self.created = datetime.datetime.now()
302
self.last_checked_ok = None
303
self.timeout = string_to_delta(config["timeout"])
304
self.interval = string_to_delta(config["interval"])
128
self.last_seen = None
130
timeout = options.timeout
131
self.timeout = timeout
133
interval = options.interval
135
interval = string_to_delta(interval)
136
self.interval = interval
305
137
self.stop_hook = stop_hook
306
138
self.checker = None
307
139
self.checker_initiator_tag = None
308
140
self.stop_initiator_tag = None
309
141
self.checker_callback_tag = None
310
self.check_command = config["checker"]
142
self.check_command = checker
313
"""Start this client's checker and timeout hooks"""
144
"""Start this clients checker and timeout hooks"""
315
145
# Schedule a new checker to be started an 'interval' from now,
316
146
# and every interval from then on.
317
147
self.checker_initiator_tag = gobject.timeout_add\
323
153
self.stop_initiator_tag = gobject.timeout_add\
324
154
(self._timeout_milliseconds,
327
self.StateChanged(True)
329
@dbus.service.signal(interface, signature="b")
330
def StateChanged(self, started):
335
"""Stop this client."""
336
if getattr(self, "started", False):
337
logger.info(u"Stopping client %s", self.name)
340
if getattr(self, "stop_initiator_tag", False):
158
The possibility that this client might be restarted is left
159
open, but not currently used."""
160
logger.debug(u"Stopping client %s", self.name)
162
if self.stop_initiator_tag:
341
163
gobject.source_remove(self.stop_initiator_tag)
342
164
self.stop_initiator_tag = None
343
if getattr(self, "checker_initiator_tag", False):
165
if self.checker_initiator_tag:
344
166
gobject.source_remove(self.checker_initiator_tag)
345
167
self.checker_initiator_tag = None
346
168
self.stop_checker()
347
169
if self.stop_hook:
348
170
self.stop_hook(self)
350
self.StateChanged(False)
351
171
# Do not run this again if called by a gobject.timeout_add
354
Stop = dbus.service.method(interface)(stop)
356
173
def __del__(self):
357
self.stop_hook = None
174
# Some code duplication here and in stop()
175
if hasattr(self, "stop_initiator_tag") \
176
and self.stop_initiator_tag:
177
gobject.source_remove(self.stop_initiator_tag)
178
self.stop_initiator_tag = None
179
if hasattr(self, "checker_initiator_tag") \
180
and self.checker_initiator_tag:
181
gobject.source_remove(self.checker_initiator_tag)
182
self.checker_initiator_tag = None
360
184
def checker_callback(self, pid, condition):
361
185
"""The checker has completed, so take appropriate actions."""
362
self.checker_callback_tag = None
186
now = datetime.datetime.now()
364
187
if os.WIFEXITED(condition) \
365
188
and (os.WEXITSTATUS(condition) == 0):
366
logger.info(u"Checker for %(name)s succeeded",
369
self.CheckerCompleted(True)
371
elif not os.WIFEXITED(condition):
189
logger.debug(u"Checker for %(name)s succeeded",
192
gobject.source_remove(self.stop_initiator_tag)
193
self.stop_initiator_tag = gobject.timeout_add\
194
(self._timeout_milliseconds,
196
if not os.WIFEXITED(condition):
372
197
logger.warning(u"Checker for %(name)s crashed?",
375
self.CheckerCompleted(False)
377
logger.info(u"Checker for %(name)s failed",
380
self.CheckerCompleted(False)
382
@dbus.service.signal(interface, signature="b")
383
def CheckerCompleted(self, success):
387
def bump_timeout(self):
388
"""Bump up the timeout for this client.
389
This should only be called when the client has been seen,
392
self.last_checked_ok = datetime.datetime.now()
393
gobject.source_remove(self.stop_initiator_tag)
394
self.stop_initiator_tag = gobject.timeout_add\
395
(self._timeout_milliseconds, self.stop)
397
bumpTimeout = dbus.service.method(interface)(bump_timeout)
200
logger.debug(u"Checker for %(name)s failed",
203
self.checker_callback_tag = None
399
204
def start_checker(self):
400
205
"""Start a new checker subprocess if one is not running.
401
206
If a checker already exists, leave it running and do
403
# The reason for not killing a running checker is that if we
404
# did that, then if a checker (for some reason) started
405
# running slowly and taking more than 'interval' time, the
406
# client would inevitably timeout, since no checker would get
407
# a chance to run to completion. If we instead leave running
408
# checkers alone, the checker would have to take more time
409
# than 'timeout' for the client to be declared invalid, which
410
# is as it should be.
411
208
if self.checker is None:
209
logger.debug(u"Starting checker for %s",
413
# In case check_command has exactly one % operator
414
command = self.check_command % self.host
212
command = self.check_command % self.fqdn
415
213
except TypeError:
416
# Escape attributes for the shell
417
214
escaped_attrs = dict((key, re.escape(str(val)))
419
216
vars(self).iteritems())
421
218
command = self.check_command % escaped_attrs
422
219
except TypeError, error:
423
logger.error(u'Could not format string "%s":'
424
u' %s', self.check_command, error)
220
logger.critical(u'Could not format string "%s": %s',
221
self.check_command, error)
425
222
return True # Try again later
427
logger.info(u"Starting checker %r for %s",
429
# We don't need to redirect stdout and stderr, since
430
# in normal mode, that is already done by daemon(),
431
# and in debug mode we don't want to. (Stdin is
432
# always replaced by /dev/null.)
433
self.checker = subprocess.Popen(command,
436
self.checker_callback_tag = gobject.child_watch_add\
438
self.checker_callback)
440
self.CheckerStarted(command)
441
except OSError, error:
224
self.checker = subprocess.\
226
stdout=subprocess.PIPE,
227
close_fds=True, shell=True,
229
self.checker_callback_tag = gobject.\
230
child_watch_add(self.checker.pid,
233
except subprocess.OSError, error:
442
234
logger.error(u"Failed to start subprocess: %s",
444
236
# Re-run this periodically if run by gobject.timeout_add
447
@dbus.service.signal(interface, signature="s")
448
def CheckerStarted(self, command):
451
@dbus.service.method(interface, out_signature="b")
452
def checkerIsRunning(self):
453
"D-Bus getter method"
454
return self.checker is not None
456
238
def stop_checker(self):
457
239
"""Force the checker process, if any, to stop."""
458
if self.checker_callback_tag:
459
gobject.source_remove(self.checker_callback_tag)
460
self.checker_callback_tag = None
461
if getattr(self, "checker", None) is None:
240
if not hasattr(self, "checker") or self.checker is None:
463
logger.debug(u"Stopping checker for %(name)s", vars(self))
465
os.kill(self.checker.pid, signal.SIGTERM)
467
#if self.checker.poll() is None:
468
# os.kill(self.checker.pid, signal.SIGKILL)
469
except OSError, error:
470
if error.errno != errno.ESRCH: # No such process
242
gobject.source_remove(self.checker_callback_tag)
243
self.checker_callback_tag = None
244
os.kill(self.checker.pid, signal.SIGTERM)
245
if self.checker.poll() is None:
246
os.kill(self.checker.pid, signal.SIGKILL)
472
247
self.checker = None
474
StopChecker = dbus.service.method(interface)(stop_checker)
476
def still_valid(self):
248
def still_valid(self, now=None):
477
249
"""Has the timeout not yet passed for this client?"""
480
now = datetime.datetime.now()
481
if self.last_checked_ok is None:
251
now = datetime.datetime.now()
252
if self.last_seen is None:
482
253
return now < (self.created + self.timeout)
484
return now < (self.last_checked_ok + self.timeout)
486
stillValid = dbus.service.method(interface, out_signature="b")\
255
return now < (self.last_seen + self.timeout)
492
258
def peer_certificate(session):
493
"Return the peer's OpenPGP certificate as a bytestring"
494
259
# If not an OpenPGP certificate...
495
260
if gnutls.library.functions.gnutls_certificate_type_get\
496
261
(session._c_object) \
509
274
def fingerprint(openpgp):
510
"Convert an OpenPGP bytestring to a hexdigit fingerprint string"
275
# New empty GnuTLS certificate
276
crt = gnutls.library.types.gnutls_openpgp_crt_t()
277
gnutls.library.functions.gnutls_openpgp_crt_init\
511
279
# New GnuTLS "datum" with the OpenPGP public key
512
280
datum = gnutls.library.types.gnutls_datum_t\
513
281
(ctypes.cast(ctypes.c_char_p(openpgp),
514
282
ctypes.POINTER(ctypes.c_ubyte)),
515
283
ctypes.c_uint(len(openpgp)))
516
# New empty GnuTLS certificate
517
crt = gnutls.library.types.gnutls_openpgp_crt_t()
518
gnutls.library.functions.gnutls_openpgp_crt_init\
520
284
# Import the OpenPGP public key into the certificate
521
gnutls.library.functions.gnutls_openpgp_crt_import\
522
(crt, ctypes.byref(datum),
523
gnutls.library.constants.GNUTLS_OPENPGP_FMT_RAW)
524
# Verify the self signature in the key
525
crtverify = ctypes.c_uint()
526
gnutls.library.functions.gnutls_openpgp_crt_verify_self\
527
(crt, 0, ctypes.byref(crtverify))
528
if crtverify.value != 0:
529
gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
530
raise gnutls.errors.CertificateSecurityError("Verify failed")
285
ret = gnutls.library.functions.gnutls_openpgp_crt_import\
288
gnutls.library.constants.GNUTLS_OPENPGP_FMT_RAW)
531
289
# New buffer for the fingerprint
532
buf = ctypes.create_string_buffer(20)
533
buf_len = ctypes.c_size_t()
290
buffer = ctypes.create_string_buffer(20)
291
buffer_length = ctypes.c_size_t()
534
292
# Get the fingerprint from the certificate into the buffer
535
293
gnutls.library.functions.gnutls_openpgp_crt_get_fingerprint\
536
(crt, ctypes.byref(buf), ctypes.byref(buf_len))
294
(crt, ctypes.byref(buffer), ctypes.byref(buffer_length))
537
295
# Deinit the certificate
538
296
gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
539
297
# Convert the buffer to a Python bytestring
540
fpr = ctypes.string_at(buf, buf_len.value)
298
fpr = ctypes.string_at(buffer, buffer_length.value)
541
299
# Convert the bytestring to hexadecimal notation
542
300
hex_fpr = u''.join(u"%02X" % ord(char) for char in fpr)
546
class TCP_handler(SocketServer.BaseRequestHandler, object):
304
class tcp_handler(SocketServer.BaseRequestHandler, object):
547
305
"""A TCP request handler class.
548
306
Instantiated by IPv6_TCPServer for each request to handle it.
549
307
Note: This will run in its own forked process."""
551
309
def handle(self):
552
logger.info(u"TCP connection from: %s",
553
unicode(self.client_address))
554
session = gnutls.connection.ClientSession\
555
(self.request, gnutls.connection.X509Credentials())
557
line = self.request.makefile().readline()
558
logger.debug(u"Protocol version: %r", line)
560
if int(line.strip().split()[0]) > 1:
562
except (ValueError, IndexError, RuntimeError), error:
563
logger.error(u"Unknown protocol version: %s", error)
566
# Note: gnutls.connection.X509Credentials is really a generic
567
# GnuTLS certificate credentials object so long as no X.509
568
# keys are added to it. Therefore, we can use it here despite
569
# using OpenPGP certificates.
310
logger.debug(u"TCP connection from: %s",
311
unicode(self.client_address))
312
session = gnutls.connection.ClientSession(self.request,
571
316
#priority = ':'.join(("NONE", "+VERS-TLS1.1", "+AES-256-CBC",
572
317
# "+SHA1", "+COMP-NULL", "+CTYPE-OPENPGP",
574
# Use a fallback default, since this MUST be set.
575
priority = self.server.settings.get("priority", "NORMAL")
319
priority = "SECURE256"
576
321
gnutls.library.functions.gnutls_priority_set_direct\
577
(session._c_object, priority, None)
322
(session._c_object, priority, None);
580
325
session.handshake()
581
326
except gnutls.errors.GNUTLSError, error:
582
logger.warning(u"Handshake failed: %s", error)
327
logger.debug(u"Handshake failed: %s", error)
583
328
# Do not run session.bye() here: the session is not
584
329
# established. Just abandon the request.
587
332
fpr = fingerprint(peer_certificate(session))
588
333
except (TypeError, gnutls.errors.GNUTLSError), error:
589
logger.warning(u"Bad certificate: %s", error)
334
logger.debug(u"Bad certificate: %s", error)
592
337
logger.debug(u"Fingerprint: %s", fpr)
594
for c in self.server.clients:
595
340
if c.fingerprint == fpr:
599
logger.warning(u"Client not found for fingerprint: %s",
603
343
# Have to check if client.still_valid(), since it is possible
604
344
# that the client timed out while establishing the GnuTLS
606
if not client.still_valid():
607
logger.warning(u"Client %(name)s is invalid",
346
if (not client) or (not client.still_valid()):
348
logger.debug(u"Client %(name)s is invalid",
351
logger.debug(u"Client not found for fingerprint: %s",
611
## This won't work here, since we're in a fork.
612
# client.bump_timeout()
614
356
while sent_size < len(client.secret):
615
357
sent = session.send(client.secret[sent_size:])
693
421
datetime.timedelta(1)
694
422
>>> string_to_delta(u'1w')
695
423
datetime.timedelta(7)
696
>>> string_to_delta('5m 30s')
697
datetime.timedelta(0, 330)
699
timevalue = datetime.timedelta(0)
700
for s in interval.split():
702
suffix = unicode(s[-1])
705
delta = datetime.timedelta(value)
707
delta = datetime.timedelta(0, value)
709
delta = datetime.timedelta(0, 0, 0, 0, value)
711
delta = datetime.timedelta(0, 0, 0, 0, 0, value)
713
delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
716
except (ValueError, IndexError):
426
suffix=unicode(interval[-1])
427
value=int(interval[:-1])
429
delta = datetime.timedelta(value)
431
delta = datetime.timedelta(0, value)
433
delta = datetime.timedelta(0, 0, 0, 0, value)
435
delta = datetime.timedelta(0, 0, 0, 0, 0, value)
437
delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
440
except (ValueError, IndexError):
446
"""From the Avahi server example code"""
447
global group, serviceName, serviceType, servicePort, serviceTXT, \
450
group = dbus.Interface(
451
bus.get_object( avahi.DBUS_NAME,
452
server.EntryGroupNew()),
453
avahi.DBUS_INTERFACE_ENTRY_GROUP)
454
group.connect_to_signal('StateChanged',
455
entry_group_state_changed)
456
logger.debug(u"Adding service '%s' of type '%s' ...",
457
serviceName, serviceType)
460
serviceInterface, # interface
461
avahi.PROTO_INET6, # protocol
462
dbus.UInt32(0), # flags
463
serviceName, serviceType,
465
dbus.UInt16(servicePort),
466
avahi.string_array_to_txt_array(serviceTXT))
470
def remove_service():
471
"""From the Avahi server example code"""
474
if not group is None:
722
478
def server_state_changed(state):
723
"""Derived from the Avahi example code"""
479
"""From the Avahi server example code"""
724
480
if state == avahi.SERVER_COLLISION:
725
logger.error(u"Zeroconf server name collision")
481
logger.warning(u"Server name collision")
727
483
elif state == avahi.SERVER_RUNNING:
731
487
def entry_group_state_changed(state, error):
732
"""Derived from the Avahi example code"""
733
logger.debug(u"Avahi state change: %i", state)
488
"""From the Avahi server example code"""
489
global serviceName, server, rename_count
491
logger.debug(u"state change: %i", state)
735
493
if state == avahi.ENTRY_GROUP_ESTABLISHED:
736
logger.debug(u"Zeroconf service established.")
494
logger.debug(u"Service established.")
737
495
elif state == avahi.ENTRY_GROUP_COLLISION:
738
logger.warning(u"Zeroconf service name collision.")
497
rename_count = rename_count - 1
499
name = server.GetAlternativeServiceName(name)
500
logger.warning(u"Service name collision, "
501
u"changing name to '%s' ...", name)
506
logger.error(u"No suitable service name found "
507
u"after %i retries, exiting.",
740
510
elif state == avahi.ENTRY_GROUP_FAILURE:
741
logger.critical(u"Avahi: Error in group state changed %s",
743
raise AvahiGroupError("State changed: %s", str(error))
511
logger.error(u"Error in group state changed %s",
745
517
def if_nametoindex(interface):
746
"""Call the C function if_nametoindex(), or equivalent"""
747
global if_nametoindex
518
"""Call the C function if_nametoindex()"""
749
if_nametoindex = ctypes.cdll.LoadLibrary\
750
(ctypes.util.find_library("c")).if_nametoindex
520
libc = ctypes.cdll.LoadLibrary("libc.so.6")
521
return libc.if_nametoindex(interface)
751
522
except (OSError, AttributeError):
752
523
if "struct" not in sys.modules:
754
525
if "fcntl" not in sys.modules:
756
def if_nametoindex(interface):
757
"Get an interface index the hard way, i.e. using fcntl()"
758
SIOCGIFINDEX = 0x8933 # From /usr/include/linux/sockios.h
759
with closing(socket.socket()) as s:
760
ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
761
struct.pack("16s16x", interface))
762
interface_index = struct.unpack("I", ifreq[16:20])[0]
763
return interface_index
764
return if_nametoindex(interface)
767
def daemon(nochdir = False, noclose = False):
768
"""See daemon(3). Standard BSD Unix function.
769
This should really exist as os.daemon, but it doesn't (yet)."""
778
# Close all standard open file descriptors
779
null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
780
if not stat.S_ISCHR(os.fstat(null).st_mode):
781
raise OSError(errno.ENODEV,
782
"/dev/null not a character device")
783
os.dup2(null, sys.stdin.fileno())
784
os.dup2(null, sys.stdout.fileno())
785
os.dup2(null, sys.stderr.fileno())
791
parser = OptionParser(version = "%%prog %s" % version)
527
SIOCGIFINDEX = 0x8933 # From /usr/include/linux/sockios.h
529
ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
530
struct.pack("16s16x", interface))
532
interface_index = struct.unpack("I", ifreq[16:20])[0]
533
return interface_index
536
if __name__ == '__main__':
537
parser = OptionParser()
792
538
parser.add_option("-i", "--interface", type="string",
793
metavar="IF", help="Bind to interface IF")
794
parser.add_option("-a", "--address", type="string",
795
help="Address to listen for requests on")
796
parser.add_option("-p", "--port", type="int",
539
default=None, metavar="IF",
540
help="Bind to interface IF")
541
parser.add_option("--cert", type="string", default="cert.pem",
543
help="Public key certificate PEM file to use")
544
parser.add_option("--key", type="string", default="key.pem",
546
help="Private key PEM file to use")
547
parser.add_option("--ca", type="string", default="ca.pem",
549
help="Certificate Authority certificate PEM file to use")
550
parser.add_option("--crl", type="string", default="crl.pem",
552
help="Certificate Revokation List PEM file to use")
553
parser.add_option("-p", "--port", type="int", default=None,
797
554
help="Port number to receive requests on")
555
parser.add_option("--timeout", type="string", # Parsed later
557
help="Amount of downtime allowed for clients")
558
parser.add_option("--interval", type="string", # Parsed later
560
help="How often to check that a client is up")
798
561
parser.add_option("--check", action="store_true", default=False,
799
562
help="Run self-test")
800
parser.add_option("--debug", action="store_true",
801
help="Debug mode; run in foreground and log to"
803
parser.add_option("--priority", type="string", help="GnuTLS"
804
" priority string (see GnuTLS documentation)")
805
parser.add_option("--servicename", type="string", metavar="NAME",
806
help="Zeroconf service name")
807
parser.add_option("--configdir", type="string",
808
default="/etc/mandos", metavar="DIR",
809
help="Directory to search for configuration"
811
options = parser.parse_args()[0]
563
parser.add_option("--debug", action="store_true", default=False,
565
(options, args) = parser.parse_args()
813
567
if options.check:
815
569
doctest.testmod()
818
# Default values for config file for server-global settings
819
server_defaults = { "interface": "",
824
"SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
825
"servicename": "Mandos",
828
# Parse config file for server-global settings
829
server_config = ConfigParser.SafeConfigParser(server_defaults)
831
server_config.read(os.path.join(options.configdir, "mandos.conf"))
832
# Convert the SafeConfigParser object to a dict
833
server_settings = server_config.defaults()
834
# Use getboolean on the boolean config option
835
server_settings["debug"] = server_config.getboolean\
839
# Override the settings from the config file with command line
841
for option in ("interface", "address", "port", "debug",
842
"priority", "servicename", "configdir"):
843
value = getattr(options, option)
844
if value is not None:
845
server_settings[option] = value
847
# Now we have our good server settings in "server_settings"
849
debug = server_settings["debug"]
852
syslogger.setLevel(logging.WARNING)
853
console.setLevel(logging.WARNING)
855
if server_settings["servicename"] != "Mandos":
856
syslogger.setFormatter(logging.Formatter\
857
('Mandos (%s): %%(levelname)s:'
859
% server_settings["servicename"]))
861
# Parse config file with clients
862
client_defaults = { "timeout": "1h",
864
"checker": "fping -q -- %(host)s",
867
client_config = ConfigParser.SafeConfigParser(client_defaults)
868
client_config.read(os.path.join(server_settings["configdir"],
872
tcp_server = IPv6_TCPServer((server_settings["address"],
873
server_settings["port"]),
875
settings=server_settings,
877
pidfilename = "/var/run/mandos.pid"
879
pidfile = open(pidfilename, "w")
880
except IOError, error:
881
logger.error("Could not open file %r", pidfilename)
886
uid = pwd.getpwnam("mandos").pw_uid
889
uid = pwd.getpwnam("nobody").pw_uid
893
gid = pwd.getpwnam("mandos").pw_gid
896
gid = pwd.getpwnam("nogroup").pw_gid
902
except OSError, error:
903
if error[0] != errno.EPERM:
907
service = AvahiService(name = server_settings["servicename"],
908
servicetype = "_mandos._tcp", )
909
if server_settings["interface"]:
910
service.interface = if_nametoindex\
911
(server_settings["interface"])
916
# From the Avahi example code
572
# Parse the time arguments
574
options.timeout = string_to_delta(options.timeout)
576
parser.error("option --timeout: Unparseable time")
578
options.interval = string_to_delta(options.interval)
580
parser.error("option --interval: Unparseable time")
583
defaults = { "checker": "sleep 1; fping -q -- %%(fqdn)s" }
584
client_config = ConfigParser.SafeConfigParser(defaults)
585
#client_config.readfp(open("secrets.conf"), "secrets.conf")
586
client_config.read("mandos-clients.conf")
588
# From the Avahi server example code
917
589
DBusGMainLoop(set_as_default=True )
918
590
main_loop = gobject.MainLoop()
919
591
bus = dbus.SystemBus()
920
server = dbus.Interface(bus.get_object(avahi.DBUS_NAME,
921
avahi.DBUS_PATH_SERVER),
922
avahi.DBUS_INTERFACE_SERVER)
592
server = dbus.Interface(
593
bus.get_object( avahi.DBUS_NAME, avahi.DBUS_PATH_SERVER ),
594
avahi.DBUS_INTERFACE_SERVER )
923
595
# End of Avahi example code
597
debug = options.debug
925
600
def remove_from_clients(client):
926
601
clients.remove(client)
928
logger.critical(u"No clients left, exiting")
603
logger.debug(u"No clients left, exiting")
931
clients.update(Set(Client(name = section,
606
clients.update(Set(Client(name=section, options=options,
932
607
stop_hook = remove_from_clients,
934
= dict(client_config.items(section)))
608
**(dict(client_config\
935
610
for section in client_config.sections()))
937
logger.critical(u"No clients defined")
941
# Redirect stdin so all checkers get /dev/null
942
null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
943
os.dup2(null, sys.stdin.fileno())
948
logger.removeHandler(console)
949
# Close all input and output, do double fork, etc.
954
pidfile.write(str(pid) + "\n")
958
logger.error(u"Could not write to file %r with PID %d",
961
# "pidfile" was never created
966
"Cleanup function; run on exit"
968
# From the Avahi example code
969
if not group is None:
972
# End of Avahi example code
975
client = clients.pop()
976
client.stop_hook = None
979
atexit.register(cleanup)
982
signal.signal(signal.SIGINT, signal.SIG_IGN)
983
signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
984
signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
986
611
for client in clients:
990
tcp_server.server_activate()
992
# Find out what port we got
993
service.port = tcp_server.socket.getsockname()[1]
994
logger.info(u"Now listening on address %r, port %d, flowinfo %d,"
995
u" scope_id %d" % tcp_server.socket.getsockname())
997
#service.interface = tcp_server.socket.getsockname()[3]
614
tcp_server = IPv6_TCPServer((None, options.port),
618
# Find out what random port we got
619
servicePort = tcp_server.socket.getsockname()[1]
620
logger.debug(u"Now listening on port %d", servicePort)
622
if options.interface is not None:
623
serviceInterface = if_nametoindex(options.interface)
625
# From the Avahi server example code
626
server.connect_to_signal("StateChanged", server_state_changed)
627
server_state_changed(server.GetState())
628
# End of Avahi example code
630
gobject.io_add_watch(tcp_server.fileno(), gobject.IO_IN,
631
lambda *args, **kwargs:
632
tcp_server.handle_request(*args[2:],
1000
# From the Avahi example code
1001
server.connect_to_signal("StateChanged", server_state_changed)
1003
server_state_changed(server.GetState())
1004
except dbus.exceptions.DBusException, error:
1005
logger.critical(u"DBusException: %s", error)
1007
# End of Avahi example code
1009
gobject.io_add_watch(tcp_server.fileno(), gobject.IO_IN,
1010
lambda *args, **kwargs:
1011
tcp_server.handle_request\
1012
(*args[2:], **kwargs) or True)
1014
logger.debug(u"Starting main loop")
1016
except AvahiError, error:
1017
logger.critical(u"AvahiError: %s" + unicode(error))
1019
636
except KeyboardInterrupt:
1023
if __name__ == '__main__':
641
# From the Avahi server example code
642
if not group is None:
644
# End of Avahi example code
646
for client in clients:
647
client.stop_hook = None