Add reload support to attr plugin
This commit is contained in:
@@ -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,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -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.
|
||||||
*/
|
*/
|
||||||
|
|||||||
Reference in New Issue
Block a user