askrene: memleak: scan reserved htable

Signed-off-by: Lagrang3 <[email protected]>
ElementsProject · Aug 12, 2024 · 5073942 · 5073942
1 parent cf3375c
commit 5073942
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/plugins/askrene/askrene.c b/plugins/askrene/askrene.c
@@ -883,7 +883,9 @@ static const struct plugin_command commands[] = {
 
 static void askrene_markmem(struct plugin *plugin, struct htable *memtable)
 {
-	layer_memleak_mark(get_askrene(plugin), memtable);
+	struct askrene *askrene = get_askrene(plugin);
+	layer_memleak_mark(askrene, memtable);
+	reserve_memleak_mark(askrene, memtable);
 }
 
 static const char *init(struct plugin *plugin,

diff --git a/plugins/askrene/reserve.c b/plugins/askrene/reserve.c
@@ -2,6 +2,8 @@
 #include <assert.h>
 #include <ccan/htable/htable_type.h>
 #include <common/gossmap.h>
+#include <common/memleak.h>
+#include <plugins/askrene/askrene.h>
 #include <plugins/askrene/reserve.h>
 
 /* Hash table for reservations */
@@ -138,3 +140,8 @@ void reserves_clear_capacities(struct reserve_htable *reserved,
 			capacities[idx] = 0;
 	}
 }
+
+void reserve_memleak_mark(struct askrene *askrene, struct htable *memtable)
+{
+	memleak_scan_htable(memtable, &askrene->reserved->raw);
+}
diff --git a/plugins/askrene/reserve.h b/plugins/askrene/reserve.h
@@ -40,4 +40,7 @@ size_t reserves_remove(struct reserve_htable *reserved,
 void reserves_clear_capacities(struct reserve_htable *reserved,
 			       const struct gossmap *gossmap,
 			       fp16_t *capacities);
+
+/* Scan for memleaks */
+void reserve_memleak_mark(struct askrene *askrene, struct htable *memtable);
 #endif /* LIGHTNING_PLUGINS_ASKRENE_RESERVE_H */