视频技术

FFmpeg red5 nginx Html5 rtsp rtmp

·您当前的位置：首页 > 技术教程 > live555技术 >

[RTSP]Windows下live555做rtsp直播

时间:2015-11-02 22:26酷播
文如其名，最近在做的项目要求利用RTSP协议转发处理完的H264视频数据给上一层客户端，环境是Windows的VS201

文如其名，最近在做的项目要求利用RTSP协议转发处理完的H264视频数据给上一层客户端，环境是Windows的VS2013，于是就各种百度谷歌找代码。结果在得到利用live555去做比较简单的结论的同时也悲情地发现，网上别人贴出来的代码基本都是Linux上面的。在修改了两份来适用于Windows无效后，又一次陷入了百度谷歌的无尽搜索中。Anyway，最后终于解决了，所以贴出代码跟大家分享下，希望能给和我需求相似的童鞋一点启发，也希望有高手指正其中的问题。

用live555进行RTSP的播放基本上是通过修改其给出来的播放本地文件的DEMO来实现的。但由于其DEMO封装的比较深，所以要直接修改他的fread处的代码变成内存拷贝来实现实时传输会显得比较别扭。本文参考了网上的一些代码，自定义了一个继承自H264VideoFileServerMediaSubsession的类来来进行处理，同时定义了一个继承自FramedSource的类来做内存的拷贝操作，该类亦是区别于读本地文件和实时流之紧要处。

一口气杂七杂八说了好多，下面贴出代码吧。如果觉得需要或者懒得自己搭建live555的环境亦可以在文中最后的链接中下载该工程（环境为VS2013），如果你的VS版本合适即可直接运行。


