netpoll.c 21.8 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11
/*
 * 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.
 */

12
#include <linux/moduleparam.h>
L
Linus Torvalds 已提交
13 14 15
#include <linux/netdevice.h>
#include <linux/etherdevice.h>
#include <linux/string.h>
16
#include <linux/if_arp.h>
L
Linus Torvalds 已提交
17 18 19 20 21 22 23 24
#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>
25
#include <linux/slab.h>
L
Linus Torvalds 已提交
26 27 28
#include <net/tcp.h>
#include <net/udp.h>
#include <asm/unaligned.h>
29
#include <trace/events/napi.h>
L
Linus Torvalds 已提交
30 31 32 33 34 35 36 37 38

/*
 * 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

39
static struct sk_buff_head skb_pool;
L
Linus Torvalds 已提交
40 41 42

static atomic_t trapped;

S
Stephen Hemminger 已提交
43
#define USEC_PER_POLL	50
44 45
#define NETPOLL_RX_ENABLED  1
#define NETPOLL_RX_DROP     2
L
Linus Torvalds 已提交
46 47 48 49 50

#define MAX_SKB_SIZE \
		(MAX_UDP_CHUNK + sizeof(struct udphdr) + \
				sizeof(struct iphdr) + sizeof(struct ethhdr))

51
static void zap_completion_queue(void);
52
static void arp_reply(struct sk_buff *skb);
L
Linus Torvalds 已提交
53

54 55 56
static unsigned int carrier_timeout = 4;
module_param(carrier_timeout, uint, 0644);

D
David Howells 已提交
57
static void queue_process(struct work_struct *work)
L
Linus Torvalds 已提交
58
{
59 60
	struct netpoll_info *npinfo =
		container_of(work, struct netpoll_info, tx_work.work);
L
Linus Torvalds 已提交
61
	struct sk_buff *skb;
I
Ingo Molnar 已提交
62
	unsigned long flags;
L
Linus Torvalds 已提交
63

S
Stephen Hemminger 已提交
64 65
	while ((skb = skb_dequeue(&npinfo->txq))) {
		struct net_device *dev = skb->dev;
66
		const struct net_device_ops *ops = dev->netdev_ops;
67
		struct netdev_queue *txq;
L
Linus Torvalds 已提交
68

S
Stephen Hemminger 已提交
69 70 71 72
		if (!netif_device_present(dev) || !netif_running(dev)) {
			__kfree_skb(skb);
			continue;
		}
L
Linus Torvalds 已提交
73

74 75
		txq = netdev_get_tx_queue(dev, skb_get_queue_mapping(skb));

I
Ingo Molnar 已提交
76
		local_irq_save(flags);
77
		__netif_tx_lock(txq, smp_processor_id());
78
		if (netif_tx_queue_frozen_or_stopped(txq) ||
79
		    ops->ndo_start_xmit(skb, dev) != NETDEV_TX_OK) {
S
Stephen Hemminger 已提交
80
			skb_queue_head(&npinfo->txq, skb);
81
			__netif_tx_unlock(txq);
I
Ingo Molnar 已提交
82
			local_irq_restore(flags);
L
Linus Torvalds 已提交
83

84
			schedule_delayed_work(&npinfo->tx_work, HZ/10);
S
Stephen Hemminger 已提交
85 86
			return;
		}
87
		__netif_tx_unlock(txq);
I
Ingo Molnar 已提交
88
		local_irq_restore(flags);
L
Linus Torvalds 已提交
89 90 91
	}
}

92 93
static __sum16 checksum_udp(struct sk_buff *skb, struct udphdr *uh,
			    unsigned short ulen, __be32 saddr, __be32 daddr)
L
Linus Torvalds 已提交
94
{
95
	__wsum psum;
96

97
	if (uh->check == 0 || skb_csum_unnecessary(skb))
L
Linus Torvalds 已提交
98 99
		return 0;

100 101
	psum = csum_tcpudp_nofold(saddr, daddr, ulen, IPPROTO_UDP, 0);

102
	if (skb->ip_summed == CHECKSUM_COMPLETE &&
103
	    !csum_fold(csum_add(psum, skb->csum)))
104
		return 0;
L
Linus Torvalds 已提交
105

106
	skb->csum = psum;
L
Linus Torvalds 已提交
107

108
	return __skb_checksum_complete(skb);
L
Linus Torvalds 已提交
109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126
}

/*
 * 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.
 */
