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