/* SPDX-License-Identifier: GPL-2.0 */
/* Copyright (c) 2016-2017 HiSilicon Limited. */

#ifndef _SEC_DRV_H_
#define _SEC_DRV_H_

#include <crypto/algapi.h>
#include <linux/kfifo.h>

#define SEC_MAX_SGE_NUM                 64
#define SEC_HW_RING_NUM                 3

#define SEC_CMD_RING                    0
#define SEC_OUTORDER_RING               1
#define SEC_DBG_RING                    2

/* A reasonable length to balance memory use against flexibility */
#define SEC_QUEUE_LEN                   512

#define SEC_MAX_SGE_NUM   64

struct sec_bd_info {
#define SEC_BD_W0_T_LEN_M                       GENMASK(4, 0)
#define SEC_BD_W0_T_LEN_S                       0

#define SEC_BD_W0_C_WIDTH_M                     GENMASK(6, 5)
#define SEC_BD_W0_C_WIDTH_S                     5
#define   SEC_C_WIDTH_AES_128BIT                0
#define   SEC_C_WIDTH_AES_8BIT          1
#define   SEC_C_WIDTH_AES_1BIT          2
#define   SEC_C_WIDTH_DES_64BIT         0
#define   SEC_C_WIDTH_DES_8BIT          1
#define   SEC_C_WIDTH_DES_1BIT          2

#define SEC_BD_W0_C_MODE_M                      GENMASK(9, 7)
#define SEC_BD_W0_C_MODE_S                      7
#define   SEC_C_MODE_ECB                        0
#define   SEC_C_MODE_CBC                        1
#define   SEC_C_MODE_CTR                        4
#define   SEC_C_MODE_CCM                        5
#define   SEC_C_MODE_GCM                        6
#define   SEC_C_MODE_XTS                        7

#define SEC_BD_W0_SEQ                           BIT(10)
#define SEC_BD_W0_DE                            BIT(11)
#define SEC_BD_W0_DAT_SKIP_M                    GENMASK(13, 12)
#define SEC_BD_W0_DAT_SKIP_S                    12
#define SEC_BD_W0_C_GRAN_SIZE_19_16_M           GENMASK(17, 14)
#define SEC_BD_W0_C_GRAN_SIZE_19_16_S           14

#define SEC_BD_W0_CIPHER_M                      GENMASK(19, 18)
#define SEC_BD_W0_CIPHER_S                      18
#define   SEC_CIPHER_NULL                       0
#define   SEC_CIPHER_ENCRYPT                    1
#define   SEC_CIPHER_DECRYPT                    2

#define SEC_BD_W0_AUTH_M                        GENMASK(21, 20)
#define SEC_BD_W0_AUTH_S                        20
#define   SEC_AUTH_NULL                         0
#define   SEC_AUTH_MAC                          1
#define   SEC_AUTH_VERIF                        2

#define SEC_BD_W0_AI_GEN                        BIT(22)
#define SEC_BD_W0_CI_GEN                        BIT(23)
#define SEC_BD_W0_NO_HPAD                       BIT(24)
#define SEC_BD_W0_HM_M                          GENMASK(26, 25)
#define SEC_BD_W0_HM_S                          25
#define SEC_BD_W0_ICV_OR_SKEY_EN_M              GENMASK(28, 27)
#define SEC_BD_W0_ICV_OR_SKEY_EN_S              27

/* Multi purpose field - gran size bits for send, flag for recv */
#define SEC_BD_W0_FLAG_M                        GENMASK(30, 29)
#define SEC_BD_W0_C_GRAN_SIZE_21_20_M           GENMASK(30, 29)
#define SEC_BD_W0_FLAG_S                        29
#define SEC_BD_W0_C_GRAN_SIZE_21_20_S           29

#define SEC_BD_W0_DONE                          BIT(31)
        u32 w0;

#define SEC_BD_W1_AUTH_GRAN_SIZE_M              GENMASK(21, 0)
#define SEC_BD_W1_AUTH_GRAN_SIZE_S              0
#define SEC_BD_W1_M_KEY_EN                      BIT(22)
#define SEC_BD_W1_BD_INVALID                    BIT(23)
#define SEC_BD_W1_ADDR_TYPE                     BIT(24)

#define SEC_BD_W1_A_ALG_M                       GENMASK(28, 25)
#define SEC_BD_W1_A_ALG_S                       25
#define   SEC_A_ALG_SHA1                        0
#define   SEC_A_ALG_SHA256                      1
#define   SEC_A_ALG_MD5                         2
#define   SEC_A_ALG_SHA224                      3
#define   SEC_A_ALG_HMAC_SHA1                   8
#define   SEC_A_ALG_HMAC_SHA224                 10
#define   SEC_A_ALG_HMAC_SHA256                 11
#define   SEC_A_ALG_HMAC_MD5                    12
#define   SEC_A_ALG_AES_XCBC                    13
#define   SEC_A_ALG_AES_CMAC                    14

#define SEC_BD_W1_C_ALG_M                       GENMASK(31, 29)
#define SEC_BD_W1_C_ALG_S                       29
#define   SEC_C_ALG_DES                         0
#define   SEC_C_ALG_3DES                        1
#define   SEC_C_ALG_AES                         2

