Commit 0646634e by Russell Belfer

Update filter registry code

This updates the git filter registry to be a little cleaner and
plugs some memory leaks.
parent a3aa5f4d
...@@ -318,6 +318,11 @@ static void crlf_cleanup( ...@@ -318,6 +318,11 @@ static void crlf_cleanup(
git__free(payload); git__free(payload);
} }
static void crlf_shutdown(git_filter *self)
{
git__free(self);
}
git_filter *git_crlf_filter_new(void) git_filter *git_crlf_filter_new(void)
{ {
struct crlf_filter *f = git__calloc(1, sizeof(struct crlf_filter)); struct crlf_filter *f = git__calloc(1, sizeof(struct crlf_filter));
...@@ -325,10 +330,10 @@ git_filter *git_crlf_filter_new(void) ...@@ -325,10 +330,10 @@ git_filter *git_crlf_filter_new(void)
f->f.version = GIT_FILTER_VERSION; f->f.version = GIT_FILTER_VERSION;
f->f.attributes = "crlf eol text"; f->f.attributes = "crlf eol text";
f->f.initialize = NULL; f->f.initialize = NULL;
f->f.shutdown = NULL; f->f.shutdown = crlf_shutdown;
f->f.check = crlf_check; f->f.check = crlf_check;
f->f.apply = crlf_apply; f->f.apply = crlf_apply;
f->f.cleanup = crlf_cleanup; f->f.cleanup = crlf_cleanup;
return (git_filter *)f; return (git_filter *)f;
} }
...@@ -10,6 +10,7 @@ ...@@ -10,6 +10,7 @@
#include "hash.h" #include "hash.h"
#include "filter.h" #include "filter.h"
#include "repository.h" #include "repository.h"
#include "global.h"
#include "git2/sys/filter.h" #include "git2/sys/filter.h"
#include "git2/config.h" #include "git2/config.h"
#include "blob.h" #include "blob.h"
...@@ -52,10 +53,64 @@ static int filter_def_priority_cmp(const void *a, const void *b) ...@@ -52,10 +53,64 @@ static int filter_def_priority_cmp(const void *a, const void *b)
return (pa < pb) ? -1 : (pa > pb) ? 1 : 0; return (pa < pb) ? -1 : (pa > pb) ? 1 : 0;
} }
static git_vector git__filter_registry = { struct filter_registry {
0, filter_def_priority_cmp, NULL, 0, 0 git_vector filters;
}; };
static struct filter_registry *git__filter_registry = NULL;
static void filter_registry_shutdown(void)
{
struct filter_registry *reg = NULL;
size_t pos;
git_filter_def *fdef;
if ((reg = git__swap(git__filter_registry, NULL)) == NULL)
return;
git_vector_foreach(&reg->filters, pos, fdef) {
if (fdef->initialized && fdef->filter && fdef->filter->shutdown) {
fdef->filter->shutdown(fdef->filter);
fdef->initialized = false;
}
git__free(fdef->attrdata);
git__free(fdef);
}
git_vector_free(&reg->filters);
git__free(reg);
}
static int filter_registry_initialize(void)
{
int error = 0;
struct filter_registry *reg;
if (git__filter_registry)
return 0;
reg = git__calloc(1, sizeof(struct filter_registry));
GITERR_CHECK_ALLOC(reg);
if ((error = git_vector_init(
&reg->filters, 2, filter_def_priority_cmp)) < 0)
goto cleanup;
reg = git__compare_and_swap(&git__filter_registry, NULL, reg);
if (reg != NULL)
goto cleanup;
git__on_shutdown(filter_registry_shutdown);
return git_filter_register(GIT_FILTER_CRLF, git_crlf_filter_new(), 0);
cleanup:
git_vector_free(&reg->filters);
git__free(reg);
return error;
}
static int filter_def_scan_attrs( static int filter_def_scan_attrs(
git_buf *attrs, size_t *nattr, size_t *nmatch, const char *attr_str) git_buf *attrs, size_t *nattr, size_t *nmatch, const char *attr_str)
{ {
...@@ -122,6 +177,29 @@ static void filter_def_set_attrs(git_filter_def *fdef) ...@@ -122,6 +177,29 @@ static void filter_def_set_attrs(git_filter_def *fdef)
} }
} }
static int filter_def_name_key_check(const void *key, const void *fdef)
{
const char *name =
fdef ? ((const git_filter_def *)fdef)->filter_name : NULL;
return name ? -1 : git__strcmp(key, name);
}
static int filter_registry_find(size_t *pos, const char *name)
{
return git_vector_search2(
pos, &git__filter_registry->filters, filter_def_name_key_check, name);
}
static git_filter_def *filter_registry_lookup(size_t *pos, const char *name)
{
git_filter_def *fdef = NULL;
if (!filter_registry_find(pos, name))
fdef = git_vector_get(&git__filter_registry->filters, *pos);
return fdef;
}
int git_filter_register( int git_filter_register(
const char *name, git_filter *filter, int priority) const char *name, git_filter *filter, int priority)
{ {
...@@ -129,7 +207,10 @@ int git_filter_register( ...@@ -129,7 +207,10 @@ int git_filter_register(
size_t nattr = 0, nmatch = 0; size_t nattr = 0, nmatch = 0;
git_buf attrs = GIT_BUF_INIT; git_buf attrs = GIT_BUF_INIT;
if (git_filter_lookup(name) != NULL) { if (filter_registry_initialize() < 0)
return -1;
if (!filter_registry_find(NULL, name)) {
giterr_set( giterr_set(
GITERR_FILTER, "Attempt to reregister existing filter '%s'", name); GITERR_FILTER, "Attempt to reregister existing filter '%s'", name);
return -1; return -1;
...@@ -151,48 +232,13 @@ int git_filter_register( ...@@ -151,48 +232,13 @@ int git_filter_register(
filter_def_set_attrs(fdef); filter_def_set_attrs(fdef);
if (git_vector_insert(&git__filter_registry, fdef) < 0) { if (git_vector_insert(&git__filter_registry->filters, fdef) < 0) {
git__free(fdef->attrdata); git__free(fdef->attrdata);
git__free(fdef); git__free(fdef);
return -1; return -1;
} }
git_vector_sort(&git__filter_registry); git_vector_sort(&git__filter_registry->filters);
return 0;
}
static int filter_def_name_key_check(const void *key, const void *fdef)
{
const char *name =
fdef ? ((const git_filter_def *)fdef)->filter_name : NULL;
return name ? -1 : git__strcmp(key, name);
}
static git_filter_def *filter_find_by_name(size_t *pos, const char *name)
{
git_filter_def *fdef = NULL;
if (!git_vector_search2(
pos, &git__filter_registry, filter_def_name_key_check, name))
fdef = git_vector_get(&git__filter_registry, *pos);
return fdef;
}
static int filter_initialize(git_filter_def *fdef)
{
int error = 0;
if (!fdef->initialized &&
fdef->filter &&
fdef->filter->initialize &&
(error = fdef->filter->initialize(fdef->filter)) < 0)
{
git_filter_unregister(fdef->filter_name);
return error;
}
fdef->initialized = true;
return 0; return 0;
} }
...@@ -207,12 +253,12 @@ int git_filter_unregister(const char *name) ...@@ -207,12 +253,12 @@ int git_filter_unregister(const char *name)
return -1; return -1;
} }
if ((fdef = filter_find_by_name(&pos, name)) == NULL) { if ((fdef = filter_registry_lookup(&pos, name)) == NULL) {
giterr_set(GITERR_FILTER, "Cannot find filter '%s' to unregister", name); giterr_set(GITERR_FILTER, "Cannot find filter '%s' to unregister", name);
return GIT_ENOTFOUND; return GIT_ENOTFOUND;
} }
(void)git_vector_remove(&git__filter_registry, pos); (void)git_vector_remove(&git__filter_registry->filters, pos);
if (fdef->initialized && fdef->filter && fdef->filter->shutdown) { if (fdef->initialized && fdef->filter && fdef->filter->shutdown) {
fdef->filter->shutdown(fdef->filter); fdef->filter->shutdown(fdef->filter);
...@@ -225,12 +271,33 @@ int git_filter_unregister(const char *name) ...@@ -225,12 +271,33 @@ int git_filter_unregister(const char *name)
return 0; return 0;
} }
static int filter_initialize(git_filter_def *fdef)
{
int error = 0;
if (!fdef->initialized &&
fdef->filter &&
fdef->filter->initialize &&
(error = fdef->filter->initialize(fdef->filter)) < 0)
{
/* auto-unregister if initialize fails */
git_filter_unregister(fdef->filter_name);
return error;
}
fdef->initialized = true;
return 0;
}
git_filter *git_filter_lookup(const char *name) git_filter *git_filter_lookup(const char *name)
{ {
size_t pos; size_t pos;
git_filter_def *fdef = filter_find_by_name(&pos, name); git_filter_def *fdef;
if (filter_registry_initialize() < 0)
return NULL;
if (!fdef) if ((fdef = filter_registry_lookup(&pos, name)) == NULL)
return NULL; return NULL;
if (!fdef->initialized && filter_initialize(fdef) < 0) if (!fdef->initialized && filter_initialize(fdef) < 0)
...@@ -239,14 +306,6 @@ git_filter *git_filter_lookup(const char *name) ...@@ -239,14 +306,6 @@ git_filter *git_filter_lookup(const char *name)
return fdef->filter; return fdef->filter;
} }
static int filter_load_defaults(void)
{
if (!git_vector_length(&git__filter_registry))
return git_filter_register(GIT_FILTER_CRLF, git_crlf_filter_new(), 0);
return 0;
}
git_repository *git_filter_source_repo(const git_filter_source *src) git_repository *git_filter_source_repo(const git_filter_source *src)
{ {
return src->repo; return src->repo;
...@@ -345,14 +404,14 @@ int git_filter_list_load( ...@@ -345,14 +404,14 @@ int git_filter_list_load(
size_t idx; size_t idx;
git_filter_def *fdef; git_filter_def *fdef;
if (filter_load_defaults() < 0) if (filter_registry_initialize() < 0)
return -1; return -1;
src.repo = repo; src.repo = repo;
src.path = path; src.path = path;
src.mode = mode; src.mode = mode;
git_vector_foreach(&git__filter_registry, idx, fdef) { git_vector_foreach(&git__filter_registry->filters, idx, fdef) {
const char **values = NULL; const char **values = NULL;
void *payload = NULL; void *payload = NULL;
......
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