Commit 4f9327fa by Patrick Steinhardt

treebuilder: fix memory leaks in `write_with_buffer`

While we detect errors in `git_treebuilder_write_with_buffer`, we just
exit directly instead of freeing allocated memory. Fix this by
remembering error codes and skipping forward to the function's cleanup
code.
parent a25df009
...@@ -818,7 +818,7 @@ int git_treebuilder_write_with_buffer(git_oid *oid, git_treebuilder *bld, git_bu ...@@ -818,7 +818,7 @@ int git_treebuilder_write_with_buffer(git_oid *oid, git_treebuilder *bld, git_bu
size_t i, entrycount; size_t i, entrycount;
git_odb *odb; git_odb *odb;
git_tree_entry *entry; git_tree_entry *entry;
git_vector entries; git_vector entries = GIT_VECTOR_INIT;
assert(bld); assert(bld);
assert(tree); assert(tree);
...@@ -826,16 +826,16 @@ int git_treebuilder_write_with_buffer(git_oid *oid, git_treebuilder *bld, git_bu ...@@ -826,16 +826,16 @@ int git_treebuilder_write_with_buffer(git_oid *oid, git_treebuilder *bld, git_bu
git_buf_clear(tree); git_buf_clear(tree);
entrycount = git_strmap_num_entries(bld->map); entrycount = git_strmap_num_entries(bld->map);
if (git_vector_init(&entries, entrycount, entry_sort_cmp) < 0) if ((error = git_vector_init(&entries, entrycount, entry_sort_cmp)) < 0)
return -1; goto out;
if (tree->asize == 0 && if (tree->asize == 0 &&
(error = git_buf_grow(tree, entrycount * 72)) < 0) (error = git_buf_grow(tree, entrycount * 72)) < 0)
return error; goto out;
git_strmap_foreach_value(bld->map, entry, { git_strmap_foreach_value(bld->map, entry, {
if (git_vector_insert(&entries, entry) < 0) if ((error = git_vector_insert(&entries, entry)) < 0)
return -1; goto out;
}); });
git_vector_sort(&entries); git_vector_sort(&entries);
...@@ -855,6 +855,7 @@ int git_treebuilder_write_with_buffer(git_oid *oid, git_treebuilder *bld, git_bu ...@@ -855,6 +855,7 @@ int git_treebuilder_write_with_buffer(git_oid *oid, git_treebuilder *bld, git_bu
!(error = git_repository_odb__weakptr(&odb, bld->repo))) !(error = git_repository_odb__weakptr(&odb, bld->repo)))
error = git_odb_write(oid, odb, tree->ptr, tree->size, GIT_OBJ_TREE); error = git_odb_write(oid, odb, tree->ptr, tree->size, GIT_OBJ_TREE);
out:
git_vector_free(&entries); git_vector_free(&entries);
return error; return error;
......
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