        u32 w1;

#define SEC_BD_W2_C_GRAN_SIZE_15_0_M            GENMASK(15, 0)
#define SEC_BD_W2_C_GRAN_SIZE_15_0_S            0
#define SEC_BD_W2_GRAN_NUM_M                    GENMASK(31, 16)
#define SEC_BD_W2_GRAN_NUM_S                    16
        u32 w2;

#define SEC_BD_W3_AUTH_LEN_OFFSET_M             GENMASK(9, 0)
#define SEC_BD_W3_AUTH_LEN_OFFSET_S             0
#define SEC_BD_W3_CIPHER_LEN_OFFSET_M           GENMASK(19, 10)
#define SEC_BD_W3_CIPHER_LEN_OFFSET_S           10
#define SEC_BD_W3_MAC_LEN_M                     GENMASK(24, 20)
#define SEC_BD_W3_MAC_LEN_S                     20
#define SEC_BD_W3_A_KEY_LEN_M                   GENMASK(29, 25)
#define SEC_BD_W3_A_KEY_LEN_S                   25
#define SEC_BD_W3_C_KEY_LEN_M                   GENMASK(31, 30)
#define SEC_BD_W3_C_KEY_LEN_S                   30
#define   SEC_KEY_LEN_AES_128                   0
#define   SEC_KEY_LEN_AES_192                   1
#define   SEC_KEY_LEN_AES_256                   2
#define   SEC_KEY_LEN_DES                       1
#define   SEC_KEY_LEN_3DES_3_KEY                1
#define   SEC_KEY_LEN_3DES_2_KEY                3
        u32 w3;

        /* W4,5 */
        union {
                u32 authkey_addr_lo;
                u32 authiv_addr_lo;
        };
        union {
                u32 authkey_addr_hi;
                u32 authiv_addr_hi;
        };

        /* W6,7 */
        u32 cipher_key_addr_lo;
        u32 cipher_key_addr_hi;

        /* W8,9 */
        u32 cipher_iv_addr_lo;
        u32 cipher_iv_addr_hi;

        /* W10,11 */
        u32 data_addr_lo;
        u32 data_addr_hi;

        /* W12,13 */
        u32 mac_addr_lo;
        u32 mac_addr_hi;

        /* W14,15 */
        u32 cipher_destin_addr_lo;
        u32 cipher_destin_addr_hi;
};

enum sec_mem_region {
        SEC_COMMON = 0,
        SEC_SAA,
        SEC_NUM_ADDR_REGIONS
};

#define SEC_NAME_SIZE                           64
#define SEC_Q_NUM                               16


/**
 * struct sec_queue_ring_cmd - store information about a SEC HW cmd ring
 * @used: Local counter used to cheaply establish if the ring is empty.
 * @lock: Protect against simultaneous adjusting of the read and write pointers.
 * @vaddr: Virtual address for the ram pages used for the ring.
 * @paddr: Physical address of the dma mapped region of ram used for the ring.
 * @callback: Callback function called on a ring element completing.
 */
struct sec_queue_ring_cmd {
        atomic_t used;
        struct mutex lock;
        struct sec_bd_info *vaddr;
        dma_addr_t paddr;
        void (*callback)(struct sec_bd_info *resp, void *ctx);
};

struct sec_debug_bd_info;
struct sec_queue_ring_db {
        struct sec_debug_bd_info *vaddr;
        dma_addr_t paddr;
};

struct sec_out_bd_info;
struct sec_queue_ring_cq {
        struct sec_out_bd_info *vaddr;
        dma_addr_t paddr;
};

