最简单的基于FFmpeg的推流器（以推送RTMP为例）

心已赠人 2022-08-20 06:13 325阅读 0赞

由于工作一部分工作是作为流媒体服务器的程序员。所以自己那块也算是处理了推流器的一块代码吧。

这边是从网上转载的文章，原文：http://blog.csdn.net/leixiaohua1020/article/details/46890487

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

最简单的基于FFmpeg的推流器系列文章列表：

[《最简单的基于FFmpeg的推流器（以推送RTMP为例）》][FFmpeg_RTMP]

[《最简单的基于FFMPEG的推流器附件：收流器》][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）通过访问流媒体服务器就可以收看实时流了。

![SouthEast][]

运行本程序之前需要先运行RTMP流媒体服务器，并在流媒体服务器上建立相应的Application。有关流媒体服务器的操作不在本文的论述范围内，在此不再详述。本程序运行后，即可通过RTMP客户端（例如 Flash Player, FFplay等等）收看推送的直播流。

## 需要要注意的地方 ##

### 封装格式 ###

RTMP采用的封装格式是FLV。因此在指定输出流媒体的时候需要指定其封装格式为“flv”。同理，其他流媒体协议也需要指定其封装格式。例如采用UDP推送流媒体的时候，可以指定其封装格式为“mpegts”。

### 延时 ###

发送流媒体的数据的时候需要延时。不然的话，FFmpeg处理数据速度很快，瞬间就能把所有的数据发送出去，流媒体服务器是接受不了的。因此需要按照视频实际的帧率发送数据。本文记录的推流器在视频帧与帧之间采用了av\_usleep()函数休眠的方式来延迟发送。这样就可以按照视频的帧率发送数据了，参考代码如下。

**\[cpp\]** [ view plain][view plain] [ copy][view plain]  [![在CODE上查看代码片][CODE]][CODE_CODE] [![派生到我的代码片][ico_fork.svg]][ico_fork.svg 1]

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等参数。这里还没有深入研究，简单写了一点代码，如下所示。

**\[cpp\]** [ view plain][view plain] [ copy][view plain]  [![在CODE上查看代码片][CODE]][CODE_CODE] [![派生到我的代码片][ico_fork.svg]][ico_fork.svg 1]

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的封装格式转换器（无编解码）][FFMPEG 1]》中的封装格式转换器比较类似。它们之间比较明显的区别在于：  
1. Streamer输出为URL  
2. Streamer包含了延时部分

![SouthEast 1][]

## 代码 ##

代码如下。

**\[cpp\]** [ view plain][view plain] [ copy][view plain]  [![在CODE上查看代码片][CODE]][CODE_CODE] [![派生到我的代码片][ico_fork.svg]][ico_fork.svg 1]

