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