Home | History | Annotate | Line # | Download | only in sysinst
disks.c revision 1.13.12.1
      1  1.13.12.1  pgoyette /*	$NetBSD: disks.c,v 1.13.12.1 2018/05/02 07:20:28 pgoyette 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.6    martin 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.10     isaki 	        inqbuf->atap_model[1] == 'E') ||
    312        1.1  dholland 	       (inqbuf->atap_model[0] == 'F' &&
    313       1.10     isaki 	        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.10     isaki 	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.10     isaki 	/*
    472       1.10     isaki 	 * partman_go: <0 - we want to see menu with extended partitioning
    473       1.10     isaki 	 *            ==0 - we want to see simple select disk menu
    474       1.10     isaki 	 *             >0 - we do not want to see any menus, just detect
    475       1.10     isaki 	 *                  all disks
    476       1.10     isaki 	 */
    477        1.2    martin 	if (partman_go <= 0) {
    478        1.2    martin 		if (numdisks == 0) {
    479        1.2    martin 			/* No disks found! */
    480        1.2    martin 			msg_display(MSG_nodisk);
    481        1.2    martin 			process_menu(MENU_ok, NULL);
    482        1.2    martin 			/*endwin();*/
    483        1.2    martin 			return -1;
    484        1.2    martin 		} else {
    485        1.2    martin 			/* One or more disks found! */
    486        1.2    martin 			for (i = 0; i < numdisks; i++) {
    487        1.2    martin 				dsk_menu[i].opt_name = disks[i].dd_descr;
    488        1.2    martin 				dsk_menu[i].opt_menu = OPT_NOMENU;
    489        1.2    martin 				dsk_menu[i].opt_flags = OPT_EXIT;
    490        1.2    martin 				dsk_menu[i].opt_action = set_menu_select;
    491        1.2    martin 			}
    492        1.2    martin 			if (partman_go < 0) {
    493        1.2    martin 				dsk_menu[i].opt_name = MSG_partman;
    494        1.2    martin 				dsk_menu[i].opt_menu = OPT_NOMENU;
    495        1.2    martin 				dsk_menu[i].opt_flags = OPT_EXIT;
    496        1.2    martin 				dsk_menu[i].opt_action = set_menu_select;
    497        1.2    martin 			}
    498        1.2    martin 			menu_no = new_menu(MSG_Available_disks,
    499        1.2    martin 				dsk_menu, numdisks + ((partman_go<0)?1:0), -1, 4, 0, 0, MC_SCROLL,
    500        1.2    martin 				NULL, NULL, NULL, NULL, NULL);
    501        1.2    martin 			if (menu_no == -1)
    502        1.2    martin 				return -1;
    503        1.2    martin 			msg_display(MSG_ask_disk, doingwhat);
    504        1.2    martin 			process_menu(menu_no, &selected_disk);
    505        1.2    martin 			free_menu(menu_no);
    506        1.2    martin 		}
    507        1.2    martin 		if (partman_go < 0 && selected_disk == numdisks) {
    508        1.2    martin 			partman_go = 1;
    509       1.10     isaki 			return -2;
    510        1.2    martin 		} else
    511        1.2    martin 			partman_go = 0;
    512        1.2    martin 		if (selected_disk < 0 || selected_disk >= numdisks)
    513       1.10     isaki 			return -1;
    514        1.2    martin 	}
    515        1.2    martin 
    516        1.2    martin 	/* Fill pm struct with device(s) info */
    517        1.2    martin 	for (i = 0; i < numdisks; i++) {
    518        1.2    martin 		if (! partman_go)
    519        1.2    martin 			disk = disks + selected_disk;
    520        1.2    martin 		else {
    521        1.2    martin 			disk = disks + i;
    522        1.2    martin 			already_found = 0;
    523        1.2    martin 			SLIST_FOREACH(pm_i, &pm_head, l) {
    524        1.2    martin 				pm_last = pm_i;
    525       1.10     isaki 				if (!already_found &&
    526       1.10     isaki 				    strcmp(pm_i->diskdev, disk->dd_name) == 0) {
    527        1.2    martin 					pm_i->found = 1;
    528        1.2    martin 					break;
    529        1.2    martin 				}
    530        1.2    martin 			}
    531        1.2    martin 			if (pm_i != NULL && pm_i->found)
    532        1.2    martin 				/* We already added this device, skipping */
    533        1.2    martin 				continue;
    534        1.1  dholland 		}
    535        1.2    martin 		pm = pm_new;
    536        1.2    martin 		pm->found = 1;
    537        1.2    martin 		pm->bootable = 0;
    538        1.2    martin 		pm->pi.menu_no = -1;
    539        1.2    martin 		pm->disktype = "unknown";
    540        1.2    martin 		pm->doessf = "";
    541        1.2    martin 		strlcpy(pm->diskdev, disk->dd_name, sizeof pm->diskdev);
    542        1.2    martin 		strlcpy(pm->diskdev_descr, disk->dd_descr, sizeof pm->diskdev_descr);
    543        1.2    martin 		/* Use as a default disk if the user has the sets on a local disk */
    544        1.2    martin 		strlcpy(localfs_dev, disk->dd_name, sizeof localfs_dev);
    545        1.2    martin 
    546        1.2    martin 		pm->gpt = is_gpt(pm->diskdev);
    547        1.2    martin 		pm->no_mbr = disk->dd_no_mbr || pm->gpt;
    548        1.2    martin 		pm->sectorsize = disk->dd_secsize;
    549        1.2    martin 		pm->dlcyl = disk->dd_cyl;
    550        1.2    martin 		pm->dlhead = disk->dd_head;
    551        1.2    martin 		pm->dlsec = disk->dd_sec;
    552        1.2    martin 		pm->dlsize = disk->dd_totsec;
    553        1.2    martin 		if (pm->dlsize == 0)
    554        1.2    martin 			pm->dlsize = disk->dd_cyl * disk->dd_head * disk->dd_sec;
    555        1.2    martin 		if (pm->dlsize > UINT32_MAX && ! partman_go) {
    556        1.2    martin 			if (logfp)
    557        1.2    martin 				fprintf(logfp, "Cannot process disk %s: too big size (%d)\n",
    558        1.2    martin 					pm->diskdev, (int)pm->dlsize);
    559        1.2    martin 			msg_display(MSG_toobigdisklabel);
    560        1.2    martin 			process_menu(MENU_ok, NULL);
    561        1.1  dholland 			return -1;
    562        1.2    martin 		}
    563        1.2    martin 		pm->dlcylsize = pm->dlhead * pm->dlsec;
    564        1.2    martin 
    565        1.2    martin 		label_read();
    566        1.2    martin 		if (partman_go) {
    567        1.2    martin 			pm_getrefdev(pm_new);
    568        1.2    martin 			if (SLIST_EMPTY(&pm_head) || pm_last == NULL)
    569        1.2    martin 				 SLIST_INSERT_HEAD(&pm_head, pm_new, l);
    570        1.2    martin 			else
    571        1.2    martin 				 SLIST_INSERT_AFTER(pm_last, pm_new, l);
    572        1.2    martin 			pm_new = malloc(sizeof (pm_devs_t));
    573        1.2    martin 			memset(pm_new, 0, sizeof *pm_new);
    574        1.2    martin 		} else
    575        1.2    martin 			/* We is not in partman and do not want to process all devices, exit */
    576        1.2    martin 			break;
    577        1.1  dholland 	}
    578        1.1  dholland 
    579        1.2    martin 	return numdisks;
    580        1.2    martin }
    581        1.2    martin 
    582        1.6    martin 
    583        1.2    martin void
    584        1.2    martin label_read(void)
    585        1.2    martin {
    586        1.8    martin 	check_available_binaries();
    587        1.8    martin 
    588        1.1  dholland 	/* Get existing/default label */
    589        1.2    martin 	memset(&pm->oldlabel, 0, sizeof pm->oldlabel);
    590        1.8    martin 	if (!have_gpt || !pm->gpt)
    591        1.2    martin 		incorelabel(pm->diskdev, pm->oldlabel);
    592        1.2    martin 	else
    593        1.2    martin 		incoregpt(pm, pm->oldlabel);
    594        1.1  dholland 	/* Set 'target' label to current label in case we don't change it */
    595        1.2    martin 	memcpy(&pm->bsdlabel, &pm->oldlabel, sizeof pm->bsdlabel);
    596        1.2    martin #ifndef NO_DISKLABEL
    597        1.2    martin 	if (! pm->gpt)
    598        1.2    martin 		savenewlabel(pm->oldlabel, getmaxpartitions());
    599        1.2    martin #endif
    600        1.1  dholland }
    601        1.1  dholland 
    602        1.1  dholland void
    603        1.1  dholland fmt_fspart(menudesc *m, int ptn, void *arg)
    604        1.1  dholland {
    605        1.1  dholland 	unsigned int poffset, psize, pend;
    606        1.1  dholland 	const char *desc;
    607        1.1  dholland 	static const char *Yes;
    608        1.2    martin 	partinfo *p = pm->bsdlabel + ptn;
    609        1.1  dholland 
    610        1.1  dholland 	if (Yes == NULL)
    611        1.1  dholland 		Yes = msg_string(MSG_Yes);
    612        1.1  dholland 
    613        1.1  dholland 	poffset = p->pi_offset / sizemult;
    614        1.1  dholland 	psize = p->pi_size / sizemult;
    615        1.1  dholland 	if (psize == 0)
    616        1.1  dholland 		pend = 0;
    617        1.1  dholland 	else
    618        1.1  dholland 		pend = (p->pi_offset + p->pi_size) / sizemult - 1;
    619        1.1  dholland 
    620        1.1  dholland 	if (p->pi_fstype == FS_BSDFFS)
    621        1.1  dholland 		if (p->pi_flags & PIF_FFSv2)
    622        1.1  dholland 			desc = "FFSv2";
    623        1.1  dholland 		else
    624        1.1  dholland 			desc = "FFSv1";
    625        1.1  dholland 	else
    626        1.1  dholland 		desc = getfslabelname(p->pi_fstype);
    627        1.1  dholland 
    628        1.1  dholland #ifdef PART_BOOT
    629        1.1  dholland 	if (ptn == PART_BOOT)
    630        1.1  dholland 		desc = msg_string(MSG_Boot_partition_cant_change);
    631        1.1  dholland #endif
    632        1.1  dholland 	if (ptn == getrawpartition())
    633        1.1  dholland 		desc = msg_string(MSG_Whole_disk_cant_change);
    634        1.1  dholland 	else {
    635        1.1  dholland 		if (ptn == PART_C)
    636        1.1  dholland 			desc = msg_string(MSG_NetBSD_partition_cant_change);
    637        1.1  dholland 	}
    638        1.1  dholland 
    639        1.1  dholland 	wprintw(m->mw, msg_string(MSG_fspart_row),
    640        1.1  dholland 			poffset, pend, psize, desc,
    641        1.1  dholland 			p->pi_flags & PIF_NEWFS ? Yes : "",
    642        1.1  dholland 			p->pi_flags & PIF_MOUNT ? Yes : "",
    643        1.1  dholland 			p->pi_mount);
    644        1.1  dholland }
    645        1.1  dholland 
    646        1.1  dholland /*
    647        1.1  dholland  * Label a disk using an MD-specific string DISKLABEL_CMD for
    648        1.1  dholland  * to invoke disklabel.
    649        1.1  dholland  * if MD code does not define DISKLABEL_CMD, this is a no-op.
    650        1.1  dholland  *
    651        1.1  dholland  * i386 port uses "/sbin/disklabel -w -r", just like i386
    652        1.1  dholland  * miniroot scripts, though this may leave a bogus incore label.
    653        1.1  dholland  *
    654        1.1  dholland  * Sun ports should use DISKLABEL_CMD "/sbin/disklabel -w"
    655        1.1  dholland  * to get incore to ondisk inode translation for the Sun proms.
    656        1.1  dholland  */
    657        1.1  dholland int
    658        1.1  dholland write_disklabel (void)
    659        1.1  dholland {
    660        1.8    martin 	int rv = 0;
    661        1.1  dholland 
    662        1.1  dholland #ifdef DISKLABEL_CMD
    663        1.1  dholland 	/* disklabel the disk */
    664        1.8    martin 	rv = run_program(RUN_DISPLAY, "%s -f /tmp/disktab %s '%s'",
    665        1.2    martin 	    DISKLABEL_CMD, pm->diskdev, pm->bsddiskname);
    666        1.8    martin 	if (rv == 0)
    667        1.8    martin 		update_wedges(pm->diskdev);
    668        1.1  dholland #endif
    669        1.8    martin 	return rv;
    670        1.1  dholland }
    671        1.1  dholland 
    672        1.1  dholland 
    673        1.1  dholland static int
    674        1.1  dholland ptn_sort(const void *a, const void *b)
    675        1.1  dholland {
    676        1.2    martin 	return strcmp(pm->bsdlabel[*(const int *)a].pi_mount,
    677        1.2    martin 		      pm->bsdlabel[*(const int *)b].pi_mount);
    678        1.1  dholland }
    679        1.1  dholland 
    680        1.1  dholland int
    681        1.1  dholland make_filesystems(void)
    682        1.1  dholland {
    683        1.1  dholland 	unsigned int i;
    684        1.1  dholland 	int ptn;
    685        1.2    martin 	int ptn_order[nelem(pm->bsdlabel)];
    686        1.1  dholland 	int error = 0;
    687        1.1  dholland 	unsigned int maxpart = getmaxpartitions();
    688        1.2    martin 	char *newfs = NULL, *dev = NULL, *devdev = NULL;
    689        1.1  dholland 	partinfo *lbl;
    690        1.1  dholland 
    691        1.2    martin 	if (maxpart > nelem(pm->bsdlabel))
    692        1.2    martin 		maxpart = nelem(pm->bsdlabel);
    693        1.1  dholland 
    694        1.1  dholland 	/* Making new file systems and mounting them */
    695        1.1  dholland 
    696        1.1  dholland 	/* sort to ensure /usr/local is mounted after /usr (etc) */
    697        1.1  dholland 	for (i = 0; i < maxpart; i++)
    698        1.1  dholland 		ptn_order[i] = i;
    699        1.1  dholland 	qsort(ptn_order, maxpart, sizeof ptn_order[0], ptn_sort);
    700        1.1  dholland 
    701        1.1  dholland 	for (i = 0; i < maxpart; i++) {
    702        1.1  dholland 		/*
    703        1.1  dholland 		 * newfs and mount. For now, process only BSD filesystems.
    704        1.1  dholland 		 * but if this is the mounted-on root, has no mount
    705        1.1  dholland 		 * point defined, or is marked preserve, don't touch it!
    706        1.1  dholland 		 */
    707        1.1  dholland 		ptn = ptn_order[i];
    708        1.2    martin 		lbl = pm->bsdlabel + ptn;
    709        1.1  dholland 
    710        1.2    martin 		if (is_active_rootpart(pm->diskdev, ptn))
    711        1.1  dholland 			continue;
    712        1.1  dholland 
    713        1.1  dholland 		if (*lbl->pi_mount == 0)
    714        1.1  dholland 			/* No mount point */
    715        1.1  dholland 			continue;
    716        1.1  dholland 
    717        1.2    martin 		if (pm->isspecial) {
    718        1.2    martin 			asprintf(&dev, "%s", pm->diskdev);
    719        1.2    martin 			ptn = 0 - 'a';
    720        1.2    martin 		} else {
    721        1.2    martin 			asprintf(&dev, "%s%c", pm->diskdev, 'a' + ptn);
    722        1.2    martin 		}
    723        1.2    martin 		if (dev == NULL)
    724        1.2    martin 			return (ENOMEM);
    725        1.2    martin 		asprintf(&devdev, "/dev/%s", dev);
    726        1.2    martin 		if (devdev == NULL)
    727        1.2    martin 			return (ENOMEM);
    728        1.2    martin 
    729        1.1  dholland 		newfs = NULL;
    730        1.2    martin 		lbl->mnt_opts = NULL;
    731        1.2    martin 		lbl->fsname = NULL;
    732        1.1  dholland 		switch (lbl->pi_fstype) {
    733        1.1  dholland 		case FS_APPLEUFS:
    734        1.1  dholland 			asprintf(&newfs, "/sbin/newfs %s%.0d",
    735        1.1  dholland 				lbl->pi_isize != 0 ? "-i" : "", lbl->pi_isize);
    736        1.2    martin 			lbl->mnt_opts = "-tffs -o async";
    737        1.2    martin 			lbl->fsname = "ffs";
    738        1.1  dholland 			break;
    739        1.1  dholland 		case FS_BSDFFS:
    740        1.1  dholland 			asprintf(&newfs,
    741        1.1  dholland 			    "/sbin/newfs -V2 -O %d -b %d -f %d%s%.0d",
    742        1.1  dholland 			    lbl->pi_flags & PIF_FFSv2 ? 2 : 1,
    743        1.1  dholland 			    lbl->pi_fsize * lbl->pi_frag, lbl->pi_fsize,
    744        1.1  dholland 			    lbl->pi_isize != 0 ? " -i " : "", lbl->pi_isize);
    745        1.1  dholland 			if (lbl->pi_flags & PIF_LOG)
    746        1.2    martin 				lbl->mnt_opts = "-tffs -o log";
    747        1.1  dholland 			else
    748        1.2    martin 				lbl->mnt_opts = "-tffs -o async";
    749        1.2    martin 			lbl->fsname = "ffs";
    750        1.1  dholland 			break;
    751        1.1  dholland 		case FS_BSDLFS:
    752        1.1  dholland 			asprintf(&newfs, "/sbin/newfs_lfs -b %d",
    753        1.1  dholland 				lbl->pi_fsize * lbl->pi_frag);
    754        1.2    martin 			lbl->mnt_opts = "-tlfs";
    755        1.2    martin 			lbl->fsname = "lfs";
    756        1.1  dholland 			break;
    757        1.1  dholland 		case FS_MSDOS:
    758        1.1  dholland #ifdef USE_NEWFS_MSDOS
    759        1.1  dholland 			asprintf(&newfs, "/sbin/newfs_msdos");
    760        1.1  dholland #endif
    761        1.2    martin 			lbl->mnt_opts = "-tmsdos";
    762        1.2    martin 			lbl->fsname = "msdos";
    763        1.1  dholland 			break;
    764        1.1  dholland #ifdef USE_SYSVBFS
    765        1.1  dholland 		case FS_SYSVBFS:
    766        1.1  dholland 			asprintf(&newfs, "/sbin/newfs_sysvbfs");
    767        1.2    martin 			lbl->mnt_opts = "-tsysvbfs";
    768        1.2    martin 			lbl->fsname = "sysvbfs";
    769        1.1  dholland 			break;
    770        1.1  dholland #endif
    771        1.1  dholland #ifdef USE_EXT2FS
    772        1.1  dholland 		case FS_EX2FS:
    773        1.1  dholland 			asprintf(&newfs, "/sbin/newfs_ext2fs");
    774        1.2    martin 			lbl->mnt_opts = "-text2fs";
    775        1.2    martin 			lbl->fsname = "ext2fs";
    776        1.1  dholland 			break;
    777        1.1  dholland #endif
    778        1.1  dholland 		}
    779        1.1  dholland 		if (lbl->pi_flags & PIF_NEWFS && newfs != NULL) {
    780        1.1  dholland #ifdef USE_NEWFS_MSDOS
    781        1.1  dholland 			if (lbl->pi_fstype == FS_MSDOS) {
    782        1.1  dholland 			        /* newfs only if mount fails */
    783        1.1  dholland 			        if (run_program(RUN_SILENT | RUN_ERROR_OK,
    784        1.2    martin 				    "mount -rt msdos /dev/%s /mnt2", dev) != 0)
    785        1.1  dholland 					error = run_program(
    786        1.1  dholland 					    RUN_DISPLAY | RUN_PROGRESS,
    787        1.2    martin 					    "%s /dev/r%s",
    788        1.2    martin 					    newfs, dev);
    789        1.1  dholland 				else {
    790       1.10     isaki 					run_program(RUN_SILENT | RUN_ERROR_OK,
    791        1.1  dholland 					    "umount /mnt2");
    792        1.1  dholland 					error = 0;
    793        1.1  dholland 				}
    794        1.1  dholland 			} else
    795        1.1  dholland #endif
    796        1.1  dholland 			error = run_program(RUN_DISPLAY | RUN_PROGRESS,
    797        1.2    martin 			    "%s /dev/r%s", newfs, dev);
    798        1.1  dholland 		} else {
    799        1.1  dholland 			/* We'd better check it isn't dirty */
    800        1.2    martin 			error = fsck_preen(pm->diskdev, ptn, lbl->fsname);
    801        1.1  dholland 		}
    802        1.1  dholland 		free(newfs);
    803       1.12    martin 		if (error != 0) {
    804       1.12    martin 			free(devdev);
    805       1.12    martin 			free(dev);
    806        1.1  dholland 			return error;
    807       1.12    martin 		}
    808        1.1  dholland 
    809        1.2    martin 		lbl->pi_flags ^= PIF_NEWFS;
    810        1.1  dholland 		md_pre_mount();
    811        1.1  dholland 
    812        1.2    martin 		if (partman_go == 0 && lbl->pi_flags & PIF_MOUNT &&
    813        1.2    martin 				lbl->mnt_opts != NULL) {
    814        1.1  dholland 			make_target_dir(lbl->pi_mount);
    815        1.2    martin 			error = target_mount_do(lbl->mnt_opts, devdev, lbl->pi_mount);
    816        1.1  dholland 			if (error) {
    817        1.2    martin 				msg_display(MSG_mountfail, dev, ' ', lbl->pi_mount);
    818        1.1  dholland 				process_menu(MENU_ok, NULL);
    819       1.12    martin 				free(devdev);
    820       1.12    martin 				free(dev);
    821        1.1  dholland 				return error;
    822        1.1  dholland 			}
    823        1.1  dholland 		}
    824        1.2    martin 		free(devdev);
    825        1.2    martin 		free(dev);
    826        1.1  dholland 	}
    827        1.1  dholland 	return 0;
    828        1.1  dholland }
    829        1.1  dholland 
    830        1.1  dholland int
    831        1.1  dholland make_fstab(void)
    832        1.1  dholland {
    833        1.1  dholland 	FILE *f;
    834        1.1  dholland 	int i, swap_dev = -1;
    835        1.1  dholland 	const char *dump_dev;
    836        1.2    martin 	char *dev = NULL;
    837        1.2    martin 	pm_devs_t *pm_i;
    838        1.2    martin #ifndef HAVE_TMPFS
    839        1.3       mrg 	pm_devs_t *pm_with_swap = NULL;
    840        1.2    martin #endif
    841        1.1  dholland 
    842        1.1  dholland 	/* Create the fstab. */
    843        1.1  dholland 	make_target_dir("/etc");
    844        1.1  dholland 	f = target_fopen("/etc/fstab", "w");
    845        1.2    martin 	scripting_fprintf(NULL, "cat <<EOF >%s/etc/fstab\n", target_prefix());
    846        1.2    martin 
    847        1.1  dholland 	if (logfp)
    848        1.1  dholland 		(void)fprintf(logfp,
    849        1.2    martin 		    "Making %s/etc/fstab (%s).\n", target_prefix(), pm->diskdev);
    850       1.10     isaki 
    851        1.1  dholland 	if (f == NULL) {
    852        1.1  dholland 		msg_display(MSG_createfstab);
    853        1.1  dholland 		if (logfp)
    854        1.1  dholland 			(void)fprintf(logfp, "Failed to make /etc/fstab!\n");
    855        1.1  dholland 		process_menu(MENU_ok, NULL);
    856        1.2    martin #ifndef DEBUG
    857        1.1  dholland 		return 1;
    858        1.1  dholland #else
    859        1.1  dholland 		f = stdout;
    860        1.1  dholland #endif
    861        1.1  dholland 	}
    862        1.1  dholland 
    863        1.1  dholland 	scripting_fprintf(f, "# NetBSD %s/etc/fstab\n# See /usr/share/examples/"
    864        1.2    martin 			"fstab/ for more examples.\n", target_prefix());
    865        1.2    martin 	if (! partman_go) {
    866        1.2    martin 		/* We want to process only one disk... */
    867        1.2    martin 		pm_i = pm;
    868        1.2    martin 		goto onlyonediskinfstab;
    869        1.2    martin 	}
    870        1.2    martin 	SLIST_FOREACH(pm_i, &pm_head, l) {
    871        1.2    martin 		onlyonediskinfstab:
    872        1.2    martin 		for (i = 0; i < getmaxpartitions(); i++) {
    873        1.2    martin 			const char *s = "";
    874        1.2    martin 			const char *mp = pm_i->bsdlabel[i].pi_mount;
    875        1.2    martin 			const char *fstype = "ffs";
    876        1.2    martin 			int fsck_pass = 0, dump_freq = 0;
    877       1.10     isaki 
    878        1.2    martin 			if (dev != NULL)
    879        1.2    martin 				free(dev);
    880        1.2    martin 			if (pm_i->isspecial)
    881        1.2    martin 				asprintf(&dev, "%s", pm_i->diskdev);
    882        1.2    martin 			else
    883        1.2    martin 				asprintf(&dev, "%s%c", pm_i->diskdev, 'a' + i);
    884        1.2    martin 			if (dev == NULL)
    885        1.2    martin 				return (ENOMEM);
    886        1.2    martin 
    887        1.2    martin 			if (!*mp) {
    888        1.2    martin 				/*
    889        1.2    martin 				 * No mount point specified, comment out line and
    890        1.2    martin 				 * use /mnt as a placeholder for the mount point.
    891        1.2    martin 				 */
    892        1.1  dholland 				s = "# ";
    893        1.2    martin 				mp = "/mnt";
    894        1.1  dholland 			}
    895        1.2    martin 
    896        1.2    martin 			switch (pm_i->bsdlabel[i].pi_fstype) {
    897        1.2    martin 			case FS_UNUSED:
    898        1.2    martin 				continue;
    899        1.2    martin 			case FS_BSDLFS:
    900        1.2    martin 				/* If there is no LFS, just comment it out. */
    901        1.2    martin 				if (!check_lfs_progs())
    902        1.2    martin 					s = "# ";
    903        1.2    martin 				fstype = "lfs";
    904        1.2    martin 				/* FALLTHROUGH */
    905        1.2    martin 			case FS_BSDFFS:
    906        1.2    martin 				fsck_pass = (strcmp(mp, "/") == 0) ? 1 : 2;
    907        1.2    martin 				dump_freq = 1;
    908        1.2    martin 				break;
    909        1.2    martin 			case FS_MSDOS:
    910        1.2    martin 				fstype = "msdos";
    911        1.2    martin 				break;
    912        1.2    martin 			case FS_SWAP:
    913        1.2    martin 				if (pm_i->isspecial)
    914        1.2    martin 					continue;
    915        1.2    martin 				if (swap_dev == -1) {
    916        1.2    martin 					swap_dev = i;
    917        1.2    martin 					dump_dev = ",dp";
    918        1.2    martin #ifndef HAVE_TMPFS
    919        1.2    martin 					pm_with_swap = pm_i;
    920        1.2    martin #endif
    921        1.2    martin 				} else {
    922        1.2    martin 					dump_dev = "";
    923        1.2    martin 				}
    924        1.2    martin 				scripting_fprintf(f, "/dev/%s\t\tnone\tswap\tsw%s\t\t 0 0\n",
    925        1.2    martin 					dev, dump_dev);
    926        1.2    martin 				continue;
    927        1.1  dholland #ifdef USE_SYSVBFS
    928        1.2    martin 			case FS_SYSVBFS:
    929        1.2    martin 				fstype = "sysvbfs";
    930        1.2    martin 				make_target_dir("/stand");
    931        1.2    martin 				break;
    932        1.1  dholland #endif
    933        1.2    martin 			default:
    934        1.2    martin 				fstype = "???";
    935        1.2    martin 				s = "# ";
    936        1.2    martin 				break;
    937        1.2    martin 			}
    938        1.2    martin 			/* The code that remounts root rw doesn't check the partition */
    939        1.2    martin 			if (strcmp(mp, "/") == 0 && !(pm_i->bsdlabel[i].pi_flags & PIF_MOUNT))
    940        1.2    martin 				s = "# ";
    941        1.2    martin 
    942        1.2    martin 	 		scripting_fprintf(f,
    943        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",
    944        1.2    martin 			   s, dev, mp, fstype,
    945        1.2    martin 			   pm_i->bsdlabel[i].pi_flags & PIF_LOG ? ",log" : "",
    946        1.2    martin 			   pm_i->bsdlabel[i].pi_flags & PIF_MOUNT ? "" : ",noauto",
    947        1.2    martin 			   pm_i->bsdlabel[i].pi_flags & PIF_ASYNC ? ",async" : "",
    948        1.2    martin 			   pm_i->bsdlabel[i].pi_flags & PIF_NOATIME ? ",noatime" : "",
    949        1.2    martin 			   pm_i->bsdlabel[i].pi_flags & PIF_NODEV ? ",nodev" : "",
    950        1.2    martin 			   pm_i->bsdlabel[i].pi_flags & PIF_NODEVMTIME ? ",nodevmtime" : "",
    951        1.2    martin 			   pm_i->bsdlabel[i].pi_flags & PIF_NOEXEC ? ",noexec" : "",
    952        1.2    martin 			   pm_i->bsdlabel[i].pi_flags & PIF_NOSUID ? ",nosuid" : "",
    953        1.2    martin 			   dump_freq, fsck_pass);
    954        1.2    martin 	 		if (pm_i->isspecial)
    955        1.2    martin 	 			/* Special device (such as dk*) have only one partition */
    956        1.2    martin 	 			break;
    957        1.2    martin 		}
    958        1.2    martin 		/* Simple install, only one disk */
    959        1.2    martin 		if (!partman_go)
    960        1.1  dholland 			break;
    961        1.1  dholland 	}
    962        1.1  dholland 	if (tmp_ramdisk_size != 0) {
    963        1.1  dholland #ifdef HAVE_TMPFS
    964        1.1  dholland 		scripting_fprintf(f, "tmpfs\t\t/tmp\ttmpfs\trw,-m=1777,-s=%"
    965        1.1  dholland 		    PRIi64 "\n",
    966        1.1  dholland 		    tmp_ramdisk_size * 512);
    967        1.1  dholland #else
    968        1.2    martin 		if (swap_dev != -1 && pm_with_swap != NULL)
    969        1.1  dholland 			scripting_fprintf(f, "/dev/%s%c\t\t/tmp\tmfs\trw,-s=%"
    970        1.1  dholland 			    PRIi64 "\n",
    971        1.2    martin 			    pm_with_swap->diskdev, 'a' + swap_dev, tmp_ramdisk_size);
    972        1.1  dholland 		else
    973        1.1  dholland 			scripting_fprintf(f, "swap\t\t/tmp\tmfs\trw,-s=%"
    974        1.1  dholland 			    PRIi64 "\n",
    975        1.1  dholland 			    tmp_ramdisk_size);
    976        1.1  dholland #endif
    977        1.1  dholland 	}
    978        1.1  dholland 
    979        1.1  dholland 	/* Add /kern, /proc and /dev/pts to fstab and make mountpoint. */
    980        1.1  dholland 	scripting_fprintf(f, "kernfs\t\t/kern\tkernfs\trw\n");
    981        1.1  dholland 	scripting_fprintf(f, "ptyfs\t\t/dev/pts\tptyfs\trw\n");
    982        1.1  dholland 	scripting_fprintf(f, "procfs\t\t/proc\tprocfs\trw\n");
    983        1.1  dholland 	scripting_fprintf(f, "/dev/%s\t\t/cdrom\tcd9660\tro,noauto\n",
    984        1.1  dholland 	    get_default_cdrom());
    985        1.1  dholland 	scripting_fprintf(f, "%stmpfs\t\t/var/shm\ttmpfs\trw,-m1777,-sram%%25\n",
    986        1.1  dholland 	    tmpfs_on_var_shm() ? "" : "#");
    987        1.1  dholland 	make_target_dir("/kern");
    988        1.1  dholland 	make_target_dir("/proc");
    989        1.1  dholland 	make_target_dir("/dev/pts");
    990        1.1  dholland 	make_target_dir("/cdrom");
    991        1.1  dholland 	make_target_dir("/var/shm");
    992        1.1  dholland 
    993        1.1  dholland 	scripting_fprintf(NULL, "EOF\n");
    994        1.1  dholland 
    995        1.2    martin 	if (dev != NULL)
    996        1.2    martin 		free(dev);
    997        1.1  dholland 	fclose(f);
    998        1.1  dholland 	fflush(NULL);
    999        1.1  dholland 	return 0;
   1000        1.1  dholland }
   1001        1.1  dholland 
   1002        1.1  dholland 
   1003        1.1  dholland 
   1004        1.1  dholland static int
   1005        1.1  dholland /*ARGSUSED*/
   1006        1.1  dholland foundffs(struct data *list, size_t num)
   1007        1.1  dholland {
   1008        1.1  dholland 	int error;
   1009        1.1  dholland 
   1010        1.1  dholland 	if (num < 2 || strcmp(list[1].u.s_val, "/") == 0 ||
   1011        1.1  dholland 	    strstr(list[2].u.s_val, "noauto") != NULL)
   1012        1.1  dholland 		return 0;
   1013        1.1  dholland 
   1014        1.1  dholland 	error = fsck_preen(list[0].u.s_val, ' '-'a', "ffs");
   1015        1.1  dholland 	if (error != 0)
   1016        1.1  dholland 		return error;
   1017        1.1  dholland 
   1018        1.1  dholland 	error = target_mount("", list[0].u.s_val, ' '-'a', list[1].u.s_val);
   1019        1.1  dholland 	if (error != 0) {
   1020        1.1  dholland 		msg_display(MSG_mount_failed, list[0].u.s_val);
   1021        1.9    martin 		if (!ask_noyes(NULL))
   1022        1.1  dholland 			return error;
   1023        1.1  dholland 	}
   1024        1.1  dholland 	return 0;
   1025        1.1  dholland }
   1026        1.1  dholland 
   1027        1.1  dholland #ifdef USE_SYSVBFS
   1028        1.1  dholland static int
   1029        1.1  dholland /*ARGSUSED*/
   1030        1.1  dholland foundsysvbfs(struct data *list, size_t num)
   1031        1.1  dholland {
   1032        1.1  dholland 	int error;
   1033        1.1  dholland 
   1034        1.1  dholland 	if (num < 2 || strcmp(list[1].u.s_val, "/") == 0 ||
   1035        1.1  dholland 	    strstr(list[2].u.s_val, "noauto") != NULL)
   1036        1.1  dholland 		return 0;
   1037        1.1  dholland 
   1038        1.1  dholland 	error = target_mount("", list[0].u.s_val, ' '-'a', list[1].u.s_val);
   1039        1.1  dholland 	if (error != 0)
   1040        1.1  dholland 		return error;
   1041        1.1  dholland 	return 0;
   1042        1.1  dholland }
   1043        1.1  dholland #endif
   1044        1.1  dholland 
   1045        1.1  dholland /*
   1046        1.1  dholland  * Do an fsck. On failure, inform the user by showing a warning
   1047        1.1  dholland  * message and doing menu_ok() before proceeding.
   1048        1.1  dholland  * Returns 0 on success, or nonzero return code from fsck() on failure.
   1049        1.1  dholland  */
   1050        1.1  dholland static int
   1051        1.1  dholland fsck_preen(const char *disk, int ptn, const char *fsname)
   1052        1.1  dholland {
   1053        1.1  dholland 	char *prog;
   1054        1.1  dholland 	int error;
   1055        1.1  dholland 
   1056        1.2    martin 	ptn = (ptn < 0)? 0 : 'a' + ptn;
   1057        1.1  dholland 	if (fsname == NULL)
   1058        1.1  dholland 		return 0;
   1059        1.1  dholland 	/* first, check if fsck program exists, if not, assume ok */
   1060        1.1  dholland 	asprintf(&prog, "/sbin/fsck_%s", fsname);
   1061        1.1  dholland 	if (prog == NULL)
   1062        1.1  dholland 		return 0;
   1063       1.12    martin 	if (access(prog, X_OK) != 0) {
   1064       1.12    martin 		free(prog);
   1065        1.1  dholland 		return 0;
   1066       1.12    martin 	}
   1067        1.1  dholland 	if (!strcmp(fsname,"ffs"))
   1068        1.1  dholland 		fixsb(prog, disk, ptn);
   1069        1.1  dholland 	error = run_program(0, "%s -p -q /dev/r%s%c", prog, disk, ptn);
   1070        1.1  dholland 	free(prog);
   1071        1.1  dholland 	if (error != 0) {
   1072        1.1  dholland 		msg_display(MSG_badfs, disk, ptn, error);
   1073        1.9    martin 		if (ask_noyes(NULL))
   1074        1.1  dholland 			error = 0;
   1075        1.1  dholland 		/* XXX at this point maybe we should run a full fsck? */
   1076        1.1  dholland 	}
   1077        1.1  dholland 	return error;
   1078        1.1  dholland }
   1079        1.1  dholland 
   1080        1.1  dholland /* This performs the same function as the etc/rc.d/fixsb script
   1081        1.1  dholland  * which attempts to correct problems with ffs1 filesystems
   1082        1.1  dholland  * which may have been introduced by booting a netbsd-current kernel
   1083        1.1  dholland  * from between April of 2003 and January 2004. For more information
   1084        1.1  dholland  * This script was developed as a response to NetBSD pr install/25138
   1085        1.1  dholland  * Additional prs regarding the original issue include:
   1086        1.1  dholland  *  bin/17910 kern/21283 kern/21404 port-macppc/23925 port-macppc/23926
   1087        1.1  dholland  */
   1088        1.1  dholland static void
   1089        1.1  dholland fixsb(const char *prog, const char *disk, char ptn)
   1090        1.1  dholland {
   1091        1.1  dholland 	int fd;
   1092        1.1  dholland 	int rval;
   1093        1.1  dholland 	union {
   1094        1.1  dholland 		struct fs fs;
   1095        1.1  dholland 		char buf[SBLOCKSIZE];
   1096        1.1  dholland 	} sblk;
   1097        1.1  dholland 	struct fs *fs = &sblk.fs;
   1098        1.1  dholland 
   1099        1.1  dholland 	snprintf(sblk.buf, sizeof(sblk.buf), "/dev/r%s%c",
   1100        1.1  dholland 		disk, ptn == ' ' ? 0 : ptn);
   1101        1.1  dholland 	fd = open(sblk.buf, O_RDONLY);
   1102        1.1  dholland 	if (fd == -1)
   1103        1.1  dholland 		return;
   1104        1.1  dholland 
   1105        1.1  dholland 	/* Read ffsv1 main superblock */
   1106        1.1  dholland 	rval = pread(fd, sblk.buf, sizeof sblk.buf, SBLOCK_UFS1);
   1107        1.1  dholland 	close(fd);
   1108        1.1  dholland 	if (rval != sizeof sblk.buf)
   1109        1.1  dholland 		return;
   1110        1.1  dholland 
   1111        1.1  dholland 	if (fs->fs_magic != FS_UFS1_MAGIC &&
   1112        1.1  dholland 	    fs->fs_magic != FS_UFS1_MAGIC_SWAPPED)
   1113        1.1  dholland 		/* Not FFSv1 */
   1114        1.1  dholland 		return;
   1115        1.1  dholland 	if (fs->fs_old_flags & FS_FLAGS_UPDATED)
   1116        1.1  dholland 		/* properly updated fslevel 4 */
   1117        1.1  dholland 		return;
   1118        1.1  dholland 	if (fs->fs_bsize != fs->fs_maxbsize)
   1119        1.1  dholland 		/* not messed up */
   1120        1.1  dholland 		return;
   1121        1.1  dholland 
   1122        1.1  dholland 	/*
   1123        1.1  dholland 	 * OK we have a munged fs, first 'upgrade' to fslevel 4,
   1124        1.1  dholland 	 * We specify -b16 in order to stop fsck bleating that the
   1125        1.1  dholland 	 * sb doesn't match the first alternate.
   1126        1.1  dholland 	 */
   1127        1.1  dholland 	run_program(RUN_DISPLAY | RUN_PROGRESS,
   1128        1.1  dholland 	    "%s -p -b 16 -c 4 /dev/r%s%c", prog, disk, ptn);
   1129        1.1  dholland 	/* Then downgrade to fslevel 3 */
   1130        1.1  dholland 	run_program(RUN_DISPLAY | RUN_PROGRESS,
   1131        1.1  dholland 	    "%s -p -c 3 /dev/r%s%c", prog, disk, ptn);
   1132        1.1  dholland }
   1133        1.1  dholland 
   1134        1.1  dholland /*
   1135        1.1  dholland  * fsck and mount the root partition.
   1136        1.1  dholland  */
   1137        1.1  dholland static int
   1138        1.1  dholland mount_root(void)
   1139        1.1  dholland {
   1140        1.1  dholland 	int	error;
   1141        1.2    martin 	int ptn = (pm->isspecial)? 0 - 'a' : pm->rootpart;
   1142        1.1  dholland 
   1143        1.2    martin 	error = fsck_preen(pm->diskdev, ptn, "ffs");
   1144        1.1  dholland 	if (error != 0)
   1145        1.1  dholland 		return error;
   1146        1.1  dholland 
   1147        1.1  dholland 	md_pre_mount();
   1148        1.1  dholland 
   1149        1.1  dholland 	/* Mount /dev/<diskdev>a on target's "".
   1150        1.1  dholland 	 * If we pass "" as mount-on, Prefixing will DTRT.
   1151        1.1  dholland 	 * for now, use no options.
   1152        1.1  dholland 	 * XXX consider -o remount in case target root is
   1153        1.1  dholland 	 * current root, still readonly from single-user?
   1154        1.1  dholland 	 */
   1155        1.2    martin 	return target_mount("", pm->diskdev, ptn, "");
   1156        1.1  dholland }
   1157        1.1  dholland 
   1158        1.1  dholland /* Get information on the file systems mounted from the root filesystem.
   1159        1.1  dholland  * Offer to convert them into 4.4BSD inodes if they are not 4.4BSD
   1160        1.1  dholland  * inodes.  Fsck them.  Mount them.
   1161        1.1  dholland  */
   1162        1.1  dholland 
   1163        1.1  dholland int
   1164        1.1  dholland mount_disks(void)
   1165        1.1  dholland {
   1166        1.1  dholland 	char *fstab;
   1167        1.1  dholland 	int   fstabsize;
   1168        1.1  dholland 	int   error;
   1169        1.1  dholland 
   1170        1.1  dholland 	static struct lookfor fstabbuf[] = {
   1171        1.1  dholland 		{"/dev/", "/dev/%s %s ffs %s", "c", NULL, 0, 0, foundffs},
   1172        1.1  dholland 		{"/dev/", "/dev/%s %s ufs %s", "c", NULL, 0, 0, foundffs},
   1173        1.1  dholland #ifdef USE_SYSVBFS
   1174        1.1  dholland 		{"/dev/", "/dev/%s %s sysvbfs %s", "c", NULL, 0, 0,
   1175        1.1  dholland 		    foundsysvbfs},
   1176        1.1  dholland #endif
   1177        1.1  dholland 	};
   1178        1.1  dholland 	static size_t numfstabbuf = sizeof(fstabbuf) / sizeof(struct lookfor);
   1179        1.1  dholland 
   1180        1.1  dholland 	/* First the root device. */
   1181        1.1  dholland 	if (target_already_root())
   1182        1.1  dholland 		/* avoid needing to call target_already_root() again */
   1183        1.1  dholland 		targetroot_mnt[0] = 0;
   1184        1.1  dholland 	else {
   1185        1.1  dholland 		error = mount_root();
   1186        1.1  dholland 		if (error != 0 && error != EBUSY)
   1187        1.1  dholland 			return -1;
   1188        1.1  dholland 	}
   1189        1.1  dholland 
   1190        1.1  dholland 	/* Check the target /etc/fstab exists before trying to parse it. */
   1191        1.1  dholland 	if (target_dir_exists_p("/etc") == 0 ||
   1192        1.1  dholland 	    target_file_exists_p("/etc/fstab") == 0) {
   1193        1.2    martin 		msg_display(MSG_noetcfstab, pm->diskdev);
   1194        1.1  dholland 		process_menu(MENU_ok, NULL);
   1195        1.1  dholland 		return -1;
   1196        1.1  dholland 	}
   1197        1.1  dholland 
   1198        1.1  dholland 
   1199        1.1  dholland 	/* Get fstab entries from the target-root /etc/fstab. */
   1200        1.1  dholland 	fstabsize = target_collect_file(T_FILE, &fstab, "/etc/fstab");
   1201        1.1  dholland 	if (fstabsize < 0) {
   1202        1.1  dholland 		/* error ! */
   1203        1.2    martin 		msg_display(MSG_badetcfstab, pm->diskdev);
   1204        1.1  dholland 		process_menu(MENU_ok, NULL);
   1205        1.2    martin 		return -2;
   1206        1.1  dholland 	}
   1207        1.1  dholland 	error = walk(fstab, (size_t)fstabsize, fstabbuf, numfstabbuf);
   1208        1.1  dholland 	free(fstab);
   1209        1.1  dholland 
   1210        1.1  dholland 	return error;
   1211        1.1  dholland }
   1212        1.1  dholland 
   1213        1.1  dholland int
   1214       1.10     isaki set_swap_if_low_ram(const char *disk, partinfo *pp)
   1215       1.10     isaki {
   1216       1.10     isaki 	if (get_ramsize() <= 32)
   1217       1.10     isaki 		return set_swap(disk, pp);
   1218       1.10     isaki 	return 0;
   1219        1.7       abs }
   1220        1.7       abs 
   1221        1.7       abs int
   1222        1.1  dholland set_swap(const char *disk, partinfo *pp)
   1223        1.1  dholland {
   1224        1.1  dholland 	int i;
   1225        1.1  dholland 	char *cp;
   1226        1.1  dholland 	int rval;
   1227        1.1  dholland 
   1228        1.1  dholland 	if (pp == NULL)
   1229        1.2    martin 		pp = pm->oldlabel;
   1230        1.1  dholland 
   1231        1.1  dholland 	for (i = 0; i < MAXPARTITIONS; i++) {
   1232        1.1  dholland 		if (pp[i].pi_fstype != FS_SWAP)
   1233        1.1  dholland 			continue;
   1234        1.1  dholland 		asprintf(&cp, "/dev/%s%c", disk, 'a' + i);
   1235        1.1  dholland 		rval = swapctl(SWAP_ON, cp, 0);
   1236        1.1  dholland 		free(cp);
   1237        1.1  dholland 		if (rval != 0)
   1238        1.1  dholland 			return -1;
   1239        1.1  dholland 	}
   1240        1.1  dholland 
   1241        1.1  dholland 	return 0;
   1242        1.1  dholland }
   1243        1.1  dholland 
   1244        1.1  dholland int
   1245        1.1  dholland check_swap(const char *disk, int remove_swap)
   1246        1.1  dholland {
   1247        1.1  dholland 	struct swapent *swap;
   1248        1.1  dholland 	char *cp;
   1249        1.1  dholland 	int nswap;
   1250        1.1  dholland 	int l;
   1251        1.1  dholland 	int rval = 0;
   1252        1.1  dholland 
   1253        1.1  dholland 	nswap = swapctl(SWAP_NSWAP, 0, 0);
   1254        1.1  dholland 	if (nswap <= 0)
   1255        1.1  dholland 		return 0;
   1256        1.1  dholland 
   1257        1.1  dholland 	swap = malloc(nswap * sizeof *swap);
   1258        1.1  dholland 	if (swap == NULL)
   1259        1.1  dholland 		return -1;
   1260        1.1  dholland 
   1261        1.1  dholland 	nswap = swapctl(SWAP_STATS, swap, nswap);
   1262        1.1  dholland 	if (nswap < 0)
   1263        1.1  dholland 		goto bad_swap;
   1264        1.1  dholland 
   1265        1.1  dholland 	l = strlen(disk);
   1266        1.1  dholland 	while (--nswap >= 0) {
   1267        1.1  dholland 		/* Should we check the se_dev or se_path? */
   1268        1.1  dholland 		cp = swap[nswap].se_path;
   1269        1.1  dholland 		if (memcmp(cp, "/dev/", 5) != 0)
   1270        1.1  dholland 			continue;
   1271        1.1  dholland 		if (memcmp(cp + 5, disk, l) != 0)
   1272        1.1  dholland 			continue;
   1273        1.1  dholland 		if (!isalpha(*(unsigned char *)(cp + 5 + l)))
   1274        1.1  dholland 			continue;
   1275        1.1  dholland 		if (cp[5 + l + 1] != 0)
   1276        1.1  dholland 			continue;
   1277        1.1  dholland 		/* ok path looks like it is for this device */
   1278        1.1  dholland 		if (!remove_swap) {
   1279        1.1  dholland 			/* count active swap areas */
   1280        1.1  dholland 			rval++;
   1281        1.1  dholland 			continue;
   1282        1.1  dholland 		}
   1283        1.1  dholland 		if (swapctl(SWAP_OFF, cp, 0) == -1)
   1284        1.1  dholland 			rval = -1;
   1285        1.1  dholland 	}
   1286        1.1  dholland 
   1287        1.1  dholland     done:
   1288        1.1  dholland 	free(swap);
   1289        1.1  dholland 	return rval;
   1290        1.1  dholland 
   1291        1.1  dholland     bad_swap:
   1292        1.1  dholland 	rval = -1;
   1293        1.1  dholland 	goto done;
   1294        1.1  dholland }
   1295        1.1  dholland 
   1296        1.1  dholland #ifdef HAVE_BOOTXX_xFS
   1297        1.1  dholland char *
   1298        1.1  dholland bootxx_name(void)
   1299        1.1  dholland {
   1300        1.1  dholland 	int fstype;
   1301        1.1  dholland 	const char *bootxxname;
   1302        1.1  dholland 	char *bootxx;
   1303        1.1  dholland 
   1304        1.1  dholland 	/* check we have boot code for the root partition type */
   1305        1.2    martin 	fstype = pm->bsdlabel[pm->rootpart].pi_fstype;
   1306        1.1  dholland 	switch (fstype) {
   1307        1.1  dholland #if defined(BOOTXX_FFSV1) || defined(BOOTXX_FFSV2)
   1308        1.1  dholland 	case FS_BSDFFS:
   1309        1.2    martin 		if (pm->bsdlabel[pm->rootpart].pi_flags & PIF_FFSv2) {
   1310        1.1  dholland #ifdef BOOTXX_FFSV2
   1311        1.1  dholland 			bootxxname = BOOTXX_FFSV2;
   1312        1.1  dholland #else
   1313        1.1  dholland 			bootxxname = NULL;
   1314        1.1  dholland #endif
   1315        1.1  dholland 		} else {
   1316        1.1  dholland #ifdef BOOTXX_FFSV1
   1317        1.1  dholland 			bootxxname = BOOTXX_FFSV1;
   1318        1.1  dholland #else
   1319        1.1  dholland 			bootxxname = NULL;
   1320        1.1  dholland #endif
   1321        1.1  dholland 		}
   1322        1.1  dholland 		break;
   1323        1.1  dholland #endif
   1324       1.11  pgoyette #ifdef BOOTXX_LFSV2
   1325        1.1  dholland 	case FS_BSDLFS:
   1326       1.11  pgoyette 		bootxxname = BOOTXX_LFSV2;
   1327        1.1  dholland 		break;
   1328        1.1  dholland #endif
   1329        1.1  dholland 	default:
   1330        1.1  dholland 		bootxxname = NULL;
   1331        1.1  dholland 		break;
   1332        1.1  dholland 	}
   1333        1.1  dholland 
   1334        1.1  dholland 	if (bootxxname == NULL)
   1335        1.1  dholland 		return NULL;
   1336        1.1  dholland 
   1337        1.1  dholland 	asprintf(&bootxx, "%s/%s", BOOTXXDIR, bootxxname);
   1338        1.1  dholland 	return bootxx;
   1339        1.1  dholland }
   1340        1.1  dholland #endif
   1341        1.2    martin 
   1342        1.2    martin static int
   1343        1.2    martin get_fsid_by_gptuuid(const char *str)
   1344        1.2    martin {
   1345        1.2    martin 	int i;
   1346        1.2    martin 	uuid_t uuid;
   1347        1.2    martin 	uint32_t status;
   1348        1.2    martin 
   1349        1.2    martin 	uuid_from_string(str, &uuid, &status);
   1350        1.2    martin 	if (status == uuid_s_ok) {
   1351        1.2    martin 		for (i = 0; gpt_filesystems[i].id > 0; i++)
   1352        1.2    martin 			if (uuid_equal(&uuid, &(gpt_filesystems[i].uuid), NULL))
   1353        1.2    martin 				return gpt_filesystems[i].id;
   1354        1.2    martin 	}
   1355        1.2    martin 	return FS_OTHER;
   1356        1.2    martin }
   1357        1.2    martin 
   1358        1.2    martin const char *
   1359        1.2    martin get_gptfs_by_id(int filesystem)
   1360        1.2    martin {
   1361        1.2    martin 	int i;
   1362        1.2    martin 	for (i = 0; gpt_filesystems[i].id > 0; i++)
   1363        1.2    martin 		if (filesystem == gpt_filesystems[i].id)
   1364        1.2    martin 			return gpt_filesystems[i].name;
   1365        1.2    martin 	return NULL;
   1366        1.2    martin }
   1367        1.2    martin 
   1368        1.2    martin /* from dkctl.c */
   1369        1.2    martin static int
   1370        1.2    martin get_dkwedges_sort(const void *a, const void *b)
   1371        1.2    martin {
   1372        1.2    martin 	const struct dkwedge_info *dkwa = a, *dkwb = b;
   1373        1.2    martin 	const daddr_t oa = dkwa->dkw_offset, ob = dkwb->dkw_offset;
   1374        1.2    martin 	return (oa < ob) ? -1 : (oa > ob) ? 1 : 0;
   1375        1.2    martin }
   1376        1.2    martin 
   1377        1.2    martin int
   1378        1.2    martin get_dkwedges(struct dkwedge_info **dkw, const char *diskdev)
   1379        1.2    martin {
   1380        1.2    martin 	int fd;
   1381        1.2    martin 	char buf[STRSIZE];
   1382        1.2    martin 	size_t bufsize;
   1383        1.2    martin 	struct dkwedge_list dkwl;
   1384        1.2    martin 
   1385        1.2    martin 	*dkw = NULL;
   1386        1.2    martin 	dkwl.dkwl_buf = *dkw;
   1387        1.2    martin 	dkwl.dkwl_bufsize = 0;
   1388        1.2    martin 	fd = opendisk(diskdev, O_RDONLY, buf, STRSIZE, 0);
   1389        1.2    martin 	if (fd < 0)
   1390        1.2    martin 		return -1;
   1391        1.2    martin 
   1392        1.2    martin 	for (;;) {
   1393        1.2    martin 		if (ioctl(fd, DIOCLWEDGES, &dkwl) == -1)
   1394        1.2    martin 			return -2;
   1395        1.2    martin 		if (dkwl.dkwl_nwedges == dkwl.dkwl_ncopied)
   1396        1.2    martin 			break;
   1397        1.2    martin 		bufsize = dkwl.dkwl_nwedges * sizeof(**dkw);
   1398        1.2    martin 		if (dkwl.dkwl_bufsize < bufsize) {
   1399        1.2    martin 			*dkw = realloc(dkwl.dkwl_buf, bufsize);
   1400        1.2    martin 			if (*dkw == NULL)
   1401        1.2    martin 				return -3;
   1402        1.2    martin 			dkwl.dkwl_buf = *dkw;
   1403        1.2    martin 			dkwl.dkwl_bufsize = bufsize;
   1404        1.2    martin 		}
   1405        1.2    martin 	}
   1406        1.2    martin 
   1407        1.5    martin 	if (dkwl.dkwl_nwedges > 0 && *dkw != NULL)
   1408        1.2    martin 		qsort(*dkw, dkwl.dkwl_nwedges, sizeof(**dkw), get_dkwedges_sort);
   1409        1.2    martin 
   1410        1.2    martin 	close(fd);
   1411        1.2    martin 	return dkwl.dkwl_nwedges;
   1412        1.2    martin }
   1413        1.2    martin 
   1414        1.2    martin /* XXX: rewrite */
   1415        1.2    martin static int
   1416        1.2    martin incoregpt(pm_devs_t *pm_cur, partinfo *lp)
   1417        1.2    martin {
   1418        1.2    martin 	int i, num;
   1419        1.4       riz 	unsigned int p_num;
   1420        1.4       riz 	uint64_t p_start, p_size;
   1421        1.2    martin 	char *textbuf, *t, *tt, p_type[STRSIZE];
   1422        1.2    martin 	struct dkwedge_info *dkw;
   1423        1.2    martin 
   1424        1.2    martin 	num = get_dkwedges(&dkw, pm_cur->diskdev);
   1425        1.2    martin 	if (dkw != NULL) {
   1426        1.2    martin 		for (i = 0; i < num && i < MAX_WEDGES; i++)
   1427        1.2    martin 			run_program(RUN_SILENT, "dkctl %s delwedge %s",
   1428        1.2    martin 				pm_cur->diskdev, dkw[i].dkw_devname);
   1429        1.2    martin 		free (dkw);
   1430        1.2    martin 	}
   1431        1.2    martin 
   1432        1.2    martin 	if (collect(T_OUTPUT, &textbuf, "gpt show -u %s 2>/dev/null", pm_cur->diskdev) < 1)
   1433        1.2    martin 		return -1;
   1434        1.2    martin 
   1435        1.2    martin 	(void)strtok(textbuf, "\n"); /* ignore first line */
   1436        1.2    martin 	while ((t = strtok(NULL, "\n")) != NULL) {
   1437        1.4       riz 		i = 0; p_start = 0; p_size = 0; p_num = 0; strcpy(p_type, ""); /* init */
   1438        1.2    martin 		while ((tt = strsep(&t, " \t")) != NULL) {
   1439        1.4       riz 			if (strlen(tt) == 0)
   1440        1.2    martin 				continue;
   1441        1.2    martin 			if (i == 0)
   1442        1.4       riz 				p_start = strtouq(tt, NULL, 10);
   1443        1.2    martin 			if (i == 1)
   1444        1.4       riz 				p_size = strtouq(tt, NULL, 10);
   1445        1.2    martin 			if (i == 2)
   1446        1.4       riz 				p_num = strtouq(tt, NULL, 10);
   1447        1.2    martin 			if (i > 2 || (i == 2 && p_num == 0))
   1448        1.2    martin 				if (
   1449        1.2    martin 					strcmp(tt, "GPT") &&
   1450        1.2    martin 					strcmp(tt, "part") &&
   1451        1.2    martin 					strcmp(tt, "-")
   1452        1.2    martin 					)
   1453       1.13  christos 						strlcat(p_type, tt, STRSIZE);
   1454        1.2    martin 			i++;
   1455        1.2    martin 		}
   1456        1.4       riz 		if (p_start == 0 || p_size == 0)
   1457        1.2    martin 			continue;
   1458        1.2    martin 		else if (! strcmp(p_type, "Pritable"))
   1459        1.2    martin 			pm_cur->ptstart = p_start + p_size;
   1460        1.2    martin 		else if (! strcmp(p_type, "Sectable"))
   1461        1.2    martin 			pm_cur->ptsize = p_start - pm_cur->ptstart - 1;
   1462        1.4       riz 		else if (p_num == 0 && strlen(p_type) > 0)
   1463        1.2    martin 			/* Utilitary entry (PMBR, etc) */
   1464        1.2    martin 			continue;
   1465        1.4       riz 		else if (p_num == 0) {
   1466        1.2    martin 			/* Free space */
   1467        1.2    martin 			continue;
   1468        1.2    martin 		} else {
   1469        1.2    martin 			/* Usual partition */
   1470        1.2    martin 			lp[p_num].pi_size = p_size;
   1471        1.2    martin 			lp[p_num].pi_offset = p_start;
   1472        1.2    martin 			lp[p_num].pi_fstype = get_fsid_by_gptuuid(p_type);
   1473        1.2    martin 		}
   1474        1.2    martin 	}
   1475        1.2    martin 	free(textbuf);
   1476        1.2    martin 
   1477        1.2    martin 	return 0;
   1478        1.2    martin }
   1479        1.2    martin 
   1480        1.6    martin static bool
   1481        1.2    martin is_gpt(const char *dev)
   1482        1.2    martin {
   1483  1.13.12.1  pgoyette 
   1484        1.8    martin 	check_available_binaries();
   1485        1.8    martin 
   1486        1.8    martin 	if (!have_gpt)
   1487        1.6    martin 		return false;
   1488        1.6    martin 
   1489  1.13.12.1  pgoyette 	return run_program(RUN_SILENT | RUN_ERROR_OK,
   1490  1.13.12.1  pgoyette 		"gpt -qr header %s", dev) == 0;
   1491        1.2    martin }
   1492