translation-table.c 63.8 KB
Newer Older
1
/* Copyright (C) 2007-2012 B.A.T.M.A.N. contributors:
2
 *
3
 * Marek Lindner, Simon Wunderlich, Antonio Quartulli
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of version 2 of the GNU General Public
 * License as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301, USA
 */

#include "main.h"
#include "translation-table.h"
#include "soft-interface.h"
23
#include "hard-interface.h"
24
#include "send.h"
25 26
#include "hash.h"
#include "originator.h"
27
#include "routing.h"
28
#include "bridge_loop_avoidance.h"
29

30 31
#include <linux/crc16.h>

32 33 34 35 36
static void batadv_send_roam_adv(struct bat_priv *bat_priv, uint8_t *client,
				 struct orig_node *orig_node);
static void batadv_tt_purge(struct work_struct *work);
static void
batadv_tt_global_del_orig_list(struct tt_global_entry *tt_global_entry);
37

38
/* returns 1 if they are the same mac addr */
39
static int batadv_compare_tt(const struct hlist_node *node, const void *data2)
40
{
41
	const void *data1 = container_of(node, struct tt_common_entry,
42
					 hash_entry);
43 44 45 46

	return (memcmp(data1, data2, ETH_ALEN) == 0 ? 1 : 0);
}

47
static void batadv_tt_start_timer(struct bat_priv *bat_priv)
48
{
49
	INIT_DELAYED_WORK(&bat_priv->tt_work, batadv_tt_purge);
50
	queue_delayed_work(batadv_event_workqueue, &bat_priv->tt_work,
51
			   msecs_to_jiffies(5000));
52 53
}

54 55
static struct tt_common_entry *batadv_tt_hash_find(struct hashtable_t *hash,
						   const void *data)
56 57 58
{
	struct hlist_head *head;
	struct hlist_node *node;
59
	struct tt_common_entry *tt_common_entry, *tt_common_entry_tmp = NULL;
60
	uint32_t index;
61 62 63 64

	if (!hash)
		return NULL;

65
	index = batadv_choose_orig(data, hash->size);
66 67 68
	head = &hash->table[index];

	rcu_read_lock();
69
	hlist_for_each_entry_rcu(tt_common_entry, node, head, hash_entry) {
70
		if (!batadv_compare_eth(tt_common_entry, data))
71 72
			continue;

73
		if (!atomic_inc_not_zero(&tt_common_entry->refcount))
74 75
			continue;

76
		tt_common_entry_tmp = tt_common_entry;
77 78 79 80
		break;
	}
	rcu_read_unlock();

81
	return tt_common_entry_tmp;
82 83
}

84 85
static struct tt_local_entry *
batadv_tt_local_hash_find(struct bat_priv *bat_priv, const void *data)
86
{
87 88
	struct tt_common_entry *tt_common_entry;
	struct tt_local_entry *tt_local_entry = NULL;
89

90
	tt_common_entry = batadv_tt_hash_find(bat_priv->tt_local_hash, data);
91 92 93 94 95
	if (tt_common_entry)
		tt_local_entry = container_of(tt_common_entry,
					      struct tt_local_entry, common);
	return tt_local_entry;
}
96

97 98
static struct tt_global_entry *
batadv_tt_global_hash_find(struct bat_priv *bat_priv, const void *data)
99 100 101
{
	struct tt_common_entry *tt_common_entry;
	struct tt_global_entry *tt_global_entry = NULL;
102

103
	tt_common_entry = batadv_tt_hash_find(bat_priv->tt_global_hash, data);
104 105 106 107
	if (tt_common_entry)
		tt_global_entry = container_of(tt_common_entry,
					       struct tt_global_entry, common);
	return tt_global_entry;
108 109 110

}

111 112
static void
batadv_tt_local_entry_free_ref(struct tt_local_entry *tt_local_entry)
113
{
114 115
	if (atomic_dec_and_test(&tt_local_entry->common.refcount))
		kfree_rcu(tt_local_entry, common.rcu);
116 117
}

118
static void batadv_tt_global_entry_free_rcu(struct rcu_head *rcu)
119
{
120
	struct tt_common_entry *tt_common_entry;
121 122
	struct tt_global_entry *tt_global_entry;

123 124 125
	tt_common_entry = container_of(rcu, struct tt_common_entry, rcu);
	tt_global_entry = container_of(tt_common_entry, struct tt_global_entry,
				       common);
126 127 128 129

	kfree(tt_global_entry);
}

130 131
static void
batadv_tt_global_entry_free_ref(struct tt_global_entry *tt_global_entry)
132
{
133
	if (atomic_dec_and_test(&tt_global_entry->common.refcount)) {
134
		batadv_tt_global_del_orig_list(tt_global_entry);
135
		call_rcu(&tt_global_entry->common.rcu,
136
			 batadv_tt_global_entry_free_rcu);
137 138 139
	}
}

140
static void batadv_tt_orig_list_entry_free_rcu(struct rcu_head *rcu)
141 142 143 144
{
	struct tt_orig_list_entry *orig_entry;

	orig_entry = container_of(rcu, struct tt_orig_list_entry, rcu);
145
	batadv_orig_node_free_ref(orig_entry->orig_node);
146 147 148
	kfree(orig_entry);
}

149 150
static void
batadv_tt_orig_list_entry_free_ref(struct tt_orig_list_entry *orig_entry)
151
{
152 153
	/* to avoid race conditions, immediately decrease the tt counter */
	atomic_dec(&orig_entry->orig_node->tt_size);
154
	call_rcu(&orig_entry->rcu, batadv_tt_orig_list_entry_free_rcu);
155 156
}

157 158
static void batadv_tt_local_event(struct bat_priv *bat_priv,
				  const uint8_t *addr, uint8_t flags)
159
{
160 161 162
	struct tt_change_node *tt_change_node, *entry, *safe;
	bool event_removed = false;
	bool del_op_requested, del_op_entry;
163 164 165 166 167 168

	tt_change_node = kmalloc(sizeof(*tt_change_node), GFP_ATOMIC);

	if (!tt_change_node)
		return;

169
	tt_change_node->change.flags = flags;
170 171
	memcpy(tt_change_node->change.addr, addr, ETH_ALEN);

172 173 174
	del_op_requested = flags & TT_CLIENT_DEL;

	/* check for ADD+DEL or DEL+ADD events */
175
	spin_lock_bh(&bat_priv->tt_changes_list_lock);
176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200
	list_for_each_entry_safe(entry, safe, &bat_priv->tt_changes_list,
				 list) {
		if (!batadv_compare_eth(entry->change.addr, addr))
			continue;

		/* DEL+ADD in the same orig interval have no effect and can be
		 * removed to avoid silly behaviour on the receiver side. The
		 * other way around (ADD+DEL) can happen in case of roaming of
		 * a client still in the NEW state. Roaming of NEW clients is
		 * now possible due to automatically recognition of "temporary"
		 * clients
		 */
		del_op_entry = entry->change.flags & TT_CLIENT_DEL;
		if (!del_op_requested && del_op_entry)
			goto del;
		if (del_op_requested && !del_op_entry)
			goto del;
		continue;
del:
		list_del(&entry->list);
		kfree(entry);
		event_removed = true;
		goto unlock;
	}

201 202
	/* track the change in the OGMinterval list */
	list_add_tail(&tt_change_node->list, &bat_priv->tt_changes_list);
203 204

unlock:
205 206
	spin_unlock_bh(&bat_priv->tt_changes_list_lock);

207 208 209 210
	if (event_removed)
		atomic_dec(&bat_priv->tt_local_changes);
	else
		atomic_inc(&bat_priv->tt_local_changes);
211 212
}

213
int batadv_tt_len(int changes_num)
214 215 216 217
{
	return changes_num * sizeof(struct tt_change);
}

218
static int batadv_tt_local_init(struct bat_priv *bat_priv)
219
{
220
	if (bat_priv->tt_local_hash)
221
		return 0;
222

223
	bat_priv->tt_local_hash = batadv_hash_new(1024);
224

225
	if (!bat_priv->tt_local_hash)
226
		return -ENOMEM;
227

228
	return 0;
229 230
}

231 232
void batadv_tt_local_add(struct net_device *soft_iface, const uint8_t *addr,
			 int ifindex)
233 234
{
	struct bat_priv *bat_priv = netdev_priv(soft_iface);
235 236
	struct tt_local_entry *tt_local_entry = NULL;
	struct tt_global_entry *tt_global_entry = NULL;
237 238 239
	struct hlist_head *head;
	struct hlist_node *node;
	struct tt_orig_list_entry *orig_entry;
240
	int hash_added;
241

242
	tt_local_entry = batadv_tt_local_hash_find(bat_priv, addr);
243

244 245
	if (tt_local_entry) {
		tt_local_entry->last_seen = jiffies;
246 247
		/* possibly unset the TT_CLIENT_PENDING flag */
		tt_local_entry->common.flags &= ~TT_CLIENT_PENDING;
248
		goto out;
249 250
	}

251
	tt_local_entry = kmalloc(sizeof(*tt_local_entry), GFP_ATOMIC);
252
	if (!tt_local_entry)
253
		goto out;
254

255 256 257
	batadv_dbg(DBG_TT, bat_priv,
		   "Creating new local tt entry: %pM (ttvn: %d)\n", addr,
		   (uint8_t)atomic_read(&bat_priv->ttvn));
258

259 260
	memcpy(tt_local_entry->common.addr, addr, ETH_ALEN);
	tt_local_entry->common.flags = NO_FLAGS;
261
	if (batadv_is_wifi_iface(ifindex))
262 263 264
		tt_local_entry->common.flags |= TT_CLIENT_WIFI;
	atomic_set(&tt_local_entry->common.refcount, 2);
	tt_local_entry->last_seen = jiffies;
265 266

	/* the batman interface mac address should never be purged */
267
	if (batadv_compare_eth(addr, soft_iface->dev_addr))
268
		tt_local_entry->common.flags |= TT_CLIENT_NOPURGE;
269

270 271
	/* The local entry has to be marked as NEW to avoid to send it in
	 * a full table response going out before the next ttvn increment
272 273
	 * (consistency check)
	 */
274 275
	tt_local_entry->common.flags |= TT_CLIENT_NEW;

276
	hash_added = batadv_hash_add(bat_priv->tt_local_hash, batadv_compare_tt,
277 278
				     batadv_choose_orig,
				     &tt_local_entry->common,
279
				     &tt_local_entry->common.hash_entry);
280 281 282

	if (unlikely(hash_added != 0)) {
		/* remove the reference for the hash */
283
		batadv_tt_local_entry_free_ref(tt_local_entry);
284 285 286
		goto out;
	}

287
	batadv_tt_local_event(bat_priv, addr, tt_local_entry->common.flags);
288

289
	/* remove address from global hash if present */
290
	tt_global_entry = batadv_tt_global_hash_find(bat_priv, addr);
291

292 293
	/* Check whether it is a roaming! */
	if (tt_global_entry) {
294 295 296 297 298 299
		/* These node are probably going to update their tt table */
		head = &tt_global_entry->orig_list;
		rcu_read_lock();
		hlist_for_each_entry_rcu(orig_entry, node, head, list) {
			orig_entry->orig_node->tt_poss_change = true;

300 301 302
			batadv_send_roam_adv(bat_priv,
					     tt_global_entry->common.addr,
					     orig_entry->orig_node);
303 304 305 306 307
		}
		rcu_read_unlock();
		/* The global entry has to be marked as ROAMING and
		 * has to be kept for consistency purpose
		 */
D
David S. Miller 已提交
308
		tt_global_entry->common.flags |= TT_CLIENT_ROAM;
309
		tt_global_entry->roam_at = jiffies;
310 311 312
	}
out:
	if (tt_local_entry)
313
		batadv_tt_local_entry_free_ref(tt_local_entry);
314
	if (tt_global_entry)
315
		batadv_tt_global_entry_free_ref(tt_global_entry);
316 317
}

