| Commit | Line | Data |
|---|---|---|
| 337ad0df DH |
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> | |
| 337ad0df DH |
41 | #include <sys/sysctl.h> |
| 42 | ||
| 43 | #include <stdio.h> | |
| 44 | #include <string.h> | |
| 45 | ||
| bc3d4063 | 46 | #include "libaura/dict.h" |
| 337ad0df DH |
47 | |
| 48 | #include "commands.h" | |
| 49 | #include "diskutil.h" | |
| 50 | #include "functions.h" | |
| 51 | ||
| 337ad0df DH |
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 | { | |
| 1c6a6c4c | 144 | unsigned long mem = 0; |
| 337ad0df DH |
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); | |
| 337ad0df | 161 | if (sysctlbyname("hw.physmem", &mem, &len, NULL, 0) < 0) { |
| 337ad0df DH |
162 | failure |= 1; |
| 163 | } else { | |
| 164 | storage_set_memsize(a->s, next_power_of_two(mem >> 20)); | |
| 165 | } | |
| 166 | len = 256; | |
| 337ad0df | 167 | if (sysctlbyname("kern.disks", disks, &len, NULL, 0) < 0) { |
| 337ad0df DH |
168 | failure |= 1; |
| 169 | } | |
| 337ad0df DH |
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 | /* | |
| d08ff9d4 | 178 | * If the disk is a memory disk, floppy or CD-ROM, skip it. |
| 337ad0df DH |
179 | */ |
| 180 | if (strncmp(disk, "md", 2) == 0 || | |
| d08ff9d4 SW |
181 | strncmp(disk, "cd", 2) == 0 || |
| 182 | strncmp(disk, "acd", 3) == 0 || | |
| 183 | strncmp(disk, "fd", 2) == 0) | |
| 337ad0df DH |
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); | |
| b299bac1 | 213 | cmd = command_add(cmds, "%s%s -w '^%s: [0-9]*MB' %s%s >>%ssurvey.txt || %s%s '%s' >>%ssurvey.txt", |
| 337ad0df DH |
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 | * Ensure that the device node for this disk exists, | |
| 337ad0df DH |
227 | */ |
| 228 | cmd = command_add_ensure_dev(a, cmds, disk); | |
| 229 | ||
| 230 | /* | |
| 231 | * Probe the disk with fdisk. | |
| 232 | */ | |
| 233 | cmd = command_add(cmds, "%s%s '@SLICES' >>%ssurvey.txt", | |
| 234 | a->os_root, cmd_name(a, "ECHO"), a->tmp); | |
| 235 | command_set_log_mode(cmd, COMMAND_LOG_SILENT); | |
| 236 | cmd = command_add(cmds, "%s%s -s %s 2>/dev/null >>%ssurvey.txt || %s%s '' >>%ssurvey.txt", | |
| 237 | a->os_root, cmd_name(a, "FDISK"), | |
| 238 | disk, | |
| 239 | a->tmp, | |
| 240 | a->os_root, cmd_name(a, "ECHO"), | |
| 241 | a->tmp); | |
| 242 | cmd = command_add(cmds, "%s%s '@END' >>%ssurvey.txt", | |
| 243 | a->os_root, cmd_name(a, "ECHO"), a->tmp); | |
| 244 | command_set_log_mode(cmd, COMMAND_LOG_SILENT); | |
| 245 | ||
| 246 | aura_dict_next(di); | |
| 247 | } | |
| 248 | ||
| 249 | cmd = command_add(cmds, "%s%s '.' >>%ssurvey.txt", | |
| 250 | a->os_root, cmd_name(a, "ECHO"), a->tmp); | |
| 251 | command_set_log_mode(cmd, COMMAND_LOG_SILENT); | |
| 252 | ||
| 253 | if (!commands_execute(a, cmds)) | |
| 254 | failure |= 1; | |
| 255 | commands_free(cmds); | |
| 256 | temp_file_add(a, "survey.txt"); | |
| 257 | ||
| 258 | aura_dict_free(di); | |
| 259 | ||
| 260 | /* | |
| 261 | * Now read in and parse the file that those commands just created. | |
| 262 | */ | |
| 263 | asprintf(&filename, "%ssurvey.txt", a->tmp); | |
| 264 | if ((f = fopen(filename, "r")) == NULL) | |
| 265 | failure |= 1; | |
| 266 | free(filename); | |
| 267 | ||
| 268 | while (!failure && fgets_chomp(line, 255, f)) { | |
| 269 | if (strcmp(line, "@DISK") == 0) { | |
| 270 | if (fgets_chomp(line, 255, f)) { | |
| 271 | d = disk_new(a->s, line); | |
| 272 | } | |
| 273 | } else if (strcmp(line, "@DESC") == 0) { | |
| 274 | while (d != NULL && strcmp(line, "@END") != 0 && fgets_chomp(line, 255, f)) { | |
| 275 | disk_set_desc(d, line); | |
| 276 | } | |
| 277 | } else if (strcmp(line, "@SLICES") == 0) { | |
| 278 | int cyl, hd, sec; | |
| 279 | int number, type, flags; | |
| 280 | unsigned long start, size; | |
| 281 | ||
| 282 | /* | |
| 283 | * /dev/ad3: 2112 cyl 16 hd 63 sec | |
| 284 | * Part Start Size Type Flags | |
| 285 | * 1: 63 2128833 0xa5 0x80 | |
| 286 | */ | |
| 287 | while (d != NULL && strcmp(line, "@END") != 0 && fgets_chomp(line, 255, f)) { | |
| 288 | if (strncmp(line, "/dev/", 5) == 0) { | |
| 289 | parse_geometry_info(line, &cyl, &hd, &sec); | |
| 290 | disk_set_geometry(d, cyl, hd, sec); | |
| 291 | } else if (strncmp(line, "Part", 4) == 0) { | |
| 292 | /* ignore it */ | |
| 293 | } else { | |
| 294 | if (parse_slice_info(line, &number, &start, &size, | |
| 295 | &type, &flags)) { | |
| 296 | /* | |
| 297 | fprintfo(log, "| Found slice #%d, sysid %d, " | |
| 298 | "start %ld, size %ld\n", number, type, start, size); | |
| 299 | */ | |
| 300 | slice_new(d, number, type, flags, start, size); | |
| 301 | } | |
| 302 | } | |
| 303 | } | |
| 304 | } | |
| 305 | } | |
| 306 | ||
| 307 | if (f != NULL) | |
| 308 | fclose(f); | |
| 309 | ||
| 310 | /* | |
| 311 | * Fix up any disk descriptions that didn't make it. | |
| 312 | */ | |
| 313 | for (d = storage_disk_first(a->s); d != NULL; d = disk_next(d)) { | |
| 314 | if (disk_get_desc(d) == NULL) | |
| 315 | disk_set_desc(d, disk_get_device_name(d)); | |
| 316 | } | |
| 317 | ||
| 318 | return(!failure); | |
| 319 | } |