127 128 129 130 131 132 133 134 135 136 137 138
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;

139
	npinfo->rx_flags |= NETPOLL_RX_DROP;
140
	atomic_inc(&trapped);
141
	set_bit(NAPI_STATE_NPSVC, &napi->state);
142 143

	work = napi->poll(napi, budget);
144
	trace_napi_poll(napi);
145

146
	clear_bit(NAPI_STATE_NPSVC, &napi->state);
147
	atomic_dec(&trapped);
148
	npinfo->rx_flags &= ~NETPOLL_RX_DROP;
149 150 151 152

	return budget - work;
}

153
static void poll_napi(struct net_device *dev)
L
Linus Torvalds 已提交
154
{
155
	struct napi_struct *napi;
L
Linus Torvalds 已提交
156 157
	int budget = 16;

158
	list_for_each_entry(napi, &dev->napi_list, dev_list) {
159
		if (napi->poll_owner != smp_processor_id() &&
160
		    spin_trylock(&napi->poll_lock)) {
161
			budget = poll_one_napi(dev->npinfo, napi, budget);
162
			spin_unlock(&napi->poll_lock);
163 164 165

			if (!budget)
				break;
166
		}
L
Linus Torvalds 已提交
167 168 169
	}
}

170 171
static void service_arp_queue(struct netpoll_info *npi)
{
172 173
	if (npi) {
		struct sk_buff *skb;
174

175 176
		while ((skb = skb_dequeue(&npi->arp_tx)))
			arp_reply(skb);
177 178 179
	}
}

180
void netpoll_poll_dev(struct net_device *dev)
L
Linus Torvalds 已提交
181
{
182
	const struct net_device_ops *ops;
183

184 185 186 187 188
	if (!dev || !netif_running(dev))
		return;

	ops = dev->netdev_ops;
	if (!ops->ndo_poll_controller)
L
Linus Torvalds 已提交
189 190 191
		return;

	/* Process pending work on NIC */
192
	ops->ndo_poll_controller(dev);
193 194

	poll_napi(dev);
L
Linus Torvalds 已提交
195

196 197 198 199 200 201 202 203 204 205 206
	if (dev->priv_flags & IFF_SLAVE) {
		if (dev->npinfo) {
			struct net_device *bond_dev = dev->master;
			struct sk_buff *skb;
			while ((skb = skb_dequeue(&dev->npinfo->arp_tx))) {
				skb->dev = bond_dev;
				skb_queue_tail(&bond_dev->npinfo->arp_tx, skb);
			}
		}
	}

207
	service_arp_queue(dev->npinfo);
208

209
	zap_completion_queue();
L
Linus Torvalds 已提交
210
}
E
Eric Dumazet 已提交
211
EXPORT_SYMBOL(netpoll_poll_dev);
L
Linus Torvalds 已提交
212

213 214 215 216
void netpoll_poll(struct netpoll *np)
{
	netpoll_poll_dev(np->dev);
}
E
Eric Dumazet 已提交
217
EXPORT_SYMBOL(netpoll_poll);
218

L
Linus Torvalds 已提交
219 220 221 222 223
static void refill_skbs(void)
{
	struct sk_buff *skb;
	unsigned long flags;

224 225
	spin_lock_irqsave(&skb_pool.lock, flags);
	while (skb_pool.qlen < MAX_SKBS) {
L
Linus Torvalds 已提交
226 227 228 229
		skb = alloc_skb(MAX_SKB_SIZE, GFP_ATOMIC);
		if (!skb)
			break;

230
		__skb_queue_tail(&skb_pool, skb);
L
Linus Torvalds 已提交
231
	}
232
	spin_unlock_irqrestore(&skb_pool.lock, flags);
L
Linus Torvalds 已提交
233 234
}

235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262
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);
}

263
static struct sk_buff *find_skb(struct netpoll *np, int len, int reserve)
L
Linus Torvalds 已提交
264
{
265 266
	int count = 0;
	struct sk_buff *skb;
L
Linus Torvalds 已提交
267

268
	zap_completion_queue();
269
	refill_skbs();
L
Linus Torvalds 已提交
270 271 272
repeat:

	skb = alloc_skb(len, GFP_ATOMIC);
273 274
	if (!skb)
		skb = skb_dequeue(&skb_pool);
L
Linus Torvalds 已提交
275 276

	if (!skb) {
277 278 279
		if (++count < 10) {
			netpoll_poll(np);
			goto repeat;
L
Linus Torvalds 已提交
280
		}
281
		return NULL;
L
Linus Torvalds 已提交
282 283 284 285 286 287 288
	}

	atomic_set(&skb->users, 1);
	skb_reserve(skb, reserve);
	return skb;
}

