From 23c9526f8689964b65b1344fab682f699a436f2c Mon Sep 17 00:00:00 2001
From: Lin Jinhan <troy.lin@rock-chips.com>
Date: Fri, 11 Mar 2022 15:04:20 +0800
Subject: [PATCH] crypto: rockchip: v2: akcipher: prevent input data overflow

Signed-off-by: Lin Jinhan <troy.lin@rock-chips.com>
Change-Id: I39442e8e8ad1897b454da451090fbe867d6b26d6
---
 drivers/crypto/rockchip/rk_crypto_v2_akcipher.c | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/drivers/crypto/rockchip/rk_crypto_v2_akcipher.c b/drivers/crypto/rockchip/rk_crypto_v2_akcipher.c
index 26acdcb2e975..7cbc04bbb0ae 100644
--- a/drivers/crypto/rockchip/rk_crypto_v2_akcipher.c
+++ b/drivers/crypto/rockchip/rk_crypto_v2_akcipher.c
@@ -149,6 +149,9 @@ static int rk_rsa_calc(struct akcipher_request *req, bool encypt)
 		return -EOVERFLOW;
 	}
 
+	if (req->src_len > key_byte_size)
+		return -EINVAL;
+
 	in = rk_bn_alloc(key_byte_size);
 	if (!in)
 		goto exit;