/mandos/trunk

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/trunk

« back to all changes in this revision

Viewing changes to mandos

  • Committer: Teddy Hogeborn
  • Date: 2009-01-31 10:33:17 UTC
  • mfrom: (24.1.129 mandos)
  • Revision ID: teddy@fukt.bsnet.se-20090131103317-wzqvyr532sjcjt7u
Merge from Björn:

* mandos-ctl: New option "--remove-client".  Only default to listing
              clients if no clients were given on the command line.
* plugins.d/mandos-client.c: Lower kernel log level while bringing up
                             network interface.  New option "--delay"
                             to control the maximum delay to wait for
                             running interface.
* plugins.d/mandos-client.xml (SYNOPSIS, OPTIONS): New option
                                                   "--delay".

Show diffs side-by-side

added added

removed removed

Lines of Context:
mandos
66
66
import ctypes
67
67
import ctypes.util
68
68
 
69
 
version = "1.0.6"
 
69
version = "1.0.5"
70
70
 
71
71
logger = logging.Logger('mandos')
72
72
syslogger = (logging.handlers.SysLogHandler
73
73
             (facility = logging.handlers.SysLogHandler.LOG_DAEMON,
74
74
              address = "/dev/log"))
75
75
syslogger.setFormatter(logging.Formatter
76
 
                       ('Mandos [%(process)d]: %(levelname)s:'
77
 
                        ' %(message)s'))
 
76
                       ('Mandos: %(levelname)s: %(message)s'))
78
77
logger.addHandler(syslogger)
79
78
 
80
79
console = logging.StreamHandler()
81
 
console.setFormatter(logging.Formatter('%(name)s [%(process)d]:'
82
 
                                       ' %(levelname)s: %(message)s'))
 
80
console.setFormatter(logging.Formatter('%(name)s: %(levelname)s:'
 
81
                                       ' %(message)s'))
83
82
logger.addHandler(console)
84
83
 
85
84
class AvahiError(Exception):
114
113
    """
115
114
    def __init__(self, interface = avahi.IF_UNSPEC, name = None,
116
115
                 servicetype = None, port = None, TXT = None,
117
 
                 domain = "", host = "", max_renames = 32768,
118
 
                 protocol = avahi.PROTO_UNSPEC):
 
116
                 domain = "", host = "", max_renames = 32768):
119
117
        self.interface = interface
120
118
        self.name = name
121
119
        self.type = servicetype
125
123
        self.host = host
126
124
        self.rename_count = 0
127
125
        self.max_renames = max_renames
128
 
        self.protocol = protocol
129
126
    def rename(self):
130
127
        """Derived from the Avahi example code"""
131
128
        if self.rename_count >= self.max_renames:
160
157
                     service.name, service.type)
161
158
        group.AddService(
162
159
                self.interface,         # interface
163
 
                self.protocol,          # protocol
 
160
                avahi.PROTO_INET6,      # protocol
164
161
                dbus.UInt32(0),         # flags
165
162
                self.name, self.type,
166
163
                self.domain, self.host,
205
202
                     client lives.  %() expansions are done at
206
203
                     runtime with vars(self) as dict, so that for
207
204
                     instance %(name)s can be used in the command.
208
 
    current_checker_command: string; current running checker_command
209
205
    use_dbus: bool(); Whether to provide D-Bus interface and signals
210
206
    dbus_object_path: dbus.ObjectPath ; only set if self.use_dbus
211
207
    """
260
256
        self.disable_initiator_tag = None
261
257
        self.checker_callback_tag = None
262
258
        self.checker_command = config["checker"]
263
 
        self.current_checker_command = None
264
259
        self.last_connect = None
265
260
        # Only now, when this client is initialized, can it show up on
266
261
        # the D-Bus
381
376
        # checkers alone, the checker would have to take more time
382
377
        # than 'timeout' for the client to be declared invalid, which
383
378
        # is as it should be.
384
 
        
385
 
        # If a checker exists, make sure it is not a zombie
386
 
        if self.checker is not None:
