FFmpeg: libavformat/sapenc.c Source File

FFmpeg

[フレーム]

libavformat

sapenc.c

Go to the documentation of this file.

1 /*

2 * Session Announcement Protocol (RFC 2974) muxer

4 *

5 * This file is part of FFmpeg.

6 *

7 * FFmpeg is free software; you can redistribute it and/or

8 * modify it under the terms of the GNU Lesser General Public

9 * License as published by the Free Software Foundation; either

10 * version 2.1 of the License, or (at your option) any later version.

11 *

12 * FFmpeg is distributed in the hope that it will be useful,

13 * but WITHOUT ANY WARRANTY; without even the implied warranty of

14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU

15 * Lesser General Public License for more details.

16 *

17 * You should have received a copy of the GNU Lesser General Public

18 * License along with FFmpeg; if not, write to the Free Software

19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

20 */

22 #include "avformat.h"

23 #include "libavutil/parseutils.h"

24 #include "libavutil/random_seed.h"

25 #include "libavutil/avstring.h"

26 #include "libavutil/dict.h"

27 #include "libavutil/intreadwrite.h"

28 #include "libavutil/time.h"

29 #include "internal.h"

30 #include "network.h"

31 #include "os_support.h"

32 #include "rtpenc_chain.h"

33 #include "url.h"

35 struct SAPState {

36 uint8_t *ann;

37 int ann_size;

38 URLContext *ann_fd;

39 int64_t last_time;

40 };

42 static int sap_write_close(AVFormatContext *s)

43 {

44 struct SAPState *sap = s->priv_data;

45 int i;

47 for (i = 0; i < s->nb_streams; i++) {

48 AVFormatContext *rtpctx = s->streams[i]->priv_data;

49 if (!rtpctx)

50 continue;

51 av_write_trailer(rtpctx);

52 avio_closep(&rtpctx->pb);

53 avformat_free_context(rtpctx);

54 s->streams[i]->priv_data = NULL;

55 }

57 if (sap->last_time && sap->ann && sap->ann_fd) {

58 sap->ann[0] |= 4; /* Session deletion*/

59 ffurl_write(sap->ann_fd, sap->ann, sap->ann_size);

60 }

62 av_freep(&sap->ann);

63 ffurl_closep(&sap->ann_fd);

64 ff_network_close();

65 return 0;

66 }

68 static int sap_write_header(AVFormatContext *s)

