-/*
- * Copyright (c) 1987, 1993
- * The Regents of the University of California. All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- * 1. Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- * 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.
- * 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.
- *
- * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
- * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
- * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
- * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
- * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
- * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
- * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
- * SUCH DAMAGE.
- */
-
-#if defined(LIBC_SCCS) && !defined(lint)
-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 $");
-
-#include <stddef.h>
-#include <stdlib.h>
-#include <string.h>
-#include <crt_externs.h>
-#include <errno.h>
-#include <sys/types.h>
-#include <fcntl.h>
-#include <malloc/malloc.h>
-
-#define ZONE_OWNS_PTR(zone, ptr) (malloc_zone_from_ptr((ptr)) == zone)
-
-extern malloc_zone_t *__zone0;
-extern void __malloc_check_env_name(const char *);
-
-__private_extern__ char *__findenv(const char *, int *, char **);
-__private_extern__ int __setenv(const char *, const char *, int, int, char ***, malloc_zone_t *);
-__private_extern__ void __unsetenv(const char *, char **, malloc_zone_t *);
-__private_extern__ int init__zone0(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 */
-
-/*
- * Create the environment malloc zone and give it a recognizable name.
- */
-__private_extern__ int
-init__zone0(int should_set_errno)
-{
- if (__zone0) return (0);
-
- __zone0 = malloc_create_zone(0, 0);
- if (!__zone0) {
- if (should_set_errno) {
- errno = ENOMEM;
- }
- return (-1);
- }
- malloc_set_zone_name(__zone0, "environ");
- 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(name, value, rewrite, copy, environp, envz)
- const char *name;
- const char *value;
- int rewrite, copy;
- char ***environp;
- malloc_zone_t *envz;
-{
- char *c;
- int offset;
-
- if ((c = __findenv(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 (copy > 0 && ZONE_OWNS_PTR(envz, e)) {
- 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;
- }
- }
- while ( (*c++ = *value++) );
- return (0);
- }
- } else { /* create new slot */
- int cnt;
- char **p;
-
- for (p = *environp, cnt = 0; *p; ++p, ++cnt);
- if (ZONE_OWNS_PTR(envz, *environp)) { /* just increase size */
- p = (char **)realloc((char *)*environp,
- (size_t)(sizeof(char *) * (cnt + 2)));
- if (!p)
- return (-1);
- *environp = p;
- }
- else { /* get new space */
- /* copy old entries into it */
- p = malloc_zone_malloc(envz, (size_t)(sizeof(char *) * (cnt + 2)));
- if (!p)
- return (-1);
- 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_zone_malloc(envz, (size_t)((int)(c - name) + strlen(value) + 2))))
- return (-1);
- 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_zone_malloc(envz, len + 1)) == NULL)
- return (-1);
- memcpy(c, name, len + 1);
- name = c;
- }
- /* if we malloc-ed the previous value, free it first */
- if ((*environp)[offset] != NULL && ZONE_OWNS_PTR(envz, (*environp)[offset]))
- free((*environp)[offset]);
- (*environp)[offset] = (char *)name;
- }
- return (0);
-}
-
-__private_extern__ void
-__unsetenv(const char *name, char **environ, malloc_zone_t *envz)
-{
- char **p;
- int offset;
-
- while (__findenv(name, &offset, environ)) { /* if set multiple times */
- /* if we malloc-ed it, free it first */
- if (ZONE_OWNS_PTR(envz, environ[offset]))
- free(environ[offset]);
- for (p = &environ[offset];; ++p)
- if (!(*p = *(p + 1)))
- break;
- }
-}
-
-/****************************************************************************/
-/*
- * _allocenvstate -- SPI that creates a new state (opaque)
- */
-void *
-_allocenvstate(void)
-{
- malloc_zone_t *zone;
- zone = malloc_create_zone(1000 /* unused */, 0 /* unused */);
- if (zone) {
- malloc_set_zone_name(zone, "environ");
- }
- return (void *)zone;
-}
-
-/*
- * _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 will
- * automatically make a copy in the zone.)
- */
-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)
-{
- malloc_zone_t *envz;
-
- if (!(envz = (malloc_zone_t *)state) || envz == __zone0) {
- errno = EINVAL;
- return -1;
- }
- malloc_destroy_zone(envz);
- 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)
-{
- if (init__zone0(1)) return (-1);
- return (__setenv(name, value, rewrite, 1, envp, (state ? (malloc_zone_t *)state : __zone0)));
-}
-
-/*
- * 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)
-{
- if (init__zone0(1)) return (-1);
- __unsetenv(name, *envp, (state ? (malloc_zone_t *)state : __zone0));
- return 0;
-}
-
-#endif /* !BUILD_VARIANT */
-
-/*
- * setenv --
- * Set the value of the environmental variable "name" to be
- * "value". If rewrite is set, replace any current value.
- */
-int
-setenv(name, value, rewrite)
- const char *name;
- const char *value;
- int rewrite;
-{
-#ifdef LEGACY_CRT1_ENVIRON
- int ret;
-#endif /* LEGACY_CRT1_ENVIRON */
-
- /* no null ptr or empty str */
- if(name == NULL || *name == 0) {
- errno = EINVAL;
- return (-1);
- }
-
-#if __DARWIN_UNIX03
- /* no '=' in name */
- if (strchr(name, '=')) {
- errno = EINVAL;
- return (-1);
- }
-#endif /* __DARWIN_UNIX03 */
-
- if (*value == '=') /* no `=' in value */
- ++value;
- /* insure __zone0 is set up before calling __malloc_check_env_name */
- if (init__zone0(1)) return (-1);
- __malloc_check_env_name(name); /* see if we are changing a malloc environment variable */
-#ifdef LEGACY_CRT1_ENVIRON
- ret = __setenv(name, value, rewrite, 1, _NSGetEnviron(), __zone0);
- _saved_environ = *_NSGetEnviron();
- return ret;
-#else /* !LEGACY_CRT1_ENVIRON */
- return (__setenv(name, value, rewrite, 1, _NSGetEnviron(), __zone0));
-#endif /* !LEGACY_CRT1_ENVIRON */
-}
-
-/*
- * unsetenv(name) --
- * Delete environmental variable "name".
- */
-#if __DARWIN_UNIX03
-int
-#else /* !__DARWIN_UNIX03 */
-void
-#endif /* __DARWIN_UNIX03 */
-unsetenv(name)
- const char *name;
-{
-#if __DARWIN_UNIX03
- /* no null ptr or empty str */
- if(name == NULL || *name == 0) {
- errno = EINVAL;
- return (-1);
- }
-
- /* no '=' in name */
- if (strchr(name, '=')) {
- errno = EINVAL;
- return (-1);
- }
- /* insure __zone0 is set up before calling __malloc_check_env_name */
- if (init__zone0(1)) return (-1);
-#else /* !__DARWIN_UNIX03 */
- /* no null ptr or empty str */
- if(name == NULL || *name == 0)
- return;
- /* insure __zone0 is set up before calling __malloc_check_env_name */
- if (init__zone0(0)) return;
-#endif /* __DARWIN_UNIX03 */
- __malloc_check_env_name(name); /* see if we are changing a malloc environment variable */
- __unsetenv(name, *_NSGetEnviron(), __zone0);
-#if __DARWIN_UNIX03
- return 0;
-#endif /* __DARWIN_UNIX03 */
-}
projects / apple / libc.git / blobdiff