To get this branch, use:
bzr branch
http://bzr.recompile.se/loggerhead/mandos/trunk
« back to all changes in this revision
Viewing changes to server.py
- browse files at revision 24
- compare with another revision
- download diff
- download tarball
- view history from revision 24
- Committer: Teddy Hogeborn
- Date: 2008-07-22 01:59:47 UTC
- Revision ID: teddy@fukt.bsnet.se-20080722015947-y7o6keji020nvrhw
* Makefile (OPTIMIZE): New; optimize for size.
(CFLAGS): Use $(OPTIMIZE).
(CFLAGS): Use $(OPTIMIZE).
- files added:
- ca.pem
- files removed:
- .bzr-builddeb
- debian
- debian/po
- files renamed:
- clients.conf => mandos-clients.conf
- plugin-runner.c => plugbasedclient.c
- plugins.d/mandos-client.c => plugins.d/mandosclient.c
- plugins.d/password-prompt.c => plugins.d/passprompt.c
- mandos => server.py
- files modified:
- Makefile
added
removed
server.py
6
6
# This program is partly derived from an example program for an Avahi
7
7
# service publisher, downloaded from
8
8
# <http://avahi.org/wiki/PythonPublishExample>. This includes the
9
# methods "add" and "remove" in the "AvahiService" class, the
10
# "server_state_changed" and "entry_group_state_changed" functions,
11
# and some lines in "main".
9
# following functions: "add_service", "remove_service",
10
# "server_state_changed", "entry_group_state_changed", and some lines
11
# in "main".
12
12
#
13
# Everything else is
14
# Copyright © 2008,2009 Teddy Hogeborn
15
# Copyright © 2008,2009 Björn Påhlsson
13
# Everything else is Copyright © 2007-2008 Teddy Hogeborn and Björn
14
# Påhlsson.
16
15
#
17
16
# This program is free software: you can redistribute it and/or modify
18
17
# it under the terms of the GNU General Public License as published by
25
24
# GNU General Public License for more details.
26
25
#
27
26
# You should have received a copy of the GNU General Public License
28
# along with this program. If not, see
29
# <http://www.gnu.org/licenses/>.
27
# along with this program. If not, see <http://www.gnu.org/licenses/>.
30
28
#
31
# Contact the authors at <mandos@fukt.bsnet.se>.
29
# Contact the authors at <https://www.fukt.bsnet.se/~belorn/> and
30
# <https://www.fukt.bsnet.se/~teddy/>.
32
31
#
33
32
34
from __future__ import division, with_statement, absolute_import
33
from __future__ import division
35
34
36
35
import SocketServer
37
36
import socket
38
import optparse
37
import select
38
from optparse import OptionParser
39
39
import datetime
40
40
import errno
41
41
import gnutls.crypto
55
55
import stat
56
56
import logging
57
57
import logging.handlers
58
import pwd
59
from contextlib import closing
60
58
61
59
import dbus
62
import dbus.service
63
60
import gobject
64
61
import avahi
65
62
from dbus.mainloop.glib import DBusGMainLoop
66
63
import ctypes
67
import ctypes.util
68
69
version = "1.0.8"
64
65
# Brief description of the operation of this program:
66
#
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.
75
70
76
71
77
logger = logging.Logger('mandos')
72
syslogger = (logging.handlers.SysLogHandler
73
(facility = logging.handlers.SysLogHandler.LOG_DAEMON,
74
address = "/dev/log"))
75
syslogger.setFormatter(logging.Formatter
76
('Mandos [%(process)d]: %(levelname)s:'
77
' %(message)s'))
78
syslogger = logging.handlers.SysLogHandler\
79
(facility = logging.handlers.SysLogHandler.LOG_DAEMON)
80
syslogger.setFormatter(logging.Formatter\
81
('%(levelname)s: %(message)s'))
78
82
logger.addHandler(syslogger)
79
80
console = logging.StreamHandler()
81
console.setFormatter(logging.Formatter('%(name)s [%(process)d]:'
82
' %(levelname)s: %(message)s'))
83
logger.addHandler(console)
84
85
class AvahiError(Exception):
86
def __init__(self, value, *args, **kwargs):
87
self.value = value
88
super(AvahiError, self).__init__(value, *args, **kwargs)
89
def __unicode__(self):
90
return unicode(repr(self.value))
91
92
class AvahiServiceError(AvahiError):
93
pass
94
95
class AvahiGroupError(AvahiError):
96
pass
97
98
99
class AvahiService(object):
100
"""An Avahi (Zeroconf) service.
101
Attributes:
102
interface: integer; avahi.IF_UNSPEC or an interface index.
103
Used to optionally bind to the specified interface.
104
name: string; Example: 'Mandos'
105
type: string; Example: '_mandos._tcp'.
106
See <http://www.dns-sd.org/ServiceTypes.html>
107
port: integer; what port to announce
108
TXT: list of strings; TXT record for the service
109
domain: string; Domain to publish on, default to .local if empty.
110
host: string; Host to publish records for, default is localhost
111
max_renames: integer; maximum number of renames
112
rename_count: integer; counter so we only rename after collisions
113
a sensible number of times
114
"""
115
def __init__(self, interface = avahi.IF_UNSPEC, name = None,
116
servicetype = None, port = None, TXT = None,
117
domain = "", host = "", max_renames = 32768,
118
protocol = avahi.PROTO_UNSPEC):
119
self.interface = interface
120
self.name = name
121
self.type = servicetype
122
self.port = port
123
self.TXT = TXT if TXT is not None else []
124
self.domain = domain
125
self.host = host
126
self.rename_count = 0
127
self.max_renames = max_renames
128
self.protocol = protocol
129
def rename(self):
130
"""Derived from the Avahi example code"""
131
if self.rename_count >= self.max_renames:
132
logger.critical(u"No suitable Zeroconf service name found"
133
u" after %i retries, exiting.",
134
self.rename_count)
135
raise AvahiServiceError(u"Too many renames")
136
self.name = server.GetAlternativeServiceName(self.name)
137
logger.info(u"Changing Zeroconf service name to %r ...",
138
str(self.name))
139
syslogger.setFormatter(logging.Formatter
140
('Mandos (%s) [%%(process)d]:'
141
' %%(levelname)s: %%(message)s'
142
% self.name))
143
self.remove()
144
self.add()
145
self.rename_count += 1
146
def remove(self):
147
"""Derived from the Avahi example code"""
148
if group is not None:
149
group.Reset()
150
def add(self):
151
"""Derived from the Avahi example code"""
152
global group
153
if group is None:
154
group = dbus.Interface(bus.get_object
155
(avahi.DBUS_NAME,
156
server.EntryGroupNew()),
157
avahi.DBUS_INTERFACE_ENTRY_GROUP)
158
group.connect_to_signal('StateChanged',
159
entry_group_state_changed)
160
logger.debug(u"Adding Zeroconf service '%s' of type '%s' ...",
161
service.name, service.type)
162
group.AddService(
163
self.interface, # interface
164
self.protocol, # protocol
165
dbus.UInt32(0), # flags
166
self.name, self.type,
167
self.domain, self.host,
168
dbus.UInt16(self.port),
169
avahi.string_array_to_txt_array(self.TXT))
170
group.Commit()
171
83
del syslogger
84
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
87
# Avahi example code.
88
serviceInterface = avahi.IF_UNSPEC
172
89
# From the Avahi example code:
173
group = None # our entry group
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
174
99
# End of Avahi example code
175
100
176
101
177
def _datetime_to_dbus(dt, variant_level=0):
178
"""Convert a UTC datetime.datetime() to a D-Bus type."""
179
return dbus.String(dt.isoformat(), variant_level=variant_level)
180
181
182
102
class Client(object):
183
103
"""A representation of a client host served by this server.
184
104
Attributes:
185
name: string; from the config file, used in log messages and
186
D-Bus identifiers
105
name: string; from the config file, used in log messages
187
106
fingerprint: string (40 or 32 hexadecimal digits); used to
188
107
uniquely identify the client
189
secret: bytestring; sent verbatim (over TLS) to client
190
host: string; available for use by the checker command
191
created: datetime.datetime(); (UTC) object creation
192
last_enabled: datetime.datetime(); (UTC)
193
enabled: bool()
194
last_checked_ok: datetime.datetime(); (UTC) or None
195
timeout: datetime.timedelta(); How long from last_checked_ok
196
until this client is invalid
197
interval: datetime.timedelta(); How often to start a new checker
198
disable_hook: If set, called by disable() as disable_hook(self)
199
checker: subprocess.Popen(); a running checker process used
200
to see if the client lives.
201
'None' if no process is running.
108
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
114
interval: datetime.timedelta(); How often to start a new checker
115
stop_hook: If set, called by stop() as stop_hook(self)
116
checker: subprocess.Popen(); a running checker process used
117
to see if the client lives.
118
Is None if no process is running.
202
119
checker_initiator_tag: a gobject event source tag, or None
203
disable_initiator_tag: - '' -
120
stop_initiator_tag: - '' -
204
121
checker_callback_tag: - '' -
205
122
checker_command: string; External command which is run to check if
206
client lives. %() expansions are done at
123
client lives. %()s expansions are done at
207
124
runtime with vars(self) as dict, so that for
208
125
instance %(name)s can be used in the command.
209
current_checker_command: string; current running checker_command
126
Private attibutes:
127
_timeout: Real variable for 'timeout'
128
_interval: Real variable for 'interval'
129
_timeout_milliseconds: Used by gobject.timeout_add()
130
_interval_milliseconds: - '' -
210
131
"""
211
def timeout_milliseconds(self):
212
"Return the 'timeout' attribute in milliseconds"
213
return ((self.timeout.days * 24 * 60 * 60 * 1000)
214
+ (self.timeout.seconds * 1000)
215
+ (self.timeout.microseconds // 1000))
216
217
def interval_milliseconds(self):
218
"Return the 'interval' attribute in milliseconds"
219
return ((self.interval.days * 24 * 60 * 60 * 1000)
220
+ (self.interval.seconds * 1000)
221
+ (self.interval.microseconds // 1000))
222
223
def __init__(self, name = None, disable_hook=None, config=None):
224
"""Note: the 'checker' key in 'config' sets the
225
'checker_command' attribute and *not* the 'checker'
226
attribute."""
132
def _set_timeout(self, timeout):
133
"Setter function for 'timeout' attribute"
134
self._timeout = timeout
135
self._timeout_milliseconds = ((self.timeout.days
136
* 24 * 60 * 60 * 1000)
137
+ (self.timeout.seconds * 1000)
138
+ (self.timeout.microseconds
139
// 1000))
140
timeout = property(lambda self: self._timeout,
141
_set_timeout)
142
del _set_timeout
143
def _set_interval(self, interval):
144
"Setter function for 'interval' attribute"
145
self._interval = interval
146
self._interval_milliseconds = ((self.interval.days
147
* 24 * 60 * 60 * 1000)
148
+ (self.interval.seconds
149
* 1000)
150
+ (self.interval.microseconds
151
// 1000))
152
interval = property(lambda self: self._interval,
153
_set_interval)
154
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.."""
227
160
self.name = name
228
if config is None:
229
config = {}
230
logger.debug(u"Creating client %r", self.name)
231
# Uppercase and remove spaces from fingerprint for later
232
# comparison purposes with return value from the fingerprint()
233
# function
234
self.fingerprint = (config["fingerprint"].upper()
235
.replace(u" ", u""))
236
logger.debug(u" Fingerprint: %s", self.fingerprint)
237
if "secret" in config:
238
self.secret = config["secret"].decode(u"base64")
239
elif "secfile" in config:
240
with closing(open(os.path.expanduser
241
(os.path.expandvars
242
(config["secfile"])))) as secfile:
243
self.secret = secfile.read()
244
else:
245
raise TypeError(u"No secret or secfile for client %s"
246
% self.name)
247
self.host = config.get("host", "")
248
self.created = datetime.datetime.utcnow()
249
self.enabled = False
250
self.last_enabled = None
251
self.last_checked_ok = None
252
self.timeout = string_to_delta(config["timeout"])
253
self.interval = string_to_delta(config["interval"])
254
self.disable_hook = disable_hook
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"")
165
if secret:
166
self.secret = secret.decode(u"base64")
167
elif secfile:
168
sf = open(secfile)
169
self.secret = sf.read()
170
sf.close()
171
else:
172
raise RuntimeError(u"No secret or secfile for client %s"
173
% self.name)
174
self.fqdn = fqdn # string
175
self.created = datetime.datetime.now()
176
self.last_seen = None
177
if timeout is None:
178
self.timeout = options.timeout
179
else:
180
self.timeout = string_to_delta(timeout)
181
if interval == -1:
182
self.interval = options.interval
183
else:
184
self.interval = string_to_delta(interval)
185
self.stop_hook = stop_hook
255
186
self.checker = None
256
187
self.checker_initiator_tag = None
257
self.disable_initiator_tag = None
188
self.stop_initiator_tag = None
258
189
self.checker_callback_tag = None
259
self.checker_command = config["checker"]
260
self.current_checker_command = None
261
self.last_connect = None
262
263
def enable(self):
190
self.check_command = checker
191
def start(self):
264
192
"""Start this client's checker and timeout hooks"""
265
self.last_enabled = datetime.datetime.utcnow()
266
193
# Schedule a new checker to be started an 'interval' from now,
267
194
# and every interval from then on.
268
self.checker_initiator_tag = (gobject.timeout_add
269
(self.interval_milliseconds(),
270
self.start_checker))
195
self.checker_initiator_tag = gobject.timeout_add\
196
(self._interval_milliseconds,
197
self.start_checker)
271
198
# Also start a new checker *right now*.
272
199
self.start_checker()
273
# Schedule a disable() when 'timeout' has passed
274
self.disable_initiator_tag = (gobject.timeout_add
275
(self.timeout_milliseconds(),
276
self.disable))
277
self.enabled = True
278
279
def disable(self):
280
"""Disable this client."""
281
if not getattr(self, "enabled", False):
200
# Schedule a stop() when 'timeout' has passed
201
self.stop_initiator_tag = gobject.timeout_add\
202
(self._timeout_milliseconds,
203
self.stop)
204
def stop(self):
205
"""Stop this client.
206
The possibility that this client might be restarted is left
207
open, but not currently used."""
208
# If this client doesn't have a secret, it is already stopped.
209
if self.secret:
210
logger.debug(u"Stopping client %s", self.name)
211
self.secret = None
212
else:
282
213
return False
283
logger.info(u"Disabling client %s", self.name)
284
if getattr(self, "disable_initiator_tag", False):
285
gobject.source_remove(self.disable_initiator_tag)
286
self.disable_initiator_tag = None
287
if getattr(self, "checker_initiator_tag", False):
214
if hasattr(self, "stop_initiator_tag") \
215
and self.stop_initiator_tag:
216
gobject.source_remove(self.stop_initiator_tag)
217
self.stop_initiator_tag = None
218
if hasattr(self, "checker_initiator_tag") \
219
and self.checker_initiator_tag:
288
220
gobject.source_remove(self.checker_initiator_tag)
289
221
self.checker_initiator_tag = None
290
222
self.stop_checker()
291
if self.disable_hook:
292
self.disable_hook(self)
293
self.enabled = False
223
if self.stop_hook:
224
self.stop_hook(self)
294
225
# Do not run this again if called by a gobject.timeout_add
295
226
return False
296
297
227
def __del__(self):
298
self.disable_hook = None
299
self.disable()
300
301
def checker_callback(self, pid, condition, command):
228
self.stop_hook = None
229
self.stop()
230
def checker_callback(self, pid, condition):
302
231
"""The checker has completed, so take appropriate actions."""
232
now = datetime.datetime.now()
303
233
self.checker_callback_tag = None
304
234
self.checker = None
305
if os.WIFEXITED(condition):
306
exitstatus = os.WEXITSTATUS(condition)
307
if exitstatus == 0:
308
logger.info(u"Checker for %(name)s succeeded",
309
vars(self))
310
self.checked_ok()
311
else:
312
logger.info(u"Checker for %(name)s failed",
313
vars(self))
314
else:
235
if os.WIFEXITED(condition) \
236
and (os.WEXITSTATUS(condition) == 0):
237
logger.debug(u"Checker for %(name)s succeeded",
238
vars(self))
239
self.last_seen = now
240
gobject.source_remove(self.stop_initiator_tag)
241
self.stop_initiator_tag = gobject.timeout_add\
242
(self._timeout_milliseconds,
243
self.stop)
244
elif not os.WIFEXITED(condition):
315
245
logger.warning(u"Checker for %(name)s crashed?",
316
246
vars(self))
317
318
def checked_ok(self):
319
"""Bump up the timeout for this client.
320
This should only be called when the client has been seen,
321
alive and well.
322
"""
323
self.last_checked_ok = datetime.datetime.utcnow()
324
gobject.source_remove(self.disable_initiator_tag)
325
self.disable_initiator_tag = (gobject.timeout_add
326
(self.timeout_milliseconds(),
327
self.disable))
328
247
else:
248
logger.debug(u"Checker for %(name)s failed",
249
vars(self))
329
250
def start_checker(self):
330
251
"""Start a new checker subprocess if one is not running.
331
252
If a checker already exists, leave it running and do
338
259
# checkers alone, the checker would have to take more time
339
260
# than 'timeout' for the client to be declared invalid, which
340
261
# is as it should be.
341
342
# If a checker exists, make sure it is not a zombie
343
if self.checker is not None:
344
pid, status = os.waitpid(self.checker.pid, os.WNOHANG)
345
if pid:
346
logger.warning("Checker was a zombie")
347
gobject.source_remove(self.checker_callback_tag)
348
self.checker_callback(pid, status,
349
self.current_checker_command)
350
# Start a new checker if needed
351
262
if self.checker is None:
352
263
try:
353
# In case checker_command has exactly one % operator
354
command = self.checker_command % self.host
264
command = self.check_command % self.fqdn
355
265
except TypeError:
356
# Escape attributes for the shell
357
266
escaped_attrs = dict((key, re.escape(str(val)))
358
267
for key, val in
359
268
vars(self).iteritems())
360
269
try:
361
command = self.checker_command % escaped_attrs
270
command = self.check_command % escaped_attrs
362
271
except TypeError, error:
363
logger.error(u'Could not format string "%s":'
364
u' %s', self.checker_command, error)
272
logger.critical(u'Could not format string "%s":'
273
u' %s', self.check_command, error)
365
274
return True # Try again later
366
self.current_checker_command = command
367
275
try:
368
logger.info(u"Starting checker %r for %s",
369
command, self.name)
370
# We don't need to redirect stdout and stderr, since
371
# in normal mode, that is already done by daemon(),
372
# and in debug mode we don't want to. (Stdin is
373
# always replaced by /dev/null.)
374
self.checker = subprocess.Popen(command,
375
close_fds=True,
376
shell=True, cwd="/")
377
self.checker_callback_tag = (gobject.child_watch_add
378
(self.checker.pid,
379
self.checker_callback,
380
data=command))
381
# The checker may have completed before the gobject
382
# watch was added. Check for this.
383
pid, status = os.waitpid(self.checker.pid, os.WNOHANG)
384
if pid:
385
gobject.source_remove(self.checker_callback_tag)
386
self.checker_callback(pid, status, command)
387
except OSError, error:
276
logger.debug(u"Starting checker %r for %s",
277
command, self.name)
278
self.checker = subprocess.\
279
Popen(command,
280
close_fds=True, shell=True,
281
cwd="/")
282
self.checker_callback_tag = gobject.child_watch_add\
283
(self.checker.pid,
284
self.checker_callback)
285
except subprocess.OSError, error:
388
286
logger.error(u"Failed to start subprocess: %s",
389
287
error)
390
288
# Re-run this periodically if run by gobject.timeout_add
391
289
return True
392
393
290
def stop_checker(self):
394
291
"""Force the checker process, if any, to stop."""
395
292
if self.checker_callback_tag:
396
293
gobject.source_remove(self.checker_callback_tag)
397
294
self.checker_callback_tag = None
398
if getattr(self, "checker", None) is None:
295
if not hasattr(self, "checker") or self.checker is None:
399
296
return
400
logger.debug(u"Stopping checker for %(name)s", vars(self))
297
logger.debug("Stopping checker for %(name)s", vars(self))
401
298
try:
402
299
os.kill(self.checker.pid, signal.SIGTERM)
403
300
#os.sleep(0.5)
404
301
#if self.checker.poll() is None:
405
302
# os.kill(self.checker.pid, signal.SIGKILL)
406
303
except OSError, error:
407
if error.errno != errno.ESRCH: # No such process
304
if error.errno != errno.ESRCH:
408
305
raise
409
306
self.checker = None
410
411
def still_valid(self):
307
def still_valid(self, now=None):
412
308
"""Has the timeout not yet passed for this client?"""
413
if not getattr(self, "enabled", False):
414
return False
415
now = datetime.datetime.utcnow()
416
if self.last_checked_ok is None:
309
if now is None:
310
now = datetime.datetime.now()
311
if self.last_seen is None:
417
312
return now < (self.created + self.timeout)
418
313
else:
419
return now < (self.last_checked_ok + self.timeout)
420
421
422
class ClientDBus(Client, dbus.service.Object):
423
"""A Client class using D-Bus
424
Attributes:
425
dbus_object_path: dbus.ObjectPath ; only set if self.use_dbus
426
"""
427
# dbus.service.Object doesn't use super(), so we can't either.
428
429
def __init__(self, *args, **kwargs):
430
Client.__init__(self, *args, **kwargs)
431
# Only now, when this client is initialized, can it show up on
432
# the D-Bus
433
self.dbus_object_path = (dbus.ObjectPath
434
("/clients/"
435
+ self.name.replace(".", "_")))
436
dbus.service.Object.__init__(self, bus,
437
self.dbus_object_path)
438
def enable(self):
439
oldstate = getattr(self, "enabled", False)
440
r = Client.enable(self)
441
if oldstate != self.enabled:
442
# Emit D-Bus signals
443
self.PropertyChanged(dbus.String(u"enabled"),
444
dbus.Boolean(True, variant_level=1))
445
self.PropertyChanged(dbus.String(u"last_enabled"),
446
(_datetime_to_dbus(self.last_enabled,
447
variant_level=1)))
448
return r
449
450
def disable(self, signal = True):
451
oldstate = getattr(self, "enabled", False)
452
r = Client.disable(self)
453
if signal and oldstate != self.enabled:
454
# Emit D-Bus signal
455
self.PropertyChanged(dbus.String(u"enabled"),
456
dbus.Boolean(False, variant_level=1))
457
return r
458
459
def __del__(self, *args, **kwargs):
460
try:
461
self.remove_from_connection()
462
except LookupError:
463
pass
464
if hasattr(dbus.service.Object, "__del__"):
465
dbus.service.Object.__del__(self, *args, **kwargs)
466
Client.__del__(self, *args, **kwargs)
467
468
def checker_callback(self, pid, condition, command,
469
*args, **kwargs):
470
self.checker_callback_tag = None
471
self.checker = None
472
# Emit D-Bus signal
473
self.PropertyChanged(dbus.String(u"checker_running"),
474
dbus.Boolean(False, variant_level=1))
475
if os.WIFEXITED(condition):
476
exitstatus = os.WEXITSTATUS(condition)
477
# Emit D-Bus signal
478
self.CheckerCompleted(dbus.Int16(exitstatus),
479
dbus.Int64(condition),
480
dbus.String(command))
481
else:
482
# Emit D-Bus signal
483
self.CheckerCompleted(dbus.Int16(-1),
484
dbus.Int64(condition),
485
dbus.String(command))
486
487
return Client.checker_callback(self, pid, condition, command,
488
*args, **kwargs)
489
490
def checked_ok(self, *args, **kwargs):
491
r = Client.checked_ok(self, *args, **kwargs)
492
# Emit D-Bus signal
493
self.PropertyChanged(
494
dbus.String(u"last_checked_ok"),
495
(_datetime_to_dbus(self.last_checked_ok,
496
variant_level=1)))
497
return r
498
499
def start_checker(self, *args, **kwargs):
500
old_checker = self.checker
501
if self.checker is not None:
502
old_checker_pid = self.checker.pid
503
else:
504
old_checker_pid = None
505
r = Client.start_checker(self, *args, **kwargs)
506
# Only if new checker process was started
507
if (self.checker is not None
508
and old_checker_pid != self.checker.pid):
509
# Emit D-Bus signal
510
self.CheckerStarted(self.current_checker_command)
511
self.PropertyChanged(
512
dbus.String("checker_running"),
513
dbus.Boolean(True, variant_level=1))
514
return r
515
516
def stop_checker(self, *args, **kwargs):
517
old_checker = getattr(self, "checker", None)
518
r = Client.stop_checker(self, *args, **kwargs)
519
if (old_checker is not None
520
and getattr(self, "checker", None) is None):
521
self.PropertyChanged(dbus.String(u"checker_running"),
522
dbus.Boolean(False, variant_level=1))
523
return r
524
525
## D-Bus methods & signals
526
_interface = u"se.bsnet.fukt.Mandos.Client"
527
528
# CheckedOK - method
529
CheckedOK = dbus.service.method(_interface)(checked_ok)
530
CheckedOK.__name__ = "CheckedOK"
531
532
# CheckerCompleted - signal
533
@dbus.service.signal(_interface, signature="nxs")
534
def CheckerCompleted(self, exitcode, waitstatus, command):
535
"D-Bus signal"
536
pass
537
538
# CheckerStarted - signal
539
@dbus.service.signal(_interface, signature="s")
540
def CheckerStarted(self, command):
541
"D-Bus signal"
542
pass
543
544
# GetAllProperties - method
545
@dbus.service.method(_interface, out_signature="a{sv}")
546
def GetAllProperties(self):
547
"D-Bus method"
548
return dbus.Dictionary({
549
dbus.String("name"):
550
dbus.String(self.name, variant_level=1),
551
dbus.String("fingerprint"):
552
dbus.String(self.fingerprint, variant_level=1),
553
dbus.String("host"):
554
dbus.String(self.host, variant_level=1),
555
dbus.String("created"):
556
_datetime_to_dbus(self.created, variant_level=1),
557
dbus.String("last_enabled"):
558
(_datetime_to_dbus(self.last_enabled,
559
variant_level=1)
560
if self.last_enabled is not None
561
else dbus.Boolean(False, variant_level=1)),
562
dbus.String("enabled"):
563
dbus.Boolean(self.enabled, variant_level=1),
564
dbus.String("last_checked_ok"):
565
(_datetime_to_dbus(self.last_checked_ok,
566
variant_level=1)
567
if self.last_checked_ok is not None
568
else dbus.Boolean (False, variant_level=1)),
569
dbus.String("timeout"):
570
dbus.UInt64(self.timeout_milliseconds(),
571
variant_level=1),
572
dbus.String("interval"):
573
dbus.UInt64(self.interval_milliseconds(),
574
variant_level=1),
575
dbus.String("checker"):
576
dbus.String(self.checker_command,
577
variant_level=1),
578
dbus.String("checker_running"):
579
dbus.Boolean(self.checker is not None,
580
variant_level=1),
581
dbus.String("object_path"):
582
dbus.ObjectPath(self.dbus_object_path,
583
variant_level=1)
584
}, signature="sv")
585
586
# IsStillValid - method
587
@dbus.service.method(_interface, out_signature="b")
588
def IsStillValid(self):
589
return self.still_valid()
590
591
# PropertyChanged - signal
592
@dbus.service.signal(_interface, signature="sv")
593
def PropertyChanged(self, property, value):
594
"D-Bus signal"
595
pass
596
597
# ReceivedSecret - signal
598
@dbus.service.signal(_interface)
599
def ReceivedSecret(self):
600
"D-Bus signal"
601
pass
602
603
# Rejected - signal
604
@dbus.service.signal(_interface)
605
def Rejected(self):
606
"D-Bus signal"
607
pass
608
609
# SetChecker - method
610
@dbus.service.method(_interface, in_signature="s")
611
def SetChecker(self, checker):
612
"D-Bus setter method"
613
self.checker_command = checker
614
# Emit D-Bus signal
615
self.PropertyChanged(dbus.String(u"checker"),
616
dbus.String(self.checker_command,
617
variant_level=1))
618
619
# SetHost - method
620
@dbus.service.method(_interface, in_signature="s")
621
def SetHost(self, host):
622
"D-Bus setter method"
623
self.host = host
624
# Emit D-Bus signal
625
self.PropertyChanged(dbus.String(u"host"),
626
dbus.String(self.host, variant_level=1))
627
628
# SetInterval - method
629
@dbus.service.method(_interface, in_signature="t")
630
def SetInterval(self, milliseconds):
631
self.interval = datetime.timedelta(0, 0, 0, milliseconds)
632
# Emit D-Bus signal
633
self.PropertyChanged(dbus.String(u"interval"),
634
(dbus.UInt64(self.interval_milliseconds(),
635
variant_level=1)))
636
637
# SetSecret - method
638
@dbus.service.method(_interface, in_signature="ay",
639
byte_arrays=True)
640
def SetSecret(self, secret):
641
"D-Bus setter method"
642
self.secret = str(secret)
643
644
# SetTimeout - method
645
@dbus.service.method(_interface, in_signature="t")
646
def SetTimeout(self, milliseconds):
647
self.timeout = datetime.timedelta(0, 0, 0, milliseconds)
648
# Emit D-Bus signal
649
self.PropertyChanged(dbus.String(u"timeout"),
650
(dbus.UInt64(self.timeout_milliseconds(),
651
variant_level=1)))
652
653
# Enable - method
654
Enable = dbus.service.method(_interface)(enable)
655
Enable.__name__ = "Enable"
656
657
# StartChecker - method
658
@dbus.service.method(_interface)
659
def StartChecker(self):
660
"D-Bus method"
661
self.start_checker()
662
663
# Disable - method
664
@dbus.service.method(_interface)
665
def Disable(self):
666
"D-Bus method"
667
self.disable()
668
669
# StopChecker - method
670
StopChecker = dbus.service.method(_interface)(stop_checker)
671
StopChecker.__name__ = "StopChecker"
672
673
del _interface
674
675
676
class TCP_handler(SocketServer.BaseRequestHandler, object):
314
return now < (self.last_seen + self.timeout)
315
316
317
def peer_certificate(session):
318
"Return the peer's OpenPGP certificate as a bytestring"
319
# If not an OpenPGP certificate...
320
if gnutls.library.functions.gnutls_certificate_type_get\
321
(session._c_object) \
322
!= gnutls.library.constants.GNUTLS_CRT_OPENPGP:
323
# ...do the normal thing
324
return session.peer_certificate
325
list_size = ctypes.c_uint()
326
cert_list = gnutls.library.functions.gnutls_certificate_get_peers\
327
(session._c_object, ctypes.byref(list_size))
328
if list_size.value == 0:
329
return None
330
cert = cert_list[0]
331
return ctypes.string_at(cert.data, cert.size)
332
333
334
def fingerprint(openpgp):
335
"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\
339
(ctypes.byref(crt))
340
# New GnuTLS "datum" with the OpenPGP public key
341
datum = gnutls.library.types.gnutls_datum_t\
342
(ctypes.cast(ctypes.c_char_p(openpgp),
343
ctypes.POINTER(ctypes.c_ubyte)),
344
ctypes.c_uint(len(openpgp)))
345
# Import the OpenPGP public key into the certificate
346
ret = gnutls.library.functions.gnutls_openpgp_crt_import\
347
(crt,
348
ctypes.byref(datum),
349
gnutls.library.constants.GNUTLS_OPENPGP_FMT_RAW)
350
# New buffer for the fingerprint
351
buffer = ctypes.create_string_buffer(20)
352
buffer_length = ctypes.c_size_t()
353
# Get the fingerprint from the certificate into the buffer
354
gnutls.library.functions.gnutls_openpgp_crt_get_fingerprint\
355
(crt, ctypes.byref(buffer), ctypes.byref(buffer_length))
356
# Deinit the certificate
357
gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
358
# Convert the buffer to a Python bytestring
359
fpr = ctypes.string_at(buffer, buffer_length.value)
360
# Convert the bytestring to hexadecimal notation
361
hex_fpr = u''.join(u"%02X" % ord(char) for char in fpr)
362
return hex_fpr
363
364
365
class tcp_handler(SocketServer.BaseRequestHandler, object):
677
366
"""A TCP request handler class.
678
367
Instantiated by IPv6_TCPServer for each request to handle it.
679
368
Note: This will run in its own forked process."""
680
369
681
370
def handle(self):
682
logger.info(u"TCP connection from: %s",
683
unicode(self.client_address))
684
logger.debug(u"IPC Pipe FD: %d", self.server.pipe[1])
685
# Open IPC pipe to parent process
686
with closing(os.fdopen(self.server.pipe[1], "w", 1)) as ipc:
687
session = (gnutls.connection
688
.ClientSession(self.request,
689
gnutls.connection
690
.X509Credentials()))
691
692
line = self.request.makefile().readline()
693
logger.debug(u"Protocol version: %r", line)
694
try:
695
if int(line.strip().split()[0]) > 1:
696
raise RuntimeError
697
except (ValueError, IndexError, RuntimeError), error:
698
logger.error(u"Unknown protocol version: %s", error)
699
return
700
701
# Note: gnutls.connection.X509Credentials is really a
702
# generic GnuTLS certificate credentials object so long as
703
# no X.509 keys are added to it. Therefore, we can use it
704
# here despite using OpenPGP certificates.
705
706
#priority = ':'.join(("NONE", "+VERS-TLS1.1",
707
# "+AES-256-CBC", "+SHA1",
708
# "+COMP-NULL", "+CTYPE-OPENPGP",
709
# "+DHE-DSS"))
710
# Use a fallback default, since this MUST be set.
711
priority = self.server.settings.get("priority", "NORMAL")
712
(gnutls.library.functions
713
.gnutls_priority_set_direct(session._c_object,
714
priority, None))
715
716
try:
717
session.handshake()
718
except gnutls.errors.GNUTLSError, error:
719
logger.warning(u"Handshake failed: %s", error)
720
# Do not run session.bye() here: the session is not
721
# established. Just abandon the request.
722
return
723
logger.debug(u"Handshake succeeded")
724
try:
725
fpr = self.fingerprint(self.peer_certificate(session))
726
except (TypeError, gnutls.errors.GNUTLSError), error:
727
logger.warning(u"Bad certificate: %s", error)
728
session.bye()
729
return
730
logger.debug(u"Fingerprint: %s", fpr)
731
732
for c in self.server.clients:
733
if c.fingerprint == fpr:
734
client = c
735
break
371
logger.debug(u"TCP connection from: %s",
372
unicode(self.client_address))
373
session = gnutls.connection.ClientSession(self.request,
374
gnutls.connection.\
375
X509Credentials())
376
377
#priority = ':'.join(("NONE", "+VERS-TLS1.1", "+AES-256-CBC",
378
# "+SHA1", "+COMP-NULL", "+CTYPE-OPENPGP",
379
# "+DHE-DSS"))
380
priority = "SECURE256"
381
382
gnutls.library.functions.gnutls_priority_set_direct\
383
(session._c_object, priority, None);
384
385
try:
386
session.handshake()
387
except gnutls.errors.GNUTLSError, error:
388
logger.debug(u"Handshake failed: %s", error)
389
# Do not run session.bye() here: the session is not
390
# established. Just abandon the request.
391
return
392
try:
393
fpr = fingerprint(peer_certificate(session))
394
except (TypeError, gnutls.errors.GNUTLSError), error:
395
logger.debug(u"Bad certificate: %s", error)
396
session.bye()
397
return
398
logger.debug(u"Fingerprint: %s", fpr)
399
client = None
400
for c in self.server.clients:
401
if c.fingerprint == fpr:
402
client = c
403
break
404
# Have to check if client.still_valid(), since it is possible
405
# that the client timed out while establishing the GnuTLS
406
# session.
407
if (not client) or (not client.still_valid()):
408
if client:
409
logger.debug(u"Client %(name)s is invalid",
410
vars(client))
736
411
else:
737
ipc.write("NOTFOUND %s\n" % fpr)
738
session.bye()
739
return
740
# Have to check if client.still_valid(), since it is
741
# possible that the client timed out while establishing
742
# the GnuTLS session.
743
if not client.still_valid():
744
ipc.write("INVALID %s\n" % client.name)
745
session.bye()
746
return
747
ipc.write("SENDING %s\n" % client.name)
748
sent_size = 0
749
while sent_size < len(client.secret):
750
sent = session.send(client.secret[sent_size:])
751
logger.debug(u"Sent: %d, remaining: %d",
752
sent, len(client.secret)
753
- (sent_size + sent))
754
sent_size += sent
412
logger.debug(u"Client not found for fingerprint: %s",
413
fpr)
755
414
session.bye()
756
757
@staticmethod
758
def peer_certificate(session):
759
"Return the peer's OpenPGP certificate as a bytestring"
760
# If not an OpenPGP certificate...
761
if (gnutls.library.functions
762
.gnutls_certificate_type_get(session._c_object)
763
!= gnutls.library.constants.GNUTLS_CRT_OPENPGP):
764
# ...do the normal thing
765
return session.peer_certificate
766
list_size = ctypes.c_uint(1)
767
cert_list = (gnutls.library.functions
768
.gnutls_certificate_get_peers
769
(session._c_object, ctypes.byref(list_size)))
770
if not bool(cert_list) and list_size.value != 0:
771
raise gnutls.errors.GNUTLSError("error getting peer"
772
" certificate")
773
if list_size.value == 0:
774
return None
775
cert = cert_list[0]
776
return ctypes.string_at(cert.data, cert.size)
777
778
@staticmethod
779
def fingerprint(openpgp):
780
"Convert an OpenPGP bytestring to a hexdigit fingerprint"
781
# New GnuTLS "datum" with the OpenPGP public key
782
datum = (gnutls.library.types
783
.gnutls_datum_t(ctypes.cast(ctypes.c_char_p(openpgp),
784
ctypes.POINTER
785
(ctypes.c_ubyte)),
786
ctypes.c_uint(len(openpgp))))
787
# New empty GnuTLS certificate
788
crt = gnutls.library.types.gnutls_openpgp_crt_t()
789
(gnutls.library.functions
790
.gnutls_openpgp_crt_init(ctypes.byref(crt)))
791
# Import the OpenPGP public key into the certificate
792
(gnutls.library.functions
793
.gnutls_openpgp_crt_import(crt, ctypes.byref(datum),
794
gnutls.library.constants
795
.GNUTLS_OPENPGP_FMT_RAW))
796
# Verify the self signature in the key
797
crtverify = ctypes.c_uint()
798
(gnutls.library.functions
799
.gnutls_openpgp_crt_verify_self(crt, 0,
800
ctypes.byref(crtverify)))
801
if crtverify.value != 0:
802
gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
803
raise (gnutls.errors.CertificateSecurityError
804
("Verify failed"))
805
# New buffer for the fingerprint
806
buf = ctypes.create_string_buffer(20)
807
buf_len = ctypes.c_size_t()
808
# Get the fingerprint from the certificate into the buffer
809
(gnutls.library.functions
810
.gnutls_openpgp_crt_get_fingerprint(crt, ctypes.byref(buf),
811
ctypes.byref(buf_len)))
812
# Deinit the certificate
813
gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
814
# Convert the buffer to a Python bytestring
815
fpr = ctypes.string_at(buf, buf_len.value)
816
# Convert the bytestring to hexadecimal notation
817
hex_fpr = u''.join(u"%02X" % ord(char) for char in fpr)
818
return hex_fpr
819
820
821
class ForkingMixInWithPipe(SocketServer.ForkingMixIn, object):
822
"""Like SocketServer.ForkingMixIn, but also pass a pipe.
823
Assumes a gobject.MainLoop event loop.
824
"""
825
def process_request(self, request, client_address):
826
"""This overrides and wraps the original process_request().
827
This function creates a new pipe in self.pipe
828
"""
829
self.pipe = os.pipe()
830
super(ForkingMixInWithPipe,
831
self).process_request(request, client_address)
832
os.close(self.pipe[1]) # close write end
833
# Call "handle_ipc" for both data and EOF events
834
gobject.io_add_watch(self.pipe[0],
835
gobject.IO_IN | gobject.IO_HUP,
836
self.handle_ipc)
837
def handle_ipc(source, condition):
838
"""Dummy function; override as necessary"""
839
os.close(source)
840
return False
841
842
843
class IPv6_TCPServer(ForkingMixInWithPipe,
844
SocketServer.TCPServer, object):
845
"""IPv6-capable TCP server. Accepts 'None' as address and/or port
415
return
416
sent_size = 0
417
while sent_size < len(client.secret):
418
sent = session.send(client.secret[sent_size:])
419
logger.debug(u"Sent: %d, remaining: %d",
420
sent, len(client.secret)
421
- (sent_size + sent))
422
sent_size += sent
423
session.bye()
424
425
426
class IPv6_TCPServer(SocketServer.ForkingTCPServer, object):
427
"""IPv6 TCP server. Accepts 'None' as address and/or port.
846
428
Attributes:
847
settings: Server settings
429
options: Command line options
848
430
clients: Set() of Client objects
849
enabled: Boolean; whether this server is activated yet
850
431
"""
851
432
address_family = socket.AF_INET6
852
433
def __init__(self, *args, **kwargs):
853
if "settings" in kwargs:
854
self.settings = kwargs["settings"]
855
del kwargs["settings"]
434
if "options" in kwargs:
435
self.options = kwargs["options"]
436
del kwargs["options"]
856
437
if "clients" in kwargs:
857
438
self.clients = kwargs["clients"]
858
439
del kwargs["clients"]
859
if "use_ipv6" in kwargs:
860
if not kwargs["use_ipv6"]:
861
self.address_family = socket.AF_INET
862
del kwargs["use_ipv6"]
863
self.enabled = False
864
super(IPv6_TCPServer, self).__init__(*args, **kwargs)
440
return super(type(self), self).__init__(*args, **kwargs)
865
441
def server_bind(self):
866
442
"""This overrides the normal server_bind() function
867
443
to bind to an interface if one was specified, and also NOT to
868
444
bind to an address or port if they were not specified."""
869
if self.settings["interface"]:
870
# 25 is from /usr/include/asm-i486/socket.h
871
SO_BINDTODEVICE = getattr(socket, "SO_BINDTODEVICE", 25)
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
872
449
try:
873
450
self.socket.setsockopt(socket.SOL_SOCKET,
874
SO_BINDTODEVICE,
875
self.settings["interface"])
451
socket.SO_BINDTODEVICE,
452
self.options.interface)
876
453
except socket.error, error:
877
454
if error[0] == errno.EPERM:
878
logger.error(u"No permission to"
879
u" bind to interface %s",
880
self.settings["interface"])
455
logger.warning(u"No permission to"
456
u" bind to interface %s",
457
self.options.interface)
881
458
else:
882
raise
459
raise error
883
460
# Only bind(2) the socket if we really need to.
884
461
if self.server_address[0] or self.server_address[1]:
885
462
if not self.server_address[0]:
886
if self.address_family == socket.AF_INET6:
887
any_address = "::" # in6addr_any
888
else:
889
any_address = socket.INADDR_ANY
890
self.server_address = (any_address,
463
in6addr_any = "::"
464
self.server_address = (in6addr_any,
891
465
self.server_address[1])
892
elif not self.server_address[1]:
466
elif self.server_address[1] is None:
893
467
self.server_address = (self.server_address[0],
894
468
0)
895
# if self.settings["interface"]:
896
# self.server_address = (self.server_address[0],
897
# 0, # port
898
# 0, # flowinfo
899
# if_nametoindex
900
# (self.settings
901
# ["interface"]))
902
return super(IPv6_TCPServer, self).server_bind()
903
def server_activate(self):
904
if self.enabled:
905
return super(IPv6_TCPServer, self).server_activate()
906
def enable(self):
907
self.enabled = True
908
def handle_ipc(self, source, condition, file_objects={}):
909
condition_names = {
910
gobject.IO_IN: "IN", # There is data to read.
911
gobject.IO_OUT: "OUT", # Data can be written (without
912
# blocking).
913
gobject.IO_PRI: "PRI", # There is urgent data to read.
914
gobject.IO_ERR: "ERR", # Error condition.
915
gobject.IO_HUP: "HUP" # Hung up (the connection has been
916
# broken, usually for pipes and
917
# sockets).
918
}
919
conditions_string = ' | '.join(name
920
for cond, name in
921
condition_names.iteritems()
922
if cond & condition)
923
logger.debug("Handling IPC: FD = %d, condition = %s", source,
924
conditions_string)
925
926
# Turn the pipe file descriptor into a Python file object
927
if source not in file_objects:
928
file_objects[source] = os.fdopen(source, "r", 1)
929
930
# Read a line from the file object
931
cmdline = file_objects[source].readline()
932
if not cmdline: # Empty line means end of file
933
# close the IPC pipe
934
file_objects[source].close()
935
del file_objects[source]
936
937
# Stop calling this function
938
return False
939
940
logger.debug("IPC command: %r", cmdline)
941
942
# Parse and act on command
943
cmd, args = cmdline.rstrip("\r\n").split(None, 1)
944
945
if cmd == "NOTFOUND":
946
logger.warning(u"Client not found for fingerprint: %s",
947
args)
948
if self.settings["use_dbus"]:
949
# Emit D-Bus signal
950
mandos_dbus_service.ClientNotFound(args)
951
elif cmd == "INVALID":
952
for client in self.clients:
953
if client.name == args:
954
logger.warning(u"Client %s is invalid", args)
955
if self.settings["use_dbus"]:
956
# Emit D-Bus signal
957
client.Rejected()
958
break
959
else:
960
logger.error(u"Unknown client %s is invalid", args)
961
elif cmd == "SENDING":
962
for client in self.clients:
963
if client.name == args:
964
logger.info(u"Sending secret to %s", client.name)
965
client.checked_ok()
966
if self.settings["use_dbus"]:
967
# Emit D-Bus signal
968
client.ReceivedSecret()
969
break
970
else:
971
logger.error(u"Sending secret to unknown client %s",
972
args)
973
else:
974
logger.error("Unknown IPC command: %r", cmdline)
975
976
# Keep calling this function
977
return True
469
return super(type(self), self).server_bind()
978
470
979
471
980
472
def string_to_delta(interval):
981
473
"""Parse a string and return a datetime.timedelta
982
474
983
475
>>> string_to_delta('7d')
984
476
datetime.timedelta(7)
985
477
>>> string_to_delta('60s')
990
482
datetime.timedelta(1)
991
483
>>> string_to_delta(u'1w')
992
484
datetime.timedelta(7)
993
>>> string_to_delta('5m 30s')
994
datetime.timedelta(0, 330)
995
485
"""
996
timevalue = datetime.timedelta(0)
997
for s in interval.split():
998
try:
999
suffix = unicode(s[-1])
1000
value = int(s[:-1])
1001
if suffix == u"d":
1002
delta = datetime.timedelta(value)
1003
elif suffix == u"s":
1004
delta = datetime.timedelta(0, value)
1005
elif suffix == u"m":
1006
delta = datetime.timedelta(0, 0, 0, 0, value)
1007
elif suffix == u"h":
1008
delta = datetime.timedelta(0, 0, 0, 0, 0, value)
1009
elif suffix == u"w":
1010
delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
1011
else:
1012
raise ValueError
1013
except (ValueError, IndexError):
486
try:
487
suffix=unicode(interval[-1])
488
value=int(interval[:-1])
489
if suffix == u"d":
490
delta = datetime.timedelta(value)
491
elif suffix == u"s":
492
delta = datetime.timedelta(0, value)
493
elif suffix == u"m":
494
delta = datetime.timedelta(0, 0, 0, 0, value)
495
elif suffix == u"h":
496
delta = datetime.timedelta(0, 0, 0, 0, 0, value)
497
elif suffix == u"w":
498
delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
499
else:
1014
500
raise ValueError
1015
timevalue += delta
1016
return timevalue
501
except (ValueError, IndexError):
502
raise ValueError
503
return delta
504
505
506
def add_service():
507
"""Derived from the Avahi example code"""
508
global group, serviceName, serviceType, servicePort, serviceTXT, \
509
domain, host
510
if group is None:
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)
519
520
group.AddService(
521
serviceInterface, # interface
522
avahi.PROTO_INET6, # protocol
523
dbus.UInt32(0), # flags
524
serviceName, serviceType,
525
domain, host,
526
dbus.UInt16(servicePort),
527
avahi.string_array_to_txt_array(serviceTXT))
528
group.Commit()
529
530
531
def remove_service():
532
"""From the Avahi example code"""
533
global group
534
535
if not group is None:
536
group.Reset()
1017
537
1018
538
1019
539
def server_state_changed(state):
1020
540
"""Derived from the Avahi example code"""
1021
541
if state == avahi.SERVER_COLLISION:
1022
logger.error(u"Zeroconf server name collision")
1023
service.remove()
542
logger.warning(u"Server name collision")
543
remove_service()
1024
544
elif state == avahi.SERVER_RUNNING:
1025
service.add()
545
add_service()
1026
546
1027
547
1028
548
def entry_group_state_changed(state, error):
1029
549
"""Derived from the Avahi example code"""
1030
logger.debug(u"Avahi state change: %i", state)
550
global serviceName, server, rename_count
551
552
logger.debug(u"state change: %i", state)
1031
553
1032
554
if state == avahi.ENTRY_GROUP_ESTABLISHED:
1033
logger.debug(u"Zeroconf service established.")
555
logger.debug(u"Service established.")
1034
556
elif state == avahi.ENTRY_GROUP_COLLISION:
1035
logger.warning(u"Zeroconf service name collision.")
1036
service.rename()
557
558
rename_count = rename_count - 1
559
if rename_count > 0:
560
name = server.GetAlternativeServiceName(name)
561
logger.warning(u"Service name collision, "
562
u"changing name to '%s' ...", name)
563
remove_service()
564
add_service()
565
566
else:
567
logger.error(u"No suitable service name found after %i"
568
u" retries, exiting.", n_rename)
569
killme(1)
1037
570
elif state == avahi.ENTRY_GROUP_FAILURE:
1038
logger.critical(u"Avahi: Error in group state changed %s",
1039
unicode(error))
1040
raise AvahiGroupError(u"State changed: %s" % unicode(error))
571
logger.error(u"Error in group state changed %s",
572
unicode(error))
573
killme(1)
574
1041
575
1042
576
def if_nametoindex(interface):
1043
"""Call the C function if_nametoindex(), or equivalent"""
1044
global if_nametoindex
577
"""Call the C function if_nametoindex()"""
1045
578
try:
1046
if_nametoindex = (ctypes.cdll.LoadLibrary
1047
(ctypes.util.find_library("c"))
1048
.if_nametoindex)
579
libc = ctypes.cdll.LoadLibrary("libc.so.6")
580
return libc.if_nametoindex(interface)
1049
581
except (OSError, AttributeError):
1050
582
if "struct" not in sys.modules:
1051
583
import struct
1052
584
if "fcntl" not in sys.modules:
1053
585
import fcntl
1054
def if_nametoindex(interface):
1055
"Get an interface index the hard way, i.e. using fcntl()"
1056
SIOCGIFINDEX = 0x8933 # From /usr/include/linux/sockios.h
1057
with closing(socket.socket()) as s:
1058
ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
1059
struct.pack("16s16x", interface))
1060
interface_index = struct.unpack("I", ifreq[16:20])[0]
1061
return interface_index
1062
return if_nametoindex(interface)
1063
1064
1065
def daemon(nochdir = False, noclose = False):
586
SIOCGIFINDEX = 0x8933 # From /usr/include/linux/sockios.h
587
s = socket.socket()
588
ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
589
struct.pack("16s16x", interface))
590
s.close()
591
interface_index = struct.unpack("I", ifreq[16:20])[0]
592
return interface_index
593
594
595
def daemon(nochdir, noclose):
1066
596
"""See daemon(3). Standard BSD Unix function.
1067
597
This should really exist as os.daemon, but it doesn't (yet)."""
1068
598
if os.fork():
1070
600
os.setsid()
1071
601
if not nochdir:
1072
602
os.chdir("/")
1073
if os.fork():
1074
sys.exit()
1075
603
if not noclose:
1076
604
# Close all standard open file descriptors
1077
null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
605
null = os.open("/dev/null", os.O_NOCTTY | os.O_RDWR)
1078
606
if not stat.S_ISCHR(os.fstat(null).st_mode):
1079
607
raise OSError(errno.ENODEV,
1080
608
"/dev/null not a character device")
1085
613
os.close(null)
1086
614
1087
615
616
def killme(status = 0):
617
logger.debug("Stopping server with exit status %d", status)
618
exitstatus = status
619
if main_loop_started:
620
main_loop.quit()
621
else:
622
sys.exit(status)
623
624
1088
625
def main():
1089
1090
######################################################################
1091
# Parsing of options, both command line and config file
1092
1093
parser = optparse.OptionParser(version = "%%prog %s" % version)
626
global exitstatus
627
exitstatus = 0
628
global main_loop_started
629
main_loop_started = False
630
631
parser = OptionParser()
1094
632
parser.add_option("-i", "--interface", type="string",
1095
metavar="IF", help="Bind to interface IF")
1096
parser.add_option("-a", "--address", type="string",
633
default=None, metavar="IF",
634
help="Bind to interface IF")
635
parser.add_option("-a", "--address", type="string", default=None,
1097
636
help="Address to listen for requests on")
1098
parser.add_option("-p", "--port", type="int",
637
parser.add_option("-p", "--port", type="int", default=None,
1099
638
help="Port number to receive requests on")
1100
parser.add_option("--check", action="store_true",
639
parser.add_option("--timeout", type="string", # Parsed later
640
default="1h",
641
help="Amount of downtime allowed for clients")
642
parser.add_option("--interval", type="string", # Parsed later
643
default="5m",
644
help="How often to check that a client is up")
645
parser.add_option("--check", action="store_true", default=False,
1101
646
help="Run self-test")
1102
parser.add_option("--debug", action="store_true",
1103
help="Debug mode; run in foreground and log to"
1104
" terminal")
1105
parser.add_option("--priority", type="string", help="GnuTLS"
1106
" priority string (see GnuTLS documentation)")
1107
parser.add_option("--servicename", type="string", metavar="NAME",
1108
help="Zeroconf service name")
1109
parser.add_option("--configdir", type="string",
1110
default="/etc/mandos", metavar="DIR",
1111
help="Directory to search for configuration"
1112
" files")
1113
parser.add_option("--no-dbus", action="store_false",
1114
dest="use_dbus",
1115
help="Do not provide D-Bus system bus"
1116
" interface")
1117
parser.add_option("--no-ipv6", action="store_false",
1118
dest="use_ipv6", help="Do not use IPv6")
1119
options = parser.parse_args()[0]
647
parser.add_option("--debug", action="store_true", default=False,
648
help="Debug mode")
649
(options, args) = parser.parse_args()
1120
650
1121
651
if options.check:
1122
652
import doctest
1123
653
doctest.testmod()
1124
654
sys.exit()
1125
655
1126
# Default values for config file for server-global settings
1127
server_defaults = { "interface": "",
1128
"address": "",
1129
"port": "",
1130
"debug": "False",
1131
"priority":
1132
"SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
1133
"servicename": "Mandos",
1134
"use_dbus": "True",
1135
"use_ipv6": "True",
1136
}
1137
1138
# Parse config file for server-global settings
1139
server_config = ConfigParser.SafeConfigParser(server_defaults)
1140
del server_defaults
1141
server_config.read(os.path.join(options.configdir, "mandos.conf"))
1142
# Convert the SafeConfigParser object to a dict
1143
server_settings = server_config.defaults()
1144
# Use the appropriate methods on the non-string config options
1145
server_settings["debug"] = server_config.getboolean("DEFAULT",
1146
"debug")
1147
server_settings["use_dbus"] = server_config.getboolean("DEFAULT",
1148
"use_dbus")
1149
server_settings["use_ipv6"] = server_config.getboolean("DEFAULT",
1150
"use_ipv6")
1151
if server_settings["port"]:
1152
server_settings["port"] = server_config.getint("DEFAULT",
1153
"port")
1154
del server_config
1155
1156
# Override the settings from the config file with command line
1157
# options, if set.
1158
for option in ("interface", "address", "port", "debug",
1159
"priority", "servicename", "configdir",
1160
"use_dbus", "use_ipv6"):
1161
value = getattr(options, option)
1162
if value is not None:
1163
server_settings[option] = value
1164
del options
1165
# Now we have our good server settings in "server_settings"
1166
1167
##################################################################
1168
1169
# For convenience
1170
debug = server_settings["debug"]
1171
use_dbus = server_settings["use_dbus"]
1172
use_ipv6 = server_settings["use_ipv6"]
1173
1174
if not debug:
1175
syslogger.setLevel(logging.WARNING)
1176
console.setLevel(logging.WARNING)
1177
1178
if server_settings["servicename"] != "Mandos":
1179
syslogger.setFormatter(logging.Formatter
1180
('Mandos (%s) [%%(process)d]:'
1181
' %%(levelname)s: %%(message)s'
1182
% server_settings["servicename"]))
1183
1184
# Parse config file with clients
1185
client_defaults = { "timeout": "1h",
1186
"interval": "5m",
1187
"checker": "fping -q -- %%(host)s",
1188
"host": "",
1189
}
1190
client_config = ConfigParser.SafeConfigParser(client_defaults)
1191
client_config.read(os.path.join(server_settings["configdir"],
1192
"clients.conf"))
1193
1194
global mandos_dbus_service
1195
mandos_dbus_service = None
1196
1197
clients = Set()
1198
tcp_server = IPv6_TCPServer((server_settings["address"],
1199
server_settings["port"]),
1200
TCP_handler,
1201
settings=server_settings,
1202
clients=clients, use_ipv6=use_ipv6)
1203
pidfilename = "/var/run/mandos.pid"
1204
try:
1205
pidfile = open(pidfilename, "w")
1206
except IOError:
1207
logger.error("Could not open file %r", pidfilename)
1208
1209
try:
1210
uid = pwd.getpwnam("_mandos").pw_uid
1211
gid = pwd.getpwnam("_mandos").pw_gid
1212
except KeyError:
1213
try:
1214
uid = pwd.getpwnam("mandos").pw_uid
1215
gid = pwd.getpwnam("mandos").pw_gid
1216
except KeyError:
1217
try:
1218
uid = pwd.getpwnam("nobody").pw_uid
1219
gid = pwd.getpwnam("nogroup").pw_gid
1220
except KeyError:
1221
uid = 65534
1222
gid = 65534
1223
try:
1224
os.setgid(gid)
1225
os.setuid(uid)
1226
except OSError, error:
1227
if error[0] != errno.EPERM:
1228
raise error
1229
1230
# Enable all possible GnuTLS debugging
1231
if debug:
1232
# "Use a log level over 10 to enable all debugging options."
1233
# - GnuTLS manual
1234
gnutls.library.functions.gnutls_global_set_log_level(11)
1235
1236
@gnutls.library.types.gnutls_log_func
1237
def debug_gnutls(level, string):
1238
logger.debug("GnuTLS: %s", string[:-1])
1239
1240
(gnutls.library.functions
1241
.gnutls_global_set_log_function(debug_gnutls))
1242
1243
global service
1244
protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
1245
service = AvahiService(name = server_settings["servicename"],
1246
servicetype = "_mandos._tcp",
1247
protocol = protocol)
1248
if server_settings["interface"]:
1249
service.interface = (if_nametoindex
1250
(server_settings["interface"]))
656
# Parse the time arguments
657
try:
658
options.timeout = string_to_delta(options.timeout)
659
except ValueError:
660
parser.error("option --timeout: Unparseable time")
661
try:
662
options.interval = string_to_delta(options.interval)
663
except ValueError:
664
parser.error("option --interval: Unparseable time")
665
666
# Parse config file
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")
1251
671
1252
672
global main_loop
1253
673
global bus
1256
676
DBusGMainLoop(set_as_default=True )
1257
677
main_loop = gobject.MainLoop()
1258
678
bus = dbus.SystemBus()
1259
server = dbus.Interface(bus.get_object(avahi.DBUS_NAME,
1260
avahi.DBUS_PATH_SERVER),
1261
avahi.DBUS_INTERFACE_SERVER)
679
server = dbus.Interface(
680
bus.get_object( avahi.DBUS_NAME, avahi.DBUS_PATH_SERVER ),
681
avahi.DBUS_INTERFACE_SERVER )
1262
682
# End of Avahi example code
1263
if use_dbus:
1264
bus_name = dbus.service.BusName(u"se.bsnet.fukt.Mandos", bus)
1265
683
1266
client_class = Client
1267
if use_dbus:
1268
client_class = ClientDBus
1269
clients.update(Set(
1270
client_class(name = section,
1271
config= dict(client_config.items(section)))
1272
for section in client_config.sections()))
1273
if not clients:
1274
logger.warning(u"No clients defined")
684
debug = options.debug
1275
685
1276
686
if debug:
1277
# Redirect stdin so all checkers get /dev/null
1278
null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
1279
os.dup2(null, sys.stdin.fileno())
1280
if null > 2:
1281
os.close(null)
1282
else:
1283
# No console logging
1284
logger.removeHandler(console)
1285
# Close all input and output, do double fork, etc.
1286
daemon()
1287
1288
try:
1289
with closing(pidfile):
1290
pid = os.getpid()
1291
pidfile.write(str(pid) + "\n")
1292
del pidfile
1293
except IOError:
1294
logger.error(u"Could not write to file %r with PID %d",
1295
pidfilename, pid)
1296
except NameError:
1297
# "pidfile" was never created
1298
pass
1299
del pidfilename
687
console = logging.StreamHandler()
688
# console.setLevel(logging.DEBUG)
689
console.setFormatter(logging.Formatter\
690
('%(levelname)s: %(message)s'))
691
logger.addHandler(console)
692
del console
693
694
clients = Set()
695
def remove_from_clients(client):
696
clients.remove(client)
697
if not clients:
698
logger.debug(u"No clients left, exiting")
699
killme()
700
701
clients.update(Set(Client(name=section, options=options,
702
stop_hook = remove_from_clients,
703
**(dict(client_config\
704
.items(section))))
705
for section in client_config.sections()))
706
707
if not debug:
708
daemon(False, False)
1300
709
1301
710
def cleanup():
1302
711
"Cleanup function; run on exit"
1309
718
1310
719
while clients:
1311
720
client = clients.pop()
1312
client.disable_hook = None
1313
client.disable()
721
client.stop_hook = None
722
client.stop()
1314
723
1315
724
atexit.register(cleanup)
1316
725
1317
726
if not debug:
1318
727
signal.signal(signal.SIGINT, signal.SIG_IGN)
1319
signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
1320
signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
1321
1322
if use_dbus:
1323
class MandosDBusService(dbus.service.Object):
1324
"""A D-Bus proxy object"""
1325
def __init__(self):
1326
dbus.service.Object.__init__(self, bus, "/")
1327
_interface = u"se.bsnet.fukt.Mandos"
1328
1329
@dbus.service.signal(_interface, signature="oa{sv}")
1330
def ClientAdded(self, objpath, properties):
1331
"D-Bus signal"
1332
pass
1333
1334
@dbus.service.signal(_interface, signature="s")
1335
def ClientNotFound(self, fingerprint):
1336
"D-Bus signal"
1337
pass
1338
1339
@dbus.service.signal(_interface, signature="os")
1340
def ClientRemoved(self, objpath, name):
1341
"D-Bus signal"
1342
pass
1343
1344
@dbus.service.method(_interface, out_signature="ao")
1345
def GetAllClients(self):
1346
"D-Bus method"
1347
return dbus.Array(c.dbus_object_path for c in clients)
1348
1349
@dbus.service.method(_interface, out_signature="a{oa{sv}}")
1350
def GetAllClientsWithProperties(self):
1351
"D-Bus method"
1352
return dbus.Dictionary(
1353
((c.dbus_object_path, c.GetAllProperties())
1354
for c in clients),
1355
signature="oa{sv}")
1356
1357
@dbus.service.method(_interface, in_signature="o")
1358
def RemoveClient(self, object_path):
1359
"D-Bus method"
1360
for c in clients:
1361
if c.dbus_object_path == object_path:
1362
clients.remove(c)
1363
c.remove_from_connection()
1364
# Don't signal anything except ClientRemoved
1365
c.disable(signal=False)
1366
# Emit D-Bus signal
1367
self.ClientRemoved(object_path, c.name)
1368
return
1369
raise KeyError
1370
1371
del _interface
1372
1373
mandos_dbus_service = MandosDBusService()
728
signal.signal(signal.SIGHUP, lambda signum, frame: killme())
729
signal.signal(signal.SIGTERM, lambda signum, frame: killme())
1374
730
1375
731
for client in clients:
1376
if use_dbus:
1377
# Emit D-Bus signal
1378
mandos_dbus_service.ClientAdded(client.dbus_object_path,
1379
client.GetAllProperties())
1380
client.enable()
1381
1382
tcp_server.enable()
1383
tcp_server.server_activate()
1384
1385
# Find out what port we got
1386
service.port = tcp_server.socket.getsockname()[1]
1387
if use_ipv6:
1388
logger.info(u"Now listening on address %r, port %d,"
1389
" flowinfo %d, scope_id %d"
1390
% tcp_server.socket.getsockname())
1391
else: # IPv4
1392
logger.info(u"Now listening on address %r, port %d"
1393
% tcp_server.socket.getsockname())
1394
1395
#service.interface = tcp_server.socket.getsockname()[3]
1396
1397
try:
1398
# From the Avahi example code
1399
server.connect_to_signal("StateChanged", server_state_changed)
1400
try:
1401
server_state_changed(server.GetState())
1402
except dbus.exceptions.DBusException, error:
1403
logger.critical(u"DBusException: %s", error)
1404
sys.exit(1)
1405
# End of Avahi example code
1406
1407
gobject.io_add_watch(tcp_server.fileno(), gobject.IO_IN,
1408
lambda *args, **kwargs:
1409
(tcp_server.handle_request
1410
(*args[2:], **kwargs) or True))
1411
1412
logger.debug(u"Starting main loop")
732
client.start()
733
734
tcp_server = IPv6_TCPServer((options.address, options.port),
735
tcp_handler,
736
options=options,
737
clients=clients)
738
# Find out what random port we got
739
global servicePort
740
servicePort = tcp_server.socket.getsockname()[1]
741
logger.debug(u"Now listening on port %d", servicePort)
742
743
if options.interface is not None:
744
global serviceInterface
745
serviceInterface = if_nametoindex(options.interface)
746
747
# From the Avahi example code
748
server.connect_to_signal("StateChanged", server_state_changed)
749
try:
750
server_state_changed(server.GetState())
751
except dbus.exceptions.DBusException, error:
752
logger.critical(u"DBusException: %s", error)
753
killme(1)
754
# End of Avahi example code
755
756
gobject.io_add_watch(tcp_server.fileno(), gobject.IO_IN,
757
lambda *args, **kwargs:
758
tcp_server.handle_request(*args[2:],
759
**kwargs) or True)
760
try:
761
logger.debug("Starting main loop")
762
main_loop_started = True
1413
763
main_loop.run()
1414
except AvahiError, error:
1415
logger.critical(u"AvahiError: %s", error)
1416
sys.exit(1)
1417
764
except KeyboardInterrupt:
1418
765
if debug:
1419
print >> sys.stderr
1420
logger.debug("Server received KeyboardInterrupt")
1421
logger.debug("Server exiting")
766
print
767
768
sys.exit(exitstatus)
1422
769
1423
770
if __name__ == '__main__':
1424
771
main()
Loggerhead is a web-based interface for Breezy
Version: 2.0.2.dev0