From 7d69ea177200bda62b403eb9ead6eee2ba5abe52 Mon Sep 17 00:00:00 2001 From: Tucker Evans Date: Mon, 6 Jul 2020 20:07:45 -0400 Subject: Add reset key function for maps Allows changing the pointer to key when they are equivalent (by cmp function), to avoid memory leaks that could happen if we assumed either pointer was to be freed or overwritten. --- collections/map/map.c | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) (limited to 'collections/map/map.c') diff --git a/collections/map/map.c b/collections/map/map.c index 83a446d..7de58e4 100644 --- a/collections/map/map.c +++ b/collections/map/map.c @@ -82,6 +82,29 @@ void *key, *val; return 0; } +void* map_reset_key(root, key) +map *root; +void *key; +{ + void *tmp; + int cmp; + + if (!root || !key) + return NULL; + + cmp = root->cmp(root->key, key); + + if (cmp < 0) + return map_reset_key(root->left, key); + + if (cmp > 0) + return map_reset_key(root->right, key); + + tmp = root->key; + root->key = key; + return tmp; +} + void map_clear(root) map *root; { -- cgit v1.1