usr.sbin/cron/lib/env.c

   1 /* Copyright 1988,1990,1993,1994 by Paul Vixie
   2  * All rights reserved
   3  *
   4  * Distribute freely, except: don't remove my name from the source or
   5  * documentation (don't take credit for my work), mark your changes (don't
   6  * get me blamed for your possible bugs), don't alter or remove this
   7  * notice.  May be sold if buildable source is provided to buyer.  No
   8  * warrantee of any kind, express or implied, is included with this
   9  * software; use at your own risk, responsibility for damages (if any) to
  10  * anyone resulting from the use of this software rests entirely with the
  11  * user.
  12  *
  13  * Send bug reports, bug fixes, enhancements, requests, flames, etc., and
  14  * I'll try to keep a version up to date.  I can be reached as follows:
  15  * Paul Vixie          <paul@vix.com>          uunet!decwrl!vixie!paul
  16  */
  17
  18 #if !defined(lint) && !defined(LINT)
  19 static const char rcsid[] =
  20   "$FreeBSD: src/usr.sbin/cron/lib/env.c,v 1.7.2.1 2000/07/01 10:35:07 ps Exp $";
  21 #endif
  22
  23
  24 #include "cron.h"
  25
  26
  27 char **
  28 env_init()
  29 {
  30         register char   **p = (char **) malloc(sizeof(char *));
  31
  32         if (p)
  33                 p[0] = NULL;
  34         return (p);
  35 }
  36
  37
  38 void
  39 env_free(envp)
  40         char    **envp;
  41 {
  42         char    **p;
  43
  44         for (p = envp;  *p;  p++)
  45                 free(*p);
  46         free(envp);
  47 }
  48
  49
  50 char **
  51 env_copy(envp)
  52         register char   **envp;
  53 {
  54         register int    count, i;
  55         register char   **p;
  56
  57         for (count = 0;  envp[count] != NULL;  count++)
  58                 ;
  59         p = (char **) malloc((count+1) * sizeof(char *)); /* 1 for the NULL */
  60         if (p == NULL) {
  61                 errno = ENOMEM;
  62                 return NULL;
  63         }
  64         for (i = 0;  i < count;  i++)
  65                 if ((p[i] = strdup(envp[i])) == NULL) {
  66                         while (--i >= 0)
  67                                 (void) free(p[i]);
  68                         free(p);
  69                         errno = ENOMEM;
  70                         return NULL;
  71                 }
  72         p[count] = NULL;
  73         return (p);
  74 }
  75
  76
  77 char **
  78 env_set(envp, envstr)
  79         char    **envp;
  80         char    *envstr;
  81 {
  82         register int    count, found;
  83         register char   **p;
  84         char            *q;
  85
  86         /*
  87          * count the number of elements, including the null pointer;
  88          * also set 'found' to -1 or index of entry if already in here.
  89          */
  90         found = -1;
  91         for (count = 0;  envp[count] != NULL;  count++) {
  92                 if (!strcmp_until(envp[count], envstr, '='))
  93                         found = count;
  94         }
  95         count++;        /* for the NULL */
  96
  97         if (found != -1) {
  98                 /*
  99                  * it exists already, so just free the existing setting,
 100                  * save our new one there, and return the existing array.
 101                  */
 102                 q = envp[found];
 103                 if ((envp[found] = strdup(envstr)) == NULL) {
 104                         envp[found] = q;
 105                         /* XXX env_free(envp); */
 106                         errno = ENOMEM;
 107                         return NULL;
 108                 }
 109                 free(q);
 110                 return (envp);
 111         }
 112
 113         /*
 114          * it doesn't exist yet, so resize the array, move null pointer over
 115          * one, save our string over the old null pointer, and return resized
 116          * array.
 117          */
 118         p = (char **) realloc((void *) envp,
 119                               (unsigned) ((count+1) * sizeof(char *)));
 120         if (p == NULL)  {
 121                 /* XXX env_free(envp); */
 122                 errno = ENOMEM;
 123                 return NULL;
 124         }
 125         p[count] = p[count-1];
 126         if ((p[count-1] = strdup(envstr)) == NULL) {
 127                 env_free(p);
 128                 errno = ENOMEM;
 129                 return NULL;
 130         }
 131         return (p);
 132 }
 133
 134
 135 /* return       ERR = end of file
 136  *              FALSE = not an env setting (file was repositioned)
 137  *              TRUE = was an env setting
 138  */
 139 int
 140 load_env(envstr, f)
 141         char    *envstr;
 142         FILE    *f;
 143 {
 144         long    filepos;
 145         int     fileline;
 146         char    name[MAX_ENVSTR], val[MAX_ENVSTR];
 147         int     fields;
 148
 149         filepos = ftell(f);
 150         fileline = LineNumber;
 151         skip_comments(f);
 152         if (EOF == get_string(envstr, MAX_ENVSTR, f, "\n"))
 153                 return (ERR);
 154
 155         Debug(DPARS, ("load_env, read <%s>\n", envstr))
 156
 157         name[0] = val[0] = '\0';
 158         fields = sscanf(envstr, "%[^ =] = %[^\n#]", name, val);
 159         if (fields != 2) {
 160                 Debug(DPARS, ("load_env, not 2 fields (%d)\n", fields))
 161                 fseek(f, filepos, 0);
 162                 Set_LineNum(fileline);
 163                 return (FALSE);
 164         }
 165
 166         /* 2 fields from scanf; looks like an env setting
 167          */
 168
 169         /*
 170          * process value string
 171          */
 172         /*local*/{
 173                 int     len = strdtb(val);
 174
 175                 if (len >= 2) {
 176                         if (val[0] == '\'' || val[0] == '"') {
 177                                 if (val[len-1] == val[0]) {
 178                                         val[len-1] = '\0';
 179                                         (void) strcpy(val, val+1);
 180                                 }
 181                         }
 182                 }
 183         }
 184
 185         if (strlen(name) + 1 + strlen(val) >= MAX_ENVSTR-1)
 186                 return (FALSE);
 187         (void) sprintf(envstr, "%s=%s", name, val);
 188         Debug(DPARS, ("load_env, <%s> <%s> -> <%s>\n", name, val, envstr))
 189         return (TRUE);
 190 }
 191
 192
 193 char *
 194 env_get(name, envp)
 195         register char   *name;
 196         register char   **envp;
 197 {
 198         register int    len = strlen(name);
 199         register char   *p, *q;
 200
 201         while ((p = *envp++)) {
 202                 if (!(q = strchr(p, '=')))
 203                         continue;
 204                 if ((q - p) == len && !strncmp(p, name, len))
 205                         return (q+1);
 206         }
 207         return (NULL);
 208 }