xsk.c 17.4 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
// SPDX-License-Identifier: GPL-2.0
/* XDP sockets
 *
 * AF_XDP sockets allows a channel between XDP programs and userspace
 * applications.
 * Copyright(c) 2018 Intel Corporation.
 *
 * Author(s): Björn Töpel <bjorn.topel@intel.com>
 *	      Magnus Karlsson <magnus.karlsson@intel.com>
 */

#define pr_fmt(fmt) "AF_XDP: %s: " fmt, __func__

#include <linux/if_xdp.h>
#include <linux/init.h>
#include <linux/sched/mm.h>
#include <linux/sched/signal.h>
#include <linux/sched/task.h>
#include <linux/socket.h>
#include <linux/file.h>
#include <linux/uaccess.h>
#include <linux/net.h>
#include <linux/netdevice.h>
24
#include <linux/rculist.h>
25
#include <net/xdp_sock.h>
26
#include <net/xdp.h>
27

28
#include "xsk_queue.h"
29
#include "xdp_umem.h"
30
#include "xsk.h"
31

M
Magnus Karlsson 已提交
32 33
#define TX_BATCH_SIZE 16

34 35
bool xsk_is_setup_for_bpf_map(struct xdp_sock *xs)
{
36 37
	return READ_ONCE(xs->rx) &&  READ_ONCE(xs->umem) &&
		READ_ONCE(xs->umem->fq);
38 39
}

40 41 42 43 44 45
bool xsk_umem_has_addrs(struct xdp_umem *umem, u32 cnt)
{
	return xskq_has_addrs(umem->fq, cnt);
}
EXPORT_SYMBOL(xsk_umem_has_addrs);

46 47 48 49 50 51 52 53 54 55 56 57 58
u64 *xsk_umem_peek_addr(struct xdp_umem *umem, u64 *addr)
{
	return xskq_peek_addr(umem->fq, addr);
}
EXPORT_SYMBOL(xsk_umem_peek_addr);

void xsk_umem_discard_addr(struct xdp_umem *umem)
{
	xskq_discard_addr(umem->fq);
}
EXPORT_SYMBOL(xsk_umem_discard_addr);

static int __xsk_rcv(struct xdp_sock *xs, struct xdp_buff *xdp, u32 len)
59
{
60 61
	void *to_buf, *from_buf;
	u32 metalen;
62
	u64 addr;
63
	int err;
64

65
	if (!xskq_peek_addr(xs->umem->fq, &addr) ||
66
	    len > xs->umem->chunk_size_nohr - XDP_PACKET_HEADROOM) {
67
		xs->rx_dropped++;
68
		return -ENOSPC;
69
	}
70

71 72
	addr += xs->umem->headroom;

73 74 75 76 77 78 79 80 81 82 83
	if (unlikely(xdp_data_meta_unsupported(xdp))) {
		from_buf = xdp->data;
		metalen = 0;
	} else {
		from_buf = xdp->data_meta;
		metalen = xdp->data - xdp->data_meta;
	}

	to_buf = xdp_umem_get_data(xs->umem, addr);
	memcpy(to_buf, from_buf, len + metalen);
	addr += metalen;
84
	err = xskq_produce_batch_desc(xs->rx, addr, len);
85
	if (!err) {
86
		xskq_discard_addr(xs->umem->fq);
87 88 89
		xdp_return_buff(xdp);
		return 0;
	}
90

91
	xs->rx_dropped++;
92 93 94
	return err;
}

95
static int __xsk_rcv_zc(struct xdp_sock *xs, struct xdp_buff *xdp, u32 len)
96
{
97
	int err = xskq_produce_batch_desc(xs->rx, (u64)xdp->handle, len);
98

99
	if (err)
100
		xs->rx_dropped++;
101 102 103 104

	return err;
}

105 106 107 108 109 110 111 112 113 114 115 116 117
int xsk_rcv(struct xdp_sock *xs, struct xdp_buff *xdp)
{
	u32 len;

	if (xs->dev != xdp->rxq->dev || xs->queue_id != xdp->rxq->queue_index)
		return -EINVAL;

	len = xdp->data_end - xdp->data;

	return (xdp->rxq->mem.type == MEM_TYPE_ZERO_COPY) ?
		__xsk_rcv_zc(xs, xdp, len) : __xsk_rcv(xs, xdp, len);
}