318 319 320 321
static void batadv_tt_realloc_packet_buff(unsigned char **packet_buff,
					  int *packet_buff_len,
					  int min_packet_len,
					  int new_packet_len)
322 323 324 325 326 327 328 329 330 331 332 333 334 335
{
	unsigned char *new_buff;

	new_buff = kmalloc(new_packet_len, GFP_ATOMIC);

	/* keep old buffer if kmalloc should fail */
	if (new_buff) {
		memcpy(new_buff, *packet_buff, min_packet_len);
		kfree(*packet_buff);
		*packet_buff = new_buff;
		*packet_buff_len = new_packet_len;
	}
}

336 337 338 339
static void batadv_tt_prepare_packet_buff(struct bat_priv *bat_priv,
					  unsigned char **packet_buff,
					  int *packet_buff_len,
					  int min_packet_len)
340 341 342 343
{
	struct hard_iface *primary_if;
	int req_len;

344
	primary_if = batadv_primary_if_get_selected(bat_priv);
345 346

	req_len = min_packet_len;
347
	req_len += batadv_tt_len(atomic_read(&bat_priv->tt_local_changes));
348 349 350 351 352 353 354

	/* if we have too many changes for one packet don't send any
	 * and wait for the tt table request which will be fragmented
	 */
	if ((!primary_if) || (req_len > primary_if->soft_iface->mtu))
		req_len = min_packet_len;

355 356
	batadv_tt_realloc_packet_buff(packet_buff, packet_buff_len,
				      min_packet_len, req_len);
357 358

	if (primary_if)
359
		batadv_hardif_free_ref(primary_if);
360 361
}

362 363 364 365
static int batadv_tt_changes_fill_buff(struct bat_priv *bat_priv,
				       unsigned char **packet_buff,
				       int *packet_buff_len,
				       int min_packet_len)
366
{
367
	struct tt_change_node *entry, *safe;
368 369 370
	int count = 0, tot_changes = 0, new_len;
	unsigned char *tt_buff;

371 372
	batadv_tt_prepare_packet_buff(bat_priv, packet_buff,
				      packet_buff_len, min_packet_len);
373

374 375 376 377
	new_len = *packet_buff_len - min_packet_len;
	tt_buff = *packet_buff + min_packet_len;

	if (new_len > 0)
378
		tot_changes = new_len / batadv_tt_len(1);
379

380 381
	spin_lock_bh(&bat_priv->tt_changes_list_lock);
	atomic_set(&bat_priv->tt_local_changes, 0);
382

383
	list_for_each_entry_safe(entry, safe, &bat_priv->tt_changes_list,
384
				 list) {
385
		if (count < tot_changes) {
386
			memcpy(tt_buff + batadv_tt_len(count),
387
			       &entry->change, sizeof(struct tt_change));
388 389
			count++;
		}
390 391
		list_del(&entry->list);
		kfree(entry);
392
	}
393 394 395 396 397 398 399
	spin_unlock_bh(&bat_priv->tt_changes_list_lock);

	/* Keep the buffer for possible tt_request */
	spin_lock_bh(&bat_priv->tt_buff_lock);
	kfree(bat_priv->tt_buff);
	bat_priv->tt_buff_len = 0;
	bat_priv->tt_buff = NULL;
400 401 402
	/* check whether this new OGM has no changes due to size problems */
	if (new_len > 0) {
		/* if kmalloc() fails we will reply with the full table
403 404
		 * instead of providing the diff
		 */
405
		bat_priv->tt_buff = kmalloc(new_len, GFP_ATOMIC);
406
		if (bat_priv->tt_buff) {
407 408
			memcpy(bat_priv->tt_buff, tt_buff, new_len);
			bat_priv->tt_buff_len = new_len;
409 410 411
		}
	}
	spin_unlock_bh(&bat_priv->tt_buff_lock);
412

413
	return count;
414 415
}

416
int batadv_tt_local_seq_print_text(struct seq_file *seq, void *offset)
417 418 419
{
	struct net_device *net_dev = (struct net_device *)seq->private;
	struct bat_priv *bat_priv = netdev_priv(net_dev);
420
	struct hashtable_t *hash = bat_priv->tt_local_hash;
421
	struct tt_common_entry *tt_common_entry;
422
	struct hard_iface *primary_if;
423
	struct hlist_node *node;
424
	struct hlist_head *head;
425 426
	uint32_t i;
	int ret = 0;
427

428
	primary_if = batadv_primary_if_get_selected(bat_priv);
429
	if (!primary_if) {
430 431
		ret = seq_printf(seq,
				 "BATMAN mesh %s disabled - please specify interfaces to enable it\n",
432 433 434
				 net_dev->name);
		goto out;
	}
435

436
	if (primary_if->if_status != IF_ACTIVE) {
437 438
		ret = seq_printf(seq,
				 "BATMAN mesh %s disabled - primary interface not active\n",
439 440
				 net_dev->name);
		goto out;
441 442
	}

443 444
	seq_printf(seq,
		   "Locally retrieved addresses (from %s) announced via TT (TTVN: %u):\n",
445
		   net_dev->name, (uint8_t)atomic_read(&bat_priv->ttvn));
446 447 448 449

	for (i = 0; i < hash->size; i++) {
		head = &hash->table[i];

450
		rcu_read_lock();
451
		hlist_for_each_entry_rcu(tt_common_entry, node,
452
					 head, hash_entry) {
453
			seq_printf(seq, " * %pM [%c%c%c%c%c]\n",
454 455 456 457 458 459 460 461 462 463 464
				   tt_common_entry->addr,
				   (tt_common_entry->flags &
				    TT_CLIENT_ROAM ? 'R' : '.'),
				   (tt_common_entry->flags &
				    TT_CLIENT_NOPURGE ? 'P' : '.'),
				   (tt_common_entry->flags &
				    TT_CLIENT_NEW ? 'N' : '.'),
				   (tt_common_entry->flags &
				    TT_CLIENT_PENDING ? 'X' : '.'),
				   (tt_common_entry->flags &
				    TT_CLIENT_WIFI ? 'W' : '.'));
465
		}
466
		rcu_read_unlock();
467
	}
468 469
out:
	if (primary_if)
470
		batadv_hardif_free_ref(primary_if);
471
	return ret;
472 473
}

474 475 476
static void batadv_tt_local_set_pending(struct bat_priv *bat_priv,
					struct tt_local_entry *tt_local_entry,
					uint16_t flags, const char *message)
477
{
478 479
	batadv_tt_local_event(bat_priv, tt_local_entry->common.addr,
			      tt_local_entry->common.flags | flags);
480

481 482
	/* The local client has to be marked as "pending to be removed" but has
	 * to be kept in the table in order to send it in a full table
483 484
	 * response issued before the net ttvn increment (consistency check)
	 */
485
	tt_local_entry->common.flags |= TT_CLIENT_PENDING;
486

487 488 489
	batadv_dbg(DBG_TT, bat_priv,
		   "Local tt entry (%pM) pending to be removed: %s\n",
		   tt_local_entry->common.addr, message);
490 491
}

492 493
void batadv_tt_local_remove(struct bat_priv *bat_priv, const uint8_t *addr,
			    const char *message, bool roaming)
494
{
495
	struct tt_local_entry *tt_local_entry = NULL;
496

497
	tt_local_entry = batadv_tt_local_hash_find(bat_priv, addr);
498 499 500
	if (!tt_local_entry)
		goto out;

501 502 503
	batadv_tt_local_set_pending(bat_priv, tt_local_entry, TT_CLIENT_DEL |
				    (roaming ? TT_CLIENT_ROAM : NO_FLAGS),
				    message);
504 505
out:
	if (tt_local_entry)
506
		batadv_tt_local_entry_free_ref(tt_local_entry);
507 508
}

509
static void batadv_tt_local_purge(struct bat_priv *bat_priv)
510
{
511 512
	struct hashtable_t *hash = bat_priv->tt_local_hash;
	struct tt_local_entry *tt_local_entry;
513
	struct tt_common_entry *tt_common_entry;
514
	struct hlist_node *node, *node_tmp;
515
	struct hlist_head *head;
516
	spinlock_t *list_lock; /* protects write access to the hash lists */
517
	uint32_t i;
518 519 520

	for (i = 0; i < hash->size; i++) {
		head = &hash->table[i];
521
		list_lock = &hash->list_locks[i];
522

523
		spin_lock_bh(list_lock);
524
		hlist_for_each_entry_safe(tt_common_entry, node, node_tmp,
525
					  head, hash_entry) {
526 527 528 529
			tt_local_entry = container_of(tt_common_entry,
						      struct tt_local_entry,
						      common);
			if (tt_local_entry->common.flags & TT_CLIENT_NOPURGE)
530
				continue;
531

532
			/* entry already marked for deletion */
533
			if (tt_local_entry->common.flags & TT_CLIENT_PENDING)
534 535
				continue;

536 537
			if (!batadv_has_timed_out(tt_local_entry->last_seen,
						  TT_LOCAL_TIMEOUT))
538 539
				continue;

540 541
			batadv_tt_local_set_pending(bat_priv, tt_local_entry,
						    TT_CLIENT_DEL, "timed out");
542
		}
543
		spin_unlock_bh(list_lock);
544 545 546 547
	}

}

548
static void batadv_tt_local_table_free(struct bat_priv *bat_priv)
549
{
550 551
	struct hashtable_t *hash;
	spinlock_t *list_lock; /* protects write access to the hash lists */
552
	struct tt_common_entry *tt_common_entry;
553
	struct tt_local_entry *tt_local_entry;
554 555
	struct hlist_node *node, *node_tmp;
	struct hlist_head *head;
556
	uint32_t i;
557

558
	if (!bat_priv->tt_local_hash)
559 560
		return;

561 562 563 564 565 566 567
	hash = bat_priv->tt_local_hash;

	for (i = 0; i < hash->size; i++) {
		head = &hash->table[i];
		list_lock = &hash->list_locks[i];

		spin_lock_bh(list_lock);
568
		hlist_for_each_entry_safe(tt_common_entry, node, node_tmp,
569 570
					  head, hash_entry) {
			hlist_del_rcu(node);
571 572 573
			tt_local_entry = container_of(tt_common_entry,
						      struct tt_local_entry,
						      common);
574
			batadv_tt_local_entry_free_ref(tt_local_entry);
575 576 577 578
		}
		spin_unlock_bh(list_lock);
	}

579
	batadv_hash_destroy(hash);
580

581
	bat_priv->tt_local_hash = NULL;
582 583
}

584
static int batadv_tt_global_init(struct bat_priv *bat_priv)
585
{
586
	if (bat_priv->tt_global_hash)
587
		return 0;
588

589
	bat_priv->tt_global_hash = batadv_hash_new(1024);
590

591
	if (!bat_priv->tt_global_hash)
592
		return -ENOMEM;
593

594
	return 0;
595 596
}

597
static void batadv_tt_changes_list_free(struct bat_priv *bat_priv)
598
{
599
	struct tt_change_node *entry, *safe;
600

601
	spin_lock_bh(&bat_priv->tt_changes_list_lock);
602

603 604 605 606 607
	list_for_each_entry_safe(entry, safe, &bat_priv->tt_changes_list,
				 list) {
		list_del(&entry->list);
		kfree(entry);
	}
608

609 610 611
	atomic_set(&bat_priv->tt_local_changes, 0);
	spin_unlock_bh(&bat_priv->tt_changes_list_lock);
}
612

