net/ipv4/fib_frontend.c

   1 /*
   2  * INET         An implementation of the TCP/IP protocol suite for the LINUX
   3  *              operating system.  INET is implemented using the  BSD Socket
   4  *              interface as the means of communication with the user level.
   5  *
   6  *              IPv4 Forwarding Information Base: FIB frontend.
   7  *
   8  * Authors:     Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
   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 #include <linux/module.h>
  17 #include <asm/uaccess.h>
  18 #include <asm/system.h>
  19 #include <linux/bitops.h>
  20 #include <linux/capability.h>
  21 #include <linux/types.h>
  22 #include <linux/kernel.h>
  23 #include <linux/mm.h>
  24 #include <linux/string.h>
  25 #include <linux/socket.h>
  26 #include <linux/sockios.h>
  27 #include <linux/errno.h>
  28 #include <linux/in.h>
  29 #include <linux/inet.h>
  30 #include <linux/inetdevice.h>
  31 #include <linux/netdevice.h>
  32 #include <linux/if_addr.h>
  33 #include <linux/if_arp.h>
  34 #include <linux/skbuff.h>
  35 #include <linux/init.h>
  36 #include <linux/list.h>
  37 #include <linux/slab.h>
  38
  39 #include <net/ip.h>
  40 #include <net/protocol.h>
  41 #include <net/route.h>
  42 #include <net/tcp.h>
  43 #include <net/sock.h>
  44 #include <net/arp.h>
  45 #include <net/ip_fib.h>
  46 #include <net/rtnetlink.h>
  47
  48 #ifndef CONFIG_IP_MULTIPLE_TABLES
  49
  50 static int __net_init fib4_rules_init(struct net *net)
  51 {
  52         struct fib_table *local_table, *main_table;
  53
  54         local_table = fib_hash_table(RT_TABLE_LOCAL);
  55         if (local_table == NULL)
  56                 return -ENOMEM;
  57
  58         main_table  = fib_hash_table(RT_TABLE_MAIN);
  59         if (main_table == NULL)
  60                 goto fail;
  61
  62         hlist_add_head_rcu(&local_table->tb_hlist,
  63                                 &net->ipv4.fib_table_hash[TABLE_LOCAL_INDEX]);
  64         hlist_add_head_rcu(&main_table->tb_hlist,
  65                                 &net->ipv4.fib_table_hash[TABLE_MAIN_INDEX]);
  66         return 0;
  67
  68 fail:
  69         kfree(local_table);
  70         return -ENOMEM;
  71 }
  72 #else
  73
  74 struct fib_table *fib_new_table(struct net *net, u32 id)
  75 {
  76         struct fib_table *tb;
  77         unsigned int h;
  78
  79         if (id == 0)
  80                 id = RT_TABLE_MAIN;
  81         tb = fib_get_table(net, id);
  82         if (tb)
  83                 return tb;
  84
  85         tb = fib_hash_table(id);
  86         if (!tb)
  87                 return NULL;
  88         h = id & (FIB_TABLE_HASHSZ - 1);
  89         hlist_add_head_rcu(&tb->tb_hlist, &net->ipv4.fib_table_hash[h]);
  90         return tb;
  91 }
  92
  93 struct fib_table *fib_get_table(struct net *net, u32 id)
  94 {
  95         struct fib_table *tb;
  96         struct hlist_node *node;
  97         struct hlist_head *head;
  98         unsigned int h;
  99
 100         if (id == 0)
 101                 id = RT_TABLE_MAIN;
 102         h = id & (FIB_TABLE_HASHSZ - 1);
 103
 104         rcu_read_lock();
 105         head = &net->ipv4.fib_table_hash[h];
 106         hlist_for_each_entry_rcu(tb, node, head, tb_hlist) {
 107                 if (tb->tb_id == id) {
 108                         rcu_read_unlock();
 109                         return tb;
 110                 }
 111         }
 112         rcu_read_unlock();
 113         return NULL;
 114 }
 115 #endif /* CONFIG_IP_MULTIPLE_TABLES */
 116
 117 void fib_select_default(struct net *net,
 118                         const struct flowi *flp, struct fib_result *res)
 119 {
 120         struct fib_table *tb;
 121         int table = RT_TABLE_MAIN;
 122 #ifdef CONFIG_IP_MULTIPLE_TABLES
 123         if (res->r == NULL || res->r->action != FR_ACT_TO_TBL)
 124                 return;
 125         table = res->r->table;
 126 #endif
 127         tb = fib_get_table(net, table);
 128         if (FIB_RES_GW(*res) && FIB_RES_NH(*res).nh_scope == RT_SCOPE_LINK)
 129                 fib_table_select_default(tb, flp, res);
 130 }
 131
 132 static void fib_flush(struct net *net)
 133 {
 134         int flushed = 0;
 135         struct fib_table *tb;
 136         struct hlist_node *node;
 137         struct hlist_head *head;
 138         unsigned int h;
 139
 140         for (h = 0; h < FIB_TABLE_HASHSZ; h++) {
 141                 head = &net->ipv4.fib_table_hash[h];
 142                 hlist_for_each_entry(tb, node, head, tb_hlist)
 143                         flushed += fib_table_flush(tb);
 144         }
 145
 146         if (flushed)
 147                 rt_cache_flush(net, -1);
 148 }
 149
 150 /**
 151  * __ip_dev_find - find the first device with a given source address.
 152  * @net: the net namespace
 153  * @addr: the source address
 154  * @devref: if true, take a reference on the found device
 155  *
 156  * If a caller uses devref=false, it should be protected by RCU, or RTNL
 157  */
 158 struct net_device *__ip_dev_find(struct net *net, __be32 addr, bool devref)
 159 {
 160         struct flowi fl = {
 161                 .nl_u = {
 162                         .ip4_u = {
 163                                 .daddr = addr
 164                         }
 165                 },
 166         };
 167         struct fib_result res = { 0 };
 168         struct net_device *dev = NULL;
 169         struct fib_table *local_table;
 170
 171 #ifdef CONFIG_IP_MULTIPLE_TABLES
 172         res.r = NULL;
 173 #endif
 174
 175         rcu_read_lock();
 176         local_table = fib_get_table(net, RT_TABLE_LOCAL);
 177         if (!local_table ||
 178             fib_table_lookup(local_table, &fl, &res, FIB_LOOKUP_NOREF)) {
 179                 rcu_read_unlock();
 180                 return NULL;
 181         }
 182         if (res.type != RTN_LOCAL)
 183                 goto out;
 184         dev = FIB_RES_DEV(res);
 185
 186         if (dev && devref)
 187                 dev_hold(dev);
 188 out:
 189         rcu_read_unlock();
 190         return dev;
 191 }
 192 EXPORT_SYMBOL(__ip_dev_find);
 193
 194 /*
 195  * Find address type as if only "dev" was present in the system. If
 196  * on_dev is NULL then all interfaces are taken into consideration.
 197  */
 198 static inline unsigned __inet_dev_addr_type(struct net *net,
 199                                             const struct net_device *dev,
 200                                             __be32 addr)
 201 {
 202         struct flowi            fl = { .nl_u = { .ip4_u = { .daddr = addr } } };
 203         struct fib_result       res;
 204         unsigned ret = RTN_BROADCAST;
 205         struct fib_table *local_table;
 206
 207         if (ipv4_is_zeronet(addr) || ipv4_is_lbcast(addr))
 208                 return RTN_BROADCAST;
 209         if (ipv4_is_multicast(addr))
 210                 return RTN_MULTICAST;
 211
 212 #ifdef CONFIG_IP_MULTIPLE_TABLES
 213         res.r = NULL;
 214 #endif
 215
 216         local_table = fib_get_table(net, RT_TABLE_LOCAL);
 217         if (local_table) {
 218                 ret = RTN_UNICAST;
 219                 rcu_read_lock();
 220                 if (!fib_table_lookup(local_table, &fl, &res, FIB_LOOKUP_NOREF)) {
 221                         if (!dev || dev == res.fi->fib_dev)
 222                                 ret = res.type;
 223                 }
 224                 rcu_read_unlock();
 225         }
 226         return ret;
 227 }
 228
 229 unsigned int inet_addr_type(struct net *net, __be32 addr)
 230 {
 231         return __inet_dev_addr_type(net, NULL, addr);
 232 }
 233 EXPORT_SYMBOL(inet_addr_type);
 234
 235 unsigned int inet_dev_addr_type(struct net *net, const struct net_device *dev,
 236                                 __be32 addr)
 237 {
 238         return __inet_dev_addr_type(net, dev, addr);
 239 }
 240 EXPORT_SYMBOL(inet_dev_addr_type);
 241
 242 /* Given (packet source, input interface) and optional (dst, oif, tos):
 243  * - (main) check, that source is valid i.e. not broadcast or our local
 244  *   address.
 245  * - figure out what "logical" interface this packet arrived
 246  *   and calculate "specific destination" address.
 247  * - check, that packet arrived from expected physical interface.
 248  * called with rcu_read_lock()
 249  */
 250 int fib_validate_source(__be32 src, __be32 dst, u8 tos, int oif,
 251                         struct net_device *dev, __be32 *spec_dst,
 252                         u32 *itag, u32 mark)
 253 {
 254         struct in_device *in_dev;
 255         struct flowi fl = {
 256                 .nl_u = {
 257                         .ip4_u = {
 258                                 .daddr = src,
 259                                 .saddr = dst,
 260                                 .tos = tos
 261                         }
 262                 },
 263                 .mark = mark,
 264                 .iif = oif
 265         };
 266         struct fib_result res;
 267         int no_addr, rpf, accept_local;
 268         bool dev_match;
 269         int ret;
 270         struct net *net;
 271
 272         no_addr = rpf = accept_local = 0;
 273         in_dev = __in_dev_get_rcu(dev);
 274         if (in_dev) {
 275                 no_addr = in_dev->ifa_list == NULL;
 276                 rpf = IN_DEV_RPFILTER(in_dev);
 277                 accept_local = IN_DEV_ACCEPT_LOCAL(in_dev);
 278                 if (mark && !IN_DEV_SRC_VMARK(in_dev))
 279                         fl.mark = 0;
 280         }
 281
 282         if (in_dev == NULL)
 283                 goto e_inval;
 284
 285         net = dev_net(dev);
 286         if (fib_lookup(net, &fl, &res))
 287                 goto last_resort;
 288         if (res.type != RTN_UNICAST) {
 289                 if (res.type != RTN_LOCAL || !accept_local)
 290                         goto e_inval;
 291         }
 292         *spec_dst = FIB_RES_PREFSRC(res);
 293         fib_combine_itag(itag, &res);
 294         dev_match = false;
 295
 296 #ifdef CONFIG_IP_ROUTE_MULTIPATH
 297         for (ret = 0; ret < res.fi->fib_nhs; ret++) {
 298                 struct fib_nh *nh = &res.fi->fib_nh[ret];
 299
 300                 if (nh->nh_dev == dev) {
 301                         dev_match = true;
 302                         break;
 303                 }
 304         }
 305 #else
 306         if (FIB_RES_DEV(res) == dev)
 307                 dev_match = true;
 308 #endif
 309         if (dev_match) {
 310                 ret = FIB_RES_NH(res).nh_scope >= RT_SCOPE_HOST;
 311                 return ret;
 312         }
 313         if (no_addr)
 314                 goto last_resort;
 315         if (rpf == 1)
 316                 goto e_rpf;
 317         fl.oif = dev->ifindex;
 318
 319         ret = 0;
 320         if (fib_lookup(net, &fl, &res) == 0) {
 321                 if (res.type == RTN_UNICAST) {
 322                         *spec_dst = FIB_RES_PREFSRC(res);
 323                         ret = FIB_RES_NH(res).nh_scope >= RT_SCOPE_HOST;
 324                 }
 325         }
 326         return ret;
 327
 328 last_resort:
 329         if (rpf)
 330                 goto e_rpf;
 331         *spec_dst = inet_select_addr(dev, 0, RT_SCOPE_UNIVERSE);
 332         *itag = 0;
 333         return 0;
 334
 335 e_inval:
 336         return -EINVAL;
 337 e_rpf:
 338         return -EXDEV;
 339 }
 340
 341 static inline __be32 sk_extract_addr(struct sockaddr *addr)
 342 {
 343         return ((struct sockaddr_in *) addr)->sin_addr.s_addr;
 344 }
 345
 346 static int put_rtax(struct nlattr *mx, int len, int type, u32 value)
 347 {
 348         struct nlattr *nla;
 349
 350         nla = (struct nlattr *) ((char *) mx + len);
 351         nla->nla_type = type;
 352         nla->nla_len = nla_attr_size(4);
 353         *(u32 *) nla_data(nla) = value;
 354
 355         return len + nla_total_size(4);
 356 }
 357
 358 static int rtentry_to_fib_config(struct net *net, int cmd, struct rtentry *rt,
 359                                  struct fib_config *cfg)
 360 {
 361         __be32 addr;
 362         int plen;
 363
 364         memset(cfg, 0, sizeof(*cfg));
 365         cfg->fc_nlinfo.nl_net = net;
 366
 367         if (rt->rt_dst.sa_family != AF_INET)
 368                 return -EAFNOSUPPORT;
 369
 370         /*
 371          * Check mask for validity:
 372          * a) it must be contiguous.
 373          * b) destination must have all host bits clear.
 374          * c) if application forgot to set correct family (AF_INET),
 375          *    reject request unless it is absolutely clear i.e.
 376          *    both family and mask are zero.
 377          */
 378         plen = 32;
 379         addr = sk_extract_addr(&rt->rt_dst);
 380         if (!(rt->rt_flags & RTF_HOST)) {
 381                 __be32 mask = sk_extract_addr(&rt->rt_genmask);
 382
 383                 if (rt->rt_genmask.sa_family != AF_INET) {
 384                         if (mask || rt->rt_genmask.sa_family)
 385                                 return -EAFNOSUPPORT;
 386                 }
 387
 388                 if (bad_mask(mask, addr))
 389                         return -EINVAL;
 390
 391                 plen = inet_mask_len(mask);
 392         }
 393
 394         cfg->fc_dst_len = plen;
 395         cfg->fc_dst = addr;
 396
 397         if (cmd != SIOCDELRT) {
 398                 cfg->fc_nlflags = NLM_F_CREATE;
 399                 cfg->fc_protocol = RTPROT_BOOT;
 400         }
 401
 402         if (rt->rt_metric)
 403                 cfg->fc_priority = rt->rt_metric - 1;
 404
 405         if (rt->rt_flags & RTF_REJECT) {
 406                 cfg->fc_scope = RT_SCOPE_HOST;
 407                 cfg->fc_type = RTN_UNREACHABLE;
 408                 return 0;
 409         }
 410
 411         cfg->fc_scope = RT_SCOPE_NOWHERE;
 412         cfg->fc_type = RTN_UNICAST;
 413
 414         if (rt->rt_dev) {
 415                 char *colon;
 416                 struct net_device *dev;
 417                 char devname[IFNAMSIZ];
 418
 419                 if (copy_from_user(devname, rt->rt_dev, IFNAMSIZ-1))
 420                         return -EFAULT;
 421
 422                 devname[IFNAMSIZ-1] = 0;
 423                 colon = strchr(devname, ':');
 424                 if (colon)
 425                         *colon = 0;
 426                 dev = __dev_get_by_name(net, devname);
 427                 if (!dev)
 428                         return -ENODEV;
 429                 cfg->fc_oif = dev->ifindex;
 430                 if (colon) {
 431                         struct in_ifaddr *ifa;
 432                         struct in_device *in_dev = __in_dev_get_rtnl(dev);
 433                         if (!in_dev)
 434                                 return -ENODEV;
 435                         *colon = ':';
 436                         for (ifa = in_dev->ifa_list; ifa; ifa = ifa->ifa_next)
 437                                 if (strcmp(ifa->ifa_label, devname) == 0)
 438                                         break;
 439                         if (ifa == NULL)
 440                                 return -ENODEV;
 441                         cfg->fc_prefsrc = ifa->ifa_local;
 442                 }
 443         }
 444
 445         addr = sk_extract_addr(&rt->rt_gateway);
 446         if (rt->rt_gateway.sa_family == AF_INET && addr) {
 447                 cfg->fc_gw = addr;
 448                 if (rt->rt_flags & RTF_GATEWAY &&
 449                     inet_addr_type(net, addr) == RTN_UNICAST)
 450                         cfg->fc_scope = RT_SCOPE_UNIVERSE;
 451         }
 452
 453         if (cmd == SIOCDELRT)
 454                 return 0;
 455
 456         if (rt->rt_flags & RTF_GATEWAY && !cfg->fc_gw)
 457                 return -EINVAL;
 458
 459         if (cfg->fc_scope == RT_SCOPE_NOWHERE)
 460                 cfg->fc_scope = RT_SCOPE_LINK;
 461
 462         if (rt->rt_flags & (RTF_MTU | RTF_WINDOW | RTF_IRTT)) {
 463                 struct nlattr *mx;
 464                 int len = 0;
 465
 466                 mx = kzalloc(3 * nla_total_size(4), GFP_KERNEL);
 467                 if (mx == NULL)
 468                         return -ENOMEM;
 469
 470                 if (rt->rt_flags & RTF_MTU)
 471                         len = put_rtax(mx, len, RTAX_ADVMSS, rt->rt_mtu - 40);
 472
 473                 if (rt->rt_flags & RTF_WINDOW)
 474                         len = put_rtax(mx, len, RTAX_WINDOW, rt->rt_window);
 475
 476                 if (rt->rt_flags & RTF_IRTT)
 477                         len = put_rtax(mx, len, RTAX_RTT, rt->rt_irtt << 3);
 478
 479                 cfg->fc_mx = mx;
 480                 cfg->fc_mx_len = len;
 481         }
 482
 483         return 0;
 484 }
 485
 486 /*
 487  * Handle IP routing ioctl calls.
 488  * These are used to manipulate the routing tables
 489  */
 490 int ip_rt_ioctl(struct net *net, unsigned int cmd, void __user *arg)
 491 {
 492         struct fib_config cfg;
 493         struct rtentry rt;
 494         int err;
 495
 496         switch (cmd) {
 497         case SIOCADDRT:         /* Add a route */
 498         case SIOCDELRT:         /* Delete a route */
 499                 if (!capable(CAP_NET_ADMIN))
 500                         return -EPERM;
 501
 502                 if (copy_from_user(&rt, arg, sizeof(rt)))
 503                         return -EFAULT;
 504
 505                 rtnl_lock();
 506                 err = rtentry_to_fib_config(net, cmd, &rt, &cfg);
 507                 if (err == 0) {
 508                         struct fib_table *tb;
 509
 510                         if (cmd == SIOCDELRT) {
 511                                 tb = fib_get_table(net, cfg.fc_table);
 512                                 if (tb)
 513                                         err = fib_table_delete(tb, &cfg);
 514                                 else
 515                                         err = -ESRCH;
 516                         } else {
 517                                 tb = fib_new_table(net, cfg.fc_table);
 518                                 if (tb)
 519                                         err = fib_table_insert(tb, &cfg);
 520                                 else
 521                                         err = -ENOBUFS;
 522                         }
 523
 524                         /* allocated by rtentry_to_fib_config() */
 525                         kfree(cfg.fc_mx);
 526                 }
 527                 rtnl_unlock();
 528                 return err;
 529         }
 530         return -EINVAL;
 531 }
 532
 533 const struct nla_policy rtm_ipv4_policy[RTA_MAX + 1] = {
 534         [RTA_DST]               = { .type = NLA_U32 },
 535         [RTA_SRC]               = { .type = NLA_U32 },
 536         [RTA_IIF]               = { .type = NLA_U32 },
 537         [RTA_OIF]               = { .type = NLA_U32 },
 538         [RTA_GATEWAY]           = { .type = NLA_U32 },
 539         [RTA_PRIORITY]          = { .type = NLA_U32 },
 540         [RTA_PREFSRC]           = { .type = NLA_U32 },
 541         [RTA_METRICS]           = { .type = NLA_NESTED },
 542         [RTA_MULTIPATH]         = { .len = sizeof(struct rtnexthop) },
 543         [RTA_FLOW]              = { .type = NLA_U32 },
 544 };
 545
 546 static int rtm_to_fib_config(struct net *net, struct sk_buff *skb,
 547                              struct nlmsghdr *nlh, struct fib_config *cfg)
 548 {
 549         struct nlattr *attr;
 550         int err, remaining;
 551         struct rtmsg *rtm;
 552
 553         err = nlmsg_validate(nlh, sizeof(*rtm), RTA_MAX, rtm_ipv4_policy);
 554         if (err < 0)
 555                 goto errout;
 556
 557         memset(cfg, 0, sizeof(*cfg));
 558
 559         rtm = nlmsg_data(nlh);
 560         cfg->fc_dst_len = rtm->rtm_dst_len;
 561         cfg->fc_tos = rtm->rtm_tos;
 562         cfg->fc_table = rtm->rtm_table;
 563         cfg->fc_protocol = rtm->rtm_protocol;
 564         cfg->fc_scope = rtm->rtm_scope;
 565         cfg->fc_type = rtm->rtm_type;
 566         cfg->fc_flags = rtm->rtm_flags;
 567         cfg->fc_nlflags = nlh->nlmsg_flags;
 568
 569         cfg->fc_nlinfo.pid = NETLINK_CB(skb).pid;
 570         cfg->fc_nlinfo.nlh = nlh;
 571         cfg->fc_nlinfo.nl_net = net;
 572
 573         if (cfg->fc_type > RTN_MAX) {
 574                 err = -EINVAL;
 575                 goto errout;
 576         }
 577
 578         nlmsg_for_each_attr(attr, nlh, sizeof(struct rtmsg), remaining) {
 579                 switch (nla_type(attr)) {
 580                 case RTA_DST:
 581                         cfg->fc_dst = nla_get_be32(attr);
 582                         break;
 583                 case RTA_OIF:
 584                         cfg->fc_oif = nla_get_u32(attr);
 585                         break;
 586                 case RTA_GATEWAY:
 587                         cfg->fc_gw = nla_get_be32(attr);
 588                         break;
 589                 case RTA_PRIORITY:
 590                         cfg->fc_priority = nla_get_u32(attr);
 591                         break;
 592                 case RTA_PREFSRC:
 593                         cfg->fc_prefsrc = nla_get_be32(attr);
 594                         break;
 595                 case RTA_METRICS:
 596                         cfg->fc_mx = nla_data(attr);
 597                         cfg->fc_mx_len = nla_len(attr);
 598                         break;
 599                 case RTA_MULTIPATH:
 600                         cfg->fc_mp = nla_data(attr);
 601                         cfg->fc_mp_len = nla_len(attr);
 602                         break;
 603                 case RTA_FLOW:
 604                         cfg->fc_flow = nla_get_u32(attr);
 605                         break;
 606                 case RTA_TABLE:
 607                         cfg->fc_table = nla_get_u32(attr);
 608                         break;
 609                 }
 610         }
 611
 612         return 0;
 613 errout:
 614         return err;
 615 }
 616
 617 static int inet_rtm_delroute(struct sk_buff *skb, struct nlmsghdr *nlh, void *arg)
 618 {
 619         struct net *net = sock_net(skb->sk);
 620         struct fib_config cfg;
 621         struct fib_table *tb;
 622         int err;
 623
 624         err = rtm_to_fib_config(net, skb, nlh, &cfg);
 625         if (err < 0)
 626                 goto errout;
 627
 628         tb = fib_get_table(net, cfg.fc_table);
 629         if (tb == NULL) {
 630                 err = -ESRCH;
 631                 goto errout;
 632         }
 633
 634         err = fib_table_delete(tb, &cfg);
 635 errout:
 636         return err;
 637 }
 638
 639 static int inet_rtm_newroute(struct sk_buff *skb, struct nlmsghdr *nlh, void *arg)
 640 {
 641         struct net *net = sock_net(skb->sk);
 642         struct fib_config cfg;
 643         struct fib_table *tb;
 644         int err;
 645
 646         err = rtm_to_fib_config(net, skb, nlh, &cfg);
 647         if (err < 0)
 648                 goto errout;
 649
 650         tb = fib_new_table(net, cfg.fc_table);
 651         if (tb == NULL) {
 652                 err = -ENOBUFS;
 653                 goto errout;
 654         }
 655
 656         err = fib_table_insert(tb, &cfg);
 657 errout:
 658         return err;
 659 }
 660
 661 static int inet_dump_fib(struct sk_buff *skb, struct netlink_callback *cb)
 662 {
 663         struct net *net = sock_net(skb->sk);
 664         unsigned int h, s_h;
 665         unsigned int e = 0, s_e;
 666         struct fib_table *tb;
 667         struct hlist_node *node;
 668         struct hlist_head *head;
 669         int dumped = 0;
 670
 671         if (nlmsg_len(cb->nlh) >= sizeof(struct rtmsg) &&
 672             ((struct rtmsg *) nlmsg_data(cb->nlh))->rtm_flags & RTM_F_CLONED)
 673                 return ip_rt_dump(skb, cb);
 674
 675         s_h = cb->args[0];
 676         s_e = cb->args[1];
 677
 678         for (h = s_h; h < FIB_TABLE_HASHSZ; h++, s_e = 0) {
 679                 e = 0;
 680                 head = &net->ipv4.fib_table_hash[h];
 681                 hlist_for_each_entry(tb, node, head, tb_hlist) {
 682                         if (e < s_e)
 683                                 goto next;
 684                         if (dumped)
 685                                 memset(&cb->args[2], 0, sizeof(cb->args) -
 686                                                  2 * sizeof(cb->args[0]));
 687                         if (fib_table_dump(tb, skb, cb) < 0)
 688                                 goto out;
 689                         dumped = 1;
 690 next:
 691                         e++;
 692                 }
 693         }
 694 out:
 695         cb->args[1] = e;
 696         cb->args[0] = h;
 697
 698         return skb->len;
 699 }
 700
 701 /* Prepare and feed intra-kernel routing request.
 702  * Really, it should be netlink message, but :-( netlink
 703  * can be not configured, so that we feed it directly
 704  * to fib engine. It is legal, because all events occur
 705  * only when netlink is already locked.
 706  */
 707 static void fib_magic(int cmd, int type, __be32 dst, int dst_len, struct in_ifaddr *ifa)
 708 {
 709         struct net *net = dev_net(ifa->ifa_dev->dev);
 710         struct fib_table *tb;
 711         struct fib_config cfg = {
 712                 .fc_protocol = RTPROT_KERNEL,
 713                 .fc_type = type,
 714                 .fc_dst = dst,
 715                 .fc_dst_len = dst_len,
 716                 .fc_prefsrc = ifa->ifa_local,
 717                 .fc_oif = ifa->ifa_dev->dev->ifindex,
 718                 .fc_nlflags = NLM_F_CREATE | NLM_F_APPEND,
 719                 .fc_nlinfo = {
 720                         .nl_net = net,
 721                 },
 722         };
 723
 724         if (type == RTN_UNICAST)
 725                 tb = fib_new_table(net, RT_TABLE_MAIN);
 726         else
 727                 tb = fib_new_table(net, RT_TABLE_LOCAL);
 728
 729         if (tb == NULL)
 730                 return;
 731
 732         cfg.fc_table = tb->tb_id;
 733
 734         if (type != RTN_LOCAL)
 735                 cfg.fc_scope = RT_SCOPE_LINK;
 736         else
 737                 cfg.fc_scope = RT_SCOPE_HOST;
 738
 739         if (cmd == RTM_NEWROUTE)
 740                 fib_table_insert(tb, &cfg);
 741         else
 742                 fib_table_delete(tb, &cfg);
 743 }
 744
 745 void fib_add_ifaddr(struct in_ifaddr *ifa)
 746 {
 747         struct in_device *in_dev = ifa->ifa_dev;
 748         struct net_device *dev = in_dev->dev;
 749         struct in_ifaddr *prim = ifa;
 750         __be32 mask = ifa->ifa_mask;
 751         __be32 addr = ifa->ifa_local;
 752         __be32 prefix = ifa->ifa_address & mask;
 753
 754         if (ifa->ifa_flags & IFA_F_SECONDARY) {
 755                 prim = inet_ifa_byprefix(in_dev, prefix, mask);
 756                 if (prim == NULL) {
 757                         printk(KERN_WARNING "fib_add_ifaddr: bug: prim == NULL\n");
 758                         return;
 759                 }
 760         }
 761
 762         fib_magic(RTM_NEWROUTE, RTN_LOCAL, addr, 32, prim);
 763
 764         if (!(dev->flags & IFF_UP))
 765                 return;
 766
 767         /* Add broadcast address, if it is explicitly assigned. */
 768         if (ifa->ifa_broadcast && ifa->ifa_broadcast != htonl(0xFFFFFFFF))
 769                 fib_magic(RTM_NEWROUTE, RTN_BROADCAST, ifa->ifa_broadcast, 32, prim);
 770
 771         if (!ipv4_is_zeronet(prefix) && !(ifa->ifa_flags & IFA_F_SECONDARY) &&
 772             (prefix != addr || ifa->ifa_prefixlen < 32)) {
 773                 fib_magic(RTM_NEWROUTE,
 774                           dev->flags & IFF_LOOPBACK ? RTN_LOCAL : RTN_UNICAST,
 775                           prefix, ifa->ifa_prefixlen, prim);
 776
 777                 /* Add network specific broadcasts, when it takes a sense */
 778                 if (ifa->ifa_prefixlen < 31) {
 779                         fib_magic(RTM_NEWROUTE, RTN_BROADCAST, prefix, 32, prim);
 780                         fib_magic(RTM_NEWROUTE, RTN_BROADCAST, prefix | ~mask,
 781                                   32, prim);
 782                 }
 783         }
 784 }
 785
 786 static void fib_del_ifaddr(struct in_ifaddr *ifa)
 787 {
 788         struct in_device *in_dev = ifa->ifa_dev;
 789         struct net_device *dev = in_dev->dev;
 790         struct in_ifaddr *ifa1;
 791         struct in_ifaddr *prim = ifa;
 792         __be32 brd = ifa->ifa_address | ~ifa->ifa_mask;
 793         __be32 any = ifa->ifa_address & ifa->ifa_mask;
 794 #define LOCAL_OK        1
 795 #define BRD_OK          2
 796 #define BRD0_OK         4
 797 #define BRD1_OK         8
 798         unsigned ok = 0;
 799
 800         if (!(ifa->ifa_flags & IFA_F_SECONDARY))
 801                 fib_magic(RTM_DELROUTE,
 802                           dev->flags & IFF_LOOPBACK ? RTN_LOCAL : RTN_UNICAST,
 803                           any, ifa->ifa_prefixlen, prim);
 804         else {
 805                 prim = inet_ifa_byprefix(in_dev, any, ifa->ifa_mask);
 806                 if (prim == NULL) {
 807                         printk(KERN_WARNING "fib_del_ifaddr: bug: prim == NULL\n");
 808                         return;
 809                 }
 810         }
 811
 812         /* Deletion is more complicated than add.
 813          * We should take care of not to delete too much :-)
 814          *
 815          * Scan address list to be sure that addresses are really gone.
 816          */
 817
 818         for (ifa1 = in_dev->ifa_list; ifa1; ifa1 = ifa1->ifa_next) {
 819                 if (ifa->ifa_local == ifa1->ifa_local)
 820                         ok |= LOCAL_OK;
 821                 if (ifa->ifa_broadcast == ifa1->ifa_broadcast)
 822                         ok |= BRD_OK;
 823                 if (brd == ifa1->ifa_broadcast)
 824                         ok |= BRD1_OK;
 825                 if (any == ifa1->ifa_broadcast)
 826                         ok |= BRD0_OK;
 827         }
 828
 829         if (!(ok & BRD_OK))
 830                 fib_magic(RTM_DELROUTE, RTN_BROADCAST, ifa->ifa_broadcast, 32, prim);
 831         if (!(ok & BRD1_OK))
 832                 fib_magic(RTM_DELROUTE, RTN_BROADCAST, brd, 32, prim);
 833         if (!(ok & BRD0_OK))
 834                 fib_magic(RTM_DELROUTE, RTN_BROADCAST, any, 32, prim);
 835         if (!(ok & LOCAL_OK)) {
 836                 fib_magic(RTM_DELROUTE, RTN_LOCAL, ifa->ifa_local, 32, prim);
 837
 838                 /* Check, that this local address finally disappeared. */
 839                 if (inet_addr_type(dev_net(dev), ifa->ifa_local) != RTN_LOCAL) {
 840                         /* And the last, but not the least thing.
 841                          * We must flush stray FIB entries.
 842                          *
 843                          * First of all, we scan fib_info list searching
 844                          * for stray nexthop entries, then ignite fib_flush.
 845                          */
 846                         if (fib_sync_down_addr(dev_net(dev), ifa->ifa_local))
 847                                 fib_flush(dev_net(dev));
 848                 }
 849         }
 850 #undef LOCAL_OK
 851 #undef BRD_OK
 852 #undef BRD0_OK
 853 #undef BRD1_OK
 854 }
 855
 856 static void nl_fib_lookup(struct fib_result_nl *frn, struct fib_table *tb)
 857 {
 858
 859         struct fib_result       res;
 860         struct flowi            fl = {
 861                 .mark = frn->fl_mark,
 862                 .nl_u = {
 863                         .ip4_u = {
 864                                 .daddr = frn->fl_addr,
 865                                 .tos = frn->fl_tos,
 866                                 .scope = frn->fl_scope
 867                         }
 868                 }
 869         };
 870
 871 #ifdef CONFIG_IP_MULTIPLE_TABLES
 872         res.r = NULL;
 873 #endif
 874
 875         frn->err = -ENOENT;
 876         if (tb) {
 877                 local_bh_disable();
 878
 879                 frn->tb_id = tb->tb_id;
 880                 rcu_read_lock();
 881                 frn->err = fib_table_lookup(tb, &fl, &res, FIB_LOOKUP_NOREF);
 882
 883                 if (!frn->err) {
 884                         frn->prefixlen = res.prefixlen;
 885                         frn->nh_sel = res.nh_sel;
 886                         frn->type = res.type;
 887                         frn->scope = res.scope;
 888                 }
 889                 rcu_read_unlock();
 890                 local_bh_enable();
 891         }
 892 }
 893
 894 static void nl_fib_input(struct sk_buff *skb)
 895 {
 896         struct net *net;
 897         struct fib_result_nl *frn;
 898         struct nlmsghdr *nlh;
 899         struct fib_table *tb;
 900         u32 pid;
 901
 902         net = sock_net(skb->sk);
 903         nlh = nlmsg_hdr(skb);
 904         if (skb->len < NLMSG_SPACE(0) || skb->len < nlh->nlmsg_len ||
 905             nlh->nlmsg_len < NLMSG_LENGTH(sizeof(*frn)))
 906                 return;
 907
 908         skb = skb_clone(skb, GFP_KERNEL);
 909         if (skb == NULL)
 910                 return;
 911         nlh = nlmsg_hdr(skb);
 912
 913         frn = (struct fib_result_nl *) NLMSG_DATA(nlh);
 914         tb = fib_get_table(net, frn->tb_id_in);
 915
 916         nl_fib_lookup(frn, tb);
 917
 918         pid = NETLINK_CB(skb).pid;      /* pid of sending process */
 919         NETLINK_CB(skb).pid = 0;        /* from kernel */
 920         NETLINK_CB(skb).dst_group = 0;  /* unicast */
 921         netlink_unicast(net->ipv4.fibnl, skb, pid, MSG_DONTWAIT);
 922 }
 923
 924 static int __net_init nl_fib_lookup_init(struct net *net)
 925 {
 926         struct sock *sk;
 927         sk = netlink_kernel_create(net, NETLINK_FIB_LOOKUP, 0,
 928                                    nl_fib_input, NULL, THIS_MODULE);
 929         if (sk == NULL)
 930                 return -EAFNOSUPPORT;
 931         net->ipv4.fibnl = sk;
 932         return 0;
 933 }
 934
 935 static void nl_fib_lookup_exit(struct net *net)
 936 {
 937         netlink_kernel_release(net->ipv4.fibnl);
 938         net->ipv4.fibnl = NULL;
 939 }
 940
 941 static void fib_disable_ip(struct net_device *dev, int force, int delay)
 942 {
 943         if (fib_sync_down_dev(dev, force))
 944                 fib_flush(dev_net(dev));
 945         rt_cache_flush(dev_net(dev), delay);
 946         arp_ifdown(dev);
 947 }
 948
 949 static int fib_inetaddr_event(struct notifier_block *this, unsigned long event, void *ptr)
 950 {
 951         struct in_ifaddr *ifa = (struct in_ifaddr *)ptr;
 952         struct net_device *dev = ifa->ifa_dev->dev;
 953
 954         switch (event) {
 955         case NETDEV_UP:
 956                 fib_add_ifaddr(ifa);
 957 #ifdef CONFIG_IP_ROUTE_MULTIPATH
 958                 fib_sync_up(dev);
 959 #endif
 960                 rt_cache_flush(dev_net(dev), -1);
 961                 break;
 962         case NETDEV_DOWN:
 963                 fib_del_ifaddr(ifa);
 964                 if (ifa->ifa_dev->ifa_list == NULL) {
 965                         /* Last address was deleted from this interface.
 966                          * Disable IP.
 967                          */
 968                         fib_disable_ip(dev, 1, 0);
 969                 } else {
 970                         rt_cache_flush(dev_net(dev), -1);
 971                 }
 972                 break;
 973         }
 974         return NOTIFY_DONE;
 975 }
 976
 977 static int fib_netdev_event(struct notifier_block *this, unsigned long event, void *ptr)
 978 {
 979         struct net_device *dev = ptr;
 980         struct in_device *in_dev = __in_dev_get_rtnl(dev);
 981
 982         if (event == NETDEV_UNREGISTER) {
 983                 fib_disable_ip(dev, 2, -1);
 984                 return NOTIFY_DONE;
 985         }
 986
 987         if (!in_dev)
 988                 return NOTIFY_DONE;
 989
 990         switch (event) {
 991         case NETDEV_UP:
 992                 for_ifa(in_dev) {
 993                         fib_add_ifaddr(ifa);
 994                 } endfor_ifa(in_dev);
 995 #ifdef CONFIG_IP_ROUTE_MULTIPATH
 996                 fib_sync_up(dev);
 997 #endif
 998                 rt_cache_flush(dev_net(dev), -1);
 999                 break;
1000         case NETDEV_DOWN:
1001                 fib_disable_ip(dev, 0, 0);
1002                 break;
1003         case NETDEV_CHANGEMTU:
1004         case NETDEV_CHANGE:
1005                 rt_cache_flush(dev_net(dev), 0);
1006                 break;
1007         case NETDEV_UNREGISTER_BATCH:
1008                 rt_cache_flush_batch();
1009                 break;
1010         }
1011         return NOTIFY_DONE;
1012 }
1013
1014 static struct notifier_block fib_inetaddr_notifier = {
1015         .notifier_call = fib_inetaddr_event,
1016 };
1017
1018 static struct notifier_block fib_netdev_notifier = {
1019         .notifier_call = fib_netdev_event,
1020 };
1021
1022 static int __net_init ip_fib_net_init(struct net *net)
1023 {
1024         int err;
1025         size_t size = sizeof(struct hlist_head) * FIB_TABLE_HASHSZ;
1026
1027         /* Avoid false sharing : Use at least a full cache line */
1028         size = max_t(size_t, size, L1_CACHE_BYTES);
1029
1030         net->ipv4.fib_table_hash = kzalloc(size, GFP_KERNEL);
1031         if (net->ipv4.fib_table_hash == NULL)
1032                 return -ENOMEM;
1033
1034         err = fib4_rules_init(net);
1035         if (err < 0)
1036                 goto fail;
1037         return 0;
1038
1039 fail:
1040         kfree(net->ipv4.fib_table_hash);
1041         return err;
1042 }
1043
1044 static void ip_fib_net_exit(struct net *net)
1045 {
1046         unsigned int i;
1047
1048 #ifdef CONFIG_IP_MULTIPLE_TABLES
1049         fib4_rules_exit(net);
1050 #endif
1051
1052         for (i = 0; i < FIB_TABLE_HASHSZ; i++) {
1053                 struct fib_table *tb;
1054                 struct hlist_head *head;
1055                 struct hlist_node *node, *tmp;
1056
1057                 head = &net->ipv4.fib_table_hash[i];
1058                 hlist_for_each_entry_safe(tb, node, tmp, head, tb_hlist) {
1059                         hlist_del(node);
1060                         fib_table_flush(tb);
1061                         fib_free_table(tb);
1062                 }
1063         }
1064         kfree(net->ipv4.fib_table_hash);
1065 }
1066
1067 static int __net_init fib_net_init(struct net *net)
1068 {
1069         int error;
1070
1071         error = ip_fib_net_init(net);
1072         if (error < 0)
1073                 goto out;
1074         error = nl_fib_lookup_init(net);
1075         if (error < 0)
1076                 goto out_nlfl;
1077         error = fib_proc_init(net);
1078         if (error < 0)
1079                 goto out_proc;
1080 out:
1081         return error;
1082
1083 out_proc:
1084         nl_fib_lookup_exit(net);
1085 out_nlfl:
1086         ip_fib_net_exit(net);
1087         goto out;
1088 }
1089
1090 static void __net_exit fib_net_exit(struct net *net)
1091 {
1092         fib_proc_exit(net);
1093         nl_fib_lookup_exit(net);
1094         ip_fib_net_exit(net);
1095 }
1096
1097 static struct pernet_operations fib_net_ops = {
1098         .init = fib_net_init,
1099         .exit = fib_net_exit,
1100 };
1101
1102 void __init ip_fib_init(void)
1103 {
1104         rtnl_register(PF_INET, RTM_NEWROUTE, inet_rtm_newroute, NULL);
1105         rtnl_register(PF_INET, RTM_DELROUTE, inet_rtm_delroute, NULL);
1106         rtnl_register(PF_INET, RTM_GETROUTE, NULL, inet_dump_fib);
1107
1108         register_pernet_subsys(&fib_net_ops);
1109         register_netdevice_notifier(&fib_netdev_notifier);
1110         register_inetaddr_notifier(&fib_inetaddr_notifier);
1111
1112         fib_hash_init();
1113 }