FreeBSD/Linux Kernel Cross Reference
sys/dev/ata/ata-dma.c

     /*-
      * Copyright (c) 1998 - 2004 Søren Schmidt <sos@FreeBSD.org>
      * All rights reserved.
      *
      * Redistribution and use in source and binary forms, with or without
      * modification, are permitted provided that the following conditions
      * are met:
      * 1. Redistributions of source code must retain the above copyright
      *    notice, this list of conditions and the following disclaimer,
     *    without modification, immediately at the beginning of the file.
     * 2. Redistributions in binary form must reproduce the above copyright
     *    notice, this list of conditions and the following disclaimer in the
     *    documentation and/or other materials provided with the distribution.
     * 3. The name of the author may not be used to endorse or promote products
     *    derived from this software without specific prior written permission.
     *
     * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     */
    
    #include <sys/cdefs.h>
    __FBSDID("$FreeBSD$");
    
    #include <sys/param.h>
    #include <sys/systm.h>
    #include <sys/ata.h>
    #include <sys/kernel.h>
    #include <sys/endian.h>
    #include <sys/malloc.h> 
    #include <sys/lock.h>
    #include <sys/sema.h>
    #include <sys/taskqueue.h>
    #include <vm/uma.h>
    #include <sys/bus.h>
    #include <machine/bus.h>
    #include <sys/rman.h>
    #include <dev/pci/pcivar.h>
    #include <dev/ata/ata-all.h>
    #include <dev/ata/ata-pci.h>
    
    /* prototypes */
    static void ata_dmaalloc(struct ata_channel *);
    static void ata_dmafree(struct ata_channel *);
    static void ata_dmasetprd(void *, bus_dma_segment_t *, int, int);
    static int ata_dmaload(struct ata_device *, caddr_t, int32_t, int);
    static int ata_dmaunload(struct ata_channel *);
    
    /* local vars */
    static MALLOC_DEFINE(M_ATADMA, "ATA DMA", "ATA driver DMA");
    
    /* misc defines */
    #define MAXTABSZ        PAGE_SIZE
    #define MAXWSPCSZ       PAGE_SIZE
    
    struct ata_dc_cb_args {
        bus_addr_t maddr;
        int error;
    };
    
    void 
    ata_dmainit(struct ata_channel *ch)
    {
        if ((ch->dma = malloc(sizeof(struct ata_dma), M_ATADMA, M_NOWAIT|M_ZERO))) {
            ch->dma->alloc = ata_dmaalloc;
            ch->dma->free = ata_dmafree;
            ch->dma->setprd = ata_dmasetprd;
            ch->dma->load = ata_dmaload;
            ch->dma->unload = ata_dmaunload;
            ch->dma->alignment = 2;
            ch->dma->max_iosize = 128 * DEV_BSIZE;
            ch->dma->boundary = 128 * DEV_BSIZE;
        }
    }
    
    static void
    ata_dmasetupc_cb(void *xsc, bus_dma_segment_t *segs, int nsegs, int error)
    {
        struct ata_dc_cb_args *cba = (struct ata_dc_cb_args *)xsc;
    
        if (!(cba->error = error))
            cba->maddr = segs[0].ds_addr;
    }
    
    static void
    ata_dmaalloc(struct ata_channel *ch)
    {
        struct ata_dc_cb_args ccba;
    
        if (bus_dma_tag_create(NULL, ch->dma->alignment, 0,
                               BUS_SPACE_MAXADDR_32BIT, BUS_SPACE_MAXADDR,
                               NULL, NULL, 256 * DEV_BSIZE,
                              ATA_DMA_ENTRIES, ch->dma->max_iosize,
                              0, NULL, NULL, &ch->dma->dmatag))
           goto error;
   
       if (bus_dma_tag_create(ch->dma->dmatag, PAGE_SIZE, PAGE_SIZE,
                              BUS_SPACE_MAXADDR_32BIT, BUS_SPACE_MAXADDR,
                              NULL, NULL, MAXTABSZ, 1, MAXTABSZ,
                              0, NULL, NULL, &ch->dma->cdmatag))
           goto error;
   
       if (bus_dma_tag_create(ch->dma->dmatag,ch->dma->alignment,ch->dma->boundary,
                              BUS_SPACE_MAXADDR_32BIT, BUS_SPACE_MAXADDR,
                              NULL, NULL, 256 * DEV_BSIZE,
                              ATA_DMA_ENTRIES, ch->dma->max_iosize,
                              BUS_DMA_ALLOCNOW, NULL, NULL, &ch->dma->ddmatag))
           goto error;
   
       if (bus_dmamem_alloc(ch->dma->cdmatag, (void **)&ch->dma->dmatab, 0,
                            &ch->dma->cdmamap))
           goto error;
   
       if (bus_dmamap_load(ch->dma->cdmatag, ch->dma->cdmamap, ch->dma->dmatab,
                           MAXTABSZ, ata_dmasetupc_cb, &ccba, 0) || ccba.error) {
           bus_dmamem_free(ch->dma->cdmatag, ch->dma->dmatab, ch->dma->cdmamap);
           goto error;
       }
       ch->dma->mdmatab = ccba.maddr;
   
       if (bus_dmamap_create(ch->dma->ddmatag, 0, &ch->dma->ddmamap))
           goto error;
   
       if (bus_dma_tag_create(ch->dma->dmatag, PAGE_SIZE, PAGE_SIZE,
                              BUS_SPACE_MAXADDR_32BIT, BUS_SPACE_MAXADDR,
                              NULL, NULL, MAXWSPCSZ, 1, MAXWSPCSZ,
                              0, NULL, NULL, &ch->dma->wdmatag))
           goto error;
   
       if (bus_dmamem_alloc(ch->dma->wdmatag, (void **)&ch->dma->workspace, 0,
                            &ch->dma->wdmamap))
           goto error;
   
       if (bus_dmamap_load(ch->dma->wdmatag, ch->dma->wdmamap, ch->dma->workspace,
                           MAXWSPCSZ, ata_dmasetupc_cb, &ccba, 0) || ccba.error) {
           bus_dmamem_free(ch->dma->wdmatag, ch->dma->workspace, ch->dma->wdmamap);
           goto error;
       }
       ch->dma->wdmatab = ccba.maddr;
   
       return;
   
   error:
       ata_printf(ch, -1, "WARNING - DMA allocation failed, disabling DMA\n");
       ata_dmafree(ch);
       free(ch->dma, M_ATADMA);
       ch->dma = NULL;
   }
   
   static void
   ata_dmafree(struct ata_channel *ch)
   {
       if (ch->dma->wdmatab) {
           bus_dmamap_unload(ch->dma->wdmatag, ch->dma->wdmamap);
           bus_dmamem_free(ch->dma->wdmatag, ch->dma->workspace, ch->dma->wdmamap);
           ch->dma->wdmatab = 0;
           ch->dma->wdmamap = NULL;
           ch->dma->workspace = NULL;
       }
       if (ch->dma->wdmatag) {
           bus_dma_tag_destroy(ch->dma->wdmatag);
           ch->dma->wdmatag = NULL;
       }
       if (ch->dma->mdmatab) {
           bus_dmamap_unload(ch->dma->cdmatag, ch->dma->cdmamap);
           bus_dmamem_free(ch->dma->cdmatag, ch->dma->dmatab, ch->dma->cdmamap);
           ch->dma->mdmatab = 0;
           ch->dma->cdmamap = NULL;
           ch->dma->dmatab = NULL;
       }
       if (ch->dma->ddmamap) {
           bus_dmamap_destroy(ch->dma->ddmatag, ch->dma->ddmamap);
           ch->dma->ddmamap = NULL;
       }
       if (ch->dma->cdmatag) {
           bus_dma_tag_destroy(ch->dma->cdmatag);
           ch->dma->cdmatag = NULL;
       }
       if (ch->dma->ddmatag) {
           bus_dma_tag_destroy(ch->dma->ddmatag);
           ch->dma->ddmatag = NULL;
       }
       if (ch->dma->dmatag) {
           bus_dma_tag_destroy(ch->dma->dmatag);
           ch->dma->dmatag = NULL;
       }
   }
   
   static void
   ata_dmasetprd(void *xsc, bus_dma_segment_t *segs, int nsegs, int error)
   {
       struct ata_dmasetprd_args *args = xsc;
       struct ata_dma_prdentry *prd = args->dmatab;
       int i;
   
       if ((args->error = error))
           return;
   
       for (i = 0; i < nsegs; i++) {
           prd[i].addr = htole32(segs[i].ds_addr);
           prd[i].count = htole32(segs[i].ds_len);
       }
       prd[i - 1].count |= htole32(ATA_DMA_EOT);
   }
   
   static int
   ata_dmaload(struct ata_device *atadev, caddr_t data, int32_t count, int dir)
   {
       struct ata_channel *ch = atadev->channel;
       struct ata_dmasetprd_args cba;
   
       if (ch->dma->flags & ATA_DMA_LOADED) {
           ata_prtdev(atadev, "FAILURE - already active DMA on this device\n");
           return -1;
       }
       if (!count) {
           ata_prtdev(atadev, "FAILURE - zero length DMA transfer attempted\n");
           return -1;
       }
       if (((uintptr_t)data & (ch->dma->alignment - 1)) ||
           (count & (ch->dma->alignment - 1))) {
           ata_prtdev(atadev, "FAILURE - non aligned DMA transfer attempted\n");
           return -1;
       }
       if (count > ch->dma->max_iosize) {
           ata_prtdev(atadev,
                      "FAILURE - oversized DMA transfer attempted %d > %d\n",
                      count, ch->dma->max_iosize);
           return -1;
       }
   
       cba.dmatab = ch->dma->dmatab;
   
       bus_dmamap_sync(ch->dma->cdmatag, ch->dma->cdmamap, BUS_DMASYNC_PREWRITE);
   
       if (bus_dmamap_load(ch->dma->ddmatag, ch->dma->ddmamap, data, count,
                           ch->dma->setprd, &cba, 0) || cba.error)
           return -1;
   
       bus_dmamap_sync(ch->dma->ddmatag, ch->dma->ddmamap,
                       dir ? BUS_DMASYNC_PREREAD : BUS_DMASYNC_PREWRITE);
   
       ch->dma->cur_iosize = count;
       ch->dma->flags = dir ? (ATA_DMA_LOADED | ATA_DMA_READ) : ATA_DMA_LOADED;
       return 0;
   }
   
   int
   ata_dmaunload(struct ata_channel *ch)
   {
       bus_dmamap_sync(ch->dma->cdmatag, ch->dma->cdmamap, BUS_DMASYNC_POSTWRITE);
   
       bus_dmamap_sync(ch->dma->ddmatag, ch->dma->ddmamap,
                       (ch->dma->flags & ATA_DMA_READ) != 0 ?
                       BUS_DMASYNC_POSTREAD : BUS_DMASYNC_POSTWRITE);
       bus_dmamap_unload(ch->dma->ddmatag, ch->dma->ddmamap);
   
       ch->dma->cur_iosize = 0;
       ch->dma->flags &= ~ATA_DMA_LOADED;
       return 0;
   }

Cache object: 9eaf08bbc7e64e0f1c4eab3b4739da51