src/journal/test-catalog.c

   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 static OrderedHashmap* test_import(const char* contents, ssize_t size, int code) {
  30         _cleanup_(unlink_tempfilep) char name[] = "/tmp/test-catalog.XXXXXX";
  31         _cleanup_close_ int fd;
  32         OrderedHashmap *h;
  33
  34         if (size < 0)
  35                 size = strlen(contents);
  36
  37         assert_se(h = ordered_hashmap_new(&catalog_hash_ops));
  38
  39         fd = mkostemp_safe(name);
  40         assert_se(fd >= 0);
  41         assert_se(write(fd, contents, size) == size);
  42
  43         assert_se(catalog_import_file(h, name) == code);
  44
  45         return h;
  46 }
  47
  48 static void test_catalog_import_invalid(void) {
  49         _cleanup_ordered_hashmap_free_free_free_ OrderedHashmap *h = NULL;
  50
  51         h = test_import("xxx", -1, -EINVAL);
  52         assert_se(ordered_hashmap_isempty(h));
  53 }
  54
  55 static void test_catalog_import_badid(void) {
  56         _cleanup_ordered_hashmap_free_free_free_ OrderedHashmap *h = NULL;
  57         const char *input =
  58 "-- 0027229ca0644181a76c4e92458afaff dededededededededededededededede\n" \
  59 "Subject: message\n" \
  60 "\n" \
  61 "payload\n";
  62         h = test_import(input, -1, -EINVAL);
  63 }
  64
  65 static void test_catalog_import_one(void) {
  66         _cleanup_ordered_hashmap_free_free_free_ OrderedHashmap *h = NULL;
  67         char *payload;
  68         Iterator j;
  69
  70         const char *input =
  71 "-- 0027229ca0644181a76c4e92458afaff dededededededededededededededed\n" \
  72 "Subject: message\n" \
  73 "\n" \
  74 "payload\n";
  75         const char *expect =
  76 "Subject: message\n" \
  77 "\n" \
  78 "payload\n";
  79
  80         h = test_import(input, -1, 0);
  81         assert_se(ordered_hashmap_size(h) == 1);
  82
  83         ORDERED_HASHMAP_FOREACH(payload, h, j) {
  84                 printf("expect: %s\n", expect);
  85                 printf("actual: %s\n", payload);
  86                 assert_se(streq(expect, payload));
  87         }
  88 }
  89
  90 static void test_catalog_import_merge(void) {
  91         _cleanup_ordered_hashmap_free_free_free_ OrderedHashmap *h = NULL;
  92         char *payload;
  93         Iterator j;
  94
  95         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         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         h = test_import(input, -1, 0);
 117         assert_se(ordered_hashmap_size(h) == 1);
 118
 119         ORDERED_HASHMAP_FOREACH(payload, h, j)
 120                 assert_se(streq(combined, payload));
 121 }
 122
 123 static void test_catalog_import_merge_no_body(void) {
 124         _cleanup_ordered_hashmap_free_free_free_ OrderedHashmap *h = NULL;
 125         char *payload;
 126         Iterator j;
 127
 128         const char *input =
 129 "-- 0027229ca0644181a76c4e92458afaff dededededededededededededededed\n" \
 130 "Subject: message\n" \
 131 "Defined-By: me\n" \
 132 "\n" \
 133 "payload\n" \
 134 "\n" \
 135 "-- 0027229ca0644181a76c4e92458afaff dededededededededededededededed\n" \
 136 "Subject: override subject\n" \
 137 "X-Header: hello\n" \
 138 "\n";
 139
 140         const char *combined =
 141 "Subject: override subject\n" \
 142 "X-Header: hello\n" \
 143 "Subject: message\n" \
 144 "Defined-By: me\n" \
 145 "\n" \
 146 "payload\n";
 147
 148         h = test_import(input, -1, 0);
 149         assert_se(ordered_hashmap_size(h) == 1);
 150
 151         ORDERED_HASHMAP_FOREACH(payload, h, j)
 152                 assert_se(streq(combined, payload));
 153 }
 154
 155 static void test_catalog_update(const char *database) {
 156         int r;
 157
 158         /* Test what happens if there are no files. */
 159         r = catalog_update(database, NULL, NULL);
 160         assert_se(r == 0);
 161
 162         /* Test what happens if there are no files in the directory. */
 163         r = catalog_update(database, NULL, no_catalog_dirs);
 164         assert_se(r == 0);
 165
 166         /* Make sure that we at least have some files loaded or the
 167          * catalog_list below will fail. */
 168         r = catalog_update(database, NULL, (const char * const *) catalog_dirs);
 169         assert_se(r == 0);
 170 }
 171
 172 static void test_catalog_file_lang(void) {
 173         _cleanup_free_ char *lang = NULL, *lang2 = NULL, *lang3 = NULL, *lang4 = NULL;
 174
 175         assert_se(catalog_file_lang("systemd.de_DE.catalog", &lang) == 1);
 176         assert_se(streq(lang, "de_DE"));
 177
 178         assert_se(catalog_file_lang("systemd..catalog", &lang2) == 0);
 179         assert_se(lang2 == NULL);
 180
 181         assert_se(catalog_file_lang("systemd.fr.catalog", &lang2) == 1);
 182         assert_se(streq(lang2, "fr"));
 183
 184         assert_se(catalog_file_lang("systemd.fr.catalog.gz", &lang3) == 0);
 185         assert_se(lang3 == NULL);
 186
 187         assert_se(catalog_file_lang("systemd.01234567890123456789012345678901.catalog", &lang3) == 0);
 188         assert_se(lang3 == NULL);
 189
 190         assert_se(catalog_file_lang("systemd.0123456789012345678901234567890.catalog", &lang3) == 1);
 191         assert_se(streq(lang3, "0123456789012345678901234567890"));
 192
 193         assert_se(catalog_file_lang("/x/y/systemd.catalog", &lang4) == 0);
 194         assert_se(lang4 == NULL);
 195
 196         assert_se(catalog_file_lang("/x/y/systemd.ru_RU.catalog", &lang4) == 1);
 197         assert_se(streq(lang4, "ru_RU"));
 198 }
 199
 200 int main(int argc, char *argv[]) {
 201         _cleanup_(unlink_tempfilep) char database[] = "/tmp/test-catalog.XXXXXX";
 202         _cleanup_free_ char *text = NULL;
 203         int r;
 204
 205         setlocale(LC_ALL, "de_DE.UTF-8");
 206
 207         test_setup_logging(LOG_DEBUG);
 208
 209         /* If test-catalog is located at the build directory, then use catalogs in that.
 210          * If it is not, e.g. installed by systemd-tests package, then use installed catalogs. */
 211         catalog_dirs = STRV_MAKE(get_catalog_dir());
 212
 213         assert_se(access(catalog_dirs[0], F_OK) >= 0);
 214         log_notice("Using catalog directory '%s'", catalog_dirs[0]);
 215
 216         test_catalog_file_lang();
 217
 218         test_catalog_import_invalid();
 219         test_catalog_import_badid();
 220         test_catalog_import_one();
 221         test_catalog_import_merge();
 222         test_catalog_import_merge_no_body();
 223
 224         assert_se(mkostemp_safe(database) >= 0);
 225
 226         test_catalog_update(database);
 227
 228         r = catalog_list(stdout, database, true);
 229         assert_se(r >= 0);
 230
 231         r = catalog_list(stdout, database, false);
 232         assert_se(r >= 0);
 233
 234         assert_se(catalog_get(database, SD_MESSAGE_COREDUMP, &text) >= 0);
 235         printf(">>>%s<<<\n", text);
 236
 237         return 0;
 238 }