Linux 4.8.3
[linux/fpc-iii.git] / net / batman-adv / network-coding.c
blob293ef4ffd4e1650318020f9e6772bcf58a9e1300
1 /* Copyright (C) 2012-2016 B.A.T.M.A.N. contributors:
3 * Martin Hundebøll, Jeppe Ledet-Pedersen
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.
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.
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/>.
18 #include "network-coding.h"
19 #include "main.h"
21 #include <linux/atomic.h>
22 #include <linux/bitops.h>
23 #include <linux/byteorder/generic.h>
24 #include <linux/compiler.h>
25 #include <linux/debugfs.h>
26 #include <linux/errno.h>
27 #include <linux/etherdevice.h>
28 #include <linux/fs.h>
29 #include <linux/if_ether.h>
30 #include <linux/if_packet.h>
31 #include <linux/init.h>
32 #include <linux/jhash.h>
33 #include <linux/jiffies.h>
34 #include <linux/kernel.h>
35 #include <linux/kref.h>
36 #include <linux/list.h>
37 #include <linux/lockdep.h>
38 #include <linux/netdevice.h>
39 #include <linux/printk.h>
40 #include <linux/random.h>
41 #include <linux/rculist.h>
42 #include <linux/rcupdate.h>
43 #include <linux/seq_file.h>
44 #include <linux/skbuff.h>
45 #include <linux/slab.h>
46 #include <linux/spinlock.h>
47 #include <linux/stat.h>
48 #include <linux/stddef.h>
49 #include <linux/string.h>
50 #include <linux/workqueue.h>
52 #include "hard-interface.h"
53 #include "hash.h"
54 #include "log.h"
55 #include "originator.h"
56 #include "packet.h"
57 #include "routing.h"
58 #include "send.h"
59 #include "tvlv.h"
61 static struct lock_class_key batadv_nc_coding_hash_lock_class_key;
62 static struct lock_class_key batadv_nc_decoding_hash_lock_class_key;
64 static void batadv_nc_worker(struct work_struct *work);
65 static int batadv_nc_recv_coded_packet(struct sk_buff *skb,
66 struct batadv_hard_iface *recv_if);
68 /**
69 * batadv_nc_init - one-time initialization for network coding
71 * Return: 0 on success or negative error number in case of failure
73 int __init batadv_nc_init(void)
75 int ret;
77 /* Register our packet type */
78 ret = batadv_recv_handler_register(BATADV_CODED,
79 batadv_nc_recv_coded_packet);
81 return ret;
84 /**
85 * batadv_nc_start_timer - initialise the nc periodic worker
86 * @bat_priv: the bat priv with all the soft interface information
88 static void batadv_nc_start_timer(struct batadv_priv *bat_priv)
90 queue_delayed_work(batadv_event_workqueue, &bat_priv->nc.work,
91 msecs_to_jiffies(10));
94 /**
95 * batadv_nc_tvlv_container_update - update the network coding tvlv container
96 * after network coding setting change
97 * @bat_priv: the bat priv with all the soft interface information
99 static void batadv_nc_tvlv_container_update(struct batadv_priv *bat_priv)
101 char nc_mode;
103 nc_mode = atomic_read(&bat_priv->network_coding);
105 switch (nc_mode) {
106 case 0:
107 batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_NC, 1);
108 break;
109 case 1:
110 batadv_tvlv_container_register(bat_priv, BATADV_TVLV_NC, 1,
111 NULL, 0);
112 break;
117 * batadv_nc_status_update - update the network coding tvlv container after
118 * network coding setting change
119 * @net_dev: the soft interface net device
121 void batadv_nc_status_update(struct net_device *net_dev)
123 struct batadv_priv *bat_priv = netdev_priv(net_dev);
125 batadv_nc_tvlv_container_update(bat_priv);
129 * batadv_nc_tvlv_ogm_handler_v1 - process incoming nc tvlv container
130 * @bat_priv: the bat priv with all the soft interface information
131 * @orig: the orig_node of the ogm
132 * @flags: flags indicating the tvlv state (see batadv_tvlv_handler_flags)
133 * @tvlv_value: tvlv buffer containing the gateway data
134 * @tvlv_value_len: tvlv buffer length
136 static void batadv_nc_tvlv_ogm_handler_v1(struct batadv_priv *bat_priv,
137 struct batadv_orig_node *orig,
138 u8 flags,
139 void *tvlv_value, u16 tvlv_value_len)
141 if (flags & BATADV_TVLV_HANDLER_OGM_CIFNOTFND)
142 clear_bit(BATADV_ORIG_CAPA_HAS_NC, &orig->capabilities);
143 else
144 set_bit(BATADV_ORIG_CAPA_HAS_NC, &orig->capabilities);
148 * batadv_nc_mesh_init - initialise coding hash table and start house keeping
149 * @bat_priv: the bat priv with all the soft interface information
151 * Return: 0 on success or negative error number in case of failure
153 int batadv_nc_mesh_init(struct batadv_priv *bat_priv)
155 bat_priv->nc.timestamp_fwd_flush = jiffies;
156 bat_priv->nc.timestamp_sniffed_purge = jiffies;
158 if (bat_priv->nc.coding_hash || bat_priv->nc.decoding_hash)
159 return 0;
161 bat_priv->nc.coding_hash = batadv_hash_new(128);
162 if (!bat_priv->nc.coding_hash)
163 goto err;
165 batadv_hash_set_lock_class(bat_priv->nc.coding_hash,
166 &batadv_nc_coding_hash_lock_class_key);
168 bat_priv->nc.decoding_hash = batadv_hash_new(128);
169 if (!bat_priv->nc.decoding_hash)
170 goto err;
172 batadv_hash_set_lock_class(bat_priv->nc.decoding_hash,
173 &batadv_nc_decoding_hash_lock_class_key);
175 INIT_DELAYED_WORK(&bat_priv->nc.work, batadv_nc_worker);
176 batadv_nc_start_timer(bat_priv);
178 batadv_tvlv_handler_register(bat_priv, batadv_nc_tvlv_ogm_handler_v1,
179 NULL, BATADV_TVLV_NC, 1,
180 BATADV_TVLV_HANDLER_OGM_CIFNOTFND);
181 batadv_nc_tvlv_container_update(bat_priv);
182 return 0;
184 err:
185 return -ENOMEM;
189 * batadv_nc_init_bat_priv - initialise the nc specific bat_priv variables
190 * @bat_priv: the bat priv with all the soft interface information
192 void batadv_nc_init_bat_priv(struct batadv_priv *bat_priv)
194 atomic_set(&bat_priv->network_coding, 0);
195 bat_priv->nc.min_tq = 200;
196 bat_priv->nc.max_fwd_delay = 10;
197 bat_priv->nc.max_buffer_time = 200;
201 * batadv_nc_init_orig - initialise the nc fields of an orig_node
202 * @orig_node: the orig_node which is going to be initialised
204 void batadv_nc_init_orig(struct batadv_orig_node *orig_node)
206 INIT_LIST_HEAD(&orig_node->in_coding_list);
207 INIT_LIST_HEAD(&orig_node->out_coding_list);
208 spin_lock_init(&orig_node->in_coding_list_lock);
209 spin_lock_init(&orig_node->out_coding_list_lock);
213 * batadv_nc_node_release - release nc_node from lists and queue for free after
214 * rcu grace period
215 * @ref: kref pointer of the nc_node
217 static void batadv_nc_node_release(struct kref *ref)
219 struct batadv_nc_node *nc_node;
221 nc_node = container_of(ref, struct batadv_nc_node, refcount);
223 batadv_orig_node_put(nc_node->orig_node);
224 kfree_rcu(nc_node, rcu);
228 * batadv_nc_node_put - decrement the nc_node refcounter and possibly
229 * release it
230 * @nc_node: nc_node to be free'd
232 static void batadv_nc_node_put(struct batadv_nc_node *nc_node)
234 kref_put(&nc_node->refcount, batadv_nc_node_release);
238 * batadv_nc_path_release - release nc_path from lists and queue for free after
239 * rcu grace period
240 * @ref: kref pointer of the nc_path
242 static void batadv_nc_path_release(struct kref *ref)
244 struct batadv_nc_path *nc_path;
246 nc_path = container_of(ref, struct batadv_nc_path, refcount);
248 kfree_rcu(nc_path, rcu);
252 * batadv_nc_path_put - decrement the nc_path refcounter and possibly
253 * release it
254 * @nc_path: nc_path to be free'd
256 static void batadv_nc_path_put(struct batadv_nc_path *nc_path)
258 kref_put(&nc_path->refcount, batadv_nc_path_release);
262 * batadv_nc_packet_free - frees nc packet
263 * @nc_packet: the nc packet to free
265 static void batadv_nc_packet_free(struct batadv_nc_packet *nc_packet)
267 kfree_skb(nc_packet->skb);
268 batadv_nc_path_put(nc_packet->nc_path);
269 kfree(nc_packet);
273 * batadv_nc_to_purge_nc_node - checks whether an nc node has to be purged
274 * @bat_priv: the bat priv with all the soft interface information
275 * @nc_node: the nc node to check
277 * Return: true if the entry has to be purged now, false otherwise
279 static bool batadv_nc_to_purge_nc_node(struct batadv_priv *bat_priv,
280 struct batadv_nc_node *nc_node)
282 if (atomic_read(&bat_priv->mesh_state) != BATADV_MESH_ACTIVE)
283 return true;
285 return batadv_has_timed_out(nc_node->last_seen, BATADV_NC_NODE_TIMEOUT);
289 * batadv_nc_to_purge_nc_path_coding - checks whether an nc path has timed out
290 * @bat_priv: the bat priv with all the soft interface information
291 * @nc_path: the nc path to check
293 * Return: true if the entry has to be purged now, false otherwise
295 static bool batadv_nc_to_purge_nc_path_coding(struct batadv_priv *bat_priv,
296 struct batadv_nc_path *nc_path)
298 if (atomic_read(&bat_priv->mesh_state) != BATADV_MESH_ACTIVE)
299 return true;
301 /* purge the path when no packets has been added for 10 times the
302 * max_fwd_delay time
304 return batadv_has_timed_out(nc_path->last_valid,
305 bat_priv->nc.max_fwd_delay * 10);
309 * batadv_nc_to_purge_nc_path_decoding - checks whether an nc path has timed out
310 * @bat_priv: the bat priv with all the soft interface information
311 * @nc_path: the nc path to check
313 * Return: true if the entry has to be purged now, false otherwise
315 static bool batadv_nc_to_purge_nc_path_decoding(struct batadv_priv *bat_priv,
316 struct batadv_nc_path *nc_path)
318 if (atomic_read(&bat_priv->mesh_state) != BATADV_MESH_ACTIVE)
319 return true;
321 /* purge the path when no packets has been added for 10 times the
322 * max_buffer time
324 return batadv_has_timed_out(nc_path->last_valid,
325 bat_priv->nc.max_buffer_time * 10);
329 * batadv_nc_purge_orig_nc_nodes - go through list of nc nodes and purge stale
330 * entries
331 * @bat_priv: the bat priv with all the soft interface information
332 * @list: list of nc nodes
333 * @lock: nc node list lock
334 * @to_purge: function in charge to decide whether an entry has to be purged or
335 * not. This function takes the nc node as argument and has to return
336 * a boolean value: true if the entry has to be deleted, false
337 * otherwise
339 static void
340 batadv_nc_purge_orig_nc_nodes(struct batadv_priv *bat_priv,
341 struct list_head *list,
342 spinlock_t *lock,
343 bool (*to_purge)(struct batadv_priv *,
344 struct batadv_nc_node *))
346 struct batadv_nc_node *nc_node, *nc_node_tmp;
348 /* For each nc_node in list */
349 spin_lock_bh(lock);
350 list_for_each_entry_safe(nc_node, nc_node_tmp, list, list) {
351 /* if an helper function has been passed as parameter,
352 * ask it if the entry has to be purged or not
354 if (to_purge && !to_purge(bat_priv, nc_node))
355 continue;
357 batadv_dbg(BATADV_DBG_NC, bat_priv,
358 "Removing nc_node %pM -> %pM\n",
359 nc_node->addr, nc_node->orig_node->orig);
360 list_del_rcu(&nc_node->list);
361 batadv_nc_node_put(nc_node);
363 spin_unlock_bh(lock);
367 * batadv_nc_purge_orig - purges all nc node data attached of the given
368 * originator
369 * @bat_priv: the bat priv with all the soft interface information
370 * @orig_node: orig_node with the nc node entries to be purged
371 * @to_purge: function in charge to decide whether an entry has to be purged or
372 * not. This function takes the nc node as argument and has to return
373 * a boolean value: true is the entry has to be deleted, false
374 * otherwise
376 void batadv_nc_purge_orig(struct batadv_priv *bat_priv,
377 struct batadv_orig_node *orig_node,
378 bool (*to_purge)(struct batadv_priv *,
379 struct batadv_nc_node *))
381 /* Check ingoing nc_node's of this orig_node */
382 batadv_nc_purge_orig_nc_nodes(bat_priv, &orig_node->in_coding_list,
383 &orig_node->in_coding_list_lock,
384 to_purge);
386 /* Check outgoing nc_node's of this orig_node */
387 batadv_nc_purge_orig_nc_nodes(bat_priv, &orig_node->out_coding_list,
388 &orig_node->out_coding_list_lock,
389 to_purge);
393 * batadv_nc_purge_orig_hash - traverse entire originator hash to check if they
394 * have timed out nc nodes
395 * @bat_priv: the bat priv with all the soft interface information
397 static void batadv_nc_purge_orig_hash(struct batadv_priv *bat_priv)
399 struct batadv_hashtable *hash = bat_priv->orig_hash;
400 struct hlist_head *head;
401 struct batadv_orig_node *orig_node;
402 u32 i;
404 if (!hash)
405 return;
407 /* For each orig_node */
408 for (i = 0; i < hash->size; i++) {
409 head = &hash->table[i];
411 rcu_read_lock();
412 hlist_for_each_entry_rcu(orig_node, head, hash_entry)
413 batadv_nc_purge_orig(bat_priv, orig_node,
414 batadv_nc_to_purge_nc_node);
415 rcu_read_unlock();
420 * batadv_nc_purge_paths - traverse all nc paths part of the hash and remove
421 * unused ones
422 * @bat_priv: the bat priv with all the soft interface information
423 * @hash: hash table containing the nc paths to check
424 * @to_purge: function in charge to decide whether an entry has to be purged or
425 * not. This function takes the nc node as argument and has to return
426 * a boolean value: true is the entry has to be deleted, false
427 * otherwise
429 static void batadv_nc_purge_paths(struct batadv_priv *bat_priv,
430 struct batadv_hashtable *hash,
431 bool (*to_purge)(struct batadv_priv *,
432 struct batadv_nc_path *))
434 struct hlist_head *head;
435 struct hlist_node *node_tmp;
436 struct batadv_nc_path *nc_path;
437 spinlock_t *lock; /* Protects lists in hash */
438 u32 i;
440 for (i = 0; i < hash->size; i++) {
441 head = &hash->table[i];
442 lock = &hash->list_locks[i];
444 /* For each nc_path in this bin */
445 spin_lock_bh(lock);
446 hlist_for_each_entry_safe(nc_path, node_tmp, head, hash_entry) {
447 /* if an helper function has been passed as parameter,
448 * ask it if the entry has to be purged or not
450 if (to_purge && !to_purge(bat_priv, nc_path))
451 continue;
453 /* purging an non-empty nc_path should never happen, but
454 * is observed under high CPU load. Delay the purging
455 * until next iteration to allow the packet_list to be
456 * emptied first.
458 if (!unlikely(list_empty(&nc_path->packet_list))) {
459 net_ratelimited_function(printk,
460 KERN_WARNING
461 "Skipping free of non-empty nc_path (%pM -> %pM)!\n",
462 nc_path->prev_hop,
463 nc_path->next_hop);
464 continue;
467 /* nc_path is unused, so remove it */
468 batadv_dbg(BATADV_DBG_NC, bat_priv,
469 "Remove nc_path %pM -> %pM\n",
470 nc_path->prev_hop, nc_path->next_hop);
471 hlist_del_rcu(&nc_path->hash_entry);
472 batadv_nc_path_put(nc_path);
474 spin_unlock_bh(lock);
479 * batadv_nc_hash_key_gen - computes the nc_path hash key
480 * @key: buffer to hold the final hash key
481 * @src: source ethernet mac address going into the hash key
482 * @dst: destination ethernet mac address going into the hash key
484 static void batadv_nc_hash_key_gen(struct batadv_nc_path *key, const char *src,
485 const char *dst)
487 memcpy(key->prev_hop, src, sizeof(key->prev_hop));
488 memcpy(key->next_hop, dst, sizeof(key->next_hop));
492 * batadv_nc_hash_choose - compute the hash value for an nc path
493 * @data: data to hash
494 * @size: size of the hash table
496 * Return: the selected index in the hash table for the given data.
498 static u32 batadv_nc_hash_choose(const void *data, u32 size)
500 const struct batadv_nc_path *nc_path = data;
501 u32 hash = 0;
503 hash = jhash(&nc_path->prev_hop, sizeof(nc_path->prev_hop), hash);
504 hash = jhash(&nc_path->next_hop, sizeof(nc_path->next_hop), hash);
506 return hash % size;
510 * batadv_nc_hash_compare - comparing function used in the network coding hash
511 * tables
512 * @node: node in the local table
513 * @data2: second object to compare the node to
515 * Return: true if the two entry are the same, false otherwise
517 static bool batadv_nc_hash_compare(const struct hlist_node *node,
518 const void *data2)
520 const struct batadv_nc_path *nc_path1, *nc_path2;
522 nc_path1 = container_of(node, struct batadv_nc_path, hash_entry);
523 nc_path2 = data2;
525 /* Return 1 if the two keys are identical */
526 if (!batadv_compare_eth(nc_path1->prev_hop, nc_path2->prev_hop))
527 return false;
529 if (!batadv_compare_eth(nc_path1->next_hop, nc_path2->next_hop))
530 return false;
532 return true;
536 * batadv_nc_hash_find - search for an existing nc path and return it
537 * @hash: hash table containing the nc path
538 * @data: search key
540 * Return: the nc_path if found, NULL otherwise.
542 static struct batadv_nc_path *
543 batadv_nc_hash_find(struct batadv_hashtable *hash,
544 void *data)
546 struct hlist_head *head;
547 struct batadv_nc_path *nc_path, *nc_path_tmp = NULL;
548 int index;
550 if (!hash)
551 return NULL;
553 index = batadv_nc_hash_choose(data, hash->size);
554 head = &hash->table[index];
556 rcu_read_lock();
557 hlist_for_each_entry_rcu(nc_path, head, hash_entry) {
558 if (!batadv_nc_hash_compare(&nc_path->hash_entry, data))
559 continue;
561 if (!kref_get_unless_zero(&nc_path->refcount))
562 continue;
564 nc_path_tmp = nc_path;
565 break;
567 rcu_read_unlock();
569 return nc_path_tmp;
573 * batadv_nc_send_packet - send non-coded packet and free nc_packet struct
574 * @nc_packet: the nc packet to send
576 static void batadv_nc_send_packet(struct batadv_nc_packet *nc_packet)
578 batadv_send_unicast_skb(nc_packet->skb, nc_packet->neigh_node);
579 nc_packet->skb = NULL;
580 batadv_nc_packet_free(nc_packet);
584 * batadv_nc_sniffed_purge - Checks timestamp of given sniffed nc_packet.
585 * @bat_priv: the bat priv with all the soft interface information
586 * @nc_path: the nc path the packet belongs to
587 * @nc_packet: the nc packet to be checked
589 * Checks whether the given sniffed (overheard) nc_packet has hit its buffering
590 * timeout. If so, the packet is no longer kept and the entry deleted from the
591 * queue. Has to be called with the appropriate locks.
593 * Return: false as soon as the entry in the fifo queue has not been timed out
594 * yet and true otherwise.
596 static bool batadv_nc_sniffed_purge(struct batadv_priv *bat_priv,
597 struct batadv_nc_path *nc_path,
598 struct batadv_nc_packet *nc_packet)
600 unsigned long timeout = bat_priv->nc.max_buffer_time;
601 bool res = false;
603 lockdep_assert_held(&nc_path->packet_list_lock);
605 /* Packets are added to tail, so the remaining packets did not time
606 * out and we can stop processing the current queue
608 if (atomic_read(&bat_priv->mesh_state) == BATADV_MESH_ACTIVE &&
609 !batadv_has_timed_out(nc_packet->timestamp, timeout))
610 goto out;
612 /* purge nc packet */
613 list_del(&nc_packet->list);
614 batadv_nc_packet_free(nc_packet);
616 res = true;
618 out:
619 return res;
623 * batadv_nc_fwd_flush - Checks the timestamp of the given nc packet.
624 * @bat_priv: the bat priv with all the soft interface information
625 * @nc_path: the nc path the packet belongs to
626 * @nc_packet: the nc packet to be checked
628 * Checks whether the given nc packet has hit its forward timeout. If so, the
629 * packet is no longer delayed, immediately sent and the entry deleted from the
630 * queue. Has to be called with the appropriate locks.
632 * Return: false as soon as the entry in the fifo queue has not been timed out
633 * yet and true otherwise.
635 static bool batadv_nc_fwd_flush(struct batadv_priv *bat_priv,
636 struct batadv_nc_path *nc_path,
637 struct batadv_nc_packet *nc_packet)
639 unsigned long timeout = bat_priv->nc.max_fwd_delay;
641 lockdep_assert_held(&nc_path->packet_list_lock);
643 /* Packets are added to tail, so the remaining packets did not time
644 * out and we can stop processing the current queue
646 if (atomic_read(&bat_priv->mesh_state) == BATADV_MESH_ACTIVE &&
647 !batadv_has_timed_out(nc_packet->timestamp, timeout))
648 return false;
650 /* Send packet */
651 batadv_inc_counter(bat_priv, BATADV_CNT_FORWARD);
652 batadv_add_counter(bat_priv, BATADV_CNT_FORWARD_BYTES,
653 nc_packet->skb->len + ETH_HLEN);
654 list_del(&nc_packet->list);
655 batadv_nc_send_packet(nc_packet);
657 return true;
661 * batadv_nc_process_nc_paths - traverse given nc packet pool and free timed out
662 * nc packets
663 * @bat_priv: the bat priv with all the soft interface information
664 * @hash: to be processed hash table
665 * @process_fn: Function called to process given nc packet. Should return true
666 * to encourage this function to proceed with the next packet.
667 * Otherwise the rest of the current queue is skipped.
669 static void
670 batadv_nc_process_nc_paths(struct batadv_priv *bat_priv,
671 struct batadv_hashtable *hash,
672 bool (*process_fn)(struct batadv_priv *,
673 struct batadv_nc_path *,
674 struct batadv_nc_packet *))
676 struct hlist_head *head;
677 struct batadv_nc_packet *nc_packet, *nc_packet_tmp;
678 struct batadv_nc_path *nc_path;
679 bool ret;
680 int i;
682 if (!hash)
683 return;
685 /* Loop hash table bins */
686 for (i = 0; i < hash->size; i++) {
687 head = &hash->table[i];
689 /* Loop coding paths */
690 rcu_read_lock();
691 hlist_for_each_entry_rcu(nc_path, head, hash_entry) {
692 /* Loop packets */
693 spin_lock_bh(&nc_path->packet_list_lock);
694 list_for_each_entry_safe(nc_packet, nc_packet_tmp,
695 &nc_path->packet_list, list) {
696 ret = process_fn(bat_priv, nc_path, nc_packet);
697 if (!ret)
698 break;
700 spin_unlock_bh(&nc_path->packet_list_lock);
702 rcu_read_unlock();
707 * batadv_nc_worker - periodic task for house keeping related to network coding
708 * @work: kernel work struct
710 static void batadv_nc_worker(struct work_struct *work)
712 struct delayed_work *delayed_work;
713 struct batadv_priv_nc *priv_nc;
714 struct batadv_priv *bat_priv;
715 unsigned long timeout;
717 delayed_work = to_delayed_work(work);
718 priv_nc = container_of(delayed_work, struct batadv_priv_nc, work);
719 bat_priv = container_of(priv_nc, struct batadv_priv, nc);
721 batadv_nc_purge_orig_hash(bat_priv);
722 batadv_nc_purge_paths(bat_priv, bat_priv->nc.coding_hash,
723 batadv_nc_to_purge_nc_path_coding);
724 batadv_nc_purge_paths(bat_priv, bat_priv->nc.decoding_hash,
725 batadv_nc_to_purge_nc_path_decoding);
727 timeout = bat_priv->nc.max_fwd_delay;
729 if (batadv_has_timed_out(bat_priv->nc.timestamp_fwd_flush, timeout)) {
730 batadv_nc_process_nc_paths(bat_priv, bat_priv->nc.coding_hash,
731 batadv_nc_fwd_flush);
732 bat_priv->nc.timestamp_fwd_flush = jiffies;
735 if (batadv_has_timed_out(bat_priv->nc.timestamp_sniffed_purge,
736 bat_priv->nc.max_buffer_time)) {
737 batadv_nc_process_nc_paths(bat_priv, bat_priv->nc.decoding_hash,
738 batadv_nc_sniffed_purge);
739 bat_priv->nc.timestamp_sniffed_purge = jiffies;
742 /* Schedule a new check */
743 batadv_nc_start_timer(bat_priv);
747 * batadv_can_nc_with_orig - checks whether the given orig node is suitable for
748 * coding or not
749 * @bat_priv: the bat priv with all the soft interface information
750 * @orig_node: neighboring orig node which may be used as nc candidate
751 * @ogm_packet: incoming ogm packet also used for the checks
753 * Return: true if:
754 * 1) The OGM must have the most recent sequence number.
755 * 2) The TTL must be decremented by one and only one.
756 * 3) The OGM must be received from the first hop from orig_node.
757 * 4) The TQ value of the OGM must be above bat_priv->nc.min_tq.
759 static bool batadv_can_nc_with_orig(struct batadv_priv *bat_priv,
760 struct batadv_orig_node *orig_node,
761 struct batadv_ogm_packet *ogm_packet)
763 struct batadv_orig_ifinfo *orig_ifinfo;
764 u32 last_real_seqno;
765 u8 last_ttl;
767 orig_ifinfo = batadv_orig_ifinfo_get(orig_node, BATADV_IF_DEFAULT);
768 if (!orig_ifinfo)
769 return false;
771 last_ttl = orig_ifinfo->last_ttl;
772 last_real_seqno = orig_ifinfo->last_real_seqno;
773 batadv_orig_ifinfo_put(orig_ifinfo);
775 if (last_real_seqno != ntohl(ogm_packet->seqno))
776 return false;
777 if (last_ttl != ogm_packet->ttl + 1)
778 return false;
779 if (!batadv_compare_eth(ogm_packet->orig, ogm_packet->prev_sender))
780 return false;
781 if (ogm_packet->tq < bat_priv->nc.min_tq)
782 return false;
784 return true;
788 * batadv_nc_find_nc_node - search for an existing nc node and return it
789 * @orig_node: orig node originating the ogm packet
790 * @orig_neigh_node: neighboring orig node from which we received the ogm packet
791 * (can be equal to orig_node)
792 * @in_coding: traverse incoming or outgoing network coding list
794 * Return: the nc_node if found, NULL otherwise.
796 static struct batadv_nc_node *
797 batadv_nc_find_nc_node(struct batadv_orig_node *orig_node,
798 struct batadv_orig_node *orig_neigh_node,
799 bool in_coding)
801 struct batadv_nc_node *nc_node, *nc_node_out = NULL;
802 struct list_head *list;
804 if (in_coding)
805 list = &orig_neigh_node->in_coding_list;
806 else
807 list = &orig_neigh_node->out_coding_list;
809 /* Traverse list of nc_nodes to orig_node */
810 rcu_read_lock();
811 list_for_each_entry_rcu(nc_node, list, list) {
812 if (!batadv_compare_eth(nc_node->addr, orig_node->orig))
813 continue;
815 if (!kref_get_unless_zero(&nc_node->refcount))
816 continue;
818 /* Found a match */
819 nc_node_out = nc_node;
820 break;
822 rcu_read_unlock();
824 return nc_node_out;
828 * batadv_nc_get_nc_node - retrieves an nc node or creates the entry if it was
829 * not found
830 * @bat_priv: the bat priv with all the soft interface information
831 * @orig_node: orig node originating the ogm packet
832 * @orig_neigh_node: neighboring orig node from which we received the ogm packet
833 * (can be equal to orig_node)
834 * @in_coding: traverse incoming or outgoing network coding list
836 * Return: the nc_node if found or created, NULL in case of an error.
838 static struct batadv_nc_node *
839 batadv_nc_get_nc_node(struct batadv_priv *bat_priv,
840 struct batadv_orig_node *orig_node,
841 struct batadv_orig_node *orig_neigh_node,
842 bool in_coding)
844 struct batadv_nc_node *nc_node;
845 spinlock_t *lock; /* Used to lock list selected by "int in_coding" */
846 struct list_head *list;
848 /* Check if nc_node is already added */
849 nc_node = batadv_nc_find_nc_node(orig_node, orig_neigh_node, in_coding);
851 /* Node found */
852 if (nc_node)
853 return nc_node;
855 nc_node = kzalloc(sizeof(*nc_node), GFP_ATOMIC);
856 if (!nc_node)
857 return NULL;
859 kref_get(&orig_neigh_node->refcount);
861 /* Initialize nc_node */
862 INIT_LIST_HEAD(&nc_node->list);
863 ether_addr_copy(nc_node->addr, orig_node->orig);
864 nc_node->orig_node = orig_neigh_node;
865 kref_init(&nc_node->refcount);
866 kref_get(&nc_node->refcount);
868 /* Select ingoing or outgoing coding node */
869 if (in_coding) {
870 lock = &orig_neigh_node->in_coding_list_lock;
871 list = &orig_neigh_node->in_coding_list;
872 } else {
873 lock = &orig_neigh_node->out_coding_list_lock;
874 list = &orig_neigh_node->out_coding_list;
877 batadv_dbg(BATADV_DBG_NC, bat_priv, "Adding nc_node %pM -> %pM\n",
878 nc_node->addr, nc_node->orig_node->orig);
880 /* Add nc_node to orig_node */
881 spin_lock_bh(lock);
882 list_add_tail_rcu(&nc_node->list, list);
883 spin_unlock_bh(lock);
885 return nc_node;
889 * batadv_nc_update_nc_node - updates stored incoming and outgoing nc node
890 * structs (best called on incoming OGMs)
891 * @bat_priv: the bat priv with all the soft interface information
892 * @orig_node: orig node originating the ogm packet
893 * @orig_neigh_node: neighboring orig node from which we received the ogm packet
894 * (can be equal to orig_node)
895 * @ogm_packet: incoming ogm packet
896 * @is_single_hop_neigh: orig_node is a single hop neighbor
898 void batadv_nc_update_nc_node(struct batadv_priv *bat_priv,
899 struct batadv_orig_node *orig_node,
900 struct batadv_orig_node *orig_neigh_node,
901 struct batadv_ogm_packet *ogm_packet,
902 int is_single_hop_neigh)
904 struct batadv_nc_node *in_nc_node = NULL;
905 struct batadv_nc_node *out_nc_node = NULL;
907 /* Check if network coding is enabled */
908 if (!atomic_read(&bat_priv->network_coding))
909 goto out;
911 /* check if orig node is network coding enabled */
912 if (!test_bit(BATADV_ORIG_CAPA_HAS_NC, &orig_node->capabilities))
913 goto out;
915 /* accept ogms from 'good' neighbors and single hop neighbors */
916 if (!batadv_can_nc_with_orig(bat_priv, orig_node, ogm_packet) &&
917 !is_single_hop_neigh)
918 goto out;
920 /* Add orig_node as in_nc_node on hop */
921 in_nc_node = batadv_nc_get_nc_node(bat_priv, orig_node,
922 orig_neigh_node, true);
923 if (!in_nc_node)
924 goto out;
926 in_nc_node->last_seen = jiffies;
928 /* Add hop as out_nc_node on orig_node */
929 out_nc_node = batadv_nc_get_nc_node(bat_priv, orig_neigh_node,
930 orig_node, false);
931 if (!out_nc_node)
932 goto out;
934 out_nc_node->last_seen = jiffies;
936 out:
937 if (in_nc_node)
938 batadv_nc_node_put(in_nc_node);
939 if (out_nc_node)
940 batadv_nc_node_put(out_nc_node);
944 * batadv_nc_get_path - get existing nc_path or allocate a new one
945 * @bat_priv: the bat priv with all the soft interface information
946 * @hash: hash table containing the nc path
947 * @src: ethernet source address - first half of the nc path search key
948 * @dst: ethernet destination address - second half of the nc path search key
950 * Return: pointer to nc_path if the path was found or created, returns NULL
951 * on error.
953 static struct batadv_nc_path *batadv_nc_get_path(struct batadv_priv *bat_priv,
954 struct batadv_hashtable *hash,
955 u8 *src,
956 u8 *dst)
958 int hash_added;
959 struct batadv_nc_path *nc_path, nc_path_key;
961 batadv_nc_hash_key_gen(&nc_path_key, src, dst);
963 /* Search for existing nc_path */
964 nc_path = batadv_nc_hash_find(hash, (void *)&nc_path_key);
966 if (nc_path) {
967 /* Set timestamp to delay removal of nc_path */
968 nc_path->last_valid = jiffies;
969 return nc_path;
972 /* No existing nc_path was found; create a new */
973 nc_path = kzalloc(sizeof(*nc_path), GFP_ATOMIC);
975 if (!nc_path)
976 return NULL;
978 /* Initialize nc_path */
979 INIT_LIST_HEAD(&nc_path->packet_list);
980 spin_lock_init(&nc_path->packet_list_lock);
981 kref_init(&nc_path->refcount);
982 kref_get(&nc_path->refcount);
983 nc_path->last_valid = jiffies;
984 ether_addr_copy(nc_path->next_hop, dst);
985 ether_addr_copy(nc_path->prev_hop, src);
987 batadv_dbg(BATADV_DBG_NC, bat_priv, "Adding nc_path %pM -> %pM\n",
988 nc_path->prev_hop,
989 nc_path->next_hop);
991 /* Add nc_path to hash table */
992 hash_added = batadv_hash_add(hash, batadv_nc_hash_compare,
993 batadv_nc_hash_choose, &nc_path_key,
994 &nc_path->hash_entry);
996 if (hash_added < 0) {
997 kfree(nc_path);
998 return NULL;
1001 return nc_path;
1005 * batadv_nc_random_weight_tq - scale the receivers TQ-value to avoid unfair
1006 * selection of a receiver with slightly lower TQ than the other
1007 * @tq: to be weighted tq value
1009 * Return: scaled tq value
1011 static u8 batadv_nc_random_weight_tq(u8 tq)
1013 u8 rand_val, rand_tq;
1015 get_random_bytes(&rand_val, sizeof(rand_val));
1017 /* randomize the estimated packet loss (max TQ - estimated TQ) */
1018 rand_tq = rand_val * (BATADV_TQ_MAX_VALUE - tq);
1020 /* normalize the randomized packet loss */
1021 rand_tq /= BATADV_TQ_MAX_VALUE;
1023 /* convert to (randomized) estimated tq again */
1024 return BATADV_TQ_MAX_VALUE - rand_tq;
1028 * batadv_nc_memxor - XOR destination with source
1029 * @dst: byte array to XOR into
1030 * @src: byte array to XOR from
1031 * @len: length of destination array
1033 static void batadv_nc_memxor(char *dst, const char *src, unsigned int len)
1035 unsigned int i;
1037 for (i = 0; i < len; ++i)
1038 dst[i] ^= src[i];
1042 * batadv_nc_code_packets - code a received unicast_packet with an nc packet
1043 * into a coded_packet and send it
1044 * @bat_priv: the bat priv with all the soft interface information
1045 * @skb: data skb to forward
1046 * @ethhdr: pointer to the ethernet header inside the skb
1047 * @nc_packet: structure containing the packet to the skb can be coded with
1048 * @neigh_node: next hop to forward packet to
1050 * Return: true if both packets are consumed, false otherwise.
1052 static bool batadv_nc_code_packets(struct batadv_priv *bat_priv,
1053 struct sk_buff *skb,
1054 struct ethhdr *ethhdr,
1055 struct batadv_nc_packet *nc_packet,
1056 struct batadv_neigh_node *neigh_node)
1058 u8 tq_weighted_neigh, tq_weighted_coding, tq_tmp;
1059 struct sk_buff *skb_dest, *skb_src;
1060 struct batadv_unicast_packet *packet1;
1061 struct batadv_unicast_packet *packet2;
1062 struct batadv_coded_packet *coded_packet;
1063 struct batadv_neigh_node *neigh_tmp, *router_neigh, *first_dest;
1064 struct batadv_neigh_node *router_coding = NULL, *second_dest;
1065 struct batadv_neigh_ifinfo *router_neigh_ifinfo = NULL;
1066 struct batadv_neigh_ifinfo *router_coding_ifinfo = NULL;
1067 u8 *first_source, *second_source;
1068 __be32 packet_id1, packet_id2;
1069 size_t count;
1070 bool res = false;
1071 int coding_len;
1072 int unicast_size = sizeof(*packet1);
1073 int coded_size = sizeof(*coded_packet);
1074 int header_add = coded_size - unicast_size;
1076 /* TODO: do we need to consider the outgoing interface for
1077 * coded packets?
1079 router_neigh = batadv_orig_router_get(neigh_node->orig_node,
1080 BATADV_IF_DEFAULT);
1081 if (!router_neigh)
1082 goto out;
1084 router_neigh_ifinfo = batadv_neigh_ifinfo_get(router_neigh,
1085 BATADV_IF_DEFAULT);
1086 if (!router_neigh_ifinfo)
1087 goto out;
1089 neigh_tmp = nc_packet->neigh_node;
1090 router_coding = batadv_orig_router_get(neigh_tmp->orig_node,
1091 BATADV_IF_DEFAULT);
1092 if (!router_coding)
1093 goto out;
1095 router_coding_ifinfo = batadv_neigh_ifinfo_get(router_coding,
1096 BATADV_IF_DEFAULT);
1097 if (!router_coding_ifinfo)
1098 goto out;
1100 tq_tmp = router_neigh_ifinfo->bat_iv.tq_avg;
1101 tq_weighted_neigh = batadv_nc_random_weight_tq(tq_tmp);
1102 tq_tmp = router_coding_ifinfo->bat_iv.tq_avg;
1103 tq_weighted_coding = batadv_nc_random_weight_tq(tq_tmp);
1105 /* Select one destination for the MAC-header dst-field based on
1106 * weighted TQ-values.
1108 if (tq_weighted_neigh >= tq_weighted_coding) {
1109 /* Destination from nc_packet is selected for MAC-header */
1110 first_dest = nc_packet->neigh_node;
1111 first_source = nc_packet->nc_path->prev_hop;
1112 second_dest = neigh_node;
1113 second_source = ethhdr->h_source;
1114 packet1 = (struct batadv_unicast_packet *)nc_packet->skb->data;
1115 packet2 = (struct batadv_unicast_packet *)skb->data;
1116 packet_id1 = nc_packet->packet_id;
1117 packet_id2 = batadv_skb_crc32(skb,
1118 skb->data + sizeof(*packet2));
1119 } else {
1120 /* Destination for skb is selected for MAC-header */
1121 first_dest = neigh_node;
1122 first_source = ethhdr->h_source;
1123 second_dest = nc_packet->neigh_node;
1124 second_source = nc_packet->nc_path->prev_hop;
1125 packet1 = (struct batadv_unicast_packet *)skb->data;
1126 packet2 = (struct batadv_unicast_packet *)nc_packet->skb->data;
1127 packet_id1 = batadv_skb_crc32(skb,
1128 skb->data + sizeof(*packet1));
1129 packet_id2 = nc_packet->packet_id;
1132 /* Instead of zero padding the smallest data buffer, we
1133 * code into the largest.
1135 if (skb->len <= nc_packet->skb->len) {
1136 skb_dest = nc_packet->skb;
1137 skb_src = skb;
1138 } else {
1139 skb_dest = skb;
1140 skb_src = nc_packet->skb;
1143 /* coding_len is used when decoding the packet shorter packet */
1144 coding_len = skb_src->len - unicast_size;
1146 if (skb_linearize(skb_dest) < 0 || skb_linearize(skb_src) < 0)
1147 goto out;
1149 skb_push(skb_dest, header_add);
1151 coded_packet = (struct batadv_coded_packet *)skb_dest->data;
1152 skb_reset_mac_header(skb_dest);
1154 coded_packet->packet_type = BATADV_CODED;
1155 coded_packet->version = BATADV_COMPAT_VERSION;
1156 coded_packet->ttl = packet1->ttl;
1158 /* Info about first unicast packet */
1159 ether_addr_copy(coded_packet->first_source, first_source);
1160 ether_addr_copy(coded_packet->first_orig_dest, packet1->dest);
1161 coded_packet->first_crc = packet_id1;
1162 coded_packet->first_ttvn = packet1->ttvn;
1164 /* Info about second unicast packet */
1165 ether_addr_copy(coded_packet->second_dest, second_dest->addr);
1166 ether_addr_copy(coded_packet->second_source, second_source);
1167 ether_addr_copy(coded_packet->second_orig_dest, packet2->dest);
1168 coded_packet->second_crc = packet_id2;
1169 coded_packet->second_ttl = packet2->ttl;
1170 coded_packet->second_ttvn = packet2->ttvn;
1171 coded_packet->coded_len = htons(coding_len);
1173 /* This is where the magic happens: Code skb_src into skb_dest */
1174 batadv_nc_memxor(skb_dest->data + coded_size,
1175 skb_src->data + unicast_size, coding_len);
1177 /* Update counters accordingly */
1178 if (BATADV_SKB_CB(skb_src)->decoded &&
1179 BATADV_SKB_CB(skb_dest)->decoded) {
1180 /* Both packets are recoded */
1181 count = skb_src->len + ETH_HLEN;
1182 count += skb_dest->len + ETH_HLEN;
1183 batadv_add_counter(bat_priv, BATADV_CNT_NC_RECODE, 2);
1184 batadv_add_counter(bat_priv, BATADV_CNT_NC_RECODE_BYTES, count);
1185 } else if (!BATADV_SKB_CB(skb_src)->decoded &&
1186 !BATADV_SKB_CB(skb_dest)->decoded) {
1187 /* Both packets are newly coded */
1188 count = skb_src->len + ETH_HLEN;
1189 count += skb_dest->len + ETH_HLEN;
1190 batadv_add_counter(bat_priv, BATADV_CNT_NC_CODE, 2);
1191 batadv_add_counter(bat_priv, BATADV_CNT_NC_CODE_BYTES, count);
1192 } else if (BATADV_SKB_CB(skb_src)->decoded &&
1193 !BATADV_SKB_CB(skb_dest)->decoded) {
1194 /* skb_src recoded and skb_dest is newly coded */
1195 batadv_inc_counter(bat_priv, BATADV_CNT_NC_RECODE);
1196 batadv_add_counter(bat_priv, BATADV_CNT_NC_RECODE_BYTES,
1197 skb_src->len + ETH_HLEN);
1198 batadv_inc_counter(bat_priv, BATADV_CNT_NC_CODE);
1199 batadv_add_counter(bat_priv, BATADV_CNT_NC_CODE_BYTES,
1200 skb_dest->len + ETH_HLEN);
1201 } else if (!BATADV_SKB_CB(skb_src)->decoded &&
1202 BATADV_SKB_CB(skb_dest)->decoded) {
1203 /* skb_src is newly coded and skb_dest is recoded */
1204 batadv_inc_counter(bat_priv, BATADV_CNT_NC_CODE);
1205 batadv_add_counter(bat_priv, BATADV_CNT_NC_CODE_BYTES,
1206 skb_src->len + ETH_HLEN);
1207 batadv_inc_counter(bat_priv, BATADV_CNT_NC_RECODE);
1208 batadv_add_counter(bat_priv, BATADV_CNT_NC_RECODE_BYTES,
1209 skb_dest->len + ETH_HLEN);
1212 /* skb_src is now coded into skb_dest, so free it */
1213 kfree_skb(skb_src);
1215 /* avoid duplicate free of skb from nc_packet */
1216 nc_packet->skb = NULL;
1217 batadv_nc_packet_free(nc_packet);
1219 /* Send the coded packet and return true */
1220 batadv_send_unicast_skb(skb_dest, first_dest);
1221 res = true;
1222 out:
1223 if (router_neigh)
1224 batadv_neigh_node_put(router_neigh);
1225 if (router_coding)
1226 batadv_neigh_node_put(router_coding);
1227 if (router_neigh_ifinfo)
1228 batadv_neigh_ifinfo_put(router_neigh_ifinfo);
1229 if (router_coding_ifinfo)
1230 batadv_neigh_ifinfo_put(router_coding_ifinfo);
1231 return res;
1235 * batadv_nc_skb_coding_possible - true if a decoded skb is available at dst.
1236 * @skb: data skb to forward
1237 * @dst: destination mac address of the other skb to code with
1238 * @src: source mac address of skb
1240 * Whenever we network code a packet we have to check whether we received it in
1241 * a network coded form. If so, we may not be able to use it for coding because
1242 * some neighbors may also have received (overheard) the packet in the network
1243 * coded form without being able to decode it. It is hard to know which of the
1244 * neighboring nodes was able to decode the packet, therefore we can only
1245 * re-code the packet if the source of the previous encoded packet is involved.
1246 * Since the source encoded the packet we can be certain it has all necessary
1247 * decode information.
1249 * Return: true if coding of a decoded packet is allowed.
1251 static bool batadv_nc_skb_coding_possible(struct sk_buff *skb, u8 *dst, u8 *src)
1253 if (BATADV_SKB_CB(skb)->decoded && !batadv_compare_eth(dst, src))
1254 return false;
1255 return true;
1259 * batadv_nc_path_search - Find the coding path matching in_nc_node and
1260 * out_nc_node to retrieve a buffered packet that can be used for coding.
1261 * @bat_priv: the bat priv with all the soft interface information
1262 * @in_nc_node: pointer to skb next hop's neighbor nc node
1263 * @out_nc_node: pointer to skb source's neighbor nc node
1264 * @skb: data skb to forward
1265 * @eth_dst: next hop mac address of skb
1267 * Return: true if coding of a decoded skb is allowed.
1269 static struct batadv_nc_packet *
1270 batadv_nc_path_search(struct batadv_priv *bat_priv,
1271 struct batadv_nc_node *in_nc_node,
1272 struct batadv_nc_node *out_nc_node,
1273 struct sk_buff *skb,
1274 u8 *eth_dst)
1276 struct batadv_nc_path *nc_path, nc_path_key;
1277 struct batadv_nc_packet *nc_packet_out = NULL;
1278 struct batadv_nc_packet *nc_packet, *nc_packet_tmp;
1279 struct batadv_hashtable *hash = bat_priv->nc.coding_hash;
1280 int idx;
1282 if (!hash)
1283 return NULL;
1285 /* Create almost path key */
1286 batadv_nc_hash_key_gen(&nc_path_key, in_nc_node->addr,
1287 out_nc_node->addr);
1288 idx = batadv_nc_hash_choose(&nc_path_key, hash->size);
1290 /* Check for coding opportunities in this nc_path */
1291 rcu_read_lock();
1292 hlist_for_each_entry_rcu(nc_path, &hash->table[idx], hash_entry) {
1293 if (!batadv_compare_eth(nc_path->prev_hop, in_nc_node->addr))
1294 continue;
1296 if (!batadv_compare_eth(nc_path->next_hop, out_nc_node->addr))
1297 continue;
1299 spin_lock_bh(&nc_path->packet_list_lock);
1300 if (list_empty(&nc_path->packet_list)) {
1301 spin_unlock_bh(&nc_path->packet_list_lock);
1302 continue;
1305 list_for_each_entry_safe(nc_packet, nc_packet_tmp,
1306 &nc_path->packet_list, list) {
1307 if (!batadv_nc_skb_coding_possible(nc_packet->skb,
1308 eth_dst,
1309 in_nc_node->addr))
1310 continue;
1312 /* Coding opportunity is found! */
1313 list_del(&nc_packet->list);
1314 nc_packet_out = nc_packet;
1315 break;
1318 spin_unlock_bh(&nc_path->packet_list_lock);
1319 break;
1321 rcu_read_unlock();
1323 return nc_packet_out;
1327 * batadv_nc_skb_src_search - Loops through the list of neighoring nodes of the
1328 * skb's sender (may be equal to the originator).
1329 * @bat_priv: the bat priv with all the soft interface information
1330 * @skb: data skb to forward
1331 * @eth_dst: next hop mac address of skb
1332 * @eth_src: source mac address of skb
1333 * @in_nc_node: pointer to skb next hop's neighbor nc node
1335 * Return: an nc packet if a suitable coding packet was found, NULL otherwise.
1337 static struct batadv_nc_packet *
1338 batadv_nc_skb_src_search(struct batadv_priv *bat_priv,
1339 struct sk_buff *skb,
1340 u8 *eth_dst,
1341 u8 *eth_src,
1342 struct batadv_nc_node *in_nc_node)
1344 struct batadv_orig_node *orig_node;
1345 struct batadv_nc_node *out_nc_node;
1346 struct batadv_nc_packet *nc_packet = NULL;
1348 orig_node = batadv_orig_hash_find(bat_priv, eth_src);
1349 if (!orig_node)
1350 return NULL;
1352 rcu_read_lock();
1353 list_for_each_entry_rcu(out_nc_node,
1354 &orig_node->out_coding_list, list) {
1355 /* Check if the skb is decoded and if recoding is possible */
1356 if (!batadv_nc_skb_coding_possible(skb,
1357 out_nc_node->addr, eth_src))
1358 continue;
1360 /* Search for an opportunity in this nc_path */
1361 nc_packet = batadv_nc_path_search(bat_priv, in_nc_node,
1362 out_nc_node, skb, eth_dst);
1363 if (nc_packet)
1364 break;
1366 rcu_read_unlock();
1368 batadv_orig_node_put(orig_node);
1369 return nc_packet;
1373 * batadv_nc_skb_store_before_coding - set the ethernet src and dst of the
1374 * unicast skb before it is stored for use in later decoding
1375 * @bat_priv: the bat priv with all the soft interface information
1376 * @skb: data skb to store
1377 * @eth_dst_new: new destination mac address of skb
1379 static void batadv_nc_skb_store_before_coding(struct batadv_priv *bat_priv,
1380 struct sk_buff *skb,
1381 u8 *eth_dst_new)
1383 struct ethhdr *ethhdr;
1385 /* Copy skb header to change the mac header */
1386 skb = pskb_copy_for_clone(skb, GFP_ATOMIC);
1387 if (!skb)
1388 return;
1390 /* Set the mac header as if we actually sent the packet uncoded */
1391 ethhdr = eth_hdr(skb);
1392 ether_addr_copy(ethhdr->h_source, ethhdr->h_dest);
1393 ether_addr_copy(ethhdr->h_dest, eth_dst_new);
1395 /* Set data pointer to MAC header to mimic packets from our tx path */
1396 skb_push(skb, ETH_HLEN);
1398 /* Add the packet to the decoding packet pool */
1399 batadv_nc_skb_store_for_decoding(bat_priv, skb);
1401 /* batadv_nc_skb_store_for_decoding() clones the skb, so we must free
1402 * our ref
1404 kfree_skb(skb);
1408 * batadv_nc_skb_dst_search - Loops through list of neighboring nodes to dst.
1409 * @skb: data skb to forward
1410 * @neigh_node: next hop to forward packet to
1411 * @ethhdr: pointer to the ethernet header inside the skb
1413 * Loops through list of neighboring nodes the next hop has a good connection to
1414 * (receives OGMs with a sufficient quality). We need to find a neighbor of our
1415 * next hop that potentially sent a packet which our next hop also received
1416 * (overheard) and has stored for later decoding.
1418 * Return: true if the skb was consumed (encoded packet sent) or false otherwise
1420 static bool batadv_nc_skb_dst_search(struct sk_buff *skb,
1421 struct batadv_neigh_node *neigh_node,
1422 struct ethhdr *ethhdr)
1424 struct net_device *netdev = neigh_node->if_incoming->soft_iface;
1425 struct batadv_priv *bat_priv = netdev_priv(netdev);
1426 struct batadv_orig_node *orig_node = neigh_node->orig_node;
1427 struct batadv_nc_node *nc_node;
1428 struct batadv_nc_packet *nc_packet = NULL;
1430 rcu_read_lock();
1431 list_for_each_entry_rcu(nc_node, &orig_node->in_coding_list, list) {
1432 /* Search for coding opportunity with this in_nc_node */
1433 nc_packet = batadv_nc_skb_src_search(bat_priv, skb,
1434 neigh_node->addr,
1435 ethhdr->h_source, nc_node);
1437 /* Opportunity was found, so stop searching */
1438 if (nc_packet)
1439 break;
1441 rcu_read_unlock();
1443 if (!nc_packet)
1444 return false;
1446 /* Save packets for later decoding */
1447 batadv_nc_skb_store_before_coding(bat_priv, skb,
1448 neigh_node->addr);
1449 batadv_nc_skb_store_before_coding(bat_priv, nc_packet->skb,
1450 nc_packet->neigh_node->addr);
1452 /* Code and send packets */
1453 if (batadv_nc_code_packets(bat_priv, skb, ethhdr, nc_packet,
1454 neigh_node))
1455 return true;
1457 /* out of mem ? Coding failed - we have to free the buffered packet
1458 * to avoid memleaks. The skb passed as argument will be dealt with
1459 * by the calling function.
1461 batadv_nc_send_packet(nc_packet);
1462 return false;
1466 * batadv_nc_skb_add_to_path - buffer skb for later encoding / decoding
1467 * @skb: skb to add to path
1468 * @nc_path: path to add skb to
1469 * @neigh_node: next hop to forward packet to
1470 * @packet_id: checksum to identify packet
1472 * Return: true if the packet was buffered or false in case of an error.
1474 static bool batadv_nc_skb_add_to_path(struct sk_buff *skb,
1475 struct batadv_nc_path *nc_path,
1476 struct batadv_neigh_node *neigh_node,
1477 __be32 packet_id)
1479 struct batadv_nc_packet *nc_packet;
1481 nc_packet = kzalloc(sizeof(*nc_packet), GFP_ATOMIC);
1482 if (!nc_packet)
1483 return false;
1485 /* Initialize nc_packet */
1486 nc_packet->timestamp = jiffies;
1487 nc_packet->packet_id = packet_id;
1488 nc_packet->skb = skb;
1489 nc_packet->neigh_node = neigh_node;
1490 nc_packet->nc_path = nc_path;
1492 /* Add coding packet to list */
1493 spin_lock_bh(&nc_path->packet_list_lock);
1494 list_add_tail(&nc_packet->list, &nc_path->packet_list);
1495 spin_unlock_bh(&nc_path->packet_list_lock);
1497 return true;
1501 * batadv_nc_skb_forward - try to code a packet or add it to the coding packet
1502 * buffer
1503 * @skb: data skb to forward
1504 * @neigh_node: next hop to forward packet to
1506 * Return: true if the skb was consumed (encoded packet sent) or false otherwise
1508 bool batadv_nc_skb_forward(struct sk_buff *skb,
1509 struct batadv_neigh_node *neigh_node)
1511 const struct net_device *netdev = neigh_node->if_incoming->soft_iface;
1512 struct batadv_priv *bat_priv = netdev_priv(netdev);
1513 struct batadv_unicast_packet *packet;
1514 struct batadv_nc_path *nc_path;
1515 struct ethhdr *ethhdr = eth_hdr(skb);
1516 __be32 packet_id;
1517 u8 *payload;
1519 /* Check if network coding is enabled */
1520 if (!atomic_read(&bat_priv->network_coding))
1521 goto out;
1523 /* We only handle unicast packets */
1524 payload = skb_network_header(skb);
1525 packet = (struct batadv_unicast_packet *)payload;
1526 if (packet->packet_type != BATADV_UNICAST)
1527 goto out;
1529 /* Try to find a coding opportunity and send the skb if one is found */
1530 if (batadv_nc_skb_dst_search(skb, neigh_node, ethhdr))
1531 return true;
1533 /* Find or create a nc_path for this src-dst pair */
1534 nc_path = batadv_nc_get_path(bat_priv,
1535 bat_priv->nc.coding_hash,
1536 ethhdr->h_source,
1537 neigh_node->addr);
1539 if (!nc_path)
1540 goto out;
1542 /* Add skb to nc_path */
1543 packet_id = batadv_skb_crc32(skb, payload + sizeof(*packet));
1544 if (!batadv_nc_skb_add_to_path(skb, nc_path, neigh_node, packet_id))
1545 goto free_nc_path;
1547 /* Packet is consumed */
1548 return true;
1550 free_nc_path:
1551 batadv_nc_path_put(nc_path);
1552 out:
1553 /* Packet is not consumed */
1554 return false;
1558 * batadv_nc_skb_store_for_decoding - save a clone of the skb which can be used
1559 * when decoding coded packets
1560 * @bat_priv: the bat priv with all the soft interface information
1561 * @skb: data skb to store
1563 void batadv_nc_skb_store_for_decoding(struct batadv_priv *bat_priv,
1564 struct sk_buff *skb)
1566 struct batadv_unicast_packet *packet;
1567 struct batadv_nc_path *nc_path;
1568 struct ethhdr *ethhdr = eth_hdr(skb);
1569 __be32 packet_id;
1570 u8 *payload;
1572 /* Check if network coding is enabled */
1573 if (!atomic_read(&bat_priv->network_coding))
1574 goto out;
1576 /* Check for supported packet type */
1577 payload = skb_network_header(skb);
1578 packet = (struct batadv_unicast_packet *)payload;
1579 if (packet->packet_type != BATADV_UNICAST)
1580 goto out;
1582 /* Find existing nc_path or create a new */
1583 nc_path = batadv_nc_get_path(bat_priv,
1584 bat_priv->nc.decoding_hash,
1585 ethhdr->h_source,
1586 ethhdr->h_dest);
1588 if (!nc_path)
1589 goto out;
1591 /* Clone skb and adjust skb->data to point at batman header */
1592 skb = skb_clone(skb, GFP_ATOMIC);
1593 if (unlikely(!skb))
1594 goto free_nc_path;
1596 if (unlikely(!pskb_may_pull(skb, ETH_HLEN)))
1597 goto free_skb;
1599 if (unlikely(!skb_pull_rcsum(skb, ETH_HLEN)))
1600 goto free_skb;
1602 /* Add skb to nc_path */
1603 packet_id = batadv_skb_crc32(skb, payload + sizeof(*packet));
1604 if (!batadv_nc_skb_add_to_path(skb, nc_path, NULL, packet_id))
1605 goto free_skb;
1607 batadv_inc_counter(bat_priv, BATADV_CNT_NC_BUFFER);
1608 return;
1610 free_skb:
1611 kfree_skb(skb);
1612 free_nc_path:
1613 batadv_nc_path_put(nc_path);
1614 out:
1615 return;
1619 * batadv_nc_skb_store_sniffed_unicast - check if a received unicast packet
1620 * should be saved in the decoding buffer and, if so, store it there
1621 * @bat_priv: the bat priv with all the soft interface information
1622 * @skb: unicast skb to store
1624 void batadv_nc_skb_store_sniffed_unicast(struct batadv_priv *bat_priv,
1625 struct sk_buff *skb)
1627 struct ethhdr *ethhdr = eth_hdr(skb);
1629 if (batadv_is_my_mac(bat_priv, ethhdr->h_dest))
1630 return;
1632 /* Set data pointer to MAC header to mimic packets from our tx path */
1633 skb_push(skb, ETH_HLEN);
1635 batadv_nc_skb_store_for_decoding(bat_priv, skb);
1639 * batadv_nc_skb_decode_packet - decode given skb using the decode data stored
1640 * in nc_packet
1641 * @bat_priv: the bat priv with all the soft interface information
1642 * @skb: unicast skb to decode
1643 * @nc_packet: decode data needed to decode the skb
1645 * Return: pointer to decoded unicast packet if the packet was decoded or NULL
1646 * in case of an error.
1648 static struct batadv_unicast_packet *
1649 batadv_nc_skb_decode_packet(struct batadv_priv *bat_priv, struct sk_buff *skb,
1650 struct batadv_nc_packet *nc_packet)
1652 const int h_size = sizeof(struct batadv_unicast_packet);
1653 const int h_diff = sizeof(struct batadv_coded_packet) - h_size;
1654 struct batadv_unicast_packet *unicast_packet;
1655 struct batadv_coded_packet coded_packet_tmp;
1656 struct ethhdr *ethhdr, ethhdr_tmp;
1657 u8 *orig_dest, ttl, ttvn;
1658 unsigned int coding_len;
1659 int err;
1661 /* Save headers temporarily */
1662 memcpy(&coded_packet_tmp, skb->data, sizeof(coded_packet_tmp));
1663 memcpy(&ethhdr_tmp, skb_mac_header(skb), sizeof(ethhdr_tmp));
1665 if (skb_cow(skb, 0) < 0)
1666 return NULL;
1668 if (unlikely(!skb_pull_rcsum(skb, h_diff)))
1669 return NULL;
1671 /* Data points to batman header, so set mac header 14 bytes before
1672 * and network to data
1674 skb_set_mac_header(skb, -ETH_HLEN);
1675 skb_reset_network_header(skb);
1677 /* Reconstruct original mac header */
1678 ethhdr = eth_hdr(skb);
1679 *ethhdr = ethhdr_tmp;
1681 /* Select the correct unicast header information based on the location
1682 * of our mac address in the coded_packet header
1684 if (batadv_is_my_mac(bat_priv, coded_packet_tmp.second_dest)) {
1685 /* If we are the second destination the packet was overheard,
1686 * so the Ethernet address must be copied to h_dest and
1687 * pkt_type changed from PACKET_OTHERHOST to PACKET_HOST
1689 ether_addr_copy(ethhdr->h_dest, coded_packet_tmp.second_dest);
1690 skb->pkt_type = PACKET_HOST;
1692 orig_dest = coded_packet_tmp.second_orig_dest;
1693 ttl = coded_packet_tmp.second_ttl;
1694 ttvn = coded_packet_tmp.second_ttvn;
1695 } else {
1696 orig_dest = coded_packet_tmp.first_orig_dest;
1697 ttl = coded_packet_tmp.ttl;
1698 ttvn = coded_packet_tmp.first_ttvn;
1701 coding_len = ntohs(coded_packet_tmp.coded_len);
1703 if (coding_len > skb->len)
1704 return NULL;
1706 /* Here the magic is reversed:
1707 * extract the missing packet from the received coded packet
1709 batadv_nc_memxor(skb->data + h_size,
1710 nc_packet->skb->data + h_size,
1711 coding_len);
1713 /* Resize decoded skb if decoded with larger packet */
1714 if (nc_packet->skb->len > coding_len + h_size) {
1715 err = pskb_trim_rcsum(skb, coding_len + h_size);
1716 if (err)
1717 return NULL;
1720 /* Create decoded unicast packet */
1721 unicast_packet = (struct batadv_unicast_packet *)skb->data;
1722 unicast_packet->packet_type = BATADV_UNICAST;
1723 unicast_packet->version = BATADV_COMPAT_VERSION;
1724 unicast_packet->ttl = ttl;
1725 ether_addr_copy(unicast_packet->dest, orig_dest);
1726 unicast_packet->ttvn = ttvn;
1728 batadv_nc_packet_free(nc_packet);
1729 return unicast_packet;
1733 * batadv_nc_find_decoding_packet - search through buffered decoding data to
1734 * find the data needed to decode the coded packet
1735 * @bat_priv: the bat priv with all the soft interface information
1736 * @ethhdr: pointer to the ethernet header inside the coded packet
1737 * @coded: coded packet we try to find decode data for
1739 * Return: pointer to nc packet if the needed data was found or NULL otherwise.
1741 static struct batadv_nc_packet *
1742 batadv_nc_find_decoding_packet(struct batadv_priv *bat_priv,
1743 struct ethhdr *ethhdr,
1744 struct batadv_coded_packet *coded)
1746 struct batadv_hashtable *hash = bat_priv->nc.decoding_hash;
1747 struct batadv_nc_packet *tmp_nc_packet, *nc_packet = NULL;
1748 struct batadv_nc_path *nc_path, nc_path_key;
1749 u8 *dest, *source;
1750 __be32 packet_id;
1751 int index;
1753 if (!hash)
1754 return NULL;
1756 /* Select the correct packet id based on the location of our mac-addr */
1757 dest = ethhdr->h_source;
1758 if (!batadv_is_my_mac(bat_priv, coded->second_dest)) {
1759 source = coded->second_source;
1760 packet_id = coded->second_crc;
1761 } else {
1762 source = coded->first_source;
1763 packet_id = coded->first_crc;
1766 batadv_nc_hash_key_gen(&nc_path_key, source, dest);
1767 index = batadv_nc_hash_choose(&nc_path_key, hash->size);
1769 /* Search for matching coding path */
1770 rcu_read_lock();
1771 hlist_for_each_entry_rcu(nc_path, &hash->table[index], hash_entry) {
1772 /* Find matching nc_packet */
1773 spin_lock_bh(&nc_path->packet_list_lock);
1774 list_for_each_entry(tmp_nc_packet,
1775 &nc_path->packet_list, list) {
1776 if (packet_id == tmp_nc_packet->packet_id) {
1777 list_del(&tmp_nc_packet->list);
1779 nc_packet = tmp_nc_packet;
1780 break;
1783 spin_unlock_bh(&nc_path->packet_list_lock);
1785 if (nc_packet)
1786 break;
1788 rcu_read_unlock();
1790 if (!nc_packet)
1791 batadv_dbg(BATADV_DBG_NC, bat_priv,
1792 "No decoding packet found for %u\n", packet_id);
1794 return nc_packet;
1798 * batadv_nc_recv_coded_packet - try to decode coded packet and enqueue the
1799 * resulting unicast packet
1800 * @skb: incoming coded packet
1801 * @recv_if: pointer to interface this packet was received on
1803 * Return: NET_RX_SUCCESS if the packet has been consumed or NET_RX_DROP
1804 * otherwise.
1806 static int batadv_nc_recv_coded_packet(struct sk_buff *skb,
1807 struct batadv_hard_iface *recv_if)
1809 struct batadv_priv *bat_priv = netdev_priv(recv_if->soft_iface);
1810 struct batadv_unicast_packet *unicast_packet;
1811 struct batadv_coded_packet *coded_packet;
1812 struct batadv_nc_packet *nc_packet;
1813 struct ethhdr *ethhdr;
1814 int hdr_size = sizeof(*coded_packet);
1816 /* Check if network coding is enabled */
1817 if (!atomic_read(&bat_priv->network_coding))
1818 return NET_RX_DROP;
1820 /* Make sure we can access (and remove) header */
1821 if (unlikely(!pskb_may_pull(skb, hdr_size)))
1822 return NET_RX_DROP;
1824 coded_packet = (struct batadv_coded_packet *)skb->data;
1825 ethhdr = eth_hdr(skb);
1827 /* Verify frame is destined for us */
1828 if (!batadv_is_my_mac(bat_priv, ethhdr->h_dest) &&
1829 !batadv_is_my_mac(bat_priv, coded_packet->second_dest))
1830 return NET_RX_DROP;
1832 /* Update stat counter */
1833 if (batadv_is_my_mac(bat_priv, coded_packet->second_dest))
1834 batadv_inc_counter(bat_priv, BATADV_CNT_NC_SNIFFED);
1836 nc_packet = batadv_nc_find_decoding_packet(bat_priv, ethhdr,
1837 coded_packet);
1838 if (!nc_packet) {
1839 batadv_inc_counter(bat_priv, BATADV_CNT_NC_DECODE_FAILED);
1840 return NET_RX_DROP;
1843 /* Make skb's linear, because decoding accesses the entire buffer */
1844 if (skb_linearize(skb) < 0)
1845 goto free_nc_packet;
1847 if (skb_linearize(nc_packet->skb) < 0)
1848 goto free_nc_packet;
1850 /* Decode the packet */
1851 unicast_packet = batadv_nc_skb_decode_packet(bat_priv, skb, nc_packet);
1852 if (!unicast_packet) {
1853 batadv_inc_counter(bat_priv, BATADV_CNT_NC_DECODE_FAILED);
1854 goto free_nc_packet;
1857 /* Mark packet as decoded to do correct recoding when forwarding */
1858 BATADV_SKB_CB(skb)->decoded = true;
1859 batadv_inc_counter(bat_priv, BATADV_CNT_NC_DECODE);
1860 batadv_add_counter(bat_priv, BATADV_CNT_NC_DECODE_BYTES,
1861 skb->len + ETH_HLEN);
1862 return batadv_recv_unicast_packet(skb, recv_if);
1864 free_nc_packet:
1865 batadv_nc_packet_free(nc_packet);
1866 return NET_RX_DROP;
1870 * batadv_nc_mesh_free - clean up network coding memory
1871 * @bat_priv: the bat priv with all the soft interface information
1873 void batadv_nc_mesh_free(struct batadv_priv *bat_priv)
1875 batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_NC, 1);
1876 batadv_tvlv_handler_unregister(bat_priv, BATADV_TVLV_NC, 1);
1877 cancel_delayed_work_sync(&bat_priv->nc.work);
1879 batadv_nc_purge_paths(bat_priv, bat_priv->nc.coding_hash, NULL);
1880 batadv_hash_destroy(bat_priv->nc.coding_hash);
1881 batadv_nc_purge_paths(bat_priv, bat_priv->nc.decoding_hash, NULL);
1882 batadv_hash_destroy(bat_priv->nc.decoding_hash);
1886 * batadv_nc_nodes_seq_print_text - print the nc node information
1887 * @seq: seq file to print on
1888 * @offset: not used
1890 * Return: always 0
1892 int batadv_nc_nodes_seq_print_text(struct seq_file *seq, void *offset)
1894 struct net_device *net_dev = (struct net_device *)seq->private;
1895 struct batadv_priv *bat_priv = netdev_priv(net_dev);
1896 struct batadv_hashtable *hash = bat_priv->orig_hash;
1897 struct batadv_hard_iface *primary_if;
1898 struct hlist_head *head;
1899 struct batadv_orig_node *orig_node;
1900 struct batadv_nc_node *nc_node;
1901 int i;
1903 primary_if = batadv_seq_print_text_primary_if_get(seq);
1904 if (!primary_if)
1905 goto out;
1907 /* Traverse list of originators */
1908 for (i = 0; i < hash->size; i++) {
1909 head = &hash->table[i];
1911 /* For each orig_node in this bin */
1912 rcu_read_lock();
1913 hlist_for_each_entry_rcu(orig_node, head, hash_entry) {
1914 /* no need to print the orig node if it does not have
1915 * network coding neighbors
1917 if (list_empty(&orig_node->in_coding_list) &&
1918 list_empty(&orig_node->out_coding_list))
1919 continue;
1921 seq_printf(seq, "Node: %pM\n", orig_node->orig);
1923 seq_puts(seq, " Ingoing: ");
1924 /* For each in_nc_node to this orig_node */
1925 list_for_each_entry_rcu(nc_node,
1926 &orig_node->in_coding_list,
1927 list)
1928 seq_printf(seq, "%pM ",
1929 nc_node->addr);
1930 seq_puts(seq, "\n");
1932 seq_puts(seq, " Outgoing: ");
1933 /* For out_nc_node to this orig_node */
1934 list_for_each_entry_rcu(nc_node,
1935 &orig_node->out_coding_list,
1936 list)
1937 seq_printf(seq, "%pM ",
1938 nc_node->addr);
1939 seq_puts(seq, "\n\n");
1941 rcu_read_unlock();
1944 out:
1945 if (primary_if)
1946 batadv_hardif_put(primary_if);
1947 return 0;
1951 * batadv_nc_init_debugfs - create nc folder and related files in debugfs
1952 * @bat_priv: the bat priv with all the soft interface information
1954 * Return: 0 on success or negative error number in case of failure
1956 int batadv_nc_init_debugfs(struct batadv_priv *bat_priv)
1958 struct dentry *nc_dir, *file;
1960 nc_dir = debugfs_create_dir("nc", bat_priv->debug_dir);
1961 if (!nc_dir)
1962 goto out;
1964 file = debugfs_create_u8("min_tq", S_IRUGO | S_IWUSR, nc_dir,
1965 &bat_priv->nc.min_tq);
1966 if (!file)
1967 goto out;
1969 file = debugfs_create_u32("max_fwd_delay", S_IRUGO | S_IWUSR, nc_dir,
1970 &bat_priv->nc.max_fwd_delay);
1971 if (!file)
1972 goto out;
1974 file = debugfs_create_u32("max_buffer_time", S_IRUGO | S_IWUSR, nc_dir,
1975 &bat_priv->nc.max_buffer_time);
1976 if (!file)
1977 goto out;
1979 return 0;
1981 out:
1982 return -ENOMEM;