struct sec_dev_info;

enum sec_cipher_alg {
        SEC_C_DES_ECB_64,
        SEC_C_DES_CBC_64,

        SEC_C_3DES_ECB_192_3KEY,
        SEC_C_3DES_ECB_192_2KEY,

        SEC_C_3DES_CBC_192_3KEY,
        SEC_C_3DES_CBC_192_2KEY,

        SEC_C_AES_ECB_128,
        SEC_C_AES_ECB_192,
        SEC_C_AES_ECB_256,

        SEC_C_AES_CBC_128,
        SEC_C_AES_CBC_192,
        SEC_C_AES_CBC_256,

        SEC_C_AES_CTR_128,
        SEC_C_AES_CTR_192,
        SEC_C_AES_CTR_256,

        SEC_C_AES_XTS_128,
        SEC_C_AES_XTS_256,

        SEC_C_NULL,
};

/**
 * struct sec_alg_tfm_ctx - hardware specific tranformation context
 * @cipher_alg: Cipher algorithm enabled include encryption mode.
 * @key: Key storage if required.
 * @pkey: DMA address for the key storage.
 * @req_template: Request template to save time on setup.
 * @queue: The hardware queue associated with this tfm context.
 * @lock: Protect key and pkey to ensure they are consistent
 * @auth_buf: Current context buffer for auth operations.
 * @backlog: The backlog queue used for cases where our buffers aren't
 * large enough.
 */
struct sec_alg_tfm_ctx {
        enum sec_cipher_alg cipher_alg;
        u8 *key;
        dma_addr_t pkey;
        struct sec_bd_info req_template;
        struct sec_queue *queue;
        struct mutex lock;
        u8 *auth_buf;
        struct list_head backlog;
};

/**
 * struct sec_request - data associate with a single crypto request
 * @elements: List of subparts of this request (hardware size restriction)
 * @num_elements: The number of subparts (used as an optimization)
 * @lock: Protect elements of this structure against concurrent change.
 * @tfm_ctx: hardware specific context.
 * @len_in: length of in sgl from upper layers
 * @len_out: length of out sgl from upper layers
 * @dma_iv: initialization vector - phsyical address
 * @err: store used to track errors across subelements of this request.
 * @req_base: pointer to base element of associate crypto context.
 * This is needed to allow shared handling skcipher, ahash etc.
 * @cb: completion callback.
 * @backlog_head: list head to allow backlog maintenance.
 *
 * The hardware is limited in the maximum size of data that it can
 * process from a single BD.  Typically this is fairly large (32MB)
 * but still requires the complexity of splitting the incoming
 * skreq up into a number of elements complete with appropriate
 * iv chaining.
 */
struct sec_request {
        struct list_head elements;
        int num_elements;
        struct mutex lock;
        struct sec_alg_tfm_ctx *tfm_ctx;
        int len_in;
        int len_out;
        dma_addr_t dma_iv;
        int err;
        struct crypto_async_request *req_base;
        void (*cb)(struct sec_bd_info *resp, struct crypto_async_request *req);
        struct list_head backlog_head;
};

/**
 * struct sec_request_el - A subpart of a request.
 * @head: allow us to attach this to the list in the sec_request
 * @req: hardware block descriptor corresponding to this request subpart
 * @in: hardware sgl for input - virtual address
 * @dma_in: hardware sgl for input - physical address
 * @sgl_in: scatterlist for this request subpart
 * @out: hardware sgl for output - virtual address
 * @dma_out: hardware sgl for output - physical address
 * @sgl_out: scatterlist for this request subpart
 * @sec_req: The request which this subpart forms a part of
 * @el_length: Number of bytes in this subpart. Needed to locate
 * last ivsize chunk for iv chaining.
 */
struct sec_request_el {
        struct list_head head;
        struct sec_bd_info req;
        struct sec_hw_sgl *in;
        dma_addr_t dma_in;
        struct scatterlist *sgl_in;
        struct sec_hw_sgl *out;
        dma_addr_t dma_out;
        struct scatterlist *sgl_out;
        struct sec_request *sec_req;
        size_t el_length;
};