613 614 615
/* find out if an orig_node is already in the list of a tt_global_entry.
 * returns 1 if found, 0 otherwise
 */
616 617
static bool batadv_tt_global_entry_has_orig(const struct tt_global_entry *entry,
					    const struct orig_node *orig_node)
618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635
{
	struct tt_orig_list_entry *tmp_orig_entry;
	const struct hlist_head *head;
	struct hlist_node *node;
	bool found = false;

	rcu_read_lock();
	head = &entry->orig_list;
	hlist_for_each_entry_rcu(tmp_orig_entry, node, head, list) {
		if (tmp_orig_entry->orig_node == orig_node) {
			found = true;
			break;
		}
	}
	rcu_read_unlock();
	return found;
}

636 637 638
static void
batadv_tt_global_add_orig_entry(struct tt_global_entry *tt_global_entry,
				struct orig_node *orig_node, int ttvn)
639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657
{
	struct tt_orig_list_entry *orig_entry;

	orig_entry = kzalloc(sizeof(*orig_entry), GFP_ATOMIC);
	if (!orig_entry)
		return;

	INIT_HLIST_NODE(&orig_entry->list);
	atomic_inc(&orig_node->refcount);
	atomic_inc(&orig_node->tt_size);
	orig_entry->orig_node = orig_node;
	orig_entry->ttvn = ttvn;

	spin_lock_bh(&tt_global_entry->list_lock);
	hlist_add_head_rcu(&orig_entry->list,
			   &tt_global_entry->orig_list);
	spin_unlock_bh(&tt_global_entry->list_lock);
}

658
/* caller must hold orig_node refcount */
659
int batadv_tt_global_add(struct bat_priv *bat_priv, struct orig_node *orig_node,
660 661
			 const unsigned char *tt_addr, uint8_t flags,
			 uint8_t ttvn)
662
{
663
	struct tt_global_entry *tt_global_entry = NULL;
664
	int ret = 0;
665
	int hash_added;
666
	struct tt_common_entry *common;
667

668
	tt_global_entry = batadv_tt_global_hash_find(bat_priv, tt_addr);
669 670

	if (!tt_global_entry) {
671
		tt_global_entry = kzalloc(sizeof(*tt_global_entry), GFP_ATOMIC);
672
		if (!tt_global_entry)
673 674
			goto out;

675 676
		common = &tt_global_entry->common;
		memcpy(common->addr, tt_addr, ETH_ALEN);
677

678
		common->flags = flags;
679
		tt_global_entry->roam_at = 0;
680
		atomic_set(&common->refcount, 2);
681 682 683

		INIT_HLIST_HEAD(&tt_global_entry->orig_list);
		spin_lock_init(&tt_global_entry->list_lock);
684

685
		hash_added = batadv_hash_add(bat_priv->tt_global_hash,
686 687 688
					     batadv_compare_tt,
					     batadv_choose_orig, common,
					     &common->hash_entry);
689 690 691

		if (unlikely(hash_added != 0)) {
			/* remove the reference for the hash */
692
			batadv_tt_global_entry_free_ref(tt_global_entry);
693 694
			goto out_remove;
		}
695

696 697
		batadv_tt_global_add_orig_entry(tt_global_entry, orig_node,
						ttvn);
698
	} else {
699 700 701 702 703 704 705 706 707 708
		/* there is already a global entry, use this one. */

		/* If there is the TT_CLIENT_ROAM flag set, there is only one
		 * originator left in the list and we previously received a
		 * delete + roaming change for this originator.
		 *
		 * We should first delete the old originator before adding the
		 * new one.
		 */
		if (tt_global_entry->common.flags & TT_CLIENT_ROAM) {
709
			batadv_tt_global_del_orig_list(tt_global_entry);
710 711
			tt_global_entry->common.flags &= ~TT_CLIENT_ROAM;
			tt_global_entry->roam_at = 0;
712
		}
713

714 715 716 717
		if (!batadv_tt_global_entry_has_orig(tt_global_entry,
						     orig_node))
			batadv_tt_global_add_orig_entry(tt_global_entry,
							orig_node, ttvn);
718
	}
719

720 721 722
	batadv_dbg(DBG_TT, bat_priv,
		   "Creating new global tt entry: %pM (via %pM)\n",
		   tt_global_entry->common.addr, orig_node->orig);
723

724
out_remove:
725
	/* remove address from local hash if present */
726
	batadv_tt_local_remove(bat_priv, tt_global_entry->common.addr,
727
			       "global tt received", flags & TT_CLIENT_ROAM);
728 729 730
	ret = 1;
out:
	if (tt_global_entry)
731
		batadv_tt_global_entry_free_ref(tt_global_entry);
732
	return ret;
733 734
}

735 736 737
/* print all orig nodes who announce the address for this global entry.
 * it is assumed that the caller holds rcu_read_lock();
 */
738 739 740
static void
batadv_tt_global_print_entry(struct tt_global_entry *tt_global_entry,
			     struct seq_file *seq)
741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763
{
	struct hlist_head *head;
	struct hlist_node *node;
	struct tt_orig_list_entry *orig_entry;
	struct tt_common_entry *tt_common_entry;
	uint16_t flags;
	uint8_t last_ttvn;

	tt_common_entry = &tt_global_entry->common;

	head = &tt_global_entry->orig_list;

	hlist_for_each_entry_rcu(orig_entry, node, head, list) {
		flags = tt_common_entry->flags;
		last_ttvn = atomic_read(&orig_entry->orig_node->last_ttvn);
		seq_printf(seq, " * %pM  (%3u) via %pM     (%3u)   [%c%c]\n",
			   tt_global_entry->common.addr, orig_entry->ttvn,
			   orig_entry->orig_node->orig, last_ttvn,
			   (flags & TT_CLIENT_ROAM ? 'R' : '.'),
			   (flags & TT_CLIENT_WIFI ? 'W' : '.'));
	}
}

764
int batadv_tt_global_seq_print_text(struct seq_file *seq, void *offset)
765 766 767
{
	struct net_device *net_dev = (struct net_device *)seq->private;
	struct bat_priv *bat_priv = netdev_priv(net_dev);
768
	struct hashtable_t *hash = bat_priv->tt_global_hash;
769
	struct tt_common_entry *tt_common_entry;
770
	struct tt_global_entry *tt_global_entry;
771
	struct hard_iface *primary_if;
772
	struct hlist_node *node;
773
	struct hlist_head *head;
774 775
	uint32_t i;
	int ret = 0;
776

777
	primary_if = batadv_primary_if_get_selected(bat_priv);
778
	if (!primary_if) {
779 780
		ret = seq_printf(seq,
				 "BATMAN mesh %s disabled - please specify interfaces to enable it\n",
781 782 783
				 net_dev->name);
		goto out;
	}
784

785
	if (primary_if->if_status != IF_ACTIVE) {
786 787
		ret = seq_printf(seq,
				 "BATMAN mesh %s disabled - primary interface not active\n",
788 789
				 net_dev->name);
		goto out;
790 791
	}

792 793
	seq_printf(seq,
		   "Globally announced TT entries received via the mesh %s\n",
794
		   net_dev->name);
795 796
	seq_printf(seq, "       %-13s %s       %-15s %s %s\n",
		   "Client", "(TTVN)", "Originator", "(Curr TTVN)", "Flags");
797 798 799 800

	for (i = 0; i < hash->size; i++) {
		head = &hash->table[i];

801
		rcu_read_lock();
802
		hlist_for_each_entry_rcu(tt_common_entry, node,
803
					 head, hash_entry) {
804 805 806
			tt_global_entry = container_of(tt_common_entry,
						       struct tt_global_entry,
						       common);
807
			batadv_tt_global_print_entry(tt_global_entry, seq);
808
		}
809
		rcu_read_unlock();
810
	}
811 812
out:
	if (primary_if)
813
		batadv_hardif_free_ref(primary_if);
814
	return ret;
815 816
}

817
/* deletes the orig list of a tt_global_entry */
818 819
static void
batadv_tt_global_del_orig_list(struct tt_global_entry *tt_global_entry)
820
{
821 822 823
	struct hlist_head *head;
	struct hlist_node *node, *safe;
	struct tt_orig_list_entry *orig_entry;
824

825 826 827 828
	spin_lock_bh(&tt_global_entry->list_lock);
	head = &tt_global_entry->orig_list;
	hlist_for_each_entry_safe(orig_entry, node, safe, head, list) {
		hlist_del_rcu(node);
829
		batadv_tt_orig_list_entry_free_ref(orig_entry);
830 831
	}
	spin_unlock_bh(&tt_global_entry->list_lock);
832

833 834
}

835 836 837 838 839
static void
batadv_tt_global_del_orig_entry(struct bat_priv *bat_priv,
				struct tt_global_entry *tt_global_entry,
				struct orig_node *orig_node,
				const char *message)
840 841 842 843 844 845 846 847 848
{
	struct hlist_head *head;
	struct hlist_node *node, *safe;
	struct tt_orig_list_entry *orig_entry;

	spin_lock_bh(&tt_global_entry->list_lock);
	head = &tt_global_entry->orig_list;
	hlist_for_each_entry_safe(orig_entry, node, safe, head, list) {
		if (orig_entry->orig_node == orig_node) {
849 850 851 852
			batadv_dbg(DBG_TT, bat_priv,
				   "Deleting %pM from global tt entry %pM: %s\n",
				   orig_node->orig,
				   tt_global_entry->common.addr, message);
853
			hlist_del_rcu(node);
854
			batadv_tt_orig_list_entry_free_ref(orig_entry);
855 856 857 858 859
		}
	}
	spin_unlock_bh(&tt_global_entry->list_lock);
}

860 861 862
static void batadv_tt_global_del_struct(struct bat_priv *bat_priv,
					struct tt_global_entry *tt_global_entry,
					const char *message)
863
{
864 865
	batadv_dbg(DBG_TT, bat_priv, "Deleting global tt entry %pM: %s\n",
		   tt_global_entry->common.addr, message);
866

867
	batadv_hash_remove(bat_priv->tt_global_hash, batadv_compare_tt,
868
			   batadv_choose_orig, tt_global_entry->common.addr);
869
	batadv_tt_global_entry_free_ref(tt_global_entry);
870

871 872
}

873 874 875 876
/* If the client is to be deleted, we check if it is the last origantor entry
 * within tt_global entry. If yes, we set the TT_CLIENT_ROAM flag and the timer,
 * otherwise we simply remove the originator scheduled for deletion.
 */
877 878 879 880
static void
batadv_tt_global_del_roaming(struct bat_priv *bat_priv,
			     struct tt_global_entry *tt_global_entry,
			     struct orig_node *orig_node, const char *message)
881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908
{
	bool last_entry = true;
	struct hlist_head *head;
	struct hlist_node *node;
	struct tt_orig_list_entry *orig_entry;

	/* no local entry exists, case 1:
	 * Check if this is the last one or if other entries exist.
	 */

	rcu_read_lock();
	head = &tt_global_entry->orig_list;
	hlist_for_each_entry_rcu(orig_entry, node, head, list) {
		if (orig_entry->orig_node != orig_node) {
			last_entry = false;
			break;
		}
	}
	rcu_read_unlock();

	if (last_entry) {
		/* its the last one, mark for roaming. */
		tt_global_entry->common.flags |= TT_CLIENT_ROAM;
		tt_global_entry->roam_at = jiffies;
	} else
		/* there is another entry, we can simply delete this
		 * one and can still use the other one.
		 */
909 910
		batadv_tt_global_del_orig_entry(bat_priv, tt_global_entry,
						orig_node, message);
911 912 913 914
}



