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