adopt-children-job: Adopt child-creating tasks from the old IKE_SA
This commit is contained in:
@@ -53,6 +53,36 @@ METHOD(job_t, destroy, void,
|
|||||||
free(this);
|
free(this);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
METHOD(adopt_children_job_t, queue_task, void,
|
||||||
|
private_adopt_children_job_t *this, task_t *task)
|
||||||
|
{
|
||||||
|
array_insert_create(&this->tasks, ARRAY_TAIL, task);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Adopt child-creating tasks from the given IKE_SA
|
||||||
|
*/
|
||||||
|
static u_int adopt_child_tasks(private_adopt_children_job_t *this,
|
||||||
|
ike_sa_t *ike_sa, task_queue_t queue)
|
||||||
|
{
|
||||||
|
enumerator_t *tasks;
|
||||||
|
task_t *task;
|
||||||
|
u_int count = 0;
|
||||||
|
|
||||||
|
tasks = ike_sa->create_task_enumerator(ike_sa, queue);
|
||||||
|
while (tasks->enumerate(tasks, &task))
|
||||||
|
{
|
||||||
|
if (task->get_type(task) == TASK_QUICK_MODE)
|
||||||
|
{
|
||||||
|
ike_sa->remove_task(ike_sa, tasks);
|
||||||
|
queue_task(this, task);
|
||||||
|
count++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
tasks->destroy(tasks);
|
||||||
|
return count;
|
||||||
|
}
|
||||||
|
|
||||||
METHOD(job_t, execute, job_requeue_t,
|
METHOD(job_t, execute, job_requeue_t,
|
||||||
private_adopt_children_job_t *this)
|
private_adopt_children_job_t *this)
|
||||||
{
|
{
|
||||||
@@ -65,6 +95,7 @@ METHOD(job_t, execute, job_requeue_t,
|
|||||||
ike_sa_t *ike_sa;
|
ike_sa_t *ike_sa;
|
||||||
child_sa_t *child_sa;
|
child_sa_t *child_sa;
|
||||||
uint32_t unique;
|
uint32_t unique;
|
||||||
|
u_int tasks = 0;
|
||||||
|
|
||||||
ike_sa = charon->ike_sa_manager->checkout(charon->ike_sa_manager, this->id);
|
ike_sa = charon->ike_sa_manager->checkout(charon->ike_sa_manager, this->id);
|
||||||
if (ike_sa)
|
if (ike_sa)
|
||||||
@@ -127,11 +158,17 @@ METHOD(job_t, execute, job_requeue_t,
|
|||||||
* it does trigger an assign_vips(FALSE) event, so we also
|
* it does trigger an assign_vips(FALSE) event, so we also
|
||||||
* trigger one below */
|
* trigger one below */
|
||||||
ike_sa->clear_virtual_ips(ike_sa, FALSE);
|
ike_sa->clear_virtual_ips(ike_sa, FALSE);
|
||||||
if (children->get_count(children) || vips->get_count(vips))
|
|
||||||
|
tasks += adopt_child_tasks(this, ike_sa, TASK_QUEUE_ACTIVE);
|
||||||
|
tasks += adopt_child_tasks(this, ike_sa, TASK_QUEUE_QUEUED);
|
||||||
|
|
||||||
|
if (children->get_count(children) || tasks ||
|
||||||
|
vips->get_count(vips))
|
||||||
{
|
{
|
||||||
DBG1(DBG_IKE, "detected reauth of existing IKE_SA, "
|
DBG1(DBG_IKE, "detected reauth of existing IKE_SA, "
|
||||||
"adopting %d children and %d virtual IPs",
|
"adopting %d children, %d child tasks, and %d "
|
||||||
children->get_count(children), vips->get_count(vips));
|
"virtual IPs", children->get_count(children),
|
||||||
|
tasks, vips->get_count(vips));
|
||||||
}
|
}
|
||||||
if (ike_sa->get_state(ike_sa) == IKE_PASSIVE)
|
if (ike_sa->get_state(ike_sa) == IKE_PASSIVE)
|
||||||
{
|
{
|
||||||
@@ -152,7 +189,8 @@ METHOD(job_t, execute, job_requeue_t,
|
|||||||
charon->ike_sa_manager->checkin(
|
charon->ike_sa_manager->checkin(
|
||||||
charon->ike_sa_manager, ike_sa);
|
charon->ike_sa_manager, ike_sa);
|
||||||
}
|
}
|
||||||
if (children->get_count(children) || vips->get_count(vips))
|
if (children->get_count(children) || tasks ||
|
||||||
|
vips->get_count(vips))
|
||||||
{
|
{
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
@@ -237,12 +275,6 @@ METHOD(job_t, get_priority, job_priority_t,
|
|||||||
return JOB_PRIO_HIGH;
|
return JOB_PRIO_HIGH;
|
||||||
}
|
}
|
||||||
|
|
||||||
METHOD(adopt_children_job_t, queue_task, void,
|
|
||||||
private_adopt_children_job_t *this, task_t *task)
|
|
||||||
{
|
|
||||||
array_insert_create(&this->tasks, ARRAY_TAIL, task);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* See header
|
* See header
|
||||||
*/
|
*/
|
||||||
|
|||||||
Reference in New Issue
Block a user