Commit 881e3da0 by Vicent Martí

Merge pull request #538 from carlosmn/ref-noleak

Don't leak when deleting or removing refs
parents d0ec3fb8 20c50b9e
...@@ -213,7 +213,7 @@ void *git_hashtable_lookup(git_hashtable *self, const void *key) ...@@ -213,7 +213,7 @@ void *git_hashtable_lookup(git_hashtable *self, const void *key)
return NULL; return NULL;
} }
int git_hashtable_remove(git_hashtable *self, const void *key) int git_hashtable_remove2(git_hashtable *self, const void *key, void **old_value)
{ {
int hash_id; int hash_id;
git_hashtable_node *node; git_hashtable_node *node;
...@@ -223,6 +223,7 @@ int git_hashtable_remove(git_hashtable *self, const void *key) ...@@ -223,6 +223,7 @@ int git_hashtable_remove(git_hashtable *self, const void *key)
for (hash_id = 0; hash_id < GIT_HASHTABLE_HASHES; ++hash_id) { for (hash_id = 0; hash_id < GIT_HASHTABLE_HASHES; ++hash_id) {
node = node_with_hash(self, key, hash_id); node = node_with_hash(self, key, hash_id);
if (node->key && self->key_equal(key, node->key) == 0) { if (node->key && self->key_equal(key, node->key) == 0) {
*old_value = node->value;
node->key = NULL; node->key = NULL;
node->value = NULL; node->value = NULL;
self->key_count--; self->key_count--;
......
...@@ -42,7 +42,15 @@ git_hashtable *git_hashtable_alloc(size_t min_size, ...@@ -42,7 +42,15 @@ git_hashtable *git_hashtable_alloc(size_t min_size,
git_hash_ptr hash, git_hash_ptr hash,
git_hash_keyeq_ptr key_eq); git_hash_keyeq_ptr key_eq);
void *git_hashtable_lookup(git_hashtable *h, const void *key); void *git_hashtable_lookup(git_hashtable *h, const void *key);
int git_hashtable_remove(git_hashtable *table, const void *key); int git_hashtable_remove2(git_hashtable *table, const void *key, void **old_value);
GIT_INLINE(int) git_hashtable_remove(git_hashtable *table, const void *key)
{
void *_unused;
return git_hashtable_remove2(table, key, &_unused);
}
void git_hashtable_free(git_hashtable *h); void git_hashtable_free(git_hashtable *h);
void git_hashtable_clear(git_hashtable *h); void git_hashtable_clear(git_hashtable *h);
int git_hashtable_merge(git_hashtable *self, git_hashtable *other); int git_hashtable_merge(git_hashtable *self, git_hashtable *other);
......
...@@ -984,14 +984,16 @@ static int reference_delete(git_reference *ref) ...@@ -984,14 +984,16 @@ static int reference_delete(git_reference *ref)
* We need to reload the packfile, remove the reference from the * We need to reload the packfile, remove the reference from the
* packing list, and repack */ * packing list, and repack */
if (ref->flags & GIT_REF_PACKED) { if (ref->flags & GIT_REF_PACKED) {
struct packref *packref;
/* load the existing packfile */ /* load the existing packfile */
if ((error = packed_load(ref->owner)) < GIT_SUCCESS) if ((error = packed_load(ref->owner)) < GIT_SUCCESS)
return git__rethrow(error, "Failed to delete reference"); return git__rethrow(error, "Failed to delete reference");
if (git_hashtable_remove(ref->owner->references.packfile, if (git_hashtable_remove2(ref->owner->references.packfile,
ref->name) < GIT_SUCCESS) ref->name, (void **) &packref) < GIT_SUCCESS)
return git__throw(GIT_ENOTFOUND, "Reference not found"); return git__throw(GIT_ENOTFOUND, "Reference not found");
git__free (packref);
error = packed_write(ref->owner); error = packed_write(ref->owner);
/* If the reference is loose, we can just remove the reference /* If the reference is loose, we can just remove the reference
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment