root/drivers/net/wireless/ath/ath12k/pci.c
// SPDX-License-Identifier: BSD-3-Clause-Clear
/*
 * Copyright (c) 2019-2021 The Linux Foundation. All rights reserved.
 * Copyright (c) Qualcomm Technologies, Inc. and/or its subsidiaries.
 */

#include <linux/module.h>
#include <linux/msi.h>
#include <linux/pci.h>
#include <linux/time.h>
#include <linux/vmalloc.h>

#include "pci.h"
#include "core.h"
#include "hif.h"
#include "mhi.h"
#include "debug.h"
#include "hal.h"

#define ATH12K_PCI_BAR_NUM              0
#define ATH12K_PCI_DMA_MASK             36

#define ATH12K_PCI_IRQ_CE0_OFFSET               3

#define WINDOW_ENABLE_BIT               0x40000000
#define WINDOW_VALUE_MASK               GENMASK(24, 19)
#define WINDOW_START                    0x80000
#define WINDOW_RANGE_MASK               GENMASK(18, 0)
#define WINDOW_STATIC_MASK              GENMASK(31, 6)

/* BAR0 + 4k is always accessible, and no
 * need to force wakeup.
 * 4K - 32 = 0xFE0
 */
#define ACCESS_ALWAYS_OFF 0xFE0

static struct ath12k_pci_driver *ath12k_pci_family_drivers[ATH12K_DEVICE_FAMILY_MAX];
static const struct ath12k_msi_config msi_config_one_msi = {
        .total_vectors = 1,
        .total_users = 4,
        .users = (struct ath12k_msi_user[]) {
                { .name = "MHI", .num_vectors = 3, .base_vector = 0 },
                { .name = "CE", .num_vectors = 1, .base_vector = 0 },
                { .name = "WAKE", .num_vectors = 1, .base_vector = 0 },
                { .name = "DP", .num_vectors = 1, .base_vector = 0 },
        },
};

static const char *irq_name[ATH12K_IRQ_NUM_MAX] = {
        "bhi",
        "mhi-er0",
        "mhi-er1",
        "ce0",
        "ce1",
        "ce2",
        "ce3",
        "ce4",
        "ce5",
        "ce6",
        "ce7",
        "ce8",
        "ce9",
        "ce10",
        "ce11",
        "ce12",
        "ce13",
        "ce14",
        "ce15",
        "host2wbm-desc-feed",
        "host2reo-re-injection",
        "host2reo-command",
        "host2rxdma-monitor-ring3",
        "host2rxdma-monitor-ring2",
        "host2rxdma-monitor-ring1",
        "reo2ost-exception",
        "wbm2host-rx-release",
        "reo2host-status",
        "reo2host-destination-ring4",
        "reo2host-destination-ring3",
        "reo2host-destination-ring2",
        "reo2host-destination-ring1",
        "rxdma2host-monitor-destination-mac3",
        "rxdma2host-monitor-destination-mac2",
        "rxdma2host-monitor-destination-mac1",
        "ppdu-end-interrupts-mac3",
        "ppdu-end-interrupts-mac2",
        "ppdu-end-interrupts-mac1",
        "rxdma2host-monitor-status-ring-mac3",
        "rxdma2host-monitor-status-ring-mac2",
        "rxdma2host-monitor-status-ring-mac1",
        "host2rxdma-host-buf-ring-mac3",
        "host2rxdma-host-buf-ring-mac2",
        "host2rxdma-host-buf-ring-mac1",
        "rxdma2host-destination-ring-mac3",
        "rxdma2host-destination-ring-mac2",
        "rxdma2host-destination-ring-mac1",
        "host2tcl-input-ring4",
        "host2tcl-input-ring3",
        "host2tcl-input-ring2",
        "host2tcl-input-ring1",
        "wbm2host-tx-completions-ring4",
        "wbm2host-tx-completions-ring3",
        "wbm2host-tx-completions-ring2",
        "wbm2host-tx-completions-ring1",
        "tcl2host-status-ring",
};

static void ath12k_pci_select_window(struct ath12k_pci *ab_pci, u32 offset)
{
        struct ath12k_base *ab = ab_pci->ab;

        u32 window = u32_get_bits(offset, WINDOW_VALUE_MASK);
        u32 static_window;

        lockdep_assert_held(&ab_pci->window_lock);

        /* Preserve the static window configuration and reset only dynamic window */
        static_window = ab_pci->register_window & WINDOW_STATIC_MASK;
        window |= static_window;

        if (window != ab_pci->register_window) {
                iowrite32(WINDOW_ENABLE_BIT | window,
                          ab->mem + ab_pci->window_reg_addr);
                ioread32(ab->mem + ab_pci->window_reg_addr);
                ab_pci->register_window = window;
        }
}

static void ath12k_pci_select_static_window(struct ath12k_pci *ab_pci)
{
        u32 umac_window;
        u32 ce_window;
        u32 window;

        umac_window = u32_get_bits(ab_pci->reg_base->umac_base, WINDOW_VALUE_MASK);
        ce_window = u32_get_bits(ab_pci->reg_base->ce_reg_base, WINDOW_VALUE_MASK);
        window = (umac_window << 12) | (ce_window << 6);

        spin_lock_bh(&ab_pci->window_lock);
        ab_pci->register_window = window;
        spin_unlock_bh(&ab_pci->window_lock);

        iowrite32(WINDOW_ENABLE_BIT | window, ab_pci->ab->mem + ab_pci->window_reg_addr);
}

static u32 ath12k_pci_get_window_start(struct ath12k_base *ab,
                                       u32 offset)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);
        u32 window_start;

        /* If offset lies within DP register range, use 3rd window */
        if ((offset ^ ab_pci->reg_base->umac_base) < WINDOW_RANGE_MASK)
                window_start = 3 * WINDOW_START;
        /* If offset lies within CE register range, use 2nd window */
        else if ((offset ^ ab_pci->reg_base->ce_reg_base) < WINDOW_RANGE_MASK)
                window_start = 2 * WINDOW_START;
        else
                window_start = WINDOW_START;

        return window_start;
}

static inline bool ath12k_pci_is_offset_within_mhi_region(u32 offset)
{
        return (offset >= PCI_MHIREGLEN_REG && offset <= PCI_MHI_REGION_END);
}

static void ath12k_pci_restore_window(struct ath12k_base *ab)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);

        spin_lock_bh(&ab_pci->window_lock);

        iowrite32(WINDOW_ENABLE_BIT | ab_pci->register_window,
                  ab->mem + ab_pci->window_reg_addr);
        ioread32(ab->mem + ab_pci->window_reg_addr);

        spin_unlock_bh(&ab_pci->window_lock);
}

static void ath12k_pci_soc_global_reset(struct ath12k_base *ab)
{
        u32 val, delay;

        val = ath12k_pci_read32(ab, PCIE_SOC_GLOBAL_RESET);

        val |= PCIE_SOC_GLOBAL_RESET_V;

        ath12k_pci_write32(ab, PCIE_SOC_GLOBAL_RESET, val);

        /* TODO: exact time to sleep is uncertain */
        delay = 10;
        mdelay(delay);

        /* Need to toggle V bit back otherwise stuck in reset status */
        val &= ~PCIE_SOC_GLOBAL_RESET_V;

        ath12k_pci_write32(ab, PCIE_SOC_GLOBAL_RESET, val);

        mdelay(delay);

        val = ath12k_pci_read32(ab, PCIE_SOC_GLOBAL_RESET);
        if (val == 0xffffffff)
                ath12k_warn(ab, "link down error during global reset\n");

        /* Restore window register as its content is cleared during
         * hardware global reset, such that it aligns with host cache.
         */
        ath12k_pci_restore_window(ab);
}

static void ath12k_pci_clear_dbg_registers(struct ath12k_base *ab)
{
        u32 val;

        /* read cookie */
        val = ath12k_pci_read32(ab, PCIE_Q6_COOKIE_ADDR);
        ath12k_dbg(ab, ATH12K_DBG_PCI, "cookie:0x%x\n", val);

        val = ath12k_pci_read32(ab, WLAON_WARM_SW_ENTRY);
        ath12k_dbg(ab, ATH12K_DBG_PCI, "WLAON_WARM_SW_ENTRY 0x%x\n", val);

        /* TODO: exact time to sleep is uncertain */
        mdelay(10);

        /* write 0 to WLAON_WARM_SW_ENTRY to prevent Q6 from
         * continuing warm path and entering dead loop.
         */
        ath12k_pci_write32(ab, WLAON_WARM_SW_ENTRY, 0);
        mdelay(10);

        val = ath12k_pci_read32(ab, WLAON_WARM_SW_ENTRY);
        ath12k_dbg(ab, ATH12K_DBG_PCI, "WLAON_WARM_SW_ENTRY 0x%x\n", val);

        /* A read clear register. clear the register to prevent
         * Q6 from entering wrong code path.
         */
        val = ath12k_pci_read32(ab, WLAON_SOC_RESET_CAUSE_REG);
        ath12k_dbg(ab, ATH12K_DBG_PCI, "soc reset cause:%d\n", val);
}

