src/journal/test-catalog.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2012 Lennart Poettering
   7   Copyright 2013 Zbigniew Jędrzejewski-Szmek
   8
   9   systemd is free software; you can redistribute it and/or modify it
  10   under the terms of the GNU Lesser General Public License as published by
  11   the Free Software Foundation; either version 2.1 of the License, or
  12   (at your option) any later version.
  13
  14   systemd is distributed in the hope that it will be useful, but
  15   WITHOUT ANY WARRANTY; without even the implied warranty of
  16   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  17   Lesser General Public License for more details.
  18
  19   You should have received a copy of the GNU Lesser General Public License
  20   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  21 ***/
  22
  23 #include <errno.h>
  24 #include <fcntl.h>
  25 #include <locale.h>
  26 #include <unistd.h>
  27
  28 #include "sd-messages.h"
  29
  30 #include "alloc-util.h"
  31 #include "catalog.h"
  32 #include "fd-util.h"
  33 #include "fileio.h"
  34 #include "log.h"
  35 #include "macro.h"
  36 #include "string-util.h"
  37 #include "util.h"
  38
  39 static const char *catalog_dirs[] = {
  40         CATALOG_DIR,
  41         NULL,
  42 };
  43
  44 static const char *no_catalog_dirs[] = {
  45         "/bin/hopefully/with/no/catalog",
  46         NULL
  47 };
  48
  49 static void test_import(Hashmap *h, const char* contents, ssize_t size, int code) {
  50         int r;
  51         char name[] = "/tmp/test-catalog.XXXXXX";
  52         _cleanup_close_ int fd;
  53
  54         fd = mkostemp_safe(name, O_RDWR|O_CLOEXEC);
  55         assert_se(fd >= 0);
  56         assert_se(write(fd, contents, size) == size);
  57
  58         r = catalog_import_file(h, name);
  59         assert_se(r == code);
  60
  61         unlink(name);
  62 }
  63
  64 static void test_catalog_importing(void) {
  65         Hashmap *h;
  66
  67         assert_se(h = hashmap_new(&catalog_hash_ops));
  68
  69 #define BUF "xxx"
  70         test_import(h, BUF, sizeof(BUF), -EINVAL);
  71 #undef BUF
  72         assert_se(hashmap_isempty(h));
  73         log_debug("----------------------------------------");
  74
  75 #define BUF \
  76 "-- 0027229ca0644181a76c4e92458afaff dededededededededededededededede\n" \
  77 "Subject: message\n" \
  78 "\n" \
  79 "payload\n"
  80         test_import(h, BUF, sizeof(BUF), -EINVAL);
  81 #undef BUF
  82
  83         log_debug("----------------------------------------");
  84
  85 #define BUF \
  86 "-- 0027229ca0644181a76c4e92458afaff dededededededededededededededed\n" \
  87 "Subject: message\n" \
  88 "\n" \
  89 "payload\n"
  90         test_import(h, BUF, sizeof(BUF), 0);
  91 #undef BUF
  92
  93         assert_se(hashmap_size(h) == 1);
  94
  95         log_debug("----------------------------------------");
  96
  97         hashmap_free_free(h);
  98 }
  99
 100
 101 static const char* database = NULL;
 102
 103 static void test_catalog_update(void) {
 104         static char name[] = "/tmp/test-catalog.XXXXXX";
 105         int r;
 106
 107         r = mkostemp_safe(name, O_RDWR|O_CLOEXEC);
 108         assert_se(r >= 0);
 109
 110         database = name;
 111
 112         /* Test what happens if there are no files. */
 113         r = catalog_update(database, NULL, NULL);
 114         assert_se(r >= 0);
 115
 116         /* Test what happens if there are no files in the directory. */
 117         r = catalog_update(database, NULL, no_catalog_dirs);
 118         assert_se(r >= 0);
 119
 120         /* Make sure that we at least have some files loaded or the
 121            catalog_list below will fail. */
 122         r = catalog_update(database, NULL, catalog_dirs);
 123         assert_se(r >= 0);
 124 }
 125
 126 static void test_catalog_file_lang(void) {
 127         _cleanup_free_ char *lang = NULL, *lang2 = NULL, *lang3 = NULL, *lang4 = NULL;
 128
 129         assert_se(catalog_file_lang("systemd.de_DE.catalog", &lang) == 1);
 130         assert_se(streq(lang, "de_DE"));
 131
 132         assert_se(catalog_file_lang("systemd..catalog", &lang2) == 0);
 133         assert_se(lang2 == NULL);
 134
 135         assert_se(catalog_file_lang("systemd.fr.catalog", &lang2) == 1);
 136         assert_se(streq(lang2, "fr"));
 137
 138         assert_se(catalog_file_lang("systemd.fr.catalog.gz", &lang3) == 0);
 139         assert_se(lang3 == NULL);
 140
 141         assert_se(catalog_file_lang("systemd.01234567890123456789012345678901.catalog", &lang3) == 0);
 142         assert_se(lang3 == NULL);
 143
 144         assert_se(catalog_file_lang("systemd.0123456789012345678901234567890.catalog", &lang3) == 1);
 145         assert_se(streq(lang3, "0123456789012345678901234567890"));
 146
 147         assert_se(catalog_file_lang("/x/y/systemd.catalog", &lang4) == 0);
 148         assert_se(lang4 == NULL);
 149
 150         assert_se(catalog_file_lang("/x/y/systemd.ru_RU.catalog", &lang4) == 1);
 151         assert_se(streq(lang4, "ru_RU"));
 152 }
 153
 154 int main(int argc, char *argv[]) {
 155         _cleanup_free_ char *text = NULL;
 156         int r;
 157
 158         setlocale(LC_ALL, "de_DE.UTF-8");
 159
 160         log_parse_environment();
 161         log_open();
 162
 163         test_catalog_file_lang();
 164
 165         test_catalog_importing();
 166
 167         test_catalog_update();
 168
 169         r = catalog_list(stdout, database, true);
 170         assert_se(r >= 0);
 171
 172         r = catalog_list(stdout, database, false);
 173         assert_se(r >= 0);
 174
 175         assert_se(catalog_get(database, SD_MESSAGE_COREDUMP, &text) >= 0);
 176         printf(">>>%s<<<\n", text);
 177
 178         if (database)
 179                 unlink(database);
 180
 181         return 0;
 182 }