src/test/test-fileio.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2013 Lennart Poettering
   7
   8   systemd is free software; you can redistribute it and/or modify it
   9   under the terms of the GNU Lesser General Public License as published by
  10   the Free Software Foundation; either version 2.1 of the License, or
  11   (at your option) any later version.
  12
  13   systemd is distributed in the hope that it will be useful, but
  14   WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16   Lesser General Public License for more details.
  17
  18   You should have received a copy of the GNU Lesser General Public License
  19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  20 ***/
  21
  22 #include <stdio.h>
  23 #include <fcntl.h>
  24 #include <unistd.h>
  25
  26 #include "util.h"
  27 #include "fileio.h"
  28 #include "strv.h"
  29 #include "env-util.h"
  30
  31 static void test_parse_env_file(void) {
  32         char    t[] = "/tmp/test-fileio-in-XXXXXX",
  33                 p[] = "/tmp/test-fileio-out-XXXXXX";
  34         int fd, r;
  35         FILE *f;
  36         _cleanup_free_ char *one = NULL, *two = NULL, *three = NULL, *four = NULL, *five = NULL,
  37                         *six = NULL, *seven = NULL, *eight = NULL, *nine = NULL, *ten = NULL;
  38         _cleanup_strv_free_ char **a = NULL, **b = NULL;
  39         char **i;
  40         unsigned k;
  41
  42         assert_se(mktemp(p));
  43
  44         fd = mkostemp(t, O_CLOEXEC);
  45         assert_se(fd >= 0);
  46
  47         f = fdopen(fd, "w");
  48         assert_se(f);
  49
  50         fputs("one=BAR   \n"
  51               "# comment\n"
  52               " # comment \n"
  53               " ; comment \n"
  54               "  two   =   bar    \n"
  55               "invalid line\n"
  56               "invalid line #comment\n"
  57               "three = \"333\n"
  58               "xxxx\"\n"
  59               "four = \'44\\\"44\'\n"
  60               "five = \'55\\\'55\' \"FIVE\" cinco   \n"
  61               "six = seis sechs\\\n"
  62               " sis\n"
  63               "seven=\"sevenval\" #nocomment\n"
  64               "eight=eightval #nocomment\n"
  65               "export nine=nineval\n"
  66               "ten=", f);
  67
  68         fflush(f);
  69         fclose(f);
  70
  71         r = load_env_file(t, NULL, &a);
  72         assert_se(r >= 0);
  73
  74         STRV_FOREACH(i, a)
  75                 log_info("Got: <%s>", *i);
  76
  77         assert_se(streq(a[0], "one=BAR"));
  78         assert_se(streq(a[1], "two=bar"));
  79         assert_se(streq(a[2], "three=333\nxxxx"));
  80         assert_se(streq(a[3], "four=44\"44"));
  81         assert_se(streq(a[4], "five=55\'55FIVEcinco"));
  82         assert_se(streq(a[5], "six=seis sechs sis"));
  83         assert_se(streq(a[6], "seven=sevenval#nocomment"));
  84         assert_se(streq(a[7], "eight=eightval #nocomment"));
  85         assert_se(streq(a[8], "export nine=nineval"));
  86         assert_se(streq(a[9], "ten="));
  87         assert_se(a[10] == NULL);
  88
  89         strv_env_clean_log(a, "test");
  90
  91         k = 0;
  92         STRV_FOREACH(i, b) {
  93                 log_info("Got2: <%s>", *i);
  94                 assert_se(streq(*i, a[k++]));
  95         }
  96
  97         r = parse_env_file(
  98                         t, NULL,
  99                        "one", &one,
 100                        "two", &two,
 101                        "three", &three,
 102                        "four", &four,
 103                        "five", &five,
 104                        "six", &six,
 105                        "seven", &seven,
 106                        "eight", &eight,
 107                        "export nine", &nine,
 108                        "ten", &ten,
 109                        NULL);
 110
 111         assert_se(r >= 0);
 112
 113         log_info("one=[%s]", strna(one));
 114         log_info("two=[%s]", strna(two));
 115         log_info("three=[%s]", strna(three));
 116         log_info("four=[%s]", strna(four));
 117         log_info("five=[%s]", strna(five));
 118         log_info("six=[%s]", strna(six));
 119         log_info("seven=[%s]", strna(seven));
 120         log_info("eight=[%s]", strna(eight));
 121         log_info("export nine=[%s]", strna(nine));
 122         log_info("ten=[%s]", strna(nine));
 123
 124         assert_se(streq(one, "BAR"));
 125         assert_se(streq(two, "bar"));
 126         assert_se(streq(three, "333\nxxxx"));
 127         assert_se(streq(four, "44\"44"));
 128         assert_se(streq(five, "55\'55FIVEcinco"));
 129         assert_se(streq(six, "seis sechs sis"));
 130         assert_se(streq(seven, "sevenval#nocomment"));
 131         assert_se(streq(eight, "eightval #nocomment"));
 132         assert_se(streq(nine, "nineval"));
 133         assert_se(ten == NULL);
 134
 135         r = write_env_file(p, a);
 136         assert_se(r >= 0);
 137
 138         r = load_env_file(p, NULL, &b);
 139         assert_se(r >= 0);
 140
 141         unlink(t);
 142         unlink(p);
 143 }
 144
 145 static void test_executable_is_script(void) {
 146         char t[] = "/tmp/test-executable-XXXXXX";
 147         int fd, r;
 148         FILE *f;
 149         char *command;
 150
 151         fd = mkostemp(t, O_CLOEXEC);
 152         assert_se(fd >= 0);
 153
 154         f = fdopen(fd, "w");
 155         assert_se(f);
 156
 157         fputs("#! /bin/script -a -b \ngoo goo", f);
 158         fflush(f);
 159
 160         r = executable_is_script(t, &command);
 161         assert_se(r > 0);
 162         assert_se(streq(command, "/bin/script"));
 163         free(command);
 164
 165         r = executable_is_script("/bin/sh", &command);
 166         assert_se(r == 0);
 167
 168         r = executable_is_script("/usr/bin/yum", &command);
 169         assert_se(r > 0 || r == -ENOENT);
 170         if (r > 0) {
 171                 assert_se(startswith(command, "/"));
 172                 free(command);
 173         }
 174
 175         fclose(f);
 176         unlink(t);
 177 }
 178
 179 int main(int argc, char *argv[]) {
 180         test_parse_env_file();
 181         test_executable_is_script();
 182         return 0;
 183 }