src/test/test-proc-cmdline.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2 /***
   3   This file is part of systemd.
   4
   5   Copyright 2010 Lennart Poettering
   6 ***/
   7
   8 #include "alloc-util.h"
   9 #include "log.h"
  10 #include "macro.h"
  11 #include "proc-cmdline.h"
  12 //#include "special.h"
  13 #include "string-util.h"
  14 #include "util.h"
  15
  16 static int obj;
  17
  18 static int parse_item(const char *key, const char *value, void *data) {
  19         assert_se(key);
  20         assert_se(data == &obj);
  21
  22         log_info("kernel cmdline option <%s> = <%s>", key, strna(value));
  23         return 0;
  24 }
  25
  26 static void test_proc_cmdline_parse(void) {
  27         assert_se(proc_cmdline_parse(parse_item, &obj, true) >= 0);
  28 }
  29
  30 #if 0 /// UNNEEDED by elogind
  31 static void test_runlevel_to_target(void) {
  32         in_initrd_force(false);
  33         assert_se(streq_ptr(runlevel_to_target(NULL), NULL));
  34         assert_se(streq_ptr(runlevel_to_target("unknown-runlevel"), NULL));
  35         assert_se(streq_ptr(runlevel_to_target("rd.unknown-runlevel"), NULL));
  36         assert_se(streq_ptr(runlevel_to_target("3"), SPECIAL_MULTI_USER_TARGET));
  37         assert_se(streq_ptr(runlevel_to_target("rd.rescue"), NULL));
  38
  39         in_initrd_force(true);
  40         assert_se(streq_ptr(runlevel_to_target(NULL), NULL));
  41         assert_se(streq_ptr(runlevel_to_target("unknown-runlevel"), NULL));
  42         assert_se(streq_ptr(runlevel_to_target("rd.unknown-runlevel"), NULL));
  43         assert_se(streq_ptr(runlevel_to_target("3"), NULL));
  44         assert_se(streq_ptr(runlevel_to_target("rd.rescue"), SPECIAL_RESCUE_TARGET));
  45 }
  46
  47 static void test_proc_cmdline_get_key(void) {
  48         _cleanup_free_ char *value = NULL;
  49
  50         putenv((char*) "SYSTEMD_PROC_CMDLINE=foo_bar=quux wuff-piep=tuet zumm");
  51
  52         assert_se(proc_cmdline_get_key("", 0, &value) == -EINVAL);
  53         assert_se(proc_cmdline_get_key("abc", 0, NULL) == 0);
  54         assert_se(proc_cmdline_get_key("abc", 0, &value) == 0 && value == NULL);
  55         assert_se(proc_cmdline_get_key("abc", PROC_CMDLINE_VALUE_OPTIONAL, &value) == 0 && value == NULL);
  56
  57         assert_se(proc_cmdline_get_key("foo_bar", 0, &value) > 0 && streq_ptr(value, "quux"));
  58         value = mfree(value);
  59         assert_se(proc_cmdline_get_key("foo_bar", PROC_CMDLINE_VALUE_OPTIONAL, &value) > 0 && streq_ptr(value, "quux"));
  60         value = mfree(value);
  61         assert_se(proc_cmdline_get_key("foo-bar", 0, &value) > 0 && streq_ptr(value, "quux"));
  62         value = mfree(value);
  63         assert_se(proc_cmdline_get_key("foo-bar", PROC_CMDLINE_VALUE_OPTIONAL, &value) > 0 && streq_ptr(value, "quux"));
  64         value = mfree(value);
  65         assert_se(proc_cmdline_get_key("foo-bar", 0, NULL) == 0);
  66         assert_se(proc_cmdline_get_key("foo-bar", PROC_CMDLINE_VALUE_OPTIONAL, NULL) == -EINVAL);
  67
  68         assert_se(proc_cmdline_get_key("wuff-piep", 0, &value) > 0 && streq_ptr(value, "tuet"));
  69         value = mfree(value);
  70         assert_se(proc_cmdline_get_key("wuff-piep", PROC_CMDLINE_VALUE_OPTIONAL, &value) > 0 && streq_ptr(value, "tuet"));
  71         value = mfree(value);
  72         assert_se(proc_cmdline_get_key("wuff_piep", 0, &value) > 0 && streq_ptr(value, "tuet"));
  73         value = mfree(value);
  74         assert_se(proc_cmdline_get_key("wuff_piep", PROC_CMDLINE_VALUE_OPTIONAL, &value) > 0 && streq_ptr(value, "tuet"));
  75         value = mfree(value);
  76         assert_se(proc_cmdline_get_key("wuff_piep", 0, NULL) == 0);
  77         assert_se(proc_cmdline_get_key("wuff_piep", PROC_CMDLINE_VALUE_OPTIONAL, NULL) == -EINVAL);
  78
  79         assert_se(proc_cmdline_get_key("zumm", 0, &value) == 0 && value == NULL);
  80         assert_se(proc_cmdline_get_key("zumm", PROC_CMDLINE_VALUE_OPTIONAL, &value) > 0 && value == NULL);
  81         assert_se(proc_cmdline_get_key("zumm", 0, NULL) > 0);
  82 }
  83
  84 static void test_proc_cmdline_get_bool(void) {
  85         bool value = false;
  86
  87         putenv((char*) "SYSTEMD_PROC_CMDLINE=foo_bar bar-waldo=1 x_y-z=0 quux=miep");
  88
  89         assert_se(proc_cmdline_get_bool("", &value) == -EINVAL);
  90         assert_se(proc_cmdline_get_bool("abc", &value) == 0 && value == false);
  91         assert_se(proc_cmdline_get_bool("foo_bar", &value) > 0 && value == true);
  92         assert_se(proc_cmdline_get_bool("foo-bar", &value) > 0 && value == true);
  93         assert_se(proc_cmdline_get_bool("bar-waldo", &value) > 0 && value == true);
  94         assert_se(proc_cmdline_get_bool("bar_waldo", &value) > 0 && value == true);
  95         assert_se(proc_cmdline_get_bool("x_y-z", &value) > 0 && value == false);
  96         assert_se(proc_cmdline_get_bool("x-y-z", &value) > 0 && value == false);
  97         assert_se(proc_cmdline_get_bool("x-y_z", &value) > 0 && value == false);
  98         assert_se(proc_cmdline_get_bool("x_y_z", &value) > 0 && value == false);
  99         assert_se(proc_cmdline_get_bool("quux", &value) == -EINVAL && value == false);
 100 }
 101 #endif // 0
 102
 103 static void test_proc_cmdline_key_streq(void) {
 104
 105         assert_se(proc_cmdline_key_streq("", ""));
 106         assert_se(proc_cmdline_key_streq("a", "a"));
 107         assert_se(!proc_cmdline_key_streq("", "a"));
 108         assert_se(!proc_cmdline_key_streq("a", ""));
 109         assert_se(proc_cmdline_key_streq("a", "a"));
 110         assert_se(!proc_cmdline_key_streq("a", "b"));
 111         assert_se(proc_cmdline_key_streq("x-y-z", "x-y-z"));
 112         assert_se(proc_cmdline_key_streq("x-y-z", "x_y_z"));
 113         assert_se(proc_cmdline_key_streq("x-y-z", "x-y_z"));
 114         assert_se(proc_cmdline_key_streq("x-y-z", "x_y-z"));
 115         assert_se(proc_cmdline_key_streq("x_y-z", "x-y_z"));
 116         assert_se(!proc_cmdline_key_streq("x_y-z", "x-z_z"));
 117 }
 118
 119 #if 0 /// UNNEEDED by elogind
 120 static void test_proc_cmdline_key_startswith(void) {
 121
 122         assert_se(proc_cmdline_key_startswith("", ""));
 123         assert_se(proc_cmdline_key_startswith("x", ""));
 124         assert_se(!proc_cmdline_key_startswith("", "x"));
 125         assert_se(proc_cmdline_key_startswith("x", "x"));
 126         assert_se(!proc_cmdline_key_startswith("x", "y"));
 127         assert_se(!proc_cmdline_key_startswith("foo-bar", "quux"));
 128         assert_se(proc_cmdline_key_startswith("foo-bar", "foo"));
 129         assert_se(proc_cmdline_key_startswith("foo-bar", "foo-bar"));
 130         assert_se(proc_cmdline_key_startswith("foo-bar", "foo_bar"));
 131         assert_se(proc_cmdline_key_startswith("foo-bar", "foo_"));
 132         assert_se(!proc_cmdline_key_startswith("foo-bar", "foo_xx"));
 133 }
 134 #endif // 0
 135
 136 int main(void) {
 137         log_parse_environment();
 138         log_open();
 139
 140         test_proc_cmdline_parse();
 141         test_proc_cmdline_key_streq();
 142 #if 0 /// UNNEEDED by elogind
 143         test_proc_cmdline_key_startswith();
 144         test_proc_cmdline_get_key();
 145         test_proc_cmdline_get_bool();
 146         test_runlevel_to_target();
 147 #endif // 0
 148
 149         return 0;
 150 }