178 lines
4.2 KiB
C
178 lines
4.2 KiB
C
/*
|
|
* Copyright (c) 2014 Chris Anderson
|
|
* Copyright (c) 2014 Brian Swetland
|
|
*
|
|
* Use of this source code is governed by a MIT-style
|
|
* license that can be found in the LICENSE file or at
|
|
* https://opensource.org/licenses/MIT
|
|
*/
|
|
#pragma once
|
|
|
|
#include <lib/minip.h>
|
|
|
|
#include <lk/compiler.h>
|
|
#include <endian.h>
|
|
#include <lk/list.h>
|
|
#include <stdint.h>
|
|
#include <string.h>
|
|
#include <lib/minip/netif.h>
|
|
|
|
/* Lib configuration */
|
|
#define MINIP_USE_UDP_CHECKSUM 1
|
|
#define MINIP_MTU_SIZE 1536
|
|
#define MINIP_USE_ARP 1
|
|
|
|
#pragma pack(push, 1)
|
|
struct arp_pkt {
|
|
uint16_t htype;
|
|
uint16_t ptype;
|
|
uint8_t hlen;
|
|
uint8_t plen;
|
|
uint16_t oper;
|
|
uint8_t sha[6];
|
|
uint32_t spa;
|
|
uint8_t tha[6];
|
|
uint32_t tpa;
|
|
};
|
|
|
|
struct ipv4_hdr {
|
|
uint8_t ver_ihl;
|
|
uint8_t dscp_ecn;
|
|
uint16_t len;
|
|
uint16_t id;
|
|
uint16_t flags_frags;
|
|
uint8_t ttl;
|
|
uint8_t proto;
|
|
uint16_t chksum;
|
|
uint32_t src_addr;
|
|
uint32_t dst_addr;
|
|
uint8_t data[];
|
|
};
|
|
|
|
struct icmp_pkt {
|
|
uint8_t type;
|
|
uint8_t code;
|
|
uint16_t chksum;
|
|
uint8_t hdr_data[4];
|
|
uint8_t data[];
|
|
};
|
|
|
|
struct eth_hdr {
|
|
uint8_t dst_mac[6];
|
|
uint8_t src_mac[6];
|
|
uint16_t type;
|
|
};
|
|
|
|
#pragma pack(pop)
|
|
|
|
enum {
|
|
ICMP_ECHO_REPLY = 0,
|
|
ICMP_ECHO_REQUEST = 8,
|
|
};
|
|
|
|
enum {
|
|
IP_PROTO_ICMP = 0x1,
|
|
IP_PROTO_TCP = 0x6,
|
|
IP_PROTO_UDP = 0x11,
|
|
};
|
|
|
|
enum {
|
|
ETH_TYPE_IPV4 = 0x0800,
|
|
ETH_TYPE_ARP = 0x0806,
|
|
ETH_TYPE_IPV6 = 0x88dd,
|
|
};
|
|
|
|
enum {
|
|
ARP_OPER_REQUEST = 0x0001,
|
|
ARP_OPER_REPLY = 0x0002,
|
|
};
|
|
|
|
typedef struct udp_hdr udp_hdr_t;
|
|
static const uint8_t bcast_mac[6] = {0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF};
|
|
typedef union {
|
|
uint32_t u;
|
|
uint8_t b[4];
|
|
} ipv4_t;
|
|
|
|
// ARP cache
|
|
void arp_cache_init(void);
|
|
void arp_cache_update(uint32_t addr, const uint8_t mac[6]);
|
|
uint8_t *arp_cache_lookup(uint32_t addr);
|
|
void arp_cache_dump(void);
|
|
int arp_send_request(netif_t *, ipv4_addr_t addr);
|
|
const uint8_t *arp_get_dest_mac(uint32_t host);
|
|
int handle_arp_pkt(netif_t *netif, pktbuf_t *p);
|
|
|
|
// checksums
|
|
uint16_t ones_sum16(uint32_t sum, const void *_buf, int len);
|
|
|
|
typedef struct ipv4_pseudo_header {
|
|
ipv4_addr_t source_addr;
|
|
ipv4_addr_t dest_addr;
|
|
uint8_t zero;
|
|
uint8_t protocol;
|
|
uint16_t tcp_length;
|
|
} __PACKED ipv4_pseudo_header_t;
|
|
|
|
static inline uint16_t cksum_pheader(const ipv4_pseudo_header_t *pheader, const void *buf, size_t len) {
|
|
uint16_t checksum = ones_sum16(0, pheader, sizeof(*pheader));
|
|
return ~ones_sum16(checksum, buf, len);
|
|
}
|
|
|
|
// ipv4 routing
|
|
typedef struct ipv4_route {
|
|
uint32_t flags;
|
|
int ref;
|
|
ipv4_addr_t dest;
|
|
ipv4_addr_t mask;
|
|
netif_t *interface;
|
|
} ipv4_route_t;
|
|
|
|
#define IPV4_ROUTE_UP (1<<0)
|
|
#define IPV4_ROUTE_DEFAULT (1<<1)
|
|
|
|
ipv4_route_t *ipv4_search_route(ipv4_addr_t dest);
|
|
void ipv4_dec_route_ref(ipv4_route_t *r);
|
|
status_t ipv4_add_route(ipv4_addr_t dest, ipv4_addr_t mask, netif_t *n);
|
|
status_t ipv4_add_default_route(ipv4_addr_t dest, netif_t *n);
|
|
|
|
void dump_ipv4_route_table(void);
|
|
|
|
// Helper methods for building headers
|
|
void minip_build_mac_hdr(netif_t *netif, struct eth_hdr *pkt, const uint8_t *dst, uint16_t type);
|
|
|
|
status_t minip_ipv4_send(pktbuf_t *p, ipv4_addr_t dest_addr, uint8_t proto);
|
|
status_t minip_ipv4_send_raw(pktbuf_t *p, ipv4_addr_t dest_addr, uint8_t proto, const uint8_t *dest_mac, netif_t *netif);
|
|
|
|
void tcp_input(netif_t *netif, pktbuf_t *p, uint32_t src_ip, uint32_t dst_ip);
|
|
void udp_input(netif_t *netif, pktbuf_t *p, uint32_t src_ip);
|
|
|
|
// interface list
|
|
void netif_init(void);
|
|
void netif_dump(void);
|
|
|
|
// timers
|
|
typedef void (*net_timer_callback_t)(void *);
|
|
|
|
typedef struct net_timer {
|
|
struct list_node node;
|
|
|
|
lk_time_t sched_time;
|
|
|
|
net_timer_callback_t cb;
|
|
void *arg;
|
|
} net_timer_t;
|
|
|
|
/* set a net timer. returns true if the timer was not set before and is now */
|
|
bool net_timer_set(net_timer_t *, net_timer_callback_t, void *callback_args, lk_time_t delay) __NONNULL((1));
|
|
|
|
/* cancels a net timer. returns true if it was previously set and is not now */
|
|
bool net_timer_cancel(net_timer_t *) __NONNULL();
|
|
|
|
void net_timer_init(void);
|
|
|
|
static inline void mac_addr_copy(uint8_t *dest, const uint8_t *src) {
|
|
*(uint32_t *)dest = *(const uint32_t *)src;
|
|
*(uint16_t *)(dest + 4) = *(const uint16_t *)(src + 4);
|
|
}
|