net/batman-adv/distributed-arp-table.c

   1 /* Copyright (C) 2011-2016  B.A.T.M.A.N. contributors:
   2  *
   3  * Antonio Quartulli
   4  *
   5  * This program is free software; you can redistribute it and/or
   6  * modify it under the terms of version 2 of the GNU General Public
   7  * License as published by the Free Software Foundation.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, see <http://www.gnu.org/licenses/>.
  16  */
  17
  18 #include "distributed-arp-table.h"
  19 #include "main.h"
  20
  21 #include <linux/atomic.h>
  22 #include <linux/bitops.h>
  23 #include <linux/byteorder/generic.h>
  24 #include <linux/errno.h>
  25 #include <linux/etherdevice.h>
  26 #include <linux/fs.h>
  27 #include <linux/if_arp.h>
  28 #include <linux/if_ether.h>
  29 #include <linux/if_vlan.h>
  30 #include <linux/in.h>
  31 #include <linux/jiffies.h>
  32 #include <linux/kernel.h>
  33 #include <linux/kref.h>
  34 #include <linux/list.h>
  35 #include <linux/rculist.h>
  36 #include <linux/rcupdate.h>
  37 #include <linux/seq_file.h>
  38 #include <linux/skbuff.h>
  39 #include <linux/slab.h>
  40 #include <linux/spinlock.h>
  41 #include <linux/stddef.h>
  42 #include <linux/string.h>
  43 #include <linux/workqueue.h>
  44 #include <net/arp.h>
  45
  46 #include "hard-interface.h"
  47 #include "hash.h"
  48 #include "log.h"
  49 #include "originator.h"
  50 #include "send.h"
  51 #include "translation-table.h"
  52 #include "tvlv.h"
  53
  54 static void batadv_dat_purge(struct work_struct *work);
  55
  56 /**
  57  * batadv_dat_start_timer - initialise the DAT periodic worker
  58  * @bat_priv: the bat priv with all the soft interface information
  59  */
  60 static void batadv_dat_start_timer(struct batadv_priv *bat_priv)
  61 {
  62         INIT_DELAYED_WORK(&bat_priv->dat.work, batadv_dat_purge);
  63         queue_delayed_work(batadv_event_workqueue, &bat_priv->dat.work,
  64                            msecs_to_jiffies(10000));
  65 }
  66
  67 /**
  68  * batadv_dat_entry_release - release dat_entry from lists and queue for free
  69  *  after rcu grace period
  70  * @ref: kref pointer of the dat_entry
  71  */
  72 static void batadv_dat_entry_release(struct kref *ref)
  73 {
  74         struct batadv_dat_entry *dat_entry;
  75
  76         dat_entry = container_of(ref, struct batadv_dat_entry, refcount);
  77
  78         kfree_rcu(dat_entry, rcu);
  79 }
  80
  81 /**
  82  * batadv_dat_entry_put - decrement the dat_entry refcounter and possibly
  83  *  release it
  84  * @dat_entry: dat_entry to be free'd
  85  */
  86 static void batadv_dat_entry_put(struct batadv_dat_entry *dat_entry)
  87 {
  88         kref_put(&dat_entry->refcount, batadv_dat_entry_release);
  89 }
  90
  91 /**
  92  * batadv_dat_to_purge - check whether a dat_entry has to be purged or not
  93  * @dat_entry: the entry to check
  94  *
  95  * Return: true if the entry has to be purged now, false otherwise.
  96  */
  97 static bool batadv_dat_to_purge(struct batadv_dat_entry *dat_entry)
  98 {
  99         return batadv_has_timed_out(dat_entry->last_update,
 100                                     BATADV_DAT_ENTRY_TIMEOUT);
 101 }
 102
 103 /**
 104  * __batadv_dat_purge - delete entries from the DAT local storage
 105  * @bat_priv: the bat priv with all the soft interface information
 106  * @to_purge: function in charge to decide whether an entry has to be purged or
 107  *            not. This function takes the dat_entry as argument and has to
 108  *            returns a boolean value: true is the entry has to be deleted,
 109  *            false otherwise
 110  *
 111  * Loops over each entry in the DAT local storage and deletes it if and only if
 112  * the to_purge function passed as argument returns true.
 113  */
 114 static void __batadv_dat_purge(struct batadv_priv *bat_priv,
 115                                bool (*to_purge)(struct batadv_dat_entry *))
 116 {
 117         spinlock_t *list_lock; /* protects write access to the hash lists */
 118         struct batadv_dat_entry *dat_entry;
 119         struct hlist_node *node_tmp;
 120         struct hlist_head *head;
 121         u32 i;
 122
 123         if (!bat_priv->dat.hash)
 124                 return;
 125
 126         for (i = 0; i < bat_priv->dat.hash->size; i++) {
 127                 head = &bat_priv->dat.hash->table[i];
 128                 list_lock = &bat_priv->dat.hash->list_locks[i];
 129
 130                 spin_lock_bh(list_lock);
 131                 hlist_for_each_entry_safe(dat_entry, node_tmp, head,
 132                                           hash_entry) {
 133                         /* if a helper function has been passed as parameter,
 134                          * ask it if the entry has to be purged or not
 135                          */
 136                         if (to_purge && !to_purge(dat_entry))
 137                                 continue;
 138
 139                         hlist_del_rcu(&dat_entry->hash_entry);
 140                         batadv_dat_entry_put(dat_entry);
 141                 }
 142                 spin_unlock_bh(list_lock);
 143         }
 144 }
 145
 146 /**
 147  * batadv_dat_purge - periodic task that deletes old entries from the local DAT
 148  * hash table
 149  * @work: kernel work struct
 150  */
 151 static void batadv_dat_purge(struct work_struct *work)
 152 {
 153         struct delayed_work *delayed_work;
 154         struct batadv_priv_dat *priv_dat;
 155         struct batadv_priv *bat_priv;
 156
 157         delayed_work = to_delayed_work(work);
 158         priv_dat = container_of(delayed_work, struct batadv_priv_dat, work);
 159         bat_priv = container_of(priv_dat, struct batadv_priv, dat);
 160
 161         __batadv_dat_purge(bat_priv, batadv_dat_to_purge);
 162         batadv_dat_start_timer(bat_priv);
 163 }
 164
 165 /**
 166  * batadv_compare_dat - comparing function used in the local DAT hash table
 167  * @node: node in the local table
 168  * @data2: second object to compare the node to
 169  *
 170  * Return: true if the two entries are the same, false otherwise.
 171  */
 172 static bool batadv_compare_dat(const struct hlist_node *node, const void *data2)
 173 {
 174         const void *data1 = container_of(node, struct batadv_dat_entry,
 175                                          hash_entry);
 176
 177         return memcmp(data1, data2, sizeof(__be32)) == 0;
 178 }
 179
 180 /**
 181  * batadv_arp_hw_src - extract the hw_src field from an ARP packet
 182  * @skb: ARP packet
 183  * @hdr_size: size of the possible header before the ARP packet
 184  *
 185  * Return: the value of the hw_src field in the ARP packet.
 186  */
 187 static u8 *batadv_arp_hw_src(struct sk_buff *skb, int hdr_size)
 188 {
 189         u8 *addr;
 190
 191         addr = (u8 *)(skb->data + hdr_size);
 192         addr += ETH_HLEN + sizeof(struct arphdr);
 193
 194         return addr;
 195 }
 196
 197 /**
 198  * batadv_arp_ip_src - extract the ip_src field from an ARP packet
 199  * @skb: ARP packet
 200  * @hdr_size: size of the possible header before the ARP packet
 201  *
 202  * Return: the value of the ip_src field in the ARP packet.
 203  */
 204 static __be32 batadv_arp_ip_src(struct sk_buff *skb, int hdr_size)
 205 {
 206         return *(__be32 *)(batadv_arp_hw_src(skb, hdr_size) + ETH_ALEN);
 207 }
 208
 209 /**
 210  * batadv_arp_hw_dst - extract the hw_dst field from an ARP packet
 211  * @skb: ARP packet
 212  * @hdr_size: size of the possible header before the ARP packet
 213  *
 214  * Return: the value of the hw_dst field in the ARP packet.
 215  */
 216 static u8 *batadv_arp_hw_dst(struct sk_buff *skb, int hdr_size)
 217 {
 218         return batadv_arp_hw_src(skb, hdr_size) + ETH_ALEN + 4;
 219 }
 220
 221 /**
 222  * batadv_arp_ip_dst - extract the ip_dst field from an ARP packet
 223  * @skb: ARP packet
 224  * @hdr_size: size of the possible header before the ARP packet
 225  *
 226  * Return: the value of the ip_dst field in the ARP packet.
 227  */
 228 static __be32 batadv_arp_ip_dst(struct sk_buff *skb, int hdr_size)
 229 {
 230         return *(__be32 *)(batadv_arp_hw_src(skb, hdr_size) + ETH_ALEN * 2 + 4);
 231 }
 232
 233 /**
 234  * batadv_hash_dat - compute the hash value for an IP address
 235  * @data: data to hash
 236  * @size: size of the hash table
 237  *
 238  * Return: the selected index in the hash table for the given data.
 239  */
 240 static u32 batadv_hash_dat(const void *data, u32 size)
 241 {
 242         u32 hash = 0;
 243         const struct batadv_dat_entry *dat = data;
 244         const unsigned char *key;
 245         __be16 vid;
 246         u32 i;
 247
 248         key = (const unsigned char *)&dat->ip;
 249         for (i = 0; i < sizeof(dat->ip); i++) {
 250                 hash += key[i];
 251                 hash += (hash << 10);
 252                 hash ^= (hash >> 6);
 253         }
 254
 255         vid = htons(dat->vid);
 256         key = (__force const unsigned char *)&vid;
 257         for (i = 0; i < sizeof(dat->vid); i++) {
 258                 hash += key[i];
 259                 hash += (hash << 10);
 260                 hash ^= (hash >> 6);
 261         }
 262
 263         hash += (hash << 3);
 264         hash ^= (hash >> 11);
 265         hash += (hash << 15);
 266
 267         return hash % size;
 268 }
 269
 270 /**
 271  * batadv_dat_entry_hash_find - look for a given dat_entry in the local hash
 272  * table
 273  * @bat_priv: the bat priv with all the soft interface information
 274  * @ip: search key
 275  * @vid: VLAN identifier
 276  *
 277  * Return: the dat_entry if found, NULL otherwise.
 278  */
 279 static struct batadv_dat_entry *
 280 batadv_dat_entry_hash_find(struct batadv_priv *bat_priv, __be32 ip,
 281                            unsigned short vid)
 282 {
 283         struct hlist_head *head;
 284         struct batadv_dat_entry to_find, *dat_entry, *dat_entry_tmp = NULL;
 285         struct batadv_hashtable *hash = bat_priv->dat.hash;
 286         u32 index;
 287
 288         if (!hash)
 289                 return NULL;
 290
 291         to_find.ip = ip;
 292         to_find.vid = vid;
 293
 294         index = batadv_hash_dat(&to_find, hash->size);
 295         head = &hash->table[index];
 296
 297         rcu_read_lock();
 298         hlist_for_each_entry_rcu(dat_entry, head, hash_entry) {
 299                 if (dat_entry->ip != ip)
 300                         continue;
 301
 302                 if (!kref_get_unless_zero(&dat_entry->refcount))
 303                         continue;
 304
 305                 dat_entry_tmp = dat_entry;
 306                 break;
 307         }
 308         rcu_read_unlock();
 309
 310         return dat_entry_tmp;
 311 }
 312
 313 /**
 314  * batadv_dat_entry_add - add a new dat entry or update it if already exists
 315  * @bat_priv: the bat priv with all the soft interface information
 316  * @ip: ipv4 to add/edit
 317  * @mac_addr: mac address to assign to the given ipv4
 318  * @vid: VLAN identifier
 319  */
 320 static void batadv_dat_entry_add(struct batadv_priv *bat_priv, __be32 ip,
 321                                  u8 *mac_addr, unsigned short vid)
 322 {
 323         struct batadv_dat_entry *dat_entry;
 324         int hash_added;
 325
 326         dat_entry = batadv_dat_entry_hash_find(bat_priv, ip, vid);
 327         /* if this entry is already known, just update it */
 328         if (dat_entry) {
 329                 if (!batadv_compare_eth(dat_entry->mac_addr, mac_addr))
 330                         ether_addr_copy(dat_entry->mac_addr, mac_addr);
 331                 dat_entry->last_update = jiffies;
 332                 batadv_dbg(BATADV_DBG_DAT, bat_priv,
 333                            "Entry updated: %pI4 %pM (vid: %d)\n",
 334                            &dat_entry->ip, dat_entry->mac_addr,
 335                            BATADV_PRINT_VID(vid));
 336                 goto out;
 337         }
 338
 339         dat_entry = kmalloc(sizeof(*dat_entry), GFP_ATOMIC);
 340         if (!dat_entry)
 341                 goto out;
 342
 343         dat_entry->ip = ip;
 344         dat_entry->vid = vid;
 345         ether_addr_copy(dat_entry->mac_addr, mac_addr);
 346         dat_entry->last_update = jiffies;
 347         kref_init(&dat_entry->refcount);
 348
 349         kref_get(&dat_entry->refcount);
 350         hash_added = batadv_hash_add(bat_priv->dat.hash, batadv_compare_dat,
 351                                      batadv_hash_dat, dat_entry,
 352                                      &dat_entry->hash_entry);
 353
 354         if (unlikely(hash_added != 0)) {
 355                 /* remove the reference for the hash */
 356                 batadv_dat_entry_put(dat_entry);
 357                 goto out;
 358         }
 359
 360         batadv_dbg(BATADV_DBG_DAT, bat_priv, "New entry added: %pI4 %pM (vid: %d)\n",
 361                    &dat_entry->ip, dat_entry->mac_addr, BATADV_PRINT_VID(vid));
 362
 363 out:
 364         if (dat_entry)
 365                 batadv_dat_entry_put(dat_entry);
 366 }
 367
 368 #ifdef CONFIG_BATMAN_ADV_DEBUG
 369
 370 /**
 371  * batadv_dbg_arp - print a debug message containing all the ARP packet details
 372  * @bat_priv: the bat priv with all the soft interface information
 373  * @skb: ARP packet
 374  * @type: ARP type
 375  * @hdr_size: size of the possible header before the ARP packet
 376  * @msg: message to print together with the debugging information
 377  */
 378 static void batadv_dbg_arp(struct batadv_priv *bat_priv, struct sk_buff *skb,
 379                            u16 type, int hdr_size, char *msg)
 380 {
 381         struct batadv_unicast_4addr_packet *unicast_4addr_packet;
 382         struct batadv_bcast_packet *bcast_pkt;
 383         u8 *orig_addr;
 384         __be32 ip_src, ip_dst;
 385
 386         if (msg)
 387                 batadv_dbg(BATADV_DBG_DAT, bat_priv, "%s\n", msg);
 388
 389         ip_src = batadv_arp_ip_src(skb, hdr_size);
 390         ip_dst = batadv_arp_ip_dst(skb, hdr_size);
 391         batadv_dbg(BATADV_DBG_DAT, bat_priv,
 392                    "ARP MSG = [src: %pM-%pI4 dst: %pM-%pI4]\n",
 393                    batadv_arp_hw_src(skb, hdr_size), &ip_src,
 394                    batadv_arp_hw_dst(skb, hdr_size), &ip_dst);
 395
 396         if (hdr_size < sizeof(struct batadv_unicast_packet))
 397                 return;
 398
 399         unicast_4addr_packet = (struct batadv_unicast_4addr_packet *)skb->data;
 400
 401         switch (unicast_4addr_packet->u.packet_type) {
 402         case BATADV_UNICAST:
 403                 batadv_dbg(BATADV_DBG_DAT, bat_priv,
 404                            "* encapsulated within a UNICAST packet\n");
 405                 break;
 406         case BATADV_UNICAST_4ADDR:
 407                 batadv_dbg(BATADV_DBG_DAT, bat_priv,
 408                            "* encapsulated within a UNICAST_4ADDR packet (src: %pM)\n",
 409                            unicast_4addr_packet->src);
 410                 switch (unicast_4addr_packet->subtype) {
 411                 case BATADV_P_DAT_DHT_PUT:
 412                         batadv_dbg(BATADV_DBG_DAT, bat_priv, "* type: DAT_DHT_PUT\n");
 413                         break;
 414                 case BATADV_P_DAT_DHT_GET:
 415                         batadv_dbg(BATADV_DBG_DAT, bat_priv, "* type: DAT_DHT_GET\n");
 416                         break;
 417                 case BATADV_P_DAT_CACHE_REPLY:
 418                         batadv_dbg(BATADV_DBG_DAT, bat_priv,
 419                                    "* type: DAT_CACHE_REPLY\n");
 420                         break;
 421                 case BATADV_P_DATA:
 422                         batadv_dbg(BATADV_DBG_DAT, bat_priv, "* type: DATA\n");
 423                         break;
 424                 default:
 425                         batadv_dbg(BATADV_DBG_DAT, bat_priv, "* type: Unknown (%u)!\n",
 426                                    unicast_4addr_packet->u.packet_type);
 427                 }
 428                 break;
 429         case BATADV_BCAST:
 430                 bcast_pkt = (struct batadv_bcast_packet *)unicast_4addr_packet;
 431                 orig_addr = bcast_pkt->orig;
 432                 batadv_dbg(BATADV_DBG_DAT, bat_priv,
 433                            "* encapsulated within a BCAST packet (src: %pM)\n",
 434                            orig_addr);
 435                 break;
 436         default:
 437                 batadv_dbg(BATADV_DBG_DAT, bat_priv,
 438                            "* encapsulated within an unknown packet type (0x%x)\n",
 439                            unicast_4addr_packet->u.packet_type);
 440         }
 441 }
 442
 443 #else
 444
 445 static void batadv_dbg_arp(struct batadv_priv *bat_priv, struct sk_buff *skb,
 446                            u16 type, int hdr_size, char *msg)
 447 {
 448 }
 449
 450 #endif /* CONFIG_BATMAN_ADV_DEBUG */
 451
 452 /**
 453  * batadv_is_orig_node_eligible - check whether a node can be a DHT candidate
 454  * @res: the array with the already selected candidates
 455  * @select: number of already selected candidates
 456  * @tmp_max: address of the currently evaluated node
 457  * @max: current round max address
 458  * @last_max: address of the last selected candidate
 459  * @candidate: orig_node under evaluation
 460  * @max_orig_node: last selected candidate
 461  *
 462  * Return: true if the node has been elected as next candidate or false
 463  * otherwise.
 464  */
 465 static bool batadv_is_orig_node_eligible(struct batadv_dat_candidate *res,
 466                                          int select, batadv_dat_addr_t tmp_max,
 467                                          batadv_dat_addr_t max,
 468                                          batadv_dat_addr_t last_max,
 469                                          struct batadv_orig_node *candidate,
 470                                          struct batadv_orig_node *max_orig_node)
 471 {
 472         bool ret = false;
 473         int j;
 474
 475         /* check if orig node candidate is running DAT */
 476         if (!test_bit(BATADV_ORIG_CAPA_HAS_DAT, &candidate->capabilities))
 477                 goto out;
 478
 479         /* Check if this node has already been selected... */
 480         for (j = 0; j < select; j++)
 481                 if (res[j].orig_node == candidate)
 482                         break;
 483         /* ..and possibly skip it */
 484         if (j < select)
 485                 goto out;
 486         /* sanity check: has it already been selected? This should not happen */
 487         if (tmp_max > last_max)
 488                 goto out;
 489         /* check if during this iteration an originator with a closer dht
 490          * address has already been found
 491          */
 492         if (tmp_max < max)
 493                 goto out;
 494         /* this is an hash collision with the temporary selected node. Choose
 495          * the one with the lowest address
 496          */
 497         if ((tmp_max == max) && max_orig_node &&
 498             (batadv_compare_eth(candidate->orig, max_orig_node->orig) > 0))
 499                 goto out;
 500
 501         ret = true;
 502 out:
 503         return ret;
 504 }
 505
 506 /**
 507  * batadv_choose_next_candidate - select the next DHT candidate
 508  * @bat_priv: the bat priv with all the soft interface information
 509  * @cands: candidates array
 510  * @select: number of candidates already present in the array
 511  * @ip_key: key to look up in the DHT
 512  * @last_max: pointer where the address of the selected candidate will be saved
 513  */
 514 static void batadv_choose_next_candidate(struct batadv_priv *bat_priv,
 515                                          struct batadv_dat_candidate *cands,
 516                                          int select, batadv_dat_addr_t ip_key,
 517                                          batadv_dat_addr_t *last_max)
 518 {
 519         batadv_dat_addr_t max = 0;
 520         batadv_dat_addr_t tmp_max = 0;
 521         struct batadv_orig_node *orig_node, *max_orig_node = NULL;
 522         struct batadv_hashtable *hash = bat_priv->orig_hash;
 523         struct hlist_head *head;
 524         int i;
 525
 526         /* if no node is eligible as candidate, leave the candidate type as
 527          * NOT_FOUND
 528          */
 529         cands[select].type = BATADV_DAT_CANDIDATE_NOT_FOUND;
 530
 531         /* iterate over the originator list and find the node with the closest
 532          * dat_address which has not been selected yet
 533          */
 534         for (i = 0; i < hash->size; i++) {
 535                 head = &hash->table[i];
 536
 537                 rcu_read_lock();
 538                 hlist_for_each_entry_rcu(orig_node, head, hash_entry) {
 539                         /* the dht space is a ring using unsigned addresses */
 540                         tmp_max = BATADV_DAT_ADDR_MAX - orig_node->dat_addr +
 541                                   ip_key;
 542
 543                         if (!batadv_is_orig_node_eligible(cands, select,
 544                                                           tmp_max, max,
 545                                                           *last_max, orig_node,
 546                                                           max_orig_node))
 547                                 continue;
 548
 549                         if (!kref_get_unless_zero(&orig_node->refcount))
 550                                 continue;
 551
 552                         max = tmp_max;
 553                         if (max_orig_node)
 554                                 batadv_orig_node_put(max_orig_node);
 555                         max_orig_node = orig_node;
 556                 }
 557                 rcu_read_unlock();
 558         }
 559         if (max_orig_node) {
 560                 cands[select].type = BATADV_DAT_CANDIDATE_ORIG;
 561                 cands[select].orig_node = max_orig_node;
 562                 batadv_dbg(BATADV_DBG_DAT, bat_priv,
 563                            "dat_select_candidates() %d: selected %pM addr=%u dist=%u\n",
 564                            select, max_orig_node->orig, max_orig_node->dat_addr,
 565                            max);
 566         }
 567         *last_max = max;
 568 }
 569
 570 /**
 571  * batadv_dat_select_candidates - select the nodes which the DHT message has to
 572  * be sent to
 573  * @bat_priv: the bat priv with all the soft interface information
 574  * @ip_dst: ipv4 to look up in the DHT
 575  * @vid: VLAN identifier
 576  *
 577  * An originator O is selected if and only if its DHT_ID value is one of three
 578  * closest values (from the LEFT, with wrap around if needed) then the hash
 579  * value of the key. ip_dst is the key.
 580  *
 581  * Return: the candidate array of size BATADV_DAT_CANDIDATE_NUM.
 582  */
 583 static struct batadv_dat_candidate *
 584 batadv_dat_select_candidates(struct batadv_priv *bat_priv, __be32 ip_dst,
 585                              unsigned short vid)
 586 {
 587         int select;
 588         batadv_dat_addr_t last_max = BATADV_DAT_ADDR_MAX, ip_key;
 589         struct batadv_dat_candidate *res;
 590         struct batadv_dat_entry dat;
 591
 592         if (!bat_priv->orig_hash)
 593                 return NULL;
 594
 595         res = kmalloc_array(BATADV_DAT_CANDIDATES_NUM, sizeof(*res),
 596                             GFP_ATOMIC);
 597         if (!res)
 598                 return NULL;
 599
 600         dat.ip = ip_dst;
 601         dat.vid = vid;
 602         ip_key = (batadv_dat_addr_t)batadv_hash_dat(&dat,
 603                                                     BATADV_DAT_ADDR_MAX);
 604
 605         batadv_dbg(BATADV_DBG_DAT, bat_priv,
 606                    "dat_select_candidates(): IP=%pI4 hash(IP)=%u\n", &ip_dst,
 607                    ip_key);
 608
 609         for (select = 0; select < BATADV_DAT_CANDIDATES_NUM; select++)
 610                 batadv_choose_next_candidate(bat_priv, res, select, ip_key,
 611                                              &last_max);
 612
 613         return res;
 614 }
 615
 616 /**
 617  * batadv_dat_send_data - send a payload to the selected candidates
 618  * @bat_priv: the bat priv with all the soft interface information
 619  * @skb: payload to send
 620  * @ip: the DHT key
 621  * @vid: VLAN identifier
 622  * @packet_subtype: unicast4addr packet subtype to use
 623  *
 624  * This function copies the skb with pskb_copy() and is sent as unicast packet
 625  * to each of the selected candidates.
 626  *
 627  * Return: true if the packet is sent to at least one candidate, false
 628  * otherwise.
 629  */
 630 static bool batadv_dat_send_data(struct batadv_priv *bat_priv,
 631                                  struct sk_buff *skb, __be32 ip,
 632                                  unsigned short vid, int packet_subtype)
 633 {
 634         int i;
 635         bool ret = false;
 636         int send_status;
 637         struct batadv_neigh_node *neigh_node = NULL;
 638         struct sk_buff *tmp_skb;
 639         struct batadv_dat_candidate *cand;
 640
 641         cand = batadv_dat_select_candidates(bat_priv, ip, vid);
 642         if (!cand)
 643                 goto out;
 644
 645         batadv_dbg(BATADV_DBG_DAT, bat_priv, "DHT_SEND for %pI4\n", &ip);
 646
 647         for (i = 0; i < BATADV_DAT_CANDIDATES_NUM; i++) {
 648                 if (cand[i].type == BATADV_DAT_CANDIDATE_NOT_FOUND)
 649                         continue;
 650
 651                 neigh_node = batadv_orig_router_get(cand[i].orig_node,
 652                                                     BATADV_IF_DEFAULT);
 653                 if (!neigh_node)
 654                         goto free_orig;
 655
 656                 tmp_skb = pskb_copy_for_clone(skb, GFP_ATOMIC);
 657                 if (!batadv_send_skb_prepare_unicast_4addr(bat_priv, tmp_skb,
 658                                                            cand[i].orig_node,
 659                                                            packet_subtype)) {
 660                         kfree_skb(tmp_skb);
 661                         goto free_neigh;
 662                 }
 663
 664                 send_status = batadv_send_unicast_skb(tmp_skb, neigh_node);
 665                 if (send_status == NET_XMIT_SUCCESS) {
 666                         /* count the sent packet */
 667                         switch (packet_subtype) {
 668                         case BATADV_P_DAT_DHT_GET:
 669                                 batadv_inc_counter(bat_priv,
 670                                                    BATADV_CNT_DAT_GET_TX);
 671                                 break;
 672                         case BATADV_P_DAT_DHT_PUT:
 673                                 batadv_inc_counter(bat_priv,
 674                                                    BATADV_CNT_DAT_PUT_TX);
 675                                 break;
 676                         }
 677
 678                         /* packet sent to a candidate: return true */
 679                         ret = true;
 680                 }
 681 free_neigh:
 682                 batadv_neigh_node_put(neigh_node);
 683 free_orig:
 684                 batadv_orig_node_put(cand[i].orig_node);
 685         }
 686
 687 out:
 688         kfree(cand);
 689         return ret;
 690 }
 691
 692 /**
 693  * batadv_dat_tvlv_container_update - update the dat tvlv container after dat
 694  *  setting change
 695  * @bat_priv: the bat priv with all the soft interface information
 696  */
 697 static void batadv_dat_tvlv_container_update(struct batadv_priv *bat_priv)
 698 {
 699         char dat_mode;
 700
 701         dat_mode = atomic_read(&bat_priv->distributed_arp_table);
 702
 703         switch (dat_mode) {
 704         case 0:
 705                 batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_DAT, 1);
 706                 break;
 707         case 1:
 708                 batadv_tvlv_container_register(bat_priv, BATADV_TVLV_DAT, 1,
 709                                                NULL, 0);
 710                 break;
 711         }
 712 }
 713
 714 /**
 715  * batadv_dat_status_update - update the dat tvlv container after dat
 716  *  setting change
 717  * @net_dev: the soft interface net device
 718  */
 719 void batadv_dat_status_update(struct net_device *net_dev)
 720 {
 721         struct batadv_priv *bat_priv = netdev_priv(net_dev);
 722
 723         batadv_dat_tvlv_container_update(bat_priv);
 724 }
 725
 726 /**
 727  * batadv_dat_tvlv_ogm_handler_v1 - process incoming dat tvlv container
 728  * @bat_priv: the bat priv with all the soft interface information
 729  * @orig: the orig_node of the ogm
 730  * @flags: flags indicating the tvlv state (see batadv_tvlv_handler_flags)
 731  * @tvlv_value: tvlv buffer containing the gateway data
 732  * @tvlv_value_len: tvlv buffer length
 733  */
 734 static void batadv_dat_tvlv_ogm_handler_v1(struct batadv_priv *bat_priv,
 735                                            struct batadv_orig_node *orig,
 736                                            u8 flags,
 737                                            void *tvlv_value, u16 tvlv_value_len)
 738 {
 739         if (flags & BATADV_TVLV_HANDLER_OGM_CIFNOTFND)
 740                 clear_bit(BATADV_ORIG_CAPA_HAS_DAT, &orig->capabilities);
 741         else
 742                 set_bit(BATADV_ORIG_CAPA_HAS_DAT, &orig->capabilities);
 743 }
 744
 745 /**
 746  * batadv_dat_hash_free - free the local DAT hash table
 747  * @bat_priv: the bat priv with all the soft interface information
 748  */
 749 static void batadv_dat_hash_free(struct batadv_priv *bat_priv)
 750 {
 751         if (!bat_priv->dat.hash)
 752                 return;
 753
 754         __batadv_dat_purge(bat_priv, NULL);
 755
 756         batadv_hash_destroy(bat_priv->dat.hash);
 757
 758         bat_priv->dat.hash = NULL;
 759 }
 760
 761 /**
 762  * batadv_dat_init - initialise the DAT internals
 763  * @bat_priv: the bat priv with all the soft interface information
 764  *
 765  * Return: 0 in case of success, a negative error code otherwise
 766  */
 767 int batadv_dat_init(struct batadv_priv *bat_priv)
 768 {
 769         if (bat_priv->dat.hash)
 770                 return 0;
 771
 772         bat_priv->dat.hash = batadv_hash_new(1024);
 773
 774         if (!bat_priv->dat.hash)
 775                 return -ENOMEM;
 776
 777         batadv_dat_start_timer(bat_priv);
 778
 779         batadv_tvlv_handler_register(bat_priv, batadv_dat_tvlv_ogm_handler_v1,
 780                                      NULL, BATADV_TVLV_DAT, 1,
 781                                      BATADV_TVLV_HANDLER_OGM_CIFNOTFND);
 782         batadv_dat_tvlv_container_update(bat_priv);
 783         return 0;
 784 }
 785
 786 /**
 787  * batadv_dat_free - free the DAT internals
 788  * @bat_priv: the bat priv with all the soft interface information
 789  */
 790 void batadv_dat_free(struct batadv_priv *bat_priv)
 791 {
 792         batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_DAT, 1);
 793         batadv_tvlv_handler_unregister(bat_priv, BATADV_TVLV_DAT, 1);
 794
 795         cancel_delayed_work_sync(&bat_priv->dat.work);
 796
 797         batadv_dat_hash_free(bat_priv);
 798 }
 799
 800 #ifdef CONFIG_BATMAN_ADV_DEBUGFS
 801 /**
 802  * batadv_dat_cache_seq_print_text - print the local DAT hash table
 803  * @seq: seq file to print on
 804  * @offset: not used
 805  *
 806  * Return: always 0
 807  */
 808 int batadv_dat_cache_seq_print_text(struct seq_file *seq, void *offset)
 809 {
 810         struct net_device *net_dev = (struct net_device *)seq->private;
 811         struct batadv_priv *bat_priv = netdev_priv(net_dev);
 812         struct batadv_hashtable *hash = bat_priv->dat.hash;
 813         struct batadv_dat_entry *dat_entry;
 814         struct batadv_hard_iface *primary_if;
 815         struct hlist_head *head;
 816         unsigned long last_seen_jiffies;
 817         int last_seen_msecs, last_seen_secs, last_seen_mins;
 818         u32 i;
 819
 820         primary_if = batadv_seq_print_text_primary_if_get(seq);
 821         if (!primary_if)
 822                 goto out;
 823
 824         seq_printf(seq, "Distributed ARP Table (%s):\n", net_dev->name);
 825         seq_puts(seq,
 826                  "          IPv4             MAC        VID   last-seen\n");
 827
 828         for (i = 0; i < hash->size; i++) {
 829                 head = &hash->table[i];
 830
 831                 rcu_read_lock();
 832                 hlist_for_each_entry_rcu(dat_entry, head, hash_entry) {
 833                         last_seen_jiffies = jiffies - dat_entry->last_update;
 834                         last_seen_msecs = jiffies_to_msecs(last_seen_jiffies);
 835                         last_seen_mins = last_seen_msecs / 60000;
 836                         last_seen_msecs = last_seen_msecs % 60000;
 837                         last_seen_secs = last_seen_msecs / 1000;
 838
 839                         seq_printf(seq, " * %15pI4 %14pM %4i %6i:%02i\n",
 840                                    &dat_entry->ip, dat_entry->mac_addr,
 841                                    BATADV_PRINT_VID(dat_entry->vid),
 842                                    last_seen_mins, last_seen_secs);
 843                 }
 844                 rcu_read_unlock();
 845         }
 846
 847 out:
 848         if (primary_if)
 849                 batadv_hardif_put(primary_if);
 850         return 0;
 851 }
 852 #endif
 853
 854 /**
 855  * batadv_arp_get_type - parse an ARP packet and gets the type
 856  * @bat_priv: the bat priv with all the soft interface information
 857  * @skb: packet to analyse
 858  * @hdr_size: size of the possible header before the ARP packet in the skb
 859  *
 860  * Return: the ARP type if the skb contains a valid ARP packet, 0 otherwise.
 861  */
 862 static u16 batadv_arp_get_type(struct batadv_priv *bat_priv,
 863                                struct sk_buff *skb, int hdr_size)
 864 {
 865         struct arphdr *arphdr;
 866         struct ethhdr *ethhdr;
 867         __be32 ip_src, ip_dst;
 868         u8 *hw_src, *hw_dst;
 869         u16 type = 0;
 870
 871         /* pull the ethernet header */
 872         if (unlikely(!pskb_may_pull(skb, hdr_size + ETH_HLEN)))
 873                 goto out;
 874
 875         ethhdr = (struct ethhdr *)(skb->data + hdr_size);
 876
 877         if (ethhdr->h_proto != htons(ETH_P_ARP))
 878                 goto out;
 879
 880         /* pull the ARP payload */
 881         if (unlikely(!pskb_may_pull(skb, hdr_size + ETH_HLEN +
 882                                     arp_hdr_len(skb->dev))))
 883                 goto out;
 884
 885         arphdr = (struct arphdr *)(skb->data + hdr_size + ETH_HLEN);
 886
 887         /* check whether the ARP packet carries a valid IP information */
 888         if (arphdr->ar_hrd != htons(ARPHRD_ETHER))
 889                 goto out;
 890
 891         if (arphdr->ar_pro != htons(ETH_P_IP))
 892                 goto out;
 893
 894         if (arphdr->ar_hln != ETH_ALEN)
 895                 goto out;
 896
 897         if (arphdr->ar_pln != 4)
 898                 goto out;
 899
 900         /* Check for bad reply/request. If the ARP message is not sane, DAT
 901          * will simply ignore it
 902          */
 903         ip_src = batadv_arp_ip_src(skb, hdr_size);
 904         ip_dst = batadv_arp_ip_dst(skb, hdr_size);
 905         if (ipv4_is_loopback(ip_src) || ipv4_is_multicast(ip_src) ||
 906             ipv4_is_loopback(ip_dst) || ipv4_is_multicast(ip_dst) ||
 907             ipv4_is_zeronet(ip_src) || ipv4_is_lbcast(ip_src) ||
 908             ipv4_is_zeronet(ip_dst) || ipv4_is_lbcast(ip_dst))
 909                 goto out;
 910
 911         hw_src = batadv_arp_hw_src(skb, hdr_size);
 912         if (is_zero_ether_addr(hw_src) || is_multicast_ether_addr(hw_src))
 913                 goto out;
 914
 915         /* don't care about the destination MAC address in ARP requests */
 916         if (arphdr->ar_op != htons(ARPOP_REQUEST)) {
 917                 hw_dst = batadv_arp_hw_dst(skb, hdr_size);
 918                 if (is_zero_ether_addr(hw_dst) ||
 919                     is_multicast_ether_addr(hw_dst))
 920                         goto out;
 921         }
 922
 923         type = ntohs(arphdr->ar_op);
 924 out:
 925         return type;
 926 }
 927
 928 /**
 929  * batadv_dat_get_vid - extract the VLAN identifier from skb if any
 930  * @skb: the buffer containing the packet to extract the VID from
 931  * @hdr_size: the size of the batman-adv header encapsulating the packet
 932  *
 933  * Return: If the packet embedded in the skb is vlan tagged this function
 934  * returns the VID with the BATADV_VLAN_HAS_TAG flag. Otherwise BATADV_NO_FLAGS
 935  * is returned.
 936  */
 937 static unsigned short batadv_dat_get_vid(struct sk_buff *skb, int *hdr_size)
 938 {
 939         unsigned short vid;
 940
 941         vid = batadv_get_vid(skb, *hdr_size);
 942
 943         /* ARP parsing functions jump forward of hdr_size + ETH_HLEN.
 944          * If the header contained in the packet is a VLAN one (which is longer)
 945          * hdr_size is updated so that the functions will still skip the
 946          * correct amount of bytes.
 947          */
 948         if (vid & BATADV_VLAN_HAS_TAG)
 949                 *hdr_size += VLAN_HLEN;
 950
 951         return vid;
 952 }
 953
 954 /**
 955  * batadv_dat_snoop_outgoing_arp_request - snoop the ARP request and try to
 956  * answer using DAT
 957  * @bat_priv: the bat priv with all the soft interface information
 958  * @skb: packet to check
 959  *
 960  * Return: true if the message has been sent to the dht candidates, false
 961  * otherwise. In case of a positive return value the message has to be enqueued
 962  * to permit the fallback.
 963  */
 964 bool batadv_dat_snoop_outgoing_arp_request(struct batadv_priv *bat_priv,
 965                                            struct sk_buff *skb)
 966 {
 967         u16 type = 0;
 968         __be32 ip_dst, ip_src;
 969         u8 *hw_src;
 970         bool ret = false;
 971         struct batadv_dat_entry *dat_entry = NULL;
 972         struct sk_buff *skb_new;
 973         int hdr_size = 0;
 974         unsigned short vid;
 975
 976         if (!atomic_read(&bat_priv->distributed_arp_table))
 977                 goto out;
 978
 979         vid = batadv_dat_get_vid(skb, &hdr_size);
 980
 981         type = batadv_arp_get_type(bat_priv, skb, hdr_size);
 982         /* If the node gets an ARP_REQUEST it has to send a DHT_GET unicast
 983          * message to the selected DHT candidates
 984          */
 985         if (type != ARPOP_REQUEST)
 986                 goto out;
 987
 988         batadv_dbg_arp(bat_priv, skb, type, hdr_size,
 989                        "Parsing outgoing ARP REQUEST");
 990
 991         ip_src = batadv_arp_ip_src(skb, hdr_size);
 992         hw_src = batadv_arp_hw_src(skb, hdr_size);
 993         ip_dst = batadv_arp_ip_dst(skb, hdr_size);
 994
 995         batadv_dat_entry_add(bat_priv, ip_src, hw_src, vid);
 996
 997         dat_entry = batadv_dat_entry_hash_find(bat_priv, ip_dst, vid);
 998         if (dat_entry) {
 999                 /* If the ARP request is destined for a local client the local
1000                  * client will answer itself. DAT would only generate a
1001                  * duplicate packet.
1002                  *
1003                  * Moreover, if the soft-interface is enslaved into a bridge, an
1004                  * additional DAT answer may trigger kernel warnings about
1005                  * a packet coming from the wrong port.
1006                  */
1007                 if (batadv_is_my_client(bat_priv, dat_entry->mac_addr, vid)) {
1008                         ret = true;
1009                         goto out;
1010                 }
1011
1012                 skb_new = arp_create(ARPOP_REPLY, ETH_P_ARP, ip_src,
1013                                      bat_priv->soft_iface, ip_dst, hw_src,
1014                                      dat_entry->mac_addr, hw_src);
1015                 if (!skb_new)
1016                         goto out;
1017
1018                 if (vid & BATADV_VLAN_HAS_TAG) {
1019                         skb_new = vlan_insert_tag(skb_new, htons(ETH_P_8021Q),
1020                                                   vid & VLAN_VID_MASK);
1021                         if (!skb_new)
1022                                 goto out;
1023                 }
1024
1025                 skb_reset_mac_header(skb_new);
1026                 skb_new->protocol = eth_type_trans(skb_new,
1027                                                    bat_priv->soft_iface);
1028                 batadv_inc_counter(bat_priv, BATADV_CNT_RX);
1029                 batadv_add_counter(bat_priv, BATADV_CNT_RX_BYTES,
1030                                    skb->len + ETH_HLEN + hdr_size);
1031                 bat_priv->soft_iface->last_rx = jiffies;
1032
1033                 netif_rx(skb_new);
1034                 batadv_dbg(BATADV_DBG_DAT, bat_priv, "ARP request replied locally\n");
1035                 ret = true;
1036         } else {
1037                 /* Send the request to the DHT */
1038                 ret = batadv_dat_send_data(bat_priv, skb, ip_dst, vid,
1039                                            BATADV_P_DAT_DHT_GET);
1040         }
1041 out:
1042         if (dat_entry)
1043                 batadv_dat_entry_put(dat_entry);
1044         return ret;
1045 }
1046
1047 /**
1048  * batadv_dat_snoop_incoming_arp_request - snoop the ARP request and try to
1049  * answer using the local DAT storage
1050  * @bat_priv: the bat priv with all the soft interface information
1051  * @skb: packet to check
1052  * @hdr_size: size of the encapsulation header
1053  *
1054  * Return: true if the request has been answered, false otherwise.
1055  */
1056 bool batadv_dat_snoop_incoming_arp_request(struct batadv_priv *bat_priv,
1057                                            struct sk_buff *skb, int hdr_size)
1058 {
1059         u16 type;
1060         __be32 ip_src, ip_dst;
1061         u8 *hw_src;
1062         struct sk_buff *skb_new;
1063         struct batadv_dat_entry *dat_entry = NULL;
1064         bool ret = false;
1065         unsigned short vid;
1066         int err;
1067
1068         if (!atomic_read(&bat_priv->distributed_arp_table))
1069                 goto out;
1070
1071         vid = batadv_dat_get_vid(skb, &hdr_size);
1072
1073         type = batadv_arp_get_type(bat_priv, skb, hdr_size);
1074         if (type != ARPOP_REQUEST)
1075                 goto out;
1076
1077         hw_src = batadv_arp_hw_src(skb, hdr_size);
1078         ip_src = batadv_arp_ip_src(skb, hdr_size);
1079         ip_dst = batadv_arp_ip_dst(skb, hdr_size);
1080
1081         batadv_dbg_arp(bat_priv, skb, type, hdr_size,
1082                        "Parsing incoming ARP REQUEST");
1083
1084         batadv_dat_entry_add(bat_priv, ip_src, hw_src, vid);
1085
1086         dat_entry = batadv_dat_entry_hash_find(bat_priv, ip_dst, vid);
1087         if (!dat_entry)
1088                 goto out;
1089
1090         skb_new = arp_create(ARPOP_REPLY, ETH_P_ARP, ip_src,
1091                              bat_priv->soft_iface, ip_dst, hw_src,
1092                              dat_entry->mac_addr, hw_src);
1093
1094         if (!skb_new)
1095                 goto out;
1096
1097         /* the rest of the TX path assumes that the mac_header offset pointing
1098          * to the inner Ethernet header has been set, therefore reset it now.
1099          */
1100         skb_reset_mac_header(skb_new);
1101
1102         if (vid & BATADV_VLAN_HAS_TAG) {
1103                 skb_new = vlan_insert_tag(skb_new, htons(ETH_P_8021Q),
1104                                           vid & VLAN_VID_MASK);
1105                 if (!skb_new)
1106                         goto out;
1107         }
1108
1109         /* To preserve backwards compatibility, the node has choose the outgoing
1110          * format based on the incoming request packet type. The assumption is
1111          * that a node not using the 4addr packet format doesn't support it.
1112          */
1113         if (hdr_size == sizeof(struct batadv_unicast_4addr_packet))
1114                 err = batadv_send_skb_via_tt_4addr(bat_priv, skb_new,
1115                                                    BATADV_P_DAT_CACHE_REPLY,
1116                                                    NULL, vid);
1117         else
1118                 err = batadv_send_skb_via_tt(bat_priv, skb_new, NULL, vid);
1119
1120         if (err != NET_XMIT_DROP) {
1121                 batadv_inc_counter(bat_priv, BATADV_CNT_DAT_CACHED_REPLY_TX);
1122                 ret = true;
1123         }
1124 out:
1125         if (dat_entry)
1126                 batadv_dat_entry_put(dat_entry);
1127         if (ret)
1128                 kfree_skb(skb);
1129         return ret;
1130 }
1131
1132 /**
1133  * batadv_dat_snoop_outgoing_arp_reply - snoop the ARP reply and fill the DHT
1134  * @bat_priv: the bat priv with all the soft interface information
1135  * @skb: packet to check
1136  */
1137 void batadv_dat_snoop_outgoing_arp_reply(struct batadv_priv *bat_priv,
1138                                          struct sk_buff *skb)
1139 {
1140         u16 type;
1141         __be32 ip_src, ip_dst;
1142         u8 *hw_src, *hw_dst;
1143         int hdr_size = 0;
1144         unsigned short vid;
1145
1146         if (!atomic_read(&bat_priv->distributed_arp_table))
1147                 return;
1148
1149         vid = batadv_dat_get_vid(skb, &hdr_size);
1150
1151         type = batadv_arp_get_type(bat_priv, skb, hdr_size);
1152         if (type != ARPOP_REPLY)
1153                 return;
1154
1155         batadv_dbg_arp(bat_priv, skb, type, hdr_size,
1156                        "Parsing outgoing ARP REPLY");
1157
1158         hw_src = batadv_arp_hw_src(skb, hdr_size);
1159         ip_src = batadv_arp_ip_src(skb, hdr_size);
1160         hw_dst = batadv_arp_hw_dst(skb, hdr_size);
1161         ip_dst = batadv_arp_ip_dst(skb, hdr_size);
1162
1163         batadv_dat_entry_add(bat_priv, ip_src, hw_src, vid);
1164         batadv_dat_entry_add(bat_priv, ip_dst, hw_dst, vid);
1165
1166         /* Send the ARP reply to the candidates for both the IP addresses that
1167          * the node obtained from the ARP reply
1168          */
1169         batadv_dat_send_data(bat_priv, skb, ip_src, vid, BATADV_P_DAT_DHT_PUT);
1170         batadv_dat_send_data(bat_priv, skb, ip_dst, vid, BATADV_P_DAT_DHT_PUT);
1171 }
1172
1173 /**
1174  * batadv_dat_snoop_incoming_arp_reply - snoop the ARP reply and fill the local
1175  * DAT storage only
1176  * @bat_priv: the bat priv with all the soft interface information
1177  * @skb: packet to check
1178  * @hdr_size: size of the encapsulation header
1179  *
1180  * Return: true if the packet was snooped and consumed by DAT. False if the
1181  * packet has to be delivered to the interface
1182  */
1183 bool batadv_dat_snoop_incoming_arp_reply(struct batadv_priv *bat_priv,
1184                                          struct sk_buff *skb, int hdr_size)
1185 {
1186         u16 type;
1187         __be32 ip_src, ip_dst;
1188         u8 *hw_src, *hw_dst;
1189         bool dropped = false;
1190         unsigned short vid;
1191
1192         if (!atomic_read(&bat_priv->distributed_arp_table))
1193                 goto out;
1194
1195         vid = batadv_dat_get_vid(skb, &hdr_size);
1196
1197         type = batadv_arp_get_type(bat_priv, skb, hdr_size);
1198         if (type != ARPOP_REPLY)
1199                 goto out;
1200
1201         batadv_dbg_arp(bat_priv, skb, type, hdr_size,
1202                        "Parsing incoming ARP REPLY");
1203
1204         hw_src = batadv_arp_hw_src(skb, hdr_size);
1205         ip_src = batadv_arp_ip_src(skb, hdr_size);
1206         hw_dst = batadv_arp_hw_dst(skb, hdr_size);
1207         ip_dst = batadv_arp_ip_dst(skb, hdr_size);
1208
1209         /* Update our internal cache with both the IP addresses the node got
1210          * within the ARP reply
1211          */
1212         batadv_dat_entry_add(bat_priv, ip_src, hw_src, vid);
1213         batadv_dat_entry_add(bat_priv, ip_dst, hw_dst, vid);
1214
1215         /* if this REPLY is directed to a client of mine, let's deliver the
1216          * packet to the interface
1217          */
1218         dropped = !batadv_is_my_client(bat_priv, hw_dst, vid);
1219
1220         /* if this REPLY is sent on behalf of a client of mine, let's drop the
1221          * packet because the client will reply by itself
1222          */
1223         dropped |= batadv_is_my_client(bat_priv, hw_src, vid);
1224 out:
1225         if (dropped)
1226                 kfree_skb(skb);
1227         /* if dropped == false -> deliver to the interface */
1228         return dropped;
1229 }
1230
1231 /**
1232  * batadv_dat_drop_broadcast_packet - check if an ARP request has to be dropped
1233  * (because the node has already obtained the reply via DAT) or not
1234  * @bat_priv: the bat priv with all the soft interface information
1235  * @forw_packet: the broadcast packet
1236  *
1237  * Return: true if the node can drop the packet, false otherwise.
1238  */
1239 bool batadv_dat_drop_broadcast_packet(struct batadv_priv *bat_priv,
1240                                       struct batadv_forw_packet *forw_packet)
1241 {
1242         u16 type;
1243         __be32 ip_dst;
1244         struct batadv_dat_entry *dat_entry = NULL;
1245         bool ret = false;
1246         int hdr_size = sizeof(struct batadv_bcast_packet);
1247         unsigned short vid;
1248
1249         if (!atomic_read(&bat_priv->distributed_arp_table))
1250                 goto out;
1251
1252         /* If this packet is an ARP_REQUEST and the node already has the
1253          * information that it is going to ask, then the packet can be dropped
1254          */
1255         if (forw_packet->num_packets)
1256                 goto out;
1257
1258         vid = batadv_dat_get_vid(forw_packet->skb, &hdr_size);
1259
1260         type = batadv_arp_get_type(bat_priv, forw_packet->skb, hdr_size);
1261         if (type != ARPOP_REQUEST)
1262                 goto out;
1263
1264         ip_dst = batadv_arp_ip_dst(forw_packet->skb, hdr_size);
1265         dat_entry = batadv_dat_entry_hash_find(bat_priv, ip_dst, vid);
1266         /* check if the node already got this entry */
1267         if (!dat_entry) {
1268                 batadv_dbg(BATADV_DBG_DAT, bat_priv,
1269                            "ARP Request for %pI4: fallback\n", &ip_dst);
1270                 goto out;
1271         }
1272
1273         batadv_dbg(BATADV_DBG_DAT, bat_priv,
1274                    "ARP Request for %pI4: fallback prevented\n", &ip_dst);
1275         ret = true;
1276
1277 out:
1278         if (dat_entry)
1279                 batadv_dat_entry_put(dat_entry);
1280         return ret;
1281 }