ath10k: else is not generally useful after a break or return
[cascardo/linux.git] / drivers / net / wireless / ath / ath10k / pci.c
index 3af462a..abd0a68 100644 (file)
@@ -59,14 +59,11 @@ MODULE_PARM_DESC(reset_mode, "0: auto, 1: warm only (default: 0)");
 
 #define QCA988X_2_0_DEVICE_ID  (0x003c)
 
-static DEFINE_PCI_DEVICE_TABLE(ath10k_pci_id_table) = {
+static const struct pci_device_id ath10k_pci_id_table[] = {
        { PCI_VDEVICE(ATHEROS, QCA988X_2_0_DEVICE_ID) }, /* PCI-E QCA988X V2 */
        {0}
 };
 
-static int ath10k_pci_diag_read_access(struct ath10k *ar, u32 address,
-                                      u32 *data);
-
 static void ath10k_pci_buffer_cleanup(struct ath10k *ar);
 static int ath10k_pci_cold_reset(struct ath10k *ar);
 static int ath10k_pci_warm_reset(struct ath10k *ar);
@@ -149,79 +146,175 @@ static const struct ce_attr host_ce_config_wlan[] = {
 static const struct ce_pipe_config target_ce_config_wlan[] = {
        /* CE0: host->target HTC control and raw streams */
        {
-               .pipenum = 0,
-               .pipedir = PIPEDIR_OUT,
-               .nentries = 32,
-               .nbytes_max = 256,
-               .flags = CE_ATTR_FLAGS,
-               .reserved = 0,
+               .pipenum = __cpu_to_le32(0),
+               .pipedir = __cpu_to_le32(PIPEDIR_OUT),
+               .nentries = __cpu_to_le32(32),
+               .nbytes_max = __cpu_to_le32(256),
+               .flags = __cpu_to_le32(CE_ATTR_FLAGS),
+               .reserved = __cpu_to_le32(0),
        },
 
        /* CE1: target->host HTT + HTC control */
        {
-               .pipenum = 1,
-               .pipedir = PIPEDIR_IN,
-               .nentries = 32,
-               .nbytes_max = 512,
-               .flags = CE_ATTR_FLAGS,
-               .reserved = 0,
+               .pipenum = __cpu_to_le32(1),
+               .pipedir = __cpu_to_le32(PIPEDIR_IN),
+               .nentries = __cpu_to_le32(32),
+               .nbytes_max = __cpu_to_le32(512),
+               .flags = __cpu_to_le32(CE_ATTR_FLAGS),
+               .reserved = __cpu_to_le32(0),
        },
 
        /* CE2: target->host WMI */
        {
-               .pipenum = 2,
-               .pipedir = PIPEDIR_IN,
-               .nentries = 32,
-               .nbytes_max = 2048,
-               .flags = CE_ATTR_FLAGS,
-               .reserved = 0,
+               .pipenum = __cpu_to_le32(2),
+               .pipedir = __cpu_to_le32(PIPEDIR_IN),
+               .nentries = __cpu_to_le32(32),
+               .nbytes_max = __cpu_to_le32(2048),
+               .flags = __cpu_to_le32(CE_ATTR_FLAGS),
+               .reserved = __cpu_to_le32(0),
        },
 
        /* CE3: host->target WMI */
        {
-               .pipenum = 3,
-               .pipedir = PIPEDIR_OUT,
-               .nentries = 32,
-               .nbytes_max = 2048,
-               .flags = CE_ATTR_FLAGS,
-               .reserved = 0,
+               .pipenum = __cpu_to_le32(3),
+               .pipedir = __cpu_to_le32(PIPEDIR_OUT),
+               .nentries = __cpu_to_le32(32),
+               .nbytes_max = __cpu_to_le32(2048),
+               .flags = __cpu_to_le32(CE_ATTR_FLAGS),
+               .reserved = __cpu_to_le32(0),
        },
 
        /* CE4: host->target HTT */
        {
-               .pipenum = 4,
-               .pipedir = PIPEDIR_OUT,
-               .nentries = 256,
-               .nbytes_max = 256,
-               .flags = CE_ATTR_FLAGS,
-               .reserved = 0,
+               .pipenum = __cpu_to_le32(4),
+               .pipedir = __cpu_to_le32(PIPEDIR_OUT),
+               .nentries = __cpu_to_le32(256),
+               .nbytes_max = __cpu_to_le32(256),
+               .flags = __cpu_to_le32(CE_ATTR_FLAGS),
+               .reserved = __cpu_to_le32(0),
        },
 
        /* NB: 50% of src nentries, since tx has 2 frags */
 
        /* CE5: unused */
        {
-               .pipenum = 5,
-               .pipedir = PIPEDIR_OUT,
-               .nentries = 32,
-               .nbytes_max = 2048,
-               .flags = CE_ATTR_FLAGS,
-               .reserved = 0,
+               .pipenum = __cpu_to_le32(5),
+               .pipedir = __cpu_to_le32(PIPEDIR_OUT),
+               .nentries = __cpu_to_le32(32),
+               .nbytes_max = __cpu_to_le32(2048),
+               .flags = __cpu_to_le32(CE_ATTR_FLAGS),
+               .reserved = __cpu_to_le32(0),
        },
 
        /* CE6: Reserved for target autonomous hif_memcpy */
        {
-               .pipenum = 6,
-               .pipedir = PIPEDIR_INOUT,
-               .nentries = 32,
-               .nbytes_max = 4096,
-               .flags = CE_ATTR_FLAGS,
-               .reserved = 0,
+               .pipenum = __cpu_to_le32(6),
+               .pipedir = __cpu_to_le32(PIPEDIR_INOUT),
+               .nentries = __cpu_to_le32(32),
+               .nbytes_max = __cpu_to_le32(4096),
+               .flags = __cpu_to_le32(CE_ATTR_FLAGS),
+               .reserved = __cpu_to_le32(0),
        },
 
        /* CE7 used only by Host */
 };
 
+/*
+ * Map from service/endpoint to Copy Engine.
+ * This table is derived from the CE_PCI TABLE, above.
+ * It is passed to the Target at startup for use by firmware.
+ */
+static const struct service_to_pipe target_service_to_ce_map_wlan[] = {
+       {
+               __cpu_to_le32(ATH10K_HTC_SVC_ID_WMI_DATA_VO),
+               __cpu_to_le32(PIPEDIR_OUT),     /* out = UL = host -> target */
+               __cpu_to_le32(3),
+       },
+       {
+               __cpu_to_le32(ATH10K_HTC_SVC_ID_WMI_DATA_VO),
+               __cpu_to_le32(PIPEDIR_IN),      /* in = DL = target -> host */
+               __cpu_to_le32(2),
+       },
+       {
+               __cpu_to_le32(ATH10K_HTC_SVC_ID_WMI_DATA_BK),
+               __cpu_to_le32(PIPEDIR_OUT),     /* out = UL = host -> target */
+               __cpu_to_le32(3),
+       },
+       {
+               __cpu_to_le32(ATH10K_HTC_SVC_ID_WMI_DATA_BK),
+               __cpu_to_le32(PIPEDIR_IN),      /* in = DL = target -> host */
+               __cpu_to_le32(2),
+       },
+       {
+               __cpu_to_le32(ATH10K_HTC_SVC_ID_WMI_DATA_BE),
+               __cpu_to_le32(PIPEDIR_OUT),     /* out = UL = host -> target */
+               __cpu_to_le32(3),
+       },
+       {
+               __cpu_to_le32(ATH10K_HTC_SVC_ID_WMI_DATA_BE),
+               __cpu_to_le32(PIPEDIR_IN),      /* in = DL = target -> host */
+               __cpu_to_le32(2),
+       },
+       {
+               __cpu_to_le32(ATH10K_HTC_SVC_ID_WMI_DATA_VI),
+               __cpu_to_le32(PIPEDIR_OUT),     /* out = UL = host -> target */
+               __cpu_to_le32(3),
+       },
+       {
+               __cpu_to_le32(ATH10K_HTC_SVC_ID_WMI_DATA_VI),
+               __cpu_to_le32(PIPEDIR_IN),      /* in = DL = target -> host */
+               __cpu_to_le32(2),
+       },
+       {
+               __cpu_to_le32(ATH10K_HTC_SVC_ID_WMI_CONTROL),
+               __cpu_to_le32(PIPEDIR_OUT),     /* out = UL = host -> target */
+               __cpu_to_le32(3),
+       },
+       {
+               __cpu_to_le32(ATH10K_HTC_SVC_ID_WMI_CONTROL),
+               __cpu_to_le32(PIPEDIR_IN),      /* in = DL = target -> host */
+               __cpu_to_le32(2),
+       },
+       {
+               __cpu_to_le32(ATH10K_HTC_SVC_ID_RSVD_CTRL),
+               __cpu_to_le32(PIPEDIR_OUT),     /* out = UL = host -> target */
+               __cpu_to_le32(0),
+       },
+       {
+               __cpu_to_le32(ATH10K_HTC_SVC_ID_RSVD_CTRL),
+               __cpu_to_le32(PIPEDIR_IN),      /* in = DL = target -> host */
+               __cpu_to_le32(1),
+       },
+       { /* not used */
+               __cpu_to_le32(ATH10K_HTC_SVC_ID_TEST_RAW_STREAMS),
+               __cpu_to_le32(PIPEDIR_OUT),     /* out = UL = host -> target */
+               __cpu_to_le32(0),
+       },
+       { /* not used */
+               __cpu_to_le32(ATH10K_HTC_SVC_ID_TEST_RAW_STREAMS),
+               __cpu_to_le32(PIPEDIR_IN),      /* in = DL = target -> host */
+               __cpu_to_le32(1),
+       },
+       {
+               __cpu_to_le32(ATH10K_HTC_SVC_ID_HTT_DATA_MSG),
+               __cpu_to_le32(PIPEDIR_OUT),     /* out = UL = host -> target */
+               __cpu_to_le32(4),
+       },
+       {
+               __cpu_to_le32(ATH10K_HTC_SVC_ID_HTT_DATA_MSG),
+               __cpu_to_le32(PIPEDIR_IN),      /* in = DL = target -> host */
+               __cpu_to_le32(1),
+       },
+
+       /* (Additions here) */
+
+       { /* must be last */
+               __cpu_to_le32(0),
+               __cpu_to_le32(0),
+               __cpu_to_le32(0),
+       },
+};
+
 static bool ath10k_pci_irq_pending(struct ath10k *ar)
 {
        u32 cause;
@@ -247,8 +340,8 @@ static void ath10k_pci_disable_and_clear_legacy_irq(struct ath10k *ar)
 
        /* IMPORTANT: this extra read transaction is required to
         * flush the posted write buffer. */
-       (void) ath10k_pci_read32(ar, SOC_CORE_BASE_ADDRESS +
-                                PCIE_INTR_ENABLE_ADDRESS);
+       (void)ath10k_pci_read32(ar, SOC_CORE_BASE_ADDRESS +
+                               PCIE_INTR_ENABLE_ADDRESS);
 }
 
 static void ath10k_pci_enable_legacy_irq(struct ath10k *ar)
@@ -259,8 +352,8 @@ static void ath10k_pci_enable_legacy_irq(struct ath10k *ar)
 
        /* IMPORTANT: this extra read transaction is required to
         * flush the posted write buffer. */
-       (void) ath10k_pci_read32(ar, SOC_CORE_BASE_ADDRESS +
-                                PCIE_INTR_ENABLE_ADDRESS);
+       (void)ath10k_pci_read32(ar, SOC_CORE_BASE_ADDRESS +
+                               PCIE_INTR_ENABLE_ADDRESS);
 }
 
 static inline const char *ath10k_pci_get_irq_method(struct ath10k *ar)
@@ -269,10 +362,11 @@ static inline const char *ath10k_pci_get_irq_method(struct ath10k *ar)
 
        if (ar_pci->num_msi_intrs > 1)
                return "msi-x";
-       else if (ar_pci->num_msi_intrs == 1)
+
+       if (ar_pci->num_msi_intrs == 1)
                return "msi";
-       else
-               return "legacy";
+
+       return "legacy";
 }
 
 static int __ath10k_pci_rx_post_buf(struct ath10k_pci_pipe *pipe)
