curo/ip.h at master · kametan0730/curo · GitHub

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
#ifndef CURO_IP_H
#define CURO_IP_H

#include "config.h"
#include <iostream>
#include <queue>

#define IP_ADDRESS_LEN 4
#define IP_ADDRESS(A, B, C, D) (A * 0x1000000u + B * 0x10000 + C * 0x100 + D)
#define IP_ADDRESS_LIMITED_BROADCAST IP_ADDRESS(255, 255, 255, 255)

#define IP_HEADER_SIZE 20

#define IP_PROTOCOL_NUM_ICMP 0x01
#define IP_PROTOCOL_NUM_TCP 0x06
#define IP_PROTOCOL_NUM_UDP 0x11

#define IP_FRAG_OFFSET_MASK_RESERVED_FLAG 0b1000000000000000
#define IP_FRAG_OFFSET_MASK_DF_FLAG 0b0100000000000000
#define IP_FRAG_OFFSET_MASK_MF_FLAG 0b0010000000000000
#define IP_FRAG_OFFSET_MASK_OFFSET 0b0001111111111111

struct ip_header {
    uint8_t header_len : 4;
    uint8_t version : 4;
    uint8_t tos;
    uint16_t total_len;
    uint16_t identify;
    uint16_t frag_offset;
    uint8_t ttl;
    uint8_t protocol;
    uint16_t header_checksum;
    uint32_t src_addr;
    uint32_t dest_addr;
} __attribute__((packed));

struct nat_device;

struct ip_device {
    uint32_t address = 0;   // デバイスのIPアドレス
    uint32_t netmask = 0;   // サブネットマスク
    uint32_t broadcast = 0; // ブロードキャストアドレス
#ifdef ENABLE_NAT
    nat_device *nat_dev = nullptr;
#endif
};

enum ip_route_type {
    connected, // 直接接続されているネットワークの経路
    network
};

struct net_device;

struct ip_route_entry {
    ip_route_type type;
    union {
        net_device *dev;
        uint32_t next_hop;
    };
};

template <typename DATA_TYPE> struct binary_trie_node;

extern binary_trie_node<ip_route_entry> *ip_fib;

void dump_ip_fib();

bool in_subnet(uint32_t subnet_prefix, uint32_t subnet_mask,
               uint32_t target_address);

void ip_input(net_device *input_dev, uint8_t *buffer, ssize_t len);

struct my_buf;

void ip_output_to_host(net_device *dev, uint32_t dest_address,
                       uint32_t src_address, my_buf *payload_mybuf);
void ip_output_to_next_hop(uint32_t next_hop, my_buf *payload_mybuf);
void ip_output(uint32_t dest_addr, uint32_t src_addr, my_buf *payload_mybuf);
void ip_encapsulate_output(uint32_t dest_addr, uint32_t src_addr,
                           my_buf *payload_mybuf, uint8_t protocol_num);

#endif // CURO_IP_H