/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 at recompile
  • Date: 2020-01-12 01:53:04 UTC
  • Revision ID: teddy@recompile.se-20200112015304-gqif9w4pbyix8w6b
mandos-ctl: Fix minor bug in tests

Fix two tests, the last assert of which would always erroneously
succeed.  (No change in non-test code needed.)

* mandos-ctl (Test_dbus_python_adapter_SystemBus): In the
  test_call_method_handles_exception() method, fix check for
  dbus.ConnectFailed exception.
  (Test_pydbus_adapter_SystemBus): - '' -

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
#!/usr/bin/python
2
 
# -*- mode: python; coding: utf-8 -*-
 
1
#!/usr/bin/python3 -bI
 
2
# -*- mode: python; after-save-hook: (lambda () (let ((command (if (fboundp 'file-local-name) (file-local-name (buffer-file-name)) (or (file-remote-p (buffer-file-name) 'localname) (buffer-file-name))))) (if (= (progn (if (get-buffer "*Test*") (kill-buffer "*Test*")) (process-file-shell-command (format "%s --check" (shell-quote-argument command)) nil "*Test*")) 0) (let ((w (get-buffer-window "*Test*"))) (if w (delete-window w))) (progn (with-current-buffer "*Test*" (compilation-mode)) (display-buffer "*Test*" '(display-buffer-in-side-window)))))); coding: utf-8 -*-
3
3
#
4
4
# Mandos server - give out binary blobs to connecting clients.
5
5
#
77
77
import itertools
78
78
import collections
79
79
import codecs
 
80
import unittest
80
81
 
81
82
import dbus
82
83
import dbus.service
 
84
import gi
83
85
from gi.repository import GLib
84
86
from dbus.mainloop.glib import DBusGMainLoop
85
87
import ctypes
87
89
import xml.dom.minidom
88
90
import inspect
89
91
 
 
92
if sys.version_info.major == 2:
 
93
    __metaclass__ = type
 
94
    str = unicode
 
95
 
 
96
# Show warnings by default
 
97
if not sys.warnoptions:
 
98
    import warnings
 
99
    warnings.simplefilter("default")
 
100
 
90
101
# Try to find the value of SO_BINDTODEVICE:
91
102
try:
92
103
    # This is where SO_BINDTODEVICE is in Python 3.3 (or 3.4?) and
112
123
            # No value found
113
124
            SO_BINDTODEVICE = None
114
125
 
115
 
if sys.version_info.major == 2:
116
 
    str = unicode
 
126
if sys.version_info < (3, 2):
 
127
    configparser.Configparser = configparser.SafeConfigParser
117
128
 
118
 
version = "1.7.20"
 
129
version = "1.8.9"
119
130
stored_state_file = "clients.pickle"
120
131
 
121
132
logger = logging.getLogger()
 
133
logging.captureWarnings(True)   # Show warnings via the logging system
122
134
syslogger = None
123
135
 
124
136
try:
179
191
    pass
180
192
 
181
193
 
182
 
class PGPEngine(object):
 
194
class PGPEngine:
183
195
    """A simple class for OpenPGP symmetric encryption & decryption"""
184
196
 
185
197
    def __init__(self):
189
201
            output = subprocess.check_output(["gpgconf"])
190
202
            for line in output.splitlines():
191
203
                name, text, path = line.split(b":")
192
 
                if name == "gpg":
 
204
                if name == b"gpg":
193
205
                    self.gpg = path
194
206
                    break
195
207
        except OSError as e:
200
212
                          '--force-mdc',
201
213
                          '--quiet']
202
214
        # Only GPG version 1 has the --no-use-agent option.
203
 
        if self.gpg == "gpg" or self.gpg.endswith("/gpg"):
 
215
        if self.gpg == b"gpg" or self.gpg.endswith(b"/gpg"):
204
216
            self.gnupgargs.append("--no-use-agent")
205
217
 
206
218
    def __enter__(self):
275
287
 
276
288
 
277
289
# Pretend that we have an Avahi module
278
 
class Avahi(object):
279
 
    """This isn't so much a class as it is a module-like namespace.
280
 
    It is instantiated once, and simulates having an Avahi module."""
 
290
class avahi:
 
291
    """This isn't so much a class as it is a module-like namespace."""
281
292
    IF_UNSPEC = -1               # avahi-common/address.h
