First commit

Signed-off-by: Mattias Andrée <maandree@kth.se>

1 files changed, 217 insertions, 0 deletions

diff --git a/libexec_setenvf_prepend.c b/libexec_setenvf_prepend.c
new file mode 100644
index 0000000..3a8a197
--- /dev/null
+++ b/libexec_setenvf_prepend.c
@@ -0,0 +1,217 @@
+/* See LICENSE file for copyright and license details. */
+#include "common.h"
+#ifndef TEST
+
+
+int
+libexec_setenvf_prepend(struct libexec_command *cmd, const char *name, const char *value_fmt, ...)
+{
+	int ret;
+	va_list args;
+	va_start(args, value_fmt);
+	ret = libexec_vsetenvf_prepend(cmd, name, value_fmt, args);
+	va_end(args);
+	return ret;
+}
+
+
+#else
+
+
+int
+main(void)
+{
+	struct libexec_command cmd, ref;
+	char **env;
+
+	errno = 0;
+	ASSERT_EQ_INT(libexec_setenvf_prepend(NULL, "X", "Y"), -1);
+	ASSERT_EQ_INT(errno, EINVAL);
+
+	libexec_init_command(&cmd);
+	memcpy(&ref, &cmd, sizeof(cmd));
+
+	errno = 0;
+	ASSERT_EQ_INT(libexec_setenvf_append(&cmd, "A=B", "C"), -1);
+	ASSERT_EQ_INT(errno, EINVAL);
+	ASSERT_IS_TRUE(!memcmp(&cmd, &ref, sizeof(cmd)));
+
+	errno = 0;
+	ASSERT_EQ_INT(libexec_setenvf_append(&cmd, "A", NULL), -1);
+	ASSERT_EQ_INT(errno, EINVAL);
+	ASSERT_IS_TRUE(!memcmp(&cmd, &ref, sizeof(cmd)));
+
+	errno = 0;
+	ASSERT_EQ_INT(libexec_setenvf_append(&cmd, NULL, "B"), -1);
+	ASSERT_EQ_INT(errno, EINVAL);
+	ASSERT_IS_TRUE(!memcmp(&cmd, &ref, sizeof(cmd)));
+
+	ASSERT_ZERO(libexec_clear_environ(&cmd));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_IS_NULL(cmd.environ[0]);
+
+	ASSERT_ZERO(libexec_setenvf_prepend(&cmd, "A", "%s=%s", "B", "C"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "A=B=C");
+	ASSERT_IS_NULL(cmd.environ[1]);
+
+	ASSERT_ZERO(libexec_setenvf_prepend(&cmd, "A", "%s", "D"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[1]);
+	ASSERT_EQ_STR(cmd.environ[1], "A=B=C");
+	ASSERT_IS_NULL(cmd.environ[2]);
+
+	ASSERT_ZERO(libexec_setenvf_prepend(&cmd, "A", "%c", 'D'));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[1]);
+	ASSERT_EQ_STR(cmd.environ[1], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[2]);
+	ASSERT_EQ_STR(cmd.environ[2], "A=B=C");
+	ASSERT_IS_NULL(cmd.environ[3]);
+
+	ASSERT_ZERO(libexec_setenvf_prepend(&cmd, "AA", "%i", 5));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "AA=5");
+	ASSERT_NOT_NULL(cmd.environ[1]);
+	ASSERT_EQ_STR(cmd.environ[1], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[2]);
+	ASSERT_EQ_STR(cmd.environ[2], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[3]);
+	ASSERT_EQ_STR(cmd.environ[3], "A=B=C");
+	ASSERT_IS_NULL(cmd.environ[4]);
+
+	ASSERT_ZERO(libexec_setenvf_prepend(&cmd, "BB", "F"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "BB=F");
+	ASSERT_NOT_NULL(cmd.environ[1]);
+	ASSERT_EQ_STR(cmd.environ[1], "AA=5");
+	ASSERT_NOT_NULL(cmd.environ[2]);
+	ASSERT_EQ_STR(cmd.environ[2], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[3]);
+	ASSERT_EQ_STR(cmd.environ[3], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[4]);
+	ASSERT_EQ_STR(cmd.environ[4], "A=B=C");
+	ASSERT_IS_NULL(cmd.environ[5]);
+
+	ASSERT_ZERO(libexec_setenvf_prepend(&cmd, "B", "F%s", ""));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "B=F");
+	ASSERT_NOT_NULL(cmd.environ[1]);
+	ASSERT_EQ_STR(cmd.environ[1], "BB=F");
+	ASSERT_NOT_NULL(cmd.environ[2]);
+	ASSERT_EQ_STR(cmd.environ[2], "AA=5");
+	ASSERT_NOT_NULL(cmd.environ[3]);
+	ASSERT_EQ_STR(cmd.environ[3], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[4]);
+	ASSERT_EQ_STR(cmd.environ[4], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[5]);
+	ASSERT_EQ_STR(cmd.environ[5], "A=B=C");
+	ASSERT_IS_NULL(cmd.environ[6]);
+
+	env = cmd.environ;
+	cmd.environ = NULL;
+	ASSERT_IS_TRUE(!memcmp(&cmd, &ref, sizeof(cmd)));
+	cmd.environ = env;
+
+	libexec_destroy_command(&cmd);
+
+	libexec_init_command(&cmd);
+	env = environ;
+	environ = calloc(2, sizeof(*environ));
+	ASSERT_NOT_NULL(environ);
+	environ[0] = strdup("X=0");
+	ASSERT_NOT_NULL(environ[0]);
+
+	ASSERT_ZERO(libexec_setenvf_prepend(&cmd, "A", "%s", "B=C"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "A=B=C");
+	ASSERT_NOT_NULL(cmd.environ[1]);
+	ASSERT_EQ_STR(cmd.environ[1], "X=0");
+	ASSERT_IS_NULL(cmd.environ[2]);
+
+	ASSERT_ZERO(libexec_setenvf_prepend(&cmd, "A", "%s", "D"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[1]);
+	ASSERT_EQ_STR(cmd.environ[1], "A=B=C");
+	ASSERT_NOT_NULL(cmd.environ[2]);
+	ASSERT_EQ_STR(cmd.environ[2], "X=0");
+	ASSERT_IS_NULL(cmd.environ[3]);
+
+	ASSERT_ZERO(libexec_setenvf_prepend(&cmd, "A", "D"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[1]);
+	ASSERT_EQ_STR(cmd.environ[1], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[2]);
+	ASSERT_EQ_STR(cmd.environ[2], "A=B=C");
+	ASSERT_NOT_NULL(cmd.environ[3]);
+	ASSERT_EQ_STR(cmd.environ[3], "X=0");
+	ASSERT_IS_NULL(cmd.environ[4]);
+
+	ASSERT_ZERO(libexec_setenvf_prepend(&cmd, "AA", "E"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "AA=E");
+	ASSERT_NOT_NULL(cmd.environ[1]);
+	ASSERT_EQ_STR(cmd.environ[1], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[2]);
+	ASSERT_EQ_STR(cmd.environ[2], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[3]);
+	ASSERT_EQ_STR(cmd.environ[3], "A=B=C");
+	ASSERT_NOT_NULL(cmd.environ[4]);
+	ASSERT_EQ_STR(cmd.environ[4], "X=0");
+	ASSERT_IS_NULL(cmd.environ[5]);
+
+	ASSERT_ZERO(libexec_setenvf_prepend(&cmd, "BB", "%s%s", "", "F"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "BB=F");
+	ASSERT_NOT_NULL(cmd.environ[1]);
+	ASSERT_EQ_STR(cmd.environ[1], "AA=E");
+	ASSERT_NOT_NULL(cmd.environ[2]);
+	ASSERT_EQ_STR(cmd.environ[2], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[3]);
+	ASSERT_EQ_STR(cmd.environ[3], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[4]);
+	ASSERT_EQ_STR(cmd.environ[4], "A=B=C");
+	ASSERT_NOT_NULL(cmd.environ[5]);
+	ASSERT_EQ_STR(cmd.environ[5], "X=0");
+	ASSERT_IS_NULL(cmd.environ[6]);
+
+	ASSERT_ZERO(libexec_setenvf_prepend(&cmd, "B", "F"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "B=F");
+	ASSERT_NOT_NULL(cmd.environ[1]);
+	ASSERT_EQ_STR(cmd.environ[1], "BB=F");
+	ASSERT_NOT_NULL(cmd.environ[2]);
+	ASSERT_EQ_STR(cmd.environ[2], "AA=E");
+	ASSERT_NOT_NULL(cmd.environ[3]);
+	ASSERT_EQ_STR(cmd.environ[3], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[4]);
+	ASSERT_EQ_STR(cmd.environ[4], "A=D");
+	ASSERT_NOT_NULL(cmd.environ[5]);
+	ASSERT_EQ_STR(cmd.environ[5], "A=B=C");
+	ASSERT_NOT_NULL(cmd.environ[6]);
+	ASSERT_EQ_STR(cmd.environ[6], "X=0");
+	ASSERT_IS_NULL(cmd.environ[7]);
+
+	libexec_destroy_command(&cmd);
+	environ = env;
+	return 0;
+}
+
+
+#endif