/* SPDX-License-Identifier: BSD-3-Clause */
/* Copyright(c) 2007-2022 Intel Corporation */
#ifndef _QAT_OCF_MEM_POOL_H_
#define _QAT_OCF_MEM_POOL_H_

/* System headers */
#include <sys/types.h>

/* QAT specific headers */
#include "cpa.h"
#include "cpa_cy_sym_dp.h"
#include "icp_qat_fw_la.h"

#define QAT_OCF_MAX_LEN (64 * 1024)
#define QAT_OCF_MAX_FLATS (32)
#define QAT_OCF_MAX_DIGEST SHA512_DIGEST_LENGTH
#define QAT_OCF_MAX_SYMREQ (256)
#define QAT_OCF_MEM_POOL_SIZE ((QAT_OCF_MAX_SYMREQ * 2 + 1) * 2)
#define QAT_OCF_MAXLEN 64 * 1024

/* Dedicated structure due to flexible arrays not allowed to be
 * allocated on stack */
struct qat_ocf_buffer_list {
        Cpa64U reserved0;
        Cpa32U numBuffers;
        Cpa32U reserved1;
        CpaPhysFlatBuffer flatBuffers[QAT_OCF_MAX_FLATS];
};

struct qat_ocf_dma_mem {
        bus_dma_tag_t dma_tag;
        bus_dmamap_t dma_map;
        bus_dma_segment_t dma_seg;
        void *dma_vaddr;
} __aligned(64);

struct qat_ocf_cookie {
        /* Source SGLs */
        struct qat_ocf_buffer_list src_buffers;
        /* Destination SGL */
        struct qat_ocf_buffer_list dst_buffers;

        /* Cache OP data */
        CpaCySymDpOpData pOpdata;

        /* IV max size taken from cryptdev */
        uint8_t qat_ocf_iv_buf[EALG_MAX_BLOCK_LEN];
        bus_addr_t qat_ocf_iv_buf_paddr;
        uint8_t qat_ocf_digest[QAT_OCF_MAX_DIGEST];
        bus_addr_t qat_ocf_digest_paddr;
        /* Used only in case of separated AAD and GCM, CCM and RC4 */
        uint8_t qat_ocf_gcm_aad[ICP_QAT_FW_CCM_GCM_AAD_SZ_MAX];
        bus_addr_t qat_ocf_gcm_aad_paddr;

        /* Source SGLs */
        struct qat_ocf_dma_mem src_dma_mem;
        bus_addr_t src_buffer_list_paddr;

        /* Destination SGL */
        struct qat_ocf_dma_mem dst_dma_mem;
        bus_addr_t dst_buffer_list_paddr;

        /* AAD - used only if separated AAD is used by OCF and HW requires
         * to have it at the beginning of source buffer */
        struct qat_ocf_dma_mem gcm_aad_dma_mem;
        bus_addr_t gcm_aad_buffer_list_paddr;
        CpaBoolean is_sep_aad_used;

        /* Cache OP data */
        bus_addr_t pOpData_paddr;
        /* misc */
        struct cryptop *crp_op;

        /* This cookie tag and map */
        bus_dma_tag_t dma_tag;
        bus_dmamap_t dma_map;
};

struct qat_ocf_session {
        CpaCySymSessionCtx sessionCtx;
        Cpa32U sessionCtxSize;
        Cpa32U authLen;
        Cpa32U aadLen;
};

struct qat_ocf_dsession {
        struct qat_ocf_instance *qatInstance;
        struct qat_ocf_session encSession;
        struct qat_ocf_session decSession;
};

struct qat_ocf_load_cb_arg {
        struct cryptop *crp_op;
        struct qat_ocf_cookie *qat_cookie;
        CpaCySymDpOpData *pOpData;
        int error;
};

struct qat_ocf_instance {
        CpaInstanceHandle cyInstHandle;
        struct mtx cyInstMtx;
        struct qat_ocf_dma_mem cookie_dmamem[QAT_OCF_MEM_POOL_SIZE];
        struct qat_ocf_cookie *cookie_pool[QAT_OCF_MEM_POOL_SIZE];
        struct qat_ocf_cookie *free_cookie[QAT_OCF_MEM_POOL_SIZE];
        int free_cookie_ptr;
        struct mtx cookie_pool_mtx;
        int32_t driver_id;
};

/* Init/deinit */
CpaStatus qat_ocf_cookie_pool_init(struct qat_ocf_instance *instance,
                                   device_t dev);
void qat_ocf_cookie_pool_deinit(struct qat_ocf_instance *instance);
/* Alloc/free */
CpaStatus qat_ocf_cookie_alloc(struct qat_ocf_instance *instance,
                               struct qat_ocf_cookie **buffers_out);
void qat_ocf_cookie_free(struct qat_ocf_instance *instance,
                         struct qat_ocf_cookie *cookie);
/* Pre/post sync */
CpaStatus qat_ocf_cookie_dma_pre_sync(struct cryptop *crp,
                                      CpaCySymDpOpData *pOpData);
CpaStatus qat_ocf_cookie_dma_post_sync(struct cryptop *crp,
                                       CpaCySymDpOpData *pOpData);
/* Bus DMA unload */
CpaStatus qat_ocf_cookie_dma_unload(struct cryptop *crp,
                                    CpaCySymDpOpData *pOpData);
/* Bus DMA load callbacks */
void qat_ocf_crypto_load_buf_cb(void *_arg,
                                bus_dma_segment_t *segs,
                                int nseg,
                                int error);
void qat_ocf_crypto_load_obuf_cb(void *_arg,
                                 bus_dma_segment_t *segs,
                                 int nseg,
                                 int error);
void qat_ocf_crypto_load_aadbuf_cb(void *_arg,
                                   bus_dma_segment_t *segs,
                                   int nseg,
                                   int error);

#endif /* _QAT_OCF_MEM_POOL_H_ */