usr.sbin/installer/libinstaller/confed.c

   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 *
  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
  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 *
  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
 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
 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
 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 }