/mandos/trunk

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/trunk
1185 by Teddy Hogeborn
Add -I option to python 3 on the interpreter script ("shebang") line
1
#!/usr/bin/python3 -bbI
1130 by Teddy Hogeborn
Update Python 3 compatibility
2
# -*- after-save-hook: (lambda () (let ((command (if (fboundp 'file-local-name) (file-local-name (buffer-file-name)) (or (file-remote-p (buffer-file-name) 'localname) (buffer-file-name))))) (if (= (progn (if (get-buffer "*Test*") (kill-buffer "*Test*")) (process-file-shell-command (format "%s --check" (shell-quote-argument command)) nil "*Test*")) 0) (let ((w (get-buffer-window "*Test*"))) (if w (delete-window w))) (progn (with-current-buffer "*Test*" (compilation-mode)) (display-buffer "*Test*" '(display-buffer-in-side-window)))))); coding: utf-8 -*-
872 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
3
#
444 by Teddy Hogeborn
Update copyright year to "2010" wherever appropriate.
4
# Mandos Monitor - Control and monitor the Mandos server
872 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
5
#
969 by Teddy Hogeborn
Update copyright year to 2019
6
# Copyright © 2008-2019 Teddy Hogeborn
7
# Copyright © 2008-2019 Björn Påhlsson
872 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
8
#
907 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
444 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
#
907 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
444 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.
872 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
20
#
444 by Teddy Hogeborn
Update copyright year to "2010" wherever appropriate.
21
# You should have received a copy of the GNU General Public License
907 by Teddy Hogeborn
Alter copyright notices slightly. Actual license is unchanged!
22
# along with Mandos.  If not, see <http://www.gnu.org/licenses/>.
872 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
23
#
505.1.2 by Teddy Hogeborn
Change "fukt.bsnet.se" to "recompile.se" throughout.
24
# Contact the authors at <mandos@recompile.se>.
872 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
25
#
24.1.116 by Björn Påhlsson
added a mandos list client program
26
463.1.9 by teddy at bsnet
* mandos-ctl: Use print function.
27
from __future__ import (division, absolute_import, print_function,
28
                        unicode_literals)
463.1.8 by teddy at bsnet
* mandos-ctl: Use unicode string literals.
29
718 by Teddy Hogeborn
mandos-ctl: Make it work in Python 3.
30
try:
31
    from future_builtins import *
32
except ImportError:
33
    pass
579 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
475 by teddy at bsnet
* mandos-ctl: Use the new argparse library instead of optparse.
36
import argparse
240 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
608 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
41
import collections
863 by Teddy Hogeborn
mandos-ctl: Implement --dump-json option
42
import json
984 by Teddy Hogeborn
Make mandos-ctl use unittest instead of doctest module
43
import unittest
987 by Teddy Hogeborn
mandos-ctl: Use logging module instead of print() for errors
44
import logging
1030 by Teddy Hogeborn
mandos-ctl: Fix bugs
45
import io
1031 by Teddy Hogeborn
mandos-ctl: Refactor tests and add more tests
46
import tempfile
1041 by Teddy Hogeborn
mandos-ctl: Add tests for option syntax checks
47
import contextlib
608 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
48
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
49
if sys.version_info.major == 2:
50
    __metaclass__ = type
1191 by teddy at recompile
Minor refactor of Python code: Move "str = unicode" to top
51
    str = unicode
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
52
1096 by Teddy Hogeborn
mandos-ctl: Add support for D-Bus module "pydbus"
53
try:
54
    import pydbus
55
    import gi
56
    dbus_python = None
57
except ImportError:
58
    import dbus as dbus_python
59
    pydbus = None
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
60
    class gi:
1096 by Teddy Hogeborn
mandos-ctl: Add support for D-Bus module "pydbus"
61
        """Dummy gi module, for the tests"""
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
62
        class repository:
63
            class GLib:
1096 by Teddy Hogeborn
mandos-ctl: Add support for D-Bus module "pydbus"
64
                class Error(Exception):
65
                    pass
240 by Teddy Hogeborn
Merge "mandos-list" from belorn.
66
988 by Teddy Hogeborn
mandos-ctl: Show warnings
67
# Show warnings by default
68
if not sys.warnoptions:
69
    import warnings
70
    warnings.simplefilter("default")
71
987 by Teddy Hogeborn
mandos-ctl: Use logging module instead of print() for errors
72
log = logging.getLogger(sys.argv[0])
73
logging.basicConfig(level="INFO", # Show info level messages
74
                    format="%(message)s") # Show basic log messages
75
988 by Teddy Hogeborn
mandos-ctl: Show warnings
76
logging.captureWarnings(True)   # Show warnings via the logging system
77
723.1.7 by Teddy Hogeborn
Use the .major attribute on sys.version_info instead of using "[0]".
78
if sys.version_info.major == 2:
1085 by Teddy Hogeborn
mandos-ctl: Refactor
79
    import StringIO
80
    io.StringIO = StringIO.StringIO
718 by Teddy Hogeborn
mandos-ctl: Make it work in Python 3.
81
463.1.8 by teddy at bsnet
* mandos-ctl: Use unicode string literals.
82
locale.setlocale(locale.LC_ALL, "")
24.1.116 by Björn Påhlsson
added a mandos list client program
83
237.4.124 by Teddy Hogeborn
* Makefile (version): Change to 1.8.9.
84
version = "1.8.9"
24.1.118 by Björn Påhlsson
Added enable/disable
85
745 by Teddy Hogeborn
mandos-ctl: Do minor formatting and whitespace adjustments.
86
1055 by Teddy Hogeborn
mandos-ctl: Refactor
87
def main():
88
    parser = argparse.ArgumentParser()
89
    add_command_line_options(parser)
90
91
    options = parser.parse_args()
92
    check_option_syntax(parser, options)
93
94
    clientnames = options.client
95
96
    if options.debug:
97
        log.setLevel(logging.DEBUG)
98
1096 by Teddy Hogeborn
mandos-ctl: Add support for D-Bus module "pydbus"
99
    if pydbus is not None:
100
        bus = pydbus_adapter.CachingBus(pydbus)
101
    else:
102
        bus = dbus_python_adapter.CachingBus(dbus_python)
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
103
104
    try:
105
        all_clients = bus.get_clients_and_properties()
106
    except dbus.ConnectFailed as e:
107
        log.critical("Could not connect to Mandos server: %s", e)
108
        sys.exit(1)
109
    except dbus.Error as e:
110
        log.critical(
111
            "Failed to access Mandos server through D-Bus:\n%s", e)
112
        sys.exit(1)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
113
1069 by Teddy Hogeborn
mandos-ctl: Refactor
114
    # Compile dict of (clientpath: properties) to process
1055 by Teddy Hogeborn
mandos-ctl: Refactor
115
    if not clientnames:
1065 by Teddy Hogeborn
mandos-ctl: Refactor
116
        clients = all_clients
1055 by Teddy Hogeborn
mandos-ctl: Refactor
117
    else:
1069 by Teddy Hogeborn
mandos-ctl: Refactor
118
        clients = {}
1055 by Teddy Hogeborn
mandos-ctl: Refactor
119
        for name in clientnames:
1065 by Teddy Hogeborn
mandos-ctl: Refactor
120
            for objpath, properties in all_clients.items():
1055 by Teddy Hogeborn
mandos-ctl: Refactor
121
                if properties["Name"] == name:
122
                    clients[objpath] = properties
123
                    break
124
            else:
125
                log.critical("Client not found on server: %r", name)
126
                sys.exit(1)
127
128
    commands = commands_from_options(options)
1076 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
129
1055 by Teddy Hogeborn
mandos-ctl: Refactor
130
    for command in commands:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
131
        command.run(clients, bus)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
132
133
134
def add_command_line_options(parser):
135
    parser.add_argument("--version", action="version",
136
                        version="%(prog)s {}".format(version),
137
                        help="show version number and exit")
138
    parser.add_argument("-a", "--all", action="store_true",
139
                        help="Select all clients")
140
    parser.add_argument("-v", "--verbose", action="store_true",
141
                        help="Print all fields")
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
142
    parser.add_argument("-j", "--dump-json", dest="commands",
143
                        action="append_const", default=[],
144
                        const=command.DumpJSON(),
1055 by Teddy Hogeborn
mandos-ctl: Refactor
145
                        help="Dump client data in JSON format")
146
    enable_disable = parser.add_mutually_exclusive_group()
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
147
    enable_disable.add_argument("-e", "--enable", dest="commands",
148
                                action="append_const", default=[],
149
                                const=command.Enable(),
1055 by Teddy Hogeborn
mandos-ctl: Refactor
150
                                help="Enable client")
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
151
    enable_disable.add_argument("-d", "--disable", dest="commands",
152
                                action="append_const", default=[],
153
                                const=command.Disable(),
1055 by Teddy Hogeborn
mandos-ctl: Refactor
154
                                help="disable client")
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
155
    parser.add_argument("-b", "--bump-timeout", dest="commands",
156
                        action="append_const", default=[],
157
                        const=command.BumpTimeout(),
1055 by Teddy Hogeborn
mandos-ctl: Refactor
158
                        help="Bump timeout for client")
159
    start_stop_checker = parser.add_mutually_exclusive_group()
160
    start_stop_checker.add_argument("--start-checker",
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
161
                                    dest="commands",
162
                                    action="append_const", default=[],
163
                                    const=command.StartChecker(),
1055 by Teddy Hogeborn
mandos-ctl: Refactor
164
                                    help="Start checker for client")
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
165
    start_stop_checker.add_argument("--stop-checker", dest="commands",
166
                                    action="append_const", default=[],
167
                                    const=command.StopChecker(),
1055 by Teddy Hogeborn
mandos-ctl: Refactor
168
                                    help="Stop checker for client")
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
169
    parser.add_argument("-V", "--is-enabled", dest="commands",
170
                        action="append_const", default=[],
171
                        const=command.IsEnabled(),
1055 by Teddy Hogeborn
mandos-ctl: Refactor
172
                        help="Check if client is enabled")
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
173
    parser.add_argument("-r", "--remove", dest="commands",
174
                        action="append_const", default=[],
175
                        const=command.Remove(),
1055 by Teddy Hogeborn
mandos-ctl: Refactor
176
                        help="Remove client")
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
177
    parser.add_argument("-c", "--checker", dest="commands",
178
                        action="append", default=[],
179
                        metavar="COMMAND", type=command.SetChecker,
1055 by Teddy Hogeborn
mandos-ctl: Refactor
180
                        help="Set checker command for client")
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
181
    parser.add_argument(
182
        "-t", "--timeout", dest="commands", action="append",
183
        default=[], metavar="TIME",
184
        type=command.SetTimeout.argparse(string_to_delta),
185
        help="Set timeout for client")
186
    parser.add_argument(
187
        "--extended-timeout", dest="commands", action="append",
188
        default=[], metavar="TIME",
189
        type=command.SetExtendedTimeout.argparse(string_to_delta),
190
        help="Set extended timeout for client")
191
    parser.add_argument(
192
        "-i", "--interval", dest="commands", action="append",
193
        default=[], metavar="TIME",
194
        type=command.SetInterval.argparse(string_to_delta),
195
        help="Set checker interval for client")
1055 by Teddy Hogeborn
mandos-ctl: Refactor
196
    approve_deny_default = parser.add_mutually_exclusive_group()
197
    approve_deny_default.add_argument(
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
198
        "--approve-by-default", dest="commands",
199
        action="append_const", default=[],
200
        const=command.ApproveByDefault(),
1055 by Teddy Hogeborn
mandos-ctl: Refactor
201
        help="Set client to be approved by default")
202
    approve_deny_default.add_argument(
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
203
        "--deny-by-default", dest="commands",
204
        action="append_const", default=[],
205
        const=command.DenyByDefault(),
1055 by Teddy Hogeborn
mandos-ctl: Refactor
206
        help="Set client to be denied by default")
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
207
    parser.add_argument(
208
        "--approval-delay", dest="commands", action="append",
209
        default=[], metavar="TIME",
210
        type=command.SetApprovalDelay.argparse(string_to_delta),
211
        help="Set delay before client approve/deny")
212
    parser.add_argument(
213
        "--approval-duration", dest="commands", action="append",
214
        default=[], metavar="TIME",
215
        type=command.SetApprovalDuration.argparse(string_to_delta),
216
        help="Set duration of one client approval")
217
    parser.add_argument("-H", "--host", dest="commands",
218
                        action="append", default=[], metavar="STRING",
219
                        type=command.SetHost,
220
                        help="Set host for client")
221
    parser.add_argument(
222
        "-s", "--secret", dest="commands", action="append",
223
        default=[], metavar="FILENAME",
224
        type=command.SetSecret.argparse(argparse.FileType(mode="rb")),
225
        help="Set password blob (file) for client")
1055 by Teddy Hogeborn
mandos-ctl: Refactor
226
    approve_deny = parser.add_mutually_exclusive_group()
227
    approve_deny.add_argument(
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
228
        "-A", "--approve", dest="commands", action="append_const",
229
        default=[], const=command.Approve(),
1055 by Teddy Hogeborn
mandos-ctl: Refactor
230
        help="Approve any current client request")
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
231
    approve_deny.add_argument("-D", "--deny", dest="commands",
232
                              action="append_const", default=[],
233
                              const=command.Deny(),
1055 by Teddy Hogeborn
mandos-ctl: Refactor
234
                              help="Deny any current client request")
235
    parser.add_argument("--debug", action="store_true",
236
                        help="Debug mode (show D-Bus commands)")
237
    parser.add_argument("--check", action="store_true",
238
                        help="Run self-test")
239
    parser.add_argument("client", nargs="*", help="Client name")
240
241
242
def string_to_delta(interval):
243
    """Parse a string and return a datetime.timedelta"""
244
245
    try:
246
        return rfc3339_duration_to_delta(interval)
247
    except ValueError as e:
248
        log.warning("%s - Parsing as pre-1.6.1 interval instead",
249
                    ' '.join(e.args))
250
    return parse_pre_1_6_1_interval(interval)
