translation-table.c 62.9 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 163 164 165 166
{
	struct tt_change_node *tt_change_node;

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

	if (!tt_change_node)
		return;

167
	tt_change_node->change.flags = flags;
168 169 170 171 172 173 174 175 176 177 178
	memcpy(tt_change_node->change.addr, addr, ETH_ALEN);

	spin_lock_bh(&bat_priv->tt_changes_list_lock);
	/* track the change in the OGMinterval list */
	list_add_tail(&tt_change_node->list, &bat_priv->tt_changes_list);
	atomic_inc(&bat_priv->tt_local_changes);
	spin_unlock_bh(&bat_priv->tt_changes_list_lock);

	atomic_set(&bat_priv->tt_ogm_append_cnt, 0);
}

179
int batadv_tt_len(int changes_num)
180 181 182 183
{
	return changes_num * sizeof(struct tt_change);
}

184
static int batadv_tt_local_init(struct bat_priv *bat_priv)
185
{
186
	if (bat_priv->tt_local_hash)
187
		return 0;
188

189
	bat_priv->tt_local_hash = batadv_hash_new(1024);
190

191
	if (!bat_priv->tt_local_hash)
192
		return -ENOMEM;
193

194
	return 0;
195 196
}

197 198
void batadv_tt_local_add(struct net_device *soft_iface, const uint8_t *addr,
			 int ifindex)
199 200
{
	struct bat_priv *bat_priv = netdev_priv(soft_iface);
201 202
	struct tt_local_entry *tt_local_entry = NULL;
	struct tt_global_entry *tt_global_entry = NULL;
203 204 205
	struct hlist_head *head;
	struct hlist_node *node;
	struct tt_orig_list_entry *orig_entry;
206
	int hash_added;
207

208
	tt_local_entry = batadv_tt_local_hash_find(bat_priv, addr);
209

210 211
	if (tt_local_entry) {
		tt_local_entry->last_seen = jiffies;
212 213
		/* possibly unset the TT_CLIENT_PENDING flag */
		tt_local_entry->common.flags &= ~TT_CLIENT_PENDING;
214
		goto out;
215 216
	}

217
	tt_local_entry = kmalloc(sizeof(*tt_local_entry), GFP_ATOMIC);
218
	if (!tt_local_entry)
219
		goto out;
220

221 222 223
	batadv_dbg(DBG_TT, bat_priv,
		   "Creating new local tt entry: %pM (ttvn: %d)\n", addr,
		   (uint8_t)atomic_read(&bat_priv->ttvn));
224

225 226
	memcpy(tt_local_entry->common.addr, addr, ETH_ALEN);
	tt_local_entry->common.flags = NO_FLAGS;
227
	if (batadv_is_wifi_iface(ifindex))
228 229 230
		tt_local_entry->common.flags |= TT_CLIENT_WIFI;
	atomic_set(&tt_local_entry->common.refcount, 2);
	tt_local_entry->last_seen = jiffies;
231 232

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

236 237
	/* 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
238 239
	 * (consistency check)
	 */
240 241
	tt_local_entry->common.flags |= TT_CLIENT_NEW;

242
	hash_added = batadv_hash_add(bat_priv->tt_local_hash, batadv_compare_tt,
243 244
				     batadv_choose_orig,
				     &tt_local_entry->common,
245
				     &tt_local_entry->common.hash_entry);
246 247 248

	if (unlikely(hash_added != 0)) {
		/* remove the reference for the hash */
249
		batadv_tt_local_entry_free_ref(tt_local_entry);
250 251 252
		goto out;
	}

253
	batadv_tt_local_event(bat_priv, addr, tt_local_entry->common.flags);
254

255
	/* remove address from global hash if present */
256
	tt_global_entry = batadv_tt_global_hash_find(bat_priv, addr);
257

258 259
	/* Check whether it is a roaming! */
	if (tt_global_entry) {
260 261 262 263 264 265
		/* 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;

266 267 268
			batadv_send_roam_adv(bat_priv,
					     tt_global_entry->common.addr,
					     orig_entry->orig_node);
269 270 271 272 273
		}
		rcu_read_unlock();
		/* The global entry has to be marked as ROAMING and
		 * has to be kept for consistency purpose
		 */
D
David S. Miller 已提交
274
		tt_global_entry->common.flags |= TT_CLIENT_ROAM;
275
		tt_global_entry->roam_at = jiffies;
276 277 278
	}
out:
	if (tt_local_entry)
279
		batadv_tt_local_entry_free_ref(tt_local_entry);
280
	if (tt_global_entry)
281
		batadv_tt_global_entry_free_ref(tt_global_entry);
282 283
}

284 285 286 287
static void batadv_tt_realloc_packet_buff(unsigned char **packet_buff,
					  int *packet_buff_len,
					  int min_packet_len,
					  int new_packet_len)
288 289 290 291 292 293 294 295 296 297 298 299 300 301
{
	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;
	}
}

302 303 304 305
static void batadv_tt_prepare_packet_buff(struct bat_priv *bat_priv,
					  unsigned char **packet_buff,
					  int *packet_buff_len,
					  int min_packet_len)
306 307 308 309
{
	struct hard_iface *primary_if;
	int req_len;

310
	primary_if = batadv_primary_if_get_selected(bat_priv);
311 312

	req_len = min_packet_len;
313
	req_len += batadv_tt_len(atomic_read(&bat_priv->tt_local_changes));
314 315 316 317 318 319 320

	/* 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;

321 322
	batadv_tt_realloc_packet_buff(packet_buff, packet_buff_len,
				      min_packet_len, req_len);
323 324

	if (primary_if)
325
		batadv_hardif_free_ref(primary_if);
326 327
}

328 329 330 331
static int batadv_tt_changes_fill_buff(struct bat_priv *bat_priv,
				       unsigned char **packet_buff,
				       int *packet_buff_len,
				       int min_packet_len)
332
{
333
	struct tt_change_node *entry, *safe;
334 335 336
	int count = 0, tot_changes = 0, new_len;
	unsigned char *tt_buff;

337 338
	batadv_tt_prepare_packet_buff(bat_priv, packet_buff,
				      packet_buff_len, min_packet_len);
339

340 341 342 343
	new_len = *packet_buff_len - min_packet_len;
	tt_buff = *packet_buff + min_packet_len;

	if (new_len > 0)
344
		tot_changes = new_len / batadv_tt_len(1);
345

346 347
	spin_lock_bh(&bat_priv->tt_changes_list_lock);
	atomic_set(&bat_priv->tt_local_changes, 0);
348

349
	list_for_each_entry_safe(entry, safe, &bat_priv->tt_changes_list,
350
				 list) {
351
		if (count < tot_changes) {
352
			memcpy(tt_buff + batadv_tt_len(count),
353
			       &entry->change, sizeof(struct tt_change));
354 355
			count++;
		}
356 357
		list_del(&entry->list);
		kfree(entry);
358
	}
359 360 361 362 363 364 365
	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;
366 367 368
	/* 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
369 370
		 * instead of providing the diff
		 */
371
		bat_priv->tt_buff = kmalloc(new_len, GFP_ATOMIC);
372
		if (bat_priv->tt_buff) {
373 374
			memcpy(bat_priv->tt_buff, tt_buff, new_len);
			bat_priv->tt_buff_len = new_len;
375 376 377
		}
	}
	spin_unlock_bh(&bat_priv->tt_buff_lock);
378

379
	return count;
380 381
}

