usr.sbin/installer/libinstaller/survey.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  * survey.c
  36  * Survey the storage capacity of the system.
  37  * $Id: survey.c,v 1.17 2005/02/06 21:05:18 cpressey Exp $
  38  */
  39
  40 #include <sys/types.h>
  41 #include <sys/sysctl.h>
  42
  43 #include <stdio.h>
  44 #include <string.h>
  45
  46 #include "libaura/dict.h"
  47
  48 #include "commands.h"
  49 #include "diskutil.h"
  50 #include "functions.h"
  51
  52 static int      fgets_chomp(char *, int, FILE *);
  53 static int      parse_geometry_info(char *, int *, int *, int *);
  54 static int      parse_slice_info(char *, int *,
  55                     unsigned long *, unsigned long *, int *, int *);
  56
  57 /*
  58  * Get a line from a file.  Remove any trailing EOL's.
  59  * Return 1 if we did not hit EOF, 0 if we did.
  60  */
  61 static int
  62 fgets_chomp(char *line, int size, FILE *f)
  63 {
  64         if (fgets(line, size, f) == NULL)
  65                 return(0);
  66         while (strlen(line) > 0 && line[strlen(line) - 1] == '\n')
  67                 line[strlen(line) - 1] = '\0';
  68         return(1);
  69 }
  70
  71 /*
  72  * Given a geometry line from fdisk's summary output, return the
  73  * number of cylinders, heads, and sectors.
  74  */
  75 static int
  76 parse_geometry_info(char *line, int *cyl, int *head, int *sec)
  77 {
  78         char *word;
  79
  80         /*
  81          * /dev/ad3: 2112 cyl 16 hd 63 sec
  82          */
  83         if ((word = strtok(line, " \t")) == NULL)       /* /dev/ad3: */
  84                 return(0);
  85         if ((word = strtok(NULL, " \t")) == NULL)       /* 2112 */
  86                 return(0);
  87         *cyl = atoi(word);
  88         if ((word = strtok(NULL, " \t")) == NULL)       /* cyl */
  89                 return(0);
  90         if ((word = strtok(NULL, " \t")) == NULL)       /* 16 */
  91                 return(0);
  92         *head = atoi(word);
  93         if ((word = strtok(NULL, " \t")) == NULL)       /* hd */
  94                 return(0);
  95         if ((word = strtok(NULL, " \t")) == NULL)       /* 63 */
  96                 return(0);
  97         *sec = atoi(word);
  98
  99         return(1);
 100 }
 101
 102 /*
 103  * Given a slice description line from fdisk's summary output, return
 104  * the number of the slice, and its start, size, type, and flags.
 105  */
 106 static int
 107 parse_slice_info(char *line, int *slice,
 108                  unsigned long *start, unsigned long *size,
 109                  int *type, int *flags)
 110 {
 111         char *word;
 112
 113         /*
 114          * Part        Start        Size Type Flags
 115          *    1:          63     2128833 0xa5 0x80
 116          */
 117         if ((word = strtok(line, " \t")) == NULL)       /* 1: */
 118                 return(0);
 119         *slice = atoi(word);
 120         if ((word = strtok(NULL, " \t")) == NULL)       /* 63 */
 121                 return(0);
 122         *start = strtoul(word, NULL, 10);
 123         if ((word = strtok(NULL, " \t")) == NULL)       /* 2128833 */
 124                 return(0);
 125         *size = strtoul(word, NULL, 10);
 126         if ((word = strtok(NULL, " \t")) == NULL)       /* 0xa5 */
 127                 return(0);
 128         if (!hex_to_int(word, type))
 129                 return(0);
 130         if ((word = strtok(NULL, " \t")) == NULL)       /* 0x80 */
 131                 return(0);
 132         if (!hex_to_int(word, flags))
 133                 return(0);
 134
 135         return(1);
 136 }
 137
 138 /*
 139  * Survey storage capacity of this system.
 140  */
 141 int
 142 survey_storage(struct i_fn_args *a)
 143 {
 144         unsigned long mem = 0;
 145         char disks[256], line[256];
 146         char *disk, *disk_ptr;
 147         struct commands *cmds;
 148         struct command *cmd;
 149         FILE *f;
 150         char *filename;
 151         struct disk *d = NULL;
 152         int failure = 0;
 153         size_t len;
 154         struct aura_dict *di;
 155         void *rk;
 156         size_t rk_len;
 157
 158         disks_free(a->s);
 159
 160         len = sizeof(mem);
 161         if (sysctlbyname("hw.physmem", &mem, &len, NULL, 0) < 0) {
 162                 failure |= 1;
 163         } else {
 164                 storage_set_memsize(a->s, next_power_of_two(mem >> 20));
 165         }
 166         len = 256;
 167         if (sysctlbyname("kern.disks", disks, &len, NULL, 0) < 0) {
 168                 failure |= 1;
 169         }
 170         disk_ptr = disks;
 171
 172         di = aura_dict_new(1, AURA_DICT_SORTED_LIST);
 173         while (!failure && (disk = strsep(&disk_ptr, " ")) != NULL) {
 174                 if (disk[0] == '\0')
 175                         continue;
 176
 177                 /*
 178                  * If the disk is a memory disk, floppy or CD-ROM, skip it.
 179                  */
 180                 if (strncmp(disk, "md", 2) == 0 ||
 181                     strncmp(disk, "cd", 2) == 0 ||
 182                     strncmp(disk, "acd", 3) == 0 ||
 183                     strncmp(disk, "fd", 2) == 0)
 184                         continue;
 185
 186                 aura_dict_store(di, disk, strlen(disk) + 1, "", 1);
 187         }
 188
 189         cmds = commands_new();
 190         cmd = command_add(cmds, "%s%s -n '' >%ssurvey.txt",
 191             a->os_root, cmd_name(a, "ECHO"), a->tmp);
 192         command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 193
 194         aura_dict_rewind(di);
 195         while (!aura_dict_eof(di)) {
 196                 aura_dict_get_current_key(di, &rk, &rk_len),
 197
 198                 disk = (char *)rk;
 199
 200                 cmd = command_add(cmds, "%s%s '@DISK' >>%ssurvey.txt",
 201                     a->os_root, cmd_name(a, "ECHO"), a->tmp);
 202                 command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 203                 cmd = command_add(cmds, "%s%s '%s' >>%ssurvey.txt",
 204                     a->os_root, cmd_name(a, "ECHO"), disk, a->tmp);
 205                 command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 206
 207                 /*
 208                  * Look for descriptions of this disk.
 209                  */
 210                 cmd = command_add(cmds, "%s%s '@DESC' >>%ssurvey.txt",
 211                     a->os_root, cmd_name(a, "ECHO"), a->tmp);
 212                 command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 213                 cmd = command_add(cmds, "%s%s -w '^%s: [0-9]*MB' %s%s >>%ssurvey.txt || %s%s '%s' >>%ssurvey.txt",
 214                     a->os_root, cmd_name(a, "GREP"),
 215                     disk,
 216                     a->os_root, cmd_name(a, "DMESG_BOOT"),
 217                     a->tmp,
 218                     a->os_root, cmd_name(a, "ECHO"),
 219                     disk,
 220                     a->tmp);
 221                 cmd = command_add(cmds, "%s%s '@END' >>%ssurvey.txt",
 222                     a->os_root, cmd_name(a, "ECHO"), a->tmp);
 223                 command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 224
 225                 /*
 226                  * Probe the disk with fdisk.
 227                  */
 228                 cmd = command_add(cmds, "%s%s '@SLICES' >>%ssurvey.txt",
 229                     a->os_root, cmd_name(a, "ECHO"), a->tmp);
 230                 command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 231                 cmd = command_add(cmds, "%s%s -s %s 2>/dev/null >>%ssurvey.txt || %s%s '' >>%ssurvey.txt",
 232                     a->os_root, cmd_name(a, "FDISK"),
 233                     disk,
 234                     a->tmp,
 235                     a->os_root, cmd_name(a, "ECHO"),
 236                     a->tmp);
 237                 cmd = command_add(cmds, "%s%s '@END' >>%ssurvey.txt",
 238                     a->os_root, cmd_name(a, "ECHO"), a->tmp);
 239                 command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 240
 241                 aura_dict_next(di);
 242         }
 243
 244         cmd = command_add(cmds, "%s%s '.' >>%ssurvey.txt",
 245             a->os_root, cmd_name(a, "ECHO"), a->tmp);
 246         command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 247
 248         if (!commands_execute(a, cmds))
 249                 failure |= 1;
 250         commands_free(cmds);
 251         temp_file_add(a, "survey.txt");
 252
 253         aura_dict_free(di);
 254
 255         /*
 256          * Now read in and parse the file that those commands just created.
 257          */
 258         asprintf(&filename, "%ssurvey.txt", a->tmp);
 259         if ((f = fopen(filename, "r")) == NULL)
 260                 failure |= 1;
 261         free(filename);
 262
 263         while (!failure && fgets_chomp(line, 255, f)) {
 264                 if (strcmp(line, "@DISK") == 0) {
 265                         if (fgets_chomp(line, 255, f)) {
 266                                 d = disk_new(a->s, line);
 267                         }
 268                 } else if (strcmp(line, "@DESC") == 0) {
 269                         while (d != NULL && strcmp(line, "@END") != 0 && fgets_chomp(line, 255, f)) {
 270                                 disk_set_desc(d, line);
 271                         }
 272                 } else if (strcmp(line, "@SLICES") == 0) {
 273                         int cyl, hd, sec;
 274                         int number, type, flags;
 275                         unsigned long start, size;
 276
 277                         /*
 278                          * /dev/ad3: 2112 cyl 16 hd 63 sec
 279                          * Part        Start        Size Type Flags
 280                          *    1:          63     2128833 0xa5 0x80
 281                          */
 282                         while (d != NULL && strcmp(line, "@END") != 0 && fgets_chomp(line, 255, f)) {
 283                                 if (strncmp(line, "/dev/", 5) == 0) {
 284                                         parse_geometry_info(line, &cyl, &hd, &sec);
 285                                         disk_set_geometry(d, cyl, hd, sec);
 286                                 } else if (strncmp(line, "Part", 4) == 0) {
 287                                         /* ignore it */
 288                                 } else {
 289                                         if (parse_slice_info(line, &number, &start, &size,
 290                                             &type, &flags)) {
 291                                                 /*
 292                                                 fprintfo(log, "| Found slice #%d, sysid %d, "
 293                                                     "start %ld, size %ld\n", number, type, start, size);
 294                                                 */
 295                                                 slice_new(d, number, type, flags, start, size);
 296                                         }
 297                                 }
 298                         }
 299                 }
 300         }
 301
 302         if (f != NULL)
 303                 fclose(f);
 304
 305         /*
 306          * Fix up any disk descriptions that didn't make it.
 307          */
 308         for (d = storage_disk_first(a->s); d != NULL; d = disk_next(d)) {
 309                 if (disk_get_desc(d) == NULL)
 310                         disk_set_desc(d, disk_get_device_name(d));
 311         }
 312
 313         return(!failure);
 314 }