289 290 291 292 293 294 295 296 297 298 299
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;
}

300 301
void netpoll_send_skb_on_dev(struct netpoll *np, struct sk_buff *skb,
			     struct net_device *dev)
L
Linus Torvalds 已提交
302
{
S
Stephen Hemminger 已提交
303 304
	int status = NETDEV_TX_BUSY;
	unsigned long tries;
305
	const struct net_device_ops *ops = dev->netdev_ops;
H
Herbert Xu 已提交
306
	/* It is up to the caller to keep npinfo alive. */
307
	struct netpoll_info *npinfo = np->dev->npinfo;
S
Stephen Hemminger 已提交
308

309 310 311 312
	if (!npinfo || !netif_running(dev) || !netif_device_present(dev)) {
		__kfree_skb(skb);
		return;
	}
S
Stephen Hemminger 已提交
313 314

	/* don't get messages out of order, and no recursion */
315
	if (skb_queue_len(&npinfo->txq) == 0 && !netpoll_owner_active(dev)) {
316
		struct netdev_queue *txq;
317 318
		unsigned long flags;

319 320
		txq = netdev_get_tx_queue(dev, skb_get_queue_mapping(skb));

321
		local_irq_save(flags);
322 323 324
		/* try until next clock tick */
		for (tries = jiffies_to_usecs(1)/USEC_PER_POLL;
		     tries > 0; --tries) {
325
			if (__netif_tx_trylock(txq)) {
E
Eric Dumazet 已提交
326
				if (!netif_tx_queue_stopped(txq)) {
327
					status = ops->ndo_start_xmit(skb, dev);
E
Eric Dumazet 已提交
328 329 330
					if (status == NETDEV_TX_OK)
						txq_trans_update(txq);
				}
331
				__netif_tx_unlock(txq);
332 333 334 335 336

				if (status == NETDEV_TX_OK)
					break;

			}
337 338 339 340 341

			/* tickle device maybe there is some cleanup */
			netpoll_poll(np);

			udelay(USEC_PER_POLL);
M
Matt Mackall 已提交
342
		}
343 344 345 346 347

		WARN_ONCE(!irqs_disabled(),
			"netpoll_send_skb(): %s enabled interrupts in poll (%pF)\n",
			dev->name, ops->ndo_start_xmit);

348
		local_irq_restore(flags);
L
Linus Torvalds 已提交
349 350
	}

S
Stephen Hemminger 已提交
351
	if (status != NETDEV_TX_OK) {
S
Stephen Hemminger 已提交
352
		skb_queue_tail(&npinfo->txq, skb);
353
		schedule_delayed_work(&npinfo->tx_work,0);
L
Linus Torvalds 已提交
354 355
	}
}
356
EXPORT_SYMBOL(netpoll_send_skb_on_dev);
L
Linus Torvalds 已提交
357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373

