Home | History | Annotate | Line # | Download | only in ata
wd.c revision 1.180
      1 /*	$NetBSD: wd.c,v 1.180 1998/09/07 21:28:22 drochner Exp $ */
      2 
      3 /*-
      4  * Copyright (c) 1998 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * This code is derived from software contributed to The NetBSD Foundation
      8  * by Charles M. Hannum and by Onno van der Linden.
      9  *
     10  * Redistribution and use in source and binary forms, with or without
     11  * modification, are permitted provided that the following conditions
     12  * are met:
     13  * 1. Redistributions of source code must retain the above copyright
     14  *    notice, this list of conditions and the following disclaimer.
     15  * 2. Redistributions in binary form must reproduce the above copyright
     16  *    notice, this list of conditions and the following disclaimer in the
     17  *    documentation and/or other materials provided with the distribution.
     18  * 3. All advertising materials mentioning features or use of this software
     19  *    must display the following acknowledgement:
     20  *        This product includes software developed by the NetBSD
     21  *        Foundation, Inc. and its contributors.
     22  * 4. Neither the name of The NetBSD Foundation nor the names of its
     23  *    contributors may be used to endorse or promote products derived
     24  *    from this software without specific prior written permission.
     25  *
     26  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     27  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     28  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     29  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     30  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     31  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     32  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     33  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     34  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     35  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     36  * POSSIBILITY OF SUCH DAMAGE.
     37  */
     38 
     39 #include "rnd.h"
     40 
     41 #include <sys/param.h>
     42 #include <sys/systm.h>
     43 #include <sys/kernel.h>
     44 #include <sys/conf.h>
     45 #include <sys/file.h>
     46 #include <sys/stat.h>
     47 #include <sys/ioctl.h>
     48 #include <sys/buf.h>
     49 #include <sys/uio.h>
     50 #include <sys/malloc.h>
     51 #include <sys/device.h>
     52 #include <sys/disklabel.h>
     53 #include <sys/disk.h>
     54 #include <sys/syslog.h>
     55 #include <sys/proc.h>
     56 #if NRND > 0
     57 #include <sys/rnd.h>
     58 #endif
     59 
     60 #include <vm/vm.h>
     61 
     62 #include <machine/intr.h>
     63 #include <machine/bus.h>
     64 
     65 #include <dev/ic/wdcreg.h>		/* XXX */
     66 #include <dev/ic/wdcvar.h>		/* XXX */
     67 #include <dev/ata/wdlink.h>
     68 #include "locators.h"
     69 
     70 #define	WAITTIME	(4 * hz)	/* time to wait for a completion */
     71 
     72 #define	WDIORETRIES	5		/* number of retries before giving up */
     73 
     74 #define	WDUNIT(dev)			DISKUNIT(dev)
     75 #define	WDPART(dev)			DISKPART(dev)
     76 #define	MAKEWDDEV(maj, unit, part)	MAKEDISKDEV(maj, unit, part)
     77 
     78 #define	WDLABELDEV(dev)	(MAKEWDDEV(major(dev), WDUNIT(dev), RAW_PART))
     79 
     80 #ifdef WDDEBUG
     81 #define WDDEBUG_PRINT(args)		printf args
     82 #else
     83 #define WDDEBUG_PRINT(args)
     84 #endif
     85 
     86 struct wd_softc {
     87 	struct device sc_dev;
     88 	struct disk sc_dk;
     89 	struct wd_link *d_link;
     90 	struct buf sc_q;
     91 #if NRND > 0
     92 	rndsource_element_t	rnd_source;
     93 #endif
     94 };
     95 
     96 int	wdprobe		__P((struct device *, struct cfdata *, void *));
     97 void	wdattach	__P((struct device *, struct device *, void *));
     98 int	wdprint		__P((void *, char *));
     99 
    100 struct cfattach wd_ca = {
    101 	sizeof(struct wd_softc), wdprobe, wdattach
    102 };
    103 
    104 extern struct cfdriver wd_cd;
    105 
    106 void	wdgetdefaultlabel __P((struct wd_softc *, struct disklabel *));
    107 void	wdgetdisklabel	__P((struct wd_softc *));
    108 int	wd_get_parms	__P((struct wd_softc *));
    109 void	wdstrategy	__P((struct buf *));
    110 void	wdstart		__P((void *));
    111 
    112 struct dkdriver wddkdriver = { wdstrategy };
    113 
    114 /* XXX: these should go elsewhere */
    115 cdev_decl(wd);
    116 bdev_decl(wd);
    117 
    118 void	wdfinish	__P((struct wd_softc *, struct buf *));
    119 int	wdsetctlr	__P((struct wd_link *));
    120 #ifdef HAS_BAD144_HANDLING
    121 static void bad144intern __P((struct wd_softc *));
    122 #endif
    123 int	wdlock		__P((struct wd_link *));
    124 void	wdunlock	__P((struct wd_link *));
    125 
    126 int
    127 wdprobe(parent, match, aux)
    128 	struct device *parent;
    129 	struct cfdata *match;
    130 
    131 	void *aux;
    132 {
    133 	struct wd_link *d_link = aux;
    134 	int drive;
    135 
    136 	if (d_link == NULL)
    137 		return 0;
    138 	if (d_link->type != ATA)
    139 		return 0;
    140 
    141 	drive = d_link->drive;
    142 	if (match->cf_loc[ATACF_DRIVE] != ATACF_DRIVE_DEFAULT &&
    143 		match->cf_loc[ATACF_DRIVE] != drive)
    144 		return 0;
    145 
    146 	return 1;
    147 }
    148 
    149 void
    150 wdattach(parent, self, aux)
    151 	struct device *parent, *self;
    152 	void *aux;
    153 {
    154 	struct wd_softc *wd = (void *)self;
    155 	struct wd_link *d_link= aux;
    156 	int i, blank;
    157 	char buf[41], c, *p, *q;
    158 
    159 	wd->d_link = d_link;
    160 	d_link->openings = 1;
    161 	d_link->wd_softc = (caddr_t)wd;
    162 
    163 	/*
    164 	 * Initialize and attach the disk structure.
    165 	 */
    166 	wd->sc_dk.dk_driver = &wddkdriver;
    167 	wd->sc_dk.dk_name = wd->sc_dev.dv_xname;
    168 	disk_attach(&wd->sc_dk);
    169 
    170 	d_link->sc_lp = wd->sc_dk.dk_label;
    171 
    172 	wdc_get_parms((struct wdc_softc *)d_link->wdc_softc, d_link);
    173 	for (blank = 0, p = d_link->sc_params.wdp_model, q = buf, i = 0;
    174 	     i < sizeof(d_link->sc_params.wdp_model); i++) {
    175 		c = *p++;
    176 		if (c == '\0')
    177 			break;
    178 		if (c != ' ') {
    179 			if (blank) {
    180 				*q++ = ' ';
    181 				blank = 0;
    182 			}
    183 			*q++ = c;
    184 		} else
    185 			blank = 1;
    186 	}
    187 	*q++ = '\0';
    188 
    189 	printf(": <%s>\n", buf);
    190 
    191 	if ((d_link->sc_params.wdp_capabilities1 & WD_CAP_DMA) != 0 &&
    192 	    d_link->sc_mode == WDM_DMA) {
    193 		d_link->sc_mode = WDM_DMA;
    194 	} else if ((d_link->sc_params.wdp_multi & 0xff) > 1) {
    195 		d_link->sc_mode = WDM_PIOMULTI;
    196 		d_link->sc_multiple =
    197 		    min(d_link->sc_params.wdp_multi & 0xff, 16);
    198 	} else {
    199 		d_link->sc_mode = WDM_PIOSINGLE;
    200 		d_link->sc_multiple = 1;
    201 	}
    202 
    203 	printf("%s: using", wd->sc_dev.dv_xname);
    204 	if (d_link->sc_mode == WDM_DMA)
    205 		printf(" dma transfers,");
    206 	else
    207 		printf(" %d-sector %d-bit pio transfers,",
    208 		    d_link->sc_multiple,
    209 		    (((struct wdc_softc *)d_link->wdc_softc)->sc_cap &
    210 		      WDC_CAPABILITY_DATA32) == 0 ? 16 : 32);
    211 
    212 	/* Prior to ATA-4, LBA was optional. */
    213 	if ((d_link->sc_params.wdp_capabilities1 & WD_CAP_LBA) != 0)
    214 		d_link->sc_flags |= WDF_LBA;
    215 #if 0
    216 	/* ATA-4 requires LBA. */
    217 	if (d_link->sc_params.wdp_ataversion != 0xffff &&
    218 	    d_link->sc_params.wdp_ataversion >= WD_VER_ATA4)
    219 		d_link->sc_flags |= WDF_LBA;
    220 #endif
    221 
    222 	if ((d_link->sc_flags & WDF_LBA) != 0) {
    223 		printf(" lba mode\n");
    224 		d_link->sc_capacity =
    225 		    (d_link->sc_params.wdp_capacity[1] << 16) |
    226 		    d_link->sc_params.wdp_capacity[0];
    227 		printf("%s %dMB, %d cyl, %d head, %d sec, %d bytes/sect x %d sectors\n",
    228 		    self->dv_xname,
    229 		    d_link->sc_capacity / (1048576 / DEV_BSIZE),
    230 		    d_link->sc_params.wdp_cylinders,
    231 		    d_link->sc_params.wdp_heads,
    232 		    d_link->sc_params.wdp_sectors,
    233 		    DEV_BSIZE,
    234 		    d_link->sc_capacity);
    235 	} else {
    236 		printf(" chs mode\n");
    237 		d_link->sc_capacity =
    238 		    d_link->sc_params.wdp_cylinders *
    239 		    d_link->sc_params.wdp_heads *
    240 		    d_link->sc_params.wdp_sectors;
    241 		printf("%s %dMB, %d cyl, %d head, %d sec, %d bytes/sect x %d sectors\n",
    242 		    self->dv_xname,
    243 		    d_link->sc_capacity / (1048576 / DEV_BSIZE),
    244 		    d_link->sc_params.wdp_cylinders,
    245 		    d_link->sc_params.wdp_heads,
    246 		    d_link->sc_params.wdp_sectors,
    247 		    DEV_BSIZE,
    248 		    d_link->sc_capacity);
    249 	}
    250 
    251 #if NRND > 0
    252 	rnd_attach_source(&wd->rnd_source, wd->sc_dev.dv_xname, RND_TYPE_DISK);
    253 #endif
    254 }
    255 
    256 /*
    257  * Read/write routine for a buffer.  Validates the arguments and schedules the
    258  * transfer.  Does not wait for the transfer to complete.
    259  */
    260 void
    261 wdstrategy(bp)
    262 	struct buf *bp;
    263 {
    264 	struct wd_softc *wd = wd_cd.cd_devs[WDUNIT(bp->b_dev)];
    265 	struct wd_link *d_link= wd->d_link;
    266 	int s;
    267 
    268 	/* Valid request?  */
    269 	if (bp->b_blkno < 0 ||
    270 	    (bp->b_bcount % wd->sc_dk.dk_label->d_secsize) != 0 ||
    271 	    (bp->b_bcount / wd->sc_dk.dk_label->d_secsize) >= (1 << NBBY)) {
    272 		bp->b_error = EINVAL;
    273 		goto bad;
    274 	}
    275 
    276 	/* If device invalidated (e.g. media change, door open), error. */
    277 	if ((d_link->sc_flags & WDF_LOADED) == 0) {
    278 		bp->b_error = EIO;
    279 		goto bad;
    280 	}
    281 
    282 	/* If it's a null transfer, return immediately. */
    283 	if (bp->b_bcount == 0)
    284 		goto done;
    285 
    286 	/*
    287 	 * Do bounds checking, adjust transfer. if error, process.
    288 	 * If end of partition, just return.
    289 	 */
    290 	if (WDPART(bp->b_dev) != RAW_PART &&
    291 	    bounds_check_with_label(bp, wd->sc_dk.dk_label,
    292 	    (d_link->sc_flags & (WDF_WLABEL|WDF_LABELLING)) != 0) <= 0)
    293 		goto done;
    294 
    295 	/* Queue transfer on drive, activate drive and controller if idle. */
    296 	s = splbio();
    297 	disksort(&wd->sc_q, bp);
    298 	wdstart(wd);
    299 	splx(s);
    300 	return;
    301 
    302 bad:
    303 	bp->b_flags |= B_ERROR;
    304 done:
    305 	/* Toss transfer; we're done early. */
    306 	bp->b_resid = bp->b_bcount;
    307 	biodone(bp);
    308 }
    309 
    310 /*
    311  * Queue a drive for I/O.
    312  */
    313 void
    314 wdstart(arg)
    315 	void *arg;
    316 {
    317 	struct wd_softc *wd = arg;
    318 	struct buf *dp, *bp=0;
    319 	struct wd_link *d_link = wd->d_link;
    320 	struct wdc_xfer *xfer;
    321 	daddr_t blkno, p_offset;
    322 
    323 	while (d_link->openings > 0) {
    324 
    325 		/* Is there a buf for us ? */
    326 		dp = &wd->sc_q;
    327 		if ((bp = dp->b_actf) == NULL)  /* yes, an assign */
    328                  	return;
    329 		dp->b_actf = bp->b_actf;
    330 
    331 		/*
    332 		 * Make the command. First lock the device
    333 		 */
    334 		d_link->openings--;
    335 		if (WDPART(bp->b_dev) != RAW_PART)
    336 			p_offset =
    337 		  wd->sc_dk.dk_label->d_partitions[WDPART(bp->b_dev)].p_offset;
    338 		else
    339 			p_offset = 0;
    340 		blkno = bp->b_blkno + p_offset;
    341 		blkno /= (wd->d_link->sc_lp->d_secsize / DEV_BSIZE);
    342 
    343 		xfer = wdc_get_xfer(0);
    344 		if (xfer == NULL)
    345 			panic("wdc_xfer");
    346 
    347 		xfer->d_link = d_link;
    348 		xfer->c_bp = bp;
    349 		xfer->databuf = bp->b_data;
    350 		xfer->c_bcount = bp->b_bcount;
    351 		xfer->c_flags |= bp->b_flags & (B_READ|B_WRITE);
    352 		xfer->c_blkno = blkno;
    353 
    354 		/* Instrumentation. */
    355 		disk_busy(&wd->sc_dk);
    356 		wdc_exec_xfer((struct wdc_softc *)wd->d_link->wdc_softc,
    357 			wd->d_link, xfer);
    358 	}
    359 }
    360 
    361 int
    362 wdread(dev, uio, flags)
    363 	dev_t dev;
    364 	struct uio *uio;
    365 	int flags;
    366 {
    367 
    368 	WDDEBUG_PRINT(("wdread\n"));
    369 	return (physio(wdstrategy, NULL, dev, B_READ, minphys, uio));
    370 }
    371 
    372 int
    373 wdwrite(dev, uio, flags)
    374 	dev_t dev;
    375 	struct uio *uio;
    376 	int flags;
    377 {
    378 
    379 	WDDEBUG_PRINT(("wdwrite\n"));
    380 	return (physio(wdstrategy, NULL, dev, B_WRITE, minphys, uio));
    381 }
    382 
    383 /*
    384  * Wait interruptibly for an exclusive lock.
    385  *
    386  * XXX
    387  * Several drivers do this; it should be abstracted and made MP-safe.
    388  */
    389 int
    390 wdlock(d_link)
    391 	struct wd_link *d_link;
    392 {
    393 	int error;
    394 	int s;
    395 
    396 	WDDEBUG_PRINT(("wdlock\n"));
    397 
    398 	s = splbio();
    399 
    400 	while ((d_link->sc_flags & WDF_LOCKED) != 0) {
    401 		d_link->sc_flags |= WDF_WANTED;
    402 		if ((error = tsleep(d_link, PRIBIO | PCATCH,
    403 		    "wdlck", 0)) != 0) {
    404 			splx(s);
    405 			return error;
    406 		}
    407 	}
    408 	d_link->sc_flags |= WDF_LOCKED;
    409 	splx(s);
    410 	return 0;
    411 }
    412 
    413 /*
    414  * Unlock and wake up any waiters.
    415  */
    416 void
    417 wdunlock(d_link)
    418 	struct wd_link *d_link;
    419 {
    420 
    421 	WDDEBUG_PRINT(("wdunlock\n"));
    422 
    423 	d_link->sc_flags &= ~WDF_LOCKED;
    424 	if ((d_link->sc_flags & WDF_WANTED) != 0) {
    425 		d_link->sc_flags &= ~WDF_WANTED;
    426 		wakeup(d_link);
    427 	}
    428 }
    429 
    430 int
    431 wdopen(dev, flag, fmt, p)
    432 	dev_t dev;
    433 	int flag, fmt;
    434 	struct proc *p;
    435 {
    436 	struct wd_softc *wd;
    437 	struct wd_link *d_link;
    438 	int unit, part;
    439 	int error;
    440 
    441 	WDDEBUG_PRINT(("wdopen\n"));
    442 
    443 	unit = WDUNIT(dev);
    444 	if (unit >= wd_cd.cd_ndevs)
    445 		return ENXIO;
    446 	wd = wd_cd.cd_devs[unit];
    447 	if (wd == NULL)
    448 		return ENXIO;
    449 
    450 	d_link = wd->d_link;
    451 	if ((error = wdlock(d_link)) != 0)
    452 		return error;
    453 
    454 	if (wd->sc_dk.dk_openmask != 0) {
    455 		/*
    456 		 * If any partition is open, but the disk has been invalidated,
    457 		 * disallow further opens.
    458 		 */
    459 		if ((d_link->sc_flags & WDF_LOADED) == 0) {
    460 			error = EIO;
    461 			goto bad3;
    462 		}
    463 	} else {
    464 		if ((d_link->sc_flags & WDF_LOADED) == 0) {
    465 			d_link->sc_flags |= WDF_LOADED;
    466 
    467 			/* Load the physical device parameters. */
    468 			if (wdc_get_parms((struct wdc_softc *)d_link->wdc_softc,
    469 				d_link) != 0) {
    470 				error = ENXIO;
    471 				goto bad2;
    472 			}
    473 
    474 			/* Load the partition info if not already loaded. */
    475 			wdgetdisklabel(wd);
    476 		}
    477 	}
    478 
    479 	part = WDPART(dev);
    480 
    481 	/* Check that the partition exists. */
    482 	if (part != RAW_PART &&
    483 	    (part >= wd->sc_dk.dk_label->d_npartitions ||
    484 	     wd->sc_dk.dk_label->d_partitions[part].p_fstype == FS_UNUSED)) {
    485 		error = ENXIO;
    486 		goto bad;
    487 	}
    488 
    489 	/* Insure only one open at a time. */
    490 	switch (fmt) {
    491 	case S_IFCHR:
    492 		wd->sc_dk.dk_copenmask |= (1 << part);
    493 		break;
    494 	case S_IFBLK:
    495 		wd->sc_dk.dk_bopenmask |= (1 << part);
    496 		break;
    497 	}
    498 	wd->sc_dk.dk_openmask = wd->sc_dk.dk_copenmask | wd->sc_dk.dk_bopenmask;
    499 
    500 	wdunlock(d_link);
    501 	return 0;
    502 
    503 bad2:
    504 	d_link->sc_flags &= ~WDF_LOADED;
    505 
    506 bad:
    507 	if (wd->sc_dk.dk_openmask == 0) {
    508 	}
    509 
    510 bad3:
    511 	wdunlock(d_link);
    512 	return error;
    513 }
    514 
    515 int
    516 wdclose(dev, flag, fmt, p)
    517 	dev_t dev;
    518 	int flag, fmt;
    519 	struct proc *p;
    520 {
    521 	struct wd_softc *wd = wd_cd.cd_devs[WDUNIT(dev)];
    522 	int part = WDPART(dev);
    523 	int error;
    524 
    525 	if ((error = wdlock(wd->d_link)) != 0)
    526 		return error;
    527 
    528 	switch (fmt) {
    529 	case S_IFCHR:
    530 		wd->sc_dk.dk_copenmask &= ~(1 << part);
    531 		break;
    532 	case S_IFBLK:
    533 		wd->sc_dk.dk_bopenmask &= ~(1 << part);
    534 		break;
    535 	}
    536 	wd->sc_dk.dk_openmask = wd->sc_dk.dk_copenmask | wd->sc_dk.dk_bopenmask;
    537 
    538 	if (wd->sc_dk.dk_openmask == 0) {
    539 		/* XXXX Must wait for I/O to complete! */
    540 	}
    541 
    542 	wdunlock(wd->d_link);
    543 	return 0;
    544 }
    545 
    546 void
    547 wdgetdefaultlabel(wd, lp)
    548 	struct wd_softc *wd;
    549 	struct disklabel *lp;
    550 {
    551 	struct wd_link *d_link = wd->d_link;
    552 
    553 	bzero(lp, sizeof(struct disklabel));
    554 
    555 	lp->d_secsize = DEV_BSIZE;
    556 	lp->d_ntracks = d_link->sc_params.wdp_heads;
    557 	lp->d_nsectors = d_link->sc_params.wdp_sectors;
    558 	lp->d_ncylinders = d_link->sc_params.wdp_cylinders;
    559 	lp->d_secpercyl = lp->d_ntracks * lp->d_nsectors;
    560 
    561 #if 0
    562 	lp->d_type = DTYPE_ST506;
    563 #endif
    564 	strncpy(lp->d_typename, d_link->sc_params.wdp_model, 16);
    565 	strncpy(lp->d_packname, "fictitious", 16);
    566 	lp->d_secperunit = d_link->sc_capacity;
    567 	lp->d_rpm = 3600;
    568 	lp->d_interleave = 1;
    569 	lp->d_flags = 0;
    570 
    571 	lp->d_partitions[RAW_PART].p_offset = 0;
    572 	lp->d_partitions[RAW_PART].p_size =
    573 	    lp->d_secperunit * (lp->d_secsize / DEV_BSIZE);
    574 	lp->d_partitions[RAW_PART].p_fstype = FS_UNUSED;
    575 	lp->d_npartitions = RAW_PART + 1;
    576 
    577 	lp->d_magic = DISKMAGIC;
    578 	lp->d_magic2 = DISKMAGIC;
    579 	lp->d_checksum = dkcksum(lp);
    580 }
    581 
    582 /*
    583  * Fabricate a default disk label, and try to read the correct one.
    584  */
    585 void
    586 wdgetdisklabel(wd)
    587 	struct wd_softc *wd;
    588 {
    589 	struct disklabel *lp = wd->sc_dk.dk_label;
    590 	struct wd_link *d_link = wd->d_link;
    591 	char *errstring;
    592 
    593 	WDDEBUG_PRINT(("wdgetdisklabel\n"));
    594 
    595 	bzero(wd->sc_dk.dk_cpulabel, sizeof(struct cpu_disklabel));
    596 
    597 	wdgetdefaultlabel(wd, lp);
    598 
    599 	d_link->sc_badsect[0] = -1;
    600 
    601 	if (d_link->sc_state > RECAL)
    602 		d_link->sc_state = RECAL;
    603 	errstring = readdisklabel(MAKEWDDEV(0, wd->sc_dev.dv_unit, RAW_PART),
    604 	    wdstrategy, lp, wd->sc_dk.dk_cpulabel);
    605 	if (errstring) {
    606 		/*
    607 		 * This probably happened because the drive's default
    608 		 * geometry doesn't match the DOS geometry.  We
    609 		 * assume the DOS geometry is now in the label and try
    610 		 * again.  XXX This is a kluge.
    611 		 */
    612 		if (d_link->sc_state > GEOMETRY)
    613 			d_link->sc_state = GEOMETRY;
    614 		errstring = readdisklabel(MAKEWDDEV(0, wd->sc_dev.dv_unit, RAW_PART),
    615 		    wdstrategy, lp, wd->sc_dk.dk_cpulabel);
    616 	}
    617 	if (errstring) {
    618 		printf("%s: %s\n", wd->sc_dev.dv_xname, errstring);
    619 		return;
    620 	}
    621 
    622 	if (d_link->sc_state > GEOMETRY)
    623 		d_link->sc_state = GEOMETRY;
    624 #ifdef HAS_BAD144_HANDLING
    625 	if ((lp->d_flags & D_BADSECT) != 0)
    626 		bad144intern(wd);
    627 #endif
    628 }
    629 
    630 
    631 /*
    632  * Tell the drive what geometry to use.
    633  */
    634 int
    635 wdsetctlr(d_link)
    636 	struct wd_link *d_link;
    637 {
    638 	struct wd_softc *wd=(struct wd_softc *)d_link->wd_softc;
    639 
    640 	WDDEBUG_PRINT(("wd(%d,%d) C%dH%dS%d\n", wd->sc_dev.dv_unit,
    641 	    d_link->drive, wd->sc_dk.dk_label->d_ncylinders,
    642 	    wd->sc_dk.dk_label->d_ntracks, wd->sc_dk.dk_label->d_nsectors));
    643 
    644 	if (wdccommand((struct wdc_softc *)d_link->wdc_softc,
    645 		d_link, WDCC_IDP, d_link->drive,
    646 	    wd->sc_dk.dk_label->d_ncylinders,
    647 	    wd->sc_dk.dk_label->d_ntracks - 1, 0,
    648 	    wd->sc_dk.dk_label->d_nsectors) != 0) {
    649 		wderror(d_link, NULL, "wdsetctlr: geometry upload failed");
    650 		return -1;
    651 	}
    652 
    653 	return 0;
    654 }
    655 
    656 int
    657 wdioctl(dev, xfer, addr, flag, p)
    658 	dev_t dev;
    659 	u_long xfer;
    660 	caddr_t addr;
    661 	int flag;
    662 	struct proc *p;
    663 {
    664 	struct wd_softc *wd = wd_cd.cd_devs[WDUNIT(dev)];
    665 	struct wd_link *d_link = wd->d_link;
    666 	int error;
    667 
    668 	WDDEBUG_PRINT(("wdioctl\n"));
    669 
    670 	if ((d_link->sc_flags & WDF_LOADED) == 0)
    671 		return EIO;
    672 
    673 	switch (xfer) {
    674 #ifdef HAS_BAD144_HANDLING
    675 	case DIOCSBAD:
    676 		if ((flag & FWRITE) == 0)
    677 			return EBADF;
    678 		wd->sc_dk.dk_cpulabel->bad = *(struct dkbad *)addr;
    679 		wd->sc_dk.dk_label->d_flags |= D_BADSECT;
    680 		bad144intern(wd);
    681 		return 0;
    682 #endif
    683 
    684 	case DIOCGDINFO:
    685 		*(struct disklabel *)addr = *(wd->sc_dk.dk_label);
    686 		return 0;
    687 
    688 	case DIOCGPART:
    689 		((struct partinfo *)addr)->disklab = wd->sc_dk.dk_label;
    690 		((struct partinfo *)addr)->part =
    691 		    &wd->sc_dk.dk_label->d_partitions[WDPART(dev)];
    692 		return 0;
    693 
    694 	case DIOCWDINFO:
    695 	case DIOCSDINFO:
    696 		if ((flag & FWRITE) == 0)
    697 			return EBADF;
    698 
    699 		if ((error = wdlock(wd->d_link)) != 0)
    700 			return error;
    701 		d_link->sc_flags |= WDF_LABELLING;
    702 
    703 		error = setdisklabel(wd->sc_dk.dk_label,
    704 		    (struct disklabel *)addr, /*wd->sc_dk.dk_openmask : */0,
    705 		    wd->sc_dk.dk_cpulabel);
    706 		if (error == 0) {
    707 			if (d_link->sc_state > GEOMETRY)
    708 				d_link->sc_state = GEOMETRY;
    709 			if (xfer == DIOCWDINFO)
    710 				error = writedisklabel(WDLABELDEV(dev),
    711 				    wdstrategy, wd->sc_dk.dk_label,
    712 				    wd->sc_dk.dk_cpulabel);
    713 		}
    714 
    715 		d_link->sc_flags &= ~WDF_LABELLING;
    716 		wdunlock(d_link);
    717 		return error;
    718 
    719 	case DIOCWLABEL:
    720 		if ((flag & FWRITE) == 0)
    721 			return EBADF;
    722 		if (*(int *)addr)
    723 			d_link->sc_flags |= WDF_WLABEL;
    724 		else
    725 			d_link->sc_flags &= ~WDF_WLABEL;
    726 		return 0;
    727 
    728 	case DIOCGDEFLABEL:
    729 		wdgetdefaultlabel(wd, (struct disklabel *)addr);
    730 		return 0;
    731 
    732 #ifdef notyet
    733 	case DIOCWFORMAT:
    734 		if ((flag & FWRITE) == 0)
    735 			return EBADF;
    736 	{
    737 		register struct format_op *fop;
    738 		struct iovec aiov;
    739 		struct uio auio;
    740 
    741 		fop = (struct format_op *)addr;
    742 		aiov.iov_base = fop->df_buf;
    743 		aiov.iov_len = fop->df_count;
    744 		auio.uio_iov = &aiov;
    745 		auio.uio_iovcnt = 1;
    746 		auio.uio_resid = fop->df_count;
    747 		auio.uio_segflg = 0;
    748 		auio.uio_offset =
    749 		    fop->df_startblk * wd->sc_dk.dk_label->d_secsize;
    750 		auio.uio_procp = p;
    751 		error = physio(wdformat, NULL, dev, B_WRITE, minphys,
    752 		    &auio);
    753 		fop->df_count -= auio.uio_resid;
    754 		fop->df_reg[0] = wdc->sc_status;
    755 		fop->df_reg[1] = wdc->sc_error;
    756 		return error;
    757 	}
    758 #endif
    759 
    760 	default:
    761 		return ENOTTY;
    762 	}
    763 
    764 #ifdef DIAGNOSTIC
    765 	panic("wdioctl: impossible");
    766 #endif
    767 }
    768 
    769 #ifdef B_FORMAT
    770 int
    771 wdformat(struct buf *bp)
    772 {
    773 
    774 	bp->b_flags |= B_FORMAT;
    775 	return wdstrategy(bp);
    776 }
    777 #endif
    778 
    779 int
    780 wdsize(dev)
    781 	dev_t dev;
    782 {
    783 	struct wd_softc *wd;
    784 	int part, unit, omask;
    785 	int size;
    786 
    787 	WDDEBUG_PRINT(("wdsize\n"));
    788 
    789 	unit = WDUNIT(dev);
    790 	if (unit >= wd_cd.cd_ndevs)
    791 		return (-1);
    792 	wd = wd_cd.cd_devs[unit];
    793 	if (wd == NULL)
    794 		return (-1);
    795 
    796 	part = WDPART(dev);
    797 	omask = wd->sc_dk.dk_openmask & (1 << part);
    798 
    799 	if (omask == 0 && wdopen(dev, 0, S_IFBLK, NULL) != 0)
    800 		return (-1);
    801 	if (wd->sc_dk.dk_label->d_partitions[part].p_fstype != FS_SWAP)
    802 		size = -1;
    803 	else
    804 		size = wd->sc_dk.dk_label->d_partitions[part].p_size *
    805 			(wd->sc_dk.dk_label->d_secsize / DEV_BSIZE);
    806 	if (omask == 0 && wdclose(dev, 0, S_IFBLK, NULL) != 0)
    807 		return (-1);
    808 	return (size);
    809 }
    810 
    811 
    812 #ifndef __BDEVSW_DUMP_OLD_TYPE
    813 /* #define WD_DUMP_NOT_TRUSTED if you just want to watch */
    814 static int wddoingadump;
    815 static int wddumprecalibrated;
    816 
    817 /*
    818  * Dump core after a system crash.
    819  *
    820  * XXX:  This needs work!  Currently, it's a major hack: the
    821  * use of wdc_softc is very bad and should go away.
    822  */
    823 int
    824 wddump(dev, blkno, va, size)
    825         dev_t dev;
    826         daddr_t blkno;
    827         caddr_t va;
    828         size_t size;
    829 {
    830 	struct wd_softc *wd;	/* disk unit to do the I/O */
    831 	struct wdc_softc *wdc;	/* disk controller to do the I/O */
    832 	struct disklabel *lp;	/* disk's disklabel */
    833 	struct wd_link *d_link;
    834 	int	unit, part;
    835 	int	nblks;		/* total number of sectors left to write */
    836 
    837 	/* Check if recursive dump; if so, punt. */
    838 	if (wddoingadump)
    839 		return EFAULT;
    840 	wddoingadump = 1;
    841 
    842 	unit = WDUNIT(dev);
    843 	if (unit >= wd_cd.cd_ndevs)
    844 		return ENXIO;
    845 	wd = wd_cd.cd_devs[unit];
    846 	if (wd == (struct wd_softc *)0)
    847 		return ENXIO;
    848 	d_link = wd->d_link;
    849 
    850 	part = WDPART(dev);
    851 
    852 	/* Make sure it was initialized. */
    853 	if (d_link->sc_state < READY)
    854 		return ENXIO;
    855 
    856 	wdc = (void *)wd->sc_dev.dv_parent;
    857 
    858         /* Convert to disk sectors.  Request must be a multiple of size. */
    859 	lp = wd->sc_dk.dk_label;
    860 	if ((size % lp->d_secsize) != 0)
    861 		return EFAULT;
    862 	nblks = size / lp->d_secsize;
    863 	blkno = blkno / (lp->d_secsize / DEV_BSIZE);
    864 
    865 	/* Check transfer bounds against partition size. */
    866 	if ((blkno < 0) || ((blkno + nblks) > lp->d_partitions[part].p_size))
    867 		return EINVAL;
    868 
    869 	/* Offset block number to start of partition. */
    870 	blkno += lp->d_partitions[part].p_offset;
    871 
    872 	/* Recalibrate, if first dump transfer. */
    873 	if (wddumprecalibrated == 0) {
    874 		wddumprecalibrated = 1;
    875 		if (wdccommandshort(wdc, d_link->drive, WDCC_RECAL) != 0 ||
    876 		    wait_for_ready(wdc) != 0 || wdsetctlr(d_link) != 0 ||
    877 		    wait_for_ready(wdc) != 0) {
    878 			wderror(d_link, NULL, "wddump: recal failed");
    879 			return EIO;
    880 		}
    881 	}
    882 
    883 	while (nblks > 0) {
    884 		daddr_t xlt_blkno = blkno;
    885 		long cylin, head, sector;
    886 
    887 		if ((lp->d_flags & D_BADSECT) != 0) {
    888 			long blkdiff;
    889 			int i;
    890 
    891 			for (i = 0; (blkdiff = d_link->sc_badsect[i]) != -1; i++) {
    892 				blkdiff -= xlt_blkno;
    893 				if (blkdiff < 0)
    894 					continue;
    895 				if (blkdiff == 0) {
    896 					/* Replace current block of transfer. */
    897 					xlt_blkno = lp->d_secperunit -
    898 					    lp->d_nsectors - i - 1;
    899 				}
    900 				break;
    901 			}
    902 			/* Tranfer is okay now. */
    903 		}
    904 
    905 		if ((d_link->sc_flags & WDF_LBA) != 0) {
    906 			sector = (xlt_blkno >> 0) & 0xff;
    907 			cylin = (xlt_blkno >> 8) & 0xffff;
    908 			head = (xlt_blkno >> 24) & 0xf;
    909 			head |= WDSD_LBA;
    910 		} else {
    911 			sector = xlt_blkno % lp->d_nsectors;
    912 			sector++;	/* Sectors begin with 1, not 0. */
    913 			xlt_blkno /= lp->d_nsectors;
    914 			head = xlt_blkno % lp->d_ntracks;
    915 			xlt_blkno /= lp->d_ntracks;
    916 			cylin = xlt_blkno;
    917 			head |= WDSD_CHS;
    918 		}
    919 
    920 #ifndef WD_DUMP_NOT_TRUSTED
    921 		if (wdccommand((struct wdc_softc *)d_link->wdc_softc, d_link,
    922 			WDCC_WRITE, d_link->drive, cylin, head, sector, 1) != 0 ||
    923 		    wait_for_drq(wdc) != 0) {
    924 			wderror(d_link, NULL, "wddump: write failed");
    925 			return EIO;
    926 		}
    927 
    928 		/* XXX XXX XXX */
    929 		bus_space_write_multi_2(wdc->sc_iot, wdc->sc_ioh, wd_data,
    930 		    (u_int16_t *)va, lp->d_secsize >> 1);
    931 
    932 		/* Check data request (should be done). */
    933 		if (wait_for_ready(wdc) != 0) {
    934 			wderror(d_link, NULL,
    935 			    "wddump: timeout waiting for ready");
    936 			return EIO;
    937 		}
    938 #else	/* WD_DUMP_NOT_TRUSTED */
    939 		/* Let's just talk about this first... */
    940 		printf("wd%d: dump addr 0x%x, cylin %d, head %d, sector %d\n",
    941 		    unit, va, cylin, head, sector);
    942 		delay(500 * 1000);	/* half a second */
    943 #endif
    944 
    945 		/* update block count */
    946 		nblks -= 1;
    947 		blkno += 1;
    948 		va += lp->d_secsize;
    949 	}
    950 
    951 	wddoingadump = 0;
    952 	return 0;
    953 }
    954 #else /* __BDEVSW_DUMP_NEW_TYPE */
    955 
    956 
    957 int
    958 wddump(dev, blkno, va, size)
    959         dev_t dev;
    960         daddr_t blkno;
    961         caddr_t va;
    962         size_t size;
    963 {
    964 
    965 	/* Not implemented. */
    966 	return ENXIO;
    967 }
    968 #endif /* __BDEVSW_DUMP_NEW_TYPE */
    969 
    970 #ifdef HAS_BAD144_HANDLING
    971 /*
    972  * Internalize the bad sector table.
    973  */
    974 void
    975 bad144intern(wd)
    976 	struct wd_softc *wd;
    977 {
    978 	struct dkbad *bt = &wd->sc_dk.dk_cpulabel->bad;
    979 	struct disklabel *lp = wd->sc_dk.dk_label;
    980 	struct wd_link *d_link = wd->d_link;
    981 	int i = 0;
    982 
    983 	WDDEBUG_PRINT(("bad144intern\n"));
    984 
    985 	for (; i < NBT_BAD; i++) {
    986 		if (bt->bt_bad[i].bt_cyl == 0xffff)
    987 			break;
    988 		d_link->sc_badsect[i] =
    989 		    bt->bt_bad[i].bt_cyl * lp->d_secpercyl +
    990 		    (bt->bt_bad[i].bt_trksec >> 8) * lp->d_nsectors +
    991 		    (bt->bt_bad[i].bt_trksec & 0xff);
    992 	}
    993 	for (; i < NBT_BAD+1; i++)
    994 		d_link->sc_badsect[i] = -1;
    995 }
    996 #endif
    997 
    998 void
    999 wderror(d_link, bp, msg)
   1000 	struct wd_link *d_link;
   1001 	struct buf *bp;
   1002 	char *msg;
   1003 {
   1004 	struct wd_softc *wd = (struct wd_softc *)d_link->wd_softc;
   1005 
   1006 	if (bp) {
   1007 		diskerr(bp, "wd", msg, LOG_PRINTF, bp->b_bcount,
   1008 		    wd->sc_dk.dk_label);
   1009 		printf("\n");
   1010 	} else
   1011 		printf("%s: %s\n", wd->sc_dev.dv_xname, msg);
   1012 }
   1013 
   1014 void
   1015 wddone(d_link, bp)
   1016 	struct wd_link *d_link;
   1017 	struct buf *bp;
   1018 {
   1019 	struct wd_softc *wd = (void *)d_link->wd_softc;
   1020 
   1021 	disk_unbusy(&wd->sc_dk, (bp->b_bcount - bp->b_resid));
   1022 #if NRND > 0
   1023 	rnd_add_uint32(&wd->rnd_source, bp->b_blkno);
   1024 #endif
   1025 }
   1026