Martin Michelsen
106 lines
2.5 KiB
C++
106 lines
2.5 KiB
C++
#pragma once
|
|
|
|
#include <stdint.h>
|
|
|
|
#include <phosg/Encoding.hh>
|
|
|
|
|
|
|
|
struct EthernetHeader {
|
|
uint8_t dest_mac[6];
|
|
uint8_t src_mac[6];
|
|
be_uint16_t protocol;
|
|
} __attribute__((packed));
|
|
|
|
struct ARPHeader {
|
|
be_uint16_t hardware_type;
|
|
be_uint16_t protocol_type; // same as EthernetHeader::protocol
|
|
uint8_t hwaddr_len;
|
|
uint8_t paddr_len;
|
|
be_uint16_t operation;
|
|
} __attribute__((packed));
|
|
|
|
struct IPv4Header {
|
|
uint8_t version_ihl;
|
|
uint8_t tos;
|
|
be_uint16_t size;
|
|
be_uint16_t id;
|
|
be_uint16_t frag_offset;
|
|
uint8_t ttl;
|
|
uint8_t protocol;
|
|
be_uint16_t checksum;
|
|
be_uint32_t src_addr;
|
|
be_uint32_t dest_addr;
|
|
} __attribute__((packed));
|
|
|
|
struct UDPHeader {
|
|
be_uint16_t src_port;
|
|
be_uint16_t dest_port;
|
|
be_uint16_t size;
|
|
be_uint16_t checksum;
|
|
} __attribute__((packed));
|
|
|
|
struct TCPHeader {
|
|
enum Flag {
|
|
NS = 0x0100,
|
|
CWR = 0x0080, // congestion window reduced
|
|
ECE = 0x0040, // ECN capable / congestion experienced
|
|
URG = 0x0020, // urgent pointer used
|
|
ACK = 0x0010, // ack_num is valid
|
|
PSH = 0x0008, // sending data
|
|
RST = 0x0004, // reset (hard disconnect)
|
|
SYN = 0x0002, // synchronize sequence numbers (open connection)
|
|
FIN = 0x0001, // close (normal disconnect)
|
|
};
|
|
|
|
be_uint16_t src_port;
|
|
be_uint16_t dest_port;
|
|
be_uint32_t seq_num;
|
|
be_uint32_t ack_num;
|
|
be_uint16_t flags;
|
|
be_uint16_t window;
|
|
be_uint16_t checksum;
|
|
be_uint16_t urgent_ptr;
|
|
} __attribute__((packed));
|
|
|
|
|
|
|
|
struct FrameInfo {
|
|
// This is always valid
|
|
const EthernetHeader* ether;
|
|
uint16_t ether_protocol;
|
|
|
|
// At most one of these is not null
|
|
const IPv4Header* ipv4;
|
|
const ARPHeader* arp;
|
|
|
|
// One of these may be not null if this->ipv4 is not null
|
|
const UDPHeader* udp;
|
|
const TCPHeader* tcp;
|
|
|
|
const void* header_start;
|
|
const void* payload;
|
|
size_t total_size;
|
|
size_t tcp_options_size;
|
|
size_t payload_size;
|
|
|
|
FrameInfo();
|
|
FrameInfo(const std::string& data);
|
|
FrameInfo(const void* data, size_t size);
|
|
|
|
std::string header_str() const;
|
|
|
|
void truncate(size_t new_total_size);
|
|
|
|
size_t size_from_header() const;
|
|
|
|
static uint16_t computed_ipv4_header_checksum(const IPv4Header& ipv4);
|
|
uint16_t computed_ipv4_header_checksum() const;
|
|
static uint16_t computed_udp4_checksum(
|
|
const IPv4Header& ipv4, const UDPHeader& udp, const void* data, size_t size);
|
|
uint16_t computed_udp4_checksum() const;
|
|
static uint16_t computed_tcp4_checksum(
|
|
const IPv4Header& ip, const TCPHeader& tcp, const void* data, size_t size);
|
|
uint16_t computed_tcp4_checksum() const;
|
|
};
|