diff options
Diffstat (limited to 'usr.sbin/sade/disks.c')
-rw-r--r-- | usr.sbin/sade/disks.c | 972 |
1 files changed, 0 insertions, 972 deletions
diff --git a/usr.sbin/sade/disks.c b/usr.sbin/sade/disks.c deleted file mode 100644 index 513643b..0000000 --- a/usr.sbin/sade/disks.c +++ /dev/null @@ -1,972 +0,0 @@ -/* - * $FreeBSD$ - * - * Copyright (c) 1995 - * Jordan Hubbard. All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions - * are met: - * 1. Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer, - * verbatim and that no modifications are made prior to this - * point in the file. - * 2. Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in the - * documentation and/or other materials provided with the distribution. - * - * THIS SOFTWARE IS PROVIDED BY JORDAN HUBBARD ``AS IS'' AND - * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE - * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE - * ARE DISCLAIMED. IN NO EVENT SHALL JORDAN HUBBARD OR HIS PETS BE LIABLE - * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL - * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS - * OR SERVICES; LOSS OF USE, DATA, LIFE OR PROFITS; OR BUSINESS INTERRUPTION) - * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT - * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY - * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF - * SUCH DAMAGE. - * - */ - -#include "sade.h" -#include <ctype.h> -#include <fcntl.h> -#include <inttypes.h> -#include <libdisk.h> -#include <sys/stat.h> -#include <sys/disklabel.h> - -#ifdef WITH_SLICES -enum size_units_t { UNIT_BLOCKS, UNIT_KILO, UNIT_MEG, UNIT_GIG, UNIT_SIZE }; - -#ifdef PC98 -#define SUBTYPE_FREEBSD 50324 -#define SUBTYPE_FAT 37218 -#else -#define SUBTYPE_FREEBSD 165 -#define SUBTYPE_FAT 6 -#endif -#define SUBTYPE_EFI 239 - -#ifdef PC98 -#define OTHER_SLICE_VALUES \ - "Other popular values are 37218 for a\n" \ - "DOS FAT partition.\n\n" -#else -#define OTHER_SLICE_VALUES \ - "Other popular values are 6 for a\n" \ - "DOS FAT partition, 131 for a Linux ext2fs partition, or\n" \ - "130 for a Linux swap partition.\n\n" -#endif -#define NON_FREEBSD_NOTE \ - "Note: If you choose a non-FreeBSD partition type, it will not\n" \ - "be formatted or otherwise prepared, it will simply reserve space\n" \ - "for you to use another tool, such as DOS format, to later format\n" \ - "and actually use the partition." - -/* Where we start displaying chunk information on the screen */ -#define CHUNK_START_ROW 5 - -/* Where we keep track of MBR chunks */ -#define CHUNK_INFO_ENTRIES 16 -static struct chunk *chunk_info[CHUNK_INFO_ENTRIES]; -static int current_chunk; - -static void diskPartitionNonInteractive(Device *dev); -#if !defined(__ia64__) -static u_char * bootalloc(char *name, size_t *size); -#endif - -static void -record_chunks(Disk *d) -{ - struct chunk *c1 = NULL; - int i = 0; - daddr_t last_free = 0; - - if (!d->chunks) - msgFatal("No chunk list found for %s!", d->name); - - for (c1 = d->chunks->part; c1; c1 = c1->next) { - if (c1->type == unused && c1->size > last_free) { - last_free = c1->size; - current_chunk = i; - } - chunk_info[i++] = c1; - } - chunk_info[i] = NULL; - if (current_chunk >= i) - current_chunk = i - 1; -} - -static daddr_t Total; - -static void -check_geometry(Disk *d) -{ - int sg; - -#ifdef PC98 - if (d->bios_cyl >= 65536 || d->bios_hd > 256 || d->bios_sect >= 256) -#else - if (d->bios_cyl > 65536 || d->bios_hd > 256 || d->bios_sect >= 64) -#endif - { - dlg_clear(); - sg = msgYesNo("WARNING: It is safe to use a geometry of %lu/%lu/%lu for %s on\n" - "computers with modern BIOS versions. If this disk is to be used\n" - "on an old machine it is recommended that it does not have more\n" - "than 65535 cylinders, more than 255 heads, or more than\n" -#ifdef PC98 - "255" -#else - "63" -#endif - " sectors per track.\n" - "\n" - "Would you like to keep using the current geometry?\n", - d->bios_cyl, d->bios_hd, d->bios_sect, d->name); - if (sg == 1) { - Sanitize_Bios_Geom(d); - msgConfirm("A geometry of %lu/%lu/%lu was calculated for %s.\n" - "\n" - "If you are not sure about this, please consult the Hardware Guide\n" - "in the Documentation submenu or use the (G)eometry command to\n" - "change it. Remember: you need to enter whatever your BIOS thinks\n" - "the geometry is! For IDE, it's what you were told in the BIOS\n" - "setup. For SCSI, it's the translation mode your controller is\n" - "using. Do NOT use a ``physical geometry''.\n", - d->bios_cyl, d->bios_hd, d->bios_sect, d->name); - } - } -} - -static void -print_chunks(Disk *d, int u) -{ - int row; - int i; - daddr_t sz; - char *szstr; - - szstr = (u == UNIT_GIG ? "GB" : (u == UNIT_MEG ? "MB" : - (u == UNIT_KILO ? "KB" : "ST"))); - - Total = 0; - for (i = 0; chunk_info[i]; i++) - Total += chunk_info[i]->size; - attrset(A_NORMAL); - mvaddstr(0, 0, "Disk name:\t"); - clrtobot(); - attrset(A_REVERSE); addstr(d->name); attrset(A_NORMAL); - attrset(A_REVERSE); mvaddstr(0, 55, "FDISK Partition Editor"); attrset(A_NORMAL); - mvprintw(1, 0, - "DISK Geometry:\t%lu cyls/%lu heads/%lu sectors = %jd sectors (%jdMB)", - d->bios_cyl, d->bios_hd, d->bios_sect, - (intmax_t)d->bios_cyl * d->bios_hd * d->bios_sect, - (intmax_t)d->bios_cyl * d->bios_hd * d->bios_sect / (1024/512) / 1024); - mvprintw(3, 0, "%6s %10s(%s) %10s %8s %6s %10s %8s %8s", - "Offset", "Size", szstr, "End", "Name", "PType", "Desc", - "Subtype", "Flags"); - for (i = 0, row = CHUNK_START_ROW; chunk_info[i]; i++, row++) { - switch(u) { - default: /* fall thru */ - case UNIT_BLOCKS: - sz = chunk_info[i]->size; - break; - case UNIT_KILO: - sz = chunk_info[i]->size / (1024/512); - break; - case UNIT_MEG: - sz = chunk_info[i]->size / (1024/512) / 1024; - break; - case UNIT_GIG: - sz = chunk_info[i]->size / (1024/512) / 1024 / 1024; - break; - } - if (i == current_chunk) - attrset(ATTR_SELECTED); - mvprintw(row, 0, "%10jd %10jd %10jd %8s %6d %10s %8d\t%-6s", - (intmax_t)chunk_info[i]->offset, (intmax_t)sz, - (intmax_t)chunk_info[i]->end, chunk_info[i]->name, - chunk_info[i]->type, - slice_type_name(chunk_info[i]->type, chunk_info[i]->subtype), - chunk_info[i]->subtype, ShowChunkFlags(chunk_info[i])); - if (i == current_chunk) - attrset(A_NORMAL); - } -} - -static void -print_command_summary(void) -{ - mvprintw(14, 0, "The following commands are supported (in upper or lower case):"); - mvprintw(16, 0, "A = Use Entire Disk G = set Drive Geometry C = Create Slice"); - mvprintw(17, 0, "D = Delete Slice Z = Toggle Size Units S = Set Bootable | = Expert m."); - mvprintw(18, 0, "T = Change Type U = Undo All Changes W = Write Changes Q = Finish"); - mvprintw(21, 0, "Use F1 or ? to get more help, arrow keys to select."); - move(0, 0); -} - -#ifdef PC98 -static void -getBootMgr(char *dname, u_char **bootipl, size_t *bootipl_size, - u_char **bootmenu, size_t *bootmenu_size) -{ - static u_char *boot0; - static size_t boot0_size; - static u_char *boot05; - static size_t boot05_size; - - char str[80]; - char *cp; - int i = 0; - - dlg_clr_result(); - cp = variable_get(VAR_BOOTMGR); - if (!cp) { - /* Figure out what kind of IPL the user wants */ - sprintf(str, "Install Boot Manager for drive %s?", dname); - MenuIPLType.title = str; - i = dmenuOpen(&MenuIPLType); - } else { - if (!strncmp(cp, "boot", 4)) - dlg_add_result(MenuIPLType.items[0].prompt); - else - dlg_add_result(MenuIPLType.items[1].prompt); - } - if (cp || i) { - if (!strcmp(dialog_vars.input_result, MenuIPLType.items[0].prompt)) { - if (!boot0) boot0 = bootalloc("boot0", &boot0_size); - *bootipl = boot0; - *bootipl_size = boot0_size; - if (!boot05) boot05 = bootalloc("boot0.5", &boot05_size); - *bootmenu = boot05; - *bootmenu_size = boot05_size; - return; - } - } - *bootipl = NULL; - *bootipl_size = 0; - *bootmenu = NULL; - *bootmenu_size = 0; -} -#else -static void -getBootMgr(char *dname, u_char **bootCode, size_t *bootCodeSize) -{ -#if defined(__i386__) || defined(__amd64__) /* only meaningful on x86 */ - static u_char *mbr, *boot0; - static size_t mbr_size, boot0_size; - char str[80]; - char *cp; - int i = 0; - - dlg_clr_result(); - cp = variable_get(VAR_BOOTMGR); - if (!cp) { - /* Figure out what kind of MBR the user wants */ - sprintf(str, "Install Boot Manager for drive %s?", dname); - MenuMBRType.title = str; - i = dmenuOpen(&MenuMBRType); - } - else { - if (!strcmp(cp, "standard")) - dlg_add_result(MenuMBRType.items[0].prompt); - if (!strncmp(cp, "boot", 4)) - dlg_add_result(MenuMBRType.items[1].prompt); - else - dlg_add_result(MenuMBRType.items[2].prompt); - } - if (cp || i) { - if (!strcmp(dialog_vars.input_result, MenuMBRType.items[0].prompt)) { - if (!mbr) mbr = bootalloc("mbr", &mbr_size); - *bootCode = mbr; - *bootCodeSize = mbr_size; - return; - } else if (!strcmp(dialog_vars.input_result, MenuMBRType.items[1].prompt)) { - if (!boot0) boot0 = bootalloc("boot0", &boot0_size); - *bootCode = boot0; - *bootCodeSize = boot0_size; - return; - } - } -#endif - *bootCode = NULL; - *bootCodeSize = 0; -} -#endif -#endif /* WITH_SLICES */ - -#ifdef WITH_SLICES -void -diskPartition(Device *dev) -{ - char *cp, *p; - int rv, key = 0; - int i; - Boolean chunking; - char *msg = NULL; -#ifdef PC98 - u_char *bootipl; - size_t bootipl_size; - u_char *bootmenu; - size_t bootmenu_size; -#else - u_char *mbrContents; - size_t mbrSize; -#endif - WINDOW *w = savescr(); - Disk *d = (Disk *)dev->private; - int size_unit; - - size_unit = UNIT_BLOCKS; - chunking = TRUE; - keypad(stdscr, TRUE); - - /* Flush both the dialog and curses library views of the screen - since we don't always know who called us */ - dlg_clear(), clear(); - current_chunk = 0; - - /* Set up the chunk array */ - record_chunks(d); - - /* Give the user a chance to sanitize the disk geometry, if necessary */ - check_geometry(d); - - while (chunking) { - char *val, geometry[80]; - - /* Now print our overall state */ - if (d) - print_chunks(d, size_unit); - print_command_summary(); - if (msg) { - attrset(title_attr); mvprintw(23, 0, msg); attrset(A_NORMAL); - beep(); - msg = NULL; - } - else { - move(23, 0); - clrtoeol(); - } - - /* Get command character */ - key = getch(); - switch (toupper(key)) { - case '\014': /* ^L (redraw) */ - clear(); - msg = NULL; - break; - - case '\020': /* ^P */ - case KEY_UP: - case '-': - if (current_chunk != 0) - --current_chunk; - break; - - case '\016': /* ^N */ - case KEY_DOWN: - case '+': - case '\r': - case '\n': - if (chunk_info[current_chunk + 1]) - ++current_chunk; - break; - - case KEY_HOME: - current_chunk = 0; - break; - - case KEY_END: - while (chunk_info[current_chunk + 1]) - ++current_chunk; - break; - - case KEY_F(1): - case '?': - systemDisplayHelp("slice"); - clear(); - break; - - case 'A': - case 'F': /* Undocumented magic Dangerously Dedicated mode */ -#if !defined(__i386__) && !defined(__amd64__) - rv = 1; -#else /* The rest is only relevant on x86 */ - cp = variable_get(VAR_DEDICATE_DISK); - if (cp && !strcasecmp(cp, "always")) - rv = 1; - else if (toupper(key) == 'A') - rv = 0; - else { - rv = msgYesNo("Do you want to do this with a true partition entry\n" - "so as to remain cooperative with any future possible\n" - "operating systems on the drive(s)?\n" - "(See also the section about ``dangerously dedicated''\n" - "disks in the FreeBSD FAQ.)"); - if (rv == -1) - rv = 0; - } -#endif - All_FreeBSD(d, rv); - variable_set2(DISK_PARTITIONED, "yes", 0); - record_chunks(d); - clear(); - break; - - case 'C': - if (chunk_info[current_chunk]->type != unused) - msg = "Slice in use, delete it first or move to an unused one."; - else { - char *val, tmp[20], name[16], *cp; - daddr_t size; - int subtype; - chunk_e partitiontype; -#ifdef PC98 - snprintf(name, sizeof (name), "%s", "FreeBSD"); - val = msgGetInput(name, - "Please specify the name for new FreeBSD slice."); - if (val) - strncpy(name, val, sizeof (name)); -#else - name[0] = '\0'; -#endif - snprintf(tmp, 20, "%jd", (intmax_t)chunk_info[current_chunk]->size); - val = msgGetInput(tmp, "Please specify the size for new FreeBSD slice in blocks\n" - "or append a trailing `M' for megabytes (e.g. 20M)."); - if (val && (size = strtoimax(val, &cp, 0)) > 0) { - if (*cp && toupper(*cp) == 'M') - size *= ONE_MEG; - else if (*cp && toupper(*cp) == 'G') - size *= ONE_GIG; - sprintf(tmp, "%d", SUBTYPE_FREEBSD); - val = msgGetInput(tmp, "Enter type of partition to create:\n\n" - "Pressing Enter will choose the default, a native FreeBSD\n" - "slice (type %u). " - OTHER_SLICE_VALUES - NON_FREEBSD_NOTE, SUBTYPE_FREEBSD); - if (val && (subtype = strtol(val, NULL, 0)) > 0) { - if (subtype == SUBTYPE_FREEBSD) - partitiontype = freebsd; - else if (subtype == SUBTYPE_FAT) - partitiontype = fat; - else if (subtype == SUBTYPE_EFI) - partitiontype = efi; - else -#ifdef PC98 - partitiontype = pc98; -#else - partitiontype = mbr; -#endif - Create_Chunk(d, chunk_info[current_chunk]->offset, size, partitiontype, subtype, - (chunk_info[current_chunk]->flags & CHUNK_ALIGN), name); - variable_set2(DISK_PARTITIONED, "yes", 0); - record_chunks(d); - } - } - clear(); - } - break; - - case KEY_DC: - case 'D': - if (chunk_info[current_chunk]->type == unused) - msg = "Slice is already unused!"; - else { - Delete_Chunk(d, chunk_info[current_chunk]); - variable_set2(DISK_PARTITIONED, "yes", 0); - record_chunks(d); - } - break; - - case 'T': - if (chunk_info[current_chunk]->type == unused) - msg = "Slice is currently unused (use create instead)"; - else { - char *val, tmp[20]; - int subtype; - chunk_e partitiontype; - - sprintf(tmp, "%d", chunk_info[current_chunk]->subtype); - val = msgGetInput(tmp, "New partition type:\n\n" - "Pressing Enter will use the current type. To choose a native\n" - "FreeBSD slice enter %u. " - OTHER_SLICE_VALUES - NON_FREEBSD_NOTE, SUBTYPE_FREEBSD); - if (val && (subtype = strtol(val, NULL, 0)) > 0) { - if (subtype == SUBTYPE_FREEBSD) - partitiontype = freebsd; - else if (subtype == SUBTYPE_FAT) - partitiontype = fat; - else if (subtype == SUBTYPE_EFI) - partitiontype = efi; - else -#ifdef PC98 - partitiontype = pc98; -#else - partitiontype = mbr; -#endif - chunk_info[current_chunk]->type = partitiontype; - chunk_info[current_chunk]->subtype = subtype; - } - } - break; - - case 'G': - snprintf(geometry, 80, "%lu/%lu/%lu", d->bios_cyl, d->bios_hd, d->bios_sect); - val = msgGetInput(geometry, "Please specify the new geometry in cyl/hd/sect format.\n" - "Don't forget to use the two slash (/) separator characters!\n" - "It's not possible to parse the field without them."); - if (val) { - long nc, nh, ns; - nc = strtol(val, &val, 0); - nh = strtol(val + 1, &val, 0); - ns = strtol(val + 1, 0, 0); - Set_Bios_Geom(d, nc, nh, ns); - } - clear(); - break; - - case 'S': - /* Clear active states so we won't have two */ - for (i = 0; (chunk_info[i] != NULL) && (i < CHUNK_INFO_ENTRIES); i++) - chunk_info[i]->flags &= !CHUNK_ACTIVE; - - /* Set Bootable */ - chunk_info[current_chunk]->flags |= CHUNK_ACTIVE; - break; - - case 'U': - if (!variable_cmp(DISK_LABELLED, "written")) { - msgConfirm("You've already written this information out - you\n" - "can't undo it."); - } - else if (!msgNoYes("Are you SURE you want to Undo everything?")) { - char cp[BUFSIZ]; - - sstrncpy(cp, d->name, sizeof cp); - Free_Disk(dev->private); - d = Open_Disk(cp); - if (!d) - msgConfirm("Can't reopen disk %s! Internal state is probably corrupted", cp); - dev->private = d; - variable_unset(DISK_PARTITIONED); - variable_unset(DISK_LABELLED); - if (d) - record_chunks(d); - } - clear(); - break; - - case 'W': - if (!msgNoYes("WARNING: You are about to modify an EXISTING installation.\n" - "You should simply type Q when you are finished\n" - "here and write to the disk from the label editor.\n\n" - "Are you absolutely sure you want to continue?")) { - variable_set2(DISK_PARTITIONED, "yes", 0); - -#ifdef PC98 - /* - * Don't trash the IPL if the first (and therefore only) chunk - * is marked for a truly dedicated disk (i.e., the disklabel - * starts at sector 0), even in cases where the user has - * requested a FreeBSD Boot Manager -- both would be fatal in - * this case. - */ - /* - * Don't offer to update the IPL on this disk if the first - * "real" chunk looks like a FreeBSD "all disk" partition, - * or the disk is entirely FreeBSD. - */ - if ((d->chunks->part->type != freebsd) || - (d->chunks->part->offset > 1)) - getBootMgr(d->name, &bootipl, &bootipl_size, - &bootmenu, &bootmenu_size); - else { - bootipl = NULL; - bootipl_size = 0; - bootmenu = NULL; - bootmenu_size = 0; - } - Set_Boot_Mgr(d, bootipl, bootipl_size, bootmenu, bootmenu_size); -#else - /* - * Don't trash the MBR if the first (and therefore only) chunk - * is marked for a truly dedicated disk (i.e., the disklabel - * starts at sector 0), even in cases where the user has - * requested booteasy or a "standard" MBR -- both would be - * fatal in this case. - */ - /* - * Don't offer to update the MBR on this disk if the first - * "real" chunk looks like a FreeBSD "all disk" partition, - * or the disk is entirely FreeBSD. - */ - if ((d->chunks->part->type != freebsd) || - (d->chunks->part->offset > 1)) - getBootMgr(d->name, &mbrContents, &mbrSize); - else { - mbrContents = NULL; - mbrSize = 0; - } - Set_Boot_Mgr(d, mbrContents, mbrSize); -#endif - - if (DITEM_STATUS(diskPartitionWrite(dev)) != DITEM_SUCCESS) - msgConfirm("Disk partition write returned an error status!"); - else - msgConfirm("Wrote FDISK partition information out successfully."); - } - clear(); - break; - - case '|': - if (!msgNoYes("Are you SURE you want to go into Wizard mode?\n" - "No seat belts whatsoever are provided!")) { - clear(); - refresh(); - slice_wizard(d); - variable_set2(DISK_PARTITIONED, "yes", 0); - record_chunks(d); - } - else - msg = "Wise choice!"; - clear(); - break; - - case '\033': /* ESC */ - case 'Q': - chunking = FALSE; -#ifdef PC98 - /* - * Don't trash the IPL if the first (and therefore only) chunk - * is marked for a truly dedicated disk (i.e., the disklabel - * starts at sector 0), even in cases where the user has requested - * a FreeBSD Boot Manager -- both would be fatal in this case. - */ - /* - * Don't offer to update the IPL on this disk if the first "real" - * chunk looks like a FreeBSD "all disk" partition, or the disk is - * entirely FreeBSD. - */ - if ((d->chunks->part->type != freebsd) || - (d->chunks->part->offset > 1)) { - if (variable_cmp(DISK_PARTITIONED, "written")) { - getBootMgr(d->name, &bootipl, &bootipl_size, - &bootmenu, &bootmenu_size); - if (bootipl != NULL && bootmenu != NULL) - Set_Boot_Mgr(d, bootipl, bootipl_size, - bootmenu, bootmenu_size); - } - } -#else - /* - * Don't trash the MBR if the first (and therefore only) chunk - * is marked for a truly dedicated disk (i.e., the disklabel - * starts at sector 0), even in cases where the user has requested - * booteasy or a "standard" MBR -- both would be fatal in this case. - */ - /* - * Don't offer to update the MBR on this disk if the first "real" - * chunk looks like a FreeBSD "all disk" partition, or the disk is - * entirely FreeBSD. - */ - if ((d->chunks->part->type != freebsd) || - (d->chunks->part->offset > 1)) { - if (variable_cmp(DISK_PARTITIONED, "written")) { - getBootMgr(d->name, &mbrContents, &mbrSize); - if (mbrContents != NULL) - Set_Boot_Mgr(d, mbrContents, mbrSize); - } - } -#endif - break; - - case 'Z': - size_unit = (size_unit + 1) % UNIT_SIZE; - break; - - default: - beep(); - msg = "Type F1 or ? for help"; - break; - } - } - p = CheckRules(d); - if (p) { - char buf[FILENAME_MAX]; - DIALOG_VARS save_vars; - - dlg_save_vars(&save_vars); - dialog_vars.help_line = "Press F1 to read more about disk slices."; - dialog_vars.help_file = systemHelpFile("partition", buf); - if (!variable_get(VAR_NO_WARN)) - xdialog_msgbox("Disk slicing warning:", p, -1, -1, 1); - free(p); - dlg_restore_vars(&save_vars); - } - restorescr(w); -} -#endif /* WITH_SLICES */ - -#if !defined(__ia64__) -static u_char * -bootalloc(char *name, size_t *size) -{ - char buf[FILENAME_MAX]; - struct stat sb; - - snprintf(buf, sizeof buf, "/boot/%s", name); - if (stat(buf, &sb) != -1) { - int fd; - - fd = open(buf, O_RDONLY); - if (fd != -1) { - u_char *cp; - - cp = malloc(sb.st_size); - if (read(fd, cp, sb.st_size) != sb.st_size) { - free(cp); - close(fd); - msgDebug("bootalloc: couldn't read %ld bytes from %s\n", (long)sb.st_size, buf); - return NULL; - } - close(fd); - if (size != NULL) - *size = sb.st_size; - return cp; - } - msgDebug("bootalloc: couldn't open %s\n", buf); - } - else - msgDebug("bootalloc: can't stat %s\n", buf); - return NULL; -} -#endif /* !__ia64__ */ - -#ifdef WITH_SLICES -static int -partitionHook(dialogMenuItem *selected) -{ - Device **devs = NULL; - - devs = deviceFind(selected->prompt, DEVICE_TYPE_DISK); - if (!devs) { - msgConfirm("Unable to find disk %s!", selected->prompt); - return DITEM_FAILURE; - } - diskPartition(devs[0]); - return DITEM_SUCCESS; -} - -int -diskPartitionEditor(dialogMenuItem *self) -{ - DMenu *menu; - Device **devs; - - devs = deviceFind(variable_get(VAR_DISK), DEVICE_TYPE_DISK); - if (devs == NULL) { - msgConfirm("No disks found! Please verify that your disk controller is being\n" - "properly probed at boot time. See the Hardware Guide on the\n" - "Documentation menu for clues on diagnosing this type of problem."); - return DITEM_FAILURE; - } - else { - /* No disks are selected, fall-back case now */ - int cnt = deviceCount(devs); - - if (cnt == 1) { - if (variable_get(VAR_NONINTERACTIVE) && - !variable_get(VAR_DISKINTERACTIVE)) - diskPartitionNonInteractive(devs[0]); - else - diskPartition(devs[0]); - return DITEM_SUCCESS; - } - else { - int result; - - menu = deviceCreateMenu(&MenuDiskDevices, DEVICE_TYPE_DISK, partitionHook); - if (!menu) { - msgConfirm("No devices suitable for installation found!\n\n" - "Please verify that your disk controller (and attached drives)\n" - "were detected properly. This can be done by pressing the\n" - "[Scroll Lock] key and using the Arrow keys to move back to\n" - "the boot messages. Press [Scroll Lock] again to return."); - return DITEM_FAILURE; - } - - result = dmenuOpen(menu) ? DITEM_SUCCESS : DITEM_FAILURE; - free(menu); - return result; - } - } - return DITEM_SUCCESS; -} -#endif /* WITH_SLICES */ - -int -diskPartitionWrite(Device *dev) -{ - Disk *d = (Disk *)dev->private; -#if !defined(__ia64__) - static u_char *boot1; -#endif -#if defined(__i386__) || defined(__amd64__) - static u_char *boot2; -#endif - - if (!variable_cmp(DISK_PARTITIONED, "written")) - return DITEM_SUCCESS; - -#if defined(__i386__) || defined(__amd64__) - if (!boot1) boot1 = bootalloc("boot1", NULL); - if (!boot2) boot2 = bootalloc("boot2", NULL); - Set_Boot_Blocks(d, boot1, boot2); -#elif !defined(__ia64__) - if (!boot1) boot1 = bootalloc("boot1", NULL); - Set_Boot_Blocks(d, boot1, NULL); -#endif - - msgNotify("Writing partition information to drive %s", d->name); - if (!Fake && Write_Disk(d)) { - msgConfirm("ERROR: Unable to write data to disk %s!", d->name); - return DITEM_FAILURE; - } - - /* Now it's not "yes", but "written" */ - variable_set2(DISK_PARTITIONED, "written", 0); - return DITEM_SUCCESS | DITEM_RESTORE; -} - -#ifdef WITH_SLICES -/* Partition a disk based wholly on which variables are set */ -static void -diskPartitionNonInteractive(Device *dev) -{ - char *cp; - int i, all_disk = 0; - daddr_t sz; -#ifdef PC98 - u_char *bootipl; - size_t bootipl_size; - u_char *bootmenu; - size_t bootmenu_size; -#else - u_char *mbrContents; - size_t mbrSize; -#endif - Disk *d = (Disk *)dev->private; - - record_chunks(d); - cp = variable_get(VAR_GEOMETRY); - if (cp) { - if (!strcasecmp(cp, "sane")) { -#ifdef PC98 - if (d->bios_cyl >= 65536 || d->bios_hd > 256 || d->bios_sect >= 256) -#else - if (d->bios_cyl > 65536 || d->bios_hd > 256 || d->bios_sect >= 64) -#endif - { - msgDebug("Warning: A geometry of %lu/%lu/%lu for %s is incorrect.\n", - d->bios_cyl, d->bios_hd, d->bios_sect, d->name); - Sanitize_Bios_Geom(d); - msgDebug("Sanitized geometry for %s is %lu/%lu/%lu.\n", - d->name, d->bios_cyl, d->bios_hd, d->bios_sect); - } - } else { - msgDebug("Setting geometry from script to: %s\n", cp); - d->bios_cyl = strtol(cp, &cp, 0); - d->bios_hd = strtol(cp + 1, &cp, 0); - d->bios_sect = strtol(cp + 1, 0, 0); - } - } - - cp = variable_get(VAR_PARTITION); - if (cp) { - if (!strcmp(cp, "free")) { - /* Do free disk space case */ - for (i = 0; chunk_info[i]; i++) { - /* If a chunk is at least 10MB in size, use it. */ - if (chunk_info[i]->type == unused && chunk_info[i]->size > (10 * ONE_MEG)) { - Create_Chunk(d, chunk_info[i]->offset, chunk_info[i]->size, - freebsd, 3, - (chunk_info[i]->flags & CHUNK_ALIGN), - "FreeBSD"); - variable_set2(DISK_PARTITIONED, "yes", 0); - break; - } - } - if (!chunk_info[i]) { - msgConfirm("Unable to find any free space on this disk!"); - return; - } - } - else if (!strcmp(cp, "all")) { - /* Do all disk space case */ - msgDebug("Warning: Devoting all of disk %s to FreeBSD.\n", d->name); - - All_FreeBSD(d, FALSE); - } - else if (!strcmp(cp, "exclusive")) { - /* Do really-all-the-disk-space case */ - msgDebug("Warning: Devoting all of disk %s to FreeBSD.\n", d->name); - - All_FreeBSD(d, all_disk = TRUE); - } - else if ((sz = strtoimax(cp, &cp, 0))) { - /* Look for sz bytes free */ - if (*cp && toupper(*cp) == 'M') - sz *= ONE_MEG; - else if (*cp && toupper(*cp) == 'G') - sz *= ONE_GIG; - for (i = 0; chunk_info[i]; i++) { - /* If a chunk is at least sz MB, use it. */ - if (chunk_info[i]->type == unused && chunk_info[i]->size >= sz) { - Create_Chunk(d, chunk_info[i]->offset, sz, freebsd, 3, - (chunk_info[i]->flags & CHUNK_ALIGN), - "FreeBSD"); - variable_set2(DISK_PARTITIONED, "yes", 0); - break; - } - } - if (!chunk_info[i]) { - msgConfirm("Unable to find %jd free blocks on this disk!", - (intmax_t)sz); - return; - } - } - else if (!strcmp(cp, "existing")) { - /* Do existing FreeBSD case */ - for (i = 0; chunk_info[i]; i++) { - if (chunk_info[i]->type == freebsd) - break; - } - if (!chunk_info[i]) { - msgConfirm("Unable to find any existing FreeBSD partitions on this disk!"); - return; - } - } - else { - msgConfirm("`%s' is an invalid value for %s - is config file valid?", cp, VAR_PARTITION); - return; - } - if (!all_disk) { -#ifdef PC98 - getBootMgr(d->name, &bootipl, &bootipl_size, - &bootmenu, &bootmenu_size); - Set_Boot_Mgr(d, bootipl, bootipl_size, bootmenu, bootmenu_size); -#else - getBootMgr(d->name, &mbrContents, &mbrSize); - Set_Boot_Mgr(d, mbrContents, mbrSize); -#endif - } - variable_set2(DISK_PARTITIONED, "yes", 0); - } -} -#endif /* WITH_SLICES */ |