/* SPDX-License-Identifier: GPL-2.0-or-later
 * Copyright (c) 2020, Nikolay Aleksandrov <nikolay@nvidia.com>
 */
#ifndef _BR_PRIVATE_MCAST_EHT_H_
#define _BR_PRIVATE_MCAST_EHT_H_

#define BR_MCAST_DEFAULT_EHT_HOSTS_LIMIT 512

union net_bridge_eht_addr {
	__be32				ip4;
#if IS_ENABLED(CONFIG_IPV6)
	struct in6_addr			ip6;
#endif
};

/* single host's list of set entries and filter_mode */
struct net_bridge_group_eht_host {
	struct rb_node			rb_node;

	union net_bridge_eht_addr	h_addr;
	struct hlist_head		set_entries;
	unsigned int			num_entries;
	unsigned char			filter_mode;
	struct net_bridge_port_group	*pg;
};

/* (host, src entry) added to a per-src set and host's list */
struct net_bridge_group_eht_set_entry {
	struct rb_node			rb_node;
	struct hlist_node		host_list;

	union net_bridge_eht_addr	h_addr;
	struct timer_list		timer;
	struct net_bridge		*br;
	struct net_bridge_group_eht_set	*eht_set;
	struct net_bridge_group_eht_host *h_parent;
	struct net_bridge_mcast_gc	mcast_gc;
};

/* per-src set */
struct net_bridge_group_eht_set {
	struct rb_node			rb_node;

	union net_bridge_eht_addr	src_addr;
	struct rb_root			entry_tree;
	struct timer_list		timer;
	struct net_bridge_port_group	*pg;
	struct net_bridge		*br;
	struct net_bridge_mcast_gc	mcast_gc;
};

#ifdef CONFIG_BRIDGE_IGMP_SNOOPING
void br_multicast_eht_clean_sets(struct net_bridge_port_group *pg);
bool br_multicast_eht_handle(const struct net_bridge_mcast *brmctx,
			     struct net_bridge_port_group *pg,
			     void *h_addr,
			     void *srcs,
			     u32 nsrcs,
			     size_t addr_size,
			     int grec_type);
int br_multicast_eht_set_hosts_limit(struct net_bridge_port *p,
				     u32 eht_hosts_limit);

static inline bool
br_multicast_eht_should_del_pg(const struct net_bridge_port_group *pg)
{
	return !!((pg->key.port->flags & BR_MULTICAST_FAST_LEAVE) &&
		  RB_EMPTY_ROOT(&pg->eht_host_tree));
}

static inline bool
br_multicast_eht_hosts_over_limit(const struct net_bridge_port_group *pg)
{
	const struct net_bridge_port *p = pg->key.port;

	return !!(p->multicast_eht_hosts_cnt >= p->multicast_eht_hosts_limit);
}

static inline void br_multicast_eht_hosts_inc(struct net_bridge_port_group *pg)
{
	struct net_bridge_port *p = pg->key.port;

	p->multicast_eht_hosts_cnt++;
}

static inline void br_multicast_eht_hosts_dec(struct net_bridge_port_group *pg)
{
	struct net_bridge_port *p = pg->key.port;

	p->multicast_eht_hosts_cnt--;
}
#endif /* CONFIG_BRIDGE_IGMP_SNOOPING */

#endif /* _BR_PRIVATE_MCAST_EHT_H_ */