118 119 120 121 122 123 124 125
void xsk_flush(struct xdp_sock *xs)
{
	xskq_produce_flush_desc(xs->rx);
	xs->sk.sk_data_ready(&xs->sk);
}

int xsk_generic_rcv(struct xdp_sock *xs, struct xdp_buff *xdp)
{
126
	u32 metalen = xdp->data - xdp->data_meta;
127 128 129
	u32 len = xdp->data_end - xdp->data;
	void *buffer;
	u64 addr;
130 131
	int err;

132 133 134
	if (xs->dev != xdp->rxq->dev || xs->queue_id != xdp->rxq->queue_index)
		return -EINVAL;

135
	if (!xskq_peek_addr(xs->umem->fq, &addr) ||
136
	    len > xs->umem->chunk_size_nohr - XDP_PACKET_HEADROOM) {
137 138 139 140 141 142 143
		xs->rx_dropped++;
		return -ENOSPC;
	}

	addr += xs->umem->headroom;

	buffer = xdp_umem_get_data(xs->umem, addr);
144 145
	memcpy(buffer, xdp->data_meta, len + metalen);
	addr += metalen;
146 147 148
	err = xskq_produce_batch_desc(xs->rx, addr, len);
	if (!err) {
		xskq_discard_addr(xs->umem->fq);
149
		xsk_flush(xs);
150 151
		return 0;
	}
152

153
	xs->rx_dropped++;
154 155 156
	return err;
}

157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209
void xsk_umem_complete_tx(struct xdp_umem *umem, u32 nb_entries)
{
	xskq_produce_flush_addr_n(umem->cq, nb_entries);
}
EXPORT_SYMBOL(xsk_umem_complete_tx);

void xsk_umem_consume_tx_done(struct xdp_umem *umem)
{
	struct xdp_sock *xs;

	rcu_read_lock();
	list_for_each_entry_rcu(xs, &umem->xsk_list, list) {
		xs->sk.sk_write_space(&xs->sk);
	}
	rcu_read_unlock();
}
EXPORT_SYMBOL(xsk_umem_consume_tx_done);

bool xsk_umem_consume_tx(struct xdp_umem *umem, dma_addr_t *dma, u32 *len)
{
	struct xdp_desc desc;
	struct xdp_sock *xs;

	rcu_read_lock();
	list_for_each_entry_rcu(xs, &umem->xsk_list, list) {
		if (!xskq_peek_desc(xs->tx, &desc))
			continue;

		if (xskq_produce_addr_lazy(umem->cq, desc.addr))
			goto out;

		*dma = xdp_umem_get_dma(umem, desc.addr);
		*len = desc.len;

		xskq_discard_desc(xs->tx);
		rcu_read_unlock();
		return true;
	}

out:
	rcu_read_unlock();
	return false;
}
EXPORT_SYMBOL(xsk_umem_consume_tx);

static int xsk_zc_xmit(struct sock *sk)
{
	struct xdp_sock *xs = xdp_sk(sk);
	struct net_device *dev = xs->dev;

	return dev->netdev_ops->ndo_xsk_async_xmit(dev, xs->queue_id);
}

M
Magnus Karlsson 已提交
210 211
static void xsk_destruct_skb(struct sk_buff *skb)
{
212
	u64 addr = (u64)(long)skb_shinfo(skb)->destructor_arg;
M
Magnus Karlsson 已提交
213
	struct xdp_sock *xs = xdp_sk(skb->sk);
214
	unsigned long flags;
M
Magnus Karlsson 已提交
215

216
	spin_lock_irqsave(&xs->tx_completion_lock, flags);
217
	WARN_ON_ONCE(xskq_produce_addr(xs->umem->cq, addr));
218
	spin_unlock_irqrestore(&xs->tx_completion_lock, flags);
M
Magnus Karlsson 已提交
219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236

	sock_wfree(skb);
}

