由於工作一部分工作是作為流媒體伺服器的程式設計師。所以自己那塊也算是處理了推流器的一塊程式碼吧。

這邊是從網上轉載的文章，原文：http://blog.csdn.net/leixiaohua1020/article/details/46890487

=====================================================

最簡單的基於FFmpeg的推流器系列文章列表：

=====================================================

本文記錄一個最簡單的基於FFmpeg的推流器（simplest ffmpeg streamer）。推流器的作用就是將本地的視訊資料推送至流媒體伺服器。本文記錄的推流器，可以將本地的 MOV / AVI / MKV / MP4 / FLV 等格式的媒體檔案，通過流媒體協議（例如RTMP，HTTP，UDP，TCP，RTP等等）以直播流的形式推送出去。由於流媒體協議種類繁多，不一一記錄。在這裡記錄將本地檔案以RTMP直播流的形式推送至RTMP流媒體伺服器（例如 Flash Media Server，Red5，Wowza等等）的方法。

在這個推流器的基礎上可以進行多種方式的修改，實現各式各樣的推流器。例如：
* 將輸入檔案改為網路流URL，可以實現轉流器。
* 將輸入的檔案改為回撥函式（記憶體讀取）的形式，可以推送記憶體中的視訊資料。
* 將輸入檔案改為系統裝置（通過libavdevice），同時加上編碼的功能，可以實現實時推流器（現場直播）。

PS：本程式並不包含視訊轉碼的功能。

簡介

RTMP推流器（Streamer）的在流媒體系統中的作用可以用下圖表示。首先將視訊資料以RTMP的形式傳送到流媒體伺服器端（Server，比如FMS，Red5，Wowza等），然後客戶端（一般為Flash Player）通過訪問流媒體伺服器就可以收看實時流了。

執行本程式之前需要先執行RTMP流媒體伺服器，並在流媒體伺服器上建立相應的Application。有關流媒體伺服器的操作不在本文的論述範圍內，在此不再詳述。本程式執行後，即可通過RTMP客戶端（例如 Flash Player, FFplay等等）收看推送的直播流。

需要要注意的地方

封裝格式

RTMP採用的封裝格式是FLV。因此在指定輸出流媒體的時候需要指定其封裝格式為“flv”。同理，其他流媒體協議也需要指定其封裝格式。例如採用UDP推送流媒體的時候，可以指定其封裝格式為“mpegts”。

延時

傳送流媒體的資料的時候需要延時。不然的話，FFmpeg處理資料速度很快，瞬間就能把所有的資料傳送出去，流媒體伺服器是接受不了的。因此需要按照視訊實際的幀率傳送資料。本文記錄的推流器在視訊幀與幀之間採用了av_usleep()函式休眠的方式來延遲傳送。這樣就可以按照視訊的幀率傳送資料了，參考程式碼如下。

1. //…
2. int64_t start_time=av_gettime();  
3. while (1) {  
4. //…
5.     //Important:Delay
6.     if(pkt.stream_index==videoindex){  
7.         AVRational time_base=ifmt_ctx->streams[videoindex]->time_base;  
8.         AVRational time_base_q={1,AV_TIME_BASE};  
9.         int64_t pts_time = av_rescale_q(pkt.dts, time_base, time_base_q);  
10.         int64_t now_time = av_gettime() - start_time;  
11.         if (pts_time > now_time)  
12.             av_usleep(pts_time - now_time);  
13.     }  
14. //…
15. }  
16. //…

PTS/DTS問題

沒有封裝格式的裸流（例如H.264裸流）是不包含PTS、DTS這些引數的。在傳送這種資料的時候，需要自己計算並寫入AVPacket的PTS，DTS，duration等引數。這裡還沒有深入研究，簡單寫了一點程式碼，如下所示。

1. //FIX：No PTS (Example: Raw H.264)
2. //Simple Write PTS
3. if(pkt.pts==AV_NOPTS_VALUE){  
4.     //Write PTS
5.     AVRational time_base1=ifmt_ctx->streams[videoindex]->time_base;  
6.     //Duration between 2 frames (us)
7.     int64_t calc_duration=(double)AV_TIME_BASE/av_q2d(ifmt_ctx->streams[videoindex]->r_frame_rate);  
8.     //Parameters
9.     pkt.pts=(double)(frame_index*calc_duration)/(double)(av_q2d(time_base1)*AV_TIME_BASE);  
10.     pkt.dts=pkt.pts;  
11.     pkt.duration=(double)calc_duration/(double)(av_q2d(time_base1)*AV_TIME_BASE);  
12. }  

