Libav
lzf.c
Go to the documentation of this file.
1 /*
2  * lzf decompression algorithm
3  * Copyright (c) 2015 Luca Barbato
4  *
5  * This file is part of Libav.
6  *
7  * Libav 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  * Libav 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 Libav; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
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  }
57 
58  bytestream2_get_buffer(gb, p, s);
59  p += s;
60  len += s;
61  } else {
62  int l = 2 + (s >> 5);
63  int off = ((s & 0x1f) << 8) + 1;
64 
65  if (l == LZF_LONG_BACKREF)
66  l += bytestream2_get_byte(gb);
67 
68  off += bytestream2_get_byte(gb);
69 
70  if (off > len)
71  return AVERROR_INVALIDDATA;
72 
73  if (l > *size - len) {
74  *size += *size / 2;
75  ret = av_reallocp(buf, *size);
76  if (ret < 0)
77  return ret;
78  }
79 
80  av_memcpy_backptr(p, off, l);
81 
82  p += l;
83  len += l;
84  }
85  }
86 
87  *size = len;
88 
89  return 0;
90 }
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:54
int size
memory handling functions
uint8_t
#define LZF_LONG_BACKREF
Definition: lzf.c:38
int av_reallocp(void *ptr, size_t size)
Allocate or reallocate a block of memory.
Definition: mem.c:140
int ff_lzf_uncompress(GetByteContext *gb, uint8_t **buf, int64_t *size)
Definition: lzf.c:40
static av_always_inline unsigned int bytestream2_get_buffer(GetByteContext *g, uint8_t *dst, unsigned int size)
Definition: bytestream.h:260
static av_always_inline unsigned int bytestream2_get_bytes_left(GetByteContext *g)
Definition: bytestream.h:151
#define LZF_LITERAL_MAX
Definition: lzf.c:37
int len
void av_memcpy_backptr(uint8_t *dst, int back, int cnt)
deliberately overlapping memcpy implementation
Definition: mem.c:325