EasyRTMP实现RTMP异步直播推送之环形缓冲区设计

本文转自EasyDarwin团队kim的博客：http://blog.csdn.net/jinlong0603

EasyRTMP的推送缓冲区设计

EasyRTMP内部也同样采用的环形缓冲的设计方法，将音视频数据都同时存入缓冲区，再由发送者从缓冲区中获取数据进行发送，这样就形成了一个异步、生产者、消费者的过程，上层调用者只需要将采集、编码后的音视频Frame数据Push到SDK的缓冲区中，即可返回继续进行上层逻辑操作，SDK内部的发送线程则从缓冲区中不断获取音视频数据推送到流媒体服务器；

EasyRTMP缓冲区

缓冲区丢帧策略

有缓冲区，势必会出现缓冲区满的情况，当这个时候，就需要合理的丢帧策略了，我们这里采用的非关键帧丢帧策略跟《EasyDarwin开源流媒体服务器低延时直播之转发缓存跟进算法》博客中的原理是一致的，当缓冲区新增一帧进入缓冲区时，此时缓冲区已经满了，那么我们就要从缓冲区的读指针部分开始丢弃非关键帧P帧，直到遇到I帧就停止丢帧，再将新来的数据帧增加进入缓冲区，这样既可以保证一个较合理的直播推送缓冲区，确保推送延时可控，又能够保证不会出现花屏的问题；

EasyRTMP推送缓冲区实现

int     SSQ_Init(SS_QUEUE_OBJ_T *pObj, unsigned int sharememory, unsigned int channelid, wchar_t *sharename, unsigned int bufsize, unsigned int prerecordsecs, unsigned int createsharememory)
{
wchar_t wszHeaderName[36] = {0,};
wchar_t wszFramelistName[36] = {0,};
wchar_t wszDataName[36] = {0,};
if (NULL==pObj)                                         return -1;
if (createsharememory==0x01 && bufsize<1)               return -1;
if ( (sharememory==0x01) && (NULL==sharename || (0==wcscmp(sharename, TEXT("\0")))) )   return -1;

memset(pObj, 0x00, sizeof(SS_QUEUE_OBJ_T));
pObj->channelid = channelid;
pObj->shareinfo.id = channelid;
wcscpy(pObj->shareinfo.name, sharename);

wchar_t wszMutexName[36] = {0,};
wsprintf(wszMutexName, TEXT("%s%d_mutex"), sharename, channelid);
pObj->hMutex = OpenMutex(NULL, FALSE, wszMutexName);
if (NULL == pObj->hMutex)
{
pObj->hMutex = CreateMutex(NULL, FALSE, wszMutexName);
if (NULL == pObj->hMutex)                           return -1;
}

//Create Header map

#ifdef _WIN32
if (sharememory == 0x01)
{
wsprintf(wszHeaderName, TEXT("%s%d_h"), sharename, channelid);
pObj->hSSHeader = OpenFileMapping(FILE_MAP_ALL_ACCESS, FALSE, wszHeaderName);
if (NULL==pObj->hSSHeader && createsharememory==0x01)
{
pObj->hSSHeader = CreateFileMapping(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE|SEC_COMMIT, 0, sizeof(SS_HEADER_T), wszHeaderName);
if (NULL==pObj->hSSHeader || pObj->hSSHeader==INVALID_HANDLE_VALUE)
{
return -1;
}
}
pObj->pQueHeader = (SS_HEADER_T*)MapViewOfFile(pObj->hSSHeader, FILE_MAP_READ|FILE_MAP_WRITE, 0, 0, 0);
if (createsharememory==0x01)
{
if (pObj->pQueHeader->bufsize < 1)
{
memset(pObj->pQueHeader, 0x00, sizeof(SS_HEADER_T));
pObj->pQueHeader->bufsize = bufsize;
}
}
else if (NULL==pObj->pQueHeader)
{
return -1;
}
else
{
bufsize = pObj->pQueHeader->bufsize;
}
}
else
{
pObj->pQueHeader = new SS_HEADER_T;
memset(pObj->pQueHeader, 0x00, sizeof(SS_HEADER_T));
}

//==========================================
//Create frame list map
if (prerecordsecs > 0)
{
wsprintf(wszFramelistName, TEXT("%s%d_f"), sharename, channelid);
unsigned int nFramelistNum = prerecordsecs * 30;    //每秒30帧
unsigned int nFrameQueSize = nFramelistNum*sizeof(FRAMEINFO_LIST_T);

if (sharememory == 0x01)
{
pObj->hSSFrameList = OpenFileMapping(FILE_MAP_ALL_ACCESS, FALSE, wszFramelistName);
if (NULL==pObj->hSSFrameList && createsharememory==0x01)
{
pObj->hSSFrameList = CreateFileMapping(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE|SEC_COMMIT, 0, nFrameQueSize, wszFramelistName);
if (NULL==pObj->hSSFrameList || pObj->hSSFrameList==INVALID_HANDLE_VALUE)
{
return -1;
}
}
pObj->pFrameinfoList = (FRAMEINFO_LIST_T*)MapViewOfFile(pObj->hSSFrameList, FILE_MAP_READ|FILE_MAP_WRITE, 0, 0, 0);
if (createsharememory==0x01)
{
memset(pObj->pFrameinfoList, 0x00, nFrameQueSize);
pObj->pQueHeader->framelistNum = nFramelistNum;
}
else if (NULL==pObj->hSSFrameList)
{
return -1;
}
}
else
{
pObj->pFrameinfoList = new FRAMEINFO_LIST_T[nFramelistNum];
memset(&pObj->pFrameinfoList[0], 0x00, sizeof(FRAMEINFO_LIST_T)*nFramelistNum);
pObj->pQueHeader->framelistNum = nFramelistNum;
}
}

//Create data map
if (sharememory == 0x01)
{
wsprintf(wszDataName, TEXT("%s%d_b"), sharename, channelid);
pObj->hSSData   = OpenFileMapping(FILE_MAP_ALL_ACCESS, FALSE, wszDataName);
if (NULL==pObj->hSSData && createsharememory==0x01)
{
pObj->hSSData = CreateFileMapping(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE|SEC_COMMIT, 0, bufsize, wszDataName);
}
if (NULL == pObj->hSSData || pObj->hSSData==INVALID_HANDLE_VALUE)
{
return -1;
}
pObj->pQueData = (char*)MapViewOfFile(pObj->hSSData, FILE_MAP_READ|FILE_MAP_WRITE, 0, 0, 0);
}
else
{
pObj->pQueData = new char [bufsize];
pObj->pQueHeader->bufsize = bufsize;
}
if (createsharememory==0x01)
{
//memset(pQueHeader, 0x00, sizeof(SS_HEADER_T));
memset(pObj->pQueData, 0x00, bufsize);
}
#else
int ret = shm_create((SYNC_VID_SHM_KEY<<8)|channelid, &pObj->shmHdrid, sizeof(SS_HEADER_T), (char**)&pObj->pQueHeader);
if (ret < 0)
{
return -1;
}
SSQ_TRACE("[%d]pQueHeader: %d\n", (SYNC_VID_SHM_KEY<<8)|channelid, pObj->shmHdrid);
ret = shm_create((SYNC_VID_SHM_KEY<<16)|channelid, &pObj->shmDatid, bufsize, (char**)&pObj->pQueData);
if (ret < 0)
{
shm_delete(&pObj->shmHdrid, (char*)pObj->pQueHeader);
return -1;
}
pObj->pQueHeader->bufsize = bufsize;
SSQ_TRACE("[%d]pQueData: %d\n", (SYNC_VID_SHM_KEY<<16)|channelid, pObj->shmDatid);
#endif

return 0;
}
int     SSQ_Deinit(SS_QUEUE_OBJ_T *pObj)
{
if (NULL == pObj)       return -1;

#ifdef _WIN32

if (NULL != pObj->hMutex)
{
CloseHandle(pObj->hMutex);
pObj->hMutex = NULL;
}

if (NULL != pObj->hSSHeader)
{
if (! UnmapViewOfFile(pObj->pQueHeader))
{
}
pObj->pQueHeader = NULL;
CloseHandle(pObj->hSSHeader);
pObj->hSSHeader = NULL;
}
if (NULL != pObj->pQueHeader)
{
delete []pObj->pQueHeader;
pObj->pQueHeader = NULL;
}

if (NULL != pObj->hSSData)
{
if (! UnmapViewOfFile(pObj->pQueData))
{
}
pObj->pQueData = NULL;
CloseHandle(pObj->hSSData);
pObj->hSSData = NULL;
}
if (NULL != pObj->pQueData)
{
delete []pObj->pQueData;
pObj->pQueData = NULL;
}
if (NULL != pObj->hSSFrameList)
{
if (! UnmapViewOfFile(pObj->pFrameinfoList))
{
}
pObj->pFrameinfoList = NULL;
CloseHandle(pObj->hSSFrameList);
pObj->hSSFrameList = NULL;
}
if (NULL != pObj->pFrameinfoList)
{
delete []pObj->pFrameinfoList;
pObj->pFrameinfoList = NULL;
}

#else
if (pObj->shmHdrid>0 && pObj->pQueHeader!=NULL)
{
shm_delete(&pObj->shmHdrid, (char*)pObj->pQueHeader);
}
if (pObj->shmDatid>0 && pObj->pQueData!=NULL)
{
shm_delete(&pObj->shmDatid, (char*)pObj->pQueData);
pObj->pQueData = NULL;
}
#endif
return 0;
}