程式流程圖

程式的流程圖如下圖所示。可以看出和《最簡單的基於FFMPEG的封裝格式轉換器（無編解碼）》中的封裝格式轉換器比較類似。它們之間比較明顯的區別在於：
1.Streamer輸出為URL
2.Streamer包含了延時部分

 

程式碼

程式碼如下。

1. /** 
2.  * 最簡單的基於FFmpeg的推流器（推送RTMP） 
3.  * Simplest FFmpeg Streamer (Send RTMP) 
4.  *  
5.  * 雷霄驊 Lei Xiaohua 
6.  * [email protected] 
7.  * 中國傳媒大學/數字電視技術 
8.  * Communication University of China / Digital TV Technology 
9.  * http://blog.csdn.net/leixiaohua1020 
10.  *  
11.  * 本例子實現了推送本地視訊至流媒體伺服器（以RTMP為例）。 
12.  * 是使用FFmpeg進行流媒體推送最簡單的教程。 
13.  * 
14.  * This example stream local media files to streaming media  
15.  * server (Use RTMP as example).  
16.  * It's the simplest FFmpeg streamer. 
17.  *  
18.  */
19. #include <stdio.h>
20. #define __STDC_CONSTANT_MACROS
21. #ifdef _WIN32
22. //Windows
23. extern"C"
24. {  
25. #include "libavformat/avformat.h"
26. #include "libavutil/mathematics.h"
27. #include "libavutil/time.h"
28. };  
29. #else
30. //Linux...
31. #ifdef __cplusplus
32. extern"C"
33. {  
34. #endif
35. #include <libavformat/avformat.h>
36. #include <libavutil/mathematics.h>
37. #include <libavutil/time.h>
38. #ifdef __cplusplus
39. };  
40. #endif
41. #endif
42. int main(int argc, char* argv[])  
43. {  
44.     AVOutputFormat *ofmt = NULL;  
45.     //輸入對應一個AVFormatContext，輸出對應一個AVFormatContext
46.     //（Input AVFormatContext and Output AVFormatContext）
47.     AVFormatContext *ifmt_ctx = NULL, *ofmt_ctx = NULL;  
48.     AVPacket pkt;  
49.     constchar *in_filename, *out_filename;  
50.     int ret, i;  
51.     int videoindex=-1;  
52.     int frame_index=0;  
53.     int64_t start_time=0;  
54.     //in_filename  = "cuc_ieschool.mov";
55.     //in_filename  = "cuc_ieschool.mkv";
56.     //in_filename  = "cuc_ieschool.ts";
57.     //in_filename  = "cuc_ieschool.mp4";
58.     //in_filename  = "cuc_ieschool.h264";
59.     in_filename  = "cuc_ieschool.flv";//輸入URL（Input file URL）
60.     //in_filename  = "shanghai03_p.h264";
61.     out_filename = "rtmp://localhost/publishlive/livestream";//輸出 URL（Output URL）[RTMP]  
62.     //out_filename = "rtp://233.233.233.233:6666";//輸出 URL（Output URL）[UDP]
63.     av_register_all();  
64.     //Network
65.     avformat_network_init();  
66.     //輸入（Input）
67.     if ((ret = avformat_open_input(&ifmt_ctx, in_filename, 0, 0)) < 0) {  
68.         printf( "Could not open input file.");  
69.         goto end;  
70.     }  
71.     if ((ret = avformat_find_stream_info(ifmt_ctx, 0)) < 0) {  
72.         printf( "Failed to retrieve input stream information");  
73.         goto end;  
74.     }  
75.     for(i=0; i<ifmt_ctx->nb_streams; i++)   
76.         if(ifmt_ctx->streams[i]->codec->codec_type==AVMEDIA_TYPE_VIDEO){  
77.             videoindex=i;  
78.             break;  
79.         }  
80.     av_dump_format(ifmt_ctx, 0, in_filename, 0);  
81.     //輸出（Output）
82.     avformat_alloc_output_context2(&ofmt_ctx, NULL, "flv", out_filename); //RTMP
83.     //avformat_alloc_output_context2(&ofmt_ctx, NULL, "mpegts", out_filename);//UDP
84.     if (!ofmt_ctx) {  
85.         printf( "Could not create output context\n");  
86.         ret = AVERROR_UNKNOWN;  
87.         goto end;  
88.     }  
89.     ofmt = ofmt_ctx->oformat;  
90.     for (i = 0; i < ifmt_ctx->nb_streams; i++) {  
91.         //根據輸入流建立輸出流（Create output AVStream according to input AVStream）
92.         AVStream *in_stream = ifmt_ctx->streams[i];  
93.         AVStream *out_stream = avformat_new_stream(ofmt_ctx, in_stream->codec->codec);  
94.         if (!out_stream) {  
95.             printf( "Failed allocating output stream\n");  
96.             ret = AVERROR_UNKNOWN;  
97.             goto end;  
98.         }  
99.         //複製AVCodecContext的設定（Copy the settings of AVCodecContext）
100.         ret = avcodec_copy_context(out_stream->codec, in_stream->codec);  
101.         if (ret < 0) {  
102.             printf( "Failed to copy context from input to output stream codec context\n");  
103.             goto end;  
104.         }  
105.         out_stream->codec->codec_tag = 0;  
106.         if (ofmt_ctx->oformat->flags & AVFMT_GLOBALHEADER)  
107.             out_stream->codec->flags |= CODEC_FLAG_GLOBAL_HEADER;  
108.     }  
109.     //Dump Format------------------
110.     av_dump_format(ofmt_ctx, 0, out_filename, 1);  
111.     //開啟輸出URL（Open output URL）
112.     if (!(ofmt->flags & AVFMT_NOFILE)) {  
113.         ret = avio_open(&ofmt_ctx->pb, out_filename, AVIO_FLAG_WRITE);  
114.         if (ret < 0) {  
115.             printf( "Could not open output URL '%s'", out_filename);  
116.             goto end;  
117.         }  
118.     }  
119.     //寫檔案頭（Write file header）
120.     ret = avformat_write_header(ofmt_ctx, NULL);  
121.     if (ret < 0) {  
122.         printf( "Error occurred when opening output URL\n");  
123.         goto end;  
124.     }  
125.     start_time=av_gettime();  
126.     while (1) {  
127.         AVStream *in_stream, *out_stream;  
128.         //獲取一個AVPacket（Get an AVPacket）
129.         ret = av_read_frame(ifmt_ctx, &pkt);  
130.         if (ret < 0)  
131.             break;  
132.         //FIX：No PTS (Example: Raw H.264)
133.         //Simple Write PTS
134.         if(pkt.pts==AV_NOPTS_VALUE){  
135.             //Write PTS
136.             AVRational time_base1=ifmt_ctx->streams[videoindex]->time_base;  
137.             //Duration between 2 frames (us)
138.             int64_t calc_duration=(double)AV_TIME_BASE/av_q2d(ifmt_ctx->streams[videoindex]->r_frame_rate);  
139.             //Parameters
140.             pkt.pts=(double)(frame_index*calc_duration)/(double)(av_q2d(time_base1)*AV_TIME_BASE);  
141.             pkt.dts=pkt.pts;  
142.             pkt.duration=(double)calc_duration/(double)(av_q2d(time_base1)*AV_TIME_BASE);  
143.         }  
144.         //Important:Delay
145.         if(pkt.stream_index==videoindex){  
146.             AVRational time_base=ifmt_ctx->streams[videoindex]->time_base;  
147.             AVRational time_base_q={1,AV_TIME_BASE};  
148.             int64_t pts_time = av_rescale_q(pkt.dts, time_base, time_base_q);  
149.             int64_t now_time = av_gettime() - start_time;  
150.             if (pts_time > now_time)  
151.                 av_usleep(pts_time - now_time);  
152.         }  
153.         in_stream  = ifmt_ctx->streams[pkt.stream_index];  
154.         out_stream = ofmt_ctx->streams[pkt.stream_index];  
155.         /* copy packet */
156.         //轉換PTS/DTS（Convert PTS/DTS）
157.         pkt.pts = av_rescale_q_rnd(pkt.pts, in_stream->time_base, out_stream->time_base, (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));  
158.         pkt.dts = av_rescale_q_rnd(pkt.dts, in_stream->time_base, out_stream->time_base, (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));  
159.         pkt.duration = av_rescale_q(pkt.duration, in_stream->time_base, out_stream->time_base);  
160.         pkt.pos = -1;  
161.         //Print to Screen
162.         if(pkt.stream_index==videoindex){  
163.             printf("Send %8d video frames to output URL\n",frame_index);  
164.             frame_index++;  
165.         }  
166.         //ret = av_write_frame(ofmt_ctx, &pkt);
167.         ret = av_interleaved_write_frame(ofmt_ctx, &pkt);  
168.         if (ret < 0) {  
169.             printf( "Error muxing packet\n");  
170.             break;  
171.         }  
172.         av_free_packet(&pkt);  
173.     }  
174.     //寫檔案尾（Write file trailer）
175.     av_write_trailer(ofmt_ctx);  
176. end:  
177.     avformat_close_input(&ifmt_ctx);  
178.     /* close output */
179.     if (ofmt_ctx && !(ofmt->flags & AVFMT_NOFILE))  
180.         avio_close(ofmt_ctx->pb);  
181.     avformat_free_context(ofmt_ctx);  
182.     if (ret < 0 && ret != AVERROR_EOF) {  
183.         printf( "Error occurred.\n");  
184.         return -1;  
185.     }  
186.     return 0;  
187. }  