69 {

70 struct SAPState *sap = s->priv_data;

71 char host[1024], path[1024], url[1024], announce_addr[50] = "";

72 char *option_list;

73 int port = 9875, base_port = 5004, i, pos = 0, same_port = 0, ttl = 255;

74 AVFormatContext **contexts = NULL;

75 int ret = 0;

76 struct sockaddr_storage localaddr;

77 socklen_t addrlen = sizeof(localaddr);

78 int udp_fd;

79 AVDictionaryEntry* title = av_dict_get(s->metadata, "title", NULL, 0);

81 if (!ff_network_init())

82 return AVERROR(EIO);

84 /* extract hostname and port */

85 av_url_split(NULL, 0, NULL, 0, host, sizeof(host), &base_port,

86 path, sizeof(path), s->url);

87 if (base_port < 0)

88 base_port = 5004;

90 /* search for options */

91 option_list = strrchr(path, '?');

92 if (option_list) {

93 char buf[50];

94 if (av_find_info_tag(buf, sizeof(buf), "announce_port", option_list)) {

95 port = strtol(buf, NULL, 10);

96 }

97 if (av_find_info_tag(buf, sizeof(buf), "same_port", option_list)) {

98 same_port = strtol(buf, NULL, 10);

99 }

100 if (av_find_info_tag(buf, sizeof(buf), "ttl", option_list)) {

101 ttl = strtol(buf, NULL, 10);

102 }

103 if (av_find_info_tag(buf, sizeof(buf), "announce_addr", option_list)) {

104 av_strlcpy(announce_addr, buf, sizeof(announce_addr));

105 }

106 }

107

108 if (!announce_addr[0]) {

109 struct addrinfo hints = { 0 }, *ai = NULL;

110 hints.ai_family = AF_UNSPEC;

111 if (getaddrinfo(host, NULL, &hints, &ai)) {

112 av_log(s, AV_LOG_ERROR, "Unable to resolve %s\n", host);

113 ret = AVERROR(EIO);

114 goto fail;

115 }

116 if (ai->ai_family == AF_INET) {

117 /* Also known as sap.mcast.net */

118 av_strlcpy(announce_addr, "224.2.127.254", sizeof(announce_addr));

119 #if HAVE_STRUCT_SOCKADDR_IN6

120 } else if (ai->ai_family == AF_INET6) {

121 /* With IPv6, you can use the same destination in many different

122 * multicast subnets, to choose how far you want it routed.

123 * This one is intended to be routed globally. */

124 av_strlcpy(announce_addr, "ff0e::2:7ffe", sizeof(announce_addr));

125 #endif

126 } else {

127 freeaddrinfo(ai);

128 av_log(s, AV_LOG_ERROR, "Host %s resolved to unsupported "

129 "address family\n", host);

130 ret = AVERROR(EIO);

131 goto fail;

132 }

133 freeaddrinfo(ai);

134 }

135

136 contexts = av_calloc(s->nb_streams, sizeof(*contexts));

137 if (!contexts) {

138 ret = AVERROR(ENOMEM);

139 goto fail;

140 }

141

142 if (s->start_time_realtime == 0 || s->start_time_realtime == AV_NOPTS_VALUE)

143 s->start_time_realtime = av_gettime();

144 for (i = 0; i < s->nb_streams; i++) {

145 URLContext *fd;

146 char *new_url;

147

148 ff_url_join(url, sizeof(url), "rtp", NULL, host, base_port,

149 "?ttl=%d", ttl);

150 if (!same_port)

151 base_port += 2;

152 ret = ffurl_open_whitelist(&fd, url, AVIO_FLAG_WRITE,

153 &s->interrupt_callback, NULL,

154 s->protocol_whitelist, s->protocol_blacklist, NULL);

155 if (ret) {

156 ret = AVERROR(EIO);

157 goto fail;

158 }

159 ret = ff_rtp_chain_mux_open(&contexts[i], s, s->streams[i], fd, 0, i);

160 if (ret < 0)

161 goto fail;

162 s->streams[i]->priv_data = contexts[i];

163 s->streams[i]->time_base = contexts[i]->streams[0]->time_base;

164 new_url = av_strdup(url);

165 if (!new_url) {

166 ret = AVERROR(ENOMEM);

167 goto fail;

168 }

169 ff_format_set_url(contexts[i], new_url);

170 }

171

172 if (s->nb_streams > 0 && title)

173 av_dict_set(&contexts[0]->metadata, "title", title->value, 0);

174

175 ff_url_join(url, sizeof(url), "udp", NULL, announce_addr, port,

176 "?ttl=%d&connect=1", ttl);

177 ret = ffurl_open_whitelist(&sap->ann_fd, url, AVIO_FLAG_WRITE,

178 &s->interrupt_callback, NULL,

179 s->protocol_whitelist, s->protocol_blacklist, NULL);

180 if (ret) {

181 ret = AVERROR(EIO);

182 goto fail;

183 }

184

185 udp_fd = ffurl_get_file_handle(sap->ann_fd);

186 if (getsockname(udp_fd, (struct sockaddr*) &localaddr, &addrlen)) {

187 ret = AVERROR(EIO);

188 goto fail;

189 }

190 if (localaddr.ss_family != AF_INET

191 #if HAVE_STRUCT_SOCKADDR_IN6

192 && localaddr.ss_family != AF_INET6

193 #endif

194 ) {

195 av_log(s, AV_LOG_ERROR, "Unsupported protocol family\n");

196 ret = AVERROR(EIO);

197 goto fail;

198 }

199 sap->ann_size = 8192;

200 sap->ann = av_mallocz(sap->ann_size);

201 if (!sap->ann) {

202 ret = AVERROR(EIO);

203 goto fail;

204 }

205 sap->ann[pos] = (1 << 5);

206 #if HAVE_STRUCT_SOCKADDR_IN6

207 if (localaddr.ss_family == AF_INET6)

208 sap->ann[pos] |= 0x10;

209 #endif

210 pos++;

211 sap->ann[pos++] = 0; /* Authentication length */

212 AV_WB16(&sap->ann[pos], av_get_random_seed());

213 pos += 2;

214 if (localaddr.ss_family == AF_INET) {

215 memcpy(&sap->ann[pos], &((struct sockaddr_in*)&localaddr)->sin_addr,

216 sizeof(struct in_addr));

217 pos += sizeof(struct in_addr);

218 #if HAVE_STRUCT_SOCKADDR_IN6

219 } else {

220 memcpy(&sap->ann[pos], &((struct sockaddr_in6*)&localaddr)->sin6_addr,

221 sizeof(struct in6_addr));

222 pos += sizeof(struct in6_addr);

223 #endif

224 }

225

226 av_strlcpy(&sap->ann[pos], "application/sdp", sap->ann_size - pos);

227 pos += strlen(&sap->ann[pos]) + 1;

228

229 if (av_sdp_create(contexts, s->nb_streams, &sap->ann[pos],

230 sap->ann_size - pos)) {

231 ret = AVERROR_INVALIDDATA;

232 goto fail;

233 }

234 av_freep(&contexts);

235 av_log(s, AV_LOG_VERBOSE, "SDP:\n%s\n", &sap->ann[pos]);

236 pos += strlen(&sap->ann[pos]);

237 sap->ann_size = pos;

238

239 if (sap->ann_size > sap->ann_fd->max_packet_size) {

240 av_log(s, AV_LOG_ERROR, "Announcement too large to send in one "

241 "packet\n");

242 goto fail;

243 }

244

245 return 0;

246

247 fail:

248 av_free(contexts);

249 sap_write_close(s);

250 return ret;

251 }