static int xsk_generic_xmit(struct sock *sk, struct msghdr *m,
			    size_t total_len)
{
	u32 max_batch = TX_BATCH_SIZE;
	struct xdp_sock *xs = xdp_sk(sk);
	bool sent_frame = false;
	struct xdp_desc desc;
	struct sk_buff *skb;
	int err = 0;

	mutex_lock(&xs->mutex);

	while (xskq_peek_desc(xs->tx, &desc)) {
		char *buffer;
237 238
		u64 addr;
		u32 len;
M
Magnus Karlsson 已提交
239 240 241 242 243 244

		if (max_batch-- == 0) {
			err = -EAGAIN;
			goto out;
		}

245
		if (xskq_reserve_addr(xs->umem->cq))
M
Magnus Karlsson 已提交
246 247
			goto out;

248
		if (xs->queue_id >= xs->dev->real_num_tx_queues)
249 250
			goto out;

251
		len = desc.len;
252
		skb = sock_alloc_send_skb(sk, len, 1, &err);
M
Magnus Karlsson 已提交
253 254 255 256 257 258
		if (unlikely(!skb)) {
			err = -EAGAIN;
			goto out;
		}

		skb_put(skb, len);
259 260
		addr = desc.addr;
		buffer = xdp_umem_get_data(xs->umem, addr);
M
Magnus Karlsson 已提交
261 262 263 264 265 266 267 268 269
		err = skb_store_bits(skb, 0, buffer, len);
		if (unlikely(err)) {
			kfree_skb(skb);
			goto out;
		}

		skb->dev = xs->dev;
		skb->priority = sk->sk_priority;
		skb->mark = sk->sk_mark;
270
		skb_shinfo(skb)->destructor_arg = (void *)(long)addr;
M
Magnus Karlsson 已提交
271 272 273
		skb->destructor = xsk_destruct_skb;

		err = dev_direct_xmit(skb, xs->queue_id);
274
		xskq_discard_desc(xs->tx);
M
Magnus Karlsson 已提交
275 276
		/* Ignore NET_XMIT_CN as packet might have been sent */
		if (err == NET_XMIT_DROP || err == NETDEV_TX_BUSY) {
277 278
			/* SKB completed but not sent */
			err = -EBUSY;
M
Magnus Karlsson 已提交
279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294
			goto out;
		}

		sent_frame = true;
	}

out:
	if (sent_frame)
		sk->sk_write_space(sk);

	mutex_unlock(&xs->mutex);
	return err;
}

static int xsk_sendmsg(struct socket *sock, struct msghdr *m, size_t total_len)
{
295
	bool need_wait = !(m->msg_flags & MSG_DONTWAIT);
M
Magnus Karlsson 已提交
296 297 298 299 300 301 302
	struct sock *sk = sock->sk;
	struct xdp_sock *xs = xdp_sk(sk);

	if (unlikely(!xs->dev))
		return -ENXIO;
	if (unlikely(!(xs->dev->flags & IFF_UP)))
		return -ENETDOWN;
303 304
	if (unlikely(!xs->tx))
		return -ENOBUFS;
305 306
	if (need_wait)
		return -EOPNOTSUPP;
M
Magnus Karlsson 已提交
307

308
	return (xs->zc) ? xsk_zc_xmit(sk) : xsk_generic_xmit(sk, m, total_len);
M
Magnus Karlsson 已提交
309 310
}

311 312
static unsigned int xsk_poll(struct file *file, struct socket *sock,
			     struct poll_table_struct *wait)
313
{
314
	unsigned int mask = datagram_poll(file, sock, wait);
315 316 317 318 319
	struct sock *sk = sock->sk;
	struct xdp_sock *xs = xdp_sk(sk);

	if (xs->rx && !xskq_empty_desc(xs->rx))
		mask |= POLLIN | POLLRDNORM;
M
Magnus Karlsson 已提交
320 321
	if (xs->tx && !xskq_full_desc(xs->tx))
		mask |= POLLOUT | POLLWRNORM;
322 323 324 325

	return mask;
}

326 327
static int xsk_init_queue(u32 entries, struct xsk_queue **queue,
			  bool umem_queue)
328 329 330 331 332 333
{
	struct xsk_queue *q;

	if (entries == 0 || *queue || !is_power_of_2(entries))
		return -EINVAL;

334
	q = xskq_create(entries, umem_queue);
335 336 337
	if (!q)
		return -ENOMEM;

338 339
	/* Make sure queue is ready before it can be seen by others */
	smp_wmb();
340 341 342 343
	*queue = q;
	return 0;
}

