Skip to content

Commit

Permalink
Fix memory leaks and use after free
Browse files Browse the repository at this point in the history
  • Loading branch information
arrbee committed May 4, 2012
1 parent f917481 commit b709e95
Show file tree
Hide file tree
Showing 7 changed files with 37 additions and 19 deletions.
6 changes: 3 additions & 3 deletions include/git2/attr.h
Original file line number Diff line number Diff line change
Expand Up @@ -160,10 +160,10 @@ GIT_EXTERN(int) git_attr_get(
* array itself if you allocated it).
*/
GIT_EXTERN(int) git_attr_get_many(
git_repository *repo,
git_repository *repo,
uint32_t flags,
const char *path,
size_t num_attr,
size_t num_attr,
const char **names,
const char **values);

Expand All @@ -186,7 +186,7 @@ GIT_EXTERN(int) git_attr_get_many(
* @param payload Passed on as extra parameter to callback function.
*/
GIT_EXTERN(int) git_attr_foreach(
git_repository *repo,
git_repository *repo,
uint32_t flags,
const char *path,
int (*callback)(const char *name, const char *value, void *payload),
Expand Down
5 changes: 3 additions & 2 deletions src/attr.c
Original file line number Diff line number Diff line change
Expand Up @@ -334,8 +334,6 @@ int git_attr_cache__push_file(
}

/* if not in cache, load data, parse, and cache */
if (git_attr_file__new(&file, source, relfile, &cache->pool) < 0)
return -1;

if (source == GIT_ATTR_FILE_FROM_FILE)
error = load_attr_file(filename, &content);
Expand All @@ -354,6 +352,9 @@ int git_attr_cache__push_file(
if (blob)
content = git_blob_rawcontent(blob);

if ((error = git_attr_file__new(&file, source, relfile, &cache->pool)) < 0)
goto finish;

if (parse && (error = parse(repo, content, file)) < 0)
goto finish;

Expand Down
1 change: 1 addition & 0 deletions src/config_file.c
Original file line number Diff line number Diff line change
Expand Up @@ -265,6 +265,7 @@ static int config_set(git_config_file *cfg, const char *name, const char *value)
cvar_free(old_var);

if (config_write(b, key, NULL, value) < 0) {
git_strmap_delete(b->values, var->key);
cvar_free(var);
return -1;
}
Expand Down
12 changes: 8 additions & 4 deletions src/diff.c
Original file line number Diff line number Diff line change
Expand Up @@ -506,7 +506,7 @@ static int diff_from_iterators(
git_diff_list **diff_ptr)
{
const git_index_entry *oitem, *nitem;
char *ignore_prefix = NULL;
git_buf ignore_prefix = GIT_BUF_INIT;
git_diff_list *diff = git_diff_list_alloc(repo, opts);
if (!diff)
goto fail;
Expand Down Expand Up @@ -536,8 +536,8 @@ static int diff_from_iterators(
git_delta_t delta_type = GIT_DELTA_ADDED;

/* contained in ignored parent directory, so this can be skipped. */
if (ignore_prefix != NULL &&
git__prefixcmp(nitem->path, ignore_prefix) == 0)
if (git_buf_len(&ignore_prefix) &&
git__prefixcmp(nitem->path, git_buf_cstr(&ignore_prefix)) == 0)
{
if (git_iterator_advance(new_iter, &nitem) < 0)
goto fail;
Expand All @@ -555,7 +555,7 @@ static int diff_from_iterators(
(oitem && git__prefixcmp(oitem->path, nitem->path) == 0))
{
if (is_ignored)
ignore_prefix = nitem->path;
git_buf_sets(&ignore_prefix, nitem->path);

if (git_iterator_advance_into_directory(new_iter, &nitem) < 0)
goto fail;
Expand Down Expand Up @@ -589,12 +589,16 @@ static int diff_from_iterators(

git_iterator_free(old_iter);
git_iterator_free(new_iter);
git_buf_free(&ignore_prefix);

*diff_ptr = diff;
return 0;

fail:
git_iterator_free(old_iter);
git_iterator_free(new_iter);
git_buf_free(&ignore_prefix);

git_diff_list_free(diff);
*diff_ptr = NULL;
return -1;
Expand Down
2 changes: 1 addition & 1 deletion src/ignore.c
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
static int parse_ignore_file(
git_repository *repo, const char *buffer, git_attr_file *ignores)
{
int error;
int error = 0;
git_attr_fnmatch *match = NULL;
const char *scan = NULL;
char *context = NULL;
Expand Down
28 changes: 19 additions & 9 deletions src/strmap.h
Original file line number Diff line number Diff line change
Expand Up @@ -36,18 +36,28 @@ typedef khash_t(str) git_strmap;
#define git_strmap_set_value_at(h, idx, v) kh_val(h, idx) = v
#define git_strmap_delete_at(h, idx) kh_del(str, h, idx)

#define git_strmap_insert(h, key, val, err) do { \
khiter_t __pos = kh_put(str, h, key, &err); \
if (err >= 0) kh_val(h, __pos) = val; \
} while (0)

#define git_strmap_insert2(h, key, val, old, err) do { \
khiter_t __pos = kh_put(str, h, key, &err); \
if (err >= 0) { \
old = (err == 0) ? kh_val(h, __pos) : NULL; \
#define git_strmap_insert(h, key, val, rval) do { \
khiter_t __pos = kh_put(str, h, key, &rval); \
if (rval >= 0) { \
if (rval == 0) kh_key(h, __pos) = key; \
kh_val(h, __pos) = val; \
} } while (0)

#define git_strmap_insert2(h, key, val, oldv, rval) do { \
khiter_t __pos = kh_put(str, h, key, &rval); \
if (rval >= 0) { \
if (rval == 0) { \
oldv = kh_val(h, __pos); \
kh_key(h, __pos) = key; \
} else { oldv = NULL; } \
kh_val(h, __pos) = val; \
} } while (0)

#define git_strmap_delete(h, key) do { \
khiter_t __pos = git_strmap_lookup_index(h, key); \
if (git_strmap_valid_index(h, __pos)) \
git_strmap_delete_at(h, __pos); } while (0)

#define git_strmap_foreach kh_foreach
#define git_strmap_foreach_value kh_foreach_value

Expand Down
2 changes: 2 additions & 0 deletions tests-clar/diff/tree.c
Original file line number Diff line number Diff line change
Expand Up @@ -205,4 +205,6 @@ void test_diff_tree__bare(void)
cl_assert(exp.line_dels == 1);

git_diff_list_free(diff);
git_tree_free(a);
git_tree_free(b);
}

0 comments on commit b709e95

Please sign in to comment.