added imc/imv_manager remove method
This commit is contained in:
@@ -67,12 +67,31 @@ METHOD(imc_manager_t, add, bool,
|
||||
{
|
||||
DBG1(DBG_TNC, "could not provide bind function for IMC '%s'",
|
||||
imc->get_name(imc));
|
||||
this->imcs->remove_last(this->imcs, (void**)&imc);
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
METHOD(imc_manager_t, remove_, imc_t*,
|
||||
private_tnc_imc_manager_t *this, TNC_IMCID id)
|
||||
{
|
||||
enumerator_t *enumerator;
|
||||
imc_t *imc;
|
||||
|
||||
enumerator = this->imcs->create_enumerator(this->imcs);
|
||||
while (enumerator->enumerate(enumerator, &imc))
|
||||
{
|
||||
if (id == imc->get_id(imc))
|
||||
{
|
||||
this->imcs->remove_at(this->imcs, enumerator);
|
||||
return imc;
|
||||
}
|
||||
}
|
||||
enumerator->destroy(enumerator);
|
||||
}
|
||||
|
||||
METHOD(imc_manager_t, notify_connection_change, void,
|
||||
private_tnc_imc_manager_t *this, TNC_ConnectionID id,
|
||||
TNC_ConnectionState state)
|
||||
@@ -178,6 +197,7 @@ imc_manager_t* tnc_imc_manager_create(void)
|
||||
INIT(this,
|
||||
.public = {
|
||||
.add = _add,
|
||||
.remove = _remove_, /* avoid name conflict with stdio.h */
|
||||
.notify_connection_change = _notify_connection_change,
|
||||
.begin_handshake = _begin_handshake,
|
||||
.set_message_types = _set_message_types,
|
||||
|
||||
@@ -67,11 +67,30 @@ METHOD(imv_manager_t, add, bool,
|
||||
{
|
||||
DBG1(DBG_TNC, "could not provide bind function for IMV '%s'",
|
||||
imv->get_name(imv));
|
||||
this->imvs->remove_last(this->imvs, (void**)&imv);
|
||||
return FALSE;
|
||||
}
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
METHOD(imv_manager_t, remove_, imv_t*,
|
||||
private_tnc_imv_manager_t *this, TNC_IMVID id)
|
||||
{
|
||||
enumerator_t *enumerator;
|
||||
imv_t *imv;
|
||||
|
||||
enumerator = this->imvs->create_enumerator(this->imvs);
|
||||
while (enumerator->enumerate(enumerator, &imv))
|
||||
{
|
||||
if (id == imv->get_id(imv))
|
||||
{
|
||||
this->imvs->remove_at(this->imvs, enumerator);
|
||||
return imv;
|
||||
}
|
||||
}
|
||||
enumerator->destroy(enumerator);
|
||||
}
|
||||
|
||||
METHOD(imv_manager_t, notify_connection_change, void,
|
||||
private_tnc_imv_manager_t *this, TNC_ConnectionID id,
|
||||
TNC_ConnectionState state)
|
||||
@@ -163,6 +182,7 @@ imv_manager_t* tnc_imv_manager_create(void)
|
||||
INIT(this,
|
||||
.public = {
|
||||
.add = _add,
|
||||
.remove = _remove_, /* avoid name conflict with stdio.h */
|
||||
.notify_connection_change = _notify_connection_change,
|
||||
.set_message_types = _set_message_types,
|
||||
.receive_message = _receive_message,
|
||||
|
||||
@@ -40,6 +40,14 @@ struct imc_manager_t {
|
||||
*/
|
||||
bool (*add)(imc_manager_t *this, imc_t *imc);
|
||||
|
||||
/**
|
||||
* Remove an IMC instance from the list and return it
|
||||
*
|
||||
* @param id ID of IMC instance
|
||||
* @return removed IMC instance
|
||||
*/
|
||||
imc_t* (*remove)(imc_manager_t *this, TNC_IMCID id);
|
||||
|
||||
/**
|
||||
* Notify all IMC instances
|
||||
*
|
||||
|
||||
@@ -40,6 +40,14 @@ struct imv_manager_t {
|
||||
*/
|
||||
bool (*add)(imv_manager_t *this, imv_t *imv);
|
||||
|
||||
/**
|
||||
* Remove an IMV instance from the list and return it
|
||||
*
|
||||
* @param id ID of IMV instance
|
||||
* @return removed IMC instance
|
||||
*/
|
||||
void (*remove)(imv_manager_t *this, TNC_IMVID id);
|
||||
|
||||
/**
|
||||
* Notify all IMV instances
|
||||
*
|
||||
|
||||
Reference in New Issue
Block a user