fixed rtmp active calls count

This commit is contained in:
William King 2013-04-19 11:13:32 -07:00
parent 7df4e52d61
commit 985c8de056

View File

@ -292,13 +292,6 @@ switch_status_t rtmp_on_hangup(switch_core_session_t *session)
switch_core_hash_delete_wrlock(rsession->session_hash, switch_core_session_get_uuid(session), rsession->session_rwlock); switch_core_hash_delete_wrlock(rsession->session_hash, switch_core_session_get_uuid(session), rsession->session_rwlock);
switch_mutex_lock(rsession->profile->mutex);
rsession->profile->calls--;
if (rsession->profile->calls < 0) {
rsession->profile->calls = 0;
}
switch_mutex_unlock(rsession->profile->mutex);
switch_mutex_lock(rsession->count_mutex); switch_mutex_lock(rsession->count_mutex);
rsession->active_sessions--; rsession->active_sessions--;
switch_mutex_unlock(rsession->count_mutex); switch_mutex_unlock(rsession->count_mutex);
@ -468,31 +461,37 @@ switch_status_t rtmp_write_frame(switch_core_session_t *session, switch_frame_t
assert(tech_pvt != NULL); assert(tech_pvt != NULL);
rsession = tech_pvt->rtmp_session; rsession = tech_pvt->rtmp_session;
if ( rsession == NULL ) {
goto error;
}
switch_thread_rwlock_wrlock(rsession->rwlock);
if (!switch_test_flag(tech_pvt, TFLAG_IO)) { if (!switch_test_flag(tech_pvt, TFLAG_IO)) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "TFLAG_IO not set\n"); switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "TFLAG_IO not set\n");
return SWITCH_STATUS_FALSE; goto error;
} }
if (switch_test_flag(tech_pvt, TFLAG_DETACHED) || !switch_test_flag(rsession, SFLAG_AUDIO)) { if (switch_test_flag(tech_pvt, TFLAG_DETACHED) || !switch_test_flag(rsession, SFLAG_AUDIO)) {
return SWITCH_STATUS_SUCCESS; goto success;
} }
if (!rsession || !tech_pvt->audio_codec || !tech_pvt->write_channel) { if (!rsession || !tech_pvt->audio_codec || !tech_pvt->write_channel) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Missing mandatory value\n"); switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Missing mandatory value\n");
return SWITCH_STATUS_FALSE; goto error;
} }
if (rsession->state >= RS_DESTROY) { if (rsession->state >= RS_DESTROY) {
return SWITCH_STATUS_FALSE; goto error;
} }
if (frame->datalen+1 > frame->buflen) { if (frame->datalen+1 > frame->buflen) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Datalen too big\n"); switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Datalen too big\n");
return SWITCH_STATUS_FALSE; goto error;
} }
if (frame->flags & SFF_CNG) { if (frame->flags & SFF_CNG) {
return SWITCH_STATUS_SUCCESS; goto success;
} }
/* Build message */ /* Build message */
@ -508,7 +507,14 @@ switch_status_t rtmp_write_frame(switch_core_session_t *session, switch_frame_t
} }
rtmp_send_message(rsession, RTMP_DEFAULT_STREAM_AUDIO, ts, RTMP_TYPE_AUDIO, rsession->media_streamid, buf, frame->datalen + 1, 0); rtmp_send_message(rsession, RTMP_DEFAULT_STREAM_AUDIO, ts, RTMP_TYPE_AUDIO, rsession->media_streamid, buf, frame->datalen + 1, 0);
success:
switch_thread_rwlock_unlock(rsession->rwlock);
return SWITCH_STATUS_SUCCESS; return SWITCH_STATUS_SUCCESS;
error:
switch_thread_rwlock_unlock(rsession->rwlock);
return SWITCH_STATUS_FALSE;
} }
@ -852,6 +858,14 @@ switch_status_t rtmp_session_destroy(rtmp_session_t **rsession)
switch_yield(500000); switch_yield(500000);
}*/ }*/
switch_mutex_lock((*rsession)->profile->mutex);
if ( (*rsession)->profile->calls < 1 ) {
(*rsession)->profile->calls = 0;
} else {
(*rsession)->profile->calls--;
}
switch_mutex_unlock((*rsession)->profile->mutex);
switch_thread_rwlock_wrlock((*rsession)->rwlock); switch_thread_rwlock_wrlock((*rsession)->rwlock);
switch_thread_rwlock_unlock((*rsession)->rwlock); switch_thread_rwlock_unlock((*rsession)->rwlock);