net/ipv6/seg6_local.c

   1 // SPDX-License-Identifier: GPL-2.0-or-later
   2 /*
   3  *  SR-IPv6 implementation
   4  *
   5  *  Authors:
   6  *  David Lebrun <[email protected]>
   7  *  eBPF support: Mathieu Xhonneux <[email protected]>
   8  */
   9
  10 #include <linux/types.h>
  11 #include <linux/skbuff.h>
  12 #include <linux/net.h>
  13 #include <linux/module.h>
  14 #include <net/ip.h>
  15 #include <net/lwtunnel.h>
  16 #include <net/netevent.h>
  17 #include <net/netns/generic.h>
  18 #include <net/ip6_fib.h>
  19 #include <net/route.h>
  20 #include <net/seg6.h>
  21 #include <linux/seg6.h>
  22 #include <linux/seg6_local.h>
  23 #include <net/addrconf.h>
  24 #include <net/ip6_route.h>
  25 #include <net/dst_cache.h>
  26 #ifdef CONFIG_IPV6_SEG6_HMAC
  27 #include <net/seg6_hmac.h>
  28 #endif
  29 #include <net/seg6_local.h>
  30 #include <linux/etherdevice.h>
  31 #include <linux/bpf.h>
  32
  33 struct seg6_local_lwt;
  34
  35 struct seg6_action_desc {
  36         int action;
  37         unsigned long attrs;
  38         int (*input)(struct sk_buff *skb, struct seg6_local_lwt *slwt);
  39         int static_headroom;
  40 };
  41
  42 struct bpf_lwt_prog {
  43         struct bpf_prog *prog;
  44         char *name;
  45 };
  46
  47 struct seg6_local_lwt {
  48         int action;
  49         struct ipv6_sr_hdr *srh;
  50         int table;
  51         struct in_addr nh4;
  52         struct in6_addr nh6;
  53         int iif;
  54         int oif;
  55         struct bpf_lwt_prog bpf;
  56
  57         int headroom;
  58         struct seg6_action_desc *desc;
  59 };
  60
  61 static struct seg6_local_lwt *seg6_local_lwtunnel(struct lwtunnel_state *lwt)
  62 {
  63         return (struct seg6_local_lwt *)lwt->data;
  64 }
  65
  66 static struct ipv6_sr_hdr *get_srh(struct sk_buff *skb)
  67 {
  68         struct ipv6_sr_hdr *srh;
  69         int len, srhoff = 0;
  70
  71         if (ipv6_find_hdr(skb, &srhoff, IPPROTO_ROUTING, NULL, NULL) < 0)
  72                 return NULL;
  73
  74         if (!pskb_may_pull(skb, srhoff + sizeof(*srh)))
  75                 return NULL;
  76
  77         srh = (struct ipv6_sr_hdr *)(skb->data + srhoff);
  78
  79         len = (srh->hdrlen + 1) << 3;
  80
  81         if (!pskb_may_pull(skb, srhoff + len))
  82                 return NULL;
  83
  84         /* note that pskb_may_pull may change pointers in header;
  85          * for this reason it is necessary to reload them when needed.
  86          */
  87         srh = (struct ipv6_sr_hdr *)(skb->data + srhoff);
  88
  89         if (!seg6_validate_srh(srh, len))
  90                 return NULL;
  91
  92         return srh;
  93 }
  94
  95 static struct ipv6_sr_hdr *get_and_validate_srh(struct sk_buff *skb)
  96 {
  97         struct ipv6_sr_hdr *srh;
  98
  99         srh = get_srh(skb);
 100         if (!srh)
 101                 return NULL;
 102
 103         if (srh->segments_left == 0)
 104                 return NULL;
 105
 106 #ifdef CONFIG_IPV6_SEG6_HMAC
 107         if (!seg6_hmac_validate_skb(skb))
 108                 return NULL;
 109 #endif
 110
 111         return srh;
 112 }
 113
 114 static bool decap_and_validate(struct sk_buff *skb, int proto)
 115 {
 116         struct ipv6_sr_hdr *srh;
 117         unsigned int off = 0;
 118
 119         srh = get_srh(skb);
 120         if (srh && srh->segments_left > 0)
 121                 return false;
 122
 123 #ifdef CONFIG_IPV6_SEG6_HMAC
 124         if (srh && !seg6_hmac_validate_skb(skb))
 125                 return false;
 126 #endif
 127
 128         if (ipv6_find_hdr(skb, &off, proto, NULL, NULL) < 0)
 129                 return false;
 130
 131         if (!pskb_pull(skb, off))
 132                 return false;
 133
 134         skb_postpull_rcsum(skb, skb_network_header(skb), off);
 135
 136         skb_reset_network_header(skb);
 137         skb_reset_transport_header(skb);
 138         skb->encapsulation = 0;
 139
 140         return true;
 141 }
 142
 143 static void advance_nextseg(struct ipv6_sr_hdr *srh, struct in6_addr *daddr)
 144 {
 145         struct in6_addr *addr;
 146
 147         srh->segments_left--;
 148         addr = srh->segments + srh->segments_left;
 149         *daddr = *addr;
 150 }
 151
 152 static int
 153 seg6_lookup_any_nexthop(struct sk_buff *skb, struct in6_addr *nhaddr,
 154                         u32 tbl_id, bool local_delivery)
 155 {
 156         struct net *net = dev_net(skb->dev);
 157         struct ipv6hdr *hdr = ipv6_hdr(skb);
 158         int flags = RT6_LOOKUP_F_HAS_SADDR;
 159         struct dst_entry *dst = NULL;
 160         struct rt6_info *rt;
 161         struct flowi6 fl6;
 162         int dev_flags = 0;
 163
 164         fl6.flowi6_iif = skb->dev->ifindex;
 165         fl6.daddr = nhaddr ? *nhaddr : hdr->daddr;
 166         fl6.saddr = hdr->saddr;
 167         fl6.flowlabel = ip6_flowinfo(hdr);
 168         fl6.flowi6_mark = skb->mark;
 169         fl6.flowi6_proto = hdr->nexthdr;
 170
 171         if (nhaddr)
 172                 fl6.flowi6_flags = FLOWI_FLAG_KNOWN_NH;
 173
 174         if (!tbl_id) {
 175                 dst = ip6_route_input_lookup(net, skb->dev, &fl6, skb, flags);
 176         } else {
 177                 struct fib6_table *table;
 178
 179                 table = fib6_get_table(net, tbl_id);
 180                 if (!table)
 181                         goto out;
 182
 183                 rt = ip6_pol_route(net, table, 0, &fl6, skb, flags);
 184                 dst = &rt->dst;
 185         }
 186
 187         /* we want to discard traffic destined for local packet processing,
 188          * if @local_delivery is set to false.
 189          */
 190         if (!local_delivery)
 191                 dev_flags |= IFF_LOOPBACK;
 192
 193         if (dst && (dst->dev->flags & dev_flags) && !dst->error) {
 194                 dst_release(dst);
 195                 dst = NULL;
 196         }
 197
 198 out:
 199         if (!dst) {
 200                 rt = net->ipv6.ip6_blk_hole_entry;
 201                 dst = &rt->dst;
 202                 dst_hold(dst);
 203         }
 204
 205         skb_dst_drop(skb);
 206         skb_dst_set(skb, dst);
 207         return dst->error;
 208 }
 209
 210 int seg6_lookup_nexthop(struct sk_buff *skb,
 211                         struct in6_addr *nhaddr, u32 tbl_id)
 212 {
 213         return seg6_lookup_any_nexthop(skb, nhaddr, tbl_id, false);
 214 }
 215
 216 /* regular endpoint function */
 217 static int input_action_end(struct sk_buff *skb, struct seg6_local_lwt *slwt)
 218 {
 219         struct ipv6_sr_hdr *srh;
 220
 221         srh = get_and_validate_srh(skb);
 222         if (!srh)
 223                 goto drop;
 224
 225         advance_nextseg(srh, &ipv6_hdr(skb)->daddr);
 226
 227         seg6_lookup_nexthop(skb, NULL, 0);
 228
 229         return dst_input(skb);
 230
 231 drop:
 232         kfree_skb(skb);
 233         return -EINVAL;
 234 }
 235
 236 /* regular endpoint, and forward to specified nexthop */
 237 static int input_action_end_x(struct sk_buff *skb, struct seg6_local_lwt *slwt)
 238 {
 239         struct ipv6_sr_hdr *srh;
 240
 241         srh = get_and_validate_srh(skb);
 242         if (!srh)
 243                 goto drop;
 244
 245         advance_nextseg(srh, &ipv6_hdr(skb)->daddr);
 246
 247         seg6_lookup_nexthop(skb, &slwt->nh6, 0);
 248
 249         return dst_input(skb);
 250
 251 drop:
 252         kfree_skb(skb);
 253         return -EINVAL;
 254 }
 255
 256 static int input_action_end_t(struct sk_buff *skb, struct seg6_local_lwt *slwt)
 257 {
 258         struct ipv6_sr_hdr *srh;
 259
 260         srh = get_and_validate_srh(skb);
 261         if (!srh)
 262                 goto drop;
 263
 264         advance_nextseg(srh, &ipv6_hdr(skb)->daddr);
 265
 266         seg6_lookup_nexthop(skb, NULL, slwt->table);
 267
 268         return dst_input(skb);
 269
 270 drop:
 271         kfree_skb(skb);
 272         return -EINVAL;
 273 }
 274
 275 /* decapsulate and forward inner L2 frame on specified interface */
 276 static int input_action_end_dx2(struct sk_buff *skb,
 277                                 struct seg6_local_lwt *slwt)
 278 {
 279         struct net *net = dev_net(skb->dev);
 280         struct net_device *odev;
 281         struct ethhdr *eth;
 282
 283         if (!decap_and_validate(skb, NEXTHDR_NONE))
 284                 goto drop;
 285
 286         if (!pskb_may_pull(skb, ETH_HLEN))
 287                 goto drop;
 288
 289         skb_reset_mac_header(skb);
 290         eth = (struct ethhdr *)skb->data;
 291
 292         /* To determine the frame's protocol, we assume it is 802.3. This avoids
 293          * a call to eth_type_trans(), which is not really relevant for our
 294          * use case.
 295          */
 296         if (!eth_proto_is_802_3(eth->h_proto))
 297                 goto drop;
 298
 299         odev = dev_get_by_index_rcu(net, slwt->oif);
 300         if (!odev)
 301                 goto drop;
 302
 303         /* As we accept Ethernet frames, make sure the egress device is of
 304          * the correct type.
 305          */
 306         if (odev->type != ARPHRD_ETHER)
 307                 goto drop;
 308
 309         if (!(odev->flags & IFF_UP) || !netif_carrier_ok(odev))
 310                 goto drop;
 311
 312         skb_orphan(skb);
 313
 314         if (skb_warn_if_lro(skb))
 315                 goto drop;
 316
 317         skb_forward_csum(skb);
 318
 319         if (skb->len - ETH_HLEN > odev->mtu)
 320                 goto drop;
 321
 322         skb->dev = odev;
 323         skb->protocol = eth->h_proto;
 324
 325         return dev_queue_xmit(skb);
 326
 327 drop:
 328         kfree_skb(skb);
 329         return -EINVAL;
 330 }
 331
 332 /* decapsulate and forward to specified nexthop */
 333 static int input_action_end_dx6(struct sk_buff *skb,
 334                                 struct seg6_local_lwt *slwt)
 335 {
 336         struct in6_addr *nhaddr = NULL;
 337
 338         /* this function accepts IPv6 encapsulated packets, with either
 339          * an SRH with SL=0, or no SRH.
 340          */
 341
 342         if (!decap_and_validate(skb, IPPROTO_IPV6))
 343                 goto drop;
 344
 345         if (!pskb_may_pull(skb, sizeof(struct ipv6hdr)))
 346                 goto drop;
 347
 348         /* The inner packet is not associated to any local interface,
 349          * so we do not call netif_rx().
 350          *
 351          * If slwt->nh6 is set to ::, then lookup the nexthop for the
 352          * inner packet's DA. Otherwise, use the specified nexthop.
 353          */
 354
 355         if (!ipv6_addr_any(&slwt->nh6))
 356                 nhaddr = &slwt->nh6;
 357
 358         skb_set_transport_header(skb, sizeof(struct ipv6hdr));
 359
 360         seg6_lookup_nexthop(skb, nhaddr, 0);
 361
 362         return dst_input(skb);
 363 drop:
 364         kfree_skb(skb);
 365         return -EINVAL;
 366 }
 367
 368 static int input_action_end_dx4(struct sk_buff *skb,
 369                                 struct seg6_local_lwt *slwt)
 370 {
 371         struct iphdr *iph;
 372         __be32 nhaddr;
 373         int err;
 374
 375         if (!decap_and_validate(skb, IPPROTO_IPIP))
 376                 goto drop;
 377
 378         if (!pskb_may_pull(skb, sizeof(struct iphdr)))
 379                 goto drop;
 380
 381         skb->protocol = htons(ETH_P_IP);
 382
 383         iph = ip_hdr(skb);
 384
 385         nhaddr = slwt->nh4.s_addr ?: iph->daddr;
 386
 387         skb_dst_drop(skb);
 388
 389         skb_set_transport_header(skb, sizeof(struct iphdr));
 390
 391         err = ip_route_input(skb, nhaddr, iph->saddr, 0, skb->dev);
 392         if (err)
 393                 goto drop;
 394
 395         return dst_input(skb);
 396
 397 drop:
 398         kfree_skb(skb);
 399         return -EINVAL;
 400 }
 401
 402 static int input_action_end_dt6(struct sk_buff *skb,
 403                                 struct seg6_local_lwt *slwt)
 404 {
 405         if (!decap_and_validate(skb, IPPROTO_IPV6))
 406                 goto drop;
 407
 408         if (!pskb_may_pull(skb, sizeof(struct ipv6hdr)))
 409                 goto drop;
 410
 411         skb_set_transport_header(skb, sizeof(struct ipv6hdr));
 412
 413         seg6_lookup_any_nexthop(skb, NULL, slwt->table, true);
 414
 415         return dst_input(skb);
 416
 417 drop:
 418         kfree_skb(skb);
 419         return -EINVAL;
 420 }
 421
 422 /* push an SRH on top of the current one */
 423 static int input_action_end_b6(struct sk_buff *skb, struct seg6_local_lwt *slwt)
 424 {
 425         struct ipv6_sr_hdr *srh;
 426         int err = -EINVAL;
 427
 428         srh = get_and_validate_srh(skb);
 429         if (!srh)
 430                 goto drop;
 431
 432         err = seg6_do_srh_inline(skb, slwt->srh);
 433         if (err)
 434                 goto drop;
 435
 436         ipv6_hdr(skb)->payload_len = htons(skb->len - sizeof(struct ipv6hdr));
 437         skb_set_transport_header(skb, sizeof(struct ipv6hdr));
 438
 439         seg6_lookup_nexthop(skb, NULL, 0);
 440
 441         return dst_input(skb);
 442
 443 drop:
 444         kfree_skb(skb);
 445         return err;
 446 }
 447
 448 /* encapsulate within an outer IPv6 header and a specified SRH */
 449 static int input_action_end_b6_encap(struct sk_buff *skb,
 450                                      struct seg6_local_lwt *slwt)
 451 {
 452         struct ipv6_sr_hdr *srh;
 453         int err = -EINVAL;
 454
 455         srh = get_and_validate_srh(skb);
 456         if (!srh)
 457                 goto drop;
 458
 459         advance_nextseg(srh, &ipv6_hdr(skb)->daddr);
 460
 461         skb_reset_inner_headers(skb);
 462         skb->encapsulation = 1;
 463
 464         err = seg6_do_srh_encap(skb, slwt->srh, IPPROTO_IPV6);
 465         if (err)
 466                 goto drop;
 467
 468         ipv6_hdr(skb)->payload_len = htons(skb->len - sizeof(struct ipv6hdr));
 469         skb_set_transport_header(skb, sizeof(struct ipv6hdr));
 470
 471         seg6_lookup_nexthop(skb, NULL, 0);
 472
 473         return dst_input(skb);
 474
 475 drop:
 476         kfree_skb(skb);
 477         return err;
 478 }
 479
 480 DEFINE_PER_CPU(struct seg6_bpf_srh_state, seg6_bpf_srh_states);
 481
 482 bool seg6_bpf_has_valid_srh(struct sk_buff *skb)
 483 {
 484         struct seg6_bpf_srh_state *srh_state =
 485                 this_cpu_ptr(&seg6_bpf_srh_states);
 486         struct ipv6_sr_hdr *srh = srh_state->srh;
 487
 488         if (unlikely(srh == NULL))
 489                 return false;
 490
 491         if (unlikely(!srh_state->valid)) {
 492                 if ((srh_state->hdrlen & 7) != 0)
 493                         return false;
 494
 495                 srh->hdrlen = (u8)(srh_state->hdrlen >> 3);
 496                 if (!seg6_validate_srh(srh, (srh->hdrlen + 1) << 3))
 497                         return false;
 498
 499                 srh_state->valid = true;
 500         }
 501
 502         return true;
 503 }
 504
 505 static int input_action_end_bpf(struct sk_buff *skb,
 506                                 struct seg6_local_lwt *slwt)
 507 {
 508         struct seg6_bpf_srh_state *srh_state =
 509                 this_cpu_ptr(&seg6_bpf_srh_states);
 510         struct ipv6_sr_hdr *srh;
 511         int ret;
 512
 513         srh = get_and_validate_srh(skb);
 514         if (!srh) {
 515                 kfree_skb(skb);
 516                 return -EINVAL;
 517         }
 518         advance_nextseg(srh, &ipv6_hdr(skb)->daddr);
 519
 520         /* preempt_disable is needed to protect the per-CPU buffer srh_state,
 521          * which is also accessed by the bpf_lwt_seg6_* helpers
 522          */
 523         preempt_disable();
 524         srh_state->srh = srh;
 525         srh_state->hdrlen = srh->hdrlen << 3;
 526         srh_state->valid = true;
 527
 528         rcu_read_lock();
 529         bpf_compute_data_pointers(skb);
 530         ret = bpf_prog_run_save_cb(slwt->bpf.prog, skb);
 531         rcu_read_unlock();
 532
 533         switch (ret) {
 534         case BPF_OK:
 535         case BPF_REDIRECT:
 536                 break;
 537         case BPF_DROP:
 538                 goto drop;
 539         default:
 540                 pr_warn_once("bpf-seg6local: Illegal return value %u\n", ret);
 541                 goto drop;
 542         }
 543
 544         if (srh_state->srh && !seg6_bpf_has_valid_srh(skb))
 545                 goto drop;
 546
 547         preempt_enable();
 548         if (ret != BPF_REDIRECT)
 549                 seg6_lookup_nexthop(skb, NULL, 0);
 550
 551         return dst_input(skb);
 552
 553 drop:
 554         preempt_enable();
 555         kfree_skb(skb);
 556         return -EINVAL;
 557 }
 558
 559 static struct seg6_action_desc seg6_action_table[] = {
 560         {
 561                 .action         = SEG6_LOCAL_ACTION_END,
 562                 .attrs          = 0,
 563                 .input          = input_action_end,
 564         },
 565         {
 566                 .action         = SEG6_LOCAL_ACTION_END_X,
 567                 .attrs          = (1 << SEG6_LOCAL_NH6),
 568                 .input          = input_action_end_x,
 569         },
 570         {
 571                 .action         = SEG6_LOCAL_ACTION_END_T,
 572                 .attrs          = (1 << SEG6_LOCAL_TABLE),
 573                 .input          = input_action_end_t,
 574         },
 575         {
 576                 .action         = SEG6_LOCAL_ACTION_END_DX2,
 577                 .attrs          = (1 << SEG6_LOCAL_OIF),
 578                 .input          = input_action_end_dx2,
 579         },
 580         {
 581                 .action         = SEG6_LOCAL_ACTION_END_DX6,
 582                 .attrs          = (1 << SEG6_LOCAL_NH6),
 583                 .input          = input_action_end_dx6,
 584         },
 585         {
 586                 .action         = SEG6_LOCAL_ACTION_END_DX4,
 587                 .attrs          = (1 << SEG6_LOCAL_NH4),
 588                 .input          = input_action_end_dx4,
 589         },
 590         {
 591                 .action         = SEG6_LOCAL_ACTION_END_DT6,
 592                 .attrs          = (1 << SEG6_LOCAL_TABLE),
 593                 .input          = input_action_end_dt6,
 594         },
 595         {
 596                 .action         = SEG6_LOCAL_ACTION_END_B6,
 597                 .attrs          = (1 << SEG6_LOCAL_SRH),
 598                 .input          = input_action_end_b6,
 599         },
 600         {
 601                 .action         = SEG6_LOCAL_ACTION_END_B6_ENCAP,
 602                 .attrs          = (1 << SEG6_LOCAL_SRH),
 603                 .input          = input_action_end_b6_encap,
 604                 .static_headroom        = sizeof(struct ipv6hdr),
 605         },
 606         {
 607                 .action         = SEG6_LOCAL_ACTION_END_BPF,
 608                 .attrs          = (1 << SEG6_LOCAL_BPF),
 609                 .input          = input_action_end_bpf,
 610         },
 611
 612 };
 613
 614 static struct seg6_action_desc *__get_action_desc(int action)
 615 {
 616         struct seg6_action_desc *desc;
 617         int i, count;
 618
 619         count = ARRAY_SIZE(seg6_action_table);
 620         for (i = 0; i < count; i++) {
 621                 desc = &seg6_action_table[i];
 622                 if (desc->action == action)
 623                         return desc;
 624         }
 625
 626         return NULL;
 627 }
 628
 629 static int seg6_local_input(struct sk_buff *skb)
 630 {
 631         struct dst_entry *orig_dst = skb_dst(skb);
 632         struct seg6_action_desc *desc;
 633         struct seg6_local_lwt *slwt;
 634
 635         if (skb->protocol != htons(ETH_P_IPV6)) {
 636                 kfree_skb(skb);
 637                 return -EINVAL;
 638         }
 639
 640         slwt = seg6_local_lwtunnel(orig_dst->lwtstate);
 641         desc = slwt->desc;
 642
 643         return desc->input(skb, slwt);
 644 }
 645
 646 static const struct nla_policy seg6_local_policy[SEG6_LOCAL_MAX + 1] = {
 647         [SEG6_LOCAL_ACTION]     = { .type = NLA_U32 },
 648         [SEG6_LOCAL_SRH]        = { .type = NLA_BINARY },
 649         [SEG6_LOCAL_TABLE]      = { .type = NLA_U32 },
 650         [SEG6_LOCAL_NH4]        = { .type = NLA_BINARY,
 651                                     .len = sizeof(struct in_addr) },
 652         [SEG6_LOCAL_NH6]        = { .type = NLA_BINARY,
 653                                     .len = sizeof(struct in6_addr) },
 654         [SEG6_LOCAL_IIF]        = { .type = NLA_U32 },
 655         [SEG6_LOCAL_OIF]        = { .type = NLA_U32 },
 656         [SEG6_LOCAL_BPF]        = { .type = NLA_NESTED },
 657 };
 658
 659 static int parse_nla_srh(struct nlattr **attrs, struct seg6_local_lwt *slwt)
 660 {
 661         struct ipv6_sr_hdr *srh;
 662         int len;
 663
 664         srh = nla_data(attrs[SEG6_LOCAL_SRH]);
 665         len = nla_len(attrs[SEG6_LOCAL_SRH]);
 666
 667         /* SRH must contain at least one segment */
 668         if (len < sizeof(*srh) + sizeof(struct in6_addr))
 669                 return -EINVAL;
 670
 671         if (!seg6_validate_srh(srh, len))
 672                 return -EINVAL;
 673
 674         slwt->srh = kmemdup(srh, len, GFP_KERNEL);
 675         if (!slwt->srh)
 676                 return -ENOMEM;
 677
 678         slwt->headroom += len;
 679
 680         return 0;
 681 }
 682
 683 static int put_nla_srh(struct sk_buff *skb, struct seg6_local_lwt *slwt)
 684 {
 685         struct ipv6_sr_hdr *srh;
 686         struct nlattr *nla;
 687         int len;
 688
 689         srh = slwt->srh;
 690         len = (srh->hdrlen + 1) << 3;
 691
 692         nla = nla_reserve(skb, SEG6_LOCAL_SRH, len);
 693         if (!nla)
 694                 return -EMSGSIZE;
 695
 696         memcpy(nla_data(nla), srh, len);
 697
 698         return 0;
 699 }
 700
 701 static int cmp_nla_srh(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
 702 {
 703         int len = (a->srh->hdrlen + 1) << 3;
 704
 705         if (len != ((b->srh->hdrlen + 1) << 3))
 706                 return 1;
 707
 708         return memcmp(a->srh, b->srh, len);
 709 }
 710
 711 static int parse_nla_table(struct nlattr **attrs, struct seg6_local_lwt *slwt)
 712 {
 713         slwt->table = nla_get_u32(attrs[SEG6_LOCAL_TABLE]);
 714
 715         return 0;
 716 }
 717
 718 static int put_nla_table(struct sk_buff *skb, struct seg6_local_lwt *slwt)
 719 {
 720         if (nla_put_u32(skb, SEG6_LOCAL_TABLE, slwt->table))
 721                 return -EMSGSIZE;
 722
 723         return 0;
 724 }
 725
 726 static int cmp_nla_table(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
 727 {
 728         if (a->table != b->table)
 729                 return 1;
 730
 731         return 0;
 732 }
 733
 734 static int parse_nla_nh4(struct nlattr **attrs, struct seg6_local_lwt *slwt)
 735 {
 736         memcpy(&slwt->nh4, nla_data(attrs[SEG6_LOCAL_NH4]),
 737                sizeof(struct in_addr));
 738
 739         return 0;
 740 }
 741
 742 static int put_nla_nh4(struct sk_buff *skb, struct seg6_local_lwt *slwt)
 743 {
 744         struct nlattr *nla;
 745
 746         nla = nla_reserve(skb, SEG6_LOCAL_NH4, sizeof(struct in_addr));
 747         if (!nla)
 748                 return -EMSGSIZE;
 749
 750         memcpy(nla_data(nla), &slwt->nh4, sizeof(struct in_addr));
 751
 752         return 0;
 753 }
 754
 755 static int cmp_nla_nh4(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
 756 {
 757         return memcmp(&a->nh4, &b->nh4, sizeof(struct in_addr));
 758 }
 759
 760 static int parse_nla_nh6(struct nlattr **attrs, struct seg6_local_lwt *slwt)
 761 {
 762         memcpy(&slwt->nh6, nla_data(attrs[SEG6_LOCAL_NH6]),
 763                sizeof(struct in6_addr));
 764
 765         return 0;
 766 }
 767
 768 static int put_nla_nh6(struct sk_buff *skb, struct seg6_local_lwt *slwt)
 769 {
 770         struct nlattr *nla;
 771
 772         nla = nla_reserve(skb, SEG6_LOCAL_NH6, sizeof(struct in6_addr));
 773         if (!nla)
 774                 return -EMSGSIZE;
 775
 776         memcpy(nla_data(nla), &slwt->nh6, sizeof(struct in6_addr));
 777
 778         return 0;
 779 }
 780
 781 static int cmp_nla_nh6(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
 782 {
 783         return memcmp(&a->nh6, &b->nh6, sizeof(struct in6_addr));
 784 }
 785
 786 static int parse_nla_iif(struct nlattr **attrs, struct seg6_local_lwt *slwt)
 787 {
 788         slwt->iif = nla_get_u32(attrs[SEG6_LOCAL_IIF]);
 789
 790         return 0;
 791 }
 792
 793 static int put_nla_iif(struct sk_buff *skb, struct seg6_local_lwt *slwt)
 794 {
 795         if (nla_put_u32(skb, SEG6_LOCAL_IIF, slwt->iif))
 796                 return -EMSGSIZE;
 797
 798         return 0;
 799 }
 800
 801 static int cmp_nla_iif(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
 802 {
 803         if (a->iif != b->iif)
 804                 return 1;
 805
 806         return 0;
 807 }
 808
 809 static int parse_nla_oif(struct nlattr **attrs, struct seg6_local_lwt *slwt)
 810 {
 811         slwt->oif = nla_get_u32(attrs[SEG6_LOCAL_OIF]);
 812
 813         return 0;
 814 }
 815
 816 static int put_nla_oif(struct sk_buff *skb, struct seg6_local_lwt *slwt)
 817 {
 818         if (nla_put_u32(skb, SEG6_LOCAL_OIF, slwt->oif))
 819                 return -EMSGSIZE;
 820
 821         return 0;
 822 }
 823
 824 static int cmp_nla_oif(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
 825 {
 826         if (a->oif != b->oif)
 827                 return 1;
 828
 829         return 0;
 830 }
 831
 832 #define MAX_PROG_NAME 256
 833 static const struct nla_policy bpf_prog_policy[SEG6_LOCAL_BPF_PROG_MAX + 1] = {
 834         [SEG6_LOCAL_BPF_PROG]      = { .type = NLA_U32, },
 835         [SEG6_LOCAL_BPF_PROG_NAME] = { .type = NLA_NUL_STRING,
 836                                        .len = MAX_PROG_NAME },
 837 };
 838
 839 static int parse_nla_bpf(struct nlattr **attrs, struct seg6_local_lwt *slwt)
 840 {
 841         struct nlattr *tb[SEG6_LOCAL_BPF_PROG_MAX + 1];
 842         struct bpf_prog *p;
 843         int ret;
 844         u32 fd;
 845
 846         ret = nla_parse_nested_deprecated(tb, SEG6_LOCAL_BPF_PROG_MAX,
 847                                           attrs[SEG6_LOCAL_BPF],
 848                                           bpf_prog_policy, NULL);
 849         if (ret < 0)
 850                 return ret;
 851
 852         if (!tb[SEG6_LOCAL_BPF_PROG] || !tb[SEG6_LOCAL_BPF_PROG_NAME])
 853                 return -EINVAL;
 854
 855         slwt->bpf.name = nla_memdup(tb[SEG6_LOCAL_BPF_PROG_NAME], GFP_KERNEL);
 856         if (!slwt->bpf.name)
 857                 return -ENOMEM;
 858
 859         fd = nla_get_u32(tb[SEG6_LOCAL_BPF_PROG]);
 860         p = bpf_prog_get_type(fd, BPF_PROG_TYPE_LWT_SEG6LOCAL);
 861         if (IS_ERR(p)) {
 862                 kfree(slwt->bpf.name);
 863                 return PTR_ERR(p);
 864         }
 865
 866         slwt->bpf.prog = p;
 867         return 0;
 868 }
 869
 870 static int put_nla_bpf(struct sk_buff *skb, struct seg6_local_lwt *slwt)
 871 {
 872         struct nlattr *nest;
 873
 874         if (!slwt->bpf.prog)
 875                 return 0;
 876
 877         nest = nla_nest_start_noflag(skb, SEG6_LOCAL_BPF);
 878         if (!nest)
 879                 return -EMSGSIZE;
 880
 881         if (nla_put_u32(skb, SEG6_LOCAL_BPF_PROG, slwt->bpf.prog->aux->id))
 882                 return -EMSGSIZE;
 883
 884         if (slwt->bpf.name &&
 885             nla_put_string(skb, SEG6_LOCAL_BPF_PROG_NAME, slwt->bpf.name))
 886                 return -EMSGSIZE;
 887
 888         return nla_nest_end(skb, nest);
 889 }
 890
 891 static int cmp_nla_bpf(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
 892 {
 893         if (!a->bpf.name && !b->bpf.name)
 894                 return 0;
 895
 896         if (!a->bpf.name || !b->bpf.name)
 897                 return 1;
 898
 899         return strcmp(a->bpf.name, b->bpf.name);
 900 }
 901
 902 struct seg6_action_param {
 903         int (*parse)(struct nlattr **attrs, struct seg6_local_lwt *slwt);
 904         int (*put)(struct sk_buff *skb, struct seg6_local_lwt *slwt);
 905         int (*cmp)(struct seg6_local_lwt *a, struct seg6_local_lwt *b);
 906 };
 907
 908 static struct seg6_action_param seg6_action_params[SEG6_LOCAL_MAX + 1] = {
 909         [SEG6_LOCAL_SRH]        = { .parse = parse_nla_srh,
 910                                     .put = put_nla_srh,
 911                                     .cmp = cmp_nla_srh },
 912
 913         [SEG6_LOCAL_TABLE]      = { .parse = parse_nla_table,
 914                                     .put = put_nla_table,
 915                                     .cmp = cmp_nla_table },
 916
 917         [SEG6_LOCAL_NH4]        = { .parse = parse_nla_nh4,
 918                                     .put = put_nla_nh4,
 919                                     .cmp = cmp_nla_nh4 },
 920
 921         [SEG6_LOCAL_NH6]        = { .parse = parse_nla_nh6,
 922                                     .put = put_nla_nh6,
 923                                     .cmp = cmp_nla_nh6 },
 924
 925         [SEG6_LOCAL_IIF]        = { .parse = parse_nla_iif,
 926                                     .put = put_nla_iif,
 927                                     .cmp = cmp_nla_iif },
 928
 929         [SEG6_LOCAL_OIF]        = { .parse = parse_nla_oif,
 930                                     .put = put_nla_oif,
 931                                     .cmp = cmp_nla_oif },
 932
 933         [SEG6_LOCAL_BPF]        = { .parse = parse_nla_bpf,
 934                                     .put = put_nla_bpf,
 935                                     .cmp = cmp_nla_bpf },
 936
 937 };
 938
 939 static int parse_nla_action(struct nlattr **attrs, struct seg6_local_lwt *slwt)
 940 {
 941         struct seg6_action_param *param;
 942         struct seg6_action_desc *desc;
 943         int i, err;
 944
 945         desc = __get_action_desc(slwt->action);
 946         if (!desc)
 947                 return -EINVAL;
 948
 949         if (!desc->input)
 950                 return -EOPNOTSUPP;
 951
 952         slwt->desc = desc;
 953         slwt->headroom += desc->static_headroom;
 954
 955         for (i = 0; i < SEG6_LOCAL_MAX + 1; i++) {
 956                 if (desc->attrs & (1 << i)) {
 957                         if (!attrs[i])
 958                                 return -EINVAL;
 959
 960                         param = &seg6_action_params[i];
 961
 962                         err = param->parse(attrs, slwt);
 963                         if (err < 0)
 964                                 return err;
 965                 }
 966         }
 967
 968         return 0;
 969 }
 970
 971 static int seg6_local_build_state(struct nlattr *nla, unsigned int family,
 972                                   const void *cfg, struct lwtunnel_state **ts,
 973                                   struct netlink_ext_ack *extack)
 974 {
 975         struct nlattr *tb[SEG6_LOCAL_MAX + 1];
 976         struct lwtunnel_state *newts;
 977         struct seg6_local_lwt *slwt;
 978         int err;
 979
 980         if (family != AF_INET6)
 981                 return -EINVAL;
 982
 983         err = nla_parse_nested_deprecated(tb, SEG6_LOCAL_MAX, nla,
 984                                           seg6_local_policy, extack);
 985
 986         if (err < 0)
 987                 return err;
 988
 989         if (!tb[SEG6_LOCAL_ACTION])
 990                 return -EINVAL;
 991
 992         newts = lwtunnel_state_alloc(sizeof(*slwt));
 993         if (!newts)
 994                 return -ENOMEM;
 995
 996         slwt = seg6_local_lwtunnel(newts);
 997         slwt->action = nla_get_u32(tb[SEG6_LOCAL_ACTION]);
 998
 999         err = parse_nla_action(tb, slwt);
1000         if (err < 0)
1001                 goto out_free;
1002
1003         newts->type = LWTUNNEL_ENCAP_SEG6_LOCAL;
1004         newts->flags = LWTUNNEL_STATE_INPUT_REDIRECT;
1005         newts->headroom = slwt->headroom;
1006
1007         *ts = newts;
1008
1009         return 0;
1010
1011 out_free:
1012         kfree(slwt->srh);
1013         kfree(newts);
1014         return err;
1015 }
1016
1017 static void seg6_local_destroy_state(struct lwtunnel_state *lwt)
1018 {
1019         struct seg6_local_lwt *slwt = seg6_local_lwtunnel(lwt);
1020
1021         kfree(slwt->srh);
1022
1023         if (slwt->desc->attrs & (1 << SEG6_LOCAL_BPF)) {
1024                 kfree(slwt->bpf.name);
1025                 bpf_prog_put(slwt->bpf.prog);
1026         }
1027
1028         return;
1029 }
1030
1031 static int seg6_local_fill_encap(struct sk_buff *skb,
1032                                  struct lwtunnel_state *lwt)
1033 {
1034         struct seg6_local_lwt *slwt = seg6_local_lwtunnel(lwt);
1035         struct seg6_action_param *param;
1036         int i, err;
1037
1038         if (nla_put_u32(skb, SEG6_LOCAL_ACTION, slwt->action))
1039                 return -EMSGSIZE;
1040
1041         for (i = 0; i < SEG6_LOCAL_MAX + 1; i++) {
1042                 if (slwt->desc->attrs & (1 << i)) {
1043                         param = &seg6_action_params[i];
1044                         err = param->put(skb, slwt);
1045                         if (err < 0)
1046                                 return err;
1047                 }
1048         }
1049
1050         return 0;
1051 }
1052
1053 static int seg6_local_get_encap_size(struct lwtunnel_state *lwt)
1054 {
1055         struct seg6_local_lwt *slwt = seg6_local_lwtunnel(lwt);
1056         unsigned long attrs;
1057         int nlsize;
1058
1059         nlsize = nla_total_size(4); /* action */
1060
1061         attrs = slwt->desc->attrs;
1062
1063         if (attrs & (1 << SEG6_LOCAL_SRH))
1064                 nlsize += nla_total_size((slwt->srh->hdrlen + 1) << 3);
1065
1066         if (attrs & (1 << SEG6_LOCAL_TABLE))
1067                 nlsize += nla_total_size(4);
1068
1069         if (attrs & (1 << SEG6_LOCAL_NH4))
1070                 nlsize += nla_total_size(4);
1071
1072         if (attrs & (1 << SEG6_LOCAL_NH6))
1073                 nlsize += nla_total_size(16);
1074
1075         if (attrs & (1 << SEG6_LOCAL_IIF))
1076                 nlsize += nla_total_size(4);
1077
1078         if (attrs & (1 << SEG6_LOCAL_OIF))
1079                 nlsize += nla_total_size(4);
1080
1081         if (attrs & (1 << SEG6_LOCAL_BPF))
1082                 nlsize += nla_total_size(sizeof(struct nlattr)) +
1083                        nla_total_size(MAX_PROG_NAME) +
1084                        nla_total_size(4);
1085
1086         return nlsize;
1087 }
1088
1089 static int seg6_local_cmp_encap(struct lwtunnel_state *a,
1090                                 struct lwtunnel_state *b)
1091 {
1092         struct seg6_local_lwt *slwt_a, *slwt_b;
1093         struct seg6_action_param *param;
1094         int i;
1095
1096         slwt_a = seg6_local_lwtunnel(a);
1097         slwt_b = seg6_local_lwtunnel(b);
1098
1099         if (slwt_a->action != slwt_b->action)
1100                 return 1;
1101
1102         if (slwt_a->desc->attrs != slwt_b->desc->attrs)
1103                 return 1;
1104
1105         for (i = 0; i < SEG6_LOCAL_MAX + 1; i++) {
1106                 if (slwt_a->desc->attrs & (1 << i)) {
1107                         param = &seg6_action_params[i];
1108                         if (param->cmp(slwt_a, slwt_b))
1109                                 return 1;
1110                 }
1111         }
1112
1113         return 0;
1114 }
1115
1116 static const struct lwtunnel_encap_ops seg6_local_ops = {
1117         .build_state    = seg6_local_build_state,
1118         .destroy_state  = seg6_local_destroy_state,
1119         .input          = seg6_local_input,
1120         .fill_encap     = seg6_local_fill_encap,
1121         .get_encap_size = seg6_local_get_encap_size,
1122         .cmp_encap      = seg6_local_cmp_encap,
1123         .owner          = THIS_MODULE,
1124 };
1125
1126 int __init seg6_local_init(void)
1127 {
1128         return lwtunnel_encap_add_ops(&seg6_local_ops,
1129                                       LWTUNNEL_ENCAP_SEG6_LOCAL);
1130 }
1131
1132 void seg6_local_exit(void)
1133 {
1134         lwtunnel_encap_del_ops(&seg6_local_ops, LWTUNNEL_ENCAP_SEG6_LOCAL);
1135 }