static void ath12k_pci_enable_ltssm(struct ath12k_base *ab)
{
        u32 val;
        int i;

        val = ath12k_pci_read32(ab, PCIE_PCIE_PARF_LTSSM);

        /* PCIE link seems very unstable after the Hot Reset*/
        for (i = 0; val != PARM_LTSSM_VALUE && i < 5; i++) {
                if (val == 0xffffffff)
                        mdelay(5);

                ath12k_pci_write32(ab, PCIE_PCIE_PARF_LTSSM, PARM_LTSSM_VALUE);
                val = ath12k_pci_read32(ab, PCIE_PCIE_PARF_LTSSM);
        }

        ath12k_dbg(ab, ATH12K_DBG_PCI, "pci ltssm 0x%x\n", val);

        val = ath12k_pci_read32(ab, GCC_GCC_PCIE_HOT_RST(ab));
        val |= GCC_GCC_PCIE_HOT_RST_VAL;
        ath12k_pci_write32(ab, GCC_GCC_PCIE_HOT_RST(ab), val);
        val = ath12k_pci_read32(ab, GCC_GCC_PCIE_HOT_RST(ab));

        ath12k_dbg(ab, ATH12K_DBG_PCI, "pci pcie_hot_rst 0x%x\n", val);

        mdelay(5);
}

static void ath12k_pci_clear_all_intrs(struct ath12k_base *ab)
{
        /* This is a WAR for PCIE Hotreset.
         * When target receive Hotreset, but will set the interrupt.
         * So when download SBL again, SBL will open Interrupt and
         * receive it, and crash immediately.
         */
        ath12k_pci_write32(ab, PCIE_PCIE_INT_ALL_CLEAR, PCIE_INT_CLEAR_ALL);
}

static void ath12k_pci_set_wlaon_pwr_ctrl(struct ath12k_base *ab)
{
        u32 val;

        val = ath12k_pci_read32(ab, WLAON_QFPROM_PWR_CTRL_REG);
        val &= ~QFPROM_PWR_CTRL_VDD4BLOW_MASK;
        ath12k_pci_write32(ab, WLAON_QFPROM_PWR_CTRL_REG, val);
}

static void ath12k_pci_force_wake(struct ath12k_base *ab)
{
        ath12k_pci_write32(ab, PCIE_SOC_WAKE_PCIE_LOCAL_REG, 1);
        mdelay(5);
}

static void ath12k_pci_sw_reset(struct ath12k_base *ab, bool power_on)
{
        if (power_on) {
                ath12k_pci_enable_ltssm(ab);
                ath12k_pci_clear_all_intrs(ab);
                ath12k_pci_set_wlaon_pwr_ctrl(ab);
        }

        ath12k_mhi_clear_vector(ab);
        ath12k_pci_clear_dbg_registers(ab);
        ath12k_pci_soc_global_reset(ab);
        ath12k_mhi_set_mhictrl_reset(ab);
}

static void ath12k_pci_free_ext_irq(struct ath12k_base *ab)
{
        int i, j;

        for (i = 0; i < ATH12K_EXT_IRQ_GRP_NUM_MAX; i++) {
                struct ath12k_ext_irq_grp *irq_grp = &ab->ext_irq_grp[i];

                for (j = 0; j < irq_grp->num_irq; j++)
                        free_irq(ab->irq_num[irq_grp->irqs[j]], irq_grp);

                netif_napi_del(&irq_grp->napi);
                free_netdev(irq_grp->napi_ndev);
        }
}

static void ath12k_pci_free_irq(struct ath12k_base *ab)
{
        int i, irq_idx;

        for (i = 0; i < ab->hw_params->ce_count; i++) {
                if (ath12k_ce_get_attr_flags(ab, i) & CE_ATTR_DIS_INTR)
                        continue;
                irq_idx = ATH12K_PCI_IRQ_CE0_OFFSET + i;
                free_irq(ab->irq_num[irq_idx], &ab->ce.ce_pipe[i]);
        }

        ath12k_pci_free_ext_irq(ab);
}

static void ath12k_pci_ce_irq_enable(struct ath12k_base *ab, u16 ce_id)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);
        u32 irq_idx;

        /* In case of one MSI vector, we handle irq enable/disable in a
         * uniform way since we only have one irq
         */
        if (!test_bit(ATH12K_PCI_FLAG_MULTI_MSI_VECTORS, &ab_pci->flags))
                return;

        irq_idx = ATH12K_PCI_IRQ_CE0_OFFSET + ce_id;
        enable_irq(ab->irq_num[irq_idx]);
}

static void ath12k_pci_ce_irq_disable(struct ath12k_base *ab, u16 ce_id)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);
        u32 irq_idx;

        /* In case of one MSI vector, we handle irq enable/disable in a
         * uniform way since we only have one irq
         */
        if (!test_bit(ATH12K_PCI_FLAG_MULTI_MSI_VECTORS, &ab_pci->flags))
                return;

        irq_idx = ATH12K_PCI_IRQ_CE0_OFFSET + ce_id;
        disable_irq_nosync(ab->irq_num[irq_idx]);
}

static void ath12k_pci_ce_irqs_disable(struct ath12k_base *ab)
{
        int i;

        clear_bit(ATH12K_FLAG_CE_IRQ_ENABLED, &ab->dev_flags);

        for (i = 0; i < ab->hw_params->ce_count; i++) {
                if (ath12k_ce_get_attr_flags(ab, i) & CE_ATTR_DIS_INTR)
                        continue;
                ath12k_pci_ce_irq_disable(ab, i);
        }
}

static void ath12k_pci_sync_ce_irqs(struct ath12k_base *ab)
{
        int i;
        int irq_idx;

        for (i = 0; i < ab->hw_params->ce_count; i++) {
                if (ath12k_ce_get_attr_flags(ab, i) & CE_ATTR_DIS_INTR)
                        continue;

                irq_idx = ATH12K_PCI_IRQ_CE0_OFFSET + i;
                synchronize_irq(ab->irq_num[irq_idx]);
        }
}

static void ath12k_pci_ce_workqueue(struct work_struct *work)
{
        struct ath12k_ce_pipe *ce_pipe = from_work(ce_pipe, work, intr_wq);
        int irq_idx = ATH12K_PCI_IRQ_CE0_OFFSET + ce_pipe->pipe_num;

        ath12k_ce_per_engine_service(ce_pipe->ab, ce_pipe->pipe_num);

        enable_irq(ce_pipe->ab->irq_num[irq_idx]);
}

static irqreturn_t ath12k_pci_ce_interrupt_handler(int irq, void *arg)
{
        struct ath12k_ce_pipe *ce_pipe = arg;
        struct ath12k_base *ab = ce_pipe->ab;
        int irq_idx = ATH12K_PCI_IRQ_CE0_OFFSET + ce_pipe->pipe_num;

        if (!test_bit(ATH12K_FLAG_CE_IRQ_ENABLED, &ab->dev_flags))
                return IRQ_HANDLED;

        /* last interrupt received for this CE */
        ce_pipe->timestamp = jiffies;

        disable_irq_nosync(ab->irq_num[irq_idx]);

        queue_work(system_bh_wq, &ce_pipe->intr_wq);

        return IRQ_HANDLED;
}

static void ath12k_pci_ext_grp_disable(struct ath12k_ext_irq_grp *irq_grp)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(irq_grp->ab);
        int i;

        /* In case of one MSI vector, we handle irq enable/disable
         * in a uniform way since we only have one irq
         */
        if (!test_bit(ATH12K_PCI_FLAG_MULTI_MSI_VECTORS, &ab_pci->flags))
                return;

        for (i = 0; i < irq_grp->num_irq; i++)
                disable_irq_nosync(irq_grp->ab->irq_num[irq_grp->irqs[i]]);
}

