2.11BSD#431 - /usr/src/libexec/tcpd/environ.c

   1: /*
   2:  * Many systems have putenv() but no setenv(). Other systems have setenv()
   3:  * but no putenv() (MIPS). Still other systems have neither (NeXT). This is a
   4:  * re-implementation that hopefully ends all problems.
   5:  *
   6:  * Author: Wietse Venema, Eindhoven University of Technology, The Netherlands.
   7:  */
   8: 
   9: #ifndef lint
  10: static char sccsid[] = "@(#) environ.c 1.2 94/03/23 16:09:46";
  11: #endif
  12: 
  13: /* System libraries. */
  14: 
  15: extern char **environ;
  16: extern char *strchr();
  17: extern char *strcpy();
  18: extern char *strncpy();
  19: extern char *malloc();
  20: extern char *realloc();
  21: extern int strncmp();
  22: extern void free();
  23: 
  24: #ifdef no_memcpy
  25: #define memcpy(d,s,l) bcopy(s,d,l)
  26: #else
  27: extern char *memcpy();
  28: #endif
  29: 
  30: /* Local stuff. */
  31: 
  32: static int addenv();            /* append entry to environment */
  33: 
  34: static int allocated = 0;       /* environ is, or is not, allocated */
  35: 
  36: #define DO_CLOBBER  1
  37: 
  38: /* namelength - determine length of name in "name=whatever" */
  39: 
  40: static int namelength(name)
  41: char   *name;
  42: {
  43:     char   *equal;
  44: 
  45:     equal = strchr(name, '=');
  46:     return ((equal == 0) ? strlen(name) : (equal - name));
  47: }
  48: 
  49: /* findenv - given name, locate name=value */
  50: 
  51: static char **findenv(name, len)
  52: char   *name;
  53: int     len;
  54: {
  55:     char  **envp;
  56: 
  57:     for (envp = environ; envp && *envp; envp++)
  58:     if (strncmp(name, *envp, len) == 0 && (*envp)[len] == '=')
  59:         return (envp);
  60:     return (0);
  61: }
  62: 
  63: /* getenv - given name, locate value */
  64: 
  65: char   *getenv(name)
  66: char   *name;
  67: {
  68:     int     len = namelength(name);
  69:     char  **envp = findenv(name, len);
  70: 
  71:     return (envp ? *envp + len + 1 : 0);
  72: }
  73: 
  74: /* putenv - update or append environment (name,value) pair */
  75: 
  76: int     putenv(nameval)
  77: char   *nameval;
  78: {
  79:     char   *equal = strchr(nameval, '=');
  80:     char   *value = (equal ? equal : "");
  81: 
  82:     return (setenv(nameval, value, DO_CLOBBER));
  83: }
  84: 
  85: /* unsetenv - remove variable from environment */
  86: 
  87: void    unsetenv(name)
  88: char   *name;
  89: {
  90:     char  **envp;
  91: 
  92:     if ((envp = findenv(name, namelength(name))) != 0)
  93:     while (envp[0] = envp[1])
  94:         envp++;
  95: }
  96: 
  97: /* setenv - update or append environment (name,value) pair */
  98: 
  99: int     setenv(name, value, clobber)
 100: char   *name;
 101: char   *value;
 102: int     clobber;
 103: {
 104:     char   *destination;
 105:     char  **envp;
 106:     int     l_name;         /* length of name part */
 107:     int     l_nameval;          /* length of name=value */
 108: 
 109:     /* Permit name= and =value. */
 110: 
 111:     l_name = namelength(name);
 112:     envp = findenv(name, l_name);
 113:     if (envp != 0 && clobber == 0)
 114:     return (0);
 115:     if (*value == '=')
 116:     value++;
 117:     l_nameval = l_name + strlen(value) + 1;
 118: 
 119:     /*
 120:      * Use available memory if the old value is long enough. Never free an
 121:      * old name=value entry because it may not be allocated.
 122:      */
 123: 
 124:     destination = (envp != 0 && strlen(*envp) >= l_nameval) ?
 125:     *envp : malloc(l_nameval + 1);
 126:     if (destination == 0)
 127:     return (-1);
 128:     strncpy(destination, name, l_name);
 129:     destination[l_name] = '=';
 130:     strcpy(destination + l_name + 1, value);
 131:     return ((envp == 0) ? addenv(destination) : (*envp = destination, 0));
 132: }
 133: 
 134: /* cmalloc - malloc and copy block of memory */
 135: 
 136: static char *cmalloc(new_len, old, old_len)
 137: char   *old;
 138: int     old_len;
 139: {
 140:     char   *new = malloc(new_len);
 141: 
 142:     if (new != 0)
 143:     memcpy(new, old, old_len);
 144:     return (new);
 145: }
 146: 
 147: /* addenv - append environment entry */
 148: 
 149: static int addenv(nameval)
 150: char   *nameval;
 151: {
 152:     char  **envp;
 153:     int     n_used;         /* number of environment entries */
 154:     int     l_used;         /* bytes used excl. terminator */
 155:     int     l_need;         /* bytes needed incl. terminator */
 156: 
 157:     for (envp = environ; envp && *envp; envp++)
 158:      /* void */ ;
 159:     n_used = envp - environ;
 160:     l_used = n_used * sizeof(*envp);
 161:     l_need = l_used + 2 * sizeof(*envp);
 162: 
 163:     envp = allocated ?
 164:     (char **) realloc((char *) environ, l_need) :
 165:     (char **) cmalloc(l_need, (char *) environ, l_used);
 166:     if (envp == 0) {
 167:     return (-1);
 168:     } else {
 169:     allocated = 1;
 170:     environ = envp;
 171:     environ[n_used++] = nameval;        /* add new entry */
 172:     environ[n_used] = 0;            /* terminate list */
 173:     return (0);
 174:     }
 175: }
 176: 
 177: #ifdef TEST
 178: 
 179:  /*
 180:   * Stand-alone program for test purposes.
 181:   */
 182: 
 183: /* printenv - display environment */
 184: 
 185: static void printenv()
 186: {
 187:     char  **envp;
 188: 
 189:     for (envp = environ; envp && *envp; envp++)
 190:     printf("%s\n", *envp);
 191: }
 192: 
 193: int     main(argc, argv)
 194: int     argc;
 195: char  **argv;
 196: {
 197:     char   *cp;
 198:     int     changed = 0;
 199: 
 200:     if (argc < 2) {
 201:     printf("usage: %s name[=value]...\n", argv[0]);
 202:     return (1);
 203:     }
 204:     while (--argc && *++argv) {
 205:     if (argv[0][0] == '-') {        /* unsetenv() test */
 206:         unsetenv(argv[0] + 1);
 207:         changed = 1;
 208:     } else if (strchr(argv[0], '=') == 0) { /* getenv() test */
 209:         cp = getenv(argv[0]);
 210:         printf("%s: %s\n", argv[0], cp ? cp : "not found");
 211:     } else {                /* putenv() test */
 212:         if (putenv(argv[0])) {
 213:         perror("putenv");
 214:         return (1);
 215:         }
 216:         changed = 1;
 217:     }
 218:     }
 219:     if (changed)
 220:     printenv();
 221:     return (0);
 222: }
 223: 
 224: #endif /* TEST */

Defined functions

Defined variables

Defined macros