LiveRTMP之对接海康、大华等IPCamera SDK进行RTMP直播推送

Demo项目介绍

LiveRTMP 是一套自行封装的高效的RTMP推流库，接收H264和AAC数据，内部提供高效的缓存，和时间戳矫正机制。库中自动打包成RTMP流并推送给给RTMP流媒体服务或者CDN。

本文介绍通过调用IP Camera自己的SDK获取到音视频流，然后通过libLiveRTMP推送到RTMP服务器进行直播。这样就可以将普通的不支持RTMP的IP Camera接入RTMP服务器进行直播推送。该Demo程序可以交叉编译内置到IP Camera内部，或者在Windows或者其他系统中执行此Demo拉取IP Camera的数据流再推送到RTMP服务器。

如下Demo里面用的IP Camera的SDK是海芯威视的IP Camera，只是为了举例LiveRTMP接口如何调用。如果想接入其他家的IP Camera（如海康、大华等）可自己将SDK库及调用方式换成相应的版本。

代码解析

下面是IP Camera SDK 数据回调函数的部分，解析回调出来的数据并使用libLiveRTMP库进行RTMP推送。

HI_S32 NETSDK_APICALL OnStreamCallback(HI_U32 u32Handle,
                                HI_U32 u32DataType,
                                HI_U8* pu8Buffer,
                                HI_U32 u32Length,
                                HI_VOID* pUserData
                                )
{
    HI_S_AVFrame* pstruAV = HI_NULL;
	HI_S_SysHeader* pstruSys = HI_NULL;

	if (u32DataType == HI_NET_DEV_AV_DATA)
	{
		pstruAV = (HI_S_AVFrame*)pu8Buffer;

		if (pstruAV->u32AVFrameFlag == HI_NET_DEV_VIDEO_FRAME_FLAG)
		{
			if(g_RtmpHandle == 0 ) return 0;

			if(pstruAV->u32AVFrameLen > 0)
			{
				unsigned char* pbuf = (unsigned char*)(pu8Buffer+sizeof(HI_S_AVFrame));

				EASY_AV_Frame  avFrame;
				memset(&avFrame, 0x00, sizeof(EASY_AV_Frame));
				avFrame.u32AVFrameLen = pstruAV->u32AVFrameLen;
				avFrame.pBuffer = (unsigned char*)pbuf;
				avFrame.u32VFrameType = (pstruAV->u32VFrameType==HI_NET_DEV_VIDEO_FRAME_I)?EASY_SDK_VIDEO_FRAME_I:EASY_SDK_VIDEO_FRAME_P;
				avFrame.u32AVFrameFlag = EASY_SDK_VIDEO_FRAME_FLAG;
				avFrame.u32TimestampSec = pstruAV->u32AVFramePTS/1000;
				avFrame.u32TimestampUsec = (pstruAV->u32AVFramePTS%1000)*1000;
				EasyRTMP_SendPacket(g_RtmpHandle, &avFrame);
			}
		}
		else
		if (pstruAV->u32AVFrameFlag == HI_NET_DEV_AUDIO_FRAME_FLAG)
		{
			if(g_RtmpHandle == 0 ) return 0;

			EASY_AV_Frame	avFrame;
			memset(&avFrame, 0x00, sizeof(EASY_AV_Frame));
			avFrame.u32AVFrameFlag = EASY_SDK_AUDIO_FRAME_FLAG;

			if(pstruAV->u32AVFrameLen <= 4) return 0;

			if(EasyInitAACEncoder(EASY_SDK_AUDIO_CODEC_G711A) == 0)
			{
				memset(m_pAACEncBufer, 0, 64*1024);
				unsigned int iAACBufferLen = 0;
				unsigned char* pbuf = (unsigned char*)(pu8Buffer+sizeof(HI_S_AVFrame)) + 4;
				if(Easy_AACEncoder_Encode(g_pAACEncoderHandle, (unsigned char*)pbuf,  pstruAV->u32AVFrameLen-4, m_pAACEncBufer, &iAACBufferLen) > 0)
				{
					avFrame.pBuffer = (Easy_U8*)(m_pAACEncBufer);
					avFrame.u32AVFrameLen  = iAACBufferLen;
					avFrame.u32TimestampSec = pstruAV->u32AVFramePTS/1000;
					avFrame.u32TimestampUsec = (pstruAV->u32AVFramePTS%1000)*1000;
					EasyRTMP_SendPacket(g_RtmpHandle, &avFrame);
				}
			}
		}
	}
	else
	if (u32DataType == HI_NET_DEV_SYS_DATA)
	{
		pstruSys = (HI_S_SysHeader*)pu8Buffer;
		printf("Video W:%u H:%u Audio: %u \n", pstruSys->struVHeader.u32Width, pstruSys->struVHeader.u32Height, pstruSys->struAHeader.u32Format);
	}
	return HI_SUCCESS;
}