382
int batadv_tt_local_seq_print_text(struct seq_file *seq, void *offset)
383 384 385
{
	struct net_device *net_dev = (struct net_device *)seq->private;
	struct bat_priv *bat_priv = netdev_priv(net_dev);
386
	struct hashtable_t *hash = bat_priv->tt_local_hash;
387
	struct tt_common_entry *tt_common_entry;
388
	struct hard_iface *primary_if;
389
	struct hlist_node *node;
390
	struct hlist_head *head;
391 392
	uint32_t i;
	int ret = 0;
393

394
	primary_if = batadv_primary_if_get_selected(bat_priv);
395
	if (!primary_if) {
396 397
		ret = seq_printf(seq,
				 "BATMAN mesh %s disabled - please specify interfaces to enable it\n",
398 399 400
				 net_dev->name);
		goto out;
	}
401

402
	if (primary_if->if_status != IF_ACTIVE) {
403 404
		ret = seq_printf(seq,
				 "BATMAN mesh %s disabled - primary interface not active\n",
405 406
				 net_dev->name);
		goto out;
407 408
	}

409 410
	seq_printf(seq,
		   "Locally retrieved addresses (from %s) announced via TT (TTVN: %u):\n",
411
		   net_dev->name, (uint8_t)atomic_read(&bat_priv->ttvn));
412 413 414 415

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

416
		rcu_read_lock();
417
		hlist_for_each_entry_rcu(tt_common_entry, node,
418
					 head, hash_entry) {
419
			seq_printf(seq, " * %pM [%c%c%c%c%c]\n",
420 421 422 423 424 425 426 427 428 429 430
				   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' : '.'));
431
		}
432
		rcu_read_unlock();
433
	}
434 435
out:
	if (primary_if)
436
		batadv_hardif_free_ref(primary_if);
437
	return ret;
438 439
}

440 441 442
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)
443
{
444 445
	batadv_tt_local_event(bat_priv, tt_local_entry->common.addr,
			      tt_local_entry->common.flags | flags);
446

447 448
	/* 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
449 450
	 * response issued before the net ttvn increment (consistency check)
	 */
451
	tt_local_entry->common.flags |= TT_CLIENT_PENDING;
452

453 454 455
	batadv_dbg(DBG_TT, bat_priv,
		   "Local tt entry (%pM) pending to be removed: %s\n",
		   tt_local_entry->common.addr, message);
456 457
}

458 459
void batadv_tt_local_remove(struct bat_priv *bat_priv, const uint8_t *addr,
			    const char *message, bool roaming)
460
{
461
	struct tt_local_entry *tt_local_entry = NULL;
462

463
	tt_local_entry = batadv_tt_local_hash_find(bat_priv, addr);
464 465 466
	if (!tt_local_entry)
		goto out;

467 468 469
	batadv_tt_local_set_pending(bat_priv, tt_local_entry, TT_CLIENT_DEL |
				    (roaming ? TT_CLIENT_ROAM : NO_FLAGS),
				    message);
470 471
out:
	if (tt_local_entry)
472
		batadv_tt_local_entry_free_ref(tt_local_entry);
473 474
}

475
static void batadv_tt_local_purge(struct bat_priv *bat_priv)
476
{
477 478
	struct hashtable_t *hash = bat_priv->tt_local_hash;
	struct tt_local_entry *tt_local_entry;
479
	struct tt_common_entry *tt_common_entry;
480
	struct hlist_node *node, *node_tmp;
481
	struct hlist_head *head;
482
	spinlock_t *list_lock; /* protects write access to the hash lists */
483
	uint32_t i;
484 485 486

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

489
		spin_lock_bh(list_lock);
490
		hlist_for_each_entry_safe(tt_common_entry, node, node_tmp,
491
					  head, hash_entry) {
492 493 494 495
			tt_local_entry = container_of(tt_common_entry,
						      struct tt_local_entry,
						      common);
			if (tt_local_entry->common.flags & TT_CLIENT_NOPURGE)
496
				continue;
497

498
			/* entry already marked for deletion */
499
			if (tt_local_entry->common.flags & TT_CLIENT_PENDING)
500 501
				continue;

502 503
			if (!batadv_has_timed_out(tt_local_entry->last_seen,
						  TT_LOCAL_TIMEOUT))
504 505
				continue;

506 507
			batadv_tt_local_set_pending(bat_priv, tt_local_entry,
						    TT_CLIENT_DEL, "timed out");
508
		}
509
		spin_unlock_bh(list_lock);
510 511 512 513
	}

}

514
static void batadv_tt_local_table_free(struct bat_priv *bat_priv)
515
{
516 517
	struct hashtable_t *hash;
	spinlock_t *list_lock; /* protects write access to the hash lists */
518
	struct tt_common_entry *tt_common_entry;
519
	struct tt_local_entry *tt_local_entry;
520 521
	struct hlist_node *node, *node_tmp;
	struct hlist_head *head;
522
	uint32_t i;
523

524
	if (!bat_priv->tt_local_hash)
525 526
		return;

527 528 529 530 531 532 533
	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);
534
		hlist_for_each_entry_safe(tt_common_entry, node, node_tmp,
535 536
					  head, hash_entry) {
			hlist_del_rcu(node);
537 538 539
			tt_local_entry = container_of(tt_common_entry,
						      struct tt_local_entry,
						      common);
540
			batadv_tt_local_entry_free_ref(tt_local_entry);
541 542 543 544
		}
		spin_unlock_bh(list_lock);
	}

545
	batadv_hash_destroy(hash);
546

547
	bat_priv->tt_local_hash = NULL;
548 549
}

550
static int batadv_tt_global_init(struct bat_priv *bat_priv)
551
{
552
	if (bat_priv->tt_global_hash)
553
		return 0;
554

555
	bat_priv->tt_global_hash = batadv_hash_new(1024);
556

557
	if (!bat_priv->tt_global_hash)
558
		return -ENOMEM;
559

560
	return 0;
561 562
}

563
static void batadv_tt_changes_list_free(struct bat_priv *bat_priv)
564
{
565
	struct tt_change_node *entry, *safe;
566

567
	spin_lock_bh(&bat_priv->tt_changes_list_lock);
568

569 570 571 572 573
	list_for_each_entry_safe(entry, safe, &bat_priv->tt_changes_list,
				 list) {
		list_del(&entry->list);
		kfree(entry);
	}
574

575 576 577
	atomic_set(&bat_priv->tt_local_changes, 0);
	spin_unlock_bh(&bat_priv->tt_changes_list_lock);
}
578

579 580 581
/* find out if an orig_node is already in the list of a tt_global_entry.
 * returns 1 if found, 0 otherwise
 */
582 583
static bool batadv_tt_global_entry_has_orig(const struct tt_global_entry *entry,
					    const struct orig_node *orig_node)
584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601
{
	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;
}

602 603 604
static void
batadv_tt_global_add_orig_entry(struct tt_global_entry *tt_global_entry,
				struct orig_node *orig_node, int ttvn)
605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623
{
	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);
}

624
/* caller must hold orig_node refcount */
625 626 627
int batadv_tt_global_add(struct bat_priv *bat_priv, struct orig_node *orig_node,
			 const unsigned char *tt_addr, uint8_t ttvn,
			 bool roaming, bool wifi)