void netpoll_send_udp(struct netpoll *np, const char *msg, int len)
{
	int total_len, eth_len, ip_len, udp_len;
	struct sk_buff *skb;
	struct udphdr *udph;
	struct iphdr *iph;
	struct ethhdr *eth;

	udp_len = len + sizeof(*udph);
	ip_len = eth_len = udp_len + sizeof(*iph);
	total_len = eth_len + ETH_HLEN + NET_IP_ALIGN;

	skb = find_skb(np, total_len, total_len - len);
	if (!skb)
		return;

374
	skb_copy_to_linear_data(skb, msg, len);
L
Linus Torvalds 已提交
375 376
	skb->len += len;

377 378 379
	skb_push(skb, sizeof(*udph));
	skb_reset_transport_header(skb);
	udph = udp_hdr(skb);
L
Linus Torvalds 已提交
380 381 382 383
	udph->source = htons(np->local_port);
	udph->dest = htons(np->remote_port);
	udph->len = htons(udp_len);
	udph->check = 0;
384 385
	udph->check = csum_tcpudp_magic(np->local_ip,
					np->remote_ip,
386
					udp_len, IPPROTO_UDP,
387
					csum_partial(udph, udp_len, 0));
388
	if (udph->check == 0)
389
		udph->check = CSUM_MANGLED_0;
L
Linus Torvalds 已提交
390

391 392
	skb_push(skb, sizeof(*iph));
	skb_reset_network_header(skb);
393
	iph = ip_hdr(skb);
L
Linus Torvalds 已提交
394 395 396 397 398 399 400 401 402 403

	/* 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       = 0;
	iph->frag_off = 0;
	iph->ttl      = 64;
	iph->protocol = IPPROTO_UDP;
	iph->check    = 0;
404 405
	put_unaligned(np->local_ip, &(iph->saddr));
	put_unaligned(np->remote_ip, &(iph->daddr));
L
Linus Torvalds 已提交
406 407 408
	iph->check    = ip_fast_csum((unsigned char *)iph, iph->ihl);

	eth = (struct ethhdr *) skb_push(skb, ETH_HLEN);
409
	skb_reset_mac_header(skb);
410
	skb->protocol = eth->h_proto = htons(ETH_P_IP);
411 412
	memcpy(eth->h_source, np->dev->dev_addr, ETH_ALEN);
	memcpy(eth->h_dest, np->remote_mac, ETH_ALEN);
L
Linus Torvalds 已提交
413 414 415 416 417

	skb->dev = np->dev;

	netpoll_send_skb(np, skb);
}
E
Eric Dumazet 已提交
418
EXPORT_SYMBOL(netpoll_send_udp);
L
Linus Torvalds 已提交
419 420 421

static void arp_reply(struct sk_buff *skb)
{
422
	struct netpoll_info *npinfo = skb->dev->npinfo;
L
Linus Torvalds 已提交
423 424 425
	struct arphdr *arp;
	unsigned char *arp_ptr;
	int size, type = ARPOP_REPLY, ptype = ETH_P_ARP;
A
Al Viro 已提交
426
	__be32 sip, tip;
427
	unsigned char *sha;
L
Linus Torvalds 已提交
428
	struct sk_buff *send_skb;
429 430 431 432 433 434 435 436 437 438 439 440 441 442 443
	struct netpoll *np, *tmp;
	unsigned long flags;
	int hits = 0;

	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);
L
Linus Torvalds 已提交
444

445 446
	/* No netpoll struct is using this dev */
	if (!hits)
447
		return;
L
Linus Torvalds 已提交
448 449 450 451 452

	/* No arp on this interface */
	if (skb->dev->flags & IFF_NOARP)
		return;

453
	if (!pskb_may_pull(skb, arp_hdr_len(skb->dev)))
L
Linus Torvalds 已提交
454 455
		return;

456
	skb_reset_network_header(skb);
457
	skb_reset_transport_header(skb);
458
	arp = arp_hdr(skb);
L
Linus Torvalds 已提交
459 460 461 462 463 464 465

	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;

466 467 468 469
	arp_ptr = (unsigned char *)(arp+1);
	/* save the location of the src hw addr */
	sha = arp_ptr;
	arp_ptr += skb->dev->addr_len;
L
Linus Torvalds 已提交
470
	memcpy(&sip, arp_ptr, 4);
471
	arp_ptr += 4;
472 473
	/* If we actually cared about dst hw addr,
	   it would get copied here */
474
	arp_ptr += skb->dev->addr_len;
L
Linus Torvalds 已提交
475 476 477
	memcpy(&tip, arp_ptr, 4);

	/* Should we ignore arp? */
478
	if (ipv4_is_loopback(tip) || ipv4_is_multicast(tip))
L
Linus Torvalds 已提交
479 480
		return;

481
	size = arp_hdr_len(skb->dev);
L
Linus Torvalds 已提交
482

483 484 485 486
	spin_lock_irqsave(&npinfo->rx_lock, flags);
	list_for_each_entry_safe(np, tmp, &npinfo->rx_np, rx) {
		if (tip != np->local_ip)
			continue;
L
Linus Torvalds 已提交
487

488 489 490 491
		send_skb = find_skb(np, size + LL_ALLOCATED_SPACE(np->dev),
				    LL_RESERVED_SPACE(np->dev));
		if (!send_skb)
			continue;
L
Linus Torvalds 已提交
492

493 494 495 496
		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);
L
Linus Torvalds 已提交
497

498 499 500 501 502 503 504
		/* Fill the device header for the ARP frame */
		if (dev_hard_header(send_skb, skb->dev, ptype,
				    sha, np->dev->dev_addr,
				    send_skb->len) < 0) {
			kfree_skb(send_skb);
			continue;
		}
L
Linus Torvalds 已提交
505

506 507 508 509 510 511 512
		/*
		 * Fill out the arp protocol part.
		 *
		 * we only support ethernet device type,
		 * which (according to RFC 1390) should
		 * always equal 1 (Ethernet).
		 */
L
Linus Torvalds 已提交
513

514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535
		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;
	}
	spin_unlock_irqrestore(&npinfo->rx_lock, flags);
