Home | History | Annotate | Line # | Download | only in sysinst
disks.c revision 1.4.4.4
      1  1.4.4.4   msaitoh /*	$NetBSD: disks.c,v 1.4.4.4 2015/05/09 13:50:15 msaitoh Exp $ */
      2      1.1  dholland 
      3      1.1  dholland /*
      4      1.1  dholland  * Copyright 1997 Piermont Information Systems Inc.
      5      1.1  dholland  * All rights reserved.
      6      1.1  dholland  *
      7      1.1  dholland  * Written by Philip A. Nelson for Piermont Information Systems Inc.
      8      1.1  dholland  *
      9      1.1  dholland  * Redistribution and use in source and binary forms, with or without
     10      1.1  dholland  * modification, are permitted provided that the following conditions
     11      1.1  dholland  * are met:
     12      1.1  dholland  * 1. Redistributions of source code must retain the above copyright
     13      1.1  dholland  *    notice, this list of conditions and the following disclaimer.
     14      1.1  dholland  * 2. Redistributions in binary form must reproduce the above copyright
     15      1.1  dholland  *    notice, this list of conditions and the following disclaimer in the
     16      1.1  dholland  *    documentation and/or other materials provided with the distribution.
     17      1.1  dholland  * 3. The name of Piermont Information Systems Inc. may not be used to endorse
     18      1.1  dholland  *    or promote products derived from this software without specific prior
     19      1.1  dholland  *    written permission.
     20      1.1  dholland  *
     21      1.1  dholland  * THIS SOFTWARE IS PROVIDED BY PIERMONT INFORMATION SYSTEMS INC. ``AS IS''
     22      1.1  dholland  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     23      1.1  dholland  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     24      1.1  dholland  * ARE DISCLAIMED. IN NO EVENT SHALL PIERMONT INFORMATION SYSTEMS INC. BE
     25      1.1  dholland  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     26      1.1  dholland  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     27      1.1  dholland  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     28      1.1  dholland  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     29      1.1  dholland  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     30      1.1  dholland  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
     31      1.1  dholland  * THE POSSIBILITY OF SUCH DAMAGE.
     32      1.1  dholland  *
     33      1.1  dholland  */
     34      1.1  dholland 
     35      1.1  dholland /* disks.c -- routines to deal with finding disks and labeling disks. */
     36      1.1  dholland 
     37      1.1  dholland 
     38      1.1  dholland #include <errno.h>
     39      1.1  dholland #include <stdio.h>
     40      1.1  dholland #include <stdlib.h>
     41      1.1  dholland #include <unistd.h>
     42      1.1  dholland #include <fcntl.h>
     43      1.1  dholland #include <util.h>
     44      1.2    martin #include <uuid.h>
     45      1.1  dholland 
     46      1.1  dholland #include <sys/param.h>
     47      1.1  dholland #include <sys/sysctl.h>
     48      1.1  dholland #include <sys/swap.h>
     49      1.1  dholland #include <ufs/ufs/dinode.h>
     50      1.1  dholland #include <ufs/ffs/fs.h>
     51      1.1  dholland #define FSTYPENAMES
     52      1.1  dholland #include <sys/disklabel.h>
     53      1.2    martin #include <sys/disklabel_gpt.h>
     54      1.1  dholland 
     55      1.1  dholland #include <dev/scsipi/scsipi_all.h>
     56      1.1  dholland #include <sys/scsiio.h>
     57      1.1  dholland 
     58      1.1  dholland #include <dev/ata/atareg.h>
     59      1.1  dholland #include <sys/ataio.h>
     60      1.1  dholland 
     61      1.1  dholland #include "defs.h"
     62      1.1  dholland #include "md.h"
     63      1.1  dholland #include "msg_defs.h"
     64      1.1  dholland #include "menu_defs.h"
     65      1.1  dholland #include "txtwalk.h"
     66      1.1  dholland 
     67      1.1  dholland /* Disk descriptions */
     68      1.1  dholland struct disk_desc {
     69      1.1  dholland 	char	dd_name[SSTRSIZE];
     70      1.1  dholland 	char	dd_descr[70];
     71      1.1  dholland 	uint	dd_no_mbr;
     72      1.1  dholland 	uint	dd_cyl;
     73      1.1  dholland 	uint	dd_head;
     74      1.1  dholland 	uint	dd_sec;
     75      1.1  dholland 	uint	dd_secsize;
     76      1.1  dholland 	uint	dd_totsec;
     77      1.1  dholland };
     78      1.1  dholland 
     79      1.2    martin /* gpt(8) use different filesystem names.
     80      1.2    martin    So, we cant use ./common/lib/libutil/getfstypename.c */
     81      1.2    martin struct gptfs_t {
     82      1.2    martin     const char *name;
     83      1.2    martin     int id;
     84      1.2    martin     uuid_t uuid;
     85      1.2    martin };
     86      1.2    martin static const struct gptfs_t gpt_filesystems[] = {
     87      1.2    martin     { "swap", FS_SWAP, GPT_ENT_TYPE_NETBSD_SWAP, },
     88      1.2    martin     { "ffs", FS_BSDFFS, GPT_ENT_TYPE_NETBSD_FFS, },
     89      1.2    martin     { "lfs", FS_BSDLFS, GPT_ENT_TYPE_NETBSD_LFS, },
     90      1.2    martin     { "linux", FS_EX2FS, GPT_ENT_TYPE_LINUX_DATA, },
     91      1.2    martin     { "windows,", FS_MSDOS, GPT_ENT_TYPE_MS_BASIC_DATA, },
     92      1.2    martin     { "hfs", FS_HFS, GPT_ENT_TYPE_APPLE_HFS, },
     93      1.2    martin     { "ufs", FS_OTHER, GPT_ENT_TYPE_APPLE_UFS, },
     94      1.2    martin     { "ccd", FS_CCD, GPT_ENT_TYPE_NETBSD_CCD, },
     95      1.2    martin     { "raid", FS_RAID, GPT_ENT_TYPE_NETBSD_RAIDFRAME, },
     96      1.2    martin     { "cgd", FS_CGD, GPT_ENT_TYPE_NETBSD_CGD, },
     97      1.2    martin     { "efi", FS_OTHER, GPT_ENT_TYPE_EFI, },
     98      1.2    martin     { "bios", FS_OTHER, GPT_ENT_TYPE_BIOS, },
     99      1.2    martin     { NULL, -1, GPT_ENT_TYPE_UNUSED, },
    100      1.2    martin };
    101      1.2    martin 
    102      1.1  dholland /* Local prototypes */
    103      1.1  dholland static int foundffs(struct data *, size_t);
    104      1.1  dholland #ifdef USE_SYSVBFS
    105      1.1  dholland static int foundsysvbfs(struct data *, size_t);
    106      1.1  dholland #endif
    107      1.1  dholland static int fsck_preen(const char *, int, const char *);
    108      1.1  dholland static void fixsb(const char *, const char *, char);
    109  1.4.4.2       snj static bool is_gpt(const char *);
    110      1.2    martin static int incoregpt(pm_devs_t *, partinfo *);
    111      1.1  dholland 
    112      1.1  dholland #ifndef DISK_NAMES
    113      1.1  dholland #define DISK_NAMES "wd", "sd", "ld", "raid"
    114      1.1  dholland #endif
    115      1.1  dholland 
    116      1.2    martin static const char *disk_names[] = { DISK_NAMES, "vnd", "cgd", NULL };
    117      1.1  dholland 
    118      1.1  dholland static bool tmpfs_on_var_shm(void);
    119      1.1  dholland 
    120      1.1  dholland const char *
    121      1.1  dholland getfslabelname(uint8_t f)
    122      1.1  dholland {
    123      1.1  dholland 	if (f >= __arraycount(fstypenames) || fstypenames[f] == NULL)
    124      1.1  dholland 		return "invalid";
    125      1.1  dholland 	return fstypenames[f];
    126      1.1  dholland }
    127      1.1  dholland 
    128      1.1  dholland /*
    129      1.1  dholland  * Decide wether we want to mount a tmpfs on /var/shm: we do this always
    130      1.1  dholland  * when the machine has more than 16 MB of user memory. On smaller machines,
    131      1.1  dholland  * shm_open() and friends will not perform well anyway.
    132      1.1  dholland  */
    133      1.1  dholland static bool
    134      1.1  dholland tmpfs_on_var_shm()
    135      1.1  dholland {
    136      1.1  dholland 	uint64_t ram;
    137      1.1  dholland 	size_t len;
    138      1.1  dholland 
    139      1.1  dholland 	len = sizeof(ram);
    140      1.1  dholland 	if (sysctlbyname("hw.usermem64", &ram, &len, NULL, 0))
    141      1.1  dholland 		return false;
    142      1.1  dholland 
    143      1.1  dholland 	return ram > 16UL*1024UL*1024UL;
    144      1.1  dholland }
    145      1.1  dholland 
    146      1.1  dholland /* from src/sbin/atactl/atactl.c
    147      1.1  dholland  * extract_string: copy a block of bytes out of ataparams and make
    148      1.1  dholland  * a proper string out of it, truncating trailing spaces and preserving
    149      1.1  dholland  * strict typing. And also, not doing unaligned accesses.
    150      1.1  dholland  */
    151      1.1  dholland static void
    152      1.1  dholland ata_extract_string(char *buf, size_t bufmax,
    153      1.1  dholland 		   uint8_t *bytes, unsigned numbytes,
    154      1.1  dholland 		   int needswap)
    155      1.1  dholland {
    156      1.1  dholland 	unsigned i;
    157      1.1  dholland 	size_t j;
    158      1.1  dholland 	unsigned char ch1, ch2;
    159      1.1  dholland 
    160      1.1  dholland 	for (i = 0, j = 0; i < numbytes; i += 2) {
    161      1.1  dholland 		ch1 = bytes[i];
    162      1.1  dholland 		ch2 = bytes[i+1];
    163      1.1  dholland 		if (needswap && j < bufmax-1) {
    164      1.1  dholland 			buf[j++] = ch2;
    165      1.1  dholland 		}
    166      1.1  dholland 		if (j < bufmax-1) {
    167      1.1  dholland 			buf[j++] = ch1;
    168      1.1  dholland 		}
    169      1.1  dholland 		if (!needswap && j < bufmax-1) {
    170      1.1  dholland 			buf[j++] = ch2;
    171      1.1  dholland 		}
    172      1.1  dholland 	}
    173      1.1  dholland 	while (j > 0 && buf[j-1] == ' ') {
    174      1.1  dholland 		j--;
    175      1.1  dholland 	}
    176      1.1  dholland 	buf[j] = '\0';
    177      1.1  dholland }
    178      1.1  dholland 
    179      1.1  dholland /*
    180      1.1  dholland  * from src/sbin/scsictl/scsi_subr.c
    181      1.1  dholland  */
    182      1.1  dholland #define STRVIS_ISWHITE(x) ((x) == ' ' || (x) == '\0' || (x) == (u_char)'\377')
    183      1.1  dholland 
    184      1.1  dholland static void
    185      1.1  dholland scsi_strvis(char *sdst, size_t dlen, const char *ssrc, size_t slen)
    186      1.1  dholland {
    187      1.1  dholland 	u_char *dst = (u_char *)sdst;
    188      1.1  dholland 	const u_char *src = (const u_char *)ssrc;
    189      1.1  dholland 
    190      1.1  dholland 	/* Trim leading and trailing blanks and NULs. */
    191      1.1  dholland 	while (slen > 0 && STRVIS_ISWHITE(src[0]))
    192      1.1  dholland 		++src, --slen;
    193      1.1  dholland 	while (slen > 0 && STRVIS_ISWHITE(src[slen - 1]))
    194      1.1  dholland 		--slen;
    195      1.1  dholland 
    196      1.1  dholland 	while (slen > 0) {
    197      1.1  dholland 		if (*src < 0x20 || *src >= 0x80) {
    198      1.1  dholland 			/* non-printable characters */
    199      1.1  dholland 			dlen -= 4;
    200      1.1  dholland 			if (dlen < 1)
    201      1.1  dholland 				break;
    202      1.1  dholland 			*dst++ = '\\';
    203      1.1  dholland 			*dst++ = ((*src & 0300) >> 6) + '0';
    204      1.1  dholland 			*dst++ = ((*src & 0070) >> 3) + '0';
    205      1.1  dholland 			*dst++ = ((*src & 0007) >> 0) + '0';
    206      1.1  dholland 		} else if (*src == '\\') {
    207      1.1  dholland 			/* quote characters */
    208      1.1  dholland 			dlen -= 2;
    209      1.1  dholland 			if (dlen < 1)
    210      1.1  dholland 				break;
    211      1.1  dholland 			*dst++ = '\\';
    212      1.1  dholland 			*dst++ = '\\';
    213      1.1  dholland 		} else {
    214      1.1  dholland 			/* normal characters */
    215      1.1  dholland 			if (--dlen < 1)
    216      1.1  dholland 				break;
    217      1.1  dholland 			*dst++ = *src;
    218      1.1  dholland 		}
    219      1.1  dholland 		++src, --slen;
    220      1.1  dholland 	}
    221      1.1  dholland 
    222      1.1  dholland 	*dst++ = 0;
    223      1.1  dholland }
    224      1.1  dholland 
    225      1.1  dholland 
    226      1.1  dholland static int
    227      1.1  dholland get_descr_scsi(struct disk_desc *dd, int fd)
    228      1.1  dholland {
    229      1.1  dholland 	struct scsipi_inquiry_data inqbuf;
    230      1.1  dholland 	struct scsipi_inquiry cmd;
    231      1.1  dholland 	scsireq_t req;
    232      1.1  dholland         /* x4 in case every character is escaped, +1 for NUL. */
    233      1.1  dholland 	char vendor[(sizeof(inqbuf.vendor) * 4) + 1],
    234      1.1  dholland 	     product[(sizeof(inqbuf.product) * 4) + 1],
    235      1.1  dholland 	     revision[(sizeof(inqbuf.revision) * 4) + 1];
    236      1.1  dholland 	char size[5];
    237      1.1  dholland 	int error;
    238      1.1  dholland 
    239      1.1  dholland 	memset(&inqbuf, 0, sizeof(inqbuf));
    240      1.1  dholland 	memset(&cmd, 0, sizeof(cmd));
    241      1.1  dholland 	memset(&req, 0, sizeof(req));
    242      1.1  dholland 
    243      1.1  dholland 	cmd.opcode = INQUIRY;
    244      1.1  dholland 	cmd.length = sizeof(inqbuf);
    245      1.1  dholland 	memcpy(req.cmd, &cmd, sizeof(cmd));
    246      1.1  dholland 	req.cmdlen = sizeof(cmd);
    247      1.1  dholland 	req.databuf = &inqbuf;
    248      1.1  dholland 	req.datalen = sizeof(inqbuf);
    249      1.1  dholland 	req.timeout = 10000;
    250      1.1  dholland 	req.flags = SCCMD_READ;
    251      1.1  dholland 	req.senselen = SENSEBUFLEN;
    252      1.1  dholland 
    253      1.1  dholland 	error = ioctl(fd, SCIOCCOMMAND, &req);
    254      1.1  dholland 	if (error == -1 || req.retsts != SCCMD_OK)
    255      1.1  dholland 		return 0;
    256      1.1  dholland 
    257      1.1  dholland 	scsi_strvis(vendor, sizeof(vendor), inqbuf.vendor,
    258      1.1  dholland 	    sizeof(inqbuf.vendor));
    259      1.1  dholland 	scsi_strvis(product, sizeof(product), inqbuf.product,
    260      1.1  dholland 	    sizeof(inqbuf.product));
    261      1.1  dholland 	scsi_strvis(revision, sizeof(revision), inqbuf.revision,
    262      1.1  dholland 	    sizeof(inqbuf.revision));
    263      1.1  dholland 
    264      1.1  dholland 	humanize_number(size, sizeof(size),
    265      1.1  dholland 	    (uint64_t)dd->dd_secsize * (uint64_t)dd->dd_totsec,
    266      1.1  dholland 	    "", HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL);
    267      1.1  dholland 
    268      1.1  dholland 	snprintf(dd->dd_descr, sizeof(dd->dd_descr),
    269      1.1  dholland 	    "%s (%s, %s %s)",
    270      1.1  dholland 	    dd->dd_name, size, vendor, product);
    271      1.1  dholland 
    272      1.1  dholland 	return 1;
    273      1.1  dholland }
    274      1.1  dholland 
    275      1.1  dholland static int
    276      1.1  dholland get_descr_ata(struct disk_desc *dd, int fd)
    277      1.1  dholland {
    278      1.1  dholland 	struct atareq req;
    279      1.1  dholland 	static union {
    280      1.1  dholland 		unsigned char inbuf[DEV_BSIZE];
    281      1.1  dholland 		struct ataparams inqbuf;
    282      1.1  dholland 	} inbuf;
    283      1.1  dholland 	struct ataparams *inqbuf = &inbuf.inqbuf;
    284      1.1  dholland 	char model[sizeof(inqbuf->atap_model)+1];
    285      1.1  dholland 	char size[5];
    286      1.1  dholland 	int error, needswap = 0;
    287      1.1  dholland 
    288      1.1  dholland 	memset(&inbuf, 0, sizeof(inbuf));
    289      1.1  dholland 	memset(&req, 0, sizeof(req));
    290      1.1  dholland 
    291      1.1  dholland 	req.flags = ATACMD_READ;
    292      1.1  dholland 	req.command = WDCC_IDENTIFY;
    293      1.1  dholland 	req.databuf = (void *)&inbuf;
    294      1.1  dholland 	req.datalen = sizeof(inbuf);
    295      1.1  dholland 	req.timeout = 1000;
    296      1.1  dholland 
    297      1.1  dholland 	error = ioctl(fd, ATAIOCCOMMAND, &req);
    298      1.1  dholland 	if (error == -1 || req.retsts != ATACMD_OK)
    299      1.1  dholland 		return 0;
    300      1.1  dholland 
    301      1.1  dholland #if BYTE_ORDER == LITTLE_ENDIAN
    302      1.1  dholland 	/*
    303      1.1  dholland 	 * On little endian machines, we need to shuffle the string
    304      1.1  dholland 	 * byte order.  However, we don't have to do this for NEC or
    305      1.1  dholland 	 * Mitsumi ATAPI devices
    306      1.1  dholland 	 */
    307      1.1  dholland 
    308      1.1  dholland 	if (!(inqbuf->atap_config != WDC_CFG_CFA_MAGIC &&
    309      1.1  dholland 	      (inqbuf->atap_config & WDC_CFG_ATAPI) &&
    310      1.1  dholland 	      ((inqbuf->atap_model[0] == 'N' &&
    311      1.1  dholland 		  inqbuf->atap_model[1] == 'E') ||
    312      1.1  dholland 	       (inqbuf->atap_model[0] == 'F' &&
    313      1.1  dholland 		  inqbuf->atap_model[1] == 'X')))) {
    314      1.1  dholland 		needswap = 1;
    315      1.1  dholland 	}
    316      1.1  dholland #endif
    317      1.1  dholland 
    318      1.1  dholland 	ata_extract_string(model, sizeof(model),
    319      1.1  dholland 	    inqbuf->atap_model, sizeof(inqbuf->atap_model), needswap);
    320      1.1  dholland 	humanize_number(size, sizeof(size),
    321      1.1  dholland 	    (uint64_t)dd->dd_secsize * (uint64_t)dd->dd_totsec,
    322      1.1  dholland 	    "", HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL);
    323      1.1  dholland 
    324      1.1  dholland 	snprintf(dd->dd_descr, sizeof(dd->dd_descr), "%s (%s, %s)",
    325      1.1  dholland 	    dd->dd_name, size, model);
    326      1.1  dholland 
    327      1.1  dholland 	return 1;
    328      1.1  dholland }
    329      1.1  dholland 
    330      1.1  dholland static void
    331      1.1  dholland get_descr(struct disk_desc *dd)
    332      1.1  dholland {
    333      1.1  dholland 	char diskpath[MAXPATHLEN];
    334      1.1  dholland 	int fd = -1;
    335      1.1  dholland 
    336      1.1  dholland 	fd = opendisk(dd->dd_name, O_RDONLY, diskpath, sizeof(diskpath), 0);
    337      1.1  dholland 	if (fd < 0)
    338      1.1  dholland 		goto done;
    339      1.1  dholland 
    340      1.1  dholland 	dd->dd_descr[0] = '\0';
    341      1.1  dholland 
    342      1.1  dholland 	/* try ATA */
    343      1.1  dholland 	if (get_descr_ata(dd, fd))
    344      1.1  dholland 		goto done;
    345      1.1  dholland 	/* try SCSI */
    346      1.1  dholland 	if (get_descr_scsi(dd, fd))
    347      1.1  dholland 		goto done;
    348      1.2    martin 	/* XXX: get description from raid, cgd, vnd... */
    349      1.1  dholland 
    350      1.1  dholland done:
    351      1.1  dholland 	if (fd >= 0)
    352      1.1  dholland 		close(fd);
    353      1.1  dholland 	if (strlen(dd->dd_descr) == 0)
    354      1.1  dholland 		strcpy(dd->dd_descr, dd->dd_name);
    355      1.1  dholland }
    356      1.1  dholland 
    357      1.1  dholland /* disknames - contains device names without partition letters
    358      1.1  dholland  * cdrom_devices - contains devices including partition letters
    359      1.1  dholland  * returns the first entry in hw.disknames matching a cdrom_device, or
    360      1.1  dholland  * first entry on error or no match
    361      1.1  dholland  */
    362      1.1  dholland const char *
    363      1.1  dholland get_default_cdrom(void)
    364      1.1  dholland {
    365      1.1  dholland 	static const char *cdrom_devices[] = { CD_NAMES, 0};
    366      1.1  dholland 	static const char mib_name[] = "hw.disknames";
    367      1.1  dholland 	size_t len;
    368      1.1  dholland 	char *disknames;
    369      1.1  dholland 	char *last;
    370      1.1  dholland 	char *name;
    371      1.1  dholland 	const char **arg;
    372      1.1  dholland 	const char *cd_dev;
    373      1.1  dholland 
    374      1.1  dholland 	/* On error just use first entry in cdrom_devices */
    375      1.1  dholland 	if (sysctlbyname(mib_name, NULL, &len, NULL, 0) == -1)
    376      1.1  dholland 		return cdrom_devices[0];
    377      1.1  dholland 	if ((disknames = malloc(len + 2)) == 0) /* skip on malloc fail */
    378      1.1  dholland 		return cdrom_devices[0];
    379      1.1  dholland 
    380      1.1  dholland 	(void)sysctlbyname(mib_name, disknames, &len, NULL, 0);
    381      1.1  dholland         for ((name = strtok_r(disknames, " ", &last)); name;
    382      1.1  dholland 	    (name = strtok_r(NULL, " ", &last))) {
    383      1.1  dholland 		for (arg = cdrom_devices; *arg; ++arg) {
    384      1.1  dholland 			cd_dev = *arg;
    385      1.1  dholland 			/* skip unit and partition */
    386      1.1  dholland 			if (strncmp(cd_dev, name, strlen(cd_dev) - 2) != 0)
    387      1.1  dholland 				continue;
    388      1.1  dholland 			if (name != disknames)
    389      1.1  dholland 				strcpy(disknames, name);
    390      1.1  dholland 			strcat(disknames, "a");
    391      1.1  dholland 			/* XXX: leaks, but so what? */
    392      1.1  dholland 			return disknames;
    393      1.1  dholland 		}
    394      1.1  dholland 	}
    395      1.1  dholland 	free(disknames);
    396      1.1  dholland 	return cdrom_devices[0];
    397      1.1  dholland }
    398      1.1  dholland 
    399      1.1  dholland static int
    400      1.1  dholland get_disks(struct disk_desc *dd)
    401      1.1  dholland {
    402      1.1  dholland 	const char **xd;
    403      1.1  dholland 	char *cp;
    404      1.1  dholland 	struct disklabel l;
    405      1.1  dholland 	int i;
    406      1.1  dholland 	int numdisks;
    407      1.1  dholland 
    408      1.1  dholland 	/* initialize */
    409      1.1  dholland 	numdisks = 0;
    410      1.1  dholland 
    411      1.1  dholland 	for (xd = disk_names; *xd != NULL; xd++) {
    412      1.1  dholland 		for (i = 0; i < MAX_DISKS; i++) {
    413      1.1  dholland 			strlcpy(dd->dd_name, *xd, sizeof dd->dd_name - 2);
    414      1.1  dholland 			cp = strchr(dd->dd_name, ':');
    415      1.1  dholland 			if (cp != NULL)
    416      1.1  dholland 				dd->dd_no_mbr = !strcmp(cp, ":no_mbr");
    417      1.1  dholland 			else {
    418      1.1  dholland 				dd->dd_no_mbr = 0;
    419      1.1  dholland 				cp = strchr(dd->dd_name, 0);
    420      1.1  dholland 			}
    421      1.1  dholland 
    422      1.1  dholland 			snprintf(cp, 2 + 1, "%d", i);
    423      1.1  dholland 			if (!get_geom(dd->dd_name, &l)) {
    424      1.1  dholland 				if (errno == ENOENT)
    425      1.1  dholland 					break;
    426      1.1  dholland 				continue;
    427      1.1  dholland 			}
    428      1.1  dholland 
    429      1.1  dholland 			/*
    430      1.1  dholland 			 * Exclude a disk mounted as root partition,
    431      1.1  dholland 			 * in case of install-image on a USB memstick.
    432      1.1  dholland 			 */
    433      1.1  dholland 			if (is_active_rootpart(dd->dd_name, 0))
    434      1.1  dholland 				continue;
    435      1.1  dholland 
    436      1.1  dholland 			dd->dd_cyl = l.d_ncylinders;
    437      1.1  dholland 			dd->dd_head = l.d_ntracks;
    438      1.1  dholland 			dd->dd_sec = l.d_nsectors;
    439      1.1  dholland 			dd->dd_secsize = l.d_secsize;
    440      1.1  dholland 			dd->dd_totsec = l.d_secperunit;
    441      1.1  dholland 			get_descr(dd);
    442      1.1  dholland 			dd++;
    443      1.1  dholland 			numdisks++;
    444      1.1  dholland 			if (numdisks >= MAX_DISKS)
    445      1.1  dholland 				return numdisks;
    446      1.1  dholland 		}
    447      1.1  dholland 	}
    448      1.1  dholland 	return numdisks;
    449      1.1  dholland }
    450      1.1  dholland 
    451      1.1  dholland int
    452      1.1  dholland find_disks(const char *doingwhat)
    453      1.1  dholland {
    454      1.1  dholland 	struct disk_desc disks[MAX_DISKS];
    455      1.2    martin 	menu_ent dsk_menu[nelem(disks) + 1]; // + 1 for extended partitioning entry
    456      1.1  dholland 	struct disk_desc *disk;
    457      1.2    martin 	int i, already_found;
    458      1.2    martin 	int numdisks, selected_disk = -1;
    459      1.1  dholland 	int menu_no;
    460      1.2    martin 	pm_devs_t *pm_i, *pm_last = NULL;
    461      1.1  dholland 
    462      1.1  dholland 	/* Find disks. */
    463      1.1  dholland 	numdisks = get_disks(disks);
    464      1.1  dholland 
    465      1.1  dholland 	/* need a redraw here, kernel messages hose everything */
    466      1.1  dholland 	touchwin(stdscr);
    467      1.1  dholland 	refresh();
    468      1.1  dholland 	/* Kill typeahead, it won't be what the user had in mind */
    469      1.1  dholland 	fpurge(stdin);
    470      1.1  dholland 
    471      1.2    martin 	/* partman_go: <0 - we want to see menu with extended partitioning
    472      1.2    martin 				  ==0 - we want to see simple select disk menu
    473      1.2    martin 				   >0 - we do not want to see any menus, just detect all disks */
    474      1.2    martin 	if (partman_go <= 0) {
    475      1.2    martin 		if (numdisks == 0) {
    476      1.2    martin 			/* No disks found! */
    477      1.2    martin 			msg_display(MSG_nodisk);
    478      1.2    martin 			process_menu(MENU_ok, NULL);
    479      1.2    martin 			/*endwin();*/
    480      1.2    martin 			return -1;
    481      1.2    martin 		} else {
    482      1.2    martin 			/* One or more disks found! */
    483      1.2    martin 			for (i = 0; i < numdisks; i++) {
    484      1.2    martin 				dsk_menu[i].opt_name = disks[i].dd_descr;
    485      1.2    martin 				dsk_menu[i].opt_menu = OPT_NOMENU;
    486      1.2    martin 				dsk_menu[i].opt_flags = OPT_EXIT;
    487      1.2    martin 				dsk_menu[i].opt_action = set_menu_select;
    488      1.2    martin 			}
    489      1.2    martin 			if (partman_go < 0) {
    490      1.2    martin 				dsk_menu[i].opt_name = MSG_partman;
    491      1.2    martin 				dsk_menu[i].opt_menu = OPT_NOMENU;
    492      1.2    martin 				dsk_menu[i].opt_flags = OPT_EXIT;
    493      1.2    martin 				dsk_menu[i].opt_action = set_menu_select;
    494      1.2    martin 			}
    495      1.2    martin 			menu_no = new_menu(MSG_Available_disks,
    496      1.2    martin 				dsk_menu, numdisks + ((partman_go<0)?1:0), -1, 4, 0, 0, MC_SCROLL,
    497      1.2    martin 				NULL, NULL, NULL, NULL, NULL);
    498      1.2    martin 			if (menu_no == -1)
    499      1.2    martin 				return -1;
    500      1.2    martin 			msg_display(MSG_ask_disk, doingwhat);
    501      1.2    martin 			process_menu(menu_no, &selected_disk);
    502      1.2    martin 			free_menu(menu_no);
    503      1.2    martin 		}
    504      1.2    martin 		if (partman_go < 0 && selected_disk == numdisks) {
    505      1.2    martin 			partman_go = 1;
    506      1.2    martin 	    	return -2;
    507      1.2    martin 		} else
    508      1.2    martin 			partman_go = 0;
    509      1.2    martin 		if (selected_disk < 0 || selected_disk >= numdisks)
    510      1.2    martin 	    	return -1;
    511      1.2    martin 	}
    512      1.2    martin 
    513      1.2    martin 	/* Fill pm struct with device(s) info */
    514      1.2    martin 	for (i = 0; i < numdisks; i++) {
    515      1.2    martin 		if (! partman_go)
    516      1.2    martin 			disk = disks + selected_disk;
    517      1.2    martin 		else {
    518      1.2    martin 			disk = disks + i;
    519      1.2    martin 			already_found = 0;
    520      1.2    martin 			SLIST_FOREACH(pm_i, &pm_head, l) {
    521      1.2    martin 				pm_last = pm_i;
    522      1.2    martin 				if (!already_found &&
    523      1.2    martin 						strcmp(pm_i->diskdev, disk->dd_name) == 0) {
    524      1.2    martin 					pm_i->found = 1;
    525      1.2    martin 					break;
    526      1.2    martin 				}
    527      1.2    martin 			}
    528      1.2    martin 			if (pm_i != NULL && pm_i->found)
    529      1.2    martin 				/* We already added this device, skipping */
    530      1.2    martin 				continue;
    531      1.1  dholland 		}
    532      1.2    martin 		pm = pm_new;
    533      1.2    martin 		pm->found = 1;
    534      1.2    martin 		pm->bootable = 0;
    535      1.2    martin 		pm->pi.menu_no = -1;
    536      1.2    martin 		pm->disktype = "unknown";
    537      1.2    martin 		pm->doessf = "";
    538      1.2    martin 		strlcpy(pm->diskdev, disk->dd_name, sizeof pm->diskdev);
    539      1.2    martin 		strlcpy(pm->diskdev_descr, disk->dd_descr, sizeof pm->diskdev_descr);
    540      1.2    martin 		/* Use as a default disk if the user has the sets on a local disk */
    541      1.2    martin 		strlcpy(localfs_dev, disk->dd_name, sizeof localfs_dev);
    542      1.2    martin 
    543      1.2    martin 		pm->gpt = is_gpt(pm->diskdev);
    544      1.2    martin 		pm->no_mbr = disk->dd_no_mbr || pm->gpt;
    545      1.2    martin 		pm->sectorsize = disk->dd_secsize;
    546      1.2    martin 		pm->dlcyl = disk->dd_cyl;
    547      1.2    martin 		pm->dlhead = disk->dd_head;
    548      1.2    martin 		pm->dlsec = disk->dd_sec;
    549      1.2    martin 		pm->dlsize = disk->dd_totsec;
    550      1.2    martin 		if (pm->dlsize == 0)
    551      1.2    martin 			pm->dlsize = disk->dd_cyl * disk->dd_head * disk->dd_sec;
    552      1.2    martin 		if (pm->dlsize > UINT32_MAX && ! partman_go) {
    553      1.2    martin 			if (logfp)
    554      1.2    martin 				fprintf(logfp, "Cannot process disk %s: too big size (%d)\n",
    555      1.2    martin 					pm->diskdev, (int)pm->dlsize);
    556      1.2    martin 			msg_display(MSG_toobigdisklabel);
    557      1.2    martin 			process_menu(MENU_ok, NULL);
    558      1.1  dholland 			return -1;
    559      1.2    martin 		}
    560      1.2    martin 		pm->dlcylsize = pm->dlhead * pm->dlsec;
    561      1.2    martin 
    562      1.2    martin 		label_read();
    563      1.2    martin 		if (partman_go) {
    564      1.2    martin 			pm_getrefdev(pm_new);
    565      1.2    martin 			if (SLIST_EMPTY(&pm_head) || pm_last == NULL)
    566      1.2    martin 				 SLIST_INSERT_HEAD(&pm_head, pm_new, l);
    567      1.2    martin 			else
    568      1.2    martin 				 SLIST_INSERT_AFTER(pm_last, pm_new, l);
    569      1.2    martin 			pm_new = malloc(sizeof (pm_devs_t));
    570      1.2    martin 			memset(pm_new, 0, sizeof *pm_new);
    571      1.2    martin 		} else
    572      1.2    martin 			/* We is not in partman and do not want to process all devices, exit */
    573      1.2    martin 			break;
    574      1.1  dholland 	}
    575      1.1  dholland 
    576      1.2    martin 	return numdisks;
    577      1.2    martin }
    578      1.2    martin 
    579  1.4.4.2       snj 
    580      1.2    martin void
    581      1.2    martin label_read(void)
    582      1.2    martin {
    583  1.4.4.4   msaitoh 	check_available_binaries();
    584  1.4.4.4   msaitoh 
    585      1.1  dholland 	/* Get existing/default label */
    586      1.2    martin 	memset(&pm->oldlabel, 0, sizeof pm->oldlabel);
    587  1.4.4.4   msaitoh 	if (!have_gpt || !pm->gpt)
    588      1.2    martin 		incorelabel(pm->diskdev, pm->oldlabel);
    589      1.2    martin 	else
    590      1.2    martin 		incoregpt(pm, pm->oldlabel);
    591      1.1  dholland 	/* Set 'target' label to current label in case we don't change it */
    592      1.2    martin 	memcpy(&pm->bsdlabel, &pm->oldlabel, sizeof pm->bsdlabel);
    593      1.2    martin #ifndef NO_DISKLABEL
    594      1.2    martin 	if (! pm->gpt)
    595      1.2    martin 		savenewlabel(pm->oldlabel, getmaxpartitions());
    596      1.2    martin #endif
    597      1.1  dholland }
    598      1.1  dholland 
    599      1.1  dholland void
    600      1.1  dholland fmt_fspart(menudesc *m, int ptn, void *arg)
    601      1.1  dholland {
    602      1.1  dholland 	unsigned int poffset, psize, pend;
    603      1.1  dholland 	const char *desc;
    604      1.1  dholland 	static const char *Yes;
    605      1.2    martin 	partinfo *p = pm->bsdlabel + ptn;
    606      1.1  dholland 
    607      1.1  dholland 	if (Yes == NULL)
    608      1.1  dholland 		Yes = msg_string(MSG_Yes);
    609      1.1  dholland 
    610      1.1  dholland 	poffset = p->pi_offset / sizemult;
    611      1.1  dholland 	psize = p->pi_size / sizemult;
    612      1.1  dholland 	if (psize == 0)
    613      1.1  dholland 		pend = 0;
    614      1.1  dholland 	else
    615      1.1  dholland 		pend = (p->pi_offset + p->pi_size) / sizemult - 1;
    616      1.1  dholland 
    617      1.1  dholland 	if (p->pi_fstype == FS_BSDFFS)
    618      1.1  dholland 		if (p->pi_flags & PIF_FFSv2)
    619      1.1  dholland 			desc = "FFSv2";
    620      1.1  dholland 		else
    621      1.1  dholland 			desc = "FFSv1";
    622      1.1  dholland 	else
    623      1.1  dholland 		desc = getfslabelname(p->pi_fstype);
    624      1.1  dholland 
    625      1.1  dholland #ifdef PART_BOOT
    626      1.1  dholland 	if (ptn == PART_BOOT)
    627      1.1  dholland 		desc = msg_string(MSG_Boot_partition_cant_change);
    628      1.1  dholland #endif
    629      1.1  dholland 	if (ptn == getrawpartition())
    630      1.1  dholland 		desc = msg_string(MSG_Whole_disk_cant_change);
    631      1.1  dholland 	else {
    632      1.1  dholland 		if (ptn == PART_C)
    633      1.1  dholland 			desc = msg_string(MSG_NetBSD_partition_cant_change);
    634      1.1  dholland 	}
    635      1.1  dholland 
    636      1.1  dholland 	wprintw(m->mw, msg_string(MSG_fspart_row),
    637      1.1  dholland 			poffset, pend, psize, desc,
    638      1.1  dholland 			p->pi_flags & PIF_NEWFS ? Yes : "",
    639      1.1  dholland 			p->pi_flags & PIF_MOUNT ? Yes : "",
    640      1.1  dholland 			p->pi_mount);
    641      1.1  dholland }
    642      1.1  dholland 
    643      1.1  dholland /*
    644      1.1  dholland  * Label a disk using an MD-specific string DISKLABEL_CMD for
    645      1.1  dholland  * to invoke disklabel.
    646      1.1  dholland  * if MD code does not define DISKLABEL_CMD, this is a no-op.
    647      1.1  dholland  *
    648      1.1  dholland  * i386 port uses "/sbin/disklabel -w -r", just like i386
    649      1.1  dholland  * miniroot scripts, though this may leave a bogus incore label.
    650      1.1  dholland  *
    651      1.1  dholland  * Sun ports should use DISKLABEL_CMD "/sbin/disklabel -w"
    652      1.1  dholland  * to get incore to ondisk inode translation for the Sun proms.
    653      1.1  dholland  */
    654      1.1  dholland int
    655      1.1  dholland write_disklabel (void)
    656      1.1  dholland {
    657  1.4.4.4   msaitoh 	int rv = 0;
    658      1.1  dholland 
    659      1.1  dholland #ifdef DISKLABEL_CMD
    660      1.1  dholland 	/* disklabel the disk */
    661  1.4.4.4   msaitoh 	rv = run_program(RUN_DISPLAY, "%s -f /tmp/disktab %s '%s'",
    662      1.2    martin 	    DISKLABEL_CMD, pm->diskdev, pm->bsddiskname);
    663  1.4.4.4   msaitoh 	if (rv == 0)
    664  1.4.4.4   msaitoh 		update_wedges(pm->diskdev);
    665      1.1  dholland #endif
    666  1.4.4.4   msaitoh 	return rv;
    667      1.1  dholland }
    668      1.1  dholland 
    669      1.1  dholland 
    670      1.1  dholland static int
    671      1.1  dholland ptn_sort(const void *a, const void *b)
    672      1.1  dholland {
    673      1.2    martin 	return strcmp(pm->bsdlabel[*(const int *)a].pi_mount,
    674      1.2    martin 		      pm->bsdlabel[*(const int *)b].pi_mount);
    675      1.1  dholland }
    676      1.1  dholland 
    677      1.1  dholland int
    678      1.1  dholland make_filesystems(void)
    679      1.1  dholland {
    680      1.1  dholland 	unsigned int i;
    681      1.1  dholland 	int ptn;
    682      1.2    martin 	int ptn_order[nelem(pm->bsdlabel)];
    683      1.1  dholland 	int error = 0;
    684      1.1  dholland 	unsigned int maxpart = getmaxpartitions();
    685      1.2    martin 	char *newfs = NULL, *dev = NULL, *devdev = NULL;
    686      1.1  dholland 	partinfo *lbl;
    687      1.1  dholland 
    688      1.2    martin 	if (maxpart > nelem(pm->bsdlabel))
    689      1.2    martin 		maxpart = nelem(pm->bsdlabel);
    690      1.1  dholland 
    691      1.1  dholland 	/* Making new file systems and mounting them */
    692      1.1  dholland 
    693      1.1  dholland 	/* sort to ensure /usr/local is mounted after /usr (etc) */
    694      1.1  dholland 	for (i = 0; i < maxpart; i++)
    695      1.1  dholland 		ptn_order[i] = i;
    696      1.1  dholland 	qsort(ptn_order, maxpart, sizeof ptn_order[0], ptn_sort);
    697      1.1  dholland 
    698      1.1  dholland 	for (i = 0; i < maxpart; i++) {
    699      1.1  dholland 		/*
    700      1.1  dholland 		 * newfs and mount. For now, process only BSD filesystems.
    701      1.1  dholland 		 * but if this is the mounted-on root, has no mount
    702      1.1  dholland 		 * point defined, or is marked preserve, don't touch it!
    703      1.1  dholland 		 */
    704      1.1  dholland 		ptn = ptn_order[i];
    705      1.2    martin 		lbl = pm->bsdlabel + ptn;
    706      1.1  dholland 
    707      1.2    martin 		if (is_active_rootpart(pm->diskdev, ptn))
    708      1.1  dholland 			continue;
    709      1.1  dholland 
    710      1.1  dholland 		if (*lbl->pi_mount == 0)
    711      1.1  dholland 			/* No mount point */
    712      1.1  dholland 			continue;
    713      1.1  dholland 
    714      1.2    martin 		if (pm->isspecial) {
    715      1.2    martin 			asprintf(&dev, "%s", pm->diskdev);
    716      1.2    martin 			ptn = 0 - 'a';
    717      1.2    martin 		} else {
    718      1.2    martin 			asprintf(&dev, "%s%c", pm->diskdev, 'a' + ptn);
    719      1.2    martin 		}
    720      1.2    martin 		if (dev == NULL)
    721      1.2    martin 			return (ENOMEM);
    722      1.2    martin 		asprintf(&devdev, "/dev/%s", dev);
    723      1.2    martin 		if (devdev == NULL)
    724      1.2    martin 			return (ENOMEM);
    725      1.2    martin 
    726      1.1  dholland 		newfs = NULL;
    727      1.2    martin 		lbl->mnt_opts = NULL;
    728      1.2    martin 		lbl->fsname = NULL;
    729      1.1  dholland 		switch (lbl->pi_fstype) {
    730      1.1  dholland 		case FS_APPLEUFS:
    731      1.1  dholland 			asprintf(&newfs, "/sbin/newfs %s%.0d",
    732      1.1  dholland 				lbl->pi_isize != 0 ? "-i" : "", lbl->pi_isize);
    733      1.2    martin 			lbl->mnt_opts = "-tffs -o async";
    734      1.2    martin 			lbl->fsname = "ffs";
    735      1.1  dholland 			break;
    736      1.1  dholland 		case FS_BSDFFS:
    737      1.1  dholland 			asprintf(&newfs,
    738      1.1  dholland 			    "/sbin/newfs -V2 -O %d -b %d -f %d%s%.0d",
    739      1.1  dholland 			    lbl->pi_flags & PIF_FFSv2 ? 2 : 1,
    740      1.1  dholland 			    lbl->pi_fsize * lbl->pi_frag, lbl->pi_fsize,
    741      1.1  dholland 			    lbl->pi_isize != 0 ? " -i " : "", lbl->pi_isize);
    742      1.1  dholland 			if (lbl->pi_flags & PIF_LOG)
    743      1.2    martin 				lbl->mnt_opts = "-tffs -o log";
    744      1.1  dholland 			else
    745      1.2    martin 				lbl->mnt_opts = "-tffs -o async";
    746      1.2    martin 			lbl->fsname = "ffs";
    747      1.1  dholland 			break;
    748      1.1  dholland 		case FS_BSDLFS:
    749      1.1  dholland 			asprintf(&newfs, "/sbin/newfs_lfs -b %d",
    750      1.1  dholland 				lbl->pi_fsize * lbl->pi_frag);
    751      1.2    martin 			lbl->mnt_opts = "-tlfs";
    752      1.2    martin 			lbl->fsname = "lfs";
    753      1.1  dholland 			break;
    754      1.1  dholland 		case FS_MSDOS:
    755      1.1  dholland #ifdef USE_NEWFS_MSDOS
    756      1.1  dholland 			asprintf(&newfs, "/sbin/newfs_msdos");
    757      1.1  dholland #endif
    758      1.2    martin 			lbl->mnt_opts = "-tmsdos";
    759      1.2    martin 			lbl->fsname = "msdos";
    760      1.1  dholland 			break;
    761      1.1  dholland #ifdef USE_SYSVBFS
    762      1.1  dholland 		case FS_SYSVBFS:
    763      1.1  dholland 			asprintf(&newfs, "/sbin/newfs_sysvbfs");
    764      1.2    martin 			lbl->mnt_opts = "-tsysvbfs";
    765      1.2    martin 			lbl->fsname = "sysvbfs";
    766      1.1  dholland 			break;
    767      1.1  dholland #endif
    768      1.1  dholland #ifdef USE_EXT2FS
    769      1.1  dholland 		case FS_EX2FS:
    770      1.1  dholland 			asprintf(&newfs, "/sbin/newfs_ext2fs");
    771      1.2    martin 			lbl->mnt_opts = "-text2fs";
    772      1.2    martin 			lbl->fsname = "ext2fs";
    773      1.1  dholland 			break;
    774      1.1  dholland #endif
    775      1.1  dholland 		}
    776      1.1  dholland 		if (lbl->pi_flags & PIF_NEWFS && newfs != NULL) {
    777      1.1  dholland #ifdef USE_NEWFS_MSDOS
    778      1.1  dholland 			if (lbl->pi_fstype == FS_MSDOS) {
    779      1.1  dholland 			        /* newfs only if mount fails */
    780      1.1  dholland 			        if (run_program(RUN_SILENT | RUN_ERROR_OK,
    781      1.2    martin 				    "mount -rt msdos /dev/%s /mnt2", dev) != 0)
    782      1.1  dholland 					error = run_program(
    783      1.1  dholland 					    RUN_DISPLAY | RUN_PROGRESS,
    784      1.2    martin 					    "%s /dev/r%s",
    785      1.2    martin 					    newfs, dev);
    786      1.1  dholland 				else {
    787      1.2    martin 			        run_program(RUN_SILENT | RUN_ERROR_OK,
    788      1.1  dholland 					    "umount /mnt2");
    789      1.1  dholland 					error = 0;
    790      1.1  dholland 				}
    791      1.1  dholland 			} else
    792      1.1  dholland #endif
    793      1.1  dholland 			error = run_program(RUN_DISPLAY | RUN_PROGRESS,
    794      1.2    martin 			    "%s /dev/r%s", newfs, dev);
    795      1.1  dholland 		} else {
    796      1.1  dholland 			/* We'd better check it isn't dirty */
    797      1.2    martin 			error = fsck_preen(pm->diskdev, ptn, lbl->fsname);
    798      1.1  dholland 		}
    799      1.1  dholland 		free(newfs);
    800      1.1  dholland 		if (error != 0)
    801      1.1  dholland 			return error;
    802      1.1  dholland 
    803      1.2    martin 		lbl->pi_flags ^= PIF_NEWFS;
    804      1.1  dholland 		md_pre_mount();
    805      1.1  dholland 
    806      1.2    martin 		if (partman_go == 0 && lbl->pi_flags & PIF_MOUNT &&
    807      1.2    martin 				lbl->mnt_opts != NULL) {
    808      1.1  dholland 			make_target_dir(lbl->pi_mount);
    809      1.2    martin 			error = target_mount_do(lbl->mnt_opts, devdev, lbl->pi_mount);
    810      1.1  dholland 			if (error) {
    811      1.2    martin 				msg_display(MSG_mountfail, dev, ' ', lbl->pi_mount);
    812      1.1  dholland 				process_menu(MENU_ok, NULL);
    813      1.1  dholland 				return error;
    814      1.1  dholland 			}
    815      1.1  dholland 		}
    816      1.2    martin 		free(devdev);
    817      1.2    martin 		free(dev);
    818      1.1  dholland 	}
    819      1.1  dholland 	return 0;
    820      1.1  dholland }
    821      1.1  dholland 
    822      1.1  dholland int
    823      1.1  dholland make_fstab(void)
    824      1.1  dholland {
    825      1.1  dholland 	FILE *f;
    826      1.1  dholland 	int i, swap_dev = -1;
    827      1.1  dholland 	const char *dump_dev;
    828      1.2    martin 	char *dev = NULL;
    829      1.2    martin 	pm_devs_t *pm_i;
    830      1.2    martin #ifndef HAVE_TMPFS
    831      1.3       mrg 	pm_devs_t *pm_with_swap = NULL;
    832      1.2    martin #endif
    833      1.1  dholland 
    834      1.1  dholland 	/* Create the fstab. */
    835      1.1  dholland 	make_target_dir("/etc");
    836      1.1  dholland 	f = target_fopen("/etc/fstab", "w");
    837      1.2    martin 	scripting_fprintf(NULL, "cat <<EOF >%s/etc/fstab\n", target_prefix());
    838      1.2    martin 
    839      1.1  dholland 	if (logfp)
    840      1.1  dholland 		(void)fprintf(logfp,
    841      1.2    martin 		    "Making %s/etc/fstab (%s).\n", target_prefix(), pm->diskdev);
    842      1.2    martin 
    843      1.1  dholland 	if (f == NULL) {
    844      1.1  dholland 		msg_display(MSG_createfstab);
    845      1.1  dholland 		if (logfp)
    846      1.1  dholland 			(void)fprintf(logfp, "Failed to make /etc/fstab!\n");
    847      1.1  dholland 		process_menu(MENU_ok, NULL);
    848      1.2    martin #ifndef DEBUG
    849      1.1  dholland 		return 1;
    850      1.1  dholland #else
    851      1.1  dholland 		f = stdout;
    852      1.1  dholland #endif
    853      1.1  dholland 	}
    854      1.1  dholland 
    855      1.1  dholland 	scripting_fprintf(f, "# NetBSD %s/etc/fstab\n# See /usr/share/examples/"
    856      1.2    martin 			"fstab/ for more examples.\n", target_prefix());
    857      1.2    martin 	if (! partman_go) {
    858      1.2    martin 		/* We want to process only one disk... */
    859      1.2    martin 		pm_i = pm;
    860      1.2    martin 		goto onlyonediskinfstab;
    861      1.2    martin 	}
    862      1.2    martin 	SLIST_FOREACH(pm_i, &pm_head, l) {
    863      1.2    martin 		onlyonediskinfstab:
    864      1.2    martin 		for (i = 0; i < getmaxpartitions(); i++) {
    865      1.2    martin 			const char *s = "";
    866      1.2    martin 			const char *mp = pm_i->bsdlabel[i].pi_mount;
    867      1.2    martin 			const char *fstype = "ffs";
    868      1.2    martin 			int fsck_pass = 0, dump_freq = 0;
    869      1.2    martin 
    870      1.2    martin 			if (dev != NULL)
    871      1.2    martin 				free(dev);
    872      1.2    martin 			if (pm_i->isspecial)
    873      1.2    martin 				asprintf(&dev, "%s", pm_i->diskdev);
    874      1.2    martin 			else
    875      1.2    martin 				asprintf(&dev, "%s%c", pm_i->diskdev, 'a' + i);
    876      1.2    martin 			if (dev == NULL)
    877      1.2    martin 				return (ENOMEM);
    878      1.2    martin 
    879      1.2    martin 			if (!*mp) {
    880      1.2    martin 				/*
    881      1.2    martin 				 * No mount point specified, comment out line and
    882      1.2    martin 				 * use /mnt as a placeholder for the mount point.
    883      1.2    martin 				 */
    884      1.1  dholland 				s = "# ";
    885      1.2    martin 				mp = "/mnt";
    886      1.1  dholland 			}
    887      1.2    martin 
    888      1.2    martin 			switch (pm_i->bsdlabel[i].pi_fstype) {
    889      1.2    martin 			case FS_UNUSED:
    890      1.2    martin 				continue;
    891      1.2    martin 			case FS_BSDLFS:
    892      1.2    martin 				/* If there is no LFS, just comment it out. */
    893      1.2    martin 				if (!check_lfs_progs())
    894      1.2    martin 					s = "# ";
    895      1.2    martin 				fstype = "lfs";
    896      1.2    martin 				/* FALLTHROUGH */
    897      1.2    martin 			case FS_BSDFFS:
    898      1.2    martin 				fsck_pass = (strcmp(mp, "/") == 0) ? 1 : 2;
    899      1.2    martin 				dump_freq = 1;
    900      1.2    martin 				break;
    901      1.2    martin 			case FS_MSDOS:
    902      1.2    martin 				fstype = "msdos";
    903      1.2    martin 				break;
    904      1.2    martin 			case FS_SWAP:
    905      1.2    martin 				if (pm_i->isspecial)
    906      1.2    martin 					continue;
    907      1.2    martin 				if (swap_dev == -1) {
    908      1.2    martin 					swap_dev = i;
    909      1.2    martin 					dump_dev = ",dp";
    910      1.2    martin #ifndef HAVE_TMPFS
    911      1.2    martin 					pm_with_swap = pm_i;
    912      1.2    martin #endif
    913      1.2    martin 				} else {
    914      1.2    martin 					dump_dev = "";
    915      1.2    martin 				}
    916      1.2    martin 				scripting_fprintf(f, "/dev/%s\t\tnone\tswap\tsw%s\t\t 0 0\n",
    917      1.2    martin 					dev, dump_dev);
    918      1.2    martin 				continue;
    919      1.1  dholland #ifdef USE_SYSVBFS
    920      1.2    martin 			case FS_SYSVBFS:
    921      1.2    martin 				fstype = "sysvbfs";
    922      1.2    martin 				make_target_dir("/stand");
    923      1.2    martin 				break;
    924      1.1  dholland #endif
    925      1.2    martin 			default:
    926      1.2    martin 				fstype = "???";
    927      1.2    martin 				s = "# ";
    928      1.2    martin 				break;
    929      1.2    martin 			}
    930      1.2    martin 			/* The code that remounts root rw doesn't check the partition */
    931      1.2    martin 			if (strcmp(mp, "/") == 0 && !(pm_i->bsdlabel[i].pi_flags & PIF_MOUNT))
    932      1.2    martin 				s = "# ";
    933      1.2    martin 
    934      1.2    martin 	 		scripting_fprintf(f,
    935      1.2    martin 			  "%s/dev/%s\t\t%s\t%s\trw%s%s%s%s%s%s%s%s\t\t %d %d\n",
    936      1.2    martin 			   s, dev, mp, fstype,
    937      1.2    martin 			   pm_i->bsdlabel[i].pi_flags & PIF_LOG ? ",log" : "",
    938      1.2    martin 			   pm_i->bsdlabel[i].pi_flags & PIF_MOUNT ? "" : ",noauto",
    939      1.2    martin 			   pm_i->bsdlabel[i].pi_flags & PIF_ASYNC ? ",async" : "",
    940      1.2    martin 			   pm_i->bsdlabel[i].pi_flags & PIF_NOATIME ? ",noatime" : "",
    941      1.2    martin 			   pm_i->bsdlabel[i].pi_flags & PIF_NODEV ? ",nodev" : "",
    942      1.2    martin 			   pm_i->bsdlabel[i].pi_flags & PIF_NODEVMTIME ? ",nodevmtime" : "",
    943      1.2    martin 			   pm_i->bsdlabel[i].pi_flags & PIF_NOEXEC ? ",noexec" : "",
    944      1.2    martin 			   pm_i->bsdlabel[i].pi_flags & PIF_NOSUID ? ",nosuid" : "",
    945      1.2    martin 			   dump_freq, fsck_pass);
    946      1.2    martin 	 		if (pm_i->isspecial)
    947      1.2    martin 	 			/* Special device (such as dk*) have only one partition */
    948      1.2    martin 	 			break;
    949      1.2    martin 		}
    950      1.2    martin 		/* Simple install, only one disk */
    951      1.2    martin 		if (!partman_go)
    952      1.1  dholland 			break;
    953      1.1  dholland 	}
    954      1.1  dholland 	if (tmp_ramdisk_size != 0) {
    955      1.1  dholland #ifdef HAVE_TMPFS
    956      1.1  dholland 		scripting_fprintf(f, "tmpfs\t\t/tmp\ttmpfs\trw,-m=1777,-s=%"
    957      1.1  dholland 		    PRIi64 "\n",
    958      1.1  dholland 		    tmp_ramdisk_size * 512);
    959      1.1  dholland #else
    960      1.2    martin 		if (swap_dev != -1 && pm_with_swap != NULL)
    961      1.1  dholland 			scripting_fprintf(f, "/dev/%s%c\t\t/tmp\tmfs\trw,-s=%"
    962      1.1  dholland 			    PRIi64 "\n",
    963      1.2    martin 			    pm_with_swap->diskdev, 'a' + swap_dev, tmp_ramdisk_size);
    964      1.1  dholland 		else
    965      1.1  dholland 			scripting_fprintf(f, "swap\t\t/tmp\tmfs\trw,-s=%"
    966      1.1  dholland 			    PRIi64 "\n",
    967      1.1  dholland 			    tmp_ramdisk_size);
    968      1.1  dholland #endif
    969      1.1  dholland 	}
    970      1.1  dholland 
    971      1.1  dholland 	/* Add /kern, /proc and /dev/pts to fstab and make mountpoint. */
    972      1.1  dholland 	scripting_fprintf(f, "kernfs\t\t/kern\tkernfs\trw\n");
    973      1.1  dholland 	scripting_fprintf(f, "ptyfs\t\t/dev/pts\tptyfs\trw\n");
    974      1.1  dholland 	scripting_fprintf(f, "procfs\t\t/proc\tprocfs\trw\n");
    975      1.1  dholland 	scripting_fprintf(f, "/dev/%s\t\t/cdrom\tcd9660\tro,noauto\n",
    976      1.1  dholland 	    get_default_cdrom());
    977      1.1  dholland 	scripting_fprintf(f, "%stmpfs\t\t/var/shm\ttmpfs\trw,-m1777,-sram%%25\n",
    978      1.1  dholland 	    tmpfs_on_var_shm() ? "" : "#");
    979      1.1  dholland 	make_target_dir("/kern");
    980      1.1  dholland 	make_target_dir("/proc");
    981      1.1  dholland 	make_target_dir("/dev/pts");
    982      1.1  dholland 	make_target_dir("/cdrom");
    983      1.1  dholland 	make_target_dir("/var/shm");
    984      1.1  dholland 
    985      1.1  dholland 	scripting_fprintf(NULL, "EOF\n");
    986      1.1  dholland 
    987      1.2    martin 	if (dev != NULL)
    988      1.2    martin 		free(dev);
    989      1.1  dholland 	fclose(f);
    990      1.1  dholland 	fflush(NULL);
    991      1.1  dholland 	return 0;
    992      1.1  dholland }
    993      1.1  dholland 
    994      1.1  dholland 
    995      1.1  dholland 
    996      1.1  dholland static int
    997      1.1  dholland /*ARGSUSED*/
    998      1.1  dholland foundffs(struct data *list, size_t num)
    999      1.1  dholland {
   1000      1.1  dholland 	int error;
   1001      1.1  dholland 
   1002      1.1  dholland 	if (num < 2 || strcmp(list[1].u.s_val, "/") == 0 ||
   1003      1.1  dholland 	    strstr(list[2].u.s_val, "noauto") != NULL)
   1004      1.1  dholland 		return 0;
   1005      1.1  dholland 
   1006      1.1  dholland 	error = fsck_preen(list[0].u.s_val, ' '-'a', "ffs");
   1007      1.1  dholland 	if (error != 0)
   1008      1.1  dholland 		return error;
   1009      1.1  dholland 
   1010      1.1  dholland 	error = target_mount("", list[0].u.s_val, ' '-'a', list[1].u.s_val);
   1011      1.1  dholland 	if (error != 0) {
   1012      1.1  dholland 		msg_display(MSG_mount_failed, list[0].u.s_val);
   1013      1.1  dholland 		process_menu(MENU_noyes, NULL);
   1014      1.1  dholland 		if (!yesno)
   1015      1.1  dholland 			return error;
   1016      1.1  dholland 	}
   1017      1.1  dholland 	return 0;
   1018      1.1  dholland }
   1019      1.1  dholland 
   1020      1.1  dholland #ifdef USE_SYSVBFS
   1021      1.1  dholland static int
   1022      1.1  dholland /*ARGSUSED*/
   1023      1.1  dholland foundsysvbfs(struct data *list, size_t num)
   1024      1.1  dholland {
   1025      1.1  dholland 	int error;
   1026      1.1  dholland 
   1027      1.1  dholland 	if (num < 2 || strcmp(list[1].u.s_val, "/") == 0 ||
   1028      1.1  dholland 	    strstr(list[2].u.s_val, "noauto") != NULL)
   1029      1.1  dholland 		return 0;
   1030      1.1  dholland 
   1031      1.1  dholland 	error = target_mount("", list[0].u.s_val, ' '-'a', list[1].u.s_val);
   1032      1.1  dholland 	if (error != 0)
   1033      1.1  dholland 		return error;
   1034      1.1  dholland 	return 0;
   1035      1.1  dholland }
   1036      1.1  dholland #endif
   1037      1.1  dholland 
   1038      1.1  dholland /*
   1039      1.1  dholland  * Do an fsck. On failure, inform the user by showing a warning
   1040      1.1  dholland  * message and doing menu_ok() before proceeding.
   1041      1.1  dholland  * Returns 0 on success, or nonzero return code from fsck() on failure.
   1042      1.1  dholland  */
   1043      1.1  dholland static int
   1044      1.1  dholland fsck_preen(const char *disk, int ptn, const char *fsname)
   1045      1.1  dholland {
   1046      1.1  dholland 	char *prog;
   1047      1.1  dholland 	int error;
   1048      1.1  dholland 
   1049      1.2    martin 	ptn = (ptn < 0)? 0 : 'a' + ptn;
   1050      1.1  dholland 	if (fsname == NULL)
   1051      1.1  dholland 		return 0;
   1052      1.1  dholland 	/* first, check if fsck program exists, if not, assume ok */
   1053      1.1  dholland 	asprintf(&prog, "/sbin/fsck_%s", fsname);
   1054      1.1  dholland 	if (prog == NULL)
   1055      1.1  dholland 		return 0;
   1056      1.1  dholland 	if (access(prog, X_OK) != 0)
   1057      1.1  dholland 		return 0;
   1058      1.1  dholland 	if (!strcmp(fsname,"ffs"))
   1059      1.1  dholland 		fixsb(prog, disk, ptn);
   1060      1.1  dholland 	error = run_program(0, "%s -p -q /dev/r%s%c", prog, disk, ptn);
   1061      1.1  dholland 	free(prog);
   1062      1.1  dholland 	if (error != 0) {
   1063      1.1  dholland 		msg_display(MSG_badfs, disk, ptn, error);
   1064      1.1  dholland 		process_menu(MENU_noyes, NULL);
   1065      1.1  dholland 		if (yesno)
   1066      1.1  dholland 			error = 0;
   1067      1.1  dholland 		/* XXX at this point maybe we should run a full fsck? */
   1068      1.1  dholland 	}
   1069      1.1  dholland 	return error;
   1070      1.1  dholland }
   1071      1.1  dholland 
   1072      1.1  dholland /* This performs the same function as the etc/rc.d/fixsb script
   1073      1.1  dholland  * which attempts to correct problems with ffs1 filesystems
   1074      1.1  dholland  * which may have been introduced by booting a netbsd-current kernel
   1075      1.1  dholland  * from between April of 2003 and January 2004. For more information
   1076      1.1  dholland  * This script was developed as a response to NetBSD pr install/25138
   1077      1.1  dholland  * Additional prs regarding the original issue include:
   1078      1.1  dholland  *  bin/17910 kern/21283 kern/21404 port-macppc/23925 port-macppc/23926
   1079      1.1  dholland  */
   1080      1.1  dholland static void
   1081      1.1  dholland fixsb(const char *prog, const char *disk, char ptn)
   1082      1.1  dholland {
   1083      1.1  dholland 	int fd;
   1084      1.1  dholland 	int rval;
   1085      1.1  dholland 	union {
   1086      1.1  dholland 		struct fs fs;
   1087      1.1  dholland 		char buf[SBLOCKSIZE];
   1088      1.1  dholland 	} sblk;
   1089      1.1  dholland 	struct fs *fs = &sblk.fs;
   1090      1.1  dholland 
   1091      1.1  dholland 	snprintf(sblk.buf, sizeof(sblk.buf), "/dev/r%s%c",
   1092      1.1  dholland 		disk, ptn == ' ' ? 0 : ptn);
   1093      1.1  dholland 	fd = open(sblk.buf, O_RDONLY);
   1094      1.1  dholland 	if (fd == -1)
   1095      1.1  dholland 		return;
   1096      1.1  dholland 
   1097      1.1  dholland 	/* Read ffsv1 main superblock */
   1098      1.1  dholland 	rval = pread(fd, sblk.buf, sizeof sblk.buf, SBLOCK_UFS1);
   1099      1.1  dholland 	close(fd);
   1100      1.1  dholland 	if (rval != sizeof sblk.buf)
   1101      1.1  dholland 		return;
   1102      1.1  dholland 
   1103      1.1  dholland 	if (fs->fs_magic != FS_UFS1_MAGIC &&
   1104      1.1  dholland 	    fs->fs_magic != FS_UFS1_MAGIC_SWAPPED)
   1105      1.1  dholland 		/* Not FFSv1 */
   1106      1.1  dholland 		return;
   1107      1.1  dholland 	if (fs->fs_old_flags & FS_FLAGS_UPDATED)
   1108      1.1  dholland 		/* properly updated fslevel 4 */
   1109      1.1  dholland 		return;
   1110      1.1  dholland 	if (fs->fs_bsize != fs->fs_maxbsize)
   1111      1.1  dholland 		/* not messed up */
   1112      1.1  dholland 		return;
   1113      1.1  dholland 
   1114      1.1  dholland 	/*
   1115      1.1  dholland 	 * OK we have a munged fs, first 'upgrade' to fslevel 4,
   1116      1.1  dholland 	 * We specify -b16 in order to stop fsck bleating that the
   1117      1.1  dholland 	 * sb doesn't match the first alternate.
   1118      1.1  dholland 	 */
   1119      1.1  dholland 	run_program(RUN_DISPLAY | RUN_PROGRESS,
   1120      1.1  dholland 	    "%s -p -b 16 -c 4 /dev/r%s%c", prog, disk, ptn);
   1121      1.1  dholland 	/* Then downgrade to fslevel 3 */
   1122      1.1  dholland 	run_program(RUN_DISPLAY | RUN_PROGRESS,
   1123      1.1  dholland 	    "%s -p -c 3 /dev/r%s%c", prog, disk, ptn);
   1124      1.1  dholland }
   1125      1.1  dholland 
   1126      1.1  dholland /*
   1127      1.1  dholland  * fsck and mount the root partition.
   1128      1.1  dholland  */
   1129      1.1  dholland static int
   1130      1.1  dholland mount_root(void)
   1131      1.1  dholland {
   1132      1.1  dholland 	int	error;
   1133      1.2    martin 	int ptn = (pm->isspecial)? 0 - 'a' : pm->rootpart;
   1134      1.1  dholland 
   1135      1.2    martin 	error = fsck_preen(pm->diskdev, ptn, "ffs");
   1136      1.1  dholland 	if (error != 0)
   1137      1.1  dholland 		return error;
   1138      1.1  dholland 
   1139      1.1  dholland 	md_pre_mount();
   1140      1.1  dholland 
   1141      1.1  dholland 	/* Mount /dev/<diskdev>a on target's "".
   1142      1.1  dholland 	 * If we pass "" as mount-on, Prefixing will DTRT.
   1143      1.1  dholland 	 * for now, use no options.
   1144      1.1  dholland 	 * XXX consider -o remount in case target root is
   1145      1.1  dholland 	 * current root, still readonly from single-user?
   1146      1.1  dholland 	 */
   1147      1.2    martin 	return target_mount("", pm->diskdev, ptn, "");
   1148      1.1  dholland }
   1149      1.1  dholland 
   1150      1.1  dholland /* Get information on the file systems mounted from the root filesystem.
   1151      1.1  dholland  * Offer to convert them into 4.4BSD inodes if they are not 4.4BSD
   1152      1.1  dholland  * inodes.  Fsck them.  Mount them.
   1153      1.1  dholland  */
   1154      1.1  dholland 
   1155      1.1  dholland int
   1156      1.1  dholland mount_disks(void)
   1157      1.1  dholland {
   1158      1.1  dholland 	char *fstab;
   1159      1.1  dholland 	int   fstabsize;
   1160      1.1  dholland 	int   error;
   1161      1.1  dholland 
   1162      1.1  dholland 	static struct lookfor fstabbuf[] = {
   1163      1.1  dholland 		{"/dev/", "/dev/%s %s ffs %s", "c", NULL, 0, 0, foundffs},
   1164      1.1  dholland 		{"/dev/", "/dev/%s %s ufs %s", "c", NULL, 0, 0, foundffs},
   1165      1.1  dholland #ifdef USE_SYSVBFS
   1166      1.1  dholland 		{"/dev/", "/dev/%s %s sysvbfs %s", "c", NULL, 0, 0,
   1167      1.1  dholland 		    foundsysvbfs},
   1168      1.1  dholland #endif
   1169      1.1  dholland 	};
   1170      1.1  dholland 	static size_t numfstabbuf = sizeof(fstabbuf) / sizeof(struct lookfor);
   1171      1.1  dholland 
   1172      1.1  dholland 	/* First the root device. */
   1173      1.1  dholland 	if (target_already_root())
   1174      1.1  dholland 		/* avoid needing to call target_already_root() again */
   1175      1.1  dholland 		targetroot_mnt[0] = 0;
   1176      1.1  dholland 	else {
   1177      1.1  dholland 		error = mount_root();
   1178      1.1  dholland 		if (error != 0 && error != EBUSY)
   1179      1.1  dholland 			return -1;
   1180      1.1  dholland 	}
   1181      1.1  dholland 
   1182      1.1  dholland 	/* Check the target /etc/fstab exists before trying to parse it. */
   1183      1.1  dholland 	if (target_dir_exists_p("/etc") == 0 ||
   1184      1.1  dholland 	    target_file_exists_p("/etc/fstab") == 0) {
   1185      1.2    martin 		msg_display(MSG_noetcfstab, pm->diskdev);
   1186      1.1  dholland 		process_menu(MENU_ok, NULL);
   1187      1.1  dholland 		return -1;
   1188      1.1  dholland 	}
   1189      1.1  dholland 
   1190      1.1  dholland 
   1191      1.1  dholland 	/* Get fstab entries from the target-root /etc/fstab. */
   1192      1.1  dholland 	fstabsize = target_collect_file(T_FILE, &fstab, "/etc/fstab");
   1193      1.1  dholland 	if (fstabsize < 0) {
   1194      1.1  dholland 		/* error ! */
   1195      1.2    martin 		msg_display(MSG_badetcfstab, pm->diskdev);
   1196      1.1  dholland 		process_menu(MENU_ok, NULL);
   1197      1.2    martin 		return -2;
   1198      1.1  dholland 	}
   1199      1.1  dholland 	error = walk(fstab, (size_t)fstabsize, fstabbuf, numfstabbuf);
   1200      1.1  dholland 	free(fstab);
   1201      1.1  dholland 
   1202      1.1  dholland 	return error;
   1203      1.1  dholland }
   1204      1.1  dholland 
   1205      1.1  dholland int
   1206  1.4.4.3       snj set_swap_if_low_ram(const char *disk, partinfo *pp) {
   1207  1.4.4.3       snj         if (get_ramsize() <= 32)
   1208  1.4.4.3       snj                 return set_swap(disk, pp);
   1209  1.4.4.3       snj         return 0;
   1210  1.4.4.3       snj }
   1211  1.4.4.3       snj 
   1212  1.4.4.3       snj int
   1213      1.1  dholland set_swap(const char *disk, partinfo *pp)
   1214      1.1  dholland {
   1215      1.1  dholland 	int i;
   1216      1.1  dholland 	char *cp;
   1217      1.1  dholland 	int rval;
   1218      1.1  dholland 
   1219      1.1  dholland 	if (pp == NULL)
   1220      1.2    martin 		pp = pm->oldlabel;
   1221      1.1  dholland 
   1222      1.1  dholland 	for (i = 0; i < MAXPARTITIONS; i++) {
   1223      1.1  dholland 		if (pp[i].pi_fstype != FS_SWAP)
   1224      1.1  dholland 			continue;
   1225      1.1  dholland 		asprintf(&cp, "/dev/%s%c", disk, 'a' + i);
   1226      1.1  dholland 		rval = swapctl(SWAP_ON, cp, 0);
   1227      1.1  dholland 		free(cp);
   1228      1.1  dholland 		if (rval != 0)
   1229      1.1  dholland 			return -1;
   1230      1.1  dholland 	}
   1231      1.1  dholland 
   1232      1.1  dholland 	return 0;
   1233      1.1  dholland }
   1234      1.1  dholland 
   1235      1.1  dholland int
   1236      1.1  dholland check_swap(const char *disk, int remove_swap)
   1237      1.1  dholland {
   1238      1.1  dholland 	struct swapent *swap;
   1239      1.1  dholland 	char *cp;
   1240      1.1  dholland 	int nswap;
   1241      1.1  dholland 	int l;
   1242      1.1  dholland 	int rval = 0;
   1243      1.1  dholland 
   1244      1.1  dholland 	nswap = swapctl(SWAP_NSWAP, 0, 0);
   1245      1.1  dholland 	if (nswap <= 0)
   1246      1.1  dholland 		return 0;
   1247      1.1  dholland 
   1248      1.1  dholland 	swap = malloc(nswap * sizeof *swap);
   1249      1.1  dholland 	if (swap == NULL)
   1250      1.1  dholland 		return -1;
   1251      1.1  dholland 
   1252      1.1  dholland 	nswap = swapctl(SWAP_STATS, swap, nswap);
   1253      1.1  dholland 	if (nswap < 0)
   1254      1.1  dholland 		goto bad_swap;
   1255      1.1  dholland 
   1256      1.1  dholland 	l = strlen(disk);
   1257      1.1  dholland 	while (--nswap >= 0) {
   1258      1.1  dholland 		/* Should we check the se_dev or se_path? */
   1259      1.1  dholland 		cp = swap[nswap].se_path;
   1260      1.1  dholland 		if (memcmp(cp, "/dev/", 5) != 0)
   1261      1.1  dholland 			continue;
   1262      1.1  dholland 		if (memcmp(cp + 5, disk, l) != 0)
   1263      1.1  dholland 			continue;
   1264      1.1  dholland 		if (!isalpha(*(unsigned char *)(cp + 5 + l)))
   1265      1.1  dholland 			continue;
   1266      1.1  dholland 		if (cp[5 + l + 1] != 0)
   1267      1.1  dholland 			continue;
   1268      1.1  dholland 		/* ok path looks like it is for this device */
   1269      1.1  dholland 		if (!remove_swap) {
   1270      1.1  dholland 			/* count active swap areas */
   1271      1.1  dholland 			rval++;
   1272      1.1  dholland 			continue;
   1273      1.1  dholland 		}
   1274      1.1  dholland 		if (swapctl(SWAP_OFF, cp, 0) == -1)
   1275      1.1  dholland 			rval = -1;
   1276      1.1  dholland 	}
   1277      1.1  dholland 
   1278      1.1  dholland     done:
   1279      1.1  dholland 	free(swap);
   1280      1.1  dholland 	return rval;
   1281      1.1  dholland 
   1282      1.1  dholland     bad_swap:
   1283      1.1  dholland 	rval = -1;
   1284      1.1  dholland 	goto done;
   1285      1.1  dholland }
   1286      1.1  dholland 
   1287      1.1  dholland #ifdef HAVE_BOOTXX_xFS
   1288      1.1  dholland char *
   1289      1.1  dholland bootxx_name(void)
   1290      1.1  dholland {
   1291      1.1  dholland 	int fstype;
   1292      1.1  dholland 	const char *bootxxname;
   1293      1.1  dholland 	char *bootxx;
   1294      1.1  dholland 
   1295      1.1  dholland 	/* check we have boot code for the root partition type */
   1296      1.2    martin 	fstype = pm->bsdlabel[pm->rootpart].pi_fstype;
   1297      1.1  dholland 	switch (fstype) {
   1298      1.1  dholland #if defined(BOOTXX_FFSV1) || defined(BOOTXX_FFSV2)
   1299      1.1  dholland 	case FS_BSDFFS:
   1300      1.2    martin 		if (pm->bsdlabel[pm->rootpart].pi_flags & PIF_FFSv2) {
   1301      1.1  dholland #ifdef BOOTXX_FFSV2
   1302      1.1  dholland 			bootxxname = BOOTXX_FFSV2;
   1303      1.1  dholland #else
   1304      1.1  dholland 			bootxxname = NULL;
   1305      1.1  dholland #endif
   1306      1.1  dholland 		} else {
   1307      1.1  dholland #ifdef BOOTXX_FFSV1
   1308      1.1  dholland 			bootxxname = BOOTXX_FFSV1;
   1309      1.1  dholland #else
   1310      1.1  dholland 			bootxxname = NULL;
   1311      1.1  dholland #endif
   1312      1.1  dholland 		}
   1313      1.1  dholland 		break;
   1314      1.1  dholland #endif
   1315      1.1  dholland #ifdef BOOTXX_LFS
   1316      1.1  dholland 	case FS_BSDLFS:
   1317      1.1  dholland 		bootxxname = BOOTXX_LFS;
   1318      1.1  dholland 		break;
   1319      1.1  dholland #endif
   1320      1.1  dholland 	default:
   1321      1.1  dholland 		bootxxname = NULL;
   1322      1.1  dholland 		break;
   1323      1.1  dholland 	}
   1324      1.1  dholland 
   1325      1.1  dholland 	if (bootxxname == NULL)
   1326      1.1  dholland 		return NULL;
   1327      1.1  dholland 
   1328      1.1  dholland 	asprintf(&bootxx, "%s/%s", BOOTXXDIR, bootxxname);
   1329      1.1  dholland 	return bootxx;
   1330      1.1  dholland }
   1331      1.1  dholland #endif
   1332      1.2    martin 
   1333      1.2    martin static int
   1334      1.2    martin get_fsid_by_gptuuid(const char *str)
   1335      1.2    martin {
   1336      1.2    martin 	int i;
   1337      1.2    martin 	uuid_t uuid;
   1338      1.2    martin 	uint32_t status;
   1339      1.2    martin 
   1340      1.2    martin 	uuid_from_string(str, &uuid, &status);
   1341      1.2    martin 	if (status == uuid_s_ok) {
   1342      1.2    martin 		for (i = 0; gpt_filesystems[i].id > 0; i++)
   1343      1.2    martin 			if (uuid_equal(&uuid, &(gpt_filesystems[i].uuid), NULL))
   1344      1.2    martin 				return gpt_filesystems[i].id;
   1345      1.2    martin 	}
   1346      1.2    martin 	return FS_OTHER;
   1347      1.2    martin }
   1348      1.2    martin 
   1349      1.2    martin const char *
   1350      1.2    martin get_gptfs_by_id(int filesystem)
   1351      1.2    martin {
   1352      1.2    martin 	int i;
   1353      1.2    martin 	for (i = 0; gpt_filesystems[i].id > 0; i++)
   1354      1.2    martin 		if (filesystem == gpt_filesystems[i].id)
   1355      1.2    martin 			return gpt_filesystems[i].name;
   1356      1.2    martin 	return NULL;
   1357      1.2    martin }
   1358      1.2    martin 
   1359      1.2    martin /* from dkctl.c */
   1360      1.2    martin static int
   1361      1.2    martin get_dkwedges_sort(const void *a, const void *b)
   1362      1.2    martin {
   1363      1.2    martin 	const struct dkwedge_info *dkwa = a, *dkwb = b;
   1364      1.2    martin 	const daddr_t oa = dkwa->dkw_offset, ob = dkwb->dkw_offset;
   1365      1.2    martin 	return (oa < ob) ? -1 : (oa > ob) ? 1 : 0;
   1366      1.2    martin }
   1367      1.2    martin 
   1368      1.2    martin int
   1369      1.2    martin get_dkwedges(struct dkwedge_info **dkw, const char *diskdev)
   1370      1.2    martin {
   1371      1.2    martin 	int fd;
   1372      1.2    martin 	char buf[STRSIZE];
   1373      1.2    martin 	size_t bufsize;
   1374      1.2    martin 	struct dkwedge_list dkwl;
   1375      1.2    martin 
   1376      1.2    martin 	*dkw = NULL;
   1377      1.2    martin 	dkwl.dkwl_buf = *dkw;
   1378      1.2    martin 	dkwl.dkwl_bufsize = 0;
   1379      1.2    martin 	fd = opendisk(diskdev, O_RDONLY, buf, STRSIZE, 0);
   1380      1.2    martin 	if (fd < 0)
   1381      1.2    martin 		return -1;
   1382      1.2    martin 
   1383      1.2    martin 	for (;;) {
   1384      1.2    martin 		if (ioctl(fd, DIOCLWEDGES, &dkwl) == -1)
   1385      1.2    martin 			return -2;
   1386      1.2    martin 		if (dkwl.dkwl_nwedges == dkwl.dkwl_ncopied)
   1387      1.2    martin 			break;
   1388      1.2    martin 		bufsize = dkwl.dkwl_nwedges * sizeof(**dkw);
   1389      1.2    martin 		if (dkwl.dkwl_bufsize < bufsize) {
   1390      1.2    martin 			*dkw = realloc(dkwl.dkwl_buf, bufsize);
   1391      1.2    martin 			if (*dkw == NULL)
   1392      1.2    martin 				return -3;
   1393      1.2    martin 			dkwl.dkwl_buf = *dkw;
   1394      1.2    martin 			dkwl.dkwl_bufsize = bufsize;
   1395      1.2    martin 		}
   1396      1.2    martin 	}
   1397      1.2    martin 
   1398  1.4.4.1       riz 	if (dkwl.dkwl_nwedges > 0 && *dkw != NULL)
   1399      1.2    martin 		qsort(*dkw, dkwl.dkwl_nwedges, sizeof(**dkw), get_dkwedges_sort);
   1400      1.2    martin 
   1401      1.2    martin 	close(fd);
   1402      1.2    martin 	return dkwl.dkwl_nwedges;
   1403      1.2    martin }
   1404      1.2    martin 
   1405      1.2    martin /* XXX: rewrite */
   1406      1.2    martin static int
   1407      1.2    martin incoregpt(pm_devs_t *pm_cur, partinfo *lp)
   1408      1.2    martin {
   1409      1.2    martin 	int i, num;
   1410      1.4       riz 	unsigned int p_num;
   1411      1.4       riz 	uint64_t p_start, p_size;
   1412      1.2    martin 	char *textbuf, *t, *tt, p_type[STRSIZE];
   1413      1.2    martin 	struct dkwedge_info *dkw;
   1414      1.2    martin 
   1415      1.2    martin 	num = get_dkwedges(&dkw, pm_cur->diskdev);
   1416      1.2    martin 	if (dkw != NULL) {
   1417      1.2    martin 		for (i = 0; i < num && i < MAX_WEDGES; i++)
   1418      1.2    martin 			run_program(RUN_SILENT, "dkctl %s delwedge %s",
   1419      1.2    martin 				pm_cur->diskdev, dkw[i].dkw_devname);
   1420      1.2    martin 		free (dkw);
   1421      1.2    martin 	}
   1422      1.2    martin 
   1423      1.2    martin 	if (collect(T_OUTPUT, &textbuf, "gpt show -u %s 2>/dev/null", pm_cur->diskdev) < 1)
   1424      1.2    martin 		return -1;
   1425      1.2    martin 
   1426      1.2    martin 	(void)strtok(textbuf, "\n"); /* ignore first line */
   1427      1.2    martin 	while ((t = strtok(NULL, "\n")) != NULL) {
   1428      1.4       riz 		i = 0; p_start = 0; p_size = 0; p_num = 0; strcpy(p_type, ""); /* init */
   1429      1.2    martin 		while ((tt = strsep(&t, " \t")) != NULL) {
   1430      1.4       riz 			if (strlen(tt) == 0)
   1431      1.2    martin 				continue;
   1432      1.2    martin 			if (i == 0)
   1433      1.4       riz 				p_start = strtouq(tt, NULL, 10);
   1434      1.2    martin 			if (i == 1)
   1435      1.4       riz 				p_size = strtouq(tt, NULL, 10);
   1436      1.2    martin 			if (i == 2)
   1437      1.4       riz 				p_num = strtouq(tt, NULL, 10);
   1438      1.2    martin 			if (i > 2 || (i == 2 && p_num == 0))
   1439      1.2    martin 				if (
   1440      1.2    martin 					strcmp(tt, "GPT") &&
   1441      1.2    martin 					strcmp(tt, "part") &&
   1442      1.2    martin 					strcmp(tt, "-")
   1443      1.2    martin 					)
   1444      1.2    martin 						strncat(p_type, tt, STRSIZE);
   1445      1.2    martin 			i++;
   1446      1.2    martin 		}
   1447      1.4       riz 		if (p_start == 0 || p_size == 0)
   1448      1.2    martin 			continue;
   1449      1.2    martin 		else if (! strcmp(p_type, "Pritable"))
   1450      1.2    martin 			pm_cur->ptstart = p_start + p_size;
   1451      1.2    martin 		else if (! strcmp(p_type, "Sectable"))
   1452      1.2    martin 			pm_cur->ptsize = p_start - pm_cur->ptstart - 1;
   1453      1.4       riz 		else if (p_num == 0 && strlen(p_type) > 0)
   1454      1.2    martin 			/* Utilitary entry (PMBR, etc) */
   1455      1.2    martin 			continue;
   1456      1.4       riz 		else if (p_num == 0) {
   1457      1.2    martin 			/* Free space */
   1458      1.2    martin 			continue;
   1459      1.2    martin 		} else {
   1460      1.2    martin 			/* Usual partition */
   1461      1.2    martin 			lp[p_num].pi_size = p_size;
   1462      1.2    martin 			lp[p_num].pi_offset = p_start;
   1463      1.2    martin 			lp[p_num].pi_fstype = get_fsid_by_gptuuid(p_type);
   1464      1.2    martin 		}
   1465      1.2    martin 	}
   1466      1.2    martin 	free(textbuf);
   1467      1.2    martin 
   1468      1.2    martin 	return 0;
   1469      1.2    martin }
   1470      1.2    martin 
   1471  1.4.4.2       snj static bool
   1472      1.2    martin is_gpt(const char *dev)
   1473      1.2    martin {
   1474  1.4.4.4   msaitoh 	check_available_binaries();
   1475  1.4.4.4   msaitoh 
   1476  1.4.4.4   msaitoh 	if (!have_gpt)
   1477  1.4.4.2       snj 		return false;
   1478  1.4.4.2       snj 
   1479  1.4.4.2       snj 	return !run_program(RUN_SILENT | RUN_ERROR_OK,
   1480      1.2    martin 		"sh -c 'gpt show %s |grep -e Pri\\ GPT\\ table'", dev);
   1481      1.2    martin }
   1482