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