netpoll.c 21.7 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 39

/*
 * 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
#define MAX_QUEUE_DEPTH (MAX_SKBS / 2)

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

static atomic_t trapped;

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

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

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

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

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

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

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

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

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

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

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

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

102 103
	psum = csum_tcpudp_nofold(saddr, daddr, ulen, IPPROTO_UDP, 0);

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

108
	skb->csum = psum;
L
Linus Torvalds 已提交
109

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

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

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

	work = napi->poll(napi, budget);
146
	trace_napi_poll(napi);
147

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

	return budget - work;
}

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

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

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

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

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

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

186 187 188 189 190
	if (!dev || !netif_running(dev))
		return;

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

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

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

198
	service_arp_queue(dev->npinfo);
199

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

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

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

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

221
		__skb_queue_tail(&skb_pool, skb);
L
Linus Torvalds 已提交
222
	}
223
	spin_unlock_irqrestore(&skb_pool.lock, flags);
L
Linus Torvalds 已提交
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 252 253
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);
}

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

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

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

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

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

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

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

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

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

310 311
		txq = netdev_get_tx_queue(dev, skb_get_queue_mapping(skb));

312
		local_irq_save(flags);
313 314 315
		/* try until next clock tick */
		for (tries = jiffies_to_usecs(1)/USEC_PER_POLL;
		     tries > 0; --tries) {
316
			if (__netif_tx_trylock(txq)) {
E
Eric Dumazet 已提交
317
				if (!netif_tx_queue_stopped(txq)) {
318
					dev->priv_flags |= IFF_IN_NETPOLL;
319
					status = ops->ndo_start_xmit(skb, dev);
320
					dev->priv_flags &= ~IFF_IN_NETPOLL;
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 332 333 334

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

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

722
	netpoll_print_options(np);
L
Linus Torvalds 已提交
723 724 725 726

	return 0;

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

733
int __netpoll_setup(struct netpoll *np)
L
Linus Torvalds 已提交
734
{
735
	struct net_device *ndev = np->dev;
736
	struct netpoll_info *npinfo;
H
Herbert Xu 已提交
737
	const struct net_device_ops *ops;
738
	unsigned long flags;
S
Stephen Hemminger 已提交
739
	int err;
L
Linus Torvalds 已提交
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 800 801 802 803
	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 已提交
804
	if (np->dev_name)
805
		ndev = dev_get_by_name(&init_net, np->dev_name);
L
Linus Torvalds 已提交
806 807 808
	if (!ndev) {
		printk(KERN_ERR "%s: %s doesn't exist, aborting.\n",
		       np->name, np->dev_name);
S
Stephen Hemminger 已提交
809
		return -ENODEV;
L
Linus Torvalds 已提交
810 811 812 813 814 815 816 817
	}

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

818
		rtnl_lock();
S
Stephen Hemminger 已提交
819 820 821 822
		err = dev_open(ndev);
		rtnl_unlock();

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

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

		/* 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();
855
		in_dev = __in_dev_get_rcu(ndev);
L
Linus Torvalds 已提交
856 857 858 859 860

		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 已提交
861
			err = -EDESTADDRREQ;
862
			goto put;
L
Linus Torvalds 已提交
863 864
		}

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

870 871 872 873 874 875
	np->dev = ndev;

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

	rtnl_lock();
876
	err = __netpoll_setup(np);
877
	rtnl_unlock();
878

879 880 881
	if (err)
		goto put;

L
Linus Torvalds 已提交
882 883
	return 0;

884
put:
L
Linus Torvalds 已提交
885
	dev_put(ndev);
S
Stephen Hemminger 已提交
886
	return err;
L
Linus Torvalds 已提交
887
}
E
Eric Dumazet 已提交
888
EXPORT_SYMBOL(netpoll_setup);
L
Linus Torvalds 已提交
889

890 891
static int __init netpoll_init(void)
{
892 893 894 895 896
	skb_queue_head_init(&skb_pool);
	return 0;
}
core_initcall(netpoll_init);

897
void __netpoll_cleanup(struct netpoll *np)
L
Linus Torvalds 已提交
898
{
899 900 901
	struct netpoll_info *npinfo;
	unsigned long flags;

902 903
	npinfo = np->dev->npinfo;
	if (!npinfo)
904
		return;
S
Stephen Hemminger 已提交
905

906 907 908 909 910 911 912
	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 已提交
913

914 915
	if (atomic_dec_and_test(&npinfo->refcnt)) {
		const struct net_device_ops *ops;
H
Herbert Xu 已提交
916

917 918 919
		ops = np->dev->netdev_ops;
		if (ops->ndo_netpoll_cleanup)
			ops->ndo_netpoll_cleanup(np->dev);
H
Herbert Xu 已提交
920

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

923 924
		/* avoid racing with NAPI reading npinfo */
		synchronize_rcu_bh();
S
Stephen Hemminger 已提交
925

926 927 928
		skb_queue_purge(&npinfo->arp_tx);
		skb_queue_purge(&npinfo->txq);
		cancel_rearming_delayed_work(&npinfo->tx_work);
S
Stephen Hemminger 已提交
929

930 931 932
		/* clean after last, unfinished work */
		__skb_queue_purge(&npinfo->txq);
		kfree(npinfo);
933
	}
934 935
}
EXPORT_SYMBOL_GPL(__netpoll_cleanup);
936

937 938 939 940
void netpoll_cleanup(struct netpoll *np)
{
	if (!np->dev)
		return;
941

942 943 944 945 946
	rtnl_lock();
	__netpoll_cleanup(np);
	rtnl_unlock();

	dev_put(np->dev);
L
Linus Torvalds 已提交
947 948
	np->dev = NULL;
}
E
Eric Dumazet 已提交
949
EXPORT_SYMBOL(netpoll_cleanup);
L
Linus Torvalds 已提交
950 951 952 953 954

int netpoll_trap(void)
{
	return atomic_read(&trapped);
}
E
Eric Dumazet 已提交
955
EXPORT_SYMBOL(netpoll_trap);
L
Linus Torvalds 已提交
956 957 958 959 960 961 962 963 964

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