| /* | 
 |  * net/tipc/monitor.c | 
 |  * | 
 |  * Copyright (c) 2016, Ericsson AB | 
 |  * All rights reserved. | 
 |  * | 
 |  * Redistribution and use in source and binary forms, with or without | 
 |  * modification, are permitted provided that the following conditions are met: | 
 |  * | 
 |  * 1. Redistributions of source code must retain the above copyright | 
 |  *    notice, this list of conditions and the following disclaimer. | 
 |  * 2. Redistributions in binary form must reproduce the above copyright | 
 |  *    notice, this list of conditions and the following disclaimer in the | 
 |  *    documentation and/or other materials provided with the distribution. | 
 |  * 3. Neither the names of the copyright holders nor the names of its | 
 |  *    contributors may be used to endorse or promote products derived from | 
 |  *    this software without specific prior written permission. | 
 |  * | 
 |  * Alternatively, this software may be distributed under the terms of the | 
 |  * GNU General Public License ("GPL") version 2 as published by the Free | 
 |  * Software Foundation. | 
 |  * | 
 |  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" | 
 |  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | 
 |  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | 
 |  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE | 
 |  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR | 
 |  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF | 
 |  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS | 
 |  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN | 
 |  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) | 
 |  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE | 
 |  * POSSIBILITY OF SUCH DAMAGE. | 
 |  */ | 
 |  | 
 | #include <net/genetlink.h> | 
 | #include "core.h" | 
 | #include "addr.h" | 
 | #include "monitor.h" | 
 | #include "bearer.h" | 
 |  | 
 | #define MAX_MON_DOMAIN       64 | 
 | #define MON_TIMEOUT          120000 | 
 | #define MAX_PEER_DOWN_EVENTS 4 | 
 |  | 
 | /* struct tipc_mon_domain: domain record to be transferred between peers | 
 |  * @len: actual size of domain record | 
 |  * @gen: current generation of sender's domain | 
 |  * @ack_gen: most recent generation of self's domain acked by peer | 
 |  * @member_cnt: number of domain member nodes described in this record | 
 |  * @up_map: bit map indicating which of the members the sender considers up | 
 |  * @members: identity of the domain members | 
 |  */ | 
 | struct tipc_mon_domain { | 
 | 	u16 len; | 
 | 	u16 gen; | 
 | 	u16 ack_gen; | 
 | 	u16 member_cnt; | 
 | 	u64 up_map; | 
 | 	u32 members[MAX_MON_DOMAIN]; | 
 | }; | 
 |  | 
 | /* struct tipc_peer: state of a peer node and its domain | 
 |  * @addr: tipc node identity of peer | 
 |  * @head_map: shows which other nodes currently consider peer 'up' | 
 |  * @domain: most recent domain record from peer | 
 |  * @hash: position in hashed lookup list | 
 |  * @list: position in linked list, in circular ascending order by 'addr' | 
 |  * @applied: number of reported domain members applied on this monitor list | 
 |  * @is_up: peer is up as seen from this node | 
 |  * @is_head: peer is assigned domain head as seen from this node | 
 |  * @is_local: peer is in local domain and should be continuously monitored | 
 |  * @down_cnt: - numbers of other peers which have reported this on lost | 
 |  */ | 
 | struct tipc_peer { | 
 | 	u32 addr; | 
 | 	struct tipc_mon_domain *domain; | 
 | 	struct hlist_node hash; | 
 | 	struct list_head list; | 
 | 	u8 applied; | 
 | 	u8 down_cnt; | 
 | 	bool is_up; | 
 | 	bool is_head; | 
 | 	bool is_local; | 
 | }; | 
 |  | 
 | struct tipc_monitor { | 
 | 	struct hlist_head peers[NODE_HTABLE_SIZE]; | 
 | 	int peer_cnt; | 
 | 	struct tipc_peer *self; | 
 | 	rwlock_t lock; | 
 | 	struct tipc_mon_domain cache; | 
 | 	u16 list_gen; | 
 | 	u16 dom_gen; | 
 | 	struct net *net; | 
 | 	struct timer_list timer; | 
 | 	unsigned long timer_intv; | 
 | }; | 
 |  | 
 | static struct tipc_monitor *tipc_monitor(struct net *net, int bearer_id) | 
 | { | 
 | 	return tipc_net(net)->monitors[bearer_id]; | 
 | } | 
 |  | 
 | const int tipc_max_domain_size = sizeof(struct tipc_mon_domain); | 
 |  | 
 | /* dom_rec_len(): actual length of domain record for transport | 
 |  */ | 
 | static int dom_rec_len(struct tipc_mon_domain *dom, u16 mcnt) | 
 | { | 
 | 	return ((void *)&dom->members - (void *)dom) + (mcnt * sizeof(u32)); | 
 | } | 
 |  | 
 | /* dom_size() : calculate size of own domain based on number of peers | 
 |  */ | 
 | static int dom_size(int peers) | 
 | { | 
 | 	int i = 0; | 
 |  | 
 | 	while ((i * i) < peers) | 
 | 		i++; | 
 | 	return i < MAX_MON_DOMAIN ? i : MAX_MON_DOMAIN; | 
 | } | 
 |  | 
 | static void map_set(u64 *up_map, int i, unsigned int v) | 
 | { | 
 | 	*up_map &= ~(1ULL << i); | 
 | 	*up_map |= ((u64)v << i); | 
 | } | 
 |  | 
 | static int map_get(u64 up_map, int i) | 
 | { | 
 | 	return (up_map & (1 << i)) >> i; | 
 | } | 
 |  | 
 | static struct tipc_peer *peer_prev(struct tipc_peer *peer) | 
 | { | 
 | 	return list_last_entry(&peer->list, struct tipc_peer, list); | 
 | } | 
 |  | 
 | static struct tipc_peer *peer_nxt(struct tipc_peer *peer) | 
 | { | 
 | 	return list_first_entry(&peer->list, struct tipc_peer, list); | 
 | } | 
 |  | 
 | static struct tipc_peer *peer_head(struct tipc_peer *peer) | 
 | { | 
 | 	while (!peer->is_head) | 
 | 		peer = peer_prev(peer); | 
 | 	return peer; | 
 | } | 
 |  | 
 | static struct tipc_peer *get_peer(struct tipc_monitor *mon, u32 addr) | 
 | { | 
 | 	struct tipc_peer *peer; | 
 | 	unsigned int thash = tipc_hashfn(addr); | 
 |  | 
 | 	hlist_for_each_entry(peer, &mon->peers[thash], hash) { | 
 | 		if (peer->addr == addr) | 
 | 			return peer; | 
 | 	} | 
 | 	return NULL; | 
 | } | 
 |  | 
 | static struct tipc_peer *get_self(struct net *net, int bearer_id) | 
 | { | 
 | 	struct tipc_monitor *mon = tipc_monitor(net, bearer_id); | 
 |  | 
 | 	return mon->self; | 
 | } | 
 |  | 
 | static inline bool tipc_mon_is_active(struct net *net, struct tipc_monitor *mon) | 
 | { | 
 | 	struct tipc_net *tn = tipc_net(net); | 
 |  | 
 | 	return mon->peer_cnt > tn->mon_threshold; | 
 | } | 
 |  | 
 | /* mon_identify_lost_members() : - identify amd mark potentially lost members | 
 |  */ | 
 | static void mon_identify_lost_members(struct tipc_peer *peer, | 
 | 				      struct tipc_mon_domain *dom_bef, | 
 | 				      int applied_bef) | 
 | { | 
 | 	struct tipc_peer *member = peer; | 
 | 	struct tipc_mon_domain *dom_aft = peer->domain; | 
 | 	int applied_aft = peer->applied; | 
 | 	int i; | 
 |  | 
 | 	for (i = 0; i < applied_bef; i++) { | 
 | 		member = peer_nxt(member); | 
 |  | 
 | 		/* Do nothing if self or peer already see member as down */ | 
 | 		if (!member->is_up || !map_get(dom_bef->up_map, i)) | 
 | 			continue; | 
 |  | 
 | 		/* Loss of local node must be detected by active probing */ | 
 | 		if (member->is_local) | 
 | 			continue; | 
 |  | 
 | 		/* Start probing if member was removed from applied domain */ | 
 | 		if (!applied_aft || (applied_aft < i)) { | 
 | 			member->down_cnt = 1; | 
 | 			continue; | 
 | 		} | 
 |  | 
 | 		/* Member loss is confirmed if it is still in applied domain */ | 
 | 		if (!map_get(dom_aft->up_map, i)) | 
 | 			member->down_cnt++; | 
 | 	} | 
 | } | 
 |  | 
 | /* mon_apply_domain() : match a peer's domain record against monitor list | 
 |  */ | 
 | static void mon_apply_domain(struct tipc_monitor *mon, | 
 | 			     struct tipc_peer *peer) | 
 | { | 
 | 	struct tipc_mon_domain *dom = peer->domain; | 
 | 	struct tipc_peer *member; | 
 | 	u32 addr; | 
 | 	int i; | 
 |  | 
 | 	if (!dom || !peer->is_up) | 
 | 		return; | 
 |  | 
 | 	/* Scan across domain members and match against monitor list */ | 
 | 	peer->applied = 0; | 
 | 	member = peer_nxt(peer); | 
 | 	for (i = 0; i < dom->member_cnt; i++) { | 
 | 		addr = dom->members[i]; | 
 | 		if (addr != member->addr) | 
 | 			return; | 
 | 		peer->applied++; | 
 | 		member = peer_nxt(member); | 
 | 	} | 
 | } | 
 |  | 
 | /* mon_update_local_domain() : update after peer addition/removal/up/down | 
 |  */ | 
 | static void mon_update_local_domain(struct tipc_monitor *mon) | 
 | { | 
 | 	struct tipc_peer *self = mon->self; | 
 | 	struct tipc_mon_domain *cache = &mon->cache; | 
 | 	struct tipc_mon_domain *dom = self->domain; | 
 | 	struct tipc_peer *peer = self; | 
 | 	u64 prev_up_map = dom->up_map; | 
 | 	u16 member_cnt, i; | 
 | 	bool diff; | 
 |  | 
 | 	/* Update local domain size based on current size of cluster */ | 
 | 	member_cnt = dom_size(mon->peer_cnt) - 1; | 
 | 	self->applied = member_cnt; | 
 |  | 
 | 	/* Update native and cached outgoing local domain records */ | 
 | 	dom->len = dom_rec_len(dom, member_cnt); | 
 | 	diff = dom->member_cnt != member_cnt; | 
 | 	dom->member_cnt = member_cnt; | 
 | 	for (i = 0; i < member_cnt; i++) { | 
 | 		peer = peer_nxt(peer); | 
 | 		diff |= dom->members[i] != peer->addr; | 
 | 		dom->members[i] = peer->addr; | 
 | 		map_set(&dom->up_map, i, peer->is_up); | 
 | 		cache->members[i] = htonl(peer->addr); | 
 | 	} | 
 | 	diff |= dom->up_map != prev_up_map; | 
 | 	if (!diff) | 
 | 		return; | 
 | 	dom->gen = ++mon->dom_gen; | 
 | 	cache->len = htons(dom->len); | 
 | 	cache->gen = htons(dom->gen); | 
 | 	cache->member_cnt = htons(member_cnt); | 
 | 	cache->up_map = cpu_to_be64(dom->up_map); | 
 | 	mon_apply_domain(mon, self); | 
 | } | 
 |  | 
 | /* mon_update_neighbors() : update preceding neighbors of added/removed peer | 
 |  */ | 
 | static void mon_update_neighbors(struct tipc_monitor *mon, | 
 | 				 struct tipc_peer *peer) | 
 | { | 
 | 	int dz, i; | 
 |  | 
 | 	dz = dom_size(mon->peer_cnt); | 
 | 	for (i = 0; i < dz; i++) { | 
 | 		mon_apply_domain(mon, peer); | 
 | 		peer = peer_prev(peer); | 
 | 	} | 
 | } | 
 |  | 
 | /* mon_assign_roles() : reassign peer roles after a network change | 
 |  * The monitor list is consistent at this stage; i.e., each peer is monitoring | 
 |  * a set of domain members as matched between domain record and the monitor list | 
 |  */ | 
 | static void mon_assign_roles(struct tipc_monitor *mon, struct tipc_peer *head) | 
 | { | 
 | 	struct tipc_peer *peer = peer_nxt(head); | 
 | 	struct tipc_peer *self = mon->self; | 
 | 	int i = 0; | 
 |  | 
 | 	for (; peer != self; peer = peer_nxt(peer)) { | 
 | 		peer->is_local = false; | 
 |  | 
 | 		/* Update domain member */ | 
 | 		if (i++ < head->applied) { | 
 | 			peer->is_head = false; | 
 | 			if (head == self) | 
 | 				peer->is_local = true; | 
 | 			continue; | 
 | 		} | 
 | 		/* Assign next domain head */ | 
 | 		if (!peer->is_up) | 
 | 			continue; | 
 | 		if (peer->is_head) | 
 | 			break; | 
 | 		head = peer; | 
 | 		head->is_head = true; | 
 | 		i = 0; | 
 | 	} | 
 | 	mon->list_gen++; | 
 | } | 
 |  | 
 | void tipc_mon_remove_peer(struct net *net, u32 addr, int bearer_id) | 
 | { | 
 | 	struct tipc_monitor *mon = tipc_monitor(net, bearer_id); | 
 | 	struct tipc_peer *self = get_self(net, bearer_id); | 
 | 	struct tipc_peer *peer, *prev, *head; | 
 |  | 
 | 	write_lock_bh(&mon->lock); | 
 | 	peer = get_peer(mon, addr); | 
 | 	if (!peer) | 
 | 		goto exit; | 
 | 	prev = peer_prev(peer); | 
 | 	list_del(&peer->list); | 
 | 	hlist_del(&peer->hash); | 
 | 	kfree(peer->domain); | 
 | 	kfree(peer); | 
 | 	mon->peer_cnt--; | 
 | 	head = peer_head(prev); | 
 | 	if (head == self) | 
 | 		mon_update_local_domain(mon); | 
 | 	mon_update_neighbors(mon, prev); | 
 |  | 
 | 	/* Revert to full-mesh monitoring if we reach threshold */ | 
 | 	if (!tipc_mon_is_active(net, mon)) { | 
 | 		list_for_each_entry(peer, &self->list, list) { | 
 | 			kfree(peer->domain); | 
 | 			peer->domain = NULL; | 
 | 			peer->applied = 0; | 
 | 		} | 
 | 	} | 
 | 	mon_assign_roles(mon, head); | 
 | exit: | 
 | 	write_unlock_bh(&mon->lock); | 
 | } | 
 |  | 
 | static bool tipc_mon_add_peer(struct tipc_monitor *mon, u32 addr, | 
 | 			      struct tipc_peer **peer) | 
 | { | 
 | 	struct tipc_peer *self = mon->self; | 
 | 	struct tipc_peer *cur, *prev, *p; | 
 |  | 
 | 	p = kzalloc(sizeof(*p), GFP_ATOMIC); | 
 | 	*peer = p; | 
 | 	if (!p) | 
 | 		return false; | 
 | 	p->addr = addr; | 
 |  | 
 | 	/* Add new peer to lookup list */ | 
 | 	INIT_LIST_HEAD(&p->list); | 
 | 	hlist_add_head(&p->hash, &mon->peers[tipc_hashfn(addr)]); | 
 |  | 
 | 	/* Sort new peer into iterator list, in ascending circular order */ | 
 | 	prev = self; | 
 | 	list_for_each_entry(cur, &self->list, list) { | 
 | 		if ((addr > prev->addr) && (addr < cur->addr)) | 
 | 			break; | 
 | 		if (((addr < cur->addr) || (addr > prev->addr)) && | 
 | 		    (prev->addr > cur->addr)) | 
 | 			break; | 
 | 		prev = cur; | 
 | 	} | 
 | 	list_add_tail(&p->list, &cur->list); | 
 | 	mon->peer_cnt++; | 
 | 	mon_update_neighbors(mon, p); | 
 | 	return true; | 
 | } | 
 |  | 
 | void tipc_mon_peer_up(struct net *net, u32 addr, int bearer_id) | 
 | { | 
 | 	struct tipc_monitor *mon = tipc_monitor(net, bearer_id); | 
 | 	struct tipc_peer *self = get_self(net, bearer_id); | 
 | 	struct tipc_peer *peer, *head; | 
 |  | 
 | 	write_lock_bh(&mon->lock); | 
 | 	peer = get_peer(mon, addr); | 
 | 	if (!peer && !tipc_mon_add_peer(mon, addr, &peer)) | 
 | 		goto exit; | 
 | 	peer->is_up = true; | 
 | 	head = peer_head(peer); | 
 | 	if (head == self) | 
 | 		mon_update_local_domain(mon); | 
 | 	mon_assign_roles(mon, head); | 
 | exit: | 
 | 	write_unlock_bh(&mon->lock); | 
 | } | 
 |  | 
 | void tipc_mon_peer_down(struct net *net, u32 addr, int bearer_id) | 
 | { | 
 | 	struct tipc_monitor *mon = tipc_monitor(net, bearer_id); | 
 | 	struct tipc_peer *self = get_self(net, bearer_id); | 
 | 	struct tipc_peer *peer, *head; | 
 | 	struct tipc_mon_domain *dom; | 
 | 	int applied; | 
 |  | 
 | 	write_lock_bh(&mon->lock); | 
 | 	peer = get_peer(mon, addr); | 
 | 	if (!peer) { | 
 | 		pr_warn("Mon: unknown link %x/%u DOWN\n", addr, bearer_id); | 
 | 		goto exit; | 
 | 	} | 
 | 	applied = peer->applied; | 
 | 	peer->applied = 0; | 
 | 	dom = peer->domain; | 
 | 	peer->domain = NULL; | 
 | 	if (peer->is_head) | 
 | 		mon_identify_lost_members(peer, dom, applied); | 
 | 	kfree(dom); | 
 | 	peer->is_up = false; | 
 | 	peer->is_head = false; | 
 | 	peer->is_local = false; | 
 | 	peer->down_cnt = 0; | 
 | 	head = peer_head(peer); | 
 | 	if (head == self) | 
 | 		mon_update_local_domain(mon); | 
 | 	mon_assign_roles(mon, head); | 
 | exit: | 
 | 	write_unlock_bh(&mon->lock); | 
 | } | 
 |  | 
 | /* tipc_mon_rcv - process monitor domain event message | 
 |  */ | 
 | void tipc_mon_rcv(struct net *net, void *data, u16 dlen, u32 addr, | 
 | 		  struct tipc_mon_state *state, int bearer_id) | 
 | { | 
 | 	struct tipc_monitor *mon = tipc_monitor(net, bearer_id); | 
 | 	struct tipc_mon_domain *arrv_dom = data; | 
 | 	struct tipc_mon_domain dom_bef; | 
 | 	struct tipc_mon_domain *dom; | 
 | 	struct tipc_peer *peer; | 
 | 	u16 new_member_cnt = ntohs(arrv_dom->member_cnt); | 
 | 	int new_dlen = dom_rec_len(arrv_dom, new_member_cnt); | 
 | 	u16 new_gen = ntohs(arrv_dom->gen); | 
 | 	u16 acked_gen = ntohs(arrv_dom->ack_gen); | 
 | 	bool probing = state->probing; | 
 | 	int i, applied_bef; | 
 |  | 
 | 	state->probing = false; | 
 |  | 
 | 	/* Sanity check received domain record */ | 
 | 	if (dlen < dom_rec_len(arrv_dom, 0)) | 
 | 		return; | 
 | 	if (dlen != dom_rec_len(arrv_dom, new_member_cnt)) | 
 | 		return; | 
 | 	if ((dlen < new_dlen) || ntohs(arrv_dom->len) != new_dlen) | 
 | 		return; | 
 |  | 
 | 	/* Synch generation numbers with peer if link just came up */ | 
 | 	if (!state->synched) { | 
 | 		state->peer_gen = new_gen - 1; | 
 | 		state->acked_gen = acked_gen; | 
 | 		state->synched = true; | 
 | 	} | 
 |  | 
 | 	if (more(acked_gen, state->acked_gen)) | 
 | 		state->acked_gen = acked_gen; | 
 |  | 
 | 	/* Drop duplicate unless we are waiting for a probe response */ | 
 | 	if (!more(new_gen, state->peer_gen) && !probing) | 
 | 		return; | 
 |  | 
 | 	write_lock_bh(&mon->lock); | 
 | 	peer = get_peer(mon, addr); | 
 | 	if (!peer || !peer->is_up) | 
 | 		goto exit; | 
 |  | 
 | 	/* Peer is confirmed, stop any ongoing probing */ | 
 | 	peer->down_cnt = 0; | 
 |  | 
 | 	/* Task is done for duplicate record */ | 
 | 	if (!more(new_gen, state->peer_gen)) | 
 | 		goto exit; | 
 |  | 
 | 	state->peer_gen = new_gen; | 
 |  | 
 | 	/* Cache current domain record for later use */ | 
 | 	dom_bef.member_cnt = 0; | 
 | 	dom = peer->domain; | 
 | 	if (dom) | 
 | 		memcpy(&dom_bef, dom, dom->len); | 
 |  | 
 | 	/* Transform and store received domain record */ | 
 | 	if (!dom || (dom->len < new_dlen)) { | 
 | 		kfree(dom); | 
 | 		dom = kmalloc(new_dlen, GFP_ATOMIC); | 
 | 		peer->domain = dom; | 
 | 		if (!dom) | 
 | 			goto exit; | 
 | 	} | 
 | 	dom->len = new_dlen; | 
 | 	dom->gen = new_gen; | 
 | 	dom->member_cnt = new_member_cnt; | 
 | 	dom->up_map = be64_to_cpu(arrv_dom->up_map); | 
 | 	for (i = 0; i < new_member_cnt; i++) | 
 | 		dom->members[i] = ntohl(arrv_dom->members[i]); | 
 |  | 
 | 	/* Update peers affected by this domain record */ | 
 | 	applied_bef = peer->applied; | 
 | 	mon_apply_domain(mon, peer); | 
 | 	mon_identify_lost_members(peer, &dom_bef, applied_bef); | 
 | 	mon_assign_roles(mon, peer_head(peer)); | 
 | exit: | 
 | 	write_unlock_bh(&mon->lock); | 
 | } | 
 |  | 
 | void tipc_mon_prep(struct net *net, void *data, int *dlen, | 
 | 		   struct tipc_mon_state *state, int bearer_id) | 
 | { | 
 | 	struct tipc_monitor *mon = tipc_monitor(net, bearer_id); | 
 | 	struct tipc_mon_domain *dom = data; | 
 | 	u16 gen = mon->dom_gen; | 
 | 	u16 len; | 
 |  | 
 | 	/* Send invalid record if not active */ | 
 | 	if (!tipc_mon_is_active(net, mon)) { | 
 | 		dom->len = 0; | 
 | 		return; | 
 | 	} | 
 |  | 
 | 	/* Send only a dummy record with ack if peer has acked our last sent */ | 
 | 	if (likely(state->acked_gen == gen)) { | 
 | 		len = dom_rec_len(dom, 0); | 
 | 		*dlen = len; | 
 | 		dom->len = htons(len); | 
 | 		dom->gen = htons(gen); | 
 | 		dom->ack_gen = htons(state->peer_gen); | 
 | 		dom->member_cnt = 0; | 
 | 		return; | 
 | 	} | 
 | 	/* Send the full record */ | 
 | 	read_lock_bh(&mon->lock); | 
 | 	len = ntohs(mon->cache.len); | 
 | 	*dlen = len; | 
 | 	memcpy(data, &mon->cache, len); | 
 | 	read_unlock_bh(&mon->lock); | 
 | 	dom->ack_gen = htons(state->peer_gen); | 
 | } | 
 |  | 
 | void tipc_mon_get_state(struct net *net, u32 addr, | 
 | 			struct tipc_mon_state *state, | 
 | 			int bearer_id) | 
 | { | 
 | 	struct tipc_monitor *mon = tipc_monitor(net, bearer_id); | 
 | 	struct tipc_peer *peer; | 
 |  | 
 | 	if (!tipc_mon_is_active(net, mon)) { | 
 | 		state->probing = false; | 
 | 		state->monitoring = true; | 
 | 		return; | 
 | 	} | 
 |  | 
 | 	/* Used cached state if table has not changed */ | 
 | 	if (!state->probing && | 
 | 	    (state->list_gen == mon->list_gen) && | 
 | 	    (state->acked_gen == mon->dom_gen)) | 
 | 		return; | 
 |  | 
 | 	read_lock_bh(&mon->lock); | 
 | 	peer = get_peer(mon, addr); | 
 | 	if (peer) { | 
 | 		state->probing = state->acked_gen != mon->dom_gen; | 
 | 		state->probing |= peer->down_cnt; | 
 | 		state->reset |= peer->down_cnt >= MAX_PEER_DOWN_EVENTS; | 
 | 		state->monitoring = peer->is_local; | 
 | 		state->monitoring |= peer->is_head; | 
 | 		state->list_gen = mon->list_gen; | 
 | 	} | 
 | 	read_unlock_bh(&mon->lock); | 
 | } | 
 |  | 
 | static void mon_timeout(struct timer_list *t) | 
 | { | 
 | 	struct tipc_monitor *mon = from_timer(mon, t, timer); | 
 | 	struct tipc_peer *self; | 
 | 	int best_member_cnt = dom_size(mon->peer_cnt) - 1; | 
 |  | 
 | 	write_lock_bh(&mon->lock); | 
 | 	self = mon->self; | 
 | 	if (self && (best_member_cnt != self->applied)) { | 
 | 		mon_update_local_domain(mon); | 
 | 		mon_assign_roles(mon, self); | 
 | 	} | 
 | 	write_unlock_bh(&mon->lock); | 
 | 	mod_timer(&mon->timer, jiffies + mon->timer_intv); | 
 | } | 
 |  | 
 | int tipc_mon_create(struct net *net, int bearer_id) | 
 | { | 
 | 	struct tipc_net *tn = tipc_net(net); | 
 | 	struct tipc_monitor *mon; | 
 | 	struct tipc_peer *self; | 
 | 	struct tipc_mon_domain *dom; | 
 |  | 
 | 	if (tn->monitors[bearer_id]) | 
 | 		return 0; | 
 |  | 
 | 	mon = kzalloc(sizeof(*mon), GFP_ATOMIC); | 
 | 	self = kzalloc(sizeof(*self), GFP_ATOMIC); | 
 | 	dom = kzalloc(sizeof(*dom), GFP_ATOMIC); | 
 | 	if (!mon || !self || !dom) { | 
 | 		kfree(mon); | 
 | 		kfree(self); | 
 | 		kfree(dom); | 
 | 		return -ENOMEM; | 
 | 	} | 
 | 	tn->monitors[bearer_id] = mon; | 
 | 	rwlock_init(&mon->lock); | 
 | 	mon->net = net; | 
 | 	mon->peer_cnt = 1; | 
 | 	mon->self = self; | 
 | 	self->domain = dom; | 
 | 	self->addr = tipc_own_addr(net); | 
 | 	self->is_up = true; | 
 | 	self->is_head = true; | 
 | 	INIT_LIST_HEAD(&self->list); | 
 | 	timer_setup(&mon->timer, mon_timeout, 0); | 
 | 	mon->timer_intv = msecs_to_jiffies(MON_TIMEOUT + (tn->random & 0xffff)); | 
 | 	mod_timer(&mon->timer, jiffies + mon->timer_intv); | 
 | 	return 0; | 
 | } | 
 |  | 
 | void tipc_mon_delete(struct net *net, int bearer_id) | 
 | { | 
 | 	struct tipc_net *tn = tipc_net(net); | 
 | 	struct tipc_monitor *mon = tipc_monitor(net, bearer_id); | 
 | 	struct tipc_peer *self; | 
 | 	struct tipc_peer *peer, *tmp; | 
 |  | 
 | 	if (!mon) | 
 | 		return; | 
 |  | 
 | 	self = get_self(net, bearer_id); | 
 | 	write_lock_bh(&mon->lock); | 
 | 	tn->monitors[bearer_id] = NULL; | 
 | 	list_for_each_entry_safe(peer, tmp, &self->list, list) { | 
 | 		list_del(&peer->list); | 
 | 		hlist_del(&peer->hash); | 
 | 		kfree(peer->domain); | 
 | 		kfree(peer); | 
 | 	} | 
 | 	mon->self = NULL; | 
 | 	write_unlock_bh(&mon->lock); | 
 | 	del_timer_sync(&mon->timer); | 
 | 	kfree(self->domain); | 
 | 	kfree(self); | 
 | 	kfree(mon); | 
 | } | 
 |  | 
 | int tipc_nl_monitor_set_threshold(struct net *net, u32 cluster_size) | 
 | { | 
 | 	struct tipc_net *tn = tipc_net(net); | 
 |  | 
 | 	if (cluster_size > TIPC_CLUSTER_SIZE) | 
 | 		return -EINVAL; | 
 |  | 
 | 	tn->mon_threshold = cluster_size; | 
 |  | 
 | 	return 0; | 
 | } | 
 |  | 
 | int tipc_nl_monitor_get_threshold(struct net *net) | 
 | { | 
 | 	struct tipc_net *tn = tipc_net(net); | 
 |  | 
 | 	return tn->mon_threshold; | 
 | } | 
 |  | 
 | static int __tipc_nl_add_monitor_peer(struct tipc_peer *peer, | 
 | 				      struct tipc_nl_msg *msg) | 
 | { | 
 | 	struct tipc_mon_domain *dom = peer->domain; | 
 | 	struct nlattr *attrs; | 
 | 	void *hdr; | 
 |  | 
 | 	hdr = genlmsg_put(msg->skb, msg->portid, msg->seq, &tipc_genl_family, | 
 | 			  NLM_F_MULTI, TIPC_NL_MON_PEER_GET); | 
 | 	if (!hdr) | 
 | 		return -EMSGSIZE; | 
 |  | 
 | 	attrs = nla_nest_start(msg->skb, TIPC_NLA_MON_PEER); | 
 | 	if (!attrs) | 
 | 		goto msg_full; | 
 |  | 
 | 	if (nla_put_u32(msg->skb, TIPC_NLA_MON_PEER_ADDR, peer->addr)) | 
 | 		goto attr_msg_full; | 
 | 	if (nla_put_u32(msg->skb, TIPC_NLA_MON_PEER_APPLIED, peer->applied)) | 
 | 		goto attr_msg_full; | 
 |  | 
 | 	if (peer->is_up) | 
 | 		if (nla_put_flag(msg->skb, TIPC_NLA_MON_PEER_UP)) | 
 | 			goto attr_msg_full; | 
 | 	if (peer->is_local) | 
 | 		if (nla_put_flag(msg->skb, TIPC_NLA_MON_PEER_LOCAL)) | 
 | 			goto attr_msg_full; | 
 | 	if (peer->is_head) | 
 | 		if (nla_put_flag(msg->skb, TIPC_NLA_MON_PEER_HEAD)) | 
 | 			goto attr_msg_full; | 
 |  | 
 | 	if (dom) { | 
 | 		if (nla_put_u32(msg->skb, TIPC_NLA_MON_PEER_DOMGEN, dom->gen)) | 
 | 			goto attr_msg_full; | 
 | 		if (nla_put_u64_64bit(msg->skb, TIPC_NLA_MON_PEER_UPMAP, | 
 | 				      dom->up_map, TIPC_NLA_MON_PEER_PAD)) | 
 | 			goto attr_msg_full; | 
 | 		if (nla_put(msg->skb, TIPC_NLA_MON_PEER_MEMBERS, | 
 | 			    dom->member_cnt * sizeof(u32), &dom->members)) | 
 | 			goto attr_msg_full; | 
 | 	} | 
 |  | 
 | 	nla_nest_end(msg->skb, attrs); | 
 | 	genlmsg_end(msg->skb, hdr); | 
 | 	return 0; | 
 |  | 
 | attr_msg_full: | 
 | 	nla_nest_cancel(msg->skb, attrs); | 
 | msg_full: | 
 | 	genlmsg_cancel(msg->skb, hdr); | 
 |  | 
 | 	return -EMSGSIZE; | 
 | } | 
 |  | 
 | int tipc_nl_add_monitor_peer(struct net *net, struct tipc_nl_msg *msg, | 
 | 			     u32 bearer_id, u32 *prev_node) | 
 | { | 
 | 	struct tipc_monitor *mon = tipc_monitor(net, bearer_id); | 
 | 	struct tipc_peer *peer; | 
 |  | 
 | 	if (!mon) | 
 | 		return -EINVAL; | 
 |  | 
 | 	read_lock_bh(&mon->lock); | 
 | 	peer = mon->self; | 
 | 	do { | 
 | 		if (*prev_node) { | 
 | 			if (peer->addr == *prev_node) | 
 | 				*prev_node = 0; | 
 | 			else | 
 | 				continue; | 
 | 		} | 
 | 		if (__tipc_nl_add_monitor_peer(peer, msg)) { | 
 | 			*prev_node = peer->addr; | 
 | 			read_unlock_bh(&mon->lock); | 
 | 			return -EMSGSIZE; | 
 | 		} | 
 | 	} while ((peer = peer_nxt(peer)) != mon->self); | 
 | 	read_unlock_bh(&mon->lock); | 
 |  | 
 | 	return 0; | 
 | } | 
 |  | 
 | int __tipc_nl_add_monitor(struct net *net, struct tipc_nl_msg *msg, | 
 | 			  u32 bearer_id) | 
 | { | 
 | 	struct tipc_monitor *mon = tipc_monitor(net, bearer_id); | 
 | 	char bearer_name[TIPC_MAX_BEARER_NAME]; | 
 | 	struct nlattr *attrs; | 
 | 	void *hdr; | 
 | 	int ret; | 
 |  | 
 | 	ret = tipc_bearer_get_name(net, bearer_name, bearer_id); | 
 | 	if (ret || !mon) | 
 | 		return 0; | 
 |  | 
 | 	hdr = genlmsg_put(msg->skb, msg->portid, msg->seq, &tipc_genl_family, | 
 | 			  NLM_F_MULTI, TIPC_NL_MON_GET); | 
 | 	if (!hdr) | 
 | 		return -EMSGSIZE; | 
 |  | 
 | 	attrs = nla_nest_start(msg->skb, TIPC_NLA_MON); | 
 | 	if (!attrs) | 
 | 		goto msg_full; | 
 |  | 
 | 	read_lock_bh(&mon->lock); | 
 | 	if (nla_put_u32(msg->skb, TIPC_NLA_MON_REF, bearer_id)) | 
 | 		goto attr_msg_full; | 
 | 	if (tipc_mon_is_active(net, mon)) | 
 | 		if (nla_put_flag(msg->skb, TIPC_NLA_MON_ACTIVE)) | 
 | 			goto attr_msg_full; | 
 | 	if (nla_put_string(msg->skb, TIPC_NLA_MON_BEARER_NAME, bearer_name)) | 
 | 		goto attr_msg_full; | 
 | 	if (nla_put_u32(msg->skb, TIPC_NLA_MON_PEERCNT, mon->peer_cnt)) | 
 | 		goto attr_msg_full; | 
 | 	if (nla_put_u32(msg->skb, TIPC_NLA_MON_LISTGEN, mon->list_gen)) | 
 | 		goto attr_msg_full; | 
 |  | 
 | 	read_unlock_bh(&mon->lock); | 
 | 	nla_nest_end(msg->skb, attrs); | 
 | 	genlmsg_end(msg->skb, hdr); | 
 |  | 
 | 	return 0; | 
 |  | 
 | attr_msg_full: | 
 | 	read_unlock_bh(&mon->lock); | 
 | 	nla_nest_cancel(msg->skb, attrs); | 
 | msg_full: | 
 | 	genlmsg_cancel(msg->skb, hdr); | 
 |  | 
 | 	return -EMSGSIZE; | 
 | } |