From: Sean Anderson <sean.anderson@seco.com>
Date: Fri, 22 Apr 2022 18:50:23 +0000 (-0400)
Subject: serial: smh: Check return value of strndup
X-Git-Tag: baikal/mips/sdk5.9~3^2~62^2~10
X-Git-Url: https://git.baikalelectronics.ru/?a=commitdiff_plain;h=0a74ec876c2dbe4bc405dcd6426b96c87272222e;p=uboot.git

serial: smh: Check return value of strndup

strndup can fail. Check for it.

Fixes: 4855b39be ("serial: smh: Implement puts for DM")
Signed-off-by: Sean Anderson <sean.anderson@seco.com>
---

diff --git a/drivers/serial/serial_semihosting.c b/drivers/serial/serial_semihosting.c
index 4328b3dac5..2561414e40 100644
--- a/drivers/serial/serial_semihosting.c
+++ b/drivers/serial/serial_semihosting.c
@@ -54,6 +54,9 @@ static ssize_t smh_serial_puts(struct udevice *dev, const char *s, size_t len)
 		}
 
 		buf = strndup(s, len);
+		if (!buf)
+			return -ENOMEM;
+
 		smh_puts(buf);
 		free(buf);
 		return len;