Windows下利用live555实现H264实时流RTSP发送

原文地址：http://blog.csdn.net/weixinhum/article/details/38067743#comments

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

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

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

主文件（程序入口）

[cpp] view plaincopy

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. }  

自定义H264VideoFileServerMediaSubsession类

H264VideoFileServerMediaSubsession.hh

[cpp] view plaincopy

1. #ifndef _H264_LIVE_VIDEO_SERVER_MEDIA_SUBSESSION_HH  
2. #define _H264_LIVE_VIDEO_SERVER_MEDIA_SUBSESSION_HH  
3. #include "H264VideoFileServerMediaSubsession.hh"  
4.   
5. class H264LiveVideoServerMediaSubssion : public H264VideoFileServerMediaSubsession {  
6.   
7. public:  
8.     static H264LiveVideoServerMediaSubssion* createNew(UsageEnvironment& env, Boolean reuseFirstSource, int *datasize, unsigned char*  databuf, bool *dosent);  
9.   
10. protected: // we're a virtual base class  
11.     H264LiveVideoServerMediaSubssion(UsageEnvironment& env, Boolean reuseFirstSource, int *datasize, unsigned char*  databuf, bool *dosent);  
12.     ~H264LiveVideoServerMediaSubssion();  
13.   
14. protected: // redefined virtual functions  
15.     FramedSource* createNewStreamSource(unsigned clientSessionId,unsigned& estBitrate);  
16. public:  
17.     char fFileName[100];  
18.   
19.     int *Server_datasize;//数据区大小指针  
20.     unsigned char*  Server_databuf;//数据区指针  
21.     bool *Server_dosent;//发送标示  
22. };  
23. #endif  

H264VideoFileServerMediaSubsession.cpp

[cpp] view plaincopy

1. #include "H264LiveVideoServerMediaSubssion.hh"  
2. #include "H264FramedLiveSource.hh"  
3. #include "H264VideoStreamFramer.hh"  
4.   
5. H264LiveVideoServerMediaSubssion* H264LiveVideoServerMediaSubssion::createNew(UsageEnvironment& env, Boolean reuseFirstSource, int *datasize, unsigned char*  databuf, bool *dosent)  
6. {  
7.     return new H264LiveVideoServerMediaSubssion(env, reuseFirstSource, datasize, databuf, dosent);  
8. }  
9.   
10. H264LiveVideoServerMediaSubssion::H264LiveVideoServerMediaSubssion(UsageEnvironment& env, Boolean reuseFirstSource, int *datasize, unsigned char*  databuf, bool *dosent)  
11. : H264VideoFileServerMediaSubsession(env, fFileName, reuseFirstSource)//H264VideoFileServerMediaSubsession不是我们需要修改的文件，  
12.                                                                       //但是我们又要用它来初始化我们的函数，  
13.                                                                       //所以给个空数组进去即可  
14. {  
15.     Server_datasize = datasize;//数据区大小指针  
16.     Server_databuf = databuf;//数据区指针  
17.     Server_dosent = dosent;//发送标示  
18. }  
19.   
20. H264LiveVideoServerMediaSubssion::~H264LiveVideoServerMediaSubssion()  
21. {  
22. }  
23.   
24. FramedSource* H264LiveVideoServerMediaSubssion::createNewStreamSource(unsigned clientSessionId, unsigned& estBitrate)  
25. {  
26.     /* Remain to do : assign estBitrate */  
27.     estBitrate = 1000; // kbps, estimate  
28.   
29.     //创建视频源  
30.     H264FramedLiveSource* liveSource = H264FramedLiveSource::createNew(envir(), Server_datasize, Server_databuf, Server_dosent);  
31.     if (liveSource == NULL)  
32.     {  
33.         return NULL;  
34.     }  
35.   
36.     // Create a framer for the Video Elementary Stream:  
37.     return H264VideoStreamFramer::createNew(envir(), liveSource);  
38. }  

自定义H264FramedLiveSource类

H264FramedLiveSource.hh

[cpp] view plaincopy

1. #ifndef _H264FRAMEDLIVESOURCE_HH  
2. #define _H264FRAMEDLIVESOURCE_HH  
3.   
4. #include <FramedSource.hh>  
5.   
6.   
7. class H264FramedLiveSource : public FramedSource  
8. {  
9. public:  
10.     static H264FramedLiveSource* createNew(UsageEnvironment& env, int *datasize, unsigned char*  databuf, bool *dosent, unsigned preferredFrameSize = 0, unsigned playTimePerFrame = 0);  
11.   
12. protected:  
13.     H264FramedLiveSource(UsageEnvironment& env, int *datasize, unsigned char*  databuf, bool *dosent, unsigned preferredFrameSize, unsigned playTimePerFrame);  
14.     ~H264FramedLiveSource();  
15.   
16. private:  
17.     virtual void doGetNextFrame();  
18.     int TransportData(unsigned char* to, unsigned maxSize);  
19.   
20. protected:  
21.     int *Framed_datasize;//数据区大小指针  
22.     unsigned char *Framed_databuf;//数据区指针  
23.     bool *Framed_dosent;//发送标示  
24.   
25.     int readbufsize;//记录已读取数据区大小  
26.     int bufsizel;//记录数据区大小  
27. };  
28.   
29. #endif  

