sys/dev/ena/ena.c

   1 /*-
   2  * BSD LICENSE
   3  *
   4  * Copyright (c) 2015-2020 Amazon.com, Inc. or its affiliates.
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  *
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  *
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in the
  16  *    documentation and/or other materials provided with the distribution.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29  */
  30 #include <sys/cdefs.h>
  31 __FBSDID("$FreeBSD$");
  32
  33 #include "opt_rss.h"
  34
  35 #include <sys/param.h>
  36 #include <sys/systm.h>
  37 #include <sys/bus.h>
  38 #include <sys/endian.h>
  39 #include <sys/kernel.h>
  40 #include <sys/kthread.h>
  41 #include <sys/malloc.h>
  42 #include <sys/mbuf.h>
  43 #include <sys/module.h>
  44 #include <sys/rman.h>
  45 #include <sys/smp.h>
  46 #include <sys/socket.h>
  47 #include <sys/sockio.h>
  48 #include <sys/sysctl.h>
  49 #include <sys/taskqueue.h>
  50 #include <sys/time.h>
  51 #include <sys/eventhandler.h>
  52
  53 #include <machine/bus.h>
  54 #include <machine/resource.h>
  55 #include <machine/in_cksum.h>
  56
  57 #include <net/bpf.h>
  58 #include <net/ethernet.h>
  59 #include <net/if.h>
  60 #include <net/if_var.h>
  61 #include <net/if_arp.h>
  62 #include <net/if_dl.h>
  63 #include <net/if_media.h>
  64 #include <net/if_types.h>
  65 #include <net/if_vlan_var.h>
  66 #ifdef RSS
  67 #include <net/rss_config.h>
  68 #endif
  69
  70 #include <netinet/in_systm.h>
  71 #include <netinet/in.h>
  72 #include <netinet/if_ether.h>
  73 #include <netinet/ip.h>
  74 #include <netinet/ip6.h>
  75 #include <netinet/tcp.h>
  76 #include <netinet/udp.h>
  77
  78 #include <dev/pci/pcivar.h>
  79 #include <dev/pci/pcireg.h>
  80
  81 #include <vm/vm.h>
  82 #include <vm/pmap.h>
  83
  84 #include "ena_datapath.h"
  85 #include "ena.h"
  86 #include "ena_sysctl.h"
  87
  88 #ifdef DEV_NETMAP
  89 #include "ena_netmap.h"
  90 #endif /* DEV_NETMAP */
  91
  92 /*********************************************************
  93  *  Function prototypes
  94  *********************************************************/
  95 static int      ena_probe(device_t);
  96 static void     ena_intr_msix_mgmnt(void *);
  97 static void     ena_free_pci_resources(struct ena_adapter *);
  98 static int      ena_change_mtu(if_t, int);
  99 static inline void ena_alloc_counters(counter_u64_t *, int);
 100 static inline void ena_free_counters(counter_u64_t *, int);
 101 static inline void ena_reset_counters(counter_u64_t *, int);
 102 static void     ena_init_io_rings_common(struct ena_adapter *,
 103     struct ena_ring *, uint16_t);
 104 static void     ena_init_io_rings_basic(struct ena_adapter *);
 105 static void     ena_init_io_rings_advanced(struct ena_adapter *);
 106 static void     ena_init_io_rings(struct ena_adapter *);
 107 static void     ena_free_io_ring_resources(struct ena_adapter *, unsigned int);
 108 static void     ena_free_all_io_rings_resources(struct ena_adapter *);
 109 static int      ena_setup_tx_dma_tag(struct ena_adapter *);
 110 static int      ena_free_tx_dma_tag(struct ena_adapter *);
 111 static int      ena_setup_rx_dma_tag(struct ena_adapter *);
 112 static int      ena_free_rx_dma_tag(struct ena_adapter *);
 113 static void     ena_release_all_tx_dmamap(struct ena_ring *);
 114 static int      ena_setup_tx_resources(struct ena_adapter *, int);
 115 static void     ena_free_tx_resources(struct ena_adapter *, int);
 116 static int      ena_setup_all_tx_resources(struct ena_adapter *);
 117 static void     ena_free_all_tx_resources(struct ena_adapter *);
 118 static int      ena_setup_rx_resources(struct ena_adapter *, unsigned int);
 119 static void     ena_free_rx_resources(struct ena_adapter *, unsigned int);
 120 static int      ena_setup_all_rx_resources(struct ena_adapter *);
 121 static void     ena_free_all_rx_resources(struct ena_adapter *);
 122 static inline int ena_alloc_rx_mbuf(struct ena_adapter *, struct ena_ring *,
 123     struct ena_rx_buffer *);
 124 static void     ena_free_rx_mbuf(struct ena_adapter *, struct ena_ring *,
 125     struct ena_rx_buffer *);
 126 static void     ena_free_rx_bufs(struct ena_adapter *, unsigned int);
 127 static void     ena_refill_all_rx_bufs(struct ena_adapter *);
 128 static void     ena_free_all_rx_bufs(struct ena_adapter *);
 129 static void     ena_free_tx_bufs(struct ena_adapter *, unsigned int);
 130 static void     ena_free_all_tx_bufs(struct ena_adapter *);
 131 static void     ena_destroy_all_tx_queues(struct ena_adapter *);
 132 static void     ena_destroy_all_rx_queues(struct ena_adapter *);
 133 static void     ena_destroy_all_io_queues(struct ena_adapter *);
 134 static int      ena_create_io_queues(struct ena_adapter *);
 135 static int      ena_handle_msix(void *);
 136 static int      ena_enable_msix(struct ena_adapter *);
 137 static void     ena_setup_mgmnt_intr(struct ena_adapter *);
 138 static int      ena_setup_io_intr(struct ena_adapter *);
 139 static int      ena_request_mgmnt_irq(struct ena_adapter *);
 140 static int      ena_request_io_irq(struct ena_adapter *);
 141 static void     ena_free_mgmnt_irq(struct ena_adapter *);
 142 static void     ena_free_io_irq(struct ena_adapter *);
 143 static void     ena_free_irqs(struct ena_adapter*);
 144 static void     ena_disable_msix(struct ena_adapter *);
 145 static void     ena_unmask_all_io_irqs(struct ena_adapter *);
 146 static int      ena_rss_configure(struct ena_adapter *);
 147 static int      ena_up_complete(struct ena_adapter *);
 148 static uint64_t ena_get_counter(if_t, ift_counter);
 149 static int      ena_media_change(if_t);
 150 static void     ena_media_status(if_t, struct ifmediareq *);
 151 static void     ena_init(void *);
 152 static int      ena_ioctl(if_t, u_long, caddr_t);
 153 static int      ena_get_dev_offloads(struct ena_com_dev_get_features_ctx *);
 154 static void     ena_update_host_info(struct ena_admin_host_info *, if_t);
 155 static void     ena_update_hwassist(struct ena_adapter *);
 156 static int      ena_setup_ifnet(device_t, struct ena_adapter *,
 157     struct ena_com_dev_get_features_ctx *);
 158 static int      ena_enable_wc(struct resource *);
 159 static int      ena_set_queues_placement_policy(device_t, struct ena_com_dev *,
 160     struct ena_admin_feature_llq_desc *, struct ena_llq_configurations *);
 161 static uint32_t ena_calc_max_io_queue_num(device_t, struct ena_com_dev *,
 162     struct ena_com_dev_get_features_ctx *);
 163 static int      ena_calc_io_queue_size(struct ena_calc_queue_size_ctx *);
 164 static int      ena_rss_init_default(struct ena_adapter *);
 165 static void     ena_rss_init_default_deferred(void *);
 166 static void     ena_config_host_info(struct ena_com_dev *, device_t);
 167 static int      ena_attach(device_t);
 168 static int      ena_detach(device_t);
 169 static int      ena_device_init(struct ena_adapter *, device_t,
 170     struct ena_com_dev_get_features_ctx *, int *);
 171 static int      ena_enable_msix_and_set_admin_interrupts(struct ena_adapter *);
 172 static void ena_update_on_link_change(void *, struct ena_admin_aenq_entry *);
 173 static void     unimplemented_aenq_handler(void *,
 174     struct ena_admin_aenq_entry *);
 175 static void     ena_timer_service(void *);
 176
 177 static char ena_version[] = DEVICE_NAME DRV_MODULE_NAME " v" DRV_MODULE_VERSION;
 178
 179 static ena_vendor_info_t ena_vendor_info_array[] = {
 180     { PCI_VENDOR_ID_AMAZON, PCI_DEV_ID_ENA_PF, 0},
 181     { PCI_VENDOR_ID_AMAZON, PCI_DEV_ID_ENA_LLQ_PF, 0},
 182     { PCI_VENDOR_ID_AMAZON, PCI_DEV_ID_ENA_VF, 0},
 183     { PCI_VENDOR_ID_AMAZON, PCI_DEV_ID_ENA_LLQ_VF, 0},
 184     /* Last entry */
 185     { 0, 0, 0 }
 186 };
 187
 188 /*
 189  * Contains pointers to event handlers, e.g. link state chage.
 190  */
 191 static struct ena_aenq_handlers aenq_handlers;
 192
 193 void
 194 ena_dmamap_callback(void *arg, bus_dma_segment_t *segs, int nseg, int error)
 195 {
 196         if (error != 0)
 197                 return;
 198         *(bus_addr_t *) arg = segs[0].ds_addr;
 199 }
 200
 201 int
 202 ena_dma_alloc(device_t dmadev, bus_size_t size,
 203     ena_mem_handle_t *dma, int mapflags, bus_size_t alignment)
 204 {
 205         struct ena_adapter* adapter = device_get_softc(dmadev);
 206         uint32_t maxsize;
 207         uint64_t dma_space_addr;
 208         int error;
 209
 210         maxsize = ((size - 1) / PAGE_SIZE + 1) * PAGE_SIZE;
 211
 212         dma_space_addr = ENA_DMA_BIT_MASK(adapter->dma_width);
 213         if (unlikely(dma_space_addr == 0))
 214                 dma_space_addr = BUS_SPACE_MAXADDR;
 215
 216         error = bus_dma_tag_create(bus_get_dma_tag(dmadev), /* parent */
 217             alignment, 0,     /* alignment, bounds              */
 218             dma_space_addr,   /* lowaddr of exclusion window    */
 219             BUS_SPACE_MAXADDR,/* highaddr of exclusion window   */
 220             NULL, NULL,       /* filter, filterarg              */
 221             maxsize,          /* maxsize                        */
 222             1,                /* nsegments                      */
 223             maxsize,          /* maxsegsize                     */
 224             BUS_DMA_ALLOCNOW, /* flags                          */
 225             NULL,             /* lockfunc                       */
 226             NULL,             /* lockarg                        */
 227             &dma->tag);
 228         if (unlikely(error != 0)) {
 229                 ena_trace(NULL, ENA_ALERT, "bus_dma_tag_create failed: %d\n", error);
 230                 goto fail_tag;
 231         }
 232
 233         error = bus_dmamem_alloc(dma->tag, (void**) &dma->vaddr,
 234             BUS_DMA_COHERENT | BUS_DMA_ZERO, &dma->map);
 235         if (unlikely(error != 0)) {
 236                 ena_trace(NULL, ENA_ALERT, "bus_dmamem_alloc(%ju) failed: %d\n",
 237                     (uintmax_t)size, error);
 238                 goto fail_map_create;
 239         }
 240
 241         dma->paddr = 0;
 242         error = bus_dmamap_load(dma->tag, dma->map, dma->vaddr,
 243             size, ena_dmamap_callback, &dma->paddr, mapflags);
 244         if (unlikely((error != 0) || (dma->paddr == 0))) {
 245                 ena_trace(NULL, ENA_ALERT, ": bus_dmamap_load failed: %d\n", error);
 246                 goto fail_map_load;
 247         }
 248
 249         bus_dmamap_sync(dma->tag, dma->map,
 250             BUS_DMASYNC_PREREAD | BUS_DMASYNC_PREWRITE);
 251
 252         return (0);
 253
 254 fail_map_load:
 255         bus_dmamem_free(dma->tag, dma->vaddr, dma->map);
 256 fail_map_create:
 257         bus_dma_tag_destroy(dma->tag);
 258 fail_tag:
 259         dma->tag = NULL;
 260         dma->vaddr = NULL;
 261         dma->paddr = 0;
 262
 263         return (error);
 264 }
 265
 266 /*
 267  * This function should generate unique key for the whole driver.
 268  * If the key was already genereated in the previous call (for example
 269  * for another adapter), then it should be returned instead.
 270  */
 271 void
 272 ena_rss_key_fill(void *key, size_t size)
 273 {
 274         static bool key_generated;
 275         static uint8_t default_key[ENA_HASH_KEY_SIZE];
 276
 277         KASSERT(size <= ENA_HASH_KEY_SIZE, ("Requested more bytes than ENA RSS key can hold"));
 278
 279         if (!key_generated) {
 280                 arc4random_buf(default_key, ENA_HASH_KEY_SIZE);
 281                 key_generated = true;
 282         }
 283
 284         memcpy(key, default_key, size);
 285 }
 286
 287 static void
 288 ena_free_pci_resources(struct ena_adapter *adapter)
 289 {
 290         device_t pdev = adapter->pdev;
 291
 292         if (adapter->memory != NULL) {
 293                 bus_release_resource(pdev, SYS_RES_MEMORY,
 294                     PCIR_BAR(ENA_MEM_BAR), adapter->memory);
 295         }
 296
 297         if (adapter->registers != NULL) {
 298                 bus_release_resource(pdev, SYS_RES_MEMORY,
 299                     PCIR_BAR(ENA_REG_BAR), adapter->registers);
 300         }
 301 }
 302
 303 static int
 304 ena_probe(device_t dev)
 305 {
 306         ena_vendor_info_t *ent;
 307         char            adapter_name[60];
 308         uint16_t        pci_vendor_id = 0;
 309         uint16_t        pci_device_id = 0;
 310
 311         pci_vendor_id = pci_get_vendor(dev);
 312         pci_device_id = pci_get_device(dev);
 313
 314         ent = ena_vendor_info_array;
 315         while (ent->vendor_id != 0) {
 316                 if ((pci_vendor_id == ent->vendor_id) &&
 317                     (pci_device_id == ent->device_id)) {
 318                         ena_trace(NULL, ENA_DBG, "vendor=%x device=%x\n",
 319                             pci_vendor_id, pci_device_id);
 320
 321                         sprintf(adapter_name, DEVICE_DESC);
 322                         device_set_desc_copy(dev, adapter_name);
 323                         return (BUS_PROBE_DEFAULT);
 324                 }
 325
 326                 ent++;
 327
 328         }
 329
 330         return (ENXIO);
 331 }
 332
 333 static int
 334 ena_change_mtu(if_t ifp, int new_mtu)
 335 {
 336         struct ena_adapter *adapter = if_getsoftc(ifp);
 337         int rc;
 338
 339         if ((new_mtu > adapter->max_mtu) || (new_mtu < ENA_MIN_MTU)) {
 340                 device_printf(adapter->pdev, "Invalid MTU setting. "
 341                     "new_mtu: %d max mtu: %d min mtu: %d\n",
 342                     new_mtu, adapter->max_mtu, ENA_MIN_MTU);
 343                 return (EINVAL);
 344         }
 345
 346         rc = ena_com_set_dev_mtu(adapter->ena_dev, new_mtu);
 347         if (likely(rc == 0)) {
 348                 ena_trace(NULL, ENA_DBG, "set MTU to %d\n", new_mtu);
 349                 if_setmtu(ifp, new_mtu);
 350         } else {
 351                 device_printf(adapter->pdev, "Failed to set MTU to %d\n",
 352                     new_mtu);
 353         }
 354
 355         return (rc);
 356 }
 357
 358 static inline void
 359 ena_alloc_counters(counter_u64_t *begin, int size)
 360 {
 361         counter_u64_t *end = (counter_u64_t *)((char *)begin + size);
 362
 363         for (; begin < end; ++begin)
 364                 *begin = counter_u64_alloc(M_WAITOK);
 365 }
 366
 367 static inline void
 368 ena_free_counters(counter_u64_t *begin, int size)
 369 {
 370         counter_u64_t *end = (counter_u64_t *)((char *)begin + size);
 371
 372         for (; begin < end; ++begin)
 373                 counter_u64_free(*begin);
 374 }
 375
 376 static inline void
 377 ena_reset_counters(counter_u64_t *begin, int size)
 378 {
 379         counter_u64_t *end = (counter_u64_t *)((char *)begin + size);
 380
 381         for (; begin < end; ++begin)
 382                 counter_u64_zero(*begin);
 383 }
 384
 385 static void
 386 ena_init_io_rings_common(struct ena_adapter *adapter, struct ena_ring *ring,
 387     uint16_t qid)
 388 {
 389
 390         ring->qid = qid;
 391         ring->adapter = adapter;
 392         ring->ena_dev = adapter->ena_dev;
 393         ring->first_interrupt = false;
 394         ring->no_interrupt_event_cnt = 0;
 395 }
 396
 397 static void
 398 ena_init_io_rings_basic(struct ena_adapter *adapter)
 399 {
 400         struct ena_com_dev *ena_dev;
 401         struct ena_ring *txr, *rxr;
 402         struct ena_que *que;
 403         int i;
 404
 405         ena_dev = adapter->ena_dev;
 406
 407         for (i = 0; i < adapter->num_io_queues; i++) {
 408                 txr = &adapter->tx_ring[i];
 409                 rxr = &adapter->rx_ring[i];
 410
 411                 /* TX/RX common ring state */
 412                 ena_init_io_rings_common(adapter, txr, i);
 413                 ena_init_io_rings_common(adapter, rxr, i);
 414
 415                 /* TX specific ring state */
 416                 txr->tx_max_header_size = ena_dev->tx_max_header_size;
 417                 txr->tx_mem_queue_type = ena_dev->tx_mem_queue_type;
 418
 419                 que = &adapter->que[i];
 420                 que->adapter = adapter;
 421                 que->id = i;
 422                 que->tx_ring = txr;
 423                 que->rx_ring = rxr;
 424
 425                 txr->que = que;
 426                 rxr->que = que;
 427
 428                 rxr->empty_rx_queue = 0;
 429                 rxr->rx_mbuf_sz = ena_mbuf_sz;
 430         }
 431 }
 432
 433 static void
 434 ena_init_io_rings_advanced(struct ena_adapter *adapter)
 435 {
 436         struct ena_ring *txr, *rxr;
 437         int i;
 438
 439         for (i = 0; i < adapter->num_io_queues; i++) {
 440                 txr = &adapter->tx_ring[i];
 441                 rxr = &adapter->rx_ring[i];
 442
 443                 /* Allocate a buf ring */
 444                 txr->buf_ring_size = adapter->buf_ring_size;
 445                 txr->br = buf_ring_alloc(txr->buf_ring_size, M_DEVBUF,
 446                     M_WAITOK, &txr->ring_mtx);
 447
 448                 /* Allocate Tx statistics. */
 449                 ena_alloc_counters((counter_u64_t *)&txr->tx_stats,
 450                     sizeof(txr->tx_stats));
 451
 452                 /* Allocate Rx statistics. */
 453                 ena_alloc_counters((counter_u64_t *)&rxr->rx_stats,
 454                     sizeof(rxr->rx_stats));
 455
 456                 /* Initialize locks */
 457                 snprintf(txr->mtx_name, nitems(txr->mtx_name), "%s:tx(%d)",
 458                     device_get_nameunit(adapter->pdev), i);
 459                 snprintf(rxr->mtx_name, nitems(rxr->mtx_name), "%s:rx(%d)",
 460                     device_get_nameunit(adapter->pdev), i);
 461
 462                 mtx_init(&txr->ring_mtx, txr->mtx_name, NULL, MTX_DEF);
 463         }
 464 }
 465
 466 static void
 467 ena_init_io_rings(struct ena_adapter *adapter)
 468 {
 469         /*
 470          * IO rings initialization can be divided into the 2 steps:
 471          *   1. Initialize variables and fields with initial values and copy
 472          *      them from adapter/ena_dev (basic)
 473          *   2. Allocate mutex, counters and buf_ring (advanced)
 474          */
 475         ena_init_io_rings_basic(adapter);
 476         ena_init_io_rings_advanced(adapter);
 477 }
 478
 479 static void
 480 ena_free_io_ring_resources(struct ena_adapter *adapter, unsigned int qid)
 481 {
 482         struct ena_ring *txr = &adapter->tx_ring[qid];
 483         struct ena_ring *rxr = &adapter->rx_ring[qid];
 484
 485         ena_free_counters((counter_u64_t *)&txr->tx_stats,
 486             sizeof(txr->tx_stats));
 487         ena_free_counters((counter_u64_t *)&rxr->rx_stats,
 488             sizeof(rxr->rx_stats));
 489
 490         ENA_RING_MTX_LOCK(txr);
 491         drbr_free(txr->br, M_DEVBUF);
 492         ENA_RING_MTX_UNLOCK(txr);
 493
 494         mtx_destroy(&txr->ring_mtx);
 495 }
 496
 497 static void
 498 ena_free_all_io_rings_resources(struct ena_adapter *adapter)
 499 {
 500         int i;
 501
 502         for (i = 0; i < adapter->num_io_queues; i++)
 503                 ena_free_io_ring_resources(adapter, i);
 504
 505 }
 506
 507 static int
 508 ena_setup_tx_dma_tag(struct ena_adapter *adapter)
 509 {
 510         int ret;
 511
 512         /* Create DMA tag for Tx buffers */
 513         ret = bus_dma_tag_create(bus_get_dma_tag(adapter->pdev),
 514             1, 0,                                 /* alignment, bounds       */
 515             ENA_DMA_BIT_MASK(adapter->dma_width), /* lowaddr of excl window  */
 516             BUS_SPACE_MAXADDR,                    /* highaddr of excl window */
 517             NULL, NULL,                           /* filter, filterarg       */
 518             ENA_TSO_MAXSIZE,                      /* maxsize                 */
 519             adapter->max_tx_sgl_size - 1,         /* nsegments               */
 520             ENA_TSO_MAXSIZE,                      /* maxsegsize              */
 521             0,                                    /* flags                   */
 522             NULL,                                 /* lockfunc                */
 523             NULL,                                 /* lockfuncarg             */
 524             &adapter->tx_buf_tag);
 525
 526         return (ret);
 527 }
 528
 529 static int
 530 ena_free_tx_dma_tag(struct ena_adapter *adapter)
 531 {
 532         int ret;
 533
 534         ret = bus_dma_tag_destroy(adapter->tx_buf_tag);
 535
 536         if (likely(ret == 0))
 537                 adapter->tx_buf_tag = NULL;
 538
 539         return (ret);
 540 }
 541
 542 static int
 543 ena_setup_rx_dma_tag(struct ena_adapter *adapter)
 544 {
 545         int ret;
 546
 547         /* Create DMA tag for Rx buffers*/
 548         ret = bus_dma_tag_create(bus_get_dma_tag(adapter->pdev), /* parent   */
 549             1, 0,                                 /* alignment, bounds       */
 550             ENA_DMA_BIT_MASK(adapter->dma_width), /* lowaddr of excl window  */
 551             BUS_SPACE_MAXADDR,                    /* highaddr of excl window */
 552             NULL, NULL,                           /* filter, filterarg       */
 553             ena_mbuf_sz,                          /* maxsize                 */
 554             adapter->max_rx_sgl_size,             /* nsegments               */
 555             ena_mbuf_sz,                          /* maxsegsize              */
 556             0,                                    /* flags                   */
 557             NULL,                                 /* lockfunc                */
 558             NULL,                                 /* lockarg                 */
 559             &adapter->rx_buf_tag);
 560
 561         return (ret);
 562 }
 563
 564 static int
 565 ena_free_rx_dma_tag(struct ena_adapter *adapter)
 566 {
 567         int ret;
 568
 569         ret = bus_dma_tag_destroy(adapter->rx_buf_tag);
 570
 571         if (likely(ret == 0))
 572                 adapter->rx_buf_tag = NULL;
 573
 574         return (ret);
 575 }
 576
 577 static void
 578 ena_release_all_tx_dmamap(struct ena_ring *tx_ring)
 579 {
 580         struct ena_adapter *adapter = tx_ring->adapter;
 581         struct ena_tx_buffer *tx_info;
 582         bus_dma_tag_t tx_tag = adapter->tx_buf_tag;;
 583         int i;
 584 #ifdef DEV_NETMAP
 585         struct ena_netmap_tx_info *nm_info;
 586         int j;
 587 #endif /* DEV_NETMAP */
 588
 589         for (i = 0; i < tx_ring->ring_size; ++i) {
 590                 tx_info = &tx_ring->tx_buffer_info[i];
 591 #ifdef DEV_NETMAP
 592                 if (adapter->ifp->if_capenable & IFCAP_NETMAP) {
 593                         nm_info = &tx_info->nm_info;
 594                         for (j = 0; j < ENA_PKT_MAX_BUFS; ++j) {
 595                                 if (nm_info->map_seg[j] != NULL) {
 596                                         bus_dmamap_destroy(tx_tag,
 597                                             nm_info->map_seg[j]);
 598                                         nm_info->map_seg[j] = NULL;
 599                                 }
 600                         }
 601                 }
 602 #endif /* DEV_NETMAP */
 603                 if (tx_info->dmamap != NULL) {
 604                         bus_dmamap_destroy(tx_tag, tx_info->dmamap);
 605                         tx_info->dmamap = NULL;
 606                 }
 607         }
 608 }
 609
 610 /**
 611  * ena_setup_tx_resources - allocate Tx resources (Descriptors)
 612  * @adapter: network interface device structure
 613  * @qid: queue index
 614  *
 615  * Returns 0 on success, otherwise on failure.
 616  **/
 617 static int
 618 ena_setup_tx_resources(struct ena_adapter *adapter, int qid)
 619 {
 620         struct ena_que *que = &adapter->que[qid];
 621         struct ena_ring *tx_ring = que->tx_ring;
 622         int size, i, err;
 623 #ifdef DEV_NETMAP
 624         bus_dmamap_t *map;
 625         int j;
 626
 627         ena_netmap_reset_tx_ring(adapter, qid);
 628 #endif /* DEV_NETMAP */
 629
 630         size = sizeof(struct ena_tx_buffer) * tx_ring->ring_size;
 631
 632         tx_ring->tx_buffer_info = malloc(size, M_DEVBUF, M_NOWAIT | M_ZERO);
 633         if (unlikely(tx_ring->tx_buffer_info == NULL))
 634                 return (ENOMEM);
 635
 636         size = sizeof(uint16_t) * tx_ring->ring_size;
 637         tx_ring->free_tx_ids = malloc(size, M_DEVBUF, M_NOWAIT | M_ZERO);
 638         if (unlikely(tx_ring->free_tx_ids == NULL))
 639                 goto err_buf_info_free;
 640
 641         size = tx_ring->tx_max_header_size;
 642         tx_ring->push_buf_intermediate_buf = malloc(size, M_DEVBUF,
 643             M_NOWAIT | M_ZERO);
 644         if (unlikely(tx_ring->push_buf_intermediate_buf == NULL))
 645                 goto err_tx_ids_free;
 646
 647         /* Req id stack for TX OOO completions */
 648         for (i = 0; i < tx_ring->ring_size; i++)
 649                 tx_ring->free_tx_ids[i] = i;
 650
 651         /* Reset TX statistics. */
 652         ena_reset_counters((counter_u64_t *)&tx_ring->tx_stats,
 653             sizeof(tx_ring->tx_stats));
 654
 655         tx_ring->next_to_use = 0;
 656         tx_ring->next_to_clean = 0;
 657         tx_ring->acum_pkts = 0;
 658
 659         /* Make sure that drbr is empty */
 660         ENA_RING_MTX_LOCK(tx_ring);
 661         drbr_flush(adapter->ifp, tx_ring->br);
 662         ENA_RING_MTX_UNLOCK(tx_ring);
 663
 664         /* ... and create the buffer DMA maps */
 665         for (i = 0; i < tx_ring->ring_size; i++) {
 666                 err = bus_dmamap_create(adapter->tx_buf_tag, 0,
 667                     &tx_ring->tx_buffer_info[i].dmamap);
 668                 if (unlikely(err != 0)) {
 669                         ena_trace(NULL, ENA_ALERT,
 670                             "Unable to create Tx DMA map for buffer %d\n",
 671                             i);
 672                         goto err_map_release;
 673                 }
 674
 675 #ifdef DEV_NETMAP
 676                 if (adapter->ifp->if_capenable & IFCAP_NETMAP) {
 677                         map = tx_ring->tx_buffer_info[i].nm_info.map_seg;
 678                         for (j = 0; j < ENA_PKT_MAX_BUFS; j++) {
 679                                 err = bus_dmamap_create(adapter->tx_buf_tag, 0,
 680                                     &map[j]);
 681                                 if (unlikely(err != 0)) {
 682                                         ena_trace(NULL, ENA_ALERT, "Unable to create "
 683                                             "Tx DMA for buffer %d %d\n", i, j);
 684                                         goto err_map_release;
 685                                 }
 686                         }
 687                 }
 688 #endif /* DEV_NETMAP */
 689         }
 690
 691         /* Allocate taskqueues */
 692         TASK_INIT(&tx_ring->enqueue_task, 0, ena_deferred_mq_start, tx_ring);
 693         tx_ring->enqueue_tq = taskqueue_create_fast("ena_tx_enque", M_NOWAIT,
 694             taskqueue_thread_enqueue, &tx_ring->enqueue_tq);
 695         if (unlikely(tx_ring->enqueue_tq == NULL)) {
 696                 ena_trace(NULL, ENA_ALERT,
 697                     "Unable to create taskqueue for enqueue task\n");
 698                 i = tx_ring->ring_size;
 699                 goto err_map_release;
 700         }
 701
 702         tx_ring->running = true;
 703
 704         taskqueue_start_threads(&tx_ring->enqueue_tq, 1, PI_NET,
 705             "%s txeq %d", device_get_nameunit(adapter->pdev), que->cpu);
 706
 707         return (0);
 708
 709 err_map_release:
 710         ena_release_all_tx_dmamap(tx_ring);
 711 err_tx_ids_free:
 712         free(tx_ring->free_tx_ids, M_DEVBUF);
 713         tx_ring->free_tx_ids = NULL;
 714 err_buf_info_free:
 715         free(tx_ring->tx_buffer_info, M_DEVBUF);
 716         tx_ring->tx_buffer_info = NULL;
 717
 718         return (ENOMEM);
 719 }
 720
 721 /**
 722  * ena_free_tx_resources - Free Tx Resources per Queue
 723  * @adapter: network interface device structure
 724  * @qid: queue index
 725  *
 726  * Free all transmit software resources
 727  **/
 728 static void
 729 ena_free_tx_resources(struct ena_adapter *adapter, int qid)
 730 {
 731         struct ena_ring *tx_ring = &adapter->tx_ring[qid];
 732 #ifdef DEV_NETMAP
 733         struct ena_netmap_tx_info *nm_info;
 734         int j;
 735 #endif /* DEV_NETMAP */
 736
 737         while (taskqueue_cancel(tx_ring->enqueue_tq, &tx_ring->enqueue_task,
 738             NULL))
 739                 taskqueue_drain(tx_ring->enqueue_tq, &tx_ring->enqueue_task);
 740
 741         taskqueue_free(tx_ring->enqueue_tq);
 742
 743         ENA_RING_MTX_LOCK(tx_ring);
 744         /* Flush buffer ring, */
 745         drbr_flush(adapter->ifp, tx_ring->br);
 746
 747         /* Free buffer DMA maps, */
 748         for (int i = 0; i < tx_ring->ring_size; i++) {
 749                 bus_dmamap_sync(adapter->tx_buf_tag,
 750                     tx_ring->tx_buffer_info[i].dmamap, BUS_DMASYNC_POSTWRITE);
 751                 bus_dmamap_unload(adapter->tx_buf_tag,
 752                     tx_ring->tx_buffer_info[i].dmamap);
 753                 bus_dmamap_destroy(adapter->tx_buf_tag,
 754                     tx_ring->tx_buffer_info[i].dmamap);
 755
 756 #ifdef DEV_NETMAP
 757                 if (adapter->ifp->if_capenable & IFCAP_NETMAP) {
 758                         nm_info = &tx_ring->tx_buffer_info[i].nm_info;
 759                         for (j = 0; j < ENA_PKT_MAX_BUFS; j++) {
 760                                 if (nm_info->socket_buf_idx[j] != 0) {
 761                                         bus_dmamap_sync(adapter->tx_buf_tag,
 762                                             nm_info->map_seg[j],
 763                                             BUS_DMASYNC_POSTWRITE);
 764                                         ena_netmap_unload(adapter,
 765                                             nm_info->map_seg[j]);
 766                                 }
 767                                 bus_dmamap_destroy(adapter->tx_buf_tag,
 768                                     nm_info->map_seg[j]);
 769                                 nm_info->socket_buf_idx[j] = 0;
 770                         }
 771                 }
 772 #endif /* DEV_NETMAP */
 773
 774                 m_freem(tx_ring->tx_buffer_info[i].mbuf);
 775                 tx_ring->tx_buffer_info[i].mbuf = NULL;
 776         }
 777         ENA_RING_MTX_UNLOCK(tx_ring);
 778
 779         /* And free allocated memory. */
 780         free(tx_ring->tx_buffer_info, M_DEVBUF);
 781         tx_ring->tx_buffer_info = NULL;
 782
 783         free(tx_ring->free_tx_ids, M_DEVBUF);
 784         tx_ring->free_tx_ids = NULL;
 785
 786         free(tx_ring->push_buf_intermediate_buf, M_DEVBUF);
 787         tx_ring->push_buf_intermediate_buf = NULL;
 788 }
 789
 790 /**
 791  * ena_setup_all_tx_resources - allocate all queues Tx resources
 792  * @adapter: network interface device structure
 793  *
 794  * Returns 0 on success, otherwise on failure.
 795  **/
 796 static int
 797 ena_setup_all_tx_resources(struct ena_adapter *adapter)
 798 {
 799         int i, rc;
 800
 801         for (i = 0; i < adapter->num_io_queues; i++) {
 802                 rc = ena_setup_tx_resources(adapter, i);
 803                 if (rc != 0) {
 804                         device_printf(adapter->pdev,
 805                             "Allocation for Tx Queue %u failed\n", i);
 806                         goto err_setup_tx;
 807                 }
 808         }
 809
 810         return (0);
 811
 812 err_setup_tx:
 813         /* Rewind the index freeing the rings as we go */
 814         while (i--)
 815                 ena_free_tx_resources(adapter, i);
 816         return (rc);
 817 }
 818
 819 /**
 820  * ena_free_all_tx_resources - Free Tx Resources for All Queues
 821  * @adapter: network interface device structure
 822  *
 823  * Free all transmit software resources
 824  **/
 825 static void
 826 ena_free_all_tx_resources(struct ena_adapter *adapter)
 827 {
 828         int i;
 829
 830         for (i = 0; i < adapter->num_io_queues; i++)
 831                 ena_free_tx_resources(adapter, i);
 832 }
 833
 834 /**
 835  * ena_setup_rx_resources - allocate Rx resources (Descriptors)
 836  * @adapter: network interface device structure
 837  * @qid: queue index
 838  *
 839  * Returns 0 on success, otherwise on failure.
 840  **/
 841 static int
 842 ena_setup_rx_resources(struct ena_adapter *adapter, unsigned int qid)
 843 {
 844         struct ena_que *que = &adapter->que[qid];
 845         struct ena_ring *rx_ring = que->rx_ring;
 846         int size, err, i;
 847
 848         size = sizeof(struct ena_rx_buffer) * rx_ring->ring_size;
 849
 850 #ifdef DEV_NETMAP
 851         ena_netmap_reset_rx_ring(adapter, qid);
 852         rx_ring->initialized = false;
 853 #endif /* DEV_NETMAP */
 854
 855         /*
 856          * Alloc extra element so in rx path
 857          * we can always prefetch rx_info + 1
 858          */
 859         size += sizeof(struct ena_rx_buffer);
 860
 861         rx_ring->rx_buffer_info = malloc(size, M_DEVBUF, M_WAITOK | M_ZERO);
 862
 863         size = sizeof(uint16_t) * rx_ring->ring_size;
 864         rx_ring->free_rx_ids = malloc(size, M_DEVBUF, M_WAITOK);
 865
 866         for (i = 0; i < rx_ring->ring_size; i++)
 867                 rx_ring->free_rx_ids[i] = i;
 868
 869         /* Reset RX statistics. */
 870         ena_reset_counters((counter_u64_t *)&rx_ring->rx_stats,
 871             sizeof(rx_ring->rx_stats));
 872
 873         rx_ring->next_to_clean = 0;
 874         rx_ring->next_to_use = 0;
 875
 876         /* ... and create the buffer DMA maps */
 877         for (i = 0; i < rx_ring->ring_size; i++) {
 878                 err = bus_dmamap_create(adapter->rx_buf_tag, 0,
 879                     &(rx_ring->rx_buffer_info[i].map));
 880                 if (err != 0) {
 881                         ena_trace(NULL, ENA_ALERT,
 882                             "Unable to create Rx DMA map for buffer %d\n", i);
 883                         goto err_buf_info_unmap;
 884                 }
 885         }
 886
 887         /* Create LRO for the ring */
 888         if ((adapter->ifp->if_capenable & IFCAP_LRO) != 0) {
 889                 int err = tcp_lro_init(&rx_ring->lro);
 890                 if (err != 0) {
 891                         device_printf(adapter->pdev,
 892                             "LRO[%d] Initialization failed!\n", qid);
 893                 } else {
 894                         ena_trace(NULL, ENA_INFO,
 895                             "RX Soft LRO[%d] Initialized\n", qid);
 896                         rx_ring->lro.ifp = adapter->ifp;
 897                 }
 898         }
 899
 900         return (0);
 901
 902 err_buf_info_unmap:
 903         while (i--) {
 904                 bus_dmamap_destroy(adapter->rx_buf_tag,
 905                     rx_ring->rx_buffer_info[i].map);
 906         }
 907
 908         free(rx_ring->free_rx_ids, M_DEVBUF);
 909         rx_ring->free_rx_ids = NULL;
 910         free(rx_ring->rx_buffer_info, M_DEVBUF);
 911         rx_ring->rx_buffer_info = NULL;
 912         return (ENOMEM);
 913 }
 914
 915 /**
 916  * ena_free_rx_resources - Free Rx Resources
 917  * @adapter: network interface device structure
 918  * @qid: queue index
 919  *
 920  * Free all receive software resources
 921  **/
 922 static void
 923 ena_free_rx_resources(struct ena_adapter *adapter, unsigned int qid)
 924 {
 925         struct ena_ring *rx_ring = &adapter->rx_ring[qid];
 926
 927         /* Free buffer DMA maps, */
 928         for (int i = 0; i < rx_ring->ring_size; i++) {
 929                 bus_dmamap_sync(adapter->rx_buf_tag,
 930                     rx_ring->rx_buffer_info[i].map, BUS_DMASYNC_POSTREAD);
 931                 m_freem(rx_ring->rx_buffer_info[i].mbuf);
 932                 rx_ring->rx_buffer_info[i].mbuf = NULL;
 933                 bus_dmamap_unload(adapter->rx_buf_tag,
 934                     rx_ring->rx_buffer_info[i].map);
 935                 bus_dmamap_destroy(adapter->rx_buf_tag,
 936                     rx_ring->rx_buffer_info[i].map);
 937         }
 938
 939         /* free LRO resources, */
 940         tcp_lro_free(&rx_ring->lro);
 941
 942         /* free allocated memory */
 943         free(rx_ring->rx_buffer_info, M_DEVBUF);
 944         rx_ring->rx_buffer_info = NULL;
 945
 946         free(rx_ring->free_rx_ids, M_DEVBUF);
 947         rx_ring->free_rx_ids = NULL;
 948 }
 949
 950 /**
 951  * ena_setup_all_rx_resources - allocate all queues Rx resources
 952  * @adapter: network interface device structure
 953  *
 954  * Returns 0 on success, otherwise on failure.
 955  **/
 956 static int
 957 ena_setup_all_rx_resources(struct ena_adapter *adapter)
 958 {
 959         int i, rc = 0;
 960
 961         for (i = 0; i < adapter->num_io_queues; i++) {
 962                 rc = ena_setup_rx_resources(adapter, i);
 963                 if (rc != 0) {
 964                         device_printf(adapter->pdev,
 965                             "Allocation for Rx Queue %u failed\n", i);
 966                         goto err_setup_rx;
 967                 }
 968         }
 969         return (0);
 970
 971 err_setup_rx:
 972         /* rewind the index freeing the rings as we go */
 973         while (i--)
 974                 ena_free_rx_resources(adapter, i);
 975         return (rc);
 976 }
 977
 978 /**
 979  * ena_free_all_rx_resources - Free Rx resources for all queues
 980  * @adapter: network interface device structure
 981  *
 982  * Free all receive software resources
 983  **/
 984 static void
 985 ena_free_all_rx_resources(struct ena_adapter *adapter)
 986 {
 987         int i;
 988
 989         for (i = 0; i < adapter->num_io_queues; i++)
 990                 ena_free_rx_resources(adapter, i);
 991 }
 992
 993 static inline int
 994 ena_alloc_rx_mbuf(struct ena_adapter *adapter,
 995     struct ena_ring *rx_ring, struct ena_rx_buffer *rx_info)
 996 {
 997         struct ena_com_buf *ena_buf;
 998         bus_dma_segment_t segs[1];
 999         int nsegs, error;
1000         int mlen;
1001
1002         /* if previous allocated frag is not used */
1003         if (unlikely(rx_info->mbuf != NULL))
1004                 return (0);
1005
1006         /* Get mbuf using UMA allocator */
1007         rx_info->mbuf = m_getjcl(M_NOWAIT, MT_DATA, M_PKTHDR,
1008             rx_ring->rx_mbuf_sz);
1009
1010         if (unlikely(rx_info->mbuf == NULL)) {
1011                 counter_u64_add(rx_ring->rx_stats.mjum_alloc_fail, 1);
1012                 rx_info->mbuf = m_getcl(M_NOWAIT, MT_DATA, M_PKTHDR);
1013                 if (unlikely(rx_info->mbuf == NULL)) {
1014                         counter_u64_add(rx_ring->rx_stats.mbuf_alloc_fail, 1);
1015                         return (ENOMEM);
1016                 }
1017                 mlen = MCLBYTES;
1018         } else {
1019                 mlen = rx_ring->rx_mbuf_sz;
1020         }
1021         /* Set mbuf length*/
1022         rx_info->mbuf->m_pkthdr.len = rx_info->mbuf->m_len = mlen;
1023
1024         /* Map packets for DMA */
1025         ena_trace(NULL, ENA_DBG | ENA_RSC | ENA_RXPTH,
1026             "Using tag %p for buffers' DMA mapping, mbuf %p len: %d\n",
1027             adapter->rx_buf_tag,rx_info->mbuf, rx_info->mbuf->m_len);
1028         error = bus_dmamap_load_mbuf_sg(adapter->rx_buf_tag, rx_info->map,
1029             rx_info->mbuf, segs, &nsegs, BUS_DMA_NOWAIT);
1030         if (unlikely((error != 0) || (nsegs != 1))) {
1031                 ena_trace(NULL, ENA_WARNING, "failed to map mbuf, error: %d, "
1032                     "nsegs: %d\n", error, nsegs);
1033                 counter_u64_add(rx_ring->rx_stats.dma_mapping_err, 1);
1034                 goto exit;
1035
1036         }
1037
1038         bus_dmamap_sync(adapter->rx_buf_tag, rx_info->map, BUS_DMASYNC_PREREAD);
1039
1040         ena_buf = &rx_info->ena_buf;
1041         ena_buf->paddr = segs[0].ds_addr;
1042         ena_buf->len = mlen;
1043
1044         ena_trace(NULL, ENA_DBG | ENA_RSC | ENA_RXPTH,
1045             "ALLOC RX BUF: mbuf %p, rx_info %p, len %d, paddr %#jx\n",
1046             rx_info->mbuf, rx_info,ena_buf->len, (uintmax_t)ena_buf->paddr);
1047
1048         return (0);
1049
1050 exit:
1051         m_freem(rx_info->mbuf);
1052         rx_info->mbuf = NULL;
1053         return (EFAULT);
1054 }
1055
1056 static void
1057 ena_free_rx_mbuf(struct ena_adapter *adapter, struct ena_ring *rx_ring,
1058     struct ena_rx_buffer *rx_info)
1059 {
1060
1061         if (rx_info->mbuf == NULL) {
1062                 ena_trace(NULL, ENA_WARNING, "Trying to free unallocated buffer\n");
1063                 return;
1064         }
1065
1066         bus_dmamap_sync(adapter->rx_buf_tag, rx_info->map,
1067             BUS_DMASYNC_POSTREAD);
1068         bus_dmamap_unload(adapter->rx_buf_tag, rx_info->map);
1069         m_freem(rx_info->mbuf);
1070         rx_info->mbuf = NULL;
1071 }
1072
1073 /**
1074  * ena_refill_rx_bufs - Refills ring with descriptors
1075  * @rx_ring: the ring which we want to feed with free descriptors
1076  * @num: number of descriptors to refill
1077  * Refills the ring with newly allocated DMA-mapped mbufs for receiving
1078  **/
1079 int
1080 ena_refill_rx_bufs(struct ena_ring *rx_ring, uint32_t num)
1081 {
1082         struct ena_adapter *adapter = rx_ring->adapter;
1083         uint16_t next_to_use, req_id;
1084         uint32_t i;
1085         int rc;
1086
1087         ena_trace(NULL, ENA_DBG | ENA_RXPTH | ENA_RSC, "refill qid: %d\n",
1088             rx_ring->qid);
1089
1090         next_to_use = rx_ring->next_to_use;
1091
1092         for (i = 0; i < num; i++) {
1093                 struct ena_rx_buffer *rx_info;
1094
1095                 ena_trace(NULL, ENA_DBG | ENA_RXPTH | ENA_RSC,
1096                     "RX buffer - next to use: %d\n", next_to_use);
1097
1098                 req_id = rx_ring->free_rx_ids[next_to_use];
1099                 rx_info = &rx_ring->rx_buffer_info[req_id];
1100 #ifdef DEV_NETMAP
1101                 if (ena_rx_ring_in_netmap(adapter, rx_ring->qid))
1102                         rc = ena_netmap_alloc_rx_slot(adapter, rx_ring, rx_info);
1103                 else
1104 #endif /* DEV_NETMAP */
1105                         rc = ena_alloc_rx_mbuf(adapter, rx_ring, rx_info);
1106                 if (unlikely(rc != 0)) {
1107                         ena_trace(NULL, ENA_WARNING,
1108                             "failed to alloc buffer for rx queue %d\n",
1109                             rx_ring->qid);
1110                         break;
1111                 }
1112                 rc = ena_com_add_single_rx_desc(rx_ring->ena_com_io_sq,
1113                     &rx_info->ena_buf, req_id);
1114                 if (unlikely(rc != 0)) {
1115                         ena_trace(NULL, ENA_WARNING,
1116                             "failed to add buffer for rx queue %d\n",
1117                             rx_ring->qid);
1118                         break;
1119                 }
1120                 next_to_use = ENA_RX_RING_IDX_NEXT(next_to_use,
1121                     rx_ring->ring_size);
1122         }
1123
1124         if (unlikely(i < num)) {
1125                 counter_u64_add(rx_ring->rx_stats.refil_partial, 1);
1126                 ena_trace(NULL, ENA_WARNING,
1127                      "refilled rx qid %d with only %d mbufs (from %d)\n",
1128                      rx_ring->qid, i, num);
1129         }
1130
1131         if (likely(i != 0))
1132                 ena_com_write_sq_doorbell(rx_ring->ena_com_io_sq);
1133
1134         rx_ring->next_to_use = next_to_use;
1135         return (i);
1136 }
1137
1138 int
1139 ena_update_buf_ring_size(struct ena_adapter *adapter,
1140     uint32_t new_buf_ring_size)
1141 {
1142         uint32_t old_buf_ring_size;
1143         int rc = 0;
1144         bool dev_was_up;
1145
1146         ENA_LOCK_LOCK(adapter);
1147
1148         old_buf_ring_size = adapter->buf_ring_size;
1149         adapter->buf_ring_size = new_buf_ring_size;
1150
1151         dev_was_up = ENA_FLAG_ISSET(ENA_FLAG_DEV_UP, adapter);
1152         ena_down(adapter);
1153
1154         /* Reconfigure buf ring for all Tx rings. */
1155         ena_free_all_io_rings_resources(adapter);
1156         ena_init_io_rings_advanced(adapter);
1157         if (dev_was_up) {
1158                 /*
1159                  * If ena_up() fails, it's not because of recent buf_ring size
1160                  * changes. Because of that, we just want to revert old drbr
1161                  * value and trigger the reset because something else had to
1162                  * go wrong.
1163                  */
1164                 rc = ena_up(adapter);
1165                 if (unlikely(rc != 0)) {
1166                         device_printf(adapter->pdev,
1167                             "Failed to configure device after setting new drbr size: %u. Reverting old value: %u and triggering the reset\n",
1168                             new_buf_ring_size, old_buf_ring_size);
1169
1170                         /* Revert old size and trigger the reset */
1171                         adapter->buf_ring_size = old_buf_ring_size;
1172                         ena_free_all_io_rings_resources(adapter);
1173                         ena_init_io_rings_advanced(adapter);
1174
1175                         ENA_FLAG_SET_ATOMIC(ENA_FLAG_DEV_UP_BEFORE_RESET,
1176                             adapter);
1177                         ena_trigger_reset(adapter, ENA_REGS_RESET_OS_TRIGGER);
1178
1179                 }
1180         }
1181
1182         ENA_LOCK_UNLOCK(adapter);
1183
1184         return (rc);
1185 }
1186
1187 int
1188 ena_update_queue_size(struct ena_adapter *adapter, uint32_t new_tx_size,
1189     uint32_t new_rx_size)
1190 {
1191         uint32_t old_tx_size, old_rx_size;
1192         int rc = 0;
1193         bool dev_was_up;
1194
1195         ENA_LOCK_LOCK(adapter);
1196
1197         old_tx_size = adapter->requested_tx_ring_size;
1198         old_rx_size = adapter->requested_rx_ring_size;
1199         adapter->requested_tx_ring_size = new_tx_size;
1200         adapter->requested_rx_ring_size = new_rx_size;
1201
1202         dev_was_up = ENA_FLAG_ISSET(ENA_FLAG_DEV_UP, adapter);
1203         ena_down(adapter);
1204
1205         /* Configure queues with new size. */
1206         ena_init_io_rings_basic(adapter);
1207         if (dev_was_up) {
1208                 rc = ena_up(adapter);
1209                 if (unlikely(rc != 0)) {
1210                         device_printf(adapter->pdev,
1211                             "Failed to configure device with the new sizes - Tx: %u Rx: %u. Reverting old values - Tx: %u Rx: %u\n",
1212                             new_tx_size, new_rx_size, old_tx_size, old_rx_size);
1213
1214                         /* Revert old size. */
1215                         adapter->requested_tx_ring_size = old_tx_size;
1216                         adapter->requested_rx_ring_size = old_rx_size;
1217                         ena_init_io_rings_basic(adapter);
1218
1219                         /* And try again. */
1220                         rc = ena_up(adapter);
1221                         if (unlikely(rc != 0)) {
1222                                 device_printf(adapter->pdev,
1223                                     "Failed to revert old queue sizes. Triggering device reset.\n");
1224                                 /*
1225                                  * If we've failed again, something had to go
1226                                  * wrong. After reset, the device should try to
1227                                  * go up
1228                                  */
1229                                 ENA_FLAG_SET_ATOMIC(
1230                                     ENA_FLAG_DEV_UP_BEFORE_RESET, adapter);
1231                                 ena_trigger_reset(adapter,
1232                                     ENA_REGS_RESET_OS_TRIGGER);
1233                         }
1234                 }
1235         }
1236
1237         ENA_LOCK_UNLOCK(adapter);
1238
1239         return (rc);
1240 }
1241
1242 static void
1243 ena_update_io_rings(struct ena_adapter *adapter, uint32_t num)
1244 {
1245         ena_free_all_io_rings_resources(adapter);
1246         /* Force indirection table to be reinitialized */
1247         ena_com_rss_destroy(adapter->ena_dev);
1248
1249         adapter->num_io_queues = num;
1250         ena_init_io_rings(adapter);
1251 }
1252
1253 /* Caller should sanitize new_num */
1254 int
1255 ena_update_io_queue_nb(struct ena_adapter *adapter, uint32_t new_num)
1256 {
1257         uint32_t old_num;
1258         int rc = 0;
1259         bool dev_was_up;
1260
1261         ENA_LOCK_LOCK(adapter);
1262
1263         dev_was_up = ENA_FLAG_ISSET(ENA_FLAG_DEV_UP, adapter);
1264         old_num = adapter->num_io_queues;
1265         ena_down(adapter);
1266
1267         ena_update_io_rings(adapter, new_num);
1268
1269         if (dev_was_up) {
1270                 rc = ena_up(adapter);
1271                 if (unlikely(rc != 0)) {
1272                         device_printf(adapter->pdev,
1273                             "Failed to configure device with %u IO queues. "
1274                             "Reverting to previous value: %u\n",
1275                             new_num, old_num);
1276
1277                         ena_update_io_rings(adapter, old_num);
1278
1279                         rc = ena_up(adapter);
1280                         if (unlikely(rc != 0)) {
1281                                 device_printf(adapter->pdev,
1282                                     "Failed to revert to previous setup IO "
1283                                     "queues. Triggering device reset.\n");
1284                                 ENA_FLAG_SET_ATOMIC(
1285                                     ENA_FLAG_DEV_UP_BEFORE_RESET, adapter);
1286                                 ena_trigger_reset(adapter,
1287                                     ENA_REGS_RESET_OS_TRIGGER);
1288                         }
1289                 }
1290         }
1291
1292         ENA_LOCK_UNLOCK(adapter);
1293
1294         return (rc);
1295 }
1296
1297 static void
1298 ena_free_rx_bufs(struct ena_adapter *adapter, unsigned int qid)
1299 {
1300         struct ena_ring *rx_ring = &adapter->rx_ring[qid];
1301         unsigned int i;
1302
1303         for (i = 0; i < rx_ring->ring_size; i++) {
1304                 struct ena_rx_buffer *rx_info = &rx_ring->rx_buffer_info[i];
1305
1306                 if (rx_info->mbuf != NULL)
1307                         ena_free_rx_mbuf(adapter, rx_ring, rx_info);
1308 #ifdef DEV_NETMAP
1309                 if (((if_getflags(adapter->ifp) & IFF_DYING) == 0) &&
1310                     (adapter->ifp->if_capenable & IFCAP_NETMAP)) {
1311                         if (rx_info->netmap_buf_idx != 0)
1312                                 ena_netmap_free_rx_slot(adapter, rx_ring,
1313                                     rx_info);
1314                 }
1315 #endif /* DEV_NETMAP */
1316         }
1317 }
1318
1319 /**
1320  * ena_refill_all_rx_bufs - allocate all queues Rx buffers
1321  * @adapter: network interface device structure
1322  *
1323  */
1324 static void
1325 ena_refill_all_rx_bufs(struct ena_adapter *adapter)
1326 {
1327         struct ena_ring *rx_ring;
1328         int i, rc, bufs_num;
1329
1330         for (i = 0; i < adapter->num_io_queues; i++) {
1331                 rx_ring = &adapter->rx_ring[i];
1332                 bufs_num = rx_ring->ring_size - 1;
1333                 rc = ena_refill_rx_bufs(rx_ring, bufs_num);
1334                 if (unlikely(rc != bufs_num))
1335                         ena_trace(NULL, ENA_WARNING, "refilling Queue %d failed. "
1336                             "Allocated %d buffers from: %d\n", i, rc, bufs_num);
1337 #ifdef DEV_NETMAP
1338                 rx_ring->initialized = true;
1339 #endif /* DEV_NETMAP */
1340         }
1341 }
1342
1343 static void
1344 ena_free_all_rx_bufs(struct ena_adapter *adapter)
1345 {
1346         int i;
1347
1348         for (i = 0; i < adapter->num_io_queues; i++)
1349                 ena_free_rx_bufs(adapter, i);
1350 }
1351
1352 /**
1353  * ena_free_tx_bufs - Free Tx Buffers per Queue
1354  * @adapter: network interface device structure
1355  * @qid: queue index
1356  **/
1357 static void
1358 ena_free_tx_bufs(struct ena_adapter *adapter, unsigned int qid)
1359 {
1360         bool print_once = true;
1361         struct ena_ring *tx_ring = &adapter->tx_ring[qid];
1362
1363         ENA_RING_MTX_LOCK(tx_ring);
1364         for (int i = 0; i < tx_ring->ring_size; i++) {
1365                 struct ena_tx_buffer *tx_info = &tx_ring->tx_buffer_info[i];
1366
1367                 if (tx_info->mbuf == NULL)
1368                         continue;
1369
1370                 if (print_once) {
1371                         device_printf(adapter->pdev,
1372                             "free uncompleted tx mbuf qid %d idx 0x%x\n",
1373                             qid, i);
1374                         print_once = false;
1375                 } else {
1376                         ena_trace(NULL, ENA_DBG,
1377                             "free uncompleted tx mbuf qid %d idx 0x%x\n",
1378                              qid, i);
1379                 }
1380
1381                 bus_dmamap_sync(adapter->tx_buf_tag, tx_info->dmamap,
1382                     BUS_DMASYNC_POSTWRITE);
1383                 bus_dmamap_unload(adapter->tx_buf_tag, tx_info->dmamap);
1384
1385                 m_free(tx_info->mbuf);
1386                 tx_info->mbuf = NULL;
1387         }
1388         ENA_RING_MTX_UNLOCK(tx_ring);
1389 }
1390
1391 static void
1392 ena_free_all_tx_bufs(struct ena_adapter *adapter)
1393 {
1394
1395         for (int i = 0; i < adapter->num_io_queues; i++)
1396                 ena_free_tx_bufs(adapter, i);
1397 }
1398
1399 static void
1400 ena_destroy_all_tx_queues(struct ena_adapter *adapter)
1401 {
1402         uint16_t ena_qid;
1403         int i;
1404
1405         for (i = 0; i < adapter->num_io_queues; i++) {
1406                 ena_qid = ENA_IO_TXQ_IDX(i);
1407                 ena_com_destroy_io_queue(adapter->ena_dev, ena_qid);
1408         }
1409 }
1410
1411 static void
1412 ena_destroy_all_rx_queues(struct ena_adapter *adapter)
1413 {
1414         uint16_t ena_qid;
1415         int i;
1416
1417         for (i = 0; i < adapter->num_io_queues; i++) {
1418                 ena_qid = ENA_IO_RXQ_IDX(i);
1419                 ena_com_destroy_io_queue(adapter->ena_dev, ena_qid);
1420         }
1421 }
1422
1423 static void
1424 ena_destroy_all_io_queues(struct ena_adapter *adapter)
1425 {
1426         struct ena_que *queue;
1427         int i;
1428
1429         for (i = 0; i < adapter->num_io_queues; i++) {
1430                 queue = &adapter->que[i];
1431                 while (taskqueue_cancel(queue->cleanup_tq,
1432                     &queue->cleanup_task, NULL))
1433                         taskqueue_drain(queue->cleanup_tq,
1434                             &queue->cleanup_task);
1435                 taskqueue_free(queue->cleanup_tq);
1436         }
1437
1438         ena_destroy_all_tx_queues(adapter);
1439         ena_destroy_all_rx_queues(adapter);
1440 }
1441
1442 static int
1443 ena_create_io_queues(struct ena_adapter *adapter)
1444 {
1445         struct ena_com_dev *ena_dev = adapter->ena_dev;
1446         struct ena_com_create_io_ctx ctx;
1447         struct ena_ring *ring;
1448         struct ena_que *queue;
1449         uint16_t ena_qid;
1450         uint32_t msix_vector;
1451         int rc, i;
1452
1453         /* Create TX queues */
1454         for (i = 0; i < adapter->num_io_queues; i++) {
1455                 msix_vector = ENA_IO_IRQ_IDX(i);
1456                 ena_qid = ENA_IO_TXQ_IDX(i);
1457                 ctx.mem_queue_type = ena_dev->tx_mem_queue_type;
1458                 ctx.direction = ENA_COM_IO_QUEUE_DIRECTION_TX;
1459                 ctx.queue_size = adapter->requested_tx_ring_size;
1460                 ctx.msix_vector = msix_vector;
1461                 ctx.qid = ena_qid;
1462                 rc = ena_com_create_io_queue(ena_dev, &ctx);
1463                 if (rc != 0) {
1464                         device_printf(adapter->pdev,
1465                             "Failed to create io TX queue #%d rc: %d\n", i, rc);
1466                         goto err_tx;
1467                 }
1468                 ring = &adapter->tx_ring[i];
1469                 rc = ena_com_get_io_handlers(ena_dev, ena_qid,
1470                     &ring->ena_com_io_sq,
1471                     &ring->ena_com_io_cq);
1472                 if (rc != 0) {
1473                         device_printf(adapter->pdev,
1474                             "Failed to get TX queue handlers. TX queue num"
1475                             " %d rc: %d\n", i, rc);
1476                         ena_com_destroy_io_queue(ena_dev, ena_qid);
1477                         goto err_tx;
1478                 }
1479         }
1480
1481         /* Create RX queues */
1482         for (i = 0; i < adapter->num_io_queues; i++) {
1483                 msix_vector = ENA_IO_IRQ_IDX(i);
1484                 ena_qid = ENA_IO_RXQ_IDX(i);
1485                 ctx.mem_queue_type = ENA_ADMIN_PLACEMENT_POLICY_HOST;
1486                 ctx.direction = ENA_COM_IO_QUEUE_DIRECTION_RX;
1487                 ctx.queue_size = adapter->requested_rx_ring_size;
1488                 ctx.msix_vector = msix_vector;
1489                 ctx.qid = ena_qid;
1490                 rc = ena_com_create_io_queue(ena_dev, &ctx);
1491                 if (unlikely(rc != 0)) {
1492                         device_printf(adapter->pdev,
1493                             "Failed to create io RX queue[%d] rc: %d\n", i, rc);
1494                         goto err_rx;
1495                 }
1496
1497                 ring = &adapter->rx_ring[i];
1498                 rc = ena_com_get_io_handlers(ena_dev, ena_qid,
1499                     &ring->ena_com_io_sq,
1500                     &ring->ena_com_io_cq);
1501                 if (unlikely(rc != 0)) {
1502                         device_printf(adapter->pdev,
1503                             "Failed to get RX queue handlers. RX queue num"
1504                             " %d rc: %d\n", i, rc);
1505                         ena_com_destroy_io_queue(ena_dev, ena_qid);
1506                         goto err_rx;
1507                 }
1508         }
1509
1510         for (i = 0; i < adapter->num_io_queues; i++) {
1511                 queue = &adapter->que[i];
1512
1513                 NET_TASK_INIT(&queue->cleanup_task, 0, ena_cleanup, queue);
1514                 queue->cleanup_tq = taskqueue_create_fast("ena cleanup",
1515                     M_WAITOK, taskqueue_thread_enqueue, &queue->cleanup_tq);
1516
1517                 taskqueue_start_threads(&queue->cleanup_tq, 1, PI_NET,
1518                     "%s queue %d cleanup",
1519                     device_get_nameunit(adapter->pdev), i);
1520         }
1521
1522         return (0);
1523
1524 err_rx:
1525         while (i--)
1526                 ena_com_destroy_io_queue(ena_dev, ENA_IO_RXQ_IDX(i));
1527         i = adapter->num_io_queues;
1528 err_tx:
1529         while (i--)
1530                 ena_com_destroy_io_queue(ena_dev, ENA_IO_TXQ_IDX(i));
1531
1532         return (ENXIO);
1533 }
1534
1535 /*********************************************************************
1536  *
1537  *  MSIX & Interrupt Service routine
1538  *
1539  **********************************************************************/
1540
1541 /**
1542  * ena_handle_msix - MSIX Interrupt Handler for admin/async queue
1543  * @arg: interrupt number
1544  **/
1545 static void
1546 ena_intr_msix_mgmnt(void *arg)
1547 {
1548         struct ena_adapter *adapter = (struct ena_adapter *)arg;
1549
1550         ena_com_admin_q_comp_intr_handler(adapter->ena_dev);
1551         if (likely(ENA_FLAG_ISSET(ENA_FLAG_DEVICE_RUNNING, adapter)))
1552                 ena_com_aenq_intr_handler(adapter->ena_dev, arg);
1553 }
1554
1555 /**
1556  * ena_handle_msix - MSIX Interrupt Handler for Tx/Rx
1557  * @arg: queue
1558  **/
1559 static int
1560 ena_handle_msix(void *arg)
1561 {
1562         struct ena_que *queue = arg;
1563         struct ena_adapter *adapter = queue->adapter;
1564         if_t ifp = adapter->ifp;
1565
1566         if (unlikely((if_getdrvflags(ifp) & IFF_DRV_RUNNING) == 0))
1567                 return (FILTER_STRAY);
1568
1569         taskqueue_enqueue(queue->cleanup_tq, &queue->cleanup_task);
1570
1571         return (FILTER_HANDLED);
1572 }
1573
1574 static int
1575 ena_enable_msix(struct ena_adapter *adapter)
1576 {
1577         device_t dev = adapter->pdev;
1578         int msix_vecs, msix_req;
1579         int i, rc = 0;
1580
1581         if (ENA_FLAG_ISSET(ENA_FLAG_MSIX_ENABLED, adapter)) {
1582                 device_printf(dev, "Error, MSI-X is already enabled\n");
1583                 return (EINVAL);
1584         }
1585
1586         /* Reserved the max msix vectors we might need */
1587         msix_vecs = ENA_MAX_MSIX_VEC(adapter->max_num_io_queues);
1588
1589         adapter->msix_entries = malloc(msix_vecs * sizeof(struct msix_entry),
1590             M_DEVBUF, M_WAITOK | M_ZERO);
1591
1592         ena_trace(NULL, ENA_DBG, "trying to enable MSI-X, vectors: %d\n", msix_vecs);
1593
1594         for (i = 0; i < msix_vecs; i++) {
1595                 adapter->msix_entries[i].entry = i;
1596                 /* Vectors must start from 1 */
1597                 adapter->msix_entries[i].vector = i + 1;
1598         }
1599
1600         msix_req = msix_vecs;
1601         rc = pci_alloc_msix(dev, &msix_vecs);
1602         if (unlikely(rc != 0)) {
1603                 device_printf(dev,
1604                     "Failed to enable MSIX, vectors %d rc %d\n", msix_vecs, rc);
1605
1606                 rc = ENOSPC;
1607                 goto err_msix_free;
1608         }
1609
1610         if (msix_vecs != msix_req) {
1611                 if (msix_vecs == ENA_ADMIN_MSIX_VEC) {
1612                         device_printf(dev,
1613                             "Not enough number of MSI-x allocated: %d\n",
1614                             msix_vecs);
1615                         pci_release_msi(dev);
1616                         rc = ENOSPC;
1617                         goto err_msix_free;
1618                 }
1619                 device_printf(dev, "Enable only %d MSI-x (out of %d), reduce "
1620                     "the number of queues\n", msix_vecs, msix_req);
1621         }
1622
1623         adapter->msix_vecs = msix_vecs;
1624         ENA_FLAG_SET_ATOMIC(ENA_FLAG_MSIX_ENABLED, adapter);
1625
1626         return (0);
1627
1628 err_msix_free:
1629         free(adapter->msix_entries, M_DEVBUF);
1630         adapter->msix_entries = NULL;
1631
1632         return (rc);
1633 }
1634
1635 static void
1636 ena_setup_mgmnt_intr(struct ena_adapter *adapter)
1637 {
1638
1639         snprintf(adapter->irq_tbl[ENA_MGMNT_IRQ_IDX].name,
1640             ENA_IRQNAME_SIZE, "ena-mgmnt@pci:%s",
1641             device_get_nameunit(adapter->pdev));
1642         /*
1643          * Handler is NULL on purpose, it will be set
1644          * when mgmnt interrupt is acquired
1645          */
1646         adapter->irq_tbl[ENA_MGMNT_IRQ_IDX].handler = NULL;
1647         adapter->irq_tbl[ENA_MGMNT_IRQ_IDX].data = adapter;
1648         adapter->irq_tbl[ENA_MGMNT_IRQ_IDX].vector =
1649             adapter->msix_entries[ENA_MGMNT_IRQ_IDX].vector;
1650 }
1651
1652 static int
1653 ena_setup_io_intr(struct ena_adapter *adapter)
1654 {
1655         static int last_bind_cpu = -1;
1656         int irq_idx;
1657
1658         if (adapter->msix_entries == NULL)
1659                 return (EINVAL);
1660
1661         for (int i = 0; i < adapter->num_io_queues; i++) {
1662                 irq_idx = ENA_IO_IRQ_IDX(i);
1663
1664                 snprintf(adapter->irq_tbl[irq_idx].name, ENA_IRQNAME_SIZE,
1665                     "%s-TxRx-%d", device_get_nameunit(adapter->pdev), i);
1666                 adapter->irq_tbl[irq_idx].handler = ena_handle_msix;
1667                 adapter->irq_tbl[irq_idx].data = &adapter->que[i];
1668                 adapter->irq_tbl[irq_idx].vector =
1669                     adapter->msix_entries[irq_idx].vector;
1670                 ena_trace(NULL, ENA_INFO | ENA_IOQ, "ena_setup_io_intr vector: %d\n",
1671                     adapter->msix_entries[irq_idx].vector);
1672
1673                 /*
1674                  * We want to bind rings to the corresponding cpu
1675                  * using something similar to the RSS round-robin technique.
1676                  */
1677                 if (unlikely(last_bind_cpu < 0))
1678                         last_bind_cpu = CPU_FIRST();
1679                 adapter->que[i].cpu = adapter->irq_tbl[irq_idx].cpu =
1680                     last_bind_cpu;
1681                 last_bind_cpu = CPU_NEXT(last_bind_cpu);
1682         }
1683
1684         return (0);
1685 }
1686
1687 static int
1688 ena_request_mgmnt_irq(struct ena_adapter *adapter)
1689 {
1690         struct ena_irq *irq;
1691         unsigned long flags;
1692         int rc, rcc;
1693
1694         flags = RF_ACTIVE | RF_SHAREABLE;
1695
1696         irq = &adapter->irq_tbl[ENA_MGMNT_IRQ_IDX];
1697         irq->res = bus_alloc_resource_any(adapter->pdev, SYS_RES_IRQ,
1698             &irq->vector, flags);
1699
1700         if (unlikely(irq->res == NULL)) {
1701                 device_printf(adapter->pdev, "could not allocate "
1702                     "irq vector: %d\n", irq->vector);
1703                 return (ENXIO);
1704         }
1705
1706         rc = bus_setup_intr(adapter->pdev, irq->res,
1707             INTR_TYPE_NET | INTR_MPSAFE, NULL, ena_intr_msix_mgmnt,
1708             irq->data, &irq->cookie);
1709         if (unlikely(rc != 0)) {
1710                 device_printf(adapter->pdev, "failed to register "
1711                     "interrupt handler for irq %ju: %d\n",
1712                     rman_get_start(irq->res), rc);
1713                 goto err_res_free;
1714         }
1715         irq->requested = true;
1716
1717         return (rc);
1718
1719 err_res_free:
1720         ena_trace(NULL, ENA_INFO | ENA_ADMQ, "releasing resource for irq %d\n",
1721             irq->vector);
1722         rcc = bus_release_resource(adapter->pdev, SYS_RES_IRQ,
1723             irq->vector, irq->res);
1724         if (unlikely(rcc != 0))
1725                 device_printf(adapter->pdev, "dev has no parent while "
1726                     "releasing res for irq: %d\n", irq->vector);
1727         irq->res = NULL;
1728
1729         return (rc);
1730 }
1731
1732 static int
1733 ena_request_io_irq(struct ena_adapter *adapter)
1734 {
1735         struct ena_irq *irq;
1736         unsigned long flags = 0;
1737         int rc = 0, i, rcc;
1738
1739         if (unlikely(!ENA_FLAG_ISSET(ENA_FLAG_MSIX_ENABLED, adapter))) {
1740                 device_printf(adapter->pdev,
1741                     "failed to request I/O IRQ: MSI-X is not enabled\n");
1742                 return (EINVAL);
1743         } else {
1744                 flags = RF_ACTIVE | RF_SHAREABLE;
1745         }
1746
1747         for (i = ENA_IO_IRQ_FIRST_IDX; i < adapter->msix_vecs; i++) {
1748                 irq = &adapter->irq_tbl[i];
1749
1750                 if (unlikely(irq->requested))
1751                         continue;
1752
1753                 irq->res = bus_alloc_resource_any(adapter->pdev, SYS_RES_IRQ,
1754                     &irq->vector, flags);
1755                 if (unlikely(irq->res == NULL)) {
1756                         rc = ENOMEM;
1757                         device_printf(adapter->pdev, "could not allocate "
1758                             "irq vector: %d\n", irq->vector);
1759                         goto err;
1760                 }
1761
1762                 rc = bus_setup_intr(adapter->pdev, irq->res,
1763                     INTR_TYPE_NET | INTR_MPSAFE, irq->handler, NULL,
1764                     irq->data, &irq->cookie);
1765                  if (unlikely(rc != 0)) {
1766                         device_printf(adapter->pdev, "failed to register "
1767                             "interrupt handler for irq %ju: %d\n",
1768                             rman_get_start(irq->res), rc);
1769                         goto err;
1770                 }
1771                 irq->requested = true;
1772
1773                 ena_trace(NULL, ENA_INFO, "queue %d - cpu %d\n",
1774                     i - ENA_IO_IRQ_FIRST_IDX, irq->cpu);
1775         }
1776
1777         return (rc);
1778
1779 err:
1780
1781         for (; i >= ENA_IO_IRQ_FIRST_IDX; i--) {
1782                 irq = &adapter->irq_tbl[i];
1783                 rcc = 0;
1784
1785                 /* Once we entered err: section and irq->requested is true we
1786                    free both intr and resources */
1787                 if (irq->requested)
1788                         rcc = bus_teardown_intr(adapter->pdev, irq->res, irq->cookie);
1789                 if (unlikely(rcc != 0))
1790                         device_printf(adapter->pdev, "could not release"
1791                             " irq: %d, error: %d\n", irq->vector, rcc);
1792
1793                 /* If we entred err: section without irq->requested set we know
1794                    it was bus_alloc_resource_any() that needs cleanup, provided
1795                    res is not NULL. In case res is NULL no work in needed in
1796                    this iteration */
1797                 rcc = 0;
1798                 if (irq->res != NULL) {
1799                         rcc = bus_release_resource(adapter->pdev, SYS_RES_IRQ,
1800                             irq->vector, irq->res);
1801                 }
1802                 if (unlikely(rcc != 0))
1803                         device_printf(adapter->pdev, "dev has no parent while "
1804                             "releasing res for irq: %d\n", irq->vector);
1805                 irq->requested = false;
1806                 irq->res = NULL;
1807         }
1808
1809         return (rc);
1810 }
1811
1812 static void
1813 ena_free_mgmnt_irq(struct ena_adapter *adapter)
1814 {
1815         struct ena_irq *irq;
1816         int rc;
1817
1818         irq = &adapter->irq_tbl[ENA_MGMNT_IRQ_IDX];
1819         if (irq->requested) {
1820                 ena_trace(NULL, ENA_INFO | ENA_ADMQ, "tear down irq: %d\n",
1821                     irq->vector);
1822                 rc = bus_teardown_intr(adapter->pdev, irq->res, irq->cookie);
1823                 if (unlikely(rc != 0))
1824                         device_printf(adapter->pdev, "failed to tear "
1825                             "down irq: %d\n", irq->vector);
1826                 irq->requested = 0;
1827         }
1828
1829         if (irq->res != NULL) {
1830                 ena_trace(NULL, ENA_INFO | ENA_ADMQ, "release resource irq: %d\n",
1831                     irq->vector);
1832                 rc = bus_release_resource(adapter->pdev, SYS_RES_IRQ,
1833                     irq->vector, irq->res);
1834                 irq->res = NULL;
1835                 if (unlikely(rc != 0))
1836                         device_printf(adapter->pdev, "dev has no parent while "
1837                             "releasing res for irq: %d\n", irq->vector);
1838         }
1839 }
1840
1841 static void
1842 ena_free_io_irq(struct ena_adapter *adapter)
1843 {
1844         struct ena_irq *irq;
1845         int rc;
1846
1847         for (int i = ENA_IO_IRQ_FIRST_IDX; i < adapter->msix_vecs; i++) {
1848                 irq = &adapter->irq_tbl[i];
1849                 if (irq->requested) {
1850                         ena_trace(NULL, ENA_INFO | ENA_IOQ, "tear down irq: %d\n",
1851                             irq->vector);
1852                         rc = bus_teardown_intr(adapter->pdev, irq->res,
1853                             irq->cookie);
1854                         if (unlikely(rc != 0)) {
1855                                 device_printf(adapter->pdev, "failed to tear "
1856                                     "down irq: %d\n", irq->vector);
1857                         }
1858                         irq->requested = 0;
1859                 }
1860
1861                 if (irq->res != NULL) {
1862                         ena_trace(NULL, ENA_INFO | ENA_IOQ, "release resource irq: %d\n",
1863                             irq->vector);
1864                         rc = bus_release_resource(adapter->pdev, SYS_RES_IRQ,
1865                             irq->vector, irq->res);
1866                         irq->res = NULL;
1867                         if (unlikely(rc != 0)) {
1868                                 device_printf(adapter->pdev, "dev has no parent"
1869                                     " while releasing res for irq: %d\n",
1870                                     irq->vector);
1871                         }
1872                 }
1873         }
1874 }
1875
1876 static void
1877 ena_free_irqs(struct ena_adapter* adapter)
1878 {
1879
1880         ena_free_io_irq(adapter);
1881         ena_free_mgmnt_irq(adapter);
1882         ena_disable_msix(adapter);
1883 }
1884
1885 static void
1886 ena_disable_msix(struct ena_adapter *adapter)
1887 {
1888
1889         if (ENA_FLAG_ISSET(ENA_FLAG_MSIX_ENABLED, adapter)) {
1890                 ENA_FLAG_CLEAR_ATOMIC(ENA_FLAG_MSIX_ENABLED, adapter);
1891                 pci_release_msi(adapter->pdev);
1892         }
1893
1894         adapter->msix_vecs = 0;
1895         if (adapter->msix_entries != NULL)
1896                 free(adapter->msix_entries, M_DEVBUF);
1897         adapter->msix_entries = NULL;
1898 }
1899
1900 static void
1901 ena_unmask_all_io_irqs(struct ena_adapter *adapter)
1902 {
1903         struct ena_com_io_cq* io_cq;
1904         struct ena_eth_io_intr_reg intr_reg;
1905         uint16_t ena_qid;
1906         int i;
1907
1908         /* Unmask interrupts for all queues */
1909         for (i = 0; i < adapter->num_io_queues; i++) {
1910                 ena_qid = ENA_IO_TXQ_IDX(i);
1911                 io_cq = &adapter->ena_dev->io_cq_queues[ena_qid];
1912                 ena_com_update_intr_reg(&intr_reg, 0, 0, true);
1913                 ena_com_unmask_intr(io_cq, &intr_reg);
1914         }
1915 }
1916
1917 /* Configure the Rx forwarding */
1918 static int
1919 ena_rss_configure(struct ena_adapter *adapter)
1920 {
1921         struct ena_com_dev *ena_dev = adapter->ena_dev;
1922         int rc;
1923
1924         /* In case the RSS table was destroyed */
1925         if (!ena_dev->rss.tbl_log_size) {
1926                 rc = ena_rss_init_default(adapter);
1927                 if (unlikely((rc != 0) && (rc != EOPNOTSUPP))) {
1928                         device_printf(adapter->pdev,
1929                             "WARNING: RSS was not properly re-initialized,"
1930                             " it will affect bandwidth\n");
1931                         ENA_FLAG_CLEAR_ATOMIC(ENA_FLAG_RSS_ACTIVE, adapter);
1932                         return (rc);
1933                 }
1934         }
1935
1936         /* Set indirect table */
1937         rc = ena_com_indirect_table_set(ena_dev);
1938         if (unlikely((rc != 0) && (rc != EOPNOTSUPP)))
1939                 return (rc);
1940
1941         /* Configure hash function (if supported) */
1942         rc = ena_com_set_hash_function(ena_dev);
1943         if (unlikely((rc != 0) && (rc != EOPNOTSUPP)))
1944                 return (rc);
1945
1946         /* Configure hash inputs (if supported) */
1947         rc = ena_com_set_hash_ctrl(ena_dev);
1948         if (unlikely((rc != 0) && (rc != EOPNOTSUPP)))
1949                 return (rc);
1950
1951         return (0);
1952 }
1953
1954 static int
1955 ena_up_complete(struct ena_adapter *adapter)
1956 {
1957         int rc;
1958
1959         if (likely(ENA_FLAG_ISSET(ENA_FLAG_RSS_ACTIVE, adapter))) {
1960                 rc = ena_rss_configure(adapter);
1961                 if (rc != 0) {
1962                         device_printf(adapter->pdev,
1963                             "Failed to configure RSS\n");
1964                         return (rc);
1965                 }
1966         }
1967
1968         rc = ena_change_mtu(adapter->ifp, adapter->ifp->if_mtu);
1969         if (unlikely(rc != 0))
1970                 return (rc);
1971
1972         ena_refill_all_rx_bufs(adapter);
1973         ena_reset_counters((counter_u64_t *)&adapter->hw_stats,
1974             sizeof(adapter->hw_stats));
1975
1976         return (0);
1977 }
1978
1979 static void
1980 set_io_rings_size(struct ena_adapter *adapter, int new_tx_size,
1981     int new_rx_size)
1982 {
1983         int i;
1984
1985         for (i = 0; i < adapter->num_io_queues; i++) {
1986                 adapter->tx_ring[i].ring_size = new_tx_size;
1987                 adapter->rx_ring[i].ring_size = new_rx_size;
1988         }
1989 }
1990
1991 static int
1992 create_queues_with_size_backoff(struct ena_adapter *adapter)
1993 {
1994         int rc;
1995         uint32_t cur_rx_ring_size, cur_tx_ring_size;
1996         uint32_t new_rx_ring_size, new_tx_ring_size;
1997
1998         /*
1999          * Current queue sizes might be set to smaller than the requested
2000          * ones due to past queue allocation failures.
2001          */
2002         set_io_rings_size(adapter, adapter->requested_tx_ring_size,
2003             adapter->requested_rx_ring_size);
2004
2005         while (1) {
2006                 /* Allocate transmit descriptors */
2007                 rc = ena_setup_all_tx_resources(adapter);
2008                 if (unlikely(rc != 0)) {
2009                         ena_trace(NULL, ENA_ALERT, "err_setup_tx\n");
2010                         goto err_setup_tx;
2011                 }
2012
2013                 /* Allocate receive descriptors */
2014                 rc = ena_setup_all_rx_resources(adapter);
2015                 if (unlikely(rc != 0)) {
2016                         ena_trace(NULL, ENA_ALERT, "err_setup_rx\n");
2017                         goto err_setup_rx;
2018                 }
2019
2020                 /* Create IO queues for Rx & Tx */
2021                 rc = ena_create_io_queues(adapter);
2022                 if (unlikely(rc != 0)) {
2023                         ena_trace(NULL, ENA_ALERT,
2024                             "create IO queues failed\n");
2025                         goto err_io_que;
2026                 }
2027
2028                 return (0);
2029
2030 err_io_que:
2031                 ena_free_all_rx_resources(adapter);
2032 err_setup_rx:
2033                 ena_free_all_tx_resources(adapter);
2034 err_setup_tx:
2035                 /*
2036                  * Lower the ring size if ENOMEM. Otherwise, return the
2037                  * error straightaway.
2038                  */
2039                 if (unlikely(rc != ENOMEM)) {
2040                         ena_trace(NULL, ENA_ALERT,
2041                             "Queue creation failed with error code: %d\n", rc);
2042                         return (rc);
2043                 }
2044
2045                 cur_tx_ring_size = adapter->tx_ring[0].ring_size;
2046                 cur_rx_ring_size = adapter->rx_ring[0].ring_size;
2047
2048                 device_printf(adapter->pdev,
2049                     "Not enough memory to create queues with sizes TX=%d, RX=%d\n",
2050                     cur_tx_ring_size, cur_rx_ring_size);
2051
2052                 new_tx_ring_size = cur_tx_ring_size;
2053                 new_rx_ring_size = cur_rx_ring_size;
2054
2055                 /*
2056                  * Decrease the size of a larger queue, or decrease both if they are
2057                  * the same size.
2058                  */
2059                 if (cur_rx_ring_size <= cur_tx_ring_size)
2060                         new_tx_ring_size = cur_tx_ring_size / 2;
2061                 if (cur_rx_ring_size >= cur_tx_ring_size)
2062                         new_rx_ring_size = cur_rx_ring_size / 2;
2063
2064                 if (new_tx_ring_size < ENA_MIN_RING_SIZE ||
2065                     new_rx_ring_size < ENA_MIN_RING_SIZE) {
2066                         device_printf(adapter->pdev,
2067                             "Queue creation failed with the smallest possible queue size"
2068                             "of %d for both queues. Not retrying with smaller queues\n",
2069                             ENA_MIN_RING_SIZE);
2070                         return (rc);
2071                 }
2072
2073                 set_io_rings_size(adapter, new_tx_ring_size, new_rx_ring_size);
2074         }
2075 }
2076
2077 int
2078 ena_up(struct ena_adapter *adapter)
2079 {
2080         int rc = 0;
2081
2082         if (unlikely(device_is_attached(adapter->pdev) == 0)) {
2083                 device_printf(adapter->pdev, "device is not attached!\n");
2084                 return (ENXIO);
2085         }
2086
2087         if (ENA_FLAG_ISSET(ENA_FLAG_DEV_UP, adapter))
2088                 return (0);
2089
2090         device_printf(adapter->pdev, "device is going UP\n");
2091
2092         /* setup interrupts for IO queues */
2093         rc = ena_setup_io_intr(adapter);
2094         if (unlikely(rc != 0)) {
2095                 ena_trace(NULL, ENA_ALERT, "error setting up IO interrupt\n");
2096                 goto error;
2097         }
2098         rc = ena_request_io_irq(adapter);
2099         if (unlikely(rc != 0)) {
2100                 ena_trace(NULL, ENA_ALERT, "err_req_irq\n");
2101                 goto error;
2102         }
2103
2104         device_printf(adapter->pdev,
2105             "Creating %u IO queues. Rx queue size: %d, Tx queue size: %d, "
2106             "LLQ is %s\n",
2107             adapter->num_io_queues,
2108             adapter->requested_rx_ring_size,
2109             adapter->requested_tx_ring_size,
2110             (adapter->ena_dev->tx_mem_queue_type ==
2111                 ENA_ADMIN_PLACEMENT_POLICY_DEV) ?  "ENABLED" : "DISABLED");
2112
2113         rc = create_queues_with_size_backoff(adapter);
2114         if (unlikely(rc != 0)) {
2115                 ena_trace(NULL, ENA_ALERT,
2116                     "error creating queues with size backoff\n");
2117                 goto err_create_queues_with_backoff;
2118         }
2119
2120         if (ENA_FLAG_ISSET(ENA_FLAG_LINK_UP, adapter))
2121                 if_link_state_change(adapter->ifp, LINK_STATE_UP);
2122
2123         rc = ena_up_complete(adapter);
2124         if (unlikely(rc != 0))
2125                 goto err_up_complete;
2126
2127         counter_u64_add(adapter->dev_stats.interface_up, 1);
2128
2129         ena_update_hwassist(adapter);
2130
2131         if_setdrvflagbits(adapter->ifp, IFF_DRV_RUNNING,
2132                 IFF_DRV_OACTIVE);
2133
2134         /* Activate timer service only if the device is running.
2135                 * If this flag is not set, it means that the driver is being
2136                 * reset and timer service will be activated afterwards.
2137                 */
2138         if (ENA_FLAG_ISSET(ENA_FLAG_DEVICE_RUNNING, adapter)) {
2139                 callout_reset_sbt(&adapter->timer_service, SBT_1S,
2140                         SBT_1S, ena_timer_service, (void *)adapter, 0);
2141         }
2142
2143         ENA_FLAG_SET_ATOMIC(ENA_FLAG_DEV_UP, adapter);
2144
2145         ena_unmask_all_io_irqs(adapter);
2146
2147         return (0);
2148
2149 err_up_complete:
2150         ena_destroy_all_io_queues(adapter);
2151         ena_free_all_rx_resources(adapter);
2152         ena_free_all_tx_resources(adapter);
2153 err_create_queues_with_backoff:
2154         ena_free_io_irq(adapter);
2155 error:
2156         return (rc);
2157 }
2158
2159 static uint64_t
2160 ena_get_counter(if_t ifp, ift_counter cnt)
2161 {
2162         struct ena_adapter *adapter;
2163         struct ena_hw_stats *stats;
2164
2165         adapter = if_getsoftc(ifp);
2166         stats = &adapter->hw_stats;
2167
2168         switch (cnt) {
2169         case IFCOUNTER_IPACKETS:
2170                 return (counter_u64_fetch(stats->rx_packets));
2171         case IFCOUNTER_OPACKETS:
2172                 return (counter_u64_fetch(stats->tx_packets));
2173         case IFCOUNTER_IBYTES:
2174                 return (counter_u64_fetch(stats->rx_bytes));
2175         case IFCOUNTER_OBYTES:
2176                 return (counter_u64_fetch(stats->tx_bytes));
2177         case IFCOUNTER_IQDROPS:
2178                 return (counter_u64_fetch(stats->rx_drops));
2179         case IFCOUNTER_OQDROPS:
2180                 return (counter_u64_fetch(stats->tx_drops));
2181         default:
2182                 return (if_get_counter_default(ifp, cnt));
2183         }
2184 }
2185
2186 static int
2187 ena_media_change(if_t ifp)
2188 {
2189         /* Media Change is not supported by firmware */
2190         return (0);
2191 }
2192
2193 static void
2194 ena_media_status(if_t ifp, struct ifmediareq *ifmr)
2195 {
2196         struct ena_adapter *adapter = if_getsoftc(ifp);
2197         ena_trace(NULL, ENA_DBG, "enter\n");
2198
2199         ENA_LOCK_LOCK(adapter);
2200
2201         ifmr->ifm_status = IFM_AVALID;
2202         ifmr->ifm_active = IFM_ETHER;
2203
2204         if (!ENA_FLAG_ISSET(ENA_FLAG_LINK_UP, adapter)) {
2205                 ENA_LOCK_UNLOCK(adapter);
2206                 ena_trace(NULL, ENA_INFO, "Link is down\n");
2207                 return;
2208         }
2209
2210         ifmr->ifm_status |= IFM_ACTIVE;
2211         ifmr->ifm_active |= IFM_UNKNOWN | IFM_FDX;
2212
2213         ENA_LOCK_UNLOCK(adapter);
2214 }
2215
2216 static void
2217 ena_init(void *arg)
2218 {
2219         struct ena_adapter *adapter = (struct ena_adapter *)arg;
2220
2221         if (!ENA_FLAG_ISSET(ENA_FLAG_DEV_UP, adapter)) {
2222                 ENA_LOCK_LOCK(adapter);
2223                 ena_up(adapter);
2224                 ENA_LOCK_UNLOCK(adapter);
2225         }
2226 }
2227
2228 static int
2229 ena_ioctl(if_t ifp, u_long command, caddr_t data)
2230 {
2231         struct ena_adapter *adapter;
2232         struct ifreq *ifr;
2233         int rc;
2234
2235         adapter = ifp->if_softc;
2236         ifr = (struct ifreq *)data;
2237
2238         /*
2239          * Acquiring lock to prevent from running up and down routines parallel.
2240          */
2241         rc = 0;
2242         switch (command) {
2243         case SIOCSIFMTU:
2244                 if (ifp->if_mtu == ifr->ifr_mtu)
2245                         break;
2246                 ENA_LOCK_LOCK(adapter);
2247                 ena_down(adapter);
2248
2249                 ena_change_mtu(ifp, ifr->ifr_mtu);
2250
2251                 rc = ena_up(adapter);
2252                 ENA_LOCK_UNLOCK(adapter);
2253                 break;
2254
2255         case SIOCSIFFLAGS:
2256                 if ((ifp->if_flags & IFF_UP) != 0) {
2257                         if ((if_getdrvflags(ifp) & IFF_DRV_RUNNING) != 0) {
2258                                 if ((ifp->if_flags & (IFF_PROMISC |
2259                                     IFF_ALLMULTI)) != 0) {
2260                                         device_printf(adapter->pdev,
2261                                             "ioctl promisc/allmulti\n");
2262                                 }
2263                         } else {
2264                                 ENA_LOCK_LOCK(adapter);
2265                                 rc = ena_up(adapter);
2266                                 ENA_LOCK_UNLOCK(adapter);
2267                         }
2268                 } else {
2269                         if ((if_getdrvflags(ifp) & IFF_DRV_RUNNING) != 0) {
2270                                 ENA_LOCK_LOCK(adapter);
2271                                 ena_down(adapter);
2272                                 ENA_LOCK_UNLOCK(adapter);
2273                         }
2274                 }
2275                 break;
2276
2277         case SIOCADDMULTI:
2278         case SIOCDELMULTI:
2279                 break;
2280
2281         case SIOCSIFMEDIA:
2282         case SIOCGIFMEDIA:
2283                 rc = ifmedia_ioctl(ifp, ifr, &adapter->media, command);
2284                 break;
2285
2286         case SIOCSIFCAP:
2287                 {
2288                         int reinit = 0;
2289
2290                         if (ifr->ifr_reqcap != ifp->if_capenable) {
2291                                 ifp->if_capenable = ifr->ifr_reqcap;
2292                                 reinit = 1;
2293                         }
2294
2295                         if ((reinit != 0) &&
2296                             ((if_getdrvflags(ifp) & IFF_DRV_RUNNING) != 0)) {
2297                                 ENA_LOCK_LOCK(adapter);
2298                                 ena_down(adapter);
2299                                 rc = ena_up(adapter);
2300                                 ENA_LOCK_UNLOCK(adapter);
2301                         }
2302                 }
2303
2304                 break;
2305         default:
2306                 rc = ether_ioctl(ifp, command, data);
2307                 break;
2308         }
2309
2310         return (rc);
2311 }
2312
2313 static int
2314 ena_get_dev_offloads(struct ena_com_dev_get_features_ctx *feat)
2315 {
2316         int caps = 0;
2317
2318         if ((feat->offload.tx &
2319             (ENA_ADMIN_FEATURE_OFFLOAD_DESC_TX_L4_IPV4_CSUM_FULL_MASK |
2320             ENA_ADMIN_FEATURE_OFFLOAD_DESC_TX_L4_IPV4_CSUM_PART_MASK |
2321             ENA_ADMIN_FEATURE_OFFLOAD_DESC_TX_L3_CSUM_IPV4_MASK)) != 0)
2322                 caps |= IFCAP_TXCSUM;
2323
2324         if ((feat->offload.tx &
2325             (ENA_ADMIN_FEATURE_OFFLOAD_DESC_TX_L4_IPV6_CSUM_FULL_MASK |
2326             ENA_ADMIN_FEATURE_OFFLOAD_DESC_TX_L4_IPV6_CSUM_PART_MASK)) != 0)
2327                 caps |= IFCAP_TXCSUM_IPV6;
2328
2329         if ((feat->offload.tx &
2330             ENA_ADMIN_FEATURE_OFFLOAD_DESC_TSO_IPV4_MASK) != 0)
2331                 caps |= IFCAP_TSO4;
2332
2333         if ((feat->offload.tx &
2334             ENA_ADMIN_FEATURE_OFFLOAD_DESC_TSO_IPV6_MASK) != 0)
2335                 caps |= IFCAP_TSO6;
2336
2337         if ((feat->offload.rx_supported &
2338             (ENA_ADMIN_FEATURE_OFFLOAD_DESC_RX_L4_IPV4_CSUM_MASK |
2339             ENA_ADMIN_FEATURE_OFFLOAD_DESC_RX_L3_CSUM_IPV4_MASK)) != 0)
2340                 caps |= IFCAP_RXCSUM;
2341
2342         if ((feat->offload.rx_supported &
2343             ENA_ADMIN_FEATURE_OFFLOAD_DESC_RX_L4_IPV6_CSUM_MASK) != 0)
2344                 caps |= IFCAP_RXCSUM_IPV6;
2345
2346         caps |= IFCAP_LRO | IFCAP_JUMBO_MTU;
2347
2348         return (caps);
2349 }
2350
2351 static void
2352 ena_update_host_info(struct ena_admin_host_info *host_info, if_t ifp)
2353 {
2354
2355         host_info->supported_network_features[0] =
2356             (uint32_t)if_getcapabilities(ifp);
2357 }
2358
2359 static void
2360 ena_update_hwassist(struct ena_adapter *adapter)
2361 {
2362         if_t ifp = adapter->ifp;
2363         uint32_t feat = adapter->tx_offload_cap;
2364         int cap = if_getcapenable(ifp);
2365         int flags = 0;
2366
2367         if_clearhwassist(ifp);
2368
2369         if ((cap & IFCAP_TXCSUM) != 0) {
2370                 if ((feat &
2371                     ENA_ADMIN_FEATURE_OFFLOAD_DESC_TX_L3_CSUM_IPV4_MASK) != 0)
2372                         flags |= CSUM_IP;
2373                 if ((feat &
2374                     (ENA_ADMIN_FEATURE_OFFLOAD_DESC_TX_L4_IPV4_CSUM_FULL_MASK |
2375                     ENA_ADMIN_FEATURE_OFFLOAD_DESC_TX_L4_IPV4_CSUM_PART_MASK)) != 0)
2376                         flags |= CSUM_IP_UDP | CSUM_IP_TCP;
2377         }
2378
2379         if ((cap & IFCAP_TXCSUM_IPV6) != 0)
2380                 flags |= CSUM_IP6_UDP | CSUM_IP6_TCP;
2381
2382         if ((cap & IFCAP_TSO4) != 0)
2383                 flags |= CSUM_IP_TSO;
2384
2385         if ((cap & IFCAP_TSO6) != 0)
2386                 flags |= CSUM_IP6_TSO;
2387
2388         if_sethwassistbits(ifp, flags, 0);
2389 }
2390
2391 static int
2392 ena_setup_ifnet(device_t pdev, struct ena_adapter *adapter,
2393     struct ena_com_dev_get_features_ctx *feat)
2394 {
2395         if_t ifp;
2396         int caps = 0;
2397
2398         ifp = adapter->ifp = if_gethandle(IFT_ETHER);
2399         if (unlikely(ifp == NULL)) {
2400                 ena_trace(NULL, ENA_ALERT, "can not allocate ifnet structure\n");
2401                 return (ENXIO);
2402         }
2403         if_initname(ifp, device_get_name(pdev), device_get_unit(pdev));
2404         if_setdev(ifp, pdev);
2405         if_setsoftc(ifp, adapter);
2406
2407         if_setflags(ifp, IFF_BROADCAST | IFF_SIMPLEX | IFF_MULTICAST |
2408             IFF_KNOWSEPOCH);
2409         if_setinitfn(ifp, ena_init);
2410         if_settransmitfn(ifp, ena_mq_start);
2411         if_setqflushfn(ifp, ena_qflush);
2412         if_setioctlfn(ifp, ena_ioctl);
2413         if_setgetcounterfn(ifp, ena_get_counter);
2414
2415         if_setsendqlen(ifp, adapter->requested_tx_ring_size);
2416         if_setsendqready(ifp);
2417         if_setmtu(ifp, ETHERMTU);
2418         if_setbaudrate(ifp, 0);
2419         /* Zeroize capabilities... */
2420         if_setcapabilities(ifp, 0);
2421         if_setcapenable(ifp, 0);
2422         /* check hardware support */
2423         caps = ena_get_dev_offloads(feat);
2424         /* ... and set them */
2425         if_setcapabilitiesbit(ifp, caps, 0);
2426
2427         /* TSO parameters */
2428         ifp->if_hw_tsomax = ENA_TSO_MAXSIZE -
2429             (ETHER_HDR_LEN + ETHER_VLAN_ENCAP_LEN);
2430         ifp->if_hw_tsomaxsegcount = adapter->max_tx_sgl_size - 1;
2431         ifp->if_hw_tsomaxsegsize = ENA_TSO_MAXSIZE;
2432
2433         if_setifheaderlen(ifp, sizeof(struct ether_vlan_header));
2434         if_setcapenable(ifp, if_getcapabilities(ifp));
2435
2436         /*
2437          * Specify the media types supported by this adapter and register
2438          * callbacks to update media and link information
2439          */
2440         ifmedia_init(&adapter->media, IFM_IMASK,
2441             ena_media_change, ena_media_status);
2442         ifmedia_add(&adapter->media, IFM_ETHER | IFM_AUTO, 0, NULL);
2443         ifmedia_set(&adapter->media, IFM_ETHER | IFM_AUTO);
2444
2445         ether_ifattach(ifp, adapter->mac_addr);
2446
2447         return (0);
2448 }
2449
2450 void
2451 ena_down(struct ena_adapter *adapter)
2452 {
2453         int rc;
2454
2455         if (!ENA_FLAG_ISSET(ENA_FLAG_DEV_UP, adapter))
2456                 return;
2457
2458         device_printf(adapter->pdev, "device is going DOWN\n");
2459
2460         callout_drain(&adapter->timer_service);
2461
2462         ENA_FLAG_CLEAR_ATOMIC(ENA_FLAG_DEV_UP, adapter);
2463         if_setdrvflagbits(adapter->ifp, IFF_DRV_OACTIVE,
2464                 IFF_DRV_RUNNING);
2465
2466         ena_free_io_irq(adapter);
2467
2468         if (ENA_FLAG_ISSET(ENA_FLAG_TRIGGER_RESET, adapter)) {
2469                 rc = ena_com_dev_reset(adapter->ena_dev,
2470                         adapter->reset_reason);
2471                 if (unlikely(rc != 0))
2472                         device_printf(adapter->pdev,
2473                                 "Device reset failed\n");
2474         }
2475
2476         ena_destroy_all_io_queues(adapter);
2477
2478         ena_free_all_tx_bufs(adapter);
2479         ena_free_all_rx_bufs(adapter);
2480         ena_free_all_tx_resources(adapter);
2481         ena_free_all_rx_resources(adapter);
2482
2483         counter_u64_add(adapter->dev_stats.interface_down, 1);
2484 }
2485
2486 static uint32_t
2487 ena_calc_max_io_queue_num(device_t pdev, struct ena_com_dev *ena_dev,
2488     struct ena_com_dev_get_features_ctx *get_feat_ctx)
2489 {
2490         uint32_t io_tx_sq_num, io_tx_cq_num, io_rx_num, max_num_io_queues;
2491
2492         /* Regular queues capabilities */
2493         if (ena_dev->supported_features & BIT(ENA_ADMIN_MAX_QUEUES_EXT)) {
2494                 struct ena_admin_queue_ext_feature_fields *max_queue_ext =
2495                     &get_feat_ctx->max_queue_ext.max_queue_ext;
2496                 io_rx_num = min_t(int, max_queue_ext->max_rx_sq_num,
2497                         max_queue_ext->max_rx_cq_num);
2498
2499                 io_tx_sq_num = max_queue_ext->max_tx_sq_num;
2500                 io_tx_cq_num = max_queue_ext->max_tx_cq_num;
2501         } else {
2502                 struct ena_admin_queue_feature_desc *max_queues =
2503                     &get_feat_ctx->max_queues;
2504                 io_tx_sq_num = max_queues->max_sq_num;
2505                 io_tx_cq_num = max_queues->max_cq_num;
2506                 io_rx_num = min_t(int, io_tx_sq_num, io_tx_cq_num);
2507         }
2508
2509         /* In case of LLQ use the llq fields for the tx SQ/CQ */
2510         if (ena_dev->tx_mem_queue_type == ENA_ADMIN_PLACEMENT_POLICY_DEV)
2511                 io_tx_sq_num = get_feat_ctx->llq.max_llq_num;
2512
2513         max_num_io_queues = min_t(uint32_t, mp_ncpus, ENA_MAX_NUM_IO_QUEUES);
2514         max_num_io_queues = min_t(uint32_t, max_num_io_queues, io_rx_num);
2515         max_num_io_queues = min_t(uint32_t, max_num_io_queues, io_tx_sq_num);
2516         max_num_io_queues = min_t(uint32_t, max_num_io_queues, io_tx_cq_num);
2517         /* 1 IRQ for for mgmnt and 1 IRQ for each TX/RX pair */
2518         max_num_io_queues = min_t(uint32_t, max_num_io_queues,
2519             pci_msix_count(pdev) - 1);
2520
2521         return (max_num_io_queues);
2522 }
2523
2524 static int
2525 ena_enable_wc(struct resource *res)
2526 {
2527 #if defined(__i386) || defined(__amd64) || defined(__aarch64__)
2528         vm_offset_t va;
2529         vm_size_t len;
2530         int rc;
2531
2532         va = (vm_offset_t)rman_get_virtual(res);
2533         len = rman_get_size(res);
2534         /* Enable write combining */
2535         rc = pmap_change_attr(va, len, VM_MEMATTR_WRITE_COMBINING);
2536         if (unlikely(rc != 0)) {
2537                 ena_trace(NULL, ENA_ALERT, "pmap_change_attr failed, %d\n", rc);
2538                 return (rc);
2539         }
2540
2541         return (0);
2542 #endif
2543         return (EOPNOTSUPP);
2544 }
2545
2546 static int
2547 ena_set_queues_placement_policy(device_t pdev, struct ena_com_dev *ena_dev,
2548     struct ena_admin_feature_llq_desc *llq,
2549     struct ena_llq_configurations *llq_default_configurations)
2550 {
2551         struct ena_adapter *adapter = device_get_softc(pdev);
2552         int rc, rid;
2553         uint32_t llq_feature_mask;
2554
2555         llq_feature_mask = 1 << ENA_ADMIN_LLQ;
2556         if (!(ena_dev->supported_features & llq_feature_mask)) {
2557                 device_printf(pdev,
2558                     "LLQ is not supported. Fallback to host mode policy.\n");
2559                 ena_dev->tx_mem_queue_type = ENA_ADMIN_PLACEMENT_POLICY_HOST;
2560                 return (0);
2561         }
2562
2563         rc = ena_com_config_dev_mode(ena_dev, llq, llq_default_configurations);
2564         if (unlikely(rc != 0)) {
2565                 device_printf(pdev, "Failed to configure the device mode. "
2566                     "Fallback to host mode policy.\n");
2567                 ena_dev->tx_mem_queue_type = ENA_ADMIN_PLACEMENT_POLICY_HOST;
2568                 return (0);
2569         }
2570
2571         /* Nothing to config, exit */
2572         if (ena_dev->tx_mem_queue_type == ENA_ADMIN_PLACEMENT_POLICY_HOST)
2573                 return (0);
2574
2575         /* Try to allocate resources for LLQ bar */
2576         rid = PCIR_BAR(ENA_MEM_BAR);
2577         adapter->memory = bus_alloc_resource_any(pdev, SYS_RES_MEMORY,
2578             &rid, RF_ACTIVE);
2579         if (unlikely(adapter->memory == NULL)) {
2580                 device_printf(pdev, "unable to allocate LLQ bar resource. "
2581                     "Fallback to host mode policy.\n");
2582                 ena_dev->tx_mem_queue_type = ENA_ADMIN_PLACEMENT_POLICY_HOST;
2583                 return (0);
2584         }
2585
2586         /* Enable write combining for better LLQ performance */
2587         rc = ena_enable_wc(adapter->memory);
2588         if (unlikely(rc != 0)) {
2589                 device_printf(pdev, "failed to enable write combining.\n");
2590                 return (rc);
2591         }
2592
2593         /*
2594          * Save virtual address of the device's memory region
2595          * for the ena_com layer.
2596          */
2597         ena_dev->mem_bar = rman_get_virtual(adapter->memory);
2598
2599         return (0);
2600 }
2601
2602 static inline
2603 void set_default_llq_configurations(struct ena_llq_configurations *llq_config)
2604 {
2605         llq_config->llq_header_location = ENA_ADMIN_INLINE_HEADER;
2606         llq_config->llq_ring_entry_size = ENA_ADMIN_LIST_ENTRY_SIZE_128B;
2607         llq_config->llq_stride_ctrl = ENA_ADMIN_MULTIPLE_DESCS_PER_ENTRY;
2608         llq_config->llq_num_decs_before_header =
2609             ENA_ADMIN_LLQ_NUM_DESCS_BEFORE_HEADER_2;
2610         llq_config->llq_ring_entry_size_value = 128;
2611 }
2612
2613 static int
2614 ena_calc_io_queue_size(struct ena_calc_queue_size_ctx *ctx)
2615 {
2616         struct ena_admin_feature_llq_desc *llq = &ctx->get_feat_ctx->llq;
2617         struct ena_com_dev *ena_dev = ctx->ena_dev;
2618         uint32_t tx_queue_size = ENA_DEFAULT_RING_SIZE;
2619         uint32_t rx_queue_size = ENA_DEFAULT_RING_SIZE;
2620         uint32_t max_tx_queue_size;
2621         uint32_t max_rx_queue_size;
2622
2623         if (ena_dev->supported_features & BIT(ENA_ADMIN_MAX_QUEUES_EXT)) {
2624                 struct ena_admin_queue_ext_feature_fields *max_queue_ext =
2625                     &ctx->get_feat_ctx->max_queue_ext.max_queue_ext;
2626                 max_rx_queue_size = min_t(uint32_t,
2627                     max_queue_ext->max_rx_cq_depth,
2628                     max_queue_ext->max_rx_sq_depth);
2629                 max_tx_queue_size = max_queue_ext->max_tx_cq_depth;
2630
2631                 if (ena_dev->tx_mem_queue_type ==
2632                     ENA_ADMIN_PLACEMENT_POLICY_DEV)
2633                         max_tx_queue_size = min_t(uint32_t, max_tx_queue_size,
2634                             llq->max_llq_depth);
2635                 else
2636                         max_tx_queue_size = min_t(uint32_t, max_tx_queue_size,
2637                             max_queue_ext->max_tx_sq_depth);
2638
2639                 ctx->max_tx_sgl_size = min_t(uint16_t, ENA_PKT_MAX_BUFS,
2640                     max_queue_ext->max_per_packet_tx_descs);
2641                 ctx->max_rx_sgl_size = min_t(uint16_t, ENA_PKT_MAX_BUFS,
2642                     max_queue_ext->max_per_packet_rx_descs);
2643         } else {
2644                 struct ena_admin_queue_feature_desc *max_queues =
2645                     &ctx->get_feat_ctx->max_queues;
2646                 max_rx_queue_size = min_t(uint32_t,
2647                     max_queues->max_cq_depth,
2648                     max_queues->max_sq_depth);
2649                 max_tx_queue_size = max_queues->max_cq_depth;
2650
2651                 if (ena_dev->tx_mem_queue_type ==
2652                     ENA_ADMIN_PLACEMENT_POLICY_DEV)
2653                         max_tx_queue_size = min_t(uint32_t, max_tx_queue_size,
2654                             llq->max_llq_depth);
2655                 else
2656                         max_tx_queue_size = min_t(uint32_t, max_tx_queue_size,
2657                             max_queues->max_sq_depth);
2658
2659                 ctx->max_tx_sgl_size = min_t(uint16_t, ENA_PKT_MAX_BUFS,
2660                     max_queues->max_packet_tx_descs);
2661                 ctx->max_rx_sgl_size = min_t(uint16_t, ENA_PKT_MAX_BUFS,
2662                     max_queues->max_packet_rx_descs);
2663         }
2664
2665         /* round down to the nearest power of 2 */
2666         max_tx_queue_size = 1 << (flsl(max_tx_queue_size) - 1);
2667         max_rx_queue_size = 1 << (flsl(max_rx_queue_size) - 1);
2668
2669         tx_queue_size = clamp_val(tx_queue_size, ENA_MIN_RING_SIZE,
2670             max_tx_queue_size);
2671         rx_queue_size = clamp_val(rx_queue_size, ENA_MIN_RING_SIZE,
2672             max_rx_queue_size);
2673
2674         tx_queue_size = 1 << (flsl(tx_queue_size) - 1);
2675         rx_queue_size = 1 << (flsl(rx_queue_size) - 1);
2676
2677         ctx->max_tx_queue_size = max_tx_queue_size;
2678         ctx->max_rx_queue_size = max_rx_queue_size;
2679         ctx->tx_queue_size = tx_queue_size;
2680         ctx->rx_queue_size = rx_queue_size;
2681
2682         return (0);
2683 }
2684
2685 static int
2686 ena_rss_init_default(struct ena_adapter *adapter)
2687 {
2688         struct ena_com_dev *ena_dev = adapter->ena_dev;
2689         device_t dev = adapter->pdev;
2690         int qid, rc, i;
2691
2692         rc = ena_com_rss_init(ena_dev, ENA_RX_RSS_TABLE_LOG_SIZE);
2693         if (unlikely(rc != 0)) {
2694                 device_printf(dev, "Cannot init indirect table\n");
2695                 return (rc);
2696         }
2697
2698         for (i = 0; i < ENA_RX_RSS_TABLE_SIZE; i++) {
2699                 qid = i % adapter->num_io_queues;
2700                 rc = ena_com_indirect_table_fill_entry(ena_dev, i,
2701                     ENA_IO_RXQ_IDX(qid));
2702                 if (unlikely((rc != 0) && (rc != EOPNOTSUPP))) {
2703                         device_printf(dev, "Cannot fill indirect table\n");
2704                         goto err_rss_destroy;
2705                 }
2706         }
2707
2708 #ifdef RSS
2709         uint8_t rss_algo = rss_gethashalgo();
2710         if (rss_algo == RSS_HASH_TOEPLITZ) {
2711                 uint8_t hash_key[RSS_KEYSIZE];
2712
2713                 rss_getkey(hash_key);
2714                 rc = ena_com_fill_hash_function(ena_dev, ENA_ADMIN_TOEPLITZ,
2715                     hash_key, RSS_KEYSIZE, 0xFFFFFFFF);
2716         } else
2717 #endif
2718         rc = ena_com_fill_hash_function(ena_dev, ENA_ADMIN_CRC32, NULL,
2719             ENA_HASH_KEY_SIZE, 0xFFFFFFFF);
2720         if (unlikely((rc != 0) && (rc != EOPNOTSUPP))) {
2721                 device_printf(dev, "Cannot fill hash function\n");
2722                 goto err_rss_destroy;
2723         }
2724
2725         rc = ena_com_set_default_hash_ctrl(ena_dev);
2726         if (unlikely((rc != 0) && (rc != EOPNOTSUPP))) {
2727                 device_printf(dev, "Cannot fill hash control\n");
2728                 goto err_rss_destroy;
2729         }
2730
2731         return (0);
2732
2733 err_rss_destroy:
2734         ena_com_rss_destroy(ena_dev);
2735         return (rc);
2736 }
2737
2738 static void
2739 ena_rss_init_default_deferred(void *arg)
2740 {
2741         struct ena_adapter *adapter;
2742         devclass_t dc;
2743         int max;
2744         int rc;
2745
2746         dc = devclass_find("ena");
2747         if (unlikely(dc == NULL)) {
2748                 ena_trace(NULL, ENA_ALERT, "No devclass ena\n");
2749                 return;
2750         }
2751
2752         max = devclass_get_maxunit(dc);
2753         while (max-- >= 0) {
2754                 adapter = devclass_get_softc(dc, max);
2755                 if (adapter != NULL) {
2756                         rc = ena_rss_init_default(adapter);
2757                         ENA_FLAG_SET_ATOMIC(ENA_FLAG_RSS_ACTIVE, adapter);
2758                         if (unlikely(rc != 0)) {
2759                                 device_printf(adapter->pdev,
2760                                     "WARNING: RSS was not properly initialized,"
2761                                     " it will affect bandwidth\n");
2762                                 ENA_FLAG_CLEAR_ATOMIC(ENA_FLAG_RSS_ACTIVE, adapter);
2763                         }
2764                 }
2765         }
2766 }
2767 SYSINIT(ena_rss_init, SI_SUB_KICK_SCHEDULER, SI_ORDER_SECOND, ena_rss_init_default_deferred, NULL);
2768
2769 static void
2770 ena_config_host_info(struct ena_com_dev *ena_dev, device_t dev)
2771 {
2772         struct ena_admin_host_info *host_info;
2773         uintptr_t rid;
2774         int rc;
2775
2776         /* Allocate only the host info */
2777         rc = ena_com_allocate_host_info(ena_dev);
2778         if (unlikely(rc != 0)) {
2779                 ena_trace(NULL, ENA_ALERT, "Cannot allocate host info\n");
2780                 return;
2781         }
2782
2783         host_info = ena_dev->host_attr.host_info;
2784
2785         if (pci_get_id(dev, PCI_ID_RID, &rid) == 0)
2786                 host_info->bdf = rid;
2787         host_info->os_type = ENA_ADMIN_OS_FREEBSD;
2788         host_info->kernel_ver = osreldate;
2789
2790         sprintf(host_info->kernel_ver_str, "%d", osreldate);
2791         host_info->os_dist = 0;
2792         strncpy(host_info->os_dist_str, osrelease,
2793             sizeof(host_info->os_dist_str) - 1);
2794
2795         host_info->driver_version =
2796                 (DRV_MODULE_VER_MAJOR) |
2797                 (DRV_MODULE_VER_MINOR << ENA_ADMIN_HOST_INFO_MINOR_SHIFT) |
2798                 (DRV_MODULE_VER_SUBMINOR << ENA_ADMIN_HOST_INFO_SUB_MINOR_SHIFT);
2799         host_info->num_cpus = mp_ncpus;
2800
2801         rc = ena_com_set_host_attributes(ena_dev);
2802         if (unlikely(rc != 0)) {
2803                 if (rc == EOPNOTSUPP)
2804                         ena_trace(NULL, ENA_WARNING, "Cannot set host attributes\n");
2805                 else
2806                         ena_trace(NULL, ENA_ALERT, "Cannot set host attributes\n");
2807
2808                 goto err;
2809         }
2810
2811         return;
2812
2813 err:
2814         ena_com_delete_host_info(ena_dev);
2815 }
2816
2817 static int
2818 ena_device_init(struct ena_adapter *adapter, device_t pdev,
2819     struct ena_com_dev_get_features_ctx *get_feat_ctx, int *wd_active)
2820 {
2821         struct ena_com_dev* ena_dev = adapter->ena_dev;
2822         bool readless_supported;
2823         uint32_t aenq_groups;
2824         int dma_width;
2825         int rc;
2826
2827         rc = ena_com_mmio_reg_read_request_init(ena_dev);
2828         if (unlikely(rc != 0)) {
2829                 device_printf(pdev, "failed to init mmio read less\n");
2830                 return (rc);
2831         }
2832
2833         /*
2834          * The PCIe configuration space revision id indicate if mmio reg
2835          * read is disabled
2836          */
2837         readless_supported = !(pci_get_revid(pdev) & ENA_MMIO_DISABLE_REG_READ);
2838         ena_com_set_mmio_read_mode(ena_dev, readless_supported);
2839
2840         rc = ena_com_dev_reset(ena_dev, ENA_REGS_RESET_NORMAL);
2841         if (unlikely(rc != 0)) {
2842                 device_printf(pdev, "Can not reset device\n");
2843                 goto err_mmio_read_less;
2844         }
2845
2846         rc = ena_com_validate_version(ena_dev);
2847         if (unlikely(rc != 0)) {
2848                 device_printf(pdev, "device version is too low\n");
2849                 goto err_mmio_read_less;
2850         }
2851
2852         dma_width = ena_com_get_dma_width(ena_dev);
2853         if (unlikely(dma_width < 0)) {
2854                 device_printf(pdev, "Invalid dma width value %d", dma_width);
2855                 rc = dma_width;
2856                 goto err_mmio_read_less;
2857         }
2858         adapter->dma_width = dma_width;
2859
2860         /* ENA admin level init */
2861         rc = ena_com_admin_init(ena_dev, &aenq_handlers);
2862         if (unlikely(rc != 0)) {
2863                 device_printf(pdev,
2864                     "Can not initialize ena admin queue with device\n");
2865                 goto err_mmio_read_less;
2866         }
2867
2868         /*
2869          * To enable the msix interrupts the driver needs to know the number
2870          * of queues. So the driver uses polling mode to retrieve this
2871          * information
2872          */
2873         ena_com_set_admin_polling_mode(ena_dev, true);
2874
2875         ena_config_host_info(ena_dev, pdev);
2876
2877         /* Get Device Attributes */
2878         rc = ena_com_get_dev_attr_feat(ena_dev, get_feat_ctx);
2879         if (unlikely(rc != 0)) {
2880                 device_printf(pdev,
2881                     "Cannot get attribute for ena device rc: %d\n", rc);
2882                 goto err_admin_init;
2883         }
2884
2885         aenq_groups = BIT(ENA_ADMIN_LINK_CHANGE) |
2886             BIT(ENA_ADMIN_FATAL_ERROR) |
2887             BIT(ENA_ADMIN_WARNING) |
2888             BIT(ENA_ADMIN_NOTIFICATION) |
2889             BIT(ENA_ADMIN_KEEP_ALIVE);
2890
2891         aenq_groups &= get_feat_ctx->aenq.supported_groups;
2892         rc = ena_com_set_aenq_config(ena_dev, aenq_groups);
2893         if (unlikely(rc != 0)) {
2894                 device_printf(pdev, "Cannot configure aenq groups rc: %d\n", rc);
2895                 goto err_admin_init;
2896         }
2897
2898         *wd_active = !!(aenq_groups & BIT(ENA_ADMIN_KEEP_ALIVE));
2899
2900         return (0);
2901
2902 err_admin_init:
2903         ena_com_delete_host_info(ena_dev);
2904         ena_com_admin_destroy(ena_dev);
2905 err_mmio_read_less:
2906         ena_com_mmio_reg_read_request_destroy(ena_dev);
2907
2908         return (rc);
2909 }
2910
2911 static int ena_enable_msix_and_set_admin_interrupts(struct ena_adapter *adapter)
2912 {
2913         struct ena_com_dev *ena_dev = adapter->ena_dev;
2914         int rc;
2915
2916         rc = ena_enable_msix(adapter);
2917         if (unlikely(rc != 0)) {
2918                 device_printf(adapter->pdev, "Error with MSI-X enablement\n");
2919                 return (rc);
2920         }
2921
2922         ena_setup_mgmnt_intr(adapter);
2923
2924         rc = ena_request_mgmnt_irq(adapter);
2925         if (unlikely(rc != 0)) {
2926                 device_printf(adapter->pdev, "Cannot setup mgmnt queue intr\n");
2927                 goto err_disable_msix;
2928         }
2929
2930         ena_com_set_admin_polling_mode(ena_dev, false);
2931
2932         ena_com_admin_aenq_enable(ena_dev);
2933
2934         return (0);
2935
2936 err_disable_msix:
2937         ena_disable_msix(adapter);
2938
2939         return (rc);
2940 }
2941
2942 /* Function called on ENA_ADMIN_KEEP_ALIVE event */
2943 static void ena_keep_alive_wd(void *adapter_data,
2944     struct ena_admin_aenq_entry *aenq_e)
2945 {
2946         struct ena_adapter *adapter = (struct ena_adapter *)adapter_data;
2947         struct ena_admin_aenq_keep_alive_desc *desc;
2948         sbintime_t stime;
2949         uint64_t rx_drops;
2950         uint64_t tx_drops;
2951
2952         desc = (struct ena_admin_aenq_keep_alive_desc *)aenq_e;
2953
2954         rx_drops = ((uint64_t)desc->rx_drops_high << 32) | desc->rx_drops_low;
2955         tx_drops = ((uint64_t)desc->tx_drops_high << 32) | desc->tx_drops_low;
2956         counter_u64_zero(adapter->hw_stats.rx_drops);
2957         counter_u64_add(adapter->hw_stats.rx_drops, rx_drops);
2958         counter_u64_zero(adapter->hw_stats.tx_drops);
2959         counter_u64_add(adapter->hw_stats.tx_drops, tx_drops);
2960
2961         stime = getsbinuptime();
2962         atomic_store_rel_64(&adapter->keep_alive_timestamp, stime);
2963 }
2964
2965 /* Check for keep alive expiration */
2966 static void check_for_missing_keep_alive(struct ena_adapter *adapter)
2967 {
2968         sbintime_t timestamp, time;
2969
2970         if (adapter->wd_active == 0)
2971                 return;
2972
2973         if (adapter->keep_alive_timeout == ENA_HW_HINTS_NO_TIMEOUT)
2974                 return;
2975
2976         timestamp = atomic_load_acq_64(&adapter->keep_alive_timestamp);
2977         time = getsbinuptime() - timestamp;
2978         if (unlikely(time > adapter->keep_alive_timeout)) {
2979                 device_printf(adapter->pdev,
2980                     "Keep alive watchdog timeout.\n");
2981                 counter_u64_add(adapter->dev_stats.wd_expired, 1);
2982                 ena_trigger_reset(adapter, ENA_REGS_RESET_KEEP_ALIVE_TO);
2983         }
2984 }
2985
2986 /* Check if admin queue is enabled */
2987 static void check_for_admin_com_state(struct ena_adapter *adapter)
2988 {
2989         if (unlikely(ena_com_get_admin_running_state(adapter->ena_dev) ==
2990             false)) {
2991                 device_printf(adapter->pdev,
2992                     "ENA admin queue is not in running state!\n");
2993                 counter_u64_add(adapter->dev_stats.admin_q_pause, 1);
2994                 ena_trigger_reset(adapter, ENA_REGS_RESET_ADMIN_TO);
2995         }
2996 }
2997
2998 static int
2999 check_for_rx_interrupt_queue(struct ena_adapter *adapter,
3000     struct ena_ring *rx_ring)
3001 {
3002         if (likely(rx_ring->first_interrupt))
3003                 return (0);
3004
3005         if (ena_com_cq_empty(rx_ring->ena_com_io_cq))
3006                 return (0);
3007
3008         rx_ring->no_interrupt_event_cnt++;
3009
3010         if (rx_ring->no_interrupt_event_cnt == ENA_MAX_NO_INTERRUPT_ITERATIONS) {
3011                 device_printf(adapter->pdev, "Potential MSIX issue on Rx side "
3012                     "Queue = %d. Reset the device\n", rx_ring->qid);
3013                 ena_trigger_reset(adapter, ENA_REGS_RESET_MISS_INTERRUPT);
3014                 return (EIO);
3015         }
3016
3017         return (0);
3018 }
3019
3020 static int
3021 check_missing_comp_in_tx_queue(struct ena_adapter *adapter,
3022     struct ena_ring *tx_ring)
3023 {
3024         struct bintime curtime, time;
3025         struct ena_tx_buffer *tx_buf;
3026         sbintime_t time_offset;
3027         uint32_t missed_tx = 0;
3028         int i, rc = 0;
3029
3030         getbinuptime(&curtime);
3031
3032         for (i = 0; i < tx_ring->ring_size; i++) {
3033                 tx_buf = &tx_ring->tx_buffer_info[i];
3034
3035                 if (bintime_isset(&tx_buf->timestamp) == 0)
3036                         continue;
3037
3038                 time = curtime;
3039                 bintime_sub(&time, &tx_buf->timestamp);
3040                 time_offset = bttosbt(time);
3041
3042                 if (unlikely(!tx_ring->first_interrupt &&
3043                     time_offset > 2 * adapter->missing_tx_timeout)) {
3044                         /*
3045                          * If after graceful period interrupt is still not
3046                          * received, we schedule a reset.
3047                          */
3048                         device_printf(adapter->pdev,
3049                             "Potential MSIX issue on Tx side Queue = %d. "
3050                             "Reset the device\n", tx_ring->qid);
3051                         ena_trigger_reset(adapter,
3052                             ENA_REGS_RESET_MISS_INTERRUPT);
3053                         return (EIO);
3054                 }
3055
3056                 /* Check again if packet is still waiting */
3057                 if (unlikely(time_offset > adapter->missing_tx_timeout)) {
3058
3059                         if (!tx_buf->print_once)
3060                                 ena_trace(NULL, ENA_WARNING, "Found a Tx that wasn't "
3061                                     "completed on time, qid %d, index %d.\n",
3062                                     tx_ring->qid, i);
3063
3064                         tx_buf->print_once = true;
3065                         missed_tx++;
3066                 }
3067         }
3068
3069         if (unlikely(missed_tx > adapter->missing_tx_threshold)) {
3070                 device_printf(adapter->pdev,
3071                     "The number of lost tx completion is above the threshold "
3072                     "(%d > %d). Reset the device\n",
3073                     missed_tx, adapter->missing_tx_threshold);
3074                 ena_trigger_reset(adapter, ENA_REGS_RESET_MISS_TX_CMPL);
3075                 rc = EIO;
3076         }
3077
3078         counter_u64_add(tx_ring->tx_stats.missing_tx_comp, missed_tx);
3079
3080         return (rc);
3081 }
3082
3083 /*
3084  * Check for TX which were not completed on time.
3085  * Timeout is defined by "missing_tx_timeout".
3086  * Reset will be performed if number of incompleted
3087  * transactions exceeds "missing_tx_threshold".
3088  */
3089 static void
3090 check_for_missing_completions(struct ena_adapter *adapter)
3091 {
3092         struct ena_ring *tx_ring;
3093         struct ena_ring *rx_ring;
3094         int i, budget, rc;
3095
3096         /* Make sure the driver doesn't turn the device in other process */
3097         rmb();
3098
3099         if (!ENA_FLAG_ISSET(ENA_FLAG_DEV_UP, adapter))
3100                 return;
3101
3102         if (ENA_FLAG_ISSET(ENA_FLAG_TRIGGER_RESET, adapter))
3103                 return;
3104
3105         if (adapter->missing_tx_timeout == ENA_HW_HINTS_NO_TIMEOUT)
3106                 return;
3107
3108         budget = adapter->missing_tx_max_queues;
3109
3110         for (i = adapter->next_monitored_tx_qid; i < adapter->num_io_queues; i++) {
3111                 tx_ring = &adapter->tx_ring[i];
3112                 rx_ring = &adapter->rx_ring[i];
3113
3114                 rc = check_missing_comp_in_tx_queue(adapter, tx_ring);
3115                 if (unlikely(rc != 0))
3116                         return;
3117
3118                 rc = check_for_rx_interrupt_queue(adapter, rx_ring);
3119                 if (unlikely(rc != 0))
3120                         return;
3121
3122                 budget--;
3123                 if (budget == 0) {
3124                         i++;
3125                         break;
3126                 }
3127         }
3128
3129         adapter->next_monitored_tx_qid = i % adapter->num_io_queues;
3130 }
3131
3132 /* trigger rx cleanup after 2 consecutive detections */
3133 #define EMPTY_RX_REFILL 2
3134 /* For the rare case where the device runs out of Rx descriptors and the
3135  * msix handler failed to refill new Rx descriptors (due to a lack of memory
3136  * for example).
3137  * This case will lead to a deadlock:
3138  * The device won't send interrupts since all the new Rx packets will be dropped
3139  * The msix handler won't allocate new Rx descriptors so the device won't be
3140  * able to send new packets.
3141  *
3142  * When such a situation is detected - execute rx cleanup task in another thread
3143  */
3144 static void
3145 check_for_empty_rx_ring(struct ena_adapter *adapter)
3146 {
3147         struct ena_ring *rx_ring;
3148         int i, refill_required;
3149
3150         if (!ENA_FLAG_ISSET(ENA_FLAG_DEV_UP, adapter))
3151                 return;
3152
3153         if (ENA_FLAG_ISSET(ENA_FLAG_TRIGGER_RESET, adapter))
3154                 return;
3155
3156         for (i = 0; i < adapter->num_io_queues; i++) {
3157                 rx_ring = &adapter->rx_ring[i];
3158
3159                 refill_required = ena_com_free_q_entries(rx_ring->ena_com_io_sq);
3160                 if (unlikely(refill_required == (rx_ring->ring_size - 1))) {
3161                         rx_ring->empty_rx_queue++;
3162
3163                         if (rx_ring->empty_rx_queue >= EMPTY_RX_REFILL) {
3164                                 counter_u64_add(rx_ring->rx_stats.empty_rx_ring,
3165                                     1);
3166
3167                                 device_printf(adapter->pdev,
3168                                     "trigger refill for ring %d\n", i);
3169
3170                                 taskqueue_enqueue(rx_ring->que->cleanup_tq,
3171                                     &rx_ring->que->cleanup_task);
3172                                 rx_ring->empty_rx_queue = 0;
3173                         }
3174                 } else {
3175                         rx_ring->empty_rx_queue = 0;
3176                 }
3177         }
3178 }
3179
3180 static void ena_update_hints(struct ena_adapter *adapter,
3181                              struct ena_admin_ena_hw_hints *hints)
3182 {
3183         struct ena_com_dev *ena_dev = adapter->ena_dev;
3184
3185         if (hints->admin_completion_tx_timeout)
3186                 ena_dev->admin_queue.completion_timeout =
3187                     hints->admin_completion_tx_timeout * 1000;
3188
3189         if (hints->mmio_read_timeout)
3190                 /* convert to usec */
3191                 ena_dev->mmio_read.reg_read_to =
3192                     hints->mmio_read_timeout * 1000;
3193
3194         if (hints->missed_tx_completion_count_threshold_to_reset)
3195                 adapter->missing_tx_threshold =
3196                     hints->missed_tx_completion_count_threshold_to_reset;
3197
3198         if (hints->missing_tx_completion_timeout) {
3199                 if (hints->missing_tx_completion_timeout ==
3200                      ENA_HW_HINTS_NO_TIMEOUT)
3201                         adapter->missing_tx_timeout = ENA_HW_HINTS_NO_TIMEOUT;
3202                 else
3203                         adapter->missing_tx_timeout =
3204                             SBT_1MS * hints->missing_tx_completion_timeout;
3205         }
3206
3207         if (hints->driver_watchdog_timeout) {
3208                 if (hints->driver_watchdog_timeout == ENA_HW_HINTS_NO_TIMEOUT)
3209                         adapter->keep_alive_timeout = ENA_HW_HINTS_NO_TIMEOUT;
3210                 else
3211                         adapter->keep_alive_timeout =
3212                             SBT_1MS * hints->driver_watchdog_timeout;
3213         }
3214 }
3215
3216 static void
3217 ena_timer_service(void *data)
3218 {
3219         struct ena_adapter *adapter = (struct ena_adapter *)data;
3220         struct ena_admin_host_info *host_info =
3221             adapter->ena_dev->host_attr.host_info;
3222
3223         check_for_missing_keep_alive(adapter);
3224
3225         check_for_admin_com_state(adapter);
3226
3227         check_for_missing_completions(adapter);
3228
3229         check_for_empty_rx_ring(adapter);
3230
3231         if (host_info != NULL)
3232                 ena_update_host_info(host_info, adapter->ifp);
3233
3234         if (unlikely(ENA_FLAG_ISSET(ENA_FLAG_TRIGGER_RESET, adapter))) {
3235                 device_printf(adapter->pdev, "Trigger reset is on\n");
3236                 taskqueue_enqueue(adapter->reset_tq, &adapter->reset_task);
3237                 return;
3238         }
3239
3240         /*
3241          * Schedule another timeout one second from now.
3242          */
3243         callout_schedule_sbt(&adapter->timer_service, SBT_1S, SBT_1S, 0);
3244 }
3245
3246 void
3247 ena_destroy_device(struct ena_adapter *adapter, bool graceful)
3248 {
3249         if_t ifp = adapter->ifp;
3250         struct ena_com_dev *ena_dev = adapter->ena_dev;
3251         bool dev_up;
3252
3253         if (!ENA_FLAG_ISSET(ENA_FLAG_DEVICE_RUNNING, adapter))
3254                 return;
3255
3256         if_link_state_change(ifp, LINK_STATE_DOWN);
3257
3258         callout_drain(&adapter->timer_service);
3259
3260         dev_up = ENA_FLAG_ISSET(ENA_FLAG_DEV_UP, adapter);
3261         if (dev_up)
3262                 ENA_FLAG_SET_ATOMIC(ENA_FLAG_DEV_UP_BEFORE_RESET, adapter);
3263
3264         if (!graceful)
3265                 ena_com_set_admin_running_state(ena_dev, false);
3266
3267         if (ENA_FLAG_ISSET(ENA_FLAG_DEV_UP, adapter))
3268                 ena_down(adapter);
3269
3270         /*
3271          * Stop the device from sending AENQ events (if the device was up, and
3272          * the trigger reset was on, ena_down already performs device reset)
3273          */
3274         if (!(ENA_FLAG_ISSET(ENA_FLAG_TRIGGER_RESET, adapter) && dev_up))
3275                 ena_com_dev_reset(adapter->ena_dev, adapter->reset_reason);
3276
3277         ena_free_mgmnt_irq(adapter);
3278
3279         ena_disable_msix(adapter);
3280
3281         /*
3282          * IO rings resources should be freed because `ena_restore_device()`
3283          * calls (not directly) `ena_enable_msix()`, which re-allocates MSIX
3284          * vectors. The amount of MSIX vectors after destroy-restore may be
3285          * different than before. Therefore, IO rings resources should be
3286          * established from scratch each time.
3287          */
3288         ena_free_all_io_rings_resources(adapter);
3289
3290         ena_com_abort_admin_commands(ena_dev);
3291
3292         ena_com_wait_for_abort_completion(ena_dev);
3293
3294         ena_com_admin_destroy(ena_dev);
3295
3296         ena_com_mmio_reg_read_request_destroy(ena_dev);
3297
3298         adapter->reset_reason = ENA_REGS_RESET_NORMAL;
3299
3300         ENA_FLAG_CLEAR_ATOMIC(ENA_FLAG_TRIGGER_RESET, adapter);
3301         ENA_FLAG_CLEAR_ATOMIC(ENA_FLAG_DEVICE_RUNNING, adapter);
3302 }
3303
3304 static int
3305 ena_device_validate_params(struct ena_adapter *adapter,
3306     struct ena_com_dev_get_features_ctx *get_feat_ctx)
3307 {
3308
3309         if (memcmp(get_feat_ctx->dev_attr.mac_addr, adapter->mac_addr,
3310             ETHER_ADDR_LEN) != 0) {
3311                 device_printf(adapter->pdev,
3312                     "Error, mac address are different\n");
3313                 return (EINVAL);
3314         }
3315
3316         if (get_feat_ctx->dev_attr.max_mtu < if_getmtu(adapter->ifp)) {
3317                 device_printf(adapter->pdev,
3318                     "Error, device max mtu is smaller than ifp MTU\n");
3319                 return (EINVAL);
3320         }
3321
3322         return 0;
3323 }
3324
3325 int
3326 ena_restore_device(struct ena_adapter *adapter)
3327 {
3328         struct ena_com_dev_get_features_ctx get_feat_ctx;
3329         struct ena_com_dev *ena_dev = adapter->ena_dev;
3330         if_t ifp = adapter->ifp;
3331         device_t dev = adapter->pdev;
3332         int wd_active;
3333         int rc;
3334
3335         ENA_FLAG_SET_ATOMIC(ENA_FLAG_ONGOING_RESET, adapter);
3336
3337         rc = ena_device_init(adapter, dev, &get_feat_ctx, &wd_active);
3338         if (rc != 0) {
3339                 device_printf(dev, "Cannot initialize device\n");
3340                 goto err;
3341         }
3342         /*
3343          * Only enable WD if it was enabled before reset, so it won't override
3344          * value set by the user by the sysctl.
3345          */
3346         if (adapter->wd_active != 0)
3347                 adapter->wd_active = wd_active;
3348
3349         rc = ena_device_validate_params(adapter, &get_feat_ctx);
3350         if (rc != 0) {
3351                 device_printf(dev, "Validation of device parameters failed\n");
3352                 goto err_device_destroy;
3353         }
3354
3355         ENA_FLAG_CLEAR_ATOMIC(ENA_FLAG_ONGOING_RESET, adapter);
3356         /* Make sure we don't have a race with AENQ Links state handler */
3357         if (ENA_FLAG_ISSET(ENA_FLAG_LINK_UP, adapter))
3358                 if_link_state_change(ifp, LINK_STATE_UP);
3359
3360         rc = ena_enable_msix_and_set_admin_interrupts(adapter);
3361         if (rc != 0) {
3362                 device_printf(dev, "Enable MSI-X failed\n");
3363                 goto err_device_destroy;
3364         }
3365
3366         /*
3367          * Effective value of used MSIX vectors should be the same as before
3368          * `ena_destroy_device()`, if possible, or closest to it if less vectors
3369          * are available.
3370          */
3371         if ((adapter->msix_vecs - ENA_ADMIN_MSIX_VEC) < adapter->num_io_queues)
3372                 adapter->num_io_queues =
3373                     adapter->msix_vecs - ENA_ADMIN_MSIX_VEC;
3374
3375         /* Re-initialize rings basic information */
3376         ena_init_io_rings(adapter);
3377
3378         /* If the interface was up before the reset bring it up */
3379         if (ENA_FLAG_ISSET(ENA_FLAG_DEV_UP_BEFORE_RESET, adapter)) {
3380                 rc = ena_up(adapter);
3381                 if (rc != 0) {
3382                         device_printf(dev, "Failed to create I/O queues\n");
3383                         goto err_disable_msix;
3384                 }
3385         }
3386
3387         /* Indicate that device is running again and ready to work */
3388         ENA_FLAG_SET_ATOMIC(ENA_FLAG_DEVICE_RUNNING, adapter);
3389
3390         if (ENA_FLAG_ISSET(ENA_FLAG_DEV_UP_BEFORE_RESET, adapter)) {
3391                 /*
3392                  * As the AENQ handlers weren't executed during reset because
3393                  * the flag ENA_FLAG_DEVICE_RUNNING was turned off, the
3394                  * timestamp must be updated again That will prevent next reset
3395                  * caused by missing keep alive.
3396                  */
3397                 adapter->keep_alive_timestamp = getsbinuptime();
3398                 callout_reset_sbt(&adapter->timer_service, SBT_1S, SBT_1S,
3399                     ena_timer_service, (void *)adapter, 0);
3400         }
3401         ENA_FLAG_CLEAR_ATOMIC(ENA_FLAG_DEV_UP_BEFORE_RESET, adapter);
3402
3403         device_printf(dev,
3404             "Device reset completed successfully, Driver info: %s\n", ena_version);
3405
3406         return (rc);
3407
3408 err_disable_msix:
3409         ena_free_mgmnt_irq(adapter);
3410         ena_disable_msix(adapter);
3411 err_device_destroy:
3412         ena_com_abort_admin_commands(ena_dev);
3413         ena_com_wait_for_abort_completion(ena_dev);
3414         ena_com_admin_destroy(ena_dev);
3415         ena_com_dev_reset(ena_dev, ENA_REGS_RESET_DRIVER_INVALID_STATE);
3416         ena_com_mmio_reg_read_request_destroy(ena_dev);
3417 err:
3418         ENA_FLAG_CLEAR_ATOMIC(ENA_FLAG_DEVICE_RUNNING, adapter);
3419         ENA_FLAG_CLEAR_ATOMIC(ENA_FLAG_ONGOING_RESET, adapter);
3420         device_printf(dev, "Reset attempt failed. Can not reset the device\n");
3421
3422         return (rc);
3423 }
3424
3425 static void
3426 ena_reset_task(void *arg, int pending)
3427 {
3428         struct ena_adapter *adapter = (struct ena_adapter *)arg;
3429
3430         if (unlikely(!ENA_FLAG_ISSET(ENA_FLAG_TRIGGER_RESET, adapter))) {
3431                 device_printf(adapter->pdev,
3432                     "device reset scheduled but trigger_reset is off\n");
3433                 return;
3434         }
3435
3436         ENA_LOCK_LOCK(adapter);
3437         ena_destroy_device(adapter, false);
3438         ena_restore_device(adapter);
3439         ENA_LOCK_UNLOCK(adapter);
3440 }
3441
3442 /**
3443  * ena_attach - Device Initialization Routine
3444  * @pdev: device information struct
3445  *
3446  * Returns 0 on success, otherwise on failure.
3447  *
3448  * ena_attach initializes an adapter identified by a device structure.
3449  * The OS initialization, configuring of the adapter private structure,
3450  * and a hardware reset occur.
3451  **/
3452 static int
3453 ena_attach(device_t pdev)
3454 {
3455         struct ena_com_dev_get_features_ctx get_feat_ctx;
3456         struct ena_llq_configurations llq_config;
3457         struct ena_calc_queue_size_ctx calc_queue_ctx = { 0 };
3458         static int version_printed;
3459         struct ena_adapter *adapter;
3460         struct ena_com_dev *ena_dev = NULL;
3461         uint32_t max_num_io_queues;
3462         int rid, rc;
3463
3464         adapter = device_get_softc(pdev);
3465         adapter->pdev = pdev;
3466
3467         ENA_LOCK_INIT(adapter);
3468
3469         /*
3470          * Set up the timer service - driver is responsible for avoiding
3471          * concurrency, as the callout won't be using any locking inside.
3472          */
3473         callout_init(&adapter->timer_service, true);
3474         adapter->keep_alive_timeout = DEFAULT_KEEP_ALIVE_TO;
3475         adapter->missing_tx_timeout = DEFAULT_TX_CMP_TO;
3476         adapter->missing_tx_max_queues = DEFAULT_TX_MONITORED_QUEUES;
3477         adapter->missing_tx_threshold = DEFAULT_TX_CMP_THRESHOLD;
3478
3479         if (version_printed++ == 0)
3480                 device_printf(pdev, "%s\n", ena_version);
3481
3482         /* Allocate memory for ena_dev structure */
3483         ena_dev = malloc(sizeof(struct ena_com_dev), M_DEVBUF,
3484             M_WAITOK | M_ZERO);
3485
3486         adapter->ena_dev = ena_dev;
3487         ena_dev->dmadev = pdev;
3488
3489         rid = PCIR_BAR(ENA_REG_BAR);
3490         adapter->memory = NULL;
3491         adapter->registers = bus_alloc_resource_any(pdev, SYS_RES_MEMORY,
3492             &rid, RF_ACTIVE);
3493         if (unlikely(adapter->registers == NULL)) {
3494                 device_printf(pdev,
3495                     "unable to allocate bus resource: registers!\n");
3496                 rc = ENOMEM;
3497                 goto err_dev_free;
3498         }
3499
3500         ena_dev->bus = malloc(sizeof(struct ena_bus), M_DEVBUF,
3501             M_WAITOK | M_ZERO);
3502
3503         /* Store register resources */
3504         ((struct ena_bus*)(ena_dev->bus))->reg_bar_t =
3505             rman_get_bustag(adapter->registers);
3506         ((struct ena_bus*)(ena_dev->bus))->reg_bar_h =
3507             rman_get_bushandle(adapter->registers);
3508
3509         if (unlikely(((struct ena_bus*)(ena_dev->bus))->reg_bar_h == 0)) {
3510                 device_printf(pdev, "failed to pmap registers bar\n");
3511                 rc = ENXIO;
3512                 goto err_bus_free;
3513         }
3514
3515         ena_dev->tx_mem_queue_type = ENA_ADMIN_PLACEMENT_POLICY_HOST;
3516
3517         /* Initially clear all the flags */
3518         ENA_FLAG_ZERO(adapter);
3519
3520         /* Device initialization */
3521         rc = ena_device_init(adapter, pdev, &get_feat_ctx, &adapter->wd_active);
3522         if (unlikely(rc != 0)) {
3523                 device_printf(pdev, "ENA device init failed! (err: %d)\n", rc);
3524                 rc = ENXIO;
3525                 goto err_bus_free;
3526         }
3527
3528         set_default_llq_configurations(&llq_config);
3529
3530         rc = ena_set_queues_placement_policy(pdev, ena_dev, &get_feat_ctx.llq,
3531              &llq_config);
3532         if (unlikely(rc != 0)) {
3533                 device_printf(pdev, "failed to set placement policy\n");
3534                 goto err_com_free;
3535         }
3536
3537         if (ena_dev->tx_mem_queue_type == ENA_ADMIN_PLACEMENT_POLICY_DEV)
3538                 adapter->disable_meta_caching =
3539                     !!(get_feat_ctx.llq.accel_mode.u.get.supported_flags &
3540                     BIT(ENA_ADMIN_DISABLE_META_CACHING));
3541
3542         adapter->keep_alive_timestamp = getsbinuptime();
3543
3544         adapter->tx_offload_cap = get_feat_ctx.offload.tx;
3545
3546         memcpy(adapter->mac_addr, get_feat_ctx.dev_attr.mac_addr,
3547             ETHER_ADDR_LEN);
3548
3549         calc_queue_ctx.pdev = pdev;
3550         calc_queue_ctx.ena_dev = ena_dev;
3551         calc_queue_ctx.get_feat_ctx = &get_feat_ctx;
3552
3553         /* Calculate initial and maximum IO queue number and size */
3554         max_num_io_queues = ena_calc_max_io_queue_num(pdev, ena_dev,
3555             &get_feat_ctx);
3556         rc = ena_calc_io_queue_size(&calc_queue_ctx);
3557         if (unlikely((rc != 0) || (max_num_io_queues <= 0))) {
3558                 rc = EFAULT;
3559                 goto err_com_free;
3560         }
3561
3562         adapter->requested_tx_ring_size = calc_queue_ctx.tx_queue_size;
3563         adapter->requested_rx_ring_size = calc_queue_ctx.rx_queue_size;
3564         adapter->max_tx_ring_size = calc_queue_ctx.max_tx_queue_size;
3565         adapter->max_rx_ring_size = calc_queue_ctx.max_rx_queue_size;
3566         adapter->max_tx_sgl_size = calc_queue_ctx.max_tx_sgl_size;
3567         adapter->max_rx_sgl_size = calc_queue_ctx.max_rx_sgl_size;
3568
3569         adapter->max_num_io_queues = max_num_io_queues;
3570
3571         adapter->buf_ring_size = ENA_DEFAULT_BUF_RING_SIZE;
3572
3573         adapter->max_mtu = get_feat_ctx.dev_attr.max_mtu;
3574
3575         adapter->reset_reason = ENA_REGS_RESET_NORMAL;
3576
3577         /* set up dma tags for rx and tx buffers */
3578         rc = ena_setup_tx_dma_tag(adapter);
3579         if (unlikely(rc != 0)) {
3580                 device_printf(pdev, "Failed to create TX DMA tag\n");
3581                 goto err_com_free;
3582         }
3583
3584         rc = ena_setup_rx_dma_tag(adapter);
3585         if (unlikely(rc != 0)) {
3586                 device_printf(pdev, "Failed to create RX DMA tag\n");
3587                 goto err_tx_tag_free;
3588         }
3589
3590         /*
3591          * The amount of requested MSIX vectors is equal to
3592          * adapter::max_num_io_queues (see `ena_enable_msix()`), plus a constant
3593          * number of admin queue interrupts. The former is initially determined
3594          * by HW capabilities (see `ena_calc_max_io_queue_num())` but may not be
3595          * achieved if there are not enough system resources. By default, the
3596          * number of effectively used IO queues is the same but later on it can
3597          * be limited by the user using sysctl interface.
3598          */
3599         rc = ena_enable_msix_and_set_admin_interrupts(adapter);
3600         if (unlikely(rc != 0)) {
3601                 device_printf(pdev,
3602                     "Failed to enable and set the admin interrupts\n");
3603                 goto err_io_free;
3604         }
3605         /* By default all of allocated MSIX vectors are actively used */
3606         adapter->num_io_queues = adapter->msix_vecs - ENA_ADMIN_MSIX_VEC;
3607
3608         /* initialize rings basic information */
3609         ena_init_io_rings(adapter);
3610
3611         /* setup network interface */
3612         rc = ena_setup_ifnet(pdev, adapter, &get_feat_ctx);
3613         if (unlikely(rc != 0)) {
3614                 device_printf(pdev, "Error with network interface setup\n");
3615                 goto err_msix_free;
3616         }
3617
3618         /* Initialize reset task queue */
3619         TASK_INIT(&adapter->reset_task, 0, ena_reset_task, adapter);
3620         adapter->reset_tq = taskqueue_create("ena_reset_enqueue",
3621             M_WAITOK | M_ZERO, taskqueue_thread_enqueue, &adapter->reset_tq);
3622         taskqueue_start_threads(&adapter->reset_tq, 1, PI_NET,
3623             "%s rstq", device_get_nameunit(adapter->pdev));
3624
3625         /* Initialize statistics */
3626         ena_alloc_counters((counter_u64_t *)&adapter->dev_stats,
3627             sizeof(struct ena_stats_dev));
3628         ena_alloc_counters((counter_u64_t *)&adapter->hw_stats,
3629             sizeof(struct ena_hw_stats));
3630         ena_sysctl_add_nodes(adapter);
3631
3632 #ifdef DEV_NETMAP
3633         rc = ena_netmap_attach(adapter);
3634         if (rc != 0) {
3635                 device_printf(pdev, "netmap attach failed: %d\n", rc);
3636                 goto err_detach;
3637         }
3638 #endif /* DEV_NETMAP */
3639
3640         /* Tell the stack that the interface is not active */
3641         if_setdrvflagbits(adapter->ifp, IFF_DRV_OACTIVE, IFF_DRV_RUNNING);
3642         ENA_FLAG_SET_ATOMIC(ENA_FLAG_DEVICE_RUNNING, adapter);
3643
3644         return (0);
3645
3646 #ifdef DEV_NETMAP
3647 err_detach:
3648         ether_ifdetach(adapter->ifp);
3649 #endif /* DEV_NETMAP */
3650 err_msix_free:
3651         ena_com_dev_reset(adapter->ena_dev, ENA_REGS_RESET_INIT_ERR);
3652         ena_free_mgmnt_irq(adapter);
3653         ena_disable_msix(adapter);
3654 err_io_free:
3655         ena_free_all_io_rings_resources(adapter);
3656         ena_free_rx_dma_tag(adapter);
3657 err_tx_tag_free:
3658         ena_free_tx_dma_tag(adapter);
3659 err_com_free:
3660         ena_com_admin_destroy(ena_dev);
3661         ena_com_delete_host_info(ena_dev);
3662         ena_com_mmio_reg_read_request_destroy(ena_dev);
3663 err_bus_free:
3664         free(ena_dev->bus, M_DEVBUF);
3665         ena_free_pci_resources(adapter);
3666 err_dev_free:
3667         free(ena_dev, M_DEVBUF);
3668
3669         return (rc);
3670 }
3671
3672 /**
3673  * ena_detach - Device Removal Routine
3674  * @pdev: device information struct
3675  *
3676  * ena_detach is called by the device subsystem to alert the driver
3677  * that it should release a PCI device.
3678  **/
3679 static int
3680 ena_detach(device_t pdev)
3681 {
3682         struct ena_adapter *adapter = device_get_softc(pdev);
3683         struct ena_com_dev *ena_dev = adapter->ena_dev;
3684         int rc;
3685
3686         /* Make sure VLANS are not using driver */
3687         if (adapter->ifp->if_vlantrunk != NULL) {
3688                 device_printf(adapter->pdev ,"VLAN is in use, detach first\n");
3689                 return (EBUSY);
3690         }
3691
3692         ether_ifdetach(adapter->ifp);
3693
3694         /* Stop timer service */
3695         ENA_LOCK_LOCK(adapter);
3696         callout_drain(&adapter->timer_service);
3697         ENA_LOCK_UNLOCK(adapter);
3698
3699         /* Release reset task */
3700         while (taskqueue_cancel(adapter->reset_tq, &adapter->reset_task, NULL))
3701                 taskqueue_drain(adapter->reset_tq, &adapter->reset_task);
3702         taskqueue_free(adapter->reset_tq);
3703
3704         ENA_LOCK_LOCK(adapter);
3705         ena_down(adapter);
3706         ena_destroy_device(adapter, true);
3707         ENA_LOCK_UNLOCK(adapter);
3708
3709 #ifdef DEV_NETMAP
3710         netmap_detach(adapter->ifp);
3711 #endif /* DEV_NETMAP */
3712
3713         ena_free_counters((counter_u64_t *)&adapter->hw_stats,
3714             sizeof(struct ena_hw_stats));
3715         ena_free_counters((counter_u64_t *)&adapter->dev_stats,
3716             sizeof(struct ena_stats_dev));
3717
3718         rc = ena_free_rx_dma_tag(adapter);
3719         if (unlikely(rc != 0))
3720                 device_printf(adapter->pdev,
3721                     "Unmapped RX DMA tag associations\n");
3722
3723         rc = ena_free_tx_dma_tag(adapter);
3724         if (unlikely(rc != 0))
3725                 device_printf(adapter->pdev,
3726                     "Unmapped TX DMA tag associations\n");
3727
3728         ena_free_irqs(adapter);
3729
3730         ena_free_pci_resources(adapter);
3731
3732         if (likely(ENA_FLAG_ISSET(ENA_FLAG_RSS_ACTIVE, adapter)))
3733                 ena_com_rss_destroy(ena_dev);
3734
3735         ena_com_delete_host_info(ena_dev);
3736
3737         ENA_LOCK_DESTROY(adapter);
3738
3739         if_free(adapter->ifp);
3740
3741         if (ena_dev->bus != NULL)
3742                 free(ena_dev->bus, M_DEVBUF);
3743
3744         if (ena_dev != NULL)
3745                 free(ena_dev, M_DEVBUF);
3746
3747         return (bus_generic_detach(pdev));
3748 }
3749
3750 /******************************************************************************
3751  ******************************** AENQ Handlers *******************************
3752  *****************************************************************************/
3753 /**
3754  * ena_update_on_link_change:
3755  * Notify the network interface about the change in link status
3756  **/
3757 static void
3758 ena_update_on_link_change(void *adapter_data,
3759     struct ena_admin_aenq_entry *aenq_e)
3760 {
3761         struct ena_adapter *adapter = (struct ena_adapter *)adapter_data;
3762         struct ena_admin_aenq_link_change_desc *aenq_desc;
3763         int status;
3764         if_t ifp;
3765
3766         aenq_desc = (struct ena_admin_aenq_link_change_desc *)aenq_e;
3767         ifp = adapter->ifp;
3768         status = aenq_desc->flags &
3769             ENA_ADMIN_AENQ_LINK_CHANGE_DESC_LINK_STATUS_MASK;
3770
3771         if (status != 0) {
3772                 device_printf(adapter->pdev, "link is UP\n");
3773                 ENA_FLAG_SET_ATOMIC(ENA_FLAG_LINK_UP, adapter);
3774                 if (!ENA_FLAG_ISSET(ENA_FLAG_ONGOING_RESET, adapter))
3775                         if_link_state_change(ifp, LINK_STATE_UP);
3776         } else {
3777                 device_printf(adapter->pdev, "link is DOWN\n");
3778                 if_link_state_change(ifp, LINK_STATE_DOWN);
3779                 ENA_FLAG_CLEAR_ATOMIC(ENA_FLAG_LINK_UP, adapter);
3780         }
3781 }
3782
3783 static void ena_notification(void *adapter_data,
3784     struct ena_admin_aenq_entry *aenq_e)
3785 {
3786         struct ena_adapter *adapter = (struct ena_adapter *)adapter_data;
3787         struct ena_admin_ena_hw_hints *hints;
3788
3789         ENA_WARN(NULL, aenq_e->aenq_common_desc.group != ENA_ADMIN_NOTIFICATION,
3790             "Invalid group(%x) expected %x\n",  aenq_e->aenq_common_desc.group,
3791             ENA_ADMIN_NOTIFICATION);
3792
3793         switch (aenq_e->aenq_common_desc.syndrome) {
3794         case ENA_ADMIN_UPDATE_HINTS:
3795                 hints =
3796                     (struct ena_admin_ena_hw_hints *)(&aenq_e->inline_data_w4);
3797                 ena_update_hints(adapter, hints);
3798                 break;
3799         default:
3800                 device_printf(adapter->pdev,
3801                     "Invalid aenq notification link state %d\n",
3802                     aenq_e->aenq_common_desc.syndrome);
3803         }
3804 }
3805
3806 /**
3807  * This handler will called for unknown event group or unimplemented handlers
3808  **/
3809 static void
3810 unimplemented_aenq_handler(void *adapter_data,
3811     struct ena_admin_aenq_entry *aenq_e)
3812 {
3813         struct ena_adapter *adapter = (struct ena_adapter *)adapter_data;
3814
3815         device_printf(adapter->pdev,
3816             "Unknown event was received or event with unimplemented handler\n");
3817 }
3818
3819 static struct ena_aenq_handlers aenq_handlers = {
3820     .handlers = {
3821             [ENA_ADMIN_LINK_CHANGE] = ena_update_on_link_change,
3822             [ENA_ADMIN_NOTIFICATION] = ena_notification,
3823             [ENA_ADMIN_KEEP_ALIVE] = ena_keep_alive_wd,
3824     },
3825     .unimplemented_handler = unimplemented_aenq_handler
3826 };
3827
3828 /*********************************************************************
3829  *  FreeBSD Device Interface Entry Points
3830  *********************************************************************/
3831
3832 static device_method_t ena_methods[] = {
3833     /* Device interface */
3834     DEVMETHOD(device_probe, ena_probe),
3835     DEVMETHOD(device_attach, ena_attach),
3836     DEVMETHOD(device_detach, ena_detach),
3837     DEVMETHOD_END
3838 };
3839
3840 static driver_t ena_driver = {
3841     "ena", ena_methods, sizeof(struct ena_adapter),
3842 };
3843
3844 devclass_t ena_devclass;
3845 DRIVER_MODULE(ena, pci, ena_driver, ena_devclass, 0, 0);
3846 MODULE_PNP_INFO("U16:vendor;U16:device", pci, ena, ena_vendor_info_array,
3847     nitems(ena_vendor_info_array) - 1);
3848 MODULE_DEPEND(ena, pci, 1, 1, 1);
3849 MODULE_DEPEND(ena, ether, 1, 1, 1);
3850 #ifdef DEV_NETMAP
3851 MODULE_DEPEND(ena, netmap, 1, 1, 1);
3852 #endif /* DEV_NETMAP */
3853
3854 /*********************************************************************/