Darwin做直播时对ReflectorSession引用数的控制
在之前的博客中，我们提到了如何用Darwin&live555实现直播过程，那么更进一步，当直播结束时，我们需要关闭所有正在收看的客户端，并且delete转发会话ReflectorSession，这样才能够在下一次再有同样名称的流推送进来时，创建新的转发Session。
我们下面所做的修改都是基于Darwin 6.0.3进行，具体下载可到https://github.com/xiejiashu/EasyDarwin获取。我们首先来对原有的ReflectorSession控制进行分析：
首先对于推送端，主要经历DoAnnounce、DoSetup、DoPlay、ProcessRTPData四个流程，在DoAnnounce时，QTSSReflectorModule并不创建ReflectorSession，所以此步跳过；
在第一次DoSetup推送流程时，转发会话ReflectorSession还没有创建，所以必然会进入FindOrCreateSession函数中的
if (theSessionRef == NULL)
{
... //创建ReflectorSession
}
在此函数中，变量isPush为true，所以不会引用到
if (!isPush)
{ OSRef* debug = sSessionMap->Resolve(inPath);
Assert(debug == theSession->GetRef());
}所以，对ReflectorSession的引用在执行第一次DoSetup推送的时候为
+0.
并且此时，还会设置推送端RTPSession的sClientBroadcastSessionAttr属性
theErr = QTSS_SetValue(inParams->inClientSession, sClientBroadcastSessionAttr, 0, &theSession, sizeof(theSession));
当第二次进入DoSetup推送流程时（如果有多个Track），同样进入到FindOrCreateSession时，ReflectorSession已经创建，所以直接 OSRef* theSessionRef = sSessionMap->Resolve(inPath);调用后，直接获取到第一次创建的ReflectorSession，引用数+1.
第N次调用DoSetup（有时候会有多个Track推送，如3D视频等），可以推算得到，引用数为 +(N-1).
在进入到DoPlay(QTSS_StandardRTSP_Params* inParams, ReflectorSession* inSession) 推送流程时，参数inSession为NULL，所以调用到过程
// do all above so we can add the session to the map with Resolve here.
// we must only do this once.
OSRef* debug = sSessionMap->Resolve(&thePathPtr);
if (debug != inSession->GetRef())
{
return QTSSModuleUtils::SendErrorResponse(inParams->inRTSPRequest, qtssClientBadRequest, 0);
}由上过程得知，DoPlay为ReflectorSession增加了一次引用，+1.
ProcessRTPData()过程并未进行sSessionMap的Resolve过程，所以此函数也可以跳过。如此一来，可以统计得到，推送端全程对ReflectorSession的引用数目为：N-1 + 1 =N（N为媒体的Track数目，对应到ReflectorSession中，也就是ReflectorStream的数量）。
再来看客户端，客户端经历DoDescribe、DoSetup、DoPlay三个流程。
客户端进入DoDescribe流程时，假设ReflectorSession已经由推送端建立，如此一来，同样是在FindOrCreateSession中，增加了一次对ReflectorSession的引用，+1.
当客户端第一次进入到DoSetup流程时
RTPSessionOutput** theOutput = NULL;
theErr = QTSS_GetValuePtr(inParams->inClientSession, sOutputAttr, 0, (void**)(void*)&theOutput, &theLen);
if (theLen != sizeof(RTPSessionOutput*))//第一次还未设置<span style="font-family:Arial,Helvetica,sans-serif">sOutputAttr属性，所以theLen为0</span>
{
if (theErr != QTSS_NoErr && !isPush)
{
// 客户端第一次进行DoSetup处理时，进入到这里
// Do the standard ReflectorSession setup, create an RTPSessionOutput
theSession = DoSessionSetup(inParams, qtssRTSPReqFilePathTrunc);
if (theSession == NULL)
return QTSS_RequestFailed;
RTPSessionOutput* theNewOutput = NEW RTPSessionOutput(inParams->inClientSession, theSession, sServerPrefs, sStreamCookieAttr );
theSession->AddOutput(theNewOutput,true);
(void)QTSS_SetValue(inParams->inClientSession, sOutputAttr, 0, &theNewOutput, sizeof(theNewOutput));//设置<span style="font-family:Arial,Helvetica,sans-serif">sOutputAttr属性</span>
}
else
{ ... 推送端处理 ... }
}同样，调用DoSessionSetup和FindOrCreateSession，会增加一次对ReflectorSession的引用，+1.
当客户端第二次进入到DoSetup流程时，QTSS_GetValuePtr(inParams->inClientSession, sOutputAttr, 0, (void**)(void*)&theOutput, &theLen)函数已经能够获取到theOutput，直接进入
else
{ theSession = (*theOutput)->GetReflectorSession();
if (theSession == NULL)
return QTSS_RequestFailed;
}这样，并不会增加对ReflectorSession的引用，对后面客户端的第N次DoSetup流程调用，都不会增加对ReflectorSession的引用。
当客户端进入DoPlay流程时，由于在DoPlay之前调用的ProcessRTSPRequest函数中
QTSS_Error ProcessRTSPRequest(QTSS_StandardRTSP_Params* inParams)
{
...
RTPSessionOutput** theOutput = NULL;
QTSS_Error theErr = QTSS_GetValuePtr(inParams->inClientSession, sOutputAttr, 0, (void**)(void*)&theOutput, &theLen);
if ((theErr != QTSS_NoErr) || (theLen != sizeof(RTPSessionOutput*))) // a broadcaster push session
{ if (*theMethod == qtssPlayMethod || *theMethod == qtssRecordMethod)
return DoPlay(inParams, NULL);
else
return QTSS_RequestFailed;
}
switch (*theMethod)
{
case qtssPlayMethod:
return DoPlay(inParams, (*theOutput)->GetReflectorSession());//此时已经赋值ReflectorSession
...
}
return QTSS_NoErr;
}
如此，在客户端流程中，一共对ReflectorSession引用了+2次，一次是在DoDescribe，一次是在DoSetup.
我们再分别看推送端和客户端离线时，对ReflectorSession的处理：
推送端，在我们之前的博客http://blog.csdn.net/xiejiashu/article/details/8065717，已经修改过，也就是，当推送端断开时，会调用到QTSSReflectorModule中的DestroySession函数
QTSS_Error DestroySession(QTSS_ClientSessionClosing_Params* inParams)
{
RTPSessionOutput** theOutput = NULL;
ReflectorOutput* outputPtr = NULL;
ReflectorSession* theSession = NULL;
OSMutexLocker locker (sSessionMap->GetMutex());
UInt32 theLen = sizeof(theSession);
QTSS_Error theErr = QTSS_GetValue(inParams->inClientSession, sClientBroadcastSessionAttr, 0, &theSession, &theLen);
if (theSession != NULL) // it is a broadcaster session
{
ReflectorSession* deletedSession = NULL;
theErr = QTSS_SetValue(inParams->inClientSession, sClientBroadcastSessionAttr, 0, &deletedSession, sizeof(deletedSession)); ...
RemoveOutput(NULL, theSession, killClients);
}
else
{ ... }
}
void RemoveOutput(ReflectorOutput* inOutput, ReflectorSession* inSession, Bool16 killClients)
{
...
if (inSession != NULL)
{
...
OSRef* theSessionRef = inSession->GetRef();
if (theSessionRef != NULL)
{
for (UInt32 x = 0; x < inSession->GetNumStreams(); x++)
{
if (inSession->GetStreamByIndex(x) == NULL)
continue;
Assert(theSessionRef->GetRefCount() > 0) //this shouldn't happen.
if (theSessionRef->GetRefCount() > 0)
sSessionMap->Release(theSessionRef); // 一路Track释放一次引用
}
if (theSessionRef->GetRefCount() == 0)
{
//当ReflectorSession无引用时，进行Delete
sSessionMap->UnRegister(theSessionRef);
delete inSession;
}
}
}
delete inOutput;
}如果单就推送端而言，上面的方法足够了，因为我们在上面的分析中，得知，DoAnnounce、DoSetup、DoPlay、ProcessRTPData总共对ReflectorSession引用次数就是媒体Track的数目（ReflectorSession->GetNumStreams()）。那么如果客户端也同样调用到RemoveOutput中来时，一路客户端是固定的2次引用ReflectorSession，只有当Track数目为2时（常见的音视频同时存在），正好达到收支平衡，如果只有视频进行推送的情况下，一路客户端会造成一次引用的堆积，做过测试的就会发现，每一次播放停止，都会对ReflectorSession的总引用数累积1个。如果有3路或者3路以上Track时，又会造成ReflectorSession被过早收回，当还有客户端在看的情况下，被动的关闭，导致错误的发生，那么我们如何来避免呢？
首先，我们对QTSSReflectorModule的DoDescribe函数进行修改，因为我们发现，当客户端进行DoDescribe创建ReflectorSession时，RTPSession并没有任何属性设置了ReflectorSession的指针，也就是说，客户端如果只获取一下媒体的sdp信息就回去了，ReflectorSession的引用岂不是一直都被白白占用着，如此，我们可以在DoDescribe函数的倒数第二行加入一句
sSessionMap->Release(theSession->GetRef());//减少DoDescribe对ReflectorSession的引用
OSRef* theSessionRef = inSession->GetRef();
if (theSessionRef != NULL)
{
if(inOutput != NULL) //如果是客户端的话，减少一次ReflectorSession的引用
{
if (theSessionRef->GetRefCount() > 0)
sSessionMap->Release(theSessionRef);
}
else // 如果是推送端的话，减少Track数目个引用
{
for (UInt32 x = 0; x < inSession->GetNumStreams(); x++)
{
if (inSession->GetStreamByIndex(x) == NULL)
continue;
Assert(theSessionRef->GetRefCount() > 0) //this shouldn't happen.
if (theSessionRef->GetRefCount() > 0)
sSessionMap->Release(theSessionRef);
}
}
if (theSessionRef->GetRefCount() == 0)
{
sSessionMap->UnRegister(theSessionRef);
delete inSession;
}
}如此一来，客户端在DoDescribe时不占用ReflectorSession引用数，DoSetup引用数在 RemoveOutput中被释放，推送端释放机制未改变，这样就完美解决了问题。
------------------------------------------------------------
本文转自www.easydarwin.org，更多开源流媒体解决方案，请关注我们的微信：EasyDarwin 
