Darwin做直播时对ReflectorSession引用数的控制

xiaoxiao2021-02-28  138

在之前的博客中,我们提到了如何用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函数中的 

[cpp]  view plain  copy   if (theSessionRef == NULL)   {      ... //创建ReflectorSession   }  

在此函数中,变量isPush为true,所以不会引用到

[cpp]  view plain  copy   if (!isPush)   {   OSRef* debug = sSessionMap->Resolve(inPath);       Assert(debug == theSession->GetRef());   }   所以,对ReflectorSession的引用在执行第一次DoSetup推送的时候为  +0 .

并且此时,还会设置推送端RTPSession的sClientBroadcastSessionAttr属性

[cpp]  view plain  copy   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,所以调用到过程

[cpp]  view plain  copy   // 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流程时

[cpp]  view plain  copy   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,直接进入

[cpp]  view plain  copy   else   {   theSession = (*theOutput)->GetReflectorSession();       if (theSession == NULL)           return QTSS_RequestFailed;     }   这样,并不会增加对ReflectorSession的引用,对后面客户端的第N次DoSetup流程调用,都不会增加对ReflectorSession的引用。

当客户端进入DoPlay流程时,由于在DoPlay之前调用的ProcessRTSPRequest函数中

[cpp]  view plain  copy   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函数

[cpp]  view plain  copy   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       { ... }   }  

[cpp]  view plain  copy   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函数的倒数第二行加入一句

[cpp]  view plain  copy   sSessionMap->Release(theSession->GetRef());//减少DoDescribe对ReflectorSession的引用  

再来修改RemoveOutput函数

[cpp]  view plain  copy       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中被释放,推送端释放机制未改变,这样就完美解决了问题。
转载请注明原文地址: https://www.6miu.com/read-17357.html

最新回复(0)