c-hyper: handle NULL from hyper_buf_copy()

Closes #7143

1 files changed, 13 insertions, 4 deletions

diff --git a/lib/c-hyper.c b/lib/c-hyper.c
index be4618eba..0f307f3e8 100644
--- a/lib/c-hyper.c
+++ b/lib/c-hyper.c
@@ -532,8 +532,12 @@ static int uploadpostfields(void *userdata, hyper_context *ctx,
     *chunk = NULL; /* nothing more to deliver */
   else {
     /* send everything off in a single go */
-    *chunk = hyper_buf_copy(data->set.postfields,
-                            (size_t)data->req.p.http->postsize);
+    hyper_buf *copy = hyper_buf_copy(data->set.postfields,
+                                     (size_t)data->req.p.http->postsize);
+    if(copy)
+      *chunk = copy;
+    else
+      return HYPER_POLL_ERROR;
     data->req.upload_done = TRUE;
   }
   return HYPER_POLL_READY;
@@ -552,8 +556,13 @@ static int uploadstreamed(void *userdata, hyper_context *ctx,
   if(!fillcount)
     /* done! */
     *chunk = NULL;
-  else
-    *chunk = hyper_buf_copy((uint8_t *)data->state.ulbuf, fillcount);
+  else {
+    hyper_buf *copy = hyper_buf_copy((uint8_t *)data->state.ulbuf, fillcount);
+    if(copy)
+      *chunk = copy;
+    else
+      return HYPER_POLL_ERROR;
+  }
   return HYPER_POLL_READY;
 }