static void __ath12k_pci_ext_irq_disable(struct ath12k_base *ab)
{
        int i;

        if (!test_and_clear_bit(ATH12K_FLAG_EXT_IRQ_ENABLED, &ab->dev_flags))
                return;

        for (i = 0; i < ATH12K_EXT_IRQ_GRP_NUM_MAX; i++) {
                struct ath12k_ext_irq_grp *irq_grp = &ab->ext_irq_grp[i];

                ath12k_pci_ext_grp_disable(irq_grp);

                if (irq_grp->napi_enabled) {
                        napi_synchronize(&irq_grp->napi);
                        napi_disable(&irq_grp->napi);
                        irq_grp->napi_enabled = false;
                }
        }
}

static void ath12k_pci_ext_grp_enable(struct ath12k_ext_irq_grp *irq_grp)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(irq_grp->ab);
        int i;

        /* In case of one MSI vector, we handle irq enable/disable in a
         * uniform way since we only have one irq
         */
        if (!test_bit(ATH12K_PCI_FLAG_MULTI_MSI_VECTORS, &ab_pci->flags))
                return;

        for (i = 0; i < irq_grp->num_irq; i++)
                enable_irq(irq_grp->ab->irq_num[irq_grp->irqs[i]]);
}

static void ath12k_pci_sync_ext_irqs(struct ath12k_base *ab)
{
        int i, j, irq_idx;

        for (i = 0; i < ATH12K_EXT_IRQ_GRP_NUM_MAX; i++) {
                struct ath12k_ext_irq_grp *irq_grp = &ab->ext_irq_grp[i];

                for (j = 0; j < irq_grp->num_irq; j++) {
                        irq_idx = irq_grp->irqs[j];
                        synchronize_irq(ab->irq_num[irq_idx]);
                }
        }
}

static int ath12k_pci_ext_grp_napi_poll(struct napi_struct *napi, int budget)
{
        struct ath12k_ext_irq_grp *irq_grp = container_of(napi,
                                                struct ath12k_ext_irq_grp,
                                                napi);
        struct ath12k_base *ab = irq_grp->ab;
        struct ath12k_dp *dp = ath12k_ab_to_dp(ab);
        int work_done;
        int i;

        work_done = ath12k_dp_service_srng(dp, irq_grp, budget);
        if (work_done < budget) {
                napi_complete_done(napi, work_done);
                for (i = 0; i < irq_grp->num_irq; i++)
                        enable_irq(irq_grp->ab->irq_num[irq_grp->irqs[i]]);
        }

        if (work_done > budget)
                work_done = budget;

        return work_done;
}

static irqreturn_t ath12k_pci_ext_interrupt_handler(int irq, void *arg)
{
        struct ath12k_ext_irq_grp *irq_grp = arg;
        struct ath12k_base *ab = irq_grp->ab;
        int i;

        if (!test_bit(ATH12K_FLAG_EXT_IRQ_ENABLED, &ab->dev_flags))
                return IRQ_HANDLED;

        ath12k_dbg(irq_grp->ab, ATH12K_DBG_PCI, "ext irq:%d\n", irq);

        /* last interrupt received for this group */
        irq_grp->timestamp = jiffies;

        for (i = 0; i < irq_grp->num_irq; i++)
                disable_irq_nosync(irq_grp->ab->irq_num[irq_grp->irqs[i]]);

        napi_schedule(&irq_grp->napi);

        return IRQ_HANDLED;
}

static int ath12k_pci_ext_irq_config(struct ath12k_base *ab)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);
        int i, j, n, ret, num_vectors = 0;
        u32 user_base_data = 0, base_vector = 0, base_idx;
        struct ath12k_ext_irq_grp *irq_grp;

        base_idx = ATH12K_PCI_IRQ_CE0_OFFSET + CE_COUNT_MAX;
        ret = ath12k_pci_get_user_msi_assignment(ab, "DP",
                                                 &num_vectors,
                                                 &user_base_data,
                                                 &base_vector);
        if (ret < 0)
                return ret;

        for (i = 0; i < ATH12K_EXT_IRQ_GRP_NUM_MAX; i++) {
                irq_grp = &ab->ext_irq_grp[i];
                u32 num_irq = 0;

                irq_grp->ab = ab;
                irq_grp->grp_id = i;
                irq_grp->napi_ndev = alloc_netdev_dummy(0);
                if (!irq_grp->napi_ndev) {
                        ret = -ENOMEM;
                        goto fail_allocate;
                }

                netif_napi_add(irq_grp->napi_ndev, &irq_grp->napi,
                               ath12k_pci_ext_grp_napi_poll);

                if (ab->hw_params->ring_mask->tx[i] ||
                    ab->hw_params->ring_mask->rx[i] ||
                    ab->hw_params->ring_mask->rx_err[i] ||
                    ab->hw_params->ring_mask->rx_wbm_rel[i] ||
                    ab->hw_params->ring_mask->reo_status[i] ||
                    ab->hw_params->ring_mask->host2rxdma[i] ||
                    ab->hw_params->ring_mask->rx_mon_dest[i] ||
                    ab->hw_params->ring_mask->rx_mon_status[i]) {
                        num_irq = 1;
                }

                irq_grp->num_irq = num_irq;
                irq_grp->irqs[0] = base_idx + i;

                for (j = 0; j < irq_grp->num_irq; j++) {
                        int irq_idx = irq_grp->irqs[j];
                        int vector = (i % num_vectors) + base_vector;
                        int irq = ath12k_pci_get_msi_irq(ab->dev, vector);

                        ab->irq_num[irq_idx] = irq;

                        ath12k_dbg(ab, ATH12K_DBG_PCI,
                                   "irq:%d group:%d\n", irq, i);

                        irq_set_status_flags(irq, IRQ_DISABLE_UNLAZY);
                        ret = request_irq(irq, ath12k_pci_ext_interrupt_handler,
                                          ab_pci->irq_flags,
                                          "DP_EXT_IRQ", irq_grp);
                        if (ret) {
                                ath12k_err(ab, "failed request irq %d: %d\n",
                                           vector, ret);
                                goto fail_request;
                        }
                }
                ath12k_pci_ext_grp_disable(irq_grp);
        }

        return 0;

fail_request:
        /* i ->napi_ndev was properly allocated. Free it also */
        i += 1;
fail_allocate:
        for (n = 0; n < i; n++) {
                irq_grp = &ab->ext_irq_grp[n];
                free_netdev(irq_grp->napi_ndev);
        }
        return ret;
}

static int ath12k_pci_set_irq_affinity_hint(struct ath12k_pci *ab_pci,
                                            const struct cpumask *m)
{
        if (test_bit(ATH12K_PCI_FLAG_MULTI_MSI_VECTORS, &ab_pci->flags))
                return 0;

        return irq_set_affinity_and_hint(ab_pci->pdev->irq, m);
}

static int ath12k_pci_config_irq(struct ath12k_base *ab)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);
        struct ath12k_ce_pipe *ce_pipe;
        u32 msi_data_start;
        u32 msi_data_count, msi_data_idx;
        u32 msi_irq_start;
        unsigned int msi_data;
        int irq, i, ret, irq_idx;

        ret = ath12k_pci_get_user_msi_assignment(ab,
                                                 "CE", &msi_data_count,
                                                 &msi_data_start, &msi_irq_start);
        if (ret)
                return ret;

        /* Configure CE irqs */

        for (i = 0, msi_data_idx = 0; i < ab->hw_params->ce_count; i++) {
                if (ath12k_ce_get_attr_flags(ab, i) & CE_ATTR_DIS_INTR)
                        continue;

                msi_data = (msi_data_idx % msi_data_count) + msi_irq_start;
                irq = ath12k_pci_get_msi_irq(ab->dev, msi_data);
                ce_pipe = &ab->ce.ce_pipe[i];

                irq_idx = ATH12K_PCI_IRQ_CE0_OFFSET + i;

                INIT_WORK(&ce_pipe->intr_wq, ath12k_pci_ce_workqueue);

                ret = request_irq(irq, ath12k_pci_ce_interrupt_handler,
                                  ab_pci->irq_flags, irq_name[irq_idx],
                                  ce_pipe);
                if (ret) {
                        ath12k_err(ab, "failed to request irq %d: %d\n",
                                   irq_idx, ret);
                        return ret;
                }

                ab->irq_num[irq_idx] = irq;
                msi_data_idx++;

                ath12k_pci_ce_irq_disable(ab, i);
        }

        ret = ath12k_pci_ext_irq_config(ab);
        if (ret)
                return ret;

        return 0;
}

