lokinet/llarp/net/ip.hpp

263 lines
5.1 KiB
C++
Raw Normal View History

#ifndef LLARP_IP_HPP
#define LLARP_IP_HPP
2019-01-11 01:19:36 +00:00
#include <ev/ev.h>
#include <net/net.hpp>
2019-02-02 23:12:42 +00:00
#include <util/buffer.hpp>
#include <util/time.hpp>
2018-09-19 13:24:36 +00:00
#ifndef _WIN32
// unix, linux
2018-09-21 14:36:06 +00:00
#include <sys/types.h> // FreeBSD needs this for uchar for ip.h
#include <netinet/in.h>
#include <netinet/ip.h>
// anything not win32
struct ip_header
{
#ifdef __LITTLE_ENDIAN__
unsigned int ihl : 4;
unsigned int version : 4;
#elif defined(__BIG_ENDIAN__)
unsigned int version : 4;
unsigned int ihl : 4;
#else
#error "Please fix <bits/endian.h>"
#endif
#if defined(__linux__)
#define ip_version version
#endif
uint8_t tos;
uint16_t tot_len;
uint16_t id;
uint16_t frag_off;
uint8_t ttl;
uint8_t protocol;
uint16_t check;
uint32_t saddr;
uint32_t daddr;
};
#else
// windows nt
#include <winsock2.h>
typedef struct ip_hdr
{
2018-08-20 10:52:47 +00:00
unsigned char
ip_header_len : 4; // 4-bit header length (in 32-bit words) normally=5
// (Means 20 Bytes may be 24 also)
unsigned char version : 4; // 4-bit IPv4 version
2018-08-20 10:52:47 +00:00
unsigned char ip_tos; // IP type of service
unsigned short ip_total_length; // Total length
unsigned short ip_id; // Unique identifier
2018-08-20 10:52:47 +00:00
unsigned char ip_frag_offset : 5; // Fragment offset field
2018-08-20 10:52:47 +00:00
unsigned char ip_more_fragment : 1;
unsigned char ip_dont_fragment : 1;
unsigned char ip_reserved_zero : 1;
2018-08-20 10:52:47 +00:00
unsigned char ip_frag_offset1; // fragment offset
2018-08-20 10:52:47 +00:00
unsigned char ip_ttl; // Time to live
unsigned char ip_protocol; // Protocol(TCP,UDP etc)
unsigned short ip_checksum; // IP checksum
unsigned int ip_srcaddr; // Source address
unsigned int ip_destaddr; // Source address
} IPV4_HDR;
#define ip_header IPV4_HDR
2018-08-21 18:31:42 +00:00
#define saddr ip_srcaddr
#define daddr ip_destaddr
2018-08-21 18:33:27 +00:00
#define check ip_checksum
2018-08-21 13:02:05 +00:00
#define ihl ip_header_len
#define protocol ip_protocol
#define frag_off ip_frag_offset
2018-09-19 13:24:36 +00:00
#endif
2019-06-11 16:44:05 +00:00
struct ipv6_header
{
unsigned char version : 4;
unsigned char pad_small : 4;
2019-06-11 21:28:55 +00:00
uint8_t pad[3];
2019-06-11 18:23:53 +00:00
uint16_t payload_len;
uint8_t proto;
uint8_t hoplimit;
2019-06-11 16:44:05 +00:00
in6_addr srcaddr;
in6_addr dstaddr;
};
#include <memory>
2019-06-11 16:44:05 +00:00
#include <service/protocol.hpp>
2019-07-30 23:42:13 +00:00
#include <utility>
struct llarp_ev_loop;
namespace llarp
{
namespace net
{
2019-06-11 16:44:05 +00:00
/// an Packet
struct IPPacket
{
2019-06-11 16:44:05 +00:00
static huint128_t
In6ToHUInt(in6_addr addr);
2019-06-11 16:44:05 +00:00
static in6_addr
HUIntToIn6(huint128_t x);
2019-06-11 16:44:05 +00:00
static huint128_t
ExpandV4(huint32_t x);
2019-06-11 16:44:05 +00:00
static huint32_t
TruncateV6(huint128_t x);
static constexpr size_t MaxSize = 1500;
2018-08-22 15:52:10 +00:00
llarp_time_t timestamp;
size_t sz;
byte_t buf[MaxSize];
ManagedBuffer
2018-08-22 15:52:10 +00:00
Buffer();
2018-08-20 19:12:12 +00:00
ManagedBuffer
2018-11-29 21:19:20 +00:00
ConstBuffer() const;
2018-08-20 19:12:12 +00:00
bool
Load(const llarp_buffer_t& buf);
struct GetTime
{
llarp_time_t
2019-06-11 16:44:05 +00:00
operator()(const IPPacket& pkt) const
{
2018-08-31 14:41:04 +00:00
return pkt.timestamp;
}
};
struct PutTime
{
2019-04-08 12:01:52 +00:00
llarp_ev_loop_ptr loop;
2019-07-30 23:42:13 +00:00
PutTime(llarp_ev_loop_ptr evloop) : loop(std::move(evloop))
2018-10-29 16:48:36 +00:00
{
}
void
2019-06-11 16:44:05 +00:00
operator()(IPPacket& pkt) const
{
2018-10-29 16:48:36 +00:00
pkt.timestamp = llarp_ev_loop_time_now_ms(loop);
}
};
struct GetNow
{
2019-04-08 12:01:52 +00:00
llarp_ev_loop_ptr loop;
2019-07-30 23:42:13 +00:00
GetNow(llarp_ev_loop_ptr evloop) : loop(std::move(evloop))
{
}
llarp_time_t
operator()() const
{
return llarp_ev_loop_time_now_ms(loop);
}
};
struct CompareSize
{
bool
2019-06-11 16:44:05 +00:00
operator()(const IPPacket& left, const IPPacket& right)
{
return left.sz < right.sz;
}
};
struct CompareOrder
{
bool
2019-06-11 16:44:05 +00:00
operator()(const IPPacket& left, const IPPacket& right)
{
2018-08-31 14:41:04 +00:00
return left.timestamp < right.timestamp;
}
};
2018-10-10 12:06:28 +00:00
inline ip_header*
Header()
{
2018-09-17 20:18:11 +00:00
return (ip_header*)&buf[0];
}
2018-10-10 12:06:28 +00:00
inline const ip_header*
Header() const
{
2018-09-17 20:18:11 +00:00
return (ip_header*)&buf[0];
}
2019-06-11 16:44:05 +00:00
inline ipv6_header*
HeaderV6()
{
return (ipv6_header*)&buf[0];
}
inline const ipv6_header*
HeaderV6() const
{
2019-06-11 16:44:05 +00:00
return (ipv6_header*)&buf[0];
}
2019-06-11 16:44:05 +00:00
inline int
Version() const
{
2019-06-11 16:44:05 +00:00
return Header()->version;
}
2019-06-11 16:44:05 +00:00
inline bool
IsV4() const
{
2019-06-11 16:44:05 +00:00
return Version() == 4;
}
2019-06-11 16:44:05 +00:00
inline bool
IsV6() const
{
2019-06-11 16:44:05 +00:00
return Version() == 6;
}
2019-06-11 16:44:05 +00:00
inline service::ProtocolType
ServiceProtocol() const
{
if(IsV4())
return service::eProtocolTrafficV4;
2019-07-30 23:42:13 +00:00
if(IsV6())
2019-06-11 16:44:05 +00:00
return service::eProtocolTrafficV6;
2019-07-30 23:42:13 +00:00
return service::eProtocolControl;
2019-06-11 16:44:05 +00:00
}
huint128_t
srcv6() const;
huint128_t
dstv6() const;
huint32_t
srcv4() const;
huint32_t
dstv4() const;
huint128_t
src4to6() const;
huint128_t
dst4to6() const;
2018-08-20 19:12:12 +00:00
void
2019-06-11 20:56:48 +00:00
UpdateIPv4Address(nuint32_t src, nuint32_t dst);
2018-10-09 14:09:03 +00:00
void
2019-06-11 20:56:48 +00:00
UpdateIPv6Address(huint128_t src, huint128_t dst);
};
} // namespace net
} // namespace llarp
#endif