Linux Audio

Check our new training course

Loading...
Note: File does not exist in v3.15.
   1// SPDX-License-Identifier: GPL-2.0-only
   2#include <linux/module.h>
   3#include <linux/errno.h>
   4#include <linux/socket.h>
   5#include <linux/skbuff.h>
   6#include <linux/ip.h>
   7#include <linux/icmp.h>
   8#include <linux/udp.h>
   9#include <linux/types.h>
  10#include <linux/kernel.h>
  11#include <net/genetlink.h>
  12#include <net/gue.h>
  13#include <net/fou.h>
  14#include <net/ip.h>
  15#include <net/protocol.h>
  16#include <net/udp.h>
  17#include <net/udp_tunnel.h>
  18#include <net/xfrm.h>
  19#include <uapi/linux/fou.h>
  20#include <uapi/linux/genetlink.h>
  21
  22struct fou {
  23	struct socket *sock;
  24	u8 protocol;
  25	u8 flags;
  26	__be16 port;
  27	u8 family;
  28	u16 type;
  29	struct list_head list;
  30	struct rcu_head rcu;
  31};
  32
  33#define FOU_F_REMCSUM_NOPARTIAL BIT(0)
  34
  35struct fou_cfg {
  36	u16 type;
  37	u8 protocol;
  38	u8 flags;
  39	struct udp_port_cfg udp_config;
  40};
  41
  42static unsigned int fou_net_id;
  43
  44struct fou_net {
  45	struct list_head fou_list;
  46	struct mutex fou_lock;
  47};
  48
  49static inline struct fou *fou_from_sock(struct sock *sk)
  50{
  51	return sk->sk_user_data;
  52}
  53
  54static int fou_recv_pull(struct sk_buff *skb, struct fou *fou, size_t len)
  55{
  56	/* Remove 'len' bytes from the packet (UDP header and
  57	 * FOU header if present).
  58	 */
  59	if (fou->family == AF_INET)
  60		ip_hdr(skb)->tot_len = htons(ntohs(ip_hdr(skb)->tot_len) - len);
  61	else
  62		ipv6_hdr(skb)->payload_len =
  63		    htons(ntohs(ipv6_hdr(skb)->payload_len) - len);
  64
  65	__skb_pull(skb, len);
  66	skb_postpull_rcsum(skb, udp_hdr(skb), len);
  67	skb_reset_transport_header(skb);
  68	return iptunnel_pull_offloads(skb);
  69}
  70
  71static int fou_udp_recv(struct sock *sk, struct sk_buff *skb)
  72{
  73	struct fou *fou = fou_from_sock(sk);
  74
  75	if (!fou)
  76		return 1;
  77
  78	if (fou_recv_pull(skb, fou, sizeof(struct udphdr)))
  79		goto drop;
  80
  81	return -fou->protocol;
  82
  83drop:
  84	kfree_skb(skb);
  85	return 0;
  86}
  87
  88static struct guehdr *gue_remcsum(struct sk_buff *skb, struct guehdr *guehdr,
  89				  void *data, size_t hdrlen, u8 ipproto,
  90				  bool nopartial)
  91{
  92	__be16 *pd = data;
  93	size_t start = ntohs(pd[0]);
  94	size_t offset = ntohs(pd[1]);
  95	size_t plen = sizeof(struct udphdr) + hdrlen +
  96	    max_t(size_t, offset + sizeof(u16), start);
  97
  98	if (skb->remcsum_offload)
  99		return guehdr;
 100
 101	if (!pskb_may_pull(skb, plen))
 102		return NULL;
 103	guehdr = (struct guehdr *)&udp_hdr(skb)[1];
 104
 105	skb_remcsum_process(skb, (void *)guehdr + hdrlen,
 106			    start, offset, nopartial);
 107
 108	return guehdr;
 109}
 110
 111static int gue_control_message(struct sk_buff *skb, struct guehdr *guehdr)
 112{
 113	/* No support yet */
 114	kfree_skb(skb);
 115	return 0;
 116}
 117
 118static int gue_udp_recv(struct sock *sk, struct sk_buff *skb)
 119{
 120	struct fou *fou = fou_from_sock(sk);
 121	size_t len, optlen, hdrlen;
 122	struct guehdr *guehdr;
 123	void *data;
 124	u16 doffset = 0;
 125	u8 proto_ctype;
 126
 127	if (!fou)
 128		return 1;
 129
 130	len = sizeof(struct udphdr) + sizeof(struct guehdr);
 131	if (!pskb_may_pull(skb, len))
 132		goto drop;
 133
 134	guehdr = (struct guehdr *)&udp_hdr(skb)[1];
 135
 136	switch (guehdr->version) {
 137	case 0: /* Full GUE header present */
 138		break;
 139
 140	case 1: {
 141		/* Direct encapsulation of IPv4 or IPv6 */
 142
 143		int prot;
 144
 145		switch (((struct iphdr *)guehdr)->version) {
 146		case 4:
 147			prot = IPPROTO_IPIP;
 148			break;
 149		case 6:
 150			prot = IPPROTO_IPV6;
 151			break;
 152		default:
 153			goto drop;
 154		}
 155
 156		if (fou_recv_pull(skb, fou, sizeof(struct udphdr)))
 157			goto drop;
 158
 159		return -prot;
 160	}
 161
 162	default: /* Undefined version */
 163		goto drop;
 164	}
 165
 166	optlen = guehdr->hlen << 2;
 167	len += optlen;
 168
 169	if (!pskb_may_pull(skb, len))
 170		goto drop;
 171
 172	/* guehdr may change after pull */
 173	guehdr = (struct guehdr *)&udp_hdr(skb)[1];
 174
 175	if (validate_gue_flags(guehdr, optlen))
 176		goto drop;
 177
 178	hdrlen = sizeof(struct guehdr) + optlen;
 179
 180	if (fou->family == AF_INET)
 181		ip_hdr(skb)->tot_len = htons(ntohs(ip_hdr(skb)->tot_len) - len);
 182	else
 183		ipv6_hdr(skb)->payload_len =
 184		    htons(ntohs(ipv6_hdr(skb)->payload_len) - len);
 185
 186	/* Pull csum through the guehdr now . This can be used if
 187	 * there is a remote checksum offload.
 188	 */
 189	skb_postpull_rcsum(skb, udp_hdr(skb), len);
 190
 191	data = &guehdr[1];
 192
 193	if (guehdr->flags & GUE_FLAG_PRIV) {
 194		__be32 flags = *(__be32 *)(data + doffset);
 195
 196		doffset += GUE_LEN_PRIV;
 197
 198		if (flags & GUE_PFLAG_REMCSUM) {
 199			guehdr = gue_remcsum(skb, guehdr, data + doffset,
 200					     hdrlen, guehdr->proto_ctype,
 201					     !!(fou->flags &
 202						FOU_F_REMCSUM_NOPARTIAL));
 203			if (!guehdr)
 204				goto drop;
 205
 206			data = &guehdr[1];
 207
 208			doffset += GUE_PLEN_REMCSUM;
 209		}
 210	}
 211
 212	if (unlikely(guehdr->control))
 213		return gue_control_message(skb, guehdr);
 214
 215	proto_ctype = guehdr->proto_ctype;
 216	__skb_pull(skb, sizeof(struct udphdr) + hdrlen);
 217	skb_reset_transport_header(skb);
 218
 219	if (iptunnel_pull_offloads(skb))
 220		goto drop;
 221
 222	return -proto_ctype;
 223
 224drop:
 225	kfree_skb(skb);
 226	return 0;
 227}
 228
 229static struct sk_buff *fou_gro_receive(struct sock *sk,
 230				       struct list_head *head,
 231				       struct sk_buff *skb)
 232{
 233	u8 proto = fou_from_sock(sk)->protocol;
 234	const struct net_offload **offloads;
 235	const struct net_offload *ops;
 236	struct sk_buff *pp = NULL;
 237
 238	/* We can clear the encap_mark for FOU as we are essentially doing
 239	 * one of two possible things.  We are either adding an L4 tunnel
 240	 * header to the outer L3 tunnel header, or we are are simply
 241	 * treating the GRE tunnel header as though it is a UDP protocol
 242	 * specific header such as VXLAN or GENEVE.
 243	 */
 244	NAPI_GRO_CB(skb)->encap_mark = 0;
 245
 246	/* Flag this frame as already having an outer encap header */
 247	NAPI_GRO_CB(skb)->is_fou = 1;
 248
 249	rcu_read_lock();
 250	offloads = NAPI_GRO_CB(skb)->is_ipv6 ? inet6_offloads : inet_offloads;
 251	ops = rcu_dereference(offloads[proto]);
 252	if (!ops || !ops->callbacks.gro_receive)
 253		goto out_unlock;
 254
 255	pp = call_gro_receive(ops->callbacks.gro_receive, head, skb);
 256
 257out_unlock:
 258	rcu_read_unlock();
 259
 260	return pp;
 261}
 262
 263static int fou_gro_complete(struct sock *sk, struct sk_buff *skb,
 264			    int nhoff)
 265{
 266	const struct net_offload *ops;
 267	u8 proto = fou_from_sock(sk)->protocol;
 268	int err = -ENOSYS;
 269	const struct net_offload **offloads;
 270
 271	rcu_read_lock();
 272	offloads = NAPI_GRO_CB(skb)->is_ipv6 ? inet6_offloads : inet_offloads;
 273	ops = rcu_dereference(offloads[proto]);
 274	if (WARN_ON(!ops || !ops->callbacks.gro_complete))
 275		goto out_unlock;
 276
 277	err = ops->callbacks.gro_complete(skb, nhoff);
 278
 279	skb_set_inner_mac_header(skb, nhoff);
 280
 281out_unlock:
 282	rcu_read_unlock();
 283
 284	return err;
 285}
 286
 287static struct guehdr *gue_gro_remcsum(struct sk_buff *skb, unsigned int off,
 288				      struct guehdr *guehdr, void *data,
 289				      size_t hdrlen, struct gro_remcsum *grc,
 290				      bool nopartial)
 291{
 292	__be16 *pd = data;
 293	size_t start = ntohs(pd[0]);
 294	size_t offset = ntohs(pd[1]);
 295
 296	if (skb->remcsum_offload)
 297		return guehdr;
 298
 299	if (!NAPI_GRO_CB(skb)->csum_valid)
 300		return NULL;
 301
 302	guehdr = skb_gro_remcsum_process(skb, (void *)guehdr, off, hdrlen,
 303					 start, offset, grc, nopartial);
 304
 305	skb->remcsum_offload = 1;
 306
 307	return guehdr;
 308}
 309
 310static struct sk_buff *gue_gro_receive(struct sock *sk,
 311				       struct list_head *head,
 312				       struct sk_buff *skb)
 313{
 314	const struct net_offload **offloads;
 315	const struct net_offload *ops;
 316	struct sk_buff *pp = NULL;
 317	struct sk_buff *p;
 318	struct guehdr *guehdr;
 319	size_t len, optlen, hdrlen, off;
 320	void *data;
 321	u16 doffset = 0;
 322	int flush = 1;
 323	struct fou *fou = fou_from_sock(sk);
 324	struct gro_remcsum grc;
 325	u8 proto;
 326
 327	skb_gro_remcsum_init(&grc);
 328
 329	off = skb_gro_offset(skb);
 330	len = off + sizeof(*guehdr);
 331
 332	guehdr = skb_gro_header_fast(skb, off);
 333	if (skb_gro_header_hard(skb, len)) {
 334		guehdr = skb_gro_header_slow(skb, len, off);
 335		if (unlikely(!guehdr))
 336			goto out;
 337	}
 338
 339	switch (guehdr->version) {
 340	case 0:
 341		break;
 342	case 1:
 343		switch (((struct iphdr *)guehdr)->version) {
 344		case 4:
 345			proto = IPPROTO_IPIP;
 346			break;
 347		case 6:
 348			proto = IPPROTO_IPV6;
 349			break;
 350		default:
 351			goto out;
 352		}
 353		goto next_proto;
 354	default:
 355		goto out;
 356	}
 357
 358	optlen = guehdr->hlen << 2;
 359	len += optlen;
 360
 361	if (skb_gro_header_hard(skb, len)) {
 362		guehdr = skb_gro_header_slow(skb, len, off);
 363		if (unlikely(!guehdr))
 364			goto out;
 365	}
 366
 367	if (unlikely(guehdr->control) || guehdr->version != 0 ||
 368	    validate_gue_flags(guehdr, optlen))
 369		goto out;
 370
 371	hdrlen = sizeof(*guehdr) + optlen;
 372
 373	/* Adjust NAPI_GRO_CB(skb)->csum to account for guehdr,
 374	 * this is needed if there is a remote checkcsum offload.
 375	 */
 376	skb_gro_postpull_rcsum(skb, guehdr, hdrlen);
 377
 378	data = &guehdr[1];
 379
 380	if (guehdr->flags & GUE_FLAG_PRIV) {
 381		__be32 flags = *(__be32 *)(data + doffset);
 382
 383		doffset += GUE_LEN_PRIV;
 384
 385		if (flags & GUE_PFLAG_REMCSUM) {
 386			guehdr = gue_gro_remcsum(skb, off, guehdr,
 387						 data + doffset, hdrlen, &grc,
 388						 !!(fou->flags &
 389						    FOU_F_REMCSUM_NOPARTIAL));
 390
 391			if (!guehdr)
 392				goto out;
 393
 394			data = &guehdr[1];
 395
 396			doffset += GUE_PLEN_REMCSUM;
 397		}
 398	}
 399
 400	skb_gro_pull(skb, hdrlen);
 401
 402	list_for_each_entry(p, head, list) {
 403		const struct guehdr *guehdr2;
 404
 405		if (!NAPI_GRO_CB(p)->same_flow)
 406			continue;
 407
 408		guehdr2 = (struct guehdr *)(p->data + off);
 409
 410		/* Compare base GUE header to be equal (covers
 411		 * hlen, version, proto_ctype, and flags.
 412		 */
 413		if (guehdr->word != guehdr2->word) {
 414			NAPI_GRO_CB(p)->same_flow = 0;
 415			continue;
 416		}
 417
 418		/* Compare optional fields are the same. */
 419		if (guehdr->hlen && memcmp(&guehdr[1], &guehdr2[1],
 420					   guehdr->hlen << 2)) {
 421			NAPI_GRO_CB(p)->same_flow = 0;
 422			continue;
 423		}
 424	}
 425
 426	proto = guehdr->proto_ctype;
 427
 428next_proto:
 429
 430	/* We can clear the encap_mark for GUE as we are essentially doing
 431	 * one of two possible things.  We are either adding an L4 tunnel
 432	 * header to the outer L3 tunnel header, or we are are simply
 433	 * treating the GRE tunnel header as though it is a UDP protocol
 434	 * specific header such as VXLAN or GENEVE.
 435	 */
 436	NAPI_GRO_CB(skb)->encap_mark = 0;
 437
 438	/* Flag this frame as already having an outer encap header */
 439	NAPI_GRO_CB(skb)->is_fou = 1;
 440
 441	rcu_read_lock();
 442	offloads = NAPI_GRO_CB(skb)->is_ipv6 ? inet6_offloads : inet_offloads;
 443	ops = rcu_dereference(offloads[proto]);
 444	if (WARN_ON_ONCE(!ops || !ops->callbacks.gro_receive))
 445		goto out_unlock;
 446
 447	pp = call_gro_receive(ops->callbacks.gro_receive, head, skb);
 448	flush = 0;
 449
 450out_unlock:
 451	rcu_read_unlock();
 452out:
 453	skb_gro_flush_final_remcsum(skb, pp, flush, &grc);
 454
 455	return pp;
 456}
 457
 458static int gue_gro_complete(struct sock *sk, struct sk_buff *skb, int nhoff)
 459{
 460	const struct net_offload **offloads;
 461	struct guehdr *guehdr = (struct guehdr *)(skb->data + nhoff);
 462	const struct net_offload *ops;
 463	unsigned int guehlen = 0;
 464	u8 proto;
 465	int err = -ENOENT;
 466
 467	switch (guehdr->version) {
 468	case 0:
 469		proto = guehdr->proto_ctype;
 470		guehlen = sizeof(*guehdr) + (guehdr->hlen << 2);
 471		break;
 472	case 1:
 473		switch (((struct iphdr *)guehdr)->version) {
 474		case 4:
 475			proto = IPPROTO_IPIP;
 476			break;
 477		case 6:
 478			proto = IPPROTO_IPV6;
 479			break;
 480		default:
 481			return err;
 482		}
 483		break;
 484	default:
 485		return err;
 486	}
 487
 488	rcu_read_lock();
 489	offloads = NAPI_GRO_CB(skb)->is_ipv6 ? inet6_offloads : inet_offloads;
 490	ops = rcu_dereference(offloads[proto]);
 491	if (WARN_ON(!ops || !ops->callbacks.gro_complete))
 492		goto out_unlock;
 493
 494	err = ops->callbacks.gro_complete(skb, nhoff + guehlen);
 495
 496	skb_set_inner_mac_header(skb, nhoff + guehlen);
 497
 498out_unlock:
 499	rcu_read_unlock();
 500	return err;
 501}
 502
 503static bool fou_cfg_cmp(struct fou *fou, struct fou_cfg *cfg)
 504{
 505	struct sock *sk = fou->sock->sk;
 506	struct udp_port_cfg *udp_cfg = &cfg->udp_config;
 507
 508	if (fou->family != udp_cfg->family ||
 509	    fou->port != udp_cfg->local_udp_port ||
 510	    sk->sk_dport != udp_cfg->peer_udp_port ||
 511	    sk->sk_bound_dev_if != udp_cfg->bind_ifindex)
 512		return false;
 513
 514	if (fou->family == AF_INET) {
 515		if (sk->sk_rcv_saddr != udp_cfg->local_ip.s_addr ||
 516		    sk->sk_daddr != udp_cfg->peer_ip.s_addr)
 517			return false;
 518		else
 519			return true;
 520#if IS_ENABLED(CONFIG_IPV6)
 521	} else {
 522		if (ipv6_addr_cmp(&sk->sk_v6_rcv_saddr, &udp_cfg->local_ip6) ||
 523		    ipv6_addr_cmp(&sk->sk_v6_daddr, &udp_cfg->peer_ip6))
 524			return false;
 525		else
 526			return true;
 527#endif
 528	}
 529
 530	return false;
 531}
 532
 533static int fou_add_to_port_list(struct net *net, struct fou *fou,
 534				struct fou_cfg *cfg)
 535{
 536	struct fou_net *fn = net_generic(net, fou_net_id);
 537	struct fou *fout;
 538
 539	mutex_lock(&fn->fou_lock);
 540	list_for_each_entry(fout, &fn->fou_list, list) {
 541		if (fou_cfg_cmp(fout, cfg)) {
 542			mutex_unlock(&fn->fou_lock);
 543			return -EALREADY;
 544		}
 545	}
 546
 547	list_add(&fou->list, &fn->fou_list);
 548	mutex_unlock(&fn->fou_lock);
 549
 550	return 0;
 551}
 552
 553static void fou_release(struct fou *fou)
 554{
 555	struct socket *sock = fou->sock;
 556
 557	list_del(&fou->list);
 558	udp_tunnel_sock_release(sock);
 559
 560	kfree_rcu(fou, rcu);
 561}
 562
 563static int fou_create(struct net *net, struct fou_cfg *cfg,
 564		      struct socket **sockp)
 565{
 566	struct socket *sock = NULL;
 567	struct fou *fou = NULL;
 568	struct sock *sk;
 569	struct udp_tunnel_sock_cfg tunnel_cfg;
 570	int err;
 571
 572	/* Open UDP socket */
 573	err = udp_sock_create(net, &cfg->udp_config, &sock);
 574	if (err < 0)
 575		goto error;
 576
 577	/* Allocate FOU port structure */
 578	fou = kzalloc(sizeof(*fou), GFP_KERNEL);
 579	if (!fou) {
 580		err = -ENOMEM;
 581		goto error;
 582	}
 583
 584	sk = sock->sk;
 585
 586	fou->port = cfg->udp_config.local_udp_port;
 587	fou->family = cfg->udp_config.family;
 588	fou->flags = cfg->flags;
 589	fou->type = cfg->type;
 590	fou->sock = sock;
 591
 592	memset(&tunnel_cfg, 0, sizeof(tunnel_cfg));
 593	tunnel_cfg.encap_type = 1;
 594	tunnel_cfg.sk_user_data = fou;
 595	tunnel_cfg.encap_destroy = NULL;
 596
 597	/* Initial for fou type */
 598	switch (cfg->type) {
 599	case FOU_ENCAP_DIRECT:
 600		tunnel_cfg.encap_rcv = fou_udp_recv;
 601		tunnel_cfg.gro_receive = fou_gro_receive;
 602		tunnel_cfg.gro_complete = fou_gro_complete;
 603		fou->protocol = cfg->protocol;
 604		break;
 605	case FOU_ENCAP_GUE:
 606		tunnel_cfg.encap_rcv = gue_udp_recv;
 607		tunnel_cfg.gro_receive = gue_gro_receive;
 608		tunnel_cfg.gro_complete = gue_gro_complete;
 609		break;
 610	default:
 611		err = -EINVAL;
 612		goto error;
 613	}
 614
 615	setup_udp_tunnel_sock(net, sock, &tunnel_cfg);
 616
 617	sk->sk_allocation = GFP_ATOMIC;
 618
 619	err = fou_add_to_port_list(net, fou, cfg);
 620	if (err)
 621		goto error;
 622
 623	if (sockp)
 624		*sockp = sock;
 625
 626	return 0;
 627
 628error:
 629	kfree(fou);
 630	if (sock)
 631		udp_tunnel_sock_release(sock);
 632
 633	return err;
 634}
 635
 636static int fou_destroy(struct net *net, struct fou_cfg *cfg)
 637{
 638	struct fou_net *fn = net_generic(net, fou_net_id);
 639	int err = -EINVAL;
 640	struct fou *fou;
 641
 642	mutex_lock(&fn->fou_lock);
 643	list_for_each_entry(fou, &fn->fou_list, list) {
 644		if (fou_cfg_cmp(fou, cfg)) {
 645			fou_release(fou);
 646			err = 0;
 647			break;
 648		}
 649	}
 650	mutex_unlock(&fn->fou_lock);
 651
 652	return err;
 653}
 654
 655static struct genl_family fou_nl_family;
 656
 657static const struct nla_policy fou_nl_policy[FOU_ATTR_MAX + 1] = {
 658	[FOU_ATTR_PORT]			= { .type = NLA_U16, },
 659	[FOU_ATTR_AF]			= { .type = NLA_U8, },
 660	[FOU_ATTR_IPPROTO]		= { .type = NLA_U8, },
 661	[FOU_ATTR_TYPE]			= { .type = NLA_U8, },
 662	[FOU_ATTR_REMCSUM_NOPARTIAL]	= { .type = NLA_FLAG, },
 663	[FOU_ATTR_LOCAL_V4]		= { .type = NLA_U32, },
 664	[FOU_ATTR_PEER_V4]		= { .type = NLA_U32, },
 665	[FOU_ATTR_LOCAL_V6]		= { .type = sizeof(struct in6_addr), },
 666	[FOU_ATTR_PEER_V6]		= { .type = sizeof(struct in6_addr), },
 667	[FOU_ATTR_PEER_PORT]		= { .type = NLA_U16, },
 668	[FOU_ATTR_IFINDEX]		= { .type = NLA_S32, },
 669};
 670
 671static int parse_nl_config(struct genl_info *info,
 672			   struct fou_cfg *cfg)
 673{
 674	bool has_local = false, has_peer = false;
 675	struct nlattr *attr;
 676	int ifindex;
 677	__be16 port;
 678
 679	memset(cfg, 0, sizeof(*cfg));
 680
 681	cfg->udp_config.family = AF_INET;
 682
 683	if (info->attrs[FOU_ATTR_AF]) {
 684		u8 family = nla_get_u8(info->attrs[FOU_ATTR_AF]);
 685
 686		switch (family) {
 687		case AF_INET:
 688			break;
 689		case AF_INET6:
 690			cfg->udp_config.ipv6_v6only = 1;
 691			break;
 692		default:
 693			return -EAFNOSUPPORT;
 694		}
 695
 696		cfg->udp_config.family = family;
 697	}
 698
 699	if (info->attrs[FOU_ATTR_PORT]) {
 700		port = nla_get_be16(info->attrs[FOU_ATTR_PORT]);
 701		cfg->udp_config.local_udp_port = port;
 702	}
 703
 704	if (info->attrs[FOU_ATTR_IPPROTO])
 705		cfg->protocol = nla_get_u8(info->attrs[FOU_ATTR_IPPROTO]);
 706
 707	if (info->attrs[FOU_ATTR_TYPE])
 708		cfg->type = nla_get_u8(info->attrs[FOU_ATTR_TYPE]);
 709
 710	if (info->attrs[FOU_ATTR_REMCSUM_NOPARTIAL])
 711		cfg->flags |= FOU_F_REMCSUM_NOPARTIAL;
 712
 713	if (cfg->udp_config.family == AF_INET) {
 714		if (info->attrs[FOU_ATTR_LOCAL_V4]) {
 715			attr = info->attrs[FOU_ATTR_LOCAL_V4];
 716			cfg->udp_config.local_ip.s_addr = nla_get_in_addr(attr);
 717			has_local = true;
 718		}
 719
 720		if (info->attrs[FOU_ATTR_PEER_V4]) {
 721			attr = info->attrs[FOU_ATTR_PEER_V4];
 722			cfg->udp_config.peer_ip.s_addr = nla_get_in_addr(attr);
 723			has_peer = true;
 724		}
 725#if IS_ENABLED(CONFIG_IPV6)
 726	} else {
 727		if (info->attrs[FOU_ATTR_LOCAL_V6]) {
 728			attr = info->attrs[FOU_ATTR_LOCAL_V6];
 729			cfg->udp_config.local_ip6 = nla_get_in6_addr(attr);
 730			has_local = true;
 731		}
 732
 733		if (info->attrs[FOU_ATTR_PEER_V6]) {
 734			attr = info->attrs[FOU_ATTR_PEER_V6];
 735			cfg->udp_config.peer_ip6 = nla_get_in6_addr(attr);
 736			has_peer = true;
 737		}
 738#endif
 739	}
 740
 741	if (has_peer) {
 742		if (info->attrs[FOU_ATTR_PEER_PORT]) {
 743			port = nla_get_be16(info->attrs[FOU_ATTR_PEER_PORT]);
 744			cfg->udp_config.peer_udp_port = port;
 745		} else {
 746			return -EINVAL;
 747		}
 748	}
 749
 750	if (info->attrs[FOU_ATTR_IFINDEX]) {
 751		if (!has_local)
 752			return -EINVAL;
 753
 754		ifindex = nla_get_s32(info->attrs[FOU_ATTR_IFINDEX]);
 755
 756		cfg->udp_config.bind_ifindex = ifindex;
 757	}
 758
 759	return 0;
 760}
 761
 762static int fou_nl_cmd_add_port(struct sk_buff *skb, struct genl_info *info)
 763{
 764	struct net *net = genl_info_net(info);
 765	struct fou_cfg cfg;
 766	int err;
 767
 768	err = parse_nl_config(info, &cfg);
 769	if (err)
 770		return err;
 771
 772	return fou_create(net, &cfg, NULL);
 773}
 774
 775static int fou_nl_cmd_rm_port(struct sk_buff *skb, struct genl_info *info)
 776{
 777	struct net *net = genl_info_net(info);
 778	struct fou_cfg cfg;
 779	int err;
 780
 781	err = parse_nl_config(info, &cfg);
 782	if (err)
 783		return err;
 784
 785	return fou_destroy(net, &cfg);
 786}
 787
 788static int fou_fill_info(struct fou *fou, struct sk_buff *msg)
 789{
 790	struct sock *sk = fou->sock->sk;
 791
 792	if (nla_put_u8(msg, FOU_ATTR_AF, fou->sock->sk->sk_family) ||
 793	    nla_put_be16(msg, FOU_ATTR_PORT, fou->port) ||
 794	    nla_put_be16(msg, FOU_ATTR_PEER_PORT, sk->sk_dport) ||
 795	    nla_put_u8(msg, FOU_ATTR_IPPROTO, fou->protocol) ||
 796	    nla_put_u8(msg, FOU_ATTR_TYPE, fou->type) ||
 797	    nla_put_s32(msg, FOU_ATTR_IFINDEX, sk->sk_bound_dev_if))
 798		return -1;
 799
 800	if (fou->flags & FOU_F_REMCSUM_NOPARTIAL)
 801		if (nla_put_flag(msg, FOU_ATTR_REMCSUM_NOPARTIAL))
 802			return -1;
 803
 804	if (fou->sock->sk->sk_family == AF_INET) {
 805		if (nla_put_in_addr(msg, FOU_ATTR_LOCAL_V4, sk->sk_rcv_saddr))
 806			return -1;
 807
 808		if (nla_put_in_addr(msg, FOU_ATTR_PEER_V4, sk->sk_daddr))
 809			return -1;
 810#if IS_ENABLED(CONFIG_IPV6)
 811	} else {
 812		if (nla_put_in6_addr(msg, FOU_ATTR_LOCAL_V6,
 813				     &sk->sk_v6_rcv_saddr))
 814			return -1;
 815
 816		if (nla_put_in6_addr(msg, FOU_ATTR_PEER_V6, &sk->sk_v6_daddr))
 817			return -1;
 818#endif
 819	}
 820
 821	return 0;
 822}
 823
 824static int fou_dump_info(struct fou *fou, u32 portid, u32 seq,
 825			 u32 flags, struct sk_buff *skb, u8 cmd)
 826{
 827	void *hdr;
 828
 829	hdr = genlmsg_put(skb, portid, seq, &fou_nl_family, flags, cmd);
 830	if (!hdr)
 831		return -ENOMEM;
 832
 833	if (fou_fill_info(fou, skb) < 0)
 834		goto nla_put_failure;
 835
 836	genlmsg_end(skb, hdr);
 837	return 0;
 838
 839nla_put_failure:
 840	genlmsg_cancel(skb, hdr);
 841	return -EMSGSIZE;
 842}
 843
 844static int fou_nl_cmd_get_port(struct sk_buff *skb, struct genl_info *info)
 845{
 846	struct net *net = genl_info_net(info);
 847	struct fou_net *fn = net_generic(net, fou_net_id);
 848	struct sk_buff *msg;
 849	struct fou_cfg cfg;
 850	struct fou *fout;
 851	__be16 port;
 852	u8 family;
 853	int ret;
 854
 855	ret = parse_nl_config(info, &cfg);
 856	if (ret)
 857		return ret;
 858	port = cfg.udp_config.local_udp_port;
 859	if (port == 0)
 860		return -EINVAL;
 861
 862	family = cfg.udp_config.family;
 863	if (family != AF_INET && family != AF_INET6)
 864		return -EINVAL;
 865
 866	msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
 867	if (!msg)
 868		return -ENOMEM;
 869
 870	ret = -ESRCH;
 871	mutex_lock(&fn->fou_lock);
 872	list_for_each_entry(fout, &fn->fou_list, list) {
 873		if (fou_cfg_cmp(fout, &cfg)) {
 874			ret = fou_dump_info(fout, info->snd_portid,
 875					    info->snd_seq, 0, msg,
 876					    info->genlhdr->cmd);
 877			break;
 878		}
 879	}
 880	mutex_unlock(&fn->fou_lock);
 881	if (ret < 0)
 882		goto out_free;
 883
 884	return genlmsg_reply(msg, info);
 885
 886out_free:
 887	nlmsg_free(msg);
 888	return ret;
 889}
 890
 891static int fou_nl_dump(struct sk_buff *skb, struct netlink_callback *cb)
 892{
 893	struct net *net = sock_net(skb->sk);
 894	struct fou_net *fn = net_generic(net, fou_net_id);
 895	struct fou *fout;
 896	int idx = 0, ret;
 897
 898	mutex_lock(&fn->fou_lock);
 899	list_for_each_entry(fout, &fn->fou_list, list) {
 900		if (idx++ < cb->args[0])
 901			continue;
 902		ret = fou_dump_info(fout, NETLINK_CB(cb->skb).portid,
 903				    cb->nlh->nlmsg_seq, NLM_F_MULTI,
 904				    skb, FOU_CMD_GET);
 905		if (ret)
 906			break;
 907	}
 908	mutex_unlock(&fn->fou_lock);
 909
 910	cb->args[0] = idx;
 911	return skb->len;
 912}
 913
 914static const struct genl_ops fou_nl_ops[] = {
 915	{
 916		.cmd = FOU_CMD_ADD,
 917		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
 918		.doit = fou_nl_cmd_add_port,
 919		.flags = GENL_ADMIN_PERM,
 920	},
 921	{
 922		.cmd = FOU_CMD_DEL,
 923		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
 924		.doit = fou_nl_cmd_rm_port,
 925		.flags = GENL_ADMIN_PERM,
 926	},
 927	{
 928		.cmd = FOU_CMD_GET,
 929		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
 930		.doit = fou_nl_cmd_get_port,
 931		.dumpit = fou_nl_dump,
 932	},
 933};
 934
 935static struct genl_family fou_nl_family __ro_after_init = {
 936	.hdrsize	= 0,
 937	.name		= FOU_GENL_NAME,
 938	.version	= FOU_GENL_VERSION,
 939	.maxattr	= FOU_ATTR_MAX,
 940	.policy = fou_nl_policy,
 941	.netnsok	= true,
 942	.module		= THIS_MODULE,
 943	.ops		= fou_nl_ops,
 944	.n_ops		= ARRAY_SIZE(fou_nl_ops),
 945};
 946
 947size_t fou_encap_hlen(struct ip_tunnel_encap *e)
 948{
 949	return sizeof(struct udphdr);
 950}
 951EXPORT_SYMBOL(fou_encap_hlen);
 952
 953size_t gue_encap_hlen(struct ip_tunnel_encap *e)
 954{
 955	size_t len;
 956	bool need_priv = false;
 957
 958	len = sizeof(struct udphdr) + sizeof(struct guehdr);
 959
 960	if (e->flags & TUNNEL_ENCAP_FLAG_REMCSUM) {
 961		len += GUE_PLEN_REMCSUM;
 962		need_priv = true;
 963	}
 964
 965	len += need_priv ? GUE_LEN_PRIV : 0;
 966
 967	return len;
 968}
 969EXPORT_SYMBOL(gue_encap_hlen);
 970
 971int __fou_build_header(struct sk_buff *skb, struct ip_tunnel_encap *e,
 972		       u8 *protocol, __be16 *sport, int type)
 973{
 974	int err;
 975
 976	err = iptunnel_handle_offloads(skb, type);
 977	if (err)
 978		return err;
 979
 980	*sport = e->sport ? : udp_flow_src_port(dev_net(skb->dev),
 981						skb, 0, 0, false);
 982
 983	return 0;
 984}
 985EXPORT_SYMBOL(__fou_build_header);
 986
 987int __gue_build_header(struct sk_buff *skb, struct ip_tunnel_encap *e,
 988		       u8 *protocol, __be16 *sport, int type)
 989{
 990	struct guehdr *guehdr;
 991	size_t hdrlen, optlen = 0;
 992	void *data;
 993	bool need_priv = false;
 994	int err;
 995
 996	if ((e->flags & TUNNEL_ENCAP_FLAG_REMCSUM) &&
 997	    skb->ip_summed == CHECKSUM_PARTIAL) {
 998		optlen += GUE_PLEN_REMCSUM;
 999		type |= SKB_GSO_TUNNEL_REMCSUM;
1000		need_priv = true;
1001	}
1002
1003	optlen += need_priv ? GUE_LEN_PRIV : 0;
1004
1005	err = iptunnel_handle_offloads(skb, type);
1006	if (err)
1007		return err;
1008
1009	/* Get source port (based on flow hash) before skb_push */
1010	*sport = e->sport ? : udp_flow_src_port(dev_net(skb->dev),
1011						skb, 0, 0, false);
1012
1013	hdrlen = sizeof(struct guehdr) + optlen;
1014
1015	skb_push(skb, hdrlen);
1016
1017	guehdr = (struct guehdr *)skb->data;
1018
1019	guehdr->control = 0;
1020	guehdr->version = 0;
1021	guehdr->hlen = optlen >> 2;
1022	guehdr->flags = 0;
1023	guehdr->proto_ctype = *protocol;
1024
1025	data = &guehdr[1];
1026
1027	if (need_priv) {
1028		__be32 *flags = data;
1029
1030		guehdr->flags |= GUE_FLAG_PRIV;
1031		*flags = 0;
1032		data += GUE_LEN_PRIV;
1033
1034		if (type & SKB_GSO_TUNNEL_REMCSUM) {
1035			u16 csum_start = skb_checksum_start_offset(skb);
1036			__be16 *pd = data;
1037
1038			if (csum_start < hdrlen)
1039				return -EINVAL;
1040
1041			csum_start -= hdrlen;
1042			pd[0] = htons(csum_start);
1043			pd[1] = htons(csum_start + skb->csum_offset);
1044
1045			if (!skb_is_gso(skb)) {
1046				skb->ip_summed = CHECKSUM_NONE;
1047				skb->encapsulation = 0;
1048			}
1049
1050			*flags |= GUE_PFLAG_REMCSUM;
1051			data += GUE_PLEN_REMCSUM;
1052		}
1053
1054	}
1055
1056	return 0;
1057}
1058EXPORT_SYMBOL(__gue_build_header);
1059
1060#ifdef CONFIG_NET_FOU_IP_TUNNELS
1061
1062static void fou_build_udp(struct sk_buff *skb, struct ip_tunnel_encap *e,
1063			  struct flowi4 *fl4, u8 *protocol, __be16 sport)
1064{
1065	struct udphdr *uh;
1066
1067	skb_push(skb, sizeof(struct udphdr));
1068	skb_reset_transport_header(skb);
1069
1070	uh = udp_hdr(skb);
1071
1072	uh->dest = e->dport;
1073	uh->source = sport;
1074	uh->len = htons(skb->len);
1075	udp_set_csum(!(e->flags & TUNNEL_ENCAP_FLAG_CSUM), skb,
1076		     fl4->saddr, fl4->daddr, skb->len);
1077
1078	*protocol = IPPROTO_UDP;
1079}
1080
1081static int fou_build_header(struct sk_buff *skb, struct ip_tunnel_encap *e,
1082			    u8 *protocol, struct flowi4 *fl4)
1083{
1084	int type = e->flags & TUNNEL_ENCAP_FLAG_CSUM ? SKB_GSO_UDP_TUNNEL_CSUM :
1085						       SKB_GSO_UDP_TUNNEL;
1086	__be16 sport;
1087	int err;
1088
1089	err = __fou_build_header(skb, e, protocol, &sport, type);
1090	if (err)
1091		return err;
1092
1093	fou_build_udp(skb, e, fl4, protocol, sport);
1094
1095	return 0;
1096}
1097
1098static int gue_build_header(struct sk_buff *skb, struct ip_tunnel_encap *e,
1099			    u8 *protocol, struct flowi4 *fl4)
1100{
1101	int type = e->flags & TUNNEL_ENCAP_FLAG_CSUM ? SKB_GSO_UDP_TUNNEL_CSUM :
1102						       SKB_GSO_UDP_TUNNEL;
1103	__be16 sport;
1104	int err;
1105
1106	err = __gue_build_header(skb, e, protocol, &sport, type);
1107	if (err)
1108		return err;
1109
1110	fou_build_udp(skb, e, fl4, protocol, sport);
1111
1112	return 0;
1113}
1114
1115static int gue_err_proto_handler(int proto, struct sk_buff *skb, u32 info)
1116{
1117	const struct net_protocol *ipprot = rcu_dereference(inet_protos[proto]);
1118
1119	if (ipprot && ipprot->err_handler) {
1120		if (!ipprot->err_handler(skb, info))
1121			return 0;
1122	}
1123
1124	return -ENOENT;
1125}
1126
1127static int gue_err(struct sk_buff *skb, u32 info)
1128{
1129	int transport_offset = skb_transport_offset(skb);
1130	struct guehdr *guehdr;
1131	size_t len, optlen;
1132	int ret;
1133
1134	len = sizeof(struct udphdr) + sizeof(struct guehdr);
1135	if (!pskb_may_pull(skb, transport_offset + len))
1136		return -EINVAL;
1137
1138	guehdr = (struct guehdr *)&udp_hdr(skb)[1];
1139
1140	switch (guehdr->version) {
1141	case 0: /* Full GUE header present */
1142		break;
1143	case 1: {
1144		/* Direct encapsulation of IPv4 or IPv6 */
1145		skb_set_transport_header(skb, -(int)sizeof(struct icmphdr));
1146
1147		switch (((struct iphdr *)guehdr)->version) {
1148		case 4:
1149			ret = gue_err_proto_handler(IPPROTO_IPIP, skb, info);
1150			goto out;
1151#if IS_ENABLED(CONFIG_IPV6)
1152		case 6:
1153			ret = gue_err_proto_handler(IPPROTO_IPV6, skb, info);
1154			goto out;
1155#endif
1156		default:
1157			ret = -EOPNOTSUPP;
1158			goto out;
1159		}
1160	}
1161	default: /* Undefined version */
1162		return -EOPNOTSUPP;
1163	}
1164
1165	if (guehdr->control)
1166		return -ENOENT;
1167
1168	optlen = guehdr->hlen << 2;
1169
1170	if (!pskb_may_pull(skb, transport_offset + len + optlen))
1171		return -EINVAL;
1172
1173	guehdr = (struct guehdr *)&udp_hdr(skb)[1];
1174	if (validate_gue_flags(guehdr, optlen))
1175		return -EINVAL;
1176
1177	/* Handling exceptions for direct UDP encapsulation in GUE would lead to
1178	 * recursion. Besides, this kind of encapsulation can't even be
1179	 * configured currently. Discard this.
1180	 */
1181	if (guehdr->proto_ctype == IPPROTO_UDP ||
1182	    guehdr->proto_ctype == IPPROTO_UDPLITE)
1183		return -EOPNOTSUPP;
1184
1185	skb_set_transport_header(skb, -(int)sizeof(struct icmphdr));
1186	ret = gue_err_proto_handler(guehdr->proto_ctype, skb, info);
1187
1188out:
1189	skb_set_transport_header(skb, transport_offset);
1190	return ret;
1191}
1192
1193
1194static const struct ip_tunnel_encap_ops fou_iptun_ops = {
1195	.encap_hlen = fou_encap_hlen,
1196	.build_header = fou_build_header,
1197	.err_handler = gue_err,
1198};
1199
1200static const struct ip_tunnel_encap_ops gue_iptun_ops = {
1201	.encap_hlen = gue_encap_hlen,
1202	.build_header = gue_build_header,
1203	.err_handler = gue_err,
1204};
1205
1206static int ip_tunnel_encap_add_fou_ops(void)
1207{
1208	int ret;
1209
1210	ret = ip_tunnel_encap_add_ops(&fou_iptun_ops, TUNNEL_ENCAP_FOU);
1211	if (ret < 0) {
1212		pr_err("can't add fou ops\n");
1213		return ret;
1214	}
1215
1216	ret = ip_tunnel_encap_add_ops(&gue_iptun_ops, TUNNEL_ENCAP_GUE);
1217	if (ret < 0) {
1218		pr_err("can't add gue ops\n");
1219		ip_tunnel_encap_del_ops(&fou_iptun_ops, TUNNEL_ENCAP_FOU);
1220		return ret;
1221	}
1222
1223	return 0;
1224}
1225
1226static void ip_tunnel_encap_del_fou_ops(void)
1227{
1228	ip_tunnel_encap_del_ops(&fou_iptun_ops, TUNNEL_ENCAP_FOU);
1229	ip_tunnel_encap_del_ops(&gue_iptun_ops, TUNNEL_ENCAP_GUE);
1230}
1231
1232#else
1233
1234static int ip_tunnel_encap_add_fou_ops(void)
1235{
1236	return 0;
1237}
1238
1239static void ip_tunnel_encap_del_fou_ops(void)
1240{
1241}
1242
1243#endif
1244
1245static __net_init int fou_init_net(struct net *net)
1246{
1247	struct fou_net *fn = net_generic(net, fou_net_id);
1248
1249	INIT_LIST_HEAD(&fn->fou_list);
1250	mutex_init(&fn->fou_lock);
1251	return 0;
1252}
1253
1254static __net_exit void fou_exit_net(struct net *net)
1255{
1256	struct fou_net *fn = net_generic(net, fou_net_id);
1257	struct fou *fou, *next;
1258
1259	/* Close all the FOU sockets */
1260	mutex_lock(&fn->fou_lock);
1261	list_for_each_entry_safe(fou, next, &fn->fou_list, list)
1262		fou_release(fou);
1263	mutex_unlock(&fn->fou_lock);
1264}
1265
1266static struct pernet_operations fou_net_ops = {
1267	.init = fou_init_net,
1268	.exit = fou_exit_net,
1269	.id   = &fou_net_id,
1270	.size = sizeof(struct fou_net),
1271};
1272
1273static int __init fou_init(void)
1274{
1275	int ret;
1276
1277	ret = register_pernet_device(&fou_net_ops);
1278	if (ret)
1279		goto exit;
1280
1281	ret = genl_register_family(&fou_nl_family);
1282	if (ret < 0)
1283		goto unregister;
1284
1285	ret = ip_tunnel_encap_add_fou_ops();
1286	if (ret == 0)
1287		return 0;
1288
1289	genl_unregister_family(&fou_nl_family);
1290unregister:
1291	unregister_pernet_device(&fou_net_ops);
1292exit:
1293	return ret;
1294}
1295
1296static void __exit fou_fini(void)
1297{
1298	ip_tunnel_encap_del_fou_ops();
1299	genl_unregister_family(&fou_nl_family);
1300	unregister_pernet_device(&fou_net_ops);
1301}
1302
1303module_init(fou_init);
1304module_exit(fou_fini);
1305MODULE_AUTHOR("Tom Herbert <therbert@google.com>");
1306MODULE_LICENSE("GPL");