quick-mode: Maintain reference to reqid while CHILD_SA is established

This commit is contained in:
Tobias Brunner
2023-11-13 12:02:11 +01:00
parent 4bfd93b8db
commit 4ea739baf4
2 changed files with 18 additions and 1 deletions
+14 -1
View File
@@ -1432,7 +1432,16 @@ METHOD(quick_mode_t, get_mid, uint32_t,
METHOD(quick_mode_t, use_reqid, void,
private_quick_mode_t *this, uint32_t reqid)
{
this->child.reqid = reqid;
uint32_t existing_reqid = this->child.reqid;
if (!reqid || charon->kernel->ref_reqid(charon->kernel, reqid) == SUCCESS)
{
this->child.reqid = reqid;
if (existing_reqid)
{
charon->kernel->release_reqid(charon->kernel, existing_reqid);
}
}
}
METHOD(quick_mode_t, use_marks, void,
@@ -1496,6 +1505,10 @@ METHOD(task_t, destroy, void,
DESTROY_IF(this->child_sa);
DESTROY_IF(this->config);
DESTROY_IF(this->dh);
if (this->child.reqid)
{
charon->kernel->release_reqid(charon->kernel, this->child.reqid);
}
free(this);
}
@@ -50,6 +50,10 @@ struct quick_mode_t {
/**
* Use a specific reqid to install this CHILD_SA.
*
* This must only be called with dynamically allocated reqids (i.e. from
* kernel_interface_t::alloc_reqid()), the method takes a reference that's
* maintained for the lifetime of the task.
*
* @param reqid reqid to use
*/
void (*use_reqid)(quick_mode_t *this, uint32_t reqid);