Branch data Line data Source code
1 : : /* SPDX-License-Identifier: LGPL-2.1+ */
2 : :
3 : : #include <errno.h>
4 : : #include <fcntl.h>
5 : : #include <locale.h>
6 : : #include <unistd.h>
7 : :
8 : : #include "sd-messages.h"
9 : :
10 : : #include "alloc-util.h"
11 : : #include "catalog.h"
12 : : #include "fd-util.h"
13 : : #include "fs-util.h"
14 : : #include "log.h"
15 : : #include "macro.h"
16 : : #include "path-util.h"
17 : : #include "string-util.h"
18 : : #include "strv.h"
19 : : #include "tests.h"
20 : : #include "tmpfile-util.h"
21 : : #include "util.h"
22 : :
23 : : static char** catalog_dirs = NULL;
24 : : static const char *no_catalog_dirs[] = {
25 : : "/bin/hopefully/with/no/catalog",
26 : : NULL
27 : : };
28 : :
29 : 20 : static Hashmap* test_import(const char* contents, ssize_t size, int code) {
30 : 20 : _cleanup_(unlink_tempfilep) char name[] = "/tmp/test-catalog.XXXXXX";
31 : 20 : _cleanup_close_ int fd;
32 : : Hashmap *h;
33 : :
34 [ + - ]: 20 : if (size < 0)
35 : 20 : size = strlen(contents);
36 : :
37 [ - + ]: 20 : assert_se(h = hashmap_new(&catalog_hash_ops));
38 : :
39 : 20 : fd = mkostemp_safe(name);
40 [ - + ]: 20 : assert_se(fd >= 0);
41 [ - + ]: 20 : assert_se(write(fd, contents, size) == size);
42 : :
43 [ - + ]: 20 : assert_se(catalog_import_file(h, name) == code);
44 : :
45 : 20 : return h;
46 : : }
47 : :
48 : 4 : static void test_catalog_import_invalid(void) {
49 : 4 : _cleanup_hashmap_free_free_free_ Hashmap *h = NULL;
50 : :
51 : 4 : h = test_import("xxx", -1, -EINVAL);
52 [ - + ]: 4 : assert_se(hashmap_isempty(h));
53 : 4 : }
54 : :
55 : 4 : static void test_catalog_import_badid(void) {
56 : 8 : _cleanup_hashmap_free_free_free_ Hashmap *h = NULL;
57 : 4 : const char *input =
58 : : "-- 0027229ca0644181a76c4e92458afaff dededededededededededededededede\n" \
59 : : "Subject: message\n" \
60 : : "\n" \
61 : : "payload\n";
62 : 4 : h = test_import(input, -1, -EINVAL);
63 : 4 : }
64 : :
65 : 4 : static void test_catalog_import_one(void) {
66 : 4 : _cleanup_hashmap_free_free_free_ Hashmap *h = NULL;
67 : : char *payload;
68 : : Iterator j;
69 : :
70 : 4 : const char *input =
71 : : "-- 0027229ca0644181a76c4e92458afaff dededededededededededededededed\n" \
72 : : "Subject: message\n" \
73 : : "\n" \
74 : : "payload\n";
75 : 4 : const char *expect =
76 : : "Subject: message\n" \
77 : : "\n" \
78 : : "payload\n";
79 : :
80 : 4 : h = test_import(input, -1, 0);
81 [ - + ]: 4 : assert_se(hashmap_size(h) == 1);
82 : :
83 [ + + ]: 8 : HASHMAP_FOREACH(payload, h, j) {
84 : 4 : printf("expect: %s\n", expect);
85 : 4 : printf("actual: %s\n", payload);
86 [ - + ]: 4 : assert_se(streq(expect, payload));
87 : : }
88 : 4 : }
89 : :
90 : 4 : static void test_catalog_import_merge(void) {
91 : 4 : _cleanup_hashmap_free_free_free_ Hashmap *h = NULL;
92 : : char *payload;
93 : : Iterator j;
94 : :
95 : 4 : const char *input =
96 : : "-- 0027229ca0644181a76c4e92458afaff dededededededededededededededed\n" \
97 : : "Subject: message\n" \
98 : : "Defined-By: me\n" \
99 : : "\n" \
100 : : "payload\n" \
101 : : "\n" \
102 : : "-- 0027229ca0644181a76c4e92458afaff dededededededededededededededed\n" \
103 : : "Subject: override subject\n" \
104 : : "X-Header: hello\n" \
105 : : "\n" \
106 : : "override payload\n";
107 : :
108 : 4 : const char *combined =
109 : : "Subject: override subject\n" \
110 : : "X-Header: hello\n" \
111 : : "Subject: message\n" \
112 : : "Defined-By: me\n" \
113 : : "\n" \
114 : : "override payload\n";
115 : :
116 : 4 : h = test_import(input, -1, 0);
117 [ - + ]: 4 : assert_se(hashmap_size(h) == 1);
118 : :
119 [ + + ]: 8 : HASHMAP_FOREACH(payload, h, j) {
120 [ - + ]: 4 : assert_se(streq(combined, payload));
121 : : }
122 : 4 : }
123 : :
124 : 4 : static void test_catalog_import_merge_no_body(void) {
125 : 4 : _cleanup_hashmap_free_free_free_ Hashmap *h = NULL;
126 : : char *payload;
127 : : Iterator j;
128 : :
129 : 4 : const char *input =
130 : : "-- 0027229ca0644181a76c4e92458afaff dededededededededededededededed\n" \
131 : : "Subject: message\n" \
132 : : "Defined-By: me\n" \
133 : : "\n" \
134 : : "payload\n" \
135 : : "\n" \
136 : : "-- 0027229ca0644181a76c4e92458afaff dededededededededededededededed\n" \
137 : : "Subject: override subject\n" \
138 : : "X-Header: hello\n" \
139 : : "\n";
140 : :
141 : 4 : const char *combined =
142 : : "Subject: override subject\n" \
143 : : "X-Header: hello\n" \
144 : : "Subject: message\n" \
145 : : "Defined-By: me\n" \
146 : : "\n" \
147 : : "payload\n";
148 : :
149 : 4 : h = test_import(input, -1, 0);
150 [ - + ]: 4 : assert_se(hashmap_size(h) == 1);
151 : :
152 [ + + ]: 8 : HASHMAP_FOREACH(payload, h, j) {
153 [ - + ]: 4 : assert_se(streq(combined, payload));
154 : : }
155 : 4 : }
156 : :
157 : 4 : static void test_catalog_update(const char *database) {
158 : : int r;
159 : :
160 : : /* Test what happens if there are no files. */
161 : 4 : r = catalog_update(database, NULL, NULL);
162 [ - + ]: 4 : assert_se(r == 0);
163 : :
164 : : /* Test what happens if there are no files in the directory. */
165 : 4 : r = catalog_update(database, NULL, no_catalog_dirs);
166 [ - + ]: 4 : assert_se(r == 0);
167 : :
168 : : /* Make sure that we at least have some files loaded or the
169 : : * catalog_list below will fail. */
170 : 4 : r = catalog_update(database, NULL, (const char * const *) catalog_dirs);
171 [ - + ]: 4 : assert_se(r == 0);
172 : 4 : }
173 : :
174 : 4 : static void test_catalog_file_lang(void) {
175 : 4 : _cleanup_free_ char *lang = NULL, *lang2 = NULL, *lang3 = NULL, *lang4 = NULL;
176 : :
177 [ - + ]: 4 : assert_se(catalog_file_lang("systemd.de_DE.catalog", &lang) == 1);
178 [ - + ]: 4 : assert_se(streq(lang, "de_DE"));
179 : :
180 [ - + ]: 4 : assert_se(catalog_file_lang("systemd..catalog", &lang2) == 0);
181 [ - + ]: 4 : assert_se(lang2 == NULL);
182 : :
183 [ - + ]: 4 : assert_se(catalog_file_lang("systemd.fr.catalog", &lang2) == 1);
184 [ - + ]: 4 : assert_se(streq(lang2, "fr"));
185 : :
186 [ - + ]: 4 : assert_se(catalog_file_lang("systemd.fr.catalog.gz", &lang3) == 0);
187 [ - + ]: 4 : assert_se(lang3 == NULL);
188 : :
189 [ - + ]: 4 : assert_se(catalog_file_lang("systemd.01234567890123456789012345678901.catalog", &lang3) == 0);
190 [ - + ]: 4 : assert_se(lang3 == NULL);
191 : :
192 [ - + ]: 4 : assert_se(catalog_file_lang("systemd.0123456789012345678901234567890.catalog", &lang3) == 1);
193 [ - + ]: 4 : assert_se(streq(lang3, "0123456789012345678901234567890"));
194 : :
195 [ - + ]: 4 : assert_se(catalog_file_lang("/x/y/systemd.catalog", &lang4) == 0);
196 [ - + ]: 4 : assert_se(lang4 == NULL);
197 : :
198 [ - + ]: 4 : assert_se(catalog_file_lang("/x/y/systemd.ru_RU.catalog", &lang4) == 1);
199 [ - + ]: 4 : assert_se(streq(lang4, "ru_RU"));
200 : 4 : }
201 : :
202 : 4 : int main(int argc, char *argv[]) {
203 : 4 : _cleanup_(unlink_tempfilep) char database[] = "/tmp/test-catalog.XXXXXX";
204 : 8 : _cleanup_free_ char *text = NULL;
205 : : int r;
206 : :
207 : 4 : setlocale(LC_ALL, "de_DE.UTF-8");
208 : :
209 : 4 : test_setup_logging(LOG_DEBUG);
210 : :
211 : : /* If test-catalog is located at the build directory, then use catalogs in that.
212 : : * If it is not, e.g. installed by systemd-tests package, then use installed catalogs. */
213 : 4 : catalog_dirs = STRV_MAKE(get_catalog_dir());
214 : :
215 [ - + ]: 4 : assert_se(access(catalog_dirs[0], F_OK) >= 0);
216 [ + - ]: 4 : log_notice("Using catalog directory '%s'", catalog_dirs[0]);
217 : :
218 : 4 : test_catalog_file_lang();
219 : :
220 : 4 : test_catalog_import_invalid();
221 : 4 : test_catalog_import_badid();
222 : 4 : test_catalog_import_one();
223 : 4 : test_catalog_import_merge();
224 : 4 : test_catalog_import_merge_no_body();
225 : :
226 [ - + ]: 4 : assert_se(mkostemp_safe(database) >= 0);
227 : :
228 : 4 : test_catalog_update(database);
229 : :
230 : 4 : r = catalog_list(stdout, database, true);
231 [ - + ]: 4 : assert_se(r >= 0);
232 : :
233 : 4 : r = catalog_list(stdout, database, false);
234 [ - + ]: 4 : assert_se(r >= 0);
235 : :
236 [ - + ]: 4 : assert_se(catalog_get(database, SD_MESSAGE_COREDUMP, &text) >= 0);
237 : 4 : printf(">>>%s<<<\n", text);
238 : :
239 : 4 : return 0;
240 : : }
|
