return 0;
 }
 
+static int nitrox_aes_gcm_setauthsize(struct crypto_aead *aead,
+                                     unsigned int authsize)
+{
+       switch (authsize) {
+       case 4:
+       case 8:
+       case 12:
+       case 13:
+       case 14:
+       case 15:
+       case 16:
+               break;
+       default:
+               return -EINVAL;
+       }
+
+       return nitrox_aead_setauthsize(aead, authsize);
+}
+
 static int alloc_src_sglist(struct nitrox_kcrypt_request *nkreq,
                            struct scatterlist *src, char *iv, int ivsize,
                            int buflen)
        areq->base.complete(&areq->base, err);
 }
 
+static inline bool nitrox_aes_gcm_assoclen_supported(unsigned int assoclen)
+{
+       if (assoclen <= 512)
+               return true;
+
+       return false;
+}
+
 static int nitrox_aes_gcm_enc(struct aead_request *areq)
 {
        struct crypto_aead *aead = crypto_aead_reqtfm(areq);
        struct flexi_crypto_context *fctx = nctx->u.fctx;
        int ret;
 
+       if (!nitrox_aes_gcm_assoclen_supported(areq->assoclen))
+               return -EINVAL;
+
        memcpy(fctx->crypto.iv, areq->iv, GCM_AES_SALT_SIZE);
 
        rctx->cryptlen = areq->cryptlen;
        struct flexi_crypto_context *fctx = nctx->u.fctx;
        int ret;
 
+       if (!nitrox_aes_gcm_assoclen_supported(areq->assoclen))
+               return -EINVAL;
+
        memcpy(fctx->crypto.iv, areq->iv, GCM_AES_SALT_SIZE);
 
        rctx->cryptlen = areq->cryptlen - aead->authsize;
                .cra_driver_name = "n5_aes_gcm",
                .cra_priority = PRIO,
                .cra_flags = CRYPTO_ALG_ASYNC,
-               .cra_blocksize = AES_BLOCK_SIZE,
+               .cra_blocksize = 1,
                .cra_ctxsize = sizeof(struct nitrox_crypto_ctx),
                .cra_alignmask = 0,
                .cra_module = THIS_MODULE,
        },
        .setkey = nitrox_aes_gcm_setkey,
-       .setauthsize = nitrox_aead_setauthsize,
+       .setauthsize = nitrox_aes_gcm_setauthsize,
        .encrypt = nitrox_aes_gcm_enc,
        .decrypt = nitrox_aes_gcm_dec,
        .init = nitrox_aes_gcm_init,
                .cra_driver_name = "n5_rfc4106",
                .cra_priority = PRIO,
                .cra_flags = CRYPTO_ALG_ASYNC,
-               .cra_blocksize = AES_BLOCK_SIZE,
+               .cra_blocksize = 1,
                .cra_ctxsize = sizeof(struct nitrox_crypto_ctx),
                .cra_alignmask = 0,
                .cra_module = THIS_MODULE,