summaryrefslogtreecommitdiff
path: root/src/checksum.h
blob: f788ff1469c2c64ddffbf82a26b4778845507ffd (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
#include <stdarg.h>
#include <stdint.h>
#include <stdlib.h>

// based on code from RFCs 1071 and 1624

/*
static inline uint16_t csum_update(const void *ptr, uint16_t new_value, uint16_t wsum) {
    uint32_t sum = *(uint16_t *)ptr + (~ntohs(*(uint16_t *)&new_value) & 0xffff) + ntohs(wsum);
    sum = (sum & 0xffff) + (sum >> 16);
    return htons(sum + (sum >> 16));
}

static inline uint16_t fold_sum(uint32_t sum) {
    while (sum >> 16)
        sum = (sum & 0xffff) + (sum >> 16);
    return sum;
}

static inline uint16_t do_csum(const void *ptr, size_t len) {
    uint32_t sum = 0;

    while (len > 1) {
        sum += *(uint16_t *)ptr++;
        len -= 2;
    }

    if (len > 0)
        sum += *(uint8_t *)ptr;

    return ~fold_sum(sum);
}

static inline uint16_t csum_partial(uint16_t sum, const void *ptr, size_t len, ...) {
    va_list ap;
    va_start(ap, len);
    do {
        sum = ~fold_sum(~sum + ~do_csum(ptr, len));
    } while ((ptr = va_arg(ap, const void *)) && (len = va_arg(ap, size_t)));
    va_end(ap);
    return sum;
}
*/

uint16_t csum_partial(const void *buff, int len, uint16_t wsum);