stdlib/putenv.c

   1 /* Copyright (C) 1991-2015 Free Software Foundation, Inc.
   2    This file is part of the GNU C Library.
   3
   4    The GNU C Library is free software; you can redistribute it and/or
   5    modify it under the terms of the GNU Lesser General Public
   6    License as published by the Free Software Foundation; either
   7    version 2.1 of the License, or (at your option) any later version.
   8
   9    The GNU C Library is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12    Lesser General Public License for more details.
  13
  14    You should have received a copy of the GNU Lesser General Public
  15    License along with the GNU C Library; if not, see
  16    <http://www.gnu.org/licenses/>.  */
  17
  18 #if defined _AIX && !defined __GNUC__
  19  #pragma alloca
  20 #endif
  21
  22 #if HAVE_CONFIG_H
  23 # include <config.h>
  24 #endif
  25
  26 #if _LIBC || HAVE_STDLIB_H
  27 # include <stdlib.h>
  28 #endif
  29 #if _LIBC || HAVE_STRING_H
  30 # include <string.h>
  31 #endif
  32
  33 #if !__GNU_LIBRARY__ && !HAVE_STRCHR
  34 # define strchr index
  35 #endif
  36
  37 #ifndef _LIBC
  38 # ifdef HAVE_ALLOCA_H
  39 #  include <alloca.h>
  40 # else
  41 #  ifdef __GNUC__
  42 #   define alloca __builtin_alloca
  43 #  else
  44 extern char *alloca ();
  45 #  endif /* __GNUC__ */
  46 # endif /* HAVE_ALLOCA_H */
  47 #endif /* _LIBC */
  48
  49
  50 /* Put STRING, which is of the form "NAME=VALUE", in the environment.  */
  51 int
  52 putenv (string)
  53      char *string;
  54 {
  55   const char *const name_end = strchr (string, '=');
  56
  57   if (name_end != NULL)
  58     {
  59       char *name;
  60 #ifdef _LIBC
  61       int use_malloc = !__libc_use_alloca (name_end - string + 1);
  62       if (__builtin_expect (use_malloc, 0))
  63         {
  64           name = strndup (string, name_end - string);
  65           if (name == NULL)
  66             return -1;
  67         }
  68       else
  69         name = strndupa (string, name_end - string);
  70 #else
  71 # define use_malloc 1
  72       name = malloc (name_end - string + 1);
  73       if (name == NULL)
  74         return -1;
  75       memcpy (name, string, name_end - string);
  76       name[name_end - string] = '\0';
  77 #endif
  78       int result = __add_to_environ (name, NULL, string, 1);
  79
  80       if (__glibc_unlikely (use_malloc))
  81         free (name);
  82
  83       return result;
  84     }
  85
  86   __unsetenv (string);
  87   return 0;
  88 }