[thirdparty/systemd.git] / src / test / test-fileio.c

/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/

/***
  This file is part of systemd.

  Copyright 2013 Lennart Poettering

  systemd is free software; you can redistribute it and/or modify it
  under the terms of the GNU Lesser General Public License as published by
  the Free Software Foundation; either version 2.1 of the License, or
  (at your option) any later version.

  systemd is distributed in the hope that it will be useful, but
  WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  Lesser General Public License for more details.

  You should have received a copy of the GNU Lesser General Public License
  along with systemd; If not, see <http://www.gnu.org/licenses/>.
***/

#include <stdio.h>
#include <fcntl.h>
#include <unistd.h>

#include "util.h"
#include "fileio.h"
#include "strv.h"
#include "env-util.h"
#include "def.h"
#include "ctype.h"

static void test_parse_env_file(void) {
        char    t[] = "/tmp/test-fileio-in-XXXXXX",
                p[] = "/tmp/test-fileio-out-XXXXXX";
        int fd, r;
        FILE *f;
        _cleanup_free_ char *one = NULL, *two = NULL, *three = NULL, *four = NULL, *five = NULL,
                        *six = NULL, *seven = NULL, *eight = NULL, *nine = NULL, *ten = NULL;
        _cleanup_strv_free_ char **a = NULL, **b = NULL;
        char **i;
        unsigned k;

        fd = mkostemp_safe(p, O_RDWR|O_CLOEXEC);
        assert_se(fd >= 0);
        close(fd);

        fd = mkostemp_safe(t, O_RDWR|O_CLOEXEC);
        assert_se(fd >= 0);

        f = fdopen(fd, "w");
        assert_se(f);

        fputs("one=BAR   \n"
              "# comment\n"
              " # comment \n"
              " ; comment \n"
              "  two   =   bar    \n"
              "invalid line\n"
              "invalid line #comment\n"
              "three = \"333\n"
              "xxxx\"\n"
              "four = \'44\\\"44\'\n"
              "five = \'55\\\'55\' \"FIVE\" cinco   \n"
              "six = seis sechs\\\n"
              " sis\n"
              "seven=\"sevenval\" #nocomment\n"
              "eight=eightval #nocomment\n"
              "export nine=nineval\n"
              "ten=", f);

        fflush(f);
        fclose(f);

        r = load_env_file(t, NULL, &a);
        assert_se(r >= 0);

        STRV_FOREACH(i, a)
                log_info("Got: <%s>", *i);

        assert_se(streq_ptr(a[0], "one=BAR"));
        assert_se(streq_ptr(a[1], "two=bar"));
        assert_se(streq_ptr(a[2], "three=333\nxxxx"));
        assert_se(streq_ptr(a[3], "four=44\"44"));
        assert_se(streq_ptr(a[4], "five=55\'55FIVEcinco"));
        assert_se(streq_ptr(a[5], "six=seis sechs sis"));
        assert_se(streq_ptr(a[6], "seven=sevenval#nocomment"));
        assert_se(streq_ptr(a[7], "eight=eightval #nocomment"));
        assert_se(streq_ptr(a[8], "export nine=nineval"));
        assert_se(streq_ptr(a[9], "ten="));
        assert_se(a[10] == NULL);

        strv_env_clean_log(a, "test");

        k = 0;
        STRV_FOREACH(i, b) {
                log_info("Got2: <%s>", *i);
                assert_se(streq(*i, a[k++]));
        }

        r = parse_env_file(
                        t, NULL,
                       "one", &one,
                       "two", &two,
                       "three", &three,
                       "four", &four,
                       "five", &five,
                       "six", &six,
                       "seven", &seven,
                       "eight", &eight,
                       "export nine", &nine,
                       "ten", &ten,
                       NULL);

        assert_se(r >= 0);

        log_info("one=[%s]", strna(one));
        log_info("two=[%s]", strna(two));
        log_info("three=[%s]", strna(three));
        log_info("four=[%s]", strna(four));
        log_info("five=[%s]", strna(five));
        log_info("six=[%s]", strna(six));
        log_info("seven=[%s]", strna(seven));
        log_info("eight=[%s]", strna(eight));
        log_info("export nine=[%s]", strna(nine));
        log_info("ten=[%s]", strna(nine));

        assert_se(streq(one, "BAR"));
        assert_se(streq(two, "bar"));
        assert_se(streq(three, "333\nxxxx"));
        assert_se(streq(four, "44\"44"));
        assert_se(streq(five, "55\'55FIVEcinco"));
        assert_se(streq(six, "seis sechs sis"));
        assert_se(streq(seven, "sevenval#nocomment"));
        assert_se(streq(eight, "eightval #nocomment"));
        assert_se(streq(nine, "nineval"));
        assert_se(ten == NULL);

        r = write_env_file(p, a);
        assert_se(r >= 0);

        r = load_env_file(p, NULL, &b);
        assert_se(r >= 0);

        unlink(t);
        unlink(p);
}

