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