62
61
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
66
logger = logging.Logger('mandos')
78
67
syslogger = logging.handlers.SysLogHandler\
79
(facility = logging.handlers.SysLogHandler.LOG_DAEMON)
68
(facility = logging.handlers.SysLogHandler.LOG_DAEMON,
80
70
syslogger.setFormatter(logging.Formatter\
81
('%(levelname)s: %(message)s'))
71
('Mandos: %(levelname)s: %(message)s'))
82
72
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
74
console = logging.StreamHandler()
75
console.setFormatter(logging.Formatter('%(name)s: %(levelname)s:'
77
logger.addHandler(console)
79
class AvahiError(Exception):
80
def __init__(self, value):
83
return repr(self.value)
85
class AvahiServiceError(AvahiError):
88
class AvahiGroupError(AvahiError):
92
class AvahiService(object):
93
"""An Avahi (Zeroconf) service.
95
interface: integer; avahi.IF_UNSPEC or an interface index.
96
Used to optionally bind to the specified interface.
97
name: string; Example: 'Mandos'
98
type: string; Example: '_mandos._tcp'.
99
See <http://www.dns-sd.org/ServiceTypes.html>
100
port: integer; what port to announce
101
TXT: list of strings; TXT record for the service
102
domain: string; Domain to publish on, default to .local if empty.
103
host: string; Host to publish records for, default is localhost
104
max_renames: integer; maximum number of renames
105
rename_count: integer; counter so we only rename after collisions
106
a sensible number of times
108
def __init__(self, interface = avahi.IF_UNSPEC, name = None,
109
type = None, port = None, TXT = None, domain = "",
110
host = "", max_renames = 32768):
111
self.interface = interface
121
self.rename_count = 0
122
self.max_renames = max_renames
124
"""Derived from the Avahi example code"""
125
if self.rename_count >= self.max_renames:
126
logger.critical(u"No suitable service name found after %i"
127
u" retries, exiting.", rename_count)
128
raise AvahiServiceError("Too many renames")
129
self.name = server.GetAlternativeServiceName(self.name)
130
logger.info(u"Changing name to %r ...", str(self.name))
131
syslogger.setFormatter(logging.Formatter\
132
('Mandos (%s): %%(levelname)s:'
133
' %%(message)s' % self.name))
136
self.rename_count += 1
138
"""Derived from the Avahi example code"""
139
if group is not None:
142
"""Derived from the Avahi example code"""
145
group = dbus.Interface\
146
(bus.get_object(avahi.DBUS_NAME,
147
server.EntryGroupNew()),
148
avahi.DBUS_INTERFACE_ENTRY_GROUP)
149
group.connect_to_signal('StateChanged',
150
entry_group_state_changed)
151
logger.debug(u"Adding service '%s' of type '%s' ...",
152
service.name, service.type)
154
self.interface, # interface
155
avahi.PROTO_INET6, # protocol
156
dbus.UInt32(0), # flags
157
self.name, self.type,
158
self.domain, self.host,
159
dbus.UInt16(self.port),
160
avahi.string_array_to_txt_array(self.TXT))
89
163
# 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
164
group = None # our entry group
99
165
# End of Avahi example code
106
172
fingerprint: string (40 or 32 hexadecimal digits); used to
107
173
uniquely identify the client
108
174
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
175
host: string; available for use by the checker command
176
created: datetime.datetime(); object creation, not client host
177
last_checked_ok: datetime.datetime() or None if not yet checked OK
178
timeout: datetime.timedelta(); How long from last_checked_ok
179
until this client is invalid
114
180
interval: datetime.timedelta(); How often to start a new checker
115
181
stop_hook: If set, called by stop() as stop_hook(self)
116
182
checker: subprocess.Popen(); a running checker process used
117
183
to see if the client lives.
118
Is None if no process is running.
184
'None' if no process is running.
119
185
checker_initiator_tag: a gobject event source tag, or None
120
186
stop_initiator_tag: - '' -
121
187
checker_callback_tag: - '' -
122
188
checker_command: string; External command which is run to check if
123
client lives. %()s expansions are done at
189
client lives. %() expansions are done at
124
190
runtime with vars(self) as dict, so that for
125
191
instance %(name)s can be used in the command.
126
192
Private attibutes:
127
193
_timeout: Real variable for 'timeout'
128
194
_interval: Real variable for 'interval'
129
_timeout_milliseconds: Used by gobject.timeout_add()
195
_timeout_milliseconds: Used when calling gobject.timeout_add()
130
196
_interval_milliseconds: - '' -
132
198
def _set_timeout(self, timeout):
152
218
interval = property(lambda self: self._interval,
154
220
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.."""
221
def __init__(self, name = None, stop_hook=None, config={}):
222
"""Note: the 'checker' key in 'config' sets the
223
'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")
226
logger.debug(u"Creating client %r", self.name)
227
# Uppercase and remove spaces from fingerprint for later
228
# comparison purposes with return value from the fingerprint()
230
self.fingerprint = config["fingerprint"].upper()\
232
logger.debug(u" Fingerprint: %s", self.fingerprint)
233
if "secret" in config:
234
self.secret = config["secret"].decode(u"base64")
235
elif "secfile" in config:
236
sf = open(config["secfile"])
169
237
self.secret = sf.read()
172
raise RuntimeError(u"No secret or secfile for client %s"
174
self.fqdn = fqdn # string
240
raise TypeError(u"No secret or secfile for client %s"
242
self.host = config.get("host", "")
175
243
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)
244
self.last_checked_ok = None
245
self.timeout = string_to_delta(config["timeout"])
246
self.interval = string_to_delta(config["interval"])
185
247
self.stop_hook = stop_hook
186
248
self.checker = None
187
249
self.checker_initiator_tag = None
188
250
self.stop_initiator_tag = None
189
251
self.checker_callback_tag = None
190
self.check_command = checker
252
self.check_command = config["checker"]
192
254
"""Start this client's checker and timeout hooks"""
193
255
# Schedule a new checker to be started an 'interval' from now,
205
267
"""Stop this client.
206
The possibility that this client might be restarted is left
207
open, but not currently used."""
268
The possibility that a client might be restarted is left open,
269
but not currently used."""
208
270
# If this client doesn't have a secret, it is already stopped.
210
logger.debug(u"Stopping client %s", self.name)
271
if hasattr(self, "secret") and self.secret:
272
logger.info(u"Stopping client %s", self.name)
211
273
self.secret = None
214
if hasattr(self, "stop_initiator_tag") \
215
and self.stop_initiator_tag:
276
if getattr(self, "stop_initiator_tag", False):
216
277
gobject.source_remove(self.stop_initiator_tag)
217
278
self.stop_initiator_tag = None
218
if hasattr(self, "checker_initiator_tag") \
219
and self.checker_initiator_tag:
279
if getattr(self, "checker_initiator_tag", False):
220
280
gobject.source_remove(self.checker_initiator_tag)
221
281
self.checker_initiator_tag = None
222
282
self.stop_checker()
261
321
# is as it should be.
262
322
if self.checker is None:
264
command = self.check_command % self.fqdn
324
# In case check_command has exactly one % operator
325
command = self.check_command % self.host
265
326
except TypeError:
327
# Escape attributes for the shell
266
328
escaped_attrs = dict((key, re.escape(str(val)))
268
330
vars(self).iteritems())
270
332
command = self.check_command % escaped_attrs
271
333
except TypeError, error:
272
logger.critical(u'Could not format string "%s":'
273
u' %s', self.check_command, error)
334
logger.error(u'Could not format string "%s":'
335
u' %s', self.check_command, error)
274
336
return True # Try again later
276
logger.debug(u"Starting checker %r for %s",
278
self.checker = subprocess.\
280
close_fds=True, shell=True,
338
logger.info(u"Starting checker %r for %s",
340
# We don't need to redirect stdout and stderr, since
341
# in normal mode, that is already done by daemon(),
342
# and in debug mode we don't want to. (Stdin is
343
# always replaced by /dev/null.)
344
self.checker = subprocess.Popen(command,
282
347
self.checker_callback_tag = gobject.child_watch_add\
283
348
(self.checker.pid,
284
349
self.checker_callback)
285
except subprocess.OSError, error:
350
except OSError, error:
286
351
logger.error(u"Failed to start subprocess: %s",
288
353
# Re-run this periodically if run by gobject.timeout_add
292
357
if self.checker_callback_tag:
293
358
gobject.source_remove(self.checker_callback_tag)
294
359
self.checker_callback_tag = None
295
if not hasattr(self, "checker") or self.checker is None:
360
if getattr(self, "checker", None) is None:
297
logger.debug("Stopping checker for %(name)s", vars(self))
362
logger.debug(u"Stopping checker for %(name)s", vars(self))
299
364
os.kill(self.checker.pid, signal.SIGTERM)
301
366
#if self.checker.poll() is None:
302
367
# os.kill(self.checker.pid, signal.SIGKILL)
303
368
except OSError, error:
304
if error.errno != errno.ESRCH:
369
if error.errno != errno.ESRCH: # No such process
306
371
self.checker = None
307
def still_valid(self, now=None):
372
def still_valid(self):
308
373
"""Has the timeout not yet passed for this client?"""
310
now = datetime.datetime.now()
311
if self.last_seen is None:
374
now = datetime.datetime.now()
375
if self.last_checked_ok is None:
312
376
return now < (self.created + self.timeout)
314
return now < (self.last_seen + self.timeout)
378
return now < (self.last_checked_ok + self.timeout)
317
381
def peer_certificate(session):
334
398
def fingerprint(openpgp):
335
399
"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
400
# New GnuTLS "datum" with the OpenPGP public key
341
401
datum = gnutls.library.types.gnutls_datum_t\
342
402
(ctypes.cast(ctypes.c_char_p(openpgp),
343
403
ctypes.POINTER(ctypes.c_ubyte)),
344
404
ctypes.c_uint(len(openpgp)))
405
# New empty GnuTLS certificate
406
crt = gnutls.library.types.gnutls_openpgp_crt_t()
407
gnutls.library.functions.gnutls_openpgp_crt_init\
345
409
# 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)
410
gnutls.library.functions.gnutls_openpgp_crt_import\
411
(crt, ctypes.byref(datum),
412
gnutls.library.constants.GNUTLS_OPENPGP_FMT_RAW)
413
# Verify the self signature in the key
414
crtverify = ctypes.c_uint();
415
gnutls.library.functions.gnutls_openpgp_crt_verify_self\
416
(crt, ctypes.c_uint(0), ctypes.byref(crtverify))
418
gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
419
raise gnutls.errors.CertificateSecurityError("Verify failed")
350
420
# New buffer for the fingerprint
351
421
buffer = ctypes.create_string_buffer(20)
352
422
buffer_length = ctypes.c_size_t()
368
438
Note: This will run in its own forked process."""
370
440
def handle(self):
371
logger.debug(u"TCP connection from: %s",
441
logger.info(u"TCP connection from: %s",
372
442
unicode(self.client_address))
373
session = gnutls.connection.ClientSession(self.request,
443
session = gnutls.connection.ClientSession\
444
(self.request, gnutls.connection.X509Credentials())
446
line = self.request.makefile().readline()
447
logger.debug(u"Protocol version: %r", line)
449
if int(line.strip().split()[0]) > 1:
451
except (ValueError, IndexError, RuntimeError), error:
452
logger.error(u"Unknown protocol version: %s", error)
455
# Note: gnutls.connection.X509Credentials is really a generic
456
# GnuTLS certificate credentials object so long as no X.509
457
# keys are added to it. Therefore, we can use it here despite
458
# using OpenPGP certificates.
377
460
#priority = ':'.join(("NONE", "+VERS-TLS1.1", "+AES-256-CBC",
378
461
# "+SHA1", "+COMP-NULL", "+CTYPE-OPENPGP",
380
priority = "SECURE256"
463
priority = "NORMAL" # Fallback default, since this
465
if self.server.settings["priority"]:
466
priority = self.server.settings["priority"]
382
467
gnutls.library.functions.gnutls_priority_set_direct\
383
468
(session._c_object, priority, None);
386
471
session.handshake()
387
472
except gnutls.errors.GNUTLSError, error:
388
logger.debug(u"Handshake failed: %s", error)
473
logger.warning(u"Handshake failed: %s", error)
389
474
# Do not run session.bye() here: the session is not
390
475
# established. Just abandon the request.
393
478
fpr = fingerprint(peer_certificate(session))
394
479
except (TypeError, gnutls.errors.GNUTLSError), error:
395
logger.debug(u"Bad certificate: %s", error)
480
logger.warning(u"Bad certificate: %s", error)
398
483
logger.debug(u"Fingerprint: %s", fpr)
442
528
"""This overrides the normal server_bind() function
443
529
to bind to an interface if one was specified, and also NOT to
444
530
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
531
if self.settings["interface"]:
532
# 25 is from /usr/include/asm-i486/socket.h
533
SO_BINDTODEVICE = getattr(socket, "SO_BINDTODEVICE", 25)
450
535
self.socket.setsockopt(socket.SOL_SOCKET,
451
socket.SO_BINDTODEVICE,
452
self.options.interface)
537
self.settings["interface"])
453
538
except socket.error, error:
454
539
if error[0] == errno.EPERM:
455
logger.warning(u"No permission to"
456
u" bind to interface %s",
457
self.options.interface)
540
logger.error(u"No permission to"
541
u" bind to interface %s",
542
self.settings["interface"])
460
545
# Only bind(2) the socket if we really need to.
482
574
datetime.timedelta(1)
483
575
>>> string_to_delta(u'1w')
484
576
datetime.timedelta(7)
577
>>> string_to_delta('5m 30s')
578
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)
580
timevalue = datetime.timedelta(0)
581
for s in interval.split():
583
suffix=unicode(s[-1])
586
delta = datetime.timedelta(value)
588
delta = datetime.timedelta(0, value)
590
delta = datetime.timedelta(0, 0, 0, 0, value)
592
delta = datetime.timedelta(0, 0, 0, 0, 0, value)
594
delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
597
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
603
def server_state_changed(state):
540
604
"""Derived from the Avahi example code"""
541
605
if state == avahi.SERVER_COLLISION:
542
logger.warning(u"Server name collision")
606
logger.error(u"Server name collision")
544
608
elif state == avahi.SERVER_RUNNING:
548
612
def entry_group_state_changed(state, error):
549
613
"""Derived from the Avahi example code"""
550
global serviceName, server, rename_count
552
614
logger.debug(u"state change: %i", state)
554
616
if state == avahi.ENTRY_GROUP_ESTABLISHED:
555
617
logger.debug(u"Service established.")
556
618
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)
619
logger.warning(u"Service name collision.")
570
621
elif state == avahi.ENTRY_GROUP_FAILURE:
571
logger.error(u"Error in group state changed %s",
622
logger.critical(u"Error in group state changed %s",
624
raise AvahiGroupError("State changed: %s", str(error))
576
626
def if_nametoindex(interface):
577
"""Call the C function if_nametoindex()"""
627
"""Call the C function if_nametoindex(), or equivalent"""
628
global if_nametoindex
579
libc = ctypes.cdll.LoadLibrary("libc.so.6")
580
return libc.if_nametoindex(interface)
630
if "ctypes.util" not in sys.modules:
632
if_nametoindex = ctypes.cdll.LoadLibrary\
633
(ctypes.util.find_library("c")).if_nametoindex
581
634
except (OSError, AttributeError):
582
635
if "struct" not in sys.modules:
584
637
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):
639
def if_nametoindex(interface):
640
"Get an interface index the hard way, i.e. using fcntl()"
641
SIOCGIFINDEX = 0x8933 # From /usr/include/linux/sockios.h
643
ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
644
struct.pack("16s16x", interface))
646
interface_index = struct.unpack("I", ifreq[16:20])[0]
647
return interface_index
648
return if_nametoindex(interface)
651
def daemon(nochdir = False, noclose = False):
596
652
"""See daemon(3). Standard BSD Unix function.
597
653
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
675
global main_loop_started
629
676
main_loop_started = False
631
parser = OptionParser()
678
parser = OptionParser(version = "%%prog %s" % version)
632
679
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,
680
metavar="IF", help="Bind to interface IF")
681
parser.add_option("-a", "--address", type="string",
636
682
help="Address to listen for requests on")
637
parser.add_option("-p", "--port", type="int", default=None,
683
parser.add_option("-p", "--port", type="int",
638
684
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
685
parser.add_option("--check", action="store_true", default=False,
646
686
help="Run self-test")
647
parser.add_option("--debug", action="store_true", default=False,
687
parser.add_option("--debug", action="store_true",
688
help="Debug mode; run in foreground and log to"
690
parser.add_option("--priority", type="string", help="GnuTLS"
691
" priority string (see GnuTLS documentation)")
692
parser.add_option("--servicename", type="string", metavar="NAME",
693
help="Zeroconf service name")
694
parser.add_option("--configdir", type="string",
695
default="/etc/mandos", metavar="DIR",
696
help="Directory to search for configuration"
649
698
(options, args) = parser.parse_args()
651
700
if options.check:
653
702
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")
705
# Default values for config file for server-global settings
706
server_defaults = { "interface": "",
711
"SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
712
"servicename": "Mandos",
715
# Parse config file for server-global settings
716
server_config = ConfigParser.SafeConfigParser(server_defaults)
718
server_config.read(os.path.join(options.configdir, "mandos.conf"))
719
# Convert the SafeConfigParser object to a dict
720
server_settings = server_config.defaults()
721
# Use getboolean on the boolean config option
722
server_settings["debug"] = server_config.getboolean\
726
# Override the settings from the config file with command line
728
for option in ("interface", "address", "port", "debug",
729
"priority", "servicename", "configdir"):
730
value = getattr(options, option)
731
if value is not None:
732
server_settings[option] = value
734
# Now we have our good server settings in "server_settings"
736
debug = server_settings["debug"]
739
syslogger.setLevel(logging.WARNING)
740
console.setLevel(logging.WARNING)
742
if server_settings["servicename"] != "Mandos":
743
syslogger.setFormatter(logging.Formatter\
744
('Mandos (%s): %%(levelname)s:'
746
% server_settings["servicename"]))
748
# Parse config file with clients
749
client_defaults = { "timeout": "1h",
751
"checker": "fping -q -- %(host)s",
754
client_config = ConfigParser.SafeConfigParser(client_defaults)
755
client_config.read(os.path.join(server_settings["configdir"],
759
service = AvahiService(name = server_settings["servicename"],
760
type = "_mandos._tcp", );
761
if server_settings["interface"]:
762
service.interface = if_nametoindex(server_settings["interface"])
681
773
avahi.DBUS_INTERFACE_SERVER )
682
774
# 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
777
def remove_from_clients(client):
696
778
clients.remove(client)
698
logger.debug(u"No clients left, exiting")
780
logger.critical(u"No clients left, exiting")
701
clients.update(Set(Client(name=section, options=options,
783
clients.update(Set(Client(name = section,
702
784
stop_hook = remove_from_clients,
703
**(dict(client_config\
786
= dict(client_config.items(section)))
705
787
for section in client_config.sections()))
789
logger.critical(u"No clients defined")
793
# Redirect stdin so all checkers get /dev/null
794
null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
795
os.dup2(null, sys.stdin.fileno())
800
logger.removeHandler(console)
801
# Close all input and output, do double fork, etc.
804
pidfilename = "/var/run/mandos/mandos.pid"
807
pidfile = open(pidfilename, "w")
808
pidfile.write(str(pid) + "\n")
812
logger.error(u"Could not write %s file with PID %d",
813
pidfilename, os.getpid())
711
816
"Cleanup function; run on exit"
727
832
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())
833
signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
834
signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
731
836
for client in clients:
734
tcp_server = IPv6_TCPServer((options.address, options.port),
839
tcp_server = IPv6_TCPServer((server_settings["address"],
840
server_settings["port"]),
842
settings=server_settings,
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")
844
# Find out what port we got
845
service.port = tcp_server.socket.getsockname()[1]
846
logger.info(u"Now listening on address %r, port %d, flowinfo %d,"
847
u" scope_id %d" % tcp_server.socket.getsockname())
849
#service.interface = tcp_server.socket.getsockname()[3]
852
# From the Avahi example code
853
server.connect_to_signal("StateChanged", server_state_changed)
855
server_state_changed(server.GetState())
856
except dbus.exceptions.DBusException, error:
857
logger.critical(u"DBusException: %s", error)
859
# End of Avahi example code
861
gobject.io_add_watch(tcp_server.fileno(), gobject.IO_IN,
862
lambda *args, **kwargs:
863
tcp_server.handle_request\
864
(*args[2:], **kwargs) or True)
866
logger.debug(u"Starting main loop")
762
867
main_loop_started = True
869
except AvahiError, error:
870
logger.critical(u"AvahiError: %s" + unicode(error))
764
872
except KeyboardInterrupt:
770
876
if __name__ == '__main__':