FFmpeg coverage

GCC Code Coverage Report

Directory:	../../../ffmpeg/		Exec	Total	Coverage
File:	src/libavcodec/ws-snd1.c	Lines:	78	89	87.6 %
Date:	2021-04-15 16:04:23	Branches:	32	40	80.0 %


/*
 * Westwood SNDx codecs
 * Copyright (c) 2005 Konstantin Shishkov
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include <stdint.h>

#include "libavutil/channel_layout.h"
#include "libavutil/common.h"
#include "libavutil/intreadwrite.h"
#include "avcodec.h"
#include "internal.h"

/**
 * @file
 * Westwood SNDx codecs
 *
 * Reference documents about VQA format and its audio codecs
 * can be found here:
 * http://www.multimedia.cx
 */

static const int8_t ws_adpcm_4bit[] = {
    -9, -8, -6, -5, -4, -3, -2, -1,
     0,  1,  2,  3,  4,  5,  6,  8
};

static av_cold int ws_snd_decode_init(AVCodecContext *avctx)
{
    avctx->channels       = 1;
    avctx->channel_layout = AV_CH_LAYOUT_MONO;
    avctx->sample_fmt     = AV_SAMPLE_FMT_U8;

    return 0;
}

static int ws_snd_decode_frame(AVCodecContext *avctx, void *data,
                               int *got_frame_ptr, AVPacket *avpkt)
{
    AVFrame *frame     = data;
    const uint8_t *buf = avpkt->data;
    int buf_size       = avpkt->size;

    int in_size, out_size, ret;
    int sample = 128;
    uint8_t *samples;
    uint8_t *samples_end;

    if (!buf_size)
        return 0;

    if (buf_size < 4) {
        av_log(avctx, AV_LOG_ERROR, "packet is too small\n");
        return AVERROR(EINVAL);
    }

    out_size = AV_RL16(&buf[0]);
    in_size  = AV_RL16(&buf[2]);
    buf += 4;

    if (in_size > buf_size) {
        av_log(avctx, AV_LOG_ERROR, "Frame data is larger than input buffer\n");
        return AVERROR_INVALIDDATA;
    }

    /* get output buffer */
    frame->nb_samples = out_size;
    if ((ret = ff_get_buffer(avctx, frame, 0)) < 0)
        return ret;
    samples     = frame->data[0];
    samples_end = samples + out_size;

    if (in_size == out_size) {
        memcpy(samples, buf, out_size);
        *got_frame_ptr = 1;
        return buf_size;
    }

    while (samples < samples_end && buf - avpkt->data < buf_size) {
        int code, smp, size;
        uint8_t count;
        code  = *buf >> 6;
        count = *buf & 0x3F;
        buf++;

        /* make sure we don't write past the output buffer */
        switch (code) {
        case 0:  smp = 4 * (count + 1);                break;
        case 1:  smp = 2 * (count + 1);                break;
        case 2:  smp = (count & 0x20) ? 1 : count + 1; break;
        default: smp = count + 1;                      break;
        }
        if (samples_end - samples < smp)
            break;

        /* make sure we don't read past the input buffer */
        size = ((code == 2 && (count & 0x20)) || code == 3) ? 0 : count + 1;
        if ((buf - avpkt->data) + size > buf_size)
            break;

        switch (code) {
        case 0: /* ADPCM 2-bit */
            for (count++; count > 0; count--) {
                code = *buf++;
                sample += ( code       & 0x3) - 2;
                sample = av_clip_uint8(sample);
                *samples++ = sample;
                sample += ((code >> 2) & 0x3) - 2;
                sample = av_clip_uint8(sample);
                *samples++ = sample;
                sample += ((code >> 4) & 0x3) - 2;
                sample = av_clip_uint8(sample);
                *samples++ = sample;
                sample +=  (code >> 6)        - 2;
                sample = av_clip_uint8(sample);
                *samples++ = sample;
            }
            break;
        case 1: /* ADPCM 4-bit */
            for (count++; count > 0; count--) {
                code = *buf++;
                sample += ws_adpcm_4bit[code & 0xF];
                sample = av_clip_uint8(sample);
                *samples++ = sample;
                sample += ws_adpcm_4bit[code >> 4];
                sample = av_clip_uint8(sample);
                *samples++ = sample;
            }
            break;
        case 2: /* no compression */
            if (count & 0x20) { /* big delta */
                int8_t t;
                t = count;
                t <<= 3;
                sample += t >> 3;
                sample = av_clip_uint8(sample);
                *samples++ = sample;
            } else { /* copy */
                memcpy(samples, buf, smp);
                samples += smp;
                buf     += smp;
                sample = buf[-1];
            }
            break;
        default: /* run */
            memset(samples, sample, smp);
            samples += smp;
        }
    }

    frame->nb_samples = samples - frame->data[0];
    *got_frame_ptr    = 1;

    return buf_size;
}

