7d1b30acd37ee281bd341d5cd37cb33cd89d8ff4
[people/mdeck/gpxe.git] / src / drivers / infiniband / hermon.c
1 /*
2  * Copyright (C) 2008 Michael Brown <mbrown@fensystems.co.uk>.
3  * Copyright (C) 2008 Mellanox Technologies Ltd.
4  *
5  * This program is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU General Public License as
7  * published by the Free Software Foundation; either version 2 of the
8  * License, or any later version.
9  *
10  * This program is distributed in the hope that it will be useful, but
11  * WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software
17  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
18  */
19
20 #include <stdint.h>
21 #include <stdlib.h>
22 #include <stdio.h>
23 #include <string.h>
24 #include <strings.h>
25 #include <unistd.h>
26 #include <errno.h>
27 #include <byteswap.h>
28 #include <gpxe/pci.h>
29 #include <gpxe/malloc.h>
30 #include <gpxe/umalloc.h>
31 #include <gpxe/iobuf.h>
32 #include <gpxe/netdevice.h>
33 #include <gpxe/process.h>
34 #include <gpxe/infiniband.h>
35 #include "hermon.h"
36
37 /**
38  * @file
39  *
40  * Mellanox Hermon Infiniband HCA
41  *
42  */
43
44 /***************************************************************************
45  *
46  * Queue number allocation
47  *
48  ***************************************************************************
49  */
50
51 /**
52  * Allocate offsets within usage bitmask
53  *
54  * @v bits              Usage bitmask
55  * @v bits_len          Length of usage bitmask
56  * @v num_bits          Number of contiguous bits to allocate within bitmask
57  * @ret bit             First free bit within bitmask, or negative error
58  */
59 static int hermon_bitmask_alloc ( hermon_bitmask_t *bits,
60                                   unsigned int bits_len,
61                                   unsigned int num_bits ) {
62         unsigned int bit = 0;
63         hermon_bitmask_t mask = 1;
64         unsigned int found = 0;
65
66         /* Search bits for num_bits contiguous free bits */
67         while ( bit < bits_len ) {
68                 if ( ( mask & *bits ) == 0 ) {
69                         if ( ++found == num_bits )
70                                 goto found;
71                 } else {
72                         found = 0;
73                 }
74                 bit++;
75                 mask = ( mask << 1 ) | ( mask >> ( 8 * sizeof ( mask ) - 1 ) );
76                 if ( mask == 1 )
77                         bits++;
78         }
79         return -ENFILE;
80
81  found:
82         /* Mark bits as in-use */
83         do {
84                 *bits |= mask;
85                 if ( mask == 1 )
86                         bits--;
87                 mask = ( mask >> 1 ) | ( mask << ( 8 * sizeof ( mask ) - 1 ) );
88         } while ( --found );
89
90         return ( bit - num_bits + 1 );
91 }
92
93 /**
94  * Free offsets within usage bitmask
95  *
96  * @v bits              Usage bitmask
97  * @v bit               Starting bit within bitmask
98  * @v num_bits          Number of contiguous bits to free within bitmask
99  */
100 static void hermon_bitmask_free ( hermon_bitmask_t *bits,
101                                   int bit, unsigned int num_bits ) {
102         hermon_bitmask_t mask;
103
104         for ( ; num_bits ; bit++, num_bits-- ) {
105                 mask = ( 1 << ( bit % ( 8 * sizeof ( mask ) ) ) );
106                 bits[ ( bit / ( 8 * sizeof ( mask ) ) ) ] &= ~mask;
107         }
108 }
109
110 /***************************************************************************
111  *
112  * HCA commands
113  *
114  ***************************************************************************
115  */
116
117 /**
118  * Wait for Hermon command completion
119  *
120  * @v hermon            Hermon device
121  * @v hcr               HCA command registers
122  * @ret rc              Return status code
123  */
124 static int hermon_cmd_wait ( struct hermon *hermon,
125                              struct hermonprm_hca_command_register *hcr ) {
126         unsigned int wait;
127
128         for ( wait = HERMON_HCR_MAX_WAIT_MS ; wait ; wait-- ) {
129                 hcr->u.dwords[6] =
130                         readl ( hermon->config + HERMON_HCR_REG ( 6 ) );
131                 if ( ( MLX_GET ( hcr, go ) == 0 ) &&
132                      ( MLX_GET ( hcr, t ) == hermon->toggle ) )
133                         return 0;
134                 mdelay ( 1 );
135         }
136         return -EBUSY;
137 }
138
139 /**
140  * Issue HCA command
141  *
142  * @v hermon            Hermon device
143  * @v command           Command opcode, flags and input/output lengths
144  * @v op_mod            Opcode modifier (0 if no modifier applicable)
145  * @v in                Input parameters
146  * @v in_mod            Input modifier (0 if no modifier applicable)
147  * @v out               Output parameters
148  * @ret rc              Return status code
149  */
150 static int hermon_cmd ( struct hermon *hermon, unsigned long command,
151                         unsigned int op_mod, const void *in,
152                         unsigned int in_mod, void *out ) {
153         struct hermonprm_hca_command_register hcr;
154         unsigned int opcode = HERMON_HCR_OPCODE ( command );
155         size_t in_len = HERMON_HCR_IN_LEN ( command );
156         size_t out_len = HERMON_HCR_OUT_LEN ( command );
157         void *in_buffer;
158         void *out_buffer;
159         unsigned int status;
160         unsigned int i;
161         int rc;
162
163         assert ( in_len <= HERMON_MBOX_SIZE );
164         assert ( out_len <= HERMON_MBOX_SIZE );
165
166         DBGC2 ( hermon, "Hermon %p command %02x in %zx%s out %zx%s\n",
167                 hermon, opcode, in_len,
168                 ( ( command & HERMON_HCR_IN_MBOX ) ? "(mbox)" : "" ), out_len,
169                 ( ( command & HERMON_HCR_OUT_MBOX ) ? "(mbox)" : "" ) );
170
171         /* Check that HCR is free */
172         if ( ( rc = hermon_cmd_wait ( hermon, &hcr ) ) != 0 ) {
173                 DBGC ( hermon, "Hermon %p command interface locked\n",
174                        hermon );
175                 return rc;
176         }
177
178         /* Flip HCR toggle */
179         hermon->toggle = ( 1 - hermon->toggle );
180
181         /* Prepare HCR */
182         memset ( &hcr, 0, sizeof ( hcr ) );
183         in_buffer = &hcr.u.dwords[0];
184         if ( in_len && ( command & HERMON_HCR_IN_MBOX ) ) {
185                 in_buffer = hermon->mailbox_in;
186                 MLX_FILL_1 ( &hcr, 1, in_param_l, virt_to_bus ( in_buffer ) );
187         }
188         memcpy ( in_buffer, in, in_len );
189         MLX_FILL_1 ( &hcr, 2, input_modifier, in_mod );
190         out_buffer = &hcr.u.dwords[3];
191         if ( out_len && ( command & HERMON_HCR_OUT_MBOX ) ) {
192                 out_buffer = hermon->mailbox_out;
193                 MLX_FILL_1 ( &hcr, 4, out_param_l,
194                              virt_to_bus ( out_buffer ) );
195         }
196         MLX_FILL_4 ( &hcr, 6,
197                      opcode, opcode,
198                      opcode_modifier, op_mod,
199                      go, 1,
200                      t, hermon->toggle );
201         DBGC ( hermon, "Hermon %p issuing command:\n", hermon );
202         DBGC_HDA ( hermon, virt_to_phys ( hermon->config + HERMON_HCR_BASE ),
203                    &hcr, sizeof ( hcr ) );
204         if ( in_len && ( command & HERMON_HCR_IN_MBOX ) ) {
205                 DBGC2 ( hermon, "Input mailbox:\n" );
206                 DBGC2_HDA ( hermon, virt_to_phys ( in_buffer ), in_buffer,
207                             ( ( in_len < 512 ) ? in_len : 512 ) );
208         }
209
210         /* Issue command */
211         for ( i = 0 ; i < ( sizeof ( hcr ) / sizeof ( hcr.u.dwords[0] ) ) ;
212               i++ ) {
213                 writel ( hcr.u.dwords[i],
214                          hermon->config + HERMON_HCR_REG ( i ) );
215                 barrier();
216         }
217
218         /* Wait for command completion */
219         if ( ( rc = hermon_cmd_wait ( hermon, &hcr ) ) != 0 ) {
220                 DBGC ( hermon, "Hermon %p timed out waiting for command:\n",
221                        hermon );
222                 DBGC_HDA ( hermon,
223                            virt_to_phys ( hermon->config + HERMON_HCR_BASE ),
224                            &hcr, sizeof ( hcr ) );
225                 return rc;
226         }
227
228         /* Check command status */
229         status = MLX_GET ( &hcr, status );
230         if ( status != 0 ) {
231                 DBGC ( hermon, "Hermon %p command failed with status %02x:\n",
232                        hermon, status );
233                 DBGC_HDA ( hermon,
234                            virt_to_phys ( hermon->config + HERMON_HCR_BASE ),
235                            &hcr, sizeof ( hcr ) );
236                 return -EIO;
237         }
238
239         /* Read output parameters, if any */
240         hcr.u.dwords[3] = readl ( hermon->config + HERMON_HCR_REG ( 3 ) );
241         hcr.u.dwords[4] = readl ( hermon->config + HERMON_HCR_REG ( 4 ) );
242         memcpy ( out, out_buffer, out_len );
243         if ( out_len ) {
244                 DBGC2 ( hermon, "Output%s:\n",
245                         ( command & HERMON_HCR_OUT_MBOX ) ? " mailbox" : "" );
246                 DBGC2_HDA ( hermon, virt_to_phys ( out_buffer ), out_buffer,
247                             ( ( out_len < 512 ) ? out_len : 512 ) );
248         }
249
250         return 0;
251 }
252
253 static inline int
254 hermon_cmd_query_dev_cap ( struct hermon *hermon,
255                            struct hermonprm_query_dev_cap *dev_cap ) {
256         return hermon_cmd ( hermon,
257                             HERMON_HCR_OUT_CMD ( HERMON_HCR_QUERY_DEV_CAP,
258                                                  1, sizeof ( *dev_cap ) ),
259                             0, NULL, 0, dev_cap );
260 }
261
262 static inline int
263 hermon_cmd_query_fw ( struct hermon *hermon, struct hermonprm_query_fw *fw ) {
264         return hermon_cmd ( hermon,
265                             HERMON_HCR_OUT_CMD ( HERMON_HCR_QUERY_FW,
266                                                  1, sizeof ( *fw ) ),
267                             0, NULL, 0, fw );
268 }
269
270 static inline int
271 hermon_cmd_init_hca ( struct hermon *hermon,
272                       const struct hermonprm_init_hca *init_hca ) {
273         return hermon_cmd ( hermon,
274                             HERMON_HCR_IN_CMD ( HERMON_HCR_INIT_HCA,
275                                                 1, sizeof ( *init_hca ) ),
276                             0, init_hca, 0, NULL );
277 }
278
279 static inline int
280 hermon_cmd_close_hca ( struct hermon *hermon ) {
281         return hermon_cmd ( hermon,
282                             HERMON_HCR_VOID_CMD ( HERMON_HCR_CLOSE_HCA ),
283                             0, NULL, 0, NULL );
284 }
285
286 static inline int
287 hermon_cmd_init_port ( struct hermon *hermon, unsigned int port,
288                        const struct hermonprm_init_port *init_port ) {
289         return hermon_cmd ( hermon,
290                             HERMON_HCR_IN_CMD ( HERMON_HCR_INIT_PORT,
291                                                 1, sizeof ( *init_port ) ),
292                             0, init_port, port, NULL );
293 }
294
295 static inline int
296 hermon_cmd_close_port ( struct hermon *hermon, unsigned int port ) {
297         return hermon_cmd ( hermon,
298                             HERMON_HCR_VOID_CMD ( HERMON_HCR_CLOSE_PORT ),
299                             0, NULL, port, NULL );
300 }
301
302 static inline int
303 hermon_cmd_sw2hw_mpt ( struct hermon *hermon, unsigned int index,
304                        const struct hermonprm_mpt *mpt ) {
305         return hermon_cmd ( hermon,
306                             HERMON_HCR_IN_CMD ( HERMON_HCR_SW2HW_MPT,
307                                                 1, sizeof ( *mpt ) ),
308                             0, mpt, index, NULL );
309 }
310
311 static inline int
312 hermon_cmd_write_mtt ( struct hermon *hermon,
313                        const struct hermonprm_write_mtt *write_mtt ) {
314         return hermon_cmd ( hermon,
315                             HERMON_HCR_IN_CMD ( HERMON_HCR_WRITE_MTT,
316                                                 1, sizeof ( *write_mtt ) ),
317                             0, write_mtt, 1, NULL );
318 }
319
320 static inline int
321 hermon_cmd_map_eq ( struct hermon *hermon, unsigned long index_map,
322                     const struct hermonprm_event_mask *mask ) {
323         return hermon_cmd ( hermon,
324                             HERMON_HCR_IN_CMD ( HERMON_HCR_MAP_EQ,
325                                                 0, sizeof ( *mask ) ),
326                             0, mask, index_map, NULL );
327 }
328
329 static inline int
330 hermon_cmd_sw2hw_eq ( struct hermon *hermon, unsigned int index,
331                       const struct hermonprm_eqc *eqctx ) {
332         return hermon_cmd ( hermon,
333                             HERMON_HCR_IN_CMD ( HERMON_HCR_SW2HW_EQ,
334                                                 1, sizeof ( *eqctx ) ),
335                             0, eqctx, index, NULL );
336 }
337
338 static inline int
339 hermon_cmd_hw2sw_eq ( struct hermon *hermon, unsigned int index,
340                       struct hermonprm_eqc *eqctx ) {
341         return hermon_cmd ( hermon,
342                             HERMON_HCR_OUT_CMD ( HERMON_HCR_HW2SW_EQ,
343                                                  1, sizeof ( *eqctx ) ),
344                             1, NULL, index, eqctx );
345 }
346
347 static inline int
348 hermon_cmd_query_eq ( struct hermon *hermon, unsigned int index,
349                       struct hermonprm_eqc *eqctx ) {
350         return hermon_cmd ( hermon,
351                             HERMON_HCR_OUT_CMD ( HERMON_HCR_QUERY_EQ,
352                                                  1, sizeof ( *eqctx ) ),
353                             0, NULL, index, eqctx );
354 }
355
356 static inline int
357 hermon_cmd_sw2hw_cq ( struct hermon *hermon, unsigned long cqn,
358                       const struct hermonprm_completion_queue_context *cqctx ){
359         return hermon_cmd ( hermon,
360                             HERMON_HCR_IN_CMD ( HERMON_HCR_SW2HW_CQ,
361                                                 1, sizeof ( *cqctx ) ),
362                             0, cqctx, cqn, NULL );
363 }
364
365 static inline int
366 hermon_cmd_hw2sw_cq ( struct hermon *hermon, unsigned long cqn,
367                       struct hermonprm_completion_queue_context *cqctx) {
368         return hermon_cmd ( hermon,
369                             HERMON_HCR_OUT_CMD ( HERMON_HCR_HW2SW_CQ,
370                                                  1, sizeof ( *cqctx ) ),
371                             0, NULL, cqn, cqctx );
372 }
373
374 static inline int
375 hermon_cmd_rst2init_qp ( struct hermon *hermon, unsigned long qpn,
376                          const struct hermonprm_qp_ee_state_transitions *ctx ){
377         return hermon_cmd ( hermon,
378                             HERMON_HCR_IN_CMD ( HERMON_HCR_RST2INIT_QP,
379                                                 1, sizeof ( *ctx ) ),
380                             0, ctx, qpn, NULL );
381 }
382
383 static inline int
384 hermon_cmd_init2rtr_qp ( struct hermon *hermon, unsigned long qpn,
385                          const struct hermonprm_qp_ee_state_transitions *ctx ){
386         return hermon_cmd ( hermon,
387                             HERMON_HCR_IN_CMD ( HERMON_HCR_INIT2RTR_QP,
388                                                 1, sizeof ( *ctx ) ),
389                             0, ctx, qpn, NULL );
390 }
391
392 static inline int
393 hermon_cmd_rtr2rts_qp ( struct hermon *hermon, unsigned long qpn,
394                         const struct hermonprm_qp_ee_state_transitions *ctx ) {
395         return hermon_cmd ( hermon,
396                             HERMON_HCR_IN_CMD ( HERMON_HCR_RTR2RTS_QP,
397                                                 1, sizeof ( *ctx ) ),
398                             0, ctx, qpn, NULL );
399 }
400
401 static inline int
402 hermon_cmd_rts2rts_qp ( struct hermon *hermon, unsigned long qpn,
403                         const struct hermonprm_qp_ee_state_transitions *ctx ) {
404         return hermon_cmd ( hermon,
405                             HERMON_HCR_IN_CMD ( HERMON_HCR_RTS2RTS_QP,
406                                                 1, sizeof ( *ctx ) ),
407                             0, ctx, qpn, NULL );
408 }
409
410 static inline int
411 hermon_cmd_2rst_qp ( struct hermon *hermon, unsigned long qpn ) {
412         return hermon_cmd ( hermon,
413                             HERMON_HCR_VOID_CMD ( HERMON_HCR_2RST_QP ),
414                             0x03, NULL, qpn, NULL );
415 }
416
417 static inline int
418 hermon_cmd_mad_ifc ( struct hermon *hermon, unsigned int port,
419                      union hermonprm_mad *mad ) {
420         return hermon_cmd ( hermon,
421                             HERMON_HCR_INOUT_CMD ( HERMON_HCR_MAD_IFC,
422                                                    1, sizeof ( *mad ),
423                                                    1, sizeof ( *mad ) ),
424                             0x03, mad, port, mad );
425 }
426
427 static inline int
428 hermon_cmd_read_mcg ( struct hermon *hermon, unsigned int index,
429                       struct hermonprm_mcg_entry *mcg ) {
430         return hermon_cmd ( hermon,
431                             HERMON_HCR_OUT_CMD ( HERMON_HCR_READ_MCG,
432                                                  1, sizeof ( *mcg ) ),
433                             0, NULL, index, mcg );
434 }
435
436 static inline int
437 hermon_cmd_write_mcg ( struct hermon *hermon, unsigned int index,
438                        const struct hermonprm_mcg_entry *mcg ) {
439         return hermon_cmd ( hermon,
440                             HERMON_HCR_IN_CMD ( HERMON_HCR_WRITE_MCG,
441                                                 1, sizeof ( *mcg ) ),
442                             0, mcg, index, NULL );
443 }
444
445 static inline int
446 hermon_cmd_mgid_hash ( struct hermon *hermon, const struct ib_gid *gid,
447                        struct hermonprm_mgm_hash *hash ) {
448         return hermon_cmd ( hermon,
449                             HERMON_HCR_INOUT_CMD ( HERMON_HCR_MGID_HASH,
450                                                    1, sizeof ( *gid ),
451                                                    0, sizeof ( *hash ) ),
452                             0, gid, 0, hash );
453 }
454
455 static inline int
456 hermon_cmd_run_fw ( struct hermon *hermon ) {
457         return hermon_cmd ( hermon,
458                             HERMON_HCR_VOID_CMD ( HERMON_HCR_RUN_FW ),
459                             0, NULL, 0, NULL );
460 }
461
462 static inline int
463 hermon_cmd_unmap_icm ( struct hermon *hermon, unsigned int page_count,
464                        const struct hermonprm_scalar_parameter *offset ) {
465         return hermon_cmd ( hermon,
466                             HERMON_HCR_IN_CMD ( HERMON_HCR_UNMAP_ICM,
467                                                 0, sizeof ( *offset ) ),
468                             0, offset, page_count, NULL );
469 }
470
471 static inline int
472 hermon_cmd_map_icm ( struct hermon *hermon,
473                      const struct hermonprm_virtual_physical_mapping *map ) {
474         return hermon_cmd ( hermon,
475                             HERMON_HCR_IN_CMD ( HERMON_HCR_MAP_ICM,
476                                                 1, sizeof ( *map ) ),
477                             0, map, 1, NULL );
478 }
479
480 static inline int
481 hermon_cmd_unmap_icm_aux ( struct hermon *hermon ) {
482         return hermon_cmd ( hermon,
483                             HERMON_HCR_VOID_CMD ( HERMON_HCR_UNMAP_ICM_AUX ),
484                             0, NULL, 0, NULL );
485 }
486
487 static inline int
488 hermon_cmd_map_icm_aux ( struct hermon *hermon,
489                        const struct hermonprm_virtual_physical_mapping *map ) {
490         return hermon_cmd ( hermon,
491                             HERMON_HCR_IN_CMD ( HERMON_HCR_MAP_ICM_AUX,
492                                                 1, sizeof ( *map ) ),
493                             0, map, 1, NULL );
494 }
495
496 static inline int
497 hermon_cmd_set_icm_size ( struct hermon *hermon,
498                           const struct hermonprm_scalar_parameter *icm_size,
499                           struct hermonprm_scalar_parameter *icm_aux_size ) {
500         return hermon_cmd ( hermon,
501                             HERMON_HCR_INOUT_CMD ( HERMON_HCR_SET_ICM_SIZE,
502                                                    0, sizeof ( *icm_size ),
503                                                    0, sizeof (*icm_aux_size) ),
504                             0, icm_size, 0, icm_aux_size );
505 }
506
507 static inline int
508 hermon_cmd_unmap_fa ( struct hermon *hermon ) {
509         return hermon_cmd ( hermon,
510                             HERMON_HCR_VOID_CMD ( HERMON_HCR_UNMAP_FA ),
511                             0, NULL, 0, NULL );
512 }
513
514 static inline int
515 hermon_cmd_map_fa ( struct hermon *hermon,
516                     const struct hermonprm_virtual_physical_mapping *map ) {
517         return hermon_cmd ( hermon,
518                             HERMON_HCR_IN_CMD ( HERMON_HCR_MAP_FA,
519                                                 1, sizeof ( *map ) ),
520                             0, map, 1, NULL );
521 }
522
523 /***************************************************************************
524  *
525  * Memory translation table operations
526  *
527  ***************************************************************************
528  */
529
530 /**
531  * Allocate MTT entries
532  *
533  * @v hermon            Hermon device
534  * @v memory            Memory to map into MTT
535  * @v len               Length of memory to map
536  * @v mtt               MTT descriptor to fill in
537  * @ret rc              Return status code
538  */
539 static int hermon_alloc_mtt ( struct hermon *hermon,
540                               const void *memory, size_t len,
541                               struct hermon_mtt *mtt ) {
542         struct hermonprm_write_mtt write_mtt;
543         physaddr_t start;
544         unsigned int page_offset;
545         unsigned int num_pages;
546         int mtt_offset;
547         unsigned int mtt_base_addr;
548         unsigned int i;
549         int rc;
550
551         /* Find available MTT entries */
552         start = virt_to_phys ( memory );
553         page_offset = ( start & ( HERMON_PAGE_SIZE - 1 ) );
554         start -= page_offset;
555         len += page_offset;
556         num_pages = ( ( len + HERMON_PAGE_SIZE - 1 ) / HERMON_PAGE_SIZE );
557         mtt_offset = hermon_bitmask_alloc ( hermon->mtt_inuse, HERMON_MAX_MTTS,
558                                             num_pages );
559         if ( mtt_offset < 0 ) {
560                 DBGC ( hermon, "Hermon %p could not allocate %d MTT entries\n",
561                        hermon, num_pages );
562                 rc = mtt_offset;
563                 goto err_mtt_offset;
564         }
565         mtt_base_addr = ( ( hermon->cap.reserved_mtts + mtt_offset ) *
566                           hermon->cap.mtt_entry_size );
567
568         /* Fill in MTT structure */
569         mtt->mtt_offset = mtt_offset;
570         mtt->num_pages = num_pages;
571         mtt->mtt_base_addr = mtt_base_addr;
572         mtt->page_offset = page_offset;
573
574         /* Construct and issue WRITE_MTT commands */
575         for ( i = 0 ; i < num_pages ; i++ ) {
576                 memset ( &write_mtt, 0, sizeof ( write_mtt ) );
577                 MLX_FILL_1 ( &write_mtt.mtt_base_addr, 1,
578                              value, mtt_base_addr );
579                 MLX_FILL_2 ( &write_mtt.mtt, 1,
580                              p, 1,
581                              ptag_l, ( start >> 3 ) );
582                 if ( ( rc = hermon_cmd_write_mtt ( hermon,
583                                                    &write_mtt ) ) != 0 ) {
584                         DBGC ( hermon, "Hermon %p could not write MTT at %x\n",
585                                hermon, mtt_base_addr );
586                         goto err_write_mtt;
587                 }
588                 start += HERMON_PAGE_SIZE;
589                 mtt_base_addr += hermon->cap.mtt_entry_size;
590         }
591
592         return 0;
593
594  err_write_mtt:
595         hermon_bitmask_free ( hermon->mtt_inuse, mtt_offset, num_pages );
596  err_mtt_offset:
597         return rc;
598 }
599
600 /**
601  * Free MTT entries
602  *
603  * @v hermon            Hermon device
604  * @v mtt               MTT descriptor
605  */
606 static void hermon_free_mtt ( struct hermon *hermon,
607                               struct hermon_mtt *mtt ) {
608         hermon_bitmask_free ( hermon->mtt_inuse, mtt->mtt_offset,
609                               mtt->num_pages );
610 }
611
612 /***************************************************************************
613  *
614  * Completion queue operations
615  *
616  ***************************************************************************
617  */
618
619 /**
620  * Create completion queue
621  *
622  * @v ibdev             Infiniband device
623  * @v cq                Completion queue
624  * @ret rc              Return status code
625  */
626 static int hermon_create_cq ( struct ib_device *ibdev,
627                               struct ib_completion_queue *cq ) {
628         struct hermon *hermon = ib_get_drvdata ( ibdev );
629         struct hermon_completion_queue *hermon_cq;
630         struct hermonprm_completion_queue_context cqctx;
631         int cqn_offset;
632         unsigned int i;
633         int rc;
634
635         /* Find a free completion queue number */
636         cqn_offset = hermon_bitmask_alloc ( hermon->cq_inuse,
637                                             HERMON_MAX_CQS, 1 );
638         if ( cqn_offset < 0 ) {
639                 DBGC ( hermon, "Hermon %p out of completion queues\n",
640                        hermon );
641                 rc = cqn_offset;
642                 goto err_cqn_offset;
643         }
644         cq->cqn = ( hermon->cap.reserved_cqs + cqn_offset );
645
646         /* Allocate control structures */
647         hermon_cq = zalloc ( sizeof ( *hermon_cq ) );
648         if ( ! hermon_cq ) {
649                 rc = -ENOMEM;
650                 goto err_hermon_cq;
651         }
652
653         /* Allocate completion queue itself */
654         hermon_cq->cqe_size = ( cq->num_cqes * sizeof ( hermon_cq->cqe[0] ) );
655         hermon_cq->cqe = malloc_dma ( hermon_cq->cqe_size,
656                                       sizeof ( hermon_cq->cqe[0] ) );
657         if ( ! hermon_cq->cqe ) {
658                 rc = -ENOMEM;
659                 goto err_cqe;
660         }
661         memset ( hermon_cq->cqe, 0, hermon_cq->cqe_size );
662         for ( i = 0 ; i < cq->num_cqes ; i++ ) {
663                 MLX_FILL_1 ( &hermon_cq->cqe[i].normal, 7, owner, 1 );
664         }
665         barrier();
666
667         /* Allocate MTT entries */
668         if ( ( rc = hermon_alloc_mtt ( hermon, hermon_cq->cqe,
669                                        hermon_cq->cqe_size,
670                                        &hermon_cq->mtt ) ) != 0 )
671                 goto err_alloc_mtt;
672
673         /* Hand queue over to hardware */
674         memset ( &cqctx, 0, sizeof ( cqctx ) );
675         MLX_FILL_1 ( &cqctx, 0, st, 0xa /* "Event fired" */ );
676         MLX_FILL_1 ( &cqctx, 2,
677                      page_offset, ( hermon_cq->mtt.page_offset >> 5 ) );
678         MLX_FILL_2 ( &cqctx, 3,
679                      usr_page, HERMON_UAR_NON_EQ_PAGE,
680                      log_cq_size, fls ( cq->num_cqes - 1 ) );
681         MLX_FILL_1 ( &cqctx, 7, mtt_base_addr_l,
682                      ( hermon_cq->mtt.mtt_base_addr >> 3 ) );
683         MLX_FILL_1 ( &cqctx, 15, db_record_addr_l,
684                      ( virt_to_phys ( &hermon_cq->doorbell ) >> 3 ) );
685         if ( ( rc = hermon_cmd_sw2hw_cq ( hermon, cq->cqn, &cqctx ) ) != 0 ) {
686                 DBGC ( hermon, "Hermon %p SW2HW_CQ failed: %s\n",
687                        hermon, strerror ( rc ) );
688                 goto err_sw2hw_cq;
689         }
690
691         DBGC ( hermon, "Hermon %p CQN %#lx ring at [%p,%p)\n",
692                hermon, cq->cqn, hermon_cq->cqe,
693                ( ( ( void * ) hermon_cq->cqe ) + hermon_cq->cqe_size ) );
694         ib_cq_set_drvdata ( cq, hermon_cq );
695         return 0;
696
697  err_sw2hw_cq:
698         hermon_free_mtt ( hermon, &hermon_cq->mtt );
699  err_alloc_mtt:
700         free_dma ( hermon_cq->cqe, hermon_cq->cqe_size );
701  err_cqe:
702         free ( hermon_cq );
703  err_hermon_cq:
704         hermon_bitmask_free ( hermon->cq_inuse, cqn_offset, 1 );
705  err_cqn_offset:
706         return rc;
707 }
708
709 /**
710  * Destroy completion queue
711  *
712  * @v ibdev             Infiniband device
713  * @v cq                Completion queue
714  */
715 static void hermon_destroy_cq ( struct ib_device *ibdev,
716                                 struct ib_completion_queue *cq ) {
717         struct hermon *hermon = ib_get_drvdata ( ibdev );
718         struct hermon_completion_queue *hermon_cq = ib_cq_get_drvdata ( cq );
719         struct hermonprm_completion_queue_context cqctx;
720         int cqn_offset;
721         int rc;
722
723         /* Take ownership back from hardware */
724         if ( ( rc = hermon_cmd_hw2sw_cq ( hermon, cq->cqn, &cqctx ) ) != 0 ) {
725                 DBGC ( hermon, "Hermon %p FATAL HW2SW_CQ failed on CQN %#lx: "
726                        "%s\n", hermon, cq->cqn, strerror ( rc ) );
727                 /* Leak memory and return; at least we avoid corruption */
728                 return;
729         }
730
731         /* Free MTT entries */
732         hermon_free_mtt ( hermon, &hermon_cq->mtt );
733
734         /* Free memory */
735         free_dma ( hermon_cq->cqe, hermon_cq->cqe_size );
736         free ( hermon_cq );
737
738         /* Mark queue number as free */
739         cqn_offset = ( cq->cqn - hermon->cap.reserved_cqs );
740         hermon_bitmask_free ( hermon->cq_inuse, cqn_offset, 1 );
741
742         ib_cq_set_drvdata ( cq, NULL );
743 }
744
745 /***************************************************************************
746  *
747  * Queue pair operations
748  *
749  ***************************************************************************
750  */
751
752 /**
753  * Create queue pair
754  *
755  * @v ibdev             Infiniband device
756  * @v qp                Queue pair
757  * @ret rc              Return status code
758  */
759 static int hermon_create_qp ( struct ib_device *ibdev,
760                               struct ib_queue_pair *qp ) {
761         struct hermon *hermon = ib_get_drvdata ( ibdev );
762         struct hermon_queue_pair *hermon_qp;
763         struct hermonprm_qp_ee_state_transitions qpctx;
764         int qpn_offset;
765         int rc;
766
767         /* Find a free queue pair number */
768         qpn_offset = hermon_bitmask_alloc ( hermon->qp_inuse,
769                                             HERMON_MAX_QPS, 1 );
770         if ( qpn_offset < 0 ) {
771                 DBGC ( hermon, "Hermon %p out of queue pairs\n", hermon );
772                 rc = qpn_offset;
773                 goto err_qpn_offset;
774         }
775         qp->qpn = ( HERMON_QPN_BASE + hermon->cap.reserved_qps +
776                     qpn_offset );
777
778         /* Allocate control structures */
779         hermon_qp = zalloc ( sizeof ( *hermon_qp ) );
780         if ( ! hermon_qp ) {
781                 rc = -ENOMEM;
782                 goto err_hermon_qp;
783         }
784
785         /* Calculate doorbell address */
786         hermon_qp->send.doorbell =
787                 ( hermon->uar + HERMON_UAR_NON_EQ_PAGE * HERMON_PAGE_SIZE +
788                   HERMON_DB_POST_SND_OFFSET );
789
790         /* Allocate work queue buffer */
791         hermon_qp->send.num_wqes = ( qp->send.num_wqes /* headroom */ + 1 +
792                                 ( 2048 / sizeof ( hermon_qp->send.wqe[0] ) ) );
793         hermon_qp->send.num_wqes =
794                 ( 1 << fls ( hermon_qp->send.num_wqes - 1 ) ); /* round up */
795         hermon_qp->send.wqe_size = ( hermon_qp->send.num_wqes *
796                                      sizeof ( hermon_qp->send.wqe[0] ) );
797         hermon_qp->recv.wqe_size = ( qp->recv.num_wqes *
798                                      sizeof ( hermon_qp->recv.wqe[0] ) );
799         hermon_qp->wqe_size = ( hermon_qp->send.wqe_size +
800                                 hermon_qp->recv.wqe_size );
801         hermon_qp->wqe = malloc_dma ( hermon_qp->wqe_size,
802                                       sizeof ( hermon_qp->send.wqe[0] ) );
803         if ( ! hermon_qp->wqe ) {
804                 rc = -ENOMEM;
805                 goto err_alloc_wqe;
806         }
807         hermon_qp->send.wqe = hermon_qp->wqe;
808         memset ( hermon_qp->send.wqe, 0xff, hermon_qp->send.wqe_size );
809         hermon_qp->recv.wqe = ( hermon_qp->wqe + hermon_qp->send.wqe_size );
810         memset ( hermon_qp->recv.wqe, 0, hermon_qp->recv.wqe_size );
811
812         /* Allocate MTT entries */
813         if ( ( rc = hermon_alloc_mtt ( hermon, hermon_qp->wqe,
814                                        hermon_qp->wqe_size,
815                                        &hermon_qp->mtt ) ) != 0 ) {
816                 goto err_alloc_mtt;
817         }
818
819         /* Transition queue to INIT state */
820         memset ( &qpctx, 0, sizeof ( qpctx ) );
821         MLX_FILL_2 ( &qpctx, 2,
822                      qpc_eec_data.pm_state, 0x03 /* Always 0x03 for UD */,
823                      qpc_eec_data.st, HERMON_ST_UD );
824         MLX_FILL_1 ( &qpctx, 3, qpc_eec_data.pd, HERMON_GLOBAL_PD );
825         MLX_FILL_4 ( &qpctx, 4,
826                      qpc_eec_data.log_rq_size, fls ( qp->recv.num_wqes - 1 ),
827                      qpc_eec_data.log_rq_stride,
828                      ( fls ( sizeof ( hermon_qp->recv.wqe[0] ) - 1 ) - 4 ),
829                      qpc_eec_data.log_sq_size,
830                      fls ( hermon_qp->send.num_wqes - 1 ),
831                      qpc_eec_data.log_sq_stride,
832                      ( fls ( sizeof ( hermon_qp->send.wqe[0] ) - 1 ) - 4 ) );
833         MLX_FILL_1 ( &qpctx, 5,
834                      qpc_eec_data.usr_page, HERMON_UAR_NON_EQ_PAGE );
835         MLX_FILL_1 ( &qpctx, 33, qpc_eec_data.cqn_snd, qp->send.cq->cqn );
836         MLX_FILL_1 ( &qpctx, 38, qpc_eec_data.page_offset,
837                      ( hermon_qp->mtt.page_offset >> 6 ) );
838         MLX_FILL_1 ( &qpctx, 41, qpc_eec_data.cqn_rcv, qp->recv.cq->cqn );
839         MLX_FILL_1 ( &qpctx, 43, qpc_eec_data.db_record_addr_l,
840                      ( virt_to_phys ( &hermon_qp->recv.doorbell ) >> 2 ) );
841         MLX_FILL_1 ( &qpctx, 44, qpc_eec_data.q_key, qp->qkey );
842         MLX_FILL_1 ( &qpctx, 53, qpc_eec_data.mtt_base_addr_l,
843                      ( hermon_qp->mtt.mtt_base_addr >> 3 ) );
844         if ( ( rc = hermon_cmd_rst2init_qp ( hermon, qp->qpn,
845                                              &qpctx ) ) != 0 ) {
846                 DBGC ( hermon, "Hermon %p RST2INIT_QP failed: %s\n",
847                        hermon, strerror ( rc ) );
848                 goto err_rst2init_qp;
849         }
850
851         /* Transition queue to RTR state */
852         memset ( &qpctx, 0, sizeof ( qpctx ) );
853         MLX_FILL_2 ( &qpctx, 4,
854                      qpc_eec_data.mtu, HERMON_MTU_2048,
855                      qpc_eec_data.msg_max, 11 /* 2^11 = 2048 */ );
856         MLX_FILL_1 ( &qpctx, 16,
857                      qpc_eec_data.primary_address_path.sched_queue,
858                      ( 0x83 /* default policy */ |
859                        ( ( ibdev->port - 1 ) << 6 ) ) );
860         if ( ( rc = hermon_cmd_init2rtr_qp ( hermon, qp->qpn,
861                                              &qpctx ) ) != 0 ) {
862                 DBGC ( hermon, "Hermon %p INIT2RTR_QP failed: %s\n",
863                        hermon, strerror ( rc ) );
864                 goto err_init2rtr_qp;
865         }
866         memset ( &qpctx, 0, sizeof ( qpctx ) );
867         if ( ( rc = hermon_cmd_rtr2rts_qp ( hermon, qp->qpn, &qpctx ) ) != 0 ){
868                 DBGC ( hermon, "Hermon %p RTR2RTS_QP failed: %s\n",
869                        hermon, strerror ( rc ) );
870                 goto err_rtr2rts_qp;
871         }
872
873         DBGC ( hermon, "Hermon %p QPN %#lx send ring at [%p,%p)\n",
874                hermon, qp->qpn, hermon_qp->send.wqe,
875                ( ((void *)hermon_qp->send.wqe ) + hermon_qp->send.wqe_size ) );
876         DBGC ( hermon, "Hermon %p QPN %#lx receive ring at [%p,%p)\n",
877                hermon, qp->qpn, hermon_qp->recv.wqe,
878                ( ((void *)hermon_qp->recv.wqe ) + hermon_qp->recv.wqe_size ) );
879         ib_qp_set_drvdata ( qp, hermon_qp );
880         return 0;
881
882  err_rtr2rts_qp:
883  err_init2rtr_qp:
884         hermon_cmd_2rst_qp ( hermon, qp->qpn );
885  err_rst2init_qp:
886         hermon_free_mtt ( hermon, &hermon_qp->mtt );
887  err_alloc_mtt:
888         free_dma ( hermon_qp->wqe, hermon_qp->wqe_size );
889  err_alloc_wqe:
890         free ( hermon_qp );
891  err_hermon_qp:
892         hermon_bitmask_free ( hermon->qp_inuse, qpn_offset, 1 );
893  err_qpn_offset:
894         return rc;
895 }
896
897 /**
898  * Modify queue pair
899  *
900  * @v ibdev             Infiniband device
901  * @v qp                Queue pair
902  * @v mod_list          Modification list
903  * @ret rc              Return status code
904  */
905 static int hermon_modify_qp ( struct ib_device *ibdev,
906                               struct ib_queue_pair *qp,
907                               unsigned long mod_list ) {
908         struct hermon *hermon = ib_get_drvdata ( ibdev );
909         struct hermonprm_qp_ee_state_transitions qpctx;
910         unsigned long optparammask = 0;
911         int rc;
912
913         /* Construct optparammask */
914         if ( mod_list & IB_MODIFY_QKEY )
915                 optparammask |= HERMON_QP_OPT_PARAM_QKEY;
916
917         /* Issue RTS2RTS_QP */
918         memset ( &qpctx, 0, sizeof ( qpctx ) );
919         MLX_FILL_1 ( &qpctx, 0, opt_param_mask, optparammask );
920         MLX_FILL_1 ( &qpctx, 44, qpc_eec_data.q_key, qp->qkey );
921         if ( ( rc = hermon_cmd_rts2rts_qp ( hermon, qp->qpn, &qpctx ) ) != 0 ){
922                 DBGC ( hermon, "Hermon %p RTS2RTS_QP failed: %s\n",
923                        hermon, strerror ( rc ) );
924                 return rc;
925         }
926
927         return 0;
928 }
929
930 /**
931  * Destroy queue pair
932  *
933  * @v ibdev             Infiniband device
934  * @v qp                Queue pair
935  */
936 static void hermon_destroy_qp ( struct ib_device *ibdev,
937                                 struct ib_queue_pair *qp ) {
938         struct hermon *hermon = ib_get_drvdata ( ibdev );
939         struct hermon_queue_pair *hermon_qp = ib_qp_get_drvdata ( qp );
940         int qpn_offset;
941         int rc;
942
943         /* Take ownership back from hardware */
944         if ( ( rc = hermon_cmd_2rst_qp ( hermon, qp->qpn ) ) != 0 ) {
945                 DBGC ( hermon, "Hermon %p FATAL 2RST_QP failed on QPN %#lx: "
946                        "%s\n", hermon, qp->qpn, strerror ( rc ) );
947                 /* Leak memory and return; at least we avoid corruption */
948                 return;
949         }
950
951         /* Free MTT entries */
952         hermon_free_mtt ( hermon, &hermon_qp->mtt );
953
954         /* Free memory */
955         free_dma ( hermon_qp->wqe, hermon_qp->wqe_size );
956         free ( hermon_qp );
957
958         /* Mark queue number as free */
959         qpn_offset = ( qp->qpn - HERMON_QPN_BASE -
960                        hermon->cap.reserved_qps );
961         hermon_bitmask_free ( hermon->qp_inuse, qpn_offset, 1 );
962
963         ib_qp_set_drvdata ( qp, NULL );
964 }
965
966 /***************************************************************************
967  *
968  * Work request operations
969  *
970  ***************************************************************************
971  */
972
973 /** GID used for GID-less send work queue entries */
974 static const struct ib_gid hermon_no_gid = {
975         { { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 } }
976 };
977
978 /**
979  * Post send work queue entry
980  *
981  * @v ibdev             Infiniband device
982  * @v qp                Queue pair
983  * @v av                Address vector
984  * @v iobuf             I/O buffer
985  * @ret rc              Return status code
986  */
987 static int hermon_post_send ( struct ib_device *ibdev,
988                               struct ib_queue_pair *qp,
989                               struct ib_address_vector *av,
990                               struct io_buffer *iobuf ) {
991         struct hermon *hermon = ib_get_drvdata ( ibdev );
992         struct hermon_queue_pair *hermon_qp = ib_qp_get_drvdata ( qp );
993         struct ib_work_queue *wq = &qp->send;
994         struct hermon_send_work_queue *hermon_send_wq = &hermon_qp->send;
995         struct hermonprm_ud_send_wqe *wqe;
996         const struct ib_gid *gid;
997         union hermonprm_doorbell_register db_reg;
998         unsigned int wqe_idx_mask;
999
1000         /* Allocate work queue entry */
1001         wqe_idx_mask = ( wq->num_wqes - 1 );
1002         if ( wq->iobufs[wq->next_idx & wqe_idx_mask] ) {
1003                 DBGC ( hermon, "Hermon %p send queue full", hermon );
1004                 return -ENOBUFS;
1005         }
1006         wq->iobufs[wq->next_idx & wqe_idx_mask] = iobuf;
1007         wqe = &hermon_send_wq->wqe[ wq->next_idx &
1008                                     ( hermon_send_wq->num_wqes - 1 ) ].ud;
1009
1010         /* Construct work queue entry */
1011         memset ( ( ( ( void * ) wqe ) + 4 /* avoid ctrl.owner */ ), 0,
1012                    ( sizeof ( *wqe ) - 4 ) );
1013         MLX_FILL_1 ( &wqe->ctrl, 1, ds, ( sizeof ( *wqe ) / 16 ) );
1014         MLX_FILL_1 ( &wqe->ctrl, 2, c, 0x03 /* generate completion */ );
1015         MLX_FILL_2 ( &wqe->ud, 0,
1016                      ud_address_vector.pd, HERMON_GLOBAL_PD,
1017                      ud_address_vector.port_number, ibdev->port );
1018         MLX_FILL_2 ( &wqe->ud, 1,
1019                      ud_address_vector.rlid, av->dlid,
1020                      ud_address_vector.g, av->gid_present );
1021         MLX_FILL_1 ( &wqe->ud, 2,
1022                      ud_address_vector.max_stat_rate,
1023                      ( ( ( av->rate < 2 ) || ( av->rate > 10 ) ) ?
1024                        8 : ( av->rate + 5 ) ) );
1025         MLX_FILL_1 ( &wqe->ud, 3, ud_address_vector.sl, av->sl );
1026         gid = ( av->gid_present ? &av->gid : &hermon_no_gid );
1027         memcpy ( &wqe->ud.u.dwords[4], gid, sizeof ( *gid ) );
1028         MLX_FILL_1 ( &wqe->ud, 8, destination_qp, av->dest_qp );
1029         MLX_FILL_1 ( &wqe->ud, 9, q_key, av->qkey );
1030         MLX_FILL_1 ( &wqe->data[0], 0, byte_count, iob_len ( iobuf ) );
1031         MLX_FILL_1 ( &wqe->data[0], 1, l_key, hermon->reserved_lkey );
1032         MLX_FILL_1 ( &wqe->data[0], 3,
1033                      local_address_l, virt_to_bus ( iobuf->data ) );
1034         barrier();
1035         MLX_FILL_2 ( &wqe->ctrl, 0,
1036                      opcode, HERMON_OPCODE_SEND,
1037                      owner,
1038                      ( ( wq->next_idx & hermon_send_wq->num_wqes ) ? 1 : 0 ) );
1039         DBGCP ( hermon, "Hermon %p posting send WQE:\n", hermon );
1040         DBGCP_HD ( hermon, wqe, sizeof ( *wqe ) );
1041         barrier();
1042
1043         /* Ring doorbell register */
1044         MLX_FILL_1 ( &db_reg.send, 0, qn, qp->qpn );
1045         DBGCP ( hermon, "Ringing doorbell %08lx with %08lx\n",
1046                 virt_to_phys ( hermon_send_wq->doorbell ), db_reg.dword[0] );
1047         writel ( db_reg.dword[0], ( hermon_send_wq->doorbell ) );
1048
1049         /* Update work queue's index */
1050         wq->next_idx++;
1051
1052         return 0;
1053 }
1054
1055 /**
1056  * Post receive work queue entry
1057  *
1058  * @v ibdev             Infiniband device
1059  * @v qp                Queue pair
1060  * @v iobuf             I/O buffer
1061  * @ret rc              Return status code
1062  */
1063 static int hermon_post_recv ( struct ib_device *ibdev,
1064                               struct ib_queue_pair *qp,
1065                               struct io_buffer *iobuf ) {
1066         struct hermon *hermon = ib_get_drvdata ( ibdev );
1067         struct hermon_queue_pair *hermon_qp = ib_qp_get_drvdata ( qp );
1068         struct ib_work_queue *wq = &qp->recv;
1069         struct hermon_recv_work_queue *hermon_recv_wq = &hermon_qp->recv;
1070         struct hermonprm_recv_wqe *wqe;
1071         unsigned int wqe_idx_mask;
1072
1073         /* Allocate work queue entry */
1074         wqe_idx_mask = ( wq->num_wqes - 1 );
1075         if ( wq->iobufs[wq->next_idx & wqe_idx_mask] ) {
1076                 DBGC ( hermon, "Hermon %p receive queue full", hermon );
1077                 return -ENOBUFS;
1078         }
1079         wq->iobufs[wq->next_idx & wqe_idx_mask] = iobuf;
1080         wqe = &hermon_recv_wq->wqe[wq->next_idx & wqe_idx_mask].recv;
1081
1082         /* Construct work queue entry */
1083         MLX_FILL_1 ( &wqe->data[0], 0, byte_count, iob_tailroom ( iobuf ) );
1084         MLX_FILL_1 ( &wqe->data[0], 1, l_key, hermon->reserved_lkey );
1085         MLX_FILL_1 ( &wqe->data[0], 3,
1086                      local_address_l, virt_to_bus ( iobuf->data ) );
1087
1088         /* Update work queue's index */
1089         wq->next_idx++;
1090
1091         /* Update doorbell record */
1092         barrier();
1093         MLX_FILL_1 ( &hermon_recv_wq->doorbell, 0, receive_wqe_counter,
1094                      ( wq->next_idx & 0xffff ) );
1095
1096         return 0;
1097 }
1098
1099 /**
1100  * Handle completion
1101  *
1102  * @v ibdev             Infiniband device
1103  * @v cq                Completion queue
1104  * @v cqe               Hardware completion queue entry
1105  * @v complete_send     Send completion handler
1106  * @v complete_recv     Receive completion handler
1107  * @ret rc              Return status code
1108  */
1109 static int hermon_complete ( struct ib_device *ibdev,
1110                              struct ib_completion_queue *cq,
1111                              union hermonprm_completion_entry *cqe,
1112                              ib_completer_t complete_send,
1113                              ib_completer_t complete_recv ) {
1114         struct hermon *hermon = ib_get_drvdata ( ibdev );
1115         struct ib_completion completion;
1116         struct ib_work_queue *wq;
1117         struct ib_queue_pair *qp;
1118         struct hermon_queue_pair *hermon_qp;
1119         struct io_buffer *iobuf;
1120         ib_completer_t complete;
1121         unsigned int opcode;
1122         unsigned long qpn;
1123         int is_send;
1124         unsigned int wqe_idx;
1125         int rc = 0;
1126
1127         /* Parse completion */
1128         memset ( &completion, 0, sizeof ( completion ) );
1129         qpn = MLX_GET ( &cqe->normal, qpn );
1130         is_send = MLX_GET ( &cqe->normal, s_r );
1131         opcode = MLX_GET ( &cqe->normal, opcode );
1132         if ( opcode >= HERMON_OPCODE_RECV_ERROR ) {
1133                 /* "s" field is not valid for error opcodes */
1134                 is_send = ( opcode == HERMON_OPCODE_SEND_ERROR );
1135                 completion.syndrome = MLX_GET ( &cqe->error, syndrome );
1136                 DBGC ( hermon, "Hermon %p CQN %lx syndrome %x vendor %lx\n",
1137                        hermon, cq->cqn, completion.syndrome,
1138                        MLX_GET ( &cqe->error, vendor_error_syndrome ) );
1139                 rc = -EIO;
1140                 /* Don't return immediately; propagate error to completer */
1141         }
1142
1143         /* Identify work queue */
1144         wq = ib_find_wq ( cq, qpn, is_send );
1145         if ( ! wq ) {
1146                 DBGC ( hermon, "Hermon %p CQN %lx unknown %s QPN %lx\n",
1147                        hermon, cq->cqn, ( is_send ? "send" : "recv" ), qpn );
1148                 return -EIO;
1149         }
1150         qp = wq->qp;
1151         hermon_qp = ib_qp_get_drvdata ( qp );
1152
1153         /* Identify I/O buffer */
1154         wqe_idx = ( MLX_GET ( &cqe->normal, wqe_counter ) &
1155                     ( wq->num_wqes - 1 ) );
1156         iobuf = wq->iobufs[wqe_idx];
1157         if ( ! iobuf ) {
1158                 DBGC ( hermon, "Hermon %p CQN %lx QPN %lx empty WQE %x\n",
1159                        hermon, cq->cqn, qpn, wqe_idx );
1160                 return -EIO;
1161         }
1162         wq->iobufs[wqe_idx] = NULL;
1163
1164         /* Fill in length for received packets */
1165         if ( ! is_send ) {
1166                 completion.len = MLX_GET ( &cqe->normal, byte_cnt );
1167                 if ( completion.len > iob_tailroom ( iobuf ) ) {
1168                         DBGC ( hermon, "Hermon %p CQN %lx QPN %lx IDX %x "
1169                                "overlength received packet length %zd\n",
1170                                hermon, cq->cqn, qpn, wqe_idx, completion.len );
1171                         return -EIO;
1172                 }
1173         }
1174
1175         /* Pass off to caller's completion handler */
1176         complete = ( is_send ? complete_send : complete_recv );
1177         complete ( ibdev, qp, &completion, iobuf );
1178
1179         return rc;
1180 }
1181
1182 /**
1183  * Poll completion queue
1184  *
1185  * @v ibdev             Infiniband device
1186  * @v cq                Completion queue
1187  * @v complete_send     Send completion handler
1188  * @v complete_recv     Receive completion handler
1189  */
1190 static void hermon_poll_cq ( struct ib_device *ibdev,
1191                              struct ib_completion_queue *cq,
1192                              ib_completer_t complete_send,
1193                              ib_completer_t complete_recv ) {
1194         struct hermon *hermon = ib_get_drvdata ( ibdev );
1195         struct hermon_completion_queue *hermon_cq = ib_cq_get_drvdata ( cq );
1196         union hermonprm_completion_entry *cqe;
1197         unsigned int cqe_idx_mask;
1198         int rc;
1199
1200         while ( 1 ) {
1201                 /* Look for completion entry */
1202                 cqe_idx_mask = ( cq->num_cqes - 1 );
1203                 cqe = &hermon_cq->cqe[cq->next_idx & cqe_idx_mask];
1204                 if ( MLX_GET ( &cqe->normal, owner ) ^
1205                      ( ( cq->next_idx & cq->num_cqes ) ? 1 : 0 ) ) {
1206                         /* Entry still owned by hardware; end of poll */
1207                         break;
1208                 }
1209                 DBGCP ( hermon, "Hermon %p completion:\n", hermon );
1210                 DBGCP_HD ( hermon, cqe, sizeof ( *cqe ) );
1211
1212                 /* Handle completion */
1213                 if ( ( rc = hermon_complete ( ibdev, cq, cqe, complete_send,
1214                                               complete_recv ) ) != 0 ) {
1215                         DBGC ( hermon, "Hermon %p failed to complete: %s\n",
1216                                hermon, strerror ( rc ) );
1217                         DBGC_HD ( hermon, cqe, sizeof ( *cqe ) );
1218                 }
1219
1220                 /* Update completion queue's index */
1221                 cq->next_idx++;
1222
1223                 /* Update doorbell record */
1224                 MLX_FILL_1 ( &hermon_cq->doorbell, 0, update_ci,
1225                              ( cq->next_idx & 0x00ffffffUL ) );
1226         }
1227 }
1228
1229 /***************************************************************************
1230  *
1231  * Infiniband link-layer operations
1232  *
1233  ***************************************************************************
1234  */
1235
1236 /**
1237  * Initialise Infiniband link
1238  *
1239  * @v ibdev             Infiniband device
1240  * @ret rc              Return status code
1241  */
1242 static int hermon_open ( struct ib_device *ibdev ) {
1243         struct hermon *hermon = ib_get_drvdata ( ibdev );
1244         struct hermonprm_init_port init_port;
1245         int rc;
1246
1247         memset ( &init_port, 0, sizeof ( init_port ) );
1248         MLX_FILL_2 ( &init_port, 0,
1249                      port_width_cap, 3,
1250                      vl_cap, 1 );
1251         MLX_FILL_2 ( &init_port, 1,
1252                      mtu, HERMON_MTU_2048,
1253                      max_gid, 1 );
1254         MLX_FILL_1 ( &init_port, 2, max_pkey, 64 );
1255         if ( ( rc = hermon_cmd_init_port ( hermon, ibdev->port,
1256                                            &init_port ) ) != 0 ) {
1257                 DBGC ( hermon, "Hermon %p could not intialise port: %s\n",
1258                        hermon, strerror ( rc ) );
1259                 return rc;
1260         }
1261
1262         return 0;
1263 }
1264
1265 /**
1266  * Close Infiniband link
1267  *
1268  * @v ibdev             Infiniband device
1269  */
1270 static void hermon_close ( struct ib_device *ibdev ) {
1271         struct hermon *hermon = ib_get_drvdata ( ibdev );
1272         int rc;
1273
1274         if ( ( rc = hermon_cmd_close_port ( hermon, ibdev->port ) ) != 0 ) {
1275                 DBGC ( hermon, "Hermon %p could not close port: %s\n",
1276                        hermon, strerror ( rc ) );
1277                 /* Nothing we can do about this */
1278         }
1279 }
1280
1281 /***************************************************************************
1282  *
1283  * Multicast group operations
1284  *
1285  ***************************************************************************
1286  */
1287
1288 /**
1289  * Attach to multicast group
1290  *
1291  * @v ibdev             Infiniband device
1292  * @v qp                Queue pair
1293  * @v gid               Multicast GID
1294  * @ret rc              Return status code
1295  */
1296 static int hermon_mcast_attach ( struct ib_device *ibdev,
1297                                  struct ib_queue_pair *qp,
1298                                  struct ib_gid *gid ) {
1299         struct hermon *hermon = ib_get_drvdata ( ibdev );
1300         struct hermonprm_mgm_hash hash;
1301         struct hermonprm_mcg_entry mcg;
1302         unsigned int index;
1303         int rc;
1304
1305         /* Generate hash table index */
1306         if ( ( rc = hermon_cmd_mgid_hash ( hermon, gid, &hash ) ) != 0 ) {
1307                 DBGC ( hermon, "Hermon %p could not hash GID: %s\n",
1308                        hermon, strerror ( rc ) );
1309                 return rc;
1310         }
1311         index = MLX_GET ( &hash, hash );
1312
1313         /* Check for existing hash table entry */
1314         if ( ( rc = hermon_cmd_read_mcg ( hermon, index, &mcg ) ) != 0 ) {
1315                 DBGC ( hermon, "Hermon %p could not read MCG %#x: %s\n",
1316                        hermon, index, strerror ( rc ) );
1317                 return rc;
1318         }
1319         if ( MLX_GET ( &mcg, hdr.members_count ) != 0 ) {
1320                 /* FIXME: this implementation allows only a single QP
1321                  * per multicast group, and doesn't handle hash
1322                  * collisions.  Sufficient for IPoIB but may need to
1323                  * be extended in future.
1324                  */
1325                 DBGC ( hermon, "Hermon %p MGID index %#x already in use\n",
1326                        hermon, index );
1327                 return -EBUSY;
1328         }
1329
1330         /* Update hash table entry */
1331         MLX_FILL_1 ( &mcg, 1, hdr.members_count, 1 );
1332         MLX_FILL_1 ( &mcg, 8, qp[0].qpn, qp->qpn );
1333         memcpy ( &mcg.u.dwords[4], gid, sizeof ( *gid ) );
1334         if ( ( rc = hermon_cmd_write_mcg ( hermon, index, &mcg ) ) != 0 ) {
1335                 DBGC ( hermon, "Hermon %p could not write MCG %#x: %s\n",
1336                        hermon, index, strerror ( rc ) );
1337                 return rc;
1338         }
1339
1340         return 0;
1341 }
1342
1343 /**
1344  * Detach from multicast group
1345  *
1346  * @v ibdev             Infiniband device
1347  * @v qp                Queue pair
1348  * @v gid               Multicast GID
1349  */
1350 static void hermon_mcast_detach ( struct ib_device *ibdev,
1351                                   struct ib_queue_pair *qp __unused,
1352                                   struct ib_gid *gid ) {
1353         struct hermon *hermon = ib_get_drvdata ( ibdev );
1354         struct hermonprm_mgm_hash hash;
1355         struct hermonprm_mcg_entry mcg;
1356         unsigned int index;
1357         int rc;
1358
1359         /* Generate hash table index */
1360         if ( ( rc = hermon_cmd_mgid_hash ( hermon, gid, &hash ) ) != 0 ) {
1361                 DBGC ( hermon, "Hermon %p could not hash GID: %s\n",
1362                        hermon, strerror ( rc ) );
1363                 return;
1364         }
1365         index = MLX_GET ( &hash, hash );
1366
1367         /* Clear hash table entry */
1368         memset ( &mcg, 0, sizeof ( mcg ) );
1369         if ( ( rc = hermon_cmd_write_mcg ( hermon, index, &mcg ) ) != 0 ) {
1370                 DBGC ( hermon, "Hermon %p could not write MCG %#x: %s\n",
1371                        hermon, index, strerror ( rc ) );
1372                 return;
1373         }
1374 }
1375
1376 /***************************************************************************
1377  *
1378  * MAD operations
1379  *
1380  ***************************************************************************
1381  */
1382
1383 /**
1384  * Issue management datagram
1385  *
1386  * @v ibdev             Infiniband device
1387  * @v mad               Management datagram
1388  * @v len               Length of management datagram
1389  * @ret rc              Return status code
1390  */
1391 static int hermon_mad ( struct ib_device *ibdev, struct ib_mad_hdr *mad,
1392                         size_t len ) {
1393         struct hermon *hermon = ib_get_drvdata ( ibdev );
1394         union hermonprm_mad mad_ifc;
1395         int rc;
1396
1397         /* Copy in request packet */
1398         memset ( &mad_ifc, 0, sizeof ( mad_ifc ) );
1399         assert ( len <= sizeof ( mad_ifc.mad ) );
1400         memcpy ( &mad_ifc.mad, mad, len );
1401
1402         /* Issue MAD */
1403         if ( ( rc = hermon_cmd_mad_ifc ( hermon, ibdev->port,
1404                                          &mad_ifc ) ) != 0 ) {
1405                 DBGC ( hermon, "Hermon %p could not issue MAD IFC: %s\n",
1406                        hermon, strerror ( rc ) );
1407                 return rc;
1408         }
1409
1410         /* Copy out reply packet */
1411         memcpy ( mad, &mad_ifc.mad, len );
1412
1413         if ( mad->status != 0 ) {
1414                 DBGC ( hermon, "Hermon %p MAD IFC status %04x\n",
1415                        hermon, ntohs ( mad->status ) );
1416                 return -EIO;
1417         }
1418         return 0;
1419 }
1420
1421 /** Hermon Infiniband operations */
1422 static struct ib_device_operations hermon_ib_operations = {
1423         .create_cq      = hermon_create_cq,
1424         .destroy_cq     = hermon_destroy_cq,
1425         .create_qp      = hermon_create_qp,
1426         .modify_qp      = hermon_modify_qp,
1427         .destroy_qp     = hermon_destroy_qp,
1428         .post_send      = hermon_post_send,
1429         .post_recv      = hermon_post_recv,
1430         .poll_cq        = hermon_poll_cq,
1431         .open           = hermon_open,
1432         .close          = hermon_close,
1433         .mcast_attach   = hermon_mcast_attach,
1434         .mcast_detach   = hermon_mcast_detach,
1435         .mad            = hermon_mad,
1436 };
1437
1438 /***************************************************************************
1439  *
1440  * Event queues
1441  *
1442  ***************************************************************************
1443  */
1444
1445 /**
1446  * Create event queue
1447  *
1448  * @v hermon            Hermon device
1449  * @ret rc              Return status code
1450  */
1451 static int hermon_create_eq ( struct hermon *hermon ) {
1452         struct hermon_event_queue *hermon_eq = &hermon->eq;
1453         struct hermonprm_eqc eqctx;
1454         struct hermonprm_event_mask mask;
1455         unsigned int i;
1456         int rc;
1457
1458         /* Select event queue number */
1459         hermon_eq->eqn = ( 4 * hermon->cap.reserved_uars );
1460         if ( hermon_eq->eqn < hermon->cap.reserved_eqs )
1461                 hermon_eq->eqn = hermon->cap.reserved_eqs;
1462
1463         /* Calculate doorbell address */
1464         hermon_eq->doorbell =
1465                 ( hermon->uar + HERMON_DB_EQ_OFFSET ( hermon_eq->eqn ) );
1466
1467         /* Allocate event queue itself */
1468         hermon_eq->eqe_size =
1469                 ( HERMON_NUM_EQES * sizeof ( hermon_eq->eqe[0] ) );
1470         hermon_eq->eqe = malloc_dma ( hermon_eq->eqe_size,
1471                                       sizeof ( hermon_eq->eqe[0] ) );
1472         if ( ! hermon_eq->eqe ) {
1473                 rc = -ENOMEM;
1474                 goto err_eqe;
1475         }
1476         memset ( hermon_eq->eqe, 0, hermon_eq->eqe_size );
1477         for ( i = 0 ; i < HERMON_NUM_EQES ; i++ ) {
1478                 MLX_FILL_1 ( &hermon_eq->eqe[i].generic, 7, owner, 1 );
1479         }
1480         barrier();
1481
1482         /* Allocate MTT entries */
1483         if ( ( rc = hermon_alloc_mtt ( hermon, hermon_eq->eqe,
1484                                        hermon_eq->eqe_size,
1485                                        &hermon_eq->mtt ) ) != 0 )
1486                 goto err_alloc_mtt;
1487
1488         /* Hand queue over to hardware */
1489         memset ( &eqctx, 0, sizeof ( eqctx ) );
1490         MLX_FILL_1 ( &eqctx, 0, st, 0xa /* "Fired" */ );
1491         MLX_FILL_1 ( &eqctx, 2,
1492                      page_offset, ( hermon_eq->mtt.page_offset >> 5 ) );
1493         MLX_FILL_1 ( &eqctx, 3, log_eq_size, fls ( HERMON_NUM_EQES - 1 ) );
1494         MLX_FILL_1 ( &eqctx, 7, mtt_base_addr_l,
1495                      ( hermon_eq->mtt.mtt_base_addr >> 3 ) );
1496         if ( ( rc = hermon_cmd_sw2hw_eq ( hermon, hermon_eq->eqn,
1497                                           &eqctx ) ) != 0 ) {
1498                 DBGC ( hermon, "Hermon %p SW2HW_EQ failed: %s\n",
1499                        hermon, strerror ( rc ) );
1500                 goto err_sw2hw_eq;
1501         }
1502
1503         /* Map events to this event queue */
1504         memset ( &mask, 0, sizeof ( mask ) );
1505         MLX_FILL_1 ( &mask, 1, port_state_change, 1 );
1506         if ( ( rc = hermon_cmd_map_eq ( hermon,
1507                                         ( HERMON_MAP_EQ | hermon_eq->eqn ),
1508                                         &mask ) ) != 0 ) {
1509                 DBGC ( hermon, "Hermon %p MAP_EQ failed: %s\n",
1510                        hermon, strerror ( rc )  );
1511                 goto err_map_eq;
1512         }
1513
1514         DBGC ( hermon, "Hermon %p EQN %#lx ring at [%p,%p])\n",
1515                hermon, hermon_eq->eqn, hermon_eq->eqe,
1516                ( ( ( void * ) hermon_eq->eqe ) + hermon_eq->eqe_size ) );
1517         return 0;
1518
1519  err_map_eq:
1520         hermon_cmd_hw2sw_eq ( hermon, hermon_eq->eqn, &eqctx );
1521  err_sw2hw_eq:
1522         hermon_free_mtt ( hermon, &hermon_eq->mtt );
1523  err_alloc_mtt:
1524         free_dma ( hermon_eq->eqe, hermon_eq->eqe_size );
1525  err_eqe:
1526         memset ( hermon_eq, 0, sizeof ( *hermon_eq ) );
1527         return rc;
1528 }
1529
1530 /**
1531  * Destroy event queue
1532  *
1533  * @v hermon            Hermon device
1534  */
1535 static void hermon_destroy_eq ( struct hermon *hermon ) {
1536         struct hermon_event_queue *hermon_eq = &hermon->eq;
1537         struct hermonprm_eqc eqctx;
1538         struct hermonprm_event_mask mask;
1539         int rc;
1540
1541         /* Unmap events from event queue */
1542         memset ( &mask, 0, sizeof ( mask ) );
1543         MLX_FILL_1 ( &mask, 1, port_state_change, 1 );
1544         if ( ( rc = hermon_cmd_map_eq ( hermon,
1545                                         ( HERMON_UNMAP_EQ | hermon_eq->eqn ),
1546                                         &mask ) ) != 0 ) {
1547                 DBGC ( hermon, "Hermon %p FATAL MAP_EQ failed to unmap: %s\n",
1548                        hermon, strerror ( rc ) );
1549                 /* Continue; HCA may die but system should survive */
1550         }
1551
1552         /* Take ownership back from hardware */
1553         if ( ( rc = hermon_cmd_hw2sw_eq ( hermon, hermon_eq->eqn,
1554                                           &eqctx ) ) != 0 ) {
1555                 DBGC ( hermon, "Hermon %p FATAL HW2SW_EQ failed: %s\n",
1556                        hermon, strerror ( rc ) );
1557                 /* Leak memory and return; at least we avoid corruption */
1558                 return;
1559         }
1560
1561         /* Free MTT entries */
1562         hermon_free_mtt ( hermon, &hermon_eq->mtt );
1563
1564         /* Free memory */
1565         free_dma ( hermon_eq->eqe, hermon_eq->eqe_size );
1566         memset ( hermon_eq, 0, sizeof ( *hermon_eq ) );
1567 }
1568
1569 /**
1570  * Handle port state event
1571  *
1572  * @v hermon            Hermon device
1573  * @v eqe               Port state change event queue entry
1574  */
1575 static void hermon_event_port_state_change ( struct hermon *hermon,
1576                                              union hermonprm_event_entry *eqe){
1577         unsigned int port;
1578         int link_up;
1579
1580         /* Get port and link status */
1581         port = ( MLX_GET ( &eqe->port_state_change, data.p ) - 1 );
1582         link_up = ( MLX_GET ( &eqe->generic, event_sub_type ) & 0x04 );
1583         DBGC ( hermon, "Hermon %p port %d link %s\n", hermon, ( port + 1 ),
1584                ( link_up ? "up" : "down" ) );
1585
1586         /* Sanity check */
1587         if ( port >= HERMON_NUM_PORTS ) {
1588                 DBGC ( hermon, "Hermon %p port %d does not exist!\n",
1589                        hermon, ( port + 1 ) );
1590                 return;
1591         }
1592
1593         /* Notify Infiniband core of link state change */
1594         ib_link_state_changed ( hermon->ibdev[port] );
1595 }
1596
1597 /**
1598  * Poll event queue
1599  *
1600  * @v hermon            Hermon device
1601  */
1602 static void hermon_poll_eq ( struct hermon *hermon ) {
1603         struct hermon_event_queue *hermon_eq = &hermon->eq;
1604         union hermonprm_event_entry *eqe;
1605         union hermonprm_doorbell_register db_reg;
1606         unsigned int eqe_idx_mask;
1607         unsigned int event_type;
1608
1609         while ( 1 ) {
1610                 /* Look for event entry */
1611                 eqe_idx_mask = ( HERMON_NUM_EQES - 1 );
1612                 eqe = &hermon_eq->eqe[hermon_eq->next_idx & eqe_idx_mask];
1613                 if ( MLX_GET ( &eqe->generic, owner ) ^
1614                      ( ( hermon_eq->next_idx & HERMON_NUM_EQES ) ? 1 : 0 ) ) {
1615                         /* Entry still owned by hardware; end of poll */
1616                         break;
1617                 }
1618                 DBGCP ( hermon, "Hermon %p event:\n", hermon );
1619                 DBGCP_HD ( hermon, eqe, sizeof ( *eqe ) );
1620
1621                 /* Handle event */
1622                 event_type = MLX_GET ( &eqe->generic, event_type );
1623                 switch ( event_type ) {
1624                 case HERMON_EV_PORT_STATE_CHANGE:
1625                         hermon_event_port_state_change ( hermon, eqe );
1626                         break;
1627                 default:
1628                         DBGC ( hermon, "Hermon %p unrecognised event type "
1629                                "%#x:\n", hermon, event_type );
1630                         DBGC_HD ( hermon, eqe, sizeof ( *eqe ) );
1631                         break;
1632                 }
1633
1634                 /* Update event queue's index */
1635                 hermon_eq->next_idx++;
1636
1637                 /* Ring doorbell */
1638                 MLX_FILL_1 ( &db_reg.event, 0,
1639                              ci, ( hermon_eq->next_idx & 0x00ffffffUL ) );
1640                 DBGCP ( hermon, "Ringing doorbell %08lx with %08lx\n",
1641                         virt_to_phys ( hermon_eq->doorbell ),
1642                         db_reg.dword[0] );
1643                 writel ( db_reg.dword[0], hermon_eq->doorbell );
1644         }
1645 }
1646
1647 /**
1648  * Event queue poll processor
1649  *
1650  * @v process           Hermon event queue process
1651  */
1652 static void hermon_step ( struct process *process ) {
1653         struct hermon *hermon =
1654                 container_of ( process, struct hermon, event_process );
1655
1656         hermon_poll_eq ( hermon );
1657 }
1658
1659 /***************************************************************************
1660  *
1661  * Firmware control
1662  *
1663  ***************************************************************************
1664  */
1665
1666 /**
1667  * Start firmware running
1668  *
1669  * @v hermon            Hermon device
1670  * @ret rc              Return status code
1671  */
1672 static int hermon_start_firmware ( struct hermon *hermon ) {
1673         struct hermonprm_query_fw fw;
1674         struct hermonprm_virtual_physical_mapping map_fa;
1675         unsigned int fw_pages;
1676         unsigned int log2_fw_pages;
1677         size_t fw_size;
1678         physaddr_t fw_base;
1679         int rc;
1680
1681         /* Get firmware parameters */
1682         if ( ( rc = hermon_cmd_query_fw ( hermon, &fw ) ) != 0 ) {
1683                 DBGC ( hermon, "Hermon %p could not query firmware: %s\n",
1684                        hermon, strerror ( rc ) );
1685                 goto err_query_fw;
1686         }
1687         DBGC ( hermon, "Hermon %p firmware version %ld.%ld.%ld\n", hermon,
1688                MLX_GET ( &fw, fw_rev_major ), MLX_GET ( &fw, fw_rev_minor ),
1689                MLX_GET ( &fw, fw_rev_subminor ) );
1690         fw_pages = MLX_GET ( &fw, fw_pages );
1691         log2_fw_pages = fls ( fw_pages - 1 );
1692         fw_pages = ( 1 << log2_fw_pages );
1693         DBGC ( hermon, "Hermon %p requires %d kB for firmware\n",
1694                hermon, ( fw_pages * 4 ) );
1695
1696         /* Allocate firmware pages and map firmware area */
1697         fw_size = ( fw_pages * HERMON_PAGE_SIZE );
1698         hermon->firmware_area = umalloc ( fw_size );
1699         if ( ! hermon->firmware_area ) {
1700                 rc = -ENOMEM;
1701                 goto err_alloc_fa;
1702         }
1703         fw_base = ( user_to_phys ( hermon->firmware_area, fw_size ) &
1704                     ~( fw_size - 1 ) );
1705         DBGC ( hermon, "Hermon %p firmware area at physical [%lx,%lx)\n",
1706                hermon, fw_base, ( fw_base + fw_size ) );
1707         memset ( &map_fa, 0, sizeof ( map_fa ) );
1708         MLX_FILL_2 ( &map_fa, 3,
1709                      log2size, log2_fw_pages,
1710                      pa_l, ( fw_base >> 12 ) );
1711         if ( ( rc = hermon_cmd_map_fa ( hermon, &map_fa ) ) != 0 ) {
1712                 DBGC ( hermon, "Hermon %p could not map firmware: %s\n",
1713                        hermon, strerror ( rc ) );
1714                 goto err_map_fa;
1715         }
1716
1717         /* Start firmware */
1718         if ( ( rc = hermon_cmd_run_fw ( hermon ) ) != 0 ) {
1719                 DBGC ( hermon, "Hermon %p could not run firmware: %s\n",
1720                        hermon, strerror ( rc ) );
1721                 goto err_run_fw;
1722         }
1723
1724         DBGC ( hermon, "Hermon %p firmware started\n", hermon );
1725         return 0;
1726
1727  err_run_fw:
1728         hermon_cmd_unmap_fa ( hermon );
1729  err_map_fa:
1730         ufree ( hermon->firmware_area );
1731         hermon->firmware_area = UNULL;
1732  err_alloc_fa:
1733  err_query_fw:
1734         return rc;
1735 }
1736
1737 /**
1738  * Stop firmware running
1739  *
1740  * @v hermon            Hermon device
1741  */
1742 static void hermon_stop_firmware ( struct hermon *hermon ) {
1743         int rc;
1744
1745         if ( ( rc = hermon_cmd_unmap_fa ( hermon ) ) != 0 ) {
1746                 DBGC ( hermon, "Hermon %p FATAL could not stop firmware: %s\n",
1747                        hermon, strerror ( rc ) );
1748                 /* Leak memory and return; at least we avoid corruption */
1749                 return;
1750         }
1751         ufree ( hermon->firmware_area );
1752         hermon->firmware_area = UNULL;
1753 }
1754
1755 /***************************************************************************
1756  *
1757  * Infinihost Context Memory management
1758  *
1759  ***************************************************************************
1760  */
1761
1762 /**
1763  * Get device limits
1764  *
1765  * @v hermon            Hermon device
1766  * @ret rc              Return status code
1767  */
1768 static int hermon_get_cap ( struct hermon *hermon ) {
1769         struct hermonprm_query_dev_cap dev_cap;
1770         int rc;
1771
1772         if ( ( rc = hermon_cmd_query_dev_cap ( hermon, &dev_cap ) ) != 0 ) {
1773                 DBGC ( hermon, "Hermon %p could not get device limits: %s\n",
1774                        hermon, strerror ( rc ) );
1775                 return rc;
1776         }
1777
1778         hermon->cap.cmpt_entry_size = MLX_GET ( &dev_cap, c_mpt_entry_sz );
1779         hermon->cap.reserved_qps =
1780                 ( 1 << MLX_GET ( &dev_cap, log2_rsvd_qps ) );
1781         hermon->cap.qpc_entry_size = MLX_GET ( &dev_cap, qpc_entry_sz );
1782         hermon->cap.altc_entry_size = MLX_GET ( &dev_cap, altc_entry_sz );
1783         hermon->cap.auxc_entry_size = MLX_GET ( &dev_cap, aux_entry_sz );
1784         hermon->cap.reserved_srqs =
1785                 ( 1 << MLX_GET ( &dev_cap, log2_rsvd_srqs ) );
1786         hermon->cap.srqc_entry_size = MLX_GET ( &dev_cap, srq_entry_sz );
1787         hermon->cap.reserved_cqs =
1788                 ( 1 << MLX_GET ( &dev_cap, log2_rsvd_cqs ) );
1789         hermon->cap.cqc_entry_size = MLX_GET ( &dev_cap, cqc_entry_sz );
1790         hermon->cap.reserved_eqs = MLX_GET ( &dev_cap, num_rsvd_eqs );
1791         hermon->cap.eqc_entry_size = MLX_GET ( &dev_cap, eqc_entry_sz );
1792         hermon->cap.reserved_mtts =
1793                 ( 1 << MLX_GET ( &dev_cap, log2_rsvd_mtts ) );
1794         hermon->cap.mtt_entry_size = MLX_GET ( &dev_cap, mtt_entry_sz );
1795         hermon->cap.reserved_mrws =
1796                 ( 1 << MLX_GET ( &dev_cap, log2_rsvd_mrws ) );
1797         hermon->cap.dmpt_entry_size = MLX_GET ( &dev_cap, d_mpt_entry_sz );
1798         hermon->cap.reserved_uars = MLX_GET ( &dev_cap, num_rsvd_uars );
1799
1800         return 0;
1801 }
1802
1803 /**
1804  * Get ICM usage
1805  *
1806  * @v log_num_entries   Log2 of the number of entries
1807  * @v entry_size        Entry size
1808  * @ret usage           Usage size in ICM
1809  */
1810 static size_t icm_usage ( unsigned int log_num_entries, size_t entry_size ) {
1811         size_t usage;
1812
1813         usage = ( ( 1 << log_num_entries ) * entry_size );
1814         usage = ( ( usage + HERMON_PAGE_SIZE - 1 ) &
1815                   ~( HERMON_PAGE_SIZE - 1 ) );
1816         return usage;
1817 }
1818
1819 /**
1820  * Allocate ICM
1821  *
1822  * @v hermon            Hermon device
1823  * @v init_hca          INIT_HCA structure to fill in
1824  * @ret rc              Return status code
1825  */
1826 static int hermon_alloc_icm ( struct hermon *hermon,
1827                               struct hermonprm_init_hca *init_hca ) {
1828         struct hermonprm_scalar_parameter icm_size;
1829         struct hermonprm_scalar_parameter icm_aux_size;
1830         struct hermonprm_virtual_physical_mapping map_icm_aux;
1831         struct hermonprm_virtual_physical_mapping map_icm;
1832         uint64_t icm_offset = 0;
1833         unsigned int log_num_qps, log_num_srqs, log_num_cqs, log_num_eqs;
1834         unsigned int log_num_mtts, log_num_mpts;
1835         size_t cmpt_max_len;
1836         size_t qp_cmpt_len, srq_cmpt_len, cq_cmpt_len, eq_cmpt_len;
1837         size_t icm_len, icm_aux_len;
1838         physaddr_t icm_phys;
1839         int i;
1840         int rc;
1841
1842         /*
1843          * Start by carving up the ICM virtual address space
1844          *
1845          */
1846
1847         /* Calculate number of each object type within ICM */
1848         log_num_qps = fls ( hermon->cap.reserved_qps + HERMON_MAX_QPS - 1 );
1849         log_num_srqs = fls ( hermon->cap.reserved_srqs - 1 );
1850         log_num_cqs = fls ( hermon->cap.reserved_cqs + HERMON_MAX_CQS - 1 );
1851         log_num_eqs = fls ( hermon->cap.reserved_eqs + HERMON_MAX_EQS - 1 );
1852         log_num_mtts = fls ( hermon->cap.reserved_mtts + HERMON_MAX_MTTS - 1 );
1853
1854         /* ICM starts with the cMPT tables, which are sparse */
1855         cmpt_max_len = ( HERMON_CMPT_MAX_ENTRIES *
1856                          ( ( uint64_t ) hermon->cap.cmpt_entry_size ) );
1857         qp_cmpt_len = icm_usage ( log_num_qps, hermon->cap.cmpt_entry_size );
1858         hermon->icm_map[HERMON_ICM_QP_CMPT].offset = icm_offset;
1859         hermon->icm_map[HERMON_ICM_QP_CMPT].len = qp_cmpt_len;
1860         icm_offset += cmpt_max_len;
1861         srq_cmpt_len = icm_usage ( log_num_srqs, hermon->cap.cmpt_entry_size );
1862         hermon->icm_map[HERMON_ICM_SRQ_CMPT].offset = icm_offset;
1863         hermon->icm_map[HERMON_ICM_SRQ_CMPT].len = srq_cmpt_len;
1864         icm_offset += cmpt_max_len;
1865         cq_cmpt_len = icm_usage ( log_num_cqs, hermon->cap.cmpt_entry_size );
1866         hermon->icm_map[HERMON_ICM_CQ_CMPT].offset = icm_offset;
1867         hermon->icm_map[HERMON_ICM_CQ_CMPT].len = cq_cmpt_len;
1868         icm_offset += cmpt_max_len;
1869         eq_cmpt_len = icm_usage ( log_num_eqs, hermon->cap.cmpt_entry_size );
1870         hermon->icm_map[HERMON_ICM_EQ_CMPT].offset = icm_offset;
1871         hermon->icm_map[HERMON_ICM_EQ_CMPT].len = eq_cmpt_len;
1872         icm_offset += cmpt_max_len;
1873
1874         hermon->icm_map[HERMON_ICM_OTHER].offset = icm_offset;
1875
1876         /* Queue pair contexts */
1877         MLX_FILL_1 ( init_hca, 12,
1878                      qpc_eec_cqc_eqc_rdb_parameters.qpc_base_addr_h,
1879                      ( icm_offset >> 32 ) );
1880         MLX_FILL_2 ( init_hca, 13,
1881                      qpc_eec_cqc_eqc_rdb_parameters.qpc_base_addr_l,
1882                      ( icm_offset >> 5 ),
1883                      qpc_eec_cqc_eqc_rdb_parameters.log_num_of_qp,
1884                      log_num_qps );
1885         DBGC ( hermon, "Hermon %p ICM QPC base = %llx\n", hermon, icm_offset );
1886         icm_offset += icm_usage ( log_num_qps, hermon->cap.qpc_entry_size );
1887
1888         /* Extended alternate path contexts */
1889         MLX_FILL_1 ( init_hca, 24,
1890                      qpc_eec_cqc_eqc_rdb_parameters.altc_base_addr_h,
1891                      ( icm_offset >> 32 ) );
1892         MLX_FILL_1 ( init_hca, 25,
1893                      qpc_eec_cqc_eqc_rdb_parameters.altc_base_addr_l,
1894                      icm_offset );
1895         DBGC ( hermon, "Hermon %p ICM ALTC base = %llx\n", hermon, icm_offset);
1896         icm_offset += icm_usage ( log_num_qps,
1897                                   hermon->cap.altc_entry_size );
1898
1899         /* Extended auxiliary contexts */
1900         MLX_FILL_1 ( init_hca, 28,
1901                      qpc_eec_cqc_eqc_rdb_parameters.auxc_base_addr_h,
1902                      ( icm_offset >> 32 ) );
1903         MLX_FILL_1 ( init_hca, 29,
1904                      qpc_eec_cqc_eqc_rdb_parameters.auxc_base_addr_l,
1905                      icm_offset );
1906         DBGC ( hermon, "Hermon %p ICM AUXC base = %llx\n", hermon, icm_offset);
1907         icm_offset += icm_usage ( log_num_qps,
1908                                   hermon->cap.auxc_entry_size );
1909
1910         /* Shared receive queue contexts */
1911         MLX_FILL_1 ( init_hca, 18,
1912                      qpc_eec_cqc_eqc_rdb_parameters.srqc_base_addr_h,
1913                      ( icm_offset >> 32 ) );
1914         MLX_FILL_2 ( init_hca, 19,
1915                      qpc_eec_cqc_eqc_rdb_parameters.srqc_base_addr_l,
1916                      ( icm_offset >> 5 ),
1917                      qpc_eec_cqc_eqc_rdb_parameters.log_num_of_srq,
1918                      log_num_srqs );
1919         DBGC ( hermon, "Hermon %p ICM SRQC base = %llx\n", hermon, icm_offset);
1920         icm_offset += icm_usage ( log_num_srqs,
1921                                   hermon->cap.srqc_entry_size );
1922
1923         /* Completion queue contexts */
1924         MLX_FILL_1 ( init_hca, 20,
1925                      qpc_eec_cqc_eqc_rdb_parameters.cqc_base_addr_h,
1926                      ( icm_offset >> 32 ) );
1927         MLX_FILL_2 ( init_hca, 21,
1928                      qpc_eec_cqc_eqc_rdb_parameters.cqc_base_addr_l,
1929                      ( icm_offset >> 5 ),
1930                      qpc_eec_cqc_eqc_rdb_parameters.log_num_of_cq,
1931                      log_num_cqs );
1932         DBGC ( hermon, "Hermon %p ICM CQC base = %llx\n", hermon, icm_offset );
1933         icm_offset += icm_usage ( log_num_cqs, hermon->cap.cqc_entry_size );
1934
1935         /* Event queue contexts */
1936         MLX_FILL_1 ( init_hca, 32,
1937                      qpc_eec_cqc_eqc_rdb_parameters.eqc_base_addr_h,
1938                      ( icm_offset >> 32 ) );
1939         MLX_FILL_2 ( init_hca, 33,
1940                      qpc_eec_cqc_eqc_rdb_parameters.eqc_base_addr_l,
1941                      ( icm_offset >> 5 ),
1942                      qpc_eec_cqc_eqc_rdb_parameters.log_num_of_eq,
1943                      log_num_eqs );
1944         DBGC ( hermon, "Hermon %p ICM EQC base = %llx\n", hermon, icm_offset );
1945         icm_offset += icm_usage ( log_num_eqs, hermon->cap.eqc_entry_size );
1946
1947         /* Memory translation table */
1948         MLX_FILL_1 ( init_hca, 64,
1949                      tpt_parameters.mtt_base_addr_h, ( icm_offset >> 32 ) );
1950         MLX_FILL_1 ( init_hca, 65,
1951                      tpt_parameters.mtt_base_addr_l, icm_offset );
1952         DBGC ( hermon, "Hermon %p ICM MTT base = %llx\n", hermon, icm_offset );
1953         icm_offset += icm_usage ( log_num_mtts,
1954                                   hermon->cap.mtt_entry_size );
1955
1956         /* Memory protection table */
1957         log_num_mpts = fls ( hermon->cap.reserved_mrws + 1 - 1 );
1958         MLX_FILL_1 ( init_hca, 60,
1959                      tpt_parameters.dmpt_base_adr_h, ( icm_offset >> 32 ) );
1960         MLX_FILL_1 ( init_hca, 61,
1961                      tpt_parameters.dmpt_base_adr_l, icm_offset );
1962         MLX_FILL_1 ( init_hca, 62,
1963                      tpt_parameters.log_dmpt_sz, log_num_mpts );
1964         DBGC ( hermon, "Hermon %p ICM DMPT base = %llx\n", hermon, icm_offset);
1965         icm_offset += icm_usage ( log_num_mpts,
1966                                   hermon->cap.dmpt_entry_size );
1967
1968         /* Multicast table */
1969         MLX_FILL_1 ( init_hca, 48,
1970                      multicast_parameters.mc_base_addr_h,
1971                      ( icm_offset >> 32 ) );
1972         MLX_FILL_1 ( init_hca, 49,
1973                      multicast_parameters.mc_base_addr_l, icm_offset );
1974         MLX_FILL_1 ( init_hca, 52,
1975                      multicast_parameters.log_mc_table_entry_sz,
1976                      fls ( sizeof ( struct hermonprm_mcg_entry ) - 1 ) );
1977         MLX_FILL_1 ( init_hca, 53,
1978                      multicast_parameters.log_mc_table_hash_sz, 3 );
1979         MLX_FILL_1 ( init_hca, 54,
1980                      multicast_parameters.log_mc_table_sz, 3 );
1981         DBGC ( hermon, "Hermon %p ICM MC base = %llx\n", hermon, icm_offset );
1982         icm_offset += ( ( 8 * sizeof ( struct hermonprm_mcg_entry ) +
1983                           HERMON_PAGE_SIZE - 1 ) & ~( HERMON_PAGE_SIZE - 1 ) );
1984
1985         hermon->icm_map[HERMON_ICM_OTHER].len =
1986                 ( icm_offset - hermon->icm_map[HERMON_ICM_OTHER].offset );
1987
1988         /*
1989          * Allocate and map physical memory for (portions of) ICM
1990          *
1991          * Map is:
1992          *   ICM AUX area (aligned to its own size)
1993          *   cMPT areas
1994          *   Other areas
1995          */
1996
1997         /* Calculate physical memory required for ICM */
1998         icm_len = 0;
1999         for ( i = 0 ; i < HERMON_ICM_NUM_REGIONS ; i++ ) {
2000                 icm_len += hermon->icm_map[i].len;
2001         }
2002
2003         /* Get ICM auxiliary area size */
2004         memset ( &icm_size, 0, sizeof ( icm_size ) );
2005         MLX_FILL_1 ( &icm_size, 0, value_hi, ( icm_offset >> 32 ) );
2006         MLX_FILL_1 ( &icm_size, 1, value, icm_offset );
2007         if ( ( rc = hermon_cmd_set_icm_size ( hermon, &icm_size,
2008                                               &icm_aux_size ) ) != 0 ) {
2009                 DBGC ( hermon, "Hermon %p could not set ICM size: %s\n",
2010                        hermon, strerror ( rc ) );
2011                 goto err_set_icm_size;
2012         }
2013         icm_aux_len = ( MLX_GET ( &icm_aux_size, value ) * HERMON_PAGE_SIZE );
2014         /* Must round up to nearest power of two :( */
2015         icm_aux_len = ( 1 << fls ( icm_aux_len - 1 ) );
2016
2017         /* Allocate ICM data and auxiliary area */
2018         DBGC ( hermon, "Hermon %p requires %zd kB ICM and %zd kB AUX ICM\n",
2019                hermon, ( icm_len / 1024 ), ( icm_aux_len / 1024 ) );
2020         hermon->icm = umalloc ( 2 * icm_aux_len + icm_len );
2021         if ( ! hermon->icm ) {
2022                 rc = -ENOMEM;
2023                 goto err_alloc;
2024         }
2025         icm_phys = user_to_phys ( hermon->icm, 0 );
2026
2027         /* Map ICM auxiliary area */
2028         icm_phys = ( ( icm_phys + icm_aux_len - 1 ) & ~( icm_aux_len - 1 ) );
2029         memset ( &map_icm_aux, 0, sizeof ( map_icm_aux ) );
2030         MLX_FILL_2 ( &map_icm_aux, 3,
2031                      log2size, fls ( ( icm_aux_len / HERMON_PAGE_SIZE ) - 1 ),
2032                      pa_l, ( icm_phys >> 12 ) );
2033         DBGC ( hermon, "Hermon %p mapping ICM AUX (2^%d pages) => %08lx\n",
2034                hermon, fls ( ( icm_aux_len / HERMON_PAGE_SIZE ) - 1 ),
2035                icm_phys );
2036         if ( ( rc = hermon_cmd_map_icm_aux ( hermon, &map_icm_aux ) ) != 0 ) {
2037                 DBGC ( hermon, "Hermon %p could not map AUX ICM: %s\n",
2038                        hermon, strerror ( rc ) );
2039                 goto err_map_icm_aux;
2040         }
2041         icm_phys += icm_aux_len;
2042
2043         /* MAP ICM area */
2044         for ( i = 0 ; i < HERMON_ICM_NUM_REGIONS ; i++ ) {
2045                 memset ( &map_icm, 0, sizeof ( map_icm ) );
2046                 MLX_FILL_1 ( &map_icm, 0,
2047                              va_h, ( hermon->icm_map[i].offset >> 32 ) );
2048                 MLX_FILL_1 ( &map_icm, 1,
2049                              va_l, ( hermon->icm_map[i].offset >> 12 ) );
2050                 MLX_FILL_2 ( &map_icm, 3,
2051                              log2size,
2052                              fls ( ( hermon->icm_map[i].len /
2053                                      HERMON_PAGE_SIZE ) - 1 ),
2054                              pa_l, ( icm_phys >> 12 ) );
2055                 DBGC ( hermon, "Hermon %p mapping ICM %llx+%zx (2^%d pages) "
2056                        "=> %08lx\n", hermon, hermon->icm_map[i].offset,
2057                        hermon->icm_map[i].len,
2058                        fls ( ( hermon->icm_map[i].len /
2059                                HERMON_PAGE_SIZE ) - 1 ), icm_phys );
2060                 if ( ( rc = hermon_cmd_map_icm ( hermon, &map_icm ) ) != 0 ) {
2061                         DBGC ( hermon, "Hermon %p could not map ICM: %s\n",
2062                                hermon, strerror ( rc ) );
2063                         goto err_map_icm;
2064                 }
2065                 icm_phys += hermon->icm_map[i].len;
2066         }
2067
2068         return 0;
2069
2070  err_map_icm:
2071         assert ( i == 0 ); /* We don't handle partial failure at present */
2072         hermon_cmd_unmap_icm_aux ( hermon );
2073  err_map_icm_aux:
2074         ufree ( hermon->icm );
2075         hermon->icm = UNULL;
2076  err_alloc:
2077  err_set_icm_size:
2078         return rc;
2079 }
2080
2081 /**
2082  * Free ICM
2083  *
2084  * @v hermon            Hermon device
2085  */
2086 static void hermon_free_icm ( struct hermon *hermon ) {
2087         struct hermonprm_scalar_parameter unmap_icm;
2088         int i;
2089
2090         for ( i = ( HERMON_ICM_NUM_REGIONS - 1 ) ; i >= 0 ; i-- ) {
2091                 memset ( &unmap_icm, 0, sizeof ( unmap_icm ) );
2092                 MLX_FILL_1 ( &unmap_icm, 0, value_hi,
2093                              ( hermon->icm_map[i].offset >> 32 ) );
2094                 MLX_FILL_1 ( &unmap_icm, 1, value,
2095                              hermon->icm_map[i].offset );
2096                 hermon_cmd_unmap_icm ( hermon,
2097                                        ( 1 << fls ( ( hermon->icm_map[i].len /
2098                                                       HERMON_PAGE_SIZE ) - 1)),
2099                                        &unmap_icm );
2100         }
2101         hermon_cmd_unmap_icm_aux ( hermon );
2102         ufree ( hermon->icm );
2103         hermon->icm = UNULL;
2104 }
2105
2106 /***************************************************************************
2107  *
2108  * PCI interface
2109  *
2110  ***************************************************************************
2111  */
2112
2113 /**
2114  * Set up memory protection table
2115  *
2116  * @v hermon            Hermon device
2117  * @ret rc              Return status code
2118  */
2119 static int hermon_setup_mpt ( struct hermon *hermon ) {
2120         struct hermonprm_mpt mpt;
2121         uint32_t key;
2122         int rc;
2123
2124         /* Derive key */
2125         key = ( hermon->cap.reserved_mrws | HERMON_MKEY_PREFIX );
2126         hermon->reserved_lkey = ( ( key << 8 ) | ( key >> 24 ) );
2127
2128         /* Initialise memory protection table */
2129         memset ( &mpt, 0, sizeof ( mpt ) );
2130         MLX_FILL_4 ( &mpt, 0,
2131                      r_w, 1,
2132                      pa, 1,
2133                      lr, 1,
2134                      lw, 1 );
2135         MLX_FILL_1 ( &mpt, 2, mem_key, key );
2136         MLX_FILL_1 ( &mpt, 3, pd, HERMON_GLOBAL_PD );
2137         MLX_FILL_1 ( &mpt, 10, len64, 1 );
2138         if ( ( rc = hermon_cmd_sw2hw_mpt ( hermon,
2139                                            hermon->cap.reserved_mrws,
2140                                            &mpt ) ) != 0 ) {
2141                 DBGC ( hermon, "Hermon %p could not set up MPT: %s\n",
2142                        hermon, strerror ( rc ) );
2143                 return rc;
2144         }
2145
2146         return 0;
2147 }
2148
2149 /**
2150  * Probe PCI device
2151  *
2152  * @v pci               PCI device
2153  * @v id                PCI ID
2154  * @ret rc              Return status code
2155  */
2156 static int hermon_probe ( struct pci_device *pci,
2157                           const struct pci_device_id *id __unused ) {
2158         struct hermon *hermon;
2159         struct ib_device *ibdev;
2160         struct hermonprm_init_hca init_hca;
2161         int i;
2162         int rc;
2163
2164         /* Allocate Hermon device */
2165         hermon = zalloc ( sizeof ( *hermon ) );
2166         if ( ! hermon ) {
2167                 rc = -ENOMEM;
2168                 goto err_alloc_hermon;
2169         }
2170         pci_set_drvdata ( pci, hermon );
2171         process_init ( &hermon->event_process, hermon_step, NULL );
2172
2173         /* Allocate Infiniband devices */
2174         for ( i = 0 ; i < HERMON_NUM_PORTS ; i++ ) {
2175                 ibdev = alloc_ibdev ( 0 );
2176                 if ( ! ibdev ) {
2177                         rc = -ENOMEM;
2178                         goto err_alloc_ibdev;
2179                 }
2180                 hermon->ibdev[i] = ibdev;
2181                 ibdev->op = &hermon_ib_operations;
2182                 ibdev->dev = &pci->dev;
2183                 ibdev->port = ( HERMON_PORT_BASE + i );
2184                 ib_set_drvdata ( ibdev, hermon );
2185         }
2186
2187         /* Fix up PCI device */
2188         adjust_pci_device ( pci );
2189
2190         /* Get PCI BARs */
2191         hermon->config = ioremap ( pci_bar_start ( pci, HERMON_PCI_CONFIG_BAR),
2192                                    HERMON_PCI_CONFIG_BAR_SIZE );
2193         hermon->uar = ioremap ( pci_bar_start ( pci, HERMON_PCI_UAR_BAR ),
2194                                 HERMON_UAR_NON_EQ_PAGE * HERMON_PAGE_SIZE );
2195
2196         /* Allocate space for mailboxes */
2197         hermon->mailbox_in = malloc_dma ( HERMON_MBOX_SIZE,
2198                                           HERMON_MBOX_ALIGN );
2199         if ( ! hermon->mailbox_in ) {
2200                 rc = -ENOMEM;
2201                 goto err_mailbox_in;
2202         }
2203         hermon->mailbox_out = malloc_dma ( HERMON_MBOX_SIZE,
2204                                            HERMON_MBOX_ALIGN );
2205         if ( ! hermon->mailbox_out ) {
2206                 rc = -ENOMEM;
2207                 goto err_mailbox_out;
2208         }
2209
2210         /* Start firmware */
2211         if ( ( rc = hermon_start_firmware ( hermon ) ) != 0 )
2212                 goto err_start_firmware;
2213
2214         /* Get device limits */
2215         if ( ( rc = hermon_get_cap ( hermon ) ) != 0 )
2216                 goto err_get_cap;
2217
2218         /* Allocate ICM */
2219         memset ( &init_hca, 0, sizeof ( init_hca ) );
2220         if ( ( rc = hermon_alloc_icm ( hermon, &init_hca ) ) != 0 )
2221                 goto err_alloc_icm;
2222
2223         /* Initialise HCA */
2224         MLX_FILL_1 ( &init_hca, 0, version, 0x02 /* "Must be 0x02" */ );
2225         MLX_FILL_1 ( &init_hca, 5, udp, 1 );
2226         MLX_FILL_1 ( &init_hca, 74, uar_parameters.log_max_uars, 8 );
2227         if ( ( rc = hermon_cmd_init_hca ( hermon, &init_hca ) ) != 0 ) {
2228                 DBGC ( hermon, "Hermon %p could not initialise HCA: %s\n",
2229                        hermon, strerror ( rc ) );
2230                 goto err_init_hca;
2231         }
2232
2233         /* Set up memory protection */
2234         if ( ( rc = hermon_setup_mpt ( hermon ) ) != 0 )
2235                 goto err_setup_mpt;
2236
2237         /* Set up event queue */
2238         if ( ( rc = hermon_create_eq ( hermon ) ) != 0 )
2239                 goto err_create_eq;
2240
2241         /* Register Infiniband devices */
2242         for ( i = 0 ; i < HERMON_NUM_PORTS ; i++ ) {
2243                 if ( ( rc = register_ibdev ( hermon->ibdev[i] ) ) != 0 ) {
2244                         DBGC ( hermon, "Hermon %p could not register IB "
2245                                "device: %s\n", hermon, strerror ( rc ) );
2246                         goto err_register_ibdev;
2247                 }
2248         }
2249
2250         return 0;
2251
2252         i = ( HERMON_NUM_PORTS - 1 );
2253  err_register_ibdev:
2254         for ( ; i >= 0 ; i-- )
2255                 unregister_ibdev ( hermon->ibdev[i] );
2256         hermon_destroy_eq ( hermon );
2257  err_create_eq:
2258  err_setup_mpt:
2259         hermon_cmd_close_hca ( hermon );
2260  err_init_hca:
2261         hermon_free_icm ( hermon );
2262  err_alloc_icm:
2263  err_get_cap:
2264         hermon_stop_firmware ( hermon );
2265  err_start_firmware:
2266         free_dma ( hermon->mailbox_out, HERMON_MBOX_SIZE );
2267  err_mailbox_out:
2268         free_dma ( hermon->mailbox_in, HERMON_MBOX_SIZE );
2269  err_mailbox_in:
2270         i = ( HERMON_NUM_PORTS - 1 );
2271  err_alloc_ibdev:
2272         for ( ; i >= 0 ; i-- )
2273                 free_ibdev ( hermon->ibdev[i] );
2274         process_del ( &hermon->event_process );
2275         free ( hermon );
2276  err_alloc_hermon:
2277         return rc;
2278 }
2279
2280 /**
2281  * Remove PCI device
2282  *
2283  * @v pci               PCI device
2284  */
2285 static void hermon_remove ( struct pci_device *pci ) {
2286         struct hermon *hermon = pci_get_drvdata ( pci );
2287         int i;
2288
2289         for ( i = ( HERMON_NUM_PORTS - 1 ) ; i >= 0 ; i-- )
2290                 unregister_ibdev ( hermon->ibdev[i] );
2291         hermon_destroy_eq ( hermon );
2292         hermon_cmd_close_hca ( hermon );
2293         hermon_free_icm ( hermon );
2294         hermon_stop_firmware ( hermon );
2295         hermon_stop_firmware ( hermon );
2296         free_dma ( hermon->mailbox_out, HERMON_MBOX_SIZE );
2297         free_dma ( hermon->mailbox_in, HERMON_MBOX_SIZE );
2298         for ( i = ( HERMON_NUM_PORTS - 1 ) ; i >= 0 ; i-- )
2299                 free_ibdev ( hermon->ibdev[i] );
2300         process_del ( &hermon->event_process );
2301         free ( hermon );
2302 }
2303
2304 static struct pci_device_id hermon_nics[] = {
2305         PCI_ROM ( 0x15b3, 0x6340, "mt25408", "MT25408 HCA driver" ),
2306         PCI_ROM ( 0x15b3, 0x634a, "mt25418", "MT25418 HCA driver" ),
2307 };
2308
2309 struct pci_driver hermon_driver __pci_driver = {
2310         .ids = hermon_nics,
2311         .id_count = ( sizeof ( hermon_nics ) / sizeof ( hermon_nics[0] ) ),
2312         .probe = hermon_probe,
2313         .remove = hermon_remove,
2314 };