252

253 static int sap_write_packet(AVFormatContext *s, AVPacket *pkt)

254 {

255 AVFormatContext *rtpctx;

256 struct SAPState *sap = s->priv_data;

257 int64_t now = av_gettime_relative();

258

259 if (!sap->last_time || now - sap->last_time > 5000000) {

260 int ret = ffurl_write(sap->ann_fd, sap->ann, sap->ann_size);

261 /* Don't abort even if we get "Destination unreachable" */

262 if (ret < 0 && ret != AVERROR(ECONNREFUSED))

263 return ret;

264 sap->last_time = now;

265 }

266 rtpctx = s->streams[pkt->stream_index]->priv_data;

267 return ff_write_chained(rtpctx, 0, pkt, s, 0);

268 }

269

270 const AVOutputFormat ff_sap_muxer = {

271 .name = "sap",

272 .long_name = NULL_IF_CONFIG_SMALL("SAP output"),

273 .priv_data_size = sizeof(struct SAPState),

274 .audio_codec = AV_CODEC_ID_AAC,

275 .video_codec = AV_CODEC_ID_MPEG4,

276 .write_header = sap_write_header,

277 .write_packet = sap_write_packet,

278 .write_trailer = sap_write_close,

279 .flags = AVFMT_NOFILE | AVFMT_GLOBALHEADER,

280 };

av_gettime_relative

int64_t av_gettime_relative(void)

Get the current time in microseconds since some unspecified starting point.

Definition: time.c:56

AVOutputFormat::name

const char * name

Definition: avformat.h:504

AVERROR

Filter the word "frame" indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions

av_find_info_tag

int av_find_info_tag(char *arg, int arg_size, const char *tag1, const char *info)

Attempt to find a specific tag in a URL.

Definition: parseutils.c:753

SAPState::ann_fd

URLContext * ann_fd

Definition: sapdec.c:36

sap_write_packet

static int sap_write_packet(AVFormatContext *s, AVPacket *pkt)

Definition: sapenc.c:253

AV_CODEC_ID_MPEG4

@ AV_CODEC_ID_MPEG4

Definition: codec_id.h:62

AVFormatContext::streams

AVStream ** streams

A list of all streams in the file.

Definition: avformat.h:1268

URLContext::max_packet_size

int max_packet_size

if non zero, the stream is packetized with this max packet size

Definition: url.h:44

SAPState

Definition: sapdec.c:35

AV_LOG_VERBOSE

#define AV_LOG_VERBOSE

Detailed information.

Definition: log.h:196

SAPState::ann

uint8_t * ann

Definition: sapenc.c:36

ff_network_close

void ff_network_close(void)

Definition: network.c:116

os_support.h

sockaddr_storage

Definition: network.h:111

ff_network_init

int ff_network_init(void)