/**
 * struct sec_queue - All the information about a HW queue
 * @dev_info: The parent SEC device to which this queue belongs.
 * @task_irq: Completion interrupt for the queue.
 * @name: Human readable queue description also used as irq name.
 * @ring: The several HW rings associated with one queue.
 * @regs: The iomapped device registers
 * @queue_id: Index of the queue used for naming and resource selection.
 * @in_use: Flag to say if the queue is in use.
 * @expected: The next expected element to finish assuming we were in order.
 * @uprocessed: A bitmap to track which OoO elements are done but not handled.
 * @softqueue: A software queue used when chaining requirements prevent direct
 *   use of the hardware queues.
 * @havesoftqueue: A flag to say we have a queues - as we may need one for the
 *   current mode.
 * @queuelock: Protect the soft queue from concurrent changes to avoid some
 *   potential loss of data races.
 * @shadow: Pointers back to the shadow copy of the hardware ring element
 *   need because we can't store any context reference in the bd element.
 */
struct sec_queue {
        struct sec_dev_info *dev_info;
        int task_irq;
        char name[SEC_NAME_SIZE];
        struct sec_queue_ring_cmd ring_cmd;
        struct sec_queue_ring_cq ring_cq;
        struct sec_queue_ring_db ring_db;
        void __iomem *regs;
        u32 queue_id;
        bool in_use;
        int expected;

        DECLARE_BITMAP(unprocessed, SEC_QUEUE_LEN);
        DECLARE_KFIFO_PTR(softqueue, typeof(struct sec_request_el *));
        bool havesoftqueue;
        struct mutex queuelock;
        void *shadow[SEC_QUEUE_LEN];
};

/**
 * struct sec_hw_sge: Track each of the 64 element SEC HW SGL entries
 * @buf: The IOV dma address for this entry.
 * @len: Length of this IOV.
 * @pad: Reserved space.
 */
struct sec_hw_sge {
        dma_addr_t buf;
        unsigned int len;
        unsigned int pad;
};

/**
 * struct sec_hw_sgl: One hardware SGL entry.
 * @next_sgl: The next entry if we need to chain dma address. Null if last.
 * @entry_sum_in_chain: The full count of SGEs - only matters for first SGL.
 * @entry_sum_in_sgl: The number of SGEs in this SGL element.
 * @flag: Unused in skciphers.
 * @serial_num: Unsued in skciphers.
 * @cpuid: Currently unused.
 * @data_bytes_in_sgl: Count of bytes from all SGEs in this SGL.
 * @next: Virtual address used to stash the next sgl - useful in completion.
 * @reserved: A reserved field not currently used.
 * @sge_entries: The (up to) 64 Scatter Gather Entries, representing IOVs.
 * @node: Currently unused.
 */
struct sec_hw_sgl {
        dma_addr_t next_sgl;
        u16 entry_sum_in_chain;
        u16 entry_sum_in_sgl;
        u32 flag;
        u64 serial_num;
        u32 cpuid;
        u32 data_bytes_in_sgl;
        struct sec_hw_sgl *next;
        u64 reserved;
        struct sec_hw_sge  sge_entries[SEC_MAX_SGE_NUM];
        u8 node[16];
};

struct dma_pool;

/**
 * struct sec_dev_info: The full SEC unit comprising queues and processors.
 * @sec_id: Index used to track which SEC this is when more than one is present.
 * @num_saas: The number of backed processors enabled.
 * @regs: iomapped register regions shared by whole SEC unit.
 * @dev_lock: Protects concurrent queue allocation / freeing for the SEC.
 * @queues: The 16 queues that this SEC instance provides.
 * @dev: Device pointer.
 * @hw_sgl_pool: DMA pool used to mimise mapping for the scatter gather lists.
 */
struct sec_dev_info {
        int sec_id;
        int num_saas;
        void __iomem *regs[SEC_NUM_ADDR_REGIONS];
        struct mutex dev_lock;
        int queues_in_use;
        struct sec_queue queues[SEC_Q_NUM];
        struct device *dev;
        struct dma_pool *hw_sgl_pool;
};

int sec_queue_send(struct sec_queue *queue, struct sec_bd_info *msg, void *ctx);
bool sec_queue_can_enqueue(struct sec_queue *queue, int num);
int sec_queue_stop_release(struct sec_queue *queue);
struct sec_queue *sec_queue_alloc_start_safe(void);
bool sec_queue_empty(struct sec_queue *queue);

/* Algorithm specific elements from sec_algs.c */
void sec_alg_callback(struct sec_bd_info *resp, void *ctx);
int sec_algs_register(void);
void sec_algs_unregister(void);

#endif /* _SEC_DRV_H_ */