主文件（程序入口）

  1. #include "H264LiveVideoServerMediaSubssion.hh" 
  2. #include "H264FramedLiveSource.hh" 
  3. #include "liveMedia.hh" 
  4. #include "BasicUsageEnvironment.hh" 
  5.  
  6. #define BUFSIZE 1024*200 
  7.  
  8. static void announceStream(RTSPServer* rtspServer, ServerMediaSession* sms,char const* streamName)//显示RTSP连接信息 
  9. { 
  10.     char* url = rtspServer->rtspURL(sms); 
  11.     UsageEnvironment& env = rtspServer->envir(); 
  12.     env <<streamName<< "\n"; 
  13.     env << "Play this stream using the URL \"" << url << "\"\n"; 
  14.     delete[] url; 
  15. } 
  16.  
  17. int main(int argc, char** argv) 
  18. { 
  19.     //设置环境 
  20.     UsageEnvironment* env; 
  21.     Boolean reuseFirstSource = False;//如果为“true”则其他接入的客户端跟第一个客户端看到一样的视频流，否则其他客户端接入的时候将重新播放 
  22.     TaskScheduler* scheduler = BasicTaskScheduler::createNew(); 
  23.     env = BasicUsageEnvironment::createNew(*scheduler); 
  24.  
  25.     //创建RTSP服务器 
  26.     UserAuthenticationDatabase* authDB = NULL; 
  27.     RTSPServer* rtspServer = RTSPServer::createNew(*env, 8554, authDB); 
  28.     if (rtspServer == NULL) { 
  29.         *env << "Failed to create RTSP server: " << env->getResultMsg() << "\n"; 
  30.         exit(1); 
  31.     } 
  32.     char const* descriptionString= "Session streamed by \"testOnDemandRTSPServer\""; 
  33.  
  34.     //模拟实时流发送相关变量 
  35.     int datasize;//数据区长度 
  36.     unsigned char*  databuf;//数据区指针 
  37.     databuf = (unsigned char*)malloc(1024*1024); 
  38.     bool dosent;//rtsp发送标志位，为true则发送，否则退出 
  39.  
  40.     //从文件中拷贝1M数据到内存中作为实时网络传输内存模拟，如果实时网络传输应该是双线程结构，记得在这里加上线程锁 
  41.     //此外实时传输的数据拷贝应该是发生在H264FramedLiveSource文件中，所以这里只是自上往下的传指针过去给它 
  42.     FILE *pf; 
  43.     fopen_s(&pf, "test.264", "rb"); 
  44.     fread(databuf, 1, BUFSIZE, pf); 
  45.     datasize = BUFSIZE; 
  46.     dosent = true; 
  47.     fclose(pf); 
  48.  
  49.     //上面的部分除了模拟网络传输的部分外其他的基本跟live555提供的demo一样，而下面则修改为网络传输的形式，为此重写addSubsession的第一个参数相关文件 
  50.     char const* streamName = "h264ESVideoTest"; 
  51.     ServerMediaSession* sms = ServerMediaSession::createNew(*env, streamName, streamName,descriptionString); 
  52.     sms->addSubsession(H264LiveVideoServerMediaSubssion::createNew(*env, reuseFirstSource, &datasize, databuf,&dosent));//修改为自己实现的H264LiveVideoServerMediaSubssion 
  53.     rtspServer->addServerMediaSession(sms); 
  54.  
  55.     announceStream(rtspServer, sms, streamName);//提示用户输入连接信息 
  56.     env->taskScheduler().doEventLoop(); //循环等待连接 
  57.  
  58.     free(databuf);//释放掉内存 
  59.     return 0; 
  60. } 
  61.  
  62. 自定义H264VideoFileServerMediaSubsession类 
  63.  
  64. H264VideoFileServerMediaSubsession.hh 
  65.  
  66.  
  67.  
  68. #ifndef _H264_LIVE_VIDEO_SERVER_MEDIA_SUBSESSION_HH 
  69. #define _H264_LIVE_VIDEO_SERVER_MEDIA_SUBSESSION_HH 
  70. #include "H264VideoFileServerMediaSubsession.hh" 
  71.  
  72. class H264LiveVideoServerMediaSubssion : public H264VideoFileServerMediaSubsession { 
  73.  
  74. public: 
  75.     static H264LiveVideoServerMediaSubssion* createNew(UsageEnvironment& env, Boolean reuseFirstSource, int *datasize, unsigned char*  databuf, bool *dosent); 
  76.  
  77. protected: // we're a virtual base class 
  78.     H264LiveVideoServerMediaSubssion(UsageEnvironment& env, Boolean reuseFirstSource, int *datasize, unsigned char*  databuf, bool *dosent); 
  79.     ~H264LiveVideoServerMediaSubssion(); 
  80.  
  81. protected: // redefined virtual functions 
  82.     FramedSource* createNewStreamSource(unsigned clientSessionId,unsigned& estBitrate); 
  83. public: 
  84.     char fFileName[100]; 
  85.  
  86.     int *Server_datasize;//数据区大小指针 
  87.     unsigned char*  Server_databuf;//数据区指针 
  88.     bool *Server_dosent;//发送标示 
  89. }; 
  90. #endifH264VideoFileServerMediaSubsession.cpp 
  91.  
  92.  
  93. #include "H264LiveVideoServerMediaSubssion.hh" 
  94. #include "H264FramedLiveSource.hh" 
  95. #include "H264VideoStreamFramer.hh" 
  96.  
  97. H264LiveVideoServerMediaSubssion* H264LiveVideoServerMediaSubssion::createNew(UsageEnvironment& env, Boolean reuseFirstSource, int *datasize, unsigned char*  databuf, bool *dosent) 
  98. { 
  99.     return new H264LiveVideoServerMediaSubssion(env, reuseFirstSource, datasize, databuf, dosent); 
  100. } 
  101.  
  102. H264LiveVideoServerMediaSubssion::H264LiveVideoServerMediaSubssion(UsageEnvironment& env, Boolean reuseFirstSource, int *datasize, unsigned char*  databuf, bool *dosent) 
  103. : H264VideoFileServerMediaSubsession(env, fFileName, reuseFirstSource)//H264VideoFileServerMediaSubsession不是我们需要修改的文件， 
  104.                                                                       //但是我们又要用它来初始化我们的函数， 
  105.                                                                       //所以给个空数组进去即可 
  106. { 
  107.     Server_datasize = datasize;//数据区大小指针 
  108.     Server_databuf = databuf;//数据区指针 
  109.     Server_dosent = dosent;//发送标示 
  110. } 
  111.  
  112. H264LiveVideoServerMediaSubssion::~H264LiveVideoServerMediaSubssion() 
  113. { 
  114. } 
  115.  
  116. FramedSource* H264LiveVideoServerMediaSubssion::createNewStreamSource(unsigned clientSessionId, unsigned& estBitrate) 
  117. { 
  118.     /* Remain to do : assign estBitrate */ 
  119.     estBitrate = 1000; // kbps, estimate 
  120.  
  121.     //创建视频源 
  122.     H264FramedLiveSource* liveSource = H264FramedLiveSource::createNew(envir(), Server_datasize, Server_databuf, Server_dosent); 
  123.     if (liveSource == NULL) 
  124.     { 
  125.         return NULL; 
  126.     } 
  127.  
  128.     // Create a framer for the Video Elementary Stream: 
  129.     return H264VideoStreamFramer::createNew(envir(), liveSource); 
  130. } 
  131. 自定义H264FramedLiveSource类 
  132.  
  133. H264FramedLiveSource.hh 
  134.  
  135.  
  136.  
  137. #ifndef _H264FRAMEDLIVESOURCE_HH 
  138. #define _H264FRAMEDLIVESOURCE_HH 
  139.  
  140. #include <FramedSource.hh> 
  141.  
  142.  
  143. class H264FramedLiveSource : public FramedSource 
  144. { 
  145. public: 
  146.     static H264FramedLiveSource* createNew(UsageEnvironment& env, int *datasize, unsigned char*  databuf, bool *dosent, unsigned preferredFrameSize = 0, unsigned playTimePerFrame = 0); 
  147.  
  148. protected: 
  149.     H264FramedLiveSource(UsageEnvironment& env, int *datasize, unsigned char*  databuf, bool *dosent, unsigned preferredFrameSize, unsigned playTimePerFrame); 
  150.     ~H264FramedLiveSource(); 
  151.  
  152. private: 
  153.     virtual void doGetNextFrame(); 
  154.     int TransportData(unsigned char* to, unsigned maxSize); 
  155.  
  156. protected: 
  157.     int *Framed_datasize;//数据区大小指针 
  158.     unsigned char *Framed_databuf;//数据区指针 
  159.     bool *Framed_dosent;//发送标示 
  160.  
  161.     int readbufsize;//记录已读取数据区大小 
  162.     int bufsizel;//记录数据区大小 
  163. }; 
  164.  
  165. #endifH264FramedLiveSource.cpp 
  166.  
  167.  
  168. #include "H264FramedLiveSource.hh" 
  169.  
  170. H264FramedLiveSource::H264FramedLiveSource(UsageEnvironment& env, int *datasize, unsigned char*  databuf, bool *dosent, unsigned preferredFrameSize, unsigned playTimePerFrame) 
  171. : FramedSource(env) 
  172. { 
  173.     Framed_datasize = datasize;//数据区大小指针 
  174.     Framed_databuf = databuf;//数据区指针 
  175.     Framed_dosent = dosent;//发送标示 
  176. } 
  177.  
  178. H264FramedLiveSource* H264FramedLiveSource::createNew(UsageEnvironment& env, int *datasize, unsigned char*  databuf, bool *dosent, unsigned preferredFrameSize, unsigned playTimePerFrame) 
  179. { 
  180.     H264FramedLiveSource* newnewSource = new H264FramedLiveSource(env, datasize, databuf, dosent, preferredFrameSize, playTimePerFrame); 
  181.     return newSource; 
  182. } 
  183.  
  184. H264FramedLiveSource::~H264FramedLiveSource() 
  185. { 
  186. } 
  187.  
  188. void H264FramedLiveSource::doGetNextFrame() 
  189. { 
  190.     if (*Framed_dosent == true) 
  191.     { 
  192.         *Framed_dosent = false; 
  193.         bufsizel = *Framed_datasize; 
  194.         readbufsize = 0; 
  195.  
  196.         fFrameSize = fMaxSize; 
  197.         memcpy(fTo, Framed_databuf + readbufsize, fFrameSize); 
  198.         readbufsize += fFrameSize; 
  199.     } 
  200.     else 
  201.     { 
  202.         if (bufsizel - readbufsize>fMaxSize) 
  203.         { 
  204.             fFrameSize = fMaxSize; 
  205.             memcpy(fTo, Framed_databuf + readbufsize, fFrameSize); 
  206.             readbufsize += fFrameSize; 
  207.         } 
  208.         else 
  209.         { 
  210.             memcpy(fTo, Framed_databuf + readbufsize, bufsizel - readbufsize); 
  211.             *Framed_dosent = true; 
  212.         } 
  213.     } 
  214.  
  215.     nextTask() = envir().taskScheduler().scheduleDelayedTask(0,(TaskFunc*)FramedSource::afterGetting, this);//表示延迟0秒后再执行 afterGetting 函数 
  216.     return; 
  217. } 

 

热门文章推荐

请稍候...

保利威视云平台-轻松实现点播直播视频应用

酷播云数据统计分析跨平台播放器