628
{
629
	struct tt_global_entry *tt_global_entry = NULL;
630
	int ret = 0;
631
	int hash_added;
632
	struct tt_common_entry *common;
633

634
	tt_global_entry = batadv_tt_global_hash_find(bat_priv, tt_addr);
635 636

	if (!tt_global_entry) {
637 638
		tt_global_entry = kzalloc(sizeof(*tt_global_entry),
					  GFP_ATOMIC);
639
		if (!tt_global_entry)
640 641
			goto out;

642 643
		common = &tt_global_entry->common;
		memcpy(common->addr, tt_addr, ETH_ALEN);
644

645
		common->flags = NO_FLAGS;
646
		tt_global_entry->roam_at = 0;
647
		atomic_set(&common->refcount, 2);
648 649 650

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

652
		hash_added = batadv_hash_add(bat_priv->tt_global_hash,
653 654 655
					     batadv_compare_tt,
					     batadv_choose_orig, common,
					     &common->hash_entry);
656 657 658

		if (unlikely(hash_added != 0)) {
			/* remove the reference for the hash */
659
			batadv_tt_global_entry_free_ref(tt_global_entry);
660 661
			goto out_remove;
		}
662

663 664
		batadv_tt_global_add_orig_entry(tt_global_entry, orig_node,
						ttvn);
665
	} else {
666 667 668 669 670 671 672 673 674 675
		/* 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) {
676
			batadv_tt_global_del_orig_list(tt_global_entry);
677 678
			tt_global_entry->common.flags &= ~TT_CLIENT_ROAM;
			tt_global_entry->roam_at = 0;
679
		}
680

681 682 683 684
		if (!batadv_tt_global_entry_has_orig(tt_global_entry,
						     orig_node))
			batadv_tt_global_add_orig_entry(tt_global_entry,
							orig_node, ttvn);
685
	}
686

687
	if (wifi)
688
		tt_global_entry->common.flags |= TT_CLIENT_WIFI;
689

690 691 692
	batadv_dbg(DBG_TT, bat_priv,
		   "Creating new global tt entry: %pM (via %pM)\n",
		   tt_global_entry->common.addr, orig_node->orig);
693

694
out_remove:
695
	/* remove address from local hash if present */
696 697
	batadv_tt_local_remove(bat_priv, tt_global_entry->common.addr,
			       "global tt received", roaming);
698 699 700
	ret = 1;
out:
	if (tt_global_entry)
701
		batadv_tt_global_entry_free_ref(tt_global_entry);
702
	return ret;
703 704
}

705 706 707
/* print all orig nodes who announce the address for this global entry.
 * it is assumed that the caller holds rcu_read_lock();
 */
708 709 710
static void
batadv_tt_global_print_entry(struct tt_global_entry *tt_global_entry,
			     struct seq_file *seq)
711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733
{
	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' : '.'));
	}
}

734
int batadv_tt_global_seq_print_text(struct seq_file *seq, void *offset)
735 736 737
{
	struct net_device *net_dev = (struct net_device *)seq->private;
	struct bat_priv *bat_priv = netdev_priv(net_dev);
738
	struct hashtable_t *hash = bat_priv->tt_global_hash;
739
	struct tt_common_entry *tt_common_entry;
740
	struct tt_global_entry *tt_global_entry;
741
	struct hard_iface *primary_if;
742
	struct hlist_node *node;
743
	struct hlist_head *head;
744 745
	uint32_t i;
	int ret = 0;
746

747
	primary_if = batadv_primary_if_get_selected(bat_priv);
748
	if (!primary_if) {
749 750
		ret = seq_printf(seq,
				 "BATMAN mesh %s disabled - please specify interfaces to enable it\n",
751 752 753
				 net_dev->name);
		goto out;
	}
754

755
	if (primary_if->if_status != IF_ACTIVE) {
756 757
		ret = seq_printf(seq,
				 "BATMAN mesh %s disabled - primary interface not active\n",
758 759
				 net_dev->name);
		goto out;
760 761
	}

762 763
	seq_printf(seq,
		   "Globally announced TT entries received via the mesh %s\n",
764
		   net_dev->name);
765 766
	seq_printf(seq, "       %-13s %s       %-15s %s %s\n",
		   "Client", "(TTVN)", "Originator", "(Curr TTVN)", "Flags");
767 768 769 770

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

771
		rcu_read_lock();
772
		hlist_for_each_entry_rcu(tt_common_entry, node,
773
					 head, hash_entry) {
774 775 776
			tt_global_entry = container_of(tt_common_entry,
						       struct tt_global_entry,
						       common);
777
			batadv_tt_global_print_entry(tt_global_entry, seq);
778
		}
779
		rcu_read_unlock();
780
	}
781 782
out:
	if (primary_if)
783
		batadv_hardif_free_ref(primary_if);
784
	return ret;
785 786
}

787
/* deletes the orig list of a tt_global_entry */
788 789
static void
batadv_tt_global_del_orig_list(struct tt_global_entry *tt_global_entry)
790
{
791 792 793
	struct hlist_head *head;
	struct hlist_node *node, *safe;
	struct tt_orig_list_entry *orig_entry;
794

795 796 797 798
	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);
799
		batadv_tt_orig_list_entry_free_ref(orig_entry);
800 801
	}
	spin_unlock_bh(&tt_global_entry->list_lock);
802

803 804
}

805 806 807 808 809
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)
810 811 812 813 814 815 816 817 818
{
	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) {
819 820 821 822
			batadv_dbg(DBG_TT, bat_priv,
				   "Deleting %pM from global tt entry %pM: %s\n",
				   orig_node->orig,
				   tt_global_entry->common.addr, message);
823
			hlist_del_rcu(node);
824
			batadv_tt_orig_list_entry_free_ref(orig_entry);
825 826 827 828 829
		}
	}
	spin_unlock_bh(&tt_global_entry->list_lock);
}

830 831 832
static void batadv_tt_global_del_struct(struct bat_priv *bat_priv,
					struct tt_global_entry *tt_global_entry,
					const char *message)
833
{
834 835
	batadv_dbg(DBG_TT, bat_priv, "Deleting global tt entry %pM: %s\n",
		   tt_global_entry->common.addr, message);
836

837
	batadv_hash_remove(bat_priv->tt_global_hash, batadv_compare_tt,
838
			   batadv_choose_orig, tt_global_entry->common.addr);
839
	batadv_tt_global_entry_free_ref(tt_global_entry);
840

841 842
}

843 844 845 846
/* 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.
 */
847 848 849 850
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)
851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878
{
	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.
		 */
879 880
		batadv_tt_global_del_orig_entry(bat_priv, tt_global_entry,
						orig_node, message);
881 882 883 884
}



885 886 887 888
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)
889
{
890
	struct tt_global_entry *tt_global_entry = NULL;
891
	struct tt_local_entry *local_entry = NULL;
892

893
	tt_global_entry = batadv_tt_global_hash_find(bat_priv, addr);
894
	if (!tt_global_entry)
895
		goto out;
896

897
	if (!roaming) {
898 899
		batadv_tt_global_del_orig_entry(bat_priv, tt_global_entry,
						orig_node, message);
900 901

		if (hlist_empty(&tt_global_entry->orig_list))
902 903
			batadv_tt_global_del_struct(bat_priv, tt_global_entry,
						    message);
904 905 906

		goto out;
	}
907 908 909

	/* if we are deleting a global entry due to a roam
	 * event, there are two possibilities:
910 911
	 * 1) the client roamed from node A to node B => if there
	 *    is only one originator left for this client, we mark
912 913 914
	 *    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.
915 916 917
	 *
	 *    If there are other originators left, we directly delete
	 *    the originator.
918
	 * 2) the client roamed to us => we can directly delete
919 920
	 *    the global entry, since it is useless now.
	 */
921 922 923
	local_entry = batadv_tt_local_hash_find(bat_priv,
						tt_global_entry->common.addr);
	if (local_entry) {
924
		/* local entry exists, case 2: client roamed to us. */
925 926
		batadv_tt_global_del_orig_list(tt_global_entry);
		batadv_tt_global_del_struct(bat_priv, tt_global_entry, message);
927 928
	} else
		/* no local entry exists, case 1: check for roaming */
929 930
		batadv_tt_global_del_roaming(bat_priv, tt_global_entry,
					     orig_node, message);
931 932


933
out:
934
	if (tt_global_entry)
935 936 937
		batadv_tt_global_entry_free_ref(tt_global_entry);
	if (local_entry)
		batadv_tt_local_entry_free_ref(local_entry);
938 939
}

