cm.c revision 5a0e3ad6af8660be21ca98a971cd00f331318c05
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	ret = cm_alloc_msg(cm_id_priv, &msg);
1992	if (ret) {
1993		cm_enter_timewait(cm_id_priv);
1994		goto out;
1995	}
1996
1997	cm_format_dreq((struct cm_dreq_msg *) msg->mad, cm_id_priv,
1998		       private_data, private_data_len);
1999	msg->timeout_ms = cm_id_priv->timeout_ms;
2000	msg->context[1] = (void *) (unsigned long) IB_CM_DREQ_SENT;
2001
2002	ret = ib_post_send_mad(msg, NULL);
2003	if (ret) {
2004		cm_enter_timewait(cm_id_priv);
2005		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2006		cm_free_msg(msg);
2007		return ret;
2008	}
2009
2010	cm_id->state = IB_CM_DREQ_SENT;
2011	cm_id_priv->msg = msg;
2012out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2013	return ret;
2014}
2015EXPORT_SYMBOL(ib_send_cm_dreq);
2016
2017static void cm_format_drep(struct cm_drep_msg *drep_msg,
2018			  struct cm_id_private *cm_id_priv,
2019			  const void *private_data,
2020			  u8 private_data_len)
2021{
2022	cm_format_mad_hdr(&drep_msg->hdr, CM_DREP_ATTR_ID, cm_id_priv->tid);
2023	drep_msg->local_comm_id = cm_id_priv->id.local_id;
2024	drep_msg->remote_comm_id = cm_id_priv->id.remote_id;
2025
2026	if (private_data && private_data_len)
2027		memcpy(drep_msg->private_data, private_data, private_data_len);
2028}
2029
2030int ib_send_cm_drep(struct ib_cm_id *cm_id,
2031		    const void *private_data,
2032		    u8 private_data_len)
2033{
2034	struct cm_id_private *cm_id_priv;
2035	struct ib_mad_send_buf *msg;
2036	unsigned long flags;
2037	void *data;
2038	int ret;
2039
2040	if (private_data && private_data_len > IB_CM_DREP_PRIVATE_DATA_SIZE)
2041		return -EINVAL;
2042
2043	data = cm_copy_private_data(private_data, private_data_len);
2044	if (IS_ERR(data))
2045		return PTR_ERR(data);
2046
2047	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2048	spin_lock_irqsave(&cm_id_priv->lock, flags);
2049	if (cm_id->state != IB_CM_DREQ_RCVD) {
2050		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2051		kfree(data);
2052		return -EINVAL;
2053	}
2054
2055	cm_set_private_data(cm_id_priv, data, private_data_len);
2056	cm_enter_timewait(cm_id_priv);
2057
2058	ret = cm_alloc_msg(cm_id_priv, &msg);
2059	if (ret)
2060		goto out;
2061
2062	cm_format_drep((struct cm_drep_msg *) msg->mad, cm_id_priv,
2063		       private_data, private_data_len);
2064
2065	ret = ib_post_send_mad(msg, NULL);
2066	if (ret) {
2067		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2068		cm_free_msg(msg);
2069		return ret;
2070	}
2071
2072out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2073	return ret;
2074}
2075EXPORT_SYMBOL(ib_send_cm_drep);
2076
2077static int cm_issue_drep(struct cm_port *port,
2078			 struct ib_mad_recv_wc *mad_recv_wc)
2079{
2080	struct ib_mad_send_buf *msg = NULL;
2081	struct cm_dreq_msg *dreq_msg;
2082	struct cm_drep_msg *drep_msg;
2083	int ret;
2084
2085	ret = cm_alloc_response_msg(port, mad_recv_wc, &msg);
2086	if (ret)
2087		return ret;
2088
2089	dreq_msg = (struct cm_dreq_msg *) mad_recv_wc->recv_buf.mad;
2090	drep_msg = (struct cm_drep_msg *) msg->mad;
2091
2092	cm_format_mad_hdr(&drep_msg->hdr, CM_DREP_ATTR_ID, dreq_msg->hdr.tid);
2093	drep_msg->remote_comm_id = dreq_msg->local_comm_id;
2094	drep_msg->local_comm_id = dreq_msg->remote_comm_id;
2095
2096	ret = ib_post_send_mad(msg, NULL);
2097	if (ret)
2098		cm_free_msg(msg);
2099
2100	return ret;
2101}
2102
2103static int cm_dreq_handler(struct cm_work *work)
2104{
2105	struct cm_id_private *cm_id_priv;
2106	struct cm_dreq_msg *dreq_msg;
2107	struct ib_mad_send_buf *msg = NULL;
2108	int ret;
2109
2110	dreq_msg = (struct cm_dreq_msg *)work->mad_recv_wc->recv_buf.mad;
2111	cm_id_priv = cm_acquire_id(dreq_msg->remote_comm_id,
2112				   dreq_msg->local_comm_id);
2113	if (!cm_id_priv) {
2114		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2115				counter[CM_DREQ_COUNTER]);
2116		cm_issue_drep(work->port, work->mad_recv_wc);
2117		return -EINVAL;
2118	}
2119
2120	work->cm_event.private_data = &dreq_msg->private_data;
2121
2122	spin_lock_irq(&cm_id_priv->lock);
2123	if (cm_id_priv->local_qpn != cm_dreq_get_remote_qpn(dreq_msg))
2124		goto unlock;
2125
2126	switch (cm_id_priv->id.state) {
2127	case IB_CM_REP_SENT:
2128	case IB_CM_DREQ_SENT:
2129		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2130		break;
2131	case IB_CM_ESTABLISHED:
2132	case IB_CM_MRA_REP_RCVD:
2133		break;
2134	case IB_CM_TIMEWAIT:
2135		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2136				counter[CM_DREQ_COUNTER]);
2137		if (cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg))
2138			goto unlock;
2139
2140		cm_format_drep((struct cm_drep_msg *) msg->mad, cm_id_priv,
2141			       cm_id_priv->private_data,
2142			       cm_id_priv->private_data_len);
2143		spin_unlock_irq(&cm_id_priv->lock);
2144
2145		if (ib_post_send_mad(msg, NULL))
2146			cm_free_msg(msg);
2147		goto deref;
2148	case IB_CM_DREQ_RCVD:
2149		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2150				counter[CM_DREQ_COUNTER]);
2151		goto unlock;
2152	default:
2153		goto unlock;
2154	}
2155	cm_id_priv->id.state = IB_CM_DREQ_RCVD;
2156	cm_id_priv->tid = dreq_msg->hdr.tid;
2157	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2158	if (!ret)
2159		list_add_tail(&work->list, &cm_id_priv->work_list);
2160	spin_unlock_irq(&cm_id_priv->lock);
2161
2162	if (ret)
2163		cm_process_work(cm_id_priv, work);
2164	else
2165		cm_deref_id(cm_id_priv);
2166	return 0;
2167
2168unlock:	spin_unlock_irq(&cm_id_priv->lock);
2169deref:	cm_deref_id(cm_id_priv);
2170	return -EINVAL;
2171}
2172
2173static int cm_drep_handler(struct cm_work *work)
2174{
2175	struct cm_id_private *cm_id_priv;
2176	struct cm_drep_msg *drep_msg;
2177	int ret;
2178
2179	drep_msg = (struct cm_drep_msg *)work->mad_recv_wc->recv_buf.mad;
2180	cm_id_priv = cm_acquire_id(drep_msg->remote_comm_id,
2181				   drep_msg->local_comm_id);
2182	if (!cm_id_priv)
2183		return -EINVAL;
2184
2185	work->cm_event.private_data = &drep_msg->private_data;
2186
2187	spin_lock_irq(&cm_id_priv->lock);
2188	if (cm_id_priv->id.state != IB_CM_DREQ_SENT &&
2189	    cm_id_priv->id.state != IB_CM_DREQ_RCVD) {
2190		spin_unlock_irq(&cm_id_priv->lock);
2191		goto out;
2192	}
2193	cm_enter_timewait(cm_id_priv);
2194
2195	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2196	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2197	if (!ret)
2198		list_add_tail(&work->list, &cm_id_priv->work_list);
2199	spin_unlock_irq(&cm_id_priv->lock);
2200
2201	if (ret)
2202		cm_process_work(cm_id_priv, work);
2203	else
2204		cm_deref_id(cm_id_priv);
2205	return 0;
2206out:
2207	cm_deref_id(cm_id_priv);
2208	return -EINVAL;
2209}
2210
2211int ib_send_cm_rej(struct ib_cm_id *cm_id,
2212		   enum ib_cm_rej_reason reason,
2213		   void *ari,
2214		   u8 ari_length,
2215		   const void *private_data,
2216		   u8 private_data_len)
2217{
2218	struct cm_id_private *cm_id_priv;
2219	struct ib_mad_send_buf *msg;
2220	unsigned long flags;
2221	int ret;
2222
2223	if ((private_data && private_data_len > IB_CM_REJ_PRIVATE_DATA_SIZE) ||
2224	    (ari && ari_length > IB_CM_REJ_ARI_LENGTH))
2225		return -EINVAL;
2226
2227	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2228
2229	spin_lock_irqsave(&cm_id_priv->lock, flags);
2230	switch (cm_id->state) {
2231	case IB_CM_REQ_SENT:
2232	case IB_CM_MRA_REQ_RCVD:
2233	case IB_CM_REQ_RCVD:
2234	case IB_CM_MRA_REQ_SENT:
2235	case IB_CM_REP_RCVD:
2236	case IB_CM_MRA_REP_SENT:
2237		ret = cm_alloc_msg(cm_id_priv, &msg);
2238		if (!ret)
2239			cm_format_rej((struct cm_rej_msg *) msg->mad,
2240				      cm_id_priv, reason, ari, ari_length,
2241				      private_data, private_data_len);
2242
2243		cm_reset_to_idle(cm_id_priv);
2244		break;
2245	case IB_CM_REP_SENT:
2246	case IB_CM_MRA_REP_RCVD:
2247		ret = cm_alloc_msg(cm_id_priv, &msg);
2248		if (!ret)
2249			cm_format_rej((struct cm_rej_msg *) msg->mad,
2250				      cm_id_priv, reason, ari, ari_length,
2251				      private_data, private_data_len);
2252
2253		cm_enter_timewait(cm_id_priv);
2254		break;
2255	default:
2256		ret = -EINVAL;
2257		goto out;
2258	}
2259
2260	if (ret)
2261		goto out;
2262
2263	ret = ib_post_send_mad(msg, NULL);
2264	if (ret)
2265		cm_free_msg(msg);
2266
2267out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2268	return ret;
2269}
2270EXPORT_SYMBOL(ib_send_cm_rej);
2271
2272static void cm_format_rej_event(struct cm_work *work)
2273{
2274	struct cm_rej_msg *rej_msg;
2275	struct ib_cm_rej_event_param *param;
2276
2277	rej_msg = (struct cm_rej_msg *)work->mad_recv_wc->recv_buf.mad;
2278	param = &work->cm_event.param.rej_rcvd;
2279	param->ari = rej_msg->ari;
2280	param->ari_length = cm_rej_get_reject_info_len(rej_msg);
2281	param->reason = __be16_to_cpu(rej_msg->reason);
2282	work->cm_event.private_data = &rej_msg->private_data;
2283}
2284
2285static struct cm_id_private * cm_acquire_rejected_id(struct cm_rej_msg *rej_msg)
2286{
2287	struct cm_timewait_info *timewait_info;
2288	struct cm_id_private *cm_id_priv;
2289	__be32 remote_id;
2290
2291	remote_id = rej_msg->local_comm_id;
2292
2293	if (__be16_to_cpu(rej_msg->reason) == IB_CM_REJ_TIMEOUT) {
2294		spin_lock_irq(&cm.lock);
2295		timewait_info = cm_find_remote_id( *((__be64 *) rej_msg->ari),
2296						  remote_id);
2297		if (!timewait_info) {
2298			spin_unlock_irq(&cm.lock);
2299			return NULL;
2300		}
2301		cm_id_priv = idr_find(&cm.local_id_table, (__force int)
2302				      (timewait_info->work.local_id ^
2303				       cm.random_id_operand));
2304		if (cm_id_priv) {
2305			if (cm_id_priv->id.remote_id == remote_id)
2306				atomic_inc(&cm_id_priv->refcount);
2307			else
2308				cm_id_priv = NULL;
2309		}
2310		spin_unlock_irq(&cm.lock);
2311	} else if (cm_rej_get_msg_rejected(rej_msg) == CM_MSG_RESPONSE_REQ)
2312		cm_id_priv = cm_acquire_id(rej_msg->remote_comm_id, 0);
2313	else
2314		cm_id_priv = cm_acquire_id(rej_msg->remote_comm_id, remote_id);
2315
2316	return cm_id_priv;
2317}
2318
2319static int cm_rej_handler(struct cm_work *work)
2320{
2321	struct cm_id_private *cm_id_priv;
2322	struct cm_rej_msg *rej_msg;
2323	int ret;
2324
2325	rej_msg = (struct cm_rej_msg *)work->mad_recv_wc->recv_buf.mad;
2326	cm_id_priv = cm_acquire_rejected_id(rej_msg);
2327	if (!cm_id_priv)
2328		return -EINVAL;
2329
2330	cm_format_rej_event(work);
2331
2332	spin_lock_irq(&cm_id_priv->lock);
2333	switch (cm_id_priv->id.state) {
2334	case IB_CM_REQ_SENT:
2335	case IB_CM_MRA_REQ_RCVD:
2336	case IB_CM_REP_SENT:
2337	case IB_CM_MRA_REP_RCVD:
2338		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2339		/* fall through */
2340	case IB_CM_REQ_RCVD:
2341	case IB_CM_MRA_REQ_SENT:
2342		if (__be16_to_cpu(rej_msg->reason) == IB_CM_REJ_STALE_CONN)
2343			cm_enter_timewait(cm_id_priv);
2344		else
2345			cm_reset_to_idle(cm_id_priv);
2346		break;
2347	case IB_CM_DREQ_SENT:
2348		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2349		/* fall through */
2350	case IB_CM_REP_RCVD:
2351	case IB_CM_MRA_REP_SENT:
2352	case IB_CM_ESTABLISHED:
2353		cm_enter_timewait(cm_id_priv);
2354		break;
2355	default:
2356		spin_unlock_irq(&cm_id_priv->lock);
2357		ret = -EINVAL;
2358		goto out;
2359	}
2360
2361	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2362	if (!ret)
2363		list_add_tail(&work->list, &cm_id_priv->work_list);
2364	spin_unlock_irq(&cm_id_priv->lock);
2365
2366	if (ret)
2367		cm_process_work(cm_id_priv, work);
2368	else
2369		cm_deref_id(cm_id_priv);
2370	return 0;
2371out:
2372	cm_deref_id(cm_id_priv);
2373	return -EINVAL;
2374}
2375
2376int ib_send_cm_mra(struct ib_cm_id *cm_id,
2377		   u8 service_timeout,
2378		   const void *private_data,
2379		   u8 private_data_len)
2380{
2381	struct cm_id_private *cm_id_priv;
2382	struct ib_mad_send_buf *msg;
2383	enum ib_cm_state cm_state;
2384	enum ib_cm_lap_state lap_state;
2385	enum cm_msg_response msg_response;
2386	void *data;
2387	unsigned long flags;
2388	int ret;
2389
2390	if (private_data && private_data_len > IB_CM_MRA_PRIVATE_DATA_SIZE)
2391		return -EINVAL;
2392
2393	data = cm_copy_private_data(private_data, private_data_len);
2394	if (IS_ERR(data))
2395		return PTR_ERR(data);
2396
2397	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2398
2399	spin_lock_irqsave(&cm_id_priv->lock, flags);
2400	switch(cm_id_priv->id.state) {
2401	case IB_CM_REQ_RCVD:
2402		cm_state = IB_CM_MRA_REQ_SENT;
2403		lap_state = cm_id->lap_state;
2404		msg_response = CM_MSG_RESPONSE_REQ;
2405		break;
2406	case IB_CM_REP_RCVD:
2407		cm_state = IB_CM_MRA_REP_SENT;
2408		lap_state = cm_id->lap_state;
2409		msg_response = CM_MSG_RESPONSE_REP;
2410		break;
2411	case IB_CM_ESTABLISHED:
2412		cm_state = cm_id->state;
2413		lap_state = IB_CM_MRA_LAP_SENT;
2414		msg_response = CM_MSG_RESPONSE_OTHER;
2415		break;
2416	default:
2417		ret = -EINVAL;
2418		goto error1;
2419	}
2420
2421	if (!(service_timeout & IB_CM_MRA_FLAG_DELAY)) {
2422		ret = cm_alloc_msg(cm_id_priv, &msg);
2423		if (ret)
2424			goto error1;
2425
2426		cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
2427			      msg_response, service_timeout,
2428			      private_data, private_data_len);
2429		ret = ib_post_send_mad(msg, NULL);
2430		if (ret)
2431			goto error2;
2432	}
2433
2434	cm_id->state = cm_state;
2435	cm_id->lap_state = lap_state;
2436	cm_id_priv->service_timeout = service_timeout;
2437	cm_set_private_data(cm_id_priv, data, private_data_len);
2438	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2439	return 0;
2440
2441error1:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2442	kfree(data);
2443	return ret;
2444
2445error2:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2446	kfree(data);
2447	cm_free_msg(msg);
2448	return ret;
2449}
2450EXPORT_SYMBOL(ib_send_cm_mra);
2451
2452static struct cm_id_private * cm_acquire_mraed_id(struct cm_mra_msg *mra_msg)
2453{
2454	switch (cm_mra_get_msg_mraed(mra_msg)) {
2455	case CM_MSG_RESPONSE_REQ:
2456		return cm_acquire_id(mra_msg->remote_comm_id, 0);
2457	case CM_MSG_RESPONSE_REP:
2458	case CM_MSG_RESPONSE_OTHER:
2459		return cm_acquire_id(mra_msg->remote_comm_id,
2460				     mra_msg->local_comm_id);
2461	default:
2462		return NULL;
2463	}
2464}
2465
2466static int cm_mra_handler(struct cm_work *work)
2467{
2468	struct cm_id_private *cm_id_priv;
2469	struct cm_mra_msg *mra_msg;
2470	int timeout, ret;
2471
2472	mra_msg = (struct cm_mra_msg *)work->mad_recv_wc->recv_buf.mad;
2473	cm_id_priv = cm_acquire_mraed_id(mra_msg);
2474	if (!cm_id_priv)
2475		return -EINVAL;
2476
2477	work->cm_event.private_data = &mra_msg->private_data;
2478	work->cm_event.param.mra_rcvd.service_timeout =
2479					cm_mra_get_service_timeout(mra_msg);
2480	timeout = cm_convert_to_ms(cm_mra_get_service_timeout(mra_msg)) +
2481		  cm_convert_to_ms(cm_id_priv->av.timeout);
2482
2483	spin_lock_irq(&cm_id_priv->lock);
2484	switch (cm_id_priv->id.state) {
2485	case IB_CM_REQ_SENT:
2486		if (cm_mra_get_msg_mraed(mra_msg) != CM_MSG_RESPONSE_REQ ||
2487		    ib_modify_mad(cm_id_priv->av.port->mad_agent,
2488				  cm_id_priv->msg, timeout))
2489			goto out;
2490		cm_id_priv->id.state = IB_CM_MRA_REQ_RCVD;
2491		break;
2492	case IB_CM_REP_SENT:
2493		if (cm_mra_get_msg_mraed(mra_msg) != CM_MSG_RESPONSE_REP ||
2494		    ib_modify_mad(cm_id_priv->av.port->mad_agent,
2495				  cm_id_priv->msg, timeout))
2496			goto out;
2497		cm_id_priv->id.state = IB_CM_MRA_REP_RCVD;
2498		break;
2499	case IB_CM_ESTABLISHED:
2500		if (cm_mra_get_msg_mraed(mra_msg) != CM_MSG_RESPONSE_OTHER ||
2501		    cm_id_priv->id.lap_state != IB_CM_LAP_SENT ||
2502		    ib_modify_mad(cm_id_priv->av.port->mad_agent,
2503				  cm_id_priv->msg, timeout)) {
2504			if (cm_id_priv->id.lap_state == IB_CM_MRA_LAP_RCVD)
2505				atomic_long_inc(&work->port->
2506						counter_group[CM_RECV_DUPLICATES].
2507						counter[CM_MRA_COUNTER]);
2508			goto out;
2509		}
2510		cm_id_priv->id.lap_state = IB_CM_MRA_LAP_RCVD;
2511		break;
2512	case IB_CM_MRA_REQ_RCVD:
2513	case IB_CM_MRA_REP_RCVD:
2514		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2515				counter[CM_MRA_COUNTER]);
2516		/* fall through */
2517	default:
2518		goto out;
2519	}
2520
2521	cm_id_priv->msg->context[1] = (void *) (unsigned long)
2522				      cm_id_priv->id.state;
2523	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2524	if (!ret)
2525		list_add_tail(&work->list, &cm_id_priv->work_list);
2526	spin_unlock_irq(&cm_id_priv->lock);
2527
2528	if (ret)
2529		cm_process_work(cm_id_priv, work);
2530	else
2531		cm_deref_id(cm_id_priv);
2532	return 0;
2533out:
2534	spin_unlock_irq(&cm_id_priv->lock);
2535	cm_deref_id(cm_id_priv);
2536	return -EINVAL;
2537}
2538
2539static void cm_format_lap(struct cm_lap_msg *lap_msg,
2540			  struct cm_id_private *cm_id_priv,
2541			  struct ib_sa_path_rec *alternate_path,
2542			  const void *private_data,
2543			  u8 private_data_len)
2544{
2545	cm_format_mad_hdr(&lap_msg->hdr, CM_LAP_ATTR_ID,
2546			  cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_LAP));
2547	lap_msg->local_comm_id = cm_id_priv->id.local_id;
2548	lap_msg->remote_comm_id = cm_id_priv->id.remote_id;
2549	cm_lap_set_remote_qpn(lap_msg, cm_id_priv->remote_qpn);
2550	/* todo: need remote CM response timeout */
2551	cm_lap_set_remote_resp_timeout(lap_msg, 0x1F);
2552	lap_msg->alt_local_lid = alternate_path->slid;
2553	lap_msg->alt_remote_lid = alternate_path->dlid;
2554	lap_msg->alt_local_gid = alternate_path->sgid;
2555	lap_msg->alt_remote_gid = alternate_path->dgid;
2556	cm_lap_set_flow_label(lap_msg, alternate_path->flow_label);
2557	cm_lap_set_traffic_class(lap_msg, alternate_path->traffic_class);
2558	lap_msg->alt_hop_limit = alternate_path->hop_limit;
2559	cm_lap_set_packet_rate(lap_msg, alternate_path->rate);
2560	cm_lap_set_sl(lap_msg, alternate_path->sl);
2561	cm_lap_set_subnet_local(lap_msg, 1); /* local only... */
2562	cm_lap_set_local_ack_timeout(lap_msg,
2563		cm_ack_timeout(cm_id_priv->av.port->cm_dev->ack_delay,
2564			       alternate_path->packet_life_time));
2565
2566	if (private_data && private_data_len)
2567		memcpy(lap_msg->private_data, private_data, private_data_len);
2568}
2569
2570int ib_send_cm_lap(struct ib_cm_id *cm_id,
2571		   struct ib_sa_path_rec *alternate_path,
2572		   const void *private_data,
2573		   u8 private_data_len)
2574{
2575	struct cm_id_private *cm_id_priv;
2576	struct ib_mad_send_buf *msg;
2577	unsigned long flags;
2578	int ret;
2579
2580	if (private_data && private_data_len > IB_CM_LAP_PRIVATE_DATA_SIZE)
2581		return -EINVAL;
2582
2583	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2584	spin_lock_irqsave(&cm_id_priv->lock, flags);
2585	if (cm_id->state != IB_CM_ESTABLISHED ||
2586	    (cm_id->lap_state != IB_CM_LAP_UNINIT &&
2587	     cm_id->lap_state != IB_CM_LAP_IDLE)) {
2588		ret = -EINVAL;
2589		goto out;
2590	}
2591
2592	ret = cm_init_av_by_path(alternate_path, &cm_id_priv->alt_av);
2593	if (ret)
2594		goto out;
2595	cm_id_priv->alt_av.timeout =
2596			cm_ack_timeout(cm_id_priv->target_ack_delay,
2597				       cm_id_priv->alt_av.timeout - 1);
2598
2599	ret = cm_alloc_msg(cm_id_priv, &msg);
2600	if (ret)
2601		goto out;
2602
2603	cm_format_lap((struct cm_lap_msg *) msg->mad, cm_id_priv,
2604		      alternate_path, private_data, private_data_len);
2605	msg->timeout_ms = cm_id_priv->timeout_ms;
2606	msg->context[1] = (void *) (unsigned long) IB_CM_ESTABLISHED;
2607
2608	ret = ib_post_send_mad(msg, NULL);
2609	if (ret) {
2610		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2611		cm_free_msg(msg);
2612		return ret;
2613	}
2614
2615	cm_id->lap_state = IB_CM_LAP_SENT;
2616	cm_id_priv->msg = msg;
2617
2618out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2619	return ret;
2620}
2621EXPORT_SYMBOL(ib_send_cm_lap);
2622
2623static void cm_format_path_from_lap(struct cm_id_private *cm_id_priv,
2624				    struct ib_sa_path_rec *path,
2625				    struct cm_lap_msg *lap_msg)
2626{
2627	memset(path, 0, sizeof *path);
2628	path->dgid = lap_msg->alt_local_gid;
2629	path->sgid = lap_msg->alt_remote_gid;
2630	path->dlid = lap_msg->alt_local_lid;
2631	path->slid = lap_msg->alt_remote_lid;
2632	path->flow_label = cm_lap_get_flow_label(lap_msg);
2633	path->hop_limit = lap_msg->alt_hop_limit;
2634	path->traffic_class = cm_lap_get_traffic_class(lap_msg);
2635	path->reversible = 1;
2636	path->pkey = cm_id_priv->pkey;
2637	path->sl = cm_lap_get_sl(lap_msg);
2638	path->mtu_selector = IB_SA_EQ;
2639	path->mtu = cm_id_priv->path_mtu;
2640	path->rate_selector = IB_SA_EQ;
2641	path->rate = cm_lap_get_packet_rate(lap_msg);
2642	path->packet_life_time_selector = IB_SA_EQ;
2643	path->packet_life_time = cm_lap_get_local_ack_timeout(lap_msg);
2644	path->packet_life_time -= (path->packet_life_time > 0);
2645}
2646
2647static int cm_lap_handler(struct cm_work *work)
2648{
2649	struct cm_id_private *cm_id_priv;
2650	struct cm_lap_msg *lap_msg;
2651	struct ib_cm_lap_event_param *param;
2652	struct ib_mad_send_buf *msg = NULL;
2653	int ret;
2654
2655	/* todo: verify LAP request and send reject APR if invalid. */
2656	lap_msg = (struct cm_lap_msg *)work->mad_recv_wc->recv_buf.mad;
2657	cm_id_priv = cm_acquire_id(lap_msg->remote_comm_id,
2658				   lap_msg->local_comm_id);
2659	if (!cm_id_priv)
2660		return -EINVAL;
2661
2662	param = &work->cm_event.param.lap_rcvd;
2663	param->alternate_path = &work->path[0];
2664	cm_format_path_from_lap(cm_id_priv, param->alternate_path, lap_msg);
2665	work->cm_event.private_data = &lap_msg->private_data;
2666
2667	spin_lock_irq(&cm_id_priv->lock);
2668	if (cm_id_priv->id.state != IB_CM_ESTABLISHED)
2669		goto unlock;
2670
2671	switch (cm_id_priv->id.lap_state) {
2672	case IB_CM_LAP_UNINIT:
2673	case IB_CM_LAP_IDLE:
2674		break;
2675	case IB_CM_MRA_LAP_SENT:
2676		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2677				counter[CM_LAP_COUNTER]);
2678		if (cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg))
2679			goto unlock;
2680
2681		cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
2682			      CM_MSG_RESPONSE_OTHER,
2683			      cm_id_priv->service_timeout,
2684			      cm_id_priv->private_data,
2685			      cm_id_priv->private_data_len);
2686		spin_unlock_irq(&cm_id_priv->lock);
2687
2688		if (ib_post_send_mad(msg, NULL))
2689			cm_free_msg(msg);
2690		goto deref;
2691	case IB_CM_LAP_RCVD:
2692		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2693				counter[CM_LAP_COUNTER]);
2694		goto unlock;
2695	default:
2696		goto unlock;
2697	}
2698
2699	cm_id_priv->id.lap_state = IB_CM_LAP_RCVD;
2700	cm_id_priv->tid = lap_msg->hdr.tid;
2701	cm_init_av_for_response(work->port, work->mad_recv_wc->wc,
2702				work->mad_recv_wc->recv_buf.grh,
2703				&cm_id_priv->av);
2704	cm_init_av_by_path(param->alternate_path, &cm_id_priv->alt_av);
2705	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2706	if (!ret)
2707		list_add_tail(&work->list, &cm_id_priv->work_list);
2708	spin_unlock_irq(&cm_id_priv->lock);
2709
2710	if (ret)
2711		cm_process_work(cm_id_priv, work);
2712	else
2713		cm_deref_id(cm_id_priv);
2714	return 0;
2715
2716unlock:	spin_unlock_irq(&cm_id_priv->lock);
2717deref:	cm_deref_id(cm_id_priv);
2718	return -EINVAL;
2719}
2720
2721static void cm_format_apr(struct cm_apr_msg *apr_msg,
2722			  struct cm_id_private *cm_id_priv,
2723			  enum ib_cm_apr_status status,
2724			  void *info,
2725			  u8 info_length,
2726			  const void *private_data,
2727			  u8 private_data_len)
2728{
2729	cm_format_mad_hdr(&apr_msg->hdr, CM_APR_ATTR_ID, cm_id_priv->tid);
2730	apr_msg->local_comm_id = cm_id_priv->id.local_id;
2731	apr_msg->remote_comm_id = cm_id_priv->id.remote_id;
2732	apr_msg->ap_status = (u8) status;
2733
2734	if (info && info_length) {
2735		apr_msg->info_length = info_length;
2736		memcpy(apr_msg->info, info, info_length);
2737	}
2738
2739	if (private_data && private_data_len)
2740		memcpy(apr_msg->private_data, private_data, private_data_len);
2741}
2742
2743int ib_send_cm_apr(struct ib_cm_id *cm_id,
2744		   enum ib_cm_apr_status status,
2745		   void *info,
2746		   u8 info_length,
2747		   const void *private_data,
2748		   u8 private_data_len)
2749{
2750	struct cm_id_private *cm_id_priv;
2751	struct ib_mad_send_buf *msg;
2752	unsigned long flags;
2753	int ret;
2754
2755	if ((private_data && private_data_len > IB_CM_APR_PRIVATE_DATA_SIZE) ||
2756	    (info && info_length > IB_CM_APR_INFO_LENGTH))
2757		return -EINVAL;
2758
2759	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2760	spin_lock_irqsave(&cm_id_priv->lock, flags);
2761	if (cm_id->state != IB_CM_ESTABLISHED ||
2762	    (cm_id->lap_state != IB_CM_LAP_RCVD &&
2763	     cm_id->lap_state != IB_CM_MRA_LAP_SENT)) {
2764		ret = -EINVAL;
2765		goto out;
2766	}
2767
2768	ret = cm_alloc_msg(cm_id_priv, &msg);
2769	if (ret)
2770		goto out;
2771
2772	cm_format_apr((struct cm_apr_msg *) msg->mad, cm_id_priv, status,
2773		      info, info_length, private_data, private_data_len);
2774	ret = ib_post_send_mad(msg, NULL);
2775	if (ret) {
2776		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2777		cm_free_msg(msg);
2778		return ret;
2779	}
2780
2781	cm_id->lap_state = IB_CM_LAP_IDLE;
2782out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2783	return ret;
2784}
2785EXPORT_SYMBOL(ib_send_cm_apr);
2786
2787static int cm_apr_handler(struct cm_work *work)
2788{
2789	struct cm_id_private *cm_id_priv;
2790	struct cm_apr_msg *apr_msg;
2791	int ret;
2792
2793	apr_msg = (struct cm_apr_msg *)work->mad_recv_wc->recv_buf.mad;
2794	cm_id_priv = cm_acquire_id(apr_msg->remote_comm_id,
2795				   apr_msg->local_comm_id);
2796	if (!cm_id_priv)
2797		return -EINVAL; /* Unmatched reply. */
2798
2799	work->cm_event.param.apr_rcvd.ap_status = apr_msg->ap_status;
2800	work->cm_event.param.apr_rcvd.apr_info = &apr_msg->info;
2801	work->cm_event.param.apr_rcvd.info_len = apr_msg->info_length;
2802	work->cm_event.private_data = &apr_msg->private_data;
2803
2804	spin_lock_irq(&cm_id_priv->lock);
2805	if (cm_id_priv->id.state != IB_CM_ESTABLISHED ||
2806	    (cm_id_priv->id.lap_state != IB_CM_LAP_SENT &&
2807	     cm_id_priv->id.lap_state != IB_CM_MRA_LAP_RCVD)) {
2808		spin_unlock_irq(&cm_id_priv->lock);
2809		goto out;
2810	}
2811	cm_id_priv->id.lap_state = IB_CM_LAP_IDLE;
2812	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2813	cm_id_priv->msg = NULL;
2814
2815	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2816	if (!ret)
2817		list_add_tail(&work->list, &cm_id_priv->work_list);
2818	spin_unlock_irq(&cm_id_priv->lock);
2819
2820	if (ret)
2821		cm_process_work(cm_id_priv, work);
2822	else
2823		cm_deref_id(cm_id_priv);
2824	return 0;
2825out:
2826	cm_deref_id(cm_id_priv);
2827	return -EINVAL;
2828}
2829
2830static int cm_timewait_handler(struct cm_work *work)
2831{
2832	struct cm_timewait_info *timewait_info;
2833	struct cm_id_private *cm_id_priv;
2834	int ret;
2835
2836	timewait_info = (struct cm_timewait_info *)work;
2837	spin_lock_irq(&cm.lock);
2838	list_del(&timewait_info->list);
2839	spin_unlock_irq(&cm.lock);
2840
2841	cm_id_priv = cm_acquire_id(timewait_info->work.local_id,
2842				   timewait_info->work.remote_id);
2843	if (!cm_id_priv)
2844		return -EINVAL;
2845
2846	spin_lock_irq(&cm_id_priv->lock);
2847	if (cm_id_priv->id.state != IB_CM_TIMEWAIT ||
2848	    cm_id_priv->remote_qpn != timewait_info->remote_qpn) {
2849		spin_unlock_irq(&cm_id_priv->lock);
2850		goto out;
2851	}
2852	cm_id_priv->id.state = IB_CM_IDLE;
2853	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2854	if (!ret)
2855		list_add_tail(&work->list, &cm_id_priv->work_list);
2856	spin_unlock_irq(&cm_id_priv->lock);
2857
2858	if (ret)
2859		cm_process_work(cm_id_priv, work);
2860	else
2861		cm_deref_id(cm_id_priv);
2862	return 0;
2863out:
2864	cm_deref_id(cm_id_priv);
2865	return -EINVAL;
2866}
2867
2868static void cm_format_sidr_req(struct cm_sidr_req_msg *sidr_req_msg,
2869			       struct cm_id_private *cm_id_priv,
2870			       struct ib_cm_sidr_req_param *param)
2871{
2872	cm_format_mad_hdr(&sidr_req_msg->hdr, CM_SIDR_REQ_ATTR_ID,
2873			  cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_SIDR));
2874	sidr_req_msg->request_id = cm_id_priv->id.local_id;
2875	sidr_req_msg->pkey = param->path->pkey;
2876	sidr_req_msg->service_id = param->service_id;
2877
2878	if (param->private_data && param->private_data_len)
2879		memcpy(sidr_req_msg->private_data, param->private_data,
2880		       param->private_data_len);
2881}
2882
2883int ib_send_cm_sidr_req(struct ib_cm_id *cm_id,
2884			struct ib_cm_sidr_req_param *param)
2885{
2886	struct cm_id_private *cm_id_priv;
2887	struct ib_mad_send_buf *msg;
2888	unsigned long flags;
2889	int ret;
2890
2891	if (!param->path || (param->private_data &&
2892	     param->private_data_len > IB_CM_SIDR_REQ_PRIVATE_DATA_SIZE))
2893		return -EINVAL;
2894
2895	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2896	ret = cm_init_av_by_path(param->path, &cm_id_priv->av);
2897	if (ret)
2898		goto out;
2899
2900	cm_id->service_id = param->service_id;
2901	cm_id->service_mask = ~cpu_to_be64(0);
2902	cm_id_priv->timeout_ms = param->timeout_ms;
2903	cm_id_priv->max_cm_retries = param->max_cm_retries;
2904	ret = cm_alloc_msg(cm_id_priv, &msg);
2905	if (ret)
2906		goto out;
2907
2908	cm_format_sidr_req((struct cm_sidr_req_msg *) msg->mad, cm_id_priv,
2909			   param);
2910	msg->timeout_ms = cm_id_priv->timeout_ms;
2911	msg->context[1] = (void *) (unsigned long) IB_CM_SIDR_REQ_SENT;
2912
2913	spin_lock_irqsave(&cm_id_priv->lock, flags);
2914	if (cm_id->state == IB_CM_IDLE)
2915		ret = ib_post_send_mad(msg, NULL);
2916	else
2917		ret = -EINVAL;
2918
2919	if (ret) {
2920		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2921		cm_free_msg(msg);
2922		goto out;
2923	}
2924	cm_id->state = IB_CM_SIDR_REQ_SENT;
2925	cm_id_priv->msg = msg;
2926	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2927out:
2928	return ret;
2929}
2930EXPORT_SYMBOL(ib_send_cm_sidr_req);
2931
2932static void cm_format_sidr_req_event(struct cm_work *work,
2933				     struct ib_cm_id *listen_id)
2934{
2935	struct cm_sidr_req_msg *sidr_req_msg;
2936	struct ib_cm_sidr_req_event_param *param;
2937
2938	sidr_req_msg = (struct cm_sidr_req_msg *)
2939				work->mad_recv_wc->recv_buf.mad;
2940	param = &work->cm_event.param.sidr_req_rcvd;
2941	param->pkey = __be16_to_cpu(sidr_req_msg->pkey);
2942	param->listen_id = listen_id;
2943	param->port = work->port->port_num;
2944	work->cm_event.private_data = &sidr_req_msg->private_data;
2945}
2946
2947static int cm_sidr_req_handler(struct cm_work *work)
2948{
2949	struct ib_cm_id *cm_id;
2950	struct cm_id_private *cm_id_priv, *cur_cm_id_priv;
2951	struct cm_sidr_req_msg *sidr_req_msg;
2952	struct ib_wc *wc;
2953
2954	cm_id = ib_create_cm_id(work->port->cm_dev->ib_device, NULL, NULL);
2955	if (IS_ERR(cm_id))
2956		return PTR_ERR(cm_id);
2957	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2958
2959	/* Record SGID/SLID and request ID for lookup. */
2960	sidr_req_msg = (struct cm_sidr_req_msg *)
2961				work->mad_recv_wc->recv_buf.mad;
2962	wc = work->mad_recv_wc->wc;
2963	cm_id_priv->av.dgid.global.subnet_prefix = cpu_to_be64(wc->slid);
2964	cm_id_priv->av.dgid.global.interface_id = 0;
2965	cm_init_av_for_response(work->port, work->mad_recv_wc->wc,
2966				work->mad_recv_wc->recv_buf.grh,
2967				&cm_id_priv->av);
2968	cm_id_priv->id.remote_id = sidr_req_msg->request_id;
2969	cm_id_priv->tid = sidr_req_msg->hdr.tid;
2970	atomic_inc(&cm_id_priv->work_count);
2971
2972	spin_lock_irq(&cm.lock);
2973	cur_cm_id_priv = cm_insert_remote_sidr(cm_id_priv);
2974	if (cur_cm_id_priv) {
2975		spin_unlock_irq(&cm.lock);
2976		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2977				counter[CM_SIDR_REQ_COUNTER]);
2978		goto out; /* Duplicate message. */
2979	}
2980	cm_id_priv->id.state = IB_CM_SIDR_REQ_RCVD;
2981	cur_cm_id_priv = cm_find_listen(cm_id->device,
2982					sidr_req_msg->service_id,
2983					sidr_req_msg->private_data);
2984	if (!cur_cm_id_priv) {
2985		spin_unlock_irq(&cm.lock);
2986		cm_reject_sidr_req(cm_id_priv, IB_SIDR_UNSUPPORTED);
2987		goto out; /* No match. */
2988	}
2989	atomic_inc(&cur_cm_id_priv->refcount);
2990	spin_unlock_irq(&cm.lock);
2991
2992	cm_id_priv->id.cm_handler = cur_cm_id_priv->id.cm_handler;
2993	cm_id_priv->id.context = cur_cm_id_priv->id.context;
2994	cm_id_priv->id.service_id = sidr_req_msg->service_id;
2995	cm_id_priv->id.service_mask = ~cpu_to_be64(0);
2996
2997	cm_format_sidr_req_event(work, &cur_cm_id_priv->id);
2998	cm_process_work(cm_id_priv, work);
2999	cm_deref_id(cur_cm_id_priv);
3000	return 0;
3001out:
3002	ib_destroy_cm_id(&cm_id_priv->id);
3003	return -EINVAL;
3004}
3005
3006static void cm_format_sidr_rep(struct cm_sidr_rep_msg *sidr_rep_msg,
3007			       struct cm_id_private *cm_id_priv,
3008			       struct ib_cm_sidr_rep_param *param)
3009{
3010	cm_format_mad_hdr(&sidr_rep_msg->hdr, CM_SIDR_REP_ATTR_ID,
3011			  cm_id_priv->tid);
3012	sidr_rep_msg->request_id = cm_id_priv->id.remote_id;
3013	sidr_rep_msg->status = param->status;
3014	cm_sidr_rep_set_qpn(sidr_rep_msg, cpu_to_be32(param->qp_num));
3015	sidr_rep_msg->service_id = cm_id_priv->id.service_id;
3016	sidr_rep_msg->qkey = cpu_to_be32(param->qkey);
3017
3018	if (param->info && param->info_length)
3019		memcpy(sidr_rep_msg->info, param->info, param->info_length);
3020
3021	if (param->private_data && param->private_data_len)
3022		memcpy(sidr_rep_msg->private_data, param->private_data,
3023		       param->private_data_len);
3024}
3025
3026int ib_send_cm_sidr_rep(struct ib_cm_id *cm_id,
3027			struct ib_cm_sidr_rep_param *param)
3028{
3029	struct cm_id_private *cm_id_priv;
3030	struct ib_mad_send_buf *msg;
3031	unsigned long flags;
3032	int ret;
3033
3034	if ((param->info && param->info_length > IB_CM_SIDR_REP_INFO_LENGTH) ||
3035	    (param->private_data &&
3036	     param->private_data_len > IB_CM_SIDR_REP_PRIVATE_DATA_SIZE))
3037		return -EINVAL;
3038
3039	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3040	spin_lock_irqsave(&cm_id_priv->lock, flags);
3041	if (cm_id->state != IB_CM_SIDR_REQ_RCVD) {
3042		ret = -EINVAL;
3043		goto error;
3044	}
3045
3046	ret = cm_alloc_msg(cm_id_priv, &msg);
3047	if (ret)
3048		goto error;
3049
3050	cm_format_sidr_rep((struct cm_sidr_rep_msg *) msg->mad, cm_id_priv,
3051			   param);
3052	ret = ib_post_send_mad(msg, NULL);
3053	if (ret) {
3054		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3055		cm_free_msg(msg);
3056		return ret;
3057	}
3058	cm_id->state = IB_CM_IDLE;
3059	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3060
3061	spin_lock_irqsave(&cm.lock, flags);
3062	rb_erase(&cm_id_priv->sidr_id_node, &cm.remote_sidr_table);
3063	spin_unlock_irqrestore(&cm.lock, flags);
3064	return 0;
3065
3066error:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3067	return ret;
3068}
3069EXPORT_SYMBOL(ib_send_cm_sidr_rep);
3070
3071static void cm_format_sidr_rep_event(struct cm_work *work)
3072{
3073	struct cm_sidr_rep_msg *sidr_rep_msg;
3074	struct ib_cm_sidr_rep_event_param *param;
3075
3076	sidr_rep_msg = (struct cm_sidr_rep_msg *)
3077				work->mad_recv_wc->recv_buf.mad;
3078	param = &work->cm_event.param.sidr_rep_rcvd;
3079	param->status = sidr_rep_msg->status;
3080	param->qkey = be32_to_cpu(sidr_rep_msg->qkey);
3081	param->qpn = be32_to_cpu(cm_sidr_rep_get_qpn(sidr_rep_msg));
3082	param->info = &sidr_rep_msg->info;
3083	param->info_len = sidr_rep_msg->info_length;
3084	work->cm_event.private_data = &sidr_rep_msg->private_data;
3085}
3086
3087static int cm_sidr_rep_handler(struct cm_work *work)
3088{
3089	struct cm_sidr_rep_msg *sidr_rep_msg;
3090	struct cm_id_private *cm_id_priv;
3091
3092	sidr_rep_msg = (struct cm_sidr_rep_msg *)
3093				work->mad_recv_wc->recv_buf.mad;
3094	cm_id_priv = cm_acquire_id(sidr_rep_msg->request_id, 0);
3095	if (!cm_id_priv)
3096		return -EINVAL; /* Unmatched reply. */
3097
3098	spin_lock_irq(&cm_id_priv->lock);
3099	if (cm_id_priv->id.state != IB_CM_SIDR_REQ_SENT) {
3100		spin_unlock_irq(&cm_id_priv->lock);
3101		goto out;
3102	}
3103	cm_id_priv->id.state = IB_CM_IDLE;
3104	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
3105	spin_unlock_irq(&cm_id_priv->lock);
3106
3107	cm_format_sidr_rep_event(work);
3108	cm_process_work(cm_id_priv, work);
3109	return 0;
3110out:
3111	cm_deref_id(cm_id_priv);
3112	return -EINVAL;
3113}
3114
3115static void cm_process_send_error(struct ib_mad_send_buf *msg,
3116				  enum ib_wc_status wc_status)
3117{
3118	struct cm_id_private *cm_id_priv;
3119	struct ib_cm_event cm_event;
3120	enum ib_cm_state state;
3121	int ret;
3122
3123	memset(&cm_event, 0, sizeof cm_event);
3124	cm_id_priv = msg->context[0];
3125
3126	/* Discard old sends or ones without a response. */
3127	spin_lock_irq(&cm_id_priv->lock);
3128	state = (enum ib_cm_state) (unsigned long) msg->context[1];
3129	if (msg != cm_id_priv->msg || state != cm_id_priv->id.state)
3130		goto discard;
3131
3132	switch (state) {
3133	case IB_CM_REQ_SENT:
3134	case IB_CM_MRA_REQ_RCVD:
3135		cm_reset_to_idle(cm_id_priv);
3136		cm_event.event = IB_CM_REQ_ERROR;
3137		break;
3138	case IB_CM_REP_SENT:
3139	case IB_CM_MRA_REP_RCVD:
3140		cm_reset_to_idle(cm_id_priv);
3141		cm_event.event = IB_CM_REP_ERROR;
3142		break;
3143	case IB_CM_DREQ_SENT:
3144		cm_enter_timewait(cm_id_priv);
3145		cm_event.event = IB_CM_DREQ_ERROR;
3146		break;
3147	case IB_CM_SIDR_REQ_SENT:
3148		cm_id_priv->id.state = IB_CM_IDLE;
3149		cm_event.event = IB_CM_SIDR_REQ_ERROR;
3150		break;
3151	default:
3152		goto discard;
3153	}
3154	spin_unlock_irq(&cm_id_priv->lock);
3155	cm_event.param.send_status = wc_status;
3156
3157	/* No other events can occur on the cm_id at this point. */
3158	ret = cm_id_priv->id.cm_handler(&cm_id_priv->id, &cm_event);
3159	cm_free_msg(msg);
3160	if (ret)
3161		ib_destroy_cm_id(&cm_id_priv->id);
3162	return;
3163discard:
3164	spin_unlock_irq(&cm_id_priv->lock);
3165	cm_free_msg(msg);
3166}
3167
3168static void cm_send_handler(struct ib_mad_agent *mad_agent,
3169			    struct ib_mad_send_wc *mad_send_wc)
3170{
3171	struct ib_mad_send_buf *msg = mad_send_wc->send_buf;
3172	struct cm_port *port;
3173	u16 attr_index;
3174
3175	port = mad_agent->context;
3176	attr_index = be16_to_cpu(((struct ib_mad_hdr *)
3177				  msg->mad)->attr_id) - CM_ATTR_ID_OFFSET;
3178
3179	/*
3180	 * If the send was in response to a received message (context[0] is not
3181	 * set to a cm_id), and is not a REJ, then it is a send that was
3182	 * manually retried.
3183	 */
3184	if (!msg->context[0] && (attr_index != CM_REJ_COUNTER))
3185		msg->retries = 1;
3186
3187	atomic_long_add(1 + msg->retries,
3188			&port->counter_group[CM_XMIT].counter[attr_index]);
3189	if (msg->retries)
3190		atomic_long_add(msg->retries,
3191				&port->counter_group[CM_XMIT_RETRIES].
3192				counter[attr_index]);
3193
3194	switch (mad_send_wc->status) {
3195	case IB_WC_SUCCESS:
3196	case IB_WC_WR_FLUSH_ERR:
3197		cm_free_msg(msg);
3198		break;
3199	default:
3200		if (msg->context[0] && msg->context[1])
3201			cm_process_send_error(msg, mad_send_wc->status);
3202		else
3203			cm_free_msg(msg);
3204		break;
3205	}
3206}
3207
3208static void cm_work_handler(struct work_struct *_work)
3209{
3210	struct cm_work *work = container_of(_work, struct cm_work, work.work);
3211	int ret;
3212
3213	switch (work->cm_event.event) {
3214	case IB_CM_REQ_RECEIVED:
3215		ret = cm_req_handler(work);
3216		break;
3217	case IB_CM_MRA_RECEIVED:
3218		ret = cm_mra_handler(work);
3219		break;
3220	case IB_CM_REJ_RECEIVED:
3221		ret = cm_rej_handler(work);
3222		break;
3223	case IB_CM_REP_RECEIVED:
3224		ret = cm_rep_handler(work);
3225		break;
3226	case IB_CM_RTU_RECEIVED:
3227		ret = cm_rtu_handler(work);
3228		break;
3229	case IB_CM_USER_ESTABLISHED:
3230		ret = cm_establish_handler(work);
3231		break;
3232	case IB_CM_DREQ_RECEIVED:
3233		ret = cm_dreq_handler(work);
3234		break;
3235	case IB_CM_DREP_RECEIVED:
3236		ret = cm_drep_handler(work);
3237		break;
3238	case IB_CM_SIDR_REQ_RECEIVED:
3239		ret = cm_sidr_req_handler(work);
3240		break;
3241	case IB_CM_SIDR_REP_RECEIVED:
3242		ret = cm_sidr_rep_handler(work);
3243		break;
3244	case IB_CM_LAP_RECEIVED:
3245		ret = cm_lap_handler(work);
3246		break;
3247	case IB_CM_APR_RECEIVED:
3248		ret = cm_apr_handler(work);
3249		break;
3250	case IB_CM_TIMEWAIT_EXIT:
3251		ret = cm_timewait_handler(work);
3252		break;
3253	default:
3254		ret = -EINVAL;
3255		break;
3256	}
3257	if (ret)
3258		cm_free_work(work);
3259}
3260
3261static int cm_establish(struct ib_cm_id *cm_id)
3262{
3263	struct cm_id_private *cm_id_priv;
3264	struct cm_work *work;
3265	unsigned long flags;
3266	int ret = 0;
3267
3268	work = kmalloc(sizeof *work, GFP_ATOMIC);
3269	if (!work)
3270		return -ENOMEM;
3271
3272	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3273	spin_lock_irqsave(&cm_id_priv->lock, flags);
3274	switch (cm_id->state)
3275	{
3276	case IB_CM_REP_SENT:
3277	case IB_CM_MRA_REP_RCVD:
3278		cm_id->state = IB_CM_ESTABLISHED;
3279		break;
3280	case IB_CM_ESTABLISHED:
3281		ret = -EISCONN;
3282		break;
3283	default:
3284		ret = -EINVAL;
3285		break;
3286	}
3287	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3288
3289	if (ret) {
3290		kfree(work);
3291		goto out;
3292	}
3293
3294	/*
3295	 * The CM worker thread may try to destroy the cm_id before it
3296	 * can execute this work item.  To prevent potential deadlock,
3297	 * we need to find the cm_id once we're in the context of the
3298	 * worker thread, rather than holding a reference on it.
3299	 */
3300	INIT_DELAYED_WORK(&work->work, cm_work_handler);
3301	work->local_id = cm_id->local_id;
3302	work->remote_id = cm_id->remote_id;
3303	work->mad_recv_wc = NULL;
3304	work->cm_event.event = IB_CM_USER_ESTABLISHED;
3305	queue_delayed_work(cm.wq, &work->work, 0);
3306out:
3307	return ret;
3308}
3309
3310static int cm_migrate(struct ib_cm_id *cm_id)
3311{
3312	struct cm_id_private *cm_id_priv;
3313	unsigned long flags;
3314	int ret = 0;
3315
3316	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3317	spin_lock_irqsave(&cm_id_priv->lock, flags);
3318	if (cm_id->state == IB_CM_ESTABLISHED &&
3319	    (cm_id->lap_state == IB_CM_LAP_UNINIT ||
3320	     cm_id->lap_state == IB_CM_LAP_IDLE)) {
3321		cm_id->lap_state = IB_CM_LAP_IDLE;
3322		cm_id_priv->av = cm_id_priv->alt_av;
3323	} else
3324		ret = -EINVAL;
3325	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3326
3327	return ret;
3328}
3329
3330int ib_cm_notify(struct ib_cm_id *cm_id, enum ib_event_type event)
3331{
3332	int ret;
3333
3334	switch (event) {
3335	case IB_EVENT_COMM_EST:
3336		ret = cm_establish(cm_id);
3337		break;
3338	case IB_EVENT_PATH_MIG:
3339		ret = cm_migrate(cm_id);
3340		break;
3341	default:
3342		ret = -EINVAL;
3343	}
3344	return ret;
3345}
3346EXPORT_SYMBOL(ib_cm_notify);
3347
3348static void cm_recv_handler(struct ib_mad_agent *mad_agent,
3349			    struct ib_mad_recv_wc *mad_recv_wc)
3350{
3351	struct cm_port *port = mad_agent->context;
3352	struct cm_work *work;
3353	enum ib_cm_event_type event;
3354	u16 attr_id;
3355	int paths = 0;
3356
3357	switch (mad_recv_wc->recv_buf.mad->mad_hdr.attr_id) {
3358	case CM_REQ_ATTR_ID:
3359		paths = 1 + (((struct cm_req_msg *) mad_recv_wc->recv_buf.mad)->
3360						    alt_local_lid != 0);
3361		event = IB_CM_REQ_RECEIVED;
3362		break;
3363	case CM_MRA_ATTR_ID:
3364		event = IB_CM_MRA_RECEIVED;
3365		break;
3366	case CM_REJ_ATTR_ID:
3367		event = IB_CM_REJ_RECEIVED;
3368		break;
3369	case CM_REP_ATTR_ID:
3370		event = IB_CM_REP_RECEIVED;
3371		break;
3372	case CM_RTU_ATTR_ID:
3373		event = IB_CM_RTU_RECEIVED;
3374		break;
3375	case CM_DREQ_ATTR_ID:
3376		event = IB_CM_DREQ_RECEIVED;
3377		break;
3378	case CM_DREP_ATTR_ID:
3379		event = IB_CM_DREP_RECEIVED;
3380		break;
3381	case CM_SIDR_REQ_ATTR_ID:
3382		event = IB_CM_SIDR_REQ_RECEIVED;
3383		break;
3384	case CM_SIDR_REP_ATTR_ID:
3385		event = IB_CM_SIDR_REP_RECEIVED;
3386		break;
3387	case CM_LAP_ATTR_ID:
3388		paths = 1;
3389		event = IB_CM_LAP_RECEIVED;
3390		break;
3391	case CM_APR_ATTR_ID:
3392		event = IB_CM_APR_RECEIVED;
3393		break;
3394	default:
3395		ib_free_recv_mad(mad_recv_wc);
3396		return;
3397	}
3398
3399	attr_id = be16_to_cpu(mad_recv_wc->recv_buf.mad->mad_hdr.attr_id);
3400	atomic_long_inc(&port->counter_group[CM_RECV].
3401			counter[attr_id - CM_ATTR_ID_OFFSET]);
3402
3403	work = kmalloc(sizeof *work + sizeof(struct ib_sa_path_rec) * paths,
3404		       GFP_KERNEL);
3405	if (!work) {
3406		ib_free_recv_mad(mad_recv_wc);
3407		return;
3408	}
3409
3410	INIT_DELAYED_WORK(&work->work, cm_work_handler);
3411	work->cm_event.event = event;
3412	work->mad_recv_wc = mad_recv_wc;
3413	work->port = port;
3414	queue_delayed_work(cm.wq, &work->work, 0);
3415}
3416
3417static int cm_init_qp_init_attr(struct cm_id_private *cm_id_priv,
3418				struct ib_qp_attr *qp_attr,
3419				int *qp_attr_mask)
3420{
3421	unsigned long flags;
3422	int ret;
3423
3424	spin_lock_irqsave(&cm_id_priv->lock, flags);
3425	switch (cm_id_priv->id.state) {
3426	case IB_CM_REQ_SENT:
3427	case IB_CM_MRA_REQ_RCVD:
3428	case IB_CM_REQ_RCVD:
3429	case IB_CM_MRA_REQ_SENT:
3430	case IB_CM_REP_RCVD:
3431	case IB_CM_MRA_REP_SENT:
3432	case IB_CM_REP_SENT:
3433	case IB_CM_MRA_REP_RCVD:
3434	case IB_CM_ESTABLISHED:
3435		*qp_attr_mask = IB_QP_STATE | IB_QP_ACCESS_FLAGS |
3436				IB_QP_PKEY_INDEX | IB_QP_PORT;
3437		qp_attr->qp_access_flags = IB_ACCESS_REMOTE_WRITE;
3438		if (cm_id_priv->responder_resources)
3439			qp_attr->qp_access_flags |= IB_ACCESS_REMOTE_READ |
3440						    IB_ACCESS_REMOTE_ATOMIC;
3441		qp_attr->pkey_index = cm_id_priv->av.pkey_index;
3442		qp_attr->port_num = cm_id_priv->av.port->port_num;
3443		ret = 0;
3444		break;
3445	default:
3446		ret = -EINVAL;
3447		break;
3448	}
3449	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3450	return ret;
3451}
3452
3453static int cm_init_qp_rtr_attr(struct cm_id_private *cm_id_priv,
3454			       struct ib_qp_attr *qp_attr,
3455			       int *qp_attr_mask)
3456{
3457	unsigned long flags;
3458	int ret;
3459
3460	spin_lock_irqsave(&cm_id_priv->lock, flags);
3461	switch (cm_id_priv->id.state) {
3462	case IB_CM_REQ_RCVD:
3463	case IB_CM_MRA_REQ_SENT:
3464	case IB_CM_REP_RCVD:
3465	case IB_CM_MRA_REP_SENT:
3466	case IB_CM_REP_SENT:
3467	case IB_CM_MRA_REP_RCVD:
3468	case IB_CM_ESTABLISHED:
3469		*qp_attr_mask = IB_QP_STATE | IB_QP_AV | IB_QP_PATH_MTU |
3470				IB_QP_DEST_QPN | IB_QP_RQ_PSN;
3471		qp_attr->ah_attr = cm_id_priv->av.ah_attr;
3472		qp_attr->path_mtu = cm_id_priv->path_mtu;
3473		qp_attr->dest_qp_num = be32_to_cpu(cm_id_priv->remote_qpn);
3474		qp_attr->rq_psn = be32_to_cpu(cm_id_priv->rq_psn);
3475		if (cm_id_priv->qp_type == IB_QPT_RC) {
3476			*qp_attr_mask |= IB_QP_MAX_DEST_RD_ATOMIC |
3477					 IB_QP_MIN_RNR_TIMER;
3478			qp_attr->max_dest_rd_atomic =
3479					cm_id_priv->responder_resources;
3480			qp_attr->min_rnr_timer = 0;
3481		}
3482		if (cm_id_priv->alt_av.ah_attr.dlid) {
3483			*qp_attr_mask |= IB_QP_ALT_PATH;
3484			qp_attr->alt_port_num = cm_id_priv->alt_av.port->port_num;
3485			qp_attr->alt_pkey_index = cm_id_priv->alt_av.pkey_index;
3486			qp_attr->alt_timeout = cm_id_priv->alt_av.timeout;
3487			qp_attr->alt_ah_attr = cm_id_priv->alt_av.ah_attr;
3488		}
3489		ret = 0;
3490		break;
3491	default:
3492		ret = -EINVAL;
3493		break;
3494	}
3495	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3496	return ret;
3497}
3498
3499static int cm_init_qp_rts_attr(struct cm_id_private *cm_id_priv,
3500			       struct ib_qp_attr *qp_attr,
3501			       int *qp_attr_mask)
3502{
3503	unsigned long flags;
3504	int ret;
3505
3506	spin_lock_irqsave(&cm_id_priv->lock, flags);
3507	switch (cm_id_priv->id.state) {
3508	/* Allow transition to RTS before sending REP */
3509	case IB_CM_REQ_RCVD:
3510	case IB_CM_MRA_REQ_SENT:
3511
3512	case IB_CM_REP_RCVD:
3513	case IB_CM_MRA_REP_SENT:
3514	case IB_CM_REP_SENT:
3515	case IB_CM_MRA_REP_RCVD:
3516	case IB_CM_ESTABLISHED:
3517		if (cm_id_priv->id.lap_state == IB_CM_LAP_UNINIT) {
3518			*qp_attr_mask = IB_QP_STATE | IB_QP_SQ_PSN;
3519			qp_attr->sq_psn = be32_to_cpu(cm_id_priv->sq_psn);
3520			if (cm_id_priv->qp_type == IB_QPT_RC) {
3521				*qp_attr_mask |= IB_QP_TIMEOUT | IB_QP_RETRY_CNT |
3522						 IB_QP_RNR_RETRY |
3523						 IB_QP_MAX_QP_RD_ATOMIC;
3524				qp_attr->timeout = cm_id_priv->av.timeout;
3525				qp_attr->retry_cnt = cm_id_priv->retry_count;
3526				qp_attr->rnr_retry = cm_id_priv->rnr_retry_count;
3527				qp_attr->max_rd_atomic =
3528					cm_id_priv->initiator_depth;
3529			}
3530			if (cm_id_priv->alt_av.ah_attr.dlid) {
3531				*qp_attr_mask |= IB_QP_PATH_MIG_STATE;
3532				qp_attr->path_mig_state = IB_MIG_REARM;
3533			}
3534		} else {
3535			*qp_attr_mask = IB_QP_ALT_PATH | IB_QP_PATH_MIG_STATE;
3536			qp_attr->alt_port_num = cm_id_priv->alt_av.port->port_num;
3537			qp_attr->alt_pkey_index = cm_id_priv->alt_av.pkey_index;
3538			qp_attr->alt_timeout = cm_id_priv->alt_av.timeout;
3539			qp_attr->alt_ah_attr = cm_id_priv->alt_av.ah_attr;
3540			qp_attr->path_mig_state = IB_MIG_REARM;
3541		}
3542		ret = 0;
3543		break;
3544	default:
3545		ret = -EINVAL;
3546		break;
3547	}
3548	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3549	return ret;
3550}
3551
3552int ib_cm_init_qp_attr(struct ib_cm_id *cm_id,
3553		       struct ib_qp_attr *qp_attr,
3554		       int *qp_attr_mask)
3555{
3556	struct cm_id_private *cm_id_priv;
3557	int ret;
3558
3559	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3560	switch (qp_attr->qp_state) {
3561	case IB_QPS_INIT:
3562		ret = cm_init_qp_init_attr(cm_id_priv, qp_attr, qp_attr_mask);
3563		break;
3564	case IB_QPS_RTR:
3565		ret = cm_init_qp_rtr_attr(cm_id_priv, qp_attr, qp_attr_mask);
3566		break;
3567	case IB_QPS_RTS:
3568		ret = cm_init_qp_rts_attr(cm_id_priv, qp_attr, qp_attr_mask);
3569		break;
3570	default:
3571		ret = -EINVAL;
3572		break;
3573	}
3574	return ret;
3575}
3576EXPORT_SYMBOL(ib_cm_init_qp_attr);
3577
3578static void cm_get_ack_delay(struct cm_device *cm_dev)
3579{
3580	struct ib_device_attr attr;
3581
3582	if (ib_query_device(cm_dev->ib_device, &attr))
3583		cm_dev->ack_delay = 0; /* acks will rely on packet life time */
3584	else
3585		cm_dev->ack_delay = attr.local_ca_ack_delay;
3586}
3587
3588static ssize_t cm_show_counter(struct kobject *obj, struct attribute *attr,
3589			       char *buf)
3590{
3591	struct cm_counter_group *group;
3592	struct cm_counter_attribute *cm_attr;
3593
3594	group = container_of(obj, struct cm_counter_group, obj);
3595	cm_attr = container_of(attr, struct cm_counter_attribute, attr);
3596
3597	return sprintf(buf, "%ld\n",
3598		       atomic_long_read(&group->counter[cm_attr->index]));
3599}
3600
3601static const struct sysfs_ops cm_counter_ops = {
3602	.show = cm_show_counter
3603};
3604
3605static struct kobj_type cm_counter_obj_type = {
3606	.sysfs_ops = &cm_counter_ops,
3607	.default_attrs = cm_counter_default_attrs
3608};
3609
3610static void cm_release_port_obj(struct kobject *obj)
3611{
3612	struct cm_port *cm_port;
3613
3614	cm_port = container_of(obj, struct cm_port, port_obj);
3615	kfree(cm_port);
3616}
3617
3618static struct kobj_type cm_port_obj_type = {
3619	.release = cm_release_port_obj
3620};
3621
3622struct class cm_class = {
3623	.name    = "infiniband_cm",
3624};
3625EXPORT_SYMBOL(cm_class);
3626
3627static int cm_create_port_fs(struct cm_port *port)
3628{
3629	int i, ret;
3630
3631	ret = kobject_init_and_add(&port->port_obj, &cm_port_obj_type,
3632				   &port->cm_dev->device->kobj,
3633				   "%d", port->port_num);
3634	if (ret) {
3635		kfree(port);
3636		return ret;
3637	}
3638
3639	for (i = 0; i < CM_COUNTER_GROUPS; i++) {
3640		ret = kobject_init_and_add(&port->counter_group[i].obj,
3641					   &cm_counter_obj_type,
3642					   &port->port_obj,
3643					   "%s", counter_group_names[i]);
3644		if (ret)
3645			goto error;
3646	}
3647
3648	return 0;
3649
3650error:
3651	while (i--)
3652		kobject_put(&port->counter_group[i].obj);
3653	kobject_put(&port->port_obj);
3654	return ret;
3655
3656}
3657
3658static void cm_remove_port_fs(struct cm_port *port)
3659{
3660	int i;
3661
3662	for (i = 0; i < CM_COUNTER_GROUPS; i++)
3663		kobject_put(&port->counter_group[i].obj);
3664
3665	kobject_put(&port->port_obj);
3666}
3667
3668static void cm_add_one(struct ib_device *ib_device)
3669{
3670	struct cm_device *cm_dev;
3671	struct cm_port *port;
3672	struct ib_mad_reg_req reg_req = {
3673		.mgmt_class = IB_MGMT_CLASS_CM,
3674		.mgmt_class_version = IB_CM_CLASS_VERSION
3675	};
3676	struct ib_port_modify port_modify = {
3677		.set_port_cap_mask = IB_PORT_CM_SUP
3678	};
3679	unsigned long flags;
3680	int ret;
3681	u8 i;
3682
3683	if (rdma_node_get_transport(ib_device->node_type) != RDMA_TRANSPORT_IB)
3684		return;
3685
3686	cm_dev = kzalloc(sizeof(*cm_dev) + sizeof(*port) *
3687			 ib_device->phys_port_cnt, GFP_KERNEL);
3688	if (!cm_dev)
3689		return;
3690
3691	cm_dev->ib_device = ib_device;
3692	cm_get_ack_delay(cm_dev);
3693
3694	cm_dev->device = device_create(&cm_class, &ib_device->dev,
3695				       MKDEV(0, 0), NULL,
3696				       "%s", ib_device->name);
3697	if (!cm_dev->device) {
3698		kfree(cm_dev);
3699		return;
3700	}
3701
3702	set_bit(IB_MGMT_METHOD_SEND, reg_req.method_mask);
3703	for (i = 1; i <= ib_device->phys_port_cnt; i++) {
3704		port = kzalloc(sizeof *port, GFP_KERNEL);
3705		if (!port)
3706			goto error1;
3707
3708		cm_dev->port[i-1] = port;
3709		port->cm_dev = cm_dev;
3710		port->port_num = i;
3711
3712		ret = cm_create_port_fs(port);
3713		if (ret)
3714			goto error1;
3715
3716		port->mad_agent = ib_register_mad_agent(ib_device, i,
3717							IB_QPT_GSI,
3718							&reg_req,
3719							0,
3720							cm_send_handler,
3721							cm_recv_handler,
3722							port);
3723		if (IS_ERR(port->mad_agent))
3724			goto error2;
3725
3726		ret = ib_modify_port(ib_device, i, 0, &port_modify);
3727		if (ret)
3728			goto error3;
3729	}
3730	ib_set_client_data(ib_device, &cm_client, cm_dev);
3731
3732	write_lock_irqsave(&cm.device_lock, flags);
3733	list_add_tail(&cm_dev->list, &cm.device_list);
3734	write_unlock_irqrestore(&cm.device_lock, flags);
3735	return;
3736
3737error3:
3738	ib_unregister_mad_agent(port->mad_agent);
3739error2:
3740	cm_remove_port_fs(port);
3741error1:
3742	port_modify.set_port_cap_mask = 0;
3743	port_modify.clr_port_cap_mask = IB_PORT_CM_SUP;
3744	while (--i) {
3745		port = cm_dev->port[i-1];
3746		ib_modify_port(ib_device, port->port_num, 0, &port_modify);
3747		ib_unregister_mad_agent(port->mad_agent);
3748		cm_remove_port_fs(port);
3749	}
3750	device_unregister(cm_dev->device);
3751	kfree(cm_dev);
3752}
3753
3754static void cm_remove_one(struct ib_device *ib_device)
3755{
3756	struct cm_device *cm_dev;
3757	struct cm_port *port;
3758	struct ib_port_modify port_modify = {
3759		.clr_port_cap_mask = IB_PORT_CM_SUP
3760	};
3761	unsigned long flags;
3762	int i;
3763
3764	cm_dev = ib_get_client_data(ib_device, &cm_client);
3765	if (!cm_dev)
3766		return;
3767
3768	write_lock_irqsave(&cm.device_lock, flags);
3769	list_del(&cm_dev->list);
3770	write_unlock_irqrestore(&cm.device_lock, flags);
3771
3772	for (i = 1; i <= ib_device->phys_port_cnt; i++) {
3773		port = cm_dev->port[i-1];
3774		ib_modify_port(ib_device, port->port_num, 0, &port_modify);
3775		ib_unregister_mad_agent(port->mad_agent);
3776		flush_workqueue(cm.wq);
3777		cm_remove_port_fs(port);
3778	}
3779	device_unregister(cm_dev->device);
3780	kfree(cm_dev);
3781}
3782
3783static int __init ib_cm_init(void)
3784{
3785	int ret;
3786
3787	memset(&cm, 0, sizeof cm);
3788	INIT_LIST_HEAD(&cm.device_list);
3789	rwlock_init(&cm.device_lock);
3790	spin_lock_init(&cm.lock);
3791	cm.listen_service_table = RB_ROOT;
3792	cm.listen_service_id = be64_to_cpu(IB_CM_ASSIGN_SERVICE_ID);
3793	cm.remote_id_table = RB_ROOT;
3794	cm.remote_qp_table = RB_ROOT;
3795	cm.remote_sidr_table = RB_ROOT;
3796	idr_init(&cm.local_id_table);
3797	get_random_bytes(&cm.random_id_operand, sizeof cm.random_id_operand);
3798	idr_pre_get(&cm.local_id_table, GFP_KERNEL);
3799	INIT_LIST_HEAD(&cm.timewait_list);
3800
3801	ret = class_register(&cm_class);
3802	if (ret)
3803		return -ENOMEM;
3804
3805	cm.wq = create_workqueue("ib_cm");
3806	if (!cm.wq) {
3807		ret = -ENOMEM;
3808		goto error1;
3809	}
3810
3811	ret = ib_register_client(&cm_client);
3812	if (ret)
3813		goto error2;
3814
3815	return 0;
3816error2:
3817	destroy_workqueue(cm.wq);
3818error1:
3819	class_unregister(&cm_class);
3820	return ret;
3821}
3822
3823static void __exit ib_cm_cleanup(void)
3824{
3825	struct cm_timewait_info *timewait_info, *tmp;
3826
3827	spin_lock_irq(&cm.lock);
3828	list_for_each_entry(timewait_info, &cm.timewait_list, list)
3829		cancel_delayed_work(&timewait_info->work.work);
3830	spin_unlock_irq(&cm.lock);
3831
3832	ib_unregister_client(&cm_client);
3833	destroy_workqueue(cm.wq);
3834
3835	list_for_each_entry_safe(timewait_info, tmp, &cm.timewait_list, list) {
3836		list_del(&timewait_info->list);
3837		kfree(timewait_info);
3838	}
3839
3840	class_unregister(&cm_class);
3841	idr_destroy(&cm.local_id_table);
3842}
3843
3844module_init(ib_cm_init);
3845module_exit(ib_cm_cleanup);
3846
3847