結果

程式開始執行後。截圖如下所示。

可以通過網頁播放器播放推送的直播流。

例如下圖所示，使用Flash Media Server 的Samples資料夾下的videoPlayer播放直播流的截圖如下圖所示。（直播地址：rtmp://localhost/publishlive/livestream）

此外，也可以通過FFplay這樣的客戶端播放直播流。

下載

simplest ffmpeg streamer

專案主頁

更新-1.1 (2015.2.13)=========================================

這次考慮到了跨平臺的要求，調整了原始碼。經過這次調整之後，原始碼可以在以下平臺編譯通過：

VC++：開啟sln檔案即可編譯，無需配置。

cl.exe：開啟compile_cl.bat即可命令列下使用cl.exe進行編譯，注意可能需要按照VC的安裝路徑調整腳本里面的引數。編譯命令如下。

1. ::VS2010 Environment  
2. call "D:\Program Files\Microsoft Visual Studio 10.0\VC\vcvarsall.bat"  
3. ::include  
4. @set INCLUDE=include;%INCLUDE%;  
5. ::lib  
6. @set LIB=lib;%LIB%;  
7. ::compile and link  
8. cl simplest_ffmpeg_streamer.cpp /link avcodec.lib avformat.lib avutil.lib ^  
9. avdevice.lib avfilter.lib postproc.lib swresample.lib swscale.lib /OPT:NOREF  

MinGW：MinGW命令列下執行compile_mingw.sh即可使用MinGW的g++進行編譯。編譯命令如下。

1. g++ simplest_ffmpeg_streamer.cpp -g -o simplest_ffmpeg_streamer.exe \  
2. -I /usr/local/include -L /usr/local/lib -lavformat -lavcodec -lavutil  

GCC：Linux或者MacOS命令列下執行compile_gcc.sh即可使用GCC進行編譯。編譯命令如下。

1. gcc simplest_ffmpeg_streamer.cpp -g -o simplest_ffmpeg_streamer.out \  
2. -I /usr/local/include -L /usr/local/lib -lavformat -lavcodec -lavutil  

PS：相關的編譯命令已經儲存到了工程資料夾中


CSDN下載地址：http://download.csdn.net/detail/leixiaohua1020/8445455

SourceForge上已經更新。

更新-1.2（2015.7.17）==============================
增加了下列工程：

simplest_ffmpeg_receiver: 將流媒體資料儲存成本地檔案。

SourceForge、Github等上面已經更新。