344 345 346
static int xsk_release(struct socket *sock)
{
	struct sock *sk = sock->sk;
347
	struct xdp_sock *xs = xdp_sk(sk);
348 349 350 351 352 353 354
	struct net *net;

	if (!sk)
		return 0;

	net = sock_net(sk);

355 356 357 358
	mutex_lock(&net->xdp.lock);
	sk_del_node_init_rcu(sk);
	mutex_unlock(&net->xdp.lock);

359 360 361 362
	local_bh_disable();
	sock_prot_inuse_add(net, sk->sk_prot, -1);
	local_bh_enable();

363
	if (xs->dev) {
364 365
		struct net_device *dev = xs->dev;

B
Björn Töpel 已提交
366
		/* Wait for driver to stop using the xdp socket. */
367
		xdp_del_sk_umem(xs->umem, xs);
368
		xs->dev = NULL;
369 370
		synchronize_net();
		dev_put(dev);
371 372
	}

373 374 375
	xskq_destroy(xs->rx);
	xskq_destroy(xs->tx);

376 377 378 379 380 381 382 383 384
	sock_orphan(sk);
	sock->sk = NULL;

	sk_refcnt_debug_release(sk);
	sock_put(sk);

	return 0;
}

385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406
static struct socket *xsk_lookup_xsk_from_fd(int fd)
{
	struct socket *sock;
	int err;

	sock = sockfd_lookup(fd, &err);
	if (!sock)
		return ERR_PTR(-ENOTSOCK);

	if (sock->sk->sk_family != PF_XDP) {
		sockfd_put(sock);
		return ERR_PTR(-ENOPROTOOPT);
	}

	return sock;
}

static int xsk_bind(struct socket *sock, struct sockaddr *addr, int addr_len)
{
	struct sockaddr_xdp *sxdp = (struct sockaddr_xdp *)addr;
	struct sock *sk = sock->sk;
	struct xdp_sock *xs = xdp_sk(sk);
B
Björn Töpel 已提交
407
	struct net_device *dev;
408
	u32 flags, qid;
409 410 411 412 413 414 415
	int err = 0;

	if (addr_len < sizeof(struct sockaddr_xdp))
		return -EINVAL;
	if (sxdp->sxdp_family != AF_XDP)
		return -EINVAL;

416 417 418 419
	flags = sxdp->sxdp_flags;
	if (flags & ~(XDP_SHARED_UMEM | XDP_COPY | XDP_ZEROCOPY))
		return -EINVAL;

420
	mutex_lock(&xs->mutex);
B
Björn Töpel 已提交
421 422 423 424 425
	if (xs->dev) {
		err = -EBUSY;
		goto out_release;
	}

426 427 428 429 430 431
	dev = dev_get_by_index(sock_net(sk), sxdp->sxdp_ifindex);
	if (!dev) {
		err = -ENODEV;
		goto out_release;
	}

432
	if (!xs->rx && !xs->tx) {
433 434 435 436
		err = -EINVAL;
		goto out_unlock;
	}

437 438 439
	qid = sxdp->sxdp_queue_id;

	if (flags & XDP_SHARED_UMEM) {
440 441 442
		struct xdp_sock *umem_xs;
		struct socket *sock;

443 444 445 446 447 448
		if ((flags & XDP_COPY) || (flags & XDP_ZEROCOPY)) {
			/* Cannot specify flags for shared sockets. */
			err = -EINVAL;
			goto out_unlock;
		}

449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466
		if (xs->umem) {
			/* We have already our own. */
			err = -EINVAL;
			goto out_unlock;
		}

		sock = xsk_lookup_xsk_from_fd(sxdp->sxdp_shared_umem_fd);
		if (IS_ERR(sock)) {
			err = PTR_ERR(sock);
			goto out_unlock;
		}

		umem_xs = xdp_sk(sock->sk);
		if (!umem_xs->umem) {
			/* No umem to inherit. */
			err = -EBADF;
			sockfd_put(sock);
			goto out_unlock;
467
		} else if (umem_xs->dev != dev || umem_xs->queue_id != qid) {
468 469 470 471 472 473 474 475 476 477 478
			err = -EINVAL;
			sockfd_put(sock);
			goto out_unlock;
		}

		xdp_get_umem(umem_xs->umem);
		xs->umem = umem_xs->umem;
		sockfd_put(sock);
	} else if (!xs->umem || !xdp_umem_validate_queues(xs->umem)) {
		err = -EINVAL;
		goto out_unlock;
479 480
	} else {
		/* This xsk has its own umem. */
481 482 483 484
		xskq_set_umem(xs->umem->fq, xs->umem->size,
			      xs->umem->chunk_mask);
		xskq_set_umem(xs->umem->cq, xs->umem->size,
			      xs->umem->chunk_mask);
485 486 487 488

		err = xdp_umem_assign_dev(xs->umem, dev, qid, flags);
		if (err)
			goto out_unlock;
489 490 491
	}

	xs->dev = dev;
492 493
	xs->zc = xs->umem->zc;
	xs->queue_id = qid;
494 495
	xskq_set_umem(xs->rx, xs->umem->size, xs->umem->chunk_mask);
	xskq_set_umem(xs->tx, xs->umem->size, xs->umem->chunk_mask);
496
	xdp_add_sk_umem(xs->umem, xs);
497 498 499 500 501 502 503 504 505

out_unlock:
	if (err)
		dev_put(dev);
out_release:
	mutex_unlock(&xs->mutex);
	return err;
}