@@ -296,7 +390,7 @@ static int __ath10k_pci_rx_post_buf(struct ath10k_pci_pipe *pipe)
                               skb->len + skb_tailroom(skb),
                               DMA_FROM_DEVICE);
        if (unlikely(dma_mapping_error(ar->dev, paddr))) {
-               ath10k_warn("failed to dma map pci rx buf\n");
+               ath10k_warn(ar, "failed to dma map pci rx buf\n");
                dev_kfree_skb_any(skb);
                return -EIO;
        }
@@ -305,7 +399,7 @@ static int __ath10k_pci_rx_post_buf(struct ath10k_pci_pipe *pipe)
 
        ret = __ath10k_ce_rx_post_buf(ce_pipe, skb, paddr);
        if (ret) {
-               ath10k_warn("failed to post pci rx buf: %d\n", ret);
+               ath10k_warn(ar, "failed to post pci rx buf: %d\n", ret);
                dma_unmap_single(ar->dev, paddr, skb->len + skb_tailroom(skb),
                                 DMA_FROM_DEVICE);
                dev_kfree_skb_any(skb);
@@ -334,7 +428,7 @@ static void __ath10k_pci_rx_post_pipe(struct ath10k_pci_pipe *pipe)
        while (num--) {
                ret = __ath10k_pci_rx_post_buf(pipe);
                if (ret) {
-                       ath10k_warn("failed to post pci rx buf: %d\n", ret);
+                       ath10k_warn(ar, "failed to post pci rx buf: %d\n", ret);
                        mod_timer(&ar_pci->rx_post_retry, jiffies +
                                  ATH10K_PCI_RX_POST_RETRY_MS);
                        break;
@@ -391,25 +485,6 @@ static int ath10k_pci_diag_read_mem(struct ath10k *ar, u32 address, void *data,
        void *data_buf = NULL;
        int i;
 
-       /*
-        * This code cannot handle reads to non-memory space. Redirect to the
-        * register read fn but preserve the multi word read capability of
-        * this fn
-        */
-       if (address < DRAM_BASE_ADDRESS) {
-               if (!IS_ALIGNED(address, 4) ||
-                   !IS_ALIGNED((unsigned long)data, 4))
-                       return -EIO;
-
-               while ((nbytes >= 4) &&  ((ret = ath10k_pci_diag_read_access(
-                                          ar, address, (u32 *)data)) == 0)) {
-                       nbytes -= sizeof(u32);
-                       address += sizeof(u32);
-                       data += sizeof(u32);
-               }
-               return ret;
-       }
-
        ce_diag = ar_pci->ce_diag;
 
        /*
@@ -453,7 +528,7 @@ static int ath10k_pci_diag_read_mem(struct ath10k *ar, u32 address, void *data,
                                                     address);
 
                ret = ath10k_ce_send(ce_diag, NULL, (u32)address, nbytes, 0,
-                                0);
+                                    0);
                if (ret)
                        goto done;
 
@@ -473,7 +548,7 @@ static int ath10k_pci_diag_read_mem(struct ath10k *ar, u32 address, void *data,
                        goto done;
                }
 
-               if (buf != (u32) address) {
+               if (buf != (u32)address) {
                        ret = -EIO;
                        goto done;
                }
@@ -506,15 +581,10 @@ static int ath10k_pci_diag_read_mem(struct ath10k *ar, u32 address, void *data,
        }
 
 done:
-       if (ret == 0) {
-               /* Copy data from allocated DMA buf to caller's buf */
-               WARN_ON_ONCE(orig_nbytes & 3);
-               for (i = 0; i < orig_nbytes / sizeof(__le32); i++) {
-                       ((u32 *)data)[i] =
-                               __le32_to_cpu(((__le32 *)data_buf)[i]);
-               }
-       } else
-               ath10k_warn("failed to read diag value at 0x%x: %d\n",
+       if (ret == 0)
+               memcpy(data, data_buf, orig_nbytes);
+       else
+               ath10k_warn(ar, "failed to read diag value at 0x%x: %d\n",
                            address, ret);
 
        if (data_buf)
@@ -526,7 +596,13 @@ done:
 
 static int ath10k_pci_diag_read32(struct ath10k *ar, u32 address, u32 *value)
 {
-       return ath10k_pci_diag_read_mem(ar, address, value, sizeof(u32));
+       __le32 val = 0;
+       int ret;
+
+       ret = ath10k_pci_diag_read_mem(ar, address, &val, sizeof(val));
+       *value = __le32_to_cpu(val);
+
+       return ret;
 }
 
 static int __ath10k_pci_diag_read_hi(struct ath10k *ar, void *dest,
@@ -539,14 +615,14 @@ static int __ath10k_pci_diag_read_hi(struct ath10k *ar, void *dest,
 
        ret = ath10k_pci_diag_read32(ar, host_addr, &addr);
        if (ret != 0) {
-               ath10k_warn("failed to get memcpy hi address for firmware address %d: %d\n",
+               ath10k_warn(ar, "failed to get memcpy hi address for firmware address %d: %d\n",
                            src, ret);
                return ret;
        }
 
        ret = ath10k_pci_diag_read_mem(ar, addr, dest, len);
        if (ret != 0) {
-               ath10k_warn("failed to memcpy firmware memory from %d (%d B): %d\n",
+               ath10k_warn(ar, "failed to memcpy firmware memory from %d (%d B): %d\n",
                            addr, len, ret);
                return ret;
        }
@@ -557,18 +633,6 @@ static int __ath10k_pci_diag_read_hi(struct ath10k *ar, void *dest,
 #define ath10k_pci_diag_read_hi(ar, dest, src, len)            \
        __ath10k_pci_diag_read_hi(ar, dest, HI_ITEM(src), len);
 
-/* Read 4-byte aligned data from Target memory or register */
-static int ath10k_pci_diag_read_access(struct ath10k *ar, u32 address,
-                                      u32 *data)
-{
-       /* Assume range doesn't cross this boundary */
-       if (address >= DRAM_BASE_ADDRESS)
-               return ath10k_pci_diag_read32(ar, address, data);
-
-       *data = ath10k_pci_read32(ar, address);
-       return 0;
-}
-
 static int ath10k_pci_diag_write_mem(struct ath10k *ar, u32 address,
                                     const void *data, int nbytes)
 {
@@ -603,9 +667,7 @@ static int ath10k_pci_diag_write_mem(struct ath10k *ar, u32 address,
        }
 
        /* Copy caller's data to allocated DMA buf */
-       WARN_ON_ONCE(orig_nbytes & 3);
-       for (i = 0; i < orig_nbytes / sizeof(__le32); i++)
-               ((__le32 *)data_buf)[i] = __cpu_to_le32(((u32 *)data)[i]);
+       memcpy(data_buf, data, orig_nbytes);
 
        /*
         * The address supplied by the caller is in the
@@ -634,7 +696,7 @@ static int ath10k_pci_diag_write_mem(struct ath10k *ar, u32 address,
                 * Request CE to send caller-supplied data that
                 * was copied to bounce buffer to Target(!) address.
                 */
-               ret = ath10k_ce_send(ce_diag, NULL, (u32) ce_data,
+               ret = ath10k_ce_send(ce_diag, NULL, (u32)ce_data,
                                     nbytes, 0, 0);
                if (ret != 0)
                        goto done;
@@ -695,23 +757,17 @@ done:
        }
 
        if (ret != 0)
-               ath10k_warn("failed to write diag value at 0x%x: %d\n",
+               ath10k_warn(ar, "failed to write diag value at 0x%x: %d\n",
                            address, ret);
 
        return ret;
 }
 
-/* Write 4B data to Target memory or register */
-static int ath10k_pci_diag_write_access(struct ath10k *ar, u32 address,
-                                       u32 data)
+static int ath10k_pci_diag_write32(struct ath10k *ar, u32 address, u32 value)
 {
-       /* Assume range doesn't cross this boundary */
-       if (address >= DRAM_BASE_ADDRESS)
-               return ath10k_pci_diag_write_mem(ar, address, &data,
-                                                sizeof(u32));
+       __le32 val = __cpu_to_le32(value);
 
-       ath10k_pci_write32(ar, address, data);
-       return 0;
+       return ath10k_pci_diag_write_mem(ar, address, &val, sizeof(val));
 }
 
 static bool ath10k_pci_is_awake(struct ath10k *ar)
@@ -798,7 +854,7 @@ static void ath10k_pci_ce_recv_data(struct ath10k_ce_pipe *ce_state)
                                 max_nbytes, DMA_FROM_DEVICE);
 
                if (unlikely(max_nbytes < nbytes)) {
-                       ath10k_warn("rxed more than expected (nbytes %d, max %d)",
+                       ath10k_warn(ar, "rxed more than expected (nbytes %d, max %d)",
                                    nbytes, max_nbytes);
                        dev_kfree_skb_any(skb);
                        continue;
@@ -836,10 +892,10 @@ static int ath10k_pci_hif_tx_sg(struct ath10k *ar, u8 pipe_id,
        }
 
        for (i = 0; i < n_items - 1; i++) {
-               ath10k_dbg(ATH10K_DBG_PCI,
+               ath10k_dbg(ar, ATH10K_DBG_PCI,
                           "pci tx item %d paddr 0x%08x len %d n_items %d\n",
                           i, items[i].paddr, items[i].len, n_items);
-               ath10k_dbg_dump(ATH10K_DBG_PCI_DUMP, NULL, "item data: ",
+               ath10k_dbg_dump(ar, ATH10K_DBG_PCI_DUMP, NULL, "pci tx data: ",
                                items[i].vaddr, items[i].len);
 
                err = ath10k_ce_send_nolock(ce_pipe,
@@ -854,10 +910,10 @@ static int ath10k_pci_hif_tx_sg(struct ath10k *ar, u8 pipe_id,
 
        /* `i` is equal to `n_items -1` after for() */
 
-       ath10k_dbg(ATH10K_DBG_PCI,
+       ath10k_dbg(ar, ATH10K_DBG_PCI,
                   "pci tx item %d paddr 0x%08x len %d n_items %d\n",
                   i, items[i].paddr, items[i].len, n_items);
-       ath10k_dbg_dump(ATH10K_DBG_PCI_DUMP, NULL, "item data: ",
+       ath10k_dbg_dump(ar, ATH10K_DBG_PCI_DUMP, NULL, "pci tx data: ",
                        items[i].vaddr, items[i].len);
 
        err = ath10k_ce_send_nolock(ce_pipe,
@@ -884,7 +940,7 @@ static u16 ath10k_pci_hif_get_free_queue_number(struct ath10k *ar, u8 pipe)
 {
        struct ath10k_pci *ar_pci = ath10k_pci_priv(ar);
 
-       ath10k_dbg(ATH10K_DBG_PCI, "pci hif get free queue number\n");
+       ath10k_dbg(ar, ATH10K_DBG_PCI, "pci hif get free queue number\n");
 
        return ath10k_ce_num_free_src_entries(ar_pci->pipe_info[pipe].ce_hdl);
 }
@@ -892,33 +948,35 @@ static u16 ath10k_pci_hif_get_free_queue_number(struct ath10k *ar, u8 pipe)
 static void ath10k_pci_dump_registers(struct ath10k *ar,
                                      struct ath10k_fw_crash_data *crash_data)
 {
-       u32 i, reg_dump_values[REG_DUMP_COUNT_QCA988X] = {};
-       int ret;
+       __le32 reg_dump_values[REG_DUMP_COUNT_QCA988X] = {};
+       int i, ret;
 
        lockdep_assert_held(&ar->data_lock);
 
        ret = ath10k_pci_diag_read_hi(ar, &reg_dump_values[0],
                                      hi_failure_state,
-                                     REG_DUMP_COUNT_QCA988X * sizeof(u32));
+                                     REG_DUMP_COUNT_QCA988X * sizeof(__le32));
        if (ret) {
-               ath10k_err("failed to read firmware dump area: %d\n", ret);
+               ath10k_err(ar, "failed to read firmware dump area: %d\n", ret);
                return;
        }
 
        BUILD_BUG_ON(REG_DUMP_COUNT_QCA988X % 4);
 
-       ath10k_err("firmware register dump:\n");
+       ath10k_err(ar, "firmware register dump:\n");
        for (i = 0; i < REG_DUMP_COUNT_QCA988X; i += 4)
-               ath10k_err("[%02d]: 0x%08X 0x%08X 0x%08X 0x%08X\n",
+               ath10k_err(ar, "[%02d]: 0x%08X 0x%08X 0x%08X 0x%08X\n",
                           i,
-                          reg_dump_values[i],
-                          reg_dump_values[i + 1],
-                          reg_dump_values[i + 2],
-                          reg_dump_values[i + 3]);
+                          __le32_to_cpu(reg_dump_values[i]),
+                          __le32_to_cpu(reg_dump_values[i + 1]),
+                          __le32_to_cpu(reg_dump_values[i + 2]),
+                          __le32_to_cpu(reg_dump_values[i + 3]));
+
+       if (!crash_data)
+               return;
 
-       /* crash_data is in little endian */
        for (i = 0; i < REG_DUMP_COUNT_QCA988X; i++)
-               crash_data->registers[i] = cpu_to_le32(reg_dump_values[i]);
+               crash_data->registers[i] = reg_dump_values[i];
 }
 
 static void ath10k_pci_fw_crashed_dump(struct ath10k *ar)
@@ -935,15 +993,10 @@ static void ath10k_pci_fw_crashed_dump(struct ath10k *ar)
        else
                scnprintf(uuid, sizeof(uuid), "n/a");
 
-       ath10k_err("firmware crashed! (uuid %s)\n", uuid);
+       ath10k_err(ar, "firmware crashed! (uuid %s)\n", uuid);
        ath10k_print_driver_info(ar);
-
-       if (!crash_data)
-               goto exit;
-
        ath10k_pci_dump_registers(ar, crash_data);
 
-exit:
        spin_unlock_bh(&ar->data_lock);
 
        queue_work(ar->workqueue, &ar->restart_work);
@@ -952,7 +1005,7 @@ exit:
 static void ath10k_pci_hif_send_complete_check(struct ath10k *ar, u8 pipe,
                                               int force)
 {
-       ath10k_dbg(ATH10K_DBG_PCI, "pci hif send complete check\n");
+       ath10k_dbg(ar, ATH10K_DBG_PCI, "pci hif send complete check\n");
 
        if (!force) {
                int resources;
@@ -980,7 +1033,7 @@ static void ath10k_pci_hif_set_callbacks(struct ath10k *ar,
 {
        struct ath10k_pci *ar_pci = ath10k_pci_priv(ar);
 
-       ath10k_dbg(ATH10K_DBG_PCI, "pci hif set callbacks\n");
+       ath10k_dbg(ar, ATH10K_DBG_PCI, "pci hif set callbacks\n");
 
        memcpy(&ar_pci->msg_callbacks_current, callbacks,
               sizeof(ar_pci->msg_callbacks_current));
@@ -1000,76 +1053,65 @@ static void ath10k_pci_kill_tasklet(struct ath10k *ar)
        del_timer_sync(&ar_pci->rx_post_retry);
 }
 
-/* TODO - temporary mapping while we have too few CE's */
 static int ath10k_pci_hif_map_service_to_pipe(struct ath10k *ar,
                                              u16 service_id, u8 *ul_pipe,
                                              u8 *dl_pipe, int *ul_is_polled,
                                              int *dl_is_polled)
 {
-       int ret = 0;
+       const struct service_to_pipe *entry;
+       bool ul_set = false, dl_set = false;
+       int i;
 
-       ath10k_dbg(ATH10K_DBG_PCI, "pci hif map service\n");
+       ath10k_dbg(ar, ATH10K_DBG_PCI, "pci hif map service\n");
 
        /* polling for received messages not supported */
        *dl_is_polled = 0;
 
-       switch (service_id) {
-       case ATH10K_HTC_SVC_ID_HTT_DATA_MSG:
-               /*
-                * Host->target HTT gets its own pipe, so it can be polled
-                * while other pipes are interrupt driven.
-                */
-               *ul_pipe = 4;
-               /*
-                * Use the same target->host pipe for HTC ctrl, HTC raw
-                * streams, and HTT.
-                */
-               *dl_pipe = 1;
-               break;
+       for (i = 0; i < ARRAY_SIZE(target_service_to_ce_map_wlan); i++) {
+               entry = &target_service_to_ce_map_wlan[i];
 
-       case ATH10K_HTC_SVC_ID_RSVD_CTRL:
-       case ATH10K_HTC_SVC_ID_TEST_RAW_STREAMS:
-               /*
-                * Note: HTC_RAW_STREAMS_SVC is currently unused, and
-                * HTC_CTRL_RSVD_SVC could share the same pipe as the
-                * WMI services.  So, if another CE is needed, change
-                * this to *ul_pipe = 3, which frees up CE 0.
-                */
-               /* *ul_pipe = 3; */
-               *ul_pipe = 0;
-               *dl_pipe = 1;
-               break;
-
-       case ATH10K_HTC_SVC_ID_WMI_DATA_BK:
-       case ATH10K_HTC_SVC_ID_WMI_DATA_BE:
-       case ATH10K_HTC_SVC_ID_WMI_DATA_VI:
-       case ATH10K_HTC_SVC_ID_WMI_DATA_VO:
+               if (__le32_to_cpu(entry->service_id) != service_id)
+                       continue;
 
-       case ATH10K_HTC_SVC_ID_WMI_CONTROL:
-               *ul_pipe = 3;
-               *dl_pipe = 2;
-               break;
+               switch (__le32_to_cpu(entry->pipedir)) {
+               case PIPEDIR_NONE:
+                       break;
+               case PIPEDIR_IN:
+                       WARN_ON(dl_set);
+                       *dl_pipe = __le32_to_cpu(entry->pipenum);
+                       dl_set = true;
+                       break;
+               case PIPEDIR_OUT:
+                       WARN_ON(ul_set);
+                       *ul_pipe = __le32_to_cpu(entry->pipenum);
+                       ul_set = true;
+                       break;
+               case PIPEDIR_INOUT:
+                       WARN_ON(dl_set);
+                       WARN_ON(ul_set);
+                       *dl_pipe = __le32_to_cpu(entry->pipenum);
+                       *ul_pipe = __le32_to_cpu(entry->pipenum);
+                       dl_set = true;
+                       ul_set = true;
+                       break;
+               }
+       }
 
-               /* pipe 5 unused   */
-               /* pipe 6 reserved */
-               /* pipe 7 reserved */
+       if (WARN_ON(!ul_set || !dl_set))
+               return -ENOENT;
 
-       default:
-               ret = -1;
-               break;
-       }
        *ul_is_polled =
                (host_ce_config_wlan[*ul_pipe].flags & CE_ATTR_DIS_INTR) != 0;
 
-       return ret;
+       return 0;
 }
 
 static void ath10k_pci_hif_get_default_pipe(struct ath10k *ar,
-                                               u8 *ul_pipe, u8 *dl_pipe)
+                                           u8 *ul_pipe, u8 *dl_pipe)
 {
        int ul_is_polled, dl_is_polled;
 
-       ath10k_dbg(ATH10K_DBG_PCI, "pci hif get default pipe\n");
+       ath10k_dbg(ar, ATH10K_DBG_PCI, "pci hif get default pipe\n");
 
        (void)ath10k_pci_hif_map_service_to_pipe(ar,
                                                 ATH10K_HTC_SVC_ID_RSVD_CTRL,
@@ -1085,16 +1127,8 @@ static void ath10k_pci_irq_disable(struct ath10k *ar)
        int i;
 
        ath10k_ce_disable_interrupts(ar);
-
-       /* Regardless how many interrupts were assigned for MSI the first one
-        * is always used for firmware indications (crashes). There's no way to
-        * mask the irq in the device so call disable_irq(). Legacy (shared)
-        * interrupts can be masked on the device though.
-        */
-       if (ar_pci->num_msi_intrs > 0)
-               disable_irq(ar_pci->pdev->irq);
-       else
-               ath10k_pci_disable_and_clear_legacy_irq(ar);
+       ath10k_pci_disable_and_clear_legacy_irq(ar);
+       /* FIXME: How to mask all MSI interrupts? */
 
        for (i = 0; i < max(1, ar_pci->num_msi_intrs); i++)
                synchronize_irq(ar_pci->pdev->irq + i);
@@ -1102,20 +1136,14 @@ static void ath10k_pci_irq_disable(struct ath10k *ar)
 
 static void ath10k_pci_irq_enable(struct ath10k *ar)
 {
-       struct ath10k_pci *ar_pci = ath10k_pci_priv(ar);
-
        ath10k_ce_enable_interrupts(ar);
-
-       /* See comment in ath10k_pci_irq_disable() */
-       if (ar_pci->num_msi_intrs > 0)
-               enable_irq(ar_pci->pdev->irq);
-       else
-               ath10k_pci_enable_legacy_irq(ar);
+       ath10k_pci_enable_legacy_irq(ar);
+       /* FIXME: How to unmask all MSI interrupts? */
 }
 
 static int ath10k_pci_hif_start(struct ath10k *ar)
 {
-       ath10k_dbg(ATH10K_DBG_BOOT, "boot hif start\n");
+       ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot hif start\n");
 
        ath10k_pci_irq_enable(ar);
        ath10k_pci_rx_post(ar);
@@ -1222,16 +1250,23 @@ static void ath10k_pci_flush(struct ath10k *ar)
 
 static void ath10k_pci_hif_stop(struct ath10k *ar)
 {
-       ath10k_dbg(ATH10K_DBG_BOOT, "boot hif stop\n");
-
-       ath10k_pci_irq_disable(ar);
-       ath10k_pci_flush(ar);
+       ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot hif stop\n");
 
        /* Most likely the device has HTT Rx ring configured. The only way to
         * prevent the device from accessing (and possible corrupting) host
         * memory is to reset the chip now.
+        *
+        * There's also no known way of masking MSI interrupts on the device.
+        * For ranged MSI the CE-related interrupts can be masked. However
+        * regardless how many MSI interrupts are assigned the first one
+        * is always used for firmware indications (crashes) and cannot be
+        * masked. To prevent the device from asserting the interrupt reset it
+        * before proceeding with cleanup.
         */
        ath10k_pci_warm_reset(ar);
+
+       ath10k_pci_irq_disable(ar);
+       ath10k_pci_flush(ar);
 }
 
 static int ath10k_pci_hif_exchange_bmi_msg(struct ath10k *ar,
@@ -1340,6 +1375,7 @@ static void ath10k_pci_bmi_send_done(struct ath10k_ce_pipe *ce_state)
 
 static void ath10k_pci_bmi_recv_data(struct ath10k_ce_pipe *ce_state)
 {
+       struct ath10k *ar = ce_state->ar;
        struct bmi_xfer *xfer;
        u32 ce_data;
        unsigned int nbytes;
@@ -1351,7 +1387,7 @@ static void ath10k_pci_bmi_recv_data(struct ath10k_ce_pipe *ce_state)
                return;
 
        if (!xfer->wait_for_resp) {
-               ath10k_warn("unexpected: BMI data received; ignoring\n");
+               ath10k_warn(ar, "unexpected: BMI data received; ignoring\n");
                return;
        }
 
@@ -1378,130 +1414,18 @@ static int ath10k_pci_bmi_wait(struct ath10k_ce_pipe *tx_pipe,
        return -ETIMEDOUT;
 }
 
-/*
- * Map from service/endpoint to Copy Engine.
- * This table is derived from the CE_PCI TABLE, above.
- * It is passed to the Target at startup for use by firmware.
- */
-static const struct service_to_pipe target_service_to_ce_map_wlan[] = {
-       {
-                ATH10K_HTC_SVC_ID_WMI_DATA_VO,
-                PIPEDIR_OUT,           /* out = UL = host -> target */
-                3,
-       },
-       {
-                ATH10K_HTC_SVC_ID_WMI_DATA_VO,
-                PIPEDIR_IN,            /* in = DL = target -> host */
-                2,
-       },
-       {
-                ATH10K_HTC_SVC_ID_WMI_DATA_BK,
-                PIPEDIR_OUT,           /* out = UL = host -> target */
-                3,
-       },
-       {
-                ATH10K_HTC_SVC_ID_WMI_DATA_BK,
-                PIPEDIR_IN,            /* in = DL = target -> host */
-                2,
-       },
-       {
-                ATH10K_HTC_SVC_ID_WMI_DATA_BE,
-                PIPEDIR_OUT,           /* out = UL = host -> target */
-                3,
-       },
-       {
-                ATH10K_HTC_SVC_ID_WMI_DATA_BE,
-                PIPEDIR_IN,            /* in = DL = target -> host */
-                2,
-       },
-       {
-                ATH10K_HTC_SVC_ID_WMI_DATA_VI,
-                PIPEDIR_OUT,           /* out = UL = host -> target */
-                3,
-       },
-       {
-                ATH10K_HTC_SVC_ID_WMI_DATA_VI,
-                PIPEDIR_IN,            /* in = DL = target -> host */
-                2,
-       },
-       {
-                ATH10K_HTC_SVC_ID_WMI_CONTROL,
-                PIPEDIR_OUT,           /* out = UL = host -> target */
-                3,
-       },
-       {
-                ATH10K_HTC_SVC_ID_WMI_CONTROL,
-                PIPEDIR_IN,            /* in = DL = target -> host */
-                2,
-       },
-       {
-                ATH10K_HTC_SVC_ID_RSVD_CTRL,
-                PIPEDIR_OUT,           /* out = UL = host -> target */
-                0,             /* could be moved to 3 (share with WMI) */
-       },
-       {
-                ATH10K_HTC_SVC_ID_RSVD_CTRL,
-                PIPEDIR_IN,            /* in = DL = target -> host */
-                1,
-       },
-       {
-                ATH10K_HTC_SVC_ID_TEST_RAW_STREAMS,    /* not currently used */
-                PIPEDIR_OUT,           /* out = UL = host -> target */
-                0,
-       },
-       {
-                ATH10K_HTC_SVC_ID_TEST_RAW_STREAMS,    /* not currently used */
-                PIPEDIR_IN,            /* in = DL = target -> host */
-                1,
-       },
-       {
-                ATH10K_HTC_SVC_ID_HTT_DATA_MSG,
-                PIPEDIR_OUT,           /* out = UL = host -> target */
-                4,
-       },
-       {
-                ATH10K_HTC_SVC_ID_HTT_DATA_MSG,
-                PIPEDIR_IN,            /* in = DL = target -> host */
-                1,
-       },
-
-       /* (Additions here) */
-
-       {                               /* Must be last */
-                0,
-                0,
-                0,
-       },
-};
-
 /*
  * Send an interrupt to the device to wake up the Target CPU
  * so it has an opportunity to notice any changed state.
  */
 static int ath10k_pci_wake_target_cpu(struct ath10k *ar)
 {
-       int ret;
-       u32 core_ctrl;
+       u32 addr, val;
 
-       ret = ath10k_pci_diag_read_access(ar, SOC_CORE_BASE_ADDRESS |
-                                             CORE_CTRL_ADDRESS,
-                                         &core_ctrl);
-       if (ret) {
-               ath10k_warn("failed to read core_ctrl: %d\n", ret);
-               return ret;
-       }
-
-       /* A_INUM_FIRMWARE interrupt to Target CPU */
-       core_ctrl |= CORE_CTRL_CPU_INTR_MASK;
-
-       ret = ath10k_pci_diag_write_access(ar, SOC_CORE_BASE_ADDRESS |
-                                              CORE_CTRL_ADDRESS,
-                                          core_ctrl);
-       if (ret) {
-               ath10k_warn("failed to set target CPU interrupt mask: %d\n",
-                           ret);
-               return ret;
-       }
+       addr = SOC_CORE_BASE_ADDRESS | CORE_CTRL_ADDRESS;
+       val = ath10k_pci_read32(ar, addr);
+       val |= CORE_CTRL_CPU_INTR_MASK;
+       ath10k_pci_write32(ar, addr, val);
 
        return 0;
 }
@@ -1524,92 +1448,92 @@ static int ath10k_pci_init_config(struct ath10k *ar)
                host_interest_item_address(HI_ITEM(hi_interconnect_state));
 
        /* Supply Target-side CE configuration */
-       ret = ath10k_pci_diag_read_access(ar, interconnect_targ_addr,
-                                         &pcie_state_targ_addr);
+       ret = ath10k_pci_diag_read32(ar, interconnect_targ_addr,
+                                    &pcie_state_targ_addr);
        if (ret != 0) {
-               ath10k_err("Failed to get pcie state addr: %d\n", ret);
+               ath10k_err(ar, "Failed to get pcie state addr: %d\n", ret);
                return ret;
        }
 
        if (pcie_state_targ_addr == 0) {
                ret = -EIO;
-               ath10k_err("Invalid pcie state addr\n");
+               ath10k_err(ar, "Invalid pcie state addr\n");
                return ret;
        }
 
-       ret = ath10k_pci_diag_read_access(ar, pcie_state_targ_addr +
+       ret = ath10k_pci_diag_read32(ar, (pcie_state_targ_addr +
                                          offsetof(struct pcie_state,
-                                                  pipe_cfg_addr),
-                                         &pipe_cfg_targ_addr);
+                                                  pipe_cfg_addr)),
+                                    &pipe_cfg_targ_addr);
        if (ret != 0) {
-               ath10k_err("Failed to get pipe cfg addr: %d\n", ret);
+               ath10k_err(ar, "Failed to get pipe cfg addr: %d\n", ret);
                return ret;
        }
 
        if (pipe_cfg_targ_addr == 0) {
                ret = -EIO;
-               ath10k_err("Invalid pipe cfg addr\n");
+               ath10k_err(ar, "Invalid pipe cfg addr\n");
                return ret;
        }
 
        ret = ath10k_pci_diag_write_mem(ar, pipe_cfg_targ_addr,
-                                target_ce_config_wlan,
-                                sizeof(target_ce_config_wlan));
+                                       target_ce_config_wlan,
+                                       sizeof(target_ce_config_wlan));
 
        if (ret != 0) {
-               ath10k_err("Failed to write pipe cfg: %d\n", ret);
+               ath10k_err(ar, "Failed to write pipe cfg: %d\n", ret);
                return ret;
        }
 
-       ret = ath10k_pci_diag_read_access(ar, pcie_state_targ_addr +
+       ret = ath10k_pci_diag_read32(ar, (pcie_state_targ_addr +
                                          offsetof(struct pcie_state,
-                                                  svc_to_pipe_map),
-                                         &svc_to_pipe_map);
+                                                  svc_to_pipe_map)),
+                                    &svc_to_pipe_map);
        if (ret != 0) {
-               ath10k_err("Failed to get svc/pipe map: %d\n", ret);
+               ath10k_err(ar, "Failed to get svc/pipe map: %d\n", ret);
                return ret;
        }
 
        if (svc_to_pipe_map == 0) {
                ret = -EIO;
-               ath10k_err("Invalid svc_to_pipe map\n");
+               ath10k_err(ar, "Invalid svc_to_pipe map\n");
                return ret;
        }
 
        ret = ath10k_pci_diag_write_mem(ar, svc_to_pipe_map,
-                                target_service_to_ce_map_wlan,
-                                sizeof(target_service_to_ce_map_wlan));
+                                       target_service_to_ce_map_wlan,
+                                       sizeof(target_service_to_ce_map_wlan));
        if (ret != 0) {
-               ath10k_err("Failed to write svc/pipe map: %d\n", ret);
+               ath10k_err(ar, "Failed to write svc/pipe map: %d\n", ret);
                return ret;
        }
 
-       ret = ath10k_pci_diag_read_access(ar, pcie_state_targ_addr +
+       ret = ath10k_pci_diag_read32(ar, (pcie_state_targ_addr +
                                          offsetof(struct pcie_state,
-                                                  config_flags),
-                                         &pcie_config_flags);
+                                                  config_flags)),
+                                    &pcie_config_flags);
        if (ret != 0) {
-               ath10k_err("Failed to get pcie config_flags: %d\n", ret);
+               ath10k_err(ar, "Failed to get pcie config_flags: %d\n", ret);
                return ret;
        }
 
        pcie_config_flags &= ~PCIE_CONFIG_FLAG_ENABLE_L1;
 
-       ret = ath10k_pci_diag_write_mem(ar, pcie_state_targ_addr +
-                                offsetof(struct pcie_state, config_flags),
-                                &pcie_config_flags,
-                                sizeof(pcie_config_flags));
+       ret = ath10k_pci_diag_write32(ar, (pcie_state_targ_addr +
+                                          offsetof(struct pcie_state,
+                                                   config_flags)),
+                                     pcie_config_flags);
        if (ret != 0) {
-               ath10k_err("Failed to write pcie config_flags: %d\n", ret);
+               ath10k_err(ar, "Failed to write pcie config_flags: %d\n", ret);
                return ret;
        }
 
        /* configure early allocation */
        ealloc_targ_addr = host_interest_item_address(HI_ITEM(hi_early_alloc));
 
-       ret = ath10k_pci_diag_read_access(ar, ealloc_targ_addr, &ealloc_value);
+       ret = ath10k_pci_diag_read32(ar, ealloc_targ_addr, &ealloc_value);
        if (ret != 0) {
-               ath10k_err("Faile to get early alloc val: %d\n", ret);
+               ath10k_err(ar, "Faile to get early alloc val: %d\n", ret);
                return ret;
        }
 
@@ -1619,26 +1543,26 @@ static int ath10k_pci_init_config(struct ath10k *ar)
        ealloc_value |= ((1 << HI_EARLY_ALLOC_IRAM_BANKS_SHIFT) &
                         HI_EARLY_ALLOC_IRAM_BANKS_MASK);
 
-       ret = ath10k_pci_diag_write_access(ar, ealloc_targ_addr, ealloc_value);
+       ret = ath10k_pci_diag_write32(ar, ealloc_targ_addr, ealloc_value);
        if (ret != 0) {
-               ath10k_err("Failed to set early alloc val: %d\n", ret);
+               ath10k_err(ar, "Failed to set early alloc val: %d\n", ret);
                return ret;
        }
 
        /* Tell Target to proceed with initialization */
        flag2_targ_addr = host_interest_item_address(HI_ITEM(hi_option_flag2));
 
-       ret = ath10k_pci_diag_read_access(ar, flag2_targ_addr, &flag2_value);
+       ret = ath10k_pci_diag_read32(ar, flag2_targ_addr, &flag2_value);
        if (ret != 0) {
-               ath10k_err("Failed to get option val: %d\n", ret);
+               ath10k_err(ar, "Failed to get option val: %d\n", ret);
                return ret;
        }
 
        flag2_value |= HI_OPTION_EARLY_CFG_DONE;
 
-       ret = ath10k_pci_diag_write_access(ar, flag2_targ_addr, flag2_value);
+       ret = ath10k_pci_diag_write32(ar, flag2_targ_addr, flag2_value);
        if (ret != 0) {
-               ath10k_err("Failed to set option val: %d\n", ret);
+               ath10k_err(ar, "Failed to set option val: %d\n", ret);
                return ret;
        }
 
@@ -1652,7 +1576,7 @@ static int ath10k_pci_alloc_ce(struct ath10k *ar)
        for (i = 0; i < CE_COUNT; i++) {
                ret = ath10k_ce_alloc_pipe(ar, i, &host_ce_config_wlan[i]);
                if (ret) {
-                       ath10k_err("failed to allocate copy engine pipe %d: %d\n",
+                       ath10k_err(ar, "failed to allocate copy engine pipe %d: %d\n",
                                   i, ret);
                        return ret;
                }
@@ -1687,7 +1611,7 @@ static int ath10k_pci_ce_init(struct ath10k *ar)
                                          ath10k_pci_ce_send_done,
                                          ath10k_pci_ce_recv_data);
                if (ret) {
-                       ath10k_err("failed to initialize copy engine pipe %d: %d\n",
+                       ath10k_err(ar, "failed to initialize copy engine pipe %d: %d\n",
                                   pipe_num, ret);
                        return ret;
                }
@@ -1701,7 +1625,7 @@ static int ath10k_pci_ce_init(struct ath10k *ar)
                        continue;
                }
 
-               pipe_info->buf_sz = (size_t) (attr->src_sz_max);
+               pipe_info->buf_sz = (size_t)(attr->src_sz_max);
        }
 
        return 0;
@@ -1746,16 +1670,17 @@ static int ath10k_pci_warm_reset(struct ath10k *ar)
 {
        u32 val;
 
-       ath10k_dbg(ATH10K_DBG_BOOT, "boot warm reset\n");
+       ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot warm reset\n");
 
        /* debug */
        val = ath10k_pci_read32(ar, SOC_CORE_BASE_ADDRESS +
                                PCIE_INTR_CAUSE_ADDRESS);
-       ath10k_dbg(ATH10K_DBG_BOOT, "boot host cpu intr cause: 0x%08x\n", val);
+       ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot host cpu intr cause: 0x%08x\n",
+                  val);
 
        val = ath10k_pci_read32(ar, SOC_CORE_BASE_ADDRESS +
                                CPU_INTR_ADDRESS);
-       ath10k_dbg(ATH10K_DBG_BOOT, "boot target cpu intr cause: 0x%08x\n",
+       ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot target cpu intr cause: 0x%08x\n",
                   val);
 
        /* disable pending irqs */
@@ -1798,11 +1723,12 @@ static int ath10k_pci_warm_reset(struct ath10k *ar)
        /* debug */
        val = ath10k_pci_read32(ar, SOC_CORE_BASE_ADDRESS +
                                PCIE_INTR_CAUSE_ADDRESS);
-       ath10k_dbg(ATH10K_DBG_BOOT, "boot host cpu intr cause: 0x%08x\n", val);
+       ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot host cpu intr cause: 0x%08x\n",
+                  val);
 
        val = ath10k_pci_read32(ar, SOC_CORE_BASE_ADDRESS +
                                CPU_INTR_ADDRESS);
-       ath10k_dbg(ATH10K_DBG_BOOT, "boot target cpu intr cause: 0x%08x\n",
+       ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot target cpu intr cause: 0x%08x\n",
                   val);
 
        /* CPU warm reset */
@@ -1813,11 +1739,12 @@ static int ath10k_pci_warm_reset(struct ath10k *ar)
 
        val = ath10k_pci_read32(ar, RTC_SOC_BASE_ADDRESS +
                                SOC_RESET_CONTROL_ADDRESS);
-       ath10k_dbg(ATH10K_DBG_BOOT, "boot target reset state: 0x%08x\n", val);
+       ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot target reset state: 0x%08x\n",
+                  val);
 
        msleep(100);
 
-       ath10k_dbg(ATH10K_DBG_BOOT, "boot warm reset complete\n");
+       ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot warm reset complete\n");
 
        return 0;
 }
@@ -1842,31 +1769,31 @@ static int __ath10k_pci_hif_power_up(struct ath10k *ar, bool cold_reset)
                ret = ath10k_pci_warm_reset(ar);
 
        if (ret) {
-               ath10k_err("failed to reset target: %d\n", ret);
+               ath10k_err(ar, "failed to reset target: %d\n", ret);
                goto err;
        }
 
        ret = ath10k_pci_ce_init(ar);
        if (ret) {
-               ath10k_err("failed to initialize CE: %d\n", ret);
+               ath10k_err(ar, "failed to initialize CE: %d\n", ret);
                goto err;
        }
 
        ret = ath10k_pci_wait_for_target_init(ar);
        if (ret) {
-               ath10k_err("failed to wait for target to init: %d\n", ret);
+               ath10k_err(ar, "failed to wait for target to init: %d\n", ret);
                goto err_ce;
        }
 
        ret = ath10k_pci_init_config(ar);
        if (ret) {
-               ath10k_err("failed to setup init config: %d\n", ret);
+               ath10k_err(ar, "failed to setup init config: %d\n", ret);
                goto err_ce;
        }
 
        ret = ath10k_pci_wake_target_cpu(ar);
        if (ret) {
-               ath10k_err("could not wake up target CPU: %d\n", ret);
+               ath10k_err(ar, "could not wake up target CPU: %d\n", ret);
                goto err_ce;
        }
 
@@ -1894,7 +1821,7 @@ static int ath10k_pci_hif_power_up_warm(struct ath10k *ar)
                if (ret == 0)
                        break;
 
-               ath10k_warn("failed to warm reset (attempt %d out of %d): %d\n",
+               ath10k_warn(ar, "failed to warm reset (attempt %d out of %d): %d\n",
                            i + 1, ATH10K_PCI_NUM_WARM_RESET_ATTEMPTS, ret);
        }
 
@@ -1905,7 +1832,7 @@ static int ath10k_pci_hif_power_up(struct ath10k *ar)
 {
        int ret;
 
-       ath10k_dbg(ATH10K_DBG_BOOT, "boot hif power up\n");
+       ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot hif power up\n");
 
        /*
         * Hardware CUS232 version 2 has some issues with cold reset and the
@@ -1917,17 +1844,17 @@ static int ath10k_pci_hif_power_up(struct ath10k *ar)
         */
        ret = ath10k_pci_hif_power_up_warm(ar);
        if (ret) {
-               ath10k_warn("failed to power up target using warm reset: %d\n",
+               ath10k_warn(ar, "failed to power up target using warm reset: %d\n",
                            ret);
 
                if (ath10k_pci_reset_mode == ATH10K_PCI_RESET_WARM_ONLY)
                        return ret;
 
-               ath10k_warn("trying cold reset\n");
+               ath10k_warn(ar, "trying cold reset\n");
 
                ret = __ath10k_pci_hif_power_up(ar, true);
                if (ret) {
-                       ath10k_err("failed to power up target using cold reset too (%d)\n",
+                       ath10k_err(ar, "failed to power up target using cold reset too (%d)\n",
                                   ret);
                        return ret;
                }
@@ -1938,7 +1865,7 @@ static int ath10k_pci_hif_power_up(struct ath10k *ar)
 
 static void ath10k_pci_hif_power_down(struct ath10k *ar)
 {
-       ath10k_dbg(ATH10K_DBG_BOOT, "boot hif power down\n");
+       ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot hif power down\n");
 
        ath10k_pci_warm_reset(ar);
 }
@@ -2023,7 +1950,7 @@ static void ath10k_msi_err_tasklet(unsigned long data)
        struct ath10k *ar = (struct ath10k *)data;
 
        if (!ath10k_pci_has_fw_crashed(ar)) {
-               ath10k_warn("received unsolicited fw crash interrupt\n");
+               ath10k_warn(ar, "received unsolicited fw crash interrupt\n");
                return;
        }
 
@@ -2042,7 +1969,8 @@ static irqreturn_t ath10k_pci_per_engine_handler(int irq, void *arg)
        int ce_id = irq - ar_pci->pdev->irq - MSI_ASSIGN_CE_INITIAL;
 
        if (ce_id < 0 || ce_id >= ARRAY_SIZE(ar_pci->pipe_info)) {
-               ath10k_warn("unexpected/invalid irq %d ce_id %d\n", irq, ce_id);
+               ath10k_warn(ar, "unexpected/invalid irq %d ce_id %d\n", irq,
+                           ce_id);
                return IRQ_HANDLED;
        }
 
@@ -2116,7 +2044,7 @@ static int ath10k_pci_request_irq_msix(struct ath10k *ar)
                          ath10k_pci_msi_fw_handler,
                          IRQF_SHARED, "ath10k_pci", ar);
        if (ret) {
-               ath10k_warn("failed to request MSI-X fw irq %d: %d\n",
+               ath10k_warn(ar, "failed to request MSI-X fw irq %d: %d\n",
                            ar_pci->pdev->irq + MSI_ASSIGN_FW, ret);
                return ret;
        }
@@ -2126,7 +2054,7 @@ static int ath10k_pci_request_irq_msix(struct ath10k *ar)
                                  ath10k_pci_per_engine_handler,
                                  IRQF_SHARED, "ath10k_pci", ar);
                if (ret) {
-                       ath10k_warn("failed to request MSI-X ce irq %d: %d\n",
+                       ath10k_warn(ar, "failed to request MSI-X ce irq %d: %d\n",
                                    ar_pci->pdev->irq + i, ret);
 
                        for (i--; i >= MSI_ASSIGN_CE_INITIAL; i--)
@@ -2149,7 +2077,7 @@ static int ath10k_pci_request_irq_msi(struct ath10k *ar)
                          ath10k_pci_interrupt_handler,
                          IRQF_SHARED, "ath10k_pci", ar);
        if (ret) {
-               ath10k_warn("failed to request MSI irq %d: %d\n",
+               ath10k_warn(ar, "failed to request MSI irq %d: %d\n",
                            ar_pci->pdev->irq, ret);
                return ret;
        }
@@ -2166,7 +2094,7 @@ static int ath10k_pci_request_irq_legacy(struct ath10k *ar)
                          ath10k_pci_interrupt_handler,
                          IRQF_SHARED, "ath10k_pci", ar);
        if (ret) {
-               ath10k_warn("failed to request legacy irq %d: %d\n",
+               ath10k_warn(ar, "failed to request legacy irq %d: %d\n",
                            ar_pci->pdev->irq, ret);
                return ret;
        }
@@ -2187,7 +2115,7 @@ static int ath10k_pci_request_irq(struct ath10k *ar)
                return ath10k_pci_request_irq_msix(ar);
        }
 
-       ath10k_warn("unknown irq configuration upon request\n");
+       ath10k_warn(ar, "unknown irq configuration upon request\n");
        return -EINVAL;
 }
 
@@ -2226,13 +2154,14 @@ static int ath10k_pci_init_irq(struct ath10k *ar)
        ath10k_pci_init_irq_tasklets(ar);
 
        if (ath10k_pci_irq_mode != ATH10K_PCI_IRQ_AUTO)
-               ath10k_info("limiting irq mode to: %d\n", ath10k_pci_irq_mode);
+               ath10k_info(ar, "limiting irq mode to: %d\n",
+                           ath10k_pci_irq_mode);
 
        /* Try MSI-X */
        if (ath10k_pci_irq_mode == ATH10K_PCI_IRQ_AUTO) {
                ar_pci->num_msi_intrs = MSI_NUM_REQUEST;
                ret = pci_enable_msi_range(ar_pci->pdev, ar_pci->num_msi_intrs,
-                                                        ar_pci->num_msi_intrs);
+                                          ar_pci->num_msi_intrs);
                if (ret > 0)
                        return 0;
 
@@ -2288,7 +2217,7 @@ static int ath10k_pci_deinit_irq(struct ath10k *ar)
                pci_disable_msi(ar_pci->pdev);
        }
 
-       ath10k_warn("unknown irq configuration upon deinit\n");
+       ath10k_warn(ar, "unknown irq configuration upon deinit\n");
        return -EINVAL;
 }
 
@@ -2298,14 +2227,15 @@ static int ath10k_pci_wait_for_target_init(struct ath10k *ar)
        unsigned long timeout;
        u32 val;
 
-       ath10k_dbg(ATH10K_DBG_BOOT, "boot waiting target to initialise\n");
+       ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot waiting target to initialise\n");
 
        timeout = jiffies + msecs_to_jiffies(ATH10K_PCI_TARGET_WAIT);
 
        do {
                val = ath10k_pci_read32(ar, FW_INDICATOR_ADDRESS);
 
-               ath10k_dbg(ATH10K_DBG_BOOT, "boot target indicator %x\n", val);
+               ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot target indicator %x\n",
+                          val);
 
                /* target should never return this */
                if (val == 0xffffffff)
@@ -2329,24 +2259,24 @@ static int ath10k_pci_wait_for_target_init(struct ath10k *ar)
        } while (time_before(jiffies, timeout));
 
        if (val == 0xffffffff) {
-               ath10k_err("failed to read device register, device is gone\n");
+               ath10k_err(ar, "failed to read device register, device is gone\n");
                return -EIO;
        }
 
        if (val & FW_IND_EVENT_PENDING) {
-               ath10k_warn("device has crashed during init\n");
+               ath10k_warn(ar, "device has crashed during init\n");
                ath10k_pci_fw_crashed_clear(ar);
                ath10k_pci_fw_crashed_dump(ar);
                return -ECOMM;
        }
 
        if (!(val & FW_IND_INITIALIZED)) {
-               ath10k_err("failed to receive initialized event from target: %08x\n",
+               ath10k_err(ar, "failed to receive initialized event from target: %08x\n",
                           val);
                return -ETIMEDOUT;
        }
 
-       ath10k_dbg(ATH10K_DBG_BOOT, "boot target initialised\n");
+       ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot target initialised\n");
        return 0;
 }
 
@@ -2355,7 +2285,7 @@ static int ath10k_pci_cold_reset(struct ath10k *ar)
        int i;
        u32 val;
 
-       ath10k_dbg(ATH10K_DBG_BOOT, "boot cold reset\n");
+       ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot cold reset\n");
 
        /* Put Target, including PCIe, into RESET. */
        val = ath10k_pci_reg_read32(ar, SOC_GLOBAL_RESET_ADDRESS);
@@ -2380,7 +2310,7 @@ static int ath10k_pci_cold_reset(struct ath10k *ar)
                msleep(1);
        }
 
-       ath10k_dbg(ATH10K_DBG_BOOT, "boot cold reset complete\n");
+       ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot cold reset complete\n");
 
        return 0;
 }
@@ -2396,13 +2326,13 @@ static int ath10k_pci_claim(struct ath10k *ar)
 
        ret = pci_enable_device(pdev);
        if (ret) {
-               ath10k_err("failed to enable pci device: %d\n", ret);
+               ath10k_err(ar, "failed to enable pci device: %d\n", ret);
                return ret;
        }
 
        ret = pci_request_region(pdev, BAR_NUM, "ath");
        if (ret) {
-               ath10k_err("failed to request region BAR%d: %d\n", BAR_NUM,
+               ath10k_err(ar, "failed to request region BAR%d: %d\n", BAR_NUM,
                           ret);
                goto err_device;
        }
@@ -2410,13 +2340,13 @@ static int ath10k_pci_claim(struct ath10k *ar)
        /* Target expects 32 bit DMA. Enforce it. */
        ret = pci_set_dma_mask(pdev, DMA_BIT_MASK(32));
        if (ret) {
-               ath10k_err("failed to set dma mask to 32-bit: %d\n", ret);
+               ath10k_err(ar, "failed to set dma mask to 32-bit: %d\n", ret);
                goto err_region;
        }
 
        ret = pci_set_consistent_dma_mask(pdev, DMA_BIT_MASK(32));
        if (ret) {
-               ath10k_err("failed to set consistent dma mask to 32-bit: %d\n",
+               ath10k_err(ar, "failed to set consistent dma mask to 32-bit: %d\n",
                           ret);
                goto err_region;
        }
@@ -2430,12 +2360,12 @@ static int ath10k_pci_claim(struct ath10k *ar)
        /* Arrange for access to Target SoC registers. */
        ar_pci->mem = pci_iomap(pdev, BAR_NUM, 0);
        if (!ar_pci->mem) {
-               ath10k_err("failed to iomap BAR%d\n", BAR_NUM);
+               ath10k_err(ar, "failed to iomap BAR%d\n", BAR_NUM);
                ret = -EIO;
                goto err_master;
        }
 
-       ath10k_dbg(ATH10K_DBG_BOOT, "boot pci_mem 0x%p\n", ar_pci->mem);
+       ath10k_dbg(ar, ATH10K_DBG_BOOT, "boot pci_mem 0x%p\n", ar_pci->mem);
        return 0;
 
 err_master:
@@ -2469,15 +2399,15 @@ static int ath10k_pci_probe(struct pci_dev *pdev,
        struct ath10k_pci *ar_pci;
        u32 chip_id;
 
-       ath10k_dbg(ATH10K_DBG_PCI, "pci probe\n");
-
        ar = ath10k_core_create(sizeof(*ar_pci), &pdev->dev,
                                &ath10k_pci_hif_ops);
        if (!ar) {
-               ath10k_err("failed to allocate core\n");
+               dev_err(&pdev->dev, "failed to allocate core\n");
                return -ENOMEM;
        }
 
+       ath10k_dbg(ar, ATH10K_DBG_PCI, "pci probe\n");
+
        ar_pci = ath10k_pci_priv(ar);
        ar_pci->pdev = pdev;
        ar_pci->dev = &pdev->dev;
@@ -2489,25 +2419,26 @@ static int ath10k_pci_probe(struct pci_dev *pdev,
 
        ret = ath10k_pci_claim(ar);
        if (ret) {
-               ath10k_err("failed to claim device: %d\n", ret);
+               ath10k_err(ar, "failed to claim device: %d\n", ret);
                goto err_core_destroy;
        }
 
        ret = ath10k_pci_wake(ar);
        if (ret) {
-               ath10k_err("failed to wake up: %d\n", ret);
+               ath10k_err(ar, "failed to wake up: %d\n", ret);
                goto err_release;
        }
 
        chip_id = ath10k_pci_soc_read32(ar, SOC_CHIP_ID_ADDRESS);
        if (chip_id == 0xffffffff) {
-               ath10k_err("failed to get chip id\n");
+               ath10k_err(ar, "failed to get chip id\n");
                goto err_sleep;
        }
 
        ret = ath10k_pci_alloc_ce(ar);
        if (ret) {
-               ath10k_err("failed to allocate copy engine pipes: %d\n", ret);
+               ath10k_err(ar, "failed to allocate copy engine pipes: %d\n",
+                          ret);
                goto err_sleep;
        }
 
@@ -2515,7 +2446,7 @@ static int ath10k_pci_probe(struct pci_dev *pdev,
 
        ret = ath10k_ce_disable_interrupts(ar);
        if (ret) {
-               ath10k_err("failed to disable copy engine interrupts: %d\n",
+               ath10k_err(ar, "failed to disable copy engine interrupts: %d\n",
                           ret);
                goto err_free_ce;
        }
@@ -2529,17 +2460,17 @@ static int ath10k_pci_probe(struct pci_dev *pdev,
 
        ret = ath10k_pci_init_irq(ar);
        if (ret) {
-               ath10k_err("failed to init irqs: %d\n", ret);
+               ath10k_err(ar, "failed to init irqs: %d\n", ret);
                goto err_free_ce;
        }
 
-       ath10k_info("pci irq %s interrupts %d irq_mode %d reset_mode %d\n",
+       ath10k_info(ar, "pci irq %s interrupts %d irq_mode %d reset_mode %d\n",
                    ath10k_pci_get_irq_method(ar), ar_pci->num_msi_intrs,
                    ath10k_pci_irq_mode, ath10k_pci_reset_mode);
 
        ret = ath10k_pci_request_irq(ar);
        if (ret) {
-               ath10k_warn("failed to request irqs: %d\n", ret);
+               ath10k_warn(ar, "failed to request irqs: %d\n", ret);
                goto err_deinit_irq;
        }
 
@@ -2548,7 +2479,7 @@ static int ath10k_pci_probe(struct pci_dev *pdev,
 
        ret = ath10k_core_register(ar, chip_id);
        if (ret) {
-               ath10k_err("failed to register driver core: %d\n", ret);
+               ath10k_err(ar, "failed to register driver core: %d\n", ret);
                goto err_free_irq;
        }
 
@@ -2556,6 +2487,7 @@ static int ath10k_pci_probe(struct pci_dev *pdev,
 
 err_free_irq:
        ath10k_pci_free_irq(ar);
+       ath10k_pci_kill_tasklet(ar);
 
 err_deinit_irq:
        ath10k_pci_deinit_irq(ar);
@@ -2580,7 +2512,7 @@ static void ath10k_pci_remove(struct pci_dev *pdev)
        struct ath10k *ar = pci_get_drvdata(pdev);
        struct ath10k_pci *ar_pci;
 
-       ath10k_dbg(ATH10K_DBG_PCI, "pci remove\n");
+       ath10k_dbg(ar, ATH10K_DBG_PCI, "pci remove\n");
 
        if (!ar)
                return;
@@ -2592,6 +2524,7 @@ static void ath10k_pci_remove(struct pci_dev *pdev)
 
        ath10k_core_unregister(ar);
        ath10k_pci_free_irq(ar);
+       ath10k_pci_kill_tasklet(ar);
        ath10k_pci_deinit_irq(ar);
        ath10k_pci_ce_deinit(ar);
        ath10k_pci_free_ce(ar);
@@ -2615,7 +2548,8 @@ static int __init ath10k_pci_init(void)
 
        ret = pci_register_driver(&ath10k_pci_driver);
        if (ret)
-               ath10k_err("failed to register PCI driver: %d\n", ret);
+               printk(KERN_ERR "failed to register ath10k pci driver: %d\n",
+                      ret);
 
        return ret;
 }