282
293
    PROTO_UNSPEC = -1            # avahi-common/address.h
283
294
    PROTO_INET = 0               # avahi-common/address.h
287
298
    DBUS_INTERFACE_SERVER = DBUS_NAME + ".Server"
288
299
    DBUS_PATH_SERVER = "/"
289
300
 
290
 
    def string_array_to_txt_array(self, t):
 
301
    @staticmethod
 
302
    def string_array_to_txt_array(t):
291
303
        return dbus.Array((dbus.ByteArray(s.encode("utf-8"))
292
304
                           for s in t), signature="ay")
293
305
    ENTRY_GROUP_ESTABLISHED = 2  # avahi-common/defs.h
298
310
    SERVER_RUNNING = 2           # avahi-common/defs.h
299
311
    SERVER_COLLISION = 3         # avahi-common/defs.h
300
312
    SERVER_FAILURE = 4           # avahi-common/defs.h
301
 
avahi = Avahi()
302
313
 
303
314
 
304
315
class AvahiError(Exception):
316
327
    pass
317
328
 
318
329
 
319
 
class AvahiService(object):
 
330
class AvahiService:
320
331
    """An Avahi (Zeroconf) service.
321
332
 
322
333
    Attributes:
504
515
 
505
516
 
506
517
# Pretend that we have a GnuTLS module
507
 
class GnuTLS(object):
508
 
    """This isn't so much a class as it is a module-like namespace.
509
 
    It is instantiated once, and simulates having a GnuTLS module."""
 
518
class gnutls:
 
519
    """This isn't so much a class as it is a module-like namespace."""
510
520
 
511
521
    library = ctypes.util.find_library("gnutls")
512
522
    if library is None:
513
523
        library = ctypes.util.find_library("gnutls-deb0")
514
524
    _library = ctypes.cdll.LoadLibrary(library)
515
525
    del library
516
 
    _need_version = b"3.3.0"
517
 
    _tls_rawpk_version = b"3.6.6"
518
 
 
519
 
    def __init__(self):
520
 
        # Need to use "self" here, since this method is called before
521
 
        # the assignment to the "gnutls" global variable happens.
522
 
        if self.check_version(self._need_version) is None:
523
 
            raise self.Error("Needs GnuTLS {} or later"
524
 
                             .format(self._need_version))
525
526
 
526
527
    # Unless otherwise indicated, the constants and types below are
527
528
    # all from the gnutls/gnutls.h C header file.
569
570
 
570
571
    # Exceptions
571
572
    class Error(Exception):
572
 
        # We need to use the class name "GnuTLS" here, since this
573
 
        # exception might be raised from within GnuTLS.__init__,
574
 
        # which is called before the assignment to the "gnutls"
575
 
        # global variable has happened.
576
573
        def __init__(self, message=None, code=None, args=()):
577
574
            # Default usage is by a message string, but if a return
578
575
            # code is passed, convert it to a string with
579
576
            # gnutls.strerror()
580
577
            self.code = code
581
578
            if message is None and code is not None:
582
 
                message = GnuTLS.strerror(code)
583
 
            return super(GnuTLS.Error, self).__init__(
 
579
                message = gnutls.strerror(code)
 
580
            return super(gnutls.Error, self).__init__(
584
581
                message, *args)
585
582
 
586
583
    class CertificateSecurityError(Error):
587
584
        pass
588
585
 
589
586
    # Classes
590
 
    class Credentials(object):
 
587
    class Credentials:
591
588
        def __init__(self):
592
589
            self._c_object = gnutls.certificate_credentials_t()
593
590
            gnutls.certificate_allocate_credentials(
597
594
        def __del__(self):
598
595
            gnutls.certificate_free_credentials(self._c_object)
599
596
 
600
 
    class ClientSession(object):
 
597
    class ClientSession:
601
598
        def __init__(self, socket, credentials=None):
602
599
            self._c_object = gnutls.session_t()
603
600
            gnutls_flags = gnutls.CLIENT
604
 
            if gnutls.check_version("3.5.6"):
 
601
            if gnutls.check_version(b"3.5.6"):
605
602
                gnutls_flags |= gnutls.NO_TICKETS
606
603
            if gnutls.has_rawpk:
607
604
                gnutls_flags |= gnutls.ENABLE_RAWPK
744
741
    check_version.argtypes = [ctypes.c_char_p]
745
742
    check_version.restype = ctypes.c_char_p
746
743
 
 
744
    _need_version = b"3.3.0"
 
745
    if check_version(_need_version) is None:
 
746
        raise self.Error("Needs GnuTLS {} or later"
 
747
                         .format(_need_version))
 
748
 
 
749
    _tls_rawpk_version = b"3.6.6"
747
750
    has_rawpk = bool(check_version(_tls_rawpk_version))
748
751
 
749
752
    if has_rawpk:
803
806
                                                    ctypes.c_size_t)]