506 507 508 509 510 511 512 513 514 515 516
static int xsk_setsockopt(struct socket *sock, int level, int optname,
			  char __user *optval, unsigned int optlen)
{
	struct sock *sk = sock->sk;
	struct xdp_sock *xs = xdp_sk(sk);
	int err;

	if (level != SOL_XDP)
		return -ENOPROTOOPT;

	switch (optname) {
517
	case XDP_RX_RING:
518
	case XDP_TX_RING:
519 520 521 522 523 524 525 526 527 528
	{
		struct xsk_queue **q;
		int entries;

		if (optlen < sizeof(entries))
			return -EINVAL;
		if (copy_from_user(&entries, optval, sizeof(entries)))
			return -EFAULT;

		mutex_lock(&xs->mutex);
529
		q = (optname == XDP_TX_RING) ? &xs->tx : &xs->rx;
530 531 532 533
		err = xsk_init_queue(entries, q, false);
		mutex_unlock(&xs->mutex);
		return err;
	}
534 535 536 537 538 539 540 541 542
	case XDP_UMEM_REG:
	{
		struct xdp_umem_reg mr;
		struct xdp_umem *umem;

		if (copy_from_user(&mr, optval, sizeof(mr)))
			return -EFAULT;

		mutex_lock(&xs->mutex);
B
Björn Töpel 已提交
543 544 545 546
		if (xs->umem) {
			mutex_unlock(&xs->mutex);
			return -EBUSY;
		}
547

B
Björn Töpel 已提交
548 549
		umem = xdp_umem_create(&mr);
		if (IS_ERR(umem)) {
550
			mutex_unlock(&xs->mutex);
B
Björn Töpel 已提交
551
			return PTR_ERR(umem);
552 553 554 555 556 557 558 559
		}

		/* Make sure umem is ready before it can be seen by others */
		smp_wmb();
		xs->umem = umem;
		mutex_unlock(&xs->mutex);
		return 0;
	}
560
	case XDP_UMEM_FILL_RING:
561
	case XDP_UMEM_COMPLETION_RING:
562 563 564 565 566 567 568 569
	{
		struct xsk_queue **q;
		int entries;

		if (copy_from_user(&entries, optval, sizeof(entries)))
			return -EFAULT;

		mutex_lock(&xs->mutex);
B
Björn Töpel 已提交
570 571 572 573 574
		if (!xs->umem) {
			mutex_unlock(&xs->mutex);
			return -EINVAL;
		}

575 576
		q = (optname == XDP_UMEM_FILL_RING) ? &xs->umem->fq :
			&xs->umem->cq;
577
		err = xsk_init_queue(entries, q, true);
578 579 580
		mutex_unlock(&xs->mutex);
		return err;
	}
581 582 583 584 585 586 587
	default:
		break;
	}

	return -ENOPROTOOPT;
}