387
 
            pid, status = os.waitpid(self.checker.pid, os.WNOHANG)
388
 
            if pid:
389
 
                logger.warning("Checker was a zombie")
390
 
                gobject.source_remove(self.checker_callback_tag)
391
 
                self.checker_callback(pid, status,
392
 
                                      self.current_checker_command)
393
 
        # Start a new checker if needed
394
379
        if self.checker is None:
395
380
            try:
396
381
                # In case checker_command has exactly one % operator
406
391
                    logger.error(u'Could not format string "%s":'
407
392
                                 u' %s', self.checker_command, error)
408
393
                    return True # Try again later
409
 
                self.current_checker_command = command
410
394
            try:
411
395
                logger.info(u"Starting checker %r for %s",
412
396
                            command, self.name)
427
411
                                             (self.checker.pid,
428
412
                                              self.checker_callback,
429
413
                                              data=command))
430
 
                # The checker may have completed before the gobject
431
 
                # watch was added.  Check for this.
432
 
                pid, status = os.waitpid(self.checker.pid, os.WNOHANG)
433
 
                if pid:
434
 
                    gobject.source_remove(self.checker_callback_tag)
435
 
                    self.checker_callback(pid, status, command)
436
414
            except OSError, error:
437
415
                logger.error(u"Failed to start subprocess: %s",
438
416
                             error)
698
676
        # using OpenPGP certificates.
699
677
        
700
678
        #priority = ':'.join(("NONE", "+VERS-TLS1.1", "+AES-256-CBC",
701
 
        #                     "+SHA1", "+COMP-NULL", "+CTYPE-OPENPGP",
702
 
        #                     "+DHE-DSS"))
 
679
        #                "+SHA1", "+COMP-NULL", "+CTYPE-OPENPGP",
 
680
        #                "+DHE-DSS"))
703
681
        # Use a fallback default, since this MUST be set.
704
682
        priority = self.server.settings.get("priority", "NORMAL")
705
683
        (gnutls.library.functions
753
731
 
754
732
class IPv6_TCPServer(SocketServer.ForkingMixIn,
755
733
                     SocketServer.TCPServer, object):
756
 
    """IPv6-capable TCP server.  Accepts 'None' as address and/or port
 
734
    """IPv6 TCP server.  Accepts 'None' as address and/or port.
757
735
    Attributes:
758
736
        settings:       Server settings
759
737
        clients:        Set() of Client objects
767
745
        if "clients" in kwargs:
768
746
            self.clients = kwargs["clients"]
769
747
            del kwargs["clients"]
770
 
        if "use_ipv6" in kwargs:
771
 
            if not kwargs["use_ipv6"]:
772
 
                self.address_family = socket.AF_INET
773
 
            del kwargs["use_ipv6"]
774
748
        self.enabled = False
775
749
        super(IPv6_TCPServer, self).__init__(*args, **kwargs)
776
750
    def server_bind(self):
790
764
                                 u" bind to interface %s",
791
765
                                 self.settings["interface"])
792
766
                else:
793
 
                    raise
 
767
                    raise error
794
768
        # Only bind(2) the socket if we really need to.
795
769
        if self.server_address[0] or self.server_address[1]:
796
770
            if not self.server_address[0]:
797
 
                if self.address_family == socket.AF_INET6:
798
 
                    any_address = "::" # in6addr_any
799
 
                else:
800
 
                    any_address = socket.INADDR_ANY
801
 
                self.server_address = (any_address,
 
771
                in6addr_any = "::"
 
772
                self.server_address = (in6addr_any,
802
773
                                       self.server_address[1])
803
774
            elif not self.server_address[1]:
804
775
                self.server_address = (self.server_address[0],
951
922
                      dest="use_dbus",
952
923
                      help="Do not provide D-Bus system bus"
953
924
                      " interface")
954
 
    parser.add_option("--no-ipv6", action="store_false",
955
 
                      dest="use_ipv6", help="Do not use IPv6")
956
925
    options = parser.parse_args()[0]
957
926
    
958
927
    if options.check:
969
938
                        "SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
970
939
                        "servicename": "Mandos",
971
940
                        "use_dbus": "True",
972
 
                        "use_ipv6": "True",
973
941
                        }