804
807
        openpgp_crt_get_fingerprint.restype = _error_code
805
808
 
806
 
    if check_version("3.6.4"):
 
809
    if check_version(b"3.6.4"):
807
810
        certificate_type_get2 = _library.gnutls_certificate_type_get2
808
811
        certificate_type_get2.argtypes = [session_t, ctypes.c_int]
809
812
        certificate_type_get2.restype = _error_code
810
813
 
811
814
    # Remove non-public functions
812
815
    del _error_code, _retry_on_error
813
 
# Create the global "gnutls" object, simulating a module
814
 
gnutls = GnuTLS()
815
816
 
816
817
 
817
818
def call_pipe(connection,       # : multiprocessing.Connection
825
826
    connection.close()
826
827
 
827
828
 
828
 
class Client(object):
 
829
class Client:
829
830
    """A representation of a client host served by this server.
830
831
 
831
832
    Attributes:
832
833
    approved:   bool(); 'None' if not yet approved/disapproved
833
834
    approval_delay: datetime.timedelta(); Time to wait for approval
834
835
    approval_duration: datetime.timedelta(); Duration of one approval
835
 
    checker:    subprocess.Popen(); a running checker process used
836
 
                                    to see if the client lives.
837
 
                                    'None' if no process is running.
 
836
    checker: multiprocessing.Process(); a running checker process used
 
837
             to see if the client lives. 'None' if no process is
 
838
             running.
838
839
    checker_callback_tag: a GLib event source tag, or None
839
840
    checker_command: string; External command which is run to check
840
841
                     if client lives.  %() expansions are done at
1047
1048
    def checker_callback(self, source, condition, connection,
1048
1049
                         command):
1049
1050
        """The checker has completed, so take appropriate actions."""
1050
 
        self.checker_callback_tag = None
1051
 
        self.checker = None
1052
1051
        # Read return code from connection (see call_pipe)
1053
1052
        returncode = connection.recv()
1054
1053
        connection.close()
 
1054
        if self.checker is not None:
 
1055
            self.checker.join()
 
1056
        self.checker_callback_tag = None
 
1057
        self.checker = None
1055
1058
 
1056
1059
        if returncode >= 0:
1057
1060
            self.last_checker_status = returncode
1146
1149
                kwargs=popen_args)
1147
1150
            self.checker.start()
1148
1151
            self.checker_callback_tag = GLib.io_add_watch(
1149
 
                pipe[0].fileno(), GLib.IO_IN,
 
1152
                GLib.IOChannel.unix_new(pipe[0].fileno()),
 
1153
                GLib.PRIORITY_DEFAULT, GLib.IO_IN,
1150
1154
                self.checker_callback, pipe[0], command)
1151
1155
        # Re-run this periodically if run by GLib.timeout_add
1152
1156
        return True
1407
1411
                raise ValueError("Byte arrays not supported for non-"
1408
1412
                                 "'ay' signature {!r}"
1409
1413
                                 .format(prop._dbus_signature))
1410
 
            value = dbus.ByteArray(b''.join(chr(byte)
1411
 
                                            for byte in value))
 
1414
            value = dbus.ByteArray(bytes(value))
1412
1415
        prop(value)
1413
1416
 
1414
1417
    @dbus.service.method(dbus.PROPERTIES_IFACE,
2219
2222
    del _interface
2220
2223
 
2221
2224
 
2222
 
class ProxyClient(object):
 
2225
class ProxyClient:
2223
2226
    def __init__(self, child_pipe, key_id, fpr, address):
2224
2227
        self._pipe = child_pipe
2225
2228
        self._pipe.send(('init', key_id, fpr, address))
2298
2301
            approval_required = False
2299
2302
            try:
2300
2303
                if gnutls.has_rawpk:
2301
 
                    fpr = ""
 
2304
                    fpr = b""
2302
2305
                    try:
2303
2306
                        key_id = self.key_id(
2304
2307
                            self.peer_certificate(session))
2308
2311
                    logger.debug("Key ID: %s", key_id)
2309
2312
 
2310
2313
                else:
2311
 
                    key_id = ""
 
2314
                    key_id = b""
2312
2315
                    try:
2313
2316
                        fpr = self.fingerprint(
2314
2317
                            self.peer_certificate(session))
2498
2501
        return hex_fpr
2499
2502
 
2500
2503
 
2501
 
class MultiprocessingMixIn(object):
 
2504
class MultiprocessingMixIn:
2502
2505
    """Like socketserver.ThreadingMixIn, but with multiprocessing"""
2503
2506
 
2504
2507
    def sub_process_main(self, request, address):
2516
2519
        return proc
2517
2520
 
2518
2521
 
2519
 
class MultiprocessingMixInWithPipe(MultiprocessingMixIn, object):
 
2522
class MultiprocessingMixInWithPipe(MultiprocessingMixIn):
2520
2523
    """ adds a pipe to the MixIn """
2521
2524
 
2522
2525
    def process_request(self, request, client_address):
2537
2540
 
2538
2541
 
2539
2542
class IPv6_TCPServer(MultiprocessingMixInWithPipe,
2540
 
                     socketserver.TCPServer, object):
 
2543
                     socketserver.TCPServer):
2541
2544
    """IPv6-capable TCP server.  Accepts 'None' as address and/or port
2542
2545
 
2543
2546
    Attributes:
2616
2619
                    raise
2617
2620
        # Only bind(2) the socket if we really need to.
2618
2621
        if self.server_address[0] or self.server_address[1]:
 
2622
            if self.server_address[1]:
 
2623
                self.allow_reuse_address = True
2619
2624
            if not self.server_address[0]:
2620
2625
                if self.address_family == socket.AF_INET6:
2621
2626
                    any_address = "::"  # in6addr_any
2674
2679
    def add_pipe(self, parent_pipe, proc):
2675
2680
        # Call "handle_ipc" for both data and EOF events
2676
2681
        GLib.io_add_watch(
2677
 
            parent_pipe.fileno(),
2678
 
            GLib.IO_IN | GLib.IO_HUP,
 
2682
            GLib.IOChannel.unix_new(parent_pipe.fileno()),
 
2683
            GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
2679
2684
            functools.partial(self.handle_ipc,
2680
2685
                              parent_pipe=parent_pipe,
2681
2686
                              proc=proc))
2700
2705
            address = request[3]
2701
2706
 
2702
2707
            for c in self.clients.values():
 
2708
                if key_id == "E3B0C44298FC1C149AFBF4C8996FB92427AE41E4649B934CA495991B7852B855":
 
2709
                    continue
2703
2710
                if key_id and c.key_id == key_id:
2704
2711
                    client = c
2705
2712
                    break
2717
2724
                return False
2718
2725
 
2719
2726
            GLib.io_add_watch(
2720
 
                parent_pipe.fileno(),
2721
 
                GLib.IO_IN | GLib.IO_HUP,
 
2727
                GLib.IOChannel.unix_new(parent_pipe.fileno()),
 
2728
                GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
2722
2729
                functools.partial(self.handle_ipc,
2723
2730
                                  parent_pipe=parent_pipe,
2724
2731
                                  proc=proc,
2756
2763
def rfc3339_duration_to_delta(duration):
2757
2764
    """Parse an RFC 3339 "duration" and return a datetime.timedelta
2758
2765
 
2759
 
    >>> rfc3339_duration_to_delta("P7D")
2760
 
    datetime.timedelta(7)
2761
 
    >>> rfc3339_duration_to_delta("PT60S")
2762
 
    datetime.timedelta(0, 60)
2763
 
    >>> rfc3339_duration_to_delta("PT60M")
2764
 
    datetime.timedelta(0, 3600)
2765
 
    >>> rfc3339_duration_to_delta("PT24H")
2766
 
    datetime.timedelta(1)
2767
 
    >>> rfc3339_duration_to_delta("P1W")
2768
 
    datetime.timedelta(7)
2769
 
    >>> rfc3339_duration_to_delta("PT5M30S")
2770
 
    datetime.timedelta(0, 330)
2771
 
    >>> rfc3339_duration_to_delta("P1DT3M20S")
2772
 
    datetime.timedelta(1, 200)
 
2766
    >>> rfc3339_duration_to_delta("P7D") == datetime.timedelta(7)
 
2767
    True
 
2768
    >>> rfc3339_duration_to_delta("PT60S") == datetime.timedelta(0, 60)
 
2769
    True
 
2770
    >>> rfc3339_duration_to_delta("PT60M") == datetime.timedelta(0, 3600)
 
2771
    True
 
2772
    >>> rfc3339_duration_to_delta("PT24H") == datetime.timedelta(1)
 
2773
    True
 
2774
    >>> rfc3339_duration_to_delta("P1W") == datetime.timedelta(7)
 
2775
    True
 
2776
    >>> rfc3339_duration_to_delta("PT5M30S") == datetime.timedelta(0, 330)
 
2777
    True
 
2778
    >>> rfc3339_duration_to_delta("P1DT3M20S") == datetime.timedelta(1, 200)
 
2779
    True
2773
2780
    """
2774
2781
 
2775
2782
    # Parsing an RFC 3339 duration with regular expressions is not
2855
2862
def string_to_delta(interval):
2856
2863
    """Parse a string and return a datetime.timedelta
2857
2864
 
2858
 
    >>> string_to_delta('7d')
2859
 
    datetime.timedelta(7)
2860
 
    >>> string_to_delta('60s')
2861
 
    datetime.timedelta(0, 60)
2862
 
    >>> string_to_delta('60m')
2863
 
    datetime.timedelta(0, 3600)
2864
 
    >>> string_to_delta('24h')
2865
 
    datetime.timedelta(1)
2866
 
    >>> string_to_delta('1w')
2867
 
    datetime.timedelta(7)
2868
 
    >>> string_to_delta('5m 30s')
2869
 
    datetime.timedelta(0, 330)
 
2865
    >>> string_to_delta('7d') == datetime.timedelta(7)
 
2866
    True
 
2867
    >>> string_to_delta('60s') == datetime.timedelta(0, 60)
 
2868
    True
 
2869
    >>> string_to_delta('60m') == datetime.timedelta(0, 3600)
 
2870
    True
 
2871
    >>> string_to_delta('24h') == datetime.timedelta(1)
 
2872
    True
 
2873
    >>> string_to_delta('1w') == datetime.timedelta(7)
 
2874
    True
 
2875
    >>> string_to_delta('5m 30s') == datetime.timedelta(0, 330)
 
2876
    True
2870
2877
    """
2871
2878
 
2872
2879
    try:
2974
2981
 
2975
2982
    options = parser.parse_args()
2976
2983
 
2977
 
    if options.check:
2978
 
        import doctest
2979
 
        fail_count, test_count = doctest.testmod()
2980
 
        sys.exit(os.EX_OK if fail_count == 0 else 1)
2981
 
 
2982
2984
    # Default values for config file for server-global settings
2983
2985
    if gnutls.has_rawpk:
2984
2986
        priority = ("SECURE128:!CTYPE-X.509:+CTYPE-RAWPK:!RSA"
3004
3006
    del priority
3005
3007
 
3006
3008
    # Parse config file for server-global settings
3007
 
    server_config = configparser.SafeConfigParser(server_defaults)
 
3009
    server_config = configparser.ConfigParser(server_defaults)
3008
3010
    del server_defaults
3009
3011
    server_config.read(os.path.join(options.configdir, "mandos.conf"))
3010
 
    # Convert the SafeConfigParser object to a dict
 
3012
    # Convert the ConfigParser object to a dict
3011
3013
    server_settings = server_config.defaults()
3012
3014
    # Use the appropriate methods on the non-string config options
3013
3015
    for option in ("debug", "use_dbus", "use_ipv6", "restore",
3085
3087
                                  server_settings["servicename"])))
3086
3088
 
3087
3089
    # Parse config file with clients
3088
 
    client_config = configparser.SafeConfigParser(Client
3089
 
                                                  .client_defaults)
 
3090
    client_config = configparser.ConfigParser(Client.client_defaults)
3090
3091
    client_config.read(os.path.join(server_settings["configdir"],
3091
3092
                                    "clients.conf"))
3092
3093
 
3163
3164
        # Close all input and output, do double fork, etc.
3164
3165
        daemon()
3165
3166
 
3166
 
    # multiprocessing will use threads, so before we use GLib we need
3167
 
    # to inform GLib that threads will be used.
3168
 
    GLib.threads_init()
 
3167
    if gi.version_info < (3, 10, 2):
 
3168
        # multiprocessing will use threads, so before we use GLib we
 
3169
        # need to inform GLib that threads will be used.
 
3170
        GLib.threads_init()
3169
3171
 
3170
3172
    global main_loop
3171
3173
    # From the Avahi example code
3247
3249
                             if isinstance(s, bytes)
3248
3250
                             else s) for s in
3249
3251
                            value["client_structure"]]
