Add support for zsdadev operations such as dev_start and dev_stop.

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

diff --git a/drivers/common/zsda/meson.build b/drivers/common/zsda/meson.build
index b738f74e31..d56b79b8e5 100644
--- a/drivers/common/zsda/meson.build
+++ b/drivers/common/zsda/meson.build
@@ -7,4 +7,5 @@ deps += ['bus_pci', 'compressdev']
 sources += files(
         'zsda_common.c',
         'zsda_logs.c',
+        'zsda_device.c',
         )
diff --git a/drivers/common/zsda/zsda_device.c b/drivers/common/zsda/zsda_device.c
new file mode 100644
index 0000000000..7e2d84cb79
--- /dev/null
+++ b/drivers/common/zsda/zsda_device.c
@@ -0,0 +1,480 @@
+/* SPDX-License-Identifier: BSD-3-Clause
+ * Copyright(c) 2024 ZTE Corporation
+ */
+
+#include <ctype.h>
+#include <errno.h>
+#include <stdint.h>
+#include <unistd.h>
+
+#include <rte_devargs.h>
+#include <rte_string_fns.h>
+
+#include "zsda_device.h"
+#include "dev_driver.h"
+
+/* per-process array of device data */
+struct zsda_device_info zsda_devs[RTE_PMD_ZSDA_MAX_PCI_DEVICES];
+static int zsda_nb_pci_devices;
+uint8_t zsda_num_used_qps;
+
+/*
+ * The set of PCI devices this driver supports
+ */
+static const struct rte_pci_id pci_id_zsda_map[] = {
+    {
+        RTE_PCI_DEVICE(0x1cf2, 0x8050),
+    },
+    {
+        RTE_PCI_DEVICE(0x1cf2, 0x8051),
+    },
+    {.device_id = 0},
+};
+
+static int
+zsda_check_write(uint8_t *addr, uint32_t dst_value)
+{
+    int times = ZSDA_TIME_NUM;
+    uint32_t val = 0;
+
+    val = ZSDA_CSR_READ32(addr);
+
+    while ((val != dst_value) && times--) {
+        val = ZSDA_CSR_READ32(addr);
+        rte_delay_us_sleep(ZSDA_TIME_SLEEP_US);
+    }
+    if (val == dst_value)
+        return ZSDA_SUCCESS;
+    else
+        return ZSDA_FAILED;
+}
+
+static uint8_t
+zsda_get_num_used_qps(const struct rte_pci_device *pci_dev)
+{
+    uint8_t *mmio_base = pci_dev->mem_resource[0].addr;
+    uint8_t num_used_qps;
+    num_used_qps = ZSDA_CSR_READ8(mmio_base + 0);
+
+    return num_used_qps;
+}
+
+int
+zsda_admin_q_start(const struct rte_pci_device *pci_dev)
+{
+    uint8_t *mmio_base = pci_dev->mem_resource[0].addr;
+    int ret = 0;
+
+    ZSDA_CSR_WRITE32(mmio_base + ZSDA_ADMIN_Q_START, 0);
+
+    ZSDA_CSR_WRITE32(mmio_base + ZSDA_ADMIN_Q_START, ZSDA_Q_START);
+    ret = zsda_check_write(mmio_base + ZSDA_ADMIN_Q_START, ZSDA_Q_START);
+
+    return ret;
+}
+
+int
+zsda_admin_q_stop(const struct rte_pci_device *pci_dev)
+{
+    uint8_t *mmio_base = pci_dev->mem_resource[0].addr;
+    int ret = 0;
+
+    ZSDA_CSR_WRITE32(mmio_base + ZSDA_ADMIN_Q_STOP_RESP, ZSDA_RESP_INVALID);
+    ZSDA_CSR_WRITE32(mmio_base + ZSDA_ADMIN_Q_STOP, ZSDA_Q_STOP);
+
+    ret = zsda_check_write(mmio_base + ZSDA_ADMIN_Q_STOP_RESP,
+                   ZSDA_RESP_VALID);
+
+    if (ret)
+        ZSDA_LOG(INFO, "Failed! zsda_admin q stop");
+
+    return ret;
+}
+
+int
+zsda_admin_q_clear(const struct rte_pci_device *pci_dev)
+{
+    uint8_t *mmio_base = pci_dev->mem_resource[0].addr;
+    int ret = 0;
+
+    ZSDA_CSR_WRITE32(mmio_base + ZSDA_ADMIN_Q_CLR_RESP, ZSDA_RESP_INVALID);
+    ZSDA_CSR_WRITE32(mmio_base + ZSDA_ADMIN_Q_CLR, ZSDA_RESP_VALID);
+
+    ret = zsda_check_write(mmio_base + ZSDA_ADMIN_Q_CLR_RESP,
+                   ZSDA_RESP_VALID);
+
+    if (ret)
+        ZSDA_LOG(INFO, "Failed! zsda_admin q clear");
+
+    return ret;
+}
+
+static int
+zsda_queue_stop_single(uint8_t *mmio_base, uint8_t id)
+{
+    int ret = ZSDA_SUCCESS;
+    uint8_t *addr_stop = mmio_base + ZSDA_IO_Q_STOP + (4 * id);
+    uint8_t *addr_resp = mmio_base + ZSDA_IO_Q_STOP_RESP + (4 * id);
+
+    ZSDA_CSR_WRITE32(addr_resp, ZSDA_RESP_INVALID);
+    ZSDA_CSR_WRITE32(addr_stop, ZSDA_Q_STOP);
+
+    ret = zsda_check_write(addr_resp, ZSDA_RESP_VALID);
+    ZSDA_CSR_WRITE32(addr_resp, ZSDA_RESP_INVALID);
+
+    return ret;
+}
+
+int
+zsda_queue_stop(const struct rte_pci_device *pci_dev)
+{
+    uint8_t *mmio_base = pci_dev->mem_resource[0].addr;
+    uint8_t id = 0;
+    int ret = ZSDA_SUCCESS;
+
+    for (id = 0; id < zsda_num_used_qps; id++)
+        ret |= zsda_queue_stop_single(mmio_base, id);
+
+    return ret;
+}
+
+static int
+zsda_queue_start_single(uint8_t *mmio_base, uint8_t id)
+{
+    uint8_t *addr_start = mmio_base + ZSDA_IO_Q_START + (4 * id);
+
+    ZSDA_CSR_WRITE32(addr_start, ZSDA_Q_START);
+    return zsda_check_write(addr_start, ZSDA_Q_START);
+}
+
+int
+zsda_queue_start(const struct rte_pci_device *pci_dev)
+{
+    uint8_t *mmio_base = pci_dev->mem_resource[0].addr;
+    uint8_t id = 0;
+    int ret = ZSDA_SUCCESS;
+
+    for (id = 0; id < zsda_num_used_qps; id++)
+        ret |= zsda_queue_start_single(mmio_base, id);
+
+    return ret;
+}
+
+static int
+zsda_queue_clear_single(uint8_t *mmio_base, uint8_t id)
+{
+    int ret = ZSDA_SUCCESS;
+    uint8_t *addr_clear = mmio_base + ZSDA_IO_Q_CLR + (4 * id);
+    uint8_t *addr_resp = mmio_base + ZSDA_IO_Q_CLR_RESP + (4 * id);
+
+    ZSDA_CSR_WRITE32(addr_resp, ZSDA_RESP_INVALID);
+    ZSDA_CSR_WRITE32(addr_clear, ZSDA_CLEAR_VALID);
+    ret = zsda_check_write(addr_resp, ZSDA_RESP_VALID);
+    ZSDA_CSR_WRITE32(addr_clear, ZSDA_CLEAR_INVALID);
+
+    return ret;
+}
+
+int
+zsda_queue_clear(const struct rte_pci_device *pci_dev)
+{
+    uint8_t *mmio_base = pci_dev->mem_resource[0].addr;
+    uint8_t id = 0;
+    int ret = ZSDA_SUCCESS;
+
+    for (id = 0; id < zsda_num_used_qps; id++)
+        ret |= zsda_queue_clear_single(mmio_base, id);
+
+    return ret;
+}
+
+static struct zsda_pci_device *
+zsda_pci_get_named_dev(const char *name)
+{
+    unsigned int i;
+
+    if (name == NULL) {
+        ZSDA_LOG(ERR, E_NULL);
+        return NULL;
+    }
+
+    for (i = 0; i < RTE_PMD_ZSDA_MAX_PCI_DEVICES; i++) {
+        if (zsda_devs[i].mz &&
+            (strcmp(((struct zsda_pci_device *)zsda_devs[i].mz->addr)
+                    ->name,
+                name) == 0))
+            return (struct zsda_pci_device *)zsda_devs[i].mz->addr;
+    }
+
+    return NULL;
+}
+
+static uint8_t
+zsda_pci_find_free_device_index(void)
+{
+    uint32_t dev_id;
+
+    for (dev_id = 0; dev_id < RTE_PMD_ZSDA_MAX_PCI_DEVICES; dev_id++)
+        if (zsda_devs[dev_id].mz == NULL)
+            break;
+
+    return dev_id & (ZSDA_MAX_DEV - 1);
+}
+
+struct zsda_pci_device *
+zsda_get_zsda_dev_from_pci_dev(struct rte_pci_device *pci_dev)
+{
+    char name[ZSDA_DEV_NAME_MAX_LEN];
+
+    rte_pci_device_name(&pci_dev->addr, name, sizeof(name));
+
+    return zsda_pci_get_named_dev(name);
+}
+
+struct zsda_pci_device *
+zsda_pci_device_allocate(struct rte_pci_device *pci_dev)
+{
+    struct zsda_pci_device *zsda_pci_dev;
+    uint8_t zsda_dev_id = 0;
+    char name[ZSDA_DEV_NAME_MAX_LEN];
+
+    rte_pci_device_name(&pci_dev->addr, name, sizeof(name));
+    snprintf(name + strlen(name), (ZSDA_DEV_NAME_MAX_LEN - strlen(name)),
+         "_zsda");
+    if (rte_eal_process_type() == RTE_PROC_SECONDARY) {
+        const struct rte_memzone *mz = rte_memzone_lookup(name);
+
+        if (mz == NULL) {
+            ZSDA_LOG(ERR, "Secondary can't find %s mz", name);
+            return NULL;
+        }
+        zsda_pci_dev = mz->addr;
+        zsda_devs[zsda_pci_dev->zsda_dev_id].mz = mz;
+        zsda_devs[zsda_pci_dev->zsda_dev_id].pci_dev = pci_dev;
+        zsda_nb_pci_devices++;
+        return zsda_pci_dev;
+    }
+
+    if (zsda_pci_get_named_dev(name) != NULL) {
+        ZSDA_LOG(ERR, E_CONFIG);
+        return NULL;
+    }
+
+    zsda_dev_id = zsda_pci_find_free_device_index();
+
+    if (zsda_dev_id == (RTE_PMD_ZSDA_MAX_PCI_DEVICES - 1)) {
+        ZSDA_LOG(ERR, "Reached maximum number of ZSDA devices");
+        return NULL;
+    }
+
+    unsigned int socket_id = rte_socket_id();
+
+    zsda_devs[zsda_dev_id].mz =
+        rte_memzone_reserve(name, sizeof(struct zsda_pci_device),
+                    (int)(socket_id & 0xfff), 0);
+
+    if (zsda_devs[zsda_dev_id].mz == NULL) {
+        ZSDA_LOG(ERR, E_MALLOC);
+        return NULL;
+    }
+
+    zsda_pci_dev = zsda_devs[zsda_dev_id].mz->addr;
+    memset(zsda_pci_dev, 0, sizeof(*zsda_pci_dev));
+    strlcpy(zsda_pci_dev->name, name, ZSDA_DEV_NAME_MAX_LEN);
+    zsda_pci_dev->zsda_dev_id = zsda_dev_id;
+    zsda_pci_dev->pci_dev = pci_dev;
+    zsda_devs[zsda_dev_id].pci_dev = pci_dev;
+
+    rte_spinlock_init(&zsda_pci_dev->arb_csr_lock);
+    zsda_nb_pci_devices++;
+
+    return zsda_pci_dev;
+}
+
+static int
+zsda_pci_device_release(struct rte_pci_device *pci_dev)
+{
+    struct zsda_pci_device *zsda_pci_dev;
+    struct zsda_device_info *inst;
+    char name[ZSDA_DEV_NAME_MAX_LEN];
+
+    if (pci_dev == NULL)
+        return -EINVAL;
+
+    rte_pci_device_name(&pci_dev->addr, name, sizeof(name));
+
+    snprintf(name + strlen(name),
+         ZSDA_DEV_NAME_MAX_LEN - (strlen(name) - 1), "_zsda");
+    zsda_pci_dev = zsda_pci_get_named_dev(name);
+    if (zsda_pci_dev != NULL) {
+        inst = &zsda_devs[zsda_pci_dev->zsda_dev_id];
+
+        if (rte_eal_process_type() == RTE_PROC_PRIMARY) {
+            if (zsda_pci_dev->comp_dev != NULL) {
+                ZSDA_LOG(DEBUG, "ZSDA device %s is busy", name);
+                return -EBUSY;
+            }
+            rte_memzone_free(inst->mz);
+        }
+        memset(inst, 0, sizeof(struct zsda_device_info));
+        zsda_nb_pci_devices--;
+    }
+    return 0;
+}
+
+static int
+zsda_pci_dev_destroy(struct zsda_pci_device *zsda_pci_dev,
+             struct rte_pci_device *pci_dev)
+{
+    zsda_comp_dev_destroy(zsda_pci_dev);
+
+    return zsda_pci_device_release(pci_dev);
+}
+
+int
+zsda_get_queue_cfg_by_id(struct zsda_pci_device *zsda_pci_dev, uint8_t qid,
+             struct qinfo *qcfg)
+{
+    struct zsda_admin_req_qcfg req = {0};
+    struct zsda_admin_resp_qcfg resp = {0};
+    int ret = 0;
+    struct rte_pci_device *pci_dev =
+        zsda_devs[zsda_pci_dev->zsda_dev_id].pci_dev;
+
+    if (qid >= MAX_QPS_ON_FUNCTION) {
+        ZSDA_LOG(ERR, "qid beyound limit!");
+        return ZSDA_FAILED;
+    }
+
+    zsda_admin_msg_init(pci_dev);
+    req.msg_type = ZSDA_ADMIN_QUEUE_CFG_REQ;
+    req.qid = qid;
+
+    ret = zsda_send_admin_msg(pci_dev, &req, sizeof(req));
+    if (ret) {
+        ZSDA_LOG(ERR, "Failed! Send msg");
+        return ret;
+    }
+
+    ret = zsda_recv_admin_msg(pci_dev, &resp, sizeof(resp));
+    if (ret) {
+        ZSDA_LOG(ERR, "Failed! Receive msg");
+        return ret;
+    }
+
+    memcpy(qcfg, &resp.qcfg, sizeof(*qcfg));
+
+    return ZSDA_SUCCESS;
+}
+
+static int
+zsda_unmask_flr(const struct zsda_pci_device *zsda_pci_dev)
+{
+    struct zsda_admin_req_qcfg req = {0};
+    struct zsda_admin_resp_qcfg resp = {0};
+
+    int ret = 0;
+    struct rte_pci_device *pci_dev =
+        zsda_devs[zsda_pci_dev->zsda_dev_id].pci_dev;
+
+    zsda_admin_msg_init(pci_dev);
+
+    req.msg_type = ZSDA_FLR_SET_FUNCTION;
+
+    ret = zsda_send_admin_msg(pci_dev, &req, sizeof(req));
+    if (ret) {
+        ZSDA_LOG(ERR, "Failed! Send msg");
+        return ret;
+    }
+
+    ret = zsda_recv_admin_msg(pci_dev, &resp, sizeof(resp));
+    if (ret) {
+        ZSDA_LOG(ERR, "Failed! Receive msg");
+        return ret;
+    }
+
+    return ZSDA_SUCCESS;
+}
+
+static int
+zsda_pci_probe(struct rte_pci_driver *pci_drv __rte_unused,
+           struct rte_pci_device *pci_dev)
+{
+    int ret;
+    struct zsda_pci_device *zsda_pci_dev;
+
+    zsda_pci_dev = zsda_pci_device_allocate(pci_dev);
+    if (zsda_pci_dev == NULL) {
+        ZSDA_LOG(ERR, E_NULL);
+        return -ENODEV;
+    }
+
+    zsda_num_used_qps = zsda_get_num_used_qps(zsda_pci_dev->pci_dev);
+    zsda_num_used_qps++;
+
+    ret = zsda_admin_q_start(zsda_pci_dev->pci_dev);
+    if (ret) {
+        ZSDA_LOG(ERR, "Failed! admin q start");
+        return ret;
+    }
+
+    ret = zsda_queue_clear(zsda_pci_dev->pci_dev);
+    if (ret) {
+        ZSDA_LOG(ERR, "Failed! used zsda_io q clear");
+        return ret;
+    }
+
+    ret = zsda_unmask_flr(zsda_pci_dev);
+    if (ret) {
+        ZSDA_LOG(ERR, "Failed! flr close");
+        return ret;
+    }
+
+    ret = zsda_get_queue_cfg(zsda_pci_dev);
+    if (ret) {
+        ZSDA_LOG(ERR, "Failed! zsda_get_queue_cfg");
+        return ret;
+    }
+
+    ret |= zsda_comp_dev_create(zsda_pci_dev);
+
+    if (ret) {
+        ZSDA_LOG(ERR, "Failed! dev create");
+        zsda_pci_dev_destroy(zsda_pci_dev, pci_dev);
+        return ret;
+    }
+
+    return ret;
+}
+
+static int
+zsda_pci_remove(struct rte_pci_device *pci_dev)
+{
+    struct zsda_pci_device *zsda_pci_dev;
+
+    if (pci_dev == NULL)
+        return -EINVAL;
+
+    zsda_pci_dev = zsda_get_zsda_dev_from_pci_dev(pci_dev);
+    if (zsda_pci_dev == NULL)
+        return 0;
+
+    if (zsda_admin_q_clear(zsda_pci_dev->pci_dev) == ZSDA_FAILED)
+        ZSDA_LOG(ERR, "Failed! q clear");
+
+    if (zsda_admin_q_stop(zsda_pci_dev->pci_dev) == ZSDA_FAILED)
+        ZSDA_LOG(ERR, "Failed! q stop");
+
+    return zsda_pci_dev_destroy(zsda_pci_dev, pci_dev);
+}
+
+static struct rte_pci_driver rte_zsda_pmd = {
+    .id_table = pci_id_zsda_map,
+    .drv_flags = RTE_PCI_DRV_NEED_MAPPING,
+    .probe = zsda_pci_probe,
+    .remove = zsda_pci_remove };
+
+RTE_PMD_REGISTER_PCI(ZSDA_PCI_NAME, rte_zsda_pmd);
+RTE_PMD_REGISTER_PCI_TABLE(ZSDA_PCI_NAME, pci_id_zsda_map);
+RTE_PMD_REGISTER_KMOD_DEP(ZSDA_PCI_NAME,
+              "* igb_uio | uio_pci_generic | vfio-pci");
diff --git a/drivers/common/zsda/zsda_device.h b/drivers/common/zsda/zsda_device.h
new file mode 100644
index 0000000000..3da701d0a1
--- /dev/null
+++ b/drivers/common/zsda/zsda_device.h
@@ -0,0 +1,106 @@
+/* SPDX-License-Identifier: BSD-3-Clause
+ * Copyright(c) 2024 ZTE Corporation
+ */
+
+#ifndef _ZSDA_DEVICE_H_
+#define _ZSDA_DEVICE_H_
+
+#include "bus_pci_driver.h"
+
+#include <rte_bus_pci.h>
+
+#include "zsda_common.h"
+#include "zsda_logs.h"
+
+struct zsda_device_info {
+    const struct rte_memzone *mz;
+    /**< mz to store theļ¼š  struct zsda_pci_device ,    so it can be
+     * shared across processes
+     */
+
+    struct rte_pci_device *pci_dev;
+
+    struct rte_device comp_rte_dev;
+    /**< This represents the compression subset of this pci device.
+     * Register with this rather than with the one in
+     * pci_dev so that its driver can have a compression-specific name
+     */
+};
+
+extern struct zsda_device_info zsda_devs[];
+
+struct zsda_comp_dev_private;
+
+struct zsda_qp_hw_data {
+    bool used;
+
+    uint8_t tx_ring_num;
+    uint8_t rx_ring_num;
+    uint16_t tx_msg_size;
+    uint16_t rx_msg_size;
+};
+
+struct zsda_qp_hw {
+    struct zsda_qp_hw_data data[MAX_QPS_ON_FUNCTION];
+};
+
+/*
+ * This struct holds all the data about a ZSDA pci device
+ * including data about all services it supports.
+ * It contains
+ *  - hw_data
+ *  - config data
+ *  - runtime data
+ * Note: as this data can be shared in a multi-process scenario,
+ * any pointers in it must also point to shared memory.
+ */
+struct zsda_pci_device {
+    /* Data used by all services */
+    char name[ZSDA_DEV_NAME_MAX_LEN];
+    /**< Name of zsda pci device */
+    uint8_t zsda_dev_id;
+    /**< Id of device instance for this zsda pci device */
+
+    rte_spinlock_t arb_csr_lock;
+    /**< lock to protect accesses to the arbiter CSR */
+
+    struct rte_pci_device *pci_dev;
+
+    /* Data relating to compression service */
+    struct zsda_comp_dev_private *comp_dev;
+    /**< link back to compressdev private data */
+
+    struct zsda_qp_hw zsda_hw_qps[ZSDA_MAX_SERVICES];
+    uint16_t zsda_qp_hw_num[ZSDA_MAX_SERVICES];
+};
+
+struct zsda_pci_device *
+zsda_pci_device_allocate(struct rte_pci_device *pci_dev);
+
+struct zsda_pci_device *
+zsda_get_zsda_dev_from_pci_dev(struct rte_pci_device *pci_dev);
+
+__rte_weak int
+zsda_get_queue_cfg(struct zsda_pci_device *zsda_pci_dev);
+
+__rte_weak int
+zsda_comp_dev_create(struct zsda_pci_device *zsda_pci_dev);
+
+__rte_weak int
+zsda_comp_dev_destroy(struct zsda_pci_device *zsda_pci_dev);
+
+int
+zsda_get_queue_cfg_by_id(struct zsda_pci_device *zsda_pci_dev, uint8_t qid,
+             struct qinfo *qcfg);
+
+int zsda_queue_start(const struct rte_pci_device *pci_dev);
+int zsda_queue_stop(const struct rte_pci_device *pci_dev);
+int zsda_queue_clear(const struct rte_pci_device *pci_dev);
+
+int zsda_admin_q_start(const struct rte_pci_device *pci_dev);
+int zsda_admin_q_stop(const struct rte_pci_device *pci_dev);
+int zsda_admin_q_clear(const struct rte_pci_device *pci_dev);
+
+int zsda_set_cycle_head_tail(struct zsda_pci_device *zsda_pci_dev);
+
+#endif /* _ZSDA_DEVICE_H_ */
-- 
2.27.0