int     SSQ_SetClearFlag(SS_QUEUE_OBJ_T *pObj, unsigned int _flag)
{
if (NULL == pObj)       return -1;
if (NULL==pObj->pQueData)       return -1;

pObj->pQueHeader->clear_flag = _flag;
return 0;
}
int     SSQ_Clear(SS_QUEUE_OBJ_T *pObj)
{
if (NULL == pObj)       return -1;
if (NULL==pObj->pQueData)       return -1;

//WaitForSingleObject(pObj->hMutex, INFINITE);
//Lock();
pObj->pQueHeader->writepos = 0;
pObj->pQueHeader->readpos  = 0;
pObj->pQueHeader->totalsize= 0;

pObj->pQueHeader->videoframes = 0;  //视频帧数

pObj->pQueHeader->maxframeno = 0;
pObj->pQueHeader->frameno = 0;

memset(pObj->pQueData, 0x00, pObj->pQueHeader->bufsize);

//ReleaseMutex(pObj->hMutex);

//Unlock();
return 0;
}

int SSQ_AddFrameInfo(SS_QUEUE_OBJ_T *pObj, unsigned int _pos, MEDIA_FRAME_INFO *frameinfo)
{
if (NULL == pObj)                   return -1;
if (NULL == pObj->pQueHeader)       return -1;
if (NULL == pObj->pFrameinfoList)   return -1;

if ( pObj->pQueHeader->frameno+1 > pObj->pQueHeader->framelistNum)
{
memmove((char*)pObj->pFrameinfoList, (char*)pObj->pFrameinfoList+sizeof(FRAMEINFO_LIST_T), sizeof(FRAMEINFO_LIST_T)*pObj->pQueHeader->framelistNum-1);
pObj->pQueHeader->frameno --;
pObj->pQueHeader->maxframeno = pObj->pQueHeader->frameno+1;
}

pObj->pFrameinfoList[pObj->pQueHeader->frameno].pos = pObj->pQueHeader->writepos;
#ifdef _DEBUG1
static int nTimestamp = 0;
pObj->pQueHeader->pFrameinfoList[pObj->pQueHeader->frameno].timestamp = ++nTimestamp;
#else
pObj->pFrameinfoList[pObj->pQueHeader->frameno].timestamp_sec = frameinfo->timestamp_sec;
pObj->pFrameinfoList[pObj->pQueHeader->frameno].rtp_timestamp = frameinfo->timestamp_sec*1000+frameinfo->timestamp_usec/1000;
#endif
//SSQ_TRACE("帧号: %d\n", pObj->pQueHeader->frameno);
pObj->pQueHeader->frameno ++;
pObj->pQueHeader->maxframeno = pObj->pQueHeader->frameno;

/*
if ( pObj->pQueHeader->frameno >= pObj->pQueHeader->framelistNum)
{
SSQ_TRACE("max frame: %d\n", pObj->pQueHeader->maxframeno);
pObj->pQueHeader->frameno = 0;
}
if (pObj->pQueHeader->frameno > pObj->pQueHeader->maxframeno)
{
pObj->pQueHeader->maxframeno = pObj->pQueHeader->frameno;
}
*/
return 0;
}

