8abda424aae311aae6ccd689e6f1c18199af4045
[cascardo/linux.git] / net / batman-adv / main.c
1 /* Copyright (C) 2007-2013 B.A.T.M.A.N. contributors:
2  *
3  * Marek Lindner, Simon Wunderlich
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 <linux/crc32c.h>
19 #include <linux/highmem.h>
20 #include <linux/if_vlan.h>
21 #include <net/ip.h>
22 #include <net/ipv6.h>
23 #include <net/dsfield.h>
24 #include "main.h"
25 #include "sysfs.h"
26 #include "debugfs.h"
27 #include "routing.h"
28 #include "send.h"
29 #include "originator.h"
30 #include "soft-interface.h"
31 #include "icmp_socket.h"
32 #include "translation-table.h"
33 #include "hard-interface.h"
34 #include "gateway_client.h"
35 #include "bridge_loop_avoidance.h"
36 #include "distributed-arp-table.h"
37 #include "gateway_common.h"
38 #include "hash.h"
39 #include "bat_algo.h"
40 #include "network-coding.h"
41 #include "fragmentation.h"
42
43
44 /* List manipulations on hardif_list have to be rtnl_lock()'ed,
45  * list traversals just rcu-locked
46  */
47 struct list_head batadv_hardif_list;
48 static int (*batadv_rx_handler[256])(struct sk_buff *,
49                                      struct batadv_hard_iface *);
50 char batadv_routing_algo[20] = "BATMAN_IV";
51 static struct hlist_head batadv_algo_list;
52
53 unsigned char batadv_broadcast_addr[] = {0xff, 0xff, 0xff, 0xff, 0xff, 0xff};
54
55 struct workqueue_struct *batadv_event_workqueue;
56
57 static void batadv_recv_handler_init(void);
58
59 static int __init batadv_init(void)
60 {
61         INIT_LIST_HEAD(&batadv_hardif_list);
62         INIT_HLIST_HEAD(&batadv_algo_list);
63
64         batadv_recv_handler_init();
65
66         batadv_iv_init();
67         batadv_nc_init();
68
69         batadv_event_workqueue = create_singlethread_workqueue("bat_events");
70
71         if (!batadv_event_workqueue)
72                 return -ENOMEM;
73
74         batadv_socket_init();
75         batadv_debugfs_init();
76
77         register_netdevice_notifier(&batadv_hard_if_notifier);
78         rtnl_link_register(&batadv_link_ops);
79
80         pr_info("B.A.T.M.A.N. advanced %s (compatibility version %i) loaded\n",
81                 BATADV_SOURCE_VERSION, BATADV_COMPAT_VERSION);
82
83         return 0;
84 }
85
86 static void __exit batadv_exit(void)
87 {
88         batadv_debugfs_destroy();
89         rtnl_link_unregister(&batadv_link_ops);
90         unregister_netdevice_notifier(&batadv_hard_if_notifier);
91         batadv_hardif_remove_interfaces();
92
93         flush_workqueue(batadv_event_workqueue);
94         destroy_workqueue(batadv_event_workqueue);
95         batadv_event_workqueue = NULL;
96
97         rcu_barrier();
98 }
99
100 int batadv_mesh_init(struct net_device *soft_iface)
101 {
102         struct batadv_priv *bat_priv = netdev_priv(soft_iface);
103         int ret;
104
105         spin_lock_init(&bat_priv->forw_bat_list_lock);
106         spin_lock_init(&bat_priv->forw_bcast_list_lock);
107         spin_lock_init(&bat_priv->tt.changes_list_lock);
108         spin_lock_init(&bat_priv->tt.req_list_lock);
109         spin_lock_init(&bat_priv->tt.roam_list_lock);
110         spin_lock_init(&bat_priv->tt.last_changeset_lock);
111         spin_lock_init(&bat_priv->tt.commit_lock);
112         spin_lock_init(&bat_priv->gw.list_lock);
113         spin_lock_init(&bat_priv->tvlv.container_list_lock);
114         spin_lock_init(&bat_priv->tvlv.handler_list_lock);
115         spin_lock_init(&bat_priv->softif_vlan_list_lock);
116
117         INIT_HLIST_HEAD(&bat_priv->forw_bat_list);
118         INIT_HLIST_HEAD(&bat_priv->forw_bcast_list);
119         INIT_HLIST_HEAD(&bat_priv->gw.list);
120         INIT_LIST_HEAD(&bat_priv->tt.changes_list);
121         INIT_LIST_HEAD(&bat_priv->tt.req_list);
122         INIT_LIST_HEAD(&bat_priv->tt.roam_list);
123         INIT_HLIST_HEAD(&bat_priv->tvlv.container_list);
124         INIT_HLIST_HEAD(&bat_priv->tvlv.handler_list);
125         INIT_HLIST_HEAD(&bat_priv->softif_vlan_list);
126
127         ret = batadv_originator_init(bat_priv);
128         if (ret < 0)
129                 goto err;
130
131         ret = batadv_tt_init(bat_priv);
132         if (ret < 0)
133                 goto err;
134
135         ret = batadv_bla_init(bat_priv);
136         if (ret < 0)
137                 goto err;
138
139         ret = batadv_dat_init(bat_priv);
140         if (ret < 0)
141                 goto err;
142
143         ret = batadv_nc_mesh_init(bat_priv);
144         if (ret < 0)
145                 goto err;
146
147         batadv_gw_init(bat_priv);
148
149         atomic_set(&bat_priv->gw.reselect, 0);
150         atomic_set(&bat_priv->mesh_state, BATADV_MESH_ACTIVE);
151
152         return 0;
153
154 err:
155         batadv_mesh_free(soft_iface);
156         return ret;
157 }
158
159 void batadv_mesh_free(struct net_device *soft_iface)
160 {
161         struct batadv_priv *bat_priv = netdev_priv(soft_iface);
162
163         atomic_set(&bat_priv->mesh_state, BATADV_MESH_DEACTIVATING);
164
165         batadv_purge_outstanding_packets(bat_priv, NULL);
166
167         batadv_gw_node_purge(bat_priv);
168         batadv_nc_mesh_free(bat_priv);
169         batadv_dat_free(bat_priv);
170         batadv_bla_free(bat_priv);
171
172         /* Free the TT and the originator tables only after having terminated
173          * all the other depending components which may use these structures for
174          * their purposes.
175          */
176         batadv_tt_free(bat_priv);
177
178         /* Since the originator table clean up routine is accessing the TT
179          * tables as well, it has to be invoked after the TT tables have been
180          * freed and marked as empty. This ensures that no cleanup RCU callbacks
181          * accessing the TT data are scheduled for later execution.
182          */
183         batadv_originator_free(bat_priv);
184
185         batadv_gw_free(bat_priv);
186
187         free_percpu(bat_priv->bat_counters);
188         bat_priv->bat_counters = NULL;
189
190         atomic_set(&bat_priv->mesh_state, BATADV_MESH_INACTIVE);
191 }
192
193 /**
194  * batadv_is_my_mac - check if the given mac address belongs to any of the real
195  * interfaces in the current mesh
196  * @bat_priv: the bat priv with all the soft interface information
197  * @addr: the address to check
198  */
199 int batadv_is_my_mac(struct batadv_priv *bat_priv, const uint8_t *addr)
200 {
201         const struct batadv_hard_iface *hard_iface;
202
203         rcu_read_lock();
204         list_for_each_entry_rcu(hard_iface, &batadv_hardif_list, list) {
205                 if (hard_iface->if_status != BATADV_IF_ACTIVE)
206                         continue;
207
208                 if (hard_iface->soft_iface != bat_priv->soft_iface)
209                         continue;
210
211                 if (batadv_compare_eth(hard_iface->net_dev->dev_addr, addr)) {
212                         rcu_read_unlock();
213                         return 1;
214                 }
215         }
216         rcu_read_unlock();
217         return 0;
218 }
219
220 /**
221  * batadv_seq_print_text_primary_if_get - called from debugfs table printing
222  *  function that requires the primary interface
223  * @seq: debugfs table seq_file struct
224  *
225  * Returns primary interface if found or NULL otherwise.
226  */
227 struct batadv_hard_iface *
228 batadv_seq_print_text_primary_if_get(struct seq_file *seq)
229 {
230         struct net_device *net_dev = (struct net_device *)seq->private;
231         struct batadv_priv *bat_priv = netdev_priv(net_dev);
232         struct batadv_hard_iface *primary_if;
233
234         primary_if = batadv_primary_if_get_selected(bat_priv);
235
236         if (!primary_if) {
237                 seq_printf(seq,
238                            "BATMAN mesh %s disabled - please specify interfaces to enable it\n",
239                            net_dev->name);
240                 goto out;
241         }
242
243         if (primary_if->if_status == BATADV_IF_ACTIVE)
244                 goto out;
245
246         seq_printf(seq,
247                    "BATMAN mesh %s disabled - primary interface not active\n",
248                    net_dev->name);
249         batadv_hardif_free_ref(primary_if);
250         primary_if = NULL;
251
252 out:
253         return primary_if;
254 }
255
256 /**
257  * batadv_max_header_len - calculate maximum encapsulation overhead for a
258  *  payload packet
259  *
260  * Return the maximum encapsulation overhead in bytes.
261  */
262 int batadv_max_header_len(void)
263 {
264         int header_len = 0;
265
266         header_len = max_t(int, header_len,
267                            sizeof(struct batadv_unicast_packet));
268         header_len = max_t(int, header_len,
269                            sizeof(struct batadv_unicast_4addr_packet));
270         header_len = max_t(int, header_len,
271                            sizeof(struct batadv_bcast_packet));
272
273 #ifdef CONFIG_BATMAN_ADV_NC
274         header_len = max_t(int, header_len,
275                            sizeof(struct batadv_coded_packet));
276 #endif
277
278         return header_len;
279 }
280
281 /**
282  * batadv_skb_set_priority - sets skb priority according to packet content
283  * @skb: the packet to be sent
284  * @offset: offset to the packet content
285  *
286  * This function sets a value between 256 and 263 (802.1d priority), which
287  * can be interpreted by the cfg80211 or other drivers.
288  */
289 void batadv_skb_set_priority(struct sk_buff *skb, int offset)
290 {
291         struct iphdr ip_hdr_tmp, *ip_hdr;
292         struct ipv6hdr ip6_hdr_tmp, *ip6_hdr;
293         struct ethhdr ethhdr_tmp, *ethhdr;
294         struct vlan_ethhdr *vhdr, vhdr_tmp;
295         u32 prio;
296
297         /* already set, do nothing */
298         if (skb->priority >= 256 && skb->priority <= 263)
299                 return;
300
301         ethhdr = skb_header_pointer(skb, offset, sizeof(*ethhdr), &ethhdr_tmp);
302         if (!ethhdr)
303                 return;
304
305         switch (ethhdr->h_proto) {
306         case htons(ETH_P_8021Q):
307                 vhdr = skb_header_pointer(skb, offset + sizeof(*vhdr),
308                                           sizeof(*vhdr), &vhdr_tmp);
309                 if (!vhdr)
310                         return;
311                 prio = ntohs(vhdr->h_vlan_TCI) & VLAN_PRIO_MASK;
312                 prio = prio >> VLAN_PRIO_SHIFT;
313                 break;
314         case htons(ETH_P_IP):
315                 ip_hdr = skb_header_pointer(skb, offset + sizeof(*ethhdr),
316                                             sizeof(*ip_hdr), &ip_hdr_tmp);
317                 if (!ip_hdr)
318                         return;
319                 prio = (ipv4_get_dsfield(ip_hdr) & 0xfc) >> 5;
320                 break;
321         case htons(ETH_P_IPV6):
322                 ip6_hdr = skb_header_pointer(skb, offset + sizeof(*ethhdr),
323                                              sizeof(*ip6_hdr), &ip6_hdr_tmp);
324                 if (!ip6_hdr)
325                         return;
326                 prio = (ipv6_get_dsfield(ip6_hdr) & 0xfc) >> 5;
327                 break;
328         default:
329                 return;
330         }
331
332         skb->priority = prio + 256;
333 }
334
335 static int batadv_recv_unhandled_packet(struct sk_buff *skb,
336                                         struct batadv_hard_iface *recv_if)
337 {
338         return NET_RX_DROP;
339 }
340
341 /* incoming packets with the batman ethertype received on any active hard
342  * interface
343  */
344 int batadv_batman_skb_recv(struct sk_buff *skb, struct net_device *dev,
345                            struct packet_type *ptype,
346                            struct net_device *orig_dev)
347 {
348         struct batadv_priv *bat_priv;
349         struct batadv_ogm_packet *batadv_ogm_packet;
350         struct batadv_hard_iface *hard_iface;
351         uint8_t idx;
352         int ret;
353
354         hard_iface = container_of(ptype, struct batadv_hard_iface,
355                                   batman_adv_ptype);
356         skb = skb_share_check(skb, GFP_ATOMIC);
357
358         /* skb was released by skb_share_check() */
359         if (!skb)
360                 goto err_out;
361
362         /* packet should hold at least type and version */
363         if (unlikely(!pskb_may_pull(skb, 2)))
364                 goto err_free;
365
366         /* expect a valid ethernet header here. */
367         if (unlikely(skb->mac_len != ETH_HLEN || !skb_mac_header(skb)))
368                 goto err_free;
369
370         if (!hard_iface->soft_iface)
371                 goto err_free;
372
373         bat_priv = netdev_priv(hard_iface->soft_iface);
374
375         if (atomic_read(&bat_priv->mesh_state) != BATADV_MESH_ACTIVE)
376                 goto err_free;
377
378         /* discard frames on not active interfaces */
379         if (hard_iface->if_status != BATADV_IF_ACTIVE)
380                 goto err_free;
381
382         batadv_ogm_packet = (struct batadv_ogm_packet *)skb->data;
383
384         if (batadv_ogm_packet->version != BATADV_COMPAT_VERSION) {
385                 batadv_dbg(BATADV_DBG_BATMAN, bat_priv,
386                            "Drop packet: incompatible batman version (%i)\n",
387                            batadv_ogm_packet->version);
388                 goto err_free;
389         }
390
391         /* all receive handlers return whether they received or reused
392          * the supplied skb. if not, we have to free the skb.
393          */
394         idx = batadv_ogm_packet->packet_type;
395         ret = (*batadv_rx_handler[idx])(skb, hard_iface);
396
397         if (ret == NET_RX_DROP)
398                 kfree_skb(skb);
399
400         /* return NET_RX_SUCCESS in any case as we
401          * most probably dropped the packet for
402          * routing-logical reasons.
403          */
404         return NET_RX_SUCCESS;
405
406 err_free:
407         kfree_skb(skb);
408 err_out:
409         return NET_RX_DROP;
410 }
411
412 static void batadv_recv_handler_init(void)
413 {
414         int i;
415
416         for (i = 0; i < ARRAY_SIZE(batadv_rx_handler); i++)
417                 batadv_rx_handler[i] = batadv_recv_unhandled_packet;
418
419         for (i = BATADV_UNICAST_MIN; i <= BATADV_UNICAST_MAX; i++)
420                 batadv_rx_handler[i] = batadv_recv_unhandled_unicast_packet;
421
422         /* compile time checks for struct member offsets */
423         BUILD_BUG_ON(offsetof(struct batadv_unicast_4addr_packet, src) != 10);
424         BUILD_BUG_ON(offsetof(struct batadv_unicast_packet, dest) != 4);
425         BUILD_BUG_ON(offsetof(struct batadv_unicast_tvlv_packet, dst) != 4);
426         BUILD_BUG_ON(offsetof(struct batadv_frag_packet, dest) != 4);
427         BUILD_BUG_ON(offsetof(struct batadv_icmp_packet, dst) != 4);
428         BUILD_BUG_ON(offsetof(struct batadv_icmp_packet_rr, dst) != 4);
429
430         /* broadcast packet */
431         batadv_rx_handler[BATADV_BCAST] = batadv_recv_bcast_packet;
432
433         /* unicast packets ... */
434         /* unicast with 4 addresses packet */
435         batadv_rx_handler[BATADV_UNICAST_4ADDR] = batadv_recv_unicast_packet;
436         /* unicast packet */
437         batadv_rx_handler[BATADV_UNICAST] = batadv_recv_unicast_packet;
438         /* unicast tvlv packet */
439         batadv_rx_handler[BATADV_UNICAST_TVLV] = batadv_recv_unicast_tvlv;
440         /* batman icmp packet */
441         batadv_rx_handler[BATADV_ICMP] = batadv_recv_icmp_packet;
442         /* Fragmented packets */
443         batadv_rx_handler[BATADV_UNICAST_FRAG] = batadv_recv_frag_packet;
444 }
445
446 int
447 batadv_recv_handler_register(uint8_t packet_type,
448                              int (*recv_handler)(struct sk_buff *,
449                                                  struct batadv_hard_iface *))
450 {
451         int (*curr)(struct sk_buff *,
452                     struct batadv_hard_iface *);
453         curr = batadv_rx_handler[packet_type];
454
455         if ((curr != batadv_recv_unhandled_packet) &&
456             (curr != batadv_recv_unhandled_unicast_packet))
457                 return -EBUSY;
458
459         batadv_rx_handler[packet_type] = recv_handler;
460         return 0;
461 }
462
463 void batadv_recv_handler_unregister(uint8_t packet_type)
464 {
465         batadv_rx_handler[packet_type] = batadv_recv_unhandled_packet;
466 }
467
468 static struct batadv_algo_ops *batadv_algo_get(char *name)
469 {
470         struct batadv_algo_ops *bat_algo_ops = NULL, *bat_algo_ops_tmp;
471
472         hlist_for_each_entry(bat_algo_ops_tmp, &batadv_algo_list, list) {
473                 if (strcmp(bat_algo_ops_tmp->name, name) != 0)
474                         continue;
475
476                 bat_algo_ops = bat_algo_ops_tmp;
477                 break;
478         }
479
480         return bat_algo_ops;
481 }
482
483 int batadv_algo_register(struct batadv_algo_ops *bat_algo_ops)
484 {
485         struct batadv_algo_ops *bat_algo_ops_tmp;
486         int ret;
487
488         bat_algo_ops_tmp = batadv_algo_get(bat_algo_ops->name);
489         if (bat_algo_ops_tmp) {
490                 pr_info("Trying to register already registered routing algorithm: %s\n",
491                         bat_algo_ops->name);
492                 ret = -EEXIST;
493                 goto out;
494         }
495
496         /* all algorithms must implement all ops (for now) */
497         if (!bat_algo_ops->bat_iface_enable ||
498             !bat_algo_ops->bat_iface_disable ||
499             !bat_algo_ops->bat_iface_update_mac ||
500             !bat_algo_ops->bat_primary_iface_set ||
501             !bat_algo_ops->bat_ogm_schedule ||
502             !bat_algo_ops->bat_ogm_emit ||
503             !bat_algo_ops->bat_neigh_cmp ||
504             !bat_algo_ops->bat_neigh_is_equiv_or_better) {
505                 pr_info("Routing algo '%s' does not implement required ops\n",
506                         bat_algo_ops->name);
507                 ret = -EINVAL;
508                 goto out;
509         }
510
511         INIT_HLIST_NODE(&bat_algo_ops->list);
512         hlist_add_head(&bat_algo_ops->list, &batadv_algo_list);
513         ret = 0;
514
515 out:
516         return ret;
517 }
518
519 int batadv_algo_select(struct batadv_priv *bat_priv, char *name)
520 {
521         struct batadv_algo_ops *bat_algo_ops;
522         int ret = -EINVAL;
523
524         bat_algo_ops = batadv_algo_get(name);
525         if (!bat_algo_ops)
526                 goto out;
527
528         bat_priv->bat_algo_ops = bat_algo_ops;
529         ret = 0;
530
531 out:
532         return ret;
533 }
534
535 int batadv_algo_seq_print_text(struct seq_file *seq, void *offset)
536 {
537         struct batadv_algo_ops *bat_algo_ops;
538
539         seq_puts(seq, "Available routing algorithms:\n");
540
541         hlist_for_each_entry(bat_algo_ops, &batadv_algo_list, list) {
542                 seq_printf(seq, "%s\n", bat_algo_ops->name);
543         }
544
545         return 0;
546 }
547
548 /**
549  * batadv_skb_crc32 - calculate CRC32 of the whole packet and skip bytes in
550  *  the header
551  * @skb: skb pointing to fragmented socket buffers
552  * @payload_ptr: Pointer to position inside the head buffer of the skb
553  *  marking the start of the data to be CRC'ed
554  *
555  * payload_ptr must always point to an address in the skb head buffer and not to
556  * a fragment.
557  */
558 __be32 batadv_skb_crc32(struct sk_buff *skb, u8 *payload_ptr)
559 {
560         u32 crc = 0;
561         unsigned int from;
562         unsigned int to = skb->len;
563         struct skb_seq_state st;
564         const u8 *data;
565         unsigned int len;
566         unsigned int consumed = 0;
567
568         from = (unsigned int)(payload_ptr - skb->data);
569
570         skb_prepare_seq_read(skb, from, to, &st);
571         while ((len = skb_seq_read(consumed, &data, &st)) != 0) {
572                 crc = crc32c(crc, data, len);
573                 consumed += len;
574         }
575
576         return htonl(crc);
577 }
578
579 /**
580  * batadv_tvlv_handler_free_ref - decrement the tvlv handler refcounter and
581  *  possibly free it
582  * @tvlv_handler: the tvlv handler to free
583  */
584 static void
585 batadv_tvlv_handler_free_ref(struct batadv_tvlv_handler *tvlv_handler)
586 {
587         if (atomic_dec_and_test(&tvlv_handler->refcount))
588                 kfree_rcu(tvlv_handler, rcu);
589 }
590
591 /**
592  * batadv_tvlv_handler_get - retrieve tvlv handler from the tvlv handler list
593  *  based on the provided type and version (both need to match)
594  * @bat_priv: the bat priv with all the soft interface information
595  * @type: tvlv handler type to look for
596  * @version: tvlv handler version to look for
597  *
598  * Returns tvlv handler if found or NULL otherwise.
599  */
600 static struct batadv_tvlv_handler
601 *batadv_tvlv_handler_get(struct batadv_priv *bat_priv,
602                          uint8_t type, uint8_t version)
603 {
604         struct batadv_tvlv_handler *tvlv_handler_tmp, *tvlv_handler = NULL;
605
606         rcu_read_lock();
607         hlist_for_each_entry_rcu(tvlv_handler_tmp,
608                                  &bat_priv->tvlv.handler_list, list) {
609                 if (tvlv_handler_tmp->type != type)
610                         continue;
611
612                 if (tvlv_handler_tmp->version != version)
613                         continue;
614
615                 if (!atomic_inc_not_zero(&tvlv_handler_tmp->refcount))
616                         continue;
617
618                 tvlv_handler = tvlv_handler_tmp;
619                 break;
620         }
621         rcu_read_unlock();
622
623         return tvlv_handler;
624 }
625
626 /**
627  * batadv_tvlv_container_free_ref - decrement the tvlv container refcounter and
628  *  possibly free it
629  * @tvlv_handler: the tvlv container to free
630  */
631 static void batadv_tvlv_container_free_ref(struct batadv_tvlv_container *tvlv)
632 {
633         if (atomic_dec_and_test(&tvlv->refcount))
634                 kfree(tvlv);
635 }
636
637 /**
638  * batadv_tvlv_container_get - retrieve tvlv container from the tvlv container
639  *  list based on the provided type and version (both need to match)
640  * @bat_priv: the bat priv with all the soft interface information
641  * @type: tvlv container type to look for
642  * @version: tvlv container version to look for
643  *
644  * Has to be called with the appropriate locks being acquired
645  * (tvlv.container_list_lock).
646  *
647  * Returns tvlv container if found or NULL otherwise.
648  */
649 static struct batadv_tvlv_container
650 *batadv_tvlv_container_get(struct batadv_priv *bat_priv,
651                            uint8_t type, uint8_t version)
652 {
653         struct batadv_tvlv_container *tvlv_tmp, *tvlv = NULL;
654
655         hlist_for_each_entry(tvlv_tmp, &bat_priv->tvlv.container_list, list) {
656                 if (tvlv_tmp->tvlv_hdr.type != type)
657                         continue;
658
659                 if (tvlv_tmp->tvlv_hdr.version != version)
660                         continue;
661
662                 if (!atomic_inc_not_zero(&tvlv_tmp->refcount))
663                         continue;
664
665                 tvlv = tvlv_tmp;
666                 break;
667         }
668
669         return tvlv;
670 }
671
672 /**
673  * batadv_tvlv_container_list_size - calculate the size of the tvlv container
674  *  list entries
675  * @bat_priv: the bat priv with all the soft interface information
676  *
677  * Has to be called with the appropriate locks being acquired
678  * (tvlv.container_list_lock).
679  *
680  * Returns size of all currently registered tvlv containers in bytes.
681  */
682 static uint16_t batadv_tvlv_container_list_size(struct batadv_priv *bat_priv)
683 {
684         struct batadv_tvlv_container *tvlv;
685         uint16_t tvlv_len = 0;
686
687         hlist_for_each_entry(tvlv, &bat_priv->tvlv.container_list, list) {
688                 tvlv_len += sizeof(struct batadv_tvlv_hdr);
689                 tvlv_len += ntohs(tvlv->tvlv_hdr.len);
690         }
691
692         return tvlv_len;
693 }
694
695 /**
696  * batadv_tvlv_container_remove - remove tvlv container from the tvlv container
697  *  list
698  * @tvlv: the to be removed tvlv container
699  *
700  * Has to be called with the appropriate locks being acquired
701  * (tvlv.container_list_lock).
702  */
703 static void batadv_tvlv_container_remove(struct batadv_tvlv_container *tvlv)
704 {
705         if (!tvlv)
706                 return;
707
708         hlist_del(&tvlv->list);
709
710         /* first call to decrement the counter, second call to free */
711         batadv_tvlv_container_free_ref(tvlv);
712         batadv_tvlv_container_free_ref(tvlv);
713 }
714
715 /**
716  * batadv_tvlv_container_unregister - unregister tvlv container based on the
717  *  provided type and version (both need to match)
718  * @bat_priv: the bat priv with all the soft interface information
719  * @type: tvlv container type to unregister
720  * @version: tvlv container type to unregister
721  */
722 void batadv_tvlv_container_unregister(struct batadv_priv *bat_priv,
723                                       uint8_t type, uint8_t version)
724 {
725         struct batadv_tvlv_container *tvlv;
726
727         spin_lock_bh(&bat_priv->tvlv.container_list_lock);
728         tvlv = batadv_tvlv_container_get(bat_priv, type, version);
729         batadv_tvlv_container_remove(tvlv);
730         spin_unlock_bh(&bat_priv->tvlv.container_list_lock);
731 }
732
733 /**
734  * batadv_tvlv_container_register - register tvlv type, version and content
735  *  to be propagated with each (primary interface) OGM
736  * @bat_priv: the bat priv with all the soft interface information
737  * @type: tvlv container type
738  * @version: tvlv container version
739  * @tvlv_value: tvlv container content
740  * @tvlv_value_len: tvlv container content length
741  *
742  * If a container of the same type and version was already registered the new
743  * content is going to replace the old one.
744  */
745 void batadv_tvlv_container_register(struct batadv_priv *bat_priv,
746                                     uint8_t type, uint8_t version,
747                                     void *tvlv_value, uint16_t tvlv_value_len)
748 {
749         struct batadv_tvlv_container *tvlv_old, *tvlv_new;
750
751         if (!tvlv_value)
752                 tvlv_value_len = 0;
753
754         tvlv_new = kzalloc(sizeof(*tvlv_new) + tvlv_value_len, GFP_ATOMIC);
755         if (!tvlv_new)
756                 return;
757
758         tvlv_new->tvlv_hdr.version = version;
759         tvlv_new->tvlv_hdr.type = type;
760         tvlv_new->tvlv_hdr.len = htons(tvlv_value_len);
761
762         memcpy(tvlv_new + 1, tvlv_value, ntohs(tvlv_new->tvlv_hdr.len));
763         INIT_HLIST_NODE(&tvlv_new->list);
764         atomic_set(&tvlv_new->refcount, 1);
765
766         spin_lock_bh(&bat_priv->tvlv.container_list_lock);
767         tvlv_old = batadv_tvlv_container_get(bat_priv, type, version);
768         batadv_tvlv_container_remove(tvlv_old);
769         hlist_add_head(&tvlv_new->list, &bat_priv->tvlv.container_list);
770         spin_unlock_bh(&bat_priv->tvlv.container_list_lock);
771 }
772
773 /**
774  * batadv_tvlv_realloc_packet_buff - reallocate packet buffer to accomodate
775  *  requested packet size
776  * @packet_buff: packet buffer
777  * @packet_buff_len: packet buffer size
778  * @packet_min_len: requested packet minimum size
779  * @additional_packet_len: requested additional packet size on top of minimum
780  *  size
781  *
782  * Returns true of the packet buffer could be changed to the requested size,
783  * false otherwise.
784  */
785 static bool batadv_tvlv_realloc_packet_buff(unsigned char **packet_buff,
786                                             int *packet_buff_len,
787                                             int min_packet_len,
788                                             int additional_packet_len)
789 {
790         unsigned char *new_buff;
791
792         new_buff = kmalloc(min_packet_len + additional_packet_len, GFP_ATOMIC);
793
794         /* keep old buffer if kmalloc should fail */
795         if (new_buff) {
796                 memcpy(new_buff, *packet_buff, min_packet_len);
797                 kfree(*packet_buff);
798                 *packet_buff = new_buff;
799                 *packet_buff_len = min_packet_len + additional_packet_len;
800                 return true;
801         }
802
803         return false;
804 }
805
806 /**
807  * batadv_tvlv_container_ogm_append - append tvlv container content to given
808  *  OGM packet buffer
809  * @bat_priv: the bat priv with all the soft interface information
810  * @packet_buff: ogm packet buffer
811  * @packet_buff_len: ogm packet buffer size including ogm header and tvlv
812  *  content
813  * @packet_min_len: ogm header size to be preserved for the OGM itself
814  *
815  * The ogm packet might be enlarged or shrunk depending on the current size
816  * and the size of the to-be-appended tvlv containers.
817  *
818  * Returns size of all appended tvlv containers in bytes.
819  */
820 uint16_t batadv_tvlv_container_ogm_append(struct batadv_priv *bat_priv,
821                                           unsigned char **packet_buff,
822                                           int *packet_buff_len,
823                                           int packet_min_len)
824 {
825         struct batadv_tvlv_container *tvlv;
826         struct batadv_tvlv_hdr *tvlv_hdr;
827         uint16_t tvlv_value_len;
828         void *tvlv_value;
829         bool ret;
830
831         spin_lock_bh(&bat_priv->tvlv.container_list_lock);
832         tvlv_value_len = batadv_tvlv_container_list_size(bat_priv);
833
834         ret = batadv_tvlv_realloc_packet_buff(packet_buff, packet_buff_len,
835                                               packet_min_len, tvlv_value_len);
836
837         if (!ret)
838                 goto end;
839
840         if (!tvlv_value_len)
841                 goto end;
842
843         tvlv_value = (*packet_buff) + packet_min_len;
844
845         hlist_for_each_entry(tvlv, &bat_priv->tvlv.container_list, list) {
846                 tvlv_hdr = tvlv_value;
847                 tvlv_hdr->type = tvlv->tvlv_hdr.type;
848                 tvlv_hdr->version = tvlv->tvlv_hdr.version;
849                 tvlv_hdr->len = tvlv->tvlv_hdr.len;
850                 tvlv_value = tvlv_hdr + 1;
851                 memcpy(tvlv_value, tvlv + 1, ntohs(tvlv->tvlv_hdr.len));
852                 tvlv_value = (uint8_t *)tvlv_value + ntohs(tvlv->tvlv_hdr.len);
853         }
854
855 end:
856         spin_unlock_bh(&bat_priv->tvlv.container_list_lock);
857         return tvlv_value_len;
858 }
859
860 /**
861  * batadv_tvlv_call_handler - parse the given tvlv buffer to call the
862  *  appropriate handlers
863  * @bat_priv: the bat priv with all the soft interface information
864  * @tvlv_handler: tvlv callback function handling the tvlv content
865  * @ogm_source: flag indicating wether the tvlv is an ogm or a unicast packet
866  * @orig_node: orig node emitting the ogm packet
867  * @src: source mac address of the unicast packet
868  * @dst: destination mac address of the unicast packet
869  * @tvlv_value: tvlv content
870  * @tvlv_value_len: tvlv content length
871  *
872  * Returns success if handler was not found or the return value of the handler
873  * callback.
874  */
875 static int batadv_tvlv_call_handler(struct batadv_priv *bat_priv,
876                                     struct batadv_tvlv_handler *tvlv_handler,
877                                     bool ogm_source,
878                                     struct batadv_orig_node *orig_node,
879                                     uint8_t *src, uint8_t *dst,
880                                     void *tvlv_value, uint16_t tvlv_value_len)
881 {
882         if (!tvlv_handler)
883                 return NET_RX_SUCCESS;
884
885         if (ogm_source) {
886                 if (!tvlv_handler->ogm_handler)
887                         return NET_RX_SUCCESS;
888
889                 if (!orig_node)
890                         return NET_RX_SUCCESS;
891
892                 tvlv_handler->ogm_handler(bat_priv, orig_node,
893                                           BATADV_NO_FLAGS,
894                                           tvlv_value, tvlv_value_len);
895                 tvlv_handler->flags |= BATADV_TVLV_HANDLER_OGM_CALLED;
896         } else {
897                 if (!src)
898                         return NET_RX_SUCCESS;
899
900                 if (!dst)
901                         return NET_RX_SUCCESS;
902
903                 if (!tvlv_handler->unicast_handler)
904                         return NET_RX_SUCCESS;
905
906                 return tvlv_handler->unicast_handler(bat_priv, src,
907                                                      dst, tvlv_value,
908                                                      tvlv_value_len);
909         }
910
911         return NET_RX_SUCCESS;
912 }
913
914 /**
915  * batadv_tvlv_containers_process - parse the given tvlv buffer to call the
916  *  appropriate handlers
917  * @bat_priv: the bat priv with all the soft interface information
918  * @ogm_source: flag indicating wether the tvlv is an ogm or a unicast packet
919  * @orig_node: orig node emitting the ogm packet
920  * @src: source mac address of the unicast packet
921  * @dst: destination mac address of the unicast packet
922  * @tvlv_value: tvlv content
923  * @tvlv_value_len: tvlv content length
924  *
925  * Returns success when processing an OGM or the return value of all called
926  * handler callbacks.
927  */
928 int batadv_tvlv_containers_process(struct batadv_priv *bat_priv,
929                                    bool ogm_source,
930                                    struct batadv_orig_node *orig_node,
931                                    uint8_t *src, uint8_t *dst,
932                                    void *tvlv_value, uint16_t tvlv_value_len)
933 {
934         struct batadv_tvlv_handler *tvlv_handler;
935         struct batadv_tvlv_hdr *tvlv_hdr;
936         uint16_t tvlv_value_cont_len;
937         uint8_t cifnotfound = BATADV_TVLV_HANDLER_OGM_CIFNOTFND;
938         int ret = NET_RX_SUCCESS;
939
940         while (tvlv_value_len >= sizeof(*tvlv_hdr)) {
941                 tvlv_hdr = tvlv_value;
942                 tvlv_value_cont_len = ntohs(tvlv_hdr->len);
943                 tvlv_value = tvlv_hdr + 1;
944                 tvlv_value_len -= sizeof(*tvlv_hdr);
945
946                 if (tvlv_value_cont_len > tvlv_value_len)
947                         break;
948
949                 tvlv_handler = batadv_tvlv_handler_get(bat_priv,
950                                                        tvlv_hdr->type,
951                                                        tvlv_hdr->version);
952
953                 ret |= batadv_tvlv_call_handler(bat_priv, tvlv_handler,
954                                                 ogm_source, orig_node,
955                                                 src, dst, tvlv_value,
956                                                 tvlv_value_cont_len);
957                 if (tvlv_handler)
958                         batadv_tvlv_handler_free_ref(tvlv_handler);
959                 tvlv_value = (uint8_t *)tvlv_value + tvlv_value_cont_len;
960                 tvlv_value_len -= tvlv_value_cont_len;
961         }
962
963         if (!ogm_source)
964                 return ret;
965
966         rcu_read_lock();
967         hlist_for_each_entry_rcu(tvlv_handler,
968                                  &bat_priv->tvlv.handler_list, list) {
969                 if ((tvlv_handler->flags & BATADV_TVLV_HANDLER_OGM_CIFNOTFND) &&
970                     !(tvlv_handler->flags & BATADV_TVLV_HANDLER_OGM_CALLED))
971                         tvlv_handler->ogm_handler(bat_priv, orig_node,
972                                                   cifnotfound, NULL, 0);
973
974                 tvlv_handler->flags &= ~BATADV_TVLV_HANDLER_OGM_CALLED;
975         }
976         rcu_read_unlock();
977
978         return NET_RX_SUCCESS;
979 }
980
981 /**
982  * batadv_tvlv_ogm_receive - process an incoming ogm and call the appropriate
983  *  handlers
984  * @bat_priv: the bat priv with all the soft interface information
985  * @batadv_ogm_packet: ogm packet containing the tvlv containers
986  * @orig_node: orig node emitting the ogm packet
987  */
988 void batadv_tvlv_ogm_receive(struct batadv_priv *bat_priv,
989                              struct batadv_ogm_packet *batadv_ogm_packet,
990                              struct batadv_orig_node *orig_node)
991 {
992         void *tvlv_value;
993         uint16_t tvlv_value_len;
994
995         if (!batadv_ogm_packet)
996                 return;
997
998         tvlv_value_len = ntohs(batadv_ogm_packet->tvlv_len);
999         if (!tvlv_value_len)
1000                 return;
1001
1002         tvlv_value = batadv_ogm_packet + 1;
1003
1004         batadv_tvlv_containers_process(bat_priv, true, orig_node, NULL, NULL,
1005                                        tvlv_value, tvlv_value_len);
1006 }
1007
1008 /**
1009  * batadv_tvlv_handler_register - register tvlv handler based on the provided
1010  *  type and version (both need to match) for ogm tvlv payload and/or unicast
1011  *  payload
1012  * @bat_priv: the bat priv with all the soft interface information
1013  * @optr: ogm tvlv handler callback function. This function receives the orig
1014  *  node, flags and the tvlv content as argument to process.
1015  * @uptr: unicast tvlv handler callback function. This function receives the
1016  *  source & destination of the unicast packet as well as the tvlv content
1017  *  to process.
1018  * @type: tvlv handler type to be registered
1019  * @version: tvlv handler version to be registered
1020  * @flags: flags to enable or disable TVLV API behavior
1021  */
1022 void batadv_tvlv_handler_register(struct batadv_priv *bat_priv,
1023                                   void (*optr)(struct batadv_priv *bat_priv,
1024                                                struct batadv_orig_node *orig,
1025                                                uint8_t flags,
1026                                                void *tvlv_value,
1027                                                uint16_t tvlv_value_len),
1028                                   int (*uptr)(struct batadv_priv *bat_priv,
1029                                               uint8_t *src, uint8_t *dst,
1030                                               void *tvlv_value,
1031                                               uint16_t tvlv_value_len),
1032                                   uint8_t type, uint8_t version, uint8_t flags)
1033 {
1034         struct batadv_tvlv_handler *tvlv_handler;
1035
1036         tvlv_handler = batadv_tvlv_handler_get(bat_priv, type, version);
1037         if (tvlv_handler) {
1038                 batadv_tvlv_handler_free_ref(tvlv_handler);
1039                 return;
1040         }
1041
1042         tvlv_handler = kzalloc(sizeof(*tvlv_handler), GFP_ATOMIC);
1043         if (!tvlv_handler)
1044                 return;
1045
1046         tvlv_handler->ogm_handler = optr;
1047         tvlv_handler->unicast_handler = uptr;
1048         tvlv_handler->type = type;
1049         tvlv_handler->version = version;
1050         tvlv_handler->flags = flags;
1051         atomic_set(&tvlv_handler->refcount, 1);
1052         INIT_HLIST_NODE(&tvlv_handler->list);
1053
1054         spin_lock_bh(&bat_priv->tvlv.handler_list_lock);
1055         hlist_add_head_rcu(&tvlv_handler->list, &bat_priv->tvlv.handler_list);
1056         spin_unlock_bh(&bat_priv->tvlv.handler_list_lock);
1057 }
1058
1059 /**
1060  * batadv_tvlv_handler_unregister - unregister tvlv handler based on the
1061  *  provided type and version (both need to match)
1062  * @bat_priv: the bat priv with all the soft interface information
1063  * @type: tvlv handler type to be unregistered
1064  * @version: tvlv handler version to be unregistered
1065  */
1066 void batadv_tvlv_handler_unregister(struct batadv_priv *bat_priv,
1067                                     uint8_t type, uint8_t version)
1068 {
1069         struct batadv_tvlv_handler *tvlv_handler;
1070
1071         tvlv_handler = batadv_tvlv_handler_get(bat_priv, type, version);
1072         if (!tvlv_handler)
1073                 return;
1074
1075         batadv_tvlv_handler_free_ref(tvlv_handler);
1076         spin_lock_bh(&bat_priv->tvlv.handler_list_lock);
1077         hlist_del_rcu(&tvlv_handler->list);
1078         spin_unlock_bh(&bat_priv->tvlv.handler_list_lock);
1079         batadv_tvlv_handler_free_ref(tvlv_handler);
1080 }
1081
1082 /**
1083  * batadv_tvlv_unicast_send - send a unicast packet with tvlv payload to the
1084  *  specified host
1085  * @bat_priv: the bat priv with all the soft interface information
1086  * @src: source mac address of the unicast packet
1087  * @dst: destination mac address of the unicast packet
1088  * @type: tvlv type
1089  * @version: tvlv version
1090  * @tvlv_value: tvlv content
1091  * @tvlv_value_len: tvlv content length
1092  */
1093 void batadv_tvlv_unicast_send(struct batadv_priv *bat_priv, uint8_t *src,
1094                               uint8_t *dst, uint8_t type, uint8_t version,
1095                               void *tvlv_value, uint16_t tvlv_value_len)
1096 {
1097         struct batadv_unicast_tvlv_packet *unicast_tvlv_packet;
1098         struct batadv_tvlv_hdr *tvlv_hdr;
1099         struct batadv_orig_node *orig_node;
1100         struct sk_buff *skb = NULL;
1101         unsigned char *tvlv_buff;
1102         unsigned int tvlv_len;
1103         ssize_t hdr_len = sizeof(*unicast_tvlv_packet);
1104         bool ret = false;
1105
1106         orig_node = batadv_orig_hash_find(bat_priv, dst);
1107         if (!orig_node)
1108                 goto out;
1109
1110         tvlv_len = sizeof(*tvlv_hdr) + tvlv_value_len;
1111
1112         skb = netdev_alloc_skb_ip_align(NULL, ETH_HLEN + hdr_len + tvlv_len);
1113         if (!skb)
1114                 goto out;
1115
1116         skb->priority = TC_PRIO_CONTROL;
1117         skb_reserve(skb, ETH_HLEN);
1118         tvlv_buff = skb_put(skb, sizeof(*unicast_tvlv_packet) + tvlv_len);
1119         unicast_tvlv_packet = (struct batadv_unicast_tvlv_packet *)tvlv_buff;
1120         unicast_tvlv_packet->packet_type = BATADV_UNICAST_TVLV;
1121         unicast_tvlv_packet->version = BATADV_COMPAT_VERSION;
1122         unicast_tvlv_packet->ttl = BATADV_TTL;
1123         unicast_tvlv_packet->reserved = 0;
1124         unicast_tvlv_packet->tvlv_len = htons(tvlv_len);
1125         unicast_tvlv_packet->align = 0;
1126         memcpy(unicast_tvlv_packet->src, src, ETH_ALEN);
1127         memcpy(unicast_tvlv_packet->dst, dst, ETH_ALEN);
1128
1129         tvlv_buff = (unsigned char *)(unicast_tvlv_packet + 1);
1130         tvlv_hdr = (struct batadv_tvlv_hdr *)tvlv_buff;
1131         tvlv_hdr->version = version;
1132         tvlv_hdr->type = type;
1133         tvlv_hdr->len = htons(tvlv_value_len);
1134         tvlv_buff += sizeof(*tvlv_hdr);
1135         memcpy(tvlv_buff, tvlv_value, tvlv_value_len);
1136
1137         if (batadv_send_skb_to_orig(skb, orig_node, NULL) != NET_XMIT_DROP)
1138                 ret = true;
1139
1140 out:
1141         if (skb && !ret)
1142                 kfree_skb(skb);
1143         if (orig_node)
1144                 batadv_orig_node_free_ref(orig_node);
1145 }
1146
1147 /**
1148  * batadv_get_vid - extract the VLAN identifier from skb if any
1149  * @skb: the buffer containing the packet
1150  * @header_len: length of the batman header preceding the ethernet header
1151  *
1152  * If the packet embedded in the skb is vlan tagged this function returns the
1153  * VID with the BATADV_VLAN_HAS_TAG flag. Otherwise BATADV_NO_FLAGS is returned.
1154  */
1155 unsigned short batadv_get_vid(struct sk_buff *skb, size_t header_len)
1156 {
1157         struct ethhdr *ethhdr = (struct ethhdr *)(skb->data + header_len);
1158         struct vlan_ethhdr *vhdr;
1159         unsigned short vid;
1160
1161         if (ethhdr->h_proto != htons(ETH_P_8021Q))
1162                 return BATADV_NO_FLAGS;
1163
1164         if (!pskb_may_pull(skb, header_len + VLAN_ETH_HLEN))
1165                 return BATADV_NO_FLAGS;
1166
1167         vhdr = (struct vlan_ethhdr *)(skb->data + header_len);
1168         vid = ntohs(vhdr->h_vlan_TCI) & VLAN_VID_MASK;
1169         vid |= BATADV_VLAN_HAS_TAG;
1170
1171         return vid;
1172 }
1173
1174 static int batadv_param_set_ra(const char *val, const struct kernel_param *kp)
1175 {
1176         struct batadv_algo_ops *bat_algo_ops;
1177         char *algo_name = (char *)val;
1178         size_t name_len = strlen(algo_name);
1179
1180         if (name_len > 0 && algo_name[name_len - 1] == '\n')
1181                 algo_name[name_len - 1] = '\0';
1182
1183         bat_algo_ops = batadv_algo_get(algo_name);
1184         if (!bat_algo_ops) {
1185                 pr_err("Routing algorithm '%s' is not supported\n", algo_name);
1186                 return -EINVAL;
1187         }
1188
1189         return param_set_copystring(algo_name, kp);
1190 }
1191
1192 static const struct kernel_param_ops batadv_param_ops_ra = {
1193         .set = batadv_param_set_ra,
1194         .get = param_get_string,
1195 };
1196
1197 static struct kparam_string batadv_param_string_ra = {
1198         .maxlen = sizeof(batadv_routing_algo),
1199         .string = batadv_routing_algo,
1200 };
1201
1202 module_param_cb(routing_algo, &batadv_param_ops_ra, &batadv_param_string_ra,
1203                 0644);
1204 module_init(batadv_init);
1205 module_exit(batadv_exit);
1206
1207 MODULE_LICENSE("GPL");
1208
1209 MODULE_AUTHOR(BATADV_DRIVER_AUTHOR);
1210 MODULE_DESCRIPTION(BATADV_DRIVER_DESC);
1211 MODULE_SUPPORTED_DEVICE(BATADV_DRIVER_DEVICE);
1212 MODULE_VERSION(BATADV_SOURCE_VERSION);