summaryrefslogtreecommitdiff
path: root/src/third_party/wiredtiger/src/checksum/zseries/crc32-s390x.c
blob: 3fcfcf69887c5f30034eb8b2eca0b307707dbe40 (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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
/*
 * CRC-32 algorithms implemented with the z/Architecture
 * Vector Extension Facility.
 *
 * Copyright IBM Corp. 2015
 * Author(s): Hendrik Brueckner <brueckner@linux.vnet.ibm.com>
 *
 */

#include <wiredtiger_config.h>
#include <sys/types.h>
#include <endian.h>
#include <inttypes.h>
#include <stddef.h>

#if defined(__linux__) && !defined(HAVE_NO_CRC32_HARDWARE)
#include <sys/auxv.h>

/* RHEL 7 has kernel support, but does not define this constant in the lib c headers. */
#ifndef HWCAP_S390_VX
#define HWCAP_S390_VX 2048
#endif

#include "crc32-s390x.h"
#include "slicing-consts.h"

#define VX_MIN_LEN 64
#define VX_ALIGNMENT 16UL
#define VX_ALIGN_MASK (VX_ALIGNMENT - 1)

/* Prototypes for functions in assembly files */
unsigned int __wt_crc32c_le_vgfm_16(unsigned int crc, const unsigned char *buf, size_t size);

/*
 * __wt_crc32c_le --
 *     Pure C implementations of CRC, one byte at a time
 */
unsigned int
__wt_crc32c_le(unsigned int crc, const unsigned char *buf, size_t len)
{
    crc = htole32(crc);
    while (len--)
        crc = crc32ctable_le[0][((crc >> 24) ^ *buf++) & 0xFF] ^ (crc << 8);
    crc = le32toh(crc);
    return crc;
}

/*
 * DEFINE_CRC32_VX() - Define a CRC-32 function using the vector extension
 *
 * Creates a function to perform a particular CRC-32 computation. Depending
 * on the message buffer, the hardware-accelerated or software implementation
 * is used. Note that the message buffer is aligned to improve fetch
 * operations of VECTOR LOAD MULTIPLE instructions.
 *
 */
#define DEFINE_CRC32_VX(___fname, ___crc32_vx, ___crc32_sw)                            \
    unsigned int ___fname(unsigned int crc, const unsigned char *data, size_t datalen) \
    {                                                                                  \
        unsigned long prealign, aligned, remaining;                                    \
                                                                                       \
        if ((unsigned long)data & VX_ALIGN_MASK) {                                     \
            prealign = VX_ALIGNMENT - ((unsigned long)data & VX_ALIGN_MASK);           \
            datalen -= prealign;                                                       \
            crc = ___crc32_sw(crc, data, prealign);                                    \
            data = data + prealign;                                                    \
        }                                                                              \
                                                                                       \
        if (datalen < VX_MIN_LEN)                                                      \
            return ___crc32_sw(crc, data, datalen);                                    \
                                                                                       \
        aligned = datalen & ~VX_ALIGN_MASK;                                            \
        remaining = datalen & VX_ALIGN_MASK;                                           \
                                                                                       \
        crc = ___crc32_vx(crc, data, aligned);                                         \
        data = data + aligned;                                                         \
                                                                                       \
        if (remaining)                                                                 \
            crc = ___crc32_sw(crc, data, remaining);                                   \
                                                                                       \
        return crc;                                                                    \
    }

/* Main CRC-32 functions */
DEFINE_CRC32_VX(__wt_crc32c_le_vx, __wt_crc32c_le_vgfm_16, __wt_crc32c_le)

/*
 * __wt_checksum_hw --
 *     WiredTiger: return a checksum for a chunk of memory.
 */
static uint32_t
__wt_checksum_hw(const void *chunk, size_t len)
{
    return (~__wt_crc32c_le_vx(0xffffffff, chunk, len));
}
#endif

extern uint32_t __wt_checksum_sw(const void *chunk, size_t len);
#if defined(__GNUC__)
extern uint32_t (*wiredtiger_crc32c_func(void))(const void *, size_t)
  __attribute__((visibility("default")));
#else
extern uint32_t (*wiredtiger_crc32c_func(void))(const void *, size_t);
#endif

/*
 * wiredtiger_crc32c_func --
 *     WiredTiger: detect CRC hardware and return the checksum function.
 */
uint32_t (*wiredtiger_crc32c_func(void))(const void *, size_t)
{
#if defined(__linux__) && !defined(HAVE_NO_CRC32_HARDWARE)
    unsigned long caps = getauxval(AT_HWCAP);

    if (caps & HWCAP_S390_VX)
        return (__wt_checksum_hw);
    else
        return (__wt_checksum_sw);
#else
    return (__wt_checksum_sw);
#endif
}