commit f0ff5817d0647bdecd1ec99488db9378e304cf83 Author: sauwming Date: Mon May 17 09:56:27 2021 +0800 Fix double free of stun session (#2709) --- a/pjnath/include/pjnath/stun_session.h +++ b/pjnath/include/pjnath/stun_session.h @@ -341,6 +341,7 @@ struct pj_stun_tx_data pj_pool_t *pool; /**< Pool. */ pj_stun_session *sess; /**< The STUN session. */ pj_stun_msg *msg; /**< The STUN message. */ + pj_bool_t is_destroying; /**< Is destroying? */ void *token; /**< The token. */ --- a/pjnath/src/pjnath/stun_session.c +++ b/pjnath/src/pjnath/stun_session.c @@ -167,16 +167,27 @@ static void tdata_on_destroy(void *arg) { pj_stun_tx_data *tdata = (pj_stun_tx_data*)arg; + if (tdata->grp_lock) { + pj_grp_lock_dec_ref(tdata->sess->grp_lock); + } + pj_pool_safe_release(&tdata->pool); } static void destroy_tdata(pj_stun_tx_data *tdata, pj_bool_t force) { - TRACE_((THIS_FILE, "tdata %p destroy request, force=%d, tsx=%p", tdata, - force, tdata->client_tsx)); + TRACE_((THIS_FILE, + "tdata %p destroy request, force=%d, tsx=%p, destroying=%d", + tdata, force, tdata->client_tsx, tdata->is_destroying)); + + /* Just return if destroy has been requested before */ + if (tdata->is_destroying) + return; /* STUN session may have been destroyed, except when tdata is cached. */ + tdata->is_destroying = PJ_TRUE; + if (tdata->res_timer.id != PJ_FALSE) { pj_timer_heap_cancel_if_active(tdata->sess->cfg->timer_heap, &tdata->res_timer, PJ_FALSE); @@ -189,7 +200,6 @@ static void destroy_tdata(pj_stun_tx_dat pj_stun_client_tsx_set_data(tdata->client_tsx, NULL); } if (tdata->grp_lock) { - pj_grp_lock_dec_ref(tdata->sess->grp_lock); pj_grp_lock_dec_ref(tdata->grp_lock); } else { tdata_on_destroy(tdata); @@ -200,11 +210,11 @@ static void destroy_tdata(pj_stun_tx_dat /* "Probably" this is to absorb retransmission */ pj_time_val delay = {0, 300}; pj_stun_client_tsx_schedule_destroy(tdata->client_tsx, &delay); + tdata->is_destroying = PJ_FALSE; } else { pj_list_erase(tdata); if (tdata->grp_lock) { - pj_grp_lock_dec_ref(tdata->sess->grp_lock); pj_grp_lock_dec_ref(tdata->grp_lock); } else { tdata_on_destroy(tdata); @@ -238,7 +248,7 @@ static void on_cache_timeout(pj_timer_he sess = tdata->sess; pj_grp_lock_acquire(sess->grp_lock); - if (sess->is_destroying) { + if (sess->is_destroying || tdata->is_destroying) { pj_grp_lock_release(sess->grp_lock); return; }