netpoll.c 21.1 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
static 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 211 212 213 214 215 216
}

static void refill_skbs(void)
{
	struct sk_buff *skb;
	unsigned long flags;

217 218
	spin_lock_irqsave(&skb_pool.lock, flags);
	while (skb_pool.qlen < MAX_SKBS) {
L
Linus Torvalds 已提交
219 220 221 222
		skb = alloc_skb(MAX_SKB_SIZE, GFP_ATOMIC);
		if (!skb)
			break;

223
		__skb_queue_tail(&skb_pool, skb);
L
Linus Torvalds 已提交
224
	}
225
	spin_unlock_irqrestore(&skb_pool.lock, flags);
L
Linus Torvalds 已提交
226 227
}

228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255
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);
}

256
static struct sk_buff *find_skb(struct netpoll *np, int len, int reserve)
L
Linus Torvalds 已提交
257
{
258 259
	int count = 0;
	struct sk_buff *skb;
L
Linus Torvalds 已提交
260

261
	zap_completion_queue();
262
	refill_skbs();
L
Linus Torvalds 已提交
263 264 265
repeat:

	skb = alloc_skb(len, GFP_ATOMIC);
266 267
	if (!skb)
		skb = skb_dequeue(&skb_pool);
L
Linus Torvalds 已提交
268 269

	if (!skb) {
270
		if (++count < 10) {
271
			netpoll_poll_dev(np->dev);
272
			goto repeat;
L
Linus Torvalds 已提交
273
		}
274
		return NULL;
L
Linus Torvalds 已提交
275 276 277 278 279 280 281
	}

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

282 283 284 285 286 287 288 289 290 291 292
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;
}

293 294
void netpoll_send_skb_on_dev(struct netpoll *np, struct sk_buff *skb,
			     struct net_device *dev)
L
Linus Torvalds 已提交
295
{
S
Stephen Hemminger 已提交
296 297
	int status = NETDEV_TX_BUSY;
	unsigned long tries;
298
	const struct net_device_ops *ops = dev->netdev_ops;
H
Herbert Xu 已提交
299
	/* It is up to the caller to keep npinfo alive. */
300
	struct netpoll_info *npinfo = np->dev->npinfo;
S
Stephen Hemminger 已提交
301

302 303 304 305
	if (!npinfo || !netif_running(dev) || !netif_device_present(dev)) {
		__kfree_skb(skb);
		return;
	}
S
Stephen Hemminger 已提交
306 307

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

312 313
		txq = netdev_get_tx_queue(dev, skb_get_queue_mapping(skb));

314
		local_irq_save(flags);
315 316 317
		/* try until next clock tick */
		for (tries = jiffies_to_usecs(1)/USEC_PER_POLL;
		     tries > 0; --tries) {
318
			if (__netif_tx_trylock(txq)) {
E
Eric Dumazet 已提交
319
				if (!netif_tx_queue_stopped(txq)) {
320
					status = ops->ndo_start_xmit(skb, dev);
E
Eric Dumazet 已提交
321 322 323
					if (status == NETDEV_TX_OK)
						txq_trans_update(txq);
				}
324
				__netif_tx_unlock(txq);
325 326 327 328 329

				if (status == NETDEV_TX_OK)
					break;

			}
330 331

			/* tickle device maybe there is some cleanup */
332
			netpoll_poll_dev(np->dev);
333 334

			udelay(USEC_PER_POLL);
M
Matt Mackall 已提交
335
		}
336 337 338 339 340

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

341
		local_irq_restore(flags);
L
Linus Torvalds 已提交
342 343
	}

S
Stephen Hemminger 已提交
344
	if (status != NETDEV_TX_OK) {
S
Stephen Hemminger 已提交
345
		skb_queue_tail(&npinfo->txq, skb);
346
		schedule_delayed_work(&npinfo->tx_work,0);
L
Linus Torvalds 已提交
347 348
	}
}
349
EXPORT_SYMBOL(netpoll_send_skb_on_dev);
L
Linus Torvalds 已提交
350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366

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;

367
	skb_copy_to_linear_data(skb, msg, len);
L
Linus Torvalds 已提交
368 369
	skb->len += len;

370 371 372
	skb_push(skb, sizeof(*udph));
	skb_reset_transport_header(skb);
	udph = udp_hdr(skb);
