First commit

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

1 files changed, 122 insertions, 0 deletions

diff --git a/libexec_getenv.c b/libexec_getenv.c
new file mode 100644
index 0000000..5099b28
--- /dev/null
+++ b/libexec_getenv.c
@@ -0,0 +1,122 @@
+/* See LICENSE file for copyright and license details. */
+#include "common.h"
+#ifndef TEST
+
+
+const char *
+libexec_getenv(struct libexec_command *cmd, const char *name)
+{
+	char **env;
+	size_t i, len;
+
+	if (!cmd || !name) {
+		errno = EINVAL;
+		return NULL;
+	}
+
+	if (strchr(name, '='))
+		return NULL;
+
+	env = cmd->environ ? cmd->environ : environ;
+	len = strlen(name);
+
+	for (i = 0; env[i]; i++)
+		if (!strncmp(env[i], name, len) && env[i][len] == '=')
+			return &env[i][len + 1];
+
+	return NULL;
+}
+
+
+#else
+
+
+int
+main(void)
+{
+	struct libexec_command cmd, ref;
+	char **env;
+
+	libexec_init_command(&cmd);
+	memcpy(&ref, &cmd, sizeof(cmd));
+
+	env = environ;
+	environ = calloc(12, sizeof(*environ));
+	ASSERT_NOT_NULL(environ);
+	environ[0] = strdup("AA=aa");
+	environ[1] = strdup("A=a0");
+	environ[2] = strdup("B=b0");
+	environ[3] = strdup("A=a1");
+	environ[4] = strdup("B=b1");
+	environ[5] = strdup("A=a2");
+	environ[6] = strdup("B=b2");
+	environ[7] = strdup("A=a3");
+	environ[8] = strdup("B=b3");
+	environ[9] = strdup("BB=bb");
+	environ[10] = strdup("X=Y=Z");
+	ASSERT_NOT_NULL(environ[0]);
+	ASSERT_NOT_NULL(environ[1]);
+	ASSERT_NOT_NULL(environ[2]);
+	ASSERT_NOT_NULL(environ[3]);
+	ASSERT_NOT_NULL(environ[4]);
+	ASSERT_NOT_NULL(environ[5]);
+	ASSERT_NOT_NULL(environ[6]);
+	ASSERT_NOT_NULL(environ[7]);
+	ASSERT_NOT_NULL(environ[8]);
+	ASSERT_NOT_NULL(environ[9]);
+	ASSERT_NOT_NULL(environ[10]);
+	ASSERT_IS_NULL(environ[11]);
+
+	errno = 0;
+	ASSERT_IS_NULL(libexec_getenv(&cmd, "PATH"));
+	ASSERT_IS_TRUE(errno == 0);
+
+	errno = 0;
+	ASSERT_IS_NULL(libexec_getenv(NULL, "x"));
+	ASSERT_EQ_INT(errno, EINVAL);
+
+	errno = 0;
+	ASSERT_IS_NULL(libexec_getenv(&cmd, "x"));
+	ASSERT_IS_TRUE(errno == 0);
+
+	errno = 0;
+	ASSERT_IS_NULL(libexec_getenv(&cmd, NULL));
+	ASSERT_EQ_INT(errno, EINVAL);
+
+	errno = 0;
+	ASSERT_IS_NULL(libexec_getenv(&cmd, "X=Y"));
+	ASSERT_IS_TRUE(errno == 0 || errno == EINVAL);
+
+	errno = 0;
+	ASSERT_IS_NULL(libexec_getenv(&cmd, ""));
+	ASSERT_IS_TRUE(errno == 0);
+
+	ASSERT_EQ_STR(libexec_getenv(&cmd, "X"), "Y=Z");
+	ASSERT_EQ_STR(libexec_getenv(&cmd, "AA"), "aa");
+	ASSERT_EQ_STR(libexec_getenv(&cmd, "BB"), "bb");
+	ASSERT_EQ_STR(libexec_getenv(&cmd, "A"), "a0");
+	ASSERT_EQ_STR(libexec_getenv(&cmd, "B"), "b0");
+
+	ASSERT_IS_TRUE(!memcmp(&cmd, &ref, sizeof(cmd)));
+	libexec_destroy_command(&cmd);
+
+	ASSERT_IS_NULL(libexec_getenv(&cmd, "PATH"));
+	ASSERT_IS_NULL(libexec_getenv(NULL, "x"));
+	ASSERT_IS_NULL(libexec_getenv(&cmd, "x"));
+	ASSERT_IS_NULL(libexec_getenv(&cmd, NULL));
+	ASSERT_IS_NULL(libexec_getenv(&cmd, "X=Y"));
+	ASSERT_IS_NULL(libexec_getenv(&cmd, ""));
+	ASSERT_EQ_STR(libexec_getenv(&cmd, "X"), "Y=Z");
+	ASSERT_EQ_STR(libexec_getenv(&cmd, "AA"), "aa");
+	ASSERT_EQ_STR(libexec_getenv(&cmd, "BB"), "bb");
+	ASSERT_EQ_STR(libexec_getenv(&cmd, "A"), "a0");
+	ASSERT_EQ_STR(libexec_getenv(&cmd, "B"), "b0");
+
+	libexec_set_environ(&cmd, environ);
+	environ = env;
+
+	return 0;
+}
+
+
+#endif