2649a4aba4d8cdd2e818719342ae793c44c12407
[people/mcb30/gpxe.git] / src / drivers / infiniband / arbel.c
1 /*
2  * Copyright (C) 2007 Michael Brown <mbrown@fensystems.co.uk>.
3  *
4  * Based in part upon the original driver by Mellanox Technologies
5  * Ltd.  Portions may be Copyright (c) Mellanox Technologies Ltd.
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 of the
10  * License, or any later version.
11  *
12  * This program is distributed in the hope that it will be useful, but
13  * WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, write to the Free Software
19  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20  */
21
22 #include <stdint.h>
23 #include <stdlib.h>
24 #include <stdio.h>
25 #include <string.h>
26 #include <strings.h>
27 #include <unistd.h>
28 #include <errno.h>
29 #include <byteswap.h>
30 #include <gpxe/pci.h>
31 #include <gpxe/malloc.h>
32 #include <gpxe/umalloc.h>
33 #include <gpxe/iobuf.h>
34 #include <gpxe/netdevice.h>
35 #include <gpxe/infiniband.h>
36 #include "arbel.h"
37
38 /**
39  * @file
40  *
41  * Mellanox Arbel Infiniband HCA
42  *
43  */
44
45 /***************************************************************************
46  *
47  * Queue number allocation
48  *
49  ***************************************************************************
50  */
51
52 /**
53  * Allocate queue number
54  *
55  * @v q_inuse           Queue usage bitmask
56  * @v max_inuse         Maximum number of in-use queues
57  * @ret qn_offset       Free queue number offset, or negative error
58  */
59 static int arbel_alloc_qn_offset ( arbel_bitmask_t *q_inuse,
60                                    unsigned int max_inuse ) {
61         unsigned int qn_offset = 0;
62         arbel_bitmask_t mask = 1;
63
64         while ( qn_offset < max_inuse ) {
65                 if ( ( mask & *q_inuse ) == 0 ) {
66                         *q_inuse |= mask;
67                         return qn_offset;
68                 }
69                 qn_offset++;
70                 mask <<= 1;
71                 if ( ! mask ) {
72                         mask = 1;
73                         q_inuse++;
74                 }
75         }
76         return -ENFILE;
77 }
78
79 /**
80  * Free queue number
81  *
82  * @v q_inuse           Queue usage bitmask
83  * @v qn_offset         Queue number offset
84  */
85 static void arbel_free_qn_offset ( arbel_bitmask_t *q_inuse, int qn_offset ) {
86         arbel_bitmask_t mask;
87
88         mask = ( 1 << ( qn_offset % ( 8 * sizeof ( mask ) ) ) );
89         q_inuse += ( qn_offset / ( 8 * sizeof ( mask ) ) );
90         *q_inuse &= ~mask;
91 }
92
93 /***************************************************************************
94  *
95  * HCA commands
96  *
97  ***************************************************************************
98  */
99
100 /**
101  * Wait for Arbel command completion
102  *
103  * @v arbel             Arbel device
104  * @ret rc              Return status code
105  */
106 static int arbel_cmd_wait ( struct arbel *arbel,
107                             struct arbelprm_hca_command_register *hcr ) {
108         unsigned int wait;
109
110         for ( wait = ARBEL_HCR_MAX_WAIT_MS ; wait ; wait-- ) {
111                 hcr->u.dwords[6] =
112                         readl ( arbel->config + ARBEL_HCR_REG ( 6 ) );
113                 if ( MLX_GET ( hcr, go ) == 0 )
114                         return 0;
115                 mdelay ( 1 );
116         }
117         return -EBUSY;
118 }
119
120 /**
121  * Issue HCA command
122  *
123  * @v arbel             Arbel device
124  * @v command           Command opcode, flags and input/output lengths
125  * @v op_mod            Opcode modifier (0 if no modifier applicable)
126  * @v in                Input parameters
127  * @v in_mod            Input modifier (0 if no modifier applicable)
128  * @v out               Output parameters
129  * @ret rc              Return status code
130  */
131 static int arbel_cmd ( struct arbel *arbel, unsigned long command,
132                        unsigned int op_mod, const void *in,
133                        unsigned int in_mod, void *out ) {
134         struct arbelprm_hca_command_register hcr;
135         unsigned int opcode = ARBEL_HCR_OPCODE ( command );
136         size_t in_len = ARBEL_HCR_IN_LEN ( command );
137         size_t out_len = ARBEL_HCR_OUT_LEN ( command );
138         void *in_buffer;
139         void *out_buffer;
140         unsigned int status;
141         unsigned int i;
142         int rc;
143
144         assert ( in_len <= ARBEL_MBOX_SIZE );
145         assert ( out_len <= ARBEL_MBOX_SIZE );
146
147         DBGC2 ( arbel, "Arbel %p command %02x in %zx%s out %zx%s\n",
148                 arbel, opcode, in_len,
149                 ( ( command & ARBEL_HCR_IN_MBOX ) ? "(mbox)" : "" ), out_len,
150                 ( ( command & ARBEL_HCR_OUT_MBOX ) ? "(mbox)" : "" ) );
151
152         /* Check that HCR is free */
153         if ( ( rc = arbel_cmd_wait ( arbel, &hcr ) ) != 0 ) {
154                 DBGC ( arbel, "Arbel %p command interface locked\n", arbel );
155                 return rc;
156         }
157
158         /* Prepare HCR */
159         memset ( &hcr, 0, sizeof ( hcr ) );
160         in_buffer = &hcr.u.dwords[0];
161         if ( in_len && ( command & ARBEL_HCR_IN_MBOX ) ) {
162                 in_buffer = arbel->mailbox_in;
163                 MLX_FILL_1 ( &hcr, 1, in_param_l, virt_to_bus ( in_buffer ) );
164         }
165         memcpy ( in_buffer, in, in_len );
166         MLX_FILL_1 ( &hcr, 2, input_modifier, in_mod );
167         out_buffer = &hcr.u.dwords[3];
168         if ( out_len && ( command & ARBEL_HCR_OUT_MBOX ) ) {
169                 out_buffer = arbel->mailbox_out;
170                 MLX_FILL_1 ( &hcr, 4, out_param_l,
171                              virt_to_bus ( out_buffer ) );
172         }
173         MLX_FILL_3 ( &hcr, 6,
174                      opcode, opcode,
175                      opcode_modifier, op_mod,
176                      go, 1 );
177         DBGC2_HD ( arbel, &hcr, sizeof ( hcr ) );
178         if ( in_len ) {
179                 DBGC2 ( arbel, "Input:\n" );
180                 DBGC2_HD ( arbel, in, ( ( in_len < 512 ) ? in_len : 512 ) );
181         }
182
183         /* Issue command */
184         for ( i = 0 ; i < ( sizeof ( hcr ) / sizeof ( hcr.u.dwords[0] ) ) ;
185               i++ ) {
186                 writel ( hcr.u.dwords[i],
187                          arbel->config + ARBEL_HCR_REG ( i ) );
188                 barrier();
189         }
190
191         /* Wait for command completion */
192         if ( ( rc = arbel_cmd_wait ( arbel, &hcr ) ) != 0 ) {
193                 DBGC ( arbel, "Arbel %p timed out waiting for command:\n",
194                        arbel );
195                 DBGC_HD ( arbel, &hcr, sizeof ( hcr ) );
196                 return rc;
197         }
198
199         /* Check command status */
200         status = MLX_GET ( &hcr, status );
201         if ( status != 0 ) {
202                 DBGC ( arbel, "Arbel %p command failed with status %02x:\n",
203                        arbel, status );
204                 DBGC_HD ( arbel, &hcr, sizeof ( hcr ) );
205                 return -EIO;
206         }
207
208         /* Read output parameters, if any */
209         hcr.u.dwords[3] = readl ( arbel->config + ARBEL_HCR_REG ( 3 ) );
210         hcr.u.dwords[4] = readl ( arbel->config + ARBEL_HCR_REG ( 4 ) );
211         memcpy ( out, out_buffer, out_len );
212         if ( out_len ) {
213                 DBGC2 ( arbel, "Output:\n" );
214                 DBGC2_HD ( arbel, out, ( ( out_len < 512 ) ? out_len : 512 ) );
215         }
216
217         return 0;
218 }
219
220 static inline int
221 arbel_cmd_query_dev_lim ( struct arbel *arbel,
222                           struct arbelprm_query_dev_lim *dev_lim ) {
223         return arbel_cmd ( arbel,
224                            ARBEL_HCR_OUT_CMD ( ARBEL_HCR_QUERY_DEV_LIM,
225                                                1, sizeof ( *dev_lim ) ),
226                            0, NULL, 0, dev_lim );
227 }
228
229 static inline int
230 arbel_cmd_query_fw ( struct arbel *arbel, struct arbelprm_query_fw *fw ) {
231         return arbel_cmd ( arbel,
232                            ARBEL_HCR_OUT_CMD ( ARBEL_HCR_QUERY_FW, 
233                                                1, sizeof ( *fw ) ),
234                            0, NULL, 0, fw );
235 }
236
237 static inline int
238 arbel_cmd_init_hca ( struct arbel *arbel,
239                      const struct arbelprm_init_hca *init_hca ) {
240         return arbel_cmd ( arbel,
241                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_INIT_HCA,
242                                               1, sizeof ( *init_hca ) ),
243                            0, init_hca, 0, NULL );
244 }
245
246 static inline int
247 arbel_cmd_close_hca ( struct arbel *arbel ) {
248         return arbel_cmd ( arbel,
249                            ARBEL_HCR_VOID_CMD ( ARBEL_HCR_CLOSE_HCA ),
250                            0, NULL, 0, NULL );
251 }
252
253 static inline int
254 arbel_cmd_init_ib ( struct arbel *arbel, unsigned int port,
255                     const struct arbelprm_init_ib *init_ib ) {
256         return arbel_cmd ( arbel,
257                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_INIT_IB,
258                                               1, sizeof ( *init_ib ) ),
259                            0, init_ib, port, NULL );
260 }
261
262 static inline int
263 arbel_cmd_close_ib ( struct arbel *arbel, unsigned int port ) {
264         return arbel_cmd ( arbel,
265                            ARBEL_HCR_VOID_CMD ( ARBEL_HCR_CLOSE_IB ),
266                            0, NULL, port, NULL );
267 }
268
269 static inline int
270 arbel_cmd_sw2hw_mpt ( struct arbel *arbel, unsigned int index,
271                       const struct arbelprm_mpt *mpt ) {
272         return arbel_cmd ( arbel,
273                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_SW2HW_MPT,
274                                               1, sizeof ( *mpt ) ),
275                            0, mpt, index, NULL );
276 }
277
278 static inline int
279 arbel_cmd_map_eq ( struct arbel *arbel, unsigned long index_map,
280                    const struct arbelprm_event_mask *mask ) {
281         return arbel_cmd ( arbel,
282                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_MAP_EQ,
283                                               0, sizeof ( *mask ) ),
284                            0, mask, index_map, NULL );
285 }
286
287 static inline int
288 arbel_cmd_sw2hw_eq ( struct arbel *arbel, unsigned int index,
289                      const struct arbelprm_eqc *eqctx ) {
290         return arbel_cmd ( arbel,
291                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_SW2HW_EQ,
292                                               1, sizeof ( *eqctx ) ),
293                            0, eqctx, index, NULL );
294 }
295
296 static inline int
297 arbel_cmd_hw2sw_eq ( struct arbel *arbel, unsigned int index,
298                      struct arbelprm_eqc *eqctx ) {
299         return arbel_cmd ( arbel,
300                            ARBEL_HCR_OUT_CMD ( ARBEL_HCR_HW2SW_EQ,
301                                                1, sizeof ( *eqctx ) ),
302                            1, NULL, index, eqctx );
303 }
304
305 static inline int
306 arbel_cmd_sw2hw_cq ( struct arbel *arbel, unsigned long cqn,
307                      const struct arbelprm_completion_queue_context *cqctx ) {
308         return arbel_cmd ( arbel,
309                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_SW2HW_CQ,
310                                               1, sizeof ( *cqctx ) ),
311                            0, cqctx, cqn, NULL );
312 }
313
314 static inline int
315 arbel_cmd_hw2sw_cq ( struct arbel *arbel, unsigned long cqn,
316                      struct arbelprm_completion_queue_context *cqctx) {
317         return arbel_cmd ( arbel,
318                            ARBEL_HCR_OUT_CMD ( ARBEL_HCR_HW2SW_CQ,
319                                                1, sizeof ( *cqctx ) ),
320                            0, NULL, cqn, cqctx );
321 }
322
323 static inline int
324 arbel_cmd_rst2init_qpee ( struct arbel *arbel, unsigned long qpn,
325                           const struct arbelprm_qp_ee_state_transitions *ctx ){
326         return arbel_cmd ( arbel,
327                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_RST2INIT_QPEE,
328                                               1, sizeof ( *ctx ) ),
329                            0, ctx, qpn, NULL );
330 }
331
332 static inline int
333 arbel_cmd_init2rtr_qpee ( struct arbel *arbel, unsigned long qpn,
334                           const struct arbelprm_qp_ee_state_transitions *ctx ){
335         return arbel_cmd ( arbel,
336                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_INIT2RTR_QPEE,
337                                               1, sizeof ( *ctx ) ),
338                            0, ctx, qpn, NULL );
339 }
340
341 static inline int
342 arbel_cmd_rtr2rts_qpee ( struct arbel *arbel, unsigned long qpn,
343                          const struct arbelprm_qp_ee_state_transitions *ctx ) {
344         return arbel_cmd ( arbel,
345                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_RTR2RTS_QPEE,
346                                               1, sizeof ( *ctx ) ),
347                            0, ctx, qpn, NULL );
348 }
349
350 static inline int
351 arbel_cmd_rts2rts_qp ( struct arbel *arbel, unsigned long qpn,
352                        const struct arbelprm_qp_ee_state_transitions *ctx ) {
353         return arbel_cmd ( arbel,
354                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_RTS2RTS_QPEE,
355                                               1, sizeof ( *ctx ) ),
356                            0, ctx, qpn, NULL );
357 }
358
359 static inline int
360 arbel_cmd_2rst_qpee ( struct arbel *arbel, unsigned long qpn ) {
361         return arbel_cmd ( arbel,
362                            ARBEL_HCR_VOID_CMD ( ARBEL_HCR_2RST_QPEE ),
363                            0x03, NULL, qpn, NULL );
364 }
365
366 static inline int
367 arbel_cmd_mad_ifc ( struct arbel *arbel, unsigned int port,
368                     union arbelprm_mad *mad ) {
369         return arbel_cmd ( arbel,
370                            ARBEL_HCR_INOUT_CMD ( ARBEL_HCR_MAD_IFC,
371                                                  1, sizeof ( *mad ),
372                                                  1, sizeof ( *mad ) ),
373                            0x03, mad, port, mad );
374 }
375
376 static inline int
377 arbel_cmd_read_mgm ( struct arbel *arbel, unsigned int index,
378                      struct arbelprm_mgm_entry *mgm ) {
379         return arbel_cmd ( arbel,
380                            ARBEL_HCR_OUT_CMD ( ARBEL_HCR_READ_MGM,
381                                                1, sizeof ( *mgm ) ),
382                            0, NULL, index, mgm );
383 }
384
385 static inline int
386 arbel_cmd_write_mgm ( struct arbel *arbel, unsigned int index,
387                       const struct arbelprm_mgm_entry *mgm ) {
388         return arbel_cmd ( arbel,
389                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_WRITE_MGM,
390                                               1, sizeof ( *mgm ) ),
391                            0, mgm, index, NULL );
392 }
393
394 static inline int
395 arbel_cmd_mgid_hash ( struct arbel *arbel, const struct ib_gid *gid,
396                       struct arbelprm_mgm_hash *hash ) {
397         return arbel_cmd ( arbel,
398                            ARBEL_HCR_INOUT_CMD ( ARBEL_HCR_MGID_HASH,
399                                                  1, sizeof ( *gid ),
400                                                  0, sizeof ( *hash ) ),
401                            0, gid, 0, hash );
402 }
403
404 static inline int
405 arbel_cmd_run_fw ( struct arbel *arbel ) {
406         return arbel_cmd ( arbel,
407                            ARBEL_HCR_VOID_CMD ( ARBEL_HCR_RUN_FW ),
408                            0, NULL, 0, NULL );
409 }
410
411 static inline int
412 arbel_cmd_disable_lam ( struct arbel *arbel ) {
413         return arbel_cmd ( arbel,
414                            ARBEL_HCR_VOID_CMD ( ARBEL_HCR_DISABLE_LAM ),
415                            0, NULL, 0, NULL );
416 }
417
418 static inline int
419 arbel_cmd_enable_lam ( struct arbel *arbel, struct arbelprm_access_lam *lam ) {
420         return arbel_cmd ( arbel,
421                            ARBEL_HCR_OUT_CMD ( ARBEL_HCR_ENABLE_LAM,
422                                                1, sizeof ( *lam ) ),
423                            1, NULL, 0, lam );
424 }
425
426 static inline int
427 arbel_cmd_unmap_icm ( struct arbel *arbel, unsigned int page_count ) {
428         return arbel_cmd ( arbel,
429                            ARBEL_HCR_VOID_CMD ( ARBEL_HCR_UNMAP_ICM ),
430                            0, NULL, page_count, NULL );
431 }
432
433 static inline int
434 arbel_cmd_map_icm ( struct arbel *arbel,
435                     const struct arbelprm_virtual_physical_mapping *map ) {
436         return arbel_cmd ( arbel,
437                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_MAP_ICM,
438                                               1, sizeof ( *map ) ),
439                            0, map, 1, NULL );
440 }
441
442 static inline int
443 arbel_cmd_unmap_icm_aux ( struct arbel *arbel ) {
444         return arbel_cmd ( arbel,
445                            ARBEL_HCR_VOID_CMD ( ARBEL_HCR_UNMAP_ICM_AUX ),
446                            0, NULL, 0, NULL );
447 }
448
449 static inline int
450 arbel_cmd_map_icm_aux ( struct arbel *arbel,
451                         const struct arbelprm_virtual_physical_mapping *map ) {
452         return arbel_cmd ( arbel,
453                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_MAP_ICM_AUX,
454                                               1, sizeof ( *map ) ),
455                            0, map, 1, NULL );
456 }
457
458 static inline int
459 arbel_cmd_set_icm_size ( struct arbel *arbel,
460                          const struct arbelprm_scalar_parameter *icm_size,
461                          struct arbelprm_scalar_parameter *icm_aux_size ) {
462         return arbel_cmd ( arbel,
463                            ARBEL_HCR_INOUT_CMD ( ARBEL_HCR_SET_ICM_SIZE,
464                                                  0, sizeof ( *icm_size ),
465                                                  0, sizeof ( *icm_aux_size ) ),
466                            0, icm_size, 0, icm_aux_size );
467 }
468
469 static inline int
470 arbel_cmd_unmap_fa ( struct arbel *arbel ) {
471         return arbel_cmd ( arbel,
472                            ARBEL_HCR_VOID_CMD ( ARBEL_HCR_UNMAP_FA ),
473                            0, NULL, 0, NULL );
474 }
475
476 static inline int
477 arbel_cmd_map_fa ( struct arbel *arbel,
478                    const struct arbelprm_virtual_physical_mapping *map ) {
479         return arbel_cmd ( arbel,
480                            ARBEL_HCR_IN_CMD ( ARBEL_HCR_MAP_FA,
481                                               1, sizeof ( *map ) ),
482                            0, map, 1, NULL );
483 }
484
485 /***************************************************************************
486  *
487  * Completion queue operations
488  *
489  ***************************************************************************
490  */
491
492 /**
493  * Create completion queue
494  *
495  * @v ibdev             Infiniband device
496  * @v cq                Completion queue
497  * @ret rc              Return status code
498  */
499 static int arbel_create_cq ( struct ib_device *ibdev,
500                              struct ib_completion_queue *cq ) {
501         struct arbel *arbel = ib_get_drvdata ( ibdev );
502         struct arbel_completion_queue *arbel_cq;
503         struct arbelprm_completion_queue_context cqctx;
504         struct arbelprm_cq_ci_db_record *ci_db_rec;
505         struct arbelprm_cq_arm_db_record *arm_db_rec;
506         int cqn_offset;
507         unsigned int i;
508         int rc;
509
510         /* Find a free completion queue number */
511         cqn_offset = arbel_alloc_qn_offset ( arbel->cq_inuse, ARBEL_MAX_CQS );
512         if ( cqn_offset < 0 ) {
513                 DBGC ( arbel, "Arbel %p out of completion queues\n", arbel );
514                 rc = cqn_offset;
515                 goto err_cqn_offset;
516         }
517         cq->cqn = ( arbel->limits.reserved_cqs + cqn_offset );
518
519         /* Allocate control structures */
520         arbel_cq = zalloc ( sizeof ( *arbel_cq ) );
521         if ( ! arbel_cq ) {
522                 rc = -ENOMEM;
523                 goto err_arbel_cq;
524         }
525         arbel_cq->ci_doorbell_idx = arbel_cq_ci_doorbell_idx ( cqn_offset );
526         arbel_cq->arm_doorbell_idx = arbel_cq_arm_doorbell_idx ( cqn_offset );
527
528         /* Allocate completion queue itself */
529         arbel_cq->cqe_size = ( cq->num_cqes * sizeof ( arbel_cq->cqe[0] ) );
530         arbel_cq->cqe = malloc_dma ( arbel_cq->cqe_size,
531                                      sizeof ( arbel_cq->cqe[0] ) );
532         if ( ! arbel_cq->cqe ) {
533                 rc = -ENOMEM;
534                 goto err_cqe;
535         }
536         memset ( arbel_cq->cqe, 0, arbel_cq->cqe_size );
537         for ( i = 0 ; i < cq->num_cqes ; i++ ) {
538                 MLX_FILL_1 ( &arbel_cq->cqe[i].normal, 7, owner, 1 );
539         }
540         barrier();
541
542         /* Initialise doorbell records */
543         ci_db_rec = &arbel->db_rec[arbel_cq->ci_doorbell_idx].cq_ci;
544         MLX_FILL_1 ( ci_db_rec, 0, counter, 0 );
545         MLX_FILL_2 ( ci_db_rec, 1,
546                      res, ARBEL_UAR_RES_CQ_CI,
547                      cq_number, cq->cqn );
548         arm_db_rec = &arbel->db_rec[arbel_cq->arm_doorbell_idx].cq_arm;
549         MLX_FILL_1 ( arm_db_rec, 0, counter, 0 );
550         MLX_FILL_2 ( arm_db_rec, 1,
551                      res, ARBEL_UAR_RES_CQ_ARM,
552                      cq_number, cq->cqn );
553
554         /* Hand queue over to hardware */
555         memset ( &cqctx, 0, sizeof ( cqctx ) );
556         MLX_FILL_1 ( &cqctx, 0, st, 0xa /* "Event fired" */ );
557         MLX_FILL_1 ( &cqctx, 2, start_address_l,
558                      virt_to_bus ( arbel_cq->cqe ) );
559         MLX_FILL_2 ( &cqctx, 3,
560                      usr_page, arbel->limits.reserved_uars,
561                      log_cq_size, fls ( cq->num_cqes - 1 ) );
562         MLX_FILL_1 ( &cqctx, 5, c_eqn, ARBEL_NO_EQ );
563         MLX_FILL_1 ( &cqctx, 6, pd, ARBEL_GLOBAL_PD );
564         MLX_FILL_1 ( &cqctx, 7, l_key, arbel->reserved_lkey );
565         MLX_FILL_1 ( &cqctx, 12, cqn, cq->cqn );
566         MLX_FILL_1 ( &cqctx, 13,
567                      cq_ci_db_record, arbel_cq->ci_doorbell_idx );
568         MLX_FILL_1 ( &cqctx, 14,
569                      cq_state_db_record, arbel_cq->arm_doorbell_idx );
570         if ( ( rc = arbel_cmd_sw2hw_cq ( arbel, cq->cqn, &cqctx ) ) != 0 ) {
571                 DBGC ( arbel, "Arbel %p SW2HW_CQ failed: %s\n",
572                        arbel, strerror ( rc ) );
573                 goto err_sw2hw_cq;
574         }
575
576         DBGC ( arbel, "Arbel %p CQN %#lx ring at [%p,%p)\n",
577                arbel, cq->cqn, arbel_cq->cqe,
578                ( ( ( void * ) arbel_cq->cqe ) + arbel_cq->cqe_size ) );
579         ib_cq_set_drvdata ( cq, arbel_cq );
580         return 0;
581
582  err_sw2hw_cq:
583         MLX_FILL_1 ( ci_db_rec, 1, res, ARBEL_UAR_RES_NONE );
584         MLX_FILL_1 ( arm_db_rec, 1, res, ARBEL_UAR_RES_NONE );
585         free_dma ( arbel_cq->cqe, arbel_cq->cqe_size );
586  err_cqe:
587         free ( arbel_cq );
588  err_arbel_cq:
589         arbel_free_qn_offset ( arbel->cq_inuse, cqn_offset );
590  err_cqn_offset:
591         return rc;
592 }
593
594 /**
595  * Destroy completion queue
596  *
597  * @v ibdev             Infiniband device
598  * @v cq                Completion queue
599  */
600 static void arbel_destroy_cq ( struct ib_device *ibdev,
601                                struct ib_completion_queue *cq ) {
602         struct arbel *arbel = ib_get_drvdata ( ibdev );
603         struct arbel_completion_queue *arbel_cq = ib_cq_get_drvdata ( cq );
604         struct arbelprm_completion_queue_context cqctx;
605         struct arbelprm_cq_ci_db_record *ci_db_rec;
606         struct arbelprm_cq_arm_db_record *arm_db_rec;
607         int cqn_offset;
608         int rc;
609
610         /* Take ownership back from hardware */
611         if ( ( rc = arbel_cmd_hw2sw_cq ( arbel, cq->cqn, &cqctx ) ) != 0 ) {
612                 DBGC ( arbel, "Arbel %p FATAL HW2SW_CQ failed on CQN %#lx: "
613                        "%s\n", arbel, cq->cqn, strerror ( rc ) );
614                 /* Leak memory and return; at least we avoid corruption */
615                 return;
616         }
617
618         /* Clear doorbell records */
619         ci_db_rec = &arbel->db_rec[arbel_cq->ci_doorbell_idx].cq_ci;
620         arm_db_rec = &arbel->db_rec[arbel_cq->arm_doorbell_idx].cq_arm;
621         MLX_FILL_1 ( ci_db_rec, 1, res, ARBEL_UAR_RES_NONE );
622         MLX_FILL_1 ( arm_db_rec, 1, res, ARBEL_UAR_RES_NONE );
623
624         /* Free memory */
625         free_dma ( arbel_cq->cqe, arbel_cq->cqe_size );
626         free ( arbel_cq );
627
628         /* Mark queue number as free */
629         cqn_offset = ( cq->cqn - arbel->limits.reserved_cqs );
630         arbel_free_qn_offset ( arbel->cq_inuse, cqn_offset );
631
632         ib_cq_set_drvdata ( cq, NULL );
633 }
634
635 /***************************************************************************
636  *
637  * Queue pair operations
638  *
639  ***************************************************************************
640  */
641
642 /**
643  * Create send work queue
644  *
645  * @v arbel_send_wq     Send work queue
646  * @v num_wqes          Number of work queue entries
647  * @ret rc              Return status code
648  */
649 static int arbel_create_send_wq ( struct arbel_send_work_queue *arbel_send_wq,
650                                   unsigned int num_wqes ) {
651         struct arbelprm_ud_send_wqe *wqe;
652         struct arbelprm_ud_send_wqe *next_wqe;
653         unsigned int wqe_idx_mask;
654         unsigned int i;
655
656         /* Allocate work queue */
657         arbel_send_wq->wqe_size = ( num_wqes *
658                                     sizeof ( arbel_send_wq->wqe[0] ) );
659         arbel_send_wq->wqe = malloc_dma ( arbel_send_wq->wqe_size,
660                                           sizeof ( arbel_send_wq->wqe[0] ) );
661         if ( ! arbel_send_wq->wqe )
662                 return -ENOMEM;
663         memset ( arbel_send_wq->wqe, 0, arbel_send_wq->wqe_size );
664
665         /* Link work queue entries */
666         wqe_idx_mask = ( num_wqes - 1 );
667         for ( i = 0 ; i < num_wqes ; i++ ) {
668                 wqe = &arbel_send_wq->wqe[i].ud;
669                 next_wqe = &arbel_send_wq->wqe[ ( i + 1 ) & wqe_idx_mask ].ud;
670                 MLX_FILL_1 ( &wqe->next, 0, nda_31_6,
671                              ( virt_to_bus ( next_wqe ) >> 6 ) );
672         }
673         
674         return 0;
675 }
676
677 /**
678  * Create receive work queue
679  *
680  * @v arbel_recv_wq     Receive work queue
681  * @v num_wqes          Number of work queue entries
682  * @ret rc              Return status code
683  */
684 static int arbel_create_recv_wq ( struct arbel_recv_work_queue *arbel_recv_wq,
685                                   unsigned int num_wqes ) {
686         struct arbelprm_recv_wqe *wqe;
687         struct arbelprm_recv_wqe *next_wqe;
688         unsigned int wqe_idx_mask;
689         size_t nds;
690         unsigned int i;
691         unsigned int j;
692
693         /* Allocate work queue */
694         arbel_recv_wq->wqe_size = ( num_wqes *
695                                     sizeof ( arbel_recv_wq->wqe[0] ) );
696         arbel_recv_wq->wqe = malloc_dma ( arbel_recv_wq->wqe_size,
697                                           sizeof ( arbel_recv_wq->wqe[0] ) );
698         if ( ! arbel_recv_wq->wqe )
699                 return -ENOMEM;
700         memset ( arbel_recv_wq->wqe, 0, arbel_recv_wq->wqe_size );
701
702         /* Link work queue entries */
703         wqe_idx_mask = ( num_wqes - 1 );
704         nds = ( ( offsetof ( typeof ( *wqe ), data ) +
705                   sizeof ( wqe->data[0] ) ) >> 4 );
706         for ( i = 0 ; i < num_wqes ; i++ ) {
707                 wqe = &arbel_recv_wq->wqe[i].recv;
708                 next_wqe = &arbel_recv_wq->wqe[( i + 1 ) & wqe_idx_mask].recv;
709                 MLX_FILL_1 ( &wqe->next, 0, nda_31_6,
710                              ( virt_to_bus ( next_wqe ) >> 6 ) );
711                 MLX_FILL_1 ( &wqe->next, 1, nds, ( sizeof ( *wqe ) / 16 ) );
712                 for ( j = 0 ; ( ( ( void * ) &wqe->data[j] ) <
713                                 ( ( void * ) ( wqe + 1 ) ) ) ; j++ ) {
714                         MLX_FILL_1 ( &wqe->data[j], 1,
715                                      l_key, ARBEL_INVALID_LKEY );
716                 }
717         }
718         
719         return 0;
720 }
721
722 /**
723  * Create queue pair
724  *
725  * @v ibdev             Infiniband device
726  * @v qp                Queue pair
727  * @ret rc              Return status code
728  */
729 static int arbel_create_qp ( struct ib_device *ibdev,
730                              struct ib_queue_pair *qp ) {
731         struct arbel *arbel = ib_get_drvdata ( ibdev );
732         struct arbel_queue_pair *arbel_qp;
733         struct arbelprm_qp_ee_state_transitions qpctx;
734         struct arbelprm_qp_db_record *send_db_rec;
735         struct arbelprm_qp_db_record *recv_db_rec;
736         int qpn_offset;
737         int rc;
738
739         /* Find a free queue pair number */
740         qpn_offset = arbel_alloc_qn_offset ( arbel->qp_inuse, ARBEL_MAX_QPS );
741         if ( qpn_offset < 0 ) {
742                 DBGC ( arbel, "Arbel %p out of queue pairs\n", arbel );
743                 rc = qpn_offset;
744                 goto err_qpn_offset;
745         }
746         qp->qpn = ( ARBEL_QPN_BASE + arbel->limits.reserved_qps + qpn_offset );
747
748         /* Allocate control structures */
749         arbel_qp = zalloc ( sizeof ( *arbel_qp ) );
750         if ( ! arbel_qp ) {
751                 rc = -ENOMEM;
752                 goto err_arbel_qp;
753         }
754         arbel_qp->send.doorbell_idx = arbel_send_doorbell_idx ( qpn_offset );
755         arbel_qp->recv.doorbell_idx = arbel_recv_doorbell_idx ( qpn_offset );
756
757         /* Create send and receive work queues */
758         if ( ( rc = arbel_create_send_wq ( &arbel_qp->send,
759                                            qp->send.num_wqes ) ) != 0 )
760                 goto err_create_send_wq;
761         if ( ( rc = arbel_create_recv_wq ( &arbel_qp->recv,
762                                            qp->recv.num_wqes ) ) != 0 )
763                 goto err_create_recv_wq;
764
765         /* Initialise doorbell records */
766         send_db_rec = &arbel->db_rec[arbel_qp->send.doorbell_idx].qp;
767         MLX_FILL_1 ( send_db_rec, 0, counter, 0 );
768         MLX_FILL_2 ( send_db_rec, 1,
769                      res, ARBEL_UAR_RES_SQ,
770                      qp_number, qp->qpn );
771         recv_db_rec = &arbel->db_rec[arbel_qp->recv.doorbell_idx].qp;
772         MLX_FILL_1 ( recv_db_rec, 0, counter, 0 );
773         MLX_FILL_2 ( recv_db_rec, 1,
774                      res, ARBEL_UAR_RES_RQ,
775                      qp_number, qp->qpn );
776
777         /* Hand queue over to hardware */
778         memset ( &qpctx, 0, sizeof ( qpctx ) );
779         MLX_FILL_3 ( &qpctx, 2,
780                      qpc_eec_data.de, 1,
781                      qpc_eec_data.pm_state, 0x03 /* Always 0x03 for UD */,
782                      qpc_eec_data.st, ARBEL_ST_UD );
783         MLX_FILL_6 ( &qpctx, 4,
784                      qpc_eec_data.mtu, ARBEL_MTU_2048,
785                      qpc_eec_data.msg_max, 11 /* 2^11 = 2048 */,
786                      qpc_eec_data.log_rq_size, fls ( qp->recv.num_wqes - 1 ),
787                      qpc_eec_data.log_rq_stride,
788                      ( fls ( sizeof ( arbel_qp->recv.wqe[0] ) - 1 ) - 4 ),
789                      qpc_eec_data.log_sq_size, fls ( qp->send.num_wqes - 1 ),
790                      qpc_eec_data.log_sq_stride,
791                      ( fls ( sizeof ( arbel_qp->send.wqe[0] ) - 1 ) - 4 ) );
792         MLX_FILL_1 ( &qpctx, 5,
793                      qpc_eec_data.usr_page, arbel->limits.reserved_uars );
794         MLX_FILL_1 ( &qpctx, 10, qpc_eec_data.primary_address_path.port_number,
795                      ibdev->port );
796         MLX_FILL_1 ( &qpctx, 27, qpc_eec_data.pd, ARBEL_GLOBAL_PD );
797         MLX_FILL_1 ( &qpctx, 29, qpc_eec_data.wqe_lkey, arbel->reserved_lkey );
798         MLX_FILL_1 ( &qpctx, 30, qpc_eec_data.ssc, 1 );
799         MLX_FILL_1 ( &qpctx, 33, qpc_eec_data.cqn_snd, qp->send.cq->cqn );
800         MLX_FILL_1 ( &qpctx, 34, qpc_eec_data.snd_wqe_base_adr_l,
801                      ( virt_to_bus ( arbel_qp->send.wqe ) >> 6 ) );
802         MLX_FILL_1 ( &qpctx, 35, qpc_eec_data.snd_db_record_index,
803                      arbel_qp->send.doorbell_idx );
804         MLX_FILL_1 ( &qpctx, 38, qpc_eec_data.rsc, 1 );
805         MLX_FILL_1 ( &qpctx, 41, qpc_eec_data.cqn_rcv, qp->recv.cq->cqn );
806         MLX_FILL_1 ( &qpctx, 42, qpc_eec_data.rcv_wqe_base_adr_l,
807                      ( virt_to_bus ( arbel_qp->recv.wqe ) >> 6 ) );
808         MLX_FILL_1 ( &qpctx, 43, qpc_eec_data.rcv_db_record_index,
809                      arbel_qp->recv.doorbell_idx );
810         MLX_FILL_1 ( &qpctx, 44, qpc_eec_data.q_key, qp->qkey );
811         if ( ( rc = arbel_cmd_rst2init_qpee ( arbel, qp->qpn, &qpctx )) != 0 ){
812                 DBGC ( arbel, "Arbel %p RST2INIT_QPEE failed: %s\n",
813                        arbel, strerror ( rc ) );
814                 goto err_rst2init_qpee;
815         }
816         memset ( &qpctx, 0, sizeof ( qpctx ) );
817         MLX_FILL_2 ( &qpctx, 4,
818                      qpc_eec_data.mtu, ARBEL_MTU_2048,
819                      qpc_eec_data.msg_max, 11 /* 2^11 = 2048 */ );
820         if ( ( rc = arbel_cmd_init2rtr_qpee ( arbel, qp->qpn, &qpctx )) != 0 ){
821                 DBGC ( arbel, "Arbel %p INIT2RTR_QPEE failed: %s\n",
822                        arbel, strerror ( rc ) );
823                 goto err_init2rtr_qpee;
824         }
825         memset ( &qpctx, 0, sizeof ( qpctx ) );
826         if ( ( rc = arbel_cmd_rtr2rts_qpee ( arbel, qp->qpn, &qpctx ) ) != 0 ){
827                 DBGC ( arbel, "Arbel %p RTR2RTS_QPEE failed: %s\n",
828                        arbel, strerror ( rc ) );
829                 goto err_rtr2rts_qpee;
830         }
831
832         DBGC ( arbel, "Arbel %p QPN %#lx send ring at [%p,%p)\n",
833                arbel, qp->qpn, arbel_qp->send.wqe,
834                ( ( (void *) arbel_qp->send.wqe ) + arbel_qp->send.wqe_size ) );
835         DBGC ( arbel, "Arbel %p QPN %#lx receive ring at [%p,%p)\n",
836                arbel, qp->qpn, arbel_qp->recv.wqe,
837                ( ( (void *) arbel_qp->recv.wqe ) + arbel_qp->recv.wqe_size ) );
838         ib_qp_set_drvdata ( qp, arbel_qp );
839         return 0;
840
841  err_rtr2rts_qpee:
842  err_init2rtr_qpee:
843         arbel_cmd_2rst_qpee ( arbel, qp->qpn );
844  err_rst2init_qpee:
845         MLX_FILL_1 ( send_db_rec, 1, res, ARBEL_UAR_RES_NONE );
846         MLX_FILL_1 ( recv_db_rec, 1, res, ARBEL_UAR_RES_NONE );
847         free_dma ( arbel_qp->recv.wqe, arbel_qp->recv.wqe_size );
848  err_create_recv_wq:
849         free_dma ( arbel_qp->send.wqe, arbel_qp->send.wqe_size );
850  err_create_send_wq:
851         free ( arbel_qp );
852  err_arbel_qp:
853         arbel_free_qn_offset ( arbel->qp_inuse, qpn_offset );
854  err_qpn_offset:
855         return rc;
856 }
857
858 /**
859  * Modify queue pair
860  *
861  * @v ibdev             Infiniband device
862  * @v qp                Queue pair
863  * @v mod_list          Modification list
864  * @ret rc              Return status code
865  */
866 static int arbel_modify_qp ( struct ib_device *ibdev,
867                              struct ib_queue_pair *qp,
868                              unsigned long mod_list ) {
869         struct arbel *arbel = ib_get_drvdata ( ibdev );
870         struct arbelprm_qp_ee_state_transitions qpctx;
871         unsigned long optparammask = 0;
872         int rc;
873
874         /* Construct optparammask */
875         if ( mod_list & IB_MODIFY_QKEY )
876                 optparammask |= ARBEL_QPEE_OPT_PARAM_QKEY;
877
878         /* Issue RTS2RTS_QP */
879         memset ( &qpctx, 0, sizeof ( qpctx ) );
880         MLX_FILL_1 ( &qpctx, 0, opt_param_mask, optparammask );
881         MLX_FILL_1 ( &qpctx, 44, qpc_eec_data.q_key, qp->qkey );
882         if ( ( rc = arbel_cmd_rts2rts_qp ( arbel, qp->qpn, &qpctx ) ) != 0 ){
883                 DBGC ( arbel, "Arbel %p RTS2RTS_QP failed: %s\n",
884                        arbel, strerror ( rc ) );
885                 return rc;
886         }
887
888         return 0;
889 }
890
891 /**
892  * Destroy queue pair
893  *
894  * @v ibdev             Infiniband device
895  * @v qp                Queue pair
896  */
897 static void arbel_destroy_qp ( struct ib_device *ibdev,
898                                struct ib_queue_pair *qp ) {
899         struct arbel *arbel = ib_get_drvdata ( ibdev );
900         struct arbel_queue_pair *arbel_qp = ib_qp_get_drvdata ( qp );
901         struct arbelprm_qp_db_record *send_db_rec;
902         struct arbelprm_qp_db_record *recv_db_rec;
903         int qpn_offset;
904         int rc;
905
906         /* Take ownership back from hardware */
907         if ( ( rc = arbel_cmd_2rst_qpee ( arbel, qp->qpn ) ) != 0 ) {
908                 DBGC ( arbel, "Arbel %p FATAL 2RST_QPEE failed on QPN %#lx: "
909                        "%s\n", arbel, qp->qpn, strerror ( rc ) );
910                 /* Leak memory and return; at least we avoid corruption */
911                 return;
912         }
913
914         /* Clear doorbell records */
915         send_db_rec = &arbel->db_rec[arbel_qp->send.doorbell_idx].qp;
916         recv_db_rec = &arbel->db_rec[arbel_qp->recv.doorbell_idx].qp;
917         MLX_FILL_1 ( send_db_rec, 1, res, ARBEL_UAR_RES_NONE );
918         MLX_FILL_1 ( recv_db_rec, 1, res, ARBEL_UAR_RES_NONE );
919
920         /* Free memory */
921         free_dma ( arbel_qp->send.wqe, arbel_qp->send.wqe_size );
922         free_dma ( arbel_qp->recv.wqe, arbel_qp->recv.wqe_size );
923         free ( arbel_qp );
924
925         /* Mark queue number as free */
926         qpn_offset = ( qp->qpn - ARBEL_QPN_BASE - arbel->limits.reserved_qps );
927         arbel_free_qn_offset ( arbel->qp_inuse, qpn_offset );
928
929         ib_qp_set_drvdata ( qp, NULL );
930 }
931
932 /***************************************************************************
933  *
934  * Work request operations
935  *
936  ***************************************************************************
937  */
938
939 /**
940  * Ring doorbell register in UAR
941  *
942  * @v arbel             Arbel device
943  * @v db_reg            Doorbell register structure
944  * @v offset            Address of doorbell
945  */
946 static void arbel_ring_doorbell ( struct arbel *arbel,
947                                   union arbelprm_doorbell_register *db_reg,
948                                   unsigned int offset ) {
949
950         DBGC2 ( arbel, "Arbel %p ringing doorbell %08lx:%08lx at %lx\n",
951                 arbel, db_reg->dword[0], db_reg->dword[1],
952                 virt_to_phys ( arbel->uar + offset ) );
953
954         barrier();
955         writel ( db_reg->dword[0], ( arbel->uar + offset + 0 ) );
956         barrier();
957         writel ( db_reg->dword[1], ( arbel->uar + offset + 4 ) );
958 }
959
960 /** GID used for GID-less send work queue entries */
961 static const struct ib_gid arbel_no_gid = {
962         { { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2, 0, 0, 0 } }
963 };
964
965 /**
966  * Post send work queue entry
967  *
968  * @v ibdev             Infiniband device
969  * @v qp                Queue pair
970  * @v av                Address vector
971  * @v iobuf             I/O buffer
972  * @ret rc              Return status code
973  */
974 static int arbel_post_send ( struct ib_device *ibdev,
975                              struct ib_queue_pair *qp,
976                              struct ib_address_vector *av,
977                              struct io_buffer *iobuf ) {
978         struct arbel *arbel = ib_get_drvdata ( ibdev );
979         struct arbel_queue_pair *arbel_qp = ib_qp_get_drvdata ( qp );
980         struct ib_work_queue *wq = &qp->send;
981         struct arbel_send_work_queue *arbel_send_wq = &arbel_qp->send;
982         struct arbelprm_ud_send_wqe *prev_wqe;
983         struct arbelprm_ud_send_wqe *wqe;
984         struct arbelprm_qp_db_record *qp_db_rec;
985         union arbelprm_doorbell_register db_reg;
986         const struct ib_gid *gid;
987         unsigned int wqe_idx_mask;
988         size_t nds;
989
990         /* Allocate work queue entry */
991         wqe_idx_mask = ( wq->num_wqes - 1 );
992         if ( wq->iobufs[wq->next_idx & wqe_idx_mask] ) {
993                 DBGC ( arbel, "Arbel %p send queue full", arbel );
994                 return -ENOBUFS;
995         }
996         wq->iobufs[wq->next_idx & wqe_idx_mask] = iobuf;
997         prev_wqe = &arbel_send_wq->wqe[(wq->next_idx - 1) & wqe_idx_mask].ud;
998         wqe = &arbel_send_wq->wqe[wq->next_idx & wqe_idx_mask].ud;
999
1000         /* Construct work queue entry */
1001         MLX_FILL_1 ( &wqe->next, 1, always1, 1 );
1002         memset ( &wqe->ctrl, 0, sizeof ( wqe->ctrl ) );
1003         MLX_FILL_1 ( &wqe->ctrl, 0, always1, 1 );
1004         memset ( &wqe->ud, 0, sizeof ( wqe->ud ) );
1005         MLX_FILL_2 ( &wqe->ud, 0,
1006                      ud_address_vector.pd, ARBEL_GLOBAL_PD,
1007                      ud_address_vector.port_number, ibdev->port );
1008         MLX_FILL_2 ( &wqe->ud, 1,
1009                      ud_address_vector.rlid, av->dlid,
1010                      ud_address_vector.g, av->gid_present );
1011         MLX_FILL_2 ( &wqe->ud, 2,
1012                      ud_address_vector.max_stat_rate,
1013                          ( ( av->rate >= 3 ) ? 0 : 1 ),
1014                      ud_address_vector.msg, 3 );
1015         MLX_FILL_1 ( &wqe->ud, 3, ud_address_vector.sl, av->sl );
1016         gid = ( av->gid_present ? &av->gid : &arbel_no_gid );
1017         memcpy ( &wqe->ud.u.dwords[4], gid, sizeof ( *gid ) );
1018         MLX_FILL_1 ( &wqe->ud, 8, destination_qp, av->dest_qp );
1019         MLX_FILL_1 ( &wqe->ud, 9, q_key, av->qkey );
1020         MLX_FILL_1 ( &wqe->data[0], 0, byte_count, iob_len ( iobuf ) );
1021         MLX_FILL_1 ( &wqe->data[0], 1, l_key, arbel->reserved_lkey );
1022         MLX_FILL_1 ( &wqe->data[0], 3,
1023                      local_address_l, virt_to_bus ( iobuf->data ) );
1024
1025         /* Update previous work queue entry's "next" field */
1026         nds = ( ( offsetof ( typeof ( *wqe ), data ) +
1027                   sizeof ( wqe->data[0] ) ) >> 4 );
1028         MLX_SET ( &prev_wqe->next, nopcode, ARBEL_OPCODE_SEND );
1029         MLX_FILL_3 ( &prev_wqe->next, 1,
1030                      nds, nds,
1031                      f, 1,
1032                      always1, 1 );
1033
1034         /* Update doorbell record */
1035         barrier();
1036         qp_db_rec = &arbel->db_rec[arbel_send_wq->doorbell_idx].qp;
1037         MLX_FILL_1 ( qp_db_rec, 0,
1038                      counter, ( ( wq->next_idx + 1 ) & 0xffff ) );
1039
1040         /* Ring doorbell register */
1041         MLX_FILL_4 ( &db_reg.send, 0,
1042                      nopcode, ARBEL_OPCODE_SEND,
1043                      f, 1,
1044                      wqe_counter, ( wq->next_idx & 0xffff ),
1045                      wqe_cnt, 1 );
1046         MLX_FILL_2 ( &db_reg.send, 1,
1047                      nds, nds,
1048                      qpn, qp->qpn );
1049         arbel_ring_doorbell ( arbel, &db_reg, ARBEL_DB_POST_SND_OFFSET );
1050
1051         /* Update work queue's index */
1052         wq->next_idx++;
1053
1054         return 0;
1055 }
1056
1057 /**
1058  * Post receive work queue entry
1059  *
1060  * @v ibdev             Infiniband device
1061  * @v qp                Queue pair
1062  * @v iobuf             I/O buffer
1063  * @ret rc              Return status code
1064  */
1065 static int arbel_post_recv ( struct ib_device *ibdev,
1066                              struct ib_queue_pair *qp,
1067                              struct io_buffer *iobuf ) {
1068         struct arbel *arbel = ib_get_drvdata ( ibdev );
1069         struct arbel_queue_pair *arbel_qp = ib_qp_get_drvdata ( qp );
1070         struct ib_work_queue *wq = &qp->recv;
1071         struct arbel_recv_work_queue *arbel_recv_wq = &arbel_qp->recv;
1072         struct arbelprm_recv_wqe *wqe;
1073         union arbelprm_doorbell_record *db_rec;
1074         unsigned int wqe_idx_mask;
1075
1076         /* Allocate work queue entry */
1077         wqe_idx_mask = ( wq->num_wqes - 1 );
1078         if ( wq->iobufs[wq->next_idx & wqe_idx_mask] ) {
1079                 DBGC ( arbel, "Arbel %p receive queue full", arbel );
1080                 return -ENOBUFS;
1081         }
1082         wq->iobufs[wq->next_idx & wqe_idx_mask] = iobuf;
1083         wqe = &arbel_recv_wq->wqe[wq->next_idx & wqe_idx_mask].recv;
1084
1085         /* Construct work queue entry */
1086         MLX_FILL_1 ( &wqe->data[0], 0, byte_count, iob_tailroom ( iobuf ) );
1087         MLX_FILL_1 ( &wqe->data[0], 1, l_key, arbel->reserved_lkey );
1088         MLX_FILL_1 ( &wqe->data[0], 3,
1089                      local_address_l, virt_to_bus ( iobuf->data ) );
1090
1091         /* Update doorbell record */
1092         barrier();
1093         db_rec = &arbel->db_rec[arbel_recv_wq->doorbell_idx];
1094         MLX_FILL_1 ( &db_rec->qp, 0,
1095                      counter, ( ( wq->next_idx + 1 ) & 0xffff ) );      
1096
1097         /* Update work queue's index */
1098         wq->next_idx++;
1099
1100         return 0;       
1101 }
1102
1103 /**
1104  * Handle completion
1105  *
1106  * @v ibdev             Infiniband device
1107  * @v cq                Completion queue
1108  * @v cqe               Hardware completion queue entry
1109  * @ret rc              Return status code
1110  */
1111 static int arbel_complete ( struct ib_device *ibdev,
1112                             struct ib_completion_queue *cq,
1113                             union arbelprm_completion_entry *cqe ) {
1114         struct arbel *arbel = ib_get_drvdata ( ibdev );
1115         struct ib_completion completion;
1116         struct ib_work_queue *wq;
1117         struct ib_queue_pair *qp;
1118         struct arbel_queue_pair *arbel_qp;
1119         struct arbel_send_work_queue *arbel_send_wq;
1120         struct arbel_recv_work_queue *arbel_recv_wq;
1121         struct arbelprm_recv_wqe *recv_wqe;
1122         struct io_buffer *iobuf;
1123         unsigned int opcode;
1124         unsigned long qpn;
1125         int is_send;
1126         unsigned long wqe_adr;
1127         unsigned int wqe_idx;
1128         int rc = 0;
1129
1130         /* Parse completion */
1131         memset ( &completion, 0, sizeof ( completion ) );
1132         qpn = MLX_GET ( &cqe->normal, my_qpn );
1133         is_send = MLX_GET ( &cqe->normal, s );
1134         wqe_adr = ( MLX_GET ( &cqe->normal, wqe_adr ) << 6 );
1135         opcode = MLX_GET ( &cqe->normal, opcode );
1136         if ( opcode >= ARBEL_OPCODE_RECV_ERROR ) {
1137                 /* "s" field is not valid for error opcodes */
1138                 is_send = ( opcode == ARBEL_OPCODE_SEND_ERROR );
1139                 completion.syndrome = MLX_GET ( &cqe->error, syndrome );
1140                 DBGC ( arbel, "Arbel %p CPN %lx syndrome %x vendor %lx\n",
1141                        arbel, cq->cqn, completion.syndrome,
1142                        MLX_GET ( &cqe->error, vendor_code ) );
1143                 rc = -EIO;
1144                 /* Don't return immediately; propagate error to completer */
1145         }
1146
1147         /* Identify work queue */
1148         wq = ib_find_wq ( cq, qpn, is_send );
1149         if ( ! wq ) {
1150                 DBGC ( arbel, "Arbel %p CQN %lx unknown %s QPN %lx\n",
1151                        arbel, cq->cqn, ( is_send ? "send" : "recv" ), qpn );
1152                 return -EIO;
1153         }
1154         qp = wq->qp;
1155         arbel_qp = ib_qp_get_drvdata ( qp );
1156         arbel_send_wq = &arbel_qp->send;
1157         arbel_recv_wq = &arbel_qp->recv;
1158
1159         /* Identify work queue entry index */
1160         if ( is_send ) {
1161                 wqe_idx = ( ( wqe_adr - virt_to_bus ( arbel_send_wq->wqe ) ) /
1162                             sizeof ( arbel_send_wq->wqe[0] ) );
1163                 assert ( wqe_idx < qp->send.num_wqes );
1164         } else {
1165                 wqe_idx = ( ( wqe_adr - virt_to_bus ( arbel_recv_wq->wqe ) ) /
1166                             sizeof ( arbel_recv_wq->wqe[0] ) );
1167                 assert ( wqe_idx < qp->recv.num_wqes );
1168         }
1169
1170         /* Identify I/O buffer */
1171         iobuf = wq->iobufs[wqe_idx];
1172         if ( ! iobuf ) {
1173                 DBGC ( arbel, "Arbel %p CQN %lx QPN %lx empty WQE %x\n",
1174                        arbel, cq->cqn, qpn, wqe_idx );
1175                 return -EIO;
1176         }
1177         wq->iobufs[wqe_idx] = NULL;
1178
1179         /* Fill in length for received packets */
1180         if ( ! is_send ) {
1181                 completion.len = MLX_GET ( &cqe->normal, byte_cnt );
1182                 recv_wqe = &arbel_recv_wq->wqe[wqe_idx].recv;
1183                 assert ( MLX_GET ( &recv_wqe->data[0], local_address_l ) ==
1184                          virt_to_bus ( iobuf->data ) );
1185                 assert ( MLX_GET ( &recv_wqe->data[0], byte_count ) ==
1186                          iob_tailroom ( iobuf ) );
1187                 MLX_FILL_1 ( &recv_wqe->data[0], 0, byte_count, 0 );
1188                 MLX_FILL_1 ( &recv_wqe->data[0], 1,
1189                              l_key, ARBEL_INVALID_LKEY );
1190                 if ( completion.len > iob_tailroom ( iobuf ) ) {
1191                         DBGC ( arbel, "Arbel %p CQN %lx QPN %lx IDX %x "
1192                                "overlength received packet length %zd\n",
1193                                arbel, cq->cqn, qpn, wqe_idx, completion.len );
1194                         return -EIO;
1195                 }
1196         }
1197
1198         /* Pass off to caller's completion handler */
1199         if ( is_send ) {
1200                 ib_complete_send ( ibdev, qp, &completion, iobuf );
1201         } else {
1202                 ib_complete_recv ( ibdev, qp, &completion, iobuf );
1203         }
1204
1205         return rc;
1206 }                            
1207
1208 /**
1209  * Poll completion queue
1210  *
1211  * @v ibdev             Infiniband device
1212  * @v cq                Completion queue
1213  */
1214 static void arbel_poll_cq ( struct ib_device *ibdev,
1215                             struct ib_completion_queue *cq ) {
1216         struct arbel *arbel = ib_get_drvdata ( ibdev );
1217         struct arbel_completion_queue *arbel_cq = ib_cq_get_drvdata ( cq );
1218         struct arbelprm_cq_ci_db_record *ci_db_rec;
1219         union arbelprm_completion_entry *cqe;
1220         unsigned int cqe_idx_mask;
1221         int rc;
1222
1223         while ( 1 ) {
1224                 /* Look for completion entry */
1225                 cqe_idx_mask = ( cq->num_cqes - 1 );
1226                 cqe = &arbel_cq->cqe[cq->next_idx & cqe_idx_mask];
1227                 if ( MLX_GET ( &cqe->normal, owner ) != 0 ) {
1228                         /* Entry still owned by hardware; end of poll */
1229                         break;
1230                 }
1231
1232                 /* Handle completion */
1233                 if ( ( rc = arbel_complete ( ibdev, cq, cqe ) ) != 0 ) {
1234                         DBGC ( arbel, "Arbel %p failed to complete: %s\n",
1235                                arbel, strerror ( rc ) );
1236                         DBGC_HD ( arbel, cqe, sizeof ( *cqe ) );
1237                 }
1238
1239                 /* Return ownership to hardware */
1240                 MLX_FILL_1 ( &cqe->normal, 7, owner, 1 );
1241                 barrier();
1242                 /* Update completion queue's index */
1243                 cq->next_idx++;
1244                 /* Update doorbell record */
1245                 ci_db_rec = &arbel->db_rec[arbel_cq->ci_doorbell_idx].cq_ci;
1246                 MLX_FILL_1 ( ci_db_rec, 0,
1247                              counter, ( cq->next_idx & 0xffffffffUL ) );
1248         }
1249 }
1250
1251 /***************************************************************************
1252  *
1253  * Event queues
1254  *
1255  ***************************************************************************
1256  */
1257
1258 /**
1259  * Create event queue
1260  *
1261  * @v arbel             Arbel device
1262  * @ret rc              Return status code
1263  */
1264 static int arbel_create_eq ( struct arbel *arbel ) {
1265         struct arbel_event_queue *arbel_eq = &arbel->eq;
1266         struct arbelprm_eqc eqctx;
1267         struct arbelprm_event_mask mask;
1268         unsigned int i;
1269         int rc;
1270
1271         /* Select event queue number */
1272         arbel_eq->eqn = arbel->limits.reserved_eqs;
1273
1274         /* Calculate doorbell address */
1275         arbel_eq->doorbell = ( arbel->eq_ci_doorbells +
1276                                ARBEL_DB_EQ_OFFSET ( arbel_eq->eqn ) );
1277
1278         /* Allocate event queue itself */
1279         arbel_eq->eqe_size =
1280                 ( ARBEL_NUM_EQES * sizeof ( arbel_eq->eqe[0] ) );
1281         arbel_eq->eqe = malloc_dma ( arbel_eq->eqe_size,
1282                                      sizeof ( arbel_eq->eqe[0] ) );
1283         if ( ! arbel_eq->eqe ) {
1284                 rc = -ENOMEM;
1285                 goto err_eqe;
1286         }
1287         memset ( arbel_eq->eqe, 0, arbel_eq->eqe_size );
1288         for ( i = 0 ; i < ARBEL_NUM_EQES ; i++ ) {
1289                 MLX_FILL_1 ( &arbel_eq->eqe[i].generic, 7, owner, 1 );
1290         }
1291         barrier();
1292
1293         /* Hand queue over to hardware */
1294         memset ( &eqctx, 0, sizeof ( eqctx ) );
1295         MLX_FILL_1 ( &eqctx, 0, st, 0xa /* "Fired" */ );
1296         MLX_FILL_1 ( &eqctx, 2,
1297                      start_address_l, virt_to_phys ( arbel_eq->eqe ) );
1298         MLX_FILL_1 ( &eqctx, 3, log_eq_size, fls ( ARBEL_NUM_EQES - 1 ) );
1299         MLX_FILL_1 ( &eqctx, 6, pd, ARBEL_GLOBAL_PD );
1300         MLX_FILL_1 ( &eqctx, 7, lkey, arbel->reserved_lkey );
1301         if ( ( rc = arbel_cmd_sw2hw_eq ( arbel, arbel_eq->eqn,
1302                                          &eqctx ) ) != 0 ) {
1303                 DBGC ( arbel, "Arbel %p SW2HW_EQ failed: %s\n",
1304                        arbel, strerror ( rc ) );
1305                 goto err_sw2hw_eq;
1306         }
1307
1308         /* Map events to this event queue */
1309         memset ( &mask, 0, sizeof ( mask ) );
1310         MLX_FILL_1 ( &mask, 1, port_state_change, 1 );
1311         if ( ( rc = arbel_cmd_map_eq ( arbel,
1312                                        ( ARBEL_MAP_EQ | arbel_eq->eqn ),
1313                                        &mask ) ) != 0 ) {
1314                 DBGC ( arbel, "Arbel %p MAP_EQ failed: %s\n",
1315                        arbel, strerror ( rc )  );
1316                 goto err_map_eq;
1317         }
1318
1319         DBGC ( arbel, "Arbel %p EQN %#lx ring at [%p,%p])\n",
1320                arbel, arbel_eq->eqn, arbel_eq->eqe,
1321                ( ( ( void * ) arbel_eq->eqe ) + arbel_eq->eqe_size ) );
1322         return 0;
1323
1324  err_map_eq:
1325         arbel_cmd_hw2sw_eq ( arbel, arbel_eq->eqn, &eqctx );
1326  err_sw2hw_eq:
1327         free_dma ( arbel_eq->eqe, arbel_eq->eqe_size );
1328  err_eqe:
1329         memset ( arbel_eq, 0, sizeof ( *arbel_eq ) );
1330         return rc;
1331 }
1332
1333 /**
1334  * Destroy event queue
1335  *
1336  * @v arbel             Arbel device
1337  */
1338 static void arbel_destroy_eq ( struct arbel *arbel ) {
1339         struct arbel_event_queue *arbel_eq = &arbel->eq;
1340         struct arbelprm_eqc eqctx;
1341         struct arbelprm_event_mask mask;
1342         int rc;
1343
1344         /* Unmap events from event queue */
1345         memset ( &mask, 0, sizeof ( mask ) );
1346         MLX_FILL_1 ( &mask, 1, port_state_change, 1 );
1347         if ( ( rc = arbel_cmd_map_eq ( arbel,
1348                                        ( ARBEL_UNMAP_EQ | arbel_eq->eqn ),
1349                                        &mask ) ) != 0 ) {
1350                 DBGC ( arbel, "Arbel %p FATAL MAP_EQ failed to unmap: %s\n",
1351                        arbel, strerror ( rc ) );
1352                 /* Continue; HCA may die but system should survive */
1353         }
1354
1355         /* Take ownership back from hardware */
1356         if ( ( rc = arbel_cmd_hw2sw_eq ( arbel, arbel_eq->eqn,
1357                                          &eqctx ) ) != 0 ) {
1358                 DBGC ( arbel, "Arbel %p FATAL HW2SW_EQ failed: %s\n",
1359                        arbel, strerror ( rc ) );
1360                 /* Leak memory and return; at least we avoid corruption */
1361                 return;
1362         }
1363
1364         /* Free memory */
1365         free_dma ( arbel_eq->eqe, arbel_eq->eqe_size );
1366         memset ( arbel_eq, 0, sizeof ( *arbel_eq ) );
1367 }
1368
1369 /**
1370  * Handle port state event
1371  *
1372  * @v arbel             Arbel device
1373  * @v eqe               Port state change event queue entry
1374  */
1375 static void arbel_event_port_state_change ( struct arbel *arbel,
1376                                             union arbelprm_event_entry *eqe){
1377         unsigned int port;
1378         int link_up;
1379
1380         /* Get port and link status */
1381         port = ( MLX_GET ( &eqe->port_state_change, data.p ) - 1 );
1382         link_up = ( MLX_GET ( &eqe->generic, event_sub_type ) & 0x04 );
1383         DBGC ( arbel, "Arbel %p port %d link %s\n", arbel, ( port + 1 ),
1384                ( link_up ? "up" : "down" ) );
1385
1386         /* Sanity check */
1387         if ( port >= ARBEL_NUM_PORTS ) {
1388                 DBGC ( arbel, "Arbel %p port %d does not exist!\n",
1389                        arbel, ( port + 1 ) );
1390                 return;
1391         }
1392
1393         /* Notify Infiniband core of link state change */
1394         ib_link_state_changed ( arbel->ibdev[port] );
1395 }
1396
1397 /**
1398  * Poll event queue
1399  *
1400  * @v ibdev             Infiniband device
1401  */
1402 static void arbel_poll_eq ( struct ib_device *ibdev ) {
1403         struct arbel *arbel = ib_get_drvdata ( ibdev );
1404         struct arbel_event_queue *arbel_eq = &arbel->eq;
1405         union arbelprm_event_entry *eqe;
1406         union arbelprm_eq_doorbell_register db_reg;
1407         unsigned int eqe_idx_mask;
1408         unsigned int event_type;
1409
1410         while ( 1 ) {
1411                 /* Look for event entry */
1412                 eqe_idx_mask = ( ARBEL_NUM_EQES - 1 );
1413                 eqe = &arbel_eq->eqe[arbel_eq->next_idx & eqe_idx_mask];
1414                 if ( MLX_GET ( &eqe->generic, owner ) != 0 ) {
1415                         /* Entry still owned by hardware; end of poll */
1416                         break;
1417                 }
1418                 DBGCP ( arbel, "Arbel %p event:\n", arbel );
1419                 DBGCP_HD ( arbel, eqe, sizeof ( *eqe ) );
1420
1421                 /* Handle event */
1422                 event_type = MLX_GET ( &eqe->generic, event_type );
1423                 switch ( event_type ) {
1424                 case ARBEL_EV_PORT_STATE_CHANGE:
1425                         arbel_event_port_state_change ( arbel, eqe );
1426                         break;
1427                 default:
1428                         DBGC ( arbel, "Arbel %p unrecognised event type "
1429                                "%#x:\n", arbel, event_type );
1430                         DBGC_HD ( arbel, eqe, sizeof ( *eqe ) );
1431                         break;
1432                 }
1433
1434                 /* Return ownership to hardware */
1435                 MLX_FILL_1 ( &eqe->generic, 7, owner, 1 );
1436                 barrier();
1437
1438                 /* Update event queue's index */
1439                 arbel_eq->next_idx++;
1440
1441                 /* Ring doorbell */
1442                 MLX_FILL_1 ( &db_reg.ci, 0, ci, arbel_eq->next_idx );
1443                 DBGCP ( arbel, "Ringing doorbell %08lx with %08lx\n",
1444                         virt_to_phys ( arbel_eq->doorbell ),
1445                         db_reg.dword[0] );
1446                 writel ( db_reg.dword[0], arbel_eq->doorbell );
1447         }
1448 }
1449
1450 /***************************************************************************
1451  *
1452  * Infiniband link-layer operations
1453  *
1454  ***************************************************************************
1455  */
1456
1457 /**
1458  * Initialise Infiniband link
1459  *
1460  * @v ibdev             Infiniband device
1461  * @ret rc              Return status code
1462  */
1463 static int arbel_open ( struct ib_device *ibdev ) {
1464         struct arbel *arbel = ib_get_drvdata ( ibdev );
1465         struct arbelprm_init_ib init_ib;
1466         int rc;
1467
1468         memset ( &init_ib, 0, sizeof ( init_ib ) );
1469         MLX_FILL_3 ( &init_ib, 0,
1470                      mtu_cap, ARBEL_MTU_2048,
1471                      port_width_cap, 3,
1472                      vl_cap, 1 );
1473         MLX_FILL_1 ( &init_ib, 1, max_gid, 1 );
1474         MLX_FILL_1 ( &init_ib, 2, max_pkey, 64 );
1475         if ( ( rc = arbel_cmd_init_ib ( arbel, ibdev->port,
1476                                         &init_ib ) ) != 0 ) {
1477                 DBGC ( arbel, "Arbel %p could not intialise IB: %s\n",
1478                        arbel, strerror ( rc ) );
1479                 return rc;
1480         }
1481
1482         return 0;
1483 }
1484
1485 /**
1486  * Close Infiniband link
1487  *
1488  * @v ibdev             Infiniband device
1489  */
1490 static void arbel_close ( struct ib_device *ibdev ) {
1491         struct arbel *arbel = ib_get_drvdata ( ibdev );
1492         int rc;
1493
1494         if ( ( rc = arbel_cmd_close_ib ( arbel, ibdev->port ) ) != 0 ) {
1495                 DBGC ( arbel, "Arbel %p could not close IB: %s\n",
1496                        arbel, strerror ( rc ) );
1497                 /* Nothing we can do about this */
1498         }
1499 }
1500
1501 /***************************************************************************
1502  *
1503  * Multicast group operations
1504  *
1505  ***************************************************************************
1506  */
1507
1508 /**
1509  * Attach to multicast group
1510  *
1511  * @v ibdev             Infiniband device
1512  * @v qp                Queue pair
1513  * @v gid               Multicast GID
1514  * @ret rc              Return status code
1515  */
1516 static int arbel_mcast_attach ( struct ib_device *ibdev,
1517                                 struct ib_queue_pair *qp,
1518                                 struct ib_gid *gid ) {
1519         struct arbel *arbel = ib_get_drvdata ( ibdev );
1520         struct arbelprm_mgm_hash hash;
1521         struct arbelprm_mgm_entry mgm;
1522         unsigned int index;
1523         int rc;
1524
1525         /* Generate hash table index */
1526         if ( ( rc = arbel_cmd_mgid_hash ( arbel, gid, &hash ) ) != 0 ) {
1527                 DBGC ( arbel, "Arbel %p could not hash GID: %s\n",
1528                        arbel, strerror ( rc ) );
1529                 return rc;
1530         }
1531         index = MLX_GET ( &hash, hash );
1532
1533         /* Check for existing hash table entry */
1534         if ( ( rc = arbel_cmd_read_mgm ( arbel, index, &mgm ) ) != 0 ) {
1535                 DBGC ( arbel, "Arbel %p could not read MGM %#x: %s\n",
1536                        arbel, index, strerror ( rc ) );
1537                 return rc;
1538         }
1539         if ( MLX_GET ( &mgm, mgmqp_0.qi ) != 0 ) {
1540                 /* FIXME: this implementation allows only a single QP
1541                  * per multicast group, and doesn't handle hash
1542                  * collisions.  Sufficient for IPoIB but may need to
1543                  * be extended in future.
1544                  */
1545                 DBGC ( arbel, "Arbel %p MGID index %#x already in use\n",
1546                        arbel, index );
1547                 return -EBUSY;
1548         }
1549
1550         /* Update hash table entry */
1551         MLX_FILL_2 ( &mgm, 8,
1552                      mgmqp_0.qpn_i, qp->qpn,
1553                      mgmqp_0.qi, 1 );
1554         memcpy ( &mgm.u.dwords[4], gid, sizeof ( *gid ) );
1555         if ( ( rc = arbel_cmd_write_mgm ( arbel, index, &mgm ) ) != 0 ) {
1556                 DBGC ( arbel, "Arbel %p could not write MGM %#x: %s\n",
1557                        arbel, index, strerror ( rc ) );
1558                 return rc;
1559         }
1560
1561         return 0;
1562 }
1563
1564 /**
1565  * Detach from multicast group
1566  *
1567  * @v ibdev             Infiniband device
1568  * @v qp                Queue pair
1569  * @v gid               Multicast GID
1570  */
1571 static void arbel_mcast_detach ( struct ib_device *ibdev,
1572                                  struct ib_queue_pair *qp __unused,
1573                                  struct ib_gid *gid ) {
1574         struct arbel *arbel = ib_get_drvdata ( ibdev );
1575         struct arbelprm_mgm_hash hash;
1576         struct arbelprm_mgm_entry mgm;
1577         unsigned int index;
1578         int rc;
1579
1580         /* Generate hash table index */
1581         if ( ( rc = arbel_cmd_mgid_hash ( arbel, gid, &hash ) ) != 0 ) {
1582                 DBGC ( arbel, "Arbel %p could not hash GID: %s\n",
1583                        arbel, strerror ( rc ) );
1584                 return;
1585         }
1586         index = MLX_GET ( &hash, hash );
1587
1588         /* Clear hash table entry */
1589         memset ( &mgm, 0, sizeof ( mgm ) );
1590         if ( ( rc = arbel_cmd_write_mgm ( arbel, index, &mgm ) ) != 0 ) {
1591                 DBGC ( arbel, "Arbel %p could not write MGM %#x: %s\n",
1592                        arbel, index, strerror ( rc ) );
1593                 return;
1594         }
1595 }
1596
1597 /***************************************************************************
1598  *
1599  * MAD operations
1600  *
1601  ***************************************************************************
1602  */
1603
1604 /**
1605  * Issue management datagram
1606  *
1607  * @v ibdev             Infiniband device
1608  * @v mad               Management datagram
1609  * @v len               Length of management datagram
1610  * @ret rc              Return status code
1611  */
1612 static int arbel_mad ( struct ib_device *ibdev, struct ib_mad_hdr *mad,
1613                        size_t len ) {
1614         struct arbel *arbel = ib_get_drvdata ( ibdev );
1615         union arbelprm_mad mad_ifc;
1616         int rc;
1617
1618         /* Copy in request packet */
1619         memset ( &mad_ifc, 0, sizeof ( mad_ifc ) );
1620         assert ( len <= sizeof ( mad_ifc.mad ) );
1621         memcpy ( &mad_ifc.mad, mad, len );
1622
1623         /* Issue MAD */
1624         if ( ( rc = arbel_cmd_mad_ifc ( arbel, ibdev->port,
1625                                         &mad_ifc ) ) != 0 ) {
1626                 DBGC ( arbel, "Arbel %p could not issue MAD IFC: %s\n",
1627                        arbel, strerror ( rc ) );
1628                 return rc;
1629         }
1630
1631         /* Copy out reply packet */
1632         memcpy ( mad, &mad_ifc.mad, len );
1633
1634         if ( mad->status != 0 ) {
1635                 DBGC ( arbel, "Arbel %p MAD IFC status %04x\n",
1636                        arbel, ntohs ( mad->status ) );
1637                 return -EIO;
1638         }
1639         return 0;
1640 }
1641
1642 /** Arbel Infiniband operations */
1643 static struct ib_device_operations arbel_ib_operations = {
1644         .create_cq      = arbel_create_cq,
1645         .destroy_cq     = arbel_destroy_cq,
1646         .create_qp      = arbel_create_qp,
1647         .modify_qp      = arbel_modify_qp,
1648         .destroy_qp     = arbel_destroy_qp,
1649         .post_send      = arbel_post_send,
1650         .post_recv      = arbel_post_recv,
1651         .poll_cq        = arbel_poll_cq,
1652         .poll_eq        = arbel_poll_eq,
1653         .open           = arbel_open,
1654         .close          = arbel_close,
1655         .mcast_attach   = arbel_mcast_attach,
1656         .mcast_detach   = arbel_mcast_detach,
1657         .mad            = arbel_mad,
1658 };
1659
1660 /***************************************************************************
1661  *
1662  * Firmware control
1663  *
1664  ***************************************************************************
1665  */
1666
1667 /**
1668  * Start firmware running
1669  *
1670  * @v arbel             Arbel device
1671  * @ret rc              Return status code
1672  */
1673 static int arbel_start_firmware ( struct arbel *arbel ) {
1674         struct arbelprm_query_fw fw;
1675         struct arbelprm_access_lam lam;
1676         struct arbelprm_virtual_physical_mapping map_fa;
1677         unsigned int fw_pages;
1678         unsigned int log2_fw_pages;
1679         size_t fw_size;
1680         physaddr_t fw_base;
1681         uint64_t eq_set_ci_base_addr;
1682         int rc;
1683
1684         /* Get firmware parameters */
1685         if ( ( rc = arbel_cmd_query_fw ( arbel, &fw ) ) != 0 ) {
1686                 DBGC ( arbel, "Arbel %p could not query firmware: %s\n",
1687                        arbel, strerror ( rc ) );
1688                 goto err_query_fw;
1689         }
1690         DBGC ( arbel, "Arbel %p firmware version %ld.%ld.%ld\n", arbel,
1691                MLX_GET ( &fw, fw_rev_major ), MLX_GET ( &fw, fw_rev_minor ),
1692                MLX_GET ( &fw, fw_rev_subminor ) );
1693         fw_pages = MLX_GET ( &fw, fw_pages );
1694         log2_fw_pages = fls ( fw_pages - 1 );
1695         fw_pages = ( 1 << log2_fw_pages );
1696         DBGC ( arbel, "Arbel %p requires %d kB for firmware\n",
1697                arbel, ( fw_pages * 4 ) );
1698         eq_set_ci_base_addr =
1699                 ( ( (uint64_t) MLX_GET ( &fw, eq_set_ci_base_addr_h ) << 32 ) |
1700                   ( (uint64_t) MLX_GET ( &fw, eq_set_ci_base_addr_l ) ) );
1701         arbel->eq_ci_doorbells = ioremap ( eq_set_ci_base_addr, 0x200 );
1702
1703         /* Enable locally-attached memory.  Ignore failure; there may
1704          * be no attached memory.
1705          */
1706         arbel_cmd_enable_lam ( arbel, &lam );
1707
1708         /* Allocate firmware pages and map firmware area */
1709         fw_size = ( fw_pages * 4096 );
1710         arbel->firmware_area = umalloc ( fw_size * 2 );
1711         if ( ! arbel->firmware_area ) {
1712                 rc = -ENOMEM;
1713                 goto err_alloc_fa;
1714         }
1715         fw_base = ( user_to_phys ( arbel->firmware_area, fw_size ) &
1716                     ~( fw_size - 1 ) );
1717         DBGC ( arbel, "Arbel %p firmware area at physical [%lx,%lx)\n",
1718                arbel, fw_base, ( fw_base + fw_size ) );
1719         memset ( &map_fa, 0, sizeof ( map_fa ) );
1720         MLX_FILL_2 ( &map_fa, 3,
1721                      log2size, log2_fw_pages,
1722                      pa_l, ( fw_base >> 12 ) );
1723         if ( ( rc = arbel_cmd_map_fa ( arbel, &map_fa ) ) != 0 ) {
1724                 DBGC ( arbel, "Arbel %p could not map firmware: %s\n",
1725                        arbel, strerror ( rc ) );
1726                 goto err_map_fa;
1727         }
1728
1729         /* Start firmware */
1730         if ( ( rc = arbel_cmd_run_fw ( arbel ) ) != 0 ) {
1731                 DBGC ( arbel, "Arbel %p could not run firmware: %s\n",
1732                        arbel, strerror ( rc ) );
1733                 goto err_run_fw;
1734         }
1735
1736         DBGC ( arbel, "Arbel %p firmware started\n", arbel );
1737         return 0;
1738
1739  err_run_fw:
1740         arbel_cmd_unmap_fa ( arbel );
1741  err_map_fa:
1742         ufree ( arbel->firmware_area );
1743         arbel->firmware_area = UNULL;
1744  err_alloc_fa:
1745  err_query_fw:
1746         return rc;
1747 }
1748
1749 /**
1750  * Stop firmware running
1751  *
1752  * @v arbel             Arbel device
1753  */
1754 static void arbel_stop_firmware ( struct arbel *arbel ) {
1755         int rc;
1756
1757         if ( ( rc = arbel_cmd_unmap_fa ( arbel ) ) != 0 ) {
1758                 DBGC ( arbel, "Arbel %p FATAL could not stop firmware: %s\n",
1759                        arbel, strerror ( rc ) );
1760                 /* Leak memory and return; at least we avoid corruption */
1761                 return;
1762         }
1763         ufree ( arbel->firmware_area );
1764         arbel->firmware_area = UNULL;
1765 }
1766
1767 /***************************************************************************
1768  *
1769  * Infinihost Context Memory management
1770  *
1771  ***************************************************************************
1772  */
1773
1774 /**
1775  * Get device limits
1776  *
1777  * @v arbel             Arbel device
1778  * @ret rc              Return status code
1779  */
1780 static int arbel_get_limits ( struct arbel *arbel ) {
1781         struct arbelprm_query_dev_lim dev_lim;
1782         int rc;
1783
1784         if ( ( rc = arbel_cmd_query_dev_lim ( arbel, &dev_lim ) ) != 0 ) {
1785                 DBGC ( arbel, "Arbel %p could not get device limits: %s\n",
1786                        arbel, strerror ( rc ) );
1787                 return rc;
1788         }
1789
1790         arbel->limits.reserved_qps =
1791                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_qps ) );
1792         arbel->limits.qpc_entry_size = MLX_GET ( &dev_lim, qpc_entry_sz );
1793         arbel->limits.eqpc_entry_size = MLX_GET ( &dev_lim, eqpc_entry_sz );
1794         arbel->limits.reserved_srqs =
1795                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_srqs ) );
1796         arbel->limits.srqc_entry_size = MLX_GET ( &dev_lim, srq_entry_sz );
1797         arbel->limits.reserved_ees =
1798                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_ees ) );
1799         arbel->limits.eec_entry_size = MLX_GET ( &dev_lim, eec_entry_sz );
1800         arbel->limits.eeec_entry_size = MLX_GET ( &dev_lim, eeec_entry_sz );
1801         arbel->limits.reserved_cqs =
1802                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_cqs ) );
1803         arbel->limits.cqc_entry_size = MLX_GET ( &dev_lim, cqc_entry_sz );
1804         arbel->limits.reserved_eqs = MLX_GET ( &dev_lim, num_rsvd_eqs );
1805         arbel->limits.reserved_mtts =
1806                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_mtts ) );
1807         arbel->limits.mtt_entry_size = MLX_GET ( &dev_lim, mtt_entry_sz );
1808         arbel->limits.reserved_mrws =
1809                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_mrws ) );
1810         arbel->limits.mpt_entry_size = MLX_GET ( &dev_lim, mpt_entry_sz );
1811         arbel->limits.reserved_rdbs =
1812                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_rdbs ) );
1813         arbel->limits.eqc_entry_size = MLX_GET ( &dev_lim, eqc_entry_sz );
1814         arbel->limits.reserved_uars = MLX_GET ( &dev_lim, num_rsvd_uars );
1815
1816         return 0;
1817 }
1818
1819 /**
1820  * Get ICM usage
1821  *
1822  * @v log_num_entries   Log2 of the number of entries
1823  * @v entry_size        Entry size
1824  * @ret usage           Usage size in ICM
1825  */
1826 static size_t icm_usage ( unsigned int log_num_entries, size_t entry_size ) {
1827         size_t usage;
1828
1829         usage = ( ( 1 << log_num_entries ) * entry_size );
1830         usage = ( ( usage + 4095 ) & ~4095 );
1831         return usage;
1832 }
1833
1834 /**
1835  * Allocate ICM
1836  *
1837  * @v arbel             Arbel device
1838  * @v init_hca          INIT_HCA structure to fill in
1839  * @ret rc              Return status code
1840  */
1841 static int arbel_alloc_icm ( struct arbel *arbel,
1842                              struct arbelprm_init_hca *init_hca ) {
1843         struct arbelprm_scalar_parameter icm_size;
1844         struct arbelprm_scalar_parameter icm_aux_size;
1845         struct arbelprm_virtual_physical_mapping map_icm_aux;
1846         struct arbelprm_virtual_physical_mapping map_icm;
1847         union arbelprm_doorbell_record *db_rec;
1848         size_t icm_offset = 0;
1849         unsigned int log_num_qps, log_num_srqs, log_num_ees, log_num_cqs;
1850         unsigned int log_num_mtts, log_num_mpts, log_num_rdbs, log_num_eqs;
1851         int rc;
1852
1853         icm_offset = ( ( arbel->limits.reserved_uars + 1 ) << 12 );
1854
1855         /* Queue pair contexts */
1856         log_num_qps = fls ( arbel->limits.reserved_qps + ARBEL_MAX_QPS - 1 );
1857         MLX_FILL_2 ( init_hca, 13,
1858                      qpc_eec_cqc_eqc_rdb_parameters.qpc_base_addr_l,
1859                      ( icm_offset >> 7 ),
1860                      qpc_eec_cqc_eqc_rdb_parameters.log_num_of_qp,
1861                      log_num_qps );
1862         DBGC ( arbel, "Arbel %p ICM QPC base = %zx\n", arbel, icm_offset );
1863         icm_offset += icm_usage ( log_num_qps, arbel->limits.qpc_entry_size );
1864
1865         /* Extended queue pair contexts */
1866         MLX_FILL_1 ( init_hca, 25,
1867                      qpc_eec_cqc_eqc_rdb_parameters.eqpc_base_addr_l,
1868                      icm_offset );
1869         DBGC ( arbel, "Arbel %p ICM EQPC base = %zx\n", arbel, icm_offset );
1870         //      icm_offset += icm_usage ( log_num_qps, arbel->limits.eqpc_entry_size );
1871         icm_offset += icm_usage ( log_num_qps, arbel->limits.qpc_entry_size );  
1872
1873         /* Shared receive queue contexts */
1874         log_num_srqs = fls ( arbel->limits.reserved_srqs - 1 );
1875         MLX_FILL_2 ( init_hca, 19,
1876                      qpc_eec_cqc_eqc_rdb_parameters.srqc_base_addr_l,
1877                      ( icm_offset >> 5 ),
1878                      qpc_eec_cqc_eqc_rdb_parameters.log_num_of_srq,
1879                      log_num_srqs );
1880         DBGC ( arbel, "Arbel %p ICM SRQC base = %zx\n", arbel, icm_offset );
1881         icm_offset += icm_usage ( log_num_srqs, arbel->limits.srqc_entry_size );
1882
1883         /* End-to-end contexts */
1884         log_num_ees = fls ( arbel->limits.reserved_ees - 1 );
1885         MLX_FILL_2 ( init_hca, 17,
1886                      qpc_eec_cqc_eqc_rdb_parameters.eec_base_addr_l,
1887                      ( icm_offset >> 7 ),
1888                      qpc_eec_cqc_eqc_rdb_parameters.log_num_of_ee,
1889                      log_num_ees );
1890         DBGC ( arbel, "Arbel %p ICM EEC base = %zx\n", arbel, icm_offset );
1891         icm_offset += icm_usage ( log_num_ees, arbel->limits.eec_entry_size );
1892
1893         /* Extended end-to-end contexts */
1894         MLX_FILL_1 ( init_hca, 29,
1895                      qpc_eec_cqc_eqc_rdb_parameters.eeec_base_addr_l,
1896                      icm_offset );
1897         DBGC ( arbel, "Arbel %p ICM EEEC base = %zx\n", arbel, icm_offset );
1898         icm_offset += icm_usage ( log_num_ees, arbel->limits.eeec_entry_size );
1899
1900         /* Completion queue contexts */
1901         log_num_cqs = fls ( arbel->limits.reserved_cqs + ARBEL_MAX_CQS - 1 );
1902         MLX_FILL_2 ( init_hca, 21,
1903                      qpc_eec_cqc_eqc_rdb_parameters.cqc_base_addr_l,
1904                      ( icm_offset >> 6 ),
1905                      qpc_eec_cqc_eqc_rdb_parameters.log_num_of_cq,
1906                      log_num_cqs );
1907         DBGC ( arbel, "Arbel %p ICM CQC base = %zx\n", arbel, icm_offset );
1908         icm_offset += icm_usage ( log_num_cqs, arbel->limits.cqc_entry_size );
1909
1910         /* Memory translation table */
1911         log_num_mtts = fls ( arbel->limits.reserved_mtts - 1 );
1912         MLX_FILL_1 ( init_hca, 65,
1913                      tpt_parameters.mtt_base_addr_l, icm_offset );
1914         DBGC ( arbel, "Arbel %p ICM MTT base = %zx\n", arbel, icm_offset );
1915         icm_offset += icm_usage ( log_num_mtts, arbel->limits.mtt_entry_size );
1916
1917         /* Memory protection table */
1918         log_num_mpts = fls ( arbel->limits.reserved_mrws + 1 - 1 );
1919         MLX_FILL_1 ( init_hca, 61,
1920                      tpt_parameters.mpt_base_adr_l, icm_offset );
1921         MLX_FILL_1 ( init_hca, 62,
1922                      tpt_parameters.log_mpt_sz, log_num_mpts );
1923         DBGC ( arbel, "Arbel %p ICM MTT base = %zx\n", arbel, icm_offset );
1924         icm_offset += icm_usage ( log_num_mpts, arbel->limits.mpt_entry_size );
1925
1926         /* RDMA something or other */
1927         log_num_rdbs = fls ( arbel->limits.reserved_rdbs - 1 );
1928         MLX_FILL_1 ( init_hca, 37,
1929                      qpc_eec_cqc_eqc_rdb_parameters.rdb_base_addr_l,
1930                      icm_offset );
1931         DBGC ( arbel, "Arbel %p ICM RDB base = %zx\n", arbel, icm_offset );
1932         icm_offset += icm_usage ( log_num_rdbs, 32 );
1933
1934         /* Event queue contexts */
1935         log_num_eqs =  fls ( arbel->limits.reserved_eqs + ARBEL_MAX_EQS - 1 );
1936         MLX_FILL_2 ( init_hca, 33,
1937                      qpc_eec_cqc_eqc_rdb_parameters.eqc_base_addr_l,
1938                      ( icm_offset >> 6 ),
1939                      qpc_eec_cqc_eqc_rdb_parameters.log_num_eq,
1940                      log_num_eqs );
1941         DBGC ( arbel, "Arbel %p ICM EQ base = %zx\n", arbel, icm_offset );
1942         icm_offset += ( ( 1 << log_num_eqs ) * arbel->limits.eqc_entry_size );
1943
1944         /* Multicast table */
1945         MLX_FILL_1 ( init_hca, 49,
1946                      multicast_parameters.mc_base_addr_l, icm_offset );
1947         MLX_FILL_1 ( init_hca, 52,
1948                      multicast_parameters.log_mc_table_entry_sz,
1949                      fls ( sizeof ( struct arbelprm_mgm_entry ) - 1 ) );
1950         MLX_FILL_1 ( init_hca, 53,
1951                      multicast_parameters.mc_table_hash_sz, 8 );
1952         MLX_FILL_1 ( init_hca, 54,
1953                      multicast_parameters.log_mc_table_sz, 3 );
1954         DBGC ( arbel, "Arbel %p ICM MC base = %zx\n", arbel, icm_offset );
1955         icm_offset += ( 8 * sizeof ( struct arbelprm_mgm_entry ) );
1956
1957         arbel->icm_len = icm_offset;
1958         arbel->icm_len = ( ( arbel->icm_len + 4095 ) & ~4095 );
1959
1960         /* Get ICM auxiliary area size */
1961         memset ( &icm_size, 0, sizeof ( icm_size ) );
1962         MLX_FILL_1 ( &icm_size, 1, value, arbel->icm_len );
1963         if ( ( rc = arbel_cmd_set_icm_size ( arbel, &icm_size,
1964                                              &icm_aux_size ) ) != 0 ) {
1965                 DBGC ( arbel, "Arbel %p could not set ICM size: %s\n",
1966                        arbel, strerror ( rc ) );
1967                 goto err_set_icm_size;
1968         }
1969         arbel->icm_aux_len = ( MLX_GET ( &icm_aux_size, value ) * 4096 );
1970
1971         /* Allocate ICM data and auxiliary area */
1972         DBGC ( arbel, "Arbel %p requires %zd kB ICM and %zd kB AUX ICM\n",
1973                arbel, ( arbel->icm_len / 1024 ),
1974                ( arbel->icm_aux_len / 1024 ) );
1975         arbel->icm = umalloc ( arbel->icm_len + arbel->icm_aux_len );
1976         if ( ! arbel->icm ) {
1977                 rc = -ENOMEM;
1978                 goto err_alloc;
1979         }
1980
1981         /* Map ICM auxiliary area */
1982         memset ( &map_icm_aux, 0, sizeof ( map_icm_aux ) );
1983         MLX_FILL_2 ( &map_icm_aux, 3,
1984                      log2size, fls ( ( arbel->icm_aux_len / 4096 ) - 1 ),
1985                      pa_l,
1986                      ( user_to_phys ( arbel->icm, arbel->icm_len ) >> 12 ) );
1987         if ( ( rc = arbel_cmd_map_icm_aux ( arbel, &map_icm_aux ) ) != 0 ) {
1988                 DBGC ( arbel, "Arbel %p could not map AUX ICM: %s\n",
1989                        arbel, strerror ( rc ) );
1990                 goto err_map_icm_aux;
1991         }
1992
1993         /* MAP ICM area */
1994         memset ( &map_icm, 0, sizeof ( map_icm ) );
1995         MLX_FILL_2 ( &map_icm, 3,
1996                      log2size, fls ( ( arbel->icm_len / 4096 ) - 1 ),
1997                      pa_l, ( user_to_phys ( arbel->icm, 0 ) >> 12 ) );
1998         if ( ( rc = arbel_cmd_map_icm ( arbel, &map_icm ) ) != 0 ) {
1999                 DBGC ( arbel, "Arbel %p could not map ICM: %s\n",
2000                        arbel, strerror ( rc ) );
2001                 goto err_map_icm;
2002         }
2003
2004         /* Initialise UAR context */
2005         arbel->db_rec = phys_to_virt ( user_to_phys ( arbel->icm, 0 ) +
2006                                        ( arbel->limits.reserved_uars *
2007                                          ARBEL_PAGE_SIZE ) );
2008         memset ( arbel->db_rec, 0, ARBEL_PAGE_SIZE );
2009         db_rec = &arbel->db_rec[ARBEL_GROUP_SEPARATOR_DOORBELL];
2010         MLX_FILL_1 ( &db_rec->qp, 1, res, ARBEL_UAR_RES_GROUP_SEP );
2011
2012         return 0;
2013
2014         arbel_cmd_unmap_icm ( arbel, ( arbel->icm_len / 4096 ) );
2015  err_map_icm:
2016         arbel_cmd_unmap_icm_aux ( arbel );
2017  err_map_icm_aux:
2018         ufree ( arbel->icm );
2019         arbel->icm = UNULL;
2020  err_alloc:
2021  err_set_icm_size:
2022         return rc;
2023 }
2024
2025 /**
2026  * Free ICM
2027  *
2028  * @v arbel             Arbel device
2029  */
2030 static void arbel_free_icm ( struct arbel *arbel ) {
2031         arbel_cmd_unmap_icm ( arbel, ( arbel->icm_len / 4096 ) );
2032         arbel_cmd_unmap_icm_aux ( arbel );
2033         ufree ( arbel->icm );
2034         arbel->icm = UNULL;
2035 }
2036
2037 /***************************************************************************
2038  *
2039  * PCI interface
2040  *
2041  ***************************************************************************
2042  */
2043
2044 /**
2045  * Set up memory protection table
2046  *
2047  * @v arbel             Arbel device
2048  * @ret rc              Return status code
2049  */
2050 static int arbel_setup_mpt ( struct arbel *arbel ) {
2051         struct arbelprm_mpt mpt;
2052         uint32_t key;
2053         int rc;
2054
2055         /* Derive key */
2056         key = ( arbel->limits.reserved_mrws | ARBEL_MKEY_PREFIX );
2057         arbel->reserved_lkey = ( ( key << 8 ) | ( key >> 24 ) );
2058
2059         /* Initialise memory protection table */
2060         memset ( &mpt, 0, sizeof ( mpt ) );
2061         MLX_FILL_4 ( &mpt, 0,
2062                      r_w, 1,
2063                      pa, 1,
2064                      lr, 1,
2065                      lw, 1 );
2066         MLX_FILL_1 ( &mpt, 2, mem_key, key );
2067         MLX_FILL_1 ( &mpt, 3, pd, ARBEL_GLOBAL_PD );
2068         MLX_FILL_1 ( &mpt, 6, reg_wnd_len_h, 0xffffffffUL );
2069         MLX_FILL_1 ( &mpt, 7, reg_wnd_len_l, 0xffffffffUL );
2070         if ( ( rc = arbel_cmd_sw2hw_mpt ( arbel, arbel->limits.reserved_mrws,
2071                                           &mpt ) ) != 0 ) {
2072                 DBGC ( arbel, "Arbel %p could not set up MPT: %s\n",
2073                        arbel, strerror ( rc ) );
2074                 return rc;
2075         }
2076
2077         return 0;
2078 }
2079         
2080 /**
2081  * Probe PCI device
2082  *
2083  * @v pci               PCI device
2084  * @v id                PCI ID
2085  * @ret rc              Return status code
2086  */
2087 static int arbel_probe ( struct pci_device *pci,
2088                          const struct pci_device_id *id __unused ) {
2089         struct arbel *arbel;
2090         struct ib_device *ibdev;
2091         struct arbelprm_init_hca init_hca;
2092         int i;
2093         int rc;
2094
2095         /* Allocate Arbel device */
2096         arbel = zalloc ( sizeof ( *arbel ) );
2097         if ( ! arbel ) {
2098                 rc = -ENOMEM;
2099                 goto err_alloc_arbel;
2100         }
2101         pci_set_drvdata ( pci, arbel );
2102
2103         /* Allocate Infiniband devices */
2104         for ( i = 0 ; i < ARBEL_NUM_PORTS ; i++ ) {
2105                 ibdev = alloc_ibdev ( 0 );
2106                 if ( ! ibdev ) {
2107                         rc = -ENOMEM;
2108                         goto err_alloc_ibdev;
2109                 }
2110                 arbel->ibdev[i] = ibdev;
2111                 ibdev->op = &arbel_ib_operations;
2112                 ibdev->dev = &pci->dev;
2113                 ibdev->port = ( ARBEL_PORT_BASE + i );
2114                 ib_set_drvdata ( ibdev, arbel );
2115         }
2116
2117         /* Fix up PCI device */
2118         adjust_pci_device ( pci );
2119
2120         /* Get PCI BARs */
2121         arbel->config = ioremap ( pci_bar_start ( pci, ARBEL_PCI_CONFIG_BAR ),
2122                                   ARBEL_PCI_CONFIG_BAR_SIZE );
2123         arbel->uar = ioremap ( ( pci_bar_start ( pci, ARBEL_PCI_UAR_BAR ) +
2124                                  ARBEL_PCI_UAR_IDX * ARBEL_PCI_UAR_SIZE ),
2125                                ARBEL_PCI_UAR_SIZE );
2126
2127         /* Allocate space for mailboxes */
2128         arbel->mailbox_in = malloc_dma ( ARBEL_MBOX_SIZE, ARBEL_MBOX_ALIGN );
2129         if ( ! arbel->mailbox_in ) {
2130                 rc = -ENOMEM;
2131                 goto err_mailbox_in;
2132         }
2133         arbel->mailbox_out = malloc_dma ( ARBEL_MBOX_SIZE, ARBEL_MBOX_ALIGN );
2134         if ( ! arbel->mailbox_out ) {
2135                 rc = -ENOMEM;
2136                 goto err_mailbox_out;
2137         }
2138
2139         /* Start firmware */
2140         if ( ( rc = arbel_start_firmware ( arbel ) ) != 0 )
2141                 goto err_start_firmware;
2142
2143         /* Get device limits */
2144         if ( ( rc = arbel_get_limits ( arbel ) ) != 0 )
2145                 goto err_get_limits;
2146
2147         /* Allocate ICM */
2148         memset ( &init_hca, 0, sizeof ( init_hca ) );
2149         if ( ( rc = arbel_alloc_icm ( arbel, &init_hca ) ) != 0 )
2150                 goto err_alloc_icm;
2151
2152         /* Initialise HCA */
2153         MLX_FILL_1 ( &init_hca, 74, uar_parameters.log_max_uars, 1 );
2154         if ( ( rc = arbel_cmd_init_hca ( arbel, &init_hca ) ) != 0 ) {
2155                 DBGC ( arbel, "Arbel %p could not initialise HCA: %s\n",
2156                        arbel, strerror ( rc ) );
2157                 goto err_init_hca;
2158         }
2159
2160         /* Set up memory protection */
2161         if ( ( rc = arbel_setup_mpt ( arbel ) ) != 0 )
2162                 goto err_setup_mpt;
2163
2164         /* Set up event queue */
2165         if ( ( rc = arbel_create_eq ( arbel ) ) != 0 )
2166                 goto err_create_eq;
2167
2168         /* Register Infiniband devices */
2169         for ( i = 0 ; i < ARBEL_NUM_PORTS ; i++ ) {
2170                 if ( ( rc = register_ibdev ( arbel->ibdev[i] ) ) != 0 ) {
2171                         DBGC ( arbel, "Arbel %p could not register IB "
2172                                "device: %s\n", arbel, strerror ( rc ) );
2173                         goto err_register_ibdev;
2174                 }
2175         }
2176
2177         return 0;
2178
2179         i = ARBEL_NUM_PORTS;
2180  err_register_ibdev:
2181         for ( i-- ; i >= 0 ; i-- )
2182                 unregister_ibdev ( arbel->ibdev[i] );
2183         arbel_destroy_eq ( arbel );
2184  err_create_eq:
2185  err_setup_mpt:
2186         arbel_cmd_close_hca ( arbel );
2187  err_init_hca:
2188         arbel_free_icm ( arbel );
2189  err_alloc_icm:
2190  err_get_limits:
2191         arbel_stop_firmware ( arbel );
2192  err_start_firmware:
2193         free_dma ( arbel->mailbox_out, ARBEL_MBOX_SIZE );
2194  err_mailbox_out:
2195         free_dma ( arbel->mailbox_in, ARBEL_MBOX_SIZE );
2196  err_mailbox_in:
2197         i = ARBEL_NUM_PORTS;
2198  err_alloc_ibdev:
2199         for ( i-- ; i >= 0 ; i-- )
2200                 ibdev_put ( arbel->ibdev[i] );
2201         free ( arbel );
2202  err_alloc_arbel:
2203         return rc;
2204 }
2205
2206 /**
2207  * Remove PCI device
2208  *
2209  * @v pci               PCI device
2210  */
2211 static void arbel_remove ( struct pci_device *pci ) {
2212         struct arbel *arbel = pci_get_drvdata ( pci );
2213         int i;
2214
2215         for ( i = ( ARBEL_NUM_PORTS - 1 ) ; i >= 0 ; i-- )
2216                 unregister_ibdev ( arbel->ibdev[i] );
2217         arbel_destroy_eq ( arbel );
2218         arbel_cmd_close_hca ( arbel );
2219         arbel_free_icm ( arbel );
2220         arbel_stop_firmware ( arbel );
2221         arbel_stop_firmware ( arbel );
2222         free_dma ( arbel->mailbox_out, ARBEL_MBOX_SIZE );
2223         free_dma ( arbel->mailbox_in, ARBEL_MBOX_SIZE );
2224         for ( i = ( ARBEL_NUM_PORTS - 1 ) ; i >= 0 ; i-- )
2225                 ibdev_put ( arbel->ibdev[i] );
2226         free ( arbel );
2227 }
2228
2229 static struct pci_device_id arbel_nics[] = {
2230         PCI_ROM ( 0x15b3, 0x6282, "mt25218", "MT25218 HCA driver" ),
2231         PCI_ROM ( 0x15b3, 0x6274, "mt25204", "MT25204 HCA driver" ),
2232 };
2233
2234 struct pci_driver arbel_driver __pci_driver = {
2235         .ids = arbel_nics,
2236         .id_count = ( sizeof ( arbel_nics ) / sizeof ( arbel_nics[0] ) ),
2237         .probe = arbel_probe,
2238         .remove = arbel_remove,
2239 };