1 /* $NetBSD: environment.c,v 1.1.1.1 2011/04/13 18:15:40 elric Exp $ */
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
48 static int
find_var(char ** env,char * assignment,size_t len)49 find_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
64 static int
read_env_file(FILE * F,char *** env,int * assigned)65 read_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
132 ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
read_environment(const char * file,char *** env)133 read_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
146 ROKEN_LIB_FUNCTION void ROKEN_LIB_CALL
free_environment(char ** env)147 free_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