/mandos/release

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/release
24.1.116 by Björn Påhlsson
added a mandos list client program
1
#!/usr/bin/python
237.7.533 by Teddy Hogeborn
Make Emacs run tests when mandos-ctl file is saved
2
# -*- mode: python; coding: utf-8; after-save-hook: (lambda () (let ((command (if (and (boundp 'tramp-file-name-structure) (string-match (car tramp-file-name-structure) (buffer-file-name))) (tramp-file-name-localname (tramp-dissect-file-name (buffer-file-name))) (buffer-file-name)))) (if (= (shell-command (format "%s --check" (shell-quote-argument command)) "*Test*") 0) (let ((w (get-buffer-window "*Test*"))) (if w (delete-window w)) (kill-buffer "*Test*")) (display-buffer "*Test*")))); -*-
237.7.420 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
3
#
237.2.207 by Teddy Hogeborn
Update copyright year to "2010" wherever appropriate.
4
# Mandos Monitor - Control and monitor the Mandos server
237.7.420 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
5
#
237.7.517 by Teddy Hogeborn
Update copyright year to 2019
6
# Copyright © 2008-2019 Teddy Hogeborn
7
# Copyright © 2008-2019 Björn Påhlsson
237.7.420 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
8
#
237.7.455 by Teddy Hogeborn
Alter copyright notices slightly. Actual license is unchanged!
9
# This file is part of Mandos.
10
#
11
# Mandos is free software: you can redistribute it and/or modify it
12
# under the terms of the GNU General Public License as published by
237.2.207 by Teddy Hogeborn
Update copyright year to "2010" wherever appropriate.
13
# the Free Software Foundation, either version 3 of the License, or
14
# (at your option) any later version.
15
#
237.7.455 by Teddy Hogeborn
Alter copyright notices slightly. Actual license is unchanged!
16
#     Mandos is distributed in the hope that it will be useful, but
17
#     WITHOUT ANY WARRANTY; without even the implied warranty of
237.2.207 by Teddy Hogeborn
Update copyright year to "2010" wherever appropriate.
18
#     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19
#     GNU General Public License for more details.
237.7.420 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
20
#
237.2.207 by Teddy Hogeborn
Update copyright year to "2010" wherever appropriate.
21
# You should have received a copy of the GNU General Public License
237.7.455 by Teddy Hogeborn
Alter copyright notices slightly. Actual license is unchanged!
22
# along with Mandos.  If not, see <http://www.gnu.org/licenses/>.
237.7.420 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
23
#
237.11.2 by Teddy Hogeborn
Change "fukt.bsnet.se" to "recompile.se" throughout.
24
# Contact the authors at <mandos@recompile.se>.
237.7.420 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
25
#
24.1.116 by Björn Påhlsson
added a mandos list client program
26
237.8.9 by teddy at bsnet
* mandos-ctl: Use print function.
27
from __future__ import (division, absolute_import, print_function,
28
                        unicode_literals)
237.8.8 by teddy at bsnet
* mandos-ctl: Use unicode string literals.
29
237.7.266 by Teddy Hogeborn
mandos-ctl: Make it work in Python 3.
30
try:
31
    from future_builtins import *
32
except ImportError:
33
    pass
237.7.127 by Teddy Hogeborn
* mandos: Use all new builtins.
34
24.1.119 by Björn Påhlsson
Added more method support for mandos clients through mandos-ctl
35
import sys
237.7.23 by teddy at bsnet
* mandos-ctl: Use the new argparse library instead of optparse.
36
import argparse
237.2.3 by Teddy Hogeborn
Merge "mandos-list" from belorn.
37
import locale
24.1.121 by Björn Påhlsson
mandos-ctl: Added support for all client calls
38
import datetime
39
import re
24.1.163 by Björn Påhlsson
mandos-client: Added never ending loop for --connect
40
import os
237.7.156 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
41
import collections
237.7.411 by Teddy Hogeborn
mandos-ctl: Implement --dump-json option
42
import json
237.7.532 by Teddy Hogeborn
Make mandos-ctl use unittest instead of doctest module
43
import unittest
237.7.535 by Teddy Hogeborn
mandos-ctl: Use logging module instead of print() for errors
44
import logging
237.7.578 by Teddy Hogeborn
mandos-ctl: Fix bugs
45
import io
237.7.579 by Teddy Hogeborn
mandos-ctl: Refactor tests and add more tests
46
import tempfile
237.7.589 by Teddy Hogeborn
mandos-ctl: Add tests for option syntax checks
47
import contextlib
237.7.156 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
48
49
import dbus
237.2.3 by Teddy Hogeborn
Merge "mandos-list" from belorn.
50
237.7.536 by Teddy Hogeborn
mandos-ctl: Show warnings
51
# Show warnings by default
52
if not sys.warnoptions:
53
    import warnings
54
    warnings.simplefilter("default")
55
237.7.535 by Teddy Hogeborn
mandos-ctl: Use logging module instead of print() for errors
56
log = logging.getLogger(sys.argv[0])
57
logging.basicConfig(level="INFO", # Show info level messages
58
                    format="%(message)s") # Show basic log messages
59
237.7.536 by Teddy Hogeborn
mandos-ctl: Show warnings
60
logging.captureWarnings(True)   # Show warnings via the logging system
61
237.23.7 by Teddy Hogeborn
Use the .major attribute on sys.version_info instead of using "[0]".
62
if sys.version_info.major == 2:
237.7.266 by Teddy Hogeborn
mandos-ctl: Make it work in Python 3.
63
    str = unicode
237.7.633 by Teddy Hogeborn
mandos-ctl: Refactor
64
    import StringIO
65
    io.StringIO = StringIO.StringIO
237.7.266 by Teddy Hogeborn
mandos-ctl: Make it work in Python 3.
66
237.8.8 by teddy at bsnet
* mandos-ctl: Use unicode string literals.
67
locale.setlocale(locale.LC_ALL, "")
24.1.116 by Björn Påhlsson
added a mandos list client program
68
237.7.600 by Teddy Hogeborn
mandos-ctl: Refactor
69
dbus_busname_domain = "se.recompile"
70
dbus_busname = dbus_busname_domain + ".Mandos"
71
server_dbus_path = "/"
72
server_dbus_interface = dbus_busname_domain + ".Mandos"
73
client_dbus_interface = dbus_busname_domain + ".Mandos.Client"
74
del dbus_busname_domain
377 by Teddy Hogeborn
* Makefile (version): Change to 1.8.3.
75
version = "1.8.3"
24.1.118 by Björn Påhlsson
Added enable/disable
76
237.7.293 by Teddy Hogeborn
mandos-ctl: Do minor formatting and whitespace adjustments.
77
237.7.333 by Teddy Hogeborn
Support the standard org.freedesktop.DBus.ObjectManager interface.
78
try:
79
    dbus.OBJECT_MANAGER_IFACE
80
except AttributeError:
81
    dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"
82
237.7.420 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
83
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
84
def main():
85
    parser = argparse.ArgumentParser()
86
87
    add_command_line_options(parser)
88
89
    options = parser.parse_args()
90
91
    check_option_syntax(parser, options)
92
93
    clientnames = options.client
94
95
    if options.debug:
96
        log.setLevel(logging.DEBUG)
97
237.7.616 by Teddy Hogeborn
mandos-ctl: Refactor
98
    bus = dbus.SystemBus()
99
100
    mandos_dbus_object = get_mandos_dbus_object(bus)
101
102
    mandos_serv = dbus.Interface(
103
        mandos_dbus_object, dbus_interface=server_dbus_interface)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
104
    mandos_serv_object_manager = dbus.Interface(
237.7.616 by Teddy Hogeborn
mandos-ctl: Refactor
105
        mandos_dbus_object, dbus_interface=dbus.OBJECT_MANAGER_IFACE)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
106
237.7.618 by Teddy Hogeborn
mandos-ctl: Refactor
107
    managed_objects = get_managed_objects(mandos_serv_object_manager)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
108
237.7.617 by Teddy Hogeborn
mandos-ctl: Refactor
109
    all_clients = {}
110
    for path, ifs_and_props in managed_objects.items():
111
        try:
112
            all_clients[path] = ifs_and_props[client_dbus_interface]
113
        except KeyError:
114
            pass
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
115
237.7.617 by Teddy Hogeborn
mandos-ctl: Refactor
116
    # Compile dict of (clientpath: properties) to process
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
117
    if not clientnames:
237.7.613 by Teddy Hogeborn
mandos-ctl: Refactor
118
        clients = all_clients
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
119
    else:
237.7.617 by Teddy Hogeborn
mandos-ctl: Refactor
120
        clients = {}
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
121
        for name in clientnames:
237.7.613 by Teddy Hogeborn
mandos-ctl: Refactor
122
            for objpath, properties in all_clients.items():
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
123
                if properties["Name"] == name:
124
                    clients[objpath] = properties
125
                    break
126
            else:
127
                log.critical("Client not found on server: %r", name)
128
                sys.exit(1)
129
130
    commands = commands_from_options(options)
237.7.624 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
131
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
132
    for command in commands:
133
        command.run(clients, bus, mandos_serv)
134
135
136
def add_command_line_options(parser):
137
    parser.add_argument("--version", action="version",
138
                        version="%(prog)s {}".format(version),
139
                        help="show version number and exit")
140
    parser.add_argument("-a", "--all", action="store_true",
141
                        help="Select all clients")
142
    parser.add_argument("-v", "--verbose", action="store_true",
143
                        help="Print all fields")
144
    parser.add_argument("-j", "--dump-json", action="store_true",
145
                        help="Dump client data in JSON format")
146
    enable_disable = parser.add_mutually_exclusive_group()
147
    enable_disable.add_argument("-e", "--enable", action="store_true",
148
                                help="Enable client")
149
    enable_disable.add_argument("-d", "--disable",
150
                                action="store_true",
151
                                help="disable client")
152
    parser.add_argument("-b", "--bump-timeout", action="store_true",
153
                        help="Bump timeout for client")
154
    start_stop_checker = parser.add_mutually_exclusive_group()
155
    start_stop_checker.add_argument("--start-checker",
156
                                    action="store_true",
157
                                    help="Start checker for client")
158
    start_stop_checker.add_argument("--stop-checker",
159
                                    action="store_true",
160
                                    help="Stop checker for client")
161
    parser.add_argument("-V", "--is-enabled", action="store_true",
162
                        help="Check if client is enabled")
163
    parser.add_argument("-r", "--remove", action="store_true",
164
                        help="Remove client")
165
    parser.add_argument("-c", "--checker",
166
                        help="Set checker command for client")
167
    parser.add_argument("-t", "--timeout", type=string_to_delta,
168
                        help="Set timeout for client")
169
    parser.add_argument("--extended-timeout", type=string_to_delta,
170
                        help="Set extended timeout for client")
171
    parser.add_argument("-i", "--interval", type=string_to_delta,
172
                        help="Set checker interval for client")
173
    approve_deny_default = parser.add_mutually_exclusive_group()
174
    approve_deny_default.add_argument(
175
        "--approve-by-default", action="store_true",
176
        default=None, dest="approved_by_default",
177
        help="Set client to be approved by default")
178
    approve_deny_default.add_argument(
179
        "--deny-by-default", action="store_false",
180
        dest="approved_by_default",
181
        help="Set client to be denied by default")
182
    parser.add_argument("--approval-delay", type=string_to_delta,
183
                        help="Set delay before client approve/deny")
184
    parser.add_argument("--approval-duration", type=string_to_delta,
185
                        help="Set duration of one client approval")
186
    parser.add_argument("-H", "--host", help="Set host for client")
