Fix potential realloc leaks

2 files changed, 12 insertions, 6 deletions

diff --git a/src/env.c b/src/env.c
index 7a9ae8fa..3d560c98 100644
--- a/src/env.c
+++ b/src/env.c
@@ -119,10 +119,13 @@ serd_env_add(SerdEnv* const        env,
       serd_node_free(&old_prefix_uri);
     }
   } else {
-    env->prefixes = (SerdPrefix*)realloc(
+    SerdPrefix* const new_prefixes = (SerdPrefix*)realloc(
       env->prefixes, (++env->n_prefixes) * sizeof(SerdPrefix));
-    env->prefixes[env->n_prefixes - 1].name = serd_node_copy(name);
-    env->prefixes[env->n_prefixes - 1].uri  = serd_node_copy(uri);
+    if (new_prefixes) {
+      env->prefixes                           = new_prefixes;
+      env->prefixes[env->n_prefixes - 1].name = serd_node_copy(name);
+      env->prefixes[env->n_prefixes - 1].uri  = serd_node_copy(uri);
+    }
   }
 }
 
diff --git a/src/writer.c b/src/writer.c
index 07edc7f4..2a6d7c31 100644
--- a/src/writer.c
+++ b/src/writer.c
@@ -1266,9 +1266,12 @@ size_t
 serd_chunk_sink(const void* buf, size_t len, void* stream)
 {
   SerdChunk* chunk = (SerdChunk*)stream;
-  chunk->buf       = (uint8_t*)realloc((uint8_t*)chunk->buf, chunk->len + len);
-  memcpy((uint8_t*)chunk->buf + chunk->len, buf, len);
-  chunk->len += len;
+  uint8_t* new_buf = (uint8_t*)realloc((uint8_t*)chunk->buf, chunk->len + len);
+  if (new_buf) {
+    memcpy(new_buf + chunk->len, buf, len);
+    chunk->buf = new_buf;
+    chunk->len += len;
+  }
   return len;
 }