static void ath12k_pci_init_qmi_ce_config(struct ath12k_base *ab)
{
        struct ath12k_qmi_ce_cfg *cfg = &ab->qmi.ce_cfg;

        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);
        struct pci_bus *bus = ab_pci->pdev->bus;

        cfg->tgt_ce = ab->hw_params->target_ce_config;
        cfg->tgt_ce_len = ab->hw_params->target_ce_count;

        cfg->svc_to_ce_map = ab->hw_params->svc_to_ce_map;
        cfg->svc_to_ce_map_len = ab->hw_params->svc_to_ce_map_len;
        ab->qmi.service_ins_id = ab->hw_params->qmi_service_ins_id;

        if (ath12k_fw_feature_supported(ab, ATH12K_FW_FEATURE_MULTI_QRTR_ID)) {
                ab_pci->qmi_instance =
                        u32_encode_bits(pci_domain_nr(bus), DOMAIN_NUMBER_MASK) |
                        u32_encode_bits(bus->number, BUS_NUMBER_MASK);
                ab->qmi.service_ins_id += ab_pci->qmi_instance;
        }
}

static void ath12k_pci_ce_irqs_enable(struct ath12k_base *ab)
{
        int i;

        set_bit(ATH12K_FLAG_CE_IRQ_ENABLED, &ab->dev_flags);

        for (i = 0; i < ab->hw_params->ce_count; i++) {
                if (ath12k_ce_get_attr_flags(ab, i) & CE_ATTR_DIS_INTR)
                        continue;
                ath12k_pci_ce_irq_enable(ab, i);
        }
}

static void ath12k_pci_msi_config(struct ath12k_pci *ab_pci, bool enable)
{
        struct pci_dev *dev = ab_pci->pdev;
        u16 control;

        pci_read_config_word(dev, dev->msi_cap + PCI_MSI_FLAGS, &control);

        if (enable)
                control |= PCI_MSI_FLAGS_ENABLE;
        else
                control &= ~PCI_MSI_FLAGS_ENABLE;

        pci_write_config_word(dev, dev->msi_cap + PCI_MSI_FLAGS, control);
}

static void ath12k_pci_msi_enable(struct ath12k_pci *ab_pci)
{
        ath12k_pci_msi_config(ab_pci, true);
}

static void ath12k_pci_msi_disable(struct ath12k_pci *ab_pci)
{
        ath12k_pci_msi_config(ab_pci, false);
}

static int ath12k_pci_msi_alloc(struct ath12k_pci *ab_pci)
{
        struct ath12k_base *ab = ab_pci->ab;
        const struct ath12k_msi_config *msi_config = ab_pci->msi_config;
        struct msi_desc *msi_desc;
        int num_vectors;
        int ret;

        num_vectors = pci_alloc_irq_vectors(ab_pci->pdev,
                                            msi_config->total_vectors,
                                            msi_config->total_vectors,
                                            PCI_IRQ_MSI);

        if (num_vectors == msi_config->total_vectors) {
                set_bit(ATH12K_PCI_FLAG_MULTI_MSI_VECTORS, &ab_pci->flags);
                ab_pci->irq_flags = IRQF_SHARED;
        } else {
                num_vectors = pci_alloc_irq_vectors(ab_pci->pdev,
                                                    1,
                                                    1,
                                                    PCI_IRQ_MSI);
                if (num_vectors < 0) {
                        ret = -EINVAL;
                        goto reset_msi_config;
                }
                clear_bit(ATH12K_PCI_FLAG_MULTI_MSI_VECTORS, &ab_pci->flags);
                ab_pci->msi_config = &msi_config_one_msi;
                ab_pci->irq_flags = IRQF_SHARED | IRQF_NOBALANCING;
                ath12k_dbg(ab, ATH12K_DBG_PCI, "request MSI one vector\n");
        }

        ath12k_info(ab, "MSI vectors: %d\n", num_vectors);

        ath12k_pci_msi_disable(ab_pci);

        msi_desc = irq_get_msi_desc(ab_pci->pdev->irq);
        if (!msi_desc) {
                ath12k_err(ab, "msi_desc is NULL!\n");
                ret = -EINVAL;
                goto free_msi_vector;
        }

        ab_pci->msi_ep_base_data = msi_desc->msg.data;
        if (msi_desc->pci.msi_attrib.is_64)
                set_bit(ATH12K_PCI_FLAG_IS_MSI_64, &ab_pci->flags);

        ath12k_dbg(ab, ATH12K_DBG_PCI, "msi base data is %d\n", ab_pci->msi_ep_base_data);

        return 0;

free_msi_vector:
        pci_free_irq_vectors(ab_pci->pdev);

reset_msi_config:
        return ret;
}

static void ath12k_pci_msi_free(struct ath12k_pci *ab_pci)
{
        pci_free_irq_vectors(ab_pci->pdev);
}

static int ath12k_pci_config_msi_data(struct ath12k_pci *ab_pci)
{
        struct msi_desc *msi_desc;

        msi_desc = irq_get_msi_desc(ab_pci->pdev->irq);
        if (!msi_desc) {
                ath12k_err(ab_pci->ab, "msi_desc is NULL!\n");
                pci_free_irq_vectors(ab_pci->pdev);
                return -EINVAL;
        }

        ab_pci->msi_ep_base_data = msi_desc->msg.data;

        ath12k_dbg(ab_pci->ab, ATH12K_DBG_PCI, "pci after request_irq msi_ep_base_data %d\n",
                   ab_pci->msi_ep_base_data);

        return 0;
}

static int ath12k_pci_claim(struct ath12k_pci *ab_pci, struct pci_dev *pdev)
{
        struct ath12k_base *ab = ab_pci->ab;
        u16 device_id;
        int ret = 0;

        pci_read_config_word(pdev, PCI_DEVICE_ID, &device_id);
        if (device_id != ab_pci->dev_id)  {
                ath12k_err(ab, "pci device id mismatch: 0x%x 0x%x\n",
                           device_id, ab_pci->dev_id);
                ret = -EIO;
                goto out;
        }

        ret = pci_assign_resource(pdev, ATH12K_PCI_BAR_NUM);
        if (ret) {
                ath12k_err(ab, "failed to assign pci resource: %d\n", ret);
                goto out;
        }

        ret = pci_enable_device(pdev);
        if (ret) {
                ath12k_err(ab, "failed to enable pci device: %d\n", ret);
                goto out;
        }

        ret = pci_request_region(pdev, ATH12K_PCI_BAR_NUM, "ath12k_pci");
        if (ret) {
                ath12k_err(ab, "failed to request pci region: %d\n", ret);
                goto disable_device;
        }

        ab_pci->dma_mask = DMA_BIT_MASK(ATH12K_PCI_DMA_MASK);
        dma_set_mask(&pdev->dev, ab_pci->dma_mask);
        dma_set_coherent_mask(&pdev->dev, DMA_BIT_MASK(32));

        pci_set_master(pdev);

        ab->mem_len = pci_resource_len(pdev, ATH12K_PCI_BAR_NUM);
        ab->mem = pci_iomap(pdev, ATH12K_PCI_BAR_NUM, 0);
        if (!ab->mem) {
                ath12k_err(ab, "failed to map pci bar %d\n", ATH12K_PCI_BAR_NUM);
                ret = -EIO;
                goto release_region;
        }

        ath12k_dbg(ab, ATH12K_DBG_BOOT, "boot pci_mem 0x%p\n", ab->mem);
        return 0;

release_region:
        pci_release_region(pdev, ATH12K_PCI_BAR_NUM);
disable_device:
        pci_disable_device(pdev);
out:
        return ret;
}

static void ath12k_pci_free_region(struct ath12k_pci *ab_pci)
{
        struct ath12k_base *ab = ab_pci->ab;
        struct pci_dev *pci_dev = ab_pci->pdev;

        pci_iounmap(pci_dev, ab->mem);
        ab->mem = NULL;
        pci_release_region(pci_dev, ATH12K_PCI_BAR_NUM);
        if (pci_is_enabled(pci_dev))
                pci_disable_device(pci_dev);
}

