/mandos/release

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

« back to all changes in this revision

Viewing changes to mandos-ctl

  • Committer: Teddy Hogeborn
  • Date: 2019-03-04 19:03:10 UTC
  • mto: (237.7.594 trunk)
  • mto: This revision was merged to the branch mainline in revision 382.
  • Revision ID: teddy@recompile.se-20190304190310-n6brlijq8s692dok
mandos-ctl: New tests for ApproveCmd and DenyCmd

* mandos-ctl (TestCmd.MockClient.Approve): New method.
  (TestApproveCmd, TestDenyCmd): New.

Show diffs side-by-side

added added

removed removed

Lines of Context:
mandos-ctl
278
278
        commands which want to operate on all clients at the same time
279
279
        can override this run() method instead."""
280
280
        self.mandos = mandos
281
 
        for client, properties in clients.items():
282
 
            self.run_on_one_client(client, properties)
 
281
        for client in clients:
 
282
            self.run_on_one_client(client)
283
283
 
284
284
class PrintCmd(Command):
285
285
    """Abstract class for commands printing client details"""
295
295
 
296
296
class PropertyCmd(Command):
297
297
    """Abstract class for Actions for setting one client property"""
298
 
    def run_on_one_client(self, client, properties):
 
298
    def run_on_one_client(self, client):
299
299
        """Set the Client's D-Bus property"""
300
300
        client.Set(client_interface, self.property, self.value_to_set,
301
301
                   dbus_interface=dbus.PROPERTIES_IFACE)
323
323
        self.verbose = verbose
324
324
 
325
325
    def output(self, clients):
326
 
        default_keywords = ("Name", "Enabled", "Timeout", "LastCheckedOK")
327
 
        keywords = default_keywords
328
326
        if self.verbose:
329
327
            keywords = self.all_keywords
 
328
        else:
 
329
            keywords = ("Name", "Enabled", "Timeout", "LastCheckedOK")
330
330
        return str(self.TableOfClients(clients.values(), keywords))
331
331
 
332
332
    class TableOfClients(object):
419
419
        return value
420
420
 
421
421
class IsEnabledCmd(Command):
422
 
    def run_on_one_client(self, client, properties):
423
 
        if self.is_enabled(client, properties):
 
422
    def run_on_one_client(self, client):
 
423
        if self.is_enabled(client):
424
424
            sys.exit(0)
425
425
        sys.exit(1)
426
 
    def is_enabled(self, client, properties):
427
 
        return bool(properties["Enabled"])
 
426
    def is_enabled(self, client):
 
427
        return client.Get(client_interface, "Enabled",
 
428
                          dbus_interface=dbus.PROPERTIES_IFACE)
428
429
 
429
430
class RemoveCmd(Command):
430
 
    def run_on_one_client(self, client, properties):
 
431
    def run_on_one_client(self, client):
431
432
        self.mandos.RemoveClient(client.__dbus_object_path__)
432
433
 
433
434
class ApproveCmd(Command):
434
 
    def run_on_one_client(self, client, properties):
 
435
    def run_on_one_client(self, client):
435
436
        client.Approve(dbus.Boolean(True),
436
437
                       dbus_interface=client_interface)
437
438
 
438
439
class DenyCmd(Command):
439
 
    def run_on_one_client(self, client, properties):
 
440
    def run_on_one_client(self, client):
440
441
        client.Approve(dbus.Boolean(False),
441
442
                       dbus_interface=client_interface)
442
443
 
580
581
    parser.add_argument("client", nargs="*", help="Client name")
581
582
 
582
583
 
583
 
def commands_from_options(options):
 
584
def commands_and_clients_from_options(options):
584
585
 
585
586
    commands = []
586
587
 
594
595
        commands.append(DisableCmd())
595
596
 
596
597
    if options.bump_timeout:
597
 
        commands.append(BumpTimeoutCmd())
 
598
        commands.append(BumpTimeoutCmd(options.bump_timeout))
598
599
 
599
600
    if options.start_checker:
600
601
        commands.append(StartCheckerCmd())
