28
62
from dbus.mainloop.glib import DBusGMainLoop
32
import logging.handlers
34
# logghandler.setFormatter(logging.Formatter('%(levelname)s %(message)s')
36
67
logger = logging.Logger('mandos')
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
68
syslogger = logging.handlers.SysLogHandler\
69
(facility = logging.handlers.SysLogHandler.LOG_DAEMON,
71
syslogger.setFormatter(logging.Formatter\
72
('Mandos: %(levelname)s: %(message)s'))
73
logger.addHandler(syslogger)
75
console = logging.StreamHandler()
76
console.setFormatter(logging.Formatter('%(name)s: %(levelname)s:'
78
logger.addHandler(console)
80
class AvahiError(Exception):
81
def __init__(self, value):
84
return repr(self.value)
86
class AvahiServiceError(AvahiError):
89
class AvahiGroupError(AvahiError):
93
class AvahiService(object):
94
"""An Avahi (Zeroconf) service.
96
interface: integer; avahi.IF_UNSPEC or an interface index.
97
Used to optionally bind to the specified interface.
98
name: string; Example: 'Mandos'
99
type: string; Example: '_mandos._tcp'.
100
See <http://www.dns-sd.org/ServiceTypes.html>
101
port: integer; what port to announce
102
TXT: list of strings; TXT record for the service
103
domain: string; Domain to publish on, default to .local if empty.
104
host: string; Host to publish records for, default is localhost
105
max_renames: integer; maximum number of renames
106
rename_count: integer; counter so we only rename after collisions
107
a sensible number of times
109
def __init__(self, interface = avahi.IF_UNSPEC, name = None,
110
type = None, port = None, TXT = None, domain = "",
111
host = "", max_renames = 32768):
112
self.interface = interface
122
self.rename_count = 0
123
self.max_renames = max_renames
125
"""Derived from the Avahi example code"""
126
if self.rename_count >= self.max_renames:
127
logger.critical(u"No suitable Zeroconf service name found"
128
u" after %i retries, exiting.",
130
raise AvahiServiceError("Too many renames")
131
self.name = server.GetAlternativeServiceName(self.name)
132
logger.info(u"Changing Zeroconf service name to %r ...",
134
syslogger.setFormatter(logging.Formatter\
135
('Mandos (%s): %%(levelname)s:'
136
' %%(message)s' % self.name))
139
self.rename_count += 1
141
"""Derived from the Avahi example code"""
142
if group is not None:
145
"""Derived from the Avahi example code"""
148
group = dbus.Interface\
149
(bus.get_object(avahi.DBUS_NAME,
150
server.EntryGroupNew()),
151
avahi.DBUS_INTERFACE_ENTRY_GROUP)
152
group.connect_to_signal('StateChanged',
153
entry_group_state_changed)
154
logger.debug(u"Adding Zeroconf service '%s' of type '%s' ...",
155
service.name, service.type)
157
self.interface, # interface
158
avahi.PROTO_INET6, # protocol
159
dbus.UInt32(0), # flags
160
self.name, self.type,
161
self.domain, self.host,
162
dbus.UInt16(self.port),
163
avahi.string_array_to_txt_array(self.TXT))
166
# From the Avahi example code:
167
group = None # our entry group
53
168
# End of Avahi example code
60
175
fingerprint: string (40 or 32 hexadecimal digits); used to
61
176
uniquely identify the client
62
177
secret: bytestring; sent verbatim (over TLS) to client
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
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
68
183
interval: datetime.timedelta(); How often to start a new checker
69
184
stop_hook: If set, called by stop() as stop_hook(self)
70
185
checker: subprocess.Popen(); a running checker process used
71
186
to see if the client lives.
72
Is None if no process is running.
187
'None' if no process is running.
73
188
checker_initiator_tag: a gobject event source tag, or None
74
189
stop_initiator_tag: - '' -
75
190
checker_callback_tag: - '' -
76
191
checker_command: string; External command which is run to check if
77
client lives. %()s expansions are done at
192
client lives. %() expansions are done at
78
193
runtime with vars(self) as dict, so that for
79
194
instance %(name)s can be used in the command.
80
195
Private attibutes:
81
196
_timeout: Real variable for 'timeout'
82
197
_interval: Real variable for 'interval'
83
_timeout_milliseconds: Used by gobject.timeout_add()
198
_timeout_milliseconds: Used when calling gobject.timeout_add()
84
199
_interval_milliseconds: - '' -
86
201
def _set_timeout(self, timeout):
106
221
interval = property(lambda self: self._interval,
108
223
del _set_interval
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):
224
def __init__(self, name = None, stop_hook=None, config={}):
225
"""Note: the 'checker' key in 'config' sets the
226
'checker_command' attribute and *not* the 'checker'
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")
229
logger.debug(u"Creating client %r", self.name)
230
# Uppercase and remove spaces from fingerprint for later
231
# comparison purposes with return value from the fingerprint()
233
self.fingerprint = config["fingerprint"].upper()\
235
logger.debug(u" Fingerprint: %s", self.fingerprint)
236
if "secret" in config:
237
self.secret = config["secret"].decode(u"base64")
238
elif "secfile" in config:
239
sf = open(config["secfile"])
121
240
self.secret = sf.read()
124
raise RuntimeError(u"No secret or secfile for client %s"
126
self.fqdn = fqdn # string
243
raise TypeError(u"No secret or secfile for client %s"
245
self.host = config.get("host", "")
127
246
self.created = datetime.datetime.now()
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
247
self.last_checked_ok = None
248
self.timeout = string_to_delta(config["timeout"])
249
self.interval = string_to_delta(config["interval"])
137
250
self.stop_hook = stop_hook
138
251
self.checker = None
139
252
self.checker_initiator_tag = None
140
253
self.stop_initiator_tag = None
141
254
self.checker_callback_tag = None
142
self.check_command = checker
255
self.check_command = config["checker"]
144
"""Start this clients checker and timeout hooks"""
257
"""Start this client's checker and timeout hooks"""
145
258
# Schedule a new checker to be started an 'interval' from now,
146
259
# and every interval from then on.
147
260
self.checker_initiator_tag = gobject.timeout_add\
171
288
# Do not run this again if called by a gobject.timeout_add
173
290
def __del__(self):
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
291
self.stop_hook = None
184
293
def checker_callback(self, pid, condition):
185
294
"""The checker has completed, so take appropriate actions."""
186
295
now = datetime.datetime.now()
296
self.checker_callback_tag = None
187
298
if os.WIFEXITED(condition) \
188
299
and (os.WEXITSTATUS(condition) == 0):
189
logger.debug(u"Checker for %(name)s succeeded",
300
logger.info(u"Checker for %(name)s succeeded",
302
self.last_checked_ok = now
192
303
gobject.source_remove(self.stop_initiator_tag)
193
304
self.stop_initiator_tag = gobject.timeout_add\
194
305
(self._timeout_milliseconds,
196
if not os.WIFEXITED(condition):
307
elif not os.WIFEXITED(condition):
197
308
logger.warning(u"Checker for %(name)s crashed?",
200
logger.debug(u"Checker for %(name)s failed",
203
self.checker_callback_tag = None
311
logger.info(u"Checker for %(name)s failed",
204
313
def start_checker(self):
205
314
"""Start a new checker subprocess if one is not running.
206
315
If a checker already exists, leave it running and do
317
# The reason for not killing a running checker is that if we
318
# did that, then if a checker (for some reason) started
319
# running slowly and taking more than 'interval' time, the
320
# client would inevitably timeout, since no checker would get
321
# a chance to run to completion. If we instead leave running
322
# checkers alone, the checker would have to take more time
323
# than 'timeout' for the client to be declared invalid, which
324
# is as it should be.
208
325
if self.checker is None:
209
logger.debug(u"Starting checker for %s",
212
command = self.check_command % self.fqdn
327
# In case check_command has exactly one % operator
328
command = self.check_command % self.host
213
329
except TypeError:
330
# Escape attributes for the shell
214
331
escaped_attrs = dict((key, re.escape(str(val)))
216
333
vars(self).iteritems())
218
335
command = self.check_command % escaped_attrs
219
336
except TypeError, error:
220
logger.critical(u'Could not format string "%s": %s',
221
self.check_command, error)
337
logger.error(u'Could not format string "%s":'
338
u' %s', self.check_command, error)
222
339
return True # Try again later
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:
341
logger.info(u"Starting checker %r for %s",
343
# We don't need to redirect stdout and stderr, since
344
# in normal mode, that is already done by daemon(),
345
# and in debug mode we don't want to. (Stdin is
346
# always replaced by /dev/null.)
347
self.checker = subprocess.Popen(command,
350
self.checker_callback_tag = gobject.child_watch_add\
352
self.checker_callback)
353
except OSError, error:
234
354
logger.error(u"Failed to start subprocess: %s",
236
356
# Re-run this periodically if run by gobject.timeout_add
238
358
def stop_checker(self):
239
359
"""Force the checker process, if any, to stop."""
240
if not hasattr(self, "checker") or self.checker is None:
360
if self.checker_callback_tag:
361
gobject.source_remove(self.checker_callback_tag)
362
self.checker_callback_tag = None
363
if getattr(self, "checker", None) is None:
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)
365
logger.debug(u"Stopping checker for %(name)s", vars(self))
367
os.kill(self.checker.pid, signal.SIGTERM)
369
#if self.checker.poll() is None:
370
# os.kill(self.checker.pid, signal.SIGKILL)
371
except OSError, error:
372
if error.errno != errno.ESRCH: # No such process
247
374
self.checker = None
248
def still_valid(self, now=None):
375
def still_valid(self):
249
376
"""Has the timeout not yet passed for this client?"""
251
now = datetime.datetime.now()
252
if self.last_seen is None:
377
now = datetime.datetime.now()
378
if self.last_checked_ok is None:
253
379
return now < (self.created + self.timeout)
255
return now < (self.last_seen + self.timeout)
381
return now < (self.last_checked_ok + self.timeout)
258
384
def peer_certificate(session):
385
"Return the peer's OpenPGP certificate as a bytestring"
259
386
# If not an OpenPGP certificate...
260
387
if gnutls.library.functions.gnutls_certificate_type_get\
261
388
(session._c_object) \
307
441
Note: This will run in its own forked process."""
309
443
def handle(self):
310
logger.debug(u"TCP connection from: %s",
444
logger.info(u"TCP connection from: %s",
311
445
unicode(self.client_address))
312
session = gnutls.connection.ClientSession(self.request,
446
session = gnutls.connection.ClientSession\
447
(self.request, gnutls.connection.X509Credentials())
449
line = self.request.makefile().readline()
450
logger.debug(u"Protocol version: %r", line)
452
if int(line.strip().split()[0]) > 1:
454
except (ValueError, IndexError, RuntimeError), error:
455
logger.error(u"Unknown protocol version: %s", error)
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.
316
463
#priority = ':'.join(("NONE", "+VERS-TLS1.1", "+AES-256-CBC",
317
464
# "+SHA1", "+COMP-NULL", "+CTYPE-OPENPGP",
319
priority = "SECURE256"
466
priority = "NORMAL" # Fallback default, since this
468
if self.server.settings["priority"]:
469
priority = self.server.settings["priority"]
321
470
gnutls.library.functions.gnutls_priority_set_direct\
322
471
(session._c_object, priority, None);
325
474
session.handshake()
326
475
except gnutls.errors.GNUTLSError, error:
327
logger.debug(u"Handshake failed: %s", error)
476
logger.warning(u"Handshake failed: %s", error)
328
477
# Do not run session.bye() here: the session is not
329
478
# established. Just abandon the request.
332
481
fpr = fingerprint(peer_certificate(session))
333
482
except (TypeError, gnutls.errors.GNUTLSError), error:
334
logger.debug(u"Bad certificate: %s", error)
483
logger.warning(u"Bad certificate: %s", error)
337
486
logger.debug(u"Fingerprint: %s", fpr)
488
for c in self.server.clients:
340
489
if c.fingerprint == fpr:
493
logger.warning(u"Client not found for fingerprint: %s",
343
497
# Have to check if client.still_valid(), since it is possible
344
498
# that the client timed out while establishing the GnuTLS
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",
500
if not client.still_valid():
501
logger.warning(u"Client %(name)s is invalid",
421
577
datetime.timedelta(1)
422
578
>>> string_to_delta(u'1w')
423
579
datetime.timedelta(7)
580
>>> string_to_delta('5m 30s')
581
datetime.timedelta(0, 330)
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)
583
timevalue = datetime.timedelta(0)
584
for s in interval.split():
586
suffix=unicode(s[-1])
589
delta = datetime.timedelta(value)
591
delta = datetime.timedelta(0, value)
593
delta = datetime.timedelta(0, 0, 0, 0, value)
595
delta = datetime.timedelta(0, 0, 0, 0, 0, value)
597
delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
600
except (ValueError, IndexError):
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:
478
606
def server_state_changed(state):
479
"""From the Avahi server example code"""
607
"""Derived from the Avahi example code"""
480
608
if state == avahi.SERVER_COLLISION:
481
logger.warning(u"Server name collision")
609
logger.error(u"Zeroconf server name collision")
483
611
elif state == avahi.SERVER_RUNNING:
487
615
def entry_group_state_changed(state, error):
488
"""From the Avahi server example code"""
489
global serviceName, server, rename_count
491
logger.debug(u"state change: %i", state)
616
"""Derived from the Avahi example code"""
617
logger.debug(u"Avahi state change: %i", state)
493
619
if state == avahi.ENTRY_GROUP_ESTABLISHED:
494
logger.debug(u"Service established.")
620
logger.debug(u"Zeroconf service established.")
495
621
elif state == avahi.ENTRY_GROUP_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.",
622
logger.warning(u"Zeroconf service name collision.")
510
624
elif state == avahi.ENTRY_GROUP_FAILURE:
511
logger.error(u"Error in group state changed %s",
625
logger.critical(u"Avahi: Error in group state changed %s",
627
raise AvahiGroupError("State changed: %s", str(error))
517
629
def if_nametoindex(interface):
518
"""Call the C function if_nametoindex()"""
630
"""Call the C function if_nametoindex(), or equivalent"""
631
global if_nametoindex
520
libc = ctypes.cdll.LoadLibrary("libc.so.6")
521
return libc.if_nametoindex(interface)
633
if "ctypes.util" not in sys.modules:
635
if_nametoindex = ctypes.cdll.LoadLibrary\
636
(ctypes.util.find_library("c")).if_nametoindex
522
637
except (OSError, AttributeError):
523
638
if "struct" not in sys.modules:
525
640
if "fcntl" not in sys.modules:
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()
642
def if_nametoindex(interface):
643
"Get an interface index the hard way, i.e. using fcntl()"
644
SIOCGIFINDEX = 0x8933 # From /usr/include/linux/sockios.h
646
ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
647
struct.pack("16s16x", interface))
649
interface_index = struct.unpack("I", ifreq[16:20])[0]
650
return interface_index
651
return if_nametoindex(interface)
654
def daemon(nochdir = False, noclose = False):
655
"""See daemon(3). Standard BSD Unix function.
656
This should really exist as os.daemon, but it doesn't (yet)."""
665
# Close all standard open file descriptors
666
null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
667
if not stat.S_ISCHR(os.fstat(null).st_mode):
668
raise OSError(errno.ENODEV,
669
"/dev/null not a character device")
670
os.dup2(null, sys.stdin.fileno())
671
os.dup2(null, sys.stdout.fileno())
672
os.dup2(null, sys.stderr.fileno())
678
global main_loop_started
679
main_loop_started = False
681
parser = OptionParser(version = "%%prog %s" % version)
538
682
parser.add_option("-i", "--interface", type="string",
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,
683
metavar="IF", help="Bind to interface IF")
684
parser.add_option("-a", "--address", type="string",
685
help="Address to listen for requests on")
686
parser.add_option("-p", "--port", type="int",
554
687
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")
561
688
parser.add_option("--check", action="store_true", default=False,
562
689
help="Run self-test")
563
parser.add_option("--debug", action="store_true", default=False,
690
parser.add_option("--debug", action="store_true",
691
help="Debug mode; run in foreground and log to"
693
parser.add_option("--priority", type="string", help="GnuTLS"
694
" priority string (see GnuTLS documentation)")
695
parser.add_option("--servicename", type="string", metavar="NAME",
696
help="Zeroconf service name")
697
parser.add_option("--configdir", type="string",
698
default="/etc/mandos", metavar="DIR",
699
help="Directory to search for configuration"
565
701
(options, args) = parser.parse_args()
567
703
if options.check:
569
705
doctest.testmod()
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
708
# Default values for config file for server-global settings
709
server_defaults = { "interface": "",
714
"SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
715
"servicename": "Mandos",
718
# Parse config file for server-global settings
719
server_config = ConfigParser.SafeConfigParser(server_defaults)
721
server_config.read(os.path.join(options.configdir, "mandos.conf"))
722
# Convert the SafeConfigParser object to a dict
723
server_settings = server_config.defaults()
724
# Use getboolean on the boolean config option
725
server_settings["debug"] = server_config.getboolean\
729
# Override the settings from the config file with command line
731
for option in ("interface", "address", "port", "debug",
732
"priority", "servicename", "configdir"):
733
value = getattr(options, option)
734
if value is not None:
735
server_settings[option] = value
737
# Now we have our good server settings in "server_settings"
739
debug = server_settings["debug"]
742
syslogger.setLevel(logging.WARNING)
743
console.setLevel(logging.WARNING)
745
if server_settings["servicename"] != "Mandos":
746
syslogger.setFormatter(logging.Formatter\
747
('Mandos (%s): %%(levelname)s:'
749
% server_settings["servicename"]))
751
# Parse config file with clients
752
client_defaults = { "timeout": "1h",
754
"checker": "fping -q -- %(host)s",
757
client_config = ConfigParser.SafeConfigParser(client_defaults)
758
client_config.read(os.path.join(server_settings["configdir"],
762
service = AvahiService(name = server_settings["servicename"],
763
type = "_mandos._tcp", );
764
if server_settings["interface"]:
765
service.interface = if_nametoindex\
766
(server_settings["interface"])
771
# From the Avahi example code
589
772
DBusGMainLoop(set_as_default=True )
590
773
main_loop = gobject.MainLoop()
591
774
bus = dbus.SystemBus()
592
server = dbus.Interface(
593
bus.get_object( avahi.DBUS_NAME, avahi.DBUS_PATH_SERVER ),
594
avahi.DBUS_INTERFACE_SERVER )
775
server = dbus.Interface(bus.get_object(avahi.DBUS_NAME,
776
avahi.DBUS_PATH_SERVER),
777
avahi.DBUS_INTERFACE_SERVER)
595
778
# End of Avahi example code
597
debug = options.debug
600
781
def remove_from_clients(client):
601
782
clients.remove(client)
603
logger.debug(u"No clients left, exiting")
784
logger.critical(u"No clients left, exiting")
606
clients.update(Set(Client(name=section, options=options,
787
clients.update(Set(Client(name = section,
607
788
stop_hook = remove_from_clients,
608
**(dict(client_config\
790
= dict(client_config.items(section)))
610
791
for section in client_config.sections()))
793
logger.critical(u"No clients defined")
797
# Redirect stdin so all checkers get /dev/null
798
null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
799
os.dup2(null, sys.stdin.fileno())
804
logger.removeHandler(console)
805
# Close all input and output, do double fork, etc.
808
pidfilename = "/var/run/mandos/mandos.pid"
811
pidfile = open(pidfilename, "w")
812
pidfile.write(str(pid) + "\n")
816
logger.error(u"Could not write %s file with PID %d",
817
pidfilename, os.getpid())
820
"Cleanup function; run on exit"
822
# From the Avahi example code
823
if not group is None:
826
# End of Avahi example code
829
client = clients.pop()
830
client.stop_hook = None
833
atexit.register(cleanup)
836
signal.signal(signal.SIGINT, signal.SIG_IGN)
837
signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
838
signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
611
840
for client in clients:
614
tcp_server = IPv6_TCPServer((None, options.port),
843
tcp_server = IPv6_TCPServer((server_settings["address"],
844
server_settings["port"]),
846
settings=server_settings,
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:],
848
# Find out what port we got
849
service.port = tcp_server.socket.getsockname()[1]
850
logger.info(u"Now listening on address %r, port %d, flowinfo %d,"
851
u" scope_id %d" % tcp_server.socket.getsockname())
853
#service.interface = tcp_server.socket.getsockname()[3]
856
# From the Avahi example code
857
server.connect_to_signal("StateChanged", server_state_changed)
859
server_state_changed(server.GetState())
860
except dbus.exceptions.DBusException, error:
861
logger.critical(u"DBusException: %s", error)
863
# End of Avahi example code
865
gobject.io_add_watch(tcp_server.fileno(), gobject.IO_IN,
866
lambda *args, **kwargs:
867
tcp_server.handle_request\
868
(*args[2:], **kwargs) or True)
870
logger.debug(u"Starting main loop")
871
main_loop_started = True
873
except AvahiError, error:
874
logger.critical(u"AvahiError: %s" + unicode(error))
636
876
except KeyboardInterrupt:
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
880
if __name__ == '__main__':