940 941
void batadv_tt_global_del_orig(struct bat_priv *bat_priv,
			       struct orig_node *orig_node, const char *message)
942
{
943
	struct tt_global_entry *global_entry;
944
	struct tt_common_entry *tt_common_entry;
945
	uint32_t i;
946 947 948
	struct hashtable_t *hash = bat_priv->tt_global_hash;
	struct hlist_node *node, *safe;
	struct hlist_head *head;
949
	spinlock_t *list_lock; /* protects write access to the hash lists */
950

951 952 953
	if (!hash)
		return;

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

958
		spin_lock_bh(list_lock);
959
		hlist_for_each_entry_safe(tt_common_entry, node, safe,
960
					  head, hash_entry) {
961 962 963
			global_entry = container_of(tt_common_entry,
						    struct tt_global_entry,
						    common);
964

965 966
			batadv_tt_global_del_orig_entry(bat_priv, global_entry,
							orig_node, message);
967

968
			if (hlist_empty(&global_entry->orig_list)) {
969 970
				batadv_dbg(DBG_TT, bat_priv,
					   "Deleting global tt entry %pM: %s\n",
971
					   global_entry->common.addr, message);
972
				hlist_del_rcu(node);
973
				batadv_tt_global_entry_free_ref(global_entry);
974
			}
975
		}
976
		spin_unlock_bh(list_lock);
977
	}
978
	orig_node->tt_initialised = false;
979 980
}

981
static void batadv_tt_global_roam_purge(struct bat_priv *bat_priv)
982 983
{
	struct hashtable_t *hash = bat_priv->tt_global_hash;
984
	struct tt_common_entry *tt_common_entry;
985 986 987
	struct tt_global_entry *tt_global_entry;
	struct hlist_node *node, *node_tmp;
	struct hlist_head *head;
988
	spinlock_t *list_lock; /* protects write access to the hash lists */
989
	uint32_t i;
990 991 992

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

995
		spin_lock_bh(list_lock);
996
		hlist_for_each_entry_safe(tt_common_entry, node, node_tmp,
997
					  head, hash_entry) {
998 999 1000 1001
			tt_global_entry = container_of(tt_common_entry,
						       struct tt_global_entry,
						       common);
			if (!(tt_global_entry->common.flags & TT_CLIENT_ROAM))
1002
				continue;
1003 1004
			if (!batadv_has_timed_out(tt_global_entry->roam_at,
						  TT_CLIENT_ROAM_TIMEOUT))
1005 1006
				continue;

1007 1008 1009
			batadv_dbg(DBG_TT, bat_priv,
				   "Deleting global tt entry (%pM): Roaming timeout\n",
				   tt_global_entry->common.addr);
1010

1011
			hlist_del_rcu(node);
1012
			batadv_tt_global_entry_free_ref(tt_global_entry);
1013
		}
1014
		spin_unlock_bh(list_lock);
1015 1016 1017 1018
	}

}

1019
static void batadv_tt_global_table_free(struct bat_priv *bat_priv)
1020
{
1021 1022
	struct hashtable_t *hash;
	spinlock_t *list_lock; /* protects write access to the hash lists */
1023
	struct tt_common_entry *tt_common_entry;
1024 1025 1026
	struct tt_global_entry *tt_global_entry;
	struct hlist_node *node, *node_tmp;
	struct hlist_head *head;
1027
	uint32_t i;
1028

1029
	if (!bat_priv->tt_global_hash)
1030 1031
		return;

1032 1033 1034 1035 1036 1037 1038
	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);
1039
		hlist_for_each_entry_safe(tt_common_entry, node, node_tmp,
1040 1041
					  head, hash_entry) {
			hlist_del_rcu(node);
1042 1043 1044
			tt_global_entry = container_of(tt_common_entry,
						       struct tt_global_entry,
						       common);
1045
			batadv_tt_global_entry_free_ref(tt_global_entry);
1046 1047 1048 1049
		}
		spin_unlock_bh(list_lock);
	}

1050
	batadv_hash_destroy(hash);
1051

1052
	bat_priv->tt_global_hash = NULL;
1053 1054
}

1055 1056
static bool _batadv_is_ap_isolated(struct tt_local_entry *tt_local_entry,
				   struct tt_global_entry *tt_global_entry)
1057 1058 1059
{
	bool ret = false;

1060 1061
	if (tt_local_entry->common.flags & TT_CLIENT_WIFI &&
	    tt_global_entry->common.flags & TT_CLIENT_WIFI)
1062 1063 1064 1065 1066
		ret = true;

	return ret;
}

1067 1068 1069
struct orig_node *batadv_transtable_search(struct bat_priv *bat_priv,
					   const uint8_t *src,
					   const uint8_t *addr)
1070
{
1071 1072
	struct tt_local_entry *tt_local_entry = NULL;
	struct tt_global_entry *tt_global_entry = NULL;
1073
	struct orig_node *orig_node = NULL;
1074 1075 1076 1077 1078
	struct neigh_node *router = NULL;
	struct hlist_head *head;
	struct hlist_node *node;
	struct tt_orig_list_entry *orig_entry;
	int best_tq;
1079

1080
	if (src && atomic_read(&bat_priv->ap_isolation)) {
1081
		tt_local_entry = batadv_tt_local_hash_find(bat_priv, src);
1082 1083 1084
		if (!tt_local_entry)
			goto out;
	}
1085

1086
	tt_global_entry = batadv_tt_global_hash_find(bat_priv, addr);
1087
	if (!tt_global_entry)
1088
		goto out;
1089

1090
	/* check whether the clients should not communicate due to AP
1091 1092
	 * isolation
	 */
1093 1094
	if (tt_local_entry &&
	    _batadv_is_ap_isolated(tt_local_entry, tt_global_entry))
1095 1096
		goto out;

1097
	best_tq = 0;
1098

1099 1100 1101
	rcu_read_lock();
	head = &tt_global_entry->orig_list;
	hlist_for_each_entry_rcu(orig_entry, node, head, list) {
1102
		router = batadv_orig_node_get_router(orig_entry->orig_node);
1103 1104
		if (!router)
			continue;
1105

1106 1107 1108 1109
		if (router->tq_avg > best_tq) {
			orig_node = orig_entry->orig_node;
			best_tq = router->tq_avg;
		}
1110
		batadv_neigh_node_free_ref(router);
1111 1112 1113 1114 1115
	}
	/* found anything? */
	if (orig_node && !atomic_inc_not_zero(&orig_node->refcount))
		orig_node = NULL;
	rcu_read_unlock();
1116
out:
1117
	if (tt_global_entry)
1118
		batadv_tt_global_entry_free_ref(tt_global_entry);
1119
	if (tt_local_entry)
1120
		batadv_tt_local_entry_free_ref(tt_local_entry);
1121

1122
	return orig_node;
1123
}
1124 1125

/* Calculates the checksum of the local table of a given orig_node */
1126 1127
static uint16_t batadv_tt_global_crc(struct bat_priv *bat_priv,
				     struct orig_node *orig_node)