651
652
    if not commands:
652
653
        commands.append(PrintTableCmd(verbose=options.verbose))
653
654
 
654
 
    return commands
 
655
    return commands, options.client
655
656
 
656
657
 
657
658
def main():
673
674
    if options.is_enabled and len(options.client) > 1:
674
675
        parser.error("--is-enabled requires exactly one client")
675
676
 
676
 
    clientnames = options.client
 
677
    commands, clientnames = commands_and_clients_from_options(options)
677
678
 
678
679
    try:
679
680
        bus = dbus.SystemBus()
726
727
                sys.exit(1)
727
728
 
728
729
    # Run all commands on clients
729
 
    commands = commands_from_options(options)
730
730
    for command in commands:
731
731
        command.run(mandos_serv, clients)
732
732
 
746
746
 
747
747
class Test_string_to_delta(unittest.TestCase):
748
748
    def test_handles_basic_rfc3339(self):
749
 
        self.assertEqual(string_to_delta("PT0S"),
750
 
                         datetime.timedelta())
751
 
        self.assertEqual(string_to_delta("P0D"),
752
 
                         datetime.timedelta())
753
 
        self.assertEqual(string_to_delta("PT1S"),
754
 
                         datetime.timedelta(0, 1))
755
749
        self.assertEqual(string_to_delta("PT2H"),
756
750
                         datetime.timedelta(0, 7200))
757
751
    def test_falls_back_to_pre_1_6_1_with_warning(self):
792
786
                testcase.assertEqual(dbus_interface,
793
787
                                     dbus.PROPERTIES_IFACE)
794
788
                self.attributes[property] = value
 
789
                self.calls.append(("Set", (interface, property, value,
 
790
                                           dbus_interface)))
795
791
            def Get(self, interface, property, dbus_interface):
796
792
                testcase.assertEqual(interface, client_interface)
797
793
                testcase.assertEqual(dbus_interface,
798
794
                                     dbus.PROPERTIES_IFACE)
 
795
                self.calls.append(("Get", (interface, property,
 
796
                                           dbus_interface)))
799
797
                return self.attributes[property]
800
798
            def Approve(self, approve, dbus_interface):
801
799
                testcase.assertEqual(dbus_interface, client_interface)
802
800
                self.calls.append(("Approve", (approve,
803
801
                                               dbus_interface)))
804
 
        self.client = MockClient(
805
 
            "foo",
806
 
            KeyID=("92ed150794387c03ce684574b1139a65"
807
 
                   "94a34f895daaaf09fd8ea90a27cddb12"),
808
 
            Secret=b"secret",
809
 
            Host="foo.example.org",
810
 
            Enabled=dbus.Boolean(True),
811
 
            Timeout=300000,
812
 
            LastCheckedOK="2019-02-03T00:00:00",
813
 
            Created="2019-01-02T00:00:00",
814
 
            Interval=120000,
815
 
            Fingerprint=("778827225BA7DE539C5A"
816
 
                         "7CFA59CFF7CDBD9A5920"),
817
 
            CheckerRunning=dbus.Boolean(False),
818
 
            LastEnabled="2019-01-03T00:00:00",
819
 
            ApprovalPending=dbus.Boolean(False),
820
 
            ApprovedByDefault=dbus.Boolean(True),
821
 
            LastApprovalRequest="",
822
 
            ApprovalDelay=0,
823
 
            ApprovalDuration=1000,
824
 
            Checker="fping -q -- %(host)s",
825
 
            ExtendedTimeout=900000,
826
 
            Expires="2019-02-04T00:00:00",
827
 
            LastCheckerStatus=0)
