133
133
u" retries, exiting.", rename_count)
134
134
raise AvahiServiceError("Too many renames")
135
135
name = server.GetAlternativeServiceName(name)
136
logger.error(u"Changing name to %r ...", name)
136
logger.notice(u"Changing name to %r ...", name)
139
139
self.rename_count += 1
221
221
interval = property(lambda self: self._interval,
223
223
del _set_interval
224
def __init__(self, name = None, stop_hook=None, config={}):
224
def __init__(self, name=None, stop_hook=None, fingerprint=None,
225
secret=None, secfile=None, fqdn=None, timeout=None,
226
interval=-1, checker=None):
225
227
"""Note: the 'checker' argument sets the 'checker_command'
226
228
attribute and not the 'checker' attribute.."""
229
231
# Uppercase and remove spaces from fingerprint
230
232
# for later comparison purposes with return value of
231
233
# the fingerprint() function
232
self.fingerprint = config["fingerprint"].upper()\
234
self.fingerprint = fingerprint.upper().replace(u" ", u"")
234
235
logger.debug(u" Fingerprint: %s", self.fingerprint)
235
if "secret" in config:
236
self.secret = config["secret"].decode(u"base64")
237
elif "secfile" in config:
238
sf = open(config["secfile"])
237
self.secret = secret.decode(u"base64")
239
240
self.secret = sf.read()
242
243
raise TypeError(u"No secret or secfile for client %s"
244
self.fqdn = config.get("fqdn", "")
245
246
self.created = datetime.datetime.now()
246
247
self.last_checked_ok = None
247
self.timeout = string_to_delta(config["timeout"])
248
self.interval = string_to_delta(config["interval"])
248
self.timeout = string_to_delta(timeout)
249
self.interval = string_to_delta(interval)
249
250
self.stop_hook = stop_hook
250
251
self.checker = None
251
252
self.checker_initiator_tag = None
252
253
self.stop_initiator_tag = None
253
254
self.checker_callback_tag = None
254
self.check_command = config["checker"]
255
self.check_command = checker
256
257
"""Start this client's checker and timeout hooks"""
257
258
# Schedule a new checker to be started an 'interval' from now,
271
272
but not currently used."""
272
273
# If this client doesn't have a secret, it is already stopped.
274
logger.info(u"Stopping client %s", self.name)
275
logger.debug(u"Stopping client %s", self.name)
275
276
self.secret = None
296
297
self.checker = None
297
298
if os.WIFEXITED(condition) \
298
299
and (os.WEXITSTATUS(condition) == 0):
299
logger.info(u"Checker for %(name)s succeeded",
300
logger.debug(u"Checker for %(name)s succeeded",
301
302
self.last_checked_ok = now
302
303
gobject.source_remove(self.stop_initiator_tag)
303
304
self.stop_initiator_tag = gobject.timeout_add\
307
308
logger.warning(u"Checker for %(name)s crashed?",
310
logger.info(u"Checker for %(name)s failed",
311
logger.debug(u"Checker for %(name)s failed",
312
313
def start_checker(self):
313
314
"""Start a new checker subprocess if one is not running.
314
315
If a checker already exists, leave it running and do
337
338
u' %s', self.check_command, error)
338
339
return True # Try again later
340
logger.info(u"Starting checker %r for %s",
341
logger.debug(u"Starting checker %r for %s",
342
343
self.checker = subprocess.Popen(command,
344
345
shell=True, cwd="/")
430
431
Note: This will run in its own forked process."""
432
433
def handle(self):
433
logger.info(u"TCP connection from: %s",
434
logger.debug(u"TCP connection from: %s",
434
435
unicode(self.client_address))
435
436
session = gnutls.connection.ClientSession\
436
437
(self.request, gnutls.connection.X509Credentials())
463
464
session.handshake()
464
465
except gnutls.errors.GNUTLSError, error:
465
logger.warning(u"Handshake failed: %s", error)
466
logger.debug(u"Handshake failed: %s", error)
466
467
# Do not run session.bye() here: the session is not
467
468
# established. Just abandon the request.
470
471
fpr = fingerprint(peer_certificate(session))
471
472
except (TypeError, gnutls.errors.GNUTLSError), error:
472
logger.warning(u"Bad certificate: %s", error)
473
logger.debug(u"Bad certificate: %s", error)
475
476
logger.debug(u"Fingerprint: %s", fpr)
482
logger.warning(u"Client not found for fingerprint: %s",
483
logger.debug(u"Client not found for fingerprint: %s", fpr)
486
486
# Have to check if client.still_valid(), since it is possible
487
487
# that the client timed out while establishing the GnuTLS
489
489
if not client.still_valid():
490
logger.warning(u"Client %(name)s is invalid",
490
logger.debug(u"Client %(name)s is invalid", vars(client))
529
528
self.settings["interface"])
530
529
except socket.error, error:
531
530
if error[0] == errno.EPERM:
532
logger.error(u"No permission to"
533
u" bind to interface %s",
534
self.settings["interface"])
531
logger.warning(u"No permission to"
532
u" bind to interface %s",
533
self.settings["interface"])
537
536
# Only bind(2) the socket if we really need to.
583
582
def server_state_changed(state):
584
583
"""Derived from the Avahi example code"""
585
584
if state == avahi.SERVER_COLLISION:
586
logger.error(u"Server name collision")
585
logger.warning(u"Server name collision")
588
587
elif state == avahi.SERVER_RUNNING:
753
752
def remove_from_clients(client):
754
753
clients.remove(client)
756
logger.critical(u"No clients left, exiting")
755
logger.debug(u"No clients left, exiting")
759
clients.update(Set(Client(name = section,
758
clients.update(Set(Client(name=section,
760
759
stop_hook = remove_from_clients,
762
= dict(client_config.items(section)))
760
**(dict(client_config\
763
762
for section in client_config.sections()))
797
796
# Find out what port we got
798
797
service.port = tcp_server.socket.getsockname()[1]
799
logger.info(u"Now listening on address %r, port %d, flowinfo %d,"
800
u" scope_id %d" % tcp_server.socket.getsockname())
798
logger.debug(u"Now listening on address %r, port %d, flowinfo %d,"
799
u" scope_id %d" % tcp_server.socket.getsockname())
802
801
#service.interface = tcp_server.socket.getsockname()[3]