static void test_parse_multiline_env_file(void) {
        char    t[] = "/tmp/test-fileio-in-XXXXXX",
                p[] = "/tmp/test-fileio-out-XXXXXX";
        int fd, r;
        FILE *f;
        _cleanup_strv_free_ char **a = NULL, **b = NULL;
        char **i;

        fd = mkostemp_safe(p, O_RDWR|O_CLOEXEC);
        assert_se(fd >= 0);
        close(fd);

        fd = mkostemp_safe(t, O_RDWR|O_CLOEXEC);
        assert_se(fd >= 0);

        f = fdopen(fd, "w");
        assert_se(f);

        fputs("one=BAR\\\n"
              "    VAR\\\n"
              "\tGAR\n"
              "#comment\n"
              "two=\"bar\\\n"
              "    var\\\n"
              "\tgar\"\n"
              "#comment\n"
              "tri=\"bar \\\n"
              "    var \\\n"
              "\tgar \"\n", f);

        fflush(f);
        fclose(f);

        r = load_env_file(t, NULL, &a);
        assert_se(r >= 0);

        STRV_FOREACH(i, a)
                log_info("Got: <%s>", *i);

        assert_se(streq_ptr(a[0], "one=BAR    VAR\tGAR"));
        assert_se(streq_ptr(a[1], "two=bar    var\tgar"));
        assert_se(streq_ptr(a[2], "tri=bar     var \tgar "));
        assert_se(a[3] == NULL);

        r = write_env_file(p, a);
        assert_se(r >= 0);

        r = load_env_file(p, NULL, &b);
        assert_se(r >= 0);

        unlink(t);
        unlink(p);
}


static void test_executable_is_script(void) {
        char t[] = "/tmp/test-executable-XXXXXX";
        int fd, r;
        FILE *f;
        char *command;

        fd = mkostemp_safe(t, O_RDWR|O_CLOEXEC);
        assert_se(fd >= 0);

        f = fdopen(fd, "w");
        assert_se(f);

        fputs("#! /bin/script -a -b \ngoo goo", f);
        fflush(f);

        r = executable_is_script(t, &command);
        assert_se(r > 0);
        assert_se(streq(command, "/bin/script"));
        free(command);

        r = executable_is_script("/bin/sh", &command);
        assert_se(r == 0);

        r = executable_is_script("/usr/bin/yum", &command);
        assert_se(r > 0 || r == -ENOENT);
        if (r > 0) {
                assert_se(startswith(command, "/"));
                free(command);
        }

        fclose(f);
        unlink(t);
}

static void test_status_field(void) {
        _cleanup_free_ char *t = NULL, *p = NULL, *s = NULL, *z = NULL;
        unsigned long long total = 0, buffers = 0;
        int r;

        assert_se(get_status_field("/proc/self/status", "\nThreads:", &t) == 0);
        puts(t);
        assert_se(streq(t, "1"));

        r = get_status_field("/proc/meminfo", "MemTotal:", &p);
        if (r != -ENOENT) {
                assert(r == 0);
                puts(p);
                assert_se(safe_atollu(p, &total) == 0);
        }

        r = get_status_field("/proc/meminfo", "\nBuffers:", &s);
        if (r != -ENOENT) {
                assert(r == 0);
                puts(s);
                assert_se(safe_atollu(s, &buffers) == 0);
        }

        if (p && t)
                assert(buffers < total);

        /* Seccomp should be a good test for field full of zeros. */
        r = get_status_field("/proc/meminfo", "\nSeccomp:", &z);
        if (r != -ENOENT) {
                assert(r == 0);
                puts(z);
                assert_se(safe_atollu(z, &buffers) == 0);
        }
}