915 916 917 918
static void batadv_tt_global_del(struct bat_priv *bat_priv,
				 struct orig_node *orig_node,
				 const unsigned char *addr,
				 const char *message, bool roaming)
919
{
920
	struct tt_global_entry *tt_global_entry = NULL;
921
	struct tt_local_entry *local_entry = NULL;
922

923
	tt_global_entry = batadv_tt_global_hash_find(bat_priv, addr);
924
	if (!tt_global_entry)
925
		goto out;
926

927
	if (!roaming) {
928 929
		batadv_tt_global_del_orig_entry(bat_priv, tt_global_entry,
						orig_node, message);
930 931

		if (hlist_empty(&tt_global_entry->orig_list))
932 933
			batadv_tt_global_del_struct(bat_priv, tt_global_entry,
						    message);
934 935 936

		goto out;
	}
937 938 939

	/* if we are deleting a global entry due to a roam
	 * event, there are two possibilities:
940 941
	 * 1) the client roamed from node A to node B => if there
	 *    is only one originator left for this client, we mark
942 943 944
	 *    it with TT_CLIENT_ROAM, we start a timer and we
	 *    wait for node B to claim it. In case of timeout
	 *    the entry is purged.
945 946 947
	 *
	 *    If there are other originators left, we directly delete
	 *    the originator.
948
	 * 2) the client roamed to us => we can directly delete
949 950
	 *    the global entry, since it is useless now.
	 */
951 952 953
	local_entry = batadv_tt_local_hash_find(bat_priv,
						tt_global_entry->common.addr);
	if (local_entry) {
954
		/* local entry exists, case 2: client roamed to us. */
955 956
		batadv_tt_global_del_orig_list(tt_global_entry);
		batadv_tt_global_del_struct(bat_priv, tt_global_entry, message);
957 958
	} else
		/* no local entry exists, case 1: check for roaming */
959 960
		batadv_tt_global_del_roaming(bat_priv, tt_global_entry,
					     orig_node, message);
961 962


963
out:
964
	if (tt_global_entry)
965 966 967
		batadv_tt_global_entry_free_ref(tt_global_entry);
	if (local_entry)
		batadv_tt_local_entry_free_ref(local_entry);
968 969
}

970 971
void batadv_tt_global_del_orig(struct bat_priv *bat_priv,
			       struct orig_node *orig_node, const char *message)
972
{
973
	struct tt_global_entry *global_entry;
974
	struct tt_common_entry *tt_common_entry;
975
	uint32_t i;
976 977 978
	struct hashtable_t *hash = bat_priv->tt_global_hash;
	struct hlist_node *node, *safe;
	struct hlist_head *head;
979
	spinlock_t *list_lock; /* protects write access to the hash lists */
980

981 982 983
	if (!hash)
		return;

984 985
	for (i = 0; i < hash->size; i++) {
		head = &hash->table[i];
986
		list_lock = &hash->list_locks[i];
987

988
		spin_lock_bh(list_lock);
989
		hlist_for_each_entry_safe(tt_common_entry, node, safe,
990
					  head, hash_entry) {
991 992 993
			global_entry = container_of(tt_common_entry,
						    struct tt_global_entry,
						    common);
994

995 996
			batadv_tt_global_del_orig_entry(bat_priv, global_entry,
							orig_node, message);
997

998
			if (hlist_empty(&global_entry->orig_list)) {
999 1000
				batadv_dbg(DBG_TT, bat_priv,
					   "Deleting global tt entry %pM: %s\n",
1001
					   global_entry->common.addr, message);
1002
				hlist_del_rcu(node);
1003
				batadv_tt_global_entry_free_ref(global_entry);
1004
			}
1005
		}
1006
		spin_unlock_bh(list_lock);
1007
	}
1008
	orig_node->tt_initialised = false;
1009 1010
}

1011
static void batadv_tt_global_roam_purge(struct bat_priv *bat_priv)
1012 1013
{
	struct hashtable_t *hash = bat_priv->tt_global_hash;
1014
	struct tt_common_entry *tt_common_entry;
1015 1016 1017
	struct tt_global_entry *tt_global_entry;
	struct hlist_node *node, *node_tmp;
	struct hlist_head *head;
1018
	spinlock_t *list_lock; /* protects write access to the hash lists */
1019
	uint32_t i;
1020 1021 1022

	for (i = 0; i < hash->size; i++) {
		head = &hash->table[i];
1023
		list_lock = &hash->list_locks[i];
1024

1025
		spin_lock_bh(list_lock);
1026
		hlist_for_each_entry_safe(tt_common_entry, node, node_tmp,
1027
					  head, hash_entry) {
1028 1029 1030 1031
			tt_global_entry = container_of(tt_common_entry,
						       struct tt_global_entry,
						       common);
			if (!(tt_global_entry->common.flags & TT_CLIENT_ROAM))
1032
				continue;
1033 1034
			if (!batadv_has_timed_out(tt_global_entry->roam_at,
						  TT_CLIENT_ROAM_TIMEOUT))
1035 1036
				continue;

1037 1038 1039
			batadv_dbg(DBG_TT, bat_priv,
				   "Deleting global tt entry (%pM): Roaming timeout\n",
				   tt_global_entry->common.addr);
1040

1041
			hlist_del_rcu(node);
1042
			batadv_tt_global_entry_free_ref(tt_global_entry);
1043
		}
1044
		spin_unlock_bh(list_lock);
1045 1046 1047 1048
	}

}

1049
static void batadv_tt_global_table_free(struct bat_priv *bat_priv)
1050
{
1051 1052
	struct hashtable_t *hash;
	spinlock_t *list_lock; /* protects write access to the hash lists */
1053
	struct tt_common_entry *tt_common_entry;
1054 1055 1056
	struct tt_global_entry *tt_global_entry;
	struct hlist_node *node, *node_tmp;
	struct hlist_head *head;
1057
	uint32_t i;
1058

1059
	if (!bat_priv->tt_global_hash)
1060 1061
		return;

1062 1063 1064 1065 1066 1067 1068
	hash = bat_priv->tt_global_hash;

	for (i = 0; i < hash->size; i++) {
		head = &hash->table[i];
		list_lock = &hash->list_locks[i];

		spin_lock_bh(list_lock);
1069
		hlist_for_each_entry_safe(tt_common_entry, node, node_tmp,
1070 1071
					  head, hash_entry) {
			hlist_del_rcu(node);
1072 1073 1074
			tt_global_entry = container_of(tt_common_entry,
						       struct tt_global_entry,
						       common);
1075
			batadv_tt_global_entry_free_ref(tt_global_entry);
1076 1077 1078 1079
		}
		spin_unlock_bh(list_lock);
	}

1080
	batadv_hash_destroy(hash);
1081

1082
	bat_priv->tt_global_hash = NULL;
1083 1084
}

1085 1086
static bool _batadv_is_ap_isolated(struct tt_local_entry *tt_local_entry,
				   struct tt_global_entry *tt_global_entry)
1087 1088 1089
{
	bool ret = false;

1090 1091
	if (tt_local_entry->common.flags & TT_CLIENT_WIFI &&
	    tt_global_entry->common.flags & TT_CLIENT_WIFI)
1092 1093 1094 1095 1096
		ret = true;

	return ret;
}

1097 1098 1099
struct orig_node *batadv_transtable_search(struct bat_priv *bat_priv,
					   const uint8_t *src,
					   const uint8_t *addr)
1100
{
1101 1102
	struct tt_local_entry *tt_local_entry = NULL;
	struct tt_global_entry *tt_global_entry = NULL;
1103
	struct orig_node *orig_node = NULL;
1104 1105 1106 1107 1108
	struct neigh_node *router = NULL;
	struct hlist_head *head;
	struct hlist_node *node;
	struct tt_orig_list_entry *orig_entry;
	int best_tq;
1109

1110
	if (src && atomic_read(&bat_priv->ap_isolation)) {
1111
		tt_local_entry = batadv_tt_local_hash_find(bat_priv, src);
1112 1113 1114
		if (!tt_local_entry)
			goto out;
	}
1115

1116
	tt_global_entry = batadv_tt_global_hash_find(bat_priv, addr);
1117
	if (!tt_global_entry)
1118
		goto out;
1119

1120
	/* check whether the clients should not communicate due to AP
1121 1122
	 * isolation
	 */
1123 1124
	if (tt_local_entry &&
	    _batadv_is_ap_isolated(tt_local_entry, tt_global_entry))
1125 1126
		goto out;

1127
	best_tq = 0;
1128

1129 1130 1131
	rcu_read_lock();
	head = &tt_global_entry->orig_list;
	hlist_for_each_entry_rcu(orig_entry, node, head, list) {
1132
		router = batadv_orig_node_get_router(orig_entry->orig_node);
1133 1134
		if (!router)
			continue;
1135

1136 1137 1138 1139
		if (router->tq_avg > best_tq) {
			orig_node = orig_entry->orig_node;
			best_tq = router->tq_avg;
		}
1140
		batadv_neigh_node_free_ref(router);
1141 1142 1143 1144 1145
	}
	/* found anything? */
	if (orig_node && !atomic_inc_not_zero(&orig_node->refcount))
		orig_node = NULL;
	rcu_read_unlock();
1146
out:
1147
	if (tt_global_entry)
1148
		batadv_tt_global_entry_free_ref(tt_global_entry);
1149
	if (tt_local_entry)
1150
		batadv_tt_local_entry_free_ref(tt_local_entry);
1151

1152
	return orig_node;
1153
}
1154 1155

/* Calculates the checksum of the local table of a given orig_node */
1156 1157
static uint16_t batadv_tt_global_crc(struct bat_priv *bat_priv,
				     struct orig_node *orig_node)
1158 1159 1160
{
	uint16_t total = 0, total_one;
	struct hashtable_t *hash = bat_priv->tt_global_hash;
1161
	struct tt_common_entry *tt_common_entry;
1162 1163 1164
	struct tt_global_entry *tt_global_entry;
	struct hlist_node *node;
	struct hlist_head *head;
1165 1166
	uint32_t i;
	int j;
1167 1168 1169 1170 1171

	for (i = 0; i < hash->size; i++) {
		head = &hash->table[i];

		rcu_read_lock();
1172
		hlist_for_each_entry_rcu(tt_common_entry, node,
1173
					 head, hash_entry) {
1174 1175 1176
			tt_global_entry = container_of(tt_common_entry,
						       struct tt_global_entry,
						       common);
1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187
			/* Roaming clients are in the global table for
			 * consistency only. They don't have to be
			 * taken into account while computing the
			 * global crc
			 */
			if (tt_global_entry->common.flags & TT_CLIENT_ROAM)
				continue;

			/* find out if this global entry is announced by this
			 * originator
			 */
1188 1189
			if (!batadv_tt_global_entry_has_orig(tt_global_entry,
							     orig_node))
1190 1191 1192 1193 1194 1195 1196
				continue;

			total_one = 0;
			for (j = 0; j < ETH_ALEN; j++)
				total_one = crc16_byte(total_one,
					tt_global_entry->common.addr[j]);
			total ^= total_one;
1197 1198 1199 1200 1201 1202 1203 1204
		}
		rcu_read_unlock();
	}

	return total;
}

