Module Name: src Committed By: martin Date: Wed Nov 14 02:30:00 UTC 2018
Modified Files: src/usr.sbin/sysinst: defs.h disks.c Log Message: Slightly enhance previous: create two utility functions checking for partitionability and boot code requirements, use a generic match helper function for both and fully check the device name. To generate a diff of this commit: cvs rdiff -u -r1.24 -r1.25 src/usr.sbin/sysinst/defs.h cvs rdiff -u -r1.26 -r1.27 src/usr.sbin/sysinst/disks.c Please note that diffs are not public domain; they are subject to the copyright notices on the relevant files.
Modified files: Index: src/usr.sbin/sysinst/defs.h diff -u src/usr.sbin/sysinst/defs.h:1.24 src/usr.sbin/sysinst/defs.h:1.25 --- src/usr.sbin/sysinst/defs.h:1.24 Sun Nov 11 10:06:09 2018 +++ src/usr.sbin/sysinst/defs.h Wed Nov 14 02:30:00 2018 @@ -1,4 +1,4 @@ -/* $NetBSD: defs.h,v 1.24 2018/11/11 10:06:09 martin Exp $ */ +/* $NetBSD: defs.h,v 1.25 2018/11/14 02:30:00 martin Exp $ */ /* * Copyright 1997 Piermont Information Systems Inc. @@ -483,6 +483,8 @@ bool get_default_cdrom(char *, size_t); int find_disks(const char *); bool enumerate_disks(void *state,bool (*func)(void *state, const char *dev)); bool is_cdrom_device(const char *dev, bool as_target); +bool is_bootable_device(const char *dev); +bool is_partitionable_device(const char *dev); struct menudesc; void fmt_fspart(struct menudesc *, int, void *); Index: src/usr.sbin/sysinst/disks.c diff -u src/usr.sbin/sysinst/disks.c:1.26 src/usr.sbin/sysinst/disks.c:1.27 --- src/usr.sbin/sysinst/disks.c:1.26 Tue Nov 13 17:22:04 2018 +++ src/usr.sbin/sysinst/disks.c Wed Nov 14 02:30:00 2018 @@ -1,4 +1,4 @@ -/* $NetBSD: disks.c,v 1.26 2018/11/13 17:22:04 bouyer Exp $ */ +/* $NetBSD: disks.c,v 1.27 2018/11/14 02:30:00 martin Exp $ */ /* * Copyright 1997 Piermont Information Systems Inc. @@ -546,6 +546,53 @@ is_cdrom_device(const char *dev, bool as return false; } +/* does this device match any entry in the driver list? */ +static bool +dev_in_list(const char *dev, const char **list) +{ + + for ( ; *list; list++) { + + size_t len = strlen(*list); + + /* start of name matches? */ + if (strncmp(dev, *list, len) == 0) { + char *endp; + int e; + + /* remainder of name is a decimal number? */ + strtou(dev+len, &endp, 10, 0, INT_MAX, &e); + if (endp && *endp == 0 && e == 0) + return true; + } + } + + return false; +} + +bool +is_bootable_device(const char *dev) +{ + static const char *non_bootable_devs[] = { + "raid", /* bootcode lives outside of raid */ + "xbd", /* xen virtual device, can not boot from that */ + NULL + }; + + return !dev_in_list(dev, non_bootable_devs); +} + +bool +is_partitionable_device(const char *dev) +{ + static const char *non_partitionable_devs[] = { + "dk", /* this is alreay a partioned slice */ + NULL + }; + + return !dev_in_list(dev, non_partitionable_devs); +} + /* * Multi-purpose helper function: * iterate all known disks, invoke a callback for each. @@ -604,27 +651,14 @@ get_disks_helper(void *arg, const char * if (is_cdrom_device(dev, true)) return true; + memset(state->dd, 0, sizeof(*state->dd)); strlcpy(state->dd->dd_name, dev, sizeof state->dd->dd_name - 2); - state->dd->dd_no_mbr = false; - state->dd->dd_no_part = false; + state->dd->dd_no_mbr = !is_bootable_device(dev); + state->dd->dd_no_part = !is_partitionable_device(dev); - if (strncmp(dev, "dk", 2) == 0) { - char *endp; - int e; - - /* if this device is dkNNNN, no partitioning is possible */ - strtou(dev+2, &endp, 10, 0, INT_MAX, &e); - if (endp && *endp == 0 && e == 0) - state->dd->dd_no_part = true; - } if (state->dd->dd_no_part && !state->with_non_partitionable) return true; - if (strncmp(dev, "xbd", 3) == 0 || strncmp(dev, "raid", 4) == 0) { - /* if this device is xbd or raid, don't set up an MBR */ - state->dd->dd_no_mbr = true; - } - if (!get_geom(state->dd->dd_name, &l)) { if (errno == ENOENT) return true;