[ARM] Move FLUSH_BASE macros to asm/arch/memory.h
[linux-2.6/x86.git] / drivers / infiniband / core / cm.c
blob7cfedb8d9bcd963dc81a04ad11eac0322f30eee8
1 /*
2 * Copyright (c) 2004, 2005 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.
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:
13 * Redistribution and use in source and binary forms, with or
14 * without modification, are permitted provided that the following
15 * conditions are met:
17 * - Redistributions of source code must retain the above
18 * copyright notice, this list of conditions and the following
19 * disclaimer.
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.
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.
35 * $Id: cm.c 2821 2005-07-08 17:07:28Z sean.hefty $
37 #include <linux/dma-mapping.h>
38 #include <linux/err.h>
39 #include <linux/idr.h>
40 #include <linux/interrupt.h>
41 #include <linux/pci.h>
42 #include <linux/rbtree.h>
43 #include <linux/spinlock.h>
44 #include <linux/workqueue.h>
46 #include <rdma/ib_cache.h>
47 #include <rdma/ib_cm.h>
48 #include "cm_msgs.h"
50 MODULE_AUTHOR("Sean Hefty");
51 MODULE_DESCRIPTION("InfiniBand CM");
52 MODULE_LICENSE("Dual BSD/GPL");
54 static void cm_add_one(struct ib_device *device);
55 static void cm_remove_one(struct ib_device *device);
57 static struct ib_client cm_client = {
58 .name = "cm",
59 .add = cm_add_one,
60 .remove = cm_remove_one
63 static struct ib_cm {
64 spinlock_t lock;
65 struct list_head device_list;
66 rwlock_t device_lock;
67 struct rb_root listen_service_table;
68 u64 listen_service_id;
69 /* struct rb_root peer_service_table; todo: fix peer to peer */
70 struct rb_root remote_qp_table;
71 struct rb_root remote_id_table;
72 struct rb_root remote_sidr_table;
73 struct idr local_id_table;
74 struct workqueue_struct *wq;
75 } cm;
77 struct cm_port {
78 struct cm_device *cm_dev;
79 struct ib_mad_agent *mad_agent;
80 u8 port_num;
83 struct cm_device {
84 struct list_head list;
85 struct ib_device *device;
86 __be64 ca_guid;
87 struct cm_port port[0];
90 struct cm_av {
91 struct cm_port *port;
92 union ib_gid dgid;
93 struct ib_ah_attr ah_attr;
94 u16 pkey_index;
95 u8 packet_life_time;
98 struct cm_work {
99 struct work_struct work;
100 struct list_head list;
101 struct cm_port *port;
102 struct ib_mad_recv_wc *mad_recv_wc; /* Received MADs */
103 __be32 local_id; /* Established / timewait */
104 __be32 remote_id;
105 struct ib_cm_event cm_event;
106 struct ib_sa_path_rec path[0];
109 struct cm_timewait_info {
110 struct cm_work work; /* Must be first. */
111 struct rb_node remote_qp_node;
112 struct rb_node remote_id_node;
113 __be64 remote_ca_guid;
114 __be32 remote_qpn;
115 u8 inserted_remote_qp;
116 u8 inserted_remote_id;
119 struct cm_id_private {
120 struct ib_cm_id id;
122 struct rb_node service_node;
123 struct rb_node sidr_id_node;
124 spinlock_t lock; /* Do not acquire inside cm.lock */
125 wait_queue_head_t wait;
126 atomic_t refcount;
128 struct ib_mad_send_buf *msg;
129 struct cm_timewait_info *timewait_info;
130 /* todo: use alternate port on send failure */
131 struct cm_av av;
132 struct cm_av alt_av;
134 void *private_data;
135 __be64 tid;
136 __be32 local_qpn;
137 __be32 remote_qpn;
138 enum ib_qp_type qp_type;
139 __be32 sq_psn;
140 __be32 rq_psn;
141 int timeout_ms;
142 enum ib_mtu path_mtu;
143 u8 private_data_len;
144 u8 max_cm_retries;
145 u8 peer_to_peer;
146 u8 responder_resources;
147 u8 initiator_depth;
148 u8 local_ack_timeout;
149 u8 retry_count;
150 u8 rnr_retry_count;
151 u8 service_timeout;
153 struct list_head work_list;
154 atomic_t work_count;
157 static void cm_work_handler(void *data);
159 static inline void cm_deref_id(struct cm_id_private *cm_id_priv)
161 if (atomic_dec_and_test(&cm_id_priv->refcount))
162 wake_up(&cm_id_priv->wait);
165 static int cm_alloc_msg(struct cm_id_private *cm_id_priv,
166 struct ib_mad_send_buf **msg)
168 struct ib_mad_agent *mad_agent;
169 struct ib_mad_send_buf *m;
170 struct ib_ah *ah;
172 mad_agent = cm_id_priv->av.port->mad_agent;
173 ah = ib_create_ah(mad_agent->qp->pd, &cm_id_priv->av.ah_attr);
174 if (IS_ERR(ah))
175 return PTR_ERR(ah);
177 m = ib_create_send_mad(mad_agent, cm_id_priv->id.remote_cm_qpn,
178 cm_id_priv->av.pkey_index,
179 0, IB_MGMT_MAD_HDR, IB_MGMT_MAD_DATA,
180 GFP_ATOMIC);
181 if (IS_ERR(m)) {
182 ib_destroy_ah(ah);
183 return PTR_ERR(m);
186 /* Timeout set by caller if response is expected. */
187 m->ah = ah;
188 m->retries = cm_id_priv->max_cm_retries;
190 atomic_inc(&cm_id_priv->refcount);
191 m->context[0] = cm_id_priv;
192 *msg = m;
193 return 0;
196 static int cm_alloc_response_msg(struct cm_port *port,
197 struct ib_mad_recv_wc *mad_recv_wc,
198 struct ib_mad_send_buf **msg)
200 struct ib_mad_send_buf *m;
201 struct ib_ah *ah;
203 ah = ib_create_ah_from_wc(port->mad_agent->qp->pd, mad_recv_wc->wc,
204 mad_recv_wc->recv_buf.grh, port->port_num);
205 if (IS_ERR(ah))
206 return PTR_ERR(ah);
208 m = ib_create_send_mad(port->mad_agent, 1, mad_recv_wc->wc->pkey_index,
209 0, IB_MGMT_MAD_HDR, IB_MGMT_MAD_DATA,
210 GFP_ATOMIC);
211 if (IS_ERR(m)) {
212 ib_destroy_ah(ah);
213 return PTR_ERR(m);
215 m->ah = ah;
216 *msg = m;
217 return 0;
220 static void cm_free_msg(struct ib_mad_send_buf *msg)
222 ib_destroy_ah(msg->ah);
223 if (msg->context[0])
224 cm_deref_id(msg->context[0]);
225 ib_free_send_mad(msg);
228 static void * cm_copy_private_data(const void *private_data,
229 u8 private_data_len)
231 void *data;
233 if (!private_data || !private_data_len)
234 return NULL;
236 data = kmalloc(private_data_len, GFP_KERNEL);
237 if (!data)
238 return ERR_PTR(-ENOMEM);
240 memcpy(data, private_data, private_data_len);
241 return data;
244 static void cm_set_private_data(struct cm_id_private *cm_id_priv,
245 void *private_data, u8 private_data_len)
247 if (cm_id_priv->private_data && cm_id_priv->private_data_len)
248 kfree(cm_id_priv->private_data);
250 cm_id_priv->private_data = private_data;
251 cm_id_priv->private_data_len = private_data_len;
254 static void cm_set_ah_attr(struct ib_ah_attr *ah_attr, u8 port_num,
255 u16 dlid, u8 sl, u16 src_path_bits)
257 memset(ah_attr, 0, sizeof ah_attr);
258 ah_attr->dlid = dlid;
259 ah_attr->sl = sl;
260 ah_attr->src_path_bits = src_path_bits;
261 ah_attr->port_num = port_num;
264 static void cm_init_av_for_response(struct cm_port *port,
265 struct ib_wc *wc, struct cm_av *av)
267 av->port = port;
268 av->pkey_index = wc->pkey_index;
269 cm_set_ah_attr(&av->ah_attr, port->port_num, wc->slid,
270 wc->sl, wc->dlid_path_bits);
273 static int cm_init_av_by_path(struct ib_sa_path_rec *path, struct cm_av *av)
275 struct cm_device *cm_dev;
276 struct cm_port *port = NULL;
277 unsigned long flags;
278 int ret;
279 u8 p;
281 read_lock_irqsave(&cm.device_lock, flags);
282 list_for_each_entry(cm_dev, &cm.device_list, list) {
283 if (!ib_find_cached_gid(cm_dev->device, &path->sgid,
284 &p, NULL)) {
285 port = &cm_dev->port[p-1];
286 break;
289 read_unlock_irqrestore(&cm.device_lock, flags);
291 if (!port)
292 return -EINVAL;
294 ret = ib_find_cached_pkey(cm_dev->device, port->port_num,
295 be16_to_cpu(path->pkey), &av->pkey_index);
296 if (ret)
297 return ret;
299 av->port = port;
300 cm_set_ah_attr(&av->ah_attr, av->port->port_num,
301 be16_to_cpu(path->dlid), path->sl,
302 be16_to_cpu(path->slid) & 0x7F);
303 av->packet_life_time = path->packet_life_time;
304 return 0;
307 static int cm_alloc_id(struct cm_id_private *cm_id_priv)
309 unsigned long flags;
310 int ret;
311 static int next_id;
313 do {
314 spin_lock_irqsave(&cm.lock, flags);
315 ret = idr_get_new_above(&cm.local_id_table, cm_id_priv, next_id++,
316 (__force int *) &cm_id_priv->id.local_id);
317 spin_unlock_irqrestore(&cm.lock, flags);
318 } while( (ret == -EAGAIN) && idr_pre_get(&cm.local_id_table, GFP_KERNEL) );
319 return ret;
322 static void cm_free_id(__be32 local_id)
324 unsigned long flags;
326 spin_lock_irqsave(&cm.lock, flags);
327 idr_remove(&cm.local_id_table, (__force int) local_id);
328 spin_unlock_irqrestore(&cm.lock, flags);
331 static struct cm_id_private * cm_get_id(__be32 local_id, __be32 remote_id)
333 struct cm_id_private *cm_id_priv;
335 cm_id_priv = idr_find(&cm.local_id_table, (__force int) local_id);
336 if (cm_id_priv) {
337 if (cm_id_priv->id.remote_id == remote_id)
338 atomic_inc(&cm_id_priv->refcount);
339 else
340 cm_id_priv = NULL;
343 return cm_id_priv;
346 static struct cm_id_private * cm_acquire_id(__be32 local_id, __be32 remote_id)
348 struct cm_id_private *cm_id_priv;
349 unsigned long flags;
351 spin_lock_irqsave(&cm.lock, flags);
352 cm_id_priv = cm_get_id(local_id, remote_id);
353 spin_unlock_irqrestore(&cm.lock, flags);
355 return cm_id_priv;
358 static struct cm_id_private * cm_insert_listen(struct cm_id_private *cm_id_priv)
360 struct rb_node **link = &cm.listen_service_table.rb_node;
361 struct rb_node *parent = NULL;
362 struct cm_id_private *cur_cm_id_priv;
363 __be64 service_id = cm_id_priv->id.service_id;
364 __be64 service_mask = cm_id_priv->id.service_mask;
366 while (*link) {
367 parent = *link;
368 cur_cm_id_priv = rb_entry(parent, struct cm_id_private,
369 service_node);
370 if ((cur_cm_id_priv->id.service_mask & service_id) ==
371 (service_mask & cur_cm_id_priv->id.service_id) &&
372 (cm_id_priv->id.device == cur_cm_id_priv->id.device))
373 return cur_cm_id_priv;
375 if (cm_id_priv->id.device < cur_cm_id_priv->id.device)
376 link = &(*link)->rb_left;
377 else if (cm_id_priv->id.device > cur_cm_id_priv->id.device)
378 link = &(*link)->rb_right;
379 else if (service_id < cur_cm_id_priv->id.service_id)
380 link = &(*link)->rb_left;
381 else
382 link = &(*link)->rb_right;
384 rb_link_node(&cm_id_priv->service_node, parent, link);
385 rb_insert_color(&cm_id_priv->service_node, &cm.listen_service_table);
386 return NULL;
389 static struct cm_id_private * cm_find_listen(struct ib_device *device,
390 __be64 service_id)
392 struct rb_node *node = cm.listen_service_table.rb_node;
393 struct cm_id_private *cm_id_priv;
395 while (node) {
396 cm_id_priv = rb_entry(node, struct cm_id_private, service_node);
397 if ((cm_id_priv->id.service_mask & service_id) ==
398 cm_id_priv->id.service_id &&
399 (cm_id_priv->id.device == device))
400 return cm_id_priv;
402 if (device < cm_id_priv->id.device)
403 node = node->rb_left;
404 else if (device > cm_id_priv->id.device)
405 node = node->rb_right;
406 else if (service_id < cm_id_priv->id.service_id)
407 node = node->rb_left;
408 else
409 node = node->rb_right;
411 return NULL;
414 static struct cm_timewait_info * cm_insert_remote_id(struct cm_timewait_info
415 *timewait_info)
417 struct rb_node **link = &cm.remote_id_table.rb_node;
418 struct rb_node *parent = NULL;
419 struct cm_timewait_info *cur_timewait_info;
420 __be64 remote_ca_guid = timewait_info->remote_ca_guid;
421 __be32 remote_id = timewait_info->work.remote_id;
423 while (*link) {
424 parent = *link;
425 cur_timewait_info = rb_entry(parent, struct cm_timewait_info,
426 remote_id_node);
427 if (remote_id < cur_timewait_info->work.remote_id)
428 link = &(*link)->rb_left;
429 else if (remote_id > cur_timewait_info->work.remote_id)
430 link = &(*link)->rb_right;
431 else if (remote_ca_guid < cur_timewait_info->remote_ca_guid)
432 link = &(*link)->rb_left;
433 else if (remote_ca_guid > cur_timewait_info->remote_ca_guid)
434 link = &(*link)->rb_right;
435 else
436 return cur_timewait_info;
438 timewait_info->inserted_remote_id = 1;
439 rb_link_node(&timewait_info->remote_id_node, parent, link);
440 rb_insert_color(&timewait_info->remote_id_node, &cm.remote_id_table);
441 return NULL;
444 static struct cm_timewait_info * cm_find_remote_id(__be64 remote_ca_guid,
445 __be32 remote_id)
447 struct rb_node *node = cm.remote_id_table.rb_node;
448 struct cm_timewait_info *timewait_info;
450 while (node) {
451 timewait_info = rb_entry(node, struct cm_timewait_info,
452 remote_id_node);
453 if (remote_id < timewait_info->work.remote_id)
454 node = node->rb_left;
455 else if (remote_id > timewait_info->work.remote_id)
456 node = node->rb_right;
457 else if (remote_ca_guid < timewait_info->remote_ca_guid)
458 node = node->rb_left;
459 else if (remote_ca_guid > timewait_info->remote_ca_guid)
460 node = node->rb_right;
461 else
462 return timewait_info;
464 return NULL;
467 static struct cm_timewait_info * cm_insert_remote_qpn(struct cm_timewait_info
468 *timewait_info)
470 struct rb_node **link = &cm.remote_qp_table.rb_node;
471 struct rb_node *parent = NULL;
472 struct cm_timewait_info *cur_timewait_info;
473 __be64 remote_ca_guid = timewait_info->remote_ca_guid;
474 __be32 remote_qpn = timewait_info->remote_qpn;
476 while (*link) {
477 parent = *link;
478 cur_timewait_info = rb_entry(parent, struct cm_timewait_info,
479 remote_qp_node);
480 if (remote_qpn < cur_timewait_info->remote_qpn)
481 link = &(*link)->rb_left;
482 else if (remote_qpn > cur_timewait_info->remote_qpn)
483 link = &(*link)->rb_right;
484 else if (remote_ca_guid < cur_timewait_info->remote_ca_guid)
485 link = &(*link)->rb_left;
486 else if (remote_ca_guid > cur_timewait_info->remote_ca_guid)
487 link = &(*link)->rb_right;
488 else
489 return cur_timewait_info;
491 timewait_info->inserted_remote_qp = 1;
492 rb_link_node(&timewait_info->remote_qp_node, parent, link);
493 rb_insert_color(&timewait_info->remote_qp_node, &cm.remote_qp_table);
494 return NULL;
497 static struct cm_id_private * cm_insert_remote_sidr(struct cm_id_private
498 *cm_id_priv)
500 struct rb_node **link = &cm.remote_sidr_table.rb_node;
501 struct rb_node *parent = NULL;
502 struct cm_id_private *cur_cm_id_priv;
503 union ib_gid *port_gid = &cm_id_priv->av.dgid;
504 __be32 remote_id = cm_id_priv->id.remote_id;
506 while (*link) {
507 parent = *link;
508 cur_cm_id_priv = rb_entry(parent, struct cm_id_private,
509 sidr_id_node);
510 if (remote_id < cur_cm_id_priv->id.remote_id)
511 link = &(*link)->rb_left;
512 else if (remote_id > cur_cm_id_priv->id.remote_id)
513 link = &(*link)->rb_right;
514 else {
515 int cmp;
516 cmp = memcmp(port_gid, &cur_cm_id_priv->av.dgid,
517 sizeof *port_gid);
518 if (cmp < 0)
519 link = &(*link)->rb_left;
520 else if (cmp > 0)
521 link = &(*link)->rb_right;
522 else
523 return cur_cm_id_priv;
526 rb_link_node(&cm_id_priv->sidr_id_node, parent, link);
527 rb_insert_color(&cm_id_priv->sidr_id_node, &cm.remote_sidr_table);
528 return NULL;
531 static void cm_reject_sidr_req(struct cm_id_private *cm_id_priv,
532 enum ib_cm_sidr_status status)
534 struct ib_cm_sidr_rep_param param;
536 memset(&param, 0, sizeof param);
537 param.status = status;
538 ib_send_cm_sidr_rep(&cm_id_priv->id, &param);
541 struct ib_cm_id *ib_create_cm_id(struct ib_device *device,
542 ib_cm_handler cm_handler,
543 void *context)
545 struct cm_id_private *cm_id_priv;
546 int ret;
548 cm_id_priv = kzalloc(sizeof *cm_id_priv, GFP_KERNEL);
549 if (!cm_id_priv)
550 return ERR_PTR(-ENOMEM);
552 cm_id_priv->id.state = IB_CM_IDLE;
553 cm_id_priv->id.device = device;
554 cm_id_priv->id.cm_handler = cm_handler;
555 cm_id_priv->id.context = context;
556 cm_id_priv->id.remote_cm_qpn = 1;
557 ret = cm_alloc_id(cm_id_priv);
558 if (ret)
559 goto error;
561 spin_lock_init(&cm_id_priv->lock);
562 init_waitqueue_head(&cm_id_priv->wait);
563 INIT_LIST_HEAD(&cm_id_priv->work_list);
564 atomic_set(&cm_id_priv->work_count, -1);
565 atomic_set(&cm_id_priv->refcount, 1);
566 return &cm_id_priv->id;
568 error:
569 kfree(cm_id_priv);
570 return ERR_PTR(-ENOMEM);
572 EXPORT_SYMBOL(ib_create_cm_id);
574 static struct cm_work * cm_dequeue_work(struct cm_id_private *cm_id_priv)
576 struct cm_work *work;
578 if (list_empty(&cm_id_priv->work_list))
579 return NULL;
581 work = list_entry(cm_id_priv->work_list.next, struct cm_work, list);
582 list_del(&work->list);
583 return work;
586 static void cm_free_work(struct cm_work *work)
588 if (work->mad_recv_wc)
589 ib_free_recv_mad(work->mad_recv_wc);
590 kfree(work);
593 static inline int cm_convert_to_ms(int iba_time)
595 /* approximate conversion to ms from 4.096us x 2^iba_time */
596 return 1 << max(iba_time - 8, 0);
599 static void cm_cleanup_timewait(struct cm_timewait_info *timewait_info)
601 unsigned long flags;
603 if (!timewait_info->inserted_remote_id &&
604 !timewait_info->inserted_remote_qp)
605 return;
607 spin_lock_irqsave(&cm.lock, flags);
608 if (timewait_info->inserted_remote_id) {
609 rb_erase(&timewait_info->remote_id_node, &cm.remote_id_table);
610 timewait_info->inserted_remote_id = 0;
613 if (timewait_info->inserted_remote_qp) {
614 rb_erase(&timewait_info->remote_qp_node, &cm.remote_qp_table);
615 timewait_info->inserted_remote_qp = 0;
617 spin_unlock_irqrestore(&cm.lock, flags);
620 static struct cm_timewait_info * cm_create_timewait_info(__be32 local_id)
622 struct cm_timewait_info *timewait_info;
624 timewait_info = kzalloc(sizeof *timewait_info, GFP_KERNEL);
625 if (!timewait_info)
626 return ERR_PTR(-ENOMEM);
628 timewait_info->work.local_id = local_id;
629 INIT_WORK(&timewait_info->work.work, cm_work_handler,
630 &timewait_info->work);
631 timewait_info->work.cm_event.event = IB_CM_TIMEWAIT_EXIT;
632 return timewait_info;
635 static void cm_enter_timewait(struct cm_id_private *cm_id_priv)
637 int wait_time;
640 * The cm_id could be destroyed by the user before we exit timewait.
641 * To protect against this, we search for the cm_id after exiting
642 * timewait before notifying the user that we've exited timewait.
644 cm_id_priv->id.state = IB_CM_TIMEWAIT;
645 wait_time = cm_convert_to_ms(cm_id_priv->local_ack_timeout);
646 queue_delayed_work(cm.wq, &cm_id_priv->timewait_info->work.work,
647 msecs_to_jiffies(wait_time));
648 cm_id_priv->timewait_info = NULL;
651 static void cm_reset_to_idle(struct cm_id_private *cm_id_priv)
653 cm_id_priv->id.state = IB_CM_IDLE;
654 if (cm_id_priv->timewait_info) {
655 cm_cleanup_timewait(cm_id_priv->timewait_info);
656 kfree(cm_id_priv->timewait_info);
657 cm_id_priv->timewait_info = NULL;
661 void ib_destroy_cm_id(struct ib_cm_id *cm_id)
663 struct cm_id_private *cm_id_priv;
664 struct cm_work *work;
665 unsigned long flags;
667 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
668 retest:
669 spin_lock_irqsave(&cm_id_priv->lock, flags);
670 switch (cm_id->state) {
671 case IB_CM_LISTEN:
672 cm_id->state = IB_CM_IDLE;
673 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
674 spin_lock_irqsave(&cm.lock, flags);
675 rb_erase(&cm_id_priv->service_node, &cm.listen_service_table);
676 spin_unlock_irqrestore(&cm.lock, flags);
677 break;
678 case IB_CM_SIDR_REQ_SENT:
679 cm_id->state = IB_CM_IDLE;
680 ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
681 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
682 break;
683 case IB_CM_SIDR_REQ_RCVD:
684 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
685 cm_reject_sidr_req(cm_id_priv, IB_SIDR_REJECT);
686 break;
687 case IB_CM_REQ_SENT:
688 ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
689 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
690 ib_send_cm_rej(cm_id, IB_CM_REJ_TIMEOUT,
691 &cm_id_priv->av.port->cm_dev->ca_guid,
692 sizeof cm_id_priv->av.port->cm_dev->ca_guid,
693 NULL, 0);
694 break;
695 case IB_CM_MRA_REQ_RCVD:
696 case IB_CM_REP_SENT:
697 case IB_CM_MRA_REP_RCVD:
698 ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
699 /* Fall through */
700 case IB_CM_REQ_RCVD:
701 case IB_CM_MRA_REQ_SENT:
702 case IB_CM_REP_RCVD:
703 case IB_CM_MRA_REP_SENT:
704 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
705 ib_send_cm_rej(cm_id, IB_CM_REJ_CONSUMER_DEFINED,
706 NULL, 0, NULL, 0);
707 break;
708 case IB_CM_ESTABLISHED:
709 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
710 ib_send_cm_dreq(cm_id, NULL, 0);
711 goto retest;
712 case IB_CM_DREQ_SENT:
713 ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
714 cm_enter_timewait(cm_id_priv);
715 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
716 break;
717 case IB_CM_DREQ_RCVD:
718 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
719 ib_send_cm_drep(cm_id, NULL, 0);
720 break;
721 default:
722 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
723 break;
726 cm_free_id(cm_id->local_id);
727 atomic_dec(&cm_id_priv->refcount);
728 wait_event(cm_id_priv->wait, !atomic_read(&cm_id_priv->refcount));
729 while ((work = cm_dequeue_work(cm_id_priv)) != NULL)
730 cm_free_work(work);
731 if (cm_id_priv->private_data && cm_id_priv->private_data_len)
732 kfree(cm_id_priv->private_data);
733 kfree(cm_id_priv);
735 EXPORT_SYMBOL(ib_destroy_cm_id);
737 int ib_cm_listen(struct ib_cm_id *cm_id,
738 __be64 service_id,
739 __be64 service_mask)
741 struct cm_id_private *cm_id_priv, *cur_cm_id_priv;
742 unsigned long flags;
743 int ret = 0;
745 service_mask = service_mask ? service_mask :
746 __constant_cpu_to_be64(~0ULL);
747 service_id &= service_mask;
748 if ((service_id & IB_SERVICE_ID_AGN_MASK) == IB_CM_ASSIGN_SERVICE_ID &&
749 (service_id != IB_CM_ASSIGN_SERVICE_ID))
750 return -EINVAL;
752 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
753 BUG_ON(cm_id->state != IB_CM_IDLE);
755 cm_id->state = IB_CM_LISTEN;
757 spin_lock_irqsave(&cm.lock, flags);
758 if (service_id == IB_CM_ASSIGN_SERVICE_ID) {
759 cm_id->service_id = cpu_to_be64(cm.listen_service_id++);
760 cm_id->service_mask = __constant_cpu_to_be64(~0ULL);
761 } else {
762 cm_id->service_id = service_id;
763 cm_id->service_mask = service_mask;
765 cur_cm_id_priv = cm_insert_listen(cm_id_priv);
766 spin_unlock_irqrestore(&cm.lock, flags);
768 if (cur_cm_id_priv) {
769 cm_id->state = IB_CM_IDLE;
770 ret = -EBUSY;
772 return ret;
774 EXPORT_SYMBOL(ib_cm_listen);
776 static __be64 cm_form_tid(struct cm_id_private *cm_id_priv,
777 enum cm_msg_sequence msg_seq)
779 u64 hi_tid, low_tid;
781 hi_tid = ((u64) cm_id_priv->av.port->mad_agent->hi_tid) << 32;
782 low_tid = (u64) ((__force u32)cm_id_priv->id.local_id |
783 (msg_seq << 30));
784 return cpu_to_be64(hi_tid | low_tid);
787 static void cm_format_mad_hdr(struct ib_mad_hdr *hdr,
788 __be16 attr_id, __be64 tid)
790 hdr->base_version = IB_MGMT_BASE_VERSION;
791 hdr->mgmt_class = IB_MGMT_CLASS_CM;
792 hdr->class_version = IB_CM_CLASS_VERSION;
793 hdr->method = IB_MGMT_METHOD_SEND;
794 hdr->attr_id = attr_id;
795 hdr->tid = tid;
798 static void cm_format_req(struct cm_req_msg *req_msg,
799 struct cm_id_private *cm_id_priv,
800 struct ib_cm_req_param *param)
802 cm_format_mad_hdr(&req_msg->hdr, CM_REQ_ATTR_ID,
803 cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_REQ));
805 req_msg->local_comm_id = cm_id_priv->id.local_id;
806 req_msg->service_id = param->service_id;
807 req_msg->local_ca_guid = cm_id_priv->av.port->cm_dev->ca_guid;
808 cm_req_set_local_qpn(req_msg, cpu_to_be32(param->qp_num));
809 cm_req_set_resp_res(req_msg, param->responder_resources);
810 cm_req_set_init_depth(req_msg, param->initiator_depth);
811 cm_req_set_remote_resp_timeout(req_msg,
812 param->remote_cm_response_timeout);
813 cm_req_set_qp_type(req_msg, param->qp_type);
814 cm_req_set_flow_ctrl(req_msg, param->flow_control);
815 cm_req_set_starting_psn(req_msg, cpu_to_be32(param->starting_psn));
816 cm_req_set_local_resp_timeout(req_msg,
817 param->local_cm_response_timeout);
818 cm_req_set_retry_count(req_msg, param->retry_count);
819 req_msg->pkey = param->primary_path->pkey;
820 cm_req_set_path_mtu(req_msg, param->primary_path->mtu);
821 cm_req_set_rnr_retry_count(req_msg, param->rnr_retry_count);
822 cm_req_set_max_cm_retries(req_msg, param->max_cm_retries);
823 cm_req_set_srq(req_msg, param->srq);
825 req_msg->primary_local_lid = param->primary_path->slid;
826 req_msg->primary_remote_lid = param->primary_path->dlid;
827 req_msg->primary_local_gid = param->primary_path->sgid;
828 req_msg->primary_remote_gid = param->primary_path->dgid;
829 cm_req_set_primary_flow_label(req_msg, param->primary_path->flow_label);
830 cm_req_set_primary_packet_rate(req_msg, param->primary_path->rate);
831 req_msg->primary_traffic_class = param->primary_path->traffic_class;
832 req_msg->primary_hop_limit = param->primary_path->hop_limit;
833 cm_req_set_primary_sl(req_msg, param->primary_path->sl);
834 cm_req_set_primary_subnet_local(req_msg, 1); /* local only... */
835 cm_req_set_primary_local_ack_timeout(req_msg,
836 min(31, param->primary_path->packet_life_time + 1));
838 if (param->alternate_path) {
839 req_msg->alt_local_lid = param->alternate_path->slid;
840 req_msg->alt_remote_lid = param->alternate_path->dlid;
841 req_msg->alt_local_gid = param->alternate_path->sgid;
842 req_msg->alt_remote_gid = param->alternate_path->dgid;
843 cm_req_set_alt_flow_label(req_msg,
844 param->alternate_path->flow_label);
845 cm_req_set_alt_packet_rate(req_msg, param->alternate_path->rate);
846 req_msg->alt_traffic_class = param->alternate_path->traffic_class;
847 req_msg->alt_hop_limit = param->alternate_path->hop_limit;
848 cm_req_set_alt_sl(req_msg, param->alternate_path->sl);
849 cm_req_set_alt_subnet_local(req_msg, 1); /* local only... */
850 cm_req_set_alt_local_ack_timeout(req_msg,
851 min(31, param->alternate_path->packet_life_time + 1));
854 if (param->private_data && param->private_data_len)
855 memcpy(req_msg->private_data, param->private_data,
856 param->private_data_len);
859 static int cm_validate_req_param(struct ib_cm_req_param *param)
861 /* peer-to-peer not supported */
862 if (param->peer_to_peer)
863 return -EINVAL;
865 if (!param->primary_path)
866 return -EINVAL;
868 if (param->qp_type != IB_QPT_RC && param->qp_type != IB_QPT_UC)
869 return -EINVAL;
871 if (param->private_data &&
872 param->private_data_len > IB_CM_REQ_PRIVATE_DATA_SIZE)
873 return -EINVAL;
875 if (param->alternate_path &&
876 (param->alternate_path->pkey != param->primary_path->pkey ||
877 param->alternate_path->mtu != param->primary_path->mtu))
878 return -EINVAL;
880 return 0;
883 int ib_send_cm_req(struct ib_cm_id *cm_id,
884 struct ib_cm_req_param *param)
886 struct cm_id_private *cm_id_priv;
887 struct cm_req_msg *req_msg;
888 unsigned long flags;
889 int ret;
891 ret = cm_validate_req_param(param);
892 if (ret)
893 return ret;
895 /* Verify that we're not in timewait. */
896 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
897 spin_lock_irqsave(&cm_id_priv->lock, flags);
898 if (cm_id->state != IB_CM_IDLE) {
899 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
900 ret = -EINVAL;
901 goto out;
903 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
905 cm_id_priv->timewait_info = cm_create_timewait_info(cm_id_priv->
906 id.local_id);
907 if (IS_ERR(cm_id_priv->timewait_info))
908 goto out;
910 ret = cm_init_av_by_path(param->primary_path, &cm_id_priv->av);
911 if (ret)
912 goto error1;
913 if (param->alternate_path) {
914 ret = cm_init_av_by_path(param->alternate_path,
915 &cm_id_priv->alt_av);
916 if (ret)
917 goto error1;
919 cm_id->service_id = param->service_id;
920 cm_id->service_mask = __constant_cpu_to_be64(~0ULL);
921 cm_id_priv->timeout_ms = cm_convert_to_ms(
922 param->primary_path->packet_life_time) * 2 +
923 cm_convert_to_ms(
924 param->remote_cm_response_timeout);
925 cm_id_priv->max_cm_retries = param->max_cm_retries;
926 cm_id_priv->initiator_depth = param->initiator_depth;
927 cm_id_priv->responder_resources = param->responder_resources;
928 cm_id_priv->retry_count = param->retry_count;
929 cm_id_priv->path_mtu = param->primary_path->mtu;
930 cm_id_priv->qp_type = param->qp_type;
932 ret = cm_alloc_msg(cm_id_priv, &cm_id_priv->msg);
933 if (ret)
934 goto error1;
936 req_msg = (struct cm_req_msg *) cm_id_priv->msg->mad;
937 cm_format_req(req_msg, cm_id_priv, param);
938 cm_id_priv->tid = req_msg->hdr.tid;
939 cm_id_priv->msg->timeout_ms = cm_id_priv->timeout_ms;
940 cm_id_priv->msg->context[1] = (void *) (unsigned long) IB_CM_REQ_SENT;
942 cm_id_priv->local_qpn = cm_req_get_local_qpn(req_msg);
943 cm_id_priv->rq_psn = cm_req_get_starting_psn(req_msg);
944 cm_id_priv->local_ack_timeout =
945 cm_req_get_primary_local_ack_timeout(req_msg);
947 spin_lock_irqsave(&cm_id_priv->lock, flags);
948 ret = ib_post_send_mad(cm_id_priv->msg, NULL);
949 if (ret) {
950 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
951 goto error2;
953 BUG_ON(cm_id->state != IB_CM_IDLE);
954 cm_id->state = IB_CM_REQ_SENT;
955 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
956 return 0;
958 error2: cm_free_msg(cm_id_priv->msg);
959 error1: kfree(cm_id_priv->timewait_info);
960 out: return ret;
962 EXPORT_SYMBOL(ib_send_cm_req);
964 static int cm_issue_rej(struct cm_port *port,
965 struct ib_mad_recv_wc *mad_recv_wc,
966 enum ib_cm_rej_reason reason,
967 enum cm_msg_response msg_rejected,
968 void *ari, u8 ari_length)
970 struct ib_mad_send_buf *msg = NULL;
971 struct cm_rej_msg *rej_msg, *rcv_msg;
972 int ret;
974 ret = cm_alloc_response_msg(port, mad_recv_wc, &msg);
975 if (ret)
976 return ret;
978 /* We just need common CM header information. Cast to any message. */
979 rcv_msg = (struct cm_rej_msg *) mad_recv_wc->recv_buf.mad;
980 rej_msg = (struct cm_rej_msg *) msg->mad;
982 cm_format_mad_hdr(&rej_msg->hdr, CM_REJ_ATTR_ID, rcv_msg->hdr.tid);
983 rej_msg->remote_comm_id = rcv_msg->local_comm_id;
984 rej_msg->local_comm_id = rcv_msg->remote_comm_id;
985 cm_rej_set_msg_rejected(rej_msg, msg_rejected);
986 rej_msg->reason = cpu_to_be16(reason);
988 if (ari && ari_length) {
989 cm_rej_set_reject_info_len(rej_msg, ari_length);
990 memcpy(rej_msg->ari, ari, ari_length);
993 ret = ib_post_send_mad(msg, NULL);
994 if (ret)
995 cm_free_msg(msg);
997 return ret;
1000 static inline int cm_is_active_peer(__be64 local_ca_guid, __be64 remote_ca_guid,
1001 __be32 local_qpn, __be32 remote_qpn)
1003 return (be64_to_cpu(local_ca_guid) > be64_to_cpu(remote_ca_guid) ||
1004 ((local_ca_guid == remote_ca_guid) &&
1005 (be32_to_cpu(local_qpn) > be32_to_cpu(remote_qpn))));
1008 static void cm_format_paths_from_req(struct cm_req_msg *req_msg,
1009 struct ib_sa_path_rec *primary_path,
1010 struct ib_sa_path_rec *alt_path)
1012 memset(primary_path, 0, sizeof *primary_path);
1013 primary_path->dgid = req_msg->primary_local_gid;
1014 primary_path->sgid = req_msg->primary_remote_gid;
1015 primary_path->dlid = req_msg->primary_local_lid;
1016 primary_path->slid = req_msg->primary_remote_lid;
1017 primary_path->flow_label = cm_req_get_primary_flow_label(req_msg);
1018 primary_path->hop_limit = req_msg->primary_hop_limit;
1019 primary_path->traffic_class = req_msg->primary_traffic_class;
1020 primary_path->reversible = 1;
1021 primary_path->pkey = req_msg->pkey;
1022 primary_path->sl = cm_req_get_primary_sl(req_msg);
1023 primary_path->mtu_selector = IB_SA_EQ;
1024 primary_path->mtu = cm_req_get_path_mtu(req_msg);
1025 primary_path->rate_selector = IB_SA_EQ;
1026 primary_path->rate = cm_req_get_primary_packet_rate(req_msg);
1027 primary_path->packet_life_time_selector = IB_SA_EQ;
1028 primary_path->packet_life_time =
1029 cm_req_get_primary_local_ack_timeout(req_msg);
1030 primary_path->packet_life_time -= (primary_path->packet_life_time > 0);
1032 if (req_msg->alt_local_lid) {
1033 memset(alt_path, 0, sizeof *alt_path);
1034 alt_path->dgid = req_msg->alt_local_gid;
1035 alt_path->sgid = req_msg->alt_remote_gid;
1036 alt_path->dlid = req_msg->alt_local_lid;
1037 alt_path->slid = req_msg->alt_remote_lid;
1038 alt_path->flow_label = cm_req_get_alt_flow_label(req_msg);
1039 alt_path->hop_limit = req_msg->alt_hop_limit;
1040 alt_path->traffic_class = req_msg->alt_traffic_class;
1041 alt_path->reversible = 1;
1042 alt_path->pkey = req_msg->pkey;
1043 alt_path->sl = cm_req_get_alt_sl(req_msg);
1044 alt_path->mtu_selector = IB_SA_EQ;
1045 alt_path->mtu = cm_req_get_path_mtu(req_msg);
1046 alt_path->rate_selector = IB_SA_EQ;
1047 alt_path->rate = cm_req_get_alt_packet_rate(req_msg);
1048 alt_path->packet_life_time_selector = IB_SA_EQ;
1049 alt_path->packet_life_time =
1050 cm_req_get_alt_local_ack_timeout(req_msg);
1051 alt_path->packet_life_time -= (alt_path->packet_life_time > 0);
1055 static void cm_format_req_event(struct cm_work *work,
1056 struct cm_id_private *cm_id_priv,
1057 struct ib_cm_id *listen_id)
1059 struct cm_req_msg *req_msg;
1060 struct ib_cm_req_event_param *param;
1062 req_msg = (struct cm_req_msg *)work->mad_recv_wc->recv_buf.mad;
1063 param = &work->cm_event.param.req_rcvd;
1064 param->listen_id = listen_id;
1065 param->port = cm_id_priv->av.port->port_num;
1066 param->primary_path = &work->path[0];
1067 if (req_msg->alt_local_lid)
1068 param->alternate_path = &work->path[1];
1069 else
1070 param->alternate_path = NULL;
1071 param->remote_ca_guid = req_msg->local_ca_guid;
1072 param->remote_qkey = be32_to_cpu(req_msg->local_qkey);
1073 param->remote_qpn = be32_to_cpu(cm_req_get_local_qpn(req_msg));
1074 param->qp_type = cm_req_get_qp_type(req_msg);
1075 param->starting_psn = be32_to_cpu(cm_req_get_starting_psn(req_msg));
1076 param->responder_resources = cm_req_get_init_depth(req_msg);
1077 param->initiator_depth = cm_req_get_resp_res(req_msg);
1078 param->local_cm_response_timeout =
1079 cm_req_get_remote_resp_timeout(req_msg);
1080 param->flow_control = cm_req_get_flow_ctrl(req_msg);
1081 param->remote_cm_response_timeout =
1082 cm_req_get_local_resp_timeout(req_msg);
1083 param->retry_count = cm_req_get_retry_count(req_msg);
1084 param->rnr_retry_count = cm_req_get_rnr_retry_count(req_msg);
1085 param->srq = cm_req_get_srq(req_msg);
1086 work->cm_event.private_data = &req_msg->private_data;
1089 static void cm_process_work(struct cm_id_private *cm_id_priv,
1090 struct cm_work *work)
1092 unsigned long flags;
1093 int ret;
1095 /* We will typically only have the current event to report. */
1096 ret = cm_id_priv->id.cm_handler(&cm_id_priv->id, &work->cm_event);
1097 cm_free_work(work);
1099 while (!ret && !atomic_add_negative(-1, &cm_id_priv->work_count)) {
1100 spin_lock_irqsave(&cm_id_priv->lock, flags);
1101 work = cm_dequeue_work(cm_id_priv);
1102 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1103 BUG_ON(!work);
1104 ret = cm_id_priv->id.cm_handler(&cm_id_priv->id,
1105 &work->cm_event);
1106 cm_free_work(work);
1108 cm_deref_id(cm_id_priv);
1109 if (ret)
1110 ib_destroy_cm_id(&cm_id_priv->id);
1113 static void cm_format_mra(struct cm_mra_msg *mra_msg,
1114 struct cm_id_private *cm_id_priv,
1115 enum cm_msg_response msg_mraed, u8 service_timeout,
1116 const void *private_data, u8 private_data_len)
1118 cm_format_mad_hdr(&mra_msg->hdr, CM_MRA_ATTR_ID, cm_id_priv->tid);
1119 cm_mra_set_msg_mraed(mra_msg, msg_mraed);
1120 mra_msg->local_comm_id = cm_id_priv->id.local_id;
1121 mra_msg->remote_comm_id = cm_id_priv->id.remote_id;
1122 cm_mra_set_service_timeout(mra_msg, service_timeout);
1124 if (private_data && private_data_len)
1125 memcpy(mra_msg->private_data, private_data, private_data_len);
1128 static void cm_format_rej(struct cm_rej_msg *rej_msg,
1129 struct cm_id_private *cm_id_priv,
1130 enum ib_cm_rej_reason reason,
1131 void *ari,
1132 u8 ari_length,
1133 const void *private_data,
1134 u8 private_data_len)
1136 cm_format_mad_hdr(&rej_msg->hdr, CM_REJ_ATTR_ID, cm_id_priv->tid);
1137 rej_msg->remote_comm_id = cm_id_priv->id.remote_id;
1139 switch(cm_id_priv->id.state) {
1140 case IB_CM_REQ_RCVD:
1141 rej_msg->local_comm_id = 0;
1142 cm_rej_set_msg_rejected(rej_msg, CM_MSG_RESPONSE_REQ);
1143 break;
1144 case IB_CM_MRA_REQ_SENT:
1145 rej_msg->local_comm_id = cm_id_priv->id.local_id;
1146 cm_rej_set_msg_rejected(rej_msg, CM_MSG_RESPONSE_REQ);
1147 break;
1148 case IB_CM_REP_RCVD:
1149 case IB_CM_MRA_REP_SENT:
1150 rej_msg->local_comm_id = cm_id_priv->id.local_id;
1151 cm_rej_set_msg_rejected(rej_msg, CM_MSG_RESPONSE_REP);
1152 break;
1153 default:
1154 rej_msg->local_comm_id = cm_id_priv->id.local_id;
1155 cm_rej_set_msg_rejected(rej_msg, CM_MSG_RESPONSE_OTHER);
1156 break;
1159 rej_msg->reason = cpu_to_be16(reason);
1160 if (ari && ari_length) {
1161 cm_rej_set_reject_info_len(rej_msg, ari_length);
1162 memcpy(rej_msg->ari, ari, ari_length);
1165 if (private_data && private_data_len)
1166 memcpy(rej_msg->private_data, private_data, private_data_len);
1169 static void cm_dup_req_handler(struct cm_work *work,
1170 struct cm_id_private *cm_id_priv)
1172 struct ib_mad_send_buf *msg = NULL;
1173 unsigned long flags;
1174 int ret;
1176 /* Quick state check to discard duplicate REQs. */
1177 if (cm_id_priv->id.state == IB_CM_REQ_RCVD)
1178 return;
1180 ret = cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg);
1181 if (ret)
1182 return;
1184 spin_lock_irqsave(&cm_id_priv->lock, flags);
1185 switch (cm_id_priv->id.state) {
1186 case IB_CM_MRA_REQ_SENT:
1187 cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
1188 CM_MSG_RESPONSE_REQ, cm_id_priv->service_timeout,
1189 cm_id_priv->private_data,
1190 cm_id_priv->private_data_len);
1191 break;
1192 case IB_CM_TIMEWAIT:
1193 cm_format_rej((struct cm_rej_msg *) msg->mad, cm_id_priv,
1194 IB_CM_REJ_STALE_CONN, NULL, 0, NULL, 0);
1195 break;
1196 default:
1197 goto unlock;
1199 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1201 ret = ib_post_send_mad(msg, NULL);
1202 if (ret)
1203 goto free;
1204 return;
1206 unlock: spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1207 free: cm_free_msg(msg);
1210 static struct cm_id_private * cm_match_req(struct cm_work *work,
1211 struct cm_id_private *cm_id_priv)
1213 struct cm_id_private *listen_cm_id_priv, *cur_cm_id_priv;
1214 struct cm_timewait_info *timewait_info;
1215 struct cm_req_msg *req_msg;
1216 unsigned long flags;
1218 req_msg = (struct cm_req_msg *)work->mad_recv_wc->recv_buf.mad;
1220 /* Check for duplicate REQ and stale connections. */
1221 spin_lock_irqsave(&cm.lock, flags);
1222 timewait_info = cm_insert_remote_id(cm_id_priv->timewait_info);
1223 if (!timewait_info)
1224 timewait_info = cm_insert_remote_qpn(cm_id_priv->timewait_info);
1226 if (timewait_info) {
1227 cur_cm_id_priv = cm_get_id(timewait_info->work.local_id,
1228 timewait_info->work.remote_id);
1229 spin_unlock_irqrestore(&cm.lock, flags);
1230 if (cur_cm_id_priv) {
1231 cm_dup_req_handler(work, cur_cm_id_priv);
1232 cm_deref_id(cur_cm_id_priv);
1233 } else
1234 cm_issue_rej(work->port, work->mad_recv_wc,
1235 IB_CM_REJ_STALE_CONN, CM_MSG_RESPONSE_REQ,
1236 NULL, 0);
1237 goto error;
1240 /* Find matching listen request. */
1241 listen_cm_id_priv = cm_find_listen(cm_id_priv->id.device,
1242 req_msg->service_id);
1243 if (!listen_cm_id_priv) {
1244 spin_unlock_irqrestore(&cm.lock, flags);
1245 cm_issue_rej(work->port, work->mad_recv_wc,
1246 IB_CM_REJ_INVALID_SERVICE_ID, CM_MSG_RESPONSE_REQ,
1247 NULL, 0);
1248 goto error;
1250 atomic_inc(&listen_cm_id_priv->refcount);
1251 atomic_inc(&cm_id_priv->refcount);
1252 cm_id_priv->id.state = IB_CM_REQ_RCVD;
1253 atomic_inc(&cm_id_priv->work_count);
1254 spin_unlock_irqrestore(&cm.lock, flags);
1255 return listen_cm_id_priv;
1257 error: cm_cleanup_timewait(cm_id_priv->timewait_info);
1258 return NULL;
1261 static int cm_req_handler(struct cm_work *work)
1263 struct ib_cm_id *cm_id;
1264 struct cm_id_private *cm_id_priv, *listen_cm_id_priv;
1265 struct cm_req_msg *req_msg;
1266 int ret;
1268 req_msg = (struct cm_req_msg *)work->mad_recv_wc->recv_buf.mad;
1270 cm_id = ib_create_cm_id(work->port->cm_dev->device, NULL, NULL);
1271 if (IS_ERR(cm_id))
1272 return PTR_ERR(cm_id);
1274 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1275 cm_id_priv->id.remote_id = req_msg->local_comm_id;
1276 cm_init_av_for_response(work->port, work->mad_recv_wc->wc,
1277 &cm_id_priv->av);
1278 cm_id_priv->timewait_info = cm_create_timewait_info(cm_id_priv->
1279 id.local_id);
1280 if (IS_ERR(cm_id_priv->timewait_info)) {
1281 ret = PTR_ERR(cm_id_priv->timewait_info);
1282 goto error1;
1284 cm_id_priv->timewait_info->work.remote_id = req_msg->local_comm_id;
1285 cm_id_priv->timewait_info->remote_ca_guid = req_msg->local_ca_guid;
1286 cm_id_priv->timewait_info->remote_qpn = cm_req_get_local_qpn(req_msg);
1288 listen_cm_id_priv = cm_match_req(work, cm_id_priv);
1289 if (!listen_cm_id_priv) {
1290 ret = -EINVAL;
1291 goto error2;
1294 cm_id_priv->id.cm_handler = listen_cm_id_priv->id.cm_handler;
1295 cm_id_priv->id.context = listen_cm_id_priv->id.context;
1296 cm_id_priv->id.service_id = req_msg->service_id;
1297 cm_id_priv->id.service_mask = __constant_cpu_to_be64(~0ULL);
1299 cm_format_paths_from_req(req_msg, &work->path[0], &work->path[1]);
1300 ret = cm_init_av_by_path(&work->path[0], &cm_id_priv->av);
1301 if (ret)
1302 goto error3;
1303 if (req_msg->alt_local_lid) {
1304 ret = cm_init_av_by_path(&work->path[1], &cm_id_priv->alt_av);
1305 if (ret)
1306 goto error3;
1308 cm_id_priv->tid = req_msg->hdr.tid;
1309 cm_id_priv->timeout_ms = cm_convert_to_ms(
1310 cm_req_get_local_resp_timeout(req_msg));
1311 cm_id_priv->max_cm_retries = cm_req_get_max_cm_retries(req_msg);
1312 cm_id_priv->remote_qpn = cm_req_get_local_qpn(req_msg);
1313 cm_id_priv->initiator_depth = cm_req_get_resp_res(req_msg);
1314 cm_id_priv->responder_resources = cm_req_get_init_depth(req_msg);
1315 cm_id_priv->path_mtu = cm_req_get_path_mtu(req_msg);
1316 cm_id_priv->sq_psn = cm_req_get_starting_psn(req_msg);
1317 cm_id_priv->local_ack_timeout =
1318 cm_req_get_primary_local_ack_timeout(req_msg);
1319 cm_id_priv->retry_count = cm_req_get_retry_count(req_msg);
1320 cm_id_priv->rnr_retry_count = cm_req_get_rnr_retry_count(req_msg);
1321 cm_id_priv->qp_type = cm_req_get_qp_type(req_msg);
1323 cm_format_req_event(work, cm_id_priv, &listen_cm_id_priv->id);
1324 cm_process_work(cm_id_priv, work);
1325 cm_deref_id(listen_cm_id_priv);
1326 return 0;
1328 error3: atomic_dec(&cm_id_priv->refcount);
1329 cm_deref_id(listen_cm_id_priv);
1330 cm_cleanup_timewait(cm_id_priv->timewait_info);
1331 error2: kfree(cm_id_priv->timewait_info);
1332 cm_id_priv->timewait_info = NULL;
1333 error1: ib_destroy_cm_id(&cm_id_priv->id);
1334 return ret;
1337 static void cm_format_rep(struct cm_rep_msg *rep_msg,
1338 struct cm_id_private *cm_id_priv,
1339 struct ib_cm_rep_param *param)
1341 cm_format_mad_hdr(&rep_msg->hdr, CM_REP_ATTR_ID, cm_id_priv->tid);
1342 rep_msg->local_comm_id = cm_id_priv->id.local_id;
1343 rep_msg->remote_comm_id = cm_id_priv->id.remote_id;
1344 cm_rep_set_local_qpn(rep_msg, cpu_to_be32(param->qp_num));
1345 cm_rep_set_starting_psn(rep_msg, cpu_to_be32(param->starting_psn));
1346 rep_msg->resp_resources = param->responder_resources;
1347 rep_msg->initiator_depth = param->initiator_depth;
1348 cm_rep_set_target_ack_delay(rep_msg, param->target_ack_delay);
1349 cm_rep_set_failover(rep_msg, param->failover_accepted);
1350 cm_rep_set_flow_ctrl(rep_msg, param->flow_control);
1351 cm_rep_set_rnr_retry_count(rep_msg, param->rnr_retry_count);
1352 cm_rep_set_srq(rep_msg, param->srq);
1353 rep_msg->local_ca_guid = cm_id_priv->av.port->cm_dev->ca_guid;
1355 if (param->private_data && param->private_data_len)
1356 memcpy(rep_msg->private_data, param->private_data,
1357 param->private_data_len);
1360 int ib_send_cm_rep(struct ib_cm_id *cm_id,
1361 struct ib_cm_rep_param *param)
1363 struct cm_id_private *cm_id_priv;
1364 struct ib_mad_send_buf *msg;
1365 struct cm_rep_msg *rep_msg;
1366 unsigned long flags;
1367 int ret;
1369 if (param->private_data &&
1370 param->private_data_len > IB_CM_REP_PRIVATE_DATA_SIZE)
1371 return -EINVAL;
1373 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1374 spin_lock_irqsave(&cm_id_priv->lock, flags);
1375 if (cm_id->state != IB_CM_REQ_RCVD &&
1376 cm_id->state != IB_CM_MRA_REQ_SENT) {
1377 ret = -EINVAL;
1378 goto out;
1381 ret = cm_alloc_msg(cm_id_priv, &msg);
1382 if (ret)
1383 goto out;
1385 rep_msg = (struct cm_rep_msg *) msg->mad;
1386 cm_format_rep(rep_msg, cm_id_priv, param);
1387 msg->timeout_ms = cm_id_priv->timeout_ms;
1388 msg->context[1] = (void *) (unsigned long) IB_CM_REP_SENT;
1390 ret = ib_post_send_mad(msg, NULL);
1391 if (ret) {
1392 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1393 cm_free_msg(msg);
1394 return ret;
1397 cm_id->state = IB_CM_REP_SENT;
1398 cm_id_priv->msg = msg;
1399 cm_id_priv->initiator_depth = param->initiator_depth;
1400 cm_id_priv->responder_resources = param->responder_resources;
1401 cm_id_priv->rq_psn = cm_rep_get_starting_psn(rep_msg);
1402 cm_id_priv->local_qpn = cm_rep_get_local_qpn(rep_msg);
1404 out: spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1405 return ret;
1407 EXPORT_SYMBOL(ib_send_cm_rep);
1409 static void cm_format_rtu(struct cm_rtu_msg *rtu_msg,
1410 struct cm_id_private *cm_id_priv,
1411 const void *private_data,
1412 u8 private_data_len)
1414 cm_format_mad_hdr(&rtu_msg->hdr, CM_RTU_ATTR_ID, cm_id_priv->tid);
1415 rtu_msg->local_comm_id = cm_id_priv->id.local_id;
1416 rtu_msg->remote_comm_id = cm_id_priv->id.remote_id;
1418 if (private_data && private_data_len)
1419 memcpy(rtu_msg->private_data, private_data, private_data_len);
1422 int ib_send_cm_rtu(struct ib_cm_id *cm_id,
1423 const void *private_data,
1424 u8 private_data_len)
1426 struct cm_id_private *cm_id_priv;
1427 struct ib_mad_send_buf *msg;
1428 unsigned long flags;
1429 void *data;
1430 int ret;
1432 if (private_data && private_data_len > IB_CM_RTU_PRIVATE_DATA_SIZE)
1433 return -EINVAL;
1435 data = cm_copy_private_data(private_data, private_data_len);
1436 if (IS_ERR(data))
1437 return PTR_ERR(data);
1439 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1440 spin_lock_irqsave(&cm_id_priv->lock, flags);
1441 if (cm_id->state != IB_CM_REP_RCVD &&
1442 cm_id->state != IB_CM_MRA_REP_SENT) {
1443 ret = -EINVAL;
1444 goto error;
1447 ret = cm_alloc_msg(cm_id_priv, &msg);
1448 if (ret)
1449 goto error;
1451 cm_format_rtu((struct cm_rtu_msg *) msg->mad, cm_id_priv,
1452 private_data, private_data_len);
1454 ret = ib_post_send_mad(msg, NULL);
1455 if (ret) {
1456 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1457 cm_free_msg(msg);
1458 kfree(data);
1459 return ret;
1462 cm_id->state = IB_CM_ESTABLISHED;
1463 cm_set_private_data(cm_id_priv, data, private_data_len);
1464 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1465 return 0;
1467 error: spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1468 kfree(data);
1469 return ret;
1471 EXPORT_SYMBOL(ib_send_cm_rtu);
1473 static void cm_format_rep_event(struct cm_work *work)
1475 struct cm_rep_msg *rep_msg;
1476 struct ib_cm_rep_event_param *param;
1478 rep_msg = (struct cm_rep_msg *)work->mad_recv_wc->recv_buf.mad;
1479 param = &work->cm_event.param.rep_rcvd;
1480 param->remote_ca_guid = rep_msg->local_ca_guid;
1481 param->remote_qkey = be32_to_cpu(rep_msg->local_qkey);
1482 param->remote_qpn = be32_to_cpu(cm_rep_get_local_qpn(rep_msg));
1483 param->starting_psn = be32_to_cpu(cm_rep_get_starting_psn(rep_msg));
1484 param->responder_resources = rep_msg->initiator_depth;
1485 param->initiator_depth = rep_msg->resp_resources;
1486 param->target_ack_delay = cm_rep_get_target_ack_delay(rep_msg);
1487 param->failover_accepted = cm_rep_get_failover(rep_msg);
1488 param->flow_control = cm_rep_get_flow_ctrl(rep_msg);
1489 param->rnr_retry_count = cm_rep_get_rnr_retry_count(rep_msg);
1490 param->srq = cm_rep_get_srq(rep_msg);
1491 work->cm_event.private_data = &rep_msg->private_data;
1494 static void cm_dup_rep_handler(struct cm_work *work)
1496 struct cm_id_private *cm_id_priv;
1497 struct cm_rep_msg *rep_msg;
1498 struct ib_mad_send_buf *msg = NULL;
1499 unsigned long flags;
1500 int ret;
1502 rep_msg = (struct cm_rep_msg *) work->mad_recv_wc->recv_buf.mad;
1503 cm_id_priv = cm_acquire_id(rep_msg->remote_comm_id,
1504 rep_msg->local_comm_id);
1505 if (!cm_id_priv)
1506 return;
1508 ret = cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg);
1509 if (ret)
1510 goto deref;
1512 spin_lock_irqsave(&cm_id_priv->lock, flags);
1513 if (cm_id_priv->id.state == IB_CM_ESTABLISHED)
1514 cm_format_rtu((struct cm_rtu_msg *) msg->mad, cm_id_priv,
1515 cm_id_priv->private_data,
1516 cm_id_priv->private_data_len);
1517 else if (cm_id_priv->id.state == IB_CM_MRA_REP_SENT)
1518 cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
1519 CM_MSG_RESPONSE_REP, cm_id_priv->service_timeout,
1520 cm_id_priv->private_data,
1521 cm_id_priv->private_data_len);
1522 else
1523 goto unlock;
1524 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1526 ret = ib_post_send_mad(msg, NULL);
1527 if (ret)
1528 goto free;
1529 goto deref;
1531 unlock: spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1532 free: cm_free_msg(msg);
1533 deref: cm_deref_id(cm_id_priv);
1536 static int cm_rep_handler(struct cm_work *work)
1538 struct cm_id_private *cm_id_priv;
1539 struct cm_rep_msg *rep_msg;
1540 unsigned long flags;
1541 int ret;
1543 rep_msg = (struct cm_rep_msg *)work->mad_recv_wc->recv_buf.mad;
1544 cm_id_priv = cm_acquire_id(rep_msg->remote_comm_id, 0);
1545 if (!cm_id_priv) {
1546 cm_dup_rep_handler(work);
1547 return -EINVAL;
1550 cm_format_rep_event(work);
1552 spin_lock_irqsave(&cm_id_priv->lock, flags);
1553 switch (cm_id_priv->id.state) {
1554 case IB_CM_REQ_SENT:
1555 case IB_CM_MRA_REQ_RCVD:
1556 break;
1557 default:
1558 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1559 ret = -EINVAL;
1560 goto error;
1563 cm_id_priv->timewait_info->work.remote_id = rep_msg->local_comm_id;
1564 cm_id_priv->timewait_info->remote_ca_guid = rep_msg->local_ca_guid;
1565 cm_id_priv->timewait_info->remote_qpn = cm_rep_get_local_qpn(rep_msg);
1567 spin_lock(&cm.lock);
1568 /* Check for duplicate REP. */
1569 if (cm_insert_remote_id(cm_id_priv->timewait_info)) {
1570 spin_unlock(&cm.lock);
1571 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1572 ret = -EINVAL;
1573 goto error;
1575 /* Check for a stale connection. */
1576 if (cm_insert_remote_qpn(cm_id_priv->timewait_info)) {
1577 rb_erase(&cm_id_priv->timewait_info->remote_id_node,
1578 &cm.remote_id_table);
1579 cm_id_priv->timewait_info->inserted_remote_id = 0;
1580 spin_unlock(&cm.lock);
1581 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1582 cm_issue_rej(work->port, work->mad_recv_wc,
1583 IB_CM_REJ_STALE_CONN, CM_MSG_RESPONSE_REP,
1584 NULL, 0);
1585 ret = -EINVAL;
1586 goto error;
1588 spin_unlock(&cm.lock);
1590 cm_id_priv->id.state = IB_CM_REP_RCVD;
1591 cm_id_priv->id.remote_id = rep_msg->local_comm_id;
1592 cm_id_priv->remote_qpn = cm_rep_get_local_qpn(rep_msg);
1593 cm_id_priv->initiator_depth = rep_msg->resp_resources;
1594 cm_id_priv->responder_resources = rep_msg->initiator_depth;
1595 cm_id_priv->sq_psn = cm_rep_get_starting_psn(rep_msg);
1596 cm_id_priv->rnr_retry_count = cm_rep_get_rnr_retry_count(rep_msg);
1598 /* todo: handle peer_to_peer */
1600 ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
1601 ret = atomic_inc_and_test(&cm_id_priv->work_count);
1602 if (!ret)
1603 list_add_tail(&work->list, &cm_id_priv->work_list);
1604 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1606 if (ret)
1607 cm_process_work(cm_id_priv, work);
1608 else
1609 cm_deref_id(cm_id_priv);
1610 return 0;
1612 error:
1613 cm_deref_id(cm_id_priv);
1614 return ret;
1617 static int cm_establish_handler(struct cm_work *work)
1619 struct cm_id_private *cm_id_priv;
1620 unsigned long flags;
1621 int ret;
1623 /* See comment in ib_cm_establish about lookup. */
1624 cm_id_priv = cm_acquire_id(work->local_id, work->remote_id);
1625 if (!cm_id_priv)
1626 return -EINVAL;
1628 spin_lock_irqsave(&cm_id_priv->lock, flags);
1629 if (cm_id_priv->id.state != IB_CM_ESTABLISHED) {
1630 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1631 goto out;
1634 ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
1635 ret = atomic_inc_and_test(&cm_id_priv->work_count);
1636 if (!ret)
1637 list_add_tail(&work->list, &cm_id_priv->work_list);
1638 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1640 if (ret)
1641 cm_process_work(cm_id_priv, work);
1642 else
1643 cm_deref_id(cm_id_priv);
1644 return 0;
1645 out:
1646 cm_deref_id(cm_id_priv);
1647 return -EINVAL;
1650 static int cm_rtu_handler(struct cm_work *work)
1652 struct cm_id_private *cm_id_priv;
1653 struct cm_rtu_msg *rtu_msg;
1654 unsigned long flags;
1655 int ret;
1657 rtu_msg = (struct cm_rtu_msg *)work->mad_recv_wc->recv_buf.mad;
1658 cm_id_priv = cm_acquire_id(rtu_msg->remote_comm_id,
1659 rtu_msg->local_comm_id);
1660 if (!cm_id_priv)
1661 return -EINVAL;
1663 work->cm_event.private_data = &rtu_msg->private_data;
1665 spin_lock_irqsave(&cm_id_priv->lock, flags);
1666 if (cm_id_priv->id.state != IB_CM_REP_SENT &&
1667 cm_id_priv->id.state != IB_CM_MRA_REP_RCVD) {
1668 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1669 goto out;
1671 cm_id_priv->id.state = IB_CM_ESTABLISHED;
1673 ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
1674 ret = atomic_inc_and_test(&cm_id_priv->work_count);
1675 if (!ret)
1676 list_add_tail(&work->list, &cm_id_priv->work_list);
1677 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1679 if (ret)
1680 cm_process_work(cm_id_priv, work);
1681 else
1682 cm_deref_id(cm_id_priv);
1683 return 0;
1684 out:
1685 cm_deref_id(cm_id_priv);
1686 return -EINVAL;
1689 static void cm_format_dreq(struct cm_dreq_msg *dreq_msg,
1690 struct cm_id_private *cm_id_priv,
1691 const void *private_data,
1692 u8 private_data_len)
1694 cm_format_mad_hdr(&dreq_msg->hdr, CM_DREQ_ATTR_ID,
1695 cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_DREQ));
1696 dreq_msg->local_comm_id = cm_id_priv->id.local_id;
1697 dreq_msg->remote_comm_id = cm_id_priv->id.remote_id;
1698 cm_dreq_set_remote_qpn(dreq_msg, cm_id_priv->remote_qpn);
1700 if (private_data && private_data_len)
1701 memcpy(dreq_msg->private_data, private_data, private_data_len);
1704 int ib_send_cm_dreq(struct ib_cm_id *cm_id,
1705 const void *private_data,
1706 u8 private_data_len)
1708 struct cm_id_private *cm_id_priv;
1709 struct ib_mad_send_buf *msg;
1710 unsigned long flags;
1711 int ret;
1713 if (private_data && private_data_len > IB_CM_DREQ_PRIVATE_DATA_SIZE)
1714 return -EINVAL;
1716 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1717 spin_lock_irqsave(&cm_id_priv->lock, flags);
1718 if (cm_id->state != IB_CM_ESTABLISHED) {
1719 ret = -EINVAL;
1720 goto out;
1723 ret = cm_alloc_msg(cm_id_priv, &msg);
1724 if (ret) {
1725 cm_enter_timewait(cm_id_priv);
1726 goto out;
1729 cm_format_dreq((struct cm_dreq_msg *) msg->mad, cm_id_priv,
1730 private_data, private_data_len);
1731 msg->timeout_ms = cm_id_priv->timeout_ms;
1732 msg->context[1] = (void *) (unsigned long) IB_CM_DREQ_SENT;
1734 ret = ib_post_send_mad(msg, NULL);
1735 if (ret) {
1736 cm_enter_timewait(cm_id_priv);
1737 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1738 cm_free_msg(msg);
1739 return ret;
1742 cm_id->state = IB_CM_DREQ_SENT;
1743 cm_id_priv->msg = msg;
1744 out: spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1745 return ret;
1747 EXPORT_SYMBOL(ib_send_cm_dreq);
1749 static void cm_format_drep(struct cm_drep_msg *drep_msg,
1750 struct cm_id_private *cm_id_priv,
1751 const void *private_data,
1752 u8 private_data_len)
1754 cm_format_mad_hdr(&drep_msg->hdr, CM_DREP_ATTR_ID, cm_id_priv->tid);
1755 drep_msg->local_comm_id = cm_id_priv->id.local_id;
1756 drep_msg->remote_comm_id = cm_id_priv->id.remote_id;
1758 if (private_data && private_data_len)
1759 memcpy(drep_msg->private_data, private_data, private_data_len);
1762 int ib_send_cm_drep(struct ib_cm_id *cm_id,
1763 const void *private_data,
1764 u8 private_data_len)
1766 struct cm_id_private *cm_id_priv;
1767 struct ib_mad_send_buf *msg;
1768 unsigned long flags;
1769 void *data;
1770 int ret;
1772 if (private_data && private_data_len > IB_CM_DREP_PRIVATE_DATA_SIZE)
1773 return -EINVAL;
1775 data = cm_copy_private_data(private_data, private_data_len);
1776 if (IS_ERR(data))
1777 return PTR_ERR(data);
1779 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1780 spin_lock_irqsave(&cm_id_priv->lock, flags);
1781 if (cm_id->state != IB_CM_DREQ_RCVD) {
1782 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1783 kfree(data);
1784 return -EINVAL;
1787 cm_set_private_data(cm_id_priv, data, private_data_len);
1788 cm_enter_timewait(cm_id_priv);
1790 ret = cm_alloc_msg(cm_id_priv, &msg);
1791 if (ret)
1792 goto out;
1794 cm_format_drep((struct cm_drep_msg *) msg->mad, cm_id_priv,
1795 private_data, private_data_len);
1797 ret = ib_post_send_mad(msg, NULL);
1798 if (ret) {
1799 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1800 cm_free_msg(msg);
1801 return ret;
1804 out: spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1805 return ret;
1807 EXPORT_SYMBOL(ib_send_cm_drep);
1809 static int cm_dreq_handler(struct cm_work *work)
1811 struct cm_id_private *cm_id_priv;
1812 struct cm_dreq_msg *dreq_msg;
1813 struct ib_mad_send_buf *msg = NULL;
1814 unsigned long flags;
1815 int ret;
1817 dreq_msg = (struct cm_dreq_msg *)work->mad_recv_wc->recv_buf.mad;
1818 cm_id_priv = cm_acquire_id(dreq_msg->remote_comm_id,
1819 dreq_msg->local_comm_id);
1820 if (!cm_id_priv)
1821 return -EINVAL;
1823 work->cm_event.private_data = &dreq_msg->private_data;
1825 spin_lock_irqsave(&cm_id_priv->lock, flags);
1826 if (cm_id_priv->local_qpn != cm_dreq_get_remote_qpn(dreq_msg))
1827 goto unlock;
1829 switch (cm_id_priv->id.state) {
1830 case IB_CM_REP_SENT:
1831 case IB_CM_DREQ_SENT:
1832 ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
1833 break;
1834 case IB_CM_ESTABLISHED:
1835 case IB_CM_MRA_REP_RCVD:
1836 break;
1837 case IB_CM_TIMEWAIT:
1838 if (cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg))
1839 goto unlock;
1841 cm_format_drep((struct cm_drep_msg *) msg->mad, cm_id_priv,
1842 cm_id_priv->private_data,
1843 cm_id_priv->private_data_len);
1844 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1846 if (ib_post_send_mad(msg, NULL))
1847 cm_free_msg(msg);
1848 goto deref;
1849 default:
1850 goto unlock;
1852 cm_id_priv->id.state = IB_CM_DREQ_RCVD;
1853 cm_id_priv->tid = dreq_msg->hdr.tid;
1854 ret = atomic_inc_and_test(&cm_id_priv->work_count);
1855 if (!ret)
1856 list_add_tail(&work->list, &cm_id_priv->work_list);
1857 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1859 if (ret)
1860 cm_process_work(cm_id_priv, work);
1861 else
1862 cm_deref_id(cm_id_priv);
1863 return 0;
1865 unlock: spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1866 deref: cm_deref_id(cm_id_priv);
1867 return -EINVAL;
1870 static int cm_drep_handler(struct cm_work *work)
1872 struct cm_id_private *cm_id_priv;
1873 struct cm_drep_msg *drep_msg;
1874 unsigned long flags;
1875 int ret;
1877 drep_msg = (struct cm_drep_msg *)work->mad_recv_wc->recv_buf.mad;
1878 cm_id_priv = cm_acquire_id(drep_msg->remote_comm_id,
1879 drep_msg->local_comm_id);
1880 if (!cm_id_priv)
1881 return -EINVAL;
1883 work->cm_event.private_data = &drep_msg->private_data;
1885 spin_lock_irqsave(&cm_id_priv->lock, flags);
1886 if (cm_id_priv->id.state != IB_CM_DREQ_SENT &&
1887 cm_id_priv->id.state != IB_CM_DREQ_RCVD) {
1888 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1889 goto out;
1891 cm_enter_timewait(cm_id_priv);
1893 ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
1894 ret = atomic_inc_and_test(&cm_id_priv->work_count);
1895 if (!ret)
1896 list_add_tail(&work->list, &cm_id_priv->work_list);
1897 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1899 if (ret)
1900 cm_process_work(cm_id_priv, work);
1901 else
1902 cm_deref_id(cm_id_priv);
1903 return 0;
1904 out:
1905 cm_deref_id(cm_id_priv);
1906 return -EINVAL;
1909 int ib_send_cm_rej(struct ib_cm_id *cm_id,
1910 enum ib_cm_rej_reason reason,
1911 void *ari,
1912 u8 ari_length,
1913 const void *private_data,
1914 u8 private_data_len)
1916 struct cm_id_private *cm_id_priv;
1917 struct ib_mad_send_buf *msg;
1918 unsigned long flags;
1919 int ret;
1921 if ((private_data && private_data_len > IB_CM_REJ_PRIVATE_DATA_SIZE) ||
1922 (ari && ari_length > IB_CM_REJ_ARI_LENGTH))
1923 return -EINVAL;
1925 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
1927 spin_lock_irqsave(&cm_id_priv->lock, flags);
1928 switch (cm_id->state) {
1929 case IB_CM_REQ_SENT:
1930 case IB_CM_MRA_REQ_RCVD:
1931 case IB_CM_REQ_RCVD:
1932 case IB_CM_MRA_REQ_SENT:
1933 case IB_CM_REP_RCVD:
1934 case IB_CM_MRA_REP_SENT:
1935 ret = cm_alloc_msg(cm_id_priv, &msg);
1936 if (!ret)
1937 cm_format_rej((struct cm_rej_msg *) msg->mad,
1938 cm_id_priv, reason, ari, ari_length,
1939 private_data, private_data_len);
1941 cm_reset_to_idle(cm_id_priv);
1942 break;
1943 case IB_CM_REP_SENT:
1944 case IB_CM_MRA_REP_RCVD:
1945 ret = cm_alloc_msg(cm_id_priv, &msg);
1946 if (!ret)
1947 cm_format_rej((struct cm_rej_msg *) msg->mad,
1948 cm_id_priv, reason, ari, ari_length,
1949 private_data, private_data_len);
1951 cm_enter_timewait(cm_id_priv);
1952 break;
1953 default:
1954 ret = -EINVAL;
1955 goto out;
1958 if (ret)
1959 goto out;
1961 ret = ib_post_send_mad(msg, NULL);
1962 if (ret)
1963 cm_free_msg(msg);
1965 out: spin_unlock_irqrestore(&cm_id_priv->lock, flags);
1966 return ret;
1968 EXPORT_SYMBOL(ib_send_cm_rej);
1970 static void cm_format_rej_event(struct cm_work *work)
1972 struct cm_rej_msg *rej_msg;
1973 struct ib_cm_rej_event_param *param;
1975 rej_msg = (struct cm_rej_msg *)work->mad_recv_wc->recv_buf.mad;
1976 param = &work->cm_event.param.rej_rcvd;
1977 param->ari = rej_msg->ari;
1978 param->ari_length = cm_rej_get_reject_info_len(rej_msg);
1979 param->reason = __be16_to_cpu(rej_msg->reason);
1980 work->cm_event.private_data = &rej_msg->private_data;
1983 static struct cm_id_private * cm_acquire_rejected_id(struct cm_rej_msg *rej_msg)
1985 struct cm_timewait_info *timewait_info;
1986 struct cm_id_private *cm_id_priv;
1987 unsigned long flags;
1988 __be32 remote_id;
1990 remote_id = rej_msg->local_comm_id;
1992 if (__be16_to_cpu(rej_msg->reason) == IB_CM_REJ_TIMEOUT) {
1993 spin_lock_irqsave(&cm.lock, flags);
1994 timewait_info = cm_find_remote_id( *((__be64 *) rej_msg->ari),
1995 remote_id);
1996 if (!timewait_info) {
1997 spin_unlock_irqrestore(&cm.lock, flags);
1998 return NULL;
2000 cm_id_priv = idr_find(&cm.local_id_table,
2001 (__force int) timewait_info->work.local_id);
2002 if (cm_id_priv) {
2003 if (cm_id_priv->id.remote_id == remote_id)
2004 atomic_inc(&cm_id_priv->refcount);
2005 else
2006 cm_id_priv = NULL;
2008 spin_unlock_irqrestore(&cm.lock, flags);
2009 } else if (cm_rej_get_msg_rejected(rej_msg) == CM_MSG_RESPONSE_REQ)
2010 cm_id_priv = cm_acquire_id(rej_msg->remote_comm_id, 0);
2011 else
2012 cm_id_priv = cm_acquire_id(rej_msg->remote_comm_id, remote_id);
2014 return cm_id_priv;
2017 static int cm_rej_handler(struct cm_work *work)
2019 struct cm_id_private *cm_id_priv;
2020 struct cm_rej_msg *rej_msg;
2021 unsigned long flags;
2022 int ret;
2024 rej_msg = (struct cm_rej_msg *)work->mad_recv_wc->recv_buf.mad;
2025 cm_id_priv = cm_acquire_rejected_id(rej_msg);
2026 if (!cm_id_priv)
2027 return -EINVAL;
2029 cm_format_rej_event(work);
2031 spin_lock_irqsave(&cm_id_priv->lock, flags);
2032 switch (cm_id_priv->id.state) {
2033 case IB_CM_REQ_SENT:
2034 case IB_CM_MRA_REQ_RCVD:
2035 case IB_CM_REP_SENT:
2036 case IB_CM_MRA_REP_RCVD:
2037 ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2038 /* fall through */
2039 case IB_CM_REQ_RCVD:
2040 case IB_CM_MRA_REQ_SENT:
2041 if (__be16_to_cpu(rej_msg->reason) == IB_CM_REJ_STALE_CONN)
2042 cm_enter_timewait(cm_id_priv);
2043 else
2044 cm_reset_to_idle(cm_id_priv);
2045 break;
2046 case IB_CM_DREQ_SENT:
2047 ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2048 /* fall through */
2049 case IB_CM_REP_RCVD:
2050 case IB_CM_MRA_REP_SENT:
2051 case IB_CM_ESTABLISHED:
2052 cm_enter_timewait(cm_id_priv);
2053 break;
2054 default:
2055 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2056 ret = -EINVAL;
2057 goto out;
2060 ret = atomic_inc_and_test(&cm_id_priv->work_count);
2061 if (!ret)
2062 list_add_tail(&work->list, &cm_id_priv->work_list);
2063 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2065 if (ret)
2066 cm_process_work(cm_id_priv, work);
2067 else
2068 cm_deref_id(cm_id_priv);
2069 return 0;
2070 out:
2071 cm_deref_id(cm_id_priv);
2072 return -EINVAL;
2075 int ib_send_cm_mra(struct ib_cm_id *cm_id,
2076 u8 service_timeout,
2077 const void *private_data,
2078 u8 private_data_len)
2080 struct cm_id_private *cm_id_priv;
2081 struct ib_mad_send_buf *msg;
2082 void *data;
2083 unsigned long flags;
2084 int ret;
2086 if (private_data && private_data_len > IB_CM_MRA_PRIVATE_DATA_SIZE)
2087 return -EINVAL;
2089 data = cm_copy_private_data(private_data, private_data_len);
2090 if (IS_ERR(data))
2091 return PTR_ERR(data);
2093 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2095 spin_lock_irqsave(&cm_id_priv->lock, flags);
2096 switch(cm_id_priv->id.state) {
2097 case IB_CM_REQ_RCVD:
2098 ret = cm_alloc_msg(cm_id_priv, &msg);
2099 if (ret)
2100 goto error1;
2102 cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
2103 CM_MSG_RESPONSE_REQ, service_timeout,
2104 private_data, private_data_len);
2105 ret = ib_post_send_mad(msg, NULL);
2106 if (ret)
2107 goto error2;
2108 cm_id->state = IB_CM_MRA_REQ_SENT;
2109 break;
2110 case IB_CM_REP_RCVD:
2111 ret = cm_alloc_msg(cm_id_priv, &msg);
2112 if (ret)
2113 goto error1;
2115 cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
2116 CM_MSG_RESPONSE_REP, service_timeout,
2117 private_data, private_data_len);
2118 ret = ib_post_send_mad(msg, NULL);
2119 if (ret)
2120 goto error2;
2121 cm_id->state = IB_CM_MRA_REP_SENT;
2122 break;
2123 case IB_CM_ESTABLISHED:
2124 ret = cm_alloc_msg(cm_id_priv, &msg);
2125 if (ret)
2126 goto error1;
2128 cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
2129 CM_MSG_RESPONSE_OTHER, service_timeout,
2130 private_data, private_data_len);
2131 ret = ib_post_send_mad(msg, NULL);
2132 if (ret)
2133 goto error2;
2134 cm_id->lap_state = IB_CM_MRA_LAP_SENT;
2135 break;
2136 default:
2137 ret = -EINVAL;
2138 goto error1;
2140 cm_id_priv->service_timeout = service_timeout;
2141 cm_set_private_data(cm_id_priv, data, private_data_len);
2142 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2143 return 0;
2145 error1: spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2146 kfree(data);
2147 return ret;
2149 error2: spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2150 kfree(data);
2151 cm_free_msg(msg);
2152 return ret;
2154 EXPORT_SYMBOL(ib_send_cm_mra);
2156 static struct cm_id_private * cm_acquire_mraed_id(struct cm_mra_msg *mra_msg)
2158 switch (cm_mra_get_msg_mraed(mra_msg)) {
2159 case CM_MSG_RESPONSE_REQ:
2160 return cm_acquire_id(mra_msg->remote_comm_id, 0);
2161 case CM_MSG_RESPONSE_REP:
2162 case CM_MSG_RESPONSE_OTHER:
2163 return cm_acquire_id(mra_msg->remote_comm_id,
2164 mra_msg->local_comm_id);
2165 default:
2166 return NULL;
2170 static int cm_mra_handler(struct cm_work *work)
2172 struct cm_id_private *cm_id_priv;
2173 struct cm_mra_msg *mra_msg;
2174 unsigned long flags;
2175 int timeout, ret;
2177 mra_msg = (struct cm_mra_msg *)work->mad_recv_wc->recv_buf.mad;
2178 cm_id_priv = cm_acquire_mraed_id(mra_msg);
2179 if (!cm_id_priv)
2180 return -EINVAL;
2182 work->cm_event.private_data = &mra_msg->private_data;
2183 work->cm_event.param.mra_rcvd.service_timeout =
2184 cm_mra_get_service_timeout(mra_msg);
2185 timeout = cm_convert_to_ms(cm_mra_get_service_timeout(mra_msg)) +
2186 cm_convert_to_ms(cm_id_priv->av.packet_life_time);
2188 spin_lock_irqsave(&cm_id_priv->lock, flags);
2189 switch (cm_id_priv->id.state) {
2190 case IB_CM_REQ_SENT:
2191 if (cm_mra_get_msg_mraed(mra_msg) != CM_MSG_RESPONSE_REQ ||
2192 ib_modify_mad(cm_id_priv->av.port->mad_agent,
2193 cm_id_priv->msg, timeout))
2194 goto out;
2195 cm_id_priv->id.state = IB_CM_MRA_REQ_RCVD;
2196 break;
2197 case IB_CM_REP_SENT:
2198 if (cm_mra_get_msg_mraed(mra_msg) != CM_MSG_RESPONSE_REP ||
2199 ib_modify_mad(cm_id_priv->av.port->mad_agent,
2200 cm_id_priv->msg, timeout))
2201 goto out;
2202 cm_id_priv->id.state = IB_CM_MRA_REP_RCVD;
2203 break;
2204 case IB_CM_ESTABLISHED:
2205 if (cm_mra_get_msg_mraed(mra_msg) != CM_MSG_RESPONSE_OTHER ||
2206 cm_id_priv->id.lap_state != IB_CM_LAP_SENT ||
2207 ib_modify_mad(cm_id_priv->av.port->mad_agent,
2208 cm_id_priv->msg, timeout))
2209 goto out;
2210 cm_id_priv->id.lap_state = IB_CM_MRA_LAP_RCVD;
2211 break;
2212 default:
2213 goto out;
2216 cm_id_priv->msg->context[1] = (void *) (unsigned long)
2217 cm_id_priv->id.state;
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_irqrestore(&cm_id_priv->lock, flags);
2223 if (ret)
2224 cm_process_work(cm_id_priv, work);
2225 else
2226 cm_deref_id(cm_id_priv);
2227 return 0;
2228 out:
2229 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2230 cm_deref_id(cm_id_priv);
2231 return -EINVAL;
2234 static void cm_format_lap(struct cm_lap_msg *lap_msg,
2235 struct cm_id_private *cm_id_priv,
2236 struct ib_sa_path_rec *alternate_path,
2237 const void *private_data,
2238 u8 private_data_len)
2240 cm_format_mad_hdr(&lap_msg->hdr, CM_LAP_ATTR_ID,
2241 cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_LAP));
2242 lap_msg->local_comm_id = cm_id_priv->id.local_id;
2243 lap_msg->remote_comm_id = cm_id_priv->id.remote_id;
2244 cm_lap_set_remote_qpn(lap_msg, cm_id_priv->remote_qpn);
2245 /* todo: need remote CM response timeout */
2246 cm_lap_set_remote_resp_timeout(lap_msg, 0x1F);
2247 lap_msg->alt_local_lid = alternate_path->slid;
2248 lap_msg->alt_remote_lid = alternate_path->dlid;
2249 lap_msg->alt_local_gid = alternate_path->sgid;
2250 lap_msg->alt_remote_gid = alternate_path->dgid;
2251 cm_lap_set_flow_label(lap_msg, alternate_path->flow_label);
2252 cm_lap_set_traffic_class(lap_msg, alternate_path->traffic_class);
2253 lap_msg->alt_hop_limit = alternate_path->hop_limit;
2254 cm_lap_set_packet_rate(lap_msg, alternate_path->rate);
2255 cm_lap_set_sl(lap_msg, alternate_path->sl);
2256 cm_lap_set_subnet_local(lap_msg, 1); /* local only... */
2257 cm_lap_set_local_ack_timeout(lap_msg,
2258 min(31, alternate_path->packet_life_time + 1));
2260 if (private_data && private_data_len)
2261 memcpy(lap_msg->private_data, private_data, private_data_len);
2264 int ib_send_cm_lap(struct ib_cm_id *cm_id,
2265 struct ib_sa_path_rec *alternate_path,
2266 const void *private_data,
2267 u8 private_data_len)
2269 struct cm_id_private *cm_id_priv;
2270 struct ib_mad_send_buf *msg;
2271 unsigned long flags;
2272 int ret;
2274 if (private_data && private_data_len > IB_CM_LAP_PRIVATE_DATA_SIZE)
2275 return -EINVAL;
2277 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2278 spin_lock_irqsave(&cm_id_priv->lock, flags);
2279 if (cm_id->state != IB_CM_ESTABLISHED ||
2280 cm_id->lap_state != IB_CM_LAP_IDLE) {
2281 ret = -EINVAL;
2282 goto out;
2285 ret = cm_alloc_msg(cm_id_priv, &msg);
2286 if (ret)
2287 goto out;
2289 cm_format_lap((struct cm_lap_msg *) msg->mad, cm_id_priv,
2290 alternate_path, private_data, private_data_len);
2291 msg->timeout_ms = cm_id_priv->timeout_ms;
2292 msg->context[1] = (void *) (unsigned long) IB_CM_ESTABLISHED;
2294 ret = ib_post_send_mad(msg, NULL);
2295 if (ret) {
2296 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2297 cm_free_msg(msg);
2298 return ret;
2301 cm_id->lap_state = IB_CM_LAP_SENT;
2302 cm_id_priv->msg = msg;
2304 out: spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2305 return ret;
2307 EXPORT_SYMBOL(ib_send_cm_lap);
2309 static void cm_format_path_from_lap(struct ib_sa_path_rec *path,
2310 struct cm_lap_msg *lap_msg)
2312 memset(path, 0, sizeof *path);
2313 path->dgid = lap_msg->alt_local_gid;
2314 path->sgid = lap_msg->alt_remote_gid;
2315 path->dlid = lap_msg->alt_local_lid;
2316 path->slid = lap_msg->alt_remote_lid;
2317 path->flow_label = cm_lap_get_flow_label(lap_msg);
2318 path->hop_limit = lap_msg->alt_hop_limit;
2319 path->traffic_class = cm_lap_get_traffic_class(lap_msg);
2320 path->reversible = 1;
2321 /* pkey is same as in REQ */
2322 path->sl = cm_lap_get_sl(lap_msg);
2323 path->mtu_selector = IB_SA_EQ;
2324 /* mtu is same as in REQ */
2325 path->rate_selector = IB_SA_EQ;
2326 path->rate = cm_lap_get_packet_rate(lap_msg);
2327 path->packet_life_time_selector = IB_SA_EQ;
2328 path->packet_life_time = cm_lap_get_local_ack_timeout(lap_msg);
2329 path->packet_life_time -= (path->packet_life_time > 0);
2332 static int cm_lap_handler(struct cm_work *work)
2334 struct cm_id_private *cm_id_priv;
2335 struct cm_lap_msg *lap_msg;
2336 struct ib_cm_lap_event_param *param;
2337 struct ib_mad_send_buf *msg = NULL;
2338 unsigned long flags;
2339 int ret;
2341 /* todo: verify LAP request and send reject APR if invalid. */
2342 lap_msg = (struct cm_lap_msg *)work->mad_recv_wc->recv_buf.mad;
2343 cm_id_priv = cm_acquire_id(lap_msg->remote_comm_id,
2344 lap_msg->local_comm_id);
2345 if (!cm_id_priv)
2346 return -EINVAL;
2348 param = &work->cm_event.param.lap_rcvd;
2349 param->alternate_path = &work->path[0];
2350 cm_format_path_from_lap(param->alternate_path, lap_msg);
2351 work->cm_event.private_data = &lap_msg->private_data;
2353 spin_lock_irqsave(&cm_id_priv->lock, flags);
2354 if (cm_id_priv->id.state != IB_CM_ESTABLISHED)
2355 goto unlock;
2357 switch (cm_id_priv->id.lap_state) {
2358 case IB_CM_LAP_IDLE:
2359 break;
2360 case IB_CM_MRA_LAP_SENT:
2361 if (cm_alloc_response_msg(work->port, work->mad_recv_wc, &msg))
2362 goto unlock;
2364 cm_format_mra((struct cm_mra_msg *) msg->mad, cm_id_priv,
2365 CM_MSG_RESPONSE_OTHER,
2366 cm_id_priv->service_timeout,
2367 cm_id_priv->private_data,
2368 cm_id_priv->private_data_len);
2369 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2371 if (ib_post_send_mad(msg, NULL))
2372 cm_free_msg(msg);
2373 goto deref;
2374 default:
2375 goto unlock;
2378 cm_id_priv->id.lap_state = IB_CM_LAP_RCVD;
2379 cm_id_priv->tid = lap_msg->hdr.tid;
2380 ret = atomic_inc_and_test(&cm_id_priv->work_count);
2381 if (!ret)
2382 list_add_tail(&work->list, &cm_id_priv->work_list);
2383 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2385 if (ret)
2386 cm_process_work(cm_id_priv, work);
2387 else
2388 cm_deref_id(cm_id_priv);
2389 return 0;
2391 unlock: spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2392 deref: cm_deref_id(cm_id_priv);
2393 return -EINVAL;
2396 static void cm_format_apr(struct cm_apr_msg *apr_msg,
2397 struct cm_id_private *cm_id_priv,
2398 enum ib_cm_apr_status status,
2399 void *info,
2400 u8 info_length,
2401 const void *private_data,
2402 u8 private_data_len)
2404 cm_format_mad_hdr(&apr_msg->hdr, CM_APR_ATTR_ID, cm_id_priv->tid);
2405 apr_msg->local_comm_id = cm_id_priv->id.local_id;
2406 apr_msg->remote_comm_id = cm_id_priv->id.remote_id;
2407 apr_msg->ap_status = (u8) status;
2409 if (info && info_length) {
2410 apr_msg->info_length = info_length;
2411 memcpy(apr_msg->info, info, info_length);
2414 if (private_data && private_data_len)
2415 memcpy(apr_msg->private_data, private_data, private_data_len);
2418 int ib_send_cm_apr(struct ib_cm_id *cm_id,
2419 enum ib_cm_apr_status status,
2420 void *info,
2421 u8 info_length,
2422 const void *private_data,
2423 u8 private_data_len)
2425 struct cm_id_private *cm_id_priv;
2426 struct ib_mad_send_buf *msg;
2427 unsigned long flags;
2428 int ret;
2430 if ((private_data && private_data_len > IB_CM_APR_PRIVATE_DATA_SIZE) ||
2431 (info && info_length > IB_CM_APR_INFO_LENGTH))
2432 return -EINVAL;
2434 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2435 spin_lock_irqsave(&cm_id_priv->lock, flags);
2436 if (cm_id->state != IB_CM_ESTABLISHED ||
2437 (cm_id->lap_state != IB_CM_LAP_RCVD &&
2438 cm_id->lap_state != IB_CM_MRA_LAP_SENT)) {
2439 ret = -EINVAL;
2440 goto out;
2443 ret = cm_alloc_msg(cm_id_priv, &msg);
2444 if (ret)
2445 goto out;
2447 cm_format_apr((struct cm_apr_msg *) msg->mad, cm_id_priv, status,
2448 info, info_length, private_data, private_data_len);
2449 ret = ib_post_send_mad(msg, NULL);
2450 if (ret) {
2451 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2452 cm_free_msg(msg);
2453 return ret;
2456 cm_id->lap_state = IB_CM_LAP_IDLE;
2457 out: spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2458 return ret;
2460 EXPORT_SYMBOL(ib_send_cm_apr);
2462 static int cm_apr_handler(struct cm_work *work)
2464 struct cm_id_private *cm_id_priv;
2465 struct cm_apr_msg *apr_msg;
2466 unsigned long flags;
2467 int ret;
2469 apr_msg = (struct cm_apr_msg *)work->mad_recv_wc->recv_buf.mad;
2470 cm_id_priv = cm_acquire_id(apr_msg->remote_comm_id,
2471 apr_msg->local_comm_id);
2472 if (!cm_id_priv)
2473 return -EINVAL; /* Unmatched reply. */
2475 work->cm_event.param.apr_rcvd.ap_status = apr_msg->ap_status;
2476 work->cm_event.param.apr_rcvd.apr_info = &apr_msg->info;
2477 work->cm_event.param.apr_rcvd.info_len = apr_msg->info_length;
2478 work->cm_event.private_data = &apr_msg->private_data;
2480 spin_lock_irqsave(&cm_id_priv->lock, flags);
2481 if (cm_id_priv->id.state != IB_CM_ESTABLISHED ||
2482 (cm_id_priv->id.lap_state != IB_CM_LAP_SENT &&
2483 cm_id_priv->id.lap_state != IB_CM_MRA_LAP_RCVD)) {
2484 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2485 goto out;
2487 cm_id_priv->id.lap_state = IB_CM_LAP_IDLE;
2488 ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2489 cm_id_priv->msg = NULL;
2491 ret = atomic_inc_and_test(&cm_id_priv->work_count);
2492 if (!ret)
2493 list_add_tail(&work->list, &cm_id_priv->work_list);
2494 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2496 if (ret)
2497 cm_process_work(cm_id_priv, work);
2498 else
2499 cm_deref_id(cm_id_priv);
2500 return 0;
2501 out:
2502 cm_deref_id(cm_id_priv);
2503 return -EINVAL;
2506 static int cm_timewait_handler(struct cm_work *work)
2508 struct cm_timewait_info *timewait_info;
2509 struct cm_id_private *cm_id_priv;
2510 unsigned long flags;
2511 int ret;
2513 timewait_info = (struct cm_timewait_info *)work;
2514 cm_cleanup_timewait(timewait_info);
2516 cm_id_priv = cm_acquire_id(timewait_info->work.local_id,
2517 timewait_info->work.remote_id);
2518 if (!cm_id_priv)
2519 return -EINVAL;
2521 spin_lock_irqsave(&cm_id_priv->lock, flags);
2522 if (cm_id_priv->id.state != IB_CM_TIMEWAIT ||
2523 cm_id_priv->remote_qpn != timewait_info->remote_qpn) {
2524 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2525 goto out;
2527 cm_id_priv->id.state = IB_CM_IDLE;
2528 ret = atomic_inc_and_test(&cm_id_priv->work_count);
2529 if (!ret)
2530 list_add_tail(&work->list, &cm_id_priv->work_list);
2531 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2533 if (ret)
2534 cm_process_work(cm_id_priv, work);
2535 else
2536 cm_deref_id(cm_id_priv);
2537 return 0;
2538 out:
2539 cm_deref_id(cm_id_priv);
2540 return -EINVAL;
2543 static void cm_format_sidr_req(struct cm_sidr_req_msg *sidr_req_msg,
2544 struct cm_id_private *cm_id_priv,
2545 struct ib_cm_sidr_req_param *param)
2547 cm_format_mad_hdr(&sidr_req_msg->hdr, CM_SIDR_REQ_ATTR_ID,
2548 cm_form_tid(cm_id_priv, CM_MSG_SEQUENCE_SIDR));
2549 sidr_req_msg->request_id = cm_id_priv->id.local_id;
2550 sidr_req_msg->pkey = cpu_to_be16(param->pkey);
2551 sidr_req_msg->service_id = param->service_id;
2553 if (param->private_data && param->private_data_len)
2554 memcpy(sidr_req_msg->private_data, param->private_data,
2555 param->private_data_len);
2558 int ib_send_cm_sidr_req(struct ib_cm_id *cm_id,
2559 struct ib_cm_sidr_req_param *param)
2561 struct cm_id_private *cm_id_priv;
2562 struct ib_mad_send_buf *msg;
2563 unsigned long flags;
2564 int ret;
2566 if (!param->path || (param->private_data &&
2567 param->private_data_len > IB_CM_SIDR_REQ_PRIVATE_DATA_SIZE))
2568 return -EINVAL;
2570 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2571 ret = cm_init_av_by_path(param->path, &cm_id_priv->av);
2572 if (ret)
2573 goto out;
2575 cm_id->service_id = param->service_id;
2576 cm_id->service_mask = __constant_cpu_to_be64(~0ULL);
2577 cm_id_priv->timeout_ms = param->timeout_ms;
2578 cm_id_priv->max_cm_retries = param->max_cm_retries;
2579 ret = cm_alloc_msg(cm_id_priv, &msg);
2580 if (ret)
2581 goto out;
2583 cm_format_sidr_req((struct cm_sidr_req_msg *) msg->mad, cm_id_priv,
2584 param);
2585 msg->timeout_ms = cm_id_priv->timeout_ms;
2586 msg->context[1] = (void *) (unsigned long) IB_CM_SIDR_REQ_SENT;
2588 spin_lock_irqsave(&cm_id_priv->lock, flags);
2589 if (cm_id->state == IB_CM_IDLE)
2590 ret = ib_post_send_mad(msg, NULL);
2591 else
2592 ret = -EINVAL;
2594 if (ret) {
2595 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2596 cm_free_msg(msg);
2597 goto out;
2599 cm_id->state = IB_CM_SIDR_REQ_SENT;
2600 cm_id_priv->msg = msg;
2601 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2602 out:
2603 return ret;
2605 EXPORT_SYMBOL(ib_send_cm_sidr_req);
2607 static void cm_format_sidr_req_event(struct cm_work *work,
2608 struct ib_cm_id *listen_id)
2610 struct cm_sidr_req_msg *sidr_req_msg;
2611 struct ib_cm_sidr_req_event_param *param;
2613 sidr_req_msg = (struct cm_sidr_req_msg *)
2614 work->mad_recv_wc->recv_buf.mad;
2615 param = &work->cm_event.param.sidr_req_rcvd;
2616 param->pkey = __be16_to_cpu(sidr_req_msg->pkey);
2617 param->listen_id = listen_id;
2618 param->port = work->port->port_num;
2619 work->cm_event.private_data = &sidr_req_msg->private_data;
2622 static int cm_sidr_req_handler(struct cm_work *work)
2624 struct ib_cm_id *cm_id;
2625 struct cm_id_private *cm_id_priv, *cur_cm_id_priv;
2626 struct cm_sidr_req_msg *sidr_req_msg;
2627 struct ib_wc *wc;
2628 unsigned long flags;
2630 cm_id = ib_create_cm_id(work->port->cm_dev->device, NULL, NULL);
2631 if (IS_ERR(cm_id))
2632 return PTR_ERR(cm_id);
2633 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2635 /* Record SGID/SLID and request ID for lookup. */
2636 sidr_req_msg = (struct cm_sidr_req_msg *)
2637 work->mad_recv_wc->recv_buf.mad;
2638 wc = work->mad_recv_wc->wc;
2639 cm_id_priv->av.dgid.global.subnet_prefix = cpu_to_be64(wc->slid);
2640 cm_id_priv->av.dgid.global.interface_id = 0;
2641 cm_init_av_for_response(work->port, work->mad_recv_wc->wc,
2642 &cm_id_priv->av);
2643 cm_id_priv->id.remote_id = sidr_req_msg->request_id;
2644 cm_id_priv->id.state = IB_CM_SIDR_REQ_RCVD;
2645 cm_id_priv->tid = sidr_req_msg->hdr.tid;
2646 atomic_inc(&cm_id_priv->work_count);
2648 spin_lock_irqsave(&cm.lock, flags);
2649 cur_cm_id_priv = cm_insert_remote_sidr(cm_id_priv);
2650 if (cur_cm_id_priv) {
2651 spin_unlock_irqrestore(&cm.lock, flags);
2652 goto out; /* Duplicate message. */
2654 cur_cm_id_priv = cm_find_listen(cm_id->device,
2655 sidr_req_msg->service_id);
2656 if (!cur_cm_id_priv) {
2657 rb_erase(&cm_id_priv->sidr_id_node, &cm.remote_sidr_table);
2658 spin_unlock_irqrestore(&cm.lock, flags);
2659 /* todo: reply with no match */
2660 goto out; /* No match. */
2662 atomic_inc(&cur_cm_id_priv->refcount);
2663 spin_unlock_irqrestore(&cm.lock, flags);
2665 cm_id_priv->id.cm_handler = cur_cm_id_priv->id.cm_handler;
2666 cm_id_priv->id.context = cur_cm_id_priv->id.context;
2667 cm_id_priv->id.service_id = sidr_req_msg->service_id;
2668 cm_id_priv->id.service_mask = __constant_cpu_to_be64(~0ULL);
2670 cm_format_sidr_req_event(work, &cur_cm_id_priv->id);
2671 cm_process_work(cm_id_priv, work);
2672 cm_deref_id(cur_cm_id_priv);
2673 return 0;
2674 out:
2675 ib_destroy_cm_id(&cm_id_priv->id);
2676 return -EINVAL;
2679 static void cm_format_sidr_rep(struct cm_sidr_rep_msg *sidr_rep_msg,
2680 struct cm_id_private *cm_id_priv,
2681 struct ib_cm_sidr_rep_param *param)
2683 cm_format_mad_hdr(&sidr_rep_msg->hdr, CM_SIDR_REP_ATTR_ID,
2684 cm_id_priv->tid);
2685 sidr_rep_msg->request_id = cm_id_priv->id.remote_id;
2686 sidr_rep_msg->status = param->status;
2687 cm_sidr_rep_set_qpn(sidr_rep_msg, cpu_to_be32(param->qp_num));
2688 sidr_rep_msg->service_id = cm_id_priv->id.service_id;
2689 sidr_rep_msg->qkey = cpu_to_be32(param->qkey);
2691 if (param->info && param->info_length)
2692 memcpy(sidr_rep_msg->info, param->info, param->info_length);
2694 if (param->private_data && param->private_data_len)
2695 memcpy(sidr_rep_msg->private_data, param->private_data,
2696 param->private_data_len);
2699 int ib_send_cm_sidr_rep(struct ib_cm_id *cm_id,
2700 struct ib_cm_sidr_rep_param *param)
2702 struct cm_id_private *cm_id_priv;
2703 struct ib_mad_send_buf *msg;
2704 unsigned long flags;
2705 int ret;
2707 if ((param->info && param->info_length > IB_CM_SIDR_REP_INFO_LENGTH) ||
2708 (param->private_data &&
2709 param->private_data_len > IB_CM_SIDR_REP_PRIVATE_DATA_SIZE))
2710 return -EINVAL;
2712 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2713 spin_lock_irqsave(&cm_id_priv->lock, flags);
2714 if (cm_id->state != IB_CM_SIDR_REQ_RCVD) {
2715 ret = -EINVAL;
2716 goto error;
2719 ret = cm_alloc_msg(cm_id_priv, &msg);
2720 if (ret)
2721 goto error;
2723 cm_format_sidr_rep((struct cm_sidr_rep_msg *) msg->mad, cm_id_priv,
2724 param);
2725 ret = ib_post_send_mad(msg, NULL);
2726 if (ret) {
2727 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2728 cm_free_msg(msg);
2729 return ret;
2731 cm_id->state = IB_CM_IDLE;
2732 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2734 spin_lock_irqsave(&cm.lock, flags);
2735 rb_erase(&cm_id_priv->sidr_id_node, &cm.remote_sidr_table);
2736 spin_unlock_irqrestore(&cm.lock, flags);
2737 return 0;
2739 error: spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2740 return ret;
2742 EXPORT_SYMBOL(ib_send_cm_sidr_rep);
2744 static void cm_format_sidr_rep_event(struct cm_work *work)
2746 struct cm_sidr_rep_msg *sidr_rep_msg;
2747 struct ib_cm_sidr_rep_event_param *param;
2749 sidr_rep_msg = (struct cm_sidr_rep_msg *)
2750 work->mad_recv_wc->recv_buf.mad;
2751 param = &work->cm_event.param.sidr_rep_rcvd;
2752 param->status = sidr_rep_msg->status;
2753 param->qkey = be32_to_cpu(sidr_rep_msg->qkey);
2754 param->qpn = be32_to_cpu(cm_sidr_rep_get_qpn(sidr_rep_msg));
2755 param->info = &sidr_rep_msg->info;
2756 param->info_len = sidr_rep_msg->info_length;
2757 work->cm_event.private_data = &sidr_rep_msg->private_data;
2760 static int cm_sidr_rep_handler(struct cm_work *work)
2762 struct cm_sidr_rep_msg *sidr_rep_msg;
2763 struct cm_id_private *cm_id_priv;
2764 unsigned long flags;
2766 sidr_rep_msg = (struct cm_sidr_rep_msg *)
2767 work->mad_recv_wc->recv_buf.mad;
2768 cm_id_priv = cm_acquire_id(sidr_rep_msg->request_id, 0);
2769 if (!cm_id_priv)
2770 return -EINVAL; /* Unmatched reply. */
2772 spin_lock_irqsave(&cm_id_priv->lock, flags);
2773 if (cm_id_priv->id.state != IB_CM_SIDR_REQ_SENT) {
2774 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2775 goto out;
2777 cm_id_priv->id.state = IB_CM_IDLE;
2778 ib_cancel_mad(cm_id_priv->av.port->mad_agent, cm_id_priv->msg);
2779 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2781 cm_format_sidr_rep_event(work);
2782 cm_process_work(cm_id_priv, work);
2783 return 0;
2784 out:
2785 cm_deref_id(cm_id_priv);
2786 return -EINVAL;
2789 static void cm_process_send_error(struct ib_mad_send_buf *msg,
2790 enum ib_wc_status wc_status)
2792 struct cm_id_private *cm_id_priv;
2793 struct ib_cm_event cm_event;
2794 enum ib_cm_state state;
2795 unsigned long flags;
2796 int ret;
2798 memset(&cm_event, 0, sizeof cm_event);
2799 cm_id_priv = msg->context[0];
2801 /* Discard old sends or ones without a response. */
2802 spin_lock_irqsave(&cm_id_priv->lock, flags);
2803 state = (enum ib_cm_state) (unsigned long) msg->context[1];
2804 if (msg != cm_id_priv->msg || state != cm_id_priv->id.state)
2805 goto discard;
2807 switch (state) {
2808 case IB_CM_REQ_SENT:
2809 case IB_CM_MRA_REQ_RCVD:
2810 cm_reset_to_idle(cm_id_priv);
2811 cm_event.event = IB_CM_REQ_ERROR;
2812 break;
2813 case IB_CM_REP_SENT:
2814 case IB_CM_MRA_REP_RCVD:
2815 cm_reset_to_idle(cm_id_priv);
2816 cm_event.event = IB_CM_REP_ERROR;
2817 break;
2818 case IB_CM_DREQ_SENT:
2819 cm_enter_timewait(cm_id_priv);
2820 cm_event.event = IB_CM_DREQ_ERROR;
2821 break;
2822 case IB_CM_SIDR_REQ_SENT:
2823 cm_id_priv->id.state = IB_CM_IDLE;
2824 cm_event.event = IB_CM_SIDR_REQ_ERROR;
2825 break;
2826 default:
2827 goto discard;
2829 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2830 cm_event.param.send_status = wc_status;
2832 /* No other events can occur on the cm_id at this point. */
2833 ret = cm_id_priv->id.cm_handler(&cm_id_priv->id, &cm_event);
2834 cm_free_msg(msg);
2835 if (ret)
2836 ib_destroy_cm_id(&cm_id_priv->id);
2837 return;
2838 discard:
2839 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2840 cm_free_msg(msg);
2843 static void cm_send_handler(struct ib_mad_agent *mad_agent,
2844 struct ib_mad_send_wc *mad_send_wc)
2846 struct ib_mad_send_buf *msg = mad_send_wc->send_buf;
2848 switch (mad_send_wc->status) {
2849 case IB_WC_SUCCESS:
2850 case IB_WC_WR_FLUSH_ERR:
2851 cm_free_msg(msg);
2852 break;
2853 default:
2854 if (msg->context[0] && msg->context[1])
2855 cm_process_send_error(msg, mad_send_wc->status);
2856 else
2857 cm_free_msg(msg);
2858 break;
2862 static void cm_work_handler(void *data)
2864 struct cm_work *work = data;
2865 int ret;
2867 switch (work->cm_event.event) {
2868 case IB_CM_REQ_RECEIVED:
2869 ret = cm_req_handler(work);
2870 break;
2871 case IB_CM_MRA_RECEIVED:
2872 ret = cm_mra_handler(work);
2873 break;
2874 case IB_CM_REJ_RECEIVED:
2875 ret = cm_rej_handler(work);
2876 break;
2877 case IB_CM_REP_RECEIVED:
2878 ret = cm_rep_handler(work);
2879 break;
2880 case IB_CM_RTU_RECEIVED:
2881 ret = cm_rtu_handler(work);
2882 break;
2883 case IB_CM_USER_ESTABLISHED:
2884 ret = cm_establish_handler(work);
2885 break;
2886 case IB_CM_DREQ_RECEIVED:
2887 ret = cm_dreq_handler(work);
2888 break;
2889 case IB_CM_DREP_RECEIVED:
2890 ret = cm_drep_handler(work);
2891 break;
2892 case IB_CM_SIDR_REQ_RECEIVED:
2893 ret = cm_sidr_req_handler(work);
2894 break;
2895 case IB_CM_SIDR_REP_RECEIVED:
2896 ret = cm_sidr_rep_handler(work);
2897 break;
2898 case IB_CM_LAP_RECEIVED:
2899 ret = cm_lap_handler(work);
2900 break;
2901 case IB_CM_APR_RECEIVED:
2902 ret = cm_apr_handler(work);
2903 break;
2904 case IB_CM_TIMEWAIT_EXIT:
2905 ret = cm_timewait_handler(work);
2906 break;
2907 default:
2908 ret = -EINVAL;
2909 break;
2911 if (ret)
2912 cm_free_work(work);
2915 int ib_cm_establish(struct ib_cm_id *cm_id)
2917 struct cm_id_private *cm_id_priv;
2918 struct cm_work *work;
2919 unsigned long flags;
2920 int ret = 0;
2922 work = kmalloc(sizeof *work, GFP_ATOMIC);
2923 if (!work)
2924 return -ENOMEM;
2926 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
2927 spin_lock_irqsave(&cm_id_priv->lock, flags);
2928 switch (cm_id->state)
2930 case IB_CM_REP_SENT:
2931 case IB_CM_MRA_REP_RCVD:
2932 cm_id->state = IB_CM_ESTABLISHED;
2933 break;
2934 case IB_CM_ESTABLISHED:
2935 ret = -EISCONN;
2936 break;
2937 default:
2938 ret = -EINVAL;
2939 break;
2941 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
2943 if (ret) {
2944 kfree(work);
2945 goto out;
2949 * The CM worker thread may try to destroy the cm_id before it
2950 * can execute this work item. To prevent potential deadlock,
2951 * we need to find the cm_id once we're in the context of the
2952 * worker thread, rather than holding a reference on it.
2954 INIT_WORK(&work->work, cm_work_handler, work);
2955 work->local_id = cm_id->local_id;
2956 work->remote_id = cm_id->remote_id;
2957 work->mad_recv_wc = NULL;
2958 work->cm_event.event = IB_CM_USER_ESTABLISHED;
2959 queue_work(cm.wq, &work->work);
2960 out:
2961 return ret;
2963 EXPORT_SYMBOL(ib_cm_establish);
2965 static void cm_recv_handler(struct ib_mad_agent *mad_agent,
2966 struct ib_mad_recv_wc *mad_recv_wc)
2968 struct cm_work *work;
2969 enum ib_cm_event_type event;
2970 int paths = 0;
2972 switch (mad_recv_wc->recv_buf.mad->mad_hdr.attr_id) {
2973 case CM_REQ_ATTR_ID:
2974 paths = 1 + (((struct cm_req_msg *) mad_recv_wc->recv_buf.mad)->
2975 alt_local_lid != 0);
2976 event = IB_CM_REQ_RECEIVED;
2977 break;
2978 case CM_MRA_ATTR_ID:
2979 event = IB_CM_MRA_RECEIVED;
2980 break;
2981 case CM_REJ_ATTR_ID:
2982 event = IB_CM_REJ_RECEIVED;
2983 break;
2984 case CM_REP_ATTR_ID:
2985 event = IB_CM_REP_RECEIVED;
2986 break;
2987 case CM_RTU_ATTR_ID:
2988 event = IB_CM_RTU_RECEIVED;
2989 break;
2990 case CM_DREQ_ATTR_ID:
2991 event = IB_CM_DREQ_RECEIVED;
2992 break;
2993 case CM_DREP_ATTR_ID:
2994 event = IB_CM_DREP_RECEIVED;
2995 break;
2996 case CM_SIDR_REQ_ATTR_ID:
2997 event = IB_CM_SIDR_REQ_RECEIVED;
2998 break;
2999 case CM_SIDR_REP_ATTR_ID:
3000 event = IB_CM_SIDR_REP_RECEIVED;
3001 break;
3002 case CM_LAP_ATTR_ID:
3003 paths = 1;
3004 event = IB_CM_LAP_RECEIVED;
3005 break;
3006 case CM_APR_ATTR_ID:
3007 event = IB_CM_APR_RECEIVED;
3008 break;
3009 default:
3010 ib_free_recv_mad(mad_recv_wc);
3011 return;
3014 work = kmalloc(sizeof *work + sizeof(struct ib_sa_path_rec) * paths,
3015 GFP_KERNEL);
3016 if (!work) {
3017 ib_free_recv_mad(mad_recv_wc);
3018 return;
3021 INIT_WORK(&work->work, cm_work_handler, work);
3022 work->cm_event.event = event;
3023 work->mad_recv_wc = mad_recv_wc;
3024 work->port = (struct cm_port *)mad_agent->context;
3025 queue_work(cm.wq, &work->work);
3028 static int cm_init_qp_init_attr(struct cm_id_private *cm_id_priv,
3029 struct ib_qp_attr *qp_attr,
3030 int *qp_attr_mask)
3032 unsigned long flags;
3033 int ret;
3035 spin_lock_irqsave(&cm_id_priv->lock, flags);
3036 switch (cm_id_priv->id.state) {
3037 case IB_CM_REQ_SENT:
3038 case IB_CM_MRA_REQ_RCVD:
3039 case IB_CM_REQ_RCVD:
3040 case IB_CM_MRA_REQ_SENT:
3041 case IB_CM_REP_RCVD:
3042 case IB_CM_MRA_REP_SENT:
3043 case IB_CM_REP_SENT:
3044 case IB_CM_MRA_REP_RCVD:
3045 case IB_CM_ESTABLISHED:
3046 *qp_attr_mask = IB_QP_STATE | IB_QP_ACCESS_FLAGS |
3047 IB_QP_PKEY_INDEX | IB_QP_PORT;
3048 qp_attr->qp_access_flags = IB_ACCESS_LOCAL_WRITE |
3049 IB_ACCESS_REMOTE_WRITE;
3050 if (cm_id_priv->responder_resources)
3051 qp_attr->qp_access_flags |= IB_ACCESS_REMOTE_READ;
3052 qp_attr->pkey_index = cm_id_priv->av.pkey_index;
3053 qp_attr->port_num = cm_id_priv->av.port->port_num;
3054 ret = 0;
3055 break;
3056 default:
3057 ret = -EINVAL;
3058 break;
3060 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3061 return ret;
3064 static int cm_init_qp_rtr_attr(struct cm_id_private *cm_id_priv,
3065 struct ib_qp_attr *qp_attr,
3066 int *qp_attr_mask)
3068 unsigned long flags;
3069 int ret;
3071 spin_lock_irqsave(&cm_id_priv->lock, flags);
3072 switch (cm_id_priv->id.state) {
3073 case IB_CM_REQ_RCVD:
3074 case IB_CM_MRA_REQ_SENT:
3075 case IB_CM_REP_RCVD:
3076 case IB_CM_MRA_REP_SENT:
3077 case IB_CM_REP_SENT:
3078 case IB_CM_MRA_REP_RCVD:
3079 case IB_CM_ESTABLISHED:
3080 *qp_attr_mask = IB_QP_STATE | IB_QP_AV | IB_QP_PATH_MTU |
3081 IB_QP_DEST_QPN | IB_QP_RQ_PSN;
3082 qp_attr->ah_attr = cm_id_priv->av.ah_attr;
3083 qp_attr->path_mtu = cm_id_priv->path_mtu;
3084 qp_attr->dest_qp_num = be32_to_cpu(cm_id_priv->remote_qpn);
3085 qp_attr->rq_psn = be32_to_cpu(cm_id_priv->rq_psn);
3086 if (cm_id_priv->qp_type == IB_QPT_RC) {
3087 *qp_attr_mask |= IB_QP_MAX_DEST_RD_ATOMIC |
3088 IB_QP_MIN_RNR_TIMER;
3089 qp_attr->max_dest_rd_atomic =
3090 cm_id_priv->responder_resources;
3091 qp_attr->min_rnr_timer = 0;
3093 if (cm_id_priv->alt_av.ah_attr.dlid) {
3094 *qp_attr_mask |= IB_QP_ALT_PATH;
3095 qp_attr->alt_ah_attr = cm_id_priv->alt_av.ah_attr;
3097 ret = 0;
3098 break;
3099 default:
3100 ret = -EINVAL;
3101 break;
3103 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3104 return ret;
3107 static int cm_init_qp_rts_attr(struct cm_id_private *cm_id_priv,
3108 struct ib_qp_attr *qp_attr,
3109 int *qp_attr_mask)
3111 unsigned long flags;
3112 int ret;
3114 spin_lock_irqsave(&cm_id_priv->lock, flags);
3115 switch (cm_id_priv->id.state) {
3116 case IB_CM_REP_RCVD:
3117 case IB_CM_MRA_REP_SENT:
3118 case IB_CM_REP_SENT:
3119 case IB_CM_MRA_REP_RCVD:
3120 case IB_CM_ESTABLISHED:
3121 *qp_attr_mask = IB_QP_STATE | IB_QP_SQ_PSN;
3122 qp_attr->sq_psn = be32_to_cpu(cm_id_priv->sq_psn);
3123 if (cm_id_priv->qp_type == IB_QPT_RC) {
3124 *qp_attr_mask |= IB_QP_TIMEOUT | IB_QP_RETRY_CNT |
3125 IB_QP_RNR_RETRY |
3126 IB_QP_MAX_QP_RD_ATOMIC;
3127 qp_attr->timeout = cm_id_priv->local_ack_timeout;
3128 qp_attr->retry_cnt = cm_id_priv->retry_count;
3129 qp_attr->rnr_retry = cm_id_priv->rnr_retry_count;
3130 qp_attr->max_rd_atomic = cm_id_priv->initiator_depth;
3132 if (cm_id_priv->alt_av.ah_attr.dlid) {
3133 *qp_attr_mask |= IB_QP_PATH_MIG_STATE;
3134 qp_attr->path_mig_state = IB_MIG_REARM;
3136 ret = 0;
3137 break;
3138 default:
3139 ret = -EINVAL;
3140 break;
3142 spin_unlock_irqrestore(&cm_id_priv->lock, flags);
3143 return ret;
3146 int ib_cm_init_qp_attr(struct ib_cm_id *cm_id,
3147 struct ib_qp_attr *qp_attr,
3148 int *qp_attr_mask)
3150 struct cm_id_private *cm_id_priv;
3151 int ret;
3153 cm_id_priv = container_of(cm_id, struct cm_id_private, id);
3154 switch (qp_attr->qp_state) {
3155 case IB_QPS_INIT:
3156 ret = cm_init_qp_init_attr(cm_id_priv, qp_attr, qp_attr_mask);
3157 break;
3158 case IB_QPS_RTR:
3159 ret = cm_init_qp_rtr_attr(cm_id_priv, qp_attr, qp_attr_mask);
3160 break;
3161 case IB_QPS_RTS:
3162 ret = cm_init_qp_rts_attr(cm_id_priv, qp_attr, qp_attr_mask);
3163 break;
3164 default:
3165 ret = -EINVAL;
3166 break;
3168 return ret;
3170 EXPORT_SYMBOL(ib_cm_init_qp_attr);
3172 static void cm_add_one(struct ib_device *device)
3174 struct cm_device *cm_dev;
3175 struct cm_port *port;
3176 struct ib_mad_reg_req reg_req = {
3177 .mgmt_class = IB_MGMT_CLASS_CM,
3178 .mgmt_class_version = IB_CM_CLASS_VERSION
3180 struct ib_port_modify port_modify = {
3181 .set_port_cap_mask = IB_PORT_CM_SUP
3183 unsigned long flags;
3184 int ret;
3185 u8 i;
3187 cm_dev = kmalloc(sizeof(*cm_dev) + sizeof(*port) *
3188 device->phys_port_cnt, GFP_KERNEL);
3189 if (!cm_dev)
3190 return;
3192 cm_dev->device = device;
3193 cm_dev->ca_guid = device->node_guid;
3195 set_bit(IB_MGMT_METHOD_SEND, reg_req.method_mask);
3196 for (i = 1; i <= device->phys_port_cnt; i++) {
3197 port = &cm_dev->port[i-1];
3198 port->cm_dev = cm_dev;
3199 port->port_num = i;
3200 port->mad_agent = ib_register_mad_agent(device, i,
3201 IB_QPT_GSI,
3202 &reg_req,
3204 cm_send_handler,
3205 cm_recv_handler,
3206 port);
3207 if (IS_ERR(port->mad_agent))
3208 goto error1;
3210 ret = ib_modify_port(device, i, 0, &port_modify);
3211 if (ret)
3212 goto error2;
3214 ib_set_client_data(device, &cm_client, cm_dev);
3216 write_lock_irqsave(&cm.device_lock, flags);
3217 list_add_tail(&cm_dev->list, &cm.device_list);
3218 write_unlock_irqrestore(&cm.device_lock, flags);
3219 return;
3221 error2:
3222 ib_unregister_mad_agent(port->mad_agent);
3223 error1:
3224 port_modify.set_port_cap_mask = 0;
3225 port_modify.clr_port_cap_mask = IB_PORT_CM_SUP;
3226 while (--i) {
3227 port = &cm_dev->port[i-1];
3228 ib_modify_port(device, port->port_num, 0, &port_modify);
3229 ib_unregister_mad_agent(port->mad_agent);
3231 kfree(cm_dev);
3234 static void cm_remove_one(struct ib_device *device)
3236 struct cm_device *cm_dev;
3237 struct cm_port *port;
3238 struct ib_port_modify port_modify = {
3239 .clr_port_cap_mask = IB_PORT_CM_SUP
3241 unsigned long flags;
3242 int i;
3244 cm_dev = ib_get_client_data(device, &cm_client);
3245 if (!cm_dev)
3246 return;
3248 write_lock_irqsave(&cm.device_lock, flags);
3249 list_del(&cm_dev->list);
3250 write_unlock_irqrestore(&cm.device_lock, flags);
3252 for (i = 1; i <= device->phys_port_cnt; i++) {
3253 port = &cm_dev->port[i-1];
3254 ib_modify_port(device, port->port_num, 0, &port_modify);
3255 ib_unregister_mad_agent(port->mad_agent);
3257 kfree(cm_dev);
3260 static int __init ib_cm_init(void)
3262 int ret;
3264 memset(&cm, 0, sizeof cm);
3265 INIT_LIST_HEAD(&cm.device_list);
3266 rwlock_init(&cm.device_lock);
3267 spin_lock_init(&cm.lock);
3268 cm.listen_service_table = RB_ROOT;
3269 cm.listen_service_id = __constant_be64_to_cpu(IB_CM_ASSIGN_SERVICE_ID);
3270 cm.remote_id_table = RB_ROOT;
3271 cm.remote_qp_table = RB_ROOT;
3272 cm.remote_sidr_table = RB_ROOT;
3273 idr_init(&cm.local_id_table);
3274 idr_pre_get(&cm.local_id_table, GFP_KERNEL);
3276 cm.wq = create_workqueue("ib_cm");
3277 if (!cm.wq)
3278 return -ENOMEM;
3280 ret = ib_register_client(&cm_client);
3281 if (ret)
3282 goto error;
3284 return 0;
3285 error:
3286 destroy_workqueue(cm.wq);
3287 return ret;
3290 static void __exit ib_cm_cleanup(void)
3292 flush_workqueue(cm.wq);
3293 destroy_workqueue(cm.wq);
3294 ib_unregister_client(&cm_client);
3295 idr_destroy(&cm.local_id_table);
3298 module_init(ib_cm_init);
3299 module_exit(ib_cm_cleanup);