3250
 
                        # .name & .host
3251
 
                        for k in ("name", "host"):
 
3252
                        # .name, .host, and .checker_command
 
3253
                        for k in ("name", "host", "checker_command"):
3252
3254
                            if isinstance(value[k], bytes):
3253
3255
                                value[k] = value[k].decode("utf-8")
3254
 
                        if not value.has_key("key_id"):
 
3256
                        if "key_id" not in value:
3255
3257
                            value["key_id"] = ""
3256
 
                        elif not value.has_key("fingerprint"):
 
3258
                        elif "fingerprint" not in value:
3257
3259
                            value["fingerprint"] = ""
3258
3260
                    #  old_client_settings
3259
3261
                    # .keys()
3264
3266
                        for key, value in
3265
3267
                        bytes_old_client_settings.items()}
3266
3268
                    del bytes_old_client_settings
3267
 
                    # .host
 
3269
                    # .host and .checker_command
3268
3270
                    for value in old_client_settings.values():
3269
 
                        if isinstance(value["host"], bytes):
3270
 
                            value["host"] = (value["host"]
3271
 
                                             .decode("utf-8"))
 
3271
                        for attribute in ("host", "checker_command"):
 
3272
                            if isinstance(value[attribute], bytes):
 
3273
                                value[attribute] = (value[attribute]
 
3274
                                                    .decode("utf-8"))