L
Linus Torvalds 已提交
536 537 538 539 540
}

int __netpoll_rx(struct sk_buff *skb)
{
	int proto, len, ulen;
541
	int hits = 0;
L
Linus Torvalds 已提交
542 543
	struct iphdr *iph;
	struct udphdr *uh;
544 545
	struct netpoll_info *npinfo = skb->dev->npinfo;
	struct netpoll *np, *tmp;
546

547
	if (list_empty(&npinfo->rx_np))
L
Linus Torvalds 已提交
548
		goto out;
549

L
Linus Torvalds 已提交
550 551 552
	if (skb->dev->type != ARPHRD_ETHER)
		goto out;

553
	/* check if netpoll clients need ARP */
554
	if (skb->protocol == htons(ETH_P_ARP) &&
L
Linus Torvalds 已提交
555
	    atomic_read(&trapped)) {
556
		skb_queue_tail(&npinfo->arp_tx, skb);
L
Linus Torvalds 已提交
557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581
		return 1;
	}

	proto = ntohs(eth_hdr(skb)->h_proto);
	if (proto != ETH_P_IP)
		goto out;
	if (skb->pkt_type == PACKET_OTHERHOST)
		goto out;
	if (skb_shared(skb))
		goto out;

	iph = (struct iphdr *)skb->data;
	if (!pskb_may_pull(skb, sizeof(struct iphdr)))
		goto out;
	if (iph->ihl < 5 || iph->version != 4)
		goto out;
	if (!pskb_may_pull(skb, iph->ihl*4))
		goto out;
	if (ip_fast_csum((u8 *)iph, iph->ihl) != 0)
		goto out;

	len = ntohs(iph->tot_len);
	if (skb->len < len || len < iph->ihl*4)
		goto out;

582 583 584 585 586 587 588
	/*
	 * 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;

L
Linus Torvalds 已提交
589 590 591 592 593 594 595 596 597
	if (iph->protocol != IPPROTO_UDP)
		goto out;

	len -= iph->ihl*4;
	uh = (struct udphdr *)(((char *)iph) + iph->ihl*4);
	ulen = ntohs(uh->len);

	if (ulen != len)
		goto out;
598
	if (checksum_udp(skb, uh, ulen, iph->saddr, iph->daddr))
L
Linus Torvalds 已提交
599 600
		goto out;

601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616
	list_for_each_entry_safe(np, tmp, &npinfo->rx_np, rx) {
		if (np->local_ip && np->local_ip != iph->daddr)
			continue;
		if (np->remote_ip && np->remote_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++;
	}

	if (!hits)
		goto out;
L
Linus Torvalds 已提交
617 618 619 620 621 622 623 624 625 626 627 628 629

	kfree_skb(skb);
	return 1;

out:
	if (atomic_read(&trapped)) {
		kfree_skb(skb);
		return 1;
	}

	return 0;
}

630 631 632 633
void netpoll_print_options(struct netpoll *np)
{
	printk(KERN_INFO "%s: local port %d\n",
			 np->name, np->local_port);
634 635
	printk(KERN_INFO "%s: local IP %pI4\n",
			 np->name, &np->local_ip);
636
	printk(KERN_INFO "%s: interface '%s'\n",
637 638 639
			 np->name, np->dev_name);
	printk(KERN_INFO "%s: remote port %d\n",
			 np->name, np->remote_port);
640 641
	printk(KERN_INFO "%s: remote IP %pI4\n",
			 np->name, &np->remote_ip);
J
Johannes Berg 已提交
642 643
	printk(KERN_INFO "%s: remote ethernet address %pM\n",
	                 np->name, np->remote_mac);
644
}
E
Eric Dumazet 已提交
645
EXPORT_SYMBOL(netpoll_print_options);
646

L
Linus Torvalds 已提交
647 648 649 650
int netpoll_parse_options(struct netpoll *np, char *opt)
{
	char *cur=opt, *delim;

651
	if (*cur != '@') {
L
Linus Torvalds 已提交
652 653
		if ((delim = strchr(cur, '@')) == NULL)
			goto parse_failed;
654 655 656
		*delim = 0;
		np->local_port = simple_strtol(cur, NULL, 10);
		cur = delim;
L
Linus Torvalds 已提交
657 658 659
	}
	cur++;

660
	if (*cur != '/') {
L
Linus Torvalds 已提交
661 662
		if ((delim = strchr(cur, '/')) == NULL)
			goto parse_failed;
663
		*delim = 0;
664
		np->local_ip = in_aton(cur);
665
		cur = delim;
L
Linus Torvalds 已提交
666 667 668
	}
	cur++;

669
	if (*cur != ',') {
L
Linus Torvalds 已提交
670 671 672
		/* parse out dev name */
		if ((delim = strchr(cur, ',')) == NULL)
			goto parse_failed;
673
		*delim = 0;
L
Linus Torvalds 已提交
674
		strlcpy(np->dev_name, cur, sizeof(np->dev_name));
675
		cur = delim;
L
Linus Torvalds 已提交
676 677 678
	}
	cur++;

