modules/stats: be more correct on ENOMEM

CZ-NIC · Dec 20, 2024 · 4238b8c · 4238b8c
1 parent efbb031
commit 4238b8c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/modules/stats/stats.c b/modules/stats/stats.c
@@ -624,6 +624,9 @@ int stats_init(struct kr_module *module)
 	/* Initialize ring buffer of recently visited upstreams */
 	array_init(data->upstreams.q);
 	if (array_reserve(data->upstreams.q, UPSTREAMS_COUNT) != 0) {
+		trie_free(data->trie);
+		lru_free(data->queries.frequent);
+		free(data);
 		return kr_error(ENOMEM);
 	}
 	data->upstreams.q.len = UPSTREAMS_COUNT; /* signify we use the entries */