1d1d32c3dSAlex Richardson /*-
2d1d32c3dSAlex Richardson  * SPDX-License-Identifier: BSD-2-Clause
3d1d32c3dSAlex Richardson  *
4d1d32c3dSAlex Richardson  * Copyright 2018-2020 Alex Richardson <arichardson@FreeBSD.org>
5d1d32c3dSAlex Richardson  *
6d1d32c3dSAlex Richardson  * This software was developed by SRI International and the University of
7d1d32c3dSAlex Richardson  * Cambridge Computer Laboratory (Department of Computer Science and
8d1d32c3dSAlex Richardson  * Technology) under DARPA contract HR0011-18-C-0016 ("ECATS"), as part of the
9d1d32c3dSAlex Richardson  * DARPA SSITH research programme.
10d1d32c3dSAlex Richardson  *
11d1d32c3dSAlex Richardson  * This software was developed by SRI International and the University of
12d1d32c3dSAlex Richardson  * Cambridge Computer Laboratory under DARPA/AFRL contract (FA8750-10-C-0237)
13d1d32c3dSAlex Richardson  * ("CTSRD"), as part of the DARPA CRASH research programme.
14d1d32c3dSAlex Richardson  *
15d1d32c3dSAlex Richardson  * Redistribution and use in source and binary forms, with or without
16d1d32c3dSAlex Richardson  * modification, are permitted provided that the following conditions
17d1d32c3dSAlex Richardson  * are met:
18d1d32c3dSAlex Richardson  * 1. Redistributions of source code must retain the above copyright
19d1d32c3dSAlex Richardson  *    notice, this list of conditions and the following disclaimer.
20d1d32c3dSAlex Richardson  * 2. Redistributions in binary form must reproduce the above copyright
21d1d32c3dSAlex Richardson  *    notice, this list of conditions and the following disclaimer in the
22d1d32c3dSAlex Richardson  *    documentation and/or other materials provided with the distribution.
23d1d32c3dSAlex Richardson  *
24d1d32c3dSAlex Richardson  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
25d1d32c3dSAlex Richardson  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26d1d32c3dSAlex Richardson  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27d1d32c3dSAlex Richardson  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
28d1d32c3dSAlex Richardson  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29d1d32c3dSAlex Richardson  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30d1d32c3dSAlex Richardson  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31d1d32c3dSAlex Richardson  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32d1d32c3dSAlex Richardson  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33d1d32c3dSAlex Richardson  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34d1d32c3dSAlex Richardson  * SUCH DAMAGE.
35d1d32c3dSAlex Richardson  */
36d1d32c3dSAlex Richardson #pragma once
37d1d32c3dSAlex Richardson 
38d1d32c3dSAlex Richardson #if __has_include_next(<sys/endian.h>)
39d1d32c3dSAlex Richardson #include_next <sys/endian.h>
40d1d32c3dSAlex Richardson #endif
41d1d32c3dSAlex Richardson 
42d1d32c3dSAlex Richardson #if __has_include(<endian.h>)
43d1d32c3dSAlex Richardson #include <endian.h>
44d1d32c3dSAlex Richardson #endif
45d1d32c3dSAlex Richardson 
46d1d32c3dSAlex Richardson /* Linux uses a double underscore, FreeBSD a single one */
47d1d32c3dSAlex Richardson #define _LITTLE_ENDIAN __LITTLE_ENDIAN
48d1d32c3dSAlex Richardson #define _BIG_ENDIAN __BIG_ENDIAN
49d1d32c3dSAlex Richardson #define _BYTE_ORDER __BYTE_ORDER
50d1d32c3dSAlex Richardson 
51d1d32c3dSAlex Richardson /*
52d1d32c3dSAlex Richardson  * Ensure all these are constant expressions (which is not the case for some
53d1d32c3dSAlex Richardson  * of the glibc versions depending on compiler optimization level)
54d1d32c3dSAlex Richardson  */
55d1d32c3dSAlex Richardson 
56d1d32c3dSAlex Richardson #undef bswap64
57d1d32c3dSAlex Richardson #define bswap64(a) __builtin_bswap64(a)
58d1d32c3dSAlex Richardson 
59d1d32c3dSAlex Richardson #undef bswap32
60d1d32c3dSAlex Richardson #define bswap32(a) __builtin_bswap32(a)
61d1d32c3dSAlex Richardson 
62d1d32c3dSAlex Richardson #undef bswap16
63d1d32c3dSAlex Richardson #define bswap16(a) __builtin_bswap16(a)
64d1d32c3dSAlex Richardson 
65d1d32c3dSAlex Richardson #undef __bswap_64
66d1d32c3dSAlex Richardson #define __bswap_64(a) __builtin_bswap64(a)
67d1d32c3dSAlex Richardson 
68d1d32c3dSAlex Richardson #undef __bswap_32
69d1d32c3dSAlex Richardson #define __bswap_32(a) __builtin_bswap32(a)
70d1d32c3dSAlex Richardson 
71d1d32c3dSAlex Richardson #undef __bswap_16
72d1d32c3dSAlex Richardson #define __bswap_16(a) __builtin_bswap16(a)
73d1d32c3dSAlex Richardson 
74d1d32c3dSAlex Richardson /* Alignment-agnostic encode/decode bytestream to/from little/big endian. */
75d1d32c3dSAlex Richardson 
76d1d32c3dSAlex Richardson static __inline uint16_t
be16dec(const void * pp)77d1d32c3dSAlex Richardson be16dec(const void *pp)
78d1d32c3dSAlex Richardson {
79d1d32c3dSAlex Richardson 	uint8_t const *p = (uint8_t const *)pp;
80d1d32c3dSAlex Richardson 
81d1d32c3dSAlex Richardson 	return ((p[0] << 8) | p[1]);
82d1d32c3dSAlex Richardson }
83d1d32c3dSAlex Richardson 
84d1d32c3dSAlex Richardson static __inline uint32_t
be32dec(const void * pp)85d1d32c3dSAlex Richardson be32dec(const void *pp)
86d1d32c3dSAlex Richardson {
87d1d32c3dSAlex Richardson 	uint8_t const *p = (uint8_t const *)pp;
88d1d32c3dSAlex Richardson 
89d1d32c3dSAlex Richardson 	return (((unsigned)p[0] << 24) | (p[1] << 16) | (p[2] << 8) | p[3]);
90d1d32c3dSAlex Richardson }
91d1d32c3dSAlex Richardson 
92d1d32c3dSAlex Richardson static __inline uint64_t
be64dec(const void * pp)93d1d32c3dSAlex Richardson be64dec(const void *pp)
94d1d32c3dSAlex Richardson {
95d1d32c3dSAlex Richardson 	uint8_t const *p = (uint8_t const *)pp;
96d1d32c3dSAlex Richardson 
97d1d32c3dSAlex Richardson 	return (((uint64_t)be32dec(p) << 32) | be32dec(p + 4));
98d1d32c3dSAlex Richardson }
99d1d32c3dSAlex Richardson 
100d1d32c3dSAlex Richardson static __inline uint16_t
le16dec(const void * pp)101d1d32c3dSAlex Richardson le16dec(const void *pp)
102d1d32c3dSAlex Richardson {
103d1d32c3dSAlex Richardson 	uint8_t const *p = (uint8_t const *)pp;
104d1d32c3dSAlex Richardson 
105d1d32c3dSAlex Richardson 	return ((p[1] << 8) | p[0]);
106d1d32c3dSAlex Richardson }
107d1d32c3dSAlex Richardson 
108d1d32c3dSAlex Richardson static __inline uint32_t
le32dec(const void * pp)109d1d32c3dSAlex Richardson le32dec(const void *pp)
110d1d32c3dSAlex Richardson {
111d1d32c3dSAlex Richardson 	uint8_t const *p = (uint8_t const *)pp;
112d1d32c3dSAlex Richardson 
113d1d32c3dSAlex Richardson 	return (((unsigned)p[3] << 24) | (p[2] << 16) | (p[1] << 8) | p[0]);
114d1d32c3dSAlex Richardson }
115d1d32c3dSAlex Richardson 
116d1d32c3dSAlex Richardson static __inline uint64_t
le64dec(const void * pp)117d1d32c3dSAlex Richardson le64dec(const void *pp)
118d1d32c3dSAlex Richardson {
119d1d32c3dSAlex Richardson 	uint8_t const *p = (uint8_t const *)pp;
120d1d32c3dSAlex Richardson 
121d1d32c3dSAlex Richardson 	return (((uint64_t)le32dec(p + 4) << 32) | le32dec(p));
122d1d32c3dSAlex Richardson }
123d1d32c3dSAlex Richardson 
124d1d32c3dSAlex Richardson static __inline void
be16enc(void * pp,uint16_t u)125d1d32c3dSAlex Richardson be16enc(void *pp, uint16_t u)
126d1d32c3dSAlex Richardson {
127d1d32c3dSAlex Richardson 	uint8_t *p = (uint8_t *)pp;
128d1d32c3dSAlex Richardson 
129d1d32c3dSAlex Richardson 	p[0] = (u >> 8) & 0xff;
130d1d32c3dSAlex Richardson 	p[1] = u & 0xff;
131d1d32c3dSAlex Richardson }
132d1d32c3dSAlex Richardson 
133d1d32c3dSAlex Richardson static __inline void
be32enc(void * pp,uint32_t u)134d1d32c3dSAlex Richardson be32enc(void *pp, uint32_t u)
135d1d32c3dSAlex Richardson {
136d1d32c3dSAlex Richardson 	uint8_t *p = (uint8_t *)pp;
137d1d32c3dSAlex Richardson 
138d1d32c3dSAlex Richardson 	p[0] = (u >> 24) & 0xff;
139d1d32c3dSAlex Richardson 	p[1] = (u >> 16) & 0xff;
140d1d32c3dSAlex Richardson 	p[2] = (u >> 8) & 0xff;
141d1d32c3dSAlex Richardson 	p[3] = u & 0xff;
142d1d32c3dSAlex Richardson }
143d1d32c3dSAlex Richardson 
144d1d32c3dSAlex Richardson static __inline void
be64enc(void * pp,uint64_t u)145d1d32c3dSAlex Richardson be64enc(void *pp, uint64_t u)
146d1d32c3dSAlex Richardson {
147d1d32c3dSAlex Richardson 	uint8_t *p = (uint8_t *)pp;
148d1d32c3dSAlex Richardson 
149d1d32c3dSAlex Richardson 	be32enc(p, (uint32_t)(u >> 32));
150d1d32c3dSAlex Richardson 	be32enc(p + 4, (uint32_t)(u & 0xffffffffU));
151d1d32c3dSAlex Richardson }
152d1d32c3dSAlex Richardson 
153d1d32c3dSAlex Richardson static __inline void
le16enc(void * pp,uint16_t u)154d1d32c3dSAlex Richardson le16enc(void *pp, uint16_t u)
155d1d32c3dSAlex Richardson {
156d1d32c3dSAlex Richardson 	uint8_t *p = (uint8_t *)pp;
157d1d32c3dSAlex Richardson 
158d1d32c3dSAlex Richardson 	p[0] = u & 0xff;
159d1d32c3dSAlex Richardson 	p[1] = (u >> 8) & 0xff;
160d1d32c3dSAlex Richardson }
161d1d32c3dSAlex Richardson 
162d1d32c3dSAlex Richardson static __inline void
le32enc(void * pp,uint32_t u)163d1d32c3dSAlex Richardson le32enc(void *pp, uint32_t u)
164d1d32c3dSAlex Richardson {
165d1d32c3dSAlex Richardson 	uint8_t *p = (uint8_t *)pp;
166d1d32c3dSAlex Richardson 
167d1d32c3dSAlex Richardson 	p[0] = u & 0xff;
168d1d32c3dSAlex Richardson 	p[1] = (u >> 8) & 0xff;
169d1d32c3dSAlex Richardson 	p[2] = (u >> 16) & 0xff;
170d1d32c3dSAlex Richardson 	p[3] = (u >> 24) & 0xff;
171d1d32c3dSAlex Richardson }
172d1d32c3dSAlex Richardson 
173d1d32c3dSAlex Richardson static __inline void
le64enc(void * pp,uint64_t u)174d1d32c3dSAlex Richardson le64enc(void *pp, uint64_t u)
175d1d32c3dSAlex Richardson {
176d1d32c3dSAlex Richardson 	uint8_t *p = (uint8_t *)pp;
177d1d32c3dSAlex Richardson 
178d1d32c3dSAlex Richardson 	le32enc(p, (uint32_t)(u & 0xffffffffU));
179d1d32c3dSAlex Richardson 	le32enc(p + 4, (uint32_t)(u >> 32));
180d1d32c3dSAlex Richardson }
181