int     SSQ_AddData(SS_QUEUE_OBJ_T *pObj, unsigned int channelid, unsigned int mediatype, MEDIA_FRAME_INFO *frameinfo, char *pbuf)
{
int ret = 0;
SS_BUF_T    bufNode;

if (NULL==pObj || NULL==frameinfo || NULL==pbuf)        return -1;
if (NULL == pObj->pQueData)                 return -1;
if (NULL == pObj->pQueHeader)               return -1;
if (frameinfo->length < 1)
{
#ifdef _DEBUG
SSQ_TRACE("frame length < 1: %d\n", frameinfo->length);
#endif
return -1;
}

#ifdef _DEBUG1
SSQ_TRACE("frame: %02x%02x%02x%02x%02x\n", (unsigned char)pbuf[0],
(unsigned char)pbuf[1],
(unsigned char)pbuf[2],
(unsigned char)pbuf[3],
(unsigned char)pbuf[4]);
#endif

if (frameinfo->length > pObj->pQueHeader->bufsize)
{
SSQ_TRACE("Buffer too low.. Current Frame Size: %d\tBuffer Size: %d\n", frameinfo->length, pObj->pQueHeader->bufsize);
return -1;
}

WaitForSingleObject(pObj->hMutex, INFINITE);        //Lock

if (pObj->pQueHeader->clear_flag == 0x01)
{
//SSQ_TRACE("Received clear signal... Clear Buffer..\n");
SSQ_TRACE("clear cache..  WritePos: %d\n", pObj->pQueHeader->writepos);
SSQ_Clear(pObj);
pObj->pQueHeader->clear_flag = 0x00;
}

if (pObj->pQueHeader->writepos == pObj->pQueHeader->readpos)
{
if (pObj->pQueHeader->totalsize < 0 || pObj->pQueHeader->videoframes < 0)
{
SSQ_TRACE("write speed==read speed...  data length exception: %d    video frames:%d\n", pObj->pQueHeader->totalsize, pObj->pQueHeader->videoframes);
}
//Clear();
}

if (sizeof(SS_BUF_T) + frameinfo->length + pObj->pQueHeader->totalsize > pObj->pQueHeader->bufsize)
{
SSQ_TRACE("beyond ssqueue length.. len:%d\ttotalsize:%d\tbufsize:%d  cache frames:%d\n", frameinfo->length, pObj->pQueHeader->totalsize, pObj->pQueHeader->bufsize, pObj->pQueHeader->videoframes);
ReleaseMutex(pObj->hMutex);
pObj->pQueHeader->isfull = 0x01;
return -1;
}

pObj->pQueHeader->isfull = 0x00;

memset(&bufNode, 0x00, sizeof(SS_BUF_T));
memcpy(&bufNode.frameinfo, frameinfo, sizeof(MEDIA_FRAME_INFO));
bufNode.channelid = channelid;//++m_FrameTally;
bufNode.mediatype = mediatype;
bufNode.flag    =   BUF_QUE_FLAG;

//_TRACE("WritePos: %d    totalSize: %d\n", pQueHeader->writepos, pQueHeader->totalsize);

//Lock();
if ((pObj->pQueHeader->writepos + sizeof(SS_BUF_T) + frameinfo->length) <= pObj->pQueHeader->bufsize)
{
//copy to queue
if (mediatype==MEDIA_TYPE_VIDEO)    SSQ_AddFrameInfo(pObj, pObj->pQueHeader->writepos, frameinfo);

unsigned int nAdd = pObj->pQueHeader->writepos;
memcpy(pObj->pQueData+nAdd, &bufNode, sizeof(SS_BUF_T));
nAdd += sizeof(SS_BUF_T);

memcpy(pObj->pQueData+nAdd, pbuf, frameinfo->length);
nAdd += frameinfo->length;
pObj->pQueHeader->writepos = nAdd;
pObj->pQueHeader->totalsize+= sizeof(SS_BUF_T);
pObj->pQueHeader->totalsize+= frameinfo->length;

if (mediatype==MEDIA_TYPE_VIDEO)    pObj->pQueHeader->videoframes ++;

//_TRACE("顺序新增..  writepos: %d   /   %d\n", pQueHeader->writepos, pQueHeader->size);
}
else if (pObj->pQueHeader->writepos == pObj->pQueHeader->bufsize)   //从头开始
{
//记录帧位置
if (mediatype==MEDIA_TYPE_VIDEO)    SSQ_AddFrameInfo(pObj, 0, frameinfo);

memcpy(pObj->pQueData, &bufNode, sizeof(SS_BUF_T));
pObj->pQueHeader->writepos = sizeof(SS_BUF_T);

memcpy(pObj->pQueData+pObj->pQueHeader->writepos, pbuf, frameinfo->length);
pObj->pQueHeader->writepos += frameinfo->length;

pObj->pQueHeader->totalsize= sizeof(SS_BUF_T);
pObj->pQueHeader->totalsize+= frameinfo->length;

if (mediatype==MEDIA_TYPE_VIDEO)    pObj->pQueHeader->videoframes ++;

SSQ_TRACE("从头开始写..  writepos: %d\n", pObj->pQueHeader->writepos);
}
//else if (pQueHeader->size - pQueHeader->writepos+pQueHeader->readpos >= (int)(frameinfo->length+sizeof(SS_BUF_T)))    //从尾写到头
//else if (pObj->pQueHeader->bufsize - pObj->pQueHeader->writepos+pObj->pQueHeader->readpos >= (int)(frameinfo->length+sizeof(SS_BUF_T)))   //从尾写到头
else// if (pObj->pQueHeader->bufsize - pObj->pQueHeader->writepos >= (int)(frameinfo->length+sizeof(SS_BUF_T))) //从尾写到头
{
unsigned int remain = pObj->pQueHeader->bufsize - pObj->pQueHeader->writepos;   //剩余空间
if (remain>=sizeof(SS_BUF_T))
{
unsigned int nAdd = pObj->pQueHeader->writepos;

//记录帧位置
if (mediatype==MEDIA_TYPE_VIDEO)    SSQ_AddFrameInfo(pObj, nAdd, frameinfo);

//_TRACE("WritePos111: %d\n", pQueHeader->writepos);
memcpy(pObj->pQueData+nAdd, &bufNode, sizeof(SS_BUF_T));
nAdd += sizeof(SS_BUF_T);
//pQueHeader->totalsize+= sizeof(SS_BUF_T);
//_TRACE("WritePos222: %d\n", pQueHeader->writepos+sizeof(SS_BUF_T));
remain = pObj->pQueHeader->bufsize - nAdd;//pQueHeader->writepos;
//_TRACE("WritePos: %d\n", pQueHeader->writepos+sizeof(SS_BUF_T)+remain);
if (remain>0)
{
memcpy(pObj->pQueData+nAdd, pbuf, remain);
memcpy(pObj->pQueData, pbuf+remain, frameinfo->length-remain);
nAdd = frameinfo->length-remain;

pObj->pQueHeader->writepos = nAdd;
pObj->pQueHeader->totalsize+= sizeof(SS_BUF_T);
pObj->pQueHeader->totalsize+= frameinfo->length;

if (pObj->pQueHeader->totalsize>pObj->pQueHeader->bufsize)
{
//原因:  rtsp server已停止读该队列(当前没有客户端访问)
SSQ_TRACE("[RTSP Server已停止读取该队列]错误 %d > %d    frameinfo->length:%d...\n", pObj->pQueHeader->totalsize, pObj->pQueHeader->bufsize, frameinfo->length);
//SSQ_Clear(pObj);
}
else
{
if (mediatype==MEDIA_TYPE_VIDEO)    pObj->pQueHeader->videoframes ++;

//SSQ_TRACE("111Header及部分帧数据位于缓存尾, 部分帧数据位于缓存首..  remain: %d   writepos: %d  totalsize: %d\n", remain, pObj->pQueHeader->writepos, pObj->pQueHeader->totalsize);
}
}
else if (remain==0)
{
memcpy(pObj->pQueData, pbuf, frameinfo->length);
nAdd = frameinfo->length;
pObj->pQueHeader->writepos = nAdd;
pObj->pQueHeader->totalsize+= sizeof(SS_BUF_T);
pObj->pQueHeader->totalsize+= frameinfo->length;

if (mediatype==MEDIA_TYPE_VIDEO)    pObj->pQueHeader->videoframes ++;

if (pObj->pQueHeader->totalsize>pObj->pQueHeader->bufsize)
{
SSQ_TRACE("错误222 %d > %d    frameinfo->length:%d...\n", pObj->pQueHeader->totalsize, pObj->pQueHeader->bufsize, frameinfo->length);
//SSQ_Clear(pObj);
}
else
{
SSQ_TRACE("22222Header位于缓存尾,帧数据位于缓存首..  writepos: %d\n", pObj->pQueHeader->writepos);
}
}
else
{
SSQ_TRACE("错误...  剩余空间小于0: %d\n", remain);
ret = -1;
}
}
else if (remain>0)
{
char *tmpbuf = (char *)&bufNode;

unsigned int nAdd = pObj->pQueHeader->writepos;

//记录帧位置
if (mediatype==MEDIA_TYPE_VIDEO)    SSQ_AddFrameInfo(pObj, nAdd, frameinfo);

//SSQ_TRACE("ADD DATA...%d\n", nAdd);
memcpy(pObj->pQueData+nAdd, tmpbuf, remain);
//SSQ_TRACE("ADD DATA222...%d\n", sizeof(SS_BUF_T)-remain);
memcpy(pObj->pQueData, tmpbuf+remain, sizeof(SS_BUF_T)-remain);
nAdd = sizeof(SS_BUF_T)-remain;
memcpy(pObj->pQueData+nAdd, pbuf, frameinfo->length);

nAdd += frameinfo->length;
pObj->pQueHeader->writepos  = nAdd;
pObj->pQueHeader->totalsize+= sizeof(SS_BUF_T);
pObj->pQueHeader->totalsize+= frameinfo->length;

if (pObj->pQueHeader->totalsize>pObj->pQueHeader->bufsize)
{
SSQ_TRACE("错误333 %d > %d    frameinfo->length:%d...\n", pObj->pQueHeader->totalsize, pObj->pQueHeader->bufsize, frameinfo->length);
//SSQ_Clear(pObj);
}

if (mediatype==MEDIA_TYPE_VIDEO)    pObj->pQueHeader->videoframes ++;
}
else
{
ret = -1;
SSQ_TRACE("ERROR...\n");
}
}
//else
{
//else if (pObj->pQueHeader->bufsize - pObj->pQueHeader->writepos+pObj->pQueHeader->readpos >= (int)(frameinfo->length+sizeof(SS_BUF_T)))   //从尾写到头
//SSQ_TRACE("写尾错误..  未处理.  写位置:%d / pObj->pQueHeader->bufsize  读位置:%d\n", pObj->pQueHeader->writepos, pObj->pQueHeader->bufsize, pObj->pQueHeader->readpos);
}
//Unlock();

ReleaseMutex(pObj->hMutex);

//SSQ_TRACE("writepos: %d\ttotalsize: %d\n", pObj->pQueHeader->writepos, pObj->pQueHeader->totalsize);

#ifdef _DEBUG1
if (mediatype==MEDIA_TYPE_VIDEO)
{
SSQ_TRACE("==========================\n");
for (int i=0; i<pObj->pQueHeader->maxframeno; i++)
{
SSQ_TRACE("[%d] times: %d   pos: %d   %02X%02X%02X%02X%02X\n", i, pObj->pQueHeader->pFrameinfoList[i].timestamp, pObj->pQueHeader->pFrameinfoList[i].pos,
(unsigned char)pObj->pQueData[pObj->pQueHeader->pFrameinfoList[i].pos+sizeof(SS_BUF_T)+0], (unsigned char)pObj->pQueData[pObj->pQueHeader->pFrameinfoList[i].pos+sizeof(SS_BUF_T)+1], (unsigned char)pObj->pQueData[pObj->pQueHeader->pFrameinfoList[i].pos+sizeof(SS_BUF_T)+2],
(unsigned char)pObj->pQueData[pObj->pQueHeader->pFrameinfoList[i].pos+sizeof(SS_BUF_T)+3], (unsigned char)pObj->pQueData[pObj->pQueHeader->pFrameinfoList[i].pos+sizeof(SS_BUF_T)+4]);
}
}
#endif

return ret;
}
int     SSQ_GetData(SS_QUEUE_OBJ_T *pObj, unsigned int *channelid, unsigned int *mediatype, MEDIA_FRAME_INFO *frameinfo, char *pbuf)
{
int ret = 0;
unsigned int remain = 0;

if (NULL == pObj)           return -1;

if (NULL == pObj->pQueHeader)   return -1;
if (NULL == pObj->pFrameinfoList)   return -1;

WaitForSingleObject(pObj->hMutex, INFINITE);        //Lock

if (pObj->pQueHeader->totalsize < 0)
{
SSQ_TRACE("pObj->pQueHeader->totalsize<0: %d\n", pObj->pQueHeader->totalsize);
ReleaseMutex(pObj->hMutex);
return -1;
}

if (pObj->pQueHeader->totalsize <= sizeof(SS_BUF_T))
{
ReleaseMutex(pObj->hMutex);
return -1;
}
//_TRACE("读位置: %d\n", pQueHeader->readpos);

//if (NULL != chid)     *chid = m_chid;

//Lock();

#if 0
ret = -1;
for (unsigned int i=0; i<pObj->pQueHeader->maxframeno; i++)
{
if (pObj->pFrameinfoList[i].rtp_timestamp > frameinfo->rtptimestamp)
{
pObj->pQueHeader->readpos = pObj->pFrameinfoList[i].pos;
ret = SSQ_GetDataByPosition(pObj, pObj->pFrameinfoList[i].pos, channelid, mediatype, frameinfo, pbuf);
break;
}
}
#else

if (pObj->pQueHeader->readpos == pObj->pQueHeader->bufsize)
{
SSQ_TRACE("重置读位置[%d / %d]..\n", pObj->pQueHeader->readpos, pObj->pQueHeader->bufsize);
pObj->pQueHeader->readpos = 0;
}

if ((pObj->pQueHeader->readpos + sizeof(SS_BUF_T)) <= pObj->pQueHeader->bufsize)
{
SS_BUF_T *pNode = (SS_BUF_T *)(pObj->pQueData + pObj->pQueHeader->readpos);
//if (pNode->id<1)
if (pNode->flag != BUF_QUE_FLAG)
{
SSQ_TRACE("标志位错误... 缓存视频帧:%d  字节数:%d  清空队列\n", pObj->pQueHeader->videoframes, pObj->pQueHeader->totalsize);
if (pObj->hSSHeader == NULL)    //同一个进程内使用
{
SSQ_Clear(pObj);
}
else
{
pObj->pQueHeader->clear_flag = 0x01;
#ifdef _WIN32
while (pObj->pQueHeader->clear_flag!=0x00)  {Sleep(1);}
#else
while (pObj->pQueHeader->clear_flag!=0x00)  {usleep(100);}
#endif
}
//Unlock();
SSQ_TRACE("111标志位错误... 缓存视频帧:%d  字节数:%d\n", pObj->pQueHeader->videoframes, pObj->pQueHeader->totalsize);
SSQ_TRACE("标志位错误.. 清空队列..  readpos: %d\n", pObj->pQueHeader->readpos);
//pObj->pQueHeader->clear_flag = 0x01;

//_TRACE("标志位错误.. 清空队列完成..\n");

ReleaseMutex(pObj->hMutex);
return -1;
}

if (NULL!=mediatype)        *mediatype = pNode->mediatype;
if (NULL != channelid)      *channelid = pNode->channelid;

memcpy(frameinfo, &pNode->frameinfo, sizeof(MEDIA_FRAME_INFO));
if ( (pObj->pQueHeader->readpos + pNode->frameinfo.length+sizeof(SS_BUF_T)) <= pObj->pQueHeader->bufsize)
{
//从头到尾读

if (pObj->pQueHeader->totalsize < frameinfo->length+sizeof(SS_BUF_T))
{
//数据量不够
SSQ_TRACE("数据量不够... 总字节数[%d]<帧长[%d]. 读位置:%d\n", pObj->pQueHeader->totalsize, frameinfo->length+sizeof(SS_BUF_T), pObj->pQueHeader->readpos);
ReleaseMutex(pObj->hMutex);
return -1;
}

if (frameinfo->length+sizeof(SS_BUF_T) > pObj->pQueHeader->totalsize)
{
SSQ_TRACE("总字节数[%d]<帧长[%d]. 读位置:%d\n", pObj->pQueHeader->totalsize, frameinfo->length+sizeof(SS_BUF_T), pObj->pQueHeader->readpos);
}

//_TRACE("读位置00000000...: %d  /  %d\n", pQueHeader->readpos, pQueHeader->size);
pObj->pQueHeader->readpos += sizeof(SS_BUF_T);

unsigned int total1 = pObj->pQueHeader->totalsize;
pObj->pQueHeader->totalsize -= sizeof(SS_BUF_T);

if (NULL!=pbuf) memcpy(pbuf, pObj->pQueData+pObj->pQueHeader->readpos, frameinfo->length);

//memset(pObj->pQueData+pObj->pQueHeader->readpos, 0x00, frameinfo->length);    //clear

pObj->pQueHeader->readpos += frameinfo->length;
unsigned int total2 = pObj->pQueHeader->totalsize;
pObj->pQueHeader->totalsize -= (frameinfo->length);

if (pObj->pQueHeader->readpos == pObj->pQueHeader->bufsize)
{
pObj->pQueHeader->readpos = 0;
}

if (pObj->pQueHeader->readpos > pObj->pQueHeader->bufsize)
{
SSQ_TRACE("读位置错误11111...: %d  /  %d\n", pObj->pQueHeader->readpos, pObj->pQueHeader->bufsize);
}
if (pObj->pQueHeader->totalsize < 0)
{
SSQ_TRACE("读位置: %d\t写位置: %d  当前帧大小:%d   total1:%d\ttotal2:%d\ttotal3:%d\n", pObj->pQueHeader->readpos, pObj->pQueHeader->writepos, frameinfo->length, total1, total2, pObj->pQueHeader->totalsize);
SSQ_TRACE("总字节数错误: %d\n", pObj->pQueHeader->totalsize);
}
}
else
{
if (pObj->pQueHeader->totalsize < (pNode->frameinfo.length+sizeof(SS_BUF_T)))
{
SSQ_TRACE("总字节数<帧长+sizeof(SS_BUF_T)..\n");
//Unlock();
ReleaseMutex(pObj->hMutex);
return -1;
}

remain = pObj->pQueHeader->bufsize-pObj->pQueHeader->readpos;
if (remain>=sizeof(SS_BUF_T))
{
pObj->pQueHeader->readpos += sizeof(SS_BUF_T);
remain = pObj->pQueHeader->bufsize - pObj->pQueHeader->readpos;
if (remain>0)
{
//SSQ_TRACE("111尾有部分数据... 首有部分数据... remain>0: %d\n", remain);
if (NULL!=pbuf) memcpy(pbuf, pObj->pQueData+pObj->pQueHeader->readpos, remain);
//memset(pObj->pQueData+pObj->pQueHeader->readpos, 0x00, remain);   //clear
if (NULL!=pbuf) memcpy(pbuf+remain, pObj->pQueData, frameinfo->length-remain);
//memset(pObj->pQueData, 0x00, frameinfo->length-remain);   //clear

pObj->pQueHeader->readpos = frameinfo->length-remain;
pObj->pQueHeader->totalsize -= frameinfo->length;
pObj->pQueHeader->totalsize -= sizeof(SS_BUF_T);

if (pObj->pQueHeader->totalsize < 0)
{
SSQ_TRACE("3333pObj->pQueHeader->totalsize<0: %d\n", pObj->pQueHeader->totalsize);
}
}
else
{
if (remain < 0)
{
SSQ_TRACE("位移错误: 剩余字节数<0:%d\n", remain);

}
else
{
SSQ_TRACE("111尾有部分数据... 首有部分数据... remain<=0: %d\n", remain);
if (NULL!=pbuf) memcpy(pbuf, pObj->pQueData, frameinfo->length);
//memset(pObj->pQueData, 0x00, frameinfo->length-remain);   //clear
pObj->pQueHeader->readpos = frameinfo->length;
pObj->pQueHeader->totalsize -= frameinfo->length;
pObj->pQueHeader->totalsize -= sizeof(SS_BUF_T);

if (pObj->pQueHeader->totalsize < 0)
{
SSQ_TRACE("4444pObj->pQueHeader->totalsize<0: %d\n", pObj->pQueHeader->totalsize);
}
}
}

//SSQ_TRACE("remain > sizeof(SS_BUF_T): %d\t\treadpos: %d\n", remain, pObj->pQueHeader->readpos);
}
else
{
//SSQ_TRACE("remain < sizeof(SS_BUF_T): %d\n", remain);

//执行到此处,说明异常

//remain = pObj->pQueHeader->bufsize - pObj->pQueHeader->readpos;
if (remain>0)
{
SSQ_TRACE("222尾有部分数据... 首有部分数据... remain>0: %d\n", remain);
if (NULL!=pbuf) memcpy(pbuf, pObj->pQueData+pObj->pQueHeader->readpos, remain);
//memset(pObj->pQueData+pObj->pQueHeader->readpos, 0x00, remain);   //clear
if (NULL!=pbuf) memcpy(pbuf+remain, pObj->pQueData, frameinfo->length-remain);
//memset(pObj->pQueData, 0x00, frameinfo->length-remain);   //clear

pObj->pQueHeader->readpos = frameinfo->length-remain;
pObj->pQueHeader->totalsize -= frameinfo->length;

if (pObj->pQueHeader->totalsize < 0)
{
SSQ_TRACE("555pObj->pQueHeader->totalsize<0: %d\n", pObj->pQueHeader->totalsize);
}

}
else
{
SSQ_TRACE("222尾有部分数据... 首有部分数据... remain<=0: %d\n", remain);
if (NULL!=pbuf) memcpy(pbuf, pObj->pQueData, frameinfo->length);
//memset(pObj->pQueData, 0x00, frameinfo->length-remain);   //clear
pObj->pQueHeader->readpos = frameinfo->length;
pObj->pQueHeader->totalsize -= frameinfo->length;
if (pObj->pQueHeader->totalsize < 0)
{
SSQ_TRACE("666pObj->pQueHeader->totalsize<0: %d\n", pObj->pQueHeader->totalsize);
}
}
pObj->pQueHeader->totalsize -= sizeof(SS_BUF_T);

if (pObj->pQueHeader->totalsize < 0)
{
SSQ_TRACE("777pObj->pQueHeader->totalsize<0: %d\n", pObj->pQueHeader->totalsize);
}
}

//pObj->pQueHeader->readpos += sizeof(SS_BUF_T);
//pObj->pQueHeader->totalsize -= sizeof(SS_BUF_T);

if (pObj->pQueHeader->readpos>pObj->pQueHeader->bufsize)
{
SSQ_TRACE("读位置错误...: %d  /  %d\n", pObj->pQueHeader->readpos, pObj->pQueHeader->bufsize);
}

/*
if (NULL!=pbuf) memcpy(pbuf, pShareMemoryBuff+pQueHeader->readpos, remain);
memset(pShareMemoryBuff+pQueHeader->readpos, 0x00, remain); //clear
if (NULL!=pbuf) memcpy(pbuf+remain, pShareMemoryBuff, frameinfo->length-remain);
memset(pShareMemoryBuff, 0x00, frameinfo->length-remain);   //clear

pQueHeader->readpos = frameinfo->length-remain;
pQueHeader->totalsize -= (frameinfo->length-remain);
*/
}

if (MEDIA_TYPE_VIDEO==pNode->mediatype) pObj->pQueHeader->videoframes --;

//memset(pNode, 0x00, sizeof(SS_BUF_T));
}
else
{
unsigned int remain = pObj->pQueHeader->bufsize-pObj->pQueHeader->readpos;
SS_BUF_T bufnode;
char *pp = (char *)&bufnode;
memset(&bufnode, 0x00, sizeof(SS_BUF_T));

//SSQ_TRACE("GET DATA...\n");

//SSQ_TRACE("1 REMAIN: %d\n", remain);
if (remain>0)
{
memcpy(pp, pObj->pQueData+pObj->pQueHeader->readpos, remain);
//memset(pObj->pQueData+pObj->pQueHeader->readpos, 0x00, remain);   //clear
//SSQ_TRACE("2 read: %d\n", sizeof(SS_BUF_T)-remain);
memcpy(pp+remain, pObj->pQueData, sizeof(SS_BUF_T)-remain);
//memset(pObj->pQueData, 0x00, sizeof(SS_BUF_T)-remain);    //clear

memcpy(frameinfo, &bufnode.frameinfo, sizeof(MEDIA_FRAME_INFO));
if (NULL != channelid)      *channelid = bufnode.channelid;

//if (bufnode.id<1)
if (bufnode.flag    != BUF_QUE_FLAG)
{
//Unlock();
SSQ_Clear(pObj);
ReleaseMutex(pObj->hMutex);
SSQ_TRACE("SSQ_标志符错误...\n");
return -1;
}

pObj->pQueHeader->readpos = sizeof(SS_BUF_T)-remain;

if (NULL!=mediatype)    *mediatype = bufnode.mediatype;
//SSQ_TRACE("3 frame length: %d\n", bufnode.frameinfo.length);
if (NULL!=pbuf) memcpy(pbuf, pObj->pQueData+pObj->pQueHeader->readpos, bufnode.frameinfo.length);
//memset(pObj->pQueData+pObj->pQueHeader->readpos, 0x00, bufnode.frameinfo.length); //clear
pObj->pQueHeader->readpos += bufnode.frameinfo.length;

pObj->pQueHeader->totalsize -= sizeof(SS_BUF_T);
//pObj->pQueHeader->totalsize -= (frameinfo->length-remain);
pObj->pQueHeader->totalsize -= (frameinfo->length);//20140521

if (pObj->pQueHeader->totalsize < 0)
{
SSQ_TRACE("888pObj->pQueHeader->totalsize<0: %d\n", pObj->pQueHeader->totalsize);
}

//SSQ_TRACE("GET DATA OK..\n");
if (MEDIA_TYPE_VIDEO==bufnode.mediatype)    pObj->pQueHeader->videoframes --;
}
else
{
SSQ_TRACE("异常...   REMAIN <= 0....\n");
}

ret = 1000;
}
#endif

//Unlock();
ReleaseMutex(pObj->hMutex);

return ret;
}

