netpoll.c 21.5 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
	service_arp_queue(dev->npinfo);
197

198
	zap_completion_queue();
L
Linus Torvalds 已提交
199
}
E
Eric Dumazet 已提交
200
EXPORT_SYMBOL(netpoll_poll_dev);
L
Linus Torvalds 已提交
201

202 203 204 205
void netpoll_poll(struct netpoll *np)
{
	netpoll_poll_dev(np->dev);
}
E
Eric Dumazet 已提交
206
EXPORT_SYMBOL(netpoll_poll);
207

L
Linus Torvalds 已提交
208 209 210 211 212
static void refill_skbs(void)
{
	struct sk_buff *skb;
	unsigned long flags;

213 214
	spin_lock_irqsave(&skb_pool.lock, flags);
	while (skb_pool.qlen < MAX_SKBS) {
L
Linus Torvalds 已提交
215 216 217 218
		skb = alloc_skb(MAX_SKB_SIZE, GFP_ATOMIC);
		if (!skb)
			break;

219
		__skb_queue_tail(&skb_pool, skb);
L
Linus Torvalds 已提交
220
	}
221
	spin_unlock_irqrestore(&skb_pool.lock, flags);
L
Linus Torvalds 已提交
222 223
}

224 225 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
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);
}

252
static struct sk_buff *find_skb(struct netpoll *np, int len, int reserve)
L
Linus Torvalds 已提交
253
{
254 255
	int count = 0;
	struct sk_buff *skb;
L
Linus Torvalds 已提交
256

257
	zap_completion_queue();
258
	refill_skbs();
L
Linus Torvalds 已提交
259 260 261
repeat:

	skb = alloc_skb(len, GFP_ATOMIC);
262 263
	if (!skb)
		skb = skb_dequeue(&skb_pool);
L
Linus Torvalds 已提交
264 265

	if (!skb) {
266 267 268
		if (++count < 10) {
			netpoll_poll(np);
			goto repeat;
L
Linus Torvalds 已提交
269
		}
270
		return NULL;
L
Linus Torvalds 已提交
271 272 273 274 275 276 277
	}

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

278 279 280 281 282 283 284 285 286 287 288
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;
}

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

298 299 300 301
	if (!npinfo || !netif_running(dev) || !netif_device_present(dev)) {
		__kfree_skb(skb);
		return;
	}
S
Stephen Hemminger 已提交
302 303

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

308 309
		txq = netdev_get_tx_queue(dev, skb_get_queue_mapping(skb));

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

				if (status == NETDEV_TX_OK)
					break;

			}
326 327 328 329 330

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

			udelay(USEC_PER_POLL);
M
Matt Mackall 已提交
331
		}
332 333 334 335 336

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

337
		local_irq_restore(flags);
L
Linus Torvalds 已提交
338 339
	}

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

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;

363
	skb_copy_to_linear_data(skb, msg, len);
L
Linus Torvalds 已提交
364 365
	skb->len += len;

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

380 381
	skb_push(skb, sizeof(*iph));
	skb_reset_network_header(skb);
382
	iph = ip_hdr(skb);
L
Linus Torvalds 已提交
383 384 385 386 387 388 389 390 391 392

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

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

	skb->dev = np->dev;

	netpoll_send_skb(np, skb);
}
E
Eric Dumazet 已提交
407
EXPORT_SYMBOL(netpoll_send_udp);
L
Linus Torvalds 已提交
408 409 410

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

434 435
	/* No netpoll struct is using this dev */
	if (!hits)
436
		return;
L
Linus Torvalds 已提交
437 438 439 440 441

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

442
	if (!pskb_may_pull(skb, arp_hdr_len(skb->dev)))
L
Linus Torvalds 已提交
443 444
		return;

445
	skb_reset_network_header(skb);
446
	skb_reset_transport_header(skb);
447
	arp = arp_hdr(skb);
L
Linus Torvalds 已提交
448 449 450 451 452 453 454

	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;

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

	/* Should we ignore arp? */
467
	if (ipv4_is_loopback(tip) || ipv4_is_multicast(tip))
L
Linus Torvalds 已提交
468 469
		return;

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

472 473 474 475
	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 已提交
476

