From 30487c277ac5d4be5786733ca7b98adb4c810ae9 Mon Sep 17 00:00:00 2001
From: David Robillard <d@drobilla.net>
Date: Wed, 27 Oct 2021 14:15:31 -0400
Subject: Add custom allocator support

---
 src/sink.c | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

(limited to 'src/sink.c')

diff --git a/src/sink.c b/src/sink.c
index 3e22f11c..f9c46f09 100644
--- a/src/sink.c
+++ b/src/sink.c
@@ -16,6 +16,7 @@
 
 #include "sink.h"
 
+#include "memory.h"
 #include "statement.h"
 
 #include "serd/serd.h"
@@ -29,13 +30,16 @@ serd_sink_new(const SerdWorld* const world,
               SerdEventFunc          event_func,
               SerdFreeFunc           free_handle)
 {
-  (void)world;
+  assert(world);
 
-  SerdSink* sink = (SerdSink*)calloc(1, sizeof(SerdSink));
+  SerdSink* sink = (SerdSink*)serd_wcalloc(world, 1, sizeof(SerdSink));
 
-  sink->handle      = handle;
-  sink->on_event    = event_func;
-  sink->free_handle = free_handle;
+  if (sink) {
+    sink->world       = world;
+    sink->handle      = handle;
+    sink->on_event    = event_func;
+    sink->free_handle = free_handle;
+  }
 
   return sink;
 }
@@ -48,7 +52,7 @@ serd_sink_free(SerdSink* sink)
       sink->free_handle(sink->handle);
     }
 
-    free(sink);
+    serd_wfree(sink->world, sink);
   }
 }
 
-- 
cgit v1.2.1