L
Linus Torvalds 已提交
373 374 375 376
	udph->source = htons(np->local_port);
	udph->dest = htons(np->remote_port);
	udph->len = htons(udp_len);
	udph->check = 0;
377 378
	udph->check = csum_tcpudp_magic(np->local_ip,
					np->remote_ip,
379
					udp_len, IPPROTO_UDP,
380
					csum_partial(udph, udp_len, 0));
381
	if (udph->check == 0)
382
		udph->check = CSUM_MANGLED_0;
L
Linus Torvalds 已提交
383

384 385
	skb_push(skb, sizeof(*iph));
	skb_reset_network_header(skb);
386
	iph = ip_hdr(skb);
L
Linus Torvalds 已提交
387 388 389 390 391 392 393 394 395 396

	/* 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;
397 398
	put_unaligned(np->local_ip, &(iph->saddr));
	put_unaligned(np->remote_ip, &(iph->daddr));
L
Linus Torvalds 已提交
399 400 401
	iph->check    = ip_fast_csum((unsigned char *)iph, iph->ihl);

	eth = (struct ethhdr *) skb_push(skb, ETH_HLEN);
402
	skb_reset_mac_header(skb);
403
	skb->protocol = eth->h_proto = htons(ETH_P_IP);
404 405
	memcpy(eth->h_source, np->dev->dev_addr, ETH_ALEN);
	memcpy(eth->h_dest, np->remote_mac, ETH_ALEN);
L
Linus Torvalds 已提交
406 407 408 409 410

	skb->dev = np->dev;

	netpoll_send_skb(np, skb);
}
E
Eric Dumazet 已提交
411
EXPORT_SYMBOL(netpoll_send_udp);
L
Linus Torvalds 已提交
412 413 414

static void arp_reply(struct sk_buff *skb)
{
415
	struct netpoll_info *npinfo = skb->dev->npinfo;
L
Linus Torvalds 已提交
416 417 418
	struct arphdr *arp;
	unsigned char *arp_ptr;
	int size, type = ARPOP_REPLY, ptype = ETH_P_ARP;
A
Al Viro 已提交
419
	__be32 sip, tip;
420
	unsigned char *sha;
L
Linus Torvalds 已提交
421
	struct sk_buff *send_skb;
422 423 424 425 426 427 428 429 430 431 432 433 434 435 436
	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 已提交
437

438 439
	/* No netpoll struct is using this dev */
	if (!hits)
440
		return;
L
Linus Torvalds 已提交
441 442 443 444 445

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

446
	if (!pskb_may_pull(skb, arp_hdr_len(skb->dev)))
L
Linus Torvalds 已提交
447 448
		return;

449
	skb_reset_network_header(skb);
450
	skb_reset_transport_header(skb);
451
	arp = arp_hdr(skb);
L
Linus Torvalds 已提交
452 453 454 455 456 457 458

	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;

459 460 461 462
	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 已提交
463
	memcpy(&sip, arp_ptr, 4);
464
	arp_ptr += 4;
465 466
	/* If we actually cared about dst hw addr,
	   it would get copied here */
467
	arp_ptr += skb->dev->addr_len;
L
Linus Torvalds 已提交
468 469 470
	memcpy(&tip, arp_ptr, 4);

	/* Should we ignore arp? */
471
	if (ipv4_is_loopback(tip) || ipv4_is_multicast(tip))
L
Linus Torvalds 已提交
472 473
		return;

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

476 477 478 479
	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 已提交
480

481 482 483 484
		send_skb = find_skb(np, size + LL_ALLOCATED_SPACE(np->dev),
				    LL_RESERVED_SPACE(np->dev));
		if (!send_skb)
			continue;
L
Linus Torvalds 已提交
485

486 487 488 489
		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 已提交
490

491 492 493 494 495 496 497
		/* 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 已提交
498

499 500 501 502 503 504 505
		/*
		 * 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 已提交
506

507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528
		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 已提交
529 530 531 532 533
}

int __netpoll_rx(struct sk_buff *skb)
{
	int proto, len, ulen;
534
	int hits = 0;
535
	const struct iphdr *iph;
L
Linus Torvalds 已提交
536
	struct udphdr *uh;
537 538
	struct netpoll_info *npinfo = skb->dev->npinfo;
	struct netpoll *np, *tmp;
539

540
	if (list_empty(&npinfo->rx_np))
L
Linus Torvalds 已提交
541
		goto out;
542

L
Linus Torvalds 已提交
543 544 545
	if (skb->dev->type != ARPHRD_ETHER)
		goto out;

546
	/* check if netpoll clients need ARP */