M
Magnus Karlsson 已提交
588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623
static int xsk_getsockopt(struct socket *sock, int level, int optname,
			  char __user *optval, int __user *optlen)
{
	struct sock *sk = sock->sk;
	struct xdp_sock *xs = xdp_sk(sk);
	int len;

	if (level != SOL_XDP)
		return -ENOPROTOOPT;

	if (get_user(len, optlen))
		return -EFAULT;
	if (len < 0)
		return -EINVAL;

	switch (optname) {
	case XDP_STATISTICS:
	{
		struct xdp_statistics stats;

		if (len < sizeof(stats))
			return -EINVAL;

		mutex_lock(&xs->mutex);
		stats.rx_dropped = xs->rx_dropped;
		stats.rx_invalid_descs = xskq_nb_invalid_descs(xs->rx);
		stats.tx_invalid_descs = xskq_nb_invalid_descs(xs->tx);
		mutex_unlock(&xs->mutex);

		if (copy_to_user(optval, &stats, sizeof(stats)))
			return -EFAULT;
		if (put_user(sizeof(stats), optlen))
			return -EFAULT;

		return 0;
	}
624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652
	case XDP_MMAP_OFFSETS:
	{
		struct xdp_mmap_offsets off;

		if (len < sizeof(off))
			return -EINVAL;

		off.rx.producer = offsetof(struct xdp_rxtx_ring, ptrs.producer);
		off.rx.consumer = offsetof(struct xdp_rxtx_ring, ptrs.consumer);
		off.rx.desc	= offsetof(struct xdp_rxtx_ring, desc);
		off.tx.producer = offsetof(struct xdp_rxtx_ring, ptrs.producer);
		off.tx.consumer = offsetof(struct xdp_rxtx_ring, ptrs.consumer);
		off.tx.desc	= offsetof(struct xdp_rxtx_ring, desc);

		off.fr.producer = offsetof(struct xdp_umem_ring, ptrs.producer);
		off.fr.consumer = offsetof(struct xdp_umem_ring, ptrs.consumer);
		off.fr.desc	= offsetof(struct xdp_umem_ring, desc);
		off.cr.producer = offsetof(struct xdp_umem_ring, ptrs.producer);
		off.cr.consumer = offsetof(struct xdp_umem_ring, ptrs.consumer);
		off.cr.desc	= offsetof(struct xdp_umem_ring, desc);

		len = sizeof(off);
		if (copy_to_user(optval, &off, len))
			return -EFAULT;
		if (put_user(len, optlen))
			return -EFAULT;

		return 0;
	}
M
Magnus Karlsson 已提交
653 654 655 656 657 658 659
	default:
		break;
	}

	return -EOPNOTSUPP;
}

660 661 662
static int xsk_mmap(struct file *file, struct socket *sock,
		    struct vm_area_struct *vma)
{
663
	loff_t offset = (loff_t)vma->vm_pgoff << PAGE_SHIFT;
664 665 666
	unsigned long size = vma->vm_end - vma->vm_start;
	struct xdp_sock *xs = xdp_sk(sock->sk);
	struct xsk_queue *q = NULL;
667
	struct xdp_umem *umem;
668 669 670
	unsigned long pfn;
	struct page *qpg;

671
	if (offset == XDP_PGOFF_RX_RING) {
672
		q = READ_ONCE(xs->rx);
673
	} else if (offset == XDP_PGOFF_TX_RING) {
674
		q = READ_ONCE(xs->tx);
675
	} else {
676 677
		umem = READ_ONCE(xs->umem);
		if (!umem)
678
			return -EINVAL;
679

680 681
		/* Matches the smp_wmb() in XDP_UMEM_REG */
		smp_rmb();
682
		if (offset == XDP_UMEM_PGOFF_FILL_RING)
683
			q = READ_ONCE(umem->fq);
684
		else if (offset == XDP_UMEM_PGOFF_COMPLETION_RING)
685
			q = READ_ONCE(umem->cq);
686
	}
687 688 689 690

	if (!q)
		return -EINVAL;

691 692
	/* Matches the smp_wmb() in xsk_init_queue */
	smp_rmb();
693 694 695 696 697 698 699 700 701
	qpg = virt_to_head_page(q->ring);
	if (size > (PAGE_SIZE << compound_order(qpg)))
		return -EINVAL;

	pfn = virt_to_phys(q->ring) >> PAGE_SHIFT;
	return remap_pfn_range(vma, vma->vm_start, pfn,
			       size, vma->vm_page_prot);
}

