LCOV - code coverage report
Current view: top level - libavformat - rtpdec_vc2hq.c (source / functions) Hit Total Coverage
Test: coverage.info Lines: 0 90 0.0 %
Date: 2017-12-13 18:07:29 Functions: 0 5 0.0 %

          Line data    Source code
       1             : /*
       2             :  * RTP parser for VC-2 HQ payload format (draft version 1) - experimental
       3             :  * Copyright (c) 2016 Thomas Volkert <thomas@netzeal.de>
       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             : 
      22             : #include "libavutil/intreadwrite.h"
      23             : #include "libavcodec/dirac.h"
      24             : 
      25             : #include "avio_internal.h"
      26             : #include "rtpdec_formats.h"
      27             : 
      28             : #define RTP_VC2HQ_PL_HEADER_SIZE             4
      29             : 
      30             : #define DIRAC_DATA_UNIT_HEADER_SIZE          13
      31             : #define DIRAC_PIC_NR_SIZE                    4
      32             : #define DIRAC_RTP_PCODE_HQ_PIC_FRAGMENT      0xEC
      33             : 
      34             : struct PayloadContext {
      35             :     AVIOContext *buf;
      36             :     uint32_t    frame_size;
      37             :     uint32_t    frame_nr;
      38             :     uint32_t    timestamp;
      39             :     uint32_t    last_unit_size;
      40             :     int         seen_sequence_header;
      41             : };
      42             : 
      43             : static const uint8_t start_sequence[] = { 'B', 'B', 'C', 'D' };
      44             : 
      45           0 : static void fill_parse_info_header(PayloadContext *pl_ctx, uint8_t *buf,
      46             :                                    uint8_t parse_code, uint32_t data_unit_size)
      47             : {
      48           0 :     memcpy(buf, start_sequence, sizeof(start_sequence));
      49           0 :     buf[4]  = parse_code;
      50           0 :     AV_WB32(&buf[5], data_unit_size);
      51           0 :     AV_WB32(&buf[9], pl_ctx->last_unit_size);
      52             : 
      53           0 :     pl_ctx->last_unit_size = data_unit_size;
      54           0 : }
      55             : 
      56           0 : static int vc2hq_handle_sequence_header(PayloadContext *pl_ctx, AVStream *st, AVPacket *pkt,
      57             :                                         const uint8_t *buf, int len)
      58             : {
      59             :     int res;
      60           0 :     uint32_t size = DIRAC_DATA_UNIT_HEADER_SIZE + len;
      61             : 
      62           0 :     if ((res = av_new_packet(pkt, DIRAC_DATA_UNIT_HEADER_SIZE + len)) < 0)
      63           0 :         return res;
      64             : 
      65           0 :     fill_parse_info_header(pl_ctx, pkt->data, 0x00, size);
      66             :     /* payload of seq. header */
      67           0 :     memcpy(pkt->data + DIRAC_DATA_UNIT_HEADER_SIZE, buf, len);
      68           0 :     pkt->stream_index = st->index;
      69             : 
      70           0 :     pl_ctx->seen_sequence_header = 1;
      71             : 
      72           0 :     return 0;
      73             : }
      74             : 
      75           0 : static int vc2hq_mark_end_of_sequence(PayloadContext *pl_ctx, AVStream *st, AVPacket *pkt)
      76             : {
      77             :     int res;
      78           0 :     uint32_t size = 0;
      79             : 
      80             :     /* create A/V packet */
      81           0 :     if ((res = av_new_packet(pkt, DIRAC_DATA_UNIT_HEADER_SIZE)) < 0)
      82           0 :         return res;
      83             : 
      84           0 :     fill_parse_info_header(pl_ctx, pkt->data, 0x10, size);
      85           0 :     pkt->stream_index = st->index;
      86             : 
      87           0 :     pl_ctx->seen_sequence_header = 0;
      88             : 
      89           0 :     return 0;
      90             : }
      91             : 
      92           0 : static int vc2hq_handle_frame_fragment(AVFormatContext *ctx, PayloadContext *pl_ctx, AVStream *st,
      93             :                                        AVPacket *pkt, uint32_t *timestamp, const uint8_t *buf, int len,
      94             :                                        int flags)
      95             : {
      96             :     int res;
      97             :     uint32_t pic_nr;
      98             :     uint16_t frag_len;
      99             :     uint16_t no_slices;
     100             : 
     101             :     /* sanity check for size of input packet: 16 bytes header in any case as minimum */
     102           0 :     if (len < 16) {
     103           0 :         av_log(ctx, AV_LOG_ERROR, "Too short RTP/VC2hq packet, got %d bytes\n", len);
     104           0 :         return AVERROR_INVALIDDATA;
     105             :     }
     106             : 
     107           0 :     pic_nr = AV_RB32(&buf[4]);
     108           0 :     frag_len = AV_RB16(&buf[12]);
     109           0 :     no_slices = AV_RB16(&buf[14]);
     110             : 
     111           0 :     if (pl_ctx->buf && pl_ctx->frame_nr != pic_nr) {
     112           0 :         av_log(ctx, AV_LOG_WARNING, "Dropping buffered RTP/VC2hq packet fragments - non-continuous picture numbers\n");
     113           0 :         ffio_free_dyn_buf(&pl_ctx->buf);
     114             :     }
     115             : 
     116             :     /* transform parameters? */
     117           0 :     if (no_slices == 0) {
     118           0 :         if (len < frag_len + 16) {
     119           0 :             av_log(ctx, AV_LOG_ERROR, "Too short RTP/VC2hq packet, got %d bytes\n", len);
     120           0 :             return AVERROR_INVALIDDATA;
     121             :         }
     122             : 
     123             :         /* start frame buffering with new dynamic buffer */
     124           0 :         if (!pl_ctx->buf) {
     125             : 
     126           0 :             res = avio_open_dyn_buf(&pl_ctx->buf);
     127           0 :             if (res < 0)
     128           0 :                 return res;
     129             : 
     130             :             /* reserve memory for frame header */
     131           0 :             res = avio_seek(pl_ctx->buf, DIRAC_DATA_UNIT_HEADER_SIZE + DIRAC_PIC_NR_SIZE, SEEK_SET);
     132           0 :             if (res < 0)
     133           0 :                 return res;
     134             : 
     135           0 :             pl_ctx->frame_nr = pic_nr;
     136           0 :             pl_ctx->timestamp = *timestamp;
     137           0 :             pl_ctx->frame_size = DIRAC_DATA_UNIT_HEADER_SIZE + DIRAC_PIC_NR_SIZE;
     138             :         }
     139             : 
     140           0 :         avio_write(pl_ctx->buf, buf + 16 /* skip pl header */, frag_len);
     141           0 :         pl_ctx->frame_size += frag_len;
     142             : 
     143           0 :         return AVERROR(EAGAIN);
     144             :     } else {
     145           0 :         if (len < frag_len + 20) {
     146           0 :             av_log(ctx, AV_LOG_ERROR, "Too short RTP/VC2hq packet, got %d bytes\n", len);
     147           0 :             return AVERROR_INVALIDDATA;
     148             :         }
     149             : 
     150             :         /* transform parameters were missed, no buffer available */
     151           0 :         if (!pl_ctx->buf)
     152           0 :             return AVERROR_INVALIDDATA;
     153             : 
     154           0 :         avio_write(pl_ctx->buf, buf + 20 /* skip pl header */, frag_len);
     155           0 :         pl_ctx->frame_size += frag_len;
     156             : 
     157             :         /* RTP marker bit means: last fragment of current frame was received;
     158             :            otherwise, an additional fragment is needed for the current frame */
     159           0 :         if (!(flags & RTP_FLAG_MARKER))
     160           0 :             return AVERROR(EAGAIN);
     161             :     }
     162             : 
     163             :     /* close frame buffering and create A/V packet */
     164           0 :     res = ff_rtp_finalize_packet(pkt, &pl_ctx->buf, st->index);
     165           0 :     if (res < 0)
     166           0 :         return res;
     167             : 
     168           0 :     fill_parse_info_header(pl_ctx, pkt->data, DIRAC_PCODE_PICTURE_HQ, pl_ctx->frame_size);
     169           0 :     AV_WB32(&pkt->data[13], pl_ctx->frame_nr);
     170             : 
     171           0 :     pl_ctx->frame_size = 0;
     172             : 
     173           0 :     return 0;
     174             : }
     175             : 
     176           0 : static int vc2hq_handle_packet(AVFormatContext *ctx, PayloadContext *pl_ctx,
     177             :                                AVStream *st, AVPacket *pkt, uint32_t *timestamp,
     178             :                                const uint8_t *buf, int len, uint16_t seq,
     179             :                                int flags)
     180             : {
     181           0 :     uint8_t parse_code = 0;
     182           0 :     int res = 0;
     183             : 
     184           0 :     if (pl_ctx->buf && pl_ctx->timestamp != *timestamp) {
     185           0 :         av_log(ctx, AV_LOG_WARNING, "Dropping buffered RTP/VC2hq packet fragments - non-continuous timestamps\n");
     186           0 :         ffio_free_dyn_buf(&pl_ctx->buf);
     187           0 :         pl_ctx->frame_size = 0;
     188             :     }
     189             : 
     190             :     /* sanity check for size of input packet: needed header data as minimum */
     191           0 :     if (len < RTP_VC2HQ_PL_HEADER_SIZE) {
     192           0 :         av_log(ctx, AV_LOG_ERROR, "Too short RTP/VC2hq packet, got %d bytes\n", len);
     193           0 :         return AVERROR_INVALIDDATA;
     194             :     }
     195             : 
     196           0 :     parse_code = buf[3];
     197             : 
     198             :     /* wait for next sequence header? */
     199           0 :     if (pl_ctx->seen_sequence_header || parse_code == DIRAC_PCODE_SEQ_HEADER) {
     200           0 :         switch(parse_code) {
     201             :         /* sequence header */
     202           0 :         case DIRAC_PCODE_SEQ_HEADER:
     203           0 :             res = vc2hq_handle_sequence_header(pl_ctx, st, pkt, buf + RTP_VC2HQ_PL_HEADER_SIZE, len - RTP_VC2HQ_PL_HEADER_SIZE);
     204           0 :             break;
     205             :         /* end of sequence */
     206           0 :         case DIRAC_PCODE_END_SEQ:
     207           0 :             res = vc2hq_mark_end_of_sequence(pl_ctx, st, pkt);
     208           0 :             break;
     209             :         /* HQ picture fragment */
     210           0 :         case DIRAC_RTP_PCODE_HQ_PIC_FRAGMENT:
     211           0 :             res = vc2hq_handle_frame_fragment(ctx, pl_ctx, st, pkt, timestamp, buf, len, flags);
     212           0 :             break;
     213             :         }
     214             :     }
     215             : 
     216           0 :     return res;
     217             : }
     218             : 
     219             : RTPDynamicProtocolHandler ff_vc2hq_dynamic_handler = {
     220             :     .enc_name         = "VC2",
     221             :     .codec_type       = AVMEDIA_TYPE_VIDEO,
     222             :     .codec_id         = AV_CODEC_ID_DIRAC,
     223             :     .priv_data_size   = sizeof(PayloadContext),
     224             :     .parse_packet     = vc2hq_handle_packet
     225             : };

Generated by: LCOV version 1.13