- refactored ike_sa_manager_t class

This commit is contained in:
Jan Hutter
2005-11-16 07:45:05 +00:00
parent 083254bb69
commit f0ddd8ae26
2 changed files with 190 additions and 60 deletions
+185 -57
View File
@@ -27,8 +27,12 @@
#include "ike_sa_id.h" #include "ike_sa_id.h"
#include "utils/allocator.h" #include "utils/allocator.h"
#include "utils/logger.h"
#include "utils/logger_manager.h"
#include "utils/linked_list.h" #include "utils/linked_list.h"
extern logger_manager_t *global_logger_manager;
/** /**
* @brief An entry in the linked list, contains IKE_SA, locking and lookup data. * @brief An entry in the linked list, contains IKE_SA, locking and lookup data.
*/ */
@@ -51,13 +55,13 @@ struct ike_sa_entry_s {
*/ */
bool checked_out; bool checked_out;
/** /**
* does this SA let new treads in? * Does this SA drives out new threads?
*/ */
bool driveout_new_threads; bool driveout_new_threads;
/** /**
* does this SA drives out new threads? * Does this SA drives out waiting threads?
*/ */
bool driveout_waiting_threads;; bool driveout_waiting_threads;
/** /**
* identifiaction of ike_sa (SPIs) * identifiaction of ike_sa (SPIs)
*/ */
@@ -79,7 +83,6 @@ static status_t ike_sa_entry_destroy(ike_sa_entry_t *this)
return SUCCESS; return SUCCESS;
} }
/** /**
* @brief creates a new entry for the ike_sa list * @brief creates a new entry for the ike_sa list
* *
@@ -92,13 +95,18 @@ static ike_sa_entry_t *ike_sa_entry_create(ike_sa_id_t *ike_sa_id)
{ {
ike_sa_entry_t *this = allocator_alloc_thing(ike_sa_entry_t); ike_sa_entry_t *this = allocator_alloc_thing(ike_sa_entry_t);
/* destroy function */
this->destroy = ike_sa_entry_destroy; this->destroy = ike_sa_entry_destroy;
this->waiting_threads = 0; this->waiting_threads = 0;
pthread_cond_init(&(this->condvar), NULL); pthread_cond_init(&(this->condvar), NULL);
/* we set checkout flag when we really give it out */ /* we set checkout flag when we really give it out */
this->checked_out = FALSE; this->checked_out = FALSE;
this->driveout_new_threads = FALSE; this->driveout_new_threads = FALSE;
this->driveout_waiting_threads = FALSE; this->driveout_waiting_threads = FALSE;
/* ike_sa_id is always cloned */
ike_sa_id->clone(ike_sa_id, &(this->ike_sa_id)); ike_sa_id->clone(ike_sa_id, &(this->ike_sa_id));
this->ike_sa = ike_sa_create(ike_sa_id); this->ike_sa = ike_sa_create(ike_sa_id);
return this; return this;
@@ -125,6 +133,7 @@ struct private_ike_sa_manager_s {
* OUT_OF_RES when we already served 2^64 SPIs ;-) * OUT_OF_RES when we already served 2^64 SPIs ;-)
*/ */
status_t (*get_next_spi) (private_ike_sa_manager_t *this, spi_t *spi); status_t (*get_next_spi) (private_ike_sa_manager_t *this, spi_t *spi);
/** /**
* @brief find the ike_sa_entry in the list by SPIs * @brief find the ike_sa_entry in the list by SPIs
* *
@@ -134,10 +143,13 @@ struct private_ike_sa_manager_s {
* @param this the ike_sa_manager containing the list * @param this the ike_sa_manager containing the list
* @param ike_sa_id id of the ike_sa, containing SPIs * @param ike_sa_id id of the ike_sa, containing SPIs
* @param entry[out] pointer to set to the found entry * @param entry[out] pointer to set to the found entry
* @return SUCCESS when found, * @return
* NOT_FOUND when no such ike_sa_id in list * - SUCCESS when found,
* - NOT_FOUND when no such ike_sa_id in list
* - OUT_OF_RES
*/ */
status_t (*get_entry_by_id) (private_ike_sa_manager_t *this, ike_sa_id_t *ike_sa_id, ike_sa_entry_t **entry); status_t (*get_entry_by_id) (private_ike_sa_manager_t *this, ike_sa_id_t *ike_sa_id, ike_sa_entry_t **entry);
/** /**
* @brief find the ike_sa_entry in the list by pointer to SA. * @brief find the ike_sa_entry in the list by pointer to SA.
* *
@@ -145,26 +157,36 @@ struct private_ike_sa_manager_s {
* would be more efficient when storing a lot of IKE_SAs... * would be more efficient when storing a lot of IKE_SAs...
* *
* @param this the ike_sa_manager containing the list * @param this the ike_sa_manager containing the list
* @param ike_sa pointer to the ike_sa * @param ike_sa pointer to the ike_sa
* @param entry[out] pointer to set to the found entry * @param entry[out] pointer to set to the found entry
* @return SUCCESS when found, * @return
* NOT_FOUND when no such ike_sa_id in list * - SUCCESS when found,
* - NOT_FOUND when no such ike_sa_id in list
* - OUT_OF_RES
*/ */
status_t (*get_entry_by_sa) (private_ike_sa_manager_t *this, ike_sa_t *ike_sa, ike_sa_entry_t **entry); status_t (*get_entry_by_sa) (private_ike_sa_manager_t *this, ike_sa_t *ike_sa, ike_sa_entry_t **entry);
/** /**
* @brief delete an entry from the linked list * @brief delete an entry from the linked list
* *
* @param this the ike_sa_manager containing the list * @param this the ike_sa_manager containing the list
* @param entry entry to delete * @param entry entry to delete
* @return SUCCESS when found, * @return
* NOT_FOUND when no such ike_sa_id in list * - SUCCESS when found,
* - NOT_FOUND when no such ike_sa_id in list
*/ */
status_t (*delete_entry) (private_ike_sa_manager_t *this, ike_sa_entry_t *entry); status_t (*delete_entry) (private_ike_sa_manager_t *this, ike_sa_entry_t *entry);
/** /**
* lock for exclusivly accessing the manager * lock for exclusivly accessing the manager
*/ */
pthread_mutex_t mutex; pthread_mutex_t mutex;
/**
* Logger used for this IKE SA Manager
*/
logger_t *logger;
/** /**
* Linked list with entries for the ike_sa * Linked list with entries for the ike_sa
*/ */
@@ -183,7 +205,16 @@ static status_t get_entry_by_id(private_ike_sa_manager_t *this, ike_sa_id_t *ike
{ {
linked_list_t *list = this->list; linked_list_t *list = this->list;
linked_list_iterator_t *iterator; linked_list_iterator_t *iterator;
list->create_iterator(list, &iterator, TRUE); status_t status;
status = list->create_iterator(list, &iterator, TRUE);
if (status != SUCCESS)
{
return status;
}
/* default status */
status = NOT_FOUND;
while (iterator->has_next(iterator)) while (iterator->has_next(iterator))
{ {
ike_sa_entry_t *current; ike_sa_entry_t *current;
@@ -193,12 +224,13 @@ static status_t get_entry_by_id(private_ike_sa_manager_t *this, ike_sa_id_t *ike
if (are_equal) if (are_equal)
{ {
*entry = current; *entry = current;
iterator->destroy(iterator); status = SUCCESS;
return SUCCESS; break;
} }
} }
iterator->destroy(iterator); iterator->destroy(iterator);
return NOT_FOUND; return status;
} }
/** /**
@@ -208,24 +240,34 @@ static status_t get_entry_by_sa(private_ike_sa_manager_t *this, ike_sa_t *ike_sa
{ {
linked_list_t *list = this->list; linked_list_t *list = this->list;
linked_list_iterator_t *iterator; linked_list_iterator_t *iterator;
list->create_iterator(list, &iterator, TRUE); status_t status;
status = list->create_iterator(list, &iterator, TRUE);
if (status != SUCCESS)
{
return status;
}
/* default status */
status = NOT_FOUND;
while (iterator->has_next(iterator)) while (iterator->has_next(iterator))
{ {
ike_sa_entry_t *current; ike_sa_entry_t *current;
iterator->current(iterator, (void**)&current); iterator->current(iterator, (void**)&current);
/* only pointers are compared */
if (current->ike_sa == ike_sa) if (current->ike_sa == ike_sa)
{ {
*entry = current; *entry = current;
iterator->destroy(iterator); status = SUCCESS;
return SUCCESS; break;
} }
} }
iterator->destroy(iterator); iterator->destroy(iterator);
return NOT_FOUND; return status;
} }
/** /**
* @see private_ike_sa_manager_t.delete_entry * @see private_ike_sa_manager_s.delete_entry
*/ */
static status_t delete_entry(private_ike_sa_manager_t *this, ike_sa_entry_t *entry) static status_t delete_entry(private_ike_sa_manager_t *this, ike_sa_entry_t *entry)
{ {
@@ -268,7 +310,6 @@ static status_t get_next_spi(private_ike_sa_manager_t *this, spi_t *spi)
return SUCCESS; return SUCCESS;
} }
/** /**
* @see ike_sa_manager_s.checkout_ike_sa * @see ike_sa_manager_s.checkout_ike_sa
*/ */
@@ -286,15 +327,17 @@ static status_t checkout(private_ike_sa_manager_t *this, ike_sa_id_t *ike_sa_id,
if (initiator_spi_set && responder_spi_set) if (initiator_spi_set && responder_spi_set)
{ {
/* we SHOULD have an IKE_SA for these SPIs in the list, /* we SHOULD have an IKE_SA for these SPIs in the list,
* if not, we cant handle the request... * if not, we can't handle the request...
*/ */
ike_sa_entry_t *entry; ike_sa_entry_t *entry;
/* look for the entry */ /* look for the entry */
if (this->get_entry_by_id(this, ike_sa_id, &entry) == SUCCESS) if (this->get_entry_by_id(this, ike_sa_id, &entry) == SUCCESS)
{ {
/* can we give this out to new requesters? */ /* can we give this out to new requesters?*/
if (entry->driveout_new_threads) if (entry->driveout_new_threads)
{ {
this->logger->log(this->logger,CONTROL_MORE,"Drive out new thread for existing IKE_SA");
/* no we can't */
retval = NOT_FOUND; retval = NOT_FOUND;
} }
else else
@@ -310,11 +353,13 @@ static status_t checkout(private_ike_sa_manager_t *this, ike_sa_id_t *ike_sa_id,
pthread_cond_wait(&(entry->condvar), &(this->mutex)); pthread_cond_wait(&(entry->condvar), &(this->mutex));
entry->waiting_threads--; entry->waiting_threads--;
} }
/* hm, a deletion request forbids us to get this SA, go home */ /* hm, a deletion request forbids us to get this SA, go home */
if (entry->driveout_waiting_threads) if (entry->driveout_waiting_threads)
{ {
/* we must signal here, others are interested that we leave */ /* we must signal here, others are interested that we leave */
pthread_cond_signal(&(entry->condvar)); pthread_cond_signal(&(entry->condvar));
this->logger->log(this->logger,CONTROL_MORE,"Drive out waiting thread for existing IKE_SA");
retval = NOT_FOUND; retval = NOT_FOUND;
} }
else else
@@ -347,20 +392,44 @@ static status_t checkout(private_ike_sa_manager_t *this, ike_sa_id_t *ike_sa_id,
ike_sa_entry_t *new_ike_sa_entry; ike_sa_entry_t *new_ike_sa_entry;
/* set SPIs, we are the responder */ /* set SPIs, we are the responder */
this->get_next_spi(this, &responder_spi); retval = this->get_next_spi(this, &responder_spi);
/* we also set arguments spi, so its still valid */
ike_sa_id->set_responder_spi(ike_sa_id, responder_spi);
/* create entry */ if (retval == SUCCESS)
new_ike_sa_entry = ike_sa_entry_create(ike_sa_id); { /* next SPI could be created */
this->list->insert_last(this->list, new_ike_sa_entry);
/* we also set arguments spi, so its still valid */
/* check ike_sa out */ ike_sa_id->set_responder_spi(ike_sa_id, responder_spi);
new_ike_sa_entry->checked_out = TRUE;
*ike_sa = new_ike_sa_entry->ike_sa; /* create entry */
new_ike_sa_entry = ike_sa_entry_create(ike_sa_id);
/* DON'T use return, we must unlock the mutex! */ if (new_ike_sa_entry != NULL)
retval = SUCCESS; {
retval = this->list->insert_last(this->list, new_ike_sa_entry);
if (retval == SUCCESS)
{
/* check ike_sa out */
new_ike_sa_entry->checked_out = TRUE;
*ike_sa = new_ike_sa_entry->ike_sa;
/* DON'T use return, we must unlock the mutex! */
}
else
{
/* ike_sa_entry could not be added to list*/
this->logger->log(this->logger,CONTROL,"Fatal error: ike_sa_entry could not be added to list");
}
}
else
{
/* new ike_sa_entry could not be created */
this->logger->log(this->logger,CONTROL,"Fatal error: ike_sa_entry could not be created");
}
}
else
{ /* next SPI could not be created */
this->logger->log(this->logger,CONTROL,"Fatal error: Next SPI could not be created");
}
} }
else if (!initiator_spi_set && !responder_spi_set) else if (!initiator_spi_set && !responder_spi_set)
{ {
@@ -370,26 +439,50 @@ static status_t checkout(private_ike_sa_manager_t *this, ike_sa_id_t *ike_sa_id,
spi_t initiator_spi; spi_t initiator_spi;
ike_sa_entry_t *new_ike_sa_entry; ike_sa_entry_t *new_ike_sa_entry;
this->get_next_spi(this, &initiator_spi); retval = this->get_next_spi(this, &initiator_spi);
if (retval == SUCCESS)
{
/* we also set arguments SPI, so its still valid */
ike_sa_id->set_initiator_spi(ike_sa_id, initiator_spi);
/* we also set arguments SPI, so its still valid */ /* create entry */
ike_sa_id->set_initiator_spi(ike_sa_id, initiator_spi); new_ike_sa_entry = ike_sa_entry_create(ike_sa_id);
if (new_ike_sa_entry != NULL)
{
retval = this->list->insert_last(this->list, new_ike_sa_entry);
/* create entry */ if (retval == SUCCESS)
new_ike_sa_entry = ike_sa_entry_create(ike_sa_id); {
this->list->insert_last(this->list, new_ike_sa_entry); /* check ike_sa out */
new_ike_sa_entry->checked_out = TRUE;
*ike_sa = new_ike_sa_entry->ike_sa;
/* DON'T use return, we must unlock the mutex! */
}
else
{
/* ike_sa_entry could not be added to list*/
this->logger->log(this->logger,CONTROL,"Fatal error: ike_sa_entry could not be added to list");
}
}
else
{
/* new ike_sa_entry could not be created */
this->logger->log(this->logger,CONTROL,"Fatal error: ike_sa_entry could not be created");
}
}
else
{
/* next SPI could not be created */
this->logger->log(this->logger,CONTROL,"Fatal error: Next SPI could not be created");
}
/* check ike_sa out */
new_ike_sa_entry->checked_out = TRUE;
*ike_sa = new_ike_sa_entry->ike_sa;
/* DON'T use return, we must unlock the mutex! */
retval = SUCCESS;
} }
else else
{ {
/* responder set, initiator not: here is something seriously wrong! */ /* responder set, initiator not: here is something seriously wrong! */
this->logger->log(this->logger,CONTROL_MORE,"Invalid IKE_SA SPI's");
/* DON'T use return, we must unlock the mutex! */ /* DON'T use return, we must unlock the mutex! */
retval = INVALID_ARG; retval = INVALID_ARG;
} }
@@ -399,6 +492,10 @@ static status_t checkout(private_ike_sa_manager_t *this, ike_sa_id_t *ike_sa_id,
return retval; return retval;
} }
/**
* Implements ike_sa_manager_t-function checkin.
* @see ike_sa_manager_t.checkin.
*/
static status_t checkin(private_ike_sa_manager_t *this, ike_sa_t *ike_sa) static status_t checkin(private_ike_sa_manager_t *this, ike_sa_t *ike_sa)
{ {
/* to check the SA back in, we look for the pointer of the ike_sa /* to check the SA back in, we look for the pointer of the ike_sa
@@ -423,6 +520,7 @@ static status_t checkin(private_ike_sa_manager_t *this, ike_sa_t *ike_sa)
} }
else else
{ {
this->logger->log(this->logger,CONTROL,"Fatal Error: Tried to checkin nonexisting IKE_SA");
/* this SA is no more, this REALLY should not happen */ /* this SA is no more, this REALLY should not happen */
retval = NOT_FOUND; retval = NOT_FOUND;
} }
@@ -431,7 +529,10 @@ static status_t checkin(private_ike_sa_manager_t *this, ike_sa_t *ike_sa)
} }
/**
* Implements ike_sa_manager_t-function checkin_and_delete.
* @see ike_sa_manager_t.checkin_and_delete.
*/
static status_t checkin_and_delete(private_ike_sa_manager_t *this, ike_sa_t *ike_sa) static status_t checkin_and_delete(private_ike_sa_manager_t *this, ike_sa_t *ike_sa)
{ {
/* deletion is a bit complex, we must garant that no thread is waiting for /* deletion is a bit complex, we must garant that no thread is waiting for
@@ -452,7 +553,7 @@ static status_t checkin_and_delete(private_ike_sa_manager_t *this, ike_sa_t *ike
entry->driveout_waiting_threads = TRUE; entry->driveout_waiting_threads = TRUE;
/* wait until all workers have done their work */ /* wait until all workers have done their work */
while (entry->waiting_threads) while (entry->waiting_threads > 0)
{ {
/* let the other threads do some work*/ /* let the other threads do some work*/
pthread_cond_signal(&(entry->condvar)); pthread_cond_signal(&(entry->condvar));
@@ -465,6 +566,7 @@ static status_t checkin_and_delete(private_ike_sa_manager_t *this, ike_sa_t *ike
} }
else else
{ {
this->logger->log(this->logger,CONTROL,"Fatal Error: Tried to checkin and delete nonexisting IKE_SA");
retval = NOT_FOUND; retval = NOT_FOUND;
} }
@@ -472,6 +574,10 @@ static status_t checkin_and_delete(private_ike_sa_manager_t *this, ike_sa_t *ike
return retval; return retval;
} }
/**
* Implements ike_sa_manager_t-function delete.
* @see ike_sa_manager_t.delete.
*/
static status_t delete(private_ike_sa_manager_t *this, ike_sa_id_t *ike_sa_id) static status_t delete(private_ike_sa_manager_t *this, ike_sa_id_t *ike_sa_id)
{ {
/* deletion is a bit complex, we must garant that no thread is waiting for /* deletion is a bit complex, we must garant that no thread is waiting for
@@ -503,6 +609,7 @@ static status_t delete(private_ike_sa_manager_t *this, ike_sa_id_t *ike_sa_id)
} }
else else
{ {
this->logger->log(this->logger,CONTROL,"Fatal Error: Tried to delete nonexisting IKE_SA");
retval = NOT_FOUND; retval = NOT_FOUND;
} }
@@ -510,16 +617,28 @@ static status_t delete(private_ike_sa_manager_t *this, ike_sa_id_t *ike_sa_id)
return retval; return retval;
} }
/**
* Implements ike_sa_manager_t-function destroy.
* @see ike_sa_manager_t.destroy.
*/
static status_t destroy(private_ike_sa_manager_t *this) static status_t destroy(private_ike_sa_manager_t *this)
{ {
/* destroy all list entries */ /* destroy all list entries */
linked_list_t *list = this->list; linked_list_t *list = this->list;
linked_list_iterator_t *iterator; linked_list_iterator_t *iterator;
status_t status;
pthread_mutex_lock(&(this->mutex)); pthread_mutex_lock(&(this->mutex));
/* Step 1: drive out all waiting threads */ /* Step 1: drive out all waiting threads */
list->create_iterator(list, &iterator, TRUE); status = list->create_iterator(list, &iterator, TRUE);
if (status != SUCCESS)
{
this->logger->log(this->logger,CONTROL,"Fatal Error: Out of Ressources to greate interator while destroying IKE_SA-Manager");
return FAILED;
}
while (iterator->has_next(iterator)) while (iterator->has_next(iterator))
{ {
ike_sa_entry_t *entry; ike_sa_entry_t *entry;
@@ -553,15 +672,15 @@ static status_t destroy(private_ike_sa_manager_t *this)
iterator->destroy(iterator); iterator->destroy(iterator);
list->destroy(list); list->destroy(list);
pthread_mutex_unlock(&(this->mutex)); pthread_mutex_unlock(&(this->mutex));
allocator_free(this); allocator_free(this);
return SUCCESS; return SUCCESS;
} }
/*
* Described in header
*/
ike_sa_manager_t *ike_sa_manager_create() ike_sa_manager_t *ike_sa_manager_create()
{ {
private_ike_sa_manager_t *this = allocator_alloc_thing(private_ike_sa_manager_t); private_ike_sa_manager_t *this = allocator_alloc_thing(private_ike_sa_manager_t);
@@ -573,18 +692,27 @@ ike_sa_manager_t *ike_sa_manager_create()
this->public.delete = (status_t(*)(ike_sa_manager_t*, ike_sa_id_t *sa_id))delete; this->public.delete = (status_t(*)(ike_sa_manager_t*, ike_sa_id_t *sa_id))delete;
this->public.checkin_and_delete = (status_t(*)(ike_sa_manager_t*, ike_sa_t *ike_sa))checkin_and_delete; this->public.checkin_and_delete = (status_t(*)(ike_sa_manager_t*, ike_sa_t *ike_sa))checkin_and_delete;
/* initialize private data */ /* initialize private functions */
this->get_next_spi = get_next_spi; this->get_next_spi = get_next_spi;
this->get_entry_by_sa = get_entry_by_sa; this->get_entry_by_sa = get_entry_by_sa;
this->get_entry_by_id = get_entry_by_id; this->get_entry_by_id = get_entry_by_id;
this->delete_entry = delete_entry; this->delete_entry = delete_entry;
/* initialize private variables */
this->list = linked_list_create(); this->list = linked_list_create();
if (this->list == NULL) if (this->list == NULL)
{ {
allocator_free(this); allocator_free(this);
return NULL; return NULL;
} }
this->logger = global_logger_manager->create_logger(global_logger_manager,IKE_SA_MANAGER,NULL);
if (this->logger == NULL)
{
this->list->destroy(this->list);
allocator_free(this);
return NULL;
}
pthread_mutex_init(&(this->mutex), NULL); pthread_mutex_init(&(this->mutex), NULL);
+5 -3
View File
@@ -55,14 +55,16 @@ struct ike_sa_manager_s {
* @param ike_sa_manager the manager object * @param ike_sa_manager the manager object
* @param ike_sa_id[in/out] the SA identifier, will be updated * @param ike_sa_id[in/out] the SA identifier, will be updated
* @param ike_sa[out] checked out SA * @param ike_sa[out] checked out SA
* @returns SUCCESS if checkout successful * @returns
* NOT_FOUND when no such SA is available * - SUCCESS if checkout successful
* - NOT_FOUND when no such SA is available
* - OUT_OF_RES
*/ */
status_t (*checkout) (ike_sa_manager_t* ike_sa_manager, ike_sa_id_t *sa_id, ike_sa_t **ike_sa); status_t (*checkout) (ike_sa_manager_t* ike_sa_manager, ike_sa_id_t *sa_id, ike_sa_t **ike_sa);
/** /**
* @brief Checkin the SA after usage * @brief Checkin the SA after usage
* *
* @warning the pointer MUST NOT be used after checkin! The SA must be checked * @warning the SA pointer MUST NOT be used after checkin! The SA must be checked
* out again! * out again!
* *
* @param ike_sa_manager the manager object * @param ike_sa_manager the manager object