679
	if (*cur != '@') {
L
Linus Torvalds 已提交
680 681 682
		/* dst port */
		if ((delim = strchr(cur, '@')) == NULL)
			goto parse_failed;
683
		*delim = 0;
684 685 686
		if (*cur == ' ' || *cur == '\t')
			printk(KERN_INFO "%s: warning: whitespace"
					"is not allowed\n", np->name);
687 688
		np->remote_port = simple_strtol(cur, NULL, 10);
		cur = delim;
L
Linus Torvalds 已提交
689 690 691 692 693 694
	}
	cur++;

	/* dst ip */
	if ((delim = strchr(cur, '/')) == NULL)
		goto parse_failed;
695
	*delim = 0;
696
	np->remote_ip = in_aton(cur);
697
	cur = delim + 1;
L
Linus Torvalds 已提交
698

699
	if (*cur != 0) {
L
Linus Torvalds 已提交
700 701 702
		/* MAC address */
		if ((delim = strchr(cur, ':')) == NULL)
			goto parse_failed;
703 704 705
		*delim = 0;
		np->remote_mac[0] = simple_strtol(cur, NULL, 16);
		cur = delim + 1;
L
Linus Torvalds 已提交
706 707
		if ((delim = strchr(cur, ':')) == NULL)
			goto parse_failed;
708 709 710
		*delim = 0;
		np->remote_mac[1] = simple_strtol(cur, NULL, 16);
		cur = delim + 1;
L
Linus Torvalds 已提交
711 712
		if ((delim = strchr(cur, ':')) == NULL)
			goto parse_failed;
713 714 715
		*delim = 0;
		np->remote_mac[2] = simple_strtol(cur, NULL, 16);
		cur = delim + 1;
L
Linus Torvalds 已提交
716 717
		if ((delim = strchr(cur, ':')) == NULL)
			goto parse_failed;
718 719 720
		*delim = 0;
		np->remote_mac[3] = simple_strtol(cur, NULL, 16);
		cur = delim + 1;
L
Linus Torvalds 已提交
721 722
		if ((delim = strchr(cur, ':')) == NULL)
			goto parse_failed;
723 724 725 726
		*delim = 0;
		np->remote_mac[4] = simple_strtol(cur, NULL, 16);
		cur = delim + 1;
		np->remote_mac[5] = simple_strtol(cur, NULL, 16);
L
Linus Torvalds 已提交
727 728
	}

729
	netpoll_print_options(np);
L
Linus Torvalds 已提交
730 731 732 733

	return 0;

 parse_failed:
734
	printk(KERN_INFO "%s: couldn't parse config at '%s'!\n",
L
Linus Torvalds 已提交
735 736 737
	       np->name, cur);
	return -1;
}
E
Eric Dumazet 已提交
738
EXPORT_SYMBOL(netpoll_parse_options);
L
Linus Torvalds 已提交
739

