home *** CD-ROM | disk | FTP | other *** search
/ Shareware 1 2 the Maxx / sw_1.zip / sw_1 / PROGRAM / DJLSR106.ZIP / LIBSRC / C / LIB / SETENV.C < prev    next >
C/C++ Source or Header  |  1992-03-02  |  3KB  |  107 lines
  1. /* This file may have been modified by DJ Delorie (Jan 1991).  If so,
  2. ** these modifications are Coyright (C) 1991 DJ Delorie, 24 Kirsten Ave,
  3. ** Rochester NH, 03867-2954, USA.
  4. */
  5.  
  6. /*
  7.  * Copyright (c) 1987 Regents of the University of California.
  8.  * All rights reserved.
  9.  *
  10.  * Redistribution and use in source and binary forms are permitted
  11.  * provided that: (1) source distributions retain this entire copyright
  12.  * notice and comment, and (2) distributions including binaries display
  13.  * the following acknowledgement:  ``This product includes software
  14.  * developed by the University of California, Berkeley and its contributors''
  15.  * in the documentation or other materials provided with the distribution
  16.  * and in all advertising materials mentioning features or use of this
  17.  * software. Neither the name of the University nor the names of its
  18.  * contributors may be used to endorse or promote products derived
  19.  * from this software without specific prior written permission.
  20.  * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR
  21.  * IMPLIED WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED
  22.  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
  23.  */
  24.  
  25. #if defined(LIBC_SCCS) && !defined(lint)
  26. static char sccsid[] = "@(#)setenv.c    5.4 (Berkeley) 6/1/90";
  27. #endif /* LIBC_SCCS and not lint */
  28.  
  29. #include <stddef.h>
  30. #include <stdlib.h>
  31.  
  32. /*
  33.  * setenv --
  34.  *    Set the value of the environmental variable "name" to be
  35.  *    "value".  If rewrite is set, replace any current value.
  36.  */
  37. setenv(name, value, rewrite)
  38.     register char *name, *value;
  39.     int rewrite;
  40. {
  41.     extern char **environ;
  42.     static int alloced;            /* if allocated space before */
  43.     register char *C;
  44.     int l_value, offset;
  45.     char *_findenv();
  46.  
  47.     if (*value == '=')            /* no `=' in value */
  48.         ++value;
  49.     l_value = strlen(value);
  50.     if ((C = _findenv(name, &offset))) {    /* find if already exists */
  51.         if (!rewrite)
  52.             return(0);
  53.         if (strlen(C) >= l_value) {    /* old larger; copy over */
  54.             while (*C++ = *value++);
  55.             return(0);
  56.         }
  57.     }
  58.     else {                    /* create new slot */
  59.         register int    cnt;
  60.         register char    **P;
  61.  
  62.         for (P = environ, cnt = 0; *P; ++P, ++cnt);
  63.         if (alloced) {            /* just increase size */
  64.             environ = (char **)realloc((char *)environ,
  65.                 (size_t)(sizeof(char *) * (cnt + 2)));
  66.             if (!environ)
  67.                 return(-1);
  68.         }
  69.         else {                /* get new space */
  70.             alloced = 1;        /* copy old entries into it */
  71.             P = (char **)malloc((size_t)(sizeof(char *) *
  72.                 (cnt + 2)));
  73.             if (!P)
  74.                 return(-1);
  75.             bcopy(environ, P, cnt * sizeof(char *));
  76.             environ = P;
  77.         }
  78.         environ[cnt + 1] = NULL;
  79.         offset = cnt;
  80.     }
  81.     for (C = name; *C && *C != '='; ++C);    /* no `=' in name */
  82.     if (!(environ[offset] =            /* name + `=' + value */
  83.         malloc((size_t)((int)(C - name) + l_value + 2))))
  84.         return(-1);
  85.     for (C = environ[offset]; (*C = *name++) && *C != '='; ++C);
  86.     for (*C++ = '='; *C++ = *value++;);
  87.     return(0);
  88. }
  89.  
  90. /*
  91.  * unsetenv(name) --
  92.  *    Delete environmental variable "name".
  93.  */
  94. void
  95. unsetenv(name)
  96.     char    *name;
  97. {
  98.     extern char **environ;
  99.     register char **P;
  100.     int offset;
  101.  
  102.     while (_findenv(name, &offset))        /* if set multiple times */
  103.         for (P = &environ[offset];; ++P)
  104.             if (!(*P = *(P + 1)))
  105.                 break;
  106. }
  107.