/* Calculates the checksum of the local table */
1205
static uint16_t batadv_tt_local_crc(struct bat_priv *bat_priv)
1206 1207 1208
{
	uint16_t total = 0, total_one;
	struct hashtable_t *hash = bat_priv->tt_local_hash;
1209
	struct tt_common_entry *tt_common_entry;
1210 1211
	struct hlist_node *node;
	struct hlist_head *head;
1212 1213
	uint32_t i;
	int j;
1214 1215 1216 1217 1218

	for (i = 0; i < hash->size; i++) {
		head = &hash->table[i];

		rcu_read_lock();
1219
		hlist_for_each_entry_rcu(tt_common_entry, node,
1220
					 head, hash_entry) {
1221
			/* not yet committed clients have not to be taken into
1222 1223
			 * account while computing the CRC
			 */
1224
			if (tt_common_entry->flags & TT_CLIENT_NEW)
1225
				continue;
1226 1227 1228
			total_one = 0;
			for (j = 0; j < ETH_ALEN; j++)
				total_one = crc16_byte(total_one,
1229
						   tt_common_entry->addr[j]);
1230 1231 1232 1233 1234 1235 1236 1237
			total ^= total_one;
		}
		rcu_read_unlock();
	}

	return total;
}

1238
static void batadv_tt_req_list_free(struct bat_priv *bat_priv)
1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251
{
	struct tt_req_node *node, *safe;

	spin_lock_bh(&bat_priv->tt_req_list_lock);

	list_for_each_entry_safe(node, safe, &bat_priv->tt_req_list, list) {
		list_del(&node->list);
		kfree(node);
	}

	spin_unlock_bh(&bat_priv->tt_req_list_lock);
}

1252 1253 1254 1255
static void batadv_tt_save_orig_buffer(struct bat_priv *bat_priv,
				       struct orig_node *orig_node,
				       const unsigned char *tt_buff,
				       uint8_t tt_num_changes)
1256
{
1257
	uint16_t tt_buff_len = batadv_tt_len(tt_num_changes);
1258 1259

	/* Replace the old buffer only if I received something in the
1260 1261
	 * last OGM (the OGM could carry no changes)
	 */
1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274
	spin_lock_bh(&orig_node->tt_buff_lock);
	if (tt_buff_len > 0) {
		kfree(orig_node->tt_buff);
		orig_node->tt_buff_len = 0;
		orig_node->tt_buff = kmalloc(tt_buff_len, GFP_ATOMIC);
		if (orig_node->tt_buff) {
			memcpy(orig_node->tt_buff, tt_buff, tt_buff_len);
			orig_node->tt_buff_len = tt_buff_len;
		}
	}
	spin_unlock_bh(&orig_node->tt_buff_lock);
}

1275
static void batadv_tt_req_purge(struct bat_priv *bat_priv)
1276 1277 1278 1279 1280
{
	struct tt_req_node *node, *safe;

	spin_lock_bh(&bat_priv->tt_req_list_lock);
	list_for_each_entry_safe(node, safe, &bat_priv->tt_req_list, list) {
1281
		if (batadv_has_timed_out(node->issued_at, TT_REQUEST_TIMEOUT)) {
1282 1283 1284 1285 1286 1287 1288 1289
			list_del(&node->list);
			kfree(node);
		}
	}
	spin_unlock_bh(&bat_priv->tt_req_list_lock);
}

/* returns the pointer to the new tt_req_node struct if no request
1290 1291
 * has already been issued for this orig_node, NULL otherwise
 */
1292 1293
static struct tt_req_node *batadv_new_tt_req_node(struct bat_priv *bat_priv,
						  struct orig_node *orig_node)
1294 1295 1296 1297 1298
{
	struct tt_req_node *tt_req_node_tmp, *tt_req_node = NULL;

	spin_lock_bh(&bat_priv->tt_req_list_lock);
	list_for_each_entry(tt_req_node_tmp, &bat_priv->tt_req_list, list) {
1299 1300 1301
		if (batadv_compare_eth(tt_req_node_tmp, orig_node) &&
		    !batadv_has_timed_out(tt_req_node_tmp->issued_at,
					  TT_REQUEST_TIMEOUT))
1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317
			goto unlock;
	}

	tt_req_node = kmalloc(sizeof(*tt_req_node), GFP_ATOMIC);
	if (!tt_req_node)
		goto unlock;

	memcpy(tt_req_node->addr, orig_node->orig, ETH_ALEN);
	tt_req_node->issued_at = jiffies;

	list_add(&tt_req_node->list, &bat_priv->tt_req_list);
unlock:
	spin_unlock_bh(&bat_priv->tt_req_list_lock);
	return tt_req_node;
}

1318
/* data_ptr is useless here, but has to be kept to respect the prototype */
1319 1320
static int batadv_tt_local_valid_entry(const void *entry_ptr,
				       const void *data_ptr)
1321
{
1322
	const struct tt_common_entry *tt_common_entry = entry_ptr;
1323

1324
	if (tt_common_entry->flags & TT_CLIENT_NEW)
1325 1326 1327 1328
		return 0;
	return 1;
}

1329 1330
static int batadv_tt_global_valid(const void *entry_ptr,
				  const void *data_ptr)
1331
{
1332 1333
	const struct tt_common_entry *tt_common_entry = entry_ptr;
	const struct tt_global_entry *tt_global_entry;
1334 1335
	const struct orig_node *orig_node = data_ptr;

1336
	if (tt_common_entry->flags & TT_CLIENT_ROAM)
1337 1338
		return 0;

1339 1340 1341
	tt_global_entry = container_of(tt_common_entry, struct tt_global_entry,
				       common);

1342
	return batadv_tt_global_entry_has_orig(tt_global_entry, orig_node);
1343 1344
}

1345 1346 1347 1348 1349 1350
static struct sk_buff *
batadv_tt_response_fill_table(uint16_t tt_len, uint8_t ttvn,
			      struct hashtable_t *hash,
			      struct hard_iface *primary_if,
			      int (*valid_cb)(const void *, const void *),
			      void *cb_data)
1351
{
1352
	struct tt_common_entry *tt_common_entry;
1353 1354 1355 1356 1357 1358 1359
	struct tt_query_packet *tt_response;
	struct tt_change *tt_change;
	struct hlist_node *node;
	struct hlist_head *head;
	struct sk_buff *skb = NULL;
	uint16_t tt_tot, tt_count;
	ssize_t tt_query_size = sizeof(struct tt_query_packet);
1360
	uint32_t i;
1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383

	if (tt_query_size + tt_len > primary_if->soft_iface->mtu) {
		tt_len = primary_if->soft_iface->mtu - tt_query_size;
		tt_len -= tt_len % sizeof(struct tt_change);
	}
	tt_tot = tt_len / sizeof(struct tt_change);

	skb = dev_alloc_skb(tt_query_size + tt_len + ETH_HLEN);
	if (!skb)
		goto out;

	skb_reserve(skb, ETH_HLEN);
	tt_response = (struct tt_query_packet *)skb_put(skb,
						     tt_query_size + tt_len);
	tt_response->ttvn = ttvn;

	tt_change = (struct tt_change *)(skb->data + tt_query_size);
	tt_count = 0;

	rcu_read_lock();
	for (i = 0; i < hash->size; i++) {
		head = &hash->table[i];

1384
		hlist_for_each_entry_rcu(tt_common_entry, node,
1385 1386 1387 1388
					 head, hash_entry) {
			if (tt_count == tt_tot)
				break;

1389
			if ((valid_cb) && (!valid_cb(tt_common_entry, cb_data)))
1390 1391
				continue;

1392 1393
			memcpy(tt_change->addr, tt_common_entry->addr,
			       ETH_ALEN);
1394 1395 1396 1397 1398 1399 1400 1401
			tt_change->flags = NO_FLAGS;

			tt_count++;
			tt_change++;
		}
	}
	rcu_read_unlock();

1402
	/* store in the message the number of entries we have successfully
1403 1404
	 * copied
	 */
1405 1406
	tt_response->tt_data = htons(tt_count);

1407 1408 1409 1410
out:
	return skb;
}

1411 1412 1413 1414
static int batadv_send_tt_request(struct bat_priv *bat_priv,
				  struct orig_node *dst_orig_node,
				  uint8_t ttvn, uint16_t tt_crc,
				  bool full_table)
1415 1416 1417 1418 1419 1420 1421 1422
{
	struct sk_buff *skb = NULL;
	struct tt_query_packet *tt_request;
	struct neigh_node *neigh_node = NULL;
	struct hard_iface *primary_if;
	struct tt_req_node *tt_req_node = NULL;
	int ret = 1;

1423
	primary_if = batadv_primary_if_get_selected(bat_priv);
1424 1425 1426 1427
	if (!primary_if)
		goto out;

	/* The new tt_req will be issued only if I'm not waiting for a
1428 1429
	 * reply from the same orig_node yet
	 */
1430
	tt_req_node = batadv_new_tt_req_node(bat_priv, dst_orig_node);
1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442
	if (!tt_req_node)
		goto out;

	skb = dev_alloc_skb(sizeof(struct tt_query_packet) + ETH_HLEN);
	if (!skb)
		goto out;

	skb_reserve(skb, ETH_HLEN);

	tt_request = (struct tt_query_packet *)skb_put(skb,
				sizeof(struct tt_query_packet));

1443
	tt_request->header.packet_type = BAT_TT_QUERY;
1444
	tt_request->header.version = BATADV_COMPAT_VERSION;
1445 1446
	memcpy(tt_request->src, primary_if->net_dev->dev_addr, ETH_ALEN);
	memcpy(tt_request->dst, dst_orig_node->orig, ETH_ALEN);
1447
	tt_request->header.ttl = TTL;
1448
	tt_request->ttvn = ttvn;
1449
	tt_request->tt_data = htons(tt_crc);
1450 1451 1452 1453 1454
	tt_request->flags = TT_REQUEST;

	if (full_table)
		tt_request->flags |= TT_FULL_TABLE;

1455
	neigh_node = batadv_orig_node_get_router(dst_orig_node);
1456 1457 1458
	if (!neigh_node)
		goto out;

1459 1460 1461 1462
	batadv_dbg(DBG_TT, bat_priv,
		   "Sending TT_REQUEST to %pM via %pM [%c]\n",
		   dst_orig_node->orig, neigh_node->addr,
		   (full_table ? 'F' : '.'));
1463

1464 1465
	batadv_inc_counter(bat_priv, BAT_CNT_TT_REQUEST_TX);

1466
	batadv_send_skb_packet(skb, neigh_node->if_incoming, neigh_node->addr);
1467 1468 1469 1470
	ret = 0;

out:
	if (neigh_node)
1471
		batadv_neigh_node_free_ref(neigh_node);
1472
	if (primary_if)
1473
		batadv_hardif_free_ref(primary_if);
1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484
	if (ret)
		kfree_skb(skb);
	if (ret && tt_req_node) {
		spin_lock_bh(&bat_priv->tt_req_list_lock);
		list_del(&tt_req_node->list);
		spin_unlock_bh(&bat_priv->tt_req_list_lock);
		kfree(tt_req_node);
	}
	return ret;
}

1485 1486
static bool batadv_send_other_tt_response(struct bat_priv *bat_priv,
					  struct tt_query_packet *tt_request)