740
int __netpoll_setup(struct netpoll *np)
L
Linus Torvalds 已提交
741
{
742
	struct net_device *ndev = np->dev;
743
	struct netpoll_info *npinfo;
H
Herbert Xu 已提交
744
	const struct net_device_ops *ops;
745
	unsigned long flags;
S
Stephen Hemminger 已提交
746
	int err;
L
Linus Torvalds 已提交
747

748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810
	if ((ndev->priv_flags & IFF_DISABLE_NETPOLL) ||
	    !ndev->netdev_ops->ndo_poll_controller) {
		printk(KERN_ERR "%s: %s doesn't support polling, aborting.\n",
		       np->name, np->dev_name);
		err = -ENOTSUPP;
		goto out;
	}

	if (!ndev->npinfo) {
		npinfo = kmalloc(sizeof(*npinfo), GFP_KERNEL);
		if (!npinfo) {
			err = -ENOMEM;
			goto out;
		}

		npinfo->rx_flags = 0;
		INIT_LIST_HEAD(&npinfo->rx_np);

		spin_lock_init(&npinfo->rx_lock);
		skb_queue_head_init(&npinfo->arp_tx);
		skb_queue_head_init(&npinfo->txq);
		INIT_DELAYED_WORK(&npinfo->tx_work, queue_process);

		atomic_set(&npinfo->refcnt, 1);

		ops = np->dev->netdev_ops;
		if (ops->ndo_netpoll_setup) {
			err = ops->ndo_netpoll_setup(ndev, npinfo);
			if (err)
				goto free_npinfo;
		}
	} else {
		npinfo = ndev->npinfo;
		atomic_inc(&npinfo->refcnt);
	}

	npinfo->netpoll = np;

	if (np->rx_hook) {
		spin_lock_irqsave(&npinfo->rx_lock, flags);
		npinfo->rx_flags |= NETPOLL_RX_ENABLED;
		list_add_tail(&np->rx, &npinfo->rx_np);
		spin_unlock_irqrestore(&npinfo->rx_lock, flags);
	}

	/* last thing to do is link it to the net device structure */
	rcu_assign_pointer(ndev->npinfo, npinfo);

	return 0;

free_npinfo:
	kfree(npinfo);
out:
	return err;
}
EXPORT_SYMBOL_GPL(__netpoll_setup);

int netpoll_setup(struct netpoll *np)
{
	struct net_device *ndev = NULL;
	struct in_device *in_dev;
	int err;

L
Linus Torvalds 已提交
811
	if (np->dev_name)
812
		ndev = dev_get_by_name(&init_net, np->dev_name);
L
Linus Torvalds 已提交
813 814 815
	if (!ndev) {
		printk(KERN_ERR "%s: %s doesn't exist, aborting.\n",
		       np->name, np->dev_name);
S
Stephen Hemminger 已提交
816
		return -ENODEV;
L
Linus Torvalds 已提交
817 818 819 820 821 822 823 824
	}

	if (!netif_running(ndev)) {
		unsigned long atmost, atleast;

		printk(KERN_INFO "%s: device %s not up yet, forcing it\n",
		       np->name, np->dev_name);

825
		rtnl_lock();
S
Stephen Hemminger 已提交
826 827 828 829
		err = dev_open(ndev);
		rtnl_unlock();

		if (err) {
L
Linus Torvalds 已提交
830
			printk(KERN_ERR "%s: failed to open %s\n",
S
Stephen Hemminger 已提交
831
			       np->name, ndev->name);
832
			goto put;
L
Linus Torvalds 已提交
833 834 835
		}

		atleast = jiffies + HZ/10;
836
		atmost = jiffies + carrier_timeout * HZ;
L
Linus Torvalds 已提交
837 838 839 840 841 842 843
		while (!netif_carrier_ok(ndev)) {
			if (time_after(jiffies, atmost)) {
				printk(KERN_NOTICE
				       "%s: timeout waiting for carrier\n",
				       np->name);
				break;
			}
844
			msleep(1);
L
Linus Torvalds 已提交
845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861
		}

		/* If carrier appears to come up instantly, we don't
		 * trust it and pause so that we don't pump all our
		 * queued console messages into the bitbucket.
		 */

		if (time_before(jiffies, atleast)) {
			printk(KERN_NOTICE "%s: carrier detect appears"
			       " untrustworthy, waiting 4 seconds\n",
			       np->name);
			msleep(4000);
		}
	}

	if (!np->local_ip) {
		rcu_read_lock();
862
		in_dev = __in_dev_get_rcu(ndev);
L
Linus Torvalds 已提交
863 864 865 866 867

		if (!in_dev || !in_dev->ifa_list) {
			rcu_read_unlock();
			printk(KERN_ERR "%s: no IP address for %s, aborting\n",
			       np->name, np->dev_name);
S
Stephen Hemminger 已提交
868
			err = -EDESTADDRREQ;
869
			goto put;
L
Linus Torvalds 已提交
870 871
		}

872
		np->local_ip = in_dev->ifa_list->ifa_local;
L
Linus Torvalds 已提交
873
		rcu_read_unlock();
874
		printk(KERN_INFO "%s: local IP %pI4\n", np->name, &np->local_ip);
L
Linus Torvalds 已提交
875 876
	}

877 878 879 880 881 882
	np->dev = ndev;

	/* fill up the skb queue */
	refill_skbs();

	rtnl_lock();
883
	err = __netpoll_setup(np);
884
	rtnl_unlock();
885

886 887 888
	if (err)
		goto put;

L
Linus Torvalds 已提交
889 890
	return 0;

891
put:
L
Linus Torvalds 已提交
892
	dev_put(ndev);
S
Stephen Hemminger 已提交
893
	return err;
L
Linus Torvalds 已提交
894
}
E
Eric Dumazet 已提交
895
EXPORT_SYMBOL(netpoll_setup);
L
Linus Torvalds 已提交
896

