First commit

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

1 files changed, 176 insertions, 0 deletions

diff --git a/libexec_setenv_replace.c b/libexec_setenv_replace.c
new file mode 100644
index 0000000..f8191c1
--- /dev/null
+++ b/libexec_setenv_replace.c
@@ -0,0 +1,176 @@
+/* See LICENSE file for copyright and license details. */
+#include "common.h"
+#ifndef TEST
+
+
+int
+libexec_setenv_replace(struct libexec_command *cmd, const char *name, const char *value)
+{
+	return libexec_setenv(cmd, LIBEXEC_REPLACE, name, value);
+}
+
+
+#else
+
+
+int
+main(void)
+{
+	struct libexec_command cmd, ref;
+	char **env;
+
+	errno = 0;
+	ASSERT_EQ_INT(libexec_setenv_replace(NULL, "X", "Y"), -1);
+	ASSERT_EQ_INT(errno, EINVAL);
+
+	libexec_init_command(&cmd);
+	memcpy(&ref, &cmd, sizeof(cmd));
+
+	errno = 0;
+	ASSERT_EQ_INT(libexec_setenv_replace(&cmd, "A=B", "C"), -1);
+	ASSERT_EQ_INT(errno, EINVAL);
+	ASSERT_IS_TRUE(!memcmp(&cmd, &ref, sizeof(cmd)));
+
+	errno = 0;
+	ASSERT_EQ_INT(libexec_setenv_replace(&cmd, "A", NULL), -1);
+	ASSERT_EQ_INT(errno, EINVAL);
+	ASSERT_IS_TRUE(!memcmp(&cmd, &ref, sizeof(cmd)));
+
+	errno = 0;
+	ASSERT_EQ_INT(libexec_setenv_replace(&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_setenv_replace(&cmd, "A", "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_setenv_replace(&cmd, "A", "D"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "A=D");
+	ASSERT_IS_NULL(cmd.environ[1]);
+
+	ASSERT_ZERO(libexec_setenv_replace(&cmd, "A", "D"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "A=D");
+	ASSERT_IS_NULL(cmd.environ[1]);
+
+	ASSERT_ZERO(libexec_setenv_replace(&cmd, "AA", "E"));
+	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], "AA=E");
+	ASSERT_IS_NULL(cmd.environ[2]);
+
+	ASSERT_ZERO(libexec_setenv_replace(&cmd, "BB", "F"));
+	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], "AA=E");
+	ASSERT_NOT_NULL(cmd.environ[2]);
+	ASSERT_EQ_STR(cmd.environ[2], "BB=F");
+	ASSERT_IS_NULL(cmd.environ[3]);
+
+	ASSERT_ZERO(libexec_setenv_replace(&cmd, "B", "F"));
+	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], "AA=E");
+	ASSERT_NOT_NULL(cmd.environ[2]);
+	ASSERT_EQ_STR(cmd.environ[2], "BB=F");
+	ASSERT_NOT_NULL(cmd.environ[3]);
+	ASSERT_EQ_STR(cmd.environ[3], "B=F");
+	ASSERT_IS_NULL(cmd.environ[4]);
+
+	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_setenv_replace(&cmd, "A", "B=C"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "X=0");
+	ASSERT_NOT_NULL(cmd.environ[1]);
+	ASSERT_EQ_STR(cmd.environ[1], "A=B=C");
+	ASSERT_IS_NULL(cmd.environ[2]);
+
+	ASSERT_ZERO(libexec_setenv_replace(&cmd, "A", "D"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "X=0");
+	ASSERT_NOT_NULL(cmd.environ[1]);
+	ASSERT_EQ_STR(cmd.environ[1], "A=D");
+	ASSERT_IS_NULL(cmd.environ[2]);
+
+	ASSERT_ZERO(libexec_setenv_replace(&cmd, "A", "D"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "X=0");
+	ASSERT_NOT_NULL(cmd.environ[1]);
+	ASSERT_EQ_STR(cmd.environ[1], "A=D");
+	ASSERT_IS_NULL(cmd.environ[2]);
+
+	ASSERT_ZERO(libexec_setenv_replace(&cmd, "AA", "E"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "X=0");
+	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], "AA=E");
+	ASSERT_IS_NULL(cmd.environ[3]);
+
+	ASSERT_ZERO(libexec_setenv_replace(&cmd, "BB", "F"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "X=0");
+	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], "AA=E");
+	ASSERT_NOT_NULL(cmd.environ[3]);
+	ASSERT_EQ_STR(cmd.environ[3], "BB=F");
+	ASSERT_IS_NULL(cmd.environ[4]);
+
+	ASSERT_ZERO(libexec_setenv_replace(&cmd, "B", "F"));
+	ASSERT_NOT_NULL(cmd.environ);
+	ASSERT_NOT_NULL(cmd.environ[0]);
+	ASSERT_EQ_STR(cmd.environ[0], "X=0");
+	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], "AA=E");
+	ASSERT_NOT_NULL(cmd.environ[3]);
+	ASSERT_EQ_STR(cmd.environ[3], "BB=F");
+	ASSERT_NOT_NULL(cmd.environ[4]);
+	ASSERT_EQ_STR(cmd.environ[4], "B=F");
+	ASSERT_IS_NULL(cmd.environ[5]);
+
+	libexec_destroy_command(&cmd);
+	environ = env;
+	return 0;
+}
+
+
+#endif