H264FramedLiveSource.cpp

[cpp] view plaincopy

1. #include "H264FramedLiveSource.hh"  
2.   
3. H264FramedLiveSource::H264FramedLiveSource(UsageEnvironment& env, int *datasize, unsigned char*  databuf, bool *dosent, unsigned preferredFrameSize, unsigned playTimePerFrame)  
4. : FramedSource(env)  
5. {  
6.     Framed_datasize = datasize;//数据区大小指针  
7.     Framed_databuf = databuf;//数据区指针  
8.     Framed_dosent = dosent;//发送标示  
9. }  
10.   
11. H264FramedLiveSource* H264FramedLiveSource::createNew(UsageEnvironment& env, int *datasize, unsigned char*  databuf, bool *dosent, unsigned preferredFrameSize, unsigned playTimePerFrame)  
12. {  
13.     H264FramedLiveSource* newSource = new H264FramedLiveSource(env, datasize, databuf, dosent, preferredFrameSize, playTimePerFrame);  
14.     return newSource;  
15. }  
16.   
17. H264FramedLiveSource::~H264FramedLiveSource()  
18. {  
19. }  
20.   
21. void H264FramedLiveSource::doGetNextFrame()  
22. {  
23.     if (*Framed_dosent == true)  
24.     {  
25.         *Framed_dosent = false;  
26.         bufsizel = *Framed_datasize;  
27.         readbufsize = 0;  
28.   
29.         fFrameSize = fMaxSize;  
30.         memcpy(fTo, Framed_databuf + readbufsize, fFrameSize);  
31.         readbufsize += fFrameSize;  
32.     }  
33.     else  
34.     {  
35.         if (bufsizel - readbufsize>fMaxSize)  
36.         {  
37.             fFrameSize = fMaxSize;  
38.             memcpy(fTo, Framed_databuf + readbufsize, fFrameSize);  
39.             readbufsize += fFrameSize;  
40.         }  
41.         else  
42.         {  
43.             fFrameSize= bufsizel - readbufsize;//fFrameSize是会告诉live555发送数据区的大小，必须赋予正确的值         memcpy(fTo, Framed_databuf + readbufsize,fFrameSize);  
44.             *Framed_dosent = true;  
45.         }  
46.     }  
47.   
48.     nextTask() = envir().taskScheduler().scheduleDelayedTask(0,(TaskFunc*)FramedSource::afterGetting, this);//表示延迟0秒后再执行 afterGetting 函数  
49.     return;  
50. }  

工程下载地址：点击打开链接

每个工程都是有bug的哈哈，之前写这个文章的时候并没有做认真的测试，在之后做项目的时候用到这一块的时候发现了一些问题，今天就更新一下这份代码吧。如果是下了我之前发布的工程的只能说抱歉了，csdn似乎上传资源一段时间就删不了了，无法更新到最新的版本。工程中H264FramedLiveSource.cpp这个文件中的函数

[cpp] view plaincopy

1. void H264FramedLiveSource::doGetNextFrame()  

中我少写了一句话，就是在

[cpp] view plaincopy

1. memcpy(fTo, Framed_databuf + readbufsize, bufsizel - readbufsize);  

后面需要加上

fFrameSize= bufsizel - readbufsize;//fFrameSize是会告诉live555发送数据区的大小，必须赋予正确的值

这个我已经更新了我博客的代码了，写在这里是提醒下下载我工程的人原来是忘了加的。

然后由于LIVE555默认OutPacketBuffer的大小只有60000

[cpp] view plaincopy

1. <span style="color:#3366ff;">unsigned OutPacketBuffer::maxSize = 60000;</span>  

所以导致如果264文件数据超过这个的时候可能会报错，这个时候可以选择包含MediaSink.hh，MediaSink.cpp这两个文件（在工程目录搜索就可以找到）到工程中，并且直接搜索整个工程。

unsigned OutPacketBuffer::maxSize = 60000
这句代码，将60000改成288000再重新编译就可以了。

对于其他的一些问题，由于本人还没进行进一步的论证就先不放出来了，如果大家使用过程中遇到什么问题可以留言跟我说。