musl/src/env/setenv.c
Rich Felker 649af9f73a fix for setenv bogus var argument handling
thanks to mikachu

per POSIX:

The setenv() function shall fail if:

[EINVAL] The name argument is a null pointer, points to an empty
string, or points to a string containing an '=' character.
2011-07-28 20:43:40 -04:00

32 lines
521 B
C

#include <stdlib.h>
#include <string.h>
#include <errno.h>
int __putenv(char *s, int a);
int setenv(const char *var, const char *value, int overwrite)
{
char *s;
int l1, l2;
if (!var || !*var || strchr(var, '=')) {
errno = EINVAL;
return -1;
}
if (!overwrite && getenv(var)) return 0;
l1 = strlen(var);
l2 = strlen(value);
s = malloc(l1+l2+2);
memcpy(s, var, l1);
s[l1] = '=';
memcpy(s+l1+1, value, l2);
s[l1+l2+1] = 0;
if (__putenv(s, 1)) {
free(s);
errno = ENOMEM;
return -1;
}
return 0;
}