AVCodec ff_ws_snd1_decoder = {
    .name           = "ws_snd1",
    .long_name      = NULL_IF_CONFIG_SMALL("Westwood Audio (SND1)"),
    .type           = AVMEDIA_TYPE_AUDIO,
    .id             = AV_CODEC_ID_WESTWOOD_SND1,
    .init           = ws_snd_decode_init,
    .decode         = ws_snd_decode_frame,
    .capabilities   = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_CHANNEL_CONF,
    .caps_internal  = FF_CODEC_CAP_INIT_THREADSAFE,
};


Generated by: GCOVR (Version 4.2)

Line	Branch	Exec	Source
1			/*
2			* Westwood SNDx codecs
3			* Copyright (c) 2005 Konstantin Shishkov
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 <stdint.h>
23
24			#include "libavutil/channel_layout.h"
25			#include "libavutil/common.h"
26			#include "libavutil/intreadwrite.h"
27			#include "avcodec.h"
28			#include "internal.h"
29
30			/**
31			* @file
32			* Westwood SNDx codecs
33			*
34			* Reference documents about VQA format and its audio codecs
35			* can be found here:
36			* http://www.multimedia.cx
37			*/
38
39			static const int8_t ws_adpcm_4bit[] = {
40			-9, -8, -6, -5, -4, -3, -2, -1,
41			0, 1, 2, 3, 4, 5, 6, 8
42			};
43
44		2	static av_cold int ws_snd_decode_init(AVCodecContext *avctx)
45			{
46		2	avctx->channels = 1;
47		2	avctx->channel_layout = AV_CH_LAYOUT_MONO;
48		2	avctx->sample_fmt = AV_SAMPLE_FMT_U8;
49
50		2	return 0;
51			}
52
53		12	static int ws_snd_decode_frame(AVCodecContext avctx, void data,
54			int got_frame_ptr, AVPacket avpkt)
55			{
56		12	AVFrame *frame = data;
57		12	const uint8_t *buf = avpkt->data;
58		12	int buf_size = avpkt->size;
59
60			int in_size, out_size, ret;
61		12	int sample = 128;
62			uint8_t *samples;
63			uint8_t *samples_end;
64
65	✗✓	12	if (!buf_size)
66			return 0;
67
68	✗✓	12	if (buf_size < 4) {
69			av_log(avctx, AV_LOG_ERROR, "packet is too small\n");
70			return AVERROR(EINVAL);
71			}
72
73		12	out_size = AV_RL16(&buf[0]);
74		12	in_size = AV_RL16(&buf[2]);
75		12	buf += 4;
76
77	✗✓	12	if (in_size > buf_size) {
78			av_log(avctx, AV_LOG_ERROR, "Frame data is larger than input buffer\n");
79			return AVERROR_INVALIDDATA;
80			}
81
82			/* get output buffer */
83		12	frame->nb_samples = out_size;
84	✗✓	12	if ((ret = ff_get_buffer(avctx, frame, 0)) < 0)
85			return ret;
86		12	samples = frame->data[0];
87		12	samples_end = samples + out_size;
88
89	✗✓	12	if (in_size == out_size) {
90			memcpy(samples, buf, out_size);
91			*got_frame_ptr = 1;
92			return buf_size;
93			}
94
95	✓✓✓✗	12139	while (samples < samples_end && buf - avpkt->data < buf_size) {
96			int code, smp, size;
97			uint8_t count;
98		12127	code = *buf >> 6;
99		12127	count = *buf & 0x3F;
100		12127	buf++;
101
102			/* make sure we don't write past the output buffer */
103	✓✓✓✓	12127	switch (code) {
104		29	case 0: smp = 4 * (count + 1); break;
105		4143	case 1: smp = 2 * (count + 1); break;
106	✓✓	7950	case 2: smp = (count & 0x20) ? 1 : count + 1; break;
107		5	default: smp = count + 1; break;
108			}
109	✗✓	12127	if (samples_end - samples < smp)
110			break;
111
112			/* make sure we don't read past the input buffer */
113	✓✓✓✓ ✓✓	12127	size = ((code == 2 && (count & 0x20)) \|\| code == 3) ? 0 : count + 1;
114	✗✓	12127	if ((buf - avpkt->data) + size > buf_size)
115			break;
116
117	✓✓✓✓	12127	switch (code) {
118		29	case 0: /* ADPCM 2-bit */
119	✓✓	58	for (count++; count > 0; count--) {
120		29	code = *buf++;
121		29	sample += ( code & 0x3) - 2;
122		29	sample = av_clip_uint8(sample);
123		29	*samples++ = sample;
124		29	sample += ((code >> 2) & 0x3) - 2;
125		29	sample = av_clip_uint8(sample);
126		29	*samples++ = sample;
127		29	sample += ((code >> 4) & 0x3) - 2;
128		29	sample = av_clip_uint8(sample);
129		29	*samples++ = sample;
130		29	sample += (code >> 6) - 2;
131		29	sample = av_clip_uint8(sample);
132		29	*samples++ = sample;
133			}
134		29	break;
135		4143	case 1: /* ADPCM 4-bit */
136	✓✓	28339	for (count++; count > 0; count--) {
137		24196	code = *buf++;
138		24196	sample += ws_adpcm_4bit[code & 0xF];
139		24196	sample = av_clip_uint8(sample);
140		24196	*samples++ = sample;
141		24196	sample += ws_adpcm_4bit[code >> 4];
142		24196	sample = av_clip_uint8(sample);
143		24196	*samples++ = sample;
144			}
145		4143	break;
146		7950	case 2: /* no compression */
147	✓✓	7950	if (count & 0x20) { /* big delta */
148			int8_t t;
149		7535	t = count;
150		7535	t <<= 3;
151		7535	sample += t >> 3;
152		7535	sample = av_clip_uint8(sample);
153		7535	*samples++ = sample;
154			} else { /* copy */
155		415	memcpy(samples, buf, smp);
156		415	samples += smp;
157		415	buf += smp;
158		415	sample = buf[-1];
159			}
160		7950	break;
161		5	default: /* run */
162		5	memset(samples, sample, smp);
163		5	samples += smp;
164			}
165			}
166
167		12	frame->nb_samples = samples - frame->data[0];
168		12	*got_frame_ptr = 1;
169
170		12	return buf_size;
171			}
172
173			AVCodec ff_ws_snd1_decoder = {
174			.name = "ws_snd1",
175			.long_name = NULL_IF_CONFIG_SMALL("Westwood Audio (SND1)"),
176			.type = AVMEDIA_TYPE_AUDIO,
177			.id = AV_CODEC_ID_WESTWOOD_SND1,
178			.init = ws_snd_decode_init,
179			.decode = ws_snd_decode_frame,
180			.capabilities = AV_CODEC_CAP_DR1 \| AV_CODEC_CAP_CHANNEL_CONF,
181			.caps_internal = FF_CODEC_CAP_INIT_THREADSAFE,
182			};