828
 
        self.other_client = MockClient(
829
 
            "barbar",
830
 
            KeyID=("0558568eedd67d622f5c83b35a115f79"
831
 
                   "6ab612cff5ad227247e46c2b020f441c"),
832
 
            Secret=b"secretbar",
833
 
            Host="192.0.2.3",
834
 
            Enabled=dbus.Boolean(True),
835
 
            Timeout=300000,
836
 
            LastCheckedOK="2019-02-04T00:00:00",
837
 
            Created="2019-01-03T00:00:00",
838
 
            Interval=120000,
839
 
            Fingerprint=("3E393AEAEFB84C7E89E2"
840
 
                         "F547B3A107558FCA3A27"),
841
 
            CheckerRunning=dbus.Boolean(True),
842
 
            LastEnabled="2019-01-04T00:00:00",
843
 
            ApprovalPending=dbus.Boolean(False),
844
 
            ApprovedByDefault=dbus.Boolean(False),
845
 
            LastApprovalRequest="2019-01-03T00:00:00",
846
 
            ApprovalDelay=30000,
847
 
            ApprovalDuration=1000,
848
 
            Checker=":",
849
 
            ExtendedTimeout=900000,
850
 
            Expires="2019-02-05T00:00:00",
851
 
            LastCheckerStatus=-2)