1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498
{
	struct orig_node *req_dst_orig_node = NULL, *res_dst_orig_node = NULL;
	struct neigh_node *neigh_node = NULL;
	struct hard_iface *primary_if = NULL;
	uint8_t orig_ttvn, req_ttvn, ttvn;
	int ret = false;
	unsigned char *tt_buff;
	bool full_table;
	uint16_t tt_len, tt_tot;
	struct sk_buff *skb = NULL;
	struct tt_query_packet *tt_response;

1499 1500 1501 1502
	batadv_dbg(DBG_TT, bat_priv,
		   "Received TT_REQUEST from %pM for ttvn: %u (%pM) [%c]\n",
		   tt_request->src, tt_request->ttvn, tt_request->dst,
		   (tt_request->flags & TT_FULL_TABLE ? 'F' : '.'));
1503 1504

	/* Let's get the orig node of the REAL destination */
1505
	req_dst_orig_node = batadv_orig_hash_find(bat_priv, tt_request->dst);
1506 1507 1508
	if (!req_dst_orig_node)
		goto out;

1509
	res_dst_orig_node = batadv_orig_hash_find(bat_priv, tt_request->src);
1510 1511 1512
	if (!res_dst_orig_node)
		goto out;

1513
	neigh_node = batadv_orig_node_get_router(res_dst_orig_node);
1514 1515 1516
	if (!neigh_node)
		goto out;

1517
	primary_if = batadv_primary_if_get_selected(bat_priv);
1518 1519 1520 1521 1522 1523
	if (!primary_if)
		goto out;

	orig_ttvn = (uint8_t)atomic_read(&req_dst_orig_node->last_ttvn);
	req_ttvn = tt_request->ttvn;

1524
	/* I don't have the requested data */
1525
	if (orig_ttvn != req_ttvn ||
1526
	    tt_request->tt_data != htons(req_dst_orig_node->tt_crc))
1527 1528
		goto out;

1529
	/* If the full table has been explicitly requested */
1530 1531 1532 1533 1534 1535 1536
	if (tt_request->flags & TT_FULL_TABLE ||
	    !req_dst_orig_node->tt_buff)
		full_table = true;
	else
		full_table = false;

	/* In this version, fragmentation is not implemented, then
1537 1538
	 * I'll send only one packet with as much TT entries as I can
	 */
1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565
	if (!full_table) {
		spin_lock_bh(&req_dst_orig_node->tt_buff_lock);
		tt_len = req_dst_orig_node->tt_buff_len;
		tt_tot = tt_len / sizeof(struct tt_change);

		skb = dev_alloc_skb(sizeof(struct tt_query_packet) +
				    tt_len + ETH_HLEN);
		if (!skb)
			goto unlock;

		skb_reserve(skb, ETH_HLEN);
		tt_response = (struct tt_query_packet *)skb_put(skb,
				sizeof(struct tt_query_packet) + tt_len);
		tt_response->ttvn = req_ttvn;
		tt_response->tt_data = htons(tt_tot);

		tt_buff = skb->data + sizeof(struct tt_query_packet);
		/* Copy the last orig_node's OGM buffer */
		memcpy(tt_buff, req_dst_orig_node->tt_buff,
		       req_dst_orig_node->tt_buff_len);

		spin_unlock_bh(&req_dst_orig_node->tt_buff_lock);
	} else {
		tt_len = (uint16_t)atomic_read(&req_dst_orig_node->tt_size) *
						sizeof(struct tt_change);
		ttvn = (uint8_t)atomic_read(&req_dst_orig_node->last_ttvn);

1566 1567 1568 1569 1570
		skb = batadv_tt_response_fill_table(tt_len, ttvn,
						    bat_priv->tt_global_hash,
						    primary_if,
						    batadv_tt_global_valid,
						    req_dst_orig_node);
1571 1572 1573 1574 1575 1576
		if (!skb)
			goto out;

		tt_response = (struct tt_query_packet *)skb->data;
	}

1577
	tt_response->header.packet_type = BAT_TT_QUERY;
1578
	tt_response->header.version = BATADV_COMPAT_VERSION;
1579
	tt_response->header.ttl = TTL;
1580 1581 1582 1583 1584 1585 1586
	memcpy(tt_response->src, req_dst_orig_node->orig, ETH_ALEN);
	memcpy(tt_response->dst, tt_request->src, ETH_ALEN);
	tt_response->flags = TT_RESPONSE;

	if (full_table)
		tt_response->flags |= TT_FULL_TABLE;

1587 1588 1589 1590
	batadv_dbg(DBG_TT, bat_priv,
		   "Sending TT_RESPONSE %pM via %pM for %pM (ttvn: %u)\n",
		   res_dst_orig_node->orig, neigh_node->addr,
		   req_dst_orig_node->orig, req_ttvn);
1591

1592 1593
	batadv_inc_counter(bat_priv, BAT_CNT_TT_RESPONSE_TX);

1594
	batadv_send_skb_packet(skb, neigh_node->if_incoming, neigh_node->addr);
1595 1596 1597 1598 1599 1600 1601 1602
	ret = true;
	goto out;

unlock:
	spin_unlock_bh(&req_dst_orig_node->tt_buff_lock);

out:
	if (res_dst_orig_node)
1603
		batadv_orig_node_free_ref(res_dst_orig_node);
1604
	if (req_dst_orig_node)
1605
		batadv_orig_node_free_ref(req_dst_orig_node);
1606
	if (neigh_node)
1607
		batadv_neigh_node_free_ref(neigh_node);
1608
	if (primary_if)
1609
		batadv_hardif_free_ref(primary_if);
1610 1611 1612 1613 1614
	if (!ret)
		kfree_skb(skb);
	return ret;

}
1615 1616
static bool batadv_send_my_tt_response(struct bat_priv *bat_priv,
				       struct tt_query_packet *tt_request)
1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628
{
	struct orig_node *orig_node = NULL;
	struct neigh_node *neigh_node = NULL;
	struct hard_iface *primary_if = NULL;
	uint8_t my_ttvn, req_ttvn, ttvn;
	int ret = false;
	unsigned char *tt_buff;
	bool full_table;
	uint16_t tt_len, tt_tot;
	struct sk_buff *skb = NULL;
	struct tt_query_packet *tt_response;

1629 1630 1631 1632
	batadv_dbg(DBG_TT, bat_priv,
		   "Received TT_REQUEST from %pM for ttvn: %u (me) [%c]\n",
		   tt_request->src, tt_request->ttvn,
		   (tt_request->flags & TT_FULL_TABLE ? 'F' : '.'));
1633 1634 1635 1636 1637


	my_ttvn = (uint8_t)atomic_read(&bat_priv->ttvn);
	req_ttvn = tt_request->ttvn;

1638
	orig_node = batadv_orig_hash_find(bat_priv, tt_request->src);
1639 1640 1641
	if (!orig_node)
		goto out;

1642
	neigh_node = batadv_orig_node_get_router(orig_node);
1643 1644 1645
	if (!neigh_node)
		goto out;

1646
	primary_if = batadv_primary_if_get_selected(bat_priv);
1647 1648 1649 1650
	if (!primary_if)
		goto out;

	/* If the full table has been explicitly requested or the gap
1651 1652
	 * is too big send the whole local translation table
	 */
1653 1654 1655 1656 1657 1658 1659
	if (tt_request->flags & TT_FULL_TABLE || my_ttvn != req_ttvn ||
	    !bat_priv->tt_buff)
		full_table = true;
	else
		full_table = false;

	/* In this version, fragmentation is not implemented, then
1660 1661
	 * I'll send only one packet with as much TT entries as I can
	 */
1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686
	if (!full_table) {
		spin_lock_bh(&bat_priv->tt_buff_lock);
		tt_len = bat_priv->tt_buff_len;
		tt_tot = tt_len / sizeof(struct tt_change);

		skb = dev_alloc_skb(sizeof(struct tt_query_packet) +
				    tt_len + ETH_HLEN);
		if (!skb)
			goto unlock;

		skb_reserve(skb, ETH_HLEN);
		tt_response = (struct tt_query_packet *)skb_put(skb,
				sizeof(struct tt_query_packet) + tt_len);
		tt_response->ttvn = req_ttvn;
		tt_response->tt_data = htons(tt_tot);

		tt_buff = skb->data + sizeof(struct tt_query_packet);
		memcpy(tt_buff, bat_priv->tt_buff,
		       bat_priv->tt_buff_len);
		spin_unlock_bh(&bat_priv->tt_buff_lock);
	} else {
		tt_len = (uint16_t)atomic_read(&bat_priv->num_local_tt) *
						sizeof(struct tt_change);
		ttvn = (uint8_t)atomic_read(&bat_priv->ttvn);

1687 1688 1689 1690 1691
		skb = batadv_tt_response_fill_table(tt_len, ttvn,
						    bat_priv->tt_local_hash,
						    primary_if,
						    batadv_tt_local_valid_entry,
						    NULL);
1692 1693 1694 1695 1696 1697
		if (!skb)
			goto out;

		tt_response = (struct tt_query_packet *)skb->data;
	}

1698
	tt_response->header.packet_type = BAT_TT_QUERY;
1699
	tt_response->header.version = BATADV_COMPAT_VERSION;
1700
	tt_response->header.ttl = TTL;
1701 1702 1703 1704 1705 1706 1707
	memcpy(tt_response->src, primary_if->net_dev->dev_addr, ETH_ALEN);
	memcpy(tt_response->dst, tt_request->src, ETH_ALEN);
	tt_response->flags = TT_RESPONSE;

	if (full_table)
		tt_response->flags |= TT_FULL_TABLE;

1708 1709 1710 1711
	batadv_dbg(DBG_TT, bat_priv,
		   "Sending TT_RESPONSE to %pM via %pM [%c]\n",
		   orig_node->orig, neigh_node->addr,
		   (tt_response->flags & TT_FULL_TABLE ? 'F' : '.'));
1712

1713 1714
	batadv_inc_counter(bat_priv, BAT_CNT_TT_RESPONSE_TX);

1715
	batadv_send_skb_packet(skb, neigh_node->if_incoming, neigh_node->addr);
1716 1717 1718 1719 1720 1721 1722
	ret = true;
	goto out;

unlock:
	spin_unlock_bh(&bat_priv->tt_buff_lock);
out:
	if (orig_node)
1723
		batadv_orig_node_free_ref(orig_node);
1724
	if (neigh_node)
1725
		batadv_neigh_node_free_ref(neigh_node);
1726
	if (primary_if)
1727
		batadv_hardif_free_ref(primary_if);
1728 1729 1730 1731 1732 1733
	if (!ret)
		kfree_skb(skb);
	/* This packet was for me, so it doesn't need to be re-routed */
	return true;
}

1734 1735
bool batadv_send_tt_response(struct bat_priv *bat_priv,
			     struct tt_query_packet *tt_request)
1736
{
1737
	if (batadv_is_my_mac(tt_request->dst)) {
1738
		/* don't answer backbone gws! */
1739
		if (batadv_bla_is_backbone_gw_orig(bat_priv, tt_request->src))
1740 1741
			return true;

1742
		return batadv_send_my_tt_response(bat_priv, tt_request);
1743
	} else {
1744
		return batadv_send_other_tt_response(bat_priv, tt_request);
1745
	}
1746 1747
}

1748 1749 1750 1751
static void _batadv_tt_update_changes(struct bat_priv *bat_priv,
				      struct orig_node *orig_node,
				      struct tt_change *tt_change,
				      uint16_t tt_num_changes, uint8_t ttvn)
