home *** CD-ROM | disk | FTP | other *** search
/ Fresh Fish 8 / FreshFishVol8-CD2.bin / bbs / gnu / sh-utils-1.12-src.lha / sh-utils-1.12 / lib / putenv.c < prev    next >
C/C++ Source or Header  |  1994-09-26  |  3KB  |  112 lines
  1. /* Copyright (C) 1991 Free Software Foundation, Inc.
  2.  
  3.    This program is free software; you can redistribute it and/or modify
  4.    it under the terms of the GNU General Public License as published by
  5.    the Free Software Foundation; either version 2, or (at your option)
  6.    any later version.
  7.  
  8.    This program is distributed in the hope that it will be useful,
  9.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  10.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11.    GNU General Public License for more details.
  12.  
  13.    You should have received a copy of the GNU General Public License
  14.    along with this program; if not, write to the Free Software
  15.    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
  16.  
  17. #ifdef HAVE_CONFIG_H
  18. #include <config.h>
  19. #endif
  20.  
  21. #include <sys/types.h>
  22. #include <errno.h>
  23. #ifndef errno
  24. extern int errno;
  25. #endif
  26.  
  27. /* Don't include stdlib.h for non-GNU C libraries because some of them
  28.    contain conflicting prototypes for getopt.
  29.    This needs to come after some library #include
  30.    to get __GNU_LIBRARY__ defined.  */
  31. #ifdef    __GNU_LIBRARY__
  32. #include <stdlib.h>
  33. #else
  34. char *malloc ();
  35. #endif    /* GNU C library.  */
  36.  
  37. #if defined(STDC_HEADERS) || defined(HAVE_STRING_H)
  38. #include <string.h>
  39. #else
  40. #include <strings.h>
  41. #ifndef strchr
  42. #define strchr index
  43. #endif
  44. #ifndef memcpy
  45. #define memcpy(d, s, n) bcopy((s), (d), (n))
  46. #endif
  47. #endif
  48.  
  49. #ifdef HAVE_UNISTD_H
  50. #include <unistd.h>
  51. #endif
  52.  
  53. #ifndef NULL
  54. #define NULL 0
  55. #endif
  56.  
  57. extern char **environ;
  58.  
  59. /* Put STRING, which is of the form "NAME=VALUE", in the environment.  */
  60. int
  61. putenv (string)
  62.      const char *string;
  63. {
  64.   char *name_end = strchr (string, '=');
  65.   register size_t size;
  66.   register char **ep;
  67.  
  68.   if (name_end == NULL)
  69.     {
  70.       /* Remove the variable from the environment.  */
  71.       size = strlen (string);
  72.       for (ep = environ; *ep != NULL; ++ep)
  73.     if (!strncmp (*ep, string, size) && (*ep)[size] == '=')
  74.       {
  75.         while (ep[1] != NULL)
  76.           {
  77.         ep[0] = ep[1];
  78.         ++ep;
  79.           }
  80.         *ep = NULL;
  81.         return 0;
  82.       }
  83.     }
  84.  
  85.   size = 0;
  86.   for (ep = environ; *ep != NULL; ++ep)
  87.     if (!strncmp (*ep, string, name_end - string) &&
  88.     (*ep)[name_end - string] == '=')
  89.       break;
  90.     else
  91.       ++size;
  92.  
  93.   if (*ep == NULL)
  94.     {
  95.       static char **last_environ = NULL;
  96.       char **new_environ = (char **) malloc ((size + 2) * sizeof (char *));
  97.       if (new_environ == NULL)
  98.     return -1;
  99.       memcpy ((char *) new_environ, (char *) environ, size * sizeof (char *));
  100.       new_environ[size] = (char *) string;
  101.       new_environ[size + 1] = NULL;
  102.       if (last_environ != NULL)
  103.     free ((char *) last_environ);
  104.       last_environ = new_environ;
  105.       environ = new_environ;
  106.     }
  107.   else
  108.     *ep = (char *) string;
  109.  
  110.   return 0;
  111. }
  112.