return ret;
 }
 
-int mv_cesa_queue_req(struct crypto_async_request *req)
+int mv_cesa_queue_req(struct crypto_async_request *req,
+                     struct mv_cesa_req *creq)
 {
        int ret;
        int i;
 
 
 /**
  * struct mv_cesa_req - CESA request
- * @type:      request type
  * @engine:    engine associated with this request
+ * @chain:     list of tdma descriptors associated  with this request
  */
 struct mv_cesa_req {
-       enum mv_cesa_req_type type;
        struct mv_cesa_engine *engine;
-};
-
-/**
- * struct mv_cesa_tdma_req - CESA TDMA request
- * @base:      base information
- * @chain:     TDMA chain
- */
-struct mv_cesa_tdma_req {
-       struct mv_cesa_req base;
        struct mv_cesa_tdma_chain chain;
 };
 
 
 /**
  * struct mv_cesa_ablkcipher_std_req - cipher standard request
- * @base:      base information
  * @op:                operation context
  * @offset:    current operation offset
  * @size:      size of the crypto operation
  */
 struct mv_cesa_ablkcipher_std_req {
-       struct mv_cesa_req base;
        struct mv_cesa_op_ctx op;
        unsigned int offset;
        unsigned int size;
  * @dst_nents: number of entries in the dest sg list
  */
 struct mv_cesa_ablkcipher_req {
-       union {
-               struct mv_cesa_req base;
-               struct mv_cesa_tdma_req dma;
-               struct mv_cesa_ablkcipher_std_req std;
-       } req;
+       struct mv_cesa_req base;
+       struct mv_cesa_ablkcipher_std_req std;
        int src_nents;
        int dst_nents;
 };
 
 /**
  * struct mv_cesa_ahash_std_req - standard hash request
- * @base:      base information
  * @offset:    current operation offset
  */
 struct mv_cesa_ahash_std_req {
-       struct mv_cesa_req base;
        unsigned int offset;
 };
 
 /**
  * struct mv_cesa_ahash_dma_req - DMA hash request
- * @base:              base information
  * @padding:           padding buffer
  * @padding_dma:       DMA address of the padding buffer
  * @cache_dma:         DMA address of the cache buffer
  */
 struct mv_cesa_ahash_dma_req {
-       struct mv_cesa_tdma_req base;
        u8 *padding;
        dma_addr_t padding_dma;
        u8 *cache;
  * @state:             hash state
  */
 struct mv_cesa_ahash_req {
+       struct mv_cesa_req base;
        union {
-               struct mv_cesa_req base;
                struct mv_cesa_ahash_dma_req dma;
                struct mv_cesa_ahash_std_req std;
        } req;
 
 extern struct mv_cesa_dev *cesa_dev;
 
+static inline enum mv_cesa_req_type
+mv_cesa_req_get_type(struct mv_cesa_req *req)
+{
+       return req->chain.first ? CESA_DMA_REQ : CESA_STD_REQ;
+}
+
 static inline void mv_cesa_update_op_cfg(struct mv_cesa_op_ctx *op,
                                         u32 cfg, u32 mask)
 {
                CESA_SA_DESC_CFG_FIRST_FRAG;
 }
 
-int mv_cesa_queue_req(struct crypto_async_request *req);
+int mv_cesa_queue_req(struct crypto_async_request *req,
+                     struct mv_cesa_req *creq);
 
 /*
  * Helper function that indicates whether a crypto request needs to be
        return iter->op_len;
 }
 
-void mv_cesa_dma_step(struct mv_cesa_tdma_req *dreq);
+void mv_cesa_dma_step(struct mv_cesa_req *dreq);
 
-static inline int mv_cesa_dma_process(struct mv_cesa_tdma_req *dreq,
+static inline int mv_cesa_dma_process(struct mv_cesa_req *dreq,
                                      u32 status)
 {
        if (!(status & CESA_SA_INT_ACC0_IDMA_DONE))
        return 0;
 }
 
-void mv_cesa_dma_prepare(struct mv_cesa_tdma_req *dreq,
+void mv_cesa_dma_prepare(struct mv_cesa_req *dreq,
                         struct mv_cesa_engine *engine);
+void mv_cesa_dma_cleanup(struct mv_cesa_req *dreq);
 
-void mv_cesa_dma_cleanup(struct mv_cesa_tdma_req *dreq);
 
 static inline void
 mv_cesa_tdma_desc_iter_init(struct mv_cesa_tdma_chain *chain)
 
                dma_unmap_sg(cesa_dev->dev, req->src, creq->src_nents,
                             DMA_BIDIRECTIONAL);
        }
-       mv_cesa_dma_cleanup(&creq->req.dma);
+       mv_cesa_dma_cleanup(&creq->base);
 }
 
 static inline void mv_cesa_ablkcipher_cleanup(struct ablkcipher_request *req)
 {
        struct mv_cesa_ablkcipher_req *creq = ablkcipher_request_ctx(req);
 
-       if (creq->req.base.type == CESA_DMA_REQ)
+       if (mv_cesa_req_get_type(&creq->base) == CESA_DMA_REQ)
                mv_cesa_ablkcipher_dma_cleanup(req);
 }
 
 static void mv_cesa_ablkcipher_std_step(struct ablkcipher_request *req)
 {
        struct mv_cesa_ablkcipher_req *creq = ablkcipher_request_ctx(req);
-       struct mv_cesa_ablkcipher_std_req *sreq = &creq->req.std;
-       struct mv_cesa_engine *engine = sreq->base.engine;
+       struct mv_cesa_ablkcipher_std_req *sreq = &creq->std;
+       struct mv_cesa_engine *engine = creq->base.engine;
        size_t  len = min_t(size_t, req->nbytes - sreq->offset,
                            CESA_SA_SRAM_PAYLOAD_SIZE);
 
                                          u32 status)
 {
        struct mv_cesa_ablkcipher_req *creq = ablkcipher_request_ctx(req);
-       struct mv_cesa_ablkcipher_std_req *sreq = &creq->req.std;
-       struct mv_cesa_engine *engine = sreq->base.engine;
+       struct mv_cesa_ablkcipher_std_req *sreq = &creq->std;
+       struct mv_cesa_engine *engine = creq->base.engine;
        size_t len;
        unsigned int ivsize;
 
 {
        struct ablkcipher_request *ablkreq = ablkcipher_request_cast(req);
        struct mv_cesa_ablkcipher_req *creq = ablkcipher_request_ctx(ablkreq);
-       struct mv_cesa_tdma_req *dreq;
+       struct mv_cesa_req *basereq = &creq->base;
        unsigned int ivsize;
        int ret;
 
-       if (creq->req.base.type == CESA_STD_REQ)
+       if (mv_cesa_req_get_type(basereq) == CESA_STD_REQ)
                return mv_cesa_ablkcipher_std_process(ablkreq, status);
 
-       ret = mv_cesa_dma_process(&creq->req.dma, status);
+       ret = mv_cesa_dma_process(basereq, status);
        if (ret)
                return ret;
 
-       dreq = &creq->req.dma;
        ivsize = crypto_ablkcipher_ivsize(crypto_ablkcipher_reqtfm(ablkreq));
-       memcpy_fromio(ablkreq->info, dreq->chain.last->data, ivsize);
+       memcpy_fromio(ablkreq->info, basereq->chain.last->data, ivsize);
 
        return 0;
 }
        struct ablkcipher_request *ablkreq = ablkcipher_request_cast(req);
        struct mv_cesa_ablkcipher_req *creq = ablkcipher_request_ctx(ablkreq);
 
-       if (creq->req.base.type == CESA_DMA_REQ)
-               mv_cesa_dma_step(&creq->req.dma);
+       if (mv_cesa_req_get_type(&creq->base) == CESA_DMA_REQ)
+               mv_cesa_dma_step(&creq->base);
        else
                mv_cesa_ablkcipher_std_step(ablkreq);
 }
 mv_cesa_ablkcipher_dma_prepare(struct ablkcipher_request *req)
 {
        struct mv_cesa_ablkcipher_req *creq = ablkcipher_request_ctx(req);
-       struct mv_cesa_tdma_req *dreq = &creq->req.dma;
+       struct mv_cesa_req *basereq = &creq->base;
 
-       mv_cesa_dma_prepare(dreq, dreq->base.engine);
+       mv_cesa_dma_prepare(basereq, basereq->engine);
 }
 
 static inline void
 mv_cesa_ablkcipher_std_prepare(struct ablkcipher_request *req)
 {
        struct mv_cesa_ablkcipher_req *creq = ablkcipher_request_ctx(req);
-       struct mv_cesa_ablkcipher_std_req *sreq = &creq->req.std;
-       struct mv_cesa_engine *engine = sreq->base.engine;
+       struct mv_cesa_ablkcipher_std_req *sreq = &creq->std;
+       struct mv_cesa_engine *engine = creq->base.engine;
 
        sreq->size = 0;
        sreq->offset = 0;
 {
        struct ablkcipher_request *ablkreq = ablkcipher_request_cast(req);
        struct mv_cesa_ablkcipher_req *creq = ablkcipher_request_ctx(ablkreq);
-       creq->req.base.engine = engine;
+       creq->base.engine = engine;
 
-       if (creq->req.base.type == CESA_DMA_REQ)
+       if (mv_cesa_req_get_type(&creq->base) == CESA_DMA_REQ)
                mv_cesa_ablkcipher_dma_prepare(ablkreq);
        else
                mv_cesa_ablkcipher_std_prepare(ablkreq);
        struct mv_cesa_ablkcipher_req *creq = ablkcipher_request_ctx(req);
        gfp_t flags = (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
                      GFP_KERNEL : GFP_ATOMIC;
-       struct mv_cesa_tdma_req *dreq = &creq->req.dma;
+       struct mv_cesa_req *basereq = &creq->base;
        struct mv_cesa_ablkcipher_dma_iter iter;
        struct mv_cesa_tdma_chain chain;
        bool skip_ctx = false;
        int ret;
        unsigned int ivsize;
 
-       dreq->base.type = CESA_DMA_REQ;
-       dreq->chain.first = NULL;
-       dreq->chain.last = NULL;
+       basereq->chain.first = NULL;
+       basereq->chain.last = NULL;
 
        if (req->src != req->dst) {
                ret = dma_map_sg(cesa_dev->dev, req->src, creq->src_nents,
        if (ret)
                goto err_free_tdma;
 
-       dreq->chain = chain;
+       basereq->chain = chain;
 
        return 0;
 
 err_free_tdma:
-       mv_cesa_dma_cleanup(dreq);
+       mv_cesa_dma_cleanup(basereq);
        if (req->dst != req->src)
                dma_unmap_sg(cesa_dev->dev, req->dst, creq->dst_nents,
                             DMA_FROM_DEVICE);
                                const struct mv_cesa_op_ctx *op_templ)
 {
        struct mv_cesa_ablkcipher_req *creq = ablkcipher_request_ctx(req);
-       struct mv_cesa_ablkcipher_std_req *sreq = &creq->req.std;
+       struct mv_cesa_ablkcipher_std_req *sreq = &creq->std;
+       struct mv_cesa_req *basereq = &creq->base;
 
-       sreq->base.type = CESA_STD_REQ;
        sreq->op = *op_templ;
        sreq->skip_ctx = false;
+       basereq->chain.first = NULL;
+       basereq->chain.last = NULL;
 
        return 0;
 }
 static int mv_cesa_des_op(struct ablkcipher_request *req,
                          struct mv_cesa_op_ctx *tmpl)
 {
+       struct mv_cesa_ablkcipher_req *creq = ablkcipher_request_ctx(req);
        struct mv_cesa_des_ctx *ctx = crypto_tfm_ctx(req->base.tfm);
        int ret;
 
        if (ret)
                return ret;
 
-       ret = mv_cesa_queue_req(&req->base);
+       ret = mv_cesa_queue_req(&req->base, &creq->base);
        if (mv_cesa_req_needs_cleanup(&req->base, ret))
                mv_cesa_ablkcipher_cleanup(req);
 
 static int mv_cesa_des3_op(struct ablkcipher_request *req,
                           struct mv_cesa_op_ctx *tmpl)
 {
+       struct mv_cesa_ablkcipher_req *creq = ablkcipher_request_ctx(req);
        struct mv_cesa_des3_ctx *ctx = crypto_tfm_ctx(req->base.tfm);
        int ret;
 
        if (ret)
                return ret;
 
-       ret = mv_cesa_queue_req(&req->base);
+       ret = mv_cesa_queue_req(&req->base, &creq->base);
        if (mv_cesa_req_needs_cleanup(&req->base, ret))
                mv_cesa_ablkcipher_cleanup(req);
 
 static int mv_cesa_aes_op(struct ablkcipher_request *req,
                          struct mv_cesa_op_ctx *tmpl)
 {
+       struct mv_cesa_ablkcipher_req *creq = ablkcipher_request_ctx(req);
        struct mv_cesa_aes_ctx *ctx = crypto_tfm_ctx(req->base.tfm);
        int ret, i;
        u32 *key;
        if (ret)
                return ret;
 
-       ret = mv_cesa_queue_req(&req->base);
+       ret = mv_cesa_queue_req(&req->base, &creq->base);
        if (mv_cesa_req_needs_cleanup(&req->base, ret))
                mv_cesa_ablkcipher_cleanup(req);
 
 
 
        dma_unmap_sg(cesa_dev->dev, req->src, creq->src_nents, DMA_TO_DEVICE);
        mv_cesa_ahash_dma_free_cache(&creq->req.dma);
-       mv_cesa_dma_cleanup(&creq->req.dma.base);
+       mv_cesa_dma_cleanup(&creq->base);
 }
 
 static inline void mv_cesa_ahash_cleanup(struct ahash_request *req)
 {
        struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 
-       if (creq->req.base.type == CESA_DMA_REQ)
+       if (mv_cesa_req_get_type(&creq->base) == CESA_DMA_REQ)
                mv_cesa_ahash_dma_cleanup(req);
 }
 
 {
        struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
 
-       if (creq->req.base.type == CESA_DMA_REQ)
+       if (mv_cesa_req_get_type(&creq->base) == CESA_DMA_REQ)
                mv_cesa_ahash_dma_last_cleanup(req);
 }
 
 {
        struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
        struct mv_cesa_ahash_std_req *sreq = &creq->req.std;
-       struct mv_cesa_engine *engine = sreq->base.engine;
+       struct mv_cesa_engine *engine = creq->base.engine;
        struct mv_cesa_op_ctx *op;
        unsigned int new_cache_ptr = 0;
        u32 frag_mode;
 static inline void mv_cesa_ahash_dma_prepare(struct ahash_request *req)
 {
        struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
-       struct mv_cesa_tdma_req *dreq = &creq->req.dma.base;
+       struct mv_cesa_req *basereq = &creq->base;
 
-       mv_cesa_dma_prepare(dreq, dreq->base.engine);
+       mv_cesa_dma_prepare(basereq, basereq->engine);
 }
 
 static void mv_cesa_ahash_std_prepare(struct ahash_request *req)
 {
        struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
        struct mv_cesa_ahash_std_req *sreq = &creq->req.std;
-       struct mv_cesa_engine *engine = sreq->base.engine;
+       struct mv_cesa_engine *engine = creq->base.engine;
 
        sreq->offset = 0;
        mv_cesa_adjust_op(engine, &creq->op_tmpl);
        struct ahash_request *ahashreq = ahash_request_cast(req);
        struct mv_cesa_ahash_req *creq = ahash_request_ctx(ahashreq);
 
-       if (creq->req.base.type == CESA_DMA_REQ)
-               mv_cesa_dma_step(&creq->req.dma.base);
+       if (mv_cesa_req_get_type(&creq->base) == CESA_DMA_REQ)
+               mv_cesa_dma_step(&creq->base);
        else
                mv_cesa_ahash_std_step(ahashreq);
 }
 {
        struct ahash_request *ahashreq = ahash_request_cast(req);
        struct mv_cesa_ahash_req *creq = ahash_request_ctx(ahashreq);
-       struct mv_cesa_engine *engine = creq->req.base.engine;
+       struct mv_cesa_engine *engine = creq->base.engine;
        unsigned int digsize;
        int ret, i;
 
-       if (creq->req.base.type == CESA_DMA_REQ)
-               ret = mv_cesa_dma_process(&creq->req.dma.base, status);
+       if (mv_cesa_req_get_type(&creq->base) == CESA_DMA_REQ)
+               ret = mv_cesa_dma_process(&creq->base, status);
        else
                ret = mv_cesa_ahash_std_process(ahashreq, status);
 
        unsigned int digsize;
        int i;
 
-       creq->req.base.engine = engine;
+       creq->base.engine = engine;
 
-       if (creq->req.base.type == CESA_DMA_REQ)
+       if (mv_cesa_req_get_type(&creq->base) == CESA_DMA_REQ)
                mv_cesa_ahash_dma_prepare(ahashreq);
        else
                mv_cesa_ahash_std_prepare(ahashreq);
        struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
        gfp_t flags = (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
                      GFP_KERNEL : GFP_ATOMIC;
-       struct mv_cesa_ahash_dma_req *ahashdreq = &creq->req.dma;
-       struct mv_cesa_tdma_req *dreq = &ahashdreq->base;
+       struct mv_cesa_req *basereq = &creq->base;
        struct mv_cesa_ahash_dma_iter iter;
        struct mv_cesa_op_ctx *op = NULL;
        unsigned int frag_len;
        int ret;
 
-       dreq->chain.first = NULL;
-       dreq->chain.last = NULL;
+       basereq->chain.first = NULL;
+       basereq->chain.last = NULL;
 
        if (creq->src_nents) {
                ret = dma_map_sg(cesa_dev->dev, req->src, creq->src_nents,
                }
        }
 
-       mv_cesa_tdma_desc_iter_init(&dreq->chain);
+       mv_cesa_tdma_desc_iter_init(&basereq->chain);
        mv_cesa_ahash_req_iter_init(&iter, req);
 
        /*
         * Add the cache (left-over data from a previous block) first.
         * This will never overflow the SRAM size.
         */
-       ret = mv_cesa_ahash_dma_add_cache(&dreq->chain, &iter, creq, flags);
+       ret = mv_cesa_ahash_dma_add_cache(&basereq->chain, &iter, creq, flags);
        if (ret)
                goto err_free_tdma;
 
                 * data. We intentionally do not add the final op block.
                 */
                while (true) {
-                       ret = mv_cesa_dma_add_op_transfers(&dreq->chain,
+                       ret = mv_cesa_dma_add_op_transfers(&basereq->chain,
                                                           &iter.base,
                                                           &iter.src, flags);
                        if (ret)
                        if (!mv_cesa_ahash_req_iter_next_op(&iter))
                                break;
 
-                       op = mv_cesa_dma_add_frag(&dreq->chain, &creq->op_tmpl,
+                       op = mv_cesa_dma_add_frag(&basereq->chain, &creq->op_tmpl,
                                                  frag_len, flags);
                        if (IS_ERR(op)) {
                                ret = PTR_ERR(op);
         * operation, which depends whether this is the final request.
         */
        if (creq->last_req)
-               op = mv_cesa_ahash_dma_last_req(&dreq->chain, &iter, creq,
+               op = mv_cesa_ahash_dma_last_req(&basereq->chain, &iter, creq,
                                                frag_len, flags);
        else if (frag_len)
-               op = mv_cesa_dma_add_frag(&dreq->chain, &creq->op_tmpl,
+               op = mv_cesa_dma_add_frag(&basereq->chain, &creq->op_tmpl,
                                          frag_len, flags);
 
        if (IS_ERR(op)) {
 
        if (op) {
                /* Add dummy desc to wait for crypto operation end */
-               ret = mv_cesa_dma_add_dummy_end(&dreq->chain, flags);
+               ret = mv_cesa_dma_add_dummy_end(&basereq->chain, flags);
                if (ret)
                        goto err_free_tdma;
        }
        return 0;
 
 err_free_tdma:
-       mv_cesa_dma_cleanup(dreq);
+       mv_cesa_dma_cleanup(basereq);
        dma_unmap_sg(cesa_dev->dev, req->src, creq->src_nents, DMA_TO_DEVICE);
 
 err:
        struct mv_cesa_ahash_req *creq = ahash_request_ctx(req);
        int ret;
 
-       if (cesa_dev->caps->has_tdma)
-               creq->req.base.type = CESA_DMA_REQ;
-       else
-               creq->req.base.type = CESA_STD_REQ;
-
        creq->src_nents = sg_nents_for_len(req->src, req->nbytes);
        if (creq->src_nents < 0) {
                dev_err(cesa_dev->dev, "Invalid number of src SG");
        if (*cached)
                return 0;
 
-       if (creq->req.base.type == CESA_DMA_REQ)
+       if (cesa_dev->caps->has_tdma)
                ret = mv_cesa_ahash_dma_req_init(req);
 
        return ret;
        if (cached)
                return 0;
 
-       ret = mv_cesa_queue_req(&req->base);
+       ret = mv_cesa_queue_req(&req->base, &creq->base);
        if (mv_cesa_req_needs_cleanup(&req->base, ret))
                mv_cesa_ahash_cleanup(req);
 
        if (cached)
                return 0;
 
-       ret = mv_cesa_queue_req(&req->base);
+       ret = mv_cesa_queue_req(&req->base, &creq->base);
        if (mv_cesa_req_needs_cleanup(&req->base, ret))
                mv_cesa_ahash_cleanup(req);
 
        if (cached)
                return 0;
 
-       ret = mv_cesa_queue_req(&req->base);
+       ret = mv_cesa_queue_req(&req->base, &creq->base);
        if (mv_cesa_req_needs_cleanup(&req->base, ret))
                mv_cesa_ahash_cleanup(req);
 
 
        return true;
 }
 
-void mv_cesa_dma_step(struct mv_cesa_tdma_req *dreq)
+void mv_cesa_dma_step(struct mv_cesa_req *dreq)
 {
-       struct mv_cesa_engine *engine = dreq->base.engine;
+       struct mv_cesa_engine *engine = dreq->engine;
 
        writel_relaxed(0, engine->regs + CESA_SA_CFG);
 
        writel(CESA_SA_CMD_EN_CESA_SA_ACCL0, engine->regs + CESA_SA_CMD);
 }
 
-void mv_cesa_dma_cleanup(struct mv_cesa_tdma_req *dreq)
+void mv_cesa_dma_cleanup(struct mv_cesa_req *dreq)
 {
        struct mv_cesa_tdma_desc *tdma;
 
        dreq->chain.last = NULL;
 }
 
-void mv_cesa_dma_prepare(struct mv_cesa_tdma_req *dreq,
+void mv_cesa_dma_prepare(struct mv_cesa_req *dreq,
                         struct mv_cesa_engine *engine)
 {
        struct mv_cesa_tdma_desc *tdma;