static void test_capeff(void) {
        int pid, p;

        for (pid = 0; pid < 2; pid++) {
                _cleanup_free_ char *capeff = NULL;
                int r;

                r = get_process_capeff(0, &capeff);
                log_info("capeff: '%s' (r=%d)", capeff, r);

                if (r == -ENOENT || r == -EPERM)
                        return;

                assert(r == 0);
                assert(*capeff);
                p = capeff[strspn(capeff, DIGITS "abcdefABCDEF")];
                assert(!p || isspace(p));
        }
}

int main(int argc, char *argv[]) {
        log_parse_environment();
        log_open();

        test_parse_env_file();
        test_parse_multiline_env_file();
        test_executable_is_script();
        test_status_field();
        test_capeff();

        return 0;
}
Commit	Line	Data
f73141d7 LP	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"
ebc05a09	29	#include "env-util.h"
1e5413f7 ZJS	30	#include "def.h"
1e5413f7 ZJS	31	#include "ctype.h"
f73141d7 LP	32
f73141d7 LP	33	static void test_parse_env_file(void) {
095b30cb ZJS	34	char t[] = "/tmp/test-fileio-in-XXXXXX",
095b30cb ZJS	35	p[] = "/tmp/test-fileio-out-XXXXXX";
f73141d7 LP	36	int fd, r;
f73141d7 LP	37	FILE *f;
ebc05a09 HH	38	_cleanup_free_ char one = NULL, two = NULL, three = NULL, four = NULL, *five = NULL,
ebc05a09 HH	39	six = NULL, seven = NULL, eight = NULL, nine = NULL, *ten = NULL;
768100ef	40	_cleanup_strv_free_ char a = NULL, b = NULL;
f73141d7	41	char **i;
768100ef	42	unsigned k;
f73141d7	43
2d5bdf5b	44	fd = mkostemp_safe(p, O_RDWR\|O_CLOEXEC);
d514feaa TA	45	assert_se(fd >= 0);
d514feaa TA	46	close(fd);
095b30cb	47
2d5bdf5b	48	fd = mkostemp_safe(t, O_RDWR\|O_CLOEXEC);
f73141d7 LP	49	assert_se(fd >= 0);
	50
	51	f = fdopen(fd, "w");
	52	assert_se(f);
	53
	54	fputs("one=BAR \n"
	55	"# comment\n"
	56	" # comment \n"
98f59e59	57	" ; comment \n"
f73141d7 LP	58	" two = bar \n"
f73141d7 LP	59	"invalid line\n"
98f59e59	60	"invalid line #comment\n"
f73141d7 LP	61	"three = \"333\n"
	62	"xxxx\"\n"
	63	"four = \'44\\\"44\'\n"
	64	"five = \'55\\\'55\' \"FIVE\" cinco \n"
	65	"six = seis sechs\\\n"
	66	" sis\n"
98f59e59 HH	67	"seven=\"sevenval\" #nocomment\n"
98f59e59 HH	68	"eight=eightval #nocomment\n"
ebc05a09 HH	69	"export nine=nineval\n"
ebc05a09 HH	70	"ten=", f);
f73141d7 LP	71
	72	fflush(f);
	73	fclose(f);
	74
ebc05a09 HH	75	r = load_env_file(t, NULL, &a);
	76	assert_se(r >= 0);
	77
	78	STRV_FOREACH(i, a)
	79	log_info("Got: <%s>", *i);
	80
5fba7bbf TA	81	assert_se(streq_ptr(a[0], "one=BAR"));
	82	assert_se(streq_ptr(a[1], "two=bar"));
	83	assert_se(streq_ptr(a[2], "three=333\nxxxx"));
	84	assert_se(streq_ptr(a[3], "four=44\"44"));
	85	assert_se(streq_ptr(a[4], "five=55\'55FIVEcinco"));
	86	assert_se(streq_ptr(a[5], "six=seis sechs sis"));
	87	assert_se(streq_ptr(a[6], "seven=sevenval#nocomment"));
	88	assert_se(streq_ptr(a[7], "eight=eightval #nocomment"));
	89	assert_se(streq_ptr(a[8], "export nine=nineval"));
	90	assert_se(streq_ptr(a[9], "ten="));
ebc05a09 HH	91	assert_se(a[10] == NULL);
ebc05a09 HH	92
095b30cb	93	strv_env_clean_log(a, "test");
ebc05a09	94
ebc05a09 HH	95	k = 0;
	96	STRV_FOREACH(i, b) {
	97	log_info("Got2: <%s>", *i);
	98	assert_se(streq(*i, a[k++]));
	99	}
	100
f73141d7 LP	101	r = parse_env_file(
	102	t, NULL,
	103	"one", &one,
	104	"two", &two,
	105	"three", &three,
	106	"four", &four,
	107	"five", &five,
	108	"six", &six,
	109	"seven", &seven,
db537209	110	"eight", &eight,
ebc05a09 HH	111	"export nine", &nine,
ebc05a09 HH	112	"ten", &ten,
f73141d7 LP	113	NULL);
	114
	115	assert_se(r >= 0);
	116
	117	log_info("one=[%s]", strna(one));
	118	log_info("two=[%s]", strna(two));
	119	log_info("three=[%s]", strna(three));
	120	log_info("four=[%s]", strna(four));
	121	log_info("five=[%s]", strna(five));
	122	log_info("six=[%s]", strna(six));
	123	log_info("seven=[%s]", strna(seven));
db537209	124	log_info("eight=[%s]", strna(eight));
ebc05a09 HH	125	log_info("export nine=[%s]", strna(nine));
ebc05a09 HH	126	log_info("ten=[%s]", strna(nine));
f73141d7 LP	127
	128	assert_se(streq(one, "BAR"));
	129	assert_se(streq(two, "bar"));
	130	assert_se(streq(three, "333\nxxxx"));
	131	assert_se(streq(four, "44\"44"));
	132	assert_se(streq(five, "55\'55FIVEcinco"));
	133	assert_se(streq(six, "seis sechs sis"));
98f59e59 HH	134	assert_se(streq(seven, "sevenval#nocomment"));
98f59e59 HH	135	assert_se(streq(eight, "eightval #nocomment"));
ebc05a09 HH	136	assert_se(streq(nine, "nineval"));
ebc05a09 HH	137	assert_se(ten == NULL);
f73141d7	138
095b30cb	139	r = write_env_file(p, a);
98f59e59 HH	140	assert_se(r >= 0);
98f59e59 HH	141
095b30cb	142	r = load_env_file(p, NULL, &b);
98f59e59 HH	143	assert_se(r >= 0);
98f59e59 HH	144
f73141d7	145	unlink(t);
095b30cb	146	unlink(p);
f73141d7 LP	147	}
f73141d7 LP	148
ac4c8d6d ZJS	149	static void test_parse_multiline_env_file(void) {
	150	char t[] = "/tmp/test-fileio-in-XXXXXX",
	151	p[] = "/tmp/test-fileio-out-XXXXXX";
	152	int fd, r;
	153	FILE *f;
	154	_cleanup_strv_free_ char a = NULL, b = NULL;
	155	char **i;
	156
2d5bdf5b	157	fd = mkostemp_safe(p, O_RDWR\|O_CLOEXEC);
d514feaa TA	158	assert_se(fd >= 0);
d514feaa TA	159	close(fd);
ac4c8d6d	160
2d5bdf5b	161	fd = mkostemp_safe(t, O_RDWR\|O_CLOEXEC);
ac4c8d6d ZJS	162	assert_se(fd >= 0);
	163
	164	f = fdopen(fd, "w");
	165	assert_se(f);
	166
	167	fputs("one=BAR\\\n"
	168	" VAR\\\n"
	169	"\tGAR\n"
	170	"#comment\n"
	171	"two=\"bar\\\n"
	172	" var\\\n"
	173	"\tgar\"\n"
	174	"#comment\n"
	175	"tri=\"bar \\\n"
	176	" var \\\n"
	177	"\tgar \"\n", f);
	178
	179	fflush(f);
	180	fclose(f);
	181
	182	r = load_env_file(t, NULL, &a);
	183	assert_se(r >= 0);
	184
	185	STRV_FOREACH(i, a)
	186	log_info("Got: <%s>", *i);
	187
5fba7bbf TA	188	assert_se(streq_ptr(a[0], "one=BAR VAR\tGAR"));
	189	assert_se(streq_ptr(a[1], "two=bar var\tgar"));
	190	assert_se(streq_ptr(a[2], "tri=bar var \tgar "));
ac4c8d6d ZJS	191	assert_se(a[3] == NULL);
	192
	193	r = write_env_file(p, a);
	194	assert_se(r >= 0);
	195
	196	r = load_env_file(p, NULL, &b);
	197	assert_se(r >= 0);
	198
	199	unlink(t);
	200	unlink(p);
	201	}
	202
	203