1128 1129 1130
{
	uint16_t total = 0, total_one;
	struct hashtable_t *hash = bat_priv->tt_global_hash;
1131
	struct tt_common_entry *tt_common_entry;
1132 1133 1134
	struct tt_global_entry *tt_global_entry;
	struct hlist_node *node;
	struct hlist_head *head;
1135 1136
	uint32_t i;
	int j;
1137 1138 1139 1140 1141

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

		rcu_read_lock();
1142
		hlist_for_each_entry_rcu(tt_common_entry, node,
1143
					 head, hash_entry) {
1144 1145 1146
			tt_global_entry = container_of(tt_common_entry,
						       struct tt_global_entry,
						       common);
1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157
			/* 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
			 */
1158 1159
			if (!batadv_tt_global_entry_has_orig(tt_global_entry,
							     orig_node))
1160 1161 1162 1163 1164 1165 1166
				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;
1167 1168 1169 1170 1171 1172 1173 1174
		}
		rcu_read_unlock();
	}

	return total;
}

/* Calculates the checksum of the local table */
1175
static uint16_t batadv_tt_local_crc(struct bat_priv *bat_priv)
1176 1177 1178
{
	uint16_t total = 0, total_one;
	struct hashtable_t *hash = bat_priv->tt_local_hash;
1179
	struct tt_common_entry *tt_common_entry;
1180 1181
	struct hlist_node *node;
	struct hlist_head *head;
1182 1183
	uint32_t i;
	int j;
1184 1185 1186 1187 1188

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

		rcu_read_lock();
1189
		hlist_for_each_entry_rcu(tt_common_entry, node,
1190
					 head, hash_entry) {
1191
			/* not yet committed clients have not to be taken into
1192 1193
			 * account while computing the CRC
			 */
1194
			if (tt_common_entry->flags & TT_CLIENT_NEW)
1195
				continue;
1196 1197 1198
			total_one = 0;
			for (j = 0; j < ETH_ALEN; j++)
				total_one = crc16_byte(total_one,
1199
						   tt_common_entry->addr[j]);
1200 1201 1202 1203 1204 1205 1206 1207
			total ^= total_one;
		}
		rcu_read_unlock();
	}

	return total;
}

1208
static void batadv_tt_req_list_free(struct bat_priv *bat_priv)
1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221
{
	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);
}

1222 1223 1224 1225
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)
1226
{
1227
	uint16_t tt_buff_len = batadv_tt_len(tt_num_changes);
1228 1229

	/* Replace the old buffer only if I received something in the
1230 1231
	 * last OGM (the OGM could carry no changes)
	 */
1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244
	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);
}

1245
static void batadv_tt_req_purge(struct bat_priv *bat_priv)
1246 1247 1248 1249 1250
{
	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) {
1251
		if (batadv_has_timed_out(node->issued_at, TT_REQUEST_TIMEOUT)) {
1252 1253 1254 1255 1256 1257 1258 1259
			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
1260 1261
 * has already been issued for this orig_node, NULL otherwise
 */
1262 1263
static struct tt_req_node *batadv_new_tt_req_node(struct bat_priv *bat_priv,
						  struct orig_node *orig_node)
1264 1265 1266 1267 1268
{
	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) {
1269 1270 1271
		if (batadv_compare_eth(tt_req_node_tmp, orig_node) &&
		    !batadv_has_timed_out(tt_req_node_tmp->issued_at,
					  TT_REQUEST_TIMEOUT))
1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287
			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;
}

1288
/* data_ptr is useless here, but has to be kept to respect the prototype */
1289 1290
static int batadv_tt_local_valid_entry(const void *entry_ptr,
				       const void *data_ptr)
1291
{
1292
	const struct tt_common_entry *tt_common_entry = entry_ptr;
1293

1294
	if (tt_common_entry->flags & TT_CLIENT_NEW)
1295 1296 1297 1298
		return 0;
	return 1;
}

1299 1300
static int batadv_tt_global_valid(const void *entry_ptr,
				  const void *data_ptr)
1301
{
1302 1303
	const struct tt_common_entry *tt_common_entry = entry_ptr;
	const struct tt_global_entry *tt_global_entry;
1304 1305
	const struct orig_node *orig_node = data_ptr;

1306
	if (tt_common_entry->flags & TT_CLIENT_ROAM)
1307 1308
		return 0;

1309 1310 1311
	tt_global_entry = container_of(tt_common_entry, struct tt_global_entry,
				       common);

1312
	return batadv_tt_global_entry_has_orig(tt_global_entry, orig_node);
1313 1314
}

1315 1316 1317 1318 1319 1320
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)
1321
{
1322
	struct tt_common_entry *tt_common_entry;
1323 1324 1325 1326 1327 1328 1329
	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);
1330
	uint32_t i;
1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353

	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];

1354
		hlist_for_each_entry_rcu(tt_common_entry, node,
1355 1356 1357 1358
					 head, hash_entry) {
			if (tt_count == tt_tot)
				break;

1359
			if ((valid_cb) && (!valid_cb(tt_common_entry, cb_data)))
1360 1361
				continue;

1362 1363
			memcpy(tt_change->addr, tt_common_entry->addr,
			       ETH_ALEN);
1364 1365 1366 1367 1368 1369 1370 1371
			tt_change->flags = NO_FLAGS;

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

1372
	/* store in the message the number of entries we have successfully
1373 1374
	 * copied
	 */
1375 1376
	tt_response->tt_data = htons(tt_count);

1377 1378 1379 1380
out:
	return skb;
}

1381 1382 1383 1384
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)
1385 1386 1387 1388 1389 1390 1391 1392
{
	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;

1393
	primary_if = batadv_primary_if_get_selected(bat_priv);
1394 1395 1396 1397
	if (!primary_if)
		goto out;

	/* The new tt_req will be issued only if I'm not waiting for a
1398 1399
	 * reply from the same orig_node yet
	 */
1400
	tt_req_node = batadv_new_tt_req_node(bat_priv, dst_orig_node);
1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412
	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));

1413 1414
	tt_request->header.packet_type = BAT_TT_QUERY;
	tt_request->header.version = COMPAT_VERSION;
1415 1416
	memcpy(tt_request->src, primary_if->net_dev->dev_addr, ETH_ALEN);
	memcpy(tt_request->dst, dst_orig_node->orig, ETH_ALEN);
1417
	tt_request->header.ttl = TTL;
1418
	tt_request->ttvn = ttvn;
1419
	tt_request->tt_data = htons(tt_crc);
1420 1421 1422 1423 1424
	tt_request->flags = TT_REQUEST;

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

1425
	neigh_node = batadv_orig_node_get_router(dst_orig_node);
1426 1427 1428
	if (!neigh_node)
		goto out;

1429 1430 1431 1432
	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' : '.'));
1433

1434 1435
	batadv_inc_counter(bat_priv, BAT_CNT_TT_REQUEST_TX);

1436
	batadv_send_skb_packet(skb, neigh_node->if_incoming, neigh_node->addr);
1437 1438 1439 1440
	ret = 0;

out:
	if (neigh_node)
1441
		batadv_neigh_node_free_ref(neigh_node);
1442
	if (primary_if)
1443
		batadv_hardif_free_ref(primary_if);
1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454
	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;
}

1455 1456
static bool batadv_send_other_tt_response(struct bat_priv *bat_priv,
					  struct tt_query_packet *tt_request)
