src/test/test-env-util.c

   1 /***
   2   This file is part of systemd.
   3
   4   Copyright 2010 Lennart Poettering
   5   Copyright 2016 Zbigniew Jędrzejewski-Szmek
   6
   7   systemd is free software; you can redistribute it and/or modify it
   8   under the terms of the GNU Lesser General Public License as published by
   9   the Free Software Foundation; either version 2.1 of the License, or
  10   (at your option) any later version.
  11
  12   systemd is distributed in the hope that it will be useful, but
  13   WITHOUT ANY WARRANTY; without even the implied warranty of
  14   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15   Lesser General Public License for more details.
  16
  17   You should have received a copy of the GNU Lesser General Public License
  18   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  19 ***/
  20
  21 #include <string.h>
  22
  23 #include "env-util.h"
  24 #include "string-util.h"
  25 #include "strv.h"
  26 #include "util.h"
  27
  28 static void test_strv_env_delete(void) {
  29         _cleanup_strv_free_ char **a = NULL, **b = NULL, **c = NULL, **d = NULL;
  30
  31         a = strv_new("FOO=BAR", "WALDO=WALDO", "WALDO=", "PIEP", "SCHLUMPF=SMURF", NULL);
  32         assert_se(a);
  33
  34         b = strv_new("PIEP", "FOO", NULL);
  35         assert_se(b);
  36
  37         c = strv_new("SCHLUMPF", NULL);
  38         assert_se(c);
  39
  40         d = strv_env_delete(a, 2, b, c);
  41         assert_se(d);
  42
  43         assert_se(streq(d[0], "WALDO=WALDO"));
  44         assert_se(streq(d[1], "WALDO="));
  45         assert_se(strv_length(d) == 2);
  46 }
  47
  48 static void test_strv_env_unset(void) {
  49         _cleanup_strv_free_ char **l = NULL;
  50
  51         l = strv_new("PIEP", "SCHLUMPF=SMURFF", "NANANANA=YES", NULL);
  52         assert_se(l);
  53
  54         assert_se(strv_env_unset(l, "SCHLUMPF") == l);
  55
  56         assert_se(streq(l[0], "PIEP"));
  57         assert_se(streq(l[1], "NANANANA=YES"));
  58         assert_se(strv_length(l) == 2);
  59 }
  60
  61 static void test_strv_env_set(void) {
  62         _cleanup_strv_free_ char **l = NULL, **r = NULL;
  63
  64         l = strv_new("PIEP", "SCHLUMPF=SMURFF", "NANANANA=YES", NULL);
  65         assert_se(l);
  66
  67         r = strv_env_set(l, "WALDO=WALDO");
  68         assert_se(r);
  69
  70         assert_se(streq(r[0], "PIEP"));
  71         assert_se(streq(r[1], "SCHLUMPF=SMURFF"));
  72         assert_se(streq(r[2], "NANANANA=YES"));
  73         assert_se(streq(r[3], "WALDO=WALDO"));
  74         assert_se(strv_length(r) == 4);
  75 }
  76
  77 static void test_strv_env_merge(void) {
  78         _cleanup_strv_free_ char **a = NULL, **b = NULL, **r = NULL;
  79
  80         a = strv_new("FOO=BAR", "WALDO=WALDO", "WALDO=", "PIEP", "SCHLUMPF=SMURF", NULL);
  81         assert_se(a);
  82
  83         b = strv_new("FOO=KKK", "FOO=", "PIEP=", "SCHLUMPF=SMURFF", "NANANANA=YES", NULL);
  84         assert_se(b);
  85
  86         r = strv_env_merge(2, a, b);
  87         assert_se(r);
  88         assert_se(streq(r[0], "FOO="));
  89         assert_se(streq(r[1], "WALDO="));
  90         assert_se(streq(r[2], "PIEP"));
  91         assert_se(streq(r[3], "SCHLUMPF=SMURFF"));
  92         assert_se(streq(r[4], "PIEP="));
  93         assert_se(streq(r[5], "NANANANA=YES"));
  94         assert_se(strv_length(r) == 6);
  95
  96         assert_se(strv_env_clean(r) == r);
  97         assert_se(streq(r[0], "FOO="));
  98         assert_se(streq(r[1], "WALDO="));
  99         assert_se(streq(r[2], "SCHLUMPF=SMURFF"));
 100         assert_se(streq(r[3], "PIEP="));
 101         assert_se(streq(r[4], "NANANANA=YES"));
 102         assert_se(strv_length(r) == 5);
 103 }
 104
 105 static void test_replace_env_arg(void) {
 106         const char *env[] = {
 107                 "FOO=BAR BAR",
 108                 "BAR=waldo",
 109                 NULL
 110         };
 111         const char *line[] = {
 112                 "FOO$FOO",
 113                 "FOO$FOOFOO",
 114                 "FOO${FOO}$FOO",
 115                 "FOO${FOO}",
 116                 "${FOO}",
 117                 "$FOO",
 118                 "$FOO$FOO",
 119                 "${FOO}${BAR}",
 120                 "${FOO",
 121                 "FOO$$${FOO}",
 122                 "$$FOO${FOO}",
 123                 NULL
 124         };
 125         _cleanup_strv_free_ char **r = NULL;
 126
 127         r = replace_env_argv((char**) line, (char**) env);
 128         assert_se(r);
 129         assert_se(streq(r[0], "FOO$FOO"));
 130         assert_se(streq(r[1], "FOO$FOOFOO"));
 131         assert_se(streq(r[2], "FOOBAR BAR$FOO"));
 132         assert_se(streq(r[3], "FOOBAR BAR"));
 133         assert_se(streq(r[4], "BAR BAR"));
 134         assert_se(streq(r[5], "BAR"));
 135         assert_se(streq(r[6], "BAR"));
 136         assert_se(streq(r[7], "BAR BARwaldo"));
 137         assert_se(streq(r[8], "${FOO"));
 138         assert_se(streq(r[9], "FOO$BAR BAR"));
 139         assert_se(streq(r[10], "$FOOBAR BAR"));
 140         assert_se(strv_length(r) == 11);
 141 }
 142
 143 static void test_env_clean(void) {
 144         _cleanup_strv_free_ char **e;
 145
 146         e = strv_new("FOOBAR=WALDO",
 147                      "FOOBAR=WALDO",
 148                      "FOOBAR",
 149                      "F",
 150                      "X=",
 151                      "F=F",
 152                      "=",
 153                      "=F",
 154                      "",
 155                      "0000=000",
 156                      "äöüß=abcd",
 157                      "abcd=äöüß",
 158                      "xyz\n=xyz",
 159                      "xyz=xyz\n",
 160                      "another=one",
 161                      "another=final one",
 162                      NULL);
 163         assert_se(e);
 164         assert_se(!strv_env_is_valid(e));
 165         assert_se(strv_env_clean(e) == e);
 166         assert_se(strv_env_is_valid(e));
 167
 168         assert_se(streq(e[0], "FOOBAR=WALDO"));
 169         assert_se(streq(e[1], "X="));
 170         assert_se(streq(e[2], "F=F"));
 171         assert_se(streq(e[3], "abcd=äöüß"));
 172         assert_se(streq(e[4], "another=final one"));
 173         assert_se(e[5] == NULL);
 174 }
 175
 176 static void test_env_name_is_valid(void) {
 177         assert_se(env_name_is_valid("test"));
 178
 179         assert_se(!env_name_is_valid(NULL));
 180         assert_se(!env_name_is_valid(""));
 181         assert_se(!env_name_is_valid("xxx\a"));
 182         assert_se(!env_name_is_valid("xxx\007b"));
 183         assert_se(!env_name_is_valid("\007\009"));
 184         assert_se(!env_name_is_valid("5_starting_with_a_number_is_wrong"));
 185         assert_se(!env_name_is_valid("#¤%&?_only_numbers_letters_and_underscore_allowed"));
 186 }
 187
 188 static void test_env_value_is_valid(void) {
 189         assert_se(env_value_is_valid(""));
 190         assert_se(env_value_is_valid("głąb kapuściany"));
 191         assert_se(env_value_is_valid("printf \"\\x1b]0;<mock-chroot>\\x07<mock-chroot>\""));
 192 }
 193
 194 static void test_env_assignment_is_valid(void) {
 195         assert_se(env_assignment_is_valid("a="));
 196         assert_se(env_assignment_is_valid("b=głąb kapuściany"));
 197         assert_se(env_assignment_is_valid("c=\\007\\009\\011"));
 198         assert_se(env_assignment_is_valid("e=printf \"\\x1b]0;<mock-chroot>\\x07<mock-chroot>\""));
 199
 200         assert_se(!env_assignment_is_valid("="));
 201         assert_se(!env_assignment_is_valid("a b="));
 202         assert_se(!env_assignment_is_valid("a ="));
 203         assert_se(!env_assignment_is_valid(" b="));
 204         /* no dots or dashes: http://tldp.org/LDP/abs/html/gotchas.html */
 205         assert_se(!env_assignment_is_valid("a.b="));
 206         assert_se(!env_assignment_is_valid("a-b="));
 207         assert_se(!env_assignment_is_valid("\007=głąb kapuściany"));
 208         assert_se(!env_assignment_is_valid("c\009=\007\009\011"));
 209         assert_se(!env_assignment_is_valid("głąb=printf \"\x1b]0;<mock-chroot>\x07<mock-chroot>\""));
 210 }
 211
 212 int main(int argc, char *argv[]) {
 213         test_strv_env_delete();
 214         test_strv_env_unset();
 215         test_strv_env_set();
 216         test_strv_env_merge();
 217         test_replace_env_arg();
 218         test_env_clean();
 219         test_env_name_is_valid();
 220         test_env_value_is_valid();
 221         test_env_assignment_is_valid();
 222
 223         return 0;
 224 }