drivers/net/gtp.c

   1 // SPDX-License-Identifier: GPL-2.0-or-later
   2 /* GTP according to GSM TS 09.60 / 3GPP TS 29.060
   3  *
   4  * (C) 2012-2014 by sysmocom - s.f.m.c. GmbH
   5  * (C) 2016 by Pablo Neira Ayuso <pablo@netfilter.org>
   6  *
   7  * Author: Harald Welte <hwelte@sysmocom.de>
   8  *         Pablo Neira Ayuso <pablo@netfilter.org>
   9  *         Andreas Schultz <aschultz@travelping.com>
  10  */
  11
  12 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
  13
  14 #include <linux/module.h>
  15 #include <linux/skbuff.h>
  16 #include <linux/udp.h>
  17 #include <linux/rculist.h>
  18 #include <linux/jhash.h>
  19 #include <linux/if_tunnel.h>
  20 #include <linux/net.h>
  21 #include <linux/file.h>
  22 #include <linux/gtp.h>
  23
  24 #include <net/net_namespace.h>
  25 #include <net/protocol.h>
  26 #include <net/ip.h>
  27 #include <net/udp.h>
  28 #include <net/udp_tunnel.h>
  29 #include <net/icmp.h>
  30 #include <net/xfrm.h>
  31 #include <net/genetlink.h>
  32 #include <net/netns/generic.h>
  33 #include <net/gtp.h>
  34
  35 /* An active session for the subscriber. */
  36 struct pdp_ctx {
  37         struct hlist_node       hlist_tid;
  38         struct hlist_node       hlist_addr;
  39
  40         union {
  41                 struct {
  42                         u64     tid;
  43                         u16     flow;
  44                 } v0;
  45                 struct {
  46                         u32     i_tei;
  47                         u32     o_tei;
  48                 } v1;
  49         } u;
  50         u8                      gtp_version;
  51         u16                     af;
  52
  53         struct in_addr          ms_addr_ip4;
  54         struct in_addr          peer_addr_ip4;
  55
  56         struct sock             *sk;
  57         struct net_device       *dev;
  58
  59         atomic_t                tx_seq;
  60         struct rcu_head         rcu_head;
  61 };
  62
  63 /* One instance of the GTP device. */
  64 struct gtp_dev {
  65         struct list_head        list;
  66
  67         struct sock             *sk0;
  68         struct sock             *sk1u;
  69
  70         struct net_device       *dev;
  71
  72         unsigned int            role;
  73         unsigned int            hash_size;
  74         struct hlist_head       *tid_hash;
  75         struct hlist_head       *addr_hash;
  76 };
  77
  78 static unsigned int gtp_net_id __read_mostly;
  79
  80 struct gtp_net {
  81         struct list_head gtp_dev_list;
  82 };
  83
  84 static u32 gtp_h_initval;
  85
  86 static void pdp_context_delete(struct pdp_ctx *pctx);
  87
  88 static inline u32 gtp0_hashfn(u64 tid)
  89 {
  90         u32 *tid32 = (u32 *) &tid;
  91         return jhash_2words(tid32[0], tid32[1], gtp_h_initval);
  92 }
  93
  94 static inline u32 gtp1u_hashfn(u32 tid)
  95 {
  96         return jhash_1word(tid, gtp_h_initval);
  97 }
  98
  99 static inline u32 ipv4_hashfn(__be32 ip)
 100 {
 101         return jhash_1word((__force u32)ip, gtp_h_initval);
 102 }
 103
 104 /* Resolve a PDP context structure based on the 64bit TID. */
 105 static struct pdp_ctx *gtp0_pdp_find(struct gtp_dev *gtp, u64 tid)
 106 {
 107         struct hlist_head *head;
 108         struct pdp_ctx *pdp;
 109
 110         head = &gtp->tid_hash[gtp0_hashfn(tid) % gtp->hash_size];
 111
 112         hlist_for_each_entry_rcu(pdp, head, hlist_tid) {
 113                 if (pdp->gtp_version == GTP_V0 &&
 114                     pdp->u.v0.tid == tid)
 115                         return pdp;
 116         }
 117         return NULL;
 118 }
 119
 120 /* Resolve a PDP context structure based on the 32bit TEI. */
 121 static struct pdp_ctx *gtp1_pdp_find(struct gtp_dev *gtp, u32 tid)
 122 {
 123         struct hlist_head *head;
 124         struct pdp_ctx *pdp;
 125
 126         head = &gtp->tid_hash[gtp1u_hashfn(tid) % gtp->hash_size];
 127
 128         hlist_for_each_entry_rcu(pdp, head, hlist_tid) {
 129                 if (pdp->gtp_version == GTP_V1 &&
 130                     pdp->u.v1.i_tei == tid)
 131                         return pdp;
 132         }
 133         return NULL;
 134 }
 135
 136 /* Resolve a PDP context based on IPv4 address of MS. */
 137 static struct pdp_ctx *ipv4_pdp_find(struct gtp_dev *gtp, __be32 ms_addr)
 138 {
 139         struct hlist_head *head;
 140         struct pdp_ctx *pdp;
 141
 142         head = &gtp->addr_hash[ipv4_hashfn(ms_addr) % gtp->hash_size];
 143
 144         hlist_for_each_entry_rcu(pdp, head, hlist_addr) {
 145                 if (pdp->af == AF_INET &&
 146                     pdp->ms_addr_ip4.s_addr == ms_addr)
 147                         return pdp;
 148         }
 149
 150         return NULL;
 151 }
 152
 153 static bool gtp_check_ms_ipv4(struct sk_buff *skb, struct pdp_ctx *pctx,
 154                                   unsigned int hdrlen, unsigned int role)
 155 {
 156         struct iphdr *iph;
 157
 158         if (!pskb_may_pull(skb, hdrlen + sizeof(struct iphdr)))
 159                 return false;
 160
 161         iph = (struct iphdr *)(skb->data + hdrlen);
 162
 163         if (role == GTP_ROLE_SGSN)
 164                 return iph->daddr == pctx->ms_addr_ip4.s_addr;
 165         else
 166                 return iph->saddr == pctx->ms_addr_ip4.s_addr;
 167 }
 168
 169 /* Check if the inner IP address in this packet is assigned to any
 170  * existing mobile subscriber.
 171  */
 172 static bool gtp_check_ms(struct sk_buff *skb, struct pdp_ctx *pctx,
 173                              unsigned int hdrlen, unsigned int role)
 174 {
 175         switch (ntohs(skb->protocol)) {
 176         case ETH_P_IP:
 177                 return gtp_check_ms_ipv4(skb, pctx, hdrlen, role);
 178         }
 179         return false;
 180 }
 181
 182 static int gtp_rx(struct pdp_ctx *pctx, struct sk_buff *skb,
 183                         unsigned int hdrlen, unsigned int role)
 184 {
 185         if (!gtp_check_ms(skb, pctx, hdrlen, role)) {
 186                 netdev_dbg(pctx->dev, "No PDP ctx for this MS\n");
 187                 return 1;
 188         }
 189
 190         /* Get rid of the GTP + UDP headers. */
 191         if (iptunnel_pull_header(skb, hdrlen, skb->protocol,
 192                                  !net_eq(sock_net(pctx->sk), dev_net(pctx->dev))))
 193                 return -1;
 194
 195         netdev_dbg(pctx->dev, "forwarding packet from GGSN to uplink\n");
 196
 197         /* Now that the UDP and the GTP header have been removed, set up the
 198          * new network header. This is required by the upper layer to
 199          * calculate the transport header.
 200          */
 201         skb_reset_network_header(skb);
 202
 203         skb->dev = pctx->dev;
 204
 205         dev_sw_netstats_rx_add(pctx->dev, skb->len);
 206
 207         netif_rx(skb);
 208         return 0;
 209 }
 210
 211 /* 1 means pass up to the stack, -1 means drop and 0 means decapsulated. */
 212 static int gtp0_udp_encap_recv(struct gtp_dev *gtp, struct sk_buff *skb)
 213 {
 214         unsigned int hdrlen = sizeof(struct udphdr) +
 215                               sizeof(struct gtp0_header);
 216         struct gtp0_header *gtp0;
 217         struct pdp_ctx *pctx;
 218
 219         if (!pskb_may_pull(skb, hdrlen))
 220                 return -1;
 221
 222         gtp0 = (struct gtp0_header *)(skb->data + sizeof(struct udphdr));
 223
 224         if ((gtp0->flags >> 5) != GTP_V0)
 225                 return 1;
 226
 227         if (gtp0->type != GTP_TPDU)
 228                 return 1;
 229
 230         pctx = gtp0_pdp_find(gtp, be64_to_cpu(gtp0->tid));
 231         if (!pctx) {
 232                 netdev_dbg(gtp->dev, "No PDP ctx to decap skb=%p\n", skb);
 233                 return 1;
 234         }
 235
 236         return gtp_rx(pctx, skb, hdrlen, gtp->role);
 237 }
 238
 239 static int gtp1u_udp_encap_recv(struct gtp_dev *gtp, struct sk_buff *skb)
 240 {
 241         unsigned int hdrlen = sizeof(struct udphdr) +
 242                               sizeof(struct gtp1_header);
 243         struct gtp1_header *gtp1;
 244         struct pdp_ctx *pctx;
 245
 246         if (!pskb_may_pull(skb, hdrlen))
 247                 return -1;
 248
 249         gtp1 = (struct gtp1_header *)(skb->data + sizeof(struct udphdr));
 250
 251         if ((gtp1->flags >> 5) != GTP_V1)
 252                 return 1;
 253
 254         if (gtp1->type != GTP_TPDU)
 255                 return 1;
 256
 257         /* From 29.060: "This field shall be present if and only if any one or
 258          * more of the S, PN and E flags are set.".
 259          *
 260          * If any of the bit is set, then the remaining ones also have to be
 261          * set.
 262          */
 263         if (gtp1->flags & GTP1_F_MASK)
 264                 hdrlen += 4;
 265
 266         /* Make sure the header is larger enough, including extensions. */
 267         if (!pskb_may_pull(skb, hdrlen))
 268                 return -1;
 269
 270         gtp1 = (struct gtp1_header *)(skb->data + sizeof(struct udphdr));
 271
 272         pctx = gtp1_pdp_find(gtp, ntohl(gtp1->tid));
 273         if (!pctx) {
 274                 netdev_dbg(gtp->dev, "No PDP ctx to decap skb=%p\n", skb);
 275                 return 1;
 276         }
 277
 278         return gtp_rx(pctx, skb, hdrlen, gtp->role);
 279 }
 280
 281 static void __gtp_encap_destroy(struct sock *sk)
 282 {
 283         struct gtp_dev *gtp;
 284
 285         lock_sock(sk);
 286         gtp = sk->sk_user_data;
 287         if (gtp) {
 288                 if (gtp->sk0 == sk)
 289                         gtp->sk0 = NULL;
 290                 else
 291                         gtp->sk1u = NULL;
 292                 udp_sk(sk)->encap_type = 0;
 293                 rcu_assign_sk_user_data(sk, NULL);
 294                 release_sock(sk);
 295                 sock_put(sk);
 296                 return;
 297         }
 298         release_sock(sk);
 299 }
 300
 301 static void gtp_encap_destroy(struct sock *sk)
 302 {
 303         rtnl_lock();
 304         __gtp_encap_destroy(sk);
 305         rtnl_unlock();
 306 }
 307
 308 static void gtp_encap_disable_sock(struct sock *sk)
 309 {
 310         if (!sk)
 311                 return;
 312
 313         __gtp_encap_destroy(sk);
 314 }
 315
 316 static void gtp_encap_disable(struct gtp_dev *gtp)
 317 {
 318         gtp_encap_disable_sock(gtp->sk0);
 319         gtp_encap_disable_sock(gtp->sk1u);
 320 }
 321
 322 /* UDP encapsulation receive handler. See net/ipv4/udp.c.
 323  * Return codes: 0: success, <0: error, >0: pass up to userspace UDP socket.
 324  */
 325 static int gtp_encap_recv(struct sock *sk, struct sk_buff *skb)
 326 {
 327         struct gtp_dev *gtp;
 328         int ret = 0;
 329
 330         gtp = rcu_dereference_sk_user_data(sk);
 331         if (!gtp)
 332                 return 1;
 333
 334         netdev_dbg(gtp->dev, "encap_recv sk=%p\n", sk);
 335
 336         switch (udp_sk(sk)->encap_type) {
 337         case UDP_ENCAP_GTP0:
 338                 netdev_dbg(gtp->dev, "received GTP0 packet\n");
 339                 ret = gtp0_udp_encap_recv(gtp, skb);
 340                 break;
 341         case UDP_ENCAP_GTP1U:
 342                 netdev_dbg(gtp->dev, "received GTP1U packet\n");
 343                 ret = gtp1u_udp_encap_recv(gtp, skb);
 344                 break;
 345         default:
 346                 ret = -1; /* Shouldn't happen. */
 347         }
 348
 349         switch (ret) {
 350         case 1:
 351                 netdev_dbg(gtp->dev, "pass up to the process\n");
 352                 break;
 353         case 0:
 354                 break;
 355         case -1:
 356                 netdev_dbg(gtp->dev, "GTP packet has been dropped\n");
 357                 kfree_skb(skb);
 358                 ret = 0;
 359                 break;
 360         }
 361
 362         return ret;
 363 }
 364
 365 static int gtp_dev_init(struct net_device *dev)
 366 {
 367         struct gtp_dev *gtp = netdev_priv(dev);
 368
 369         gtp->dev = dev;
 370
 371         dev->tstats = netdev_alloc_pcpu_stats(struct pcpu_sw_netstats);
 372         if (!dev->tstats)
 373                 return -ENOMEM;
 374
 375         return 0;
 376 }
 377
 378 static void gtp_dev_uninit(struct net_device *dev)
 379 {
 380         struct gtp_dev *gtp = netdev_priv(dev);
 381
 382         gtp_encap_disable(gtp);
 383         free_percpu(dev->tstats);
 384 }
 385
 386 static struct rtable *ip4_route_output_gtp(struct flowi4 *fl4,
 387                                            const struct sock *sk,
 388                                            __be32 daddr)
 389 {
 390         memset(fl4, 0, sizeof(*fl4));
 391         fl4->flowi4_oif         = sk->sk_bound_dev_if;
 392         fl4->daddr              = daddr;
 393         fl4->saddr              = inet_sk(sk)->inet_saddr;
 394         fl4->flowi4_tos         = RT_CONN_FLAGS(sk);
 395         fl4->flowi4_proto       = sk->sk_protocol;
 396
 397         return ip_route_output_key(sock_net(sk), fl4);
 398 }
 399
 400 static inline void gtp0_push_header(struct sk_buff *skb, struct pdp_ctx *pctx)
 401 {
 402         int payload_len = skb->len;
 403         struct gtp0_header *gtp0;
 404
 405         gtp0 = skb_push(skb, sizeof(*gtp0));
 406
 407         gtp0->flags     = 0x1e; /* v0, GTP-non-prime. */
 408         gtp0->type      = GTP_TPDU;
 409         gtp0->length    = htons(payload_len);
 410         gtp0->seq       = htons((atomic_inc_return(&pctx->tx_seq) - 1) % 0xffff);
 411         gtp0->flow      = htons(pctx->u.v0.flow);
 412         gtp0->number    = 0xff;
 413         gtp0->spare[0]  = gtp0->spare[1] = gtp0->spare[2] = 0xff;
 414         gtp0->tid       = cpu_to_be64(pctx->u.v0.tid);
 415 }
 416
 417 static inline void gtp1_push_header(struct sk_buff *skb, struct pdp_ctx *pctx)
 418 {
 419         int payload_len = skb->len;
 420         struct gtp1_header *gtp1;
 421
 422         gtp1 = skb_push(skb, sizeof(*gtp1));
 423
 424         /* Bits    8  7  6  5  4  3  2  1
 425          *        +--+--+--+--+--+--+--+--+
 426          *        |version |PT| 0| E| S|PN|
 427          *        +--+--+--+--+--+--+--+--+
 428          *          0  0  1  1  1  0  0  0
 429          */
 430         gtp1->flags     = 0x30; /* v1, GTP-non-prime. */
 431         gtp1->type      = GTP_TPDU;
 432         gtp1->length    = htons(payload_len);
 433         gtp1->tid       = htonl(pctx->u.v1.o_tei);
 434
 435         /* TODO: Suppport for extension header, sequence number and N-PDU.
 436          *       Update the length field if any of them is available.
 437          */
 438 }
 439
 440 struct gtp_pktinfo {
 441         struct sock             *sk;
 442         struct iphdr            *iph;
 443         struct flowi4           fl4;
 444         struct rtable           *rt;
 445         struct pdp_ctx          *pctx;
 446         struct net_device       *dev;
 447         __be16                  gtph_port;
 448 };
 449
 450 static void gtp_push_header(struct sk_buff *skb, struct gtp_pktinfo *pktinfo)
 451 {
 452         switch (pktinfo->pctx->gtp_version) {
 453         case GTP_V0:
 454                 pktinfo->gtph_port = htons(GTP0_PORT);
 455                 gtp0_push_header(skb, pktinfo->pctx);
 456                 break;
 457         case GTP_V1:
 458                 pktinfo->gtph_port = htons(GTP1U_PORT);
 459                 gtp1_push_header(skb, pktinfo->pctx);
 460                 break;
 461         }
 462 }
 463
 464 static inline void gtp_set_pktinfo_ipv4(struct gtp_pktinfo *pktinfo,
 465                                         struct sock *sk, struct iphdr *iph,
 466                                         struct pdp_ctx *pctx, struct rtable *rt,
 467                                         struct flowi4 *fl4,
 468                                         struct net_device *dev)
 469 {
 470         pktinfo->sk     = sk;
 471         pktinfo->iph    = iph;
 472         pktinfo->pctx   = pctx;
 473         pktinfo->rt     = rt;
 474         pktinfo->fl4    = *fl4;
 475         pktinfo->dev    = dev;
 476 }
 477
 478 static int gtp_build_skb_ip4(struct sk_buff *skb, struct net_device *dev,
 479                              struct gtp_pktinfo *pktinfo)
 480 {
 481         struct gtp_dev *gtp = netdev_priv(dev);
 482         struct pdp_ctx *pctx;
 483         struct rtable *rt;
 484         struct flowi4 fl4;
 485         struct iphdr *iph;
 486         __be16 df;
 487         int mtu;
 488
 489         /* Read the IP destination address and resolve the PDP context.
 490          * Prepend PDP header with TEI/TID from PDP ctx.
 491          */
 492         iph = ip_hdr(skb);
 493         if (gtp->role == GTP_ROLE_SGSN)
 494                 pctx = ipv4_pdp_find(gtp, iph->saddr);
 495         else
 496                 pctx = ipv4_pdp_find(gtp, iph->daddr);
 497
 498         if (!pctx) {
 499                 netdev_dbg(dev, "no PDP ctx found for %pI4, skip\n",
 500                            &iph->daddr);
 501                 return -ENOENT;
 502         }
 503         netdev_dbg(dev, "found PDP context %p\n", pctx);
 504
 505         rt = ip4_route_output_gtp(&fl4, pctx->sk, pctx->peer_addr_ip4.s_addr);
 506         if (IS_ERR(rt)) {
 507                 netdev_dbg(dev, "no route to SSGN %pI4\n",
 508                            &pctx->peer_addr_ip4.s_addr);
 509                 dev->stats.tx_carrier_errors++;
 510                 goto err;
 511         }
 512
 513         if (rt->dst.dev == dev) {
 514                 netdev_dbg(dev, "circular route to SSGN %pI4\n",
 515                            &pctx->peer_addr_ip4.s_addr);
 516                 dev->stats.collisions++;
 517                 goto err_rt;
 518         }
 519
 520         skb_dst_drop(skb);
 521
 522         /* This is similar to tnl_update_pmtu(). */
 523         df = iph->frag_off;
 524         if (df) {
 525                 mtu = dst_mtu(&rt->dst) - dev->hard_header_len -
 526                         sizeof(struct iphdr) - sizeof(struct udphdr);
 527                 switch (pctx->gtp_version) {
 528                 case GTP_V0:
 529                         mtu -= sizeof(struct gtp0_header);
 530                         break;
 531                 case GTP_V1:
 532                         mtu -= sizeof(struct gtp1_header);
 533                         break;
 534                 }
 535         } else {
 536                 mtu = dst_mtu(&rt->dst);
 537         }
 538
 539         rt->dst.ops->update_pmtu(&rt->dst, NULL, skb, mtu, false);
 540
 541         if (iph->frag_off & htons(IP_DF) &&
 542             ((!skb_is_gso(skb) && skb->len > mtu) ||
 543              (skb_is_gso(skb) && !skb_gso_validate_network_len(skb, mtu)))) {
 544                 netdev_dbg(dev, "packet too big, fragmentation needed\n");
 545                 icmp_ndo_send(skb, ICMP_DEST_UNREACH, ICMP_FRAG_NEEDED,
 546                               htonl(mtu));
 547                 goto err_rt;
 548         }
 549
 550         gtp_set_pktinfo_ipv4(pktinfo, pctx->sk, iph, pctx, rt, &fl4, dev);
 551         gtp_push_header(skb, pktinfo);
 552
 553         return 0;
 554 err_rt:
 555         ip_rt_put(rt);
 556 err:
 557         return -EBADMSG;
 558 }
 559
 560 static netdev_tx_t gtp_dev_xmit(struct sk_buff *skb, struct net_device *dev)
 561 {
 562         unsigned int proto = ntohs(skb->protocol);
 563         struct gtp_pktinfo pktinfo;
 564         int err;
 565
 566         /* Ensure there is sufficient headroom. */
 567         if (skb_cow_head(skb, dev->needed_headroom))
 568                 goto tx_err;
 569
 570         skb_reset_inner_headers(skb);
 571
 572         /* PDP context lookups in gtp_build_skb_*() need rcu read-side lock. */
 573         rcu_read_lock();
 574         switch (proto) {
 575         case ETH_P_IP:
 576                 err = gtp_build_skb_ip4(skb, dev, &pktinfo);
 577                 break;
 578         default:
 579                 err = -EOPNOTSUPP;
 580                 break;
 581         }
 582         rcu_read_unlock();
 583
 584         if (err < 0)
 585                 goto tx_err;
 586
 587         switch (proto) {
 588         case ETH_P_IP:
 589                 netdev_dbg(pktinfo.dev, "gtp -> IP src: %pI4 dst: %pI4\n",
 590                            &pktinfo.iph->saddr, &pktinfo.iph->daddr);
 591                 udp_tunnel_xmit_skb(pktinfo.rt, pktinfo.sk, skb,
 592                                     pktinfo.fl4.saddr, pktinfo.fl4.daddr,
 593                                     pktinfo.iph->tos,
 594                                     ip4_dst_hoplimit(&pktinfo.rt->dst),
 595                                     0,
 596                                     pktinfo.gtph_port, pktinfo.gtph_port,
 597                                     true, false);
 598                 break;
 599         }
 600
 601         return NETDEV_TX_OK;
 602 tx_err:
 603         dev->stats.tx_errors++;
 604         dev_kfree_skb(skb);
 605         return NETDEV_TX_OK;
 606 }
 607
 608 static const struct net_device_ops gtp_netdev_ops = {
 609         .ndo_init               = gtp_dev_init,
 610         .ndo_uninit             = gtp_dev_uninit,
 611         .ndo_start_xmit         = gtp_dev_xmit,
 612         .ndo_get_stats64        = ip_tunnel_get_stats64,
 613 };
 614
 615 static void gtp_link_setup(struct net_device *dev)
 616 {
 617         dev->netdev_ops         = &gtp_netdev_ops;
 618         dev->needs_free_netdev  = true;
 619
 620         dev->hard_header_len = 0;
 621         dev->addr_len = 0;
 622
 623         /* Zero header length. */
 624         dev->type = ARPHRD_NONE;
 625         dev->flags = IFF_POINTOPOINT | IFF_NOARP | IFF_MULTICAST;
 626
 627         dev->priv_flags |= IFF_NO_QUEUE;
 628         dev->features   |= NETIF_F_LLTX;
 629         netif_keep_dst(dev);
 630
 631         /* Assume largest header, ie. GTPv0. */
 632         dev->needed_headroom    = LL_MAX_HEADER +
 633                                   sizeof(struct iphdr) +
 634                                   sizeof(struct udphdr) +
 635                                   sizeof(struct gtp0_header);
 636 }
 637
 638 static int gtp_hashtable_new(struct gtp_dev *gtp, int hsize);
 639 static int gtp_encap_enable(struct gtp_dev *gtp, struct nlattr *data[]);
 640
 641 static void gtp_destructor(struct net_device *dev)
 642 {
 643         struct gtp_dev *gtp = netdev_priv(dev);
 644
 645         kfree(gtp->addr_hash);
 646         kfree(gtp->tid_hash);
 647 }
 648
 649 static int gtp_newlink(struct net *src_net, struct net_device *dev,
 650                        struct nlattr *tb[], struct nlattr *data[],
 651                        struct netlink_ext_ack *extack)
 652 {
 653         struct gtp_dev *gtp;
 654         struct gtp_net *gn;
 655         int hashsize, err;
 656
 657         if (!data[IFLA_GTP_FD0] && !data[IFLA_GTP_FD1])
 658                 return -EINVAL;
 659
 660         gtp = netdev_priv(dev);
 661
 662         if (!data[IFLA_GTP_PDP_HASHSIZE]) {
 663                 hashsize = 1024;
 664         } else {
 665                 hashsize = nla_get_u32(data[IFLA_GTP_PDP_HASHSIZE]);
 666                 if (!hashsize)
 667                         hashsize = 1024;
 668         }
 669
 670         err = gtp_hashtable_new(gtp, hashsize);
 671         if (err < 0)
 672                 return err;
 673
 674         err = gtp_encap_enable(gtp, data);
 675         if (err < 0)
 676                 goto out_hashtable;
 677
 678         err = register_netdevice(dev);
 679         if (err < 0) {
 680                 netdev_dbg(dev, "failed to register new netdev %d\n", err);
 681                 goto out_encap;
 682         }
 683
 684         gn = net_generic(dev_net(dev), gtp_net_id);
 685         list_add_rcu(&gtp->list, &gn->gtp_dev_list);
 686         dev->priv_destructor = gtp_destructor;
 687
 688         netdev_dbg(dev, "registered new GTP interface\n");
 689
 690         return 0;
 691
 692 out_encap:
 693         gtp_encap_disable(gtp);
 694 out_hashtable:
 695         kfree(gtp->addr_hash);
 696         kfree(gtp->tid_hash);
 697         return err;
 698 }
 699
 700 static void gtp_dellink(struct net_device *dev, struct list_head *head)
 701 {
 702         struct gtp_dev *gtp = netdev_priv(dev);
 703         struct pdp_ctx *pctx;
 704         int i;
 705
 706         for (i = 0; i < gtp->hash_size; i++)
 707                 hlist_for_each_entry_rcu(pctx, &gtp->tid_hash[i], hlist_tid)
 708                         pdp_context_delete(pctx);
 709
 710         list_del_rcu(&gtp->list);
 711         unregister_netdevice_queue(dev, head);
 712 }
 713
 714 static const struct nla_policy gtp_policy[IFLA_GTP_MAX + 1] = {
 715         [IFLA_GTP_FD0]                  = { .type = NLA_U32 },
 716         [IFLA_GTP_FD1]                  = { .type = NLA_U32 },
 717         [IFLA_GTP_PDP_HASHSIZE]         = { .type = NLA_U32 },
 718         [IFLA_GTP_ROLE]                 = { .type = NLA_U32 },
 719 };
 720
 721 static int gtp_validate(struct nlattr *tb[], struct nlattr *data[],
 722                         struct netlink_ext_ack *extack)
 723 {
 724         if (!data)
 725                 return -EINVAL;
 726
 727         return 0;
 728 }
 729
 730 static size_t gtp_get_size(const struct net_device *dev)
 731 {
 732         return nla_total_size(sizeof(__u32));   /* IFLA_GTP_PDP_HASHSIZE */
 733 }
 734
 735 static int gtp_fill_info(struct sk_buff *skb, const struct net_device *dev)
 736 {
 737         struct gtp_dev *gtp = netdev_priv(dev);
 738
 739         if (nla_put_u32(skb, IFLA_GTP_PDP_HASHSIZE, gtp->hash_size))
 740                 goto nla_put_failure;
 741
 742         return 0;
 743
 744 nla_put_failure:
 745         return -EMSGSIZE;
 746 }
 747
 748 static struct rtnl_link_ops gtp_link_ops __read_mostly = {
 749         .kind           = "gtp",
 750         .maxtype        = IFLA_GTP_MAX,
 751         .policy         = gtp_policy,
 752         .priv_size      = sizeof(struct gtp_dev),
 753         .setup          = gtp_link_setup,
 754         .validate       = gtp_validate,
 755         .newlink        = gtp_newlink,
 756         .dellink        = gtp_dellink,
 757         .get_size       = gtp_get_size,
 758         .fill_info      = gtp_fill_info,
 759 };
 760
 761 static int gtp_hashtable_new(struct gtp_dev *gtp, int hsize)
 762 {
 763         int i;
 764
 765         gtp->addr_hash = kmalloc_array(hsize, sizeof(struct hlist_head),
 766                                        GFP_KERNEL | __GFP_NOWARN);
 767         if (gtp->addr_hash == NULL)
 768                 return -ENOMEM;
 769
 770         gtp->tid_hash = kmalloc_array(hsize, sizeof(struct hlist_head),
 771                                       GFP_KERNEL | __GFP_NOWARN);
 772         if (gtp->tid_hash == NULL)
 773                 goto err1;
 774
 775         gtp->hash_size = hsize;
 776
 777         for (i = 0; i < hsize; i++) {
 778                 INIT_HLIST_HEAD(&gtp->addr_hash[i]);
 779                 INIT_HLIST_HEAD(&gtp->tid_hash[i]);
 780         }
 781         return 0;
 782 err1:
 783         kfree(gtp->addr_hash);
 784         return -ENOMEM;
 785 }
 786
 787 static struct sock *gtp_encap_enable_socket(int fd, int type,
 788                                             struct gtp_dev *gtp)
 789 {
 790         struct udp_tunnel_sock_cfg tuncfg = {NULL};
 791         struct socket *sock;
 792         struct sock *sk;
 793         int err;
 794
 795         pr_debug("enable gtp on %d, %d\n", fd, type);
 796
 797         sock = sockfd_lookup(fd, &err);
 798         if (!sock) {
 799                 pr_debug("gtp socket fd=%d not found\n", fd);
 800                 return NULL;
 801         }
 802
 803         sk = sock->sk;
 804         if (sk->sk_protocol != IPPROTO_UDP ||
 805             sk->sk_type != SOCK_DGRAM ||
 806             (sk->sk_family != AF_INET && sk->sk_family != AF_INET6)) {
 807                 pr_debug("socket fd=%d not UDP\n", fd);
 808                 sk = ERR_PTR(-EINVAL);
 809                 goto out_sock;
 810         }
 811
 812         lock_sock(sk);
 813         if (sk->sk_user_data) {
 814                 sk = ERR_PTR(-EBUSY);
 815                 goto out_rel_sock;
 816         }
 817
 818         sock_hold(sk);
 819
 820         tuncfg.sk_user_data = gtp;
 821         tuncfg.encap_type = type;
 822         tuncfg.encap_rcv = gtp_encap_recv;
 823         tuncfg.encap_destroy = gtp_encap_destroy;
 824
 825         setup_udp_tunnel_sock(sock_net(sock->sk), sock, &tuncfg);
 826
 827 out_rel_sock:
 828         release_sock(sock->sk);
 829 out_sock:
 830         sockfd_put(sock);
 831         return sk;
 832 }
 833
 834 static int gtp_encap_enable(struct gtp_dev *gtp, struct nlattr *data[])
 835 {
 836         struct sock *sk1u = NULL;
 837         struct sock *sk0 = NULL;
 838         unsigned int role = GTP_ROLE_GGSN;
 839
 840         if (data[IFLA_GTP_FD0]) {
 841                 u32 fd0 = nla_get_u32(data[IFLA_GTP_FD0]);
 842
 843                 sk0 = gtp_encap_enable_socket(fd0, UDP_ENCAP_GTP0, gtp);
 844                 if (IS_ERR(sk0))
 845                         return PTR_ERR(sk0);
 846         }
 847
 848         if (data[IFLA_GTP_FD1]) {
 849                 u32 fd1 = nla_get_u32(data[IFLA_GTP_FD1]);
 850
 851                 sk1u = gtp_encap_enable_socket(fd1, UDP_ENCAP_GTP1U, gtp);
 852                 if (IS_ERR(sk1u)) {
 853                         gtp_encap_disable_sock(sk0);
 854                         return PTR_ERR(sk1u);
 855                 }
 856         }
 857
 858         if (data[IFLA_GTP_ROLE]) {
 859                 role = nla_get_u32(data[IFLA_GTP_ROLE]);
 860                 if (role > GTP_ROLE_SGSN) {
 861                         gtp_encap_disable_sock(sk0);
 862                         gtp_encap_disable_sock(sk1u);
 863                         return -EINVAL;
 864                 }
 865         }
 866
 867         gtp->sk0 = sk0;
 868         gtp->sk1u = sk1u;
 869         gtp->role = role;
 870
 871         return 0;
 872 }
 873
 874 static struct gtp_dev *gtp_find_dev(struct net *src_net, struct nlattr *nla[])
 875 {
 876         struct gtp_dev *gtp = NULL;
 877         struct net_device *dev;
 878         struct net *net;
 879
 880         /* Examine the link attributes and figure out which network namespace
 881          * we are talking about.
 882          */
 883         if (nla[GTPA_NET_NS_FD])
 884                 net = get_net_ns_by_fd(nla_get_u32(nla[GTPA_NET_NS_FD]));
 885         else
 886                 net = get_net(src_net);
 887
 888         if (IS_ERR(net))
 889                 return NULL;
 890
 891         /* Check if there's an existing gtpX device to configure */
 892         dev = dev_get_by_index_rcu(net, nla_get_u32(nla[GTPA_LINK]));
 893         if (dev && dev->netdev_ops == &gtp_netdev_ops)
 894                 gtp = netdev_priv(dev);
 895
 896         put_net(net);
 897         return gtp;
 898 }
 899
 900 static void ipv4_pdp_fill(struct pdp_ctx *pctx, struct genl_info *info)
 901 {
 902         pctx->gtp_version = nla_get_u32(info->attrs[GTPA_VERSION]);
 903         pctx->af = AF_INET;
 904         pctx->peer_addr_ip4.s_addr =
 905                 nla_get_be32(info->attrs[GTPA_PEER_ADDRESS]);
 906         pctx->ms_addr_ip4.s_addr =
 907                 nla_get_be32(info->attrs[GTPA_MS_ADDRESS]);
 908
 909         switch (pctx->gtp_version) {
 910         case GTP_V0:
 911                 /* According to TS 09.60, sections 7.5.1 and 7.5.2, the flow
 912                  * label needs to be the same for uplink and downlink packets,
 913                  * so let's annotate this.
 914                  */
 915                 pctx->u.v0.tid = nla_get_u64(info->attrs[GTPA_TID]);
 916                 pctx->u.v0.flow = nla_get_u16(info->attrs[GTPA_FLOW]);
 917                 break;
 918         case GTP_V1:
 919                 pctx->u.v1.i_tei = nla_get_u32(info->attrs[GTPA_I_TEI]);
 920                 pctx->u.v1.o_tei = nla_get_u32(info->attrs[GTPA_O_TEI]);
 921                 break;
 922         default:
 923                 break;
 924         }
 925 }
 926
 927 static struct pdp_ctx *gtp_pdp_add(struct gtp_dev *gtp, struct sock *sk,
 928                                    struct genl_info *info)
 929 {
 930         struct pdp_ctx *pctx, *pctx_tid = NULL;
 931         struct net_device *dev = gtp->dev;
 932         u32 hash_ms, hash_tid = 0;
 933         unsigned int version;
 934         bool found = false;
 935         __be32 ms_addr;
 936
 937         ms_addr = nla_get_be32(info->attrs[GTPA_MS_ADDRESS]);
 938         hash_ms = ipv4_hashfn(ms_addr) % gtp->hash_size;
 939         version = nla_get_u32(info->attrs[GTPA_VERSION]);
 940
 941         pctx = ipv4_pdp_find(gtp, ms_addr);
 942         if (pctx)
 943                 found = true;
 944         if (version == GTP_V0)
 945                 pctx_tid = gtp0_pdp_find(gtp,
 946                                          nla_get_u64(info->attrs[GTPA_TID]));
 947         else if (version == GTP_V1)
 948                 pctx_tid = gtp1_pdp_find(gtp,
 949                                          nla_get_u32(info->attrs[GTPA_I_TEI]));
 950         if (pctx_tid)
 951                 found = true;
 952
 953         if (found) {
 954                 if (info->nlhdr->nlmsg_flags & NLM_F_EXCL)
 955                         return ERR_PTR(-EEXIST);
 956                 if (info->nlhdr->nlmsg_flags & NLM_F_REPLACE)
 957                         return ERR_PTR(-EOPNOTSUPP);
 958
 959                 if (pctx && pctx_tid)
 960                         return ERR_PTR(-EEXIST);
 961                 if (!pctx)
 962                         pctx = pctx_tid;
 963
 964                 ipv4_pdp_fill(pctx, info);
 965
 966                 if (pctx->gtp_version == GTP_V0)
 967                         netdev_dbg(dev, "GTPv0-U: update tunnel id = %llx (pdp %p)\n",
 968                                    pctx->u.v0.tid, pctx);
 969                 else if (pctx->gtp_version == GTP_V1)
 970                         netdev_dbg(dev, "GTPv1-U: update tunnel id = %x/%x (pdp %p)\n",
 971                                    pctx->u.v1.i_tei, pctx->u.v1.o_tei, pctx);
 972
 973                 return pctx;
 974
 975         }
 976
 977         pctx = kmalloc(sizeof(*pctx), GFP_ATOMIC);
 978         if (pctx == NULL)
 979                 return ERR_PTR(-ENOMEM);
 980
 981         sock_hold(sk);
 982         pctx->sk = sk;
 983         pctx->dev = gtp->dev;
 984         ipv4_pdp_fill(pctx, info);
 985         atomic_set(&pctx->tx_seq, 0);
 986
 987         switch (pctx->gtp_version) {
 988         case GTP_V0:
 989                 /* TS 09.60: "The flow label identifies unambiguously a GTP
 990                  * flow.". We use the tid for this instead, I cannot find a
 991                  * situation in which this doesn't unambiguosly identify the
 992                  * PDP context.
 993                  */
 994                 hash_tid = gtp0_hashfn(pctx->u.v0.tid) % gtp->hash_size;
 995                 break;
 996         case GTP_V1:
 997                 hash_tid = gtp1u_hashfn(pctx->u.v1.i_tei) % gtp->hash_size;
 998                 break;
 999         }
1000
1001         hlist_add_head_rcu(&pctx->hlist_addr, &gtp->addr_hash[hash_ms]);
1002         hlist_add_head_rcu(&pctx->hlist_tid, &gtp->tid_hash[hash_tid]);
1003
1004         switch (pctx->gtp_version) {
1005         case GTP_V0:
1006                 netdev_dbg(dev, "GTPv0-U: new PDP ctx id=%llx ssgn=%pI4 ms=%pI4 (pdp=%p)\n",
1007                            pctx->u.v0.tid, &pctx->peer_addr_ip4,
1008                            &pctx->ms_addr_ip4, pctx);
1009                 break;
1010         case GTP_V1:
1011                 netdev_dbg(dev, "GTPv1-U: new PDP ctx id=%x/%x ssgn=%pI4 ms=%pI4 (pdp=%p)\n",
1012                            pctx->u.v1.i_tei, pctx->u.v1.o_tei,
1013                            &pctx->peer_addr_ip4, &pctx->ms_addr_ip4, pctx);
1014                 break;
1015         }
1016
1017         return pctx;
1018 }
1019
1020 static void pdp_context_free(struct rcu_head *head)
1021 {
1022         struct pdp_ctx *pctx = container_of(head, struct pdp_ctx, rcu_head);
1023
1024         sock_put(pctx->sk);
1025         kfree(pctx);
1026 }
1027
1028 static void pdp_context_delete(struct pdp_ctx *pctx)
1029 {
1030         hlist_del_rcu(&pctx->hlist_tid);
1031         hlist_del_rcu(&pctx->hlist_addr);
1032         call_rcu(&pctx->rcu_head, pdp_context_free);
1033 }
1034
1035 static int gtp_tunnel_notify(struct pdp_ctx *pctx, u8 cmd, gfp_t allocation);
1036
1037 static int gtp_genl_new_pdp(struct sk_buff *skb, struct genl_info *info)
1038 {
1039         unsigned int version;
1040         struct pdp_ctx *pctx;
1041         struct gtp_dev *gtp;
1042         struct sock *sk;
1043         int err;
1044
1045         if (!info->attrs[GTPA_VERSION] ||
1046             !info->attrs[GTPA_LINK] ||
1047             !info->attrs[GTPA_PEER_ADDRESS] ||
1048             !info->attrs[GTPA_MS_ADDRESS])
1049                 return -EINVAL;
1050
1051         version = nla_get_u32(info->attrs[GTPA_VERSION]);
1052
1053         switch (version) {
1054         case GTP_V0:
1055                 if (!info->attrs[GTPA_TID] ||
1056                     !info->attrs[GTPA_FLOW])
1057                         return -EINVAL;
1058                 break;
1059         case GTP_V1:
1060                 if (!info->attrs[GTPA_I_TEI] ||
1061                     !info->attrs[GTPA_O_TEI])
1062                         return -EINVAL;
1063                 break;
1064
1065         default:
1066                 return -EINVAL;
1067         }
1068
1069         rtnl_lock();
1070
1071         gtp = gtp_find_dev(sock_net(skb->sk), info->attrs);
1072         if (!gtp) {
1073                 err = -ENODEV;
1074                 goto out_unlock;
1075         }
1076
1077         if (version == GTP_V0)
1078                 sk = gtp->sk0;
1079         else if (version == GTP_V1)
1080                 sk = gtp->sk1u;
1081         else
1082                 sk = NULL;
1083
1084         if (!sk) {
1085                 err = -ENODEV;
1086                 goto out_unlock;
1087         }
1088
1089         pctx = gtp_pdp_add(gtp, sk, info);
1090         if (IS_ERR(pctx)) {
1091                 err = PTR_ERR(pctx);
1092         } else {
1093                 gtp_tunnel_notify(pctx, GTP_CMD_NEWPDP, GFP_KERNEL);
1094                 err = 0;
1095         }
1096
1097 out_unlock:
1098         rtnl_unlock();
1099         return err;
1100 }
1101
1102 static struct pdp_ctx *gtp_find_pdp_by_link(struct net *net,
1103                                             struct nlattr *nla[])
1104 {
1105         struct gtp_dev *gtp;
1106
1107         gtp = gtp_find_dev(net, nla);
1108         if (!gtp)
1109                 return ERR_PTR(-ENODEV);
1110
1111         if (nla[GTPA_MS_ADDRESS]) {
1112                 __be32 ip = nla_get_be32(nla[GTPA_MS_ADDRESS]);
1113
1114                 return ipv4_pdp_find(gtp, ip);
1115         } else if (nla[GTPA_VERSION]) {
1116                 u32 gtp_version = nla_get_u32(nla[GTPA_VERSION]);
1117
1118                 if (gtp_version == GTP_V0 && nla[GTPA_TID])
1119                         return gtp0_pdp_find(gtp, nla_get_u64(nla[GTPA_TID]));
1120                 else if (gtp_version == GTP_V1 && nla[GTPA_I_TEI])
1121                         return gtp1_pdp_find(gtp, nla_get_u32(nla[GTPA_I_TEI]));
1122         }
1123
1124         return ERR_PTR(-EINVAL);
1125 }
1126
1127 static struct pdp_ctx *gtp_find_pdp(struct net *net, struct nlattr *nla[])
1128 {
1129         struct pdp_ctx *pctx;
1130
1131         if (nla[GTPA_LINK])
1132                 pctx = gtp_find_pdp_by_link(net, nla);
1133         else
1134                 pctx = ERR_PTR(-EINVAL);
1135
1136         if (!pctx)
1137                 pctx = ERR_PTR(-ENOENT);
1138
1139         return pctx;
1140 }
1141
1142 static int gtp_genl_del_pdp(struct sk_buff *skb, struct genl_info *info)
1143 {
1144         struct pdp_ctx *pctx;
1145         int err = 0;
1146
1147         if (!info->attrs[GTPA_VERSION])
1148                 return -EINVAL;
1149
1150         rcu_read_lock();
1151
1152         pctx = gtp_find_pdp(sock_net(skb->sk), info->attrs);
1153         if (IS_ERR(pctx)) {
1154                 err = PTR_ERR(pctx);
1155                 goto out_unlock;
1156         }
1157
1158         if (pctx->gtp_version == GTP_V0)
1159                 netdev_dbg(pctx->dev, "GTPv0-U: deleting tunnel id = %llx (pdp %p)\n",
1160                            pctx->u.v0.tid, pctx);
1161         else if (pctx->gtp_version == GTP_V1)
1162                 netdev_dbg(pctx->dev, "GTPv1-U: deleting tunnel id = %x/%x (pdp %p)\n",
1163                            pctx->u.v1.i_tei, pctx->u.v1.o_tei, pctx);
1164
1165         gtp_tunnel_notify(pctx, GTP_CMD_DELPDP, GFP_ATOMIC);
1166         pdp_context_delete(pctx);
1167
1168 out_unlock:
1169         rcu_read_unlock();
1170         return err;
1171 }
1172
1173 static struct genl_family gtp_genl_family;
1174
1175 enum gtp_multicast_groups {
1176         GTP_GENL_MCGRP,
1177 };
1178
1179 static const struct genl_multicast_group gtp_genl_mcgrps[] = {
1180         [GTP_GENL_MCGRP] = { .name = GTP_GENL_MCGRP_NAME },
1181 };
1182
1183 static int gtp_genl_fill_info(struct sk_buff *skb, u32 snd_portid, u32 snd_seq,
1184                               int flags, u32 type, struct pdp_ctx *pctx)
1185 {
1186         void *genlh;
1187
1188         genlh = genlmsg_put(skb, snd_portid, snd_seq, &gtp_genl_family, flags,
1189                             type);
1190         if (genlh == NULL)
1191                 goto nlmsg_failure;
1192
1193         if (nla_put_u32(skb, GTPA_VERSION, pctx->gtp_version) ||
1194             nla_put_u32(skb, GTPA_LINK, pctx->dev->ifindex) ||
1195             nla_put_be32(skb, GTPA_PEER_ADDRESS, pctx->peer_addr_ip4.s_addr) ||
1196             nla_put_be32(skb, GTPA_MS_ADDRESS, pctx->ms_addr_ip4.s_addr))
1197                 goto nla_put_failure;
1198
1199         switch (pctx->gtp_version) {
1200         case GTP_V0:
1201                 if (nla_put_u64_64bit(skb, GTPA_TID, pctx->u.v0.tid, GTPA_PAD) ||
1202                     nla_put_u16(skb, GTPA_FLOW, pctx->u.v0.flow))
1203                         goto nla_put_failure;
1204                 break;
1205         case GTP_V1:
1206                 if (nla_put_u32(skb, GTPA_I_TEI, pctx->u.v1.i_tei) ||
1207                     nla_put_u32(skb, GTPA_O_TEI, pctx->u.v1.o_tei))
1208                         goto nla_put_failure;
1209                 break;
1210         }
1211         genlmsg_end(skb, genlh);
1212         return 0;
1213
1214 nlmsg_failure:
1215 nla_put_failure:
1216         genlmsg_cancel(skb, genlh);
1217         return -EMSGSIZE;
1218 }
1219
1220 static int gtp_tunnel_notify(struct pdp_ctx *pctx, u8 cmd, gfp_t allocation)
1221 {
1222         struct sk_buff *msg;
1223         int ret;
1224
1225         msg = nlmsg_new(NLMSG_DEFAULT_SIZE, allocation);
1226         if (!msg)
1227                 return -ENOMEM;
1228
1229         ret = gtp_genl_fill_info(msg, 0, 0, 0, cmd, pctx);
1230         if (ret < 0) {
1231                 nlmsg_free(msg);
1232                 return ret;
1233         }
1234
1235         ret = genlmsg_multicast_netns(&gtp_genl_family, dev_net(pctx->dev), msg,
1236                                       0, GTP_GENL_MCGRP, GFP_ATOMIC);
1237         return ret;
1238 }
1239
1240 static int gtp_genl_get_pdp(struct sk_buff *skb, struct genl_info *info)
1241 {
1242         struct pdp_ctx *pctx = NULL;
1243         struct sk_buff *skb2;
1244         int err;
1245
1246         if (!info->attrs[GTPA_VERSION])
1247                 return -EINVAL;
1248
1249         rcu_read_lock();
1250
1251         pctx = gtp_find_pdp(sock_net(skb->sk), info->attrs);
1252         if (IS_ERR(pctx)) {
1253                 err = PTR_ERR(pctx);
1254                 goto err_unlock;
1255         }
1256
1257         skb2 = genlmsg_new(NLMSG_GOODSIZE, GFP_ATOMIC);
1258         if (skb2 == NULL) {
1259                 err = -ENOMEM;
1260                 goto err_unlock;
1261         }
1262
1263         err = gtp_genl_fill_info(skb2, NETLINK_CB(skb).portid, info->snd_seq,
1264                                  0, info->nlhdr->nlmsg_type, pctx);
1265         if (err < 0)
1266                 goto err_unlock_free;
1267
1268         rcu_read_unlock();
1269         return genlmsg_unicast(genl_info_net(info), skb2, info->snd_portid);
1270
1271 err_unlock_free:
1272         kfree_skb(skb2);
1273 err_unlock:
1274         rcu_read_unlock();
1275         return err;
1276 }
1277
1278 static int gtp_genl_dump_pdp(struct sk_buff *skb,
1279                                 struct netlink_callback *cb)
1280 {
1281         struct gtp_dev *last_gtp = (struct gtp_dev *)cb->args[2], *gtp;
1282         int i, j, bucket = cb->args[0], skip = cb->args[1];
1283         struct net *net = sock_net(skb->sk);
1284         struct pdp_ctx *pctx;
1285         struct gtp_net *gn;
1286
1287         gn = net_generic(net, gtp_net_id);
1288
1289         if (cb->args[4])
1290                 return 0;
1291
1292         rcu_read_lock();
1293         list_for_each_entry_rcu(gtp, &gn->gtp_dev_list, list) {
1294                 if (last_gtp && last_gtp != gtp)
1295                         continue;
1296                 else
1297                         last_gtp = NULL;
1298
1299                 for (i = bucket; i < gtp->hash_size; i++) {
1300                         j = 0;
1301                         hlist_for_each_entry_rcu(pctx, &gtp->tid_hash[i],
1302                                                  hlist_tid) {
1303                                 if (j >= skip &&
1304                                     gtp_genl_fill_info(skb,
1305                                             NETLINK_CB(cb->skb).portid,
1306                                             cb->nlh->nlmsg_seq,
1307                                             NLM_F_MULTI,
1308                                             cb->nlh->nlmsg_type, pctx)) {
1309                                         cb->args[0] = i;
1310                                         cb->args[1] = j;
1311                                         cb->args[2] = (unsigned long)gtp;
1312                                         goto out;
1313                                 }
1314                                 j++;
1315                         }
1316                         skip = 0;
1317                 }
1318                 bucket = 0;
1319         }
1320         cb->args[4] = 1;
1321 out:
1322         rcu_read_unlock();
1323         return skb->len;
1324 }
1325
1326 static const struct nla_policy gtp_genl_policy[GTPA_MAX + 1] = {
1327         [GTPA_LINK]             = { .type = NLA_U32, },
1328         [GTPA_VERSION]          = { .type = NLA_U32, },
1329         [GTPA_TID]              = { .type = NLA_U64, },
1330         [GTPA_PEER_ADDRESS]     = { .type = NLA_U32, },
1331         [GTPA_MS_ADDRESS]       = { .type = NLA_U32, },
1332         [GTPA_FLOW]             = { .type = NLA_U16, },
1333         [GTPA_NET_NS_FD]        = { .type = NLA_U32, },
1334         [GTPA_I_TEI]            = { .type = NLA_U32, },
1335         [GTPA_O_TEI]            = { .type = NLA_U32, },
1336 };
1337
1338 static const struct genl_small_ops gtp_genl_ops[] = {
1339         {
1340                 .cmd = GTP_CMD_NEWPDP,
1341                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
1342                 .doit = gtp_genl_new_pdp,
1343                 .flags = GENL_ADMIN_PERM,
1344         },
1345         {
1346                 .cmd = GTP_CMD_DELPDP,
1347                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
1348                 .doit = gtp_genl_del_pdp,
1349                 .flags = GENL_ADMIN_PERM,
1350         },
1351         {
1352                 .cmd = GTP_CMD_GETPDP,
1353                 .validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
1354                 .doit = gtp_genl_get_pdp,
1355                 .dumpit = gtp_genl_dump_pdp,
1356                 .flags = GENL_ADMIN_PERM,
1357         },
1358 };
1359
1360 static struct genl_family gtp_genl_family __ro_after_init = {
1361         .name           = "gtp",
1362         .version        = 0,
1363         .hdrsize        = 0,
1364         .maxattr        = GTPA_MAX,
1365         .policy = gtp_genl_policy,
1366         .netnsok        = true,
1367         .module         = THIS_MODULE,
1368         .small_ops      = gtp_genl_ops,
1369         .n_small_ops    = ARRAY_SIZE(gtp_genl_ops),
1370         .mcgrps         = gtp_genl_mcgrps,
1371         .n_mcgrps       = ARRAY_SIZE(gtp_genl_mcgrps),
1372 };
1373
1374 static int __net_init gtp_net_init(struct net *net)
1375 {
1376         struct gtp_net *gn = net_generic(net, gtp_net_id);
1377
1378         INIT_LIST_HEAD(&gn->gtp_dev_list);
1379         return 0;
1380 }
1381
1382 static void __net_exit gtp_net_exit(struct net *net)
1383 {
1384         struct gtp_net *gn = net_generic(net, gtp_net_id);
1385         struct gtp_dev *gtp;
1386         LIST_HEAD(list);
1387
1388         rtnl_lock();
1389         list_for_each_entry(gtp, &gn->gtp_dev_list, list)
1390                 gtp_dellink(gtp->dev, &list);
1391
1392         unregister_netdevice_many(&list);
1393         rtnl_unlock();
1394 }
1395
1396 static struct pernet_operations gtp_net_ops = {
1397         .init   = gtp_net_init,
1398         .exit   = gtp_net_exit,
1399         .id     = &gtp_net_id,
1400         .size   = sizeof(struct gtp_net),
1401 };
1402
1403 static int __init gtp_init(void)
1404 {
1405         int err;
1406
1407         get_random_bytes(&gtp_h_initval, sizeof(gtp_h_initval));
1408
1409         err = register_pernet_subsys(&gtp_net_ops);
1410         if (err < 0)
1411                 goto error_out;
1412
1413         err = rtnl_link_register(&gtp_link_ops);
1414         if (err < 0)
1415                 goto unreg_pernet_subsys;
1416
1417         err = genl_register_family(&gtp_genl_family);
1418         if (err < 0)
1419                 goto unreg_rtnl_link;
1420
1421         pr_info("GTP module loaded (pdp ctx size %zd bytes)\n",
1422                 sizeof(struct pdp_ctx));
1423         return 0;
1424
1425 unreg_rtnl_link:
1426         rtnl_link_unregister(&gtp_link_ops);
1427 unreg_pernet_subsys:
1428         unregister_pernet_subsys(&gtp_net_ops);
1429 error_out:
1430         pr_err("error loading GTP module loaded\n");
1431         return err;
1432 }
1433 late_initcall(gtp_init);
1434
1435 static void __exit gtp_fini(void)
1436 {
1437         genl_unregister_family(&gtp_genl_family);
1438         rtnl_link_unregister(&gtp_link_ops);
1439         unregister_pernet_subsys(&gtp_net_ops);
1440
1441         pr_info("GTP module unloaded\n");
1442 }
1443 module_exit(gtp_fini);
1444
1445 MODULE_LICENSE("GPL");
1446 MODULE_AUTHOR("Harald Welte <hwelte@sysmocom.de>");
1447 MODULE_DESCRIPTION("Interface driver for GTP encapsulated traffic");
1448 MODULE_ALIAS_RTNL_LINK("gtp");
1449 MODULE_ALIAS_GENL_FAMILY("gtp");