2 * Copyright (c)2004 The DragonFly Project. All rights reserved.
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
8 * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
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
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.
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.
36 * Generic functions for installer.
37 * $Id: functions.c,v 1.22 2005/02/06 21:05:18 cpressey Exp $
47 #include "libaura/mem.h"
48 #include "libaura/dict.h"
50 #include "libdfui/dfui.h"
52 #include "functions.h"
56 /*** INSTALLER CONTEXT CONSTRUCTOR ***/
59 i_fn_args_new(const char *os_root, const char *def_tmp_dir, int transport, const char *rendezvous)
64 AURA_MALLOC(a, i_fn_args);
67 a->os_root = aura_strdup(os_root);
79 asprintf(&filename, "%sinstall.log", def_tmp_dir);
80 a->log = fopen(filename, "w");
87 i_log(a, "Installer started");
88 i_log(a, "-----------------");
90 i_log(a, "+ Creating DFUI connection on ``%s''\n", rendezvous);
92 if ((a->c = dfui_connection_new(transport, rendezvous)) == NULL) {
93 i_log(a, "! ERROR: Couldn't create connection on ``%s''\n", rendezvous);
98 i_log(a, "+ Connecting on ``%s''\n", rendezvous);
100 if (!dfui_be_start(a->c)) {
101 i_log(a, "! ERROR: Couldn't connect to frontend on ``%s''\n", rendezvous);
106 if ((a->s = storage_new()) == NULL) {
107 i_log(a, "! ERROR: Couldn't create storage descriptor");
112 a->tmp = def_tmp_dir; /* XXX temporarily set to this */
113 a->temp_files = aura_dict_new(23, AURA_DICT_HASH);
114 a->cmd_names = config_vars_new();
115 if (!config_vars_read(a, a->cmd_names, CONFIG_TYPE_SH,
116 "usr/share/installer/cmdnames.conf")) {
117 i_log(a, "! ERROR: Couldn't read cmdnames config file");
122 a->tmp = cmd_name(a, "INSTALLER_TEMP");
124 i_log(a, "+ Starting installer state machine");
130 i_fn_args_free(struct i_fn_args *a)
133 if (a->temp_files != NULL) {
135 aura_dict_free(a->temp_files);
137 if (a->cmd_names != NULL) {
138 config_vars_free(a->cmd_names);
146 if (a->log != NULL) {
149 AURA_FREE(a, i_fn_args);
153 /*** INSTALLER CONTEXT FUNCTIONS ***/
156 i_log(struct i_fn_args *a, const char *fmt, ...)
161 vfprintf(stderr, fmt, args);
162 fprintf(stderr, "\n");
163 if (a->log != NULL) {
164 vfprintf(a->log, fmt, args);
165 fprintf(a->log, "\n");
180 assert_clean(struct dfui_connection *c, const char *name, const char *field,
181 const char *not_allowed)
183 if (strpbrk(field, not_allowed) != NULL) {
184 inform(c, "The %s field may not contain any of the "
185 "following characters:\n\n%s",
194 * Expects a leading 0x.
197 hex_to_int(const char *hex, int *result)
202 if (strncmp(hex, "0x", 2) != 0)
205 for (i = 2; hex[i] != '\0'; i++) {
210 a = a * 16 + (d - '0');
211 else if (d >= 'A' && d <= 'F')
212 a = a * 16 + (d + 10 - 'A');
222 first_non_space_char_is(const char *line, char x)
226 for (i = 0; line[i] != '\0'; i++) {
227 if (isspace(line[i]))
238 capacity_to_string(long capacity)
240 static char string[256];
243 strlcpy(string, "*", 2);
245 snprintf(string, 255, "%ldM", capacity);
251 string_to_capacity(const char *string, long *capacity)
255 unit = string[strlen(string) - 1];
256 if (!strcmp(string, "*")) {
259 } else if (unit == 'm' || unit == 'M') {
260 *capacity = strtol(string, NULL, 10);
262 } else if (unit == 'g' || unit == 'G') {
263 *capacity = strtol(string, NULL, 10) * 1024;
271 * Round a number up to the nearest power of two.
274 next_power_of_two(unsigned long n)
280 while (p < n && p > op) {
285 return(p > op ? p : n);
289 * Returns file name without extension.
291 * ru.koi8-r.kbd -> ru.koi8-r
294 * Caller is responsible for freeing the string returned.
297 filename_noext(const char *filename)
302 buffer = aura_strdup(filename);
304 if (strlen(filename) == 0) {
309 p = strrchr(filename, '.');
312 i = strlen(filename) - strlen(p);
324 temp_file_add(struct i_fn_args *a, const char *filename)
326 aura_dict_store(a->temp_files, filename, strlen(filename) + 1, "", 1);
331 temp_files_clean(struct i_fn_args *a)
337 aura_dict_rewind(a->temp_files);
338 while (!aura_dict_eof(a->temp_files)) {
339 aura_dict_get_current_key(a->temp_files, &rk, &rk_len);
340 asprintf(&filename, "%s%s", a->tmp, (char *)rk);
341 (void)unlink(filename); /* not much we can do if it fails */
343 aura_dict_next(a->temp_files);
352 cmd_name(const struct i_fn_args *a, const char *cmd_key)
356 name = config_var_get(a->cmd_names, cmd_key);
357 if (strcmp(name, "") == 0)
358 return("bin/echo"); /* XXX usr/local/sbin/error? */