wd.c revision 1.284 1 /* $NetBSD: wd.c,v 1.284 2004/08/03 22:37:19 bouyer Exp $ */
2
3 /*
4 * Copyright (c) 1998, 2001 Manuel Bouyer. All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 * 3. All advertising materials mentioning features or use of this software
15 * must display the following acknowledgement:
16 * This product includes software developed by Manuel Bouyer.
17 * 4. The name of the author may not be used to endorse or promote products
18 * derived from this software without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
21 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
22 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
23 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
24 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
25 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
29 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30 */
31
32 /*-
33 * Copyright (c) 1998, 2003, 2004 The NetBSD Foundation, Inc.
34 * All rights reserved.
35 *
36 * This code is derived from software contributed to The NetBSD Foundation
37 * by Charles M. Hannum and by Onno van der Linden.
38 *
39 * Redistribution and use in source and binary forms, with or without
40 * modification, are permitted provided that the following conditions
41 * are met:
42 * 1. Redistributions of source code must retain the above copyright
43 * notice, this list of conditions and the following disclaimer.
44 * 2. Redistributions in binary form must reproduce the above copyright
45 * notice, this list of conditions and the following disclaimer in the
46 * documentation and/or other materials provided with the distribution.
47 * 3. All advertising materials mentioning features or use of this software
48 * must display the following acknowledgement:
49 * This product includes software developed by the NetBSD
50 * Foundation, Inc. and its contributors.
51 * 4. Neither the name of The NetBSD Foundation nor the names of its
52 * contributors may be used to endorse or promote products derived
53 * from this software without specific prior written permission.
54 *
55 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
56 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
57 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
58 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
59 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
60 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
61 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
62 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
63 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
64 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
65 * POSSIBILITY OF SUCH DAMAGE.
66 */
67
68 #include <sys/cdefs.h>
69 __KERNEL_RCSID(0, "$NetBSD: wd.c,v 1.284 2004/08/03 22:37:19 bouyer Exp $");
70
71 #ifndef WDCDEBUG
72 #define WDCDEBUG
73 #endif /* WDCDEBUG */
74
75 #include "rnd.h"
76
77 #include <sys/param.h>
78 #include <sys/systm.h>
79 #include <sys/kernel.h>
80 #include <sys/conf.h>
81 #include <sys/file.h>
82 #include <sys/stat.h>
83 #include <sys/ioctl.h>
84 #include <sys/buf.h>
85 #include <sys/uio.h>
86 #include <sys/malloc.h>
87 #include <sys/device.h>
88 #include <sys/disklabel.h>
89 #include <sys/disk.h>
90 #include <sys/syslog.h>
91 #include <sys/proc.h>
92 #include <sys/vnode.h>
93 #if NRND > 0
94 #include <sys/rnd.h>
95 #endif
96
97 #include <machine/intr.h>
98 #include <machine/bus.h>
99
100 #include <dev/ata/atareg.h>
101 #include <dev/ata/atavar.h>
102 #include <dev/ata/wdvar.h>
103 #include <dev/ic/wdcreg.h>
104 #include <sys/ataio.h>
105 #include "locators.h"
106
107 #define LBA48_THRESHOLD (0xfffffff) /* 128GB / DEV_BSIZE */
108
109 #define WDIORETRIES_SINGLE 4 /* number of retries before single-sector */
110 #define WDIORETRIES 5 /* number of retries before giving up */
111 #define RECOVERYTIME hz/2 /* time to wait before retrying a cmd */
112
113 #define WDUNIT(dev) DISKUNIT(dev)
114 #define WDPART(dev) DISKPART(dev)
115 #define WDMINOR(unit, part) DISKMINOR(unit, part)
116 #define MAKEWDDEV(maj, unit, part) MAKEDISKDEV(maj, unit, part)
117
118 #define WDLABELDEV(dev) (MAKEWDDEV(major(dev), WDUNIT(dev), RAW_PART))
119
120 #define DEBUG_INTR 0x01
121 #define DEBUG_XFERS 0x02
122 #define DEBUG_STATUS 0x04
123 #define DEBUG_FUNCS 0x08
124 #define DEBUG_PROBE 0x10
125 #ifdef WDCDEBUG
126 int wdcdebug_wd_mask = 0x0;
127 #define WDCDEBUG_PRINT(args, level) \
128 if (wdcdebug_wd_mask & (level)) \
129 printf args
130 #else
131 #define WDCDEBUG_PRINT(args, level)
132 #endif
133
134 int wdprobe(struct device *, struct cfdata *, void *);
135 void wdattach(struct device *, struct device *, void *);
136 int wddetach(struct device *, int);
137 int wdactivate(struct device *, enum devact);
138 int wdprint(void *, char *);
139 void wdperror(const struct wd_softc *);
140
141 CFATTACH_DECL(wd, sizeof(struct wd_softc),
142 wdprobe, wdattach, wddetach, wdactivate);
143
144 extern struct cfdriver wd_cd;
145
146 dev_type_open(wdopen);
147 dev_type_close(wdclose);
148 dev_type_read(wdread);
149 dev_type_write(wdwrite);
150 dev_type_ioctl(wdioctl);
151 dev_type_strategy(wdstrategy);
152 dev_type_dump(wddump);
153 dev_type_size(wdsize);
154
155 const struct bdevsw wd_bdevsw = {
156 wdopen, wdclose, wdstrategy, wdioctl, wddump, wdsize, D_DISK
157 };
158
159 const struct cdevsw wd_cdevsw = {
160 wdopen, wdclose, wdread, wdwrite, wdioctl,
161 nostop, notty, nopoll, nommap, nokqfilter, D_DISK
162 };
163
164 /*
165 * Glue necessary to hook WDCIOCCOMMAND into physio
166 */
167
168 struct wd_ioctl {
169 LIST_ENTRY(wd_ioctl) wi_list;
170 struct buf wi_bp;
171 struct uio wi_uio;
172 struct iovec wi_iov;
173 atareq_t wi_atareq;
174 struct wd_softc *wi_softc;
175 };
176
177 LIST_HEAD(, wd_ioctl) wi_head;
178
179 struct wd_ioctl *wi_find(struct buf *);
180 void wi_free(struct wd_ioctl *);
181 struct wd_ioctl *wi_get(void);
182 void wdioctlstrategy(struct buf *);
183
184 void wdgetdefaultlabel(struct wd_softc *, struct disklabel *);
185 void wdgetdisklabel(struct wd_softc *);
186 void wdstart(void *);
187 void __wdstart(struct wd_softc*, struct buf *);
188 void wdrestart(void *);
189 void wddone(void *);
190 int wd_get_params(struct wd_softc *, u_int8_t, struct ataparams *);
191 int wd_flushcache(struct wd_softc *, int);
192 void wd_shutdown(void *);
193
194 int wd_getcache(struct wd_softc *, int *);
195 int wd_setcache(struct wd_softc *, int);
196
197 struct dkdriver wddkdriver = { wdstrategy };
198
199 #ifdef HAS_BAD144_HANDLING
200 static void bad144intern(struct wd_softc *);
201 #endif
202
203 #define WD_QUIRK_SPLIT_MOD15_WRITE 0x0001 /* must split certain writes */
204
205 /*
206 * Quirk table for IDE drives. Put more-specific matches first, since
207 * a simple globbing routine is used for matching.
208 */
209 static const struct wd_quirk {
210 const char *wdq_match; /* inquiry pattern to match */
211 int wdq_quirks; /* drive quirks */
212 } wd_quirk_table[] = {
213 /*
214 * Some Seagate S-ATA drives have a PHY which can get confused
215 * with the way data is packetized by some S-ATA controllers.
216 *
217 * The work-around is to split in two any write transfer whose
218 * sector count % 15 == 1 (assuming 512 byte sectors).
219 *
220 * XXX This is an incomplete list. There are at least a couple
221 * XXX more model numbers. If you have trouble with such transfers
222 * XXX (8K is the most common) on Seagate S-ATA drives, please
223 * XXX notify thorpej (at) NetBSD.org.
224 */
225 { "ST3120023AS",
226 WD_QUIRK_SPLIT_MOD15_WRITE },
227 { "ST380023AS",
228 WD_QUIRK_SPLIT_MOD15_WRITE },
229
230 { NULL,
231 0 }
232 };
233
234 static const struct wd_quirk *
235 wd_lookup_quirks(const char *name)
236 {
237 const struct wd_quirk *wdq;
238 const char *estr;
239
240 for (wdq = wd_quirk_table; wdq->wdq_match != NULL; wdq++) {
241 /*
242 * We only want exact matches (which include matches
243 * against globbing characters).
244 */
245 if (pmatch(name, wdq->wdq_match, &estr) == 2)
246 return (wdq);
247 }
248 return (NULL);
249 }
250
251 int
252 wdprobe(struct device *parent, struct cfdata *match, void *aux)
253 {
254 struct ata_device *adev = aux;
255
256 if (adev == NULL)
257 return 0;
258 if (adev->adev_bustype->bustype_type != SCSIPI_BUSTYPE_ATA)
259 return 0;
260
261 if (match->cf_loc[ATA_HLCF_DRIVE] != ATA_HLCF_DRIVE_DEFAULT &&
262 match->cf_loc[ATA_HLCF_DRIVE] != adev->adev_drv_data->drive)
263 return 0;
264 return 1;
265 }
266
267 void
268 wdattach(struct device *parent, struct device *self, void *aux)
269 {
270 struct wd_softc *wd = (void *)self;
271 struct ata_device *adev= aux;
272 int i, blank;
273 char buf[41], pbuf[9], c, *p, *q;
274 const struct wd_quirk *wdq;
275 WDCDEBUG_PRINT(("wdattach\n"), DEBUG_FUNCS | DEBUG_PROBE);
276
277 lockinit(&wd->sc_lock, PRIBIO | PCATCH, "wdlock", 0, 0);
278
279 callout_init(&wd->sc_restart_ch);
280 bufq_alloc(&wd->sc_q, BUFQ_DISK_DEFAULT_STRAT()|BUFQ_SORT_RAWBLOCK);
281 SLIST_INIT(&wd->sc_bslist);
282
283 wd->atabus = adev->adev_bustype;
284 wd->openings = adev->adev_openings;
285 wd->drvp = adev->adev_drv_data;
286
287 wd->drvp->drv_done = wddone;
288 wd->drvp->drv_softc = &wd->sc_dev;
289
290 aprint_naive("\n");
291
292 /* read our drive info */
293 if (wd_get_params(wd, AT_WAIT, &wd->sc_params) != 0) {
294 aprint_error("\n%s: IDENTIFY failed\n", wd->sc_dev.dv_xname);
295 return;
296 }
297
298 for (blank = 0, p = wd->sc_params.atap_model, q = buf, i = 0;
299 i < sizeof(wd->sc_params.atap_model); i++) {
300 c = *p++;
301 if (c == '\0')
302 break;
303 if (c != ' ') {
304 if (blank) {
305 *q++ = ' ';
306 blank = 0;
307 }
308 *q++ = c;
309 } else
310 blank = 1;
311 }
312 *q++ = '\0';
313
314 aprint_normal(": <%s>\n", buf);
315
316 wdq = wd_lookup_quirks(buf);
317 if (wdq != NULL)
318 wd->sc_quirks = wdq->wdq_quirks;
319
320 if ((wd->sc_params.atap_multi & 0xff) > 1) {
321 wd->sc_multi = wd->sc_params.atap_multi & 0xff;
322 } else {
323 wd->sc_multi = 1;
324 }
325
326 aprint_normal("%s: drive supports %d-sector PIO transfers,",
327 wd->sc_dev.dv_xname, wd->sc_multi);
328
329 /* 48-bit LBA addressing */
330 if ((wd->sc_params.atap_cmd2_en & ATA_CMD2_LBA48) != 0)
331 wd->sc_flags |= WDF_LBA48;
332
333 /* Prior to ATA-4, LBA was optional. */
334 if ((wd->sc_params.atap_capabilities1 & WDC_CAP_LBA) != 0)
335 wd->sc_flags |= WDF_LBA;
336 #if 0
337 /* ATA-4 requires LBA. */
338 if (wd->sc_params.atap_ataversion != 0xffff &&
339 wd->sc_params.atap_ataversion >= WDC_VER_ATA4)
340 wd->sc_flags |= WDF_LBA;
341 #endif
342
343 if ((wd->sc_flags & WDF_LBA48) != 0) {
344 aprint_normal(" LBA48 addressing\n");
345 wd->sc_capacity =
346 ((u_int64_t) wd->sc_params.__reserved6[11] << 48) |
347 ((u_int64_t) wd->sc_params.__reserved6[10] << 32) |
348 ((u_int64_t) wd->sc_params.__reserved6[9] << 16) |
349 ((u_int64_t) wd->sc_params.__reserved6[8] << 0);
350 } else if ((wd->sc_flags & WDF_LBA) != 0) {
351 aprint_normal(" LBA addressing\n");
352 wd->sc_capacity =
353 (wd->sc_params.atap_capacity[1] << 16) |
354 wd->sc_params.atap_capacity[0];
355 } else {
356 aprint_normal(" chs addressing\n");
357 wd->sc_capacity =
358 wd->sc_params.atap_cylinders *
359 wd->sc_params.atap_heads *
360 wd->sc_params.atap_sectors;
361 }
362 format_bytes(pbuf, sizeof(pbuf), wd->sc_capacity * DEV_BSIZE);
363 aprint_normal("%s: %s, %d cyl, %d head, %d sec, "
364 "%d bytes/sect x %llu sectors\n",
365 self->dv_xname, pbuf,
366 (wd->sc_flags & WDF_LBA) ? (int)(wd->sc_capacity /
367 (wd->sc_params.atap_heads * wd->sc_params.atap_sectors)) :
368 wd->sc_params.atap_cylinders,
369 wd->sc_params.atap_heads, wd->sc_params.atap_sectors,
370 DEV_BSIZE, (unsigned long long)wd->sc_capacity);
371
372 WDCDEBUG_PRINT(("%s: atap_dmatiming_mimi=%d, atap_dmatiming_recom=%d\n",
373 self->dv_xname, wd->sc_params.atap_dmatiming_mimi,
374 wd->sc_params.atap_dmatiming_recom), DEBUG_PROBE);
375 /*
376 * Initialize and attach the disk structure.
377 */
378 wd->sc_dk.dk_driver = &wddkdriver;
379 wd->sc_dk.dk_name = wd->sc_dev.dv_xname;
380 disk_attach(&wd->sc_dk);
381 wd->sc_wdc_bio.lp = wd->sc_dk.dk_label;
382 wd->sc_sdhook = shutdownhook_establish(wd_shutdown, wd);
383 if (wd->sc_sdhook == NULL)
384 aprint_error("%s: WARNING: unable to establish shutdown hook\n",
385 wd->sc_dev.dv_xname);
386 #if NRND > 0
387 rnd_attach_source(&wd->rnd_source, wd->sc_dev.dv_xname,
388 RND_TYPE_DISK, 0);
389 #endif
390 }
391
392 int
393 wdactivate(struct device *self, enum devact act)
394 {
395 int rv = 0;
396
397 switch (act) {
398 case DVACT_ACTIVATE:
399 rv = EOPNOTSUPP;
400 break;
401
402 case DVACT_DEACTIVATE:
403 /*
404 * Nothing to do; we key off the device's DVF_ACTIVATE.
405 */
406 break;
407 }
408 return (rv);
409 }
410
411 int
412 wddetach(struct device *self, int flags)
413 {
414 struct wd_softc *sc = (struct wd_softc *)self;
415 struct buf *bp;
416 int s, bmaj, cmaj, i, mn;
417
418 /* locate the major number */
419 bmaj = bdevsw_lookup_major(&wd_bdevsw);
420 cmaj = cdevsw_lookup_major(&wd_cdevsw);
421
422 s = splbio();
423
424 /* Kill off any queued buffers. */
425 while ((bp = BUFQ_GET(&sc->sc_q)) != NULL) {
426 bp->b_error = EIO;
427 bp->b_flags |= B_ERROR;
428 bp->b_resid = bp->b_bcount;
429 biodone(bp);
430 }
431
432 bufq_free(&sc->sc_q);
433
434 splx(s);
435
436 /* Nuke the vnodes for any open instances. */
437 for (i = 0; i < MAXPARTITIONS; i++) {
438 mn = WDMINOR(self->dv_unit, i);
439 vdevgone(bmaj, mn, mn, VBLK);
440 vdevgone(cmaj, mn, mn, VCHR);
441 }
442
443 /* Detach disk. */
444 disk_detach(&sc->sc_dk);
445
446 /* Clean out the bad sector list */
447 while (!SLIST_EMPTY(&sc->sc_bslist)) {
448 void *head = SLIST_FIRST(&sc->sc_bslist);
449 SLIST_REMOVE_HEAD(&sc->sc_bslist, dbs_next);
450 free(head, M_TEMP);
451 }
452 sc->sc_bscount = 0;
453
454 /* Get rid of the shutdown hook. */
455 if (sc->sc_sdhook != NULL)
456 shutdownhook_disestablish(sc->sc_sdhook);
457
458 #if NRND > 0
459 /* Unhook the entropy source. */
460 rnd_detach_source(&sc->rnd_source);
461 #endif
462
463 lockmgr(&sc->sc_lock, LK_DRAIN, NULL);
464 sc->drvp->drive_flags = 0; /* no drive any more here */
465
466 return (0);
467 }
468
469 /*
470 * Read/write routine for a buffer. Validates the arguments and schedules the
471 * transfer. Does not wait for the transfer to complete.
472 */
473 void
474 wdstrategy(struct buf *bp)
475 {
476 struct wd_softc *wd = device_lookup(&wd_cd, WDUNIT(bp->b_dev));
477 struct disklabel *lp = wd->sc_dk.dk_label;
478 daddr_t blkno;
479 int s;
480
481 WDCDEBUG_PRINT(("wdstrategy (%s)\n", wd->sc_dev.dv_xname),
482 DEBUG_XFERS);
483
484 /* Valid request? */
485 if (bp->b_blkno < 0 ||
486 (bp->b_bcount % lp->d_secsize) != 0 ||
487 (bp->b_bcount / lp->d_secsize) >= (1 << NBBY)) {
488 bp->b_error = EINVAL;
489 goto bad;
490 }
491
492 /* If device invalidated (e.g. media change, door open), error. */
493 if ((wd->sc_flags & WDF_LOADED) == 0) {
494 bp->b_error = EIO;
495 goto bad;
496 }
497
498 /* If it's a null transfer, return immediately. */
499 if (bp->b_bcount == 0)
500 goto done;
501
502 /*
503 * Do bounds checking, adjust transfer. if error, process.
504 * If end of partition, just return.
505 */
506 if (WDPART(bp->b_dev) == RAW_PART) {
507 if (bounds_check_with_mediasize(bp, DEV_BSIZE,
508 wd->sc_capacity) <= 0)
509 goto done;
510 } else {
511 if (bounds_check_with_label(&wd->sc_dk, bp,
512 (wd->sc_flags & (WDF_WLABEL|WDF_LABELLING)) != 0) <= 0)
513 goto done;
514 }
515
516 /*
517 * Now convert the block number to absolute and put it in
518 * terms of the device's logical block size.
519 */
520 if (lp->d_secsize >= DEV_BSIZE)
521 blkno = bp->b_blkno / (lp->d_secsize / DEV_BSIZE);
522 else
523 blkno = bp->b_blkno * (DEV_BSIZE / lp->d_secsize);
524
525 if (WDPART(bp->b_dev) != RAW_PART)
526 blkno += lp->d_partitions[WDPART(bp->b_dev)].p_offset;
527
528 bp->b_rawblkno = blkno;
529
530 /*
531 * If the transfer about to be attempted contains only a block that
532 * is known to be bad then return an error for the transfer without
533 * even attempting to start a transfer up under the premis that we
534 * will just end up doing more retries for a transfer that will end
535 * up failing again.
536 * XXX:SMP - mutex required to protect with DIOCBSFLUSH
537 */
538 if (__predict_false(!SLIST_EMPTY(&wd->sc_bslist))) {
539 struct disk_badsectors *dbs;
540 daddr_t maxblk = blkno + (bp->b_bcount >> DEV_BSHIFT) - 1;
541
542 SLIST_FOREACH(dbs, &wd->sc_bslist, dbs_next)
543 if ((dbs->dbs_min <= blkno && blkno <= dbs->dbs_max) ||
544 (dbs->dbs_min <= maxblk && maxblk <= dbs->dbs_max)){
545 bp->b_error = EIO;
546 goto bad;
547 }
548 }
549
550 /* Queue transfer on drive, activate drive and controller if idle. */
551 s = splbio();
552 BUFQ_PUT(&wd->sc_q, bp);
553 wdstart(wd);
554 splx(s);
555 return;
556 bad:
557 bp->b_flags |= B_ERROR;
558 done:
559 /* Toss transfer; we're done early. */
560 bp->b_resid = bp->b_bcount;
561 biodone(bp);
562 }
563
564 /*
565 * Queue a drive for I/O.
566 */
567 void
568 wdstart(void *arg)
569 {
570 struct wd_softc *wd = arg;
571 struct buf *bp = NULL;
572
573 WDCDEBUG_PRINT(("wdstart %s\n", wd->sc_dev.dv_xname),
574 DEBUG_XFERS);
575 while (wd->openings > 0) {
576
577 /* Is there a buf for us ? */
578 if ((bp = BUFQ_GET(&wd->sc_q)) == NULL)
579 return;
580
581 /*
582 * Make the command. First lock the device
583 */
584 wd->openings--;
585
586 wd->retries = 0;
587 __wdstart(wd, bp);
588 }
589 }
590
591 static void
592 wd_split_mod15_write(struct buf *bp)
593 {
594 struct buf *obp = bp->b_private;
595 struct wd_softc *sc = wd_cd.cd_devs[DISKUNIT(obp->b_dev)];
596
597 if (__predict_false(bp->b_flags & B_ERROR) != 0) {
598 /*
599 * Propagate the error. If this was the first half of
600 * the original transfer, make sure to account for that
601 * in the residual.
602 */
603 if (bp->b_data == obp->b_data)
604 bp->b_resid += bp->b_bcount;
605 goto done;
606 }
607
608 /*
609 * If this was the second half of the transfer, we're all done!
610 */
611 if (bp->b_data != obp->b_data)
612 goto done;
613
614 /*
615 * Advance the pointer to the second half and issue that command
616 * using the same opening.
617 */
618 bp->b_flags = obp->b_flags | B_CALL;
619 bp->b_data += bp->b_bcount;
620 bp->b_blkno += (bp->b_bcount / 512);
621 bp->b_rawblkno += (bp->b_bcount / 512);
622 __wdstart(sc, bp);
623 return;
624
625 done:
626 obp->b_flags |= (bp->b_flags & (B_EINTR|B_ERROR));
627 obp->b_error = bp->b_error;
628 obp->b_resid = bp->b_resid;
629 pool_put(&bufpool, bp);
630 biodone(obp);
631 sc->openings++;
632 /* wddone() will call wdstart() */
633 }
634
635 void
636 __wdstart(struct wd_softc *wd, struct buf *bp)
637 {
638
639 /*
640 * Deal with the "split mod15 write" quirk. We just divide the
641 * transfer in two, doing the first half and then then second half
642 * with the same command opening.
643 *
644 * Note we MUST do this here, because we can't let insertion
645 * into the bufq cause the transfers to be re-merged.
646 */
647 if (__predict_false((wd->sc_quirks & WD_QUIRK_SPLIT_MOD15_WRITE) != 0 &&
648 (bp->b_flags & B_READ) == 0 &&
649 bp->b_bcount > 512 &&
650 ((bp->b_bcount / 512) % 15) == 1)) {
651 struct buf *nbp;
652
653 /* already at splbio */
654 nbp = pool_get(&bufpool, PR_NOWAIT);
655 if (__predict_false(nbp == NULL)) {
656 /* No memory -- fail the iop. */
657 bp->b_error = ENOMEM;
658 bp->b_flags |= B_ERROR;
659 bp->b_resid = bp->b_bcount;
660 biodone(bp);
661 wd->openings++;
662 return;
663 }
664
665 BUF_INIT(nbp);
666 nbp->b_error = 0;
667 nbp->b_proc = bp->b_proc;
668 nbp->b_vp = NULLVP;
669 nbp->b_dev = bp->b_dev;
670
671 nbp->b_bcount = bp->b_bcount / 2;
672 nbp->b_bufsize = bp->b_bcount / 2;
673 nbp->b_data = bp->b_data;
674
675 nbp->b_blkno = bp->b_blkno;
676 nbp->b_rawblkno = bp->b_rawblkno;
677
678 nbp->b_flags = bp->b_flags | B_CALL;
679 nbp->b_iodone = wd_split_mod15_write;
680
681 /* Put ptr to orig buf in b_private and use new buf */
682 nbp->b_private = bp;
683
684 BIO_COPYPRIO(nbp, bp);
685
686 bp = nbp;
687 }
688
689 wd->sc_wdc_bio.blkno = bp->b_rawblkno;
690 wd->sc_wdc_bio.blkdone =0;
691 wd->sc_bp = bp;
692 /*
693 * If we're retrying, retry in single-sector mode. This will give us
694 * the sector number of the problem, and will eventually allow the
695 * transfer to succeed.
696 */
697 if (wd->retries >= WDIORETRIES_SINGLE)
698 wd->sc_wdc_bio.flags = ATA_SINGLE;
699 else
700 wd->sc_wdc_bio.flags = 0;
701 if (wd->sc_flags & WDF_LBA48 && wd->sc_wdc_bio.blkno > LBA48_THRESHOLD)
702 wd->sc_wdc_bio.flags |= ATA_LBA48;
703 if (wd->sc_flags & WDF_LBA)
704 wd->sc_wdc_bio.flags |= ATA_LBA;
705 if (bp->b_flags & B_READ)
706 wd->sc_wdc_bio.flags |= ATA_READ;
707 wd->sc_wdc_bio.bcount = bp->b_bcount;
708 wd->sc_wdc_bio.databuf = bp->b_data;
709 /* Instrumentation. */
710 disk_busy(&wd->sc_dk);
711 switch (wd->atabus->ata_bio(wd->drvp, &wd->sc_wdc_bio)) {
712 case WDC_TRY_AGAIN:
713 callout_reset(&wd->sc_restart_ch, hz, wdrestart, wd);
714 break;
715 case WDC_QUEUED:
716 case WDC_COMPLETE:
717 break;
718 default:
719 panic("__wdstart: bad return code from ata_bio()");
720 }
721 }
722
723 void
724 wddone(void *v)
725 {
726 struct wd_softc *wd = v;
727 struct buf *bp = wd->sc_bp;
728 const char *errmsg;
729 int do_perror = 0;
730 WDCDEBUG_PRINT(("wddone %s\n", wd->sc_dev.dv_xname),
731 DEBUG_XFERS);
732
733 if (bp == NULL)
734 return;
735 bp->b_resid = wd->sc_wdc_bio.bcount;
736 switch (wd->sc_wdc_bio.error) {
737 case ERR_DMA:
738 errmsg = "DMA error";
739 goto retry;
740 case ERR_DF:
741 errmsg = "device fault";
742 goto retry;
743 case TIMEOUT:
744 errmsg = "device timeout";
745 goto retry;
746 case ERR_RESET:
747 errmsg = "channel reset";
748 goto retry2;
749 case ERROR:
750 /* Don't care about media change bits */
751 if (wd->sc_wdc_bio.r_error != 0 &&
752 (wd->sc_wdc_bio.r_error & ~(WDCE_MC | WDCE_MCR)) == 0)
753 goto noerror;
754 errmsg = "error";
755 do_perror = 1;
756 retry: /* Just reset and retry. Can we do more ? */
757 wd->atabus->ata_reset_channel(wd->drvp, 0);
758 retry2:
759 diskerr(bp, "wd", errmsg, LOG_PRINTF,
760 wd->sc_wdc_bio.blkdone, wd->sc_dk.dk_label);
761 if (wd->retries < WDIORETRIES)
762 printf(", retrying\n");
763 if (do_perror)
764 wdperror(wd);
765 if (wd->retries < WDIORETRIES) {
766 wd->retries++;
767 callout_reset(&wd->sc_restart_ch, RECOVERYTIME,
768 wdrestart, wd);
769 return;
770 }
771 printf("\n");
772
773 /*
774 * Not all errors indicate a failed block but those that do,
775 * put the block on the bad-block list for the device. Only
776 * do this for reads because the drive should do it for writes,
777 * itself, according to Manuel.
778 */
779 if ((bp->b_flags & B_READ) &&
780 ((wd->drvp->ata_vers >= 4 && wd->sc_wdc_bio.r_error & 64) ||
781 (wd->drvp->ata_vers < 4 && wd->sc_wdc_bio.r_error & 192))) {
782 struct disk_badsectors *dbs;
783
784 dbs = malloc(sizeof *dbs, M_TEMP, M_WAITOK);
785 dbs->dbs_min = bp->b_rawblkno;
786 dbs->dbs_max = dbs->dbs_min + (bp->b_bcount >> DEV_BSHIFT) - 1;
787 microtime(&dbs->dbs_failedat);
788 SLIST_INSERT_HEAD(&wd->sc_bslist, dbs, dbs_next);
789 wd->sc_bscount++;
790 }
791
792 bp->b_flags |= B_ERROR;
793 bp->b_error = EIO;
794 break;
795 case NOERROR:
796 noerror: if ((wd->sc_wdc_bio.flags & ATA_CORR) || wd->retries > 0)
797 printf("%s: soft error (corrected)\n",
798 wd->sc_dev.dv_xname);
799 break;
800 case ERR_NODEV:
801 bp->b_flags |= B_ERROR;
802 bp->b_error = EIO;
803 break;
804 }
805 disk_unbusy(&wd->sc_dk, (bp->b_bcount - bp->b_resid),
806 (bp->b_flags & B_READ));
807 #if NRND > 0
808 rnd_add_uint32(&wd->rnd_source, bp->b_blkno);
809 #endif
810 /* XXX Yuck, but we don't want to increment openings in this case */
811 if (__predict_false((bp->b_flags & B_CALL) != 0 &&
812 bp->b_iodone == wd_split_mod15_write))
813 biodone(bp);
814 else {
815 biodone(bp);
816 wd->openings++;
817 }
818 wdstart(wd);
819 }
820
821 void
822 wdrestart(void *v)
823 {
824 struct wd_softc *wd = v;
825 struct buf *bp = wd->sc_bp;
826 int s;
827 WDCDEBUG_PRINT(("wdrestart %s\n", wd->sc_dev.dv_xname),
828 DEBUG_XFERS);
829
830 s = splbio();
831 __wdstart(v, bp);
832 splx(s);
833 }
834
835 int
836 wdread(dev_t dev, struct uio *uio, int flags)
837 {
838
839 WDCDEBUG_PRINT(("wdread\n"), DEBUG_XFERS);
840 return (physio(wdstrategy, NULL, dev, B_READ, minphys, uio));
841 }
842
843 int
844 wdwrite(dev_t dev, struct uio *uio, int flags)
845 {
846
847 WDCDEBUG_PRINT(("wdwrite\n"), DEBUG_XFERS);
848 return (physio(wdstrategy, NULL, dev, B_WRITE, minphys, uio));
849 }
850
851 int
852 wdopen(dev_t dev, int flag, int fmt, struct proc *p)
853 {
854 struct wd_softc *wd;
855 int part, error;
856
857 WDCDEBUG_PRINT(("wdopen\n"), DEBUG_FUNCS);
858 wd = device_lookup(&wd_cd, WDUNIT(dev));
859 if (wd == NULL)
860 return (ENXIO);
861
862 if ((wd->sc_dev.dv_flags & DVF_ACTIVE) == 0)
863 return (ENODEV);
864
865 /*
866 * If this is the first open of this device, add a reference
867 * to the adapter.
868 */
869 if (wd->sc_dk.dk_openmask == 0 &&
870 (error = wd->atabus->ata_addref(wd->drvp)) != 0)
871 return (error);
872
873 if ((error = lockmgr(&wd->sc_lock, LK_EXCLUSIVE, NULL)) != 0)
874 goto bad4;
875
876 if (wd->sc_dk.dk_openmask != 0) {
877 /*
878 * If any partition is open, but the disk has been invalidated,
879 * disallow further opens.
880 */
881 if ((wd->sc_flags & WDF_LOADED) == 0) {
882 error = EIO;
883 goto bad3;
884 }
885 } else {
886 if ((wd->sc_flags & WDF_LOADED) == 0) {
887 wd->sc_flags |= WDF_LOADED;
888
889 /* Load the physical device parameters. */
890 wd_get_params(wd, AT_WAIT, &wd->sc_params);
891
892 /* Load the partition info if not already loaded. */
893 wdgetdisklabel(wd);
894 }
895 }
896
897 part = WDPART(dev);
898
899 /* Check that the partition exists. */
900 if (part != RAW_PART &&
901 (part >= wd->sc_dk.dk_label->d_npartitions ||
902 wd->sc_dk.dk_label->d_partitions[part].p_fstype == FS_UNUSED)) {
903 error = ENXIO;
904 goto bad;
905 }
906
907 /* Insure only one open at a time. */
908 switch (fmt) {
909 case S_IFCHR:
910 wd->sc_dk.dk_copenmask |= (1 << part);
911 break;
912 case S_IFBLK:
913 wd->sc_dk.dk_bopenmask |= (1 << part);
914 break;
915 }
916 wd->sc_dk.dk_openmask =
917 wd->sc_dk.dk_copenmask | wd->sc_dk.dk_bopenmask;
918
919 lockmgr(&wd->sc_lock, LK_RELEASE, NULL);
920 return 0;
921
922 bad:
923 if (wd->sc_dk.dk_openmask == 0) {
924 }
925
926 bad3:
927 lockmgr(&wd->sc_lock, LK_RELEASE, NULL);
928 bad4:
929 if (wd->sc_dk.dk_openmask == 0)
930 wd->atabus->ata_delref(wd->drvp);
931 return error;
932 }
933
934 int
935 wdclose(dev_t dev, int flag, int fmt, struct proc *p)
936 {
937 struct wd_softc *wd = device_lookup(&wd_cd, WDUNIT(dev));
938 int part = WDPART(dev);
939 int error;
940
941 WDCDEBUG_PRINT(("wdclose\n"), DEBUG_FUNCS);
942 if ((error = lockmgr(&wd->sc_lock, LK_EXCLUSIVE, NULL)) != 0)
943 return error;
944
945 switch (fmt) {
946 case S_IFCHR:
947 wd->sc_dk.dk_copenmask &= ~(1 << part);
948 break;
949 case S_IFBLK:
950 wd->sc_dk.dk_bopenmask &= ~(1 << part);
951 break;
952 }
953 wd->sc_dk.dk_openmask =
954 wd->sc_dk.dk_copenmask | wd->sc_dk.dk_bopenmask;
955
956 if (wd->sc_dk.dk_openmask == 0) {
957 wd_flushcache(wd, AT_WAIT);
958 /* XXXX Must wait for I/O to complete! */
959
960 if (! (wd->sc_flags & WDF_KLABEL))
961 wd->sc_flags &= ~WDF_LOADED;
962
963 wd->atabus->ata_delref(wd->drvp);
964 }
965
966 lockmgr(&wd->sc_lock, LK_RELEASE, NULL);
967 return 0;
968 }
969
970 void
971 wdgetdefaultlabel(struct wd_softc *wd, struct disklabel *lp)
972 {
973
974 WDCDEBUG_PRINT(("wdgetdefaultlabel\n"), DEBUG_FUNCS);
975 memset(lp, 0, sizeof(struct disklabel));
976
977 lp->d_secsize = DEV_BSIZE;
978 lp->d_ntracks = wd->sc_params.atap_heads;
979 lp->d_nsectors = wd->sc_params.atap_sectors;
980 lp->d_ncylinders = (wd->sc_flags & WDF_LBA) ? wd->sc_capacity /
981 (wd->sc_params.atap_heads * wd->sc_params.atap_sectors) :
982 wd->sc_params.atap_cylinders;
983 lp->d_secpercyl = lp->d_ntracks * lp->d_nsectors;
984
985 if (strcmp(wd->sc_params.atap_model, "ST506") == 0)
986 lp->d_type = DTYPE_ST506;
987 else
988 lp->d_type = DTYPE_ESDI;
989
990 strncpy(lp->d_typename, wd->sc_params.atap_model, 16);
991 strncpy(lp->d_packname, "fictitious", 16);
992 if (wd->sc_capacity > UINT32_MAX)
993 lp->d_secperunit = UINT32_MAX;
994 else
995 lp->d_secperunit = wd->sc_capacity;
996 lp->d_rpm = 3600;
997 lp->d_interleave = 1;
998 lp->d_flags = 0;
999
1000 lp->d_partitions[RAW_PART].p_offset = 0;
1001 lp->d_partitions[RAW_PART].p_size =
1002 lp->d_secperunit * (lp->d_secsize / DEV_BSIZE);
1003 lp->d_partitions[RAW_PART].p_fstype = FS_UNUSED;
1004 lp->d_npartitions = RAW_PART + 1;
1005
1006 lp->d_magic = DISKMAGIC;
1007 lp->d_magic2 = DISKMAGIC;
1008 lp->d_checksum = dkcksum(lp);
1009 }
1010
1011 /*
1012 * Fabricate a default disk label, and try to read the correct one.
1013 */
1014 void
1015 wdgetdisklabel(struct wd_softc *wd)
1016 {
1017 struct disklabel *lp = wd->sc_dk.dk_label;
1018 const char *errstring;
1019
1020 WDCDEBUG_PRINT(("wdgetdisklabel\n"), DEBUG_FUNCS);
1021
1022 memset(wd->sc_dk.dk_cpulabel, 0, sizeof(struct cpu_disklabel));
1023
1024 wdgetdefaultlabel(wd, lp);
1025
1026 wd->sc_badsect[0] = -1;
1027
1028 if (wd->drvp->state > RESET)
1029 wd->drvp->drive_flags |= DRIVE_RESET;
1030 errstring = readdisklabel(MAKEWDDEV(0, wd->sc_dev.dv_unit, RAW_PART),
1031 wdstrategy, lp, wd->sc_dk.dk_cpulabel);
1032 if (errstring) {
1033 /*
1034 * This probably happened because the drive's default
1035 * geometry doesn't match the DOS geometry. We
1036 * assume the DOS geometry is now in the label and try
1037 * again. XXX This is a kluge.
1038 */
1039 if (wd->drvp->state > RESET)
1040 wd->drvp->drive_flags |= DRIVE_RESET;
1041 errstring = readdisklabel(MAKEWDDEV(0, wd->sc_dev.dv_unit,
1042 RAW_PART), wdstrategy, lp, wd->sc_dk.dk_cpulabel);
1043 }
1044 if (errstring) {
1045 printf("%s: %s\n", wd->sc_dev.dv_xname, errstring);
1046 return;
1047 }
1048
1049 if (wd->drvp->state > RESET)
1050 wd->drvp->drive_flags |= DRIVE_RESET;
1051 #ifdef HAS_BAD144_HANDLING
1052 if ((lp->d_flags & D_BADSECT) != 0)
1053 bad144intern(wd);
1054 #endif
1055 }
1056
1057 void
1058 wdperror(const struct wd_softc *wd)
1059 {
1060 static const char *const errstr0_3[] = {"address mark not found",
1061 "track 0 not found", "aborted command", "media change requested",
1062 "id not found", "media changed", "uncorrectable data error",
1063 "bad block detected"};
1064 static const char *const errstr4_5[] = {
1065 "obsolete (address mark not found)",
1066 "no media/write protected", "aborted command",
1067 "media change requested", "id not found", "media changed",
1068 "uncorrectable data error", "interface CRC error"};
1069 const char *const *errstr;
1070 int i;
1071 char *sep = "";
1072
1073 const char *devname = wd->sc_dev.dv_xname;
1074 struct ata_drive_datas *drvp = wd->drvp;
1075 int errno = wd->sc_wdc_bio.r_error;
1076
1077 if (drvp->ata_vers >= 4)
1078 errstr = errstr4_5;
1079 else
1080 errstr = errstr0_3;
1081
1082 printf("%s: (", devname);
1083
1084 if (errno == 0)
1085 printf("error not notified");
1086
1087 for (i = 0; i < 8; i++) {
1088 if (errno & (1 << i)) {
1089 printf("%s%s", sep, errstr[i]);
1090 sep = ", ";
1091 }
1092 }
1093 printf(")\n");
1094 }
1095
1096 int
1097 wdioctl(dev_t dev, u_long xfer, caddr_t addr, int flag, struct proc *p)
1098 {
1099 struct wd_softc *wd = device_lookup(&wd_cd, WDUNIT(dev));
1100 int error = 0;
1101 #ifdef __HAVE_OLD_DISKLABEL
1102 struct disklabel *newlabel = NULL;
1103 #endif
1104
1105 WDCDEBUG_PRINT(("wdioctl\n"), DEBUG_FUNCS);
1106
1107 if ((wd->sc_flags & WDF_LOADED) == 0)
1108 return EIO;
1109
1110 switch (xfer) {
1111 #ifdef HAS_BAD144_HANDLING
1112 case DIOCSBAD:
1113 if ((flag & FWRITE) == 0)
1114 return EBADF;
1115 wd->sc_dk.dk_cpulabel->bad = *(struct dkbad *)addr;
1116 wd->sc_dk.dk_label->d_flags |= D_BADSECT;
1117 bad144intern(wd);
1118 return 0;
1119 #endif
1120
1121 case DIOCBSLIST :
1122 {
1123 u_int32_t count, missing, skip;
1124 struct disk_badsecinfo dbsi;
1125 struct disk_badsectors *dbs;
1126 size_t available;
1127 caddr_t laddr;
1128
1129 dbsi = *(struct disk_badsecinfo *)addr;
1130 missing = wd->sc_bscount;
1131 count = 0;
1132 available = dbsi.dbsi_bufsize;
1133 skip = dbsi.dbsi_skip;
1134 laddr = dbsi.dbsi_buffer;
1135
1136 /*
1137 * We start this loop with the expectation that all of the
1138 * entries will be missed and decrement this counter each
1139 * time we either skip over one (already copied out) or
1140 * we actually copy it back to user space. The structs
1141 * holding the bad sector information are copied directly
1142 * back to user space whilst the summary is returned via
1143 * the struct passed in via the ioctl.
1144 */
1145 SLIST_FOREACH(dbs, &wd->sc_bslist, dbs_next) {
1146 if (skip > 0) {
1147 missing--;
1148 skip--;
1149 continue;
1150 }
1151 if (available < sizeof(*dbs))
1152 break;
1153 available -= sizeof(*dbs);
1154 copyout(dbs, laddr, sizeof(*dbs));
1155 laddr += sizeof(*dbs);
1156 missing--;
1157 count++;
1158 }
1159 dbsi.dbsi_left = missing;
1160 dbsi.dbsi_copied = count;
1161 *(struct disk_badsecinfo *)addr = dbsi;
1162 return 0;
1163 }
1164
1165 case DIOCBSFLUSH :
1166 /* Clean out the bad sector list */
1167 while (!SLIST_EMPTY(&wd->sc_bslist)) {
1168 void *head = SLIST_FIRST(&wd->sc_bslist);
1169 SLIST_REMOVE_HEAD(&wd->sc_bslist, dbs_next);
1170 free(head, M_TEMP);
1171 }
1172 wd->sc_bscount = 0;
1173 return 0;
1174
1175 case DIOCGDINFO:
1176 *(struct disklabel *)addr = *(wd->sc_dk.dk_label);
1177 return 0;
1178 #ifdef __HAVE_OLD_DISKLABEL
1179 case ODIOCGDINFO:
1180 newlabel = malloc(sizeof *newlabel, M_TEMP, M_WAITOK);
1181 if (newlabel == NULL)
1182 return EIO;
1183 *newlabel = *(wd->sc_dk.dk_label);
1184 if (newlabel->d_npartitions <= OLDMAXPARTITIONS)
1185 memcpy(addr, newlabel, sizeof (struct olddisklabel));
1186 else
1187 error = ENOTTY;
1188 free(newlabel, M_TEMP);
1189 return error;
1190 #endif
1191
1192 case DIOCGPART:
1193 ((struct partinfo *)addr)->disklab = wd->sc_dk.dk_label;
1194 ((struct partinfo *)addr)->part =
1195 &wd->sc_dk.dk_label->d_partitions[WDPART(dev)];
1196 return 0;
1197
1198 case DIOCWDINFO:
1199 case DIOCSDINFO:
1200 #ifdef __HAVE_OLD_DISKLABEL
1201 case ODIOCWDINFO:
1202 case ODIOCSDINFO:
1203 #endif
1204 {
1205 struct disklabel *lp;
1206
1207 if ((flag & FWRITE) == 0)
1208 return EBADF;
1209
1210 #ifdef __HAVE_OLD_DISKLABEL
1211 if (xfer == ODIOCSDINFO || xfer == ODIOCWDINFO) {
1212 newlabel = malloc(sizeof *newlabel, M_TEMP, M_WAITOK);
1213 if (newlabel == NULL)
1214 return EIO;
1215 memset(newlabel, 0, sizeof newlabel);
1216 memcpy(newlabel, addr, sizeof (struct olddisklabel));
1217 lp = newlabel;
1218 } else
1219 #endif
1220 lp = (struct disklabel *)addr;
1221
1222 if ((error = lockmgr(&wd->sc_lock, LK_EXCLUSIVE, NULL)) != 0)
1223 goto bad;
1224 wd->sc_flags |= WDF_LABELLING;
1225
1226 error = setdisklabel(wd->sc_dk.dk_label,
1227 lp, /*wd->sc_dk.dk_openmask : */0,
1228 wd->sc_dk.dk_cpulabel);
1229 if (error == 0) {
1230 if (wd->drvp->state > RESET)
1231 wd->drvp->drive_flags |= DRIVE_RESET;
1232 if (xfer == DIOCWDINFO
1233 #ifdef __HAVE_OLD_DISKLABEL
1234 || xfer == ODIOCWDINFO
1235 #endif
1236 )
1237 error = writedisklabel(WDLABELDEV(dev),
1238 wdstrategy, wd->sc_dk.dk_label,
1239 wd->sc_dk.dk_cpulabel);
1240 }
1241
1242 wd->sc_flags &= ~WDF_LABELLING;
1243 lockmgr(&wd->sc_lock, LK_RELEASE, NULL);
1244 bad:
1245 #ifdef __HAVE_OLD_DISKLABEL
1246 if (newlabel != NULL)
1247 free(newlabel, M_TEMP);
1248 #endif
1249 return error;
1250 }
1251
1252 case DIOCKLABEL:
1253 if (*(int *)addr)
1254 wd->sc_flags |= WDF_KLABEL;
1255 else
1256 wd->sc_flags &= ~WDF_KLABEL;
1257 return 0;
1258
1259 case DIOCWLABEL:
1260 if ((flag & FWRITE) == 0)
1261 return EBADF;
1262 if (*(int *)addr)
1263 wd->sc_flags |= WDF_WLABEL;
1264 else
1265 wd->sc_flags &= ~WDF_WLABEL;
1266 return 0;
1267
1268 case DIOCGDEFLABEL:
1269 wdgetdefaultlabel(wd, (struct disklabel *)addr);
1270 return 0;
1271 #ifdef __HAVE_OLD_DISKLABEL
1272 case ODIOCGDEFLABEL:
1273 newlabel = malloc(sizeof *newlabel, M_TEMP, M_WAITOK);
1274 if (newlabel == NULL)
1275 return EIO;
1276 wdgetdefaultlabel(wd, newlabel);
1277 if (newlabel->d_npartitions <= OLDMAXPARTITIONS)
1278 memcpy(addr, &newlabel, sizeof (struct olddisklabel));
1279 else
1280 error = ENOTTY;
1281 free(newlabel, M_TEMP);
1282 return error;
1283 #endif
1284
1285 #ifdef notyet
1286 case DIOCWFORMAT:
1287 if ((flag & FWRITE) == 0)
1288 return EBADF;
1289 {
1290 register struct format_op *fop;
1291 struct iovec aiov;
1292 struct uio auio;
1293
1294 fop = (struct format_op *)addr;
1295 aiov.iov_base = fop->df_buf;
1296 aiov.iov_len = fop->df_count;
1297 auio.uio_iov = &aiov;
1298 auio.uio_iovcnt = 1;
1299 auio.uio_resid = fop->df_count;
1300 auio.uio_segflg = 0;
1301 auio.uio_offset =
1302 fop->df_startblk * wd->sc_dk.dk_label->d_secsize;
1303 auio.uio_procp = p;
1304 error = physio(wdformat, NULL, dev, B_WRITE, minphys,
1305 &auio);
1306 fop->df_count -= auio.uio_resid;
1307 fop->df_reg[0] = wdc->sc_status;
1308 fop->df_reg[1] = wdc->sc_error;
1309 return error;
1310 }
1311 #endif
1312 case DIOCGCACHE:
1313 return wd_getcache(wd, (int *)addr);
1314
1315 case DIOCSCACHE:
1316 return wd_setcache(wd, *(int *)addr);
1317
1318 case DIOCCACHESYNC:
1319 return wd_flushcache(wd, AT_WAIT);
1320
1321 case ATAIOCCOMMAND:
1322 /*
1323 * Make sure this command is (relatively) safe first
1324 */
1325 if ((((atareq_t *) addr)->flags & ATACMD_READ) == 0 &&
1326 (flag & FWRITE) == 0)
1327 return (EBADF);
1328 {
1329 struct wd_ioctl *wi;
1330 atareq_t *atareq = (atareq_t *) addr;
1331 int error;
1332
1333 wi = wi_get();
1334 wi->wi_softc = wd;
1335 wi->wi_atareq = *atareq;
1336
1337 if (atareq->datalen && atareq->flags &
1338 (ATACMD_READ | ATACMD_WRITE)) {
1339 wi->wi_iov.iov_base = atareq->databuf;
1340 wi->wi_iov.iov_len = atareq->datalen;
1341 wi->wi_uio.uio_iov = &wi->wi_iov;
1342 wi->wi_uio.uio_iovcnt = 1;
1343 wi->wi_uio.uio_resid = atareq->datalen;
1344 wi->wi_uio.uio_offset = 0;
1345 wi->wi_uio.uio_segflg = UIO_USERSPACE;
1346 wi->wi_uio.uio_rw =
1347 (atareq->flags & ATACMD_READ) ? B_READ : B_WRITE;
1348 wi->wi_uio.uio_procp = p;
1349 error = physio(wdioctlstrategy, &wi->wi_bp, dev,
1350 (atareq->flags & ATACMD_READ) ? B_READ : B_WRITE,
1351 minphys, &wi->wi_uio);
1352 } else {
1353 /* No need to call physio if we don't have any
1354 user data */
1355 wi->wi_bp.b_flags = 0;
1356 wi->wi_bp.b_data = 0;
1357 wi->wi_bp.b_bcount = 0;
1358 wi->wi_bp.b_dev = 0;
1359 wi->wi_bp.b_proc = p;
1360 wdioctlstrategy(&wi->wi_bp);
1361 error = wi->wi_bp.b_error;
1362 }
1363 *atareq = wi->wi_atareq;
1364 wi_free(wi);
1365 return(error);
1366 }
1367
1368 default:
1369 return ENOTTY;
1370 }
1371
1372 #ifdef DIAGNOSTIC
1373 panic("wdioctl: impossible");
1374 #endif
1375 }
1376
1377 #ifdef B_FORMAT
1378 int
1379 wdformat(struct buf *bp)
1380 {
1381
1382 bp->b_flags |= B_FORMAT;
1383 return wdstrategy(bp);
1384 }
1385 #endif
1386
1387 int
1388 wdsize(dev_t dev)
1389 {
1390 struct wd_softc *wd;
1391 int part, omask;
1392 int size;
1393
1394 WDCDEBUG_PRINT(("wdsize\n"), DEBUG_FUNCS);
1395
1396 wd = device_lookup(&wd_cd, WDUNIT(dev));
1397 if (wd == NULL)
1398 return (-1);
1399
1400 part = WDPART(dev);
1401 omask = wd->sc_dk.dk_openmask & (1 << part);
1402
1403 if (omask == 0 && wdopen(dev, 0, S_IFBLK, NULL) != 0)
1404 return (-1);
1405 if (wd->sc_dk.dk_label->d_partitions[part].p_fstype != FS_SWAP)
1406 size = -1;
1407 else
1408 size = wd->sc_dk.dk_label->d_partitions[part].p_size *
1409 (wd->sc_dk.dk_label->d_secsize / DEV_BSIZE);
1410 if (omask == 0 && wdclose(dev, 0, S_IFBLK, NULL) != 0)
1411 return (-1);
1412 return (size);
1413 }
1414
1415 /* #define WD_DUMP_NOT_TRUSTED if you just want to watch */
1416 static int wddoingadump = 0;
1417 static int wddumprecalibrated = 0;
1418 static int wddumpmulti = 1;
1419
1420 /*
1421 * Dump core after a system crash.
1422 */
1423 int
1424 wddump(dev_t dev, daddr_t blkno, caddr_t va, size_t size)
1425 {
1426 struct wd_softc *wd; /* disk unit to do the I/O */
1427 struct disklabel *lp; /* disk's disklabel */
1428 int part, err;
1429 int nblks; /* total number of sectors left to write */
1430
1431 /* Check if recursive dump; if so, punt. */
1432 if (wddoingadump)
1433 return EFAULT;
1434 wddoingadump = 1;
1435
1436 wd = device_lookup(&wd_cd, WDUNIT(dev));
1437 if (wd == NULL)
1438 return (ENXIO);
1439
1440 part = WDPART(dev);
1441
1442 /* Convert to disk sectors. Request must be a multiple of size. */
1443 lp = wd->sc_dk.dk_label;
1444 if ((size % lp->d_secsize) != 0)
1445 return EFAULT;
1446 nblks = size / lp->d_secsize;
1447 blkno = blkno / (lp->d_secsize / DEV_BSIZE);
1448
1449 /* Check transfer bounds against partition size. */
1450 if ((blkno < 0) || ((blkno + nblks) > lp->d_partitions[part].p_size))
1451 return EINVAL;
1452
1453 /* Offset block number to start of partition. */
1454 blkno += lp->d_partitions[part].p_offset;
1455
1456 /* Recalibrate, if first dump transfer. */
1457 if (wddumprecalibrated == 0) {
1458 wddumpmulti = wd->sc_multi;
1459 wddumprecalibrated = 1;
1460 wd->atabus->ata_reset_channel(wd->drvp, AT_POLL | AT_RST_EMERG);
1461 wd->drvp->state = RESET;
1462 }
1463
1464 while (nblks > 0) {
1465 wd->sc_bp = NULL;
1466 wd->sc_wdc_bio.blkno = blkno;
1467 wd->sc_wdc_bio.flags = ATA_POLL;
1468 if (wd->sc_flags & WDF_LBA48 && blkno > LBA48_THRESHOLD)
1469 wd->sc_wdc_bio.flags |= ATA_LBA48;
1470 if (wd->sc_flags & WDF_LBA)
1471 wd->sc_wdc_bio.flags |= ATA_LBA;
1472 wd->sc_wdc_bio.bcount =
1473 min(nblks, wddumpmulti) * lp->d_secsize;
1474 wd->sc_wdc_bio.databuf = va;
1475 #ifndef WD_DUMP_NOT_TRUSTED
1476 switch (wd->atabus->ata_bio(wd->drvp, &wd->sc_wdc_bio)) {
1477 case WDC_TRY_AGAIN:
1478 panic("wddump: try again");
1479 break;
1480 case WDC_QUEUED:
1481 panic("wddump: polled command has been queued");
1482 break;
1483 case WDC_COMPLETE:
1484 break;
1485 }
1486 switch(wd->sc_wdc_bio.error) {
1487 case TIMEOUT:
1488 printf("wddump: device timed out");
1489 err = EIO;
1490 break;
1491 case ERR_DF:
1492 printf("wddump: drive fault");
1493 err = EIO;
1494 break;
1495 case ERR_DMA:
1496 printf("wddump: DMA error");
1497 err = EIO;
1498 break;
1499 case ERROR:
1500 printf("wddump: ");
1501 wdperror(wd);
1502 err = EIO;
1503 break;
1504 case NOERROR:
1505 err = 0;
1506 break;
1507 default:
1508 panic("wddump: unknown error type");
1509 }
1510 if (err != 0) {
1511 printf("\n");
1512 return err;
1513 }
1514 #else /* WD_DUMP_NOT_TRUSTED */
1515 /* Let's just talk about this first... */
1516 printf("wd%d: dump addr 0x%x, cylin %d, head %d, sector %d\n",
1517 unit, va, cylin, head, sector);
1518 delay(500 * 1000); /* half a second */
1519 #endif
1520
1521 /* update block count */
1522 nblks -= min(nblks, wddumpmulti);
1523 blkno += min(nblks, wddumpmulti);
1524 va += min(nblks, wddumpmulti) * lp->d_secsize;
1525 }
1526
1527 wddoingadump = 0;
1528 return 0;
1529 }
1530
1531 #ifdef HAS_BAD144_HANDLING
1532 /*
1533 * Internalize the bad sector table.
1534 */
1535 void
1536 bad144intern(struct wd_softc *wd)
1537 {
1538 struct dkbad *bt = &wd->sc_dk.dk_cpulabel->bad;
1539 struct disklabel *lp = wd->sc_dk.dk_label;
1540 int i = 0;
1541
1542 WDCDEBUG_PRINT(("bad144intern\n"), DEBUG_XFERS);
1543
1544 for (; i < NBT_BAD; i++) {
1545 if (bt->bt_bad[i].bt_cyl == 0xffff)
1546 break;
1547 wd->sc_badsect[i] =
1548 bt->bt_bad[i].bt_cyl * lp->d_secpercyl +
1549 (bt->bt_bad[i].bt_trksec >> 8) * lp->d_nsectors +
1550 (bt->bt_bad[i].bt_trksec & 0xff);
1551 }
1552 for (; i < NBT_BAD+1; i++)
1553 wd->sc_badsect[i] = -1;
1554 }
1555 #endif
1556
1557 int
1558 wd_get_params(struct wd_softc *wd, u_int8_t flags, struct ataparams *params)
1559 {
1560 switch (wd->atabus->ata_get_params(wd->drvp, flags, params)) {
1561 case CMD_AGAIN:
1562 return 1;
1563 case CMD_ERR:
1564 /*
1565 * We `know' there's a drive here; just assume it's old.
1566 * This geometry is only used to read the MBR and print a
1567 * (false) attach message.
1568 */
1569 strncpy(params->atap_model, "ST506",
1570 sizeof params->atap_model);
1571 params->atap_config = ATA_CFG_FIXED;
1572 params->atap_cylinders = 1024;
1573 params->atap_heads = 8;
1574 params->atap_sectors = 17;
1575 params->atap_multi = 1;
1576 params->atap_capabilities1 = params->atap_capabilities2 = 0;
1577 wd->drvp->ata_vers = -1; /* Mark it as pre-ATA */
1578 return 0;
1579 case CMD_OK:
1580 return 0;
1581 default:
1582 panic("wd_get_params: bad return code from ata_get_params");
1583 /* NOTREACHED */
1584 }
1585 }
1586
1587 int
1588 wd_getcache(struct wd_softc *wd, int *bitsp)
1589 {
1590 struct ataparams params;
1591
1592 if (wd_get_params(wd, AT_WAIT, ¶ms) != 0)
1593 return EIO;
1594 if (params.atap_cmd_set1 == 0x0000 ||
1595 params.atap_cmd_set1 == 0xffff ||
1596 (params.atap_cmd_set1 & WDC_CMD1_CACHE) == 0) {
1597 *bitsp = 0;
1598 return 0;
1599 }
1600 *bitsp = DKCACHE_WCHANGE | DKCACHE_READ;
1601 if (params.atap_cmd1_en & WDC_CMD1_CACHE)
1602 *bitsp |= DKCACHE_WRITE;
1603
1604 return 0;
1605 }
1606
1607 int
1608 wd_setcache(struct wd_softc *wd, int bits)
1609 {
1610 struct ataparams params;
1611 struct wdc_command wdc_c;
1612
1613 if (wd_get_params(wd, AT_WAIT, ¶ms) != 0)
1614 return EIO;
1615
1616 if (params.atap_cmd_set1 == 0x0000 ||
1617 params.atap_cmd_set1 == 0xffff ||
1618 (params.atap_cmd_set1 & WDC_CMD1_CACHE) == 0)
1619 return EOPNOTSUPP;
1620
1621 if ((bits & DKCACHE_READ) == 0 ||
1622 (bits & DKCACHE_SAVE) != 0)
1623 return EOPNOTSUPP;
1624
1625 memset(&wdc_c, 0, sizeof(struct wdc_command));
1626 wdc_c.r_command = SET_FEATURES;
1627 wdc_c.r_st_bmask = 0;
1628 wdc_c.r_st_pmask = 0;
1629 wdc_c.timeout = 30000; /* 30s timeout */
1630 wdc_c.flags = AT_WAIT;
1631 if (bits & DKCACHE_WRITE)
1632 wdc_c.r_features = WDSF_WRITE_CACHE_EN;
1633 else
1634 wdc_c.r_features = WDSF_WRITE_CACHE_DS;
1635 if (wd->atabus->ata_exec_command(wd->drvp, &wdc_c) != WDC_COMPLETE) {
1636 printf("%s: wd_setcache command not complete\n",
1637 wd->sc_dev.dv_xname);
1638 return EIO;
1639 }
1640 if (wdc_c.flags & (AT_ERROR | AT_TIMEOU | AT_DF)) {
1641 printf("%s: wd_setcache command error 0x%x\n",
1642 wd->sc_dev.dv_xname, wdc_c.flags);
1643 return EIO;
1644 }
1645 if (wdc_c.flags & ERR_NODEV)
1646 return ENODEV;
1647 return 0;
1648 }
1649
1650 int
1651 wd_flushcache(struct wd_softc *wd, int flags)
1652 {
1653 struct wdc_command wdc_c;
1654
1655 if (wd->drvp->ata_vers < 4) /* WDCC_FLUSHCACHE is here since ATA-4 */
1656 return ENODEV;
1657 memset(&wdc_c, 0, sizeof(struct wdc_command));
1658 if ((wd->sc_params.atap_cmd2_en & ATA_CMD2_LBA48) != 0 &&
1659 (wd->sc_params.atap_cmd2_en & ATA_CMD2_FCE) != 0)
1660 wdc_c.r_command = WDCC_FLUSHCACHE_EXT;
1661 else
1662 wdc_c.r_command = WDCC_FLUSHCACHE;
1663 wdc_c.r_st_bmask = WDCS_DRDY;
1664 wdc_c.r_st_pmask = WDCS_DRDY;
1665 wdc_c.flags = flags;
1666 wdc_c.timeout = 30000; /* 30s timeout */
1667 if (wd->atabus->ata_exec_command(wd->drvp, &wdc_c) != WDC_COMPLETE) {
1668 printf("%s: flush cache command didn't complete\n",
1669 wd->sc_dev.dv_xname);
1670 return EIO;
1671 }
1672 if (wdc_c.flags & ERR_NODEV)
1673 return ENODEV;
1674 if (wdc_c.flags & AT_TIMEOU) {
1675 printf("%s: flush cache command timeout\n",
1676 wd->sc_dev.dv_xname);
1677 return EIO;
1678 }
1679 if (wdc_c.flags & AT_ERROR) {
1680 if (wdc_c.r_error == WDCE_ABRT) /* command not supported */
1681 return ENODEV;
1682 printf("%s: flush cache command: error 0x%x\n",
1683 wd->sc_dev.dv_xname, wdc_c.r_error);
1684 return EIO;
1685 }
1686 if (wdc_c.flags & AT_DF) {
1687 printf("%s: flush cache command: drive fault\n",
1688 wd->sc_dev.dv_xname);
1689 return EIO;
1690 }
1691 return 0;
1692 }
1693
1694 void
1695 wd_shutdown(void *arg)
1696 {
1697 struct wd_softc *wd = arg;
1698 wd_flushcache(wd, AT_POLL);
1699 }
1700
1701 /*
1702 * Allocate space for a ioctl queue structure. Mostly taken from
1703 * scsipi_ioctl.c
1704 */
1705 struct wd_ioctl *
1706 wi_get(void)
1707 {
1708 struct wd_ioctl *wi;
1709 int s;
1710
1711 wi = malloc(sizeof(struct wd_ioctl), M_TEMP, M_WAITOK|M_ZERO);
1712 simple_lock_init(&wi->wi_bp.b_interlock);
1713 s = splbio();
1714 LIST_INSERT_HEAD(&wi_head, wi, wi_list);
1715 splx(s);
1716 return (wi);
1717 }
1718
1719 /*
1720 * Free an ioctl structure and remove it from our list
1721 */
1722
1723 void
1724 wi_free(struct wd_ioctl *wi)
1725 {
1726 int s;
1727
1728 s = splbio();
1729 LIST_REMOVE(wi, wi_list);
1730 splx(s);
1731 free(wi, M_TEMP);
1732 }
1733
1734 /*
1735 * Find a wd_ioctl structure based on the struct buf.
1736 */
1737
1738 struct wd_ioctl *
1739 wi_find(struct buf *bp)
1740 {
1741 struct wd_ioctl *wi;
1742 int s;
1743
1744 s = splbio();
1745 for (wi = wi_head.lh_first; wi != 0; wi = wi->wi_list.le_next)
1746 if (bp == &wi->wi_bp)
1747 break;
1748 splx(s);
1749 return (wi);
1750 }
1751
1752 /*
1753 * Ioctl pseudo strategy routine
1754 *
1755 * This is mostly stolen from scsipi_ioctl.c:scsistrategy(). What
1756 * happens here is:
1757 *
1758 * - wdioctl() queues a wd_ioctl structure.
1759 *
1760 * - wdioctl() calls physio/wdioctlstrategy based on whether or not
1761 * user space I/O is required. If physio() is called, physio() eventually
1762 * calls wdioctlstrategy().
1763 *
1764 * - In either case, wdioctlstrategy() calls wd->atabus->ata_exec_command()
1765 * to perform the actual command
1766 *
1767 * The reason for the use of the pseudo strategy routine is because
1768 * when doing I/O to/from user space, physio _really_ wants to be in
1769 * the loop. We could put the entire buffer into the ioctl request
1770 * structure, but that won't scale if we want to do things like download
1771 * microcode.
1772 */
1773
1774 void
1775 wdioctlstrategy(struct buf *bp)
1776 {
1777 struct wd_ioctl *wi;
1778 struct wdc_command wdc_c;
1779 int error = 0;
1780
1781 wi = wi_find(bp);
1782 if (wi == NULL) {
1783 printf("user_strat: No ioctl\n");
1784 error = EINVAL;
1785 goto bad;
1786 }
1787
1788 memset(&wdc_c, 0, sizeof(wdc_c));
1789
1790 /*
1791 * Abort if physio broke up the transfer
1792 */
1793
1794 if (bp->b_bcount != wi->wi_atareq.datalen) {
1795 printf("physio split wd ioctl request... cannot proceed\n");
1796 error = EIO;
1797 goto bad;
1798 }
1799
1800 /*
1801 * Abort if we didn't get a buffer size that was a multiple of
1802 * our sector size (or was larger than NBBY)
1803 */
1804
1805 if ((bp->b_bcount % wi->wi_softc->sc_dk.dk_label->d_secsize) != 0 ||
1806 (bp->b_bcount / wi->wi_softc->sc_dk.dk_label->d_secsize) >=
1807 (1 << NBBY)) {
1808 error = EINVAL;
1809 goto bad;
1810 }
1811
1812 /*
1813 * Make sure a timeout was supplied in the ioctl request
1814 */
1815
1816 if (wi->wi_atareq.timeout == 0) {
1817 error = EINVAL;
1818 goto bad;
1819 }
1820
1821 if (wi->wi_atareq.flags & ATACMD_READ)
1822 wdc_c.flags |= AT_READ;
1823 else if (wi->wi_atareq.flags & ATACMD_WRITE)
1824 wdc_c.flags |= AT_WRITE;
1825
1826 if (wi->wi_atareq.flags & ATACMD_READREG)
1827 wdc_c.flags |= AT_READREG;
1828
1829 wdc_c.flags |= AT_WAIT;
1830
1831 wdc_c.timeout = wi->wi_atareq.timeout;
1832 wdc_c.r_command = wi->wi_atareq.command;
1833 wdc_c.r_head = wi->wi_atareq.head & 0x0f;
1834 wdc_c.r_cyl = wi->wi_atareq.cylinder;
1835 wdc_c.r_sector = wi->wi_atareq.sec_num;
1836 wdc_c.r_count = wi->wi_atareq.sec_count;
1837 wdc_c.r_features = wi->wi_atareq.features;
1838 wdc_c.r_st_bmask = WDCS_DRDY;
1839 wdc_c.r_st_pmask = WDCS_DRDY;
1840 wdc_c.data = wi->wi_bp.b_data;
1841 wdc_c.bcount = wi->wi_bp.b_bcount;
1842
1843 if (wi->wi_softc->atabus->ata_exec_command(wi->wi_softc->drvp, &wdc_c)
1844 != WDC_COMPLETE) {
1845 wi->wi_atareq.retsts = ATACMD_ERROR;
1846 goto bad;
1847 }
1848
1849 if (wdc_c.flags & (AT_ERROR | AT_TIMEOU | AT_DF)) {
1850 if (wdc_c.flags & AT_ERROR) {
1851 wi->wi_atareq.retsts = ATACMD_ERROR;
1852 wi->wi_atareq.error = wdc_c.r_error;
1853 } else if (wdc_c.flags & AT_DF)
1854 wi->wi_atareq.retsts = ATACMD_DF;
1855 else
1856 wi->wi_atareq.retsts = ATACMD_TIMEOUT;
1857 } else {
1858 wi->wi_atareq.retsts = ATACMD_OK;
1859 if (wi->wi_atareq.flags & ATACMD_READREG) {
1860 wi->wi_atareq.head = wdc_c.r_head ;
1861 wi->wi_atareq.cylinder = wdc_c.r_cyl;
1862 wi->wi_atareq.sec_num = wdc_c.r_sector;
1863 wi->wi_atareq.sec_count = wdc_c.r_count;
1864 wi->wi_atareq.features = wdc_c.r_features;
1865 wi->wi_atareq.error = wdc_c.r_error;
1866 }
1867 }
1868
1869 bp->b_error = 0;
1870 biodone(bp);
1871 return;
1872 bad:
1873 bp->b_flags |= B_ERROR;
1874 bp->b_error = error;
1875 biodone(bp);
1876 }
1877