struct crypto_queue     queue;
        struct ahash_request    *req;
        bool                    is_async;
+       atmel_sha_fn_t          resume;
 
        struct atmel_sha_dma    dma_lch_in;
 
        return (start_async) ? ret : err;
 }
 
+static int atmel_sha_done(struct atmel_sha_dev *dd);
+
 static int atmel_sha_start(struct atmel_sha_dev *dd)
 {
        struct ahash_request *req = dd->req;
        if (err)
                goto err1;
 
+       dd->resume = atmel_sha_done;
        if (ctx->op == SHA_OP_UPDATE) {
                err = atmel_sha_update_req(dd);
                if (err != -EINPROGRESS && (ctx->flags & SHA_FLAGS_FINUP))
        atmel_sha_handle_queue(dd, NULL);
 }
 
-static void atmel_sha_done_task(unsigned long data)
+static int atmel_sha_done(struct atmel_sha_dev *dd)
 {
-       struct atmel_sha_dev *dd = (struct atmel_sha_dev *)data;
        int err = 0;
 
-       dd->is_async = true;
-
        if (SHA_FLAGS_CPU & dd->flags) {
                if (SHA_FLAGS_OUTPUT_READY & dd->flags) {
                        dd->flags &= ~SHA_FLAGS_OUTPUT_READY;
                                goto finish;
                }
        }
-       return;
+       return err;
 
 finish:
        /* finish curent request */
        atmel_sha_finish_req(dd->req, err);
+
+       return err;
+}
+
+static void atmel_sha_done_task(unsigned long data)
+{
+       struct atmel_sha_dev *dd = (struct atmel_sha_dev *)data;
+
+       dd->is_async = true;
+       (void)dd->resume(dd);
 }
 
 static irqreturn_t atmel_sha_irq(int irq, void *dev_id)