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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
// MIT License, Copyright (c) 2020 Marvin Borner
#include <def.h>
#include <mem.h>
#include <net.h>
#include <pci.h>
#include <print.h>
#include <rtl8139.h>
/**
* Checksums
*/
u16 ip_calculate_checksum(struct ip_packet *packet)
{
u32 sum = 0;
u16 *s = (u16 *)packet;
for (int i = 0; i < 10; ++i)
sum += ntohs(s[i]);
if (sum > 0xFFFF)
sum = (sum >> 16) + (sum & 0xFFFF);
return sum;
}
/**
* Requests
*/
int ethernet_send_packet(u8 *dst, u8 *data, int len, int prot)
{
struct ethernet_packet *packet = malloc(sizeof(*packet) + len);
memcpy(packet->src, rtl8139_get_mac(), 6);
memcpy(packet->dst, dst, 6);
memcpy(packet->data, data, len);
packet->type = htons(prot);
rtl8139_send_packet(packet, sizeof(*packet) + len);
free(packet);
return len;
}
void ip_send_packet(u32 dst, void *data, int len, int prot)
{
struct ip_packet *packet = malloc(sizeof(*packet) + len);
memset(packet, 0, sizeof(*packet));
packet->version_ihl = ((0x4 << 4) | (0x5 << 0));
packet->length = sizeof(*packet) + len;
packet->id = 0; // TODO: IP fragmentation
packet->ttl = 64;
packet->protocol = prot;
packet->src = htonl(0x0a00022a);
packet->dst = dst;
/* void *packet_data = (u32 *)packet + 0x5 * 4; */
memcpy(packet->data, data, len);
packet->length = htons(sizeof(*packet) + len);
packet->checksum = htons(ip_calculate_checksum(packet));
// TODO: arp destination lookup
ethernet_send_packet((u8 *)0x424242424242, (u8 *)packet, htons(packet->length),
ETHERNET_TYPE_IP4);
free(packet);
}
/**
* Responses
*/
void icmp_handle_packet(u32 dst, int len)
{
struct icmp_packet *packet = malloc(sizeof(*packet) + len);
memset(packet, 0, sizeof(*packet));
packet->type = 0; // Ping reponse
packet->version = 0;
packet->checksum = 0;
ip_send_packet(dst, packet, sizeof(*packet) + len, IP_PROT_ICMP);
free(packet);
}
void ip_handle_packet(struct ip_packet *packet, int len)
{
switch (packet->protocol) {
case IP_PROT_ICMP:
print("ICMP Packet!\n");
icmp_handle_packet(packet->src, len);
break;
case IP_PROT_TCP:
print("TCP Packet!\n");
break;
case IP_PROT_UDP:
print("UDP Packet!\n");
break;
default:
printf("Unknown IP protocol %d\n", packet->protocol);
}
}
void ethernet_handle_packet(struct ethernet_packet *packet, int len)
{
void *data = packet->data;
int data_len = len - sizeof(*packet);
if (ntohs(packet->type) == ETHERNET_TYPE_ARP) {
print("ARP PACKET\n");
} else if (ntohs(packet->type) == ETHERNET_TYPE_IP4) {
print("IP4 PACKET\n");
ip_handle_packet(data, data_len);
} else if (ntohs(packet->type) == ETHERNET_TYPE_IP6) {
print("IP6 PACKET\n");
/* ip_handle_packet(data, data_len); */
} else {
printf("UNKNOWN PACKET %x\n", ntohs(packet->type));
}
}
/**
* Install
*/
void net_install()
{
rtl8139_install();
}
|
