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