usr/klibc/__put_env.c

/*
 * __put_env.c - common code for putenv() and setenv()
 */

#include <errno.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "env.h"

static size_t __environ_size;
static char **__environ_alloc;

/* str should be a duplicated version of the input string;
   len is the length of the key including the = sign */

int __put_env(char *str, size_t len, int overwrite)
{
	static char *const null_environ = { NULL };
	char **p, *q;
	char **newenv;
	size_t n;

	if (!environ)
		environ = (char **)null_environ;

	n = 1;			/* Include space for final NULL */
	for (p = environ; (q = *p); p++) {
		n++;
		if (!strncmp(q, str, len)) {
			if (!overwrite)
				free(str);
			else
				*p = str;	/* Possible memory leak... */
			return 0;
		}
	}

	if (__environ_alloc && environ != __environ_alloc) {
		free(__environ_alloc);
		__environ_alloc = NULL;
	}

	/* Need to extend the environment */
	if (n < __environ_size) {
		p[1] = NULL;
		*p = str;
		return 0;
	} else {
		if (__environ_alloc) {
			newenv =
			    realloc(__environ_alloc,
				    (__environ_size << 1) * sizeof(char *));
			if (!newenv)
				return -1;

			__environ_size <<= 1;
		} else {
			/* Make a reasonable guess how much more space
			   we need */
			size_t newsize = n + 32;
			newenv = malloc(newsize * sizeof(char *));
			if (!newenv)
				return -1;

			memcpy(newenv, environ, n * sizeof(char *));
			__environ_size = newsize;
		}
		newenv[n-1] = str;	/* Old NULL position */
		newenv[n]   = NULL;
		environ	= newenv;
	}
	return 0;
}
1	/*
2	* __put_env.c - common code for putenv() and setenv()
3	*/
4
5	#include <errno.h>
6	#include <stdlib.h>
7	#include <string.h>
8	#include <unistd.h>
9	#include "env.h"
10
11	static size_t __environ_size;
12	static char **__environ_alloc;
13
14	/* str should be a duplicated version of the input string;
15	len is the length of the key including the = sign */
16
17	int __put_env(char *str, size_t len, int overwrite)
18	{
19	static char *const null_environ = { NULL };
20	char *p, q;
21	char **newenv;
22	size_t n;
23
24	if (!environ)
25	environ = (char **)null_environ;
26
27	n = 1; /* Include space for final NULL */
28	for (p = environ; (q = *p); p++) {
29	n++;
30	if (!strncmp(q, str, len)) {
31	if (!overwrite)
32	free(str);
33	else
34	p = str; / Possible memory leak... */
35	return 0;
36	}
37	}
38
39	if (__environ_alloc && environ != __environ_alloc) {
40	free(__environ_alloc);
41	__environ_alloc = NULL;
42	}
43
44	/* Need to extend the environment */
45	if (n < __environ_size) {
46	p[1] = NULL;
47	*p = str;
48	return 0;
49	} else {
50	if (__environ_alloc) {
51	newenv =
52	realloc(__environ_alloc,
53	(__environ_size << 1) * sizeof(char *));
54	if (!newenv)
55	return -1;
56
57	__environ_size <<= 1;
58	} else {
59	/* Make a reasonable guess how much more space
60	we need */
61	size_t newsize = n + 32;
62	newenv = malloc(newsize * sizeof(char *));
63	if (!newenv)
64	return -1;
65
66	memcpy(newenv, environ, n * sizeof(char *));
67	__environ_size = newsize;
68	}
69	newenv[n-1] = str; /* Old NULL position */
70	newenv[n] = NULL;
71	environ = newenv;
72	}
73	return 0;
74	}