src/env/setenv.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

#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 (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;
}