avcodec/mpeg4_unpack_bframes_bsf: Copy packet props
[ffmpeg.git] / libavcodec / dump_extradata_bsf.c
1 /*
2  * copyright (c) 2006 Michael Niedermayer <michaelni@gmx.at>
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * FFmpeg is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20
21 #include <string.h>
22
23 #include "avcodec.h"
24 #include "bsf.h"
25
26 #include "libavutil/log.h"
27 #include "libavutil/mem.h"
28 #include "libavutil/opt.h"
29
30 enum DumpFreq {
31     DUMP_FREQ_KEYFRAME,
32     DUMP_FREQ_ALL,
33 };
34
35 typedef struct DumpExtradataContext {
36     const AVClass *class;
37     int freq;
38 } DumpExtradataContext;
39
40 static int dump_extradata(AVBSFContext *ctx, AVPacket *out)
41 {
42     DumpExtradataContext *s = ctx->priv_data;
43     AVPacket *in;
44     int ret = 0;
45
46     ret = ff_bsf_get_packet(ctx, &in);
47     if (ret < 0)
48         return ret;
49
50     if (ctx->par_in->extradata &&
51         (s->freq == DUMP_FREQ_ALL ||
52          (s->freq == DUMP_FREQ_KEYFRAME && in->flags & AV_PKT_FLAG_KEY))) {
53         if (in->size >= INT_MAX - ctx->par_in->extradata_size) {
54             ret = AVERROR(ERANGE);
55             goto fail;
56         }
57
58         ret = av_new_packet(out, in->size + ctx->par_in->extradata_size);
59         if (ret < 0)
60             goto fail;
61
62         ret = av_packet_copy_props(out, in);
63         if (ret < 0) {
64             av_packet_unref(out);
65             goto fail;
66         }
67
68         memcpy(out->data, ctx->par_in->extradata, ctx->par_in->extradata_size);
69         memcpy(out->data + ctx->par_in->extradata_size, in->data, in->size);
70     } else {
71         av_packet_move_ref(out, in);
72     }
73
74 fail:
75     av_packet_free(&in);
76
77     return ret;
78 }
79
80 #define OFFSET(x) offsetof(DumpExtradataContext, x)
81 static const AVOption options[] = {
82     { "freq", "When do dump extradata", OFFSET(freq), AV_OPT_TYPE_INT,
83         { .i64 = DUMP_FREQ_KEYFRAME }, DUMP_FREQ_KEYFRAME, DUMP_FREQ_ALL, 0, "freq" },
84         { "k",        NULL, 0, AV_OPT_TYPE_CONST, { .i64 = DUMP_FREQ_KEYFRAME }, .unit = "freq" },
85         { "keyframe", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = DUMP_FREQ_KEYFRAME }, .unit = "freq" },
86         { "e",        NULL, 0, AV_OPT_TYPE_CONST, { .i64 = DUMP_FREQ_ALL      }, .unit = "freq" },
87         { "all",      NULL, 0, AV_OPT_TYPE_CONST, { .i64 = DUMP_FREQ_ALL      }, .unit = "freq" },
88     { NULL },
89 };
90
91 static const AVClass dump_extradata_class = {
92     .class_name = "dump_extradata bsf",
93     .item_name  = av_default_item_name,
94     .option     = options,
95     .version    = LIBAVUTIL_VERSION_MAJOR,
96 };
97
98 const AVBitStreamFilter ff_dump_extradata_bsf = {
99     .name           = "dump_extra",
100     .priv_data_size = sizeof(DumpExtradataContext),
101     .priv_class     = &dump_extradata_class,
102     .filter         = dump_extradata,
103 };