brintos

brintos / linux-shallow public Read only

0
0
Text · 35.3 KiB · 33f525b Raw
1268 lines · c
1/*2 * Copyright (c) 2012 Mellanox Technologies. All rights reserved.3 *4 * This software is available to you under a choice of one of two5 * licenses.  You may choose to be licensed under the terms of the GNU6 * General Public License (GPL) Version 2, available from the file7 * COPYING in the main directory of this source tree, or the8 * OpenIB.org BSD license below:9 *10 *     Redistribution and use in source and binary forms, with or11 *     without modification, are permitted provided that the following12 *     conditions are met:13 *14 *      - Redistributions of source code must retain the above15 *        copyright notice, this list of conditions and the following16 *        disclaimer.17 *18 *      - Redistributions in binary form must reproduce the above19 *        copyright notice, this list of conditions and the following20 *        disclaimer in the documentation and/or other materials21 *        provided with the distribution.22 *23 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,24 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF25 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND26 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS27 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN28 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN29 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE30 * SOFTWARE.31 */32 33#include <rdma/ib_mad.h>34#include <rdma/ib_smi.h>35#include <rdma/ib_cache.h>36#include <rdma/ib_sa.h>37 38#include <linux/mlx4/cmd.h>39#include <linux/rbtree.h>40#include <linux/delay.h>41 42#include "mlx4_ib.h"43 44#define MAX_VFS		8045#define MAX_PEND_REQS_PER_FUNC 446#define MAD_TIMEOUT_MS	200047 48#define mcg_warn(fmt, arg...)	pr_warn("MCG WARNING: " fmt, ##arg)49#define mcg_error(fmt, arg...)	pr_err(fmt, ##arg)50#define mcg_warn_group(group, format, arg...) \51	pr_warn("%s-%d: %16s (port %d): WARNING: " format, __func__, __LINE__,\52	(group)->name, group->demux->port, ## arg)53 54#define mcg_debug_group(group, format, arg...) \55	pr_debug("%s-%d: %16s (port %d): WARNING: " format, __func__, __LINE__,\56		 (group)->name, (group)->demux->port, ## arg)57 58#define mcg_error_group(group, format, arg...) \59	pr_err("  %16s: " format, (group)->name, ## arg)60 61 62static union ib_gid mgid0;63 64static struct workqueue_struct *clean_wq;65 66enum mcast_state {67	MCAST_NOT_MEMBER = 0,68	MCAST_MEMBER,69};70 71enum mcast_group_state {72	MCAST_IDLE,73	MCAST_JOIN_SENT,74	MCAST_LEAVE_SENT,75	MCAST_RESP_READY76};77 78struct mcast_member {79	enum mcast_state state;80	uint8_t			join_state;81	int			num_pend_reqs;82	struct list_head	pending;83};84 85struct ib_sa_mcmember_data {86	union ib_gid	mgid;87	union ib_gid	port_gid;88	__be32		qkey;89	__be16		mlid;90	u8		mtusel_mtu;91	u8		tclass;92	__be16		pkey;93	u8		ratesel_rate;94	u8		lifetmsel_lifetm;95	__be32		sl_flowlabel_hoplimit;96	u8		scope_join_state;97	u8		proxy_join;98	u8		reserved[2];99} __packed __aligned(4);100 101struct mcast_group {102	struct ib_sa_mcmember_data rec;103	struct rb_node		node;104	struct list_head	mgid0_list;105	struct mlx4_ib_demux_ctx *demux;106	struct mcast_member	func[MAX_VFS];107	struct mutex		lock;108	struct work_struct	work;109	struct list_head	pending_list;110	int			members[3];111	enum mcast_group_state	state;112	enum mcast_group_state	prev_state;113	struct ib_sa_mad	response_sa_mad;114	__be64			last_req_tid;115 116	char			name[33]; /* MGID string */117	struct device_attribute	dentry;118 119	/* refcount is the reference count for the following:120	   1. Each queued request121	   2. Each invocation of the worker thread122	   3. Membership of the port at the SA123	*/124	atomic_t		refcount;125 126	/* delayed work to clean pending SM request */127	struct delayed_work	timeout_work;128	struct list_head	cleanup_list;129};130 131struct mcast_req {132	int			func;133	struct ib_sa_mad	sa_mad;134	struct list_head	group_list;135	struct list_head	func_list;136	struct mcast_group	*group;137	int			clean;138};139 140 141#define safe_atomic_dec(ref) \142	do {\143		if (atomic_dec_and_test(ref)) \144			mcg_warn_group(group, "did not expect to reach zero\n"); \145	} while (0)146 147static const char *get_state_string(enum mcast_group_state state)148{149	switch (state) {150	case MCAST_IDLE:151		return "MCAST_IDLE";152	case MCAST_JOIN_SENT:153		return "MCAST_JOIN_SENT";154	case MCAST_LEAVE_SENT:155		return "MCAST_LEAVE_SENT";156	case MCAST_RESP_READY:157		return "MCAST_RESP_READY";158	}159	return "Invalid State";160}161 162static struct mcast_group *mcast_find(struct mlx4_ib_demux_ctx *ctx,163				      union ib_gid *mgid)164{165	struct rb_node *node = ctx->mcg_table.rb_node;166	struct mcast_group *group;167	int ret;168 169	while (node) {170		group = rb_entry(node, struct mcast_group, node);171		ret = memcmp(mgid->raw, group->rec.mgid.raw, sizeof *mgid);172		if (!ret)173			return group;174 175		if (ret < 0)176			node = node->rb_left;177		else178			node = node->rb_right;179	}180	return NULL;181}182 183static struct mcast_group *mcast_insert(struct mlx4_ib_demux_ctx *ctx,184					struct mcast_group *group)185{186	struct rb_node **link = &ctx->mcg_table.rb_node;187	struct rb_node *parent = NULL;188	struct mcast_group *cur_group;189	int ret;190 191	while (*link) {192		parent = *link;193		cur_group = rb_entry(parent, struct mcast_group, node);194 195		ret = memcmp(group->rec.mgid.raw, cur_group->rec.mgid.raw,196			     sizeof group->rec.mgid);197		if (ret < 0)198			link = &(*link)->rb_left;199		else if (ret > 0)200			link = &(*link)->rb_right;201		else202			return cur_group;203	}204	rb_link_node(&group->node, parent, link);205	rb_insert_color(&group->node, &ctx->mcg_table);206	return NULL;207}208 209static int send_mad_to_wire(struct mlx4_ib_demux_ctx *ctx, struct ib_mad *mad)210{211	struct mlx4_ib_dev *dev = ctx->dev;212	struct rdma_ah_attr	ah_attr;213	unsigned long flags;214 215	spin_lock_irqsave(&dev->sm_lock, flags);216	if (!dev->sm_ah[ctx->port - 1]) {217		/* port is not yet Active, sm_ah not ready */218		spin_unlock_irqrestore(&dev->sm_lock, flags);219		return -EAGAIN;220	}221	mlx4_ib_query_ah(dev->sm_ah[ctx->port - 1], &ah_attr);222	spin_unlock_irqrestore(&dev->sm_lock, flags);223	return mlx4_ib_send_to_wire(dev, mlx4_master_func_num(dev->dev),224				    ctx->port, IB_QPT_GSI, 0, 1, IB_QP1_QKEY,225				    &ah_attr, NULL, 0xffff, mad);226}227 228static int send_mad_to_slave(int slave, struct mlx4_ib_demux_ctx *ctx,229			     struct ib_mad *mad)230{231	struct mlx4_ib_dev *dev = ctx->dev;232	struct ib_mad_agent *agent = dev->send_agent[ctx->port - 1][1];233	struct ib_wc wc;234	struct rdma_ah_attr ah_attr;235 236	/* Our agent might not yet be registered when mads start to arrive */237	if (!agent)238		return -EAGAIN;239 240	rdma_query_ah(dev->sm_ah[ctx->port - 1], &ah_attr);241 242	if (ib_find_cached_pkey(&dev->ib_dev, ctx->port, IB_DEFAULT_PKEY_FULL, &wc.pkey_index))243		return -EINVAL;244	wc.sl = 0;245	wc.dlid_path_bits = 0;246	wc.port_num = ctx->port;247	wc.slid = rdma_ah_get_dlid(&ah_attr);  /* opensm lid */248	wc.src_qp = 1;249	return mlx4_ib_send_to_slave(dev, slave, ctx->port, IB_QPT_GSI, &wc, NULL, mad);250}251 252static int send_join_to_wire(struct mcast_group *group, struct ib_sa_mad *sa_mad)253{254	struct ib_sa_mad mad;255	struct ib_sa_mcmember_data *sa_mad_data = (struct ib_sa_mcmember_data *)&mad.data;256	int ret;257 258	/* we rely on a mad request as arrived from a VF */259	memcpy(&mad, sa_mad, sizeof mad);260 261	/* fix port GID to be the real one (slave 0) */262	sa_mad_data->port_gid.global.interface_id = group->demux->guid_cache[0];263 264	/* assign our own TID */265	mad.mad_hdr.tid = mlx4_ib_get_new_demux_tid(group->demux);266	group->last_req_tid = mad.mad_hdr.tid; /* keep it for later validation */267 268	ret = send_mad_to_wire(group->demux, (struct ib_mad *)&mad);269	/* set timeout handler */270	if (!ret) {271		/* calls mlx4_ib_mcg_timeout_handler */272		queue_delayed_work(group->demux->mcg_wq, &group->timeout_work,273				msecs_to_jiffies(MAD_TIMEOUT_MS));274	}275 276	return ret;277}278 279static int send_leave_to_wire(struct mcast_group *group, u8 join_state)280{281	struct ib_sa_mad mad;282	struct ib_sa_mcmember_data *sa_data = (struct ib_sa_mcmember_data *)&mad.data;283	int ret;284 285	memset(&mad, 0, sizeof mad);286	mad.mad_hdr.base_version = 1;287	mad.mad_hdr.mgmt_class = IB_MGMT_CLASS_SUBN_ADM;288	mad.mad_hdr.class_version = 2;289	mad.mad_hdr.method = IB_SA_METHOD_DELETE;290	mad.mad_hdr.status = cpu_to_be16(0);291	mad.mad_hdr.class_specific = cpu_to_be16(0);292	mad.mad_hdr.tid = mlx4_ib_get_new_demux_tid(group->demux);293	group->last_req_tid = mad.mad_hdr.tid; /* keep it for later validation */294	mad.mad_hdr.attr_id = cpu_to_be16(IB_SA_ATTR_MC_MEMBER_REC);295	mad.mad_hdr.attr_mod = cpu_to_be32(0);296	mad.sa_hdr.sm_key = 0x0;297	mad.sa_hdr.attr_offset = cpu_to_be16(7);298	mad.sa_hdr.comp_mask = IB_SA_MCMEMBER_REC_MGID |299		IB_SA_MCMEMBER_REC_PORT_GID | IB_SA_MCMEMBER_REC_JOIN_STATE;300 301	*sa_data = group->rec;302	sa_data->scope_join_state = join_state;303 304	ret = send_mad_to_wire(group->demux, (struct ib_mad *)&mad);305	if (ret)306		group->state = MCAST_IDLE;307 308	/* set timeout handler */309	if (!ret) {310		/* calls mlx4_ib_mcg_timeout_handler */311		queue_delayed_work(group->demux->mcg_wq, &group->timeout_work,312				msecs_to_jiffies(MAD_TIMEOUT_MS));313	}314 315	return ret;316}317 318static int send_reply_to_slave(int slave, struct mcast_group *group,319		struct ib_sa_mad *req_sa_mad, u16 status)320{321	struct ib_sa_mad mad;322	struct ib_sa_mcmember_data *sa_data = (struct ib_sa_mcmember_data *)&mad.data;323	struct ib_sa_mcmember_data *req_sa_data = (struct ib_sa_mcmember_data *)&req_sa_mad->data;324	int ret;325 326	memset(&mad, 0, sizeof mad);327	mad.mad_hdr.base_version = 1;328	mad.mad_hdr.mgmt_class = IB_MGMT_CLASS_SUBN_ADM;329	mad.mad_hdr.class_version = 2;330	mad.mad_hdr.method = IB_MGMT_METHOD_GET_RESP;331	mad.mad_hdr.status = cpu_to_be16(status);332	mad.mad_hdr.class_specific = cpu_to_be16(0);333	mad.mad_hdr.tid = req_sa_mad->mad_hdr.tid;334	*(u8 *)&mad.mad_hdr.tid = 0; /* resetting tid to 0 */335	mad.mad_hdr.attr_id = cpu_to_be16(IB_SA_ATTR_MC_MEMBER_REC);336	mad.mad_hdr.attr_mod = cpu_to_be32(0);337	mad.sa_hdr.sm_key = req_sa_mad->sa_hdr.sm_key;338	mad.sa_hdr.attr_offset = cpu_to_be16(7);339	mad.sa_hdr.comp_mask = 0; /* ignored on responses, see IBTA spec */340 341	*sa_data = group->rec;342 343	/* reconstruct VF's requested join_state and port_gid */344	sa_data->scope_join_state &= 0xf0;345	sa_data->scope_join_state |= (group->func[slave].join_state & 0x0f);346	memcpy(&sa_data->port_gid, &req_sa_data->port_gid, sizeof req_sa_data->port_gid);347 348	ret = send_mad_to_slave(slave, group->demux, (struct ib_mad *)&mad);349	return ret;350}351 352static int check_selector(ib_sa_comp_mask comp_mask,353			  ib_sa_comp_mask selector_mask,354			  ib_sa_comp_mask value_mask,355			  u8 src_value, u8 dst_value)356{357	int err;358	u8 selector = dst_value >> 6;359	dst_value &= 0x3f;360	src_value &= 0x3f;361 362	if (!(comp_mask & selector_mask) || !(comp_mask & value_mask))363		return 0;364 365	switch (selector) {366	case IB_SA_GT:367		err = (src_value <= dst_value);368		break;369	case IB_SA_LT:370		err = (src_value >= dst_value);371		break;372	case IB_SA_EQ:373		err = (src_value != dst_value);374		break;375	default:376		err = 0;377		break;378	}379 380	return err;381}382 383static u16 cmp_rec(struct ib_sa_mcmember_data *src,384		   struct ib_sa_mcmember_data *dst, ib_sa_comp_mask comp_mask)385{386	/* src is group record, dst is request record */387	/* MGID must already match */388	/* Port_GID we always replace to our Port_GID, so it is a match */389 390#define MAD_STATUS_REQ_INVALID 0x0200391	if (comp_mask & IB_SA_MCMEMBER_REC_QKEY && src->qkey != dst->qkey)392		return MAD_STATUS_REQ_INVALID;393	if (comp_mask & IB_SA_MCMEMBER_REC_MLID && src->mlid != dst->mlid)394		return MAD_STATUS_REQ_INVALID;395	if (check_selector(comp_mask, IB_SA_MCMEMBER_REC_MTU_SELECTOR,396				 IB_SA_MCMEMBER_REC_MTU,397				 src->mtusel_mtu, dst->mtusel_mtu))398		return MAD_STATUS_REQ_INVALID;399	if (comp_mask & IB_SA_MCMEMBER_REC_TRAFFIC_CLASS &&400	    src->tclass != dst->tclass)401		return MAD_STATUS_REQ_INVALID;402	if (comp_mask & IB_SA_MCMEMBER_REC_PKEY && src->pkey != dst->pkey)403		return MAD_STATUS_REQ_INVALID;404	if (check_selector(comp_mask, IB_SA_MCMEMBER_REC_RATE_SELECTOR,405				 IB_SA_MCMEMBER_REC_RATE,406				 src->ratesel_rate, dst->ratesel_rate))407		return MAD_STATUS_REQ_INVALID;408	if (check_selector(comp_mask,409				 IB_SA_MCMEMBER_REC_PACKET_LIFE_TIME_SELECTOR,410				 IB_SA_MCMEMBER_REC_PACKET_LIFE_TIME,411				 src->lifetmsel_lifetm, dst->lifetmsel_lifetm))412		return MAD_STATUS_REQ_INVALID;413	if (comp_mask & IB_SA_MCMEMBER_REC_SL &&414			(be32_to_cpu(src->sl_flowlabel_hoplimit) & 0xf0000000) !=415			(be32_to_cpu(dst->sl_flowlabel_hoplimit) & 0xf0000000))416		return MAD_STATUS_REQ_INVALID;417	if (comp_mask & IB_SA_MCMEMBER_REC_FLOW_LABEL &&418			(be32_to_cpu(src->sl_flowlabel_hoplimit) & 0x0fffff00) !=419			(be32_to_cpu(dst->sl_flowlabel_hoplimit) & 0x0fffff00))420		return MAD_STATUS_REQ_INVALID;421	if (comp_mask & IB_SA_MCMEMBER_REC_HOP_LIMIT &&422			(be32_to_cpu(src->sl_flowlabel_hoplimit) & 0x000000ff) !=423			(be32_to_cpu(dst->sl_flowlabel_hoplimit) & 0x000000ff))424		return MAD_STATUS_REQ_INVALID;425	if (comp_mask & IB_SA_MCMEMBER_REC_SCOPE &&426			(src->scope_join_state & 0xf0) !=427			(dst->scope_join_state & 0xf0))428		return MAD_STATUS_REQ_INVALID;429 430	/* join_state checked separately, proxy_join ignored */431 432	return 0;433}434 435/* release group, return 1 if this was last release and group is destroyed436 * timout work is canceled sync */437static int release_group(struct mcast_group *group, int from_timeout_handler)438{439	struct mlx4_ib_demux_ctx *ctx = group->demux;440	int nzgroup;441 442	mutex_lock(&ctx->mcg_table_lock);443	mutex_lock(&group->lock);444	if (atomic_dec_and_test(&group->refcount)) {445		if (!from_timeout_handler) {446			if (group->state != MCAST_IDLE &&447			    !cancel_delayed_work(&group->timeout_work)) {448				atomic_inc(&group->refcount);449				mutex_unlock(&group->lock);450				mutex_unlock(&ctx->mcg_table_lock);451				return 0;452			}453		}454 455		nzgroup = memcmp(&group->rec.mgid, &mgid0, sizeof mgid0);456		if (nzgroup)457			del_sysfs_port_mcg_attr(ctx->dev, ctx->port, &group->dentry.attr);458		if (!list_empty(&group->pending_list))459			mcg_warn_group(group, "releasing a group with non empty pending list\n");460		if (nzgroup)461			rb_erase(&group->node, &ctx->mcg_table);462		list_del_init(&group->mgid0_list);463		mutex_unlock(&group->lock);464		mutex_unlock(&ctx->mcg_table_lock);465		kfree(group);466		return 1;467	} else {468		mutex_unlock(&group->lock);469		mutex_unlock(&ctx->mcg_table_lock);470	}471	return 0;472}473 474static void adjust_membership(struct mcast_group *group, u8 join_state, int inc)475{476	int i;477 478	for (i = 0; i < 3; i++, join_state >>= 1)479		if (join_state & 0x1)480			group->members[i] += inc;481}482 483static u8 get_leave_state(struct mcast_group *group)484{485	u8 leave_state = 0;486	int i;487 488	for (i = 0; i < 3; i++)489		if (!group->members[i])490			leave_state |= (1 << i);491 492	return leave_state & (group->rec.scope_join_state & 0xf);493}494 495static int join_group(struct mcast_group *group, int slave, u8 join_mask)496{497	int ret = 0;498	u8 join_state;499 500	/* remove bits that slave is already member of, and adjust */501	join_state = join_mask & (~group->func[slave].join_state);502	adjust_membership(group, join_state, 1);503	group->func[slave].join_state |= join_state;504	if (group->func[slave].state != MCAST_MEMBER && join_state) {505		group->func[slave].state = MCAST_MEMBER;506		ret = 1;507	}508	return ret;509}510 511static int leave_group(struct mcast_group *group, int slave, u8 leave_state)512{513	int ret = 0;514 515	adjust_membership(group, leave_state, -1);516	group->func[slave].join_state &= ~leave_state;517	if (!group->func[slave].join_state) {518		group->func[slave].state = MCAST_NOT_MEMBER;519		ret = 1;520	}521	return ret;522}523 524static int check_leave(struct mcast_group *group, int slave, u8 leave_mask)525{526	if (group->func[slave].state != MCAST_MEMBER)527		return MAD_STATUS_REQ_INVALID;528 529	/* make sure we're not deleting unset bits */530	if (~group->func[slave].join_state & leave_mask)531		return MAD_STATUS_REQ_INVALID;532 533	if (!leave_mask)534		return MAD_STATUS_REQ_INVALID;535 536	return 0;537}538 539static void mlx4_ib_mcg_timeout_handler(struct work_struct *work)540{541	struct delayed_work *delay = to_delayed_work(work);542	struct mcast_group *group;543	struct mcast_req *req = NULL;544 545	group = container_of(delay, typeof(*group), timeout_work);546 547	mutex_lock(&group->lock);548	if (group->state == MCAST_JOIN_SENT) {549		if (!list_empty(&group->pending_list)) {550			req = list_first_entry(&group->pending_list, struct mcast_req, group_list);551			list_del(&req->group_list);552			list_del(&req->func_list);553			--group->func[req->func].num_pend_reqs;554			mutex_unlock(&group->lock);555			kfree(req);556			if (memcmp(&group->rec.mgid, &mgid0, sizeof mgid0)) {557				if (release_group(group, 1))558					return;559			} else {560				kfree(group);561				return;562			}563			mutex_lock(&group->lock);564		} else565			mcg_warn_group(group, "DRIVER BUG\n");566	} else if (group->state == MCAST_LEAVE_SENT) {567		if (group->rec.scope_join_state & 0xf)568			group->rec.scope_join_state &= 0xf0;569		group->state = MCAST_IDLE;570		mutex_unlock(&group->lock);571		if (release_group(group, 1))572			return;573		mutex_lock(&group->lock);574	} else575		mcg_warn_group(group, "invalid state %s\n", get_state_string(group->state));576	group->state = MCAST_IDLE;577	atomic_inc(&group->refcount);578	if (!queue_work(group->demux->mcg_wq, &group->work))579		safe_atomic_dec(&group->refcount);580 581	mutex_unlock(&group->lock);582}583 584static int handle_leave_req(struct mcast_group *group, u8 leave_mask,585			    struct mcast_req *req)586{587	u16 status;588 589	if (req->clean)590		leave_mask = group->func[req->func].join_state;591 592	status = check_leave(group, req->func, leave_mask);593	if (!status)594		leave_group(group, req->func, leave_mask);595 596	if (!req->clean)597		send_reply_to_slave(req->func, group, &req->sa_mad, status);598	--group->func[req->func].num_pend_reqs;599	list_del(&req->group_list);600	list_del(&req->func_list);601	kfree(req);602	return 1;603}604 605static int handle_join_req(struct mcast_group *group, u8 join_mask,606			   struct mcast_req *req)607{608	u8 group_join_state = group->rec.scope_join_state & 0xf;609	int ref = 0;610	u16 status;611	struct ib_sa_mcmember_data *sa_data = (struct ib_sa_mcmember_data *)req->sa_mad.data;612 613	if (join_mask == (group_join_state & join_mask)) {614		/* port's membership need not change */615		status = cmp_rec(&group->rec, sa_data, req->sa_mad.sa_hdr.comp_mask);616		if (!status)617			join_group(group, req->func, join_mask);618 619		--group->func[req->func].num_pend_reqs;620		send_reply_to_slave(req->func, group, &req->sa_mad, status);621		list_del(&req->group_list);622		list_del(&req->func_list);623		kfree(req);624		++ref;625	} else {626		/* port's membership needs to be updated */627		group->prev_state = group->state;628		if (send_join_to_wire(group, &req->sa_mad)) {629			--group->func[req->func].num_pend_reqs;630			list_del(&req->group_list);631			list_del(&req->func_list);632			kfree(req);633			ref = 1;634			group->state = group->prev_state;635		} else636			group->state = MCAST_JOIN_SENT;637	}638 639	return ref;640}641 642static void mlx4_ib_mcg_work_handler(struct work_struct *work)643{644	struct mcast_group *group;645	struct mcast_req *req = NULL;646	struct ib_sa_mcmember_data *sa_data;647	u8 req_join_state;648	int rc = 1; /* release_count - this is for the scheduled work */649	u16 status;650	u8 method;651 652	group = container_of(work, typeof(*group), work);653 654	mutex_lock(&group->lock);655 656	/* First, let's see if a response from SM is waiting regarding this group.657	 * If so, we need to update the group's REC. If this is a bad response, we658	 * may need to send a bad response to a VF waiting for it. If VF is waiting659	 * and this is a good response, the VF will be answered later in this func. */660	if (group->state == MCAST_RESP_READY) {661		/* cancels mlx4_ib_mcg_timeout_handler */662		cancel_delayed_work(&group->timeout_work);663		status = be16_to_cpu(group->response_sa_mad.mad_hdr.status);664		method = group->response_sa_mad.mad_hdr.method;665		if (group->last_req_tid != group->response_sa_mad.mad_hdr.tid) {666			mcg_warn_group(group, "Got MAD response to existing MGID but wrong TID, dropping. Resp TID=%llx, group TID=%llx\n",667				be64_to_cpu(group->response_sa_mad.mad_hdr.tid),668				be64_to_cpu(group->last_req_tid));669			group->state = group->prev_state;670			goto process_requests;671		}672		if (status) {673			if (!list_empty(&group->pending_list))674				req = list_first_entry(&group->pending_list,675						struct mcast_req, group_list);676			if (method == IB_MGMT_METHOD_GET_RESP) {677					if (req) {678						send_reply_to_slave(req->func, group, &req->sa_mad, status);679						--group->func[req->func].num_pend_reqs;680						list_del(&req->group_list);681						list_del(&req->func_list);682						kfree(req);683						++rc;684					} else685						mcg_warn_group(group, "no request for failed join\n");686			} else if (method == IB_SA_METHOD_DELETE_RESP && group->demux->flushing)687				++rc;688		} else {689			u8 resp_join_state;690			u8 cur_join_state;691 692			resp_join_state = ((struct ib_sa_mcmember_data *)693						group->response_sa_mad.data)->scope_join_state & 0xf;694			cur_join_state = group->rec.scope_join_state & 0xf;695 696			if (method == IB_MGMT_METHOD_GET_RESP) {697				/* successfull join */698				if (!cur_join_state && resp_join_state)699					--rc;700			} else if (!resp_join_state)701					++rc;702			memcpy(&group->rec, group->response_sa_mad.data, sizeof group->rec);703		}704		group->state = MCAST_IDLE;705	}706 707process_requests:708	/* We should now go over pending join/leave requests, as long as we are idle. */709	while (!list_empty(&group->pending_list) && group->state == MCAST_IDLE) {710		req = list_first_entry(&group->pending_list, struct mcast_req,711				       group_list);712		sa_data = (struct ib_sa_mcmember_data *)req->sa_mad.data;713		req_join_state = sa_data->scope_join_state & 0xf;714 715		/* For a leave request, we will immediately answer the VF, and716		 * update our internal counters. The actual leave will be sent717		 * to SM later, if at all needed. We dequeue the request now. */718		if (req->sa_mad.mad_hdr.method == IB_SA_METHOD_DELETE)719			rc += handle_leave_req(group, req_join_state, req);720		else721			rc += handle_join_req(group, req_join_state, req);722	}723 724	/* Handle leaves */725	if (group->state == MCAST_IDLE) {726		req_join_state = get_leave_state(group);727		if (req_join_state) {728			group->rec.scope_join_state &= ~req_join_state;729			group->prev_state = group->state;730			if (send_leave_to_wire(group, req_join_state)) {731				group->state = group->prev_state;732				++rc;733			} else734				group->state = MCAST_LEAVE_SENT;735		}736	}737 738	if (!list_empty(&group->pending_list) && group->state == MCAST_IDLE)739		goto process_requests;740	mutex_unlock(&group->lock);741 742	while (rc--)743		release_group(group, 0);744}745 746static struct mcast_group *search_relocate_mgid0_group(struct mlx4_ib_demux_ctx *ctx,747						       __be64 tid,748						       union ib_gid *new_mgid)749{750	struct mcast_group *group = NULL, *cur_group, *n;751	struct mcast_req *req;752 753	mutex_lock(&ctx->mcg_table_lock);754	list_for_each_entry_safe(group, n, &ctx->mcg_mgid0_list, mgid0_list) {755		mutex_lock(&group->lock);756		if (group->last_req_tid == tid) {757			if (memcmp(new_mgid, &mgid0, sizeof mgid0)) {758				group->rec.mgid = *new_mgid;759				sprintf(group->name, "%016llx%016llx",760						be64_to_cpu(group->rec.mgid.global.subnet_prefix),761						be64_to_cpu(group->rec.mgid.global.interface_id));762				list_del_init(&group->mgid0_list);763				cur_group = mcast_insert(ctx, group);764				if (cur_group) {765					/* A race between our code and SM. Silently cleaning the new one */766					req = list_first_entry(&group->pending_list,767							       struct mcast_req, group_list);768					--group->func[req->func].num_pend_reqs;769					list_del(&req->group_list);770					list_del(&req->func_list);771					kfree(req);772					mutex_unlock(&group->lock);773					mutex_unlock(&ctx->mcg_table_lock);774					release_group(group, 0);775					return NULL;776				}777 778				atomic_inc(&group->refcount);779				add_sysfs_port_mcg_attr(ctx->dev, ctx->port, &group->dentry.attr);780				mutex_unlock(&group->lock);781				mutex_unlock(&ctx->mcg_table_lock);782				return group;783			} else {784				struct mcast_req *tmp1, *tmp2;785 786				list_del(&group->mgid0_list);787				if (!list_empty(&group->pending_list) && group->state != MCAST_IDLE)788					cancel_delayed_work_sync(&group->timeout_work);789 790				list_for_each_entry_safe(tmp1, tmp2, &group->pending_list, group_list) {791					list_del(&tmp1->group_list);792					kfree(tmp1);793				}794				mutex_unlock(&group->lock);795				mutex_unlock(&ctx->mcg_table_lock);796				kfree(group);797				return NULL;798			}799		}800		mutex_unlock(&group->lock);801	}802	mutex_unlock(&ctx->mcg_table_lock);803 804	return NULL;805}806 807static ssize_t sysfs_show_group(struct device *dev,808		struct device_attribute *attr, char *buf);809 810static struct mcast_group *acquire_group(struct mlx4_ib_demux_ctx *ctx,811					 union ib_gid *mgid, int create)812{813	struct mcast_group *group, *cur_group;814	int is_mgid0;815	int i;816 817	is_mgid0 = !memcmp(&mgid0, mgid, sizeof mgid0);818	if (!is_mgid0) {819		group = mcast_find(ctx, mgid);820		if (group)821			goto found;822	}823 824	if (!create)825		return ERR_PTR(-ENOENT);826 827	group = kzalloc(sizeof(*group), GFP_KERNEL);828	if (!group)829		return ERR_PTR(-ENOMEM);830 831	group->demux = ctx;832	group->rec.mgid = *mgid;833	INIT_LIST_HEAD(&group->pending_list);834	INIT_LIST_HEAD(&group->mgid0_list);835	for (i = 0; i < MAX_VFS; ++i)836		INIT_LIST_HEAD(&group->func[i].pending);837	INIT_WORK(&group->work, mlx4_ib_mcg_work_handler);838	INIT_DELAYED_WORK(&group->timeout_work, mlx4_ib_mcg_timeout_handler);839	mutex_init(&group->lock);840	sprintf(group->name, "%016llx%016llx",841			be64_to_cpu(group->rec.mgid.global.subnet_prefix),842			be64_to_cpu(group->rec.mgid.global.interface_id));843	sysfs_attr_init(&group->dentry.attr);844	group->dentry.show = sysfs_show_group;845	group->dentry.store = NULL;846	group->dentry.attr.name = group->name;847	group->dentry.attr.mode = 0400;848	group->state = MCAST_IDLE;849 850	if (is_mgid0) {851		list_add(&group->mgid0_list, &ctx->mcg_mgid0_list);852		goto found;853	}854 855	cur_group = mcast_insert(ctx, group);856	if (cur_group) {857		mcg_warn("group just showed up %s - confused\n", cur_group->name);858		kfree(group);859		return ERR_PTR(-EINVAL);860	}861 862	add_sysfs_port_mcg_attr(ctx->dev, ctx->port, &group->dentry.attr);863 864found:865	atomic_inc(&group->refcount);866	return group;867}868 869static void queue_req(struct mcast_req *req)870{871	struct mcast_group *group = req->group;872 873	atomic_inc(&group->refcount); /* for the request */874	atomic_inc(&group->refcount); /* for scheduling the work */875	list_add_tail(&req->group_list, &group->pending_list);876	list_add_tail(&req->func_list, &group->func[req->func].pending);877	/* calls mlx4_ib_mcg_work_handler */878	if (!queue_work(group->demux->mcg_wq, &group->work))879		safe_atomic_dec(&group->refcount);880}881 882int mlx4_ib_mcg_demux_handler(struct ib_device *ibdev, int port, int slave,883			      struct ib_sa_mad *mad)884{885	struct mlx4_ib_dev *dev = to_mdev(ibdev);886	struct ib_sa_mcmember_data *rec = (struct ib_sa_mcmember_data *)mad->data;887	struct mlx4_ib_demux_ctx *ctx = &dev->sriov.demux[port - 1];888	struct mcast_group *group;889 890	switch (mad->mad_hdr.method) {891	case IB_MGMT_METHOD_GET_RESP:892	case IB_SA_METHOD_DELETE_RESP:893		mutex_lock(&ctx->mcg_table_lock);894		group = acquire_group(ctx, &rec->mgid, 0);895		mutex_unlock(&ctx->mcg_table_lock);896		if (IS_ERR(group)) {897			if (mad->mad_hdr.method == IB_MGMT_METHOD_GET_RESP) {898				__be64 tid = mad->mad_hdr.tid;899				*(u8 *)(&tid) = (u8)slave; /* in group we kept the modified TID */900				group = search_relocate_mgid0_group(ctx, tid, &rec->mgid);901			} else902				group = NULL;903		}904 905		if (!group)906			return 1;907 908		mutex_lock(&group->lock);909		group->response_sa_mad = *mad;910		group->prev_state = group->state;911		group->state = MCAST_RESP_READY;912		/* calls mlx4_ib_mcg_work_handler */913		atomic_inc(&group->refcount);914		if (!queue_work(ctx->mcg_wq, &group->work))915			safe_atomic_dec(&group->refcount);916		mutex_unlock(&group->lock);917		release_group(group, 0);918		return 1; /* consumed */919	case IB_MGMT_METHOD_SET:920	case IB_SA_METHOD_GET_TABLE:921	case IB_SA_METHOD_GET_TABLE_RESP:922	case IB_SA_METHOD_DELETE:923		return 0; /* not consumed, pass-through to guest over tunnel */924	default:925		mcg_warn("In demux, port %d: unexpected MCMember method: 0x%x, dropping\n",926			port, mad->mad_hdr.method);927		return 1; /* consumed */928	}929}930 931int mlx4_ib_mcg_multiplex_handler(struct ib_device *ibdev, int port,932				  int slave, struct ib_sa_mad *sa_mad)933{934	struct mlx4_ib_dev *dev = to_mdev(ibdev);935	struct ib_sa_mcmember_data *rec = (struct ib_sa_mcmember_data *)sa_mad->data;936	struct mlx4_ib_demux_ctx *ctx = &dev->sriov.demux[port - 1];937	struct mcast_group *group;938	struct mcast_req *req;939	int may_create = 0;940 941	if (ctx->flushing)942		return -EAGAIN;943 944	switch (sa_mad->mad_hdr.method) {945	case IB_MGMT_METHOD_SET:946		may_create = 1;947		fallthrough;948	case IB_SA_METHOD_DELETE:949		req = kzalloc(sizeof *req, GFP_KERNEL);950		if (!req)951			return -ENOMEM;952 953		req->func = slave;954		req->sa_mad = *sa_mad;955 956		mutex_lock(&ctx->mcg_table_lock);957		group = acquire_group(ctx, &rec->mgid, may_create);958		mutex_unlock(&ctx->mcg_table_lock);959		if (IS_ERR(group)) {960			kfree(req);961			return PTR_ERR(group);962		}963		mutex_lock(&group->lock);964		if (group->func[slave].num_pend_reqs > MAX_PEND_REQS_PER_FUNC) {965			mutex_unlock(&group->lock);966			mcg_debug_group(group, "Port %d, Func %d has too many pending requests (%d), dropping\n",967					port, slave, MAX_PEND_REQS_PER_FUNC);968			release_group(group, 0);969			kfree(req);970			return -ENOMEM;971		}972		++group->func[slave].num_pend_reqs;973		req->group = group;974		queue_req(req);975		mutex_unlock(&group->lock);976		release_group(group, 0);977		return 1; /* consumed */978	case IB_SA_METHOD_GET_TABLE:979	case IB_MGMT_METHOD_GET_RESP:980	case IB_SA_METHOD_GET_TABLE_RESP:981	case IB_SA_METHOD_DELETE_RESP:982		return 0; /* not consumed, pass-through */983	default:984		mcg_warn("In multiplex, port %d, func %d: unexpected MCMember method: 0x%x, dropping\n",985			port, slave, sa_mad->mad_hdr.method);986		return 1; /* consumed */987	}988}989 990static ssize_t sysfs_show_group(struct device *dev,991				struct device_attribute *attr, char *buf)992{993	struct mcast_group *group =994		container_of(attr, struct mcast_group, dentry);995	struct mcast_req *req = NULL;996	char state_str[40];997	char pending_str[40];998	int len;999	int i;1000	u32 hoplimit;1001 1002	if (group->state == MCAST_IDLE)1003		scnprintf(state_str, sizeof(state_str), "%s",1004			  get_state_string(group->state));1005	else1006		scnprintf(state_str, sizeof(state_str), "%s(TID=0x%llx)",1007			  get_state_string(group->state),1008			  be64_to_cpu(group->last_req_tid));1009 1010	if (list_empty(&group->pending_list)) {1011		scnprintf(pending_str, sizeof(pending_str), "No");1012	} else {1013		req = list_first_entry(&group->pending_list, struct mcast_req,1014				       group_list);1015		scnprintf(pending_str, sizeof(pending_str), "Yes(TID=0x%llx)",1016			  be64_to_cpu(req->sa_mad.mad_hdr.tid));1017	}1018 1019	len = sysfs_emit(buf, "%1d [%02d,%02d,%02d] %4d %4s %5s     ",1020			 group->rec.scope_join_state & 0xf,1021			 group->members[2],1022			 group->members[1],1023			 group->members[0],1024			 atomic_read(&group->refcount),1025			 pending_str,1026			 state_str);1027 1028	for (i = 0; i < MAX_VFS; i++) {1029		if (group->func[i].state == MCAST_MEMBER)1030			len += sysfs_emit_at(buf, len, "%d[%1x] ", i,1031					     group->func[i].join_state);1032	}1033 1034	hoplimit = be32_to_cpu(group->rec.sl_flowlabel_hoplimit);1035	len += sysfs_emit_at(buf, len,1036			     "\t\t(%4hx %4x %2x %2x %2x %2x %2x %4x %4x %2x %2x)\n",1037			     be16_to_cpu(group->rec.pkey),1038			     be32_to_cpu(group->rec.qkey),1039			     (group->rec.mtusel_mtu & 0xc0) >> 6,1040			     (group->rec.mtusel_mtu & 0x3f),1041			     group->rec.tclass,1042			     (group->rec.ratesel_rate & 0xc0) >> 6,1043			     (group->rec.ratesel_rate & 0x3f),1044			     (hoplimit & 0xf0000000) >> 28,1045			     (hoplimit & 0x0fffff00) >> 8,1046			     (hoplimit & 0x000000ff),1047			     group->rec.proxy_join);1048 1049	return len;1050}1051 1052int mlx4_ib_mcg_port_init(struct mlx4_ib_demux_ctx *ctx)1053{1054	char name[20];1055 1056	atomic_set(&ctx->tid, 0);1057	sprintf(name, "mlx4_ib_mcg%d", ctx->port);1058	ctx->mcg_wq = alloc_ordered_workqueue(name, WQ_MEM_RECLAIM);1059	if (!ctx->mcg_wq)1060		return -ENOMEM;1061 1062	mutex_init(&ctx->mcg_table_lock);1063	ctx->mcg_table = RB_ROOT;1064	INIT_LIST_HEAD(&ctx->mcg_mgid0_list);1065	ctx->flushing = 0;1066 1067	return 0;1068}1069 1070static void force_clean_group(struct mcast_group *group)1071{1072	struct mcast_req *req, *tmp1073		;1074	list_for_each_entry_safe(req, tmp, &group->pending_list, group_list) {1075		list_del(&req->group_list);1076		kfree(req);1077	}1078	del_sysfs_port_mcg_attr(group->demux->dev, group->demux->port, &group->dentry.attr);1079	rb_erase(&group->node, &group->demux->mcg_table);1080	kfree(group);1081}1082 1083static void _mlx4_ib_mcg_port_cleanup(struct mlx4_ib_demux_ctx *ctx, int destroy_wq)1084{1085	int i;1086	struct rb_node *p;1087	struct mcast_group *group;1088	unsigned long end;1089	int count;1090 1091	for (i = 0; i < MAX_VFS; ++i)1092		clean_vf_mcast(ctx, i);1093 1094	end = jiffies + msecs_to_jiffies(MAD_TIMEOUT_MS + 3000);1095	do {1096		count = 0;1097		mutex_lock(&ctx->mcg_table_lock);1098		for (p = rb_first(&ctx->mcg_table); p; p = rb_next(p))1099			++count;1100		mutex_unlock(&ctx->mcg_table_lock);1101		if (!count)1102			break;1103 1104		usleep_range(1000, 2000);1105	} while (time_after(end, jiffies));1106 1107	flush_workqueue(ctx->mcg_wq);1108	if (destroy_wq)1109		destroy_workqueue(ctx->mcg_wq);1110 1111	mutex_lock(&ctx->mcg_table_lock);1112	while ((p = rb_first(&ctx->mcg_table)) != NULL) {1113		group = rb_entry(p, struct mcast_group, node);1114		if (atomic_read(&group->refcount))1115			mcg_debug_group(group, "group refcount %d!!! (pointer %p)\n",1116					atomic_read(&group->refcount), group);1117 1118		force_clean_group(group);1119	}1120	mutex_unlock(&ctx->mcg_table_lock);1121}1122 1123struct clean_work {1124	struct work_struct work;1125	struct mlx4_ib_demux_ctx *ctx;1126	int destroy_wq;1127};1128 1129static void mcg_clean_task(struct work_struct *work)1130{1131	struct clean_work *cw = container_of(work, struct clean_work, work);1132 1133	_mlx4_ib_mcg_port_cleanup(cw->ctx, cw->destroy_wq);1134	cw->ctx->flushing = 0;1135	kfree(cw);1136}1137 1138void mlx4_ib_mcg_port_cleanup(struct mlx4_ib_demux_ctx *ctx, int destroy_wq)1139{1140	struct clean_work *work;1141 1142	if (ctx->flushing)1143		return;1144 1145	ctx->flushing = 1;1146 1147	if (destroy_wq) {1148		_mlx4_ib_mcg_port_cleanup(ctx, destroy_wq);1149		ctx->flushing = 0;1150		return;1151	}1152 1153	work = kmalloc(sizeof *work, GFP_KERNEL);1154	if (!work) {1155		ctx->flushing = 0;1156		return;1157	}1158 1159	work->ctx = ctx;1160	work->destroy_wq = destroy_wq;1161	INIT_WORK(&work->work, mcg_clean_task);1162	queue_work(clean_wq, &work->work);1163}1164 1165static void build_leave_mad(struct mcast_req *req)1166{1167	struct ib_sa_mad *mad = &req->sa_mad;1168 1169	mad->mad_hdr.method = IB_SA_METHOD_DELETE;1170}1171 1172 1173static void clear_pending_reqs(struct mcast_group *group, int vf)1174{1175	struct mcast_req *req, *tmp, *group_first = NULL;1176	int clear;1177	int pend = 0;1178 1179	if (!list_empty(&group->pending_list))1180		group_first = list_first_entry(&group->pending_list, struct mcast_req, group_list);1181 1182	list_for_each_entry_safe(req, tmp, &group->func[vf].pending, func_list) {1183		clear = 1;1184		if (group_first == req &&1185		    (group->state == MCAST_JOIN_SENT ||1186		     group->state == MCAST_LEAVE_SENT)) {1187			clear = cancel_delayed_work(&group->timeout_work);1188			pend = !clear;1189			group->state = MCAST_IDLE;1190		}1191		if (clear) {1192			--group->func[vf].num_pend_reqs;1193			list_del(&req->group_list);1194			list_del(&req->func_list);1195			kfree(req);1196			atomic_dec(&group->refcount);1197		}1198	}1199 1200	if (!pend && (!list_empty(&group->func[vf].pending) || group->func[vf].num_pend_reqs)) {1201		mcg_warn_group(group, "DRIVER BUG: list_empty %d, num_pend_reqs %d\n",1202			       list_empty(&group->func[vf].pending), group->func[vf].num_pend_reqs);1203	}1204}1205 1206static int push_deleteing_req(struct mcast_group *group, int slave)1207{1208	struct mcast_req *req;1209	struct mcast_req *pend_req;1210 1211	if (!group->func[slave].join_state)1212		return 0;1213 1214	req = kzalloc(sizeof *req, GFP_KERNEL);1215	if (!req)1216		return -ENOMEM;1217 1218	if (!list_empty(&group->func[slave].pending)) {1219		pend_req = list_entry(group->func[slave].pending.prev, struct mcast_req, group_list);1220		if (pend_req->clean) {1221			kfree(req);1222			return 0;1223		}1224	}1225 1226	req->clean = 1;1227	req->func = slave;1228	req->group = group;1229	++group->func[slave].num_pend_reqs;1230	build_leave_mad(req);1231	queue_req(req);1232	return 0;1233}1234 1235void clean_vf_mcast(struct mlx4_ib_demux_ctx *ctx, int slave)1236{1237	struct mcast_group *group;1238	struct rb_node *p;1239 1240	mutex_lock(&ctx->mcg_table_lock);1241	for (p = rb_first(&ctx->mcg_table); p; p = rb_next(p)) {1242		group = rb_entry(p, struct mcast_group, node);1243		mutex_lock(&group->lock);1244		if (atomic_read(&group->refcount)) {1245			/* clear pending requests of this VF */1246			clear_pending_reqs(group, slave);1247			push_deleteing_req(group, slave);1248		}1249		mutex_unlock(&group->lock);1250	}1251	mutex_unlock(&ctx->mcg_table_lock);1252}1253 1254 1255int mlx4_ib_mcg_init(void)1256{1257	clean_wq = alloc_ordered_workqueue("mlx4_ib_mcg", WQ_MEM_RECLAIM);1258	if (!clean_wq)1259		return -ENOMEM;1260 1261	return 0;1262}1263 1264void mlx4_ib_mcg_destroy(void)1265{1266	destroy_workqueue(clean_wq);1267}1268