drivers/raw/ntb/ntb.c

   1 /* SPDX-License-Identifier: BSD-3-Clause
   2  * Copyright(c) 2019 Intel Corporation.
   3  */
   4 #include <stdint.h>
   5 #include <stdio.h>
   6 #include <string.h>
   7 #include <errno.h>
   8
   9 #include <rte_common.h>
  10 #include <rte_lcore.h>
  11 #include <rte_cycles.h>
  12 #include <rte_eal.h>
  13 #include <rte_log.h>
  14 #include <rte_pci.h>
  15 #include <rte_mbuf.h>
  16 #include <rte_bus_pci.h>
  17 #include <rte_memzone.h>
  18 #include <rte_memcpy.h>
  19 #include <rte_rawdev.h>
  20 #include <rte_rawdev_pmd.h>
  21
  22 #include "ntb_hw_intel.h"
  23 #include "rte_pmd_ntb.h"
  24 #include "ntb.h"
  25
  26 static const struct rte_pci_id pci_id_ntb_map[] = {
  27         { RTE_PCI_DEVICE(NTB_INTEL_VENDOR_ID, NTB_INTEL_DEV_ID_B2B_SKX) },
  28         { RTE_PCI_DEVICE(NTB_INTEL_VENDOR_ID, NTB_INTEL_DEV_ID_B2B_ICX) },
  29         { .vendor_id = 0, /* sentinel */ },
  30 };
  31
  32 /* Align with enum ntb_xstats_idx */
  33 static struct rte_rawdev_xstats_name ntb_xstats_names[] = {
  34         {"Tx-packets"},
  35         {"Tx-bytes"},
  36         {"Tx-errors"},
  37         {"Rx-packets"},
  38         {"Rx-bytes"},
  39         {"Rx-missed"},
  40 };
  41 #define NTB_XSTATS_NUM RTE_DIM(ntb_xstats_names)
  42
  43 static inline void
  44 ntb_link_cleanup(struct rte_rawdev *dev)
  45 {
  46         struct ntb_hw *hw = dev->dev_private;
  47         int status, i;
  48
  49         if (hw->ntb_ops->spad_write == NULL ||
  50             hw->ntb_ops->mw_set_trans == NULL) {
  51                 NTB_LOG(ERR, "Not supported to clean up link.");
  52                 return;
  53         }
  54
  55         /* Clean spad registers. */
  56         for (i = 0; i < hw->spad_cnt; i++) {
  57                 status = (*hw->ntb_ops->spad_write)(dev, i, 0, 0);
  58                 if (status)
  59                         NTB_LOG(ERR, "Failed to clean local spad.");
  60         }
  61
  62         /* Clear mw so that peer cannot access local memory.*/
  63         for (i = 0; i < hw->used_mw_num; i++) {
  64                 status = (*hw->ntb_ops->mw_set_trans)(dev, i, 0, 0);
  65                 if (status)
  66                         NTB_LOG(ERR, "Failed to clean mw.");
  67         }
  68 }
  69
  70 static inline int
  71 ntb_handshake_work(const struct rte_rawdev *dev)
  72 {
  73         struct ntb_hw *hw = dev->dev_private;
  74         uint32_t val;
  75         int ret, i;
  76
  77         if (hw->ntb_ops->spad_write == NULL ||
  78             hw->ntb_ops->mw_set_trans == NULL) {
  79                 NTB_LOG(ERR, "Scratchpad/MW setting is not supported.");
  80                 return -ENOTSUP;
  81         }
  82
  83         /* Tell peer the mw info of local side. */
  84         ret = (*hw->ntb_ops->spad_write)(dev, SPAD_NUM_MWS, 1, hw->mw_cnt);
  85         if (ret < 0)
  86                 return ret;
  87         for (i = 0; i < hw->mw_cnt; i++) {
  88                 NTB_LOG(INFO, "Local %u mw size: 0x%"PRIx64"", i,
  89                                 hw->mw_size[i]);
  90                 val = hw->mw_size[i] >> 32;
  91                 ret = (*hw->ntb_ops->spad_write)(dev, SPAD_MW0_SZ_H + 2 * i,
  92                                                  1, val);
  93                 if (ret < 0)
  94                         return ret;
  95                 val = hw->mw_size[i];
  96                 ret = (*hw->ntb_ops->spad_write)(dev, SPAD_MW0_SZ_L + 2 * i,
  97                                                  1, val);
  98                 if (ret < 0)
  99                         return ret;
 100         }
 101
 102         /* Tell peer about the queue info and map memory to the peer. */
 103         ret = (*hw->ntb_ops->spad_write)(dev, SPAD_Q_SZ, 1, hw->queue_size);
 104         if (ret < 0)
 105                 return ret;
 106         ret = (*hw->ntb_ops->spad_write)(dev, SPAD_NUM_QPS, 1,
 107                                          hw->queue_pairs);
 108         if (ret < 0)
 109                 return ret;
 110         ret = (*hw->ntb_ops->spad_write)(dev, SPAD_USED_MWS, 1,
 111                                          hw->used_mw_num);
 112         if (ret < 0)
 113                 return ret;
 114         for (i = 0; i < hw->used_mw_num; i++) {
 115                 val = (uint64_t)(size_t)(hw->mz[i]->addr) >> 32;
 116                 ret = (*hw->ntb_ops->spad_write)(dev, SPAD_MW0_BA_H + 2 * i,
 117                                                  1, val);
 118                 if (ret < 0)
 119                         return ret;
 120                 val = (uint64_t)(size_t)(hw->mz[i]->addr);
 121                 ret = (*hw->ntb_ops->spad_write)(dev, SPAD_MW0_BA_L + 2 * i,
 122                                                  1, val);
 123                 if (ret < 0)
 124                         return ret;
 125         }
 126
 127         for (i = 0; i < hw->used_mw_num; i++) {
 128                 ret = (*hw->ntb_ops->mw_set_trans)(dev, i, hw->mz[i]->iova,
 129                                                    hw->mz[i]->len);
 130                 if (ret < 0)
 131                         return ret;
 132         }
 133
 134         /* Ring doorbell 0 to tell peer the device is ready. */
 135         ret = (*hw->ntb_ops->peer_db_set)(dev, 0);
 136         if (ret < 0)
 137                 return ret;
 138
 139         return 0;
 140 }
 141
 142 static void
 143 ntb_dev_intr_handler(void *param)
 144 {
 145         struct rte_rawdev *dev = (struct rte_rawdev *)param;
 146         struct ntb_hw *hw = dev->dev_private;
 147         uint32_t val_h, val_l;
 148         uint64_t peer_mw_size;
 149         uint64_t db_bits = 0;
 150         uint8_t peer_mw_cnt;
 151         int i = 0;
 152
 153         if (hw->ntb_ops->db_read == NULL ||
 154             hw->ntb_ops->db_clear == NULL ||
 155             hw->ntb_ops->peer_db_set == NULL) {
 156                 NTB_LOG(ERR, "Doorbell is not supported.");
 157                 return;
 158         }
 159
 160         db_bits = (*hw->ntb_ops->db_read)(dev);
 161         if (!db_bits)
 162                 NTB_LOG(ERR, "No doorbells");
 163
 164         /* Doorbell 0 is for peer device ready. */
 165         if (db_bits & 1) {
 166                 NTB_LOG(INFO, "DB0: Peer device is up.");
 167                 /* Clear received doorbell. */
 168                 (*hw->ntb_ops->db_clear)(dev, 1);
 169
 170                 /**
 171                  * Peer dev is already up. All mw settings are already done.
 172                  * Skip them.
 173                  */
 174                 if (hw->peer_dev_up)
 175                         return;
 176
 177                 if (hw->ntb_ops->spad_read == NULL) {
 178                         NTB_LOG(ERR, "Scratchpad read is not supported.");
 179                         return;
 180                 }
 181
 182                 /* Check if mw setting on the peer is the same as local. */
 183                 peer_mw_cnt = (*hw->ntb_ops->spad_read)(dev, SPAD_NUM_MWS, 0);
 184                 if (peer_mw_cnt != hw->mw_cnt) {
 185                         NTB_LOG(ERR, "Both mw cnt must be the same.");
 186                         return;
 187                 }
 188
 189                 for (i = 0; i < hw->mw_cnt; i++) {
 190                         val_h = (*hw->ntb_ops->spad_read)
 191                                 (dev, SPAD_MW0_SZ_H + 2 * i, 0);
 192                         val_l = (*hw->ntb_ops->spad_read)
 193                                 (dev, SPAD_MW0_SZ_L + 2 * i, 0);
 194                         peer_mw_size = ((uint64_t)val_h << 32) | val_l;
 195                         NTB_LOG(DEBUG, "Peer %u mw size: 0x%"PRIx64"", i,
 196                                         peer_mw_size);
 197                         if (peer_mw_size != hw->mw_size[i]) {
 198                                 NTB_LOG(ERR, "Mw config must be the same.");
 199                                 return;
 200                         }
 201                 }
 202
 203                 hw->peer_dev_up = 1;
 204
 205                 /**
 206                  * Handshake with peer. Spad_write & mw_set_trans only works
 207                  * when both devices are up. So write spad again when db is
 208                  * received. And set db again for the later device who may miss
 209                  * the 1st db.
 210                  */
 211                 if (ntb_handshake_work(dev) < 0) {
 212                         NTB_LOG(ERR, "Handshake work failed.");
 213                         return;
 214                 }
 215
 216                 /* To get the link info. */
 217                 if (hw->ntb_ops->get_link_status == NULL) {
 218                         NTB_LOG(ERR, "Not supported to get link status.");
 219                         return;
 220                 }
 221                 (*hw->ntb_ops->get_link_status)(dev);
 222                 NTB_LOG(INFO, "Link is up. Link speed: %u. Link width: %u",
 223                         hw->link_speed, hw->link_width);
 224                 return;
 225         }
 226
 227         if (db_bits & (1 << 1)) {
 228                 NTB_LOG(INFO, "DB1: Peer device is down.");
 229                 /* Clear received doorbell. */
 230                 (*hw->ntb_ops->db_clear)(dev, 2);
 231
 232                 /* Peer device will be down, So clean local side too. */
 233                 ntb_link_cleanup(dev);
 234
 235                 hw->peer_dev_up = 0;
 236                 /* Response peer's dev_stop request. */
 237                 (*hw->ntb_ops->peer_db_set)(dev, 2);
 238                 return;
 239         }
 240
 241         if (db_bits & (1 << 2)) {
 242                 NTB_LOG(INFO, "DB2: Peer device agrees dev to be down.");
 243                 /* Clear received doorbell. */
 244                 (*hw->ntb_ops->db_clear)(dev, (1 << 2));
 245                 hw->peer_dev_up = 0;
 246                 return;
 247         }
 248
 249         /* Clear other received doorbells. */
 250         (*hw->ntb_ops->db_clear)(dev, db_bits);
 251 }
 252
 253 static int
 254 ntb_queue_conf_get(struct rte_rawdev *dev,
 255                    uint16_t queue_id,
 256                    rte_rawdev_obj_t queue_conf,
 257                    size_t conf_size)
 258 {
 259         struct ntb_queue_conf *q_conf = queue_conf;
 260         struct ntb_hw *hw = dev->dev_private;
 261
 262         if (conf_size != sizeof(*q_conf))
 263                 return -EINVAL;
 264
 265         q_conf->tx_free_thresh = hw->tx_queues[queue_id]->tx_free_thresh;
 266         q_conf->nb_desc = hw->rx_queues[queue_id]->nb_rx_desc;
 267         q_conf->rx_mp = hw->rx_queues[queue_id]->mpool;
 268
 269         return 0;
 270 }
 271
 272 static void
 273 ntb_rxq_release_mbufs(struct ntb_rx_queue *q)
 274 {
 275         int i;
 276
 277         if (!q || !q->sw_ring) {
 278                 NTB_LOG(ERR, "Pointer to rxq or sw_ring is NULL");
 279                 return;
 280         }
 281
 282         for (i = 0; i < q->nb_rx_desc; i++) {
 283                 if (q->sw_ring[i].mbuf) {
 284                         rte_pktmbuf_free_seg(q->sw_ring[i].mbuf);
 285                         q->sw_ring[i].mbuf = NULL;
 286                 }
 287         }
 288 }
 289
 290 static void
 291 ntb_rxq_release(struct ntb_rx_queue *rxq)
 292 {
 293         if (!rxq) {
 294                 NTB_LOG(ERR, "Pointer to rxq is NULL");
 295                 return;
 296         }
 297
 298         ntb_rxq_release_mbufs(rxq);
 299
 300         rte_free(rxq->sw_ring);
 301         rte_free(rxq);
 302 }
 303
 304 static int
 305 ntb_rxq_setup(struct rte_rawdev *dev,
 306               uint16_t qp_id,
 307               rte_rawdev_obj_t queue_conf,
 308               size_t conf_size)
 309 {
 310         struct ntb_queue_conf *rxq_conf = queue_conf;
 311         struct ntb_hw *hw = dev->dev_private;
 312         struct ntb_rx_queue *rxq;
 313
 314         if (conf_size != sizeof(*rxq_conf))
 315                 return -EINVAL;
 316
 317         /* Allocate the rx queue data structure */
 318         rxq = rte_zmalloc_socket("ntb rx queue",
 319                                  sizeof(struct ntb_rx_queue),
 320                                  RTE_CACHE_LINE_SIZE,
 321                                  dev->socket_id);
 322         if (!rxq) {
 323                 NTB_LOG(ERR, "Failed to allocate memory for "
 324                             "rx queue data structure.");
 325                 return -ENOMEM;
 326         }
 327
 328         if (rxq_conf->rx_mp == NULL) {
 329                 NTB_LOG(ERR, "Invalid null mempool pointer.");
 330                 return -EINVAL;
 331         }
 332         rxq->nb_rx_desc = rxq_conf->nb_desc;
 333         rxq->mpool = rxq_conf->rx_mp;
 334         rxq->port_id = dev->dev_id;
 335         rxq->queue_id = qp_id;
 336         rxq->hw = hw;
 337
 338         /* Allocate the software ring. */
 339         rxq->sw_ring =
 340                 rte_zmalloc_socket("ntb rx sw ring",
 341                                    sizeof(struct ntb_rx_entry) *
 342                                    rxq->nb_rx_desc,
 343                                    RTE_CACHE_LINE_SIZE,
 344                                    dev->socket_id);
 345         if (!rxq->sw_ring) {
 346                 ntb_rxq_release(rxq);
 347                 rxq = NULL;
 348                 NTB_LOG(ERR, "Failed to allocate memory for SW ring");
 349                 return -ENOMEM;
 350         }
 351
 352         hw->rx_queues[qp_id] = rxq;
 353
 354         return 0;
 355 }
 356
 357 static void
 358 ntb_txq_release_mbufs(struct ntb_tx_queue *q)
 359 {
 360         int i;
 361
 362         if (!q || !q->sw_ring) {
 363                 NTB_LOG(ERR, "Pointer to txq or sw_ring is NULL");
 364                 return;
 365         }
 366
 367         for (i = 0; i < q->nb_tx_desc; i++) {
 368                 if (q->sw_ring[i].mbuf) {
 369                         rte_pktmbuf_free_seg(q->sw_ring[i].mbuf);
 370                         q->sw_ring[i].mbuf = NULL;
 371                 }
 372         }
 373 }
 374
 375 static void
 376 ntb_txq_release(struct ntb_tx_queue *txq)
 377 {
 378         if (!txq) {
 379                 NTB_LOG(ERR, "Pointer to txq is NULL");
 380                 return;
 381         }
 382
 383         ntb_txq_release_mbufs(txq);
 384
 385         rte_free(txq->sw_ring);
 386         rte_free(txq);
 387 }
 388
 389 static int
 390 ntb_txq_setup(struct rte_rawdev *dev,
 391               uint16_t qp_id,
 392               rte_rawdev_obj_t queue_conf,
 393               size_t conf_size)
 394 {
 395         struct ntb_queue_conf *txq_conf = queue_conf;
 396         struct ntb_hw *hw = dev->dev_private;
 397         struct ntb_tx_queue *txq;
 398         uint16_t i, prev;
 399
 400         if (conf_size != sizeof(*txq_conf))
 401                 return -EINVAL;
 402
 403         /* Allocate the TX queue data structure. */
 404         txq = rte_zmalloc_socket("ntb tx queue",
 405                                   sizeof(struct ntb_tx_queue),
 406                                   RTE_CACHE_LINE_SIZE,
 407                                   dev->socket_id);
 408         if (!txq) {
 409                 NTB_LOG(ERR, "Failed to allocate memory for "
 410                             "tx queue structure");
 411                 return -ENOMEM;
 412         }
 413
 414         txq->nb_tx_desc = txq_conf->nb_desc;
 415         txq->port_id = dev->dev_id;
 416         txq->queue_id = qp_id;
 417         txq->hw = hw;
 418
 419         /* Allocate software ring */
 420         txq->sw_ring =
 421                 rte_zmalloc_socket("ntb tx sw ring",
 422                                    sizeof(struct ntb_tx_entry) *
 423                                    txq->nb_tx_desc,
 424                                    RTE_CACHE_LINE_SIZE,
 425                                    dev->socket_id);
 426         if (!txq->sw_ring) {
 427                 ntb_txq_release(txq);
 428                 txq = NULL;
 429                 NTB_LOG(ERR, "Failed to allocate memory for SW TX ring");
 430                 return -ENOMEM;
 431         }
 432
 433         prev = txq->nb_tx_desc - 1;
 434         for (i = 0; i < txq->nb_tx_desc; i++) {
 435                 txq->sw_ring[i].mbuf = NULL;
 436                 txq->sw_ring[i].last_id = i;
 437                 txq->sw_ring[prev].next_id = i;
 438                 prev = i;
 439         }
 440
 441         txq->tx_free_thresh = txq_conf->tx_free_thresh ?
 442                               txq_conf->tx_free_thresh :
 443                               NTB_DFLT_TX_FREE_THRESH;
 444         if (txq->tx_free_thresh >= txq->nb_tx_desc - 3) {
 445                 NTB_LOG(ERR, "tx_free_thresh must be less than nb_desc - 3. "
 446                         "(tx_free_thresh=%u qp_id=%u)", txq->tx_free_thresh,
 447                         qp_id);
 448                 return -EINVAL;
 449         }
 450
 451         hw->tx_queues[qp_id] = txq;
 452
 453         return 0;
 454 }
 455
 456
 457 static int
 458 ntb_queue_setup(struct rte_rawdev *dev,
 459                 uint16_t queue_id,
 460                 rte_rawdev_obj_t queue_conf,
 461                 size_t conf_size)
 462 {
 463         struct ntb_hw *hw = dev->dev_private;
 464         int ret;
 465
 466         if (queue_id >= hw->queue_pairs)
 467                 return -EINVAL;
 468
 469         ret = ntb_txq_setup(dev, queue_id, queue_conf, conf_size);
 470         if (ret < 0)
 471                 return ret;
 472
 473         ret = ntb_rxq_setup(dev, queue_id, queue_conf, conf_size);
 474
 475         return ret;
 476 }
 477
 478 static int
 479 ntb_queue_release(struct rte_rawdev *dev, uint16_t queue_id)
 480 {
 481         struct ntb_hw *hw = dev->dev_private;
 482
 483         if (queue_id >= hw->queue_pairs)
 484                 return -EINVAL;
 485
 486         ntb_txq_release(hw->tx_queues[queue_id]);
 487         hw->tx_queues[queue_id] = NULL;
 488         ntb_rxq_release(hw->rx_queues[queue_id]);
 489         hw->rx_queues[queue_id] = NULL;
 490
 491         return 0;
 492 }
 493
 494 static uint16_t
 495 ntb_queue_count(struct rte_rawdev *dev)
 496 {
 497         struct ntb_hw *hw = dev->dev_private;
 498         return hw->queue_pairs;
 499 }
 500
 501 static int
 502 ntb_queue_init(struct rte_rawdev *dev, uint16_t qp_id)
 503 {
 504         struct ntb_hw *hw = dev->dev_private;
 505         struct ntb_rx_queue *rxq = hw->rx_queues[qp_id];
 506         struct ntb_tx_queue *txq = hw->tx_queues[qp_id];
 507         volatile struct ntb_header *local_hdr;
 508         struct ntb_header *remote_hdr;
 509         uint16_t q_size = hw->queue_size;
 510         uint32_t hdr_offset;
 511         void *bar_addr;
 512         uint16_t i;
 513
 514         if (hw->ntb_ops->get_peer_mw_addr == NULL) {
 515                 NTB_LOG(ERR, "Getting peer mw addr is not supported.");
 516                 return -EINVAL;
 517         }
 518
 519         /* Put queue info into the start of shared memory. */
 520         hdr_offset = hw->hdr_size_per_queue * qp_id;
 521         local_hdr = (volatile struct ntb_header *)
 522                     ((size_t)hw->mz[0]->addr + hdr_offset);
 523         bar_addr = (*hw->ntb_ops->get_peer_mw_addr)(dev, 0);
 524         if (bar_addr == NULL)
 525                 return -EINVAL;
 526         remote_hdr = (struct ntb_header *)
 527                      ((size_t)bar_addr + hdr_offset);
 528
 529         /* rxq init. */
 530         rxq->rx_desc_ring = (struct ntb_desc *)
 531                             (&remote_hdr->desc_ring);
 532         rxq->rx_used_ring = (volatile struct ntb_used *)
 533                             (&local_hdr->desc_ring[q_size]);
 534         rxq->avail_cnt = &remote_hdr->avail_cnt;
 535         rxq->used_cnt = &local_hdr->used_cnt;
 536
 537         for (i = 0; i < rxq->nb_rx_desc - 1; i++) {
 538                 struct rte_mbuf *mbuf = rte_mbuf_raw_alloc(rxq->mpool);
 539                 if (unlikely(!mbuf)) {
 540                         NTB_LOG(ERR, "Failed to allocate mbuf for RX");
 541                         return -ENOMEM;
 542                 }
 543                 mbuf->port = dev->dev_id;
 544
 545                 rxq->sw_ring[i].mbuf = mbuf;
 546
 547                 rxq->rx_desc_ring[i].addr = rte_pktmbuf_mtod(mbuf, size_t);
 548                 rxq->rx_desc_ring[i].len = mbuf->buf_len - RTE_PKTMBUF_HEADROOM;
 549         }
 550         rte_wmb();
 551         *rxq->avail_cnt = rxq->nb_rx_desc - 1;
 552         rxq->last_avail = rxq->nb_rx_desc - 1;
 553         rxq->last_used = 0;
 554
 555         /* txq init */
 556         txq->tx_desc_ring = (volatile struct ntb_desc *)
 557                             (&local_hdr->desc_ring);
 558         txq->tx_used_ring = (struct ntb_used *)
 559                             (&remote_hdr->desc_ring[q_size]);
 560         txq->avail_cnt = &local_hdr->avail_cnt;
 561         txq->used_cnt = &remote_hdr->used_cnt;
 562
 563         rte_wmb();
 564         *txq->used_cnt = 0;
 565         txq->last_used = 0;
 566         txq->last_avail = 0;
 567         txq->nb_tx_free = txq->nb_tx_desc - 1;
 568
 569         /* Set per queue stats. */
 570         for (i = 0; i < NTB_XSTATS_NUM; i++) {
 571                 hw->ntb_xstats[i + NTB_XSTATS_NUM * (qp_id + 1)] = 0;
 572                 hw->ntb_xstats_off[i + NTB_XSTATS_NUM * (qp_id + 1)] = 0;
 573         }
 574
 575         return 0;
 576 }
 577
 578 static inline void
 579 ntb_enqueue_cleanup(struct ntb_tx_queue *txq)
 580 {
 581         struct ntb_tx_entry *sw_ring = txq->sw_ring;
 582         uint16_t tx_free = txq->last_avail;
 583         uint16_t nb_to_clean, i;
 584
 585         /* avail_cnt + 1 represents where to rx next in the peer. */
 586         nb_to_clean = (*txq->avail_cnt - txq->last_avail + 1 +
 587                         txq->nb_tx_desc) & (txq->nb_tx_desc - 1);
 588         nb_to_clean = RTE_MIN(nb_to_clean, txq->tx_free_thresh);
 589         for (i = 0; i < nb_to_clean; i++) {
 590                 if (sw_ring[tx_free].mbuf)
 591                         rte_pktmbuf_free_seg(sw_ring[tx_free].mbuf);
 592                 tx_free = (tx_free + 1) & (txq->nb_tx_desc - 1);
 593         }
 594
 595         txq->nb_tx_free += nb_to_clean;
 596         txq->last_avail = tx_free;
 597 }
 598
 599 static int
 600 ntb_enqueue_bufs(struct rte_rawdev *dev,
 601                  struct rte_rawdev_buf **buffers,
 602                  unsigned int count,
 603                  rte_rawdev_obj_t context)
 604 {
 605         struct ntb_hw *hw = dev->dev_private;
 606         struct ntb_tx_queue *txq = hw->tx_queues[(size_t)context];
 607         struct ntb_tx_entry *sw_ring = txq->sw_ring;
 608         struct rte_mbuf *txm;
 609         struct ntb_used tx_used[NTB_MAX_DESC_SIZE];
 610         volatile struct ntb_desc *tx_item;
 611         uint16_t tx_last, nb_segs, off, last_used, avail_cnt;
 612         uint16_t nb_mbufs = 0;
 613         uint16_t nb_tx = 0;
 614         uint64_t bytes = 0;
 615         void *buf_addr;
 616         int i;
 617
 618         if (unlikely(hw->ntb_ops->ioremap == NULL)) {
 619                 NTB_LOG(ERR, "Ioremap not supported.");
 620                 return nb_tx;
 621         }
 622
 623         if (unlikely(dev->started == 0 || hw->peer_dev_up == 0)) {
 624                 NTB_LOG(DEBUG, "Link is not up.");
 625                 return nb_tx;
 626         }
 627
 628         if (txq->nb_tx_free < txq->tx_free_thresh)
 629                 ntb_enqueue_cleanup(txq);
 630
 631         off = NTB_XSTATS_NUM * ((size_t)context + 1);
 632         last_used = txq->last_used;
 633         avail_cnt = *txq->avail_cnt;/* Where to alloc next. */
 634         for (nb_tx = 0; nb_tx < count; nb_tx++) {
 635                 txm = (struct rte_mbuf *)(buffers[nb_tx]->buf_addr);
 636                 if (txm == NULL || txq->nb_tx_free < txm->nb_segs)
 637                         break;
 638
 639                 tx_last = (txq->last_used + txm->nb_segs - 1) &
 640                           (txq->nb_tx_desc - 1);
 641                 nb_segs = txm->nb_segs;
 642                 for (i = 0; i < nb_segs; i++) {
 643                         /* Not enough ring space for tx. */
 644                         if (txq->last_used == avail_cnt)
 645                                 goto end_of_tx;
 646                         sw_ring[txq->last_used].mbuf = txm;
 647                         tx_item = txq->tx_desc_ring + txq->last_used;
 648
 649                         if (!tx_item->len) {
 650                                 (hw->ntb_xstats[NTB_TX_ERRS_ID + off])++;
 651                                 goto end_of_tx;
 652                         }
 653                         if (txm->data_len > tx_item->len) {
 654                                 NTB_LOG(ERR, "Data length exceeds buf length."
 655                                         " Only %u data would be transmitted.",
 656                                         tx_item->len);
 657                                 txm->data_len = tx_item->len;
 658                         }
 659
 660                         /* translate remote virtual addr to bar virtual addr */
 661                         buf_addr = (*hw->ntb_ops->ioremap)(dev, tx_item->addr);
 662                         if (buf_addr == NULL) {
 663                                 (hw->ntb_xstats[NTB_TX_ERRS_ID + off])++;
 664                                 NTB_LOG(ERR, "Null remap addr.");
 665                                 goto end_of_tx;
 666                         }
 667                         rte_memcpy(buf_addr, rte_pktmbuf_mtod(txm, void *),
 668                                    txm->data_len);
 669
 670                         tx_used[nb_mbufs].len = txm->data_len;
 671                         tx_used[nb_mbufs++].flags = (txq->last_used ==
 672                                                     tx_last) ?
 673                                                     NTB_FLAG_EOP : 0;
 674
 675                         /* update stats */
 676                         bytes += txm->data_len;
 677
 678                         txm = txm->next;
 679
 680                         sw_ring[txq->last_used].next_id = (txq->last_used + 1) &
 681                                                   (txq->nb_tx_desc - 1);
 682                         sw_ring[txq->last_used].last_id = tx_last;
 683                         txq->last_used = (txq->last_used + 1) &
 684                                          (txq->nb_tx_desc - 1);
 685                 }
 686                 txq->nb_tx_free -= nb_segs;
 687         }
 688
 689 end_of_tx:
 690         if (nb_tx) {
 691                 uint16_t nb1, nb2;
 692                 if (nb_mbufs > txq->nb_tx_desc - last_used) {
 693                         nb1 = txq->nb_tx_desc - last_used;
 694                         nb2 = nb_mbufs - txq->nb_tx_desc + last_used;
 695                 } else {
 696                         nb1 = nb_mbufs;
 697                         nb2 = 0;
 698                 }
 699                 rte_memcpy(txq->tx_used_ring + last_used, tx_used,
 700                            sizeof(struct ntb_used) * nb1);
 701                 rte_memcpy(txq->tx_used_ring, tx_used + nb1,
 702                            sizeof(struct ntb_used) * nb2);
 703                 rte_wmb();
 704                 *txq->used_cnt = txq->last_used;
 705
 706                 /* update queue stats */
 707                 hw->ntb_xstats[NTB_TX_BYTES_ID + off] += bytes;
 708                 hw->ntb_xstats[NTB_TX_PKTS_ID + off] += nb_tx;
 709         }
 710
 711         return nb_tx;
 712 }
 713
 714 static int
 715 ntb_dequeue_bufs(struct rte_rawdev *dev,
 716                  struct rte_rawdev_buf **buffers,
 717                  unsigned int count,
 718                  rte_rawdev_obj_t context)
 719 {
 720         struct ntb_hw *hw = dev->dev_private;
 721         struct ntb_rx_queue *rxq = hw->rx_queues[(size_t)context];
 722         struct ntb_rx_entry *sw_ring = rxq->sw_ring;
 723         struct ntb_desc rx_desc[NTB_MAX_DESC_SIZE];
 724         struct rte_mbuf *first, *rxm_t;
 725         struct rte_mbuf *prev = NULL;
 726         volatile struct ntb_used *rx_item;
 727         uint16_t nb_mbufs = 0;
 728         uint16_t nb_rx = 0;
 729         uint64_t bytes = 0;
 730         uint16_t off, last_avail, used_cnt, used_nb;
 731         int i;
 732
 733         if (unlikely(dev->started == 0 || hw->peer_dev_up == 0)) {
 734                 NTB_LOG(DEBUG, "Link is not up");
 735                 return nb_rx;
 736         }
 737
 738         used_cnt = *rxq->used_cnt;
 739
 740         if (rxq->last_used == used_cnt)
 741                 return nb_rx;
 742
 743         last_avail = rxq->last_avail;
 744         used_nb = (used_cnt - rxq->last_used) & (rxq->nb_rx_desc - 1);
 745         count = RTE_MIN(count, used_nb);
 746         for (nb_rx = 0; nb_rx < count; nb_rx++) {
 747                 i = 0;
 748                 while (true) {
 749                         rx_item = rxq->rx_used_ring + rxq->last_used;
 750                         rxm_t = sw_ring[rxq->last_used].mbuf;
 751                         rxm_t->data_len = rx_item->len;
 752                         rxm_t->data_off = RTE_PKTMBUF_HEADROOM;
 753                         rxm_t->port = rxq->port_id;
 754
 755                         if (!i) {
 756                                 rxm_t->nb_segs = 1;
 757                                 first = rxm_t;
 758                                 first->pkt_len = 0;
 759                                 buffers[nb_rx]->buf_addr = rxm_t;
 760                         } else {
 761                                 prev->next = rxm_t;
 762                                 first->nb_segs++;
 763                         }
 764
 765                         prev = rxm_t;
 766                         first->pkt_len += prev->data_len;
 767                         rxq->last_used = (rxq->last_used + 1) &
 768                                          (rxq->nb_rx_desc - 1);
 769
 770                         /* alloc new mbuf */
 771                         rxm_t = rte_mbuf_raw_alloc(rxq->mpool);
 772                         if (unlikely(rxm_t == NULL)) {
 773                                 NTB_LOG(ERR, "recv alloc mbuf failed.");
 774                                 goto end_of_rx;
 775                         }
 776                         rxm_t->port = rxq->port_id;
 777                         sw_ring[rxq->last_avail].mbuf = rxm_t;
 778                         i++;
 779
 780                         /* fill new desc */
 781                         rx_desc[nb_mbufs].addr =
 782                                         rte_pktmbuf_mtod(rxm_t, size_t);
 783                         rx_desc[nb_mbufs++].len = rxm_t->buf_len -
 784                                                   RTE_PKTMBUF_HEADROOM;
 785                         rxq->last_avail = (rxq->last_avail + 1) &
 786                                           (rxq->nb_rx_desc - 1);
 787
 788                         if (rx_item->flags & NTB_FLAG_EOP)
 789                                 break;
 790                 }
 791                 /* update stats */
 792                 bytes += first->pkt_len;
 793         }
 794
 795 end_of_rx:
 796         if (nb_rx) {
 797                 uint16_t nb1, nb2;
 798                 if (nb_mbufs > rxq->nb_rx_desc - last_avail) {
 799                         nb1 = rxq->nb_rx_desc - last_avail;
 800                         nb2 = nb_mbufs - rxq->nb_rx_desc + last_avail;
 801                 } else {
 802                         nb1 = nb_mbufs;
 803                         nb2 = 0;
 804                 }
 805                 rte_memcpy(rxq->rx_desc_ring + last_avail, rx_desc,
 806                            sizeof(struct ntb_desc) * nb1);
 807                 rte_memcpy(rxq->rx_desc_ring, rx_desc + nb1,
 808                            sizeof(struct ntb_desc) * nb2);
 809                 rte_wmb();
 810                 *rxq->avail_cnt = rxq->last_avail;
 811
 812                 /* update queue stats */
 813                 off = NTB_XSTATS_NUM * ((size_t)context + 1);
 814                 hw->ntb_xstats[NTB_RX_BYTES_ID + off] += bytes;
 815                 hw->ntb_xstats[NTB_RX_PKTS_ID + off] += nb_rx;
 816                 hw->ntb_xstats[NTB_RX_MISS_ID + off] += (count - nb_rx);
 817         }
 818
 819         return nb_rx;
 820 }
 821
 822 static int
 823 ntb_dev_info_get(struct rte_rawdev *dev, rte_rawdev_obj_t dev_info,
 824                 size_t dev_info_size)
 825 {
 826         struct ntb_hw *hw = dev->dev_private;
 827         struct ntb_dev_info *info = dev_info;
 828
 829         if (dev_info_size != sizeof(*info)) {
 830                 NTB_LOG(ERR, "Invalid size parameter to %s", __func__);
 831                 return -EINVAL;
 832         }
 833
 834         info->mw_cnt = hw->mw_cnt;
 835         info->mw_size = hw->mw_size;
 836
 837         /**
 838          * Intel hardware requires that mapped memory base address should be
 839          * aligned with EMBARSZ and needs continuous memzone.
 840          */
 841         info->mw_size_align = (uint8_t)(hw->pci_dev->id.vendor_id ==
 842                                         NTB_INTEL_VENDOR_ID);
 843
 844         if (!hw->queue_size || !hw->queue_pairs) {
 845                 NTB_LOG(ERR, "No queue size and queue num assigned.");
 846                 return -EAGAIN;
 847         }
 848
 849         hw->hdr_size_per_queue = RTE_ALIGN(sizeof(struct ntb_header) +
 850                                 hw->queue_size * sizeof(struct ntb_desc) +
 851                                 hw->queue_size * sizeof(struct ntb_used),
 852                                 RTE_CACHE_LINE_SIZE);
 853         info->ntb_hdr_size = hw->hdr_size_per_queue * hw->queue_pairs;
 854
 855         return 0;
 856 }
 857
 858 static int
 859 ntb_dev_configure(const struct rte_rawdev *dev, rte_rawdev_obj_t config,
 860                 size_t config_size)
 861 {
 862         struct ntb_dev_config *conf = config;
 863         struct ntb_hw *hw = dev->dev_private;
 864         uint32_t xstats_num;
 865         int ret;
 866
 867         if (conf == NULL || config_size != sizeof(*conf))
 868                 return -EINVAL;
 869
 870         hw->queue_pairs = conf->num_queues;
 871         hw->queue_size = conf->queue_size;
 872         hw->used_mw_num = conf->mz_num;
 873         hw->mz = conf->mz_list;
 874         hw->rx_queues = rte_zmalloc("ntb_rx_queues",
 875                         sizeof(struct ntb_rx_queue *) * hw->queue_pairs, 0);
 876         hw->tx_queues = rte_zmalloc("ntb_tx_queues",
 877                         sizeof(struct ntb_tx_queue *) * hw->queue_pairs, 0);
 878         /* First total stats, then per queue stats. */
 879         xstats_num = (hw->queue_pairs + 1) * NTB_XSTATS_NUM;
 880         hw->ntb_xstats = rte_zmalloc("ntb_xstats", xstats_num *
 881                                      sizeof(uint64_t), 0);
 882         hw->ntb_xstats_off = rte_zmalloc("ntb_xstats_off", xstats_num *
 883                                          sizeof(uint64_t), 0);
 884
 885         /* Start handshake with the peer. */
 886         ret = ntb_handshake_work(dev);
 887         if (ret < 0) {
 888                 rte_free(hw->rx_queues);
 889                 rte_free(hw->tx_queues);
 890                 hw->rx_queues = NULL;
 891                 hw->tx_queues = NULL;
 892                 return ret;
 893         }
 894
 895         return 0;
 896 }
 897
 898 static int
 899 ntb_dev_start(struct rte_rawdev *dev)
 900 {
 901         struct ntb_hw *hw = dev->dev_private;
 902         uint32_t peer_base_l, peer_val;
 903         uint64_t peer_base_h;
 904         uint32_t i;
 905         int ret;
 906
 907         if (!hw->link_status || !hw->peer_dev_up)
 908                 return -EINVAL;
 909
 910         /* Set total stats. */
 911         for (i = 0; i < NTB_XSTATS_NUM; i++) {
 912                 hw->ntb_xstats[i] = 0;
 913                 hw->ntb_xstats_off[i] = 0;
 914         }
 915
 916         for (i = 0; i < hw->queue_pairs; i++) {
 917                 ret = ntb_queue_init(dev, i);
 918                 if (ret) {
 919                         NTB_LOG(ERR, "Failed to init queue.");
 920                         goto err_q_init;
 921                 }
 922         }
 923
 924         hw->peer_mw_base = rte_zmalloc("ntb_peer_mw_base", hw->mw_cnt *
 925                                         sizeof(uint64_t), 0);
 926
 927         if (hw->ntb_ops->spad_read == NULL) {
 928                 ret = -ENOTSUP;
 929                 goto err_up;
 930         }
 931
 932         peer_val = (*hw->ntb_ops->spad_read)(dev, SPAD_Q_SZ, 0);
 933         if (peer_val != hw->queue_size) {
 934                 NTB_LOG(ERR, "Inconsistent queue size! (local: %u peer: %u)",
 935                         hw->queue_size, peer_val);
 936                 ret = -EINVAL;
 937                 goto err_up;
 938         }
 939
 940         peer_val = (*hw->ntb_ops->spad_read)(dev, SPAD_NUM_QPS, 0);
 941         if (peer_val != hw->queue_pairs) {
 942                 NTB_LOG(ERR, "Inconsistent number of queues! (local: %u peer:"
 943                         " %u)", hw->queue_pairs, peer_val);
 944                 ret = -EINVAL;
 945                 goto err_up;
 946         }
 947
 948         hw->peer_used_mws = (*hw->ntb_ops->spad_read)(dev, SPAD_USED_MWS, 0);
 949
 950         for (i = 0; i < hw->peer_used_mws; i++) {
 951                 peer_base_h = (*hw->ntb_ops->spad_read)(dev,
 952                                 SPAD_MW0_BA_H + 2 * i, 0);
 953                 peer_base_l = (*hw->ntb_ops->spad_read)(dev,
 954                                 SPAD_MW0_BA_L + 2 * i, 0);
 955                 hw->peer_mw_base[i] = (peer_base_h << 32) + peer_base_l;
 956         }
 957
 958         dev->started = 1;
 959
 960         return 0;
 961
 962 err_up:
 963         rte_free(hw->peer_mw_base);
 964 err_q_init:
 965         for (i = 0; i < hw->queue_pairs; i++) {
 966                 ntb_rxq_release_mbufs(hw->rx_queues[i]);
 967                 ntb_txq_release_mbufs(hw->tx_queues[i]);
 968         }
 969
 970         return ret;
 971 }
 972
 973 static void
 974 ntb_dev_stop(struct rte_rawdev *dev)
 975 {
 976         struct ntb_hw *hw = dev->dev_private;
 977         uint32_t time_out;
 978         int status, i;
 979
 980         if (!hw->peer_dev_up)
 981                 goto clean;
 982
 983         ntb_link_cleanup(dev);
 984
 985         /* Notify the peer that device will be down. */
 986         if (hw->ntb_ops->peer_db_set == NULL) {
 987                 NTB_LOG(ERR, "Peer doorbell setting is not supported.");
 988                 return;
 989         }
 990         status = (*hw->ntb_ops->peer_db_set)(dev, 1);
 991         if (status) {
 992                 NTB_LOG(ERR, "Failed to tell peer device is down.");
 993                 return;
 994         }
 995
 996         /*
 997          * Set time out as 1s in case that the peer is stopped accidently
 998          * without any notification.
 999          */
1000         time_out = 1000000;
1001
1002         /* Wait for cleanup work down before db mask clear. */
1003         while (hw->peer_dev_up && time_out) {
1004                 time_out -= 10;
1005                 rte_delay_us(10);
1006         }
1007
1008 clean:
1009         /* Clear doorbells mask. */
1010         if (hw->ntb_ops->db_set_mask == NULL) {
1011                 NTB_LOG(ERR, "Doorbell mask setting is not supported.");
1012                 return;
1013         }
1014         status = (*hw->ntb_ops->db_set_mask)(dev,
1015                                 (((uint64_t)1 << hw->db_cnt) - 1));
1016         if (status)
1017                 NTB_LOG(ERR, "Failed to clear doorbells.");
1018
1019         for (i = 0; i < hw->queue_pairs; i++) {
1020                 ntb_rxq_release_mbufs(hw->rx_queues[i]);
1021                 ntb_txq_release_mbufs(hw->tx_queues[i]);
1022         }
1023
1024         dev->started = 0;
1025 }
1026
1027 static int
1028 ntb_dev_close(struct rte_rawdev *dev)
1029 {
1030         struct ntb_hw *hw = dev->dev_private;
1031         struct rte_intr_handle *intr_handle;
1032         int i;
1033
1034         if (dev->started)
1035                 ntb_dev_stop(dev);
1036
1037         /* free queues */
1038         for (i = 0; i < hw->queue_pairs; i++)
1039                 ntb_queue_release(dev, i);
1040         hw->queue_pairs = 0;
1041
1042         intr_handle = &hw->pci_dev->intr_handle;
1043         /* Clean datapath event and vec mapping */
1044         rte_intr_efd_disable(intr_handle);
1045         if (intr_handle->intr_vec) {
1046                 rte_free(intr_handle->intr_vec);
1047                 intr_handle->intr_vec = NULL;
1048         }
1049         /* Disable uio intr before callback unregister */
1050         rte_intr_disable(intr_handle);
1051
1052         /* Unregister callback func to eal lib */
1053         rte_intr_callback_unregister(intr_handle,
1054                                      ntb_dev_intr_handler, dev);
1055
1056         return 0;
1057 }
1058
1059 static int
1060 ntb_dev_reset(struct rte_rawdev *rawdev __rte_unused)
1061 {
1062         return 0;
1063 }
1064
1065 static int
1066 ntb_attr_set(struct rte_rawdev *dev, const char *attr_name,
1067              uint64_t attr_value)
1068 {
1069         struct ntb_hw *hw;
1070         int index;
1071
1072         if (dev == NULL || attr_name == NULL) {
1073                 NTB_LOG(ERR, "Invalid arguments for setting attributes");
1074                 return -EINVAL;
1075         }
1076
1077         hw = dev->dev_private;
1078
1079         if (!strncmp(attr_name, NTB_SPAD_USER, NTB_SPAD_USER_LEN)) {
1080                 if (hw->ntb_ops->spad_write == NULL)
1081                         return -ENOTSUP;
1082                 index = atoi(&attr_name[NTB_SPAD_USER_LEN]);
1083                 if (index < 0 || index >= NTB_SPAD_USER_MAX_NUM) {
1084                         NTB_LOG(ERR, "Invalid attribute (%s)", attr_name);
1085                         return -EINVAL;
1086                 }
1087                 (*hw->ntb_ops->spad_write)(dev, hw->spad_user_list[index],
1088                                            1, attr_value);
1089                 NTB_LOG(DEBUG, "Set attribute (%s) Value (%" PRIu64 ")",
1090                         attr_name, attr_value);
1091                 return 0;
1092         }
1093
1094         if (!strncmp(attr_name, NTB_QUEUE_SZ_NAME, NTB_ATTR_NAME_LEN)) {
1095                 hw->queue_size = attr_value;
1096                 NTB_LOG(DEBUG, "Set attribute (%s) Value (%" PRIu64 ")",
1097                         attr_name, attr_value);
1098                 return 0;
1099         }
1100
1101         if (!strncmp(attr_name, NTB_QUEUE_NUM_NAME, NTB_ATTR_NAME_LEN)) {
1102                 hw->queue_pairs = attr_value;
1103                 NTB_LOG(DEBUG, "Set attribute (%s) Value (%" PRIu64 ")",
1104                         attr_name, attr_value);
1105                 return 0;
1106         }
1107
1108         /* Attribute not found. */
1109         NTB_LOG(ERR, "Attribute not found.");
1110         return -EINVAL;
1111 }
1112
1113 static int
1114 ntb_attr_get(struct rte_rawdev *dev, const char *attr_name,
1115              uint64_t *attr_value)
1116 {
1117         struct ntb_hw *hw;
1118         int index;
1119
1120         if (dev == NULL || attr_name == NULL || attr_value == NULL) {
1121                 NTB_LOG(ERR, "Invalid arguments for getting attributes");
1122                 return -EINVAL;
1123         }
1124
1125         hw = dev->dev_private;
1126
1127         if (!strncmp(attr_name, NTB_TOPO_NAME, NTB_ATTR_NAME_LEN)) {
1128                 *attr_value = hw->topo;
1129                 NTB_LOG(DEBUG, "Attribute (%s) Value (%" PRIu64 ")",
1130                         attr_name, *attr_value);
1131                 return 0;
1132         }
1133
1134         if (!strncmp(attr_name, NTB_LINK_STATUS_NAME, NTB_ATTR_NAME_LEN)) {
1135                 /* hw->link_status only indicates hw link status. */
1136                 *attr_value = hw->link_status && hw->peer_dev_up;
1137                 NTB_LOG(DEBUG, "Attribute (%s) Value (%" PRIu64 ")",
1138                         attr_name, *attr_value);
1139                 return 0;
1140         }
1141
1142         if (!strncmp(attr_name, NTB_SPEED_NAME, NTB_ATTR_NAME_LEN)) {
1143                 *attr_value = hw->link_speed;
1144                 NTB_LOG(DEBUG, "Attribute (%s) Value (%" PRIu64 ")",
1145                         attr_name, *attr_value);
1146                 return 0;
1147         }
1148
1149         if (!strncmp(attr_name, NTB_WIDTH_NAME, NTB_ATTR_NAME_LEN)) {
1150                 *attr_value = hw->link_width;
1151                 NTB_LOG(DEBUG, "Attribute (%s) Value (%" PRIu64 ")",
1152                         attr_name, *attr_value);
1153                 return 0;
1154         }
1155
1156         if (!strncmp(attr_name, NTB_MW_CNT_NAME, NTB_ATTR_NAME_LEN)) {
1157                 *attr_value = hw->mw_cnt;
1158                 NTB_LOG(DEBUG, "Attribute (%s) Value (%" PRIu64 ")",
1159                         attr_name, *attr_value);
1160                 return 0;
1161         }
1162
1163         if (!strncmp(attr_name, NTB_DB_CNT_NAME, NTB_ATTR_NAME_LEN)) {
1164                 *attr_value = hw->db_cnt;
1165                 NTB_LOG(DEBUG, "Attribute (%s) Value (%" PRIu64 ")",
1166                         attr_name, *attr_value);
1167                 return 0;
1168         }
1169
1170         if (!strncmp(attr_name, NTB_SPAD_CNT_NAME, NTB_ATTR_NAME_LEN)) {
1171                 *attr_value = hw->spad_cnt;
1172                 NTB_LOG(DEBUG, "Attribute (%s) Value (%" PRIu64 ")",
1173                         attr_name, *attr_value);
1174                 return 0;
1175         }
1176
1177         if (!strncmp(attr_name, NTB_SPAD_USER, NTB_SPAD_USER_LEN)) {
1178                 if (hw->ntb_ops->spad_read == NULL)
1179                         return -ENOTSUP;
1180                 index = atoi(&attr_name[NTB_SPAD_USER_LEN]);
1181                 if (index < 0 || index >= NTB_SPAD_USER_MAX_NUM) {
1182                         NTB_LOG(ERR, "Attribute (%s) out of range", attr_name);
1183                         return -EINVAL;
1184                 }
1185                 *attr_value = (*hw->ntb_ops->spad_read)(dev,
1186                                 hw->spad_user_list[index], 0);
1187                 NTB_LOG(DEBUG, "Attribute (%s) Value (%" PRIu64 ")",
1188                         attr_name, *attr_value);
1189                 return 0;
1190         }
1191
1192         /* Attribute not found. */
1193         NTB_LOG(ERR, "Attribute not found.");
1194         return -EINVAL;
1195 }
1196
1197 static inline uint64_t
1198 ntb_stats_update(uint64_t offset, uint64_t stat)
1199 {
1200         if (stat >= offset)
1201                 return (stat - offset);
1202         else
1203                 return (uint64_t)(((uint64_t)-1) - offset + stat + 1);
1204 }
1205
1206 static int
1207 ntb_xstats_get(const struct rte_rawdev *dev,
1208                const unsigned int ids[],
1209                uint64_t values[],
1210                unsigned int n)
1211 {
1212         struct ntb_hw *hw = dev->dev_private;
1213         uint32_t i, j, off, xstats_num;
1214
1215         /* Calculate total stats of all queues. */
1216         for (i = 0; i < NTB_XSTATS_NUM; i++) {
1217                 hw->ntb_xstats[i] = 0;
1218                 for (j = 0; j < hw->queue_pairs; j++) {
1219                         off = NTB_XSTATS_NUM * (j + 1) + i;
1220                         hw->ntb_xstats[i] +=
1221                         ntb_stats_update(hw->ntb_xstats_off[off],
1222                                          hw->ntb_xstats[off]);
1223                 }
1224         }
1225
1226         xstats_num = NTB_XSTATS_NUM * (hw->queue_pairs + 1);
1227         for (i = 0; i < n && ids[i] < xstats_num; i++) {
1228                 if (ids[i] < NTB_XSTATS_NUM)
1229                         values[i] = hw->ntb_xstats[ids[i]];
1230                 else
1231                         values[i] =
1232                         ntb_stats_update(hw->ntb_xstats_off[ids[i]],
1233                                          hw->ntb_xstats[ids[i]]);
1234         }
1235
1236         return i;
1237 }
1238
1239 static int
1240 ntb_xstats_get_names(const struct rte_rawdev *dev,
1241                      struct rte_rawdev_xstats_name *xstats_names,
1242                      unsigned int size)
1243 {
1244         struct ntb_hw *hw = dev->dev_private;
1245         uint32_t xstats_num, i, j, off;
1246
1247         xstats_num = NTB_XSTATS_NUM * (hw->queue_pairs + 1);
1248         if (xstats_names == NULL || size < xstats_num)
1249                 return xstats_num;
1250
1251         /* Total stats names */
1252         memcpy(xstats_names, ntb_xstats_names, sizeof(ntb_xstats_names));
1253
1254         /* Queue stats names */
1255         for (i = 0; i < hw->queue_pairs; i++) {
1256                 for (j = 0; j < NTB_XSTATS_NUM; j++) {
1257                         off = j + (i + 1) * NTB_XSTATS_NUM;
1258                         snprintf(xstats_names[off].name,
1259                                 sizeof(xstats_names[0].name),
1260                                 "%s_q%u", ntb_xstats_names[j].name, i);
1261                 }
1262         }
1263
1264         return xstats_num;
1265 }
1266
1267 static uint64_t
1268 ntb_xstats_get_by_name(const struct rte_rawdev *dev,
1269                        const char *name, unsigned int *id)
1270 {
1271         struct rte_rawdev_xstats_name *xstats_names;
1272         struct ntb_hw *hw = dev->dev_private;
1273         uint32_t xstats_num, i, j, off;
1274
1275         if (name == NULL)
1276                 return -EINVAL;
1277
1278         xstats_num = NTB_XSTATS_NUM * (hw->queue_pairs + 1);
1279         xstats_names = rte_zmalloc("ntb_stats_name",
1280                                    sizeof(struct rte_rawdev_xstats_name) *
1281                                    xstats_num, 0);
1282         ntb_xstats_get_names(dev, xstats_names, xstats_num);
1283
1284         /* Calculate total stats of all queues. */
1285         for (i = 0; i < NTB_XSTATS_NUM; i++) {
1286                 for (j = 0; j < hw->queue_pairs; j++) {
1287                         off = NTB_XSTATS_NUM * (j + 1) + i;
1288                         hw->ntb_xstats[i] +=
1289                         ntb_stats_update(hw->ntb_xstats_off[off],
1290                                          hw->ntb_xstats[off]);
1291                 }
1292         }
1293
1294         for (i = 0; i < xstats_num; i++) {
1295                 if (!strncmp(name, xstats_names[i].name,
1296                     RTE_RAW_DEV_XSTATS_NAME_SIZE)) {
1297                         *id = i;
1298                         rte_free(xstats_names);
1299                         if (i < NTB_XSTATS_NUM)
1300                                 return hw->ntb_xstats[i];
1301                         else
1302                                 return ntb_stats_update(hw->ntb_xstats_off[i],
1303                                                         hw->ntb_xstats[i]);
1304                 }
1305         }
1306
1307         NTB_LOG(ERR, "Cannot find the xstats name.");
1308
1309         return -EINVAL;
1310 }
1311
1312 static int
1313 ntb_xstats_reset(struct rte_rawdev *dev,
1314                  const uint32_t ids[],
1315                  uint32_t nb_ids)
1316 {
1317         struct ntb_hw *hw = dev->dev_private;
1318         uint32_t i, j, off, xstats_num;
1319
1320         xstats_num = NTB_XSTATS_NUM * (hw->queue_pairs + 1);
1321         for (i = 0; i < nb_ids && ids[i] < xstats_num; i++) {
1322                 if (ids[i] < NTB_XSTATS_NUM) {
1323                         for (j = 0; j < hw->queue_pairs; j++) {
1324                                 off = NTB_XSTATS_NUM * (j + 1) + ids[i];
1325                                 hw->ntb_xstats_off[off] = hw->ntb_xstats[off];
1326                         }
1327                 } else {
1328                         hw->ntb_xstats_off[ids[i]] = hw->ntb_xstats[ids[i]];
1329                 }
1330         }
1331
1332         return i;
1333 }
1334
1335 static const struct rte_rawdev_ops ntb_ops = {
1336         .dev_info_get         = ntb_dev_info_get,
1337         .dev_configure        = ntb_dev_configure,
1338         .dev_start            = ntb_dev_start,
1339         .dev_stop             = ntb_dev_stop,
1340         .dev_close            = ntb_dev_close,
1341         .dev_reset            = ntb_dev_reset,
1342
1343         .queue_def_conf       = ntb_queue_conf_get,
1344         .queue_setup          = ntb_queue_setup,
1345         .queue_release        = ntb_queue_release,
1346         .queue_count          = ntb_queue_count,
1347
1348         .enqueue_bufs         = ntb_enqueue_bufs,
1349         .dequeue_bufs         = ntb_dequeue_bufs,
1350
1351         .attr_get             = ntb_attr_get,
1352         .attr_set             = ntb_attr_set,
1353
1354         .xstats_get           = ntb_xstats_get,
1355         .xstats_get_names     = ntb_xstats_get_names,
1356         .xstats_get_by_name   = ntb_xstats_get_by_name,
1357         .xstats_reset         = ntb_xstats_reset,
1358 };
1359
1360 static int
1361 ntb_init_hw(struct rte_rawdev *dev, struct rte_pci_device *pci_dev)
1362 {
1363         struct ntb_hw *hw = dev->dev_private;
1364         struct rte_intr_handle *intr_handle;
1365         int ret, i;
1366
1367         hw->pci_dev = pci_dev;
1368         hw->peer_dev_up = 0;
1369         hw->link_status = NTB_LINK_DOWN;
1370         hw->link_speed = NTB_SPEED_NONE;
1371         hw->link_width = NTB_WIDTH_NONE;
1372
1373         switch (pci_dev->id.device_id) {
1374         case NTB_INTEL_DEV_ID_B2B_SKX:
1375         case NTB_INTEL_DEV_ID_B2B_ICX:
1376                 hw->ntb_ops = &intel_ntb_ops;
1377                 break;
1378         default:
1379                 NTB_LOG(ERR, "Not supported device.");
1380                 return -EINVAL;
1381         }
1382
1383         if (hw->ntb_ops->ntb_dev_init == NULL)
1384                 return -ENOTSUP;
1385         ret = (*hw->ntb_ops->ntb_dev_init)(dev);
1386         if (ret) {
1387                 NTB_LOG(ERR, "Unable to init ntb dev.");
1388                 return ret;
1389         }
1390
1391         if (hw->ntb_ops->set_link == NULL)
1392                 return -ENOTSUP;
1393         ret = (*hw->ntb_ops->set_link)(dev, 1);
1394         if (ret)
1395                 return ret;
1396
1397         /* Init doorbell. */
1398         hw->db_valid_mask = RTE_LEN2MASK(hw->db_cnt, uint64_t);
1399
1400         intr_handle = &pci_dev->intr_handle;
1401         /* Register callback func to eal lib */
1402         rte_intr_callback_register(intr_handle,
1403                                    ntb_dev_intr_handler, dev);
1404
1405         ret = rte_intr_efd_enable(intr_handle, hw->db_cnt);
1406         if (ret)
1407                 return ret;
1408
1409         /* To clarify, the interrupt for each doorbell is already mapped
1410          * by default for intel gen3. They are mapped to msix vec 1-32,
1411          * and hardware intr is mapped to 0. Map all to 0 for uio.
1412          */
1413         if (!rte_intr_cap_multiple(intr_handle)) {
1414                 for (i = 0; i < hw->db_cnt; i++) {
1415                         if (hw->ntb_ops->vector_bind == NULL)
1416                                 return -ENOTSUP;
1417                         ret = (*hw->ntb_ops->vector_bind)(dev, i, 0);
1418                         if (ret)
1419                                 return ret;
1420                 }
1421         }
1422
1423         if (hw->ntb_ops->db_set_mask == NULL ||
1424             hw->ntb_ops->peer_db_set == NULL) {
1425                 NTB_LOG(ERR, "Doorbell is not supported.");
1426                 return -ENOTSUP;
1427         }
1428         hw->db_mask = 0;
1429         ret = (*hw->ntb_ops->db_set_mask)(dev, hw->db_mask);
1430         if (ret) {
1431                 NTB_LOG(ERR, "Unable to enable intr for all dbs.");
1432                 return ret;
1433         }
1434
1435         /* enable uio intr after callback register */
1436         rte_intr_enable(intr_handle);
1437
1438         return ret;
1439 }
1440
1441 static int
1442 ntb_create(struct rte_pci_device *pci_dev, int socket_id)
1443 {
1444         char name[RTE_RAWDEV_NAME_MAX_LEN];
1445         struct rte_rawdev *rawdev = NULL;
1446         int ret;
1447
1448         if (pci_dev == NULL) {
1449                 NTB_LOG(ERR, "Invalid pci_dev.");
1450                 return -EINVAL;
1451         }
1452
1453         memset(name, 0, sizeof(name));
1454         snprintf(name, RTE_RAWDEV_NAME_MAX_LEN, "NTB:%x:%02x.%x",
1455                  pci_dev->addr.bus, pci_dev->addr.devid,
1456                  pci_dev->addr.function);
1457
1458         NTB_LOG(INFO, "Init %s on NUMA node %d", name, socket_id);
1459
1460         /* Allocate device structure. */
1461         rawdev = rte_rawdev_pmd_allocate(name, sizeof(struct ntb_hw),
1462                                          socket_id);
1463         if (rawdev == NULL) {
1464                 NTB_LOG(ERR, "Unable to allocate rawdev.");
1465                 return -EINVAL;
1466         }
1467
1468         rawdev->dev_ops = &ntb_ops;
1469         rawdev->device = &pci_dev->device;
1470         rawdev->driver_name = pci_dev->driver->driver.name;
1471
1472         ret = ntb_init_hw(rawdev, pci_dev);
1473         if (ret < 0) {
1474                 NTB_LOG(ERR, "Unable to init ntb hw.");
1475                 goto fail;
1476         }
1477
1478         return ret;
1479
1480 fail:
1481         if (rawdev != NULL)
1482                 rte_rawdev_pmd_release(rawdev);
1483
1484         return ret;
1485 }
1486
1487 static int
1488 ntb_destroy(struct rte_pci_device *pci_dev)
1489 {
1490         char name[RTE_RAWDEV_NAME_MAX_LEN];
1491         struct rte_rawdev *rawdev;
1492         int ret;
1493
1494         if (pci_dev == NULL) {
1495                 NTB_LOG(ERR, "Invalid pci_dev.");
1496                 ret = -EINVAL;
1497                 return ret;
1498         }
1499
1500         memset(name, 0, sizeof(name));
1501         snprintf(name, RTE_RAWDEV_NAME_MAX_LEN, "NTB:%x:%02x.%x",
1502                  pci_dev->addr.bus, pci_dev->addr.devid,
1503                  pci_dev->addr.function);
1504
1505         NTB_LOG(INFO, "Closing %s on NUMA node %d", name, rte_socket_id());
1506
1507         rawdev = rte_rawdev_pmd_get_named_dev(name);
1508         if (rawdev == NULL) {
1509                 NTB_LOG(ERR, "Invalid device name (%s)", name);
1510                 ret = -EINVAL;
1511                 return ret;
1512         }
1513
1514         ret = rte_rawdev_pmd_release(rawdev);
1515         if (ret)
1516                 NTB_LOG(ERR, "Failed to destroy ntb rawdev.");
1517
1518         return ret;
1519 }
1520
1521 static int
1522 ntb_probe(struct rte_pci_driver *pci_drv __rte_unused,
1523         struct rte_pci_device *pci_dev)
1524 {
1525         return ntb_create(pci_dev, rte_socket_id());
1526 }
1527
1528 static int
1529 ntb_remove(struct rte_pci_device *pci_dev)
1530 {
1531         return ntb_destroy(pci_dev);
1532 }
1533
1534
1535 static struct rte_pci_driver rte_ntb_pmd = {
1536         .id_table = pci_id_ntb_map,
1537         .drv_flags = RTE_PCI_DRV_NEED_MAPPING | RTE_PCI_DRV_WC_ACTIVATE,
1538         .probe = ntb_probe,
1539         .remove = ntb_remove,
1540 };
1541
1542 RTE_PMD_REGISTER_PCI(raw_ntb, rte_ntb_pmd);
1543 RTE_PMD_REGISTER_PCI_TABLE(raw_ntb, pci_id_ntb_map);
1544 RTE_PMD_REGISTER_KMOD_DEP(raw_ntb, "* igb_uio | uio_pci_generic | vfio-pci");
1545 RTE_LOG_REGISTER(ntb_logtype, pmd.raw.ntb, INFO);