最简单的基于FFMPEG的推流器附件：收流器

Love The Way You Lie 2022-08-20 06:13 155阅读 0赞

这块工作中，我没怎么用到，因为公司分的太细了，所以我只做发的那块。所以看了雷哥的文章就了解了代码了。

原文地址：http://blog.csdn.net/leixiaohua1020/article/details/46890487

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

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

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

[《最简单的基于FFMPEG的推流器附件：收流器》][FFMPEG]

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

出于对《最简单的基于FFmpeg的推流器》的补充，本文记录一个最简单的基于FFmpeg的收流器。收流器和推流器的作用正好相反：推流器用于将本地文件以流媒体的形式发送出去，而收流器用于将流媒体内容保存为本地文件。

本文记录的推流器可以将RTMP流媒体保存成为一个本地的FLV文件。由于FFmpeg本身支持很多的流媒体协议和封装格式，所以也支持其它的封装格式和流媒体协议。

## 源代码 ##

**\[cpp\]** [ view plain][view plain] [ copy][view plain]

1.  /\*\*
2.   \* 最简单的基于FFmpeg的收流器（接收RTMP）
3.   \* Simplest FFmpeg Receiver (Receive 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 saves streaming media data (Use RTMP as example)
15.  \* as a local file.
16.  \* It's the simplest FFmpeg stream receiver.
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. //'1': Use H.264 Bitstream Filter 
47. \#define USE\_H264BSF 0
48. 
49. int main(int argc, char\* argv\[\])
50. \{
51. AVOutputFormat \*ofmt = NULL;
52. //Input AVFormatContext and Output AVFormatContext
53. AVFormatContext \*ifmt\_ctx = NULL, \*ofmt\_ctx = NULL;
54. AVPacket pkt;
55. const char \*in\_filename, \*out\_filename;
56. int ret, i;
57. int videoindex=-1;
58. int frame\_index=0;
59. in\_filename = "rtmp://live.hkstv.hk.lxdns.com/live/hks";
60. //in\_filename = "rtp://233.233.233.233:6666";
61. //out\_filename = "receive.ts";
62. //out\_filename = "receive.mkv";
63. out\_filename = "receive.flv";
64. 
65. av\_register\_all();
66. //Network
67. avformat\_network\_init();
68. //Input
69. if ((ret = avformat\_open\_input(&ifmt\_ctx, in\_filename, 0, 0)) < 0) \{
70. printf( "Could not open input file.");
71. goto end;
72. \}
73. if ((ret = avformat\_find\_stream\_info(ifmt\_ctx, 0)) < 0) \{
74. printf( "Failed to retrieve input stream information");
75. goto end;
76. \}
77. 
78. for(i=0; i<ifmt\_ctx->nb\_streams; i++)
79. if(ifmt\_ctx->streams\[i\]->codec->codec\_type==AVMEDIA\_TYPE\_VIDEO)\{
80. videoindex=i;
81. break;
82. \}
83. 
84. av\_dump\_format(ifmt\_ctx, 0, in\_filename, 0);
85. 
86. //Output
87. avformat\_alloc\_output\_context2(&ofmt\_ctx, NULL, NULL, out\_filename); //RTMP
88. 
89. if (!ofmt\_ctx) \{
90. printf( "Could not create output context\\n");
91. ret = AVERROR\_UNKNOWN;
92. goto end;
93. \}
94. ofmt = ofmt\_ctx->oformat;
95. for (i = 0; i < ifmt\_ctx->nb\_streams; i++) \{
96. //Create output AVStream according to input AVStream
97. AVStream \*in\_stream = ifmt\_ctx->streams\[i\];
98. AVStream \*out\_stream = avformat\_new\_stream(ofmt\_ctx, in\_stream->codec->codec);
99. if (!out\_stream) \{
100. printf( "Failed allocating output stream\\n");
101. ret = AVERROR\_UNKNOWN;
102. goto end;
103. \}
104. //Copy the settings of AVCodecContext
105. ret = avcodec\_copy\_context(out\_stream->codec, in\_stream->codec);
106. if (ret < 0) \{
107. printf( "Failed to copy context from input to output stream codec context\\n");
108. goto end;
109. \}
110. out\_stream->codec->codec\_tag = 0;
111. if (ofmt\_ctx->oformat->flags & AVFMT\_GLOBALHEADER)
112. out\_stream->codec->flags |= CODEC\_FLAG\_GLOBAL\_HEADER;
113. \}
114. //Dump Format------------------
115. av\_dump\_format(ofmt\_ctx, 0, out\_filename, 1);
116. //Open output URL
117. if (!(ofmt->flags & AVFMT\_NOFILE)) \{
118. ret = avio\_open(&ofmt\_ctx->pb, out\_filename, AVIO\_FLAG\_WRITE);
119. if (ret < 0) \{
120. printf( "Could not open output URL '%s'", out\_filename);
121. goto end;
122. \}
123. \}
124. //Write file header
125. ret = avformat\_write\_header(ofmt\_ctx, NULL);
126. if (ret < 0) \{
127. printf( "Error occurred when opening output URL\\n");
128. goto end;
129. \}
130. 
131. \#if USE\_H264BSF
132. AVBitStreamFilterContext\* h264bsfc = av\_bitstream\_filter\_init("h264\_mp4toannexb");
133. \#endif
134. 
135. while (1) \{
136. AVStream \*in\_stream, \*out\_stream;
137. //Get an AVPacket
138. ret = av\_read\_frame(ifmt\_ctx, &pkt);
139. if (ret < 0)
140. break;
141. 
142. in\_stream = ifmt\_ctx->streams\[pkt.stream\_index\];
143. out\_stream = ofmt\_ctx->streams\[pkt.stream\_index\];
144. /\* copy packet \*/
145. //Convert PTS/DTS
146. 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));
147. 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));
148. pkt.duration = av\_rescale\_q(pkt.duration, in\_stream->time\_base, out\_stream->time\_base);
149. pkt.pos = -1;
150. //Print to Screen
151. if(pkt.stream\_index==videoindex)\{
152. printf("Receive %8d video frames from input URL\\n",frame\_index);
153. frame\_index++;
154. 
155. \#if USE\_H264BSF
156. av\_bitstream\_filter\_filter(h264bsfc, in\_stream->codec, NULL, &pkt.data, &pkt.size, pkt.data, pkt.size, 0);
157. \#endif
158. \}
159. //ret = av\_write\_frame(ofmt\_ctx, &pkt);
160. ret = av\_interleaved\_write\_frame(ofmt\_ctx, &pkt);
161. 
162. if (ret < 0) \{
163. printf( "Error muxing packet\\n");
164. break;
165. \}
166. 
167. av\_free\_packet(&pkt);
168. 
169. \}
170. 
171. \#if USE\_H264BSF
172. av\_bitstream\_filter\_close(h264bsfc);
173. \#endif
174. 
175. //Write file trailer
176. av\_write\_trailer(ofmt\_ctx);
177. end:
178. avformat\_close\_input(&ifmt\_ctx);
179. /\* close output \*/
180. if (ofmt\_ctx && !(ofmt->flags & AVFMT\_NOFILE))
181. avio\_close(ofmt\_ctx->pb);
182. avformat\_free\_context(ofmt\_ctx);
183. if (ret < 0 && ret != AVERROR\_EOF) \{
184. printf( "Error occurred.\\n");
185. return \-1;
186. \}
187. return 0;
188. \}

## 运行结果 ##

程序运行之后，即可获取流媒体数据并且在本地保存成一个视频文件。

## 下载 ##

**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/8924345][http_download.csdn.net_detail_leixiaohua1020_8924345]

解决方案包含2个项目：

> simplest\_ffmpeg\_streamer: 将本地视频文件推送至流媒体服务器。
> 
> simplest\_ffmpeg\_receiver: 将流媒体数据保存成本地文件。

[FFmpeg_RTMP]: http://blog.csdn.net/leixiaohua1020/article/details/39803457
[FFMPEG]: http://blog.csdn.net/leixiaohua1020/article/details/46890487
[view plain]: http://blog.csdn.net/leixiaohua1020/article/details/46890487#
[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_8924345]: http://download.csdn.net/detail/leixiaohua1020/8924345