24.1.121 by Björn Påhlsson
mandos-ctl: Added support for all client calls
251
608 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
252
253
def rfc3339_duration_to_delta(duration):
609 by Teddy Hogeborn
* clients.conf: Convert all time intervals to new RFC 3339 syntax.
254
    """Parse an RFC 3339 "duration" and return a datetime.timedelta
872 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
255
1130 by Teddy Hogeborn
Update Python 3 compatibility
256
    >>> rfc3339_duration_to_delta("P7D") == datetime.timedelta(7)
257
    True
258
    >>> rfc3339_duration_to_delta("PT60S") == datetime.timedelta(0, 60)
259
    True
260
    >>> rfc3339_duration_to_delta("PT60M") == datetime.timedelta(hours=1)
261
    True
262
    >>> # 60 months
263
    >>> rfc3339_duration_to_delta("P60M") == datetime.timedelta(1680)
264
    True
265
    >>> rfc3339_duration_to_delta("PT24H") == datetime.timedelta(1)
266
    True
267
    >>> rfc3339_duration_to_delta("P1W") == datetime.timedelta(7)
268
    True
269
    >>> rfc3339_duration_to_delta("PT5M30S") == datetime.timedelta(0, 330)
270
    True
271
    >>> rfc3339_duration_to_delta("P1DT3M20S") == datetime.timedelta(1, 200)
272
    True
990 by Teddy Hogeborn
mandos-ctl (rfc3339_duration_to_delta): Improve tests
273
    >>> # Can not be empty:
274
    >>> rfc3339_duration_to_delta("")
275
    Traceback (most recent call last):
276
    ...
1076 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
277
    ValueError: Invalid RFC 3339 duration: ""
990 by Teddy Hogeborn
mandos-ctl (rfc3339_duration_to_delta): Improve tests
278
    >>> # Must start with "P":
279
    >>> rfc3339_duration_to_delta("1D")
280
    Traceback (most recent call last):
281
    ...
1076 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
282
    ValueError: Invalid RFC 3339 duration: "1D"
990 by Teddy Hogeborn
mandos-ctl (rfc3339_duration_to_delta): Improve tests
283
    >>> # Must use correct order
284
    >>> rfc3339_duration_to_delta("PT1S2M")
285
    Traceback (most recent call last):
286
    ...
1076 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
287
    ValueError: Invalid RFC 3339 duration: "PT1S2M"
990 by Teddy Hogeborn
mandos-ctl (rfc3339_duration_to_delta): Improve tests
288
    >>> # Time needs time marker
289
    >>> rfc3339_duration_to_delta("P1H2S")
290
    Traceback (most recent call last):
291
    ...
1076 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
292
    ValueError: Invalid RFC 3339 duration: "P1H2S"
990 by Teddy Hogeborn
mandos-ctl (rfc3339_duration_to_delta): Improve tests
293
    >>> # Weeks can not be combined with anything else
294
    >>> rfc3339_duration_to_delta("P1D2W")
295
    Traceback (most recent call last):
296
    ...
1076 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
297
    ValueError: Invalid RFC 3339 duration: "P1D2W"
990 by Teddy Hogeborn
mandos-ctl (rfc3339_duration_to_delta): Improve tests
298
    >>> rfc3339_duration_to_delta("P2W2H")
299
    Traceback (most recent call last):
300
    ...
1076 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
301
    ValueError: Invalid RFC 3339 duration: "P2W2H"
608 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
302
    """
872 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
303
609 by Teddy Hogeborn
* clients.conf: Convert all time intervals to new RFC 3339 syntax.
304
    # Parsing an RFC 3339 duration with regular expressions is not
608 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
305
    # possible - there would have to be multiple places for the same
609 by Teddy Hogeborn
* clients.conf: Convert all time intervals to new RFC 3339 syntax.
306
    # values, like seconds.  The current code, while more esoteric, is
307
    # cleaner without depending on a parsing library.  If Python had a
608 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
308
    # built-in library for parsing we would use it, but we'd like to
309
    # avoid excessive use of external libraries.
872 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
310
608 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
311
    # New type for defining tokens, syntax, and semantics all-in-one
753 by Teddy Hogeborn
mandos-ctl: Generate better messages in exceptions.
312
    Token = collections.namedtuple("Token", (
313
        "regexp",  # To match token; if "value" is not None, must have
314
                   # a "group" containing digits
315
        "value",   # datetime.timedelta or None
316
        "followers"))           # Tokens valid after this token
608 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
317
    # RFC 3339 "duration" tokens, syntax, and semantics; taken from
318
    # the "duration" ABNF definition in RFC 3339, Appendix A.
319
    token_end = Token(re.compile(r"$"), None, frozenset())
320
    token_second = Token(re.compile(r"(\d+)S"),
321
                         datetime.timedelta(seconds=1),
745 by Teddy Hogeborn
mandos-ctl: Do minor formatting and whitespace adjustments.
322
                         frozenset((token_end, )))
608 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
323
    token_minute = Token(re.compile(r"(\d+)M"),
324
                         datetime.timedelta(minutes=1),
325
                         frozenset((token_second, token_end)))
326
    token_hour = Token(re.compile(r"(\d+)H"),
327
                       datetime.timedelta(hours=1),
328
                       frozenset((token_minute, token_end)))
329
    token_time = Token(re.compile(r"T"),
330
                       None,
331
                       frozenset((token_hour, token_minute,
332
                                  token_second)))
333
    token_day = Token(re.compile(r"(\d+)D"),
334
                      datetime.timedelta(days=1),
335
                      frozenset((token_time, token_end)))
336
    token_month = Token(re.compile(r"(\d+)M"),
337
                        datetime.timedelta(weeks=4),
338
                        frozenset((token_day, token_end)))
339
    token_year = Token(re.compile(r"(\d+)Y"),
340
                       datetime.timedelta(weeks=52),
341
                       frozenset((token_month, token_end)))
342
    token_week = Token(re.compile(r"(\d+)W"),
343
                       datetime.timedelta(weeks=1),
745 by Teddy Hogeborn
mandos-ctl: Do minor formatting and whitespace adjustments.
344
                       frozenset((token_end, )))
608 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
345
    token_duration = Token(re.compile(r"P"), None,
346
                           frozenset((token_year, token_month,
347
                                      token_day, token_time,
721 by Teddy Hogeborn
Fix two mutually cancelling bugs.
348
                                      token_week)))
872 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
349
    # Define starting values:
350
    # Value so far
351
    value = datetime.timedelta()
608 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
352
    found_token = None
872 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
353
    # Following valid tokens
354
    followers = frozenset((token_duration, ))
355
    # String left to parse
356
    s = duration
608 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
357
    # Loop until end token is found
358
    while found_token is not token_end:
359
        # Search for any currently valid tokens
360
        for token in followers:
361
            match = token.regexp.match(s)
362
            if match is not None:
363
                # Token found
364
                if token.value is not None:
365
                    # Value found, parse digits
366
                    factor = int(match.group(1), 10)
367
                    # Add to value so far
368
                    value += factor * token.value
369
                # Strip token from string
370
                s = token.regexp.sub("", s, 1)
371
                # Go to found token
372
                found_token = token
373
                # Set valid next tokens
374
                followers = found_token.followers
375
                break
376
        else:
377
            # No currently valid tokens were found
1076 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
378
            raise ValueError("Invalid RFC 3339 duration: \"{}\""
753 by Teddy Hogeborn
mandos-ctl: Generate better messages in exceptions.
379
                             .format(duration))
608 by Teddy Hogeborn
* Makefile (check): Also check mandos-ctl.
380
    # End token found
381
    return value
382
383
991 by Teddy Hogeborn
mandos-ctl: Refactor and add more tests
384
def parse_pre_1_6_1_interval(interval):
1001 by Teddy Hogeborn
mandos-ctl: White space changes only
385
    """Parse an interval string as documented by Mandos before 1.6.1,
386
    and return a datetime.timedelta
387
1130 by Teddy Hogeborn
Update Python 3 compatibility
388
    >>> parse_pre_1_6_1_interval('7d') == datetime.timedelta(days=7)
389
    True
390
    >>> parse_pre_1_6_1_interval('60s') == datetime.timedelta(0, 60)
391
    True
392
    >>> parse_pre_1_6_1_interval('60m') == datetime.timedelta(hours=1)
393
    True
394
    >>> parse_pre_1_6_1_interval('24h') == datetime.timedelta(days=1)
395
    True
396
    >>> parse_pre_1_6_1_interval('1w') == datetime.timedelta(days=7)
397
    True
398
    >>> parse_pre_1_6_1_interval('5m 30s') == datetime.timedelta(0, 330)
399
    True
400
    >>> parse_pre_1_6_1_interval('') == datetime.timedelta(0)
401
    True
991 by Teddy Hogeborn
mandos-ctl: Refactor and add more tests
402
    >>> # Ignore unknown characters, allow any order and repetitions
1130 by Teddy Hogeborn
Update Python 3 compatibility
403
    >>> parse_pre_1_6_1_interval('2dxy7zz11y3m5m') == datetime.timedelta(2, 480, 18000)
404
    True
991 by Teddy Hogeborn
mandos-ctl: Refactor and add more tests
405
406
    """
872 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
407
616 by Teddy Hogeborn
* mandos-ctl (string_to_delta): Try to parse RFC 3339 duration before
408
    value = datetime.timedelta(0)
409
    regexp = re.compile(r"(\d+)([dsmhw]?)")