547
	if (skb->protocol == htons(ETH_P_ARP) &&
L
Linus Torvalds 已提交
548
	    atomic_read(&trapped)) {
549
		skb_queue_tail(&npinfo->arp_tx, skb);
L
Linus Torvalds 已提交
550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574
		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;

575 576 577 578 579 580 581
	/*
	 * 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 已提交
582 583 584 585 586 587 588 589 590
	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;
591
	if (checksum_udp(skb, uh, ulen, iph->saddr, iph->daddr))
L
Linus Torvalds 已提交
592 593
		goto out;

594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609
	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 已提交
610 611 612 613 614 615 616 617 618 619 620 621 622

	kfree_skb(skb);
	return 1;

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

	return 0;
}

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

L
Linus Torvalds 已提交
640 641 642 643
int netpoll_parse_options(struct netpoll *np, char *opt)
{
	char *cur=opt, *delim;

644
	if (*cur != '@') {
L
Linus Torvalds 已提交
645 646
		if ((delim = strchr(cur, '@')) == NULL)
			goto parse_failed;
647 648 649
		*delim = 0;
		np->local_port = simple_strtol(cur, NULL, 10);
		cur = delim;
L
Linus Torvalds 已提交
650 651 652
	}
	cur++;

653
	if (*cur != '/') {
L
Linus Torvalds 已提交
654 655
		if ((delim = strchr(cur, '/')) == NULL)
			goto parse_failed;
656
		*delim = 0;
657
		np->local_ip = in_aton(cur);
658
		cur = delim;
L
Linus Torvalds 已提交
659 660 661
	}
	cur++;

662
	if (*cur != ',') {
L
Linus Torvalds 已提交
663 664 665
		/* parse out dev name */
		if ((delim = strchr(cur, ',')) == NULL)
			goto parse_failed;
666
		*delim = 0;
L
Linus Torvalds 已提交
667
		strlcpy(np->dev_name, cur, sizeof(np->dev_name));
668
		cur = delim;
L
Linus Torvalds 已提交
669 670 671
	}
	cur++;

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

	/* dst ip */
	if ((delim = strchr(cur, '/')) == NULL)
		goto parse_failed;
688
	*delim = 0;
689
	np->remote_ip = in_aton(cur);
690
	cur = delim + 1;
L
Linus Torvalds 已提交
691

692
	if (*cur != 0) {
L
Linus Torvalds 已提交
693
		/* MAC address */
694
		if (!mac_pton(cur, np->remote_mac))
L
Linus Torvalds 已提交
695 696 697
			goto parse_failed;
	}

698
	netpoll_print_options(np);
L
Linus Torvalds 已提交
699 700 701 702

	return 0;

 parse_failed:
703
	printk(KERN_INFO "%s: couldn't parse config at '%s'!\n",
L
Linus Torvalds 已提交
704 705 706
	       np->name, cur);
	return -1;
}
E
Eric Dumazet 已提交
707
EXPORT_SYMBOL(netpoll_parse_options);
L
Linus Torvalds 已提交
708

709
int __netpoll_setup(struct netpoll *np)
L
Linus Torvalds 已提交
710
{
711
	struct net_device *ndev = np->dev;
712
	struct netpoll_info *npinfo;
H
Herbert Xu 已提交
713
	const struct net_device_ops *ops;
714
	unsigned long flags;
S
Stephen Hemminger 已提交
715
	int err;
L
Linus Torvalds 已提交
716

717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 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
	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 已提交
780
	if (np->dev_name)
781
		ndev = dev_get_by_name(&init_net, np->dev_name);
L
Linus Torvalds 已提交
782 783 784
	if (!ndev) {
		printk(KERN_ERR "%s: %s doesn't exist, aborting.\n",
		       np->name, np->dev_name);
S
Stephen Hemminger 已提交
785
		return -ENODEV;
L
Linus Torvalds 已提交
786 787
	}

788 789 790
	if (ndev->master) {
		printk(KERN_ERR "%s: %s is a slave device, aborting.\n",
		       np->name, np->dev_name);
791 792
		err = -EBUSY;
		goto put;
793 794
	}

L
Linus Torvalds 已提交
795 796 797 798 799 800
	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);

801
		rtnl_lock();
