summaryrefslogtreecommitdiff
path: root/src/leb128.c
blob: 14b79672aa45cafdddd86d573ed434cdfbc2de10 (plain) (blame)
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
#include <leb128.h>
#include <stdbool.h>
#include <stdint.h>
#include <stdio.h>

/* The functions below are (somewhat heavily) modified versions of those
 * provided by the wac project: https://github.com/kanaka/wac */

/*
 * Copyright (C) Joel Martin <github@martintribe.org>
 * The wac project is licensed under the MPL 2.0 (Mozilla Public License
 * 2.0). The text of the MPL 2.0 license is included below and can be
 * found at https://www.mozilla.org/MPL/2.0/
 */

static int read_LEB(FILE *const f, const unsigned maxbits,
    const bool sign, unsigned long long *const out)
{
    unsigned long long result = 0;
    unsigned shift = 0, bcnt = 0;
    uint8_t byte;

    for (;;)
    {
        if (!fread(&byte, sizeof byte, 1, f))
            return -1;

        result |= (unsigned long long)(byte & 0x7f) << shift;
        shift += 7;

        if (!(byte & 0x80))
            break;
        else if (++bcnt > (maxbits + 7 - 1) / 7)
        {
            fprintf(stderr, "%s: overflow\n", __func__);
            return -1;
        }
    }

    if (sign && (shift < maxbits) && (byte & 0x40))
        result |= -1ll << shift;

    *out = result;
    return 0;
}

int leb128_read_unsigned(FILE *const f, const unsigned maxbits,
    unsigned long long *const out)
{
    return read_LEB(f, maxbits, false, out);
}

int leb128_read_signed(FILE *const f, const unsigned maxbits,
    long long *const out)
{
    unsigned long long value;
    const int ret = read_LEB(f, maxbits, true, &value);

    *out = value;
    return ret;
}