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
|
#include <assert.h>
#include <drivers/rtl8139.h>
#include <interrupts.h>
#include <kmalloc.h>
#include <network/arp.h>
#include <network/bytes.h>
#include <network/ethernet.h>
#include <network/ipv4.h>
#include <stdio.h>
#include <string.h>
u32 crc32(const char *buf, size_t len) {
static u32 table[256];
static int have_table = 0;
u32 rem;
u8 octet;
int i, j;
const char *p, *q;
if (have_table == 0) {
for (i = 0; i < 256; i++) {
rem = i;
for (j = 0; j < 8; j++) {
if (rem & 1) {
rem >>= 1;
rem ^= 0xedb88320;
} else {
rem >>= 1;
}
}
table[i] = rem;
}
have_table = 1;
}
u32 crc = 0xFFFFFFFF;
q = buf + len;
for (p = buf; p < q; p++) {
octet = *p;
crc = (crc >> 8) ^ table[(crc & 0xff) ^ octet];
}
return ~crc;
}
void handle_ethernet(const u8 *packet, u64 packet_length) {
struct EthernetHeader *eth_header = (struct EthernetHeader *)packet;
packet += sizeof(struct EthernetHeader);
const u8 *payload = packet;
packet += packet_length - sizeof(struct EthernetHeader);
u16 type = ntohs(eth_header->type);
switch (type) {
case 0x0806:
handle_arp(payload);
break;
case 0x0800:
handle_ipv4(payload, packet_length - sizeof(struct EthernetHeader));
break;
default:
kprintf("Can't handle ethernet type 0x%x\n", type);
break;
}
}
void send_ethernet_packet2(u8 mac_dst[6], u16 type, u64 payload_length) {
u64 buffer_size =
sizeof(struct EthernetHeader) + payload_length + sizeof(u32);
assert(buffer_size < 0x1000);
u8 *buffer = nic_get_buffer();
struct EthernetHeader *eth_header = (struct EthernetHeader *)buffer;
memset(eth_header, 0, sizeof(struct EthernetHeader));
buffer += sizeof(struct EthernetHeader);
buffer += payload_length;
memcpy(eth_header->mac_dst, mac_dst, sizeof(u8[6]));
get_mac_address(eth_header->mac_src);
eth_header->type = htons(type);
*(u32 *)(buffer) =
htonl(crc32((const char *)nic_get_buffer(), buffer_size - 4));
nic_send_buffer(buffer_size);
// rtl8139_send_data(ethernet_buffer, buffer_size);
}
u8 ethernet_buffer[0x1000];
void send_ethernet_packet(u8 mac_dst[6], u16 type, u8 *payload,
u64 payload_length) {
u64 buffer_size =
sizeof(struct EthernetHeader) + payload_length + sizeof(u32);
assert(buffer_size < 0x1000);
memset(ethernet_buffer, 0, buffer_size);
u8 *buffer = ethernet_buffer;
struct EthernetHeader *eth_header = (struct EthernetHeader *)buffer;
buffer += sizeof(struct EthernetHeader);
memcpy(buffer, payload, payload_length);
buffer += payload_length;
memcpy(eth_header->mac_dst, mac_dst, sizeof(u8[6]));
get_mac_address(eth_header->mac_src);
eth_header->type = htons(type);
*(u32 *)(buffer) =
htonl(crc32((const char *)ethernet_buffer, buffer_size - 4));
rtl8139_send_data(ethernet_buffer, buffer_size);
}
|
