1 // SPDX-License-Identifier: GPL-2.0-only 2 /* net/core/xdp.c 3 * 4 * Copyright (c) 2017 Jesper Dangaard Brouer, Red Hat Inc. 5 */ 6 #include <linux/bpf.h> 7 #include <linux/btf.h> 8 #include <linux/btf_ids.h> 9 #include <linux/filter.h> 10 #include <linux/types.h> 11 #include <linux/mm.h> 12 #include <linux/netdevice.h> 13 #include <linux/slab.h> 14 #include <linux/idr.h> 15 #include <linux/rhashtable.h> 16 #include <linux/bug.h> 17 #include <net/page_pool/helpers.h> 18 19 #include <net/hotdata.h> 20 #include <net/xdp.h> 21 #include <net/xdp_priv.h> /* struct xdp_mem_allocator */ 22 #include <trace/events/xdp.h> 23 #include <net/xdp_sock_drv.h> 24 25 #define REG_STATE_NEW 0x0 26 #define REG_STATE_REGISTERED 0x1 27 #define REG_STATE_UNREGISTERED 0x2 28 #define REG_STATE_UNUSED 0x3 29 30 static DEFINE_IDA(mem_id_pool); 31 static DEFINE_MUTEX(mem_id_lock); 32 #define MEM_ID_MAX 0xFFFE 33 #define MEM_ID_MIN 1 34 static int mem_id_next = MEM_ID_MIN; 35 36 static bool mem_id_init; /* false */ 37 static struct rhashtable *mem_id_ht; 38 39 static u32 xdp_mem_id_hashfn(const void *data, u32 len, u32 seed) 40 { 41 const u32 *k = data; 42 const u32 key = *k; 43 44 BUILD_BUG_ON(sizeof_field(struct xdp_mem_allocator, mem.id) 45 != sizeof(u32)); 46 47 /* Use cyclic increasing ID as direct hash key */ 48 return key; 49 } 50 51 static int xdp_mem_id_cmp(struct rhashtable_compare_arg *arg, 52 const void *ptr) 53 { 54 const struct xdp_mem_allocator *xa = ptr; 55 u32 mem_id = *(u32 *)arg->key; 56 57 return xa->mem.id != mem_id; 58 } 59 60 static const struct rhashtable_params mem_id_rht_params = { 61 .nelem_hint = 64, 62 .head_offset = offsetof(struct xdp_mem_allocator, node), 63 .key_offset = offsetof(struct xdp_mem_allocator, mem.id), 64 .key_len = sizeof_field(struct xdp_mem_allocator, mem.id), 65 .max_size = MEM_ID_MAX, 66 .min_size = 8, 67 .automatic_shrinking = true, 68 .hashfn = xdp_mem_id_hashfn, 69 .obj_cmpfn = xdp_mem_id_cmp, 70 }; 71 72 static void __xdp_mem_allocator_rcu_free(struct rcu_head *rcu) 73 { 74 struct xdp_mem_allocator *xa; 75 76 xa = container_of(rcu, struct xdp_mem_allocator, rcu); 77 78 /* Allow this ID to be reused */ 79 ida_free(&mem_id_pool, xa->mem.id); 80 81 kfree(xa); 82 } 83 84 static void mem_xa_remove(struct xdp_mem_allocator *xa) 85 { 86 trace_mem_disconnect(xa); 87 88 if (!rhashtable_remove_fast(mem_id_ht, &xa->node, mem_id_rht_params)) 89 call_rcu(&xa->rcu, __xdp_mem_allocator_rcu_free); 90 } 91 92 static void mem_allocator_disconnect(void *allocator) 93 { 94 struct xdp_mem_allocator *xa; 95 struct rhashtable_iter iter; 96 97 mutex_lock(&mem_id_lock); 98 99 rhashtable_walk_enter(mem_id_ht, &iter); 100 do { 101 rhashtable_walk_start(&iter); 102 103 while ((xa = rhashtable_walk_next(&iter)) && !IS_ERR(xa)) { 104 if (xa->allocator == allocator) 105 mem_xa_remove(xa); 106 } 107 108 rhashtable_walk_stop(&iter); 109 110 } while (xa == ERR_PTR(-EAGAIN)); 111 rhashtable_walk_exit(&iter); 112 113 mutex_unlock(&mem_id_lock); 114 } 115 116 void xdp_unreg_mem_model(struct xdp_mem_info *mem) 117 { 118 struct xdp_mem_allocator *xa; 119 int type = mem->type; 120 int id = mem->id; 121 122 /* Reset mem info to defaults */ 123 mem->id = 0; 124 mem->type = 0; 125 126 if (id == 0) 127 return; 128 129 if (type == MEM_TYPE_PAGE_POOL) { 130 xa = rhashtable_lookup_fast(mem_id_ht, &id, mem_id_rht_params); 131 page_pool_destroy(xa->page_pool); 132 } 133 } 134 EXPORT_SYMBOL_GPL(xdp_unreg_mem_model); 135 136 void xdp_rxq_info_unreg_mem_model(struct xdp_rxq_info *xdp_rxq) 137 { 138 if (xdp_rxq->reg_state != REG_STATE_REGISTERED) { 139 WARN(1, "Missing register, driver bug"); 140 return; 141 } 142 143 xdp_unreg_mem_model(&xdp_rxq->mem); 144 } 145 EXPORT_SYMBOL_GPL(xdp_rxq_info_unreg_mem_model); 146 147 void xdp_rxq_info_unreg(struct xdp_rxq_info *xdp_rxq) 148 { 149 /* Simplify driver cleanup code paths, allow unreg "unused" */ 150 if (xdp_rxq->reg_state == REG_STATE_UNUSED) 151 return; 152 153 xdp_rxq_info_unreg_mem_model(xdp_rxq); 154 155 xdp_rxq->reg_state = REG_STATE_UNREGISTERED; 156 xdp_rxq->dev = NULL; 157 } 158 EXPORT_SYMBOL_GPL(xdp_rxq_info_unreg); 159 160 static void xdp_rxq_info_init(struct xdp_rxq_info *xdp_rxq) 161 { 162 memset(xdp_rxq, 0, sizeof(*xdp_rxq)); 163 } 164 165 /* Returns 0 on success, negative on failure */ 166 int __xdp_rxq_info_reg(struct xdp_rxq_info *xdp_rxq, 167 struct net_device *dev, u32 queue_index, 168 unsigned int napi_id, u32 frag_size) 169 { 170 if (!dev) { 171 WARN(1, "Missing net_device from driver"); 172 return -ENODEV; 173 } 174 175 if (xdp_rxq->reg_state == REG_STATE_UNUSED) { 176 WARN(1, "Driver promised not to register this"); 177 return -EINVAL; 178 } 179 180 if (xdp_rxq->reg_state == REG_STATE_REGISTERED) { 181 WARN(1, "Missing unregister, handled but fix driver"); 182 xdp_rxq_info_unreg(xdp_rxq); 183 } 184 185 /* State either UNREGISTERED or NEW */ 186 xdp_rxq_info_init(xdp_rxq); 187 xdp_rxq->dev = dev; 188 xdp_rxq->queue_index = queue_index; 189 xdp_rxq->napi_id = napi_id; 190 xdp_rxq->frag_size = frag_size; 191 192 xdp_rxq->reg_state = REG_STATE_REGISTERED; 193 return 0; 194 } 195 EXPORT_SYMBOL_GPL(__xdp_rxq_info_reg); 196 197 void xdp_rxq_info_unused(struct xdp_rxq_info *xdp_rxq) 198 { 199 xdp_rxq->reg_state = REG_STATE_UNUSED; 200 } 201 EXPORT_SYMBOL_GPL(xdp_rxq_info_unused); 202 203 bool xdp_rxq_info_is_reg(struct xdp_rxq_info *xdp_rxq) 204 { 205 return (xdp_rxq->reg_state == REG_STATE_REGISTERED); 206 } 207 EXPORT_SYMBOL_GPL(xdp_rxq_info_is_reg); 208 209 static int __mem_id_init_hash_table(void) 210 { 211 struct rhashtable *rht; 212 int ret; 213 214 if (unlikely(mem_id_init)) 215 return 0; 216 217 rht = kzalloc(sizeof(*rht), GFP_KERNEL); 218 if (!rht) 219 return -ENOMEM; 220 221 ret = rhashtable_init(rht, &mem_id_rht_params); 222 if (ret < 0) { 223 kfree(rht); 224 return ret; 225 } 226 mem_id_ht = rht; 227 smp_mb(); /* mutex lock should provide enough pairing */ 228 mem_id_init = true; 229 230 return 0; 231 } 232 233 /* Allocate a cyclic ID that maps to allocator pointer. 234 * See: https://www.kernel.org/doc/html/latest/core-api/idr.html 235 * 236 * Caller must lock mem_id_lock. 237 */ 238 static int __mem_id_cyclic_get(gfp_t gfp) 239 { 240 int retries = 1; 241 int id; 242 243 again: 244 id = ida_alloc_range(&mem_id_pool, mem_id_next, MEM_ID_MAX - 1, gfp); 245 if (id < 0) { 246 if (id == -ENOSPC) { 247 /* Cyclic allocator, reset next id */ 248 if (retries--) { 249 mem_id_next = MEM_ID_MIN; 250 goto again; 251 } 252 } 253 return id; /* errno */ 254 } 255 mem_id_next = id + 1; 256 257 return id; 258 } 259 260 static bool __is_supported_mem_type(enum xdp_mem_type type) 261 { 262 if (type == MEM_TYPE_PAGE_POOL) 263 return is_page_pool_compiled_in(); 264 265 if (type >= MEM_TYPE_MAX) 266 return false; 267 268 return true; 269 } 270 271 static struct xdp_mem_allocator *__xdp_reg_mem_model(struct xdp_mem_info *mem, 272 enum xdp_mem_type type, 273 void *allocator) 274 { 275 struct xdp_mem_allocator *xdp_alloc; 276 gfp_t gfp = GFP_KERNEL; 277 int id, errno, ret; 278 void *ptr; 279 280 if (!__is_supported_mem_type(type)) 281 return ERR_PTR(-EOPNOTSUPP); 282 283 mem->type = type; 284 285 if (!allocator) { 286 if (type == MEM_TYPE_PAGE_POOL) 287 return ERR_PTR(-EINVAL); /* Setup time check page_pool req */ 288 return NULL; 289 } 290 291 /* Delay init of rhashtable to save memory if feature isn't used */ 292 if (!mem_id_init) { 293 mutex_lock(&mem_id_lock); 294 ret = __mem_id_init_hash_table(); 295 mutex_unlock(&mem_id_lock); 296 if (ret < 0) 297 return ERR_PTR(ret); 298 } 299 300 xdp_alloc = kzalloc(sizeof(*xdp_alloc), gfp); 301 if (!xdp_alloc) 302 return ERR_PTR(-ENOMEM); 303 304 mutex_lock(&mem_id_lock); 305 id = __mem_id_cyclic_get(gfp); 306 if (id < 0) { 307 errno = id; 308 goto err; 309 } 310 mem->id = id; 311 xdp_alloc->mem = *mem; 312 xdp_alloc->allocator = allocator; 313 314 /* Insert allocator into ID lookup table */ 315 ptr = rhashtable_insert_slow(mem_id_ht, &id, &xdp_alloc->node); 316 if (IS_ERR(ptr)) { 317 ida_free(&mem_id_pool, mem->id); 318 mem->id = 0; 319 errno = PTR_ERR(ptr); 320 goto err; 321 } 322 323 if (type == MEM_TYPE_PAGE_POOL) 324 page_pool_use_xdp_mem(allocator, mem_allocator_disconnect, mem); 325 326 mutex_unlock(&mem_id_lock); 327 328 return xdp_alloc; 329 err: 330 mutex_unlock(&mem_id_lock); 331 kfree(xdp_alloc); 332 return ERR_PTR(errno); 333 } 334 335 int xdp_reg_mem_model(struct xdp_mem_info *mem, 336 enum xdp_mem_type type, void *allocator) 337 { 338 struct xdp_mem_allocator *xdp_alloc; 339 340 xdp_alloc = __xdp_reg_mem_model(mem, type, allocator); 341 if (IS_ERR(xdp_alloc)) 342 return PTR_ERR(xdp_alloc); 343 return 0; 344 } 345 EXPORT_SYMBOL_GPL(xdp_reg_mem_model); 346 347 int xdp_rxq_info_reg_mem_model(struct xdp_rxq_info *xdp_rxq, 348 enum xdp_mem_type type, void *allocator) 349 { 350 struct xdp_mem_allocator *xdp_alloc; 351 352 if (xdp_rxq->reg_state != REG_STATE_REGISTERED) { 353 WARN(1, "Missing register, driver bug"); 354 return -EFAULT; 355 } 356 357 xdp_alloc = __xdp_reg_mem_model(&xdp_rxq->mem, type, allocator); 358 if (IS_ERR(xdp_alloc)) 359 return PTR_ERR(xdp_alloc); 360 361 if (type == MEM_TYPE_XSK_BUFF_POOL && allocator) 362 xsk_pool_set_rxq_info(allocator, xdp_rxq); 363 364 if (trace_mem_connect_enabled() && xdp_alloc) 365 trace_mem_connect(xdp_alloc, xdp_rxq); 366 return 0; 367 } 368 369 EXPORT_SYMBOL_GPL(xdp_rxq_info_reg_mem_model); 370 371 /** 372 * xdp_reg_page_pool - register &page_pool as a memory provider for XDP 373 * @pool: &page_pool to register 374 * 375 * Can be used to register pools manually without connecting to any XDP RxQ 376 * info, so that the XDP layer will be aware of them. Then, they can be 377 * attached to an RxQ info manually via xdp_rxq_info_attach_page_pool(). 378 * 379 * Return: %0 on success, -errno on error. 380 */ 381 int xdp_reg_page_pool(struct page_pool *pool) 382 { 383 struct xdp_mem_info mem; 384 385 return xdp_reg_mem_model(&mem, MEM_TYPE_PAGE_POOL, pool); 386 } 387 EXPORT_SYMBOL_GPL(xdp_reg_page_pool); 388 389 /** 390 * xdp_unreg_page_pool - unregister &page_pool from the memory providers list 391 * @pool: &page_pool to unregister 392 * 393 * A shorthand for manual unregistering page pools. If the pool was previously 394 * attached to an RxQ info, it must be detached first. 395 */ 396 void xdp_unreg_page_pool(const struct page_pool *pool) 397 { 398 struct xdp_mem_info mem = { 399 .type = MEM_TYPE_PAGE_POOL, 400 .id = pool->xdp_mem_id, 401 }; 402 403 xdp_unreg_mem_model(&mem); 404 } 405 EXPORT_SYMBOL_GPL(xdp_unreg_page_pool); 406 407 /** 408 * xdp_rxq_info_attach_page_pool - attach registered pool to RxQ info 409 * @xdp_rxq: XDP RxQ info to attach the pool to 410 * @pool: pool to attach 411 * 412 * If the pool was registered manually, this function must be called instead 413 * of xdp_rxq_info_reg_mem_model() to connect it to the RxQ info. 414 */ 415 void xdp_rxq_info_attach_page_pool(struct xdp_rxq_info *xdp_rxq, 416 const struct page_pool *pool) 417 { 418 struct xdp_mem_info mem = { 419 .type = MEM_TYPE_PAGE_POOL, 420 .id = pool->xdp_mem_id, 421 }; 422 423 xdp_rxq_info_attach_mem_model(xdp_rxq, &mem); 424 } 425 EXPORT_SYMBOL_GPL(xdp_rxq_info_attach_page_pool); 426 427 /* XDP RX runs under NAPI protection, and in different delivery error 428 * scenarios (e.g. queue full), it is possible to return the xdp_frame 429 * while still leveraging this protection. The @napi_direct boolean 430 * is used for those calls sites. Thus, allowing for faster recycling 431 * of xdp_frames/pages in those cases. 432 */ 433 void __xdp_return(netmem_ref netmem, enum xdp_mem_type mem_type, 434 bool napi_direct, struct xdp_buff *xdp) 435 { 436 switch (mem_type) { 437 case MEM_TYPE_PAGE_POOL: 438 netmem = netmem_compound_head(netmem); 439 if (napi_direct && xdp_return_frame_no_direct()) 440 napi_direct = false; 441 /* No need to check ((page->pp_magic & ~0x3UL) == PP_SIGNATURE) 442 * as mem->type knows this a page_pool page 443 */ 444 page_pool_put_full_netmem(netmem_get_pp(netmem), netmem, 445 napi_direct); 446 break; 447 case MEM_TYPE_PAGE_SHARED: 448 page_frag_free(__netmem_address(netmem)); 449 break; 450 case MEM_TYPE_PAGE_ORDER0: 451 put_page(__netmem_to_page(netmem)); 452 break; 453 case MEM_TYPE_XSK_BUFF_POOL: 454 /* NB! Only valid from an xdp_buff! */ 455 xsk_buff_free(xdp); 456 break; 457 default: 458 /* Not possible, checked in xdp_rxq_info_reg_mem_model() */ 459 WARN(1, "Incorrect XDP memory type (%d) usage", mem_type); 460 break; 461 } 462 } 463 464 void xdp_return_frame(struct xdp_frame *xdpf) 465 { 466 struct skb_shared_info *sinfo; 467 468 if (likely(!xdp_frame_has_frags(xdpf))) 469 goto out; 470 471 sinfo = xdp_get_shared_info_from_frame(xdpf); 472 for (u32 i = 0; i < sinfo->nr_frags; i++) 473 __xdp_return(skb_frag_netmem(&sinfo->frags[i]), xdpf->mem_type, 474 false, NULL); 475 476 out: 477 __xdp_return(virt_to_netmem(xdpf->data), xdpf->mem_type, false, NULL); 478 } 479 EXPORT_SYMBOL_GPL(xdp_return_frame); 480 481 void xdp_return_frame_rx_napi(struct xdp_frame *xdpf) 482 { 483 struct skb_shared_info *sinfo; 484 485 if (likely(!xdp_frame_has_frags(xdpf))) 486 goto out; 487 488 sinfo = xdp_get_shared_info_from_frame(xdpf); 489 for (u32 i = 0; i < sinfo->nr_frags; i++) 490 __xdp_return(skb_frag_netmem(&sinfo->frags[i]), xdpf->mem_type, 491 true, NULL); 492 493 out: 494 __xdp_return(virt_to_netmem(xdpf->data), xdpf->mem_type, true, NULL); 495 } 496 EXPORT_SYMBOL_GPL(xdp_return_frame_rx_napi); 497 498 /* XDP bulk APIs introduce a defer/flush mechanism to return 499 * pages belonging to the same xdp_mem_allocator object 500 * (identified via the mem.id field) in bulk to optimize 501 * I-cache and D-cache. 502 * The bulk queue size is set to 16 to be aligned to how 503 * XDP_REDIRECT bulking works. The bulk is flushed when 504 * it is full or when mem.id changes. 505 * xdp_frame_bulk is usually stored/allocated on the function 506 * call-stack to avoid locking penalties. 507 */ 508 509 /* Must be called with rcu_read_lock held */ 510 void xdp_return_frame_bulk(struct xdp_frame *xdpf, 511 struct xdp_frame_bulk *bq) 512 { 513 if (xdpf->mem_type != MEM_TYPE_PAGE_POOL) { 514 xdp_return_frame(xdpf); 515 return; 516 } 517 518 if (bq->count == XDP_BULK_QUEUE_SIZE) 519 xdp_flush_frame_bulk(bq); 520 521 if (unlikely(xdp_frame_has_frags(xdpf))) { 522 struct skb_shared_info *sinfo; 523 int i; 524 525 sinfo = xdp_get_shared_info_from_frame(xdpf); 526 for (i = 0; i < sinfo->nr_frags; i++) { 527 skb_frag_t *frag = &sinfo->frags[i]; 528 529 bq->q[bq->count++] = skb_frag_netmem(frag); 530 if (bq->count == XDP_BULK_QUEUE_SIZE) 531 xdp_flush_frame_bulk(bq); 532 } 533 } 534 bq->q[bq->count++] = virt_to_netmem(xdpf->data); 535 } 536 EXPORT_SYMBOL_GPL(xdp_return_frame_bulk); 537 538 void xdp_return_buff(struct xdp_buff *xdp) 539 { 540 struct skb_shared_info *sinfo; 541 542 if (likely(!xdp_buff_has_frags(xdp))) 543 goto out; 544 545 sinfo = xdp_get_shared_info_from_buff(xdp); 546 for (u32 i = 0; i < sinfo->nr_frags; i++) 547 __xdp_return(skb_frag_netmem(&sinfo->frags[i]), 548 xdp->rxq->mem.type, true, xdp); 549 550 out: 551 __xdp_return(virt_to_netmem(xdp->data), xdp->rxq->mem.type, true, xdp); 552 } 553 EXPORT_SYMBOL_GPL(xdp_return_buff); 554 555 void xdp_attachment_setup(struct xdp_attachment_info *info, 556 struct netdev_bpf *bpf) 557 { 558 if (info->prog) 559 bpf_prog_put(info->prog); 560 info->prog = bpf->prog; 561 info->flags = bpf->flags; 562 } 563 EXPORT_SYMBOL_GPL(xdp_attachment_setup); 564 565 struct xdp_frame *xdp_convert_zc_to_xdp_frame(struct xdp_buff *xdp) 566 { 567 unsigned int metasize, totsize; 568 void *addr, *data_to_copy; 569 struct xdp_frame *xdpf; 570 struct page *page; 571 572 /* Clone into a MEM_TYPE_PAGE_ORDER0 xdp_frame. */ 573 metasize = xdp_data_meta_unsupported(xdp) ? 0 : 574 xdp->data - xdp->data_meta; 575 totsize = xdp->data_end - xdp->data + metasize; 576 577 if (sizeof(*xdpf) + totsize > PAGE_SIZE) 578 return NULL; 579 580 page = dev_alloc_page(); 581 if (!page) 582 return NULL; 583 584 addr = page_to_virt(page); 585 xdpf = addr; 586 memset(xdpf, 0, sizeof(*xdpf)); 587 588 addr += sizeof(*xdpf); 589 data_to_copy = metasize ? xdp->data_meta : xdp->data; 590 memcpy(addr, data_to_copy, totsize); 591 592 xdpf->data = addr + metasize; 593 xdpf->len = totsize - metasize; 594 xdpf->headroom = 0; 595 xdpf->metasize = metasize; 596 xdpf->frame_sz = PAGE_SIZE; 597 xdpf->mem_type = MEM_TYPE_PAGE_ORDER0; 598 599 xsk_buff_free(xdp); 600 return xdpf; 601 } 602 EXPORT_SYMBOL_GPL(xdp_convert_zc_to_xdp_frame); 603 604 /* Used by XDP_WARN macro, to avoid inlining WARN() in fast-path */ 605 void xdp_warn(const char *msg, const char *func, const int line) 606 { 607 WARN(1, "XDP_WARN: %s(line:%d): %s\n", func, line, msg); 608 }; 609 EXPORT_SYMBOL_GPL(xdp_warn); 610 611 int xdp_alloc_skb_bulk(void **skbs, int n_skb, gfp_t gfp) 612 { 613 n_skb = kmem_cache_alloc_bulk(net_hotdata.skbuff_cache, gfp, n_skb, skbs); 614 if (unlikely(!n_skb)) 615 return -ENOMEM; 616 617 return 0; 618 } 619 EXPORT_SYMBOL_GPL(xdp_alloc_skb_bulk); 620 621 struct sk_buff *__xdp_build_skb_from_frame(struct xdp_frame *xdpf, 622 struct sk_buff *skb, 623 struct net_device *dev) 624 { 625 struct skb_shared_info *sinfo = xdp_get_shared_info_from_frame(xdpf); 626 unsigned int headroom, frame_size; 627 void *hard_start; 628 u8 nr_frags; 629 630 /* xdp frags frame */ 631 if (unlikely(xdp_frame_has_frags(xdpf))) 632 nr_frags = sinfo->nr_frags; 633 634 /* Part of headroom was reserved to xdpf */ 635 headroom = sizeof(*xdpf) + xdpf->headroom; 636 637 /* Memory size backing xdp_frame data already have reserved 638 * room for build_skb to place skb_shared_info in tailroom. 639 */ 640 frame_size = xdpf->frame_sz; 641 642 hard_start = xdpf->data - headroom; 643 skb = build_skb_around(skb, hard_start, frame_size); 644 if (unlikely(!skb)) 645 return NULL; 646 647 skb_reserve(skb, headroom); 648 __skb_put(skb, xdpf->len); 649 if (xdpf->metasize) 650 skb_metadata_set(skb, xdpf->metasize); 651 652 if (unlikely(xdp_frame_has_frags(xdpf))) 653 xdp_update_skb_shared_info(skb, nr_frags, 654 sinfo->xdp_frags_size, 655 nr_frags * xdpf->frame_sz, 656 xdp_frame_is_frag_pfmemalloc(xdpf)); 657 658 /* Essential SKB info: protocol and skb->dev */ 659 skb->protocol = eth_type_trans(skb, dev); 660 661 /* Optional SKB info, currently missing: 662 * - HW checksum info (skb->ip_summed) 663 * - HW RX hash (skb_set_hash) 664 * - RX ring dev queue index (skb_record_rx_queue) 665 */ 666 667 if (xdpf->mem_type == MEM_TYPE_PAGE_POOL) 668 skb_mark_for_recycle(skb); 669 670 /* Allow SKB to reuse area used by xdp_frame */ 671 xdp_scrub_frame(xdpf); 672 673 return skb; 674 } 675 EXPORT_SYMBOL_GPL(__xdp_build_skb_from_frame); 676 677 struct sk_buff *xdp_build_skb_from_frame(struct xdp_frame *xdpf, 678 struct net_device *dev) 679 { 680 struct sk_buff *skb; 681 682 skb = kmem_cache_alloc(net_hotdata.skbuff_cache, GFP_ATOMIC); 683 if (unlikely(!skb)) 684 return NULL; 685 686 memset(skb, 0, offsetof(struct sk_buff, tail)); 687 688 return __xdp_build_skb_from_frame(xdpf, skb, dev); 689 } 690 EXPORT_SYMBOL_GPL(xdp_build_skb_from_frame); 691 692 struct xdp_frame *xdpf_clone(struct xdp_frame *xdpf) 693 { 694 unsigned int headroom, totalsize; 695 struct xdp_frame *nxdpf; 696 struct page *page; 697 void *addr; 698 699 headroom = xdpf->headroom + sizeof(*xdpf); 700 totalsize = headroom + xdpf->len; 701 702 if (unlikely(totalsize > PAGE_SIZE)) 703 return NULL; 704 page = dev_alloc_page(); 705 if (!page) 706 return NULL; 707 addr = page_to_virt(page); 708 709 memcpy(addr, xdpf, totalsize); 710 711 nxdpf = addr; 712 nxdpf->data = addr + headroom; 713 nxdpf->frame_sz = PAGE_SIZE; 714 nxdpf->mem_type = MEM_TYPE_PAGE_ORDER0; 715 716 return nxdpf; 717 } 718 719 __bpf_kfunc_start_defs(); 720 721 /** 722 * bpf_xdp_metadata_rx_timestamp - Read XDP frame RX timestamp. 723 * @ctx: XDP context pointer. 724 * @timestamp: Return value pointer. 725 * 726 * Return: 727 * * Returns 0 on success or ``-errno`` on error. 728 * * ``-EOPNOTSUPP`` : means device driver does not implement kfunc 729 * * ``-ENODATA`` : means no RX-timestamp available for this frame 730 */ 731 __bpf_kfunc int bpf_xdp_metadata_rx_timestamp(const struct xdp_md *ctx, u64 *timestamp) 732 { 733 return -EOPNOTSUPP; 734 } 735 736 /** 737 * bpf_xdp_metadata_rx_hash - Read XDP frame RX hash. 738 * @ctx: XDP context pointer. 739 * @hash: Return value pointer. 740 * @rss_type: Return value pointer for RSS type. 741 * 742 * The RSS hash type (@rss_type) specifies what portion of packet headers NIC 743 * hardware used when calculating RSS hash value. The RSS type can be decoded 744 * via &enum xdp_rss_hash_type either matching on individual L3/L4 bits 745 * ``XDP_RSS_L*`` or by combined traditional *RSS Hashing Types* 746 * ``XDP_RSS_TYPE_L*``. 747 * 748 * Return: 749 * * Returns 0 on success or ``-errno`` on error. 750 * * ``-EOPNOTSUPP`` : means device driver doesn't implement kfunc 751 * * ``-ENODATA`` : means no RX-hash available for this frame 752 */ 753 __bpf_kfunc int bpf_xdp_metadata_rx_hash(const struct xdp_md *ctx, u32 *hash, 754 enum xdp_rss_hash_type *rss_type) 755 { 756 return -EOPNOTSUPP; 757 } 758 759 /** 760 * bpf_xdp_metadata_rx_vlan_tag - Get XDP packet outermost VLAN tag 761 * @ctx: XDP context pointer. 762 * @vlan_proto: Destination pointer for VLAN Tag protocol identifier (TPID). 763 * @vlan_tci: Destination pointer for VLAN TCI (VID + DEI + PCP) 764 * 765 * In case of success, ``vlan_proto`` contains *Tag protocol identifier (TPID)*, 766 * usually ``ETH_P_8021Q`` or ``ETH_P_8021AD``, but some networks can use 767 * custom TPIDs. ``vlan_proto`` is stored in **network byte order (BE)** 768 * and should be used as follows: 769 * ``if (vlan_proto == bpf_htons(ETH_P_8021Q)) do_something();`` 770 * 771 * ``vlan_tci`` contains the remaining 16 bits of a VLAN tag. 772 * Driver is expected to provide those in **host byte order (usually LE)**, 773 * so the bpf program should not perform byte conversion. 774 * According to 802.1Q standard, *VLAN TCI (Tag control information)* 775 * is a bit field that contains: 776 * *VLAN identifier (VID)* that can be read with ``vlan_tci & 0xfff``, 777 * *Drop eligible indicator (DEI)* - 1 bit, 778 * *Priority code point (PCP)* - 3 bits. 779 * For detailed meaning of DEI and PCP, please refer to other sources. 780 * 781 * Return: 782 * * Returns 0 on success or ``-errno`` on error. 783 * * ``-EOPNOTSUPP`` : device driver doesn't implement kfunc 784 * * ``-ENODATA`` : VLAN tag was not stripped or is not available 785 */ 786 __bpf_kfunc int bpf_xdp_metadata_rx_vlan_tag(const struct xdp_md *ctx, 787 __be16 *vlan_proto, u16 *vlan_tci) 788 { 789 return -EOPNOTSUPP; 790 } 791 792 __bpf_kfunc_end_defs(); 793 794 BTF_KFUNCS_START(xdp_metadata_kfunc_ids) 795 #define XDP_METADATA_KFUNC(_, __, name, ___) BTF_ID_FLAGS(func, name, KF_TRUSTED_ARGS) 796 XDP_METADATA_KFUNC_xxx 797 #undef XDP_METADATA_KFUNC 798 BTF_KFUNCS_END(xdp_metadata_kfunc_ids) 799 800 static const struct btf_kfunc_id_set xdp_metadata_kfunc_set = { 801 .owner = THIS_MODULE, 802 .set = &xdp_metadata_kfunc_ids, 803 }; 804 805 BTF_ID_LIST(xdp_metadata_kfunc_ids_unsorted) 806 #define XDP_METADATA_KFUNC(name, _, str, __) BTF_ID(func, str) 807 XDP_METADATA_KFUNC_xxx 808 #undef XDP_METADATA_KFUNC 809 810 u32 bpf_xdp_metadata_kfunc_id(int id) 811 { 812 /* xdp_metadata_kfunc_ids is sorted and can't be used */ 813 return xdp_metadata_kfunc_ids_unsorted[id]; 814 } 815 816 bool bpf_dev_bound_kfunc_id(u32 btf_id) 817 { 818 return btf_id_set8_contains(&xdp_metadata_kfunc_ids, btf_id); 819 } 820 821 static int __init xdp_metadata_init(void) 822 { 823 return register_btf_kfunc_id_set(BPF_PROG_TYPE_XDP, &xdp_metadata_kfunc_set); 824 } 825 late_initcall(xdp_metadata_init); 826 827 void xdp_set_features_flag(struct net_device *dev, xdp_features_t val) 828 { 829 val &= NETDEV_XDP_ACT_MASK; 830 if (dev->xdp_features == val) 831 return; 832 833 dev->xdp_features = val; 834 835 if (dev->reg_state == NETREG_REGISTERED) 836 call_netdevice_notifiers(NETDEV_XDP_FEAT_CHANGE, dev); 837 } 838 EXPORT_SYMBOL_GPL(xdp_set_features_flag); 839 840 void xdp_features_set_redirect_target(struct net_device *dev, bool support_sg) 841 { 842 xdp_features_t val = (dev->xdp_features | NETDEV_XDP_ACT_NDO_XMIT); 843 844 if (support_sg) 845 val |= NETDEV_XDP_ACT_NDO_XMIT_SG; 846 xdp_set_features_flag(dev, val); 847 } 848 EXPORT_SYMBOL_GPL(xdp_features_set_redirect_target); 849 850 void xdp_features_clear_redirect_target(struct net_device *dev) 851 { 852 xdp_features_t val = dev->xdp_features; 853 854 val &= ~(NETDEV_XDP_ACT_NDO_XMIT | NETDEV_XDP_ACT_NDO_XMIT_SG); 855 xdp_set_features_flag(dev, val); 856 } 857 EXPORT_SYMBOL_GPL(xdp_features_clear_redirect_target); 858