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.

654 lines
22KB

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