/* Copyright (C) 2007-2014 B.A.T.M.A.N. contributors:

   *
   * Marek Lindner, Simon Wunderlich
   *
   * This program is free software; you can redistribute it and/or
   * modify it under the terms of version 2 of the GNU General Public
   * License as published by the Free Software Foundation.
   *
   * This program is distributed in the hope that it will be useful, but
   * WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
   * General Public License for more details.
   *
   * You should have received a copy of the GNU General Public License

   * along with this program; if not, see <http://www.gnu.org/licenses/>.

   */
  
  #include "main.h"
  #include "routing.h"
  #include "send.h"

  #include "soft-interface.h"
  #include "hard-interface.h"
  #include "icmp_socket.h"
  #include "translation-table.h"
  #include "originator.h"

  #include "bridge_loop_avoidance.h"

  #include "distributed-arp-table.h"

  #include "network-coding.h"

  #include "fragmentation.h"

  #include <linux/if_vlan.h>

  static int batadv_route_unicast_packet(struct sk_buff *skb,

  				       struct batadv_hard_iface *recv_if);

  /**
   * _batadv_update_route - set the router for this originator
   * @bat_priv: the bat priv with all the soft interface information
   * @orig_node: orig node which is to be configured
   * @recv_if: the receive interface for which this route is set
   * @neigh_node: neighbor which should be the next router
   *
   * This function does not perform any error checks
   */

  static void _batadv_update_route(struct batadv_priv *bat_priv,
  				 struct batadv_orig_node *orig_node,

  				 struct batadv_hard_iface *recv_if,

  				 struct batadv_neigh_node *neigh_node)

  {

  	struct batadv_orig_ifinfo *orig_ifinfo;

  	struct batadv_neigh_node *curr_router;

  	orig_ifinfo = batadv_orig_ifinfo_get(orig_node, recv_if);
  	if (!orig_ifinfo)
  		return;
  
  	rcu_read_lock();
  	curr_router = rcu_dereference(orig_ifinfo->router);
  	if (curr_router && !atomic_inc_not_zero(&curr_router->refcount))
  		curr_router = NULL;
  	rcu_read_unlock();

  	/* route deleted */

  	if ((curr_router) && (!neigh_node)) {

  		batadv_dbg(BATADV_DBG_ROUTES, bat_priv,
  			   "Deleting route towards: %pM
  ", orig_node->orig);

  		batadv_tt_global_del_orig(bat_priv, orig_node, -1,

  					  "Deleted route towards originator");

  	/* route added */
  	} else if ((!curr_router) && (neigh_node)) {

  		batadv_dbg(BATADV_DBG_ROUTES, bat_priv,

  			   "Adding route towards: %pM (via %pM)
  ",
  			   orig_node->orig, neigh_node->addr);

  	/* route changed */

  	} else if (neigh_node && curr_router) {

  		batadv_dbg(BATADV_DBG_ROUTES, bat_priv,

  			   "Changing route towards: %pM (now via %pM - was via %pM)
  ",
  			   orig_node->orig, neigh_node->addr,
  			   curr_router->addr);

  	}

  	if (curr_router)

  		batadv_neigh_node_free_ref(curr_router);

  
  	/* increase refcount of new best neighbor */

  	if (neigh_node && !atomic_inc_not_zero(&neigh_node->refcount))
  		neigh_node = NULL;

  
  	spin_lock_bh(&orig_node->neigh_list_lock);

  	rcu_assign_pointer(orig_ifinfo->router, neigh_node);

  	spin_unlock_bh(&orig_node->neigh_list_lock);

  	batadv_orig_ifinfo_free_ref(orig_ifinfo);

  
  	/* decrease refcount of previous best neighbor */
  	if (curr_router)

  		batadv_neigh_node_free_ref(curr_router);

  }

  /**
   * batadv_update_route - set the router for this originator
   * @bat_priv: the bat priv with all the soft interface information
   * @orig_node: orig node which is to be configured
   * @recv_if: the receive interface for which this route is set
   * @neigh_node: neighbor which should be the next router
   */

  void batadv_update_route(struct batadv_priv *bat_priv,
  			 struct batadv_orig_node *orig_node,

  			 struct batadv_hard_iface *recv_if,

  			 struct batadv_neigh_node *neigh_node)

  {

  	struct batadv_neigh_node *router = NULL;

  
  	if (!orig_node)

  		goto out;

  	router = batadv_orig_router_get(orig_node, recv_if);

  	if (router != neigh_node)

  		_batadv_update_route(bat_priv, orig_node, recv_if, neigh_node);

  
  out:
  	if (router)

  		batadv_neigh_node_free_ref(router);

  }

  /* checks whether the host restarted and is in the protection time.
   * returns:
   *  0 if the packet is to be accepted
   *  1 if the packet is to be ignored.
   */

  int batadv_window_protected(struct batadv_priv *bat_priv, int32_t seq_num_diff,

  			    unsigned long *last_reset)

  {

  	if (seq_num_diff <= -BATADV_TQ_LOCAL_WINDOW_SIZE ||
  	    seq_num_diff >= BATADV_EXPECTED_SEQNO_RANGE) {
  		if (!batadv_has_timed_out(*last_reset,
  					  BATADV_RESET_PROTECTION_MS))

  			return 1;

  
  		*last_reset = jiffies;

  		batadv_dbg(BATADV_DBG_BATMAN, bat_priv,

  			   "old packet received, start protection
  ");

  	}

  	return 0;
  }

  bool batadv_check_management_packet(struct sk_buff *skb,

  				    struct batadv_hard_iface *hard_iface,

  				    int header_len)

  {

  	struct ethhdr *ethhdr;
  
  	/* drop packet if it has not necessary minimum size */

  	if (unlikely(!pskb_may_pull(skb, header_len)))
  		return false;

  	ethhdr = eth_hdr(skb);

  
  	/* packet with broadcast indication but unicast recipient */
  	if (!is_broadcast_ether_addr(ethhdr->h_dest))

  		return false;

  
  	/* packet with broadcast sender address */
  	if (is_broadcast_ether_addr(ethhdr->h_source))

  		return false;

  
  	/* create a copy of the skb, if needed, to modify it. */
  	if (skb_cow(skb, 0) < 0)

  		return false;

  
  	/* keep skb linear */
  	if (skb_linearize(skb) < 0)

  		return false;

  	return true;

  }

  /**
   * batadv_recv_my_icmp_packet - receive an icmp packet locally
   * @bat_priv: the bat priv with all the soft interface information
   * @skb: icmp packet to process
   *
   * Returns NET_RX_SUCCESS if the packet has been consumed or NET_RX_DROP
   * otherwise.
   */

  static int batadv_recv_my_icmp_packet(struct batadv_priv *bat_priv,

  				      struct sk_buff *skb)

  {

  	struct batadv_hard_iface *primary_if = NULL;
  	struct batadv_orig_node *orig_node = NULL;

  	struct batadv_icmp_header *icmph;
  	int res, ret = NET_RX_DROP;

  	icmph = (struct batadv_icmp_header *)skb->data;

  	switch (icmph->msg_type) {
  	case BATADV_ECHO_REPLY:
  	case BATADV_DESTINATION_UNREACHABLE:
  	case BATADV_TTL_EXCEEDED:
  		/* receive the packet */
  		if (skb_linearize(skb) < 0)
  			break;

  		batadv_socket_receive_packet(icmph, skb->len);
  		break;
  	case BATADV_ECHO_REQUEST:
  		/* answer echo request (ping) */
  		primary_if = batadv_primary_if_get_selected(bat_priv);
  		if (!primary_if)
  			goto out;
  
  		/* get routing information */
  		orig_node = batadv_orig_hash_find(bat_priv, icmph->orig);
  		if (!orig_node)
  			goto out;
  
  		/* create a copy of the skb, if needed, to modify it. */
  		if (skb_cow(skb, ETH_HLEN) < 0)
  			goto out;
  
  		icmph = (struct batadv_icmp_header *)skb->data;

  		ether_addr_copy(icmph->dst, icmph->orig);
  		ether_addr_copy(icmph->orig, primary_if->net_dev->dev_addr);

  		icmph->msg_type = BATADV_ECHO_REPLY;

  		icmph->ttl = BATADV_TTL;

  
  		res = batadv_send_skb_to_orig(skb, orig_node, NULL);
  		if (res != NET_XMIT_DROP)
  			ret = NET_RX_SUCCESS;
  
  		break;
  	default:
  		/* drop unknown type */

  		goto out;

  	}

  out:

  	if (primary_if)

  		batadv_hardif_free_ref(primary_if);

  	if (orig_node)

  		batadv_orig_node_free_ref(orig_node);

  	return ret;
  }

  static int batadv_recv_icmp_ttl_exceeded(struct batadv_priv *bat_priv,

  					 struct sk_buff *skb)

  {

  	struct batadv_hard_iface *primary_if = NULL;
  	struct batadv_orig_node *orig_node = NULL;

  	struct batadv_icmp_packet *icmp_packet;

  	int ret = NET_RX_DROP;

  	icmp_packet = (struct batadv_icmp_packet *)skb->data;

  
  	/* send TTL exceeded if packet is an echo request (traceroute) */

  	if (icmp_packet->msg_type != BATADV_ECHO_REQUEST) {

  		pr_debug("Warning - can't forward icmp packet from %pM to %pM: ttl exceeded
  ",

  			 icmp_packet->orig, icmp_packet->dst);

  		goto out;

  	}

  	primary_if = batadv_primary_if_get_selected(bat_priv);

  	if (!primary_if)

  		goto out;

  
  	/* get routing information */

  	orig_node = batadv_orig_hash_find(bat_priv, icmp_packet->orig);

  	if (!orig_node)

  		goto out;

  	/* create a copy of the skb, if needed, to modify it. */

  	if (skb_cow(skb, ETH_HLEN) < 0)

  		goto out;

  	icmp_packet = (struct batadv_icmp_packet *)skb->data;

  	ether_addr_copy(icmp_packet->dst, icmp_packet->orig);
  	ether_addr_copy(icmp_packet->orig, primary_if->net_dev->dev_addr);

  	icmp_packet->msg_type = BATADV_TTL_EXCEEDED;
  	icmp_packet->ttl = BATADV_TTL;

  	if (batadv_send_skb_to_orig(skb, orig_node, NULL) != NET_XMIT_DROP)

  		ret = NET_RX_SUCCESS;

  out:

  	if (primary_if)

  		batadv_hardif_free_ref(primary_if);

  	if (orig_node)

  		batadv_orig_node_free_ref(orig_node);

  	return ret;
  }

  int batadv_recv_icmp_packet(struct sk_buff *skb,
  			    struct batadv_hard_iface *recv_if)

  {

  	struct batadv_priv *bat_priv = netdev_priv(recv_if->soft_iface);

  	struct batadv_icmp_header *icmph;
  	struct batadv_icmp_packet_rr *icmp_packet_rr;

  	struct ethhdr *ethhdr;

  	struct batadv_orig_node *orig_node = NULL;

  	int hdr_size = sizeof(struct batadv_icmp_header);

  	int ret = NET_RX_DROP;

  	/* drop packet if it has not necessary minimum size */
  	if (unlikely(!pskb_may_pull(skb, hdr_size)))

  		goto out;

  	ethhdr = eth_hdr(skb);

  
  	/* packet with unicast indication but broadcast recipient */
  	if (is_broadcast_ether_addr(ethhdr->h_dest))

  		goto out;

  
  	/* packet with broadcast sender address */
  	if (is_broadcast_ether_addr(ethhdr->h_source))

  		goto out;

  
  	/* not for me */

  	if (!batadv_is_my_mac(bat_priv, ethhdr->h_dest))

  		goto out;

  	icmph = (struct batadv_icmp_header *)skb->data;

  
  	/* add record route information if not full */

  	if ((icmph->msg_type == BATADV_ECHO_REPLY ||
  	     icmph->msg_type == BATADV_ECHO_REQUEST) &&
  	    (skb->len >= sizeof(struct batadv_icmp_packet_rr))) {
  		if (skb_linearize(skb) < 0)
  			goto out;
  
  		/* create a copy of the skb, if needed, to modify it. */
  		if (skb_cow(skb, ETH_HLEN) < 0)
  			goto out;
  
  		icmph = (struct batadv_icmp_header *)skb->data;
  		icmp_packet_rr = (struct batadv_icmp_packet_rr *)icmph;
  		if (icmp_packet_rr->rr_cur >= BATADV_RR_LEN)
  			goto out;

  		ether_addr_copy(icmp_packet_rr->rr[icmp_packet_rr->rr_cur],
  				ethhdr->h_dest);

  		icmp_packet_rr->rr_cur++;

  	}
  
  	/* packet for me */

  	if (batadv_is_my_mac(bat_priv, icmph->dst))
  		return batadv_recv_my_icmp_packet(bat_priv, skb);

  
  	/* TTL exceeded */

  	if (icmph->ttl < 2)

  		return batadv_recv_icmp_ttl_exceeded(bat_priv, skb);

  	/* get routing information */

  	orig_node = batadv_orig_hash_find(bat_priv, icmph->dst);

  	if (!orig_node)

  		goto out;

  	/* create a copy of the skb, if needed, to modify it. */

  	if (skb_cow(skb, ETH_HLEN) < 0)

  		goto out;

  	icmph = (struct batadv_icmp_header *)skb->data;

  	/* decrement ttl */

  	icmph->ttl--;

  
  	/* route it */

  	if (batadv_send_skb_to_orig(skb, orig_node, recv_if) != NET_XMIT_DROP)

  		ret = NET_RX_SUCCESS;

  out:

  	if (orig_node)

  		batadv_orig_node_free_ref(orig_node);

  	return ret;
  }

  /**
   * batadv_check_unicast_packet - Check for malformed unicast packets

   * @bat_priv: the bat priv with all the soft interface information

   * @skb: packet to check
   * @hdr_size: size of header to pull
   *
   * Check for short header and bad addresses in given packet. Returns negative
   * value when check fails and 0 otherwise. The negative value depends on the
   * reason: -ENODATA for bad header, -EBADR for broadcast destination or source,
   * and -EREMOTE for non-local (other host) destination.
   */

  static int batadv_check_unicast_packet(struct batadv_priv *bat_priv,
  				       struct sk_buff *skb, int hdr_size)

  {
  	struct ethhdr *ethhdr;
  
  	/* drop packet if it has not necessary minimum size */
  	if (unlikely(!pskb_may_pull(skb, hdr_size)))

  		return -ENODATA;

  	ethhdr = eth_hdr(skb);

  
  	/* packet with unicast indication but broadcast recipient */
  	if (is_broadcast_ether_addr(ethhdr->h_dest))

  		return -EBADR;

  
  	/* packet with broadcast sender address */
  	if (is_broadcast_ether_addr(ethhdr->h_source))

  		return -EBADR;

  
  	/* not for me */

  	if (!batadv_is_my_mac(bat_priv, ethhdr->h_dest))

  		return -EREMOTE;

  
  	return 0;
  }

  /**
   * batadv_find_router - find a suitable router for this originator
   * @bat_priv: the bat priv with all the soft interface information
   * @orig_node: the destination node
   * @recv_if: pointer to interface this packet was received on
   *
   * Returns the router which should be used for this orig_node on
   * this interface, or NULL if not available.

   */

  struct batadv_neigh_node *
  batadv_find_router(struct batadv_priv *bat_priv,
  		   struct batadv_orig_node *orig_node,

  		   struct batadv_hard_iface *recv_if)

  {

  	struct batadv_algo_ops *bao = bat_priv->bat_algo_ops;
  	struct batadv_neigh_node *first_candidate_router = NULL;
  	struct batadv_neigh_node *next_candidate_router = NULL;
  	struct batadv_neigh_node *router, *cand_router = NULL;
  	struct batadv_neigh_node *last_cand_router = NULL;
  	struct batadv_orig_ifinfo *cand, *first_candidate = NULL;
  	struct batadv_orig_ifinfo *next_candidate = NULL;
  	struct batadv_orig_ifinfo *last_candidate;
  	bool last_candidate_found = false;

  
  	if (!orig_node)
  		return NULL;

  	router = batadv_orig_router_get(orig_node, recv_if);

  	/* only consider bonding for recv_if == BATADV_IF_DEFAULT (first hop)
  	 * and if activated.
  	 */
  	if (recv_if == BATADV_IF_DEFAULT || !atomic_read(&bat_priv->bonding) ||
  	    !router)
  		return router;
  
  	/* bonding: loop through the list of possible routers found
  	 * for the various outgoing interfaces and find a candidate after
  	 * the last chosen bonding candidate (next_candidate). If no such
  	 * router is found, use the first candidate found (the previously
  	 * chosen bonding candidate might have been the last one in the list).
  	 * If this can't be found either, return the previously choosen
  	 * router - obviously there are no other candidates.
  	 */
  	rcu_read_lock();
  	last_candidate = orig_node->last_bonding_candidate;
  	if (last_candidate)
  		last_cand_router = rcu_dereference(last_candidate->router);
  
  	hlist_for_each_entry_rcu(cand, &orig_node->ifinfo_list, list) {
  		/* acquire some structures and references ... */
  		if (!atomic_inc_not_zero(&cand->refcount))
  			continue;
  
  		cand_router = rcu_dereference(cand->router);
  		if (!cand_router)
  			goto next;
  
  		if (!atomic_inc_not_zero(&cand_router->refcount)) {
  			cand_router = NULL;
  			goto next;
  		}
  
  		/* alternative candidate should be good enough to be
  		 * considered
  		 */
  		if (!bao->bat_neigh_is_equiv_or_better(cand_router,
  						       cand->if_outgoing,
  						       router, recv_if))
  			goto next;
  
  		/* don't use the same router twice */
  		if (last_cand_router == cand_router)
  			goto next;
  
  		/* mark the first possible candidate */
  		if (!first_candidate) {
  			atomic_inc(&cand_router->refcount);
  			atomic_inc(&cand->refcount);
  			first_candidate = cand;
  			first_candidate_router = cand_router;
  		}
  
  		/* check if the loop has already passed the previously selected
  		 * candidate ... this function should select the next candidate
  		 * AFTER the previously used bonding candidate.
  		 */
  		if (!last_candidate || last_candidate_found) {
  			next_candidate = cand;
  			next_candidate_router = cand_router;
  			break;
  		}
  
  		if (last_candidate == cand)
  			last_candidate_found = true;
  next:
  		/* free references */
  		if (cand_router) {
  			batadv_neigh_node_free_ref(cand_router);
  			cand_router = NULL;
  		}
  		batadv_orig_ifinfo_free_ref(cand);
  	}
  	rcu_read_unlock();
  
  	/* last_bonding_candidate is reset below, remove the old reference. */
  	if (orig_node->last_bonding_candidate)
  		batadv_orig_ifinfo_free_ref(orig_node->last_bonding_candidate);
  
  	/* After finding candidates, handle the three cases:
  	 * 1) there is a next candidate, use that
  	 * 2) there is no next candidate, use the first of the list
  	 * 3) there is no candidate at all, return the default router
  	 */
  	if (next_candidate) {
  		batadv_neigh_node_free_ref(router);
  
  		/* remove references to first candidate, we don't need it. */
  		if (first_candidate) {
  			batadv_neigh_node_free_ref(first_candidate_router);
  			batadv_orig_ifinfo_free_ref(first_candidate);
  		}
  		router = next_candidate_router;
  		orig_node->last_bonding_candidate = next_candidate;
  	} else if (first_candidate) {
  		batadv_neigh_node_free_ref(router);
  
  		/* refcounting has already been done in the loop above. */
  		router = first_candidate_router;
  		orig_node->last_bonding_candidate = first_candidate;
  	} else {
  		orig_node->last_bonding_candidate = NULL;
  	}

  	return router;
  }

  static int batadv_route_unicast_packet(struct sk_buff *skb,

  				       struct batadv_hard_iface *recv_if)

  {

  	struct batadv_priv *bat_priv = netdev_priv(recv_if->soft_iface);
  	struct batadv_orig_node *orig_node = NULL;

  	struct batadv_unicast_packet *unicast_packet;

  	struct ethhdr *ethhdr = eth_hdr(skb);

  	int res, hdr_len, ret = NET_RX_DROP;

  	unicast_packet = (struct batadv_unicast_packet *)skb->data;

  
  	/* TTL exceeded */

  	if (unicast_packet->ttl < 2) {

  		pr_debug("Warning - can't forward unicast packet from %pM to %pM: ttl exceeded
  ",
  			 ethhdr->h_source, unicast_packet->dest);

  		goto out;

  	}
  
  	/* get routing information */

  	orig_node = batadv_orig_hash_find(bat_priv, unicast_packet->dest);

  	if (!orig_node)

  		goto out;

  	/* create a copy of the skb, if needed, to modify it. */

  	if (skb_cow(skb, ETH_HLEN) < 0)

  		goto out;

  	/* decrement ttl */

  	unicast_packet = (struct batadv_unicast_packet *)skb->data;

  	unicast_packet->ttl--;

  	switch (unicast_packet->packet_type) {

  	case BATADV_UNICAST_4ADDR:
  		hdr_len = sizeof(struct batadv_unicast_4addr_packet);
  		break;
  	case BATADV_UNICAST:
  		hdr_len = sizeof(struct batadv_unicast_packet);
  		break;
  	default:
  		/* other packet types not supported - yet */
  		hdr_len = -1;
  		break;
  	}
  
  	if (hdr_len > 0)
  		batadv_skb_set_priority(skb, hdr_len);

  	res = batadv_send_skb_to_orig(skb, orig_node, recv_if);

  	/* translate transmit result into receive result */
  	if (res == NET_XMIT_SUCCESS) {
  		/* skb was transmitted and consumed */

  		batadv_inc_counter(bat_priv, BATADV_CNT_FORWARD);
  		batadv_add_counter(bat_priv, BATADV_CNT_FORWARD_BYTES,
  				   skb->len + ETH_HLEN);

  
  		ret = NET_RX_SUCCESS;
  	} else if (res == NET_XMIT_POLICED) {
  		/* skb was buffered and consumed */
  		ret = NET_RX_SUCCESS;

  	}

  out:

  	if (orig_node)

  		batadv_orig_node_free_ref(orig_node);

  	return ret;

  }

  /**
   * batadv_reroute_unicast_packet - update the unicast header for re-routing
   * @bat_priv: the bat priv with all the soft interface information
   * @unicast_packet: the unicast header to be updated
   * @dst_addr: the payload destination

   * @vid: VLAN identifier

   *
   * Search the translation table for dst_addr and update the unicast header with
   * the new corresponding information (originator address where the destination
   * client currently is and its known TTVN)
   *
   * Returns true if the packet header has been updated, false otherwise
   */
  static bool
  batadv_reroute_unicast_packet(struct batadv_priv *bat_priv,
  			      struct batadv_unicast_packet *unicast_packet,

  			      uint8_t *dst_addr, unsigned short vid)

  {
  	struct batadv_orig_node *orig_node = NULL;
  	struct batadv_hard_iface *primary_if = NULL;
  	bool ret = false;
  	uint8_t *orig_addr, orig_ttvn;

  	if (batadv_is_my_client(bat_priv, dst_addr, vid)) {

  		primary_if = batadv_primary_if_get_selected(bat_priv);
  		if (!primary_if)
  			goto out;
  		orig_addr = primary_if->net_dev->dev_addr;
  		orig_ttvn = (uint8_t)atomic_read(&bat_priv->tt.vn);
  	} else {

  		orig_node = batadv_transtable_search(bat_priv, NULL, dst_addr,
  						     vid);

  		if (!orig_node)
  			goto out;
  
  		if (batadv_compare_eth(orig_node->orig, unicast_packet->dest))
  			goto out;
  
  		orig_addr = orig_node->orig;
  		orig_ttvn = (uint8_t)atomic_read(&orig_node->last_ttvn);
  	}
  
  	/* update the packet header */

  	ether_addr_copy(unicast_packet->dest, orig_addr);

  	unicast_packet->ttvn = orig_ttvn;
  
  	ret = true;
  out:
  	if (primary_if)
  		batadv_hardif_free_ref(primary_if);
  	if (orig_node)
  		batadv_orig_node_free_ref(orig_node);
  
  	return ret;
  }

  static int batadv_check_unicast_ttvn(struct batadv_priv *bat_priv,

  				     struct sk_buff *skb, int hdr_len) {

  	struct batadv_unicast_packet *unicast_packet;
  	struct batadv_hard_iface *primary_if;

  	struct batadv_orig_node *orig_node;

  	uint8_t curr_ttvn, old_ttvn;

  	struct ethhdr *ethhdr;

  	unsigned short vid;

  	int is_old_ttvn;

  	/* check if there is enough data before accessing it */

  	if (!pskb_may_pull(skb, hdr_len + ETH_HLEN))

  		return 0;
  
  	/* create a copy of the skb (in case of for re-routing) to modify it. */
  	if (skb_cow(skb, sizeof(*unicast_packet)) < 0)

  		return 0;

  	unicast_packet = (struct batadv_unicast_packet *)skb->data;

  	vid = batadv_get_vid(skb, hdr_len);

  	ethhdr = (struct ethhdr *)(skb->data + hdr_len);

  	/* check if the destination client was served by this node and it is now
  	 * roaming. In this case, it means that the node has got a ROAM_ADV
  	 * message and that it knows the new destination in the mesh to re-route
  	 * the packet to
  	 */

  	if (batadv_tt_local_client_is_roaming(bat_priv, ethhdr->h_dest, vid)) {

  		if (batadv_reroute_unicast_packet(bat_priv, unicast_packet,

  						  ethhdr->h_dest, vid))

  			net_ratelimited_function(batadv_dbg, BATADV_DBG_TT,
  						 bat_priv,
  						 "Rerouting unicast packet to %pM (dst=%pM): Local Roaming
  ",
  						 unicast_packet->dest,
  						 ethhdr->h_dest);
  		/* at this point the mesh destination should have been
  		 * substituted with the originator address found in the global
  		 * table. If not, let the packet go untouched anyway because
  		 * there is nothing the node can do
  		 */
  		return 1;
  	}
  
  	/* retrieve the TTVN known by this node for the packet destination. This
  	 * value is used later to check if the node which sent (or re-routed
  	 * last time) the packet had an updated information or not
  	 */
  	curr_ttvn = (uint8_t)atomic_read(&bat_priv->tt.vn);

  	if (!batadv_is_my_mac(bat_priv, unicast_packet->dest)) {

  		orig_node = batadv_orig_hash_find(bat_priv,
  						  unicast_packet->dest);

  		/* if it is not possible to find the orig_node representing the
  		 * destination, the packet can immediately be dropped as it will
  		 * not be possible to deliver it
  		 */

  		if (!orig_node)
  			return 0;
  
  		curr_ttvn = (uint8_t)atomic_read(&orig_node->last_ttvn);

  		batadv_orig_node_free_ref(orig_node);

  	}

  	/* check if the TTVN contained in the packet is fresher than what the
  	 * node knows
  	 */

  	is_old_ttvn = batadv_seq_before(unicast_packet->ttvn, curr_ttvn);

  	if (!is_old_ttvn)
  		return 1;

  	old_ttvn = unicast_packet->ttvn;
  	/* the packet was forged based on outdated network information. Its
  	 * destination can possibly be updated and forwarded towards the new
  	 * target host
  	 */
  	if (batadv_reroute_unicast_packet(bat_priv, unicast_packet,

  					  ethhdr->h_dest, vid)) {

  		net_ratelimited_function(batadv_dbg, BATADV_DBG_TT, bat_priv,
  					 "Rerouting unicast packet to %pM (dst=%pM): TTVN mismatch old_ttvn=%u new_ttvn=%u
  ",
  					 unicast_packet->dest, ethhdr->h_dest,
  					 old_ttvn, curr_ttvn);
  		return 1;
  	}

  	/* the packet has not been re-routed: either the destination is
  	 * currently served by this node or there is no destination at all and
  	 * it is possible to drop the packet
  	 */

  	if (!batadv_is_my_client(bat_priv, ethhdr->h_dest, vid))

  		return 0;

  	/* update the header in order to let the packet be delivered to this
  	 * node's soft interface
  	 */
  	primary_if = batadv_primary_if_get_selected(bat_priv);
  	if (!primary_if)
  		return 0;

  	ether_addr_copy(unicast_packet->dest, primary_if->net_dev->dev_addr);

  
  	batadv_hardif_free_ref(primary_if);
  
  	unicast_packet->ttvn = curr_ttvn;

  	return 1;
  }

  /**
   * batadv_recv_unhandled_unicast_packet - receive and process packets which
   *	are in the unicast number space but not yet known to the implementation
   * @skb: unicast tvlv packet to process
   * @recv_if: pointer to interface this packet was received on
   *
   * Returns NET_RX_SUCCESS if the packet has been consumed or NET_RX_DROP
   * otherwise.
   */
  int batadv_recv_unhandled_unicast_packet(struct sk_buff *skb,
  					 struct batadv_hard_iface *recv_if)
  {
  	struct batadv_unicast_packet *unicast_packet;
  	struct batadv_priv *bat_priv = netdev_priv(recv_if->soft_iface);
  	int check, hdr_size = sizeof(*unicast_packet);
  
  	check = batadv_check_unicast_packet(bat_priv, skb, hdr_size);
  	if (check < 0)
  		return NET_RX_DROP;
  
  	/* we don't know about this type, drop it. */
  	unicast_packet = (struct batadv_unicast_packet *)skb->data;
  	if (batadv_is_my_mac(bat_priv, unicast_packet->dest))
  		return NET_RX_DROP;
  
  	return batadv_route_unicast_packet(skb, recv_if);
  }

  int batadv_recv_unicast_packet(struct sk_buff *skb,
  			       struct batadv_hard_iface *recv_if)

  {

  	struct batadv_priv *bat_priv = netdev_priv(recv_if->soft_iface);

  	struct batadv_unicast_packet *unicast_packet;

  	struct batadv_unicast_4addr_packet *unicast_4addr_packet;

  	uint8_t *orig_addr;
  	struct batadv_orig_node *orig_node = NULL;

  	int check, hdr_size = sizeof(*unicast_packet);

  	bool is4addr;

  	unicast_packet = (struct batadv_unicast_packet *)skb->data;

  	unicast_4addr_packet = (struct batadv_unicast_4addr_packet *)skb->data;

  	is4addr = unicast_packet->packet_type == BATADV_UNICAST_4ADDR;

  	/* the caller function should have already pulled 2 bytes */

  	if (is4addr)

  		hdr_size = sizeof(*unicast_4addr_packet);

  	/* function returns -EREMOTE for promiscuous packets */

  	check = batadv_check_unicast_packet(bat_priv, skb, hdr_size);

  
  	/* Even though the packet is not for us, we might save it to use for
  	 * decoding a later received coded packet
  	 */
  	if (check == -EREMOTE)
  		batadv_nc_skb_store_sniffed_unicast(bat_priv, skb);
  
  	if (check < 0)

  		return NET_RX_DROP;

  	if (!batadv_check_unicast_ttvn(bat_priv, skb, hdr_size))

  		return NET_RX_DROP;

  	/* packet for me */

  	if (batadv_is_my_mac(bat_priv, unicast_packet->dest)) {

  		if (is4addr) {

  			batadv_dat_inc_counter(bat_priv,
  					       unicast_4addr_packet->subtype);

  			orig_addr = unicast_4addr_packet->src;
  			orig_node = batadv_orig_hash_find(bat_priv, orig_addr);
  		}

  		if (batadv_dat_snoop_incoming_arp_request(bat_priv, skb,
  							  hdr_size))
  			goto rx_success;
  		if (batadv_dat_snoop_incoming_arp_reply(bat_priv, skb,
  							hdr_size))
  			goto rx_success;

  		batadv_interface_rx(recv_if->soft_iface, skb, recv_if, hdr_size,

  				    orig_node);

  rx_success:

  		if (orig_node)
  			batadv_orig_node_free_ref(orig_node);

  		return NET_RX_SUCCESS;
  	}

  	return batadv_route_unicast_packet(skb, recv_if);

  }

  /**
   * batadv_recv_unicast_tvlv - receive and process unicast tvlv packets
   * @skb: unicast tvlv packet to process
   * @recv_if: pointer to interface this packet was received on
   * @dst_addr: the payload destination
   *
   * Returns NET_RX_SUCCESS if the packet has been consumed or NET_RX_DROP
   * otherwise.
   */
  int batadv_recv_unicast_tvlv(struct sk_buff *skb,
  			     struct batadv_hard_iface *recv_if)
  {
  	struct batadv_priv *bat_priv = netdev_priv(recv_if->soft_iface);
  	struct batadv_unicast_tvlv_packet *unicast_tvlv_packet;
  	unsigned char *tvlv_buff;
  	uint16_t tvlv_buff_len;
  	int hdr_size = sizeof(*unicast_tvlv_packet);
  	int ret = NET_RX_DROP;
  
  	if (batadv_check_unicast_packet(bat_priv, skb, hdr_size) < 0)
  		return NET_RX_DROP;
  
  	/* the header is likely to be modified while forwarding */
  	if (skb_cow(skb, hdr_size) < 0)
  		return NET_RX_DROP;
  
  	/* packet needs to be linearized to access the tvlv content */
  	if (skb_linearize(skb) < 0)
  		return NET_RX_DROP;
  
  	unicast_tvlv_packet = (struct batadv_unicast_tvlv_packet *)skb->data;
  
  	tvlv_buff = (unsigned char *)(skb->data + hdr_size);
  	tvlv_buff_len = ntohs(unicast_tvlv_packet->tvlv_len);
  
  	if (tvlv_buff_len > skb->len - hdr_size)
  		return NET_RX_DROP;
  
  	ret = batadv_tvlv_containers_process(bat_priv, false, NULL,
  					     unicast_tvlv_packet->src,
  					     unicast_tvlv_packet->dst,
  					     tvlv_buff, tvlv_buff_len);
  
  	if (ret != NET_RX_SUCCESS)
  		ret = batadv_route_unicast_packet(skb, recv_if);

  	else
  		consume_skb(skb);

  
  	return ret;
  }

  /**
   * batadv_recv_frag_packet - process received fragment
   * @skb: the received fragment
   * @recv_if: interface that the skb is received on
   *
   * This function does one of the three following things: 1) Forward fragment, if
   * the assembled packet will exceed our MTU; 2) Buffer fragment, if we till
   * lack further fragments; 3) Merge fragments, if we have all needed parts.
   *
   * Return NET_RX_DROP if the skb is not consumed, NET_RX_SUCCESS otherwise.
   */
  int batadv_recv_frag_packet(struct sk_buff *skb,
  			    struct batadv_hard_iface *recv_if)
  {
  	struct batadv_priv *bat_priv = netdev_priv(recv_if->soft_iface);
  	struct batadv_orig_node *orig_node_src = NULL;
  	struct batadv_frag_packet *frag_packet;
  	int ret = NET_RX_DROP;
  
  	if (batadv_check_unicast_packet(bat_priv, skb,
  					sizeof(*frag_packet)) < 0)
  		goto out;
  
  	frag_packet = (struct batadv_frag_packet *)skb->data;
  	orig_node_src = batadv_orig_hash_find(bat_priv, frag_packet->orig);
  	if (!orig_node_src)
  		goto out;
  
  	/* Route the fragment if it is not for us and too big to be merged. */
  	if (!batadv_is_my_mac(bat_priv, frag_packet->dest) &&
  	    batadv_frag_skb_fwd(skb, recv_if, orig_node_src)) {
  		ret = NET_RX_SUCCESS;
  		goto out;
  	}
  
  	batadv_inc_counter(bat_priv, BATADV_CNT_FRAG_RX);
  	batadv_add_counter(bat_priv, BATADV_CNT_FRAG_RX_BYTES, skb->len);
  
  	/* Add fragment to buffer and merge if possible. */
  	if (!batadv_frag_skb_buffer(&skb, orig_node_src))
  		goto out;
  
  	/* Deliver merged packet to the appropriate handler, if it was
  	 * merged
  	 */
  	if (skb)
  		batadv_batman_skb_recv(skb, recv_if->net_dev,
  				       &recv_if->batman_adv_ptype, NULL);
  
  	ret = NET_RX_SUCCESS;
  
  out:
  	if (orig_node_src)
  		batadv_orig_node_free_ref(orig_node_src);
  
  	return ret;
  }

  int batadv_recv_bcast_packet(struct sk_buff *skb,
  			     struct batadv_hard_iface *recv_if)

  {

  	struct batadv_priv *bat_priv = netdev_priv(recv_if->soft_iface);
  	struct batadv_orig_node *orig_node = NULL;

  	struct batadv_bcast_packet *bcast_packet;

  	struct ethhdr *ethhdr;

  	int hdr_size = sizeof(*bcast_packet);

  	int ret = NET_RX_DROP;

  	int32_t seq_diff;

  	uint32_t seqno;

  
  	/* drop packet if it has not necessary minimum size */
  	if (unlikely(!pskb_may_pull(skb, hdr_size)))

  		goto out;

  	ethhdr = eth_hdr(skb);

  
  	/* packet with broadcast indication but unicast recipient */
  	if (!is_broadcast_ether_addr(ethhdr->h_dest))

  		goto out;

  
  	/* packet with broadcast sender address */
  	if (is_broadcast_ether_addr(ethhdr->h_source))

  		goto out;

  
  	/* ignore broadcasts sent by myself */

  	if (batadv_is_my_mac(bat_priv, ethhdr->h_source))

  		goto out;

  	bcast_packet = (struct batadv_bcast_packet *)skb->data;

  
  	/* ignore broadcasts originated by myself */

  	if (batadv_is_my_mac(bat_priv, bcast_packet->orig))

  		goto out;

  	if (bcast_packet->ttl < 2)

  		goto out;

  	orig_node = batadv_orig_hash_find(bat_priv, bcast_packet->orig);

  
  	if (!orig_node)

  		goto out;

  	spin_lock_bh(&orig_node->bcast_seqno_lock);

  	seqno = ntohl(bcast_packet->seqno);

  	/* check whether the packet is a duplicate */

  	if (batadv_test_bit(orig_node->bcast_bits, orig_node->last_bcast_seqno,

  			    seqno))

  		goto spin_unlock;

  	seq_diff = seqno - orig_node->last_bcast_seqno;

  
  	/* check whether the packet is old and the host just restarted. */

  	if (batadv_window_protected(bat_priv, seq_diff,
  				    &orig_node->bcast_seqno_reset))

  		goto spin_unlock;

  
  	/* mark broadcast in flood history, update window position

  	 * if required.
  	 */

  	if (batadv_bit_get_packet(bat_priv, orig_node->bcast_bits, seq_diff, 1))

  		orig_node->last_bcast_seqno = seqno;

  	spin_unlock_bh(&orig_node->bcast_seqno_lock);

  	/* check whether this has been sent by another originator before */

  	if (batadv_bla_check_bcast_duplist(bat_priv, skb))

  		goto out;

  	batadv_skb_set_priority(skb, sizeof(struct batadv_bcast_packet));

  	/* rebroadcast packet */

  	batadv_add_bcast_packet_to_list(bat_priv, skb, 1);

  	/* don't hand the broadcast up if it is from an originator
  	 * from the same backbone.
  	 */

  	if (batadv_bla_is_backbone_gw(skb, orig_node, hdr_size))

  		goto out;

  	if (batadv_dat_snoop_incoming_arp_request(bat_priv, skb, hdr_size))
  		goto rx_success;
  	if (batadv_dat_snoop_incoming_arp_reply(bat_priv, skb, hdr_size))
  		goto rx_success;

  	/* broadcast for me */

  	batadv_interface_rx(recv_if->soft_iface, skb, recv_if, hdr_size,
  			    orig_node);

  
  rx_success:

  	ret = NET_RX_SUCCESS;
  	goto out;

  spin_unlock:
  	spin_unlock_bh(&orig_node->bcast_seqno_lock);

  out:
  	if (orig_node)

  		batadv_orig_node_free_ref(orig_node);

  	return ret;

  }