static void ath12k_pci_aspm_disable(struct ath12k_pci *ab_pci)
{
        struct ath12k_base *ab = ab_pci->ab;

        pcie_capability_read_word(ab_pci->pdev, PCI_EXP_LNKCTL,
                                  &ab_pci->link_ctl);

        ath12k_dbg(ab, ATH12K_DBG_PCI, "pci link_ctl 0x%04x L0s %d L1 %d\n",
                   ab_pci->link_ctl,
                   u16_get_bits(ab_pci->link_ctl, PCI_EXP_LNKCTL_ASPM_L0S),
                   u16_get_bits(ab_pci->link_ctl, PCI_EXP_LNKCTL_ASPM_L1));

        /* disable L0s and L1 */
        pcie_capability_clear_word(ab_pci->pdev, PCI_EXP_LNKCTL,
                                   PCI_EXP_LNKCTL_ASPMC);

        set_bit(ATH12K_PCI_ASPM_RESTORE, &ab_pci->flags);
}

static void ath12k_pci_update_qrtr_node_id(struct ath12k_base *ab)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);
        u32 reg;

        /* On platforms with two or more identical mhi devices, qmi service run
         * with identical qrtr-node-id. Because of this identical ID qrtr-lookup
         * cannot register more than one qmi service with identical node ID.
         *
         * This generates a unique instance ID from PCIe domain number and bus number,
         * writes to the given register, it is available for firmware when the QMI service
         * is spawned.
         */
        reg = PCIE_LOCAL_REG_QRTR_NODE_ID(ab) & WINDOW_RANGE_MASK;
        ath12k_pci_write32(ab, reg, ab_pci->qmi_instance);

        ath12k_dbg(ab, ATH12K_DBG_PCI, "pci reg 0x%x instance 0x%x read val 0x%x\n",
                   reg, ab_pci->qmi_instance, ath12k_pci_read32(ab, reg));
}

static void ath12k_pci_aspm_restore(struct ath12k_pci *ab_pci)
{
        if (ab_pci->ab->hw_params->supports_aspm &&
            test_and_clear_bit(ATH12K_PCI_ASPM_RESTORE, &ab_pci->flags))
                pcie_capability_clear_and_set_word(ab_pci->pdev, PCI_EXP_LNKCTL,
                                                   PCI_EXP_LNKCTL_ASPMC,
                                                   ab_pci->link_ctl &
                                                   PCI_EXP_LNKCTL_ASPMC);
}

static void ath12k_pci_cancel_workqueue(struct ath12k_base *ab)
{
        int i;

        for (i = 0; i < ab->hw_params->ce_count; i++) {
                struct ath12k_ce_pipe *ce_pipe = &ab->ce.ce_pipe[i];

                if (ath12k_ce_get_attr_flags(ab, i) & CE_ATTR_DIS_INTR)
                        continue;

                cancel_work_sync(&ce_pipe->intr_wq);
        }
}

static void ath12k_pci_ce_irq_disable_sync(struct ath12k_base *ab)
{
        ath12k_pci_ce_irqs_disable(ab);
        ath12k_pci_sync_ce_irqs(ab);
        ath12k_pci_cancel_workqueue(ab);
}

int ath12k_pci_map_service_to_pipe(struct ath12k_base *ab, u16 service_id,
                                   u8 *ul_pipe, u8 *dl_pipe)
{
        const struct service_to_pipe *entry;
        bool ul_set = false, dl_set = false;
        int i;

        for (i = 0; i < ab->hw_params->svc_to_ce_map_len; i++) {
                entry = &ab->hw_params->svc_to_ce_map[i];

                if (__le32_to_cpu(entry->service_id) != service_id)
                        continue;

                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;
                }
        }

        if (WARN_ON(!ul_set || !dl_set))
                return -ENOENT;

        return 0;
}

int ath12k_pci_get_msi_irq(struct device *dev, unsigned int vector)
{
        struct pci_dev *pci_dev = to_pci_dev(dev);

        return pci_irq_vector(pci_dev, vector);
}

int ath12k_pci_get_user_msi_assignment(struct ath12k_base *ab, char *user_name,
                                       int *num_vectors, u32 *user_base_data,
                                       u32 *base_vector)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);
        const struct ath12k_msi_config *msi_config = ab_pci->msi_config;
        int idx;

        for (idx = 0; idx < msi_config->total_users; idx++) {
                if (strcmp(user_name, msi_config->users[idx].name) == 0) {
                        *num_vectors = msi_config->users[idx].num_vectors;
                        *base_vector =  msi_config->users[idx].base_vector;
                        *user_base_data = *base_vector + ab_pci->msi_ep_base_data;

                        ath12k_dbg(ab, ATH12K_DBG_PCI,
                                   "Assign MSI to user: %s, num_vectors: %d, user_base_data: %u, base_vector: %u\n",
                                   user_name, *num_vectors, *user_base_data,
                                   *base_vector);

                        return 0;
                }
        }

        ath12k_err(ab, "Failed to find MSI assignment for %s!\n", user_name);

        return -EINVAL;
}

void ath12k_pci_get_msi_address(struct ath12k_base *ab, u32 *msi_addr_lo,
                                u32 *msi_addr_hi)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);
        struct pci_dev *pci_dev = to_pci_dev(ab->dev);

        pci_read_config_dword(pci_dev, pci_dev->msi_cap + PCI_MSI_ADDRESS_LO,
                              msi_addr_lo);

        if (test_bit(ATH12K_PCI_FLAG_IS_MSI_64, &ab_pci->flags)) {
                pci_read_config_dword(pci_dev, pci_dev->msi_cap + PCI_MSI_ADDRESS_HI,
                                      msi_addr_hi);
        } else {
                *msi_addr_hi = 0;
        }
}

void ath12k_pci_get_ce_msi_idx(struct ath12k_base *ab, u32 ce_id,
                               u32 *msi_idx)
{
        u32 i, msi_data_idx;

        for (i = 0, msi_data_idx = 0; i < ab->hw_params->ce_count; i++) {
                if (ath12k_ce_get_attr_flags(ab, i) & CE_ATTR_DIS_INTR)
                        continue;

                if (ce_id == i)
                        break;

                msi_data_idx++;
        }
        *msi_idx = msi_data_idx;
}

void ath12k_pci_hif_ce_irq_enable(struct ath12k_base *ab)
{
        ath12k_pci_ce_irqs_enable(ab);
}

void ath12k_pci_hif_ce_irq_disable(struct ath12k_base *ab)
{
        ath12k_pci_ce_irq_disable_sync(ab);
}

void ath12k_pci_ext_irq_enable(struct ath12k_base *ab)
{
        int i;

        for (i = 0; i < ATH12K_EXT_IRQ_GRP_NUM_MAX; i++) {
                struct ath12k_ext_irq_grp *irq_grp = &ab->ext_irq_grp[i];

                if (!irq_grp->napi_enabled) {
                        napi_enable(&irq_grp->napi);
                        irq_grp->napi_enabled = true;
                }

                ath12k_pci_ext_grp_enable(irq_grp);
        }

        set_bit(ATH12K_FLAG_EXT_IRQ_ENABLED, &ab->dev_flags);
}

void ath12k_pci_ext_irq_disable(struct ath12k_base *ab)
{
        if (!test_bit(ATH12K_FLAG_EXT_IRQ_ENABLED, &ab->dev_flags))
                return;

        __ath12k_pci_ext_irq_disable(ab);
        ath12k_pci_sync_ext_irqs(ab);
}

int ath12k_pci_hif_suspend(struct ath12k_base *ab)
{
        struct ath12k_pci *ar_pci = ath12k_pci_priv(ab);

        ath12k_mhi_suspend(ar_pci);

        return 0;
}

int ath12k_pci_hif_resume(struct ath12k_base *ab)
{
        struct ath12k_pci *ar_pci = ath12k_pci_priv(ab);

        ath12k_mhi_resume(ar_pci);

        return 0;
}

void ath12k_pci_stop(struct ath12k_base *ab)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);

        if (!test_bit(ATH12K_PCI_FLAG_INIT_DONE, &ab_pci->flags))
                return;

        ath12k_pci_ce_irq_disable_sync(ab);
        ath12k_ce_cleanup_pipes(ab);
}

int ath12k_pci_start(struct ath12k_base *ab)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);

        set_bit(ATH12K_PCI_FLAG_INIT_DONE, &ab_pci->flags);

        if (test_bit(ATH12K_PCI_FLAG_MULTI_MSI_VECTORS, &ab_pci->flags))
                ath12k_pci_aspm_restore(ab_pci);
        else
                ath12k_info(ab, "leaving PCI ASPM disabled to avoid MHI M2 problems\n");

        ath12k_pci_ce_irqs_enable(ab);
        ath12k_ce_rx_post_buf(ab);

        return 0;
}