1.  /\*\*
2.   \* 最简单的基于FFmpeg的推流器（推送RTMP）
3.   \* Simplest FFmpeg Streamer (Send RTMP)
4.   \* 
5.   \* 雷霄骅 Lei Xiaohua
6.   \* leixiaohua1020@126.com
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. 
20. \#include <stdio.h>
21. 
22. \#define \_\_STDC\_CONSTANT\_MACROS
23. 
24. \#ifdef \_WIN32
25. //Windows
26. extern "C"
27. \{
28. \#include "libavformat/avformat.h"
29. \#include "libavutil/mathematics.h"
30. \#include "libavutil/time.h"
31. \};
32. \#else
33. //Linux...
34. \#ifdef \_\_cplusplus
35. extern "C"
36. \{
37. \#endif
38. \#include <libavformat/avformat.h>
39. \#include <libavutil/mathematics.h>
40. \#include <libavutil/time.h>
41. \#ifdef \_\_cplusplus
42. \};
43. \#endif
44. \#endif
45. 
46. int main(int argc, char\* argv\[\])
47. \{
48. AVOutputFormat \*ofmt = NULL;
49. //输入对应一个AVFormatContext，输出对应一个AVFormatContext
50. //（Input AVFormatContext and Output AVFormatContext）
51. AVFormatContext \*ifmt\_ctx = NULL, \*ofmt\_ctx = NULL;
52. AVPacket pkt;
53. const char \*in\_filename, \*out\_filename;
54. int ret, i;
55. int videoindex=-1;
56. int frame\_index=0;
57. int64\_t start\_time=0;
58. //in\_filename = "cuc\_ieschool.mov";
59. //in\_filename = "cuc\_ieschool.mkv";
60. //in\_filename = "cuc\_ieschool.ts";
61. //in\_filename = "cuc\_ieschool.mp4";
62. //in\_filename = "cuc\_ieschool.h264";
63. in\_filename = "cuc\_ieschool.flv";//输入URL（Input file URL）
64. //in\_filename = "shanghai03\_p.h264";
65. 
66. out\_filename = "rtmp://localhost/publishlive/livestream";//输出 URL（Output URL）\[RTMP\]
67. //out\_filename = "rtp://233.233.233.233:6666";//输出 URL（Output URL）\[UDP\]
68. 
69. av\_register\_all();
70. //Network
71. avformat\_network\_init();
72. //输入（Input）
73. if ((ret = avformat\_open\_input(&ifmt\_ctx, in\_filename, 0, 0)) < 0) \{
74. printf( "Could not open input file.");
75. goto end;
76. \}
77. if ((ret = avformat\_find\_stream\_info(ifmt\_ctx, 0)) < 0) \{
78. printf( "Failed to retrieve input stream information");
79. goto end;
80. \}
81. 
82. for(i=0; i<ifmt\_ctx->nb\_streams; i++)
83. if(ifmt\_ctx->streams\[i\]->codec->codec\_type==AVMEDIA\_TYPE\_VIDEO)\{
84. videoindex=i;
85. break;
86. \}
87. 
88. av\_dump\_format(ifmt\_ctx, 0, in\_filename, 0);
89. 
90. //输出（Output）
91. 
92. avformat\_alloc\_output\_context2(&ofmt\_ctx, NULL, "flv", out\_filename); //RTMP
93. //avformat\_alloc\_output\_context2(&ofmt\_ctx, NULL, "mpegts", out\_filename);//UDP
94. 
95. if (!ofmt\_ctx) \{
96. printf( "Could not create output context\\n");
97. ret = AVERROR\_UNKNOWN;
98. goto end;
99. \}
100. ofmt = ofmt\_ctx->oformat;
101. for (i = 0; i < ifmt\_ctx->nb\_streams; i++) \{
102. //根据输入流创建输出流（Create output AVStream according to input AVStream）
103. AVStream \*in\_stream = ifmt\_ctx->streams\[i\];
104. AVStream \*out\_stream = avformat\_new\_stream(ofmt\_ctx, in\_stream->codec->codec);
105. if (!out\_stream) \{
106. printf( "Failed allocating output stream\\n");
107. ret = AVERROR\_UNKNOWN;
108. goto end;
109. \}
110. //复制AVCodecContext的设置（Copy the settings of AVCodecContext）
111. ret = avcodec\_copy\_context(out\_stream->codec, in\_stream->codec);
112. if (ret < 0) \{
113. printf( "Failed to copy context from input to output stream codec context\\n");
114. goto end;
115. \}
116. out\_stream->codec->codec\_tag = 0;
117. if (ofmt\_ctx->oformat->flags & AVFMT\_GLOBALHEADER)
118. out\_stream->codec->flags |= CODEC\_FLAG\_GLOBAL\_HEADER;
119. \}
120. //Dump Format------------------
121. av\_dump\_format(ofmt\_ctx, 0, out\_filename, 1);
122. //打开输出URL（Open output URL）
123. if (!(ofmt->flags & AVFMT\_NOFILE)) \{
124. ret = avio\_open(&ofmt\_ctx->pb, out\_filename, AVIO\_FLAG\_WRITE);
125. if (ret < 0) \{
126. printf( "Could not open output URL '%s'", out\_filename);
127. goto end;
128. \}
129. \}
130. //写文件头（Write file header）
131. ret = avformat\_write\_header(ofmt\_ctx, NULL);
132. if (ret < 0) \{
133. printf( "Error occurred when opening output URL\\n");
134. goto end;
135. \}
136. 
137. start\_time=av\_gettime();
138. while (1) \{
139. AVStream \*in\_stream, \*out\_stream;
140. //获取一个AVPacket（Get an AVPacket）
141. ret = av\_read\_frame(ifmt\_ctx, &pkt);
142. if (ret < 0)
143. break;
144. //FIX：No PTS (Example: Raw H.264)
145. //Simple Write PTS
146. if(pkt.pts==AV\_NOPTS\_VALUE)\{
147. //Write PTS
148. AVRational time\_base1=ifmt\_ctx->streams\[videoindex\]->time\_base;
149. //Duration between 2 frames (us)
150. int64\_t calc\_duration=(double)AV\_TIME\_BASE/av\_q2d(ifmt\_ctx->streams\[videoindex\]->r\_frame\_rate);
151. //Parameters
152. pkt.pts=(double)(frame\_index\*calc\_duration)/(double)(av\_q2d(time\_base1)\*AV\_TIME\_BASE);
153. pkt.dts=pkt.pts;
154. pkt.duration=(double)calc\_duration/(double)(av\_q2d(time\_base1)\*AV\_TIME\_BASE);
155. \}
156. //Important:Delay
157. if(pkt.stream\_index==videoindex)\{
158. AVRational time\_base=ifmt\_ctx->streams\[videoindex\]->time\_base;
159. AVRational time\_base\_q=\{1,AV\_TIME\_BASE\};
160. int64\_t pts\_time = av\_rescale\_q(pkt.dts, time\_base, time\_base\_q);
161. int64\_t now\_time = av\_gettime() - start\_time;
162. if (pts\_time > now\_time)
163. av\_usleep(pts\_time - now\_time);
164. 
165. \}
166. 
167. in\_stream = ifmt\_ctx->streams\[pkt.stream\_index\];
168. out\_stream = ofmt\_ctx->streams\[pkt.stream\_index\];
169. /\* copy packet \*/
170. //转换PTS/DTS（Convert PTS/DTS）
171. 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));
172. 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));
173. pkt.duration = av\_rescale\_q(pkt.duration, in\_stream->time\_base, out\_stream->time\_base);
174. pkt.pos = -1;
175. //Print to Screen
176. if(pkt.stream\_index==videoindex)\{
177. printf("Send %8d video frames to output URL\\n",frame\_index);
178. frame\_index++;
179. \}
180. //ret = av\_write\_frame(ofmt\_ctx, &pkt);
181. ret = av\_interleaved\_write\_frame(ofmt\_ctx, &pkt);
182. 
183. if (ret < 0) \{
184. printf( "Error muxing packet\\n");
185. break;
186. \}
187. 
188. av\_free\_packet(&pkt);
189. 
190. \}
191. //写文件尾（Write file trailer）
192. av\_write\_trailer(ofmt\_ctx);
193. end:
194. avformat\_close\_input(&ifmt\_ctx);
195. /\* close output \*/
196. if (ofmt\_ctx && !(ofmt->flags & AVFMT\_NOFILE))
197. avio\_close(ofmt\_ctx->pb);
198. avformat\_free\_context(ofmt\_ctx);
199. if (ret < 0 && ret != AVERROR\_EOF) \{
200. printf( "Error occurred.\\n");
201. return \-1;
202. \}
203. return 0;
204. \}