702 703 704 705 706 707 708
static struct proto xsk_proto = {
	.name =		"XDP",
	.owner =	THIS_MODULE,
	.obj_size =	sizeof(struct xdp_sock),
};

static const struct proto_ops xsk_proto_ops = {
B
Björn Töpel 已提交
709 710 711 712 713 714 715 716
	.family		= PF_XDP,
	.owner		= THIS_MODULE,
	.release	= xsk_release,
	.bind		= xsk_bind,
	.connect	= sock_no_connect,
	.socketpair	= sock_no_socketpair,
	.accept		= sock_no_accept,
	.getname	= sock_no_getname,
717
	.poll		= xsk_poll,
B
Björn Töpel 已提交
718 719 720 721 722 723 724 725 726
	.ioctl		= sock_no_ioctl,
	.listen		= sock_no_listen,
	.shutdown	= sock_no_shutdown,
	.setsockopt	= xsk_setsockopt,
	.getsockopt	= xsk_getsockopt,
	.sendmsg	= xsk_sendmsg,
	.recvmsg	= sock_no_recvmsg,
	.mmap		= xsk_mmap,
	.sendpage	= sock_no_sendpage,
727 728
};

729 730 731 732 733 734 735 736 737 738 739 740
static void xsk_destruct(struct sock *sk)
{
	struct xdp_sock *xs = xdp_sk(sk);

	if (!sock_flag(sk, SOCK_DEAD))
		return;

	xdp_put_umem(xs->umem);

	sk_refcnt_debug_dec(sk);
}

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
static int xsk_create(struct net *net, struct socket *sock, int protocol,
		      int kern)
{
	struct sock *sk;
	struct xdp_sock *xs;

	if (!ns_capable(net->user_ns, CAP_NET_RAW))
		return -EPERM;
	if (sock->type != SOCK_RAW)
		return -ESOCKTNOSUPPORT;

	if (protocol)
		return -EPROTONOSUPPORT;

	sock->state = SS_UNCONNECTED;

	sk = sk_alloc(net, PF_XDP, GFP_KERNEL, &xsk_proto, kern);
	if (!sk)
		return -ENOBUFS;

	sock->ops = &xsk_proto_ops;

	sock_init_data(sock, sk);

	sk->sk_family = PF_XDP;

767 768 769
	sk->sk_destruct = xsk_destruct;
	sk_refcnt_debug_inc(sk);

770 771
	sock_set_flag(sk, SOCK_RCU_FREE);

772 773
	xs = xdp_sk(sk);
	mutex_init(&xs->mutex);
774
	spin_lock_init(&xs->tx_completion_lock);
775

776 777 778 779
	mutex_lock(&net->xdp.lock);
	sk_add_node_rcu(sk, &net->xdp.list);
	mutex_unlock(&net->xdp.lock);

780 781 782 783 784 785 786 787 788 789 790 791 792
	local_bh_disable();
	sock_prot_inuse_add(net, &xsk_proto, 1);
	local_bh_enable();

	return 0;
}

static const struct net_proto_family xsk_family_ops = {
	.family = PF_XDP,
	.create = xsk_create,
	.owner	= THIS_MODULE,
};

793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809
static int __net_init xsk_net_init(struct net *net)
{
	mutex_init(&net->xdp.lock);
	INIT_HLIST_HEAD(&net->xdp.list);
	return 0;
}

static void __net_exit xsk_net_exit(struct net *net)
{
	WARN_ON_ONCE(!hlist_empty(&net->xdp.list));
}

static struct pernet_operations xsk_net_ops = {
	.init = xsk_net_init,
	.exit = xsk_net_exit,
};

810 811 812 813 814 815 816 817 818 819 820 821
static int __init xsk_init(void)
{
	int err;

	err = proto_register(&xsk_proto, 0 /* no slab */);
	if (err)
		goto out;

	err = sock_register(&xsk_family_ops);
	if (err)
		goto out_proto;

822 823 824
	err = register_pernet_subsys(&xsk_net_ops);
	if (err)
		goto out_sk;
825 826
	return 0;

827 828
out_sk:
	sock_unregister(PF_XDP);
829 830 831 832 833 834 835
out_proto:
	proto_unregister(&xsk_proto);
out:
	return err;
}

fs_initcall(xsk_init);