Definition: network.c:58

av_get_random_seed

uint32_t av_get_random_seed(void)

Get a seed to use in conjunction with random functions.

Definition: random_seed.c:120

freeaddrinfo

#define freeaddrinfo

Definition: network.h:218

fail

#define fail()

Definition: checkasm.h:127

rtpenc_chain.h

pkt

AVPacket * pkt

Definition: movenc.c:59

AV_LOG_ERROR

#define AV_LOG_ERROR

Something went wrong and cannot losslessly be recovered.

Definition: log.h:180

ff_write_chained

int ff_write_chained(AVFormatContext *dst, int dst_stream, AVPacket *pkt, AVFormatContext *src, int interleave)

Write a packet to another muxer than the one the user originally intended.

Definition: mux.c:1300

av_dict_get

AVDictionaryEntry * av_dict_get(const AVDictionary *m, const char *key, const AVDictionaryEntry *prev, int flags)

Get a dictionary entry with matching key.

Definition: dict.c:40

ffurl_open_whitelist

int ffurl_open_whitelist(URLContext **puc, const char *filename, int flags, const AVIOInterruptCB *int_cb, AVDictionary **options, const char *whitelist, const char *blacklist, URLContext *parent)

Create an URLContext for accessing to the resource indicated by url, and open it.

Definition: avio.c:306

intreadwrite.h

#define s(width, name)

Definition: cbs_vp9.c:257

ff_url_join

int ff_url_join(char *str, int size, const char *proto, const char *authorization, const char *hostname, int port, const char *fmt,...)

Definition: url.c:38

AVIO_FLAG_WRITE

#define AVIO_FLAG_WRITE

write-only

Definition: avio.h:622

AVFormatContext

Format I/O context.

Definition: avformat.h:1200

internal.h

AVStream::time_base

AVRational time_base

This is the fundamental unit of time (in seconds) in terms of which frame timestamps are represented.

Definition: avformat.h:965

NULL

#define NULL

Definition: coverity.c:32

AV_WB16

#define AV_WB16(p, v)

Definition: intreadwrite.h:405

AVFormatContext::pb

AVIOContext * pb

I/O context.

Definition: avformat.h:1242

parseutils.h

time.h

AV_CODEC_ID_AAC

@ AV_CODEC_ID_AAC

Definition: codec_id.h:425

addrinfo::ai_family

int ai_family

Definition: network.h:139

NULL_IF_CONFIG_SMALL

#define NULL_IF_CONFIG_SMALL(x)

Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.

Definition: internal.h:117

AV_NOPTS_VALUE

#define AV_NOPTS_VALUE

Undefined timestamp value.

Definition: avutil.h:248

AVFMT_NOFILE

#define AVFMT_NOFILE

Demuxer will use avio_open, no opened file should be provided by the caller.

Definition: avformat.h:464

av_sdp_create

int av_sdp_create(AVFormatContext *ac[], int n_files, char *buf, int size)

Generate an SDP for an RTP session.

Definition: sdp.c:905

getaddrinfo

#define getaddrinfo

Definition: network.h:217

avio_closep

int avio_closep(AVIOContext **s)

Close the resource accessed by the AVIOContext *s, free it and set the pointer pointing to it to NULL...

Definition: aviobuf.c:1285

av_write_trailer

int av_write_trailer(AVFormatContext *s)

Write the stream trailer to an output media file and free the file private data.

Definition: mux.c:1243

URLContext

Definition: url.h:38

AVFMT_GLOBALHEADER

#define AVFMT_GLOBALHEADER

Format wants global header.

Definition: avformat.h:474

AVOutputFormat

Definition: avformat.h:503

#define i(width, name, range_min, range_max)

Definition: cbs_h2645.c:271

av_url_split

void av_url_split(char *proto, int proto_size, char *authorization, int authorization_size, char *hostname, int hostname_size, int *port_ptr, char *path, int path_size, const char *url)

Split a URL string into components.

Definition: utils.c:1050

url.h

av_mallocz

void * av_mallocz(size_t size)

Allocate a memory block with alignment suitable for all memory accesses (including vectors if availab...

Definition: mem.c:263

ff_sap_muxer

const AVOutputFormat ff_sap_muxer

Definition: sapenc.c:270

av_calloc

void * av_calloc(size_t nmemb, size_t size)