diff options
Diffstat (limited to 'firmware/zpu/lib/net_common.c')
-rw-r--r-- | firmware/zpu/lib/net_common.c | 503 |
1 files changed, 0 insertions, 503 deletions
diff --git a/firmware/zpu/lib/net_common.c b/firmware/zpu/lib/net_common.c deleted file mode 100644 index 9b75006d3..000000000 --- a/firmware/zpu/lib/net_common.c +++ /dev/null @@ -1,503 +0,0 @@ -/* - * Copyright 2009-2012 Ettus Research LLC - * - * This program is free software: you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation, either version 3 of the License, or - * (at your option) any later version. - * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License - * along with this program. If not, see <http://www.gnu.org/licenses/>. - */ - -#ifdef HAVE_CONFIG_H -#include <config.h> -#endif -#include "net_common.h" -#include "banal.h" -#include <hal_io.h> -#include <memory_map.h> -#include <memcpy_wa.h> -#include <ethernet.h> -#include <net/padded_eth_hdr.h> -#include <lwip/ip.h> -#include <lwip/udp.h> -#include <lwip/icmp.h> -#include <stdlib.h> -#include <nonstdio.h> -#include "arp_cache.h" -#include "if_arp.h" -#include <ethertype.h> -#include <string.h> -#include "pkt_ctrl.h" - -/*********************************************************************** - * Constants + Globals - **********************************************************************/ -static const bool debug = false; -static const size_t out_buff_size = 2048; -static const eth_mac_addr_t BCAST_MAC_ADDR = {{0xff, 0xff, 0xff, 0xff, 0xff, 0xff}}; -#define MAX_UDP_LISTENERS 10 - -/*********************************************************************** - * 16-bit one's complement sum - **********************************************************************/ -static uint32_t chksum_buffer( - uint16_t *buf, size_t nshorts, - uint32_t initial_chksum -){ - uint32_t chksum = initial_chksum; - for (size_t i = 0; i < nshorts; i++) chksum += buf[i]; - - while (chksum >> 16) chksum = (chksum & 0xffff) + (chksum >> 16); - - return chksum; -} - -/*********************************************************************** - * Listener registry - **********************************************************************/ -static eth_mac_addr_t _local_mac_addr; -static struct ip_addr _local_ip_addr; -void register_addrs(const eth_mac_addr_t *mac_addr, const struct ip_addr *ip_addr){ - _local_mac_addr = *mac_addr; - _local_ip_addr = *ip_addr; -} - -struct listener_entry { - unsigned short port; - udp_receiver_t rcvr; -}; - -static struct listener_entry listeners[MAX_UDP_LISTENERS]; - -void init_udp_listeners(void){ - for (int i = 0; i < MAX_UDP_LISTENERS; i++){ - listeners[i].rcvr = NULL; - } -} - -static struct listener_entry * -find_listener_by_port(unsigned short port) -{ - for (int i = 0; i < MAX_UDP_LISTENERS; i++){ - if (port == listeners[i].port) - return &listeners[i]; - } - return 0; -} - -static struct listener_entry * -find_free_listener(void) -{ - for (int i = 0; i < MAX_UDP_LISTENERS; i++){ - if (listeners[i].rcvr == NULL) - return &listeners[i]; - } - abort(); -} - -void -register_udp_listener(int port, udp_receiver_t rcvr) -{ - struct listener_entry *lx = find_listener_by_port(port); - if (lx) - lx->rcvr = rcvr; - else { - lx = find_free_listener(); - lx->port = port; - lx->rcvr = rcvr; - } -} - -/*********************************************************************** - * Protocol framer - **********************************************************************/ -void setup_framer( - eth_mac_addr_t eth_dst, - eth_mac_addr_t eth_src, - struct socket_address sock_dst, - struct socket_address sock_src, - size_t which -){ - struct { - padded_eth_hdr_t eth; - struct ip_hdr ip; - struct udp_hdr udp; - } frame; - - //-- load Ethernet header --// - frame.eth.dst = eth_dst; - frame.eth.src = eth_src; - frame.eth.ethertype = ETHERTYPE_IPV4; - - //-- load IPv4 header --// - IPH_VHLTOS_SET(&frame.ip, 4, 5, 0); - IPH_LEN_SET(&frame.ip, 0); - IPH_ID_SET(&frame.ip, 0); - IPH_OFFSET_SET(&frame.ip, IP_DF); // don't fragment - IPH_TTL_SET(&frame.ip, 32); - IPH_PROTO_SET(&frame.ip, IP_PROTO_UDP); - IPH_CHKSUM_SET(&frame.ip, 0); - frame.ip.src = sock_src.addr; - frame.ip.dest = sock_dst.addr; - IPH_CHKSUM_SET(&frame.ip, chksum_buffer( - (unsigned short *) &frame.ip, - sizeof(frame.ip)/sizeof(short), 0 - )); - - //-- load UDP header --// - frame.udp.src = sock_src.port; - frame.udp.dest = sock_dst.port; - frame.udp.len = 0; - frame.udp.chksum = 0; - - //copy into the framer table registers - memcpy_wa((void *)(sr_proto_framer_regs->table[which].entry + 1), &frame, sizeof(frame)); -} - -/*********************************************************************** - * Slow-path packet framing and transmission - **********************************************************************/ -/*! - * low level routine to assembly an ethernet frame and send it. - * - * \param dst destination mac address - * \param ethertype ethertype field - * \param buf0 first part of data - * \param len0 length of first part of data - * \param buf1 second part of data - * \param len1 length of second part of data - * \param buf2 third part of data - * \param len2 length of third part of data - */ -static void -send_pkt( - eth_mac_addr_t dst, int ethertype, - const void *buf0, size_t len0, - const void *buf1, size_t len1, - const void *buf2, size_t len2 -){ - - //control word for framed data - uint32_t ctrl_word = 0x0; - - //assemble the ethernet header - padded_eth_hdr_t ehdr; - ehdr.pad = 0; - ehdr.dst = dst; - ehdr.src = _local_mac_addr; - ehdr.ethertype = ethertype; - - //grab an out buffer and pointer - uint8_t *buff = (uint8_t *)pkt_ctrl_claim_outgoing_buffer(); - uint8_t *p = buff; - size_t total_len = 0; - - //create a list of all buffers to copy - const void *buffs[] = {&ctrl_word, &ehdr, buf0, buf1, buf2}; - size_t lens[] = {sizeof(ctrl_word), sizeof(ehdr), len0, len1, (len2 + 3) & ~3}; - - //copy each buffer into the out buffer - for (size_t i = 0; i < sizeof(buffs)/sizeof(buffs[0]); i++){ - total_len += lens[i]; //use full length (not clipped) - size_t bytes_remaining = out_buff_size - (size_t)(p - buff); - if (lens[i] > bytes_remaining) lens[i] = bytes_remaining; - if (lens[i] && ((lens[i] & 0x3) || (intptr_t) buffs[i] & 0x3)) - printf("send_pkt: bad alignment of len and/or buf\n"); - memcpy_wa(p, buffs[i], lens[i]); - p += lens[i]; - } - - //ensure that minimum length requirements are met - if (total_len < 64) total_len = 64; //60 + ctrl word - - pkt_ctrl_commit_outgoing_buffer(total_len/sizeof(uint32_t)); - if (debug) printf("sent %d bytes\n", (int)total_len); -} - -void -send_ip_pkt(struct ip_addr dst, int protocol, - const void *buf0, size_t len0, - const void *buf1, size_t len1) -{ - struct ip_hdr ip; - IPH_VHLTOS_SET(&ip, 4, 5, 0); - IPH_LEN_SET(&ip, IP_HLEN + len0 + len1); - IPH_ID_SET(&ip, 0); - IPH_OFFSET_SET(&ip, IP_DF); /* don't fragment */ - IPH_TTL_SET(&ip, 32); - IPH_PROTO_SET(&ip, protocol); - IPH_CHKSUM_SET(&ip, 0); - ip.src = _local_ip_addr; - ip.dest = dst; - - IPH_CHKSUM_SET(&ip, ~chksum_buffer( - (unsigned short *) &ip, sizeof(ip)/sizeof(short), 0 - )); - - eth_mac_addr_t dst_mac; - bool found = arp_cache_lookup_mac(&ip.dest, &dst_mac); - if (!found){ - printf("net_common: failed to hit cache looking for "); - print_ip_addr(&ip.dest); - newline(); - return; - } - - send_pkt(dst_mac, ETHERTYPE_IPV4, - &ip, sizeof(ip), buf0, len0, buf1, len1); -} - -void -send_udp_pkt(int src_port, struct socket_address dst, - const void *buf, size_t len) -{ - struct udp_hdr udp _AL4; - udp.src = src_port; - udp.dest = dst.port; - udp.len = UDP_HLEN + len; - udp.chksum = 0; - - send_ip_pkt(dst.addr, IP_PROTO_UDP, - &udp, sizeof(udp), buf, len); -} - -static void -handle_udp_packet(struct ip_addr src_ip, struct ip_addr dst_ip, - struct udp_hdr *udp, size_t len) -{ - if (len != udp->len){ - printf("UDP inconsistent lengths: %d %d\n", (int)len, udp->len); - return; - } - - unsigned char *payload = ((unsigned char *) udp) + UDP_HLEN; - int payload_len = len - UDP_HLEN; - - if (0){ - printf("\nUDP: src = %d dst = %d len = %d\n", - udp->src, udp->dest, udp->len); - - //print_bytes(0, payload, payload_len); - } - - struct listener_entry *lx = find_listener_by_port(udp->dest); - if (lx){ - struct socket_address src = make_socket_address(src_ip, udp->src); - struct socket_address dst = make_socket_address(dst_ip, udp->dest); - lx->rcvr(src, dst, payload, payload_len); - } -} - -static void -handle_icmp_packet(struct ip_addr src, struct ip_addr dst, - struct icmp_echo_hdr *icmp, size_t len) -{ - switch (icmp->type){ - case ICMP_DUR: // Destinatino Unreachable - if (icmp->code == ICMP_DUR_PORT){ // port unreachable - //handle destination port unreachable (the host ctrl+c'd the app): - - //filter out non udp data response - struct ip_hdr *ip = (struct ip_hdr *)(((uint8_t*)icmp) + sizeof(struct icmp_echo_hdr)); - struct udp_hdr *udp = (struct udp_hdr *)(((char *)ip) + IP_HLEN); - if (IPH_PROTO(ip) != IP_PROTO_UDP) break; - - struct listener_entry *lx = find_listener_by_port(udp->src); - if (lx){ - struct socket_address src = make_socket_address(ip->src, udp->src); - struct socket_address dst = make_socket_address(ip->dest, udp->dest); - lx->rcvr(src, dst, NULL, 0); - } - - putchar('i'); - } - else { - //printf("icmp dst unr (code: %d)", icmp->code); - putchar('i'); - } - break; - - case ICMP_ECHO:{ - const void *icmp_data_buff = ((uint8_t*)icmp) + sizeof(struct icmp_echo_hdr); - size_t icmp_data_len = len - sizeof(struct icmp_echo_hdr); - - struct icmp_echo_hdr echo_reply; - echo_reply.type = 0; - echo_reply.code = 0; - echo_reply.chksum = 0; - echo_reply.id = icmp->id; - echo_reply.seqno = icmp->seqno; - echo_reply.chksum = ~chksum_buffer( //data checksum - (unsigned short *)icmp_data_buff, - icmp_data_len/sizeof(short), - chksum_buffer( //header checksum - (unsigned short *)&echo_reply, - sizeof(echo_reply)/sizeof(short), - 0) - ); - - send_ip_pkt( - src, IP_PROTO_ICMP, - &echo_reply, sizeof(echo_reply), - icmp_data_buff, icmp_data_len - ); - break; - } - default: - break; - } -} - -static void -send_arp_reply(struct arp_eth_ipv4 *req, eth_mac_addr_t our_mac) -{ - struct arp_eth_ipv4 reply _AL4; - reply.ar_hrd = req->ar_hrd; - reply.ar_pro = req->ar_pro; - reply.ar_hln = req->ar_hln; - reply.ar_pln = req->ar_pln; - reply.ar_op = ARPOP_REPLY; - memcpy(reply.ar_sha, &our_mac, 6); - memcpy(reply.ar_sip, req->ar_tip, 4); - memcpy(reply.ar_tha, req->ar_sha, 6); - memcpy(reply.ar_tip, req->ar_sip, 4); - - eth_mac_addr_t t; - memcpy(t.addr, reply.ar_tha, 6); - send_pkt(t, ETHERTYPE_ARP, &reply, sizeof(reply), 0, 0, 0, 0); -} - -void net_common_send_arp_request(const struct ip_addr *addr){ - struct arp_eth_ipv4 req _AL4; - req.ar_hrd = ARPHRD_ETHER; - req.ar_pro = ETHERTYPE_IPV4; - req.ar_hln = sizeof(eth_mac_addr_t); - req.ar_pln = sizeof(struct ip_addr); - req.ar_op = ARPOP_REQUEST; - memcpy(req.ar_sha, ethernet_mac_addr(), sizeof(eth_mac_addr_t)); - memcpy(req.ar_sip, get_ip_addr(), sizeof(struct ip_addr)); - memset(req.ar_tha, 0x00, sizeof(eth_mac_addr_t)); - memcpy(req.ar_tip, addr, sizeof(struct ip_addr)); - - //send the request with a broadcast ethernet mac address - send_pkt(BCAST_MAC_ADDR, ETHERTYPE_ARP, &req, sizeof(req), 0, 0, 0, 0); -} - -void send_gratuitous_arp(void){ - struct arp_eth_ipv4 req _AL4; - req.ar_hrd = ARPHRD_ETHER; - req.ar_pro = ETHERTYPE_IPV4; - req.ar_hln = sizeof(eth_mac_addr_t); - req.ar_pln = sizeof(struct ip_addr); - req.ar_op = ARPOP_REQUEST; - memcpy(req.ar_sha, ethernet_mac_addr(), sizeof(eth_mac_addr_t)); - memcpy(req.ar_sip, get_ip_addr(), sizeof(struct ip_addr)); - memset(req.ar_tha, 0x00, sizeof(eth_mac_addr_t)); - memcpy(req.ar_tip, get_ip_addr(), sizeof(struct ip_addr)); - - //send the request with a broadcast ethernet mac address - send_pkt(BCAST_MAC_ADDR, ETHERTYPE_ARP, &req, sizeof(req), 0, 0, 0, 0); -} - -static void -handle_arp_packet(struct arp_eth_ipv4 *p, size_t size) -{ - if (size < sizeof(struct arp_eth_ipv4)){ - printf("\nhandle_arp: weird size = %d\n", (int)size); - return; - } - - if (0){ - printf("ar_hrd = %d\n", p->ar_hrd); - printf("ar_pro = %d\n", p->ar_pro); - printf("ar_hln = %d\n", p->ar_hln); - printf("ar_pln = %d\n", p->ar_pln); - printf("ar_op = %d\n", p->ar_op); - printf("ar_sha = "); print_mac_addr(p->ar_sha); newline(); - printf("ar_sip = "); print_ip_addr (p->ar_sip); newline(); - printf("ar_tha = "); print_mac_addr(p->ar_tha); newline(); - printf("ar_tip = "); print_ip_addr (p->ar_tip); newline(); - } - - if (p->ar_hrd != ARPHRD_ETHER - || p->ar_pro != ETHERTYPE_IPV4 - || p->ar_hln != 6 - || p->ar_pln != 4) - return; - - if (p->ar_op == ARPOP_REPLY){ - struct ip_addr ip_addr; - memcpy(&ip_addr, p->ar_sip, sizeof(ip_addr)); - eth_mac_addr_t mac_addr; - memcpy(&mac_addr, p->ar_sha, sizeof(mac_addr)); - arp_cache_update(&ip_addr, &mac_addr); - } - - if (p->ar_op != ARPOP_REQUEST) - return; - - struct ip_addr sip; - struct ip_addr tip; - - sip.addr = get_int32(p->ar_sip); - tip.addr = get_int32(p->ar_tip); - - if (memcmp(&tip, &_local_ip_addr, sizeof(_local_ip_addr)) == 0){ // They're looking for us... - send_arp_reply(p, _local_mac_addr); - } -} - -void -handle_eth_packet(uint32_t *p, size_t nlines) -{ - static size_t bcount = 0; - if (debug) printf("===> %d\n", (int)bcount++); - if (debug) print_buffer(p, nlines); - - padded_eth_hdr_t *eth_hdr = (padded_eth_hdr_t *)p; - - if (eth_hdr->ethertype == ETHERTYPE_ARP){ - struct arp_eth_ipv4 *arp = (struct arp_eth_ipv4 *)(p + 4); - handle_arp_packet(arp, nlines*sizeof(uint32_t) - 14); - } - else if (eth_hdr->ethertype == ETHERTYPE_IPV4){ - struct ip_hdr *ip = (struct ip_hdr *)(p + 4); - if (IPH_V(ip) != 4 || IPH_HL(ip) != 5) // ignore pkts w/ bad version or options - return; - - if (IPH_OFFSET(ip) & (IP_MF | IP_OFFMASK)) // ignore fragmented packets - return; - - // filter on dest ip addr (should be broadcast or for us) - bool is_bcast = memcmp(ð_hdr->dst, &BCAST_MAC_ADDR, sizeof(BCAST_MAC_ADDR)) == 0; - bool is_my_ip = memcmp(&ip->dest, &_local_ip_addr, sizeof(_local_ip_addr)) == 0; - if (!is_bcast && !is_my_ip) return; - - arp_cache_update(&ip->src, (eth_mac_addr_t *)(((char *)p)+8)); - - int protocol = IPH_PROTO(ip); - int len = IPH_LEN(ip) - IP_HLEN; - - switch (protocol){ - case IP_PROTO_UDP: - handle_udp_packet(ip->src, ip->dest, (struct udp_hdr *)(((char *)ip) + IP_HLEN), len); - break; - - case IP_PROTO_ICMP: - handle_icmp_packet(ip->src, ip->dest, (struct icmp_echo_hdr *)(((char *)ip) + IP_HLEN), len); - break; - - default: // ignore - break; - } - } - else - return; // Not ARP or IPV4, ignore -} |