LCOV - code coverage report
Current view: top level - libavformat - fifo_test.c (source / functions) Hit Total Coverage
Test: coverage.info Lines: 36 37 97.3 %
Date: 2018-05-20 11:54:08 Functions: 4 4 100.0 %

          Line data    Source code
       1             : /*
       2             :  * FIFO test pseudo-muxer
       3             :  * Copyright (c) 2016 Jan Sebechlebsky
       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 License
       9             :  * 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
      15             :  * GNU Lesser General Public License for more details.
      16             :  *
      17             :  * You should have received a copy of the GNU Lesser General Public License
      18             :  * along with FFmpeg; if not, write to the Free Software * Foundation, Inc.,
      19             :  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
      20             :  */
      21             : 
      22             : #include <stdlib.h>
      23             : 
      24             : #include "libavutil/opt.h"
      25             : #include "libavutil/time.h"
      26             : #include "libavutil/avassert.h"
      27             : 
      28             : #include "avformat.h"
      29             : #include "url.h"
      30             : 
      31             : /* Implementation of mock muxer to simulate real muxer failures */
      32             : 
      33             : #define MAX_TST_PACKETS 128
      34             : #define SLEEPTIME_50_MS 50000
      35             : #define SLEEPTIME_10_MS 10000
      36             : 
      37             : /* Implementation of mock muxer to simulate real muxer failures */
      38             : 
      39             : /* This is structure of data sent in packets to
      40             :  * failing muxer */
      41             : typedef struct FailingMuxerPacketData {
      42             :     int ret;             /* return value of write_packet call*/
      43             :     int recover_after;   /* set ret to zero after this number of recovery attempts */
      44             :     unsigned sleep_time; /* sleep for this long in write_packet to simulate long I/O operation */
      45             : } FailingMuxerPacketData;
      46             : 
      47             : 
      48             : typedef struct FailingMuxerContext {
      49             :     AVClass *class;
      50             :     int write_header_ret;
      51             :     int write_trailer_ret;
      52             :     /* If non-zero, summary of processed packets will be printed in deinit */
      53             :     int print_deinit_summary;
      54             : 
      55             :     int flush_count;
      56             :     int pts_written[MAX_TST_PACKETS];
      57             :     int pts_written_nr;
      58             : } FailingMuxerContext;
      59             : 
      60          49 : static int failing_write_header(AVFormatContext *avf)
      61             : {
      62          49 :     FailingMuxerContext *ctx = avf->priv_data;
      63          49 :     return ctx->write_header_ret;
      64             : }
      65             : 
      66          93 : static int failing_write_packet(AVFormatContext *avf, AVPacket *pkt)
      67             : {
      68          93 :     FailingMuxerContext *ctx = avf->priv_data;
      69          93 :     int ret = 0;
      70          93 :     if (!pkt) {
      71           3 :         ctx->flush_count++;
      72             :     } else {
      73          90 :         FailingMuxerPacketData *data = (FailingMuxerPacketData*) pkt->data;
      74             : 
      75          90 :         if (!data->recover_after) {
      76          45 :             data->ret = 0;
      77             :         } else {
      78          45 :             data->recover_after--;
      79             :         }
      80             : 
      81          90 :         ret = data->ret;
      82             : 
      83          90 :         if (data->sleep_time) {
      84          15 :             int64_t slept = 0;
      85          45 :             while (slept < data->sleep_time) {
      86          15 :                 if (ff_check_interrupt(&avf->interrupt_callback))
      87           0 :                     return AVERROR_EXIT;
      88          15 :                 av_usleep(SLEEPTIME_10_MS);
      89          15 :                 slept += SLEEPTIME_10_MS;
      90             :             }
      91             :         }
      92             : 
      93          90 :         if (!ret) {
      94          45 :             ctx->pts_written[ctx->pts_written_nr++] = pkt->pts;
      95          45 :             av_packet_unref(pkt);
      96             :         }
      97             :     }
      98          93 :     return ret;
      99             : }
     100             : 
     101          49 : static int failing_write_trailer(AVFormatContext *avf)
     102             : {
     103          49 :     FailingMuxerContext *ctx = avf->priv_data;
     104          49 :     return ctx->write_trailer_ret;
     105             : }
     106             : 
     107          49 : static void failing_deinit(AVFormatContext *avf)
     108             : {
     109             :     int i;
     110          49 :     FailingMuxerContext *ctx = avf->priv_data;
     111             : 
     112          49 :     if (!ctx->print_deinit_summary)
     113          47 :         return;
     114             : 
     115           2 :     printf("flush count: %d\n", ctx->flush_count);
     116           2 :     printf("pts seen nr: %d\n", ctx->pts_written_nr);
     117           2 :     printf("pts seen: ");
     118          32 :     for (i = 0; i < ctx->pts_written_nr; ++i ) {
     119          30 :         printf(i ? ",%d" : "%d", ctx->pts_written[i]);
     120             :     }
     121           2 :     printf("\n");
     122             : }
     123             : #define OFFSET(x) offsetof(FailingMuxerContext, x)
     124             : static const AVOption options[] = {
     125             :         {"write_header_ret", "write_header() return value", OFFSET(write_header_ret),
     126             :          AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM},
     127             :         {"write_trailer_ret", "write_trailer() return value", OFFSET(write_trailer_ret),
     128             :          AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM},
     129             :         {"print_deinit_summary", "print summary when deinitializing muxer", OFFSET(print_deinit_summary),
     130             :          AV_OPT_TYPE_BOOL, {.i64 = 1}, 0, 1, AV_OPT_FLAG_ENCODING_PARAM},
     131             :         {NULL}
     132             :     };
     133             : 
     134             : static const AVClass failing_muxer_class = {
     135             :     .class_name = "Fifo test muxer",
     136             :     .item_name  = av_default_item_name,
     137             :     .option     = options,
     138             :     .version    = LIBAVUTIL_VERSION_INT,
     139             : };
     140             : 
     141             : AVOutputFormat ff_fifo_test_muxer = {
     142             :     .name           = "fifo_test",
     143             :     .long_name      = NULL_IF_CONFIG_SMALL("Fifo test muxer"),
     144             :     .priv_data_size = sizeof(FailingMuxerContext),
     145             :     .write_header   = failing_write_header,
     146             :     .write_packet   = failing_write_packet,
     147             :     .write_trailer  = failing_write_trailer,
     148             :     .deinit         = failing_deinit,
     149             :     .priv_class     = &failing_muxer_class,
     150             :     .flags          = AVFMT_NOFILE | AVFMT_ALLOW_FLUSH,
     151             : };
     152             : 

Generated by: LCOV version 1.13