187
    parser.add_argument("-s", "--secret",
188
                        type=argparse.FileType(mode="rb"),
189
                        help="Set password blob (file) for client")
190
    approve_deny = parser.add_mutually_exclusive_group()
191
    approve_deny.add_argument(
192
        "-A", "--approve", action="store_true",
193
        help="Approve any current client request")
194
    approve_deny.add_argument("-D", "--deny", action="store_true",
195
                              help="Deny any current client request")
196
    parser.add_argument("--debug", action="store_true",
197
                        help="Debug mode (show D-Bus commands)")
198
    parser.add_argument("--check", action="store_true",
199
                        help="Run self-test")
200
    parser.add_argument("client", nargs="*", help="Client name")
201
202
203
def string_to_delta(interval):
204
    """Parse a string and return a datetime.timedelta"""
205
206
    try:
207
        return rfc3339_duration_to_delta(interval)
208
    except ValueError as e:
209
        log.warning("%s - Parsing as pre-1.6.1 interval instead",
210
                    ' '.join(e.args))
211
    return parse_pre_1_6_1_interval(interval)
24.1.121 by Björn Påhlsson
mandos-ctl: Added support for all client calls
212
237.7.156 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
213
214
def rfc3339_duration_to_delta(duration):
237.7.157 by Teddy Hogeborn
* clients.conf: Convert all time intervals to new RFC 3339 syntax.
215
    """Parse an RFC 3339 "duration" and return a datetime.timedelta
237.7.420 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
216
237.7.156 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
217
    >>> rfc3339_duration_to_delta("P7D")
218
    datetime.timedelta(7)
219
    >>> rfc3339_duration_to_delta("PT60S")
220
    datetime.timedelta(0, 60)
221
    >>> rfc3339_duration_to_delta("PT60M")
222
    datetime.timedelta(0, 3600)
237.7.538 by Teddy Hogeborn
mandos-ctl (rfc3339_duration_to_delta): Improve tests
223
    >>> rfc3339_duration_to_delta("P60M")
224
    datetime.timedelta(1680)
237.7.156 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
225
    >>> rfc3339_duration_to_delta("PT24H")
226
    datetime.timedelta(1)
227
    >>> rfc3339_duration_to_delta("P1W")
228
    datetime.timedelta(7)
229
    >>> rfc3339_duration_to_delta("PT5M30S")
230
    datetime.timedelta(0, 330)
231
    >>> rfc3339_duration_to_delta("P1DT3M20S")
232
    datetime.timedelta(1, 200)
237.7.538 by Teddy Hogeborn
mandos-ctl (rfc3339_duration_to_delta): Improve tests
233
    >>> # Can not be empty:
234
    >>> rfc3339_duration_to_delta("")
235
    Traceback (most recent call last):
236
    ...
237.7.624 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
237
    ValueError: Invalid RFC 3339 duration: ""
237.7.538 by Teddy Hogeborn
mandos-ctl (rfc3339_duration_to_delta): Improve tests
238
    >>> # Must start with "P":
239
    >>> rfc3339_duration_to_delta("1D")
240
    Traceback (most recent call last):
241
    ...
237.7.624 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
242
    ValueError: Invalid RFC 3339 duration: "1D"
237.7.538 by Teddy Hogeborn
mandos-ctl (rfc3339_duration_to_delta): Improve tests
243
    >>> # Must use correct order
244
    >>> rfc3339_duration_to_delta("PT1S2M")
245
    Traceback (most recent call last):
246
    ...
237.7.624 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
247
    ValueError: Invalid RFC 3339 duration: "PT1S2M"
237.7.538 by Teddy Hogeborn
mandos-ctl (rfc3339_duration_to_delta): Improve tests
248
    >>> # Time needs time marker
249
    >>> rfc3339_duration_to_delta("P1H2S")
250
    Traceback (most recent call last):
251
    ...
237.7.624 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
252
    ValueError: Invalid RFC 3339 duration: "P1H2S"
237.7.538 by Teddy Hogeborn
mandos-ctl (rfc3339_duration_to_delta): Improve tests
253
    >>> # Weeks can not be combined with anything else
254
    >>> rfc3339_duration_to_delta("P1D2W")
255
    Traceback (most recent call last):
256
    ...
237.7.624 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
257
    ValueError: Invalid RFC 3339 duration: "P1D2W"
237.7.538 by Teddy Hogeborn
mandos-ctl (rfc3339_duration_to_delta): Improve tests
258
    >>> rfc3339_duration_to_delta("P2W2H")
259
    Traceback (most recent call last):
260
    ...
237.7.624 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
261
    ValueError: Invalid RFC 3339 duration: "P2W2H"
237.7.156 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
262
    """
237.7.420 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
263
237.7.157 by Teddy Hogeborn
* clients.conf: Convert all time intervals to new RFC 3339 syntax.
264
    # Parsing an RFC 3339 duration with regular expressions is not
237.7.156 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
265
    # possible - there would have to be multiple places for the same
237.7.157 by Teddy Hogeborn
* clients.conf: Convert all time intervals to new RFC 3339 syntax.
266
    # values, like seconds.  The current code, while more esoteric, is
267
    # cleaner without depending on a parsing library.  If Python had a
237.7.156 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
268
    # built-in library for parsing we would use it, but we'd like to
269
    # avoid excessive use of external libraries.
237.7.420 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
270
237.7.156 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
271
    # New type for defining tokens, syntax, and semantics all-in-one
237.7.301 by Teddy Hogeborn
mandos-ctl: Generate better messages in exceptions.
272
    Token = collections.namedtuple("Token", (
273
        "regexp",  # To match token; if "value" is not None, must have
274
                   # a "group" containing digits
275
        "value",   # datetime.timedelta or None
276
        "followers"))           # Tokens valid after this token
237.7.156 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
277
    # RFC 3339 "duration" tokens, syntax, and semantics; taken from
278
    # the "duration" ABNF definition in RFC 3339, Appendix A.
279
    token_end = Token(re.compile(r"$"), None, frozenset())
280
    token_second = Token(re.compile(r"(\d+)S"),
281
                         datetime.timedelta(seconds=1),
237.7.293 by Teddy Hogeborn
mandos-ctl: Do minor formatting and whitespace adjustments.
282
                         frozenset((token_end, )))
237.7.156 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
283
    token_minute = Token(re.compile(r"(\d+)M"),
284
                         datetime.timedelta(minutes=1),
285
                         frozenset((token_second, token_end)))
286
    token_hour = Token(re.compile(r"(\d+)H"),
287
                       datetime.timedelta(hours=1),
288
                       frozenset((token_minute, token_end)))
289
    token_time = Token(re.compile(r"T"),
290
                       None,
291
                       frozenset((token_hour, token_minute,
292
                                  token_second)))
293
    token_day = Token(re.compile(r"(\d+)D"),
294
                      datetime.timedelta(days=1),
295
                      frozenset((token_time, token_end)))
296
    token_month = Token(re.compile(r"(\d+)M"),
297
                        datetime.timedelta(weeks=4),
298
                        frozenset((token_day, token_end)))
299
    token_year = Token(re.compile(r"(\d+)Y"),
300
                       datetime.timedelta(weeks=52),
301
                       frozenset((token_month, token_end)))
302
    token_week = Token(re.compile(r"(\d+)W"),
303
                       datetime.timedelta(weeks=1),
237.7.293 by Teddy Hogeborn
mandos-ctl: Do minor formatting and whitespace adjustments.
304
                       frozenset((token_end, )))
237.7.156 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
305
    token_duration = Token(re.compile(r"P"), None,
306
                           frozenset((token_year, token_month,
307
                                      token_day, token_time,
237.7.269 by Teddy Hogeborn
Fix two mutually cancelling bugs.
308
                                      token_week)))
237.7.420 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
309
    # Define starting values:
310
    # Value so far
311
    value = datetime.timedelta()
237.7.156 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
312
    found_token = None
237.7.420 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
313
    # Following valid tokens
314
    followers = frozenset((token_duration, ))
315
    # String left to parse
316
    s = duration
237.7.156 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
317
    # Loop until end token is found
318
    while found_token is not token_end:
319
        # Search for any currently valid tokens
320
        for token in followers:
321
            match = token.regexp.match(s)
322
            if match is not None:
323
                # Token found
324
                if token.value is not None:
325
                    # Value found, parse digits
326
                    factor = int(match.group(1), 10)
327
                    # Add to value so far
328
                    value += factor * token.value
329
                # Strip token from string
330
                s = token.regexp.sub("", s, 1)
331
                # Go to found token
332
                found_token = token
333
                # Set valid next tokens
334
                followers = found_token.followers
335
                break
336
        else:
337
            # No currently valid tokens were found
237.7.624 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
338
            raise ValueError("Invalid RFC 3339 duration: \"{}\""
237.7.301 by Teddy Hogeborn
mandos-ctl: Generate better messages in exceptions.
339
                             .format(duration))
237.7.156 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
340
    # End token found
341
    return value
342
343
237.7.539 by Teddy Hogeborn
mandos-ctl: Refactor and add more tests
344
def parse_pre_1_6_1_interval(interval):
237.7.549 by Teddy Hogeborn
mandos-ctl: White space changes only
345
    """Parse an interval string as documented by Mandos before 1.6.1,
346
    and return a datetime.timedelta
347
237.7.539 by Teddy Hogeborn
mandos-ctl: Refactor and add more tests
348
    >>> parse_pre_1_6_1_interval('7d')
349
    datetime.timedelta(7)
350
    >>> parse_pre_1_6_1_interval('60s')
351
    datetime.timedelta(0, 60)
352
    >>> parse_pre_1_6_1_interval('60m')
353
    datetime.timedelta(0, 3600)
354
    >>> parse_pre_1_6_1_interval('24h')
355
    datetime.timedelta(1)
356
    >>> parse_pre_1_6_1_interval('1w')
357
    datetime.timedelta(7)
358
    >>> parse_pre_1_6_1_interval('5m 30s')
359
    datetime.timedelta(0, 330)
360
    >>> parse_pre_1_6_1_interval('')
361
    datetime.timedelta(0)
362
    >>> # Ignore unknown characters, allow any order and repetitions
363
    >>> parse_pre_1_6_1_interval('2dxy7zz11y3m5m')
364
    datetime.timedelta(2, 480, 18000)
365
366
    """
237.7.420 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
367
237.7.164 by Teddy Hogeborn
* mandos-ctl (string_to_delta): Try to parse RFC 3339 duration before
368
    value = datetime.timedelta(0)
369
    regexp = re.compile(r"(\d+)([dsmhw]?)")
