drivers/net/gtp.c

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