Linux Audio

Check our new training course

Loading...
v3.1
   1/*
   2 * Copyright (c) 2004-2007 Intel Corporation.  All rights reserved.
   3 * Copyright (c) 2004 Topspin Corporation.  All rights reserved.
   4 * Copyright (c) 2004, 2005 Voltaire Corporation.  All rights reserved.
   5 * Copyright (c) 2005 Sun Microsystems, Inc. All rights reserved.
   6 *
   7 * This software is available to you under a choice of one of two
   8 * licenses.  You may choose to be licensed under the terms of the GNU
   9 * General Public License (GPL) Version 2, available from the file
  10 * COPYING in the main directory of this source tree, or the
  11 * OpenIB.org BSD license below:
  12 *
  13 *     Redistribution and use in source and binary forms, with or
  14 *     without modification, are permitted provided that the following
  15 *     conditions are met:
  16 *
  17 *      - Redistributions of source code must retain the above
  18 *        copyright notice, this list of conditions and the following
  19 *        disclaimer.
  20 *
  21 *      - Redistributions in binary form must reproduce the above
  22 *        copyright notice, this list of conditions and the following
  23 *        disclaimer in the documentation and/or other materials
  24 *        provided with the distribution.
  25 *
  26 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  27 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  28 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  29 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  30 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  31 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  32 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  33 * SOFTWARE.
  34 */
  35
  36#include <linux/completion.h>
  37#include <linux/dma-mapping.h>
  38#include <linux/device.h>
 
  39#include <linux/err.h>
  40#include <linux/idr.h>
  41#include <linux/interrupt.h>
  42#include <linux/random.h>
  43#include <linux/rbtree.h>
  44#include <linux/spinlock.h>
  45#include <linux/slab.h>
  46#include <linux/sysfs.h>
  47#include <linux/workqueue.h>
  48#include <linux/kdev_t.h>
 
  49
  50#include <rdma/ib_cache.h>
  51#include <rdma/ib_cm.h>
  52#include "cm_msgs.h"
  53
  54MODULE_AUTHOR("Sean Hefty");
  55MODULE_DESCRIPTION("InfiniBand CM");
  56MODULE_LICENSE("Dual BSD/GPL");
  57
  58static void cm_add_one(struct ib_device *device);
  59static void cm_remove_one(struct ib_device *device);
  60
  61static struct ib_client cm_client = {
  62	.name   = "cm",
  63	.add    = cm_add_one,
  64	.remove = cm_remove_one
  65};
  66
  67static struct ib_cm {
  68	spinlock_t lock;
  69	struct list_head device_list;
  70	rwlock_t device_lock;
  71	struct rb_root listen_service_table;
  72	u64 listen_service_id;
  73	/* struct rb_root peer_service_table; todo: fix peer to peer */
  74	struct rb_root remote_qp_table;
  75	struct rb_root remote_id_table;
  76	struct rb_root remote_sidr_table;
  77	struct idr local_id_table;
  78	__be32 random_id_operand;
  79	struct list_head timewait_list;
  80	struct workqueue_struct *wq;
  81} cm;
  82
  83/* Counter indexes ordered by attribute ID */
  84enum {
  85	CM_REQ_COUNTER,
  86	CM_MRA_COUNTER,
  87	CM_REJ_COUNTER,
  88	CM_REP_COUNTER,
  89	CM_RTU_COUNTER,
  90	CM_DREQ_COUNTER,
  91	CM_DREP_COUNTER,
  92	CM_SIDR_REQ_COUNTER,
  93	CM_SIDR_REP_COUNTER,
  94	CM_LAP_COUNTER,
  95	CM_APR_COUNTER,
  96	CM_ATTR_COUNT,
  97	CM_ATTR_ID_OFFSET = 0x0010,
  98};
  99
 100enum {
 101	CM_XMIT,
 102	CM_XMIT_RETRIES,
 103	CM_RECV,
 104	CM_RECV_DUPLICATES,
 105	CM_COUNTER_GROUPS
 106};
 107
 108static char const counter_group_names[CM_COUNTER_GROUPS]
 109				     [sizeof("cm_rx_duplicates")] = {
 110	"cm_tx_msgs", "cm_tx_retries",
 111	"cm_rx_msgs", "cm_rx_duplicates"
 112};
 113
 114struct cm_counter_group {
 115	struct kobject obj;
 116	atomic_long_t counter[CM_ATTR_COUNT];
 117};
 118
 119struct cm_counter_attribute {
 120	struct attribute attr;
 121	int index;
 122};
 123
 124#define CM_COUNTER_ATTR(_name, _index) \
 125struct cm_counter_attribute cm_##_name##_counter_attr = { \
 126	.attr = { .name = __stringify(_name), .mode = 0444 }, \
 127	.index = _index \
 128}
 129
 130static CM_COUNTER_ATTR(req, CM_REQ_COUNTER);
 131static CM_COUNTER_ATTR(mra, CM_MRA_COUNTER);
 132static CM_COUNTER_ATTR(rej, CM_REJ_COUNTER);
 133static CM_COUNTER_ATTR(rep, CM_REP_COUNTER);
 134static CM_COUNTER_ATTR(rtu, CM_RTU_COUNTER);
 135static CM_COUNTER_ATTR(dreq, CM_DREQ_COUNTER);
 136static CM_COUNTER_ATTR(drep, CM_DREP_COUNTER);
 137static CM_COUNTER_ATTR(sidr_req, CM_SIDR_REQ_COUNTER);
 138static CM_COUNTER_ATTR(sidr_rep, CM_SIDR_REP_COUNTER);
 139static CM_COUNTER_ATTR(lap, CM_LAP_COUNTER);
 140static CM_COUNTER_ATTR(apr, CM_APR_COUNTER);
 141
 142static struct attribute *cm_counter_default_attrs[] = {
 143	&cm_req_counter_attr.attr,
 144	&cm_mra_counter_attr.attr,
 145	&cm_rej_counter_attr.attr,
 146	&cm_rep_counter_attr.attr,
 147	&cm_rtu_counter_attr.attr,
 148	&cm_dreq_counter_attr.attr,
 149	&cm_drep_counter_attr.attr,
 150	&cm_sidr_req_counter_attr.attr,
 151	&cm_sidr_rep_counter_attr.attr,
 152	&cm_lap_counter_attr.attr,
 153	&cm_apr_counter_attr.attr,
 154	NULL
 155};
 156
 157struct cm_port {
 158	struct cm_device *cm_dev;
 159	struct ib_mad_agent *mad_agent;
 160	struct kobject port_obj;
 161	u8 port_num;
 162	struct cm_counter_group counter_group[CM_COUNTER_GROUPS];
 163};
 164
 165struct cm_device {
 166	struct list_head list;
 167	struct ib_device *ib_device;
 168	struct device *device;
 169	u8 ack_delay;
 
 170	struct cm_port *port[0];
 171};
 172
 173struct cm_av {
 174	struct cm_port *port;
 175	union ib_gid dgid;
 176	struct ib_ah_attr ah_attr;
 177	u16 pkey_index;
 178	u8 timeout;
 179};
 180
 181struct cm_work {
 182	struct delayed_work work;
 183	struct list_head list;
 184	struct cm_port *port;
 185	struct ib_mad_recv_wc *mad_recv_wc;	/* Received MADs */
 186	__be32 local_id;			/* Established / timewait */
 187	__be32 remote_id;
 188	struct ib_cm_event cm_event;
 189	struct ib_sa_path_rec path[0];
 190};
 191
 192struct cm_timewait_info {
 193	struct cm_work work;			/* Must be first. */
 194	struct list_head list;
 195	struct rb_node remote_qp_node;
 196	struct rb_node remote_id_node;
 197	__be64 remote_ca_guid;
 198	__be32 remote_qpn;
 199	u8 inserted_remote_qp;
 200	u8 inserted_remote_id;
 201};
 202
 203struct cm_id_private {
 204	struct ib_cm_id	id;
 205
 206	struct rb_node service_node;
 207	struct rb_node sidr_id_node;
 208	spinlock_t lock;	/* Do not acquire inside cm.lock */
 209	struct completion comp;
 210	atomic_t refcount;
 
 
 
 211
 212	struct ib_mad_send_buf *msg;
 213	struct cm_timewait_info *timewait_info;
 214	/* todo: use alternate port on send failure */
 215	struct cm_av av;
 216	struct cm_av alt_av;
 217	struct ib_cm_compare_data *compare_data;
 218
 219	void *private_data;
 220	__be64 tid;
 221	__be32 local_qpn;
 222	__be32 remote_qpn;
 223	enum ib_qp_type qp_type;
 224	__be32 sq_psn;
 225	__be32 rq_psn;
 226	int timeout_ms;
 227	enum ib_mtu path_mtu;
 228	__be16 pkey;
 229	u8 private_data_len;
 230	u8 max_cm_retries;
 231	u8 peer_to_peer;
 232	u8 responder_resources;
 233	u8 initiator_depth;
 234	u8 retry_count;
 235	u8 rnr_retry_count;
 236	u8 service_timeout;
 237	u8 target_ack_delay;
 238
 239	struct list_head work_list;
 240	atomic_t work_count;
 241};
 242
 243static void cm_work_handler(struct work_struct *work);
 244
 245static inline void cm_deref_id(struct cm_id_private *cm_id_priv)
 246{
 247	if (atomic_dec_and_test(&cm_id_priv->refcount))
 248		complete(&cm_id_priv->comp);
 249}
 250
 251static int cm_alloc_msg(struct cm_id_private *cm_id_priv,
 252			struct ib_mad_send_buf **msg)
 253{
 254	struct ib_mad_agent *mad_agent;
 255	struct ib_mad_send_buf *m;
 256	struct ib_ah *ah;
 257
 258	mad_agent = cm_id_priv->av.port->mad_agent;
 259	ah = ib_create_ah(mad_agent->qp->pd, &cm_id_priv->av.ah_attr);
 260	if (IS_ERR(ah))
 261		return PTR_ERR(ah);
 262
 263	m = ib_create_send_mad(mad_agent, cm_id_priv->id.remote_cm_qpn,
 264			       cm_id_priv->av.pkey_index,
 265			       0, IB_MGMT_MAD_HDR, IB_MGMT_MAD_DATA,
 266			       GFP_ATOMIC);
 
 267	if (IS_ERR(m)) {
 268		ib_destroy_ah(ah);
 269		return PTR_ERR(m);
 270	}
 271
 272	/* Timeout set by caller if response is expected. */
 273	m->ah = ah;
 274	m->retries = cm_id_priv->max_cm_retries;
 275
 276	atomic_inc(&cm_id_priv->refcount);
 277	m->context[0] = cm_id_priv;
 278	*msg = m;
 279	return 0;
 280}
 281
 282static int cm_alloc_response_msg(struct cm_port *port,
 283				 struct ib_mad_recv_wc *mad_recv_wc,
 284				 struct ib_mad_send_buf **msg)
 285{
 286	struct ib_mad_send_buf *m;
 287	struct ib_ah *ah;
 288
 289	ah = ib_create_ah_from_wc(port->mad_agent->qp->pd, mad_recv_wc->wc,
 290				  mad_recv_wc->recv_buf.grh, port->port_num);
 291	if (IS_ERR(ah))
 292		return PTR_ERR(ah);
 293
 294	m = ib_create_send_mad(port->mad_agent, 1, mad_recv_wc->wc->pkey_index,
 295			       0, IB_MGMT_MAD_HDR, IB_MGMT_MAD_DATA,
 296			       GFP_ATOMIC);
 
 297	if (IS_ERR(m)) {
 298		ib_destroy_ah(ah);
 299		return PTR_ERR(m);
 300	}
 301	m->ah = ah;
 302	*msg = m;
 303	return 0;
 304}
 305
 306static void cm_free_msg(struct ib_mad_send_buf *msg)
 307{
 308	ib_destroy_ah(msg->ah);
 309	if (msg->context[0])
 310		cm_deref_id(msg->context[0]);
 311	ib_free_send_mad(msg);
 312}
 313
 314static void * cm_copy_private_data(const void *private_data,
 315				   u8 private_data_len)
 316{
 317	void *data;
 318
 319	if (!private_data || !private_data_len)
 320		return NULL;
 321
 322	data = kmemdup(private_data, private_data_len, GFP_KERNEL);
 323	if (!data)
 324		return ERR_PTR(-ENOMEM);
 325
 326	return data;
 327}
 328
 329static void cm_set_private_data(struct cm_id_private *cm_id_priv,
 330				 void *private_data, u8 private_data_len)
 331{
 332	if (cm_id_priv->private_data && cm_id_priv->private_data_len)
 333		kfree(cm_id_priv->private_data);
 334
 335	cm_id_priv->private_data = private_data;
 336	cm_id_priv->private_data_len = private_data_len;
 337}
 338
 339static void cm_init_av_for_response(struct cm_port *port, struct ib_wc *wc,
 340				    struct ib_grh *grh, struct cm_av *av)
 341{
 342	av->port = port;
 343	av->pkey_index = wc->pkey_index;
 344	ib_init_ah_from_wc(port->cm_dev->ib_device, port->port_num, wc,
 345			   grh, &av->ah_attr);
 346}
 347
 348static int cm_init_av_by_path(struct ib_sa_path_rec *path, struct cm_av *av)
 349{
 350	struct cm_device *cm_dev;
 351	struct cm_port *port = NULL;
 352	unsigned long flags;
 353	int ret;
 354	u8 p;
 
 355
 356	read_lock_irqsave(&cm.device_lock, flags);
 357	list_for_each_entry(cm_dev, &cm.device_list, list) {
 358		if (!ib_find_cached_gid(cm_dev->ib_device, &path->sgid,
 359					&p, NULL)) {
 360			port = cm_dev->port[p-1];
 361			break;
 362		}
 363	}
 364	read_unlock_irqrestore(&cm.device_lock, flags);
 365
 
 
 
 366	if (!port)
 367		return -EINVAL;
 368
 369	ret = ib_find_cached_pkey(cm_dev->ib_device, port->port_num,
 370				  be16_to_cpu(path->pkey), &av->pkey_index);
 371	if (ret)
 372		return ret;
 373
 374	av->port = port;
 375	ib_init_ah_from_path(cm_dev->ib_device, port->port_num, path,
 376			     &av->ah_attr);
 377	av->timeout = path->packet_life_time + 1;
 
 378	return 0;
 379}
 380
 381static int cm_alloc_id(struct cm_id_private *cm_id_priv)
 382{
 383	unsigned long flags;
 384	int ret, id;
 385	static int next_id;
 386
 387	do {
 388		spin_lock_irqsave(&cm.lock, flags);
 389		ret = idr_get_new_above(&cm.local_id_table, cm_id_priv,
 390					next_id, &id);
 391		if (!ret)
 392			next_id = ((unsigned) id + 1) & MAX_ID_MASK;
 393		spin_unlock_irqrestore(&cm.lock, flags);
 394	} while( (ret == -EAGAIN) && idr_pre_get(&cm.local_id_table, GFP_KERNEL) );
 395
 396	cm_id_priv->id.local_id = (__force __be32)id ^ cm.random_id_operand;
 397	return ret;
 398}
 399
 400static void cm_free_id(__be32 local_id)
 401{
 402	spin_lock_irq(&cm.lock);
 403	idr_remove(&cm.local_id_table,
 404		   (__force int) (local_id ^ cm.random_id_operand));
 405	spin_unlock_irq(&cm.lock);
 406}
 407
 408static struct cm_id_private * cm_get_id(__be32 local_id, __be32 remote_id)
 409{
 410	struct cm_id_private *cm_id_priv;
 411
 412	cm_id_priv = idr_find(&cm.local_id_table,
 413			      (__force int) (local_id ^ cm.random_id_operand));
 414	if (cm_id_priv) {
 415		if (cm_id_priv->id.remote_id == remote_id)
 416			atomic_inc(&cm_id_priv->refcount);
 417		else
 418			cm_id_priv = NULL;
 419	}
 420
 421	return cm_id_priv;
 422}
 423
 424static struct cm_id_private * cm_acquire_id(__be32 local_id, __be32 remote_id)
 425{
 426	struct cm_id_private *cm_id_priv;
 427
 428	spin_lock_irq(&cm.lock);
 429	cm_id_priv = cm_get_id(local_id, remote_id);
 430	spin_unlock_irq(&cm.lock);
 431
 432	return cm_id_priv;
 433}
 434
 435static void cm_mask_copy(u8 *dst, u8 *src, u8 *mask)
 436{
 437	int i;
 438
 439	for (i = 0; i < IB_CM_COMPARE_SIZE / sizeof(unsigned long); i++)
 440		((unsigned long *) dst)[i] = ((unsigned long *) src)[i] &
 441					     ((unsigned long *) mask)[i];
 442}
 443
 444static int cm_compare_data(struct ib_cm_compare_data *src_data,
 445			   struct ib_cm_compare_data *dst_data)
 446{
 447	u8 src[IB_CM_COMPARE_SIZE];
 448	u8 dst[IB_CM_COMPARE_SIZE];
 449
 450	if (!src_data || !dst_data)
 451		return 0;
 452
 453	cm_mask_copy(src, src_data->data, dst_data->mask);
 454	cm_mask_copy(dst, dst_data->data, src_data->mask);
 455	return memcmp(src, dst, IB_CM_COMPARE_SIZE);
 456}
 457
 458static int cm_compare_private_data(u8 *private_data,
 459				   struct ib_cm_compare_data *dst_data)
 460{
 461	u8 src[IB_CM_COMPARE_SIZE];
 462
 463	if (!dst_data)
 464		return 0;
 465
 466	cm_mask_copy(src, private_data, dst_data->mask);
 467	return memcmp(src, dst_data->data, IB_CM_COMPARE_SIZE);
 468}
 469
 470/*
 471 * Trivial helpers to strip endian annotation and compare; the
 472 * endianness doesn't actually matter since we just need a stable
 473 * order for the RB tree.
 474 */
 475static int be32_lt(__be32 a, __be32 b)
 476{
 477	return (__force u32) a < (__force u32) b;
 478}
 479
 480static int be32_gt(__be32 a, __be32 b)
 481{
 482	return (__force u32) a > (__force u32) b;
 483}
 484
 485static int be64_lt(__be64 a, __be64 b)
 486{
 487	return (__force u64) a < (__force u64) b;
 488}
 489
 490static int be64_gt(__be64 a, __be64 b)
 491{
 492	return (__force u64) a > (__force u64) b;
 493}
 494
 495static struct cm_id_private * cm_insert_listen(struct cm_id_private *cm_id_priv)
 496{
 497	struct rb_node **link = &cm.listen_service_table.rb_node;
 498	struct rb_node *parent = NULL;
 499	struct cm_id_private *cur_cm_id_priv;
 500	__be64 service_id = cm_id_priv->id.service_id;
 501	__be64 service_mask = cm_id_priv->id.service_mask;
 502	int data_cmp;
 503
 504	while (*link) {
 505		parent = *link;
 506		cur_cm_id_priv = rb_entry(parent, struct cm_id_private,
 507					  service_node);
 508		data_cmp = cm_compare_data(cm_id_priv->compare_data,
 509					   cur_cm_id_priv->compare_data);
 510		if ((cur_cm_id_priv->id.service_mask & service_id) ==
 511		    (service_mask & cur_cm_id_priv->id.service_id) &&
 512		    (cm_id_priv->id.device == cur_cm_id_priv->id.device) &&
 513		    !data_cmp)
 514			return cur_cm_id_priv;
 515
 516		if (cm_id_priv->id.device < cur_cm_id_priv->id.device)
 517			link = &(*link)->rb_left;
 518		else if (cm_id_priv->id.device > cur_cm_id_priv->id.device)
 519			link = &(*link)->rb_right;
 520		else if (be64_lt(service_id, cur_cm_id_priv->id.service_id))
 521			link = &(*link)->rb_left;
 522		else if (be64_gt(service_id, cur_cm_id_priv->id.service_id))
 523			link = &(*link)->rb_right;
 524		else if (data_cmp < 0)
 525			link = &(*link)->rb_left;
 526		else
 527			link = &(*link)->rb_right;
 528	}
 529	rb_link_node(&cm_id_priv->service_node, parent, link);
 530	rb_insert_color(&cm_id_priv->service_node, &cm.listen_service_table);
 531	return NULL;
 532}
 533
 534static struct cm_id_private * cm_find_listen(struct ib_device *device,
 535					     __be64 service_id,
 536					     u8 *private_data)
 537{
 538	struct rb_node *node = cm.listen_service_table.rb_node;
 539	struct cm_id_private *cm_id_priv;
 540	int data_cmp;
 541
 542	while (node) {
 543		cm_id_priv = rb_entry(node, struct cm_id_private, service_node);
 544		data_cmp = cm_compare_private_data(private_data,
 545						   cm_id_priv->compare_data);
 546		if ((cm_id_priv->id.service_mask & service_id) ==
 547		     cm_id_priv->id.service_id &&
 548		    (cm_id_priv->id.device == device) && !data_cmp)
 549			return cm_id_priv;
 550
 551		if (device < cm_id_priv->id.device)
 552			node = node->rb_left;
 553		else if (device > cm_id_priv->id.device)
 554			node = node->rb_right;
 555		else if (be64_lt(service_id, cm_id_priv->id.service_id))
 556			node = node->rb_left;
 557		else if (be64_gt(service_id, cm_id_priv->id.service_id))
 558			node = node->rb_right;
 559		else if (data_cmp < 0)
 560			node = node->rb_left;
 561		else
 562			node = node->rb_right;
 563	}
 564	return NULL;
 565}
 566
 567static struct cm_timewait_info * cm_insert_remote_id(struct cm_timewait_info
 568						     *timewait_info)
 569{
 570	struct rb_node **link = &cm.remote_id_table.rb_node;
 571	struct rb_node *parent = NULL;
 572	struct cm_timewait_info *cur_timewait_info;
 573	__be64 remote_ca_guid = timewait_info->remote_ca_guid;
 574	__be32 remote_id = timewait_info->work.remote_id;
 575
 576	while (*link) {
 577		parent = *link;
 578		cur_timewait_info = rb_entry(parent, struct cm_timewait_info,
 579					     remote_id_node);
 580		if (be32_lt(remote_id, cur_timewait_info->work.remote_id))
 581			link = &(*link)->rb_left;
 582		else if (be32_gt(remote_id, cur_timewait_info->work.remote_id))
 583			link = &(*link)->rb_right;
 584		else if (be64_lt(remote_ca_guid, cur_timewait_info->remote_ca_guid))
 585			link = &(*link)->rb_left;
 586		else if (be64_gt(remote_ca_guid, cur_timewait_info->remote_ca_guid))
 587			link = &(*link)->rb_right;
 588		else
 589			return cur_timewait_info;
 590	}
 591	timewait_info->inserted_remote_id = 1;
 592	rb_link_node(&timewait_info->remote_id_node, parent, link);
 593	rb_insert_color(&timewait_info->remote_id_node, &cm.remote_id_table);
 594	return NULL;
 595}
 596
 597static struct cm_timewait_info * cm_find_remote_id(__be64 remote_ca_guid,
 598						   __be32 remote_id)
 599{
 600	struct rb_node *node = cm.remote_id_table.rb_node;
 601	struct cm_timewait_info *timewait_info;
 602
 603	while (node) {
 604		timewait_info = rb_entry(node, struct cm_timewait_info,
 605					 remote_id_node);
 606		if (be32_lt(remote_id, timewait_info->work.remote_id))
 607			node = node->rb_left;
 608		else if (be32_gt(remote_id, timewait_info->work.remote_id))
 609			node = node->rb_right;
 610		else if (be64_lt(remote_ca_guid, timewait_info->remote_ca_guid))
 611			node = node->rb_left;
 612		else if (be64_gt(remote_ca_guid, timewait_info->remote_ca_guid))
 613			node = node->rb_right;
 614		else
 615			return timewait_info;
 616	}
 617	return NULL;
 618}
 619
 620static struct cm_timewait_info * cm_insert_remote_qpn(struct cm_timewait_info
 621						      *timewait_info)
 622{
 623	struct rb_node **link = &cm.remote_qp_table.rb_node;
 624	struct rb_node *parent = NULL;
 625	struct cm_timewait_info *cur_timewait_info;
 626	__be64 remote_ca_guid = timewait_info->remote_ca_guid;
 627	__be32 remote_qpn = timewait_info->remote_qpn;
 628
 629	while (*link) {
 630		parent = *link;
 631		cur_timewait_info = rb_entry(parent, struct cm_timewait_info,
 632					     remote_qp_node);
 633		if (be32_lt(remote_qpn, cur_timewait_info->remote_qpn))
 634			link = &(*link)->rb_left;
 635		else if (be32_gt(remote_qpn, cur_timewait_info->remote_qpn))
 636			link = &(*link)->rb_right;
 637		else if (be64_lt(remote_ca_guid, cur_timewait_info->remote_ca_guid))
 638			link = &(*link)->rb_left;
 639		else if (be64_gt(remote_ca_guid, cur_timewait_info->remote_ca_guid))
 640			link = &(*link)->rb_right;
 641		else
 642			return cur_timewait_info;
 643	}
 644	timewait_info->inserted_remote_qp = 1;
 645	rb_link_node(&timewait_info->remote_qp_node, parent, link);
 646	rb_insert_color(&timewait_info->remote_qp_node, &cm.remote_qp_table);
 647	return NULL;
 648}
 649
 650static struct cm_id_private * cm_insert_remote_sidr(struct cm_id_private
 651						    *cm_id_priv)
 652{
 653	struct rb_node **link = &cm.remote_sidr_table.rb_node;
 654	struct rb_node *parent = NULL;
 655	struct cm_id_private *cur_cm_id_priv;
 656	union ib_gid *port_gid = &cm_id_priv->av.dgid;
 657	__be32 remote_id = cm_id_priv->id.remote_id;
 658
 659	while (*link) {
 660		parent = *link;
 661		cur_cm_id_priv = rb_entry(parent, struct cm_id_private,
 662					  sidr_id_node);
 663		if (be32_lt(remote_id, cur_cm_id_priv->id.remote_id))
 664			link = &(*link)->rb_left;
 665		else if (be32_gt(remote_id, cur_cm_id_priv->id.remote_id))
 666			link = &(*link)->rb_right;
 667		else {
 668			int cmp;
 669			cmp = memcmp(port_gid, &cur_cm_id_priv->av.dgid,
 670				     sizeof *port_gid);
 671			if (cmp < 0)
 672				link = &(*link)->rb_left;
 673			else if (cmp > 0)
 674				link = &(*link)->rb_right;
 675			else
 676				return cur_cm_id_priv;
 677		}
 678	}
 679	rb_link_node(&cm_id_priv->sidr_id_node, parent, link);
 680	rb_insert_color(&cm_id_priv->sidr_id_node, &cm.remote_sidr_table);
 681	return NULL;
 682}
 683
 684static void cm_reject_sidr_req(struct cm_id_private *cm_id_priv,
 685			       enum ib_cm_sidr_status status)
 686{
 687	struct ib_cm_sidr_rep_param param;
 688
 689	memset(&param, 0, sizeof param);
 690	param.status = status;
 691	ib_send_cm_sidr_rep(&cm_id_priv->id, &param);
 692}
 693
 694struct ib_cm_id *ib_create_cm_id(struct ib_device *device,
 695				 ib_cm_handler cm_handler,
 696				 void *context)
 697{
 698	struct cm_id_private *cm_id_priv;
 699	int ret;
 700
 701	cm_id_priv = kzalloc(sizeof *cm_id_priv, GFP_KERNEL);
 702	if (!cm_id_priv)
 703		return ERR_PTR(-ENOMEM);
 704
 705	cm_id_priv->id.state = IB_CM_IDLE;
 706	cm_id_priv->id.device = device;
 707	cm_id_priv->id.cm_handler = cm_handler;
 708	cm_id_priv->id.context = context;
 709	cm_id_priv->id.remote_cm_qpn = 1;
 710	ret = cm_alloc_id(cm_id_priv);
 711	if (ret)
 712		goto error;
 713
 714	spin_lock_init(&cm_id_priv->lock);
 715	init_completion(&cm_id_priv->comp);
 716	INIT_LIST_HEAD(&cm_id_priv->work_list);
 717	atomic_set(&cm_id_priv->work_count, -1);
 718	atomic_set(&cm_id_priv->refcount, 1);
 719	return &cm_id_priv->id;
 720
 721error:
 722	kfree(cm_id_priv);
 723	return ERR_PTR(-ENOMEM);
 724}
 725EXPORT_SYMBOL(ib_create_cm_id);
 726
 727static struct cm_work * cm_dequeue_work(struct cm_id_private *cm_id_priv)
 728{
 729	struct cm_work *work;
 730
 731	if (list_empty(&cm_id_priv->work_list))
 732		return NULL;
 733
 734	work = list_entry(cm_id_priv->work_list.next, struct cm_work, list);
 735	list_del(&work->list);
 736	return work;
 737}
 738
 739static void cm_free_work(struct cm_work *work)
 740{
 741	if (work->mad_recv_wc)
 742		ib_free_recv_mad(work->mad_recv_wc);
 743	kfree(work);
 744}
 745
 746static inline int cm_convert_to_ms(int iba_time)
 747{
 748	/* approximate conversion to ms from 4.096us x 2^iba_time */
 749	return 1 << max(iba_time - 8, 0);
 750}
 751
 752/*
 753 * calculate: 4.096x2^ack_timeout = 4.096x2^ack_delay + 2x4.096x2^life_time
 754 * Because of how ack_timeout is stored, adding one doubles the timeout.
 755 * To avoid large timeouts, select the max(ack_delay, life_time + 1), and
 756 * increment it (round up) only if the other is within 50%.
 757 */
 758static u8 cm_ack_timeout(u8 ca_ack_delay, u8 packet_life_time)
 759{
 760	int ack_timeout = packet_life_time + 1;
 761
 762	if (ack_timeout >= ca_ack_delay)
 763		ack_timeout += (ca_ack_delay >= (ack_timeout - 1));
 764	else
 765		ack_timeout = ca_ack_delay +
 766			      (ack_timeout >= (ca_ack_delay - 1));
 767
 768	return min(31, ack_timeout);
 769}
 770
 771static void cm_cleanup_timewait(struct cm_timewait_info *timewait_info)
 772{
 773	if (timewait_info->inserted_remote_id) {
 774		rb_erase(&timewait_info->remote_id_node, &cm.remote_id_table);
 775		timewait_info->inserted_remote_id = 0;
 776	}
 777
 778	if (timewait_info->inserted_remote_qp) {
 779		rb_erase(&timewait_info->remote_qp_node, &cm.remote_qp_table);
 780		timewait_info->inserted_remote_qp = 0;
 781	}
 782}
 783
 784static struct cm_timewait_info * cm_create_timewait_info(__be32 local_id)
 785{
 786	struct cm_timewait_info *timewait_info;
 787
 788	timewait_info = kzalloc(sizeof *timewait_info, GFP_KERNEL);
 789	if (!timewait_info)
 790		return ERR_PTR(-ENOMEM);
 791
 792	timewait_info->work.local_id = local_id;
 793	INIT_DELAYED_WORK(&timewait_info->work.work, cm_work_handler);
 794	timewait_info->work.cm_event.event = IB_CM_TIMEWAIT_EXIT;
 795	return timewait_info;
 796}
 797
 798static void cm_enter_timewait(struct cm_id_private *cm_id_priv)
 799{
 800	int wait_time;
 801	unsigned long flags;
 
 
 
 
 
 802
 803	spin_lock_irqsave(&cm.lock, flags);
 804	cm_cleanup_timewait(cm_id_priv->timewait_info);
 805	list_add_tail(&cm_id_priv->timewait_info->list, &cm.timewait_list);
 806	spin_unlock_irqrestore(&cm.lock, flags);
 807
 808	/*
 809	 * The cm_id could be destroyed by the user before we exit timewait.
 810	 * To protect against this, we search for the cm_id after exiting
 811	 * timewait before notifying the user that we've exited timewait.
 812	 */
 813	cm_id_priv->id.state = IB_CM_TIMEWAIT;
 814	wait_time = cm_convert_to_ms(cm_id_priv->av.timeout);
 815	queue_delayed_work(cm.wq, &cm_id_priv->timewait_info->work.work,
 816			   msecs_to_jiffies(wait_time));
 
 
 
 
 
 
 817	cm_id_priv->timewait_info = NULL;
 818}
 819
 820static void cm_reset_to_idle(struct cm_id_private *cm_id_priv)
 821{
 822	unsigned long flags;
 823
 824	cm_id_priv->id.state = IB_CM_IDLE;
 825	if (cm_id_priv->timewait_info) {
 826		spin_lock_irqsave(&cm.lock, flags);
 827		cm_cleanup_timewait(cm_id_priv->timewait_info);
 828		spin_unlock_irqrestore(&cm.lock, flags);
 829		kfree(cm_id_priv->timewait_info);
 830		cm_id_priv->timewait_info = NULL;
 831	}
 832}
 833
 834static void cm_destroy_id(struct ib_cm_id *cm_id, int err)
 835{
 836	struct cm_id_private *cm_id_priv;
 837	struct cm_work *work;
 838
 839	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
 840retest:
 841	spin_lock_irq(&cm_id_priv->lock);
 842	switch (cm_id->state) {
 843	case IB_CM_LISTEN:
 844		cm_id->state = IB_CM_IDLE;
 845		spin_unlock_irq(&cm_id_priv->lock);
 
 846		spin_lock_irq(&cm.lock);
 
 
 
 
 
 
 847		rb_erase(&cm_id_priv->service_node, &cm.listen_service_table);
 848		spin_unlock_irq(&cm.lock);
 849		break;
 850	case IB_CM_SIDR_REQ_SENT:
 851		cm_id->state = IB_CM_IDLE;
 852		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
 853		spin_unlock_irq(&cm_id_priv->lock);
 854		break;
 855	case IB_CM_SIDR_REQ_RCVD:
 856		spin_unlock_irq(&cm_id_priv->lock);
 857		cm_reject_sidr_req(cm_id_priv, IB_SIDR_REJECT);
 
 
 
 
 
 858		break;
 859	case IB_CM_REQ_SENT:
 
 860		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
 861		spin_unlock_irq(&cm_id_priv->lock);
 862		ib_send_cm_rej(cm_id, IB_CM_REJ_TIMEOUT,
 863			       &cm_id_priv->id.device->node_guid,
 864			       sizeof cm_id_priv->id.device->node_guid,
 865			       NULL, 0);
 866		break;
 867	case IB_CM_REQ_RCVD:
 868		if (err == -ENOMEM) {
 869			/* Do not reject to allow future retries. */
 870			cm_reset_to_idle(cm_id_priv);
 871			spin_unlock_irq(&cm_id_priv->lock);
 872		} else {
 873			spin_unlock_irq(&cm_id_priv->lock);
 874			ib_send_cm_rej(cm_id, IB_CM_REJ_CONSUMER_DEFINED,
 875				       NULL, 0, NULL, 0);
 876		}
 877		break;
 878	case IB_CM_MRA_REQ_RCVD:
 879	case IB_CM_REP_SENT:
 880	case IB_CM_MRA_REP_RCVD:
 881		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
 882		/* Fall through */
 883	case IB_CM_MRA_REQ_SENT:
 884	case IB_CM_REP_RCVD:
 885	case IB_CM_MRA_REP_SENT:
 886		spin_unlock_irq(&cm_id_priv->lock);
 887		ib_send_cm_rej(cm_id, IB_CM_REJ_CONSUMER_DEFINED,
 888			       NULL, 0, NULL, 0);
 889		break;
 890	case IB_CM_ESTABLISHED:
 891		spin_unlock_irq(&cm_id_priv->lock);
 
 
 892		ib_send_cm_dreq(cm_id, NULL, 0);
 893		goto retest;
 894	case IB_CM_DREQ_SENT:
 895		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
 896		cm_enter_timewait(cm_id_priv);
 897		spin_unlock_irq(&cm_id_priv->lock);
 898		break;
 899	case IB_CM_DREQ_RCVD:
 900		spin_unlock_irq(&cm_id_priv->lock);
 901		ib_send_cm_drep(cm_id, NULL, 0);
 902		break;
 903	default:
 904		spin_unlock_irq(&cm_id_priv->lock);
 905		break;
 906	}
 907
 908	cm_free_id(cm_id->local_id);
 909	cm_deref_id(cm_id_priv);
 910	wait_for_completion(&cm_id_priv->comp);
 911	while ((work = cm_dequeue_work(cm_id_priv)) != NULL)
 912		cm_free_work(work);
 913	kfree(cm_id_priv->compare_data);
 914	kfree(cm_id_priv->private_data);
 915	kfree(cm_id_priv);
 916}
 917
 918void ib_destroy_cm_id(struct ib_cm_id *cm_id)
 919{
 920	cm_destroy_id(cm_id, 0);
 921}
 922EXPORT_SYMBOL(ib_destroy_cm_id);
 923
 924int ib_cm_listen(struct ib_cm_id *cm_id, __be64 service_id, __be64 service_mask,
 925		 struct ib_cm_compare_data *compare_data)
 
 
 
 
 
 
 
 
 
 
 
 
 
 926{
 927	struct cm_id_private *cm_id_priv, *cur_cm_id_priv;
 928	unsigned long flags;
 929	int ret = 0;
 930
 931	service_mask = service_mask ? service_mask : ~cpu_to_be64(0);
 932	service_id &= service_mask;
 933	if ((service_id & IB_SERVICE_ID_AGN_MASK) == IB_CM_ASSIGN_SERVICE_ID &&
 934	    (service_id != IB_CM_ASSIGN_SERVICE_ID))
 935		return -EINVAL;
 936
 937	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
 938	if (cm_id->state != IB_CM_IDLE)
 939		return -EINVAL;
 940
 941	if (compare_data) {
 942		cm_id_priv->compare_data = kzalloc(sizeof *compare_data,
 943						   GFP_KERNEL);
 944		if (!cm_id_priv->compare_data)
 945			return -ENOMEM;
 946		cm_mask_copy(cm_id_priv->compare_data->data,
 947			     compare_data->data, compare_data->mask);
 948		memcpy(cm_id_priv->compare_data->mask, compare_data->mask,
 949		       IB_CM_COMPARE_SIZE);
 950	}
 951
 952	cm_id->state = IB_CM_LISTEN;
 
 953
 954	spin_lock_irqsave(&cm.lock, flags);
 955	if (service_id == IB_CM_ASSIGN_SERVICE_ID) {
 956		cm_id->service_id = cpu_to_be64(cm.listen_service_id++);
 957		cm_id->service_mask = ~cpu_to_be64(0);
 958	} else {
 959		cm_id->service_id = service_id;
 960		cm_id->service_mask = service_mask;
 961	}
 962	cur_cm_id_priv = cm_insert_listen(cm_id_priv);
 963	spin_unlock_irqrestore(&cm.lock, flags);
 964
 965	if (cur_cm_id_priv) {
 966		cm_id->state = IB_CM_IDLE;
 967		kfree(cm_id_priv->compare_data);
 968		cm_id_priv->compare_data = NULL;
 969		ret = -EBUSY;
 970	}
 971	return ret;
 972}
 
 
 
 
 
 
 
 
 
 
 
 
 973EXPORT_SYMBOL(ib_cm_listen);
 974
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 975static __be64 cm_form_tid(struct cm_id_private *cm_id_priv,
 976			  enum cm_msg_sequence msg_seq)
 977{
 978	u64 hi_tid, low_tid;
 979
 980	hi_tid   = ((u64) cm_id_priv->av.port->mad_agent->hi_tid) << 32;
 981	low_tid  = (u64) ((__force u32)cm_id_priv->id.local_id |
 982			  (msg_seq << 30));
 983	return cpu_to_be64(hi_tid | low_tid);
 984}
 985
 986static void cm_format_mad_hdr(struct ib_mad_hdr *hdr,
 987			      __be16 attr_id, __be64 tid)
 988{
 989	hdr->base_version  = IB_MGMT_BASE_VERSION;
 990	hdr->mgmt_class	   = IB_MGMT_CLASS_CM;
 991	hdr->class_version = IB_CM_CLASS_VERSION;
 992	hdr->method	   = IB_MGMT_METHOD_SEND;
 993	hdr->attr_id	   = attr_id;
 994	hdr->tid	   = tid;
 995}
 996
 997static void cm_format_req(struct cm_req_msg *req_msg,
 998			  struct cm_id_private *cm_id_priv,
 999			  struct ib_cm_req_param *param)
1000{
1001	struct ib_sa_path_rec *pri_path = param->primary_path;
1002	struct ib_sa_path_rec *alt_path = param->alternate_path;
1003
1004	cm_format_mad_hdr(&req_msg->hdr, CM_REQ_ATTR_ID,
1005			  cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_REQ));
1006
1007	req_msg->local_comm_id = cm_id_priv->id.local_id;
1008	req_msg->service_id = param->service_id;
1009	req_msg->local_ca_guid = cm_id_priv->id.device->node_guid;
1010	cm_req_set_local_qpn(req_msg, cpu_to_be32(param->qp_num));
1011	cm_req_set_resp_res(req_msg, param->responder_resources);
1012	cm_req_set_init_depth(req_msg, param->initiator_depth);
1013	cm_req_set_remote_resp_timeout(req_msg,
1014				       param->remote_cm_response_timeout);
1015	cm_req_set_qp_type(req_msg, param->qp_type);
1016	cm_req_set_flow_ctrl(req_msg, param->flow_control);
1017	cm_req_set_starting_psn(req_msg, cpu_to_be32(param->starting_psn));
1018	cm_req_set_local_resp_timeout(req_msg,
1019				      param->local_cm_response_timeout);
1020	cm_req_set_retry_count(req_msg, param->retry_count);
1021	req_msg->pkey = param->primary_path->pkey;
1022	cm_req_set_path_mtu(req_msg, param->primary_path->mtu);
1023	cm_req_set_rnr_retry_count(req_msg, param->rnr_retry_count);
1024	cm_req_set_max_cm_retries(req_msg, param->max_cm_retries);
1025	cm_req_set_srq(req_msg, param->srq);
 
 
 
 
 
 
1026
1027	if (pri_path->hop_limit <= 1) {
1028		req_msg->primary_local_lid = pri_path->slid;
1029		req_msg->primary_remote_lid = pri_path->dlid;
1030	} else {
1031		/* Work-around until there's a way to obtain remote LID info */
1032		req_msg->primary_local_lid = IB_LID_PERMISSIVE;
1033		req_msg->primary_remote_lid = IB_LID_PERMISSIVE;
1034	}
1035	req_msg->primary_local_gid = pri_path->sgid;
1036	req_msg->primary_remote_gid = pri_path->dgid;
1037	cm_req_set_primary_flow_label(req_msg, pri_path->flow_label);
1038	cm_req_set_primary_packet_rate(req_msg, pri_path->rate);
1039	req_msg->primary_traffic_class = pri_path->traffic_class;
1040	req_msg->primary_hop_limit = pri_path->hop_limit;
1041	cm_req_set_primary_sl(req_msg, pri_path->sl);
1042	cm_req_set_primary_subnet_local(req_msg, (pri_path->hop_limit <= 1));
1043	cm_req_set_primary_local_ack_timeout(req_msg,
1044		cm_ack_timeout(cm_id_priv->av.port->cm_dev->ack_delay,
1045			       pri_path->packet_life_time));
1046
1047	if (alt_path) {
1048		if (alt_path->hop_limit <= 1) {
1049			req_msg->alt_local_lid = alt_path->slid;
1050			req_msg->alt_remote_lid = alt_path->dlid;
1051		} else {
1052			req_msg->alt_local_lid = IB_LID_PERMISSIVE;
1053			req_msg->alt_remote_lid = IB_LID_PERMISSIVE;
1054		}
1055		req_msg->alt_local_gid = alt_path->sgid;
1056		req_msg->alt_remote_gid = alt_path->dgid;
1057		cm_req_set_alt_flow_label(req_msg,
1058					  alt_path->flow_label);
1059		cm_req_set_alt_packet_rate(req_msg, alt_path->rate);
1060		req_msg->alt_traffic_class = alt_path->traffic_class;
1061		req_msg->alt_hop_limit = alt_path->hop_limit;
1062		cm_req_set_alt_sl(req_msg, alt_path->sl);
1063		cm_req_set_alt_subnet_local(req_msg, (alt_path->hop_limit <= 1));
1064		cm_req_set_alt_local_ack_timeout(req_msg,
1065			cm_ack_timeout(cm_id_priv->av.port->cm_dev->ack_delay,
1066				       alt_path->packet_life_time));
1067	}
1068
1069	if (param->private_data && param->private_data_len)
1070		memcpy(req_msg->private_data, param->private_data,
1071		       param->private_data_len);
1072}
1073
1074static int cm_validate_req_param(struct ib_cm_req_param *param)
1075{
1076	/* peer-to-peer not supported */
1077	if (param->peer_to_peer)
1078		return -EINVAL;
1079
1080	if (!param->primary_path)
1081		return -EINVAL;
1082
1083	if (param->qp_type != IB_QPT_RC && param->qp_type != IB_QPT_UC)
 
1084		return -EINVAL;
1085
1086	if (param->private_data &&
1087	    param->private_data_len > IB_CM_REQ_PRIVATE_DATA_SIZE)
1088		return -EINVAL;
1089
1090	if (param->alternate_path &&
1091	    (param->alternate_path->pkey != param->primary_path->pkey ||
1092	     param->alternate_path->mtu != param->primary_path->mtu))
1093		return -EINVAL;
1094
1095	return 0;
1096}
1097
1098int ib_send_cm_req(struct ib_cm_id *cm_id,
1099		   struct ib_cm_req_param *param)
1100{
1101	struct cm_id_private *cm_id_priv;
1102	struct cm_req_msg *req_msg;
1103	unsigned long flags;
1104	int ret;
1105
1106	ret = cm_validate_req_param(param);
1107	if (ret)
1108		return ret;
1109
1110	/* Verify that we're not in timewait. */
1111	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1112	spin_lock_irqsave(&cm_id_priv->lock, flags);
1113	if (cm_id->state != IB_CM_IDLE) {
1114		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1115		ret = -EINVAL;
1116		goto out;
1117	}
1118	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1119
1120	cm_id_priv->timewait_info = cm_create_timewait_info(cm_id_priv->
1121							    id.local_id);
1122	if (IS_ERR(cm_id_priv->timewait_info)) {
1123		ret = PTR_ERR(cm_id_priv->timewait_info);
1124		goto out;
1125	}
1126
1127	ret = cm_init_av_by_path(param->primary_path, &cm_id_priv->av);
1128	if (ret)
1129		goto error1;
1130	if (param->alternate_path) {
1131		ret = cm_init_av_by_path(param->alternate_path,
1132					 &cm_id_priv->alt_av);
1133		if (ret)
1134			goto error1;
1135	}
1136	cm_id->service_id = param->service_id;
1137	cm_id->service_mask = ~cpu_to_be64(0);
1138	cm_id_priv->timeout_ms = cm_convert_to_ms(
1139				    param->primary_path->packet_life_time) * 2 +
1140				 cm_convert_to_ms(
1141				    param->remote_cm_response_timeout);
1142	cm_id_priv->max_cm_retries = param->max_cm_retries;
1143	cm_id_priv->initiator_depth = param->initiator_depth;
1144	cm_id_priv->responder_resources = param->responder_resources;
1145	cm_id_priv->retry_count = param->retry_count;
1146	cm_id_priv->path_mtu = param->primary_path->mtu;
1147	cm_id_priv->pkey = param->primary_path->pkey;
1148	cm_id_priv->qp_type = param->qp_type;
1149
1150	ret = cm_alloc_msg(cm_id_priv, &cm_id_priv->msg);
1151	if (ret)
1152		goto error1;
1153
1154	req_msg = (struct cm_req_msg *) cm_id_priv->msg->mad;
1155	cm_format_req(req_msg, cm_id_priv, param);
1156	cm_id_priv->tid = req_msg->hdr.tid;
1157	cm_id_priv->msg->timeout_ms = cm_id_priv->timeout_ms;
1158	cm_id_priv->msg->context[1] = (void *) (unsigned long) IB_CM_REQ_SENT;
1159
1160	cm_id_priv->local_qpn = cm_req_get_local_qpn(req_msg);
1161	cm_id_priv->rq_psn = cm_req_get_starting_psn(req_msg);
1162
1163	spin_lock_irqsave(&cm_id_priv->lock, flags);
1164	ret = ib_post_send_mad(cm_id_priv->msg, NULL);
1165	if (ret) {
1166		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1167		goto error2;
1168	}
1169	BUG_ON(cm_id->state != IB_CM_IDLE);
1170	cm_id->state = IB_CM_REQ_SENT;
1171	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1172	return 0;
1173
1174error2:	cm_free_msg(cm_id_priv->msg);
1175error1:	kfree(cm_id_priv->timewait_info);
1176out:	return ret;
1177}
1178EXPORT_SYMBOL(ib_send_cm_req);
1179
1180static int cm_issue_rej(struct cm_port *port,
1181			struct ib_mad_recv_wc *mad_recv_wc,
1182			enum ib_cm_rej_reason reason,
1183			enum cm_msg_response msg_rejected,
1184			void *ari, u8 ari_length)
1185{
1186	struct ib_mad_send_buf *msg = NULL;
1187	struct cm_rej_msg *rej_msg, *rcv_msg;
1188	int ret;
1189
1190	ret = cm_alloc_response_msg(port, mad_recv_wc, &msg);
1191	if (ret)
1192		return ret;
1193
1194	/* We just need common CM header information.  Cast to any message. */
1195	rcv_msg = (struct cm_rej_msg *) mad_recv_wc->recv_buf.mad;
1196	rej_msg = (struct cm_rej_msg *) msg->mad;
1197
1198	cm_format_mad_hdr(&rej_msg->hdr, CM_REJ_ATTR_ID, rcv_msg->hdr.tid);
1199	rej_msg->remote_comm_id = rcv_msg->local_comm_id;
1200	rej_msg->local_comm_id = rcv_msg->remote_comm_id;
1201	cm_rej_set_msg_rejected(rej_msg, msg_rejected);
1202	rej_msg->reason = cpu_to_be16(reason);
1203
1204	if (ari && ari_length) {
1205		cm_rej_set_reject_info_len(rej_msg, ari_length);
1206		memcpy(rej_msg->ari, ari, ari_length);
1207	}
1208
1209	ret = ib_post_send_mad(msg, NULL);
1210	if (ret)
1211		cm_free_msg(msg);
1212
1213	return ret;
1214}
1215
1216static inline int cm_is_active_peer(__be64 local_ca_guid, __be64 remote_ca_guid,
1217				    __be32 local_qpn, __be32 remote_qpn)
1218{
1219	return (be64_to_cpu(local_ca_guid) > be64_to_cpu(remote_ca_guid) ||
1220		((local_ca_guid == remote_ca_guid) &&
1221		 (be32_to_cpu(local_qpn) > be32_to_cpu(remote_qpn))));
1222}
1223
1224static void cm_format_paths_from_req(struct cm_req_msg *req_msg,
1225					    struct ib_sa_path_rec *primary_path,
1226					    struct ib_sa_path_rec *alt_path)
1227{
1228	memset(primary_path, 0, sizeof *primary_path);
1229	primary_path->dgid = req_msg->primary_local_gid;
1230	primary_path->sgid = req_msg->primary_remote_gid;
1231	primary_path->dlid = req_msg->primary_local_lid;
1232	primary_path->slid = req_msg->primary_remote_lid;
1233	primary_path->flow_label = cm_req_get_primary_flow_label(req_msg);
1234	primary_path->hop_limit = req_msg->primary_hop_limit;
1235	primary_path->traffic_class = req_msg->primary_traffic_class;
1236	primary_path->reversible = 1;
1237	primary_path->pkey = req_msg->pkey;
1238	primary_path->sl = cm_req_get_primary_sl(req_msg);
1239	primary_path->mtu_selector = IB_SA_EQ;
1240	primary_path->mtu = cm_req_get_path_mtu(req_msg);
1241	primary_path->rate_selector = IB_SA_EQ;
1242	primary_path->rate = cm_req_get_primary_packet_rate(req_msg);
1243	primary_path->packet_life_time_selector = IB_SA_EQ;
1244	primary_path->packet_life_time =
1245		cm_req_get_primary_local_ack_timeout(req_msg);
1246	primary_path->packet_life_time -= (primary_path->packet_life_time > 0);
 
1247
1248	if (req_msg->alt_local_lid) {
1249		memset(alt_path, 0, sizeof *alt_path);
1250		alt_path->dgid = req_msg->alt_local_gid;
1251		alt_path->sgid = req_msg->alt_remote_gid;
1252		alt_path->dlid = req_msg->alt_local_lid;
1253		alt_path->slid = req_msg->alt_remote_lid;
1254		alt_path->flow_label = cm_req_get_alt_flow_label(req_msg);
1255		alt_path->hop_limit = req_msg->alt_hop_limit;
1256		alt_path->traffic_class = req_msg->alt_traffic_class;
1257		alt_path->reversible = 1;
1258		alt_path->pkey = req_msg->pkey;
1259		alt_path->sl = cm_req_get_alt_sl(req_msg);
1260		alt_path->mtu_selector = IB_SA_EQ;
1261		alt_path->mtu = cm_req_get_path_mtu(req_msg);
1262		alt_path->rate_selector = IB_SA_EQ;
1263		alt_path->rate = cm_req_get_alt_packet_rate(req_msg);
1264		alt_path->packet_life_time_selector = IB_SA_EQ;
1265		alt_path->packet_life_time =
1266			cm_req_get_alt_local_ack_timeout(req_msg);
1267		alt_path->packet_life_time -= (alt_path->packet_life_time > 0);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1268	}
 
 
1269}
1270
1271static void cm_format_req_event(struct cm_work *work,
1272				struct cm_id_private *cm_id_priv,
1273				struct ib_cm_id *listen_id)
1274{
1275	struct cm_req_msg *req_msg;
1276	struct ib_cm_req_event_param *param;
1277
1278	req_msg = (struct cm_req_msg *)work->mad_recv_wc->recv_buf.mad;
1279	param = &work->cm_event.param.req_rcvd;
1280	param->listen_id = listen_id;
 
1281	param->port = cm_id_priv->av.port->port_num;
1282	param->primary_path = &work->path[0];
1283	if (req_msg->alt_local_lid)
1284		param->alternate_path = &work->path[1];
1285	else
1286		param->alternate_path = NULL;
1287	param->remote_ca_guid = req_msg->local_ca_guid;
1288	param->remote_qkey = be32_to_cpu(req_msg->local_qkey);
1289	param->remote_qpn = be32_to_cpu(cm_req_get_local_qpn(req_msg));
1290	param->qp_type = cm_req_get_qp_type(req_msg);
1291	param->starting_psn = be32_to_cpu(cm_req_get_starting_psn(req_msg));
1292	param->responder_resources = cm_req_get_init_depth(req_msg);
1293	param->initiator_depth = cm_req_get_resp_res(req_msg);
1294	param->local_cm_response_timeout =
1295					cm_req_get_remote_resp_timeout(req_msg);
1296	param->flow_control = cm_req_get_flow_ctrl(req_msg);
1297	param->remote_cm_response_timeout =
1298					cm_req_get_local_resp_timeout(req_msg);
1299	param->retry_count = cm_req_get_retry_count(req_msg);
1300	param->rnr_retry_count = cm_req_get_rnr_retry_count(req_msg);
1301	param->srq = cm_req_get_srq(req_msg);
1302	work->cm_event.private_data = &req_msg->private_data;
1303}
1304
1305static void cm_process_work(struct cm_id_private *cm_id_priv,
1306			    struct cm_work *work)
1307{
1308	int ret;
1309
1310	/* We will typically only have the current event to report. */
1311	ret = cm_id_priv->id.cm_handler(&cm_id_priv->id, &work->cm_event);
1312	cm_free_work(work);
1313
1314	while (!ret && !atomic_add_negative(-1, &cm_id_priv->work_count)) {
1315		spin_lock_irq(&cm_id_priv->lock);
1316		work = cm_dequeue_work(cm_id_priv);
1317		spin_unlock_irq(&cm_id_priv->lock);
1318		BUG_ON(!work);
1319		ret = cm_id_priv->id.cm_handler(&cm_id_priv->id,
1320						&work->cm_event);
1321		cm_free_work(work);
1322	}
1323	cm_deref_id(cm_id_priv);
1324	if (ret)
1325		cm_destroy_id(&cm_id_priv->id, ret);
1326}
1327
1328static void cm_format_mra(struct cm_mra_msg *mra_msg,
1329			  struct cm_id_private *cm_id_priv,
1330			  enum cm_msg_response msg_mraed, u8 service_timeout,
1331			  const void *private_data, u8 private_data_len)
1332{
1333	cm_format_mad_hdr(&mra_msg->hdr, CM_MRA_ATTR_ID, cm_id_priv->tid);
1334	cm_mra_set_msg_mraed(mra_msg, msg_mraed);
1335	mra_msg->local_comm_id = cm_id_priv->id.local_id;
1336	mra_msg->remote_comm_id = cm_id_priv->id.remote_id;
1337	cm_mra_set_service_timeout(mra_msg, service_timeout);
1338
1339	if (private_data && private_data_len)
1340		memcpy(mra_msg->private_data, private_data, private_data_len);
1341}
1342
1343static void cm_format_rej(struct cm_rej_msg *rej_msg,
1344			  struct cm_id_private *cm_id_priv,
1345			  enum ib_cm_rej_reason reason,
1346			  void *ari,
1347			  u8 ari_length,
1348			  const void *private_data,
1349			  u8 private_data_len)
1350{
1351	cm_format_mad_hdr(&rej_msg->hdr, CM_REJ_ATTR_ID, cm_id_priv->tid);
1352	rej_msg->remote_comm_id = cm_id_priv->id.remote_id;
1353
1354	switch(cm_id_priv->id.state) {
1355	case IB_CM_REQ_RCVD:
1356		rej_msg->local_comm_id = 0;
1357		cm_rej_set_msg_rejected(rej_msg, CM_MSG_RESPONSE_REQ);
1358		break;
1359	case IB_CM_MRA_REQ_SENT:
1360		rej_msg->local_comm_id = cm_id_priv->id.local_id;
1361		cm_rej_set_msg_rejected(rej_msg, CM_MSG_RESPONSE_REQ);
1362		break;
1363	case IB_CM_REP_RCVD:
1364	case IB_CM_MRA_REP_SENT:
1365		rej_msg->local_comm_id = cm_id_priv->id.local_id;
1366		cm_rej_set_msg_rejected(rej_msg, CM_MSG_RESPONSE_REP);
1367		break;
1368	default:
1369		rej_msg->local_comm_id = cm_id_priv->id.local_id;
1370		cm_rej_set_msg_rejected(rej_msg, CM_MSG_RESPONSE_OTHER);
1371		break;
1372	}
1373
1374	rej_msg->reason = cpu_to_be16(reason);
1375	if (ari && ari_length) {
1376		cm_rej_set_reject_info_len(rej_msg, ari_length);
1377		memcpy(rej_msg->ari, ari, ari_length);
1378	}
1379
1380	if (private_data && private_data_len)
1381		memcpy(rej_msg->private_data, private_data, private_data_len);
1382}
1383
1384static void cm_dup_req_handler(struct cm_work *work,
1385			       struct cm_id_private *cm_id_priv)
1386{
1387	struct ib_mad_send_buf *msg = NULL;
1388	int ret;
1389
1390	atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
1391			counter[CM_REQ_COUNTER]);
1392
1393	/* Quick state check to discard duplicate REQs. */
1394	if (cm_id_priv->id.state == IB_CM_REQ_RCVD)
1395		return;
1396
1397	ret = cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg);
1398	if (ret)
1399		return;
1400
1401	spin_lock_irq(&cm_id_priv->lock);
1402	switch (cm_id_priv->id.state) {
1403	case IB_CM_MRA_REQ_SENT:
1404		cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
1405			      CM_MSG_RESPONSE_REQ, cm_id_priv->service_timeout,
1406			      cm_id_priv->private_data,
1407			      cm_id_priv->private_data_len);
1408		break;
1409	case IB_CM_TIMEWAIT:
1410		cm_format_rej((struct cm_rej_msg *) msg->mad, cm_id_priv,
1411			      IB_CM_REJ_STALE_CONN, NULL, 0, NULL, 0);
1412		break;
1413	default:
1414		goto unlock;
1415	}
1416	spin_unlock_irq(&cm_id_priv->lock);
1417
1418	ret = ib_post_send_mad(msg, NULL);
1419	if (ret)
1420		goto free;
1421	return;
1422
1423unlock:	spin_unlock_irq(&cm_id_priv->lock);
1424free:	cm_free_msg(msg);
1425}
1426
1427static struct cm_id_private * cm_match_req(struct cm_work *work,
1428					   struct cm_id_private *cm_id_priv)
1429{
1430	struct cm_id_private *listen_cm_id_priv, *cur_cm_id_priv;
1431	struct cm_timewait_info *timewait_info;
1432	struct cm_req_msg *req_msg;
1433
1434	req_msg = (struct cm_req_msg *)work->mad_recv_wc->recv_buf.mad;
1435
1436	/* Check for possible duplicate REQ. */
1437	spin_lock_irq(&cm.lock);
1438	timewait_info = cm_insert_remote_id(cm_id_priv->timewait_info);
1439	if (timewait_info) {
1440		cur_cm_id_priv = cm_get_id(timewait_info->work.local_id,
1441					   timewait_info->work.remote_id);
1442		spin_unlock_irq(&cm.lock);
1443		if (cur_cm_id_priv) {
1444			cm_dup_req_handler(work, cur_cm_id_priv);
1445			cm_deref_id(cur_cm_id_priv);
1446		}
1447		return NULL;
1448	}
1449
1450	/* Check for stale connections. */
1451	timewait_info = cm_insert_remote_qpn(cm_id_priv->timewait_info);
1452	if (timewait_info) {
1453		cm_cleanup_timewait(cm_id_priv->timewait_info);
1454		spin_unlock_irq(&cm.lock);
1455		cm_issue_rej(work->port, work->mad_recv_wc,
1456			     IB_CM_REJ_STALE_CONN, CM_MSG_RESPONSE_REQ,
1457			     NULL, 0);
1458		return NULL;
1459	}
1460
1461	/* Find matching listen request. */
1462	listen_cm_id_priv = cm_find_listen(cm_id_priv->id.device,
1463					   req_msg->service_id,
1464					   req_msg->private_data);
1465	if (!listen_cm_id_priv) {
1466		cm_cleanup_timewait(cm_id_priv->timewait_info);
1467		spin_unlock_irq(&cm.lock);
1468		cm_issue_rej(work->port, work->mad_recv_wc,
1469			     IB_CM_REJ_INVALID_SERVICE_ID, CM_MSG_RESPONSE_REQ,
1470			     NULL, 0);
1471		goto out;
1472	}
1473	atomic_inc(&listen_cm_id_priv->refcount);
1474	atomic_inc(&cm_id_priv->refcount);
1475	cm_id_priv->id.state = IB_CM_REQ_RCVD;
1476	atomic_inc(&cm_id_priv->work_count);
1477	spin_unlock_irq(&cm.lock);
1478out:
1479	return listen_cm_id_priv;
1480}
1481
1482/*
1483 * Work-around for inter-subnet connections.  If the LIDs are permissive,
1484 * we need to override the LID/SL data in the REQ with the LID information
1485 * in the work completion.
1486 */
1487static void cm_process_routed_req(struct cm_req_msg *req_msg, struct ib_wc *wc)
1488{
1489	if (!cm_req_get_primary_subnet_local(req_msg)) {
1490		if (req_msg->primary_local_lid == IB_LID_PERMISSIVE) {
1491			req_msg->primary_local_lid = cpu_to_be16(wc->slid);
1492			cm_req_set_primary_sl(req_msg, wc->sl);
1493		}
1494
1495		if (req_msg->primary_remote_lid == IB_LID_PERMISSIVE)
1496			req_msg->primary_remote_lid = cpu_to_be16(wc->dlid_path_bits);
1497	}
1498
1499	if (!cm_req_get_alt_subnet_local(req_msg)) {
1500		if (req_msg->alt_local_lid == IB_LID_PERMISSIVE) {
1501			req_msg->alt_local_lid = cpu_to_be16(wc->slid);
1502			cm_req_set_alt_sl(req_msg, wc->sl);
1503		}
1504
1505		if (req_msg->alt_remote_lid == IB_LID_PERMISSIVE)
1506			req_msg->alt_remote_lid = cpu_to_be16(wc->dlid_path_bits);
1507	}
1508}
1509
1510static int cm_req_handler(struct cm_work *work)
1511{
1512	struct ib_cm_id *cm_id;
1513	struct cm_id_private *cm_id_priv, *listen_cm_id_priv;
1514	struct cm_req_msg *req_msg;
 
 
1515	int ret;
1516
1517	req_msg = (struct cm_req_msg *)work->mad_recv_wc->recv_buf.mad;
1518
1519	cm_id = ib_create_cm_id(work->port->cm_dev->ib_device, NULL, NULL);
1520	if (IS_ERR(cm_id))
1521		return PTR_ERR(cm_id);
1522
1523	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1524	cm_id_priv->id.remote_id = req_msg->local_comm_id;
1525	cm_init_av_for_response(work->port, work->mad_recv_wc->wc,
1526				work->mad_recv_wc->recv_buf.grh,
1527				&cm_id_priv->av);
1528	cm_id_priv->timewait_info = cm_create_timewait_info(cm_id_priv->
1529							    id.local_id);
1530	if (IS_ERR(cm_id_priv->timewait_info)) {
1531		ret = PTR_ERR(cm_id_priv->timewait_info);
1532		goto destroy;
1533	}
1534	cm_id_priv->timewait_info->work.remote_id = req_msg->local_comm_id;
1535	cm_id_priv->timewait_info->remote_ca_guid = req_msg->local_ca_guid;
1536	cm_id_priv->timewait_info->remote_qpn = cm_req_get_local_qpn(req_msg);
1537
1538	listen_cm_id_priv = cm_match_req(work, cm_id_priv);
1539	if (!listen_cm_id_priv) {
1540		ret = -EINVAL;
1541		kfree(cm_id_priv->timewait_info);
1542		goto destroy;
1543	}
1544
1545	cm_id_priv->id.cm_handler = listen_cm_id_priv->id.cm_handler;
1546	cm_id_priv->id.context = listen_cm_id_priv->id.context;
1547	cm_id_priv->id.service_id = req_msg->service_id;
1548	cm_id_priv->id.service_mask = ~cpu_to_be64(0);
1549
1550	cm_process_routed_req(req_msg, work->mad_recv_wc->wc);
1551	cm_format_paths_from_req(req_msg, &work->path[0], &work->path[1]);
1552	ret = cm_init_av_by_path(&work->path[0], &cm_id_priv->av);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1553	if (ret) {
1554		ib_get_cached_gid(work->port->cm_dev->ib_device,
1555				  work->port->port_num, 0, &work->path[0].sgid);
 
 
 
 
 
 
 
 
1556		ib_send_cm_rej(cm_id, IB_CM_REJ_INVALID_GID,
1557			       &work->path[0].sgid, sizeof work->path[0].sgid,
1558			       NULL, 0);
1559		goto rejected;
1560	}
1561	if (req_msg->alt_local_lid) {
1562		ret = cm_init_av_by_path(&work->path[1], &cm_id_priv->alt_av);
1563		if (ret) {
1564			ib_send_cm_rej(cm_id, IB_CM_REJ_INVALID_ALT_GID,
1565				       &work->path[0].sgid,
1566				       sizeof work->path[0].sgid, NULL, 0);
1567			goto rejected;
1568		}
1569	}
1570	cm_id_priv->tid = req_msg->hdr.tid;
1571	cm_id_priv->timeout_ms = cm_convert_to_ms(
1572					cm_req_get_local_resp_timeout(req_msg));
1573	cm_id_priv->max_cm_retries = cm_req_get_max_cm_retries(req_msg);
1574	cm_id_priv->remote_qpn = cm_req_get_local_qpn(req_msg);
1575	cm_id_priv->initiator_depth = cm_req_get_resp_res(req_msg);
1576	cm_id_priv->responder_resources = cm_req_get_init_depth(req_msg);
1577	cm_id_priv->path_mtu = cm_req_get_path_mtu(req_msg);
1578	cm_id_priv->pkey = req_msg->pkey;
1579	cm_id_priv->sq_psn = cm_req_get_starting_psn(req_msg);
1580	cm_id_priv->retry_count = cm_req_get_retry_count(req_msg);
1581	cm_id_priv->rnr_retry_count = cm_req_get_rnr_retry_count(req_msg);
1582	cm_id_priv->qp_type = cm_req_get_qp_type(req_msg);
1583
1584	cm_format_req_event(work, cm_id_priv, &listen_cm_id_priv->id);
1585	cm_process_work(cm_id_priv, work);
1586	cm_deref_id(listen_cm_id_priv);
1587	return 0;
1588
1589rejected:
1590	atomic_dec(&cm_id_priv->refcount);
1591	cm_deref_id(listen_cm_id_priv);
1592destroy:
1593	ib_destroy_cm_id(cm_id);
1594	return ret;
1595}
1596
1597static void cm_format_rep(struct cm_rep_msg *rep_msg,
1598			  struct cm_id_private *cm_id_priv,
1599			  struct ib_cm_rep_param *param)
1600{
1601	cm_format_mad_hdr(&rep_msg->hdr, CM_REP_ATTR_ID, cm_id_priv->tid);
1602	rep_msg->local_comm_id = cm_id_priv->id.local_id;
1603	rep_msg->remote_comm_id = cm_id_priv->id.remote_id;
1604	cm_rep_set_local_qpn(rep_msg, cpu_to_be32(param->qp_num));
1605	cm_rep_set_starting_psn(rep_msg, cpu_to_be32(param->starting_psn));
1606	rep_msg->resp_resources = param->responder_resources;
1607	rep_msg->initiator_depth = param->initiator_depth;
1608	cm_rep_set_target_ack_delay(rep_msg,
1609				    cm_id_priv->av.port->cm_dev->ack_delay);
1610	cm_rep_set_failover(rep_msg, param->failover_accepted);
1611	cm_rep_set_flow_ctrl(rep_msg, param->flow_control);
1612	cm_rep_set_rnr_retry_count(rep_msg, param->rnr_retry_count);
1613	cm_rep_set_srq(rep_msg, param->srq);
1614	rep_msg->local_ca_guid = cm_id_priv->id.device->node_guid;
1615
 
 
 
 
 
 
 
 
 
 
1616	if (param->private_data && param->private_data_len)
1617		memcpy(rep_msg->private_data, param->private_data,
1618		       param->private_data_len);
1619}
1620
1621int ib_send_cm_rep(struct ib_cm_id *cm_id,
1622		   struct ib_cm_rep_param *param)
1623{
1624	struct cm_id_private *cm_id_priv;
1625	struct ib_mad_send_buf *msg;
1626	struct cm_rep_msg *rep_msg;
1627	unsigned long flags;
1628	int ret;
1629
1630	if (param->private_data &&
1631	    param->private_data_len > IB_CM_REP_PRIVATE_DATA_SIZE)
1632		return -EINVAL;
1633
1634	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1635	spin_lock_irqsave(&cm_id_priv->lock, flags);
1636	if (cm_id->state != IB_CM_REQ_RCVD &&
1637	    cm_id->state != IB_CM_MRA_REQ_SENT) {
1638		ret = -EINVAL;
1639		goto out;
1640	}
1641
1642	ret = cm_alloc_msg(cm_id_priv, &msg);
1643	if (ret)
1644		goto out;
1645
1646	rep_msg = (struct cm_rep_msg *) msg->mad;
1647	cm_format_rep(rep_msg, cm_id_priv, param);
1648	msg->timeout_ms = cm_id_priv->timeout_ms;
1649	msg->context[1] = (void *) (unsigned long) IB_CM_REP_SENT;
1650
1651	ret = ib_post_send_mad(msg, NULL);
1652	if (ret) {
1653		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1654		cm_free_msg(msg);
1655		return ret;
1656	}
1657
1658	cm_id->state = IB_CM_REP_SENT;
1659	cm_id_priv->msg = msg;
1660	cm_id_priv->initiator_depth = param->initiator_depth;
1661	cm_id_priv->responder_resources = param->responder_resources;
1662	cm_id_priv->rq_psn = cm_rep_get_starting_psn(rep_msg);
1663	cm_id_priv->local_qpn = cm_rep_get_local_qpn(rep_msg);
1664
1665out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1666	return ret;
1667}
1668EXPORT_SYMBOL(ib_send_cm_rep);
1669
1670static void cm_format_rtu(struct cm_rtu_msg *rtu_msg,
1671			  struct cm_id_private *cm_id_priv,
1672			  const void *private_data,
1673			  u8 private_data_len)
1674{
1675	cm_format_mad_hdr(&rtu_msg->hdr, CM_RTU_ATTR_ID, cm_id_priv->tid);
1676	rtu_msg->local_comm_id = cm_id_priv->id.local_id;
1677	rtu_msg->remote_comm_id = cm_id_priv->id.remote_id;
1678
1679	if (private_data && private_data_len)
1680		memcpy(rtu_msg->private_data, private_data, private_data_len);
1681}
1682
1683int ib_send_cm_rtu(struct ib_cm_id *cm_id,
1684		   const void *private_data,
1685		   u8 private_data_len)
1686{
1687	struct cm_id_private *cm_id_priv;
1688	struct ib_mad_send_buf *msg;
1689	unsigned long flags;
1690	void *data;
1691	int ret;
1692
1693	if (private_data && private_data_len > IB_CM_RTU_PRIVATE_DATA_SIZE)
1694		return -EINVAL;
1695
1696	data = cm_copy_private_data(private_data, private_data_len);
1697	if (IS_ERR(data))
1698		return PTR_ERR(data);
1699
1700	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1701	spin_lock_irqsave(&cm_id_priv->lock, flags);
1702	if (cm_id->state != IB_CM_REP_RCVD &&
1703	    cm_id->state != IB_CM_MRA_REP_SENT) {
1704		ret = -EINVAL;
1705		goto error;
1706	}
1707
1708	ret = cm_alloc_msg(cm_id_priv, &msg);
1709	if (ret)
1710		goto error;
1711
1712	cm_format_rtu((struct cm_rtu_msg *) msg->mad, cm_id_priv,
1713		      private_data, private_data_len);
1714
1715	ret = ib_post_send_mad(msg, NULL);
1716	if (ret) {
1717		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1718		cm_free_msg(msg);
1719		kfree(data);
1720		return ret;
1721	}
1722
1723	cm_id->state = IB_CM_ESTABLISHED;
1724	cm_set_private_data(cm_id_priv, data, private_data_len);
1725	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1726	return 0;
1727
1728error:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1729	kfree(data);
1730	return ret;
1731}
1732EXPORT_SYMBOL(ib_send_cm_rtu);
1733
1734static void cm_format_rep_event(struct cm_work *work)
1735{
1736	struct cm_rep_msg *rep_msg;
1737	struct ib_cm_rep_event_param *param;
1738
1739	rep_msg = (struct cm_rep_msg *)work->mad_recv_wc->recv_buf.mad;
1740	param = &work->cm_event.param.rep_rcvd;
1741	param->remote_ca_guid = rep_msg->local_ca_guid;
1742	param->remote_qkey = be32_to_cpu(rep_msg->local_qkey);
1743	param->remote_qpn = be32_to_cpu(cm_rep_get_local_qpn(rep_msg));
1744	param->starting_psn = be32_to_cpu(cm_rep_get_starting_psn(rep_msg));
1745	param->responder_resources = rep_msg->initiator_depth;
1746	param->initiator_depth = rep_msg->resp_resources;
1747	param->target_ack_delay = cm_rep_get_target_ack_delay(rep_msg);
1748	param->failover_accepted = cm_rep_get_failover(rep_msg);
1749	param->flow_control = cm_rep_get_flow_ctrl(rep_msg);
1750	param->rnr_retry_count = cm_rep_get_rnr_retry_count(rep_msg);
1751	param->srq = cm_rep_get_srq(rep_msg);
1752	work->cm_event.private_data = &rep_msg->private_data;
1753}
1754
1755static void cm_dup_rep_handler(struct cm_work *work)
1756{
1757	struct cm_id_private *cm_id_priv;
1758	struct cm_rep_msg *rep_msg;
1759	struct ib_mad_send_buf *msg = NULL;
1760	int ret;
1761
1762	rep_msg = (struct cm_rep_msg *) work->mad_recv_wc->recv_buf.mad;
1763	cm_id_priv = cm_acquire_id(rep_msg->remote_comm_id,
1764				   rep_msg->local_comm_id);
1765	if (!cm_id_priv)
1766		return;
1767
1768	atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
1769			counter[CM_REP_COUNTER]);
1770	ret = cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg);
1771	if (ret)
1772		goto deref;
1773
1774	spin_lock_irq(&cm_id_priv->lock);
1775	if (cm_id_priv->id.state == IB_CM_ESTABLISHED)
1776		cm_format_rtu((struct cm_rtu_msg *) msg->mad, cm_id_priv,
1777			      cm_id_priv->private_data,
1778			      cm_id_priv->private_data_len);
1779	else if (cm_id_priv->id.state == IB_CM_MRA_REP_SENT)
1780		cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
1781			      CM_MSG_RESPONSE_REP, cm_id_priv->service_timeout,
1782			      cm_id_priv->private_data,
1783			      cm_id_priv->private_data_len);
1784	else
1785		goto unlock;
1786	spin_unlock_irq(&cm_id_priv->lock);
1787
1788	ret = ib_post_send_mad(msg, NULL);
1789	if (ret)
1790		goto free;
1791	goto deref;
1792
1793unlock:	spin_unlock_irq(&cm_id_priv->lock);
1794free:	cm_free_msg(msg);
1795deref:	cm_deref_id(cm_id_priv);
1796}
1797
1798static int cm_rep_handler(struct cm_work *work)
1799{
1800	struct cm_id_private *cm_id_priv;
1801	struct cm_rep_msg *rep_msg;
1802	int ret;
1803
1804	rep_msg = (struct cm_rep_msg *)work->mad_recv_wc->recv_buf.mad;
1805	cm_id_priv = cm_acquire_id(rep_msg->remote_comm_id, 0);
1806	if (!cm_id_priv) {
1807		cm_dup_rep_handler(work);
1808		return -EINVAL;
1809	}
1810
1811	cm_format_rep_event(work);
1812
1813	spin_lock_irq(&cm_id_priv->lock);
1814	switch (cm_id_priv->id.state) {
1815	case IB_CM_REQ_SENT:
1816	case IB_CM_MRA_REQ_RCVD:
1817		break;
1818	default:
1819		spin_unlock_irq(&cm_id_priv->lock);
1820		ret = -EINVAL;
1821		goto error;
1822	}
1823
1824	cm_id_priv->timewait_info->work.remote_id = rep_msg->local_comm_id;
1825	cm_id_priv->timewait_info->remote_ca_guid = rep_msg->local_ca_guid;
1826	cm_id_priv->timewait_info->remote_qpn = cm_rep_get_local_qpn(rep_msg);
1827
1828	spin_lock(&cm.lock);
1829	/* Check for duplicate REP. */
1830	if (cm_insert_remote_id(cm_id_priv->timewait_info)) {
1831		spin_unlock(&cm.lock);
1832		spin_unlock_irq(&cm_id_priv->lock);
1833		ret = -EINVAL;
1834		goto error;
1835	}
1836	/* Check for a stale connection. */
1837	if (cm_insert_remote_qpn(cm_id_priv->timewait_info)) {
1838		rb_erase(&cm_id_priv->timewait_info->remote_id_node,
1839			 &cm.remote_id_table);
1840		cm_id_priv->timewait_info->inserted_remote_id = 0;
1841		spin_unlock(&cm.lock);
1842		spin_unlock_irq(&cm_id_priv->lock);
1843		cm_issue_rej(work->port, work->mad_recv_wc,
1844			     IB_CM_REJ_STALE_CONN, CM_MSG_RESPONSE_REP,
1845			     NULL, 0);
1846		ret = -EINVAL;
1847		goto error;
1848	}
1849	spin_unlock(&cm.lock);
1850
1851	cm_id_priv->id.state = IB_CM_REP_RCVD;
1852	cm_id_priv->id.remote_id = rep_msg->local_comm_id;
1853	cm_id_priv->remote_qpn = cm_rep_get_local_qpn(rep_msg);
1854	cm_id_priv->initiator_depth = rep_msg->resp_resources;
1855	cm_id_priv->responder_resources = rep_msg->initiator_depth;
1856	cm_id_priv->sq_psn = cm_rep_get_starting_psn(rep_msg);
1857	cm_id_priv->rnr_retry_count = cm_rep_get_rnr_retry_count(rep_msg);
1858	cm_id_priv->target_ack_delay = cm_rep_get_target_ack_delay(rep_msg);
1859	cm_id_priv->av.timeout =
1860			cm_ack_timeout(cm_id_priv->target_ack_delay,
1861				       cm_id_priv->av.timeout - 1);
1862	cm_id_priv->alt_av.timeout =
1863			cm_ack_timeout(cm_id_priv->target_ack_delay,
1864				       cm_id_priv->alt_av.timeout - 1);
1865
1866	/* todo: handle peer_to_peer */
1867
1868	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
1869	ret = atomic_inc_and_test(&cm_id_priv->work_count);
1870	if (!ret)
1871		list_add_tail(&work->list, &cm_id_priv->work_list);
1872	spin_unlock_irq(&cm_id_priv->lock);
1873
1874	if (ret)
1875		cm_process_work(cm_id_priv, work);
1876	else
1877		cm_deref_id(cm_id_priv);
1878	return 0;
1879
1880error:
1881	cm_deref_id(cm_id_priv);
1882	return ret;
1883}
1884
1885static int cm_establish_handler(struct cm_work *work)
1886{
1887	struct cm_id_private *cm_id_priv;
1888	int ret;
1889
1890	/* See comment in cm_establish about lookup. */
1891	cm_id_priv = cm_acquire_id(work->local_id, work->remote_id);
1892	if (!cm_id_priv)
1893		return -EINVAL;
1894
1895	spin_lock_irq(&cm_id_priv->lock);
1896	if (cm_id_priv->id.state != IB_CM_ESTABLISHED) {
1897		spin_unlock_irq(&cm_id_priv->lock);
1898		goto out;
1899	}
1900
1901	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
1902	ret = atomic_inc_and_test(&cm_id_priv->work_count);
1903	if (!ret)
1904		list_add_tail(&work->list, &cm_id_priv->work_list);
1905	spin_unlock_irq(&cm_id_priv->lock);
1906
1907	if (ret)
1908		cm_process_work(cm_id_priv, work);
1909	else
1910		cm_deref_id(cm_id_priv);
1911	return 0;
1912out:
1913	cm_deref_id(cm_id_priv);
1914	return -EINVAL;
1915}
1916
1917static int cm_rtu_handler(struct cm_work *work)
1918{
1919	struct cm_id_private *cm_id_priv;
1920	struct cm_rtu_msg *rtu_msg;
1921	int ret;
1922
1923	rtu_msg = (struct cm_rtu_msg *)work->mad_recv_wc->recv_buf.mad;
1924	cm_id_priv = cm_acquire_id(rtu_msg->remote_comm_id,
1925				   rtu_msg->local_comm_id);
1926	if (!cm_id_priv)
1927		return -EINVAL;
1928
1929	work->cm_event.private_data = &rtu_msg->private_data;
1930
1931	spin_lock_irq(&cm_id_priv->lock);
1932	if (cm_id_priv->id.state != IB_CM_REP_SENT &&
1933	    cm_id_priv->id.state != IB_CM_MRA_REP_RCVD) {
1934		spin_unlock_irq(&cm_id_priv->lock);
1935		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
1936				counter[CM_RTU_COUNTER]);
1937		goto out;
1938	}
1939	cm_id_priv->id.state = IB_CM_ESTABLISHED;
1940
1941	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
1942	ret = atomic_inc_and_test(&cm_id_priv->work_count);
1943	if (!ret)
1944		list_add_tail(&work->list, &cm_id_priv->work_list);
1945	spin_unlock_irq(&cm_id_priv->lock);
1946
1947	if (ret)
1948		cm_process_work(cm_id_priv, work);
1949	else
1950		cm_deref_id(cm_id_priv);
1951	return 0;
1952out:
1953	cm_deref_id(cm_id_priv);
1954	return -EINVAL;
1955}
1956
1957static void cm_format_dreq(struct cm_dreq_msg *dreq_msg,
1958			  struct cm_id_private *cm_id_priv,
1959			  const void *private_data,
1960			  u8 private_data_len)
1961{
1962	cm_format_mad_hdr(&dreq_msg->hdr, CM_DREQ_ATTR_ID,
1963			  cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_DREQ));
1964	dreq_msg->local_comm_id = cm_id_priv->id.local_id;
1965	dreq_msg->remote_comm_id = cm_id_priv->id.remote_id;
1966	cm_dreq_set_remote_qpn(dreq_msg, cm_id_priv->remote_qpn);
1967
1968	if (private_data && private_data_len)
1969		memcpy(dreq_msg->private_data, private_data, private_data_len);
1970}
1971
1972int ib_send_cm_dreq(struct ib_cm_id *cm_id,
1973		    const void *private_data,
1974		    u8 private_data_len)
1975{
1976	struct cm_id_private *cm_id_priv;
1977	struct ib_mad_send_buf *msg;
1978	unsigned long flags;
1979	int ret;
1980
1981	if (private_data && private_data_len > IB_CM_DREQ_PRIVATE_DATA_SIZE)
1982		return -EINVAL;
1983
1984	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1985	spin_lock_irqsave(&cm_id_priv->lock, flags);
1986	if (cm_id->state != IB_CM_ESTABLISHED) {
1987		ret = -EINVAL;
1988		goto out;
1989	}
1990
1991	if (cm_id->lap_state == IB_CM_LAP_SENT ||
1992	    cm_id->lap_state == IB_CM_MRA_LAP_RCVD)
1993		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
1994
1995	ret = cm_alloc_msg(cm_id_priv, &msg);
1996	if (ret) {
1997		cm_enter_timewait(cm_id_priv);
1998		goto out;
1999	}
2000
2001	cm_format_dreq((struct cm_dreq_msg *) msg->mad, cm_id_priv,
2002		       private_data, private_data_len);
2003	msg->timeout_ms = cm_id_priv->timeout_ms;
2004	msg->context[1] = (void *) (unsigned long) IB_CM_DREQ_SENT;
2005
2006	ret = ib_post_send_mad(msg, NULL);
2007	if (ret) {
2008		cm_enter_timewait(cm_id_priv);
2009		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2010		cm_free_msg(msg);
2011		return ret;
2012	}
2013
2014	cm_id->state = IB_CM_DREQ_SENT;
2015	cm_id_priv->msg = msg;
2016out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2017	return ret;
2018}
2019EXPORT_SYMBOL(ib_send_cm_dreq);
2020
2021static void cm_format_drep(struct cm_drep_msg *drep_msg,
2022			  struct cm_id_private *cm_id_priv,
2023			  const void *private_data,
2024			  u8 private_data_len)
2025{
2026	cm_format_mad_hdr(&drep_msg->hdr, CM_DREP_ATTR_ID, cm_id_priv->tid);
2027	drep_msg->local_comm_id = cm_id_priv->id.local_id;
2028	drep_msg->remote_comm_id = cm_id_priv->id.remote_id;
2029
2030	if (private_data && private_data_len)
2031		memcpy(drep_msg->private_data, private_data, private_data_len);
2032}
2033
2034int ib_send_cm_drep(struct ib_cm_id *cm_id,
2035		    const void *private_data,
2036		    u8 private_data_len)
2037{
2038	struct cm_id_private *cm_id_priv;
2039	struct ib_mad_send_buf *msg;
2040	unsigned long flags;
2041	void *data;
2042	int ret;
2043
2044	if (private_data && private_data_len > IB_CM_DREP_PRIVATE_DATA_SIZE)
2045		return -EINVAL;
2046
2047	data = cm_copy_private_data(private_data, private_data_len);
2048	if (IS_ERR(data))
2049		return PTR_ERR(data);
2050
2051	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2052	spin_lock_irqsave(&cm_id_priv->lock, flags);
2053	if (cm_id->state != IB_CM_DREQ_RCVD) {
2054		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2055		kfree(data);
2056		return -EINVAL;
2057	}
2058
2059	cm_set_private_data(cm_id_priv, data, private_data_len);
2060	cm_enter_timewait(cm_id_priv);
2061
2062	ret = cm_alloc_msg(cm_id_priv, &msg);
2063	if (ret)
2064		goto out;
2065
2066	cm_format_drep((struct cm_drep_msg *) msg->mad, cm_id_priv,
2067		       private_data, private_data_len);
2068
2069	ret = ib_post_send_mad(msg, NULL);
2070	if (ret) {
2071		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2072		cm_free_msg(msg);
2073		return ret;
2074	}
2075
2076out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2077	return ret;
2078}
2079EXPORT_SYMBOL(ib_send_cm_drep);
2080
2081static int cm_issue_drep(struct cm_port *port,
2082			 struct ib_mad_recv_wc *mad_recv_wc)
2083{
2084	struct ib_mad_send_buf *msg = NULL;
2085	struct cm_dreq_msg *dreq_msg;
2086	struct cm_drep_msg *drep_msg;
2087	int ret;
2088
2089	ret = cm_alloc_response_msg(port, mad_recv_wc, &msg);
2090	if (ret)
2091		return ret;
2092
2093	dreq_msg = (struct cm_dreq_msg *) mad_recv_wc->recv_buf.mad;
2094	drep_msg = (struct cm_drep_msg *) msg->mad;
2095
2096	cm_format_mad_hdr(&drep_msg->hdr, CM_DREP_ATTR_ID, dreq_msg->hdr.tid);
2097	drep_msg->remote_comm_id = dreq_msg->local_comm_id;
2098	drep_msg->local_comm_id = dreq_msg->remote_comm_id;
2099
2100	ret = ib_post_send_mad(msg, NULL);
2101	if (ret)
2102		cm_free_msg(msg);
2103
2104	return ret;
2105}
2106
2107static int cm_dreq_handler(struct cm_work *work)
2108{
2109	struct cm_id_private *cm_id_priv;
2110	struct cm_dreq_msg *dreq_msg;
2111	struct ib_mad_send_buf *msg = NULL;
2112	int ret;
2113
2114	dreq_msg = (struct cm_dreq_msg *)work->mad_recv_wc->recv_buf.mad;
2115	cm_id_priv = cm_acquire_id(dreq_msg->remote_comm_id,
2116				   dreq_msg->local_comm_id);
2117	if (!cm_id_priv) {
2118		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2119				counter[CM_DREQ_COUNTER]);
2120		cm_issue_drep(work->port, work->mad_recv_wc);
2121		return -EINVAL;
2122	}
2123
2124	work->cm_event.private_data = &dreq_msg->private_data;
2125
2126	spin_lock_irq(&cm_id_priv->lock);
2127	if (cm_id_priv->local_qpn != cm_dreq_get_remote_qpn(dreq_msg))
2128		goto unlock;
2129
2130	switch (cm_id_priv->id.state) {
2131	case IB_CM_REP_SENT:
2132	case IB_CM_DREQ_SENT:
2133		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2134		break;
2135	case IB_CM_ESTABLISHED:
2136		if (cm_id_priv->id.lap_state == IB_CM_LAP_SENT ||
2137		    cm_id_priv->id.lap_state == IB_CM_MRA_LAP_RCVD)
2138			ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2139		break;
2140	case IB_CM_MRA_REP_RCVD:
2141		break;
2142	case IB_CM_TIMEWAIT:
2143		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2144				counter[CM_DREQ_COUNTER]);
2145		if (cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg))
2146			goto unlock;
2147
2148		cm_format_drep((struct cm_drep_msg *) msg->mad, cm_id_priv,
2149			       cm_id_priv->private_data,
2150			       cm_id_priv->private_data_len);
2151		spin_unlock_irq(&cm_id_priv->lock);
2152
2153		if (ib_post_send_mad(msg, NULL))
2154			cm_free_msg(msg);
2155		goto deref;
2156	case IB_CM_DREQ_RCVD:
2157		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2158				counter[CM_DREQ_COUNTER]);
2159		goto unlock;
2160	default:
2161		goto unlock;
2162	}
2163	cm_id_priv->id.state = IB_CM_DREQ_RCVD;
2164	cm_id_priv->tid = dreq_msg->hdr.tid;
2165	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2166	if (!ret)
2167		list_add_tail(&work->list, &cm_id_priv->work_list);
2168	spin_unlock_irq(&cm_id_priv->lock);
2169
2170	if (ret)
2171		cm_process_work(cm_id_priv, work);
2172	else
2173		cm_deref_id(cm_id_priv);
2174	return 0;
2175
2176unlock:	spin_unlock_irq(&cm_id_priv->lock);
2177deref:	cm_deref_id(cm_id_priv);
2178	return -EINVAL;
2179}
2180
2181static int cm_drep_handler(struct cm_work *work)
2182{
2183	struct cm_id_private *cm_id_priv;
2184	struct cm_drep_msg *drep_msg;
2185	int ret;
2186
2187	drep_msg = (struct cm_drep_msg *)work->mad_recv_wc->recv_buf.mad;
2188	cm_id_priv = cm_acquire_id(drep_msg->remote_comm_id,
2189				   drep_msg->local_comm_id);
2190	if (!cm_id_priv)
2191		return -EINVAL;
2192
2193	work->cm_event.private_data = &drep_msg->private_data;
2194
2195	spin_lock_irq(&cm_id_priv->lock);
2196	if (cm_id_priv->id.state != IB_CM_DREQ_SENT &&
2197	    cm_id_priv->id.state != IB_CM_DREQ_RCVD) {
2198		spin_unlock_irq(&cm_id_priv->lock);
2199		goto out;
2200	}
2201	cm_enter_timewait(cm_id_priv);
2202
2203	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2204	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2205	if (!ret)
2206		list_add_tail(&work->list, &cm_id_priv->work_list);
2207	spin_unlock_irq(&cm_id_priv->lock);
2208
2209	if (ret)
2210		cm_process_work(cm_id_priv, work);
2211	else
2212		cm_deref_id(cm_id_priv);
2213	return 0;
2214out:
2215	cm_deref_id(cm_id_priv);
2216	return -EINVAL;
2217}
2218
2219int ib_send_cm_rej(struct ib_cm_id *cm_id,
2220		   enum ib_cm_rej_reason reason,
2221		   void *ari,
2222		   u8 ari_length,
2223		   const void *private_data,
2224		   u8 private_data_len)
2225{
2226	struct cm_id_private *cm_id_priv;
2227	struct ib_mad_send_buf *msg;
2228	unsigned long flags;
2229	int ret;
2230
2231	if ((private_data && private_data_len > IB_CM_REJ_PRIVATE_DATA_SIZE) ||
2232	    (ari && ari_length > IB_CM_REJ_ARI_LENGTH))
2233		return -EINVAL;
2234
2235	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2236
2237	spin_lock_irqsave(&cm_id_priv->lock, flags);
2238	switch (cm_id->state) {
2239	case IB_CM_REQ_SENT:
2240	case IB_CM_MRA_REQ_RCVD:
2241	case IB_CM_REQ_RCVD:
2242	case IB_CM_MRA_REQ_SENT:
2243	case IB_CM_REP_RCVD:
2244	case IB_CM_MRA_REP_SENT:
2245		ret = cm_alloc_msg(cm_id_priv, &msg);
2246		if (!ret)
2247			cm_format_rej((struct cm_rej_msg *) msg->mad,
2248				      cm_id_priv, reason, ari, ari_length,
2249				      private_data, private_data_len);
2250
2251		cm_reset_to_idle(cm_id_priv);
2252		break;
2253	case IB_CM_REP_SENT:
2254	case IB_CM_MRA_REP_RCVD:
2255		ret = cm_alloc_msg(cm_id_priv, &msg);
2256		if (!ret)
2257			cm_format_rej((struct cm_rej_msg *) msg->mad,
2258				      cm_id_priv, reason, ari, ari_length,
2259				      private_data, private_data_len);
2260
2261		cm_enter_timewait(cm_id_priv);
2262		break;
2263	default:
2264		ret = -EINVAL;
2265		goto out;
2266	}
2267
2268	if (ret)
2269		goto out;
2270
2271	ret = ib_post_send_mad(msg, NULL);
2272	if (ret)
2273		cm_free_msg(msg);
2274
2275out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2276	return ret;
2277}
2278EXPORT_SYMBOL(ib_send_cm_rej);
2279
2280static void cm_format_rej_event(struct cm_work *work)
2281{
2282	struct cm_rej_msg *rej_msg;
2283	struct ib_cm_rej_event_param *param;
2284
2285	rej_msg = (struct cm_rej_msg *)work->mad_recv_wc->recv_buf.mad;
2286	param = &work->cm_event.param.rej_rcvd;
2287	param->ari = rej_msg->ari;
2288	param->ari_length = cm_rej_get_reject_info_len(rej_msg);
2289	param->reason = __be16_to_cpu(rej_msg->reason);
2290	work->cm_event.private_data = &rej_msg->private_data;
2291}
2292
2293static struct cm_id_private * cm_acquire_rejected_id(struct cm_rej_msg *rej_msg)
2294{
2295	struct cm_timewait_info *timewait_info;
2296	struct cm_id_private *cm_id_priv;
2297	__be32 remote_id;
2298
2299	remote_id = rej_msg->local_comm_id;
2300
2301	if (__be16_to_cpu(rej_msg->reason) == IB_CM_REJ_TIMEOUT) {
2302		spin_lock_irq(&cm.lock);
2303		timewait_info = cm_find_remote_id( *((__be64 *) rej_msg->ari),
2304						  remote_id);
2305		if (!timewait_info) {
2306			spin_unlock_irq(&cm.lock);
2307			return NULL;
2308		}
2309		cm_id_priv = idr_find(&cm.local_id_table, (__force int)
2310				      (timewait_info->work.local_id ^
2311				       cm.random_id_operand));
2312		if (cm_id_priv) {
2313			if (cm_id_priv->id.remote_id == remote_id)
2314				atomic_inc(&cm_id_priv->refcount);
2315			else
2316				cm_id_priv = NULL;
2317		}
2318		spin_unlock_irq(&cm.lock);
2319	} else if (cm_rej_get_msg_rejected(rej_msg) == CM_MSG_RESPONSE_REQ)
2320		cm_id_priv = cm_acquire_id(rej_msg->remote_comm_id, 0);
2321	else
2322		cm_id_priv = cm_acquire_id(rej_msg->remote_comm_id, remote_id);
2323
2324	return cm_id_priv;
2325}
2326
2327static int cm_rej_handler(struct cm_work *work)
2328{
2329	struct cm_id_private *cm_id_priv;
2330	struct cm_rej_msg *rej_msg;
2331	int ret;
2332
2333	rej_msg = (struct cm_rej_msg *)work->mad_recv_wc->recv_buf.mad;
2334	cm_id_priv = cm_acquire_rejected_id(rej_msg);
2335	if (!cm_id_priv)
2336		return -EINVAL;
2337
2338	cm_format_rej_event(work);
2339
2340	spin_lock_irq(&cm_id_priv->lock);
2341	switch (cm_id_priv->id.state) {
2342	case IB_CM_REQ_SENT:
2343	case IB_CM_MRA_REQ_RCVD:
2344	case IB_CM_REP_SENT:
2345	case IB_CM_MRA_REP_RCVD:
2346		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2347		/* fall through */
2348	case IB_CM_REQ_RCVD:
2349	case IB_CM_MRA_REQ_SENT:
2350		if (__be16_to_cpu(rej_msg->reason) == IB_CM_REJ_STALE_CONN)
2351			cm_enter_timewait(cm_id_priv);
2352		else
2353			cm_reset_to_idle(cm_id_priv);
2354		break;
2355	case IB_CM_DREQ_SENT:
2356		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2357		/* fall through */
2358	case IB_CM_REP_RCVD:
2359	case IB_CM_MRA_REP_SENT:
2360		cm_enter_timewait(cm_id_priv);
2361		break;
2362	case IB_CM_ESTABLISHED:
2363		if (cm_id_priv->id.lap_state == IB_CM_LAP_UNINIT ||
2364		    cm_id_priv->id.lap_state == IB_CM_LAP_SENT) {
2365			if (cm_id_priv->id.lap_state == IB_CM_LAP_SENT)
2366				ib_cancel_mad(cm_id_priv->av.port->mad_agent,
2367					      cm_id_priv->msg);
2368			cm_enter_timewait(cm_id_priv);
2369			break;
2370		}
2371		/* fall through */
2372	default:
2373		spin_unlock_irq(&cm_id_priv->lock);
2374		ret = -EINVAL;
2375		goto out;
2376	}
2377
2378	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2379	if (!ret)
2380		list_add_tail(&work->list, &cm_id_priv->work_list);
2381	spin_unlock_irq(&cm_id_priv->lock);
2382
2383	if (ret)
2384		cm_process_work(cm_id_priv, work);
2385	else
2386		cm_deref_id(cm_id_priv);
2387	return 0;
2388out:
2389	cm_deref_id(cm_id_priv);
2390	return -EINVAL;
2391}
2392
2393int ib_send_cm_mra(struct ib_cm_id *cm_id,
2394		   u8 service_timeout,
2395		   const void *private_data,
2396		   u8 private_data_len)
2397{
2398	struct cm_id_private *cm_id_priv;
2399	struct ib_mad_send_buf *msg;
2400	enum ib_cm_state cm_state;
2401	enum ib_cm_lap_state lap_state;
2402	enum cm_msg_response msg_response;
2403	void *data;
2404	unsigned long flags;
2405	int ret;
2406
2407	if (private_data && private_data_len > IB_CM_MRA_PRIVATE_DATA_SIZE)
2408		return -EINVAL;
2409
2410	data = cm_copy_private_data(private_data, private_data_len);
2411	if (IS_ERR(data))
2412		return PTR_ERR(data);
2413
2414	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2415
2416	spin_lock_irqsave(&cm_id_priv->lock, flags);
2417	switch(cm_id_priv->id.state) {
2418	case IB_CM_REQ_RCVD:
2419		cm_state = IB_CM_MRA_REQ_SENT;
2420		lap_state = cm_id->lap_state;
2421		msg_response = CM_MSG_RESPONSE_REQ;
2422		break;
2423	case IB_CM_REP_RCVD:
2424		cm_state = IB_CM_MRA_REP_SENT;
2425		lap_state = cm_id->lap_state;
2426		msg_response = CM_MSG_RESPONSE_REP;
2427		break;
2428	case IB_CM_ESTABLISHED:
2429		if (cm_id->lap_state == IB_CM_LAP_RCVD) {
2430			cm_state = cm_id->state;
2431			lap_state = IB_CM_MRA_LAP_SENT;
2432			msg_response = CM_MSG_RESPONSE_OTHER;
2433			break;
2434		}
2435	default:
2436		ret = -EINVAL;
2437		goto error1;
2438	}
2439
2440	if (!(service_timeout & IB_CM_MRA_FLAG_DELAY)) {
2441		ret = cm_alloc_msg(cm_id_priv, &msg);
2442		if (ret)
2443			goto error1;
2444
2445		cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
2446			      msg_response, service_timeout,
2447			      private_data, private_data_len);
2448		ret = ib_post_send_mad(msg, NULL);
2449		if (ret)
2450			goto error2;
2451	}
2452
2453	cm_id->state = cm_state;
2454	cm_id->lap_state = lap_state;
2455	cm_id_priv->service_timeout = service_timeout;
2456	cm_set_private_data(cm_id_priv, data, private_data_len);
2457	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2458	return 0;
2459
2460error1:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2461	kfree(data);
2462	return ret;
2463
2464error2:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2465	kfree(data);
2466	cm_free_msg(msg);
2467	return ret;
2468}
2469EXPORT_SYMBOL(ib_send_cm_mra);
2470
2471static struct cm_id_private * cm_acquire_mraed_id(struct cm_mra_msg *mra_msg)
2472{
2473	switch (cm_mra_get_msg_mraed(mra_msg)) {
2474	case CM_MSG_RESPONSE_REQ:
2475		return cm_acquire_id(mra_msg->remote_comm_id, 0);
2476	case CM_MSG_RESPONSE_REP:
2477	case CM_MSG_RESPONSE_OTHER:
2478		return cm_acquire_id(mra_msg->remote_comm_id,
2479				     mra_msg->local_comm_id);
2480	default:
2481		return NULL;
2482	}
2483}
2484
2485static int cm_mra_handler(struct cm_work *work)
2486{
2487	struct cm_id_private *cm_id_priv;
2488	struct cm_mra_msg *mra_msg;
2489	int timeout, ret;
2490
2491	mra_msg = (struct cm_mra_msg *)work->mad_recv_wc->recv_buf.mad;
2492	cm_id_priv = cm_acquire_mraed_id(mra_msg);
2493	if (!cm_id_priv)
2494		return -EINVAL;
2495
2496	work->cm_event.private_data = &mra_msg->private_data;
2497	work->cm_event.param.mra_rcvd.service_timeout =
2498					cm_mra_get_service_timeout(mra_msg);
2499	timeout = cm_convert_to_ms(cm_mra_get_service_timeout(mra_msg)) +
2500		  cm_convert_to_ms(cm_id_priv->av.timeout);
2501
2502	spin_lock_irq(&cm_id_priv->lock);
2503	switch (cm_id_priv->id.state) {
2504	case IB_CM_REQ_SENT:
2505		if (cm_mra_get_msg_mraed(mra_msg) != CM_MSG_RESPONSE_REQ ||
2506		    ib_modify_mad(cm_id_priv->av.port->mad_agent,
2507				  cm_id_priv->msg, timeout))
2508			goto out;
2509		cm_id_priv->id.state = IB_CM_MRA_REQ_RCVD;
2510		break;
2511	case IB_CM_REP_SENT:
2512		if (cm_mra_get_msg_mraed(mra_msg) != CM_MSG_RESPONSE_REP ||
2513		    ib_modify_mad(cm_id_priv->av.port->mad_agent,
2514				  cm_id_priv->msg, timeout))
2515			goto out;
2516		cm_id_priv->id.state = IB_CM_MRA_REP_RCVD;
2517		break;
2518	case IB_CM_ESTABLISHED:
2519		if (cm_mra_get_msg_mraed(mra_msg) != CM_MSG_RESPONSE_OTHER ||
2520		    cm_id_priv->id.lap_state != IB_CM_LAP_SENT ||
2521		    ib_modify_mad(cm_id_priv->av.port->mad_agent,
2522				  cm_id_priv->msg, timeout)) {
2523			if (cm_id_priv->id.lap_state == IB_CM_MRA_LAP_RCVD)
2524				atomic_long_inc(&work->port->
2525						counter_group[CM_RECV_DUPLICATES].
2526						counter[CM_MRA_COUNTER]);
2527			goto out;
2528		}
2529		cm_id_priv->id.lap_state = IB_CM_MRA_LAP_RCVD;
2530		break;
2531	case IB_CM_MRA_REQ_RCVD:
2532	case IB_CM_MRA_REP_RCVD:
2533		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2534				counter[CM_MRA_COUNTER]);
2535		/* fall through */
2536	default:
2537		goto out;
2538	}
2539
2540	cm_id_priv->msg->context[1] = (void *) (unsigned long)
2541				      cm_id_priv->id.state;
2542	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2543	if (!ret)
2544		list_add_tail(&work->list, &cm_id_priv->work_list);
2545	spin_unlock_irq(&cm_id_priv->lock);
2546
2547	if (ret)
2548		cm_process_work(cm_id_priv, work);
2549	else
2550		cm_deref_id(cm_id_priv);
2551	return 0;
2552out:
2553	spin_unlock_irq(&cm_id_priv->lock);
2554	cm_deref_id(cm_id_priv);
2555	return -EINVAL;
2556}
2557
2558static void cm_format_lap(struct cm_lap_msg *lap_msg,
2559			  struct cm_id_private *cm_id_priv,
2560			  struct ib_sa_path_rec *alternate_path,
2561			  const void *private_data,
2562			  u8 private_data_len)
2563{
2564	cm_format_mad_hdr(&lap_msg->hdr, CM_LAP_ATTR_ID,
2565			  cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_LAP));
2566	lap_msg->local_comm_id = cm_id_priv->id.local_id;
2567	lap_msg->remote_comm_id = cm_id_priv->id.remote_id;
2568	cm_lap_set_remote_qpn(lap_msg, cm_id_priv->remote_qpn);
2569	/* todo: need remote CM response timeout */
2570	cm_lap_set_remote_resp_timeout(lap_msg, 0x1F);
2571	lap_msg->alt_local_lid = alternate_path->slid;
2572	lap_msg->alt_remote_lid = alternate_path->dlid;
2573	lap_msg->alt_local_gid = alternate_path->sgid;
2574	lap_msg->alt_remote_gid = alternate_path->dgid;
2575	cm_lap_set_flow_label(lap_msg, alternate_path->flow_label);
2576	cm_lap_set_traffic_class(lap_msg, alternate_path->traffic_class);
2577	lap_msg->alt_hop_limit = alternate_path->hop_limit;
2578	cm_lap_set_packet_rate(lap_msg, alternate_path->rate);
2579	cm_lap_set_sl(lap_msg, alternate_path->sl);
2580	cm_lap_set_subnet_local(lap_msg, 1); /* local only... */
2581	cm_lap_set_local_ack_timeout(lap_msg,
2582		cm_ack_timeout(cm_id_priv->av.port->cm_dev->ack_delay,
2583			       alternate_path->packet_life_time));
2584
2585	if (private_data && private_data_len)
2586		memcpy(lap_msg->private_data, private_data, private_data_len);
2587}
2588
2589int ib_send_cm_lap(struct ib_cm_id *cm_id,
2590		   struct ib_sa_path_rec *alternate_path,
2591		   const void *private_data,
2592		   u8 private_data_len)
2593{
2594	struct cm_id_private *cm_id_priv;
2595	struct ib_mad_send_buf *msg;
2596	unsigned long flags;
2597	int ret;
2598
2599	if (private_data && private_data_len > IB_CM_LAP_PRIVATE_DATA_SIZE)
2600		return -EINVAL;
2601
2602	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2603	spin_lock_irqsave(&cm_id_priv->lock, flags);
2604	if (cm_id->state != IB_CM_ESTABLISHED ||
2605	    (cm_id->lap_state != IB_CM_LAP_UNINIT &&
2606	     cm_id->lap_state != IB_CM_LAP_IDLE)) {
2607		ret = -EINVAL;
2608		goto out;
2609	}
2610
2611	ret = cm_init_av_by_path(alternate_path, &cm_id_priv->alt_av);
2612	if (ret)
2613		goto out;
2614	cm_id_priv->alt_av.timeout =
2615			cm_ack_timeout(cm_id_priv->target_ack_delay,
2616				       cm_id_priv->alt_av.timeout - 1);
2617
2618	ret = cm_alloc_msg(cm_id_priv, &msg);
2619	if (ret)
2620		goto out;
2621
2622	cm_format_lap((struct cm_lap_msg *) msg->mad, cm_id_priv,
2623		      alternate_path, private_data, private_data_len);
2624	msg->timeout_ms = cm_id_priv->timeout_ms;
2625	msg->context[1] = (void *) (unsigned long) IB_CM_ESTABLISHED;
2626
2627	ret = ib_post_send_mad(msg, NULL);
2628	if (ret) {
2629		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2630		cm_free_msg(msg);
2631		return ret;
2632	}
2633
2634	cm_id->lap_state = IB_CM_LAP_SENT;
2635	cm_id_priv->msg = msg;
2636
2637out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2638	return ret;
2639}
2640EXPORT_SYMBOL(ib_send_cm_lap);
2641
2642static void cm_format_path_from_lap(struct cm_id_private *cm_id_priv,
2643				    struct ib_sa_path_rec *path,
2644				    struct cm_lap_msg *lap_msg)
2645{
2646	memset(path, 0, sizeof *path);
2647	path->dgid = lap_msg->alt_local_gid;
2648	path->sgid = lap_msg->alt_remote_gid;
2649	path->dlid = lap_msg->alt_local_lid;
2650	path->slid = lap_msg->alt_remote_lid;
2651	path->flow_label = cm_lap_get_flow_label(lap_msg);
2652	path->hop_limit = lap_msg->alt_hop_limit;
2653	path->traffic_class = cm_lap_get_traffic_class(lap_msg);
2654	path->reversible = 1;
2655	path->pkey = cm_id_priv->pkey;
2656	path->sl = cm_lap_get_sl(lap_msg);
2657	path->mtu_selector = IB_SA_EQ;
2658	path->mtu = cm_id_priv->path_mtu;
2659	path->rate_selector = IB_SA_EQ;
2660	path->rate = cm_lap_get_packet_rate(lap_msg);
2661	path->packet_life_time_selector = IB_SA_EQ;
2662	path->packet_life_time = cm_lap_get_local_ack_timeout(lap_msg);
2663	path->packet_life_time -= (path->packet_life_time > 0);
2664}
2665
2666static int cm_lap_handler(struct cm_work *work)
2667{
2668	struct cm_id_private *cm_id_priv;
2669	struct cm_lap_msg *lap_msg;
2670	struct ib_cm_lap_event_param *param;
2671	struct ib_mad_send_buf *msg = NULL;
2672	int ret;
2673
2674	/* todo: verify LAP request and send reject APR if invalid. */
2675	lap_msg = (struct cm_lap_msg *)work->mad_recv_wc->recv_buf.mad;
2676	cm_id_priv = cm_acquire_id(lap_msg->remote_comm_id,
2677				   lap_msg->local_comm_id);
2678	if (!cm_id_priv)
2679		return -EINVAL;
2680
2681	param = &work->cm_event.param.lap_rcvd;
2682	param->alternate_path = &work->path[0];
2683	cm_format_path_from_lap(cm_id_priv, param->alternate_path, lap_msg);
2684	work->cm_event.private_data = &lap_msg->private_data;
2685
2686	spin_lock_irq(&cm_id_priv->lock);
2687	if (cm_id_priv->id.state != IB_CM_ESTABLISHED)
2688		goto unlock;
2689
2690	switch (cm_id_priv->id.lap_state) {
2691	case IB_CM_LAP_UNINIT:
2692	case IB_CM_LAP_IDLE:
2693		break;
2694	case IB_CM_MRA_LAP_SENT:
2695		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2696				counter[CM_LAP_COUNTER]);
2697		if (cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg))
2698			goto unlock;
2699
2700		cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
2701			      CM_MSG_RESPONSE_OTHER,
2702			      cm_id_priv->service_timeout,
2703			      cm_id_priv->private_data,
2704			      cm_id_priv->private_data_len);
2705		spin_unlock_irq(&cm_id_priv->lock);
2706
2707		if (ib_post_send_mad(msg, NULL))
2708			cm_free_msg(msg);
2709		goto deref;
2710	case IB_CM_LAP_RCVD:
2711		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2712				counter[CM_LAP_COUNTER]);
2713		goto unlock;
2714	default:
2715		goto unlock;
2716	}
2717
2718	cm_id_priv->id.lap_state = IB_CM_LAP_RCVD;
2719	cm_id_priv->tid = lap_msg->hdr.tid;
2720	cm_init_av_for_response(work->port, work->mad_recv_wc->wc,
2721				work->mad_recv_wc->recv_buf.grh,
2722				&cm_id_priv->av);
2723	cm_init_av_by_path(param->alternate_path, &cm_id_priv->alt_av);
2724	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2725	if (!ret)
2726		list_add_tail(&work->list, &cm_id_priv->work_list);
2727	spin_unlock_irq(&cm_id_priv->lock);
2728
2729	if (ret)
2730		cm_process_work(cm_id_priv, work);
2731	else
2732		cm_deref_id(cm_id_priv);
2733	return 0;
2734
2735unlock:	spin_unlock_irq(&cm_id_priv->lock);
2736deref:	cm_deref_id(cm_id_priv);
2737	return -EINVAL;
2738}
2739
2740static void cm_format_apr(struct cm_apr_msg *apr_msg,
2741			  struct cm_id_private *cm_id_priv,
2742			  enum ib_cm_apr_status status,
2743			  void *info,
2744			  u8 info_length,
2745			  const void *private_data,
2746			  u8 private_data_len)
2747{
2748	cm_format_mad_hdr(&apr_msg->hdr, CM_APR_ATTR_ID, cm_id_priv->tid);
2749	apr_msg->local_comm_id = cm_id_priv->id.local_id;
2750	apr_msg->remote_comm_id = cm_id_priv->id.remote_id;
2751	apr_msg->ap_status = (u8) status;
2752
2753	if (info && info_length) {
2754		apr_msg->info_length = info_length;
2755		memcpy(apr_msg->info, info, info_length);
2756	}
2757
2758	if (private_data && private_data_len)
2759		memcpy(apr_msg->private_data, private_data, private_data_len);
2760}
2761
2762int ib_send_cm_apr(struct ib_cm_id *cm_id,
2763		   enum ib_cm_apr_status status,
2764		   void *info,
2765		   u8 info_length,
2766		   const void *private_data,
2767		   u8 private_data_len)
2768{
2769	struct cm_id_private *cm_id_priv;
2770	struct ib_mad_send_buf *msg;
2771	unsigned long flags;
2772	int ret;
2773
2774	if ((private_data && private_data_len > IB_CM_APR_PRIVATE_DATA_SIZE) ||
2775	    (info && info_length > IB_CM_APR_INFO_LENGTH))
2776		return -EINVAL;
2777
2778	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2779	spin_lock_irqsave(&cm_id_priv->lock, flags);
2780	if (cm_id->state != IB_CM_ESTABLISHED ||
2781	    (cm_id->lap_state != IB_CM_LAP_RCVD &&
2782	     cm_id->lap_state != IB_CM_MRA_LAP_SENT)) {
2783		ret = -EINVAL;
2784		goto out;
2785	}
2786
2787	ret = cm_alloc_msg(cm_id_priv, &msg);
2788	if (ret)
2789		goto out;
2790
2791	cm_format_apr((struct cm_apr_msg *) msg->mad, cm_id_priv, status,
2792		      info, info_length, private_data, private_data_len);
2793	ret = ib_post_send_mad(msg, NULL);
2794	if (ret) {
2795		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2796		cm_free_msg(msg);
2797		return ret;
2798	}
2799
2800	cm_id->lap_state = IB_CM_LAP_IDLE;
2801out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2802	return ret;
2803}
2804EXPORT_SYMBOL(ib_send_cm_apr);
2805
2806static int cm_apr_handler(struct cm_work *work)
2807{
2808	struct cm_id_private *cm_id_priv;
2809	struct cm_apr_msg *apr_msg;
2810	int ret;
2811
2812	apr_msg = (struct cm_apr_msg *)work->mad_recv_wc->recv_buf.mad;
2813	cm_id_priv = cm_acquire_id(apr_msg->remote_comm_id,
2814				   apr_msg->local_comm_id);
2815	if (!cm_id_priv)
2816		return -EINVAL; /* Unmatched reply. */
2817
2818	work->cm_event.param.apr_rcvd.ap_status = apr_msg->ap_status;
2819	work->cm_event.param.apr_rcvd.apr_info = &apr_msg->info;
2820	work->cm_event.param.apr_rcvd.info_len = apr_msg->info_length;
2821	work->cm_event.private_data = &apr_msg->private_data;
2822
2823	spin_lock_irq(&cm_id_priv->lock);
2824	if (cm_id_priv->id.state != IB_CM_ESTABLISHED ||
2825	    (cm_id_priv->id.lap_state != IB_CM_LAP_SENT &&
2826	     cm_id_priv->id.lap_state != IB_CM_MRA_LAP_RCVD)) {
2827		spin_unlock_irq(&cm_id_priv->lock);
2828		goto out;
2829	}
2830	cm_id_priv->id.lap_state = IB_CM_LAP_IDLE;
2831	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2832	cm_id_priv->msg = NULL;
2833
2834	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2835	if (!ret)
2836		list_add_tail(&work->list, &cm_id_priv->work_list);
2837	spin_unlock_irq(&cm_id_priv->lock);
2838
2839	if (ret)
2840		cm_process_work(cm_id_priv, work);
2841	else
2842		cm_deref_id(cm_id_priv);
2843	return 0;
2844out:
2845	cm_deref_id(cm_id_priv);
2846	return -EINVAL;
2847}
2848
2849static int cm_timewait_handler(struct cm_work *work)
2850{
2851	struct cm_timewait_info *timewait_info;
2852	struct cm_id_private *cm_id_priv;
2853	int ret;
2854
2855	timewait_info = (struct cm_timewait_info *)work;
2856	spin_lock_irq(&cm.lock);
2857	list_del(&timewait_info->list);
2858	spin_unlock_irq(&cm.lock);
2859
2860	cm_id_priv = cm_acquire_id(timewait_info->work.local_id,
2861				   timewait_info->work.remote_id);
2862	if (!cm_id_priv)
2863		return -EINVAL;
2864
2865	spin_lock_irq(&cm_id_priv->lock);
2866	if (cm_id_priv->id.state != IB_CM_TIMEWAIT ||
2867	    cm_id_priv->remote_qpn != timewait_info->remote_qpn) {
2868		spin_unlock_irq(&cm_id_priv->lock);
2869		goto out;
2870	}
2871	cm_id_priv->id.state = IB_CM_IDLE;
2872	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2873	if (!ret)
2874		list_add_tail(&work->list, &cm_id_priv->work_list);
2875	spin_unlock_irq(&cm_id_priv->lock);
2876
2877	if (ret)
2878		cm_process_work(cm_id_priv, work);
2879	else
2880		cm_deref_id(cm_id_priv);
2881	return 0;
2882out:
2883	cm_deref_id(cm_id_priv);
2884	return -EINVAL;
2885}
2886
2887static void cm_format_sidr_req(struct cm_sidr_req_msg *sidr_req_msg,
2888			       struct cm_id_private *cm_id_priv,
2889			       struct ib_cm_sidr_req_param *param)
2890{
2891	cm_format_mad_hdr(&sidr_req_msg->hdr, CM_SIDR_REQ_ATTR_ID,
2892			  cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_SIDR));
2893	sidr_req_msg->request_id = cm_id_priv->id.local_id;
2894	sidr_req_msg->pkey = param->path->pkey;
2895	sidr_req_msg->service_id = param->service_id;
2896
2897	if (param->private_data && param->private_data_len)
2898		memcpy(sidr_req_msg->private_data, param->private_data,
2899		       param->private_data_len);
2900}
2901
2902int ib_send_cm_sidr_req(struct ib_cm_id *cm_id,
2903			struct ib_cm_sidr_req_param *param)
2904{
2905	struct cm_id_private *cm_id_priv;
2906	struct ib_mad_send_buf *msg;
2907	unsigned long flags;
2908	int ret;
2909
2910	if (!param->path || (param->private_data &&
2911	     param->private_data_len > IB_CM_SIDR_REQ_PRIVATE_DATA_SIZE))
2912		return -EINVAL;
2913
2914	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2915	ret = cm_init_av_by_path(param->path, &cm_id_priv->av);
2916	if (ret)
2917		goto out;
2918
2919	cm_id->service_id = param->service_id;
2920	cm_id->service_mask = ~cpu_to_be64(0);
2921	cm_id_priv->timeout_ms = param->timeout_ms;
2922	cm_id_priv->max_cm_retries = param->max_cm_retries;
2923	ret = cm_alloc_msg(cm_id_priv, &msg);
2924	if (ret)
2925		goto out;
2926
2927	cm_format_sidr_req((struct cm_sidr_req_msg *) msg->mad, cm_id_priv,
2928			   param);
2929	msg->timeout_ms = cm_id_priv->timeout_ms;
2930	msg->context[1] = (void *) (unsigned long) IB_CM_SIDR_REQ_SENT;
2931
2932	spin_lock_irqsave(&cm_id_priv->lock, flags);
2933	if (cm_id->state == IB_CM_IDLE)
2934		ret = ib_post_send_mad(msg, NULL);
2935	else
2936		ret = -EINVAL;
2937
2938	if (ret) {
2939		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2940		cm_free_msg(msg);
2941		goto out;
2942	}
2943	cm_id->state = IB_CM_SIDR_REQ_SENT;
2944	cm_id_priv->msg = msg;
2945	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2946out:
2947	return ret;
2948}
2949EXPORT_SYMBOL(ib_send_cm_sidr_req);
2950
2951static void cm_format_sidr_req_event(struct cm_work *work,
2952				     struct ib_cm_id *listen_id)
2953{
2954	struct cm_sidr_req_msg *sidr_req_msg;
2955	struct ib_cm_sidr_req_event_param *param;
2956
2957	sidr_req_msg = (struct cm_sidr_req_msg *)
2958				work->mad_recv_wc->recv_buf.mad;
2959	param = &work->cm_event.param.sidr_req_rcvd;
2960	param->pkey = __be16_to_cpu(sidr_req_msg->pkey);
2961	param->listen_id = listen_id;
 
 
2962	param->port = work->port->port_num;
2963	work->cm_event.private_data = &sidr_req_msg->private_data;
2964}
2965
2966static int cm_sidr_req_handler(struct cm_work *work)
2967{
2968	struct ib_cm_id *cm_id;
2969	struct cm_id_private *cm_id_priv, *cur_cm_id_priv;
2970	struct cm_sidr_req_msg *sidr_req_msg;
2971	struct ib_wc *wc;
2972
2973	cm_id = ib_create_cm_id(work->port->cm_dev->ib_device, NULL, NULL);
2974	if (IS_ERR(cm_id))
2975		return PTR_ERR(cm_id);
2976	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2977
2978	/* Record SGID/SLID and request ID for lookup. */
2979	sidr_req_msg = (struct cm_sidr_req_msg *)
2980				work->mad_recv_wc->recv_buf.mad;
2981	wc = work->mad_recv_wc->wc;
2982	cm_id_priv->av.dgid.global.subnet_prefix = cpu_to_be64(wc->slid);
2983	cm_id_priv->av.dgid.global.interface_id = 0;
2984	cm_init_av_for_response(work->port, work->mad_recv_wc->wc,
2985				work->mad_recv_wc->recv_buf.grh,
2986				&cm_id_priv->av);
2987	cm_id_priv->id.remote_id = sidr_req_msg->request_id;
2988	cm_id_priv->tid = sidr_req_msg->hdr.tid;
2989	atomic_inc(&cm_id_priv->work_count);
2990
2991	spin_lock_irq(&cm.lock);
2992	cur_cm_id_priv = cm_insert_remote_sidr(cm_id_priv);
2993	if (cur_cm_id_priv) {
2994		spin_unlock_irq(&cm.lock);
2995		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2996				counter[CM_SIDR_REQ_COUNTER]);
2997		goto out; /* Duplicate message. */
2998	}
2999	cm_id_priv->id.state = IB_CM_SIDR_REQ_RCVD;
3000	cur_cm_id_priv = cm_find_listen(cm_id->device,
3001					sidr_req_msg->service_id,
3002					sidr_req_msg->private_data);
3003	if (!cur_cm_id_priv) {
3004		spin_unlock_irq(&cm.lock);
3005		cm_reject_sidr_req(cm_id_priv, IB_SIDR_UNSUPPORTED);
3006		goto out; /* No match. */
3007	}
3008	atomic_inc(&cur_cm_id_priv->refcount);
3009	atomic_inc(&cm_id_priv->refcount);
3010	spin_unlock_irq(&cm.lock);
3011
3012	cm_id_priv->id.cm_handler = cur_cm_id_priv->id.cm_handler;
3013	cm_id_priv->id.context = cur_cm_id_priv->id.context;
3014	cm_id_priv->id.service_id = sidr_req_msg->service_id;
3015	cm_id_priv->id.service_mask = ~cpu_to_be64(0);
3016
3017	cm_format_sidr_req_event(work, &cur_cm_id_priv->id);
3018	cm_process_work(cm_id_priv, work);
3019	cm_deref_id(cur_cm_id_priv);
3020	return 0;
3021out:
3022	ib_destroy_cm_id(&cm_id_priv->id);
3023	return -EINVAL;
3024}
3025
3026static void cm_format_sidr_rep(struct cm_sidr_rep_msg *sidr_rep_msg,
3027			       struct cm_id_private *cm_id_priv,
3028			       struct ib_cm_sidr_rep_param *param)
3029{
3030	cm_format_mad_hdr(&sidr_rep_msg->hdr, CM_SIDR_REP_ATTR_ID,
3031			  cm_id_priv->tid);
3032	sidr_rep_msg->request_id = cm_id_priv->id.remote_id;
3033	sidr_rep_msg->status = param->status;
3034	cm_sidr_rep_set_qpn(sidr_rep_msg, cpu_to_be32(param->qp_num));
3035	sidr_rep_msg->service_id = cm_id_priv->id.service_id;
3036	sidr_rep_msg->qkey = cpu_to_be32(param->qkey);
3037
3038	if (param->info && param->info_length)
3039		memcpy(sidr_rep_msg->info, param->info, param->info_length);
3040
3041	if (param->private_data && param->private_data_len)
3042		memcpy(sidr_rep_msg->private_data, param->private_data,
3043		       param->private_data_len);
3044}
3045
3046int ib_send_cm_sidr_rep(struct ib_cm_id *cm_id,
3047			struct ib_cm_sidr_rep_param *param)
3048{
3049	struct cm_id_private *cm_id_priv;
3050	struct ib_mad_send_buf *msg;
3051	unsigned long flags;
3052	int ret;
3053
3054	if ((param->info && param->info_length > IB_CM_SIDR_REP_INFO_LENGTH) ||
3055	    (param->private_data &&
3056	     param->private_data_len > IB_CM_SIDR_REP_PRIVATE_DATA_SIZE))
3057		return -EINVAL;
3058
3059	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3060	spin_lock_irqsave(&cm_id_priv->lock, flags);
3061	if (cm_id->state != IB_CM_SIDR_REQ_RCVD) {
3062		ret = -EINVAL;
3063		goto error;
3064	}
3065
3066	ret = cm_alloc_msg(cm_id_priv, &msg);
3067	if (ret)
3068		goto error;
3069
3070	cm_format_sidr_rep((struct cm_sidr_rep_msg *) msg->mad, cm_id_priv,
3071			   param);
3072	ret = ib_post_send_mad(msg, NULL);
3073	if (ret) {
3074		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3075		cm_free_msg(msg);
3076		return ret;
3077	}
3078	cm_id->state = IB_CM_IDLE;
3079	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3080
3081	spin_lock_irqsave(&cm.lock, flags);
3082	rb_erase(&cm_id_priv->sidr_id_node, &cm.remote_sidr_table);
 
 
 
3083	spin_unlock_irqrestore(&cm.lock, flags);
3084	return 0;
3085
3086error:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3087	return ret;
3088}
3089EXPORT_SYMBOL(ib_send_cm_sidr_rep);
3090
3091static void cm_format_sidr_rep_event(struct cm_work *work)
3092{
3093	struct cm_sidr_rep_msg *sidr_rep_msg;
3094	struct ib_cm_sidr_rep_event_param *param;
3095
3096	sidr_rep_msg = (struct cm_sidr_rep_msg *)
3097				work->mad_recv_wc->recv_buf.mad;
3098	param = &work->cm_event.param.sidr_rep_rcvd;
3099	param->status = sidr_rep_msg->status;
3100	param->qkey = be32_to_cpu(sidr_rep_msg->qkey);
3101	param->qpn = be32_to_cpu(cm_sidr_rep_get_qpn(sidr_rep_msg));
3102	param->info = &sidr_rep_msg->info;
3103	param->info_len = sidr_rep_msg->info_length;
3104	work->cm_event.private_data = &sidr_rep_msg->private_data;
3105}
3106
3107static int cm_sidr_rep_handler(struct cm_work *work)
3108{
3109	struct cm_sidr_rep_msg *sidr_rep_msg;
3110	struct cm_id_private *cm_id_priv;
3111
3112	sidr_rep_msg = (struct cm_sidr_rep_msg *)
3113				work->mad_recv_wc->recv_buf.mad;
3114	cm_id_priv = cm_acquire_id(sidr_rep_msg->request_id, 0);
3115	if (!cm_id_priv)
3116		return -EINVAL; /* Unmatched reply. */
3117
3118	spin_lock_irq(&cm_id_priv->lock);
3119	if (cm_id_priv->id.state != IB_CM_SIDR_REQ_SENT) {
3120		spin_unlock_irq(&cm_id_priv->lock);
3121		goto out;
3122	}
3123	cm_id_priv->id.state = IB_CM_IDLE;
3124	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
3125	spin_unlock_irq(&cm_id_priv->lock);
3126
3127	cm_format_sidr_rep_event(work);
3128	cm_process_work(cm_id_priv, work);
3129	return 0;
3130out:
3131	cm_deref_id(cm_id_priv);
3132	return -EINVAL;
3133}
3134
3135static void cm_process_send_error(struct ib_mad_send_buf *msg,
3136				  enum ib_wc_status wc_status)
3137{
3138	struct cm_id_private *cm_id_priv;
3139	struct ib_cm_event cm_event;
3140	enum ib_cm_state state;
3141	int ret;
3142
3143	memset(&cm_event, 0, sizeof cm_event);
3144	cm_id_priv = msg->context[0];
3145
3146	/* Discard old sends or ones without a response. */
3147	spin_lock_irq(&cm_id_priv->lock);
3148	state = (enum ib_cm_state) (unsigned long) msg->context[1];
3149	if (msg != cm_id_priv->msg || state != cm_id_priv->id.state)
3150		goto discard;
3151
3152	switch (state) {
3153	case IB_CM_REQ_SENT:
3154	case IB_CM_MRA_REQ_RCVD:
3155		cm_reset_to_idle(cm_id_priv);
3156		cm_event.event = IB_CM_REQ_ERROR;
3157		break;
3158	case IB_CM_REP_SENT:
3159	case IB_CM_MRA_REP_RCVD:
3160		cm_reset_to_idle(cm_id_priv);
3161		cm_event.event = IB_CM_REP_ERROR;
3162		break;
3163	case IB_CM_DREQ_SENT:
3164		cm_enter_timewait(cm_id_priv);
3165		cm_event.event = IB_CM_DREQ_ERROR;
3166		break;
3167	case IB_CM_SIDR_REQ_SENT:
3168		cm_id_priv->id.state = IB_CM_IDLE;
3169		cm_event.event = IB_CM_SIDR_REQ_ERROR;
3170		break;
3171	default:
3172		goto discard;
3173	}
3174	spin_unlock_irq(&cm_id_priv->lock);
3175	cm_event.param.send_status = wc_status;
3176
3177	/* No other events can occur on the cm_id at this point. */
3178	ret = cm_id_priv->id.cm_handler(&cm_id_priv->id, &cm_event);
3179	cm_free_msg(msg);
3180	if (ret)
3181		ib_destroy_cm_id(&cm_id_priv->id);
3182	return;
3183discard:
3184	spin_unlock_irq(&cm_id_priv->lock);
3185	cm_free_msg(msg);
3186}
3187
3188static void cm_send_handler(struct ib_mad_agent *mad_agent,
3189			    struct ib_mad_send_wc *mad_send_wc)
3190{
3191	struct ib_mad_send_buf *msg = mad_send_wc->send_buf;
3192	struct cm_port *port;
3193	u16 attr_index;
3194
3195	port = mad_agent->context;
3196	attr_index = be16_to_cpu(((struct ib_mad_hdr *)
3197				  msg->mad)->attr_id) - CM_ATTR_ID_OFFSET;
3198
3199	/*
3200	 * If the send was in response to a received message (context[0] is not
3201	 * set to a cm_id), and is not a REJ, then it is a send that was
3202	 * manually retried.
3203	 */
3204	if (!msg->context[0] && (attr_index != CM_REJ_COUNTER))
3205		msg->retries = 1;
3206
3207	atomic_long_add(1 + msg->retries,
3208			&port->counter_group[CM_XMIT].counter[attr_index]);
3209	if (msg->retries)
3210		atomic_long_add(msg->retries,
3211				&port->counter_group[CM_XMIT_RETRIES].
3212				counter[attr_index]);
3213
3214	switch (mad_send_wc->status) {
3215	case IB_WC_SUCCESS:
3216	case IB_WC_WR_FLUSH_ERR:
3217		cm_free_msg(msg);
3218		break;
3219	default:
3220		if (msg->context[0] && msg->context[1])
3221			cm_process_send_error(msg, mad_send_wc->status);
3222		else
3223			cm_free_msg(msg);
3224		break;
3225	}
3226}
3227
3228static void cm_work_handler(struct work_struct *_work)
3229{
3230	struct cm_work *work = container_of(_work, struct cm_work, work.work);
3231	int ret;
3232
3233	switch (work->cm_event.event) {
3234	case IB_CM_REQ_RECEIVED:
3235		ret = cm_req_handler(work);
3236		break;
3237	case IB_CM_MRA_RECEIVED:
3238		ret = cm_mra_handler(work);
3239		break;
3240	case IB_CM_REJ_RECEIVED:
3241		ret = cm_rej_handler(work);
3242		break;
3243	case IB_CM_REP_RECEIVED:
3244		ret = cm_rep_handler(work);
3245		break;
3246	case IB_CM_RTU_RECEIVED:
3247		ret = cm_rtu_handler(work);
3248		break;
3249	case IB_CM_USER_ESTABLISHED:
3250		ret = cm_establish_handler(work);
3251		break;
3252	case IB_CM_DREQ_RECEIVED:
3253		ret = cm_dreq_handler(work);
3254		break;
3255	case IB_CM_DREP_RECEIVED:
3256		ret = cm_drep_handler(work);
3257		break;
3258	case IB_CM_SIDR_REQ_RECEIVED:
3259		ret = cm_sidr_req_handler(work);
3260		break;
3261	case IB_CM_SIDR_REP_RECEIVED:
3262		ret = cm_sidr_rep_handler(work);
3263		break;
3264	case IB_CM_LAP_RECEIVED:
3265		ret = cm_lap_handler(work);
3266		break;
3267	case IB_CM_APR_RECEIVED:
3268		ret = cm_apr_handler(work);
3269		break;
3270	case IB_CM_TIMEWAIT_EXIT:
3271		ret = cm_timewait_handler(work);
3272		break;
3273	default:
3274		ret = -EINVAL;
3275		break;
3276	}
3277	if (ret)
3278		cm_free_work(work);
3279}
3280
3281static int cm_establish(struct ib_cm_id *cm_id)
3282{
3283	struct cm_id_private *cm_id_priv;
3284	struct cm_work *work;
3285	unsigned long flags;
3286	int ret = 0;
 
 
 
 
 
3287
3288	work = kmalloc(sizeof *work, GFP_ATOMIC);
3289	if (!work)
3290		return -ENOMEM;
3291
3292	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3293	spin_lock_irqsave(&cm_id_priv->lock, flags);
3294	switch (cm_id->state)
3295	{
3296	case IB_CM_REP_SENT:
3297	case IB_CM_MRA_REP_RCVD:
3298		cm_id->state = IB_CM_ESTABLISHED;
3299		break;
3300	case IB_CM_ESTABLISHED:
3301		ret = -EISCONN;
3302		break;
3303	default:
3304		ret = -EINVAL;
3305		break;
3306	}
3307	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3308
3309	if (ret) {
3310		kfree(work);
3311		goto out;
3312	}
3313
3314	/*
3315	 * The CM worker thread may try to destroy the cm_id before it
3316	 * can execute this work item.  To prevent potential deadlock,
3317	 * we need to find the cm_id once we're in the context of the
3318	 * worker thread, rather than holding a reference on it.
3319	 */
3320	INIT_DELAYED_WORK(&work->work, cm_work_handler);
3321	work->local_id = cm_id->local_id;
3322	work->remote_id = cm_id->remote_id;
3323	work->mad_recv_wc = NULL;
3324	work->cm_event.event = IB_CM_USER_ESTABLISHED;
3325	queue_delayed_work(cm.wq, &work->work, 0);
 
 
 
 
 
 
 
 
 
 
3326out:
3327	return ret;
3328}
3329
3330static int cm_migrate(struct ib_cm_id *cm_id)
3331{
3332	struct cm_id_private *cm_id_priv;
3333	unsigned long flags;
3334	int ret = 0;
3335
3336	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3337	spin_lock_irqsave(&cm_id_priv->lock, flags);
3338	if (cm_id->state == IB_CM_ESTABLISHED &&
3339	    (cm_id->lap_state == IB_CM_LAP_UNINIT ||
3340	     cm_id->lap_state == IB_CM_LAP_IDLE)) {
3341		cm_id->lap_state = IB_CM_LAP_IDLE;
3342		cm_id_priv->av = cm_id_priv->alt_av;
3343	} else
3344		ret = -EINVAL;
3345	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3346
3347	return ret;
3348}
3349
3350int ib_cm_notify(struct ib_cm_id *cm_id, enum ib_event_type event)
3351{
3352	int ret;
3353
3354	switch (event) {
3355	case IB_EVENT_COMM_EST:
3356		ret = cm_establish(cm_id);
3357		break;
3358	case IB_EVENT_PATH_MIG:
3359		ret = cm_migrate(cm_id);
3360		break;
3361	default:
3362		ret = -EINVAL;
3363	}
3364	return ret;
3365}
3366EXPORT_SYMBOL(ib_cm_notify);
3367
3368static void cm_recv_handler(struct ib_mad_agent *mad_agent,
 
3369			    struct ib_mad_recv_wc *mad_recv_wc)
3370{
3371	struct cm_port *port = mad_agent->context;
3372	struct cm_work *work;
3373	enum ib_cm_event_type event;
3374	u16 attr_id;
3375	int paths = 0;
 
3376
3377	switch (mad_recv_wc->recv_buf.mad->mad_hdr.attr_id) {
3378	case CM_REQ_ATTR_ID:
3379		paths = 1 + (((struct cm_req_msg *) mad_recv_wc->recv_buf.mad)->
3380						    alt_local_lid != 0);
3381		event = IB_CM_REQ_RECEIVED;
3382		break;
3383	case CM_MRA_ATTR_ID:
3384		event = IB_CM_MRA_RECEIVED;
3385		break;
3386	case CM_REJ_ATTR_ID:
3387		event = IB_CM_REJ_RECEIVED;
3388		break;
3389	case CM_REP_ATTR_ID:
3390		event = IB_CM_REP_RECEIVED;
3391		break;
3392	case CM_RTU_ATTR_ID:
3393		event = IB_CM_RTU_RECEIVED;
3394		break;
3395	case CM_DREQ_ATTR_ID:
3396		event = IB_CM_DREQ_RECEIVED;
3397		break;
3398	case CM_DREP_ATTR_ID:
3399		event = IB_CM_DREP_RECEIVED;
3400		break;
3401	case CM_SIDR_REQ_ATTR_ID:
3402		event = IB_CM_SIDR_REQ_RECEIVED;
3403		break;
3404	case CM_SIDR_REP_ATTR_ID:
3405		event = IB_CM_SIDR_REP_RECEIVED;
3406		break;
3407	case CM_LAP_ATTR_ID:
3408		paths = 1;
3409		event = IB_CM_LAP_RECEIVED;
3410		break;
3411	case CM_APR_ATTR_ID:
3412		event = IB_CM_APR_RECEIVED;
3413		break;
3414	default:
3415		ib_free_recv_mad(mad_recv_wc);
3416		return;
3417	}
3418
3419	attr_id = be16_to_cpu(mad_recv_wc->recv_buf.mad->mad_hdr.attr_id);
3420	atomic_long_inc(&port->counter_group[CM_RECV].
3421			counter[attr_id - CM_ATTR_ID_OFFSET]);
3422
3423	work = kmalloc(sizeof *work + sizeof(struct ib_sa_path_rec) * paths,
3424		       GFP_KERNEL);
3425	if (!work) {
3426		ib_free_recv_mad(mad_recv_wc);
3427		return;
3428	}
3429
3430	INIT_DELAYED_WORK(&work->work, cm_work_handler);
3431	work->cm_event.event = event;
3432	work->mad_recv_wc = mad_recv_wc;
3433	work->port = port;
3434	queue_delayed_work(cm.wq, &work->work, 0);
 
 
 
 
 
 
 
 
 
 
 
 
3435}
3436
3437static int cm_init_qp_init_attr(struct cm_id_private *cm_id_priv,
3438				struct ib_qp_attr *qp_attr,
3439				int *qp_attr_mask)
3440{
3441	unsigned long flags;
3442	int ret;
3443
3444	spin_lock_irqsave(&cm_id_priv->lock, flags);
3445	switch (cm_id_priv->id.state) {
3446	case IB_CM_REQ_SENT:
3447	case IB_CM_MRA_REQ_RCVD:
3448	case IB_CM_REQ_RCVD:
3449	case IB_CM_MRA_REQ_SENT:
3450	case IB_CM_REP_RCVD:
3451	case IB_CM_MRA_REP_SENT:
3452	case IB_CM_REP_SENT:
3453	case IB_CM_MRA_REP_RCVD:
3454	case IB_CM_ESTABLISHED:
3455		*qp_attr_mask = IB_QP_STATE | IB_QP_ACCESS_FLAGS |
3456				IB_QP_PKEY_INDEX | IB_QP_PORT;
3457		qp_attr->qp_access_flags = IB_ACCESS_REMOTE_WRITE;
3458		if (cm_id_priv->responder_resources)
3459			qp_attr->qp_access_flags |= IB_ACCESS_REMOTE_READ |
3460						    IB_ACCESS_REMOTE_ATOMIC;
3461		qp_attr->pkey_index = cm_id_priv->av.pkey_index;
3462		qp_attr->port_num = cm_id_priv->av.port->port_num;
3463		ret = 0;
3464		break;
3465	default:
3466		ret = -EINVAL;
3467		break;
3468	}
3469	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3470	return ret;
3471}
3472
3473static int cm_init_qp_rtr_attr(struct cm_id_private *cm_id_priv,
3474			       struct ib_qp_attr *qp_attr,
3475			       int *qp_attr_mask)
3476{
3477	unsigned long flags;
3478	int ret;
3479
3480	spin_lock_irqsave(&cm_id_priv->lock, flags);
3481	switch (cm_id_priv->id.state) {
3482	case IB_CM_REQ_RCVD:
3483	case IB_CM_MRA_REQ_SENT:
3484	case IB_CM_REP_RCVD:
3485	case IB_CM_MRA_REP_SENT:
3486	case IB_CM_REP_SENT:
3487	case IB_CM_MRA_REP_RCVD:
3488	case IB_CM_ESTABLISHED:
3489		*qp_attr_mask = IB_QP_STATE | IB_QP_AV | IB_QP_PATH_MTU |
3490				IB_QP_DEST_QPN | IB_QP_RQ_PSN;
3491		qp_attr->ah_attr = cm_id_priv->av.ah_attr;
3492		qp_attr->path_mtu = cm_id_priv->path_mtu;
3493		qp_attr->dest_qp_num = be32_to_cpu(cm_id_priv->remote_qpn);
3494		qp_attr->rq_psn = be32_to_cpu(cm_id_priv->rq_psn);
3495		if (cm_id_priv->qp_type == IB_QPT_RC) {
 
3496			*qp_attr_mask |= IB_QP_MAX_DEST_RD_ATOMIC |
3497					 IB_QP_MIN_RNR_TIMER;
3498			qp_attr->max_dest_rd_atomic =
3499					cm_id_priv->responder_resources;
3500			qp_attr->min_rnr_timer = 0;
3501		}
3502		if (cm_id_priv->alt_av.ah_attr.dlid) {
3503			*qp_attr_mask |= IB_QP_ALT_PATH;
3504			qp_attr->alt_port_num = cm_id_priv->alt_av.port->port_num;
3505			qp_attr->alt_pkey_index = cm_id_priv->alt_av.pkey_index;
3506			qp_attr->alt_timeout = cm_id_priv->alt_av.timeout;
3507			qp_attr->alt_ah_attr = cm_id_priv->alt_av.ah_attr;
3508		}
3509		ret = 0;
3510		break;
3511	default:
3512		ret = -EINVAL;
3513		break;
3514	}
3515	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3516	return ret;
3517}
3518
3519static int cm_init_qp_rts_attr(struct cm_id_private *cm_id_priv,
3520			       struct ib_qp_attr *qp_attr,
3521			       int *qp_attr_mask)
3522{
3523	unsigned long flags;
3524	int ret;
3525
3526	spin_lock_irqsave(&cm_id_priv->lock, flags);
3527	switch (cm_id_priv->id.state) {
3528	/* Allow transition to RTS before sending REP */
3529	case IB_CM_REQ_RCVD:
3530	case IB_CM_MRA_REQ_SENT:
3531
3532	case IB_CM_REP_RCVD:
3533	case IB_CM_MRA_REP_SENT:
3534	case IB_CM_REP_SENT:
3535	case IB_CM_MRA_REP_RCVD:
3536	case IB_CM_ESTABLISHED:
3537		if (cm_id_priv->id.lap_state == IB_CM_LAP_UNINIT) {
3538			*qp_attr_mask = IB_QP_STATE | IB_QP_SQ_PSN;
3539			qp_attr->sq_psn = be32_to_cpu(cm_id_priv->sq_psn);
3540			if (cm_id_priv->qp_type == IB_QPT_RC) {
3541				*qp_attr_mask |= IB_QP_TIMEOUT | IB_QP_RETRY_CNT |
3542						 IB_QP_RNR_RETRY |
 
3543						 IB_QP_MAX_QP_RD_ATOMIC;
3544				qp_attr->timeout = cm_id_priv->av.timeout;
3545				qp_attr->retry_cnt = cm_id_priv->retry_count;
3546				qp_attr->rnr_retry = cm_id_priv->rnr_retry_count;
3547				qp_attr->max_rd_atomic =
3548					cm_id_priv->initiator_depth;
 
 
 
 
 
 
3549			}
3550			if (cm_id_priv->alt_av.ah_attr.dlid) {
3551				*qp_attr_mask |= IB_QP_PATH_MIG_STATE;
3552				qp_attr->path_mig_state = IB_MIG_REARM;
3553			}
3554		} else {
3555			*qp_attr_mask = IB_QP_ALT_PATH | IB_QP_PATH_MIG_STATE;
3556			qp_attr->alt_port_num = cm_id_priv->alt_av.port->port_num;
3557			qp_attr->alt_pkey_index = cm_id_priv->alt_av.pkey_index;
3558			qp_attr->alt_timeout = cm_id_priv->alt_av.timeout;
3559			qp_attr->alt_ah_attr = cm_id_priv->alt_av.ah_attr;
3560			qp_attr->path_mig_state = IB_MIG_REARM;
3561		}
3562		ret = 0;
3563		break;
3564	default:
3565		ret = -EINVAL;
3566		break;
3567	}
3568	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3569	return ret;
3570}
3571
3572int ib_cm_init_qp_attr(struct ib_cm_id *cm_id,
3573		       struct ib_qp_attr *qp_attr,
3574		       int *qp_attr_mask)
3575{
3576	struct cm_id_private *cm_id_priv;
3577	int ret;
3578
3579	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3580	switch (qp_attr->qp_state) {
3581	case IB_QPS_INIT:
3582		ret = cm_init_qp_init_attr(cm_id_priv, qp_attr, qp_attr_mask);
3583		break;
3584	case IB_QPS_RTR:
3585		ret = cm_init_qp_rtr_attr(cm_id_priv, qp_attr, qp_attr_mask);
3586		break;
3587	case IB_QPS_RTS:
3588		ret = cm_init_qp_rts_attr(cm_id_priv, qp_attr, qp_attr_mask);
3589		break;
3590	default:
3591		ret = -EINVAL;
3592		break;
3593	}
3594	return ret;
3595}
3596EXPORT_SYMBOL(ib_cm_init_qp_attr);
3597
3598static void cm_get_ack_delay(struct cm_device *cm_dev)
3599{
3600	struct ib_device_attr attr;
3601
3602	if (ib_query_device(cm_dev->ib_device, &attr))
3603		cm_dev->ack_delay = 0; /* acks will rely on packet life time */
3604	else
3605		cm_dev->ack_delay = attr.local_ca_ack_delay;
3606}
3607
3608static ssize_t cm_show_counter(struct kobject *obj, struct attribute *attr,
3609			       char *buf)
3610{
3611	struct cm_counter_group *group;
3612	struct cm_counter_attribute *cm_attr;
3613
3614	group = container_of(obj, struct cm_counter_group, obj);
3615	cm_attr = container_of(attr, struct cm_counter_attribute, attr);
3616
3617	return sprintf(buf, "%ld\n",
3618		       atomic_long_read(&group->counter[cm_attr->index]));
3619}
3620
3621static const struct sysfs_ops cm_counter_ops = {
3622	.show = cm_show_counter
3623};
3624
3625static struct kobj_type cm_counter_obj_type = {
3626	.sysfs_ops = &cm_counter_ops,
3627	.default_attrs = cm_counter_default_attrs
3628};
3629
3630static void cm_release_port_obj(struct kobject *obj)
3631{
3632	struct cm_port *cm_port;
3633
3634	cm_port = container_of(obj, struct cm_port, port_obj);
3635	kfree(cm_port);
3636}
3637
3638static struct kobj_type cm_port_obj_type = {
3639	.release = cm_release_port_obj
3640};
3641
3642static char *cm_devnode(struct device *dev, mode_t *mode)
3643{
3644	if (mode)
3645		*mode = 0666;
3646	return kasprintf(GFP_KERNEL, "infiniband/%s", dev_name(dev));
3647}
3648
3649struct class cm_class = {
3650	.owner   = THIS_MODULE,
3651	.name    = "infiniband_cm",
3652	.devnode = cm_devnode,
3653};
3654EXPORT_SYMBOL(cm_class);
3655
3656static int cm_create_port_fs(struct cm_port *port)
3657{
3658	int i, ret;
3659
3660	ret = kobject_init_and_add(&port->port_obj, &cm_port_obj_type,
3661				   &port->cm_dev->device->kobj,
3662				   "%d", port->port_num);
3663	if (ret) {
3664		kfree(port);
3665		return ret;
3666	}
3667
3668	for (i = 0; i < CM_COUNTER_GROUPS; i++) {
3669		ret = kobject_init_and_add(&port->counter_group[i].obj,
3670					   &cm_counter_obj_type,
3671					   &port->port_obj,
3672					   "%s", counter_group_names[i]);
3673		if (ret)
3674			goto error;
3675	}
3676
3677	return 0;
3678
3679error:
3680	while (i--)
3681		kobject_put(&port->counter_group[i].obj);
3682	kobject_put(&port->port_obj);
3683	return ret;
3684
3685}
3686
3687static void cm_remove_port_fs(struct cm_port *port)
3688{
3689	int i;
3690
3691	for (i = 0; i < CM_COUNTER_GROUPS; i++)
3692		kobject_put(&port->counter_group[i].obj);
3693
3694	kobject_put(&port->port_obj);
3695}
3696
3697static void cm_add_one(struct ib_device *ib_device)
3698{
3699	struct cm_device *cm_dev;
3700	struct cm_port *port;
3701	struct ib_mad_reg_req reg_req = {
3702		.mgmt_class = IB_MGMT_CLASS_CM,
3703		.mgmt_class_version = IB_CM_CLASS_VERSION
3704	};
3705	struct ib_port_modify port_modify = {
3706		.set_port_cap_mask = IB_PORT_CM_SUP
3707	};
3708	unsigned long flags;
3709	int ret;
 
3710	u8 i;
3711
3712	if (rdma_node_get_transport(ib_device->node_type) != RDMA_TRANSPORT_IB)
3713		return;
3714
3715	cm_dev = kzalloc(sizeof(*cm_dev) + sizeof(*port) *
3716			 ib_device->phys_port_cnt, GFP_KERNEL);
3717	if (!cm_dev)
3718		return;
3719
3720	cm_dev->ib_device = ib_device;
3721	cm_get_ack_delay(cm_dev);
3722
3723	cm_dev->device = device_create(&cm_class, &ib_device->dev,
3724				       MKDEV(0, 0), NULL,
3725				       "%s", ib_device->name);
3726	if (IS_ERR(cm_dev->device)) {
3727		kfree(cm_dev);
3728		return;
3729	}
3730
3731	set_bit(IB_MGMT_METHOD_SEND, reg_req.method_mask);
3732	for (i = 1; i <= ib_device->phys_port_cnt; i++) {
 
 
 
3733		port = kzalloc(sizeof *port, GFP_KERNEL);
3734		if (!port)
3735			goto error1;
3736
3737		cm_dev->port[i-1] = port;
3738		port->cm_dev = cm_dev;
3739		port->port_num = i;
3740
3741		ret = cm_create_port_fs(port);
3742		if (ret)
3743			goto error1;
3744
3745		port->mad_agent = ib_register_mad_agent(ib_device, i,
3746							IB_QPT_GSI,
3747							&reg_req,
3748							0,
3749							cm_send_handler,
3750							cm_recv_handler,
3751							port);
 
3752		if (IS_ERR(port->mad_agent))
3753			goto error2;
3754
3755		ret = ib_modify_port(ib_device, i, 0, &port_modify);
3756		if (ret)
3757			goto error3;
 
 
3758	}
 
 
 
 
3759	ib_set_client_data(ib_device, &cm_client, cm_dev);
3760
3761	write_lock_irqsave(&cm.device_lock, flags);
3762	list_add_tail(&cm_dev->list, &cm.device_list);
3763	write_unlock_irqrestore(&cm.device_lock, flags);
3764	return;
3765
3766error3:
3767	ib_unregister_mad_agent(port->mad_agent);
3768error2:
3769	cm_remove_port_fs(port);
3770error1:
3771	port_modify.set_port_cap_mask = 0;
3772	port_modify.clr_port_cap_mask = IB_PORT_CM_SUP;
3773	while (--i) {
 
 
 
3774		port = cm_dev->port[i-1];
3775		ib_modify_port(ib_device, port->port_num, 0, &port_modify);
3776		ib_unregister_mad_agent(port->mad_agent);
3777		cm_remove_port_fs(port);
3778	}
 
3779	device_unregister(cm_dev->device);
3780	kfree(cm_dev);
3781}
3782
3783static void cm_remove_one(struct ib_device *ib_device)
3784{
3785	struct cm_device *cm_dev;
3786	struct cm_port *port;
3787	struct ib_port_modify port_modify = {
3788		.clr_port_cap_mask = IB_PORT_CM_SUP
3789	};
3790	unsigned long flags;
3791	int i;
3792
3793	cm_dev = ib_get_client_data(ib_device, &cm_client);
3794	if (!cm_dev)
3795		return;
3796
3797	write_lock_irqsave(&cm.device_lock, flags);
3798	list_del(&cm_dev->list);
3799	write_unlock_irqrestore(&cm.device_lock, flags);
3800
 
 
 
 
3801	for (i = 1; i <= ib_device->phys_port_cnt; i++) {
 
 
 
3802		port = cm_dev->port[i-1];
3803		ib_modify_port(ib_device, port->port_num, 0, &port_modify);
3804		ib_unregister_mad_agent(port->mad_agent);
 
 
 
 
3805		flush_workqueue(cm.wq);
 
3806		cm_remove_port_fs(port);
3807	}
3808	device_unregister(cm_dev->device);
3809	kfree(cm_dev);
3810}
3811
3812static int __init ib_cm_init(void)
3813{
3814	int ret;
3815
3816	memset(&cm, 0, sizeof cm);
3817	INIT_LIST_HEAD(&cm.device_list);
3818	rwlock_init(&cm.device_lock);
3819	spin_lock_init(&cm.lock);
3820	cm.listen_service_table = RB_ROOT;
3821	cm.listen_service_id = be64_to_cpu(IB_CM_ASSIGN_SERVICE_ID);
3822	cm.remote_id_table = RB_ROOT;
3823	cm.remote_qp_table = RB_ROOT;
3824	cm.remote_sidr_table = RB_ROOT;
3825	idr_init(&cm.local_id_table);
3826	get_random_bytes(&cm.random_id_operand, sizeof cm.random_id_operand);
3827	idr_pre_get(&cm.local_id_table, GFP_KERNEL);
3828	INIT_LIST_HEAD(&cm.timewait_list);
3829
3830	ret = class_register(&cm_class);
3831	if (ret)
3832		return -ENOMEM;
 
 
3833
3834	cm.wq = create_workqueue("ib_cm");
3835	if (!cm.wq) {
3836		ret = -ENOMEM;
3837		goto error1;
3838	}
3839
3840	ret = ib_register_client(&cm_client);
3841	if (ret)
3842		goto error2;
3843
3844	return 0;
3845error2:
3846	destroy_workqueue(cm.wq);
3847error1:
3848	class_unregister(&cm_class);
 
 
3849	return ret;
3850}
3851
3852static void __exit ib_cm_cleanup(void)
3853{
3854	struct cm_timewait_info *timewait_info, *tmp;
3855
3856	spin_lock_irq(&cm.lock);
3857	list_for_each_entry(timewait_info, &cm.timewait_list, list)
3858		cancel_delayed_work(&timewait_info->work.work);
3859	spin_unlock_irq(&cm.lock);
3860
3861	ib_unregister_client(&cm_client);
3862	destroy_workqueue(cm.wq);
3863
3864	list_for_each_entry_safe(timewait_info, tmp, &cm.timewait_list, list) {
3865		list_del(&timewait_info->list);
3866		kfree(timewait_info);
3867	}
3868
3869	class_unregister(&cm_class);
3870	idr_destroy(&cm.local_id_table);
3871}
3872
3873module_init(ib_cm_init);
3874module_exit(ib_cm_cleanup);
3875
v4.6
   1/*
   2 * Copyright (c) 2004-2007 Intel Corporation.  All rights reserved.
   3 * Copyright (c) 2004 Topspin Corporation.  All rights reserved.
   4 * Copyright (c) 2004, 2005 Voltaire Corporation.  All rights reserved.
   5 * Copyright (c) 2005 Sun Microsystems, Inc. All rights reserved.
   6 *
   7 * This software is available to you under a choice of one of two
   8 * licenses.  You may choose to be licensed under the terms of the GNU
   9 * General Public License (GPL) Version 2, available from the file
  10 * COPYING in the main directory of this source tree, or the
  11 * OpenIB.org BSD license below:
  12 *
  13 *     Redistribution and use in source and binary forms, with or
  14 *     without modification, are permitted provided that the following
  15 *     conditions are met:
  16 *
  17 *      - Redistributions of source code must retain the above
  18 *        copyright notice, this list of conditions and the following
  19 *        disclaimer.
  20 *
  21 *      - Redistributions in binary form must reproduce the above
  22 *        copyright notice, this list of conditions and the following
  23 *        disclaimer in the documentation and/or other materials
  24 *        provided with the distribution.
  25 *
  26 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  27 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  28 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  29 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  30 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  31 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  32 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  33 * SOFTWARE.
  34 */
  35
  36#include <linux/completion.h>
  37#include <linux/dma-mapping.h>
  38#include <linux/device.h>
  39#include <linux/module.h>
  40#include <linux/err.h>
  41#include <linux/idr.h>
  42#include <linux/interrupt.h>
  43#include <linux/random.h>
  44#include <linux/rbtree.h>
  45#include <linux/spinlock.h>
  46#include <linux/slab.h>
  47#include <linux/sysfs.h>
  48#include <linux/workqueue.h>
  49#include <linux/kdev_t.h>
  50#include <linux/etherdevice.h>
  51
  52#include <rdma/ib_cache.h>
  53#include <rdma/ib_cm.h>
  54#include "cm_msgs.h"
  55
  56MODULE_AUTHOR("Sean Hefty");
  57MODULE_DESCRIPTION("InfiniBand CM");
  58MODULE_LICENSE("Dual BSD/GPL");
  59
  60static void cm_add_one(struct ib_device *device);
  61static void cm_remove_one(struct ib_device *device, void *client_data);
  62
  63static struct ib_client cm_client = {
  64	.name   = "cm",
  65	.add    = cm_add_one,
  66	.remove = cm_remove_one
  67};
  68
  69static struct ib_cm {
  70	spinlock_t lock;
  71	struct list_head device_list;
  72	rwlock_t device_lock;
  73	struct rb_root listen_service_table;
  74	u64 listen_service_id;
  75	/* struct rb_root peer_service_table; todo: fix peer to peer */
  76	struct rb_root remote_qp_table;
  77	struct rb_root remote_id_table;
  78	struct rb_root remote_sidr_table;
  79	struct idr local_id_table;
  80	__be32 random_id_operand;
  81	struct list_head timewait_list;
  82	struct workqueue_struct *wq;
  83} cm;
  84
  85/* Counter indexes ordered by attribute ID */
  86enum {
  87	CM_REQ_COUNTER,
  88	CM_MRA_COUNTER,
  89	CM_REJ_COUNTER,
  90	CM_REP_COUNTER,
  91	CM_RTU_COUNTER,
  92	CM_DREQ_COUNTER,
  93	CM_DREP_COUNTER,
  94	CM_SIDR_REQ_COUNTER,
  95	CM_SIDR_REP_COUNTER,
  96	CM_LAP_COUNTER,
  97	CM_APR_COUNTER,
  98	CM_ATTR_COUNT,
  99	CM_ATTR_ID_OFFSET = 0x0010,
 100};
 101
 102enum {
 103	CM_XMIT,
 104	CM_XMIT_RETRIES,
 105	CM_RECV,
 106	CM_RECV_DUPLICATES,
 107	CM_COUNTER_GROUPS
 108};
 109
 110static char const counter_group_names[CM_COUNTER_GROUPS]
 111				     [sizeof("cm_rx_duplicates")] = {
 112	"cm_tx_msgs", "cm_tx_retries",
 113	"cm_rx_msgs", "cm_rx_duplicates"
 114};
 115
 116struct cm_counter_group {
 117	struct kobject obj;
 118	atomic_long_t counter[CM_ATTR_COUNT];
 119};
 120
 121struct cm_counter_attribute {
 122	struct attribute attr;
 123	int index;
 124};
 125
 126#define CM_COUNTER_ATTR(_name, _index) \
 127struct cm_counter_attribute cm_##_name##_counter_attr = { \
 128	.attr = { .name = __stringify(_name), .mode = 0444 }, \
 129	.index = _index \
 130}
 131
 132static CM_COUNTER_ATTR(req, CM_REQ_COUNTER);
 133static CM_COUNTER_ATTR(mra, CM_MRA_COUNTER);
 134static CM_COUNTER_ATTR(rej, CM_REJ_COUNTER);
 135static CM_COUNTER_ATTR(rep, CM_REP_COUNTER);
 136static CM_COUNTER_ATTR(rtu, CM_RTU_COUNTER);
 137static CM_COUNTER_ATTR(dreq, CM_DREQ_COUNTER);
 138static CM_COUNTER_ATTR(drep, CM_DREP_COUNTER);
 139static CM_COUNTER_ATTR(sidr_req, CM_SIDR_REQ_COUNTER);
 140static CM_COUNTER_ATTR(sidr_rep, CM_SIDR_REP_COUNTER);
 141static CM_COUNTER_ATTR(lap, CM_LAP_COUNTER);
 142static CM_COUNTER_ATTR(apr, CM_APR_COUNTER);
 143
 144static struct attribute *cm_counter_default_attrs[] = {
 145	&cm_req_counter_attr.attr,
 146	&cm_mra_counter_attr.attr,
 147	&cm_rej_counter_attr.attr,
 148	&cm_rep_counter_attr.attr,
 149	&cm_rtu_counter_attr.attr,
 150	&cm_dreq_counter_attr.attr,
 151	&cm_drep_counter_attr.attr,
 152	&cm_sidr_req_counter_attr.attr,
 153	&cm_sidr_rep_counter_attr.attr,
 154	&cm_lap_counter_attr.attr,
 155	&cm_apr_counter_attr.attr,
 156	NULL
 157};
 158
 159struct cm_port {
 160	struct cm_device *cm_dev;
 161	struct ib_mad_agent *mad_agent;
 162	struct kobject port_obj;
 163	u8 port_num;
 164	struct cm_counter_group counter_group[CM_COUNTER_GROUPS];
 165};
 166
 167struct cm_device {
 168	struct list_head list;
 169	struct ib_device *ib_device;
 170	struct device *device;
 171	u8 ack_delay;
 172	int going_down;
 173	struct cm_port *port[0];
 174};
 175
 176struct cm_av {
 177	struct cm_port *port;
 178	union ib_gid dgid;
 179	struct ib_ah_attr ah_attr;
 180	u16 pkey_index;
 181	u8 timeout;
 182};
 183
 184struct cm_work {
 185	struct delayed_work work;
 186	struct list_head list;
 187	struct cm_port *port;
 188	struct ib_mad_recv_wc *mad_recv_wc;	/* Received MADs */
 189	__be32 local_id;			/* Established / timewait */
 190	__be32 remote_id;
 191	struct ib_cm_event cm_event;
 192	struct ib_sa_path_rec path[0];
 193};
 194
 195struct cm_timewait_info {
 196	struct cm_work work;			/* Must be first. */
 197	struct list_head list;
 198	struct rb_node remote_qp_node;
 199	struct rb_node remote_id_node;
 200	__be64 remote_ca_guid;
 201	__be32 remote_qpn;
 202	u8 inserted_remote_qp;
 203	u8 inserted_remote_id;
 204};
 205
 206struct cm_id_private {
 207	struct ib_cm_id	id;
 208
 209	struct rb_node service_node;
 210	struct rb_node sidr_id_node;
 211	spinlock_t lock;	/* Do not acquire inside cm.lock */
 212	struct completion comp;
 213	atomic_t refcount;
 214	/* Number of clients sharing this ib_cm_id. Only valid for listeners.
 215	 * Protected by the cm.lock spinlock. */
 216	int listen_sharecount;
 217
 218	struct ib_mad_send_buf *msg;
 219	struct cm_timewait_info *timewait_info;
 220	/* todo: use alternate port on send failure */
 221	struct cm_av av;
 222	struct cm_av alt_av;
 
 223
 224	void *private_data;
 225	__be64 tid;
 226	__be32 local_qpn;
 227	__be32 remote_qpn;
 228	enum ib_qp_type qp_type;
 229	__be32 sq_psn;
 230	__be32 rq_psn;
 231	int timeout_ms;
 232	enum ib_mtu path_mtu;
 233	__be16 pkey;
 234	u8 private_data_len;
 235	u8 max_cm_retries;
 236	u8 peer_to_peer;
 237	u8 responder_resources;
 238	u8 initiator_depth;
 239	u8 retry_count;
 240	u8 rnr_retry_count;
 241	u8 service_timeout;
 242	u8 target_ack_delay;
 243
 244	struct list_head work_list;
 245	atomic_t work_count;
 246};
 247
 248static void cm_work_handler(struct work_struct *work);
 249
 250static inline void cm_deref_id(struct cm_id_private *cm_id_priv)
 251{
 252	if (atomic_dec_and_test(&cm_id_priv->refcount))
 253		complete(&cm_id_priv->comp);
 254}
 255
 256static int cm_alloc_msg(struct cm_id_private *cm_id_priv,
 257			struct ib_mad_send_buf **msg)
 258{
 259	struct ib_mad_agent *mad_agent;
 260	struct ib_mad_send_buf *m;
 261	struct ib_ah *ah;
 262
 263	mad_agent = cm_id_priv->av.port->mad_agent;
 264	ah = ib_create_ah(mad_agent->qp->pd, &cm_id_priv->av.ah_attr);
 265	if (IS_ERR(ah))
 266		return PTR_ERR(ah);
 267
 268	m = ib_create_send_mad(mad_agent, cm_id_priv->id.remote_cm_qpn,
 269			       cm_id_priv->av.pkey_index,
 270			       0, IB_MGMT_MAD_HDR, IB_MGMT_MAD_DATA,
 271			       GFP_ATOMIC,
 272			       IB_MGMT_BASE_VERSION);
 273	if (IS_ERR(m)) {
 274		ib_destroy_ah(ah);
 275		return PTR_ERR(m);
 276	}
 277
 278	/* Timeout set by caller if response is expected. */
 279	m->ah = ah;
 280	m->retries = cm_id_priv->max_cm_retries;
 281
 282	atomic_inc(&cm_id_priv->refcount);
 283	m->context[0] = cm_id_priv;
 284	*msg = m;
 285	return 0;
 286}
 287
 288static int cm_alloc_response_msg(struct cm_port *port,
 289				 struct ib_mad_recv_wc *mad_recv_wc,
 290				 struct ib_mad_send_buf **msg)
 291{
 292	struct ib_mad_send_buf *m;
 293	struct ib_ah *ah;
 294
 295	ah = ib_create_ah_from_wc(port->mad_agent->qp->pd, mad_recv_wc->wc,
 296				  mad_recv_wc->recv_buf.grh, port->port_num);
 297	if (IS_ERR(ah))
 298		return PTR_ERR(ah);
 299
 300	m = ib_create_send_mad(port->mad_agent, 1, mad_recv_wc->wc->pkey_index,
 301			       0, IB_MGMT_MAD_HDR, IB_MGMT_MAD_DATA,
 302			       GFP_ATOMIC,
 303			       IB_MGMT_BASE_VERSION);
 304	if (IS_ERR(m)) {
 305		ib_destroy_ah(ah);
 306		return PTR_ERR(m);
 307	}
 308	m->ah = ah;
 309	*msg = m;
 310	return 0;
 311}
 312
 313static void cm_free_msg(struct ib_mad_send_buf *msg)
 314{
 315	ib_destroy_ah(msg->ah);
 316	if (msg->context[0])
 317		cm_deref_id(msg->context[0]);
 318	ib_free_send_mad(msg);
 319}
 320
 321static void * cm_copy_private_data(const void *private_data,
 322				   u8 private_data_len)
 323{
 324	void *data;
 325
 326	if (!private_data || !private_data_len)
 327		return NULL;
 328
 329	data = kmemdup(private_data, private_data_len, GFP_KERNEL);
 330	if (!data)
 331		return ERR_PTR(-ENOMEM);
 332
 333	return data;
 334}
 335
 336static void cm_set_private_data(struct cm_id_private *cm_id_priv,
 337				 void *private_data, u8 private_data_len)
 338{
 339	if (cm_id_priv->private_data && cm_id_priv->private_data_len)
 340		kfree(cm_id_priv->private_data);
 341
 342	cm_id_priv->private_data = private_data;
 343	cm_id_priv->private_data_len = private_data_len;
 344}
 345
 346static void cm_init_av_for_response(struct cm_port *port, struct ib_wc *wc,
 347				    struct ib_grh *grh, struct cm_av *av)
 348{
 349	av->port = port;
 350	av->pkey_index = wc->pkey_index;
 351	ib_init_ah_from_wc(port->cm_dev->ib_device, port->port_num, wc,
 352			   grh, &av->ah_attr);
 353}
 354
 355static int cm_init_av_by_path(struct ib_sa_path_rec *path, struct cm_av *av)
 356{
 357	struct cm_device *cm_dev;
 358	struct cm_port *port = NULL;
 359	unsigned long flags;
 360	int ret;
 361	u8 p;
 362	struct net_device *ndev = ib_get_ndev_from_path(path);
 363
 364	read_lock_irqsave(&cm.device_lock, flags);
 365	list_for_each_entry(cm_dev, &cm.device_list, list) {
 366		if (!ib_find_cached_gid(cm_dev->ib_device, &path->sgid,
 367					path->gid_type, ndev, &p, NULL)) {
 368			port = cm_dev->port[p-1];
 369			break;
 370		}
 371	}
 372	read_unlock_irqrestore(&cm.device_lock, flags);
 373
 374	if (ndev)
 375		dev_put(ndev);
 376
 377	if (!port)
 378		return -EINVAL;
 379
 380	ret = ib_find_cached_pkey(cm_dev->ib_device, port->port_num,
 381				  be16_to_cpu(path->pkey), &av->pkey_index);
 382	if (ret)
 383		return ret;
 384
 385	av->port = port;
 386	ib_init_ah_from_path(cm_dev->ib_device, port->port_num, path,
 387			     &av->ah_attr);
 388	av->timeout = path->packet_life_time + 1;
 389
 390	return 0;
 391}
 392
 393static int cm_alloc_id(struct cm_id_private *cm_id_priv)
 394{
 395	unsigned long flags;
 396	int id;
 
 397
 398	idr_preload(GFP_KERNEL);
 399	spin_lock_irqsave(&cm.lock, flags);
 400
 401	id = idr_alloc_cyclic(&cm.local_id_table, cm_id_priv, 0, 0, GFP_NOWAIT);
 402
 403	spin_unlock_irqrestore(&cm.lock, flags);
 404	idr_preload_end();
 
 405
 406	cm_id_priv->id.local_id = (__force __be32)id ^ cm.random_id_operand;
 407	return id < 0 ? id : 0;
 408}
 409
 410static void cm_free_id(__be32 local_id)
 411{
 412	spin_lock_irq(&cm.lock);
 413	idr_remove(&cm.local_id_table,
 414		   (__force int) (local_id ^ cm.random_id_operand));
 415	spin_unlock_irq(&cm.lock);
 416}
 417
 418static struct cm_id_private * cm_get_id(__be32 local_id, __be32 remote_id)
 419{
 420	struct cm_id_private *cm_id_priv;
 421
 422	cm_id_priv = idr_find(&cm.local_id_table,
 423			      (__force int) (local_id ^ cm.random_id_operand));
 424	if (cm_id_priv) {
 425		if (cm_id_priv->id.remote_id == remote_id)
 426			atomic_inc(&cm_id_priv->refcount);
 427		else
 428			cm_id_priv = NULL;
 429	}
 430
 431	return cm_id_priv;
 432}
 433
 434static struct cm_id_private * cm_acquire_id(__be32 local_id, __be32 remote_id)
 435{
 436	struct cm_id_private *cm_id_priv;
 437
 438	spin_lock_irq(&cm.lock);
 439	cm_id_priv = cm_get_id(local_id, remote_id);
 440	spin_unlock_irq(&cm.lock);
 441
 442	return cm_id_priv;
 443}
 444
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 445/*
 446 * Trivial helpers to strip endian annotation and compare; the
 447 * endianness doesn't actually matter since we just need a stable
 448 * order for the RB tree.
 449 */
 450static int be32_lt(__be32 a, __be32 b)
 451{
 452	return (__force u32) a < (__force u32) b;
 453}
 454
 455static int be32_gt(__be32 a, __be32 b)
 456{
 457	return (__force u32) a > (__force u32) b;
 458}
 459
 460static int be64_lt(__be64 a, __be64 b)
 461{
 462	return (__force u64) a < (__force u64) b;
 463}
 464
 465static int be64_gt(__be64 a, __be64 b)
 466{
 467	return (__force u64) a > (__force u64) b;
 468}
 469
 470static struct cm_id_private * cm_insert_listen(struct cm_id_private *cm_id_priv)
 471{
 472	struct rb_node **link = &cm.listen_service_table.rb_node;
 473	struct rb_node *parent = NULL;
 474	struct cm_id_private *cur_cm_id_priv;
 475	__be64 service_id = cm_id_priv->id.service_id;
 476	__be64 service_mask = cm_id_priv->id.service_mask;
 
 477
 478	while (*link) {
 479		parent = *link;
 480		cur_cm_id_priv = rb_entry(parent, struct cm_id_private,
 481					  service_node);
 
 
 482		if ((cur_cm_id_priv->id.service_mask & service_id) ==
 483		    (service_mask & cur_cm_id_priv->id.service_id) &&
 484		    (cm_id_priv->id.device == cur_cm_id_priv->id.device))
 
 485			return cur_cm_id_priv;
 486
 487		if (cm_id_priv->id.device < cur_cm_id_priv->id.device)
 488			link = &(*link)->rb_left;
 489		else if (cm_id_priv->id.device > cur_cm_id_priv->id.device)
 490			link = &(*link)->rb_right;
 491		else if (be64_lt(service_id, cur_cm_id_priv->id.service_id))
 492			link = &(*link)->rb_left;
 493		else if (be64_gt(service_id, cur_cm_id_priv->id.service_id))
 494			link = &(*link)->rb_right;
 
 
 495		else
 496			link = &(*link)->rb_right;
 497	}
 498	rb_link_node(&cm_id_priv->service_node, parent, link);
 499	rb_insert_color(&cm_id_priv->service_node, &cm.listen_service_table);
 500	return NULL;
 501}
 502
 503static struct cm_id_private * cm_find_listen(struct ib_device *device,
 504					     __be64 service_id)
 
 505{
 506	struct rb_node *node = cm.listen_service_table.rb_node;
 507	struct cm_id_private *cm_id_priv;
 
 508
 509	while (node) {
 510		cm_id_priv = rb_entry(node, struct cm_id_private, service_node);
 
 
 511		if ((cm_id_priv->id.service_mask & service_id) ==
 512		     cm_id_priv->id.service_id &&
 513		    (cm_id_priv->id.device == device))
 514			return cm_id_priv;
 515
 516		if (device < cm_id_priv->id.device)
 517			node = node->rb_left;
 518		else if (device > cm_id_priv->id.device)
 519			node = node->rb_right;
 520		else if (be64_lt(service_id, cm_id_priv->id.service_id))
 521			node = node->rb_left;
 522		else if (be64_gt(service_id, cm_id_priv->id.service_id))
 523			node = node->rb_right;
 
 
 524		else
 525			node = node->rb_right;
 526	}
 527	return NULL;
 528}
 529
 530static struct cm_timewait_info * cm_insert_remote_id(struct cm_timewait_info
 531						     *timewait_info)
 532{
 533	struct rb_node **link = &cm.remote_id_table.rb_node;
 534	struct rb_node *parent = NULL;
 535	struct cm_timewait_info *cur_timewait_info;
 536	__be64 remote_ca_guid = timewait_info->remote_ca_guid;
 537	__be32 remote_id = timewait_info->work.remote_id;
 538
 539	while (*link) {
 540		parent = *link;
 541		cur_timewait_info = rb_entry(parent, struct cm_timewait_info,
 542					     remote_id_node);
 543		if (be32_lt(remote_id, cur_timewait_info->work.remote_id))
 544			link = &(*link)->rb_left;
 545		else if (be32_gt(remote_id, cur_timewait_info->work.remote_id))
 546			link = &(*link)->rb_right;
 547		else if (be64_lt(remote_ca_guid, cur_timewait_info->remote_ca_guid))
 548			link = &(*link)->rb_left;
 549		else if (be64_gt(remote_ca_guid, cur_timewait_info->remote_ca_guid))
 550			link = &(*link)->rb_right;
 551		else
 552			return cur_timewait_info;
 553	}
 554	timewait_info->inserted_remote_id = 1;
 555	rb_link_node(&timewait_info->remote_id_node, parent, link);
 556	rb_insert_color(&timewait_info->remote_id_node, &cm.remote_id_table);
 557	return NULL;
 558}
 559
 560static struct cm_timewait_info * cm_find_remote_id(__be64 remote_ca_guid,
 561						   __be32 remote_id)
 562{
 563	struct rb_node *node = cm.remote_id_table.rb_node;
 564	struct cm_timewait_info *timewait_info;
 565
 566	while (node) {
 567		timewait_info = rb_entry(node, struct cm_timewait_info,
 568					 remote_id_node);
 569		if (be32_lt(remote_id, timewait_info->work.remote_id))
 570			node = node->rb_left;
 571		else if (be32_gt(remote_id, timewait_info->work.remote_id))
 572			node = node->rb_right;
 573		else if (be64_lt(remote_ca_guid, timewait_info->remote_ca_guid))
 574			node = node->rb_left;
 575		else if (be64_gt(remote_ca_guid, timewait_info->remote_ca_guid))
 576			node = node->rb_right;
 577		else
 578			return timewait_info;
 579	}
 580	return NULL;
 581}
 582
 583static struct cm_timewait_info * cm_insert_remote_qpn(struct cm_timewait_info
 584						      *timewait_info)
 585{
 586	struct rb_node **link = &cm.remote_qp_table.rb_node;
 587	struct rb_node *parent = NULL;
 588	struct cm_timewait_info *cur_timewait_info;
 589	__be64 remote_ca_guid = timewait_info->remote_ca_guid;
 590	__be32 remote_qpn = timewait_info->remote_qpn;
 591
 592	while (*link) {
 593		parent = *link;
 594		cur_timewait_info = rb_entry(parent, struct cm_timewait_info,
 595					     remote_qp_node);
 596		if (be32_lt(remote_qpn, cur_timewait_info->remote_qpn))
 597			link = &(*link)->rb_left;
 598		else if (be32_gt(remote_qpn, cur_timewait_info->remote_qpn))
 599			link = &(*link)->rb_right;
 600		else if (be64_lt(remote_ca_guid, cur_timewait_info->remote_ca_guid))
 601			link = &(*link)->rb_left;
 602		else if (be64_gt(remote_ca_guid, cur_timewait_info->remote_ca_guid))
 603			link = &(*link)->rb_right;
 604		else
 605			return cur_timewait_info;
 606	}
 607	timewait_info->inserted_remote_qp = 1;
 608	rb_link_node(&timewait_info->remote_qp_node, parent, link);
 609	rb_insert_color(&timewait_info->remote_qp_node, &cm.remote_qp_table);
 610	return NULL;
 611}
 612
 613static struct cm_id_private * cm_insert_remote_sidr(struct cm_id_private
 614						    *cm_id_priv)
 615{
 616	struct rb_node **link = &cm.remote_sidr_table.rb_node;
 617	struct rb_node *parent = NULL;
 618	struct cm_id_private *cur_cm_id_priv;
 619	union ib_gid *port_gid = &cm_id_priv->av.dgid;
 620	__be32 remote_id = cm_id_priv->id.remote_id;
 621
 622	while (*link) {
 623		parent = *link;
 624		cur_cm_id_priv = rb_entry(parent, struct cm_id_private,
 625					  sidr_id_node);
 626		if (be32_lt(remote_id, cur_cm_id_priv->id.remote_id))
 627			link = &(*link)->rb_left;
 628		else if (be32_gt(remote_id, cur_cm_id_priv->id.remote_id))
 629			link = &(*link)->rb_right;
 630		else {
 631			int cmp;
 632			cmp = memcmp(port_gid, &cur_cm_id_priv->av.dgid,
 633				     sizeof *port_gid);
 634			if (cmp < 0)
 635				link = &(*link)->rb_left;
 636			else if (cmp > 0)
 637				link = &(*link)->rb_right;
 638			else
 639				return cur_cm_id_priv;
 640		}
 641	}
 642	rb_link_node(&cm_id_priv->sidr_id_node, parent, link);
 643	rb_insert_color(&cm_id_priv->sidr_id_node, &cm.remote_sidr_table);
 644	return NULL;
 645}
 646
 647static void cm_reject_sidr_req(struct cm_id_private *cm_id_priv,
 648			       enum ib_cm_sidr_status status)
 649{
 650	struct ib_cm_sidr_rep_param param;
 651
 652	memset(&param, 0, sizeof param);
 653	param.status = status;
 654	ib_send_cm_sidr_rep(&cm_id_priv->id, &param);
 655}
 656
 657struct ib_cm_id *ib_create_cm_id(struct ib_device *device,
 658				 ib_cm_handler cm_handler,
 659				 void *context)
 660{
 661	struct cm_id_private *cm_id_priv;
 662	int ret;
 663
 664	cm_id_priv = kzalloc(sizeof *cm_id_priv, GFP_KERNEL);
 665	if (!cm_id_priv)
 666		return ERR_PTR(-ENOMEM);
 667
 668	cm_id_priv->id.state = IB_CM_IDLE;
 669	cm_id_priv->id.device = device;
 670	cm_id_priv->id.cm_handler = cm_handler;
 671	cm_id_priv->id.context = context;
 672	cm_id_priv->id.remote_cm_qpn = 1;
 673	ret = cm_alloc_id(cm_id_priv);
 674	if (ret)
 675		goto error;
 676
 677	spin_lock_init(&cm_id_priv->lock);
 678	init_completion(&cm_id_priv->comp);
 679	INIT_LIST_HEAD(&cm_id_priv->work_list);
 680	atomic_set(&cm_id_priv->work_count, -1);
 681	atomic_set(&cm_id_priv->refcount, 1);
 682	return &cm_id_priv->id;
 683
 684error:
 685	kfree(cm_id_priv);
 686	return ERR_PTR(-ENOMEM);
 687}
 688EXPORT_SYMBOL(ib_create_cm_id);
 689
 690static struct cm_work * cm_dequeue_work(struct cm_id_private *cm_id_priv)
 691{
 692	struct cm_work *work;
 693
 694	if (list_empty(&cm_id_priv->work_list))
 695		return NULL;
 696
 697	work = list_entry(cm_id_priv->work_list.next, struct cm_work, list);
 698	list_del(&work->list);
 699	return work;
 700}
 701
 702static void cm_free_work(struct cm_work *work)
 703{
 704	if (work->mad_recv_wc)
 705		ib_free_recv_mad(work->mad_recv_wc);
 706	kfree(work);
 707}
 708
 709static inline int cm_convert_to_ms(int iba_time)
 710{
 711	/* approximate conversion to ms from 4.096us x 2^iba_time */
 712	return 1 << max(iba_time - 8, 0);
 713}
 714
 715/*
 716 * calculate: 4.096x2^ack_timeout = 4.096x2^ack_delay + 2x4.096x2^life_time
 717 * Because of how ack_timeout is stored, adding one doubles the timeout.
 718 * To avoid large timeouts, select the max(ack_delay, life_time + 1), and
 719 * increment it (round up) only if the other is within 50%.
 720 */
 721static u8 cm_ack_timeout(u8 ca_ack_delay, u8 packet_life_time)
 722{
 723	int ack_timeout = packet_life_time + 1;
 724
 725	if (ack_timeout >= ca_ack_delay)
 726		ack_timeout += (ca_ack_delay >= (ack_timeout - 1));
 727	else
 728		ack_timeout = ca_ack_delay +
 729			      (ack_timeout >= (ca_ack_delay - 1));
 730
 731	return min(31, ack_timeout);
 732}
 733
 734static void cm_cleanup_timewait(struct cm_timewait_info *timewait_info)
 735{
 736	if (timewait_info->inserted_remote_id) {
 737		rb_erase(&timewait_info->remote_id_node, &cm.remote_id_table);
 738		timewait_info->inserted_remote_id = 0;
 739	}
 740
 741	if (timewait_info->inserted_remote_qp) {
 742		rb_erase(&timewait_info->remote_qp_node, &cm.remote_qp_table);
 743		timewait_info->inserted_remote_qp = 0;
 744	}
 745}
 746
 747static struct cm_timewait_info * cm_create_timewait_info(__be32 local_id)
 748{
 749	struct cm_timewait_info *timewait_info;
 750
 751	timewait_info = kzalloc(sizeof *timewait_info, GFP_KERNEL);
 752	if (!timewait_info)
 753		return ERR_PTR(-ENOMEM);
 754
 755	timewait_info->work.local_id = local_id;
 756	INIT_DELAYED_WORK(&timewait_info->work.work, cm_work_handler);
 757	timewait_info->work.cm_event.event = IB_CM_TIMEWAIT_EXIT;
 758	return timewait_info;
 759}
 760
 761static void cm_enter_timewait(struct cm_id_private *cm_id_priv)
 762{
 763	int wait_time;
 764	unsigned long flags;
 765	struct cm_device *cm_dev;
 766
 767	cm_dev = ib_get_client_data(cm_id_priv->id.device, &cm_client);
 768	if (!cm_dev)
 769		return;
 770
 771	spin_lock_irqsave(&cm.lock, flags);
 772	cm_cleanup_timewait(cm_id_priv->timewait_info);
 773	list_add_tail(&cm_id_priv->timewait_info->list, &cm.timewait_list);
 774	spin_unlock_irqrestore(&cm.lock, flags);
 775
 776	/*
 777	 * The cm_id could be destroyed by the user before we exit timewait.
 778	 * To protect against this, we search for the cm_id after exiting
 779	 * timewait before notifying the user that we've exited timewait.
 780	 */
 781	cm_id_priv->id.state = IB_CM_TIMEWAIT;
 782	wait_time = cm_convert_to_ms(cm_id_priv->av.timeout);
 783
 784	/* Check if the device started its remove_one */
 785	spin_lock_irqsave(&cm.lock, flags);
 786	if (!cm_dev->going_down)
 787		queue_delayed_work(cm.wq, &cm_id_priv->timewait_info->work.work,
 788				   msecs_to_jiffies(wait_time));
 789	spin_unlock_irqrestore(&cm.lock, flags);
 790
 791	cm_id_priv->timewait_info = NULL;
 792}
 793
 794static void cm_reset_to_idle(struct cm_id_private *cm_id_priv)
 795{
 796	unsigned long flags;
 797
 798	cm_id_priv->id.state = IB_CM_IDLE;
 799	if (cm_id_priv->timewait_info) {
 800		spin_lock_irqsave(&cm.lock, flags);
 801		cm_cleanup_timewait(cm_id_priv->timewait_info);
 802		spin_unlock_irqrestore(&cm.lock, flags);
 803		kfree(cm_id_priv->timewait_info);
 804		cm_id_priv->timewait_info = NULL;
 805	}
 806}
 807
 808static void cm_destroy_id(struct ib_cm_id *cm_id, int err)
 809{
 810	struct cm_id_private *cm_id_priv;
 811	struct cm_work *work;
 812
 813	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
 814retest:
 815	spin_lock_irq(&cm_id_priv->lock);
 816	switch (cm_id->state) {
 817	case IB_CM_LISTEN:
 
 818		spin_unlock_irq(&cm_id_priv->lock);
 819
 820		spin_lock_irq(&cm.lock);
 821		if (--cm_id_priv->listen_sharecount > 0) {
 822			/* The id is still shared. */
 823			cm_deref_id(cm_id_priv);
 824			spin_unlock_irq(&cm.lock);
 825			return;
 826		}
 827		rb_erase(&cm_id_priv->service_node, &cm.listen_service_table);
 828		spin_unlock_irq(&cm.lock);
 829		break;
 830	case IB_CM_SIDR_REQ_SENT:
 831		cm_id->state = IB_CM_IDLE;
 832		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
 833		spin_unlock_irq(&cm_id_priv->lock);
 834		break;
 835	case IB_CM_SIDR_REQ_RCVD:
 836		spin_unlock_irq(&cm_id_priv->lock);
 837		cm_reject_sidr_req(cm_id_priv, IB_SIDR_REJECT);
 838		spin_lock_irq(&cm.lock);
 839		if (!RB_EMPTY_NODE(&cm_id_priv->sidr_id_node))
 840			rb_erase(&cm_id_priv->sidr_id_node,
 841				 &cm.remote_sidr_table);
 842		spin_unlock_irq(&cm.lock);
 843		break;
 844	case IB_CM_REQ_SENT:
 845	case IB_CM_MRA_REQ_RCVD:
 846		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
 847		spin_unlock_irq(&cm_id_priv->lock);
 848		ib_send_cm_rej(cm_id, IB_CM_REJ_TIMEOUT,
 849			       &cm_id_priv->id.device->node_guid,
 850			       sizeof cm_id_priv->id.device->node_guid,
 851			       NULL, 0);
 852		break;
 853	case IB_CM_REQ_RCVD:
 854		if (err == -ENOMEM) {
 855			/* Do not reject to allow future retries. */
 856			cm_reset_to_idle(cm_id_priv);
 857			spin_unlock_irq(&cm_id_priv->lock);
 858		} else {
 859			spin_unlock_irq(&cm_id_priv->lock);
 860			ib_send_cm_rej(cm_id, IB_CM_REJ_CONSUMER_DEFINED,
 861				       NULL, 0, NULL, 0);
 862		}
 863		break;
 
 864	case IB_CM_REP_SENT:
 865	case IB_CM_MRA_REP_RCVD:
 866		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
 867		/* Fall through */
 868	case IB_CM_MRA_REQ_SENT:
 869	case IB_CM_REP_RCVD:
 870	case IB_CM_MRA_REP_SENT:
 871		spin_unlock_irq(&cm_id_priv->lock);
 872		ib_send_cm_rej(cm_id, IB_CM_REJ_CONSUMER_DEFINED,
 873			       NULL, 0, NULL, 0);
 874		break;
 875	case IB_CM_ESTABLISHED:
 876		spin_unlock_irq(&cm_id_priv->lock);
 877		if (cm_id_priv->qp_type == IB_QPT_XRC_TGT)
 878			break;
 879		ib_send_cm_dreq(cm_id, NULL, 0);
 880		goto retest;
 881	case IB_CM_DREQ_SENT:
 882		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
 883		cm_enter_timewait(cm_id_priv);
 884		spin_unlock_irq(&cm_id_priv->lock);
 885		break;
 886	case IB_CM_DREQ_RCVD:
 887		spin_unlock_irq(&cm_id_priv->lock);
 888		ib_send_cm_drep(cm_id, NULL, 0);
 889		break;
 890	default:
 891		spin_unlock_irq(&cm_id_priv->lock);
 892		break;
 893	}
 894
 895	cm_free_id(cm_id->local_id);
 896	cm_deref_id(cm_id_priv);
 897	wait_for_completion(&cm_id_priv->comp);
 898	while ((work = cm_dequeue_work(cm_id_priv)) != NULL)
 899		cm_free_work(work);
 
 900	kfree(cm_id_priv->private_data);
 901	kfree(cm_id_priv);
 902}
 903
 904void ib_destroy_cm_id(struct ib_cm_id *cm_id)
 905{
 906	cm_destroy_id(cm_id, 0);
 907}
 908EXPORT_SYMBOL(ib_destroy_cm_id);
 909
 910/**
 911 * __ib_cm_listen - Initiates listening on the specified service ID for
 912 *   connection and service ID resolution requests.
 913 * @cm_id: Connection identifier associated with the listen request.
 914 * @service_id: Service identifier matched against incoming connection
 915 *   and service ID resolution requests.  The service ID should be specified
 916 *   network-byte order.  If set to IB_CM_ASSIGN_SERVICE_ID, the CM will
 917 *   assign a service ID to the caller.
 918 * @service_mask: Mask applied to service ID used to listen across a
 919 *   range of service IDs.  If set to 0, the service ID is matched
 920 *   exactly.  This parameter is ignored if %service_id is set to
 921 *   IB_CM_ASSIGN_SERVICE_ID.
 922 */
 923static int __ib_cm_listen(struct ib_cm_id *cm_id, __be64 service_id,
 924			  __be64 service_mask)
 925{
 926	struct cm_id_private *cm_id_priv, *cur_cm_id_priv;
 
 927	int ret = 0;
 928
 929	service_mask = service_mask ? service_mask : ~cpu_to_be64(0);
 930	service_id &= service_mask;
 931	if ((service_id & IB_SERVICE_ID_AGN_MASK) == IB_CM_ASSIGN_SERVICE_ID &&
 932	    (service_id != IB_CM_ASSIGN_SERVICE_ID))
 933		return -EINVAL;
 934
 935	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
 936	if (cm_id->state != IB_CM_IDLE)
 937		return -EINVAL;
 938
 
 
 
 
 
 
 
 
 
 
 
 939	cm_id->state = IB_CM_LISTEN;
 940	++cm_id_priv->listen_sharecount;
 941
 
 942	if (service_id == IB_CM_ASSIGN_SERVICE_ID) {
 943		cm_id->service_id = cpu_to_be64(cm.listen_service_id++);
 944		cm_id->service_mask = ~cpu_to_be64(0);
 945	} else {
 946		cm_id->service_id = service_id;
 947		cm_id->service_mask = service_mask;
 948	}
 949	cur_cm_id_priv = cm_insert_listen(cm_id_priv);
 
 950
 951	if (cur_cm_id_priv) {
 952		cm_id->state = IB_CM_IDLE;
 953		--cm_id_priv->listen_sharecount;
 
 954		ret = -EBUSY;
 955	}
 956	return ret;
 957}
 958
 959int ib_cm_listen(struct ib_cm_id *cm_id, __be64 service_id, __be64 service_mask)
 960{
 961	unsigned long flags;
 962	int ret;
 963
 964	spin_lock_irqsave(&cm.lock, flags);
 965	ret = __ib_cm_listen(cm_id, service_id, service_mask);
 966	spin_unlock_irqrestore(&cm.lock, flags);
 967
 968	return ret;
 969}
 970EXPORT_SYMBOL(ib_cm_listen);
 971
 972/**
 973 * Create a new listening ib_cm_id and listen on the given service ID.
 974 *
 975 * If there's an existing ID listening on that same device and service ID,
 976 * return it.
 977 *
 978 * @device: Device associated with the cm_id.  All related communication will
 979 * be associated with the specified device.
 980 * @cm_handler: Callback invoked to notify the user of CM events.
 981 * @service_id: Service identifier matched against incoming connection
 982 *   and service ID resolution requests.  The service ID should be specified
 983 *   network-byte order.  If set to IB_CM_ASSIGN_SERVICE_ID, the CM will
 984 *   assign a service ID to the caller.
 985 *
 986 * Callers should call ib_destroy_cm_id when done with the listener ID.
 987 */
 988struct ib_cm_id *ib_cm_insert_listen(struct ib_device *device,
 989				     ib_cm_handler cm_handler,
 990				     __be64 service_id)
 991{
 992	struct cm_id_private *cm_id_priv;
 993	struct ib_cm_id *cm_id;
 994	unsigned long flags;
 995	int err = 0;
 996
 997	/* Create an ID in advance, since the creation may sleep */
 998	cm_id = ib_create_cm_id(device, cm_handler, NULL);
 999	if (IS_ERR(cm_id))
1000		return cm_id;
1001
1002	spin_lock_irqsave(&cm.lock, flags);
1003
1004	if (service_id == IB_CM_ASSIGN_SERVICE_ID)
1005		goto new_id;
1006
1007	/* Find an existing ID */
1008	cm_id_priv = cm_find_listen(device, service_id);
1009	if (cm_id_priv) {
1010		if (cm_id->cm_handler != cm_handler || cm_id->context) {
1011			/* Sharing an ib_cm_id with different handlers is not
1012			 * supported */
1013			spin_unlock_irqrestore(&cm.lock, flags);
1014			return ERR_PTR(-EINVAL);
1015		}
1016		atomic_inc(&cm_id_priv->refcount);
1017		++cm_id_priv->listen_sharecount;
1018		spin_unlock_irqrestore(&cm.lock, flags);
1019
1020		ib_destroy_cm_id(cm_id);
1021		cm_id = &cm_id_priv->id;
1022		return cm_id;
1023	}
1024
1025new_id:
1026	/* Use newly created ID */
1027	err = __ib_cm_listen(cm_id, service_id, 0);
1028
1029	spin_unlock_irqrestore(&cm.lock, flags);
1030
1031	if (err) {
1032		ib_destroy_cm_id(cm_id);
1033		return ERR_PTR(err);
1034	}
1035	return cm_id;
1036}
1037EXPORT_SYMBOL(ib_cm_insert_listen);
1038
1039static __be64 cm_form_tid(struct cm_id_private *cm_id_priv,
1040			  enum cm_msg_sequence msg_seq)
1041{
1042	u64 hi_tid, low_tid;
1043
1044	hi_tid   = ((u64) cm_id_priv->av.port->mad_agent->hi_tid) << 32;
1045	low_tid  = (u64) ((__force u32)cm_id_priv->id.local_id |
1046			  (msg_seq << 30));
1047	return cpu_to_be64(hi_tid | low_tid);
1048}
1049
1050static void cm_format_mad_hdr(struct ib_mad_hdr *hdr,
1051			      __be16 attr_id, __be64 tid)
1052{
1053	hdr->base_version  = IB_MGMT_BASE_VERSION;
1054	hdr->mgmt_class	   = IB_MGMT_CLASS_CM;
1055	hdr->class_version = IB_CM_CLASS_VERSION;
1056	hdr->method	   = IB_MGMT_METHOD_SEND;
1057	hdr->attr_id	   = attr_id;
1058	hdr->tid	   = tid;
1059}
1060
1061static void cm_format_req(struct cm_req_msg *req_msg,
1062			  struct cm_id_private *cm_id_priv,
1063			  struct ib_cm_req_param *param)
1064{
1065	struct ib_sa_path_rec *pri_path = param->primary_path;
1066	struct ib_sa_path_rec *alt_path = param->alternate_path;
1067
1068	cm_format_mad_hdr(&req_msg->hdr, CM_REQ_ATTR_ID,
1069			  cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_REQ));
1070
1071	req_msg->local_comm_id = cm_id_priv->id.local_id;
1072	req_msg->service_id = param->service_id;
1073	req_msg->local_ca_guid = cm_id_priv->id.device->node_guid;
1074	cm_req_set_local_qpn(req_msg, cpu_to_be32(param->qp_num));
 
1075	cm_req_set_init_depth(req_msg, param->initiator_depth);
1076	cm_req_set_remote_resp_timeout(req_msg,
1077				       param->remote_cm_response_timeout);
1078	cm_req_set_qp_type(req_msg, param->qp_type);
1079	cm_req_set_flow_ctrl(req_msg, param->flow_control);
1080	cm_req_set_starting_psn(req_msg, cpu_to_be32(param->starting_psn));
1081	cm_req_set_local_resp_timeout(req_msg,
1082				      param->local_cm_response_timeout);
 
1083	req_msg->pkey = param->primary_path->pkey;
1084	cm_req_set_path_mtu(req_msg, param->primary_path->mtu);
 
1085	cm_req_set_max_cm_retries(req_msg, param->max_cm_retries);
1086
1087	if (param->qp_type != IB_QPT_XRC_INI) {
1088		cm_req_set_resp_res(req_msg, param->responder_resources);
1089		cm_req_set_retry_count(req_msg, param->retry_count);
1090		cm_req_set_rnr_retry_count(req_msg, param->rnr_retry_count);
1091		cm_req_set_srq(req_msg, param->srq);
1092	}
1093
1094	if (pri_path->hop_limit <= 1) {
1095		req_msg->primary_local_lid = pri_path->slid;
1096		req_msg->primary_remote_lid = pri_path->dlid;
1097	} else {
1098		/* Work-around until there's a way to obtain remote LID info */
1099		req_msg->primary_local_lid = IB_LID_PERMISSIVE;
1100		req_msg->primary_remote_lid = IB_LID_PERMISSIVE;
1101	}
1102	req_msg->primary_local_gid = pri_path->sgid;
1103	req_msg->primary_remote_gid = pri_path->dgid;
1104	cm_req_set_primary_flow_label(req_msg, pri_path->flow_label);
1105	cm_req_set_primary_packet_rate(req_msg, pri_path->rate);
1106	req_msg->primary_traffic_class = pri_path->traffic_class;
1107	req_msg->primary_hop_limit = pri_path->hop_limit;
1108	cm_req_set_primary_sl(req_msg, pri_path->sl);
1109	cm_req_set_primary_subnet_local(req_msg, (pri_path->hop_limit <= 1));
1110	cm_req_set_primary_local_ack_timeout(req_msg,
1111		cm_ack_timeout(cm_id_priv->av.port->cm_dev->ack_delay,
1112			       pri_path->packet_life_time));
1113
1114	if (alt_path) {
1115		if (alt_path->hop_limit <= 1) {
1116			req_msg->alt_local_lid = alt_path->slid;
1117			req_msg->alt_remote_lid = alt_path->dlid;
1118		} else {
1119			req_msg->alt_local_lid = IB_LID_PERMISSIVE;
1120			req_msg->alt_remote_lid = IB_LID_PERMISSIVE;
1121		}
1122		req_msg->alt_local_gid = alt_path->sgid;
1123		req_msg->alt_remote_gid = alt_path->dgid;
1124		cm_req_set_alt_flow_label(req_msg,
1125					  alt_path->flow_label);
1126		cm_req_set_alt_packet_rate(req_msg, alt_path->rate);
1127		req_msg->alt_traffic_class = alt_path->traffic_class;
1128		req_msg->alt_hop_limit = alt_path->hop_limit;
1129		cm_req_set_alt_sl(req_msg, alt_path->sl);
1130		cm_req_set_alt_subnet_local(req_msg, (alt_path->hop_limit <= 1));
1131		cm_req_set_alt_local_ack_timeout(req_msg,
1132			cm_ack_timeout(cm_id_priv->av.port->cm_dev->ack_delay,
1133				       alt_path->packet_life_time));
1134	}
1135
1136	if (param->private_data && param->private_data_len)
1137		memcpy(req_msg->private_data, param->private_data,
1138		       param->private_data_len);
1139}
1140
1141static int cm_validate_req_param(struct ib_cm_req_param *param)
1142{
1143	/* peer-to-peer not supported */
1144	if (param->peer_to_peer)
1145		return -EINVAL;
1146
1147	if (!param->primary_path)
1148		return -EINVAL;
1149
1150	if (param->qp_type != IB_QPT_RC && param->qp_type != IB_QPT_UC &&
1151	    param->qp_type != IB_QPT_XRC_INI)
1152		return -EINVAL;
1153
1154	if (param->private_data &&
1155	    param->private_data_len > IB_CM_REQ_PRIVATE_DATA_SIZE)
1156		return -EINVAL;
1157
1158	if (param->alternate_path &&
1159	    (param->alternate_path->pkey != param->primary_path->pkey ||
1160	     param->alternate_path->mtu != param->primary_path->mtu))
1161		return -EINVAL;
1162
1163	return 0;
1164}
1165
1166int ib_send_cm_req(struct ib_cm_id *cm_id,
1167		   struct ib_cm_req_param *param)
1168{
1169	struct cm_id_private *cm_id_priv;
1170	struct cm_req_msg *req_msg;
1171	unsigned long flags;
1172	int ret;
1173
1174	ret = cm_validate_req_param(param);
1175	if (ret)
1176		return ret;
1177
1178	/* Verify that we're not in timewait. */
1179	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1180	spin_lock_irqsave(&cm_id_priv->lock, flags);
1181	if (cm_id->state != IB_CM_IDLE) {
1182		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1183		ret = -EINVAL;
1184		goto out;
1185	}
1186	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1187
1188	cm_id_priv->timewait_info = cm_create_timewait_info(cm_id_priv->
1189							    id.local_id);
1190	if (IS_ERR(cm_id_priv->timewait_info)) {
1191		ret = PTR_ERR(cm_id_priv->timewait_info);
1192		goto out;
1193	}
1194
1195	ret = cm_init_av_by_path(param->primary_path, &cm_id_priv->av);
1196	if (ret)
1197		goto error1;
1198	if (param->alternate_path) {
1199		ret = cm_init_av_by_path(param->alternate_path,
1200					 &cm_id_priv->alt_av);
1201		if (ret)
1202			goto error1;
1203	}
1204	cm_id->service_id = param->service_id;
1205	cm_id->service_mask = ~cpu_to_be64(0);
1206	cm_id_priv->timeout_ms = cm_convert_to_ms(
1207				    param->primary_path->packet_life_time) * 2 +
1208				 cm_convert_to_ms(
1209				    param->remote_cm_response_timeout);
1210	cm_id_priv->max_cm_retries = param->max_cm_retries;
1211	cm_id_priv->initiator_depth = param->initiator_depth;
1212	cm_id_priv->responder_resources = param->responder_resources;
1213	cm_id_priv->retry_count = param->retry_count;
1214	cm_id_priv->path_mtu = param->primary_path->mtu;
1215	cm_id_priv->pkey = param->primary_path->pkey;
1216	cm_id_priv->qp_type = param->qp_type;
1217
1218	ret = cm_alloc_msg(cm_id_priv, &cm_id_priv->msg);
1219	if (ret)
1220		goto error1;
1221
1222	req_msg = (struct cm_req_msg *) cm_id_priv->msg->mad;
1223	cm_format_req(req_msg, cm_id_priv, param);
1224	cm_id_priv->tid = req_msg->hdr.tid;
1225	cm_id_priv->msg->timeout_ms = cm_id_priv->timeout_ms;
1226	cm_id_priv->msg->context[1] = (void *) (unsigned long) IB_CM_REQ_SENT;
1227
1228	cm_id_priv->local_qpn = cm_req_get_local_qpn(req_msg);
1229	cm_id_priv->rq_psn = cm_req_get_starting_psn(req_msg);
1230
1231	spin_lock_irqsave(&cm_id_priv->lock, flags);
1232	ret = ib_post_send_mad(cm_id_priv->msg, NULL);
1233	if (ret) {
1234		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1235		goto error2;
1236	}
1237	BUG_ON(cm_id->state != IB_CM_IDLE);
1238	cm_id->state = IB_CM_REQ_SENT;
1239	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1240	return 0;
1241
1242error2:	cm_free_msg(cm_id_priv->msg);
1243error1:	kfree(cm_id_priv->timewait_info);
1244out:	return ret;
1245}
1246EXPORT_SYMBOL(ib_send_cm_req);
1247
1248static int cm_issue_rej(struct cm_port *port,
1249			struct ib_mad_recv_wc *mad_recv_wc,
1250			enum ib_cm_rej_reason reason,
1251			enum cm_msg_response msg_rejected,
1252			void *ari, u8 ari_length)
1253{
1254	struct ib_mad_send_buf *msg = NULL;
1255	struct cm_rej_msg *rej_msg, *rcv_msg;
1256	int ret;
1257
1258	ret = cm_alloc_response_msg(port, mad_recv_wc, &msg);
1259	if (ret)
1260		return ret;
1261
1262	/* We just need common CM header information.  Cast to any message. */
1263	rcv_msg = (struct cm_rej_msg *) mad_recv_wc->recv_buf.mad;
1264	rej_msg = (struct cm_rej_msg *) msg->mad;
1265
1266	cm_format_mad_hdr(&rej_msg->hdr, CM_REJ_ATTR_ID, rcv_msg->hdr.tid);
1267	rej_msg->remote_comm_id = rcv_msg->local_comm_id;
1268	rej_msg->local_comm_id = rcv_msg->remote_comm_id;
1269	cm_rej_set_msg_rejected(rej_msg, msg_rejected);
1270	rej_msg->reason = cpu_to_be16(reason);
1271
1272	if (ari && ari_length) {
1273		cm_rej_set_reject_info_len(rej_msg, ari_length);
1274		memcpy(rej_msg->ari, ari, ari_length);
1275	}
1276
1277	ret = ib_post_send_mad(msg, NULL);
1278	if (ret)
1279		cm_free_msg(msg);
1280
1281	return ret;
1282}
1283
1284static inline int cm_is_active_peer(__be64 local_ca_guid, __be64 remote_ca_guid,
1285				    __be32 local_qpn, __be32 remote_qpn)
1286{
1287	return (be64_to_cpu(local_ca_guid) > be64_to_cpu(remote_ca_guid) ||
1288		((local_ca_guid == remote_ca_guid) &&
1289		 (be32_to_cpu(local_qpn) > be32_to_cpu(remote_qpn))));
1290}
1291
1292static void cm_format_paths_from_req(struct cm_req_msg *req_msg,
1293					    struct ib_sa_path_rec *primary_path,
1294					    struct ib_sa_path_rec *alt_path)
1295{
1296	memset(primary_path, 0, sizeof *primary_path);
1297	primary_path->dgid = req_msg->primary_local_gid;
1298	primary_path->sgid = req_msg->primary_remote_gid;
1299	primary_path->dlid = req_msg->primary_local_lid;
1300	primary_path->slid = req_msg->primary_remote_lid;
1301	primary_path->flow_label = cm_req_get_primary_flow_label(req_msg);
1302	primary_path->hop_limit = req_msg->primary_hop_limit;
1303	primary_path->traffic_class = req_msg->primary_traffic_class;
1304	primary_path->reversible = 1;
1305	primary_path->pkey = req_msg->pkey;
1306	primary_path->sl = cm_req_get_primary_sl(req_msg);
1307	primary_path->mtu_selector = IB_SA_EQ;
1308	primary_path->mtu = cm_req_get_path_mtu(req_msg);
1309	primary_path->rate_selector = IB_SA_EQ;
1310	primary_path->rate = cm_req_get_primary_packet_rate(req_msg);
1311	primary_path->packet_life_time_selector = IB_SA_EQ;
1312	primary_path->packet_life_time =
1313		cm_req_get_primary_local_ack_timeout(req_msg);
1314	primary_path->packet_life_time -= (primary_path->packet_life_time > 0);
1315	primary_path->service_id = req_msg->service_id;
1316
1317	if (req_msg->alt_local_lid) {
1318		memset(alt_path, 0, sizeof *alt_path);
1319		alt_path->dgid = req_msg->alt_local_gid;
1320		alt_path->sgid = req_msg->alt_remote_gid;
1321		alt_path->dlid = req_msg->alt_local_lid;
1322		alt_path->slid = req_msg->alt_remote_lid;
1323		alt_path->flow_label = cm_req_get_alt_flow_label(req_msg);
1324		alt_path->hop_limit = req_msg->alt_hop_limit;
1325		alt_path->traffic_class = req_msg->alt_traffic_class;
1326		alt_path->reversible = 1;
1327		alt_path->pkey = req_msg->pkey;
1328		alt_path->sl = cm_req_get_alt_sl(req_msg);
1329		alt_path->mtu_selector = IB_SA_EQ;
1330		alt_path->mtu = cm_req_get_path_mtu(req_msg);
1331		alt_path->rate_selector = IB_SA_EQ;
1332		alt_path->rate = cm_req_get_alt_packet_rate(req_msg);
1333		alt_path->packet_life_time_selector = IB_SA_EQ;
1334		alt_path->packet_life_time =
1335			cm_req_get_alt_local_ack_timeout(req_msg);
1336		alt_path->packet_life_time -= (alt_path->packet_life_time > 0);
1337		alt_path->service_id = req_msg->service_id;
1338	}
1339}
1340
1341static u16 cm_get_bth_pkey(struct cm_work *work)
1342{
1343	struct ib_device *ib_dev = work->port->cm_dev->ib_device;
1344	u8 port_num = work->port->port_num;
1345	u16 pkey_index = work->mad_recv_wc->wc->pkey_index;
1346	u16 pkey;
1347	int ret;
1348
1349	ret = ib_get_cached_pkey(ib_dev, port_num, pkey_index, &pkey);
1350	if (ret) {
1351		dev_warn_ratelimited(&ib_dev->dev, "ib_cm: Couldn't retrieve pkey for incoming request (port %d, pkey index %d). %d\n",
1352				     port_num, pkey_index, ret);
1353		return 0;
1354	}
1355
1356	return pkey;
1357}
1358
1359static void cm_format_req_event(struct cm_work *work,
1360				struct cm_id_private *cm_id_priv,
1361				struct ib_cm_id *listen_id)
1362{
1363	struct cm_req_msg *req_msg;
1364	struct ib_cm_req_event_param *param;
1365
1366	req_msg = (struct cm_req_msg *)work->mad_recv_wc->recv_buf.mad;
1367	param = &work->cm_event.param.req_rcvd;
1368	param->listen_id = listen_id;
1369	param->bth_pkey = cm_get_bth_pkey(work);
1370	param->port = cm_id_priv->av.port->port_num;
1371	param->primary_path = &work->path[0];
1372	if (req_msg->alt_local_lid)
1373		param->alternate_path = &work->path[1];
1374	else
1375		param->alternate_path = NULL;
1376	param->remote_ca_guid = req_msg->local_ca_guid;
1377	param->remote_qkey = be32_to_cpu(req_msg->local_qkey);
1378	param->remote_qpn = be32_to_cpu(cm_req_get_local_qpn(req_msg));
1379	param->qp_type = cm_req_get_qp_type(req_msg);
1380	param->starting_psn = be32_to_cpu(cm_req_get_starting_psn(req_msg));
1381	param->responder_resources = cm_req_get_init_depth(req_msg);
1382	param->initiator_depth = cm_req_get_resp_res(req_msg);
1383	param->local_cm_response_timeout =
1384					cm_req_get_remote_resp_timeout(req_msg);
1385	param->flow_control = cm_req_get_flow_ctrl(req_msg);
1386	param->remote_cm_response_timeout =
1387					cm_req_get_local_resp_timeout(req_msg);
1388	param->retry_count = cm_req_get_retry_count(req_msg);
1389	param->rnr_retry_count = cm_req_get_rnr_retry_count(req_msg);
1390	param->srq = cm_req_get_srq(req_msg);
1391	work->cm_event.private_data = &req_msg->private_data;
1392}
1393
1394static void cm_process_work(struct cm_id_private *cm_id_priv,
1395			    struct cm_work *work)
1396{
1397	int ret;
1398
1399	/* We will typically only have the current event to report. */
1400	ret = cm_id_priv->id.cm_handler(&cm_id_priv->id, &work->cm_event);
1401	cm_free_work(work);
1402
1403	while (!ret && !atomic_add_negative(-1, &cm_id_priv->work_count)) {
1404		spin_lock_irq(&cm_id_priv->lock);
1405		work = cm_dequeue_work(cm_id_priv);
1406		spin_unlock_irq(&cm_id_priv->lock);
1407		BUG_ON(!work);
1408		ret = cm_id_priv->id.cm_handler(&cm_id_priv->id,
1409						&work->cm_event);
1410		cm_free_work(work);
1411	}
1412	cm_deref_id(cm_id_priv);
1413	if (ret)
1414		cm_destroy_id(&cm_id_priv->id, ret);
1415}
1416
1417static void cm_format_mra(struct cm_mra_msg *mra_msg,
1418			  struct cm_id_private *cm_id_priv,
1419			  enum cm_msg_response msg_mraed, u8 service_timeout,
1420			  const void *private_data, u8 private_data_len)
1421{
1422	cm_format_mad_hdr(&mra_msg->hdr, CM_MRA_ATTR_ID, cm_id_priv->tid);
1423	cm_mra_set_msg_mraed(mra_msg, msg_mraed);
1424	mra_msg->local_comm_id = cm_id_priv->id.local_id;
1425	mra_msg->remote_comm_id = cm_id_priv->id.remote_id;
1426	cm_mra_set_service_timeout(mra_msg, service_timeout);
1427
1428	if (private_data && private_data_len)
1429		memcpy(mra_msg->private_data, private_data, private_data_len);
1430}
1431
1432static void cm_format_rej(struct cm_rej_msg *rej_msg,
1433			  struct cm_id_private *cm_id_priv,
1434			  enum ib_cm_rej_reason reason,
1435			  void *ari,
1436			  u8 ari_length,
1437			  const void *private_data,
1438			  u8 private_data_len)
1439{
1440	cm_format_mad_hdr(&rej_msg->hdr, CM_REJ_ATTR_ID, cm_id_priv->tid);
1441	rej_msg->remote_comm_id = cm_id_priv->id.remote_id;
1442
1443	switch(cm_id_priv->id.state) {
1444	case IB_CM_REQ_RCVD:
1445		rej_msg->local_comm_id = 0;
1446		cm_rej_set_msg_rejected(rej_msg, CM_MSG_RESPONSE_REQ);
1447		break;
1448	case IB_CM_MRA_REQ_SENT:
1449		rej_msg->local_comm_id = cm_id_priv->id.local_id;
1450		cm_rej_set_msg_rejected(rej_msg, CM_MSG_RESPONSE_REQ);
1451		break;
1452	case IB_CM_REP_RCVD:
1453	case IB_CM_MRA_REP_SENT:
1454		rej_msg->local_comm_id = cm_id_priv->id.local_id;
1455		cm_rej_set_msg_rejected(rej_msg, CM_MSG_RESPONSE_REP);
1456		break;
1457	default:
1458		rej_msg->local_comm_id = cm_id_priv->id.local_id;
1459		cm_rej_set_msg_rejected(rej_msg, CM_MSG_RESPONSE_OTHER);
1460		break;
1461	}
1462
1463	rej_msg->reason = cpu_to_be16(reason);
1464	if (ari && ari_length) {
1465		cm_rej_set_reject_info_len(rej_msg, ari_length);
1466		memcpy(rej_msg->ari, ari, ari_length);
1467	}
1468
1469	if (private_data && private_data_len)
1470		memcpy(rej_msg->private_data, private_data, private_data_len);
1471}
1472
1473static void cm_dup_req_handler(struct cm_work *work,
1474			       struct cm_id_private *cm_id_priv)
1475{
1476	struct ib_mad_send_buf *msg = NULL;
1477	int ret;
1478
1479	atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
1480			counter[CM_REQ_COUNTER]);
1481
1482	/* Quick state check to discard duplicate REQs. */
1483	if (cm_id_priv->id.state == IB_CM_REQ_RCVD)
1484		return;
1485
1486	ret = cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg);
1487	if (ret)
1488		return;
1489
1490	spin_lock_irq(&cm_id_priv->lock);
1491	switch (cm_id_priv->id.state) {
1492	case IB_CM_MRA_REQ_SENT:
1493		cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
1494			      CM_MSG_RESPONSE_REQ, cm_id_priv->service_timeout,
1495			      cm_id_priv->private_data,
1496			      cm_id_priv->private_data_len);
1497		break;
1498	case IB_CM_TIMEWAIT:
1499		cm_format_rej((struct cm_rej_msg *) msg->mad, cm_id_priv,
1500			      IB_CM_REJ_STALE_CONN, NULL, 0, NULL, 0);
1501		break;
1502	default:
1503		goto unlock;
1504	}
1505	spin_unlock_irq(&cm_id_priv->lock);
1506
1507	ret = ib_post_send_mad(msg, NULL);
1508	if (ret)
1509		goto free;
1510	return;
1511
1512unlock:	spin_unlock_irq(&cm_id_priv->lock);
1513free:	cm_free_msg(msg);
1514}
1515
1516static struct cm_id_private * cm_match_req(struct cm_work *work,
1517					   struct cm_id_private *cm_id_priv)
1518{
1519	struct cm_id_private *listen_cm_id_priv, *cur_cm_id_priv;
1520	struct cm_timewait_info *timewait_info;
1521	struct cm_req_msg *req_msg;
1522
1523	req_msg = (struct cm_req_msg *)work->mad_recv_wc->recv_buf.mad;
1524
1525	/* Check for possible duplicate REQ. */
1526	spin_lock_irq(&cm.lock);
1527	timewait_info = cm_insert_remote_id(cm_id_priv->timewait_info);
1528	if (timewait_info) {
1529		cur_cm_id_priv = cm_get_id(timewait_info->work.local_id,
1530					   timewait_info->work.remote_id);
1531		spin_unlock_irq(&cm.lock);
1532		if (cur_cm_id_priv) {
1533			cm_dup_req_handler(work, cur_cm_id_priv);
1534			cm_deref_id(cur_cm_id_priv);
1535		}
1536		return NULL;
1537	}
1538
1539	/* Check for stale connections. */
1540	timewait_info = cm_insert_remote_qpn(cm_id_priv->timewait_info);
1541	if (timewait_info) {
1542		cm_cleanup_timewait(cm_id_priv->timewait_info);
1543		spin_unlock_irq(&cm.lock);
1544		cm_issue_rej(work->port, work->mad_recv_wc,
1545			     IB_CM_REJ_STALE_CONN, CM_MSG_RESPONSE_REQ,
1546			     NULL, 0);
1547		return NULL;
1548	}
1549
1550	/* Find matching listen request. */
1551	listen_cm_id_priv = cm_find_listen(cm_id_priv->id.device,
1552					   req_msg->service_id);
 
1553	if (!listen_cm_id_priv) {
1554		cm_cleanup_timewait(cm_id_priv->timewait_info);
1555		spin_unlock_irq(&cm.lock);
1556		cm_issue_rej(work->port, work->mad_recv_wc,
1557			     IB_CM_REJ_INVALID_SERVICE_ID, CM_MSG_RESPONSE_REQ,
1558			     NULL, 0);
1559		goto out;
1560	}
1561	atomic_inc(&listen_cm_id_priv->refcount);
1562	atomic_inc(&cm_id_priv->refcount);
1563	cm_id_priv->id.state = IB_CM_REQ_RCVD;
1564	atomic_inc(&cm_id_priv->work_count);
1565	spin_unlock_irq(&cm.lock);
1566out:
1567	return listen_cm_id_priv;
1568}
1569
1570/*
1571 * Work-around for inter-subnet connections.  If the LIDs are permissive,
1572 * we need to override the LID/SL data in the REQ with the LID information
1573 * in the work completion.
1574 */
1575static void cm_process_routed_req(struct cm_req_msg *req_msg, struct ib_wc *wc)
1576{
1577	if (!cm_req_get_primary_subnet_local(req_msg)) {
1578		if (req_msg->primary_local_lid == IB_LID_PERMISSIVE) {
1579			req_msg->primary_local_lid = cpu_to_be16(wc->slid);
1580			cm_req_set_primary_sl(req_msg, wc->sl);
1581		}
1582
1583		if (req_msg->primary_remote_lid == IB_LID_PERMISSIVE)
1584			req_msg->primary_remote_lid = cpu_to_be16(wc->dlid_path_bits);
1585	}
1586
1587	if (!cm_req_get_alt_subnet_local(req_msg)) {
1588		if (req_msg->alt_local_lid == IB_LID_PERMISSIVE) {
1589			req_msg->alt_local_lid = cpu_to_be16(wc->slid);
1590			cm_req_set_alt_sl(req_msg, wc->sl);
1591		}
1592
1593		if (req_msg->alt_remote_lid == IB_LID_PERMISSIVE)
1594			req_msg->alt_remote_lid = cpu_to_be16(wc->dlid_path_bits);
1595	}
1596}
1597
1598static int cm_req_handler(struct cm_work *work)
1599{
1600	struct ib_cm_id *cm_id;
1601	struct cm_id_private *cm_id_priv, *listen_cm_id_priv;
1602	struct cm_req_msg *req_msg;
1603	union ib_gid gid;
1604	struct ib_gid_attr gid_attr;
1605	int ret;
1606
1607	req_msg = (struct cm_req_msg *)work->mad_recv_wc->recv_buf.mad;
1608
1609	cm_id = ib_create_cm_id(work->port->cm_dev->ib_device, NULL, NULL);
1610	if (IS_ERR(cm_id))
1611		return PTR_ERR(cm_id);
1612
1613	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1614	cm_id_priv->id.remote_id = req_msg->local_comm_id;
1615	cm_init_av_for_response(work->port, work->mad_recv_wc->wc,
1616				work->mad_recv_wc->recv_buf.grh,
1617				&cm_id_priv->av);
1618	cm_id_priv->timewait_info = cm_create_timewait_info(cm_id_priv->
1619							    id.local_id);
1620	if (IS_ERR(cm_id_priv->timewait_info)) {
1621		ret = PTR_ERR(cm_id_priv->timewait_info);
1622		goto destroy;
1623	}
1624	cm_id_priv->timewait_info->work.remote_id = req_msg->local_comm_id;
1625	cm_id_priv->timewait_info->remote_ca_guid = req_msg->local_ca_guid;
1626	cm_id_priv->timewait_info->remote_qpn = cm_req_get_local_qpn(req_msg);
1627
1628	listen_cm_id_priv = cm_match_req(work, cm_id_priv);
1629	if (!listen_cm_id_priv) {
1630		ret = -EINVAL;
1631		kfree(cm_id_priv->timewait_info);
1632		goto destroy;
1633	}
1634
1635	cm_id_priv->id.cm_handler = listen_cm_id_priv->id.cm_handler;
1636	cm_id_priv->id.context = listen_cm_id_priv->id.context;
1637	cm_id_priv->id.service_id = req_msg->service_id;
1638	cm_id_priv->id.service_mask = ~cpu_to_be64(0);
1639
1640	cm_process_routed_req(req_msg, work->mad_recv_wc->wc);
1641	cm_format_paths_from_req(req_msg, &work->path[0], &work->path[1]);
1642
1643	memcpy(work->path[0].dmac, cm_id_priv->av.ah_attr.dmac, ETH_ALEN);
1644	work->path[0].hop_limit = cm_id_priv->av.ah_attr.grh.hop_limit;
1645	ret = ib_get_cached_gid(work->port->cm_dev->ib_device,
1646				work->port->port_num,
1647				cm_id_priv->av.ah_attr.grh.sgid_index,
1648				&gid, &gid_attr);
1649	if (!ret) {
1650		if (gid_attr.ndev) {
1651			work->path[0].ifindex = gid_attr.ndev->ifindex;
1652			work->path[0].net = dev_net(gid_attr.ndev);
1653			dev_put(gid_attr.ndev);
1654		}
1655		work->path[0].gid_type = gid_attr.gid_type;
1656		ret = cm_init_av_by_path(&work->path[0], &cm_id_priv->av);
1657	}
1658	if (ret) {
1659		int err = ib_get_cached_gid(work->port->cm_dev->ib_device,
1660					    work->port->port_num, 0,
1661					    &work->path[0].sgid,
1662					    &gid_attr);
1663		if (!err && gid_attr.ndev) {
1664			work->path[0].ifindex = gid_attr.ndev->ifindex;
1665			work->path[0].net = dev_net(gid_attr.ndev);
1666			dev_put(gid_attr.ndev);
1667		}
1668		work->path[0].gid_type = gid_attr.gid_type;
1669		ib_send_cm_rej(cm_id, IB_CM_REJ_INVALID_GID,
1670			       &work->path[0].sgid, sizeof work->path[0].sgid,
1671			       NULL, 0);
1672		goto rejected;
1673	}
1674	if (req_msg->alt_local_lid) {
1675		ret = cm_init_av_by_path(&work->path[1], &cm_id_priv->alt_av);
1676		if (ret) {
1677			ib_send_cm_rej(cm_id, IB_CM_REJ_INVALID_ALT_GID,
1678				       &work->path[0].sgid,
1679				       sizeof work->path[0].sgid, NULL, 0);
1680			goto rejected;
1681		}
1682	}
1683	cm_id_priv->tid = req_msg->hdr.tid;
1684	cm_id_priv->timeout_ms = cm_convert_to_ms(
1685					cm_req_get_local_resp_timeout(req_msg));
1686	cm_id_priv->max_cm_retries = cm_req_get_max_cm_retries(req_msg);
1687	cm_id_priv->remote_qpn = cm_req_get_local_qpn(req_msg);
1688	cm_id_priv->initiator_depth = cm_req_get_resp_res(req_msg);
1689	cm_id_priv->responder_resources = cm_req_get_init_depth(req_msg);
1690	cm_id_priv->path_mtu = cm_req_get_path_mtu(req_msg);
1691	cm_id_priv->pkey = req_msg->pkey;
1692	cm_id_priv->sq_psn = cm_req_get_starting_psn(req_msg);
1693	cm_id_priv->retry_count = cm_req_get_retry_count(req_msg);
1694	cm_id_priv->rnr_retry_count = cm_req_get_rnr_retry_count(req_msg);
1695	cm_id_priv->qp_type = cm_req_get_qp_type(req_msg);
1696
1697	cm_format_req_event(work, cm_id_priv, &listen_cm_id_priv->id);
1698	cm_process_work(cm_id_priv, work);
1699	cm_deref_id(listen_cm_id_priv);
1700	return 0;
1701
1702rejected:
1703	atomic_dec(&cm_id_priv->refcount);
1704	cm_deref_id(listen_cm_id_priv);
1705destroy:
1706	ib_destroy_cm_id(cm_id);
1707	return ret;
1708}
1709
1710static void cm_format_rep(struct cm_rep_msg *rep_msg,
1711			  struct cm_id_private *cm_id_priv,
1712			  struct ib_cm_rep_param *param)
1713{
1714	cm_format_mad_hdr(&rep_msg->hdr, CM_REP_ATTR_ID, cm_id_priv->tid);
1715	rep_msg->local_comm_id = cm_id_priv->id.local_id;
1716	rep_msg->remote_comm_id = cm_id_priv->id.remote_id;
 
1717	cm_rep_set_starting_psn(rep_msg, cpu_to_be32(param->starting_psn));
1718	rep_msg->resp_resources = param->responder_resources;
 
1719	cm_rep_set_target_ack_delay(rep_msg,
1720				    cm_id_priv->av.port->cm_dev->ack_delay);
1721	cm_rep_set_failover(rep_msg, param->failover_accepted);
 
1722	cm_rep_set_rnr_retry_count(rep_msg, param->rnr_retry_count);
 
1723	rep_msg->local_ca_guid = cm_id_priv->id.device->node_guid;
1724
1725	if (cm_id_priv->qp_type != IB_QPT_XRC_TGT) {
1726		rep_msg->initiator_depth = param->initiator_depth;
1727		cm_rep_set_flow_ctrl(rep_msg, param->flow_control);
1728		cm_rep_set_srq(rep_msg, param->srq);
1729		cm_rep_set_local_qpn(rep_msg, cpu_to_be32(param->qp_num));
1730	} else {
1731		cm_rep_set_srq(rep_msg, 1);
1732		cm_rep_set_local_eecn(rep_msg, cpu_to_be32(param->qp_num));
1733	}
1734
1735	if (param->private_data && param->private_data_len)
1736		memcpy(rep_msg->private_data, param->private_data,
1737		       param->private_data_len);
1738}
1739
1740int ib_send_cm_rep(struct ib_cm_id *cm_id,
1741		   struct ib_cm_rep_param *param)
1742{
1743	struct cm_id_private *cm_id_priv;
1744	struct ib_mad_send_buf *msg;
1745	struct cm_rep_msg *rep_msg;
1746	unsigned long flags;
1747	int ret;
1748
1749	if (param->private_data &&
1750	    param->private_data_len > IB_CM_REP_PRIVATE_DATA_SIZE)
1751		return -EINVAL;
1752
1753	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1754	spin_lock_irqsave(&cm_id_priv->lock, flags);
1755	if (cm_id->state != IB_CM_REQ_RCVD &&
1756	    cm_id->state != IB_CM_MRA_REQ_SENT) {
1757		ret = -EINVAL;
1758		goto out;
1759	}
1760
1761	ret = cm_alloc_msg(cm_id_priv, &msg);
1762	if (ret)
1763		goto out;
1764
1765	rep_msg = (struct cm_rep_msg *) msg->mad;
1766	cm_format_rep(rep_msg, cm_id_priv, param);
1767	msg->timeout_ms = cm_id_priv->timeout_ms;
1768	msg->context[1] = (void *) (unsigned long) IB_CM_REP_SENT;
1769
1770	ret = ib_post_send_mad(msg, NULL);
1771	if (ret) {
1772		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1773		cm_free_msg(msg);
1774		return ret;
1775	}
1776
1777	cm_id->state = IB_CM_REP_SENT;
1778	cm_id_priv->msg = msg;
1779	cm_id_priv->initiator_depth = param->initiator_depth;
1780	cm_id_priv->responder_resources = param->responder_resources;
1781	cm_id_priv->rq_psn = cm_rep_get_starting_psn(rep_msg);
1782	cm_id_priv->local_qpn = cpu_to_be32(param->qp_num & 0xFFFFFF);
1783
1784out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1785	return ret;
1786}
1787EXPORT_SYMBOL(ib_send_cm_rep);
1788
1789static void cm_format_rtu(struct cm_rtu_msg *rtu_msg,
1790			  struct cm_id_private *cm_id_priv,
1791			  const void *private_data,
1792			  u8 private_data_len)
1793{
1794	cm_format_mad_hdr(&rtu_msg->hdr, CM_RTU_ATTR_ID, cm_id_priv->tid);
1795	rtu_msg->local_comm_id = cm_id_priv->id.local_id;
1796	rtu_msg->remote_comm_id = cm_id_priv->id.remote_id;
1797
1798	if (private_data && private_data_len)
1799		memcpy(rtu_msg->private_data, private_data, private_data_len);
1800}
1801
1802int ib_send_cm_rtu(struct ib_cm_id *cm_id,
1803		   const void *private_data,
1804		   u8 private_data_len)
1805{
1806	struct cm_id_private *cm_id_priv;
1807	struct ib_mad_send_buf *msg;
1808	unsigned long flags;
1809	void *data;
1810	int ret;
1811
1812	if (private_data && private_data_len > IB_CM_RTU_PRIVATE_DATA_SIZE)
1813		return -EINVAL;
1814
1815	data = cm_copy_private_data(private_data, private_data_len);
1816	if (IS_ERR(data))
1817		return PTR_ERR(data);
1818
1819	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1820	spin_lock_irqsave(&cm_id_priv->lock, flags);
1821	if (cm_id->state != IB_CM_REP_RCVD &&
1822	    cm_id->state != IB_CM_MRA_REP_SENT) {
1823		ret = -EINVAL;
1824		goto error;
1825	}
1826
1827	ret = cm_alloc_msg(cm_id_priv, &msg);
1828	if (ret)
1829		goto error;
1830
1831	cm_format_rtu((struct cm_rtu_msg *) msg->mad, cm_id_priv,
1832		      private_data, private_data_len);
1833
1834	ret = ib_post_send_mad(msg, NULL);
1835	if (ret) {
1836		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1837		cm_free_msg(msg);
1838		kfree(data);
1839		return ret;
1840	}
1841
1842	cm_id->state = IB_CM_ESTABLISHED;
1843	cm_set_private_data(cm_id_priv, data, private_data_len);
1844	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1845	return 0;
1846
1847error:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1848	kfree(data);
1849	return ret;
1850}
1851EXPORT_SYMBOL(ib_send_cm_rtu);
1852
1853static void cm_format_rep_event(struct cm_work *work, enum ib_qp_type qp_type)
1854{
1855	struct cm_rep_msg *rep_msg;
1856	struct ib_cm_rep_event_param *param;
1857
1858	rep_msg = (struct cm_rep_msg *)work->mad_recv_wc->recv_buf.mad;
1859	param = &work->cm_event.param.rep_rcvd;
1860	param->remote_ca_guid = rep_msg->local_ca_guid;
1861	param->remote_qkey = be32_to_cpu(rep_msg->local_qkey);
1862	param->remote_qpn = be32_to_cpu(cm_rep_get_qpn(rep_msg, qp_type));
1863	param->starting_psn = be32_to_cpu(cm_rep_get_starting_psn(rep_msg));
1864	param->responder_resources = rep_msg->initiator_depth;
1865	param->initiator_depth = rep_msg->resp_resources;
1866	param->target_ack_delay = cm_rep_get_target_ack_delay(rep_msg);
1867	param->failover_accepted = cm_rep_get_failover(rep_msg);
1868	param->flow_control = cm_rep_get_flow_ctrl(rep_msg);
1869	param->rnr_retry_count = cm_rep_get_rnr_retry_count(rep_msg);
1870	param->srq = cm_rep_get_srq(rep_msg);
1871	work->cm_event.private_data = &rep_msg->private_data;
1872}
1873
1874static void cm_dup_rep_handler(struct cm_work *work)
1875{
1876	struct cm_id_private *cm_id_priv;
1877	struct cm_rep_msg *rep_msg;
1878	struct ib_mad_send_buf *msg = NULL;
1879	int ret;
1880
1881	rep_msg = (struct cm_rep_msg *) work->mad_recv_wc->recv_buf.mad;
1882	cm_id_priv = cm_acquire_id(rep_msg->remote_comm_id,
1883				   rep_msg->local_comm_id);
1884	if (!cm_id_priv)
1885		return;
1886
1887	atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
1888			counter[CM_REP_COUNTER]);
1889	ret = cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg);
1890	if (ret)
1891		goto deref;
1892
1893	spin_lock_irq(&cm_id_priv->lock);
1894	if (cm_id_priv->id.state == IB_CM_ESTABLISHED)
1895		cm_format_rtu((struct cm_rtu_msg *) msg->mad, cm_id_priv,
1896			      cm_id_priv->private_data,
1897			      cm_id_priv->private_data_len);
1898	else if (cm_id_priv->id.state == IB_CM_MRA_REP_SENT)
1899		cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
1900			      CM_MSG_RESPONSE_REP, cm_id_priv->service_timeout,
1901			      cm_id_priv->private_data,
1902			      cm_id_priv->private_data_len);
1903	else
1904		goto unlock;
1905	spin_unlock_irq(&cm_id_priv->lock);
1906
1907	ret = ib_post_send_mad(msg, NULL);
1908	if (ret)
1909		goto free;
1910	goto deref;
1911
1912unlock:	spin_unlock_irq(&cm_id_priv->lock);
1913free:	cm_free_msg(msg);
1914deref:	cm_deref_id(cm_id_priv);
1915}
1916
1917static int cm_rep_handler(struct cm_work *work)
1918{
1919	struct cm_id_private *cm_id_priv;
1920	struct cm_rep_msg *rep_msg;
1921	int ret;
1922
1923	rep_msg = (struct cm_rep_msg *)work->mad_recv_wc->recv_buf.mad;
1924	cm_id_priv = cm_acquire_id(rep_msg->remote_comm_id, 0);
1925	if (!cm_id_priv) {
1926		cm_dup_rep_handler(work);
1927		return -EINVAL;
1928	}
1929
1930	cm_format_rep_event(work, cm_id_priv->qp_type);
1931
1932	spin_lock_irq(&cm_id_priv->lock);
1933	switch (cm_id_priv->id.state) {
1934	case IB_CM_REQ_SENT:
1935	case IB_CM_MRA_REQ_RCVD:
1936		break;
1937	default:
1938		spin_unlock_irq(&cm_id_priv->lock);
1939		ret = -EINVAL;
1940		goto error;
1941	}
1942
1943	cm_id_priv->timewait_info->work.remote_id = rep_msg->local_comm_id;
1944	cm_id_priv->timewait_info->remote_ca_guid = rep_msg->local_ca_guid;
1945	cm_id_priv->timewait_info->remote_qpn = cm_rep_get_qpn(rep_msg, cm_id_priv->qp_type);
1946
1947	spin_lock(&cm.lock);
1948	/* Check for duplicate REP. */
1949	if (cm_insert_remote_id(cm_id_priv->timewait_info)) {
1950		spin_unlock(&cm.lock);
1951		spin_unlock_irq(&cm_id_priv->lock);
1952		ret = -EINVAL;
1953		goto error;
1954	}
1955	/* Check for a stale connection. */
1956	if (cm_insert_remote_qpn(cm_id_priv->timewait_info)) {
1957		rb_erase(&cm_id_priv->timewait_info->remote_id_node,
1958			 &cm.remote_id_table);
1959		cm_id_priv->timewait_info->inserted_remote_id = 0;
1960		spin_unlock(&cm.lock);
1961		spin_unlock_irq(&cm_id_priv->lock);
1962		cm_issue_rej(work->port, work->mad_recv_wc,
1963			     IB_CM_REJ_STALE_CONN, CM_MSG_RESPONSE_REP,
1964			     NULL, 0);
1965		ret = -EINVAL;
1966		goto error;
1967	}
1968	spin_unlock(&cm.lock);
1969
1970	cm_id_priv->id.state = IB_CM_REP_RCVD;
1971	cm_id_priv->id.remote_id = rep_msg->local_comm_id;
1972	cm_id_priv->remote_qpn = cm_rep_get_qpn(rep_msg, cm_id_priv->qp_type);
1973	cm_id_priv->initiator_depth = rep_msg->resp_resources;
1974	cm_id_priv->responder_resources = rep_msg->initiator_depth;
1975	cm_id_priv->sq_psn = cm_rep_get_starting_psn(rep_msg);
1976	cm_id_priv->rnr_retry_count = cm_rep_get_rnr_retry_count(rep_msg);
1977	cm_id_priv->target_ack_delay = cm_rep_get_target_ack_delay(rep_msg);
1978	cm_id_priv->av.timeout =
1979			cm_ack_timeout(cm_id_priv->target_ack_delay,
1980				       cm_id_priv->av.timeout - 1);
1981	cm_id_priv->alt_av.timeout =
1982			cm_ack_timeout(cm_id_priv->target_ack_delay,
1983				       cm_id_priv->alt_av.timeout - 1);
1984
1985	/* todo: handle peer_to_peer */
1986
1987	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
1988	ret = atomic_inc_and_test(&cm_id_priv->work_count);
1989	if (!ret)
1990		list_add_tail(&work->list, &cm_id_priv->work_list);
1991	spin_unlock_irq(&cm_id_priv->lock);
1992
1993	if (ret)
1994		cm_process_work(cm_id_priv, work);
1995	else
1996		cm_deref_id(cm_id_priv);
1997	return 0;
1998
1999error:
2000	cm_deref_id(cm_id_priv);
2001	return ret;
2002}
2003
2004static int cm_establish_handler(struct cm_work *work)
2005{
2006	struct cm_id_private *cm_id_priv;
2007	int ret;
2008
2009	/* See comment in cm_establish about lookup. */
2010	cm_id_priv = cm_acquire_id(work->local_id, work->remote_id);
2011	if (!cm_id_priv)
2012		return -EINVAL;
2013
2014	spin_lock_irq(&cm_id_priv->lock);
2015	if (cm_id_priv->id.state != IB_CM_ESTABLISHED) {
2016		spin_unlock_irq(&cm_id_priv->lock);
2017		goto out;
2018	}
2019
2020	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2021	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2022	if (!ret)
2023		list_add_tail(&work->list, &cm_id_priv->work_list);
2024	spin_unlock_irq(&cm_id_priv->lock);
2025
2026	if (ret)
2027		cm_process_work(cm_id_priv, work);
2028	else
2029		cm_deref_id(cm_id_priv);
2030	return 0;
2031out:
2032	cm_deref_id(cm_id_priv);
2033	return -EINVAL;
2034}
2035
2036static int cm_rtu_handler(struct cm_work *work)
2037{
2038	struct cm_id_private *cm_id_priv;
2039	struct cm_rtu_msg *rtu_msg;
2040	int ret;
2041
2042	rtu_msg = (struct cm_rtu_msg *)work->mad_recv_wc->recv_buf.mad;
2043	cm_id_priv = cm_acquire_id(rtu_msg->remote_comm_id,
2044				   rtu_msg->local_comm_id);
2045	if (!cm_id_priv)
2046		return -EINVAL;
2047
2048	work->cm_event.private_data = &rtu_msg->private_data;
2049
2050	spin_lock_irq(&cm_id_priv->lock);
2051	if (cm_id_priv->id.state != IB_CM_REP_SENT &&
2052	    cm_id_priv->id.state != IB_CM_MRA_REP_RCVD) {
2053		spin_unlock_irq(&cm_id_priv->lock);
2054		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2055				counter[CM_RTU_COUNTER]);
2056		goto out;
2057	}
2058	cm_id_priv->id.state = IB_CM_ESTABLISHED;
2059
2060	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2061	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2062	if (!ret)
2063		list_add_tail(&work->list, &cm_id_priv->work_list);
2064	spin_unlock_irq(&cm_id_priv->lock);
2065
2066	if (ret)
2067		cm_process_work(cm_id_priv, work);
2068	else
2069		cm_deref_id(cm_id_priv);
2070	return 0;
2071out:
2072	cm_deref_id(cm_id_priv);
2073	return -EINVAL;
2074}
2075
2076static void cm_format_dreq(struct cm_dreq_msg *dreq_msg,
2077			  struct cm_id_private *cm_id_priv,
2078			  const void *private_data,
2079			  u8 private_data_len)
2080{
2081	cm_format_mad_hdr(&dreq_msg->hdr, CM_DREQ_ATTR_ID,
2082			  cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_DREQ));
2083	dreq_msg->local_comm_id = cm_id_priv->id.local_id;
2084	dreq_msg->remote_comm_id = cm_id_priv->id.remote_id;
2085	cm_dreq_set_remote_qpn(dreq_msg, cm_id_priv->remote_qpn);
2086
2087	if (private_data && private_data_len)
2088		memcpy(dreq_msg->private_data, private_data, private_data_len);
2089}
2090
2091int ib_send_cm_dreq(struct ib_cm_id *cm_id,
2092		    const void *private_data,
2093		    u8 private_data_len)
2094{
2095	struct cm_id_private *cm_id_priv;
2096	struct ib_mad_send_buf *msg;
2097	unsigned long flags;
2098	int ret;
2099
2100	if (private_data && private_data_len > IB_CM_DREQ_PRIVATE_DATA_SIZE)
2101		return -EINVAL;
2102
2103	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2104	spin_lock_irqsave(&cm_id_priv->lock, flags);
2105	if (cm_id->state != IB_CM_ESTABLISHED) {
2106		ret = -EINVAL;
2107		goto out;
2108	}
2109
2110	if (cm_id->lap_state == IB_CM_LAP_SENT ||
2111	    cm_id->lap_state == IB_CM_MRA_LAP_RCVD)
2112		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2113
2114	ret = cm_alloc_msg(cm_id_priv, &msg);
2115	if (ret) {
2116		cm_enter_timewait(cm_id_priv);
2117		goto out;
2118	}
2119
2120	cm_format_dreq((struct cm_dreq_msg *) msg->mad, cm_id_priv,
2121		       private_data, private_data_len);
2122	msg->timeout_ms = cm_id_priv->timeout_ms;
2123	msg->context[1] = (void *) (unsigned long) IB_CM_DREQ_SENT;
2124
2125	ret = ib_post_send_mad(msg, NULL);
2126	if (ret) {
2127		cm_enter_timewait(cm_id_priv);
2128		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2129		cm_free_msg(msg);
2130		return ret;
2131	}
2132
2133	cm_id->state = IB_CM_DREQ_SENT;
2134	cm_id_priv->msg = msg;
2135out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2136	return ret;
2137}
2138EXPORT_SYMBOL(ib_send_cm_dreq);
2139
2140static void cm_format_drep(struct cm_drep_msg *drep_msg,
2141			  struct cm_id_private *cm_id_priv,
2142			  const void *private_data,
2143			  u8 private_data_len)
2144{
2145	cm_format_mad_hdr(&drep_msg->hdr, CM_DREP_ATTR_ID, cm_id_priv->tid);
2146	drep_msg->local_comm_id = cm_id_priv->id.local_id;
2147	drep_msg->remote_comm_id = cm_id_priv->id.remote_id;
2148
2149	if (private_data && private_data_len)
2150		memcpy(drep_msg->private_data, private_data, private_data_len);
2151}
2152
2153int ib_send_cm_drep(struct ib_cm_id *cm_id,
2154		    const void *private_data,
2155		    u8 private_data_len)
2156{
2157	struct cm_id_private *cm_id_priv;
2158	struct ib_mad_send_buf *msg;
2159	unsigned long flags;
2160	void *data;
2161	int ret;
2162
2163	if (private_data && private_data_len > IB_CM_DREP_PRIVATE_DATA_SIZE)
2164		return -EINVAL;
2165
2166	data = cm_copy_private_data(private_data, private_data_len);
2167	if (IS_ERR(data))
2168		return PTR_ERR(data);
2169
2170	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2171	spin_lock_irqsave(&cm_id_priv->lock, flags);
2172	if (cm_id->state != IB_CM_DREQ_RCVD) {
2173		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2174		kfree(data);
2175		return -EINVAL;
2176	}
2177
2178	cm_set_private_data(cm_id_priv, data, private_data_len);
2179	cm_enter_timewait(cm_id_priv);
2180
2181	ret = cm_alloc_msg(cm_id_priv, &msg);
2182	if (ret)
2183		goto out;
2184
2185	cm_format_drep((struct cm_drep_msg *) msg->mad, cm_id_priv,
2186		       private_data, private_data_len);
2187
2188	ret = ib_post_send_mad(msg, NULL);
2189	if (ret) {
2190		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2191		cm_free_msg(msg);
2192		return ret;
2193	}
2194
2195out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2196	return ret;
2197}
2198EXPORT_SYMBOL(ib_send_cm_drep);
2199
2200static int cm_issue_drep(struct cm_port *port,
2201			 struct ib_mad_recv_wc *mad_recv_wc)
2202{
2203	struct ib_mad_send_buf *msg = NULL;
2204	struct cm_dreq_msg *dreq_msg;
2205	struct cm_drep_msg *drep_msg;
2206	int ret;
2207
2208	ret = cm_alloc_response_msg(port, mad_recv_wc, &msg);
2209	if (ret)
2210		return ret;
2211
2212	dreq_msg = (struct cm_dreq_msg *) mad_recv_wc->recv_buf.mad;
2213	drep_msg = (struct cm_drep_msg *) msg->mad;
2214
2215	cm_format_mad_hdr(&drep_msg->hdr, CM_DREP_ATTR_ID, dreq_msg->hdr.tid);
2216	drep_msg->remote_comm_id = dreq_msg->local_comm_id;
2217	drep_msg->local_comm_id = dreq_msg->remote_comm_id;
2218
2219	ret = ib_post_send_mad(msg, NULL);
2220	if (ret)
2221		cm_free_msg(msg);
2222
2223	return ret;
2224}
2225
2226static int cm_dreq_handler(struct cm_work *work)
2227{
2228	struct cm_id_private *cm_id_priv;
2229	struct cm_dreq_msg *dreq_msg;
2230	struct ib_mad_send_buf *msg = NULL;
2231	int ret;
2232
2233	dreq_msg = (struct cm_dreq_msg *)work->mad_recv_wc->recv_buf.mad;
2234	cm_id_priv = cm_acquire_id(dreq_msg->remote_comm_id,
2235				   dreq_msg->local_comm_id);
2236	if (!cm_id_priv) {
2237		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2238				counter[CM_DREQ_COUNTER]);
2239		cm_issue_drep(work->port, work->mad_recv_wc);
2240		return -EINVAL;
2241	}
2242
2243	work->cm_event.private_data = &dreq_msg->private_data;
2244
2245	spin_lock_irq(&cm_id_priv->lock);
2246	if (cm_id_priv->local_qpn != cm_dreq_get_remote_qpn(dreq_msg))
2247		goto unlock;
2248
2249	switch (cm_id_priv->id.state) {
2250	case IB_CM_REP_SENT:
2251	case IB_CM_DREQ_SENT:
2252		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2253		break;
2254	case IB_CM_ESTABLISHED:
2255		if (cm_id_priv->id.lap_state == IB_CM_LAP_SENT ||
2256		    cm_id_priv->id.lap_state == IB_CM_MRA_LAP_RCVD)
2257			ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2258		break;
2259	case IB_CM_MRA_REP_RCVD:
2260		break;
2261	case IB_CM_TIMEWAIT:
2262		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2263				counter[CM_DREQ_COUNTER]);
2264		if (cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg))
2265			goto unlock;
2266
2267		cm_format_drep((struct cm_drep_msg *) msg->mad, cm_id_priv,
2268			       cm_id_priv->private_data,
2269			       cm_id_priv->private_data_len);
2270		spin_unlock_irq(&cm_id_priv->lock);
2271
2272		if (ib_post_send_mad(msg, NULL))
2273			cm_free_msg(msg);
2274		goto deref;
2275	case IB_CM_DREQ_RCVD:
2276		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2277				counter[CM_DREQ_COUNTER]);
2278		goto unlock;
2279	default:
2280		goto unlock;
2281	}
2282	cm_id_priv->id.state = IB_CM_DREQ_RCVD;
2283	cm_id_priv->tid = dreq_msg->hdr.tid;
2284	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2285	if (!ret)
2286		list_add_tail(&work->list, &cm_id_priv->work_list);
2287	spin_unlock_irq(&cm_id_priv->lock);
2288
2289	if (ret)
2290		cm_process_work(cm_id_priv, work);
2291	else
2292		cm_deref_id(cm_id_priv);
2293	return 0;
2294
2295unlock:	spin_unlock_irq(&cm_id_priv->lock);
2296deref:	cm_deref_id(cm_id_priv);
2297	return -EINVAL;
2298}
2299
2300static int cm_drep_handler(struct cm_work *work)
2301{
2302	struct cm_id_private *cm_id_priv;
2303	struct cm_drep_msg *drep_msg;
2304	int ret;
2305
2306	drep_msg = (struct cm_drep_msg *)work->mad_recv_wc->recv_buf.mad;
2307	cm_id_priv = cm_acquire_id(drep_msg->remote_comm_id,
2308				   drep_msg->local_comm_id);
2309	if (!cm_id_priv)
2310		return -EINVAL;
2311
2312	work->cm_event.private_data = &drep_msg->private_data;
2313
2314	spin_lock_irq(&cm_id_priv->lock);
2315	if (cm_id_priv->id.state != IB_CM_DREQ_SENT &&
2316	    cm_id_priv->id.state != IB_CM_DREQ_RCVD) {
2317		spin_unlock_irq(&cm_id_priv->lock);
2318		goto out;
2319	}
2320	cm_enter_timewait(cm_id_priv);
2321
2322	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2323	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2324	if (!ret)
2325		list_add_tail(&work->list, &cm_id_priv->work_list);
2326	spin_unlock_irq(&cm_id_priv->lock);
2327
2328	if (ret)
2329		cm_process_work(cm_id_priv, work);
2330	else
2331		cm_deref_id(cm_id_priv);
2332	return 0;
2333out:
2334	cm_deref_id(cm_id_priv);
2335	return -EINVAL;
2336}
2337
2338int ib_send_cm_rej(struct ib_cm_id *cm_id,
2339		   enum ib_cm_rej_reason reason,
2340		   void *ari,
2341		   u8 ari_length,
2342		   const void *private_data,
2343		   u8 private_data_len)
2344{
2345	struct cm_id_private *cm_id_priv;
2346	struct ib_mad_send_buf *msg;
2347	unsigned long flags;
2348	int ret;
2349
2350	if ((private_data && private_data_len > IB_CM_REJ_PRIVATE_DATA_SIZE) ||
2351	    (ari && ari_length > IB_CM_REJ_ARI_LENGTH))
2352		return -EINVAL;
2353
2354	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2355
2356	spin_lock_irqsave(&cm_id_priv->lock, flags);
2357	switch (cm_id->state) {
2358	case IB_CM_REQ_SENT:
2359	case IB_CM_MRA_REQ_RCVD:
2360	case IB_CM_REQ_RCVD:
2361	case IB_CM_MRA_REQ_SENT:
2362	case IB_CM_REP_RCVD:
2363	case IB_CM_MRA_REP_SENT:
2364		ret = cm_alloc_msg(cm_id_priv, &msg);
2365		if (!ret)
2366			cm_format_rej((struct cm_rej_msg *) msg->mad,
2367				      cm_id_priv, reason, ari, ari_length,
2368				      private_data, private_data_len);
2369
2370		cm_reset_to_idle(cm_id_priv);
2371		break;
2372	case IB_CM_REP_SENT:
2373	case IB_CM_MRA_REP_RCVD:
2374		ret = cm_alloc_msg(cm_id_priv, &msg);
2375		if (!ret)
2376			cm_format_rej((struct cm_rej_msg *) msg->mad,
2377				      cm_id_priv, reason, ari, ari_length,
2378				      private_data, private_data_len);
2379
2380		cm_enter_timewait(cm_id_priv);
2381		break;
2382	default:
2383		ret = -EINVAL;
2384		goto out;
2385	}
2386
2387	if (ret)
2388		goto out;
2389
2390	ret = ib_post_send_mad(msg, NULL);
2391	if (ret)
2392		cm_free_msg(msg);
2393
2394out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2395	return ret;
2396}
2397EXPORT_SYMBOL(ib_send_cm_rej);
2398
2399static void cm_format_rej_event(struct cm_work *work)
2400{
2401	struct cm_rej_msg *rej_msg;
2402	struct ib_cm_rej_event_param *param;
2403
2404	rej_msg = (struct cm_rej_msg *)work->mad_recv_wc->recv_buf.mad;
2405	param = &work->cm_event.param.rej_rcvd;
2406	param->ari = rej_msg->ari;
2407	param->ari_length = cm_rej_get_reject_info_len(rej_msg);
2408	param->reason = __be16_to_cpu(rej_msg->reason);
2409	work->cm_event.private_data = &rej_msg->private_data;
2410}
2411
2412static struct cm_id_private * cm_acquire_rejected_id(struct cm_rej_msg *rej_msg)
2413{
2414	struct cm_timewait_info *timewait_info;
2415	struct cm_id_private *cm_id_priv;
2416	__be32 remote_id;
2417
2418	remote_id = rej_msg->local_comm_id;
2419
2420	if (__be16_to_cpu(rej_msg->reason) == IB_CM_REJ_TIMEOUT) {
2421		spin_lock_irq(&cm.lock);
2422		timewait_info = cm_find_remote_id( *((__be64 *) rej_msg->ari),
2423						  remote_id);
2424		if (!timewait_info) {
2425			spin_unlock_irq(&cm.lock);
2426			return NULL;
2427		}
2428		cm_id_priv = idr_find(&cm.local_id_table, (__force int)
2429				      (timewait_info->work.local_id ^
2430				       cm.random_id_operand));
2431		if (cm_id_priv) {
2432			if (cm_id_priv->id.remote_id == remote_id)
2433				atomic_inc(&cm_id_priv->refcount);
2434			else
2435				cm_id_priv = NULL;
2436		}
2437		spin_unlock_irq(&cm.lock);
2438	} else if (cm_rej_get_msg_rejected(rej_msg) == CM_MSG_RESPONSE_REQ)
2439		cm_id_priv = cm_acquire_id(rej_msg->remote_comm_id, 0);
2440	else
2441		cm_id_priv = cm_acquire_id(rej_msg->remote_comm_id, remote_id);
2442
2443	return cm_id_priv;
2444}
2445
2446static int cm_rej_handler(struct cm_work *work)
2447{
2448	struct cm_id_private *cm_id_priv;
2449	struct cm_rej_msg *rej_msg;
2450	int ret;
2451
2452	rej_msg = (struct cm_rej_msg *)work->mad_recv_wc->recv_buf.mad;
2453	cm_id_priv = cm_acquire_rejected_id(rej_msg);
2454	if (!cm_id_priv)
2455		return -EINVAL;
2456
2457	cm_format_rej_event(work);
2458
2459	spin_lock_irq(&cm_id_priv->lock);
2460	switch (cm_id_priv->id.state) {
2461	case IB_CM_REQ_SENT:
2462	case IB_CM_MRA_REQ_RCVD:
2463	case IB_CM_REP_SENT:
2464	case IB_CM_MRA_REP_RCVD:
2465		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2466		/* fall through */
2467	case IB_CM_REQ_RCVD:
2468	case IB_CM_MRA_REQ_SENT:
2469		if (__be16_to_cpu(rej_msg->reason) == IB_CM_REJ_STALE_CONN)
2470			cm_enter_timewait(cm_id_priv);
2471		else
2472			cm_reset_to_idle(cm_id_priv);
2473		break;
2474	case IB_CM_DREQ_SENT:
2475		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2476		/* fall through */
2477	case IB_CM_REP_RCVD:
2478	case IB_CM_MRA_REP_SENT:
2479		cm_enter_timewait(cm_id_priv);
2480		break;
2481	case IB_CM_ESTABLISHED:
2482		if (cm_id_priv->id.lap_state == IB_CM_LAP_UNINIT ||
2483		    cm_id_priv->id.lap_state == IB_CM_LAP_SENT) {
2484			if (cm_id_priv->id.lap_state == IB_CM_LAP_SENT)
2485				ib_cancel_mad(cm_id_priv->av.port->mad_agent,
2486					      cm_id_priv->msg);
2487			cm_enter_timewait(cm_id_priv);
2488			break;
2489		}
2490		/* fall through */
2491	default:
2492		spin_unlock_irq(&cm_id_priv->lock);
2493		ret = -EINVAL;
2494		goto out;
2495	}
2496
2497	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2498	if (!ret)
2499		list_add_tail(&work->list, &cm_id_priv->work_list);
2500	spin_unlock_irq(&cm_id_priv->lock);
2501
2502	if (ret)
2503		cm_process_work(cm_id_priv, work);
2504	else
2505		cm_deref_id(cm_id_priv);
2506	return 0;
2507out:
2508	cm_deref_id(cm_id_priv);
2509	return -EINVAL;
2510}
2511
2512int ib_send_cm_mra(struct ib_cm_id *cm_id,
2513		   u8 service_timeout,
2514		   const void *private_data,
2515		   u8 private_data_len)
2516{
2517	struct cm_id_private *cm_id_priv;
2518	struct ib_mad_send_buf *msg;
2519	enum ib_cm_state cm_state;
2520	enum ib_cm_lap_state lap_state;
2521	enum cm_msg_response msg_response;
2522	void *data;
2523	unsigned long flags;
2524	int ret;
2525
2526	if (private_data && private_data_len > IB_CM_MRA_PRIVATE_DATA_SIZE)
2527		return -EINVAL;
2528
2529	data = cm_copy_private_data(private_data, private_data_len);
2530	if (IS_ERR(data))
2531		return PTR_ERR(data);
2532
2533	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2534
2535	spin_lock_irqsave(&cm_id_priv->lock, flags);
2536	switch(cm_id_priv->id.state) {
2537	case IB_CM_REQ_RCVD:
2538		cm_state = IB_CM_MRA_REQ_SENT;
2539		lap_state = cm_id->lap_state;
2540		msg_response = CM_MSG_RESPONSE_REQ;
2541		break;
2542	case IB_CM_REP_RCVD:
2543		cm_state = IB_CM_MRA_REP_SENT;
2544		lap_state = cm_id->lap_state;
2545		msg_response = CM_MSG_RESPONSE_REP;
2546		break;
2547	case IB_CM_ESTABLISHED:
2548		if (cm_id->lap_state == IB_CM_LAP_RCVD) {
2549			cm_state = cm_id->state;
2550			lap_state = IB_CM_MRA_LAP_SENT;
2551			msg_response = CM_MSG_RESPONSE_OTHER;
2552			break;
2553		}
2554	default:
2555		ret = -EINVAL;
2556		goto error1;
2557	}
2558
2559	if (!(service_timeout & IB_CM_MRA_FLAG_DELAY)) {
2560		ret = cm_alloc_msg(cm_id_priv, &msg);
2561		if (ret)
2562			goto error1;
2563
2564		cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
2565			      msg_response, service_timeout,
2566			      private_data, private_data_len);
2567		ret = ib_post_send_mad(msg, NULL);
2568		if (ret)
2569			goto error2;
2570	}
2571
2572	cm_id->state = cm_state;
2573	cm_id->lap_state = lap_state;
2574	cm_id_priv->service_timeout = service_timeout;
2575	cm_set_private_data(cm_id_priv, data, private_data_len);
2576	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2577	return 0;
2578
2579error1:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2580	kfree(data);
2581	return ret;
2582
2583error2:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2584	kfree(data);
2585	cm_free_msg(msg);
2586	return ret;
2587}
2588EXPORT_SYMBOL(ib_send_cm_mra);
2589
2590static struct cm_id_private * cm_acquire_mraed_id(struct cm_mra_msg *mra_msg)
2591{
2592	switch (cm_mra_get_msg_mraed(mra_msg)) {
2593	case CM_MSG_RESPONSE_REQ:
2594		return cm_acquire_id(mra_msg->remote_comm_id, 0);
2595	case CM_MSG_RESPONSE_REP:
2596	case CM_MSG_RESPONSE_OTHER:
2597		return cm_acquire_id(mra_msg->remote_comm_id,
2598				     mra_msg->local_comm_id);
2599	default:
2600		return NULL;
2601	}
2602}
2603
2604static int cm_mra_handler(struct cm_work *work)
2605{
2606	struct cm_id_private *cm_id_priv;
2607	struct cm_mra_msg *mra_msg;
2608	int timeout, ret;
2609
2610	mra_msg = (struct cm_mra_msg *)work->mad_recv_wc->recv_buf.mad;
2611	cm_id_priv = cm_acquire_mraed_id(mra_msg);
2612	if (!cm_id_priv)
2613		return -EINVAL;
2614
2615	work->cm_event.private_data = &mra_msg->private_data;
2616	work->cm_event.param.mra_rcvd.service_timeout =
2617					cm_mra_get_service_timeout(mra_msg);
2618	timeout = cm_convert_to_ms(cm_mra_get_service_timeout(mra_msg)) +
2619		  cm_convert_to_ms(cm_id_priv->av.timeout);
2620
2621	spin_lock_irq(&cm_id_priv->lock);
2622	switch (cm_id_priv->id.state) {
2623	case IB_CM_REQ_SENT:
2624		if (cm_mra_get_msg_mraed(mra_msg) != CM_MSG_RESPONSE_REQ ||
2625		    ib_modify_mad(cm_id_priv->av.port->mad_agent,
2626				  cm_id_priv->msg, timeout))
2627			goto out;
2628		cm_id_priv->id.state = IB_CM_MRA_REQ_RCVD;
2629		break;
2630	case IB_CM_REP_SENT:
2631		if (cm_mra_get_msg_mraed(mra_msg) != CM_MSG_RESPONSE_REP ||
2632		    ib_modify_mad(cm_id_priv->av.port->mad_agent,
2633				  cm_id_priv->msg, timeout))
2634			goto out;
2635		cm_id_priv->id.state = IB_CM_MRA_REP_RCVD;
2636		break;
2637	case IB_CM_ESTABLISHED:
2638		if (cm_mra_get_msg_mraed(mra_msg) != CM_MSG_RESPONSE_OTHER ||
2639		    cm_id_priv->id.lap_state != IB_CM_LAP_SENT ||
2640		    ib_modify_mad(cm_id_priv->av.port->mad_agent,
2641				  cm_id_priv->msg, timeout)) {
2642			if (cm_id_priv->id.lap_state == IB_CM_MRA_LAP_RCVD)
2643				atomic_long_inc(&work->port->
2644						counter_group[CM_RECV_DUPLICATES].
2645						counter[CM_MRA_COUNTER]);
2646			goto out;
2647		}
2648		cm_id_priv->id.lap_state = IB_CM_MRA_LAP_RCVD;
2649		break;
2650	case IB_CM_MRA_REQ_RCVD:
2651	case IB_CM_MRA_REP_RCVD:
2652		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2653				counter[CM_MRA_COUNTER]);
2654		/* fall through */
2655	default:
2656		goto out;
2657	}
2658
2659	cm_id_priv->msg->context[1] = (void *) (unsigned long)
2660				      cm_id_priv->id.state;
2661	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2662	if (!ret)
2663		list_add_tail(&work->list, &cm_id_priv->work_list);
2664	spin_unlock_irq(&cm_id_priv->lock);
2665
2666	if (ret)
2667		cm_process_work(cm_id_priv, work);
2668	else
2669		cm_deref_id(cm_id_priv);
2670	return 0;
2671out:
2672	spin_unlock_irq(&cm_id_priv->lock);
2673	cm_deref_id(cm_id_priv);
2674	return -EINVAL;
2675}
2676
2677static void cm_format_lap(struct cm_lap_msg *lap_msg,
2678			  struct cm_id_private *cm_id_priv,
2679			  struct ib_sa_path_rec *alternate_path,
2680			  const void *private_data,
2681			  u8 private_data_len)
2682{
2683	cm_format_mad_hdr(&lap_msg->hdr, CM_LAP_ATTR_ID,
2684			  cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_LAP));
2685	lap_msg->local_comm_id = cm_id_priv->id.local_id;
2686	lap_msg->remote_comm_id = cm_id_priv->id.remote_id;
2687	cm_lap_set_remote_qpn(lap_msg, cm_id_priv->remote_qpn);
2688	/* todo: need remote CM response timeout */
2689	cm_lap_set_remote_resp_timeout(lap_msg, 0x1F);
2690	lap_msg->alt_local_lid = alternate_path->slid;
2691	lap_msg->alt_remote_lid = alternate_path->dlid;
2692	lap_msg->alt_local_gid = alternate_path->sgid;
2693	lap_msg->alt_remote_gid = alternate_path->dgid;
2694	cm_lap_set_flow_label(lap_msg, alternate_path->flow_label);
2695	cm_lap_set_traffic_class(lap_msg, alternate_path->traffic_class);
2696	lap_msg->alt_hop_limit = alternate_path->hop_limit;
2697	cm_lap_set_packet_rate(lap_msg, alternate_path->rate);
2698	cm_lap_set_sl(lap_msg, alternate_path->sl);
2699	cm_lap_set_subnet_local(lap_msg, 1); /* local only... */
2700	cm_lap_set_local_ack_timeout(lap_msg,
2701		cm_ack_timeout(cm_id_priv->av.port->cm_dev->ack_delay,
2702			       alternate_path->packet_life_time));
2703
2704	if (private_data && private_data_len)
2705		memcpy(lap_msg->private_data, private_data, private_data_len);
2706}
2707
2708int ib_send_cm_lap(struct ib_cm_id *cm_id,
2709		   struct ib_sa_path_rec *alternate_path,
2710		   const void *private_data,
2711		   u8 private_data_len)
2712{
2713	struct cm_id_private *cm_id_priv;
2714	struct ib_mad_send_buf *msg;
2715	unsigned long flags;
2716	int ret;
2717
2718	if (private_data && private_data_len > IB_CM_LAP_PRIVATE_DATA_SIZE)
2719		return -EINVAL;
2720
2721	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2722	spin_lock_irqsave(&cm_id_priv->lock, flags);
2723	if (cm_id->state != IB_CM_ESTABLISHED ||
2724	    (cm_id->lap_state != IB_CM_LAP_UNINIT &&
2725	     cm_id->lap_state != IB_CM_LAP_IDLE)) {
2726		ret = -EINVAL;
2727		goto out;
2728	}
2729
2730	ret = cm_init_av_by_path(alternate_path, &cm_id_priv->alt_av);
2731	if (ret)
2732		goto out;
2733	cm_id_priv->alt_av.timeout =
2734			cm_ack_timeout(cm_id_priv->target_ack_delay,
2735				       cm_id_priv->alt_av.timeout - 1);
2736
2737	ret = cm_alloc_msg(cm_id_priv, &msg);
2738	if (ret)
2739		goto out;
2740
2741	cm_format_lap((struct cm_lap_msg *) msg->mad, cm_id_priv,
2742		      alternate_path, private_data, private_data_len);
2743	msg->timeout_ms = cm_id_priv->timeout_ms;
2744	msg->context[1] = (void *) (unsigned long) IB_CM_ESTABLISHED;
2745
2746	ret = ib_post_send_mad(msg, NULL);
2747	if (ret) {
2748		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2749		cm_free_msg(msg);
2750		return ret;
2751	}
2752
2753	cm_id->lap_state = IB_CM_LAP_SENT;
2754	cm_id_priv->msg = msg;
2755
2756out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2757	return ret;
2758}
2759EXPORT_SYMBOL(ib_send_cm_lap);
2760
2761static void cm_format_path_from_lap(struct cm_id_private *cm_id_priv,
2762				    struct ib_sa_path_rec *path,
2763				    struct cm_lap_msg *lap_msg)
2764{
2765	memset(path, 0, sizeof *path);
2766	path->dgid = lap_msg->alt_local_gid;
2767	path->sgid = lap_msg->alt_remote_gid;
2768	path->dlid = lap_msg->alt_local_lid;
2769	path->slid = lap_msg->alt_remote_lid;
2770	path->flow_label = cm_lap_get_flow_label(lap_msg);
2771	path->hop_limit = lap_msg->alt_hop_limit;
2772	path->traffic_class = cm_lap_get_traffic_class(lap_msg);
2773	path->reversible = 1;
2774	path->pkey = cm_id_priv->pkey;
2775	path->sl = cm_lap_get_sl(lap_msg);
2776	path->mtu_selector = IB_SA_EQ;
2777	path->mtu = cm_id_priv->path_mtu;
2778	path->rate_selector = IB_SA_EQ;
2779	path->rate = cm_lap_get_packet_rate(lap_msg);
2780	path->packet_life_time_selector = IB_SA_EQ;
2781	path->packet_life_time = cm_lap_get_local_ack_timeout(lap_msg);
2782	path->packet_life_time -= (path->packet_life_time > 0);
2783}
2784
2785static int cm_lap_handler(struct cm_work *work)
2786{
2787	struct cm_id_private *cm_id_priv;
2788	struct cm_lap_msg *lap_msg;
2789	struct ib_cm_lap_event_param *param;
2790	struct ib_mad_send_buf *msg = NULL;
2791	int ret;
2792
2793	/* todo: verify LAP request and send reject APR if invalid. */
2794	lap_msg = (struct cm_lap_msg *)work->mad_recv_wc->recv_buf.mad;
2795	cm_id_priv = cm_acquire_id(lap_msg->remote_comm_id,
2796				   lap_msg->local_comm_id);
2797	if (!cm_id_priv)
2798		return -EINVAL;
2799
2800	param = &work->cm_event.param.lap_rcvd;
2801	param->alternate_path = &work->path[0];
2802	cm_format_path_from_lap(cm_id_priv, param->alternate_path, lap_msg);
2803	work->cm_event.private_data = &lap_msg->private_data;
2804
2805	spin_lock_irq(&cm_id_priv->lock);
2806	if (cm_id_priv->id.state != IB_CM_ESTABLISHED)
2807		goto unlock;
2808
2809	switch (cm_id_priv->id.lap_state) {
2810	case IB_CM_LAP_UNINIT:
2811	case IB_CM_LAP_IDLE:
2812		break;
2813	case IB_CM_MRA_LAP_SENT:
2814		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2815				counter[CM_LAP_COUNTER]);
2816		if (cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg))
2817			goto unlock;
2818
2819		cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
2820			      CM_MSG_RESPONSE_OTHER,
2821			      cm_id_priv->service_timeout,
2822			      cm_id_priv->private_data,
2823			      cm_id_priv->private_data_len);
2824		spin_unlock_irq(&cm_id_priv->lock);
2825
2826		if (ib_post_send_mad(msg, NULL))
2827			cm_free_msg(msg);
2828		goto deref;
2829	case IB_CM_LAP_RCVD:
2830		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2831				counter[CM_LAP_COUNTER]);
2832		goto unlock;
2833	default:
2834		goto unlock;
2835	}
2836
2837	cm_id_priv->id.lap_state = IB_CM_LAP_RCVD;
2838	cm_id_priv->tid = lap_msg->hdr.tid;
2839	cm_init_av_for_response(work->port, work->mad_recv_wc->wc,
2840				work->mad_recv_wc->recv_buf.grh,
2841				&cm_id_priv->av);
2842	cm_init_av_by_path(param->alternate_path, &cm_id_priv->alt_av);
2843	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2844	if (!ret)
2845		list_add_tail(&work->list, &cm_id_priv->work_list);
2846	spin_unlock_irq(&cm_id_priv->lock);
2847
2848	if (ret)
2849		cm_process_work(cm_id_priv, work);
2850	else
2851		cm_deref_id(cm_id_priv);
2852	return 0;
2853
2854unlock:	spin_unlock_irq(&cm_id_priv->lock);
2855deref:	cm_deref_id(cm_id_priv);
2856	return -EINVAL;
2857}
2858
2859static void cm_format_apr(struct cm_apr_msg *apr_msg,
2860			  struct cm_id_private *cm_id_priv,
2861			  enum ib_cm_apr_status status,
2862			  void *info,
2863			  u8 info_length,
2864			  const void *private_data,
2865			  u8 private_data_len)
2866{
2867	cm_format_mad_hdr(&apr_msg->hdr, CM_APR_ATTR_ID, cm_id_priv->tid);
2868	apr_msg->local_comm_id = cm_id_priv->id.local_id;
2869	apr_msg->remote_comm_id = cm_id_priv->id.remote_id;
2870	apr_msg->ap_status = (u8) status;
2871
2872	if (info && info_length) {
2873		apr_msg->info_length = info_length;
2874		memcpy(apr_msg->info, info, info_length);
2875	}
2876
2877	if (private_data && private_data_len)
2878		memcpy(apr_msg->private_data, private_data, private_data_len);
2879}
2880
2881int ib_send_cm_apr(struct ib_cm_id *cm_id,
2882		   enum ib_cm_apr_status status,
2883		   void *info,
2884		   u8 info_length,
2885		   const void *private_data,
2886		   u8 private_data_len)
2887{
2888	struct cm_id_private *cm_id_priv;
2889	struct ib_mad_send_buf *msg;
2890	unsigned long flags;
2891	int ret;
2892
2893	if ((private_data && private_data_len > IB_CM_APR_PRIVATE_DATA_SIZE) ||
2894	    (info && info_length > IB_CM_APR_INFO_LENGTH))
2895		return -EINVAL;
2896
2897	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2898	spin_lock_irqsave(&cm_id_priv->lock, flags);
2899	if (cm_id->state != IB_CM_ESTABLISHED ||
2900	    (cm_id->lap_state != IB_CM_LAP_RCVD &&
2901	     cm_id->lap_state != IB_CM_MRA_LAP_SENT)) {
2902		ret = -EINVAL;
2903		goto out;
2904	}
2905
2906	ret = cm_alloc_msg(cm_id_priv, &msg);
2907	if (ret)
2908		goto out;
2909
2910	cm_format_apr((struct cm_apr_msg *) msg->mad, cm_id_priv, status,
2911		      info, info_length, private_data, private_data_len);
2912	ret = ib_post_send_mad(msg, NULL);
2913	if (ret) {
2914		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2915		cm_free_msg(msg);
2916		return ret;
2917	}
2918
2919	cm_id->lap_state = IB_CM_LAP_IDLE;
2920out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2921	return ret;
2922}
2923EXPORT_SYMBOL(ib_send_cm_apr);
2924
2925static int cm_apr_handler(struct cm_work *work)
2926{
2927	struct cm_id_private *cm_id_priv;
2928	struct cm_apr_msg *apr_msg;
2929	int ret;
2930
2931	apr_msg = (struct cm_apr_msg *)work->mad_recv_wc->recv_buf.mad;
2932	cm_id_priv = cm_acquire_id(apr_msg->remote_comm_id,
2933				   apr_msg->local_comm_id);
2934	if (!cm_id_priv)
2935		return -EINVAL; /* Unmatched reply. */
2936
2937	work->cm_event.param.apr_rcvd.ap_status = apr_msg->ap_status;
2938	work->cm_event.param.apr_rcvd.apr_info = &apr_msg->info;
2939	work->cm_event.param.apr_rcvd.info_len = apr_msg->info_length;
2940	work->cm_event.private_data = &apr_msg->private_data;
2941
2942	spin_lock_irq(&cm_id_priv->lock);
2943	if (cm_id_priv->id.state != IB_CM_ESTABLISHED ||
2944	    (cm_id_priv->id.lap_state != IB_CM_LAP_SENT &&
2945	     cm_id_priv->id.lap_state != IB_CM_MRA_LAP_RCVD)) {
2946		spin_unlock_irq(&cm_id_priv->lock);
2947		goto out;
2948	}
2949	cm_id_priv->id.lap_state = IB_CM_LAP_IDLE;
2950	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2951	cm_id_priv->msg = NULL;
2952
2953	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2954	if (!ret)
2955		list_add_tail(&work->list, &cm_id_priv->work_list);
2956	spin_unlock_irq(&cm_id_priv->lock);
2957
2958	if (ret)
2959		cm_process_work(cm_id_priv, work);
2960	else
2961		cm_deref_id(cm_id_priv);
2962	return 0;
2963out:
2964	cm_deref_id(cm_id_priv);
2965	return -EINVAL;
2966}
2967
2968static int cm_timewait_handler(struct cm_work *work)
2969{
2970	struct cm_timewait_info *timewait_info;
2971	struct cm_id_private *cm_id_priv;
2972	int ret;
2973
2974	timewait_info = (struct cm_timewait_info *)work;
2975	spin_lock_irq(&cm.lock);
2976	list_del(&timewait_info->list);
2977	spin_unlock_irq(&cm.lock);
2978
2979	cm_id_priv = cm_acquire_id(timewait_info->work.local_id,
2980				   timewait_info->work.remote_id);
2981	if (!cm_id_priv)
2982		return -EINVAL;
2983
2984	spin_lock_irq(&cm_id_priv->lock);
2985	if (cm_id_priv->id.state != IB_CM_TIMEWAIT ||
2986	    cm_id_priv->remote_qpn != timewait_info->remote_qpn) {
2987		spin_unlock_irq(&cm_id_priv->lock);
2988		goto out;
2989	}
2990	cm_id_priv->id.state = IB_CM_IDLE;
2991	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2992	if (!ret)
2993		list_add_tail(&work->list, &cm_id_priv->work_list);
2994	spin_unlock_irq(&cm_id_priv->lock);
2995
2996	if (ret)
2997		cm_process_work(cm_id_priv, work);
2998	else
2999		cm_deref_id(cm_id_priv);
3000	return 0;
3001out:
3002	cm_deref_id(cm_id_priv);
3003	return -EINVAL;
3004}
3005
3006static void cm_format_sidr_req(struct cm_sidr_req_msg *sidr_req_msg,
3007			       struct cm_id_private *cm_id_priv,
3008			       struct ib_cm_sidr_req_param *param)
3009{
3010	cm_format_mad_hdr(&sidr_req_msg->hdr, CM_SIDR_REQ_ATTR_ID,
3011			  cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_SIDR));
3012	sidr_req_msg->request_id = cm_id_priv->id.local_id;
3013	sidr_req_msg->pkey = param->path->pkey;
3014	sidr_req_msg->service_id = param->service_id;
3015
3016	if (param->private_data && param->private_data_len)
3017		memcpy(sidr_req_msg->private_data, param->private_data,
3018		       param->private_data_len);
3019}
3020
3021int ib_send_cm_sidr_req(struct ib_cm_id *cm_id,
3022			struct ib_cm_sidr_req_param *param)
3023{
3024	struct cm_id_private *cm_id_priv;
3025	struct ib_mad_send_buf *msg;
3026	unsigned long flags;
3027	int ret;
3028
3029	if (!param->path || (param->private_data &&
3030	     param->private_data_len > IB_CM_SIDR_REQ_PRIVATE_DATA_SIZE))
3031		return -EINVAL;
3032
3033	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3034	ret = cm_init_av_by_path(param->path, &cm_id_priv->av);
3035	if (ret)
3036		goto out;
3037
3038	cm_id->service_id = param->service_id;
3039	cm_id->service_mask = ~cpu_to_be64(0);
3040	cm_id_priv->timeout_ms = param->timeout_ms;
3041	cm_id_priv->max_cm_retries = param->max_cm_retries;
3042	ret = cm_alloc_msg(cm_id_priv, &msg);
3043	if (ret)
3044		goto out;
3045
3046	cm_format_sidr_req((struct cm_sidr_req_msg *) msg->mad, cm_id_priv,
3047			   param);
3048	msg->timeout_ms = cm_id_priv->timeout_ms;
3049	msg->context[1] = (void *) (unsigned long) IB_CM_SIDR_REQ_SENT;
3050
3051	spin_lock_irqsave(&cm_id_priv->lock, flags);
3052	if (cm_id->state == IB_CM_IDLE)
3053		ret = ib_post_send_mad(msg, NULL);
3054	else
3055		ret = -EINVAL;
3056
3057	if (ret) {
3058		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3059		cm_free_msg(msg);
3060		goto out;
3061	}
3062	cm_id->state = IB_CM_SIDR_REQ_SENT;
3063	cm_id_priv->msg = msg;
3064	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3065out:
3066	return ret;
3067}
3068EXPORT_SYMBOL(ib_send_cm_sidr_req);
3069
3070static void cm_format_sidr_req_event(struct cm_work *work,
3071				     struct ib_cm_id *listen_id)
3072{
3073	struct cm_sidr_req_msg *sidr_req_msg;
3074	struct ib_cm_sidr_req_event_param *param;
3075
3076	sidr_req_msg = (struct cm_sidr_req_msg *)
3077				work->mad_recv_wc->recv_buf.mad;
3078	param = &work->cm_event.param.sidr_req_rcvd;
3079	param->pkey = __be16_to_cpu(sidr_req_msg->pkey);
3080	param->listen_id = listen_id;
3081	param->service_id = sidr_req_msg->service_id;
3082	param->bth_pkey = cm_get_bth_pkey(work);
3083	param->port = work->port->port_num;
3084	work->cm_event.private_data = &sidr_req_msg->private_data;
3085}
3086
3087static int cm_sidr_req_handler(struct cm_work *work)
3088{
3089	struct ib_cm_id *cm_id;
3090	struct cm_id_private *cm_id_priv, *cur_cm_id_priv;
3091	struct cm_sidr_req_msg *sidr_req_msg;
3092	struct ib_wc *wc;
3093
3094	cm_id = ib_create_cm_id(work->port->cm_dev->ib_device, NULL, NULL);
3095	if (IS_ERR(cm_id))
3096		return PTR_ERR(cm_id);
3097	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3098
3099	/* Record SGID/SLID and request ID for lookup. */
3100	sidr_req_msg = (struct cm_sidr_req_msg *)
3101				work->mad_recv_wc->recv_buf.mad;
3102	wc = work->mad_recv_wc->wc;
3103	cm_id_priv->av.dgid.global.subnet_prefix = cpu_to_be64(wc->slid);
3104	cm_id_priv->av.dgid.global.interface_id = 0;
3105	cm_init_av_for_response(work->port, work->mad_recv_wc->wc,
3106				work->mad_recv_wc->recv_buf.grh,
3107				&cm_id_priv->av);
3108	cm_id_priv->id.remote_id = sidr_req_msg->request_id;
3109	cm_id_priv->tid = sidr_req_msg->hdr.tid;
3110	atomic_inc(&cm_id_priv->work_count);
3111
3112	spin_lock_irq(&cm.lock);
3113	cur_cm_id_priv = cm_insert_remote_sidr(cm_id_priv);
3114	if (cur_cm_id_priv) {
3115		spin_unlock_irq(&cm.lock);
3116		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
3117				counter[CM_SIDR_REQ_COUNTER]);
3118		goto out; /* Duplicate message. */
3119	}
3120	cm_id_priv->id.state = IB_CM_SIDR_REQ_RCVD;
3121	cur_cm_id_priv = cm_find_listen(cm_id->device,
3122					sidr_req_msg->service_id);
 
3123	if (!cur_cm_id_priv) {
3124		spin_unlock_irq(&cm.lock);
3125		cm_reject_sidr_req(cm_id_priv, IB_SIDR_UNSUPPORTED);
3126		goto out; /* No match. */
3127	}
3128	atomic_inc(&cur_cm_id_priv->refcount);
3129	atomic_inc(&cm_id_priv->refcount);
3130	spin_unlock_irq(&cm.lock);
3131
3132	cm_id_priv->id.cm_handler = cur_cm_id_priv->id.cm_handler;
3133	cm_id_priv->id.context = cur_cm_id_priv->id.context;
3134	cm_id_priv->id.service_id = sidr_req_msg->service_id;
3135	cm_id_priv->id.service_mask = ~cpu_to_be64(0);
3136
3137	cm_format_sidr_req_event(work, &cur_cm_id_priv->id);
3138	cm_process_work(cm_id_priv, work);
3139	cm_deref_id(cur_cm_id_priv);
3140	return 0;
3141out:
3142	ib_destroy_cm_id(&cm_id_priv->id);
3143	return -EINVAL;
3144}
3145
3146static void cm_format_sidr_rep(struct cm_sidr_rep_msg *sidr_rep_msg,
3147			       struct cm_id_private *cm_id_priv,
3148			       struct ib_cm_sidr_rep_param *param)
3149{
3150	cm_format_mad_hdr(&sidr_rep_msg->hdr, CM_SIDR_REP_ATTR_ID,
3151			  cm_id_priv->tid);
3152	sidr_rep_msg->request_id = cm_id_priv->id.remote_id;
3153	sidr_rep_msg->status = param->status;
3154	cm_sidr_rep_set_qpn(sidr_rep_msg, cpu_to_be32(param->qp_num));
3155	sidr_rep_msg->service_id = cm_id_priv->id.service_id;
3156	sidr_rep_msg->qkey = cpu_to_be32(param->qkey);
3157
3158	if (param->info && param->info_length)
3159		memcpy(sidr_rep_msg->info, param->info, param->info_length);
3160
3161	if (param->private_data && param->private_data_len)
3162		memcpy(sidr_rep_msg->private_data, param->private_data,
3163		       param->private_data_len);
3164}
3165
3166int ib_send_cm_sidr_rep(struct ib_cm_id *cm_id,
3167			struct ib_cm_sidr_rep_param *param)
3168{
3169	struct cm_id_private *cm_id_priv;
3170	struct ib_mad_send_buf *msg;
3171	unsigned long flags;
3172	int ret;
3173
3174	if ((param->info && param->info_length > IB_CM_SIDR_REP_INFO_LENGTH) ||
3175	    (param->private_data &&
3176	     param->private_data_len > IB_CM_SIDR_REP_PRIVATE_DATA_SIZE))
3177		return -EINVAL;
3178
3179	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3180	spin_lock_irqsave(&cm_id_priv->lock, flags);
3181	if (cm_id->state != IB_CM_SIDR_REQ_RCVD) {
3182		ret = -EINVAL;
3183		goto error;
3184	}
3185
3186	ret = cm_alloc_msg(cm_id_priv, &msg);
3187	if (ret)
3188		goto error;
3189
3190	cm_format_sidr_rep((struct cm_sidr_rep_msg *) msg->mad, cm_id_priv,
3191			   param);
3192	ret = ib_post_send_mad(msg, NULL);
3193	if (ret) {
3194		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3195		cm_free_msg(msg);
3196		return ret;
3197	}
3198	cm_id->state = IB_CM_IDLE;
3199	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3200
3201	spin_lock_irqsave(&cm.lock, flags);
3202	if (!RB_EMPTY_NODE(&cm_id_priv->sidr_id_node)) {
3203		rb_erase(&cm_id_priv->sidr_id_node, &cm.remote_sidr_table);
3204		RB_CLEAR_NODE(&cm_id_priv->sidr_id_node);
3205	}
3206	spin_unlock_irqrestore(&cm.lock, flags);
3207	return 0;
3208
3209error:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3210	return ret;
3211}
3212EXPORT_SYMBOL(ib_send_cm_sidr_rep);
3213
3214static void cm_format_sidr_rep_event(struct cm_work *work)
3215{
3216	struct cm_sidr_rep_msg *sidr_rep_msg;
3217	struct ib_cm_sidr_rep_event_param *param;
3218
3219	sidr_rep_msg = (struct cm_sidr_rep_msg *)
3220				work->mad_recv_wc->recv_buf.mad;
3221	param = &work->cm_event.param.sidr_rep_rcvd;
3222	param->status = sidr_rep_msg->status;
3223	param->qkey = be32_to_cpu(sidr_rep_msg->qkey);
3224	param->qpn = be32_to_cpu(cm_sidr_rep_get_qpn(sidr_rep_msg));
3225	param->info = &sidr_rep_msg->info;
3226	param->info_len = sidr_rep_msg->info_length;
3227	work->cm_event.private_data = &sidr_rep_msg->private_data;
3228}
3229
3230static int cm_sidr_rep_handler(struct cm_work *work)
3231{
3232	struct cm_sidr_rep_msg *sidr_rep_msg;
3233	struct cm_id_private *cm_id_priv;
3234
3235	sidr_rep_msg = (struct cm_sidr_rep_msg *)
3236				work->mad_recv_wc->recv_buf.mad;
3237	cm_id_priv = cm_acquire_id(sidr_rep_msg->request_id, 0);
3238	if (!cm_id_priv)
3239		return -EINVAL; /* Unmatched reply. */
3240
3241	spin_lock_irq(&cm_id_priv->lock);
3242	if (cm_id_priv->id.state != IB_CM_SIDR_REQ_SENT) {
3243		spin_unlock_irq(&cm_id_priv->lock);
3244		goto out;
3245	}
3246	cm_id_priv->id.state = IB_CM_IDLE;
3247	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
3248	spin_unlock_irq(&cm_id_priv->lock);
3249
3250	cm_format_sidr_rep_event(work);
3251	cm_process_work(cm_id_priv, work);
3252	return 0;
3253out:
3254	cm_deref_id(cm_id_priv);
3255	return -EINVAL;
3256}
3257
3258static void cm_process_send_error(struct ib_mad_send_buf *msg,
3259				  enum ib_wc_status wc_status)
3260{
3261	struct cm_id_private *cm_id_priv;
3262	struct ib_cm_event cm_event;
3263	enum ib_cm_state state;
3264	int ret;
3265
3266	memset(&cm_event, 0, sizeof cm_event);
3267	cm_id_priv = msg->context[0];
3268
3269	/* Discard old sends or ones without a response. */
3270	spin_lock_irq(&cm_id_priv->lock);
3271	state = (enum ib_cm_state) (unsigned long) msg->context[1];
3272	if (msg != cm_id_priv->msg || state != cm_id_priv->id.state)
3273		goto discard;
3274
3275	switch (state) {
3276	case IB_CM_REQ_SENT:
3277	case IB_CM_MRA_REQ_RCVD:
3278		cm_reset_to_idle(cm_id_priv);
3279		cm_event.event = IB_CM_REQ_ERROR;
3280		break;
3281	case IB_CM_REP_SENT:
3282	case IB_CM_MRA_REP_RCVD:
3283		cm_reset_to_idle(cm_id_priv);
3284		cm_event.event = IB_CM_REP_ERROR;
3285		break;
3286	case IB_CM_DREQ_SENT:
3287		cm_enter_timewait(cm_id_priv);
3288		cm_event.event = IB_CM_DREQ_ERROR;
3289		break;
3290	case IB_CM_SIDR_REQ_SENT:
3291		cm_id_priv->id.state = IB_CM_IDLE;
3292		cm_event.event = IB_CM_SIDR_REQ_ERROR;
3293		break;
3294	default:
3295		goto discard;
3296	}
3297	spin_unlock_irq(&cm_id_priv->lock);
3298	cm_event.param.send_status = wc_status;
3299
3300	/* No other events can occur on the cm_id at this point. */
3301	ret = cm_id_priv->id.cm_handler(&cm_id_priv->id, &cm_event);
3302	cm_free_msg(msg);
3303	if (ret)
3304		ib_destroy_cm_id(&cm_id_priv->id);
3305	return;
3306discard:
3307	spin_unlock_irq(&cm_id_priv->lock);
3308	cm_free_msg(msg);
3309}
3310
3311static void cm_send_handler(struct ib_mad_agent *mad_agent,
3312			    struct ib_mad_send_wc *mad_send_wc)
3313{
3314	struct ib_mad_send_buf *msg = mad_send_wc->send_buf;
3315	struct cm_port *port;
3316	u16 attr_index;
3317
3318	port = mad_agent->context;
3319	attr_index = be16_to_cpu(((struct ib_mad_hdr *)
3320				  msg->mad)->attr_id) - CM_ATTR_ID_OFFSET;
3321
3322	/*
3323	 * If the send was in response to a received message (context[0] is not
3324	 * set to a cm_id), and is not a REJ, then it is a send that was
3325	 * manually retried.
3326	 */
3327	if (!msg->context[0] && (attr_index != CM_REJ_COUNTER))
3328		msg->retries = 1;
3329
3330	atomic_long_add(1 + msg->retries,
3331			&port->counter_group[CM_XMIT].counter[attr_index]);
3332	if (msg->retries)
3333		atomic_long_add(msg->retries,
3334				&port->counter_group[CM_XMIT_RETRIES].
3335				counter[attr_index]);
3336
3337	switch (mad_send_wc->status) {
3338	case IB_WC_SUCCESS:
3339	case IB_WC_WR_FLUSH_ERR:
3340		cm_free_msg(msg);
3341		break;
3342	default:
3343		if (msg->context[0] && msg->context[1])
3344			cm_process_send_error(msg, mad_send_wc->status);
3345		else
3346			cm_free_msg(msg);
3347		break;
3348	}
3349}
3350
3351static void cm_work_handler(struct work_struct *_work)
3352{
3353	struct cm_work *work = container_of(_work, struct cm_work, work.work);
3354	int ret;
3355
3356	switch (work->cm_event.event) {
3357	case IB_CM_REQ_RECEIVED:
3358		ret = cm_req_handler(work);
3359		break;
3360	case IB_CM_MRA_RECEIVED:
3361		ret = cm_mra_handler(work);
3362		break;
3363	case IB_CM_REJ_RECEIVED:
3364		ret = cm_rej_handler(work);
3365		break;
3366	case IB_CM_REP_RECEIVED:
3367		ret = cm_rep_handler(work);
3368		break;
3369	case IB_CM_RTU_RECEIVED:
3370		ret = cm_rtu_handler(work);
3371		break;
3372	case IB_CM_USER_ESTABLISHED:
3373		ret = cm_establish_handler(work);
3374		break;
3375	case IB_CM_DREQ_RECEIVED:
3376		ret = cm_dreq_handler(work);
3377		break;
3378	case IB_CM_DREP_RECEIVED:
3379		ret = cm_drep_handler(work);
3380		break;
3381	case IB_CM_SIDR_REQ_RECEIVED:
3382		ret = cm_sidr_req_handler(work);
3383		break;
3384	case IB_CM_SIDR_REP_RECEIVED:
3385		ret = cm_sidr_rep_handler(work);
3386		break;
3387	case IB_CM_LAP_RECEIVED:
3388		ret = cm_lap_handler(work);
3389		break;
3390	case IB_CM_APR_RECEIVED:
3391		ret = cm_apr_handler(work);
3392		break;
3393	case IB_CM_TIMEWAIT_EXIT:
3394		ret = cm_timewait_handler(work);
3395		break;
3396	default:
3397		ret = -EINVAL;
3398		break;
3399	}
3400	if (ret)
3401		cm_free_work(work);
3402}
3403
3404static int cm_establish(struct ib_cm_id *cm_id)
3405{
3406	struct cm_id_private *cm_id_priv;
3407	struct cm_work *work;
3408	unsigned long flags;
3409	int ret = 0;
3410	struct cm_device *cm_dev;
3411
3412	cm_dev = ib_get_client_data(cm_id->device, &cm_client);
3413	if (!cm_dev)
3414		return -ENODEV;
3415
3416	work = kmalloc(sizeof *work, GFP_ATOMIC);
3417	if (!work)
3418		return -ENOMEM;
3419
3420	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3421	spin_lock_irqsave(&cm_id_priv->lock, flags);
3422	switch (cm_id->state)
3423	{
3424	case IB_CM_REP_SENT:
3425	case IB_CM_MRA_REP_RCVD:
3426		cm_id->state = IB_CM_ESTABLISHED;
3427		break;
3428	case IB_CM_ESTABLISHED:
3429		ret = -EISCONN;
3430		break;
3431	default:
3432		ret = -EINVAL;
3433		break;
3434	}
3435	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3436
3437	if (ret) {
3438		kfree(work);
3439		goto out;
3440	}
3441
3442	/*
3443	 * The CM worker thread may try to destroy the cm_id before it
3444	 * can execute this work item.  To prevent potential deadlock,
3445	 * we need to find the cm_id once we're in the context of the
3446	 * worker thread, rather than holding a reference on it.
3447	 */
3448	INIT_DELAYED_WORK(&work->work, cm_work_handler);
3449	work->local_id = cm_id->local_id;
3450	work->remote_id = cm_id->remote_id;
3451	work->mad_recv_wc = NULL;
3452	work->cm_event.event = IB_CM_USER_ESTABLISHED;
3453
3454	/* Check if the device started its remove_one */
3455	spin_lock_irq(&cm.lock);
3456	if (!cm_dev->going_down) {
3457		queue_delayed_work(cm.wq, &work->work, 0);
3458	} else {
3459		kfree(work);
3460		ret = -ENODEV;
3461	}
3462	spin_unlock_irq(&cm.lock);
3463
3464out:
3465	return ret;
3466}
3467
3468static int cm_migrate(struct ib_cm_id *cm_id)
3469{
3470	struct cm_id_private *cm_id_priv;
3471	unsigned long flags;
3472	int ret = 0;
3473
3474	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3475	spin_lock_irqsave(&cm_id_priv->lock, flags);
3476	if (cm_id->state == IB_CM_ESTABLISHED &&
3477	    (cm_id->lap_state == IB_CM_LAP_UNINIT ||
3478	     cm_id->lap_state == IB_CM_LAP_IDLE)) {
3479		cm_id->lap_state = IB_CM_LAP_IDLE;
3480		cm_id_priv->av = cm_id_priv->alt_av;
3481	} else
3482		ret = -EINVAL;
3483	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3484
3485	return ret;
3486}
3487
3488int ib_cm_notify(struct ib_cm_id *cm_id, enum ib_event_type event)
3489{
3490	int ret;
3491
3492	switch (event) {
3493	case IB_EVENT_COMM_EST:
3494		ret = cm_establish(cm_id);
3495		break;
3496	case IB_EVENT_PATH_MIG:
3497		ret = cm_migrate(cm_id);
3498		break;
3499	default:
3500		ret = -EINVAL;
3501	}
3502	return ret;
3503}
3504EXPORT_SYMBOL(ib_cm_notify);
3505
3506static void cm_recv_handler(struct ib_mad_agent *mad_agent,
3507			    struct ib_mad_send_buf *send_buf,
3508			    struct ib_mad_recv_wc *mad_recv_wc)
3509{
3510	struct cm_port *port = mad_agent->context;
3511	struct cm_work *work;
3512	enum ib_cm_event_type event;
3513	u16 attr_id;
3514	int paths = 0;
3515	int going_down = 0;
3516
3517	switch (mad_recv_wc->recv_buf.mad->mad_hdr.attr_id) {
3518	case CM_REQ_ATTR_ID:
3519		paths = 1 + (((struct cm_req_msg *) mad_recv_wc->recv_buf.mad)->
3520						    alt_local_lid != 0);
3521		event = IB_CM_REQ_RECEIVED;
3522		break;
3523	case CM_MRA_ATTR_ID:
3524		event = IB_CM_MRA_RECEIVED;
3525		break;
3526	case CM_REJ_ATTR_ID:
3527		event = IB_CM_REJ_RECEIVED;
3528		break;
3529	case CM_REP_ATTR_ID:
3530		event = IB_CM_REP_RECEIVED;
3531		break;
3532	case CM_RTU_ATTR_ID:
3533		event = IB_CM_RTU_RECEIVED;
3534		break;
3535	case CM_DREQ_ATTR_ID:
3536		event = IB_CM_DREQ_RECEIVED;
3537		break;
3538	case CM_DREP_ATTR_ID:
3539		event = IB_CM_DREP_RECEIVED;
3540		break;
3541	case CM_SIDR_REQ_ATTR_ID:
3542		event = IB_CM_SIDR_REQ_RECEIVED;
3543		break;
3544	case CM_SIDR_REP_ATTR_ID:
3545		event = IB_CM_SIDR_REP_RECEIVED;
3546		break;
3547	case CM_LAP_ATTR_ID:
3548		paths = 1;
3549		event = IB_CM_LAP_RECEIVED;
3550		break;
3551	case CM_APR_ATTR_ID:
3552		event = IB_CM_APR_RECEIVED;
3553		break;
3554	default:
3555		ib_free_recv_mad(mad_recv_wc);
3556		return;
3557	}
3558
3559	attr_id = be16_to_cpu(mad_recv_wc->recv_buf.mad->mad_hdr.attr_id);
3560	atomic_long_inc(&port->counter_group[CM_RECV].
3561			counter[attr_id - CM_ATTR_ID_OFFSET]);
3562
3563	work = kmalloc(sizeof *work + sizeof(struct ib_sa_path_rec) * paths,
3564		       GFP_KERNEL);
3565	if (!work) {
3566		ib_free_recv_mad(mad_recv_wc);
3567		return;
3568	}
3569
3570	INIT_DELAYED_WORK(&work->work, cm_work_handler);
3571	work->cm_event.event = event;
3572	work->mad_recv_wc = mad_recv_wc;
3573	work->port = port;
3574
3575	/* Check if the device started its remove_one */
3576	spin_lock_irq(&cm.lock);
3577	if (!port->cm_dev->going_down)
3578		queue_delayed_work(cm.wq, &work->work, 0);
3579	else
3580		going_down = 1;
3581	spin_unlock_irq(&cm.lock);
3582
3583	if (going_down) {
3584		kfree(work);
3585		ib_free_recv_mad(mad_recv_wc);
3586	}
3587}
3588
3589static int cm_init_qp_init_attr(struct cm_id_private *cm_id_priv,
3590				struct ib_qp_attr *qp_attr,
3591				int *qp_attr_mask)
3592{
3593	unsigned long flags;
3594	int ret;
3595
3596	spin_lock_irqsave(&cm_id_priv->lock, flags);
3597	switch (cm_id_priv->id.state) {
3598	case IB_CM_REQ_SENT:
3599	case IB_CM_MRA_REQ_RCVD:
3600	case IB_CM_REQ_RCVD:
3601	case IB_CM_MRA_REQ_SENT:
3602	case IB_CM_REP_RCVD:
3603	case IB_CM_MRA_REP_SENT:
3604	case IB_CM_REP_SENT:
3605	case IB_CM_MRA_REP_RCVD:
3606	case IB_CM_ESTABLISHED:
3607		*qp_attr_mask = IB_QP_STATE | IB_QP_ACCESS_FLAGS |
3608				IB_QP_PKEY_INDEX | IB_QP_PORT;
3609		qp_attr->qp_access_flags = IB_ACCESS_REMOTE_WRITE;
3610		if (cm_id_priv->responder_resources)
3611			qp_attr->qp_access_flags |= IB_ACCESS_REMOTE_READ |
3612						    IB_ACCESS_REMOTE_ATOMIC;
3613		qp_attr->pkey_index = cm_id_priv->av.pkey_index;
3614		qp_attr->port_num = cm_id_priv->av.port->port_num;
3615		ret = 0;
3616		break;
3617	default:
3618		ret = -EINVAL;
3619		break;
3620	}
3621	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3622	return ret;
3623}
3624
3625static int cm_init_qp_rtr_attr(struct cm_id_private *cm_id_priv,
3626			       struct ib_qp_attr *qp_attr,
3627			       int *qp_attr_mask)
3628{
3629	unsigned long flags;
3630	int ret;
3631
3632	spin_lock_irqsave(&cm_id_priv->lock, flags);
3633	switch (cm_id_priv->id.state) {
3634	case IB_CM_REQ_RCVD:
3635	case IB_CM_MRA_REQ_SENT:
3636	case IB_CM_REP_RCVD:
3637	case IB_CM_MRA_REP_SENT:
3638	case IB_CM_REP_SENT:
3639	case IB_CM_MRA_REP_RCVD:
3640	case IB_CM_ESTABLISHED:
3641		*qp_attr_mask = IB_QP_STATE | IB_QP_AV | IB_QP_PATH_MTU |
3642				IB_QP_DEST_QPN | IB_QP_RQ_PSN;
3643		qp_attr->ah_attr = cm_id_priv->av.ah_attr;
3644		qp_attr->path_mtu = cm_id_priv->path_mtu;
3645		qp_attr->dest_qp_num = be32_to_cpu(cm_id_priv->remote_qpn);
3646		qp_attr->rq_psn = be32_to_cpu(cm_id_priv->rq_psn);
3647		if (cm_id_priv->qp_type == IB_QPT_RC ||
3648		    cm_id_priv->qp_type == IB_QPT_XRC_TGT) {
3649			*qp_attr_mask |= IB_QP_MAX_DEST_RD_ATOMIC |
3650					 IB_QP_MIN_RNR_TIMER;
3651			qp_attr->max_dest_rd_atomic =
3652					cm_id_priv->responder_resources;
3653			qp_attr->min_rnr_timer = 0;
3654		}
3655		if (cm_id_priv->alt_av.ah_attr.dlid) {
3656			*qp_attr_mask |= IB_QP_ALT_PATH;
3657			qp_attr->alt_port_num = cm_id_priv->alt_av.port->port_num;
3658			qp_attr->alt_pkey_index = cm_id_priv->alt_av.pkey_index;
3659			qp_attr->alt_timeout = cm_id_priv->alt_av.timeout;
3660			qp_attr->alt_ah_attr = cm_id_priv->alt_av.ah_attr;
3661		}
3662		ret = 0;
3663		break;
3664	default:
3665		ret = -EINVAL;
3666		break;
3667	}
3668	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3669	return ret;
3670}
3671
3672static int cm_init_qp_rts_attr(struct cm_id_private *cm_id_priv,
3673			       struct ib_qp_attr *qp_attr,
3674			       int *qp_attr_mask)
3675{
3676	unsigned long flags;
3677	int ret;
3678
3679	spin_lock_irqsave(&cm_id_priv->lock, flags);
3680	switch (cm_id_priv->id.state) {
3681	/* Allow transition to RTS before sending REP */
3682	case IB_CM_REQ_RCVD:
3683	case IB_CM_MRA_REQ_SENT:
3684
3685	case IB_CM_REP_RCVD:
3686	case IB_CM_MRA_REP_SENT:
3687	case IB_CM_REP_SENT:
3688	case IB_CM_MRA_REP_RCVD:
3689	case IB_CM_ESTABLISHED:
3690		if (cm_id_priv->id.lap_state == IB_CM_LAP_UNINIT) {
3691			*qp_attr_mask = IB_QP_STATE | IB_QP_SQ_PSN;
3692			qp_attr->sq_psn = be32_to_cpu(cm_id_priv->sq_psn);
3693			switch (cm_id_priv->qp_type) {
3694			case IB_QPT_RC:
3695			case IB_QPT_XRC_INI:
3696				*qp_attr_mask |= IB_QP_RETRY_CNT | IB_QP_RNR_RETRY |
3697						 IB_QP_MAX_QP_RD_ATOMIC;
 
3698				qp_attr->retry_cnt = cm_id_priv->retry_count;
3699				qp_attr->rnr_retry = cm_id_priv->rnr_retry_count;
3700				qp_attr->max_rd_atomic = cm_id_priv->initiator_depth;
3701				/* fall through */
3702			case IB_QPT_XRC_TGT:
3703				*qp_attr_mask |= IB_QP_TIMEOUT;
3704				qp_attr->timeout = cm_id_priv->av.timeout;
3705				break;
3706			default:
3707				break;
3708			}
3709			if (cm_id_priv->alt_av.ah_attr.dlid) {
3710				*qp_attr_mask |= IB_QP_PATH_MIG_STATE;
3711				qp_attr->path_mig_state = IB_MIG_REARM;
3712			}
3713		} else {
3714			*qp_attr_mask = IB_QP_ALT_PATH | IB_QP_PATH_MIG_STATE;
3715			qp_attr->alt_port_num = cm_id_priv->alt_av.port->port_num;
3716			qp_attr->alt_pkey_index = cm_id_priv->alt_av.pkey_index;
3717			qp_attr->alt_timeout = cm_id_priv->alt_av.timeout;
3718			qp_attr->alt_ah_attr = cm_id_priv->alt_av.ah_attr;
3719			qp_attr->path_mig_state = IB_MIG_REARM;
3720		}
3721		ret = 0;
3722		break;
3723	default:
3724		ret = -EINVAL;
3725		break;
3726	}
3727	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3728	return ret;
3729}
3730
3731int ib_cm_init_qp_attr(struct ib_cm_id *cm_id,
3732		       struct ib_qp_attr *qp_attr,
3733		       int *qp_attr_mask)
3734{
3735	struct cm_id_private *cm_id_priv;
3736	int ret;
3737
3738	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3739	switch (qp_attr->qp_state) {
3740	case IB_QPS_INIT:
3741		ret = cm_init_qp_init_attr(cm_id_priv, qp_attr, qp_attr_mask);
3742		break;
3743	case IB_QPS_RTR:
3744		ret = cm_init_qp_rtr_attr(cm_id_priv, qp_attr, qp_attr_mask);
3745		break;
3746	case IB_QPS_RTS:
3747		ret = cm_init_qp_rts_attr(cm_id_priv, qp_attr, qp_attr_mask);
3748		break;
3749	default:
3750		ret = -EINVAL;
3751		break;
3752	}
3753	return ret;
3754}
3755EXPORT_SYMBOL(ib_cm_init_qp_attr);
3756
 
 
 
 
 
 
 
 
 
 
3757static ssize_t cm_show_counter(struct kobject *obj, struct attribute *attr,
3758			       char *buf)
3759{
3760	struct cm_counter_group *group;
3761	struct cm_counter_attribute *cm_attr;
3762
3763	group = container_of(obj, struct cm_counter_group, obj);
3764	cm_attr = container_of(attr, struct cm_counter_attribute, attr);
3765
3766	return sprintf(buf, "%ld\n",
3767		       atomic_long_read(&group->counter[cm_attr->index]));
3768}
3769
3770static const struct sysfs_ops cm_counter_ops = {
3771	.show = cm_show_counter
3772};
3773
3774static struct kobj_type cm_counter_obj_type = {
3775	.sysfs_ops = &cm_counter_ops,
3776	.default_attrs = cm_counter_default_attrs
3777};
3778
3779static void cm_release_port_obj(struct kobject *obj)
3780{
3781	struct cm_port *cm_port;
3782
3783	cm_port = container_of(obj, struct cm_port, port_obj);
3784	kfree(cm_port);
3785}
3786
3787static struct kobj_type cm_port_obj_type = {
3788	.release = cm_release_port_obj
3789};
3790
3791static char *cm_devnode(struct device *dev, umode_t *mode)
3792{
3793	if (mode)
3794		*mode = 0666;
3795	return kasprintf(GFP_KERNEL, "infiniband/%s", dev_name(dev));
3796}
3797
3798struct class cm_class = {
3799	.owner   = THIS_MODULE,
3800	.name    = "infiniband_cm",
3801	.devnode = cm_devnode,
3802};
3803EXPORT_SYMBOL(cm_class);
3804
3805static int cm_create_port_fs(struct cm_port *port)
3806{
3807	int i, ret;
3808
3809	ret = kobject_init_and_add(&port->port_obj, &cm_port_obj_type,
3810				   &port->cm_dev->device->kobj,
3811				   "%d", port->port_num);
3812	if (ret) {
3813		kfree(port);
3814		return ret;
3815	}
3816
3817	for (i = 0; i < CM_COUNTER_GROUPS; i++) {
3818		ret = kobject_init_and_add(&port->counter_group[i].obj,
3819					   &cm_counter_obj_type,
3820					   &port->port_obj,
3821					   "%s", counter_group_names[i]);
3822		if (ret)
3823			goto error;
3824	}
3825
3826	return 0;
3827
3828error:
3829	while (i--)
3830		kobject_put(&port->counter_group[i].obj);
3831	kobject_put(&port->port_obj);
3832	return ret;
3833
3834}
3835
3836static void cm_remove_port_fs(struct cm_port *port)
3837{
3838	int i;
3839
3840	for (i = 0; i < CM_COUNTER_GROUPS; i++)
3841		kobject_put(&port->counter_group[i].obj);
3842
3843	kobject_put(&port->port_obj);
3844}
3845
3846static void cm_add_one(struct ib_device *ib_device)
3847{
3848	struct cm_device *cm_dev;
3849	struct cm_port *port;
3850	struct ib_mad_reg_req reg_req = {
3851		.mgmt_class = IB_MGMT_CLASS_CM,
3852		.mgmt_class_version = IB_CM_CLASS_VERSION,
3853	};
3854	struct ib_port_modify port_modify = {
3855		.set_port_cap_mask = IB_PORT_CM_SUP
3856	};
3857	unsigned long flags;
3858	int ret;
3859	int count = 0;
3860	u8 i;
3861
 
 
 
3862	cm_dev = kzalloc(sizeof(*cm_dev) + sizeof(*port) *
3863			 ib_device->phys_port_cnt, GFP_KERNEL);
3864	if (!cm_dev)
3865		return;
3866
3867	cm_dev->ib_device = ib_device;
3868	cm_dev->ack_delay = ib_device->attrs.local_ca_ack_delay;
3869	cm_dev->going_down = 0;
3870	cm_dev->device = device_create(&cm_class, &ib_device->dev,
3871				       MKDEV(0, 0), NULL,
3872				       "%s", ib_device->name);
3873	if (IS_ERR(cm_dev->device)) {
3874		kfree(cm_dev);
3875		return;
3876	}
3877
3878	set_bit(IB_MGMT_METHOD_SEND, reg_req.method_mask);
3879	for (i = 1; i <= ib_device->phys_port_cnt; i++) {
3880		if (!rdma_cap_ib_cm(ib_device, i))
3881			continue;
3882
3883		port = kzalloc(sizeof *port, GFP_KERNEL);
3884		if (!port)
3885			goto error1;
3886
3887		cm_dev->port[i-1] = port;
3888		port->cm_dev = cm_dev;
3889		port->port_num = i;
3890
3891		ret = cm_create_port_fs(port);
3892		if (ret)
3893			goto error1;
3894
3895		port->mad_agent = ib_register_mad_agent(ib_device, i,
3896							IB_QPT_GSI,
3897							&reg_req,
3898							0,
3899							cm_send_handler,
3900							cm_recv_handler,
3901							port,
3902							0);
3903		if (IS_ERR(port->mad_agent))
3904			goto error2;
3905
3906		ret = ib_modify_port(ib_device, i, 0, &port_modify);
3907		if (ret)
3908			goto error3;
3909
3910		count++;
3911	}
3912
3913	if (!count)
3914		goto free;
3915
3916	ib_set_client_data(ib_device, &cm_client, cm_dev);
3917
3918	write_lock_irqsave(&cm.device_lock, flags);
3919	list_add_tail(&cm_dev->list, &cm.device_list);
3920	write_unlock_irqrestore(&cm.device_lock, flags);
3921	return;
3922
3923error3:
3924	ib_unregister_mad_agent(port->mad_agent);
3925error2:
3926	cm_remove_port_fs(port);
3927error1:
3928	port_modify.set_port_cap_mask = 0;
3929	port_modify.clr_port_cap_mask = IB_PORT_CM_SUP;
3930	while (--i) {
3931		if (!rdma_cap_ib_cm(ib_device, i))
3932			continue;
3933
3934		port = cm_dev->port[i-1];
3935		ib_modify_port(ib_device, port->port_num, 0, &port_modify);
3936		ib_unregister_mad_agent(port->mad_agent);
3937		cm_remove_port_fs(port);
3938	}
3939free:
3940	device_unregister(cm_dev->device);
3941	kfree(cm_dev);
3942}
3943
3944static void cm_remove_one(struct ib_device *ib_device, void *client_data)
3945{
3946	struct cm_device *cm_dev = client_data;
3947	struct cm_port *port;
3948	struct ib_port_modify port_modify = {
3949		.clr_port_cap_mask = IB_PORT_CM_SUP
3950	};
3951	unsigned long flags;
3952	int i;
3953
 
3954	if (!cm_dev)
3955		return;
3956
3957	write_lock_irqsave(&cm.device_lock, flags);
3958	list_del(&cm_dev->list);
3959	write_unlock_irqrestore(&cm.device_lock, flags);
3960
3961	spin_lock_irq(&cm.lock);
3962	cm_dev->going_down = 1;
3963	spin_unlock_irq(&cm.lock);
3964
3965	for (i = 1; i <= ib_device->phys_port_cnt; i++) {
3966		if (!rdma_cap_ib_cm(ib_device, i))
3967			continue;
3968
3969		port = cm_dev->port[i-1];
3970		ib_modify_port(ib_device, port->port_num, 0, &port_modify);
3971		/*
3972		 * We flush the queue here after the going_down set, this
3973		 * verify that no new works will be queued in the recv handler,
3974		 * after that we can call the unregister_mad_agent
3975		 */
3976		flush_workqueue(cm.wq);
3977		ib_unregister_mad_agent(port->mad_agent);
3978		cm_remove_port_fs(port);
3979	}
3980	device_unregister(cm_dev->device);
3981	kfree(cm_dev);
3982}
3983
3984static int __init ib_cm_init(void)
3985{
3986	int ret;
3987
3988	memset(&cm, 0, sizeof cm);
3989	INIT_LIST_HEAD(&cm.device_list);
3990	rwlock_init(&cm.device_lock);
3991	spin_lock_init(&cm.lock);
3992	cm.listen_service_table = RB_ROOT;
3993	cm.listen_service_id = be64_to_cpu(IB_CM_ASSIGN_SERVICE_ID);
3994	cm.remote_id_table = RB_ROOT;
3995	cm.remote_qp_table = RB_ROOT;
3996	cm.remote_sidr_table = RB_ROOT;
3997	idr_init(&cm.local_id_table);
3998	get_random_bytes(&cm.random_id_operand, sizeof cm.random_id_operand);
 
3999	INIT_LIST_HEAD(&cm.timewait_list);
4000
4001	ret = class_register(&cm_class);
4002	if (ret) {
4003		ret = -ENOMEM;
4004		goto error1;
4005	}
4006
4007	cm.wq = create_workqueue("ib_cm");
4008	if (!cm.wq) {
4009		ret = -ENOMEM;
4010		goto error2;
4011	}
4012
4013	ret = ib_register_client(&cm_client);
4014	if (ret)
4015		goto error3;
4016
4017	return 0;
4018error3:
4019	destroy_workqueue(cm.wq);
4020error2:
4021	class_unregister(&cm_class);
4022error1:
4023	idr_destroy(&cm.local_id_table);
4024	return ret;
4025}
4026
4027static void __exit ib_cm_cleanup(void)
4028{
4029	struct cm_timewait_info *timewait_info, *tmp;
4030
4031	spin_lock_irq(&cm.lock);
4032	list_for_each_entry(timewait_info, &cm.timewait_list, list)
4033		cancel_delayed_work(&timewait_info->work.work);
4034	spin_unlock_irq(&cm.lock);
4035
4036	ib_unregister_client(&cm_client);
4037	destroy_workqueue(cm.wq);
4038
4039	list_for_each_entry_safe(timewait_info, tmp, &cm.timewait_list, list) {
4040		list_del(&timewait_info->list);
4041		kfree(timewait_info);
4042	}
4043
4044	class_unregister(&cm_class);
4045	idr_destroy(&cm.local_id_table);
4046}
4047
4048module_init(ib_cm_init);
4049module_exit(ib_cm_cleanup);
4050