]>
Commit | Line | Data |
---|---|---|
ae54b3a7 EK |
1 | /* |
2 | * Copyright (C) 2020 Ed Kellett | |
3 | * | |
4 | * This program is free software; you can redistribute it and/or modify | |
5 | * it under the terms of the GNU General Public License as published by | |
6 | * the Free Software Foundation; either version 2 of the License, or | |
7 | * (at your option) any later version. | |
8 | * | |
9 | * This program is distributed in the hope that it will be useful, | |
10 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
12 | * GNU General Public License for more details. | |
13 | * | |
14 | * You should have received a copy of the GNU General Public License | |
15 | * along with this program; if not, write to the Free Software | |
16 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 | |
17 | * USA | |
18 | */ | |
19 | #include <stdio.h> | |
20 | #include <string.h> | |
21 | #include <stdlib.h> | |
22 | #include <unistd.h> | |
23 | #include "tap/basic.h" | |
24 | ||
25 | #include "stdinc.h" | |
26 | #include "client.h" | |
27 | #include "privilege.h" | |
28 | ||
29 | #define MSG "%s:%d (%s)", __FILE__, __LINE__, __FUNCTION__ | |
30 | ||
31 | struct Client me; | |
32 | ||
33 | static void cleanup(void) | |
34 | { | |
35 | privilegeset_prepare_rehash(); | |
36 | privilegeset_cleanup_rehash(); | |
37 | } | |
38 | ||
39 | static void test_privset_membership(void) | |
40 | { | |
41 | struct PrivilegeSet *set = privilegeset_set_new("test", "foo bar", 0); | |
42 | ||
43 | is_bool(true, privilegeset_in_set(set, "foo"), MSG); | |
44 | is_bool(true, privilegeset_in_set(set, "bar"), MSG); | |
45 | ||
46 | is_bool(false, privilegeset_in_set(set, "qux"), MSG); | |
47 | ||
48 | cleanup(); | |
49 | } | |
50 | ||
51 | static void test_privset_extend(void) | |
52 | { | |
53 | struct PrivilegeSet *parent = privilegeset_set_new("parent", "foo bar", 0); | |
54 | struct PrivilegeSet *child = privilegeset_extend(parent, "child", "qux", 0); | |
55 | ||
56 | is_bool(true, privilegeset_in_set(child, "foo"), MSG); | |
57 | is_bool(true, privilegeset_in_set(child, "bar"), MSG); | |
58 | ||
59 | is_bool(false, privilegeset_in_set(parent, "qux"), MSG); | |
60 | is_bool(true, privilegeset_in_set(child, "qux"), MSG); | |
61 | ||
62 | cleanup(); | |
63 | } | |
64 | ||
65 | static void test_privset_persistence(void) | |
66 | { | |
67 | struct PrivilegeSet *set = privilegeset_set_new("test", "foo", 0); | |
68 | privilegeset_ref(set); | |
69 | ||
70 | /* should survive rehash since it's referenced, but become empty */ | |
71 | privilegeset_prepare_rehash(); | |
72 | privilegeset_cleanup_rehash(); | |
73 | is_bool(false, privilegeset_in_set(set, "foo"), MSG); | |
74 | ||
75 | /* and have its contents replaced by the equal name */ | |
76 | privilegeset_set_new("test", "bar", 0); | |
77 | is_bool(true, privilegeset_in_set(set, "bar"), MSG); | |
78 | ||
79 | privilegeset_unref(set); | |
80 | cleanup(); | |
81 | } | |
82 | ||
83 | static void test_privset_diff(void) | |
84 | { | |
85 | struct PrivilegeSet *old = privilegeset_set_new("old", "foo bar", 0); | |
86 | struct PrivilegeSet *new = privilegeset_set_new("new", "foo qux", 0); | |
87 | const struct PrivilegeSet *added, *removed, *unchanged; | |
88 | const struct PrivilegeSet **result = privilegeset_diff(old, new); | |
89 | unchanged = result[0]; | |
90 | added = result[1]; | |
91 | removed = result[2]; | |
92 | ||
93 | is_bool(true, privilegeset_in_set(unchanged, "foo"), MSG); | |
94 | is_bool(false, privilegeset_in_set(added, "foo"), MSG); | |
95 | is_bool(false, privilegeset_in_set(removed, "foo"), MSG); | |
96 | ||
97 | is_bool(false, privilegeset_in_set(unchanged, "bar"), MSG); | |
98 | is_bool(false, privilegeset_in_set(added, "bar"), MSG); | |
99 | is_bool(true, privilegeset_in_set(removed, "bar"), MSG); | |
100 | ||
101 | is_bool(false, privilegeset_in_set(unchanged, "qux"), MSG); | |
102 | is_bool(true, privilegeset_in_set(added, "qux"), MSG); | |
103 | is_bool(false, privilegeset_in_set(removed, "qux"), MSG); | |
104 | ||
105 | cleanup(); | |
106 | } | |
107 | ||
108 | static void test_privset_diff_rehash(void) | |
109 | { | |
110 | struct PrivilegeSet *set = privilegeset_set_new("test", "foo bar", 0); | |
111 | const struct PrivilegeSet *added, *removed, *unchanged; | |
112 | const struct PrivilegeSet **result; | |
113 | privilegeset_ref(set); | |
114 | ||
115 | privilegeset_prepare_rehash(); | |
116 | ||
117 | /* should have changed from foo, bar to nothing, i.e. -foo -bar */ | |
118 | result = privilegeset_diff(set->shadow, set); | |
119 | unchanged = result[0]; | |
120 | added = result[1]; | |
121 | removed = result[2]; | |
122 | ||
123 | is_bool(false, privilegeset_in_set(unchanged, "foo"), MSG); | |
124 | is_bool(false, privilegeset_in_set(added, "foo"), MSG); | |
125 | is_bool(true, privilegeset_in_set(removed, "foo"), MSG); | |
126 | ||
127 | is_bool(false, privilegeset_in_set(unchanged, "bar"), MSG); | |
128 | is_bool(false, privilegeset_in_set(added, "bar"), MSG); | |
129 | is_bool(true, privilegeset_in_set(removed, "bar"), MSG); | |
130 | ||
131 | privilegeset_set_new("test", "foo qux", 0); | |
132 | result = privilegeset_diff(set->shadow, set); | |
133 | unchanged = result[0]; | |
134 | added = result[1]; | |
135 | removed = result[2]; | |
136 | ||
137 | /* should have changed from foo, bar to foo, qux, i.e. =foo -bar +qux */ | |
138 | is_bool(true, privilegeset_in_set(unchanged, "foo"), MSG); | |
139 | is_bool(false, privilegeset_in_set(added, "foo"), MSG); | |
140 | is_bool(false, privilegeset_in_set(removed, "foo"), MSG); | |
141 | ||
142 | is_bool(false, privilegeset_in_set(unchanged, "bar"), MSG); | |
143 | is_bool(false, privilegeset_in_set(added, "bar"), MSG); | |
144 | is_bool(true, privilegeset_in_set(removed, "bar"), MSG); | |
145 | ||
146 | is_bool(false, privilegeset_in_set(unchanged, "qux"), MSG); | |
147 | is_bool(true, privilegeset_in_set(added, "qux"), MSG); | |
148 | is_bool(false, privilegeset_in_set(removed, "qux"), MSG); | |
149 | ||
150 | privilegeset_cleanup_rehash(); | |
151 | ||
152 | privilegeset_unref(set); | |
153 | cleanup(); | |
154 | } | |
155 | ||
156 | int main(int argc, char *argv[]) | |
157 | { | |
158 | plan_lazy(); | |
159 | ||
160 | test_privset_membership(); | |
161 | test_privset_extend(); | |
162 | test_privset_persistence(); | |
163 | test_privset_diff(); | |
164 | test_privset_diff_rehash(); | |
165 | ||
166 | return 0; | |
167 | } |