home *** CD-ROM | disk | FTP | other *** search
/ Linux Cubed Series 4: GNU Archives / Linux Cubed Series 4 - GNU Archives.iso / gnu / glibc-1.09 / glibc-1 / glibc-1.09.1 / sysdeps / generic / setenv.c < prev    next >
Encoding:
C/C++ Source or Header  |  1994-07-18  |  2.4 KB  |  87 lines
  1. /* Copyright (C) 1992 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 Library General Public License as
  6. published by the Free Software Foundation; either version 2 of the
  7. 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. Library General Public License for more details.
  13.  
  14. You should have received a copy of the GNU Library General Public
  15. License along with the GNU C Library; see the file COPYING.LIB.  If
  16. not, write to the Free Software Foundation, Inc., 675 Mass Ave,
  17. Cambridge, MA 02139, USA.  */
  18.  
  19. #include <ansidecl.h>
  20. #include <stdlib.h>
  21. #include <string.h>
  22. #include <unistd.h>
  23.  
  24. #ifndef    HAVE_GNU_LD
  25. #define    __environ    environ
  26. #endif
  27.  
  28. int
  29. DEFUN(setenv, (name, value, replace),
  30.       CONST char *name AND CONST char *value AND int replace)
  31. {
  32.   register char **ep;
  33.   register size_t size;
  34.   CONST size_t namelen = strlen (name);
  35.   CONST size_t vallen = strlen (value);
  36.  
  37.   size = 0;
  38.   for (ep = __environ; *ep != NULL; ++ep)
  39.     if (!strncmp (*ep, name, namelen) && (*ep)[namelen] == '=')
  40.       break;
  41.     else
  42.       ++size;
  43.   
  44.   if (*ep == NULL)
  45.     {
  46.       static char **last_environ = NULL;
  47.       char **new_environ = (char **) malloc((size + 2) * sizeof(char *));
  48.       if (new_environ == NULL)
  49.     return -1;
  50.       (void) memcpy((PTR) new_environ, (PTR) __environ, size * sizeof(char *));
  51.  
  52.       new_environ[size] = malloc (namelen + 1 + vallen + 1);
  53.       if (new_environ[size] == NULL)
  54.     {
  55.       free (new_environ);
  56.       errno = ENOMEM;
  57.       return -1;
  58.     }
  59.       memcpy (new_environ[size], name, namelen);
  60.       new_environ[size][namelen] = '=';
  61.       memcpy (&new_environ[size][namelen + 1], value, vallen + 1);
  62.  
  63.       new_environ[size + 1] = NULL;
  64.  
  65.       if (last_environ != NULL)
  66.     free ((PTR) last_environ);
  67.       last_environ = new_environ;
  68.       __environ = new_environ;
  69.     }
  70.   else if (replace)
  71.     {
  72.       size_t len = strlen (*ep);
  73.       if (len < namelen + 1 + vallen)
  74.     {
  75.       char *new = malloc (namelen + 1 + vallen);
  76.       if (new == NULL)
  77.         return -1;
  78.       *ep = new;
  79.     }
  80.       memcpy (*ep, name, namelen);
  81.       (*ep)[namelen] = '=';
  82.       memcpy (&(*ep)[namelen + 1], value, vallen + 1);
  83.     }
  84.  
  85.   return 0;
  86. }
  87.