## 结果 ##

程序开始运行后。截图如下所示。

![SouthEast 2][]

可以通过网页播放器播放推送的直播流。

例如下图所示，使用Flash Media Server 的Samples文件夹下的videoPlayer播放直播流的截图如下图所示。（直播地址：rtmp://localhost/publishlive/livestream）

![SouthEast 3][]

此外，也可以通过FFplay这样的客户端播放直播流。

## 下载 ##

**simplest ffmpeg streamer**

**项目主页**

SourceForge：[https://sourceforge.net/projects/simplestffmpegstreamer/][https_sourceforge.net_projects_simplestffmpegstreamer]

Github：[https://github.com/leixiaohua1020/simplest\_ffmpeg\_streamer][https_github.com_leixiaohua1020_simplest_ffmpeg_streamer]

开源中国：[http://git.oschina.net/leixiaohua1020/simplest\_ffmpeg\_streamer][http_git.oschina.net_leixiaohua1020_simplest_ffmpeg_streamer]

CSDN下载地址：  
[http://download.csdn.net/detail/leixiaohua1020/8005311][http_download.csdn.net_detail_leixiaohua1020_8005311]

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

这次考虑到了跨平台的要求，调整了源代码。经过这次调整之后，源代码可以在以下平台编译通过：

> *VC++：打开sln文件即可编译，无需配置。*

> *cl.exe：打开compile\_cl.bat即可命令行下使用cl.exe进行编译，注意可能需要按照VC的安装路径调整脚本里面的参数。编译命令如下。*

**\[plain\]** [ view plain][view plain] [ copy][view plain]  [![在CODE上查看代码片][CODE]][CODE_CODE] [![派生到我的代码片][ico_fork.svg]][ico_fork.svg 1]

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++进行编译。编译命令如下。*

**\[plain\]** [ view plain][view plain] [ copy][view plain]  [![在CODE上查看代码片][CODE]][CODE_CODE] [![派生到我的代码片][ico_fork.svg]][ico_fork.svg 1]

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进行编译。编译命令如下。*

**\[plain\]** [ view plain][view plain] [ copy][view plain]  [![在CODE上查看代码片][CODE]][CODE_CODE] [![派生到我的代码片][ico_fork.svg]][ico_fork.svg 1]

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][http_download.csdn.net_detail_leixiaohua1020_8445455]

SourceForge上已经更新。

**更新-1.2（2015.7.17）==============================**

增加了下列工程：

> simplest\_ffmpeg\_receiver: 将流媒体数据保存成本地文件。  
>

CSDN项目下载地址： [http://download.csdn.net/detail/leixiaohua1020/8924345][http_download.csdn.net_detail_leixiaohua1020_8924345]

SourceForge、Github等上面已经更新。

[FFmpeg_RTMP]: http://blog.csdn.net/leixiaohua1020/article/details/39803457
[FFMPEG]: http://blog.csdn.net/leixiaohua1020/article/details/46890487
[SouthEast]: /images/20220731/988bbcb4ede14d09aa4ab1105f314e65.png
[view plain]: http://blog.csdn.net/leixiaohua1020/article/details/39803457#
[CODE]: https://code.csdn.net/assets/CODE_ico.png
[CODE_CODE]: https://code.csdn.net/snippets/479697
[ico_fork.svg]: https://code.csdn.net/assets/ico_fork.svg
[ico_fork.svg 1]: https://code.csdn.net/snippets/479697/fork
[FFMPEG 1]: http://blog.csdn.net/leixiaohua1020/article/details/25422685
[SouthEast 1]: /images/20220731/939564fa34d04935b4719a8dd4b71939.png
[SouthEast 2]: /images/20220731/3d27c419eb9e4728be7e956c61dd708c.png
[SouthEast 3]: /images/20220731/dfae76ee0c024c0da7b65ed2203db5a0.png
[https_sourceforge.net_projects_simplestffmpegstreamer]: https://sourceforge.net/projects/simplestffmpegstreamer/
[https_github.com_leixiaohua1020_simplest_ffmpeg_streamer]: https://github.com/leixiaohua1020/simplest_ffmpeg_streamer
[http_git.oschina.net_leixiaohua1020_simplest_ffmpeg_streamer]: http://git.oschina.net/leixiaohua1020/simplest_ffmpeg_streamer
[http_download.csdn.net_detail_leixiaohua1020_8005311]: http://download.csdn.net/detail/leixiaohua1020/8005311
[http_download.csdn.net_detail_leixiaohua1020_8445455]: http://download.csdn.net/detail/leixiaohua1020/8445455
[http_download.csdn.net_detail_leixiaohua1020_8924345]: http://download.csdn.net/detail/leixiaohua1020/8924345