drivers/dma/ioat_dma.c

   1 /*
   2  * Intel I/OAT DMA Linux driver
   3  * Copyright(c) 2004 - 2007 Intel Corporation.
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms and conditions of the GNU General Public License,
   7  * version 2, as published by the Free Software Foundation.
   8  *
   9  * This program is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  12  * more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along with
  15  * this program; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin St - Fifth Floor, Boston, MA 02110-1301 USA.
  17  *
  18  * The full GNU General Public License is included in this distribution in
  19  * the file called "COPYING".
  20  *
  21  */
  22
  23 /*
  24  * This driver supports an Intel I/OAT DMA engine, which does asynchronous
  25  * copy operations.
  26  */
  27
  28 #include <linux/init.h>
  29 #include <linux/module.h>
  30 #include <linux/pci.h>
  31 #include <linux/interrupt.h>
  32 #include <linux/dmaengine.h>
  33 #include <linux/delay.h>
  34 #include <linux/dma-mapping.h>
  35 #include "ioatdma.h"
  36 #include "ioatdma_registers.h"
  37 #include "ioatdma_hw.h"
  38
  39 #define to_ioat_chan(chan) container_of(chan, struct ioat_dma_chan, common)
  40 #define to_ioatdma_device(dev) container_of(dev, struct ioatdma_device, common)
  41 #define to_ioat_desc(lh) container_of(lh, struct ioat_desc_sw, node)
  42 #define tx_to_ioat_desc(tx) container_of(tx, struct ioat_desc_sw, async_tx)
  43
  44 static int ioat_pending_level = 4;
  45 module_param(ioat_pending_level, int, 0644);
  46 MODULE_PARM_DESC(ioat_pending_level,
  47                  "high-water mark for pushing ioat descriptors (default: 4)");
  48
  49 /* internal functions */
  50 static void ioat_dma_start_null_desc(struct ioat_dma_chan *ioat_chan);
  51 static void ioat_dma_memcpy_cleanup(struct ioat_dma_chan *ioat_chan);
  52
  53 static struct ioat_desc_sw *
  54 ioat1_dma_get_next_descriptor(struct ioat_dma_chan *ioat_chan);
  55 static struct ioat_desc_sw *
  56 ioat2_dma_get_next_descriptor(struct ioat_dma_chan *ioat_chan);
  57
  58 static inline struct ioat_dma_chan *ioat_lookup_chan_by_index(
  59                                                 struct ioatdma_device *device,
  60                                                 int index)
  61 {
  62         return device->idx[index];
  63 }
  64
  65 /**
  66  * ioat_dma_do_interrupt - handler used for single vector interrupt mode
  67  * @irq: interrupt id
  68  * @data: interrupt data
  69  */
  70 static irqreturn_t ioat_dma_do_interrupt(int irq, void *data)
  71 {
  72         struct ioatdma_device *instance = data;
  73         struct ioat_dma_chan *ioat_chan;
  74         unsigned long attnstatus;
  75         int bit;
  76         u8 intrctrl;
  77
  78         intrctrl = readb(instance->reg_base + IOAT_INTRCTRL_OFFSET);
  79
  80         if (!(intrctrl & IOAT_INTRCTRL_MASTER_INT_EN))
  81                 return IRQ_NONE;
  82
  83         if (!(intrctrl & IOAT_INTRCTRL_INT_STATUS)) {
  84                 writeb(intrctrl, instance->reg_base + IOAT_INTRCTRL_OFFSET);
  85                 return IRQ_NONE;
  86         }
  87
  88         attnstatus = readl(instance->reg_base + IOAT_ATTNSTATUS_OFFSET);
  89         for_each_bit(bit, &attnstatus, BITS_PER_LONG) {
  90                 ioat_chan = ioat_lookup_chan_by_index(instance, bit);
  91                 tasklet_schedule(&ioat_chan->cleanup_task);
  92         }
  93
  94         writeb(intrctrl, instance->reg_base + IOAT_INTRCTRL_OFFSET);
  95         return IRQ_HANDLED;
  96 }
  97
  98 /**
  99  * ioat_dma_do_interrupt_msix - handler used for vector-per-channel interrupt mode
 100  * @irq: interrupt id
 101  * @data: interrupt data
 102  */
 103 static irqreturn_t ioat_dma_do_interrupt_msix(int irq, void *data)
 104 {
 105         struct ioat_dma_chan *ioat_chan = data;
 106
 107         tasklet_schedule(&ioat_chan->cleanup_task);
 108
 109         return IRQ_HANDLED;
 110 }
 111
 112 static void ioat_dma_cleanup_tasklet(unsigned long data);
 113
 114 /**
 115  * ioat_dma_enumerate_channels - find and initialize the device's channels
 116  * @device: the device to be enumerated
 117  */
 118 static int ioat_dma_enumerate_channels(struct ioatdma_device *device)
 119 {
 120         u8 xfercap_scale;
 121         u32 xfercap;
 122         int i;
 123         struct ioat_dma_chan *ioat_chan;
 124
 125         device->common.chancnt = readb(device->reg_base + IOAT_CHANCNT_OFFSET);
 126         xfercap_scale = readb(device->reg_base + IOAT_XFERCAP_OFFSET);
 127         xfercap = (xfercap_scale == 0 ? -1 : (1UL << xfercap_scale));
 128
 129         for (i = 0; i < device->common.chancnt; i++) {
 130                 ioat_chan = kzalloc(sizeof(*ioat_chan), GFP_KERNEL);
 131                 if (!ioat_chan) {
 132                         device->common.chancnt = i;
 133                         break;
 134                 }
 135
 136                 ioat_chan->device = device;
 137                 ioat_chan->reg_base = device->reg_base + (0x80 * (i + 1));
 138                 ioat_chan->xfercap = xfercap;
 139                 ioat_chan->desccount = 0;
 140                 if (ioat_chan->device->version != IOAT_VER_1_2) {
 141                         writel(IOAT_DCACTRL_CMPL_WRITE_ENABLE
 142                                         | IOAT_DMA_DCA_ANY_CPU,
 143                                 ioat_chan->reg_base + IOAT_DCACTRL_OFFSET);
 144                 }
 145                 spin_lock_init(&ioat_chan->cleanup_lock);
 146                 spin_lock_init(&ioat_chan->desc_lock);
 147                 INIT_LIST_HEAD(&ioat_chan->free_desc);
 148                 INIT_LIST_HEAD(&ioat_chan->used_desc);
 149                 /* This should be made common somewhere in dmaengine.c */
 150                 ioat_chan->common.device = &device->common;
 151                 list_add_tail(&ioat_chan->common.device_node,
 152                               &device->common.channels);
 153                 device->idx[i] = ioat_chan;
 154                 tasklet_init(&ioat_chan->cleanup_task,
 155                              ioat_dma_cleanup_tasklet,
 156                              (unsigned long) ioat_chan);
 157                 tasklet_disable(&ioat_chan->cleanup_task);
 158         }
 159         return device->common.chancnt;
 160 }
 161
 162 static void ioat_set_src(dma_addr_t addr,
 163                          struct dma_async_tx_descriptor *tx,
 164                          int index)
 165 {
 166         tx_to_ioat_desc(tx)->src = addr;
 167 }
 168
 169 static void ioat_set_dest(dma_addr_t addr,
 170                           struct dma_async_tx_descriptor *tx,
 171                           int index)
 172 {
 173         tx_to_ioat_desc(tx)->dst = addr;
 174 }
 175
 176 /**
 177  * ioat_dma_memcpy_issue_pending - push potentially unrecognized appended
 178  *                                 descriptors to hw
 179  * @chan: DMA channel handle
 180  */
 181 static inline void __ioat1_dma_memcpy_issue_pending(
 182                                                 struct ioat_dma_chan *ioat_chan)
 183 {
 184         ioat_chan->pending = 0;
 185         writeb(IOAT_CHANCMD_APPEND, ioat_chan->reg_base + IOAT1_CHANCMD_OFFSET);
 186 }
 187
 188 static void ioat1_dma_memcpy_issue_pending(struct dma_chan *chan)
 189 {
 190         struct ioat_dma_chan *ioat_chan = to_ioat_chan(chan);
 191
 192         if (ioat_chan->pending != 0) {
 193                 spin_lock_bh(&ioat_chan->desc_lock);
 194                 __ioat1_dma_memcpy_issue_pending(ioat_chan);
 195                 spin_unlock_bh(&ioat_chan->desc_lock);
 196         }
 197 }
 198
 199 static inline void __ioat2_dma_memcpy_issue_pending(
 200                                                 struct ioat_dma_chan *ioat_chan)
 201 {
 202         ioat_chan->pending = 0;
 203         writew(ioat_chan->dmacount,
 204                ioat_chan->reg_base + IOAT_CHAN_DMACOUNT_OFFSET);
 205 }
 206
 207 static void ioat2_dma_memcpy_issue_pending(struct dma_chan *chan)
 208 {
 209         struct ioat_dma_chan *ioat_chan = to_ioat_chan(chan);
 210
 211         if (ioat_chan->pending != 0) {
 212                 spin_lock_bh(&ioat_chan->desc_lock);
 213                 __ioat2_dma_memcpy_issue_pending(ioat_chan);
 214                 spin_unlock_bh(&ioat_chan->desc_lock);
 215         }
 216 }
 217
 218 static dma_cookie_t ioat1_tx_submit(struct dma_async_tx_descriptor *tx)
 219 {
 220         struct ioat_dma_chan *ioat_chan = to_ioat_chan(tx->chan);
 221         struct ioat_desc_sw *first = tx_to_ioat_desc(tx);
 222         struct ioat_desc_sw *prev, *new;
 223         struct ioat_dma_descriptor *hw;
 224         dma_cookie_t cookie;
 225         LIST_HEAD(new_chain);
 226         u32 copy;
 227         size_t len;
 228         dma_addr_t src, dst;
 229         int orig_ack;
 230         unsigned int desc_count = 0;
 231
 232         /* src and dest and len are stored in the initial descriptor */
 233         len = first->len;
 234         src = first->src;
 235         dst = first->dst;
 236         orig_ack = first->async_tx.ack;
 237         new = first;
 238
 239         spin_lock_bh(&ioat_chan->desc_lock);
 240         prev = to_ioat_desc(ioat_chan->used_desc.prev);
 241         prefetch(prev->hw);
 242         do {
 243                 copy = min_t(size_t, len, ioat_chan->xfercap);
 244
 245                 new->async_tx.ack = 1;
 246
 247                 hw = new->hw;
 248                 hw->size = copy;
 249                 hw->ctl = 0;
 250                 hw->src_addr = src;
 251                 hw->dst_addr = dst;
 252                 hw->next = 0;
 253
 254                 /* chain together the physical address list for the HW */
 255                 wmb();
 256                 prev->hw->next = (u64) new->async_tx.phys;
 257
 258                 len -= copy;
 259                 dst += copy;
 260                 src += copy;
 261
 262                 list_add_tail(&new->node, &new_chain);
 263                 desc_count++;
 264                 prev = new;
 265         } while (len && (new = ioat1_dma_get_next_descriptor(ioat_chan)));
 266
 267         hw->ctl = IOAT_DMA_DESCRIPTOR_CTL_CP_STS;
 268         if (new->async_tx.callback) {
 269                 hw->ctl |= IOAT_DMA_DESCRIPTOR_CTL_INT_GN;
 270                 if (first != new) {
 271                         /* move callback into to last desc */
 272                         new->async_tx.callback = first->async_tx.callback;
 273                         new->async_tx.callback_param
 274                                         = first->async_tx.callback_param;
 275                         first->async_tx.callback = NULL;
 276                         first->async_tx.callback_param = NULL;
 277                 }
 278         }
 279
 280         new->tx_cnt = desc_count;
 281         new->async_tx.ack = orig_ack; /* client is in control of this ack */
 282
 283         /* store the original values for use in later cleanup */
 284         if (new != first) {
 285                 new->src = first->src;
 286                 new->dst = first->dst;
 287                 new->len = first->len;
 288         }
 289
 290         /* cookie incr and addition to used_list must be atomic */
 291         cookie = ioat_chan->common.cookie;
 292         cookie++;
 293         if (cookie < 0)
 294                 cookie = 1;
 295         ioat_chan->common.cookie = new->async_tx.cookie = cookie;
 296
 297         /* write address into NextDescriptor field of last desc in chain */
 298         to_ioat_desc(ioat_chan->used_desc.prev)->hw->next =
 299                                                         first->async_tx.phys;
 300         __list_splice(&new_chain, ioat_chan->used_desc.prev);
 301
 302         ioat_chan->dmacount += desc_count;
 303         ioat_chan->pending += desc_count;
 304         if (ioat_chan->pending >= ioat_pending_level)
 305                 __ioat1_dma_memcpy_issue_pending(ioat_chan);
 306         spin_unlock_bh(&ioat_chan->desc_lock);
 307
 308         return cookie;
 309 }
 310
 311 static dma_cookie_t ioat2_tx_submit(struct dma_async_tx_descriptor *tx)
 312 {
 313         struct ioat_dma_chan *ioat_chan = to_ioat_chan(tx->chan);
 314         struct ioat_desc_sw *first = tx_to_ioat_desc(tx);
 315         struct ioat_desc_sw *new;
 316         struct ioat_dma_descriptor *hw;
 317         dma_cookie_t cookie;
 318         u32 copy;
 319         size_t len;
 320         dma_addr_t src, dst;
 321         int orig_ack;
 322         unsigned int desc_count = 0;
 323
 324         /* src and dest and len are stored in the initial descriptor */
 325         len = first->len;
 326         src = first->src;
 327         dst = first->dst;
 328         orig_ack = first->async_tx.ack;
 329         new = first;
 330
 331         /*
 332          * ioat_chan->desc_lock is still in force in version 2 path
 333          * it gets unlocked at end of this function
 334          */
 335         do {
 336                 copy = min_t(size_t, len, ioat_chan->xfercap);
 337
 338                 new->async_tx.ack = 1;
 339
 340                 hw = new->hw;
 341                 hw->size = copy;
 342                 hw->ctl = 0;
 343                 hw->src_addr = src;
 344                 hw->dst_addr = dst;
 345
 346                 len -= copy;
 347                 dst += copy;
 348                 src += copy;
 349                 desc_count++;
 350         } while (len && (new = ioat2_dma_get_next_descriptor(ioat_chan)));
 351
 352         hw->ctl = IOAT_DMA_DESCRIPTOR_CTL_CP_STS;
 353         if (new->async_tx.callback) {
 354                 hw->ctl |= IOAT_DMA_DESCRIPTOR_CTL_INT_GN;
 355                 if (first != new) {
 356                         /* move callback into to last desc */
 357                         new->async_tx.callback = first->async_tx.callback;
 358                         new->async_tx.callback_param
 359                                         = first->async_tx.callback_param;
 360                         first->async_tx.callback = NULL;
 361                         first->async_tx.callback_param = NULL;
 362                 }
 363         }
 364
 365         new->tx_cnt = desc_count;
 366         new->async_tx.ack = orig_ack; /* client is in control of this ack */
 367
 368         /* store the original values for use in later cleanup */
 369         if (new != first) {
 370                 new->src = first->src;
 371                 new->dst = first->dst;
 372                 new->len = first->len;
 373         }
 374
 375         /* cookie incr and addition to used_list must be atomic */
 376         cookie = ioat_chan->common.cookie;
 377         cookie++;
 378         if (cookie < 0)
 379                 cookie = 1;
 380         ioat_chan->common.cookie = new->async_tx.cookie = cookie;
 381
 382         ioat_chan->dmacount += desc_count;
 383         ioat_chan->pending += desc_count;
 384         if (ioat_chan->pending >= ioat_pending_level)
 385                 __ioat2_dma_memcpy_issue_pending(ioat_chan);
 386         spin_unlock_bh(&ioat_chan->desc_lock);
 387
 388         return cookie;
 389 }
 390
 391 /**
 392  * ioat_dma_alloc_descriptor - allocate and return a sw and hw descriptor pair
 393  * @ioat_chan: the channel supplying the memory pool for the descriptors
 394  * @flags: allocation flags
 395  */
 396 static struct ioat_desc_sw *ioat_dma_alloc_descriptor(
 397                                         struct ioat_dma_chan *ioat_chan,
 398                                         gfp_t flags)
 399 {
 400         struct ioat_dma_descriptor *desc;
 401         struct ioat_desc_sw *desc_sw;
 402         struct ioatdma_device *ioatdma_device;
 403         dma_addr_t phys;
 404
 405         ioatdma_device = to_ioatdma_device(ioat_chan->common.device);
 406         desc = pci_pool_alloc(ioatdma_device->dma_pool, flags, &phys);
 407         if (unlikely(!desc))
 408                 return NULL;
 409
 410         desc_sw = kzalloc(sizeof(*desc_sw), flags);
 411         if (unlikely(!desc_sw)) {
 412                 pci_pool_free(ioatdma_device->dma_pool, desc, phys);
 413                 return NULL;
 414         }
 415
 416         memset(desc, 0, sizeof(*desc));
 417         dma_async_tx_descriptor_init(&desc_sw->async_tx, &ioat_chan->common);
 418         desc_sw->async_tx.tx_set_src = ioat_set_src;
 419         desc_sw->async_tx.tx_set_dest = ioat_set_dest;
 420         switch (ioat_chan->device->version) {
 421         case IOAT_VER_1_2:
 422                 desc_sw->async_tx.tx_submit = ioat1_tx_submit;
 423                 break;
 424         case IOAT_VER_2_0:
 425                 desc_sw->async_tx.tx_submit = ioat2_tx_submit;
 426                 break;
 427         }
 428         INIT_LIST_HEAD(&desc_sw->async_tx.tx_list);
 429
 430         desc_sw->hw = desc;
 431         desc_sw->async_tx.phys = phys;
 432
 433         return desc_sw;
 434 }
 435
 436 static int ioat_initial_desc_count = 256;
 437 module_param(ioat_initial_desc_count, int, 0644);
 438 MODULE_PARM_DESC(ioat_initial_desc_count,
 439                  "initial descriptors per channel (default: 256)");
 440
 441 /**
 442  * ioat2_dma_massage_chan_desc - link the descriptors into a circle
 443  * @ioat_chan: the channel to be massaged
 444  */
 445 static void ioat2_dma_massage_chan_desc(struct ioat_dma_chan *ioat_chan)
 446 {
 447         struct ioat_desc_sw *desc, *_desc;
 448
 449         /* setup used_desc */
 450         ioat_chan->used_desc.next = ioat_chan->free_desc.next;
 451         ioat_chan->used_desc.prev = NULL;
 452
 453         /* pull free_desc out of the circle so that every node is a hw
 454          * descriptor, but leave it pointing to the list
 455          */
 456         ioat_chan->free_desc.prev->next = ioat_chan->free_desc.next;
 457         ioat_chan->free_desc.next->prev = ioat_chan->free_desc.prev;
 458
 459         /* circle link the hw descriptors */
 460         desc = to_ioat_desc(ioat_chan->free_desc.next);
 461         desc->hw->next = to_ioat_desc(desc->node.next)->async_tx.phys;
 462         list_for_each_entry_safe(desc, _desc, ioat_chan->free_desc.next, node) {
 463                 desc->hw->next = to_ioat_desc(desc->node.next)->async_tx.phys;
 464         }
 465 }
 466
 467 /**
 468  * ioat_dma_alloc_chan_resources - returns the number of allocated descriptors
 469  * @chan: the channel to be filled out
 470  */
 471 static int ioat_dma_alloc_chan_resources(struct dma_chan *chan)
 472 {
 473         struct ioat_dma_chan *ioat_chan = to_ioat_chan(chan);
 474         struct ioat_desc_sw *desc;
 475         u16 chanctrl;
 476         u32 chanerr;
 477         int i;
 478         LIST_HEAD(tmp_list);
 479
 480         /* have we already been set up? */
 481         if (!list_empty(&ioat_chan->free_desc))
 482                 return ioat_chan->desccount;
 483
 484         /* Setup register to interrupt and write completion status on error */
 485         chanctrl = IOAT_CHANCTRL_ERR_INT_EN |
 486                 IOAT_CHANCTRL_ANY_ERR_ABORT_EN |
 487                 IOAT_CHANCTRL_ERR_COMPLETION_EN;
 488         writew(chanctrl, ioat_chan->reg_base + IOAT_CHANCTRL_OFFSET);
 489
 490         chanerr = readl(ioat_chan->reg_base + IOAT_CHANERR_OFFSET);
 491         if (chanerr) {
 492                 dev_err(&ioat_chan->device->pdev->dev,
 493                         "CHANERR = %x, clearing\n", chanerr);
 494                 writel(chanerr, ioat_chan->reg_base + IOAT_CHANERR_OFFSET);
 495         }
 496
 497         /* Allocate descriptors */
 498         for (i = 0; i < ioat_initial_desc_count; i++) {
 499                 desc = ioat_dma_alloc_descriptor(ioat_chan, GFP_KERNEL);
 500                 if (!desc) {
 501                         dev_err(&ioat_chan->device->pdev->dev,
 502                                 "Only %d initial descriptors\n", i);
 503                         break;
 504                 }
 505                 list_add_tail(&desc->node, &tmp_list);
 506         }
 507         spin_lock_bh(&ioat_chan->desc_lock);
 508         ioat_chan->desccount = i;
 509         list_splice(&tmp_list, &ioat_chan->free_desc);
 510         if (ioat_chan->device->version != IOAT_VER_1_2)
 511                 ioat2_dma_massage_chan_desc(ioat_chan);
 512         spin_unlock_bh(&ioat_chan->desc_lock);
 513
 514         /* allocate a completion writeback area */
 515         /* doing 2 32bit writes to mmio since 1 64b write doesn't work */
 516         ioat_chan->completion_virt =
 517                 pci_pool_alloc(ioat_chan->device->completion_pool,
 518                                GFP_KERNEL,
 519                                &ioat_chan->completion_addr);
 520         memset(ioat_chan->completion_virt, 0,
 521                sizeof(*ioat_chan->completion_virt));
 522         writel(((u64) ioat_chan->completion_addr) & 0x00000000FFFFFFFF,
 523                ioat_chan->reg_base + IOAT_CHANCMP_OFFSET_LOW);
 524         writel(((u64) ioat_chan->completion_addr) >> 32,
 525                ioat_chan->reg_base + IOAT_CHANCMP_OFFSET_HIGH);
 526
 527         tasklet_enable(&ioat_chan->cleanup_task);
 528         ioat_dma_start_null_desc(ioat_chan);  /* give chain to dma device */
 529         return ioat_chan->desccount;
 530 }
 531
 532 /**
 533  * ioat_dma_free_chan_resources - release all the descriptors
 534  * @chan: the channel to be cleaned
 535  */
 536 static void ioat_dma_free_chan_resources(struct dma_chan *chan)
 537 {
 538         struct ioat_dma_chan *ioat_chan = to_ioat_chan(chan);
 539         struct ioatdma_device *ioatdma_device = to_ioatdma_device(chan->device);
 540         struct ioat_desc_sw *desc, *_desc;
 541         int in_use_descs = 0;
 542
 543         tasklet_disable(&ioat_chan->cleanup_task);
 544         ioat_dma_memcpy_cleanup(ioat_chan);
 545
 546         /* Delay 100ms after reset to allow internal DMA logic to quiesce
 547          * before removing DMA descriptor resources.
 548          */
 549         writeb(IOAT_CHANCMD_RESET,
 550                ioat_chan->reg_base
 551                         + IOAT_CHANCMD_OFFSET(ioat_chan->device->version));
 552         mdelay(100);
 553
 554         spin_lock_bh(&ioat_chan->desc_lock);
 555         switch (ioat_chan->device->version) {
 556         case IOAT_VER_1_2:
 557                 list_for_each_entry_safe(desc, _desc,
 558                                          &ioat_chan->used_desc, node) {
 559                         in_use_descs++;
 560                         list_del(&desc->node);
 561                         pci_pool_free(ioatdma_device->dma_pool, desc->hw,
 562                                       desc->async_tx.phys);
 563                         kfree(desc);
 564                 }
 565                 list_for_each_entry_safe(desc, _desc,
 566                                          &ioat_chan->free_desc, node) {
 567                         list_del(&desc->node);
 568                         pci_pool_free(ioatdma_device->dma_pool, desc->hw,
 569                                       desc->async_tx.phys);
 570                         kfree(desc);
 571                 }
 572                 break;
 573         case IOAT_VER_2_0:
 574                 list_for_each_entry_safe(desc, _desc,
 575                                          ioat_chan->free_desc.next, node) {
 576                         list_del(&desc->node);
 577                         pci_pool_free(ioatdma_device->dma_pool, desc->hw,
 578                                       desc->async_tx.phys);
 579                         kfree(desc);
 580                 }
 581                 desc = to_ioat_desc(ioat_chan->free_desc.next);
 582                 pci_pool_free(ioatdma_device->dma_pool, desc->hw,
 583                               desc->async_tx.phys);
 584                 kfree(desc);
 585                 INIT_LIST_HEAD(&ioat_chan->free_desc);
 586                 INIT_LIST_HEAD(&ioat_chan->used_desc);
 587                 break;
 588         }
 589         spin_unlock_bh(&ioat_chan->desc_lock);
 590
 591         pci_pool_free(ioatdma_device->completion_pool,
 592                       ioat_chan->completion_virt,
 593                       ioat_chan->completion_addr);
 594
 595         /* one is ok since we left it on there on purpose */
 596         if (in_use_descs > 1)
 597                 dev_err(&ioat_chan->device->pdev->dev,
 598                         "Freeing %d in use descriptors!\n",
 599                         in_use_descs - 1);
 600
 601         ioat_chan->last_completion = ioat_chan->completion_addr = 0;
 602         ioat_chan->pending = 0;
 603         ioat_chan->dmacount = 0;
 604 }
 605
 606 /**
 607  * ioat_dma_get_next_descriptor - return the next available descriptor
 608  * @ioat_chan: IOAT DMA channel handle
 609  *
 610  * Gets the next descriptor from the chain, and must be called with the
 611  * channel's desc_lock held.  Allocates more descriptors if the channel
 612  * has run out.
 613  */
 614 static struct ioat_desc_sw *
 615 ioat1_dma_get_next_descriptor(struct ioat_dma_chan *ioat_chan)
 616 {
 617         struct ioat_desc_sw *new;
 618
 619         if (!list_empty(&ioat_chan->free_desc)) {
 620                 new = to_ioat_desc(ioat_chan->free_desc.next);
 621                 list_del(&new->node);
 622         } else {
 623                 /* try to get another desc */
 624                 new = ioat_dma_alloc_descriptor(ioat_chan, GFP_ATOMIC);
 625                 if (!new) {
 626                         dev_err(&ioat_chan->device->pdev->dev,
 627                                 "alloc failed\n");
 628                         return NULL;
 629                 }
 630         }
 631
 632         prefetch(new->hw);
 633         return new;
 634 }
 635
 636 static struct ioat_desc_sw *
 637 ioat2_dma_get_next_descriptor(struct ioat_dma_chan *ioat_chan)
 638 {
 639         struct ioat_desc_sw *new;
 640
 641         /*
 642          * used.prev points to where to start processing
 643          * used.next points to next free descriptor
 644          * if used.prev == NULL, there are none waiting to be processed
 645          * if used.next == used.prev.prev, there is only one free descriptor,
 646          *      and we need to use it to as a noop descriptor before
 647          *      linking in a new set of descriptors, since the device
 648          *      has probably already read the pointer to it
 649          */
 650         if (ioat_chan->used_desc.prev &&
 651             ioat_chan->used_desc.next == ioat_chan->used_desc.prev->prev) {
 652
 653                 struct ioat_desc_sw *desc;
 654                 struct ioat_desc_sw *noop_desc;
 655                 int i;
 656
 657                 /* set up the noop descriptor */
 658                 noop_desc = to_ioat_desc(ioat_chan->used_desc.next);
 659                 noop_desc->hw->size = 0;
 660                 noop_desc->hw->ctl = IOAT_DMA_DESCRIPTOR_NUL;
 661                 noop_desc->hw->src_addr = 0;
 662                 noop_desc->hw->dst_addr = 0;
 663
 664                 ioat_chan->used_desc.next = ioat_chan->used_desc.next->next;
 665                 ioat_chan->pending++;
 666                 ioat_chan->dmacount++;
 667
 668                 /* try to get a few more descriptors */
 669                 for (i = 16; i; i--) {
 670                         desc = ioat_dma_alloc_descriptor(ioat_chan, GFP_ATOMIC);
 671                         if (!desc) {
 672                                 dev_err(&ioat_chan->device->pdev->dev,
 673                                         "alloc failed\n");
 674                                 break;
 675                         }
 676                         list_add_tail(&desc->node, ioat_chan->used_desc.next);
 677
 678                         desc->hw->next
 679                                 = to_ioat_desc(desc->node.next)->async_tx.phys;
 680                         to_ioat_desc(desc->node.prev)->hw->next
 681                                 = desc->async_tx.phys;
 682                         ioat_chan->desccount++;
 683                 }
 684
 685                 ioat_chan->used_desc.next = noop_desc->node.next;
 686         }
 687         new = to_ioat_desc(ioat_chan->used_desc.next);
 688         prefetch(new);
 689         ioat_chan->used_desc.next = new->node.next;
 690
 691         if (ioat_chan->used_desc.prev == NULL)
 692                 ioat_chan->used_desc.prev = &new->node;
 693
 694         prefetch(new->hw);
 695         return new;
 696 }
 697
 698 static struct ioat_desc_sw *ioat_dma_get_next_descriptor(
 699                                                 struct ioat_dma_chan *ioat_chan)
 700 {
 701         if (!ioat_chan)
 702                 return NULL;
 703
 704         switch (ioat_chan->device->version) {
 705         case IOAT_VER_1_2:
 706                 return ioat1_dma_get_next_descriptor(ioat_chan);
 707                 break;
 708         case IOAT_VER_2_0:
 709                 return ioat2_dma_get_next_descriptor(ioat_chan);
 710                 break;
 711         }
 712         return NULL;
 713 }
 714
 715 static struct dma_async_tx_descriptor *ioat1_dma_prep_memcpy(
 716                                                 struct dma_chan *chan,
 717                                                 size_t len,
 718                                                 int int_en)
 719 {
 720         struct ioat_dma_chan *ioat_chan = to_ioat_chan(chan);
 721         struct ioat_desc_sw *new;
 722
 723         spin_lock_bh(&ioat_chan->desc_lock);
 724         new = ioat_dma_get_next_descriptor(ioat_chan);
 725         spin_unlock_bh(&ioat_chan->desc_lock);
 726
 727         if (new) {
 728                 new->len = len;
 729                 return &new->async_tx;
 730         } else
 731                 return NULL;
 732 }
 733
 734 static struct dma_async_tx_descriptor *ioat2_dma_prep_memcpy(
 735                                                 struct dma_chan *chan,
 736                                                 size_t len,
 737                                                 int int_en)
 738 {
 739         struct ioat_dma_chan *ioat_chan = to_ioat_chan(chan);
 740         struct ioat_desc_sw *new;
 741
 742         spin_lock_bh(&ioat_chan->desc_lock);
 743         new = ioat2_dma_get_next_descriptor(ioat_chan);
 744
 745         /*
 746          * leave ioat_chan->desc_lock set in ioat 2 path
 747          * it will get unlocked at end of tx_submit
 748          */
 749
 750         if (new) {
 751                 new->len = len;
 752                 return &new->async_tx;
 753         } else
 754                 return NULL;
 755 }
 756
 757 static void ioat_dma_cleanup_tasklet(unsigned long data)
 758 {
 759         struct ioat_dma_chan *chan = (void *)data;
 760         ioat_dma_memcpy_cleanup(chan);
 761         writew(IOAT_CHANCTRL_INT_DISABLE,
 762                chan->reg_base + IOAT_CHANCTRL_OFFSET);
 763 }
 764
 765 /**
 766  * ioat_dma_memcpy_cleanup - cleanup up finished descriptors
 767  * @chan: ioat channel to be cleaned up
 768  */
 769 static void ioat_dma_memcpy_cleanup(struct ioat_dma_chan *ioat_chan)
 770 {
 771         unsigned long phys_complete;
 772         struct ioat_desc_sw *desc, *_desc;
 773         dma_cookie_t cookie = 0;
 774         unsigned long desc_phys;
 775         struct ioat_desc_sw *latest_desc;
 776
 777         prefetch(ioat_chan->completion_virt);
 778
 779         if (!spin_trylock_bh(&ioat_chan->cleanup_lock))
 780                 return;
 781
 782         /* The completion writeback can happen at any time,
 783            so reads by the driver need to be atomic operations
 784            The descriptor physical addresses are limited to 32-bits
 785            when the CPU can only do a 32-bit mov */
 786
 787 #if (BITS_PER_LONG == 64)
 788         phys_complete =
 789                 ioat_chan->completion_virt->full
 790                 & IOAT_CHANSTS_COMPLETED_DESCRIPTOR_ADDR;
 791 #else
 792         phys_complete =
 793                 ioat_chan->completion_virt->low & IOAT_LOW_COMPLETION_MASK;
 794 #endif
 795
 796         if ((ioat_chan->completion_virt->full
 797                 & IOAT_CHANSTS_DMA_TRANSFER_STATUS) ==
 798                                 IOAT_CHANSTS_DMA_TRANSFER_STATUS_HALTED) {
 799                 dev_err(&ioat_chan->device->pdev->dev,
 800                         "Channel halted, chanerr = %x\n",
 801                         readl(ioat_chan->reg_base + IOAT_CHANERR_OFFSET));
 802
 803                 /* TODO do something to salvage the situation */
 804         }
 805
 806         if (phys_complete == ioat_chan->last_completion) {
 807                 spin_unlock_bh(&ioat_chan->cleanup_lock);
 808                 return;
 809         }
 810
 811         cookie = 0;
 812         spin_lock_bh(&ioat_chan->desc_lock);
 813         switch (ioat_chan->device->version) {
 814         case IOAT_VER_1_2:
 815                 list_for_each_entry_safe(desc, _desc,
 816                                          &ioat_chan->used_desc, node) {
 817
 818                         /*
 819                          * Incoming DMA requests may use multiple descriptors,
 820                          * due to exceeding xfercap, perhaps. If so, only the
 821                          * last one will have a cookie, and require unmapping.
 822                          */
 823                         if (desc->async_tx.cookie) {
 824                                 cookie = desc->async_tx.cookie;
 825
 826                                 /*
 827                                  * yes we are unmapping both _page and _single
 828                                  * alloc'd regions with unmap_page. Is this
 829                                  * *really* that bad?
 830                                  */
 831                                 pci_unmap_page(ioat_chan->device->pdev,
 832                                                 pci_unmap_addr(desc, dst),
 833                                                 pci_unmap_len(desc, len),
 834                                                 PCI_DMA_FROMDEVICE);
 835                                 pci_unmap_page(ioat_chan->device->pdev,
 836                                                 pci_unmap_addr(desc, src),
 837                                                 pci_unmap_len(desc, len),
 838                                                 PCI_DMA_TODEVICE);
 839
 840                                 if (desc->async_tx.callback) {
 841                                         desc->async_tx.callback(desc->async_tx.callback_param);
 842                                         desc->async_tx.callback = NULL;
 843                                 }
 844                         }
 845
 846                         if (desc->async_tx.phys != phys_complete) {
 847                                 /*
 848                                  * a completed entry, but not the last, so clean
 849                                  * up if the client is done with the descriptor
 850                                  */
 851                                 if (desc->async_tx.ack) {
 852                                         list_del(&desc->node);
 853                                         list_add_tail(&desc->node,
 854                                                       &ioat_chan->free_desc);
 855                                 } else
 856                                         desc->async_tx.cookie = 0;
 857                         } else {
 858                                 /*
 859                                  * last used desc. Do not remove, so we can
 860                                  * append from it, but don't look at it next
 861                                  * time, either
 862                                  */
 863                                 desc->async_tx.cookie = 0;
 864
 865                                 /* TODO check status bits? */
 866                                 break;
 867                         }
 868                 }
 869                 break;
 870         case IOAT_VER_2_0:
 871                 /* has some other thread has already cleaned up? */
 872                 if (ioat_chan->used_desc.prev == NULL)
 873                         break;
 874
 875                 /* work backwards to find latest finished desc */
 876                 desc = to_ioat_desc(ioat_chan->used_desc.next);
 877                 latest_desc = NULL;
 878                 do {
 879                         desc = to_ioat_desc(desc->node.prev);
 880                         desc_phys = (unsigned long)desc->async_tx.phys
 881                                        & IOAT_CHANSTS_COMPLETED_DESCRIPTOR_ADDR;
 882                         if (desc_phys == phys_complete) {
 883                                 latest_desc = desc;
 884                                 break;
 885                         }
 886                 } while (&desc->node != ioat_chan->used_desc.prev);
 887
 888                 if (latest_desc != NULL) {
 889
 890                         /* work forwards to clear finished descriptors */
 891                         for (desc = to_ioat_desc(ioat_chan->used_desc.prev);
 892                              &desc->node != latest_desc->node.next &&
 893                              &desc->node != ioat_chan->used_desc.next;
 894                              desc = to_ioat_desc(desc->node.next)) {
 895                                 if (desc->async_tx.cookie) {
 896                                         cookie = desc->async_tx.cookie;
 897                                         desc->async_tx.cookie = 0;
 898
 899                                         pci_unmap_page(ioat_chan->device->pdev,
 900                                                       pci_unmap_addr(desc, dst),
 901                                                       pci_unmap_len(desc, len),
 902                                                       PCI_DMA_FROMDEVICE);
 903                                         pci_unmap_page(ioat_chan->device->pdev,
 904                                                       pci_unmap_addr(desc, src),
 905                                                       pci_unmap_len(desc, len),
 906                                                       PCI_DMA_TODEVICE);
 907
 908                                         if (desc->async_tx.callback) {
 909                                                 desc->async_tx.callback(desc->async_tx.callback_param);
 910                                                 desc->async_tx.callback = NULL;
 911                                         }
 912                                 }
 913                         }
 914
 915                         /* move used.prev up beyond those that are finished */
 916                         if (&desc->node == ioat_chan->used_desc.next)
 917                                 ioat_chan->used_desc.prev = NULL;
 918                         else
 919                                 ioat_chan->used_desc.prev = &desc->node;
 920                 }
 921                 break;
 922         }
 923
 924         spin_unlock_bh(&ioat_chan->desc_lock);
 925
 926         ioat_chan->last_completion = phys_complete;
 927         if (cookie != 0)
 928                 ioat_chan->completed_cookie = cookie;
 929
 930         spin_unlock_bh(&ioat_chan->cleanup_lock);
 931 }
 932
 933 static void ioat_dma_dependency_added(struct dma_chan *chan)
 934 {
 935         struct ioat_dma_chan *ioat_chan = to_ioat_chan(chan);
 936         spin_lock_bh(&ioat_chan->desc_lock);
 937         if (ioat_chan->pending == 0) {
 938                 spin_unlock_bh(&ioat_chan->desc_lock);
 939                 ioat_dma_memcpy_cleanup(ioat_chan);
 940         } else
 941                 spin_unlock_bh(&ioat_chan->desc_lock);
 942 }
 943
 944 /**
 945  * ioat_dma_is_complete - poll the status of a IOAT DMA transaction
 946  * @chan: IOAT DMA channel handle
 947  * @cookie: DMA transaction identifier
 948  * @done: if not %NULL, updated with last completed transaction
 949  * @used: if not %NULL, updated with last used transaction
 950  */
 951 static enum dma_status ioat_dma_is_complete(struct dma_chan *chan,
 952                                             dma_cookie_t cookie,
 953                                             dma_cookie_t *done,
 954                                             dma_cookie_t *used)
 955 {
 956         struct ioat_dma_chan *ioat_chan = to_ioat_chan(chan);
 957         dma_cookie_t last_used;
 958         dma_cookie_t last_complete;
 959         enum dma_status ret;
 960
 961         last_used = chan->cookie;
 962         last_complete = ioat_chan->completed_cookie;
 963
 964         if (done)
 965                 *done = last_complete;
 966         if (used)
 967                 *used = last_used;
 968
 969         ret = dma_async_is_complete(cookie, last_complete, last_used);
 970         if (ret == DMA_SUCCESS)
 971                 return ret;
 972
 973         ioat_dma_memcpy_cleanup(ioat_chan);
 974
 975         last_used = chan->cookie;
 976         last_complete = ioat_chan->completed_cookie;
 977
 978         if (done)
 979                 *done = last_complete;
 980         if (used)
 981                 *used = last_used;
 982
 983         return dma_async_is_complete(cookie, last_complete, last_used);
 984 }
 985
 986 static void ioat_dma_start_null_desc(struct ioat_dma_chan *ioat_chan)
 987 {
 988         struct ioat_desc_sw *desc;
 989
 990         spin_lock_bh(&ioat_chan->desc_lock);
 991
 992         desc = ioat_dma_get_next_descriptor(ioat_chan);
 993         desc->hw->ctl = IOAT_DMA_DESCRIPTOR_NUL
 994                                 | IOAT_DMA_DESCRIPTOR_CTL_INT_GN
 995                                 | IOAT_DMA_DESCRIPTOR_CTL_CP_STS;
 996         desc->hw->size = 0;
 997         desc->hw->src_addr = 0;
 998         desc->hw->dst_addr = 0;
 999         desc->async_tx.ack = 1;
1000         switch (ioat_chan->device->version) {
1001         case IOAT_VER_1_2:
1002                 desc->hw->next = 0;
1003                 list_add_tail(&desc->node, &ioat_chan->used_desc);
1004
1005                 writel(((u64) desc->async_tx.phys) & 0x00000000FFFFFFFF,
1006                        ioat_chan->reg_base + IOAT1_CHAINADDR_OFFSET_LOW);
1007                 writel(((u64) desc->async_tx.phys) >> 32,
1008                        ioat_chan->reg_base + IOAT1_CHAINADDR_OFFSET_HIGH);
1009
1010                 writeb(IOAT_CHANCMD_START, ioat_chan->reg_base
1011                         + IOAT_CHANCMD_OFFSET(ioat_chan->device->version));
1012                 break;
1013         case IOAT_VER_2_0:
1014                 writel(((u64) desc->async_tx.phys) & 0x00000000FFFFFFFF,
1015                        ioat_chan->reg_base + IOAT2_CHAINADDR_OFFSET_LOW);
1016                 writel(((u64) desc->async_tx.phys) >> 32,
1017                        ioat_chan->reg_base + IOAT2_CHAINADDR_OFFSET_HIGH);
1018
1019                 ioat_chan->dmacount++;
1020                 __ioat2_dma_memcpy_issue_pending(ioat_chan);
1021                 break;
1022         }
1023         spin_unlock_bh(&ioat_chan->desc_lock);
1024 }
1025
1026 /*
1027  * Perform a IOAT transaction to verify the HW works.
1028  */
1029 #define IOAT_TEST_SIZE 2000
1030
1031 static void ioat_dma_test_callback(void *dma_async_param)
1032 {
1033         printk(KERN_ERR "ioatdma: ioat_dma_test_callback(%p)\n",
1034                 dma_async_param);
1035 }
1036
1037 /**
1038  * ioat_dma_self_test - Perform a IOAT transaction to verify the HW works.
1039  * @device: device to be tested
1040  */
1041 static int ioat_dma_self_test(struct ioatdma_device *device)
1042 {
1043         int i;
1044         u8 *src;
1045         u8 *dest;
1046         struct dma_chan *dma_chan;
1047         struct dma_async_tx_descriptor *tx;
1048         dma_addr_t addr;
1049         dma_cookie_t cookie;
1050         int err = 0;
1051
1052         src = kzalloc(sizeof(u8) * IOAT_TEST_SIZE, GFP_KERNEL);
1053         if (!src)
1054                 return -ENOMEM;
1055         dest = kzalloc(sizeof(u8) * IOAT_TEST_SIZE, GFP_KERNEL);
1056         if (!dest) {
1057                 kfree(src);
1058                 return -ENOMEM;
1059         }
1060
1061         /* Fill in src buffer */
1062         for (i = 0; i < IOAT_TEST_SIZE; i++)
1063                 src[i] = (u8)i;
1064
1065         /* Start copy, using first DMA channel */
1066         dma_chan = container_of(device->common.channels.next,
1067                                 struct dma_chan,
1068                                 device_node);
1069         if (device->common.device_alloc_chan_resources(dma_chan) < 1) {
1070                 dev_err(&device->pdev->dev,
1071                         "selftest cannot allocate chan resource\n");
1072                 err = -ENODEV;
1073                 goto out;
1074         }
1075
1076         tx = device->common.device_prep_dma_memcpy(dma_chan, IOAT_TEST_SIZE, 0);
1077         if (!tx) {
1078                 dev_err(&device->pdev->dev,
1079                         "Self-test prep failed, disabling\n");
1080                 err = -ENODEV;
1081                 goto free_resources;
1082         }
1083
1084         async_tx_ack(tx);
1085         addr = dma_map_single(dma_chan->device->dev, src, IOAT_TEST_SIZE,
1086                               DMA_TO_DEVICE);
1087         tx->tx_set_src(addr, tx, 0);
1088         addr = dma_map_single(dma_chan->device->dev, dest, IOAT_TEST_SIZE,
1089                               DMA_FROM_DEVICE);
1090         tx->tx_set_dest(addr, tx, 0);
1091         tx->callback = ioat_dma_test_callback;
1092         tx->callback_param = (void *)0x8086;
1093         cookie = tx->tx_submit(tx);
1094         if (cookie < 0) {
1095                 dev_err(&device->pdev->dev,
1096                         "Self-test setup failed, disabling\n");
1097                 err = -ENODEV;
1098                 goto free_resources;
1099         }
1100         device->common.device_issue_pending(dma_chan);
1101         msleep(1);
1102
1103         if (device->common.device_is_tx_complete(dma_chan, cookie, NULL, NULL)
1104                                         != DMA_SUCCESS) {
1105                 dev_err(&device->pdev->dev,
1106                         "Self-test copy timed out, disabling\n");
1107                 err = -ENODEV;
1108                 goto free_resources;
1109         }
1110         if (memcmp(src, dest, IOAT_TEST_SIZE)) {
1111                 dev_err(&device->pdev->dev,
1112                         "Self-test copy failed compare, disabling\n");
1113                 err = -ENODEV;
1114                 goto free_resources;
1115         }
1116
1117 free_resources:
1118         device->common.device_free_chan_resources(dma_chan);
1119 out:
1120         kfree(src);
1121         kfree(dest);
1122         return err;
1123 }
1124
1125 static char ioat_interrupt_style[32] = "msix";
1126 module_param_string(ioat_interrupt_style, ioat_interrupt_style,
1127                     sizeof(ioat_interrupt_style), 0644);
1128 MODULE_PARM_DESC(ioat_interrupt_style,
1129                  "set ioat interrupt style: msix (default), "
1130                  "msix-single-vector, msi, intx)");
1131
1132 /**
1133  * ioat_dma_setup_interrupts - setup interrupt handler
1134  * @device: ioat device
1135  */
1136 static int ioat_dma_setup_interrupts(struct ioatdma_device *device)
1137 {
1138         struct ioat_dma_chan *ioat_chan;
1139         int err, i, j, msixcnt;
1140         u8 intrctrl = 0;
1141
1142         if (!strcmp(ioat_interrupt_style, "msix"))
1143                 goto msix;
1144         if (!strcmp(ioat_interrupt_style, "msix-single-vector"))
1145                 goto msix_single_vector;
1146         if (!strcmp(ioat_interrupt_style, "msi"))
1147                 goto msi;
1148         if (!strcmp(ioat_interrupt_style, "intx"))
1149                 goto intx;
1150         dev_err(&device->pdev->dev, "invalid ioat_interrupt_style %s\n",
1151                 ioat_interrupt_style);
1152         goto err_no_irq;
1153
1154 msix:
1155         /* The number of MSI-X vectors should equal the number of channels */
1156         msixcnt = device->common.chancnt;
1157         for (i = 0; i < msixcnt; i++)
1158                 device->msix_entries[i].entry = i;
1159
1160         err = pci_enable_msix(device->pdev, device->msix_entries, msixcnt);
1161         if (err < 0)
1162                 goto msi;
1163         if (err > 0)
1164                 goto msix_single_vector;
1165
1166         for (i = 0; i < msixcnt; i++) {
1167                 ioat_chan = ioat_lookup_chan_by_index(device, i);
1168                 err = request_irq(device->msix_entries[i].vector,
1169                                   ioat_dma_do_interrupt_msix,
1170                                   0, "ioat-msix", ioat_chan);
1171                 if (err) {
1172                         for (j = 0; j < i; j++) {
1173                                 ioat_chan =
1174                                         ioat_lookup_chan_by_index(device, j);
1175                                 free_irq(device->msix_entries[j].vector,
1176                                          ioat_chan);
1177                         }
1178                         goto msix_single_vector;
1179                 }
1180         }
1181         intrctrl |= IOAT_INTRCTRL_MSIX_VECTOR_CONTROL;
1182         device->irq_mode = msix_multi_vector;
1183         goto done;
1184
1185 msix_single_vector:
1186         device->msix_entries[0].entry = 0;
1187         err = pci_enable_msix(device->pdev, device->msix_entries, 1);
1188         if (err)
1189                 goto msi;
1190
1191         err = request_irq(device->msix_entries[0].vector, ioat_dma_do_interrupt,
1192                           0, "ioat-msix", device);
1193         if (err) {
1194                 pci_disable_msix(device->pdev);
1195                 goto msi;
1196         }
1197         device->irq_mode = msix_single_vector;
1198         goto done;
1199
1200 msi:
1201         err = pci_enable_msi(device->pdev);
1202         if (err)
1203                 goto intx;
1204
1205         err = request_irq(device->pdev->irq, ioat_dma_do_interrupt,
1206                           0, "ioat-msi", device);
1207         if (err) {
1208                 pci_disable_msi(device->pdev);
1209                 goto intx;
1210         }
1211         /*
1212          * CB 1.2 devices need a bit set in configuration space to enable MSI
1213          */
1214         if (device->version == IOAT_VER_1_2) {
1215                 u32 dmactrl;
1216                 pci_read_config_dword(device->pdev,
1217                                       IOAT_PCI_DMACTRL_OFFSET, &dmactrl);
1218                 dmactrl |= IOAT_PCI_DMACTRL_MSI_EN;
1219                 pci_write_config_dword(device->pdev,
1220                                        IOAT_PCI_DMACTRL_OFFSET, dmactrl);
1221         }
1222         device->irq_mode = msi;
1223         goto done;
1224
1225 intx:
1226         err = request_irq(device->pdev->irq, ioat_dma_do_interrupt,
1227                           IRQF_SHARED, "ioat-intx", device);
1228         if (err)
1229                 goto err_no_irq;
1230         device->irq_mode = intx;
1231
1232 done:
1233         intrctrl |= IOAT_INTRCTRL_MASTER_INT_EN;
1234         writeb(intrctrl, device->reg_base + IOAT_INTRCTRL_OFFSET);
1235         return 0;
1236
1237 err_no_irq:
1238         /* Disable all interrupt generation */
1239         writeb(0, device->reg_base + IOAT_INTRCTRL_OFFSET);
1240         dev_err(&device->pdev->dev, "no usable interrupts\n");
1241         device->irq_mode = none;
1242         return -1;
1243 }
1244
1245 /**
1246  * ioat_dma_remove_interrupts - remove whatever interrupts were set
1247  * @device: ioat device
1248  */
1249 static void ioat_dma_remove_interrupts(struct ioatdma_device *device)
1250 {
1251         struct ioat_dma_chan *ioat_chan;
1252         int i;
1253
1254         /* Disable all interrupt generation */
1255         writeb(0, device->reg_base + IOAT_INTRCTRL_OFFSET);
1256
1257         switch (device->irq_mode) {
1258         case msix_multi_vector:
1259                 for (i = 0; i < device->common.chancnt; i++) {
1260                         ioat_chan = ioat_lookup_chan_by_index(device, i);
1261                         free_irq(device->msix_entries[i].vector, ioat_chan);
1262                 }
1263                 pci_disable_msix(device->pdev);
1264                 break;
1265         case msix_single_vector:
1266                 free_irq(device->msix_entries[0].vector, device);
1267                 pci_disable_msix(device->pdev);
1268                 break;
1269         case msi:
1270                 free_irq(device->pdev->irq, device);
1271                 pci_disable_msi(device->pdev);
1272                 break;
1273         case intx:
1274                 free_irq(device->pdev->irq, device);
1275                 break;
1276         case none:
1277                 dev_warn(&device->pdev->dev,
1278                          "call to %s without interrupts setup\n", __func__);
1279         }
1280         device->irq_mode = none;
1281 }
1282
1283 struct ioatdma_device *ioat_dma_probe(struct pci_dev *pdev,
1284                                       void __iomem *iobase)
1285 {
1286         int err;
1287         struct ioatdma_device *device;
1288
1289         device = kzalloc(sizeof(*device), GFP_KERNEL);
1290         if (!device) {
1291                 err = -ENOMEM;
1292                 goto err_kzalloc;
1293         }
1294         device->pdev = pdev;
1295         device->reg_base = iobase;
1296         device->version = readb(device->reg_base + IOAT_VER_OFFSET);
1297
1298         /* DMA coherent memory pool for DMA descriptor allocations */
1299         device->dma_pool = pci_pool_create("dma_desc_pool", pdev,
1300                                            sizeof(struct ioat_dma_descriptor),
1301                                            64, 0);
1302         if (!device->dma_pool) {
1303                 err = -ENOMEM;
1304                 goto err_dma_pool;
1305         }
1306
1307         device->completion_pool = pci_pool_create("completion_pool", pdev,
1308                                                   sizeof(u64), SMP_CACHE_BYTES,
1309                                                   SMP_CACHE_BYTES);
1310         if (!device->completion_pool) {
1311                 err = -ENOMEM;
1312                 goto err_completion_pool;
1313         }
1314
1315         INIT_LIST_HEAD(&device->common.channels);
1316         ioat_dma_enumerate_channels(device);
1317
1318         device->common.device_alloc_chan_resources =
1319                                                 ioat_dma_alloc_chan_resources;
1320         device->common.device_free_chan_resources =
1321                                                 ioat_dma_free_chan_resources;
1322         device->common.dev = &pdev->dev;
1323
1324         dma_cap_set(DMA_MEMCPY, device->common.cap_mask);
1325         device->common.device_is_tx_complete = ioat_dma_is_complete;
1326         device->common.device_dependency_added = ioat_dma_dependency_added;
1327         switch (device->version) {
1328         case IOAT_VER_1_2:
1329                 device->common.device_prep_dma_memcpy = ioat1_dma_prep_memcpy;
1330                 device->common.device_issue_pending =
1331                                                 ioat1_dma_memcpy_issue_pending;
1332                 break;
1333         case IOAT_VER_2_0:
1334                 device->common.device_prep_dma_memcpy = ioat2_dma_prep_memcpy;
1335                 device->common.device_issue_pending =
1336                                                 ioat2_dma_memcpy_issue_pending;
1337                 break;
1338         }
1339
1340         dev_err(&device->pdev->dev,
1341                 "Intel(R) I/OAT DMA Engine found,"
1342                 " %d channels, device version 0x%02x, driver version %s\n",
1343                 device->common.chancnt, device->version, IOAT_DMA_VERSION);
1344
1345         err = ioat_dma_setup_interrupts(device);
1346         if (err)
1347                 goto err_setup_interrupts;
1348
1349         err = ioat_dma_self_test(device);
1350         if (err)
1351                 goto err_self_test;
1352
1353         dma_async_device_register(&device->common);
1354
1355         return device;
1356
1357 err_self_test:
1358         ioat_dma_remove_interrupts(device);
1359 err_setup_interrupts:
1360         pci_pool_destroy(device->completion_pool);
1361 err_completion_pool:
1362         pci_pool_destroy(device->dma_pool);
1363 err_dma_pool:
1364         kfree(device);
1365 err_kzalloc:
1366         dev_err(&pdev->dev,
1367                 "Intel(R) I/OAT DMA Engine initialization failed\n");
1368         return NULL;
1369 }
1370
1371 void ioat_dma_remove(struct ioatdma_device *device)
1372 {
1373         struct dma_chan *chan, *_chan;
1374         struct ioat_dma_chan *ioat_chan;
1375
1376         ioat_dma_remove_interrupts(device);
1377
1378         dma_async_device_unregister(&device->common);
1379
1380         pci_pool_destroy(device->dma_pool);
1381         pci_pool_destroy(device->completion_pool);
1382
1383         iounmap(device->reg_base);
1384         pci_release_regions(device->pdev);
1385         pci_disable_device(device->pdev);
1386
1387         list_for_each_entry_safe(chan, _chan,
1388                                  &device->common.channels, device_node) {
1389                 ioat_chan = to_ioat_chan(chan);
1390                 list_del(&chan->device_node);
1391                 kfree(ioat_chan);
1392         }
1393         kfree(device);
1394 }
1395