974
942
    
975
943
    # Parse config file for server-global settings
983
951
                                                        "debug")
984
952
    server_settings["use_dbus"] = server_config.getboolean("DEFAULT",
985
953
                                                           "use_dbus")
986
 
    server_settings["use_ipv6"] = server_config.getboolean("DEFAULT",
987
 
                                                           "use_ipv6")
988
954
    if server_settings["port"]:
989
955
        server_settings["port"] = server_config.getint("DEFAULT",
990
956
                                                       "port")
994
960
    # options, if set.
995
961
    for option in ("interface", "address", "port", "debug",
996
962
                   "priority", "servicename", "configdir",
997
 
                   "use_dbus", "use_ipv6"):
 
963
                   "use_dbus"):
998
964
        value = getattr(options, option)
999
965
        if value is not None:
1000
966
            server_settings[option] = value
1004
970
    # For convenience
1005
971
    debug = server_settings["debug"]
1006
972
    use_dbus = server_settings["use_dbus"]
1007
 
    use_ipv6 = server_settings["use_ipv6"]
 
973
 
 
974
    def sigsegvhandler(signum, frame):
 
975
        raise RuntimeError('Segmentation fault')
1008
976
    
1009
977
    if not debug:
1010
978
        syslogger.setLevel(logging.WARNING)
1011
979
        console.setLevel(logging.WARNING)
 
980
    else:
 
981
        signal.signal(signal.SIGSEGV, sigsegvhandler)
1012
982
    
1013
983
    if server_settings["servicename"] != "Mandos":
1014
984
        syslogger.setFormatter(logging.Formatter
1031
1001
                                 server_settings["port"]),
1032
1002
                                TCP_handler,
1033
1003
                                settings=server_settings,
1034
 
                                clients=clients, use_ipv6=use_ipv6)
 
1004
                                clients=clients)
1035
1005
    pidfilename = "/var/run/mandos.pid"
1036
1006
    try:
1037
1007
        pidfile = open(pidfilename, "w")
1038
 
    except IOError:
 
1008
    except IOError, error:
1039
1009
        logger.error("Could not open file %r", pidfilename)
1040
1010
    
1041
1011
    try:
1073
1043
         .gnutls_global_set_log_function(debug_gnutls))
1074
1044
    
1075
1045
    global service
1076
 
    protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
1077
1046
    service = AvahiService(name = server_settings["servicename"],
1078
 
                           servicetype = "_mandos._tcp",
1079
 
                           protocol = protocol)
 
1047
                           servicetype = "_mandos._tcp", )
1080
1048
    if server_settings["interface"]:
1081
1049
        service.interface = (if_nametoindex
1082
1050
                             (server_settings["interface"]))
1209
1177
    
1210
1178
    # Find out what port we got
1211
1179
    service.port = tcp_server.socket.getsockname()[1]
1212
 
    if use_ipv6:
1213
 
        logger.info(u"Now listening on address %r, port %d,"
1214
 
                    " flowinfo %d, scope_id %d"
1215
 
                    % tcp_server.socket.getsockname())
1216
 
    else:                       # IPv4
1217
 
        logger.info(u"Now listening on address %r, port %d"
1218
 
                    % tcp_server.socket.getsockname())
 
1180
    logger.info(u"Now listening on address %r, port %d, flowinfo %d,"
 
1181
                u" scope_id %d" % tcp_server.socket.getsockname())
1219
1182
    
1220
1183
    #service.interface = tcp_server.socket.getsockname()[3]
1221
1184
    
1241
1204
        sys.exit(1)
1242
1205
    except KeyboardInterrupt:
1243
1206
        if debug:
1244
 
            print >> sys.stderr
1245
 
        logger.debug("Server received KeyboardInterrupt")
1246
 
    logger.debug("Server exiting")
 
1207
            print
1247
1208
 
1248
1209
if __name__ == '__main__':
1249
1210
    main()