S
Stephen Hemminger 已提交
802 803 804 805
		err = dev_open(ndev);
		rtnl_unlock();

		if (err) {
L
Linus Torvalds 已提交
806
			printk(KERN_ERR "%s: failed to open %s\n",
S
Stephen Hemminger 已提交
807
			       np->name, ndev->name);
808
			goto put;
L
Linus Torvalds 已提交
809 810 811
		}

		atleast = jiffies + HZ/10;
812
		atmost = jiffies + carrier_timeout * HZ;
L
Linus Torvalds 已提交
813 814 815 816 817 818 819
		while (!netif_carrier_ok(ndev)) {
			if (time_after(jiffies, atmost)) {
				printk(KERN_NOTICE
				       "%s: timeout waiting for carrier\n",
				       np->name);
				break;
			}
820
			msleep(1);
L
Linus Torvalds 已提交
821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837
		}

		/* 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();
838
		in_dev = __in_dev_get_rcu(ndev);
L
Linus Torvalds 已提交
839 840 841 842 843

		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 已提交
844
			err = -EDESTADDRREQ;
845
			goto put;
L
Linus Torvalds 已提交
846 847
		}

848
		np->local_ip = in_dev->ifa_list->ifa_local;
L
Linus Torvalds 已提交
849
		rcu_read_unlock();
850
		printk(KERN_INFO "%s: local IP %pI4\n", np->name, &np->local_ip);
L
Linus Torvalds 已提交
851 852
	}

853 854 855 856 857 858
	np->dev = ndev;

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

	rtnl_lock();
859
	err = __netpoll_setup(np);
860
	rtnl_unlock();
861

862 863 864
	if (err)
		goto put;

L
Linus Torvalds 已提交
865 866
	return 0;

867
put:
L
Linus Torvalds 已提交
868
	dev_put(ndev);
S
Stephen Hemminger 已提交
869
	return err;
L
Linus Torvalds 已提交
870
}
E
Eric Dumazet 已提交
871
EXPORT_SYMBOL(netpoll_setup);
L
Linus Torvalds 已提交
872

873 874
static int __init netpoll_init(void)
{
875 876 877 878 879
	skb_queue_head_init(&skb_pool);
	return 0;
}
core_initcall(netpoll_init);

880
void __netpoll_cleanup(struct netpoll *np)
L
Linus Torvalds 已提交
881
{
882 883 884
	struct netpoll_info *npinfo;
	unsigned long flags;

885 886
	npinfo = np->dev->npinfo;
	if (!npinfo)
887
		return;
S
Stephen Hemminger 已提交
888

889 890 891 892 893 894 895
	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 已提交
896

897 898
	if (atomic_dec_and_test(&npinfo->refcnt)) {
		const struct net_device_ops *ops;
H
Herbert Xu 已提交
899

900 901 902
		ops = np->dev->netdev_ops;
		if (ops->ndo_netpoll_cleanup)
			ops->ndo_netpoll_cleanup(np->dev);
H
Herbert Xu 已提交
903

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

906 907
		/* avoid racing with NAPI reading npinfo */
		synchronize_rcu_bh();
S
Stephen Hemminger 已提交
908

909 910
		skb_queue_purge(&npinfo->arp_tx);
		skb_queue_purge(&npinfo->txq);
911
		cancel_delayed_work_sync(&npinfo->tx_work);
S
Stephen Hemminger 已提交
912

913 914 915
		/* clean after last, unfinished work */
		__skb_queue_purge(&npinfo->txq);
		kfree(npinfo);
916
	}
917 918
}
EXPORT_SYMBOL_GPL(__netpoll_cleanup);
919

920 921 922 923
void netpoll_cleanup(struct netpoll *np)
{
	if (!np->dev)
		return;
924

925 926 927 928 929
	rtnl_lock();
	__netpoll_cleanup(np);
	rtnl_unlock();

	dev_put(np->dev);
L
Linus Torvalds 已提交
930 931
	np->dev = NULL;
}
E
Eric Dumazet 已提交
932
EXPORT_SYMBOL(netpoll_cleanup);
L
Linus Torvalds 已提交
933 934 935 936 937

int netpoll_trap(void)
{
	return atomic_read(&trapped);
}
E
Eric Dumazet 已提交
938
EXPORT_SYMBOL(netpoll_trap);
L
Linus Torvalds 已提交
939 940 941 942 943 944 945 946 947

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