872 by Teddy Hogeborn
PEP8 compliance: mandos-ctl
410
518.2.9 by Teddy Hogeborn
* mandos (ClientDBus.approval_delay, ClientDBus.approval_duration,
411
    for num, suffix in regexp.findall(interval):
412
        if suffix == "d":
413
            value += datetime.timedelta(int(num))
414
        elif suffix == "s":
415
            value += datetime.timedelta(0, int(num))
416
        elif suffix == "m":
417
            value += datetime.timedelta(0, 0, 0, 0, int(num))
418
        elif suffix == "h":
419
            value += datetime.timedelta(0, 0, 0, 0, 0, int(num))
420
        elif suffix == "w":
421
            value += datetime.timedelta(0, 0, 0, 0, 0, 0, int(num))
422
        elif suffix == "":
423
            value += datetime.timedelta(0, 0, 0, int(num))
424
    return value
24.1.121 by Björn Påhlsson
mandos-ctl: Added support for all client calls
425
745 by Teddy Hogeborn
mandos-ctl: Do minor formatting and whitespace adjustments.
426
1055 by Teddy Hogeborn
mandos-ctl: Refactor
427
def check_option_syntax(parser, options):
428
    """Apply additional restrictions on options, not expressible in
429
argparse"""
430
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
431
    def has_commands(options, commands=None):
432
        if commands is None:
433
            commands = (command.Enable,
434
                        command.Disable,
435
                        command.BumpTimeout,
436
                        command.StartChecker,
437
                        command.StopChecker,
438
                        command.IsEnabled,
439
                        command.Remove,
440
                        command.SetChecker,
441
                        command.SetTimeout,
442
                        command.SetExtendedTimeout,
443
                        command.SetInterval,
444
                        command.ApproveByDefault,
445
                        command.DenyByDefault,
446
                        command.SetApprovalDelay,
447
                        command.SetApprovalDuration,
448
                        command.SetHost,
449
                        command.SetSecret,
450
                        command.Approve,
451
                        command.Deny)
452
        return any(isinstance(cmd, commands)
453
                   for cmd in options.commands)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
454
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
455
    if has_commands(options) and not (options.client or options.all):
1055 by Teddy Hogeborn
mandos-ctl: Refactor
456
        parser.error("Options require clients names or --all.")
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
457
    if options.verbose and has_commands(options):
1055 by Teddy Hogeborn
mandos-ctl: Refactor
458
        parser.error("--verbose can only be used alone.")
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
459
    if (has_commands(options, (command.DumpJSON,))
460
        and (options.verbose or len(options.commands) > 1)):
1055 by Teddy Hogeborn
mandos-ctl: Refactor
461
        parser.error("--dump-json can only be used alone.")
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
462
    if options.all and not has_commands(options):
1055 by Teddy Hogeborn
mandos-ctl: Refactor
463
        parser.error("--all requires an action.")
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
464
    if (has_commands(options, (command.IsEnabled,))
465
        and len(options.client) > 1):
1055 by Teddy Hogeborn
mandos-ctl: Refactor
466
        parser.error("--is-enabled requires exactly one client")
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
467
    if (len(options.commands) > 1
468
        and has_commands(options, (command.Remove,))
469
        and not has_commands(options, (command.Deny,))):
470
        parser.error("--remove can only be combined with --deny")
1055 by Teddy Hogeborn
mandos-ctl: Refactor
471
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
472
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
473
class dbus:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
474
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
475
    class SystemBus:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
476
477
        object_manager_iface = "org.freedesktop.DBus.ObjectManager"
478
        def get_managed_objects(self, busname, objectpath):
479
            return self.call_method("GetManagedObjects", busname,
480
                                    objectpath,
481
                                    self.object_manager_iface)
482
483
        properties_iface = "org.freedesktop.DBus.Properties"
484
        def set_property(self, busname, objectpath, interface, key,
485
                         value):
486
            self.call_method("Set", busname, objectpath,
487
                             self.properties_iface, interface, key,
488
                             value)
489
490
491
    class MandosBus(SystemBus):
492
        busname_domain = "se.recompile"
493
        busname = busname_domain + ".Mandos"
494
        server_path = "/"
495
        server_interface = busname_domain + ".Mandos"
496
        client_interface = busname_domain + ".Mandos.Client"
497
        del busname_domain
498
499
        def get_clients_and_properties(self):
500
            managed_objects = self.get_managed_objects(
501
                self.busname, self.server_path)
502
            return {objpath: properties[self.client_interface]
503
                    for objpath, properties in managed_objects.items()
504
                    if self.client_interface in properties}
505
506
        def set_client_property(self, objectpath, key, value):
507
            return self.set_property(self.busname, objectpath,
508
                                     self.client_interface, key,
509
                                     value)
510
511
        def call_client_method(self, objectpath, method, *args):
512
            return self.call_method(method, self.busname, objectpath,
513
                                    self.client_interface, *args)
514
515
        def call_server_method(self, method, *args):
516
            return self.call_method(method, self.busname,
517
                                    self.server_path,
518
                                    self.server_interface, *args)
519
520
    class Error(Exception):
521
        pass
522
523
    class ConnectFailed(Error):
524
        pass
525
526
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
527
class dbus_python_adapter:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
528
529
    class SystemBus(dbus.MandosBus):
530
        """Use dbus-python"""
531
532
        def __init__(self, module=dbus_python):
533
            self.dbus_python = module
534
            self.bus = self.dbus_python.SystemBus()
535
536
        @contextlib.contextmanager
537
        def convert_exception(self, exception_class=dbus.Error):
538
            try:
539
                yield
540
            except self.dbus_python.exceptions.DBusException as e:
541
                # This does what "raise from" would do
542
                exc = exception_class(*e.args)
543
                exc.__cause__ = e
544
                raise exc
545
546
        def call_method(self, methodname, busname, objectpath,
547
                        interface, *args):
548
            proxy_object = self.get_object(busname, objectpath)
549
            log.debug("D-Bus: %s:%s:%s.%s(%s)", busname, objectpath,
550
                      interface, methodname,
551
                      ", ".join(repr(a) for a in args))
552
            method = getattr(proxy_object, methodname)
553
            with self.convert_exception():
554
                with dbus_python_adapter.SilenceLogger(
555
                        "dbus.proxies"):
556
                    value = method(*args, dbus_interface=interface)
557
            return self.type_filter(value)
558
559
        def get_object(self, busname, objectpath):
560
            log.debug("D-Bus: Connect to: (busname=%r, path=%r)",
561
                      busname, objectpath)
562
            with self.convert_exception(dbus.ConnectFailed):
563
                return self.bus.get_object(busname, objectpath)
564
565
        def type_filter(self, value):
566
            """Convert the most bothersome types to Python types"""
567
            if isinstance(value, self.dbus_python.Boolean):
568
                return bool(value)
569
            if isinstance(value, self.dbus_python.ObjectPath):
570
                return str(value)
571
            # Also recurse into dictionaries
572
            if isinstance(value, self.dbus_python.Dictionary):
573
                return {self.type_filter(key):
574
                        self.type_filter(subval)
575
                        for key, subval in value.items()}
576
            return value
577
1098 by Teddy Hogeborn
mandos-ctl: Fix --secret when using the dbus-python module
578
        def set_client_property(self, objectpath, key, value):
579
            if key == "Secret":
580
                if not isinstance(value, bytes):
581
                    value = value.encode("utf-8")
582
                value = self.dbus_python.ByteArray(value)
583
            return self.set_property(self.busname, objectpath,
584
                                     self.client_interface, key,
585
                                     value)
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
586
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
587
    class SilenceLogger:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
588
        "Simple context manager to silence a particular logger"
589
        def __init__(self, loggername):
590
            self.logger = logging.getLogger(loggername)
591
592
        def __enter__(self):
593
            self.logger.addFilter(self.nullfilter)
594
595
        class NullFilter(logging.Filter):
596
            def filter(self, record):
597
                return False
598
599
        nullfilter = NullFilter()
600
601
        def __exit__(self, exc_type, exc_val, exc_tb):
602
            self.logger.removeFilter(self.nullfilter)
603
604
605
    class CachingBus(SystemBus):
606
        """A caching layer for dbus_python_adapter.SystemBus"""
607
        def __init__(self, *args, **kwargs):
608
            self.object_cache = {}
609
            super(dbus_python_adapter.CachingBus,
610
                  self).__init__(*args, **kwargs)
611
        def get_object(self, busname, objectpath):
612
            try:
613
                return self.object_cache[(busname, objectpath)]
614
            except KeyError:
615
                new_object = super(
616
                    dbus_python_adapter.CachingBus,
617
                    self).get_object(busname, objectpath)
618
                self.object_cache[(busname, objectpath)]  = new_object
619
                return new_object
1066 by Teddy Hogeborn
mandos-ctl: Refactor
620
621
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
622
class pydbus_adapter:
1096 by Teddy Hogeborn
mandos-ctl: Add support for D-Bus module "pydbus"
623
    class SystemBus(dbus.MandosBus):
624
        def __init__(self, module=pydbus):
625
            self.pydbus = module
626
            self.bus = self.pydbus.SystemBus()
627
628
        @contextlib.contextmanager
629
        def convert_exception(self, exception_class=dbus.Error):
630
            try:
631
                yield
632
            except gi.repository.GLib.Error as e:
633
                # This does what "raise from" would do
634
                exc = exception_class(*e.args)
635
                exc.__cause__ = e
636
                raise exc
637
638
        def call_method(self, methodname, busname, objectpath,
639
                        interface, *args):
640
            proxy_object = self.get(busname, objectpath)
641
            log.debug("D-Bus: %s:%s:%s.%s(%s)", busname, objectpath,
642
                      interface, methodname,
643
                      ", ".join(repr(a) for a in args))
644
            method = getattr(proxy_object[interface], methodname)
645
            with self.convert_exception():
646
                return method(*args)
647
648
        def get(self, busname, objectpath):
649
            log.debug("D-Bus: Connect to: (busname=%r, path=%r)",
650
                      busname, objectpath)
651
            with self.convert_exception(dbus.ConnectFailed):
652
                if sys.version_info.major <= 2:
653
                    with warnings.catch_warnings():
654
                        warnings.filterwarnings(
655
                            "ignore", "", DeprecationWarning,
656
                            r"^xml\.etree\.ElementTree$")
657
                        return self.bus.get(busname, objectpath)
658
                else:
659
                    return self.bus.get(busname, objectpath)
660
661
        def set_property(self, busname, objectpath, interface, key,
662
                         value):
663
            proxy_object = self.get(busname, objectpath)
664
            log.debug("D-Bus: %s:%s:%s.Set(%r, %r, %r)", busname,
665
                      objectpath, self.properties_iface, interface,
666
                      key, value)
667
            setattr(proxy_object[interface], key, value)
668
669
    class CachingBus(SystemBus):
670
        """A caching layer for pydbus_adapter.SystemBus"""
671
        def __init__(self, *args, **kwargs):
672
            self.object_cache = {}
673
            super(pydbus_adapter.CachingBus,
674
                  self).__init__(*args, **kwargs)
675
        def get(self, busname, objectpath):
676
            try:
677
                return self.object_cache[(busname, objectpath)]
678
            except KeyError:
679
                new_object = (super(pydbus_adapter.CachingBus, self)
680
                              .get(busname, objectpath))
681
                self.object_cache[(busname, objectpath)]  = new_object
682
                return new_object
683
684
1055 by Teddy Hogeborn
mandos-ctl: Refactor
685
def commands_from_options(options):
686
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
687
    commands = list(options.commands)
688
689
    def find_cmd(cmd, commands):
690
        i = 0
691
        for i, c in enumerate(commands):
692
            if isinstance(c, cmd):
693
                return i
694
        return i+1
695
696
    # If command.Remove is present, move any instances of command.Deny
697
    # to occur ahead of command.Remove.
698
    index_of_remove = find_cmd(command.Remove, commands)
699
    before_remove = commands[:index_of_remove]
700
    after_remove = commands[index_of_remove:]
701
    cleaned_after = []
702
    for cmd in after_remove:
703
        if isinstance(cmd, command.Deny):
704
            before_remove.append(cmd)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
705
        else:
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
706
            cleaned_after.append(cmd)
707
    if cleaned_after != after_remove:
708
        commands = before_remove + cleaned_after
1055 by Teddy Hogeborn
mandos-ctl: Refactor
709
710
    # If no command option has been given, show table of clients,
711
    # optionally verbosely
712
    if not commands:
1077 by Teddy Hogeborn
mandos-ctl: Refactor
713
        commands.append(command.PrintTable(verbose=options.verbose))
1055 by Teddy Hogeborn
mandos-ctl: Refactor
714
715
    return commands
716
717
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
718
class command:
1077 by Teddy Hogeborn
mandos-ctl: Refactor
719
    """A namespace for command classes"""
720
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
721
    class Base:
1077 by Teddy Hogeborn
mandos-ctl: Refactor
722
        """Abstract base class for commands"""
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
723
        def run(self, clients, bus=None):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
724
            """Normal commands should implement run_on_one_client(),
725
but commands which want to operate on all clients at the same time can
726
override this run() method instead.
727
"""
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
728
            self.bus = bus
729
            for client, properties in clients.items():
1077 by Teddy Hogeborn
mandos-ctl: Refactor
730
                self.run_on_one_client(client, properties)
731
732
733
    class IsEnabled(Base):
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
734
        def run(self, clients, bus=None):
735
            properties = next(iter(clients.values()))
736
            if properties["Enabled"]:
1077 by Teddy Hogeborn
mandos-ctl: Refactor
737
                sys.exit(0)
738
            sys.exit(1)
739
740
741
    class Approve(Base):
742
        def run_on_one_client(self, client, properties):
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
743
            self.bus.call_client_method(client, "Approve", True)
1077 by Teddy Hogeborn
mandos-ctl: Refactor
744
745
746
    class Deny(Base):
747
        def run_on_one_client(self, client, properties):
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
748
            self.bus.call_client_method(client, "Approve", False)
1077 by Teddy Hogeborn
mandos-ctl: Refactor
749
750
751
    class Remove(Base):
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
752
        def run(self, clients, bus):
753
            for clientpath in frozenset(clients.keys()):
754
                bus.call_server_method("RemoveClient", clientpath)
1077 by Teddy Hogeborn
mandos-ctl: Refactor
755
756
757
    class Output(Base):
758
        """Abstract class for commands outputting client details"""
759
        all_keywords = ("Name", "Enabled", "Timeout", "LastCheckedOK",
760
                        "Created", "Interval", "Host", "KeyID",
761
                        "Fingerprint", "CheckerRunning",
762
                        "LastEnabled", "ApprovalPending",
763
                        "ApprovedByDefault", "LastApprovalRequest",
764
                        "ApprovalDelay", "ApprovalDuration",
765
                        "Checker", "ExtendedTimeout", "Expires",
766
                        "LastCheckerStatus")
767
768
769
    class DumpJSON(Output):
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
770
        def run(self, clients, bus=None):
771
            data = {properties["Name"]:
772
                    {key: properties[key]
1077 by Teddy Hogeborn
mandos-ctl: Refactor
773
                     for key in self.all_keywords}
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
774
                    for properties in clients.values()}
1085 by Teddy Hogeborn
mandos-ctl: Refactor
775
            print(json.dumps(data, indent=4, separators=(',', ': ')))
1077 by Teddy Hogeborn
mandos-ctl: Refactor
776
777
778
    class PrintTable(Output):
779
        def __init__(self, verbose=False):
780
            self.verbose = verbose
781
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
782
        def run(self, clients, bus=None):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
783
            default_keywords = ("Name", "Enabled", "Timeout",
784
                                "LastCheckedOK")
785
            keywords = default_keywords
786
            if self.verbose:
787
                keywords = self.all_keywords
1085 by Teddy Hogeborn
mandos-ctl: Refactor
788
            print(self.TableOfClients(clients.values(), keywords))
1077 by Teddy Hogeborn
mandos-ctl: Refactor
789
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
790
        class TableOfClients:
1077 by Teddy Hogeborn
mandos-ctl: Refactor
791
            tableheaders = {
792
                "Name": "Name",
793
                "Enabled": "Enabled",
794
                "Timeout": "Timeout",
795
                "LastCheckedOK": "Last Successful Check",
796
                "LastApprovalRequest": "Last Approval Request",
797
                "Created": "Created",
798
                "Interval": "Interval",
799
                "Host": "Host",
800
                "Fingerprint": "Fingerprint",
801
                "KeyID": "Key ID",
802
                "CheckerRunning": "Check Is Running",
803
                "LastEnabled": "Last Enabled",
804
                "ApprovalPending": "Approval Is Pending",
805
                "ApprovedByDefault": "Approved By Default",
806
                "ApprovalDelay": "Approval Delay",
807
                "ApprovalDuration": "Approval Duration",
808
                "Checker": "Checker",
809
                "ExtendedTimeout": "Extended Timeout",
810
                "Expires": "Expires",
811
                "LastCheckerStatus": "Last Checker Status",
812
            }
813
814
            def __init__(self, clients, keywords):
815
                self.clients = clients
816
                self.keywords = keywords
817
1011 by Teddy Hogeborn
mandos-ctl: Refactor; move TableOfClients into PrintTableCmd
818
            def __str__(self):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
819
                return "\n".join(self.rows())
820
821
            if sys.version_info.major == 2:
822
                __unicode__ = __str__
823
                def __str__(self):
824
                    return str(self).encode(
825
                        locale.getpreferredencoding())
826
827
            def rows(self):
828
                format_string = self.row_formatting_string()
829
                rows = [self.header_line(format_string)]
830
                rows.extend(self.client_line(client, format_string)
831
                            for client in self.clients)
832
                return rows
833
834
            def row_formatting_string(self):
835
                "Format string used to format table rows"
836
                return " ".join("{{{key}:{width}}}".format(
837
                    width=max(len(self.tableheaders[key]),
838
                              *(len(self.string_from_client(client,
839
                                                            key))
840
                                for client in self.clients)),
841
                    key=key)
842
                                for key in self.keywords)
843
844
            def string_from_client(self, client, key):
845
                return self.valuetostring(client[key], key)
846
847
            @classmethod
848
            def valuetostring(cls, value, keyword):
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
849
                if isinstance(value, bool):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
850
                    return "Yes" if value else "No"
851
                if keyword in ("Timeout", "Interval", "ApprovalDelay",
852
                               "ApprovalDuration", "ExtendedTimeout"):
853
                    return cls.milliseconds_to_string(value)
854
                return str(value)
855
856
            def header_line(self, format_string):
857
                return format_string.format(**self.tableheaders)
858
859
            def client_line(self, client, format_string):
860
                return format_string.format(
861
                    **{key: self.string_from_client(client, key)
862
                       for key in self.keywords})
863
864
            @staticmethod
865
            def milliseconds_to_string(ms):
866
                td = datetime.timedelta(0, 0, 0, ms)
867
                return ("{days}{hours:02}:{minutes:02}:{seconds:02}"
868
                        .format(days="{}T".format(td.days)
869
                                if td.days else "",
870
                                hours=td.seconds // 3600,
871
                                minutes=(td.seconds % 3600) // 60,
872
                                seconds=td.seconds % 60))
873
874
1087 by Teddy Hogeborn
mandos-ctl: Refactor
875
    class PropertySetter(Base):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
876
        "Abstract class for Actions for setting one client property"
877
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
878
        def run_on_one_client(self, client, properties=None):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
879
            """Set the Client's D-Bus property"""
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
880
            self.bus.set_client_property(client, self.propname,
881
                                         self.value_to_set)
1077 by Teddy Hogeborn
mandos-ctl: Refactor
882
883
        @property
884
        def propname(self):
885
            raise NotImplementedError()
886
887
1087 by Teddy Hogeborn
mandos-ctl: Refactor
888
    class Enable(PropertySetter):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
889
        propname = "Enabled"
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
890
        value_to_set = True
1077 by Teddy Hogeborn
mandos-ctl: Refactor
891
892
1087 by Teddy Hogeborn
mandos-ctl: Refactor
893
    class Disable(PropertySetter):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
894
        propname = "Enabled"
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
895
        value_to_set = False
1077 by Teddy Hogeborn
mandos-ctl: Refactor
896
897
1087 by Teddy Hogeborn
mandos-ctl: Refactor
898
    class BumpTimeout(PropertySetter):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
899
        propname = "LastCheckedOK"
900
        value_to_set = ""
901
902
1087 by Teddy Hogeborn
mandos-ctl: Refactor
903
    class StartChecker(PropertySetter):
904
        propname = "CheckerRunning"
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
905
        value_to_set = True
1087 by Teddy Hogeborn
mandos-ctl: Refactor
906
907
908
    class StopChecker(PropertySetter):
909
        propname = "CheckerRunning"
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
910
        value_to_set = False
1087 by Teddy Hogeborn
mandos-ctl: Refactor
911
912
913
    class ApproveByDefault(PropertySetter):
914
        propname = "ApprovedByDefault"
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
915
        value_to_set = True
1087 by Teddy Hogeborn
mandos-ctl: Refactor
916
917
918
    class DenyByDefault(PropertySetter):
919
        propname = "ApprovedByDefault"
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
920
        value_to_set = False
1087 by Teddy Hogeborn
mandos-ctl: Refactor
921
922
923
    class PropertySetterValue(PropertySetter):
924
        """Abstract class for PropertySetter recieving a value as
925
constructor argument instead of a class attribute."""
1077 by Teddy Hogeborn
mandos-ctl: Refactor
926
        def __init__(self, value):
927
            self.value_to_set = value
928
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
929
        @classmethod
930
        def argparse(cls, argtype):
931
            def cmdtype(arg):
932
                return cls(argtype(arg))
933
            return cmdtype
1077 by Teddy Hogeborn
mandos-ctl: Refactor
934
1087 by Teddy Hogeborn
mandos-ctl: Refactor
935
    class SetChecker(PropertySetterValue):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
936
        propname = "Checker"
937
938
1087 by Teddy Hogeborn
mandos-ctl: Refactor
939
    class SetHost(PropertySetterValue):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
940
        propname = "Host"
941
942
1087 by Teddy Hogeborn
mandos-ctl: Refactor
943
    class SetSecret(PropertySetterValue):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
944
        propname = "Secret"
945
946
        @property
947
        def value_to_set(self):
948
            return self._vts
949
950
        @value_to_set.setter
951
        def value_to_set(self, value):
952
            """When setting, read data from supplied file object"""
953
            self._vts = value.read()
954
            value.close()
955
956
1087 by Teddy Hogeborn
mandos-ctl: Refactor
957
    class PropertySetterValueMilliseconds(PropertySetterValue):
958
        """Abstract class for PropertySetterValue taking a value
959
argument as a datetime.timedelta() but should store it as
960
milliseconds."""
1061 by Teddy Hogeborn
mandos-ctl: White space changes only
961
1077 by Teddy Hogeborn
mandos-ctl: Refactor
962
        @property
963
        def value_to_set(self):
964
            return self._vts
965
966
        @value_to_set.setter
967
        def value_to_set(self, value):
968
            "When setting, convert value from a datetime.timedelta"
969
            self._vts = int(round(value.total_seconds() * 1000))
970
971
1087 by Teddy Hogeborn
mandos-ctl: Refactor
972
    class SetTimeout(PropertySetterValueMilliseconds):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
973
        propname = "Timeout"
974
975
1087 by Teddy Hogeborn
mandos-ctl: Refactor
976
    class SetExtendedTimeout(PropertySetterValueMilliseconds):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
977
        propname = "ExtendedTimeout"
978
979
1087 by Teddy Hogeborn
mandos-ctl: Refactor
980
    class SetInterval(PropertySetterValueMilliseconds):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
981
        propname = "Interval"
982
983
1087 by Teddy Hogeborn
mandos-ctl: Refactor
984
    class SetApprovalDelay(PropertySetterValueMilliseconds):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
985
        propname = "ApprovalDelay"
986
987
1087 by Teddy Hogeborn
mandos-ctl: Refactor
988
    class SetApprovalDuration(PropertySetterValueMilliseconds):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
989
        propname = "ApprovalDuration"
1003 by Teddy Hogeborn
mandos-ctl: Separate determining what to do and actually doing it
990
24.1.163 by Björn Påhlsson
mandos-client: Added never ending loop for --connect
991
984 by Teddy Hogeborn
Make mandos-ctl use unittest instead of doctest module
992

1072 by Teddy Hogeborn
mandos-ctl: Refactor tests
993
class TestCaseWithAssertLogs(unittest.TestCase):
994
    """unittest.TestCase.assertLogs only exists in Python 3.4"""
995
996
    if not hasattr(unittest.TestCase, "assertLogs"):
997
        @contextlib.contextmanager
998
        def assertLogs(self, logger, level=logging.INFO):
999
            capturing_handler = self.CapturingLevelHandler(level)
1000
            old_level = logger.level
1001
            old_propagate = logger.propagate
1002
            logger.addHandler(capturing_handler)
1003
            logger.setLevel(level)
1004
            logger.propagate = False
1005
            try:
1006
                yield capturing_handler.watcher
1007
            finally:
1008
                logger.propagate = old_propagate
1009
                logger.removeHandler(capturing_handler)
1010
                logger.setLevel(old_level)
1011
            self.assertGreater(len(capturing_handler.watcher.records),
1012
                               0)
1013
1014
        class CapturingLevelHandler(logging.Handler):
1015
            def __init__(self, level, *args, **kwargs):
1016
                logging.Handler.__init__(self, *args, **kwargs)
1017
                self.watcher = self.LoggingWatcher([], [])
1018
            def emit(self, record):
1019
                self.watcher.records.append(record)
1020
                self.watcher.output.append(self.format(record))
1021
1022
            LoggingWatcher = collections.namedtuple("LoggingWatcher",
1023
                                                    ("records",
1024
                                                     "output"))
1025
1076 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
1026
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1027
class Unique:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1028
    """Class for objects which exist only to be unique objects, since
1029
unittest.mock.sentinel only exists in Python 3.3"""
1030
1031
1072 by Teddy Hogeborn
mandos-ctl: Refactor tests
1032
class Test_string_to_delta(TestCaseWithAssertLogs):
1080 by Teddy Hogeborn
mandos-ctl: Refactor tests
1033
    # Just test basic RFC 3339 functionality here, the doc string for
1034
    # rfc3339_duration_to_delta() already has more comprehensive
1091 by Teddy Hogeborn
mandos-ctl: Fix grammar.
1035
    # tests, which are run by doctest.
1080 by Teddy Hogeborn
mandos-ctl: Refactor tests
1036
1037
    def test_rfc3339_zero_seconds(self):
1082 by Teddy Hogeborn
mandos-ctl: Refactor tests
1038
        self.assertEqual(datetime.timedelta(),
1039
                         string_to_delta("PT0S"))
1080 by Teddy Hogeborn
mandos-ctl: Refactor tests
1040
1041
    def test_rfc3339_zero_days(self):
1082 by Teddy Hogeborn
mandos-ctl: Refactor tests
1042
        self.assertEqual(datetime.timedelta(), string_to_delta("P0D"))
1080 by Teddy Hogeborn
mandos-ctl: Refactor tests
1043
1044
    def test_rfc3339_one_second(self):
1082 by Teddy Hogeborn
mandos-ctl: Refactor tests
1045
        self.assertEqual(datetime.timedelta(0, 1),
1046
                         string_to_delta("PT1S"))
1080 by Teddy Hogeborn
mandos-ctl: Refactor tests
1047
1048
    def test_rfc3339_two_hours(self):
1082 by Teddy Hogeborn
mandos-ctl: Refactor tests
1049
        self.assertEqual(datetime.timedelta(0, 7200),
1050
                         string_to_delta("PT2H"))
1061 by Teddy Hogeborn
mandos-ctl: White space changes only
1051
992 by Teddy Hogeborn
mandos-ctl: Add more tests
1052
    def test_falls_back_to_pre_1_6_1_with_warning(self):
1072 by Teddy Hogeborn
mandos-ctl: Refactor tests
1053
        with self.assertLogs(log, logging.WARNING):
1054
            value = string_to_delta("2h")
1082 by Teddy Hogeborn
mandos-ctl: Refactor tests
1055
        self.assertEqual(datetime.timedelta(0, 7200), value)
992 by Teddy Hogeborn
mandos-ctl: Add more tests
1056
1010 by Teddy Hogeborn
mandos-ctl: Refactor; test PrintTableCmd instead of TableOfClients
1057
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1058
class Test_check_option_syntax(unittest.TestCase):
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1059
    def setUp(self):
1060
        self.parser = argparse.ArgumentParser()
1061
        add_command_line_options(self.parser)
1062
1063
    def test_actions_requires_client_or_all(self):
1064
        for action, value in self.actions.items():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1065
            args = self.actionargs(action, value)
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1066
            with self.assertParseError():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1067
                self.parse_args(args)
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1068
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1069
    # This mostly corresponds to the definition from has_commands() in
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1070
    # check_option_syntax()
1071
    actions = {
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1072
        "--enable": None,
1073
        "--disable": None,
1074
        "--bump-timeout": None,
1075
        "--start-checker": None,
1076
        "--stop-checker": None,
1077
        "--is-enabled": None,
1078
        "--remove": None,
1079
        "--checker": "x",
1080
        "--timeout": "PT0S",
1081
        "--extended-timeout": "PT0S",
1082
        "--interval": "PT0S",
1083
        "--approve-by-default": None,
1084
        "--deny-by-default": None,
1085
        "--approval-delay": "PT0S",
1086
        "--approval-duration": "PT0S",
1087
        "--host": "hostname",
1088
        "--secret": "/dev/null",
1089
        "--approve": None,
1090
        "--deny": None,
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1091
    }
1092
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1093
    @staticmethod
1094
    def actionargs(action, value, *args):
1095
        if value is not None:
1096
            return [action, value] + list(args)
1097
        else:
1098
            return [action] + list(args)
1099
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1100
    @contextlib.contextmanager
1101
    def assertParseError(self):
1102
        with self.assertRaises(SystemExit) as e:
1076 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
1103
            with self.redirect_stderr_to_devnull():
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1104
                yield
1105
        # Exit code from argparse is guaranteed to be "2".  Reference:
1106
        # https://docs.python.org/3/library
1107
        # /argparse.html#exiting-methods
1082 by Teddy Hogeborn
mandos-ctl: Refactor tests
1108
        self.assertEqual(2, e.exception.code)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1109
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1110
    def parse_args(self, args):
1111
        options = self.parser.parse_args(args)
1112
        check_option_syntax(self.parser, options)
1113
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1114
    @staticmethod
1115
    @contextlib.contextmanager
1076 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
1116
    def redirect_stderr_to_devnull():
1083 by Teddy Hogeborn
mandos-ctl: Refactor tests
1117
        old_stderr = sys.stderr
1118
        with contextlib.closing(open(os.devnull, "w")) as null:
1119
            sys.stderr = null
1120
            try:
1121
                yield
1122
            finally:
1123
                sys.stderr = old_stderr
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1124
1125
    def check_option_syntax(self, options):
1126
        check_option_syntax(self.parser, options)
1127
1075 by Teddy Hogeborn
mandos-ctl: Refactor and fix bug in tests.
1128
    def test_actions_all_conflicts_with_verbose(self):
1129
        for action, value in self.actions.items():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1130
            args = self.actionargs(action, value, "--all",
1131
                                   "--verbose")
1075 by Teddy Hogeborn
mandos-ctl: Refactor and fix bug in tests.
1132
            with self.assertParseError():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1133
                self.parse_args(args)
1075 by Teddy Hogeborn
mandos-ctl: Refactor and fix bug in tests.
1134
1135
    def test_actions_with_client_conflicts_with_verbose(self):
1136
        for action, value in self.actions.items():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1137
            args = self.actionargs(action, value, "--verbose",
1138
                                   "client")
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1139
            with self.assertParseError():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1140
                self.parse_args(args)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1141
1142
    def test_dump_json_conflicts_with_verbose(self):
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1143
        args = ["--dump-json", "--verbose"]
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1144
        with self.assertParseError():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1145
            self.parse_args(args)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1146
1147
    def test_dump_json_conflicts_with_action(self):
1148
        for action, value in self.actions.items():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1149
            args = self.actionargs(action, value, "--dump-json")
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1150
            with self.assertParseError():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1151
                self.parse_args(args)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1152
1153
    def test_all_can_not_be_alone(self):
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1154
        args = ["--all"]
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1155
        with self.assertParseError():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1156
            self.parse_args(args)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1157
1158
    def test_all_is_ok_with_any_action(self):
1159
        for action, value in self.actions.items():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1160
            args = self.actionargs(action, value, "--all")
1161
            self.parse_args(args)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1162
1075 by Teddy Hogeborn
mandos-ctl: Refactor and fix bug in tests.
1163
    def test_any_action_is_ok_with_one_client(self):
1164
        for action, value in self.actions.items():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1165
            args = self.actionargs(action, value, "client")
1166
            self.parse_args(args)
1075 by Teddy Hogeborn
mandos-ctl: Refactor and fix bug in tests.
1167
1076 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
1168
    def test_one_client_with_all_actions_except_is_enabled(self):
1169
        for action, value in self.actions.items():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1170
            if action == "--is-enabled":
1076 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
1171
                continue
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1172
            args = self.actionargs(action, value, "client")
1173
            self.parse_args(args)
1076 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
1174
1175
    def test_two_clients_with_all_actions_except_is_enabled(self):
1176
        for action, value in self.actions.items():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1177
            if action == "--is-enabled":
1076 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
1178
                continue
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1179
            args = self.actionargs(action, value, "client1",
1180
                                   "client2")
1181
            self.parse_args(args)
1076 by Teddy Hogeborn
mandos-ctl: Refactor and add a few more tests
1182
1183
    def test_two_clients_are_ok_with_actions_except_is_enabled(self):
1075 by Teddy Hogeborn
mandos-ctl: Refactor and fix bug in tests.
1184
        for action, value in self.actions.items():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1185
            if action == "--is-enabled":
1075 by Teddy Hogeborn
mandos-ctl: Refactor and fix bug in tests.
1186
                continue
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1187
            args = self.actionargs(action, value, "client1",
1188
                                   "client2")
1189
            self.parse_args(args)
1075 by Teddy Hogeborn
mandos-ctl: Refactor and fix bug in tests.
1190
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1191
    def test_is_enabled_fails_without_client(self):
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1192
        args = ["--is-enabled"]
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1193
        with self.assertParseError():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1194
            self.parse_args(args)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1195
1196
    def test_is_enabled_fails_with_two_clients(self):
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1197
        args = ["--is-enabled", "client1", "client2"]
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1198
        with self.assertParseError():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1199
            self.parse_args(args)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1200
1201
    def test_remove_can_only_be_combined_with_action_deny(self):
1202
        for action, value in self.actions.items():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1203
            if action in {"--remove", "--deny"}:
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1204
                continue
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1205
            args = self.actionargs(action, value, "--all",
1206
                                   "--remove")
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1207
            with self.assertParseError():
1094 by Teddy Hogeborn
mandos-ctl: Refactor: Move command list generation into argparse
1208
                self.parse_args(args)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1209
1210
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1211
class Test_dbus_exceptions(unittest.TestCase):
1212
1213
    def test_dbus_ConnectFailed_is_Error(self):
1214
        with self.assertRaises(dbus.Error):
1215
            raise dbus.ConnectFailed()
1216
1217
1218
class Test_dbus_MandosBus(unittest.TestCase):
1219
1220
    class MockMandosBus(dbus.MandosBus):
1221
        def __init__(self):
1222
            self._name = "se.recompile.Mandos"
1223
            self._server_path = "/"
1224
            self._server_interface = "se.recompile.Mandos"
1225
            self._client_interface = "se.recompile.Mandos.Client"
1226
            self.calls = []
1227
            self.call_method_return = Unique()
1228
1229
        def call_method(self, methodname, busname, objectpath,
1230
                        interface, *args):
1231
            self.calls.append((methodname, busname, objectpath,
1232
                               interface, args))
1233
            return self.call_method_return
1234
1235
    def setUp(self):
1236
        self.bus = self.MockMandosBus()
1237
1238
    def test_set_client_property(self):
1239
        self.bus.set_client_property("objectpath", "key", "value")
1240
        expected_call = ("Set", self.bus._name, "objectpath",
1241
                         "org.freedesktop.DBus.Properties",
1242
                         (self.bus._client_interface, "key", "value"))
1243
        self.assertIn(expected_call, self.bus.calls)
1244
1245
    def test_call_client_method(self):
1246
        ret = self.bus.call_client_method("objectpath", "methodname")
1247
        self.assertIs(self.bus.call_method_return, ret)
1248
        expected_call = ("methodname", self.bus._name, "objectpath",
1249
                         self.bus._client_interface, ())
1250
        self.assertIn(expected_call, self.bus.calls)
1251
1252
    def test_call_client_method_with_args(self):
1253
        args = (Unique(), Unique())
1254
        ret = self.bus.call_client_method("objectpath", "methodname",
1255
                                          *args)
1256
        self.assertIs(self.bus.call_method_return, ret)
1257
        expected_call = ("methodname", self.bus._name, "objectpath",
1258
                         self.bus._client_interface,
1259
                         (args[0], args[1]))
1260
        self.assertIn(expected_call, self.bus.calls)
1261
1262
    def test_get_clients_and_properties(self):
1263
        managed_objects = {
1264
            "objectpath": {
1265
                self.bus._client_interface: {
1266
                    "key": "value",
1267
                    "bool": True,
1268
                },
1269
                "irrelevant_interface": {
1270
                    "key": "othervalue",
1271
                    "bool": False,
1272
                },
1273
            },
1274
            "other_objectpath": {
1275
                "other_irrelevant_interface": {
1276
                    "key": "value 3",
1277
                    "bool": None,
1278
                },
1279
            },
1280
        }
1281
        expected_clients_and_properties = {
1282
            "objectpath": {
1283
                "key": "value",
1284
                "bool": True,
1285
            }
1286
        }
1287
        self.bus.call_method_return = managed_objects
1288
        ret = self.bus.get_clients_and_properties()
1289
        self.assertDictEqual(expected_clients_and_properties, ret)
1290
        expected_call = ("GetManagedObjects", self.bus._name,
1291
                         self.bus._server_path,
1292
                         "org.freedesktop.DBus.ObjectManager", ())
1293
        self.assertIn(expected_call, self.bus.calls)
1294
1295
    def test_call_server_method(self):
1296
        ret = self.bus.call_server_method("methodname")
1297
        self.assertIs(self.bus.call_method_return, ret)
1298
        expected_call = ("methodname", self.bus._name,
1299
                         self.bus._server_path,
1300
                         self.bus._server_interface, ())
1301
        self.assertIn(expected_call, self.bus.calls)
1302
1303
    def test_call_server_method_with_args(self):
1304
        args = (Unique(), Unique())
1305
        ret = self.bus.call_server_method("methodname", *args)
1306
        self.assertIs(self.bus.call_method_return, ret)
1307
        expected_call = ("methodname", self.bus._name,
1308
                         self.bus._server_path,
1309
                         self.bus._server_interface,
1310
                         (args[0], args[1]))
1311
        self.assertIn(expected_call, self.bus.calls)
1312
1313
1314
class Test_dbus_python_adapter_SystemBus(TestCaseWithAssertLogs):
1315
1316
    def MockDBusPython_func(self, func):
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1317
        class mock_dbus_python:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1318
            """mock dbus-python module"""
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1319
            class exceptions:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1320
                """Pseudo-namespace"""
1321
                class DBusException(Exception):
1322
                    pass
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1323
            class SystemBus:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1324
                @staticmethod
1325
                def get_object(busname, objectpath):
1326
                    DBusObject = collections.namedtuple(
1098 by Teddy Hogeborn
mandos-ctl: Fix --secret when using the dbus-python module
1327
                        "DBusObject", ("methodname", "Set"))
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1328
                    def method(*args, **kwargs):
1329
                        self.assertEqual({"dbus_interface":
1330
                                          "interface"},
1331
                                         kwargs)
1332
                        return func(*args)
1098 by Teddy Hogeborn
mandos-ctl: Fix --secret when using the dbus-python module
1333
                    def set_property(interface, key, value,
1334
                                     dbus_interface=None):
1335
                        self.assertEqual(
1336
                            "org.freedesktop.DBus.Properties",
1337
                            dbus_interface)
1338
                        self.assertEqual("Secret", key)
1339
                        return func(interface, key, value,
1340
                                    dbus_interface=dbus_interface)
1341
                    return DBusObject(methodname=method,
1342
                                      Set=set_property)
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1343
            class Boolean:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1344
                def __init__(self, value):
1345
                    self.value = bool(value)
1346
                def __bool__(self):
1347
                    return self.value
1348
                if sys.version_info.major == 2:
1349
                    __nonzero__ = __bool__
1350
            class ObjectPath(str):
1351
                pass
1352
            class Dictionary(dict):
1353
                pass
1098 by Teddy Hogeborn
mandos-ctl: Fix --secret when using the dbus-python module
1354
            class ByteArray(bytes):
1355
                pass
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1356
        return mock_dbus_python
1357
1358
    def call_method(self, bus, methodname, busname, objectpath,
1359
                    interface, *args):
1360
        with self.assertLogs(log, logging.DEBUG):
1361
            return bus.call_method(methodname, busname, objectpath,
1362
                                   interface, *args)
1363
1364
    def test_call_method_returns(self):
1365
        expected_method_return = Unique()
1366
        method_args = (Unique(), Unique())
1367
        def func(*args):
1368
            self.assertEqual(len(method_args), len(args))
1369
            for marg, arg in zip(method_args, args):
1370
                self.assertIs(marg, arg)
1371
            return expected_method_return
1372
        mock_dbus_python = self.MockDBusPython_func(func)
1373
        bus = dbus_python_adapter.SystemBus(mock_dbus_python)
1374
        ret = self.call_method(bus, "methodname", "busname",
1375
                               "objectpath", "interface",
1376
                               *method_args)
1377
        self.assertIs(ret, expected_method_return)
1378
1379
    def test_call_method_filters_bool_true(self):
1380
        def func():
1381
            return method_return
1382
        mock_dbus_python = self.MockDBusPython_func(func)
1383
        bus = dbus_python_adapter.SystemBus(mock_dbus_python)
1384
        method_return = mock_dbus_python.Boolean(True)
1385
        ret = self.call_method(bus, "methodname", "busname",
1386
                               "objectpath", "interface")
1387
        self.assertTrue(ret)
1388
        self.assertNotIsInstance(ret, mock_dbus_python.Boolean)
1389
1390
    def test_call_method_filters_bool_false(self):
1391
        def func():
1392
            return method_return
1393
        mock_dbus_python = self.MockDBusPython_func(func)
1394
        bus = dbus_python_adapter.SystemBus(mock_dbus_python)
1395
        method_return = mock_dbus_python.Boolean(False)
1396
        ret = self.call_method(bus, "methodname", "busname",
1397
                               "objectpath", "interface")
1398
        self.assertFalse(ret)
1399
        self.assertNotIsInstance(ret, mock_dbus_python.Boolean)
1400
1401
    def test_call_method_filters_objectpath(self):
1402
        def func():
1403
            return method_return
1404
        mock_dbus_python = self.MockDBusPython_func(func)
1405
        bus = dbus_python_adapter.SystemBus(mock_dbus_python)
1406
        method_return = mock_dbus_python.ObjectPath("objectpath")
1407
        ret = self.call_method(bus, "methodname", "busname",
1408
                               "objectpath", "interface")
1409
        self.assertEqual("objectpath", ret)
1410
        self.assertIsNot("objectpath", ret)
1411
        self.assertNotIsInstance(ret, mock_dbus_python.ObjectPath)
1412
1413
    def test_call_method_filters_booleans_in_dict(self):
1414
        def func():
1415
            return method_return
1416
        mock_dbus_python = self.MockDBusPython_func(func)
1417
        bus = dbus_python_adapter.SystemBus(mock_dbus_python)
1418
        method_return = mock_dbus_python.Dictionary(
1419
        {mock_dbus_python.Boolean(True):
1420
         mock_dbus_python.Boolean(False),
1421
         mock_dbus_python.Boolean(False):
1422
         mock_dbus_python.Boolean(True)})
1423
        ret = self.call_method(bus, "methodname", "busname",
1424
                               "objectpath", "interface")
1425
        expected_method_return = {True: False,
1426
                                  False: True}
1427
        self.assertEqual(expected_method_return, ret)
1428
        self.assertNotIsInstance(ret, mock_dbus_python.Dictionary)
1429
1430
    def test_call_method_filters_objectpaths_in_dict(self):
1431
        def func():
1432
            return method_return
1433
        mock_dbus_python = self.MockDBusPython_func(func)
1434
        bus = dbus_python_adapter.SystemBus(mock_dbus_python)
1435
        method_return = mock_dbus_python.Dictionary(
1436
        {mock_dbus_python.ObjectPath("objectpath_key_1"):
1437
         mock_dbus_python.ObjectPath("objectpath_value_1"),
1438
         mock_dbus_python.ObjectPath("objectpath_key_2"):
1439
         mock_dbus_python.ObjectPath("objectpath_value_2")})
1440
        ret = self.call_method(bus, "methodname", "busname",
1441
                               "objectpath", "interface")
1442
        expected_method_return = {str(key): str(value)
1443
                                  for key, value in
1444
                                  method_return.items()}
1445
        self.assertEqual(expected_method_return, ret)
1446
        self.assertIsInstance(ret, dict)
1447
        self.assertNotIsInstance(ret, mock_dbus_python.Dictionary)
1448
1449
    def test_call_method_filters_dict_in_dict(self):
1450
        def func():
1451
            return method_return
1452
        mock_dbus_python = self.MockDBusPython_func(func)
1453
        bus = dbus_python_adapter.SystemBus(mock_dbus_python)
1454
        method_return = mock_dbus_python.Dictionary(
1455
        {"key1": mock_dbus_python.Dictionary({"key11": "value11",
1456
                                              "key12": "value12"}),
1457
         "key2": mock_dbus_python.Dictionary({"key21": "value21",
1458
                                              "key22": "value22"})})
1459
        ret = self.call_method(bus, "methodname", "busname",
1460
                               "objectpath", "interface")
1461
        expected_method_return = {
1462
            "key1": {"key11": "value11",
1463
                     "key12": "value12"},
1464
            "key2": {"key21": "value21",
1465
                     "key22": "value22"},
1466
        }
1467
        self.assertEqual(expected_method_return, ret)
1468
        self.assertIsInstance(ret, dict)
1469
        self.assertNotIsInstance(ret, mock_dbus_python.Dictionary)
1470
        for key, value in ret.items():
1471
            self.assertIsInstance(value, dict)
1472
            self.assertEqual(expected_method_return[key], value)
1473
            self.assertNotIsInstance(value,
1474
                                     mock_dbus_python.Dictionary)
1475
1476
    def test_call_method_filters_dict_three_deep(self):
1477
        def func():
1478
            return method_return
1479
        mock_dbus_python = self.MockDBusPython_func(func)
1480
        bus = dbus_python_adapter.SystemBus(mock_dbus_python)
1481
        method_return = mock_dbus_python.Dictionary(
1482
            {"key1":
1483
             mock_dbus_python.Dictionary(
1484
                 {"key2":
1485
                  mock_dbus_python.Dictionary(
1486
                      {"key3":
1487
                       mock_dbus_python.Boolean(True),
1488
                       }),
1489
                  }),
1490
             })
1491
        ret = self.call_method(bus, "methodname", "busname",
1492
                               "objectpath", "interface")
1493
        expected_method_return = {"key1": {"key2": {"key3": True}}}
1494
        self.assertEqual(expected_method_return, ret)
1495
        self.assertIsInstance(ret, dict)
1496
        self.assertNotIsInstance(ret, mock_dbus_python.Dictionary)
1497
        self.assertIsInstance(ret["key1"], dict)
1498
        self.assertNotIsInstance(ret["key1"],
1499
                                 mock_dbus_python.Dictionary)
1500
        self.assertIsInstance(ret["key1"]["key2"], dict)
1501
        self.assertNotIsInstance(ret["key1"]["key2"],
1502
                                 mock_dbus_python.Dictionary)
1503
        self.assertTrue(ret["key1"]["key2"]["key3"])
1504
        self.assertNotIsInstance(ret["key1"]["key2"]["key3"],
1505
                                 mock_dbus_python.Boolean)
1506
1507
    def test_call_method_handles_exception(self):
1072 by Teddy Hogeborn
mandos-ctl: Refactor tests
1508
        dbus_logger = logging.getLogger("dbus.proxies")
1509
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1510
        def func():
1511
            dbus_logger.error("Test")
1512
            raise mock_dbus_python.exceptions.DBusException()
1513
1514
        mock_dbus_python = self.MockDBusPython_func(func)
1515
        bus = dbus_python_adapter.SystemBus(mock_dbus_python)
1070 by Teddy Hogeborn
mandos-ctl: Refactor
1516
1072 by Teddy Hogeborn
mandos-ctl: Refactor tests
1517
        class CountingHandler(logging.Handler):
1518
            count = 0
1519
            def emit(self, record):
1520
                self.count += 1
1521
1522
        counting_handler = CountingHandler()
1523
1524
        dbus_logger.addHandler(counting_handler)
1525
1526
        try:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1527
            with self.assertRaises(dbus.Error) as e:
1528
                self.call_method(bus, "methodname", "busname",
1529
                                 "objectpath", "interface")
1072 by Teddy Hogeborn
mandos-ctl: Refactor tests
1530
        finally:
1531
            dbus_logger.removeFilter(counting_handler)
1074 by Teddy Hogeborn
mandos-ctl: Add comment
1532
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1533
        self.assertNotIsInstance(e, dbus.ConnectFailed)
1534
1074 by Teddy Hogeborn
mandos-ctl: Add comment
1535
        # Make sure the dbus logger was suppressed
1082 by Teddy Hogeborn
mandos-ctl: Refactor tests
1536
        self.assertEqual(0, counting_handler.count)
1072 by Teddy Hogeborn
mandos-ctl: Refactor tests
1537
1098 by Teddy Hogeborn
mandos-ctl: Fix --secret when using the dbus-python module
1538
    def test_Set_Secret_sends_bytearray(self):
1539
        ret = [None]
1540
        def func(*args, **kwargs):
1541
            ret[0] = (args, kwargs)
1542
        mock_dbus_python = self.MockDBusPython_func(func)
1543
        bus = dbus_python_adapter.SystemBus(mock_dbus_python)
1544
        bus.set_client_property("objectpath", "Secret", "value")
1545
        expected_call = (("se.recompile.Mandos.Client", "Secret",
1546
                          mock_dbus_python.ByteArray(b"value")),
1547
                         {"dbus_interface":
1548
                          "org.freedesktop.DBus.Properties"})
1549
        self.assertEqual(expected_call, ret[0])
1550
        if sys.version_info.major == 2:
1551
            self.assertIsInstance(ret[0][0][-1],
1552
                                  mock_dbus_python.ByteArray)
1553
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1554
    def test_get_object_converts_to_correct_exception(self):
1555
        bus = dbus_python_adapter.SystemBus(
1556
            self.fake_dbus_python_raises_exception_on_connect)
1557
        with self.assertRaises(dbus.ConnectFailed):
1558
            self.call_method(bus, "methodname", "busname",
1559
                             "objectpath", "interface")
1560
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1561
    class fake_dbus_python_raises_exception_on_connect:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1562
        """fake dbus-python module"""
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1563
        class exceptions:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1564
            """Pseudo-namespace"""
1565
            class DBusException(Exception):
1566
                pass
1567
1568
        @classmethod
1569
        def SystemBus(cls):
1570
            def get_object(busname, objectpath):
1571
                raise cls.exceptions.DBusException()
1572
            Bus = collections.namedtuple("Bus", ["get_object"])
1573
            return Bus(get_object=get_object)
1574
1575
1576
class Test_dbus_python_adapter_CachingBus(unittest.TestCase):
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1577
    class mock_dbus_python:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1578
        """mock dbus-python modules"""
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1579
        class SystemBus:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1580
            @staticmethod
1581
            def get_object(busname, objectpath):
1582
                return Unique()
1583
1584
    def setUp(self):
1585
        self.bus = dbus_python_adapter.CachingBus(
1586
            self.mock_dbus_python)
1587
1588
    def test_returns_distinct_objectpaths(self):
1589
        obj1 = self.bus.get_object("busname", "objectpath1")
1590
        self.assertIsInstance(obj1, Unique)
1591
        obj2 = self.bus.get_object("busname", "objectpath2")
1592
        self.assertIsInstance(obj2, Unique)
1593
        self.assertIsNot(obj1, obj2)
1594
1595
    def test_returns_distinct_busnames(self):
1596
        obj1 = self.bus.get_object("busname1", "objectpath")
1597
        self.assertIsInstance(obj1, Unique)
1598
        obj2 = self.bus.get_object("busname2", "objectpath")
1599
        self.assertIsInstance(obj2, Unique)
1600
        self.assertIsNot(obj1, obj2)
1601
1602
    def test_returns_distinct_both(self):
1603
        obj1 = self.bus.get_object("busname1", "objectpath")
1604
        self.assertIsInstance(obj1, Unique)
1605
        obj2 = self.bus.get_object("busname2", "objectpath")
1606
        self.assertIsInstance(obj2, Unique)
1607
        self.assertIsNot(obj1, obj2)
1608
1609
    def test_returns_same(self):
1610
        obj1 = self.bus.get_object("busname", "objectpath")
1611
        self.assertIsInstance(obj1, Unique)
1612
        obj2 = self.bus.get_object("busname", "objectpath")
1613
        self.assertIsInstance(obj2, Unique)
1614
        self.assertIs(obj1, obj2)
1615
1616
    def test_returns_same_old(self):
1617
        obj1 = self.bus.get_object("busname1", "objectpath1")
1618
        self.assertIsInstance(obj1, Unique)
1619
        obj2 = self.bus.get_object("busname2", "objectpath2")
1620
        self.assertIsInstance(obj2, Unique)
1621
        obj1b = self.bus.get_object("busname1", "objectpath1")
1622
        self.assertIsInstance(obj1b, Unique)
1623
        self.assertIsNot(obj1, obj2)
1624
        self.assertIsNot(obj2, obj1b)
1625
        self.assertIs(obj1, obj1b)
1070 by Teddy Hogeborn
mandos-ctl: Refactor
1626
1066 by Teddy Hogeborn
mandos-ctl: Refactor
1627
1096 by Teddy Hogeborn
mandos-ctl: Add support for D-Bus module "pydbus"
1628
class Test_pydbus_adapter_SystemBus(TestCaseWithAssertLogs):
1629
1630
    def Stub_pydbus_func(self, func):
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1631
        class stub_pydbus:
1096 by Teddy Hogeborn
mandos-ctl: Add support for D-Bus module "pydbus"
1632
            """stub pydbus module"""
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1633
            class SystemBus:
1096 by Teddy Hogeborn
mandos-ctl: Add support for D-Bus module "pydbus"
1634
                @staticmethod
1635
                def get(busname, objectpath):
1636
                    DBusObject = collections.namedtuple(
1637
                        "DBusObject", ("methodname",))
1638
                    return {"interface":
1639
                            DBusObject(methodname=func)}
1640
        return stub_pydbus
1641
1642
    def call_method(self, bus, methodname, busname, objectpath,
1643
                    interface, *args):
1644
        with self.assertLogs(log, logging.DEBUG):
1645
            return bus.call_method(methodname, busname, objectpath,
1646
                                   interface, *args)
1647
1648
    def test_call_method_returns(self):
1649
        expected_method_return = Unique()
1650
        method_args = (Unique(), Unique())
1651
        def func(*args):
1652
            self.assertEqual(len(method_args), len(args))
1653
            for marg, arg in zip(method_args, args):
1654
                self.assertIs(marg, arg)
1655
            return expected_method_return
1656
        stub_pydbus = self.Stub_pydbus_func(func)
1657
        bus = pydbus_adapter.SystemBus(stub_pydbus)
1658
        ret = self.call_method(bus, "methodname", "busname",
1659
                               "objectpath", "interface",
1660
                               *method_args)
1661
        self.assertIs(ret, expected_method_return)
1662
1663
    def test_call_method_handles_exception(self):
1664
        dbus_logger = logging.getLogger("dbus.proxies")
1665
1666
        def func():
1667
            raise gi.repository.GLib.Error()
1668
1669
        stub_pydbus = self.Stub_pydbus_func(func)
1670
        bus = pydbus_adapter.SystemBus(stub_pydbus)
1671
1672
        with self.assertRaises(dbus.Error) as e:
1673
            self.call_method(bus, "methodname", "busname",
1674
                             "objectpath", "interface")
1675
1676
        self.assertNotIsInstance(e, dbus.ConnectFailed)
1677
1678
    def test_get_converts_to_correct_exception(self):
1679
        bus = pydbus_adapter.SystemBus(
1680
            self.fake_pydbus_raises_exception_on_connect)
1681
        with self.assertRaises(dbus.ConnectFailed):
1682
            self.call_method(bus, "methodname", "busname",
1683
                             "objectpath", "interface")
1684
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1685
    class fake_pydbus_raises_exception_on_connect:
1096 by Teddy Hogeborn
mandos-ctl: Add support for D-Bus module "pydbus"
1686
        """fake dbus-python module"""
1687
        @classmethod
1688
        def SystemBus(cls):
1689
            def get(busname, objectpath):
1690
                raise gi.repository.GLib.Error()
1691
            Bus = collections.namedtuple("Bus", ["get"])
1692
            return Bus(get=get)
1693
1694
    def test_set_property_uses_setattr(self):
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1695
        class Object:
1096 by Teddy Hogeborn
mandos-ctl: Add support for D-Bus module "pydbus"
1696
            pass
1697
        obj = Object()
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1698
        class pydbus_spy:
1699
            class SystemBus:
1096 by Teddy Hogeborn
mandos-ctl: Add support for D-Bus module "pydbus"
1700
                @staticmethod
1701
                def get(busname, objectpath):
1702
                    return {"interface": obj}
1703
        bus = pydbus_adapter.SystemBus(pydbus_spy)
1704
        value = Unique()
1705
        bus.set_property("busname", "objectpath", "interface", "key",
1706
                         value)
1707
        self.assertIs(value, obj.key)
1708
1709
    def test_get_suppresses_xml_deprecation_warning(self):
1710
        if sys.version_info.major >= 3:
1711
            return
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1712
        class stub_pydbus_get:
1713
            class SystemBus:
1096 by Teddy Hogeborn
mandos-ctl: Add support for D-Bus module "pydbus"
1714
                @staticmethod
1715
                def get(busname, objectpath):
1716
                    warnings.warn_explicit(
1717
                        "deprecated", DeprecationWarning,
1718
                        "xml.etree.ElementTree", 0)
1719
        bus = pydbus_adapter.SystemBus(stub_pydbus_get)
1720
        with warnings.catch_warnings(record=True) as w:
1721
            warnings.simplefilter("always")
1722
            bus.get("busname", "objectpath")
1723
            self.assertEqual(0, len(w))
1724
1725
1726
class Test_pydbus_adapter_CachingBus(unittest.TestCase):
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1727
    class stub_pydbus:
1096 by Teddy Hogeborn
mandos-ctl: Add support for D-Bus module "pydbus"
1728
        """stub pydbus module"""
1154 by Teddy Hogeborn
Simplification of Python 3 compatibility code
1729
        class SystemBus:
1096 by Teddy Hogeborn
mandos-ctl: Add support for D-Bus module "pydbus"
1730
            @staticmethod
1731
            def get(busname, objectpath):
1732
                return Unique()
1733
1734
    def setUp(self):
1735
        self.bus = pydbus_adapter.CachingBus(self.stub_pydbus)
1736
1737
    def test_returns_distinct_objectpaths(self):
1738
        obj1 = self.bus.get("busname", "objectpath1")
1739
        self.assertIsInstance(obj1, Unique)
1740
        obj2 = self.bus.get("busname", "objectpath2")
1741
        self.assertIsInstance(obj2, Unique)
1742
        self.assertIsNot(obj1, obj2)
1743
1744
    def test_returns_distinct_busnames(self):
1745
        obj1 = self.bus.get("busname1", "objectpath")
1746
        self.assertIsInstance(obj1, Unique)
1747
        obj2 = self.bus.get("busname2", "objectpath")
1748
        self.assertIsInstance(obj2, Unique)
1749
        self.assertIsNot(obj1, obj2)
1750
1751
    def test_returns_distinct_both(self):
1752
        obj1 = self.bus.get("busname1", "objectpath")
1753
        self.assertIsInstance(obj1, Unique)
1754
        obj2 = self.bus.get("busname2", "objectpath")
1755
        self.assertIsInstance(obj2, Unique)
1756
        self.assertIsNot(obj1, obj2)
1757
1758
    def test_returns_same(self):
1759
        obj1 = self.bus.get("busname", "objectpath")
1760
        self.assertIsInstance(obj1, Unique)
1761
        obj2 = self.bus.get("busname", "objectpath")
1762
        self.assertIsInstance(obj2, Unique)
1763
        self.assertIs(obj1, obj2)
1764
1765
    def test_returns_same_old(self):
1766
        obj1 = self.bus.get("busname1", "objectpath1")
1767
        self.assertIsInstance(obj1, Unique)
1768
        obj2 = self.bus.get("busname2", "objectpath2")
1769
        self.assertIsInstance(obj2, Unique)
1770
        obj1b = self.bus.get("busname1", "objectpath1")
1771
        self.assertIsInstance(obj1b, Unique)
1772
        self.assertIsNot(obj1, obj2)
1773
        self.assertIsNot(obj2, obj1b)
1774
        self.assertIs(obj1, obj1b)
1775
1776
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1777
class Test_commands_from_options(unittest.TestCase):
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
1778
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1779
    def setUp(self):
1780
        self.parser = argparse.ArgumentParser()
1781
        add_command_line_options(self.parser)
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1782
1783
    def test_is_enabled(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1784
        self.assert_command_from_args(["--is-enabled", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1785
                                      command.IsEnabled)
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1786
1104 by Teddy Hogeborn
mandos-ctl: Add tests for all examples in the manual page
1787
    def assert_command_from_args(self, args, command_cls, length=1,
1788
                                 clients=None, **cmd_attrs):
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1789
        """Assert that parsing ARGS should result in an instance of
1790
COMMAND_CLS with (optionally) all supplied attributes (CMD_ATTRS)."""
1791
        options = self.parser.parse_args(args)
1792
        check_option_syntax(self.parser, options)
1793
        commands = commands_from_options(options)
1104 by Teddy Hogeborn
mandos-ctl: Add tests for all examples in the manual page
1794
        self.assertEqual(length, len(commands))
1795
        for command in commands:
1796
            if isinstance(command, command_cls):
1797
                break
1798
        else:
1799
            self.assertIsInstance(command, command_cls)
1800
        if clients is not None:
1801
            self.assertEqual(clients, options.client)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1802
        for key, value in cmd_attrs.items():
1082 by Teddy Hogeborn
mandos-ctl: Refactor tests
1803
            self.assertEqual(value, getattr(command, key))
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1804
1104 by Teddy Hogeborn
mandos-ctl: Add tests for all examples in the manual page
1805
    def assert_commands_from_args(self, args, commands, clients=None):
1806
        for cmd in commands:
1807
            self.assert_command_from_args(args, cmd,
1808
                                          length=len(commands),
1809
                                          clients=clients)
1810
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1811
    def test_is_enabled_short(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1812
        self.assert_command_from_args(["-V", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1813
                                      command.IsEnabled)
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1814
1815
    def test_approve(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1816
        self.assert_command_from_args(["--approve", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1817
                                      command.Approve)
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1818
1819
    def test_approve_short(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1820
        self.assert_command_from_args(["-A", "client"],
1821
                                      command.Approve)
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1822
1823
    def test_deny(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1824
        self.assert_command_from_args(["--deny", "client"],
1825
                                      command.Deny)
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1826
1827
    def test_deny_short(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1828
        self.assert_command_from_args(["-D", "client"], command.Deny)
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1829
1830
    def test_remove(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1831
        self.assert_command_from_args(["--remove", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1832
                                      command.Remove)
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1833
1834
    def test_deny_before_remove(self):
1835
        options = self.parser.parse_args(["--deny", "--remove",
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1836
                                          "client"])
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1837
        check_option_syntax(self.parser, options)
1838
        commands = commands_from_options(options)
1082 by Teddy Hogeborn
mandos-ctl: Refactor tests
1839
        self.assertEqual(2, len(commands))
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1840
        self.assertIsInstance(commands[0], command.Deny)
1841
        self.assertIsInstance(commands[1], command.Remove)
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1842
1843
    def test_deny_before_remove_reversed(self):
1844
        options = self.parser.parse_args(["--remove", "--deny",
1845
                                          "--all"])
1846
        check_option_syntax(self.parser, options)
1847
        commands = commands_from_options(options)
1082 by Teddy Hogeborn
mandos-ctl: Refactor tests
1848
        self.assertEqual(2, len(commands))
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1849
        self.assertIsInstance(commands[0], command.Deny)
1850
        self.assertIsInstance(commands[1], command.Remove)
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1851
1852
    def test_remove_short(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1853
        self.assert_command_from_args(["-r", "client"],
1854
                                      command.Remove)
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1855
1856
    def test_dump_json(self):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1857
        self.assert_command_from_args(["--dump-json"],
1858
                                      command.DumpJSON)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1859
1860
    def test_enable(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1861
        self.assert_command_from_args(["--enable", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1862
                                      command.Enable)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1863
1864
    def test_enable_short(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1865
        self.assert_command_from_args(["-e", "client"],
1866
                                      command.Enable)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1867
1868
    def test_disable(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1869
        self.assert_command_from_args(["--disable", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1870
                                      command.Disable)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1871
1872
    def test_disable_short(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1873
        self.assert_command_from_args(["-d", "client"],
1874
                                      command.Disable)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1875
1876
    def test_bump_timeout(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1877
        self.assert_command_from_args(["--bump-timeout", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1878
                                      command.BumpTimeout)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1879
1880
    def test_bump_timeout_short(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1881
        self.assert_command_from_args(["-b", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1882
                                      command.BumpTimeout)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1883
1884
    def test_start_checker(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1885
        self.assert_command_from_args(["--start-checker", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1886
                                      command.StartChecker)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1887
1888
    def test_stop_checker(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1889
        self.assert_command_from_args(["--stop-checker", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1890
                                      command.StopChecker)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1891
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1892
    def test_approve_by_default(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1893
        self.assert_command_from_args(["--approve-by-default",
1894
                                       "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1895
                                      command.ApproveByDefault)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1896
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1897
    def test_deny_by_default(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1898
        self.assert_command_from_args(["--deny-by-default", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1899
                                      command.DenyByDefault)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1900
1901
    def test_checker(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1902
        self.assert_command_from_args(["--checker", ":", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1903
                                      command.SetChecker,
1904
                                      value_to_set=":")
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1905
1906
    def test_checker_empty(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1907
        self.assert_command_from_args(["--checker", "", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1908
                                      command.SetChecker,
1909
                                      value_to_set="")
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1910
1911
    def test_checker_short(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1912
        self.assert_command_from_args(["-c", ":", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1913
                                      command.SetChecker,
1914
                                      value_to_set=":")
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1915
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1916
    def test_host(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1917
        self.assert_command_from_args(
1918
            ["--host", "client.example.org", "client"],
1919
            command.SetHost, value_to_set="client.example.org")
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1920
1921
    def test_host_short(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1922
        self.assert_command_from_args(
1923
            ["-H", "client.example.org", "client"], command.SetHost,
1924
            value_to_set="client.example.org")
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1925
1926
    def test_secret_devnull(self):
1927
        self.assert_command_from_args(["--secret", os.path.devnull,
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1928
                                       "client"], command.SetSecret,
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1929
                                      value_to_set=b"")
1930
1931
    def test_secret_tempfile(self):
1932
        with tempfile.NamedTemporaryFile(mode="r+b") as f:
1933
            value = b"secret\0xyzzy\nbar"
1934
            f.write(value)
1935
            f.seek(0)
1936
            self.assert_command_from_args(["--secret", f.name,
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1937
                                           "client"],
1938
                                          command.SetSecret,
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1939
                                          value_to_set=value)
1940
1941
    def test_secret_devnull_short(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1942
        self.assert_command_from_args(["-s", os.path.devnull,
1943
                                       "client"], command.SetSecret,
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1944
                                      value_to_set=b"")
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1945
1946
    def test_secret_tempfile_short(self):
1947
        with tempfile.NamedTemporaryFile(mode="r+b") as f:
1948
            value = b"secret\0xyzzy\nbar"
1949
            f.write(value)
1950
            f.seek(0)
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1951
            self.assert_command_from_args(["-s", f.name, "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1952
                                          command.SetSecret,
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1953
                                          value_to_set=value)
1954
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1955
    def test_timeout(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1956
        self.assert_command_from_args(["--timeout", "PT5M", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1957
                                      command.SetTimeout,
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1958
                                      value_to_set=300000)
1959
1960
    def test_timeout_short(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1961
        self.assert_command_from_args(["-t", "PT5M", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1962
                                      command.SetTimeout,
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1963
                                      value_to_set=300000)
1964
1965
    def test_extended_timeout(self):
1966
        self.assert_command_from_args(["--extended-timeout", "PT15M",
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1967
                                       "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1968
                                      command.SetExtendedTimeout,
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1969
                                      value_to_set=900000)
1970
1971
    def test_interval(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1972
        self.assert_command_from_args(["--interval", "PT2M",
1973
                                       "client"], command.SetInterval,
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1974
                                      value_to_set=120000)
1975
1976
    def test_interval_short(self):
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1977
        self.assert_command_from_args(["-i", "PT2M", "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1978
                                      command.SetInterval,
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1979
                                      value_to_set=120000)
1980
1981
    def test_approval_delay(self):
1982
        self.assert_command_from_args(["--approval-delay", "PT30S",
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1983
                                       "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1984
                                      command.SetApprovalDelay,
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1985
                                      value_to_set=30000)
1986
1987
    def test_approval_duration(self):
1988
        self.assert_command_from_args(["--approval-duration", "PT1S",
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
1989
                                       "client"],
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1990
                                      command.SetApprovalDuration,
1055 by Teddy Hogeborn
mandos-ctl: Refactor
1991
                                      value_to_set=1000)
1992
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1993
    def test_print_table(self):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1994
        self.assert_command_from_args([], command.PrintTable,
1060 by Teddy Hogeborn
mandos-ctl: Refactor
1995
                                      verbose=False)
1996
1997
    def test_print_table_verbose(self):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
1998
        self.assert_command_from_args(["--verbose"],
1999
                                      command.PrintTable,
1060 by Teddy Hogeborn
mandos-ctl: Refactor
2000
                                      verbose=True)
2001
2002
    def test_print_table_verbose_short(self):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2003
        self.assert_command_from_args(["-v"], command.PrintTable,
1060 by Teddy Hogeborn
mandos-ctl: Refactor
2004
                                      verbose=True)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2005
2006
1104 by Teddy Hogeborn
mandos-ctl: Add tests for all examples in the manual page
2007
    def test_manual_page_example_1(self):
1131 by Teddy Hogeborn
Make tests and man page examples match
2008
        self.assert_command_from_args("",
1104 by Teddy Hogeborn
mandos-ctl: Add tests for all examples in the manual page
2009
                                      command.PrintTable,
2010
                                      clients=[],
1131 by Teddy Hogeborn
Make tests and man page examples match
2011
                                      verbose=False)
1104 by Teddy Hogeborn
mandos-ctl: Add tests for all examples in the manual page
2012
2013
    def test_manual_page_example_2(self):
2014
        self.assert_command_from_args(
2015
            "--verbose foo1.example.org foo2.example.org".split(),
2016
            command.PrintTable, clients=["foo1.example.org",
2017
                                         "foo2.example.org"],
2018
            verbose=True)
2019
2020
    def test_manual_page_example_3(self):
2021
        self.assert_command_from_args("--enable --all".split(),
2022
                                      command.Enable,
2023
                                      clients=[])
2024
2025
    def test_manual_page_example_4(self):
2026
        self.assert_commands_from_args(
2027
            ("--timeout=PT5M --interval=PT1M foo1.example.org"
2028
             " foo2.example.org").split(),
2029
            [command.SetTimeout, command.SetInterval],
2030
            clients=["foo1.example.org", "foo2.example.org"])
2031
2032
    def test_manual_page_example_5(self):
2033
        self.assert_command_from_args("--approve --all".split(),
2034
                                      command.Approve,
2035
                                      clients=[])
2036
2037
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2038
class TestCommand(unittest.TestCase):
1010 by Teddy Hogeborn
mandos-ctl: Refactor; test PrintTableCmd instead of TableOfClients
2039
    """Abstract class for tests of command classes"""
1061 by Teddy Hogeborn
mandos-ctl: White space changes only
2040
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
2041
    class FakeMandosBus(dbus.MandosBus):
2042
        def __init__(self, testcase):
2043
            self.client_properties = {
2044
                "Name": "foo",
2045
                "KeyID": ("92ed150794387c03ce684574b1139a65"
2046
                          "94a34f895daaaf09fd8ea90a27cddb12"),
2047
                "Secret": b"secret",
2048
                "Host": "foo.example.org",
2049
                "Enabled": True,
2050
                "Timeout": 300000,
2051
                "LastCheckedOK": "2019-02-03T00:00:00",
2052
                "Created": "2019-01-02T00:00:00",
2053
                "Interval": 120000,
2054
                "Fingerprint": ("778827225BA7DE539C5A"
2055
                                "7CFA59CFF7CDBD9A5920"),
2056
                "CheckerRunning": False,
2057
                "LastEnabled": "2019-01-03T00:00:00",
2058
                "ApprovalPending": False,
2059
                "ApprovedByDefault": True,
2060
                "LastApprovalRequest": "",
2061
                "ApprovalDelay": 0,
2062
                "ApprovalDuration": 1000,
2063
                "Checker": "fping -q -- %(host)s",
2064
                "ExtendedTimeout": 900000,
2065
                "Expires": "2019-02-04T00:00:00",
2066
                "LastCheckerStatus": 0,
2067
            }
2068
            self.other_client_properties = {
2069
                "Name": "barbar",
2070
                "KeyID": ("0558568eedd67d622f5c83b35a115f79"
2071
                          "6ab612cff5ad227247e46c2b020f441c"),
2072
                "Secret": b"secretbar",
2073
                "Host": "192.0.2.3",
2074
                "Enabled": True,
2075
                "Timeout": 300000,
2076
                "LastCheckedOK": "2019-02-04T00:00:00",
2077
                "Created": "2019-01-03T00:00:00",
2078
                "Interval": 120000,
2079
                "Fingerprint": ("3E393AEAEFB84C7E89E2"
2080
                                "F547B3A107558FCA3A27"),
2081
                "CheckerRunning": True,
2082
                "LastEnabled": "2019-01-04T00:00:00",
2083
                "ApprovalPending": False,
2084
                "ApprovedByDefault": False,
2085
                "LastApprovalRequest": "2019-01-03T00:00:00",
2086
                "ApprovalDelay": 30000,
2087
                "ApprovalDuration": 93785000,
2088
                "Checker": ":",
2089
                "ExtendedTimeout": 900000,
2090
                "Expires": "2019-02-05T00:00:00",
2091
                "LastCheckerStatus": -2,
2092
            }
2093
            self.clients =  collections.OrderedDict(
2094
                [
2095
                    ("client_objectpath", self.client_properties),
2096
                    ("other_client_objectpath",
2097
                     self.other_client_properties),
2098
                ])
2099
            self.one_client = {"client_objectpath":
2100
                               self.client_properties}
2101
            self.testcase = testcase
2102
            self.calls = []
2103
2104
        def call_method(self, methodname, busname, objectpath,
2105
                        interface, *args):
2106
            self.testcase.assertEqual("se.recompile.Mandos", busname)
2107
            self.calls.append((methodname, busname, objectpath,
2108
                               interface, args))
2109
            if interface == "org.freedesktop.DBus.Properties":
2110
                if methodname == "Set":
2111
                    self.testcase.assertEqual(3, len(args))
2112
                    interface, key, value = args
2113
                    self.testcase.assertEqual(
2114
                        "se.recompile.Mandos.Client", interface)
2115
                    self.clients[objectpath][key] = value
2116
                    return
2117
            elif interface == "se.recompile.Mandos":
2118
                self.testcase.assertEqual("RemoveClient", methodname)
2119
                self.testcase.assertEqual(1, len(args))
2120
                clientpath = args[0]
2121
                del self.clients[clientpath]
2122
                return
2123
            elif interface == "se.recompile.Mandos.Client":
2124
                if methodname == "Approve":
2125
                    self.testcase.assertEqual(1, len(args))
2126
                    return
2127
            raise ValueError()
2128
994 by Teddy Hogeborn
mandos-ctl: Add tests for table_rows_of_clients()
2129
    def setUp(self):
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
2130
        self.bus = self.FakeMandosBus(self)
1010 by Teddy Hogeborn
mandos-ctl: Refactor; test PrintTableCmd instead of TableOfClients
2131
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2132
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2133
class TestBaseCommands(TestCommand):
2134
1081 by Teddy Hogeborn
mandos-ctl: Refactor tests
2135
    def test_IsEnabled_exits_successfully(self):
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2136
        with self.assertRaises(SystemExit) as e:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
2137
            command.IsEnabled().run(self.bus.one_client)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2138
        if e.exception.code is not None:
1082 by Teddy Hogeborn
mandos-ctl: Refactor tests
2139
            self.assertEqual(0, e.exception.code)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2140
        else:
2141
            self.assertIsNone(e.exception.code)
1061 by Teddy Hogeborn
mandos-ctl: White space changes only
2142
1081 by Teddy Hogeborn
mandos-ctl: Refactor tests
2143
    def test_IsEnabled_exits_with_failure(self):
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
2144
        self.bus.client_properties["Enabled"] = False
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2145
        with self.assertRaises(SystemExit) as e:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
2146
            command.IsEnabled().run(self.bus.one_client)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2147
        if isinstance(e.exception.code, int):
1082 by Teddy Hogeborn
mandos-ctl: Refactor tests
2148
            self.assertNotEqual(0, e.exception.code)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2149
        else:
2150
            self.assertIsNotNone(e.exception.code)
2151
1079 by Teddy Hogeborn
mandos-ctl: Refactor
2152
    def test_Approve(self):
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
2153
        busname = "se.recompile.Mandos"
2154
        client_interface = "se.recompile.Mandos.Client"
2155
        command.Approve().run(self.bus.clients, self.bus)
2156
        for clientpath in self.bus.clients:
2157
            self.assertIn(("Approve", busname, clientpath,
2158
                           client_interface, (True,)), self.bus.calls)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2159
1079 by Teddy Hogeborn
mandos-ctl: Refactor
2160
    def test_Deny(self):
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
2161
        busname = "se.recompile.Mandos"
2162
        client_interface = "se.recompile.Mandos.Client"
2163
        command.Deny().run(self.bus.clients, self.bus)
2164
        for clientpath in self.bus.clients:
2165
            self.assertIn(("Approve", busname, clientpath,
2166
                           client_interface, (False,)),
2167
                          self.bus.calls)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2168
1079 by Teddy Hogeborn
mandos-ctl: Refactor
2169
    def test_Remove(self):
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
2170
        command.Remove().run(self.bus.clients, self.bus)
2171
        for clientpath in self.bus.clients:
2172
            self.assertIn(("RemoveClient", dbus_busname,
2173
                           dbus_server_path, dbus_server_interface,
2174
                           (clientpath,)), self.bus.calls)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2175
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2176
    expected_json = {
2177
        "foo": {
2178
            "Name": "foo",
2179
            "KeyID": ("92ed150794387c03ce684574b1139a65"
2180
                      "94a34f895daaaf09fd8ea90a27cddb12"),
2181
            "Host": "foo.example.org",
2182
            "Enabled": True,
2183
            "Timeout": 300000,
2184
            "LastCheckedOK": "2019-02-03T00:00:00",
2185
            "Created": "2019-01-02T00:00:00",
2186
            "Interval": 120000,
2187
            "Fingerprint": ("778827225BA7DE539C5A"
2188
                            "7CFA59CFF7CDBD9A5920"),
2189
            "CheckerRunning": False,
2190
            "LastEnabled": "2019-01-03T00:00:00",
2191
            "ApprovalPending": False,
2192
            "ApprovedByDefault": True,
2193
            "LastApprovalRequest": "",
2194
            "ApprovalDelay": 0,
2195
            "ApprovalDuration": 1000,
2196
            "Checker": "fping -q -- %(host)s",
2197
            "ExtendedTimeout": 900000,
2198
            "Expires": "2019-02-04T00:00:00",
2199
            "LastCheckerStatus": 0,
2200
        },
2201
        "barbar": {
2202
            "Name": "barbar",
2203
            "KeyID": ("0558568eedd67d622f5c83b35a115f79"
2204
                      "6ab612cff5ad227247e46c2b020f441c"),
2205
            "Host": "192.0.2.3",
2206
            "Enabled": True,
2207
            "Timeout": 300000,
2208
            "LastCheckedOK": "2019-02-04T00:00:00",
2209
            "Created": "2019-01-03T00:00:00",
2210
            "Interval": 120000,
2211
            "Fingerprint": ("3E393AEAEFB84C7E89E2"
2212
                            "F547B3A107558FCA3A27"),
2213
            "CheckerRunning": True,
2214
            "LastEnabled": "2019-01-04T00:00:00",
2215
            "ApprovalPending": False,
2216
            "ApprovedByDefault": False,
2217
            "LastApprovalRequest": "2019-01-03T00:00:00",
2218
            "ApprovalDelay": 30000,
2219
            "ApprovalDuration": 93785000,
2220
            "Checker": ":",
2221
            "ExtendedTimeout": 900000,
2222
            "Expires": "2019-02-05T00:00:00",
2223
            "LastCheckerStatus": -2,
2224
        },
2225
    }
2226
2227
    def test_DumpJSON_normal(self):
1084 by Teddy Hogeborn
mandos-ctl: Refactor tests
2228
        with self.capture_stdout_to_buffer() as buffer:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
2229
            command.DumpJSON().run(self.bus.clients)
1084 by Teddy Hogeborn
mandos-ctl: Refactor tests
2230
        json_data = json.loads(buffer.getvalue())
1082 by Teddy Hogeborn
mandos-ctl: Refactor tests
2231
        self.assertDictEqual(self.expected_json, json_data)
1061 by Teddy Hogeborn
mandos-ctl: White space changes only
2232
1084 by Teddy Hogeborn
mandos-ctl: Refactor tests
2233
    @staticmethod
2234
    @contextlib.contextmanager
2235
    def capture_stdout_to_buffer():
2236
        capture_buffer = io.StringIO()
2237
        old_stdout = sys.stdout
2238
        sys.stdout = capture_buffer
2239
        try:
2240
            yield capture_buffer
2241
        finally:
2242
            sys.stdout = old_stdout
2243
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2244
    def test_DumpJSON_one_client(self):
1084 by Teddy Hogeborn
mandos-ctl: Refactor tests
2245
        with self.capture_stdout_to_buffer() as buffer:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
2246
            command.DumpJSON().run(self.bus.one_client)
1084 by Teddy Hogeborn
mandos-ctl: Refactor tests
2247
        json_data = json.loads(buffer.getvalue())
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2248
        expected_json = {"foo": self.expected_json["foo"]}
1082 by Teddy Hogeborn
mandos-ctl: Refactor tests
2249
        self.assertDictEqual(expected_json, json_data)
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2250
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2251
    def test_PrintTable_normal(self):
1084 by Teddy Hogeborn
mandos-ctl: Refactor tests
2252
        with self.capture_stdout_to_buffer() as buffer:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
2253
            command.PrintTable().run(self.bus.clients)
1053 by Teddy Hogeborn
mandos-ctl: White space and other non-semantic changes only
2254
        expected_output = "\n".join((
2255
            "Name   Enabled Timeout  Last Successful Check",
2256
            "foo    Yes     00:05:00 2019-02-03T00:00:00  ",
2257
            "barbar Yes     00:05:00 2019-02-04T00:00:00  ",
1084 by Teddy Hogeborn
mandos-ctl: Refactor tests
2258
        )) + "\n"
2259
        self.assertEqual(expected_output, buffer.getvalue())
1061 by Teddy Hogeborn
mandos-ctl: White space changes only
2260
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2261
    def test_PrintTable_verbose(self):
1084 by Teddy Hogeborn
mandos-ctl: Refactor tests
2262
        with self.capture_stdout_to_buffer() as buffer:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
2263
            command.PrintTable(verbose=True).run(self.bus.clients)
1054 by Teddy Hogeborn
mandos-ctl: Refactor
2264
        columns = (
2265
            (
2266
                "Name   ",
2267
                "foo    ",
2268
                "barbar ",
2269
            ),(
2270
                "Enabled ",
2271
                "Yes     ",
2272
                "Yes     ",
2273
            ),(
2274
                "Timeout  ",
2275
                "00:05:00 ",
2276
                "00:05:00 ",
2277
            ),(
2278
                "Last Successful Check ",
2279
                "2019-02-03T00:00:00   ",
2280
                "2019-02-04T00:00:00   ",
2281
            ),(
2282
                "Created             ",
2283
                "2019-01-02T00:00:00 ",
2284
                "2019-01-03T00:00:00 ",
2285
            ),(
2286
                "Interval ",
2287
                "00:02:00 ",
2288
                "00:02:00 ",
2289
            ),(
2290
                "Host            ",
2291
                "foo.example.org ",
2292
                "192.0.2.3       ",
2293
            ),(
2294
                ("Key ID                                             "
2295
                 "              "),
2296
                ("92ed150794387c03ce684574b1139a6594a34f895daaaf09fd8"
2297
                 "ea90a27cddb12 "),
2298
                ("0558568eedd67d622f5c83b35a115f796ab612cff5ad227247e"
2299
                 "46c2b020f441c "),
2300
            ),(
2301
                "Fingerprint                              ",
2302
                "778827225BA7DE539C5A7CFA59CFF7CDBD9A5920 ",
2303
                "3E393AEAEFB84C7E89E2F547B3A107558FCA3A27 ",
2304
            ),(
2305
                "Check Is Running ",
2306
                "No               ",
2307
                "Yes              ",
2308
            ),(
2309
                "Last Enabled        ",
2310
                "2019-01-03T00:00:00 ",
2311
                "2019-01-04T00:00:00 ",
2312
            ),(
2313
                "Approval Is Pending ",
2314
                "No                  ",
2315
                "No                  ",
2316
            ),(
2317
                "Approved By Default ",
2318
                "Yes                 ",
2319
                "No                  ",
2320
            ),(
2321
                "Last Approval Request ",
2322
                "                      ",
2323
                "2019-01-03T00:00:00   ",
2324
            ),(
2325
                "Approval Delay ",
2326
                "00:00:00       ",
2327
                "00:00:30       ",
2328
            ),(
2329
                "Approval Duration ",
2330
                "00:00:01          ",
1056 by Teddy Hogeborn
mandos-ctl: Refactor
2331
                "1T02:03:05        ",
1054 by Teddy Hogeborn
mandos-ctl: Refactor
2332
            ),(
2333
                "Checker              ",
2334
                "fping -q -- %(host)s ",
2335
                ":                    ",
2336
            ),(
2337
                "Extended Timeout ",
2338
                "00:15:00         ",
2339
                "00:15:00         ",
2340
            ),(
2341
                "Expires             ",
2342
                "2019-02-04T00:00:00 ",
2343
                "2019-02-05T00:00:00 ",
2344
            ),(
2345
                "Last Checker Status",
2346
                "0                  ",
2347
                "-2                 ",
2348
            )
2349
        )
2350
        num_lines = max(len(rows) for rows in columns)
1084 by Teddy Hogeborn
mandos-ctl: Refactor tests
2351
        expected_output = ("\n".join("".join(rows[line]
2352
                                             for rows in columns)
2353
                                     for line in range(num_lines))
2354
                           + "\n")
2355
        self.assertEqual(expected_output, buffer.getvalue())
1061 by Teddy Hogeborn
mandos-ctl: White space changes only
2356
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2357
    def test_PrintTable_one_client(self):
1084 by Teddy Hogeborn
mandos-ctl: Refactor tests
2358
        with self.capture_stdout_to_buffer() as buffer:
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
2359
            command.PrintTable().run(self.bus.one_client)
1057 by Teddy Hogeborn
mandos-ctl: Refactor
2360
        expected_output = "\n".join((
2361
            "Name Enabled Timeout  Last Successful Check",
2362
            "foo  Yes     00:05:00 2019-02-03T00:00:00  ",
1084 by Teddy Hogeborn
mandos-ctl: Refactor tests
2363
        )) + "\n"
2364
        self.assertEqual(expected_output, buffer.getvalue())
994 by Teddy Hogeborn
mandos-ctl: Add tests for table_rows_of_clients()
2365
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2366
1087 by Teddy Hogeborn
mandos-ctl: Refactor
2367
class TestPropertySetterCmd(TestCommand):
2368
    """Abstract class for tests of command.PropertySetter classes"""
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
2369
1024 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
2370
    def runTest(self):
2371
        if not hasattr(self, "command"):
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
2372
            return              # Abstract TestCase class
1093 by Teddy Hogeborn
mandos-ctl: Refactor: fix confusion about values_to_set/values_to_get
2373
2374
        if hasattr(self, "values_to_set"):
2375
            cmd_args = [(value,) for value in self.values_to_set]
2376
            values_to_get = getattr(self, "values_to_get",
2377
                                    self.values_to_set)
2378
        else:
2379
            cmd_args = [() for x in range(len(self.values_to_get))]
2380
            values_to_get = self.values_to_get
2381
        for value_to_get, cmd_arg in zip(values_to_get, cmd_args):
2382
            for clientpath in self.bus.clients:
2383
                self.bus.clients[clientpath][self.propname] = (
2384
                    Unique())
2385
            self.command(*cmd_arg).run(self.bus.clients, self.bus)
2386
            for clientpath in self.bus.clients:
2387
                value = (self.bus.clients[clientpath]
2388
                         [self.propname])
1090 by Teddy Hogeborn
mandos-ctl: Refactor D-Bus operations
2389
                self.assertNotIsInstance(value, Unique)
1082 by Teddy Hogeborn
mandos-ctl: Refactor tests
2390
                self.assertEqual(value_to_get, value)
1062 by Teddy Hogeborn
mandos-ctl: Refactor
2391
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2392
1087 by Teddy Hogeborn
mandos-ctl: Refactor
2393
class TestEnableCmd(TestPropertySetterCmd):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2394
    command = command.Enable
1059 by Teddy Hogeborn
mandos-ctl: Refactor
2395
    propname = "Enabled"
1093 by Teddy Hogeborn
mandos-ctl: Refactor: fix confusion about values_to_set/values_to_get
2396
    values_to_get = [True]
1059 by Teddy Hogeborn
mandos-ctl: Refactor
2397
2398
1087 by Teddy Hogeborn
mandos-ctl: Refactor
2399
class TestDisableCmd(TestPropertySetterCmd):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2400
    command = command.Disable
1059 by Teddy Hogeborn
mandos-ctl: Refactor
2401
    propname = "Enabled"
1093 by Teddy Hogeborn
mandos-ctl: Refactor: fix confusion about values_to_set/values_to_get
2402
    values_to_get = [False]
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2403
2404
1087 by Teddy Hogeborn
mandos-ctl: Refactor
2405
class TestBumpTimeoutCmd(TestPropertySetterCmd):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2406
    command = command.BumpTimeout
1047 by Teddy Hogeborn
mandos-ctl: Refactor
2407
    propname = "LastCheckedOK"
1093 by Teddy Hogeborn
mandos-ctl: Refactor: fix confusion about values_to_set/values_to_get
2408
    values_to_get = [""]
1024 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
2409
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2410
1087 by Teddy Hogeborn
mandos-ctl: Refactor
2411
class TestStartCheckerCmd(TestPropertySetterCmd):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2412
    command = command.StartChecker
1047 by Teddy Hogeborn
mandos-ctl: Refactor
2413
    propname = "CheckerRunning"
1093 by Teddy Hogeborn
mandos-ctl: Refactor: fix confusion about values_to_set/values_to_get
2414
    values_to_get = [True]
1024 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
2415
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2416
1087 by Teddy Hogeborn
mandos-ctl: Refactor
2417
class TestStopCheckerCmd(TestPropertySetterCmd):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2418
    command = command.StopChecker
1047 by Teddy Hogeborn
mandos-ctl: Refactor
2419
    propname = "CheckerRunning"
1093 by Teddy Hogeborn
mandos-ctl: Refactor: fix confusion about values_to_set/values_to_get
2420
    values_to_get = [False]
1024 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
2421
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2422
1087 by Teddy Hogeborn
mandos-ctl: Refactor
2423
class TestApproveByDefaultCmd(TestPropertySetterCmd):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2424
    command = command.ApproveByDefault
1047 by Teddy Hogeborn
mandos-ctl: Refactor
2425
    propname = "ApprovedByDefault"
1093 by Teddy Hogeborn
mandos-ctl: Refactor: fix confusion about values_to_set/values_to_get
2426
    values_to_get = [True]
1024 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
2427
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2428
1087 by Teddy Hogeborn
mandos-ctl: Refactor
2429
class TestDenyByDefaultCmd(TestPropertySetterCmd):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2430
    command = command.DenyByDefault
1047 by Teddy Hogeborn
mandos-ctl: Refactor
2431
    propname = "ApprovedByDefault"
1093 by Teddy Hogeborn
mandos-ctl: Refactor: fix confusion about values_to_set/values_to_get
2432
    values_to_get = [False]
2433
2434
2435
class TestSetCheckerCmd(TestPropertySetterCmd):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2436
    command = command.SetChecker
1047 by Teddy Hogeborn
mandos-ctl: Refactor
2437
    propname = "Checker"
1024 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
2438
    values_to_set = ["", ":", "fping -q -- %s"]
2439
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2440
1093 by Teddy Hogeborn
mandos-ctl: Refactor: fix confusion about values_to_set/values_to_get
2441
class TestSetHostCmd(TestPropertySetterCmd):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2442
    command = command.SetHost
1047 by Teddy Hogeborn
mandos-ctl: Refactor
2443
    propname = "Host"
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
2444
    values_to_set = ["192.0.2.3", "client.example.org"]
1024 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
2445
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2446
1093 by Teddy Hogeborn
mandos-ctl: Refactor: fix confusion about values_to_set/values_to_get
2447
class TestSetSecretCmd(TestPropertySetterCmd):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2448
    command = command.SetSecret
1047 by Teddy Hogeborn
mandos-ctl: Refactor
2449
    propname = "Secret"
1042 by Teddy Hogeborn
mandos-ctl: Bug fix: close an open file
2450
    values_to_set = [io.BytesIO(b""),
1030 by Teddy Hogeborn
mandos-ctl: Fix bugs
2451
                     io.BytesIO(b"secret\0xyzzy\nbar")]
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
2452
    values_to_get = [f.getvalue() for f in values_to_set]
1024 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
2453
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2454
1093 by Teddy Hogeborn
mandos-ctl: Refactor: fix confusion about values_to_set/values_to_get
2455
class TestSetTimeoutCmd(TestPropertySetterCmd):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2456
    command = command.SetTimeout
1047 by Teddy Hogeborn
mandos-ctl: Refactor
2457
    propname = "Timeout"
1035 by Teddy Hogeborn
mandos-ctl: Refactor; move parsing of intervals into argument parsing
2458
    values_to_set = [datetime.timedelta(),
2459
                     datetime.timedelta(minutes=5),
2460
                     datetime.timedelta(seconds=1),
2461
                     datetime.timedelta(weeks=1),
2462
                     datetime.timedelta(weeks=52)]
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
2463
    values_to_get = [dt.total_seconds()*1000 for dt in values_to_set]
1024 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
2464
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2465
1093 by Teddy Hogeborn
mandos-ctl: Refactor: fix confusion about values_to_set/values_to_get
2466
class TestSetExtendedTimeoutCmd(TestPropertySetterCmd):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2467
    command = command.SetExtendedTimeout
1047 by Teddy Hogeborn
mandos-ctl: Refactor
2468
    propname = "ExtendedTimeout"
1035 by Teddy Hogeborn
mandos-ctl: Refactor; move parsing of intervals into argument parsing
2469
    values_to_set = [datetime.timedelta(),
2470
                     datetime.timedelta(minutes=5),
2471
                     datetime.timedelta(seconds=1),
2472
                     datetime.timedelta(weeks=1),
2473
                     datetime.timedelta(weeks=52)]
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
2474
    values_to_get = [dt.total_seconds()*1000 for dt in values_to_set]
1024 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
2475
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2476
1093 by Teddy Hogeborn
mandos-ctl: Refactor: fix confusion about values_to_set/values_to_get
2477
class TestSetIntervalCmd(TestPropertySetterCmd):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2478
    command = command.SetInterval
1047 by Teddy Hogeborn
mandos-ctl: Refactor
2479
    propname = "Interval"
1035 by Teddy Hogeborn
mandos-ctl: Refactor; move parsing of intervals into argument parsing
2480
    values_to_set = [datetime.timedelta(),
2481
                     datetime.timedelta(minutes=5),
2482
                     datetime.timedelta(seconds=1),
2483
                     datetime.timedelta(weeks=1),
2484
                     datetime.timedelta(weeks=52)]
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
2485
    values_to_get = [dt.total_seconds()*1000 for dt in values_to_set]
1024 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
2486
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2487
1093 by Teddy Hogeborn
mandos-ctl: Refactor: fix confusion about values_to_set/values_to_get
2488
class TestSetApprovalDelayCmd(TestPropertySetterCmd):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2489
    command = command.SetApprovalDelay
1047 by Teddy Hogeborn
mandos-ctl: Refactor
2490
    propname = "ApprovalDelay"
1035 by Teddy Hogeborn
mandos-ctl: Refactor; move parsing of intervals into argument parsing
2491
    values_to_set = [datetime.timedelta(),
2492
                     datetime.timedelta(minutes=5),
2493
                     datetime.timedelta(seconds=1),
2494
                     datetime.timedelta(weeks=1),
2495
                     datetime.timedelta(weeks=52)]
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
2496
    values_to_get = [dt.total_seconds()*1000 for dt in values_to_set]
1024 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
2497
1055 by Teddy Hogeborn
mandos-ctl: Refactor
2498
1093 by Teddy Hogeborn
mandos-ctl: Refactor: fix confusion about values_to_set/values_to_get
2499
class TestSetApprovalDurationCmd(TestPropertySetterCmd):
1077 by Teddy Hogeborn
mandos-ctl: Refactor
2500
    command = command.SetApprovalDuration
1047 by Teddy Hogeborn
mandos-ctl: Refactor
2501
    propname = "ApprovalDuration"
1035 by Teddy Hogeborn
mandos-ctl: Refactor; move parsing of intervals into argument parsing
2502
    values_to_set = [datetime.timedelta(),
2503
                     datetime.timedelta(minutes=5),
2504
                     datetime.timedelta(seconds=1),
2505
                     datetime.timedelta(weeks=1),
2506
                     datetime.timedelta(weeks=52)]
1086 by Teddy Hogeborn
mandos-ctl: Refactor tests
2507
    values_to_get = [dt.total_seconds()*1000 for dt in values_to_set]
1024 by Teddy Hogeborn
mandos-ctl: Add more tests, including tests for all commands
2508
1041 by Teddy Hogeborn
mandos-ctl: Add tests for option syntax checks
2509
986 by Teddy Hogeborn
Add tests to mandos-ctl's milliseconds_to_string function
2510

984 by Teddy Hogeborn
Make mandos-ctl use unittest instead of doctest module
2511
def should_only_run_tests():
2512
    parser = argparse.ArgumentParser(add_help=False)
2513
    parser.add_argument("--check", action='store_true')
2514
    args, unknown_args = parser.parse_known_args()
2515
    run_tests = args.check
2516
    if run_tests:
2517
        # Remove --check argument from sys.argv
2518
        sys.argv[1:] = unknown_args
2519
    return run_tests
2520
2521
# Add all tests from doctest strings
2522
def load_tests(loader, tests, none):
2523
    import doctest
2524
    tests.addTests(doctest.DocTestSuite())
2525
    return tests
745 by Teddy Hogeborn
mandos-ctl: Do minor formatting and whitespace adjustments.
2526
463.1.8 by teddy at bsnet
* mandos-ctl: Use unicode string literals.
2527
if __name__ == "__main__":
1067 by Teddy Hogeborn
mandos-ctl: Bug fix: always shutdown logging.
2528
    try:
2529
        if should_only_run_tests():
2530
            # Call using ./tdd-python-script --check [--verbose]
2531
            unittest.main()
2532
        else:
2533
            main()
2534
    finally:
2535
        logging.shutdown()