r8152: move enabling PHY
[cascardo/linux.git] / net / batman-adv / network-coding.c
1 /* Copyright (C) 2012-2016  B.A.T.M.A.N. contributors:
2  *
3  * Martin Hundebøll, Jeppe Ledet-Pedersen
4  *
5  * This program is free software; you can redistribute it and/or
6  * modify it under the terms of version 2 of the GNU General Public
7  * License as published by the Free Software Foundation.
8  *
9  * This program is distributed in the hope that it will be useful, but
10  * WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12  * General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, see <http://www.gnu.org/licenses/>.
16  */
17
18 #include "network-coding.h"
19 #include "main.h"
20
21 #include <linux/atomic.h>
22 #include <linux/bitops.h>
23 #include <linux/byteorder/generic.h>
24 #include <linux/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>
51
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"
60
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;
63
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);
67
68 /**
69  * batadv_nc_init - one-time initialization for network coding
70  *
71  * Return: 0 on success or negative error number in case of failure
72  */
73 int __init batadv_nc_init(void)
74 {
75         int ret;
76
77         /* Register our packet type */
78         ret = batadv_recv_handler_register(BATADV_CODED,
79                                            batadv_nc_recv_coded_packet);
80
81         return ret;
82 }
83
84 /**
85  * batadv_nc_start_timer - initialise the nc periodic worker
86  * @bat_priv: the bat priv with all the soft interface information
87  */
88 static void batadv_nc_start_timer(struct batadv_priv *bat_priv)
89 {
90         queue_delayed_work(batadv_event_workqueue, &bat_priv->nc.work,
91                            msecs_to_jiffies(10));
92 }
93
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
98  */
99 static void batadv_nc_tvlv_container_update(struct batadv_priv *bat_priv)
100 {
101         char nc_mode;
102
103         nc_mode = atomic_read(&bat_priv->network_coding);
104
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;
113         }
114 }
115
116 /**
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
120  */
121 void batadv_nc_status_update(struct net_device *net_dev)
122 {
123         struct batadv_priv *bat_priv = netdev_priv(net_dev);
124
125         batadv_nc_tvlv_container_update(bat_priv);
126 }
127
128 /**
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
135  */
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)
140 {
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);
145 }
146
147 /**
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
150  *
151  * Return: 0 on success or negative error number in case of failure
152  */
153 int batadv_nc_mesh_init(struct batadv_priv *bat_priv)
154 {
155         bat_priv->nc.timestamp_fwd_flush = jiffies;
156         bat_priv->nc.timestamp_sniffed_purge = jiffies;
157
158         if (bat_priv->nc.coding_hash || bat_priv->nc.decoding_hash)
159                 return 0;
160
161         bat_priv->nc.coding_hash = batadv_hash_new(128);
162         if (!bat_priv->nc.coding_hash)
163                 goto err;
164
165         batadv_hash_set_lock_class(bat_priv->nc.coding_hash,
166                                    &batadv_nc_coding_hash_lock_class_key);
167
168         bat_priv->nc.decoding_hash = batadv_hash_new(128);
169         if (!bat_priv->nc.decoding_hash)
170                 goto err;
171
172         batadv_hash_set_lock_class(bat_priv->nc.decoding_hash,
173                                    &batadv_nc_decoding_hash_lock_class_key);
174
175         INIT_DELAYED_WORK(&bat_priv->nc.work, batadv_nc_worker);
176         batadv_nc_start_timer(bat_priv);
177
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;
183
184 err:
185         return -ENOMEM;
186 }
187
188 /**
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
191  */
192 void batadv_nc_init_bat_priv(struct batadv_priv *bat_priv)
193 {
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;
198 }
199
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
203  */
204 void batadv_nc_init_orig(struct batadv_orig_node *orig_node)
205 {
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);
210 }
211
212 /**
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
216  */
217 static void batadv_nc_node_release(struct kref *ref)
218 {
219         struct batadv_nc_node *nc_node;
220
221         nc_node = container_of(ref, struct batadv_nc_node, refcount);
222
223         batadv_orig_node_put(nc_node->orig_node);
224         kfree_rcu(nc_node, rcu);
225 }
226
227 /**
228  * batadv_nc_node_put - decrement the nc_node refcounter and possibly
229  *  release it
230  * @nc_node: nc_node to be free'd
231  */
232 static void batadv_nc_node_put(struct batadv_nc_node *nc_node)
233 {
234         kref_put(&nc_node->refcount, batadv_nc_node_release);
235 }
236
237 /**
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
241  */
242 static void batadv_nc_path_release(struct kref *ref)
243 {
244         struct batadv_nc_path *nc_path;
245
246         nc_path = container_of(ref, struct batadv_nc_path, refcount);
247
248         kfree_rcu(nc_path, rcu);
249 }
250
251 /**
252  * batadv_nc_path_put - decrement the nc_path refcounter and possibly
253  *  release it
254  * @nc_path: nc_path to be free'd
255  */
256 static void batadv_nc_path_put(struct batadv_nc_path *nc_path)
257 {
258         kref_put(&nc_path->refcount, batadv_nc_path_release);
259 }
260
261 /**
262  * batadv_nc_packet_free - frees nc packet
263  * @nc_packet: the nc packet to free
264  */
265 static void batadv_nc_packet_free(struct batadv_nc_packet *nc_packet)
266 {
267         kfree_skb(nc_packet->skb);
268         batadv_nc_path_put(nc_packet->nc_path);
269         kfree(nc_packet);
270 }
271
272 /**
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
276  *
277  * Return: true if the entry has to be purged now, false otherwise
278  */
279 static bool batadv_nc_to_purge_nc_node(struct batadv_priv *bat_priv,
280                                        struct batadv_nc_node *nc_node)
281 {
282         if (atomic_read(&bat_priv->mesh_state) != BATADV_MESH_ACTIVE)
283                 return true;
284
285         return batadv_has_timed_out(nc_node->last_seen, BATADV_NC_NODE_TIMEOUT);
286 }
287
288 /**
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
292  *
293  * Return: true if the entry has to be purged now, false otherwise
294  */
295 static bool batadv_nc_to_purge_nc_path_coding(struct batadv_priv *bat_priv,
296                                               struct batadv_nc_path *nc_path)
297 {
298         if (atomic_read(&bat_priv->mesh_state) != BATADV_MESH_ACTIVE)
299                 return true;
300
301         /* purge the path when no packets has been added for 10 times the
302          * max_fwd_delay time
303          */
304         return batadv_has_timed_out(nc_path->last_valid,
305                                     bat_priv->nc.max_fwd_delay * 10);
306 }
307
308 /**
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
312  *
313  * Return: true if the entry has to be purged now, false otherwise
314  */
315 static bool batadv_nc_to_purge_nc_path_decoding(struct batadv_priv *bat_priv,
316                                                 struct batadv_nc_path *nc_path)
317 {
318         if (atomic_read(&bat_priv->mesh_state) != BATADV_MESH_ACTIVE)
319                 return true;
320
321         /* purge the path when no packets has been added for 10 times the
322          * max_buffer time
323          */
324         return batadv_has_timed_out(nc_path->last_valid,
325                                     bat_priv->nc.max_buffer_time * 10);
326 }
327
328 /**
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
338  */
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 *))
345 {
346         struct batadv_nc_node *nc_node, *nc_node_tmp;
347
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
353                  */
354                 if (to_purge && !to_purge(bat_priv, nc_node))
355                         continue;
356
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);
362         }
363         spin_unlock_bh(lock);
364 }
365
366 /**
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
375  */
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 *))
380 {
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);
385
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);
390 }
391
392 /**
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
396  */
397 static void batadv_nc_purge_orig_hash(struct batadv_priv *bat_priv)
398 {
399         struct batadv_hashtable *hash = bat_priv->orig_hash;
400         struct hlist_head *head;
401         struct batadv_orig_node *orig_node;
402         u32 i;
403
404         if (!hash)
405                 return;
406
407         /* For each orig_node */
408         for (i = 0; i < hash->size; i++) {
409                 head = &hash->table[i];
410
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();
416         }
417 }
418
419 /**
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
428  */
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 *))
433 {
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;
439
440         for (i = 0; i < hash->size; i++) {
441                 head = &hash->table[i];
442                 lock = &hash->list_locks[i];
443
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
449                          */
450                         if (to_purge && !to_purge(bat_priv, nc_path))
451                                 continue;
452
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.
457                          */
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;
465                         }
466
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);
473                 }
474                 spin_unlock_bh(lock);
475         }
476 }
477
478 /**
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
483  */
484 static void batadv_nc_hash_key_gen(struct batadv_nc_path *key, const char *src,
485                                    const char *dst)
486 {
487         memcpy(key->prev_hop, src, sizeof(key->prev_hop));
488         memcpy(key->next_hop, dst, sizeof(key->next_hop));
489 }
490
491 /**
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
495  *
496  * Return: the selected index in the hash table for the given data.
497  */
498 static u32 batadv_nc_hash_choose(const void *data, u32 size)
499 {
500         const struct batadv_nc_path *nc_path = data;
501         u32 hash = 0;
502
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);
505
506         return hash % size;
507 }
508
509 /**
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
514  *
515  * Return: true if the two entry are the same, false otherwise
516  */
517 static bool batadv_nc_hash_compare(const struct hlist_node *node,
518                                    const void *data2)
519 {
520         const struct batadv_nc_path *nc_path1, *nc_path2;
521
522         nc_path1 = container_of(node, struct batadv_nc_path, hash_entry);
523         nc_path2 = data2;
524
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;
528
529         if (!batadv_compare_eth(nc_path1->next_hop, nc_path2->next_hop))
530                 return false;
531
532         return true;
533 }
534
535 /**
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
539  *
540  * Return: the nc_path if found, NULL otherwise.
541  */
542 static struct batadv_nc_path *
543 batadv_nc_hash_find(struct batadv_hashtable *hash,
544                     void *data)
545 {
546         struct hlist_head *head;
547         struct batadv_nc_path *nc_path, *nc_path_tmp = NULL;
548         int index;
549
550         if (!hash)
551                 return NULL;
552
553         index = batadv_nc_hash_choose(data, hash->size);
554         head = &hash->table[index];
555
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;
560
561                 if (!kref_get_unless_zero(&nc_path->refcount))
562                         continue;
563
564                 nc_path_tmp = nc_path;
565                 break;
566         }
567         rcu_read_unlock();
568
569         return nc_path_tmp;
570 }
571
572 /**
573  * batadv_nc_send_packet - send non-coded packet and free nc_packet struct
574  * @nc_packet: the nc packet to send
575  */
576 static void batadv_nc_send_packet(struct batadv_nc_packet *nc_packet)
577 {
578         batadv_send_unicast_skb(nc_packet->skb, nc_packet->neigh_node);
579         nc_packet->skb = NULL;
580         batadv_nc_packet_free(nc_packet);
581 }
582
583 /**
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
588  *
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.
592  *
593  * Return: false as soon as the entry in the fifo queue has not been timed out
594  * yet and true otherwise.
595  */
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)
599 {
600         unsigned long timeout = bat_priv->nc.max_buffer_time;
601         bool res = false;
602
603         lockdep_assert_held(&nc_path->packet_list_lock);
604
605         /* Packets are added to tail, so the remaining packets did not time
606          * out and we can stop processing the current queue
607          */
608         if (atomic_read(&bat_priv->mesh_state) == BATADV_MESH_ACTIVE &&
609             !batadv_has_timed_out(nc_packet->timestamp, timeout))
610                 goto out;
611
612         /* purge nc packet */
613         list_del(&nc_packet->list);
614         batadv_nc_packet_free(nc_packet);
615
616         res = true;
617
618 out:
619         return res;
620 }
621
622 /**
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
627  *
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.
631  *
632  * Return: false as soon as the entry in the fifo queue has not been timed out
633  * yet and true otherwise.
634  */
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)
638 {
639         unsigned long timeout = bat_priv->nc.max_fwd_delay;
640
641         lockdep_assert_held(&nc_path->packet_list_lock);
642
643         /* Packets are added to tail, so the remaining packets did not time
644          * out and we can stop processing the current queue
645          */
646         if (atomic_read(&bat_priv->mesh_state) == BATADV_MESH_ACTIVE &&
647             !batadv_has_timed_out(nc_packet->timestamp, timeout))
648                 return false;
649
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);
656
657         return true;
658 }
659
660 /**
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.
668  */
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 *))
675 {
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;
681
682         if (!hash)
683                 return;
684
685         /* Loop hash table bins */
686         for (i = 0; i < hash->size; i++) {
687                 head = &hash->table[i];
688
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;
699                         }
700                         spin_unlock_bh(&nc_path->packet_list_lock);
701                 }
702                 rcu_read_unlock();
703         }
704 }
705
706 /**
707  * batadv_nc_worker - periodic task for house keeping related to network coding
708  * @work: kernel work struct
709  */
710 static void batadv_nc_worker(struct work_struct *work)
711 {
712         struct delayed_work *delayed_work;
713         struct batadv_priv_nc *priv_nc;
714         struct batadv_priv *bat_priv;
715         unsigned long timeout;
716
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);
720
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);
726
727         timeout = bat_priv->nc.max_fwd_delay;
728
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;
733         }
734
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;
740         }
741
742         /* Schedule a new check */
743         batadv_nc_start_timer(bat_priv);
744 }
745
746 /**
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
752  *
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.
758  */
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)
762 {
763         struct batadv_orig_ifinfo *orig_ifinfo;
764         u32 last_real_seqno;
765         u8 last_ttl;
766
767         orig_ifinfo = batadv_orig_ifinfo_get(orig_node, BATADV_IF_DEFAULT);
768         if (!orig_ifinfo)
769                 return false;
770
771         last_ttl = orig_ifinfo->last_ttl;
772         last_real_seqno = orig_ifinfo->last_real_seqno;
773         batadv_orig_ifinfo_put(orig_ifinfo);
774
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;
783
784         return true;
785 }
786
787 /**
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
793  *
794  * Return: the nc_node if found, NULL otherwise.
795  */
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)
800 {
801         struct batadv_nc_node *nc_node, *nc_node_out = NULL;
802         struct list_head *list;
803
804         if (in_coding)
805                 list = &orig_neigh_node->in_coding_list;
806         else
807                 list = &orig_neigh_node->out_coding_list;
808
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;
814
815                 if (!kref_get_unless_zero(&nc_node->refcount))
816                         continue;
817
818                 /* Found a match */
819                 nc_node_out = nc_node;
820                 break;
821         }
822         rcu_read_unlock();
823
824         return nc_node_out;
825 }
826
827 /**
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
835  *
836  * Return: the nc_node if found or created, NULL in case of an error.
837  */
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)
843 {
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;
847
848         /* Check if nc_node is already added */
849         nc_node = batadv_nc_find_nc_node(orig_node, orig_neigh_node, in_coding);
850
851         /* Node found */
852         if (nc_node)
853                 return nc_node;
854
855         nc_node = kzalloc(sizeof(*nc_node), GFP_ATOMIC);
856         if (!nc_node)
857                 return NULL;
858
859         kref_get(&orig_neigh_node->refcount);
860
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);
867
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;
875         }
876
877         batadv_dbg(BATADV_DBG_NC, bat_priv, "Adding nc_node %pM -> %pM\n",
878                    nc_node->addr, nc_node->orig_node->orig);
879
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);
884
885         return nc_node;
886 }
887
888 /**
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
897  */
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)
903 {
904         struct batadv_nc_node *in_nc_node = NULL;
905         struct batadv_nc_node *out_nc_node = NULL;
906
907         /* Check if network coding is enabled */
908         if (!atomic_read(&bat_priv->network_coding))
909                 goto out;
910
911         /* check if orig node is network coding enabled */
912         if (!test_bit(BATADV_ORIG_CAPA_HAS_NC, &orig_node->capabilities))
913                 goto out;
914
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;
919
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;
925
926         in_nc_node->last_seen = jiffies;
927
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;
933
934         out_nc_node->last_seen = jiffies;
935
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);
941 }
942
943 /**
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
949  *
950  * Return: pointer to nc_path if the path was found or created, returns NULL
951  * on error.
952  */
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)
957 {
958         int hash_added;
959         struct batadv_nc_path *nc_path, nc_path_key;
960
961         batadv_nc_hash_key_gen(&nc_path_key, src, dst);
962
963         /* Search for existing nc_path */
964         nc_path = batadv_nc_hash_find(hash, (void *)&nc_path_key);
965
966         if (nc_path) {
967                 /* Set timestamp to delay removal of nc_path */
968                 nc_path->last_valid = jiffies;
969                 return nc_path;
970         }
971
972         /* No existing nc_path was found; create a new */
973         nc_path = kzalloc(sizeof(*nc_path), GFP_ATOMIC);
974
975         if (!nc_path)
976                 return NULL;
977
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);
986
987         batadv_dbg(BATADV_DBG_NC, bat_priv, "Adding nc_path %pM -> %pM\n",
988                    nc_path->prev_hop,
989                    nc_path->next_hop);
990
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);
995
996         if (hash_added < 0) {
997                 kfree(nc_path);
998                 return NULL;
999         }
1000
1001         return nc_path;
1002 }
1003
1004 /**
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
1008  *
1009  * Return: scaled tq value
1010  */
1011 static u8 batadv_nc_random_weight_tq(u8 tq)
1012 {
1013         u8 rand_val, rand_tq;
1014
1015         get_random_bytes(&rand_val, sizeof(rand_val));
1016
1017         /* randomize the estimated packet loss (max TQ - estimated TQ) */
1018         rand_tq = rand_val * (BATADV_TQ_MAX_VALUE - tq);
1019
1020         /* normalize the randomized packet loss */
1021         rand_tq /= BATADV_TQ_MAX_VALUE;
1022
1023         /* convert to (randomized) estimated tq again */
1024         return BATADV_TQ_MAX_VALUE - rand_tq;
1025 }
1026
1027 /**
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
1032  */
1033 static void batadv_nc_memxor(char *dst, const char *src, unsigned int len)
1034 {
1035         unsigned int i;
1036
1037         for (i = 0; i < len; ++i)
1038                 dst[i] ^= src[i];
1039 }
1040
1041 /**
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
1049  *
1050  * Return: true if both packets are consumed, false otherwise.
1051  */
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)
1057 {
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;
1075
1076         /* TODO: do we need to consider the outgoing interface for
1077          * coded packets?
1078          */
1079         router_neigh = batadv_orig_router_get(neigh_node->orig_node,
1080                                               BATADV_IF_DEFAULT);
1081         if (!router_neigh)
1082                 goto out;
1083
1084         router_neigh_ifinfo = batadv_neigh_ifinfo_get(router_neigh,
1085                                                       BATADV_IF_DEFAULT);
1086         if (!router_neigh_ifinfo)
1087                 goto out;
1088
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;
1094
1095         router_coding_ifinfo = batadv_neigh_ifinfo_get(router_coding,
1096                                                        BATADV_IF_DEFAULT);
1097         if (!router_coding_ifinfo)
1098                 goto out;
1099
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);
1104
1105         /* Select one destination for the MAC-header dst-field based on
1106          * weighted TQ-values.
1107          */
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;
1130         }
1131
1132         /* Instead of zero padding the smallest data buffer, we
1133          * code into the largest.
1134          */
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;
1141         }
1142
1143         /* coding_len is used when decoding the packet shorter packet */
1144         coding_len = skb_src->len - unicast_size;
1145
1146         if (skb_linearize(skb_dest) < 0 || skb_linearize(skb_src) < 0)
1147                 goto out;
1148
1149         skb_push(skb_dest, header_add);
1150
1151         coded_packet = (struct batadv_coded_packet *)skb_dest->data;
1152         skb_reset_mac_header(skb_dest);
1153
1154         coded_packet->packet_type = BATADV_CODED;
1155         coded_packet->version = BATADV_COMPAT_VERSION;
1156         coded_packet->ttl = packet1->ttl;
1157
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;
1163
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);
1172
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);
1176
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);
1210         }
1211
1212         /* skb_src is now coded into skb_dest, so free it */
1213         kfree_skb(skb_src);
1214
1215         /* avoid duplicate free of skb from nc_packet */
1216         nc_packet->skb = NULL;
1217         batadv_nc_packet_free(nc_packet);
1218
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;
1232 }
1233
1234 /**
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
1239  *
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.
1248  *
1249  * Return: true if coding of a decoded packet is allowed.
1250  */
1251 static bool batadv_nc_skb_coding_possible(struct sk_buff *skb, u8 *dst, u8 *src)
1252 {
1253         if (BATADV_SKB_CB(skb)->decoded && !batadv_compare_eth(dst, src))
1254                 return false;
1255         return true;
1256 }
1257
1258 /**
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
1266  *
1267  * Return: true if coding of a decoded skb is allowed.
1268  */
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)
1275 {
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;
1281
1282         if (!hash)
1283                 return NULL;
1284
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);
1289
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;
1295
1296                 if (!batadv_compare_eth(nc_path->next_hop, out_nc_node->addr))
1297                         continue;
1298
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;
1303                 }
1304
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;
1311
1312                         /* Coding opportunity is found! */
1313                         list_del(&nc_packet->list);
1314                         nc_packet_out = nc_packet;
1315                         break;
1316                 }
1317
1318                 spin_unlock_bh(&nc_path->packet_list_lock);
1319                 break;
1320         }
1321         rcu_read_unlock();
1322
1323         return nc_packet_out;
1324 }
1325
1326 /**
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
1334  *
1335  * Return: an nc packet if a suitable coding packet was found, NULL otherwise.
1336  */
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)
1343 {
1344         struct batadv_orig_node *orig_node;
1345         struct batadv_nc_node *out_nc_node;
1346         struct batadv_nc_packet *nc_packet = NULL;
1347
1348         orig_node = batadv_orig_hash_find(bat_priv, eth_src);
1349         if (!orig_node)
1350                 return NULL;
1351
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;
1359
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;
1365         }
1366         rcu_read_unlock();
1367
1368         batadv_orig_node_put(orig_node);
1369         return nc_packet;
1370 }
1371
1372 /**
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
1378  */
1379 static void batadv_nc_skb_store_before_coding(struct batadv_priv *bat_priv,
1380                                               struct sk_buff *skb,
1381                                               u8 *eth_dst_new)
1382 {
1383         struct ethhdr *ethhdr;
1384
1385         /* Copy skb header to change the mac header */
1386         skb = pskb_copy_for_clone(skb, GFP_ATOMIC);
1387         if (!skb)
1388                 return;
1389
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);
1394
1395         /* Set data pointer to MAC header to mimic packets from our tx path */
1396         skb_push(skb, ETH_HLEN);
1397
1398         /* Add the packet to the decoding packet pool */
1399         batadv_nc_skb_store_for_decoding(bat_priv, skb);
1400
1401         /* batadv_nc_skb_store_for_decoding() clones the skb, so we must free
1402          * our ref
1403          */
1404         kfree_skb(skb);
1405 }
1406
1407 /**
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
1412  *
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.
1417  *
1418  * Return: true if the skb was consumed (encoded packet sent) or false otherwise
1419  */
1420 static bool batadv_nc_skb_dst_search(struct sk_buff *skb,
1421                                      struct batadv_neigh_node *neigh_node,
1422                                      struct ethhdr *ethhdr)
1423 {
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;
1429
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);
1436
1437                 /* Opportunity was found, so stop searching */
1438                 if (nc_packet)
1439                         break;
1440         }
1441         rcu_read_unlock();
1442
1443         if (!nc_packet)
1444                 return false;
1445
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);
1451
1452         /* Code and send packets */
1453         if (batadv_nc_code_packets(bat_priv, skb, ethhdr, nc_packet,
1454                                    neigh_node))
1455                 return true;
1456
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.
1460          */
1461         batadv_nc_send_packet(nc_packet);
1462         return false;
1463 }
1464
1465 /**
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
1471  *
1472  * Return: true if the packet was buffered or false in case of an error.
1473  */
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)
1478 {
1479         struct batadv_nc_packet *nc_packet;
1480
1481         nc_packet = kzalloc(sizeof(*nc_packet), GFP_ATOMIC);
1482         if (!nc_packet)
1483                 return false;
1484
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;
1491
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);
1496
1497         return true;
1498 }
1499
1500 /**
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
1505  *
1506  * Return: true if the skb was consumed (encoded packet sent) or false otherwise
1507  */
1508 bool batadv_nc_skb_forward(struct sk_buff *skb,
1509                            struct batadv_neigh_node *neigh_node)
1510 {
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;
1518
1519         /* Check if network coding is enabled */
1520         if (!atomic_read(&bat_priv->network_coding))
1521                 goto out;
1522
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;
1528
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;
1532
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);
1538
1539         if (!nc_path)
1540                 goto out;
1541
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;
1546
1547         /* Packet is consumed */
1548         return true;
1549
1550 free_nc_path:
1551         batadv_nc_path_put(nc_path);
1552 out:
1553         /* Packet is not consumed */
1554         return false;
1555 }
1556
1557 /**
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
1562  */
1563 void batadv_nc_skb_store_for_decoding(struct batadv_priv *bat_priv,
1564                                       struct sk_buff *skb)
1565 {
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;
1571
1572         /* Check if network coding is enabled */
1573         if (!atomic_read(&bat_priv->network_coding))
1574                 goto out;
1575
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;
1581
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);
1587
1588         if (!nc_path)
1589                 goto out;
1590
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;
1595
1596         if (unlikely(!pskb_may_pull(skb, ETH_HLEN)))
1597                 goto free_skb;
1598
1599         if (unlikely(!skb_pull_rcsum(skb, ETH_HLEN)))
1600                 goto free_skb;
1601
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;
1606
1607         batadv_inc_counter(bat_priv, BATADV_CNT_NC_BUFFER);
1608         return;
1609
1610 free_skb:
1611         kfree_skb(skb);
1612 free_nc_path:
1613         batadv_nc_path_put(nc_path);
1614 out:
1615         return;
1616 }
1617
1618 /**
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
1623  */
1624 void batadv_nc_skb_store_sniffed_unicast(struct batadv_priv *bat_priv,
1625                                          struct sk_buff *skb)
1626 {
1627         struct ethhdr *ethhdr = eth_hdr(skb);
1628
1629         if (batadv_is_my_mac(bat_priv, ethhdr->h_dest))
1630                 return;
1631
1632         /* Set data pointer to MAC header to mimic packets from our tx path */
1633         skb_push(skb, ETH_HLEN);
1634
1635         batadv_nc_skb_store_for_decoding(bat_priv, skb);
1636 }
1637
1638 /**
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
1644  *
1645  * Return: pointer to decoded unicast packet if the packet was decoded or NULL
1646  * in case of an error.
1647  */
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)
1651 {
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;
1660
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));
1664
1665         if (skb_cow(skb, 0) < 0)
1666                 return NULL;
1667
1668         if (unlikely(!skb_pull_rcsum(skb, h_diff)))
1669                 return NULL;
1670
1671         /* Data points to batman header, so set mac header 14 bytes before
1672          * and network to data
1673          */
1674         skb_set_mac_header(skb, -ETH_HLEN);
1675         skb_reset_network_header(skb);
1676
1677         /* Reconstruct original mac header */
1678         ethhdr = eth_hdr(skb);
1679         *ethhdr = ethhdr_tmp;
1680
1681         /* Select the correct unicast header information based on the location
1682          * of our mac address in the coded_packet header
1683          */
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
1688                  */
1689                 ether_addr_copy(ethhdr->h_dest, coded_packet_tmp.second_dest);
1690                 skb->pkt_type = PACKET_HOST;
1691
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;
1699         }
1700
1701         coding_len = ntohs(coded_packet_tmp.coded_len);
1702
1703         if (coding_len > skb->len)
1704                 return NULL;
1705
1706         /* Here the magic is reversed:
1707          *   extract the missing packet from the received coded packet
1708          */
1709         batadv_nc_memxor(skb->data + h_size,
1710                          nc_packet->skb->data + h_size,
1711                          coding_len);
1712
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;
1718         }
1719
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;
1727
1728         batadv_nc_packet_free(nc_packet);
1729         return unicast_packet;
1730 }
1731
1732 /**
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
1738  *
1739  * Return: pointer to nc packet if the needed data was found or NULL otherwise.
1740  */
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)
1745 {
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;
1752
1753         if (!hash)
1754                 return NULL;
1755
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;
1764         }
1765
1766         batadv_nc_hash_key_gen(&nc_path_key, source, dest);
1767         index = batadv_nc_hash_choose(&nc_path_key, hash->size);
1768
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);
1778
1779                                 nc_packet = tmp_nc_packet;
1780                                 break;
1781                         }
1782                 }
1783                 spin_unlock_bh(&nc_path->packet_list_lock);
1784
1785                 if (nc_packet)
1786                         break;
1787         }
1788         rcu_read_unlock();
1789
1790         if (!nc_packet)
1791                 batadv_dbg(BATADV_DBG_NC, bat_priv,
1792                            "No decoding packet found for %u\n", packet_id);
1793
1794         return nc_packet;
1795 }
1796
1797 /**
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
1802  *
1803  * Return: NET_RX_SUCCESS if the packet has been consumed or NET_RX_DROP
1804  * otherwise.
1805  */
1806 static int batadv_nc_recv_coded_packet(struct sk_buff *skb,
1807                                        struct batadv_hard_iface *recv_if)
1808 {
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);
1815
1816         /* Check if network coding is enabled */
1817         if (!atomic_read(&bat_priv->network_coding))
1818                 return NET_RX_DROP;
1819
1820         /* Make sure we can access (and remove) header */
1821         if (unlikely(!pskb_may_pull(skb, hdr_size)))
1822                 return NET_RX_DROP;
1823
1824         coded_packet = (struct batadv_coded_packet *)skb->data;
1825         ethhdr = eth_hdr(skb);
1826
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;
1831
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);
1835
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;
1841         }
1842
1843         /* Make skb's linear, because decoding accesses the entire buffer */
1844         if (skb_linearize(skb) < 0)
1845                 goto free_nc_packet;
1846
1847         if (skb_linearize(nc_packet->skb) < 0)
1848                 goto free_nc_packet;
1849
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;
1855         }
1856
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);
1863
1864 free_nc_packet:
1865         batadv_nc_packet_free(nc_packet);
1866         return NET_RX_DROP;
1867 }
1868
1869 /**
1870  * batadv_nc_mesh_free - clean up network coding memory
1871  * @bat_priv: the bat priv with all the soft interface information
1872  */
1873 void batadv_nc_mesh_free(struct batadv_priv *bat_priv)
1874 {
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);
1878
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);
1883 }
1884
1885 /**
1886  * batadv_nc_nodes_seq_print_text - print the nc node information
1887  * @seq: seq file to print on
1888  * @offset: not used
1889  *
1890  * Return: always 0
1891  */
1892 int batadv_nc_nodes_seq_print_text(struct seq_file *seq, void *offset)
1893 {
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;
1902
1903         primary_if = batadv_seq_print_text_primary_if_get(seq);
1904         if (!primary_if)
1905                 goto out;
1906
1907         /* Traverse list of originators */
1908         for (i = 0; i < hash->size; i++) {
1909                 head = &hash->table[i];
1910
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
1916                          */
1917                         if (list_empty(&orig_node->in_coding_list) &&
1918                             list_empty(&orig_node->out_coding_list))
1919                                 continue;
1920
1921                         seq_printf(seq, "Node:      %pM\n", orig_node->orig);
1922
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");
1931
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");
1940                 }
1941                 rcu_read_unlock();
1942         }
1943
1944 out:
1945         if (primary_if)
1946                 batadv_hardif_put(primary_if);
1947         return 0;
1948 }
1949
1950 /**
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
1953  *
1954  * Return: 0 on success or negative error number in case of failure
1955  */
1956 int batadv_nc_init_debugfs(struct batadv_priv *bat_priv)
1957 {
1958         struct dentry *nc_dir, *file;
1959
1960         nc_dir = debugfs_create_dir("nc", bat_priv->debug_dir);
1961         if (!nc_dir)
1962                 goto out;
1963
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;
1968
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;
1973
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;
1978
1979         return 0;
1980
1981 out:
1982         return -ENOMEM;
1983 }