drivers/net/ethernet/microsoft/mana/mana_en.c

   1 // SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
   2 /* Copyright (c) 2021, Microsoft Corporation. */
   3
   4 #include <uapi/linux/bpf.h>
   5
   6 #include <linux/inetdevice.h>
   7 #include <linux/etherdevice.h>
   8 #include <linux/ethtool.h>
   9 #include <linux/filter.h>
  10 #include <linux/mm.h>
  11
  12 #include <net/checksum.h>
  13 #include <net/ip6_checksum.h>
  14 #include <net/xdp.h>
  15
  16 #include <net/mana/mana.h>
  17 #include <net/mana/mana_auxiliary.h>
  18
  19 static DEFINE_IDA(mana_adev_ida);
  20
  21 static int mana_adev_idx_alloc(void)
  22 {
  23         return ida_alloc(&mana_adev_ida, GFP_KERNEL);
  24 }
  25
  26 static void mana_adev_idx_free(int idx)
  27 {
  28         ida_free(&mana_adev_ida, idx);
  29 }
  30
  31 /* Microsoft Azure Network Adapter (MANA) functions */
  32
  33 static int mana_open(struct net_device *ndev)
  34 {
  35         struct mana_port_context *apc = netdev_priv(ndev);
  36         int err;
  37
  38         err = mana_alloc_queues(ndev);
  39         if (err)
  40                 return err;
  41
  42         apc->port_is_up = true;
  43
  44         /* Ensure port state updated before txq state */
  45         smp_wmb();
  46
  47         netif_carrier_on(ndev);
  48         netif_tx_wake_all_queues(ndev);
  49
  50         return 0;
  51 }
  52
  53 static int mana_close(struct net_device *ndev)
  54 {
  55         struct mana_port_context *apc = netdev_priv(ndev);
  56
  57         if (!apc->port_is_up)
  58                 return 0;
  59
  60         return mana_detach(ndev, true);
  61 }
  62
  63 static bool mana_can_tx(struct gdma_queue *wq)
  64 {
  65         return mana_gd_wq_avail_space(wq) >= MAX_TX_WQE_SIZE;
  66 }
  67
  68 static unsigned int mana_checksum_info(struct sk_buff *skb)
  69 {
  70         if (skb->protocol == htons(ETH_P_IP)) {
  71                 struct iphdr *ip = ip_hdr(skb);
  72
  73                 if (ip->protocol == IPPROTO_TCP)
  74                         return IPPROTO_TCP;
  75
  76                 if (ip->protocol == IPPROTO_UDP)
  77                         return IPPROTO_UDP;
  78         } else if (skb->protocol == htons(ETH_P_IPV6)) {
  79                 struct ipv6hdr *ip6 = ipv6_hdr(skb);
  80
  81                 if (ip6->nexthdr == IPPROTO_TCP)
  82                         return IPPROTO_TCP;
  83
  84                 if (ip6->nexthdr == IPPROTO_UDP)
  85                         return IPPROTO_UDP;
  86         }
  87
  88         /* No csum offloading */
  89         return 0;
  90 }
  91
  92 static int mana_map_skb(struct sk_buff *skb, struct mana_port_context *apc,
  93                         struct mana_tx_package *tp)
  94 {
  95         struct mana_skb_head *ash = (struct mana_skb_head *)skb->head;
  96         struct gdma_dev *gd = apc->ac->gdma_dev;
  97         struct gdma_context *gc;
  98         struct device *dev;
  99         skb_frag_t *frag;
 100         dma_addr_t da;
 101         int i;
 102
 103         gc = gd->gdma_context;
 104         dev = gc->dev;
 105         da = dma_map_single(dev, skb->data, skb_headlen(skb), DMA_TO_DEVICE);
 106
 107         if (dma_mapping_error(dev, da))
 108                 return -ENOMEM;
 109
 110         ash->dma_handle[0] = da;
 111         ash->size[0] = skb_headlen(skb);
 112
 113         tp->wqe_req.sgl[0].address = ash->dma_handle[0];
 114         tp->wqe_req.sgl[0].mem_key = gd->gpa_mkey;
 115         tp->wqe_req.sgl[0].size = ash->size[0];
 116
 117         for (i = 0; i < skb_shinfo(skb)->nr_frags; i++) {
 118                 frag = &skb_shinfo(skb)->frags[i];
 119                 da = skb_frag_dma_map(dev, frag, 0, skb_frag_size(frag),
 120                                       DMA_TO_DEVICE);
 121
 122                 if (dma_mapping_error(dev, da))
 123                         goto frag_err;
 124
 125                 ash->dma_handle[i + 1] = da;
 126                 ash->size[i + 1] = skb_frag_size(frag);
 127
 128                 tp->wqe_req.sgl[i + 1].address = ash->dma_handle[i + 1];
 129                 tp->wqe_req.sgl[i + 1].mem_key = gd->gpa_mkey;
 130                 tp->wqe_req.sgl[i + 1].size = ash->size[i + 1];
 131         }
 132
 133         return 0;
 134
 135 frag_err:
 136         for (i = i - 1; i >= 0; i--)
 137                 dma_unmap_page(dev, ash->dma_handle[i + 1], ash->size[i + 1],
 138                                DMA_TO_DEVICE);
 139
 140         dma_unmap_single(dev, ash->dma_handle[0], ash->size[0], DMA_TO_DEVICE);
 141
 142         return -ENOMEM;
 143 }
 144
 145 netdev_tx_t mana_start_xmit(struct sk_buff *skb, struct net_device *ndev)
 146 {
 147         enum mana_tx_pkt_format pkt_fmt = MANA_SHORT_PKT_FMT;
 148         struct mana_port_context *apc = netdev_priv(ndev);
 149         u16 txq_idx = skb_get_queue_mapping(skb);
 150         struct gdma_dev *gd = apc->ac->gdma_dev;
 151         bool ipv4 = false, ipv6 = false;
 152         struct mana_tx_package pkg = {};
 153         struct netdev_queue *net_txq;
 154         struct mana_stats_tx *tx_stats;
 155         struct gdma_queue *gdma_sq;
 156         unsigned int csum_type;
 157         struct mana_txq *txq;
 158         struct mana_cq *cq;
 159         int err, len;
 160         u16 ihs;
 161
 162         if (unlikely(!apc->port_is_up))
 163                 goto tx_drop;
 164
 165         if (skb_cow_head(skb, MANA_HEADROOM))
 166                 goto tx_drop_count;
 167
 168         txq = &apc->tx_qp[txq_idx].txq;
 169         gdma_sq = txq->gdma_sq;
 170         cq = &apc->tx_qp[txq_idx].tx_cq;
 171         tx_stats = &txq->stats;
 172
 173         pkg.tx_oob.s_oob.vcq_num = cq->gdma_id;
 174         pkg.tx_oob.s_oob.vsq_frame = txq->vsq_frame;
 175
 176         if (txq->vp_offset > MANA_SHORT_VPORT_OFFSET_MAX) {
 177                 pkg.tx_oob.l_oob.long_vp_offset = txq->vp_offset;
 178                 pkt_fmt = MANA_LONG_PKT_FMT;
 179         } else {
 180                 pkg.tx_oob.s_oob.short_vp_offset = txq->vp_offset;
 181         }
 182
 183         if (skb_vlan_tag_present(skb)) {
 184                 pkt_fmt = MANA_LONG_PKT_FMT;
 185                 pkg.tx_oob.l_oob.inject_vlan_pri_tag = 1;
 186                 pkg.tx_oob.l_oob.pcp = skb_vlan_tag_get_prio(skb);
 187                 pkg.tx_oob.l_oob.dei = skb_vlan_tag_get_cfi(skb);
 188                 pkg.tx_oob.l_oob.vlan_id = skb_vlan_tag_get_id(skb);
 189         }
 190
 191         pkg.tx_oob.s_oob.pkt_fmt = pkt_fmt;
 192
 193         if (pkt_fmt == MANA_SHORT_PKT_FMT) {
 194                 pkg.wqe_req.inline_oob_size = sizeof(struct mana_tx_short_oob);
 195                 u64_stats_update_begin(&tx_stats->syncp);
 196                 tx_stats->short_pkt_fmt++;
 197                 u64_stats_update_end(&tx_stats->syncp);
 198         } else {
 199                 pkg.wqe_req.inline_oob_size = sizeof(struct mana_tx_oob);
 200                 u64_stats_update_begin(&tx_stats->syncp);
 201                 tx_stats->long_pkt_fmt++;
 202                 u64_stats_update_end(&tx_stats->syncp);
 203         }
 204
 205         pkg.wqe_req.inline_oob_data = &pkg.tx_oob;
 206         pkg.wqe_req.flags = 0;
 207         pkg.wqe_req.client_data_unit = 0;
 208
 209         pkg.wqe_req.num_sge = 1 + skb_shinfo(skb)->nr_frags;
 210         WARN_ON_ONCE(pkg.wqe_req.num_sge > MAX_TX_WQE_SGL_ENTRIES);
 211
 212         if (pkg.wqe_req.num_sge <= ARRAY_SIZE(pkg.sgl_array)) {
 213                 pkg.wqe_req.sgl = pkg.sgl_array;
 214         } else {
 215                 pkg.sgl_ptr = kmalloc_array(pkg.wqe_req.num_sge,
 216                                             sizeof(struct gdma_sge),
 217                                             GFP_ATOMIC);
 218                 if (!pkg.sgl_ptr)
 219                         goto tx_drop_count;
 220
 221                 pkg.wqe_req.sgl = pkg.sgl_ptr;
 222         }
 223
 224         if (skb->protocol == htons(ETH_P_IP))
 225                 ipv4 = true;
 226         else if (skb->protocol == htons(ETH_P_IPV6))
 227                 ipv6 = true;
 228
 229         if (skb_is_gso(skb)) {
 230                 pkg.tx_oob.s_oob.is_outer_ipv4 = ipv4;
 231                 pkg.tx_oob.s_oob.is_outer_ipv6 = ipv6;
 232
 233                 pkg.tx_oob.s_oob.comp_iphdr_csum = 1;
 234                 pkg.tx_oob.s_oob.comp_tcp_csum = 1;
 235                 pkg.tx_oob.s_oob.trans_off = skb_transport_offset(skb);
 236
 237                 pkg.wqe_req.client_data_unit = skb_shinfo(skb)->gso_size;
 238                 pkg.wqe_req.flags = GDMA_WR_OOB_IN_SGL | GDMA_WR_PAD_BY_SGE0;
 239                 if (ipv4) {
 240                         ip_hdr(skb)->tot_len = 0;
 241                         ip_hdr(skb)->check = 0;
 242                         tcp_hdr(skb)->check =
 243                                 ~csum_tcpudp_magic(ip_hdr(skb)->saddr,
 244                                                    ip_hdr(skb)->daddr, 0,
 245                                                    IPPROTO_TCP, 0);
 246                 } else {
 247                         ipv6_hdr(skb)->payload_len = 0;
 248                         tcp_hdr(skb)->check =
 249                                 ~csum_ipv6_magic(&ipv6_hdr(skb)->saddr,
 250                                                  &ipv6_hdr(skb)->daddr, 0,
 251                                                  IPPROTO_TCP, 0);
 252                 }
 253
 254                 if (skb->encapsulation) {
 255                         ihs = skb_inner_tcp_all_headers(skb);
 256                         u64_stats_update_begin(&tx_stats->syncp);
 257                         tx_stats->tso_inner_packets++;
 258                         tx_stats->tso_inner_bytes += skb->len - ihs;
 259                         u64_stats_update_end(&tx_stats->syncp);
 260                 } else {
 261                         if (skb_shinfo(skb)->gso_type & SKB_GSO_UDP_L4) {
 262                                 ihs = skb_transport_offset(skb) + sizeof(struct udphdr);
 263                         } else {
 264                                 ihs = skb_tcp_all_headers(skb);
 265                                 if (ipv6_has_hopopt_jumbo(skb))
 266                                         ihs -= sizeof(struct hop_jumbo_hdr);
 267                         }
 268
 269                         u64_stats_update_begin(&tx_stats->syncp);
 270                         tx_stats->tso_packets++;
 271                         tx_stats->tso_bytes += skb->len - ihs;
 272                         u64_stats_update_end(&tx_stats->syncp);
 273                 }
 274
 275         } else if (skb->ip_summed == CHECKSUM_PARTIAL) {
 276                 csum_type = mana_checksum_info(skb);
 277
 278                 u64_stats_update_begin(&tx_stats->syncp);
 279                 tx_stats->csum_partial++;
 280                 u64_stats_update_end(&tx_stats->syncp);
 281
 282                 if (csum_type == IPPROTO_TCP) {
 283                         pkg.tx_oob.s_oob.is_outer_ipv4 = ipv4;
 284                         pkg.tx_oob.s_oob.is_outer_ipv6 = ipv6;
 285
 286                         pkg.tx_oob.s_oob.comp_tcp_csum = 1;
 287                         pkg.tx_oob.s_oob.trans_off = skb_transport_offset(skb);
 288
 289                 } else if (csum_type == IPPROTO_UDP) {
 290                         pkg.tx_oob.s_oob.is_outer_ipv4 = ipv4;
 291                         pkg.tx_oob.s_oob.is_outer_ipv6 = ipv6;
 292
 293                         pkg.tx_oob.s_oob.comp_udp_csum = 1;
 294                 } else {
 295                         /* Can't do offload of this type of checksum */
 296                         if (skb_checksum_help(skb))
 297                                 goto free_sgl_ptr;
 298                 }
 299         }
 300
 301         if (mana_map_skb(skb, apc, &pkg)) {
 302                 u64_stats_update_begin(&tx_stats->syncp);
 303                 tx_stats->mana_map_err++;
 304                 u64_stats_update_end(&tx_stats->syncp);
 305                 goto free_sgl_ptr;
 306         }
 307
 308         skb_queue_tail(&txq->pending_skbs, skb);
 309
 310         len = skb->len;
 311         net_txq = netdev_get_tx_queue(ndev, txq_idx);
 312
 313         err = mana_gd_post_work_request(gdma_sq, &pkg.wqe_req,
 314                                         (struct gdma_posted_wqe_info *)skb->cb);
 315         if (!mana_can_tx(gdma_sq)) {
 316                 netif_tx_stop_queue(net_txq);
 317                 apc->eth_stats.stop_queue++;
 318         }
 319
 320         if (err) {
 321                 (void)skb_dequeue_tail(&txq->pending_skbs);
 322                 netdev_warn(ndev, "Failed to post TX OOB: %d\n", err);
 323                 err = NETDEV_TX_BUSY;
 324                 goto tx_busy;
 325         }
 326
 327         err = NETDEV_TX_OK;
 328         atomic_inc(&txq->pending_sends);
 329
 330         mana_gd_wq_ring_doorbell(gd->gdma_context, gdma_sq);
 331
 332         /* skb may be freed after mana_gd_post_work_request. Do not use it. */
 333         skb = NULL;
 334
 335         tx_stats = &txq->stats;
 336         u64_stats_update_begin(&tx_stats->syncp);
 337         tx_stats->packets++;
 338         tx_stats->bytes += len;
 339         u64_stats_update_end(&tx_stats->syncp);
 340
 341 tx_busy:
 342         if (netif_tx_queue_stopped(net_txq) && mana_can_tx(gdma_sq)) {
 343                 netif_tx_wake_queue(net_txq);
 344                 apc->eth_stats.wake_queue++;
 345         }
 346
 347         kfree(pkg.sgl_ptr);
 348         return err;
 349
 350 free_sgl_ptr:
 351         kfree(pkg.sgl_ptr);
 352 tx_drop_count:
 353         ndev->stats.tx_dropped++;
 354 tx_drop:
 355         dev_kfree_skb_any(skb);
 356         return NETDEV_TX_OK;
 357 }
 358
 359 static void mana_get_stats64(struct net_device *ndev,
 360                              struct rtnl_link_stats64 *st)
 361 {
 362         struct mana_port_context *apc = netdev_priv(ndev);
 363         unsigned int num_queues = apc->num_queues;
 364         struct mana_stats_rx *rx_stats;
 365         struct mana_stats_tx *tx_stats;
 366         unsigned int start;
 367         u64 packets, bytes;
 368         int q;
 369
 370         if (!apc->port_is_up)
 371                 return;
 372
 373         netdev_stats_to_stats64(st, &ndev->stats);
 374
 375         for (q = 0; q < num_queues; q++) {
 376                 rx_stats = &apc->rxqs[q]->stats;
 377
 378                 do {
 379                         start = u64_stats_fetch_begin(&rx_stats->syncp);
 380                         packets = rx_stats->packets;
 381                         bytes = rx_stats->bytes;
 382                 } while (u64_stats_fetch_retry(&rx_stats->syncp, start));
 383
 384                 st->rx_packets += packets;
 385                 st->rx_bytes += bytes;
 386         }
 387
 388         for (q = 0; q < num_queues; q++) {
 389                 tx_stats = &apc->tx_qp[q].txq.stats;
 390
 391                 do {
 392                         start = u64_stats_fetch_begin(&tx_stats->syncp);
 393                         packets = tx_stats->packets;
 394                         bytes = tx_stats->bytes;
 395                 } while (u64_stats_fetch_retry(&tx_stats->syncp, start));
 396
 397                 st->tx_packets += packets;
 398                 st->tx_bytes += bytes;
 399         }
 400 }
 401
 402 static int mana_get_tx_queue(struct net_device *ndev, struct sk_buff *skb,
 403                              int old_q)
 404 {
 405         struct mana_port_context *apc = netdev_priv(ndev);
 406         u32 hash = skb_get_hash(skb);
 407         struct sock *sk = skb->sk;
 408         int txq;
 409
 410         txq = apc->indir_table[hash & MANA_INDIRECT_TABLE_MASK];
 411
 412         if (txq != old_q && sk && sk_fullsock(sk) &&
 413             rcu_access_pointer(sk->sk_dst_cache))
 414                 sk_tx_queue_set(sk, txq);
 415
 416         return txq;
 417 }
 418
 419 static u16 mana_select_queue(struct net_device *ndev, struct sk_buff *skb,
 420                              struct net_device *sb_dev)
 421 {
 422         int txq;
 423
 424         if (ndev->real_num_tx_queues == 1)
 425                 return 0;
 426
 427         txq = sk_tx_queue_get(skb->sk);
 428
 429         if (txq < 0 || skb->ooo_okay || txq >= ndev->real_num_tx_queues) {
 430                 if (skb_rx_queue_recorded(skb))
 431                         txq = skb_get_rx_queue(skb);
 432                 else
 433                         txq = mana_get_tx_queue(ndev, skb, txq);
 434         }
 435
 436         return txq;
 437 }
 438
 439 /* Release pre-allocated RX buffers */
 440 static void mana_pre_dealloc_rxbufs(struct mana_port_context *mpc)
 441 {
 442         struct device *dev;
 443         int i;
 444
 445         dev = mpc->ac->gdma_dev->gdma_context->dev;
 446
 447         if (!mpc->rxbufs_pre)
 448                 goto out1;
 449
 450         if (!mpc->das_pre)
 451                 goto out2;
 452
 453         while (mpc->rxbpre_total) {
 454                 i = --mpc->rxbpre_total;
 455                 dma_unmap_single(dev, mpc->das_pre[i], mpc->rxbpre_datasize,
 456                                  DMA_FROM_DEVICE);
 457                 put_page(virt_to_head_page(mpc->rxbufs_pre[i]));
 458         }
 459
 460         kfree(mpc->das_pre);
 461         mpc->das_pre = NULL;
 462
 463 out2:
 464         kfree(mpc->rxbufs_pre);
 465         mpc->rxbufs_pre = NULL;
 466
 467 out1:
 468         mpc->rxbpre_datasize = 0;
 469         mpc->rxbpre_alloc_size = 0;
 470         mpc->rxbpre_headroom = 0;
 471 }
 472
 473 /* Get a buffer from the pre-allocated RX buffers */
 474 static void *mana_get_rxbuf_pre(struct mana_rxq *rxq, dma_addr_t *da)
 475 {
 476         struct net_device *ndev = rxq->ndev;
 477         struct mana_port_context *mpc;
 478         void *va;
 479
 480         mpc = netdev_priv(ndev);
 481
 482         if (!mpc->rxbufs_pre || !mpc->das_pre || !mpc->rxbpre_total) {
 483                 netdev_err(ndev, "No RX pre-allocated bufs\n");
 484                 return NULL;
 485         }
 486
 487         /* Check sizes to catch unexpected coding error */
 488         if (mpc->rxbpre_datasize != rxq->datasize) {
 489                 netdev_err(ndev, "rxbpre_datasize mismatch: %u: %u\n",
 490                            mpc->rxbpre_datasize, rxq->datasize);
 491                 return NULL;
 492         }
 493
 494         if (mpc->rxbpre_alloc_size != rxq->alloc_size) {
 495                 netdev_err(ndev, "rxbpre_alloc_size mismatch: %u: %u\n",
 496                            mpc->rxbpre_alloc_size, rxq->alloc_size);
 497                 return NULL;
 498         }
 499
 500         if (mpc->rxbpre_headroom != rxq->headroom) {
 501                 netdev_err(ndev, "rxbpre_headroom mismatch: %u: %u\n",
 502                            mpc->rxbpre_headroom, rxq->headroom);
 503                 return NULL;
 504         }
 505
 506         mpc->rxbpre_total--;
 507
 508         *da = mpc->das_pre[mpc->rxbpre_total];
 509         va = mpc->rxbufs_pre[mpc->rxbpre_total];
 510         mpc->rxbufs_pre[mpc->rxbpre_total] = NULL;
 511
 512         /* Deallocate the array after all buffers are gone */
 513         if (!mpc->rxbpre_total)
 514                 mana_pre_dealloc_rxbufs(mpc);
 515
 516         return va;
 517 }
 518
 519 /* Get RX buffer's data size, alloc size, XDP headroom based on MTU */
 520 static void mana_get_rxbuf_cfg(int mtu, u32 *datasize, u32 *alloc_size,
 521                                u32 *headroom)
 522 {
 523         if (mtu > MANA_XDP_MTU_MAX)
 524                 *headroom = 0; /* no support for XDP */
 525         else
 526                 *headroom = XDP_PACKET_HEADROOM;
 527
 528         *alloc_size = mtu + MANA_RXBUF_PAD + *headroom;
 529
 530         *datasize = ALIGN(mtu + ETH_HLEN, MANA_RX_DATA_ALIGN);
 531 }
 532
 533 static int mana_pre_alloc_rxbufs(struct mana_port_context *mpc, int new_mtu)
 534 {
 535         struct device *dev;
 536         struct page *page;
 537         dma_addr_t da;
 538         int num_rxb;
 539         void *va;
 540         int i;
 541
 542         mana_get_rxbuf_cfg(new_mtu, &mpc->rxbpre_datasize,
 543                            &mpc->rxbpre_alloc_size, &mpc->rxbpre_headroom);
 544
 545         dev = mpc->ac->gdma_dev->gdma_context->dev;
 546
 547         num_rxb = mpc->num_queues * RX_BUFFERS_PER_QUEUE;
 548
 549         WARN(mpc->rxbufs_pre, "mana rxbufs_pre exists\n");
 550         mpc->rxbufs_pre = kmalloc_array(num_rxb, sizeof(void *), GFP_KERNEL);
 551         if (!mpc->rxbufs_pre)
 552                 goto error;
 553
 554         mpc->das_pre = kmalloc_array(num_rxb, sizeof(dma_addr_t), GFP_KERNEL);
 555         if (!mpc->das_pre)
 556                 goto error;
 557
 558         mpc->rxbpre_total = 0;
 559
 560         for (i = 0; i < num_rxb; i++) {
 561                 if (mpc->rxbpre_alloc_size > PAGE_SIZE) {
 562                         va = netdev_alloc_frag(mpc->rxbpre_alloc_size);
 563                         if (!va)
 564                                 goto error;
 565
 566                         page = virt_to_head_page(va);
 567                         /* Check if the frag falls back to single page */
 568                         if (compound_order(page) <
 569                             get_order(mpc->rxbpre_alloc_size)) {
 570                                 put_page(page);
 571                                 goto error;
 572                         }
 573                 } else {
 574                         page = dev_alloc_page();
 575                         if (!page)
 576                                 goto error;
 577
 578                         va = page_to_virt(page);
 579                 }
 580
 581                 da = dma_map_single(dev, va + mpc->rxbpre_headroom,
 582                                     mpc->rxbpre_datasize, DMA_FROM_DEVICE);
 583                 if (dma_mapping_error(dev, da)) {
 584                         put_page(virt_to_head_page(va));
 585                         goto error;
 586                 }
 587
 588                 mpc->rxbufs_pre[i] = va;
 589                 mpc->das_pre[i] = da;
 590                 mpc->rxbpre_total = i + 1;
 591         }
 592
 593         return 0;
 594
 595 error:
 596         mana_pre_dealloc_rxbufs(mpc);
 597         return -ENOMEM;
 598 }
 599
 600 static int mana_change_mtu(struct net_device *ndev, int new_mtu)
 601 {
 602         struct mana_port_context *mpc = netdev_priv(ndev);
 603         unsigned int old_mtu = ndev->mtu;
 604         int err;
 605
 606         /* Pre-allocate buffers to prevent failure in mana_attach later */
 607         err = mana_pre_alloc_rxbufs(mpc, new_mtu);
 608         if (err) {
 609                 netdev_err(ndev, "Insufficient memory for new MTU\n");
 610                 return err;
 611         }
 612
 613         err = mana_detach(ndev, false);
 614         if (err) {
 615                 netdev_err(ndev, "mana_detach failed: %d\n", err);
 616                 goto out;
 617         }
 618
 619         ndev->mtu = new_mtu;
 620
 621         err = mana_attach(ndev);
 622         if (err) {
 623                 netdev_err(ndev, "mana_attach failed: %d\n", err);
 624                 ndev->mtu = old_mtu;
 625         }
 626
 627 out:
 628         mana_pre_dealloc_rxbufs(mpc);
 629         return err;
 630 }
 631
 632 static const struct net_device_ops mana_devops = {
 633         .ndo_open               = mana_open,
 634         .ndo_stop               = mana_close,
 635         .ndo_select_queue       = mana_select_queue,
 636         .ndo_start_xmit         = mana_start_xmit,
 637         .ndo_validate_addr      = eth_validate_addr,
 638         .ndo_get_stats64        = mana_get_stats64,
 639         .ndo_bpf                = mana_bpf,
 640         .ndo_xdp_xmit           = mana_xdp_xmit,
 641         .ndo_change_mtu         = mana_change_mtu,
 642 };
 643
 644 static void mana_cleanup_port_context(struct mana_port_context *apc)
 645 {
 646         kfree(apc->rxqs);
 647         apc->rxqs = NULL;
 648 }
 649
 650 static int mana_init_port_context(struct mana_port_context *apc)
 651 {
 652         apc->rxqs = kcalloc(apc->num_queues, sizeof(struct mana_rxq *),
 653                             GFP_KERNEL);
 654
 655         return !apc->rxqs ? -ENOMEM : 0;
 656 }
 657
 658 static int mana_send_request(struct mana_context *ac, void *in_buf,
 659                              u32 in_len, void *out_buf, u32 out_len)
 660 {
 661         struct gdma_context *gc = ac->gdma_dev->gdma_context;
 662         struct gdma_resp_hdr *resp = out_buf;
 663         struct gdma_req_hdr *req = in_buf;
 664         struct device *dev = gc->dev;
 665         static atomic_t activity_id;
 666         int err;
 667
 668         req->dev_id = gc->mana.dev_id;
 669         req->activity_id = atomic_inc_return(&activity_id);
 670
 671         err = mana_gd_send_request(gc, in_len, in_buf, out_len,
 672                                    out_buf);
 673         if (err || resp->status) {
 674                 dev_err(dev, "Failed to send mana message: %d, 0x%x\n",
 675                         err, resp->status);
 676                 return err ? err : -EPROTO;
 677         }
 678
 679         if (req->dev_id.as_uint32 != resp->dev_id.as_uint32 ||
 680             req->activity_id != resp->activity_id) {
 681                 dev_err(dev, "Unexpected mana message response: %x,%x,%x,%x\n",
 682                         req->dev_id.as_uint32, resp->dev_id.as_uint32,
 683                         req->activity_id, resp->activity_id);
 684                 return -EPROTO;
 685         }
 686
 687         return 0;
 688 }
 689
 690 static int mana_verify_resp_hdr(const struct gdma_resp_hdr *resp_hdr,
 691                                 const enum mana_command_code expected_code,
 692                                 const u32 min_size)
 693 {
 694         if (resp_hdr->response.msg_type != expected_code)
 695                 return -EPROTO;
 696
 697         if (resp_hdr->response.msg_version < GDMA_MESSAGE_V1)
 698                 return -EPROTO;
 699
 700         if (resp_hdr->response.msg_size < min_size)
 701                 return -EPROTO;
 702
 703         return 0;
 704 }
 705
 706 static int mana_pf_register_hw_vport(struct mana_port_context *apc)
 707 {
 708         struct mana_register_hw_vport_resp resp = {};
 709         struct mana_register_hw_vport_req req = {};
 710         int err;
 711
 712         mana_gd_init_req_hdr(&req.hdr, MANA_REGISTER_HW_PORT,
 713                              sizeof(req), sizeof(resp));
 714         req.attached_gfid = 1;
 715         req.is_pf_default_vport = 1;
 716         req.allow_all_ether_types = 1;
 717
 718         err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
 719                                 sizeof(resp));
 720         if (err) {
 721                 netdev_err(apc->ndev, "Failed to register hw vPort: %d\n", err);
 722                 return err;
 723         }
 724
 725         err = mana_verify_resp_hdr(&resp.hdr, MANA_REGISTER_HW_PORT,
 726                                    sizeof(resp));
 727         if (err || resp.hdr.status) {
 728                 netdev_err(apc->ndev, "Failed to register hw vPort: %d, 0x%x\n",
 729                            err, resp.hdr.status);
 730                 return err ? err : -EPROTO;
 731         }
 732
 733         apc->port_handle = resp.hw_vport_handle;
 734         return 0;
 735 }
 736
 737 static void mana_pf_deregister_hw_vport(struct mana_port_context *apc)
 738 {
 739         struct mana_deregister_hw_vport_resp resp = {};
 740         struct mana_deregister_hw_vport_req req = {};
 741         int err;
 742
 743         mana_gd_init_req_hdr(&req.hdr, MANA_DEREGISTER_HW_PORT,
 744                              sizeof(req), sizeof(resp));
 745         req.hw_vport_handle = apc->port_handle;
 746
 747         err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
 748                                 sizeof(resp));
 749         if (err) {
 750                 netdev_err(apc->ndev, "Failed to unregister hw vPort: %d\n",
 751                            err);
 752                 return;
 753         }
 754
 755         err = mana_verify_resp_hdr(&resp.hdr, MANA_DEREGISTER_HW_PORT,
 756                                    sizeof(resp));
 757         if (err || resp.hdr.status)
 758                 netdev_err(apc->ndev,
 759                            "Failed to deregister hw vPort: %d, 0x%x\n",
 760                            err, resp.hdr.status);
 761 }
 762
 763 static int mana_pf_register_filter(struct mana_port_context *apc)
 764 {
 765         struct mana_register_filter_resp resp = {};
 766         struct mana_register_filter_req req = {};
 767         int err;
 768
 769         mana_gd_init_req_hdr(&req.hdr, MANA_REGISTER_FILTER,
 770                              sizeof(req), sizeof(resp));
 771         req.vport = apc->port_handle;
 772         memcpy(req.mac_addr, apc->mac_addr, ETH_ALEN);
 773
 774         err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
 775                                 sizeof(resp));
 776         if (err) {
 777                 netdev_err(apc->ndev, "Failed to register filter: %d\n", err);
 778                 return err;
 779         }
 780
 781         err = mana_verify_resp_hdr(&resp.hdr, MANA_REGISTER_FILTER,
 782                                    sizeof(resp));
 783         if (err || resp.hdr.status) {
 784                 netdev_err(apc->ndev, "Failed to register filter: %d, 0x%x\n",
 785                            err, resp.hdr.status);
 786                 return err ? err : -EPROTO;
 787         }
 788
 789         apc->pf_filter_handle = resp.filter_handle;
 790         return 0;
 791 }
 792
 793 static void mana_pf_deregister_filter(struct mana_port_context *apc)
 794 {
 795         struct mana_deregister_filter_resp resp = {};
 796         struct mana_deregister_filter_req req = {};
 797         int err;
 798
 799         mana_gd_init_req_hdr(&req.hdr, MANA_DEREGISTER_FILTER,
 800                              sizeof(req), sizeof(resp));
 801         req.filter_handle = apc->pf_filter_handle;
 802
 803         err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
 804                                 sizeof(resp));
 805         if (err) {
 806                 netdev_err(apc->ndev, "Failed to unregister filter: %d\n",
 807                            err);
 808                 return;
 809         }
 810
 811         err = mana_verify_resp_hdr(&resp.hdr, MANA_DEREGISTER_FILTER,
 812                                    sizeof(resp));
 813         if (err || resp.hdr.status)
 814                 netdev_err(apc->ndev,
 815                            "Failed to deregister filter: %d, 0x%x\n",
 816                            err, resp.hdr.status);
 817 }
 818
 819 static int mana_query_device_cfg(struct mana_context *ac, u32 proto_major_ver,
 820                                  u32 proto_minor_ver, u32 proto_micro_ver,
 821                                  u16 *max_num_vports)
 822 {
 823         struct gdma_context *gc = ac->gdma_dev->gdma_context;
 824         struct mana_query_device_cfg_resp resp = {};
 825         struct mana_query_device_cfg_req req = {};
 826         struct device *dev = gc->dev;
 827         int err = 0;
 828
 829         mana_gd_init_req_hdr(&req.hdr, MANA_QUERY_DEV_CONFIG,
 830                              sizeof(req), sizeof(resp));
 831
 832         req.hdr.resp.msg_version = GDMA_MESSAGE_V2;
 833
 834         req.proto_major_ver = proto_major_ver;
 835         req.proto_minor_ver = proto_minor_ver;
 836         req.proto_micro_ver = proto_micro_ver;
 837
 838         err = mana_send_request(ac, &req, sizeof(req), &resp, sizeof(resp));
 839         if (err) {
 840                 dev_err(dev, "Failed to query config: %d", err);
 841                 return err;
 842         }
 843
 844         err = mana_verify_resp_hdr(&resp.hdr, MANA_QUERY_DEV_CONFIG,
 845                                    sizeof(resp));
 846         if (err || resp.hdr.status) {
 847                 dev_err(dev, "Invalid query result: %d, 0x%x\n", err,
 848                         resp.hdr.status);
 849                 if (!err)
 850                         err = -EPROTO;
 851                 return err;
 852         }
 853
 854         *max_num_vports = resp.max_num_vports;
 855
 856         if (resp.hdr.response.msg_version == GDMA_MESSAGE_V2)
 857                 gc->adapter_mtu = resp.adapter_mtu;
 858         else
 859                 gc->adapter_mtu = ETH_FRAME_LEN;
 860
 861         return 0;
 862 }
 863
 864 static int mana_query_vport_cfg(struct mana_port_context *apc, u32 vport_index,
 865                                 u32 *max_sq, u32 *max_rq, u32 *num_indir_entry)
 866 {
 867         struct mana_query_vport_cfg_resp resp = {};
 868         struct mana_query_vport_cfg_req req = {};
 869         int err;
 870
 871         mana_gd_init_req_hdr(&req.hdr, MANA_QUERY_VPORT_CONFIG,
 872                              sizeof(req), sizeof(resp));
 873
 874         req.vport_index = vport_index;
 875
 876         err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
 877                                 sizeof(resp));
 878         if (err)
 879                 return err;
 880
 881         err = mana_verify_resp_hdr(&resp.hdr, MANA_QUERY_VPORT_CONFIG,
 882                                    sizeof(resp));
 883         if (err)
 884                 return err;
 885
 886         if (resp.hdr.status)
 887                 return -EPROTO;
 888
 889         *max_sq = resp.max_num_sq;
 890         *max_rq = resp.max_num_rq;
 891         *num_indir_entry = resp.num_indirection_ent;
 892
 893         apc->port_handle = resp.vport;
 894         ether_addr_copy(apc->mac_addr, resp.mac_addr);
 895
 896         return 0;
 897 }
 898
 899 void mana_uncfg_vport(struct mana_port_context *apc)
 900 {
 901         mutex_lock(&apc->vport_mutex);
 902         apc->vport_use_count--;
 903         WARN_ON(apc->vport_use_count < 0);
 904         mutex_unlock(&apc->vport_mutex);
 905 }
 906 EXPORT_SYMBOL_NS(mana_uncfg_vport, NET_MANA);
 907
 908 int mana_cfg_vport(struct mana_port_context *apc, u32 protection_dom_id,
 909                    u32 doorbell_pg_id)
 910 {
 911         struct mana_config_vport_resp resp = {};
 912         struct mana_config_vport_req req = {};
 913         int err;
 914
 915         /* This function is used to program the Ethernet port in the hardware
 916          * table. It can be called from the Ethernet driver or the RDMA driver.
 917          *
 918          * For Ethernet usage, the hardware supports only one active user on a
 919          * physical port. The driver checks on the port usage before programming
 920          * the hardware when creating the RAW QP (RDMA driver) or exposing the
 921          * device to kernel NET layer (Ethernet driver).
 922          *
 923          * Because the RDMA driver doesn't know in advance which QP type the
 924          * user will create, it exposes the device with all its ports. The user
 925          * may not be able to create RAW QP on a port if this port is already
 926          * in used by the Ethernet driver from the kernel.
 927          *
 928          * This physical port limitation only applies to the RAW QP. For RC QP,
 929          * the hardware doesn't have this limitation. The user can create RC
 930          * QPs on a physical port up to the hardware limits independent of the
 931          * Ethernet usage on the same port.
 932          */
 933         mutex_lock(&apc->vport_mutex);
 934         if (apc->vport_use_count > 0) {
 935                 mutex_unlock(&apc->vport_mutex);
 936                 return -EBUSY;
 937         }
 938         apc->vport_use_count++;
 939         mutex_unlock(&apc->vport_mutex);
 940
 941         mana_gd_init_req_hdr(&req.hdr, MANA_CONFIG_VPORT_TX,
 942                              sizeof(req), sizeof(resp));
 943         req.vport = apc->port_handle;
 944         req.pdid = protection_dom_id;
 945         req.doorbell_pageid = doorbell_pg_id;
 946
 947         err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
 948                                 sizeof(resp));
 949         if (err) {
 950                 netdev_err(apc->ndev, "Failed to configure vPort: %d\n", err);
 951                 goto out;
 952         }
 953
 954         err = mana_verify_resp_hdr(&resp.hdr, MANA_CONFIG_VPORT_TX,
 955                                    sizeof(resp));
 956         if (err || resp.hdr.status) {
 957                 netdev_err(apc->ndev, "Failed to configure vPort: %d, 0x%x\n",
 958                            err, resp.hdr.status);
 959                 if (!err)
 960                         err = -EPROTO;
 961
 962                 goto out;
 963         }
 964
 965         apc->tx_shortform_allowed = resp.short_form_allowed;
 966         apc->tx_vp_offset = resp.tx_vport_offset;
 967
 968         netdev_info(apc->ndev, "Configured vPort %llu PD %u DB %u\n",
 969                     apc->port_handle, protection_dom_id, doorbell_pg_id);
 970 out:
 971         if (err)
 972                 mana_uncfg_vport(apc);
 973
 974         return err;
 975 }
 976 EXPORT_SYMBOL_NS(mana_cfg_vport, NET_MANA);
 977
 978 static int mana_cfg_vport_steering(struct mana_port_context *apc,
 979                                    enum TRI_STATE rx,
 980                                    bool update_default_rxobj, bool update_key,
 981                                    bool update_tab)
 982 {
 983         u16 num_entries = MANA_INDIRECT_TABLE_SIZE;
 984         struct mana_cfg_rx_steer_req_v2 *req;
 985         struct mana_cfg_rx_steer_resp resp = {};
 986         struct net_device *ndev = apc->ndev;
 987         mana_handle_t *req_indir_tab;
 988         u32 req_buf_size;
 989         int err;
 990
 991         req_buf_size = sizeof(*req) + sizeof(mana_handle_t) * num_entries;
 992         req = kzalloc(req_buf_size, GFP_KERNEL);
 993         if (!req)
 994                 return -ENOMEM;
 995
 996         mana_gd_init_req_hdr(&req->hdr, MANA_CONFIG_VPORT_RX, req_buf_size,
 997                              sizeof(resp));
 998
 999         req->hdr.req.msg_version = GDMA_MESSAGE_V2;
1000
1001         req->vport = apc->port_handle;
1002         req->num_indir_entries = num_entries;
1003         req->indir_tab_offset = sizeof(*req);
1004         req->rx_enable = rx;
1005         req->rss_enable = apc->rss_state;
1006         req->update_default_rxobj = update_default_rxobj;
1007         req->update_hashkey = update_key;
1008         req->update_indir_tab = update_tab;
1009         req->default_rxobj = apc->default_rxobj;
1010         req->cqe_coalescing_enable = 0;
1011
1012         if (update_key)
1013                 memcpy(&req->hashkey, apc->hashkey, MANA_HASH_KEY_SIZE);
1014
1015         if (update_tab) {
1016                 req_indir_tab = (mana_handle_t *)(req + 1);
1017                 memcpy(req_indir_tab, apc->rxobj_table,
1018                        req->num_indir_entries * sizeof(mana_handle_t));
1019         }
1020
1021         err = mana_send_request(apc->ac, req, req_buf_size, &resp,
1022                                 sizeof(resp));
1023         if (err) {
1024                 netdev_err(ndev, "Failed to configure vPort RX: %d\n", err);
1025                 goto out;
1026         }
1027
1028         err = mana_verify_resp_hdr(&resp.hdr, MANA_CONFIG_VPORT_RX,
1029                                    sizeof(resp));
1030         if (err) {
1031                 netdev_err(ndev, "vPort RX configuration failed: %d\n", err);
1032                 goto out;
1033         }
1034
1035         if (resp.hdr.status) {
1036                 netdev_err(ndev, "vPort RX configuration failed: 0x%x\n",
1037                            resp.hdr.status);
1038                 err = -EPROTO;
1039         }
1040
1041         netdev_info(ndev, "Configured steering vPort %llu entries %u\n",
1042                     apc->port_handle, num_entries);
1043 out:
1044         kfree(req);
1045         return err;
1046 }
1047
1048 int mana_create_wq_obj(struct mana_port_context *apc,
1049                        mana_handle_t vport,
1050                        u32 wq_type, struct mana_obj_spec *wq_spec,
1051                        struct mana_obj_spec *cq_spec,
1052                        mana_handle_t *wq_obj)
1053 {
1054         struct mana_create_wqobj_resp resp = {};
1055         struct mana_create_wqobj_req req = {};
1056         struct net_device *ndev = apc->ndev;
1057         int err;
1058
1059         mana_gd_init_req_hdr(&req.hdr, MANA_CREATE_WQ_OBJ,
1060                              sizeof(req), sizeof(resp));
1061         req.vport = vport;
1062         req.wq_type = wq_type;
1063         req.wq_gdma_region = wq_spec->gdma_region;
1064         req.cq_gdma_region = cq_spec->gdma_region;
1065         req.wq_size = wq_spec->queue_size;
1066         req.cq_size = cq_spec->queue_size;
1067         req.cq_moderation_ctx_id = cq_spec->modr_ctx_id;
1068         req.cq_parent_qid = cq_spec->attached_eq;
1069
1070         err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
1071                                 sizeof(resp));
1072         if (err) {
1073                 netdev_err(ndev, "Failed to create WQ object: %d\n", err);
1074                 goto out;
1075         }
1076
1077         err = mana_verify_resp_hdr(&resp.hdr, MANA_CREATE_WQ_OBJ,
1078                                    sizeof(resp));
1079         if (err || resp.hdr.status) {
1080                 netdev_err(ndev, "Failed to create WQ object: %d, 0x%x\n", err,
1081                            resp.hdr.status);
1082                 if (!err)
1083                         err = -EPROTO;
1084                 goto out;
1085         }
1086
1087         if (resp.wq_obj == INVALID_MANA_HANDLE) {
1088                 netdev_err(ndev, "Got an invalid WQ object handle\n");
1089                 err = -EPROTO;
1090                 goto out;
1091         }
1092
1093         *wq_obj = resp.wq_obj;
1094         wq_spec->queue_index = resp.wq_id;
1095         cq_spec->queue_index = resp.cq_id;
1096
1097         return 0;
1098 out:
1099         return err;
1100 }
1101 EXPORT_SYMBOL_NS(mana_create_wq_obj, NET_MANA);
1102
1103 void mana_destroy_wq_obj(struct mana_port_context *apc, u32 wq_type,
1104                          mana_handle_t wq_obj)
1105 {
1106         struct mana_destroy_wqobj_resp resp = {};
1107         struct mana_destroy_wqobj_req req = {};
1108         struct net_device *ndev = apc->ndev;
1109         int err;
1110
1111         mana_gd_init_req_hdr(&req.hdr, MANA_DESTROY_WQ_OBJ,
1112                              sizeof(req), sizeof(resp));
1113         req.wq_type = wq_type;
1114         req.wq_obj_handle = wq_obj;
1115
1116         err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
1117                                 sizeof(resp));
1118         if (err) {
1119                 netdev_err(ndev, "Failed to destroy WQ object: %d\n", err);
1120                 return;
1121         }
1122
1123         err = mana_verify_resp_hdr(&resp.hdr, MANA_DESTROY_WQ_OBJ,
1124                                    sizeof(resp));
1125         if (err || resp.hdr.status)
1126                 netdev_err(ndev, "Failed to destroy WQ object: %d, 0x%x\n", err,
1127                            resp.hdr.status);
1128 }
1129 EXPORT_SYMBOL_NS(mana_destroy_wq_obj, NET_MANA);
1130
1131 static void mana_destroy_eq(struct mana_context *ac)
1132 {
1133         struct gdma_context *gc = ac->gdma_dev->gdma_context;
1134         struct gdma_queue *eq;
1135         int i;
1136
1137         if (!ac->eqs)
1138                 return;
1139
1140         for (i = 0; i < gc->max_num_queues; i++) {
1141                 eq = ac->eqs[i].eq;
1142                 if (!eq)
1143                         continue;
1144
1145                 mana_gd_destroy_queue(gc, eq);
1146         }
1147
1148         kfree(ac->eqs);
1149         ac->eqs = NULL;
1150 }
1151
1152 static int mana_create_eq(struct mana_context *ac)
1153 {
1154         struct gdma_dev *gd = ac->gdma_dev;
1155         struct gdma_context *gc = gd->gdma_context;
1156         struct gdma_queue_spec spec = {};
1157         int err;
1158         int i;
1159
1160         ac->eqs = kcalloc(gc->max_num_queues, sizeof(struct mana_eq),
1161                           GFP_KERNEL);
1162         if (!ac->eqs)
1163                 return -ENOMEM;
1164
1165         spec.type = GDMA_EQ;
1166         spec.monitor_avl_buf = false;
1167         spec.queue_size = EQ_SIZE;
1168         spec.eq.callback = NULL;
1169         spec.eq.context = ac->eqs;
1170         spec.eq.log2_throttle_limit = LOG2_EQ_THROTTLE;
1171
1172         for (i = 0; i < gc->max_num_queues; i++) {
1173                 err = mana_gd_create_mana_eq(gd, &spec, &ac->eqs[i].eq);
1174                 if (err)
1175                         goto out;
1176         }
1177
1178         return 0;
1179 out:
1180         mana_destroy_eq(ac);
1181         return err;
1182 }
1183
1184 static int mana_fence_rq(struct mana_port_context *apc, struct mana_rxq *rxq)
1185 {
1186         struct mana_fence_rq_resp resp = {};
1187         struct mana_fence_rq_req req = {};
1188         int err;
1189
1190         init_completion(&rxq->fence_event);
1191
1192         mana_gd_init_req_hdr(&req.hdr, MANA_FENCE_RQ,
1193                              sizeof(req), sizeof(resp));
1194         req.wq_obj_handle =  rxq->rxobj;
1195
1196         err = mana_send_request(apc->ac, &req, sizeof(req), &resp,
1197                                 sizeof(resp));
1198         if (err) {
1199                 netdev_err(apc->ndev, "Failed to fence RQ %u: %d\n",
1200                            rxq->rxq_idx, err);
1201                 return err;
1202         }
1203
1204         err = mana_verify_resp_hdr(&resp.hdr, MANA_FENCE_RQ, sizeof(resp));
1205         if (err || resp.hdr.status) {
1206                 netdev_err(apc->ndev, "Failed to fence RQ %u: %d, 0x%x\n",
1207                            rxq->rxq_idx, err, resp.hdr.status);
1208                 if (!err)
1209                         err = -EPROTO;
1210
1211                 return err;
1212         }
1213
1214         if (wait_for_completion_timeout(&rxq->fence_event, 10 * HZ) == 0) {
1215                 netdev_err(apc->ndev, "Failed to fence RQ %u: timed out\n",
1216                            rxq->rxq_idx);
1217                 return -ETIMEDOUT;
1218         }
1219
1220         return 0;
1221 }
1222
1223 static void mana_fence_rqs(struct mana_port_context *apc)
1224 {
1225         unsigned int rxq_idx;
1226         struct mana_rxq *rxq;
1227         int err;
1228
1229         for (rxq_idx = 0; rxq_idx < apc->num_queues; rxq_idx++) {
1230                 rxq = apc->rxqs[rxq_idx];
1231                 err = mana_fence_rq(apc, rxq);
1232
1233                 /* In case of any error, use sleep instead. */
1234                 if (err)
1235                         msleep(100);
1236         }
1237 }
1238
1239 static int mana_move_wq_tail(struct gdma_queue *wq, u32 num_units)
1240 {
1241         u32 used_space_old;
1242         u32 used_space_new;
1243
1244         used_space_old = wq->head - wq->tail;
1245         used_space_new = wq->head - (wq->tail + num_units);
1246
1247         if (WARN_ON_ONCE(used_space_new > used_space_old))
1248                 return -ERANGE;
1249
1250         wq->tail += num_units;
1251         return 0;
1252 }
1253
1254 static void mana_unmap_skb(struct sk_buff *skb, struct mana_port_context *apc)
1255 {
1256         struct mana_skb_head *ash = (struct mana_skb_head *)skb->head;
1257         struct gdma_context *gc = apc->ac->gdma_dev->gdma_context;
1258         struct device *dev = gc->dev;
1259         int i;
1260
1261         dma_unmap_single(dev, ash->dma_handle[0], ash->size[0], DMA_TO_DEVICE);
1262
1263         for (i = 1; i < skb_shinfo(skb)->nr_frags + 1; i++)
1264                 dma_unmap_page(dev, ash->dma_handle[i], ash->size[i],
1265                                DMA_TO_DEVICE);
1266 }
1267
1268 static void mana_poll_tx_cq(struct mana_cq *cq)
1269 {
1270         struct gdma_comp *completions = cq->gdma_comp_buf;
1271         struct gdma_posted_wqe_info *wqe_info;
1272         unsigned int pkt_transmitted = 0;
1273         unsigned int wqe_unit_cnt = 0;
1274         struct mana_txq *txq = cq->txq;
1275         struct mana_port_context *apc;
1276         struct netdev_queue *net_txq;
1277         struct gdma_queue *gdma_wq;
1278         unsigned int avail_space;
1279         struct net_device *ndev;
1280         struct sk_buff *skb;
1281         bool txq_stopped;
1282         int comp_read;
1283         int i;
1284
1285         ndev = txq->ndev;
1286         apc = netdev_priv(ndev);
1287
1288         comp_read = mana_gd_poll_cq(cq->gdma_cq, completions,
1289                                     CQE_POLLING_BUFFER);
1290
1291         if (comp_read < 1)
1292                 return;
1293
1294         for (i = 0; i < comp_read; i++) {
1295                 struct mana_tx_comp_oob *cqe_oob;
1296
1297                 if (WARN_ON_ONCE(!completions[i].is_sq))
1298                         return;
1299
1300                 cqe_oob = (struct mana_tx_comp_oob *)completions[i].cqe_data;
1301                 if (WARN_ON_ONCE(cqe_oob->cqe_hdr.client_type !=
1302                                  MANA_CQE_COMPLETION))
1303                         return;
1304
1305                 switch (cqe_oob->cqe_hdr.cqe_type) {
1306                 case CQE_TX_OKAY:
1307                         break;
1308
1309                 case CQE_TX_SA_DROP:
1310                 case CQE_TX_MTU_DROP:
1311                 case CQE_TX_INVALID_OOB:
1312                 case CQE_TX_INVALID_ETH_TYPE:
1313                 case CQE_TX_HDR_PROCESSING_ERROR:
1314                 case CQE_TX_VF_DISABLED:
1315                 case CQE_TX_VPORT_IDX_OUT_OF_RANGE:
1316                 case CQE_TX_VPORT_DISABLED:
1317                 case CQE_TX_VLAN_TAGGING_VIOLATION:
1318                         WARN_ONCE(1, "TX: CQE error %d: ignored.\n",
1319                                   cqe_oob->cqe_hdr.cqe_type);
1320                         apc->eth_stats.tx_cqe_err++;
1321                         break;
1322
1323                 default:
1324                         /* If the CQE type is unexpected, log an error, assert,
1325                          * and go through the error path.
1326                          */
1327                         WARN_ONCE(1, "TX: Unexpected CQE type %d: HW BUG?\n",
1328                                   cqe_oob->cqe_hdr.cqe_type);
1329                         apc->eth_stats.tx_cqe_unknown_type++;
1330                         return;
1331                 }
1332
1333                 if (WARN_ON_ONCE(txq->gdma_txq_id != completions[i].wq_num))
1334                         return;
1335
1336                 skb = skb_dequeue(&txq->pending_skbs);
1337                 if (WARN_ON_ONCE(!skb))
1338                         return;
1339
1340                 wqe_info = (struct gdma_posted_wqe_info *)skb->cb;
1341                 wqe_unit_cnt += wqe_info->wqe_size_in_bu;
1342
1343                 mana_unmap_skb(skb, apc);
1344
1345                 napi_consume_skb(skb, cq->budget);
1346
1347                 pkt_transmitted++;
1348         }
1349
1350         if (WARN_ON_ONCE(wqe_unit_cnt == 0))
1351                 return;
1352
1353         mana_move_wq_tail(txq->gdma_sq, wqe_unit_cnt);
1354
1355         gdma_wq = txq->gdma_sq;
1356         avail_space = mana_gd_wq_avail_space(gdma_wq);
1357
1358         /* Ensure tail updated before checking q stop */
1359         smp_mb();
1360
1361         net_txq = txq->net_txq;
1362         txq_stopped = netif_tx_queue_stopped(net_txq);
1363
1364         /* Ensure checking txq_stopped before apc->port_is_up. */
1365         smp_rmb();
1366
1367         if (txq_stopped && apc->port_is_up && avail_space >= MAX_TX_WQE_SIZE) {
1368                 netif_tx_wake_queue(net_txq);
1369                 apc->eth_stats.wake_queue++;
1370         }
1371
1372         if (atomic_sub_return(pkt_transmitted, &txq->pending_sends) < 0)
1373                 WARN_ON_ONCE(1);
1374
1375         cq->work_done = pkt_transmitted;
1376 }
1377
1378 static void mana_post_pkt_rxq(struct mana_rxq *rxq)
1379 {
1380         struct mana_recv_buf_oob *recv_buf_oob;
1381         u32 curr_index;
1382         int err;
1383
1384         curr_index = rxq->buf_index++;
1385         if (rxq->buf_index == rxq->num_rx_buf)
1386                 rxq->buf_index = 0;
1387
1388         recv_buf_oob = &rxq->rx_oobs[curr_index];
1389
1390         err = mana_gd_post_work_request(rxq->gdma_rq, &recv_buf_oob->wqe_req,
1391                                         &recv_buf_oob->wqe_inf);
1392         if (WARN_ON_ONCE(err))
1393                 return;
1394
1395         WARN_ON_ONCE(recv_buf_oob->wqe_inf.wqe_size_in_bu != 1);
1396 }
1397
1398 static struct sk_buff *mana_build_skb(struct mana_rxq *rxq, void *buf_va,
1399                                       uint pkt_len, struct xdp_buff *xdp)
1400 {
1401         struct sk_buff *skb = napi_build_skb(buf_va, rxq->alloc_size);
1402
1403         if (!skb)
1404                 return NULL;
1405
1406         if (xdp->data_hard_start) {
1407                 skb_reserve(skb, xdp->data - xdp->data_hard_start);
1408                 skb_put(skb, xdp->data_end - xdp->data);
1409                 return skb;
1410         }
1411
1412         skb_reserve(skb, rxq->headroom);
1413         skb_put(skb, pkt_len);
1414
1415         return skb;
1416 }
1417
1418 static void mana_rx_skb(void *buf_va, bool from_pool,
1419                         struct mana_rxcomp_oob *cqe, struct mana_rxq *rxq)
1420 {
1421         struct mana_stats_rx *rx_stats = &rxq->stats;
1422         struct net_device *ndev = rxq->ndev;
1423         uint pkt_len = cqe->ppi[0].pkt_len;
1424         u16 rxq_idx = rxq->rxq_idx;
1425         struct napi_struct *napi;
1426         struct xdp_buff xdp = {};
1427         struct sk_buff *skb;
1428         u32 hash_value;
1429         u32 act;
1430
1431         rxq->rx_cq.work_done++;
1432         napi = &rxq->rx_cq.napi;
1433
1434         if (!buf_va) {
1435                 ++ndev->stats.rx_dropped;
1436                 return;
1437         }
1438
1439         act = mana_run_xdp(ndev, rxq, &xdp, buf_va, pkt_len);
1440
1441         if (act == XDP_REDIRECT && !rxq->xdp_rc)
1442                 return;
1443
1444         if (act != XDP_PASS && act != XDP_TX)
1445                 goto drop_xdp;
1446
1447         skb = mana_build_skb(rxq, buf_va, pkt_len, &xdp);
1448
1449         if (!skb)
1450                 goto drop;
1451
1452         if (from_pool)
1453                 skb_mark_for_recycle(skb);
1454
1455         skb->dev = napi->dev;
1456
1457         skb->protocol = eth_type_trans(skb, ndev);
1458         skb_checksum_none_assert(skb);
1459         skb_record_rx_queue(skb, rxq_idx);
1460
1461         if ((ndev->features & NETIF_F_RXCSUM) && cqe->rx_iphdr_csum_succeed) {
1462                 if (cqe->rx_tcp_csum_succeed || cqe->rx_udp_csum_succeed)
1463                         skb->ip_summed = CHECKSUM_UNNECESSARY;
1464         }
1465
1466         if (cqe->rx_hashtype != 0 && (ndev->features & NETIF_F_RXHASH)) {
1467                 hash_value = cqe->ppi[0].pkt_hash;
1468
1469                 if (cqe->rx_hashtype & MANA_HASH_L4)
1470                         skb_set_hash(skb, hash_value, PKT_HASH_TYPE_L4);
1471                 else
1472                         skb_set_hash(skb, hash_value, PKT_HASH_TYPE_L3);
1473         }
1474
1475         if (cqe->rx_vlantag_present) {
1476                 u16 vlan_tci = cqe->rx_vlan_id;
1477
1478                 __vlan_hwaccel_put_tag(skb, htons(ETH_P_8021Q), vlan_tci);
1479         }
1480
1481         u64_stats_update_begin(&rx_stats->syncp);
1482         rx_stats->packets++;
1483         rx_stats->bytes += pkt_len;
1484
1485         if (act == XDP_TX)
1486                 rx_stats->xdp_tx++;
1487         u64_stats_update_end(&rx_stats->syncp);
1488
1489         if (act == XDP_TX) {
1490                 skb_set_queue_mapping(skb, rxq_idx);
1491                 mana_xdp_tx(skb, ndev);
1492                 return;
1493         }
1494
1495         napi_gro_receive(napi, skb);
1496
1497         return;
1498
1499 drop_xdp:
1500         u64_stats_update_begin(&rx_stats->syncp);
1501         rx_stats->xdp_drop++;
1502         u64_stats_update_end(&rx_stats->syncp);
1503
1504 drop:
1505         if (from_pool) {
1506                 page_pool_recycle_direct(rxq->page_pool,
1507                                          virt_to_head_page(buf_va));
1508         } else {
1509                 WARN_ON_ONCE(rxq->xdp_save_va);
1510                 /* Save for reuse */
1511                 rxq->xdp_save_va = buf_va;
1512         }
1513
1514         ++ndev->stats.rx_dropped;
1515
1516         return;
1517 }
1518
1519 static void *mana_get_rxfrag(struct mana_rxq *rxq, struct device *dev,
1520                              dma_addr_t *da, bool *from_pool, bool is_napi)
1521 {
1522         struct page *page;
1523         void *va;
1524
1525         *from_pool = false;
1526
1527         /* Reuse XDP dropped page if available */
1528         if (rxq->xdp_save_va) {
1529                 va = rxq->xdp_save_va;
1530                 rxq->xdp_save_va = NULL;
1531         } else if (rxq->alloc_size > PAGE_SIZE) {
1532                 if (is_napi)
1533                         va = napi_alloc_frag(rxq->alloc_size);
1534                 else
1535                         va = netdev_alloc_frag(rxq->alloc_size);
1536
1537                 if (!va)
1538                         return NULL;
1539
1540                 page = virt_to_head_page(va);
1541                 /* Check if the frag falls back to single page */
1542                 if (compound_order(page) < get_order(rxq->alloc_size)) {
1543                         put_page(page);
1544                         return NULL;
1545                 }
1546         } else {
1547                 page = page_pool_dev_alloc_pages(rxq->page_pool);
1548                 if (!page)
1549                         return NULL;
1550
1551                 *from_pool = true;
1552                 va = page_to_virt(page);
1553         }
1554
1555         *da = dma_map_single(dev, va + rxq->headroom, rxq->datasize,
1556                              DMA_FROM_DEVICE);
1557         if (dma_mapping_error(dev, *da)) {
1558                 if (*from_pool)
1559                         page_pool_put_full_page(rxq->page_pool, page, false);
1560                 else
1561                         put_page(virt_to_head_page(va));
1562
1563                 return NULL;
1564         }
1565
1566         return va;
1567 }
1568
1569 /* Allocate frag for rx buffer, and save the old buf */
1570 static void mana_refill_rx_oob(struct device *dev, struct mana_rxq *rxq,
1571                                struct mana_recv_buf_oob *rxoob, void **old_buf,
1572                                bool *old_fp)
1573 {
1574         bool from_pool;
1575         dma_addr_t da;
1576         void *va;
1577
1578         va = mana_get_rxfrag(rxq, dev, &da, &from_pool, true);
1579         if (!va)
1580                 return;
1581
1582         dma_unmap_single(dev, rxoob->sgl[0].address, rxq->datasize,
1583                          DMA_FROM_DEVICE);
1584         *old_buf = rxoob->buf_va;
1585         *old_fp = rxoob->from_pool;
1586
1587         rxoob->buf_va = va;
1588         rxoob->sgl[0].address = da;
1589         rxoob->from_pool = from_pool;
1590 }
1591
1592 static void mana_process_rx_cqe(struct mana_rxq *rxq, struct mana_cq *cq,
1593                                 struct gdma_comp *cqe)
1594 {
1595         struct mana_rxcomp_oob *oob = (struct mana_rxcomp_oob *)cqe->cqe_data;
1596         struct gdma_context *gc = rxq->gdma_rq->gdma_dev->gdma_context;
1597         struct net_device *ndev = rxq->ndev;
1598         struct mana_recv_buf_oob *rxbuf_oob;
1599         struct mana_port_context *apc;
1600         struct device *dev = gc->dev;
1601         void *old_buf = NULL;
1602         u32 curr, pktlen;
1603         bool old_fp;
1604
1605         apc = netdev_priv(ndev);
1606
1607         switch (oob->cqe_hdr.cqe_type) {
1608         case CQE_RX_OKAY:
1609                 break;
1610
1611         case CQE_RX_TRUNCATED:
1612                 ++ndev->stats.rx_dropped;
1613                 rxbuf_oob = &rxq->rx_oobs[rxq->buf_index];
1614                 netdev_warn_once(ndev, "Dropped a truncated packet\n");
1615                 goto drop;
1616
1617         case CQE_RX_COALESCED_4:
1618                 netdev_err(ndev, "RX coalescing is unsupported\n");
1619                 apc->eth_stats.rx_coalesced_err++;
1620                 return;
1621
1622         case CQE_RX_OBJECT_FENCE:
1623                 complete(&rxq->fence_event);
1624                 return;
1625
1626         default:
1627                 netdev_err(ndev, "Unknown RX CQE type = %d\n",
1628                            oob->cqe_hdr.cqe_type);
1629                 apc->eth_stats.rx_cqe_unknown_type++;
1630                 return;
1631         }
1632
1633         pktlen = oob->ppi[0].pkt_len;
1634
1635         if (pktlen == 0) {
1636                 /* data packets should never have packetlength of zero */
1637                 netdev_err(ndev, "RX pkt len=0, rq=%u, cq=%u, rxobj=0x%llx\n",
1638                            rxq->gdma_id, cq->gdma_id, rxq->rxobj);
1639                 return;
1640         }
1641
1642         curr = rxq->buf_index;
1643         rxbuf_oob = &rxq->rx_oobs[curr];
1644         WARN_ON_ONCE(rxbuf_oob->wqe_inf.wqe_size_in_bu != 1);
1645
1646         mana_refill_rx_oob(dev, rxq, rxbuf_oob, &old_buf, &old_fp);
1647
1648         /* Unsuccessful refill will have old_buf == NULL.
1649          * In this case, mana_rx_skb() will drop the packet.
1650          */
1651         mana_rx_skb(old_buf, old_fp, oob, rxq);
1652
1653 drop:
1654         mana_move_wq_tail(rxq->gdma_rq, rxbuf_oob->wqe_inf.wqe_size_in_bu);
1655
1656         mana_post_pkt_rxq(rxq);
1657 }
1658
1659 static void mana_poll_rx_cq(struct mana_cq *cq)
1660 {
1661         struct gdma_comp *comp = cq->gdma_comp_buf;
1662         struct mana_rxq *rxq = cq->rxq;
1663         int comp_read, i;
1664
1665         comp_read = mana_gd_poll_cq(cq->gdma_cq, comp, CQE_POLLING_BUFFER);
1666         WARN_ON_ONCE(comp_read > CQE_POLLING_BUFFER);
1667
1668         rxq->xdp_flush = false;
1669
1670         for (i = 0; i < comp_read; i++) {
1671                 if (WARN_ON_ONCE(comp[i].is_sq))
1672                         return;
1673
1674                 /* verify recv cqe references the right rxq */
1675                 if (WARN_ON_ONCE(comp[i].wq_num != cq->rxq->gdma_id))
1676                         return;
1677
1678                 mana_process_rx_cqe(rxq, cq, &comp[i]);
1679         }
1680
1681         if (comp_read > 0) {
1682                 struct gdma_context *gc = rxq->gdma_rq->gdma_dev->gdma_context;
1683
1684                 mana_gd_wq_ring_doorbell(gc, rxq->gdma_rq);
1685         }
1686
1687         if (rxq->xdp_flush)
1688                 xdp_do_flush();
1689 }
1690
1691 static int mana_cq_handler(void *context, struct gdma_queue *gdma_queue)
1692 {
1693         struct mana_cq *cq = context;
1694         u8 arm_bit;
1695         int w;
1696
1697         WARN_ON_ONCE(cq->gdma_cq != gdma_queue);
1698
1699         if (cq->type == MANA_CQ_TYPE_RX)
1700                 mana_poll_rx_cq(cq);
1701         else
1702                 mana_poll_tx_cq(cq);
1703
1704         w = cq->work_done;
1705
1706         if (w < cq->budget &&
1707             napi_complete_done(&cq->napi, w)) {
1708                 arm_bit = SET_ARM_BIT;
1709         } else {
1710                 arm_bit = 0;
1711         }
1712
1713         mana_gd_ring_cq(gdma_queue, arm_bit);
1714
1715         return w;
1716 }
1717
1718 static int mana_poll(struct napi_struct *napi, int budget)
1719 {
1720         struct mana_cq *cq = container_of(napi, struct mana_cq, napi);
1721         int w;
1722
1723         cq->work_done = 0;
1724         cq->budget = budget;
1725
1726         w = mana_cq_handler(cq, cq->gdma_cq);
1727
1728         return min(w, budget);
1729 }
1730
1731 static void mana_schedule_napi(void *context, struct gdma_queue *gdma_queue)
1732 {
1733         struct mana_cq *cq = context;
1734
1735         napi_schedule_irqoff(&cq->napi);
1736 }
1737
1738 static void mana_deinit_cq(struct mana_port_context *apc, struct mana_cq *cq)
1739 {
1740         struct gdma_dev *gd = apc->ac->gdma_dev;
1741
1742         if (!cq->gdma_cq)
1743                 return;
1744
1745         mana_gd_destroy_queue(gd->gdma_context, cq->gdma_cq);
1746 }
1747
1748 static void mana_deinit_txq(struct mana_port_context *apc, struct mana_txq *txq)
1749 {
1750         struct gdma_dev *gd = apc->ac->gdma_dev;
1751
1752         if (!txq->gdma_sq)
1753                 return;
1754
1755         mana_gd_destroy_queue(gd->gdma_context, txq->gdma_sq);
1756 }
1757
1758 static void mana_destroy_txq(struct mana_port_context *apc)
1759 {
1760         struct napi_struct *napi;
1761         int i;
1762
1763         if (!apc->tx_qp)
1764                 return;
1765
1766         for (i = 0; i < apc->num_queues; i++) {
1767                 napi = &apc->tx_qp[i].tx_cq.napi;
1768                 napi_synchronize(napi);
1769                 napi_disable(napi);
1770                 netif_napi_del(napi);
1771
1772                 mana_destroy_wq_obj(apc, GDMA_SQ, apc->tx_qp[i].tx_object);
1773
1774                 mana_deinit_cq(apc, &apc->tx_qp[i].tx_cq);
1775
1776                 mana_deinit_txq(apc, &apc->tx_qp[i].txq);
1777         }
1778
1779         kfree(apc->tx_qp);
1780         apc->tx_qp = NULL;
1781 }
1782
1783 static int mana_create_txq(struct mana_port_context *apc,
1784                            struct net_device *net)
1785 {
1786         struct mana_context *ac = apc->ac;
1787         struct gdma_dev *gd = ac->gdma_dev;
1788         struct mana_obj_spec wq_spec;
1789         struct mana_obj_spec cq_spec;
1790         struct gdma_queue_spec spec;
1791         struct gdma_context *gc;
1792         struct mana_txq *txq;
1793         struct mana_cq *cq;
1794         u32 txq_size;
1795         u32 cq_size;
1796         int err;
1797         int i;
1798
1799         apc->tx_qp = kcalloc(apc->num_queues, sizeof(struct mana_tx_qp),
1800                              GFP_KERNEL);
1801         if (!apc->tx_qp)
1802                 return -ENOMEM;
1803
1804         /*  The minimum size of the WQE is 32 bytes, hence
1805          *  MAX_SEND_BUFFERS_PER_QUEUE represents the maximum number of WQEs
1806          *  the SQ can store. This value is then used to size other queues
1807          *  to prevent overflow.
1808          */
1809         txq_size = MAX_SEND_BUFFERS_PER_QUEUE * 32;
1810         BUILD_BUG_ON(!PAGE_ALIGNED(txq_size));
1811
1812         cq_size = MAX_SEND_BUFFERS_PER_QUEUE * COMP_ENTRY_SIZE;
1813         cq_size = PAGE_ALIGN(cq_size);
1814
1815         gc = gd->gdma_context;
1816
1817         for (i = 0; i < apc->num_queues; i++) {
1818                 apc->tx_qp[i].tx_object = INVALID_MANA_HANDLE;
1819
1820                 /* Create SQ */
1821                 txq = &apc->tx_qp[i].txq;
1822
1823                 u64_stats_init(&txq->stats.syncp);
1824                 txq->ndev = net;
1825                 txq->net_txq = netdev_get_tx_queue(net, i);
1826                 txq->vp_offset = apc->tx_vp_offset;
1827                 skb_queue_head_init(&txq->pending_skbs);
1828
1829                 memset(&spec, 0, sizeof(spec));
1830                 spec.type = GDMA_SQ;
1831                 spec.monitor_avl_buf = true;
1832                 spec.queue_size = txq_size;
1833                 err = mana_gd_create_mana_wq_cq(gd, &spec, &txq->gdma_sq);
1834                 if (err)
1835                         goto out;
1836
1837                 /* Create SQ's CQ */
1838                 cq = &apc->tx_qp[i].tx_cq;
1839                 cq->type = MANA_CQ_TYPE_TX;
1840
1841                 cq->txq = txq;
1842
1843                 memset(&spec, 0, sizeof(spec));
1844                 spec.type = GDMA_CQ;
1845                 spec.monitor_avl_buf = false;
1846                 spec.queue_size = cq_size;
1847                 spec.cq.callback = mana_schedule_napi;
1848                 spec.cq.parent_eq = ac->eqs[i].eq;
1849                 spec.cq.context = cq;
1850                 err = mana_gd_create_mana_wq_cq(gd, &spec, &cq->gdma_cq);
1851                 if (err)
1852                         goto out;
1853
1854                 memset(&wq_spec, 0, sizeof(wq_spec));
1855                 memset(&cq_spec, 0, sizeof(cq_spec));
1856
1857                 wq_spec.gdma_region = txq->gdma_sq->mem_info.dma_region_handle;
1858                 wq_spec.queue_size = txq->gdma_sq->queue_size;
1859
1860                 cq_spec.gdma_region = cq->gdma_cq->mem_info.dma_region_handle;
1861                 cq_spec.queue_size = cq->gdma_cq->queue_size;
1862                 cq_spec.modr_ctx_id = 0;
1863                 cq_spec.attached_eq = cq->gdma_cq->cq.parent->id;
1864
1865                 err = mana_create_wq_obj(apc, apc->port_handle, GDMA_SQ,
1866                                          &wq_spec, &cq_spec,
1867                                          &apc->tx_qp[i].tx_object);
1868
1869                 if (err)
1870                         goto out;
1871
1872                 txq->gdma_sq->id = wq_spec.queue_index;
1873                 cq->gdma_cq->id = cq_spec.queue_index;
1874
1875                 txq->gdma_sq->mem_info.dma_region_handle =
1876                         GDMA_INVALID_DMA_REGION;
1877                 cq->gdma_cq->mem_info.dma_region_handle =
1878                         GDMA_INVALID_DMA_REGION;
1879
1880                 txq->gdma_txq_id = txq->gdma_sq->id;
1881
1882                 cq->gdma_id = cq->gdma_cq->id;
1883
1884                 if (WARN_ON(cq->gdma_id >= gc->max_num_cqs)) {
1885                         err = -EINVAL;
1886                         goto out;
1887                 }
1888
1889                 gc->cq_table[cq->gdma_id] = cq->gdma_cq;
1890
1891                 netif_napi_add_tx(net, &cq->napi, mana_poll);
1892                 napi_enable(&cq->napi);
1893
1894                 mana_gd_ring_cq(cq->gdma_cq, SET_ARM_BIT);
1895         }
1896
1897         return 0;
1898 out:
1899         mana_destroy_txq(apc);
1900         return err;
1901 }
1902
1903 static void mana_destroy_rxq(struct mana_port_context *apc,
1904                              struct mana_rxq *rxq, bool validate_state)
1905
1906 {
1907         struct gdma_context *gc = apc->ac->gdma_dev->gdma_context;
1908         struct mana_recv_buf_oob *rx_oob;
1909         struct device *dev = gc->dev;
1910         struct napi_struct *napi;
1911         struct page *page;
1912         int i;
1913
1914         if (!rxq)
1915                 return;
1916
1917         napi = &rxq->rx_cq.napi;
1918
1919         if (validate_state)
1920                 napi_synchronize(napi);
1921
1922         napi_disable(napi);
1923
1924         xdp_rxq_info_unreg(&rxq->xdp_rxq);
1925
1926         netif_napi_del(napi);
1927
1928         mana_destroy_wq_obj(apc, GDMA_RQ, rxq->rxobj);
1929
1930         mana_deinit_cq(apc, &rxq->rx_cq);
1931
1932         if (rxq->xdp_save_va)
1933                 put_page(virt_to_head_page(rxq->xdp_save_va));
1934
1935         for (i = 0; i < rxq->num_rx_buf; i++) {
1936                 rx_oob = &rxq->rx_oobs[i];
1937
1938                 if (!rx_oob->buf_va)
1939                         continue;
1940
1941                 dma_unmap_single(dev, rx_oob->sgl[0].address,
1942                                  rx_oob->sgl[0].size, DMA_FROM_DEVICE);
1943
1944                 page = virt_to_head_page(rx_oob->buf_va);
1945
1946                 if (rx_oob->from_pool)
1947                         page_pool_put_full_page(rxq->page_pool, page, false);
1948                 else
1949                         put_page(page);
1950
1951                 rx_oob->buf_va = NULL;
1952         }
1953
1954         page_pool_destroy(rxq->page_pool);
1955
1956         if (rxq->gdma_rq)
1957                 mana_gd_destroy_queue(gc, rxq->gdma_rq);
1958
1959         kfree(rxq);
1960 }
1961
1962 static int mana_fill_rx_oob(struct mana_recv_buf_oob *rx_oob, u32 mem_key,
1963                             struct mana_rxq *rxq, struct device *dev)
1964 {
1965         struct mana_port_context *mpc = netdev_priv(rxq->ndev);
1966         bool from_pool = false;
1967         dma_addr_t da;
1968         void *va;
1969
1970         if (mpc->rxbufs_pre)
1971                 va = mana_get_rxbuf_pre(rxq, &da);
1972         else
1973                 va = mana_get_rxfrag(rxq, dev, &da, &from_pool, false);
1974
1975         if (!va)
1976                 return -ENOMEM;
1977
1978         rx_oob->buf_va = va;
1979         rx_oob->from_pool = from_pool;
1980
1981         rx_oob->sgl[0].address = da;
1982         rx_oob->sgl[0].size = rxq->datasize;
1983         rx_oob->sgl[0].mem_key = mem_key;
1984
1985         return 0;
1986 }
1987
1988 #define MANA_WQE_HEADER_SIZE 16
1989 #define MANA_WQE_SGE_SIZE 16
1990
1991 static int mana_alloc_rx_wqe(struct mana_port_context *apc,
1992                              struct mana_rxq *rxq, u32 *rxq_size, u32 *cq_size)
1993 {
1994         struct gdma_context *gc = apc->ac->gdma_dev->gdma_context;
1995         struct mana_recv_buf_oob *rx_oob;
1996         struct device *dev = gc->dev;
1997         u32 buf_idx;
1998         int ret;
1999
2000         WARN_ON(rxq->datasize == 0);
2001
2002         *rxq_size = 0;
2003         *cq_size = 0;
2004
2005         for (buf_idx = 0; buf_idx < rxq->num_rx_buf; buf_idx++) {
2006                 rx_oob = &rxq->rx_oobs[buf_idx];
2007                 memset(rx_oob, 0, sizeof(*rx_oob));
2008
2009                 rx_oob->num_sge = 1;
2010
2011                 ret = mana_fill_rx_oob(rx_oob, apc->ac->gdma_dev->gpa_mkey, rxq,
2012                                        dev);
2013                 if (ret)
2014                         return ret;
2015
2016                 rx_oob->wqe_req.sgl = rx_oob->sgl;
2017                 rx_oob->wqe_req.num_sge = rx_oob->num_sge;
2018                 rx_oob->wqe_req.inline_oob_size = 0;
2019                 rx_oob->wqe_req.inline_oob_data = NULL;
2020                 rx_oob->wqe_req.flags = 0;
2021                 rx_oob->wqe_req.client_data_unit = 0;
2022
2023                 *rxq_size += ALIGN(MANA_WQE_HEADER_SIZE +
2024                                    MANA_WQE_SGE_SIZE * rx_oob->num_sge, 32);
2025                 *cq_size += COMP_ENTRY_SIZE;
2026         }
2027
2028         return 0;
2029 }
2030
2031 static int mana_push_wqe(struct mana_rxq *rxq)
2032 {
2033         struct mana_recv_buf_oob *rx_oob;
2034         u32 buf_idx;
2035         int err;
2036
2037         for (buf_idx = 0; buf_idx < rxq->num_rx_buf; buf_idx++) {
2038                 rx_oob = &rxq->rx_oobs[buf_idx];
2039
2040                 err = mana_gd_post_and_ring(rxq->gdma_rq, &rx_oob->wqe_req,
2041                                             &rx_oob->wqe_inf);
2042                 if (err)
2043                         return -ENOSPC;
2044         }
2045
2046         return 0;
2047 }
2048
2049 static int mana_create_page_pool(struct mana_rxq *rxq, struct gdma_context *gc)
2050 {
2051         struct page_pool_params pprm = {};
2052         int ret;
2053
2054         pprm.pool_size = RX_BUFFERS_PER_QUEUE;
2055         pprm.nid = gc->numa_node;
2056         pprm.napi = &rxq->rx_cq.napi;
2057
2058         rxq->page_pool = page_pool_create(&pprm);
2059
2060         if (IS_ERR(rxq->page_pool)) {
2061                 ret = PTR_ERR(rxq->page_pool);
2062                 rxq->page_pool = NULL;
2063                 return ret;
2064         }
2065
2066         return 0;
2067 }
2068
2069 static struct mana_rxq *mana_create_rxq(struct mana_port_context *apc,
2070                                         u32 rxq_idx, struct mana_eq *eq,
2071                                         struct net_device *ndev)
2072 {
2073         struct gdma_dev *gd = apc->ac->gdma_dev;
2074         struct mana_obj_spec wq_spec;
2075         struct mana_obj_spec cq_spec;
2076         struct gdma_queue_spec spec;
2077         struct mana_cq *cq = NULL;
2078         struct gdma_context *gc;
2079         u32 cq_size, rq_size;
2080         struct mana_rxq *rxq;
2081         int err;
2082
2083         gc = gd->gdma_context;
2084
2085         rxq = kzalloc(struct_size(rxq, rx_oobs, RX_BUFFERS_PER_QUEUE),
2086                       GFP_KERNEL);
2087         if (!rxq)
2088                 return NULL;
2089
2090         rxq->ndev = ndev;
2091         rxq->num_rx_buf = RX_BUFFERS_PER_QUEUE;
2092         rxq->rxq_idx = rxq_idx;
2093         rxq->rxobj = INVALID_MANA_HANDLE;
2094
2095         mana_get_rxbuf_cfg(ndev->mtu, &rxq->datasize, &rxq->alloc_size,
2096                            &rxq->headroom);
2097
2098         /* Create page pool for RX queue */
2099         err = mana_create_page_pool(rxq, gc);
2100         if (err) {
2101                 netdev_err(ndev, "Create page pool err:%d\n", err);
2102                 goto out;
2103         }
2104
2105         err = mana_alloc_rx_wqe(apc, rxq, &rq_size, &cq_size);
2106         if (err)
2107                 goto out;
2108
2109         rq_size = PAGE_ALIGN(rq_size);
2110         cq_size = PAGE_ALIGN(cq_size);
2111
2112         /* Create RQ */
2113         memset(&spec, 0, sizeof(spec));
2114         spec.type = GDMA_RQ;
2115         spec.monitor_avl_buf = true;
2116         spec.queue_size = rq_size;
2117         err = mana_gd_create_mana_wq_cq(gd, &spec, &rxq->gdma_rq);
2118         if (err)
2119                 goto out;
2120
2121         /* Create RQ's CQ */
2122         cq = &rxq->rx_cq;
2123         cq->type = MANA_CQ_TYPE_RX;
2124         cq->rxq = rxq;
2125
2126         memset(&spec, 0, sizeof(spec));
2127         spec.type = GDMA_CQ;
2128         spec.monitor_avl_buf = false;
2129         spec.queue_size = cq_size;
2130         spec.cq.callback = mana_schedule_napi;
2131         spec.cq.parent_eq = eq->eq;
2132         spec.cq.context = cq;
2133         err = mana_gd_create_mana_wq_cq(gd, &spec, &cq->gdma_cq);
2134         if (err)
2135                 goto out;
2136
2137         memset(&wq_spec, 0, sizeof(wq_spec));
2138         memset(&cq_spec, 0, sizeof(cq_spec));
2139         wq_spec.gdma_region = rxq->gdma_rq->mem_info.dma_region_handle;
2140         wq_spec.queue_size = rxq->gdma_rq->queue_size;
2141
2142         cq_spec.gdma_region = cq->gdma_cq->mem_info.dma_region_handle;
2143         cq_spec.queue_size = cq->gdma_cq->queue_size;
2144         cq_spec.modr_ctx_id = 0;
2145         cq_spec.attached_eq = cq->gdma_cq->cq.parent->id;
2146
2147         err = mana_create_wq_obj(apc, apc->port_handle, GDMA_RQ,
2148                                  &wq_spec, &cq_spec, &rxq->rxobj);
2149         if (err)
2150                 goto out;
2151
2152         rxq->gdma_rq->id = wq_spec.queue_index;
2153         cq->gdma_cq->id = cq_spec.queue_index;
2154
2155         rxq->gdma_rq->mem_info.dma_region_handle = GDMA_INVALID_DMA_REGION;
2156         cq->gdma_cq->mem_info.dma_region_handle = GDMA_INVALID_DMA_REGION;
2157
2158         rxq->gdma_id = rxq->gdma_rq->id;
2159         cq->gdma_id = cq->gdma_cq->id;
2160
2161         err = mana_push_wqe(rxq);
2162         if (err)
2163                 goto out;
2164
2165         if (WARN_ON(cq->gdma_id >= gc->max_num_cqs)) {
2166                 err = -EINVAL;
2167                 goto out;
2168         }
2169
2170         gc->cq_table[cq->gdma_id] = cq->gdma_cq;
2171
2172         netif_napi_add_weight(ndev, &cq->napi, mana_poll, 1);
2173
2174         WARN_ON(xdp_rxq_info_reg(&rxq->xdp_rxq, ndev, rxq_idx,
2175                                  cq->napi.napi_id));
2176         WARN_ON(xdp_rxq_info_reg_mem_model(&rxq->xdp_rxq, MEM_TYPE_PAGE_POOL,
2177                                            rxq->page_pool));
2178
2179         napi_enable(&cq->napi);
2180
2181         mana_gd_ring_cq(cq->gdma_cq, SET_ARM_BIT);
2182 out:
2183         if (!err)
2184                 return rxq;
2185
2186         netdev_err(ndev, "Failed to create RXQ: err = %d\n", err);
2187
2188         mana_destroy_rxq(apc, rxq, false);
2189
2190         if (cq)
2191                 mana_deinit_cq(apc, cq);
2192
2193         return NULL;
2194 }
2195
2196 static int mana_add_rx_queues(struct mana_port_context *apc,
2197                               struct net_device *ndev)
2198 {
2199         struct mana_context *ac = apc->ac;
2200         struct mana_rxq *rxq;
2201         int err = 0;
2202         int i;
2203
2204         for (i = 0; i < apc->num_queues; i++) {
2205                 rxq = mana_create_rxq(apc, i, &ac->eqs[i], ndev);
2206                 if (!rxq) {
2207                         err = -ENOMEM;
2208                         goto out;
2209                 }
2210
2211                 u64_stats_init(&rxq->stats.syncp);
2212
2213                 apc->rxqs[i] = rxq;
2214         }
2215
2216         apc->default_rxobj = apc->rxqs[0]->rxobj;
2217 out:
2218         return err;
2219 }
2220
2221 static void mana_destroy_vport(struct mana_port_context *apc)
2222 {
2223         struct gdma_dev *gd = apc->ac->gdma_dev;
2224         struct mana_rxq *rxq;
2225         u32 rxq_idx;
2226
2227         for (rxq_idx = 0; rxq_idx < apc->num_queues; rxq_idx++) {
2228                 rxq = apc->rxqs[rxq_idx];
2229                 if (!rxq)
2230                         continue;
2231
2232                 mana_destroy_rxq(apc, rxq, true);
2233                 apc->rxqs[rxq_idx] = NULL;
2234         }
2235
2236         mana_destroy_txq(apc);
2237         mana_uncfg_vport(apc);
2238
2239         if (gd->gdma_context->is_pf)
2240                 mana_pf_deregister_hw_vport(apc);
2241 }
2242
2243 static int mana_create_vport(struct mana_port_context *apc,
2244                              struct net_device *net)
2245 {
2246         struct gdma_dev *gd = apc->ac->gdma_dev;
2247         int err;
2248
2249         apc->default_rxobj = INVALID_MANA_HANDLE;
2250
2251         if (gd->gdma_context->is_pf) {
2252                 err = mana_pf_register_hw_vport(apc);
2253                 if (err)
2254                         return err;
2255         }
2256
2257         err = mana_cfg_vport(apc, gd->pdid, gd->doorbell);
2258         if (err)
2259                 return err;
2260
2261         return mana_create_txq(apc, net);
2262 }
2263
2264 static void mana_rss_table_init(struct mana_port_context *apc)
2265 {
2266         int i;
2267
2268         for (i = 0; i < MANA_INDIRECT_TABLE_SIZE; i++)
2269                 apc->indir_table[i] =
2270                         ethtool_rxfh_indir_default(i, apc->num_queues);
2271 }
2272
2273 int mana_config_rss(struct mana_port_context *apc, enum TRI_STATE rx,
2274                     bool update_hash, bool update_tab)
2275 {
2276         u32 queue_idx;
2277         int err;
2278         int i;
2279
2280         if (update_tab) {
2281                 for (i = 0; i < MANA_INDIRECT_TABLE_SIZE; i++) {
2282                         queue_idx = apc->indir_table[i];
2283                         apc->rxobj_table[i] = apc->rxqs[queue_idx]->rxobj;
2284                 }
2285         }
2286
2287         err = mana_cfg_vport_steering(apc, rx, true, update_hash, update_tab);
2288         if (err)
2289                 return err;
2290
2291         mana_fence_rqs(apc);
2292
2293         return 0;
2294 }
2295
2296 static int mana_init_port(struct net_device *ndev)
2297 {
2298         struct mana_port_context *apc = netdev_priv(ndev);
2299         u32 max_txq, max_rxq, max_queues;
2300         int port_idx = apc->port_idx;
2301         u32 num_indirect_entries;
2302         int err;
2303
2304         err = mana_init_port_context(apc);
2305         if (err)
2306                 return err;
2307
2308         err = mana_query_vport_cfg(apc, port_idx, &max_txq, &max_rxq,
2309                                    &num_indirect_entries);
2310         if (err) {
2311                 netdev_err(ndev, "Failed to query info for vPort %d\n",
2312                            port_idx);
2313                 goto reset_apc;
2314         }
2315
2316         max_queues = min_t(u32, max_txq, max_rxq);
2317         if (apc->max_queues > max_queues)
2318                 apc->max_queues = max_queues;
2319
2320         if (apc->num_queues > apc->max_queues)
2321                 apc->num_queues = apc->max_queues;
2322
2323         eth_hw_addr_set(ndev, apc->mac_addr);
2324
2325         return 0;
2326
2327 reset_apc:
2328         kfree(apc->rxqs);
2329         apc->rxqs = NULL;
2330         return err;
2331 }
2332
2333 int mana_alloc_queues(struct net_device *ndev)
2334 {
2335         struct mana_port_context *apc = netdev_priv(ndev);
2336         struct gdma_dev *gd = apc->ac->gdma_dev;
2337         int err;
2338
2339         err = mana_create_vport(apc, ndev);
2340         if (err)
2341                 return err;
2342
2343         err = netif_set_real_num_tx_queues(ndev, apc->num_queues);
2344         if (err)
2345                 goto destroy_vport;
2346
2347         err = mana_add_rx_queues(apc, ndev);
2348         if (err)
2349                 goto destroy_vport;
2350
2351         apc->rss_state = apc->num_queues > 1 ? TRI_STATE_TRUE : TRI_STATE_FALSE;
2352
2353         err = netif_set_real_num_rx_queues(ndev, apc->num_queues);
2354         if (err)
2355                 goto destroy_vport;
2356
2357         mana_rss_table_init(apc);
2358
2359         err = mana_config_rss(apc, TRI_STATE_TRUE, true, true);
2360         if (err)
2361                 goto destroy_vport;
2362
2363         if (gd->gdma_context->is_pf) {
2364                 err = mana_pf_register_filter(apc);
2365                 if (err)
2366                         goto destroy_vport;
2367         }
2368
2369         mana_chn_setxdp(apc, mana_xdp_get(apc));
2370
2371         return 0;
2372
2373 destroy_vport:
2374         mana_destroy_vport(apc);
2375         return err;
2376 }
2377
2378 int mana_attach(struct net_device *ndev)
2379 {
2380         struct mana_port_context *apc = netdev_priv(ndev);
2381         int err;
2382
2383         ASSERT_RTNL();
2384
2385         err = mana_init_port(ndev);
2386         if (err)
2387                 return err;
2388
2389         if (apc->port_st_save) {
2390                 err = mana_alloc_queues(ndev);
2391                 if (err) {
2392                         mana_cleanup_port_context(apc);
2393                         return err;
2394                 }
2395         }
2396
2397         apc->port_is_up = apc->port_st_save;
2398
2399         /* Ensure port state updated before txq state */
2400         smp_wmb();
2401
2402         if (apc->port_is_up)
2403                 netif_carrier_on(ndev);
2404
2405         netif_device_attach(ndev);
2406
2407         return 0;
2408 }
2409
2410 static int mana_dealloc_queues(struct net_device *ndev)
2411 {
2412         struct mana_port_context *apc = netdev_priv(ndev);
2413         struct gdma_dev *gd = apc->ac->gdma_dev;
2414         struct mana_txq *txq;
2415         int i, err;
2416
2417         if (apc->port_is_up)
2418                 return -EINVAL;
2419
2420         mana_chn_setxdp(apc, NULL);
2421
2422         if (gd->gdma_context->is_pf)
2423                 mana_pf_deregister_filter(apc);
2424
2425         /* No packet can be transmitted now since apc->port_is_up is false.
2426          * There is still a tiny chance that mana_poll_tx_cq() can re-enable
2427          * a txq because it may not timely see apc->port_is_up being cleared
2428          * to false, but it doesn't matter since mana_start_xmit() drops any
2429          * new packets due to apc->port_is_up being false.
2430          *
2431          * Drain all the in-flight TX packets
2432          */
2433         for (i = 0; i < apc->num_queues; i++) {
2434                 txq = &apc->tx_qp[i].txq;
2435
2436                 while (atomic_read(&txq->pending_sends) > 0)
2437                         usleep_range(1000, 2000);
2438         }
2439
2440         /* We're 100% sure the queues can no longer be woken up, because
2441          * we're sure now mana_poll_tx_cq() can't be running.
2442          */
2443
2444         apc->rss_state = TRI_STATE_FALSE;
2445         err = mana_config_rss(apc, TRI_STATE_FALSE, false, false);
2446         if (err) {
2447                 netdev_err(ndev, "Failed to disable vPort: %d\n", err);
2448                 return err;
2449         }
2450
2451         mana_destroy_vport(apc);
2452
2453         return 0;
2454 }
2455
2456 int mana_detach(struct net_device *ndev, bool from_close)
2457 {
2458         struct mana_port_context *apc = netdev_priv(ndev);
2459         int err;
2460
2461         ASSERT_RTNL();
2462
2463         apc->port_st_save = apc->port_is_up;
2464         apc->port_is_up = false;
2465
2466         /* Ensure port state updated before txq state */
2467         smp_wmb();
2468
2469         netif_tx_disable(ndev);
2470         netif_carrier_off(ndev);
2471
2472         if (apc->port_st_save) {
2473                 err = mana_dealloc_queues(ndev);
2474                 if (err)
2475                         return err;
2476         }
2477
2478         if (!from_close) {
2479                 netif_device_detach(ndev);
2480                 mana_cleanup_port_context(apc);
2481         }
2482
2483         return 0;
2484 }
2485
2486 static int mana_probe_port(struct mana_context *ac, int port_idx,
2487                            struct net_device **ndev_storage)
2488 {
2489         struct gdma_context *gc = ac->gdma_dev->gdma_context;
2490         struct mana_port_context *apc;
2491         struct net_device *ndev;
2492         int err;
2493
2494         ndev = alloc_etherdev_mq(sizeof(struct mana_port_context),
2495                                  gc->max_num_queues);
2496         if (!ndev)
2497                 return -ENOMEM;
2498
2499         *ndev_storage = ndev;
2500
2501         apc = netdev_priv(ndev);
2502         apc->ac = ac;
2503         apc->ndev = ndev;
2504         apc->max_queues = gc->max_num_queues;
2505         apc->num_queues = gc->max_num_queues;
2506         apc->port_handle = INVALID_MANA_HANDLE;
2507         apc->pf_filter_handle = INVALID_MANA_HANDLE;
2508         apc->port_idx = port_idx;
2509
2510         mutex_init(&apc->vport_mutex);
2511         apc->vport_use_count = 0;
2512
2513         ndev->netdev_ops = &mana_devops;
2514         ndev->ethtool_ops = &mana_ethtool_ops;
2515         ndev->mtu = ETH_DATA_LEN;
2516         ndev->max_mtu = gc->adapter_mtu - ETH_HLEN;
2517         ndev->min_mtu = ETH_MIN_MTU;
2518         ndev->needed_headroom = MANA_HEADROOM;
2519         ndev->dev_port = port_idx;
2520         SET_NETDEV_DEV(ndev, gc->dev);
2521
2522         netif_carrier_off(ndev);
2523
2524         netdev_rss_key_fill(apc->hashkey, MANA_HASH_KEY_SIZE);
2525
2526         err = mana_init_port(ndev);
2527         if (err)
2528                 goto free_net;
2529
2530         netdev_lockdep_set_classes(ndev);
2531
2532         ndev->hw_features = NETIF_F_SG | NETIF_F_IP_CSUM | NETIF_F_IPV6_CSUM;
2533         ndev->hw_features |= NETIF_F_RXCSUM;
2534         ndev->hw_features |= NETIF_F_TSO | NETIF_F_TSO6;
2535         ndev->hw_features |= NETIF_F_RXHASH;
2536         ndev->features = ndev->hw_features | NETIF_F_HW_VLAN_CTAG_TX |
2537                          NETIF_F_HW_VLAN_CTAG_RX;
2538         ndev->vlan_features = ndev->features;
2539         ndev->xdp_features = NETDEV_XDP_ACT_BASIC | NETDEV_XDP_ACT_REDIRECT |
2540                              NETDEV_XDP_ACT_NDO_XMIT;
2541
2542         err = register_netdev(ndev);
2543         if (err) {
2544                 netdev_err(ndev, "Unable to register netdev.\n");
2545                 goto reset_apc;
2546         }
2547
2548         return 0;
2549
2550 reset_apc:
2551         kfree(apc->rxqs);
2552         apc->rxqs = NULL;
2553 free_net:
2554         *ndev_storage = NULL;
2555         netdev_err(ndev, "Failed to probe vPort %d: %d\n", port_idx, err);
2556         free_netdev(ndev);
2557         return err;
2558 }
2559
2560 static void adev_release(struct device *dev)
2561 {
2562         struct mana_adev *madev = container_of(dev, struct mana_adev, adev.dev);
2563
2564         kfree(madev);
2565 }
2566
2567 static void remove_adev(struct gdma_dev *gd)
2568 {
2569         struct auxiliary_device *adev = gd->adev;
2570         int id = adev->id;
2571
2572         auxiliary_device_delete(adev);
2573         auxiliary_device_uninit(adev);
2574
2575         mana_adev_idx_free(id);
2576         gd->adev = NULL;
2577 }
2578
2579 static int add_adev(struct gdma_dev *gd)
2580 {
2581         struct auxiliary_device *adev;
2582         struct mana_adev *madev;
2583         int ret;
2584
2585         madev = kzalloc(sizeof(*madev), GFP_KERNEL);
2586         if (!madev)
2587                 return -ENOMEM;
2588
2589         adev = &madev->adev;
2590         ret = mana_adev_idx_alloc();
2591         if (ret < 0)
2592                 goto idx_fail;
2593         adev->id = ret;
2594
2595         adev->name = "rdma";
2596         adev->dev.parent = gd->gdma_context->dev;
2597         adev->dev.release = adev_release;
2598         madev->mdev = gd;
2599
2600         ret = auxiliary_device_init(adev);
2601         if (ret)
2602                 goto init_fail;
2603
2604         ret = auxiliary_device_add(adev);
2605         if (ret)
2606                 goto add_fail;
2607
2608         gd->adev = adev;
2609         return 0;
2610
2611 add_fail:
2612         auxiliary_device_uninit(adev);
2613
2614 init_fail:
2615         mana_adev_idx_free(adev->id);
2616
2617 idx_fail:
2618         kfree(madev);
2619
2620         return ret;
2621 }
2622
2623 int mana_probe(struct gdma_dev *gd, bool resuming)
2624 {
2625         struct gdma_context *gc = gd->gdma_context;
2626         struct mana_context *ac = gd->driver_data;
2627         struct device *dev = gc->dev;
2628         u16 num_ports = 0;
2629         int err;
2630         int i;
2631
2632         dev_info(dev,
2633                  "Microsoft Azure Network Adapter protocol version: %d.%d.%d\n",
2634                  MANA_MAJOR_VERSION, MANA_MINOR_VERSION, MANA_MICRO_VERSION);
2635
2636         err = mana_gd_register_device(gd);
2637         if (err)
2638                 return err;
2639
2640         if (!resuming) {
2641                 ac = kzalloc(sizeof(*ac), GFP_KERNEL);
2642                 if (!ac)
2643                         return -ENOMEM;
2644
2645                 ac->gdma_dev = gd;
2646                 gd->driver_data = ac;
2647         }
2648
2649         err = mana_create_eq(ac);
2650         if (err)
2651                 goto out;
2652
2653         err = mana_query_device_cfg(ac, MANA_MAJOR_VERSION, MANA_MINOR_VERSION,
2654                                     MANA_MICRO_VERSION, &num_ports);
2655         if (err)
2656                 goto out;
2657
2658         if (!resuming) {
2659                 ac->num_ports = num_ports;
2660         } else {
2661                 if (ac->num_ports != num_ports) {
2662                         dev_err(dev, "The number of vPorts changed: %d->%d\n",
2663                                 ac->num_ports, num_ports);
2664                         err = -EPROTO;
2665                         goto out;
2666                 }
2667         }
2668
2669         if (ac->num_ports == 0)
2670                 dev_err(dev, "Failed to detect any vPort\n");
2671
2672         if (ac->num_ports > MAX_PORTS_IN_MANA_DEV)
2673                 ac->num_ports = MAX_PORTS_IN_MANA_DEV;
2674
2675         if (!resuming) {
2676                 for (i = 0; i < ac->num_ports; i++) {
2677                         err = mana_probe_port(ac, i, &ac->ports[i]);
2678                         if (err)
2679                                 break;
2680                 }
2681         } else {
2682                 for (i = 0; i < ac->num_ports; i++) {
2683                         rtnl_lock();
2684                         err = mana_attach(ac->ports[i]);
2685                         rtnl_unlock();
2686                         if (err)
2687                                 break;
2688                 }
2689         }
2690
2691         err = add_adev(gd);
2692 out:
2693         if (err)
2694                 mana_remove(gd, false);
2695
2696         return err;
2697 }
2698
2699 void mana_remove(struct gdma_dev *gd, bool suspending)
2700 {
2701         struct gdma_context *gc = gd->gdma_context;
2702         struct mana_context *ac = gd->driver_data;
2703         struct device *dev = gc->dev;
2704         struct net_device *ndev;
2705         int err;
2706         int i;
2707
2708         /* adev currently doesn't support suspending, always remove it */
2709         if (gd->adev)
2710                 remove_adev(gd);
2711
2712         for (i = 0; i < ac->num_ports; i++) {
2713                 ndev = ac->ports[i];
2714                 if (!ndev) {
2715                         if (i == 0)
2716                                 dev_err(dev, "No net device to remove\n");
2717                         goto out;
2718                 }
2719
2720                 /* All cleanup actions should stay after rtnl_lock(), otherwise
2721                  * other functions may access partially cleaned up data.
2722                  */
2723                 rtnl_lock();
2724
2725                 err = mana_detach(ndev, false);
2726                 if (err)
2727                         netdev_err(ndev, "Failed to detach vPort %d: %d\n",
2728                                    i, err);
2729
2730                 if (suspending) {
2731                         /* No need to unregister the ndev. */
2732                         rtnl_unlock();
2733                         continue;
2734                 }
2735
2736                 unregister_netdevice(ndev);
2737
2738                 rtnl_unlock();
2739
2740                 free_netdev(ndev);
2741         }
2742
2743         mana_destroy_eq(ac);
2744 out:
2745         mana_gd_deregister_device(gd);
2746
2747         if (suspending)
2748                 return;
2749
2750         gd->driver_data = NULL;
2751         gd->gdma_context = NULL;
2752         kfree(ac);
2753 }