diff --git a/drivers/staging/unisys/virthba/virthba.c b/drivers/staging/unisys/virthba/virthba.c
index c292293718f0b..277851fff443e 100644
--- a/drivers/staging/unisys/virthba/virthba.c
+++ b/drivers/staging/unisys/virthba/virthba.c
@@ -1439,12 +1439,15 @@ static ssize_t
 enable_ints_write(struct file *file, const char __user *buffer,
 		  size_t count, loff_t *ppos)
 {
-	char buf[count + 1];
+	char buf[4];
 	int i, new_value;
 	struct virthba_info *virthbainfo;
 	U64 *Features_addr;
 	U64 mask;
 
+	if (count >= ARRAY_SIZE(buf))
+		return -EINVAL;
+
 	buf[count] = '\0';
 	if (copy_from_user(buf, buffer, count)) {
 		LOGERR("copy_from_user failed. buf<<%.*s>> count<<%lu>>\n",