• Home
  • History
  • Annotate
  • Line#
  • Navigate
  • Raw
  • Download
  • only in /asuswrt-rt-n18u-9.0.0.4.380.2695/release/src-rt-6.x.4708/linux/linux-2.6/drivers/infiniband/core/
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		req_msg->primary_local_lid = IB_LID_PERMISSIVE;
1032		req_msg->primary_remote_lid = IB_LID_PERMISSIVE;
1033	}
1034	req_msg->primary_local_gid = pri_path->sgid;
1035	req_msg->primary_remote_gid = pri_path->dgid;
1036	cm_req_set_primary_flow_label(req_msg, pri_path->flow_label);
1037	cm_req_set_primary_packet_rate(req_msg, pri_path->rate);
1038	req_msg->primary_traffic_class = pri_path->traffic_class;
1039	req_msg->primary_hop_limit = pri_path->hop_limit;
1040	cm_req_set_primary_sl(req_msg, pri_path->sl);
1041	cm_req_set_primary_subnet_local(req_msg, (pri_path->hop_limit <= 1));
1042	cm_req_set_primary_local_ack_timeout(req_msg,
1043		cm_ack_timeout(cm_id_priv->av.port->cm_dev->ack_delay,
1044			       pri_path->packet_life_time));
1045
1046	if (alt_path) {
1047		if (alt_path->hop_limit <= 1) {
1048			req_msg->alt_local_lid = alt_path->slid;
1049			req_msg->alt_remote_lid = alt_path->dlid;
1050		} else {
1051			req_msg->alt_local_lid = IB_LID_PERMISSIVE;
1052			req_msg->alt_remote_lid = IB_LID_PERMISSIVE;
1053		}
1054		req_msg->alt_local_gid = alt_path->sgid;
1055		req_msg->alt_remote_gid = alt_path->dgid;
1056		cm_req_set_alt_flow_label(req_msg,
1057					  alt_path->flow_label);
1058		cm_req_set_alt_packet_rate(req_msg, alt_path->rate);
1059		req_msg->alt_traffic_class = alt_path->traffic_class;
1060		req_msg->alt_hop_limit = alt_path->hop_limit;
1061		cm_req_set_alt_sl(req_msg, alt_path->sl);
1062		cm_req_set_alt_subnet_local(req_msg, (alt_path->hop_limit <= 1));
1063		cm_req_set_alt_local_ack_timeout(req_msg,
1064			cm_ack_timeout(cm_id_priv->av.port->cm_dev->ack_delay,
1065				       alt_path->packet_life_time));
1066	}
1067
1068	if (param->private_data && param->private_data_len)
1069		memcpy(req_msg->private_data, param->private_data,
1070		       param->private_data_len);
1071}
1072
1073static int cm_validate_req_param(struct ib_cm_req_param *param)
1074{
1075	/* peer-to-peer not supported */
1076	if (param->peer_to_peer)
1077		return -EINVAL;
1078
1079	if (!param->primary_path)
1080		return -EINVAL;
1081
1082	if (param->qp_type != IB_QPT_RC && param->qp_type != IB_QPT_UC)
1083		return -EINVAL;
1084
1085	if (param->private_data &&
1086	    param->private_data_len > IB_CM_REQ_PRIVATE_DATA_SIZE)
1087		return -EINVAL;
1088
1089	if (param->alternate_path &&
1090	    (param->alternate_path->pkey != param->primary_path->pkey ||
1091	     param->alternate_path->mtu != param->primary_path->mtu))
1092		return -EINVAL;
1093
1094	return 0;
1095}
1096
1097int ib_send_cm_req(struct ib_cm_id *cm_id,
1098		   struct ib_cm_req_param *param)
1099{
1100	struct cm_id_private *cm_id_priv;
1101	struct cm_req_msg *req_msg;
1102	unsigned long flags;
1103	int ret;
1104
1105	ret = cm_validate_req_param(param);
1106	if (ret)
1107		return ret;
1108
1109	/* Verify that we're not in timewait. */
1110	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1111	spin_lock_irqsave(&cm_id_priv->lock, flags);
1112	if (cm_id->state != IB_CM_IDLE) {
1113		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1114		ret = -EINVAL;
1115		goto out;
1116	}
1117	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1118
1119	cm_id_priv->timewait_info = cm_create_timewait_info(cm_id_priv->
1120							    id.local_id);
1121	if (IS_ERR(cm_id_priv->timewait_info)) {
1122		ret = PTR_ERR(cm_id_priv->timewait_info);
1123		goto out;
1124	}
1125
1126	ret = cm_init_av_by_path(param->primary_path, &cm_id_priv->av);
1127	if (ret)
1128		goto error1;
1129	if (param->alternate_path) {
1130		ret = cm_init_av_by_path(param->alternate_path,
1131					 &cm_id_priv->alt_av);
1132		if (ret)
1133			goto error1;
1134	}
1135	cm_id->service_id = param->service_id;
1136	cm_id->service_mask = ~cpu_to_be64(0);
1137	cm_id_priv->timeout_ms = cm_convert_to_ms(
1138				    param->primary_path->packet_life_time) * 2 +
1139				 cm_convert_to_ms(
1140				    param->remote_cm_response_timeout);
1141	cm_id_priv->max_cm_retries = param->max_cm_retries;
1142	cm_id_priv->initiator_depth = param->initiator_depth;
1143	cm_id_priv->responder_resources = param->responder_resources;
1144	cm_id_priv->retry_count = param->retry_count;
1145	cm_id_priv->path_mtu = param->primary_path->mtu;
1146	cm_id_priv->pkey = param->primary_path->pkey;
1147	cm_id_priv->qp_type = param->qp_type;
1148
1149	ret = cm_alloc_msg(cm_id_priv, &cm_id_priv->msg);
1150	if (ret)
1151		goto error1;
1152
1153	req_msg = (struct cm_req_msg *) cm_id_priv->msg->mad;
1154	cm_format_req(req_msg, cm_id_priv, param);
1155	cm_id_priv->tid = req_msg->hdr.tid;
1156	cm_id_priv->msg->timeout_ms = cm_id_priv->timeout_ms;
1157	cm_id_priv->msg->context[1] = (void *) (unsigned long) IB_CM_REQ_SENT;
1158
1159	cm_id_priv->local_qpn = cm_req_get_local_qpn(req_msg);
1160	cm_id_priv->rq_psn = cm_req_get_starting_psn(req_msg);
1161
1162	spin_lock_irqsave(&cm_id_priv->lock, flags);
1163	ret = ib_post_send_mad(cm_id_priv->msg, NULL);
1164	if (ret) {
1165		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1166		goto error2;
1167	}
1168	BUG_ON(cm_id->state != IB_CM_IDLE);
1169	cm_id->state = IB_CM_REQ_SENT;
1170	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1171	return 0;
1172
1173error2:	cm_free_msg(cm_id_priv->msg);
1174error1:	kfree(cm_id_priv->timewait_info);
1175out:	return ret;
1176}
1177EXPORT_SYMBOL(ib_send_cm_req);
1178
1179static int cm_issue_rej(struct cm_port *port,
1180			struct ib_mad_recv_wc *mad_recv_wc,
1181			enum ib_cm_rej_reason reason,
1182			enum cm_msg_response msg_rejected,
1183			void *ari, u8 ari_length)
1184{
1185	struct ib_mad_send_buf *msg = NULL;
1186	struct cm_rej_msg *rej_msg, *rcv_msg;
1187	int ret;
1188
1189	ret = cm_alloc_response_msg(port, mad_recv_wc, &msg);
1190	if (ret)
1191		return ret;
1192
1193	/* We just need common CM header information.  Cast to any message. */
1194	rcv_msg = (struct cm_rej_msg *) mad_recv_wc->recv_buf.mad;
1195	rej_msg = (struct cm_rej_msg *) msg->mad;
1196
1197	cm_format_mad_hdr(&rej_msg->hdr, CM_REJ_ATTR_ID, rcv_msg->hdr.tid);
1198	rej_msg->remote_comm_id = rcv_msg->local_comm_id;
1199	rej_msg->local_comm_id = rcv_msg->remote_comm_id;
1200	cm_rej_set_msg_rejected(rej_msg, msg_rejected);
1201	rej_msg->reason = cpu_to_be16(reason);
1202
1203	if (ari && ari_length) {
1204		cm_rej_set_reject_info_len(rej_msg, ari_length);
1205		memcpy(rej_msg->ari, ari, ari_length);
1206	}
1207
1208	ret = ib_post_send_mad(msg, NULL);
1209	if (ret)
1210		cm_free_msg(msg);
1211
1212	return ret;
1213}
1214
1215static inline int cm_is_active_peer(__be64 local_ca_guid, __be64 remote_ca_guid,
1216				    __be32 local_qpn, __be32 remote_qpn)
1217{
1218	return (be64_to_cpu(local_ca_guid) > be64_to_cpu(remote_ca_guid) ||
1219		((local_ca_guid == remote_ca_guid) &&
1220		 (be32_to_cpu(local_qpn) > be32_to_cpu(remote_qpn))));
1221}
1222
1223static void cm_format_paths_from_req(struct cm_req_msg *req_msg,
1224					    struct ib_sa_path_rec *primary_path,
1225					    struct ib_sa_path_rec *alt_path)
1226{
1227	memset(primary_path, 0, sizeof *primary_path);
1228	primary_path->dgid = req_msg->primary_local_gid;
1229	primary_path->sgid = req_msg->primary_remote_gid;
1230	primary_path->dlid = req_msg->primary_local_lid;
1231	primary_path->slid = req_msg->primary_remote_lid;
1232	primary_path->flow_label = cm_req_get_primary_flow_label(req_msg);
1233	primary_path->hop_limit = req_msg->primary_hop_limit;
1234	primary_path->traffic_class = req_msg->primary_traffic_class;
1235	primary_path->reversible = 1;
1236	primary_path->pkey = req_msg->pkey;
1237	primary_path->sl = cm_req_get_primary_sl(req_msg);
1238	primary_path->mtu_selector = IB_SA_EQ;
1239	primary_path->mtu = cm_req_get_path_mtu(req_msg);
1240	primary_path->rate_selector = IB_SA_EQ;
1241	primary_path->rate = cm_req_get_primary_packet_rate(req_msg);
1242	primary_path->packet_life_time_selector = IB_SA_EQ;
1243	primary_path->packet_life_time =
1244		cm_req_get_primary_local_ack_timeout(req_msg);
1245	primary_path->packet_life_time -= (primary_path->packet_life_time > 0);
1246
1247	if (req_msg->alt_local_lid) {
1248		memset(alt_path, 0, sizeof *alt_path);
1249		alt_path->dgid = req_msg->alt_local_gid;
1250		alt_path->sgid = req_msg->alt_remote_gid;
1251		alt_path->dlid = req_msg->alt_local_lid;
1252		alt_path->slid = req_msg->alt_remote_lid;
1253		alt_path->flow_label = cm_req_get_alt_flow_label(req_msg);
1254		alt_path->hop_limit = req_msg->alt_hop_limit;
1255		alt_path->traffic_class = req_msg->alt_traffic_class;
1256		alt_path->reversible = 1;
1257		alt_path->pkey = req_msg->pkey;
1258		alt_path->sl = cm_req_get_alt_sl(req_msg);
1259		alt_path->mtu_selector = IB_SA_EQ;
1260		alt_path->mtu = cm_req_get_path_mtu(req_msg);
1261		alt_path->rate_selector = IB_SA_EQ;
1262		alt_path->rate = cm_req_get_alt_packet_rate(req_msg);
1263		alt_path->packet_life_time_selector = IB_SA_EQ;
1264		alt_path->packet_life_time =
1265			cm_req_get_alt_local_ack_timeout(req_msg);
1266		alt_path->packet_life_time -= (alt_path->packet_life_time > 0);
1267	}
1268}
1269
1270static void cm_format_req_event(struct cm_work *work,
1271				struct cm_id_private *cm_id_priv,
1272				struct ib_cm_id *listen_id)
1273{
1274	struct cm_req_msg *req_msg;
1275	struct ib_cm_req_event_param *param;
1276
1277	req_msg = (struct cm_req_msg *)work->mad_recv_wc->recv_buf.mad;
1278	param = &work->cm_event.param.req_rcvd;
1279	param->listen_id = listen_id;
1280	param->port = cm_id_priv->av.port->port_num;
1281	param->primary_path = &work->path[0];
1282	if (req_msg->alt_local_lid)
1283		param->alternate_path = &work->path[1];
1284	else
1285		param->alternate_path = NULL;
1286	param->remote_ca_guid = req_msg->local_ca_guid;
1287	param->remote_qkey = be32_to_cpu(req_msg->local_qkey);
1288	param->remote_qpn = be32_to_cpu(cm_req_get_local_qpn(req_msg));
1289	param->qp_type = cm_req_get_qp_type(req_msg);
1290	param->starting_psn = be32_to_cpu(cm_req_get_starting_psn(req_msg));
1291	param->responder_resources = cm_req_get_init_depth(req_msg);
1292	param->initiator_depth = cm_req_get_resp_res(req_msg);
1293	param->local_cm_response_timeout =
1294					cm_req_get_remote_resp_timeout(req_msg);
1295	param->flow_control = cm_req_get_flow_ctrl(req_msg);
1296	param->remote_cm_response_timeout =
1297					cm_req_get_local_resp_timeout(req_msg);
1298	param->retry_count = cm_req_get_retry_count(req_msg);
1299	param->rnr_retry_count = cm_req_get_rnr_retry_count(req_msg);
1300	param->srq = cm_req_get_srq(req_msg);
1301	work->cm_event.private_data = &req_msg->private_data;
1302}
1303
1304static void cm_process_work(struct cm_id_private *cm_id_priv,
1305			    struct cm_work *work)
1306{
1307	int ret;
1308
1309	/* We will typically only have the current event to report. */
1310	ret = cm_id_priv->id.cm_handler(&cm_id_priv->id, &work->cm_event);
1311	cm_free_work(work);
1312
1313	while (!ret && !atomic_add_negative(-1, &cm_id_priv->work_count)) {
1314		spin_lock_irq(&cm_id_priv->lock);
1315		work = cm_dequeue_work(cm_id_priv);
1316		spin_unlock_irq(&cm_id_priv->lock);
1317		BUG_ON(!work);
1318		ret = cm_id_priv->id.cm_handler(&cm_id_priv->id,
1319						&work->cm_event);
1320		cm_free_work(work);
1321	}
1322	cm_deref_id(cm_id_priv);
1323	if (ret)
1324		cm_destroy_id(&cm_id_priv->id, ret);
1325}
1326
1327static void cm_format_mra(struct cm_mra_msg *mra_msg,
1328			  struct cm_id_private *cm_id_priv,
1329			  enum cm_msg_response msg_mraed, u8 service_timeout,
1330			  const void *private_data, u8 private_data_len)
1331{
1332	cm_format_mad_hdr(&mra_msg->hdr, CM_MRA_ATTR_ID, cm_id_priv->tid);
1333	cm_mra_set_msg_mraed(mra_msg, msg_mraed);
1334	mra_msg->local_comm_id = cm_id_priv->id.local_id;
1335	mra_msg->remote_comm_id = cm_id_priv->id.remote_id;
1336	cm_mra_set_service_timeout(mra_msg, service_timeout);
1337
1338	if (private_data && private_data_len)
1339		memcpy(mra_msg->private_data, private_data, private_data_len);
1340}
1341
1342static void cm_format_rej(struct cm_rej_msg *rej_msg,
1343			  struct cm_id_private *cm_id_priv,
1344			  enum ib_cm_rej_reason reason,
1345			  void *ari,
1346			  u8 ari_length,
1347			  const void *private_data,
1348			  u8 private_data_len)
1349{
1350	cm_format_mad_hdr(&rej_msg->hdr, CM_REJ_ATTR_ID, cm_id_priv->tid);
1351	rej_msg->remote_comm_id = cm_id_priv->id.remote_id;
1352
1353	switch(cm_id_priv->id.state) {
1354	case IB_CM_REQ_RCVD:
1355		rej_msg->local_comm_id = 0;
1356		cm_rej_set_msg_rejected(rej_msg, CM_MSG_RESPONSE_REQ);
1357		break;
1358	case IB_CM_MRA_REQ_SENT:
1359		rej_msg->local_comm_id = cm_id_priv->id.local_id;
1360		cm_rej_set_msg_rejected(rej_msg, CM_MSG_RESPONSE_REQ);
1361		break;
1362	case IB_CM_REP_RCVD:
1363	case IB_CM_MRA_REP_SENT:
1364		rej_msg->local_comm_id = cm_id_priv->id.local_id;
1365		cm_rej_set_msg_rejected(rej_msg, CM_MSG_RESPONSE_REP);
1366		break;
1367	default:
1368		rej_msg->local_comm_id = cm_id_priv->id.local_id;
1369		cm_rej_set_msg_rejected(rej_msg, CM_MSG_RESPONSE_OTHER);
1370		break;
1371	}
1372
1373	rej_msg->reason = cpu_to_be16(reason);
1374	if (ari && ari_length) {
1375		cm_rej_set_reject_info_len(rej_msg, ari_length);
1376		memcpy(rej_msg->ari, ari, ari_length);
1377	}
1378
1379	if (private_data && private_data_len)
1380		memcpy(rej_msg->private_data, private_data, private_data_len);
1381}
1382
1383static void cm_dup_req_handler(struct cm_work *work,
1384			       struct cm_id_private *cm_id_priv)
1385{
1386	struct ib_mad_send_buf *msg = NULL;
1387	int ret;
1388
1389	atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
1390			counter[CM_REQ_COUNTER]);
1391
1392	/* Quick state check to discard duplicate REQs. */
1393	if (cm_id_priv->id.state == IB_CM_REQ_RCVD)
1394		return;
1395
1396	ret = cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg);
1397	if (ret)
1398		return;
1399
1400	spin_lock_irq(&cm_id_priv->lock);
1401	switch (cm_id_priv->id.state) {
1402	case IB_CM_MRA_REQ_SENT:
1403		cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
1404			      CM_MSG_RESPONSE_REQ, cm_id_priv->service_timeout,
1405			      cm_id_priv->private_data,
1406			      cm_id_priv->private_data_len);
1407		break;
1408	case IB_CM_TIMEWAIT:
1409		cm_format_rej((struct cm_rej_msg *) msg->mad, cm_id_priv,
1410			      IB_CM_REJ_STALE_CONN, NULL, 0, NULL, 0);
1411		break;
1412	default:
1413		goto unlock;
1414	}
1415	spin_unlock_irq(&cm_id_priv->lock);
1416
1417	ret = ib_post_send_mad(msg, NULL);
1418	if (ret)
1419		goto free;
1420	return;
1421
1422unlock:	spin_unlock_irq(&cm_id_priv->lock);
1423free:	cm_free_msg(msg);
1424}
1425
1426static struct cm_id_private * cm_match_req(struct cm_work *work,
1427					   struct cm_id_private *cm_id_priv)
1428{
1429	struct cm_id_private *listen_cm_id_priv, *cur_cm_id_priv;
1430	struct cm_timewait_info *timewait_info;
1431	struct cm_req_msg *req_msg;
1432
1433	req_msg = (struct cm_req_msg *)work->mad_recv_wc->recv_buf.mad;
1434
1435	/* Check for possible duplicate REQ. */
1436	spin_lock_irq(&cm.lock);
1437	timewait_info = cm_insert_remote_id(cm_id_priv->timewait_info);
1438	if (timewait_info) {
1439		cur_cm_id_priv = cm_get_id(timewait_info->work.local_id,
1440					   timewait_info->work.remote_id);
1441		spin_unlock_irq(&cm.lock);
1442		if (cur_cm_id_priv) {
1443			cm_dup_req_handler(work, cur_cm_id_priv);
1444			cm_deref_id(cur_cm_id_priv);
1445		}
1446		return NULL;
1447	}
1448
1449	/* Check for stale connections. */
1450	timewait_info = cm_insert_remote_qpn(cm_id_priv->timewait_info);
1451	if (timewait_info) {
1452		cm_cleanup_timewait(cm_id_priv->timewait_info);
1453		spin_unlock_irq(&cm.lock);
1454		cm_issue_rej(work->port, work->mad_recv_wc,
1455			     IB_CM_REJ_STALE_CONN, CM_MSG_RESPONSE_REQ,
1456			     NULL, 0);
1457		return NULL;
1458	}
1459
1460	/* Find matching listen request. */
1461	listen_cm_id_priv = cm_find_listen(cm_id_priv->id.device,
1462					   req_msg->service_id,
1463					   req_msg->private_data);
1464	if (!listen_cm_id_priv) {
1465		cm_cleanup_timewait(cm_id_priv->timewait_info);
1466		spin_unlock_irq(&cm.lock);
1467		cm_issue_rej(work->port, work->mad_recv_wc,
1468			     IB_CM_REJ_INVALID_SERVICE_ID, CM_MSG_RESPONSE_REQ,
1469			     NULL, 0);
1470		goto out;
1471	}
1472	atomic_inc(&listen_cm_id_priv->refcount);
1473	atomic_inc(&cm_id_priv->refcount);
1474	cm_id_priv->id.state = IB_CM_REQ_RCVD;
1475	atomic_inc(&cm_id_priv->work_count);
1476	spin_unlock_irq(&cm.lock);
1477out:
1478	return listen_cm_id_priv;
1479}
1480
1481static void cm_process_routed_req(struct cm_req_msg *req_msg, struct ib_wc *wc)
1482{
1483	if (!cm_req_get_primary_subnet_local(req_msg)) {
1484		if (req_msg->primary_local_lid == IB_LID_PERMISSIVE) {
1485			req_msg->primary_local_lid = cpu_to_be16(wc->slid);
1486			cm_req_set_primary_sl(req_msg, wc->sl);
1487		}
1488
1489		if (req_msg->primary_remote_lid == IB_LID_PERMISSIVE)
1490			req_msg->primary_remote_lid = cpu_to_be16(wc->dlid_path_bits);
1491	}
1492
1493	if (!cm_req_get_alt_subnet_local(req_msg)) {
1494		if (req_msg->alt_local_lid == IB_LID_PERMISSIVE) {
1495			req_msg->alt_local_lid = cpu_to_be16(wc->slid);
1496			cm_req_set_alt_sl(req_msg, wc->sl);
1497		}
1498
1499		if (req_msg->alt_remote_lid == IB_LID_PERMISSIVE)
1500			req_msg->alt_remote_lid = cpu_to_be16(wc->dlid_path_bits);
1501	}
1502}
1503
1504static int cm_req_handler(struct cm_work *work)
1505{
1506	struct ib_cm_id *cm_id;
1507	struct cm_id_private *cm_id_priv, *listen_cm_id_priv;
1508	struct cm_req_msg *req_msg;
1509	int ret;
1510
1511	req_msg = (struct cm_req_msg *)work->mad_recv_wc->recv_buf.mad;
1512
1513	cm_id = ib_create_cm_id(work->port->cm_dev->ib_device, NULL, NULL);
1514	if (IS_ERR(cm_id))
1515		return PTR_ERR(cm_id);
1516
1517	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1518	cm_id_priv->id.remote_id = req_msg->local_comm_id;
1519	cm_init_av_for_response(work->port, work->mad_recv_wc->wc,
1520				work->mad_recv_wc->recv_buf.grh,
1521				&cm_id_priv->av);
1522	cm_id_priv->timewait_info = cm_create_timewait_info(cm_id_priv->
1523							    id.local_id);
1524	if (IS_ERR(cm_id_priv->timewait_info)) {
1525		ret = PTR_ERR(cm_id_priv->timewait_info);
1526		goto destroy;
1527	}
1528	cm_id_priv->timewait_info->work.remote_id = req_msg->local_comm_id;
1529	cm_id_priv->timewait_info->remote_ca_guid = req_msg->local_ca_guid;
1530	cm_id_priv->timewait_info->remote_qpn = cm_req_get_local_qpn(req_msg);
1531
1532	listen_cm_id_priv = cm_match_req(work, cm_id_priv);
1533	if (!listen_cm_id_priv) {
1534		ret = -EINVAL;
1535		kfree(cm_id_priv->timewait_info);
1536		goto destroy;
1537	}
1538
1539	cm_id_priv->id.cm_handler = listen_cm_id_priv->id.cm_handler;
1540	cm_id_priv->id.context = listen_cm_id_priv->id.context;
1541	cm_id_priv->id.service_id = req_msg->service_id;
1542	cm_id_priv->id.service_mask = ~cpu_to_be64(0);
1543
1544	cm_process_routed_req(req_msg, work->mad_recv_wc->wc);
1545	cm_format_paths_from_req(req_msg, &work->path[0], &work->path[1]);
1546	ret = cm_init_av_by_path(&work->path[0], &cm_id_priv->av);
1547	if (ret) {
1548		ib_get_cached_gid(work->port->cm_dev->ib_device,
1549				  work->port->port_num, 0, &work->path[0].sgid);
1550		ib_send_cm_rej(cm_id, IB_CM_REJ_INVALID_GID,
1551			       &work->path[0].sgid, sizeof work->path[0].sgid,
1552			       NULL, 0);
1553		goto rejected;
1554	}
1555	if (req_msg->alt_local_lid) {
1556		ret = cm_init_av_by_path(&work->path[1], &cm_id_priv->alt_av);
1557		if (ret) {
1558			ib_send_cm_rej(cm_id, IB_CM_REJ_INVALID_ALT_GID,
1559				       &work->path[0].sgid,
1560				       sizeof work->path[0].sgid, NULL, 0);
1561			goto rejected;
1562		}
1563	}
1564	cm_id_priv->tid = req_msg->hdr.tid;
1565	cm_id_priv->timeout_ms = cm_convert_to_ms(
1566					cm_req_get_local_resp_timeout(req_msg));
1567	cm_id_priv->max_cm_retries = cm_req_get_max_cm_retries(req_msg);
1568	cm_id_priv->remote_qpn = cm_req_get_local_qpn(req_msg);
1569	cm_id_priv->initiator_depth = cm_req_get_resp_res(req_msg);
1570	cm_id_priv->responder_resources = cm_req_get_init_depth(req_msg);
1571	cm_id_priv->path_mtu = cm_req_get_path_mtu(req_msg);
1572	cm_id_priv->pkey = req_msg->pkey;
1573	cm_id_priv->sq_psn = cm_req_get_starting_psn(req_msg);
1574	cm_id_priv->retry_count = cm_req_get_retry_count(req_msg);
1575	cm_id_priv->rnr_retry_count = cm_req_get_rnr_retry_count(req_msg);
1576	cm_id_priv->qp_type = cm_req_get_qp_type(req_msg);
1577
1578	cm_format_req_event(work, cm_id_priv, &listen_cm_id_priv->id);
1579	cm_process_work(cm_id_priv, work);
1580	cm_deref_id(listen_cm_id_priv);
1581	return 0;
1582
1583rejected:
1584	atomic_dec(&cm_id_priv->refcount);
1585	cm_deref_id(listen_cm_id_priv);
1586destroy:
1587	ib_destroy_cm_id(cm_id);
1588	return ret;
1589}
1590
1591static void cm_format_rep(struct cm_rep_msg *rep_msg,
1592			  struct cm_id_private *cm_id_priv,
1593			  struct ib_cm_rep_param *param)
1594{
1595	cm_format_mad_hdr(&rep_msg->hdr, CM_REP_ATTR_ID, cm_id_priv->tid);
1596	rep_msg->local_comm_id = cm_id_priv->id.local_id;
1597	rep_msg->remote_comm_id = cm_id_priv->id.remote_id;
1598	cm_rep_set_local_qpn(rep_msg, cpu_to_be32(param->qp_num));
1599	cm_rep_set_starting_psn(rep_msg, cpu_to_be32(param->starting_psn));
1600	rep_msg->resp_resources = param->responder_resources;
1601	rep_msg->initiator_depth = param->initiator_depth;
1602	cm_rep_set_target_ack_delay(rep_msg,
1603				    cm_id_priv->av.port->cm_dev->ack_delay);
1604	cm_rep_set_failover(rep_msg, param->failover_accepted);
1605	cm_rep_set_flow_ctrl(rep_msg, param->flow_control);
1606	cm_rep_set_rnr_retry_count(rep_msg, param->rnr_retry_count);
1607	cm_rep_set_srq(rep_msg, param->srq);
1608	rep_msg->local_ca_guid = cm_id_priv->id.device->node_guid;
1609
1610	if (param->private_data && param->private_data_len)
1611		memcpy(rep_msg->private_data, param->private_data,
1612		       param->private_data_len);
1613}
1614
1615int ib_send_cm_rep(struct ib_cm_id *cm_id,
1616		   struct ib_cm_rep_param *param)
1617{
1618	struct cm_id_private *cm_id_priv;
1619	struct ib_mad_send_buf *msg;
1620	struct cm_rep_msg *rep_msg;
1621	unsigned long flags;
1622	int ret;
1623
1624	if (param->private_data &&
1625	    param->private_data_len > IB_CM_REP_PRIVATE_DATA_SIZE)
1626		return -EINVAL;
1627
1628	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1629	spin_lock_irqsave(&cm_id_priv->lock, flags);
1630	if (cm_id->state != IB_CM_REQ_RCVD &&
1631	    cm_id->state != IB_CM_MRA_REQ_SENT) {
1632		ret = -EINVAL;
1633		goto out;
1634	}
1635
1636	ret = cm_alloc_msg(cm_id_priv, &msg);
1637	if (ret)
1638		goto out;
1639
1640	rep_msg = (struct cm_rep_msg *) msg->mad;
1641	cm_format_rep(rep_msg, cm_id_priv, param);
1642	msg->timeout_ms = cm_id_priv->timeout_ms;
1643	msg->context[1] = (void *) (unsigned long) IB_CM_REP_SENT;
1644
1645	ret = ib_post_send_mad(msg, NULL);
1646	if (ret) {
1647		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1648		cm_free_msg(msg);
1649		return ret;
1650	}
1651
1652	cm_id->state = IB_CM_REP_SENT;
1653	cm_id_priv->msg = msg;
1654	cm_id_priv->initiator_depth = param->initiator_depth;
1655	cm_id_priv->responder_resources = param->responder_resources;
1656	cm_id_priv->rq_psn = cm_rep_get_starting_psn(rep_msg);
1657	cm_id_priv->local_qpn = cm_rep_get_local_qpn(rep_msg);
1658
1659out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1660	return ret;
1661}
1662EXPORT_SYMBOL(ib_send_cm_rep);
1663
1664static void cm_format_rtu(struct cm_rtu_msg *rtu_msg,
1665			  struct cm_id_private *cm_id_priv,
1666			  const void *private_data,
1667			  u8 private_data_len)
1668{
1669	cm_format_mad_hdr(&rtu_msg->hdr, CM_RTU_ATTR_ID, cm_id_priv->tid);
1670	rtu_msg->local_comm_id = cm_id_priv->id.local_id;
1671	rtu_msg->remote_comm_id = cm_id_priv->id.remote_id;
1672
1673	if (private_data && private_data_len)
1674		memcpy(rtu_msg->private_data, private_data, private_data_len);
1675}
1676
1677int ib_send_cm_rtu(struct ib_cm_id *cm_id,
1678		   const void *private_data,
1679		   u8 private_data_len)
1680{
1681	struct cm_id_private *cm_id_priv;
1682	struct ib_mad_send_buf *msg;
1683	unsigned long flags;
1684	void *data;
1685	int ret;
1686
1687	if (private_data && private_data_len > IB_CM_RTU_PRIVATE_DATA_SIZE)
1688		return -EINVAL;
1689
1690	data = cm_copy_private_data(private_data, private_data_len);
1691	if (IS_ERR(data))
1692		return PTR_ERR(data);
1693
1694	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1695	spin_lock_irqsave(&cm_id_priv->lock, flags);
1696	if (cm_id->state != IB_CM_REP_RCVD &&
1697	    cm_id->state != IB_CM_MRA_REP_SENT) {
1698		ret = -EINVAL;
1699		goto error;
1700	}
1701
1702	ret = cm_alloc_msg(cm_id_priv, &msg);
1703	if (ret)
1704		goto error;
1705
1706	cm_format_rtu((struct cm_rtu_msg *) msg->mad, cm_id_priv,
1707		      private_data, private_data_len);
1708
1709	ret = ib_post_send_mad(msg, NULL);
1710	if (ret) {
1711		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1712		cm_free_msg(msg);
1713		kfree(data);
1714		return ret;
1715	}
1716
1717	cm_id->state = IB_CM_ESTABLISHED;
1718	cm_set_private_data(cm_id_priv, data, private_data_len);
1719	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1720	return 0;
1721
1722error:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1723	kfree(data);
1724	return ret;
1725}
1726EXPORT_SYMBOL(ib_send_cm_rtu);
1727
1728static void cm_format_rep_event(struct cm_work *work)
1729{
1730	struct cm_rep_msg *rep_msg;
1731	struct ib_cm_rep_event_param *param;
1732
1733	rep_msg = (struct cm_rep_msg *)work->mad_recv_wc->recv_buf.mad;
1734	param = &work->cm_event.param.rep_rcvd;
1735	param->remote_ca_guid = rep_msg->local_ca_guid;
1736	param->remote_qkey = be32_to_cpu(rep_msg->local_qkey);
1737	param->remote_qpn = be32_to_cpu(cm_rep_get_local_qpn(rep_msg));
1738	param->starting_psn = be32_to_cpu(cm_rep_get_starting_psn(rep_msg));
1739	param->responder_resources = rep_msg->initiator_depth;
1740	param->initiator_depth = rep_msg->resp_resources;
1741	param->target_ack_delay = cm_rep_get_target_ack_delay(rep_msg);
1742	param->failover_accepted = cm_rep_get_failover(rep_msg);
1743	param->flow_control = cm_rep_get_flow_ctrl(rep_msg);
1744	param->rnr_retry_count = cm_rep_get_rnr_retry_count(rep_msg);
1745	param->srq = cm_rep_get_srq(rep_msg);
1746	work->cm_event.private_data = &rep_msg->private_data;
1747}
1748
1749static void cm_dup_rep_handler(struct cm_work *work)
1750{
1751	struct cm_id_private *cm_id_priv;
1752	struct cm_rep_msg *rep_msg;
1753	struct ib_mad_send_buf *msg = NULL;
1754	int ret;
1755
1756	rep_msg = (struct cm_rep_msg *) work->mad_recv_wc->recv_buf.mad;
1757	cm_id_priv = cm_acquire_id(rep_msg->remote_comm_id,
1758				   rep_msg->local_comm_id);
1759	if (!cm_id_priv)
1760		return;
1761
1762	atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
1763			counter[CM_REP_COUNTER]);
1764	ret = cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg);
1765	if (ret)
1766		goto deref;
1767
1768	spin_lock_irq(&cm_id_priv->lock);
1769	if (cm_id_priv->id.state == IB_CM_ESTABLISHED)
1770		cm_format_rtu((struct cm_rtu_msg *) msg->mad, cm_id_priv,
1771			      cm_id_priv->private_data,
1772			      cm_id_priv->private_data_len);
1773	else if (cm_id_priv->id.state == IB_CM_MRA_REP_SENT)
1774		cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
1775			      CM_MSG_RESPONSE_REP, cm_id_priv->service_timeout,
1776			      cm_id_priv->private_data,
1777			      cm_id_priv->private_data_len);
1778	else
1779		goto unlock;
1780	spin_unlock_irq(&cm_id_priv->lock);
1781
1782	ret = ib_post_send_mad(msg, NULL);
1783	if (ret)
1784		goto free;
1785	goto deref;
1786
1787unlock:	spin_unlock_irq(&cm_id_priv->lock);
1788free:	cm_free_msg(msg);
1789deref:	cm_deref_id(cm_id_priv);
1790}
1791
1792static int cm_rep_handler(struct cm_work *work)
1793{
1794	struct cm_id_private *cm_id_priv;
1795	struct cm_rep_msg *rep_msg;
1796	int ret;
1797
1798	rep_msg = (struct cm_rep_msg *)work->mad_recv_wc->recv_buf.mad;
1799	cm_id_priv = cm_acquire_id(rep_msg->remote_comm_id, 0);
1800	if (!cm_id_priv) {
1801		cm_dup_rep_handler(work);
1802		return -EINVAL;
1803	}
1804
1805	cm_format_rep_event(work);
1806
1807	spin_lock_irq(&cm_id_priv->lock);
1808	switch (cm_id_priv->id.state) {
1809	case IB_CM_REQ_SENT:
1810	case IB_CM_MRA_REQ_RCVD:
1811		break;
1812	default:
1813		spin_unlock_irq(&cm_id_priv->lock);
1814		ret = -EINVAL;
1815		goto error;
1816	}
1817
1818	cm_id_priv->timewait_info->work.remote_id = rep_msg->local_comm_id;
1819	cm_id_priv->timewait_info->remote_ca_guid = rep_msg->local_ca_guid;
1820	cm_id_priv->timewait_info->remote_qpn = cm_rep_get_local_qpn(rep_msg);
1821
1822	spin_lock(&cm.lock);
1823	/* Check for duplicate REP. */
1824	if (cm_insert_remote_id(cm_id_priv->timewait_info)) {
1825		spin_unlock(&cm.lock);
1826		spin_unlock_irq(&cm_id_priv->lock);
1827		ret = -EINVAL;
1828		goto error;
1829	}
1830	/* Check for a stale connection. */
1831	if (cm_insert_remote_qpn(cm_id_priv->timewait_info)) {
1832		rb_erase(&cm_id_priv->timewait_info->remote_id_node,
1833			 &cm.remote_id_table);
1834		cm_id_priv->timewait_info->inserted_remote_id = 0;
1835		spin_unlock(&cm.lock);
1836		spin_unlock_irq(&cm_id_priv->lock);
1837		cm_issue_rej(work->port, work->mad_recv_wc,
1838			     IB_CM_REJ_STALE_CONN, CM_MSG_RESPONSE_REP,
1839			     NULL, 0);
1840		ret = -EINVAL;
1841		goto error;
1842	}
1843	spin_unlock(&cm.lock);
1844
1845	cm_id_priv->id.state = IB_CM_REP_RCVD;
1846	cm_id_priv->id.remote_id = rep_msg->local_comm_id;
1847	cm_id_priv->remote_qpn = cm_rep_get_local_qpn(rep_msg);
1848	cm_id_priv->initiator_depth = rep_msg->resp_resources;
1849	cm_id_priv->responder_resources = rep_msg->initiator_depth;
1850	cm_id_priv->sq_psn = cm_rep_get_starting_psn(rep_msg);
1851	cm_id_priv->rnr_retry_count = cm_rep_get_rnr_retry_count(rep_msg);
1852	cm_id_priv->target_ack_delay = cm_rep_get_target_ack_delay(rep_msg);
1853	cm_id_priv->av.timeout =
1854			cm_ack_timeout(cm_id_priv->target_ack_delay,
1855				       cm_id_priv->av.timeout - 1);
1856	cm_id_priv->alt_av.timeout =
1857			cm_ack_timeout(cm_id_priv->target_ack_delay,
1858				       cm_id_priv->alt_av.timeout - 1);
1859
1860	/* todo: handle peer_to_peer */
1861
1862	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
1863	ret = atomic_inc_and_test(&cm_id_priv->work_count);
1864	if (!ret)
1865		list_add_tail(&work->list, &cm_id_priv->work_list);
1866	spin_unlock_irq(&cm_id_priv->lock);
1867
1868	if (ret)
1869		cm_process_work(cm_id_priv, work);
1870	else
1871		cm_deref_id(cm_id_priv);
1872	return 0;
1873
1874error:
1875	cm_deref_id(cm_id_priv);
1876	return ret;
1877}
1878
1879static int cm_establish_handler(struct cm_work *work)
1880{
1881	struct cm_id_private *cm_id_priv;
1882	int ret;
1883
1884	/* See comment in cm_establish about lookup. */
1885	cm_id_priv = cm_acquire_id(work->local_id, work->remote_id);
1886	if (!cm_id_priv)
1887		return -EINVAL;
1888
1889	spin_lock_irq(&cm_id_priv->lock);
1890	if (cm_id_priv->id.state != IB_CM_ESTABLISHED) {
1891		spin_unlock_irq(&cm_id_priv->lock);
1892		goto out;
1893	}
1894
1895	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
1896	ret = atomic_inc_and_test(&cm_id_priv->work_count);
1897	if (!ret)
1898		list_add_tail(&work->list, &cm_id_priv->work_list);
1899	spin_unlock_irq(&cm_id_priv->lock);
1900
1901	if (ret)
1902		cm_process_work(cm_id_priv, work);
1903	else
1904		cm_deref_id(cm_id_priv);
1905	return 0;
1906out:
1907	cm_deref_id(cm_id_priv);
1908	return -EINVAL;
1909}
1910
1911static int cm_rtu_handler(struct cm_work *work)
1912{
1913	struct cm_id_private *cm_id_priv;
1914	struct cm_rtu_msg *rtu_msg;
1915	int ret;
1916
1917	rtu_msg = (struct cm_rtu_msg *)work->mad_recv_wc->recv_buf.mad;
1918	cm_id_priv = cm_acquire_id(rtu_msg->remote_comm_id,
1919				   rtu_msg->local_comm_id);
1920	if (!cm_id_priv)
1921		return -EINVAL;
1922
1923	work->cm_event.private_data = &rtu_msg->private_data;
1924
1925	spin_lock_irq(&cm_id_priv->lock);
1926	if (cm_id_priv->id.state != IB_CM_REP_SENT &&
1927	    cm_id_priv->id.state != IB_CM_MRA_REP_RCVD) {
1928		spin_unlock_irq(&cm_id_priv->lock);
1929		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
1930				counter[CM_RTU_COUNTER]);
1931		goto out;
1932	}
1933	cm_id_priv->id.state = IB_CM_ESTABLISHED;
1934
1935	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
1936	ret = atomic_inc_and_test(&cm_id_priv->work_count);
1937	if (!ret)
1938		list_add_tail(&work->list, &cm_id_priv->work_list);
1939	spin_unlock_irq(&cm_id_priv->lock);
1940
1941	if (ret)
1942		cm_process_work(cm_id_priv, work);
1943	else
1944		cm_deref_id(cm_id_priv);
1945	return 0;
1946out:
1947	cm_deref_id(cm_id_priv);
1948	return -EINVAL;
1949}
1950
1951static void cm_format_dreq(struct cm_dreq_msg *dreq_msg,
1952			  struct cm_id_private *cm_id_priv,
1953			  const void *private_data,
1954			  u8 private_data_len)
1955{
1956	cm_format_mad_hdr(&dreq_msg->hdr, CM_DREQ_ATTR_ID,
1957			  cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_DREQ));
1958	dreq_msg->local_comm_id = cm_id_priv->id.local_id;
1959	dreq_msg->remote_comm_id = cm_id_priv->id.remote_id;
1960	cm_dreq_set_remote_qpn(dreq_msg, cm_id_priv->remote_qpn);
1961
1962	if (private_data && private_data_len)
1963		memcpy(dreq_msg->private_data, private_data, private_data_len);
1964}
1965
1966int ib_send_cm_dreq(struct ib_cm_id *cm_id,
1967		    const void *private_data,
1968		    u8 private_data_len)
1969{
1970	struct cm_id_private *cm_id_priv;
1971	struct ib_mad_send_buf *msg;
1972	unsigned long flags;
1973	int ret;
1974
1975	if (private_data && private_data_len > IB_CM_DREQ_PRIVATE_DATA_SIZE)
1976		return -EINVAL;
1977
1978	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1979	spin_lock_irqsave(&cm_id_priv->lock, flags);
1980	if (cm_id->state != IB_CM_ESTABLISHED) {
1981		ret = -EINVAL;
1982		goto out;
1983	}
1984
1985	ret = cm_alloc_msg(cm_id_priv, &msg);
1986	if (ret) {
1987		cm_enter_timewait(cm_id_priv);
1988		goto out;
1989	}
1990
1991	cm_format_dreq((struct cm_dreq_msg *) msg->mad, cm_id_priv,
1992		       private_data, private_data_len);
1993	msg->timeout_ms = cm_id_priv->timeout_ms;
1994	msg->context[1] = (void *) (unsigned long) IB_CM_DREQ_SENT;
1995
1996	ret = ib_post_send_mad(msg, NULL);
1997	if (ret) {
1998		cm_enter_timewait(cm_id_priv);
1999		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2000		cm_free_msg(msg);
2001		return ret;
2002	}
2003
2004	cm_id->state = IB_CM_DREQ_SENT;
2005	cm_id_priv->msg = msg;
2006out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2007	return ret;
2008}
2009EXPORT_SYMBOL(ib_send_cm_dreq);
2010
2011static void cm_format_drep(struct cm_drep_msg *drep_msg,
2012			  struct cm_id_private *cm_id_priv,
2013			  const void *private_data,
2014			  u8 private_data_len)
2015{
2016	cm_format_mad_hdr(&drep_msg->hdr, CM_DREP_ATTR_ID, cm_id_priv->tid);
2017	drep_msg->local_comm_id = cm_id_priv->id.local_id;
2018	drep_msg->remote_comm_id = cm_id_priv->id.remote_id;
2019
2020	if (private_data && private_data_len)
2021		memcpy(drep_msg->private_data, private_data, private_data_len);
2022}
2023
2024int ib_send_cm_drep(struct ib_cm_id *cm_id,
2025		    const void *private_data,
2026		    u8 private_data_len)
2027{
2028	struct cm_id_private *cm_id_priv;
2029	struct ib_mad_send_buf *msg;
2030	unsigned long flags;
2031	void *data;
2032	int ret;
2033
2034	if (private_data && private_data_len > IB_CM_DREP_PRIVATE_DATA_SIZE)
2035		return -EINVAL;
2036
2037	data = cm_copy_private_data(private_data, private_data_len);
2038	if (IS_ERR(data))
2039		return PTR_ERR(data);
2040
2041	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2042	spin_lock_irqsave(&cm_id_priv->lock, flags);
2043	if (cm_id->state != IB_CM_DREQ_RCVD) {
2044		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2045		kfree(data);
2046		return -EINVAL;
2047	}
2048
2049	cm_set_private_data(cm_id_priv, data, private_data_len);
2050	cm_enter_timewait(cm_id_priv);
2051
2052	ret = cm_alloc_msg(cm_id_priv, &msg);
2053	if (ret)
2054		goto out;
2055
2056	cm_format_drep((struct cm_drep_msg *) msg->mad, cm_id_priv,
2057		       private_data, private_data_len);
2058
2059	ret = ib_post_send_mad(msg, NULL);
2060	if (ret) {
2061		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2062		cm_free_msg(msg);
2063		return ret;
2064	}
2065
2066out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2067	return ret;
2068}
2069EXPORT_SYMBOL(ib_send_cm_drep);
2070
2071static int cm_issue_drep(struct cm_port *port,
2072			 struct ib_mad_recv_wc *mad_recv_wc)
2073{
2074	struct ib_mad_send_buf *msg = NULL;
2075	struct cm_dreq_msg *dreq_msg;
2076	struct cm_drep_msg *drep_msg;
2077	int ret;
2078
2079	ret = cm_alloc_response_msg(port, mad_recv_wc, &msg);
2080	if (ret)
2081		return ret;
2082
2083	dreq_msg = (struct cm_dreq_msg *) mad_recv_wc->recv_buf.mad;
2084	drep_msg = (struct cm_drep_msg *) msg->mad;
2085
2086	cm_format_mad_hdr(&drep_msg->hdr, CM_DREP_ATTR_ID, dreq_msg->hdr.tid);
2087	drep_msg->remote_comm_id = dreq_msg->local_comm_id;
2088	drep_msg->local_comm_id = dreq_msg->remote_comm_id;
2089
2090	ret = ib_post_send_mad(msg, NULL);
2091	if (ret)
2092		cm_free_msg(msg);
2093
2094	return ret;
2095}
2096
2097static int cm_dreq_handler(struct cm_work *work)
2098{
2099	struct cm_id_private *cm_id_priv;
2100	struct cm_dreq_msg *dreq_msg;
2101	struct ib_mad_send_buf *msg = NULL;
2102	int ret;
2103
2104	dreq_msg = (struct cm_dreq_msg *)work->mad_recv_wc->recv_buf.mad;
2105	cm_id_priv = cm_acquire_id(dreq_msg->remote_comm_id,
2106				   dreq_msg->local_comm_id);
2107	if (!cm_id_priv) {
2108		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2109				counter[CM_DREQ_COUNTER]);
2110		cm_issue_drep(work->port, work->mad_recv_wc);
2111		return -EINVAL;
2112	}
2113
2114	work->cm_event.private_data = &dreq_msg->private_data;
2115
2116	spin_lock_irq(&cm_id_priv->lock);
2117	if (cm_id_priv->local_qpn != cm_dreq_get_remote_qpn(dreq_msg))
2118		goto unlock;
2119
2120	switch (cm_id_priv->id.state) {
2121	case IB_CM_REP_SENT:
2122	case IB_CM_DREQ_SENT:
2123		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2124		break;
2125	case IB_CM_ESTABLISHED:
2126	case IB_CM_MRA_REP_RCVD:
2127		break;
2128	case IB_CM_TIMEWAIT:
2129		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2130				counter[CM_DREQ_COUNTER]);
2131		if (cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg))
2132			goto unlock;
2133
2134		cm_format_drep((struct cm_drep_msg *) msg->mad, cm_id_priv,
2135			       cm_id_priv->private_data,
2136			       cm_id_priv->private_data_len);
2137		spin_unlock_irq(&cm_id_priv->lock);
2138
2139		if (ib_post_send_mad(msg, NULL))
2140			cm_free_msg(msg);
2141		goto deref;
2142	case IB_CM_DREQ_RCVD:
2143		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2144				counter[CM_DREQ_COUNTER]);
2145		goto unlock;
2146	default:
2147		goto unlock;
2148	}
2149	cm_id_priv->id.state = IB_CM_DREQ_RCVD;
2150	cm_id_priv->tid = dreq_msg->hdr.tid;
2151	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2152	if (!ret)
2153		list_add_tail(&work->list, &cm_id_priv->work_list);
2154	spin_unlock_irq(&cm_id_priv->lock);
2155
2156	if (ret)
2157		cm_process_work(cm_id_priv, work);
2158	else
2159		cm_deref_id(cm_id_priv);
2160	return 0;
2161
2162unlock:	spin_unlock_irq(&cm_id_priv->lock);
2163deref:	cm_deref_id(cm_id_priv);
2164	return -EINVAL;
2165}
2166
2167static int cm_drep_handler(struct cm_work *work)
2168{
2169	struct cm_id_private *cm_id_priv;
2170	struct cm_drep_msg *drep_msg;
2171	int ret;
2172
2173	drep_msg = (struct cm_drep_msg *)work->mad_recv_wc->recv_buf.mad;
2174	cm_id_priv = cm_acquire_id(drep_msg->remote_comm_id,
2175				   drep_msg->local_comm_id);
2176	if (!cm_id_priv)
2177		return -EINVAL;
2178
2179	work->cm_event.private_data = &drep_msg->private_data;
2180
2181	spin_lock_irq(&cm_id_priv->lock);
2182	if (cm_id_priv->id.state != IB_CM_DREQ_SENT &&
2183	    cm_id_priv->id.state != IB_CM_DREQ_RCVD) {
2184		spin_unlock_irq(&cm_id_priv->lock);
2185		goto out;
2186	}
2187	cm_enter_timewait(cm_id_priv);
2188
2189	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2190	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2191	if (!ret)
2192		list_add_tail(&work->list, &cm_id_priv->work_list);
2193	spin_unlock_irq(&cm_id_priv->lock);
2194
2195	if (ret)
2196		cm_process_work(cm_id_priv, work);
2197	else
2198		cm_deref_id(cm_id_priv);
2199	return 0;
2200out:
2201	cm_deref_id(cm_id_priv);
2202	return -EINVAL;
2203}
2204
2205int ib_send_cm_rej(struct ib_cm_id *cm_id,
2206		   enum ib_cm_rej_reason reason,
2207		   void *ari,
2208		   u8 ari_length,
2209		   const void *private_data,
2210		   u8 private_data_len)
2211{
2212	struct cm_id_private *cm_id_priv;
2213	struct ib_mad_send_buf *msg;
2214	unsigned long flags;
2215	int ret;
2216
2217	if ((private_data && private_data_len > IB_CM_REJ_PRIVATE_DATA_SIZE) ||
2218	    (ari && ari_length > IB_CM_REJ_ARI_LENGTH))
2219		return -EINVAL;
2220
2221	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2222
2223	spin_lock_irqsave(&cm_id_priv->lock, flags);
2224	switch (cm_id->state) {
2225	case IB_CM_REQ_SENT:
2226	case IB_CM_MRA_REQ_RCVD:
2227	case IB_CM_REQ_RCVD:
2228	case IB_CM_MRA_REQ_SENT:
2229	case IB_CM_REP_RCVD:
2230	case IB_CM_MRA_REP_SENT:
2231		ret = cm_alloc_msg(cm_id_priv, &msg);
2232		if (!ret)
2233			cm_format_rej((struct cm_rej_msg *) msg->mad,
2234				      cm_id_priv, reason, ari, ari_length,
2235				      private_data, private_data_len);
2236
2237		cm_reset_to_idle(cm_id_priv);
2238		break;
2239	case IB_CM_REP_SENT:
2240	case IB_CM_MRA_REP_RCVD:
2241		ret = cm_alloc_msg(cm_id_priv, &msg);
2242		if (!ret)
2243			cm_format_rej((struct cm_rej_msg *) msg->mad,
2244				      cm_id_priv, reason, ari, ari_length,
2245				      private_data, private_data_len);
2246
2247		cm_enter_timewait(cm_id_priv);
2248		break;
2249	default:
2250		ret = -EINVAL;
2251		goto out;
2252	}
2253
2254	if (ret)
2255		goto out;
2256
2257	ret = ib_post_send_mad(msg, NULL);
2258	if (ret)
2259		cm_free_msg(msg);
2260
2261out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2262	return ret;
2263}
2264EXPORT_SYMBOL(ib_send_cm_rej);
2265
2266static void cm_format_rej_event(struct cm_work *work)
2267{
2268	struct cm_rej_msg *rej_msg;
2269	struct ib_cm_rej_event_param *param;
2270
2271	rej_msg = (struct cm_rej_msg *)work->mad_recv_wc->recv_buf.mad;
2272	param = &work->cm_event.param.rej_rcvd;
2273	param->ari = rej_msg->ari;
2274	param->ari_length = cm_rej_get_reject_info_len(rej_msg);
2275	param->reason = __be16_to_cpu(rej_msg->reason);
2276	work->cm_event.private_data = &rej_msg->private_data;
2277}
2278
2279static struct cm_id_private * cm_acquire_rejected_id(struct cm_rej_msg *rej_msg)
2280{
2281	struct cm_timewait_info *timewait_info;
2282	struct cm_id_private *cm_id_priv;
2283	__be32 remote_id;
2284
2285	remote_id = rej_msg->local_comm_id;
2286
2287	if (__be16_to_cpu(rej_msg->reason) == IB_CM_REJ_TIMEOUT) {
2288		spin_lock_irq(&cm.lock);
2289		timewait_info = cm_find_remote_id( *((__be64 *) rej_msg->ari),
2290						  remote_id);
2291		if (!timewait_info) {
2292			spin_unlock_irq(&cm.lock);
2293			return NULL;
2294		}
2295		cm_id_priv = idr_find(&cm.local_id_table, (__force int)
2296				      (timewait_info->work.local_id ^
2297				       cm.random_id_operand));
2298		if (cm_id_priv) {
2299			if (cm_id_priv->id.remote_id == remote_id)
2300				atomic_inc(&cm_id_priv->refcount);
2301			else
2302				cm_id_priv = NULL;
2303		}
2304		spin_unlock_irq(&cm.lock);
2305	} else if (cm_rej_get_msg_rejected(rej_msg) == CM_MSG_RESPONSE_REQ)
2306		cm_id_priv = cm_acquire_id(rej_msg->remote_comm_id, 0);
2307	else
2308		cm_id_priv = cm_acquire_id(rej_msg->remote_comm_id, remote_id);
2309
2310	return cm_id_priv;
2311}
2312
2313static int cm_rej_handler(struct cm_work *work)
2314{
2315	struct cm_id_private *cm_id_priv;
2316	struct cm_rej_msg *rej_msg;
2317	int ret;
2318
2319	rej_msg = (struct cm_rej_msg *)work->mad_recv_wc->recv_buf.mad;
2320	cm_id_priv = cm_acquire_rejected_id(rej_msg);
2321	if (!cm_id_priv)
2322		return -EINVAL;
2323
2324	cm_format_rej_event(work);
2325
2326	spin_lock_irq(&cm_id_priv->lock);
2327	switch (cm_id_priv->id.state) {
2328	case IB_CM_REQ_SENT:
2329	case IB_CM_MRA_REQ_RCVD:
2330	case IB_CM_REP_SENT:
2331	case IB_CM_MRA_REP_RCVD:
2332		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2333		/* fall through */
2334	case IB_CM_REQ_RCVD:
2335	case IB_CM_MRA_REQ_SENT:
2336		if (__be16_to_cpu(rej_msg->reason) == IB_CM_REJ_STALE_CONN)
2337			cm_enter_timewait(cm_id_priv);
2338		else
2339			cm_reset_to_idle(cm_id_priv);
2340		break;
2341	case IB_CM_DREQ_SENT:
2342		ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2343		/* fall through */
2344	case IB_CM_REP_RCVD:
2345	case IB_CM_MRA_REP_SENT:
2346	case IB_CM_ESTABLISHED:
2347		cm_enter_timewait(cm_id_priv);
2348		break;
2349	default:
2350		spin_unlock_irq(&cm_id_priv->lock);
2351		ret = -EINVAL;
2352		goto out;
2353	}
2354
2355	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2356	if (!ret)
2357		list_add_tail(&work->list, &cm_id_priv->work_list);
2358	spin_unlock_irq(&cm_id_priv->lock);
2359
2360	if (ret)
2361		cm_process_work(cm_id_priv, work);
2362	else
2363		cm_deref_id(cm_id_priv);
2364	return 0;
2365out:
2366	cm_deref_id(cm_id_priv);
2367	return -EINVAL;
2368}
2369
2370int ib_send_cm_mra(struct ib_cm_id *cm_id,
2371		   u8 service_timeout,
2372		   const void *private_data,
2373		   u8 private_data_len)
2374{
2375	struct cm_id_private *cm_id_priv;
2376	struct ib_mad_send_buf *msg;
2377	enum ib_cm_state cm_state;
2378	enum ib_cm_lap_state lap_state;
2379	enum cm_msg_response msg_response;
2380	void *data;
2381	unsigned long flags;
2382	int ret;
2383
2384	if (private_data && private_data_len > IB_CM_MRA_PRIVATE_DATA_SIZE)
2385		return -EINVAL;
2386
2387	data = cm_copy_private_data(private_data, private_data_len);
2388	if (IS_ERR(data))
2389		return PTR_ERR(data);
2390
2391	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2392
2393	spin_lock_irqsave(&cm_id_priv->lock, flags);
2394	switch(cm_id_priv->id.state) {
2395	case IB_CM_REQ_RCVD:
2396		cm_state = IB_CM_MRA_REQ_SENT;
2397		lap_state = cm_id->lap_state;
2398		msg_response = CM_MSG_RESPONSE_REQ;
2399		break;
2400	case IB_CM_REP_RCVD:
2401		cm_state = IB_CM_MRA_REP_SENT;
2402		lap_state = cm_id->lap_state;
2403		msg_response = CM_MSG_RESPONSE_REP;
2404		break;
2405	case IB_CM_ESTABLISHED:
2406		if (cm_id->lap_state == IB_CM_LAP_RCVD) {
2407			cm_state = cm_id->state;
2408			lap_state = IB_CM_MRA_LAP_SENT;
2409			msg_response = CM_MSG_RESPONSE_OTHER;
2410			break;
2411		}
2412	default:
2413		ret = -EINVAL;
2414		goto error1;
2415	}
2416
2417	if (!(service_timeout & IB_CM_MRA_FLAG_DELAY)) {
2418		ret = cm_alloc_msg(cm_id_priv, &msg);
2419		if (ret)
2420			goto error1;
2421
2422		cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
2423			      msg_response, service_timeout,
2424			      private_data, private_data_len);
2425		ret = ib_post_send_mad(msg, NULL);
2426		if (ret)
2427			goto error2;
2428	}
2429
2430	cm_id->state = cm_state;
2431	cm_id->lap_state = lap_state;
2432	cm_id_priv->service_timeout = service_timeout;
2433	cm_set_private_data(cm_id_priv, data, private_data_len);
2434	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2435	return 0;
2436
2437error1:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2438	kfree(data);
2439	return ret;
2440
2441error2:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2442	kfree(data);
2443	cm_free_msg(msg);
2444	return ret;
2445}
2446EXPORT_SYMBOL(ib_send_cm_mra);
2447
2448static struct cm_id_private * cm_acquire_mraed_id(struct cm_mra_msg *mra_msg)
2449{
2450	switch (cm_mra_get_msg_mraed(mra_msg)) {
2451	case CM_MSG_RESPONSE_REQ:
2452		return cm_acquire_id(mra_msg->remote_comm_id, 0);
2453	case CM_MSG_RESPONSE_REP:
2454	case CM_MSG_RESPONSE_OTHER:
2455		return cm_acquire_id(mra_msg->remote_comm_id,
2456				     mra_msg->local_comm_id);
2457	default:
2458		return NULL;
2459	}
2460}
2461
2462static int cm_mra_handler(struct cm_work *work)
2463{
2464	struct cm_id_private *cm_id_priv;
2465	struct cm_mra_msg *mra_msg;
2466	int timeout, ret;
2467
2468	mra_msg = (struct cm_mra_msg *)work->mad_recv_wc->recv_buf.mad;
2469	cm_id_priv = cm_acquire_mraed_id(mra_msg);
2470	if (!cm_id_priv)
2471		return -EINVAL;
2472
2473	work->cm_event.private_data = &mra_msg->private_data;
2474	work->cm_event.param.mra_rcvd.service_timeout =
2475					cm_mra_get_service_timeout(mra_msg);
2476	timeout = cm_convert_to_ms(cm_mra_get_service_timeout(mra_msg)) +
2477		  cm_convert_to_ms(cm_id_priv->av.timeout);
2478
2479	spin_lock_irq(&cm_id_priv->lock);
2480	switch (cm_id_priv->id.state) {
2481	case IB_CM_REQ_SENT:
2482		if (cm_mra_get_msg_mraed(mra_msg) != CM_MSG_RESPONSE_REQ ||
2483		    ib_modify_mad(cm_id_priv->av.port->mad_agent,
2484				  cm_id_priv->msg, timeout))
2485			goto out;
2486		cm_id_priv->id.state = IB_CM_MRA_REQ_RCVD;
2487		break;
2488	case IB_CM_REP_SENT:
2489		if (cm_mra_get_msg_mraed(mra_msg) != CM_MSG_RESPONSE_REP ||
2490		    ib_modify_mad(cm_id_priv->av.port->mad_agent,
2491				  cm_id_priv->msg, timeout))
2492			goto out;
2493		cm_id_priv->id.state = IB_CM_MRA_REP_RCVD;
2494		break;
2495	case IB_CM_ESTABLISHED:
2496		if (cm_mra_get_msg_mraed(mra_msg) != CM_MSG_RESPONSE_OTHER ||
2497		    cm_id_priv->id.lap_state != IB_CM_LAP_SENT ||
2498		    ib_modify_mad(cm_id_priv->av.port->mad_agent,
2499				  cm_id_priv->msg, timeout)) {
2500			if (cm_id_priv->id.lap_state == IB_CM_MRA_LAP_RCVD)
2501				atomic_long_inc(&work->port->
2502						counter_group[CM_RECV_DUPLICATES].
2503						counter[CM_MRA_COUNTER]);
2504			goto out;
2505		}
2506		cm_id_priv->id.lap_state = IB_CM_MRA_LAP_RCVD;
2507		break;
2508	case IB_CM_MRA_REQ_RCVD:
2509	case IB_CM_MRA_REP_RCVD:
2510		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2511				counter[CM_MRA_COUNTER]);
2512		/* fall through */
2513	default:
2514		goto out;
2515	}
2516
2517	cm_id_priv->msg->context[1] = (void *) (unsigned long)
2518				      cm_id_priv->id.state;
2519	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2520	if (!ret)
2521		list_add_tail(&work->list, &cm_id_priv->work_list);
2522	spin_unlock_irq(&cm_id_priv->lock);
2523
2524	if (ret)
2525		cm_process_work(cm_id_priv, work);
2526	else
2527		cm_deref_id(cm_id_priv);
2528	return 0;
2529out:
2530	spin_unlock_irq(&cm_id_priv->lock);
2531	cm_deref_id(cm_id_priv);
2532	return -EINVAL;
2533}
2534
2535static void cm_format_lap(struct cm_lap_msg *lap_msg,
2536			  struct cm_id_private *cm_id_priv,
2537			  struct ib_sa_path_rec *alternate_path,
2538			  const void *private_data,
2539			  u8 private_data_len)
2540{
2541	cm_format_mad_hdr(&lap_msg->hdr, CM_LAP_ATTR_ID,
2542			  cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_LAP));
2543	lap_msg->local_comm_id = cm_id_priv->id.local_id;
2544	lap_msg->remote_comm_id = cm_id_priv->id.remote_id;
2545	cm_lap_set_remote_qpn(lap_msg, cm_id_priv->remote_qpn);
2546	/* todo: need remote CM response timeout */
2547	cm_lap_set_remote_resp_timeout(lap_msg, 0x1F);
2548	lap_msg->alt_local_lid = alternate_path->slid;
2549	lap_msg->alt_remote_lid = alternate_path->dlid;
2550	lap_msg->alt_local_gid = alternate_path->sgid;
2551	lap_msg->alt_remote_gid = alternate_path->dgid;
2552	cm_lap_set_flow_label(lap_msg, alternate_path->flow_label);
2553	cm_lap_set_traffic_class(lap_msg, alternate_path->traffic_class);
2554	lap_msg->alt_hop_limit = alternate_path->hop_limit;
2555	cm_lap_set_packet_rate(lap_msg, alternate_path->rate);
2556	cm_lap_set_sl(lap_msg, alternate_path->sl);
2557	cm_lap_set_subnet_local(lap_msg, 1); /* local only... */
2558	cm_lap_set_local_ack_timeout(lap_msg,
2559		cm_ack_timeout(cm_id_priv->av.port->cm_dev->ack_delay,
2560			       alternate_path->packet_life_time));
2561
2562	if (private_data && private_data_len)
2563		memcpy(lap_msg->private_data, private_data, private_data_len);
2564}
2565
2566int ib_send_cm_lap(struct ib_cm_id *cm_id,
2567		   struct ib_sa_path_rec *alternate_path,
2568		   const void *private_data,
2569		   u8 private_data_len)
2570{
2571	struct cm_id_private *cm_id_priv;
2572	struct ib_mad_send_buf *msg;
2573	unsigned long flags;
2574	int ret;
2575
2576	if (private_data && private_data_len > IB_CM_LAP_PRIVATE_DATA_SIZE)
2577		return -EINVAL;
2578
2579	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2580	spin_lock_irqsave(&cm_id_priv->lock, flags);
2581	if (cm_id->state != IB_CM_ESTABLISHED ||
2582	    (cm_id->lap_state != IB_CM_LAP_UNINIT &&
2583	     cm_id->lap_state != IB_CM_LAP_IDLE)) {
2584		ret = -EINVAL;
2585		goto out;
2586	}
2587
2588	ret = cm_init_av_by_path(alternate_path, &cm_id_priv->alt_av);
2589	if (ret)
2590		goto out;
2591	cm_id_priv->alt_av.timeout =
2592			cm_ack_timeout(cm_id_priv->target_ack_delay,
2593				       cm_id_priv->alt_av.timeout - 1);
2594
2595	ret = cm_alloc_msg(cm_id_priv, &msg);
2596	if (ret)
2597		goto out;
2598
2599	cm_format_lap((struct cm_lap_msg *) msg->mad, cm_id_priv,
2600		      alternate_path, private_data, private_data_len);
2601	msg->timeout_ms = cm_id_priv->timeout_ms;
2602	msg->context[1] = (void *) (unsigned long) IB_CM_ESTABLISHED;
2603
2604	ret = ib_post_send_mad(msg, NULL);
2605	if (ret) {
2606		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2607		cm_free_msg(msg);
2608		return ret;
2609	}
2610
2611	cm_id->lap_state = IB_CM_LAP_SENT;
2612	cm_id_priv->msg = msg;
2613
2614out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2615	return ret;
2616}
2617EXPORT_SYMBOL(ib_send_cm_lap);
2618
2619static void cm_format_path_from_lap(struct cm_id_private *cm_id_priv,
2620				    struct ib_sa_path_rec *path,
2621				    struct cm_lap_msg *lap_msg)
2622{
2623	memset(path, 0, sizeof *path);
2624	path->dgid = lap_msg->alt_local_gid;
2625	path->sgid = lap_msg->alt_remote_gid;
2626	path->dlid = lap_msg->alt_local_lid;
2627	path->slid = lap_msg->alt_remote_lid;
2628	path->flow_label = cm_lap_get_flow_label(lap_msg);
2629	path->hop_limit = lap_msg->alt_hop_limit;
2630	path->traffic_class = cm_lap_get_traffic_class(lap_msg);
2631	path->reversible = 1;
2632	path->pkey = cm_id_priv->pkey;
2633	path->sl = cm_lap_get_sl(lap_msg);
2634	path->mtu_selector = IB_SA_EQ;
2635	path->mtu = cm_id_priv->path_mtu;
2636	path->rate_selector = IB_SA_EQ;
2637	path->rate = cm_lap_get_packet_rate(lap_msg);
2638	path->packet_life_time_selector = IB_SA_EQ;
2639	path->packet_life_time = cm_lap_get_local_ack_timeout(lap_msg);
2640	path->packet_life_time -= (path->packet_life_time > 0);
2641}
2642
2643static int cm_lap_handler(struct cm_work *work)
2644{
2645	struct cm_id_private *cm_id_priv;
2646	struct cm_lap_msg *lap_msg;
2647	struct ib_cm_lap_event_param *param;
2648	struct ib_mad_send_buf *msg = NULL;
2649	int ret;
2650
2651	/* todo: verify LAP request and send reject APR if invalid. */
2652	lap_msg = (struct cm_lap_msg *)work->mad_recv_wc->recv_buf.mad;
2653	cm_id_priv = cm_acquire_id(lap_msg->remote_comm_id,
2654				   lap_msg->local_comm_id);
2655	if (!cm_id_priv)
2656		return -EINVAL;
2657
2658	param = &work->cm_event.param.lap_rcvd;
2659	param->alternate_path = &work->path[0];
2660	cm_format_path_from_lap(cm_id_priv, param->alternate_path, lap_msg);
2661	work->cm_event.private_data = &lap_msg->private_data;
2662
2663	spin_lock_irq(&cm_id_priv->lock);
2664	if (cm_id_priv->id.state != IB_CM_ESTABLISHED)
2665		goto unlock;
2666
2667	switch (cm_id_priv->id.lap_state) {
2668	case IB_CM_LAP_UNINIT:
2669	case IB_CM_LAP_IDLE:
2670		break;
2671	case IB_CM_MRA_LAP_SENT:
2672		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2673				counter[CM_LAP_COUNTER]);
2674		if (cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg))
2675			goto unlock;
2676
2677		cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
2678			      CM_MSG_RESPONSE_OTHER,
2679			      cm_id_priv->service_timeout,
2680			      cm_id_priv->private_data,
2681			      cm_id_priv->private_data_len);
2682		spin_unlock_irq(&cm_id_priv->lock);
2683
2684		if (ib_post_send_mad(msg, NULL))
2685			cm_free_msg(msg);
2686		goto deref;
2687	case IB_CM_LAP_RCVD:
2688		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2689				counter[CM_LAP_COUNTER]);
2690		goto unlock;
2691	default:
2692		goto unlock;
2693	}
2694
2695	cm_id_priv->id.lap_state = IB_CM_LAP_RCVD;
2696	cm_id_priv->tid = lap_msg->hdr.tid;
2697	cm_init_av_for_response(work->port, work->mad_recv_wc->wc,
2698				work->mad_recv_wc->recv_buf.grh,
2699				&cm_id_priv->av);
2700	cm_init_av_by_path(param->alternate_path, &cm_id_priv->alt_av);
2701	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2702	if (!ret)
2703		list_add_tail(&work->list, &cm_id_priv->work_list);
2704	spin_unlock_irq(&cm_id_priv->lock);
2705
2706	if (ret)
2707		cm_process_work(cm_id_priv, work);
2708	else
2709		cm_deref_id(cm_id_priv);
2710	return 0;
2711
2712unlock:	spin_unlock_irq(&cm_id_priv->lock);
2713deref:	cm_deref_id(cm_id_priv);
2714	return -EINVAL;
2715}
2716
2717static void cm_format_apr(struct cm_apr_msg *apr_msg,
2718			  struct cm_id_private *cm_id_priv,
2719			  enum ib_cm_apr_status status,
2720			  void *info,
2721			  u8 info_length,
2722			  const void *private_data,
2723			  u8 private_data_len)
2724{
2725	cm_format_mad_hdr(&apr_msg->hdr, CM_APR_ATTR_ID, cm_id_priv->tid);
2726	apr_msg->local_comm_id = cm_id_priv->id.local_id;
2727	apr_msg->remote_comm_id = cm_id_priv->id.remote_id;
2728	apr_msg->ap_status = (u8) status;
2729
2730	if (info && info_length) {
2731		apr_msg->info_length = info_length;
2732		memcpy(apr_msg->info, info, info_length);
2733	}
2734
2735	if (private_data && private_data_len)
2736		memcpy(apr_msg->private_data, private_data, private_data_len);
2737}
2738
2739int ib_send_cm_apr(struct ib_cm_id *cm_id,
2740		   enum ib_cm_apr_status status,
2741		   void *info,
2742		   u8 info_length,
2743		   const void *private_data,
2744		   u8 private_data_len)
2745{
2746	struct cm_id_private *cm_id_priv;
2747	struct ib_mad_send_buf *msg;
2748	unsigned long flags;
2749	int ret;
2750
2751	if ((private_data && private_data_len > IB_CM_APR_PRIVATE_DATA_SIZE) ||
2752	    (info && info_length > IB_CM_APR_INFO_LENGTH))
2753		return -EINVAL;
2754
2755	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2756	spin_lock_irqsave(&cm_id_priv->lock, flags);
2757	if (cm_id->state != IB_CM_ESTABLISHED ||
2758	    (cm_id->lap_state != IB_CM_LAP_RCVD &&
2759	     cm_id->lap_state != IB_CM_MRA_LAP_SENT)) {
2760		ret = -EINVAL;
2761		goto out;
2762	}
2763
2764	ret = cm_alloc_msg(cm_id_priv, &msg);
2765	if (ret)
2766		goto out;
2767
2768	cm_format_apr((struct cm_apr_msg *) msg->mad, cm_id_priv, status,
2769		      info, info_length, private_data, private_data_len);
2770	ret = ib_post_send_mad(msg, NULL);
2771	if (ret) {
2772		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2773		cm_free_msg(msg);
2774		return ret;
2775	}
2776
2777	cm_id->lap_state = IB_CM_LAP_IDLE;
2778out:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2779	return ret;
2780}
2781EXPORT_SYMBOL(ib_send_cm_apr);
2782
2783static int cm_apr_handler(struct cm_work *work)
2784{
2785	struct cm_id_private *cm_id_priv;
2786	struct cm_apr_msg *apr_msg;
2787	int ret;
2788
2789	apr_msg = (struct cm_apr_msg *)work->mad_recv_wc->recv_buf.mad;
2790	cm_id_priv = cm_acquire_id(apr_msg->remote_comm_id,
2791				   apr_msg->local_comm_id);
2792	if (!cm_id_priv)
2793		return -EINVAL; /* Unmatched reply. */
2794
2795	work->cm_event.param.apr_rcvd.ap_status = apr_msg->ap_status;
2796	work->cm_event.param.apr_rcvd.apr_info = &apr_msg->info;
2797	work->cm_event.param.apr_rcvd.info_len = apr_msg->info_length;
2798	work->cm_event.private_data = &apr_msg->private_data;
2799
2800	spin_lock_irq(&cm_id_priv->lock);
2801	if (cm_id_priv->id.state != IB_CM_ESTABLISHED ||
2802	    (cm_id_priv->id.lap_state != IB_CM_LAP_SENT &&
2803	     cm_id_priv->id.lap_state != IB_CM_MRA_LAP_RCVD)) {
2804		spin_unlock_irq(&cm_id_priv->lock);
2805		goto out;
2806	}
2807	cm_id_priv->id.lap_state = IB_CM_LAP_IDLE;
2808	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2809	cm_id_priv->msg = NULL;
2810
2811	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2812	if (!ret)
2813		list_add_tail(&work->list, &cm_id_priv->work_list);
2814	spin_unlock_irq(&cm_id_priv->lock);
2815
2816	if (ret)
2817		cm_process_work(cm_id_priv, work);
2818	else
2819		cm_deref_id(cm_id_priv);
2820	return 0;
2821out:
2822	cm_deref_id(cm_id_priv);
2823	return -EINVAL;
2824}
2825
2826static int cm_timewait_handler(struct cm_work *work)
2827{
2828	struct cm_timewait_info *timewait_info;
2829	struct cm_id_private *cm_id_priv;
2830	int ret;
2831
2832	timewait_info = (struct cm_timewait_info *)work;
2833	spin_lock_irq(&cm.lock);
2834	list_del(&timewait_info->list);
2835	spin_unlock_irq(&cm.lock);
2836
2837	cm_id_priv = cm_acquire_id(timewait_info->work.local_id,
2838				   timewait_info->work.remote_id);
2839	if (!cm_id_priv)
2840		return -EINVAL;
2841
2842	spin_lock_irq(&cm_id_priv->lock);
2843	if (cm_id_priv->id.state != IB_CM_TIMEWAIT ||
2844	    cm_id_priv->remote_qpn != timewait_info->remote_qpn) {
2845		spin_unlock_irq(&cm_id_priv->lock);
2846		goto out;
2847	}
2848	cm_id_priv->id.state = IB_CM_IDLE;
2849	ret = atomic_inc_and_test(&cm_id_priv->work_count);
2850	if (!ret)
2851		list_add_tail(&work->list, &cm_id_priv->work_list);
2852	spin_unlock_irq(&cm_id_priv->lock);
2853
2854	if (ret)
2855		cm_process_work(cm_id_priv, work);
2856	else
2857		cm_deref_id(cm_id_priv);
2858	return 0;
2859out:
2860	cm_deref_id(cm_id_priv);
2861	return -EINVAL;
2862}
2863
2864static void cm_format_sidr_req(struct cm_sidr_req_msg *sidr_req_msg,
2865			       struct cm_id_private *cm_id_priv,
2866			       struct ib_cm_sidr_req_param *param)
2867{
2868	cm_format_mad_hdr(&sidr_req_msg->hdr, CM_SIDR_REQ_ATTR_ID,
2869			  cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_SIDR));
2870	sidr_req_msg->request_id = cm_id_priv->id.local_id;
2871	sidr_req_msg->pkey = param->path->pkey;
2872	sidr_req_msg->service_id = param->service_id;
2873
2874	if (param->private_data && param->private_data_len)
2875		memcpy(sidr_req_msg->private_data, param->private_data,
2876		       param->private_data_len);
2877}
2878
2879int ib_send_cm_sidr_req(struct ib_cm_id *cm_id,
2880			struct ib_cm_sidr_req_param *param)
2881{
2882	struct cm_id_private *cm_id_priv;
2883	struct ib_mad_send_buf *msg;
2884	unsigned long flags;
2885	int ret;
2886
2887	if (!param->path || (param->private_data &&
2888	     param->private_data_len > IB_CM_SIDR_REQ_PRIVATE_DATA_SIZE))
2889		return -EINVAL;
2890
2891	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2892	ret = cm_init_av_by_path(param->path, &cm_id_priv->av);
2893	if (ret)
2894		goto out;
2895
2896	cm_id->service_id = param->service_id;
2897	cm_id->service_mask = ~cpu_to_be64(0);
2898	cm_id_priv->timeout_ms = param->timeout_ms;
2899	cm_id_priv->max_cm_retries = param->max_cm_retries;
2900	ret = cm_alloc_msg(cm_id_priv, &msg);
2901	if (ret)
2902		goto out;
2903
2904	cm_format_sidr_req((struct cm_sidr_req_msg *) msg->mad, cm_id_priv,
2905			   param);
2906	msg->timeout_ms = cm_id_priv->timeout_ms;
2907	msg->context[1] = (void *) (unsigned long) IB_CM_SIDR_REQ_SENT;
2908
2909	spin_lock_irqsave(&cm_id_priv->lock, flags);
2910	if (cm_id->state == IB_CM_IDLE)
2911		ret = ib_post_send_mad(msg, NULL);
2912	else
2913		ret = -EINVAL;
2914
2915	if (ret) {
2916		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2917		cm_free_msg(msg);
2918		goto out;
2919	}
2920	cm_id->state = IB_CM_SIDR_REQ_SENT;
2921	cm_id_priv->msg = msg;
2922	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2923out:
2924	return ret;
2925}
2926EXPORT_SYMBOL(ib_send_cm_sidr_req);
2927
2928static void cm_format_sidr_req_event(struct cm_work *work,
2929				     struct ib_cm_id *listen_id)
2930{
2931	struct cm_sidr_req_msg *sidr_req_msg;
2932	struct ib_cm_sidr_req_event_param *param;
2933
2934	sidr_req_msg = (struct cm_sidr_req_msg *)
2935				work->mad_recv_wc->recv_buf.mad;
2936	param = &work->cm_event.param.sidr_req_rcvd;
2937	param->pkey = __be16_to_cpu(sidr_req_msg->pkey);
2938	param->listen_id = listen_id;
2939	param->port = work->port->port_num;
2940	work->cm_event.private_data = &sidr_req_msg->private_data;
2941}
2942
2943static int cm_sidr_req_handler(struct cm_work *work)
2944{
2945	struct ib_cm_id *cm_id;
2946	struct cm_id_private *cm_id_priv, *cur_cm_id_priv;
2947	struct cm_sidr_req_msg *sidr_req_msg;
2948	struct ib_wc *wc;
2949
2950	cm_id = ib_create_cm_id(work->port->cm_dev->ib_device, NULL, NULL);
2951	if (IS_ERR(cm_id))
2952		return PTR_ERR(cm_id);
2953	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2954
2955	/* Record SGID/SLID and request ID for lookup. */
2956	sidr_req_msg = (struct cm_sidr_req_msg *)
2957				work->mad_recv_wc->recv_buf.mad;
2958	wc = work->mad_recv_wc->wc;
2959	cm_id_priv->av.dgid.global.subnet_prefix = cpu_to_be64(wc->slid);
2960	cm_id_priv->av.dgid.global.interface_id = 0;
2961	cm_init_av_for_response(work->port, work->mad_recv_wc->wc,
2962				work->mad_recv_wc->recv_buf.grh,
2963				&cm_id_priv->av);
2964	cm_id_priv->id.remote_id = sidr_req_msg->request_id;
2965	cm_id_priv->tid = sidr_req_msg->hdr.tid;
2966	atomic_inc(&cm_id_priv->work_count);
2967
2968	spin_lock_irq(&cm.lock);
2969	cur_cm_id_priv = cm_insert_remote_sidr(cm_id_priv);
2970	if (cur_cm_id_priv) {
2971		spin_unlock_irq(&cm.lock);
2972		atomic_long_inc(&work->port->counter_group[CM_RECV_DUPLICATES].
2973				counter[CM_SIDR_REQ_COUNTER]);
2974		goto out; /* Duplicate message. */
2975	}
2976	cm_id_priv->id.state = IB_CM_SIDR_REQ_RCVD;
2977	cur_cm_id_priv = cm_find_listen(cm_id->device,
2978					sidr_req_msg->service_id,
2979					sidr_req_msg->private_data);
2980	if (!cur_cm_id_priv) {
2981		spin_unlock_irq(&cm.lock);
2982		cm_reject_sidr_req(cm_id_priv, IB_SIDR_UNSUPPORTED);
2983		goto out; /* No match. */
2984	}
2985	atomic_inc(&cur_cm_id_priv->refcount);
2986	spin_unlock_irq(&cm.lock);
2987
2988	cm_id_priv->id.cm_handler = cur_cm_id_priv->id.cm_handler;
2989	cm_id_priv->id.context = cur_cm_id_priv->id.context;
2990	cm_id_priv->id.service_id = sidr_req_msg->service_id;
2991	cm_id_priv->id.service_mask = ~cpu_to_be64(0);
2992
2993	cm_format_sidr_req_event(work, &cur_cm_id_priv->id);
2994	cm_process_work(cm_id_priv, work);
2995	cm_deref_id(cur_cm_id_priv);
2996	return 0;
2997out:
2998	ib_destroy_cm_id(&cm_id_priv->id);
2999	return -EINVAL;
3000}
3001
3002static void cm_format_sidr_rep(struct cm_sidr_rep_msg *sidr_rep_msg,
3003			       struct cm_id_private *cm_id_priv,
3004			       struct ib_cm_sidr_rep_param *param)
3005{
3006	cm_format_mad_hdr(&sidr_rep_msg->hdr, CM_SIDR_REP_ATTR_ID,
3007			  cm_id_priv->tid);
3008	sidr_rep_msg->request_id = cm_id_priv->id.remote_id;
3009	sidr_rep_msg->status = param->status;
3010	cm_sidr_rep_set_qpn(sidr_rep_msg, cpu_to_be32(param->qp_num));
3011	sidr_rep_msg->service_id = cm_id_priv->id.service_id;
3012	sidr_rep_msg->qkey = cpu_to_be32(param->qkey);
3013
3014	if (param->info && param->info_length)
3015		memcpy(sidr_rep_msg->info, param->info, param->info_length);
3016
3017	if (param->private_data && param->private_data_len)
3018		memcpy(sidr_rep_msg->private_data, param->private_data,
3019		       param->private_data_len);
3020}
3021
3022int ib_send_cm_sidr_rep(struct ib_cm_id *cm_id,
3023			struct ib_cm_sidr_rep_param *param)
3024{
3025	struct cm_id_private *cm_id_priv;
3026	struct ib_mad_send_buf *msg;
3027	unsigned long flags;
3028	int ret;
3029
3030	if ((param->info && param->info_length > IB_CM_SIDR_REP_INFO_LENGTH) ||
3031	    (param->private_data &&
3032	     param->private_data_len > IB_CM_SIDR_REP_PRIVATE_DATA_SIZE))
3033		return -EINVAL;
3034
3035	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3036	spin_lock_irqsave(&cm_id_priv->lock, flags);
3037	if (cm_id->state != IB_CM_SIDR_REQ_RCVD) {
3038		ret = -EINVAL;
3039		goto error;
3040	}
3041
3042	ret = cm_alloc_msg(cm_id_priv, &msg);
3043	if (ret)
3044		goto error;
3045
3046	cm_format_sidr_rep((struct cm_sidr_rep_msg *) msg->mad, cm_id_priv,
3047			   param);
3048	ret = ib_post_send_mad(msg, NULL);
3049	if (ret) {
3050		spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3051		cm_free_msg(msg);
3052		return ret;
3053	}
3054	cm_id->state = IB_CM_IDLE;
3055	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3056
3057	spin_lock_irqsave(&cm.lock, flags);
3058	rb_erase(&cm_id_priv->sidr_id_node, &cm.remote_sidr_table);
3059	spin_unlock_irqrestore(&cm.lock, flags);
3060	return 0;
3061
3062error:	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3063	return ret;
3064}
3065EXPORT_SYMBOL(ib_send_cm_sidr_rep);
3066
3067static void cm_format_sidr_rep_event(struct cm_work *work)
3068{
3069	struct cm_sidr_rep_msg *sidr_rep_msg;
3070	struct ib_cm_sidr_rep_event_param *param;
3071
3072	sidr_rep_msg = (struct cm_sidr_rep_msg *)
3073				work->mad_recv_wc->recv_buf.mad;
3074	param = &work->cm_event.param.sidr_rep_rcvd;
3075	param->status = sidr_rep_msg->status;
3076	param->qkey = be32_to_cpu(sidr_rep_msg->qkey);
3077	param->qpn = be32_to_cpu(cm_sidr_rep_get_qpn(sidr_rep_msg));
3078	param->info = &sidr_rep_msg->info;
3079	param->info_len = sidr_rep_msg->info_length;
3080	work->cm_event.private_data = &sidr_rep_msg->private_data;
3081}
3082
3083static int cm_sidr_rep_handler(struct cm_work *work)
3084{
3085	struct cm_sidr_rep_msg *sidr_rep_msg;
3086	struct cm_id_private *cm_id_priv;
3087
3088	sidr_rep_msg = (struct cm_sidr_rep_msg *)
3089				work->mad_recv_wc->recv_buf.mad;
3090	cm_id_priv = cm_acquire_id(sidr_rep_msg->request_id, 0);
3091	if (!cm_id_priv)
3092		return -EINVAL; /* Unmatched reply. */
3093
3094	spin_lock_irq(&cm_id_priv->lock);
3095	if (cm_id_priv->id.state != IB_CM_SIDR_REQ_SENT) {
3096		spin_unlock_irq(&cm_id_priv->lock);
3097		goto out;
3098	}
3099	cm_id_priv->id.state = IB_CM_IDLE;
3100	ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
3101	spin_unlock_irq(&cm_id_priv->lock);
3102
3103	cm_format_sidr_rep_event(work);
3104	cm_process_work(cm_id_priv, work);
3105	return 0;
3106out:
3107	cm_deref_id(cm_id_priv);
3108	return -EINVAL;
3109}
3110
3111static void cm_process_send_error(struct ib_mad_send_buf *msg,
3112				  enum ib_wc_status wc_status)
3113{
3114	struct cm_id_private *cm_id_priv;
3115	struct ib_cm_event cm_event;
3116	enum ib_cm_state state;
3117	int ret;
3118
3119	memset(&cm_event, 0, sizeof cm_event);
3120	cm_id_priv = msg->context[0];
3121
3122	/* Discard old sends or ones without a response. */
3123	spin_lock_irq(&cm_id_priv->lock);
3124	state = (enum ib_cm_state) (unsigned long) msg->context[1];
3125	if (msg != cm_id_priv->msg || state != cm_id_priv->id.state)
3126		goto discard;
3127
3128	switch (state) {
3129	case IB_CM_REQ_SENT:
3130	case IB_CM_MRA_REQ_RCVD:
3131		cm_reset_to_idle(cm_id_priv);
3132		cm_event.event = IB_CM_REQ_ERROR;
3133		break;
3134	case IB_CM_REP_SENT:
3135	case IB_CM_MRA_REP_RCVD:
3136		cm_reset_to_idle(cm_id_priv);
3137		cm_event.event = IB_CM_REP_ERROR;
3138		break;
3139	case IB_CM_DREQ_SENT:
3140		cm_enter_timewait(cm_id_priv);
3141		cm_event.event = IB_CM_DREQ_ERROR;
3142		break;
3143	case IB_CM_SIDR_REQ_SENT:
3144		cm_id_priv->id.state = IB_CM_IDLE;
3145		cm_event.event = IB_CM_SIDR_REQ_ERROR;
3146		break;
3147	default:
3148		goto discard;
3149	}
3150	spin_unlock_irq(&cm_id_priv->lock);
3151	cm_event.param.send_status = wc_status;
3152
3153	/* No other events can occur on the cm_id at this point. */
3154	ret = cm_id_priv->id.cm_handler(&cm_id_priv->id, &cm_event);
3155	cm_free_msg(msg);
3156	if (ret)
3157		ib_destroy_cm_id(&cm_id_priv->id);
3158	return;
3159discard:
3160	spin_unlock_irq(&cm_id_priv->lock);
3161	cm_free_msg(msg);
3162}
3163
3164static void cm_send_handler(struct ib_mad_agent *mad_agent,
3165			    struct ib_mad_send_wc *mad_send_wc)
3166{
3167	struct ib_mad_send_buf *msg = mad_send_wc->send_buf;
3168	struct cm_port *port;
3169	u16 attr_index;
3170
3171	port = mad_agent->context;
3172	attr_index = be16_to_cpu(((struct ib_mad_hdr *)
3173				  msg->mad)->attr_id) - CM_ATTR_ID_OFFSET;
3174
3175	/*
3176	 * If the send was in response to a received message (context[0] is not
3177	 * set to a cm_id), and is not a REJ, then it is a send that was
3178	 * manually retried.
3179	 */
3180	if (!msg->context[0] && (attr_index != CM_REJ_COUNTER))
3181		msg->retries = 1;
3182
3183	atomic_long_add(1 + msg->retries,
3184			&port->counter_group[CM_XMIT].counter[attr_index]);
3185	if (msg->retries)
3186		atomic_long_add(msg->retries,
3187				&port->counter_group[CM_XMIT_RETRIES].
3188				counter[attr_index]);
3189
3190	switch (mad_send_wc->status) {
3191	case IB_WC_SUCCESS:
3192	case IB_WC_WR_FLUSH_ERR:
3193		cm_free_msg(msg);
3194		break;
3195	default:
3196		if (msg->context[0] && msg->context[1])
3197			cm_process_send_error(msg, mad_send_wc->status);
3198		else
3199			cm_free_msg(msg);
3200		break;
3201	}
3202}
3203
3204static void cm_work_handler(struct work_struct *_work)
3205{
3206	struct cm_work *work = container_of(_work, struct cm_work, work.work);
3207	int ret;
3208
3209	switch (work->cm_event.event) {
3210	case IB_CM_REQ_RECEIVED:
3211		ret = cm_req_handler(work);
3212		break;
3213	case IB_CM_MRA_RECEIVED:
3214		ret = cm_mra_handler(work);
3215		break;
3216	case IB_CM_REJ_RECEIVED:
3217		ret = cm_rej_handler(work);
3218		break;
3219	case IB_CM_REP_RECEIVED:
3220		ret = cm_rep_handler(work);
3221		break;
3222	case IB_CM_RTU_RECEIVED:
3223		ret = cm_rtu_handler(work);
3224		break;
3225	case IB_CM_USER_ESTABLISHED:
3226		ret = cm_establish_handler(work);
3227		break;
3228	case IB_CM_DREQ_RECEIVED:
3229		ret = cm_dreq_handler(work);
3230		break;
3231	case IB_CM_DREP_RECEIVED:
3232		ret = cm_drep_handler(work);
3233		break;
3234	case IB_CM_SIDR_REQ_RECEIVED:
3235		ret = cm_sidr_req_handler(work);
3236		break;
3237	case IB_CM_SIDR_REP_RECEIVED:
3238		ret = cm_sidr_rep_handler(work);
3239		break;
3240	case IB_CM_LAP_RECEIVED:
3241		ret = cm_lap_handler(work);
3242		break;
3243	case IB_CM_APR_RECEIVED:
3244		ret = cm_apr_handler(work);
3245		break;
3246	case IB_CM_TIMEWAIT_EXIT:
3247		ret = cm_timewait_handler(work);
3248		break;
3249	default:
3250		ret = -EINVAL;
3251		break;
3252	}
3253	if (ret)
3254		cm_free_work(work);
3255}
3256
3257static int cm_establish(struct ib_cm_id *cm_id)
3258{
3259	struct cm_id_private *cm_id_priv;
3260	struct cm_work *work;
3261	unsigned long flags;
3262	int ret = 0;
3263
3264	work = kmalloc(sizeof *work, GFP_ATOMIC);
3265	if (!work)
3266		return -ENOMEM;
3267
3268	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3269	spin_lock_irqsave(&cm_id_priv->lock, flags);
3270	switch (cm_id->state)
3271	{
3272	case IB_CM_REP_SENT:
3273	case IB_CM_MRA_REP_RCVD:
3274		cm_id->state = IB_CM_ESTABLISHED;
3275		break;
3276	case IB_CM_ESTABLISHED:
3277		ret = -EISCONN;
3278		break;
3279	default:
3280		ret = -EINVAL;
3281		break;
3282	}
3283	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3284
3285	if (ret) {
3286		kfree(work);
3287		goto out;
3288	}
3289
3290	/*
3291	 * The CM worker thread may try to destroy the cm_id before it
3292	 * can execute this work item.  To prevent potential deadlock,
3293	 * we need to find the cm_id once we're in the context of the
3294	 * worker thread, rather than holding a reference on it.
3295	 */
3296	INIT_DELAYED_WORK(&work->work, cm_work_handler);
3297	work->local_id = cm_id->local_id;
3298	work->remote_id = cm_id->remote_id;
3299	work->mad_recv_wc = NULL;
3300	work->cm_event.event = IB_CM_USER_ESTABLISHED;
3301	queue_delayed_work(cm.wq, &work->work, 0);
3302out:
3303	return ret;
3304}
3305
3306static int cm_migrate(struct ib_cm_id *cm_id)
3307{
3308	struct cm_id_private *cm_id_priv;
3309	unsigned long flags;
3310	int ret = 0;
3311
3312	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3313	spin_lock_irqsave(&cm_id_priv->lock, flags);
3314	if (cm_id->state == IB_CM_ESTABLISHED &&
3315	    (cm_id->lap_state == IB_CM_LAP_UNINIT ||
3316	     cm_id->lap_state == IB_CM_LAP_IDLE)) {
3317		cm_id->lap_state = IB_CM_LAP_IDLE;
3318		cm_id_priv->av = cm_id_priv->alt_av;
3319	} else
3320		ret = -EINVAL;
3321	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3322
3323	return ret;
3324}
3325
3326int ib_cm_notify(struct ib_cm_id *cm_id, enum ib_event_type event)
3327{
3328	int ret;
3329
3330	switch (event) {
3331	case IB_EVENT_COMM_EST:
3332		ret = cm_establish(cm_id);
3333		break;
3334	case IB_EVENT_PATH_MIG:
3335		ret = cm_migrate(cm_id);
3336		break;
3337	default:
3338		ret = -EINVAL;
3339	}
3340	return ret;
3341}
3342EXPORT_SYMBOL(ib_cm_notify);
3343
3344static void cm_recv_handler(struct ib_mad_agent *mad_agent,
3345			    struct ib_mad_recv_wc *mad_recv_wc)
3346{
3347	struct cm_port *port = mad_agent->context;
3348	struct cm_work *work;
3349	enum ib_cm_event_type event;
3350	u16 attr_id;
3351	int paths = 0;
3352
3353	switch (mad_recv_wc->recv_buf.mad->mad_hdr.attr_id) {
3354	case CM_REQ_ATTR_ID:
3355		paths = 1 + (((struct cm_req_msg *) mad_recv_wc->recv_buf.mad)->
3356						    alt_local_lid != 0);
3357		event = IB_CM_REQ_RECEIVED;
3358		break;
3359	case CM_MRA_ATTR_ID:
3360		event = IB_CM_MRA_RECEIVED;
3361		break;
3362	case CM_REJ_ATTR_ID:
3363		event = IB_CM_REJ_RECEIVED;
3364		break;
3365	case CM_REP_ATTR_ID:
3366		event = IB_CM_REP_RECEIVED;
3367		break;
3368	case CM_RTU_ATTR_ID:
3369		event = IB_CM_RTU_RECEIVED;
3370		break;
3371	case CM_DREQ_ATTR_ID:
3372		event = IB_CM_DREQ_RECEIVED;
3373		break;
3374	case CM_DREP_ATTR_ID:
3375		event = IB_CM_DREP_RECEIVED;
3376		break;
3377	case CM_SIDR_REQ_ATTR_ID:
3378		event = IB_CM_SIDR_REQ_RECEIVED;
3379		break;
3380	case CM_SIDR_REP_ATTR_ID:
3381		event = IB_CM_SIDR_REP_RECEIVED;
3382		break;
3383	case CM_LAP_ATTR_ID:
3384		paths = 1;
3385		event = IB_CM_LAP_RECEIVED;
3386		break;
3387	case CM_APR_ATTR_ID:
3388		event = IB_CM_APR_RECEIVED;
3389		break;
3390	default:
3391		ib_free_recv_mad(mad_recv_wc);
3392		return;
3393	}
3394
3395	attr_id = be16_to_cpu(mad_recv_wc->recv_buf.mad->mad_hdr.attr_id);
3396	atomic_long_inc(&port->counter_group[CM_RECV].
3397			counter[attr_id - CM_ATTR_ID_OFFSET]);
3398
3399	work = kmalloc(sizeof *work + sizeof(struct ib_sa_path_rec) * paths,
3400		       GFP_KERNEL);
3401	if (!work) {
3402		ib_free_recv_mad(mad_recv_wc);
3403		return;
3404	}
3405
3406	INIT_DELAYED_WORK(&work->work, cm_work_handler);
3407	work->cm_event.event = event;
3408	work->mad_recv_wc = mad_recv_wc;
3409	work->port = port;
3410	queue_delayed_work(cm.wq, &work->work, 0);
3411}
3412
3413static int cm_init_qp_init_attr(struct cm_id_private *cm_id_priv,
3414				struct ib_qp_attr *qp_attr,
3415				int *qp_attr_mask)
3416{
3417	unsigned long flags;
3418	int ret;
3419
3420	spin_lock_irqsave(&cm_id_priv->lock, flags);
3421	switch (cm_id_priv->id.state) {
3422	case IB_CM_REQ_SENT:
3423	case IB_CM_MRA_REQ_RCVD:
3424	case IB_CM_REQ_RCVD:
3425	case IB_CM_MRA_REQ_SENT:
3426	case IB_CM_REP_RCVD:
3427	case IB_CM_MRA_REP_SENT:
3428	case IB_CM_REP_SENT:
3429	case IB_CM_MRA_REP_RCVD:
3430	case IB_CM_ESTABLISHED:
3431		*qp_attr_mask = IB_QP_STATE | IB_QP_ACCESS_FLAGS |
3432				IB_QP_PKEY_INDEX | IB_QP_PORT;
3433		qp_attr->qp_access_flags = IB_ACCESS_REMOTE_WRITE;
3434		if (cm_id_priv->responder_resources)
3435			qp_attr->qp_access_flags |= IB_ACCESS_REMOTE_READ |
3436						    IB_ACCESS_REMOTE_ATOMIC;
3437		qp_attr->pkey_index = cm_id_priv->av.pkey_index;
3438		qp_attr->port_num = cm_id_priv->av.port->port_num;
3439		ret = 0;
3440		break;
3441	default:
3442		ret = -EINVAL;
3443		break;
3444	}
3445	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3446	return ret;
3447}
3448
3449static int cm_init_qp_rtr_attr(struct cm_id_private *cm_id_priv,
3450			       struct ib_qp_attr *qp_attr,
3451			       int *qp_attr_mask)
3452{
3453	unsigned long flags;
3454	int ret;
3455
3456	spin_lock_irqsave(&cm_id_priv->lock, flags);
3457	switch (cm_id_priv->id.state) {
3458	case IB_CM_REQ_RCVD:
3459	case IB_CM_MRA_REQ_SENT:
3460	case IB_CM_REP_RCVD:
3461	case IB_CM_MRA_REP_SENT:
3462	case IB_CM_REP_SENT:
3463	case IB_CM_MRA_REP_RCVD:
3464	case IB_CM_ESTABLISHED:
3465		*qp_attr_mask = IB_QP_STATE | IB_QP_AV | IB_QP_PATH_MTU |
3466				IB_QP_DEST_QPN | IB_QP_RQ_PSN;
3467		qp_attr->ah_attr = cm_id_priv->av.ah_attr;
3468		qp_attr->path_mtu = cm_id_priv->path_mtu;
3469		qp_attr->dest_qp_num = be32_to_cpu(cm_id_priv->remote_qpn);
3470		qp_attr->rq_psn = be32_to_cpu(cm_id_priv->rq_psn);
3471		if (cm_id_priv->qp_type == IB_QPT_RC) {
3472			*qp_attr_mask |= IB_QP_MAX_DEST_RD_ATOMIC |
3473					 IB_QP_MIN_RNR_TIMER;
3474			qp_attr->max_dest_rd_atomic =
3475					cm_id_priv->responder_resources;
3476			qp_attr->min_rnr_timer = 0;
3477		}
3478		if (cm_id_priv->alt_av.ah_attr.dlid) {
3479			*qp_attr_mask |= IB_QP_ALT_PATH;
3480			qp_attr->alt_port_num = cm_id_priv->alt_av.port->port_num;
3481			qp_attr->alt_pkey_index = cm_id_priv->alt_av.pkey_index;
3482			qp_attr->alt_timeout = cm_id_priv->alt_av.timeout;
3483			qp_attr->alt_ah_attr = cm_id_priv->alt_av.ah_attr;
3484		}
3485		ret = 0;
3486		break;
3487	default:
3488		ret = -EINVAL;
3489		break;
3490	}
3491	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3492	return ret;
3493}
3494
3495static int cm_init_qp_rts_attr(struct cm_id_private *cm_id_priv,
3496			       struct ib_qp_attr *qp_attr,
3497			       int *qp_attr_mask)
3498{
3499	unsigned long flags;
3500	int ret;
3501
3502	spin_lock_irqsave(&cm_id_priv->lock, flags);
3503	switch (cm_id_priv->id.state) {
3504	/* Allow transition to RTS before sending REP */
3505	case IB_CM_REQ_RCVD:
3506	case IB_CM_MRA_REQ_SENT:
3507
3508	case IB_CM_REP_RCVD:
3509	case IB_CM_MRA_REP_SENT:
3510	case IB_CM_REP_SENT:
3511	case IB_CM_MRA_REP_RCVD:
3512	case IB_CM_ESTABLISHED:
3513		if (cm_id_priv->id.lap_state == IB_CM_LAP_UNINIT) {
3514			*qp_attr_mask = IB_QP_STATE | IB_QP_SQ_PSN;
3515			qp_attr->sq_psn = be32_to_cpu(cm_id_priv->sq_psn);
3516			if (cm_id_priv->qp_type == IB_QPT_RC) {
3517				*qp_attr_mask |= IB_QP_TIMEOUT | IB_QP_RETRY_CNT |
3518						 IB_QP_RNR_RETRY |
3519						 IB_QP_MAX_QP_RD_ATOMIC;
3520				qp_attr->timeout = cm_id_priv->av.timeout;
3521				qp_attr->retry_cnt = cm_id_priv->retry_count;
3522				qp_attr->rnr_retry = cm_id_priv->rnr_retry_count;
3523				qp_attr->max_rd_atomic =
3524					cm_id_priv->initiator_depth;
3525			}
3526			if (cm_id_priv->alt_av.ah_attr.dlid) {
3527				*qp_attr_mask |= IB_QP_PATH_MIG_STATE;
3528				qp_attr->path_mig_state = IB_MIG_REARM;
3529			}
3530		} else {
3531			*qp_attr_mask = IB_QP_ALT_PATH | IB_QP_PATH_MIG_STATE;
3532			qp_attr->alt_port_num = cm_id_priv->alt_av.port->port_num;
3533			qp_attr->alt_pkey_index = cm_id_priv->alt_av.pkey_index;
3534			qp_attr->alt_timeout = cm_id_priv->alt_av.timeout;
3535			qp_attr->alt_ah_attr = cm_id_priv->alt_av.ah_attr;
3536			qp_attr->path_mig_state = IB_MIG_REARM;
3537		}
3538		ret = 0;
3539		break;
3540	default:
3541		ret = -EINVAL;
3542		break;
3543	}
3544	spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3545	return ret;
3546}
3547
3548int ib_cm_init_qp_attr(struct ib_cm_id *cm_id,
3549		       struct ib_qp_attr *qp_attr,
3550		       int *qp_attr_mask)
3551{
3552	struct cm_id_private *cm_id_priv;
3553	int ret;
3554
3555	cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3556	switch (qp_attr->qp_state) {
3557	case IB_QPS_INIT:
3558		ret = cm_init_qp_init_attr(cm_id_priv, qp_attr, qp_attr_mask);
3559		break;
3560	case IB_QPS_RTR:
3561		ret = cm_init_qp_rtr_attr(cm_id_priv, qp_attr, qp_attr_mask);
3562		break;
3563	case IB_QPS_RTS:
3564		ret = cm_init_qp_rts_attr(cm_id_priv, qp_attr, qp_attr_mask);
3565		break;
3566	default:
3567		ret = -EINVAL;
3568		break;
3569	}
3570	return ret;
3571}
3572EXPORT_SYMBOL(ib_cm_init_qp_attr);
3573
3574static void cm_get_ack_delay(struct cm_device *cm_dev)
3575{
3576	struct ib_device_attr attr;
3577
3578	if (ib_query_device(cm_dev->ib_device, &attr))
3579		cm_dev->ack_delay = 0; /* acks will rely on packet life time */
3580	else
3581		cm_dev->ack_delay = attr.local_ca_ack_delay;
3582}
3583
3584static ssize_t cm_show_counter(struct kobject *obj, struct attribute *attr,
3585			       char *buf)
3586{
3587	struct cm_counter_group *group;
3588	struct cm_counter_attribute *cm_attr;
3589
3590	group = container_of(obj, struct cm_counter_group, obj);
3591	cm_attr = container_of(attr, struct cm_counter_attribute, attr);
3592
3593	return sprintf(buf, "%ld\n",
3594		       atomic_long_read(&group->counter[cm_attr->index]));
3595}
3596
3597static const struct sysfs_ops cm_counter_ops = {
3598	.show = cm_show_counter
3599};
3600
3601static struct kobj_type cm_counter_obj_type = {
3602	.sysfs_ops = &cm_counter_ops,
3603	.default_attrs = cm_counter_default_attrs
3604};
3605
3606static void cm_release_port_obj(struct kobject *obj)
3607{
3608	struct cm_port *cm_port;
3609
3610	cm_port = container_of(obj, struct cm_port, port_obj);
3611	kfree(cm_port);
3612}
3613
3614static struct kobj_type cm_port_obj_type = {
3615	.release = cm_release_port_obj
3616};
3617
3618struct class cm_class = {
3619	.name    = "infiniband_cm",
3620};
3621EXPORT_SYMBOL(cm_class);
3622
3623static int cm_create_port_fs(struct cm_port *port)
3624{
3625	int i, ret;
3626
3627	ret = kobject_init_and_add(&port->port_obj, &cm_port_obj_type,
3628				   &port->cm_dev->device->kobj,
3629				   "%d", port->port_num);
3630	if (ret) {
3631		kfree(port);
3632		return ret;
3633	}
3634
3635	for (i = 0; i < CM_COUNTER_GROUPS; i++) {
3636		ret = kobject_init_and_add(&port->counter_group[i].obj,
3637					   &cm_counter_obj_type,
3638					   &port->port_obj,
3639					   "%s", counter_group_names[i]);
3640		if (ret)
3641			goto error;
3642	}
3643
3644	return 0;
3645
3646error:
3647	while (i--)
3648		kobject_put(&port->counter_group[i].obj);
3649	kobject_put(&port->port_obj);
3650	return ret;
3651
3652}
3653
3654static void cm_remove_port_fs(struct cm_port *port)
3655{
3656	int i;
3657
3658	for (i = 0; i < CM_COUNTER_GROUPS; i++)
3659		kobject_put(&port->counter_group[i].obj);
3660
3661	kobject_put(&port->port_obj);
3662}
3663
3664static void cm_add_one(struct ib_device *ib_device)
3665{
3666	struct cm_device *cm_dev;
3667	struct cm_port *port;
3668	struct ib_mad_reg_req reg_req = {
3669		.mgmt_class = IB_MGMT_CLASS_CM,
3670		.mgmt_class_version = IB_CM_CLASS_VERSION
3671	};
3672	struct ib_port_modify port_modify = {
3673		.set_port_cap_mask = IB_PORT_CM_SUP
3674	};
3675	unsigned long flags;
3676	int ret;
3677	u8 i;
3678
3679	if (rdma_node_get_transport(ib_device->node_type) != RDMA_TRANSPORT_IB)
3680		return;
3681
3682	cm_dev = kzalloc(sizeof(*cm_dev) + sizeof(*port) *
3683			 ib_device->phys_port_cnt, GFP_KERNEL);
3684	if (!cm_dev)
3685		return;
3686
3687	cm_dev->ib_device = ib_device;
3688	cm_get_ack_delay(cm_dev);
3689
3690	cm_dev->device = device_create(&cm_class, &ib_device->dev,
3691				       MKDEV(0, 0), NULL,
3692				       "%s", ib_device->name);
3693	if (IS_ERR(cm_dev->device)) {
3694		kfree(cm_dev);
3695		return;
3696	}
3697
3698	set_bit(IB_MGMT_METHOD_SEND, reg_req.method_mask);
3699	for (i = 1; i <= ib_device->phys_port_cnt; i++) {
3700		port = kzalloc(sizeof *port, GFP_KERNEL);
3701		if (!port)
3702			goto error1;
3703
3704		cm_dev->port[i-1] = port;
3705		port->cm_dev = cm_dev;
3706		port->port_num = i;
3707
3708		ret = cm_create_port_fs(port);
3709		if (ret)
3710			goto error1;
3711
3712		port->mad_agent = ib_register_mad_agent(ib_device, i,
3713							IB_QPT_GSI,
3714							&reg_req,
3715							0,
3716							cm_send_handler,
3717							cm_recv_handler,
3718							port);
3719		if (IS_ERR(port->mad_agent))
3720			goto error2;
3721
3722		ret = ib_modify_port(ib_device, i, 0, &port_modify);
3723		if (ret)
3724			goto error3;
3725	}
3726	ib_set_client_data(ib_device, &cm_client, cm_dev);
3727
3728	write_lock_irqsave(&cm.device_lock, flags);
3729	list_add_tail(&cm_dev->list, &cm.device_list);
3730	write_unlock_irqrestore(&cm.device_lock, flags);
3731	return;
3732
3733error3:
3734	ib_unregister_mad_agent(port->mad_agent);
3735error2:
3736	cm_remove_port_fs(port);
3737error1:
3738	port_modify.set_port_cap_mask = 0;
3739	port_modify.clr_port_cap_mask = IB_PORT_CM_SUP;
3740	while (--i) {
3741		port = cm_dev->port[i-1];
3742		ib_modify_port(ib_device, port->port_num, 0, &port_modify);
3743		ib_unregister_mad_agent(port->mad_agent);
3744		cm_remove_port_fs(port);
3745	}
3746	device_unregister(cm_dev->device);
3747	kfree(cm_dev);
3748}
3749
3750static void cm_remove_one(struct ib_device *ib_device)
3751{
3752	struct cm_device *cm_dev;
3753	struct cm_port *port;
3754	struct ib_port_modify port_modify = {
3755		.clr_port_cap_mask = IB_PORT_CM_SUP
3756	};
3757	unsigned long flags;
3758	int i;
3759
3760	cm_dev = ib_get_client_data(ib_device, &cm_client);
3761	if (!cm_dev)
3762		return;
3763
3764	write_lock_irqsave(&cm.device_lock, flags);
3765	list_del(&cm_dev->list);
3766	write_unlock_irqrestore(&cm.device_lock, flags);
3767
3768	for (i = 1; i <= ib_device->phys_port_cnt; i++) {
3769		port = cm_dev->port[i-1];
3770		ib_modify_port(ib_device, port->port_num, 0, &port_modify);
3771		ib_unregister_mad_agent(port->mad_agent);
3772		flush_workqueue(cm.wq);
3773		cm_remove_port_fs(port);
3774	}
3775	device_unregister(cm_dev->device);
3776	kfree(cm_dev);
3777}
3778
3779static int __init ib_cm_init(void)
3780{
3781	int ret;
3782
3783	memset(&cm, 0, sizeof cm);
3784	INIT_LIST_HEAD(&cm.device_list);
3785	rwlock_init(&cm.device_lock);
3786	spin_lock_init(&cm.lock);
3787	cm.listen_service_table = RB_ROOT;
3788	cm.listen_service_id = be64_to_cpu(IB_CM_ASSIGN_SERVICE_ID);
3789	cm.remote_id_table = RB_ROOT;
3790	cm.remote_qp_table = RB_ROOT;
3791	cm.remote_sidr_table = RB_ROOT;
3792	idr_init(&cm.local_id_table);
3793	get_random_bytes(&cm.random_id_operand, sizeof cm.random_id_operand);
3794	idr_pre_get(&cm.local_id_table, GFP_KERNEL);
3795	INIT_LIST_HEAD(&cm.timewait_list);
3796
3797	ret = class_register(&cm_class);
3798	if (ret)
3799		return -ENOMEM;
3800
3801	cm.wq = create_workqueue("ib_cm");
3802	if (!cm.wq) {
3803		ret = -ENOMEM;
3804		goto error1;
3805	}
3806
3807	ret = ib_register_client(&cm_client);
3808	if (ret)
3809		goto error2;
3810
3811	return 0;
3812error2:
3813	destroy_workqueue(cm.wq);
3814error1:
3815	class_unregister(&cm_class);
3816	return ret;
3817}
3818
3819static void __exit ib_cm_cleanup(void)
3820{
3821	struct cm_timewait_info *timewait_info, *tmp;
3822
3823	spin_lock_irq(&cm.lock);
3824	list_for_each_entry(timewait_info, &cm.timewait_list, list)
3825		cancel_delayed_work(&timewait_info->work.work);
3826	spin_unlock_irq(&cm.lock);
3827
3828	ib_unregister_client(&cm_client);
3829	destroy_workqueue(cm.wq);
3830
3831	list_for_each_entry_safe(timewait_info, tmp, &cm.timewait_list, list) {
3832		list_del(&timewait_info->list);
3833		kfree(timewait_info);
3834	}
3835
3836	class_unregister(&cm_class);
3837	idr_destroy(&cm.local_id_table);
3838}
3839
3840module_init(ib_cm_init);
3841module_exit(ib_cm_cleanup);
3842