0001
0002
0003
0004
0005
0006
0007 #include "network-coding.h"
0008 #include "main.h"
0009
0010 #include <linux/atomic.h>
0011 #include <linux/bitops.h>
0012 #include <linux/byteorder/generic.h>
0013 #include <linux/compiler.h>
0014 #include <linux/container_of.h>
0015 #include <linux/errno.h>
0016 #include <linux/etherdevice.h>
0017 #include <linux/gfp.h>
0018 #include <linux/if_ether.h>
0019 #include <linux/if_packet.h>
0020 #include <linux/init.h>
0021 #include <linux/jhash.h>
0022 #include <linux/jiffies.h>
0023 #include <linux/kref.h>
0024 #include <linux/list.h>
0025 #include <linux/lockdep.h>
0026 #include <linux/net.h>
0027 #include <linux/netdevice.h>
0028 #include <linux/prandom.h>
0029 #include <linux/printk.h>
0030 #include <linux/rculist.h>
0031 #include <linux/rcupdate.h>
0032 #include <linux/skbuff.h>
0033 #include <linux/slab.h>
0034 #include <linux/spinlock.h>
0035 #include <linux/stddef.h>
0036 #include <linux/string.h>
0037 #include <linux/workqueue.h>
0038 #include <uapi/linux/batadv_packet.h>
0039
0040 #include "hash.h"
0041 #include "log.h"
0042 #include "originator.h"
0043 #include "routing.h"
0044 #include "send.h"
0045 #include "tvlv.h"
0046
0047 static struct lock_class_key batadv_nc_coding_hash_lock_class_key;
0048 static struct lock_class_key batadv_nc_decoding_hash_lock_class_key;
0049
0050 static void batadv_nc_worker(struct work_struct *work);
0051 static int batadv_nc_recv_coded_packet(struct sk_buff *skb,
0052 struct batadv_hard_iface *recv_if);
0053
0054
0055
0056
0057
0058
0059 int __init batadv_nc_init(void)
0060 {
0061
0062 return batadv_recv_handler_register(BATADV_CODED,
0063 batadv_nc_recv_coded_packet);
0064 }
0065
0066
0067
0068
0069
0070 static void batadv_nc_start_timer(struct batadv_priv *bat_priv)
0071 {
0072 queue_delayed_work(batadv_event_workqueue, &bat_priv->nc.work,
0073 msecs_to_jiffies(10));
0074 }
0075
0076
0077
0078
0079
0080
0081 static void batadv_nc_tvlv_container_update(struct batadv_priv *bat_priv)
0082 {
0083 char nc_mode;
0084
0085 nc_mode = atomic_read(&bat_priv->network_coding);
0086
0087 switch (nc_mode) {
0088 case 0:
0089 batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_NC, 1);
0090 break;
0091 case 1:
0092 batadv_tvlv_container_register(bat_priv, BATADV_TVLV_NC, 1,
0093 NULL, 0);
0094 break;
0095 }
0096 }
0097
0098
0099
0100
0101
0102
0103 void batadv_nc_status_update(struct net_device *net_dev)
0104 {
0105 struct batadv_priv *bat_priv = netdev_priv(net_dev);
0106
0107 batadv_nc_tvlv_container_update(bat_priv);
0108 }
0109
0110
0111
0112
0113
0114
0115
0116
0117
0118 static void batadv_nc_tvlv_ogm_handler_v1(struct batadv_priv *bat_priv,
0119 struct batadv_orig_node *orig,
0120 u8 flags,
0121 void *tvlv_value, u16 tvlv_value_len)
0122 {
0123 if (flags & BATADV_TVLV_HANDLER_OGM_CIFNOTFND)
0124 clear_bit(BATADV_ORIG_CAPA_HAS_NC, &orig->capabilities);
0125 else
0126 set_bit(BATADV_ORIG_CAPA_HAS_NC, &orig->capabilities);
0127 }
0128
0129
0130
0131
0132
0133
0134
0135 int batadv_nc_mesh_init(struct batadv_priv *bat_priv)
0136 {
0137 bat_priv->nc.timestamp_fwd_flush = jiffies;
0138 bat_priv->nc.timestamp_sniffed_purge = jiffies;
0139
0140 if (bat_priv->nc.coding_hash || bat_priv->nc.decoding_hash)
0141 return 0;
0142
0143 bat_priv->nc.coding_hash = batadv_hash_new(128);
0144 if (!bat_priv->nc.coding_hash)
0145 goto err;
0146
0147 batadv_hash_set_lock_class(bat_priv->nc.coding_hash,
0148 &batadv_nc_coding_hash_lock_class_key);
0149
0150 bat_priv->nc.decoding_hash = batadv_hash_new(128);
0151 if (!bat_priv->nc.decoding_hash) {
0152 batadv_hash_destroy(bat_priv->nc.coding_hash);
0153 goto err;
0154 }
0155
0156 batadv_hash_set_lock_class(bat_priv->nc.decoding_hash,
0157 &batadv_nc_decoding_hash_lock_class_key);
0158
0159 INIT_DELAYED_WORK(&bat_priv->nc.work, batadv_nc_worker);
0160 batadv_nc_start_timer(bat_priv);
0161
0162 batadv_tvlv_handler_register(bat_priv, batadv_nc_tvlv_ogm_handler_v1,
0163 NULL, BATADV_TVLV_NC, 1,
0164 BATADV_TVLV_HANDLER_OGM_CIFNOTFND);
0165 batadv_nc_tvlv_container_update(bat_priv);
0166 return 0;
0167
0168 err:
0169 return -ENOMEM;
0170 }
0171
0172
0173
0174
0175
0176 void batadv_nc_init_bat_priv(struct batadv_priv *bat_priv)
0177 {
0178 atomic_set(&bat_priv->network_coding, 0);
0179 bat_priv->nc.min_tq = 200;
0180 bat_priv->nc.max_fwd_delay = 10;
0181 bat_priv->nc.max_buffer_time = 200;
0182 }
0183
0184
0185
0186
0187
0188 void batadv_nc_init_orig(struct batadv_orig_node *orig_node)
0189 {
0190 INIT_LIST_HEAD(&orig_node->in_coding_list);
0191 INIT_LIST_HEAD(&orig_node->out_coding_list);
0192 spin_lock_init(&orig_node->in_coding_list_lock);
0193 spin_lock_init(&orig_node->out_coding_list_lock);
0194 }
0195
0196
0197
0198
0199
0200
0201 static void batadv_nc_node_release(struct kref *ref)
0202 {
0203 struct batadv_nc_node *nc_node;
0204
0205 nc_node = container_of(ref, struct batadv_nc_node, refcount);
0206
0207 batadv_orig_node_put(nc_node->orig_node);
0208 kfree_rcu(nc_node, rcu);
0209 }
0210
0211
0212
0213
0214
0215
0216 static void batadv_nc_node_put(struct batadv_nc_node *nc_node)
0217 {
0218 if (!nc_node)
0219 return;
0220
0221 kref_put(&nc_node->refcount, batadv_nc_node_release);
0222 }
0223
0224
0225
0226
0227
0228
0229 static void batadv_nc_path_release(struct kref *ref)
0230 {
0231 struct batadv_nc_path *nc_path;
0232
0233 nc_path = container_of(ref, struct batadv_nc_path, refcount);
0234
0235 kfree_rcu(nc_path, rcu);
0236 }
0237
0238
0239
0240
0241
0242
0243 static void batadv_nc_path_put(struct batadv_nc_path *nc_path)
0244 {
0245 if (!nc_path)
0246 return;
0247
0248 kref_put(&nc_path->refcount, batadv_nc_path_release);
0249 }
0250
0251
0252
0253
0254
0255
0256 static void batadv_nc_packet_free(struct batadv_nc_packet *nc_packet,
0257 bool dropped)
0258 {
0259 if (dropped)
0260 kfree_skb(nc_packet->skb);
0261 else
0262 consume_skb(nc_packet->skb);
0263
0264 batadv_nc_path_put(nc_packet->nc_path);
0265 kfree(nc_packet);
0266 }
0267
0268
0269
0270
0271
0272
0273
0274
0275 static bool batadv_nc_to_purge_nc_node(struct batadv_priv *bat_priv,
0276 struct batadv_nc_node *nc_node)
0277 {
0278 if (atomic_read(&bat_priv->mesh_state) != BATADV_MESH_ACTIVE)
0279 return true;
0280
0281 return batadv_has_timed_out(nc_node->last_seen, BATADV_NC_NODE_TIMEOUT);
0282 }
0283
0284
0285
0286
0287
0288
0289
0290
0291 static bool batadv_nc_to_purge_nc_path_coding(struct batadv_priv *bat_priv,
0292 struct batadv_nc_path *nc_path)
0293 {
0294 if (atomic_read(&bat_priv->mesh_state) != BATADV_MESH_ACTIVE)
0295 return true;
0296
0297
0298
0299
0300 return batadv_has_timed_out(nc_path->last_valid,
0301 bat_priv->nc.max_fwd_delay * 10);
0302 }
0303
0304
0305
0306
0307
0308
0309
0310
0311
0312 static bool batadv_nc_to_purge_nc_path_decoding(struct batadv_priv *bat_priv,
0313 struct batadv_nc_path *nc_path)
0314 {
0315 if (atomic_read(&bat_priv->mesh_state) != BATADV_MESH_ACTIVE)
0316 return true;
0317
0318
0319
0320
0321 return batadv_has_timed_out(nc_path->last_valid,
0322 bat_priv->nc.max_buffer_time * 10);
0323 }
0324
0325
0326
0327
0328
0329
0330
0331
0332
0333
0334
0335
0336 static void
0337 batadv_nc_purge_orig_nc_nodes(struct batadv_priv *bat_priv,
0338 struct list_head *list,
0339 spinlock_t *lock,
0340 bool (*to_purge)(struct batadv_priv *,
0341 struct batadv_nc_node *))
0342 {
0343 struct batadv_nc_node *nc_node, *nc_node_tmp;
0344
0345
0346 spin_lock_bh(lock);
0347 list_for_each_entry_safe(nc_node, nc_node_tmp, list, list) {
0348
0349
0350
0351 if (to_purge && !to_purge(bat_priv, nc_node))
0352 continue;
0353
0354 batadv_dbg(BATADV_DBG_NC, bat_priv,
0355 "Removing nc_node %pM -> %pM\n",
0356 nc_node->addr, nc_node->orig_node->orig);
0357 list_del_rcu(&nc_node->list);
0358 batadv_nc_node_put(nc_node);
0359 }
0360 spin_unlock_bh(lock);
0361 }
0362
0363
0364
0365
0366
0367
0368
0369
0370
0371
0372
0373 void batadv_nc_purge_orig(struct batadv_priv *bat_priv,
0374 struct batadv_orig_node *orig_node,
0375 bool (*to_purge)(struct batadv_priv *,
0376 struct batadv_nc_node *))
0377 {
0378
0379 batadv_nc_purge_orig_nc_nodes(bat_priv, &orig_node->in_coding_list,
0380 &orig_node->in_coding_list_lock,
0381 to_purge);
0382
0383
0384 batadv_nc_purge_orig_nc_nodes(bat_priv, &orig_node->out_coding_list,
0385 &orig_node->out_coding_list_lock,
0386 to_purge);
0387 }
0388
0389
0390
0391
0392
0393
0394 static void batadv_nc_purge_orig_hash(struct batadv_priv *bat_priv)
0395 {
0396 struct batadv_hashtable *hash = bat_priv->orig_hash;
0397 struct hlist_head *head;
0398 struct batadv_orig_node *orig_node;
0399 u32 i;
0400
0401 if (!hash)
0402 return;
0403
0404
0405 for (i = 0; i < hash->size; i++) {
0406 head = &hash->table[i];
0407
0408 rcu_read_lock();
0409 hlist_for_each_entry_rcu(orig_node, head, hash_entry)
0410 batadv_nc_purge_orig(bat_priv, orig_node,
0411 batadv_nc_to_purge_nc_node);
0412 rcu_read_unlock();
0413 }
0414 }
0415
0416
0417
0418
0419
0420
0421
0422
0423
0424
0425
0426 static void batadv_nc_purge_paths(struct batadv_priv *bat_priv,
0427 struct batadv_hashtable *hash,
0428 bool (*to_purge)(struct batadv_priv *,
0429 struct batadv_nc_path *))
0430 {
0431 struct hlist_head *head;
0432 struct hlist_node *node_tmp;
0433 struct batadv_nc_path *nc_path;
0434 spinlock_t *lock;
0435 u32 i;
0436
0437 for (i = 0; i < hash->size; i++) {
0438 head = &hash->table[i];
0439 lock = &hash->list_locks[i];
0440
0441
0442 spin_lock_bh(lock);
0443 hlist_for_each_entry_safe(nc_path, node_tmp, head, hash_entry) {
0444
0445
0446
0447 if (to_purge && !to_purge(bat_priv, nc_path))
0448 continue;
0449
0450
0451
0452
0453
0454
0455 if (!unlikely(list_empty(&nc_path->packet_list))) {
0456 net_ratelimited_function(printk,
0457 KERN_WARNING
0458 "Skipping free of non-empty nc_path (%pM -> %pM)!\n",
0459 nc_path->prev_hop,
0460 nc_path->next_hop);
0461 continue;
0462 }
0463
0464
0465 batadv_dbg(BATADV_DBG_NC, bat_priv,
0466 "Remove nc_path %pM -> %pM\n",
0467 nc_path->prev_hop, nc_path->next_hop);
0468 hlist_del_rcu(&nc_path->hash_entry);
0469 batadv_nc_path_put(nc_path);
0470 }
0471 spin_unlock_bh(lock);
0472 }
0473 }
0474
0475
0476
0477
0478
0479
0480
0481 static void batadv_nc_hash_key_gen(struct batadv_nc_path *key, const char *src,
0482 const char *dst)
0483 {
0484 memcpy(key->prev_hop, src, sizeof(key->prev_hop));
0485 memcpy(key->next_hop, dst, sizeof(key->next_hop));
0486 }
0487
0488
0489
0490
0491
0492
0493
0494
0495 static u32 batadv_nc_hash_choose(const void *data, u32 size)
0496 {
0497 const struct batadv_nc_path *nc_path = data;
0498 u32 hash = 0;
0499
0500 hash = jhash(&nc_path->prev_hop, sizeof(nc_path->prev_hop), hash);
0501 hash = jhash(&nc_path->next_hop, sizeof(nc_path->next_hop), hash);
0502
0503 return hash % size;
0504 }
0505
0506
0507
0508
0509
0510
0511
0512
0513
0514 static bool batadv_nc_hash_compare(const struct hlist_node *node,
0515 const void *data2)
0516 {
0517 const struct batadv_nc_path *nc_path1, *nc_path2;
0518
0519 nc_path1 = container_of(node, struct batadv_nc_path, hash_entry);
0520 nc_path2 = data2;
0521
0522
0523 if (!batadv_compare_eth(nc_path1->prev_hop, nc_path2->prev_hop))
0524 return false;
0525
0526 if (!batadv_compare_eth(nc_path1->next_hop, nc_path2->next_hop))
0527 return false;
0528
0529 return true;
0530 }
0531
0532
0533
0534
0535
0536
0537
0538
0539 static struct batadv_nc_path *
0540 batadv_nc_hash_find(struct batadv_hashtable *hash,
0541 void *data)
0542 {
0543 struct hlist_head *head;
0544 struct batadv_nc_path *nc_path, *nc_path_tmp = NULL;
0545 int index;
0546
0547 if (!hash)
0548 return NULL;
0549
0550 index = batadv_nc_hash_choose(data, hash->size);
0551 head = &hash->table[index];
0552
0553 rcu_read_lock();
0554 hlist_for_each_entry_rcu(nc_path, head, hash_entry) {
0555 if (!batadv_nc_hash_compare(&nc_path->hash_entry, data))
0556 continue;
0557
0558 if (!kref_get_unless_zero(&nc_path->refcount))
0559 continue;
0560
0561 nc_path_tmp = nc_path;
0562 break;
0563 }
0564 rcu_read_unlock();
0565
0566 return nc_path_tmp;
0567 }
0568
0569
0570
0571
0572
0573 static void batadv_nc_send_packet(struct batadv_nc_packet *nc_packet)
0574 {
0575 batadv_send_unicast_skb(nc_packet->skb, nc_packet->neigh_node);
0576 nc_packet->skb = NULL;
0577 batadv_nc_packet_free(nc_packet, false);
0578 }
0579
0580
0581
0582
0583
0584
0585
0586
0587
0588
0589
0590
0591
0592
0593 static bool batadv_nc_sniffed_purge(struct batadv_priv *bat_priv,
0594 struct batadv_nc_path *nc_path,
0595 struct batadv_nc_packet *nc_packet)
0596 {
0597 unsigned long timeout = bat_priv->nc.max_buffer_time;
0598 bool res = false;
0599
0600 lockdep_assert_held(&nc_path->packet_list_lock);
0601
0602
0603
0604
0605 if (atomic_read(&bat_priv->mesh_state) == BATADV_MESH_ACTIVE &&
0606 !batadv_has_timed_out(nc_packet->timestamp, timeout))
0607 goto out;
0608
0609
0610 list_del(&nc_packet->list);
0611 batadv_nc_packet_free(nc_packet, true);
0612
0613 res = true;
0614
0615 out:
0616 return res;
0617 }
0618
0619
0620
0621
0622
0623
0624
0625
0626
0627
0628
0629
0630
0631
0632 static bool batadv_nc_fwd_flush(struct batadv_priv *bat_priv,
0633 struct batadv_nc_path *nc_path,
0634 struct batadv_nc_packet *nc_packet)
0635 {
0636 unsigned long timeout = bat_priv->nc.max_fwd_delay;
0637
0638 lockdep_assert_held(&nc_path->packet_list_lock);
0639
0640
0641
0642
0643 if (atomic_read(&bat_priv->mesh_state) == BATADV_MESH_ACTIVE &&
0644 !batadv_has_timed_out(nc_packet->timestamp, timeout))
0645 return false;
0646
0647
0648 batadv_inc_counter(bat_priv, BATADV_CNT_FORWARD);
0649 batadv_add_counter(bat_priv, BATADV_CNT_FORWARD_BYTES,
0650 nc_packet->skb->len + ETH_HLEN);
0651 list_del(&nc_packet->list);
0652 batadv_nc_send_packet(nc_packet);
0653
0654 return true;
0655 }
0656
0657
0658
0659
0660
0661
0662
0663
0664
0665
0666 static void
0667 batadv_nc_process_nc_paths(struct batadv_priv *bat_priv,
0668 struct batadv_hashtable *hash,
0669 bool (*process_fn)(struct batadv_priv *,
0670 struct batadv_nc_path *,
0671 struct batadv_nc_packet *))
0672 {
0673 struct hlist_head *head;
0674 struct batadv_nc_packet *nc_packet, *nc_packet_tmp;
0675 struct batadv_nc_path *nc_path;
0676 bool ret;
0677 int i;
0678
0679 if (!hash)
0680 return;
0681
0682
0683 for (i = 0; i < hash->size; i++) {
0684 head = &hash->table[i];
0685
0686
0687 rcu_read_lock();
0688 hlist_for_each_entry_rcu(nc_path, head, hash_entry) {
0689
0690 spin_lock_bh(&nc_path->packet_list_lock);
0691 list_for_each_entry_safe(nc_packet, nc_packet_tmp,
0692 &nc_path->packet_list, list) {
0693 ret = process_fn(bat_priv, nc_path, nc_packet);
0694 if (!ret)
0695 break;
0696 }
0697 spin_unlock_bh(&nc_path->packet_list_lock);
0698 }
0699 rcu_read_unlock();
0700 }
0701 }
0702
0703
0704
0705
0706
0707
0708 static void batadv_nc_worker(struct work_struct *work)
0709 {
0710 struct delayed_work *delayed_work;
0711 struct batadv_priv_nc *priv_nc;
0712 struct batadv_priv *bat_priv;
0713 unsigned long timeout;
0714
0715 delayed_work = to_delayed_work(work);
0716 priv_nc = container_of(delayed_work, struct batadv_priv_nc, work);
0717 bat_priv = container_of(priv_nc, struct batadv_priv, nc);
0718
0719 batadv_nc_purge_orig_hash(bat_priv);
0720 batadv_nc_purge_paths(bat_priv, bat_priv->nc.coding_hash,
0721 batadv_nc_to_purge_nc_path_coding);
0722 batadv_nc_purge_paths(bat_priv, bat_priv->nc.decoding_hash,
0723 batadv_nc_to_purge_nc_path_decoding);
0724
0725 timeout = bat_priv->nc.max_fwd_delay;
0726
0727 if (batadv_has_timed_out(bat_priv->nc.timestamp_fwd_flush, timeout)) {
0728 batadv_nc_process_nc_paths(bat_priv, bat_priv->nc.coding_hash,
0729 batadv_nc_fwd_flush);
0730 bat_priv->nc.timestamp_fwd_flush = jiffies;
0731 }
0732
0733 if (batadv_has_timed_out(bat_priv->nc.timestamp_sniffed_purge,
0734 bat_priv->nc.max_buffer_time)) {
0735 batadv_nc_process_nc_paths(bat_priv, bat_priv->nc.decoding_hash,
0736 batadv_nc_sniffed_purge);
0737 bat_priv->nc.timestamp_sniffed_purge = jiffies;
0738 }
0739
0740
0741 batadv_nc_start_timer(bat_priv);
0742 }
0743
0744
0745
0746
0747
0748
0749
0750
0751
0752
0753
0754
0755
0756
0757 static bool batadv_can_nc_with_orig(struct batadv_priv *bat_priv,
0758 struct batadv_orig_node *orig_node,
0759 struct batadv_ogm_packet *ogm_packet)
0760 {
0761 struct batadv_orig_ifinfo *orig_ifinfo;
0762 u32 last_real_seqno;
0763 u8 last_ttl;
0764
0765 orig_ifinfo = batadv_orig_ifinfo_get(orig_node, BATADV_IF_DEFAULT);
0766 if (!orig_ifinfo)
0767 return false;
0768
0769 last_ttl = orig_ifinfo->last_ttl;
0770 last_real_seqno = orig_ifinfo->last_real_seqno;
0771 batadv_orig_ifinfo_put(orig_ifinfo);
0772
0773 if (last_real_seqno != ntohl(ogm_packet->seqno))
0774 return false;
0775 if (last_ttl != ogm_packet->ttl + 1)
0776 return false;
0777 if (!batadv_compare_eth(ogm_packet->orig, ogm_packet->prev_sender))
0778 return false;
0779 if (ogm_packet->tq < bat_priv->nc.min_tq)
0780 return false;
0781
0782 return true;
0783 }
0784
0785
0786
0787
0788
0789
0790
0791
0792
0793
0794 static struct batadv_nc_node *
0795 batadv_nc_find_nc_node(struct batadv_orig_node *orig_node,
0796 struct batadv_orig_node *orig_neigh_node,
0797 bool in_coding)
0798 {
0799 struct batadv_nc_node *nc_node, *nc_node_out = NULL;
0800 struct list_head *list;
0801
0802 if (in_coding)
0803 list = &orig_neigh_node->in_coding_list;
0804 else
0805 list = &orig_neigh_node->out_coding_list;
0806
0807
0808 rcu_read_lock();
0809 list_for_each_entry_rcu(nc_node, list, list) {
0810 if (!batadv_compare_eth(nc_node->addr, orig_node->orig))
0811 continue;
0812
0813 if (!kref_get_unless_zero(&nc_node->refcount))
0814 continue;
0815
0816
0817 nc_node_out = nc_node;
0818 break;
0819 }
0820 rcu_read_unlock();
0821
0822 return nc_node_out;
0823 }
0824
0825
0826
0827
0828
0829
0830
0831
0832
0833
0834
0835
0836 static struct batadv_nc_node *
0837 batadv_nc_get_nc_node(struct batadv_priv *bat_priv,
0838 struct batadv_orig_node *orig_node,
0839 struct batadv_orig_node *orig_neigh_node,
0840 bool in_coding)
0841 {
0842 struct batadv_nc_node *nc_node;
0843 spinlock_t *lock;
0844 struct list_head *list;
0845
0846
0847 if (in_coding) {
0848 lock = &orig_neigh_node->in_coding_list_lock;
0849 list = &orig_neigh_node->in_coding_list;
0850 } else {
0851 lock = &orig_neigh_node->out_coding_list_lock;
0852 list = &orig_neigh_node->out_coding_list;
0853 }
0854
0855 spin_lock_bh(lock);
0856
0857
0858 nc_node = batadv_nc_find_nc_node(orig_node, orig_neigh_node, in_coding);
0859
0860
0861 if (nc_node)
0862 goto unlock;
0863
0864 nc_node = kzalloc(sizeof(*nc_node), GFP_ATOMIC);
0865 if (!nc_node)
0866 goto unlock;
0867
0868
0869 INIT_LIST_HEAD(&nc_node->list);
0870 kref_init(&nc_node->refcount);
0871 ether_addr_copy(nc_node->addr, orig_node->orig);
0872 kref_get(&orig_neigh_node->refcount);
0873 nc_node->orig_node = orig_neigh_node;
0874
0875 batadv_dbg(BATADV_DBG_NC, bat_priv, "Adding nc_node %pM -> %pM\n",
0876 nc_node->addr, nc_node->orig_node->orig);
0877
0878
0879 kref_get(&nc_node->refcount);
0880 list_add_tail_rcu(&nc_node->list, list);
0881
0882 unlock:
0883 spin_unlock_bh(lock);
0884
0885 return nc_node;
0886 }
0887
0888
0889
0890
0891
0892
0893
0894
0895
0896
0897
0898 void batadv_nc_update_nc_node(struct batadv_priv *bat_priv,
0899 struct batadv_orig_node *orig_node,
0900 struct batadv_orig_node *orig_neigh_node,
0901 struct batadv_ogm_packet *ogm_packet,
0902 int is_single_hop_neigh)
0903 {
0904 struct batadv_nc_node *in_nc_node = NULL;
0905 struct batadv_nc_node *out_nc_node = NULL;
0906
0907
0908 if (!atomic_read(&bat_priv->network_coding))
0909 goto out;
0910
0911
0912 if (!test_bit(BATADV_ORIG_CAPA_HAS_NC, &orig_node->capabilities))
0913 goto out;
0914
0915
0916 if (!batadv_can_nc_with_orig(bat_priv, orig_node, ogm_packet) &&
0917 !is_single_hop_neigh)
0918 goto out;
0919
0920
0921 in_nc_node = batadv_nc_get_nc_node(bat_priv, orig_node,
0922 orig_neigh_node, true);
0923 if (!in_nc_node)
0924 goto out;
0925
0926 in_nc_node->last_seen = jiffies;
0927
0928
0929 out_nc_node = batadv_nc_get_nc_node(bat_priv, orig_neigh_node,
0930 orig_node, false);
0931 if (!out_nc_node)
0932 goto out;
0933
0934 out_nc_node->last_seen = jiffies;
0935
0936 out:
0937 batadv_nc_node_put(in_nc_node);
0938 batadv_nc_node_put(out_nc_node);
0939 }
0940
0941
0942
0943
0944
0945
0946
0947
0948
0949
0950
0951 static struct batadv_nc_path *batadv_nc_get_path(struct batadv_priv *bat_priv,
0952 struct batadv_hashtable *hash,
0953 u8 *src,
0954 u8 *dst)
0955 {
0956 int hash_added;
0957 struct batadv_nc_path *nc_path, nc_path_key;
0958
0959 batadv_nc_hash_key_gen(&nc_path_key, src, dst);
0960
0961
0962 nc_path = batadv_nc_hash_find(hash, (void *)&nc_path_key);
0963
0964 if (nc_path) {
0965
0966 nc_path->last_valid = jiffies;
0967 return nc_path;
0968 }
0969
0970
0971 nc_path = kzalloc(sizeof(*nc_path), GFP_ATOMIC);
0972
0973 if (!nc_path)
0974 return NULL;
0975
0976
0977 INIT_LIST_HEAD(&nc_path->packet_list);
0978 spin_lock_init(&nc_path->packet_list_lock);
0979 kref_init(&nc_path->refcount);
0980 nc_path->last_valid = jiffies;
0981 ether_addr_copy(nc_path->next_hop, dst);
0982 ether_addr_copy(nc_path->prev_hop, src);
0983
0984 batadv_dbg(BATADV_DBG_NC, bat_priv, "Adding nc_path %pM -> %pM\n",
0985 nc_path->prev_hop,
0986 nc_path->next_hop);
0987
0988
0989 kref_get(&nc_path->refcount);
0990 hash_added = batadv_hash_add(hash, batadv_nc_hash_compare,
0991 batadv_nc_hash_choose, &nc_path_key,
0992 &nc_path->hash_entry);
0993
0994 if (hash_added < 0) {
0995 kfree(nc_path);
0996 return NULL;
0997 }
0998
0999 return nc_path;
1000 }
1001
1002
1003
1004
1005
1006
1007
1008
1009 static u8 batadv_nc_random_weight_tq(u8 tq)
1010 {
1011
1012 u8 rand_tq = prandom_u32_max(BATADV_TQ_MAX_VALUE + 1 - tq);
1013
1014
1015 return BATADV_TQ_MAX_VALUE - rand_tq;
1016 }
1017
1018
1019
1020
1021
1022
1023
1024 static void batadv_nc_memxor(char *dst, const char *src, unsigned int len)
1025 {
1026 unsigned int i;
1027
1028 for (i = 0; i < len; ++i)
1029 dst[i] ^= src[i];
1030 }
1031
1032
1033
1034
1035
1036
1037
1038
1039
1040
1041
1042
1043 static bool batadv_nc_code_packets(struct batadv_priv *bat_priv,
1044 struct sk_buff *skb,
1045 struct ethhdr *ethhdr,
1046 struct batadv_nc_packet *nc_packet,
1047 struct batadv_neigh_node *neigh_node)
1048 {
1049 u8 tq_weighted_neigh, tq_weighted_coding, tq_tmp;
1050 struct sk_buff *skb_dest, *skb_src;
1051 struct batadv_unicast_packet *packet1;
1052 struct batadv_unicast_packet *packet2;
1053 struct batadv_coded_packet *coded_packet;
1054 struct batadv_neigh_node *neigh_tmp, *router_neigh, *first_dest;
1055 struct batadv_neigh_node *router_coding = NULL, *second_dest;
1056 struct batadv_neigh_ifinfo *router_neigh_ifinfo = NULL;
1057 struct batadv_neigh_ifinfo *router_coding_ifinfo = NULL;
1058 u8 *first_source, *second_source;
1059 __be32 packet_id1, packet_id2;
1060 size_t count;
1061 bool res = false;
1062 int coding_len;
1063 int unicast_size = sizeof(*packet1);
1064 int coded_size = sizeof(*coded_packet);
1065 int header_add = coded_size - unicast_size;
1066
1067
1068
1069
1070 router_neigh = batadv_orig_router_get(neigh_node->orig_node,
1071 BATADV_IF_DEFAULT);
1072 if (!router_neigh)
1073 goto out;
1074
1075 router_neigh_ifinfo = batadv_neigh_ifinfo_get(router_neigh,
1076 BATADV_IF_DEFAULT);
1077 if (!router_neigh_ifinfo)
1078 goto out;
1079
1080 neigh_tmp = nc_packet->neigh_node;
1081 router_coding = batadv_orig_router_get(neigh_tmp->orig_node,
1082 BATADV_IF_DEFAULT);
1083 if (!router_coding)
1084 goto out;
1085
1086 router_coding_ifinfo = batadv_neigh_ifinfo_get(router_coding,
1087 BATADV_IF_DEFAULT);
1088 if (!router_coding_ifinfo)
1089 goto out;
1090
1091 tq_tmp = router_neigh_ifinfo->bat_iv.tq_avg;
1092 tq_weighted_neigh = batadv_nc_random_weight_tq(tq_tmp);
1093 tq_tmp = router_coding_ifinfo->bat_iv.tq_avg;
1094 tq_weighted_coding = batadv_nc_random_weight_tq(tq_tmp);
1095
1096
1097
1098
1099 if (tq_weighted_neigh >= tq_weighted_coding) {
1100
1101 first_dest = nc_packet->neigh_node;
1102 first_source = nc_packet->nc_path->prev_hop;
1103 second_dest = neigh_node;
1104 second_source = ethhdr->h_source;
1105 packet1 = (struct batadv_unicast_packet *)nc_packet->skb->data;
1106 packet2 = (struct batadv_unicast_packet *)skb->data;
1107 packet_id1 = nc_packet->packet_id;
1108 packet_id2 = batadv_skb_crc32(skb,
1109 skb->data + sizeof(*packet2));
1110 } else {
1111
1112 first_dest = neigh_node;
1113 first_source = ethhdr->h_source;
1114 second_dest = nc_packet->neigh_node;
1115 second_source = nc_packet->nc_path->prev_hop;
1116 packet1 = (struct batadv_unicast_packet *)skb->data;
1117 packet2 = (struct batadv_unicast_packet *)nc_packet->skb->data;
1118 packet_id1 = batadv_skb_crc32(skb,
1119 skb->data + sizeof(*packet1));
1120 packet_id2 = nc_packet->packet_id;
1121 }
1122
1123
1124
1125
1126 if (skb->len <= nc_packet->skb->len) {
1127 skb_dest = nc_packet->skb;
1128 skb_src = skb;
1129 } else {
1130 skb_dest = skb;
1131 skb_src = nc_packet->skb;
1132 }
1133
1134
1135 coding_len = skb_src->len - unicast_size;
1136
1137 if (skb_linearize(skb_dest) < 0 || skb_linearize(skb_src) < 0)
1138 goto out;
1139
1140 skb_push(skb_dest, header_add);
1141
1142 coded_packet = (struct batadv_coded_packet *)skb_dest->data;
1143 skb_reset_mac_header(skb_dest);
1144
1145 coded_packet->packet_type = BATADV_CODED;
1146 coded_packet->version = BATADV_COMPAT_VERSION;
1147 coded_packet->ttl = packet1->ttl;
1148
1149
1150 ether_addr_copy(coded_packet->first_source, first_source);
1151 ether_addr_copy(coded_packet->first_orig_dest, packet1->dest);
1152 coded_packet->first_crc = packet_id1;
1153 coded_packet->first_ttvn = packet1->ttvn;
1154
1155
1156 ether_addr_copy(coded_packet->second_dest, second_dest->addr);
1157 ether_addr_copy(coded_packet->second_source, second_source);
1158 ether_addr_copy(coded_packet->second_orig_dest, packet2->dest);
1159 coded_packet->second_crc = packet_id2;
1160 coded_packet->second_ttl = packet2->ttl;
1161 coded_packet->second_ttvn = packet2->ttvn;
1162 coded_packet->coded_len = htons(coding_len);
1163
1164
1165 batadv_nc_memxor(skb_dest->data + coded_size,
1166 skb_src->data + unicast_size, coding_len);
1167
1168
1169 if (BATADV_SKB_CB(skb_src)->decoded &&
1170 BATADV_SKB_CB(skb_dest)->decoded) {
1171
1172 count = skb_src->len + ETH_HLEN;
1173 count += skb_dest->len + ETH_HLEN;
1174 batadv_add_counter(bat_priv, BATADV_CNT_NC_RECODE, 2);
1175 batadv_add_counter(bat_priv, BATADV_CNT_NC_RECODE_BYTES, count);
1176 } else if (!BATADV_SKB_CB(skb_src)->decoded &&
1177 !BATADV_SKB_CB(skb_dest)->decoded) {
1178
1179 count = skb_src->len + ETH_HLEN;
1180 count += skb_dest->len + ETH_HLEN;
1181 batadv_add_counter(bat_priv, BATADV_CNT_NC_CODE, 2);
1182 batadv_add_counter(bat_priv, BATADV_CNT_NC_CODE_BYTES, count);
1183 } else if (BATADV_SKB_CB(skb_src)->decoded &&
1184 !BATADV_SKB_CB(skb_dest)->decoded) {
1185
1186 batadv_inc_counter(bat_priv, BATADV_CNT_NC_RECODE);
1187 batadv_add_counter(bat_priv, BATADV_CNT_NC_RECODE_BYTES,
1188 skb_src->len + ETH_HLEN);
1189 batadv_inc_counter(bat_priv, BATADV_CNT_NC_CODE);
1190 batadv_add_counter(bat_priv, BATADV_CNT_NC_CODE_BYTES,
1191 skb_dest->len + ETH_HLEN);
1192 } else if (!BATADV_SKB_CB(skb_src)->decoded &&
1193 BATADV_SKB_CB(skb_dest)->decoded) {
1194
1195 batadv_inc_counter(bat_priv, BATADV_CNT_NC_CODE);
1196 batadv_add_counter(bat_priv, BATADV_CNT_NC_CODE_BYTES,
1197 skb_src->len + ETH_HLEN);
1198 batadv_inc_counter(bat_priv, BATADV_CNT_NC_RECODE);
1199 batadv_add_counter(bat_priv, BATADV_CNT_NC_RECODE_BYTES,
1200 skb_dest->len + ETH_HLEN);
1201 }
1202
1203
1204 consume_skb(skb_src);
1205
1206
1207 nc_packet->skb = NULL;
1208 batadv_nc_packet_free(nc_packet, false);
1209
1210
1211 batadv_send_unicast_skb(skb_dest, first_dest);
1212 res = true;
1213 out:
1214 batadv_neigh_node_put(router_neigh);
1215 batadv_neigh_node_put(router_coding);
1216 batadv_neigh_ifinfo_put(router_neigh_ifinfo);
1217 batadv_neigh_ifinfo_put(router_coding_ifinfo);
1218 return res;
1219 }
1220
1221
1222
1223
1224
1225
1226
1227
1228
1229
1230
1231
1232
1233
1234
1235
1236
1237
1238 static bool batadv_nc_skb_coding_possible(struct sk_buff *skb, u8 *dst, u8 *src)
1239 {
1240 if (BATADV_SKB_CB(skb)->decoded && !batadv_compare_eth(dst, src))
1241 return false;
1242 return true;
1243 }
1244
1245
1246
1247
1248
1249
1250
1251
1252
1253
1254
1255
1256 static struct batadv_nc_packet *
1257 batadv_nc_path_search(struct batadv_priv *bat_priv,
1258 struct batadv_nc_node *in_nc_node,
1259 struct batadv_nc_node *out_nc_node,
1260 struct sk_buff *skb,
1261 u8 *eth_dst)
1262 {
1263 struct batadv_nc_path *nc_path, nc_path_key;
1264 struct batadv_nc_packet *nc_packet_out = NULL;
1265 struct batadv_nc_packet *nc_packet, *nc_packet_tmp;
1266 struct batadv_hashtable *hash = bat_priv->nc.coding_hash;
1267 int idx;
1268
1269 if (!hash)
1270 return NULL;
1271
1272
1273 batadv_nc_hash_key_gen(&nc_path_key, in_nc_node->addr,
1274 out_nc_node->addr);
1275 idx = batadv_nc_hash_choose(&nc_path_key, hash->size);
1276
1277
1278 rcu_read_lock();
1279 hlist_for_each_entry_rcu(nc_path, &hash->table[idx], hash_entry) {
1280 if (!batadv_compare_eth(nc_path->prev_hop, in_nc_node->addr))
1281 continue;
1282
1283 if (!batadv_compare_eth(nc_path->next_hop, out_nc_node->addr))
1284 continue;
1285
1286 spin_lock_bh(&nc_path->packet_list_lock);
1287 if (list_empty(&nc_path->packet_list)) {
1288 spin_unlock_bh(&nc_path->packet_list_lock);
1289 continue;
1290 }
1291
1292 list_for_each_entry_safe(nc_packet, nc_packet_tmp,
1293 &nc_path->packet_list, list) {
1294 if (!batadv_nc_skb_coding_possible(nc_packet->skb,
1295 eth_dst,
1296 in_nc_node->addr))
1297 continue;
1298
1299
1300 list_del(&nc_packet->list);
1301 nc_packet_out = nc_packet;
1302 break;
1303 }
1304
1305 spin_unlock_bh(&nc_path->packet_list_lock);
1306 break;
1307 }
1308 rcu_read_unlock();
1309
1310 return nc_packet_out;
1311 }
1312
1313
1314
1315
1316
1317
1318
1319
1320
1321
1322
1323
1324 static struct batadv_nc_packet *
1325 batadv_nc_skb_src_search(struct batadv_priv *bat_priv,
1326 struct sk_buff *skb,
1327 u8 *eth_dst,
1328 u8 *eth_src,
1329 struct batadv_nc_node *in_nc_node)
1330 {
1331 struct batadv_orig_node *orig_node;
1332 struct batadv_nc_node *out_nc_node;
1333 struct batadv_nc_packet *nc_packet = NULL;
1334
1335 orig_node = batadv_orig_hash_find(bat_priv, eth_src);
1336 if (!orig_node)
1337 return NULL;
1338
1339 rcu_read_lock();
1340 list_for_each_entry_rcu(out_nc_node,
1341 &orig_node->out_coding_list, list) {
1342
1343 if (!batadv_nc_skb_coding_possible(skb,
1344 out_nc_node->addr, eth_src))
1345 continue;
1346
1347
1348 nc_packet = batadv_nc_path_search(bat_priv, in_nc_node,
1349 out_nc_node, skb, eth_dst);
1350 if (nc_packet)
1351 break;
1352 }
1353 rcu_read_unlock();
1354
1355 batadv_orig_node_put(orig_node);
1356 return nc_packet;
1357 }
1358
1359
1360
1361
1362
1363
1364
1365
1366 static void batadv_nc_skb_store_before_coding(struct batadv_priv *bat_priv,
1367 struct sk_buff *skb,
1368 u8 *eth_dst_new)
1369 {
1370 struct ethhdr *ethhdr;
1371
1372
1373 skb = pskb_copy_for_clone(skb, GFP_ATOMIC);
1374 if (!skb)
1375 return;
1376
1377
1378 ethhdr = eth_hdr(skb);
1379 ether_addr_copy(ethhdr->h_source, ethhdr->h_dest);
1380 ether_addr_copy(ethhdr->h_dest, eth_dst_new);
1381
1382
1383 skb_push(skb, ETH_HLEN);
1384
1385
1386 batadv_nc_skb_store_for_decoding(bat_priv, skb);
1387
1388
1389
1390
1391 consume_skb(skb);
1392 }
1393
1394
1395
1396
1397
1398
1399
1400
1401
1402
1403
1404
1405
1406
1407 static bool batadv_nc_skb_dst_search(struct sk_buff *skb,
1408 struct batadv_neigh_node *neigh_node,
1409 struct ethhdr *ethhdr)
1410 {
1411 struct net_device *netdev = neigh_node->if_incoming->soft_iface;
1412 struct batadv_priv *bat_priv = netdev_priv(netdev);
1413 struct batadv_orig_node *orig_node = neigh_node->orig_node;
1414 struct batadv_nc_node *nc_node;
1415 struct batadv_nc_packet *nc_packet = NULL;
1416
1417 rcu_read_lock();
1418 list_for_each_entry_rcu(nc_node, &orig_node->in_coding_list, list) {
1419
1420 nc_packet = batadv_nc_skb_src_search(bat_priv, skb,
1421 neigh_node->addr,
1422 ethhdr->h_source, nc_node);
1423
1424
1425 if (nc_packet)
1426 break;
1427 }
1428 rcu_read_unlock();
1429
1430 if (!nc_packet)
1431 return false;
1432
1433
1434 batadv_nc_skb_store_before_coding(bat_priv, skb,
1435 neigh_node->addr);
1436 batadv_nc_skb_store_before_coding(bat_priv, nc_packet->skb,
1437 nc_packet->neigh_node->addr);
1438
1439
1440 if (batadv_nc_code_packets(bat_priv, skb, ethhdr, nc_packet,
1441 neigh_node))
1442 return true;
1443
1444
1445
1446
1447
1448 batadv_nc_send_packet(nc_packet);
1449 return false;
1450 }
1451
1452
1453
1454
1455
1456
1457
1458
1459
1460
1461 static bool batadv_nc_skb_add_to_path(struct sk_buff *skb,
1462 struct batadv_nc_path *nc_path,
1463 struct batadv_neigh_node *neigh_node,
1464 __be32 packet_id)
1465 {
1466 struct batadv_nc_packet *nc_packet;
1467
1468 nc_packet = kzalloc(sizeof(*nc_packet), GFP_ATOMIC);
1469 if (!nc_packet)
1470 return false;
1471
1472
1473 nc_packet->timestamp = jiffies;
1474 nc_packet->packet_id = packet_id;
1475 nc_packet->skb = skb;
1476 nc_packet->neigh_node = neigh_node;
1477 nc_packet->nc_path = nc_path;
1478
1479
1480 spin_lock_bh(&nc_path->packet_list_lock);
1481 list_add_tail(&nc_packet->list, &nc_path->packet_list);
1482 spin_unlock_bh(&nc_path->packet_list_lock);
1483
1484 return true;
1485 }
1486
1487
1488
1489
1490
1491
1492
1493
1494
1495 bool batadv_nc_skb_forward(struct sk_buff *skb,
1496 struct batadv_neigh_node *neigh_node)
1497 {
1498 const struct net_device *netdev = neigh_node->if_incoming->soft_iface;
1499 struct batadv_priv *bat_priv = netdev_priv(netdev);
1500 struct batadv_unicast_packet *packet;
1501 struct batadv_nc_path *nc_path;
1502 struct ethhdr *ethhdr = eth_hdr(skb);
1503 __be32 packet_id;
1504 u8 *payload;
1505
1506
1507 if (!atomic_read(&bat_priv->network_coding))
1508 goto out;
1509
1510
1511 payload = skb_network_header(skb);
1512 packet = (struct batadv_unicast_packet *)payload;
1513 if (packet->packet_type != BATADV_UNICAST)
1514 goto out;
1515
1516
1517 if (batadv_nc_skb_dst_search(skb, neigh_node, ethhdr))
1518 return true;
1519
1520
1521 nc_path = batadv_nc_get_path(bat_priv,
1522 bat_priv->nc.coding_hash,
1523 ethhdr->h_source,
1524 neigh_node->addr);
1525
1526 if (!nc_path)
1527 goto out;
1528
1529
1530 packet_id = batadv_skb_crc32(skb, payload + sizeof(*packet));
1531 if (!batadv_nc_skb_add_to_path(skb, nc_path, neigh_node, packet_id))
1532 goto free_nc_path;
1533
1534
1535 return true;
1536
1537 free_nc_path:
1538 batadv_nc_path_put(nc_path);
1539 out:
1540
1541 return false;
1542 }
1543
1544
1545
1546
1547
1548
1549
1550 void batadv_nc_skb_store_for_decoding(struct batadv_priv *bat_priv,
1551 struct sk_buff *skb)
1552 {
1553 struct batadv_unicast_packet *packet;
1554 struct batadv_nc_path *nc_path;
1555 struct ethhdr *ethhdr = eth_hdr(skb);
1556 __be32 packet_id;
1557 u8 *payload;
1558
1559
1560 if (!atomic_read(&bat_priv->network_coding))
1561 goto out;
1562
1563
1564 payload = skb_network_header(skb);
1565 packet = (struct batadv_unicast_packet *)payload;
1566 if (packet->packet_type != BATADV_UNICAST)
1567 goto out;
1568
1569
1570 nc_path = batadv_nc_get_path(bat_priv,
1571 bat_priv->nc.decoding_hash,
1572 ethhdr->h_source,
1573 ethhdr->h_dest);
1574
1575 if (!nc_path)
1576 goto out;
1577
1578
1579 skb = skb_clone(skb, GFP_ATOMIC);
1580 if (unlikely(!skb))
1581 goto free_nc_path;
1582
1583 if (unlikely(!pskb_may_pull(skb, ETH_HLEN)))
1584 goto free_skb;
1585
1586 if (unlikely(!skb_pull_rcsum(skb, ETH_HLEN)))
1587 goto free_skb;
1588
1589
1590 packet_id = batadv_skb_crc32(skb, payload + sizeof(*packet));
1591 if (!batadv_nc_skb_add_to_path(skb, nc_path, NULL, packet_id))
1592 goto free_skb;
1593
1594 batadv_inc_counter(bat_priv, BATADV_CNT_NC_BUFFER);
1595 return;
1596
1597 free_skb:
1598 kfree_skb(skb);
1599 free_nc_path:
1600 batadv_nc_path_put(nc_path);
1601 out:
1602 return;
1603 }
1604
1605
1606
1607
1608
1609
1610
1611 void batadv_nc_skb_store_sniffed_unicast(struct batadv_priv *bat_priv,
1612 struct sk_buff *skb)
1613 {
1614 struct ethhdr *ethhdr = eth_hdr(skb);
1615
1616 if (batadv_is_my_mac(bat_priv, ethhdr->h_dest))
1617 return;
1618
1619
1620 skb_push(skb, ETH_HLEN);
1621
1622 batadv_nc_skb_store_for_decoding(bat_priv, skb);
1623 }
1624
1625
1626
1627
1628
1629
1630
1631
1632
1633
1634
1635 static struct batadv_unicast_packet *
1636 batadv_nc_skb_decode_packet(struct batadv_priv *bat_priv, struct sk_buff *skb,
1637 struct batadv_nc_packet *nc_packet)
1638 {
1639 const int h_size = sizeof(struct batadv_unicast_packet);
1640 const int h_diff = sizeof(struct batadv_coded_packet) - h_size;
1641 struct batadv_unicast_packet *unicast_packet;
1642 struct batadv_coded_packet coded_packet_tmp;
1643 struct ethhdr *ethhdr, ethhdr_tmp;
1644 u8 *orig_dest, ttl, ttvn;
1645 unsigned int coding_len;
1646 int err;
1647
1648
1649 memcpy(&coded_packet_tmp, skb->data, sizeof(coded_packet_tmp));
1650 memcpy(ðhdr_tmp, skb_mac_header(skb), sizeof(ethhdr_tmp));
1651
1652 if (skb_cow(skb, 0) < 0)
1653 return NULL;
1654
1655 if (unlikely(!skb_pull_rcsum(skb, h_diff)))
1656 return NULL;
1657
1658
1659
1660
1661 skb_set_mac_header(skb, -ETH_HLEN);
1662 skb_reset_network_header(skb);
1663
1664
1665 ethhdr = eth_hdr(skb);
1666 *ethhdr = ethhdr_tmp;
1667
1668
1669
1670
1671 if (batadv_is_my_mac(bat_priv, coded_packet_tmp.second_dest)) {
1672
1673
1674
1675
1676 ether_addr_copy(ethhdr->h_dest, coded_packet_tmp.second_dest);
1677 skb->pkt_type = PACKET_HOST;
1678
1679 orig_dest = coded_packet_tmp.second_orig_dest;
1680 ttl = coded_packet_tmp.second_ttl;
1681 ttvn = coded_packet_tmp.second_ttvn;
1682 } else {
1683 orig_dest = coded_packet_tmp.first_orig_dest;
1684 ttl = coded_packet_tmp.ttl;
1685 ttvn = coded_packet_tmp.first_ttvn;
1686 }
1687
1688 coding_len = ntohs(coded_packet_tmp.coded_len);
1689
1690 if (coding_len > skb->len)
1691 return NULL;
1692
1693
1694
1695
1696 batadv_nc_memxor(skb->data + h_size,
1697 nc_packet->skb->data + h_size,
1698 coding_len);
1699
1700
1701 if (nc_packet->skb->len > coding_len + h_size) {
1702 err = pskb_trim_rcsum(skb, coding_len + h_size);
1703 if (err)
1704 return NULL;
1705 }
1706
1707
1708 unicast_packet = (struct batadv_unicast_packet *)skb->data;
1709 unicast_packet->packet_type = BATADV_UNICAST;
1710 unicast_packet->version = BATADV_COMPAT_VERSION;
1711 unicast_packet->ttl = ttl;
1712 ether_addr_copy(unicast_packet->dest, orig_dest);
1713 unicast_packet->ttvn = ttvn;
1714
1715 batadv_nc_packet_free(nc_packet, false);
1716 return unicast_packet;
1717 }
1718
1719
1720
1721
1722
1723
1724
1725
1726
1727
1728 static struct batadv_nc_packet *
1729 batadv_nc_find_decoding_packet(struct batadv_priv *bat_priv,
1730 struct ethhdr *ethhdr,
1731 struct batadv_coded_packet *coded)
1732 {
1733 struct batadv_hashtable *hash = bat_priv->nc.decoding_hash;
1734 struct batadv_nc_packet *tmp_nc_packet, *nc_packet = NULL;
1735 struct batadv_nc_path *nc_path, nc_path_key;
1736 u8 *dest, *source;
1737 __be32 packet_id;
1738 int index;
1739
1740 if (!hash)
1741 return NULL;
1742
1743
1744 dest = ethhdr->h_source;
1745 if (!batadv_is_my_mac(bat_priv, coded->second_dest)) {
1746 source = coded->second_source;
1747 packet_id = coded->second_crc;
1748 } else {
1749 source = coded->first_source;
1750 packet_id = coded->first_crc;
1751 }
1752
1753 batadv_nc_hash_key_gen(&nc_path_key, source, dest);
1754 index = batadv_nc_hash_choose(&nc_path_key, hash->size);
1755
1756
1757 rcu_read_lock();
1758 hlist_for_each_entry_rcu(nc_path, &hash->table[index], hash_entry) {
1759
1760 spin_lock_bh(&nc_path->packet_list_lock);
1761 list_for_each_entry(tmp_nc_packet,
1762 &nc_path->packet_list, list) {
1763 if (packet_id == tmp_nc_packet->packet_id) {
1764 list_del(&tmp_nc_packet->list);
1765
1766 nc_packet = tmp_nc_packet;
1767 break;
1768 }
1769 }
1770 spin_unlock_bh(&nc_path->packet_list_lock);
1771
1772 if (nc_packet)
1773 break;
1774 }
1775 rcu_read_unlock();
1776
1777 if (!nc_packet)
1778 batadv_dbg(BATADV_DBG_NC, bat_priv,
1779 "No decoding packet found for %u\n", packet_id);
1780
1781 return nc_packet;
1782 }
1783
1784
1785
1786
1787
1788
1789
1790
1791
1792
1793 static int batadv_nc_recv_coded_packet(struct sk_buff *skb,
1794 struct batadv_hard_iface *recv_if)
1795 {
1796 struct batadv_priv *bat_priv = netdev_priv(recv_if->soft_iface);
1797 struct batadv_unicast_packet *unicast_packet;
1798 struct batadv_coded_packet *coded_packet;
1799 struct batadv_nc_packet *nc_packet;
1800 struct ethhdr *ethhdr;
1801 int hdr_size = sizeof(*coded_packet);
1802
1803
1804 if (!atomic_read(&bat_priv->network_coding))
1805 goto free_skb;
1806
1807
1808 if (unlikely(!pskb_may_pull(skb, hdr_size)))
1809 goto free_skb;
1810
1811 coded_packet = (struct batadv_coded_packet *)skb->data;
1812 ethhdr = eth_hdr(skb);
1813
1814
1815 if (!batadv_is_my_mac(bat_priv, ethhdr->h_dest) &&
1816 !batadv_is_my_mac(bat_priv, coded_packet->second_dest))
1817 goto free_skb;
1818
1819
1820 if (batadv_is_my_mac(bat_priv, coded_packet->second_dest))
1821 batadv_inc_counter(bat_priv, BATADV_CNT_NC_SNIFFED);
1822
1823 nc_packet = batadv_nc_find_decoding_packet(bat_priv, ethhdr,
1824 coded_packet);
1825 if (!nc_packet) {
1826 batadv_inc_counter(bat_priv, BATADV_CNT_NC_DECODE_FAILED);
1827 goto free_skb;
1828 }
1829
1830
1831 if (skb_linearize(skb) < 0)
1832 goto free_nc_packet;
1833
1834 if (skb_linearize(nc_packet->skb) < 0)
1835 goto free_nc_packet;
1836
1837
1838 unicast_packet = batadv_nc_skb_decode_packet(bat_priv, skb, nc_packet);
1839 if (!unicast_packet) {
1840 batadv_inc_counter(bat_priv, BATADV_CNT_NC_DECODE_FAILED);
1841 goto free_nc_packet;
1842 }
1843
1844
1845 BATADV_SKB_CB(skb)->decoded = true;
1846 batadv_inc_counter(bat_priv, BATADV_CNT_NC_DECODE);
1847 batadv_add_counter(bat_priv, BATADV_CNT_NC_DECODE_BYTES,
1848 skb->len + ETH_HLEN);
1849 return batadv_recv_unicast_packet(skb, recv_if);
1850
1851 free_nc_packet:
1852 batadv_nc_packet_free(nc_packet, true);
1853 free_skb:
1854 kfree_skb(skb);
1855
1856 return NET_RX_DROP;
1857 }
1858
1859
1860
1861
1862
1863 void batadv_nc_mesh_free(struct batadv_priv *bat_priv)
1864 {
1865 batadv_tvlv_container_unregister(bat_priv, BATADV_TVLV_NC, 1);
1866 batadv_tvlv_handler_unregister(bat_priv, BATADV_TVLV_NC, 1);
1867 cancel_delayed_work_sync(&bat_priv->nc.work);
1868
1869 batadv_nc_purge_paths(bat_priv, bat_priv->nc.coding_hash, NULL);
1870 batadv_hash_destroy(bat_priv->nc.coding_hash);
1871 batadv_nc_purge_paths(bat_priv, bat_priv->nc.decoding_hash, NULL);
1872 batadv_hash_destroy(bat_priv->nc.decoding_hash);
1873 }