Added an almost obscene amount of debugging and assertion code while
[people/pcmattman/gpxe.git] / src / drivers / net / mlx_ipoib / mt25218.c
1 /**************************************************************************
2 Etherboot -  BOOTP/TFTP Bootstrap Program
3 Skeleton NIC driver for Etherboot
4 ***************************************************************************/
5
6 /*
7  * This program is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU General Public License as
9  * published by the Free Software Foundation; either version 2, or (at
10  * your option) any later version.
11  */
12
13 #include <errno.h>
14 #include <gpxe/pci.h>
15 #include <gpxe/malloc.h>
16 #include <gpxe/iobuf.h>
17 #include <gpxe/netdevice.h>
18 #include <gpxe/infiniband.h>
19 #include <gpxe/ipoib.h>
20
21 /* to get some global routines like printf */
22 #include "etherboot.h"
23 /* to get the interface to the body of the program */
24 #include "nic.h"
25
26 #define CREATE_OWN 1
27
28 #include "mt25218_imp.c"
29
30 #include "arbel.h"
31
32
33 struct ib_address_vector hack_ipoib_bcast_av;
34
35
36
37
38
39
40
41
42 /***************************************************************************
43  *
44  * Queue number allocation
45  *
46  ***************************************************************************
47  */
48
49 /**
50  * Allocate queue number
51  *
52  * @v q_inuse           Queue usage bitmask
53  * @v max_inuse         Maximum number of in-use queues
54  * @ret qn_offset       Free queue number offset, or negative error
55  */
56 static int arbel_alloc_qn_offset ( arbel_bitmask_t *q_inuse,
57                                    unsigned int max_inuse ) {
58         unsigned int qn_offset = 0;
59         arbel_bitmask_t mask = 1;
60
61         while ( qn_offset < max_inuse ) {
62                 if ( ( mask & *q_inuse ) == 0 ) {
63                         *q_inuse |= mask;
64                         return qn_offset;
65                 }
66                 qn_offset++;
67                 mask <<= 1;
68                 if ( ! mask ) {
69                         mask = 1;
70                         q_inuse++;
71                 }
72         }
73         return -ENFILE;
74 }
75
76 /**
77  * Free queue number
78  *
79  * @v q_inuse           Queue usage bitmask
80  * @v qn_offset         Queue number offset
81  */
82 static void arbel_free_qn_offset ( arbel_bitmask_t *q_inuse, int qn_offset ) {
83         arbel_bitmask_t mask;
84
85         mask = ( 1 << ( qn_offset % ( 8 * sizeof ( mask ) ) ) );
86         q_inuse += ( qn_offset / ( 8 * sizeof ( mask ) ) );
87         *q_inuse &= ~mask;
88 }
89
90 /***************************************************************************
91  *
92  * HCA commands
93  *
94  ***************************************************************************
95  */
96
97 /**
98  * Wait for Arbel command completion
99  *
100  * @v arbel             Arbel device
101  * @ret rc              Return status code
102  */
103 static int arbel_cmd_wait ( struct arbel *arbel,
104                             struct arbelprm_hca_command_register *hcr ) {
105         unsigned int wait;
106
107         for ( wait = ARBEL_HCR_MAX_WAIT_MS ; wait ; wait-- ) {
108                 hcr->u.dwords[6] =
109                         readl ( arbel->config + ARBEL_HCR_REG ( 6 ) );
110                 if ( MLX_GET ( hcr, go ) == 0 )
111                         return 0;
112                 mdelay ( 1 );
113         }
114         return -EBUSY;
115 }
116
117 /**
118  * Issue HCA command
119  *
120  * @v arbel             Arbel device
121  * @v command           Command opcode, flags and input/output lengths
122  * @v op_mod            Opcode modifier (0 if no modifier applicable)
123  * @v in                Input parameters
124  * @v in_mod            Input modifier (0 if no modifier applicable)
125  * @v out               Output parameters
126  * @ret rc              Return status code
127  */
128 static int arbel_cmd ( struct arbel *arbel, unsigned long command,
129                        unsigned int op_mod, const void *in,
130                        unsigned int in_mod, void *out ) {
131         struct arbelprm_hca_command_register hcr;
132         unsigned int opcode = ARBEL_HCR_OPCODE ( command );
133         size_t in_len = ARBEL_HCR_IN_LEN ( command );
134         size_t out_len = ARBEL_HCR_OUT_LEN ( command );
135         void *in_buffer;
136         void *out_buffer;
137         unsigned int status;
138         unsigned int i;
139         int rc;
140
141         DBGC2 ( arbel, "Arbel %p command %02x in %zx%s out %zx%s\n",
142                 arbel, opcode, in_len,
143                 ( ( command & ARBEL_HCR_IN_MBOX ) ? "(mbox)" : "" ), out_len,
144                 ( ( command & ARBEL_HCR_OUT_MBOX ) ? "(mbox)" : "" ) );
145
146         /* Check that HCR is free */
147         if ( ( rc = arbel_cmd_wait ( arbel, &hcr ) ) != 0 ) {
148                 DBGC ( arbel, "Arbel %p command interface locked\n", arbel );
149                 return rc;
150         }
151
152         /* Prepare HCR */
153         memset ( &hcr, 0, sizeof ( hcr ) );
154         in_buffer = &hcr.u.dwords[0];
155         if ( in_len && ( command & ARBEL_HCR_IN_MBOX ) ) {
156                 in_buffer = arbel->mailbox_in;
157                 MLX_FILL_1 ( &hcr, 1, in_param_l, virt_to_bus ( in_buffer ) );
158         }
159         memcpy ( in_buffer, in, in_len );
160         MLX_FILL_1 ( &hcr, 2, input_modifier, in_mod );
161         out_buffer = &hcr.u.dwords[3];
162         if ( out_len && ( command & ARBEL_HCR_OUT_MBOX ) ) {
163                 out_buffer = arbel->mailbox_out;
164                 MLX_FILL_1 ( &hcr, 4, out_param_l,
165                              virt_to_bus ( out_buffer ) );
166         }
167         MLX_FILL_3 ( &hcr, 6,
168                      opcode, opcode,
169                      opcode_modifier, op_mod,
170                      go, 1 );
171         DBGC2_HD ( arbel, &hcr, sizeof ( hcr ) );
172         if ( in_len ) {
173                 DBGC2 ( arbel, "Input:\n" );
174                 DBGC2_HD ( arbel, in, ( ( in_len < 256 ) ? in_len : 256 ) );
175         }
176
177         /* Issue command */
178         for ( i = 0 ; i < ( sizeof ( hcr ) / sizeof ( hcr.u.dwords[0] ) ) ;
179               i++ ) {
180                 writel ( hcr.u.dwords[i],
181                          arbel->config + ARBEL_HCR_REG ( i ) );
182                 barrier();
183         }
184
185         /* Wait for command completion */
186         if ( ( rc = arbel_cmd_wait ( arbel, &hcr ) ) != 0 ) {
187                 DBGC ( arbel, "Arbel %p timed out waiting for command:\n",
188                        arbel );
189                 DBGC_HD ( arbel, &hcr, sizeof ( hcr ) );
190                 return rc;
191         }
192
193         /* Check command status */
194         status = MLX_GET ( &hcr, status );
195         if ( status != 0 ) {
196                 DBGC ( arbel, "Arbel %p command failed with status %02x:\n",
197                        arbel, status );
198                 DBGC_HD ( arbel, &hcr, sizeof ( hcr ) );
199                 return -EIO;
200         }
201
202         /* Read output parameters, if any */
203         hcr.u.dwords[3] = readl ( arbel->config + ARBEL_HCR_REG ( 3 ) );
204         hcr.u.dwords[4] = readl ( arbel->config + ARBEL_HCR_REG ( 4 ) );
205         memcpy ( out, out_buffer, out_len );
206         if ( out_len ) {
207                 DBGC2 ( arbel, "Output:\n" );
208                 DBGC2_HD ( arbel, out, ( ( out_len < 256 ) ? out_len : 256 ) );
209         }
210
211         return 0;
212 }
213
214 static inline int
215 arbel_cmd_query_dev_lim ( struct arbel *arbel,
216                           struct arbelprm_query_dev_lim *dev_lim ) {
217         return arbel_cmd ( arbel,
218                            ARBEL_HCR_OUT_CMD ( ARBEL_HCR_QUERY_DEV_LIM, 
219                                                1, sizeof ( *dev_lim ) ),
220                            0, NULL, 0, dev_lim );
221 }
222
223 static inline int
224 arbel_cmd_sw2hw_cq ( struct arbel *arbel, unsigned long cqn,
225                      const struct arbelprm_completion_queue_context *cqctx ) {
226         return arbel_cmd ( arbel,
227                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_SW2HW_CQ,
228                                               1, sizeof ( *cqctx ) ),
229                            0, cqctx, cqn, NULL );
230 }
231
232 static inline int
233 arbel_cmd_hw2sw_cq ( struct arbel *arbel, unsigned long cqn ) {
234         return arbel_cmd ( arbel,
235                            ARBEL_HCR_VOID_CMD ( ARBEL_HCR_HW2SW_CQ ),
236                            1, NULL, cqn, NULL );
237 }
238
239 static inline int
240 arbel_cmd_rst2init_qpee ( struct arbel *arbel, unsigned long qpn,
241                           const struct arbelprm_qp_ee_state_transitions *ctx ){
242         return arbel_cmd ( arbel,
243                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_RST2INIT_QPEE,
244                                               1, sizeof ( *ctx ) ),
245                            0, ctx, qpn, NULL );
246 }
247
248 static inline int
249 arbel_cmd_init2rtr_qpee ( struct arbel *arbel, unsigned long qpn,
250                           const struct arbelprm_qp_ee_state_transitions *ctx ){
251         return arbel_cmd ( arbel,
252                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_INIT2RTR_QPEE,
253                                               1, sizeof ( *ctx ) ),
254                            0, ctx, qpn, NULL );
255 }
256
257 static inline int
258 arbel_cmd_rtr2rts_qpee ( struct arbel *arbel, unsigned long qpn,
259                          const struct arbelprm_qp_ee_state_transitions *ctx ) {
260         return arbel_cmd ( arbel,
261                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_RTR2RTS_QPEE,
262                                               1, sizeof ( *ctx ) ),
263                            0, ctx, qpn, NULL );
264 }
265
266 static inline int
267 arbel_cmd_2rst_qpee ( struct arbel *arbel, unsigned long qpn ) {
268         return arbel_cmd ( arbel,
269                            ARBEL_HCR_VOID_CMD ( ARBEL_HCR_2RST_QPEE ),
270                            0x03, NULL, qpn, NULL );
271 }
272
273 static inline int
274 arbel_cmd_mad_ifc ( struct arbel *arbel, union arbelprm_mad *mad ) {
275         return arbel_cmd ( arbel,
276                            ARBEL_HCR_INOUT_CMD ( ARBEL_HCR_MAD_IFC,
277                                                  1, sizeof ( *mad ),
278                                                  1, sizeof ( *mad ) ),
279                            0x03, mad, PXE_IB_PORT, mad );
280 }
281
282 static inline int
283 arbel_cmd_read_mgm ( struct arbel *arbel, unsigned int index,
284                      struct arbelprm_mgm_entry *mgm ) {
285         return arbel_cmd ( arbel,
286                            ARBEL_HCR_OUT_CMD ( ARBEL_HCR_READ_MGM,
287                                                1, sizeof ( *mgm ) ),
288                            0, NULL, index, mgm );
289 }
290
291 static inline int
292 arbel_cmd_write_mgm ( struct arbel *arbel, unsigned int index,
293                       const struct arbelprm_mgm_entry *mgm ) {
294         return arbel_cmd ( arbel,
295                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_WRITE_MGM,
296                                               1, sizeof ( *mgm ) ),
297                            0, mgm, index, NULL );
298 }
299
300 static inline int
301 arbel_cmd_mgid_hash ( struct arbel *arbel, const struct ib_gid *gid,
302                       struct arbelprm_mgm_hash *hash ) {
303         return arbel_cmd ( arbel,
304                            ARBEL_HCR_INOUT_CMD ( ARBEL_HCR_MGID_HASH,
305                                                  1, sizeof ( *gid ),
306                                                  0, sizeof ( *hash ) ),
307                            0, gid, 0, hash );
308 }
309
310 /***************************************************************************
311  *
312  * Completion queue operations
313  *
314  ***************************************************************************
315  */
316
317 /**
318  * Create completion queue
319  *
320  * @v ibdev             Infiniband device
321  * @v cq                Completion queue
322  * @ret rc              Return status code
323  */
324 static int arbel_create_cq ( struct ib_device *ibdev,
325                              struct ib_completion_queue *cq ) {
326         struct arbel *arbel = ibdev->dev_priv;
327         struct arbel_completion_queue *arbel_cq;
328         struct arbelprm_completion_queue_context cqctx;
329         struct arbelprm_cq_ci_db_record *ci_db_rec;
330         struct arbelprm_cq_arm_db_record *arm_db_rec;
331         int cqn_offset;
332         unsigned int i;
333         int rc;
334
335         /* Find a free completion queue number */
336         cqn_offset = arbel_alloc_qn_offset ( arbel->cq_inuse, ARBEL_MAX_CQS );
337         if ( cqn_offset < 0 ) {
338                 DBGC ( arbel, "Arbel %p out of completion queues\n", arbel );
339                 rc = cqn_offset;
340                 goto err_cqn_offset;
341         }
342         cq->cqn = ( arbel->limits.reserved_cqs + cqn_offset );
343
344         /* Allocate control structures */
345         arbel_cq = zalloc ( sizeof ( *arbel_cq ) );
346         if ( ! arbel_cq ) {
347                 rc = -ENOMEM;
348                 goto err_arbel_cq;
349         }
350         arbel_cq->ci_doorbell_idx = arbel_cq_ci_doorbell_idx ( cqn_offset );
351         arbel_cq->arm_doorbell_idx = arbel_cq_arm_doorbell_idx ( cqn_offset );
352
353         /* Allocate completion queue itself */
354         arbel_cq->cqe_size = ( cq->num_cqes * sizeof ( arbel_cq->cqe[0] ) );
355         arbel_cq->cqe = malloc_dma ( arbel_cq->cqe_size,
356                                      sizeof ( arbel_cq->cqe[0] ) );
357         if ( ! arbel_cq->cqe ) {
358                 rc = -ENOMEM;
359                 goto err_cqe;
360         }
361         memset ( arbel_cq->cqe, 0, arbel_cq->cqe_size );
362         for ( i = 0 ; i < cq->num_cqes ; i++ ) {
363                 MLX_FILL_1 ( &arbel_cq->cqe[i].normal, 7, owner, 1 );
364         }
365         barrier();
366
367         /* Initialise doorbell records */
368         ci_db_rec = &arbel->db_rec[arbel_cq->ci_doorbell_idx].cq_ci;
369         MLX_FILL_1 ( ci_db_rec, 0, counter, 0 );
370         MLX_FILL_2 ( ci_db_rec, 1,
371                      res, ARBEL_UAR_RES_CQ_CI,
372                      cq_number, cq->cqn );
373         arm_db_rec = &arbel->db_rec[arbel_cq->arm_doorbell_idx].cq_arm;
374         MLX_FILL_1 ( arm_db_rec, 0, counter, 0 );
375         MLX_FILL_2 ( arm_db_rec, 1,
376                      res, ARBEL_UAR_RES_CQ_ARM,
377                      cq_number, cq->cqn );
378
379         /* Hand queue over to hardware */
380         memset ( &cqctx, 0, sizeof ( cqctx ) );
381         MLX_FILL_1 ( &cqctx, 0, st, 0xa /* "Event fired" */ );
382         MLX_FILL_1 ( &cqctx, 2, start_address_l,
383                      virt_to_bus ( arbel_cq->cqe ) );
384         MLX_FILL_2 ( &cqctx, 3,
385                      usr_page, arbel->limits.reserved_uars,
386                      log_cq_size, fls ( cq->num_cqes - 1 ) );
387         MLX_FILL_1 ( &cqctx, 5, c_eqn, arbel->eqn );
388         MLX_FILL_1 ( &cqctx, 6, pd, ARBEL_GLOBAL_PD );
389         MLX_FILL_1 ( &cqctx, 7, l_key, arbel->reserved_lkey );
390         MLX_FILL_1 ( &cqctx, 12, cqn, cq->cqn );
391         MLX_FILL_1 ( &cqctx, 13,
392                      cq_ci_db_record, arbel_cq->ci_doorbell_idx );
393         MLX_FILL_1 ( &cqctx, 14,
394                      cq_state_db_record, arbel_cq->arm_doorbell_idx );
395         if ( ( rc = arbel_cmd_sw2hw_cq ( arbel, cq->cqn, &cqctx ) ) != 0 ) {
396                 DBGC ( arbel, "Arbel %p SW2HW_CQ failed: %s\n",
397                        arbel, strerror ( rc ) );
398                 goto err_sw2hw_cq;
399         }
400
401         DBGC ( arbel, "Arbel %p CQN %#lx ring at [%p,%p)\n",
402                arbel, cq->cqn, arbel_cq->cqe,
403                ( ( ( void * ) arbel_cq->cqe ) + arbel_cq->cqe_size ) );
404         cq->dev_priv = arbel_cq;
405         return 0;
406
407  err_sw2hw_cq:
408         MLX_FILL_1 ( ci_db_rec, 1, res, ARBEL_UAR_RES_NONE );
409         MLX_FILL_1 ( arm_db_rec, 1, res, ARBEL_UAR_RES_NONE );
410         free_dma ( arbel_cq->cqe, arbel_cq->cqe_size );
411  err_cqe:
412         free ( arbel_cq );
413  err_arbel_cq:
414         arbel_free_qn_offset ( arbel->cq_inuse, cqn_offset );
415  err_cqn_offset:
416         return rc;
417 }
418
419 /**
420  * Destroy completion queue
421  *
422  * @v ibdev             Infiniband device
423  * @v cq                Completion queue
424  */
425 static void arbel_destroy_cq ( struct ib_device *ibdev,
426                                struct ib_completion_queue *cq ) {
427         struct arbel *arbel = ibdev->dev_priv;
428         struct arbel_completion_queue *arbel_cq = cq->dev_priv;
429         struct arbelprm_cq_ci_db_record *ci_db_rec;
430         struct arbelprm_cq_arm_db_record *arm_db_rec;
431         int cqn_offset;
432         int rc;
433
434         /* Take ownership back from hardware */
435         if ( ( rc = arbel_cmd_hw2sw_cq ( arbel, cq->cqn ) ) != 0 ) {
436                 DBGC ( arbel, "Arbel %p FATAL HW2SW_CQ failed on CQN %#lx: "
437                        "%s\n", arbel, cq->cqn, strerror ( rc ) );
438                 /* Leak memory and return; at least we avoid corruption */
439                 return;
440         }
441
442         /* Clear doorbell records */
443         ci_db_rec = &arbel->db_rec[arbel_cq->ci_doorbell_idx].cq_ci;
444         arm_db_rec = &arbel->db_rec[arbel_cq->arm_doorbell_idx].cq_arm;
445         MLX_FILL_1 ( ci_db_rec, 1, res, ARBEL_UAR_RES_NONE );
446         MLX_FILL_1 ( arm_db_rec, 1, res, ARBEL_UAR_RES_NONE );
447
448         /* Free memory */
449         free_dma ( arbel_cq->cqe, arbel_cq->cqe_size );
450         free ( arbel_cq );
451
452         /* Mark queue number as free */
453         cqn_offset = ( cq->cqn - arbel->limits.reserved_cqs );
454         arbel_free_qn_offset ( arbel->cq_inuse, cqn_offset );
455
456         cq->dev_priv = NULL;
457 }
458
459 /***************************************************************************
460  *
461  * Queue pair operations
462  *
463  ***************************************************************************
464  */
465
466 /**
467  * Create send work queue
468  *
469  * @v arbel_send_wq     Send work queue
470  * @v num_wqes          Number of work queue entries
471  * @ret rc              Return status code
472  */
473 static int arbel_create_send_wq ( struct arbel_send_work_queue *arbel_send_wq,
474                                   unsigned int num_wqes ) {
475         struct arbelprm_ud_send_wqe *wqe;
476         struct arbelprm_ud_send_wqe *next_wqe;
477         unsigned int wqe_idx_mask;
478         unsigned int i;
479
480         /* Allocate work queue */
481         arbel_send_wq->wqe_size = ( num_wqes *
482                                     sizeof ( arbel_send_wq->wqe[0] ) );
483         arbel_send_wq->wqe = malloc_dma ( arbel_send_wq->wqe_size,
484                                           sizeof ( arbel_send_wq->wqe[0] ) );
485         if ( ! arbel_send_wq->wqe )
486                 return -ENOMEM;
487         memset ( arbel_send_wq->wqe, 0, arbel_send_wq->wqe_size );
488
489         /* Link work queue entries */
490         wqe_idx_mask = ( num_wqes - 1 );
491         for ( i = 0 ; i < num_wqes ; i++ ) {
492                 wqe = &arbel_send_wq->wqe[i].ud;
493                 next_wqe = &arbel_send_wq->wqe[ ( i + 1 ) & wqe_idx_mask ].ud;
494                 MLX_FILL_1 ( &wqe->next, 0, nda_31_6,
495                              ( virt_to_bus ( next_wqe ) >> 6 ) );
496         }
497         
498         return 0;
499 }
500
501 /**
502  * Create receive work queue
503  *
504  * @v arbel_recv_wq     Receive work queue
505  * @v num_wqes          Number of work queue entries
506  * @ret rc              Return status code
507  */
508 static int arbel_create_recv_wq ( struct arbel_recv_work_queue *arbel_recv_wq,
509                                   unsigned int num_wqes ) {
510         struct arbelprm_recv_wqe *wqe;
511         struct arbelprm_recv_wqe *next_wqe;
512         unsigned int wqe_idx_mask;
513         size_t nds;
514         unsigned int i;
515         unsigned int j;
516
517         /* Allocate work queue */
518         arbel_recv_wq->wqe_size = ( num_wqes *
519                                     sizeof ( arbel_recv_wq->wqe[0] ) );
520         arbel_recv_wq->wqe = malloc_dma ( arbel_recv_wq->wqe_size,
521                                           sizeof ( arbel_recv_wq->wqe[0] ) );
522         if ( ! arbel_recv_wq->wqe )
523                 return -ENOMEM;
524         memset ( arbel_recv_wq->wqe, 0, arbel_recv_wq->wqe_size );
525
526         /* Link work queue entries */
527         wqe_idx_mask = ( num_wqes - 1 );
528         nds = ( ( offsetof ( typeof ( *wqe ), data ) +
529                   sizeof ( wqe->data[0] ) ) >> 4 );
530         for ( i = 0 ; i < num_wqes ; i++ ) {
531                 wqe = &arbel_recv_wq->wqe[i].recv;
532                 next_wqe = &arbel_recv_wq->wqe[( i + 1 ) & wqe_idx_mask].recv;
533                 MLX_FILL_1 ( &wqe->next, 0, nda_31_6,
534                              ( virt_to_bus ( next_wqe ) >> 6 ) );
535                 MLX_FILL_1 ( &wqe->next, 1, nds, ( sizeof ( *wqe ) / 16 ) );
536                 for ( j = 0 ; ( ( ( void * ) &wqe->data[j] ) <
537                                 ( ( void * ) ( wqe + 1 ) ) ) ; j++ ) {
538                         MLX_FILL_1 ( &wqe->data[j], 1,
539                                      l_key, ARBEL_INVALID_LKEY );
540                 }
541         }
542         
543         return 0;
544 }
545
546 /**
547  * Create queue pair
548  *
549  * @v ibdev             Infiniband device
550  * @v qp                Queue pair
551  * @ret rc              Return status code
552  */
553 static int arbel_create_qp ( struct ib_device *ibdev,
554                              struct ib_queue_pair *qp ) {
555         struct arbel *arbel = ibdev->dev_priv;
556         struct arbel_queue_pair *arbel_qp;
557         struct arbelprm_qp_ee_state_transitions qpctx;
558         struct arbelprm_qp_db_record *send_db_rec;
559         struct arbelprm_qp_db_record *recv_db_rec;
560         int qpn_offset;
561         int rc;
562
563         /* Find a free queue pair number */
564         qpn_offset = arbel_alloc_qn_offset ( arbel->qp_inuse, ARBEL_MAX_QPS );
565         if ( qpn_offset < 0 ) {
566                 DBGC ( arbel, "Arbel %p out of queue pairs\n", arbel );
567                 rc = qpn_offset;
568                 goto err_qpn_offset;
569         }
570         qp->qpn = ( ARBEL_QPN_BASE + arbel->limits.reserved_qps + qpn_offset );
571
572         /* Allocate control structures */
573         arbel_qp = zalloc ( sizeof ( *arbel_qp ) );
574         if ( ! arbel_qp ) {
575                 rc = -ENOMEM;
576                 goto err_arbel_qp;
577         }
578         arbel_qp->send.doorbell_idx = arbel_send_doorbell_idx ( qpn_offset );
579         arbel_qp->recv.doorbell_idx = arbel_recv_doorbell_idx ( qpn_offset );
580
581         /* Create send and receive work queues */
582         if ( ( rc = arbel_create_send_wq ( &arbel_qp->send,
583                                            qp->send.num_wqes ) ) != 0 )
584                 goto err_create_send_wq;
585         if ( ( rc = arbel_create_recv_wq ( &arbel_qp->recv,
586                                            qp->recv.num_wqes ) ) != 0 )
587                 goto err_create_recv_wq;
588
589         /* Initialise doorbell records */
590         send_db_rec = &arbel->db_rec[arbel_qp->send.doorbell_idx].qp;
591         MLX_FILL_1 ( send_db_rec, 0, counter, 0 );
592         MLX_FILL_2 ( send_db_rec, 1,
593                      res, ARBEL_UAR_RES_SQ,
594                      qp_number, qp->qpn );
595         recv_db_rec = &arbel->db_rec[arbel_qp->recv.doorbell_idx].qp;
596         MLX_FILL_1 ( recv_db_rec, 0, counter, 0 );
597         MLX_FILL_2 ( recv_db_rec, 1,
598                      res, ARBEL_UAR_RES_RQ,
599                      qp_number, qp->qpn );
600
601         /* Hand queue over to hardware */
602         memset ( &qpctx, 0, sizeof ( qpctx ) );
603         MLX_FILL_3 ( &qpctx, 2,
604                      qpc_eec_data.de, 1,
605                      qpc_eec_data.pm_state, 0x03 /* Always 0x03 for UD */,
606                      qpc_eec_data.st, ARBEL_ST_UD );
607         MLX_FILL_6 ( &qpctx, 4,
608                      qpc_eec_data.mtu, ARBEL_MTU_2048,
609                      qpc_eec_data.msg_max, 11 /* 2^11 = 2048 */,
610                      qpc_eec_data.log_rq_size, fls ( qp->recv.num_wqes - 1 ),
611                      qpc_eec_data.log_rq_stride,
612                      ( fls ( sizeof ( arbel_qp->recv.wqe[0] ) - 1 ) - 4 ),
613                      qpc_eec_data.log_sq_size, fls ( qp->send.num_wqes - 1 ),
614                      qpc_eec_data.log_sq_stride,
615                      ( fls ( sizeof ( arbel_qp->send.wqe[0] ) - 1 ) - 4 ) );
616         MLX_FILL_1 ( &qpctx, 5,
617                      qpc_eec_data.usr_page, arbel->limits.reserved_uars );
618         MLX_FILL_1 ( &qpctx, 10, qpc_eec_data.primary_address_path.port_number,
619                      PXE_IB_PORT );
620         MLX_FILL_1 ( &qpctx, 27, qpc_eec_data.pd, ARBEL_GLOBAL_PD );
621         MLX_FILL_1 ( &qpctx, 29, qpc_eec_data.wqe_lkey, arbel->reserved_lkey );
622         MLX_FILL_1 ( &qpctx, 30, qpc_eec_data.ssc, 1 );
623         MLX_FILL_1 ( &qpctx, 33, qpc_eec_data.cqn_snd, qp->send.cq->cqn );
624         MLX_FILL_1 ( &qpctx, 34, qpc_eec_data.snd_wqe_base_adr_l,
625                      ( virt_to_bus ( arbel_qp->send.wqe ) >> 6 ) );
626         MLX_FILL_1 ( &qpctx, 35, qpc_eec_data.snd_db_record_index,
627                      arbel_qp->send.doorbell_idx );
628         MLX_FILL_1 ( &qpctx, 38, qpc_eec_data.rsc, 1 );
629         MLX_FILL_1 ( &qpctx, 41, qpc_eec_data.cqn_rcv, qp->recv.cq->cqn );
630         MLX_FILL_1 ( &qpctx, 42, qpc_eec_data.rcv_wqe_base_adr_l,
631                      ( virt_to_bus ( arbel_qp->recv.wqe ) >> 6 ) );
632         MLX_FILL_1 ( &qpctx, 43, qpc_eec_data.rcv_db_record_index,
633                      arbel_qp->recv.doorbell_idx );
634         MLX_FILL_1 ( &qpctx, 44, qpc_eec_data.q_key, qp->qkey );
635         if ( ( rc = arbel_cmd_rst2init_qpee ( arbel, qp->qpn, &qpctx )) != 0 ){
636                 DBGC ( arbel, "Arbel %p RST2INIT_QPEE failed: %s\n",
637                        arbel, strerror ( rc ) );
638                 goto err_rst2init_qpee;
639         }
640         memset ( &qpctx, 0, sizeof ( qpctx ) );
641         MLX_FILL_2 ( &qpctx, 4,
642                      qpc_eec_data.mtu, ARBEL_MTU_2048,
643                      qpc_eec_data.msg_max, 11 /* 2^11 = 2048 */ );
644         if ( ( rc = arbel_cmd_init2rtr_qpee ( arbel, qp->qpn, &qpctx )) != 0 ){
645                 DBGC ( arbel, "Arbel %p INIT2RTR_QPEE failed: %s\n",
646                        arbel, strerror ( rc ) );
647                 goto err_init2rtr_qpee;
648         }
649         memset ( &qpctx, 0, sizeof ( qpctx ) );
650         if ( ( rc = arbel_cmd_rtr2rts_qpee ( arbel, qp->qpn, &qpctx ) ) != 0 ){
651                 DBGC ( arbel, "Arbel %p RTR2RTS_QPEE failed: %s\n",
652                        arbel, strerror ( rc ) );
653                 goto err_rtr2rts_qpee;
654         }
655
656         DBGC ( arbel, "Arbel %p QPN %#lx send ring at [%p,%p)\n",
657                arbel, qp->qpn, arbel_qp->send.wqe,
658                ( ( (void *) arbel_qp->send.wqe ) + arbel_qp->send.wqe_size ) );
659         DBGC ( arbel, "Arbel %p QPN %#lx receive ring at [%p,%p)\n",
660                arbel, qp->qpn, arbel_qp->recv.wqe,
661                ( ( (void *) arbel_qp->recv.wqe ) + arbel_qp->recv.wqe_size ) );
662         qp->dev_priv = arbel_qp;
663         return 0;
664
665  err_rtr2rts_qpee:
666  err_init2rtr_qpee:
667         arbel_cmd_2rst_qpee ( arbel, qp->qpn );
668  err_rst2init_qpee:
669         MLX_FILL_1 ( send_db_rec, 1, res, ARBEL_UAR_RES_NONE );
670         MLX_FILL_1 ( recv_db_rec, 1, res, ARBEL_UAR_RES_NONE );
671         free_dma ( arbel_qp->recv.wqe, arbel_qp->recv.wqe_size );
672  err_create_recv_wq:
673         free_dma ( arbel_qp->send.wqe, arbel_qp->send.wqe_size );
674  err_create_send_wq:
675         free ( arbel_qp );
676  err_arbel_qp:
677         arbel_free_qn_offset ( arbel->qp_inuse, qpn_offset );
678  err_qpn_offset:
679         return rc;
680 }
681
682 /**
683  * Destroy queue pair
684  *
685  * @v ibdev             Infiniband device
686  * @v qp                Queue pair
687  */
688 static void arbel_destroy_qp ( struct ib_device *ibdev,
689                                struct ib_queue_pair *qp ) {
690         struct arbel *arbel = ibdev->dev_priv;
691         struct arbel_queue_pair *arbel_qp = qp->dev_priv;
692         struct arbelprm_qp_db_record *send_db_rec;
693         struct arbelprm_qp_db_record *recv_db_rec;
694         int qpn_offset;
695         int rc;
696
697         /* Take ownership back from hardware */
698         if ( ( rc = arbel_cmd_2rst_qpee ( arbel, qp->qpn ) ) != 0 ) {
699                 DBGC ( arbel, "Arbel %p FATAL 2RST_QPEE failed on QPN %#lx: "
700                        "%s\n", arbel, qp->qpn, strerror ( rc ) );
701                 /* Leak memory and return; at least we avoid corruption */
702                 return;
703         }
704
705         /* Clear doorbell records */
706         send_db_rec = &arbel->db_rec[arbel_qp->send.doorbell_idx].qp;
707         recv_db_rec = &arbel->db_rec[arbel_qp->recv.doorbell_idx].qp;
708         MLX_FILL_1 ( send_db_rec, 1, res, ARBEL_UAR_RES_NONE );
709         MLX_FILL_1 ( recv_db_rec, 1, res, ARBEL_UAR_RES_NONE );
710
711         /* Free memory */
712         free_dma ( arbel_qp->send.wqe, arbel_qp->send.wqe_size );
713         free_dma ( arbel_qp->recv.wqe, arbel_qp->recv.wqe_size );
714         free ( arbel_qp );
715
716         /* Mark queue number as free */
717         qpn_offset = ( qp->qpn - ARBEL_QPN_BASE - arbel->limits.reserved_qps );
718         arbel_free_qn_offset ( arbel->qp_inuse, qpn_offset );
719
720         qp->dev_priv = NULL;
721 }
722
723 /***************************************************************************
724  *
725  * Work request operations
726  *
727  ***************************************************************************
728  */
729
730 /**
731  * Ring doorbell register in UAR
732  *
733  * @v arbel             Arbel device
734  * @v db_reg            Doorbell register structure
735  * @v offset            Address of doorbell
736  */
737 static void arbel_ring_doorbell ( struct arbel *arbel,
738                                   union arbelprm_doorbell_register *db_reg,
739                                   unsigned int offset ) {
740
741         DBGC2 ( arbel, "Arbel %p ringing doorbell %08lx:%08lx at %lx\n",
742                 arbel, db_reg->dword[0], db_reg->dword[1],
743                 virt_to_phys ( arbel->uar + offset ) );
744
745         barrier();
746         writel ( db_reg->dword[0], ( arbel->uar + offset + 0 ) );
747         barrier();
748         writel ( db_reg->dword[1], ( arbel->uar + offset + 4 ) );
749 }
750
751 /** GID used for GID-less send work queue entries */
752 static const struct ib_gid arbel_no_gid = {
753         { { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2, 0, 0, 0 } }
754 };
755
756 /**
757  * Post send work queue entry
758  *
759  * @v ibdev             Infiniband device
760  * @v qp                Queue pair
761  * @v av                Address vector
762  * @v iobuf             I/O buffer
763  * @ret rc              Return status code
764  */
765 static int arbel_post_send ( struct ib_device *ibdev,
766                              struct ib_queue_pair *qp,
767                              struct ib_address_vector *av,
768                              struct io_buffer *iobuf ) {
769         struct arbel *arbel = ibdev->dev_priv;
770         struct arbel_queue_pair *arbel_qp = qp->dev_priv;
771         struct ib_work_queue *wq = &qp->send;
772         struct arbel_send_work_queue *arbel_send_wq = &arbel_qp->send;
773         struct arbelprm_ud_send_wqe *prev_wqe;
774         struct arbelprm_ud_send_wqe *wqe;
775         struct arbelprm_qp_db_record *qp_db_rec;
776         union arbelprm_doorbell_register db_reg;
777         const struct ib_gid *gid;
778         unsigned int wqe_idx_mask;
779         size_t nds;
780
781         /* Allocate work queue entry */
782         wqe_idx_mask = ( wq->num_wqes - 1 );
783         if ( wq->iobufs[wq->next_idx & wqe_idx_mask] ) {
784                 DBGC ( arbel, "Arbel %p send queue full", arbel );
785                 return -ENOBUFS;
786         }
787         wq->iobufs[wq->next_idx & wqe_idx_mask] = iobuf;
788         prev_wqe = &arbel_send_wq->wqe[(wq->next_idx - 1) & wqe_idx_mask].ud;
789         wqe = &arbel_send_wq->wqe[wq->next_idx & wqe_idx_mask].ud;
790
791         /* Construct work queue entry */
792         MLX_FILL_1 ( &wqe->next, 1, always1, 1 );
793         memset ( &wqe->ctrl, 0, sizeof ( wqe->ctrl ) );
794         MLX_FILL_1 ( &wqe->ctrl, 0, always1, 1 );
795         memset ( &wqe->ud, 0, sizeof ( wqe->ud ) );
796         MLX_FILL_2 ( &wqe->ud, 0,
797                      ud_address_vector.pd, ARBEL_GLOBAL_PD,
798                      ud_address_vector.port_number, PXE_IB_PORT );
799         MLX_FILL_2 ( &wqe->ud, 1,
800                      ud_address_vector.rlid, av->dlid,
801                      ud_address_vector.g, av->gid_present );
802         MLX_FILL_2 ( &wqe->ud, 2,
803                      ud_address_vector.max_stat_rate,
804                          ( ( av->rate >= 3 ) ? 0 : 1 ),
805                      ud_address_vector.msg, 3 );
806         MLX_FILL_1 ( &wqe->ud, 3, ud_address_vector.sl, av->sl );
807         gid = ( av->gid_present ? &av->gid : &arbel_no_gid );
808         memcpy ( &wqe->ud.u.dwords[4], gid, sizeof ( *gid ) );
809         MLX_FILL_1 ( &wqe->ud, 8, destination_qp, av->dest_qp );
810         MLX_FILL_1 ( &wqe->ud, 9, q_key, av->qkey );
811         MLX_FILL_1 ( &wqe->data[0], 0, byte_count, iob_len ( iobuf ) );
812         MLX_FILL_1 ( &wqe->data[0], 1, l_key, arbel->reserved_lkey );
813         MLX_FILL_1 ( &wqe->data[0], 3,
814                      local_address_l, virt_to_bus ( iobuf->data ) );
815
816         /* Update previous work queue entry's "next" field */
817         nds = ( ( offsetof ( typeof ( *wqe ), data ) +
818                   sizeof ( wqe->data[0] ) ) >> 4 );
819         MLX_SET ( &prev_wqe->next, nopcode, ARBEL_OPCODE_SEND );
820         MLX_FILL_3 ( &prev_wqe->next, 1,
821                      nds, nds,
822                      f, 1,
823                      always1, 1 );
824
825         /* Update doorbell record */
826         barrier();
827         qp_db_rec = &arbel->db_rec[arbel_send_wq->doorbell_idx].qp;
828         MLX_FILL_1 ( qp_db_rec, 0,
829                      counter, ( ( wq->next_idx + 1 ) & 0xffff ) );
830
831         /* Ring doorbell register */
832         MLX_FILL_4 ( &db_reg.send, 0,
833                      nopcode, ARBEL_OPCODE_SEND,
834                      f, 1,
835                      wqe_counter, ( wq->next_idx & 0xffff ),
836                      wqe_cnt, 1 );
837         MLX_FILL_2 ( &db_reg.send, 1,
838                      nds, nds,
839                      qpn, qp->qpn );
840         arbel_ring_doorbell ( arbel, &db_reg, POST_SND_OFFSET );
841
842         /* Update work queue's index */
843         wq->next_idx++;
844
845         return 0;
846 }
847
848 /**
849  * Post receive work queue entry
850  *
851  * @v ibdev             Infiniband device
852  * @v qp                Queue pair
853  * @v iobuf             I/O buffer
854  * @ret rc              Return status code
855  */
856 static int arbel_post_recv ( struct ib_device *ibdev,
857                              struct ib_queue_pair *qp,
858                              struct io_buffer *iobuf ) {
859         struct arbel *arbel = ibdev->dev_priv;
860         struct arbel_queue_pair *arbel_qp = qp->dev_priv;
861         struct ib_work_queue *wq = &qp->recv;
862         struct arbel_recv_work_queue *arbel_recv_wq = &arbel_qp->recv;
863         struct arbelprm_recv_wqe *wqe;
864         union arbelprm_doorbell_record *db_rec;
865         unsigned int wqe_idx_mask;
866
867         /* Allocate work queue entry */
868         wqe_idx_mask = ( wq->num_wqes - 1 );
869         if ( wq->iobufs[wq->next_idx & wqe_idx_mask] ) {
870                 DBGC ( arbel, "Arbel %p receive queue full", arbel );
871                 return -ENOBUFS;
872         }
873         wq->iobufs[wq->next_idx & wqe_idx_mask] = iobuf;
874         wqe = &arbel_recv_wq->wqe[wq->next_idx & wqe_idx_mask].recv;
875
876         /* Construct work queue entry */
877         MLX_FILL_1 ( &wqe->data[0], 0, byte_count, iob_tailroom ( iobuf ) );
878         MLX_FILL_1 ( &wqe->data[0], 1, l_key, arbel->reserved_lkey );
879         MLX_FILL_1 ( &wqe->data[0], 3,
880                      local_address_l, virt_to_bus ( iobuf->data ) );
881
882         /* Update doorbell record */
883         barrier();
884         db_rec = &arbel->db_rec[arbel_recv_wq->doorbell_idx];
885         MLX_FILL_1 ( &db_rec->qp, 0,
886                      counter, ( ( wq->next_idx + 1 ) & 0xffff ) );      
887
888         /* Update work queue's index */
889         wq->next_idx++;
890
891         return 0;       
892 }
893
894 /**
895  * Handle completion
896  *
897  * @v ibdev             Infiniband device
898  * @v cq                Completion queue
899  * @v cqe               Hardware completion queue entry
900  * @v complete_send     Send completion handler
901  * @v complete_recv     Receive completion handler
902  * @ret rc              Return status code
903  */
904 static int arbel_complete ( struct ib_device *ibdev,
905                             struct ib_completion_queue *cq,
906                             union arbelprm_completion_entry *cqe,
907                             ib_completer_t complete_send,
908                             ib_completer_t complete_recv ) {
909         struct arbel *arbel = ibdev->dev_priv;
910         struct ib_completion completion;
911         struct ib_work_queue *wq;
912         struct ib_queue_pair *qp;
913         struct arbel_queue_pair *arbel_qp;
914         struct arbel_send_work_queue *arbel_send_wq;
915         struct arbel_recv_work_queue *arbel_recv_wq;
916         struct arbelprm_recv_wqe *recv_wqe;
917         struct io_buffer *iobuf;
918         ib_completer_t complete;
919         unsigned int opcode;
920         unsigned long qpn;
921         int is_send;
922         unsigned long wqe_adr;
923         unsigned int wqe_idx;
924         int rc = 0;
925
926         /* Parse completion */
927         memset ( &completion, 0, sizeof ( completion ) );
928         qpn = MLX_GET ( &cqe->normal, my_qpn );
929         is_send = MLX_GET ( &cqe->normal, s );
930         wqe_adr = ( MLX_GET ( &cqe->normal, wqe_adr ) << 6 );
931         opcode = MLX_GET ( &cqe->normal, opcode );
932         if ( opcode >= ARBEL_OPCODE_RECV_ERROR ) {
933                 /* "s" field is not valid for error opcodes */
934                 is_send = ( opcode == ARBEL_OPCODE_SEND_ERROR );
935                 completion.syndrome = MLX_GET ( &cqe->error, syndrome );
936                 DBGC ( arbel, "Arbel %p CPN %lx syndrome %x vendor %lx\n",
937                        arbel, cq->cqn, completion.syndrome,
938                        MLX_GET ( &cqe->error, vendor_code ) );
939                 rc = -EIO;
940                 /* Don't return immediately; propagate error to completer */
941         }
942
943         /* Identify work queue */
944         wq = ib_find_wq ( cq, qpn, is_send );
945         if ( ! wq ) {
946                 DBGC ( arbel, "Arbel %p CQN %lx unknown %s QPN %lx\n",
947                        arbel, cq->cqn, ( is_send ? "send" : "recv" ), qpn );
948                 return -EIO;
949         }
950         qp = wq->qp;
951         arbel_qp = qp->dev_priv;
952
953         /* Identify work queue entry index */
954         if ( is_send ) {
955                 arbel_send_wq = &arbel_qp->send;
956                 wqe_idx = ( ( wqe_adr - virt_to_bus ( arbel_send_wq->wqe ) ) /
957                             sizeof ( arbel_send_wq->wqe[0] ) );
958                 assert ( wqe_idx < qp->send.num_wqes );
959         } else {
960                 arbel_recv_wq = &arbel_qp->recv;
961                 wqe_idx = ( ( wqe_adr - virt_to_bus ( arbel_recv_wq->wqe ) ) /
962                             sizeof ( arbel_recv_wq->wqe[0] ) );
963                 assert ( wqe_idx < qp->recv.num_wqes );
964         }
965
966         /* Identify I/O buffer */
967         iobuf = wq->iobufs[wqe_idx];
968         if ( ! iobuf ) {
969                 DBGC ( arbel, "Arbel %p CQN %lx QPN %lx empty WQE %x\n",
970                        arbel, cq->cqn, qpn, wqe_idx );
971                 return -EIO;
972         }
973         wq->iobufs[wqe_idx] = NULL;
974
975         /* Fill in length for received packets */
976         if ( ! is_send ) {
977                 completion.len = MLX_GET ( &cqe->normal, byte_cnt );
978                 recv_wqe = &arbel_recv_wq->wqe[wqe_idx].recv;
979                 assert ( MLX_GET ( &recv_wqe->data[0], local_address_l ) ==
980                          virt_to_bus ( iobuf->data ) );
981                 assert ( MLX_GET ( &recv_wqe->data[0], byte_count ) ==
982                          iob_tailroom ( iobuf ) );
983                 DBG ( "CPQ %lx QPN %lx WQE %x\n", cq->cqn, qp->qpn, wqe_idx );
984                 //              DBG_HD ( iobuf, sizeof ( *iobuf ) );
985                 MLX_FILL_1 ( &recv_wqe->data[0], 0, byte_count, 0 );
986                 MLX_FILL_1 ( &recv_wqe->data[0], 1,
987                              l_key, ARBEL_INVALID_LKEY );
988                 if ( completion.len > iob_tailroom ( iobuf ) ) {
989                         DBGC ( arbel, "Arbel %p CQN %lx QPN %lx IDX %x "
990                                "overlength received packet length %zd\n",
991                                arbel, cq->cqn, qpn, wqe_idx, completion.len );
992                         return -EIO;
993                 }
994         }
995
996         /* Pass off to caller's completion handler */
997         complete = ( is_send ? complete_send : complete_recv );
998         complete ( ibdev, qp, &completion, iobuf );
999
1000         return rc;
1001 }                            
1002
1003 /**
1004  * Drain event queue
1005  *
1006  * @v arbel             Arbel device
1007  */
1008 static void arbel_drain_eq ( struct arbel *arbel ) {
1009 #warning "drain the event queue"
1010         drain_eq();
1011 }
1012
1013 /**
1014  * Poll completion queue
1015  *
1016  * @v ibdev             Infiniband device
1017  * @v cq                Completion queue
1018  * @v complete_send     Send completion handler
1019  * @v complete_recv     Receive completion handler
1020  */
1021 static void arbel_poll_cq ( struct ib_device *ibdev,
1022                             struct ib_completion_queue *cq,
1023                             ib_completer_t complete_send,
1024                             ib_completer_t complete_recv ) {
1025         struct arbel *arbel = ibdev->dev_priv;
1026         struct arbel_completion_queue *arbel_cq = cq->dev_priv;
1027         struct arbelprm_cq_ci_db_record *ci_db_rec;
1028         union arbelprm_completion_entry *cqe;
1029         unsigned int cqe_idx_mask;
1030         int rc;
1031
1032         /* Drain the event queue */
1033         arbel_drain_eq ( arbel );
1034
1035         while ( 1 ) {
1036                 /* Look for completion entry */
1037                 cqe_idx_mask = ( cq->num_cqes - 1 );
1038                 cqe = &arbel_cq->cqe[cq->next_idx & cqe_idx_mask];
1039                 if ( MLX_GET ( &cqe->normal, owner ) != 0 ) {
1040                         /* Entry still owned by hardware; end of poll */
1041                         break;
1042                 }
1043
1044                 /* Handle completion */
1045                 if ( ( rc = arbel_complete ( ibdev, cq, cqe, complete_send,
1046                                              complete_recv ) ) != 0 ) {
1047                         DBGC ( arbel, "Arbel %p failed to complete: %s\n",
1048                                arbel, strerror ( rc ) );
1049                         DBGC_HD ( arbel, cqe, sizeof ( *cqe ) );
1050                 }
1051
1052                 /* Return ownership to hardware */
1053                 MLX_FILL_1 ( &cqe->normal, 7, owner, 1 );
1054                 barrier();
1055                 /* Update completion queue's index */
1056                 cq->next_idx++;
1057                 /* Update doorbell record */
1058                 ci_db_rec = &arbel->db_rec[arbel_cq->ci_doorbell_idx].cq_ci;
1059                 MLX_FILL_1 ( ci_db_rec, 0,
1060                              counter, ( cq->next_idx & 0xffffffffUL ) );
1061         }
1062 }
1063
1064 /***************************************************************************
1065  *
1066  * Multicast group operations
1067  *
1068  ***************************************************************************
1069  */
1070
1071 /**
1072  * Attach to multicast group
1073  *
1074  * @v ibdev             Infiniband device
1075  * @v qp                Queue pair
1076  * @v gid               Multicast GID
1077  * @ret rc              Return status code
1078  */
1079 static int arbel_mcast_attach ( struct ib_device *ibdev,
1080                                 struct ib_queue_pair *qp,
1081                                 struct ib_gid *gid ) {
1082         struct arbel *arbel = ibdev->dev_priv;
1083         struct arbelprm_mgm_hash hash;
1084         struct arbelprm_mgm_entry mgm;
1085         unsigned int index;
1086         int rc;
1087
1088         /* Generate hash table index */
1089         if ( ( rc = arbel_cmd_mgid_hash ( arbel, gid, &hash ) ) != 0 ) {
1090                 DBGC ( arbel, "Arbel %p could not hash GID: %s\n",
1091                        arbel, strerror ( rc ) );
1092                 return rc;
1093         }
1094         index = MLX_GET ( &hash, hash );
1095
1096         /* Check for existing hash table entry */
1097         if ( ( rc = arbel_cmd_read_mgm ( arbel, index, &mgm ) ) != 0 ) {
1098                 DBGC ( arbel, "Arbel %p could not read MGM %#x: %s\n",
1099                        arbel, index, strerror ( rc ) );
1100                 return rc;
1101         }
1102         if ( MLX_GET ( &mgm, mgmqp_0.qi ) != 0 ) {
1103                 /* FIXME: this implementation allows only a single QP
1104                  * per multicast group, and doesn't handle hash
1105                  * collisions.  Sufficient for IPoIB but may need to
1106                  * be extended in future.
1107                  */
1108                 DBGC ( arbel, "Arbel %p MGID index %#x already in use\n",
1109                        arbel, index );
1110                 return -EBUSY;
1111         }
1112
1113         /* Update hash table entry */
1114         MLX_FILL_2 ( &mgm, 8,
1115                      mgmqp_0.qpn_i, qp->qpn,
1116                      mgmqp_0.qi, 1 );
1117         memcpy ( &mgm.u.dwords[4], gid, sizeof ( *gid ) );
1118         if ( ( rc = arbel_cmd_write_mgm ( arbel, index, &mgm ) ) != 0 ) {
1119                 DBGC ( arbel, "Arbel %p could not write MGM %#x: %s\n",
1120                        arbel, index, strerror ( rc ) );
1121                 return rc;
1122         }
1123
1124         return 0;
1125 }
1126
1127 /**
1128  * Detach from multicast group
1129  *
1130  * @v ibdev             Infiniband device
1131  * @v qp                Queue pair
1132  * @v gid               Multicast GID
1133  */
1134 static void arbel_mcast_detach ( struct ib_device *ibdev,
1135                                  struct ib_queue_pair *qp __unused,
1136                                  struct ib_gid *gid ) {
1137         struct arbel *arbel = ibdev->dev_priv;
1138         struct arbelprm_mgm_hash hash;
1139         struct arbelprm_mgm_entry mgm;
1140         unsigned int index;
1141         int rc;
1142
1143         /* Generate hash table index */
1144         if ( ( rc = arbel_cmd_mgid_hash ( arbel, gid, &hash ) ) != 0 ) {
1145                 DBGC ( arbel, "Arbel %p could not hash GID: %s\n",
1146                        arbel, strerror ( rc ) );
1147                 return;
1148         }
1149         index = MLX_GET ( &hash, hash );
1150
1151         /* Clear hash table entry */
1152         memset ( &mgm, 0, sizeof ( mgm ) );
1153         if ( ( rc = arbel_cmd_write_mgm ( arbel, index, &mgm ) ) != 0 ) {
1154                 DBGC ( arbel, "Arbel %p could not write MGM %#x: %s\n",
1155                        arbel, index, strerror ( rc ) );
1156                 return;
1157         }
1158 }
1159
1160 /** Arbel Infiniband operations */
1161 static struct ib_device_operations arbel_ib_operations = {
1162         .create_cq      = arbel_create_cq,
1163         .destroy_cq     = arbel_destroy_cq,
1164         .create_qp      = arbel_create_qp,
1165         .destroy_qp     = arbel_destroy_qp,
1166         .post_send      = arbel_post_send,
1167         .post_recv      = arbel_post_recv,
1168         .poll_cq        = arbel_poll_cq,
1169         .mcast_attach   = arbel_mcast_attach,
1170         .mcast_detach   = arbel_mcast_detach,
1171 };
1172
1173
1174 static int arbel_mad_ifc ( struct arbel *arbel,
1175                            union arbelprm_mad *mad ) {
1176         struct ib_mad_hdr *hdr = &mad->mad.mad_hdr;
1177         int rc;
1178
1179         hdr->base_version = IB_MGMT_BASE_VERSION;
1180         if ( ( rc = arbel_cmd_mad_ifc ( arbel, mad ) ) != 0 ) {
1181                 DBGC ( arbel, "Arbel %p could not issue MAD IFC: %s\n",
1182                        arbel, strerror ( rc ) );
1183                 return rc;
1184         }
1185         if ( hdr->status != 0 ) {
1186                 DBGC ( arbel, "Arbel %p MAD IFC status %04x\n",
1187                        arbel, ntohs ( hdr->status ) );
1188                 return -EIO;
1189         }
1190         return 0;
1191 }
1192
1193 static int arbel_get_port_info ( struct arbel *arbel,
1194                                  struct ib_mad_port_info *port_info ) {
1195         union arbelprm_mad mad;
1196         struct ib_mad_hdr *hdr = &mad.mad.mad_hdr;
1197         int rc;
1198
1199         memset ( &mad, 0, sizeof ( mad ) );
1200         hdr->mgmt_class = IB_MGMT_CLASS_SUBN_LID_ROUTED;
1201         hdr->class_version = 1;
1202         hdr->method = IB_MGMT_METHOD_GET;
1203         hdr->attr_id = htons ( IB_SMP_ATTR_PORT_INFO );
1204         hdr->attr_mod = htonl ( PXE_IB_PORT );
1205         if ( ( rc = arbel_mad_ifc ( arbel, &mad ) ) != 0 ) {
1206                 DBGC ( arbel, "Arbel %p could not get port info: %s\n",
1207                        arbel, strerror ( rc ) );
1208                 return rc;
1209         }
1210         memcpy ( port_info, &mad.mad.port_info, sizeof ( *port_info ) );
1211         return 0;
1212 }
1213
1214 static int arbel_get_guid_info ( struct arbel *arbel,
1215                                  struct ib_mad_guid_info *guid_info ) {
1216         union arbelprm_mad mad;
1217         struct ib_mad_hdr *hdr = &mad.mad.mad_hdr;
1218         int rc;
1219
1220         memset ( &mad, 0, sizeof ( mad ) );
1221         hdr->mgmt_class = IB_MGMT_CLASS_SUBN_LID_ROUTED;
1222         hdr->class_version = 1;
1223         hdr->method = IB_MGMT_METHOD_GET;
1224         hdr->attr_id = htons ( IB_SMP_ATTR_GUID_INFO );
1225         if ( ( rc = arbel_mad_ifc ( arbel, &mad ) ) != 0 ) {
1226                 DBGC ( arbel, "Arbel %p could not get GUID info: %s\n",
1227                        arbel, strerror ( rc ) );
1228                 return rc;
1229         }
1230         memcpy ( guid_info, &mad.mad.guid_info, sizeof ( *guid_info ) );
1231         return 0;
1232 }
1233
1234 static int arbel_get_pkey_table ( struct arbel *arbel,
1235                                   struct ib_mad_pkey_table *pkey_table ) {
1236         union arbelprm_mad mad;
1237         struct ib_mad_hdr *hdr = &mad.mad.mad_hdr;
1238         int rc;
1239
1240         memset ( &mad, 0, sizeof ( mad ) );
1241         hdr->mgmt_class = IB_MGMT_CLASS_SUBN_LID_ROUTED;
1242         hdr->class_version = 1;
1243         hdr->method = IB_MGMT_METHOD_GET;
1244         hdr->attr_id = htons ( IB_SMP_ATTR_PKEY_TABLE );
1245         if ( ( rc = arbel_mad_ifc ( arbel, &mad ) ) != 0 ) {
1246                 DBGC ( arbel, "Arbel %p could not get pkey table: %s\n",
1247                        arbel, strerror ( rc ) );
1248                 return rc;
1249         }
1250         memcpy ( pkey_table, &mad.mad.pkey_table, sizeof ( *pkey_table ) );
1251         return 0;
1252 }
1253
1254 static int arbel_get_port_gid ( struct arbel *arbel,
1255                                 struct ib_gid *port_gid ) {
1256         union {
1257                 /* This union exists just to save stack space */
1258                 struct ib_mad_port_info port_info;
1259                 struct ib_mad_guid_info guid_info;
1260         } u;
1261         int rc;
1262
1263         /* Port info gives us the first half of the port GID */
1264         if ( ( rc = arbel_get_port_info ( arbel, &u.port_info ) ) != 0 )
1265                 return rc;
1266         memcpy ( &port_gid->u.bytes[0], u.port_info.gid_prefix, 8 );
1267         
1268         /* GUID info gives us the second half of the port GID */
1269         if ( ( rc = arbel_get_guid_info ( arbel, &u.guid_info ) ) != 0 )
1270                 return rc;
1271         memcpy ( &port_gid->u.bytes[8], u.guid_info.gid_local, 8 );
1272
1273         return 0;
1274 }
1275
1276 static int arbel_get_sm_lid ( struct arbel *arbel,
1277                               unsigned long *sm_lid ) {
1278         struct ib_mad_port_info port_info;
1279         int rc;
1280
1281         if ( ( rc = arbel_get_port_info ( arbel, &port_info ) ) != 0 )
1282                 return rc;
1283         *sm_lid = ntohs ( port_info.mastersm_lid );
1284         return 0;
1285 }
1286
1287 static int arbel_get_pkey ( struct arbel *arbel, unsigned int *pkey ) {
1288         struct ib_mad_pkey_table pkey_table;
1289         int rc;
1290
1291         if ( ( rc = arbel_get_pkey_table ( arbel, &pkey_table ) ) != 0 )
1292                 return rc;
1293         *pkey = ntohs ( pkey_table.pkey[0][0] );
1294         return 0;
1295 }
1296
1297 /**
1298  * Probe PCI device
1299  *
1300  * @v pci               PCI device
1301  * @v id                PCI ID
1302  * @ret rc              Return status code
1303  */
1304 static int arbel_probe ( struct pci_device *pci,
1305                          const struct pci_device_id *id __unused ) {
1306         struct ib_device *ibdev;
1307         struct arbelprm_query_dev_lim dev_lim;
1308         struct arbel *arbel;
1309         udqp_t qph;
1310         int rc;
1311
1312         /* Allocate Infiniband device */
1313         ibdev = alloc_ibdev ( sizeof ( *arbel ) );
1314         if ( ! ibdev )
1315                 return -ENOMEM;
1316         ibdev->op = &arbel_ib_operations;
1317         pci_set_drvdata ( pci, ibdev );
1318         ibdev->dev = &pci->dev;
1319         arbel = ibdev->dev_priv;
1320         memset ( arbel, 0, sizeof ( *arbel ) );
1321
1322         /* Fix up PCI device */
1323         adjust_pci_device ( pci );
1324
1325         /* Initialise hardware */
1326         if ( ( rc = ib_driver_init ( pci, &qph ) ) != 0 )
1327                 goto err_ib_driver_init;
1328
1329         /* Hack up IB structures */
1330         arbel->config = memfree_pci_dev.cr_space;
1331         arbel->mailbox_in = dev_buffers_p->inprm_buf;
1332         arbel->mailbox_out = dev_buffers_p->outprm_buf;
1333         arbel->uar = memfree_pci_dev.uar;
1334         arbel->db_rec = dev_ib_data.uar_context_base;
1335         arbel->reserved_lkey = dev_ib_data.mkey;
1336         arbel->eqn = dev_ib_data.eq.eqn;
1337
1338         /* Get device limits */
1339         if ( ( rc = arbel_cmd_query_dev_lim ( arbel, &dev_lim ) ) != 0 ) {
1340                 DBGC ( arbel, "Arbel %p could not get device limits: %s\n",
1341                        arbel, strerror ( rc ) );
1342                 goto err_query_dev_lim;
1343         }
1344         arbel->limits.reserved_uars = MLX_GET ( &dev_lim, num_rsvd_uars );
1345         arbel->limits.reserved_cqs =
1346                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_cqs ) );
1347         arbel->limits.reserved_qps =
1348                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_qps ) );
1349
1350         /* Get subnet manager LID */
1351         if ( ( rc = arbel_get_sm_lid ( arbel, &ibdev->sm_lid ) ) != 0 ) {
1352                 DBGC ( arbel, "Arbel %p could not determine subnet manager "
1353                        "LID: %s\n", arbel, strerror ( rc ) );
1354                 goto err_get_sm_lid;
1355         }
1356
1357         /* Get port GID */
1358         if ( ( rc = arbel_get_port_gid ( arbel, &ibdev->port_gid ) ) != 0 ) {
1359                 DBGC ( arbel, "Arbel %p could not determine port GID: %s\n",
1360                        arbel, strerror ( rc ) );
1361                 goto err_get_port_gid;
1362         }
1363
1364         /* Get partition key */
1365         if ( ( rc = arbel_get_pkey ( arbel, &ibdev->pkey ) ) != 0 ) {
1366                 DBGC ( arbel, "Arbel %p could not determine partition key: "
1367                        "%s\n", arbel, strerror ( rc ) );
1368                 goto err_get_pkey;
1369         }
1370
1371         struct ud_av_st *bcast_av = ib_data.bcast_av;
1372         struct arbelprm_ud_address_vector *bav =
1373                 ( struct arbelprm_ud_address_vector * ) &bcast_av->av;
1374         struct ib_address_vector *av = &hack_ipoib_bcast_av;
1375         av->dest_qp = bcast_av->dest_qp;
1376         av->qkey = bcast_av->qkey;
1377         av->dlid = MLX_GET ( bav, rlid );
1378         av->rate = ( MLX_GET ( bav, max_stat_rate ) ? 1 : 4 );
1379         av->sl = MLX_GET ( bav, sl );
1380         av->gid_present = 1;
1381         memcpy ( &av->gid, ( ( void * ) bav ) + 16, 16 );
1382
1383         /* Add IPoIB device */
1384         if ( ( rc = ipoib_probe ( ibdev ) ) != 0 ) {
1385                 DBGC ( arbel, "Arbel %p could not add IPoIB device: %s\n",
1386                        arbel, strerror ( rc ) );
1387                 goto err_ipoib_probe;
1388         }
1389
1390         return 0;
1391
1392  err_ipoib_probe:
1393  err_get_pkey:
1394  err_get_port_gid:
1395  err_get_sm_lid:
1396  err_query_dev_lim:
1397         ib_driver_close ( 0 );
1398  err_ib_driver_init:
1399         free_ibdev ( ibdev );
1400         return rc;
1401 }
1402
1403 /**
1404  * Remove PCI device
1405  *
1406  * @v pci               PCI device
1407  */
1408 static void arbel_remove ( struct pci_device *pci ) {
1409         struct ib_device *ibdev = pci_get_drvdata ( pci );
1410
1411         ipoib_remove ( ibdev );
1412         ib_driver_close ( 0 );
1413 }
1414
1415 static struct pci_device_id arbel_nics[] = {
1416         PCI_ROM ( 0x15b3, 0x6282, "MT25218", "MT25218 HCA driver" ),
1417         PCI_ROM ( 0x15b3, 0x6274, "MT25204", "MT25204 HCA driver" ),
1418 };
1419
1420 struct pci_driver arbel_driver __pci_driver = {
1421         .ids = arbel_nics,
1422         .id_count = ( sizeof ( arbel_nics ) / sizeof ( arbel_nics[0] ) ),
1423         .probe = arbel_probe,
1424         .remove = arbel_remove,
1425 };