denozord/strongswan-ext
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ikev2: Allow tasks to do work after processing requests/responses

Browse Source
This commit is contained in:
Tobias Brunner
2022-06-29 10:28:50 +02:00
parent 210e2177aa
commit 36312808d1
1 changed files with 51 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/libcharon/sa/ikev2/task_manager_v2.c
+51 -1
View File
@@ -1,5 +1,5 @@
/* /*
* Copyright (C) 2007-2018 Tobias Brunner * Copyright (C) 2007-2019 Tobias Brunner
* Copyright (C) 2007-2010 Martin Willi * Copyright (C) 2007-2010 Martin Willi
* *
* Copyright (C) secunet Security Networks AG * Copyright (C) secunet Security Networks AG
@@ -837,6 +837,31 @@ static status_t process_response(private_task_manager_t *this,
} }
enumerator->destroy(enumerator); enumerator->destroy(enumerator);
enumerator = array_create_enumerator(this->active_tasks);
while (enumerator->enumerate(enumerator, &task))
{
if (!task->post_process)
{
continue;
}
switch (task->post_process(task, message))
{
case SUCCESS:
array_remove_at(this->active_tasks, enumerator);
task->destroy(task);
break;
case NEED_MORE:
break;
default:
/* critical failure, destroy IKE_SA */
array_remove_at(this->active_tasks, enumerator);
enumerator->destroy(enumerator);
task->destroy(task);
return DESTROY_ME;
}
}
enumerator->destroy(enumerator);
this->initiating.mid++; this->initiating.mid++;
this->initiating.type = EXCHANGE_TYPE_UNDEFINED; this->initiating.type = EXCHANGE_TYPE_UNDEFINED;
clear_packets(this->initiating.packets); clear_packets(this->initiating.packets);
@@ -1314,6 +1339,31 @@ static status_t process_request(private_task_manager_t *this,
} }
enumerator->destroy(enumerator); enumerator->destroy(enumerator);
enumerator = array_create_enumerator(this->passive_tasks);
while (enumerator->enumerate(enumerator, &task))
{
if (!task->post_process)
{
continue;
}
switch (task->post_process(task, message))
{
case SUCCESS:
array_remove_at(this->passive_tasks, enumerator);
task->destroy(task);
break;
case NEED_MORE:
break;
default:
/* critical failure, destroy IKE_SA */
array_remove_at(this->passive_tasks, enumerator);
enumerator->destroy(enumerator);
task->destroy(task);
return DESTROY_ME;
}
}
enumerator->destroy(enumerator);
return build_response(this, message); return build_response(this, message);
} }