Add queue initialization, release, enqueue, dequeue and other interface.

Signed-off-by: Hanxiao Li <li.hanxiao@zte.com.cn>
---
 drivers/common/zsda/meson.build |   1 +
 drivers/common/zsda/zsda_qp.c   | 727 ++++++++++++++++++++++++++++++++
 drivers/common/zsda/zsda_qp.h   | 164 +++++++
 3 files changed, 892 insertions(+)
 create mode 100644 drivers/common/zsda/zsda_qp.c
 create mode 100644 drivers/common/zsda/zsda_qp.h

diff --git a/drivers/common/zsda/meson.build b/drivers/common/zsda/meson.build
index d56b79b8e5..ee7e9e1720 100644
--- a/drivers/common/zsda/meson.build
+++ b/drivers/common/zsda/meson.build
@@ -8,4 +8,5 @@ sources += files(
         'zsda_common.c',
         'zsda_logs.c',
         'zsda_device.c',
+        'zsda_qp.c',
         )
diff --git a/drivers/common/zsda/zsda_qp.c b/drivers/common/zsda/zsda_qp.c
new file mode 100644
index 0000000000..afea2f3640
--- /dev/null
+++ b/drivers/common/zsda/zsda_qp.c
@@ -0,0 +1,727 @@
+/* SPDX-License-Identifier: BSD-3-Clause
+ * Copyright(c) 2024 ZTE Corporation
+ */
+
+#include <asm-generic/errno.h>
+#include <bits/stdint-uintn.h>
+#include <limits.h>
+#include <stdint.h>
+
+#include <rte_atomic.h>
+#include <rte_bus_pci.h>
+#include <rte_common.h>
+#include <rte_cycles.h>
+#include <rte_dev.h>
+#include <rte_malloc.h>
+#include <rte_memzone.h>
+#include <rte_pci.h>
+#include <rte_prefetch.h>
+
+#include "zsda_common.h"
+#include "zsda_logs.h"
+#include "zsda_device.h"
+#include "zsda_qp.h"
+
+#define RING_DIR_TX 0
+#define RING_DIR_RX 1
+
+struct ring_size {
+    uint16_t tx_msg_size;
+    uint16_t rx_msg_size;
+};
+
+struct ring_size zsda_qp_hw_ring_size[ZSDA_MAX_SERVICES] = {
+    [ZSDA_SERVICE_COMPRESSION] = {32, 16},
+    [ZSDA_SERVICE_DECOMPRESSION] = {32, 16},
+};
+
+static void
+zsda_set_queue_head_tail(struct zsda_pci_device *zsda_pci_dev, uint8_t qid)
+{
+    struct rte_pci_device *pci_dev =
+        zsda_devs[zsda_pci_dev->zsda_dev_id].pci_dev;
+    uint8_t *mmio_base = pci_dev->mem_resource[0].addr;
+
+    ZSDA_CSR_WRITE32(mmio_base + IO_DB_INITIAL_CONFIG + (qid * 4),
+             SET_HEAD_INTI);
+}
+
+int
+zsda_get_queue_cfg(struct zsda_pci_device *zsda_pci_dev)
+{
+    uint8_t i = 0;
+    uint32_t index = 0;
+    enum zsda_service_type type = ZSDA_SERVICE_INVALID;
+    struct zsda_qp_hw *zsda_hw_qps = zsda_pci_dev->zsda_hw_qps;
+    struct qinfo qcfg;
+    int ret = 0;
+
+    for (i = 0; i < zsda_num_used_qps; i++) {
+        zsda_set_queue_head_tail(zsda_pci_dev, i);
+        ret = zsda_get_queue_cfg_by_id(zsda_pci_dev, i, &qcfg);
+        type = qcfg.q_type;
+        if (ret) {
+            ZSDA_LOG(ERR, "get queue cfg!");
+            return ret;
+        }
+        if (type >= ZSDA_SERVICE_INVALID)
+            continue;
+
+        index = zsda_pci_dev->zsda_qp_hw_num[type];
+        zsda_hw_qps[type].data[index].used = true;
+        zsda_hw_qps[type].data[index].tx_ring_num = i;
+        zsda_hw_qps[type].data[index].rx_ring_num = i;
+        zsda_hw_qps[type].data[index].tx_msg_size =
+            zsda_qp_hw_ring_size[type].tx_msg_size;
+        zsda_hw_qps[type].data[index].rx_msg_size =
+            zsda_qp_hw_ring_size[type].rx_msg_size;
+
+        zsda_pci_dev->zsda_qp_hw_num[type]++;
+    }
+
+    return ret;
+}
+
+struct zsda_qp_hw *
+zsda_qps_hw_per_service(struct zsda_pci_device *zsda_pci_dev,
+            enum zsda_service_type service)
+{
+    struct zsda_qp_hw *qp_hw = NULL;
+
+    if (service < ZSDA_SERVICE_INVALID)
+        qp_hw = &(zsda_pci_dev->zsda_hw_qps[service]);
+
+    return qp_hw;
+}
+
+uint16_t
+zsda_qps_per_service(struct zsda_pci_device *zsda_pci_dev,
+             enum zsda_service_type service)
+{
+    uint16_t qp_hw_num = 0;
+    if (service < ZSDA_SERVICE_INVALID)
+        qp_hw_num = zsda_pci_dev->zsda_qp_hw_num[service];
+
+    return qp_hw_num;
+}
+
+uint16_t
+zsda_comp_max_nb_qps(struct zsda_pci_device *zsda_pci_dev)
+{
+    uint16_t comp =
+        zsda_qps_per_service(zsda_pci_dev, ZSDA_SERVICE_COMPRESSION);
+    uint16_t decomp =
+        zsda_qps_per_service(zsda_pci_dev, ZSDA_SERVICE_DECOMPRESSION);
+    uint16_t min = 0;
+
+    if ((comp == MAX_QPS_ON_FUNCTION) ||
+        (decomp == MAX_QPS_ON_FUNCTION))
+        min = MAX_QPS_ON_FUNCTION;
+    else
+        min = (comp < decomp) ? comp : decomp;
+    if (min == 0)
+        return MAX_QPS_ON_FUNCTION;
+    return min;
+}
+
+
+void
+zsda_stats_get(void **queue_pairs, uint32_t nb_queue_pairs,
+          struct zsda_common_stat *stats)
+{
+    enum zsda_service_type type;
+    uint32_t i = 0;
+    struct zsda_qp *qp;
+
+    if ((stats == NULL) || (queue_pairs == NULL)) {
+        ZSDA_LOG(ERR, E_NULL);
+        return;
+    }
+
+    for (i = 0; i < nb_queue_pairs; i++) {
+        qp = queue_pairs[i];
+
+        if (qp == NULL) {
+            ZSDA_LOG(ERR, E_NULL);
+            break;
+        }
+
+        for (type = 0; type < ZSDA_SERVICE_INVALID; type++) {
+            if (qp->srv[type].used) {
+                stats->enqueued_count +=
+                    qp->srv[type].stats.enqueued_count;
+                stats->dequeued_count +=
+                    qp->srv[type].stats.dequeued_count;
+                stats->enqueue_err_count +=
+                    qp->srv[type].stats.enqueue_err_count;
+                stats->dequeue_err_count +=
+                    qp->srv[type].stats.dequeue_err_count;
+            }
+        }
+    }
+}
+
+void
+zsda_stats_reset(void **queue_pairs, uint32_t nb_queue_pairs)
+{
+    enum zsda_service_type type;
+    uint32_t i = 0;
+    struct zsda_qp *qp;
+
+    if (queue_pairs == NULL) {
+        ZSDA_LOG(ERR, E_NULL);
+        return;
+    }
+
+    for (i = 0; i < nb_queue_pairs; i++) {
+        qp = queue_pairs[i];
+
+        if (qp == NULL) {
+            ZSDA_LOG(ERR, E_NULL);
+            break;
+        }
+        for (type = 0; type < ZSDA_MAX_SERVICES; type++) {
+            if (qp->srv[type].used)
+                memset(&(qp->srv[type].stats), 0,
+                       sizeof(struct zsda_common_stat));
+        }
+    }
+}
+
+static const struct rte_memzone *
+queue_dma_zone_reserve(const char *queue_name, unsigned int queue_size,
+               unsigned int socket_id)
+{
+    const struct rte_memzone *mz;
+
+    mz = rte_memzone_lookup(queue_name);
+    if (mz != 0) {
+        if (((size_t)queue_size <= mz->len) &&
+            ((socket_id == (SOCKET_ID_ANY & 0xffff)) ||
+             (socket_id == (mz->socket_id & 0xffff)))) {
+            ZSDA_LOG(DEBUG,
+                 "re-use memzone already allocated for %s",
+                 queue_name);
+            return mz;
+        }
+        ZSDA_LOG(ERR, E_MALLOC);
+        return NULL;
+    }
+
+    mz = rte_memzone_reserve_aligned(queue_name, queue_size,
+                       (int)(socket_id & 0xfff),
+                       RTE_MEMZONE_IOVA_CONTIG, queue_size);
+
+    return mz;
+}
+
+static int
+zsda_queue_create(uint32_t dev_id, struct zsda_queue *queue,
+          struct zsda_qp_config *qp_conf, uint8_t dir)
+{
+    void *io_addr;
+    const struct rte_memzone *qp_mz;
+    struct qinfo qcfg;
+
+    uint16_t desc_size = ((dir == RING_DIR_TX) ? qp_conf->hw->tx_msg_size
+                           : qp_conf->hw->rx_msg_size);
+    unsigned int queue_size_bytes = qp_conf->nb_descriptors * desc_size;
+
+    queue->hw_queue_number =
+        ((dir == RING_DIR_TX) ? qp_conf->hw->tx_ring_num
+                      : qp_conf->hw->rx_ring_num);
+
+    struct rte_pci_device *pci_dev = zsda_devs[dev_id].pci_dev;
+    struct zsda_pci_device *zsda_dev =
+        (struct zsda_pci_device *)zsda_devs[dev_id].mz->addr;
+
+    zsda_get_queue_cfg_by_id(zsda_dev, queue->hw_queue_number, &qcfg);
+
+    if (dir == RING_DIR_TX)
+        snprintf(queue->memz_name, sizeof(queue->memz_name),
+             "%s_%d_%s_%s_%d", pci_dev->driver->driver.name, dev_id,
+             qp_conf->service_str, "qptxmem",
+             queue->hw_queue_number);
+    else
+        snprintf(queue->memz_name, sizeof(queue->memz_name),
+             "%s_%d_%s_%s_%d", pci_dev->driver->driver.name, dev_id,
+             qp_conf->service_str, "qprxmem",
+             queue->hw_queue_number);
+
+    qp_mz = queue_dma_zone_reserve(queue->memz_name, queue_size_bytes,
+                       rte_socket_id());
+    if (qp_mz == NULL) {
+        ZSDA_LOG(ERR, E_MALLOC);
+        return -ENOMEM;
+    }
+
+    queue->base_addr = (uint8_t *)qp_mz->addr;
+    queue->base_phys_addr = qp_mz->iova;
+    queue->modulo_mask = MAX_NUM_OPS;
+    queue->msg_size = desc_size;
+
+    queue->head = (dir == RING_DIR_TX) ? qcfg.wq_head : qcfg.cq_head;
+    queue->tail = (dir == RING_DIR_TX) ? qcfg.wq_tail : qcfg.cq_tail;
+
+    if ((queue->head == 0) && (queue->tail == 0))
+        qcfg.cycle += 1;
+
+    queue->valid = qcfg.cycle & (ZSDA_MAX_CYCLE - 1);
+    queue->queue_size = ZSDA_MAX_DESC;
+    queue->cycle_size = ZSDA_MAX_CYCLE;
+    queue->io_addr = pci_dev->mem_resource[0].addr;
+
+    memset(queue->base_addr, 0x0, queue_size_bytes);
+    io_addr = pci_dev->mem_resource[0].addr;
+
+    if (dir == RING_DIR_TX)
+        ZSDA_CSR_WQ_RING_BASE(io_addr, queue->hw_queue_number,
+                      queue->base_phys_addr);
+    else
+        ZSDA_CSR_CQ_RING_BASE(io_addr, queue->hw_queue_number,
+                      queue->base_phys_addr);
+
+    return 0;
+}
+
+static void
+zsda_queue_delete(const struct zsda_queue *queue)
+{
+    const struct rte_memzone *mz;
+    int status = 0;
+
+    if (queue == NULL) {
+        ZSDA_LOG(DEBUG, "Invalid queue");
+        return;
+    }
+
+    mz = rte_memzone_lookup(queue->memz_name);
+    if (mz != NULL) {
+        memset(queue->base_addr, 0x0,
+               (uint16_t)(queue->queue_size * queue->msg_size));
+        status = rte_memzone_free(mz);
+        if (status != 0)
+            ZSDA_LOG(ERR, E_FREE);
+    } else
+        ZSDA_LOG(DEBUG, "queue %s doesn't exist", queue->memz_name);
+}
+
+static int
+cookie_init(uint32_t dev_id, struct zsda_qp **qp_addr, uint16_t queue_pair_id,
+        struct zsda_qp_config *zsda_qp_conf)
+{
+    struct zsda_qp *qp = *qp_addr;
+    struct rte_pci_device *pci_dev = zsda_devs[dev_id].pci_dev;
+    char op_cookie_pool_name[RTE_RING_NAMESIZE];
+    uint32_t i;
+    enum zsda_service_type type = zsda_qp_conf->service_type;
+
+    if (zsda_qp_conf->nb_descriptors != ZSDA_MAX_DESC)
+        ZSDA_LOG(ERR, "Can't create qp for %u descriptors",
+             zsda_qp_conf->nb_descriptors);
+
+    qp->srv[type].nb_descriptors = zsda_qp_conf->nb_descriptors;
+
+    qp->srv[type].op_cookies = rte_zmalloc_socket(
+        "zsda PMD op cookie pointer",
+        zsda_qp_conf->nb_descriptors *
+            sizeof(*qp->srv[type].op_cookies),
+        RTE_CACHE_LINE_SIZE, zsda_qp_conf->socket_id);
+
+    if (qp->srv[type].op_cookies == NULL) {
+        ZSDA_LOG(ERR, E_MALLOC);
+        return -ENOMEM;
+    }
+
+    snprintf(op_cookie_pool_name, RTE_RING_NAMESIZE, "%s%d_cks_%s_qp%hu",
+         pci_dev->driver->driver.name, dev_id,
+         zsda_qp_conf->service_str, queue_pair_id);
+
+    qp->srv[type].op_cookie_pool = rte_mempool_lookup(op_cookie_pool_name);
+    if (qp->srv[type].op_cookie_pool == NULL)
+        qp->srv[type].op_cookie_pool = rte_mempool_create(
+            op_cookie_pool_name, qp->srv[type].nb_descriptors,
+            zsda_qp_conf->cookie_size, 64, 0, NULL, NULL, NULL,
+            NULL, (int)(rte_socket_id() & 0xfff), 0);
+    if (!qp->srv[type].op_cookie_pool) {
+        ZSDA_LOG(ERR, E_CREATE);
+        goto exit;
+    }
+
+    for (i = 0; i < qp->srv[type].nb_descriptors; i++) {
+        if (rte_mempool_get(qp->srv[type].op_cookie_pool,
+                    &qp->srv[type].op_cookies[i])) {
+            ZSDA_LOG(ERR, "ZSDA PMD Cannot get op_cookie");
+            goto exit;
+        }
+        memset(qp->srv[type].op_cookies[i], 0,
+               zsda_qp_conf->cookie_size);
+    }
+    return 0;
+
+exit:
+    if (qp->srv[type].op_cookie_pool)
+        rte_mempool_free(qp->srv[type].op_cookie_pool);
+    rte_free(qp->srv[type].op_cookies);
+
+    return -EFAULT;
+}
+
+int
+zsda_queue_pair_setup(uint32_t dev_id, struct zsda_qp **qp_addr,
+              uint16_t queue_pair_id,
+              struct zsda_qp_config *zsda_qp_conf)
+{
+    struct zsda_qp *qp = *qp_addr;
+    struct rte_pci_device *pci_dev = zsda_devs[dev_id].pci_dev;
+    int ret = 0;
+    enum zsda_service_type type = zsda_qp_conf->service_type;
+
+    if (type >= ZSDA_SERVICE_INVALID) {
+        ZSDA_LOG(ERR, "Failed! servie type");
+        return -EINVAL;
+    }
+
+    if (pci_dev->mem_resource[0].addr == NULL) {
+        ZSDA_LOG(ERR, E_NULL);
+        return -EINVAL;
+    }
+
+    if (zsda_queue_create(dev_id, &(qp->srv[type].tx_q), zsda_qp_conf,
+                  RING_DIR_TX) != 0) {
+        ZSDA_LOG(ERR, E_CREATE);
+        return -EFAULT;
+    }
+
+    if (zsda_queue_create(dev_id, &(qp->srv[type].rx_q), zsda_qp_conf,
+                  RING_DIR_RX) != 0) {
+        ZSDA_LOG(ERR, E_CREATE);
+        zsda_queue_delete(&(qp->srv[type].tx_q));
+        return -EFAULT;
+    }
+
+    ret = cookie_init(dev_id, qp_addr, queue_pair_id, zsda_qp_conf);
+    if (ret) {
+        zsda_queue_delete(&(qp->srv[type].tx_q));
+        zsda_queue_delete(&(qp->srv[type].rx_q));
+        qp->srv[type].used = false;
+    }
+    qp->srv[type].used = true;
+    return ret;
+}
+
+int
+zsda_queue_pair_release(struct zsda_qp **qp_addr)
+{
+    struct zsda_qp *qp = *qp_addr;
+    uint32_t i;
+    enum zsda_service_type type = 0;
+
+    if (qp == NULL) {
+        ZSDA_LOG(DEBUG, "qp already freed");
+        return 0;
+    }
+
+    for (type = 0; type < ZSDA_SERVICE_INVALID; type++) {
+        if (!qp->srv[type].used)
+            continue;
+
+        zsda_queue_delete(&(qp->srv[type].tx_q));
+        zsda_queue_delete(&(qp->srv[type].rx_q));
+        qp->srv[type].used = false;
+        for (i = 0; i < qp->srv[type].nb_descriptors; i++)
+            rte_mempool_put(qp->srv[type].op_cookie_pool,
+                    qp->srv[type].op_cookies[i]);
+
+        if (qp->srv[type].op_cookie_pool)
+            rte_mempool_free(qp->srv[type].op_cookie_pool);
+
+        rte_free(qp->srv[type].op_cookies);
+    }
+
+    rte_free(qp);
+    *qp_addr = NULL;
+
+    return 0;
+}
+
+int
+zsda_fill_sgl(struct rte_mbuf *buf, uint32_t offset,
+              struct zsda_sgl *sgl, phys_addr_t sgl_phy_addr,
+              uint32_t remain_len, struct comp_head_info *comp_head_info)
+{
+    uint32_t nr = 0;
+    uint16_t put_in_len = 0;
+    bool head_set = false;
+
+    for (nr = 0; (buf && (nr < (ZSDA_SGL_MAX_NUMBER - 1)));) {
+        if (offset >= rte_pktmbuf_data_len(buf)) {
+            offset -= rte_pktmbuf_data_len(buf);
+            buf = buf->next;
+            continue;
+        }
+        memset(&(sgl->buffers[nr]), 0, sizeof(struct zsda_buf));
+        if ((nr > 0) && (((nr + 1) % ZSDA_SGL_FRAGMENT_SIZE) == 0) &&
+            (buf->next != NULL)) {
+            sgl->buffers[nr].len = SGL_TYPE_PHYS_ADDR;
+            sgl->buffers[nr].addr =
+                sgl_phy_addr +
+                ((nr + 1) * sizeof(struct zsda_buf));
+            sgl->buffers[nr].type = SGL_TYPE_NEXT_LIST;
+            ++nr;
+            continue;
+        }
+        if (comp_head_info && !head_set) {
+            sgl->buffers[nr].len = comp_head_info->head_len;
+            sgl->buffers[nr].addr = comp_head_info->head_phys_addr;
+            sgl->buffers[nr].type = SGL_TYPE_PHYS_ADDR;
+            ++nr;
+            head_set = true;
+            remain_len -= comp_head_info->head_len;
+            continue;
+        } else {
+            put_in_len = rte_pktmbuf_data_len(buf) - (offset & 0xffff);
+            if (remain_len <= put_in_len)
+                put_in_len = remain_len;
+            remain_len -= put_in_len;
+
+            sgl->buffers[nr].len = put_in_len;
+            sgl->buffers[nr].addr = rte_pktmbuf_iova_offset(buf, offset);
+            sgl->buffers[nr].type = SGL_TYPE_PHYS_ADDR;
+        }
+        offset = 0;
+        ++nr;
+        buf = buf->next;
+
+        if (remain_len == 0)
+            break;
+    }
+
+    if (nr == 0) {
+        ZSDA_LOG(ERR, "In fill_sgl, nr == 0");
+        return ZSDA_FAILED;
+    }
+
+    sgl->buffers[nr - 1].type = SGL_TYPE_LAST_PHYS_ADDR;
+
+    if (buf) {
+        if (unlikely(buf->next)) {
+            if (nr == (ZSDA_SGL_MAX_NUMBER - 1)) {
+                ZSDA_LOG(ERR, "ERR! segs size (%u)",
+                     (ZSDA_SGL_MAX_NUMBER));
+                return -EINVAL;
+            }
+        }
+    }
+
+    return ZSDA_SUCCESS;
+}
+
+int
+zsda_get_sgl_num(struct zsda_sgl *sgl)
+{
+    int sgl_num = 0;
+
+    while (sgl->buffers[sgl_num].type != 1) {
+        sgl_num++;
+        if (sgl_num >= ZSDA_SGL_MAX_NUMBER)
+            return ZSDA_FAILED;
+    }
+    sgl_num++;
+    return sgl_num;
+}
+
+int
+find_next_free_cookie(struct zsda_queue *queue, void **op_cookie, uint16_t *idx)
+{
+    uint16_t old_tail = queue->tail;
+    uint16_t tail = queue->tail;
+    struct zsda_op_cookie *cookie = NULL;
+
+    do {
+        cookie = (struct zsda_op_cookie *)op_cookie[tail];
+        if (!cookie->used) {
+            *idx = tail & (queue->queue_size - 1);
+            return 0;
+        }
+        tail = zsda_modulo_16(tail++, queue->modulo_mask);
+    } while (old_tail != tail);
+
+    return -EINVAL;
+}
+
+static int
+enqueue(void *op, struct zsda_qp *qp)
+{
+    uint16_t new_tail;
+    enum zsda_service_type type;
+    void **op_cookie;
+    int ret = 0;
+    struct zsda_queue *queue;
+
+    for (type = 0; type < ZSDA_SERVICE_INVALID; type++) {
+        if (qp->srv[type].used) {
+            if (!qp->srv[type].match(op))
+                continue;
+            queue = &qp->srv[type].tx_q;
+            op_cookie = qp->srv[type].op_cookies;
+
+            if (find_next_free_cookie(queue, op_cookie,
+                          &new_tail)) {
+                ret = -EBUSY;
+                break;
+            }
+            ret = qp->srv[type].tx_cb(op, queue, op_cookie,
+                          new_tail);
+            if (ret) {
+                qp->srv[type].stats.enqueue_err_count++;
+                ZSDA_LOG(ERR, "Failed! config wqe");
+                break;
+            }
+            qp->srv[type].stats.enqueued_count++;
+
+            queue->tail = zsda_modulo_16(new_tail + 1,
+                             queue->queue_size - 1);
+
+            if (new_tail > queue->tail)
+                queue->valid =
+                    zsda_modulo_8(queue->valid + 1,
+                    (uint8_t)(queue->cycle_size - 1));
+
+            queue->pushed_wqe++;
+            break;
+        }
+    }
+
+    return ret;
+}
+
+void
+tx_write_tail(struct zsda_queue *queue)
+{
+    if (queue->pushed_wqe)
+        WRITE_CSR_WQ_TAIL(queue->io_addr, queue->hw_queue_number,
+                  queue->tail);
+
+    queue->pushed_wqe = 0;
+}
+
+uint16_t
+zsda_enqueue_op_burst(struct zsda_qp *qp, void **ops, uint16_t nb_ops)
+{
+    int ret = -1;
+    enum zsda_service_type type = 0;
+    uint16_t i = 0;
+    uint16_t nb_send = 0;
+    void *op;
+
+    if (nb_ops > ZSDA_MAX_DESC) {
+        ZSDA_LOG(ERR, "Enqueue number bigger than %d", ZSDA_MAX_DESC);
+        return 0;
+    }
+
+    for (i = 0; i < nb_ops; i++) {
+        op = ops[i];
+        ret = enqueue(op, qp);
+        if (ret < 0)
+            break;
+        nb_send++;
+    }
+
+    for (type = 0; type < ZSDA_SERVICE_INVALID; type++)
+        if (qp->srv[type].used)
+            tx_write_tail(&qp->srv[type].tx_q);
+
+    return nb_send;
+}
+
+static void
+dequeue(struct qp_srv *srv, void **ops, uint32_t nb_ops, uint32_t *nb)
+{
+    uint16_t head;
+    struct zsda_cqe *cqe = NULL;
+    struct zsda_queue *queue = &srv->rx_q;
+    struct zsda_op_cookie *cookie = NULL;
+    head = queue->head;
+
+    while (*nb < nb_ops) {
+        cqe = (struct zsda_cqe *)((uint8_t *)queue->base_addr + head * queue->msg_size);
+
+        if (!CQE_VALID(cqe->err1))
+            break;
+        cookie = (struct zsda_op_cookie *)srv->op_cookies[cqe->sid];
+
+        if (cookie->decomp_no_tail && CQE_ERR0_RIGHT(cqe->err0))
+            cqe->err0 = 0x0000;
+
+        if (CQE_ERR0(cqe->err0) || CQE_ERR1(cqe->err1)) {
+            ZSDA_LOG(ERR,
+                 "ERR! Cqe, opcode 0x%x, sid 0x%x, "
+                 "tx_real_length 0x%x, err0 0x%x, err1 0x%x",
+                 cqe->op_code, cqe->sid, cqe->tx_real_length,
+                 cqe->err0, cqe->err1);
+            srv->stats.dequeue_err_count++;
+        } else
+            srv->stats.dequeued_count++;
+
+        ops[*nb] = cookie->op;
+        srv->rx_cb(cookie, cqe);
+        (*nb)++;
+        cookie->used = false;
+
+        head = zsda_modulo_16(head + 1, queue->modulo_mask);
+        queue->head = head;
+        WRITE_CSR_CQ_HEAD(queue->io_addr, queue->hw_queue_number, head);
+        memset(cqe, 0x0, sizeof(struct zsda_cqe));
+    }
+}
+
+uint16_t
+zsda_dequeue_op_burst(struct zsda_qp *qp, void **ops, uint16_t nb_ops)
+{
+    uint32_t nb = 0;
+    uint32_t type = 0;
+    struct qp_srv *srv;
+
+    for (type = 0; type < ZSDA_SERVICE_INVALID; type++) {
+        if (!qp->srv[type].used)
+            continue;
+        srv = &qp->srv[type];
+        dequeue(srv, ops, nb_ops, &nb);
+        if (nb >= nb_ops)
+            return nb_ops;
+    }
+    return nb;
+}
+
+int
+common_setup_qp(uint32_t zsda_dev_id, struct zsda_qp **qp_addr,
+        uint16_t queue_pair_id, struct zsda_qp_config *conf)
+{
+    uint32_t i;
+    int ret = 0;
+    struct zsda_qp *qp;
+    rte_iova_t cookie_phys_addr;
+
+    ret = zsda_queue_pair_setup(zsda_dev_id, qp_addr, queue_pair_id, conf);
+    if (ret)
+        return ret;
+
+    qp = (struct zsda_qp *)*qp_addr;
+
+    for (i = 0; i < qp->srv[conf->service_type].nb_descriptors; i++) {
+        struct zsda_op_cookie *cookie =
+            qp->srv[conf->service_type].op_cookies[i];
+        cookie_phys_addr = rte_mempool_virt2iova(cookie);
+
+        cookie->comp_head_phys_addr = cookie_phys_addr +
+            offsetof(struct zsda_op_cookie, comp_head);
+
+        cookie->sgl_src_phys_addr = cookie_phys_addr +
+            offsetof(struct zsda_op_cookie, sgl_src);
+
+        cookie->sgl_dst_phys_addr = cookie_phys_addr +
+            offsetof(struct zsda_op_cookie, sgl_dst);
+    }
+    return ret;
+}
diff --git a/drivers/common/zsda/zsda_qp.h b/drivers/common/zsda/zsda_qp.h
new file mode 100644
index 0000000000..27d033128c
--- /dev/null
+++ b/drivers/common/zsda/zsda_qp.h
@@ -0,0 +1,164 @@
+/* SPDX-License-Identifier: BSD-3-Clause
+ * Copyright(c) 2024 ZTE Corporation
+ */
+
+#ifndef _ZSDA_QP_H_
+#define _ZSDA_QP_H_
+
+#include <rte_bus_pci.h>
+
+#define WQ_CSR_LBASE 0x1000
+#define WQ_CSR_UBASE 0x1004
+#define CQ_CSR_LBASE 0x1400
+#define CQ_CSR_UBASE 0x1404
+#define WQ_TAIL         0x1800
+#define CQ_HEAD         0x1804
+
+/**
+ * Structure associated with each queue.
+ */
+struct zsda_queue {
+    char memz_name[RTE_MEMZONE_NAMESIZE];
+    uint8_t *io_addr;
+    uint8_t *base_addr;       /* Base address */
+    rte_iova_t base_phys_addr; /* Queue physical address */
+    uint16_t head;           /* Shadow copy of the head */
+    uint16_t tail;           /* Shadow copy of the tail */
+    uint16_t modulo_mask;
+    uint16_t msg_size;
+    uint16_t queue_size;
+    uint16_t cycle_size;
+    uint16_t pushed_wqe;
+
+    uint8_t hw_queue_number;
+    uint32_t csr_head; /* last written head value */
+    uint32_t csr_tail; /* last written tail value */
+
+    uint8_t valid;
+    uint16_t sid;
+};
+
+typedef void (*rx_callback)(void *cookie_in, struct zsda_cqe *cqe);
+typedef int (*tx_callback)(void *op_in, const struct zsda_queue *queue,
+               void **op_cookies, uint16_t new_tail);
+typedef int (*srv_match)(void *op_in);
+
+struct qp_srv {
+    bool used;
+    struct zsda_queue tx_q;
+    struct zsda_queue rx_q;
+    rx_callback rx_cb;
+    tx_callback tx_cb;
+    srv_match match;
+    struct zsda_common_stat stats;
+    struct rte_mempool *op_cookie_pool;
+    void **op_cookies;
+    uint16_t nb_descriptors;
+};
+
+struct zsda_qp {
+    void *mmap_bar_addr;
+    struct qp_srv srv[ZSDA_MAX_SERVICES];
+
+    uint16_t max_inflights;
+    uint16_t min_enq_burst_threshold;
+
+} __rte_cache_aligned;
+
+struct zsda_qp_config {
+    enum zsda_service_type service_type;
+    const struct zsda_qp_hw_data *hw;
+    uint16_t nb_descriptors;
+    uint32_t cookie_size;
+    int socket_id;
+    const char *service_str;
+};
+
+struct comp_head_info {
+    uint32_t head_len;
+    phys_addr_t head_phys_addr;
+};
+
+extern uint8_t zsda_num_used_qps;
+
+struct zsda_qp_hw *zsda_qps_hw_per_service(struct zsda_pci_device *zsda_pci_dev,
+                       enum zsda_service_type service);
+uint16_t zsda_qps_per_service(struct zsda_pci_device *zsda_pci_dev,
+                  enum zsda_service_type service);
+
+uint16_t zsda_comp_max_nb_qps(struct zsda_pci_device *zsda_pci_dev);
+uint16_t zsda_crypto_max_nb_qps(struct zsda_pci_device *zsda_pci_dev);
+
+int zsda_get_queue_cfg(struct zsda_pci_device *zsda_pci_dev);
+
+/* CSR write macro */
+#define ZSDA_CSR_WR(csrAddr, csrOffset, val)                                   \
+    rte_write32(val, (((uint8_t *)csrAddr) + csrOffset))
+#define ZSDA_CSR_WC_WR(csrAddr, csrOffset, val)                                \
+    rte_write32_wc(val, (((uint8_t *)csrAddr) + csrOffset))
+
+/* CSR read macro */
+#define ZSDA_CSR_RD(csrAddr, csrOffset)                                        \
+    rte_read32((((uint8_t *)csrAddr) + csrOffset))
+
+#define ZSDA_CSR_WQ_RING_BASE(csr_base_addr, ring, value)                      \
+    do {                                                                   \
+        uint32_t l_base = 0, u_base = 0;                               \
+        l_base = (uint32_t)(value & 0xFFFFFFFF);                       \
+        u_base = (uint32_t)((value & 0xFFFFFFFF00000000ULL) >> 32);    \
+        ZSDA_CSR_WR(csr_base_addr, (ring << 3) + WQ_CSR_LBASE,         \
+                l_base);                                           \
+        ZSDA_LOG(INFO, "l_basg - offest:0x%x, value:0x%x",             \
+             ((ring << 3) + WQ_CSR_LBASE), l_base);                \
+        ZSDA_CSR_WR(csr_base_addr, (ring << 3) + WQ_CSR_UBASE,         \
+                u_base);                                           \
+        ZSDA_LOG(INFO, "h_base - offest:0x%x, value:0x%x",             \
+             ((ring << 3) + WQ_CSR_UBASE), u_base);                \
+    } while (0)
+
+#define ZSDA_CSR_CQ_RING_BASE(csr_base_addr, ring, value)                      \
+    do {                                                                   \
+        uint32_t l_base = 0, u_base = 0;                               \
+        l_base = (uint32_t)(value & 0xFFFFFFFF);                       \
+        u_base = (uint32_t)((value & 0xFFFFFFFF00000000ULL) >> 32);    \
+        ZSDA_CSR_WR(csr_base_addr, (ring << 3) + CQ_CSR_LBASE,         \
+                l_base);                                           \
+        ZSDA_CSR_WR(csr_base_addr, (ring << 3) + CQ_CSR_UBASE,         \
+                u_base);                                           \
+    } while (0)
+
+#define READ_CSR_WQ_HEAD(csr_base_addr, ring)                                  \
+    ZSDA_CSR_RD(csr_base_addr, WQ_TAIL + (ring << 3))
+#define WRITE_CSR_WQ_TAIL(csr_base_addr, ring, value)                          \
+    ZSDA_CSR_WC_WR(csr_base_addr, WQ_TAIL + (ring << 3), value)
+#define READ_CSR_CQ_HEAD(csr_base_addr, ring)                                  \
+    ZSDA_CSR_RD(csr_base_addr, WQ_TAIL + (ring << 3))
+#define WRITE_CSR_CQ_HEAD(csr_base_addr, ring, value)                          \
+    ZSDA_CSR_WC_WR(csr_base_addr, CQ_HEAD + (ring << 3), value)
+
+uint16_t zsda_enqueue_op_burst(struct zsda_qp *qp, void **ops, uint16_t nb_ops);
+uint16_t zsda_dequeue_op_burst(struct zsda_qp *qp, void **ops, uint16_t nb_ops);
+
+void tx_write_tail(struct zsda_queue *queue);
+int zsda_queue_pair_setup(uint32_t dev_id, struct zsda_qp **qp_addr,
+              uint16_t queue_pair_id,
+              struct zsda_qp_config *zsda_qp_conf);
+
+int zsda_queue_pair_release(struct zsda_qp **qp_addr);
+int zsda_fill_sgl(struct rte_mbuf *buf, uint32_t offset,
+             struct zsda_sgl *sgl, phys_addr_t sgl_phy_addr,
+             uint32_t remain_len, struct comp_head_info *comp_head_info);
+
+int zsda_get_sgl_num(struct zsda_sgl *sgl);
+int zsda_sgl_opt_addr_lost(struct rte_mbuf *mbuf);
+
+int find_next_free_cookie(struct zsda_queue *queue, void **op_cookie,
+              uint16_t *idx);
+int common_setup_qp(uint32_t dev_id, struct zsda_qp **qp_addr,
+            uint16_t queue_pair_id, struct zsda_qp_config *conf);
+
+void zsda_stats_get(void **queue_pairs, uint32_t nb_queue_pairs,
+           struct zsda_common_stat *stats);
+void zsda_stats_reset(void **queue_pairs, uint32_t nb_queue_pairs);
+
+#endif /* _ZSDA_QP_H_ */
-- 
2.27.0