[sheepdog] [PATCH v10 3/4] lib: correct a way of freeing sockfd_cache_entry
Hitoshi Mitake
mitake.hitoshi at lab.ntt.co.jp
Wed Jul 31 11:44:11 CEST 2013
Previous implementation of the sockfd caching mechanism has two memory
leaks. The leak is caused when sockfd_cache_destroy(),
sockfd_cache_add_nolock() try to free a sockfd cache entry. They
forget to free entry->fds.
This patch adds a new function, free_cache_entry() for safe freeing of
sockfd cache entries. And let the above two functions to use it.
Also this patch lets sockfd_cache_add() use free_cache_entry() for
consistency.
Signed-off-by: Hitoshi Mitake <mitake.hitoshi at lab.ntt.co.jp>
---
v10:
- let sockfd_cache_add() use free_cache_entry() for consistent
v9:
- remove an unnecessary check in free_cache_entry()
lib/sockfd_cache.c | 13 +++++++++----
1 file changed, 9 insertions(+), 4 deletions(-)
diff --git a/lib/sockfd_cache.c b/lib/sockfd_cache.c
index 64c44d8..f4efd1b 100644
--- a/lib/sockfd_cache.c
+++ b/lib/sockfd_cache.c
@@ -179,6 +179,12 @@ static inline void destroy_all_slots(struct sockfd_cache_entry *entry)
close(entry->fds[i].fd);
}
+static void free_cache_entry(struct sockfd_cache_entry *entry)
+{
+ free(entry->fds);
+ free(entry);
+}
+
/*
* Destroy all the Cached FDs of the node
*
@@ -206,7 +212,7 @@ static bool sockfd_cache_destroy(const struct node_id *nid)
pthread_rwlock_unlock(&sockfd_cache.lock);
destroy_all_slots(entry);
- free(entry);
+ free_cache_entry(entry);
return true;
false_out:
@@ -225,7 +231,7 @@ static void sockfd_cache_add_nolock(const struct node_id *nid)
memcpy(&new->nid, nid, sizeof(struct node_id));
if (sockfd_cache_insert(new)) {
- free(new);
+ free_cache_entry(new);
return;
}
sockfd_cache.count++;
@@ -260,8 +266,7 @@ void sockfd_cache_add(const struct node_id *nid)
memcpy(&new->nid, nid, sizeof(struct node_id));
if (sockfd_cache_insert(new)) {
- free(new->fds);
- free(new);
+ free_cache_entry(new);
pthread_rwlock_unlock(&sockfd_cache.lock);
return;
}
--
1.7.10.4
More information about the sheepdog
mailing list