yading@11: /* yading@11: * RTP/Quicktime support. yading@11: * Copyright (c) 2009 Ronald S. Bultje yading@11: * yading@11: * This file is part of FFmpeg. yading@11: * yading@11: * FFmpeg is free software; you can redistribute it and/or yading@11: * modify it under the terms of the GNU Lesser General Public yading@11: * License as published by the Free Software Foundation; either yading@11: * version 2.1 of the License, or (at your option) any later version. yading@11: * yading@11: * FFmpeg is distributed in the hope that it will be useful, yading@11: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@11: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@11: * Lesser General Public License for more details. yading@11: * yading@11: * You should have received a copy of the GNU Lesser General Public yading@11: * License along with FFmpeg; if not, write to the Free Software yading@11: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@11: */ yading@11: yading@11: /** yading@11: * @file yading@11: * @brief Quicktime-style RTP support yading@11: * @author Ronald S. Bultje yading@11: */ yading@11: yading@11: #include "avformat.h" yading@11: #include "internal.h" yading@11: #include "avio_internal.h" yading@11: #include "rtp.h" yading@11: #include "rtpdec.h" yading@11: #include "isom.h" yading@11: #include "libavcodec/get_bits.h" yading@11: yading@11: struct PayloadContext { yading@11: AVPacket pkt; yading@11: int bytes_per_frame, remaining; yading@11: uint32_t timestamp; yading@11: }; yading@11: yading@11: static int qt_rtp_parse_packet(AVFormatContext *s, PayloadContext *qt, yading@11: AVStream *st, AVPacket *pkt, yading@11: uint32_t *timestamp, const uint8_t *buf, yading@11: int len, uint16_t seq, int flags) yading@11: { yading@11: AVIOContext pb; yading@11: GetBitContext gb; yading@11: int packing_scheme, has_payload_desc, has_packet_info, alen, yading@11: has_marker_bit = flags & RTP_FLAG_MARKER; yading@11: yading@11: if (qt->remaining) { yading@11: int num = qt->pkt.size / qt->bytes_per_frame; yading@11: yading@11: if (av_new_packet(pkt, qt->bytes_per_frame)) yading@11: return AVERROR(ENOMEM); yading@11: pkt->stream_index = st->index; yading@11: pkt->flags = qt->pkt.flags; yading@11: memcpy(pkt->data, yading@11: &qt->pkt.data[(num - qt->remaining) * qt->bytes_per_frame], yading@11: qt->bytes_per_frame); yading@11: if (--qt->remaining == 0) { yading@11: av_freep(&qt->pkt.data); yading@11: qt->pkt.size = 0; yading@11: } yading@11: return qt->remaining > 0; yading@11: } yading@11: yading@11: /** yading@11: * The RTP payload is described in: yading@11: * http://developer.apple.com/quicktime/icefloe/dispatch026.html yading@11: */ yading@11: init_get_bits(&gb, buf, len << 3); yading@11: ffio_init_context(&pb, buf, len, 0, NULL, NULL, NULL, NULL); yading@11: yading@11: if (len < 4) yading@11: return AVERROR_INVALIDDATA; yading@11: yading@11: skip_bits(&gb, 4); // version yading@11: if ((packing_scheme = get_bits(&gb, 2)) == 0) yading@11: return AVERROR_INVALIDDATA; yading@11: if (get_bits1(&gb)) yading@11: flags |= RTP_FLAG_KEY; yading@11: has_payload_desc = get_bits1(&gb); yading@11: has_packet_info = get_bits1(&gb); yading@11: skip_bits(&gb, 23); // reserved:7, cache payload info:1, payload ID:15 yading@11: yading@11: if (has_payload_desc) { yading@11: int data_len, pos, is_start, is_finish; yading@11: uint32_t tag; yading@11: yading@11: pos = get_bits_count(&gb) >> 3; yading@11: if (pos + 12 > len) yading@11: return AVERROR_INVALIDDATA; yading@11: yading@11: skip_bits(&gb, 2); // has non-I frames:1, is sparse:1 yading@11: is_start = get_bits1(&gb); yading@11: is_finish = get_bits1(&gb); yading@11: if (!is_start || !is_finish) { yading@11: avpriv_request_sample(s, "RTP-X-QT with payload description " yading@11: "split over several packets"); yading@11: return AVERROR_PATCHWELCOME; yading@11: } yading@11: skip_bits(&gb, 12); // reserved yading@11: data_len = get_bits(&gb, 16); yading@11: yading@11: avio_seek(&pb, pos + 4, SEEK_SET); yading@11: tag = avio_rl32(&pb); yading@11: if ((st->codec->codec_type == AVMEDIA_TYPE_VIDEO && yading@11: tag != MKTAG('v','i','d','e')) || yading@11: (st->codec->codec_type == AVMEDIA_TYPE_AUDIO && yading@11: tag != MKTAG('s','o','u','n'))) yading@11: return AVERROR_INVALIDDATA; yading@11: avpriv_set_pts_info(st, 32, 1, avio_rb32(&pb)); yading@11: yading@11: if (pos + data_len > len) yading@11: return AVERROR_INVALIDDATA; yading@11: /* TLVs */ yading@11: while (avio_tell(&pb) + 4 < pos + data_len) { yading@11: int tlv_len = avio_rb16(&pb); yading@11: tag = avio_rl16(&pb); yading@11: if (avio_tell(&pb) + tlv_len > pos + data_len) yading@11: return AVERROR_INVALIDDATA; yading@11: yading@11: #define MKTAG16(a,b) MKTAG(a,b,0,0) yading@11: switch (tag) { yading@11: case MKTAG16('s','d'): { yading@11: MOVStreamContext *msc; yading@11: void *priv_data = st->priv_data; yading@11: int nb_streams = s->nb_streams; yading@11: MOVContext *mc = av_mallocz(sizeof(*mc)); yading@11: if (!mc) yading@11: return AVERROR(ENOMEM); yading@11: mc->fc = s; yading@11: st->priv_data = msc = av_mallocz(sizeof(MOVStreamContext)); yading@11: if (!msc) { yading@11: av_free(mc); yading@11: st->priv_data = priv_data; yading@11: return AVERROR(ENOMEM); yading@11: } yading@11: /* ff_mov_read_stsd_entries updates stream s->nb_streams-1, yading@11: * so set it temporarily to indicate which stream to update. */ yading@11: s->nb_streams = st->index + 1; yading@11: ff_mov_read_stsd_entries(mc, &pb, 1); yading@11: qt->bytes_per_frame = msc->bytes_per_frame; yading@11: av_free(msc); yading@11: av_free(mc); yading@11: st->priv_data = priv_data; yading@11: s->nb_streams = nb_streams; yading@11: break; yading@11: } yading@11: default: yading@11: avio_skip(&pb, tlv_len); yading@11: break; yading@11: } yading@11: } yading@11: yading@11: /* 32-bit alignment */ yading@11: avio_skip(&pb, ((avio_tell(&pb) + 3) & ~3) - avio_tell(&pb)); yading@11: } else yading@11: avio_seek(&pb, 4, SEEK_SET); yading@11: yading@11: if (has_packet_info) { yading@11: avpriv_request_sample(s, "RTP-X-QT with packet-specific info"); yading@11: return AVERROR_PATCHWELCOME; yading@11: } yading@11: yading@11: alen = len - avio_tell(&pb); yading@11: if (alen <= 0) yading@11: return AVERROR_INVALIDDATA; yading@11: yading@11: switch (packing_scheme) { yading@11: case 3: /* one data packet spread over 1 or multiple RTP packets */ yading@11: if (qt->pkt.size > 0 && qt->timestamp == *timestamp) { yading@11: qt->pkt.data = av_realloc(qt->pkt.data, qt->pkt.size + alen + yading@11: FF_INPUT_BUFFER_PADDING_SIZE); yading@11: } else { yading@11: av_freep(&qt->pkt.data); yading@11: av_init_packet(&qt->pkt); yading@11: qt->pkt.data = av_malloc(alen + FF_INPUT_BUFFER_PADDING_SIZE); yading@11: qt->pkt.size = 0; yading@11: qt->timestamp = *timestamp; yading@11: } yading@11: if (!qt->pkt.data) yading@11: return AVERROR(ENOMEM); yading@11: memcpy(qt->pkt.data + qt->pkt.size, buf + avio_tell(&pb), alen); yading@11: qt->pkt.size += alen; yading@11: if (has_marker_bit) { yading@11: int ret = av_packet_from_data(pkt, qt->pkt.data, qt->pkt.size); yading@11: if (ret < 0) yading@11: return ret; yading@11: yading@11: qt->pkt.size = 0; yading@11: qt->pkt.data = NULL; yading@11: pkt->flags = flags & RTP_FLAG_KEY ? AV_PKT_FLAG_KEY : 0; yading@11: pkt->stream_index = st->index; yading@11: memset(pkt->data + pkt->size, 0, FF_INPUT_BUFFER_PADDING_SIZE); yading@11: return 0; yading@11: } yading@11: return AVERROR(EAGAIN); yading@11: yading@11: case 1: /* constant packet size, multiple packets per RTP packet */ yading@11: if (qt->bytes_per_frame == 0 || yading@11: alen % qt->bytes_per_frame != 0) yading@11: return AVERROR_INVALIDDATA; /* wrongly padded */ yading@11: qt->remaining = (alen / qt->bytes_per_frame) - 1; yading@11: if (av_new_packet(pkt, qt->bytes_per_frame)) yading@11: return AVERROR(ENOMEM); yading@11: memcpy(pkt->data, buf + avio_tell(&pb), qt->bytes_per_frame); yading@11: pkt->flags = flags & RTP_FLAG_KEY ? AV_PKT_FLAG_KEY : 0; yading@11: pkt->stream_index = st->index; yading@11: if (qt->remaining > 0) { yading@11: av_freep(&qt->pkt.data); yading@11: qt->pkt.data = av_malloc(qt->remaining * qt->bytes_per_frame); yading@11: if (!qt->pkt.data) { yading@11: av_free_packet(pkt); yading@11: return AVERROR(ENOMEM); yading@11: } yading@11: qt->pkt.size = qt->remaining * qt->bytes_per_frame; yading@11: memcpy(qt->pkt.data, yading@11: buf + avio_tell(&pb) + qt->bytes_per_frame, yading@11: qt->remaining * qt->bytes_per_frame); yading@11: qt->pkt.flags = pkt->flags; yading@11: return 1; yading@11: } yading@11: return 0; yading@11: yading@11: default: /* unimplemented */ yading@11: avpriv_request_sample(NULL, "RTP-X-QT with packing scheme 2"); yading@11: return AVERROR_PATCHWELCOME; yading@11: } yading@11: } yading@11: yading@11: static PayloadContext *qt_rtp_new(void) yading@11: { yading@11: return av_mallocz(sizeof(PayloadContext)); yading@11: } yading@11: yading@11: static void qt_rtp_free(PayloadContext *qt) yading@11: { yading@11: av_freep(&qt->pkt.data); yading@11: av_free(qt); yading@11: } yading@11: yading@11: #define RTP_QT_HANDLER(m, n, s, t) \ yading@11: RTPDynamicProtocolHandler ff_ ## m ## _rtp_ ## n ## _handler = { \ yading@11: .enc_name = s, \ yading@11: .codec_type = t, \ yading@11: .codec_id = AV_CODEC_ID_NONE, \ yading@11: .alloc = qt_rtp_new, \ yading@11: .free = qt_rtp_free, \ yading@11: .parse_packet = qt_rtp_parse_packet, \ yading@11: } yading@11: yading@11: RTP_QT_HANDLER(qt, vid, "X-QT", AVMEDIA_TYPE_VIDEO); yading@11: RTP_QT_HANDLER(qt, aud, "X-QT", AVMEDIA_TYPE_AUDIO); yading@11: RTP_QT_HANDLER(quicktime, vid, "X-QUICKTIME", AVMEDIA_TYPE_VIDEO); yading@11: RTP_QT_HANDLER(quicktime, aud, "X-QUICKTIME", AVMEDIA_TYPE_AUDIO);