diff options
Diffstat (limited to 'kernel/network')
-rw-r--r-- | kernel/network/ethernet.c | 3 | ||||
-rw-r--r-- | kernel/network/tcp.c | 6 |
2 files changed, 8 insertions, 1 deletions
diff --git a/kernel/network/ethernet.c b/kernel/network/ethernet.c index 6255ecb..940802f 100644 --- a/kernel/network/ethernet.c +++ b/kernel/network/ethernet.c @@ -67,6 +67,7 @@ void handle_ethernet(const u8 *packet, u64 packet_length) { void send_ethernet_packet(u8 mac_dst[6], u16 type, u8 *payload, u64 payload_length) { + assert(payload_length <= 1500); // FIXME: Janky allocation, do this better u64 buffer_size = sizeof(struct ETHERNET_HEADER) + payload_length + sizeof(u32); @@ -82,6 +83,6 @@ void send_ethernet_packet(u8 mac_dst[6], u16 type, u8 *payload, eth_header->type = htons(type); *(u32 *)(buffer) = htonl(crc32((const char *)buffer_start, buffer_size - 4)); - assert(rtl8139_send_data(buffer_start, buffer_size)); + rtl8139_send_data(buffer_start, buffer_size); kfree(buffer_start); } diff --git a/kernel/network/tcp.c b/kernel/network/tcp.c index 14bd0cb..e4b0891 100644 --- a/kernel/network/tcp.c +++ b/kernel/network/tcp.c @@ -113,6 +113,12 @@ void tcp_close_connection(struct INCOMING_TCP_CONNECTION *inc) { void send_tcp_packet(struct INCOMING_TCP_CONNECTION *inc, u8 *payload, u16 payload_length) { + if (payload_length > 1500 - 20 - sizeof(struct TCP_HEADER)) { + send_tcp_packet(inc, payload, 1500 - 20 - sizeof(struct TCP_HEADER)); + payload_length -= 1500 - 20 - sizeof(struct TCP_HEADER); + payload += 1500 - 20 - sizeof(struct TCP_HEADER); + return send_tcp_packet(inc, payload, payload_length); + } struct TCP_HEADER header = {0}; header.src_port = htons(inc->dst_port); header.dst_port = inc->n_port; |