477 478 479 480
		send_skb = find_skb(np, size + LL_ALLOCATED_SPACE(np->dev),
				    LL_RESERVED_SPACE(np->dev));
		if (!send_skb)
			continue;
L
Linus Torvalds 已提交
481

482 483 484 485
		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 已提交
486

487 488 489 490 491 492 493
		/* 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 已提交
494

495 496 497 498 499 500 501
		/*
		 * 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 已提交
502

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

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

536
	if (list_empty(&npinfo->rx_np))
L
Linus Torvalds 已提交
537
		goto out;
538

L
Linus Torvalds 已提交
539 540 541
	if (skb->dev->type != ARPHRD_ETHER)
		goto out;

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

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

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

	kfree_skb(skb);
	return 1;

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

	return 0;
}

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

L
Linus Torvalds 已提交
636 637 638 639
int netpoll_parse_options(struct netpoll *np, char *opt)
{
	char *cur=opt, *delim;

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

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

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

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

	/* dst ip */
	if ((delim = strchr(cur, '/')) == NULL)
		goto parse_failed;
684
	*delim = 0;
685
	np->remote_ip = in_aton(cur);
686
	cur = delim + 1;
L
Linus Torvalds 已提交
687

688
	if (*cur != 0) {
L
Linus Torvalds 已提交
689 690 691
		/* MAC address */
		if ((delim = strchr(cur, ':')) == NULL)
			goto parse_failed;
692 693 694
		*delim = 0;
		np->remote_mac[0] = simple_strtol(cur, NULL, 16);
		cur = delim + 1;
L
Linus Torvalds 已提交
695 696
		if ((delim = strchr(cur, ':')) == NULL)
			goto parse_failed;
697 698 699
		*delim = 0;
		np->remote_mac[1] = simple_strtol(cur, NULL, 16);
		cur = delim + 1;
L
Linus Torvalds 已提交
700 701
		if ((delim = strchr(cur, ':')) == NULL)
			goto parse_failed;
702 703 704
		*delim = 0;
		np->remote_mac[2] = simple_strtol(cur, NULL, 16);
		cur = delim + 1;
L
Linus Torvalds 已提交
705 706
		if ((delim = strchr(cur, ':')) == NULL)
			goto parse_failed;
707 708 709
		*delim = 0;
		np->remote_mac[3] = simple_strtol(cur, NULL, 16);
		cur = delim + 1;
L
Linus Torvalds 已提交
710 711
		if ((delim = strchr(cur, ':')) == NULL)
			goto parse_failed;
712 713 714 715
		*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 已提交
716 717
	}

718
	netpoll_print_options(np);
L
Linus Torvalds 已提交
719 720 721 722

	return 0;

 parse_failed:
723
	printk(KERN_INFO "%s: couldn't parse config at '%s'!\n",
L
Linus Torvalds 已提交
724 725 726
	       np->name, cur);
	return -1;
}
E
Eric Dumazet 已提交
727
EXPORT_SYMBOL(netpoll_parse_options);
L
Linus Torvalds 已提交
728

729
int __netpoll_setup(struct netpoll *np)
L
Linus Torvalds 已提交
730
{
731
	struct net_device *ndev = np->dev;
732
	struct netpoll_info *npinfo;
H
Herbert Xu 已提交
733
	const struct net_device_ops *ops;
734
	unsigned long flags;
S
Stephen Hemminger 已提交
735
	int err;
L
Linus Torvalds 已提交
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 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799
	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 已提交
800
	if (np->dev_name)
801
		ndev = dev_get_by_name(&init_net, np->dev_name);
L
Linus Torvalds 已提交
802 803 804
	if (!ndev) {
		printk(KERN_ERR "%s: %s doesn't exist, aborting.\n",
		       np->name, np->dev_name);
S
Stephen Hemminger 已提交
805
		return -ENODEV;
L
Linus Torvalds 已提交
806 807 808 809 810 811 812 813
	}

	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);

814
		rtnl_lock();
