src/journal/test-journal.c

   1 /***
   2   This file is part of systemd.
   3
   4   Copyright 2011 Lennart Poettering
   5
   6   systemd is free software; you can redistribute it and/or modify it
   7   under the terms of the GNU Lesser General Public License as published by
   8   the Free Software Foundation; either version 2.1 of the License, or
   9   (at your option) any later version.
  10
  11   systemd is distributed in the hope that it will be useful, but
  12   WITHOUT ANY WARRANTY; without even the implied warranty of
  13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14   Lesser General Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public License
  17   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  18 ***/
  19
  20 #include <fcntl.h>
  21 #include <unistd.h>
  22
  23 #include "journal-authenticate.h"
  24 #include "journal-file.h"
  25 #include "journal-vacuum.h"
  26 #include "log.h"
  27 #include "rm-rf.h"
  28
  29 static bool arg_keep = false;
  30
  31 static void test_non_empty(void) {
  32         dual_timestamp ts;
  33         JournalFile *f;
  34         struct iovec iovec;
  35         static const char test[] = "TEST1=1", test2[] = "TEST2=2";
  36         Object *o;
  37         uint64_t p;
  38         char t[] = "/tmp/journal-XXXXXX";
  39
  40         log_set_max_level(LOG_DEBUG);
  41
  42         assert_se(mkdtemp(t));
  43         assert_se(chdir(t) >= 0);
  44
  45         assert_se(journal_file_open(-1, "test.journal", O_RDWR|O_CREAT, 0666, true, true, NULL, NULL, NULL, NULL, &f) == 0);
  46
  47         dual_timestamp_get(&ts);
  48
  49         iovec.iov_base = (void*) test;
  50         iovec.iov_len = strlen(test);
  51         assert_se(journal_file_append_entry(f, &ts, &iovec, 1, NULL, NULL, NULL) == 0);
  52
  53         iovec.iov_base = (void*) test2;
  54         iovec.iov_len = strlen(test2);
  55         assert_se(journal_file_append_entry(f, &ts, &iovec, 1, NULL, NULL, NULL) == 0);
  56
  57         iovec.iov_base = (void*) test;
  58         iovec.iov_len = strlen(test);
  59         assert_se(journal_file_append_entry(f, &ts, &iovec, 1, NULL, NULL, NULL) == 0);
  60
  61 #ifdef HAVE_GCRYPT
  62         journal_file_append_tag(f);
  63 #endif
  64         journal_file_dump(f);
  65
  66         assert_se(journal_file_next_entry(f, 0, DIRECTION_DOWN, &o, &p) == 1);
  67         assert_se(le64toh(o->entry.seqnum) == 1);
  68
  69         assert_se(journal_file_next_entry(f, p, DIRECTION_DOWN, &o, &p) == 1);
  70         assert_se(le64toh(o->entry.seqnum) == 2);
  71
  72         assert_se(journal_file_next_entry(f, p, DIRECTION_DOWN, &o, &p) == 1);
  73         assert_se(le64toh(o->entry.seqnum) == 3);
  74
  75         assert_se(journal_file_next_entry(f, p, DIRECTION_DOWN, &o, &p) == 0);
  76
  77         assert_se(journal_file_next_entry(f, 0, DIRECTION_DOWN, &o, &p) == 1);
  78         assert_se(le64toh(o->entry.seqnum) == 1);
  79
  80         assert_se(journal_file_find_data_object(f, test, strlen(test), NULL, &p) == 1);
  81         assert_se(journal_file_next_entry_for_data(f, NULL, 0, p, DIRECTION_DOWN, &o, NULL) == 1);
  82         assert_se(le64toh(o->entry.seqnum) == 1);
  83
  84         assert_se(journal_file_next_entry_for_data(f, NULL, 0, p, DIRECTION_UP, &o, NULL) == 1);
  85         assert_se(le64toh(o->entry.seqnum) == 3);
  86
  87         assert_se(journal_file_find_data_object(f, test2, strlen(test2), NULL, &p) == 1);
  88         assert_se(journal_file_next_entry_for_data(f, NULL, 0, p, DIRECTION_UP, &o, NULL) == 1);
  89         assert_se(le64toh(o->entry.seqnum) == 2);
  90
  91         assert_se(journal_file_next_entry_for_data(f, NULL, 0, p, DIRECTION_DOWN, &o, NULL) == 1);
  92         assert_se(le64toh(o->entry.seqnum) == 2);
  93
  94         assert_se(journal_file_find_data_object(f, "quux", 4, NULL, &p) == 0);
  95
  96         assert_se(journal_file_move_to_entry_by_seqnum(f, 1, DIRECTION_DOWN, &o, NULL) == 1);
  97         assert_se(le64toh(o->entry.seqnum) == 1);
  98
  99         assert_se(journal_file_move_to_entry_by_seqnum(f, 3, DIRECTION_DOWN, &o, NULL) == 1);
 100         assert_se(le64toh(o->entry.seqnum) == 3);
 101
 102         assert_se(journal_file_move_to_entry_by_seqnum(f, 2, DIRECTION_DOWN, &o, NULL) == 1);
 103         assert_se(le64toh(o->entry.seqnum) == 2);
 104
 105         assert_se(journal_file_move_to_entry_by_seqnum(f, 10, DIRECTION_DOWN, &o, NULL) == 0);
 106
 107         journal_file_rotate(&f, true, true, NULL);
 108         journal_file_rotate(&f, true, true, NULL);
 109
 110         (void) journal_file_close(f);
 111
 112         log_info("Done...");
 113
 114         if (arg_keep)
 115                 log_info("Not removing %s", t);
 116         else {
 117                 journal_directory_vacuum(".", 3000000, 0, 0, NULL, true);
 118
 119                 assert_se(rm_rf(t, REMOVE_ROOT|REMOVE_PHYSICAL) >= 0);
 120         }
 121
 122         puts("------------------------------------------------------------");
 123 }
 124
 125 static void test_empty(void) {
 126         JournalFile *f1, *f2, *f3, *f4;
 127         char t[] = "/tmp/journal-XXXXXX";
 128
 129         log_set_max_level(LOG_DEBUG);
 130
 131         assert_se(mkdtemp(t));
 132         assert_se(chdir(t) >= 0);
 133
 134         assert_se(journal_file_open(-1, "test.journal", O_RDWR|O_CREAT, 0666, false, false, NULL, NULL, NULL, NULL, &f1) == 0);
 135
 136         assert_se(journal_file_open(-1, "test-compress.journal", O_RDWR|O_CREAT, 0666, true, false, NULL, NULL, NULL, NULL, &f2) == 0);
 137
 138         assert_se(journal_file_open(-1, "test-seal.journal", O_RDWR|O_CREAT, 0666, false, true, NULL, NULL, NULL, NULL, &f3) == 0);
 139
 140         assert_se(journal_file_open(-1, "test-seal-compress.journal", O_RDWR|O_CREAT, 0666, true, true, NULL, NULL, NULL, NULL, &f4) == 0);
 141
 142         journal_file_print_header(f1);
 143         puts("");
 144         journal_file_print_header(f2);
 145         puts("");
 146         journal_file_print_header(f3);
 147         puts("");
 148         journal_file_print_header(f4);
 149         puts("");
 150
 151         log_info("Done...");
 152
 153         if (arg_keep)
 154                 log_info("Not removing %s", t);
 155         else {
 156                 journal_directory_vacuum(".", 3000000, 0, 0, NULL, true);
 157
 158                 assert_se(rm_rf(t, REMOVE_ROOT|REMOVE_PHYSICAL) >= 0);
 159         }
 160
 161         (void) journal_file_close(f1);
 162         (void) journal_file_close(f2);
 163         (void) journal_file_close(f3);
 164         (void) journal_file_close(f4);
 165 }
 166
 167 int main(int argc, char *argv[]) {
 168         arg_keep = argc > 1;
 169
 170         /* journal_file_open requires a valid machine id */
 171         if (access("/etc/machine-id", F_OK) != 0)
 172                 return EXIT_TEST_SKIP;
 173
 174         test_non_empty();
 175         test_empty();
 176
 177         return 0;
 178 }