237.7.420 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
370
237.14.9 by Teddy Hogeborn
* mandos (ClientDBus.approval_delay, ClientDBus.approval_duration,
371
    for num, suffix in regexp.findall(interval):
372
        if suffix == "d":
373
            value += datetime.timedelta(int(num))
374
        elif suffix == "s":
375
            value += datetime.timedelta(0, int(num))
376
        elif suffix == "m":
377
            value += datetime.timedelta(0, 0, 0, 0, int(num))
378
        elif suffix == "h":
379
            value += datetime.timedelta(0, 0, 0, 0, 0, int(num))
380
        elif suffix == "w":
381
            value += datetime.timedelta(0, 0, 0, 0, 0, 0, int(num))
382
        elif suffix == "":
383
            value += datetime.timedelta(0, 0, 0, int(num))
384
    return value
24.1.121 by Björn Påhlsson
mandos-ctl: Added support for all client calls
385
237.7.293 by Teddy Hogeborn
mandos-ctl: Do minor formatting and whitespace adjustments.
386
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
387
def check_option_syntax(parser, options):
388
    """Apply additional restrictions on options, not expressible in
389
argparse"""
390
391
    def has_actions(options):
392
        return any((options.enable,
393
                    options.disable,
394
                    options.bump_timeout,
395
                    options.start_checker,
396
                    options.stop_checker,
397
                    options.is_enabled,
398
                    options.remove,
399
                    options.checker is not None,
400
                    options.timeout is not None,
401
                    options.extended_timeout is not None,
402
                    options.interval is not None,
403
                    options.approved_by_default is not None,
404
                    options.approval_delay is not None,
405
                    options.approval_duration is not None,
406
                    options.host is not None,
407
                    options.secret is not None,
408
                    options.approve,
409
                    options.deny))
410
411
    if has_actions(options) and not (options.client or options.all):
412
        parser.error("Options require clients names or --all.")
413
    if options.verbose and has_actions(options):
414
        parser.error("--verbose can only be used alone.")
415
    if options.dump_json and (options.verbose
416
                              or has_actions(options)):
417
        parser.error("--dump-json can only be used alone.")
418
    if options.all and not has_actions(options):
419
        parser.error("--all requires an action.")
420
    if options.is_enabled and len(options.client) > 1:
421
        parser.error("--is-enabled requires exactly one client")
422
    if options.remove:
423
        options.remove = False
424
        if has_actions(options) and not options.deny:
425
            parser.error("--remove can only be combined with --deny")
426
        options.remove = True
427
428
237.7.616 by Teddy Hogeborn
mandos-ctl: Refactor
429
def get_mandos_dbus_object(bus):
237.7.619 by Teddy Hogeborn
mandos-ctl: Refactor
430
    log.debug("D-Bus: Connect to: (busname=%r, path=%r)",
431
              dbus_busname, server_dbus_path)
432
    with if_dbus_exception_log_with_exception_and_exit(
433
            "Could not connect to Mandos server: %s"):
237.7.616 by Teddy Hogeborn
mandos-ctl: Refactor
434
        mandos_dbus_object = bus.get_object(dbus_busname,
435
                                            server_dbus_path)
436
    return mandos_dbus_object
437
438
237.7.619 by Teddy Hogeborn
mandos-ctl: Refactor
439
@contextlib.contextmanager
440
def if_dbus_exception_log_with_exception_and_exit(*args, **kwargs):
441
    try:
442
        yield
443
    except dbus.exceptions.DBusException as e:
444
        log.critical(*(args + (e,)), **kwargs)
445
        sys.exit(1)
446
447
237.7.618 by Teddy Hogeborn
mandos-ctl: Refactor
448
def get_managed_objects(object_manager):
449
    log.debug("D-Bus: %s:%s:%s.GetManagedObjects()", dbus_busname,
450
              server_dbus_path, dbus.OBJECT_MANAGER_IFACE)
237.7.619 by Teddy Hogeborn
mandos-ctl: Refactor
451
    with if_dbus_exception_log_with_exception_and_exit(
452
            "Failed to access Mandos server through D-Bus:\n%s"):
237.7.618 by Teddy Hogeborn
mandos-ctl: Refactor
453
        with SilenceLogger("dbus.proxies"):
454
            managed_objects = object_manager.GetManagedObjects()
455
    return managed_objects
456
457
237.7.614 by Teddy Hogeborn
mandos-ctl: Refactor
458
class SilenceLogger(object):
459
    "Simple context manager to silence a particular logger"
460
    def __init__(self, loggername):
461
        self.logger = logging.getLogger(loggername)
462
463
    def __enter__(self):
464
        self.logger.addFilter(self.nullfilter)
465
        return self
466
467
    class NullFilter(logging.Filter):
468
        def filter(self, record):
469
            return False
470
471
    nullfilter = NullFilter()
472
473
    def __exit__(self, exc_type, exc_val, exc_tb):
474
        self.logger.removeFilter(self.nullfilter)
475
476
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
477
def commands_from_options(options):
478
479
    commands = []
480
481
    if options.is_enabled:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
482
        commands.append(command.IsEnabled())
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
483
484
    if options.approve:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
485
        commands.append(command.Approve())
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
486
487
    if options.deny:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
488
        commands.append(command.Deny())
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
489
490
    if options.remove:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
491
        commands.append(command.Remove())
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
492
493
    if options.dump_json:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
494
        commands.append(command.DumpJSON())
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
495
496
    if options.enable:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
497
        commands.append(command.Enable())
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
498
499
    if options.disable:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
500
        commands.append(command.Disable())
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
501
502
    if options.bump_timeout:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
503
        commands.append(command.BumpTimeout())
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
504
505
    if options.start_checker:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
506
        commands.append(command.StartChecker())
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
507
508
    if options.stop_checker:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
509
        commands.append(command.StopChecker())
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
510
511
    if options.approved_by_default is not None:
512
        if options.approved_by_default:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
513
            commands.append(command.ApproveByDefault())
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
514
        else:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
515
            commands.append(command.DenyByDefault())
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
516
517
    if options.checker is not None:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
518
        commands.append(command.SetChecker(options.checker))
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
519
520
    if options.host is not None:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
521
        commands.append(command.SetHost(options.host))
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
522
523
    if options.secret is not None:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
524
        commands.append(command.SetSecret(options.secret))
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
525
526
    if options.timeout is not None:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
527
        commands.append(command.SetTimeout(options.timeout))
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
528
529
    if options.extended_timeout:
530
        commands.append(
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
531
            command.SetExtendedTimeout(options.extended_timeout))
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
532
533
    if options.interval is not None:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
534
        commands.append(command.SetInterval(options.interval))
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
535
536
    if options.approval_delay is not None:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
537
        commands.append(
538
            command.SetApprovalDelay(options.approval_delay))
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
539
540
    if options.approval_duration is not None:
541
        commands.append(
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
542
            command.SetApprovalDuration(options.approval_duration))
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
543
544
    # If no command option has been given, show table of clients,
545
    # optionally verbosely
546
    if not commands:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
547
        commands.append(command.PrintTable(verbose=options.verbose))
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
548
549
    return commands
550
551
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
552
class command(object):
553
    """A namespace for command classes"""
554
555
    class Base(object):
556
        """Abstract base class for commands"""
557
        def run(self, clients, bus=None, mandos=None):
558
            """Normal commands should implement run_on_one_client(),
559
but commands which want to operate on all clients at the same time can
560
override this run() method instead.
561
"""
562
            self.mandos = mandos
563
            for clientpath, properties in clients.items():
564
                log.debug("D-Bus: Connect to: (busname=%r, path=%r)",
565
                          dbus_busname, str(clientpath))
566
                client = bus.get_object(dbus_busname, clientpath)
567
                self.run_on_one_client(client, properties)
568
569
570
    class IsEnabled(Base):
571
        def run(self, clients, bus=None, mandos=None):
572
            client, properties = next(iter(clients.items()))
573
            if self.is_enabled(client, properties):
574
                sys.exit(0)
575
            sys.exit(1)
576
        def is_enabled(self, client, properties):
577
            return properties["Enabled"]
578
579
580
    class Approve(Base):
581
        def run_on_one_client(self, client, properties):
582
            log.debug("D-Bus: %s:%s:%s.Approve(True)", dbus_busname,
583
                      client.__dbus_object_path__,
584
                      client_dbus_interface)
585
            client.Approve(dbus.Boolean(True),
586
                           dbus_interface=client_dbus_interface)
587
588
589
    class Deny(Base):
590
        def run_on_one_client(self, client, properties):
591
            log.debug("D-Bus: %s:%s:%s.Approve(False)", dbus_busname,
592
                      client.__dbus_object_path__,
593
                      client_dbus_interface)
594
            client.Approve(dbus.Boolean(False),
595
                           dbus_interface=client_dbus_interface)
596
597
598
    class Remove(Base):
599
        def run_on_one_client(self, client, properties):
600
            log.debug("D-Bus: %s:%s:%s.RemoveClient(%r)",
601
                      dbus_busname, server_dbus_path,
602
                      server_dbus_interface,
603
                      str(client.__dbus_object_path__))
604
            self.mandos.RemoveClient(client.__dbus_object_path__)
605
606
607
    class Output(Base):
608
        """Abstract class for commands outputting client details"""
609
        all_keywords = ("Name", "Enabled", "Timeout", "LastCheckedOK",
610
                        "Created", "Interval", "Host", "KeyID",
611
                        "Fingerprint", "CheckerRunning",
612
                        "LastEnabled", "ApprovalPending",
613
                        "ApprovedByDefault", "LastApprovalRequest",
614
                        "ApprovalDelay", "ApprovalDuration",
615
                        "Checker", "ExtendedTimeout", "Expires",
616
                        "LastCheckerStatus")
617
618
619
    class DumpJSON(Output):
237.7.633 by Teddy Hogeborn
mandos-ctl: Refactor
620
        def run(self, clients, bus=None, mandos=None):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
621
            data = {client["Name"]:
622
                    {key: self.dbus_boolean_to_bool(client[key])
623
                     for key in self.all_keywords}
237.7.633 by Teddy Hogeborn
mandos-ctl: Refactor
624
                    for client in clients.values()}
625
            print(json.dumps(data, indent=4, separators=(',', ': ')))
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
626
627
        @staticmethod
628
        def dbus_boolean_to_bool(value):
629
            if isinstance(value, dbus.Boolean):
630
                value = bool(value)
631
            return value
632
633
634
    class PrintTable(Output):
635
        def __init__(self, verbose=False):
636
            self.verbose = verbose
637
237.7.633 by Teddy Hogeborn
mandos-ctl: Refactor
638
        def run(self, clients, bus=None, mandos=None):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
639
            default_keywords = ("Name", "Enabled", "Timeout",
640
                                "LastCheckedOK")
641
            keywords = default_keywords
642
            if self.verbose:
643
                keywords = self.all_keywords
237.7.633 by Teddy Hogeborn
mandos-ctl: Refactor
644
            print(self.TableOfClients(clients.values(), keywords))
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
645
646
        class TableOfClients(object):
647
            tableheaders = {
648
                "Name": "Name",
649
                "Enabled": "Enabled",
650
                "Timeout": "Timeout",
651
                "LastCheckedOK": "Last Successful Check",
652
                "LastApprovalRequest": "Last Approval Request",
653
                "Created": "Created",
654
                "Interval": "Interval",
655
                "Host": "Host",
656
                "Fingerprint": "Fingerprint",
657
                "KeyID": "Key ID",
658
                "CheckerRunning": "Check Is Running",
659
                "LastEnabled": "Last Enabled",
660
                "ApprovalPending": "Approval Is Pending",
661
                "ApprovedByDefault": "Approved By Default",
662
                "ApprovalDelay": "Approval Delay",
663
                "ApprovalDuration": "Approval Duration",
664
                "Checker": "Checker",
665
                "ExtendedTimeout": "Extended Timeout",
666
                "Expires": "Expires",
667
                "LastCheckerStatus": "Last Checker Status",
668
            }
669
670
            def __init__(self, clients, keywords):
671
                self.clients = clients
672
                self.keywords = keywords
673
237.7.559 by Teddy Hogeborn
mandos-ctl: Refactor; move TableOfClients into PrintTableCmd
674
            def __str__(self):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
675
                return "\n".join(self.rows())
676
677
            if sys.version_info.major == 2:
678
                __unicode__ = __str__
679
                def __str__(self):
680
                    return str(self).encode(
681
                        locale.getpreferredencoding())
682
683
            def rows(self):
684
                format_string = self.row_formatting_string()
685
                rows = [self.header_line(format_string)]
686
                rows.extend(self.client_line(client, format_string)
687
                            for client in self.clients)
688
                return rows
689
690
            def row_formatting_string(self):
691
                "Format string used to format table rows"
692
                return " ".join("{{{key}:{width}}}".format(
693
                    width=max(len(self.tableheaders[key]),
694
                              *(len(self.string_from_client(client,
695
                                                            key))
696
                                for client in self.clients)),
697
                    key=key)
698
                                for key in self.keywords)
699
700
            def string_from_client(self, client, key):
701
                return self.valuetostring(client[key], key)
702
703
            @classmethod
704
            def valuetostring(cls, value, keyword):
705
                if isinstance(value, dbus.Boolean):
706
                    return "Yes" if value else "No"
707
                if keyword in ("Timeout", "Interval", "ApprovalDelay",
708
                               "ApprovalDuration", "ExtendedTimeout"):
709
                    return cls.milliseconds_to_string(value)
710
                return str(value)
711
712
            def header_line(self, format_string):
713
                return format_string.format(**self.tableheaders)
714
715
            def client_line(self, client, format_string):
716
                return format_string.format(
717
                    **{key: self.string_from_client(client, key)
718
                       for key in self.keywords})
719
720
            @staticmethod
721
            def milliseconds_to_string(ms):
722
                td = datetime.timedelta(0, 0, 0, ms)
723
                return ("{days}{hours:02}:{minutes:02}:{seconds:02}"
724
                        .format(days="{}T".format(td.days)
725
                                if td.days else "",
726
                                hours=td.seconds // 3600,
727
                                minutes=(td.seconds % 3600) // 60,
728
                                seconds=td.seconds % 60))
729
730
731
    class Property(Base):
732
        "Abstract class for Actions for setting one client property"
733
734
        def run_on_one_client(self, client, properties):
735
            """Set the Client's D-Bus property"""
736
            log.debug("D-Bus: %s:%s:%s.Set(%r, %r, %r)", dbus_busname,
737
                      client.__dbus_object_path__,
738
                      dbus.PROPERTIES_IFACE, client_dbus_interface,
739
                      self.propname, self.value_to_set
740
                      if not isinstance(self.value_to_set,
741
                                        dbus.Boolean)
742
                      else bool(self.value_to_set))
743
            client.Set(client_dbus_interface, self.propname,
744
                       self.value_to_set,
745
                       dbus_interface=dbus.PROPERTIES_IFACE)
746
747
        @property
748
        def propname(self):
749
            raise NotImplementedError()
750
751
752
    class Enable(Property):
753
        propname = "Enabled"
754
        value_to_set = dbus.Boolean(True)
755
756
757
    class Disable(Property):
758
        propname = "Enabled"
759
        value_to_set = dbus.Boolean(False)
760
761
762
    class BumpTimeout(Property):
763
        propname = "LastCheckedOK"
764
        value_to_set = ""
765
766
767
    class StartChecker(Property):
768
        propname = "CheckerRunning"
769
        value_to_set = dbus.Boolean(True)
770
771
772
    class StopChecker(Property):
773
        propname = "CheckerRunning"
774
        value_to_set = dbus.Boolean(False)
775
776
777
    class ApproveByDefault(Property):
778
        propname = "ApprovedByDefault"
779
        value_to_set = dbus.Boolean(True)
780
781
782
    class DenyByDefault(Property):
783
        propname = "ApprovedByDefault"
784
        value_to_set = dbus.Boolean(False)
785
786
787
    class PropertyValue(Property):
788
        "Abstract class for Property recieving a value as argument"
789
        def __init__(self, value):
790
            self.value_to_set = value
791
792
793
    class SetChecker(PropertyValue):
794
        propname = "Checker"
795
796
797
    class SetHost(PropertyValue):
798
        propname = "Host"
799
800
801
    class SetSecret(PropertyValue):
802
        propname = "Secret"
803
804
        @property
805
        def value_to_set(self):
806
            return self._vts
807
808
        @value_to_set.setter
809
        def value_to_set(self, value):
810
            """When setting, read data from supplied file object"""
811
            self._vts = value.read()
812
            value.close()
813
814
815
    class MillisecondsPropertyValueArgument(PropertyValue):
816
        """Abstract class for PropertyValue taking a value argument as
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
817
a datetime.timedelta() but should store it as milliseconds."""
237.7.609 by Teddy Hogeborn
mandos-ctl: White space changes only
818
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
819
        @property
820
        def value_to_set(self):
821
            return self._vts
822
823
        @value_to_set.setter
824
        def value_to_set(self, value):
825
            "When setting, convert value from a datetime.timedelta"
826
            self._vts = int(round(value.total_seconds() * 1000))
827
828
829
    class SetTimeout(MillisecondsPropertyValueArgument):
830
        propname = "Timeout"
831
832
833
    class SetExtendedTimeout(MillisecondsPropertyValueArgument):
834
        propname = "ExtendedTimeout"
835
836
837
    class SetInterval(MillisecondsPropertyValueArgument):
838
        propname = "Interval"
839
840
841
    class SetApprovalDelay(MillisecondsPropertyValueArgument):
842
        propname = "ApprovalDelay"
843
844
845
    class SetApprovalDuration(MillisecondsPropertyValueArgument):
846
        propname = "ApprovalDuration"
237.7.551 by Teddy Hogeborn
mandos-ctl: Separate determining what to do and actually doing it
847
24.1.163 by Björn Påhlsson
mandos-client: Added never ending loop for --connect
848
237.7.532 by Teddy Hogeborn
Make mandos-ctl use unittest instead of doctest module
849

237.7.620 by Teddy Hogeborn
mandos-ctl: Refactor tests
850
class TestCaseWithAssertLogs(unittest.TestCase):
851
    """unittest.TestCase.assertLogs only exists in Python 3.4"""
852
853
    if not hasattr(unittest.TestCase, "assertLogs"):
854
        @contextlib.contextmanager
855
        def assertLogs(self, logger, level=logging.INFO):
856
            capturing_handler = self.CapturingLevelHandler(level)
857
            old_level = logger.level
858
            old_propagate = logger.propagate
859
            logger.addHandler(capturing_handler)
860
            logger.setLevel(level)
861
            logger.propagate = False
862
            try:
863
                yield capturing_handler.watcher
864
            finally:
865
                logger.propagate = old_propagate
866
                logger.removeHandler(capturing_handler)
867
                logger.setLevel(old_level)
868
            self.assertGreater(len(capturing_handler.watcher.records),
869
                               0)
870
871
        class CapturingLevelHandler(logging.Handler):
872
            def __init__(self, level, *args, **kwargs):
873
                logging.Handler.__init__(self, *args, **kwargs)
874
                self.watcher = self.LoggingWatcher([], [])
875
            def emit(self, record):
876
                self.watcher.records.append(record)
877
                self.watcher.output.append(self.format(record))
878
879
            LoggingWatcher = collections.namedtuple("LoggingWatcher",
880
                                                    ("records",
881
                                                     "output"))
882
237.7.624 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
883
237.7.620 by Teddy Hogeborn
mandos-ctl: Refactor tests
884
class Test_string_to_delta(TestCaseWithAssertLogs):
237.7.628 by Teddy Hogeborn
mandos-ctl: Refactor tests
885
    # Just test basic RFC 3339 functionality here, the doc string for
886
    # rfc3339_duration_to_delta() already has more comprehensive
887
    # tests, which is run by doctest.
888
889
    def test_rfc3339_zero_seconds(self):
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
890
        self.assertEqual(datetime.timedelta(),
891
                         string_to_delta("PT0S"))
237.7.628 by Teddy Hogeborn
mandos-ctl: Refactor tests
892
893
    def test_rfc3339_zero_days(self):
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
894
        self.assertEqual(datetime.timedelta(), string_to_delta("P0D"))
237.7.628 by Teddy Hogeborn
mandos-ctl: Refactor tests
895
896
    def test_rfc3339_one_second(self):
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
897
        self.assertEqual(datetime.timedelta(0, 1),
898
                         string_to_delta("PT1S"))
237.7.628 by Teddy Hogeborn
mandos-ctl: Refactor tests
899
900
    def test_rfc3339_two_hours(self):
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
901
        self.assertEqual(datetime.timedelta(0, 7200),
902
                         string_to_delta("PT2H"))
237.7.609 by Teddy Hogeborn
mandos-ctl: White space changes only
903
237.7.540 by Teddy Hogeborn
mandos-ctl: Add more tests
904
    def test_falls_back_to_pre_1_6_1_with_warning(self):
237.7.620 by Teddy Hogeborn
mandos-ctl: Refactor tests
905
        with self.assertLogs(log, logging.WARNING):
906
            value = string_to_delta("2h")
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
907
        self.assertEqual(datetime.timedelta(0, 7200), value)
237.7.540 by Teddy Hogeborn
mandos-ctl: Add more tests
908
237.7.558 by Teddy Hogeborn
mandos-ctl: Refactor; test PrintTableCmd instead of TableOfClients
909
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
910
class Test_check_option_syntax(unittest.TestCase):
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
911
    def setUp(self):
912
        self.parser = argparse.ArgumentParser()
913
        add_command_line_options(self.parser)
914
915
    def test_actions_requires_client_or_all(self):
916
        for action, value in self.actions.items():
917
            options = self.parser.parse_args()
918
            setattr(options, action, value)
919
            with self.assertParseError():
920
                self.check_option_syntax(options)
921
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
922
    # This mostly corresponds to the definition from has_actions() in
923
    # check_option_syntax()
924
    actions = {
925
        # The actual values set here are not that important, but we do
926
        # at least stick to the correct types, even though they are
927
        # never used
928
        "enable": True,
929
        "disable": True,
930
        "bump_timeout": True,
931
        "start_checker": True,
932
        "stop_checker": True,
933
        "is_enabled": True,
934
        "remove": True,
935
        "checker": "x",
936
        "timeout": datetime.timedelta(),
937
        "extended_timeout": datetime.timedelta(),
938
        "interval": datetime.timedelta(),
939
        "approved_by_default": True,
940
        "approval_delay": datetime.timedelta(),
941
        "approval_duration": datetime.timedelta(),
942
        "host": "x",
943
        "secret": io.BytesIO(b"x"),
944
        "approve": True,
945
        "deny": True,
946
    }
947
948
    @contextlib.contextmanager
949
    def assertParseError(self):
950
        with self.assertRaises(SystemExit) as e:
237.7.624 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
951
            with self.redirect_stderr_to_devnull():
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
952
                yield
953
        # Exit code from argparse is guaranteed to be "2".  Reference:
954
        # https://docs.python.org/3/library
955
        # /argparse.html#exiting-methods
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
956
        self.assertEqual(2, e.exception.code)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
957
958
    @staticmethod
959
    @contextlib.contextmanager
237.7.624 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
960
    def redirect_stderr_to_devnull():
237.7.631 by Teddy Hogeborn
mandos-ctl: Refactor tests
961
        old_stderr = sys.stderr
962
        with contextlib.closing(open(os.devnull, "w")) as null:
963
            sys.stderr = null
964
            try:
965
                yield
966
            finally:
967
                sys.stderr = old_stderr
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
968
969
    def check_option_syntax(self, options):
970
        check_option_syntax(self.parser, options)
971
237.7.623 by Teddy Hogeborn
mandos-ctl: Refactor and fix bug in tests.
972
    def test_actions_all_conflicts_with_verbose(self):
973
        for action, value in self.actions.items():
974
            options = self.parser.parse_args()
975
            setattr(options, action, value)
976
            options.all = True
977
            options.verbose = True
978
            with self.assertParseError():
979
                self.check_option_syntax(options)
980
981
    def test_actions_with_client_conflicts_with_verbose(self):
982
        for action, value in self.actions.items():
983
            options = self.parser.parse_args()
984
            setattr(options, action, value)
985
            options.verbose = True
986
            options.client = ["foo"]
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
987
            with self.assertParseError():
988
                self.check_option_syntax(options)
989
990
    def test_dump_json_conflicts_with_verbose(self):
991
        options = self.parser.parse_args()
992
        options.dump_json = True
993
        options.verbose = True
994
        with self.assertParseError():
995
            self.check_option_syntax(options)
996
997
    def test_dump_json_conflicts_with_action(self):
998
        for action, value in self.actions.items():
999
            options = self.parser.parse_args()
1000
            setattr(options, action, value)
1001
            options.dump_json = True
1002
            with self.assertParseError():
1003
                self.check_option_syntax(options)
1004
1005
    def test_all_can_not_be_alone(self):
1006
        options = self.parser.parse_args()
1007
        options.all = True
1008
        with self.assertParseError():
1009
            self.check_option_syntax(options)
1010
1011
    def test_all_is_ok_with_any_action(self):
1012
        for action, value in self.actions.items():
1013
            options = self.parser.parse_args()
1014
            setattr(options, action, value)
1015
            options.all = True
1016
            self.check_option_syntax(options)
1017
237.7.623 by Teddy Hogeborn
mandos-ctl: Refactor and fix bug in tests.
1018
    def test_any_action_is_ok_with_one_client(self):
1019
        for action, value in self.actions.items():
1020
            options = self.parser.parse_args()
1021
            setattr(options, action, value)
1022
            options.client = ["foo"]
1023
            self.check_option_syntax(options)
1024
237.7.624 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
1025
    def test_one_client_with_all_actions_except_is_enabled(self):
1026
        options = self.parser.parse_args()
1027
        for action, value in self.actions.items():
1028
            if action == "is_enabled":
1029
                continue
1030
            setattr(options, action, value)
1031
        options.client = ["foo"]
1032
        self.check_option_syntax(options)
1033
1034
    def test_two_clients_with_all_actions_except_is_enabled(self):
1035
        options = self.parser.parse_args()
1036
        for action, value in self.actions.items():
1037
            if action == "is_enabled":
1038
                continue
1039
            setattr(options, action, value)
1040
        options.client = ["foo", "barbar"]
1041
        self.check_option_syntax(options)
1042
1043
    def test_two_clients_are_ok_with_actions_except_is_enabled(self):
237.7.623 by Teddy Hogeborn
mandos-ctl: Refactor and fix bug in tests.
1044
        for action, value in self.actions.items():
1045
            if action == "is_enabled":
1046
                continue
1047
            options = self.parser.parse_args()
1048
            setattr(options, action, value)
1049
            options.client = ["foo", "barbar"]
1050
            self.check_option_syntax(options)
1051
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1052
    def test_is_enabled_fails_without_client(self):
1053
        options = self.parser.parse_args()
1054
        options.is_enabled = True
1055
        with self.assertParseError():
1056
            self.check_option_syntax(options)
1057
1058
    def test_is_enabled_fails_with_two_clients(self):
1059
        options = self.parser.parse_args()
1060
        options.is_enabled = True
1061
        options.client = ["foo", "barbar"]
1062
        with self.assertParseError():
1063
            self.check_option_syntax(options)
1064
1065
    def test_remove_can_only_be_combined_with_action_deny(self):
1066
        for action, value in self.actions.items():
1067
            if action in {"remove", "deny"}:
1068
                continue
1069
            options = self.parser.parse_args()
1070
            setattr(options, action, value)
1071
            options.all = True
1072
            options.remove = True
1073
            with self.assertParseError():
1074
                self.check_option_syntax(options)
1075
1076
237.7.620 by Teddy Hogeborn
mandos-ctl: Refactor tests
1077
class Test_get_mandos_dbus_object(TestCaseWithAssertLogs):
237.7.616 by Teddy Hogeborn
mandos-ctl: Refactor
1078
    def test_calls_and_returns_get_object_on_bus(self):
1079
        class MockBus(object):
1080
            called = False
1081
            def get_object(mockbus_self, busname, dbus_path):
1082
                # Note that "self" is still the testcase instance,
1083
                # this MockBus instance is in "mockbus_self".
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
1084
                self.assertEqual(dbus_busname, busname)
1085
                self.assertEqual(server_dbus_path, dbus_path)
237.7.616 by Teddy Hogeborn
mandos-ctl: Refactor
1086
                mockbus_self.called = True
1087
                return mockbus_self
1088
1089
        mockbus = get_mandos_dbus_object(bus=MockBus())
1090
        self.assertIsInstance(mockbus, MockBus)
1091
        self.assertTrue(mockbus.called)
1092
1093
    def test_logs_and_exits_on_dbus_error(self):
1094
        class MockBusFailing(object):
1095
            def get_object(self, busname, dbus_path):
1096
                raise dbus.exceptions.DBusException("Test")
1097
237.7.620 by Teddy Hogeborn
mandos-ctl: Refactor tests
1098
        with self.assertLogs(log, logging.CRITICAL):
1099
            with self.assertRaises(SystemExit) as e:
1100
                bus = get_mandos_dbus_object(bus=MockBusFailing())
1101
237.7.616 by Teddy Hogeborn
mandos-ctl: Refactor
1102
        if isinstance(e.exception.code, int):
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
1103
            self.assertNotEqual(0, e.exception.code)
237.7.616 by Teddy Hogeborn
mandos-ctl: Refactor
1104
        else:
1105
            self.assertIsNotNone(e.exception.code)
1106
237.7.620 by Teddy Hogeborn
mandos-ctl: Refactor tests
1107
1108
class Test_get_managed_objects(TestCaseWithAssertLogs):
237.7.618 by Teddy Hogeborn
mandos-ctl: Refactor
1109
    def test_calls_and_returns_GetManagedObjects(self):
1110
        managed_objects = {"/clients/foo": { "Name": "foo"}}
1111
        class MockObjectManager(object):
237.7.621 by Teddy Hogeborn
mandos-ctl: Refactor tests
1112
            def GetManagedObjects(self):
237.7.618 by Teddy Hogeborn
mandos-ctl: Refactor
1113
                return managed_objects
1114
        retval = get_managed_objects(MockObjectManager())
1115
        self.assertDictEqual(managed_objects, retval)
1116
1117
    def test_logs_and_exits_on_dbus_error(self):
237.7.620 by Teddy Hogeborn
mandos-ctl: Refactor tests
1118
        dbus_logger = logging.getLogger("dbus.proxies")
1119
237.7.618 by Teddy Hogeborn
mandos-ctl: Refactor
1120
        class MockObjectManagerFailing(object):
237.7.621 by Teddy Hogeborn
mandos-ctl: Refactor tests
1121
            def GetManagedObjects(self):
237.7.620 by Teddy Hogeborn
mandos-ctl: Refactor tests
1122
                dbus_logger.error("Test")
237.7.618 by Teddy Hogeborn
mandos-ctl: Refactor
1123
                raise dbus.exceptions.DBusException("Test")
1124
237.7.620 by Teddy Hogeborn
mandos-ctl: Refactor tests
1125
        class CountingHandler(logging.Handler):
1126
            count = 0
1127
            def emit(self, record):
1128
                self.count += 1
1129
1130
        counting_handler = CountingHandler()
1131
1132
        dbus_logger.addHandler(counting_handler)
1133
1134
        try:
1135
            with self.assertLogs(log, logging.CRITICAL) as watcher:
237.7.618 by Teddy Hogeborn
mandos-ctl: Refactor
1136
                with self.assertRaises(SystemExit) as e:
1137
                    get_managed_objects(MockObjectManagerFailing())
237.7.620 by Teddy Hogeborn
mandos-ctl: Refactor tests
1138
        finally:
1139
            dbus_logger.removeFilter(counting_handler)
237.7.622 by Teddy Hogeborn
mandos-ctl: Add comment
1140
1141
        # Make sure the dbus logger was suppressed
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
1142
        self.assertEqual(0, counting_handler.count)
237.7.620 by Teddy Hogeborn
mandos-ctl: Refactor tests
1143
1144
        # Test that the dbus_logger still works
1145
        with self.assertLogs(dbus_logger, logging.ERROR):
1146
            dbus_logger.error("Test")
1147
237.7.618 by Teddy Hogeborn
mandos-ctl: Refactor
1148
        if isinstance(e.exception.code, int):
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
1149
            self.assertNotEqual(0, e.exception.code)
237.7.618 by Teddy Hogeborn
mandos-ctl: Refactor
1150
        else:
1151
            self.assertIsNotNone(e.exception.code)
1152
237.7.614 by Teddy Hogeborn
mandos-ctl: Refactor
1153
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1154
class Test_commands_from_options(unittest.TestCase):
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1155
    def setUp(self):
1156
        self.parser = argparse.ArgumentParser()
1157
        add_command_line_options(self.parser)
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1158
1159
    def test_is_enabled(self):
1160
        self.assert_command_from_args(["--is-enabled", "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1161
                                      command.IsEnabled)
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1162
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1163
    def assert_command_from_args(self, args, command_cls,
1164
                                 **cmd_attrs):
1165
        """Assert that parsing ARGS should result in an instance of
1166
COMMAND_CLS with (optionally) all supplied attributes (CMD_ATTRS)."""
1167
        options = self.parser.parse_args(args)
1168
        check_option_syntax(self.parser, options)
1169
        commands = commands_from_options(options)
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
1170
        self.assertEqual(1, len(commands))
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1171
        command = commands[0]
1172
        self.assertIsInstance(command, command_cls)
1173
        for key, value in cmd_attrs.items():
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
1174
            self.assertEqual(value, getattr(command, key))
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1175
1176
    def test_is_enabled_short(self):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1177
        self.assert_command_from_args(["-V", "foo"],
1178
                                      command.IsEnabled)
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1179
1180
    def test_approve(self):
1181
        self.assert_command_from_args(["--approve", "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1182
                                      command.Approve)
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1183
1184
    def test_approve_short(self):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1185
        self.assert_command_from_args(["-A", "foo"], command.Approve)
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1186
1187
    def test_deny(self):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1188
        self.assert_command_from_args(["--deny", "foo"], command.Deny)
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1189
1190
    def test_deny_short(self):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1191
        self.assert_command_from_args(["-D", "foo"], command.Deny)
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1192
1193
    def test_remove(self):
1194
        self.assert_command_from_args(["--remove", "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1195
                                      command.Remove)
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1196
1197
    def test_deny_before_remove(self):
1198
        options = self.parser.parse_args(["--deny", "--remove",
1199
                                          "foo"])
1200
        check_option_syntax(self.parser, options)
1201
        commands = commands_from_options(options)
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
1202
        self.assertEqual(2, len(commands))
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1203
        self.assertIsInstance(commands[0], command.Deny)
1204
        self.assertIsInstance(commands[1], command.Remove)
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1205
1206
    def test_deny_before_remove_reversed(self):
1207
        options = self.parser.parse_args(["--remove", "--deny",
1208
                                          "--all"])
1209
        check_option_syntax(self.parser, options)
1210
        commands = commands_from_options(options)
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
1211
        self.assertEqual(2, len(commands))
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1212
        self.assertIsInstance(commands[0], command.Deny)
1213
        self.assertIsInstance(commands[1], command.Remove)
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1214
1215
    def test_remove_short(self):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1216
        self.assert_command_from_args(["-r", "foo"], command.Remove)
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1217
1218
    def test_dump_json(self):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1219
        self.assert_command_from_args(["--dump-json"],
1220
                                      command.DumpJSON)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1221
1222
    def test_enable(self):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1223
        self.assert_command_from_args(["--enable", "foo"],
1224
                                      command.Enable)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1225
1226
    def test_enable_short(self):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1227
        self.assert_command_from_args(["-e", "foo"], command.Enable)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1228
1229
    def test_disable(self):
1230
        self.assert_command_from_args(["--disable", "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1231
                                      command.Disable)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1232
1233
    def test_disable_short(self):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1234
        self.assert_command_from_args(["-d", "foo"], command.Disable)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1235
1236
    def test_bump_timeout(self):
1237
        self.assert_command_from_args(["--bump-timeout", "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1238
                                      command.BumpTimeout)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1239
1240
    def test_bump_timeout_short(self):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1241
        self.assert_command_from_args(["-b", "foo"],
1242
                                      command.BumpTimeout)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1243
1244
    def test_start_checker(self):
1245
        self.assert_command_from_args(["--start-checker", "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1246
                                      command.StartChecker)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1247
1248
    def test_stop_checker(self):
1249
        self.assert_command_from_args(["--stop-checker", "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1250
                                      command.StopChecker)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1251
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1252
    def test_approve_by_default(self):
1253
        self.assert_command_from_args(["--approve-by-default", "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1254
                                      command.ApproveByDefault)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1255
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1256
    def test_deny_by_default(self):
1257
        self.assert_command_from_args(["--deny-by-default", "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1258
                                      command.DenyByDefault)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1259
1260
    def test_checker(self):
1261
        self.assert_command_from_args(["--checker", ":", "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1262
                                      command.SetChecker,
1263
                                      value_to_set=":")
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1264
1265
    def test_checker_empty(self):
1266
        self.assert_command_from_args(["--checker", "", "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1267
                                      command.SetChecker,
1268
                                      value_to_set="")
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1269
1270
    def test_checker_short(self):
1271
        self.assert_command_from_args(["-c", ":", "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1272
                                      command.SetChecker,
1273
                                      value_to_set=":")
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1274
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1275
    def test_host(self):
1276
        self.assert_command_from_args(["--host", "foo.example.org",
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1277
                                       "foo"], command.SetHost,
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1278
                                      value_to_set="foo.example.org")
1279
1280
    def test_host_short(self):
1281
        self.assert_command_from_args(["-H", "foo.example.org",
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1282
                                       "foo"], command.SetHost,
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1283
                                      value_to_set="foo.example.org")
1284
1285
    def test_secret_devnull(self):
1286
        self.assert_command_from_args(["--secret", os.path.devnull,
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1287
                                       "foo"], command.SetSecret,
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1288
                                      value_to_set=b"")
1289
1290
    def test_secret_tempfile(self):
1291
        with tempfile.NamedTemporaryFile(mode="r+b") as f:
1292
            value = b"secret\0xyzzy\nbar"
1293
            f.write(value)
1294
            f.seek(0)
1295
            self.assert_command_from_args(["--secret", f.name,
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1296
                                           "foo"], command.SetSecret,
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1297
                                          value_to_set=value)
1298
1299
    def test_secret_devnull_short(self):
1300
        self.assert_command_from_args(["-s", os.path.devnull, "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1301
                                      command.SetSecret,
1302
                                      value_to_set=b"")
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1303
1304
    def test_secret_tempfile_short(self):
1305
        with tempfile.NamedTemporaryFile(mode="r+b") as f:
1306
            value = b"secret\0xyzzy\nbar"
1307
            f.write(value)
1308
            f.seek(0)
1309
            self.assert_command_from_args(["-s", f.name, "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1310
                                          command.SetSecret,
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1311
                                          value_to_set=value)
1312
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1313
    def test_timeout(self):
1314
        self.assert_command_from_args(["--timeout", "PT5M", "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1315
                                      command.SetTimeout,
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1316
                                      value_to_set=300000)
1317
1318
    def test_timeout_short(self):
1319
        self.assert_command_from_args(["-t", "PT5M", "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1320
                                      command.SetTimeout,
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1321
                                      value_to_set=300000)
1322
1323
    def test_extended_timeout(self):
1324
        self.assert_command_from_args(["--extended-timeout", "PT15M",
1325
                                       "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1326
                                      command.SetExtendedTimeout,
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1327
                                      value_to_set=900000)
1328
1329
    def test_interval(self):
1330
        self.assert_command_from_args(["--interval", "PT2M", "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1331
                                      command.SetInterval,
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1332
                                      value_to_set=120000)
1333
1334
    def test_interval_short(self):
1335
        self.assert_command_from_args(["-i", "PT2M", "foo"],
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1336
                                      command.SetInterval,
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1337
                                      value_to_set=120000)
1338
1339
    def test_approval_delay(self):
1340
        self.assert_command_from_args(["--approval-delay", "PT30S",
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1341
                                       "foo"],
1342
                                      command.SetApprovalDelay,
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1343
                                      value_to_set=30000)
1344
1345
    def test_approval_duration(self):
1346
        self.assert_command_from_args(["--approval-duration", "PT1S",
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1347
                                       "foo"],
1348
                                      command.SetApprovalDuration,
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1349
                                      value_to_set=1000)
1350
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1351
    def test_print_table(self):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1352
        self.assert_command_from_args([], command.PrintTable,
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1353
                                      verbose=False)
1354
1355
    def test_print_table_verbose(self):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1356
        self.assert_command_from_args(["--verbose"],
1357
                                      command.PrintTable,
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1358
                                      verbose=True)
1359
1360
    def test_print_table_verbose_short(self):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1361
        self.assert_command_from_args(["-v"], command.PrintTable,
237.7.608 by Teddy Hogeborn
mandos-ctl: Refactor
1362
                                      verbose=True)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1363
1364
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1365
class TestCommand(unittest.TestCase):
237.7.558 by Teddy Hogeborn
mandos-ctl: Refactor; test PrintTableCmd instead of TableOfClients
1366
    """Abstract class for tests of command classes"""
237.7.609 by Teddy Hogeborn
mandos-ctl: White space changes only
1367
237.7.542 by Teddy Hogeborn
mandos-ctl: Add tests for table_rows_of_clients()
1368
    def setUp(self):
237.7.558 by Teddy Hogeborn
mandos-ctl: Refactor; test PrintTableCmd instead of TableOfClients
1369
        testcase = self
1370
        class MockClient(object):
1371
            def __init__(self, name, **attributes):
237.7.598 by Teddy Hogeborn
mandos-ctl: Minimize number of D-Bus calls
1372
                self.__dbus_object_path__ = "/clients/{}".format(name)
237.7.558 by Teddy Hogeborn
mandos-ctl: Refactor; test PrintTableCmd instead of TableOfClients
1373
                self.attributes = attributes
1374
                self.attributes["Name"] = name
237.7.561 by Teddy Hogeborn
mandos-ctl: Add test for IsEnabledCmd class
1375
                self.calls = []
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1376
            def Set(self, interface, propname, value, dbus_interface):
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
1377
                testcase.assertEqual(client_dbus_interface, interface)
1378
                testcase.assertEqual(dbus.PROPERTIES_IFACE,
1379
                                     dbus_interface)
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1380
                self.attributes[propname] = value
237.7.567 by Teddy Hogeborn
mandos-ctl: New tests for ApproveCmd and DenyCmd
1381
            def Approve(self, approve, dbus_interface):
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
1382
                testcase.assertEqual(client_dbus_interface,
1383
                                     dbus_interface)
237.7.567 by Teddy Hogeborn
mandos-ctl: New tests for ApproveCmd and DenyCmd
1384
                self.calls.append(("Approve", (approve,
1385
                                               dbus_interface)))
237.7.568 by Teddy Hogeborn
mandos-ctl: Bug fix: fix client/properties confusion
1386
        self.client = MockClient(
1387
            "foo",
1388
            KeyID=("92ed150794387c03ce684574b1139a65"
1389
                   "94a34f895daaaf09fd8ea90a27cddb12"),
1390
            Secret=b"secret",
1391
            Host="foo.example.org",
1392
            Enabled=dbus.Boolean(True),
1393
            Timeout=300000,
1394
            LastCheckedOK="2019-02-03T00:00:00",
1395
            Created="2019-01-02T00:00:00",
1396
            Interval=120000,
1397
            Fingerprint=("778827225BA7DE539C5A"
1398
                         "7CFA59CFF7CDBD9A5920"),
1399
            CheckerRunning=dbus.Boolean(False),
1400
            LastEnabled="2019-01-03T00:00:00",
1401
            ApprovalPending=dbus.Boolean(False),
1402
            ApprovedByDefault=dbus.Boolean(True),
1403
            LastApprovalRequest="",
1404
            ApprovalDelay=0,
1405
            ApprovalDuration=1000,
1406
            Checker="fping -q -- %(host)s",
1407
            ExtendedTimeout=900000,
1408
            Expires="2019-02-04T00:00:00",
1409
            LastCheckerStatus=0)
1410
        self.other_client = MockClient(
1411
            "barbar",
1412
            KeyID=("0558568eedd67d622f5c83b35a115f79"
1413
                   "6ab612cff5ad227247e46c2b020f441c"),
1414
            Secret=b"secretbar",
1415
            Host="192.0.2.3",
1416
            Enabled=dbus.Boolean(True),
1417
            Timeout=300000,
1418
            LastCheckedOK="2019-02-04T00:00:00",
1419
            Created="2019-01-03T00:00:00",
1420
            Interval=120000,
1421
            Fingerprint=("3E393AEAEFB84C7E89E2"
1422
                         "F547B3A107558FCA3A27"),
1423
            CheckerRunning=dbus.Boolean(True),
1424
            LastEnabled="2019-01-04T00:00:00",
1425
            ApprovalPending=dbus.Boolean(False),
1426
            ApprovedByDefault=dbus.Boolean(False),
1427
            LastApprovalRequest="2019-01-03T00:00:00",
1428
            ApprovalDelay=30000,
237.7.604 by Teddy Hogeborn
mandos-ctl: Refactor
1429
            ApprovalDuration=93785000,
237.7.568 by Teddy Hogeborn
mandos-ctl: Bug fix: fix client/properties confusion
1430
            Checker=":",
1431
            ExtendedTimeout=900000,
1432
            Expires="2019-02-05T00:00:00",
1433
            LastCheckerStatus=-2)
1434
        self.clients =  collections.OrderedDict(
1435
            [
237.7.598 by Teddy Hogeborn
mandos-ctl: Minimize number of D-Bus calls
1436
                ("/clients/foo", self.client.attributes),
1437
                ("/clients/barbar", self.other_client.attributes),
237.7.558 by Teddy Hogeborn
mandos-ctl: Refactor; test PrintTableCmd instead of TableOfClients
1438
            ])
237.7.598 by Teddy Hogeborn
mandos-ctl: Minimize number of D-Bus calls
1439
        self.one_client = {"/clients/foo": self.client.attributes}
237.7.609 by Teddy Hogeborn
mandos-ctl: White space changes only
1440
237.7.598 by Teddy Hogeborn
mandos-ctl: Minimize number of D-Bus calls
1441
    @property
1442
    def bus(self):
1443
        class Bus(object):
1444
            @staticmethod
1445
            def get_object(client_bus_name, path):
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
1446
                self.assertEqual(dbus_busname, client_bus_name)
237.7.598 by Teddy Hogeborn
mandos-ctl: Minimize number of D-Bus calls
1447
                return {
237.7.610 by Teddy Hogeborn
mandos-ctl: Refactor
1448
                    # Note: "self" here is the TestCmd instance, not
1449
                    # the Bus instance, since this is a static method!
237.7.598 by Teddy Hogeborn
mandos-ctl: Minimize number of D-Bus calls
1450
                    "/clients/foo": self.client,
1451
                    "/clients/barbar": self.other_client,
1452
                }[path]
1453
        return Bus()
237.7.558 by Teddy Hogeborn
mandos-ctl: Refactor; test PrintTableCmd instead of TableOfClients
1454
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1455
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1456
class TestBaseCommands(TestCommand):
1457
237.7.629 by Teddy Hogeborn
mandos-ctl: Refactor tests
1458
    def test_IsEnabled_exits_successfully(self):
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1459
        with self.assertRaises(SystemExit) as e:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1460
            command.IsEnabled().run(self.one_client)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1461
        if e.exception.code is not None:
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
1462
            self.assertEqual(0, e.exception.code)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1463
        else:
1464
            self.assertIsNone(e.exception.code)
237.7.609 by Teddy Hogeborn
mandos-ctl: White space changes only
1465
237.7.629 by Teddy Hogeborn
mandos-ctl: Refactor tests
1466
    def test_IsEnabled_exits_with_failure(self):
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1467
        self.client.attributes["Enabled"] = dbus.Boolean(False)
1468
        with self.assertRaises(SystemExit) as e:
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1469
            command.IsEnabled().run(self.one_client)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1470
        if isinstance(e.exception.code, int):
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
1471
            self.assertNotEqual(0, e.exception.code)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1472
        else:
1473
            self.assertIsNotNone(e.exception.code)
1474
237.7.627 by Teddy Hogeborn
mandos-ctl: Refactor
1475
    def test_Approve(self):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1476
        command.Approve().run(self.clients, self.bus)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1477
        for clientpath in self.clients:
1478
            client = self.bus.get_object(dbus_busname, clientpath)
1479
            self.assertIn(("Approve", (True, client_dbus_interface)),
1480
                          client.calls)
1481
237.7.627 by Teddy Hogeborn
mandos-ctl: Refactor
1482
    def test_Deny(self):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1483
        command.Deny().run(self.clients, self.bus)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1484
        for clientpath in self.clients:
1485
            client = self.bus.get_object(dbus_busname, clientpath)
1486
            self.assertIn(("Approve", (False, client_dbus_interface)),
1487
                          client.calls)
1488
237.7.627 by Teddy Hogeborn
mandos-ctl: Refactor
1489
    def test_Remove(self):
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1490
        class MockMandos(object):
1491
            def __init__(self):
1492
                self.calls = []
1493
            def RemoveClient(self, dbus_path):
1494
                self.calls.append(("RemoveClient", (dbus_path,)))
1495
        mandos = MockMandos()
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1496
        command.Remove().run(self.clients, self.bus, mandos)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1497
        for clientpath in self.clients:
1498
            self.assertIn(("RemoveClient", (clientpath,)),
1499
                          mandos.calls)
1500
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1501
    expected_json = {
1502
        "foo": {
1503
            "Name": "foo",
1504
            "KeyID": ("92ed150794387c03ce684574b1139a65"
1505
                      "94a34f895daaaf09fd8ea90a27cddb12"),
1506
            "Host": "foo.example.org",
1507
            "Enabled": True,
1508
            "Timeout": 300000,
1509
            "LastCheckedOK": "2019-02-03T00:00:00",
1510
            "Created": "2019-01-02T00:00:00",
1511
            "Interval": 120000,
1512
            "Fingerprint": ("778827225BA7DE539C5A"
1513
                            "7CFA59CFF7CDBD9A5920"),
1514
            "CheckerRunning": False,
1515
            "LastEnabled": "2019-01-03T00:00:00",
1516
            "ApprovalPending": False,
1517
            "ApprovedByDefault": True,
1518
            "LastApprovalRequest": "",
1519
            "ApprovalDelay": 0,
1520
            "ApprovalDuration": 1000,
1521
            "Checker": "fping -q -- %(host)s",
1522
            "ExtendedTimeout": 900000,
1523
            "Expires": "2019-02-04T00:00:00",
1524
            "LastCheckerStatus": 0,
1525
        },
1526
        "barbar": {
1527
            "Name": "barbar",
1528
            "KeyID": ("0558568eedd67d622f5c83b35a115f79"
1529
                      "6ab612cff5ad227247e46c2b020f441c"),
1530
            "Host": "192.0.2.3",
1531
            "Enabled": True,
1532
            "Timeout": 300000,
1533
            "LastCheckedOK": "2019-02-04T00:00:00",
1534
            "Created": "2019-01-03T00:00:00",
1535
            "Interval": 120000,
1536
            "Fingerprint": ("3E393AEAEFB84C7E89E2"
1537
                            "F547B3A107558FCA3A27"),
1538
            "CheckerRunning": True,
1539
            "LastEnabled": "2019-01-04T00:00:00",
1540
            "ApprovalPending": False,
1541
            "ApprovedByDefault": False,
1542
            "LastApprovalRequest": "2019-01-03T00:00:00",
1543
            "ApprovalDelay": 30000,
1544
            "ApprovalDuration": 93785000,
1545
            "Checker": ":",
1546
            "ExtendedTimeout": 900000,
1547
            "Expires": "2019-02-05T00:00:00",
1548
            "LastCheckerStatus": -2,
1549
        },
1550
    }
1551
1552
    def test_DumpJSON_normal(self):
237.7.632 by Teddy Hogeborn
mandos-ctl: Refactor tests
1553
        with self.capture_stdout_to_buffer() as buffer:
1554
            command.DumpJSON().run(self.clients)
1555
        json_data = json.loads(buffer.getvalue())
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
1556
        self.assertDictEqual(self.expected_json, json_data)
237.7.609 by Teddy Hogeborn
mandos-ctl: White space changes only
1557
237.7.632 by Teddy Hogeborn
mandos-ctl: Refactor tests
1558
    @staticmethod
1559
    @contextlib.contextmanager
1560
    def capture_stdout_to_buffer():
1561
        capture_buffer = io.StringIO()
1562
        old_stdout = sys.stdout
1563
        sys.stdout = capture_buffer
1564
        try:
1565
            yield capture_buffer
1566
        finally:
1567
            sys.stdout = old_stdout
1568
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1569
    def test_DumpJSON_one_client(self):
237.7.632 by Teddy Hogeborn
mandos-ctl: Refactor tests
1570
        with self.capture_stdout_to_buffer() as buffer:
1571
            command.DumpJSON().run(self.one_client)
1572
        json_data = json.loads(buffer.getvalue())
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1573
        expected_json = {"foo": self.expected_json["foo"]}
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
1574
        self.assertDictEqual(expected_json, json_data)
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1575
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1576
    def test_PrintTable_normal(self):
237.7.632 by Teddy Hogeborn
mandos-ctl: Refactor tests
1577
        with self.capture_stdout_to_buffer() as buffer:
1578
            command.PrintTable().run(self.clients)
237.7.601 by Teddy Hogeborn
mandos-ctl: White space and other non-semantic changes only
1579
        expected_output = "\n".join((
1580
            "Name   Enabled Timeout  Last Successful Check",
1581
            "foo    Yes     00:05:00 2019-02-03T00:00:00  ",
1582
            "barbar Yes     00:05:00 2019-02-04T00:00:00  ",
237.7.632 by Teddy Hogeborn
mandos-ctl: Refactor tests
1583
        )) + "\n"
1584
        self.assertEqual(expected_output, buffer.getvalue())
237.7.609 by Teddy Hogeborn
mandos-ctl: White space changes only
1585
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1586
    def test_PrintTable_verbose(self):
237.7.632 by Teddy Hogeborn
mandos-ctl: Refactor tests
1587
        with self.capture_stdout_to_buffer() as buffer:
1588
            command.PrintTable(verbose=True).run(self.clients)
237.7.602 by Teddy Hogeborn
mandos-ctl: Refactor
1589
        columns = (
1590
            (
1591
                "Name   ",
1592
                "foo    ",
1593
                "barbar ",
1594
            ),(
1595
                "Enabled ",
1596
                "Yes     ",
1597
                "Yes     ",
1598
            ),(
1599
                "Timeout  ",
1600
                "00:05:00 ",
1601
                "00:05:00 ",
1602
            ),(
1603
                "Last Successful Check ",
1604
                "2019-02-03T00:00:00   ",
1605
                "2019-02-04T00:00:00   ",
1606
            ),(
1607
                "Created             ",
1608
                "2019-01-02T00:00:00 ",
1609
                "2019-01-03T00:00:00 ",
1610
            ),(
1611
                "Interval ",
1612
                "00:02:00 ",
1613
                "00:02:00 ",
1614
            ),(
1615
                "Host            ",
1616
                "foo.example.org ",
1617
                "192.0.2.3       ",
1618
            ),(
1619
                ("Key ID                                             "
1620
                 "              "),
1621
                ("92ed150794387c03ce684574b1139a6594a34f895daaaf09fd8"
1622
                 "ea90a27cddb12 "),
1623
                ("0558568eedd67d622f5c83b35a115f796ab612cff5ad227247e"
1624
                 "46c2b020f441c "),
1625
            ),(
1626
                "Fingerprint                              ",
1627
                "778827225BA7DE539C5A7CFA59CFF7CDBD9A5920 ",
1628
                "3E393AEAEFB84C7E89E2F547B3A107558FCA3A27 ",
1629
            ),(
1630
                "Check Is Running ",
1631
                "No               ",
1632
                "Yes              ",
1633
            ),(
1634
                "Last Enabled        ",
1635
                "2019-01-03T00:00:00 ",
1636
                "2019-01-04T00:00:00 ",
1637
            ),(
1638
                "Approval Is Pending ",
1639
                "No                  ",
1640
                "No                  ",
1641
            ),(
1642
                "Approved By Default ",
1643
                "Yes                 ",
1644
                "No                  ",
1645
            ),(
1646
                "Last Approval Request ",
1647
                "                      ",
1648
                "2019-01-03T00:00:00   ",
1649
            ),(
1650
                "Approval Delay ",
1651
                "00:00:00       ",
1652
                "00:00:30       ",
1653
            ),(
1654
                "Approval Duration ",
1655
                "00:00:01          ",
237.7.604 by Teddy Hogeborn
mandos-ctl: Refactor
1656
                "1T02:03:05        ",
237.7.602 by Teddy Hogeborn
mandos-ctl: Refactor
1657
            ),(
1658
                "Checker              ",
1659
                "fping -q -- %(host)s ",
1660
                ":                    ",
1661
            ),(
1662
                "Extended Timeout ",
1663
                "00:15:00         ",
1664
                "00:15:00         ",
1665
            ),(
1666
                "Expires             ",
1667
                "2019-02-04T00:00:00 ",
1668
                "2019-02-05T00:00:00 ",
1669
            ),(
1670
                "Last Checker Status",
1671
                "0                  ",
1672
                "-2                 ",
1673
            )
1674
        )
1675
        num_lines = max(len(rows) for rows in columns)
237.7.632 by Teddy Hogeborn
mandos-ctl: Refactor tests
1676
        expected_output = ("\n".join("".join(rows[line]
1677
                                             for rows in columns)
1678
                                     for line in range(num_lines))
1679
                           + "\n")
1680
        self.assertEqual(expected_output, buffer.getvalue())
237.7.609 by Teddy Hogeborn
mandos-ctl: White space changes only
1681
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1682
    def test_PrintTable_one_client(self):
237.7.632 by Teddy Hogeborn
mandos-ctl: Refactor tests
1683
        with self.capture_stdout_to_buffer() as buffer:
1684
            command.PrintTable().run(self.one_client)
237.7.605 by Teddy Hogeborn
mandos-ctl: Refactor
1685
        expected_output = "\n".join((
1686
            "Name Enabled Timeout  Last Successful Check",
1687
            "foo  Yes     00:05:00 2019-02-03T00:00:00  ",
237.7.632 by Teddy Hogeborn
mandos-ctl: Refactor tests
1688
        )) + "\n"
1689
        self.assertEqual(expected_output, buffer.getvalue())
237.7.542 by Teddy Hogeborn
mandos-ctl: Add tests for table_rows_of_clients()
1690
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1691
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1692
class TestPropertyCmd(TestCommand):
1693
    """Abstract class for tests of command.Property classes"""
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1694
    def runTest(self):
1695
        if not hasattr(self, "command"):
1696
            return
1697
        values_to_get = getattr(self, "values_to_get",
1698
                                self.values_to_set)
1699
        for value_to_set, value_to_get in zip(self.values_to_set,
1700
                                              values_to_get):
237.7.598 by Teddy Hogeborn
mandos-ctl: Minimize number of D-Bus calls
1701
            for clientpath in self.clients:
237.7.600 by Teddy Hogeborn
mandos-ctl: Refactor
1702
                client = self.bus.get_object(dbus_busname, clientpath)
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1703
                old_value = client.attributes[self.propname]
237.7.610 by Teddy Hogeborn
mandos-ctl: Refactor
1704
                client.attributes[self.propname] = self.Unique()
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1705
            self.run_command(value_to_set, self.clients)
237.7.598 by Teddy Hogeborn
mandos-ctl: Minimize number of D-Bus calls
1706
            for clientpath in self.clients:
237.7.600 by Teddy Hogeborn
mandos-ctl: Refactor
1707
                client = self.bus.get_object(dbus_busname, clientpath)
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1708
                value = client.attributes[self.propname]
237.7.610 by Teddy Hogeborn
mandos-ctl: Refactor
1709
                self.assertNotIsInstance(value, self.Unique)
237.7.630 by Teddy Hogeborn
mandos-ctl: Refactor tests
1710
                self.assertEqual(value_to_get, value)
237.7.610 by Teddy Hogeborn
mandos-ctl: Refactor
1711
1712
    class Unique(object):
1713
        """Class for objects which exist only to be unique objects,
1714
since unittest.mock.sentinel only exists in Python 3.3"""
1715
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1716
    def run_command(self, value, clients):
237.7.598 by Teddy Hogeborn
mandos-ctl: Minimize number of D-Bus calls
1717
        self.command().run(clients, self.bus)
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1718
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1719
237.7.607 by Teddy Hogeborn
mandos-ctl: Refactor
1720
class TestEnableCmd(TestPropertyCmd):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1721
    command = command.Enable
237.7.607 by Teddy Hogeborn
mandos-ctl: Refactor
1722
    propname = "Enabled"
1723
    values_to_set = [dbus.Boolean(True)]
1724
1725
1726
class TestDisableCmd(TestPropertyCmd):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1727
    command = command.Disable
237.7.607 by Teddy Hogeborn
mandos-ctl: Refactor
1728
    propname = "Enabled"
1729
    values_to_set = [dbus.Boolean(False)]
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1730
1731
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1732
class TestBumpTimeoutCmd(TestPropertyCmd):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1733
    command = command.BumpTimeout
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1734
    propname = "LastCheckedOK"
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1735
    values_to_set = [""]
1736
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1737
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1738
class TestStartCheckerCmd(TestPropertyCmd):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1739
    command = command.StartChecker
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1740
    propname = "CheckerRunning"
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1741
    values_to_set = [dbus.Boolean(True)]
1742
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1743
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1744
class TestStopCheckerCmd(TestPropertyCmd):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1745
    command = command.StopChecker
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1746
    propname = "CheckerRunning"
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1747
    values_to_set = [dbus.Boolean(False)]
1748
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1749
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1750
class TestApproveByDefaultCmd(TestPropertyCmd):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1751
    command = command.ApproveByDefault
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1752
    propname = "ApprovedByDefault"
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1753
    values_to_set = [dbus.Boolean(True)]
1754
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1755
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1756
class TestDenyByDefaultCmd(TestPropertyCmd):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1757
    command = command.DenyByDefault
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1758
    propname = "ApprovedByDefault"
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1759
    values_to_set = [dbus.Boolean(False)]
1760
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1761
237.7.599 by Teddy Hogeborn
mandos-ctl: Refactor
1762
class TestPropertyValueCmd(TestPropertyCmd):
1763
    """Abstract class for tests of PropertyValueCmd classes"""
237.7.609 by Teddy Hogeborn
mandos-ctl: White space changes only
1764
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1765
    def runTest(self):
237.7.599 by Teddy Hogeborn
mandos-ctl: Refactor
1766
        if type(self) is TestPropertyValueCmd:
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1767
            return
237.7.599 by Teddy Hogeborn
mandos-ctl: Refactor
1768
        return super(TestPropertyValueCmd, self).runTest()
237.7.609 by Teddy Hogeborn
mandos-ctl: White space changes only
1769
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1770
    def run_command(self, value, clients):
237.7.598 by Teddy Hogeborn
mandos-ctl: Minimize number of D-Bus calls
1771
        self.command(value).run(clients, self.bus)
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1772
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1773
237.7.599 by Teddy Hogeborn
mandos-ctl: Refactor
1774
class TestSetCheckerCmd(TestPropertyValueCmd):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1775
    command = command.SetChecker
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1776
    propname = "Checker"
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1777
    values_to_set = ["", ":", "fping -q -- %s"]
1778
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1779
237.7.599 by Teddy Hogeborn
mandos-ctl: Refactor
1780
class TestSetHostCmd(TestPropertyValueCmd):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1781
    command = command.SetHost
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1782
    propname = "Host"
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1783
    values_to_set = ["192.0.2.3", "foo.example.org"]
1784
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1785
237.7.599 by Teddy Hogeborn
mandos-ctl: Refactor
1786
class TestSetSecretCmd(TestPropertyValueCmd):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1787
    command = command.SetSecret
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1788
    propname = "Secret"
237.7.590 by Teddy Hogeborn
mandos-ctl: Bug fix: close an open file
1789
    values_to_set = [io.BytesIO(b""),
237.7.578 by Teddy Hogeborn
mandos-ctl: Fix bugs
1790
                     io.BytesIO(b"secret\0xyzzy\nbar")]
1791
    values_to_get = [b"", b"secret\0xyzzy\nbar"]
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1792
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1793
237.7.599 by Teddy Hogeborn
mandos-ctl: Refactor
1794
class TestSetTimeoutCmd(TestPropertyValueCmd):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1795
    command = command.SetTimeout
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1796
    propname = "Timeout"
237.7.583 by Teddy Hogeborn
mandos-ctl: Refactor; move parsing of intervals into argument parsing
1797
    values_to_set = [datetime.timedelta(),
1798
                     datetime.timedelta(minutes=5),
1799
                     datetime.timedelta(seconds=1),
1800
                     datetime.timedelta(weeks=1),
1801
                     datetime.timedelta(weeks=52)]
1802
    values_to_get = [0, 300000, 1000, 604800000, 31449600000]
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1803
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1804
237.7.599 by Teddy Hogeborn
mandos-ctl: Refactor
1805
class TestSetExtendedTimeoutCmd(TestPropertyValueCmd):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1806
    command = command.SetExtendedTimeout
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1807
    propname = "ExtendedTimeout"
237.7.583 by Teddy Hogeborn
mandos-ctl: Refactor; move parsing of intervals into argument parsing
1808
    values_to_set = [datetime.timedelta(),
1809
                     datetime.timedelta(minutes=5),
1810
                     datetime.timedelta(seconds=1),
1811
                     datetime.timedelta(weeks=1),
1812
                     datetime.timedelta(weeks=52)]
1813
    values_to_get = [0, 300000, 1000, 604800000, 31449600000]
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1814
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1815
237.7.599 by Teddy Hogeborn
mandos-ctl: Refactor
1816
class TestSetIntervalCmd(TestPropertyValueCmd):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1817
    command = command.SetInterval
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1818
    propname = "Interval"
237.7.583 by Teddy Hogeborn
mandos-ctl: Refactor; move parsing of intervals into argument parsing
1819
    values_to_set = [datetime.timedelta(),
1820
                     datetime.timedelta(minutes=5),
1821
                     datetime.timedelta(seconds=1),
1822
                     datetime.timedelta(weeks=1),
1823
                     datetime.timedelta(weeks=52)]
1824
    values_to_get = [0, 300000, 1000, 604800000, 31449600000]
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1825
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1826
237.7.599 by Teddy Hogeborn
mandos-ctl: Refactor
1827
class TestSetApprovalDelayCmd(TestPropertyValueCmd):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1828
    command = command.SetApprovalDelay
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1829
    propname = "ApprovalDelay"
237.7.583 by Teddy Hogeborn
mandos-ctl: Refactor; move parsing of intervals into argument parsing
1830
    values_to_set = [datetime.timedelta(),
1831
                     datetime.timedelta(minutes=5),
1832
                     datetime.timedelta(seconds=1),
1833
                     datetime.timedelta(weeks=1),
1834
                     datetime.timedelta(weeks=52)]
1835
    values_to_get = [0, 300000, 1000, 604800000, 31449600000]
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1836
237.7.603 by Teddy Hogeborn
mandos-ctl: Refactor
1837
237.7.599 by Teddy Hogeborn
mandos-ctl: Refactor
1838
class TestSetApprovalDurationCmd(TestPropertyValueCmd):
237.7.625 by Teddy Hogeborn
mandos-ctl: Refactor
1839
    command = command.SetApprovalDuration
237.7.595 by Teddy Hogeborn
mandos-ctl: Refactor
1840
    propname = "ApprovalDuration"
237.7.583 by Teddy Hogeborn
mandos-ctl: Refactor; move parsing of intervals into argument parsing
1841
    values_to_set = [datetime.timedelta(),
1842
                     datetime.timedelta(minutes=5),
1843
                     datetime.timedelta(seconds=1),
1844
                     datetime.timedelta(weeks=1),
1845
                     datetime.timedelta(weeks=52)]
1846
    values_to_get = [0, 300000, 1000, 604800000, 31449600000]
237.7.572 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
1847
237.7.589 by Teddy Hogeborn
mandos-ctl: Add tests for option syntax checks
1848
237.7.534 by Teddy Hogeborn
Add tests to mandos-ctl's milliseconds_to_string function
1849

237.7.532 by Teddy Hogeborn
Make mandos-ctl use unittest instead of doctest module
1850
def should_only_run_tests():
1851
    parser = argparse.ArgumentParser(add_help=False)
1852
    parser.add_argument("--check", action='store_true')
1853
    args, unknown_args = parser.parse_known_args()
1854
    run_tests = args.check
1855
    if run_tests:
1856
        # Remove --check argument from sys.argv
1857
        sys.argv[1:] = unknown_args
1858
    return run_tests
1859
1860
# Add all tests from doctest strings
1861
def load_tests(loader, tests, none):
1862
    import doctest
1863
    tests.addTests(doctest.DocTestSuite())
1864
    return tests
237.7.293 by Teddy Hogeborn
mandos-ctl: Do minor formatting and whitespace adjustments.
1865
237.8.8 by teddy at bsnet
* mandos-ctl: Use unicode string literals.
1866
if __name__ == "__main__":
237.7.615 by Teddy Hogeborn
mandos-ctl: Bug fix: always shutdown logging.
1867
    try:
1868
        if should_only_run_tests():
1869
            # Call using ./tdd-python-script --check [--verbose]
1870
            unittest.main()
1871
        else:
1872
            main()
1873
    finally:
1874
        logging.shutdown()