68fee104 ZJS	204	static void test_executable_is_script(void) {
	205	char t[] = "/tmp/test-executable-XXXXXX";
	206	int fd, r;
	207	FILE *f;
	208	char *command;
	209
2d5bdf5b	210	fd = mkostemp_safe(t, O_RDWR\|O_CLOEXEC);
68fee104 ZJS	211	assert_se(fd >= 0);
	212
	213	f = fdopen(fd, "w");
	214	assert_se(f);
	215
	216	fputs("#! /bin/script -a -b \ngoo goo", f);
	217	fflush(f);
	218
	219	r = executable_is_script(t, &command);
	220	assert_se(r > 0);
	221	assert_se(streq(command, "/bin/script"));
	222	free(command);
	223
	224	r = executable_is_script("/bin/sh", &command);
	225	assert_se(r == 0);
	226
	227	r = executable_is_script("/usr/bin/yum", &command);
	228	assert_se(r > 0 \|\| r == -ENOENT);
	229	if (r > 0) {
	230	assert_se(startswith(command, "/"));
	231	free(command);
	232	}
	233
	234	fclose(f);
	235	unlink(t);
	236	}
	237
69ab8088	238	static void test_status_field(void) {
1e5413f7 ZJS	239	_cleanup_free_ char t = NULL, p = NULL, s = NULL, z = NULL;
1e5413f7 ZJS	240	unsigned long long total = 0, buffers = 0;
442e0083	241	int r;
69ab8088 ZJS	242
	243	assert_se(get_status_field("/proc/self/status", "\nThreads:", &t) == 0);
	244	puts(t);
	245	assert_se(streq(t, "1"));
	246
442e0083	247	r = get_status_field("/proc/meminfo", "MemTotal:", &p);
1e5413f7 ZJS	248	if (r != -ENOENT) {
	249	assert(r == 0);
	250	puts(p);
	251	assert_se(safe_atollu(p, &total) == 0);
	252	}
69ab8088	253
d8a11003	254	r = get_status_field("/proc/meminfo", "\nBuffers:", &s);
1e5413f7 ZJS	255	if (r != -ENOENT) {
	256	assert(r == 0);
	257	puts(s);
	258	assert_se(safe_atollu(s, &buffers) == 0);
	259	}
69ab8088	260
1e5413f7 ZJS	261	if (p && t)
	262	assert(buffers < total);
	263
	264	/* Seccomp should be a good test for field full of zeros. */
	265	r = get_status_field("/proc/meminfo", "\nSeccomp:", &z);
	266	if (r != -ENOENT) {
	267	assert(r == 0);
	268	puts(z);
	269	assert_se(safe_atollu(z, &buffers) == 0);
	270	}
	271	}
	272
	273	static void test_capeff(void) {
	274	int pid, p;
	275
	276	for (pid = 0; pid < 2; pid++) {
	277	_cleanup_free_ char *capeff = NULL;
	278	int r;
	279
	280	r = get_process_capeff(0, &capeff);
	281	log_info("capeff: '%s' (r=%d)", capeff, r);
	282
	283	if (r == -ENOENT \|\| r == -EPERM)
	284	return;
	285
	286	assert(r == 0);
	287	assert(*capeff);
	288	p = capeff[strspn(capeff, DIGITS "abcdefABCDEF")];
	289	assert(!p \|\| isspace(p));
	290	}
69ab8088 ZJS	291	}
69ab8088 ZJS	292
f73141d7	293	int main(int argc, char *argv[]) {
1e5413f7 ZJS	294	log_parse_environment();
	295	log_open();
	296
f73141d7	297	test_parse_env_file();
ac4c8d6d	298	test_parse_multiline_env_file();
68fee104	299	test_executable_is_script();
69ab8088	300	test_status_field();
1e5413f7 ZJS	301	test_capeff();
1e5413f7 ZJS	302
f73141d7 LP	303	return 0;
f73141d7 LP	304	}