RTMP/RTSP推流组件推送H264+AAC到EasyDarwin

/******************************************************************************* Copyright (c) wubihe Tech. Co., Ltd. All rights reserved. -------------------------------------------------------------------------------- Date Created: 2014-10-25 Author: wubihe QQ:1269122125 Email:[email protected] Description: 推流工具库，协议支持RTMP/RTSP,RTSP支持RTP OVER TCP/RTP OVER UDP 视频编码格式支持H264,H265,音频编码格式支持AAC,G711A/G711U 封装格式支持MPEG2-PS,MPEG2-TS,FLV格式,本Demo模拟AAC与H264编码器, 从文件循环读取AAC和H264数据发送给EasyDarwin服务器,并保证音视频 同步 -------------------------------------------------------------------------------- Modification History DATE AUTHOR DESCRIPTION -------------------------------------------------------------------------------- ********************************************************************************/ #include "IPushStream.h" #include "XBase/XThread.h" #include <queue> #define INPUT_H264_FILE_NAME ("huangdun.264") #define INPUT_AAC_FILE_NAME ("huangdun.aac") //推流地址 #define PUSH_STREAM_URL ("rtsp://192.168.1.110/live.sdp") #define MAX_BUFFER_SIZE (1024*8) //读取H264文件 static FILE* gInputH264File = NULL; //读取Aac文件 static FILE* gInputAacFile = NULL; //推流句柄 PSTREAM_HANDLE gPushHandle ; //H264流句柄 int gH264StreamId; //Aac流句柄 int gAacStreamId; unsigned char gszReadBuffer[MAX_BUFFER_SIZE]; //日志回调 void CALLBACK LogCBFun(PSTREAM_LOG_LEVEL nLogLevel, const char *szMessage, void* pUserData ) { printf("%s\n",szMessage); } //消息回调 void CALLBACK MsgCBFun(PSTREAM_HANDLE lHandle, PSTREAM_MSG_TYPE eType,void* pUserData ) { switch (eType) { case PSTREAM_MSG_TYPE_OFFLINE: printf("收到离线消息\n"); break; default: break; } } //线程循环读取本地文件 模拟从编码器获取的编码数据 #ifdef __WINDOWS__ unsigned XAPI Thread1Handle(void* pParam) #endif//__WINDOWS__ #ifdef __GNUC__ void* XAPI Thread1Handle(void* pParam) #endif//__GNUC__ { XThread* pThread = (XThread*)pParam; int iReadSize; //或者采用阻塞版本TryWaitQuit(uint32) while(!pThread->IsExitThread()) { //调用者 只要读取数据送个PushStream即可 同步问题PushStream内部会处理 //读取H264文件 iReadSize = fread(gszReadBuffer, 1, MAX_BUFFER_SIZE, gInputH264File); if(iReadSize > 0) { //流送入库 PSTREAM_DataInput(gPushHandle,gH264StreamId,gszReadBuffer,iReadSize); } else { //文件结束从头读 fseek(gInputH264File,0,SEEK_SET); } //读取H264文件 iReadSize = fread(gszReadBuffer, 1, MAX_BUFFER_SIZE, gInputAacFile); if(iReadSize > 0) { //流送入库 PSTREAM_DataInput(gPushHandle,gAacStreamId,gszReadBuffer,iReadSize); } else { //文件结束从头读 fseek(gInputAacFile,0,SEEK_SET); } XThread::Sleep(20); } printf("线程:%ld 运行结束\n",XThread::SelfID()); return 0; } int main() { gInputH264File = fopen(INPUT_H264_FILE_NAME, "rb"); if (!gInputH264File) { printf("read input file :%s failed!\n",INPUT_H264_FILE_NAME); getchar(); return 0; } gInputAacFile = fopen(INPUT_AAC_FILE_NAME, "rb"); if (!gInputAacFile) { printf("read input file :%s failed!\n",INPUT_AAC_FILE_NAME); getchar(); return 0; } PSTREAM_SetLogCallBack(LogCBFun,NULL); //RTSP/RTP OVER TCP 推流 gPushHandle = PSTREAM_CreatePushHandle(PSTREAM_RTSP_RTP_TCP); if(!gPushHandle) { printf("PSTREAM_CreatePushHandle Error!\n"); getchar(); return 0; } //设置消息回调 PSTREAM_SetMsgCallBack(gPushHandle,MsgCBFun, NULL); //添加音视频轨道 这里添加H264轨道 gH264StreamId = PSTREAM_AddStream(gPushHandle, PSTREAM_VIDEO_TYPE_H264); //添加音视频轨道 这里添加AAC轨道 gAacStreamId = PSTREAM_AddStream(gPushHandle, PSTREAM_AUDIO_TYPE_AAC); if(gH264StreamId<0 || gAacStreamId<0) { printf("PSTREAM_AddStream Error!\n"); PSTREAM_DestroyPushHandle(gPushHandle); getchar(); return 0; } //创建数据读取线程 XThread ReadThread; ReadThread.Start(Thread1Handle,&ReadThread); XThread::Sleep(1000); //探测是否可以打开推流器 int iStartPushRst = PSTREAM_StartPush(gPushHandle,(unsigned char*)PUSH_STREAM_URL); while(iStartPushRst != E_PUSH_SUCCESS) { //数据未准备好 休息一会继续探测 if(iStartPushRst == E_PUSH_NOTREADY) { XThread::Sleep(100); printf("数据探测失败 再次尝试...!\n"); iStartPushRst = PSTREAM_StartPush(gPushHandle,(unsigned char*)PUSH_STREAM_URL); } else { printf("PSTREAM_StartPush Error!\n"); ReadThread.PostStop(); ReadThread.Join(); PSTREAM_DestroyPushHandle(gPushHandle); getchar(); return 0; } } //推流300S后结束 XThread::Sleep(1000*500); printf("推流完成..."); PSTREAM_StopPush(gPushHandle); ReadThread.PostStop(); ReadThread.Join(); PSTREAM_DestroyPushHandle(gPushHandle); printf("Demo 测试完成..."); getchar(); return 1; }