transport: af-mctp: Ensure malloc() succeeds in *_recv()
Avoid a potential NULL-pointer dereference - don't assume we're
executing under Linux.
Fixes: ba6971b9ae69 ("requester: Add af_mctp transport")
Signed-off-by: Andrew Jeffery <andrew@aj.id.au>
Change-Id: Ieb73ace2b981eb94b710000b25e8bed3a6e34e7e
diff --git a/src/transport/af-mctp.c b/src/transport/af-mctp.c
index 8c7046b..08d8322 100644
--- a/src/transport/af-mctp.c
+++ b/src/transport/af-mctp.c
@@ -101,6 +101,10 @@
}
msg = malloc(length);
+ if (!msg) {
+ return PLDM_REQUESTER_RECV_FAIL;
+ }
+
length = recv(af_mctp->socket, msg, length, MSG_TRUNC);
if (length < (ssize_t)sizeof(struct pldm_msg_hdr)) {
free(msg);