src/journal/test-catalog.c

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