u32 ath12k_pci_read32(struct ath12k_base *ab, u32 offset)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);
        u32 val, window_start;
        int ret = 0;

        /* for offset beyond BAR + 4K - 32, may
         * need to wakeup MHI to access.
         */
        if (test_bit(ATH12K_PCI_FLAG_INIT_DONE, &ab_pci->flags) &&
            offset >= ACCESS_ALWAYS_OFF && ab_pci->pci_ops->wakeup)
                ret = ab_pci->pci_ops->wakeup(ab);

        if (offset < WINDOW_START) {
                val = ioread32(ab->mem + offset);
        } else {
                if (ab->static_window_map)
                        window_start = ath12k_pci_get_window_start(ab, offset);
                else
                        window_start = WINDOW_START;

                if (window_start == WINDOW_START) {
                        spin_lock_bh(&ab_pci->window_lock);
                        ath12k_pci_select_window(ab_pci, offset);

                        if (ath12k_pci_is_offset_within_mhi_region(offset)) {
                                offset = offset - PCI_MHIREGLEN_REG;
                                val = ioread32(ab->mem +
                                               (offset & WINDOW_RANGE_MASK));
                        } else {
                                val = ioread32(ab->mem + window_start +
                                               (offset & WINDOW_RANGE_MASK));
                        }
                        spin_unlock_bh(&ab_pci->window_lock);
                } else {
                        val = ioread32(ab->mem + window_start +
                                       (offset & WINDOW_RANGE_MASK));
                }
        }

        if (test_bit(ATH12K_PCI_FLAG_INIT_DONE, &ab_pci->flags) &&
            offset >= ACCESS_ALWAYS_OFF && ab_pci->pci_ops->release &&
            !ret)
                ab_pci->pci_ops->release(ab);
        return val;
}
EXPORT_SYMBOL(ath12k_pci_read32);

void ath12k_pci_write32(struct ath12k_base *ab, u32 offset, u32 value)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);
        u32 window_start;
        int ret = 0;

        /* for offset beyond BAR + 4K - 32, may
         * need to wakeup MHI to access.
         */
        if (test_bit(ATH12K_PCI_FLAG_INIT_DONE, &ab_pci->flags) &&
            offset >= ACCESS_ALWAYS_OFF && ab_pci->pci_ops->wakeup)
                ret = ab_pci->pci_ops->wakeup(ab);

        if (offset < WINDOW_START) {
                iowrite32(value, ab->mem + offset);
        } else {
                if (ab->static_window_map)
                        window_start = ath12k_pci_get_window_start(ab, offset);
                else
                        window_start = WINDOW_START;

                if (window_start == WINDOW_START) {
                        spin_lock_bh(&ab_pci->window_lock);
                        ath12k_pci_select_window(ab_pci, offset);

                        if (ath12k_pci_is_offset_within_mhi_region(offset)) {
                                offset = offset - PCI_MHIREGLEN_REG;
                                iowrite32(value, ab->mem +
                                          (offset & WINDOW_RANGE_MASK));
                        } else {
                                iowrite32(value, ab->mem + window_start +
                                          (offset & WINDOW_RANGE_MASK));
                        }
                        spin_unlock_bh(&ab_pci->window_lock);
                } else {
                        iowrite32(value, ab->mem + window_start +
                                  (offset & WINDOW_RANGE_MASK));
                }
        }

        if (test_bit(ATH12K_PCI_FLAG_INIT_DONE, &ab_pci->flags) &&
            offset >= ACCESS_ALWAYS_OFF && ab_pci->pci_ops->release &&
            !ret)
                ab_pci->pci_ops->release(ab);
}

#ifdef CONFIG_ATH12K_COREDUMP
static int ath12k_pci_coredump_calculate_size(struct ath12k_base *ab, u32 *dump_seg_sz)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);
        struct mhi_controller *mhi_ctrl = ab_pci->mhi_ctrl;
        struct image_info *rddm_img, *fw_img;
        struct ath12k_tlv_dump_data *dump_tlv;
        enum ath12k_fw_crash_dump_type mem_type;
        u32 len = 0, rddm_tlv_sz = 0, paging_tlv_sz = 0;
        struct ath12k_dump_file_data *file_data;
        int i;

        rddm_img = mhi_ctrl->rddm_image;
        if (!rddm_img) {
                ath12k_err(ab, "No RDDM dump found\n");
                return 0;
        }

        fw_img = mhi_ctrl->fbc_image;

        for (i = 0; i < fw_img->entries ; i++) {
                if (!fw_img->mhi_buf[i].buf)
                        continue;

                paging_tlv_sz += fw_img->mhi_buf[i].len;
        }
        dump_seg_sz[FW_CRASH_DUMP_PAGING_DATA] = paging_tlv_sz;

        for (i = 0; i < rddm_img->entries; i++) {
                if (!rddm_img->mhi_buf[i].buf)
                        continue;

                rddm_tlv_sz += rddm_img->mhi_buf[i].len;
        }
        dump_seg_sz[FW_CRASH_DUMP_RDDM_DATA] = rddm_tlv_sz;

        for (i = 0; i < ab->qmi.mem_seg_count; i++) {
                mem_type = ath12k_coredump_get_dump_type(ab->qmi.target_mem[i].type);

                if (mem_type == FW_CRASH_DUMP_NONE)
                        continue;

                if (mem_type == FW_CRASH_DUMP_TYPE_MAX) {
                        ath12k_dbg(ab, ATH12K_DBG_PCI,
                                   "target mem region type %d not supported",
                                   ab->qmi.target_mem[i].type);
                        continue;
                }

                if (!ab->qmi.target_mem[i].paddr)
                        continue;

                dump_seg_sz[mem_type] += ab->qmi.target_mem[i].size;
        }

        for (i = 0; i < FW_CRASH_DUMP_TYPE_MAX; i++) {
                if (!dump_seg_sz[i])
                        continue;

                len += sizeof(*dump_tlv) + dump_seg_sz[i];
        }

        if (len)
                len += sizeof(*file_data);

        return len;
}

static void ath12k_pci_coredump_download(struct ath12k_base *ab)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);
        struct mhi_controller *mhi_ctrl = ab_pci->mhi_ctrl;
        struct image_info *rddm_img, *fw_img;
        struct timespec64 timestamp;
        int i, len, mem_idx;
        enum ath12k_fw_crash_dump_type mem_type;
        struct ath12k_dump_file_data *file_data;
        struct ath12k_tlv_dump_data *dump_tlv;
        size_t hdr_len = sizeof(*file_data);
        void *buf;
        u32 dump_seg_sz[FW_CRASH_DUMP_TYPE_MAX] = {};

        ath12k_mhi_coredump(mhi_ctrl, false);

        len = ath12k_pci_coredump_calculate_size(ab, dump_seg_sz);
        if (!len) {
                ath12k_warn(ab, "No crash dump data found for devcoredump");
                return;
        }

        rddm_img = mhi_ctrl->rddm_image;
        fw_img = mhi_ctrl->fbc_image;

        /* dev_coredumpv() requires vmalloc data */
        buf = vzalloc(len);
        if (!buf)
                return;

        ab->dump_data = buf;
        ab->ath12k_coredump_len = len;
        file_data = ab->dump_data;
        strscpy(file_data->df_magic, "ATH12K-FW-DUMP", sizeof(file_data->df_magic));
        file_data->len = cpu_to_le32(len);
        file_data->version = cpu_to_le32(ATH12K_FW_CRASH_DUMP_V2);
        file_data->chip_id = cpu_to_le32(ab_pci->dev_id);
        file_data->qrtr_id = cpu_to_le32(ab_pci->ab->qmi.service_ins_id);
        file_data->bus_id = cpu_to_le32(pci_domain_nr(ab_pci->pdev->bus));
        guid_gen(&file_data->guid);
        ktime_get_real_ts64(&timestamp);
        file_data->tv_sec = cpu_to_le64(timestamp.tv_sec);
        file_data->tv_nsec = cpu_to_le64(timestamp.tv_nsec);
        buf += hdr_len;
        dump_tlv = buf;
        dump_tlv->type = cpu_to_le32(FW_CRASH_DUMP_PAGING_DATA);
        dump_tlv->tlv_len = cpu_to_le32(dump_seg_sz[FW_CRASH_DUMP_PAGING_DATA]);
        buf += COREDUMP_TLV_HDR_SIZE;

        /* append all segments together as they are all part of a single contiguous
         * block of memory
         */
        for (i = 0; i < fw_img->entries ; i++) {
                if (!fw_img->mhi_buf[i].buf)
                        continue;

                memcpy_fromio(buf, (void const __iomem *)fw_img->mhi_buf[i].buf,
                              fw_img->mhi_buf[i].len);
                buf += fw_img->mhi_buf[i].len;
        }

        dump_tlv = buf;
        dump_tlv->type = cpu_to_le32(FW_CRASH_DUMP_RDDM_DATA);
        dump_tlv->tlv_len = cpu_to_le32(dump_seg_sz[FW_CRASH_DUMP_RDDM_DATA]);
        buf += COREDUMP_TLV_HDR_SIZE;

        /* append all segments together as they are all part of a single contiguous
         * block of memory
         */
        for (i = 0; i < rddm_img->entries; i++) {
                if (!rddm_img->mhi_buf[i].buf)
                        continue;

                memcpy_fromio(buf, (void const __iomem *)rddm_img->mhi_buf[i].buf,
                              rddm_img->mhi_buf[i].len);
                buf += rddm_img->mhi_buf[i].len;
        }

        mem_idx = FW_CRASH_DUMP_REMOTE_MEM_DATA;
        for (; mem_idx < FW_CRASH_DUMP_TYPE_MAX; mem_idx++) {
                if (!dump_seg_sz[mem_idx] || mem_idx == FW_CRASH_DUMP_NONE)
                        continue;

                dump_tlv = buf;
                dump_tlv->type = cpu_to_le32(mem_idx);
                dump_tlv->tlv_len = cpu_to_le32(dump_seg_sz[mem_idx]);
                buf += COREDUMP_TLV_HDR_SIZE;

                for (i = 0; i < ab->qmi.mem_seg_count; i++) {
                        mem_type = ath12k_coredump_get_dump_type
                                                        (ab->qmi.target_mem[i].type);

                        if (mem_type != mem_idx)
                                continue;

                        if (!ab->qmi.target_mem[i].paddr) {
                                ath12k_dbg(ab, ATH12K_DBG_PCI,
                                           "Skipping mem region type %d",
                                           ab->qmi.target_mem[i].type);
                                continue;
                        }

                        memcpy_fromio(buf, ab->qmi.target_mem[i].v.ioaddr,
                                      ab->qmi.target_mem[i].size);
                        buf += ab->qmi.target_mem[i].size;
                }
        }

        queue_work(ab->workqueue, &ab->dump_work);
}
#endif

