]> git.sesse.net Git - ffmpeg/blob - libavcodec/lzf.c
avcodec: Fix reference data type for nvdec vc1 hwaccel
[ffmpeg] / libavcodec / lzf.c
1 /*
2  * lzf decompression algorithm
3  * Copyright (c) 2015 Luca Barbato
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 /**
23  * @file
24  * lzf decompression
25  *
26  * LZF is a fast compression/decompression algorithm that takes very little
27  * code space and working memory, ideal for real-time and block compression.
28  *
29  * https://en.wikibooks.org/wiki/Data_Compression/Dictionary_compression#LZF
30  */
31
32 #include "libavutil/mem.h"
33
34 #include "bytestream.h"
35 #include "lzf.h"
36
37 #define LZF_LITERAL_MAX (1 << 5)
38 #define LZF_LONG_BACKREF 7 + 2
39
40 int ff_lzf_uncompress(GetByteContext *gb, uint8_t **buf, int64_t *size)
41 {
42     int ret     = 0;
43     uint8_t *p  = *buf;
44     int64_t len = 0;
45
46     while (bytestream2_get_bytes_left(gb) > 2) {
47         uint8_t s = bytestream2_get_byte(gb);
48
49         if (s < LZF_LITERAL_MAX) {
50             s++;
51             if (s > *size - len) {
52                 *size += *size /2;
53                 ret = av_reallocp(buf, *size);
54                 if (ret < 0)
55                     return ret;
56                 p = *buf + len;
57             }
58
59             bytestream2_get_buffer(gb, p, s);
60             p   += s;
61             len += s;
62         } else {
63             int l   = 2 + (s >> 5);
64             int off = ((s & 0x1f) << 8) + 1;
65
66             if (l == LZF_LONG_BACKREF)
67                 l += bytestream2_get_byte(gb);
68
69             off += bytestream2_get_byte(gb);
70
71             if (off > len)
72                 return AVERROR_INVALIDDATA;
73
74             if (l > *size - len) {
75                 *size += *size / 2;
76                 ret = av_reallocp(buf, *size);
77                 if (ret < 0)
78                     return ret;
79                 p = *buf + len;
80             }
81
82             av_memcpy_backptr(p, off, l);
83
84             p   += l;
85             len += l;
86         }
87     }
88
89     *size = len;
90
91     return 0;
92 }