61
61
from dbus.mainloop.glib import DBusGMainLoop
66
65
logger = logging.Logger('mandos')
67
66
syslogger = logging.handlers.SysLogHandler\
68
(facility = logging.handlers.SysLogHandler.LOG_DAEMON,
67
(facility = logging.handlers.SysLogHandler.LOG_DAEMON)
70
68
syslogger.setFormatter(logging.Formatter\
71
('Mandos: %(levelname)s: %(message)s'))
69
('%(levelname)s: %(message)s'))
72
70
logger.addHandler(syslogger)
74
console = logging.StreamHandler()
75
console.setFormatter(logging.Formatter('%(name)s: %(levelname)s:'
77
logger.addHandler(console)
79
74
class AvahiError(Exception):
80
75
def __init__(self, value):
108
103
def __init__(self, interface = avahi.IF_UNSPEC, name = None,
109
104
type = None, port = None, TXT = None, domain = "",
110
host = "", max_renames = 32768):
105
host = "", max_renames = 12):
111
106
self.interface = interface
119
114
self.domain = domain
121
116
self.rename_count = 0
122
self.max_renames = max_renames
123
117
def rename(self):
124
118
"""Derived from the Avahi example code"""
125
119
if self.rename_count >= self.max_renames:
126
120
logger.critical(u"No suitable service name found after %i"
127
121
u" retries, exiting.", rename_count)
128
122
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))
123
name = server.GetAlternativeServiceName(name)
124
logger.error(u"Changing name to %r ...", name)
136
127
self.rename_count += 1
172
163
fingerprint: string (40 or 32 hexadecimal digits); used to
173
164
uniquely identify the client
174
165
secret: bytestring; sent verbatim (over TLS) to client
175
host: string; available for use by the checker command
166
fqdn: string (FQDN); available for use by the checker command
176
167
created: datetime.datetime(); object creation, not client host
177
168
last_checked_ok: datetime.datetime() or None if not yet checked OK
178
169
timeout: datetime.timedelta(); How long from last_checked_ok
240
231
raise TypeError(u"No secret or secfile for client %s"
242
self.host = config.get("host", "")
233
self.fqdn = config.get("fqdn", "")
243
234
self.created = datetime.datetime.now()
244
235
self.last_checked_ok = None
245
236
self.timeout = string_to_delta(config["timeout"])
268
259
The possibility that a client might be restarted is left open,
269
260
but not currently used."""
270
261
# If this client doesn't have a secret, it is already stopped.
271
if hasattr(self, "secret") and self.secret:
272
263
logger.info(u"Stopping client %s", self.name)
273
264
self.secret = None
322
313
if self.checker is None:
324
315
# In case check_command has exactly one % operator
325
command = self.check_command % self.host
316
command = self.check_command % self.fqdn
326
317
except TypeError:
327
318
# Escape attributes for the shell
328
319
escaped_attrs = dict((key, re.escape(str(val)))
338
329
logger.info(u"Starting checker %r for %s",
339
330
command, self.name)
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
331
self.checker = subprocess.Popen(command,
346
333
shell=True, cwd="/")
347
334
self.checker_callback_tag = gobject.child_watch_add\
348
335
(self.checker.pid,
349
336
self.checker_callback)
350
except OSError, error:
337
except subprocess.OSError, error:
351
338
logger.error(u"Failed to start subprocess: %s",
353
340
# Re-run this periodically if run by gobject.timeout_add
359
346
self.checker_callback_tag = None
360
347
if getattr(self, "checker", None) is None:
362
logger.debug(u"Stopping checker for %(name)s", vars(self))
349
logger.debug("Stopping checker for %(name)s", vars(self))
364
351
os.kill(self.checker.pid, signal.SIGTERM)
567
554
datetime.timedelta(1)
568
555
>>> string_to_delta(u'1w')
569
556
datetime.timedelta(7)
570
>>> string_to_delta('5m 30s')
571
datetime.timedelta(0, 330)
573
timevalue = datetime.timedelta(0)
574
for s in interval.split():
576
suffix=unicode(s[-1])
579
delta = datetime.timedelta(value)
581
delta = datetime.timedelta(0, value)
583
delta = datetime.timedelta(0, 0, 0, 0, value)
585
delta = datetime.timedelta(0, 0, 0, 0, 0, value)
587
delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
590
except (ValueError, IndexError):
559
suffix=unicode(interval[-1])
560
value=int(interval[:-1])
562
delta = datetime.timedelta(value)
564
delta = datetime.timedelta(0, value)
566
delta = datetime.timedelta(0, 0, 0, 0, value)
568
delta = datetime.timedelta(0, 0, 0, 0, 0, value)
570
delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
573
except (ValueError, IndexError):
596
578
def server_state_changed(state):
668
650
global main_loop_started
669
651
main_loop_started = False
671
parser = OptionParser(version = "%%prog %s" % version)
653
parser = OptionParser()
672
654
parser.add_option("-i", "--interface", type="string",
673
655
metavar="IF", help="Bind to interface IF")
674
656
parser.add_option("-a", "--address", type="string",
709
691
server_config = ConfigParser.SafeConfigParser(server_defaults)
710
692
del server_defaults
711
693
server_config.read(os.path.join(options.configdir, "mandos.conf"))
694
server_section = "server"
712
695
# Convert the SafeConfigParser object to a dict
713
server_settings = server_config.defaults()
696
server_settings = dict(server_config.items(server_section))
714
697
# Use getboolean on the boolean config option
715
698
server_settings["debug"] = server_config.getboolean\
699
(server_section, "debug")
717
700
del server_config
719
702
# Override the settings from the config file with command line
727
710
# Now we have our good server settings in "server_settings"
729
debug = server_settings["debug"]
732
syslogger.setLevel(logging.WARNING)
733
console.setLevel(logging.WARNING)
735
if server_settings["servicename"] != "Mandos":
736
syslogger.setFormatter(logging.Formatter\
737
('Mandos (%s): %%(levelname)s:'
739
% server_settings["servicename"]))
741
712
# Parse config file with clients
742
713
client_defaults = { "timeout": "1h",
743
714
"interval": "5m",
744
"checker": "fping -q -- %(host)s",
715
"checker": "fping -q -- %%(fqdn)s",
747
717
client_config = ConfigParser.SafeConfigParser(client_defaults)
748
718
client_config.read(os.path.join(server_settings["configdir"],
766
736
avahi.DBUS_INTERFACE_SERVER )
767
737
# End of Avahi example code
739
debug = server_settings["debug"]
742
console = logging.StreamHandler()
743
# console.setLevel(logging.DEBUG)
744
console.setFormatter(logging.Formatter\
745
('%(levelname)s: %(message)s'))
746
logger.addHandler(console)
770
750
def remove_from_clients(client):
771
751
clients.remove(client)
779
759
= dict(client_config.items(section)))
780
760
for section in client_config.sections()))
782
logger.critical(u"No clients defined")
786
# Redirect stdin so all checkers get /dev/null
787
null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
788
os.dup2(null, sys.stdin.fileno())
793
logger.removeHandler(console)
794
# Close all input and output, do double fork, etc.
797
pidfilename = "/var/run/mandos/mandos.pid"
800
pidfile = open(pidfilename, "w")
801
pidfile.write(str(pid) + "\n")
805
logger.error(u"Could not write %s file with PID %d",
806
pidfilename, os.getpid())
809
766
"Cleanup function; run on exit"
856
813
tcp_server.handle_request\
857
814
(*args[2:], **kwargs) or True)
859
logger.debug(u"Starting main loop")
816
logger.debug("Starting main loop")
860
817
main_loop_started = True
862
819
except AvahiError, error: