Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | * WebP parser | ||
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 | /** | ||
22 | * @file | ||
23 | * WebP parser | ||
24 | */ | ||
25 | |||
26 | #include "libavutil/bswap.h" | ||
27 | #include "libavutil/common.h" | ||
28 | |||
29 | #include "parser.h" | ||
30 | |||
31 | typedef struct WebPParseContext { | ||
32 | ParseContext pc; | ||
33 | uint32_t fsize; | ||
34 | uint32_t remaining_size; | ||
35 | } WebPParseContext; | ||
36 | |||
37 | 45 | static int webp_parse(AVCodecParserContext *s, AVCodecContext *avctx, | |
38 | const uint8_t **poutbuf, int *poutbuf_size, | ||
39 | const uint8_t *buf, int buf_size) | ||
40 | { | ||
41 | 45 | WebPParseContext *ctx = s->priv_data; | |
42 | 45 | uint64_t state = ctx->pc.state64; | |
43 | 45 | int next = END_NOT_FOUND; | |
44 | 45 | int i = 0; | |
45 | |||
46 | 45 | *poutbuf = NULL; | |
47 | 45 | *poutbuf_size = 0; | |
48 | |||
49 | 16 | restart: | |
50 |
2/2✓ Branch 0 taken 32 times.
✓ Branch 1 taken 29 times.
|
61 | if (ctx->pc.frame_start_found <= 8) { |
51 |
2/2✓ Branch 0 taken 128 times.
✓ Branch 1 taken 24 times.
|
152 | for (; i < buf_size; i++) { |
52 | 128 | state = (state << 8) | buf[i]; | |
53 |
2/2✓ Branch 0 taken 64 times.
✓ Branch 1 taken 64 times.
|
128 | if (ctx->pc.frame_start_found == 0) { |
54 |
2/2✓ Branch 0 taken 8 times.
✓ Branch 1 taken 56 times.
|
64 | if ((state >> 32) == MKBETAG('R', 'I', 'F', 'F')) { |
55 | 8 | ctx->fsize = av_bswap32(state); | |
56 |
2/4✓ Branch 0 taken 8 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 8 times.
✗ Branch 3 not taken.
|
8 | if (ctx->fsize > 15 && ctx->fsize <= UINT32_MAX - 10) { |
57 | 8 | ctx->pc.frame_start_found = 1; | |
58 | 8 | ctx->fsize += 8; | |
59 | } | ||
60 | } | ||
61 |
2/2✓ Branch 0 taken 8 times.
✓ Branch 1 taken 56 times.
|
64 | } else if (ctx->pc.frame_start_found == 8) { |
62 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
|
8 | if ((state >> 32) != MKBETAG('W', 'E', 'B', 'P')) { |
63 | ✗ | ctx->pc.frame_start_found = 0; | |
64 | ✗ | continue; | |
65 | } | ||
66 | 8 | ctx->pc.frame_start_found++; | |
67 | 8 | ctx->remaining_size = ctx->fsize + i - 15; | |
68 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
|
8 | if (ctx->pc.index + i > 15) { |
69 | ✗ | next = i - 15; | |
70 | ✗ | state = 0; | |
71 | ✗ | break; | |
72 | } else { | ||
73 | 8 | ctx->pc.state64 = 0; | |
74 | 8 | goto restart; | |
75 | } | ||
76 |
1/2✓ Branch 0 taken 56 times.
✗ Branch 1 not taken.
|
56 | } else if (ctx->pc.frame_start_found) |
77 | 56 | ctx->pc.frame_start_found++; | |
78 | } | ||
79 | 24 | ctx->pc.state64 = state; | |
80 | } else { | ||
81 |
1/2✓ Branch 0 taken 29 times.
✗ Branch 1 not taken.
|
29 | if (ctx->remaining_size) { |
82 | 29 | i = FFMIN(ctx->remaining_size, buf_size); | |
83 | 29 | ctx->remaining_size -= i; | |
84 |
2/2✓ Branch 0 taken 21 times.
✓ Branch 1 taken 8 times.
|
29 | if (ctx->remaining_size) |
85 | 21 | goto flush; | |
86 | |||
87 | 8 | ctx->pc.frame_start_found = 0; | |
88 | 8 | goto restart; | |
89 | } | ||
90 | } | ||
91 | |||
92 | ✗ | flush: | |
93 |
2/2✓ Branch 1 taken 29 times.
✓ Branch 2 taken 16 times.
|
45 | if (ff_combine_frame(&ctx->pc, next, &buf, &buf_size) < 0) |
94 | 29 | return buf_size; | |
95 | |||
96 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
16 | if (next != END_NOT_FOUND && next < 0) |
97 | ✗ | ctx->pc.frame_start_found = FFMAX(ctx->pc.frame_start_found - i - 1, 0); | |
98 | else | ||
99 | 16 | ctx->pc.frame_start_found = 0; | |
100 | |||
101 | 16 | *poutbuf = buf; | |
102 | 16 | *poutbuf_size = buf_size; | |
103 | |||
104 | 16 | return next; | |
105 | } | ||
106 | |||
107 | const AVCodecParser ff_webp_parser = { | ||
108 | .codec_ids = { AV_CODEC_ID_WEBP }, | ||
109 | .priv_data_size = sizeof(WebPParseContext), | ||
110 | .parser_parse = webp_parse, | ||
111 | .parser_close = ff_parse_close, | ||
112 | }; | ||
113 |