X-Git-Url: http://v3vee.org/palacios/gitweb/gitweb.cgi?a=blobdiff_plain;f=linux_usr%2Fv3_user_keyed_stream_example.c;h=1886b7393dad456c8643d03ac269296ae74ec220;hb=bd229a86fee7c3cc33a81986ac46186ad67c869e;hp=1e126da7333617fbe7e0fa12d7370150c5dbd192;hpb=7b67d47d87463b237db5ad43941742c5a5d4b0bf;p=palacios.releases.git

diff --git a/linux_usr/v3_user_keyed_stream_example.c b/linux_usr/v3_user_keyed_stream_example.c
index 1e126da..1886b73 100644
--- a/linux_usr/v3_user_keyed_stream_example.c
+++ b/linux_usr/v3_user_keyed_stream_example.c
@@ -35,6 +35,11 @@ int do_work(struct palacios_user_keyed_stream_op *req,
 
     // now built a response
     *resp = malloc(sizeof(struct palacios_user_keyed_stream_op) + datasize);
+    if (!*resp) {
+        fprintf(stderr, "ERROR: could not allocate space for response\n");
+        return -1;
+    }
+
     (*resp)->len = sizeof(struct palacios_user_keyed_stream_op) + datasize;
     (*resp)->buf_len = datasize;
     (*resp)->type = req->type;