#include <linux/types.h>
 #include <asm/byteorder.h>
+#include <linux/crypto.h>
 #include <linux/socket.h>
 #include <linux/tcp.h>
 #include <net/tcp.h>
 
 struct tls_sw_context {
        struct crypto_aead *aead_send;
+       struct crypto_wait async_wait;
 
        /* Sending context */
        char aad_space[TLS_AAD_SPACE_SIZE];
 
        aead_request_set_ad(aead_req, TLS_AAD_SPACE_SIZE);
        aead_request_set_crypt(aead_req, ctx->sg_aead_in, ctx->sg_aead_out,
                               data_len, tls_ctx->iv);
-       rc = crypto_aead_encrypt(aead_req);
+
+       aead_request_set_callback(aead_req, CRYPTO_TFM_REQ_MAY_BACKLOG,
+                                 crypto_req_done, &ctx->async_wait);
+
+       rc = crypto_wait_req(crypto_aead_encrypt(aead_req), &ctx->async_wait);
 
        ctx->sg_encrypted_data[0].offset -= tls_ctx->prepend_size;
        ctx->sg_encrypted_data[0].length += tls_ctx->prepend_size;
                goto out;
        }
 
+       crypto_init_wait(&sw_ctx->async_wait);
+
        ctx->priv_ctx = (struct tls_offload_context *)sw_ctx;
 
        crypto_info = &ctx->crypto_send;