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