From edd8f67873fdeb78f2d642dc65e185f978d9affb Mon Sep 17 00:00:00 2001 From: Olga Kornievskaia Date: Mon, 22 Aug 2011 11:49:42 -0400 Subject: [PATCH] [daemon] checking for memory allocate in downcall --- daemon/nfs41_daemon.c | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/daemon/nfs41_daemon.c b/daemon/nfs41_daemon.c index 2bf3f1e..5d3796e 100644 --- a/daemon/nfs41_daemon.c +++ b/daemon/nfs41_daemon.c @@ -136,6 +136,12 @@ write_downcall: inbuf_len = UPCALL_BUF_SIZE; inbuf = malloc(inbuf_len); + if (inbuf == NULL) { + upcall.status = GetLastError(); + upcall_cancel(&upcall); + eprintf("Failed to allocate memory for downcall buffer... Exiting\n"); + break; + } upcall_marshall(&upcall, inbuf, (uint32_t)inbuf_len, (uint32_t*)&outbuf_len); dprintf(2, "making a downcall: outbuf_len %ld\n\n", outbuf_len);