/* See LICENSE file for copyright and license details. */
#include "libsimple.h"
#ifndef TEST
int
libsimple_vputenvf(const char *fmt, va_list ap)
{
va_list ap2;
int n, r;
char *s, *p;
va_copy(ap2, ap);
n = vsnprintf(NULL, 0, fmt, ap2);
va_end(ap2);
if (n < 0)
return -1;
s = malloc((size_t)n + 1);
if (!s)
return -1;
vsprintf(s, fmt, ap);
p = strchr(s, '=');
if (p) {
*p++ = '\0';
r = setenv(s, p, 1);
free(s);
} else {
r = putenv(s);
}
return r;
}
#else
#include "test.h"
int
main(void)
{
unsetenv("X");
assert(!getenv("X"));
unsetenv("Y");
assert(!getenv("Y"));
putenvf("X=xyz");
assert(!strcmpnul(getenv("X"), "xyz"));
putenvf("Y=xyz");
assert(!strcmpnul(getenv("Y"), "xyz"));
putenvf("X=x%sz", "abc");
assert(!strcmpnul(getenv("X"), "xabcz"));
putenvf("Y=x%sz", "abc");
assert(!strcmpnul(getenv("Y"), "xabcz"));
putenvf("X=%ix%sz%i", 10, "abc", -11);
assert(!strcmpnul(getenv("X"), "10xabcz-11"));
putenvf("Y=%ix%sz%i", 10, "abc", -11);
assert(!strcmpnul(getenv("Y"), "10xabcz-11"));
return 0;
}
#endif