1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468
{
	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;

1469 1470 1471 1472
	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' : '.'));
1473 1474

	/* Let's get the orig node of the REAL destination */
1475
	req_dst_orig_node = batadv_orig_hash_find(bat_priv, tt_request->dst);
1476 1477 1478
	if (!req_dst_orig_node)
		goto out;

1479
	res_dst_orig_node = batadv_orig_hash_find(bat_priv, tt_request->src);
1480 1481 1482
	if (!res_dst_orig_node)
		goto out;

1483
	neigh_node = batadv_orig_node_get_router(res_dst_orig_node);
1484 1485 1486
	if (!neigh_node)
		goto out;

1487
	primary_if = batadv_primary_if_get_selected(bat_priv);
1488 1489 1490 1491 1492 1493
	if (!primary_if)
		goto out;

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

1494
	/* I don't have the requested data */
1495
	if (orig_ttvn != req_ttvn ||
1496
	    tt_request->tt_data != htons(req_dst_orig_node->tt_crc))
1497 1498
		goto out;

1499
	/* If the full table has been explicitly requested */
1500 1501 1502 1503 1504 1505 1506
	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
1507 1508
	 * I'll send only one packet with as much TT entries as I can
	 */
1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535
	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);

1536 1537 1538 1539 1540
		skb = batadv_tt_response_fill_table(tt_len, ttvn,
						    bat_priv->tt_global_hash,
						    primary_if,
						    batadv_tt_global_valid,
						    req_dst_orig_node);
1541 1542 1543 1544 1545 1546
		if (!skb)
			goto out;

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

1547 1548 1549
	tt_response->header.packet_type = BAT_TT_QUERY;
	tt_response->header.version = COMPAT_VERSION;
	tt_response->header.ttl = TTL;
1550 1551 1552 1553 1554 1555 1556
	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;

1557 1558 1559 1560
	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);
1561

1562 1563
	batadv_inc_counter(bat_priv, BAT_CNT_TT_RESPONSE_TX);

1564
	batadv_send_skb_packet(skb, neigh_node->if_incoming, neigh_node->addr);
1565 1566 1567 1568 1569 1570 1571 1572
	ret = true;
	goto out;

unlock:
	spin_unlock_bh(&req_dst_orig_node->tt_buff_lock);

out:
	if (res_dst_orig_node)
1573
		batadv_orig_node_free_ref(res_dst_orig_node);
1574
	if (req_dst_orig_node)
1575
		batadv_orig_node_free_ref(req_dst_orig_node);
1576
	if (neigh_node)
1577
		batadv_neigh_node_free_ref(neigh_node);
1578
	if (primary_if)
1579
		batadv_hardif_free_ref(primary_if);
1580 1581 1582 1583 1584
	if (!ret)
		kfree_skb(skb);
	return ret;

}
1585 1586
static bool batadv_send_my_tt_response(struct bat_priv *bat_priv,
				       struct tt_query_packet *tt_request)
1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598
{
	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;

1599 1600 1601 1602
	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' : '.'));
1603 1604 1605 1606 1607


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

1608
	orig_node = batadv_orig_hash_find(bat_priv, tt_request->src);
1609 1610 1611
	if (!orig_node)
		goto out;

1612
	neigh_node = batadv_orig_node_get_router(orig_node);
1613 1614 1615
	if (!neigh_node)
		goto out;

1616
	primary_if = batadv_primary_if_get_selected(bat_priv);
1617 1618 1619 1620
	if (!primary_if)
		goto out;

	/* If the full table has been explicitly requested or the gap
1621 1622
	 * is too big send the whole local translation table
	 */
1623 1624 1625 1626 1627 1628 1629
	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
1630 1631
	 * I'll send only one packet with as much TT entries as I can
	 */
1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656
	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);

1657 1658 1659 1660 1661
		skb = batadv_tt_response_fill_table(tt_len, ttvn,
						    bat_priv->tt_local_hash,
						    primary_if,
						    batadv_tt_local_valid_entry,
						    NULL);
1662 1663 1664 1665 1666 1667
		if (!skb)
			goto out;

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

1668 1669 1670
	tt_response->header.packet_type = BAT_TT_QUERY;
	tt_response->header.version = COMPAT_VERSION;
	tt_response->header.ttl = TTL;
1671 1672 1673 1674 1675 1676 1677
	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;

1678 1679 1680 1681
	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' : '.'));
1682

1683 1684
	batadv_inc_counter(bat_priv, BAT_CNT_TT_RESPONSE_TX);

1685
	batadv_send_skb_packet(skb, neigh_node->if_incoming, neigh_node->addr);
1686 1687 1688 1689 1690 1691 1692
	ret = true;
	goto out;

unlock:
	spin_unlock_bh(&bat_priv->tt_buff_lock);
out:
	if (orig_node)
1693
		batadv_orig_node_free_ref(orig_node);
1694
	if (neigh_node)
1695
		batadv_neigh_node_free_ref(neigh_node);
1696
	if (primary_if)
1697
		batadv_hardif_free_ref(primary_if);
1698 1699 1700 1701 1702 1703
	if (!ret)
		kfree_skb(skb);
	/* This packet was for me, so it doesn't need to be re-routed */
	return true;
}

1704 1705
bool batadv_send_tt_response(struct bat_priv *bat_priv,
			     struct tt_query_packet *tt_request)
1706
{
1707
	if (batadv_is_my_mac(tt_request->dst)) {
1708
		/* don't answer backbone gws! */
1709
		if (batadv_bla_is_backbone_gw_orig(bat_priv, tt_request->src))
1710 1711
			return true;

1712
		return batadv_send_my_tt_response(bat_priv, tt_request);
1713
	} else {
1714
		return batadv_send_other_tt_response(bat_priv, tt_request);
1715
	}
1716 1717
}

