1/*	$NetBSD$	*/
2
3/*
4 * Copyright (c) 2000, 2005 Kungliga Tekniska H��gskolan
5 * (Royal Institute of Technology, Stockholm, Sweden).
6 * All rights reserved.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 *
12 * 1. Redistributions of source code must retain the above copyright
13 *    notice, this list of conditions and the following disclaimer.
14 *
15 * 2. Redistributions in binary form must reproduce the above copyright
16 *    notice, this list of conditions and the following disclaimer in the
17 *    documentation and/or other materials provided with the distribution.
18 *
19 * 3. Neither the name of the Institute nor the names of its contributors
20 *    may be used to endorse or promote products derived from this software
21 *    without specific prior written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
24 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26 * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
27 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33 * SUCH DAMAGE.
34 */
35
36
37#include <config.h>
38
39#include <stdio.h>
40#include <string.h>
41#include <ctype.h>
42#include <krb5/roken.h>
43
44/* find assignment in env list; len is length of variable including
45 * equal
46 */
47
48static int
49find_var(char **env, char *assignment, size_t len)
50{
51    int i;
52    for(i = 0; env != NULL && env[i] != NULL; i++)
53	if(strncmp(env[i], assignment, len) == 0)
54	    return i;
55    return -1;
56}
57
58/*
59 * return count of environment assignments from open file F in
60 * assigned and list of malloced strings in env, return 0 or errno
61 * number
62 */
63
64static int
65read_env_file(FILE *F, char ***env, int *assigned)
66{
67    int idx = 0;
68    int i;
69    char **l;
70    char buf[BUFSIZ], *p, *r;
71    char **tmp;
72    int ret = 0;
73
74    *assigned = 0;
75
76    for(idx = 0; *env != NULL && (*env)[idx] != NULL; idx++);
77    l = *env;
78
79    /* This is somewhat more relaxed on what it accepts then
80     * Wietses sysv_environ from K4 was...
81     */
82    while (fgets(buf, BUFSIZ, F) != NULL) {
83	buf[strcspn(buf, "#\n")] = '\0';
84
85	for(p = buf; isspace((unsigned char)*p); p++);
86	if (*p == '\0')
87	    continue;
88
89	/* Here one should check that it's a 'valid' env string... */
90	r = strchr(p, '=');
91	if (r == NULL)
92	    continue;
93
94	if((i = find_var(l, p, r - p + 1)) >= 0) {
95	    char *val = strdup(p);
96	    if(val == NULL) {
97		ret = ENOMEM;
98		break;
99	    }
100	    free(l[i]);
101	    l[i] = val;
102	    (*assigned)++;
103	    continue;
104	}
105
106	tmp = realloc(l, (idx+2) * sizeof (char *));
107	if(tmp == NULL) {
108	    ret = ENOMEM;
109	    break;
110	}
111
112	l = tmp;
113	l[idx] = strdup(p);
114	if(l[idx] == NULL) {
115	    ret = ENOMEM;
116	    break;
117	}
118	l[++idx] = NULL;
119	(*assigned)++;
120    }
121    if(ferror(F))
122	ret = errno;
123    *env = l;
124    return ret;
125}
126
127/*
128 * return count of environment assignments from file and
129 * list of malloced strings in `env'
130 */
131
132ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
133read_environment(const char *file, char ***env)
134{
135    int assigned;
136    FILE *F;
137
138    if ((F = fopen(file, "r")) == NULL)
139	return 0;
140
141    read_env_file(F, env, &assigned);
142    fclose(F);
143    return assigned;
144}
145
146ROKEN_LIB_FUNCTION void ROKEN_LIB_CALL
147free_environment(char **env)
148{
149    int i;
150    if (env == NULL)
151	return;
152    for (i = 0; env[i]; i++)
153	free(env[i]);
154    free(env);
155}
156