1752 1753
{
	int i;
1754
	int roams;
1755 1756

	for (i = 0; i < tt_num_changes; i++) {
1757
		if ((tt_change + i)->flags & TT_CLIENT_DEL) {
1758 1759 1760
			roams = (tt_change + i)->flags & TT_CLIENT_ROAM;
			batadv_tt_global_del(bat_priv, orig_node,
					     (tt_change + i)->addr,
1761 1762
					     "tt removed by changes",
					     roams);
1763 1764
		} else {
			if (!batadv_tt_global_add(bat_priv, orig_node,
1765 1766
						  (tt_change + i)->addr,
						  (tt_change + i)->flags, ttvn))
1767 1768 1769 1770 1771 1772 1773
				/* In case of problem while storing a
				 * global_entry, we stop the updating
				 * procedure without committing the
				 * ttvn change. This will avoid to send
				 * corrupted data on tt_request
				 */
				return;
1774
		}
1775
	}
1776
	orig_node->tt_initialised = true;
1777 1778
}

1779 1780
static void batadv_tt_fill_gtable(struct bat_priv *bat_priv,
				  struct tt_query_packet *tt_response)
1781 1782 1783
{
	struct orig_node *orig_node = NULL;

1784
	orig_node = batadv_orig_hash_find(bat_priv, tt_response->src);
1785 1786 1787 1788
	if (!orig_node)
		goto out;

	/* Purge the old table first.. */
1789
	batadv_tt_global_del_orig(bat_priv, orig_node, "Received full table");
1790

1791 1792 1793 1794
	_batadv_tt_update_changes(bat_priv, orig_node,
				  (struct tt_change *)(tt_response + 1),
				  ntohs(tt_response->tt_data),
				  tt_response->ttvn);
1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805

	spin_lock_bh(&orig_node->tt_buff_lock);
	kfree(orig_node->tt_buff);
	orig_node->tt_buff_len = 0;
	orig_node->tt_buff = NULL;
	spin_unlock_bh(&orig_node->tt_buff_lock);

	atomic_set(&orig_node->last_ttvn, tt_response->ttvn);

out:
	if (orig_node)
1806
		batadv_orig_node_free_ref(orig_node);
1807 1808
}

1809 1810 1811 1812
static void batadv_tt_update_changes(struct bat_priv *bat_priv,
				     struct orig_node *orig_node,
				     uint16_t tt_num_changes, uint8_t ttvn,
				     struct tt_change *tt_change)
1813
{
1814 1815
	_batadv_tt_update_changes(bat_priv, orig_node, tt_change,
				  tt_num_changes, ttvn);
1816

1817 1818
	batadv_tt_save_orig_buffer(bat_priv, orig_node,
				   (unsigned char *)tt_change, tt_num_changes);
1819 1820 1821
	atomic_set(&orig_node->last_ttvn, ttvn);
}

1822
bool batadv_is_my_client(struct bat_priv *bat_priv, const uint8_t *addr)
1823
{
1824 1825
	struct tt_local_entry *tt_local_entry = NULL;
	bool ret = false;
1826

1827
	tt_local_entry = batadv_tt_local_hash_find(bat_priv, addr);
1828 1829
	if (!tt_local_entry)
		goto out;
1830
	/* Check if the client has been logically deleted (but is kept for
1831 1832
	 * consistency purpose)
	 */
1833
	if (tt_local_entry->common.flags & TT_CLIENT_PENDING)
1834
		goto out;
1835 1836
	ret = true;
out:
1837
	if (tt_local_entry)
1838
		batadv_tt_local_entry_free_ref(tt_local_entry);
1839
	return ret;
1840 1841
}

1842 1843
void batadv_handle_tt_response(struct bat_priv *bat_priv,
			       struct tt_query_packet *tt_response)
1844 1845 1846 1847
{
	struct tt_req_node *node, *safe;
	struct orig_node *orig_node = NULL;

1848 1849 1850 1851 1852
	batadv_dbg(DBG_TT, bat_priv,
		   "Received TT_RESPONSE from %pM for ttvn %d t_size: %d [%c]\n",
		   tt_response->src, tt_response->ttvn,
		   ntohs(tt_response->tt_data),
		   (tt_response->flags & TT_FULL_TABLE ? 'F' : '.'));
1853

1854
	/* we should have never asked a backbone gw */
1855
	if (batadv_bla_is_backbone_gw_orig(bat_priv, tt_response->src))
1856 1857
		goto out;

1858
	orig_node = batadv_orig_hash_find(bat_priv, tt_response->src);
1859 1860 1861 1862
	if (!orig_node)
		goto out;

	if (tt_response->flags & TT_FULL_TABLE)
1863
		batadv_tt_fill_gtable(bat_priv, tt_response);
1864
	else
1865 1866 1867 1868
		batadv_tt_update_changes(bat_priv, orig_node,
					 ntohs(tt_response->tt_data),
					 tt_response->ttvn,
					 (struct tt_change *)(tt_response + 1));
1869 1870 1871 1872

	/* Delete the tt_req_node from pending tt_requests list */
	spin_lock_bh(&bat_priv->tt_req_list_lock);
	list_for_each_entry_safe(node, safe, &bat_priv->tt_req_list, list) {
1873
		if (!batadv_compare_eth(node->addr, tt_response->src))
1874 1875 1876 1877 1878 1879 1880
			continue;
		list_del(&node->list);
		kfree(node);
	}
	spin_unlock_bh(&bat_priv->tt_req_list_lock);

	/* Recalculate the CRC for this orig_node and store it */
1881
	orig_node->tt_crc = batadv_tt_global_crc(bat_priv, orig_node);
1882
	/* Roaming phase is over: tables are in sync again. I can
1883 1884
	 * unset the flag
	 */
1885
	orig_node->tt_poss_change = false;
1886 1887
out:
	if (orig_node)
1888
		batadv_orig_node_free_ref(orig_node);
1889 1890
}

1891
int batadv_tt_init(struct bat_priv *bat_priv)
1892
{
1893
	int ret;
1894

1895
	ret = batadv_tt_local_init(bat_priv);
1896 1897 1898
	if (ret < 0)
		return ret;

1899
	ret = batadv_tt_global_init(bat_priv);
1900 1901
	if (ret < 0)
		return ret;
1902

1903
	batadv_tt_start_timer(bat_priv);
1904 1905 1906 1907

	return 1;
}

1908
static void batadv_tt_roam_list_free(struct bat_priv *bat_priv)
1909
{
1910
	struct tt_roam_node *node, *safe;
1911

1912
	spin_lock_bh(&bat_priv->tt_roam_list_lock);
1913

1914 1915 1916 1917 1918 1919 1920 1921
	list_for_each_entry_safe(node, safe, &bat_priv->tt_roam_list, list) {
		list_del(&node->list);
		kfree(node);
	}

	spin_unlock_bh(&bat_priv->tt_roam_list_lock);
}

1922
static void batadv_tt_roam_purge(struct bat_priv *bat_priv)
1923 1924 1925 1926 1927
{
	struct tt_roam_node *node, *safe;

	spin_lock_bh(&bat_priv->tt_roam_list_lock);
	list_for_each_entry_safe(node, safe, &bat_priv->tt_roam_list, list) {
1928
		if (!batadv_has_timed_out(node->first_time, ROAMING_MAX_TIME))
1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940
			continue;

		list_del(&node->list);
		kfree(node);
	}
	spin_unlock_bh(&bat_priv->tt_roam_list_lock);
}

/* This function checks whether the client already reached the
 * maximum number of possible roaming phases. In this case the ROAMING_ADV
 * will not be sent.
 *
1941 1942
 * returns true if the ROAMING_ADV can be sent, false otherwise
 */
1943 1944
static bool batadv_tt_check_roam_count(struct bat_priv *bat_priv,
				       uint8_t *client)
1945 1946 1947 1948 1949 1950
{
	struct tt_roam_node *tt_roam_node;
	bool ret = false;

	spin_lock_bh(&bat_priv->tt_roam_list_lock);
	/* The new tt_req will be issued only if I'm not waiting for a
1951 1952
	 * reply from the same orig_node yet
	 */
1953
	list_for_each_entry(tt_roam_node, &bat_priv->tt_roam_list, list) {
1954
		if (!batadv_compare_eth(tt_roam_node->addr, client))
1955 1956
			continue;

1957 1958
		if (batadv_has_timed_out(tt_roam_node->first_time,
					 ROAMING_MAX_TIME))
1959 1960
			continue;

1961
		if (!batadv_atomic_dec_not_zero(&tt_roam_node->counter))
1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985
			/* Sorry, you roamed too many times! */
			goto unlock;
		ret = true;
		break;
	}

	if (!ret) {
		tt_roam_node = kmalloc(sizeof(*tt_roam_node), GFP_ATOMIC);
		if (!tt_roam_node)
			goto unlock;

		tt_roam_node->first_time = jiffies;
		atomic_set(&tt_roam_node->counter, ROAMING_MAX_COUNT - 1);
		memcpy(tt_roam_node->addr, client, ETH_ALEN);

		list_add(&tt_roam_node->list, &bat_priv->tt_roam_list);
		ret = true;
	}

unlock:
	spin_unlock_bh(&bat_priv->tt_roam_list_lock);
	return ret;
}

1986 1987
static void batadv_send_roam_adv(struct bat_priv *bat_priv, uint8_t *client,
				 struct orig_node *orig_node)
1988 1989 1990 1991 1992 1993 1994 1995
{
	struct neigh_node *neigh_node = NULL;
	struct sk_buff *skb = NULL;
	struct roam_adv_packet *roam_adv_packet;
	int ret = 1;
	struct hard_iface *primary_if;

	/* before going on we have to check whether the client has
1996 1997
	 * already roamed to us too many times
	 */
1998
	if (!batadv_tt_check_roam_count(bat_priv, client))
1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009
		goto out;

	skb = dev_alloc_skb(sizeof(struct roam_adv_packet) + ETH_HLEN);
	if (!skb)
		goto out;

	skb_reserve(skb, ETH_HLEN);

	roam_adv_packet = (struct roam_adv_packet *)skb_put(skb,
					sizeof(struct roam_adv_packet));

2010
	roam_adv_packet->header.packet_type = BAT_ROAM_ADV;
2011
	roam_adv_packet->header.version = BATADV_COMPAT_VERSION;
2012
	roam_adv_packet->header.ttl = TTL;
2013
	primary_if = batadv_primary_if_get_selected(bat_priv);
2014 2015 2016
	if (!primary_if)
		goto out;
	memcpy(roam_adv_packet->src, primary_if->net_dev->dev_addr, ETH_ALEN);
2017
	batadv_hardif_free_ref(primary_if);
2018 2019 2020
	memcpy(roam_adv_packet->dst, orig_node->orig, ETH_ALEN);
	memcpy(roam_adv_packet->client, client, ETH_ALEN);

2021
	neigh_node = batadv_orig_node_get_router(orig_node);
2022 2023 2024
	if (!neigh_node)
		goto out;

2025 2026 2027
	batadv_dbg(DBG_TT, bat_priv,
		   "Sending ROAMING_ADV to %pM (client %pM) via %pM\n",
		   orig_node->orig, client, neigh_node->addr);
2028

2029 2030
	batadv_inc_counter(bat_priv, BAT_CNT_TT_ROAM_ADV_TX);

2031
	batadv_send_skb_packet(skb, neigh_node->if_incoming, neigh_node->addr);
2032 2033 2034 2035
	ret = 0;

out:
	if (neigh_node)
2036
		batadv_neigh_node_free_ref(neigh_node);
2037 2038 2039
	if (ret)
		kfree_skb(skb);
	return;
2040 2041
}