int ath12k_pci_power_up(struct ath12k_base *ab)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);
        int ret;

        ab_pci->register_window = 0;
        clear_bit(ATH12K_PCI_FLAG_INIT_DONE, &ab_pci->flags);
        ath12k_pci_sw_reset(ab_pci->ab, true);

        /* Disable ASPM during firmware download due to problems switching
         * to AMSS state.
         */
        ath12k_pci_aspm_disable(ab_pci);

        ath12k_pci_msi_enable(ab_pci);

        if (ath12k_fw_feature_supported(ab, ATH12K_FW_FEATURE_MULTI_QRTR_ID))
                ath12k_pci_update_qrtr_node_id(ab);

        ret = ath12k_mhi_start(ab_pci);
        if (ret) {
                ath12k_err(ab, "failed to start mhi: %d\n", ret);
                return ret;
        }

        if (ab->static_window_map)
                ath12k_pci_select_static_window(ab_pci);

        return 0;
}

void ath12k_pci_power_down(struct ath12k_base *ab, bool is_suspend)
{
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);

        if (!test_bit(ATH12K_PCI_FLAG_INIT_DONE, &ab_pci->flags))
                return;

        /* restore aspm in case firmware bootup fails */
        ath12k_pci_aspm_restore(ab_pci);

        ath12k_pci_force_wake(ab_pci->ab);
        ath12k_pci_msi_disable(ab_pci);
        ath12k_mhi_stop(ab_pci, is_suspend);
        clear_bit(ATH12K_PCI_FLAG_INIT_DONE, &ab_pci->flags);
        ath12k_pci_sw_reset(ab_pci->ab, false);
}

static int ath12k_pci_panic_handler(struct ath12k_base *ab)
{
        ath12k_pci_sw_reset(ab, false);

        return NOTIFY_OK;
}

static const struct ath12k_hif_ops ath12k_pci_hif_ops = {
        .start = ath12k_pci_start,
        .stop = ath12k_pci_stop,
        .read32 = ath12k_pci_read32,
        .write32 = ath12k_pci_write32,
        .power_down = ath12k_pci_power_down,
        .power_up = ath12k_pci_power_up,
        .suspend = ath12k_pci_hif_suspend,
        .resume = ath12k_pci_hif_resume,
        .irq_enable = ath12k_pci_ext_irq_enable,
        .irq_disable = ath12k_pci_ext_irq_disable,
        .get_msi_address = ath12k_pci_get_msi_address,
        .get_user_msi_vector = ath12k_pci_get_user_msi_assignment,
        .map_service_to_pipe = ath12k_pci_map_service_to_pipe,
        .ce_irq_enable = ath12k_pci_hif_ce_irq_enable,
        .ce_irq_disable = ath12k_pci_hif_ce_irq_disable,
        .get_ce_msi_idx = ath12k_pci_get_ce_msi_idx,
        .panic_handler = ath12k_pci_panic_handler,
#ifdef CONFIG_ATH12K_COREDUMP
        .coredump_download = ath12k_pci_coredump_download,
#endif
};

static enum ath12k_device_family
ath12k_get_device_family(const struct pci_device_id *pci_dev)
{
        enum ath12k_device_family device_family_id;
        const struct pci_device_id *id;

        for (device_family_id = ATH12K_DEVICE_FAMILY_START;
             device_family_id < ATH12K_DEVICE_FAMILY_MAX; device_family_id++) {
                if (!ath12k_pci_family_drivers[device_family_id])
                        continue;

                id = ath12k_pci_family_drivers[device_family_id]->id_table;
                while (id->device) {
                        if (id->device == pci_dev->device)
                                return device_family_id;
                        id += 1;
                }
        }

        return ATH12K_DEVICE_FAMILY_MAX;
}

