1 /*
2 * Copyright (c)2004 The DragonFly Project. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 *
8 * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 *
11 * Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in
13 * the documentation and/or other materials provided with the
14 * distribution.
15 *
16 * Neither the name of the DragonFly Project nor the names of its
17 * contributors may be used to endorse or promote products derived
18 * from this software without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
23 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
24 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
25 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
26 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
27 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
29 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
31 * OF THE POSSIBILITY OF SUCH DAMAGE.
32 */
33
34 /*
35 * confed.c
36 * Functions for working with configuration files.
37 * Inspired by (but not derived from) sysinstall's variable.c
38 * $Id: confed.c,v 1.16 2005/02/06 21:05:18 cpressey Exp $
39 */
40
41 #include <sys/stat.h>
42
43 #include <ctype.h>
44 #include <stdarg.h>
45 #include <stdio.h>
46 #include <stdlib.h>
47 #include <string.h>
48 #include <time.h>
49 #include <unistd.h>
50
51 #include "libaura/mem.h"
52 #include "libaura/dict.h"
53 #include "libdfui/system.h"
54
55 #include "confed.h"
56 #include "commands.h"
57 #include "functions.h"
58
59 /*
60 * Create a new, empty set of in-memory config variable settings.
61 */
62 struct config_vars *
config_vars_new(void)63 config_vars_new(void)
64 {
65 struct config_vars *cvs;
66
67 AURA_MALLOC(cvs, config_vars);
68
69 cvs->d = aura_dict_new(1, AURA_DICT_SORTED_LIST);
70
71 return(cvs);
72 }
73
74 /*
75 * Deallocate the memory used by a set of config variable settings.
76 */
77 void
config_vars_free(struct config_vars * cvs)78 config_vars_free(struct config_vars *cvs)
79 {
80 if (cvs == NULL)
81 return;
82
83 aura_dict_free(cvs->d);
84
85 AURA_FREE(cvs, config_vars);
86 }
87
88 /*
89 * Get the value of a configuration variable in a set of settings
90 * and return it, or a (constant) 0-length string if not found.
91 */
92 const char *
config_var_get(const struct config_vars * cvs,const char * name)93 config_var_get(const struct config_vars *cvs, const char *name)
94 {
95 void *rv;
96 size_t rv_len;
97
98 aura_dict_fetch(cvs->d, name, strlen(name) + 1, &rv, &rv_len);
99 if (rv == NULL)
100 return("");
101 else
102 return(rv);
103 }
104
105 /*
106 * Set the value of a configuration variable. If the named variable
107 * does not exist within the given set, a new one is created.
108 */
109 int
config_var_set(struct config_vars * cvs,const char * name,const char * value)110 config_var_set(struct config_vars *cvs, const char *name, const char *value)
111 {
112 aura_dict_store(cvs->d,
113 name, strlen(name) + 1,
114 value, strlen(value) + 1);
115 return(1);
116 }
117
118 /*
119 * Write a set of configuration variable settings to a file.
120 */
121 int
config_vars_write(const struct config_vars * cvs,int config_type,const char * fmt,...)122 config_vars_write(const struct config_vars *cvs, int config_type,
123 const char *fmt, ...)
124 {
125 FILE *f;
126 va_list args;
127 char *filename;
128 void *rk, *rv;
129 size_t rk_len, rv_len;
130
131 va_start(args, fmt);
132 vasprintf(&filename, fmt, args);
133 va_end(args);
134
135 if ((f = fopen(filename, "a")) == NULL)
136 return(0);
137
138 switch (config_type) {
139 case CONFIG_TYPE_SH:
140
141 aura_dict_rewind(cvs->d);
142 while (!aura_dict_eof(cvs->d)) {
143 aura_dict_get_current_key(cvs->d, &rk, &rk_len),
144 aura_dict_fetch(cvs->d, rk, rk_len, &rv, &rv_len);
145 fprintf(f, "%s=\"%s\"\t# via installer configuration\n",
146 (char *)rk, (char *)rv);
147 aura_dict_next(cvs->d);
148 }
149 break;
150 case CONFIG_TYPE_RESOLV:
151 aura_dict_rewind(cvs->d);
152 while (!aura_dict_eof(cvs->d)) {
153 aura_dict_get_current_key(cvs->d, &rk, &rk_len),
154 aura_dict_fetch(cvs->d, rk, rk_len, &rv, &rv_len);
155 fprintf(f, "%s\t\t%s\n", (char *)rk, (char *)rv);
156 aura_dict_next(cvs->d);
157 }
158 break;
159 default:
160 fclose(f);
161 return(0);
162 }
163
164 fclose(f);
165 return(1);
166 }
167
168 /*
169 * Read variables from a file.
170 * Returns 1 if the variables could be read successfully, 0 if not.
171 */
172 int
config_vars_read(struct i_fn_args * a,struct config_vars * cvs,int config_type __unused,const char * fmt,...)173 config_vars_read(struct i_fn_args *a, struct config_vars *cvs,
174 int config_type __unused, const char *fmt, ...)
175 {
176 struct commands *cmds;
177 char *filename, *tmp_filename, line[1024], *value;
178 FILE *f, *script;
179 va_list args;
180
181 va_start(args, fmt);
182 vasprintf(&filename, fmt, args);
183 va_end(args);
184
185 asprintf(&tmp_filename, "%sextract_vars", a->tmp);
186 script = fopen(tmp_filename, "w");
187 free(tmp_filename);
188 if (script == NULL)
189 return(0);
190
191 fprintf(script, "set | %susr/bin/sort >%senv.before\n", a->os_root, a->tmp);
192 fprintf(script, ". %s%s\n", a->os_root, filename);
193 fprintf(script, "set | %susr/bin/sort >%senv.after\n", a->os_root, a->tmp);
194 fprintf(script, "%susr/bin/comm -1 -3 %senv.before %senv.after | \\\n",
195 a->os_root, a->tmp, a->tmp);
196 fprintf(script, " %susr/bin/awk -F= '{ print $1 }' | \\\n", a->os_root);
197 fprintf(script, " while read __VARNAME; do\n");
198 fprintf(script, " echo -n ${__VARNAME}=\n");
199 fprintf(script, " eval echo ' $'${__VARNAME}\n");
200 fprintf(script, " done\n");
201 fprintf(script, "%sbin/rm -f %senv.before %senv.after\n",
202 a->os_root, a->tmp, a->tmp);
203 fclose(script);
204
205 cmds = commands_new();
206 command_add(cmds, "%sbin/sh %sextract_vars >%sextracted_vars.txt",
207 a->os_root, a->tmp, a->tmp);
208 temp_file_add(a, "extracted_vars.txt");
209 if (!commands_execute(a, cmds)) {
210 commands_free(cmds);
211 return(0);
212 }
213 commands_free(cmds);
214
215 /*
216 * Delete the script immediately.
217 */
218 asprintf(&tmp_filename, "%sextract_vars", a->tmp);
219 (void)unlink(tmp_filename); /* not much we can do if it fails */
220 free(tmp_filename);
221
222 asprintf(&tmp_filename, "%sextracted_vars.txt", a->tmp);
223 f = fopen(tmp_filename, "r");
224 free(tmp_filename);
225 if (f == NULL)
226 return(0);
227 while (fgets(line, 1024, f) != NULL) {
228 if (strlen(line) > 0)
229 line[strlen(line) - 1] = '\0';
230 /* split line at first = */
231 for (value = line; *value != '=' && *value != '\0'; value++)
232 ;
233 if (*value == '\0')
234 break;
235 *value = '\0';
236 value++;
237 config_var_set(cvs, line, value);
238 }
239 fclose(f);
240
241 return(1);
242 }
243