Skip to content
Snippets Groups Projects
netpoll.c 29.3 KiB
Newer Older
  • Learn to ignore specific revisions
  • Linus Torvalds's avatar
    Linus Torvalds committed
    /*
     * Common framework for low-level network console, dump, and debugger code
     *
     * Sep 8 2003  Matt Mackall <mpm@selenic.com>
     *
     * based on the netconsole code from:
     *
     * Copyright (C) 2001  Ingo Molnar <mingo@redhat.com>
     * Copyright (C) 2002  Red Hat, Inc.
     */
    
    
    #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
    
    
    #include <linux/moduleparam.h>
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    #include <linux/netdevice.h>
    #include <linux/etherdevice.h>
    #include <linux/string.h>
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    #include <linux/inetdevice.h>
    #include <linux/inet.h>
    #include <linux/interrupt.h>
    #include <linux/netpoll.h>
    #include <linux/sched.h>
    #include <linux/delay.h>
    #include <linux/rcupdate.h>
    #include <linux/workqueue.h>
    
    #include <linux/if_vlan.h>
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    #include <net/tcp.h>
    #include <net/udp.h>
    
    Cong Wang's avatar
    Cong Wang committed
    #include <net/addrconf.h>
    #include <net/ndisc.h>
    #include <net/ip6_checksum.h>
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    #include <asm/unaligned.h>
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    /*
     * We maintain a small pool of fully-sized skbs, to make sure the
     * message gets out even in extreme OOM situations.
     */
    
    #define MAX_UDP_CHUNK 1460
    #define MAX_SKBS 32
    
    
    static struct sk_buff_head skb_pool;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    static atomic_t trapped;
    
    
    Stephen Hemminger's avatar
    Stephen Hemminger committed
    #define USEC_PER_POLL	50
    
    #define NETPOLL_RX_ENABLED  1
    #define NETPOLL_RX_DROP     2
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    #define MAX_SKB_SIZE							\
    	(sizeof(struct ethhdr) +					\
    	 sizeof(struct iphdr) +						\
    	 sizeof(struct udphdr) +					\
    	 MAX_UDP_CHUNK)
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    static void zap_completion_queue(void);
    
    Cong Wang's avatar
    Cong Wang committed
    static void netpoll_neigh_reply(struct sk_buff *skb, struct netpoll_info *npinfo);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    static unsigned int carrier_timeout = 4;
    module_param(carrier_timeout, uint, 0644);
    
    
    #define np_info(np, fmt, ...)				\
    	pr_info("%s: " fmt, np->name, ##__VA_ARGS__)
    #define np_err(np, fmt, ...)				\
    	pr_err("%s: " fmt, np->name, ##__VA_ARGS__)
    #define np_notice(np, fmt, ...)				\
    	pr_notice("%s: " fmt, np->name, ##__VA_ARGS__)
    
    
    static void queue_process(struct work_struct *work)
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    {
    
    	struct netpoll_info *npinfo =
    		container_of(work, struct netpoll_info, tx_work.work);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	struct sk_buff *skb;
    
    	unsigned long flags;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    	while ((skb = skb_dequeue(&npinfo->txq))) {
    		struct net_device *dev = skb->dev;
    
    		const struct net_device_ops *ops = dev->netdev_ops;
    
    		struct netdev_queue *txq;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    		if (!netif_device_present(dev) || !netif_running(dev)) {
    			__kfree_skb(skb);
    			continue;
    		}
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    		txq = netdev_get_tx_queue(dev, skb_get_queue_mapping(skb));
    
    
    		local_irq_save(flags);
    
    		__netif_tx_lock(txq, smp_processor_id());
    
    		if (netif_xmit_frozen_or_stopped(txq) ||
    
    		    ops->ndo_start_xmit(skb, dev) != NETDEV_TX_OK) {
    
    			skb_queue_head(&npinfo->txq, skb);
    
    			__netif_tx_unlock(txq);
    
    			local_irq_restore(flags);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    			schedule_delayed_work(&npinfo->tx_work, HZ/10);
    
    		__netif_tx_unlock(txq);
    
    		local_irq_restore(flags);
    
    static __sum16 checksum_udp(struct sk_buff *skb, struct udphdr *uh,
    			    unsigned short ulen, __be32 saddr, __be32 daddr)
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    {
    
    	if (uh->check == 0 || skb_csum_unnecessary(skb))
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    		return 0;
    
    
    	psum = csum_tcpudp_nofold(saddr, daddr, ulen, IPPROTO_UDP, 0);
    
    
    	if (skb->ip_summed == CHECKSUM_COMPLETE &&
    
    	    !csum_fold(csum_add(psum, skb->csum)))
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    	return __skb_checksum_complete(skb);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    }
    
    /*
     * Check whether delayed processing was scheduled for our NIC. If so,
     * we attempt to grab the poll lock and use ->poll() to pump the card.
     * If this fails, either we've recursed in ->poll() or it's already
     * running on another CPU.
     *
     * Note: we don't mask interrupts with this lock because we're using
     * trylock here and interrupts are already disabled in the softirq
     * case. Further, we test the poll_owner to avoid recursion on UP
     * systems where the lock doesn't exist.
     *
     * In cases where there is bi-directional communications, reading only
     * one message at a time can lead to packets being dropped by the
     * network adapter, forcing superfluous retries and possibly timeouts.
     * Thus, we set our budget to greater than 1.
     */
    
    static int poll_one_napi(struct netpoll_info *npinfo,
    			 struct napi_struct *napi, int budget)
    {
    	int work;
    
    	/* net_rx_action's ->poll() invocations and our's are
    	 * synchronized by this test which is only made while
    	 * holding the napi->poll_lock.
    	 */
    	if (!test_bit(NAPI_STATE_SCHED, &napi->state))
    		return budget;
    
    
    	npinfo->rx_flags |= NETPOLL_RX_DROP;
    
    	set_bit(NAPI_STATE_NPSVC, &napi->state);
    
    
    	work = napi->poll(napi, budget);
    
    	trace_napi_poll(napi);
    
    	clear_bit(NAPI_STATE_NPSVC, &napi->state);
    
    	npinfo->rx_flags &= ~NETPOLL_RX_DROP;
    
    static void poll_napi(struct net_device *dev)
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    {
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	int budget = 16;
    
    
    	list_for_each_entry(napi, &dev->napi_list, dev_list) {
    
    		if (napi->poll_owner != smp_processor_id() &&
    
    		    spin_trylock(&napi->poll_lock)) {
    
    			budget = poll_one_napi(rcu_dereference_bh(dev->npinfo),
    					       napi, budget);
    
    Cong Wang's avatar
    Cong Wang committed
    static void service_neigh_queue(struct netpoll_info *npi)
    
    	if (npi) {
    		struct sk_buff *skb;
    
    Cong Wang's avatar
    Cong Wang committed
    		while ((skb = skb_dequeue(&npi->neigh_tx)))
    			netpoll_neigh_reply(skb, npi);
    
    static void netpoll_poll_dev(struct net_device *dev)
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    {
    
    	const struct net_device_ops *ops;
    
    	struct netpoll_info *ni = rcu_dereference_bh(dev->npinfo);
    
    	if (!dev || !netif_running(dev))
    		return;
    
    	ops = dev->netdev_ops;
    	if (!ops->ndo_poll_controller)
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    		return;
    
    	/* Process pending work on NIC */
    
    	ops->ndo_poll_controller(dev);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    	if (dev->flags & IFF_SLAVE) {
    
    			struct net_device *bond_dev;
    
    			struct sk_buff *skb;
    
    			struct netpoll_info *bond_ni;
    
    			bond_dev = netdev_master_upper_dev_get_rcu(dev);
    			bond_ni = rcu_dereference_bh(bond_dev->npinfo);
    
    Cong Wang's avatar
    Cong Wang committed
    			while ((skb = skb_dequeue(&ni->neigh_tx))) {
    
    				skb->dev = bond_dev;
    
    Cong Wang's avatar
    Cong Wang committed
    				skb_queue_tail(&bond_ni->neigh_tx, skb);
    
    Cong Wang's avatar
    Cong Wang committed
    	service_neigh_queue(ni);
    
    	zap_completion_queue();
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    }
    
    static void refill_skbs(void)
    {
    	struct sk_buff *skb;
    	unsigned long flags;
    
    
    	spin_lock_irqsave(&skb_pool.lock, flags);
    	while (skb_pool.qlen < MAX_SKBS) {
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    		skb = alloc_skb(MAX_SKB_SIZE, GFP_ATOMIC);
    		if (!skb)
    			break;
    
    
    		__skb_queue_tail(&skb_pool, skb);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	}
    
    	spin_unlock_irqrestore(&skb_pool.lock, flags);
    
    static void zap_completion_queue(void)
    {
    	unsigned long flags;
    	struct softnet_data *sd = &get_cpu_var(softnet_data);
    
    	if (sd->completion_queue) {
    		struct sk_buff *clist;
    
    		local_irq_save(flags);
    		clist = sd->completion_queue;
    		sd->completion_queue = NULL;
    		local_irq_restore(flags);
    
    		while (clist != NULL) {
    			struct sk_buff *skb = clist;
    			clist = clist->next;
    			if (skb->destructor) {
    				atomic_inc(&skb->users);
    				dev_kfree_skb_any(skb); /* put this one back */
    			} else {
    				__kfree_skb(skb);
    			}
    		}
    	}
    
    	put_cpu_var(softnet_data);
    }
    
    
    static struct sk_buff *find_skb(struct netpoll *np, int len, int reserve)
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    {
    
    	int count = 0;
    	struct sk_buff *skb;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    	zap_completion_queue();
    
    	refill_skbs();
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    repeat:
    
    	skb = alloc_skb(len, GFP_ATOMIC);
    
    	if (!skb)
    		skb = skb_dequeue(&skb_pool);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    	if (!skb) {
    
    		if (++count < 10) {
    
    			netpoll_poll_dev(np->dev);
    
    			goto repeat;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    		}
    
    		return NULL;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	}
    
    	atomic_set(&skb->users, 1);
    	skb_reserve(skb, reserve);
    	return skb;
    }
    
    
    static int netpoll_owner_active(struct net_device *dev)
    {
    	struct napi_struct *napi;
    
    	list_for_each_entry(napi, &dev->napi_list, dev_list) {
    		if (napi->poll_owner == smp_processor_id())
    			return 1;
    	}
    	return 0;
    }
    
    
    /* call with IRQ disabled */
    
    void netpoll_send_skb_on_dev(struct netpoll *np, struct sk_buff *skb,
    			     struct net_device *dev)
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    {
    
    Stephen Hemminger's avatar
    Stephen Hemminger committed
    	int status = NETDEV_TX_BUSY;
    	unsigned long tries;
    
    	const struct net_device_ops *ops = dev->netdev_ops;
    
    Herbert Xu's avatar
    Herbert Xu committed
    	/* It is up to the caller to keep npinfo alive. */
    
    	struct netpoll_info *npinfo;
    
    	WARN_ON_ONCE(!irqs_disabled());
    
    	npinfo = rcu_dereference_bh(np->dev->npinfo);
    
    	if (!npinfo || !netif_running(dev) || !netif_device_present(dev)) {
    		__kfree_skb(skb);
    		return;
    	}
    
    Stephen Hemminger's avatar
    Stephen Hemminger committed
    
    	/* don't get messages out of order, and no recursion */
    
    	if (skb_queue_len(&npinfo->txq) == 0 && !netpoll_owner_active(dev)) {
    
    		struct netdev_queue *txq;
    
    		txq = netdev_pick_tx(dev, skb);
    
    		/* try until next clock tick */
    		for (tries = jiffies_to_usecs(1)/USEC_PER_POLL;
    		     tries > 0; --tries) {
    
    			if (__netif_tx_trylock(txq)) {
    
    				if (!netif_xmit_stopped(txq)) {
    
    					if (vlan_tx_tag_present(skb) &&
    					    !(netif_skb_features(skb) & NETIF_F_HW_VLAN_TX)) {
    						skb = __vlan_put_tag(skb, vlan_tx_tag_get(skb));
    						if (unlikely(!skb))
    							break;
    						skb->vlan_tci = 0;
    					}
    
    
    					status = ops->ndo_start_xmit(skb, dev);
    
    					if (status == NETDEV_TX_OK)
    						txq_trans_update(txq);
    				}
    
    				__netif_tx_unlock(txq);
    
    
    			/* tickle device maybe there is some cleanup */
    
    			netpoll_poll_dev(np->dev);
    
    
    			udelay(USEC_PER_POLL);
    
    			"netpoll_send_skb_on_dev(): %s enabled interrupts in poll (%pF)\n",
    
    Stephen Hemminger's avatar
    Stephen Hemminger committed
    	if (status != NETDEV_TX_OK) {
    
    Stephen Hemminger's avatar
    Stephen Hemminger committed
    		skb_queue_tail(&npinfo->txq, skb);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	}
    }
    
    EXPORT_SYMBOL(netpoll_send_skb_on_dev);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    void netpoll_send_udp(struct netpoll *np, const char *msg, int len)
    {
    
    	int total_len, ip_len, udp_len;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	struct sk_buff *skb;
    	struct udphdr *udph;
    	struct iphdr *iph;
    	struct ethhdr *eth;
    
    	static atomic_t ip_ident;
    
    Cong Wang's avatar
    Cong Wang committed
    	struct ipv6hdr *ip6h;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    	udp_len = len + sizeof(*udph);
    
    Cong Wang's avatar
    Cong Wang committed
    	if (np->ipv6)
    		ip_len = udp_len + sizeof(*ip6h);
    	else
    
    Cong Wang's avatar
    Cong Wang committed
    		ip_len = udp_len + sizeof(*iph);
    
    
    	total_len = ip_len + LL_RESERVED_SPACE(np->dev);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    	skb = find_skb(np, total_len + np->dev->needed_tailroom,
    		       total_len - len);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	if (!skb)
    		return;
    
    
    	skb_copy_to_linear_data(skb, msg, len);
    
    	skb_put(skb, len);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    	skb_push(skb, sizeof(*udph));
    	skb_reset_transport_header(skb);
    	udph = udp_hdr(skb);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	udph->source = htons(np->local_port);
    	udph->dest = htons(np->remote_port);
    	udph->len = htons(udp_len);
    
    Cong Wang's avatar
    Cong Wang committed
    	if (np->ipv6) {
    		udph->check = 0;
    		udph->check = csum_ipv6_magic(&np->local_ip.in6,
    					      &np->remote_ip.in6,
    					      udp_len, IPPROTO_UDP,
    					      csum_partial(udph, udp_len, 0));
    		if (udph->check == 0)
    			udph->check = CSUM_MANGLED_0;
    
    		skb_push(skb, sizeof(*ip6h));
    		skb_reset_network_header(skb);
    		ip6h = ipv6_hdr(skb);
    
    		/* ip6h->version = 6; ip6h->priority = 0; */
    		put_unaligned(0x60, (unsigned char *)ip6h);
    		ip6h->flow_lbl[0] = 0;
    		ip6h->flow_lbl[1] = 0;
    		ip6h->flow_lbl[2] = 0;
    
    		ip6h->payload_len = htons(sizeof(struct udphdr) + len);
    		ip6h->nexthdr = IPPROTO_UDP;
    		ip6h->hop_limit = 32;
    		ip6h->saddr = np->local_ip.in6;
    		ip6h->daddr = np->remote_ip.in6;
    
    		eth = (struct ethhdr *) skb_push(skb, ETH_HLEN);
    		skb_reset_mac_header(skb);
    		skb->protocol = eth->h_proto = htons(ETH_P_IPV6);
    	} else {
    
    Cong Wang's avatar
    Cong Wang committed
    		udph->check = 0;
    		udph->check = csum_tcpudp_magic(np->local_ip.ip,
    						np->remote_ip.ip,
    						udp_len, IPPROTO_UDP,
    						csum_partial(udph, udp_len, 0));
    		if (udph->check == 0)
    			udph->check = CSUM_MANGLED_0;
    
    		skb_push(skb, sizeof(*iph));
    		skb_reset_network_header(skb);
    		iph = ip_hdr(skb);
    
    		/* iph->version = 4; iph->ihl = 5; */
    		put_unaligned(0x45, (unsigned char *)iph);
    		iph->tos      = 0;
    		put_unaligned(htons(ip_len), &(iph->tot_len));
    		iph->id       = htons(atomic_inc_return(&ip_ident));
    		iph->frag_off = 0;
    		iph->ttl      = 64;
    		iph->protocol = IPPROTO_UDP;
    		iph->check    = 0;
    		put_unaligned(np->local_ip.ip, &(iph->saddr));
    		put_unaligned(np->remote_ip.ip, &(iph->daddr));
    		iph->check    = ip_fast_csum((unsigned char *)iph, iph->ihl);
    
    		eth = (struct ethhdr *) skb_push(skb, ETH_HLEN);
    		skb_reset_mac_header(skb);
    		skb->protocol = eth->h_proto = htons(ETH_P_IP);
    	}
    
    
    	memcpy(eth->h_source, np->dev->dev_addr, ETH_ALEN);
    	memcpy(eth->h_dest, np->remote_mac, ETH_ALEN);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    	skb->dev = np->dev;
    
    	netpoll_send_skb(np, skb);
    }
    
    EXPORT_SYMBOL(netpoll_send_udp);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    Cong Wang's avatar
    Cong Wang committed
    static void netpoll_neigh_reply(struct sk_buff *skb, struct netpoll_info *npinfo)
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    {
    
    Cong Wang's avatar
    Cong Wang committed
    	int size, type = ARPOP_REPLY;
    
    Al Viro's avatar
    Al Viro committed
    	__be32 sip, tip;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	struct sk_buff *send_skb;
    
    	struct netpoll *np, *tmp;
    	unsigned long flags;
    
    	int hlen, tlen;
    
    Cong Wang's avatar
    Cong Wang committed
    	int hits = 0, proto;
    
    
    	if (list_empty(&npinfo->rx_np))
    		return;
    
    	/* Before checking the packet, we do some early
    	   inspection whether this is interesting at all */
    	spin_lock_irqsave(&npinfo->rx_lock, flags);
    	list_for_each_entry_safe(np, tmp, &npinfo->rx_np, rx) {
    		if (np->dev == skb->dev)
    			hits++;
    	}
    	spin_unlock_irqrestore(&npinfo->rx_lock, flags);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    	/* No netpoll struct is using this dev */
    	if (!hits)
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    Cong Wang's avatar
    Cong Wang committed
    	proto = ntohs(eth_hdr(skb)->h_proto);
    	if (proto == ETH_P_IP) {
    
    Cong Wang's avatar
    Cong Wang committed
    		struct arphdr *arp;
    		unsigned char *arp_ptr;
    
    Cong Wang's avatar
    Cong Wang committed
    		/* No arp on this interface */
    		if (skb->dev->flags & IFF_NOARP)
    			return;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    Cong Wang's avatar
    Cong Wang committed
    		if (!pskb_may_pull(skb, arp_hdr_len(skb->dev)))
    			return;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    Cong Wang's avatar
    Cong Wang committed
    		skb_reset_network_header(skb);
    		skb_reset_transport_header(skb);
    		arp = arp_hdr(skb);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    Cong Wang's avatar
    Cong Wang committed
    		if ((arp->ar_hrd != htons(ARPHRD_ETHER) &&
    		     arp->ar_hrd != htons(ARPHRD_IEEE802)) ||
    		    arp->ar_pro != htons(ETH_P_IP) ||
    		    arp->ar_op != htons(ARPOP_REQUEST))
    			return;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    Cong Wang's avatar
    Cong Wang committed
    		arp_ptr = (unsigned char *)(arp+1);
    		/* save the location of the src hw addr */
    		sha = arp_ptr;
    		arp_ptr += skb->dev->addr_len;
    		memcpy(&sip, arp_ptr, 4);
    		arp_ptr += 4;
    		/* If we actually cared about dst hw addr,
    		   it would get copied here */
    		arp_ptr += skb->dev->addr_len;
    		memcpy(&tip, arp_ptr, 4);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    Cong Wang's avatar
    Cong Wang committed
    		/* Should we ignore arp? */
    		if (ipv4_is_loopback(tip) || ipv4_is_multicast(tip))
    			return;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    Cong Wang's avatar
    Cong Wang committed
    		size = arp_hdr_len(skb->dev);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    Cong Wang's avatar
    Cong Wang committed
    		spin_lock_irqsave(&npinfo->rx_lock, flags);
    		list_for_each_entry_safe(np, tmp, &npinfo->rx_np, rx) {
    			if (tip != np->local_ip.ip)
    				continue;
    
    			hlen = LL_RESERVED_SPACE(np->dev);
    			tlen = np->dev->needed_tailroom;
    			send_skb = find_skb(np, size + hlen + tlen, hlen);
    			if (!send_skb)
    				continue;
    
    			skb_reset_network_header(send_skb);
    			arp = (struct arphdr *) skb_put(send_skb, size);
    			send_skb->dev = skb->dev;
    			send_skb->protocol = htons(ETH_P_ARP);
    
    			/* Fill the device header for the ARP frame */
    
    Cong Wang's avatar
    Cong Wang committed
    			if (dev_hard_header(send_skb, skb->dev, ETH_P_ARP,
    
    Cong Wang's avatar
    Cong Wang committed
    					    sha, np->dev->dev_addr,
    					    send_skb->len) < 0) {
    				kfree_skb(send_skb);
    				continue;
    			}
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    Cong Wang's avatar
    Cong Wang committed
    			/*
    			 * Fill out the arp protocol part.
    			 *
    			 * we only support ethernet device type,
    			 * which (according to RFC 1390) should
    			 * always equal 1 (Ethernet).
    			 */
    
    			arp->ar_hrd = htons(np->dev->type);
    			arp->ar_pro = htons(ETH_P_IP);
    			arp->ar_hln = np->dev->addr_len;
    			arp->ar_pln = 4;
    			arp->ar_op = htons(type);
    
    			arp_ptr = (unsigned char *)(arp + 1);
    			memcpy(arp_ptr, np->dev->dev_addr, np->dev->addr_len);
    			arp_ptr += np->dev->addr_len;
    			memcpy(arp_ptr, &tip, 4);
    			arp_ptr += 4;
    			memcpy(arp_ptr, sha, np->dev->addr_len);
    			arp_ptr += np->dev->addr_len;
    			memcpy(arp_ptr, &sip, 4);
    
    			netpoll_send_skb(np, send_skb);
    
    			/* If there are several rx_hooks for the same address,
    			   we're fine by sending a single reply */
    			break;
    
    Cong Wang's avatar
    Cong Wang committed
    		spin_unlock_irqrestore(&npinfo->rx_lock, flags);
    
    Cong Wang's avatar
    Cong Wang committed
    	} else if( proto == ETH_P_IPV6) {
    #if IS_ENABLED(CONFIG_IPV6)
    		struct nd_msg *msg;
    		u8 *lladdr = NULL;
    		struct ipv6hdr *hdr;
    		struct icmp6hdr *icmp6h;
    		const struct in6_addr *saddr;
    		const struct in6_addr *daddr;
    		struct inet6_dev *in6_dev = NULL;
    		struct in6_addr *target;
    
    		in6_dev = in6_dev_get(skb->dev);
    		if (!in6_dev || !in6_dev->cnf.accept_ra)
    			return;
    
    		if (!pskb_may_pull(skb, skb->len))
    			return;
    
    		msg = (struct nd_msg *)skb_transport_header(skb);
    
    		__skb_push(skb, skb->data - skb_transport_header(skb));
    
    		if (ipv6_hdr(skb)->hop_limit != 255)
    			return;
    		if (msg->icmph.icmp6_code != 0)
    			return;
    		if (msg->icmph.icmp6_type != NDISC_NEIGHBOUR_SOLICITATION)
    			return;
    
    		saddr = &ipv6_hdr(skb)->saddr;
    		daddr = &ipv6_hdr(skb)->daddr;
    
    		size = sizeof(struct icmp6hdr) + sizeof(struct in6_addr);
    
    		spin_lock_irqsave(&npinfo->rx_lock, flags);
    		list_for_each_entry_safe(np, tmp, &npinfo->rx_np, rx) {
    
    			if (!ipv6_addr_equal(daddr, &np->local_ip.in6))
    
    Cong Wang's avatar
    Cong Wang committed
    				continue;
    
    			hlen = LL_RESERVED_SPACE(np->dev);
    			tlen = np->dev->needed_tailroom;
    			send_skb = find_skb(np, size + hlen + tlen, hlen);
    			if (!send_skb)
    				continue;
    
    			send_skb->protocol = htons(ETH_P_IPV6);
    			send_skb->dev = skb->dev;
    
    			skb_reset_network_header(send_skb);
    			skb_put(send_skb, sizeof(struct ipv6hdr));
    			hdr = ipv6_hdr(send_skb);
    
    			*(__be32*)hdr = htonl(0x60000000);
    
    			hdr->payload_len = htons(size);
    			hdr->nexthdr = IPPROTO_ICMPV6;
    			hdr->hop_limit = 255;
    			hdr->saddr = *saddr;
    			hdr->daddr = *daddr;
    
    			send_skb->transport_header = send_skb->tail;
    			skb_put(send_skb, size);
    
    			icmp6h = (struct icmp6hdr *)skb_transport_header(skb);
    			icmp6h->icmp6_type = NDISC_NEIGHBOUR_ADVERTISEMENT;
    			icmp6h->icmp6_router = 0;
    			icmp6h->icmp6_solicited = 1;
    			target = (struct in6_addr *)skb_transport_header(send_skb) + sizeof(struct icmp6hdr);
    			*target = msg->target;
    			icmp6h->icmp6_cksum = csum_ipv6_magic(saddr, daddr, size,
    							      IPPROTO_ICMPV6,
    							      csum_partial(icmp6h,
    									   size, 0));
    
    			if (dev_hard_header(send_skb, skb->dev, ETH_P_IPV6,
    					    lladdr, np->dev->dev_addr,
    					    send_skb->len) < 0) {
    				kfree_skb(send_skb);
    				continue;
    			}
    
    			netpoll_send_skb(np, send_skb);
    
    			/* If there are several rx_hooks for the same address,
    			   we're fine by sending a single reply */
    			break;
    		}
    		spin_unlock_irqrestore(&npinfo->rx_lock, flags);
    #endif
    
    Cong Wang's avatar
    Cong Wang committed
    static bool pkt_is_ns(struct sk_buff *skb)
    {
    	struct nd_msg *msg;
    	struct ipv6hdr *hdr;
    
    	if (skb->protocol != htons(ETH_P_ARP))
    		return false;
    	if (!pskb_may_pull(skb, sizeof(struct ipv6hdr) + sizeof(struct nd_msg)))
    		return false;
    
    	msg = (struct nd_msg *)skb_transport_header(skb);
    	__skb_push(skb, skb->data - skb_transport_header(skb));
    	hdr = ipv6_hdr(skb);
    
    	if (hdr->nexthdr != IPPROTO_ICMPV6)
    		return false;
    	if (hdr->hop_limit != 255)
    		return false;
    	if (msg->icmph.icmp6_code != 0)
    		return false;
    	if (msg->icmph.icmp6_type != NDISC_NEIGHBOUR_SOLICITATION)
    		return false;
    
    	return true;
    }
    
    
    int __netpoll_rx(struct sk_buff *skb, struct netpoll_info *npinfo)
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    {
    	int proto, len, ulen;
    
    	const struct iphdr *iph;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	struct udphdr *uh;
    
    	if (list_empty(&npinfo->rx_np))
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    		goto out;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	if (skb->dev->type != ARPHRD_ETHER)
    		goto out;
    
    
    	/* check if netpoll clients need ARP */
    
    Cong Wang's avatar
    Cong Wang committed
    	if (skb->protocol == htons(ETH_P_ARP) && atomic_read(&trapped)) {
    		skb_queue_tail(&npinfo->neigh_tx, skb);
    		return 1;
    	} else if (pkt_is_ns(skb) && atomic_read(&trapped)) {
    
    Cong Wang's avatar
    Cong Wang committed
    		skb_queue_tail(&npinfo->neigh_tx, skb);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    		return 1;
    	}
    
    
    	if (skb->protocol == cpu_to_be16(ETH_P_8021Q)) {
    		skb = vlan_untag(skb);
    		if (unlikely(!skb))
    			goto out;
    	}
    
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	proto = ntohs(eth_hdr(skb)->h_proto);
    
    Cong Wang's avatar
    Cong Wang committed
    	if (proto != ETH_P_IP && proto != ETH_P_IPV6)
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    		goto out;
    	if (skb->pkt_type == PACKET_OTHERHOST)
    		goto out;
    	if (skb_shared(skb))
    		goto out;
    
    
    Cong Wang's avatar
    Cong Wang committed
    	if (proto == ETH_P_IP) {
    		if (!pskb_may_pull(skb, sizeof(struct iphdr)))
    			goto out;
    		iph = (struct iphdr *)skb->data;
    		if (iph->ihl < 5 || iph->version != 4)
    			goto out;
    		if (!pskb_may_pull(skb, iph->ihl*4))
    			goto out;
    		iph = (struct iphdr *)skb->data;
    		if (ip_fast_csum((u8 *)iph, iph->ihl) != 0)
    			goto out;
    
    Cong Wang's avatar
    Cong Wang committed
    		len = ntohs(iph->tot_len);
    		if (skb->len < len || len < iph->ihl*4)
    			goto out;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    Cong Wang's avatar
    Cong Wang committed
    		/*
    		 * Our transport medium may have padded the buffer out.
    		 * Now We trim to the true length of the frame.
    		 */
    		if (pskb_trim_rcsum(skb, len))
    			goto out;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    Cong Wang's avatar
    Cong Wang committed
    		iph = (struct iphdr *)skb->data;
    		if (iph->protocol != IPPROTO_UDP)
    			goto out;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    Cong Wang's avatar
    Cong Wang committed
    		len -= iph->ihl*4;
    		uh = (struct udphdr *)(((char *)iph) + iph->ihl*4);
    		ulen = ntohs(uh->len);
    
    Cong Wang's avatar
    Cong Wang committed
    		if (ulen != len)
    			goto out;
    		if (checksum_udp(skb, uh, ulen, iph->saddr, iph->daddr))
    			goto out;
    		list_for_each_entry_safe(np, tmp, &npinfo->rx_np, rx) {
    			if (np->local_ip.ip && np->local_ip.ip != iph->daddr)
    				continue;
    			if (np->remote_ip.ip && np->remote_ip.ip != iph->saddr)
    				continue;
    			if (np->local_port && np->local_port != ntohs(uh->dest))
    				continue;
    
    			np->rx_hook(np, ntohs(uh->source),
    				       (char *)(uh+1),
    				       ulen - sizeof(struct udphdr));
    			hits++;
    		}
    
    Cong Wang's avatar
    Cong Wang committed
    	} else {
    #if IS_ENABLED(CONFIG_IPV6)
    		const struct ipv6hdr *ip6h;
    
    		if (!pskb_may_pull(skb, sizeof(struct ipv6hdr)))
    			goto out;
    		ip6h = (struct ipv6hdr *)skb->data;
    		if (ip6h->version != 6)
    			goto out;
    		len = ntohs(ip6h->payload_len);
    		if (!len)
    			goto out;
    		if (len + sizeof(struct ipv6hdr) > skb->len)
    			goto out;
    		if (pskb_trim_rcsum(skb, len + sizeof(struct ipv6hdr)))
    			goto out;
    		ip6h = ipv6_hdr(skb);
    		if (!pskb_may_pull(skb, sizeof(struct udphdr)))
    			goto out;
    		uh = udp_hdr(skb);
    		ulen = ntohs(uh->len);
    		if (ulen != skb->len)
    			goto out;
    		if (udp6_csum_init(skb, uh, IPPROTO_UDP))
    			goto out;
    		list_for_each_entry_safe(np, tmp, &npinfo->rx_np, rx) {
    
    			if (!ipv6_addr_equal(&np->local_ip.in6, &ip6h->daddr))
    
    Cong Wang's avatar
    Cong Wang committed
    				continue;
    
    			if (!ipv6_addr_equal(&np->remote_ip.in6, &ip6h->saddr))
    
    Cong Wang's avatar
    Cong Wang committed
    				continue;
    			if (np->local_port && np->local_port != ntohs(uh->dest))
    				continue;
    
    			np->rx_hook(np, ntohs(uh->source),
    				       (char *)(uh+1),
    				       ulen - sizeof(struct udphdr));
    			hits++;
    		}
    #endif
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    	kfree_skb(skb);
    	return 1;
    
    out:
    	if (atomic_read(&trapped)) {
    		kfree_skb(skb);
    		return 1;
    	}
    
    	return 0;
    }
    
    
    void netpoll_print_options(struct netpoll *np)
    {
    
    	np_info(np, "local port %d\n", np->local_port);
    
    Cong Wang's avatar
    Cong Wang committed
    	if (np->ipv6)
    		np_info(np, "local IPv6 address %pI6c\n", &np->local_ip.in6);
    	else
    
    Cong Wang's avatar
    Cong Wang committed
    		np_info(np, "local IPv4 address %pI4\n", &np->local_ip.ip);
    
    	np_info(np, "interface '%s'\n", np->dev_name);
    	np_info(np, "remote port %d\n", np->remote_port);
    
    Cong Wang's avatar
    Cong Wang committed
    	if (np->ipv6)
    		np_info(np, "remote IPv6 address %pI6c\n", &np->remote_ip.in6);
    	else
    
    Cong Wang's avatar
    Cong Wang committed
    		np_info(np, "remote IPv4 address %pI4\n", &np->remote_ip.ip);
    
    	np_info(np, "remote ethernet address %pM\n", np->remote_mac);
    
    EXPORT_SYMBOL(netpoll_print_options);
    
    Cong Wang's avatar
    Cong Wang committed
    static int netpoll_parse_ip_addr(const char *str, union inet_addr *addr)
    {
    	const char *end;
    
    	if (!strchr(str, ':') &&
    	    in4_pton(str, -1, (void *)addr, -1, &end) > 0) {
    		if (!*end)
    			return 0;
    	}
    	if (in6_pton(str, -1, addr->in6.s6_addr, -1, &end) > 0) {
    #if IS_ENABLED(CONFIG_IPV6)
    		if (!*end)
    			return 1;
    #else
    		return -1;
    #endif
    	}
    	return -1;
    }
    
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    int netpoll_parse_options(struct netpoll *np, char *opt)
    {
    	char *cur=opt, *delim;
    
    Cong Wang's avatar
    Cong Wang committed
    	int ipv6;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    	if (*cur != '@') {
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    		if ((delim = strchr(cur, '@')) == NULL)
    			goto parse_failed;
    
    		if (kstrtou16(cur, 10, &np->local_port))
    			goto parse_failed;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	}
    	cur++;
    
    
    	if (*cur != '/') {
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    		if ((delim = strchr(cur, '/')) == NULL)
    			goto parse_failed;
    
    Cong Wang's avatar
    Cong Wang committed
    		ipv6 = netpoll_parse_ip_addr(cur, &np->local_ip);
    		if (ipv6 < 0)
    			goto parse_failed;
    		else
    			np->ipv6 = (bool)ipv6;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	}
    	cur++;
    
    
    	if (*cur != ',') {
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    		/* parse out dev name */
    		if ((delim = strchr(cur, ',')) == NULL)
    			goto parse_failed;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    		strlcpy(np->dev_name, cur, sizeof(np->dev_name));
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	}
    	cur++;
    
    
    	if (*cur != '@') {
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    		/* dst port */
    		if ((delim = strchr(cur, '@')) == NULL)
    			goto parse_failed;
    
    		if (*cur == ' ' || *cur == '\t')
    
    			np_info(np, "warning: whitespace is not allowed\n");
    
    		if (kstrtou16(cur, 10, &np->remote_port))
    			goto parse_failed;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	}
    	cur++;
    
    	/* dst ip */
    	if ((delim = strchr(cur, '/')) == NULL)
    		goto parse_failed;
    
    Cong Wang's avatar
    Cong Wang committed
    	ipv6 = netpoll_parse_ip_addr(cur, &np->remote_ip);
    	if (ipv6 < 0)
    		goto parse_failed;
    	else if (np->ipv6 != (bool)ipv6)
    		goto parse_failed;
    	else
    		np->ipv6 = (bool)ipv6;
    
    	cur = delim + 1;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    	if (*cur != 0) {
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    		/* MAC address */
    
    		if (!mac_pton(cur, np->remote_mac))
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    			goto parse_failed;
    	}
    
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    	return 0;
    
     parse_failed:
    
    	np_info(np, "couldn't parse config at '%s'!\n", cur);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    	return -1;
    }
    
    EXPORT_SYMBOL(netpoll_parse_options);
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    int __netpoll_setup(struct netpoll *np, struct net_device *ndev, gfp_t gfp)
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    {
    
    	struct netpoll_info *npinfo;
    
    	const struct net_device_ops *ops;
    
    	int err;
    
    Linus Torvalds's avatar
    Linus Torvalds committed
    
    
    	np->dev = ndev;
    	strlcpy(np->dev_name, ndev->name, IFNAMSIZ);
    
    
    	if ((ndev->priv_flags & IFF_DISABLE_NETPOLL) ||
    	    !ndev->netdev_ops->ndo_poll_controller) {
    
    		np_err(np, "%s doesn't support polling, aborting\n",
    		       np->dev_name);
    
    		err = -ENOTSUPP;
    		goto out;
    	}
    
    	if (!ndev->npinfo) {
    
    		npinfo = kmalloc(sizeof(*npinfo), gfp);
    
    		if (!npinfo) {
    			err = -ENOMEM;
    			goto out;