1718 1719 1720 1721
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)
1722 1723
{
	int i;
1724
	int is_wifi;
1725
	int roams;
1726 1727

	for (i = 0; i < tt_num_changes; i++) {
1728
		if ((tt_change + i)->flags & TT_CLIENT_DEL) {
1729 1730 1731 1732 1733
			roams = (tt_change + i)->flags & TT_CLIENT_ROAM;
			batadv_tt_global_del(bat_priv, orig_node,
					     (tt_change + i)->addr,
					      "tt removed by changes",
					      roams);
1734 1735 1736 1737 1738
		} else {
			is_wifi = (tt_change + i)->flags & TT_CLIENT_WIFI;
			if (!batadv_tt_global_add(bat_priv, orig_node,
						  (tt_change + i)->addr, ttvn,
						  false, is_wifi))
1739 1740 1741 1742 1743 1744 1745
				/* 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;
1746
		}
1747
	}
1748
	orig_node->tt_initialised = true;
1749 1750
}

1751 1752
static void batadv_tt_fill_gtable(struct bat_priv *bat_priv,
				  struct tt_query_packet *tt_response)
1753 1754 1755
{
	struct orig_node *orig_node = NULL;

1756
	orig_node = batadv_orig_hash_find(bat_priv, tt_response->src);
1757 1758 1759 1760
	if (!orig_node)
		goto out;

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

1763 1764 1765 1766
	_batadv_tt_update_changes(bat_priv, orig_node,
				  (struct tt_change *)(tt_response + 1),
				  ntohs(tt_response->tt_data),
				  tt_response->ttvn);
1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777

	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)
1778
		batadv_orig_node_free_ref(orig_node);
1779 1780
}

1781 1782 1783 1784
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)
1785
{
1786 1787
	_batadv_tt_update_changes(bat_priv, orig_node, tt_change,
				  tt_num_changes, ttvn);
1788

1789 1790
	batadv_tt_save_orig_buffer(bat_priv, orig_node,
				   (unsigned char *)tt_change, tt_num_changes);
1791 1792 1793
	atomic_set(&orig_node->last_ttvn, ttvn);
}

1794
bool batadv_is_my_client(struct bat_priv *bat_priv, const uint8_t *addr)
1795
{
1796 1797
	struct tt_local_entry *tt_local_entry = NULL;
	bool ret = false;
1798

1799
	tt_local_entry = batadv_tt_local_hash_find(bat_priv, addr);
1800 1801
	if (!tt_local_entry)
		goto out;
1802
	/* Check if the client has been logically deleted (but is kept for
1803 1804
	 * consistency purpose)
	 */
1805
	if (tt_local_entry->common.flags & TT_CLIENT_PENDING)
1806
		goto out;
1807 1808
	ret = true;
out:
1809
	if (tt_local_entry)
1810
		batadv_tt_local_entry_free_ref(tt_local_entry);
1811
	return ret;
1812 1813
}

1814 1815
void batadv_handle_tt_response(struct bat_priv *bat_priv,
			       struct tt_query_packet *tt_response)
1816 1817 1818 1819
{
	struct tt_req_node *node, *safe;
	struct orig_node *orig_node = NULL;

1820 1821 1822 1823 1824
	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' : '.'));
1825

1826
	/* we should have never asked a backbone gw */
1827
	if (batadv_bla_is_backbone_gw_orig(bat_priv, tt_response->src))
1828 1829
		goto out;

1830
	orig_node = batadv_orig_hash_find(bat_priv, tt_response->src);
1831 1832 1833 1834
	if (!orig_node)
		goto out;

	if (tt_response->flags & TT_FULL_TABLE)
1835
		batadv_tt_fill_gtable(bat_priv, tt_response);
1836
	else
1837 1838 1839 1840
		batadv_tt_update_changes(bat_priv, orig_node,
					 ntohs(tt_response->tt_data),
					 tt_response->ttvn,
					 (struct tt_change *)(tt_response + 1));
1841 1842 1843 1844

	/* 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) {
1845
		if (!batadv_compare_eth(node->addr, tt_response->src))
1846 1847 1848 1849 1850 1851 1852
			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 */
1853
	orig_node->tt_crc = batadv_tt_global_crc(bat_priv, orig_node);
1854
	/* Roaming phase is over: tables are in sync again. I can
1855 1856
	 * unset the flag
	 */
1857
	orig_node->tt_poss_change = false;
1858 1859
out:
	if (orig_node)
1860
		batadv_orig_node_free_ref(orig_node);
1861 1862
}

1863
int batadv_tt_init(struct bat_priv *bat_priv)
1864
{
1865
	int ret;
1866

1867
	ret = batadv_tt_local_init(bat_priv);
1868 1869 1870
	if (ret < 0)
		return ret;

1871
	ret = batadv_tt_global_init(bat_priv);
1872 1873
	if (ret < 0)
		return ret;
1874

1875
	batadv_tt_start_timer(bat_priv);
1876 1877 1878 1879

	return 1;
}

1880
static void batadv_tt_roam_list_free(struct bat_priv *bat_priv)
1881
{
1882
	struct tt_roam_node *node, *safe;
1883

1884
	spin_lock_bh(&bat_priv->tt_roam_list_lock);
1885

1886 1887 1888 1889 1890 1891 1892 1893
	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);
}

1894
static void batadv_tt_roam_purge(struct bat_priv *bat_priv)
1895 1896 1897 1898 1899
{
	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) {
1900
		if (!batadv_has_timed_out(node->first_time, ROAMING_MAX_TIME))
1901 1902 1903 1904 1905 1906 1907 1908 1909 1910 1911 1912
			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.
 *
1913 1914
 * returns true if the ROAMING_ADV can be sent, false otherwise
 */
1915 1916
static bool batadv_tt_check_roam_count(struct bat_priv *bat_priv,
				       uint8_t *client)
1917 1918 1919 1920 1921 1922
{
	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
1923 1924
	 * reply from the same orig_node yet
	 */
1925
	list_for_each_entry(tt_roam_node, &bat_priv->tt_roam_list, list) {
1926
		if (!batadv_compare_eth(tt_roam_node->addr, client))
1927 1928
			continue;

1929 1930
		if (batadv_has_timed_out(tt_roam_node->first_time,
					 ROAMING_MAX_TIME))
1931 1932
			continue;

1933
		if (!batadv_atomic_dec_not_zero(&tt_roam_node->counter))
1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957
			/* 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;
}

1958 1959
static void batadv_send_roam_adv(struct bat_priv *bat_priv, uint8_t *client,
				 struct orig_node *orig_node)
1960 1961 1962 1963 1964 1965 1966 1967
{
	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
1968 1969
	 * already roamed to us too many times
	 */
1970
	if (!batadv_tt_check_roam_count(bat_priv, client))
1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981
		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));

1982 1983 1984
	roam_adv_packet->header.packet_type = BAT_ROAM_ADV;
	roam_adv_packet->header.version = COMPAT_VERSION;
	roam_adv_packet->header.ttl = TTL;
1985
	primary_if = batadv_primary_if_get_selected(bat_priv);
1986 1987 1988
	if (!primary_if)
		goto out;
	memcpy(roam_adv_packet->src, primary_if->net_dev->dev_addr, ETH_ALEN);
1989
	batadv_hardif_free_ref(primary_if);
1990 1991 1992
	memcpy(roam_adv_packet->dst, orig_node->orig, ETH_ALEN);
	memcpy(roam_adv_packet->client, client, ETH_ALEN);

1993
	neigh_node = batadv_orig_node_get_router(orig_node);
1994 1995 1996
	if (!neigh_node)
		goto out;

1997 1998 1999
	batadv_dbg(DBG_TT, bat_priv,
		   "Sending ROAMING_ADV to %pM (client %pM) via %pM\n",
		   orig_node->orig, client, neigh_node->addr);
2000

2001 2002
	batadv_inc_counter(bat_priv, BAT_CNT_TT_ROAM_ADV_TX);

2003
	batadv_send_skb_packet(skb, neigh_node->if_incoming, neigh_node->addr);
2004 2005 2006 2007
	ret = 0;

out:
	if (neigh_node)
2008
		batadv_neigh_node_free_ref(neigh_node);
2009 2010 2011
	if (ret)
		kfree_skb(skb);
	return;
2012 2013
}

2014
static void batadv_tt_purge(struct work_struct *work)
2015 2016 2017 2018 2019 2020
{
	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);

2021 2022 2023 2024
	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);
2025

2026
	batadv_tt_start_timer(bat_priv);
2027
}
2028

2029
void batadv_tt_free(struct bat_priv *bat_priv)
2030 2031 2032
{
	cancel_delayed_work_sync(&bat_priv->tt_work);

2033 2034 2035 2036 2037
	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);
2038 2039 2040

	kfree(bat_priv->tt_buff);
}
2041

2042
/* This function will enable or disable the specified flags for all the entries
2043 2044
 * in the given hash table and returns the number of modified entries
 */
2045 2046
static uint16_t batadv_tt_set_flags(struct hashtable_t *hash, uint16_t flags,
				    bool enable)
2047
{
2048
	uint32_t i;
2049
	uint16_t changed_num = 0;
2050 2051
	struct hlist_head *head;
	struct hlist_node *node;
2052
	struct tt_common_entry *tt_common_entry;
2053 2054

	if (!hash)
2055
		goto out;
2056 2057 2058 2059 2060

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

		rcu_read_lock();
2061
		hlist_for_each_entry_rcu(tt_common_entry, node,
2062
					 head, hash_entry) {
2063 2064 2065 2066 2067 2068 2069 2070 2071 2072
			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++;
2073 2074 2075
		}
		rcu_read_unlock();
	}
