Branch data Line data Source code
1 : : /* SPDX-License-Identifier: LGPL-2.1+ */
2 : :
3 : : #include <fcntl.h>
4 : : #include <unistd.h>
5 : :
6 : : #include "chattr-util.h"
7 : : #include "io-util.h"
8 : : #include "journal-authenticate.h"
9 : : #include "journal-file.h"
10 : : #include "journal-vacuum.h"
11 : : #include "log.h"
12 : : #include "rm-rf.h"
13 : : #include "tests.h"
14 : :
15 : : static bool arg_keep = false;
16 : :
17 : 36 : static void mkdtemp_chdir_chattr(char *path) {
18 [ - + ]: 36 : assert_se(mkdtemp(path));
19 [ - + ]: 36 : assert_se(chdir(path) >= 0);
20 : :
21 : : /* Speed up things a bit on btrfs, ensuring that CoW is turned off for all files created in our
22 : : * directory during the test run */
23 : 36 : (void) chattr_path(path, FS_NOCOW_FL, FS_NOCOW_FL, NULL);
24 : 36 : }
25 : :
26 : 4 : static void test_non_empty(void) {
27 : : dual_timestamp ts;
28 : : JournalFile *f;
29 : : struct iovec iovec;
30 : : static const char test[] = "TEST1=1", test2[] = "TEST2=2";
31 : : Object *o;
32 : : uint64_t p;
33 : : sd_id128_t fake_boot_id;
34 : 4 : char t[] = "/var/tmp/journal-XXXXXX";
35 : :
36 : 4 : test_setup_logging(LOG_DEBUG);
37 : :
38 : 4 : mkdtemp_chdir_chattr(t);
39 : :
40 [ - + ]: 4 : assert_se(journal_file_open(-1, "test.journal", O_RDWR|O_CREAT, 0666, true, (uint64_t) -1, true, NULL, NULL, NULL, NULL, &f) == 0);
41 : :
42 [ - + ]: 4 : assert_se(dual_timestamp_get(&ts));
43 [ - + ]: 4 : assert_se(sd_id128_randomize(&fake_boot_id) == 0);
44 : :
45 : 4 : iovec = IOVEC_MAKE_STRING(test);
46 [ - + ]: 4 : assert_se(journal_file_append_entry(f, &ts, NULL, &iovec, 1, NULL, NULL, NULL) == 0);
47 : :
48 : 4 : iovec = IOVEC_MAKE_STRING(test2);
49 [ - + ]: 4 : assert_se(journal_file_append_entry(f, &ts, NULL, &iovec, 1, NULL, NULL, NULL) == 0);
50 : :
51 : 4 : iovec = IOVEC_MAKE_STRING(test);
52 [ - + ]: 4 : assert_se(journal_file_append_entry(f, &ts, &fake_boot_id, &iovec, 1, NULL, NULL, NULL) == 0);
53 : :
54 : : #if HAVE_GCRYPT
55 : 4 : journal_file_append_tag(f);
56 : : #endif
57 : 4 : journal_file_dump(f);
58 : :
59 [ - + ]: 4 : assert_se(journal_file_next_entry(f, 0, DIRECTION_DOWN, &o, &p) == 1);
60 [ - + ]: 4 : assert_se(le64toh(o->entry.seqnum) == 1);
61 : :
62 [ - + ]: 4 : assert_se(journal_file_next_entry(f, p, DIRECTION_DOWN, &o, &p) == 1);
63 [ - + ]: 4 : assert_se(le64toh(o->entry.seqnum) == 2);
64 : :
65 [ - + ]: 4 : assert_se(journal_file_next_entry(f, p, DIRECTION_DOWN, &o, &p) == 1);
66 [ - + ]: 4 : assert_se(le64toh(o->entry.seqnum) == 3);
67 [ - + ]: 4 : assert_se(sd_id128_equal(o->entry.boot_id, fake_boot_id));
68 : :
69 [ - + ]: 4 : assert_se(journal_file_next_entry(f, p, DIRECTION_DOWN, &o, &p) == 0);
70 : :
71 [ - + ]: 4 : assert_se(journal_file_next_entry(f, 0, DIRECTION_DOWN, &o, &p) == 1);
72 [ - + ]: 4 : assert_se(le64toh(o->entry.seqnum) == 1);
73 : :
74 [ - + ]: 4 : assert_se(journal_file_find_data_object(f, test, strlen(test), NULL, &p) == 1);
75 [ - + ]: 4 : assert_se(journal_file_next_entry_for_data(f, NULL, 0, p, DIRECTION_DOWN, &o, NULL) == 1);
76 [ - + ]: 4 : assert_se(le64toh(o->entry.seqnum) == 1);
77 : :
78 [ - + ]: 4 : assert_se(journal_file_next_entry_for_data(f, NULL, 0, p, DIRECTION_UP, &o, NULL) == 1);
79 [ - + ]: 4 : assert_se(le64toh(o->entry.seqnum) == 3);
80 : :
81 [ - + ]: 4 : assert_se(journal_file_find_data_object(f, test2, strlen(test2), NULL, &p) == 1);
82 [ - + ]: 4 : assert_se(journal_file_next_entry_for_data(f, NULL, 0, p, DIRECTION_UP, &o, NULL) == 1);
83 [ - + ]: 4 : assert_se(le64toh(o->entry.seqnum) == 2);
84 : :
85 [ - + ]: 4 : assert_se(journal_file_next_entry_for_data(f, NULL, 0, p, DIRECTION_DOWN, &o, NULL) == 1);
86 [ - + ]: 4 : assert_se(le64toh(o->entry.seqnum) == 2);
87 : :
88 [ - + ]: 4 : assert_se(journal_file_find_data_object(f, "quux", 4, NULL, &p) == 0);
89 : :
90 [ - + ]: 4 : assert_se(journal_file_move_to_entry_by_seqnum(f, 1, DIRECTION_DOWN, &o, NULL) == 1);
91 [ - + ]: 4 : assert_se(le64toh(o->entry.seqnum) == 1);
92 : :
93 [ - + ]: 4 : assert_se(journal_file_move_to_entry_by_seqnum(f, 3, DIRECTION_DOWN, &o, NULL) == 1);
94 [ - + ]: 4 : assert_se(le64toh(o->entry.seqnum) == 3);
95 : :
96 [ - + ]: 4 : assert_se(journal_file_move_to_entry_by_seqnum(f, 2, DIRECTION_DOWN, &o, NULL) == 1);
97 [ - + ]: 4 : assert_se(le64toh(o->entry.seqnum) == 2);
98 : :
99 [ - + ]: 4 : assert_se(journal_file_move_to_entry_by_seqnum(f, 10, DIRECTION_DOWN, &o, NULL) == 0);
100 : :
101 : 4 : journal_file_rotate(&f, true, (uint64_t) -1, true, NULL);
102 : 4 : journal_file_rotate(&f, true, (uint64_t) -1, true, NULL);
103 : :
104 : 4 : (void) journal_file_close(f);
105 : :
106 [ + - ]: 4 : log_info("Done...");
107 : :
108 [ - + ]: 4 : if (arg_keep)
109 [ # # ]: 0 : log_info("Not removing %s", t);
110 : : else {
111 : 4 : journal_directory_vacuum(".", 3000000, 0, 0, NULL, true);
112 : :
113 [ - + ]: 4 : assert_se(rm_rf(t, REMOVE_ROOT|REMOVE_PHYSICAL) >= 0);
114 : : }
115 : :
116 : 4 : puts("------------------------------------------------------------");
117 : 4 : }
118 : :
119 : 4 : static void test_empty(void) {
120 : : JournalFile *f1, *f2, *f3, *f4;
121 : 4 : char t[] = "/var/tmp/journal-XXXXXX";
122 : :
123 : 4 : test_setup_logging(LOG_DEBUG);
124 : :
125 : 4 : mkdtemp_chdir_chattr(t);
126 : :
127 [ - + ]: 4 : assert_se(journal_file_open(-1, "test.journal", O_RDWR|O_CREAT, 0666, false, (uint64_t) -1, false, NULL, NULL, NULL, NULL, &f1) == 0);
128 : :
129 [ - + ]: 4 : assert_se(journal_file_open(-1, "test-compress.journal", O_RDWR|O_CREAT, 0666, true, (uint64_t) -1, false, NULL, NULL, NULL, NULL, &f2) == 0);
130 : :
131 [ - + ]: 4 : assert_se(journal_file_open(-1, "test-seal.journal", O_RDWR|O_CREAT, 0666, false, (uint64_t) -1, true, NULL, NULL, NULL, NULL, &f3) == 0);
132 : :
133 [ - + ]: 4 : assert_se(journal_file_open(-1, "test-seal-compress.journal", O_RDWR|O_CREAT, 0666, true, (uint64_t) -1, true, NULL, NULL, NULL, NULL, &f4) == 0);
134 : :
135 : 4 : journal_file_print_header(f1);
136 : 4 : puts("");
137 : 4 : journal_file_print_header(f2);
138 : 4 : puts("");
139 : 4 : journal_file_print_header(f3);
140 : 4 : puts("");
141 : 4 : journal_file_print_header(f4);
142 : 4 : puts("");
143 : :
144 [ + - ]: 4 : log_info("Done...");
145 : :
146 [ - + ]: 4 : if (arg_keep)
147 [ # # ]: 0 : log_info("Not removing %s", t);
148 : : else {
149 : 4 : journal_directory_vacuum(".", 3000000, 0, 0, NULL, true);
150 : :
151 [ - + ]: 4 : assert_se(rm_rf(t, REMOVE_ROOT|REMOVE_PHYSICAL) >= 0);
152 : : }
153 : :
154 : 4 : (void) journal_file_close(f1);
155 : 4 : (void) journal_file_close(f2);
156 : 4 : (void) journal_file_close(f3);
157 : 4 : (void) journal_file_close(f4);
158 : 4 : }
159 : :
160 : : #if HAVE_XZ || HAVE_LZ4
161 : 28 : static bool check_compressed(uint64_t compress_threshold, uint64_t data_size) {
162 : : dual_timestamp ts;
163 : : JournalFile *f;
164 : : struct iovec iovec;
165 : : Object *o;
166 : : uint64_t p;
167 : 28 : char t[] = "/var/tmp/journal-XXXXXX";
168 : 28 : char data[2048] = {0};
169 : : bool is_compressed;
170 : : int r;
171 : :
172 [ - + ]: 28 : assert_se(data_size <= sizeof(data));
173 : :
174 : 28 : test_setup_logging(LOG_DEBUG);
175 : :
176 : 28 : mkdtemp_chdir_chattr(t);
177 : :
178 [ - + ]: 28 : assert_se(journal_file_open(-1, "test.journal", O_RDWR|O_CREAT, 0666, true, compress_threshold, true, NULL, NULL, NULL, NULL, &f) == 0);
179 : :
180 : 28 : dual_timestamp_get(&ts);
181 : :
182 : 28 : iovec = IOVEC_MAKE(data, data_size);
183 [ - + ]: 28 : assert_se(journal_file_append_entry(f, &ts, NULL, &iovec, 1, NULL, NULL, NULL) == 0);
184 : :
185 : : #if HAVE_GCRYPT
186 : 28 : journal_file_append_tag(f);
187 : : #endif
188 : 28 : journal_file_dump(f);
189 : :
190 : : /* We have to partially reimplement some of the dump logic, because the normal next_entry does the
191 : : * decompression for us. */
192 : 28 : p = le64toh(f->header->header_size);
193 : : for (;;) {
194 : 84 : r = journal_file_move_to_object(f, OBJECT_UNUSED, p, &o);
195 [ - + ]: 84 : assert_se(r == 0);
196 [ + + ]: 84 : if (o->object.type == OBJECT_DATA)
197 : 28 : break;
198 : :
199 [ - + ]: 56 : assert_se(p < le64toh(f->header->tail_object_offset));
200 : 56 : p = p + ALIGN64(le64toh(o->object.size));
201 : : }
202 : :
203 : 28 : is_compressed = (o->object.flags & OBJECT_COMPRESSION_MASK) != 0;
204 : :
205 : 28 : (void) journal_file_close(f);
206 : :
207 [ + - ]: 28 : log_info("Done...");
208 : :
209 [ - + ]: 28 : if (arg_keep)
210 [ # # ]: 0 : log_info("Not removing %s", t);
211 : : else {
212 : 28 : journal_directory_vacuum(".", 3000000, 0, 0, NULL, true);
213 : :
214 [ - + ]: 28 : assert_se(rm_rf(t, REMOVE_ROOT|REMOVE_PHYSICAL) >= 0);
215 : : }
216 : :
217 : 28 : puts("------------------------------------------------------------");
218 : :
219 : 28 : return is_compressed;
220 : : }
221 : :
222 : 4 : static void test_min_compress_size(void) {
223 : : /* Note that XZ will actually fail to compress anything under 80 bytes, so you have to choose the limits
224 : : * carefully */
225 : :
226 : : /* DEFAULT_MIN_COMPRESS_SIZE is 512 */
227 [ - + ]: 4 : assert_se(!check_compressed((uint64_t) -1, 255));
228 [ - + ]: 4 : assert_se(check_compressed((uint64_t) -1, 513));
229 : :
230 : : /* compress everything */
231 [ - + ]: 4 : assert_se(check_compressed(0, 96));
232 [ - + ]: 4 : assert_se(check_compressed(8, 96));
233 : :
234 : : /* Ensure we don't try to compress less than 8 bytes */
235 [ - + ]: 4 : assert_se(!check_compressed(0, 7));
236 : :
237 : : /* check boundary conditions */
238 [ - + ]: 4 : assert_se(check_compressed(256, 256));
239 [ - + ]: 4 : assert_se(!check_compressed(256, 255));
240 : 4 : }
241 : : #endif
242 : :
243 : 4 : int main(int argc, char *argv[]) {
244 : 4 : arg_keep = argc > 1;
245 : :
246 : 4 : test_setup_logging(LOG_INFO);
247 : :
248 : : /* journal_file_open requires a valid machine id */
249 [ - + ]: 4 : if (access("/etc/machine-id", F_OK) != 0)
250 : 0 : return log_tests_skipped("/etc/machine-id not found");
251 : :
252 : 4 : test_non_empty();
253 : 4 : test_empty();
254 : : #if HAVE_XZ || HAVE_LZ4
255 : 4 : test_min_compress_size();
256 : : #endif
257 : :
258 : 4 : return 0;
259 : : }
|
