fix failure to check malloc result in setenv

This commit is contained in:
Rich Felker 2013-10-04 15:49:55 -04:00
commit 5437870390

18
src/env/setenv.c vendored
View file

@ -18,14 +18,14 @@ int setenv(const char *var, const char *value, int overwrite)
l1 = strlen(var); l1 = strlen(var);
l2 = strlen(value); l2 = strlen(value);
s = malloc(l1+l2+2); s = malloc(l1+l2+2);
memcpy(s, var, l1); if (s) {
s[l1] = '='; memcpy(s, var, l1);
memcpy(s+l1+1, value, l2); s[l1] = '=';
s[l1+l2+1] = 0; memcpy(s+l1+1, value, l2);
if (__putenv(s, 1)) { s[l1+l2+1] = 0;
free(s); if (!__putenv(s, 1)) return 0;
errno = ENOMEM;
return -1;
} }
return 0; free(s);
errno = ENOMEM;
return -1;
} }