852
 
        self.clients =  collections.OrderedDict(
853
 
            [
854
 
                (self.client, self.client.attributes),
855
 
                (self.other_client, self.other_client.attributes),
 
802
            def __getitem__(self, key):
 
803
                return self.attributes[key]
 
804
            def __setitem__(self, key, value):
 
805
                self.attributes[key] = value
 
806
        self.clients = collections.OrderedDict([
 
807
            ("foo",
 
808
             MockClient(
 
809
                 "foo",
 
810
                 KeyID=("92ed150794387c03ce684574b1139a65"
 
811
                        "94a34f895daaaf09fd8ea90a27cddb12"),
 
812
                 Secret=b"secret",
 
813
                 Host="foo.example.org",
 
814
                 Enabled=dbus.Boolean(True),
 
815
                 Timeout=300000,
 
816
                 LastCheckedOK="2019-02-03T00:00:00",
 
817
                 Created="2019-01-02T00:00:00",
 
818
                 Interval=120000,
 
819
                 Fingerprint=("778827225BA7DE539C5A"
 
820
                              "7CFA59CFF7CDBD9A5920"),
 
821
                 CheckerRunning=dbus.Boolean(False),
 
822
                 LastEnabled="2019-01-03T00:00:00",
 
823
                 ApprovalPending=dbus.Boolean(False),
 
824
                 ApprovedByDefault=dbus.Boolean(True),
 
825
                 LastApprovalRequest="",
 
826
                 ApprovalDelay=0,
 
827
                 ApprovalDuration=1000,
 
828
                 Checker="fping -q -- %(host)s",
 
829
                 ExtendedTimeout=900000,
 
830
                 Expires="2019-02-04T00:00:00",
 
831
                 LastCheckerStatus=0)),
 
832
            ("barbar",
 
833
             MockClient(
 
834
                 "barbar",
 
835
                 KeyID=("0558568eedd67d622f5c83b35a115f79"
 
836
                        "6ab612cff5ad227247e46c2b020f441c"),
 
837
                 Secret=b"secretbar",
 
838
                 Host="192.0.2.3",
 
839
                 Enabled=dbus.Boolean(True),
 
840
                 Timeout=300000,
 
841
                 LastCheckedOK="2019-02-04T00:00:00",
 
842
                 Created="2019-01-03T00:00:00",
 
843
                 Interval=120000,
 
844
                 Fingerprint=("3E393AEAEFB84C7E89E2"
 
845
                              "F547B3A107558FCA3A27"),
 
846
                 CheckerRunning=dbus.Boolean(True),
 
847
                 LastEnabled="2019-01-04T00:00:00",
 
848
                 ApprovalPending=dbus.Boolean(False),
 
849
                 ApprovedByDefault=dbus.Boolean(False),
 
850
                 LastApprovalRequest="2019-01-03T00:00:00",
 
851
                 ApprovalDelay=30000,
 
852
                 ApprovalDuration=1000,
 
853
                 Checker=":",
 
854
                 ExtendedTimeout=900000,
 
855
                 Expires="2019-02-05T00:00:00",
 
856
                 LastCheckerStatus=-2)),
856
857
            ])
857
 
        self.one_client = {self.client: self.client.attributes}
 
858
        self.client = self.clients["foo"]
858
859
 
859
860
class TestPrintTableCmd(TestCmd):
860
861
    def test_normal(self):
874
875
"""[1:-1]
875
876
        self.assertEqual(output, expected_output)
876
877
    def test_one_client(self):
877
 
        output = PrintTableCmd().output(self.one_client)
 
878
        output = PrintTableCmd().output({"foo": self.client})
878
879
        expected_output = """
879
880
Name Enabled Timeout  Last Successful Check
880
881
foo  Yes     00:05:00 2019-02-03T00:00:00  
938
939
        json_data = json.loads(DumpJSONCmd().output(self.clients))
939
940
        self.assertDictEqual(json_data, self.expected_json)
940
941
    def test_one_client(self):
941
 
        clients = self.one_client
 
942
        clients = {"foo": self.client}
942
943
        json_data = json.loads(DumpJSONCmd().output(clients))
943
944
        expected_json = {"foo": self.expected_json["foo"]}
944
945
        self.assertDictEqual(json_data, expected_json)
945
946
 
946
947
class TestIsEnabledCmd(TestCmd):
947
948
    def test_is_enabled(self):
948
 
        self.assertTrue(all(IsEnabledCmd().is_enabled(client, properties)
949
 
                            for client, properties in self.clients.items()))
 
949
        self.assertTrue(all(IsEnabledCmd().is_enabled(client)
 
950
                            for client in self.clients.values()))
 
951
    def test_is_enabled_does_get_attribute(self):
 
952
        self.assertTrue(IsEnabledCmd().is_enabled(self.client))
 
953
        self.assertListEqual(self.client.calls,
 
954
                             [("Get",
 
955
                               ("se.recompile.Mandos.Client",
 
956
                                "Enabled",
 
957
                                "org.freedesktop.DBus.Properties"))])
950
958
    def test_is_enabled_run_exits_successfully(self):
951
959
        with self.assertRaises(SystemExit) as e:
952
 
            IsEnabledCmd().run(None, self.one_client)
 
960
            IsEnabledCmd().run(None, [self.client])
953
961
        if e.exception.code is not None:
954
962
            self.assertEqual(e.exception.code, 0)
955
963
        else:
956
964
            self.assertIsNone(e.exception.code)
957
965
    def test_is_enabled_run_exits_with_failure(self):
958
 
        self.client.attributes["Enabled"] = dbus.Boolean(False)
 
966
        self.client["Enabled"] = dbus.Boolean(False)
959
967
        with self.assertRaises(SystemExit) as e:
960
 
            IsEnabledCmd().run(None, self.one_client)
 
968
            IsEnabledCmd().run(None, [self.client])
961
969
        if isinstance(e.exception.code, int):
962
970
            self.assertNotEqual(e.exception.code, 0)
963
971
        else:
971
979
            def RemoveClient(self, dbus_path):
972
980
                self.calls.append(("RemoveClient", (dbus_path,)))
973
981
        mandos = MockMandos()
974
 
        super(TestRemoveCmd, self).setUp()
975
 
        RemoveCmd().run(mandos, self.clients)
976
 
        self.assertEqual(len(mandos.calls), 2)
977
 
        for client in self.clients:
978
 
            self.assertIn(("RemoveClient",
979
 
                           (client.__dbus_object_path__,)),
980
 
                          mandos.calls)
 
982
        RemoveCmd().run(mandos, [self.client])
 
983
        self.assertEqual(len(mandos.calls), 1)
 
984
        self.assertListEqual(mandos.calls,
 
985
                             [("RemoveClient",
 
986
                               (self.client.__dbus_object_path__,))])
981
987
 
982
988
class TestApproveCmd(TestCmd):
983
989
    def test_approve(self):
984
 
        ApproveCmd().run(None, self.clients)
985
 
        for client in self.clients:
986
 
            self.assertIn(("Approve", (True, client_interface)),
987
 
                          client.calls)
988
 
 
 
990
        ApproveCmd().run(None, [self.client])
 
991
        self.assertListEqual(self.client.calls,
 
992
                             [("Approve", (True, client_interface))])
989
993
class TestDenyCmd(TestCmd):
990
 
    def test_deny(self):
991
 
        DenyCmd().run(None, self.clients)
992
 
        for client in self.clients:
993
 
            self.assertIn(("Approve", (False, client_interface)),
994
 
                          client.calls)
995
 
 
996
 
class TestEnableCmd(TestCmd):
997
 
    def test_enable(self):
998
 
        for client in self.clients:
999
 
            client.attributes["Enabled"] = False
1000
 
 
1001
 
        EnableCmd().run(None, self.clients)
1002
 
 
1003
 
        for client in self.clients:
1004
 
            self.assertTrue(client.attributes["Enabled"])
1005
 
 
1006
 
class TestDisableCmd(TestCmd):
1007
 
    def test_disable(self):
1008
 
        DisableCmd().run(None, self.clients)
1009
 
 
1010
 
        for client in self.clients:
1011
 
            self.assertFalse(client.attributes["Enabled"])
1012
 
 
1013
 
class Unique(object):
1014
 
    """Class for objects which exist only to be unique objects, since
1015
 
unittest.mock.sentinel only exists in Python 3.3"""
1016
 
 
1017
 
class TestPropertyCmd(TestCmd):
1018
 
    """Abstract class for tests of PropertyCmd classes"""
1019
 
    def runTest(self):
1020
 
        if not hasattr(self, "command"):
1021
 
            return
1022
 
        values_to_get = getattr(self, "values_to_get",
1023
 
                                self.values_to_set)
1024
 
        for value_to_set, value_to_get in zip(self.values_to_set,
1025
 
                                              values_to_get):
1026
 
            for client in self.clients:
1027
 
                old_value = client.attributes[self.property]
1028
 
                self.assertNotIsInstance(old_value, Unique)
1029
 
                client.attributes[self.property] = Unique()
1030
 
            self.run_command(value_to_set, self.clients)
1031
 
            for client in self.clients:
1032
 
                value = client.attributes[self.property]
1033
 
                self.assertNotIsInstance(value, Unique)
1034
 
                self.assertEqual(value, value_to_get)
1035
 
    def run_command(self, value, clients):
1036
 
        self.command().run(None, clients)
1037
 
 
1038
 
class TestBumpTimeoutCmd(TestPropertyCmd):
1039
 
    command = BumpTimeoutCmd
1040
 
    property = "LastCheckedOK"
1041
 
    values_to_set = [""]
1042
 
 
1043
 
class TestStartCheckerCmd(TestPropertyCmd):
1044
 
    command = StartCheckerCmd
1045
 
    property = "CheckerRunning"
1046
 
    values_to_set = [dbus.Boolean(True)]
1047
 
 
1048
 
class TestStopCheckerCmd(TestPropertyCmd):
1049
 
    command = StopCheckerCmd
1050
 
    property = "CheckerRunning"
1051
 
    values_to_set = [dbus.Boolean(False)]
1052
 
 
1053
 
class TestApproveByDefaultCmd(TestPropertyCmd):
1054
 
    command = ApproveByDefaultCmd
1055
 
    property = "ApprovedByDefault"
1056
 
    values_to_set = [dbus.Boolean(True)]
1057
 
 
1058
 
class TestDenyByDefaultCmd(TestPropertyCmd):
1059
 
    command = DenyByDefaultCmd
1060
 
    property = "ApprovedByDefault"
1061
 
    values_to_set = [dbus.Boolean(False)]
1062
 
 
1063
 
class TestValueArgumentPropertyCmd(TestPropertyCmd):
1064
 
    """Abstract class for tests of PropertyCmd classes using the
1065
 
ValueArgumentMixIn"""
1066
 
    def runTest(self):
1067
 
        if type(self) is TestValueArgumentPropertyCmd:
1068
 
            return
1069
 
        return super(TestValueArgumentPropertyCmd, self).runTest()
1070
 
    def run_command(self, value, clients):
1071
 
        self.command(value).run(None, clients)
1072
 
 
1073
 
class TestSetCheckerCmd(TestValueArgumentPropertyCmd):
1074
 
    command = SetCheckerCmd
1075
 
    property = "Checker"
1076
 
    values_to_set = ["", ":", "fping -q -- %s"]
1077
 
 
1078
 
class TestSetHostCmd(TestValueArgumentPropertyCmd):
1079
 
    command = SetHostCmd
1080
 
    property = "Host"
1081
 
    values_to_set = ["192.0.2.3", "foo.example.org"]
1082
 
 
1083
 
class TestSetSecretCmd(TestValueArgumentPropertyCmd):
1084
 
    command = SetSecretCmd
1085
 
    property = "Secret"
1086
 
    values_to_set = [b"", b"secret"]
1087
 
 
1088
 
class TestSetTimeoutCmd(TestValueArgumentPropertyCmd):
1089
 
    command = SetTimeoutCmd
1090
 
    property = "Timeout"
1091
 
    values_to_set = ["P0D", "PT5M", "PT1S", "PT120S", "P1Y"]
1092
 
    values_to_get = [0, 300000, 1000, 120000, 31449600000]
1093
 
 
1094
 
class TestSetExtendedTimeoutCmd(TestValueArgumentPropertyCmd):
1095
 
    command = SetExtendedTimeoutCmd
1096
 
    property = "ExtendedTimeout"
1097
 
    values_to_set = ["P0D", "PT5M", "PT1S", "PT120S", "P1Y"]
1098
 
    values_to_get = [0, 300000, 1000, 120000, 31449600000]
1099
 
 
1100
 
class TestSetIntervalCmd(TestValueArgumentPropertyCmd):
1101
 
    command = SetIntervalCmd
1102
 
    property = "Interval"
1103
 
    values_to_set = ["P0D", "PT5M", "PT1S", "PT120S", "P1Y"]
1104
 
    values_to_get = [0, 300000, 1000, 120000, 31449600000]
1105
 
 
1106
 
class TestSetApprovalDelayCmd(TestValueArgumentPropertyCmd):
1107
 
    command = SetApprovalDelayCmd
1108
 
    property = "ApprovalDelay"
1109
 
    values_to_set = ["P0D", "PT5M", "PT1S", "PT120S", "P1Y"]
1110
 
    values_to_get = [0, 300000, 1000, 120000, 31449600000]
1111
 
 
1112
 
class TestSetApprovalDurationCmd(TestValueArgumentPropertyCmd):
1113
 
    command = SetApprovalDurationCmd
1114
 
    property = "ApprovalDuration"
1115
 
    values_to_set = ["P0D", "PT5M", "PT1S", "PT120S", "P1Y"]
1116
 
    values_to_get = [0, 300000, 1000, 120000, 31449600000]
1117
 
 
1118
 
class TestOptions(unittest.TestCase):
1119
 
    def setUp(self):
1120
 
        self.parser = argparse.ArgumentParser()
1121
 
        add_command_line_options(self.parser)
1122
 
    def commands_from_args(self, args):
1123
 
        options = self.parser.parse_args(args)
1124
 
        return commands_from_options(options)
1125
 
    def test_default_is_show_table(self):
1126
 
        commands = self.commands_from_args([])
1127
 
        self.assertEqual(len(commands), 1)
1128
 
        command = commands[0]
1129
 
        self.assertIsInstance(command, PrintTableCmd)
1130
 
        self.assertEqual(command.verbose, False)
1131
 
    def test_show_table_verbose(self):
1132
 
        commands = self.commands_from_args(["--verbose"])
1133
 
        self.assertEqual(len(commands), 1)
1134
 
        command = commands[0]
1135
 
        self.assertIsInstance(command, PrintTableCmd)
1136
 
        self.assertEqual(command.verbose, True)
 
994
    def test_approve(self):
 
995
        DenyCmd().run(None, [self.client])
 
996
        self.assertListEqual(self.client.calls,
 
997
                             [("Approve", (False, client_interface))])
1137
998
 
1138
999
 
1139
1000