linux-user: check return value of malloc()

Signed-off-by: zhanghailiang <[email protected]>
Acked-by: Riku Voipio <[email protected]>
Signed-off-by: Riku Voipio <[email protected]>

diff --git a/linux-user/syscall.c b/linux-user/syscall.c
index ebdc70e4ca54fb0b8b75395d05513903328f66a4..7ff7c2125536bc0087ccf427942527847bc1ad92 100644 (file)
--- a/linux-user/syscall.c
+++ b/linux-user/syscall.c
@@ -2904,6 +2904,10 @@ static inline abi_long do_msgsnd(int msqid, abi_long msgp,
     if (!lock_user_struct(VERIFY_READ, target_mb, msgp, 0))
         return -TARGET_EFAULT;
     host_mb = malloc(msgsz+sizeof(long));
+    if (!host_mb) {
+        unlock_user_struct(target_mb, msgp, 0);
+        return -TARGET_ENOMEM;
+    }
     host_mb->mtype = (abi_long) tswapal(target_mb->mtype);
     memcpy(host_mb->mtext, target_mb->mtext, msgsz);
     ret = get_errno(msgsnd(msqid, host_mb, msgsz, msgflg));