static int ath12k_pci_probe(struct pci_dev *pdev,
                            const struct pci_device_id *pci_dev)
{
        enum ath12k_device_family device_id;
        struct ath12k_pci *ab_pci;
        struct ath12k_base *ab;
        int ret;

        ab = ath12k_core_alloc(&pdev->dev, sizeof(*ab_pci), ATH12K_BUS_PCI);
        if (!ab) {
                dev_err(&pdev->dev, "failed to allocate ath12k base\n");
                return -ENOMEM;
        }

        ab->dev = &pdev->dev;
        ab_pci = ath12k_pci_priv(ab);
        ab_pci->dev_id = pci_dev->device;
        ab_pci->ab = ab;
        ab_pci->pdev = pdev;
        ab->hif.ops = &ath12k_pci_hif_ops;
        ab->fw_mode = ATH12K_FIRMWARE_MODE_NORMAL;
        pci_set_drvdata(pdev, ab);
        spin_lock_init(&ab_pci->window_lock);

        ret = ath12k_pci_claim(ab_pci, pdev);
        if (ret) {
                ath12k_err(ab, "failed to claim device: %d\n", ret);
                goto err_free_core;
        }

        ath12k_dbg(ab, ATH12K_DBG_BOOT, "pci probe %04x:%04x %04x:%04x\n",
                   pdev->vendor, pdev->device,
                   pdev->subsystem_vendor, pdev->subsystem_device);

        ab->id.vendor = pdev->vendor;
        ab->id.device = pdev->device;
        ab->id.subsystem_vendor = pdev->subsystem_vendor;
        ab->id.subsystem_device = pdev->subsystem_device;

        device_id = ath12k_get_device_family(pci_dev);
        if (device_id >= ATH12K_DEVICE_FAMILY_MAX) {
                ath12k_err(ab, "failed to get device family id\n");
                ret = -EINVAL;
                goto err_pci_free_region;
        }

        ath12k_dbg(ab, ATH12K_DBG_PCI, "PCI device family id: %d\n", device_id);

        ab_pci->device_family_ops = &ath12k_pci_family_drivers[device_id]->ops;
        ab_pci->reg_base = ath12k_pci_family_drivers[device_id]->reg_base;

        /* Call device specific probe. This is the callback that can
         * be used to override any ops in future
         * probe is validated for NULL during registration.
         */
        ret = ab_pci->device_family_ops->probe(pdev, pci_dev);
        if (ret) {
                ath12k_err(ab, "failed to probe device: %d\n", ret);
                goto err_pci_free_region;
        }

        ret = ath12k_pci_msi_alloc(ab_pci);
        if (ret) {
                ath12k_err(ab, "failed to alloc msi: %d\n", ret);
                goto err_pci_free_region;
        }

        ret = ath12k_core_pre_init(ab);
        if (ret)
                goto err_pci_msi_free;

        ret = ath12k_pci_set_irq_affinity_hint(ab_pci, cpumask_of(0));
        if (ret) {
                ath12k_err(ab, "failed to set irq affinity %d\n", ret);
                goto err_pci_msi_free;
        }

        ret = ath12k_mhi_register(ab_pci);
        if (ret) {
                ath12k_err(ab, "failed to register mhi: %d\n", ret);
                goto err_irq_affinity_cleanup;
        }

        ret = ath12k_hal_srng_init(ab);
        if (ret)
                goto err_mhi_unregister;

        ret = ath12k_ce_alloc_pipes(ab);
        if (ret) {
                ath12k_err(ab, "failed to allocate ce pipes: %d\n", ret);
                goto err_hal_srng_deinit;
        }

        ath12k_pci_init_qmi_ce_config(ab);

        ret = ath12k_pci_config_irq(ab);
        if (ret) {
                ath12k_err(ab, "failed to config irq: %d\n", ret);
                goto err_ce_free;
        }

        /* kernel may allocate a dummy vector before request_irq and
         * then allocate a real vector when request_irq is called.
         * So get msi_data here again to avoid spurious interrupt
         * as msi_data will configured to srngs.
         */
        ret = ath12k_pci_config_msi_data(ab_pci);
        if (ret) {
                ath12k_err(ab, "failed to config msi_data: %d\n", ret);
                goto err_free_irq;
        }

        /* Invoke arch_init here so that arch-specific init operations
         * can utilize already initialized ab fields, such as HAL SRNGs.
         */
        ret = ab_pci->device_family_ops->arch_init(ab);
        if (ret) {
                ath12k_err(ab, "PCI arch_init failed %d\n", ret);
                goto err_pci_msi_free;
        }

        ret = ath12k_core_init(ab);
        if (ret) {
                ath12k_err(ab, "failed to init core: %d\n", ret);
                goto err_deinit_arch;
        }
        return 0;

err_deinit_arch:
        ab_pci->device_family_ops->arch_deinit(ab);

err_free_irq:
        /* __free_irq() expects the caller to have cleared the affinity hint */
        ath12k_pci_set_irq_affinity_hint(ab_pci, NULL);
        ath12k_pci_free_irq(ab);

err_ce_free:
        ath12k_ce_free_pipes(ab);

err_hal_srng_deinit:
        ath12k_hal_srng_deinit(ab);

err_mhi_unregister:
        ath12k_mhi_unregister(ab_pci);

err_irq_affinity_cleanup:
        ath12k_pci_set_irq_affinity_hint(ab_pci, NULL);

err_pci_msi_free:
        ath12k_pci_msi_free(ab_pci);

err_pci_free_region:
        ath12k_pci_free_region(ab_pci);

err_free_core:
        ath12k_core_free(ab);

        return ret;
}

static void ath12k_pci_remove(struct pci_dev *pdev)
{
        struct ath12k_base *ab = pci_get_drvdata(pdev);
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);

        ath12k_pci_set_irq_affinity_hint(ab_pci, NULL);

        if (test_bit(ATH12K_FLAG_QMI_FAIL, &ab->dev_flags)) {
                ath12k_pci_power_down(ab, false);
                goto qmi_fail;
        }

        set_bit(ATH12K_FLAG_UNREGISTERING, &ab->dev_flags);

        cancel_work_sync(&ab->reset_work);
        cancel_work_sync(&ab->dump_work);
        ath12k_core_hw_group_cleanup(ab->ag);

qmi_fail:
        ath12k_core_deinit(ab);
        ath12k_fw_unmap(ab);
        ath12k_mhi_unregister(ab_pci);

        ath12k_pci_free_irq(ab);
        ath12k_pci_msi_free(ab_pci);
        ath12k_pci_free_region(ab_pci);

        ath12k_hal_srng_deinit(ab);
        ath12k_ce_free_pipes(ab);

        ab_pci->device_family_ops->arch_deinit(ab);

        ath12k_core_free(ab);
}

static void ath12k_pci_hw_group_power_down(struct ath12k_hw_group *ag)
{
        struct ath12k_base *ab;
        int i;

        if (!ag)
                return;

        mutex_lock(&ag->mutex);

        for (i = 0; i < ag->num_devices; i++) {
                ab = ag->ab[i];
                if (!ab)
                        continue;

                ath12k_pci_power_down(ab, false);
        }

        mutex_unlock(&ag->mutex);
}

static void ath12k_pci_shutdown(struct pci_dev *pdev)
{
        struct ath12k_base *ab = pci_get_drvdata(pdev);
        struct ath12k_pci *ab_pci = ath12k_pci_priv(ab);

        ath12k_pci_set_irq_affinity_hint(ab_pci, NULL);
        ath12k_pci_hw_group_power_down(ab->ag);
}

static __maybe_unused int ath12k_pci_pm_suspend(struct device *dev)
{
        struct ath12k_base *ab = dev_get_drvdata(dev);
        int ret;

        ret = ath12k_core_suspend(ab);
        if (ret)
                ath12k_warn(ab, "failed to suspend core: %d\n", ret);

        return ret;
}

static __maybe_unused int ath12k_pci_pm_resume(struct device *dev)
{
        struct ath12k_base *ab = dev_get_drvdata(dev);
        int ret;

        ret = ath12k_core_resume(ab);
        if (ret)
                ath12k_warn(ab, "failed to resume core: %d\n", ret);

        return ret;
}

static __maybe_unused int ath12k_pci_pm_suspend_late(struct device *dev)
{
        struct ath12k_base *ab = dev_get_drvdata(dev);
        int ret;

        ret = ath12k_core_suspend_late(ab);
        if (ret)
                ath12k_warn(ab, "failed to late suspend core: %d\n", ret);

        return ret;
}

static __maybe_unused int ath12k_pci_pm_resume_early(struct device *dev)
{
        struct ath12k_base *ab = dev_get_drvdata(dev);
        int ret;

        ret = ath12k_core_resume_early(ab);
        if (ret)
                ath12k_warn(ab, "failed to early resume core: %d\n", ret);

        return ret;
}

static const struct dev_pm_ops __maybe_unused ath12k_pci_pm_ops = {
        SET_SYSTEM_SLEEP_PM_OPS(ath12k_pci_pm_suspend,
                                ath12k_pci_pm_resume)
        SET_LATE_SYSTEM_SLEEP_PM_OPS(ath12k_pci_pm_suspend_late,
                                     ath12k_pci_pm_resume_early)
};

int ath12k_pci_register_driver(const enum ath12k_device_family device_id,
                               struct ath12k_pci_driver *driver)
{
        struct pci_driver *pci_driver;

        if (device_id >= ATH12K_DEVICE_FAMILY_MAX)
                return -EINVAL;

        if (!driver || !driver->ops.probe ||
            !driver->ops.arch_init || !driver->ops.arch_deinit)
                return -EINVAL;

        if (ath12k_pci_family_drivers[device_id]) {
                pr_err("Driver already registered for %d\n", device_id);
                return -EALREADY;
        }

        ath12k_pci_family_drivers[device_id] = driver;

        pci_driver = &ath12k_pci_family_drivers[device_id]->driver;
        pci_driver->name = driver->name;
        pci_driver->id_table = driver->id_table;
        pci_driver->probe = ath12k_pci_probe;
        pci_driver->remove = ath12k_pci_remove;
        pci_driver->shutdown = ath12k_pci_shutdown;
        pci_driver->driver.pm = &ath12k_pci_pm_ops;

        return pci_register_driver(pci_driver);
}
EXPORT_SYMBOL(ath12k_pci_register_driver);

void ath12k_pci_unregister_driver(const enum ath12k_device_family device_id)
{
        if (device_id >= ATH12K_DEVICE_FAMILY_MAX ||
            !ath12k_pci_family_drivers[device_id])
                return;

        pci_unregister_driver(&ath12k_pci_family_drivers[device_id]->driver);
        ath12k_pci_family_drivers[device_id] = NULL;
}
EXPORT_SYMBOL(ath12k_pci_unregister_driver);

/* firmware files */
MODULE_FIRMWARE(ATH12K_FW_DIR "/QCN9274/hw2.0/*");
MODULE_FIRMWARE(ATH12K_FW_DIR "/WCN7850/hw2.0/*");