/* 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