3272
3275
            os.remove(stored_state_path)
3273
3276
        except IOError as e:
3274
3277
            if e.errno == errno.ENOENT:
3599
3602
                sys.exit(1)
3600
3603
            # End of Avahi example code
3601
3604
 
3602
 
        GLib.io_add_watch(tcp_server.fileno(), GLib.IO_IN,
3603
 
                          lambda *args, **kwargs:
3604
 
                          (tcp_server.handle_request
3605
 
                           (*args[2:], **kwargs) or True))
 
3605
        GLib.io_add_watch(
 
3606
            GLib.IOChannel.unix_new(tcp_server.fileno()),
 
3607
            GLib.PRIORITY_DEFAULT, GLib.IO_IN,
 
3608
            lambda *args, **kwargs: (tcp_server.handle_request
 
3609
                                     (*args[2:], **kwargs) or True))
3606
3610
 
3607
3611
        logger.debug("Starting main loop")
3608
3612
        main_loop.run()
3618
3622
    # Must run before the D-Bus bus name gets deregistered
3619
3623
    cleanup()
3620
3624
 
 
3625
 
 
3626
def should_only_run_tests():
 
3627
    parser = argparse.ArgumentParser(add_help=False)
 
3628
    parser.add_argument("--check", action='store_true')
 
3629
    args, unknown_args = parser.parse_known_args()
 
3630
    run_tests = args.check
 
3631
    if run_tests:
 
3632
        # Remove --check argument from sys.argv
 
3633
        sys.argv[1:] = unknown_args
 
3634
    return run_tests
 
3635
 
 
3636
# Add all tests from doctest strings
 
3637
def load_tests(loader, tests, none):
 
3638
    import doctest
 
3639
    tests.addTests(doctest.DocTestSuite())
 
3640
    return tests
3621
3641
 
3622
3642
if __name__ == '__main__':
3623
 
    main()
 
3643
    try:
 
3644
        if should_only_run_tests():
 
3645
            # Call using ./mandos --check [--verbose]
 
3646
            unittest.main()
 
3647
        else:
 
3648
            main()
 
3649
    finally:
 
3650
        logging.shutdown()