Loading...
--- Libc/Libc-1725.40.4/stdlib/FreeBSD/setenv.c
+++ Libc/Libc-391/stdlib/FreeBSD/setenv.c
@@ -10,6 +10,10 @@
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
+ * 3. All advertising materials mentioning features or use of this software
+ * must display the following acknowledgement:
+ * This product includes software developed by the University of
+ * California, Berkeley and its contributors.
* 4. Neither the name of the University nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
@@ -31,411 +35,13 @@
static char sccsid[] = "@(#)setenv.c 8.1 (Berkeley) 6/4/93";
#endif /* LIBC_SCCS and not lint */
#include <sys/cdefs.h>
-__FBSDID("$FreeBSD: src/lib/libc/stdlib/setenv.c,v 1.14 2007/05/01 16:02:41 ache Exp $");
+__FBSDID("$FreeBSD: src/lib/libc/stdlib/setenv.c,v 1.9 2002/03/22 21:53:10 obrien Exp $");
#include <stddef.h>
#include <stdlib.h>
#include <string.h>
-#include <crt_externs.h>
-#include <errno.h>
-#include <sys/types.h>
-#include <fcntl.h>
-#include "libc_private.h"
-
-struct owned_ptr;
-extern char *__findenv_locked(const char *, int *, char **);
-extern int __setenv_locked(const char *, const char *, int, int, char ***, struct owned_ptr *);
-extern void __unsetenv_locked(const char *, char **, struct owned_ptr *);
-
-extern struct owned_ptr *__env_owned;
-extern int __init__env_owned_locked(int);
-
-/*
- * _cthread_init_routine used to be called from crt1.o to initialize threads.
- * This is no longer needed, as initialization happens in dylib initializers,
- * but is provided to maintain backwards compatibility. Normally, for 10.5
- * or greater, _cthread_init_routine does nothing.
- *
- * Before 10.5, the _start routine in crt1.o clobbers environ with the original
- * stack value, which effectively undoes any environment changes made in
- * initializers. When LEGACY_CRT1_ENVIRON is defined, we replace the
- * do-nothing routine with one that attempts to restore the environ value.
- * But this only works if the setenv (and family) routines were used
- * exclusively, (no direct manipulation of environ). Note that according to
- * SUSv3, direct manipulation of environ may result in undefined behavior in
- * setenv and family, so we don't support that (on less than 10.5).
- */
-#ifdef BUILDING_VARIANT
-# ifdef LEGACY_CRT1_ENVIRON
-extern char **_saved_environ;
-# endif /* LEGACY_CRT1_ENVIRON */
-#else /* !BUILDING_VARIANT */
-# ifdef LEGACY_CRT1_ENVIRON
-__private_extern__ char **_saved_environ = NULL;
-
-static int
-_legacy_crt1_environ(void)
-{
- if (_saved_environ) *_NSGetEnviron() = _saved_environ;
- return 0;
-}
-int (*_cthread_init_routine)(void) = _legacy_crt1_environ;
-
-# else /* !LEGACY_CRT1_ENVIRON */
-static int _do_nothing(void) { return 0; }
-int (*_cthread_init_routine)(void) = _do_nothing;
-# endif /* !LEGACY_CRT1_ENVIRON */
-
-__private_extern__ struct owned_ptr *__env_owned = NULL;
-
-/*
- * The owned_ptr structure is a table of pointers that we own, and can
- * realloc/free as we choose. table[0] is always NULL, so it is always
- * less that any real pointer. "used" is the number of table entries in use
- * (including the initial NULL), while "size" is the allocated size of the
- * table (used to enlarge the size of the table when needed).
- */
-struct owned_ptr {
- const void **table;
- int used;
- int size;
-};
-
-#define OWNED_PTR_INITIAL_SIZE 8
-
-__private_extern__ int _owned_ptr_search(struct owned_ptr * __restrict, const void * __restrict, int * __restrict);
-
-__private_extern__ void
-_owned_ptr_add(struct owned_ptr * __restrict owned, const void * __restrict ptr)
-{
- int index;
-
- if (_owned_ptr_search(owned, ptr, &index) == 0) return; /* already there */
- if (owned->used >= owned->size) {
- int new_size = 2 * owned->size;
- const void **new_table = (const void **)realloc(owned->table,
- new_size * sizeof(const void *));
- if (!new_table) {
- /* no memory to enlarge the table, so just drop */
- return;
- }
- owned->table = new_table;
- owned->size = new_size;
- }
- memmove(owned->table + index + 2, owned->table + index + 1,
- sizeof(void *) * (owned->used - index - 1));
- owned->table[index + 1] = ptr;
- owned->used++;
-}
-
-__private_extern__ struct owned_ptr *
-_owned_ptr_alloc(void)
-{
- struct owned_ptr *owned;
-
- owned = (struct owned_ptr *)malloc(sizeof(struct owned_ptr));
- if (!owned) return NULL;
- owned->table = (const void **)malloc(OWNED_PTR_INITIAL_SIZE *
- sizeof(const void *));
- if (!owned->table) {
- int save = errno;
- free(owned);
- errno = save;
- return NULL;
- }
- owned->table[0] = NULL;
- owned->used = 1;
- owned->size = OWNED_PTR_INITIAL_SIZE;
- return owned;
-}
-
-__private_extern__ void
-_owned_ptr_delete(struct owned_ptr *owned, int index)
-{
- if (!index || index >= owned->used) return;
- memmove(owned->table + index, owned->table + index + 1,
- sizeof(void *) * (owned->used - index - 1));
- owned->used--;
-}
-
-__private_extern__ void
-_owned_ptr_free(struct owned_ptr *owned)
-{
- free(owned->table);
- free(owned);
-}
-
-/*
- * Search owned->table for "ptr". Zero is returned if found, non-zero means
- * not found. If "result" is non-NULL, the found index is returned in it, or
- * if not found, the index of the immediate lower value (ptr can be inserted
- * after this index).
- */
-__private_extern__ int
-_owned_ptr_search(struct owned_ptr * __restrict owned, const void * __restrict ptr, int * __restrict result)
-{
- int low = 0;
- int high = owned->used - 1;
- int cur;
-
- if (owned->table[high] < ptr) {
- if (result) *result = high;
- return -1;
- } else if (owned->table[high] == ptr) {
- if (result) *result = high;
- return 0;
- }
- while (high - low > 1) {
- cur = (low + high) / 2;
- if (ptr > owned->table[cur]) {
- low = cur;
- } else if (ptr < owned->table[cur]) {
- high = cur;
- } else {
- /* match found */
- if (result) *result = cur;
- return 0;
- }
- }
- /* no match found; *result will be the insert-after position */
- if (result) *result = low;
- return -1;
-}
-
-/*
- * Initialize the process's __env_owned structure
- */
-__private_extern__ int
-__init__env_owned_locked(int should_set_errno)
-{
- int save;
-
- if (__env_owned) return 0;
-
- if (!should_set_errno)
- save = errno;
- __env_owned = _owned_ptr_alloc();
- if (!__env_owned) {
- if (!should_set_errno)
- errno = save;
- return -1;
- }
- return 0;
-}
-
-/*
- * The copy flag may have 3 values:
- * 1 - make a copy of the name/value pair
- * 0 - take the name as a user-supplied name=value string
- * -1 - like 0, except we copy of the name=value string in name
- */
-__private_extern__ int
-__setenv_locked(const char *name, const char *value, int rewrite, int copy,
- char ***environp, struct owned_ptr *owned)
-{
- char *c;
- int offset;
- int oindex;
-
- if ((c = __findenv_locked(name, &offset, *environp))) { /* find if already exists */
- char *e;
- if (!rewrite)
- return (0);
- /*
- * In UNIX03, we can overwrite only if we allocated the
- * string. Then we can realloc it if it is too small.
- */
- e = (*environp)[offset];
- if (_owned_ptr_search(owned, e, &oindex) == 0) {
- if (copy > 0) {
- size_t l_value = strlen(value);
- if (strlen(c) < l_value) { /* old smaller; resize*/
- char *r;
- size_t len = c - e;
- if ((r = realloc(e, l_value + len + 1)) == NULL)
- return (-1);
- if (r != e) {
- (*environp)[offset] = r;
- c = r + len;
- _owned_ptr_delete(owned, oindex);
- _owned_ptr_add(owned, r);
- }
- }
- while ( (*c++ = *value++) );
- return (0);
- }
- /* Free up the slot for later use (putenv) */
- _owned_ptr_delete(owned, oindex);
- free(e);
- }
- } else { /* create new slot */
- int cnt;
- char **p;
-
- for (p = *environp, cnt = 0; *p; ++p, ++cnt);
- if (_owned_ptr_search(owned, *environp, &oindex) == 0) { /* just increase size */
- p = (char **)realloc((char *)*environp,
- (size_t)(sizeof(char *) * (cnt + 2)));
- if (!p)
- return (-1);
- if (*environp != p) {
- _owned_ptr_delete(owned, oindex);
- _owned_ptr_add(owned, p);
- *environp = p;
- }
- }
- else { /* get new space */
- /* copy old entries into it */
- p = malloc((size_t)(sizeof(char *) * (cnt + 2)));
- if (!p)
- return (-1);
- _owned_ptr_add(owned, p);
- bcopy(*environp, p, cnt * sizeof(char *));
- *environp = p;
- }
- (*environp)[cnt + 1] = NULL;
- offset = cnt;
- }
- /* For non Unix03, or UnixO3 setenv(), we make a copy of the user's
- * strings. For Unix03 putenv(), we put the string directly in
- * the environment. */
- if (copy > 0) {
- for (c = (char *)name; *c && *c != '='; ++c); /* no `=' in name */
- if (!((*environp)[offset] = /* name + `=' + value */
- malloc((size_t)((int)(c - name) + strlen(value) + 2))))
- return (-1);
- _owned_ptr_add(owned, (*environp)[offset]);
- for (c = (*environp)[offset]; (*c = *name++) && *c != '='; ++c);
- for (*c++ = '='; (*c++ = *value++); );
- } else {
- /* the legacy behavior copies the string */
- if (copy < 0) {
- size_t len = strlen(name);
- if((c = malloc(len + 1)) == NULL)
- return (-1);
- _owned_ptr_add(owned, c);
- memcpy(c, name, len + 1);
- name = c;
- }
- (*environp)[offset] = (char *)name;
- }
- return (0);
-}
-
-__private_extern__ void
-__unsetenv_locked(const char *name, char **environ, struct owned_ptr *owned)
-{
- char **p;
- int offset;
- int oindex;
-
- while (__findenv_locked(name, &offset, environ)) { /* if set multiple times */
- /* if we malloc-ed it, free it first */
- if (_owned_ptr_search(owned, environ[offset], &oindex) == 0) {
- _owned_ptr_delete(owned, oindex);
- free(environ[offset]);
- }
- for (p = &environ[offset];; ++p)
- if (!(*p = *(p + 1)))
- break;
- }
-}
-
-/****************************************************************************/
-/*
- * _allocenvstate -- SPI that creates a new state (opaque)
- */
-void *
-_allocenvstate(void)
-{
- return _owned_ptr_alloc();
-}
-
-/*
- * _copyenv -- SPI that copies a NULL-tereminated char * array in a newly
- * allocated buffer, compatible with the other SPI env routines. If env
- * is NULL, a char * array composed of a single NULL is returned. NULL
- * is returned on error. (This isn't needed anymore, as __setenv_locked will
- * automatically make a copy.)
- */
-char **
-_copyenv(char **env)
-{
- char **p;
- int cnt = 1;
-
- if (env)
- for (p = env; *p; ++p, ++cnt);
- p = (char **)malloc((size_t)(sizeof(char *) * cnt));
- if (!p)
- return (NULL);
- if (env)
- bcopy(env, p, cnt * sizeof(char *));
- else
- *p = NULL;
- return p;
-}
-
-/*
- * _deallocenvstate -- SPI that frees all the memory associated with the state
- * and all allocated strings, including the environment array itself if it
- * was copied.
- */
-int
-_deallocenvstate(void *state)
-{
- struct owned_ptr *owned;
-
- if (!(owned = (struct owned_ptr *)state) || owned == __env_owned) {
- errno = EINVAL;
- return -1;
- }
- _owned_ptr_free(owned);
- return 0;
-}
-
-/*
- * setenvp -- SPI using an arbitrary pointer to string array and an env state,
- * created by _allocenvstate(). Initial checking is not done.
- *
- * Set the value of the environmental variable "name" to be
- * "value". If rewrite is set, replace any current value.
- */
-int
-_setenvp(const char *name, const char *value, int rewrite, char ***envp, void *state)
-{
- environ_lock_np();
- if (__init__env_owned_locked(1)) {
- environ_unlock_np();
- return (-1);
- }
- int ret = __setenv_locked(name, value, rewrite, 1, envp,
- (state ? (struct owned_ptr *)state : __env_owned));
- environ_unlock_np();
- return ret;
-}
-
-/*
- * unsetenv(name) -- SPI using an arbitrary pointer to string array and an env
- * state, created by _allocenvstate(). Initial checking is not done.
- *
- * Delete environmental variable "name".
- */
-int
-_unsetenvp(const char *name, char ***envp, void *state)
-{
- environ_lock_np();
- if (__init__env_owned_locked(1)) {
- environ_unlock_np();
- return (-1);
- }
- __unsetenv_locked(name, *envp, (state ? (struct owned_ptr *)state : __env_owned));
- environ_unlock_np();
- return 0;
-}
-
-#endif /* !BUILD_VARIANT */
+char *__findenv(const char *, int *);
/*
* setenv --
@@ -443,79 +49,72 @@
* "value". If rewrite is set, replace any current value.
*/
int
-setenv(const char *name, const char *value, int rewrite)
+setenv(name, value, rewrite)
+ const char *name;
+ const char *value;
+ int rewrite;
{
- int ret;
+ extern char **environ;
+ static char **alloced; /* if allocated space before */
+ char *c;
+ int l_value, offset;
- /* no null ptr or empty str */
- if(name == NULL || *name == 0) {
- errno = EINVAL;
+ if (*value == '=') /* no `=' in value */
+ ++value;
+ l_value = strlen(value);
+ if ((c = __findenv(name, &offset))) { /* find if already exists */
+ if (!rewrite)
+ return (0);
+ if (strlen(c) >= l_value) { /* old larger; copy over */
+ while ( (*c++ = *value++) );
+ return (0);
+ }
+ } else { /* create new slot */
+ int cnt;
+ char **p;
+
+ for (p = environ, cnt = 0; *p; ++p, ++cnt);
+ if (alloced == environ) { /* just increase size */
+ p = (char **)realloc((char *)environ,
+ (size_t)(sizeof(char *) * (cnt + 2)));
+ if (!p)
+ return (-1);
+ alloced = environ = p;
+ }
+ else { /* get new space */
+ /* copy old entries into it */
+ p = malloc((size_t)(sizeof(char *) * (cnt + 2)));
+ if (!p)
+ return (-1);
+ bcopy(environ, p, cnt * sizeof(char *));
+ alloced = environ = p;
+ }
+ environ[cnt + 1] = NULL;
+ offset = cnt;
+ }
+ for (c = (char *)name; *c && *c != '='; ++c); /* no `=' in name */
+ if (!(environ[offset] = /* name + `=' + value */
+ malloc((size_t)((int)(c - name) + l_value + 2))))
return (-1);
- }
-
-#if __DARWIN_UNIX03
- /* no '=' in name */
- if (strchr(name, '=')) {
- errno = EINVAL;
- return (-1);
- }
-#endif /* __DARWIN_UNIX03 */
-
- environ_lock_np();
- if (__init__env_owned_locked(1)) {
- environ_unlock_np();
- return (-1);
- }
- ret = __setenv_locked(name, value, rewrite, 1, _NSGetEnviron(), __env_owned);
-#ifdef LEGACY_CRT1_ENVIRON
- _saved_environ = *_NSGetEnviron();
-#endif /* !LEGACY_CRT1_ENVIRON */
- environ_unlock_np();
-
- return ret;
-}
-
-static inline __attribute__((always_inline)) int
-_unsetenv(const char *name, int should_set_errno)
-{
- environ_lock_np();
- if (__init__env_owned_locked(should_set_errno)) {
- environ_unlock_np();
- return (-1);
- }
- __unsetenv_locked(name, *_NSGetEnviron(), __env_owned);
- environ_unlock_np();
- return 0;
+ for (c = environ[offset]; (*c = *name++) && *c != '='; ++c);
+ for (*c++ = '='; (*c++ = *value++); );
+ return (0);
}
/*
* unsetenv(name) --
* Delete environmental variable "name".
*/
-#if __DARWIN_UNIX03
-int
-unsetenv(const char *name)
+void
+unsetenv(name)
+ const char *name;
{
- /* no null ptr or empty str */
- if(name == NULL || *name == 0) {
- errno = EINVAL;
- return (-1);
- }
+ extern char **environ;
+ char **p;
+ int offset;
- /* no '=' in name */
- if (strchr(name, '=')) {
- errno = EINVAL;
- return (-1);
- }
- return _unsetenv(name, 1);
+ while (__findenv(name, &offset)) /* if set multiple times */
+ for (p = &environ[offset];; ++p)
+ if (!(*p = *(p + 1)))
+ break;
}
-#else /* !__DARWIN_UNIX03 */
-void
-unsetenv(const char *name)
-{
- /* no null ptr or empty str */
- if(name == NULL || *name == 0)
- return;
- _unsetenv(name, 0);
-}
-#endif /* __DARWIN_UNIX03 */