2042
static void batadv_tt_purge(struct work_struct *work)
2043 2044 2045 2046 2047 2048
{
	struct delayed_work *delayed_work =
		container_of(work, struct delayed_work, work);
	struct bat_priv *bat_priv =
		container_of(delayed_work, struct bat_priv, tt_work);

2049 2050 2051 2052
	batadv_tt_local_purge(bat_priv);
	batadv_tt_global_roam_purge(bat_priv);
	batadv_tt_req_purge(bat_priv);
	batadv_tt_roam_purge(bat_priv);
2053

2054
	batadv_tt_start_timer(bat_priv);
2055
}
2056

2057
void batadv_tt_free(struct bat_priv *bat_priv)
2058 2059 2060
{
	cancel_delayed_work_sync(&bat_priv->tt_work);

2061 2062 2063 2064 2065
	batadv_tt_local_table_free(bat_priv);
	batadv_tt_global_table_free(bat_priv);
	batadv_tt_req_list_free(bat_priv);
	batadv_tt_changes_list_free(bat_priv);
	batadv_tt_roam_list_free(bat_priv);
2066 2067 2068

	kfree(bat_priv->tt_buff);
}
2069

2070
/* This function will enable or disable the specified flags for all the entries
2071 2072
 * in the given hash table and returns the number of modified entries
 */
2073 2074
static uint16_t batadv_tt_set_flags(struct hashtable_t *hash, uint16_t flags,
				    bool enable)
2075
{
2076
	uint32_t i;
2077
	uint16_t changed_num = 0;
2078 2079
	struct hlist_head *head;
	struct hlist_node *node;
2080
	struct tt_common_entry *tt_common_entry;
2081 2082

	if (!hash)
2083
		goto out;
2084 2085 2086 2087 2088

	for (i = 0; i < hash->size; i++) {
		head = &hash->table[i];

		rcu_read_lock();
2089
		hlist_for_each_entry_rcu(tt_common_entry, node,
2090
					 head, hash_entry) {
2091 2092 2093 2094 2095 2096 2097 2098 2099 2100
			if (enable) {
				if ((tt_common_entry->flags & flags) == flags)
					continue;
				tt_common_entry->flags |= flags;
			} else {
				if (!(tt_common_entry->flags & flags))
					continue;
				tt_common_entry->flags &= ~flags;
			}
			changed_num++;
2101 2102 2103
		}
		rcu_read_unlock();
	}
2104 2105
out:
	return changed_num;
2106 2107 2108
}

/* Purge out all the tt local entries marked with TT_CLIENT_PENDING */
2109
static void batadv_tt_local_purge_pending_clients(struct bat_priv *bat_priv)
2110 2111
{
	struct hashtable_t *hash = bat_priv->tt_local_hash;
2112
	struct tt_common_entry *tt_common_entry;
2113 2114 2115 2116
	struct tt_local_entry *tt_local_entry;
	struct hlist_node *node, *node_tmp;
	struct hlist_head *head;
	spinlock_t *list_lock; /* protects write access to the hash lists */
2117
	uint32_t i;
2118 2119 2120 2121 2122 2123 2124 2125 2126

	if (!hash)
		return;

	for (i = 0; i < hash->size; i++) {
		head = &hash->table[i];
		list_lock = &hash->list_locks[i];

		spin_lock_bh(list_lock);
2127
		hlist_for_each_entry_safe(tt_common_entry, node, node_tmp,
2128
					  head, hash_entry) {
2129
			if (!(tt_common_entry->flags & TT_CLIENT_PENDING))
2130 2131
				continue;

2132 2133 2134
			batadv_dbg(DBG_TT, bat_priv,
				   "Deleting local tt entry (%pM): pending\n",
				   tt_common_entry->addr);
2135 2136 2137

			atomic_dec(&bat_priv->num_local_tt);
			hlist_del_rcu(node);
2138 2139 2140
			tt_local_entry = container_of(tt_common_entry,
						      struct tt_local_entry,
						      common);
2141
			batadv_tt_local_entry_free_ref(tt_local_entry);
2142 2143 2144 2145 2146 2147
		}
		spin_unlock_bh(list_lock);
	}

}

2148 2149 2150
static int batadv_tt_commit_changes(struct bat_priv *bat_priv,
				    unsigned char **packet_buff,
				    int *packet_buff_len, int packet_min_len)
2151
{
2152 2153 2154 2155 2156
	uint16_t changed_num = 0;

	if (atomic_read(&bat_priv->tt_local_changes) < 1)
		return -ENOENT;

2157 2158
	changed_num = batadv_tt_set_flags(bat_priv->tt_local_hash,
					  TT_CLIENT_NEW, false);
2159 2160

	/* all reset entries have to be counted as local entries */
2161
	atomic_add(changed_num, &bat_priv->num_local_tt);
2162
	batadv_tt_local_purge_pending_clients(bat_priv);
2163
	bat_priv->tt_crc = batadv_tt_local_crc(bat_priv);
2164 2165 2166

	/* Increment the TTVN only once per OGM interval */
	atomic_inc(&bat_priv->ttvn);
2167 2168 2169
	batadv_dbg(DBG_TT, bat_priv,
		   "Local changes committed, updating to ttvn %u\n",
		   (uint8_t)atomic_read(&bat_priv->ttvn));
2170
	bat_priv->tt_poss_change = false;
2171 2172 2173 2174

	/* reset the sending counter */
	atomic_set(&bat_priv->tt_ogm_append_cnt, TT_OGM_APPEND_MAX);

2175 2176
	return batadv_tt_changes_fill_buff(bat_priv, packet_buff,
					   packet_buff_len, packet_min_len);
2177 2178 2179 2180 2181 2182 2183 2184 2185 2186
}

/* when calling this function (hard_iface == primary_if) has to be true */
int batadv_tt_append_diff(struct bat_priv *bat_priv,
			  unsigned char **packet_buff, int *packet_buff_len,
			  int packet_min_len)
{
	int tt_num_changes;

	/* if at least one change happened */
2187 2188 2189
	tt_num_changes = batadv_tt_commit_changes(bat_priv, packet_buff,
						  packet_buff_len,
						  packet_min_len);
2190 2191 2192

	/* if the changes have been sent often enough */
	if ((tt_num_changes < 0) &&
2193
	    (!batadv_atomic_dec_not_zero(&bat_priv->tt_ogm_append_cnt))) {
2194 2195
		batadv_tt_realloc_packet_buff(packet_buff, packet_buff_len,
					      packet_min_len, packet_min_len);
2196 2197 2198 2199
		tt_num_changes = 0;
	}

	return tt_num_changes;
2200
}
2201

2202 2203
bool batadv_is_ap_isolated(struct bat_priv *bat_priv, uint8_t *src,
			   uint8_t *dst)
2204 2205 2206
{
	struct tt_local_entry *tt_local_entry = NULL;
	struct tt_global_entry *tt_global_entry = NULL;
2207
	bool ret = false;
2208 2209

	if (!atomic_read(&bat_priv->ap_isolation))
2210
		goto out;
2211

2212
	tt_local_entry = batadv_tt_local_hash_find(bat_priv, dst);
2213 2214 2215
	if (!tt_local_entry)
		goto out;

2216
	tt_global_entry = batadv_tt_global_hash_find(bat_priv, src);
2217 2218 2219
	if (!tt_global_entry)
		goto out;

2220
	if (!_batadv_is_ap_isolated(tt_local_entry, tt_global_entry))
2221 2222
		goto out;

2223
	ret = true;
2224 2225 2226

out:
	if (tt_global_entry)
2227
		batadv_tt_global_entry_free_ref(tt_global_entry);
2228
	if (tt_local_entry)
2229
		batadv_tt_local_entry_free_ref(tt_local_entry);
2230 2231
	return ret;
}
2232

2233 2234 2235 2236
void batadv_tt_update_orig(struct bat_priv *bat_priv,
			   struct orig_node *orig_node,
			   const unsigned char *tt_buff, uint8_t tt_num_changes,
			   uint8_t ttvn, uint16_t tt_crc)
2237 2238 2239 2240
{
	uint8_t orig_ttvn = (uint8_t)atomic_read(&orig_node->last_ttvn);
	bool full_table = true;

2241
	/* don't care about a backbone gateways updates. */
2242
	if (batadv_bla_is_backbone_gw_orig(bat_priv, orig_node->orig))
2243 2244
		return;

2245
	/* orig table not initialised AND first diff is in the OGM OR the ttvn
2246 2247
	 * increased by one -> we can apply the attached changes
	 */
2248 2249
	if ((!orig_node->tt_initialised && ttvn == 1) ||
	    ttvn - orig_ttvn == 1) {
2250 2251
		/* the OGM could not contain the changes due to their size or
		 * because they have already been sent TT_OGM_APPEND_MAX times.
2252 2253
		 * In this case send a tt request
		 */
2254 2255 2256 2257 2258
		if (!tt_num_changes) {
			full_table = false;
			goto request_table;
		}

2259 2260
		batadv_tt_update_changes(bat_priv, orig_node, tt_num_changes,
					 ttvn, (struct tt_change *)tt_buff);
2261 2262 2263

		/* Even if we received the precomputed crc with the OGM, we
		 * prefer to recompute it to spot any possible inconsistency
2264 2265
		 * in the global table
		 */
2266
		orig_node->tt_crc = batadv_tt_global_crc(bat_priv, orig_node);
2267 2268 2269 2270 2271 2272 2273 2274

		/* The ttvn alone is not enough to guarantee consistency
		 * because a single value could represent different states
		 * (due to the wrap around). Thus a node has to check whether
		 * the resulting table (after applying the changes) is still
		 * consistent or not. E.g. a node could disconnect while its
		 * ttvn is X and reconnect on ttvn = X + TTVN_MAX: in this case
		 * checking the CRC value is mandatory to detect the
2275 2276
		 * inconsistency
		 */
2277 2278 2279 2280
		if (orig_node->tt_crc != tt_crc)
			goto request_table;

		/* Roaming phase is over: tables are in sync again. I can
2281 2282
		 * unset the flag
		 */
2283 2284 2285
		orig_node->tt_poss_change = false;
	} else {
		/* if we missed more than one change or our tables are not
2286 2287
		 * in sync anymore -> request fresh tt data
		 */
2288 2289
		if (!orig_node->tt_initialised || ttvn != orig_ttvn ||
		    orig_node->tt_crc != tt_crc) {
2290
request_table:
2291 2292 2293 2294
			batadv_dbg(DBG_TT, bat_priv,
				   "TT inconsistency for %pM. Need to retrieve the correct information (ttvn: %u last_ttvn: %u crc: %u last_crc: %u num_changes: %u)\n",
				   orig_node->orig, ttvn, orig_ttvn, tt_crc,
				   orig_node->tt_crc, tt_num_changes);
2295 2296
			batadv_send_tt_request(bat_priv, orig_node, ttvn,
					       tt_crc, full_table);
2297 2298 2299 2300
			return;
		}
	}
}
2301 2302 2303 2304 2305

/* returns true whether we know that the client has moved from its old
 * originator to another one. This entry is kept is still kept for consistency
 * purposes
 */
2306 2307
bool batadv_tt_global_client_is_roaming(struct bat_priv *bat_priv,
					uint8_t *addr)
2308 2309 2310 2311
{
	struct tt_global_entry *tt_global_entry;
	bool ret = false;

2312
	tt_global_entry = batadv_tt_global_hash_find(bat_priv, addr);
2313 2314 2315 2316
	if (!tt_global_entry)
		goto out;

	ret = tt_global_entry->common.flags & TT_CLIENT_ROAM;
2317
	batadv_tt_global_entry_free_ref(tt_global_entry);
2318 2319 2320
out:
	return ret;
}