897 898
static int __init netpoll_init(void)
{
899 900 901 902 903
	skb_queue_head_init(&skb_pool);
	return 0;
}
core_initcall(netpoll_init);

904
void __netpoll_cleanup(struct netpoll *np)
L
Linus Torvalds 已提交
905
{
906 907 908
	struct netpoll_info *npinfo;
	unsigned long flags;

909 910
	npinfo = np->dev->npinfo;
	if (!npinfo)
911
		return;
S
Stephen Hemminger 已提交
912

913 914 915 916 917 918 919
	if (!list_empty(&npinfo->rx_np)) {
		spin_lock_irqsave(&npinfo->rx_lock, flags);
		list_del(&np->rx);
		if (list_empty(&npinfo->rx_np))
			npinfo->rx_flags &= ~NETPOLL_RX_ENABLED;
		spin_unlock_irqrestore(&npinfo->rx_lock, flags);
	}
H
Herbert Xu 已提交
920

921 922
	if (atomic_dec_and_test(&npinfo->refcnt)) {
		const struct net_device_ops *ops;
H
Herbert Xu 已提交
923

924 925 926
		ops = np->dev->netdev_ops;
		if (ops->ndo_netpoll_cleanup)
			ops->ndo_netpoll_cleanup(np->dev);
H
Herbert Xu 已提交
927

928
		rcu_assign_pointer(np->dev->npinfo, NULL);
H
Herbert Xu 已提交
929

930 931
		/* avoid racing with NAPI reading npinfo */
		synchronize_rcu_bh();
S
Stephen Hemminger 已提交
932

933 934
		skb_queue_purge(&npinfo->arp_tx);
		skb_queue_purge(&npinfo->txq);
935
		cancel_delayed_work_sync(&npinfo->tx_work);
S
Stephen Hemminger 已提交
936

937 938 939
		/* clean after last, unfinished work */
		__skb_queue_purge(&npinfo->txq);
		kfree(npinfo);
940
	}
941 942
}
EXPORT_SYMBOL_GPL(__netpoll_cleanup);
943

944 945 946 947
void netpoll_cleanup(struct netpoll *np)
{
	if (!np->dev)
		return;
948

949 950 951 952 953
	rtnl_lock();
	__netpoll_cleanup(np);
	rtnl_unlock();

	dev_put(np->dev);
L
Linus Torvalds 已提交
954 955
	np->dev = NULL;
}
E
Eric Dumazet 已提交
956
EXPORT_SYMBOL(netpoll_cleanup);
L
Linus Torvalds 已提交
957 958 959 960 961

int netpoll_trap(void)
{
	return atomic_read(&trapped);
}
E
Eric Dumazet 已提交
962
EXPORT_SYMBOL(netpoll_trap);
L
Linus Torvalds 已提交
963 964 965 966 967 968 969 970 971

void netpoll_set_trap(int trap)
{
	if (trap)
		atomic_inc(&trapped);
	else
		atomic_dec(&trapped);
}
EXPORT_SYMBOL(netpoll_set_trap);