src/journal/test-catalog.c

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