GNU Linux-libre 4.19.242-gnu1
[releases.git] / drivers / net / geneve.c
1 /*
2  * GENEVE: Generic Network Virtualization Encapsulation
3  *
4  * Copyright (c) 2015 Red Hat, Inc.
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License version 2 as
8  * published by the Free Software Foundation.
9  */
10
11 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
12
13 #include <linux/kernel.h>
14 #include <linux/module.h>
15 #include <linux/etherdevice.h>
16 #include <linux/hash.h>
17 #include <net/dst_metadata.h>
18 #include <net/gro_cells.h>
19 #include <net/rtnetlink.h>
20 #include <net/geneve.h>
21 #include <net/protocol.h>
22
23 #define GENEVE_NETDEV_VER       "0.6"
24
25 #define GENEVE_UDP_PORT         6081
26
27 #define GENEVE_N_VID            (1u << 24)
28 #define GENEVE_VID_MASK         (GENEVE_N_VID - 1)
29
30 #define VNI_HASH_BITS           10
31 #define VNI_HASH_SIZE           (1<<VNI_HASH_BITS)
32
33 static bool log_ecn_error = true;
34 module_param(log_ecn_error, bool, 0644);
35 MODULE_PARM_DESC(log_ecn_error, "Log packets received with corrupted ECN");
36
37 #define GENEVE_VER 0
38 #define GENEVE_BASE_HLEN (sizeof(struct udphdr) + sizeof(struct genevehdr))
39 #define GENEVE_IPV4_HLEN (ETH_HLEN + sizeof(struct iphdr) + GENEVE_BASE_HLEN)
40 #define GENEVE_IPV6_HLEN (ETH_HLEN + sizeof(struct ipv6hdr) + GENEVE_BASE_HLEN)
41
42 /* per-network namespace private data for this module */
43 struct geneve_net {
44         struct list_head        geneve_list;
45         struct list_head        sock_list;
46 };
47
48 static unsigned int geneve_net_id;
49
50 struct geneve_dev_node {
51         struct hlist_node hlist;
52         struct geneve_dev *geneve;
53 };
54
55 /* Pseudo network device */
56 struct geneve_dev {
57         struct geneve_dev_node hlist4;  /* vni hash table for IPv4 socket */
58 #if IS_ENABLED(CONFIG_IPV6)
59         struct geneve_dev_node hlist6;  /* vni hash table for IPv6 socket */
60 #endif
61         struct net         *net;        /* netns for packet i/o */
62         struct net_device  *dev;        /* netdev for geneve tunnel */
63         struct ip_tunnel_info info;
64         struct geneve_sock __rcu *sock4;        /* IPv4 socket used for geneve tunnel */
65 #if IS_ENABLED(CONFIG_IPV6)
66         struct geneve_sock __rcu *sock6;        /* IPv6 socket used for geneve tunnel */
67 #endif
68         struct list_head   next;        /* geneve's per namespace list */
69         struct gro_cells   gro_cells;
70         bool               collect_md;
71         bool               use_udp6_rx_checksums;
72 };
73
74 struct geneve_sock {
75         bool                    collect_md;
76         struct list_head        list;
77         struct socket           *sock;
78         struct rcu_head         rcu;
79         int                     refcnt;
80         struct hlist_head       vni_list[VNI_HASH_SIZE];
81 };
82
83 static inline __u32 geneve_net_vni_hash(u8 vni[3])
84 {
85         __u32 vnid;
86
87         vnid = (vni[0] << 16) | (vni[1] << 8) | vni[2];
88         return hash_32(vnid, VNI_HASH_BITS);
89 }
90
91 static __be64 vni_to_tunnel_id(const __u8 *vni)
92 {
93 #ifdef __BIG_ENDIAN
94         return (vni[0] << 16) | (vni[1] << 8) | vni[2];
95 #else
96         return (__force __be64)(((__force u64)vni[0] << 40) |
97                                 ((__force u64)vni[1] << 48) |
98                                 ((__force u64)vni[2] << 56));
99 #endif
100 }
101
102 /* Convert 64 bit tunnel ID to 24 bit VNI. */
103 static void tunnel_id_to_vni(__be64 tun_id, __u8 *vni)
104 {
105 #ifdef __BIG_ENDIAN
106         vni[0] = (__force __u8)(tun_id >> 16);
107         vni[1] = (__force __u8)(tun_id >> 8);
108         vni[2] = (__force __u8)tun_id;
109 #else
110         vni[0] = (__force __u8)((__force u64)tun_id >> 40);
111         vni[1] = (__force __u8)((__force u64)tun_id >> 48);
112         vni[2] = (__force __u8)((__force u64)tun_id >> 56);
113 #endif
114 }
115
116 static bool eq_tun_id_and_vni(u8 *tun_id, u8 *vni)
117 {
118         return !memcmp(vni, &tun_id[5], 3);
119 }
120
121 static sa_family_t geneve_get_sk_family(struct geneve_sock *gs)
122 {
123         return gs->sock->sk->sk_family;
124 }
125
126 static struct geneve_dev *geneve_lookup(struct geneve_sock *gs,
127                                         __be32 addr, u8 vni[])
128 {
129         struct hlist_head *vni_list_head;
130         struct geneve_dev_node *node;
131         __u32 hash;
132
133         /* Find the device for this VNI */
134         hash = geneve_net_vni_hash(vni);
135         vni_list_head = &gs->vni_list[hash];
136         hlist_for_each_entry_rcu(node, vni_list_head, hlist) {
137                 if (eq_tun_id_and_vni((u8 *)&node->geneve->info.key.tun_id, vni) &&
138                     addr == node->geneve->info.key.u.ipv4.dst)
139                         return node->geneve;
140         }
141         return NULL;
142 }
143
144 #if IS_ENABLED(CONFIG_IPV6)
145 static struct geneve_dev *geneve6_lookup(struct geneve_sock *gs,
146                                          struct in6_addr addr6, u8 vni[])
147 {
148         struct hlist_head *vni_list_head;
149         struct geneve_dev_node *node;
150         __u32 hash;
151
152         /* Find the device for this VNI */
153         hash = geneve_net_vni_hash(vni);
154         vni_list_head = &gs->vni_list[hash];
155         hlist_for_each_entry_rcu(node, vni_list_head, hlist) {
156                 if (eq_tun_id_and_vni((u8 *)&node->geneve->info.key.tun_id, vni) &&
157                     ipv6_addr_equal(&addr6, &node->geneve->info.key.u.ipv6.dst))
158                         return node->geneve;
159         }
160         return NULL;
161 }
162 #endif
163
164 static inline struct genevehdr *geneve_hdr(const struct sk_buff *skb)
165 {
166         return (struct genevehdr *)(udp_hdr(skb) + 1);
167 }
168
169 static struct geneve_dev *geneve_lookup_skb(struct geneve_sock *gs,
170                                             struct sk_buff *skb)
171 {
172         static u8 zero_vni[3];
173         u8 *vni;
174
175         if (geneve_get_sk_family(gs) == AF_INET) {
176                 struct iphdr *iph;
177                 __be32 addr;
178
179                 iph = ip_hdr(skb); /* outer IP header... */
180
181                 if (gs->collect_md) {
182                         vni = zero_vni;
183                         addr = 0;
184                 } else {
185                         vni = geneve_hdr(skb)->vni;
186                         addr = iph->saddr;
187                 }
188
189                 return geneve_lookup(gs, addr, vni);
190 #if IS_ENABLED(CONFIG_IPV6)
191         } else if (geneve_get_sk_family(gs) == AF_INET6) {
192                 static struct in6_addr zero_addr6;
193                 struct ipv6hdr *ip6h;
194                 struct in6_addr addr6;
195
196                 ip6h = ipv6_hdr(skb); /* outer IPv6 header... */
197
198                 if (gs->collect_md) {
199                         vni = zero_vni;
200                         addr6 = zero_addr6;
201                 } else {
202                         vni = geneve_hdr(skb)->vni;
203                         addr6 = ip6h->saddr;
204                 }
205
206                 return geneve6_lookup(gs, addr6, vni);
207 #endif
208         }
209         return NULL;
210 }
211
212 /* geneve receive/decap routine */
213 static void geneve_rx(struct geneve_dev *geneve, struct geneve_sock *gs,
214                       struct sk_buff *skb)
215 {
216         struct genevehdr *gnvh = geneve_hdr(skb);
217         struct metadata_dst *tun_dst = NULL;
218         struct pcpu_sw_netstats *stats;
219         unsigned int len;
220         int err = 0;
221         void *oiph;
222
223         if (ip_tunnel_collect_metadata() || gs->collect_md) {
224                 __be16 flags;
225
226                 flags = TUNNEL_KEY | (gnvh->oam ? TUNNEL_OAM : 0) |
227                         (gnvh->critical ? TUNNEL_CRIT_OPT : 0);
228
229                 tun_dst = udp_tun_rx_dst(skb, geneve_get_sk_family(gs), flags,
230                                          vni_to_tunnel_id(gnvh->vni),
231                                          gnvh->opt_len * 4);
232                 if (!tun_dst) {
233                         geneve->dev->stats.rx_dropped++;
234                         goto drop;
235                 }
236                 /* Update tunnel dst according to Geneve options. */
237                 ip_tunnel_info_opts_set(&tun_dst->u.tun_info,
238                                         gnvh->options, gnvh->opt_len * 4,
239                                         TUNNEL_GENEVE_OPT);
240         } else {
241                 /* Drop packets w/ critical options,
242                  * since we don't support any...
243                  */
244                 if (gnvh->critical) {
245                         geneve->dev->stats.rx_frame_errors++;
246                         geneve->dev->stats.rx_errors++;
247                         goto drop;
248                 }
249         }
250
251         skb_reset_mac_header(skb);
252         skb->protocol = eth_type_trans(skb, geneve->dev);
253         skb_postpull_rcsum(skb, eth_hdr(skb), ETH_HLEN);
254
255         if (tun_dst)
256                 skb_dst_set(skb, &tun_dst->dst);
257
258         /* Ignore packet loops (and multicast echo) */
259         if (ether_addr_equal(eth_hdr(skb)->h_source, geneve->dev->dev_addr)) {
260                 geneve->dev->stats.rx_errors++;
261                 goto drop;
262         }
263
264         oiph = skb_network_header(skb);
265         skb_reset_network_header(skb);
266
267         if (geneve_get_sk_family(gs) == AF_INET)
268                 err = IP_ECN_decapsulate(oiph, skb);
269 #if IS_ENABLED(CONFIG_IPV6)
270         else
271                 err = IP6_ECN_decapsulate(oiph, skb);
272 #endif
273
274         if (unlikely(err)) {
275                 if (log_ecn_error) {
276                         if (geneve_get_sk_family(gs) == AF_INET)
277                                 net_info_ratelimited("non-ECT from %pI4 "
278                                                      "with TOS=%#x\n",
279                                                      &((struct iphdr *)oiph)->saddr,
280                                                      ((struct iphdr *)oiph)->tos);
281 #if IS_ENABLED(CONFIG_IPV6)
282                         else
283                                 net_info_ratelimited("non-ECT from %pI6\n",
284                                                      &((struct ipv6hdr *)oiph)->saddr);
285 #endif
286                 }
287                 if (err > 1) {
288                         ++geneve->dev->stats.rx_frame_errors;
289                         ++geneve->dev->stats.rx_errors;
290                         goto drop;
291                 }
292         }
293
294         len = skb->len;
295         err = gro_cells_receive(&geneve->gro_cells, skb);
296         if (likely(err == NET_RX_SUCCESS)) {
297                 stats = this_cpu_ptr(geneve->dev->tstats);
298                 u64_stats_update_begin(&stats->syncp);
299                 stats->rx_packets++;
300                 stats->rx_bytes += len;
301                 u64_stats_update_end(&stats->syncp);
302         }
303         return;
304 drop:
305         /* Consume bad packet */
306         kfree_skb(skb);
307 }
308
309 /* Setup stats when device is created */
310 static int geneve_init(struct net_device *dev)
311 {
312         struct geneve_dev *geneve = netdev_priv(dev);
313         int err;
314
315         dev->tstats = netdev_alloc_pcpu_stats(struct pcpu_sw_netstats);
316         if (!dev->tstats)
317                 return -ENOMEM;
318
319         err = gro_cells_init(&geneve->gro_cells, dev);
320         if (err) {
321                 free_percpu(dev->tstats);
322                 return err;
323         }
324
325         err = dst_cache_init(&geneve->info.dst_cache, GFP_KERNEL);
326         if (err) {
327                 free_percpu(dev->tstats);
328                 gro_cells_destroy(&geneve->gro_cells);
329                 return err;
330         }
331         return 0;
332 }
333
334 static void geneve_uninit(struct net_device *dev)
335 {
336         struct geneve_dev *geneve = netdev_priv(dev);
337
338         dst_cache_destroy(&geneve->info.dst_cache);
339         gro_cells_destroy(&geneve->gro_cells);
340         free_percpu(dev->tstats);
341 }
342
343 /* Callback from net/ipv4/udp.c to receive packets */
344 static int geneve_udp_encap_recv(struct sock *sk, struct sk_buff *skb)
345 {
346         struct genevehdr *geneveh;
347         struct geneve_dev *geneve;
348         struct geneve_sock *gs;
349         int opts_len;
350
351         /* Need UDP and Geneve header to be present */
352         if (unlikely(!pskb_may_pull(skb, GENEVE_BASE_HLEN)))
353                 goto drop;
354
355         /* Return packets with reserved bits set */
356         geneveh = geneve_hdr(skb);
357         if (unlikely(geneveh->ver != GENEVE_VER))
358                 goto drop;
359
360         if (unlikely(geneveh->proto_type != htons(ETH_P_TEB)))
361                 goto drop;
362
363         gs = rcu_dereference_sk_user_data(sk);
364         if (!gs)
365                 goto drop;
366
367         geneve = geneve_lookup_skb(gs, skb);
368         if (!geneve)
369                 goto drop;
370
371         opts_len = geneveh->opt_len * 4;
372         if (iptunnel_pull_header(skb, GENEVE_BASE_HLEN + opts_len,
373                                  htons(ETH_P_TEB),
374                                  !net_eq(geneve->net, dev_net(geneve->dev)))) {
375                 geneve->dev->stats.rx_dropped++;
376                 goto drop;
377         }
378
379         geneve_rx(geneve, gs, skb);
380         return 0;
381
382 drop:
383         /* Consume bad packet */
384         kfree_skb(skb);
385         return 0;
386 }
387
388 static struct socket *geneve_create_sock(struct net *net, bool ipv6,
389                                          __be16 port, bool ipv6_rx_csum)
390 {
391         struct socket *sock;
392         struct udp_port_cfg udp_conf;
393         int err;
394
395         memset(&udp_conf, 0, sizeof(udp_conf));
396
397         if (ipv6) {
398                 udp_conf.family = AF_INET6;
399                 udp_conf.ipv6_v6only = 1;
400                 udp_conf.use_udp6_rx_checksums = ipv6_rx_csum;
401         } else {
402                 udp_conf.family = AF_INET;
403                 udp_conf.local_ip.s_addr = htonl(INADDR_ANY);
404         }
405
406         udp_conf.local_udp_port = port;
407
408         /* Open UDP socket */
409         err = udp_sock_create(net, &udp_conf, &sock);
410         if (err < 0)
411                 return ERR_PTR(err);
412
413         return sock;
414 }
415
416 static int geneve_hlen(struct genevehdr *gh)
417 {
418         return sizeof(*gh) + gh->opt_len * 4;
419 }
420
421 static struct sk_buff *geneve_gro_receive(struct sock *sk,
422                                           struct list_head *head,
423                                           struct sk_buff *skb)
424 {
425         struct sk_buff *pp = NULL;
426         struct sk_buff *p;
427         struct genevehdr *gh, *gh2;
428         unsigned int hlen, gh_len, off_gnv;
429         const struct packet_offload *ptype;
430         __be16 type;
431         int flush = 1;
432
433         off_gnv = skb_gro_offset(skb);
434         hlen = off_gnv + sizeof(*gh);
435         gh = skb_gro_header_fast(skb, off_gnv);
436         if (skb_gro_header_hard(skb, hlen)) {
437                 gh = skb_gro_header_slow(skb, hlen, off_gnv);
438                 if (unlikely(!gh))
439                         goto out;
440         }
441
442         if (gh->ver != GENEVE_VER || gh->oam)
443                 goto out;
444         gh_len = geneve_hlen(gh);
445
446         hlen = off_gnv + gh_len;
447         if (skb_gro_header_hard(skb, hlen)) {
448                 gh = skb_gro_header_slow(skb, hlen, off_gnv);
449                 if (unlikely(!gh))
450                         goto out;
451         }
452
453         list_for_each_entry(p, head, list) {
454                 if (!NAPI_GRO_CB(p)->same_flow)
455                         continue;
456
457                 gh2 = (struct genevehdr *)(p->data + off_gnv);
458                 if (gh->opt_len != gh2->opt_len ||
459                     memcmp(gh, gh2, gh_len)) {
460                         NAPI_GRO_CB(p)->same_flow = 0;
461                         continue;
462                 }
463         }
464
465         type = gh->proto_type;
466
467         rcu_read_lock();
468         ptype = gro_find_receive_by_type(type);
469         if (!ptype)
470                 goto out_unlock;
471
472         skb_gro_pull(skb, gh_len);
473         skb_gro_postpull_rcsum(skb, gh, gh_len);
474         pp = call_gro_receive(ptype->callbacks.gro_receive, head, skb);
475         flush = 0;
476
477 out_unlock:
478         rcu_read_unlock();
479 out:
480         skb_gro_flush_final(skb, pp, flush);
481
482         return pp;
483 }
484
485 static int geneve_gro_complete(struct sock *sk, struct sk_buff *skb,
486                                int nhoff)
487 {
488         struct genevehdr *gh;
489         struct packet_offload *ptype;
490         __be16 type;
491         int gh_len;
492         int err = -ENOSYS;
493
494         gh = (struct genevehdr *)(skb->data + nhoff);
495         gh_len = geneve_hlen(gh);
496         type = gh->proto_type;
497
498         rcu_read_lock();
499         ptype = gro_find_complete_by_type(type);
500         if (ptype)
501                 err = ptype->callbacks.gro_complete(skb, nhoff + gh_len);
502
503         rcu_read_unlock();
504
505         skb_set_inner_mac_header(skb, nhoff + gh_len);
506
507         return err;
508 }
509
510 /* Create new listen socket if needed */
511 static struct geneve_sock *geneve_socket_create(struct net *net, __be16 port,
512                                                 bool ipv6, bool ipv6_rx_csum)
513 {
514         struct geneve_net *gn = net_generic(net, geneve_net_id);
515         struct geneve_sock *gs;
516         struct socket *sock;
517         struct udp_tunnel_sock_cfg tunnel_cfg;
518         int h;
519
520         gs = kzalloc(sizeof(*gs), GFP_KERNEL);
521         if (!gs)
522                 return ERR_PTR(-ENOMEM);
523
524         sock = geneve_create_sock(net, ipv6, port, ipv6_rx_csum);
525         if (IS_ERR(sock)) {
526                 kfree(gs);
527                 return ERR_CAST(sock);
528         }
529
530         gs->sock = sock;
531         gs->refcnt = 1;
532         for (h = 0; h < VNI_HASH_SIZE; ++h)
533                 INIT_HLIST_HEAD(&gs->vni_list[h]);
534
535         /* Initialize the geneve udp offloads structure */
536         udp_tunnel_notify_add_rx_port(gs->sock, UDP_TUNNEL_TYPE_GENEVE);
537
538         /* Mark socket as an encapsulation socket */
539         memset(&tunnel_cfg, 0, sizeof(tunnel_cfg));
540         tunnel_cfg.sk_user_data = gs;
541         tunnel_cfg.encap_type = 1;
542         tunnel_cfg.gro_receive = geneve_gro_receive;
543         tunnel_cfg.gro_complete = geneve_gro_complete;
544         tunnel_cfg.encap_rcv = geneve_udp_encap_recv;
545         tunnel_cfg.encap_destroy = NULL;
546         setup_udp_tunnel_sock(net, sock, &tunnel_cfg);
547         list_add(&gs->list, &gn->sock_list);
548         return gs;
549 }
550
551 static void __geneve_sock_release(struct geneve_sock *gs)
552 {
553         if (!gs || --gs->refcnt)
554                 return;
555
556         list_del(&gs->list);
557         udp_tunnel_notify_del_rx_port(gs->sock, UDP_TUNNEL_TYPE_GENEVE);
558         udp_tunnel_sock_release(gs->sock);
559         kfree_rcu(gs, rcu);
560 }
561
562 static void geneve_sock_release(struct geneve_dev *geneve)
563 {
564         struct geneve_sock *gs4 = rtnl_dereference(geneve->sock4);
565 #if IS_ENABLED(CONFIG_IPV6)
566         struct geneve_sock *gs6 = rtnl_dereference(geneve->sock6);
567
568         rcu_assign_pointer(geneve->sock6, NULL);
569 #endif
570
571         rcu_assign_pointer(geneve->sock4, NULL);
572         synchronize_net();
573
574         __geneve_sock_release(gs4);
575 #if IS_ENABLED(CONFIG_IPV6)
576         __geneve_sock_release(gs6);
577 #endif
578 }
579
580 static struct geneve_sock *geneve_find_sock(struct geneve_net *gn,
581                                             sa_family_t family,
582                                             __be16 dst_port)
583 {
584         struct geneve_sock *gs;
585
586         list_for_each_entry(gs, &gn->sock_list, list) {
587                 if (inet_sk(gs->sock->sk)->inet_sport == dst_port &&
588                     geneve_get_sk_family(gs) == family) {
589                         return gs;
590                 }
591         }
592         return NULL;
593 }
594
595 static int geneve_sock_add(struct geneve_dev *geneve, bool ipv6)
596 {
597         struct net *net = geneve->net;
598         struct geneve_net *gn = net_generic(net, geneve_net_id);
599         struct geneve_dev_node *node;
600         struct geneve_sock *gs;
601         __u8 vni[3];
602         __u32 hash;
603
604         gs = geneve_find_sock(gn, ipv6 ? AF_INET6 : AF_INET, geneve->info.key.tp_dst);
605         if (gs) {
606                 gs->refcnt++;
607                 goto out;
608         }
609
610         gs = geneve_socket_create(net, geneve->info.key.tp_dst, ipv6,
611                                   geneve->use_udp6_rx_checksums);
612         if (IS_ERR(gs))
613                 return PTR_ERR(gs);
614
615 out:
616         gs->collect_md = geneve->collect_md;
617 #if IS_ENABLED(CONFIG_IPV6)
618         if (ipv6) {
619                 rcu_assign_pointer(geneve->sock6, gs);
620                 node = &geneve->hlist6;
621         } else
622 #endif
623         {
624                 rcu_assign_pointer(geneve->sock4, gs);
625                 node = &geneve->hlist4;
626         }
627         node->geneve = geneve;
628
629         tunnel_id_to_vni(geneve->info.key.tun_id, vni);
630         hash = geneve_net_vni_hash(vni);
631         hlist_add_head_rcu(&node->hlist, &gs->vni_list[hash]);
632         return 0;
633 }
634
635 static int geneve_open(struct net_device *dev)
636 {
637         struct geneve_dev *geneve = netdev_priv(dev);
638         bool metadata = geneve->collect_md;
639         bool ipv4, ipv6;
640         int ret = 0;
641
642         ipv6 = geneve->info.mode & IP_TUNNEL_INFO_IPV6 || metadata;
643         ipv4 = !ipv6 || metadata;
644 #if IS_ENABLED(CONFIG_IPV6)
645         if (ipv6) {
646                 ret = geneve_sock_add(geneve, true);
647                 if (ret < 0 && ret != -EAFNOSUPPORT)
648                         ipv4 = false;
649         }
650 #endif
651         if (ipv4)
652                 ret = geneve_sock_add(geneve, false);
653         if (ret < 0)
654                 geneve_sock_release(geneve);
655
656         return ret;
657 }
658
659 static int geneve_stop(struct net_device *dev)
660 {
661         struct geneve_dev *geneve = netdev_priv(dev);
662
663         hlist_del_init_rcu(&geneve->hlist4.hlist);
664 #if IS_ENABLED(CONFIG_IPV6)
665         hlist_del_init_rcu(&geneve->hlist6.hlist);
666 #endif
667         geneve_sock_release(geneve);
668         return 0;
669 }
670
671 static void geneve_build_header(struct genevehdr *geneveh,
672                                 const struct ip_tunnel_info *info)
673 {
674         geneveh->ver = GENEVE_VER;
675         geneveh->opt_len = info->options_len / 4;
676         geneveh->oam = !!(info->key.tun_flags & TUNNEL_OAM);
677         geneveh->critical = !!(info->key.tun_flags & TUNNEL_CRIT_OPT);
678         geneveh->rsvd1 = 0;
679         tunnel_id_to_vni(info->key.tun_id, geneveh->vni);
680         geneveh->proto_type = htons(ETH_P_TEB);
681         geneveh->rsvd2 = 0;
682
683         if (info->key.tun_flags & TUNNEL_GENEVE_OPT)
684                 ip_tunnel_info_opts_get(geneveh->options, info);
685 }
686
687 static int geneve_build_skb(struct dst_entry *dst, struct sk_buff *skb,
688                             const struct ip_tunnel_info *info,
689                             bool xnet, int ip_hdr_len)
690 {
691         bool udp_sum = !!(info->key.tun_flags & TUNNEL_CSUM);
692         struct genevehdr *gnvh;
693         int min_headroom;
694         int err;
695
696         skb_reset_mac_header(skb);
697         skb_scrub_packet(skb, xnet);
698
699         min_headroom = LL_RESERVED_SPACE(dst->dev) + dst->header_len +
700                        GENEVE_BASE_HLEN + info->options_len + ip_hdr_len;
701         err = skb_cow_head(skb, min_headroom);
702         if (unlikely(err))
703                 goto free_dst;
704
705         err = udp_tunnel_handle_offloads(skb, udp_sum);
706         if (err)
707                 goto free_dst;
708
709         gnvh = __skb_push(skb, sizeof(*gnvh) + info->options_len);
710         geneve_build_header(gnvh, info);
711         skb_set_inner_protocol(skb, htons(ETH_P_TEB));
712         return 0;
713
714 free_dst:
715         dst_release(dst);
716         return err;
717 }
718
719 static struct rtable *geneve_get_v4_rt(struct sk_buff *skb,
720                                        struct net_device *dev,
721                                        struct geneve_sock *gs4,
722                                        struct flowi4 *fl4,
723                                        const struct ip_tunnel_info *info,
724                                        __be16 dport, __be16 sport)
725 {
726         bool use_cache = ip_tunnel_dst_cache_usable(skb, info);
727         struct geneve_dev *geneve = netdev_priv(dev);
728         struct dst_cache *dst_cache;
729         struct rtable *rt = NULL;
730         __u8 tos;
731
732         if (!gs4)
733                 return ERR_PTR(-EIO);
734
735         memset(fl4, 0, sizeof(*fl4));
736         fl4->flowi4_mark = skb->mark;
737         fl4->flowi4_proto = IPPROTO_UDP;
738         fl4->daddr = info->key.u.ipv4.dst;
739         fl4->saddr = info->key.u.ipv4.src;
740         fl4->fl4_dport = dport;
741         fl4->fl4_sport = sport;
742
743         tos = info->key.tos;
744         if ((tos == 1) && !geneve->collect_md) {
745                 tos = ip_tunnel_get_dsfield(ip_hdr(skb), skb);
746                 use_cache = false;
747         }
748         fl4->flowi4_tos = RT_TOS(tos);
749
750         dst_cache = (struct dst_cache *)&info->dst_cache;
751         if (use_cache) {
752                 rt = dst_cache_get_ip4(dst_cache, &fl4->saddr);
753                 if (rt)
754                         return rt;
755         }
756         rt = ip_route_output_key(geneve->net, fl4);
757         if (IS_ERR(rt)) {
758                 netdev_dbg(dev, "no route to %pI4\n", &fl4->daddr);
759                 return ERR_PTR(-ENETUNREACH);
760         }
761         if (rt->dst.dev == dev) { /* is this necessary? */
762                 netdev_dbg(dev, "circular route to %pI4\n", &fl4->daddr);
763                 ip_rt_put(rt);
764                 return ERR_PTR(-ELOOP);
765         }
766         if (use_cache)
767                 dst_cache_set_ip4(dst_cache, &rt->dst, fl4->saddr);
768         return rt;
769 }
770
771 #if IS_ENABLED(CONFIG_IPV6)
772 static struct dst_entry *geneve_get_v6_dst(struct sk_buff *skb,
773                                            struct net_device *dev,
774                                            struct geneve_sock *gs6,
775                                            struct flowi6 *fl6,
776                                            const struct ip_tunnel_info *info,
777                                            __be16 dport, __be16 sport)
778 {
779         bool use_cache = ip_tunnel_dst_cache_usable(skb, info);
780         struct geneve_dev *geneve = netdev_priv(dev);
781         struct dst_entry *dst = NULL;
782         struct dst_cache *dst_cache;
783         __u8 prio;
784
785         if (!gs6)
786                 return ERR_PTR(-EIO);
787
788         memset(fl6, 0, sizeof(*fl6));
789         fl6->flowi6_mark = skb->mark;
790         fl6->flowi6_proto = IPPROTO_UDP;
791         fl6->daddr = info->key.u.ipv6.dst;
792         fl6->saddr = info->key.u.ipv6.src;
793         fl6->fl6_dport = dport;
794         fl6->fl6_sport = sport;
795
796         prio = info->key.tos;
797         if ((prio == 1) && !geneve->collect_md) {
798                 prio = ip_tunnel_get_dsfield(ip_hdr(skb), skb);
799                 use_cache = false;
800         }
801
802         fl6->flowlabel = ip6_make_flowinfo(RT_TOS(prio),
803                                            info->key.label);
804         dst_cache = (struct dst_cache *)&info->dst_cache;
805         if (use_cache) {
806                 dst = dst_cache_get_ip6(dst_cache, &fl6->saddr);
807                 if (dst)
808                         return dst;
809         }
810         dst = ipv6_stub->ipv6_dst_lookup_flow(geneve->net, gs6->sock->sk, fl6,
811                                               NULL);
812         if (IS_ERR(dst)) {
813                 netdev_dbg(dev, "no route to %pI6\n", &fl6->daddr);
814                 return ERR_PTR(-ENETUNREACH);
815         }
816         if (dst->dev == dev) { /* is this necessary? */
817                 netdev_dbg(dev, "circular route to %pI6\n", &fl6->daddr);
818                 dst_release(dst);
819                 return ERR_PTR(-ELOOP);
820         }
821
822         if (use_cache)
823                 dst_cache_set_ip6(dst_cache, dst, &fl6->saddr);
824         return dst;
825 }
826 #endif
827
828 static int geneve_xmit_skb(struct sk_buff *skb, struct net_device *dev,
829                            struct geneve_dev *geneve,
830                            const struct ip_tunnel_info *info)
831 {
832         bool xnet = !net_eq(geneve->net, dev_net(geneve->dev));
833         struct geneve_sock *gs4 = rcu_dereference(geneve->sock4);
834         const struct ip_tunnel_key *key = &info->key;
835         struct rtable *rt;
836         struct flowi4 fl4;
837         __u8 tos, ttl;
838         __be16 sport;
839         __be16 df;
840         int err;
841
842         if (!pskb_inet_may_pull(skb))
843                 return -EINVAL;
844
845         sport = udp_flow_src_port(geneve->net, skb, 1, USHRT_MAX, true);
846         rt = geneve_get_v4_rt(skb, dev, gs4, &fl4, info,
847                               geneve->info.key.tp_dst, sport);
848         if (IS_ERR(rt))
849                 return PTR_ERR(rt);
850
851         skb_tunnel_check_pmtu(skb, &rt->dst,
852                               GENEVE_IPV4_HLEN + info->options_len);
853
854         if (geneve->collect_md) {
855                 tos = ip_tunnel_ecn_encap(key->tos, ip_hdr(skb), skb);
856                 ttl = key->ttl;
857         } else {
858                 tos = ip_tunnel_ecn_encap(fl4.flowi4_tos, ip_hdr(skb), skb);
859                 ttl = key->ttl ? : ip4_dst_hoplimit(&rt->dst);
860         }
861         df = key->tun_flags & TUNNEL_DONT_FRAGMENT ? htons(IP_DF) : 0;
862
863         err = geneve_build_skb(&rt->dst, skb, info, xnet, sizeof(struct iphdr));
864         if (unlikely(err))
865                 return err;
866
867         udp_tunnel_xmit_skb(rt, gs4->sock->sk, skb, fl4.saddr, fl4.daddr,
868                             tos, ttl, df, sport, geneve->info.key.tp_dst,
869                             !net_eq(geneve->net, dev_net(geneve->dev)),
870                             !(info->key.tun_flags & TUNNEL_CSUM));
871         return 0;
872 }
873
874 #if IS_ENABLED(CONFIG_IPV6)
875 static int geneve6_xmit_skb(struct sk_buff *skb, struct net_device *dev,
876                             struct geneve_dev *geneve,
877                             const struct ip_tunnel_info *info)
878 {
879         bool xnet = !net_eq(geneve->net, dev_net(geneve->dev));
880         struct geneve_sock *gs6 = rcu_dereference(geneve->sock6);
881         const struct ip_tunnel_key *key = &info->key;
882         struct dst_entry *dst = NULL;
883         struct flowi6 fl6;
884         __u8 prio, ttl;
885         __be16 sport;
886         int err;
887
888         if (!pskb_inet_may_pull(skb))
889                 return -EINVAL;
890
891         sport = udp_flow_src_port(geneve->net, skb, 1, USHRT_MAX, true);
892         dst = geneve_get_v6_dst(skb, dev, gs6, &fl6, info,
893                                 geneve->info.key.tp_dst, sport);
894         if (IS_ERR(dst))
895                 return PTR_ERR(dst);
896
897         skb_tunnel_check_pmtu(skb, dst, GENEVE_IPV6_HLEN + info->options_len);
898
899         if (geneve->collect_md) {
900                 prio = ip_tunnel_ecn_encap(key->tos, ip_hdr(skb), skb);
901                 ttl = key->ttl;
902         } else {
903                 prio = ip_tunnel_ecn_encap(ip6_tclass(fl6.flowlabel),
904                                            ip_hdr(skb), skb);
905                 ttl = key->ttl ? : ip6_dst_hoplimit(dst);
906         }
907         err = geneve_build_skb(dst, skb, info, xnet, sizeof(struct ipv6hdr));
908         if (unlikely(err))
909                 return err;
910
911         udp_tunnel6_xmit_skb(dst, gs6->sock->sk, skb, dev,
912                              &fl6.saddr, &fl6.daddr, prio, ttl,
913                              info->key.label, sport, geneve->info.key.tp_dst,
914                              !(info->key.tun_flags & TUNNEL_CSUM));
915         return 0;
916 }
917 #endif
918
919 static netdev_tx_t geneve_xmit(struct sk_buff *skb, struct net_device *dev)
920 {
921         struct geneve_dev *geneve = netdev_priv(dev);
922         struct ip_tunnel_info *info = NULL;
923         int err;
924
925         if (geneve->collect_md) {
926                 info = skb_tunnel_info(skb);
927                 if (unlikely(!info || !(info->mode & IP_TUNNEL_INFO_TX))) {
928                         netdev_dbg(dev, "no tunnel metadata\n");
929                         dev_kfree_skb(skb);
930                         dev->stats.tx_dropped++;
931                         return NETDEV_TX_OK;
932                 }
933         } else {
934                 info = &geneve->info;
935         }
936
937         rcu_read_lock();
938 #if IS_ENABLED(CONFIG_IPV6)
939         if (info->mode & IP_TUNNEL_INFO_IPV6)
940                 err = geneve6_xmit_skb(skb, dev, geneve, info);
941         else
942 #endif
943                 err = geneve_xmit_skb(skb, dev, geneve, info);
944         rcu_read_unlock();
945
946         if (likely(!err))
947                 return NETDEV_TX_OK;
948
949         dev_kfree_skb(skb);
950
951         if (err == -ELOOP)
952                 dev->stats.collisions++;
953         else if (err == -ENETUNREACH)
954                 dev->stats.tx_carrier_errors++;
955
956         dev->stats.tx_errors++;
957         return NETDEV_TX_OK;
958 }
959
960 static int geneve_change_mtu(struct net_device *dev, int new_mtu)
961 {
962         if (new_mtu > dev->max_mtu)
963                 new_mtu = dev->max_mtu;
964         else if (new_mtu < dev->min_mtu)
965                 new_mtu = dev->min_mtu;
966
967         dev->mtu = new_mtu;
968         return 0;
969 }
970
971 static int geneve_fill_metadata_dst(struct net_device *dev, struct sk_buff *skb)
972 {
973         struct ip_tunnel_info *info = skb_tunnel_info(skb);
974         struct geneve_dev *geneve = netdev_priv(dev);
975         __be16 sport;
976
977         if (ip_tunnel_info_af(info) == AF_INET) {
978                 struct rtable *rt;
979                 struct flowi4 fl4;
980
981                 struct geneve_sock *gs4 = rcu_dereference(geneve->sock4);
982                 sport = udp_flow_src_port(geneve->net, skb,
983                                           1, USHRT_MAX, true);
984
985                 rt = geneve_get_v4_rt(skb, dev, gs4, &fl4, info,
986                                       geneve->info.key.tp_dst, sport);
987                 if (IS_ERR(rt))
988                         return PTR_ERR(rt);
989
990                 ip_rt_put(rt);
991                 info->key.u.ipv4.src = fl4.saddr;
992 #if IS_ENABLED(CONFIG_IPV6)
993         } else if (ip_tunnel_info_af(info) == AF_INET6) {
994                 struct dst_entry *dst;
995                 struct flowi6 fl6;
996
997                 struct geneve_sock *gs6 = rcu_dereference(geneve->sock6);
998                 sport = udp_flow_src_port(geneve->net, skb,
999                                           1, USHRT_MAX, true);
1000
1001                 dst = geneve_get_v6_dst(skb, dev, gs6, &fl6, info,
1002                                         geneve->info.key.tp_dst, sport);
1003                 if (IS_ERR(dst))
1004                         return PTR_ERR(dst);
1005
1006                 dst_release(dst);
1007                 info->key.u.ipv6.src = fl6.saddr;
1008 #endif
1009         } else {
1010                 return -EINVAL;
1011         }
1012
1013         info->key.tp_src = sport;
1014         info->key.tp_dst = geneve->info.key.tp_dst;
1015         return 0;
1016 }
1017
1018 static const struct net_device_ops geneve_netdev_ops = {
1019         .ndo_init               = geneve_init,
1020         .ndo_uninit             = geneve_uninit,
1021         .ndo_open               = geneve_open,
1022         .ndo_stop               = geneve_stop,
1023         .ndo_start_xmit         = geneve_xmit,
1024         .ndo_get_stats64        = ip_tunnel_get_stats64,
1025         .ndo_change_mtu         = geneve_change_mtu,
1026         .ndo_validate_addr      = eth_validate_addr,
1027         .ndo_set_mac_address    = eth_mac_addr,
1028         .ndo_fill_metadata_dst  = geneve_fill_metadata_dst,
1029 };
1030
1031 static void geneve_get_drvinfo(struct net_device *dev,
1032                                struct ethtool_drvinfo *drvinfo)
1033 {
1034         strlcpy(drvinfo->version, GENEVE_NETDEV_VER, sizeof(drvinfo->version));
1035         strlcpy(drvinfo->driver, "geneve", sizeof(drvinfo->driver));
1036 }
1037
1038 static const struct ethtool_ops geneve_ethtool_ops = {
1039         .get_drvinfo    = geneve_get_drvinfo,
1040         .get_link       = ethtool_op_get_link,
1041 };
1042
1043 /* Info for udev, that this is a virtual tunnel endpoint */
1044 static struct device_type geneve_type = {
1045         .name = "geneve",
1046 };
1047
1048 /* Calls the ndo_udp_tunnel_add of the caller in order to
1049  * supply the listening GENEVE udp ports. Callers are expected
1050  * to implement the ndo_udp_tunnel_add.
1051  */
1052 static void geneve_offload_rx_ports(struct net_device *dev, bool push)
1053 {
1054         struct net *net = dev_net(dev);
1055         struct geneve_net *gn = net_generic(net, geneve_net_id);
1056         struct geneve_sock *gs;
1057
1058         rcu_read_lock();
1059         list_for_each_entry_rcu(gs, &gn->sock_list, list) {
1060                 if (push) {
1061                         udp_tunnel_push_rx_port(dev, gs->sock,
1062                                                 UDP_TUNNEL_TYPE_GENEVE);
1063                 } else {
1064                         udp_tunnel_drop_rx_port(dev, gs->sock,
1065                                                 UDP_TUNNEL_TYPE_GENEVE);
1066                 }
1067         }
1068         rcu_read_unlock();
1069 }
1070
1071 /* Initialize the device structure. */
1072 static void geneve_setup(struct net_device *dev)
1073 {
1074         ether_setup(dev);
1075
1076         dev->netdev_ops = &geneve_netdev_ops;
1077         dev->ethtool_ops = &geneve_ethtool_ops;
1078         dev->needs_free_netdev = true;
1079
1080         SET_NETDEV_DEVTYPE(dev, &geneve_type);
1081
1082         dev->features    |= NETIF_F_LLTX;
1083         dev->features    |= NETIF_F_SG | NETIF_F_HW_CSUM;
1084         dev->features    |= NETIF_F_RXCSUM;
1085         dev->features    |= NETIF_F_GSO_SOFTWARE;
1086
1087         dev->hw_features |= NETIF_F_SG | NETIF_F_HW_CSUM | NETIF_F_RXCSUM;
1088         dev->hw_features |= NETIF_F_GSO_SOFTWARE;
1089
1090         /* MTU range: 68 - (something less than 65535) */
1091         dev->min_mtu = ETH_MIN_MTU;
1092         /* The max_mtu calculation does not take account of GENEVE
1093          * options, to avoid excluding potentially valid
1094          * configurations. This will be further reduced by IPvX hdr size.
1095          */
1096         dev->max_mtu = IP_MAX_MTU - GENEVE_BASE_HLEN - dev->hard_header_len;
1097
1098         netif_keep_dst(dev);
1099         dev->priv_flags &= ~IFF_TX_SKB_SHARING;
1100         dev->priv_flags |= IFF_LIVE_ADDR_CHANGE | IFF_NO_QUEUE;
1101         eth_hw_addr_random(dev);
1102 }
1103
1104 static const struct nla_policy geneve_policy[IFLA_GENEVE_MAX + 1] = {
1105         [IFLA_GENEVE_ID]                = { .type = NLA_U32 },
1106         [IFLA_GENEVE_REMOTE]            = { .len = FIELD_SIZEOF(struct iphdr, daddr) },
1107         [IFLA_GENEVE_REMOTE6]           = { .len = sizeof(struct in6_addr) },
1108         [IFLA_GENEVE_TTL]               = { .type = NLA_U8 },
1109         [IFLA_GENEVE_TOS]               = { .type = NLA_U8 },
1110         [IFLA_GENEVE_LABEL]             = { .type = NLA_U32 },
1111         [IFLA_GENEVE_PORT]              = { .type = NLA_U16 },
1112         [IFLA_GENEVE_COLLECT_METADATA]  = { .type = NLA_FLAG },
1113         [IFLA_GENEVE_UDP_CSUM]          = { .type = NLA_U8 },
1114         [IFLA_GENEVE_UDP_ZERO_CSUM6_TX] = { .type = NLA_U8 },
1115         [IFLA_GENEVE_UDP_ZERO_CSUM6_RX] = { .type = NLA_U8 },
1116 };
1117
1118 static int geneve_validate(struct nlattr *tb[], struct nlattr *data[],
1119                            struct netlink_ext_ack *extack)
1120 {
1121         if (tb[IFLA_ADDRESS]) {
1122                 if (nla_len(tb[IFLA_ADDRESS]) != ETH_ALEN) {
1123                         NL_SET_ERR_MSG_ATTR(extack, tb[IFLA_ADDRESS],
1124                                             "Provided link layer address is not Ethernet");
1125                         return -EINVAL;
1126                 }
1127
1128                 if (!is_valid_ether_addr(nla_data(tb[IFLA_ADDRESS]))) {
1129                         NL_SET_ERR_MSG_ATTR(extack, tb[IFLA_ADDRESS],
1130                                             "Provided Ethernet address is not unicast");
1131                         return -EADDRNOTAVAIL;
1132                 }
1133         }
1134
1135         if (!data) {
1136                 NL_SET_ERR_MSG(extack,
1137                                "Not enough attributes provided to perform the operation");
1138                 return -EINVAL;
1139         }
1140
1141         if (data[IFLA_GENEVE_ID]) {
1142                 __u32 vni =  nla_get_u32(data[IFLA_GENEVE_ID]);
1143
1144                 if (vni >= GENEVE_N_VID) {
1145                         NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_ID],
1146                                             "Geneve ID must be lower than 16777216");
1147                         return -ERANGE;
1148                 }
1149         }
1150
1151         return 0;
1152 }
1153
1154 static struct geneve_dev *geneve_find_dev(struct geneve_net *gn,
1155                                           const struct ip_tunnel_info *info,
1156                                           bool *tun_on_same_port,
1157                                           bool *tun_collect_md)
1158 {
1159         struct geneve_dev *geneve, *t = NULL;
1160
1161         *tun_on_same_port = false;
1162         *tun_collect_md = false;
1163         list_for_each_entry(geneve, &gn->geneve_list, next) {
1164                 if (info->key.tp_dst == geneve->info.key.tp_dst) {
1165                         *tun_collect_md = geneve->collect_md;
1166                         *tun_on_same_port = true;
1167                 }
1168                 if (info->key.tun_id == geneve->info.key.tun_id &&
1169                     info->key.tp_dst == geneve->info.key.tp_dst &&
1170                     !memcmp(&info->key.u, &geneve->info.key.u, sizeof(info->key.u)))
1171                         t = geneve;
1172         }
1173         return t;
1174 }
1175
1176 static bool is_tnl_info_zero(const struct ip_tunnel_info *info)
1177 {
1178         return !(info->key.tun_id || info->key.tun_flags || info->key.tos ||
1179                  info->key.ttl || info->key.label || info->key.tp_src ||
1180                  memchr_inv(&info->key.u, 0, sizeof(info->key.u)));
1181 }
1182
1183 static bool geneve_dst_addr_equal(struct ip_tunnel_info *a,
1184                                   struct ip_tunnel_info *b)
1185 {
1186         if (ip_tunnel_info_af(a) == AF_INET)
1187                 return a->key.u.ipv4.dst == b->key.u.ipv4.dst;
1188         else
1189                 return ipv6_addr_equal(&a->key.u.ipv6.dst, &b->key.u.ipv6.dst);
1190 }
1191
1192 static int geneve_configure(struct net *net, struct net_device *dev,
1193                             struct netlink_ext_ack *extack,
1194                             const struct ip_tunnel_info *info,
1195                             bool metadata, bool ipv6_rx_csum)
1196 {
1197         struct geneve_net *gn = net_generic(net, geneve_net_id);
1198         struct geneve_dev *t, *geneve = netdev_priv(dev);
1199         bool tun_collect_md, tun_on_same_port;
1200         int err, encap_len;
1201
1202         if (metadata && !is_tnl_info_zero(info)) {
1203                 NL_SET_ERR_MSG(extack,
1204                                "Device is externally controlled, so attributes (VNI, Port, and so on) must not be specified");
1205                 return -EINVAL;
1206         }
1207
1208         geneve->net = net;
1209         geneve->dev = dev;
1210
1211         t = geneve_find_dev(gn, info, &tun_on_same_port, &tun_collect_md);
1212         if (t)
1213                 return -EBUSY;
1214
1215         /* make enough headroom for basic scenario */
1216         encap_len = GENEVE_BASE_HLEN + ETH_HLEN;
1217         if (!metadata && ip_tunnel_info_af(info) == AF_INET) {
1218                 encap_len += sizeof(struct iphdr);
1219                 dev->max_mtu -= sizeof(struct iphdr);
1220         } else {
1221                 encap_len += sizeof(struct ipv6hdr);
1222                 dev->max_mtu -= sizeof(struct ipv6hdr);
1223         }
1224         dev->needed_headroom = encap_len + ETH_HLEN;
1225
1226         if (metadata) {
1227                 if (tun_on_same_port) {
1228                         NL_SET_ERR_MSG(extack,
1229                                        "There can be only one externally controlled device on a destination port");
1230                         return -EPERM;
1231                 }
1232         } else {
1233                 if (tun_collect_md) {
1234                         NL_SET_ERR_MSG(extack,
1235                                        "There already exists an externally controlled device on this destination port");
1236                         return -EPERM;
1237                 }
1238         }
1239
1240         dst_cache_reset(&geneve->info.dst_cache);
1241         geneve->info = *info;
1242         geneve->collect_md = metadata;
1243         geneve->use_udp6_rx_checksums = ipv6_rx_csum;
1244
1245         err = register_netdevice(dev);
1246         if (err)
1247                 return err;
1248
1249         list_add(&geneve->next, &gn->geneve_list);
1250         return 0;
1251 }
1252
1253 static void init_tnl_info(struct ip_tunnel_info *info, __u16 dst_port)
1254 {
1255         memset(info, 0, sizeof(*info));
1256         info->key.tp_dst = htons(dst_port);
1257 }
1258
1259 static int geneve_nl2info(struct nlattr *tb[], struct nlattr *data[],
1260                           struct netlink_ext_ack *extack,
1261                           struct ip_tunnel_info *info, bool *metadata,
1262                           bool *use_udp6_rx_checksums, bool changelink)
1263 {
1264         int attrtype;
1265
1266         if (data[IFLA_GENEVE_REMOTE] && data[IFLA_GENEVE_REMOTE6]) {
1267                 NL_SET_ERR_MSG(extack,
1268                                "Cannot specify both IPv4 and IPv6 Remote addresses");
1269                 return -EINVAL;
1270         }
1271
1272         if (data[IFLA_GENEVE_REMOTE]) {
1273                 if (changelink && (ip_tunnel_info_af(info) == AF_INET6)) {
1274                         attrtype = IFLA_GENEVE_REMOTE;
1275                         goto change_notsup;
1276                 }
1277
1278                 info->key.u.ipv4.dst =
1279                         nla_get_in_addr(data[IFLA_GENEVE_REMOTE]);
1280
1281                 if (IN_MULTICAST(ntohl(info->key.u.ipv4.dst))) {
1282                         NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_REMOTE],
1283                                             "Remote IPv4 address cannot be Multicast");
1284                         return -EINVAL;
1285                 }
1286         }
1287
1288         if (data[IFLA_GENEVE_REMOTE6]) {
1289 #if IS_ENABLED(CONFIG_IPV6)
1290                 if (changelink && (ip_tunnel_info_af(info) == AF_INET)) {
1291                         attrtype = IFLA_GENEVE_REMOTE6;
1292                         goto change_notsup;
1293                 }
1294
1295                 info->mode = IP_TUNNEL_INFO_IPV6;
1296                 info->key.u.ipv6.dst =
1297                         nla_get_in6_addr(data[IFLA_GENEVE_REMOTE6]);
1298
1299                 if (ipv6_addr_type(&info->key.u.ipv6.dst) &
1300                     IPV6_ADDR_LINKLOCAL) {
1301                         NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_REMOTE6],
1302                                             "Remote IPv6 address cannot be link-local");
1303                         return -EINVAL;
1304                 }
1305                 if (ipv6_addr_is_multicast(&info->key.u.ipv6.dst)) {
1306                         NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_REMOTE6],
1307                                             "Remote IPv6 address cannot be Multicast");
1308                         return -EINVAL;
1309                 }
1310                 info->key.tun_flags |= TUNNEL_CSUM;
1311                 *use_udp6_rx_checksums = true;
1312 #else
1313                 NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_REMOTE6],
1314                                     "IPv6 support not enabled in the kernel");
1315                 return -EPFNOSUPPORT;
1316 #endif
1317         }
1318
1319         if (data[IFLA_GENEVE_ID]) {
1320                 __u32 vni;
1321                 __u8 tvni[3];
1322                 __be64 tunid;
1323
1324                 vni = nla_get_u32(data[IFLA_GENEVE_ID]);
1325                 tvni[0] = (vni & 0x00ff0000) >> 16;
1326                 tvni[1] = (vni & 0x0000ff00) >> 8;
1327                 tvni[2] =  vni & 0x000000ff;
1328
1329                 tunid = vni_to_tunnel_id(tvni);
1330                 if (changelink && (tunid != info->key.tun_id)) {
1331                         attrtype = IFLA_GENEVE_ID;
1332                         goto change_notsup;
1333                 }
1334                 info->key.tun_id = tunid;
1335         }
1336
1337         if (data[IFLA_GENEVE_TTL])
1338                 info->key.ttl = nla_get_u8(data[IFLA_GENEVE_TTL]);
1339
1340         if (data[IFLA_GENEVE_TOS])
1341                 info->key.tos = nla_get_u8(data[IFLA_GENEVE_TOS]);
1342
1343         if (data[IFLA_GENEVE_LABEL]) {
1344                 info->key.label = nla_get_be32(data[IFLA_GENEVE_LABEL]) &
1345                                   IPV6_FLOWLABEL_MASK;
1346                 if (info->key.label && (!(info->mode & IP_TUNNEL_INFO_IPV6))) {
1347                         NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_LABEL],
1348                                             "Label attribute only applies for IPv6 Geneve devices");
1349                         return -EINVAL;
1350                 }
1351         }
1352
1353         if (data[IFLA_GENEVE_PORT]) {
1354                 if (changelink) {
1355                         attrtype = IFLA_GENEVE_PORT;
1356                         goto change_notsup;
1357                 }
1358                 info->key.tp_dst = nla_get_be16(data[IFLA_GENEVE_PORT]);
1359         }
1360
1361         if (data[IFLA_GENEVE_COLLECT_METADATA]) {
1362                 if (changelink) {
1363                         attrtype = IFLA_GENEVE_COLLECT_METADATA;
1364                         goto change_notsup;
1365                 }
1366                 *metadata = true;
1367         }
1368
1369         if (data[IFLA_GENEVE_UDP_CSUM]) {
1370                 if (changelink) {
1371                         attrtype = IFLA_GENEVE_UDP_CSUM;
1372                         goto change_notsup;
1373                 }
1374                 if (nla_get_u8(data[IFLA_GENEVE_UDP_CSUM]))
1375                         info->key.tun_flags |= TUNNEL_CSUM;
1376         }
1377
1378         if (data[IFLA_GENEVE_UDP_ZERO_CSUM6_TX]) {
1379 #if IS_ENABLED(CONFIG_IPV6)
1380                 if (changelink) {
1381                         attrtype = IFLA_GENEVE_UDP_ZERO_CSUM6_TX;
1382                         goto change_notsup;
1383                 }
1384                 if (nla_get_u8(data[IFLA_GENEVE_UDP_ZERO_CSUM6_TX]))
1385                         info->key.tun_flags &= ~TUNNEL_CSUM;
1386 #else
1387                 NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_UDP_ZERO_CSUM6_TX],
1388                                     "IPv6 support not enabled in the kernel");
1389                 return -EPFNOSUPPORT;
1390 #endif
1391         }
1392
1393         if (data[IFLA_GENEVE_UDP_ZERO_CSUM6_RX]) {
1394 #if IS_ENABLED(CONFIG_IPV6)
1395                 if (changelink) {
1396                         attrtype = IFLA_GENEVE_UDP_ZERO_CSUM6_RX;
1397                         goto change_notsup;
1398                 }
1399                 if (nla_get_u8(data[IFLA_GENEVE_UDP_ZERO_CSUM6_RX]))
1400                         *use_udp6_rx_checksums = false;
1401 #else
1402                 NL_SET_ERR_MSG_ATTR(extack, data[IFLA_GENEVE_UDP_ZERO_CSUM6_RX],
1403                                     "IPv6 support not enabled in the kernel");
1404                 return -EPFNOSUPPORT;
1405 #endif
1406         }
1407
1408         return 0;
1409 change_notsup:
1410         NL_SET_ERR_MSG_ATTR(extack, data[attrtype],
1411                             "Changing VNI, Port, endpoint IP address family, external, and UDP checksum attributes are not supported");
1412         return -EOPNOTSUPP;
1413 }
1414
1415 static void geneve_link_config(struct net_device *dev,
1416                                struct ip_tunnel_info *info, struct nlattr *tb[])
1417 {
1418         struct geneve_dev *geneve = netdev_priv(dev);
1419         int ldev_mtu = 0;
1420
1421         if (tb[IFLA_MTU]) {
1422                 geneve_change_mtu(dev, nla_get_u32(tb[IFLA_MTU]));
1423                 return;
1424         }
1425
1426         switch (ip_tunnel_info_af(info)) {
1427         case AF_INET: {
1428                 struct flowi4 fl4 = { .daddr = info->key.u.ipv4.dst };
1429                 struct rtable *rt = ip_route_output_key(geneve->net, &fl4);
1430
1431                 if (!IS_ERR(rt) && rt->dst.dev) {
1432                         ldev_mtu = rt->dst.dev->mtu - GENEVE_IPV4_HLEN;
1433                         ip_rt_put(rt);
1434                 }
1435                 break;
1436         }
1437 #if IS_ENABLED(CONFIG_IPV6)
1438         case AF_INET6: {
1439                 struct rt6_info *rt;
1440
1441                 if (!__in6_dev_get(dev))
1442                         break;
1443
1444                 rt = rt6_lookup(geneve->net, &info->key.u.ipv6.dst, NULL, 0,
1445                                 NULL, 0);
1446
1447                 if (rt && rt->dst.dev)
1448                         ldev_mtu = rt->dst.dev->mtu - GENEVE_IPV6_HLEN;
1449                 ip6_rt_put(rt);
1450                 break;
1451         }
1452 #endif
1453         }
1454
1455         if (ldev_mtu <= 0)
1456                 return;
1457
1458         geneve_change_mtu(dev, ldev_mtu - info->options_len);
1459 }
1460
1461 static int geneve_newlink(struct net *net, struct net_device *dev,
1462                           struct nlattr *tb[], struct nlattr *data[],
1463                           struct netlink_ext_ack *extack)
1464 {
1465         bool use_udp6_rx_checksums = false;
1466         struct ip_tunnel_info info;
1467         bool metadata = false;
1468         int err;
1469
1470         init_tnl_info(&info, GENEVE_UDP_PORT);
1471         err = geneve_nl2info(tb, data, extack, &info, &metadata,
1472                              &use_udp6_rx_checksums, false);
1473         if (err)
1474                 return err;
1475
1476         err = geneve_configure(net, dev, extack, &info, metadata,
1477                                use_udp6_rx_checksums);
1478         if (err)
1479                 return err;
1480
1481         geneve_link_config(dev, &info, tb);
1482
1483         return 0;
1484 }
1485
1486 /* Quiesces the geneve device data path for both TX and RX.
1487  *
1488  * On transmit geneve checks for non-NULL geneve_sock before it proceeds.
1489  * So, if we set that socket to NULL under RCU and wait for synchronize_net()
1490  * to complete for the existing set of in-flight packets to be transmitted,
1491  * then we would have quiesced the transmit data path. All the future packets
1492  * will get dropped until we unquiesce the data path.
1493  *
1494  * On receive geneve dereference the geneve_sock stashed in the socket. So,
1495  * if we set that to NULL under RCU and wait for synchronize_net() to
1496  * complete, then we would have quiesced the receive data path.
1497  */
1498 static void geneve_quiesce(struct geneve_dev *geneve, struct geneve_sock **gs4,
1499                            struct geneve_sock **gs6)
1500 {
1501         *gs4 = rtnl_dereference(geneve->sock4);
1502         rcu_assign_pointer(geneve->sock4, NULL);
1503         if (*gs4)
1504                 rcu_assign_sk_user_data((*gs4)->sock->sk, NULL);
1505 #if IS_ENABLED(CONFIG_IPV6)
1506         *gs6 = rtnl_dereference(geneve->sock6);
1507         rcu_assign_pointer(geneve->sock6, NULL);
1508         if (*gs6)
1509                 rcu_assign_sk_user_data((*gs6)->sock->sk, NULL);
1510 #else
1511         *gs6 = NULL;
1512 #endif
1513         synchronize_net();
1514 }
1515
1516 /* Resumes the geneve device data path for both TX and RX. */
1517 static void geneve_unquiesce(struct geneve_dev *geneve, struct geneve_sock *gs4,
1518                              struct geneve_sock __maybe_unused *gs6)
1519 {
1520         rcu_assign_pointer(geneve->sock4, gs4);
1521         if (gs4)
1522                 rcu_assign_sk_user_data(gs4->sock->sk, gs4);
1523 #if IS_ENABLED(CONFIG_IPV6)
1524         rcu_assign_pointer(geneve->sock6, gs6);
1525         if (gs6)
1526                 rcu_assign_sk_user_data(gs6->sock->sk, gs6);
1527 #endif
1528         synchronize_net();
1529 }
1530
1531 static int geneve_changelink(struct net_device *dev, struct nlattr *tb[],
1532                              struct nlattr *data[],
1533                              struct netlink_ext_ack *extack)
1534 {
1535         struct geneve_dev *geneve = netdev_priv(dev);
1536         struct geneve_sock *gs4, *gs6;
1537         struct ip_tunnel_info info;
1538         bool metadata;
1539         bool use_udp6_rx_checksums;
1540         int err;
1541
1542         /* If the geneve device is configured for metadata (or externally
1543          * controlled, for example, OVS), then nothing can be changed.
1544          */
1545         if (geneve->collect_md)
1546                 return -EOPNOTSUPP;
1547
1548         /* Start with the existing info. */
1549         memcpy(&info, &geneve->info, sizeof(info));
1550         metadata = geneve->collect_md;
1551         use_udp6_rx_checksums = geneve->use_udp6_rx_checksums;
1552         err = geneve_nl2info(tb, data, extack, &info, &metadata,
1553                              &use_udp6_rx_checksums, true);
1554         if (err)
1555                 return err;
1556
1557         if (!geneve_dst_addr_equal(&geneve->info, &info)) {
1558                 dst_cache_reset(&info.dst_cache);
1559                 geneve_link_config(dev, &info, tb);
1560         }
1561
1562         geneve_quiesce(geneve, &gs4, &gs6);
1563         geneve->info = info;
1564         geneve->collect_md = metadata;
1565         geneve->use_udp6_rx_checksums = use_udp6_rx_checksums;
1566         geneve_unquiesce(geneve, gs4, gs6);
1567
1568         return 0;
1569 }
1570
1571 static void geneve_dellink(struct net_device *dev, struct list_head *head)
1572 {
1573         struct geneve_dev *geneve = netdev_priv(dev);
1574
1575         list_del(&geneve->next);
1576         unregister_netdevice_queue(dev, head);
1577 }
1578
1579 static size_t geneve_get_size(const struct net_device *dev)
1580 {
1581         return nla_total_size(sizeof(__u32)) +  /* IFLA_GENEVE_ID */
1582                 nla_total_size(sizeof(struct in6_addr)) + /* IFLA_GENEVE_REMOTE{6} */
1583                 nla_total_size(sizeof(__u8)) +  /* IFLA_GENEVE_TTL */
1584                 nla_total_size(sizeof(__u8)) +  /* IFLA_GENEVE_TOS */
1585                 nla_total_size(sizeof(__be32)) +  /* IFLA_GENEVE_LABEL */
1586                 nla_total_size(sizeof(__be16)) +  /* IFLA_GENEVE_PORT */
1587                 nla_total_size(0) +      /* IFLA_GENEVE_COLLECT_METADATA */
1588                 nla_total_size(sizeof(__u8)) + /* IFLA_GENEVE_UDP_CSUM */
1589                 nla_total_size(sizeof(__u8)) + /* IFLA_GENEVE_UDP_ZERO_CSUM6_TX */
1590                 nla_total_size(sizeof(__u8)) + /* IFLA_GENEVE_UDP_ZERO_CSUM6_RX */
1591                 0;
1592 }
1593
1594 static int geneve_fill_info(struct sk_buff *skb, const struct net_device *dev)
1595 {
1596         struct geneve_dev *geneve = netdev_priv(dev);
1597         struct ip_tunnel_info *info = &geneve->info;
1598         bool metadata = geneve->collect_md;
1599         __u8 tmp_vni[3];
1600         __u32 vni;
1601
1602         tunnel_id_to_vni(info->key.tun_id, tmp_vni);
1603         vni = (tmp_vni[0] << 16) | (tmp_vni[1] << 8) | tmp_vni[2];
1604         if (nla_put_u32(skb, IFLA_GENEVE_ID, vni))
1605                 goto nla_put_failure;
1606
1607         if (!metadata && ip_tunnel_info_af(info) == AF_INET) {
1608                 if (nla_put_in_addr(skb, IFLA_GENEVE_REMOTE,
1609                                     info->key.u.ipv4.dst))
1610                         goto nla_put_failure;
1611                 if (nla_put_u8(skb, IFLA_GENEVE_UDP_CSUM,
1612                                !!(info->key.tun_flags & TUNNEL_CSUM)))
1613                         goto nla_put_failure;
1614
1615 #if IS_ENABLED(CONFIG_IPV6)
1616         } else if (!metadata) {
1617                 if (nla_put_in6_addr(skb, IFLA_GENEVE_REMOTE6,
1618                                      &info->key.u.ipv6.dst))
1619                         goto nla_put_failure;
1620                 if (nla_put_u8(skb, IFLA_GENEVE_UDP_ZERO_CSUM6_TX,
1621                                !(info->key.tun_flags & TUNNEL_CSUM)))
1622                         goto nla_put_failure;
1623 #endif
1624         }
1625
1626         if (nla_put_u8(skb, IFLA_GENEVE_TTL, info->key.ttl) ||
1627             nla_put_u8(skb, IFLA_GENEVE_TOS, info->key.tos) ||
1628             nla_put_be32(skb, IFLA_GENEVE_LABEL, info->key.label))
1629                 goto nla_put_failure;
1630
1631         if (nla_put_be16(skb, IFLA_GENEVE_PORT, info->key.tp_dst))
1632                 goto nla_put_failure;
1633
1634         if (metadata && nla_put_flag(skb, IFLA_GENEVE_COLLECT_METADATA))
1635                 goto nla_put_failure;
1636
1637 #if IS_ENABLED(CONFIG_IPV6)
1638         if (nla_put_u8(skb, IFLA_GENEVE_UDP_ZERO_CSUM6_RX,
1639                        !geneve->use_udp6_rx_checksums))
1640                 goto nla_put_failure;
1641 #endif
1642
1643         return 0;
1644
1645 nla_put_failure:
1646         return -EMSGSIZE;
1647 }
1648
1649 static struct rtnl_link_ops geneve_link_ops __read_mostly = {
1650         .kind           = "geneve",
1651         .maxtype        = IFLA_GENEVE_MAX,
1652         .policy         = geneve_policy,
1653         .priv_size      = sizeof(struct geneve_dev),
1654         .setup          = geneve_setup,
1655         .validate       = geneve_validate,
1656         .newlink        = geneve_newlink,
1657         .changelink     = geneve_changelink,
1658         .dellink        = geneve_dellink,
1659         .get_size       = geneve_get_size,
1660         .fill_info      = geneve_fill_info,
1661 };
1662
1663 struct net_device *geneve_dev_create_fb(struct net *net, const char *name,
1664                                         u8 name_assign_type, u16 dst_port)
1665 {
1666         struct nlattr *tb[IFLA_MAX + 1];
1667         struct ip_tunnel_info info;
1668         struct net_device *dev;
1669         LIST_HEAD(list_kill);
1670         int err;
1671
1672         memset(tb, 0, sizeof(tb));
1673         dev = rtnl_create_link(net, name, name_assign_type,
1674                                &geneve_link_ops, tb);
1675         if (IS_ERR(dev))
1676                 return dev;
1677
1678         init_tnl_info(&info, dst_port);
1679         err = geneve_configure(net, dev, NULL, &info, true, true);
1680         if (err) {
1681                 free_netdev(dev);
1682                 return ERR_PTR(err);
1683         }
1684
1685         /* openvswitch users expect packet sizes to be unrestricted,
1686          * so set the largest MTU we can.
1687          */
1688         err = geneve_change_mtu(dev, IP_MAX_MTU);
1689         if (err)
1690                 goto err;
1691
1692         err = rtnl_configure_link(dev, NULL);
1693         if (err < 0)
1694                 goto err;
1695
1696         return dev;
1697 err:
1698         geneve_dellink(dev, &list_kill);
1699         unregister_netdevice_many(&list_kill);
1700         return ERR_PTR(err);
1701 }
1702 EXPORT_SYMBOL_GPL(geneve_dev_create_fb);
1703
1704 static int geneve_netdevice_event(struct notifier_block *unused,
1705                                   unsigned long event, void *ptr)
1706 {
1707         struct net_device *dev = netdev_notifier_info_to_dev(ptr);
1708
1709         if (event == NETDEV_UDP_TUNNEL_PUSH_INFO ||
1710             event == NETDEV_UDP_TUNNEL_DROP_INFO) {
1711                 geneve_offload_rx_ports(dev, event == NETDEV_UDP_TUNNEL_PUSH_INFO);
1712         } else if (event == NETDEV_UNREGISTER) {
1713                 geneve_offload_rx_ports(dev, false);
1714         } else if (event == NETDEV_REGISTER) {
1715                 geneve_offload_rx_ports(dev, true);
1716         }
1717
1718         return NOTIFY_DONE;
1719 }
1720
1721 static struct notifier_block geneve_notifier_block __read_mostly = {
1722         .notifier_call = geneve_netdevice_event,
1723 };
1724
1725 static __net_init int geneve_init_net(struct net *net)
1726 {
1727         struct geneve_net *gn = net_generic(net, geneve_net_id);
1728
1729         INIT_LIST_HEAD(&gn->geneve_list);
1730         INIT_LIST_HEAD(&gn->sock_list);
1731         return 0;
1732 }
1733
1734 static void geneve_destroy_tunnels(struct net *net, struct list_head *head)
1735 {
1736         struct geneve_net *gn = net_generic(net, geneve_net_id);
1737         struct geneve_dev *geneve, *next;
1738         struct net_device *dev, *aux;
1739
1740         /* gather any geneve devices that were moved into this ns */
1741         for_each_netdev_safe(net, dev, aux)
1742                 if (dev->rtnl_link_ops == &geneve_link_ops)
1743                         unregister_netdevice_queue(dev, head);
1744
1745         /* now gather any other geneve devices that were created in this ns */
1746         list_for_each_entry_safe(geneve, next, &gn->geneve_list, next) {
1747                 /* If geneve->dev is in the same netns, it was already added
1748                  * to the list by the previous loop.
1749                  */
1750                 if (!net_eq(dev_net(geneve->dev), net))
1751                         unregister_netdevice_queue(geneve->dev, head);
1752         }
1753 }
1754
1755 static void __net_exit geneve_exit_batch_net(struct list_head *net_list)
1756 {
1757         struct net *net;
1758         LIST_HEAD(list);
1759
1760         rtnl_lock();
1761         list_for_each_entry(net, net_list, exit_list)
1762                 geneve_destroy_tunnels(net, &list);
1763
1764         /* unregister the devices gathered above */
1765         unregister_netdevice_many(&list);
1766         rtnl_unlock();
1767
1768         list_for_each_entry(net, net_list, exit_list) {
1769                 const struct geneve_net *gn = net_generic(net, geneve_net_id);
1770
1771                 WARN_ON_ONCE(!list_empty(&gn->sock_list));
1772         }
1773 }
1774
1775 static struct pernet_operations geneve_net_ops = {
1776         .init = geneve_init_net,
1777         .exit_batch = geneve_exit_batch_net,
1778         .id   = &geneve_net_id,
1779         .size = sizeof(struct geneve_net),
1780 };
1781
1782 static int __init geneve_init_module(void)
1783 {
1784         int rc;
1785
1786         rc = register_pernet_subsys(&geneve_net_ops);
1787         if (rc)
1788                 goto out1;
1789
1790         rc = register_netdevice_notifier(&geneve_notifier_block);
1791         if (rc)
1792                 goto out2;
1793
1794         rc = rtnl_link_register(&geneve_link_ops);
1795         if (rc)
1796                 goto out3;
1797
1798         return 0;
1799 out3:
1800         unregister_netdevice_notifier(&geneve_notifier_block);
1801 out2:
1802         unregister_pernet_subsys(&geneve_net_ops);
1803 out1:
1804         return rc;
1805 }
1806 late_initcall(geneve_init_module);
1807
1808 static void __exit geneve_cleanup_module(void)
1809 {
1810         rtnl_link_unregister(&geneve_link_ops);
1811         unregister_netdevice_notifier(&geneve_notifier_block);
1812         unregister_pernet_subsys(&geneve_net_ops);
1813 }
1814 module_exit(geneve_cleanup_module);
1815
1816 MODULE_LICENSE("GPL");
1817 MODULE_VERSION(GENEVE_NETDEV_VER);
1818 MODULE_AUTHOR("John W. Linville <linville@tuxdriver.com>");
1819 MODULE_DESCRIPTION("Interface driver for GENEVE encapsulated traffic");
1820 MODULE_ALIAS_RTNL_LINK("geneve");