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 4328b3dac5b7f9b2dfa64e5325f63f5f6739bea6..2561414e40f8f866d7328c46f3b5a34327afa934 100644 (file)
--- 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;