implemented batch_ending() and solicit_recommendation() functions

This commit is contained in:
Andreas Steffen
2010-11-09 20:43:51 +01:00
parent 08235c7ded
commit d599902a3a
5 changed files with 79 additions and 1 deletions
@@ -175,6 +175,23 @@ METHOD(imc_manager_t, receive_message, void,
enumerator->destroy(enumerator);
}
METHOD(imc_manager_t, batch_ending, void,
private_tnc_imc_manager_t *this, TNC_ConnectionID id)
{
enumerator_t *enumerator;
imc_t *imc;
enumerator = this->imcs->create_enumerator(this->imcs);
while (enumerator->enumerate(enumerator, &imc))
{
if (imc->batch_ending)
{
imc->batch_ending(imc->get_id(imc), id);
}
}
enumerator->destroy(enumerator);
}
METHOD(imc_manager_t, destroy, void,
private_tnc_imc_manager_t *this)
{
@@ -210,6 +227,7 @@ imc_manager_t* tnc_imc_manager_create(void)
.begin_handshake = _begin_handshake,
.set_message_types = _set_message_types,
.receive_message = _receive_message,
.batch_ending = _batch_ending,
.destroy = _destroy,
},
.imcs = linked_list_create(),
@@ -140,6 +140,20 @@ METHOD(imv_manager_t, set_message_types, TNC_Result,
return result;
}
METHOD(imv_manager_t, solicit_recommendation, void,
private_tnc_imv_manager_t *this, TNC_ConnectionID id)
{
enumerator_t *enumerator;
imv_t *imv;
enumerator = this->imvs->create_enumerator(this->imvs);
while (enumerator->enumerate(enumerator, &imv))
{
imv->solicit_recommendation(imv->get_id(imv), id);
}
enumerator->destroy(enumerator);
}
METHOD(imv_manager_t, receive_message, void,
private_tnc_imv_manager_t *this, TNC_ConnectionID connection_id,
TNC_BufferReference message,
@@ -161,6 +175,23 @@ METHOD(imv_manager_t, receive_message, void,
enumerator->destroy(enumerator);
}
METHOD(imv_manager_t, batch_ending, void,
private_tnc_imv_manager_t *this, TNC_ConnectionID id)
{
enumerator_t *enumerator;
imv_t *imv;
enumerator = this->imvs->create_enumerator(this->imvs);
while (enumerator->enumerate(enumerator, &imv))
{
if (imv->batch_ending)
{
imv->batch_ending(imv->get_id(imv), id);
}
}
enumerator->destroy(enumerator);
}
METHOD(imv_manager_t, destroy, void,
private_tnc_imv_manager_t *this)
{
@@ -194,7 +225,9 @@ imv_manager_t* tnc_imv_manager_create(void)
.get_count = _get_count,
.notify_connection_change = _notify_connection_change,
.set_message_types = _set_message_types,
.solicit_recommendation = _solicit_recommendation,
.receive_message = _receive_message,
.batch_ending = _batch_ending,
.destroy = _destroy,
},
.imvs = linked_list_create(),
@@ -100,11 +100,13 @@ METHOD(tls_t, process, status_t,
{
charon->imvs->receive_message(charon->imvs, this->connection_id,
pos, len, 0x0080ab31);
charon->imvs->batch_ending(charon->imvs, this->connection_id);
}
else
{
charon->imcs->receive_message(charon->imcs, this->connection_id,
pos, len, 0x0080ab31);
charon->imcs->batch_ending(charon->imcs, this->connection_id);
}
return NEED_MORE;
}