S
Stephen Hemminger 已提交
815 816 817 818
		err = dev_open(ndev);
		rtnl_unlock();

		if (err) {
L
Linus Torvalds 已提交
819
			printk(KERN_ERR "%s: failed to open %s\n",
S
Stephen Hemminger 已提交
820
			       np->name, ndev->name);
821
			goto put;
L
Linus Torvalds 已提交
822 823 824
		}

		atleast = jiffies + HZ/10;
825
		atmost = jiffies + carrier_timeout * HZ;
L
Linus Torvalds 已提交
826 827 828 829 830 831 832
		while (!netif_carrier_ok(ndev)) {
			if (time_after(jiffies, atmost)) {
				printk(KERN_NOTICE
				       "%s: timeout waiting for carrier\n",
				       np->name);
				break;
			}
833
			msleep(1);
L
Linus Torvalds 已提交
834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850
		}

		/* 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();
851
		in_dev = __in_dev_get_rcu(ndev);
L
Linus Torvalds 已提交
852 853 854 855 856

		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 已提交
857
			err = -EDESTADDRREQ;
858
			goto put;
L
Linus Torvalds 已提交
859 860
		}

861
		np->local_ip = in_dev->ifa_list->ifa_local;
L
Linus Torvalds 已提交
862
		rcu_read_unlock();
863
		printk(KERN_INFO "%s: local IP %pI4\n", np->name, &np->local_ip);
L
Linus Torvalds 已提交
864 865
	}

866 867 868 869 870 871
	np->dev = ndev;

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

	rtnl_lock();
872
	err = __netpoll_setup(np);
873
	rtnl_unlock();
874

875 876 877
	if (err)
		goto put;

L
Linus Torvalds 已提交
878 879
	return 0;

880
put:
L
Linus Torvalds 已提交
881
	dev_put(ndev);
S
Stephen Hemminger 已提交
882
	return err;
L
Linus Torvalds 已提交
883
}
E
Eric Dumazet 已提交
884
EXPORT_SYMBOL(netpoll_setup);
L
Linus Torvalds 已提交
885

886 887
static int __init netpoll_init(void)
{
888 889 890 891 892
	skb_queue_head_init(&skb_pool);
	return 0;
}
core_initcall(netpoll_init);

893
void __netpoll_cleanup(struct netpoll *np)
L
Linus Torvalds 已提交
894
{
895 896 897
	struct netpoll_info *npinfo;
	unsigned long flags;

898 899
	npinfo = np->dev->npinfo;
	if (!npinfo)
900
		return;
S
Stephen Hemminger 已提交
901

902 903 904 905 906 907 908
	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 已提交
909

910 911
	if (atomic_dec_and_test(&npinfo->refcnt)) {
		const struct net_device_ops *ops;
H
Herbert Xu 已提交
912

913 914 915
		ops = np->dev->netdev_ops;
		if (ops->ndo_netpoll_cleanup)
			ops->ndo_netpoll_cleanup(np->dev);
H
Herbert Xu 已提交
916

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

919 920
		/* avoid racing with NAPI reading npinfo */
		synchronize_rcu_bh();
S
Stephen Hemminger 已提交
921

922 923
		skb_queue_purge(&npinfo->arp_tx);
		skb_queue_purge(&npinfo->txq);
924
		cancel_delayed_work_sync(&npinfo->tx_work);
S
Stephen Hemminger 已提交
925

926 927 928
		/* clean after last, unfinished work */
		__skb_queue_purge(&npinfo->txq);
		kfree(npinfo);
929
	}
930 931
}
EXPORT_SYMBOL_GPL(__netpoll_cleanup);
932

933 934 935 936
void netpoll_cleanup(struct netpoll *np)
{
	if (!np->dev)
		return;
937

938 939 940 941 942
	rtnl_lock();
	__netpoll_cleanup(np);
	rtnl_unlock();

	dev_put(np->dev);
L
Linus Torvalds 已提交
943 944
	np->dev = NULL;
}
E
Eric Dumazet 已提交
945
EXPORT_SYMBOL(netpoll_cleanup);
L
Linus Torvalds 已提交
946 947 948 949 950

int netpoll_trap(void)
{
	return atomic_read(&trapped);
}
E
Eric Dumazet 已提交
951
EXPORT_SYMBOL(netpoll_trap);
L
Linus Torvalds 已提交
952 953 954 955 956 957 958 959 960

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