b1e3415174efc15a7eae41fed31e6ee8cf154d3b
[people/xl0/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
34
35
36
37 /***************************************************************************
38  *
39  * Queue number allocation
40  *
41  ***************************************************************************
42  */
43
44 /**
45  * Allocate queue number
46  *
47  * @v q_inuse           Queue usage bitmask
48  * @v max_inuse         Maximum number of in-use queues
49  * @ret qn_offset       Free queue number offset, or negative error
50  */
51 static int arbel_alloc_qn_offset ( arbel_bitmask_t *q_inuse,
52                                    unsigned int max_inuse ) {
53         unsigned int qn_offset = 0;
54         arbel_bitmask_t mask = 1;
55
56         while ( qn_offset < max_inuse ) {
57                 if ( ( mask & *q_inuse ) == 0 ) {
58                         *q_inuse |= mask;
59                         return qn_offset;
60                 }
61                 qn_offset++;
62                 mask <<= 1;
63                 if ( ! mask ) {
64                         mask = 1;
65                         q_inuse++;
66                 }
67         }
68         return -ENFILE;
69 }
70
71 /**
72  * Free queue number
73  *
74  * @v q_inuse           Queue usage bitmask
75  * @v qn_offset         Queue number offset
76  */
77 static void arbel_free_qn_offset ( arbel_bitmask_t *q_inuse, int qn_offset ) {
78         arbel_bitmask_t mask;
79
80         mask = ( 1 << ( qn_offset % ( 8 * sizeof ( mask ) ) ) );
81         q_inuse += ( qn_offset / ( 8 * sizeof ( mask ) ) );
82         *q_inuse &= ~mask;
83 }
84
85 /***************************************************************************
86  *
87  * HCA commands
88  *
89  ***************************************************************************
90  */
91
92 /**
93  * Wait for Arbel command completion
94  *
95  * @v arbel             Arbel device
96  * @ret rc              Return status code
97  */
98 static int arbel_cmd_wait ( struct arbel *arbel,
99                             struct arbelprm_hca_command_register *hcr ) {
100         unsigned int wait;
101
102         for ( wait = ARBEL_HCR_MAX_WAIT_MS ; wait ; wait-- ) {
103                 hcr->u.dwords[6] =
104                         readl ( arbel->config + ARBEL_HCR_REG ( 6 ) );
105                 if ( MLX_GET ( hcr, go ) == 0 )
106                         return 0;
107                 mdelay ( 1 );
108         }
109         return -EBUSY;
110 }
111
112 /**
113  * Issue HCA command
114  *
115  * @v arbel             Arbel device
116  * @v command           Command opcode, flags and input/output lengths
117  * @v op_mod            Opcode modifier (0 if no modifier applicable)
118  * @v in                Input parameters
119  * @v in_mod            Input modifier (0 if no modifier applicable)
120  * @v out               Output parameters
121  * @ret rc              Return status code
122  */
123 static int arbel_cmd ( struct arbel *arbel, unsigned long command,
124                        unsigned int op_mod, const void *in,
125                        unsigned int in_mod, void *out ) {
126         struct arbelprm_hca_command_register hcr;
127         unsigned int opcode = ARBEL_HCR_OPCODE ( command );
128         size_t in_len = ARBEL_HCR_IN_LEN ( command );
129         size_t out_len = ARBEL_HCR_OUT_LEN ( command );
130         void *in_buffer;
131         void *out_buffer;
132         unsigned int status;
133         unsigned int i;
134         int rc;
135
136         assert ( in_len <= ARBEL_MBOX_SIZE );
137         assert ( out_len <= ARBEL_MBOX_SIZE );
138
139         DBGC2 ( arbel, "Arbel %p command %02x in %zx%s out %zx%s\n",
140                 arbel, opcode, in_len,
141                 ( ( command & ARBEL_HCR_IN_MBOX ) ? "(mbox)" : "" ), out_len,
142                 ( ( command & ARBEL_HCR_OUT_MBOX ) ? "(mbox)" : "" ) );
143
144         /* Check that HCR is free */
145         if ( ( rc = arbel_cmd_wait ( arbel, &hcr ) ) != 0 ) {
146                 DBGC ( arbel, "Arbel %p command interface locked\n", arbel );
147                 return rc;
148         }
149
150         /* Prepare HCR */
151         memset ( &hcr, 0, sizeof ( hcr ) );
152         in_buffer = &hcr.u.dwords[0];
153         if ( in_len && ( command & ARBEL_HCR_IN_MBOX ) ) {
154                 in_buffer = arbel->mailbox_in;
155                 MLX_FILL_1 ( &hcr, 1, in_param_l, virt_to_bus ( in_buffer ) );
156         }
157         memcpy ( in_buffer, in, in_len );
158         MLX_FILL_1 ( &hcr, 2, input_modifier, in_mod );
159         out_buffer = &hcr.u.dwords[3];
160         if ( out_len && ( command & ARBEL_HCR_OUT_MBOX ) ) {
161                 out_buffer = arbel->mailbox_out;
162                 MLX_FILL_1 ( &hcr, 4, out_param_l,
163                              virt_to_bus ( out_buffer ) );
164         }
165         MLX_FILL_3 ( &hcr, 6,
166                      opcode, opcode,
167                      opcode_modifier, op_mod,
168                      go, 1 );
169         DBGC2_HD ( arbel, &hcr, sizeof ( hcr ) );
170         if ( in_len ) {
171                 DBGC2 ( arbel, "Input:\n" );
172                 DBGC2_HD ( arbel, in, ( ( in_len < 256 ) ? in_len : 256 ) );
173         }
174
175         /* Issue command */
176         for ( i = 0 ; i < ( sizeof ( hcr ) / sizeof ( hcr.u.dwords[0] ) ) ;
177               i++ ) {
178                 writel ( hcr.u.dwords[i],
179                          arbel->config + ARBEL_HCR_REG ( i ) );
180                 barrier();
181         }
182
183         /* Wait for command completion */
184         if ( ( rc = arbel_cmd_wait ( arbel, &hcr ) ) != 0 ) {
185                 DBGC ( arbel, "Arbel %p timed out waiting for command:\n",
186                        arbel );
187                 DBGC_HD ( arbel, &hcr, sizeof ( hcr ) );
188                 return rc;
189         }
190
191         /* Check command status */
192         status = MLX_GET ( &hcr, status );
193         if ( status != 0 ) {
194                 DBGC ( arbel, "Arbel %p command failed with status %02x:\n",
195                        arbel, status );
196                 DBGC_HD ( arbel, &hcr, sizeof ( hcr ) );
197                 return -EIO;
198         }
199
200         /* Read output parameters, if any */
201         hcr.u.dwords[3] = readl ( arbel->config + ARBEL_HCR_REG ( 3 ) );
202         hcr.u.dwords[4] = readl ( arbel->config + ARBEL_HCR_REG ( 4 ) );
203         memcpy ( out, out_buffer, out_len );
204         if ( out_len ) {
205                 DBGC2 ( arbel, "Output:\n" );
206                 DBGC2_HD ( arbel, out, ( ( out_len < 256 ) ? out_len : 256 ) );
207         }
208
209         return 0;
210 }
211
212 static inline int
213 arbel_cmd_query_dev_lim ( struct arbel *arbel,
214                           struct arbelprm_query_dev_lim *dev_lim ) {
215         return arbel_cmd ( arbel,
216                            ARBEL_HCR_OUT_CMD ( ARBEL_HCR_QUERY_DEV_LIM,
217                                                1, sizeof ( *dev_lim ) ),
218                            0, NULL, 0, dev_lim );
219 }
220
221 static inline int
222 arbel_cmd_query_fw ( struct arbel *arbel, struct arbelprm_query_fw *fw ) {
223         return arbel_cmd ( arbel,
224                            ARBEL_HCR_OUT_CMD ( ARBEL_HCR_QUERY_FW, 
225                                                1, sizeof ( *fw ) ),
226                            0, NULL, 0, fw );
227 }
228
229 static inline int
230 arbel_cmd_init_hca ( struct arbel *arbel,
231                      const struct arbelprm_init_hca *init_hca ) {
232         return arbel_cmd ( arbel,
233                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_INIT_HCA,
234                                               1, sizeof ( *init_hca ) ),
235                            0, init_hca, 0, NULL );
236 }
237
238 static inline int
239 arbel_cmd_sw2hw_cq ( struct arbel *arbel, unsigned long cqn,
240                      const struct arbelprm_completion_queue_context *cqctx ) {
241         return arbel_cmd ( arbel,
242                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_SW2HW_CQ,
243                                               1, sizeof ( *cqctx ) ),
244                            0, cqctx, cqn, NULL );
245 }
246
247 static inline int
248 arbel_cmd_hw2sw_cq ( struct arbel *arbel, unsigned long cqn,
249                      struct arbelprm_completion_queue_context *cqctx) {
250         return arbel_cmd ( arbel,
251                            ARBEL_HCR_OUT_CMD ( ARBEL_HCR_HW2SW_CQ,
252                                                1, sizeof ( *cqctx ) ),
253                            0, NULL, cqn, cqctx );
254 }
255
256 static inline int
257 arbel_cmd_rst2init_qpee ( struct arbel *arbel, unsigned long qpn,
258                           const struct arbelprm_qp_ee_state_transitions *ctx ){
259         return arbel_cmd ( arbel,
260                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_RST2INIT_QPEE,
261                                               1, sizeof ( *ctx ) ),
262                            0, ctx, qpn, NULL );
263 }
264
265 static inline int
266 arbel_cmd_init2rtr_qpee ( struct arbel *arbel, unsigned long qpn,
267                           const struct arbelprm_qp_ee_state_transitions *ctx ){
268         return arbel_cmd ( arbel,
269                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_INIT2RTR_QPEE,
270                                               1, sizeof ( *ctx ) ),
271                            0, ctx, qpn, NULL );
272 }
273
274 static inline int
275 arbel_cmd_rtr2rts_qpee ( struct arbel *arbel, unsigned long qpn,
276                          const struct arbelprm_qp_ee_state_transitions *ctx ) {
277         return arbel_cmd ( arbel,
278                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_RTR2RTS_QPEE,
279                                               1, sizeof ( *ctx ) ),
280                            0, ctx, qpn, NULL );
281 }
282
283 static inline int
284 arbel_cmd_2rst_qpee ( struct arbel *arbel, unsigned long qpn ) {
285         return arbel_cmd ( arbel,
286                            ARBEL_HCR_VOID_CMD ( ARBEL_HCR_2RST_QPEE ),
287                            0x03, NULL, qpn, NULL );
288 }
289
290 static inline int
291 arbel_cmd_mad_ifc ( struct arbel *arbel, union arbelprm_mad *mad ) {
292         return arbel_cmd ( arbel,
293                            ARBEL_HCR_INOUT_CMD ( ARBEL_HCR_MAD_IFC,
294                                                  1, sizeof ( *mad ),
295                                                  1, sizeof ( *mad ) ),
296                            0x03, mad, PXE_IB_PORT, mad );
297 }
298
299 static inline int
300 arbel_cmd_read_mgm ( struct arbel *arbel, unsigned int index,
301                      struct arbelprm_mgm_entry *mgm ) {
302         return arbel_cmd ( arbel,
303                            ARBEL_HCR_OUT_CMD ( ARBEL_HCR_READ_MGM,
304                                                1, sizeof ( *mgm ) ),
305                            0, NULL, index, mgm );
306 }
307
308 static inline int
309 arbel_cmd_write_mgm ( struct arbel *arbel, unsigned int index,
310                       const struct arbelprm_mgm_entry *mgm ) {
311         return arbel_cmd ( arbel,
312                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_WRITE_MGM,
313                                               1, sizeof ( *mgm ) ),
314                            0, mgm, index, NULL );
315 }
316
317 static inline int
318 arbel_cmd_mgid_hash ( struct arbel *arbel, const struct ib_gid *gid,
319                       struct arbelprm_mgm_hash *hash ) {
320         return arbel_cmd ( arbel,
321                            ARBEL_HCR_INOUT_CMD ( ARBEL_HCR_MGID_HASH,
322                                                  1, sizeof ( *gid ),
323                                                  0, sizeof ( *hash ) ),
324                            0, gid, 0, hash );
325 }
326
327 static inline int
328 arbel_cmd_run_fw ( struct arbel *arbel ) {
329         return arbel_cmd ( arbel,
330                            ARBEL_HCR_VOID_CMD ( ARBEL_HCR_RUN_FW ),
331                            0, NULL, 0, NULL );
332 }
333
334 static inline int
335 arbel_cmd_unmap_fa ( struct arbel *arbel ) {
336         return arbel_cmd ( arbel,
337                            ARBEL_HCR_VOID_CMD ( ARBEL_HCR_UNMAP_FA ),
338                            0, NULL, 0, NULL );
339 }
340
341 static inline int
342 arbel_cmd_map_fa ( struct arbel *arbel,
343                    const struct arbelprm_virtual_physical_mapping *map_fa ) {
344         return arbel_cmd ( arbel,
345                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_MAP_FA,
346                                               1, sizeof ( *map_fa ) ),
347                            0, map_fa, 1, NULL );
348 }
349
350 /***************************************************************************
351  *
352  * Completion queue operations
353  *
354  ***************************************************************************
355  */
356
357 /**
358  * Create completion queue
359  *
360  * @v ibdev             Infiniband device
361  * @v cq                Completion queue
362  * @ret rc              Return status code
363  */
364 static int arbel_create_cq ( struct ib_device *ibdev,
365                              struct ib_completion_queue *cq ) {
366         struct arbel *arbel = ibdev->dev_priv;
367         struct arbel_completion_queue *arbel_cq;
368         struct arbelprm_completion_queue_context cqctx;
369         struct arbelprm_cq_ci_db_record *ci_db_rec;
370         struct arbelprm_cq_arm_db_record *arm_db_rec;
371         int cqn_offset;
372         unsigned int i;
373         int rc;
374
375         /* Find a free completion queue number */
376         cqn_offset = arbel_alloc_qn_offset ( arbel->cq_inuse, ARBEL_MAX_CQS );
377         if ( cqn_offset < 0 ) {
378                 DBGC ( arbel, "Arbel %p out of completion queues\n", arbel );
379                 rc = cqn_offset;
380                 goto err_cqn_offset;
381         }
382         cq->cqn = ( arbel->limits.reserved_cqs + cqn_offset );
383
384         /* Allocate control structures */
385         arbel_cq = zalloc ( sizeof ( *arbel_cq ) );
386         if ( ! arbel_cq ) {
387                 rc = -ENOMEM;
388                 goto err_arbel_cq;
389         }
390         arbel_cq->ci_doorbell_idx = arbel_cq_ci_doorbell_idx ( cqn_offset );
391         arbel_cq->arm_doorbell_idx = arbel_cq_arm_doorbell_idx ( cqn_offset );
392
393         /* Allocate completion queue itself */
394         arbel_cq->cqe_size = ( cq->num_cqes * sizeof ( arbel_cq->cqe[0] ) );
395         arbel_cq->cqe = malloc_dma ( arbel_cq->cqe_size,
396                                      sizeof ( arbel_cq->cqe[0] ) );
397         if ( ! arbel_cq->cqe ) {
398                 rc = -ENOMEM;
399                 goto err_cqe;
400         }
401         memset ( arbel_cq->cqe, 0, arbel_cq->cqe_size );
402         for ( i = 0 ; i < cq->num_cqes ; i++ ) {
403                 MLX_FILL_1 ( &arbel_cq->cqe[i].normal, 7, owner, 1 );
404         }
405         barrier();
406
407         /* Initialise doorbell records */
408         ci_db_rec = &arbel->db_rec[arbel_cq->ci_doorbell_idx].cq_ci;
409         MLX_FILL_1 ( ci_db_rec, 0, counter, 0 );
410         MLX_FILL_2 ( ci_db_rec, 1,
411                      res, ARBEL_UAR_RES_CQ_CI,
412                      cq_number, cq->cqn );
413         arm_db_rec = &arbel->db_rec[arbel_cq->arm_doorbell_idx].cq_arm;
414         MLX_FILL_1 ( arm_db_rec, 0, counter, 0 );
415         MLX_FILL_2 ( arm_db_rec, 1,
416                      res, ARBEL_UAR_RES_CQ_ARM,
417                      cq_number, cq->cqn );
418
419         /* Hand queue over to hardware */
420         memset ( &cqctx, 0, sizeof ( cqctx ) );
421         MLX_FILL_1 ( &cqctx, 0, st, 0xa /* "Event fired" */ );
422         MLX_FILL_1 ( &cqctx, 2, start_address_l,
423                      virt_to_bus ( arbel_cq->cqe ) );
424         MLX_FILL_2 ( &cqctx, 3,
425                      usr_page, arbel->limits.reserved_uars,
426                      log_cq_size, fls ( cq->num_cqes - 1 ) );
427         MLX_FILL_1 ( &cqctx, 5, c_eqn, arbel->eqn );
428         MLX_FILL_1 ( &cqctx, 6, pd, ARBEL_GLOBAL_PD );
429         MLX_FILL_1 ( &cqctx, 7, l_key, arbel->reserved_lkey );
430         MLX_FILL_1 ( &cqctx, 12, cqn, cq->cqn );
431         MLX_FILL_1 ( &cqctx, 13,
432                      cq_ci_db_record, arbel_cq->ci_doorbell_idx );
433         MLX_FILL_1 ( &cqctx, 14,
434                      cq_state_db_record, arbel_cq->arm_doorbell_idx );
435         if ( ( rc = arbel_cmd_sw2hw_cq ( arbel, cq->cqn, &cqctx ) ) != 0 ) {
436                 DBGC ( arbel, "Arbel %p SW2HW_CQ failed: %s\n",
437                        arbel, strerror ( rc ) );
438                 goto err_sw2hw_cq;
439         }
440
441         DBGC ( arbel, "Arbel %p CQN %#lx ring at [%p,%p)\n",
442                arbel, cq->cqn, arbel_cq->cqe,
443                ( ( ( void * ) arbel_cq->cqe ) + arbel_cq->cqe_size ) );
444         cq->dev_priv = arbel_cq;
445         return 0;
446
447  err_sw2hw_cq:
448         MLX_FILL_1 ( ci_db_rec, 1, res, ARBEL_UAR_RES_NONE );
449         MLX_FILL_1 ( arm_db_rec, 1, res, ARBEL_UAR_RES_NONE );
450         free_dma ( arbel_cq->cqe, arbel_cq->cqe_size );
451  err_cqe:
452         free ( arbel_cq );
453  err_arbel_cq:
454         arbel_free_qn_offset ( arbel->cq_inuse, cqn_offset );
455  err_cqn_offset:
456         return rc;
457 }
458
459 /**
460  * Destroy completion queue
461  *
462  * @v ibdev             Infiniband device
463  * @v cq                Completion queue
464  */
465 static void arbel_destroy_cq ( struct ib_device *ibdev,
466                                struct ib_completion_queue *cq ) {
467         struct arbel *arbel = ibdev->dev_priv;
468         struct arbel_completion_queue *arbel_cq = cq->dev_priv;
469         struct arbelprm_completion_queue_context cqctx;
470         struct arbelprm_cq_ci_db_record *ci_db_rec;
471         struct arbelprm_cq_arm_db_record *arm_db_rec;
472         int cqn_offset;
473         int rc;
474
475         /* Take ownership back from hardware */
476         if ( ( rc = arbel_cmd_hw2sw_cq ( arbel, cq->cqn, &cqctx ) ) != 0 ) {
477                 DBGC ( arbel, "Arbel %p FATAL HW2SW_CQ failed on CQN %#lx: "
478                        "%s\n", arbel, cq->cqn, strerror ( rc ) );
479                 /* Leak memory and return; at least we avoid corruption */
480                 return;
481         }
482
483         /* Clear doorbell records */
484         ci_db_rec = &arbel->db_rec[arbel_cq->ci_doorbell_idx].cq_ci;
485         arm_db_rec = &arbel->db_rec[arbel_cq->arm_doorbell_idx].cq_arm;
486         MLX_FILL_1 ( ci_db_rec, 1, res, ARBEL_UAR_RES_NONE );
487         MLX_FILL_1 ( arm_db_rec, 1, res, ARBEL_UAR_RES_NONE );
488
489         /* Free memory */
490         free_dma ( arbel_cq->cqe, arbel_cq->cqe_size );
491         free ( arbel_cq );
492
493         /* Mark queue number as free */
494         cqn_offset = ( cq->cqn - arbel->limits.reserved_cqs );
495         arbel_free_qn_offset ( arbel->cq_inuse, cqn_offset );
496
497         cq->dev_priv = NULL;
498 }
499
500 /***************************************************************************
501  *
502  * Queue pair operations
503  *
504  ***************************************************************************
505  */
506
507 /**
508  * Create send work queue
509  *
510  * @v arbel_send_wq     Send work queue
511  * @v num_wqes          Number of work queue entries
512  * @ret rc              Return status code
513  */
514 static int arbel_create_send_wq ( struct arbel_send_work_queue *arbel_send_wq,
515                                   unsigned int num_wqes ) {
516         struct arbelprm_ud_send_wqe *wqe;
517         struct arbelprm_ud_send_wqe *next_wqe;
518         unsigned int wqe_idx_mask;
519         unsigned int i;
520
521         /* Allocate work queue */
522         arbel_send_wq->wqe_size = ( num_wqes *
523                                     sizeof ( arbel_send_wq->wqe[0] ) );
524         arbel_send_wq->wqe = malloc_dma ( arbel_send_wq->wqe_size,
525                                           sizeof ( arbel_send_wq->wqe[0] ) );
526         if ( ! arbel_send_wq->wqe )
527                 return -ENOMEM;
528         memset ( arbel_send_wq->wqe, 0, arbel_send_wq->wqe_size );
529
530         /* Link work queue entries */
531         wqe_idx_mask = ( num_wqes - 1 );
532         for ( i = 0 ; i < num_wqes ; i++ ) {
533                 wqe = &arbel_send_wq->wqe[i].ud;
534                 next_wqe = &arbel_send_wq->wqe[ ( i + 1 ) & wqe_idx_mask ].ud;
535                 MLX_FILL_1 ( &wqe->next, 0, nda_31_6,
536                              ( virt_to_bus ( next_wqe ) >> 6 ) );
537         }
538         
539         return 0;
540 }
541
542 /**
543  * Create receive work queue
544  *
545  * @v arbel_recv_wq     Receive work queue
546  * @v num_wqes          Number of work queue entries
547  * @ret rc              Return status code
548  */
549 static int arbel_create_recv_wq ( struct arbel_recv_work_queue *arbel_recv_wq,
550                                   unsigned int num_wqes ) {
551         struct arbelprm_recv_wqe *wqe;
552         struct arbelprm_recv_wqe *next_wqe;
553         unsigned int wqe_idx_mask;
554         size_t nds;
555         unsigned int i;
556         unsigned int j;
557
558         /* Allocate work queue */
559         arbel_recv_wq->wqe_size = ( num_wqes *
560                                     sizeof ( arbel_recv_wq->wqe[0] ) );
561         arbel_recv_wq->wqe = malloc_dma ( arbel_recv_wq->wqe_size,
562                                           sizeof ( arbel_recv_wq->wqe[0] ) );
563         if ( ! arbel_recv_wq->wqe )
564                 return -ENOMEM;
565         memset ( arbel_recv_wq->wqe, 0, arbel_recv_wq->wqe_size );
566
567         /* Link work queue entries */
568         wqe_idx_mask = ( num_wqes - 1 );
569         nds = ( ( offsetof ( typeof ( *wqe ), data ) +
570                   sizeof ( wqe->data[0] ) ) >> 4 );
571         for ( i = 0 ; i < num_wqes ; i++ ) {
572                 wqe = &arbel_recv_wq->wqe[i].recv;
573                 next_wqe = &arbel_recv_wq->wqe[( i + 1 ) & wqe_idx_mask].recv;
574                 MLX_FILL_1 ( &wqe->next, 0, nda_31_6,
575                              ( virt_to_bus ( next_wqe ) >> 6 ) );
576                 MLX_FILL_1 ( &wqe->next, 1, nds, ( sizeof ( *wqe ) / 16 ) );
577                 for ( j = 0 ; ( ( ( void * ) &wqe->data[j] ) <
578                                 ( ( void * ) ( wqe + 1 ) ) ) ; j++ ) {
579                         MLX_FILL_1 ( &wqe->data[j], 1,
580                                      l_key, ARBEL_INVALID_LKEY );
581                 }
582         }
583         
584         return 0;
585 }
586
587 /**
588  * Create queue pair
589  *
590  * @v ibdev             Infiniband device
591  * @v qp                Queue pair
592  * @ret rc              Return status code
593  */
594 static int arbel_create_qp ( struct ib_device *ibdev,
595                              struct ib_queue_pair *qp ) {
596         struct arbel *arbel = ibdev->dev_priv;
597         struct arbel_queue_pair *arbel_qp;
598         struct arbelprm_qp_ee_state_transitions qpctx;
599         struct arbelprm_qp_db_record *send_db_rec;
600         struct arbelprm_qp_db_record *recv_db_rec;
601         int qpn_offset;
602         int rc;
603
604         /* Find a free queue pair number */
605         qpn_offset = arbel_alloc_qn_offset ( arbel->qp_inuse, ARBEL_MAX_QPS );
606         if ( qpn_offset < 0 ) {
607                 DBGC ( arbel, "Arbel %p out of queue pairs\n", arbel );
608                 rc = qpn_offset;
609                 goto err_qpn_offset;
610         }
611         qp->qpn = ( ARBEL_QPN_BASE + arbel->limits.reserved_qps + qpn_offset );
612
613         /* Allocate control structures */
614         arbel_qp = zalloc ( sizeof ( *arbel_qp ) );
615         if ( ! arbel_qp ) {
616                 rc = -ENOMEM;
617                 goto err_arbel_qp;
618         }
619         arbel_qp->send.doorbell_idx = arbel_send_doorbell_idx ( qpn_offset );
620         arbel_qp->recv.doorbell_idx = arbel_recv_doorbell_idx ( qpn_offset );
621
622         /* Create send and receive work queues */
623         if ( ( rc = arbel_create_send_wq ( &arbel_qp->send,
624                                            qp->send.num_wqes ) ) != 0 )
625                 goto err_create_send_wq;
626         if ( ( rc = arbel_create_recv_wq ( &arbel_qp->recv,
627                                            qp->recv.num_wqes ) ) != 0 )
628                 goto err_create_recv_wq;
629
630         /* Initialise doorbell records */
631         send_db_rec = &arbel->db_rec[arbel_qp->send.doorbell_idx].qp;
632         MLX_FILL_1 ( send_db_rec, 0, counter, 0 );
633         MLX_FILL_2 ( send_db_rec, 1,
634                      res, ARBEL_UAR_RES_SQ,
635                      qp_number, qp->qpn );
636         recv_db_rec = &arbel->db_rec[arbel_qp->recv.doorbell_idx].qp;
637         MLX_FILL_1 ( recv_db_rec, 0, counter, 0 );
638         MLX_FILL_2 ( recv_db_rec, 1,
639                      res, ARBEL_UAR_RES_RQ,
640                      qp_number, qp->qpn );
641
642         /* Hand queue over to hardware */
643         memset ( &qpctx, 0, sizeof ( qpctx ) );
644         MLX_FILL_3 ( &qpctx, 2,
645                      qpc_eec_data.de, 1,
646                      qpc_eec_data.pm_state, 0x03 /* Always 0x03 for UD */,
647                      qpc_eec_data.st, ARBEL_ST_UD );
648         MLX_FILL_6 ( &qpctx, 4,
649                      qpc_eec_data.mtu, ARBEL_MTU_2048,
650                      qpc_eec_data.msg_max, 11 /* 2^11 = 2048 */,
651                      qpc_eec_data.log_rq_size, fls ( qp->recv.num_wqes - 1 ),
652                      qpc_eec_data.log_rq_stride,
653                      ( fls ( sizeof ( arbel_qp->recv.wqe[0] ) - 1 ) - 4 ),
654                      qpc_eec_data.log_sq_size, fls ( qp->send.num_wqes - 1 ),
655                      qpc_eec_data.log_sq_stride,
656                      ( fls ( sizeof ( arbel_qp->send.wqe[0] ) - 1 ) - 4 ) );
657         MLX_FILL_1 ( &qpctx, 5,
658                      qpc_eec_data.usr_page, arbel->limits.reserved_uars );
659         MLX_FILL_1 ( &qpctx, 10, qpc_eec_data.primary_address_path.port_number,
660                      PXE_IB_PORT );
661         MLX_FILL_1 ( &qpctx, 27, qpc_eec_data.pd, ARBEL_GLOBAL_PD );
662         MLX_FILL_1 ( &qpctx, 29, qpc_eec_data.wqe_lkey, arbel->reserved_lkey );
663         MLX_FILL_1 ( &qpctx, 30, qpc_eec_data.ssc, 1 );
664         MLX_FILL_1 ( &qpctx, 33, qpc_eec_data.cqn_snd, qp->send.cq->cqn );
665         MLX_FILL_1 ( &qpctx, 34, qpc_eec_data.snd_wqe_base_adr_l,
666                      ( virt_to_bus ( arbel_qp->send.wqe ) >> 6 ) );
667         MLX_FILL_1 ( &qpctx, 35, qpc_eec_data.snd_db_record_index,
668                      arbel_qp->send.doorbell_idx );
669         MLX_FILL_1 ( &qpctx, 38, qpc_eec_data.rsc, 1 );
670         MLX_FILL_1 ( &qpctx, 41, qpc_eec_data.cqn_rcv, qp->recv.cq->cqn );
671         MLX_FILL_1 ( &qpctx, 42, qpc_eec_data.rcv_wqe_base_adr_l,
672                      ( virt_to_bus ( arbel_qp->recv.wqe ) >> 6 ) );
673         MLX_FILL_1 ( &qpctx, 43, qpc_eec_data.rcv_db_record_index,
674                      arbel_qp->recv.doorbell_idx );
675         MLX_FILL_1 ( &qpctx, 44, qpc_eec_data.q_key, qp->qkey );
676         if ( ( rc = arbel_cmd_rst2init_qpee ( arbel, qp->qpn, &qpctx )) != 0 ){
677                 DBGC ( arbel, "Arbel %p RST2INIT_QPEE failed: %s\n",
678                        arbel, strerror ( rc ) );
679                 goto err_rst2init_qpee;
680         }
681         memset ( &qpctx, 0, sizeof ( qpctx ) );
682         MLX_FILL_2 ( &qpctx, 4,
683                      qpc_eec_data.mtu, ARBEL_MTU_2048,
684                      qpc_eec_data.msg_max, 11 /* 2^11 = 2048 */ );
685         if ( ( rc = arbel_cmd_init2rtr_qpee ( arbel, qp->qpn, &qpctx )) != 0 ){
686                 DBGC ( arbel, "Arbel %p INIT2RTR_QPEE failed: %s\n",
687                        arbel, strerror ( rc ) );
688                 goto err_init2rtr_qpee;
689         }
690         memset ( &qpctx, 0, sizeof ( qpctx ) );
691         if ( ( rc = arbel_cmd_rtr2rts_qpee ( arbel, qp->qpn, &qpctx ) ) != 0 ){
692                 DBGC ( arbel, "Arbel %p RTR2RTS_QPEE failed: %s\n",
693                        arbel, strerror ( rc ) );
694                 goto err_rtr2rts_qpee;
695         }
696
697         DBGC ( arbel, "Arbel %p QPN %#lx send ring at [%p,%p)\n",
698                arbel, qp->qpn, arbel_qp->send.wqe,
699                ( ( (void *) arbel_qp->send.wqe ) + arbel_qp->send.wqe_size ) );
700         DBGC ( arbel, "Arbel %p QPN %#lx receive ring at [%p,%p)\n",
701                arbel, qp->qpn, arbel_qp->recv.wqe,
702                ( ( (void *) arbel_qp->recv.wqe ) + arbel_qp->recv.wqe_size ) );
703         qp->dev_priv = arbel_qp;
704         return 0;
705
706  err_rtr2rts_qpee:
707  err_init2rtr_qpee:
708         arbel_cmd_2rst_qpee ( arbel, qp->qpn );
709  err_rst2init_qpee:
710         MLX_FILL_1 ( send_db_rec, 1, res, ARBEL_UAR_RES_NONE );
711         MLX_FILL_1 ( recv_db_rec, 1, res, ARBEL_UAR_RES_NONE );
712         free_dma ( arbel_qp->recv.wqe, arbel_qp->recv.wqe_size );
713  err_create_recv_wq:
714         free_dma ( arbel_qp->send.wqe, arbel_qp->send.wqe_size );
715  err_create_send_wq:
716         free ( arbel_qp );
717  err_arbel_qp:
718         arbel_free_qn_offset ( arbel->qp_inuse, qpn_offset );
719  err_qpn_offset:
720         return rc;
721 }
722
723 /**
724  * Destroy queue pair
725  *
726  * @v ibdev             Infiniband device
727  * @v qp                Queue pair
728  */
729 static void arbel_destroy_qp ( struct ib_device *ibdev,
730                                struct ib_queue_pair *qp ) {
731         struct arbel *arbel = ibdev->dev_priv;
732         struct arbel_queue_pair *arbel_qp = qp->dev_priv;
733         struct arbelprm_qp_db_record *send_db_rec;
734         struct arbelprm_qp_db_record *recv_db_rec;
735         int qpn_offset;
736         int rc;
737
738         /* Take ownership back from hardware */
739         if ( ( rc = arbel_cmd_2rst_qpee ( arbel, qp->qpn ) ) != 0 ) {
740                 DBGC ( arbel, "Arbel %p FATAL 2RST_QPEE failed on QPN %#lx: "
741                        "%s\n", arbel, qp->qpn, strerror ( rc ) );
742                 /* Leak memory and return; at least we avoid corruption */
743                 return;
744         }
745
746         /* Clear doorbell records */
747         send_db_rec = &arbel->db_rec[arbel_qp->send.doorbell_idx].qp;
748         recv_db_rec = &arbel->db_rec[arbel_qp->recv.doorbell_idx].qp;
749         MLX_FILL_1 ( send_db_rec, 1, res, ARBEL_UAR_RES_NONE );
750         MLX_FILL_1 ( recv_db_rec, 1, res, ARBEL_UAR_RES_NONE );
751
752         /* Free memory */
753         free_dma ( arbel_qp->send.wqe, arbel_qp->send.wqe_size );
754         free_dma ( arbel_qp->recv.wqe, arbel_qp->recv.wqe_size );
755         free ( arbel_qp );
756
757         /* Mark queue number as free */
758         qpn_offset = ( qp->qpn - ARBEL_QPN_BASE - arbel->limits.reserved_qps );
759         arbel_free_qn_offset ( arbel->qp_inuse, qpn_offset );
760
761         qp->dev_priv = NULL;
762 }
763
764 /***************************************************************************
765  *
766  * Work request operations
767  *
768  ***************************************************************************
769  */
770
771 /**
772  * Ring doorbell register in UAR
773  *
774  * @v arbel             Arbel device
775  * @v db_reg            Doorbell register structure
776  * @v offset            Address of doorbell
777  */
778 static void arbel_ring_doorbell ( struct arbel *arbel,
779                                   union arbelprm_doorbell_register *db_reg,
780                                   unsigned int offset ) {
781
782         DBGC2 ( arbel, "Arbel %p ringing doorbell %08lx:%08lx at %lx\n",
783                 arbel, db_reg->dword[0], db_reg->dword[1],
784                 virt_to_phys ( arbel->uar + offset ) );
785
786         barrier();
787         writel ( db_reg->dword[0], ( arbel->uar + offset + 0 ) );
788         barrier();
789         writel ( db_reg->dword[1], ( arbel->uar + offset + 4 ) );
790 }
791
792 /** GID used for GID-less send work queue entries */
793 static const struct ib_gid arbel_no_gid = {
794         { { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2, 0, 0, 0 } }
795 };
796
797 /**
798  * Post send work queue entry
799  *
800  * @v ibdev             Infiniband device
801  * @v qp                Queue pair
802  * @v av                Address vector
803  * @v iobuf             I/O buffer
804  * @ret rc              Return status code
805  */
806 static int arbel_post_send ( struct ib_device *ibdev,
807                              struct ib_queue_pair *qp,
808                              struct ib_address_vector *av,
809                              struct io_buffer *iobuf ) {
810         struct arbel *arbel = ibdev->dev_priv;
811         struct arbel_queue_pair *arbel_qp = qp->dev_priv;
812         struct ib_work_queue *wq = &qp->send;
813         struct arbel_send_work_queue *arbel_send_wq = &arbel_qp->send;
814         struct arbelprm_ud_send_wqe *prev_wqe;
815         struct arbelprm_ud_send_wqe *wqe;
816         struct arbelprm_qp_db_record *qp_db_rec;
817         union arbelprm_doorbell_register db_reg;
818         const struct ib_gid *gid;
819         unsigned int wqe_idx_mask;
820         size_t nds;
821
822         /* Allocate work queue entry */
823         wqe_idx_mask = ( wq->num_wqes - 1 );
824         if ( wq->iobufs[wq->next_idx & wqe_idx_mask] ) {
825                 DBGC ( arbel, "Arbel %p send queue full", arbel );
826                 return -ENOBUFS;
827         }
828         wq->iobufs[wq->next_idx & wqe_idx_mask] = iobuf;
829         prev_wqe = &arbel_send_wq->wqe[(wq->next_idx - 1) & wqe_idx_mask].ud;
830         wqe = &arbel_send_wq->wqe[wq->next_idx & wqe_idx_mask].ud;
831
832         /* Construct work queue entry */
833         MLX_FILL_1 ( &wqe->next, 1, always1, 1 );
834         memset ( &wqe->ctrl, 0, sizeof ( wqe->ctrl ) );
835         MLX_FILL_1 ( &wqe->ctrl, 0, always1, 1 );
836         memset ( &wqe->ud, 0, sizeof ( wqe->ud ) );
837         MLX_FILL_2 ( &wqe->ud, 0,
838                      ud_address_vector.pd, ARBEL_GLOBAL_PD,
839                      ud_address_vector.port_number, PXE_IB_PORT );
840         MLX_FILL_2 ( &wqe->ud, 1,
841                      ud_address_vector.rlid, av->dlid,
842                      ud_address_vector.g, av->gid_present );
843         MLX_FILL_2 ( &wqe->ud, 2,
844                      ud_address_vector.max_stat_rate,
845                          ( ( av->rate >= 3 ) ? 0 : 1 ),
846                      ud_address_vector.msg, 3 );
847         MLX_FILL_1 ( &wqe->ud, 3, ud_address_vector.sl, av->sl );
848         gid = ( av->gid_present ? &av->gid : &arbel_no_gid );
849         memcpy ( &wqe->ud.u.dwords[4], gid, sizeof ( *gid ) );
850         MLX_FILL_1 ( &wqe->ud, 8, destination_qp, av->dest_qp );
851         MLX_FILL_1 ( &wqe->ud, 9, q_key, av->qkey );
852         MLX_FILL_1 ( &wqe->data[0], 0, byte_count, iob_len ( iobuf ) );
853         MLX_FILL_1 ( &wqe->data[0], 1, l_key, arbel->reserved_lkey );
854         MLX_FILL_1 ( &wqe->data[0], 3,
855                      local_address_l, virt_to_bus ( iobuf->data ) );
856
857         /* Update previous work queue entry's "next" field */
858         nds = ( ( offsetof ( typeof ( *wqe ), data ) +
859                   sizeof ( wqe->data[0] ) ) >> 4 );
860         MLX_SET ( &prev_wqe->next, nopcode, ARBEL_OPCODE_SEND );
861         MLX_FILL_3 ( &prev_wqe->next, 1,
862                      nds, nds,
863                      f, 1,
864                      always1, 1 );
865
866         /* Update doorbell record */
867         barrier();
868         qp_db_rec = &arbel->db_rec[arbel_send_wq->doorbell_idx].qp;
869         MLX_FILL_1 ( qp_db_rec, 0,
870                      counter, ( ( wq->next_idx + 1 ) & 0xffff ) );
871
872         /* Ring doorbell register */
873         MLX_FILL_4 ( &db_reg.send, 0,
874                      nopcode, ARBEL_OPCODE_SEND,
875                      f, 1,
876                      wqe_counter, ( wq->next_idx & 0xffff ),
877                      wqe_cnt, 1 );
878         MLX_FILL_2 ( &db_reg.send, 1,
879                      nds, nds,
880                      qpn, qp->qpn );
881         arbel_ring_doorbell ( arbel, &db_reg, POST_SND_OFFSET );
882
883         /* Update work queue's index */
884         wq->next_idx++;
885
886         return 0;
887 }
888
889 /**
890  * Post receive work queue entry
891  *
892  * @v ibdev             Infiniband device
893  * @v qp                Queue pair
894  * @v iobuf             I/O buffer
895  * @ret rc              Return status code
896  */
897 static int arbel_post_recv ( struct ib_device *ibdev,
898                              struct ib_queue_pair *qp,
899                              struct io_buffer *iobuf ) {
900         struct arbel *arbel = ibdev->dev_priv;
901         struct arbel_queue_pair *arbel_qp = qp->dev_priv;
902         struct ib_work_queue *wq = &qp->recv;
903         struct arbel_recv_work_queue *arbel_recv_wq = &arbel_qp->recv;
904         struct arbelprm_recv_wqe *wqe;
905         union arbelprm_doorbell_record *db_rec;
906         unsigned int wqe_idx_mask;
907
908         /* Allocate work queue entry */
909         wqe_idx_mask = ( wq->num_wqes - 1 );
910         if ( wq->iobufs[wq->next_idx & wqe_idx_mask] ) {
911                 DBGC ( arbel, "Arbel %p receive queue full", arbel );
912                 return -ENOBUFS;
913         }
914         wq->iobufs[wq->next_idx & wqe_idx_mask] = iobuf;
915         wqe = &arbel_recv_wq->wqe[wq->next_idx & wqe_idx_mask].recv;
916
917         /* Construct work queue entry */
918         MLX_FILL_1 ( &wqe->data[0], 0, byte_count, iob_tailroom ( iobuf ) );
919         MLX_FILL_1 ( &wqe->data[0], 1, l_key, arbel->reserved_lkey );
920         MLX_FILL_1 ( &wqe->data[0], 3,
921                      local_address_l, virt_to_bus ( iobuf->data ) );
922
923         /* Update doorbell record */
924         barrier();
925         db_rec = &arbel->db_rec[arbel_recv_wq->doorbell_idx];
926         MLX_FILL_1 ( &db_rec->qp, 0,
927                      counter, ( ( wq->next_idx + 1 ) & 0xffff ) );      
928
929         /* Update work queue's index */
930         wq->next_idx++;
931
932         return 0;       
933 }
934
935 /**
936  * Handle completion
937  *
938  * @v ibdev             Infiniband device
939  * @v cq                Completion queue
940  * @v cqe               Hardware completion queue entry
941  * @v complete_send     Send completion handler
942  * @v complete_recv     Receive completion handler
943  * @ret rc              Return status code
944  */
945 static int arbel_complete ( struct ib_device *ibdev,
946                             struct ib_completion_queue *cq,
947                             union arbelprm_completion_entry *cqe,
948                             ib_completer_t complete_send,
949                             ib_completer_t complete_recv ) {
950         struct arbel *arbel = ibdev->dev_priv;
951         struct ib_completion completion;
952         struct ib_work_queue *wq;
953         struct ib_queue_pair *qp;
954         struct arbel_queue_pair *arbel_qp;
955         struct arbel_send_work_queue *arbel_send_wq;
956         struct arbel_recv_work_queue *arbel_recv_wq;
957         struct arbelprm_recv_wqe *recv_wqe;
958         struct io_buffer *iobuf;
959         ib_completer_t complete;
960         unsigned int opcode;
961         unsigned long qpn;
962         int is_send;
963         unsigned long wqe_adr;
964         unsigned int wqe_idx;
965         int rc = 0;
966
967         /* Parse completion */
968         memset ( &completion, 0, sizeof ( completion ) );
969         qpn = MLX_GET ( &cqe->normal, my_qpn );
970         is_send = MLX_GET ( &cqe->normal, s );
971         wqe_adr = ( MLX_GET ( &cqe->normal, wqe_adr ) << 6 );
972         opcode = MLX_GET ( &cqe->normal, opcode );
973         if ( opcode >= ARBEL_OPCODE_RECV_ERROR ) {
974                 /* "s" field is not valid for error opcodes */
975                 is_send = ( opcode == ARBEL_OPCODE_SEND_ERROR );
976                 completion.syndrome = MLX_GET ( &cqe->error, syndrome );
977                 DBGC ( arbel, "Arbel %p CPN %lx syndrome %x vendor %lx\n",
978                        arbel, cq->cqn, completion.syndrome,
979                        MLX_GET ( &cqe->error, vendor_code ) );
980                 rc = -EIO;
981                 /* Don't return immediately; propagate error to completer */
982         }
983
984         /* Identify work queue */
985         wq = ib_find_wq ( cq, qpn, is_send );
986         if ( ! wq ) {
987                 DBGC ( arbel, "Arbel %p CQN %lx unknown %s QPN %lx\n",
988                        arbel, cq->cqn, ( is_send ? "send" : "recv" ), qpn );
989                 return -EIO;
990         }
991         qp = wq->qp;
992         arbel_qp = qp->dev_priv;
993         arbel_send_wq = &arbel_qp->send;
994         arbel_recv_wq = &arbel_qp->recv;
995
996         /* Identify work queue entry index */
997         if ( is_send ) {
998                 wqe_idx = ( ( wqe_adr - virt_to_bus ( arbel_send_wq->wqe ) ) /
999                             sizeof ( arbel_send_wq->wqe[0] ) );
1000                 assert ( wqe_idx < qp->send.num_wqes );
1001         } else {
1002                 wqe_idx = ( ( wqe_adr - virt_to_bus ( arbel_recv_wq->wqe ) ) /
1003                             sizeof ( arbel_recv_wq->wqe[0] ) );
1004                 assert ( wqe_idx < qp->recv.num_wqes );
1005         }
1006
1007         /* Identify I/O buffer */
1008         iobuf = wq->iobufs[wqe_idx];
1009         if ( ! iobuf ) {
1010                 DBGC ( arbel, "Arbel %p CQN %lx QPN %lx empty WQE %x\n",
1011                        arbel, cq->cqn, qpn, wqe_idx );
1012                 return -EIO;
1013         }
1014         wq->iobufs[wqe_idx] = NULL;
1015
1016         /* Fill in length for received packets */
1017         if ( ! is_send ) {
1018                 completion.len = MLX_GET ( &cqe->normal, byte_cnt );
1019                 recv_wqe = &arbel_recv_wq->wqe[wqe_idx].recv;
1020                 assert ( MLX_GET ( &recv_wqe->data[0], local_address_l ) ==
1021                          virt_to_bus ( iobuf->data ) );
1022                 assert ( MLX_GET ( &recv_wqe->data[0], byte_count ) ==
1023                          iob_tailroom ( iobuf ) );
1024                 MLX_FILL_1 ( &recv_wqe->data[0], 0, byte_count, 0 );
1025                 MLX_FILL_1 ( &recv_wqe->data[0], 1,
1026                              l_key, ARBEL_INVALID_LKEY );
1027                 if ( completion.len > iob_tailroom ( iobuf ) ) {
1028                         DBGC ( arbel, "Arbel %p CQN %lx QPN %lx IDX %x "
1029                                "overlength received packet length %zd\n",
1030                                arbel, cq->cqn, qpn, wqe_idx, completion.len );
1031                         return -EIO;
1032                 }
1033         }
1034
1035         /* Pass off to caller's completion handler */
1036         complete = ( is_send ? complete_send : complete_recv );
1037         complete ( ibdev, qp, &completion, iobuf );
1038
1039         return rc;
1040 }                            
1041
1042 /**
1043  * Drain event queue
1044  *
1045  * @v arbel             Arbel device
1046  */
1047 static void arbel_drain_eq ( struct arbel *arbel ) {
1048 #warning "drain the event queue"
1049         drain_eq();
1050 }
1051
1052 /**
1053  * Poll completion queue
1054  *
1055  * @v ibdev             Infiniband device
1056  * @v cq                Completion queue
1057  * @v complete_send     Send completion handler
1058  * @v complete_recv     Receive completion handler
1059  */
1060 static void arbel_poll_cq ( struct ib_device *ibdev,
1061                             struct ib_completion_queue *cq,
1062                             ib_completer_t complete_send,
1063                             ib_completer_t complete_recv ) {
1064         struct arbel *arbel = ibdev->dev_priv;
1065         struct arbel_completion_queue *arbel_cq = cq->dev_priv;
1066         struct arbelprm_cq_ci_db_record *ci_db_rec;
1067         union arbelprm_completion_entry *cqe;
1068         unsigned int cqe_idx_mask;
1069         int rc;
1070
1071         /* Drain the event queue */
1072         arbel_drain_eq ( arbel );
1073
1074         while ( 1 ) {
1075                 /* Look for completion entry */
1076                 cqe_idx_mask = ( cq->num_cqes - 1 );
1077                 cqe = &arbel_cq->cqe[cq->next_idx & cqe_idx_mask];
1078                 if ( MLX_GET ( &cqe->normal, owner ) != 0 ) {
1079                         /* Entry still owned by hardware; end of poll */
1080                         break;
1081                 }
1082
1083                 /* Handle completion */
1084                 if ( ( rc = arbel_complete ( ibdev, cq, cqe, complete_send,
1085                                              complete_recv ) ) != 0 ) {
1086                         DBGC ( arbel, "Arbel %p failed to complete: %s\n",
1087                                arbel, strerror ( rc ) );
1088                         DBGC_HD ( arbel, cqe, sizeof ( *cqe ) );
1089                 }
1090
1091                 /* Return ownership to hardware */
1092                 MLX_FILL_1 ( &cqe->normal, 7, owner, 1 );
1093                 barrier();
1094                 /* Update completion queue's index */
1095                 cq->next_idx++;
1096                 /* Update doorbell record */
1097                 ci_db_rec = &arbel->db_rec[arbel_cq->ci_doorbell_idx].cq_ci;
1098                 MLX_FILL_1 ( ci_db_rec, 0,
1099                              counter, ( cq->next_idx & 0xffffffffUL ) );
1100         }
1101 }
1102
1103 /***************************************************************************
1104  *
1105  * Multicast group operations
1106  *
1107  ***************************************************************************
1108  */
1109
1110 /**
1111  * Attach to multicast group
1112  *
1113  * @v ibdev             Infiniband device
1114  * @v qp                Queue pair
1115  * @v gid               Multicast GID
1116  * @ret rc              Return status code
1117  */
1118 static int arbel_mcast_attach ( struct ib_device *ibdev,
1119                                 struct ib_queue_pair *qp,
1120                                 struct ib_gid *gid ) {
1121         struct arbel *arbel = ibdev->dev_priv;
1122         struct arbelprm_mgm_hash hash;
1123         struct arbelprm_mgm_entry mgm;
1124         unsigned int index;
1125         int rc;
1126
1127         /* Generate hash table index */
1128         if ( ( rc = arbel_cmd_mgid_hash ( arbel, gid, &hash ) ) != 0 ) {
1129                 DBGC ( arbel, "Arbel %p could not hash GID: %s\n",
1130                        arbel, strerror ( rc ) );
1131                 return rc;
1132         }
1133         index = MLX_GET ( &hash, hash );
1134
1135         /* Check for existing hash table entry */
1136         if ( ( rc = arbel_cmd_read_mgm ( arbel, index, &mgm ) ) != 0 ) {
1137                 DBGC ( arbel, "Arbel %p could not read MGM %#x: %s\n",
1138                        arbel, index, strerror ( rc ) );
1139                 return rc;
1140         }
1141         if ( MLX_GET ( &mgm, mgmqp_0.qi ) != 0 ) {
1142                 /* FIXME: this implementation allows only a single QP
1143                  * per multicast group, and doesn't handle hash
1144                  * collisions.  Sufficient for IPoIB but may need to
1145                  * be extended in future.
1146                  */
1147                 DBGC ( arbel, "Arbel %p MGID index %#x already in use\n",
1148                        arbel, index );
1149                 return -EBUSY;
1150         }
1151
1152         /* Update hash table entry */
1153         MLX_FILL_2 ( &mgm, 8,
1154                      mgmqp_0.qpn_i, qp->qpn,
1155                      mgmqp_0.qi, 1 );
1156         memcpy ( &mgm.u.dwords[4], gid, sizeof ( *gid ) );
1157         if ( ( rc = arbel_cmd_write_mgm ( arbel, index, &mgm ) ) != 0 ) {
1158                 DBGC ( arbel, "Arbel %p could not write MGM %#x: %s\n",
1159                        arbel, index, strerror ( rc ) );
1160                 return rc;
1161         }
1162
1163         return 0;
1164 }
1165
1166 /**
1167  * Detach from multicast group
1168  *
1169  * @v ibdev             Infiniband device
1170  * @v qp                Queue pair
1171  * @v gid               Multicast GID
1172  */
1173 static void arbel_mcast_detach ( struct ib_device *ibdev,
1174                                  struct ib_queue_pair *qp __unused,
1175                                  struct ib_gid *gid ) {
1176         struct arbel *arbel = ibdev->dev_priv;
1177         struct arbelprm_mgm_hash hash;
1178         struct arbelprm_mgm_entry mgm;
1179         unsigned int index;
1180         int rc;
1181
1182         /* Generate hash table index */
1183         if ( ( rc = arbel_cmd_mgid_hash ( arbel, gid, &hash ) ) != 0 ) {
1184                 DBGC ( arbel, "Arbel %p could not hash GID: %s\n",
1185                        arbel, strerror ( rc ) );
1186                 return;
1187         }
1188         index = MLX_GET ( &hash, hash );
1189
1190         /* Clear hash table entry */
1191         memset ( &mgm, 0, sizeof ( mgm ) );
1192         if ( ( rc = arbel_cmd_write_mgm ( arbel, index, &mgm ) ) != 0 ) {
1193                 DBGC ( arbel, "Arbel %p could not write MGM %#x: %s\n",
1194                        arbel, index, strerror ( rc ) );
1195                 return;
1196         }
1197 }
1198
1199 /** Arbel Infiniband operations */
1200 static struct ib_device_operations arbel_ib_operations = {
1201         .create_cq      = arbel_create_cq,
1202         .destroy_cq     = arbel_destroy_cq,
1203         .create_qp      = arbel_create_qp,
1204         .destroy_qp     = arbel_destroy_qp,
1205         .post_send      = arbel_post_send,
1206         .post_recv      = arbel_post_recv,
1207         .poll_cq        = arbel_poll_cq,
1208         .mcast_attach   = arbel_mcast_attach,
1209         .mcast_detach   = arbel_mcast_detach,
1210 };
1211
1212 /***************************************************************************
1213  *
1214  * MAD IFC operations
1215  *
1216  ***************************************************************************
1217  */
1218
1219 static int arbel_mad_ifc ( struct arbel *arbel,
1220                            union arbelprm_mad *mad ) {
1221         struct ib_mad_hdr *hdr = &mad->mad.mad_hdr;
1222         int rc;
1223
1224         hdr->base_version = IB_MGMT_BASE_VERSION;
1225         if ( ( rc = arbel_cmd_mad_ifc ( arbel, mad ) ) != 0 ) {
1226                 DBGC ( arbel, "Arbel %p could not issue MAD IFC: %s\n",
1227                        arbel, strerror ( rc ) );
1228                 return rc;
1229         }
1230         if ( hdr->status != 0 ) {
1231                 DBGC ( arbel, "Arbel %p MAD IFC status %04x\n",
1232                        arbel, ntohs ( hdr->status ) );
1233                 return -EIO;
1234         }
1235         return 0;
1236 }
1237
1238 static int arbel_get_port_info ( struct arbel *arbel,
1239                                  struct ib_mad_port_info *port_info ) {
1240         union arbelprm_mad mad;
1241         struct ib_mad_hdr *hdr = &mad.mad.mad_hdr;
1242         int rc;
1243
1244         memset ( &mad, 0, sizeof ( mad ) );
1245         hdr->mgmt_class = IB_MGMT_CLASS_SUBN_LID_ROUTED;
1246         hdr->class_version = 1;
1247         hdr->method = IB_MGMT_METHOD_GET;
1248         hdr->attr_id = htons ( IB_SMP_ATTR_PORT_INFO );
1249         hdr->attr_mod = htonl ( PXE_IB_PORT );
1250         if ( ( rc = arbel_mad_ifc ( arbel, &mad ) ) != 0 ) {
1251                 DBGC ( arbel, "Arbel %p could not get port info: %s\n",
1252                        arbel, strerror ( rc ) );
1253                 return rc;
1254         }
1255         memcpy ( port_info, &mad.mad.port_info, sizeof ( *port_info ) );
1256         return 0;
1257 }
1258
1259 static int arbel_get_guid_info ( struct arbel *arbel,
1260                                  struct ib_mad_guid_info *guid_info ) {
1261         union arbelprm_mad mad;
1262         struct ib_mad_hdr *hdr = &mad.mad.mad_hdr;
1263         int rc;
1264
1265         memset ( &mad, 0, sizeof ( mad ) );
1266         hdr->mgmt_class = IB_MGMT_CLASS_SUBN_LID_ROUTED;
1267         hdr->class_version = 1;
1268         hdr->method = IB_MGMT_METHOD_GET;
1269         hdr->attr_id = htons ( IB_SMP_ATTR_GUID_INFO );
1270         if ( ( rc = arbel_mad_ifc ( arbel, &mad ) ) != 0 ) {
1271                 DBGC ( arbel, "Arbel %p could not get GUID info: %s\n",
1272                        arbel, strerror ( rc ) );
1273                 return rc;
1274         }
1275         memcpy ( guid_info, &mad.mad.guid_info, sizeof ( *guid_info ) );
1276         return 0;
1277 }
1278
1279 static int arbel_get_pkey_table ( struct arbel *arbel,
1280                                   struct ib_mad_pkey_table *pkey_table ) {
1281         union arbelprm_mad mad;
1282         struct ib_mad_hdr *hdr = &mad.mad.mad_hdr;
1283         int rc;
1284
1285         memset ( &mad, 0, sizeof ( mad ) );
1286         hdr->mgmt_class = IB_MGMT_CLASS_SUBN_LID_ROUTED;
1287         hdr->class_version = 1;
1288         hdr->method = IB_MGMT_METHOD_GET;
1289         hdr->attr_id = htons ( IB_SMP_ATTR_PKEY_TABLE );
1290         if ( ( rc = arbel_mad_ifc ( arbel, &mad ) ) != 0 ) {
1291                 DBGC ( arbel, "Arbel %p could not get pkey table: %s\n",
1292                        arbel, strerror ( rc ) );
1293                 return rc;
1294         }
1295         memcpy ( pkey_table, &mad.mad.pkey_table, sizeof ( *pkey_table ) );
1296         return 0;
1297 }
1298
1299 static int arbel_get_port_gid ( struct arbel *arbel,
1300                                 struct ib_gid *port_gid ) {
1301         union {
1302                 /* This union exists just to save stack space */
1303                 struct ib_mad_port_info port_info;
1304                 struct ib_mad_guid_info guid_info;
1305         } u;
1306         int rc;
1307
1308         /* Port info gives us the first half of the port GID */
1309         if ( ( rc = arbel_get_port_info ( arbel, &u.port_info ) ) != 0 )
1310                 return rc;
1311         memcpy ( &port_gid->u.bytes[0], u.port_info.gid_prefix, 8 );
1312         
1313         /* GUID info gives us the second half of the port GID */
1314         if ( ( rc = arbel_get_guid_info ( arbel, &u.guid_info ) ) != 0 )
1315                 return rc;
1316         memcpy ( &port_gid->u.bytes[8], u.guid_info.gid_local, 8 );
1317
1318         return 0;
1319 }
1320
1321 static int arbel_get_sm_lid ( struct arbel *arbel,
1322                               unsigned long *sm_lid ) {
1323         struct ib_mad_port_info port_info;
1324         int rc;
1325
1326         if ( ( rc = arbel_get_port_info ( arbel, &port_info ) ) != 0 )
1327                 return rc;
1328         *sm_lid = ntohs ( port_info.mastersm_lid );
1329         return 0;
1330 }
1331
1332 static int arbel_get_pkey ( struct arbel *arbel, unsigned int *pkey ) {
1333         struct ib_mad_pkey_table pkey_table;
1334         int rc;
1335
1336         if ( ( rc = arbel_get_pkey_table ( arbel, &pkey_table ) ) != 0 )
1337                 return rc;
1338         *pkey = ntohs ( pkey_table.pkey[0][0] );
1339         return 0;
1340 }
1341
1342 /**
1343  * Get MAD parameters
1344  *
1345  * @v arbel             Arbel device
1346  * @ret rc              Return status code
1347  */
1348 static int arbel_get_mad_params ( struct ib_device *ibdev ) {
1349         struct arbel *arbel = ibdev->dev_priv;
1350         int rc;
1351
1352         /* Get subnet manager LID */
1353         if ( ( rc = arbel_get_sm_lid ( arbel, &ibdev->sm_lid ) ) != 0 ) {
1354                 DBGC ( arbel, "Arbel %p could not determine subnet manager "
1355                        "LID: %s\n", arbel, strerror ( rc ) );
1356                 return rc;
1357         }
1358
1359         /* Get port GID */
1360         if ( ( rc = arbel_get_port_gid ( arbel, &ibdev->port_gid ) ) != 0 ) {
1361                 DBGC ( arbel, "Arbel %p could not determine port GID: %s\n",
1362                        arbel, strerror ( rc ) );
1363                 return rc;
1364         }
1365
1366         /* Get partition key */
1367         if ( ( rc = arbel_get_pkey ( arbel, &ibdev->pkey ) ) != 0 ) {
1368                 DBGC ( arbel, "Arbel %p could not determine partition key: "
1369                        "%s\n", arbel, strerror ( rc ) );
1370                 return rc;
1371         }
1372
1373         return 0;
1374 }
1375
1376 /***************************************************************************
1377  *
1378  * Firmware control
1379  *
1380  ***************************************************************************
1381  */
1382
1383 /**
1384  * Start firmware running
1385  *
1386  * @v arbel             Arbel device
1387  * @ret rc              Return status code
1388  */
1389 static int arbel_start_firmware ( struct arbel *arbel ) {
1390         struct arbelprm_query_fw fw;
1391         struct arbelprm_virtual_physical_mapping map_fa;
1392         unsigned int fw_pages;
1393         unsigned int log2_fw_pages;
1394         size_t fw_size;
1395         physaddr_t fw_base;
1396         int rc;
1397
1398         /* Get firmware parameters */
1399         if ( ( rc = arbel_cmd_query_fw ( arbel, &fw ) ) != 0 ) {
1400                 DBGC ( arbel, "Arbel %p could not query firmware: %s\n",
1401                        arbel, strerror ( rc ) );
1402                 goto err_query_fw;
1403         }
1404         DBGC ( arbel, "Arbel %p firmware version %ld.%ld.%ld\n", arbel,
1405                MLX_GET ( &fw, fw_rev_major ), MLX_GET ( &fw, fw_rev_minor ),
1406                MLX_GET ( &fw, fw_rev_subminor ) );
1407         fw_pages = MLX_GET ( &fw, fw_pages );
1408         log2_fw_pages = fls ( fw_pages - 1 );
1409         fw_pages = ( 1 << log2_fw_pages );
1410         DBGC ( arbel, "Arbel %p requires %d kB for firmware\n",
1411                arbel, ( fw_pages * 4 ) );
1412
1413         /* Allocate firmware pages and map firmware area */
1414         fw_size = ( fw_pages * 4096 );
1415         arbel->firmware_area = umalloc ( fw_size );
1416         if ( ! arbel->firmware_area ) {
1417                 rc = -ENOMEM;
1418                 goto err_alloc_fa;
1419         }
1420         fw_base = ( user_to_phys ( arbel->firmware_area, fw_size ) &
1421                     ~( fw_size - 1 ) );
1422         DBGC ( arbel, "Arbel %p firmware area at physical [%lx,%lx)\n",
1423                arbel, fw_base, ( fw_base + fw_size ) );
1424         memset ( &map_fa, 0, sizeof ( map_fa ) );
1425         MLX_FILL_2 ( &map_fa, 3,
1426                      log2size, log2_fw_pages,
1427                      pa_l, ( fw_base >> 12 ) );
1428         if ( ( rc = arbel_cmd_map_fa ( arbel, &map_fa ) ) != 0 ) {
1429                 DBGC ( arbel, "Arbel %p could not map firmware: %s\n",
1430                        arbel, strerror ( rc ) );
1431                 goto err_map_fa;
1432         }
1433
1434         /* Start firmware */
1435         if ( ( rc = arbel_cmd_run_fw ( arbel ) ) != 0 ) {
1436                 DBGC ( arbel, "Arbel %p could not run firmware: %s\n",
1437                        arbel, strerror ( rc ) );
1438                 goto err_run_fw;
1439         }
1440
1441         DBGC ( arbel, "Arbel %p firmware started\n", arbel );
1442         return 0;
1443
1444  err_run_fw:
1445         arbel_cmd_unmap_fa ( arbel );
1446  err_map_fa:
1447         ufree ( arbel->firmware_area );
1448         arbel->firmware_area = UNULL;
1449  err_alloc_fa:
1450  err_query_fw:
1451         return rc;
1452 }
1453
1454 /**
1455  * Stop firmware running
1456  *
1457  * @v arbel             Arbel device
1458  */
1459 static void arbel_stop_firmware ( struct arbel *arbel ) {
1460         int rc;
1461
1462         if ( ( rc = arbel_cmd_unmap_fa ( arbel ) ) != 0 ) {
1463                 DBGC ( arbel, "Arbel %p FATAL could not stop firmware: %s\n",
1464                        arbel, strerror ( rc ) );
1465                 /* Leak memory and return; at least we avoid corruption */
1466                 return;
1467         }
1468         ufree ( arbel->firmware_area );
1469         arbel->firmware_area = UNULL;
1470 }
1471
1472 /***************************************************************************
1473  *
1474  * Infinihost Context Memory management
1475  *
1476  ***************************************************************************
1477  */
1478
1479 /**
1480  * Get device limits
1481  *
1482  * @v arbel             Arbel device
1483  * @ret rc              Return status code
1484  */
1485 static int arbel_get_limits ( struct arbel *arbel ) {
1486         struct arbelprm_query_dev_lim dev_lim;
1487         int rc;
1488
1489         if ( ( rc = arbel_cmd_query_dev_lim ( arbel, &dev_lim ) ) != 0 ) {
1490                 DBGC ( arbel, "Arbel %p could not get device limits: %s\n",
1491                        arbel, strerror ( rc ) );
1492                 return rc;
1493         }
1494
1495         arbel->limits.reserved_qps =
1496                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_qps ) );
1497         arbel->limits.qpc_entry_size = MLX_GET ( &dev_lim, qpc_entry_sz );
1498         arbel->limits.eqpc_entry_size = MLX_GET ( &dev_lim, eqpc_entry_sz );
1499         arbel->limits.reserved_srqs =
1500                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_srqs ) );
1501         arbel->limits.srqc_entry_size = MLX_GET ( &dev_lim, srq_entry_sz );
1502         arbel->limits.reserved_ees =
1503                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_ees ) );
1504         arbel->limits.eec_entry_size = MLX_GET ( &dev_lim, eec_entry_sz );
1505         arbel->limits.eeec_entry_size = MLX_GET ( &dev_lim, eeec_entry_sz );
1506         arbel->limits.reserved_cqs =
1507                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_cqs ) );
1508         arbel->limits.cqc_entry_size = MLX_GET ( &dev_lim, cqc_entry_sz );
1509         arbel->limits.reserved_mtts =
1510                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_mtts ) );
1511         arbel->limits.mtt_entry_size = MLX_GET ( &dev_lim, mtt_entry_sz );
1512         arbel->limits.reserved_mrws =
1513                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_mrws ) );
1514         arbel->limits.mpt_entry_size = MLX_GET ( &dev_lim, mpt_entry_sz );
1515         arbel->limits.reserved_rdbs =
1516                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_rdbs ) );
1517         arbel->limits.eqc_entry_size = MLX_GET ( &dev_lim, eqc_entry_sz );
1518         arbel->limits.reserved_uars = MLX_GET ( &dev_lim, num_rsvd_uars );
1519
1520         return 0;
1521 }
1522
1523 /**
1524  * Get ICM usage
1525  *
1526  * @v log_num_entries   Log2 of the number of entries
1527  * @v entry_size        Entry size
1528  * @ret usage           Usage size in ICM
1529  */
1530 static size_t icm_usage ( unsigned int log_num_entries, size_t entry_size ) {
1531         size_t usage;
1532
1533         usage = ( ( 1 << log_num_entries ) * entry_size );
1534         usage = ( ( usage + 4095 ) & ~4095 );
1535         return usage;
1536 }
1537
1538 /**
1539  * Partition ICM
1540  *
1541  * @v arbel             Arbel device
1542  * @ret rc              Return status code
1543  */
1544 static int arbel_alloc_icm ( struct arbel *arbel ) {
1545         struct arbelprm_init_hca init_hca;
1546         size_t icm_offset = 0;
1547         unsigned int log_num_qps, log_num_srqs, log_num_ees, log_num_cqs;
1548         unsigned int log_num_mtts, log_num_mpts, log_num_rdbs, log_num_eqs;
1549
1550         memset ( &init_hca, 0, sizeof ( init_hca ) );
1551         icm_offset = ( ( arbel->limits.reserved_uars + 1 ) << 12 );
1552
1553         /* Queue pair contexts */
1554         log_num_qps = fls ( arbel->limits.reserved_qps + ARBEL_MAX_QPS - 1 );
1555         MLX_FILL_2 ( &init_hca, 13,
1556                      qpc_eec_cqc_eqc_rdb_parameters.qpc_base_addr_l,
1557                      ( icm_offset >> 7 ),
1558                      qpc_eec_cqc_eqc_rdb_parameters.log_num_of_qp,
1559                      log_num_qps );
1560         icm_offset += icm_usage ( log_num_qps, arbel->limits.qpc_entry_size );
1561
1562         /* Extended queue pair contexts */
1563         MLX_FILL_1 ( &init_hca, 25,
1564                      qpc_eec_cqc_eqc_rdb_parameters.eqpc_base_addr_l,
1565                      icm_offset );
1566         icm_offset += icm_usage ( log_num_qps, arbel->limits.eqpc_entry_size );
1567
1568         /* Shared receive queue contexts */
1569         log_num_srqs = fls ( arbel->limits.reserved_srqs - 1 );
1570         MLX_FILL_2 ( &init_hca, 19,
1571                      qpc_eec_cqc_eqc_rdb_parameters.srqc_base_addr_l,
1572                      ( icm_offset >> 5 ),
1573                      qpc_eec_cqc_eqc_rdb_parameters.log_num_of_srq,
1574                      log_num_srqs );
1575         icm_offset += icm_usage ( log_num_srqs, arbel->limits.srqc_entry_size );
1576
1577         /* End-to-end contexts */
1578         log_num_ees = fls ( arbel->limits.reserved_ees - 1 );
1579         MLX_FILL_2 ( &init_hca, 17,
1580                      qpc_eec_cqc_eqc_rdb_parameters.eec_base_addr_l,
1581                      ( icm_offset >> 7 ),
1582                      qpc_eec_cqc_eqc_rdb_parameters.log_num_of_ee,
1583                      log_num_ees );
1584         icm_offset += icm_usage ( log_num_ees, arbel->limits.eec_entry_size );
1585
1586         /* Extended end-to-end contexts */
1587         MLX_FILL_1 ( &init_hca, 29,
1588                      qpc_eec_cqc_eqc_rdb_parameters.eeec_base_addr_l,
1589                      icm_offset );
1590         icm_offset += icm_usage ( log_num_ees, arbel->limits.eeec_entry_size );
1591
1592         /* Completion queue contexts */
1593         log_num_cqs = fls ( arbel->limits.reserved_cqs + ARBEL_MAX_CQS - 1 );
1594         MLX_FILL_2 ( &init_hca, 21,
1595                      qpc_eec_cqc_eqc_rdb_parameters.cqc_base_addr_l,
1596                      ( icm_offset >> 6 ),
1597                      qpc_eec_cqc_eqc_rdb_parameters.log_num_of_cq,
1598                      log_num_cqs );
1599         icm_offset += icm_usage ( log_num_cqs, arbel->limits.cqc_entry_size );
1600
1601         /* Memory translation table */
1602         log_num_mtts = fls ( arbel->limits.reserved_mtts - 1 );
1603         MLX_FILL_1 ( &init_hca, 65,
1604                      tpt_parameters.mtt_base_addr_l, icm_offset );
1605         icm_offset += icm_usage ( log_num_mtts, arbel->limits.mtt_entry_size );
1606
1607         /* Memory protection table */
1608         log_num_mpts = fls ( arbel->limits.reserved_mrws - 1 );
1609         MLX_FILL_1 ( &init_hca, 61,
1610                      tpt_parameters.mpt_base_adr_l, icm_offset );
1611         MLX_FILL_1 ( &init_hca, 62,
1612                      tpt_parameters.log_mpt_sz, log_num_mpts );
1613         icm_offset += icm_usage ( log_num_mpts, arbel->limits.mpt_entry_size );
1614
1615         /* RDMA something or other */
1616         log_num_rdbs = fls ( arbel->limits.reserved_rdbs - 1 );
1617         MLX_FILL_1 ( &init_hca, 37,
1618                      qpc_eec_cqc_eqc_rdb_parameters.rdb_base_addr_l,
1619                      icm_offset );
1620         icm_offset += icm_usage ( log_num_rdbs, 32 );
1621
1622         /* Event queue contexts */
1623         log_num_eqs = 6;
1624         MLX_FILL_2 ( &init_hca, 33,
1625                      qpc_eec_cqc_eqc_rdb_parameters.eqc_base_addr_l,
1626                      ( icm_offset >> 6 ),
1627                      qpc_eec_cqc_eqc_rdb_parameters.log_num_eq,
1628                      log_num_eqs );
1629         icm_offset += ( ( 1 << log_num_eqs ) * arbel->limits.eqc_entry_size );
1630
1631         /* Multicast table */
1632         MLX_FILL_1 ( &init_hca, 49,
1633                      multicast_parameters.mc_base_addr_l, icm_offset );
1634         MLX_FILL_1 ( &init_hca, 52,
1635                      multicast_parameters.log_mc_table_entry_sz,
1636                      fls ( sizeof ( struct arbelprm_mgm_entry ) - 1 ) );
1637         MLX_FILL_1 ( &init_hca, 53,
1638                      multicast_parameters.mc_table_hash_sz, 8 );
1639         MLX_FILL_1 ( &init_hca, 54,
1640                      multicast_parameters.log_mc_table_sz, 3 );
1641         icm_offset += ( 8 * sizeof ( struct arbelprm_mgm_entry ) );
1642
1643         return 0;
1644 }
1645
1646 /***************************************************************************
1647  *
1648  * PCI interface
1649  *
1650  ***************************************************************************
1651  */
1652
1653 /**
1654  * Probe PCI device
1655  *
1656  * @v pci               PCI device
1657  * @v id                PCI ID
1658  * @ret rc              Return status code
1659  */
1660 static int arbel_probe ( struct pci_device *pci,
1661                          const struct pci_device_id *id __unused ) {
1662         struct ib_device *ibdev;
1663         struct arbel *arbel;
1664         int rc;
1665
1666         /* Allocate Infiniband device */
1667         ibdev = alloc_ibdev ( sizeof ( *arbel ) );
1668         if ( ! ibdev ) {
1669                 rc = -ENOMEM;
1670                 goto err_ibdev;
1671         }
1672         ibdev->op = &arbel_ib_operations;
1673         pci_set_drvdata ( pci, ibdev );
1674         ibdev->dev = &pci->dev;
1675         arbel = ibdev->dev_priv;
1676         memset ( arbel, 0, sizeof ( *arbel ) );
1677
1678         /* Fix up PCI device */
1679         adjust_pci_device ( pci );
1680
1681         /* Get PCI BARs */
1682         arbel->config = ioremap ( pci_bar_start ( pci, ARBEL_PCI_CONFIG_BAR ),
1683                                   ARBEL_PCI_CONFIG_BAR_SIZE );
1684         arbel->uar = ioremap ( ( pci_bar_start ( pci, ARBEL_PCI_UAR_BAR ) +
1685                                  ARBEL_PCI_UAR_IDX * ARBEL_PCI_UAR_SIZE ),
1686                                ARBEL_PCI_UAR_SIZE );
1687
1688         /* Allocate space for mailboxes */
1689         arbel->mailbox_in = malloc_dma ( ARBEL_MBOX_SIZE, ARBEL_MBOX_ALIGN );
1690         if ( ! arbel->mailbox_in ) {
1691                 rc = -ENOMEM;
1692                 goto err_mailbox_in;
1693         }
1694         arbel->mailbox_out = malloc_dma ( ARBEL_MBOX_SIZE, ARBEL_MBOX_ALIGN );
1695         if ( ! arbel->mailbox_out ) {
1696                 rc = -ENOMEM;
1697                 goto err_mailbox_out;
1698         }
1699
1700         /* Start firmware */
1701         if ( ( rc = arbel_start_firmware ( arbel ) ) != 0 )
1702                 goto err_start_firmware;
1703
1704         /* Get device limits */
1705         if ( ( rc = arbel_get_limits ( arbel ) ) != 0 )
1706                 goto err_get_limits;
1707
1708         while ( 1 ) {}
1709
1710 #if 0
1711         /* Initialise hardware */
1712         if ( ( rc = ib_driver_init ( pci, &qph ) ) != 0 )
1713                 goto err_ib_driver_init;
1714
1715         /* Hack up IB structures */
1716
1717         arbel->config = memfree_pci_dev.cr_space;
1718         arbel->uar = memfree_pci_dev.uar;
1719         arbel->mailbox_in = dev_buffers_p->inprm_buf;
1720         arbel->mailbox_out = dev_buffers_p->outprm_buf;
1721 #endif
1722         arbel->db_rec = dev_ib_data.uar_context_base;
1723         arbel->reserved_lkey = dev_ib_data.mkey;
1724         arbel->eqn = dev_ib_data.eq.eqn;
1725
1726
1727         /* Get MAD parameters */
1728         if ( ( rc = arbel_get_mad_params ( ibdev ) ) != 0 )
1729                 goto err_get_mad_params;
1730
1731
1732         /* Add IPoIB device */
1733         if ( ( rc = ipoib_probe ( ibdev ) ) != 0 ) {
1734                 DBGC ( arbel, "Arbel %p could not add IPoIB device: %s\n",
1735                        arbel, strerror ( rc ) );
1736                 goto err_ipoib_probe;
1737         }
1738
1739         return 0;
1740
1741  err_ipoib_probe:
1742  err_get_mad_params:
1743         ib_driver_close ( 0 );
1744  err_ib_driver_init:
1745
1746
1747  err_get_limits:
1748         arbel_stop_firmware ( arbel );
1749  err_start_firmware:
1750         free_dma ( arbel->mailbox_out, ARBEL_MBOX_SIZE );
1751  err_mailbox_out:
1752         free_dma ( arbel->mailbox_in, ARBEL_MBOX_SIZE );
1753  err_mailbox_in:
1754         free_ibdev ( ibdev );
1755  err_ibdev:
1756         return rc;
1757 }
1758
1759 /**
1760  * Remove PCI device
1761  *
1762  * @v pci               PCI device
1763  */
1764 static void arbel_remove ( struct pci_device *pci ) {
1765         struct ib_device *ibdev = pci_get_drvdata ( pci );
1766         struct arbel *arbel = ibdev->dev_priv;
1767
1768
1769 #warning "check error sequence for probe()"
1770         ipoib_remove ( ibdev );
1771         ib_driver_close ( 0 );
1772         arbel_stop_firmware ( arbel );
1773         free_dma ( arbel->mailbox_out, ARBEL_MBOX_SIZE );
1774         free_dma ( arbel->mailbox_in, ARBEL_MBOX_SIZE );
1775         free_ibdev ( ibdev );
1776 }
1777
1778 static struct pci_device_id arbel_nics[] = {
1779         PCI_ROM ( 0x15b3, 0x6282, "MT25218", "MT25218 HCA driver" ),
1780         PCI_ROM ( 0x15b3, 0x6274, "MT25204", "MT25204 HCA driver" ),
1781 };
1782
1783 struct pci_driver arbel_driver __pci_driver = {
1784         .ids = arbel_nics,
1785         .id_count = ( sizeof ( arbel_nics ) / sizeof ( arbel_nics[0] ) ),
1786         .probe = arbel_probe,
1787         .remove = arbel_remove,
1788 };