#if !defined(lint) && !defined(LINT)
static const char rcsid[] =
"$FreeBSD: src/usr.sbin/cron/lib/env.c,v 1.12 2003/02/10 11:20:58 thomas Exp $";
#endif
#include "cron.h"
char **
env_init()
{
register char **p = (char **) malloc(sizeof(char *));
if (p)
p[0] = NULL;
return (p);
}
void
env_free(envp)
char **envp;
{
char **p;
if ((p = envp))
for (; *p; p++)
free(*p);
free(envp);
}
char **
env_copy(envp)
register char **envp;
{
register int count, i;
register char **p;
for (count = 0; envp[count] != NULL; count++)
;
p = (char **) malloc((count+1) * sizeof(char *));
if (p == NULL) {
errno = ENOMEM;
return NULL;
}
for (i = 0; i < count; i++)
if ((p[i] = strdup(envp[i])) == NULL) {
while (--i >= 0)
(void) free(p[i]);
free(p);
errno = ENOMEM;
return NULL;
}
p[count] = NULL;
return (p);
}
char **
env_set(envp, envstr)
char **envp;
char *envstr;
{
register int count, found;
register char **p;
char *q;
found = -1;
for (count = 0; envp[count] != NULL; count++) {
if (!strcmp_until(envp[count], envstr, '='))
found = count;
}
count++;
if (found != -1) {
q = envp[found];
if ((envp[found] = strdup(envstr)) == NULL) {
envp[found] = q;
errno = ENOMEM;
return NULL;
}
free(q);
return (envp);
}
p = (char **) realloc((void *) envp,
(unsigned) ((count+1) * sizeof(char *)));
if (p == NULL) {
errno = ENOMEM;
return NULL;
}
p[count] = p[count-1];
if ((p[count-1] = strdup(envstr)) == NULL) {
env_free(p);
errno = ENOMEM;
return NULL;
}
return (p);
}
int
load_env(envstr, f)
char *envstr;
FILE *f;
{
long filepos;
int fileline;
char name[MAX_ENVSTR], val[MAX_ENVSTR];
char quotechar, *c, *str;
int state;
#define NAMEI 0
#define NAME 1
#define EQ1 2
#define EQ2 3
#define VALUEI 4
#define VALUE 5
#define FINI 6
#define ERROR 7
filepos = ftell(f);
fileline = LineNumber;
skip_comments(f);
if (EOF == get_string(envstr, MAX_ENVSTR, f, "\n"))
return (ERR);
Debug(DPARS, ("load_env, read <%s>\n", envstr));
bzero (name, sizeof name);
bzero (val, sizeof val);
str = name;
state = NAMEI;
quotechar = '\0';
c = envstr;
while (state != ERROR && *c) {
switch (state) {
case NAMEI:
case VALUEI:
if (*c == '\'' || *c == '"')
quotechar = *c++;
++state;
case NAME:
case VALUE:
if (quotechar) {
if (*c == quotechar) {
state++;
c++;
break;
}
if (state == NAME && *c == '=') {
state = ERROR;
break;
}
} else {
if (state == NAME) {
if (isspace (*c)) {
c++;
state++;
break;
}
if (*c == '=') {
state++;
break;
}
}
}
*str++ = *c++;
break;
case EQ1:
if (*c == '=') {
state++;
str = val;
quotechar = '\0';
} else {
if (!isspace (*c))
state = ERROR;
}
c++;
break;
case EQ2:
case FINI:
if (isspace (*c))
c++;
else
state++;
break;
}
}
if (state != FINI && !(state == VALUE && !quotechar)) {
Debug(DPARS, ("load_env, parse error, state = %d\n", state))
fseek(f, filepos, 0);
Set_LineNum(fileline);
return (FALSE);
}
if (state == VALUE) {
c = val + strlen (val);
while (c > val && isspace (*(c - 1)))
*(--c) = '\0';
}
if (strlen(name) + 1 + strlen(val) >= MAX_ENVSTR-1)
return (FALSE);
(void) sprintf(envstr, "%s=%s", name, val);
Debug(DPARS, ("load_env, <%s> <%s> -> <%s>\n", name, val, envstr))
return (TRUE);
}
char *
env_get(name, envp)
register char *name;
register char **envp;
{
register int len = strlen(name);
register char *p, *q;
while ((p = *envp++)) {
if (!(q = strchr(p, '=')))
continue;
if ((q - p) == len && !strncmp(p, name, len))
return (q+1);
}
return (NULL);
}