sys/dev/disk/nata/ata-all.c

   1 /*-
   2  * Copyright (c) 1998 - 2008 Søren Schmidt <sos@FreeBSD.org>
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer,
  10  *    without modification, immediately at the beginning of the file.
  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  *
  15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  18  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25  *
  26  * $FreeBSD: src/sys/dev/ata/ata-all.c,v 1.279 2007/02/23 16:25:08 jhb Exp $
  27  */
  28
  29 #include "opt_ata.h"
  30
  31 #include <sys/param.h>
  32 #include <sys/bus.h>
  33 #include <sys/callout.h>
  34 #include <sys/conf.h>
  35 #include <sys/ctype.h>
  36 #include <sys/device.h>
  37 #include <sys/endian.h>
  38 #include <sys/kernel.h>
  39 #include <sys/libkern.h>
  40 #include <sys/lock.h>           /* for {get,rel}_mplock() */
  41 #include <sys/malloc.h>
  42 #include <sys/module.h>
  43 #include <sys/nata.h>
  44 #include <sys/objcache.h>
  45 #include <sys/queue.h>
  46 #include <sys/sysctl.h>
  47 #include <sys/systm.h>
  48
  49 #include <sys/mplock2.h>
  50
  51 #include "ata-all.h"
  52 #include "ata_if.h"
  53
  54 /* device structure */
  55 static  d_ioctl_t       ata_ioctl;
  56 static struct dev_ops ata_ops = {
  57         { "ata", 159, 0 },
  58         .d_open =       nullopen,
  59         .d_close =      nullclose,
  60         .d_ioctl =      ata_ioctl,
  61 };
  62
  63 /* prototypes */
  64 #if 0
  65 static void ata_boot_attach(void);
  66 #endif
  67 static device_t ata_add_child(device_t, struct ata_device *, int);
  68 static int ata_getparam(struct ata_device *, int);
  69 static void bswap(int8_t *, int);
  70 static void btrim(int8_t *, int);
  71 static void bpack(int8_t *, int8_t *, int);
  72
  73 /* global vars */
  74 MALLOC_DEFINE(M_ATA, "ata_generic", "ATA driver generic layer");
  75 int (*ata_raid_ioctl_func)(u_long cmd, caddr_t data) = NULL;
  76 devclass_t ata_devclass;
  77 struct objcache *ata_request_cache;
  78 struct objcache *ata_composite_cache;
  79 struct objcache_malloc_args ata_request_malloc_args = {
  80         sizeof(struct ata_request), M_ATA };
  81 struct objcache_malloc_args ata_composite_malloc_args = {
  82         sizeof(struct ata_composite), M_ATA };
  83 int ata_wc = 1;
  84
  85 /* local vars */
  86 static int ata_dma = 1;
  87 static int atapi_dma = 1;
  88
  89 /* sysctl vars */
  90 SYSCTL_NODE(_hw, OID_AUTO, ata, CTLFLAG_RD, 0, "ATA driver parameters");
  91 TUNABLE_INT("hw.ata.ata_dma", &ata_dma);
  92 SYSCTL_INT(_hw_ata, OID_AUTO, ata_dma, CTLFLAG_RW, &ata_dma, 0,
  93            "ATA disk DMA mode control");
  94 TUNABLE_INT("hw.ata.atapi_dma", &atapi_dma);
  95 SYSCTL_INT(_hw_ata, OID_AUTO, atapi_dma, CTLFLAG_RW, &atapi_dma, 0,
  96            "ATAPI device DMA mode control");
  97 TUNABLE_INT("hw.ata.wc", &ata_wc);
  98 SYSCTL_INT(_hw_ata, OID_AUTO, ata_wc, CTLFLAG_RW, &ata_wc, 0,
  99            "ATA disk write caching");
 100
 101 /*
 102  * newbus device interface related functions
 103  */
 104 int
 105 ata_probe(device_t dev)
 106 {
 107     return 0;
 108 }
 109
 110 int
 111 ata_attach(device_t dev)
 112 {
 113     struct ata_channel *ch = device_get_softc(dev);
 114     int error, rid;
 115
 116     /* check that we have a virgin channel to attach */
 117     if (ch->r_irq)
 118         return EEXIST;
 119
 120     /* initialize the softc basics */
 121     ch->dev = dev;
 122     ch->state = ATA_IDLE;
 123     lockinit(&ch->state_mtx, "ataattach_state", 0, 0);
 124     lockinit(&ch->queue_mtx, "ataattach_queue", 0, 0);
 125     ata_queue_init(ch);
 126
 127     /* reset the controller HW, the channel and device(s) */
 128     while (ATA_LOCKING(dev, ATA_LF_LOCK) != ch->unit)
 129         tsleep(&error, 0, "ataatch", 1);
 130     ATA_RESET(dev);
 131     ATA_LOCKING(dev, ATA_LF_UNLOCK);
 132
 133     /* setup interrupt delivery */
 134     rid = ATA_IRQ_RID;
 135     ch->r_irq = bus_alloc_resource_any(dev, SYS_RES_IRQ, &rid,
 136                                        RF_SHAREABLE | RF_ACTIVE);
 137     if (!ch->r_irq) {
 138         device_printf(dev, "unable to allocate interrupt\n");
 139         return ENXIO;
 140     }
 141     if ((error = bus_setup_intr(dev, ch->r_irq, ATA_INTR_FLAGS,
 142                                 (driver_intr_t *)ata_interrupt, ch, &ch->ih,
 143                                 NULL))) {
 144         device_printf(dev, "unable to setup interrupt\n");
 145         return error;
 146     }
 147
 148     /* probe and attach devices on this channel unless we are in early boot */
 149     ata_identify(dev);
 150     return 0;
 151 }
 152
 153 int
 154 ata_detach(device_t dev)
 155 {
 156     struct ata_channel *ch = device_get_softc(dev);
 157     device_t *children;
 158     int nchildren, i;
 159
 160     /* check that we have a valid channel to detach */
 161     if (!ch->r_irq)
 162         return ENXIO;
 163
 164     /* grap the channel lock so no new requests gets launched */
 165     lockmgr(&ch->state_mtx, LK_EXCLUSIVE);
 166     ch->state |= ATA_STALL_QUEUE;
 167     lockmgr(&ch->state_mtx, LK_RELEASE);
 168
 169     /* detach & delete all children */
 170     if (!device_get_children(dev, &children, &nchildren)) {
 171         for (i = 0; i < nchildren; i++)
 172             if (children[i])
 173                 device_delete_child(dev, children[i]);
 174         kfree(children, M_TEMP);
 175     }
 176
 177     /* release resources */
 178     bus_teardown_intr(dev, ch->r_irq, ch->ih);
 179     bus_release_resource(dev, SYS_RES_IRQ, ATA_IRQ_RID, ch->r_irq);
 180     ch->r_irq = NULL;
 181     lockuninit(&ch->state_mtx);
 182     lockuninit(&ch->queue_mtx);
 183     return 0;
 184 }
 185
 186 int
 187 ata_reinit(device_t dev)
 188 {
 189     struct ata_channel *ch = device_get_softc(dev);
 190     struct ata_request *request;
 191     device_t *children;
 192     int nchildren, i;
 193
 194     /* check that we have a valid channel to reinit */
 195     if (!ch || !ch->r_irq)
 196         return ENXIO;
 197
 198     if (bootverbose)
 199         device_printf(dev, "reiniting channel ..\n");
 200
 201     /* poll for locking the channel */
 202     while (ATA_LOCKING(dev, ATA_LF_LOCK) != ch->unit)
 203         tsleep(&dev, 0, "atarini", 1);
 204
 205     /* catch eventual request in ch->running */
 206     lockmgr(&ch->state_mtx, LK_EXCLUSIVE);
 207     if ((request = ch->running))
 208         callout_cancel(&request->callout);
 209     ch->running = NULL;
 210
 211     /* unconditionally grap the channel lock */
 212     ch->state |= ATA_STALL_QUEUE;
 213     lockmgr(&ch->state_mtx, LK_RELEASE);
 214
 215     /* reset the controller HW, the channel and device(s) */
 216     ATA_RESET(dev);
 217
 218     /* reinit the children and delete any that fails */
 219     if (!device_get_children(dev, &children, &nchildren)) {
 220         get_mplock();
 221         for (i = 0; i < nchildren; i++) {
 222             /* did any children go missing ? */
 223             if (children[i] && device_is_attached(children[i]) &&
 224                 ATA_REINIT(children[i])) {
 225                 /*
 226                  * if we had a running request and its device matches
 227                  * this child we need to inform the request that the
 228                  * device is gone.
 229                  */
 230                 if (request && request->dev == children[i]) {
 231                     request->result = ENXIO;
 232                     device_printf(request->dev, "FAILURE - device detached\n");
 233
 234                     /* if not timeout finish request here */
 235                     if (!(request->flags & ATA_R_TIMEOUT))
 236                             ata_finish(request);
 237                     request = NULL;
 238                 }
 239                 device_delete_child(dev, children[i]);
 240             }
 241         }
 242         kfree(children, M_TEMP);
 243         rel_mplock();
 244     }
 245
 246     /* if we still have a good request put it on the queue again */
 247     if (request && !(request->flags & ATA_R_TIMEOUT)) {
 248         device_printf(request->dev,
 249                       "WARNING - %s requeued due to channel reset",
 250                       ata_cmd2str(request));
 251         if (!(request->flags & (ATA_R_ATAPI | ATA_R_CONTROL)))
 252             kprintf(" LBA=%ju", request->u.ata.lba);
 253         kprintf("\n");
 254         request->flags |= ATA_R_REQUEUE;
 255         ata_queue_request(request);
 256     }
 257
 258     /* we're done release the channel for new work */
 259     lockmgr(&ch->state_mtx, LK_EXCLUSIVE);
 260     ch->state = ATA_IDLE;
 261     lockmgr(&ch->state_mtx, LK_RELEASE);
 262     ATA_LOCKING(dev, ATA_LF_UNLOCK);
 263
 264     if (bootverbose)
 265         device_printf(dev, "reinit done ..\n");
 266
 267     /* kick off requests on the queue */
 268     ata_start(dev);
 269     return 0;
 270 }
 271
 272 int
 273 ata_suspend(device_t dev)
 274 {
 275     struct ata_channel *ch;
 276
 277     /* check for valid device */
 278     if (!dev || !(ch = device_get_softc(dev)))
 279         return ENXIO;
 280
 281     /* wait for the channel to be IDLE or detached before suspending */
 282     while (ch->r_irq) {
 283         lockmgr(&ch->state_mtx, LK_EXCLUSIVE);
 284         if (ch->state == ATA_IDLE) {
 285             ch->state = ATA_ACTIVE;
 286             lockmgr(&ch->state_mtx, LK_RELEASE);
 287             break;
 288         }
 289         lockmgr(&ch->state_mtx, LK_RELEASE);
 290         tsleep(ch, 0, "atasusp", hz/10);
 291     }
 292     ATA_LOCKING(dev, ATA_LF_UNLOCK);
 293     return 0;
 294 }
 295
 296 int
 297 ata_resume(device_t dev)
 298 {
 299     int error;
 300
 301     /* check for valid device */
 302     if (!dev || !device_get_softc(dev))
 303         return ENXIO;
 304
 305     /* reinit the devices, we dont know what mode/state they are in */
 306     error = ata_reinit(dev);
 307
 308     /* kick off requests on the queue */
 309     ata_start(dev);
 310     return error;
 311 }
 312
 313 int
 314 ata_interrupt(void *data)
 315 {
 316     struct ata_channel *ch = (struct ata_channel *)data;
 317     struct ata_request *request;
 318
 319     lockmgr(&ch->state_mtx, LK_EXCLUSIVE);
 320     do {
 321         /*
 322          * Ignore interrupt if its not for us.  This may also have the
 323          * side effect of processing events unrelated to I/O requests.
 324          */
 325         if (ch->hw.status && !ch->hw.status(ch->dev))
 326             break;
 327
 328         /*
 329          * Check if we have a running request, and make sure it has been
 330          * completely queued.  Otherwise the channel status may indicate
 331          * not-busy when, in fact, the command had not yet been issued.
 332          */
 333         if ((request = ch->running) == NULL)
 334             break;
 335         if ((request->flags & ATA_R_HWCMDQUEUED) == 0) {
 336             kprintf("ata_interrupt: early interrupt\n");
 337             break;
 338         }
 339
 340         ATA_DEBUG_RQ(request, "interrupt");
 341
 342         /* safetycheck for the right state */
 343         if (ch->state == ATA_IDLE) {
 344             device_printf(request->dev, "interrupt on idle channel ignored\n");
 345             break;
 346         }
 347
 348         /*
 349          * we have the HW locks, so end the transaction for this request
 350          * if it finishes immediately otherwise wait for next interrupt
 351          */
 352         if (ch->hw.end_transaction(request) == ATA_OP_FINISHED) {
 353             ch->running = NULL;
 354             if (ch->state == ATA_ACTIVE)
 355                 ch->state = ATA_IDLE;
 356             lockmgr(&ch->state_mtx, LK_RELEASE);
 357             ATA_LOCKING(ch->dev, ATA_LF_UNLOCK);
 358             ata_finish(request);
 359             return 1;
 360         }
 361     } while (0);
 362     lockmgr(&ch->state_mtx, LK_RELEASE);
 363     return 0;
 364 }
 365
 366 /*
 367  * device related interfaces
 368  */
 369 static int
 370 ata_ioctl(struct dev_ioctl_args *ap)
 371 {
 372     device_t device, *children;
 373     struct ata_ioc_devices *devices = (struct ata_ioc_devices *)ap->a_data;
 374     int *value = (int *)ap->a_data;
 375     int i, nchildren, error = ENOTTY;
 376
 377     switch (ap->a_cmd) {
 378     case IOCATAGMAXCHANNEL:
 379         /* In case we have channel 0..n this will return n+1. */
 380         *value = devclass_get_maxunit(ata_devclass);
 381         error = 0;
 382         break;
 383
 384     case IOCATAREINIT:
 385         if (*value >= devclass_get_maxunit(ata_devclass) ||
 386             !(device = devclass_get_device(ata_devclass, *value)))
 387             return ENXIO;
 388         error = ata_reinit(device);
 389         ata_start(device);
 390         break;
 391
 392     case IOCATAATTACH:
 393         if (*value >= devclass_get_maxunit(ata_devclass) ||
 394             !(device = devclass_get_device(ata_devclass, *value)))
 395             return ENXIO;
 396         /* XXX SOS should enable channel HW on controller */
 397         error = ata_attach(device);
 398         break;
 399
 400     case IOCATADETACH:
 401         if (*value >= devclass_get_maxunit(ata_devclass) ||
 402             !(device = devclass_get_device(ata_devclass, *value)))
 403             return ENXIO;
 404         error = ata_detach(device);
 405         /* XXX SOS should disable channel HW on controller */
 406         break;
 407
 408     case IOCATADEVICES:
 409         if (devices->channel >= devclass_get_maxunit(ata_devclass) ||
 410             !(device = devclass_get_device(ata_devclass, devices->channel)))
 411             return ENXIO;
 412         bzero(devices->name[0], 32);
 413         bzero(&devices->params[0], sizeof(struct ata_params));
 414         bzero(devices->name[1], 32);
 415         bzero(&devices->params[1], sizeof(struct ata_params));
 416         if (!device_get_children(device, &children, &nchildren)) {
 417             for (i = 0; i < nchildren; i++) {
 418                 if (children[i] && device_is_attached(children[i])) {
 419                     struct ata_device *atadev = device_get_softc(children[i]);
 420
 421                     if (atadev->unit == ATA_MASTER) {
 422                         strncpy(devices->name[0],
 423                                 device_get_nameunit(children[i]), 32);
 424                         bcopy(&atadev->param, &devices->params[0],
 425                               sizeof(struct ata_params));
 426                     }
 427                     if (atadev->unit == ATA_SLAVE) {
 428                         strncpy(devices->name[1],
 429                                 device_get_nameunit(children[i]), 32);
 430                         bcopy(&atadev->param, &devices->params[1],
 431                               sizeof(struct ata_params));
 432                     }
 433                 }
 434             }
 435             kfree(children, M_TEMP);
 436             error = 0;
 437         }
 438         else
 439             error = ENODEV;
 440         break;
 441
 442     default:
 443         if (ata_raid_ioctl_func)
 444             error = ata_raid_ioctl_func(ap->a_cmd, ap->a_data);
 445     }
 446     return error;
 447 }
 448
 449 int
 450 ata_device_ioctl(device_t dev, u_long cmd, caddr_t data)
 451 {
 452     struct ata_device *atadev = device_get_softc(dev);
 453     struct ata_ioc_request *ioc_request = (struct ata_ioc_request *)data;
 454     struct ata_params *params = (struct ata_params *)data;
 455     int *mode = (int *)data;
 456     struct ata_request *request;
 457     caddr_t buf;
 458     int error;
 459
 460     switch (cmd) {
 461     case IOCATAREQUEST:
 462         if (!(buf = kmalloc(ioc_request->count, M_ATA, M_WAITOK | M_NULLOK))) {
 463             return ENOMEM;
 464         }
 465         if (!(request = ata_alloc_request())) {
 466             kfree(buf, M_ATA);
 467             return  ENOMEM;
 468         }
 469         request->dev = atadev->dev;
 470         if (ioc_request->flags & ATA_CMD_WRITE) {
 471             error = copyin(ioc_request->data, buf, ioc_request->count);
 472             if (error) {
 473                 kfree(buf, M_ATA);
 474                 ata_free_request(request);
 475                 return error;
 476             }
 477         }
 478         if (ioc_request->flags & ATA_CMD_ATAPI) {
 479             request->flags = ATA_R_ATAPI;
 480             bcopy(ioc_request->u.atapi.ccb, request->u.atapi.ccb, 16);
 481         }
 482         else {
 483             request->u.ata.command = ioc_request->u.ata.command;
 484             request->u.ata.feature = ioc_request->u.ata.feature;
 485             request->u.ata.lba = ioc_request->u.ata.lba;
 486             request->u.ata.count = ioc_request->u.ata.count;
 487         }
 488         request->timeout = ioc_request->timeout;
 489         request->data = buf;
 490         request->bytecount = ioc_request->count;
 491         request->transfersize = request->bytecount;
 492         if (ioc_request->flags & ATA_CMD_CONTROL)
 493             request->flags |= ATA_R_CONTROL;
 494         if (ioc_request->flags & ATA_CMD_READ)
 495             request->flags |= ATA_R_READ;
 496         if (ioc_request->flags & ATA_CMD_WRITE)
 497             request->flags |= ATA_R_WRITE;
 498         ata_queue_request(request);
 499         if (request->flags & ATA_R_ATAPI) {
 500             bcopy(&request->u.atapi.sense, &ioc_request->u.atapi.sense,
 501                   sizeof(struct atapi_sense));
 502         }
 503         else {
 504             ioc_request->u.ata.command = request->u.ata.command;
 505             ioc_request->u.ata.feature = request->u.ata.feature;
 506             ioc_request->u.ata.lba = request->u.ata.lba;
 507             ioc_request->u.ata.count = request->u.ata.count;
 508         }
 509         ioc_request->error = request->result;
 510         if (ioc_request->flags & ATA_CMD_READ)
 511             error = copyout(buf, ioc_request->data, ioc_request->count);
 512         else
 513             error = 0;
 514         kfree(buf, M_ATA);
 515         ata_free_request(request);
 516         return error;
 517
 518     case IOCATAGPARM:
 519         ata_getparam(atadev, 0);
 520         bcopy(&atadev->param, params, sizeof(struct ata_params));
 521         return 0;
 522
 523     case IOCATASMODE:
 524         atadev->mode = *mode;
 525         ATA_SETMODE(device_get_parent(dev), dev);
 526         return 0;
 527
 528     case IOCATAGMODE:
 529         *mode = atadev->mode;
 530         return 0;
 531     case IOCATASSPINDOWN:
 532         atadev->spindown = *mode;
 533         return 0;
 534     case IOCATAGSPINDOWN:
 535         *mode = atadev->spindown;
 536         return 0;
 537     default:
 538         return ENOTTY;
 539     }
 540 }
 541
 542 #if 0
 543
 544 static void
 545 ata_boot_attach(void)
 546 {
 547     struct ata_channel *ch;
 548     int ctlr;
 549
 550     get_mplock();
 551
 552     /* kick of probe and attach on all channels */
 553     for (ctlr = 0; ctlr < devclass_get_maxunit(ata_devclass); ctlr++) {
 554         if ((ch = devclass_get_softc(ata_devclass, ctlr))) {
 555             ata_identify(ch->dev);
 556         }
 557     }
 558
 559     rel_mplock();
 560 }
 561
 562 #endif
 563
 564 void
 565 ata_print_cable(device_t dev, u_int8_t *who)
 566 {
 567     device_printf(dev,
 568                   "DMA limited to UDMA33, %s found non-ATA66 cable\n", who);
 569 }
 570
 571 /*
 572  * misc support functions
 573  */
 574 static device_t
 575 ata_add_child(device_t parent, struct ata_device *atadev, int unit)
 576 {
 577     device_t child;
 578
 579     if ((child = device_add_child(parent, NULL, unit))) {
 580         device_set_softc(child, atadev);
 581         device_quiet(child);
 582         atadev->dev = child;
 583         atadev->max_iosize = DEV_BSIZE;
 584         atadev->mode = ATA_PIO_MAX;
 585     }
 586     return child;
 587 }
 588
 589 static int
 590 ata_getparam(struct ata_device *atadev, int init)
 591 {
 592     struct ata_channel *ch = device_get_softc(device_get_parent(atadev->dev));
 593     struct ata_request *request;
 594     u_int8_t command = 0;
 595     int error = ENOMEM, retries = 2;
 596
 597     if (ch->devices & (ATA_ATA_MASTER << atadev->unit))
 598         command = ATA_ATA_IDENTIFY;
 599     if (ch->devices & (ATA_ATAPI_MASTER << atadev->unit))
 600         command = ATA_ATAPI_IDENTIFY;
 601     if (!command)
 602         return ENXIO;
 603
 604     while (retries-- > 0 && error) {
 605         if (!(request = ata_alloc_request()))
 606             break;
 607         request->dev = atadev->dev;
 608         request->timeout = 1;
 609         request->retries = 0;
 610         request->u.ata.command = command;
 611         request->flags = (ATA_R_READ|ATA_R_AT_HEAD|ATA_R_DIRECT|ATA_R_QUIET);
 612         request->data = (void *)&atadev->param;
 613         request->bytecount = sizeof(struct ata_params);
 614         request->donecount = 0;
 615         request->transfersize = DEV_BSIZE;
 616         ata_queue_request(request);
 617         error = request->result;
 618         ata_free_request(request);
 619     }
 620
 621     if (!error && (isprint(atadev->param.model[0]) ||
 622                    isprint(atadev->param.model[1]))) {
 623         struct ata_params *atacap = &atadev->param;
 624         int16_t *ptr;
 625
 626         for (ptr = (int16_t *)atacap;
 627              ptr < (int16_t *)atacap + sizeof(struct ata_params)/2; ptr++) {
 628             *ptr = le16toh(*ptr);
 629         }
 630         if (!(!strncmp(atacap->model, "FX", 2) ||
 631               !strncmp(atacap->model, "NEC", 3) ||
 632               !strncmp(atacap->model, "Pioneer", 7) ||
 633               !strncmp(atacap->model, "SHARP", 5))) {
 634             bswap(atacap->model, sizeof(atacap->model));
 635             bswap(atacap->revision, sizeof(atacap->revision));
 636             bswap(atacap->serial, sizeof(atacap->serial));
 637         }
 638         btrim(atacap->model, sizeof(atacap->model));
 639         bpack(atacap->model, atacap->model, sizeof(atacap->model));
 640         btrim(atacap->revision, sizeof(atacap->revision));
 641         bpack(atacap->revision, atacap->revision, sizeof(atacap->revision));
 642         btrim(atacap->serial, sizeof(atacap->serial));
 643         bpack(atacap->serial, atacap->serial, sizeof(atacap->serial));
 644
 645         if (bootverbose)
 646             kprintf("ata%d-%s: pio=%s wdma=%s udma=%s cable=%s wire\n",
 647                    device_get_unit(ch->dev),
 648                    ata_unit2str(atadev),
 649                    ata_mode2str(ata_pmode(atacap)),
 650                    ata_mode2str(ata_wmode(atacap)),
 651                    ata_mode2str(ata_umode(atacap)),
 652                    (atacap->hwres & ATA_CABLE_ID) ? "80":"40");
 653
 654         if (init) {
 655             char buffer[64];
 656
 657             ksprintf(buffer, "%.40s/%.8s", atacap->model, atacap->revision);
 658             device_set_desc_copy(atadev->dev, buffer);
 659             if ((atadev->param.config & ATA_PROTO_ATAPI) &&
 660                 (atadev->param.config != ATA_CFA_MAGIC1) &&
 661                 (atadev->param.config != ATA_CFA_MAGIC2)) {
 662                 if (atapi_dma && ch->dma &&
 663                     (atadev->param.config & ATA_DRQ_MASK) != ATA_DRQ_INTR &&
 664                     ata_umode(&atadev->param) >= ATA_UDMA2)
 665                     atadev->mode = ATA_DMA_MAX;
 666             }
 667             else {
 668                 if (ata_dma && ch->dma &&
 669                     (ata_umode(&atadev->param) > 0 ||
 670                      ata_wmode(&atadev->param) > 0))
 671                     atadev->mode = ATA_DMA_MAX;
 672             }
 673         }
 674     }
 675     else {
 676         if (!error)
 677             error = ENXIO;
 678     }
 679     return error;
 680 }
 681
 682 int
 683 ata_identify(device_t dev)
 684 {
 685     struct ata_channel *ch = device_get_softc(dev);
 686     struct ata_device *atadev;
 687     device_t child;
 688     int i;
 689
 690     if (bootverbose)
 691         device_printf(dev, "identify ch->devices=%08x\n", ch->devices);
 692
 693     for (i = 0; i < ATA_PM; ++i) {
 694         if (ch->devices & (((ATA_ATA_MASTER | ATA_ATAPI_MASTER) << i))) {
 695             int unit = -1;
 696
 697             if (!(atadev = kmalloc(sizeof(struct ata_device),
 698                                    M_ATA, M_INTWAIT | M_ZERO))) {
 699                 device_printf(dev, "out of memory\n");
 700                 return ENOMEM;
 701             }
 702             atadev->unit = i;
 703 #ifdef ATA_STATIC_ID
 704             if (ch->devices & ((ATA_ATA_MASTER << i)))
 705                 unit = (device_get_unit(dev) << 1) + i;
 706 #endif
 707             if ((child = ata_add_child(dev, atadev, unit))) {
 708                 if (ata_getparam(atadev, 1)) {
 709                     device_delete_child(dev, child);
 710                     kfree(atadev, M_ATA);
 711                 }
 712             }
 713             else
 714                 kfree(atadev, M_ATA);
 715         }
 716     }
 717     bus_generic_probe(dev);
 718     bus_generic_attach(dev);
 719     return 0;
 720 }
 721
 722 void
 723 ata_default_registers(device_t dev)
 724 {
 725     struct ata_channel *ch = device_get_softc(dev);
 726
 727     /* fill in the defaults from whats setup already */
 728     ch->r_io[ATA_ERROR].res = ch->r_io[ATA_FEATURE].res;
 729     ch->r_io[ATA_ERROR].offset = ch->r_io[ATA_FEATURE].offset;
 730     ch->r_io[ATA_IREASON].res = ch->r_io[ATA_COUNT].res;
 731     ch->r_io[ATA_IREASON].offset = ch->r_io[ATA_COUNT].offset;
 732     ch->r_io[ATA_STATUS].res = ch->r_io[ATA_COMMAND].res;
 733     ch->r_io[ATA_STATUS].offset = ch->r_io[ATA_COMMAND].offset;
 734     ch->r_io[ATA_ALTSTAT].res = ch->r_io[ATA_CONTROL].res;
 735     ch->r_io[ATA_ALTSTAT].offset = ch->r_io[ATA_CONTROL].offset;
 736 }
 737
 738 void
 739 ata_modify_if_48bit(struct ata_request *request)
 740 {
 741     struct ata_channel *ch = device_get_softc(device_get_parent(request->dev));
 742     struct ata_device *atadev = device_get_softc(request->dev);
 743
 744     atadev->flags &= ~ATA_D_48BIT_ACTIVE;
 745
 746     if (((request->u.ata.lba + request->u.ata.count) >= ATA_MAX_28BIT_LBA ||
 747          request->u.ata.count > 256) &&
 748         atadev->param.support.command2 & ATA_SUPPORT_ADDRESS48) {
 749
 750         /* translate command into 48bit version */
 751         switch (request->u.ata.command) {
 752         case ATA_READ:
 753             request->u.ata.command = ATA_READ48;
 754             break;
 755         case ATA_READ_MUL:
 756             request->u.ata.command = ATA_READ_MUL48;
 757             break;
 758         case ATA_READ_DMA:
 759             if (ch->flags & ATA_NO_48BIT_DMA) {
 760                 if (request->transfersize > DEV_BSIZE)
 761                     request->u.ata.command = ATA_READ_MUL48;
 762                 else
 763                     request->u.ata.command = ATA_READ48;
 764                 request->flags &= ~ATA_R_DMA;
 765             }
 766             else
 767                 request->u.ata.command = ATA_READ_DMA48;
 768             break;
 769         case ATA_READ_DMA_QUEUED:
 770             if (ch->flags & ATA_NO_48BIT_DMA) {
 771                 if (request->transfersize > DEV_BSIZE)
 772                     request->u.ata.command = ATA_READ_MUL48;
 773                 else
 774                     request->u.ata.command = ATA_READ48;
 775                 request->flags &= ~ATA_R_DMA;
 776             }
 777             else
 778                 request->u.ata.command = ATA_READ_DMA_QUEUED48;
 779             break;
 780         case ATA_WRITE:
 781             request->u.ata.command = ATA_WRITE48;
 782             break;
 783         case ATA_WRITE_MUL:
 784             request->u.ata.command = ATA_WRITE_MUL48;
 785             break;
 786         case ATA_WRITE_DMA:
 787             if (ch->flags & ATA_NO_48BIT_DMA) {
 788                 if (request->transfersize > DEV_BSIZE)
 789                     request->u.ata.command = ATA_WRITE_MUL48;
 790                 else
 791                     request->u.ata.command = ATA_WRITE48;
 792                 request->flags &= ~ATA_R_DMA;
 793             }
 794             else
 795                 request->u.ata.command = ATA_WRITE_DMA48;
 796             break;
 797         case ATA_WRITE_DMA_QUEUED:
 798             if (ch->flags & ATA_NO_48BIT_DMA) {
 799                 if (request->transfersize > DEV_BSIZE)
 800                     request->u.ata.command = ATA_WRITE_MUL48;
 801                 else
 802                     request->u.ata.command = ATA_WRITE48;
 803                 request->u.ata.command = ATA_WRITE48;
 804                 request->flags &= ~ATA_R_DMA;
 805             }
 806             else
 807                 request->u.ata.command = ATA_WRITE_DMA_QUEUED48;
 808             break;
 809         case ATA_FLUSHCACHE:
 810             request->u.ata.command = ATA_FLUSHCACHE48;
 811             break;
 812         case ATA_READ_NATIVE_MAX_ADDRESS:
 813             request->u.ata.command = ATA_READ_NATIVE_MAX_ADDRESS48;
 814             break;
 815         case ATA_SET_MAX_ADDRESS:
 816             request->u.ata.command = ATA_SET_MAX_ADDRESS48;
 817             break;
 818         default:
 819             return;
 820         }
 821         atadev->flags |= ATA_D_48BIT_ACTIVE;
 822     }
 823 }
 824
 825 void
 826 ata_udelay(int interval)
 827 {
 828     /*
 829      * We can't use tsleep here, because we might be called from callout
 830      * context.
 831      */
 832     if (1 || interval < (1000000/hz))
 833         DELAY(interval);
 834     else
 835         tsleep(&interval, 0, "ataslp", 1 + interval / (1000000 / hz));
 836 }
 837
 838 const char *
 839 ata_unit2str(struct ata_device *atadev)
 840 {
 841     static char str[8];
 842
 843     ksprintf(str, "%s", atadev->unit == ATA_MASTER ? "master" : "slave");
 844     return str;
 845 }
 846
 847 const char *
 848 ata_mode2str(int mode)
 849 {
 850     switch (mode) {
 851     case -1: return "UNSUPPORTED";
 852     case ATA_PIO0: return "PIO0";
 853     case ATA_PIO1: return "PIO1";
 854     case ATA_PIO2: return "PIO2";
 855     case ATA_PIO3: return "PIO3";
 856     case ATA_PIO4: return "PIO4";
 857     case ATA_WDMA0: return "WDMA0";
 858     case ATA_WDMA1: return "WDMA1";
 859     case ATA_WDMA2: return "WDMA2";
 860     case ATA_UDMA0: return "UDMA16";
 861     case ATA_UDMA1: return "UDMA25";
 862     case ATA_UDMA2: return "UDMA33";
 863     case ATA_UDMA3: return "UDMA40";
 864     case ATA_UDMA4: return "UDMA66";
 865     case ATA_UDMA5: return "UDMA100";
 866     case ATA_UDMA6: return "UDMA133";
 867     case ATA_SA150: return "SATA150";
 868     case ATA_SA300: return "SATA300";
 869     case ATA_USB: return "USB";
 870     case ATA_USB1: return "USB1";
 871     case ATA_USB2: return "USB2";
 872     default:
 873         if (mode & ATA_DMA_MASK)
 874             return "BIOSDMA";
 875         else
 876             return "BIOSPIO";
 877     }
 878 }
 879
 880 int
 881 ata_atapi(device_t dev)
 882 {
 883     struct ata_channel *ch = device_get_softc(device_get_parent(dev));
 884     struct ata_device *atadev = device_get_softc(dev);
 885
 886     return ((atadev->unit == ATA_MASTER && ch->devices & ATA_ATAPI_MASTER) ||
 887             (atadev->unit == ATA_SLAVE && ch->devices & ATA_ATAPI_SLAVE));
 888 }
 889
 890 int
 891 ata_pmode(struct ata_params *ap)
 892 {
 893     if (ap->atavalid & ATA_FLAG_64_70) {
 894         if (ap->apiomodes & 0x02)
 895             return ATA_PIO4;
 896         if (ap->apiomodes & 0x01)
 897             return ATA_PIO3;
 898     }
 899     if (ap->mwdmamodes & 0x04)
 900         return ATA_PIO4;
 901     if (ap->mwdmamodes & 0x02)
 902         return ATA_PIO3;
 903     if (ap->mwdmamodes & 0x01)
 904         return ATA_PIO2;
 905     if ((ap->retired_piomode & ATA_RETIRED_PIO_MASK) == 0x200)
 906         return ATA_PIO2;
 907     if ((ap->retired_piomode & ATA_RETIRED_PIO_MASK) == 0x100)
 908         return ATA_PIO1;
 909     if ((ap->retired_piomode & ATA_RETIRED_PIO_MASK) == 0x000)
 910         return ATA_PIO0;
 911     return ATA_PIO0;
 912 }
 913
 914 int
 915 ata_wmode(struct ata_params *ap)
 916 {
 917     if (ap->mwdmamodes & 0x04)
 918         return ATA_WDMA2;
 919     if (ap->mwdmamodes & 0x02)
 920         return ATA_WDMA1;
 921     if (ap->mwdmamodes & 0x01)
 922         return ATA_WDMA0;
 923     return -1;
 924 }
 925
 926 int
 927 ata_umode(struct ata_params *ap)
 928 {
 929     if (ap->atavalid & ATA_FLAG_88) {
 930         if (ap->udmamodes & 0x40)
 931             return ATA_UDMA6;
 932         if (ap->udmamodes & 0x20)
 933             return ATA_UDMA5;
 934         if (ap->udmamodes & 0x10)
 935             return ATA_UDMA4;
 936         if (ap->udmamodes & 0x08)
 937             return ATA_UDMA3;
 938         if (ap->udmamodes & 0x04)
 939             return ATA_UDMA2;
 940         if (ap->udmamodes & 0x02)
 941             return ATA_UDMA1;
 942         if (ap->udmamodes & 0x01)
 943             return ATA_UDMA0;
 944     }
 945     return -1;
 946 }
 947
 948 int
 949 ata_limit_mode(device_t dev, int mode, int maxmode)
 950 {
 951     struct ata_device *atadev = device_get_softc(dev);
 952
 953     if (maxmode && mode > maxmode)
 954         mode = maxmode;
 955
 956     if (mode >= ATA_UDMA0 && ata_umode(&atadev->param) > 0)
 957         return min(mode, ata_umode(&atadev->param));
 958
 959     if (mode >= ATA_WDMA0 && ata_wmode(&atadev->param) > 0)
 960         return min(mode, ata_wmode(&atadev->param));
 961
 962     if (mode > ata_pmode(&atadev->param))
 963         return min(mode, ata_pmode(&atadev->param));
 964
 965     return mode;
 966 }
 967
 968 static void
 969 bswap(int8_t *buf, int len)
 970 {
 971     u_int16_t *ptr = (u_int16_t*)(buf + len);
 972
 973     while (--ptr >= (u_int16_t*)buf)
 974         *ptr = ntohs(*ptr);
 975 }
 976
 977 static void
 978 btrim(int8_t *buf, int len)
 979 {
 980     int8_t *ptr;
 981
 982     for (ptr = buf; ptr < buf+len; ++ptr)
 983         if (!*ptr || *ptr == '_')
 984             *ptr = ' ';
 985     for (ptr = buf + len - 1; ptr >= buf && *ptr == ' '; --ptr)
 986         *ptr = 0;
 987 }
 988
 989 static void
 990 bpack(int8_t *src, int8_t *dst, int len)
 991 {
 992     int i, j, blank;
 993
 994     for (i = j = blank = 0 ; i < len; i++) {
 995         if (blank && src[i] == ' ') continue;
 996         if (blank && src[i] != ' ') {
 997             dst[j++] = src[i];
 998             blank = 0;
 999             continue;
1000         }
1001         if (src[i] == ' ') {
1002             blank = 1;
1003             if (i == 0)
1004                 continue;
1005         }
1006         dst[j++] = src[i];
1007     }
1008     if (j < len)
1009         dst[j] = 0x00;
1010 }
1011
1012
1013 /*
1014  * module handeling
1015  */
1016 static int
1017 ata_module_event_handler(module_t mod, int what, void *arg)
1018 {
1019     /* static because we need the reference at destruction time */
1020     static cdev_t atacdev;
1021
1022     switch (what) {
1023     case MOD_LOAD:
1024         /* register controlling device */
1025         atacdev = make_dev(&ata_ops, 0, UID_ROOT, GID_OPERATOR, 0600, "ata");
1026         reference_dev(atacdev);
1027         return 0;
1028
1029     case MOD_UNLOAD:
1030         /* deregister controlling device */
1031         destroy_dev(atacdev);
1032         dev_ops_remove_all(&ata_ops);
1033         return 0;
1034
1035     default:
1036         return EOPNOTSUPP;
1037     }
1038 }
1039
1040 static moduledata_t ata_moduledata = { "ata", ata_module_event_handler, NULL };
1041 DECLARE_MODULE(ata, ata_moduledata, SI_SUB_CONFIGURE, SI_ORDER_SECOND);
1042 MODULE_VERSION(ata, 1);
1043
1044 /*
1045  * Construct a completely zero'ed ata_request. On objcache_put(), an
1046  * ata_request object is also zero'ed, so objcache_get() is guaranteed to give
1047  * completely zero'ed objects without spending too much time.
1048  */
1049 static boolean_t
1050 ata_request_cache_ctor(void *obj, void *private, int ocflags)
1051 {
1052     struct ata_request *arp = obj;
1053
1054     bzero(arp, sizeof(struct ata_request));
1055     return(TRUE);
1056 }
1057
1058 /*
1059  * Construct a completely zero'ed ata_composite. On objcache_put(), an
1060  * ata_composite object is also zero'ed, so objcache_get() is guaranteed to give
1061  * completely zero'ed objects without spending too much time.
1062  */
1063 static boolean_t
1064 ata_composite_cache_ctor(void *obj, void *private, int ocflags)
1065 {
1066     struct ata_composite *acp = obj;
1067
1068     bzero(acp, sizeof(struct ata_composite));
1069     return(TRUE);
1070 }
1071
1072 static void
1073 ata_init(void)
1074 {
1075     ata_request_cache = objcache_create("ata_request", 0, 0,
1076                                         ata_request_cache_ctor, NULL, NULL,
1077                                         objcache_malloc_alloc,
1078                                         objcache_malloc_free,
1079                                         &ata_request_malloc_args);
1080     ata_composite_cache = objcache_create("ata_composite", 0, 0,
1081                                           ata_composite_cache_ctor, NULL, NULL,
1082                                           objcache_malloc_alloc,
1083                                           objcache_malloc_free,
1084                                           &ata_composite_malloc_args);
1085 }
1086 SYSINIT(ata_register, SI_SUB_DRIVERS, SI_ORDER_SECOND, ata_init, NULL);
1087
1088 static void
1089 ata_uninit(void)
1090 {
1091     objcache_destroy(ata_composite_cache);
1092     objcache_destroy(ata_request_cache);
1093 }
1094 SYSUNINIT(ata_unregister, SI_SUB_DRIVERS, SI_ORDER_SECOND, ata_uninit, NULL);