git.gag.com Git - debian/sudo/blob - compat/setenv.c

   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 }