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/param.h>
  41 #include <sys/diskslice.h>
  42 #include <sys/sysctl.h>
  43
  44 #include <fcntl.h>
  45 #include <stdio.h>
  46 #include <string.h>
  47 #include <unistd.h>
  48
  49 #include "libaura/dict.h"
  50
  51 #include "commands.h"
  52 #include "diskutil.h"
  53 #include "functions.h"
  54
  55 static int      fgets_chomp(char *, int, FILE *);
  56 static int      parse_geometry_info(char *, int *, int *, int *);
  57 static int      parse_slice_info(char *, int *,
  58                     unsigned long *, unsigned long *, int *, int *);
  59
  60 /*
  61  * Get a line from a file.  Remove any trailing EOL's.
  62  * Return 1 if we did not hit EOF, 0 if we did.
  63  */
  64 static int
  65 fgets_chomp(char *line, int size, FILE *f)
  66 {
  67         if (fgets(line, size, f) == NULL)
  68                 return(0);
  69         while (strlen(line) > 0 && line[strlen(line) - 1] == '\n')
  70                 line[strlen(line) - 1] = '\0';
  71         return(1);
  72 }
  73
  74 /*
  75  * Given a geometry line from fdisk's summary output, return the
  76  * number of cylinders, heads, and sectors.
  77  */
  78 static int
  79 parse_geometry_info(char *line, int *cyl, int *head, int *sec)
  80 {
  81         char *word;
  82
  83         /*
  84          * /dev/ad3: 2112 cyl 16 hd 63 sec
  85          */
  86         if ((word = strtok(line, " \t")) == NULL)       /* /dev/ad3: */
  87                 return(0);
  88         if ((word = strtok(NULL, " \t")) == NULL)       /* 2112 */
  89                 return(0);
  90         *cyl = atoi(word);
  91         if ((word = strtok(NULL, " \t")) == NULL)       /* cyl */
  92                 return(0);
  93         if ((word = strtok(NULL, " \t")) == NULL)       /* 16 */
  94                 return(0);
  95         *head = atoi(word);
  96         if ((word = strtok(NULL, " \t")) == NULL)       /* hd */
  97                 return(0);
  98         if ((word = strtok(NULL, " \t")) == NULL)       /* 63 */
  99                 return(0);
 100         *sec = atoi(word);
 101
 102         return(1);
 103 }
 104
 105 /*
 106  * Given a slice description line from fdisk's summary output, return
 107  * the number of the slice, and its start, size, type, and flags.
 108  */
 109 static int
 110 parse_slice_info(char *line, int *slice,
 111                  unsigned long *start, unsigned long *size,
 112                  int *type, int *flags)
 113 {
 114         char *word;
 115
 116         /*
 117          * Part        Start        Size Type Flags
 118          *    1:          63     2128833 0x6c 0x80
 119          */
 120         if ((word = strtok(line, " \t")) == NULL)       /* 1: */
 121                 return(0);
 122         *slice = atoi(word);
 123         if ((word = strtok(NULL, " \t")) == NULL)       /* 63 */
 124                 return(0);
 125         *start = strtoul(word, NULL, 10);
 126         if ((word = strtok(NULL, " \t")) == NULL)       /* 2128833 */
 127                 return(0);
 128         *size = strtoul(word, NULL, 10);
 129         if ((word = strtok(NULL, " \t")) == NULL)       /* 0x6c */
 130                 return(0);
 131         if (!hex_to_int(word, type))
 132                 return(0);
 133         if ((word = strtok(NULL, " \t")) == NULL)       /* 0x80 */
 134                 return(0);
 135         if (!hex_to_int(word, flags))
 136                 return(0);
 137
 138         return(1);
 139 }
 140
 141 /*
 142  * Survey storage capacity of this system.
 143  */
 144 int
 145 survey_storage(struct i_fn_args *a)
 146 {
 147         unsigned long mem = 0;
 148         char disks[256], line[256];
 149         char *disk, *disk_ptr;
 150         struct commands *cmds;
 151         struct command *cmd;
 152         FILE *f;
 153         char *filename;
 154         struct disk *d = NULL;
 155         int number = 0;
 156         int failure = 0;
 157         int ndisks = 0;
 158         int fd;
 159         size_t len;
 160         struct aura_dict *di;
 161         void *rk;
 162         size_t rk_len;
 163         struct partinfo diskpart;
 164         char diskpath[PATH_MAX];
 165
 166         sleep(1);               /* give devfs time to probe */
 167         disks_free(a->s);
 168
 169         len = sizeof(mem);
 170         if (sysctlbyname("hw.physmem", &mem, &len, NULL, 0) < 0) {
 171                 failure |= 1;
 172         } else {
 173                 storage_set_memsize(a->s, next_power_of_two(mem >> 20));
 174         }
 175         len = 256;
 176         if (sysctlbyname("kern.disks", disks, &len, NULL, 0) < 0) {
 177                 failure |= 1;
 178         }
 179         disk_ptr = disks;
 180
 181         di = aura_dict_new(1, AURA_DICT_SORTED_LIST);
 182         while (!failure && (disk = strsep(&disk_ptr, " ")) != NULL) {
 183                 if (disk[0] == '\0')
 184                         continue;
 185
 186                 /*
 187                  * If the disk is a memory disk, floppy or CD-ROM, skip it.
 188                  */
 189                 if (strncmp(disk, "md", 2) == 0 ||
 190                     strncmp(disk, "cd", 2) == 0 ||
 191                     strncmp(disk, "acd", 3) == 0 ||
 192                     strncmp(disk, "fd", 2) == 0)
 193                         continue;
 194
 195                 aura_dict_store(di, disk, strlen(disk) + 1, "", 1);
 196                 ndisks++;
 197         }
 198
 199         if (ndisks == 0)
 200                 failure |= 1;
 201
 202         cmds = commands_new();
 203         cmd = command_add(cmds, "%s%s -n '' >%ssurvey.txt",
 204             a->os_root, cmd_name(a, "ECHO"), a->tmp);
 205         command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 206
 207         for (aura_dict_rewind(di); !aura_dict_eof(di); aura_dict_next(di) ) {
 208                 aura_dict_get_current_key(di, &rk, &rk_len);
 209                 disk = (char *)rk;
 210
 211                 /*
 212                  * Attempt to get media information from the disk. This information
 213                  * might be used later on for partitioning. Any disk that does not
 214                  * provide the information will be discarded as not suitable for
 215                  * an installation.
 216                  */
 217                 bzero(&diskpart, sizeof(diskpart));
 218                 snprintf(diskpath, PATH_MAX, "/dev/%s", disk);
 219                 if ((fd = open(diskpath, O_RDONLY)) < 0)
 220                         continue;
 221                 if (ioctl(fd, DIOCGPART, &diskpart) < 0)
 222                         continue;
 223
 224                 cmd = command_add(cmds, "%s%s '@DISK' >>%ssurvey.txt",
 225                     a->os_root, cmd_name(a, "ECHO"), a->tmp);
 226                 command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 227                 cmd = command_add(cmds, "%s%s '%s' >>%ssurvey.txt",
 228                     a->os_root, cmd_name(a, "ECHO"), disk, a->tmp);
 229                 command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 230
 231                 /*
 232                  * Look for descriptions of this disk.
 233                  */
 234                 cmd = command_add(cmds, "%s%s '@DESC' >>%ssurvey.txt",
 235                     a->os_root, cmd_name(a, "ECHO"), a->tmp);
 236                 command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 237                 cmd = command_add(cmds, "%s%s '%s: %luMB' >>%ssurvey.txt",
 238                     a->os_root, cmd_name(a, "ECHO"),
 239                     disk,
 240                     diskpart.media_size / 1024 / 1024,
 241                     a->tmp);
 242                 close(fd);
 243                 cmd = command_add(cmds, "%s%s '@END' >>%ssurvey.txt",
 244                     a->os_root, cmd_name(a, "ECHO"), a->tmp);
 245                 command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 246
 247                 /*
 248                  * Look for the disk's serial number.
 249                  */
 250                 cmd = command_add(cmds, "%s%s '@SERNO' >>%ssurvey.txt",
 251                     a->os_root, cmd_name(a, "ECHO"), a->tmp);
 252                 command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 253                 cmd = command_add(cmds, "if %s%s -d /dev/serno; then %s%s -l /dev/serno | %s%s \"`%s%s -l /dev/%s | %s%s '{print $5, $6;}'`\" | %s%s '{print $10;}' >>%ssurvey.txt; fi",
 254                     a->os_root, cmd_name(a, "TEST"),
 255                     a->os_root, cmd_name(a, "LS"),
 256                     a->os_root, cmd_name(a, "GREP"),
 257                     a->os_root, cmd_name(a, "LS"),
 258                     disk,
 259                     a->os_root, cmd_name(a, "AWK"),
 260                     a->os_root, cmd_name(a, "AWK"),
 261                     a->tmp);
 262                 cmd = command_add(cmds, "%s%s '@END' >>%ssurvey.txt",
 263                     a->os_root, cmd_name(a, "ECHO"), a->tmp);
 264                 command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 265
 266                 /*
 267                  * Probe the disk with fdisk.
 268                  */
 269                 cmd = command_add(cmds, "%s%s '@SLICES' >>%ssurvey.txt",
 270                     a->os_root, cmd_name(a, "ECHO"), a->tmp);
 271                 command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 272                 cmd = command_add(cmds, "%s%s -s %s 2>/dev/null >>%ssurvey.txt || %s%s '' >>%ssurvey.txt",
 273                     a->os_root, cmd_name(a, "FDISK"),
 274                     disk,
 275                     a->tmp,
 276                     a->os_root, cmd_name(a, "ECHO"),
 277                     a->tmp);
 278                 cmd = command_add(cmds, "%s%s '@END' >>%ssurvey.txt",
 279                     a->os_root, cmd_name(a, "ECHO"), a->tmp);
 280                 command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 281         }
 282
 283         cmd = command_add(cmds, "%s%s '.' >>%ssurvey.txt",
 284             a->os_root, cmd_name(a, "ECHO"), a->tmp);
 285         command_set_log_mode(cmd, COMMAND_LOG_SILENT);
 286
 287         if (!commands_execute(a, cmds))
 288                 failure |= 1;
 289         commands_free(cmds);
 290         temp_file_add(a, "survey.txt");
 291
 292         aura_dict_free(di);
 293
 294         /*
 295          * Now read in and parse the file that those commands just created.
 296          */
 297         asprintf(&filename, "%ssurvey.txt", a->tmp);
 298         if ((f = fopen(filename, "r")) == NULL)
 299                 failure |= 1;
 300         free(filename);
 301
 302         while (!failure && fgets_chomp(line, 255, f)) {
 303                 if (strcmp(line, "@DISK") == 0) {
 304                         if (fgets_chomp(line, 255, f)) {
 305                                 d = disk_new(a->s, line);
 306                                 disk_set_number(d, number++);
 307                         }
 308                 } else if (strcmp(line, "@DESC") == 0) {
 309                         while (d != NULL && strcmp(line, "@END") != 0 && fgets_chomp(line, 255, f)) {
 310                                 disk_set_desc(d, line);
 311                         }
 312                 } else if (strcmp(line, "@SERNO") == 0) {
 313                         fgets_chomp(line, 255, f);
 314                         if (line[0] != '\0' && strcmp(line, "@END") != 0)
 315                                 disk_set_serno(d, line);
 316                 } else if (strcmp(line, "@SLICES") == 0) {
 317                         int cyl, hd, sec;
 318                         int sliceno, type, flags;
 319                         unsigned long start, size;
 320
 321                         /*
 322                          * /dev/ad3: 2112 cyl 16 hd 63 sec
 323                          * Part        Start        Size Type Flags
 324                          *    1:          63     2128833 0x6c 0x80
 325                          */
 326                         while (d != NULL && strcmp(line, "@END") != 0 && fgets_chomp(line, 255, f)) {
 327                                 if (strncmp(line, "/dev/", 5) == 0) {
 328                                         cyl = hd = sec = 0;
 329                                         parse_geometry_info(line, &cyl, &hd, &sec);
 330                                         disk_set_geometry(d, cyl, hd, sec);
 331                                 } else if (strncmp(line, "Part", 4) == 0) {
 332                                         /* ignore it */
 333                                 } else {
 334                                         if (parse_slice_info(line, &sliceno, &start, &size,
 335                                             &type, &flags)) {
 336                                                 /*
 337                                                 fprintfo(log, "| Found slice #%d, sysid %d, "
 338                                                     "start %ld, size %ld\n", sliceno, type, start, size);
 339                                                 */
 340                                                 slice_new(d, sliceno, type, flags, start, size);
 341                                         }
 342                                 }
 343                         }
 344                 }
 345         }
 346
 347         if (f != NULL)
 348                 fclose(f);
 349
 350         /*
 351          * Fix up any disk descriptions that didn't make it.
 352          */
 353         for (d = storage_disk_first(a->s); d != NULL; d = disk_next(d)) {
 354                 if (disk_get_desc(d) == NULL)
 355                         disk_set_desc(d, disk_get_device_name(d));
 356         }
 357
 358         return(!failure);
 359 }