mirror of
https://github.com/scylladb/scylladb.git
synced 2026-04-22 17:40:34 +00:00
132 lines
2.9 KiB
C++
132 lines
2.9 KiB
C++
/*
|
|
* Copyright (C) 2014 Cloudius Systems, Ltd.
|
|
*
|
|
*/
|
|
|
|
#ifndef IP_HH_
|
|
#define IP_HH_
|
|
|
|
#include <arpa/inet.h>
|
|
#include <cstdint>
|
|
#include <array>
|
|
|
|
namespace net {
|
|
|
|
using ethernet_address = std::array<uint8_t, 6>;
|
|
|
|
inline uint64_t ntohq(uint64_t v) {
|
|
return __builtin_bswap64(v);
|
|
}
|
|
|
|
uint16_t ip_checksum(void* data, size_t len);
|
|
|
|
inline void ntoh() {}
|
|
inline void hton() {}
|
|
|
|
inline void ntoh(uint16_t& x) { x = ntohs(x); }
|
|
inline void hton(uint16_t& x) { x = htons(x); }
|
|
inline void ntoh(uint32_t& x) { x = ntohl(x); }
|
|
inline void hton(uint32_t& x) { x = htonl(x); }
|
|
|
|
// Wrapper around a primitive type to provide an unaligned version.
|
|
// This is because gcc (correctly) doesn't allow binding an unaligned
|
|
// scalar variable to a reference, and (unfortunately) doesn't allow
|
|
// specifying unaligned references.
|
|
//
|
|
// So, packed<uint32_t>& is our way of passing a reference (or pointer)
|
|
// to a uint32_t around, without losing the knowledge about its alignment
|
|
// or lack thereof.
|
|
template <typename T>
|
|
struct packed {
|
|
T raw;
|
|
packed() = default;
|
|
packed(T x) : raw(x) {}
|
|
packed& operator=(const T& x) { raw = x; return *this; }
|
|
operator T() const { return raw; }
|
|
} __attribute__((packed));
|
|
|
|
template <typename T>
|
|
inline void ntoh(packed<T>& x) {
|
|
T v = x;
|
|
ntoh(v);
|
|
x = v;
|
|
}
|
|
|
|
template <typename T>
|
|
inline void hton(packed<T>& x) {
|
|
T v = x;
|
|
hton(v);
|
|
x = v;
|
|
}
|
|
|
|
template <typename First, typename... Rest>
|
|
inline
|
|
void ntoh(First& first, Rest&... rest) {
|
|
ntoh(first);
|
|
ntoh(std::forward<Rest&>(rest)...);
|
|
}
|
|
|
|
template <typename First, typename... Rest>
|
|
inline
|
|
void hton(First& first, Rest&... rest) {
|
|
hton(first);
|
|
hton(std::forward<Rest&>(rest)...);
|
|
}
|
|
|
|
template <class T>
|
|
inline
|
|
void ntoh(T& x) {
|
|
x.adjust_endianness([] (auto&... what) { ntoh(std::forward<decltype(what)&>(what)...); });
|
|
}
|
|
|
|
template <class T>
|
|
inline
|
|
void hton(T& x) {
|
|
x.adjust_endianness([] (auto&... what) { hton(std::forward<decltype(what)&>(what)...); });
|
|
}
|
|
|
|
struct eth_hdr {
|
|
ethernet_address dst_mac;
|
|
ethernet_address src_mac;
|
|
packed<uint16_t> eth_proto;
|
|
template <typename Adjuster>
|
|
auto adjust_endianness(Adjuster a) {
|
|
return a(eth_proto);
|
|
}
|
|
} __attribute__((packed));
|
|
|
|
struct ip_hdr {
|
|
uint8_t ihl : 4;
|
|
uint8_t ver : 4;
|
|
uint8_t dscp : 6;
|
|
uint8_t ecn : 2;
|
|
packed<uint16_t> len;
|
|
packed<uint16_t> id;
|
|
packed<uint16_t> frag;
|
|
uint8_t ttl;
|
|
uint8_t ip_proto;
|
|
packed<uint16_t> csum;
|
|
packed<uint32_t> src_ip;
|
|
packed<uint32_t> dst_ip;
|
|
uint8_t options[0];
|
|
template <typename Adjuster>
|
|
auto adjust_endianness(Adjuster a) {
|
|
return a(len, id, frag, csum, src_ip, dst_ip);
|
|
}
|
|
} __attribute__((packed));
|
|
|
|
struct icmp_hdr {
|
|
uint8_t type;
|
|
uint8_t code;
|
|
packed<uint16_t> csum;
|
|
packed<uint32_t> rest;
|
|
template <typename Adjuster>
|
|
auto adjust_endianness(Adjuster a) {
|
|
return a(csum);
|
|
}
|
|
} __attribute__((packed));
|
|
|
|
}
|
|
|
|
#endif /* IP_HH_ */
|