ipseckey: Use plugin features and depend on RESOLVER

Also fixed a double-free of the resolver instance.
This commit is contained in:
Tobias Brunner
2013-06-11 11:18:18 +02:00
parent d895721489
commit 82d3f5122b
2 changed files with 53 additions and 28 deletions
@@ -49,7 +49,7 @@ struct ipseckey_cred_t {
* Create an ipseckey_cred instance which uses the given resolver
* to query the DNS for IPSECKEY resource records.
*
* @param res resolver to use
* @param res resolver to use (gets adopted)
* @return credential set
*/
ipseckey_cred_t *ipseckey_cred_create(resolver_t *res);
@@ -1,4 +1,5 @@
/*
* Copyright (C) 2013 Tobias Brunner
* Copyright (C) 2012 Reto Guadagnini
* Hochschule fuer Technik Rapperswil
*
@@ -31,11 +32,6 @@ struct private_ipseckey_plugin_t {
*/
ipseckey_plugin_t public;
/**
* DNS resolver instance
*/
resolver_t *res;
/**
* credential set
*/
@@ -53,15 +49,59 @@ METHOD(plugin_t, get_name, char*,
return "ipseckey";
}
/**
* Create resolver and register credential set
*/
static bool plugin_cb(private_ipseckey_plugin_t *this,
plugin_feature_t *feature, bool reg, void *cb_data)
{
if (reg)
{
resolver_t *res;
res = lib->resolver->create(lib->resolver);
if (!res)
{
DBG1(DBG_CFG, "failed to create a DNS resolver instance");
return FALSE;
}
if (this->enabled)
{
this->cred = ipseckey_cred_create(res);
lib->credmgr->add_set(lib->credmgr, &this->cred->set);
}
else
{
res->destroy(res);
}
}
else
{
if (this->enabled)
{
lib->credmgr->remove_set(lib->credmgr, &this->cred->set);
this->cred->destroy(this->cred);
}
}
return TRUE;
}
METHOD(plugin_t, get_features, int,
private_ipseckey_plugin_t *this, plugin_feature_t *features[])
{
static plugin_feature_t f[] = {
PLUGIN_CALLBACK((plugin_feature_callback_t)plugin_cb, NULL),
PLUGIN_PROVIDE(CUSTOM, "ipseckey"),
PLUGIN_DEPENDS(RESOLVER),
};
*features = f;
return countof(f);
}
METHOD(plugin_t, destroy, void,
private_ipseckey_plugin_t *this)
{
if (this->enabled)
{
lib->credmgr->remove_set(lib->credmgr, &this->cred->set);
}
DESTROY_IF(this->res);
DESTROY_IF(this->cred);
free(this);
}
@@ -76,28 +116,13 @@ plugin_t *ipseckey_plugin_create()
.public = {
.plugin = {
.get_name = _get_name,
.reload = (void*)return_false,
.get_features = _get_features,
.destroy = _destroy,
},
},
.res = lib->resolver->create(lib->resolver),
.enabled = lib->settings->get_bool(lib->settings,
"%s.plugins.ipseckey.enable", FALSE, charon->name),
);
if (!this->res)
{
DBG1(DBG_CFG, "failed to create a DNS resolver instance");
destroy(this);
return NULL;
}
if (this->enabled)
{
this->cred = ipseckey_cred_create(this->res);
lib->credmgr->add_set(lib->credmgr, &this->cred->set);
}
return &this->public.plugin;
}