Add reload support to attr plugin

This commit is contained in:
Martin Willi
2011-04-15 10:07:13 +02:00
parent 00b4b4b0d1
commit 96409be320
3 changed files with 64 additions and 19 deletions
+8 -1
View File
@@ -42,6 +42,13 @@ METHOD(plugin_t, get_name, char*,
return "attr"; return "attr";
} }
METHOD(plugin_t, reload, bool,
private_attr_plugin_t *this)
{
this->provider->reload(this->provider);
return TRUE;
}
METHOD(plugin_t, destroy, void, METHOD(plugin_t, destroy, void,
private_attr_plugin_t *this) private_attr_plugin_t *this)
{ {
@@ -61,7 +68,7 @@ plugin_t *attr_plugin_create()
.public = { .public = {
.plugin = { .plugin = {
.get_name = _get_name, .get_name = _get_name,
.reload = (void*)return_false, .reload = _reload,
.destroy = _destroy, .destroy = _destroy,
}, },
}, },
+51 -18
View File
@@ -21,6 +21,7 @@
#include <hydra.h> #include <hydra.h>
#include <debug.h> #include <debug.h>
#include <utils/linked_list.h> #include <utils/linked_list.h>
#include <threading/rwlock.h>
#define SERVER_MAX 2 #define SERVER_MAX 2
@@ -41,6 +42,11 @@ struct private_attr_provider_t {
* List of attributes, attribute_entry_t * List of attributes, attribute_entry_t
*/ */
linked_list_t *attributes; linked_list_t *attributes;
/**
* Lock for attribute list
*/
rwlock_t *lock;
}; };
struct attribute_entry_t { struct attribute_entry_t {
@@ -50,6 +56,15 @@ struct attribute_entry_t {
chunk_t value; chunk_t value;
}; };
/**
* Destroy an entry
*/
static void attribute_destroy(attribute_entry_t *this)
{
free(this->value.ptr);
free(this);
}
/** /**
* convert enumerator value from attribute_entry * convert enumerator value from attribute_entry
*/ */
@@ -67,9 +82,10 @@ METHOD(attribute_provider_t, create_attribute_enumerator, enumerator_t*,
{ {
if (vip) if (vip)
{ {
this->lock->read_lock(this->lock);
return enumerator_create_filter( return enumerator_create_filter(
this->attributes->create_enumerator(this->attributes), this->attributes->create_enumerator(this->attributes),
(void*)attr_enum_filter, NULL, NULL); (void*)attr_enum_filter, this->lock, (void*)this->lock->unlock);
} }
return enumerator_create_empty(); return enumerator_create_empty();
} }
@@ -77,15 +93,9 @@ METHOD(attribute_provider_t, create_attribute_enumerator, enumerator_t*,
METHOD(attr_provider_t, destroy, void, METHOD(attr_provider_t, destroy, void,
private_attr_provider_t *this) private_attr_provider_t *this)
{ {
attribute_entry_t *entry; this->attributes->destroy_function(this->attributes,
(void*)attribute_destroy);
while (this->attributes->remove_last(this->attributes, this->lock->destroy(this->lock);
(void**)&entry) == SUCCESS)
{
free(entry->value.ptr);
free(entry);
}
this->attributes->destroy(this->attributes);
free(this); free(this);
} }
@@ -125,6 +135,8 @@ static void add_legacy_entry(private_attr_provider_t *this, char *key, int nr,
entry->type = type; entry->type = type;
entry->value = chunk_clone(host->get_address(host)); entry->value = chunk_clone(host->get_address(host));
host->destroy(host); host->destroy(host);
DBG2(DBG_CFG, "loaded legacy entry attribute %N: %#B",
configuration_attribute_type_names, entry->type, &entry->value);
this->attributes->insert_last(this->attributes, entry); this->attributes->insert_last(this->attributes, entry);
} }
} }
@@ -154,6 +166,13 @@ static void load_entries(private_attr_provider_t *this)
{ {
enumerator_t *enumerator, *tokens; enumerator_t *enumerator, *tokens;
char *key, *value, *token; char *key, *value, *token;
int i;
for (i = 1; i <= SERVER_MAX; i++)
{
add_legacy_entry(this, "dns", i, INTERNAL_IP4_DNS);
add_legacy_entry(this, "nbns", i, INTERNAL_IP4_NBNS);
}
enumerator = lib->settings->create_key_value_enumerator(lib->settings, enumerator = lib->settings->create_key_value_enumerator(lib->settings,
"%s.plugins.attr", hydra->daemon); "%s.plugins.attr", hydra->daemon);
@@ -227,6 +246,8 @@ static void load_entries(private_attr_provider_t *this)
} }
} }
host->destroy(host); host->destroy(host);
DBG2(DBG_CFG, "loaded attribute %N: %#B",
configuration_attribute_type_names, entry->type, &entry->value);
this->attributes->insert_last(this->attributes, entry); this->attributes->insert_last(this->attributes, entry);
} }
tokens->destroy(tokens); tokens->destroy(tokens);
@@ -234,13 +255,29 @@ static void load_entries(private_attr_provider_t *this)
enumerator->destroy(enumerator); enumerator->destroy(enumerator);
} }
METHOD(attr_provider_t, reload, void,
private_attr_provider_t *this)
{
this->lock->write_lock(this->lock);
this->attributes->destroy_function(this->attributes, (void*)attribute_destroy);
this->attributes = linked_list_create();
load_entries(this);
DBG1(DBG_CFG, "loaded %d entr%s for attr plugin configuration",
this->attributes->get_count(this->attributes),
this->attributes->get_count(this->attributes) == 1 ? "y" : "ies");
this->lock->unlock(this->lock);
}
/* /*
* see header file * see header file
*/ */
attr_provider_t *attr_provider_create(database_t *db) attr_provider_t *attr_provider_create(database_t *db)
{ {
private_attr_provider_t *this; private_attr_provider_t *this;
int i;
INIT(this, INIT(this,
.public = { .public = {
@@ -249,17 +286,13 @@ attr_provider_t *attr_provider_create(database_t *db)
.release_address = (void*)return_false, .release_address = (void*)return_false,
.create_attribute_enumerator = _create_attribute_enumerator, .create_attribute_enumerator = _create_attribute_enumerator,
}, },
.reload = _reload,
.destroy = _destroy, .destroy = _destroy,
}, },
.attributes = linked_list_create(), .attributes = linked_list_create(),
.lock = rwlock_create(RWLOCK_TYPE_DEFAULT),
); );
for (i = 1; i <= SERVER_MAX; i++)
{
add_legacy_entry(this, "dns", i, INTERNAL_IP4_DNS);
add_legacy_entry(this, "nbns", i, INTERNAL_IP4_NBNS);
}
load_entries(this); load_entries(this);
return &this->public; return &this->public;
@@ -35,6 +35,11 @@ struct attr_provider_t {
*/ */
attribute_provider_t provider; attribute_provider_t provider;
/**
* Reload configuration from strongswan.conf.
*/
void (*reload)(attr_provider_t *this);
/** /**
* Destroy a attr_provider instance. * Destroy a attr_provider instance.
*/ */