2076 2077
out:
	return changed_num;
2078 2079 2080
}

/* Purge out all the tt local entries marked with TT_CLIENT_PENDING */
2081
static void batadv_tt_local_purge_pending_clients(struct bat_priv *bat_priv)
2082 2083
{
	struct hashtable_t *hash = bat_priv->tt_local_hash;
2084
	struct tt_common_entry *tt_common_entry;
2085 2086 2087 2088
	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 */
2089
	uint32_t i;
2090 2091 2092 2093 2094 2095 2096 2097 2098

	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);
2099
		hlist_for_each_entry_safe(tt_common_entry, node, node_tmp,
2100
					  head, hash_entry) {
2101
			if (!(tt_common_entry->flags & TT_CLIENT_PENDING))
2102 2103
				continue;

2104 2105 2106
			batadv_dbg(DBG_TT, bat_priv,
				   "Deleting local tt entry (%pM): pending\n",
				   tt_common_entry->addr);
2107 2108 2109

			atomic_dec(&bat_priv->num_local_tt);
			hlist_del_rcu(node);
2110 2111 2112
			tt_local_entry = container_of(tt_common_entry,
						      struct tt_local_entry,
						      common);
2113
			batadv_tt_local_entry_free_ref(tt_local_entry);
2114 2115 2116 2117 2118 2119
		}
		spin_unlock_bh(list_lock);
	}

}

2120 2121 2122
static int batadv_tt_commit_changes(struct bat_priv *bat_priv,
				    unsigned char **packet_buff,
				    int *packet_buff_len, int packet_min_len)
2123
{
2124 2125 2126 2127 2128
	uint16_t changed_num = 0;

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

2129 2130
	changed_num = batadv_tt_set_flags(bat_priv->tt_local_hash,
					  TT_CLIENT_NEW, false);
2131 2132

	/* all reset entries have to be counted as local entries */
2133
	atomic_add(changed_num, &bat_priv->num_local_tt);
2134
	batadv_tt_local_purge_pending_clients(bat_priv);
2135
	bat_priv->tt_crc = batadv_tt_local_crc(bat_priv);
2136 2137 2138

	/* Increment the TTVN only once per OGM interval */
	atomic_inc(&bat_priv->ttvn);
2139 2140 2141
	batadv_dbg(DBG_TT, bat_priv,
		   "Local changes committed, updating to ttvn %u\n",
		   (uint8_t)atomic_read(&bat_priv->ttvn));
2142
	bat_priv->tt_poss_change = false;
2143 2144 2145 2146

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

2147 2148
	return batadv_tt_changes_fill_buff(bat_priv, packet_buff,
					   packet_buff_len, packet_min_len);
2149 2150 2151 2152 2153 2154 2155 2156 2157 2158
}

/* 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 */
2159 2160 2161
	tt_num_changes = batadv_tt_commit_changes(bat_priv, packet_buff,
						  packet_buff_len,
						  packet_min_len);
2162 2163 2164

	/* if the changes have been sent often enough */
	if ((tt_num_changes < 0) &&
2165
	    (!batadv_atomic_dec_not_zero(&bat_priv->tt_ogm_append_cnt))) {
2166 2167
		batadv_tt_realloc_packet_buff(packet_buff, packet_buff_len,
					      packet_min_len, packet_min_len);
2168 2169 2170 2171
		tt_num_changes = 0;
	}

	return tt_num_changes;
2172
}
2173

2174 2175
bool batadv_is_ap_isolated(struct bat_priv *bat_priv, uint8_t *src,
			   uint8_t *dst)
2176 2177 2178
{
	struct tt_local_entry *tt_local_entry = NULL;
	struct tt_global_entry *tt_global_entry = NULL;
2179
	bool ret = false;
2180 2181

	if (!atomic_read(&bat_priv->ap_isolation))
2182
		goto out;
2183

2184
	tt_local_entry = batadv_tt_local_hash_find(bat_priv, dst);
2185 2186 2187
	if (!tt_local_entry)
		goto out;

2188
	tt_global_entry = batadv_tt_global_hash_find(bat_priv, src);
2189 2190 2191
	if (!tt_global_entry)
		goto out;

2192
	if (!_batadv_is_ap_isolated(tt_local_entry, tt_global_entry))
2193 2194
		goto out;

2195
	ret = true;
2196 2197 2198

out:
	if (tt_global_entry)
2199
		batadv_tt_global_entry_free_ref(tt_global_entry);
2200
	if (tt_local_entry)
2201
		batadv_tt_local_entry_free_ref(tt_local_entry);
2202 2203
	return ret;
}
2204

2205 2206 2207 2208
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)
2209 2210 2211 2212
{
	uint8_t orig_ttvn = (uint8_t)atomic_read(&orig_node->last_ttvn);
	bool full_table = true;

2213
	/* don't care about a backbone gateways updates. */
2214
	if (batadv_bla_is_backbone_gw_orig(bat_priv, orig_node->orig))
2215 2216
		return;

2217
	/* orig table not initialised AND first diff is in the OGM OR the ttvn
2218 2219
	 * increased by one -> we can apply the attached changes
	 */
2220 2221
	if ((!orig_node->tt_initialised && ttvn == 1) ||
	    ttvn - orig_ttvn == 1) {
2222 2223
		/* the OGM could not contain the changes due to their size or
		 * because they have already been sent TT_OGM_APPEND_MAX times.
2224 2225
		 * In this case send a tt request
		 */
2226 2227 2228 2229 2230
		if (!tt_num_changes) {
			full_table = false;
			goto request_table;
		}

2231 2232
		batadv_tt_update_changes(bat_priv, orig_node, tt_num_changes,
					 ttvn, (struct tt_change *)tt_buff);
2233 2234 2235

		/* Even if we received the precomputed crc with the OGM, we
		 * prefer to recompute it to spot any possible inconsistency
2236 2237
		 * in the global table
		 */
2238
		orig_node->tt_crc = batadv_tt_global_crc(bat_priv, orig_node);
2239 2240 2241 2242 2243 2244 2245 2246

		/* 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
2247 2248
		 * inconsistency
		 */
2249 2250 2251 2252
		if (orig_node->tt_crc != tt_crc)
			goto request_table;

		/* Roaming phase is over: tables are in sync again. I can
2253 2254
		 * unset the flag
		 */
2255 2256 2257
		orig_node->tt_poss_change = false;
	} else {
		/* if we missed more than one change or our tables are not
2258 2259
		 * in sync anymore -> request fresh tt data
		 */
2260 2261
		if (!orig_node->tt_initialised || ttvn != orig_ttvn ||
		    orig_node->tt_crc != tt_crc) {
2262
request_table:
2263 2264 2265 2266
			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);
2267 2268
			batadv_send_tt_request(bat_priv, orig_node, ttvn,
					       tt_crc, full_table);
2269 2270 2271 2272
			return;
		}
	}
}
2273 2274 2275 2276 2277

/* 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
 */
2278 2279
bool batadv_tt_global_client_is_roaming(struct bat_priv *bat_priv,
					uint8_t *addr)
2280 2281 2282 2283
{
	struct tt_global_entry *tt_global_entry;
	bool ret = false;

2284
	tt_global_entry = batadv_tt_global_hash_find(bat_priv, addr);
2285 2286 2287 2288
	if (!tt_global_entry)
		goto out;

	ret = tt_global_entry->common.flags & TT_CLIENT_ROAM;
2289
	batadv_tt_global_entry_free_ref(tt_global_entry);
2290 2291 2292
out:
	return ret;
}