You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

658 lines
22KB

  1. /*
  2. * RTP output format
  3. * Copyright (c) 2002 Fabrice Bellard
  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. */
  21. #include "avformat.h"
  22. #include "mpegts.h"
  23. #include "internal.h"
  24. #include "libavutil/mathematics.h"
  25. #include "libavutil/random_seed.h"
  26. #include "libavutil/opt.h"
  27. #include "rtpenc.h"
  28. static const AVOption options[] = {
  29. FF_RTP_FLAG_OPTS(RTPMuxContext, flags),
  30. { "payload_type", "Specify RTP payload type", offsetof(RTPMuxContext, payload_type), AV_OPT_TYPE_INT, {.i64 = -1 }, -1, 127, AV_OPT_FLAG_ENCODING_PARAM },
  31. { "ssrc", "Stream identifier", offsetof(RTPMuxContext, ssrc), AV_OPT_TYPE_INT, { .i64 = 0 }, INT_MIN, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM },
  32. { "cname", "CNAME to include in RTCP SR packets", offsetof(RTPMuxContext, cname), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, AV_OPT_FLAG_ENCODING_PARAM },
  33. { "seq", "Starting sequence number", offsetof(RTPMuxContext, seq), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, 65535, AV_OPT_FLAG_ENCODING_PARAM },
  34. { NULL },
  35. };
  36. static const AVClass rtp_muxer_class = {
  37. .class_name = "RTP muxer",
  38. .item_name = av_default_item_name,
  39. .option = options,
  40. .version = LIBAVUTIL_VERSION_INT,
  41. };
  42. #define RTCP_SR_SIZE 28
  43. static int is_supported(enum AVCodecID id)
  44. {
  45. switch(id) {
  46. case AV_CODEC_ID_DIRAC:
  47. case AV_CODEC_ID_H261:
  48. case AV_CODEC_ID_H263:
  49. case AV_CODEC_ID_H263P:
  50. case AV_CODEC_ID_H264:
  51. case AV_CODEC_ID_HEVC:
  52. case AV_CODEC_ID_MPEG1VIDEO:
  53. case AV_CODEC_ID_MPEG2VIDEO:
  54. case AV_CODEC_ID_MPEG4:
  55. case AV_CODEC_ID_AAC:
  56. case AV_CODEC_ID_MP2:
  57. case AV_CODEC_ID_MP3:
  58. case AV_CODEC_ID_PCM_ALAW:
  59. case AV_CODEC_ID_PCM_MULAW:
  60. case AV_CODEC_ID_PCM_S8:
  61. case AV_CODEC_ID_PCM_S16BE:
  62. case AV_CODEC_ID_PCM_S16LE:
  63. case AV_CODEC_ID_PCM_U16BE:
  64. case AV_CODEC_ID_PCM_U16LE:
  65. case AV_CODEC_ID_PCM_U8:
  66. case AV_CODEC_ID_MPEG2TS:
  67. case AV_CODEC_ID_AMR_NB:
  68. case AV_CODEC_ID_AMR_WB:
  69. case AV_CODEC_ID_VORBIS:
  70. case AV_CODEC_ID_THEORA:
  71. case AV_CODEC_ID_VP8:
  72. case AV_CODEC_ID_VP9:
  73. case AV_CODEC_ID_ADPCM_G722:
  74. case AV_CODEC_ID_ADPCM_G726:
  75. case AV_CODEC_ID_ILBC:
  76. case AV_CODEC_ID_MJPEG:
  77. case AV_CODEC_ID_SPEEX:
  78. case AV_CODEC_ID_OPUS:
  79. return 1;
  80. default:
  81. return 0;
  82. }
  83. }
  84. static int rtp_write_header(AVFormatContext *s1)
  85. {
  86. RTPMuxContext *s = s1->priv_data;
  87. int n, ret = AVERROR(EINVAL);
  88. AVStream *st;
  89. if (s1->nb_streams != 1) {
  90. av_log(s1, AV_LOG_ERROR, "Only one stream supported in the RTP muxer\n");
  91. return AVERROR(EINVAL);
  92. }
  93. st = s1->streams[0];
  94. if (!is_supported(st->codecpar->codec_id)) {
  95. av_log(s1, AV_LOG_ERROR, "Unsupported codec %s\n", avcodec_get_name(st->codecpar->codec_id));
  96. return -1;
  97. }
  98. if (s->payload_type < 0) {
  99. /* Re-validate non-dynamic payload types */
  100. if (st->id < RTP_PT_PRIVATE)
  101. st->id = ff_rtp_get_payload_type(s1, st->codecpar, -1);
  102. s->payload_type = st->id;
  103. } else {
  104. /* private option takes priority */
  105. st->id = s->payload_type;
  106. }
  107. s->base_timestamp = av_get_random_seed();
  108. s->timestamp = s->base_timestamp;
  109. s->cur_timestamp = 0;
  110. if (!s->ssrc)
  111. s->ssrc = av_get_random_seed();
  112. s->first_packet = 1;
  113. s->first_rtcp_ntp_time = ff_ntp_time();
  114. if (s1->start_time_realtime != 0 && s1->start_time_realtime != AV_NOPTS_VALUE)
  115. /* Round the NTP time to whole milliseconds. */
  116. s->first_rtcp_ntp_time = (s1->start_time_realtime / 1000) * 1000 +
  117. NTP_OFFSET_US;
  118. // Pick a random sequence start number, but in the lower end of the
  119. // available range, so that any wraparound doesn't happen immediately.
  120. // (Immediate wraparound would be an issue for SRTP.)
  121. if (s->seq < 0) {
  122. if (s1->flags & AVFMT_FLAG_BITEXACT) {
  123. s->seq = 0;
  124. } else
  125. s->seq = av_get_random_seed() & 0x0fff;
  126. } else
  127. s->seq &= 0xffff; // Use the given parameter, wrapped to the right interval
  128. if (s1->packet_size) {
  129. if (s1->pb->max_packet_size)
  130. s1->packet_size = FFMIN(s1->packet_size,
  131. s1->pb->max_packet_size);
  132. } else
  133. s1->packet_size = s1->pb->max_packet_size;
  134. if (s1->packet_size <= 12) {
  135. av_log(s1, AV_LOG_ERROR, "Max packet size %d too low\n", s1->packet_size);
  136. return AVERROR(EIO);
  137. }
  138. s->buf = av_malloc(s1->packet_size);
  139. if (!s->buf) {
  140. return AVERROR(ENOMEM);
  141. }
  142. s->max_payload_size = s1->packet_size - 12;
  143. if (st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) {
  144. avpriv_set_pts_info(st, 32, 1, st->codecpar->sample_rate);
  145. } else {
  146. avpriv_set_pts_info(st, 32, 1, 90000);
  147. }
  148. s->buf_ptr = s->buf;
  149. switch(st->codecpar->codec_id) {
  150. case AV_CODEC_ID_MP2:
  151. case AV_CODEC_ID_MP3:
  152. s->buf_ptr = s->buf + 4;
  153. avpriv_set_pts_info(st, 32, 1, 90000);
  154. break;
  155. case AV_CODEC_ID_MPEG1VIDEO:
  156. case AV_CODEC_ID_MPEG2VIDEO:
  157. break;
  158. case AV_CODEC_ID_MPEG2TS:
  159. n = s->max_payload_size / TS_PACKET_SIZE;
  160. if (n < 1)
  161. n = 1;
  162. s->max_payload_size = n * TS_PACKET_SIZE;
  163. break;
  164. case AV_CODEC_ID_DIRAC:
  165. if (s1->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL) {
  166. av_log(s, AV_LOG_ERROR,
  167. "Packetizing VC-2 is experimental and does not use all values "
  168. "of the specification "
  169. "(even though most receivers may handle it just fine). "
  170. "Please set -strict experimental in order to enable it.\n");
  171. ret = AVERROR_EXPERIMENTAL;
  172. goto fail;
  173. }
  174. break;
  175. case AV_CODEC_ID_H261:
  176. if (s1->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL) {
  177. av_log(s, AV_LOG_ERROR,
  178. "Packetizing H261 is experimental and produces incorrect "
  179. "packetization for cases where GOBs don't fit into packets "
  180. "(even though most receivers may handle it just fine). "
  181. "Please set -f_strict experimental in order to enable it.\n");
  182. ret = AVERROR_EXPERIMENTAL;
  183. goto fail;
  184. }
  185. break;
  186. case AV_CODEC_ID_H264:
  187. /* check for H.264 MP4 syntax */
  188. if (st->codecpar->extradata_size > 4 && st->codecpar->extradata[0] == 1) {
  189. s->nal_length_size = (st->codecpar->extradata[4] & 0x03) + 1;
  190. }
  191. break;
  192. case AV_CODEC_ID_HEVC:
  193. /* Only check for the standardized hvcC version of extradata, keeping
  194. * things simple and similar to the avcC/H264 case above, instead
  195. * of trying to handle the pre-standardization versions (as in
  196. * libavcodec/hevc.c). */
  197. if (st->codecpar->extradata_size > 21 && st->codecpar->extradata[0] == 1) {
  198. s->nal_length_size = (st->codecpar->extradata[21] & 0x03) + 1;
  199. }
  200. break;
  201. case AV_CODEC_ID_VP9:
  202. if (s1->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL) {
  203. av_log(s, AV_LOG_ERROR,
  204. "Packetizing VP9 is experimental and its specification is "
  205. "still in draft state. "
  206. "Please set -strict experimental in order to enable it.\n");
  207. ret = AVERROR_EXPERIMENTAL;
  208. goto fail;
  209. }
  210. break;
  211. case AV_CODEC_ID_VORBIS:
  212. case AV_CODEC_ID_THEORA:
  213. s->max_frames_per_packet = 15;
  214. break;
  215. case AV_CODEC_ID_ADPCM_G722:
  216. /* Due to a historical error, the clock rate for G722 in RTP is
  217. * 8000, even if the sample rate is 16000. See RFC 3551. */
  218. avpriv_set_pts_info(st, 32, 1, 8000);
  219. break;
  220. case AV_CODEC_ID_OPUS:
  221. if (st->codecpar->channels > 2) {
  222. av_log(s1, AV_LOG_ERROR, "Multistream opus not supported in RTP\n");
  223. goto fail;
  224. }
  225. /* The opus RTP RFC says that all opus streams should use 48000 Hz
  226. * as clock rate, since all opus sample rates can be expressed in
  227. * this clock rate, and sample rate changes on the fly are supported. */
  228. avpriv_set_pts_info(st, 32, 1, 48000);
  229. break;
  230. case AV_CODEC_ID_ILBC:
  231. if (st->codecpar->block_align != 38 && st->codecpar->block_align != 50) {
  232. av_log(s1, AV_LOG_ERROR, "Incorrect iLBC block size specified\n");
  233. goto fail;
  234. }
  235. s->max_frames_per_packet = s->max_payload_size / st->codecpar->block_align;
  236. break;
  237. case AV_CODEC_ID_AMR_NB:
  238. case AV_CODEC_ID_AMR_WB:
  239. s->max_frames_per_packet = 50;
  240. if (st->codecpar->codec_id == AV_CODEC_ID_AMR_NB)
  241. n = 31;
  242. else
  243. n = 61;
  244. /* max_header_toc_size + the largest AMR payload must fit */
  245. if (1 + s->max_frames_per_packet + n > s->max_payload_size) {
  246. av_log(s1, AV_LOG_ERROR, "RTP max payload size too small for AMR\n");
  247. goto fail;
  248. }
  249. if (st->codecpar->channels != 1) {
  250. av_log(s1, AV_LOG_ERROR, "Only mono is supported\n");
  251. goto fail;
  252. }
  253. break;
  254. case AV_CODEC_ID_AAC:
  255. s->max_frames_per_packet = 50;
  256. break;
  257. default:
  258. break;
  259. }
  260. return 0;
  261. fail:
  262. av_freep(&s->buf);
  263. return ret;
  264. }
  265. /* send an rtcp sender report packet */
  266. static void rtcp_send_sr(AVFormatContext *s1, int64_t ntp_time, int bye)
  267. {
  268. RTPMuxContext *s = s1->priv_data;
  269. uint32_t rtp_ts;
  270. av_log(s1, AV_LOG_TRACE, "RTCP: %02x %"PRIx64" %x\n", s->payload_type, ntp_time, s->timestamp);
  271. s->last_rtcp_ntp_time = ntp_time;
  272. rtp_ts = av_rescale_q(ntp_time - s->first_rtcp_ntp_time, (AVRational){1, 1000000},
  273. s1->streams[0]->time_base) + s->base_timestamp;
  274. avio_w8(s1->pb, RTP_VERSION << 6);
  275. avio_w8(s1->pb, RTCP_SR);
  276. avio_wb16(s1->pb, 6); /* length in words - 1 */
  277. avio_wb32(s1->pb, s->ssrc);
  278. avio_wb32(s1->pb, ntp_time / 1000000);
  279. avio_wb32(s1->pb, ((ntp_time % 1000000) << 32) / 1000000);
  280. avio_wb32(s1->pb, rtp_ts);
  281. avio_wb32(s1->pb, s->packet_count);
  282. avio_wb32(s1->pb, s->octet_count);
  283. if (s->cname) {
  284. int len = FFMIN(strlen(s->cname), 255);
  285. avio_w8(s1->pb, (RTP_VERSION << 6) + 1);
  286. avio_w8(s1->pb, RTCP_SDES);
  287. avio_wb16(s1->pb, (7 + len + 3) / 4); /* length in words - 1 */
  288. avio_wb32(s1->pb, s->ssrc);
  289. avio_w8(s1->pb, 0x01); /* CNAME */
  290. avio_w8(s1->pb, len);
  291. avio_write(s1->pb, s->cname, len);
  292. avio_w8(s1->pb, 0); /* END */
  293. for (len = (7 + len) % 4; len % 4; len++)
  294. avio_w8(s1->pb, 0);
  295. }
  296. if (bye) {
  297. avio_w8(s1->pb, (RTP_VERSION << 6) | 1);
  298. avio_w8(s1->pb, RTCP_BYE);
  299. avio_wb16(s1->pb, 1); /* length in words - 1 */
  300. avio_wb32(s1->pb, s->ssrc);
  301. }
  302. avio_flush(s1->pb);
  303. }
  304. /* send an rtp packet. sequence number is incremented, but the caller
  305. must update the timestamp itself */
  306. void ff_rtp_send_data(AVFormatContext *s1, const uint8_t *buf1, int len, int m)
  307. {
  308. RTPMuxContext *s = s1->priv_data;
  309. av_log(s1, AV_LOG_TRACE, "rtp_send_data size=%d\n", len);
  310. /* build the RTP header */
  311. avio_w8(s1->pb, RTP_VERSION << 6);
  312. avio_w8(s1->pb, (s->payload_type & 0x7f) | ((m & 0x01) << 7));
  313. avio_wb16(s1->pb, s->seq);
  314. avio_wb32(s1->pb, s->timestamp);
  315. avio_wb32(s1->pb, s->ssrc);
  316. avio_write(s1->pb, buf1, len);
  317. avio_flush(s1->pb);
  318. s->seq = (s->seq + 1) & 0xffff;
  319. s->octet_count += len;
  320. s->packet_count++;
  321. }
  322. /* send an integer number of samples and compute time stamp and fill
  323. the rtp send buffer before sending. */
  324. static int rtp_send_samples(AVFormatContext *s1,
  325. const uint8_t *buf1, int size, int sample_size_bits)
  326. {
  327. RTPMuxContext *s = s1->priv_data;
  328. int len, max_packet_size, n;
  329. /* Calculate the number of bytes to get samples aligned on a byte border */
  330. int aligned_samples_size = sample_size_bits/av_gcd(sample_size_bits, 8);
  331. max_packet_size = (s->max_payload_size / aligned_samples_size) * aligned_samples_size;
  332. /* Not needed, but who knows. Don't check if samples aren't an even number of bytes. */
  333. if ((sample_size_bits % 8) == 0 && ((8 * size) % sample_size_bits) != 0)
  334. return AVERROR(EINVAL);
  335. n = 0;
  336. while (size > 0) {
  337. s->buf_ptr = s->buf;
  338. len = FFMIN(max_packet_size, size);
  339. /* copy data */
  340. memcpy(s->buf_ptr, buf1, len);
  341. s->buf_ptr += len;
  342. buf1 += len;
  343. size -= len;
  344. s->timestamp = s->cur_timestamp + n * 8 / sample_size_bits;
  345. ff_rtp_send_data(s1, s->buf, s->buf_ptr - s->buf, 0);
  346. n += (s->buf_ptr - s->buf);
  347. }
  348. return 0;
  349. }
  350. static void rtp_send_mpegaudio(AVFormatContext *s1,
  351. const uint8_t *buf1, int size)
  352. {
  353. RTPMuxContext *s = s1->priv_data;
  354. int len, count, max_packet_size;
  355. max_packet_size = s->max_payload_size;
  356. /* test if we must flush because not enough space */
  357. len = (s->buf_ptr - s->buf);
  358. if ((len + size) > max_packet_size) {
  359. if (len > 4) {
  360. ff_rtp_send_data(s1, s->buf, s->buf_ptr - s->buf, 0);
  361. s->buf_ptr = s->buf + 4;
  362. }
  363. }
  364. if (s->buf_ptr == s->buf + 4) {
  365. s->timestamp = s->cur_timestamp;
  366. }
  367. /* add the packet */
  368. if (size > max_packet_size) {
  369. /* big packet: fragment */
  370. count = 0;
  371. while (size > 0) {
  372. len = max_packet_size - 4;
  373. if (len > size)
  374. len = size;
  375. /* build fragmented packet */
  376. s->buf[0] = 0;
  377. s->buf[1] = 0;
  378. s->buf[2] = count >> 8;
  379. s->buf[3] = count;
  380. memcpy(s->buf + 4, buf1, len);
  381. ff_rtp_send_data(s1, s->buf, len + 4, 0);
  382. size -= len;
  383. buf1 += len;
  384. count += len;
  385. }
  386. } else {
  387. if (s->buf_ptr == s->buf + 4) {
  388. /* no fragmentation possible */
  389. s->buf[0] = 0;
  390. s->buf[1] = 0;
  391. s->buf[2] = 0;
  392. s->buf[3] = 0;
  393. }
  394. memcpy(s->buf_ptr, buf1, size);
  395. s->buf_ptr += size;
  396. }
  397. }
  398. static void rtp_send_raw(AVFormatContext *s1,
  399. const uint8_t *buf1, int size)
  400. {
  401. RTPMuxContext *s = s1->priv_data;
  402. int len, max_packet_size;
  403. max_packet_size = s->max_payload_size;
  404. while (size > 0) {
  405. len = max_packet_size;
  406. if (len > size)
  407. len = size;
  408. s->timestamp = s->cur_timestamp;
  409. ff_rtp_send_data(s1, buf1, len, (len == size));
  410. buf1 += len;
  411. size -= len;
  412. }
  413. }
  414. /* NOTE: size is assumed to be an integer multiple of TS_PACKET_SIZE */
  415. static void rtp_send_mpegts_raw(AVFormatContext *s1,
  416. const uint8_t *buf1, int size)
  417. {
  418. RTPMuxContext *s = s1->priv_data;
  419. int len, out_len;
  420. s->timestamp = s->cur_timestamp;
  421. while (size >= TS_PACKET_SIZE) {
  422. len = s->max_payload_size - (s->buf_ptr - s->buf);
  423. if (len > size)
  424. len = size;
  425. memcpy(s->buf_ptr, buf1, len);
  426. buf1 += len;
  427. size -= len;
  428. s->buf_ptr += len;
  429. out_len = s->buf_ptr - s->buf;
  430. if (out_len >= s->max_payload_size) {
  431. ff_rtp_send_data(s1, s->buf, out_len, 0);
  432. s->buf_ptr = s->buf;
  433. }
  434. }
  435. }
  436. static int rtp_send_ilbc(AVFormatContext *s1, const uint8_t *buf, int size)
  437. {
  438. RTPMuxContext *s = s1->priv_data;
  439. AVStream *st = s1->streams[0];
  440. int frame_duration = av_get_audio_frame_duration2(st->codecpar, 0);
  441. int frame_size = st->codecpar->block_align;
  442. int frames = size / frame_size;
  443. while (frames > 0) {
  444. if (s->num_frames > 0 &&
  445. av_compare_ts(s->cur_timestamp - s->timestamp, st->time_base,
  446. s1->max_delay, AV_TIME_BASE_Q) >= 0) {
  447. ff_rtp_send_data(s1, s->buf, s->buf_ptr - s->buf, 1);
  448. s->num_frames = 0;
  449. }
  450. if (!s->num_frames) {
  451. s->buf_ptr = s->buf;
  452. s->timestamp = s->cur_timestamp;
  453. }
  454. memcpy(s->buf_ptr, buf, frame_size);
  455. frames--;
  456. s->num_frames++;
  457. s->buf_ptr += frame_size;
  458. buf += frame_size;
  459. s->cur_timestamp += frame_duration;
  460. if (s->num_frames == s->max_frames_per_packet) {
  461. ff_rtp_send_data(s1, s->buf, s->buf_ptr - s->buf, 1);
  462. s->num_frames = 0;
  463. }
  464. }
  465. return 0;
  466. }
  467. static int rtp_write_packet(AVFormatContext *s1, AVPacket *pkt)
  468. {
  469. RTPMuxContext *s = s1->priv_data;
  470. AVStream *st = s1->streams[0];
  471. int rtcp_bytes;
  472. int size= pkt->size;
  473. av_log(s1, AV_LOG_TRACE, "%d: write len=%d\n", pkt->stream_index, size);
  474. rtcp_bytes = ((s->octet_count - s->last_octet_count) * RTCP_TX_RATIO_NUM) /
  475. RTCP_TX_RATIO_DEN;
  476. if ((s->first_packet || ((rtcp_bytes >= RTCP_SR_SIZE) &&
  477. (ff_ntp_time() - s->last_rtcp_ntp_time > 5000000))) &&
  478. !(s->flags & FF_RTP_FLAG_SKIP_RTCP)) {
  479. rtcp_send_sr(s1, ff_ntp_time(), 0);
  480. s->last_octet_count = s->octet_count;
  481. s->first_packet = 0;
  482. }
  483. s->cur_timestamp = s->base_timestamp + pkt->pts;
  484. switch(st->codecpar->codec_id) {
  485. case AV_CODEC_ID_PCM_MULAW:
  486. case AV_CODEC_ID_PCM_ALAW:
  487. case AV_CODEC_ID_PCM_U8:
  488. case AV_CODEC_ID_PCM_S8:
  489. return rtp_send_samples(s1, pkt->data, size, 8 * st->codecpar->channels);
  490. case AV_CODEC_ID_PCM_U16BE:
  491. case AV_CODEC_ID_PCM_U16LE:
  492. case AV_CODEC_ID_PCM_S16BE:
  493. case AV_CODEC_ID_PCM_S16LE:
  494. return rtp_send_samples(s1, pkt->data, size, 16 * st->codecpar->channels);
  495. case AV_CODEC_ID_ADPCM_G722:
  496. /* The actual sample size is half a byte per sample, but since the
  497. * stream clock rate is 8000 Hz while the sample rate is 16000 Hz,
  498. * the correct parameter for send_samples_bits is 8 bits per stream
  499. * clock. */
  500. return rtp_send_samples(s1, pkt->data, size, 8 * st->codecpar->channels);
  501. case AV_CODEC_ID_ADPCM_G726:
  502. return rtp_send_samples(s1, pkt->data, size,
  503. st->codecpar->bits_per_coded_sample * st->codecpar->channels);
  504. case AV_CODEC_ID_MP2:
  505. case AV_CODEC_ID_MP3:
  506. rtp_send_mpegaudio(s1, pkt->data, size);
  507. break;
  508. case AV_CODEC_ID_MPEG1VIDEO:
  509. case AV_CODEC_ID_MPEG2VIDEO:
  510. ff_rtp_send_mpegvideo(s1, pkt->data, size);
  511. break;
  512. case AV_CODEC_ID_AAC:
  513. if (s->flags & FF_RTP_FLAG_MP4A_LATM)
  514. ff_rtp_send_latm(s1, pkt->data, size);
  515. else
  516. ff_rtp_send_aac(s1, pkt->data, size);
  517. break;
  518. case AV_CODEC_ID_AMR_NB:
  519. case AV_CODEC_ID_AMR_WB:
  520. ff_rtp_send_amr(s1, pkt->data, size);
  521. break;
  522. case AV_CODEC_ID_MPEG2TS:
  523. rtp_send_mpegts_raw(s1, pkt->data, size);
  524. break;
  525. case AV_CODEC_ID_DIRAC:
  526. ff_rtp_send_vc2hq(s1, pkt->data, size, st->codecpar->field_order != AV_FIELD_PROGRESSIVE ? 1 : 0);
  527. break;
  528. case AV_CODEC_ID_H264:
  529. ff_rtp_send_h264_hevc(s1, pkt->data, size);
  530. break;
  531. case AV_CODEC_ID_H261:
  532. ff_rtp_send_h261(s1, pkt->data, size);
  533. break;
  534. case AV_CODEC_ID_H263:
  535. if (s->flags & FF_RTP_FLAG_RFC2190) {
  536. int mb_info_size = 0;
  537. const uint8_t *mb_info =
  538. av_packet_get_side_data(pkt, AV_PKT_DATA_H263_MB_INFO,
  539. &mb_info_size);
  540. ff_rtp_send_h263_rfc2190(s1, pkt->data, size, mb_info, mb_info_size);
  541. break;
  542. }
  543. /* Fallthrough */
  544. case AV_CODEC_ID_H263P:
  545. ff_rtp_send_h263(s1, pkt->data, size);
  546. break;
  547. case AV_CODEC_ID_HEVC:
  548. ff_rtp_send_h264_hevc(s1, pkt->data, size);
  549. break;
  550. case AV_CODEC_ID_VORBIS:
  551. case AV_CODEC_ID_THEORA:
  552. ff_rtp_send_xiph(s1, pkt->data, size);
  553. break;
  554. case AV_CODEC_ID_VP8:
  555. ff_rtp_send_vp8(s1, pkt->data, size);
  556. break;
  557. case AV_CODEC_ID_VP9:
  558. ff_rtp_send_vp9(s1, pkt->data, size);
  559. break;
  560. case AV_CODEC_ID_ILBC:
  561. rtp_send_ilbc(s1, pkt->data, size);
  562. break;
  563. case AV_CODEC_ID_MJPEG:
  564. ff_rtp_send_jpeg(s1, pkt->data, size);
  565. break;
  566. case AV_CODEC_ID_OPUS:
  567. if (size > s->max_payload_size) {
  568. av_log(s1, AV_LOG_ERROR,
  569. "Packet size %d too large for max RTP payload size %d\n",
  570. size, s->max_payload_size);
  571. return AVERROR(EINVAL);
  572. }
  573. /* Intentional fallthrough */
  574. default:
  575. /* better than nothing : send the codec raw data */
  576. rtp_send_raw(s1, pkt->data, size);
  577. break;
  578. }
  579. return 0;
  580. }
  581. static int rtp_write_trailer(AVFormatContext *s1)
  582. {
  583. RTPMuxContext *s = s1->priv_data;
  584. /* If the caller closes and recreates ->pb, this might actually
  585. * be NULL here even if it was successfully allocated at the start. */
  586. if (s1->pb && (s->flags & FF_RTP_FLAG_SEND_BYE))
  587. rtcp_send_sr(s1, ff_ntp_time(), 1);
  588. av_freep(&s->buf);
  589. return 0;
  590. }
  591. AVOutputFormat ff_rtp_muxer = {
  592. .name = "rtp",
  593. .long_name = NULL_IF_CONFIG_SMALL("RTP output"),
  594. .priv_data_size = sizeof(RTPMuxContext),
  595. .audio_codec = AV_CODEC_ID_PCM_MULAW,
  596. .video_codec = AV_CODEC_ID_MPEG4,
  597. .write_header = rtp_write_header,
  598. .write_packet = rtp_write_packet,
  599. .write_trailer = rtp_write_trailer,
  600. .priv_class = &rtp_muxer_class,
  601. .flags = AVFMT_TS_NONSTRICT,
  602. };