blob: 3859006a94f11b532fd7c9ea1ce08c8f03be8d1f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
// SPDX-License-Identifier: 0BSD
///////////////////////////////////////////////////////////////////////////////
//
/// \file vli_encoder.c
/// \brief Encodes variable-length integers
//
// Author: Lasse Collin
//
///////////////////////////////////////////////////////////////////////////////
#include "common.h"
extern LZMA_API(lzma_ret)
lzma_vli_encode(lzma_vli vli, size_t *vli_pos,
uint8_t *restrict out, size_t *restrict out_pos,
size_t out_size)
{
// If we haven't been given vli_pos, work in single-call mode.
size_t vli_pos_internal = 0;
if (vli_pos == NULL) {
vli_pos = &vli_pos_internal;
// In single-call mode, we expect that the caller has
// reserved enough output space.
if (*out_pos >= out_size)
return LZMA_PROG_ERROR;
} else {
// This never happens when we are called by liblzma, but
// may happen if called directly from an application.
if (*out_pos >= out_size)
return LZMA_BUF_ERROR;
}
// Validate the arguments.
if (*vli_pos >= LZMA_VLI_BYTES_MAX || vli > LZMA_VLI_MAX)
return LZMA_PROG_ERROR;
// Shift vli so that the next bits to encode are the lowest. In
// single-call mode this never changes vli since *vli_pos is zero.
vli >>= *vli_pos * 7;
// Write the non-last bytes in a loop.
while (vli >= 0x80) {
// We don't need *vli_pos during this function call anymore,
// but update it here so that it is ready if we need to
// return before the whole integer has been decoded.
++*vli_pos;
assert(*vli_pos < LZMA_VLI_BYTES_MAX);
// Write the next byte.
out[*out_pos] = (uint8_t)(vli) | 0x80;
vli >>= 7;
if (++*out_pos == out_size)
return vli_pos == &vli_pos_internal
? LZMA_PROG_ERROR : LZMA_OK;
}
// Write the last byte.
out[*out_pos] = (uint8_t)(vli);
++*out_pos;
++*vli_pos;
return vli_pos == &vli_pos_internal ? LZMA_OK : LZMA_STREAM_END;
}
|
