[infiniband] Pass address vector in receive completions
[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->lid,
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->qpn );
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_work_queue *wq;
1116         struct ib_queue_pair *qp;
1117         struct arbel_queue_pair *arbel_qp;
1118         struct arbel_send_work_queue *arbel_send_wq;
1119         struct arbel_recv_work_queue *arbel_recv_wq;
1120         struct arbelprm_recv_wqe *recv_wqe;
1121         struct io_buffer *iobuf;
1122         struct ib_address_vector av;
1123         struct ib_global_route_header *grh;
1124         unsigned int opcode;
1125         unsigned long qpn;
1126         int is_send;
1127         unsigned long wqe_adr;
1128         unsigned int wqe_idx;
1129         size_t len;
1130         int rc = 0;
1131
1132         /* Parse completion */
1133         qpn = MLX_GET ( &cqe->normal, my_qpn );
1134         is_send = MLX_GET ( &cqe->normal, s );
1135         wqe_adr = ( MLX_GET ( &cqe->normal, wqe_adr ) << 6 );
1136         opcode = MLX_GET ( &cqe->normal, opcode );
1137         if ( opcode >= ARBEL_OPCODE_RECV_ERROR ) {
1138                 /* "s" field is not valid for error opcodes */
1139                 is_send = ( opcode == ARBEL_OPCODE_SEND_ERROR );
1140                 DBGC ( arbel, "Arbel %p CPN %lx syndrome %lx vendor %lx\n",
1141                        arbel, cq->cqn, MLX_GET ( &cqe->error, 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         if ( is_send ) {
1180                 /* Hand off to completion handler */
1181                 ib_complete_send ( ibdev, qp, iobuf, rc );
1182         } else {
1183                 /* Set received length */
1184                 len = MLX_GET ( &cqe->normal, byte_cnt );
1185                 recv_wqe = &arbel_recv_wq->wqe[wqe_idx].recv;
1186                 assert ( MLX_GET ( &recv_wqe->data[0], local_address_l ) ==
1187                          virt_to_bus ( iobuf->data ) );
1188                 assert ( MLX_GET ( &recv_wqe->data[0], byte_count ) ==
1189                          iob_tailroom ( iobuf ) );
1190                 MLX_FILL_1 ( &recv_wqe->data[0], 0, byte_count, 0 );
1191                 MLX_FILL_1 ( &recv_wqe->data[0], 1,
1192                              l_key, ARBEL_INVALID_LKEY );
1193                 assert ( len <= iob_tailroom ( iobuf ) );
1194                 iob_put ( iobuf, len );
1195                 assert ( iob_len ( iobuf ) >= sizeof ( *grh ) );
1196                 grh = iobuf->data;
1197                 iob_pull ( iobuf, sizeof ( *grh ) );
1198                 /* Construct address vector */
1199                 memset ( &av, 0, sizeof ( av ) );
1200                 av.qpn = MLX_GET ( &cqe->normal, rqpn );
1201                 av.lid = MLX_GET ( &cqe->normal, rlid );
1202                 av.sl = MLX_GET ( &cqe->normal, sl );
1203                 av.gid_present = MLX_GET ( &cqe->normal, g );
1204                 memcpy ( &av.gid, &grh->sgid, sizeof ( av.gid ) );
1205                 /* Hand off to completion handler */
1206                 ib_complete_recv ( ibdev, qp, &av, iobuf, rc );
1207         }
1208
1209         return rc;
1210 }                            
1211
1212 /**
1213  * Poll completion queue
1214  *
1215  * @v ibdev             Infiniband device
1216  * @v cq                Completion queue
1217  */
1218 static void arbel_poll_cq ( struct ib_device *ibdev,
1219                             struct ib_completion_queue *cq ) {
1220         struct arbel *arbel = ib_get_drvdata ( ibdev );
1221         struct arbel_completion_queue *arbel_cq = ib_cq_get_drvdata ( cq );
1222         struct arbelprm_cq_ci_db_record *ci_db_rec;
1223         union arbelprm_completion_entry *cqe;
1224         unsigned int cqe_idx_mask;
1225         int rc;
1226
1227         while ( 1 ) {
1228                 /* Look for completion entry */
1229                 cqe_idx_mask = ( cq->num_cqes - 1 );
1230                 cqe = &arbel_cq->cqe[cq->next_idx & cqe_idx_mask];
1231                 if ( MLX_GET ( &cqe->normal, owner ) != 0 ) {
1232                         /* Entry still owned by hardware; end of poll */
1233                         break;
1234                 }
1235
1236                 /* Handle completion */
1237                 if ( ( rc = arbel_complete ( ibdev, cq, cqe ) ) != 0 ) {
1238                         DBGC ( arbel, "Arbel %p failed to complete: %s\n",
1239                                arbel, strerror ( rc ) );
1240                         DBGC_HD ( arbel, cqe, sizeof ( *cqe ) );
1241                 }
1242
1243                 /* Return ownership to hardware */
1244                 MLX_FILL_1 ( &cqe->normal, 7, owner, 1 );
1245                 barrier();
1246                 /* Update completion queue's index */
1247                 cq->next_idx++;
1248                 /* Update doorbell record */
1249                 ci_db_rec = &arbel->db_rec[arbel_cq->ci_doorbell_idx].cq_ci;
1250                 MLX_FILL_1 ( ci_db_rec, 0,
1251                              counter, ( cq->next_idx & 0xffffffffUL ) );
1252         }
1253 }
1254
1255 /***************************************************************************
1256  *
1257  * Event queues
1258  *
1259  ***************************************************************************
1260  */
1261
1262 /**
1263  * Create event queue
1264  *
1265  * @v arbel             Arbel device
1266  * @ret rc              Return status code
1267  */
1268 static int arbel_create_eq ( struct arbel *arbel ) {
1269         struct arbel_event_queue *arbel_eq = &arbel->eq;
1270         struct arbelprm_eqc eqctx;
1271         struct arbelprm_event_mask mask;
1272         unsigned int i;
1273         int rc;
1274
1275         /* Select event queue number */
1276         arbel_eq->eqn = arbel->limits.reserved_eqs;
1277
1278         /* Calculate doorbell address */
1279         arbel_eq->doorbell = ( arbel->eq_ci_doorbells +
1280                                ARBEL_DB_EQ_OFFSET ( arbel_eq->eqn ) );
1281
1282         /* Allocate event queue itself */
1283         arbel_eq->eqe_size =
1284                 ( ARBEL_NUM_EQES * sizeof ( arbel_eq->eqe[0] ) );
1285         arbel_eq->eqe = malloc_dma ( arbel_eq->eqe_size,
1286                                      sizeof ( arbel_eq->eqe[0] ) );
1287         if ( ! arbel_eq->eqe ) {
1288                 rc = -ENOMEM;
1289                 goto err_eqe;
1290         }
1291         memset ( arbel_eq->eqe, 0, arbel_eq->eqe_size );
1292         for ( i = 0 ; i < ARBEL_NUM_EQES ; i++ ) {
1293                 MLX_FILL_1 ( &arbel_eq->eqe[i].generic, 7, owner, 1 );
1294         }
1295         barrier();
1296
1297         /* Hand queue over to hardware */
1298         memset ( &eqctx, 0, sizeof ( eqctx ) );
1299         MLX_FILL_1 ( &eqctx, 0, st, 0xa /* "Fired" */ );
1300         MLX_FILL_1 ( &eqctx, 2,
1301                      start_address_l, virt_to_phys ( arbel_eq->eqe ) );
1302         MLX_FILL_1 ( &eqctx, 3, log_eq_size, fls ( ARBEL_NUM_EQES - 1 ) );
1303         MLX_FILL_1 ( &eqctx, 6, pd, ARBEL_GLOBAL_PD );
1304         MLX_FILL_1 ( &eqctx, 7, lkey, arbel->reserved_lkey );
1305         if ( ( rc = arbel_cmd_sw2hw_eq ( arbel, arbel_eq->eqn,
1306                                          &eqctx ) ) != 0 ) {
1307                 DBGC ( arbel, "Arbel %p SW2HW_EQ failed: %s\n",
1308                        arbel, strerror ( rc ) );
1309                 goto err_sw2hw_eq;
1310         }
1311
1312         /* Map events to this event queue */
1313         memset ( &mask, 0, sizeof ( mask ) );
1314         MLX_FILL_1 ( &mask, 1, port_state_change, 1 );
1315         if ( ( rc = arbel_cmd_map_eq ( arbel,
1316                                        ( ARBEL_MAP_EQ | arbel_eq->eqn ),
1317                                        &mask ) ) != 0 ) {
1318                 DBGC ( arbel, "Arbel %p MAP_EQ failed: %s\n",
1319                        arbel, strerror ( rc )  );
1320                 goto err_map_eq;
1321         }
1322
1323         DBGC ( arbel, "Arbel %p EQN %#lx ring at [%p,%p])\n",
1324                arbel, arbel_eq->eqn, arbel_eq->eqe,
1325                ( ( ( void * ) arbel_eq->eqe ) + arbel_eq->eqe_size ) );
1326         return 0;
1327
1328  err_map_eq:
1329         arbel_cmd_hw2sw_eq ( arbel, arbel_eq->eqn, &eqctx );
1330  err_sw2hw_eq:
1331         free_dma ( arbel_eq->eqe, arbel_eq->eqe_size );
1332  err_eqe:
1333         memset ( arbel_eq, 0, sizeof ( *arbel_eq ) );
1334         return rc;
1335 }
1336
1337 /**
1338  * Destroy event queue
1339  *
1340  * @v arbel             Arbel device
1341  */
1342 static void arbel_destroy_eq ( struct arbel *arbel ) {
1343         struct arbel_event_queue *arbel_eq = &arbel->eq;
1344         struct arbelprm_eqc eqctx;
1345         struct arbelprm_event_mask mask;
1346         int rc;
1347
1348         /* Unmap events from event queue */
1349         memset ( &mask, 0, sizeof ( mask ) );
1350         MLX_FILL_1 ( &mask, 1, port_state_change, 1 );
1351         if ( ( rc = arbel_cmd_map_eq ( arbel,
1352                                        ( ARBEL_UNMAP_EQ | arbel_eq->eqn ),
1353                                        &mask ) ) != 0 ) {
1354                 DBGC ( arbel, "Arbel %p FATAL MAP_EQ failed to unmap: %s\n",
1355                        arbel, strerror ( rc ) );
1356                 /* Continue; HCA may die but system should survive */
1357         }
1358
1359         /* Take ownership back from hardware */
1360         if ( ( rc = arbel_cmd_hw2sw_eq ( arbel, arbel_eq->eqn,
1361                                          &eqctx ) ) != 0 ) {
1362                 DBGC ( arbel, "Arbel %p FATAL HW2SW_EQ failed: %s\n",
1363                        arbel, strerror ( rc ) );
1364                 /* Leak memory and return; at least we avoid corruption */
1365                 return;
1366         }
1367
1368         /* Free memory */
1369         free_dma ( arbel_eq->eqe, arbel_eq->eqe_size );
1370         memset ( arbel_eq, 0, sizeof ( *arbel_eq ) );
1371 }
1372
1373 /**
1374  * Handle port state event
1375  *
1376  * @v arbel             Arbel device
1377  * @v eqe               Port state change event queue entry
1378  */
1379 static void arbel_event_port_state_change ( struct arbel *arbel,
1380                                             union arbelprm_event_entry *eqe){
1381         unsigned int port;
1382         int link_up;
1383
1384         /* Get port and link status */
1385         port = ( MLX_GET ( &eqe->port_state_change, data.p ) - 1 );
1386         link_up = ( MLX_GET ( &eqe->generic, event_sub_type ) & 0x04 );
1387         DBGC ( arbel, "Arbel %p port %d link %s\n", arbel, ( port + 1 ),
1388                ( link_up ? "up" : "down" ) );
1389
1390         /* Sanity check */
1391         if ( port >= ARBEL_NUM_PORTS ) {
1392                 DBGC ( arbel, "Arbel %p port %d does not exist!\n",
1393                        arbel, ( port + 1 ) );
1394                 return;
1395         }
1396
1397         /* Notify Infiniband core of link state change */
1398         ib_link_state_changed ( arbel->ibdev[port] );
1399 }
1400
1401 /**
1402  * Poll event queue
1403  *
1404  * @v ibdev             Infiniband device
1405  */
1406 static void arbel_poll_eq ( struct ib_device *ibdev ) {
1407         struct arbel *arbel = ib_get_drvdata ( ibdev );
1408         struct arbel_event_queue *arbel_eq = &arbel->eq;
1409         union arbelprm_event_entry *eqe;
1410         union arbelprm_eq_doorbell_register db_reg;
1411         unsigned int eqe_idx_mask;
1412         unsigned int event_type;
1413
1414         while ( 1 ) {
1415                 /* Look for event entry */
1416                 eqe_idx_mask = ( ARBEL_NUM_EQES - 1 );
1417                 eqe = &arbel_eq->eqe[arbel_eq->next_idx & eqe_idx_mask];
1418                 if ( MLX_GET ( &eqe->generic, owner ) != 0 ) {
1419                         /* Entry still owned by hardware; end of poll */
1420                         break;
1421                 }
1422                 DBGCP ( arbel, "Arbel %p event:\n", arbel );
1423                 DBGCP_HD ( arbel, eqe, sizeof ( *eqe ) );
1424
1425                 /* Handle event */
1426                 event_type = MLX_GET ( &eqe->generic, event_type );
1427                 switch ( event_type ) {
1428                 case ARBEL_EV_PORT_STATE_CHANGE:
1429                         arbel_event_port_state_change ( arbel, eqe );
1430                         break;
1431                 default:
1432                         DBGC ( arbel, "Arbel %p unrecognised event type "
1433                                "%#x:\n", arbel, event_type );
1434                         DBGC_HD ( arbel, eqe, sizeof ( *eqe ) );
1435                         break;
1436                 }
1437
1438                 /* Return ownership to hardware */
1439                 MLX_FILL_1 ( &eqe->generic, 7, owner, 1 );
1440                 barrier();
1441
1442                 /* Update event queue's index */
1443                 arbel_eq->next_idx++;
1444
1445                 /* Ring doorbell */
1446                 MLX_FILL_1 ( &db_reg.ci, 0, ci, arbel_eq->next_idx );
1447                 DBGCP ( arbel, "Ringing doorbell %08lx with %08lx\n",
1448                         virt_to_phys ( arbel_eq->doorbell ),
1449                         db_reg.dword[0] );
1450                 writel ( db_reg.dword[0], arbel_eq->doorbell );
1451         }
1452 }
1453
1454 /***************************************************************************
1455  *
1456  * Infiniband link-layer operations
1457  *
1458  ***************************************************************************
1459  */
1460
1461 /**
1462  * Initialise Infiniband link
1463  *
1464  * @v ibdev             Infiniband device
1465  * @ret rc              Return status code
1466  */
1467 static int arbel_open ( struct ib_device *ibdev ) {
1468         struct arbel *arbel = ib_get_drvdata ( ibdev );
1469         struct arbelprm_init_ib init_ib;
1470         int rc;
1471
1472         memset ( &init_ib, 0, sizeof ( init_ib ) );
1473         MLX_FILL_3 ( &init_ib, 0,
1474                      mtu_cap, ARBEL_MTU_2048,
1475                      port_width_cap, 3,
1476                      vl_cap, 1 );
1477         MLX_FILL_1 ( &init_ib, 1, max_gid, 1 );
1478         MLX_FILL_1 ( &init_ib, 2, max_pkey, 64 );
1479         if ( ( rc = arbel_cmd_init_ib ( arbel, ibdev->port,
1480                                         &init_ib ) ) != 0 ) {
1481                 DBGC ( arbel, "Arbel %p could not intialise IB: %s\n",
1482                        arbel, strerror ( rc ) );
1483                 return rc;
1484         }
1485
1486         return 0;
1487 }
1488
1489 /**
1490  * Close Infiniband link
1491  *
1492  * @v ibdev             Infiniband device
1493  */
1494 static void arbel_close ( struct ib_device *ibdev ) {
1495         struct arbel *arbel = ib_get_drvdata ( ibdev );
1496         int rc;
1497
1498         if ( ( rc = arbel_cmd_close_ib ( arbel, ibdev->port ) ) != 0 ) {
1499                 DBGC ( arbel, "Arbel %p could not close IB: %s\n",
1500                        arbel, strerror ( rc ) );
1501                 /* Nothing we can do about this */
1502         }
1503 }
1504
1505 /***************************************************************************
1506  *
1507  * Multicast group operations
1508  *
1509  ***************************************************************************
1510  */
1511
1512 /**
1513  * Attach to multicast group
1514  *
1515  * @v ibdev             Infiniband device
1516  * @v qp                Queue pair
1517  * @v gid               Multicast GID
1518  * @ret rc              Return status code
1519  */
1520 static int arbel_mcast_attach ( struct ib_device *ibdev,
1521                                 struct ib_queue_pair *qp,
1522                                 struct ib_gid *gid ) {
1523         struct arbel *arbel = ib_get_drvdata ( ibdev );
1524         struct arbelprm_mgm_hash hash;
1525         struct arbelprm_mgm_entry mgm;
1526         unsigned int index;
1527         int rc;
1528
1529         /* Generate hash table index */
1530         if ( ( rc = arbel_cmd_mgid_hash ( arbel, gid, &hash ) ) != 0 ) {
1531                 DBGC ( arbel, "Arbel %p could not hash GID: %s\n",
1532                        arbel, strerror ( rc ) );
1533                 return rc;
1534         }
1535         index = MLX_GET ( &hash, hash );
1536
1537         /* Check for existing hash table entry */
1538         if ( ( rc = arbel_cmd_read_mgm ( arbel, index, &mgm ) ) != 0 ) {
1539                 DBGC ( arbel, "Arbel %p could not read MGM %#x: %s\n",
1540                        arbel, index, strerror ( rc ) );
1541                 return rc;
1542         }
1543         if ( MLX_GET ( &mgm, mgmqp_0.qi ) != 0 ) {
1544                 /* FIXME: this implementation allows only a single QP
1545                  * per multicast group, and doesn't handle hash
1546                  * collisions.  Sufficient for IPoIB but may need to
1547                  * be extended in future.
1548                  */
1549                 DBGC ( arbel, "Arbel %p MGID index %#x already in use\n",
1550                        arbel, index );
1551                 return -EBUSY;
1552         }
1553
1554         /* Update hash table entry */
1555         MLX_FILL_2 ( &mgm, 8,
1556                      mgmqp_0.qpn_i, qp->qpn,
1557                      mgmqp_0.qi, 1 );
1558         memcpy ( &mgm.u.dwords[4], gid, sizeof ( *gid ) );
1559         if ( ( rc = arbel_cmd_write_mgm ( arbel, index, &mgm ) ) != 0 ) {
1560                 DBGC ( arbel, "Arbel %p could not write MGM %#x: %s\n",
1561                        arbel, index, strerror ( rc ) );
1562                 return rc;
1563         }
1564
1565         return 0;
1566 }
1567
1568 /**
1569  * Detach from multicast group
1570  *
1571  * @v ibdev             Infiniband device
1572  * @v qp                Queue pair
1573  * @v gid               Multicast GID
1574  */
1575 static void arbel_mcast_detach ( struct ib_device *ibdev,
1576                                  struct ib_queue_pair *qp __unused,
1577                                  struct ib_gid *gid ) {
1578         struct arbel *arbel = ib_get_drvdata ( ibdev );
1579         struct arbelprm_mgm_hash hash;
1580         struct arbelprm_mgm_entry mgm;
1581         unsigned int index;
1582         int rc;
1583
1584         /* Generate hash table index */
1585         if ( ( rc = arbel_cmd_mgid_hash ( arbel, gid, &hash ) ) != 0 ) {
1586                 DBGC ( arbel, "Arbel %p could not hash GID: %s\n",
1587                        arbel, strerror ( rc ) );
1588                 return;
1589         }
1590         index = MLX_GET ( &hash, hash );
1591
1592         /* Clear hash table entry */
1593         memset ( &mgm, 0, sizeof ( mgm ) );
1594         if ( ( rc = arbel_cmd_write_mgm ( arbel, index, &mgm ) ) != 0 ) {
1595                 DBGC ( arbel, "Arbel %p could not write MGM %#x: %s\n",
1596                        arbel, index, strerror ( rc ) );
1597                 return;
1598         }
1599 }
1600
1601 /***************************************************************************
1602  *
1603  * MAD operations
1604  *
1605  ***************************************************************************
1606  */
1607
1608 /**
1609  * Issue management datagram
1610  *
1611  * @v ibdev             Infiniband device
1612  * @v mad               Management datagram
1613  * @v len               Length of management datagram
1614  * @ret rc              Return status code
1615  */
1616 static int arbel_mad ( struct ib_device *ibdev, struct ib_mad_hdr *mad,
1617                        size_t len ) {
1618         struct arbel *arbel = ib_get_drvdata ( ibdev );
1619         union arbelprm_mad mad_ifc;
1620         int rc;
1621
1622         /* Copy in request packet */
1623         memset ( &mad_ifc, 0, sizeof ( mad_ifc ) );
1624         assert ( len <= sizeof ( mad_ifc.mad ) );
1625         memcpy ( &mad_ifc.mad, mad, len );
1626
1627         /* Issue MAD */
1628         if ( ( rc = arbel_cmd_mad_ifc ( arbel, ibdev->port,
1629                                         &mad_ifc ) ) != 0 ) {
1630                 DBGC ( arbel, "Arbel %p could not issue MAD IFC: %s\n",
1631                        arbel, strerror ( rc ) );
1632                 return rc;
1633         }
1634
1635         /* Copy out reply packet */
1636         memcpy ( mad, &mad_ifc.mad, len );
1637
1638         if ( mad->status != 0 ) {
1639                 DBGC ( arbel, "Arbel %p MAD IFC status %04x\n",
1640                        arbel, ntohs ( mad->status ) );
1641                 return -EIO;
1642         }
1643         return 0;
1644 }
1645
1646 /** Arbel Infiniband operations */
1647 static struct ib_device_operations arbel_ib_operations = {
1648         .create_cq      = arbel_create_cq,
1649         .destroy_cq     = arbel_destroy_cq,
1650         .create_qp      = arbel_create_qp,
1651         .modify_qp      = arbel_modify_qp,
1652         .destroy_qp     = arbel_destroy_qp,
1653         .post_send      = arbel_post_send,
1654         .post_recv      = arbel_post_recv,
1655         .poll_cq        = arbel_poll_cq,
1656         .poll_eq        = arbel_poll_eq,
1657         .open           = arbel_open,
1658         .close          = arbel_close,
1659         .mcast_attach   = arbel_mcast_attach,
1660         .mcast_detach   = arbel_mcast_detach,
1661         .mad            = arbel_mad,
1662 };
1663
1664 /***************************************************************************
1665  *
1666  * Firmware control
1667  *
1668  ***************************************************************************
1669  */
1670
1671 /**
1672  * Start firmware running
1673  *
1674  * @v arbel             Arbel device
1675  * @ret rc              Return status code
1676  */
1677 static int arbel_start_firmware ( struct arbel *arbel ) {
1678         struct arbelprm_query_fw fw;
1679         struct arbelprm_access_lam lam;
1680         struct arbelprm_virtual_physical_mapping map_fa;
1681         unsigned int fw_pages;
1682         unsigned int log2_fw_pages;
1683         size_t fw_size;
1684         physaddr_t fw_base;
1685         uint64_t eq_set_ci_base_addr;
1686         int rc;
1687
1688         /* Get firmware parameters */
1689         if ( ( rc = arbel_cmd_query_fw ( arbel, &fw ) ) != 0 ) {
1690                 DBGC ( arbel, "Arbel %p could not query firmware: %s\n",
1691                        arbel, strerror ( rc ) );
1692                 goto err_query_fw;
1693         }
1694         DBGC ( arbel, "Arbel %p firmware version %ld.%ld.%ld\n", arbel,
1695                MLX_GET ( &fw, fw_rev_major ), MLX_GET ( &fw, fw_rev_minor ),
1696                MLX_GET ( &fw, fw_rev_subminor ) );
1697         fw_pages = MLX_GET ( &fw, fw_pages );
1698         log2_fw_pages = fls ( fw_pages - 1 );
1699         fw_pages = ( 1 << log2_fw_pages );
1700         DBGC ( arbel, "Arbel %p requires %d kB for firmware\n",
1701                arbel, ( fw_pages * 4 ) );
1702         eq_set_ci_base_addr =
1703                 ( ( (uint64_t) MLX_GET ( &fw, eq_set_ci_base_addr_h ) << 32 ) |
1704                   ( (uint64_t) MLX_GET ( &fw, eq_set_ci_base_addr_l ) ) );
1705         arbel->eq_ci_doorbells = ioremap ( eq_set_ci_base_addr, 0x200 );
1706
1707         /* Enable locally-attached memory.  Ignore failure; there may
1708          * be no attached memory.
1709          */
1710         arbel_cmd_enable_lam ( arbel, &lam );
1711
1712         /* Allocate firmware pages and map firmware area */
1713         fw_size = ( fw_pages * 4096 );
1714         arbel->firmware_area = umalloc ( fw_size * 2 );
1715         if ( ! arbel->firmware_area ) {
1716                 rc = -ENOMEM;
1717                 goto err_alloc_fa;
1718         }
1719         fw_base = ( user_to_phys ( arbel->firmware_area, fw_size ) &
1720                     ~( fw_size - 1 ) );
1721         DBGC ( arbel, "Arbel %p firmware area at physical [%lx,%lx)\n",
1722                arbel, fw_base, ( fw_base + fw_size ) );
1723         memset ( &map_fa, 0, sizeof ( map_fa ) );
1724         MLX_FILL_2 ( &map_fa, 3,
1725                      log2size, log2_fw_pages,
1726                      pa_l, ( fw_base >> 12 ) );
1727         if ( ( rc = arbel_cmd_map_fa ( arbel, &map_fa ) ) != 0 ) {
1728                 DBGC ( arbel, "Arbel %p could not map firmware: %s\n",
1729                        arbel, strerror ( rc ) );
1730                 goto err_map_fa;
1731         }
1732
1733         /* Start firmware */
1734         if ( ( rc = arbel_cmd_run_fw ( arbel ) ) != 0 ) {
1735                 DBGC ( arbel, "Arbel %p could not run firmware: %s\n",
1736                        arbel, strerror ( rc ) );
1737                 goto err_run_fw;
1738         }
1739
1740         DBGC ( arbel, "Arbel %p firmware started\n", arbel );
1741         return 0;
1742
1743  err_run_fw:
1744         arbel_cmd_unmap_fa ( arbel );
1745  err_map_fa:
1746         ufree ( arbel->firmware_area );
1747         arbel->firmware_area = UNULL;
1748  err_alloc_fa:
1749  err_query_fw:
1750         return rc;
1751 }
1752
1753 /**
1754  * Stop firmware running
1755  *
1756  * @v arbel             Arbel device
1757  */
1758 static void arbel_stop_firmware ( struct arbel *arbel ) {
1759         int rc;
1760
1761         if ( ( rc = arbel_cmd_unmap_fa ( arbel ) ) != 0 ) {
1762                 DBGC ( arbel, "Arbel %p FATAL could not stop firmware: %s\n",
1763                        arbel, strerror ( rc ) );
1764                 /* Leak memory and return; at least we avoid corruption */
1765                 return;
1766         }
1767         ufree ( arbel->firmware_area );
1768         arbel->firmware_area = UNULL;
1769 }
1770
1771 /***************************************************************************
1772  *
1773  * Infinihost Context Memory management
1774  *
1775  ***************************************************************************
1776  */
1777
1778 /**
1779  * Get device limits
1780  *
1781  * @v arbel             Arbel device
1782  * @ret rc              Return status code
1783  */
1784 static int arbel_get_limits ( struct arbel *arbel ) {
1785         struct arbelprm_query_dev_lim dev_lim;
1786         int rc;
1787
1788         if ( ( rc = arbel_cmd_query_dev_lim ( arbel, &dev_lim ) ) != 0 ) {
1789                 DBGC ( arbel, "Arbel %p could not get device limits: %s\n",
1790                        arbel, strerror ( rc ) );
1791                 return rc;
1792         }
1793
1794         arbel->limits.reserved_qps =
1795                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_qps ) );
1796         arbel->limits.qpc_entry_size = MLX_GET ( &dev_lim, qpc_entry_sz );
1797         arbel->limits.eqpc_entry_size = MLX_GET ( &dev_lim, eqpc_entry_sz );
1798         arbel->limits.reserved_srqs =
1799                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_srqs ) );
1800         arbel->limits.srqc_entry_size = MLX_GET ( &dev_lim, srq_entry_sz );
1801         arbel->limits.reserved_ees =
1802                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_ees ) );
1803         arbel->limits.eec_entry_size = MLX_GET ( &dev_lim, eec_entry_sz );
1804         arbel->limits.eeec_entry_size = MLX_GET ( &dev_lim, eeec_entry_sz );
1805         arbel->limits.reserved_cqs =
1806                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_cqs ) );
1807         arbel->limits.cqc_entry_size = MLX_GET ( &dev_lim, cqc_entry_sz );
1808         arbel->limits.reserved_eqs = MLX_GET ( &dev_lim, num_rsvd_eqs );
1809         arbel->limits.reserved_mtts =
1810                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_mtts ) );
1811         arbel->limits.mtt_entry_size = MLX_GET ( &dev_lim, mtt_entry_sz );
1812         arbel->limits.reserved_mrws =
1813                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_mrws ) );
1814         arbel->limits.mpt_entry_size = MLX_GET ( &dev_lim, mpt_entry_sz );
1815         arbel->limits.reserved_rdbs =
1816                 ( 1 << MLX_GET ( &dev_lim, log2_rsvd_rdbs ) );
1817         arbel->limits.eqc_entry_size = MLX_GET ( &dev_lim, eqc_entry_sz );
1818         arbel->limits.reserved_uars = MLX_GET ( &dev_lim, num_rsvd_uars );
1819
1820         return 0;
1821 }
1822
1823 /**
1824  * Get ICM usage
1825  *
1826  * @v log_num_entries   Log2 of the number of entries
1827  * @v entry_size        Entry size
1828  * @ret usage           Usage size in ICM
1829  */
1830 static size_t icm_usage ( unsigned int log_num_entries, size_t entry_size ) {
1831         size_t usage;
1832
1833         usage = ( ( 1 << log_num_entries ) * entry_size );
1834         usage = ( ( usage + 4095 ) & ~4095 );
1835         return usage;
1836 }
1837
1838 /**
1839  * Allocate ICM
1840  *
1841  * @v arbel             Arbel device
1842  * @v init_hca          INIT_HCA structure to fill in
1843  * @ret rc              Return status code
1844  */
1845 static int arbel_alloc_icm ( struct arbel *arbel,
1846                              struct arbelprm_init_hca *init_hca ) {
1847         struct arbelprm_scalar_parameter icm_size;
1848         struct arbelprm_scalar_parameter icm_aux_size;
1849         struct arbelprm_virtual_physical_mapping map_icm_aux;
1850         struct arbelprm_virtual_physical_mapping map_icm;
1851         union arbelprm_doorbell_record *db_rec;
1852         size_t icm_offset = 0;
1853         unsigned int log_num_qps, log_num_srqs, log_num_ees, log_num_cqs;
1854         unsigned int log_num_mtts, log_num_mpts, log_num_rdbs, log_num_eqs;
1855         int rc;
1856
1857         icm_offset = ( ( arbel->limits.reserved_uars + 1 ) << 12 );
1858
1859         /* Queue pair contexts */
1860         log_num_qps = fls ( arbel->limits.reserved_qps + ARBEL_MAX_QPS - 1 );
1861         MLX_FILL_2 ( init_hca, 13,
1862                      qpc_eec_cqc_eqc_rdb_parameters.qpc_base_addr_l,
1863                      ( icm_offset >> 7 ),
1864                      qpc_eec_cqc_eqc_rdb_parameters.log_num_of_qp,
1865                      log_num_qps );
1866         DBGC ( arbel, "Arbel %p ICM QPC base = %zx\n", arbel, icm_offset );
1867         icm_offset += icm_usage ( log_num_qps, arbel->limits.qpc_entry_size );
1868
1869         /* Extended queue pair contexts */
1870         MLX_FILL_1 ( init_hca, 25,
1871                      qpc_eec_cqc_eqc_rdb_parameters.eqpc_base_addr_l,
1872                      icm_offset );
1873         DBGC ( arbel, "Arbel %p ICM EQPC base = %zx\n", arbel, icm_offset );
1874         //      icm_offset += icm_usage ( log_num_qps, arbel->limits.eqpc_entry_size );
1875         icm_offset += icm_usage ( log_num_qps, arbel->limits.qpc_entry_size );  
1876
1877         /* Shared receive queue contexts */
1878         log_num_srqs = fls ( arbel->limits.reserved_srqs - 1 );
1879         MLX_FILL_2 ( init_hca, 19,
1880                      qpc_eec_cqc_eqc_rdb_parameters.srqc_base_addr_l,
1881                      ( icm_offset >> 5 ),
1882                      qpc_eec_cqc_eqc_rdb_parameters.log_num_of_srq,
1883                      log_num_srqs );
1884         DBGC ( arbel, "Arbel %p ICM SRQC base = %zx\n", arbel, icm_offset );
1885         icm_offset += icm_usage ( log_num_srqs, arbel->limits.srqc_entry_size );
1886
1887         /* End-to-end contexts */
1888         log_num_ees = fls ( arbel->limits.reserved_ees - 1 );
1889         MLX_FILL_2 ( init_hca, 17,
1890                      qpc_eec_cqc_eqc_rdb_parameters.eec_base_addr_l,
1891                      ( icm_offset >> 7 ),
1892                      qpc_eec_cqc_eqc_rdb_parameters.log_num_of_ee,
1893                      log_num_ees );
1894         DBGC ( arbel, "Arbel %p ICM EEC base = %zx\n", arbel, icm_offset );
1895         icm_offset += icm_usage ( log_num_ees, arbel->limits.eec_entry_size );
1896
1897         /* Extended end-to-end contexts */
1898         MLX_FILL_1 ( init_hca, 29,
1899                      qpc_eec_cqc_eqc_rdb_parameters.eeec_base_addr_l,
1900                      icm_offset );
1901         DBGC ( arbel, "Arbel %p ICM EEEC base = %zx\n", arbel, icm_offset );
1902         icm_offset += icm_usage ( log_num_ees, arbel->limits.eeec_entry_size );
1903
1904         /* Completion queue contexts */
1905         log_num_cqs = fls ( arbel->limits.reserved_cqs + ARBEL_MAX_CQS - 1 );
1906         MLX_FILL_2 ( init_hca, 21,
1907                      qpc_eec_cqc_eqc_rdb_parameters.cqc_base_addr_l,
1908                      ( icm_offset >> 6 ),
1909                      qpc_eec_cqc_eqc_rdb_parameters.log_num_of_cq,
1910                      log_num_cqs );
1911         DBGC ( arbel, "Arbel %p ICM CQC base = %zx\n", arbel, icm_offset );
1912         icm_offset += icm_usage ( log_num_cqs, arbel->limits.cqc_entry_size );
1913
1914         /* Memory translation table */
1915         log_num_mtts = fls ( arbel->limits.reserved_mtts - 1 );
1916         MLX_FILL_1 ( init_hca, 65,
1917                      tpt_parameters.mtt_base_addr_l, icm_offset );
1918         DBGC ( arbel, "Arbel %p ICM MTT base = %zx\n", arbel, icm_offset );
1919         icm_offset += icm_usage ( log_num_mtts, arbel->limits.mtt_entry_size );
1920
1921         /* Memory protection table */
1922         log_num_mpts = fls ( arbel->limits.reserved_mrws + 1 - 1 );
1923         MLX_FILL_1 ( init_hca, 61,
1924                      tpt_parameters.mpt_base_adr_l, icm_offset );
1925         MLX_FILL_1 ( init_hca, 62,
1926                      tpt_parameters.log_mpt_sz, log_num_mpts );
1927         DBGC ( arbel, "Arbel %p ICM MTT base = %zx\n", arbel, icm_offset );
1928         icm_offset += icm_usage ( log_num_mpts, arbel->limits.mpt_entry_size );
1929
1930         /* RDMA something or other */
1931         log_num_rdbs = fls ( arbel->limits.reserved_rdbs - 1 );
1932         MLX_FILL_1 ( init_hca, 37,
1933                      qpc_eec_cqc_eqc_rdb_parameters.rdb_base_addr_l,
1934                      icm_offset );
1935         DBGC ( arbel, "Arbel %p ICM RDB base = %zx\n", arbel, icm_offset );
1936         icm_offset += icm_usage ( log_num_rdbs, 32 );
1937
1938         /* Event queue contexts */
1939         log_num_eqs =  fls ( arbel->limits.reserved_eqs + ARBEL_MAX_EQS - 1 );
1940         MLX_FILL_2 ( init_hca, 33,
1941                      qpc_eec_cqc_eqc_rdb_parameters.eqc_base_addr_l,
1942                      ( icm_offset >> 6 ),
1943                      qpc_eec_cqc_eqc_rdb_parameters.log_num_eq,
1944                      log_num_eqs );
1945         DBGC ( arbel, "Arbel %p ICM EQ base = %zx\n", arbel, icm_offset );
1946         icm_offset += ( ( 1 << log_num_eqs ) * arbel->limits.eqc_entry_size );
1947
1948         /* Multicast table */
1949         MLX_FILL_1 ( init_hca, 49,
1950                      multicast_parameters.mc_base_addr_l, icm_offset );
1951         MLX_FILL_1 ( init_hca, 52,
1952                      multicast_parameters.log_mc_table_entry_sz,
1953                      fls ( sizeof ( struct arbelprm_mgm_entry ) - 1 ) );
1954         MLX_FILL_1 ( init_hca, 53,
1955                      multicast_parameters.mc_table_hash_sz, 8 );
1956         MLX_FILL_1 ( init_hca, 54,
1957                      multicast_parameters.log_mc_table_sz, 3 );
1958         DBGC ( arbel, "Arbel %p ICM MC base = %zx\n", arbel, icm_offset );
1959         icm_offset += ( 8 * sizeof ( struct arbelprm_mgm_entry ) );
1960
1961         arbel->icm_len = icm_offset;
1962         arbel->icm_len = ( ( arbel->icm_len + 4095 ) & ~4095 );
1963
1964         /* Get ICM auxiliary area size */
1965         memset ( &icm_size, 0, sizeof ( icm_size ) );
1966         MLX_FILL_1 ( &icm_size, 1, value, arbel->icm_len );
1967         if ( ( rc = arbel_cmd_set_icm_size ( arbel, &icm_size,
1968                                              &icm_aux_size ) ) != 0 ) {
1969                 DBGC ( arbel, "Arbel %p could not set ICM size: %s\n",
1970                        arbel, strerror ( rc ) );
1971                 goto err_set_icm_size;
1972         }
1973         arbel->icm_aux_len = ( MLX_GET ( &icm_aux_size, value ) * 4096 );
1974
1975         /* Allocate ICM data and auxiliary area */
1976         DBGC ( arbel, "Arbel %p requires %zd kB ICM and %zd kB AUX ICM\n",
1977                arbel, ( arbel->icm_len / 1024 ),
1978                ( arbel->icm_aux_len / 1024 ) );
1979         arbel->icm = umalloc ( arbel->icm_len + arbel->icm_aux_len );
1980         if ( ! arbel->icm ) {
1981                 rc = -ENOMEM;
1982                 goto err_alloc;
1983         }
1984
1985         /* Map ICM auxiliary area */
1986         memset ( &map_icm_aux, 0, sizeof ( map_icm_aux ) );
1987         MLX_FILL_2 ( &map_icm_aux, 3,
1988                      log2size, fls ( ( arbel->icm_aux_len / 4096 ) - 1 ),
1989                      pa_l,
1990                      ( user_to_phys ( arbel->icm, arbel->icm_len ) >> 12 ) );
1991         if ( ( rc = arbel_cmd_map_icm_aux ( arbel, &map_icm_aux ) ) != 0 ) {
1992                 DBGC ( arbel, "Arbel %p could not map AUX ICM: %s\n",
1993                        arbel, strerror ( rc ) );
1994                 goto err_map_icm_aux;
1995         }
1996
1997         /* MAP ICM area */
1998         memset ( &map_icm, 0, sizeof ( map_icm ) );
1999         MLX_FILL_2 ( &map_icm, 3,
2000                      log2size, fls ( ( arbel->icm_len / 4096 ) - 1 ),
2001                      pa_l, ( user_to_phys ( arbel->icm, 0 ) >> 12 ) );
2002         if ( ( rc = arbel_cmd_map_icm ( arbel, &map_icm ) ) != 0 ) {
2003                 DBGC ( arbel, "Arbel %p could not map ICM: %s\n",
2004                        arbel, strerror ( rc ) );
2005                 goto err_map_icm;
2006         }
2007
2008         /* Initialise UAR context */
2009         arbel->db_rec = phys_to_virt ( user_to_phys ( arbel->icm, 0 ) +
2010                                        ( arbel->limits.reserved_uars *
2011                                          ARBEL_PAGE_SIZE ) );
2012         memset ( arbel->db_rec, 0, ARBEL_PAGE_SIZE );
2013         db_rec = &arbel->db_rec[ARBEL_GROUP_SEPARATOR_DOORBELL];
2014         MLX_FILL_1 ( &db_rec->qp, 1, res, ARBEL_UAR_RES_GROUP_SEP );
2015
2016         return 0;
2017
2018         arbel_cmd_unmap_icm ( arbel, ( arbel->icm_len / 4096 ) );
2019  err_map_icm:
2020         arbel_cmd_unmap_icm_aux ( arbel );
2021  err_map_icm_aux:
2022         ufree ( arbel->icm );
2023         arbel->icm = UNULL;
2024  err_alloc:
2025  err_set_icm_size:
2026         return rc;
2027 }
2028
2029 /**
2030  * Free ICM
2031  *
2032  * @v arbel             Arbel device
2033  */
2034 static void arbel_free_icm ( struct arbel *arbel ) {
2035         arbel_cmd_unmap_icm ( arbel, ( arbel->icm_len / 4096 ) );
2036         arbel_cmd_unmap_icm_aux ( arbel );
2037         ufree ( arbel->icm );
2038         arbel->icm = UNULL;
2039 }
2040
2041 /***************************************************************************
2042  *
2043  * PCI interface
2044  *
2045  ***************************************************************************
2046  */
2047
2048 /**
2049  * Set up memory protection table
2050  *
2051  * @v arbel             Arbel device
2052  * @ret rc              Return status code
2053  */
2054 static int arbel_setup_mpt ( struct arbel *arbel ) {
2055         struct arbelprm_mpt mpt;
2056         uint32_t key;
2057         int rc;
2058
2059         /* Derive key */
2060         key = ( arbel->limits.reserved_mrws | ARBEL_MKEY_PREFIX );
2061         arbel->reserved_lkey = ( ( key << 8 ) | ( key >> 24 ) );
2062
2063         /* Initialise memory protection table */
2064         memset ( &mpt, 0, sizeof ( mpt ) );
2065         MLX_FILL_4 ( &mpt, 0,
2066                      r_w, 1,
2067                      pa, 1,
2068                      lr, 1,
2069                      lw, 1 );
2070         MLX_FILL_1 ( &mpt, 2, mem_key, key );
2071         MLX_FILL_1 ( &mpt, 3, pd, ARBEL_GLOBAL_PD );
2072         MLX_FILL_1 ( &mpt, 6, reg_wnd_len_h, 0xffffffffUL );
2073         MLX_FILL_1 ( &mpt, 7, reg_wnd_len_l, 0xffffffffUL );
2074         if ( ( rc = arbel_cmd_sw2hw_mpt ( arbel, arbel->limits.reserved_mrws,
2075                                           &mpt ) ) != 0 ) {
2076                 DBGC ( arbel, "Arbel %p could not set up MPT: %s\n",
2077                        arbel, strerror ( rc ) );
2078                 return rc;
2079         }
2080
2081         return 0;
2082 }
2083         
2084 /**
2085  * Probe PCI device
2086  *
2087  * @v pci               PCI device
2088  * @v id                PCI ID
2089  * @ret rc              Return status code
2090  */
2091 static int arbel_probe ( struct pci_device *pci,
2092                          const struct pci_device_id *id __unused ) {
2093         struct arbel *arbel;
2094         struct ib_device *ibdev;
2095         struct arbelprm_init_hca init_hca;
2096         int i;
2097         int rc;
2098
2099         /* Allocate Arbel device */
2100         arbel = zalloc ( sizeof ( *arbel ) );
2101         if ( ! arbel ) {
2102                 rc = -ENOMEM;
2103                 goto err_alloc_arbel;
2104         }
2105         pci_set_drvdata ( pci, arbel );
2106
2107         /* Allocate Infiniband devices */
2108         for ( i = 0 ; i < ARBEL_NUM_PORTS ; i++ ) {
2109                 ibdev = alloc_ibdev ( 0 );
2110                 if ( ! ibdev ) {
2111                         rc = -ENOMEM;
2112                         goto err_alloc_ibdev;
2113                 }
2114                 arbel->ibdev[i] = ibdev;
2115                 ibdev->op = &arbel_ib_operations;
2116                 ibdev->dev = &pci->dev;
2117                 ibdev->port = ( ARBEL_PORT_BASE + i );
2118                 ib_set_drvdata ( ibdev, arbel );
2119         }
2120
2121         /* Fix up PCI device */
2122         adjust_pci_device ( pci );
2123
2124         /* Get PCI BARs */
2125         arbel->config = ioremap ( pci_bar_start ( pci, ARBEL_PCI_CONFIG_BAR ),
2126                                   ARBEL_PCI_CONFIG_BAR_SIZE );
2127         arbel->uar = ioremap ( ( pci_bar_start ( pci, ARBEL_PCI_UAR_BAR ) +
2128                                  ARBEL_PCI_UAR_IDX * ARBEL_PCI_UAR_SIZE ),
2129                                ARBEL_PCI_UAR_SIZE );
2130
2131         /* Allocate space for mailboxes */
2132         arbel->mailbox_in = malloc_dma ( ARBEL_MBOX_SIZE, ARBEL_MBOX_ALIGN );
2133         if ( ! arbel->mailbox_in ) {
2134                 rc = -ENOMEM;
2135                 goto err_mailbox_in;
2136         }
2137         arbel->mailbox_out = malloc_dma ( ARBEL_MBOX_SIZE, ARBEL_MBOX_ALIGN );
2138         if ( ! arbel->mailbox_out ) {
2139                 rc = -ENOMEM;
2140                 goto err_mailbox_out;
2141         }
2142
2143         /* Start firmware */
2144         if ( ( rc = arbel_start_firmware ( arbel ) ) != 0 )
2145                 goto err_start_firmware;
2146
2147         /* Get device limits */
2148         if ( ( rc = arbel_get_limits ( arbel ) ) != 0 )
2149                 goto err_get_limits;
2150
2151         /* Allocate ICM */
2152         memset ( &init_hca, 0, sizeof ( init_hca ) );
2153         if ( ( rc = arbel_alloc_icm ( arbel, &init_hca ) ) != 0 )
2154                 goto err_alloc_icm;
2155
2156         /* Initialise HCA */
2157         MLX_FILL_1 ( &init_hca, 74, uar_parameters.log_max_uars, 1 );
2158         if ( ( rc = arbel_cmd_init_hca ( arbel, &init_hca ) ) != 0 ) {
2159                 DBGC ( arbel, "Arbel %p could not initialise HCA: %s\n",
2160                        arbel, strerror ( rc ) );
2161                 goto err_init_hca;
2162         }
2163
2164         /* Set up memory protection */
2165         if ( ( rc = arbel_setup_mpt ( arbel ) ) != 0 )
2166                 goto err_setup_mpt;
2167
2168         /* Set up event queue */
2169         if ( ( rc = arbel_create_eq ( arbel ) ) != 0 )
2170                 goto err_create_eq;
2171
2172         /* Register Infiniband devices */
2173         for ( i = 0 ; i < ARBEL_NUM_PORTS ; i++ ) {
2174                 if ( ( rc = register_ibdev ( arbel->ibdev[i] ) ) != 0 ) {
2175                         DBGC ( arbel, "Arbel %p could not register IB "
2176                                "device: %s\n", arbel, strerror ( rc ) );
2177                         goto err_register_ibdev;
2178                 }
2179         }
2180
2181         return 0;
2182
2183         i = ARBEL_NUM_PORTS;
2184  err_register_ibdev:
2185         for ( i-- ; i >= 0 ; i-- )
2186                 unregister_ibdev ( arbel->ibdev[i] );
2187         arbel_destroy_eq ( arbel );
2188  err_create_eq:
2189  err_setup_mpt:
2190         arbel_cmd_close_hca ( arbel );
2191  err_init_hca:
2192         arbel_free_icm ( arbel );
2193  err_alloc_icm:
2194  err_get_limits:
2195         arbel_stop_firmware ( arbel );
2196  err_start_firmware:
2197         free_dma ( arbel->mailbox_out, ARBEL_MBOX_SIZE );
2198  err_mailbox_out:
2199         free_dma ( arbel->mailbox_in, ARBEL_MBOX_SIZE );
2200  err_mailbox_in:
2201         i = ARBEL_NUM_PORTS;
2202  err_alloc_ibdev:
2203         for ( i-- ; i >= 0 ; i-- )
2204                 ibdev_put ( arbel->ibdev[i] );
2205         free ( arbel );
2206  err_alloc_arbel:
2207         return rc;
2208 }
2209
2210 /**
2211  * Remove PCI device
2212  *
2213  * @v pci               PCI device
2214  */
2215 static void arbel_remove ( struct pci_device *pci ) {
2216         struct arbel *arbel = pci_get_drvdata ( pci );
2217         int i;
2218
2219         for ( i = ( ARBEL_NUM_PORTS - 1 ) ; i >= 0 ; i-- )
2220                 unregister_ibdev ( arbel->ibdev[i] );
2221         arbel_destroy_eq ( arbel );
2222         arbel_cmd_close_hca ( arbel );
2223         arbel_free_icm ( arbel );
2224         arbel_stop_firmware ( arbel );
2225         arbel_stop_firmware ( arbel );
2226         free_dma ( arbel->mailbox_out, ARBEL_MBOX_SIZE );
2227         free_dma ( arbel->mailbox_in, ARBEL_MBOX_SIZE );
2228         for ( i = ( ARBEL_NUM_PORTS - 1 ) ; i >= 0 ; i-- )
2229                 ibdev_put ( arbel->ibdev[i] );
2230         free ( arbel );
2231 }
2232
2233 static struct pci_device_id arbel_nics[] = {
2234         PCI_ROM ( 0x15b3, 0x6282, "mt25218", "MT25218 HCA driver" ),
2235         PCI_ROM ( 0x15b3, 0x6274, "mt25204", "MT25204 HCA driver" ),
2236 };
2237
2238 struct pci_driver arbel_driver __pci_driver = {
2239         .ids = arbel_nics,
2240         .id_count = ( sizeof ( arbel_nics ) / sizeof ( arbel_nics[0] ) ),
2241         .probe = arbel_probe,
2242         .remove = arbel_remove,
2243 };