Paranoid error checking in userspace utils

[palacios.git] / linux_usr / v3_user_host_dev_example.c

diff --git a/linux_usr/v3_user_host_dev_example.c b/linux_usr/v3_user_host_dev_example.c
index 833f3c5..4f8144e 100644 (file)
--- a/linux_usr/v3_user_host_dev_example.c
+++ b/linux_usr/v3_user_host_dev_example.c
@@ -31,6 +31,10 @@ int do_work(struct palacios_host_dev_host_request_response *req,
     //
     // now built a response
     *resp = malloc(sizeof(struct palacios_host_dev_host_request_response) + datasize);
+    if (!*resp) {
+        fprintf(stderr, "ERROR: could not allocate memory for response\n");
+        return -1;
+    }
     (*resp)->data_len = sizeof(struct palacios_host_dev_host_request_response) + datasize;
 
     //