57
56
import logging.handlers
62
62
from dbus.mainloop.glib import DBusGMainLoop
65
# Brief description of the operation of this program:
67
# This server announces itself as a Zeroconf service. Connecting
68
# clients use the TLS protocol, with the unusual quirk that this
69
# server program acts as a TLS "client" while the connecting clients
70
# acts as a TLS "server". The clients (acting as a TLS "server") must
71
# supply an OpenPGP certificate, and the fingerprint of this
72
# certificate is used by this server to look up (in a list read from a
73
# file at start time) which binary blob to give the client. No other
74
# authentication or authorization is done by this server.
77
68
logger = logging.Logger('mandos')
78
69
syslogger = logging.handlers.SysLogHandler\
79
(facility = logging.handlers.SysLogHandler.LOG_DAEMON)
70
(facility = logging.handlers.SysLogHandler.LOG_DAEMON,
80
72
syslogger.setFormatter(logging.Formatter\
81
('%(levelname)s: %(message)s'))
73
('Mandos: %(levelname)s: %(message)s'))
82
74
logger.addHandler(syslogger)
85
# This variable is used to optionally bind to a specified interface.
86
# It is a global variable to fit in with the other variables from the
88
serviceInterface = avahi.IF_UNSPEC
76
console = logging.StreamHandler()
77
console.setFormatter(logging.Formatter('%(name)s: %(levelname)s:'
79
logger.addHandler(console)
81
class AvahiError(Exception):
82
def __init__(self, value):
84
super(AvahiError, self).__init__()
86
return repr(self.value)
88
class AvahiServiceError(AvahiError):
91
class AvahiGroupError(AvahiError):
95
class AvahiService(object):
96
"""An Avahi (Zeroconf) service.
98
interface: integer; avahi.IF_UNSPEC or an interface index.
99
Used to optionally bind to the specified interface.
100
name: string; Example: 'Mandos'
101
type: string; Example: '_mandos._tcp'.
102
See <http://www.dns-sd.org/ServiceTypes.html>
103
port: integer; what port to announce
104
TXT: list of strings; TXT record for the service
105
domain: string; Domain to publish on, default to .local if empty.
106
host: string; Host to publish records for, default is localhost
107
max_renames: integer; maximum number of renames
108
rename_count: integer; counter so we only rename after collisions
109
a sensible number of times
111
def __init__(self, interface = avahi.IF_UNSPEC, name = None,
112
servicetype = None, port = None, TXT = None, domain = "",
113
host = "", max_renames = 32768):
114
self.interface = interface
116
self.type = servicetype
124
self.rename_count = 0
125
self.max_renames = max_renames
127
"""Derived from the Avahi example code"""
128
if self.rename_count >= self.max_renames:
129
logger.critical(u"No suitable Zeroconf service name found"
130
u" after %i retries, exiting.",
132
raise AvahiServiceError("Too many renames")
133
self.name = server.GetAlternativeServiceName(self.name)
134
logger.info(u"Changing Zeroconf service name to %r ...",
136
syslogger.setFormatter(logging.Formatter\
137
('Mandos (%s): %%(levelname)s:'
138
' %%(message)s' % self.name))
141
self.rename_count += 1
143
"""Derived from the Avahi example code"""
144
if group is not None:
147
"""Derived from the Avahi example code"""
150
group = dbus.Interface\
151
(bus.get_object(avahi.DBUS_NAME,
152
server.EntryGroupNew()),
153
avahi.DBUS_INTERFACE_ENTRY_GROUP)
154
group.connect_to_signal('StateChanged',
155
entry_group_state_changed)
156
logger.debug(u"Adding Zeroconf service '%s' of type '%s' ...",
157
service.name, service.type)
159
self.interface, # interface
160
avahi.PROTO_INET6, # protocol
161
dbus.UInt32(0), # flags
162
self.name, self.type,
163
self.domain, self.host,
164
dbus.UInt16(self.port),
165
avahi.string_array_to_txt_array(self.TXT))
89
168
# From the Avahi example code:
90
serviceName = "Mandos"
91
serviceType = "_mandos._tcp" # http://www.dns-sd.org/ServiceTypes.html
92
servicePort = None # Not known at startup
93
serviceTXT = [] # TXT record for the service
94
domain = "" # Domain to publish on, default to .local
95
host = "" # Host to publish records for, default to localhost
96
group = None #our entry group
97
rename_count = 12 # Counter so we only rename after collisions a
98
# sensible number of times
169
group = None # our entry group
99
170
# End of Avahi example code
106
177
fingerprint: string (40 or 32 hexadecimal digits); used to
107
178
uniquely identify the client
108
179
secret: bytestring; sent verbatim (over TLS) to client
109
fqdn: string (FQDN); available for use by the checker command
110
created: datetime.datetime()
111
last_seen: datetime.datetime() or None if not yet seen
112
timeout: datetime.timedelta(); How long from last_seen until
113
this client is invalid
180
host: string; available for use by the checker command
181
created: datetime.datetime(); object creation, not client host
182
last_checked_ok: datetime.datetime() or None if not yet checked OK
183
timeout: datetime.timedelta(); How long from last_checked_ok
184
until this client is invalid
114
185
interval: datetime.timedelta(); How often to start a new checker
115
186
stop_hook: If set, called by stop() as stop_hook(self)
116
187
checker: subprocess.Popen(); a running checker process used
117
188
to see if the client lives.
118
Is None if no process is running.
189
'None' if no process is running.
119
190
checker_initiator_tag: a gobject event source tag, or None
120
191
stop_initiator_tag: - '' -
121
192
checker_callback_tag: - '' -
122
193
checker_command: string; External command which is run to check if
123
client lives. %()s expansions are done at
194
client lives. %() expansions are done at
124
195
runtime with vars(self) as dict, so that for
125
196
instance %(name)s can be used in the command.
126
197
Private attibutes:
127
198
_timeout: Real variable for 'timeout'
128
199
_interval: Real variable for 'interval'
129
_timeout_milliseconds: Used by gobject.timeout_add()
200
_timeout_milliseconds: Used when calling gobject.timeout_add()
130
201
_interval_milliseconds: - '' -
132
203
def _set_timeout(self, timeout):
152
223
interval = property(lambda self: self._interval,
154
225
del _set_interval
155
def __init__(self, name=None, options=None, stop_hook=None,
156
fingerprint=None, secret=None, secfile=None,
157
fqdn=None, timeout=None, interval=-1, checker=None):
158
"""Note: the 'checker' argument sets the 'checker_command'
159
attribute and not the 'checker' attribute.."""
226
def __init__(self, name = None, stop_hook=None, config=None):
227
"""Note: the 'checker' key in 'config' sets the
228
'checker_command' attribute and *not* the 'checker'
161
# Uppercase and remove spaces from fingerprint
162
# for later comparison purposes with return value of
163
# the fingerprint() function
164
self.fingerprint = fingerprint.upper().replace(u" ", u"")
166
self.secret = secret.decode(u"base64")
169
self.secret = sf.read()
233
logger.debug(u"Creating client %r", self.name)
234
# Uppercase and remove spaces from fingerprint for later
235
# comparison purposes with return value from the fingerprint()
237
self.fingerprint = config["fingerprint"].upper()\
239
logger.debug(u" Fingerprint: %s", self.fingerprint)
240
if "secret" in config:
241
self.secret = config["secret"].decode(u"base64")
242
elif "secfile" in config:
243
secfile = open(config["secfile"])
244
self.secret = secfile.read()
172
raise RuntimeError(u"No secret or secfile for client %s"
174
self.fqdn = fqdn # string
247
raise TypeError(u"No secret or secfile for client %s"
249
self.host = config.get("host", "")
175
250
self.created = datetime.datetime.now()
176
self.last_seen = None
178
self.timeout = options.timeout
180
self.timeout = string_to_delta(timeout)
182
self.interval = options.interval
184
self.interval = string_to_delta(interval)
251
self.last_checked_ok = None
252
self.timeout = string_to_delta(config["timeout"])
253
self.interval = string_to_delta(config["interval"])
185
254
self.stop_hook = stop_hook
186
255
self.checker = None
187
256
self.checker_initiator_tag = None
188
257
self.stop_initiator_tag = None
189
258
self.checker_callback_tag = None
190
self.check_command = checker
259
self.check_command = config["checker"]
192
261
"""Start this client's checker and timeout hooks"""
193
262
# Schedule a new checker to be started an 'interval' from now,
205
274
"""Stop this client.
206
The possibility that this client might be restarted is left
207
open, but not currently used."""
275
The possibility that a client might be restarted is left open,
276
but not currently used."""
208
277
# If this client doesn't have a secret, it is already stopped.
210
logger.debug(u"Stopping client %s", self.name)
278
if hasattr(self, "secret") and self.secret:
279
logger.info(u"Stopping client %s", self.name)
211
280
self.secret = None
214
if hasattr(self, "stop_initiator_tag") \
215
and self.stop_initiator_tag:
283
if getattr(self, "stop_initiator_tag", False):
216
284
gobject.source_remove(self.stop_initiator_tag)
217
285
self.stop_initiator_tag = None
218
if hasattr(self, "checker_initiator_tag") \
219
and self.checker_initiator_tag:
286
if getattr(self, "checker_initiator_tag", False):
220
287
gobject.source_remove(self.checker_initiator_tag)
221
288
self.checker_initiator_tag = None
222
289
self.stop_checker()
261
328
# is as it should be.
262
329
if self.checker is None:
264
command = self.check_command % self.fqdn
331
# In case check_command has exactly one % operator
332
command = self.check_command % self.host
265
333
except TypeError:
334
# Escape attributes for the shell
266
335
escaped_attrs = dict((key, re.escape(str(val)))
268
337
vars(self).iteritems())
270
339
command = self.check_command % escaped_attrs
271
340
except TypeError, error:
272
logger.critical(u'Could not format string "%s":'
273
u' %s', self.check_command, error)
341
logger.error(u'Could not format string "%s":'
342
u' %s', self.check_command, error)
274
343
return True # Try again later
276
logger.debug(u"Starting checker %r for %s",
278
self.checker = subprocess.\
280
close_fds=True, shell=True,
345
logger.info(u"Starting checker %r for %s",
347
# We don't need to redirect stdout and stderr, since
348
# in normal mode, that is already done by daemon(),
349
# and in debug mode we don't want to. (Stdin is
350
# always replaced by /dev/null.)
351
self.checker = subprocess.Popen(command,
282
354
self.checker_callback_tag = gobject.child_watch_add\
283
355
(self.checker.pid,
284
356
self.checker_callback)
285
except subprocess.OSError, error:
357
except OSError, error:
286
358
logger.error(u"Failed to start subprocess: %s",
288
360
# Re-run this periodically if run by gobject.timeout_add
292
364
if self.checker_callback_tag:
293
365
gobject.source_remove(self.checker_callback_tag)
294
366
self.checker_callback_tag = None
295
if not hasattr(self, "checker") or self.checker is None:
367
if getattr(self, "checker", None) is None:
297
logger.debug("Stopping checker for %(name)s", vars(self))
369
logger.debug(u"Stopping checker for %(name)s", vars(self))
299
371
os.kill(self.checker.pid, signal.SIGTERM)
301
373
#if self.checker.poll() is None:
302
374
# os.kill(self.checker.pid, signal.SIGKILL)
303
375
except OSError, error:
304
if error.errno != errno.ESRCH:
376
if error.errno != errno.ESRCH: # No such process
306
378
self.checker = None
307
def still_valid(self, now=None):
379
def still_valid(self):
308
380
"""Has the timeout not yet passed for this client?"""
310
now = datetime.datetime.now()
311
if self.last_seen is None:
381
now = datetime.datetime.now()
382
if self.last_checked_ok is None:
312
383
return now < (self.created + self.timeout)
314
return now < (self.last_seen + self.timeout)
385
return now < (self.last_checked_ok + self.timeout)
317
388
def peer_certificate(session):
334
405
def fingerprint(openpgp):
335
406
"Convert an OpenPGP bytestring to a hexdigit fingerprint string"
336
# New empty GnuTLS certificate
337
crt = gnutls.library.types.gnutls_openpgp_crt_t()
338
gnutls.library.functions.gnutls_openpgp_crt_init\
340
407
# New GnuTLS "datum" with the OpenPGP public key
341
408
datum = gnutls.library.types.gnutls_datum_t\
342
409
(ctypes.cast(ctypes.c_char_p(openpgp),
343
410
ctypes.POINTER(ctypes.c_ubyte)),
344
411
ctypes.c_uint(len(openpgp)))
412
# New empty GnuTLS certificate
413
crt = gnutls.library.types.gnutls_openpgp_crt_t()
414
gnutls.library.functions.gnutls_openpgp_crt_init\
345
416
# Import the OpenPGP public key into the certificate
346
ret = gnutls.library.functions.gnutls_openpgp_crt_import\
349
gnutls.library.constants.GNUTLS_OPENPGP_FMT_RAW)
417
gnutls.library.functions.gnutls_openpgp_crt_import\
418
(crt, ctypes.byref(datum),
419
gnutls.library.constants.GNUTLS_OPENPGP_FMT_RAW)
420
# Verify the self signature in the key
421
crtverify = ctypes.c_uint()
422
gnutls.library.functions.gnutls_openpgp_crt_verify_self\
423
(crt, 0, ctypes.byref(crtverify))
424
if crtverify.value != 0:
425
gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
426
raise gnutls.errors.CertificateSecurityError("Verify failed")
350
427
# New buffer for the fingerprint
351
buffer = ctypes.create_string_buffer(20)
352
buffer_length = ctypes.c_size_t()
428
buf = ctypes.create_string_buffer(20)
429
buf_len = ctypes.c_size_t()
353
430
# Get the fingerprint from the certificate into the buffer
354
431
gnutls.library.functions.gnutls_openpgp_crt_get_fingerprint\
355
(crt, ctypes.byref(buffer), ctypes.byref(buffer_length))
432
(crt, ctypes.byref(buf), ctypes.byref(buf_len))
356
433
# Deinit the certificate
357
434
gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
358
435
# Convert the buffer to a Python bytestring
359
fpr = ctypes.string_at(buffer, buffer_length.value)
436
fpr = ctypes.string_at(buf, buf_len.value)
360
437
# Convert the bytestring to hexadecimal notation
361
438
hex_fpr = u''.join(u"%02X" % ord(char) for char in fpr)
365
class tcp_handler(SocketServer.BaseRequestHandler, object):
442
class TCP_handler(SocketServer.BaseRequestHandler, object):
366
443
"""A TCP request handler class.
367
444
Instantiated by IPv6_TCPServer for each request to handle it.
368
445
Note: This will run in its own forked process."""
370
447
def handle(self):
371
logger.debug(u"TCP connection from: %s",
448
logger.info(u"TCP connection from: %s",
372
449
unicode(self.client_address))
373
session = gnutls.connection.ClientSession(self.request,
450
session = gnutls.connection.ClientSession\
451
(self.request, gnutls.connection.X509Credentials())
453
line = self.request.makefile().readline()
454
logger.debug(u"Protocol version: %r", line)
456
if int(line.strip().split()[0]) > 1:
458
except (ValueError, IndexError, RuntimeError), error:
459
logger.error(u"Unknown protocol version: %s", error)
462
# Note: gnutls.connection.X509Credentials is really a generic
463
# GnuTLS certificate credentials object so long as no X.509
464
# keys are added to it. Therefore, we can use it here despite
465
# using OpenPGP certificates.
377
467
#priority = ':'.join(("NONE", "+VERS-TLS1.1", "+AES-256-CBC",
378
468
# "+SHA1", "+COMP-NULL", "+CTYPE-OPENPGP",
380
priority = "SECURE256"
470
priority = "NORMAL" # Fallback default, since this
472
if self.server.settings["priority"]:
473
priority = self.server.settings["priority"]
382
474
gnutls.library.functions.gnutls_priority_set_direct\
383
(session._c_object, priority, None);
475
(session._c_object, priority, None)
386
478
session.handshake()
387
479
except gnutls.errors.GNUTLSError, error:
388
logger.debug(u"Handshake failed: %s", error)
480
logger.warning(u"Handshake failed: %s", error)
389
481
# Do not run session.bye() here: the session is not
390
482
# established. Just abandon the request.
393
485
fpr = fingerprint(peer_certificate(session))
394
486
except (TypeError, gnutls.errors.GNUTLSError), error:
395
logger.debug(u"Bad certificate: %s", error)
487
logger.warning(u"Bad certificate: %s", error)
398
490
logger.debug(u"Fingerprint: %s", fpr)
426
519
class IPv6_TCPServer(SocketServer.ForkingTCPServer, object):
427
520
"""IPv6 TCP server. Accepts 'None' as address and/or port.
429
options: Command line options
522
settings: Server settings
430
523
clients: Set() of Client objects
524
enabled: Boolean; whether this server is activated yet
432
526
address_family = socket.AF_INET6
433
527
def __init__(self, *args, **kwargs):
434
if "options" in kwargs:
435
self.options = kwargs["options"]
436
del kwargs["options"]
528
if "settings" in kwargs:
529
self.settings = kwargs["settings"]
530
del kwargs["settings"]
437
531
if "clients" in kwargs:
438
532
self.clients = kwargs["clients"]
439
533
del kwargs["clients"]
440
return super(type(self), self).__init__(*args, **kwargs)
535
super(IPv6_TCPServer, self).__init__(*args, **kwargs)
441
536
def server_bind(self):
442
537
"""This overrides the normal server_bind() function
443
538
to bind to an interface if one was specified, and also NOT to
444
539
bind to an address or port if they were not specified."""
445
if self.options.interface:
446
if not hasattr(socket, "SO_BINDTODEVICE"):
447
# From /usr/include/asm-i486/socket.h
448
socket.SO_BINDTODEVICE = 25
540
if self.settings["interface"]:
541
# 25 is from /usr/include/asm-i486/socket.h
542
SO_BINDTODEVICE = getattr(socket, "SO_BINDTODEVICE", 25)
450
544
self.socket.setsockopt(socket.SOL_SOCKET,
451
socket.SO_BINDTODEVICE,
452
self.options.interface)
546
self.settings["interface"])
453
547
except socket.error, error:
454
548
if error[0] == errno.EPERM:
455
logger.warning(u"No permission to"
456
u" bind to interface %s",
457
self.options.interface)
549
logger.error(u"No permission to"
550
u" bind to interface %s",
551
self.settings["interface"])
460
554
# Only bind(2) the socket if we really need to.
482
588
datetime.timedelta(1)
483
589
>>> string_to_delta(u'1w')
484
590
datetime.timedelta(7)
591
>>> string_to_delta('5m 30s')
592
datetime.timedelta(0, 330)
487
suffix=unicode(interval[-1])
488
value=int(interval[:-1])
490
delta = datetime.timedelta(value)
492
delta = datetime.timedelta(0, value)
494
delta = datetime.timedelta(0, 0, 0, 0, value)
496
delta = datetime.timedelta(0, 0, 0, 0, 0, value)
498
delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
594
timevalue = datetime.timedelta(0)
595
for s in interval.split():
597
suffix = unicode(s[-1])
600
delta = datetime.timedelta(value)
602
delta = datetime.timedelta(0, value)
604
delta = datetime.timedelta(0, 0, 0, 0, value)
606
delta = datetime.timedelta(0, 0, 0, 0, 0, value)
608
delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
611
except (ValueError, IndexError):
501
except (ValueError, IndexError):
507
"""Derived from the Avahi example code"""
508
global group, serviceName, serviceType, servicePort, serviceTXT, \
511
group = dbus.Interface(
512
bus.get_object( avahi.DBUS_NAME,
513
server.EntryGroupNew()),
514
avahi.DBUS_INTERFACE_ENTRY_GROUP)
515
group.connect_to_signal('StateChanged',
516
entry_group_state_changed)
517
logger.debug(u"Adding service '%s' of type '%s' ...",
518
serviceName, serviceType)
521
serviceInterface, # interface
522
avahi.PROTO_INET6, # protocol
523
dbus.UInt32(0), # flags
524
serviceName, serviceType,
526
dbus.UInt16(servicePort),
527
avahi.string_array_to_txt_array(serviceTXT))
531
def remove_service():
532
"""From the Avahi example code"""
535
if not group is None:
539
617
def server_state_changed(state):
540
618
"""Derived from the Avahi example code"""
541
619
if state == avahi.SERVER_COLLISION:
542
logger.warning(u"Server name collision")
620
logger.error(u"Zeroconf server name collision")
544
622
elif state == avahi.SERVER_RUNNING:
548
626
def entry_group_state_changed(state, error):
549
627
"""Derived from the Avahi example code"""
550
global serviceName, server, rename_count
552
logger.debug(u"state change: %i", state)
628
logger.debug(u"Avahi state change: %i", state)
554
630
if state == avahi.ENTRY_GROUP_ESTABLISHED:
555
logger.debug(u"Service established.")
631
logger.debug(u"Zeroconf service established.")
556
632
elif state == avahi.ENTRY_GROUP_COLLISION:
558
rename_count = rename_count - 1
560
name = server.GetAlternativeServiceName(name)
561
logger.warning(u"Service name collision, "
562
u"changing name to '%s' ...", name)
567
logger.error(u"No suitable service name found after %i"
568
u" retries, exiting.", n_rename)
633
logger.warning(u"Zeroconf service name collision.")
570
635
elif state == avahi.ENTRY_GROUP_FAILURE:
571
logger.error(u"Error in group state changed %s",
636
logger.critical(u"Avahi: Error in group state changed %s",
638
raise AvahiGroupError("State changed: %s", str(error))
576
640
def if_nametoindex(interface):
577
"""Call the C function if_nametoindex()"""
641
"""Call the C function if_nametoindex(), or equivalent"""
642
global if_nametoindex
579
libc = ctypes.cdll.LoadLibrary("libc.so.6")
580
return libc.if_nametoindex(interface)
644
if_nametoindex = ctypes.cdll.LoadLibrary\
645
(ctypes.util.find_library("c")).if_nametoindex
581
646
except (OSError, AttributeError):
582
647
if "struct" not in sys.modules:
584
649
if "fcntl" not in sys.modules:
586
SIOCGIFINDEX = 0x8933 # From /usr/include/linux/sockios.h
588
ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
589
struct.pack("16s16x", interface))
591
interface_index = struct.unpack("I", ifreq[16:20])[0]
592
return interface_index
595
def daemon(nochdir, noclose):
651
def if_nametoindex(interface):
652
"Get an interface index the hard way, i.e. using fcntl()"
653
SIOCGIFINDEX = 0x8933 # From /usr/include/linux/sockios.h
655
ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
656
struct.pack("16s16x", interface))
658
interface_index = struct.unpack("I", ifreq[16:20])[0]
659
return interface_index
660
return if_nametoindex(interface)
663
def daemon(nochdir = False, noclose = False):
596
664
"""See daemon(3). Standard BSD Unix function.
597
665
This should really exist as os.daemon, but it doesn't (yet)."""
616
def killme(status = 0):
617
logger.debug("Stopping server with exit status %d", status)
619
if main_loop_started:
628
global main_loop_started
629
main_loop_started = False
631
parser = OptionParser()
687
parser = OptionParser(version = "%%prog %s" % version)
632
688
parser.add_option("-i", "--interface", type="string",
633
default=None, metavar="IF",
634
help="Bind to interface IF")
635
parser.add_option("-a", "--address", type="string", default=None,
689
metavar="IF", help="Bind to interface IF")
690
parser.add_option("-a", "--address", type="string",
636
691
help="Address to listen for requests on")
637
parser.add_option("-p", "--port", type="int", default=None,
692
parser.add_option("-p", "--port", type="int",
638
693
help="Port number to receive requests on")
639
parser.add_option("--timeout", type="string", # Parsed later
641
help="Amount of downtime allowed for clients")
642
parser.add_option("--interval", type="string", # Parsed later
644
help="How often to check that a client is up")
645
694
parser.add_option("--check", action="store_true", default=False,
646
695
help="Run self-test")
647
parser.add_option("--debug", action="store_true", default=False,
649
(options, args) = parser.parse_args()
696
parser.add_option("--debug", action="store_true",
697
help="Debug mode; run in foreground and log to"
699
parser.add_option("--priority", type="string", help="GnuTLS"
700
" priority string (see GnuTLS documentation)")
701
parser.add_option("--servicename", type="string", metavar="NAME",
702
help="Zeroconf service name")
703
parser.add_option("--configdir", type="string",
704
default="/etc/mandos", metavar="DIR",
705
help="Directory to search for configuration"
707
options = parser.parse_args()[0]
651
709
if options.check:
653
711
doctest.testmod()
656
# Parse the time arguments
658
options.timeout = string_to_delta(options.timeout)
660
parser.error("option --timeout: Unparseable time")
662
options.interval = string_to_delta(options.interval)
664
parser.error("option --interval: Unparseable time")
667
defaults = { "checker": "fping -q -- %%(fqdn)s" }
668
client_config = ConfigParser.SafeConfigParser(defaults)
669
#client_config.readfp(open("global.conf"), "global.conf")
670
client_config.read("mandos-clients.conf")
714
# Default values for config file for server-global settings
715
server_defaults = { "interface": "",
720
"SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
721
"servicename": "Mandos",
724
# Parse config file for server-global settings
725
server_config = ConfigParser.SafeConfigParser(server_defaults)
727
server_config.read(os.path.join(options.configdir, "mandos.conf"))
728
# Convert the SafeConfigParser object to a dict
729
server_settings = server_config.defaults()
730
# Use getboolean on the boolean config option
731
server_settings["debug"] = server_config.getboolean\
735
# Override the settings from the config file with command line
737
for option in ("interface", "address", "port", "debug",
738
"priority", "servicename", "configdir"):
739
value = getattr(options, option)
740
if value is not None:
741
server_settings[option] = value
743
# Now we have our good server settings in "server_settings"
745
debug = server_settings["debug"]
748
syslogger.setLevel(logging.WARNING)
749
console.setLevel(logging.WARNING)
751
if server_settings["servicename"] != "Mandos":
752
syslogger.setFormatter(logging.Formatter\
753
('Mandos (%s): %%(levelname)s:'
755
% server_settings["servicename"]))
757
# Parse config file with clients
758
client_defaults = { "timeout": "1h",
760
"checker": "fping -q -- %(host)s",
763
client_config = ConfigParser.SafeConfigParser(client_defaults)
764
client_config.read(os.path.join(server_settings["configdir"],
768
tcp_server = IPv6_TCPServer((server_settings["address"],
769
server_settings["port"]),
771
settings=server_settings,
773
pidfilename = "/var/run/mandos.pid"
775
pidfile = open(pidfilename, "w")
776
except IOError, error:
777
logger.error("Could not open file %r", pidfilename)
782
uid = pwd.getpwnam("mandos").pw_uid
785
uid = pwd.getpwnam("nobody").pw_uid
789
gid = pwd.getpwnam("mandos").pw_gid
792
gid = pwd.getpwnam("nogroup").pw_gid
798
except OSError, error:
799
if error[0] != errno.EPERM:
803
service = AvahiService(name = server_settings["servicename"],
804
servicetype = "_mandos._tcp", )
805
if server_settings["interface"]:
806
service.interface = if_nametoindex\
807
(server_settings["interface"])
676
813
DBusGMainLoop(set_as_default=True )
677
814
main_loop = gobject.MainLoop()
678
815
bus = dbus.SystemBus()
679
server = dbus.Interface(
680
bus.get_object( avahi.DBUS_NAME, avahi.DBUS_PATH_SERVER ),
681
avahi.DBUS_INTERFACE_SERVER )
816
server = dbus.Interface(bus.get_object(avahi.DBUS_NAME,
817
avahi.DBUS_PATH_SERVER),
818
avahi.DBUS_INTERFACE_SERVER)
682
819
# End of Avahi example code
684
debug = options.debug
687
console = logging.StreamHandler()
688
# console.setLevel(logging.DEBUG)
689
console.setFormatter(logging.Formatter\
690
('%(levelname)s: %(message)s'))
691
logger.addHandler(console)
695
821
def remove_from_clients(client):
696
822
clients.remove(client)
698
logger.debug(u"No clients left, exiting")
824
logger.critical(u"No clients left, exiting")
701
clients.update(Set(Client(name=section, options=options,
827
clients.update(Set(Client(name = section,
702
828
stop_hook = remove_from_clients,
703
**(dict(client_config\
830
= dict(client_config.items(section)))
705
831
for section in client_config.sections()))
833
logger.critical(u"No clients defined")
837
# Redirect stdin so all checkers get /dev/null
838
null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
839
os.dup2(null, sys.stdin.fileno())
844
logger.removeHandler(console)
845
# Close all input and output, do double fork, etc.
850
pidfile.write(str(pid) + "\n")
854
logger.error(u"Could not write to file %r with PID %d",
857
# "pidfile" was never created
711
862
"Cleanup function; run on exit"
727
878
signal.signal(signal.SIGINT, signal.SIG_IGN)
728
signal.signal(signal.SIGHUP, lambda signum, frame: killme())
729
signal.signal(signal.SIGTERM, lambda signum, frame: killme())
879
signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
880
signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
731
882
for client in clients:
734
tcp_server = IPv6_TCPServer((options.address, options.port),
738
# Find out what random port we got
740
servicePort = tcp_server.socket.getsockname()[1]
741
logger.debug(u"Now listening on port %d", servicePort)
743
if options.interface is not None:
744
global serviceInterface
745
serviceInterface = if_nametoindex(options.interface)
747
# From the Avahi example code
748
server.connect_to_signal("StateChanged", server_state_changed)
750
server_state_changed(server.GetState())
751
except dbus.exceptions.DBusException, error:
752
logger.critical(u"DBusException: %s", error)
754
# End of Avahi example code
756
gobject.io_add_watch(tcp_server.fileno(), gobject.IO_IN,
757
lambda *args, **kwargs:
758
tcp_server.handle_request(*args[2:],
761
logger.debug("Starting main loop")
762
main_loop_started = True
886
tcp_server.server_activate()
888
# Find out what port we got
889
service.port = tcp_server.socket.getsockname()[1]
890
logger.info(u"Now listening on address %r, port %d, flowinfo %d,"
891
u" scope_id %d" % tcp_server.socket.getsockname())
893
#service.interface = tcp_server.socket.getsockname()[3]
896
# From the Avahi example code
897
server.connect_to_signal("StateChanged", server_state_changed)
899
server_state_changed(server.GetState())
900
except dbus.exceptions.DBusException, error:
901
logger.critical(u"DBusException: %s", error)
903
# End of Avahi example code
905
gobject.io_add_watch(tcp_server.fileno(), gobject.IO_IN,
906
lambda *args, **kwargs:
907
tcp_server.handle_request\
908
(*args[2:], **kwargs) or True)
910
logger.debug(u"Starting main loop")
912
except AvahiError, error:
913
logger.critical(u"AvahiError: %s" + unicode(error))
764
915
except KeyboardInterrupt:
770
919
if __name__ == '__main__':