Return to setenv.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / sudo / compat

Annotation of embedaddon/sudo/compat/setenv.c, revision 1.1.1.1

1.1       misho       1: /*
                      2:  * Copyright (c) 2010 Todd C. Miller <Todd.Miller@courtesan.com>
                      3:  *
                      4:  * Permission to use, copy, modify, and distribute this software for any
                      5:  * purpose with or without fee is hereby granted, provided that the above
                      6:  * copyright notice and this permission notice appear in all copies.
                      7:  *
                      8:  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
                      9:  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
                     10:  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
                     11:  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
                     12:  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
                     13:  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
                     14:  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
                     15:  */
                     16: 
                     17: #include <config.h>
                     18: 
                     19: #include <sys/types.h>
                     20: 
                     21: #include <stdio.h>
                     22: #ifdef STDC_HEADERS
                     23: # include <stdlib.h>
                     24: # include <stddef.h>
                     25: #else
                     26: # ifdef HAVE_STDLIB_H
                     27: #  include <stdlib.h>
                     28: # endif
                     29: #endif /* STDC_HEADERS */
                     30: #ifdef HAVE_STRING_H
                     31: # include <string.h>
                     32: #endif /* HAVE_STRING_H */
                     33: #ifdef HAVE_STRINGS_H
                     34: # include <strings.h>
                     35: #endif /* HAVE_STRINGS_H */
                     36: #if defined(HAVE_MALLOC_H) && !defined(STDC_HEADERS)
                     37: # include <malloc.h>
                     38: #endif /* HAVE_MALLOC_H && !STDC_HEADERS */
                     39: #include <errno.h>
                     40: 
                     41: #include "missing.h"
                     42: 
                     43: int
                     44: setenv(const char *var, const char *val, int overwrite)
                     45: {
                     46:     char *envstr, *dst;
                     47:     const char *src;
                     48:     size_t esize;
                     49: 
                     50:     if (!var || *var == '\0') {
                     51:        errno = EINVAL;
                     52:        return -1;
                     53:     }
                     54: 
                     55:     /*
                     56:      * POSIX says a var name with '=' is an error but BSD
                     57:      * just ignores the '=' and anything after it.
                     58:      */
                     59:     for (src = var; *src != '\0' && *src != '='; src++)
                     60:        ;
                     61:     esize = (size_t)(src - var) + 2;
                     62:     if (val) {
                     63:         esize += strlen(val);  /* glibc treats a NULL val as "" */
                     64:     }
                     65: 
                     66:     /* Allocate and fill in envstr. */
                     67:     if ((envstr = malloc(esize)) == NULL)
                     68:        return -1;
                     69:     for (src = var, dst = envstr; *src != '\0' && *src != '=';)
                     70:        *dst++ = *src++;
                     71:     *dst++ = '=';
                     72:     if (val) {
                     73:        for (src = val; *src != '\0';)
                     74:            *dst++ = *src++;
                     75:     }
                     76:     *dst = '\0';
                     77: 
                     78:     if (!overwrite && getenv(var) != NULL) {
                     79:        free(envstr);
                     80:        return 0;
                     81:     }
                     82:     return putenv(envstr);
                     83: }