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;
}

在DoPlay函数时，inSession参数已经赋值，所以后面就没有再从sSessionMap中增加引用。

如此，在客户端流程中，一共对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的引用

再来修改RemoveOutput函数

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