//===========================================
//根据位置获取对应的帧数据
int     SSQ_GetDataByPosition(SS_QUEUE_OBJ_T *pObj, unsigned int position, unsigned int clearflag, unsigned int *channelid, unsigned int *mediatype, MEDIA_FRAME_INFO *frameinfo, char *pbuf)
{
int ret = 0;
unsigned int remain = 0;

if (NULL == pObj)                   return -1;
if (NULL == pObj->pQueHeader)       return -1;
if (NULL == pObj->pFrameinfoList)   return -1;

unsigned int *pOffset = (unsigned int *)&position;
unsigned int totalsize = pObj->pQueHeader->totalsize;
unsigned int *pTotalSize = (unsigned int*)&totalsize;

if (clearflag == 0x01)
{
pOffset    = (unsigned int*)&pObj->pQueHeader->readpos;
pTotalSize = (unsigned int*)&pObj->pQueHeader->totalsize;
}

WaitForSingleObject(pObj->hMutex, INFINITE);        //Lock

if (*pOffset == pObj->pQueHeader->bufsize)
{
SSQ_TRACE("重置读位置[%d / %d]..\n", *pOffset, pObj->pQueHeader->bufsize);
*pOffset = 0;
}

if (clearflag==0x01)
{
if (pObj->pQueHeader->totalsize <= sizeof(SS_BUF_T))
{
ReleaseMutex(pObj->hMutex);
return -1;
}

if (pObj->pQueHeader->readpos == pObj->pQueHeader->bufsize)
{
pObj->pQueHeader->readpos = 0;
}
}

if ((*pOffset + sizeof(SS_BUF_T)) <= pObj->pQueHeader->bufsize)
{
SS_BUF_T *pNode = (SS_BUF_T *)(pObj->pQueData + *pOffset);
if (pNode->flag != BUF_QUE_FLAG)
{
SSQ_TRACE("[SSQ_GetDataByPosition]标志位错误...\n");
if (clearflag == 0x01)
{
if (pObj->hSSHeader==NULL)  //同一进程
{
SSQ_Clear(pObj);
}
else
{
pObj->pQueHeader->clear_flag    =   0x01;
while (pObj->pQueHeader->clear_flag!=0x00)  {Sleep(1);}
}
}
ReleaseMutex(pObj->hMutex);
return -1;
}

if (NULL!=mediatype)        *mediatype = pNode->mediatype;
if (NULL != channelid)      *channelid = pNode->channelid;

memcpy(frameinfo, &pNode->frameinfo, sizeof(MEDIA_FRAME_INFO));
if ( (*pOffset + pNode->frameinfo.length+sizeof(SS_BUF_T)) <= pObj->pQueHeader->bufsize)
{
//从头到尾读
*pOffset += sizeof(SS_BUF_T);

if (NULL!=pbuf) memcpy(pbuf, pObj->pQueData+*pOffset, frameinfo->length);
*pOffset += frameinfo->length;
*pTotalSize -= (frameinfo->length+sizeof(SS_BUF_T));

if (*pOffset == pObj->pQueHeader->bufsize)
{
*pOffset = 0;
}

if (*pOffset > pObj->pQueHeader->bufsize)
{
SSQ_TRACE("[SSQ_GetDataByPosition]读位置错误11111...: %d  /  %d\n", *pOffset, pObj->pQueHeader->bufsize);
}
}
else
{
remain = pObj->pQueHeader->bufsize - *pOffset;
if (remain>=sizeof(SS_BUF_T))
{
*pOffset += sizeof(SS_BUF_T);
remain = pObj->pQueHeader->bufsize - *pOffset;
if (remain>0)
{
if (NULL!=pbuf) memcpy(pbuf, pObj->pQueData + *pOffset, remain);
if (NULL!=pbuf) memcpy(pbuf+remain, pObj->pQueData, frameinfo->length-remain);

*pOffset = frameinfo->length-remain;
*pTotalSize -= (frameinfo->length+sizeof(SS_BUF_T));
}
else
{
if (remain < 0)
{
SSQ_TRACE("[SSQ_GetDataByPosition]位移错误: 剩余字节数<0:%d\n", remain);
}
else
{
SSQ_TRACE("[SSQ_GetDataByPosition]尾有部分数据... 首有部分数据... remain<=0: %d\n", remain);
if (NULL!=pbuf) memcpy(pbuf, pObj->pQueData, frameinfo->length);
//memset(pObj->pQueData, 0x00, frameinfo->length-remain);   //clear
*pOffset = frameinfo->length;
*pTotalSize -= (frameinfo->length+sizeof(SS_BUF_T));
}
}
}
else
{
//SSQ_TRACE("remain < sizeof(SS_BUF_T): %d\n", remain);

//执行到此处,说明异常
SSQ_TRACE("[SSQ_GetDataByPosition]  异常   #########... remain>0: %d\n", remain);
if (remain>0)
{
SSQ_TRACE("[SSQ_GetDataByPosition]222尾有部分数据... 首有部分数据... remain>0: %d\n", remain);
if (NULL!=pbuf) memcpy(pbuf, pObj->pQueData + *pOffset, remain);
if (NULL!=pbuf) memcpy(pbuf+remain, pObj->pQueData, frameinfo->length-remain);

*pOffset = frameinfo->length-remain;
*pTotalSize -= (frameinfo->length);
}
else
{
SSQ_TRACE("[SSQ_GetDataByPosition]222尾有部分数据... 首有部分数据... remain<=0: %d\n", remain);
if (NULL!=pbuf) memcpy(pbuf, pObj->pQueData, frameinfo->length);
//memset(pObj->pQueData, 0x00, frameinfo->length-remain);   //clear
*pOffset = frameinfo->length;
}
}

if (*pOffset > pObj->pQueHeader->bufsize)
{
SSQ_TRACE("[SSQ_GetDataByPosition]读位置错误...: %d  /  %d\n", *pOffset, pObj->pQueHeader->bufsize);
}
}

if (MEDIA_TYPE_VIDEO==pNode->mediatype && clearflag==0x01)  pObj->pQueHeader->videoframes --;
}
else
{
unsigned int remain = pObj->pQueHeader->bufsize - *pOffset;
SS_BUF_T bufnode;
char *pp = (char *)&bufnode;
memset(&bufnode, 0x00, sizeof(SS_BUF_T));

//SSQ_TRACE("GET DATA...\n");

SSQ_TRACE("[SSQ_GetDataByPosition]1 REMAIN: %d\n", remain);
if (remain>0)
{
memcpy(pp, pObj->pQueData + *pOffset, remain);
SSQ_TRACE("[SSQ_GetDataByPosition]2 read: %d\n", sizeof(SS_BUF_T)-remain);
memcpy(pp+remain, pObj->pQueData, sizeof(SS_BUF_T)-remain);

memcpy(frameinfo, &bufnode.frameinfo, sizeof(MEDIA_FRAME_INFO));
if (NULL != channelid)      *channelid = bufnode.channelid;

if (bufnode.flag    != BUF_QUE_FLAG)
{
//Unlock();
SSQ_Clear(pObj);
ReleaseMutex(pObj->hMutex);
SSQ_TRACE("[SSQ_GetDataByPosition]SSQ_标志符错误...\n");
return -1;
}

*pOffset = sizeof(SS_BUF_T)-remain;

if (NULL!=mediatype)    *mediatype = bufnode.mediatype;
SSQ_TRACE("[SSQ_GetDataByPosition]3 frame length: %d\n", bufnode.frameinfo.length);
if (NULL!=pbuf) memcpy(pbuf, pObj->pQueData + *pOffset, bufnode.frameinfo.length);
*pOffset += bufnode.frameinfo.length;
*pTotalSize -= (frameinfo->length+sizeof(SS_BUF_T));

SSQ_TRACE("[SSQ_GetDataByPosition]GET DATA OK..\n");
if (MEDIA_TYPE_VIDEO==bufnode.mediatype && clearflag==0x01) pObj->pQueHeader->videoframes --;
}
else
{
SSQ_TRACE("[SSQ_GetDataByPosition]异常...   REMAIN <= 0....\n");
}

ret = 1000;
}

//Unlock();
ReleaseMutex(pObj->hMutex);

return ret;
}

int     SSQ_TRACE(char* szFormat, ...)
{
#ifdef _DEBUG
char buff[1024] = {0,};
wchar_t wszbuff[1024] = {0,};
va_list args;
va_start(args,szFormat);
_vsnprintf(buff, 1023, szFormat,args);
va_end(args);

MByteToWChar(buff, wszbuff, sizeof(wszbuff)/sizeof(wszbuff[0]));
#ifdef _WIN32
OutputDebugString(wszbuff);
#endif
printf("TRACE: %s", buff);
#endif

return 0;
}

获取更多信息

邮件：support@easydarwin.org

WEB：www.EasyDarwin.org

Copyright © EasyDarwin.org 2012-2016