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.

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