drivers/ntb/test/ntb_perf.c

   1 /*
   2  * This file is provided under a dual BSD/GPLv2 license.  When using or
   3  *   redistributing this file, you may do so under either license.
   4  *
   5  *   GPL LICENSE SUMMARY
   6  *
   7  *   Copyright(c) 2015 Intel Corporation. All rights reserved.
   8  *   Copyright(c) 2017 T-Platforms. All Rights Reserved.
   9  *
  10  *   This program is free software; you can redistribute it and/or modify
  11  *   it under the terms of version 2 of the GNU General Public License as
  12  *   published by the Free Software Foundation.
  13  *
  14  *   BSD LICENSE
  15  *
  16  *   Copyright(c) 2015 Intel Corporation. All rights reserved.
  17  *   Copyright(c) 2017 T-Platforms. All Rights Reserved.
  18  *
  19  *   Redistribution and use in source and binary forms, with or without
  20  *   modification, are permitted provided that the following conditions
  21  *   are met:
  22  *
  23  *     * Redistributions of source code must retain the above copyright
  24  *       notice, this list of conditions and the following disclaimer.
  25  *     * Redistributions in binary form must reproduce the above copy
  26  *       notice, this list of conditions and the following disclaimer in
  27  *       the documentation and/or other materials provided with the
  28  *       distribution.
  29  *     * Neither the name of Intel Corporation nor the names of its
  30  *       contributors may be used to endorse or promote products derived
  31  *       from this software without specific prior written permission.
  32  *
  33  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  34  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  35  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  36  *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  37  *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  38  *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  39  *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  40  *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  41  *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  42  *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  43  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  44  *
  45  * PCIe NTB Perf Linux driver
  46  */
  47
  48 /*
  49  * How to use this tool, by example.
  50  *
  51  * Assuming $DBG_DIR is something like:
  52  * '/sys/kernel/debug/ntb_perf/0000:00:03.0'
  53  * Suppose aside from local device there is at least one remote device
  54  * connected to NTB with index 0.
  55  *-----------------------------------------------------------------------------
  56  * Eg: install driver with specified chunk/total orders and dma-enabled flag
  57  *
  58  * root@self# insmod ntb_perf.ko chunk_order=19 total_order=28 use_dma
  59  *-----------------------------------------------------------------------------
  60  * Eg: check NTB ports (index) and MW mapping information
  61  *
  62  * root@self# cat $DBG_DIR/info
  63  *-----------------------------------------------------------------------------
  64  * Eg: start performance test with peer (index 0) and get the test metrics
  65  *
  66  * root@self# echo 0 > $DBG_DIR/run
  67  * root@self# cat $DBG_DIR/run
  68  */
  69
  70 #include <linux/init.h>
  71 #include <linux/kernel.h>
  72 #include <linux/module.h>
  73 #include <linux/sched.h>
  74 #include <linux/wait.h>
  75 #include <linux/dma-mapping.h>
  76 #include <linux/dmaengine.h>
  77 #include <linux/pci.h>
  78 #include <linux/ktime.h>
  79 #include <linux/slab.h>
  80 #include <linux/delay.h>
  81 #include <linux/sizes.h>
  82 #include <linux/workqueue.h>
  83 #include <linux/debugfs.h>
  84 #include <linux/random.h>
  85 #include <linux/ntb.h>
  86
  87 #define DRIVER_NAME             "ntb_perf"
  88 #define DRIVER_VERSION          "2.0"
  89
  90 MODULE_LICENSE("Dual BSD/GPL");
  91 MODULE_VERSION(DRIVER_VERSION);
  92 MODULE_AUTHOR("Dave Jiang <dave.jiang@intel.com>");
  93 MODULE_DESCRIPTION("PCIe NTB Performance Measurement Tool");
  94
  95 #define MAX_THREADS_CNT         32
  96 #define DEF_THREADS_CNT         1
  97 #define MAX_CHUNK_SIZE          SZ_1M
  98 #define MAX_CHUNK_ORDER         20 /* no larger than 1M */
  99
 100 #define DMA_TRIES               100
 101 #define DMA_MDELAY              10
 102
 103 #define MSG_TRIES               1000
 104 #define MSG_UDELAY_LOW          1000
 105 #define MSG_UDELAY_HIGH         2000
 106
 107 #define PERF_BUF_LEN 1024
 108
 109 static unsigned long max_mw_size;
 110 module_param(max_mw_size, ulong, 0644);
 111 MODULE_PARM_DESC(max_mw_size, "Upper limit of memory window size");
 112
 113 static unsigned char chunk_order = 19; /* 512K */
 114 module_param(chunk_order, byte, 0644);
 115 MODULE_PARM_DESC(chunk_order, "Data chunk order [2^n] to transfer");
 116
 117 static unsigned char total_order = 30; /* 1G */
 118 module_param(total_order, byte, 0644);
 119 MODULE_PARM_DESC(total_order, "Total data order [2^n] to transfer");
 120
 121 static bool use_dma; /* default to 0 */
 122 module_param(use_dma, bool, 0644);
 123 MODULE_PARM_DESC(use_dma, "Use DMA engine to measure performance");
 124
 125 /*==============================================================================
 126  *                         Perf driver data definition
 127  *==============================================================================
 128  */
 129
 130 enum perf_cmd {
 131         PERF_CMD_INVAL = -1,/* invalid spad command */
 132         PERF_CMD_SSIZE = 0, /* send out buffer size */
 133         PERF_CMD_RSIZE = 1, /* recv in  buffer size */
 134         PERF_CMD_SXLAT = 2, /* send in  buffer xlat */
 135         PERF_CMD_RXLAT = 3, /* recv out buffer xlat */
 136         PERF_CMD_CLEAR = 4, /* clear allocated memory */
 137         PERF_STS_DONE  = 5, /* init is done */
 138         PERF_STS_LNKUP = 6, /* link up state flag */
 139 };
 140
 141 struct perf_ctx;
 142
 143 struct perf_peer {
 144         struct perf_ctx *perf;
 145         int pidx;
 146         int gidx;
 147
 148         /* Outbound MW params */
 149         u64 outbuf_xlat;
 150         resource_size_t outbuf_size;
 151         void __iomem *outbuf;
 152
 153         /* Inbound MW params */
 154         dma_addr_t inbuf_xlat;
 155         resource_size_t inbuf_size;
 156         void            *inbuf;
 157
 158         /* NTB connection setup service */
 159         struct work_struct      service;
 160         unsigned long           sts;
 161 };
 162 #define to_peer_service(__work) \
 163         container_of(__work, struct perf_peer, service)
 164
 165 struct perf_thread {
 166         struct perf_ctx *perf;
 167         int tidx;
 168
 169         /* DMA-based test sync parameters */
 170         atomic_t dma_sync;
 171         wait_queue_head_t dma_wait;
 172         struct dma_chan *dma_chan;
 173
 174         /* Data source and measured statistics */
 175         void *src;
 176         u64 copied;
 177         ktime_t duration;
 178         int status;
 179         struct work_struct work;
 180 };
 181 #define to_thread_work(__work) \
 182         container_of(__work, struct perf_thread, work)
 183
 184 struct perf_ctx {
 185         struct ntb_dev *ntb;
 186
 187         /* Global device index and peers descriptors */
 188         int gidx;
 189         int pcnt;
 190         struct perf_peer *peers;
 191
 192         /* Performance measuring work-threads interface */
 193         unsigned long busy_flag;
 194         wait_queue_head_t twait;
 195         atomic_t tsync;
 196         u8 tcnt;
 197         struct perf_peer *test_peer;
 198         struct perf_thread threads[MAX_THREADS_CNT];
 199
 200         /* Scratchpad/Message IO operations */
 201         int (*cmd_send)(struct perf_peer *peer, enum perf_cmd cmd, u64 data);
 202         int (*cmd_recv)(struct perf_ctx *perf, int *pidx, enum perf_cmd *cmd,
 203                         u64 *data);
 204
 205         struct dentry *dbgfs_dir;
 206 };
 207
 208 /*
 209  * Scratchpads-base commands interface
 210  */
 211 #define PERF_SPAD_CNT(_pcnt) \
 212         (3*((_pcnt) + 1))
 213 #define PERF_SPAD_CMD(_gidx) \
 214         (3*(_gidx))
 215 #define PERF_SPAD_LDATA(_gidx) \
 216         (3*(_gidx) + 1)
 217 #define PERF_SPAD_HDATA(_gidx) \
 218         (3*(_gidx) + 2)
 219 #define PERF_SPAD_NOTIFY(_gidx) \
 220         (BIT_ULL(_gidx))
 221
 222 /*
 223  * Messages-base commands interface
 224  */
 225 #define PERF_MSG_CNT            3
 226 #define PERF_MSG_CMD            0
 227 #define PERF_MSG_LDATA          1
 228 #define PERF_MSG_HDATA          2
 229
 230 /*==============================================================================
 231  *                           Static data declarations
 232  *==============================================================================
 233  */
 234
 235 static struct dentry *perf_dbgfs_topdir;
 236
 237 static struct workqueue_struct *perf_wq __read_mostly;
 238
 239 /*==============================================================================
 240  *                  NTB cross-link commands execution service
 241  *==============================================================================
 242  */
 243
 244 static void perf_terminate_test(struct perf_ctx *perf);
 245
 246 static inline bool perf_link_is_up(struct perf_peer *peer)
 247 {
 248         u64 link;
 249
 250         link = ntb_link_is_up(peer->perf->ntb, NULL, NULL);
 251         return !!(link & BIT_ULL_MASK(peer->pidx));
 252 }
 253
 254 static int perf_spad_cmd_send(struct perf_peer *peer, enum perf_cmd cmd,
 255                               u64 data)
 256 {
 257         struct perf_ctx *perf = peer->perf;
 258         int try;
 259         u32 sts;
 260
 261         dev_dbg(&perf->ntb->dev, "CMD send: %d 0x%llx\n", cmd, data);
 262
 263         /*
 264          * Perform predefined number of attempts before give up.
 265          * We are sending the data to the port specific scratchpad, so
 266          * to prevent a multi-port access race-condition. Additionally
 267          * there is no need in local locking since only thread-safe
 268          * service work is using this method.
 269          */
 270         for (try = 0; try < MSG_TRIES; try++) {
 271                 if (!perf_link_is_up(peer))
 272                         return -ENOLINK;
 273
 274                 sts = ntb_peer_spad_read(perf->ntb, peer->pidx,
 275                                          PERF_SPAD_CMD(perf->gidx));
 276                 if (sts != PERF_CMD_INVAL) {
 277                         usleep_range(MSG_UDELAY_LOW, MSG_UDELAY_HIGH);
 278                         continue;
 279                 }
 280
 281                 ntb_peer_spad_write(perf->ntb, peer->pidx,
 282                                     PERF_SPAD_LDATA(perf->gidx),
 283                                     lower_32_bits(data));
 284                 ntb_peer_spad_write(perf->ntb, peer->pidx,
 285                                     PERF_SPAD_HDATA(perf->gidx),
 286                                     upper_32_bits(data));
 287                 ntb_peer_spad_write(perf->ntb, peer->pidx,
 288                                     PERF_SPAD_CMD(perf->gidx),
 289                                     cmd);
 290                 ntb_peer_db_set(perf->ntb, PERF_SPAD_NOTIFY(peer->gidx));
 291
 292                 dev_dbg(&perf->ntb->dev, "DB ring peer %#llx\n",
 293                         PERF_SPAD_NOTIFY(peer->gidx));
 294
 295                 break;
 296         }
 297
 298         return try < MSG_TRIES ? 0 : -EAGAIN;
 299 }
 300
 301 static int perf_spad_cmd_recv(struct perf_ctx *perf, int *pidx,
 302                               enum perf_cmd *cmd, u64 *data)
 303 {
 304         struct perf_peer *peer;
 305         u32 val;
 306
 307         ntb_db_clear(perf->ntb, PERF_SPAD_NOTIFY(perf->gidx));
 308
 309         /*
 310          * We start scanning all over, since cleared DB may have been set
 311          * by any peer. Yes, it makes peer with smaller index being
 312          * serviced with greater priority, but it's convenient for spad
 313          * and message code unification and simplicity.
 314          */
 315         for (*pidx = 0; *pidx < perf->pcnt; (*pidx)++) {
 316                 peer = &perf->peers[*pidx];
 317
 318                 if (!perf_link_is_up(peer))
 319                         continue;
 320
 321                 val = ntb_spad_read(perf->ntb, PERF_SPAD_CMD(peer->gidx));
 322                 if (val == PERF_CMD_INVAL)
 323                         continue;
 324
 325                 *cmd = val;
 326
 327                 val = ntb_spad_read(perf->ntb, PERF_SPAD_LDATA(peer->gidx));
 328                 *data = val;
 329
 330                 val = ntb_spad_read(perf->ntb, PERF_SPAD_HDATA(peer->gidx));
 331                 *data |= (u64)val << 32;
 332
 333                 /* Next command can be retrieved from now */
 334                 ntb_spad_write(perf->ntb, PERF_SPAD_CMD(peer->gidx),
 335                                PERF_CMD_INVAL);
 336
 337                 dev_dbg(&perf->ntb->dev, "CMD recv: %d 0x%llx\n", *cmd, *data);
 338
 339                 return 0;
 340         }
 341
 342         return -ENODATA;
 343 }
 344
 345 static int perf_msg_cmd_send(struct perf_peer *peer, enum perf_cmd cmd,
 346                              u64 data)
 347 {
 348         struct perf_ctx *perf = peer->perf;
 349         int try, ret;
 350         u64 outbits;
 351
 352         dev_dbg(&perf->ntb->dev, "CMD send: %d 0x%llx\n", cmd, data);
 353
 354         /*
 355          * Perform predefined number of attempts before give up. Message
 356          * registers are free of race-condition problem when accessed
 357          * from different ports, so we don't need splitting registers
 358          * by global device index. We also won't have local locking,
 359          * since the method is used from service work only.
 360          */
 361         outbits = ntb_msg_outbits(perf->ntb);
 362         for (try = 0; try < MSG_TRIES; try++) {
 363                 if (!perf_link_is_up(peer))
 364                         return -ENOLINK;
 365
 366                 ret = ntb_msg_clear_sts(perf->ntb, outbits);
 367                 if (ret)
 368                         return ret;
 369
 370                 ntb_peer_msg_write(perf->ntb, peer->pidx, PERF_MSG_LDATA,
 371                                    lower_32_bits(data));
 372
 373                 if (ntb_msg_read_sts(perf->ntb) & outbits) {
 374                         usleep_range(MSG_UDELAY_LOW, MSG_UDELAY_HIGH);
 375                         continue;
 376                 }
 377
 378                 ntb_peer_msg_write(perf->ntb, peer->pidx, PERF_MSG_HDATA,
 379                                    upper_32_bits(data));
 380
 381                 /* This call shall trigger peer message event */
 382                 ntb_peer_msg_write(perf->ntb, peer->pidx, PERF_MSG_CMD, cmd);
 383
 384                 break;
 385         }
 386
 387         return try < MSG_TRIES ? 0 : -EAGAIN;
 388 }
 389
 390 static int perf_msg_cmd_recv(struct perf_ctx *perf, int *pidx,
 391                              enum perf_cmd *cmd, u64 *data)
 392 {
 393         u64 inbits;
 394         u32 val;
 395
 396         inbits = ntb_msg_inbits(perf->ntb);
 397
 398         if (hweight64(ntb_msg_read_sts(perf->ntb) & inbits) < 3)
 399                 return -ENODATA;
 400
 401         val = ntb_msg_read(perf->ntb, pidx, PERF_MSG_CMD);
 402         *cmd = val;
 403
 404         val = ntb_msg_read(perf->ntb, pidx, PERF_MSG_LDATA);
 405         *data = val;
 406
 407         val = ntb_msg_read(perf->ntb, pidx, PERF_MSG_HDATA);
 408         *data |= (u64)val << 32;
 409
 410         /* Next command can be retrieved from now */
 411         ntb_msg_clear_sts(perf->ntb, inbits);
 412
 413         dev_dbg(&perf->ntb->dev, "CMD recv: %d 0x%llx\n", *cmd, *data);
 414
 415         return 0;
 416 }
 417
 418 static int perf_cmd_send(struct perf_peer *peer, enum perf_cmd cmd, u64 data)
 419 {
 420         struct perf_ctx *perf = peer->perf;
 421
 422         if (cmd == PERF_CMD_SSIZE || cmd == PERF_CMD_SXLAT)
 423                 return perf->cmd_send(peer, cmd, data);
 424
 425         dev_err(&perf->ntb->dev, "Send invalid command\n");
 426         return -EINVAL;
 427 }
 428
 429 static int perf_cmd_exec(struct perf_peer *peer, enum perf_cmd cmd)
 430 {
 431         switch (cmd) {
 432         case PERF_CMD_SSIZE:
 433         case PERF_CMD_RSIZE:
 434         case PERF_CMD_SXLAT:
 435         case PERF_CMD_RXLAT:
 436         case PERF_CMD_CLEAR:
 437                 break;
 438         default:
 439                 dev_err(&peer->perf->ntb->dev, "Exec invalid command\n");
 440                 return -EINVAL;
 441         }
 442
 443         /* No need of memory barrier, since bit ops have invernal lock */
 444         set_bit(cmd, &peer->sts);
 445
 446         dev_dbg(&peer->perf->ntb->dev, "CMD exec: %d\n", cmd);
 447
 448         (void)queue_work(system_highpri_wq, &peer->service);
 449
 450         return 0;
 451 }
 452
 453 static int perf_cmd_recv(struct perf_ctx *perf)
 454 {
 455         struct perf_peer *peer;
 456         int ret, pidx, cmd;
 457         u64 data;
 458
 459         while (!(ret = perf->cmd_recv(perf, &pidx, &cmd, &data))) {
 460                 peer = &perf->peers[pidx];
 461
 462                 switch (cmd) {
 463                 case PERF_CMD_SSIZE:
 464                         peer->inbuf_size = data;
 465                         return perf_cmd_exec(peer, PERF_CMD_RSIZE);
 466                 case PERF_CMD_SXLAT:
 467                         peer->outbuf_xlat = data;
 468                         return perf_cmd_exec(peer, PERF_CMD_RXLAT);
 469                 default:
 470                         dev_err(&perf->ntb->dev, "Recv invalid command\n");
 471                         return -EINVAL;
 472                 }
 473         }
 474
 475         /* Return 0 if no data left to process, otherwise an error */
 476         return ret == -ENODATA ? 0 : ret;
 477 }
 478
 479 static void perf_link_event(void *ctx)
 480 {
 481         struct perf_ctx *perf = ctx;
 482         struct perf_peer *peer;
 483         bool lnk_up;
 484         int pidx;
 485
 486         for (pidx = 0; pidx < perf->pcnt; pidx++) {
 487                 peer = &perf->peers[pidx];
 488
 489                 lnk_up = perf_link_is_up(peer);
 490
 491                 if (lnk_up &&
 492                     !test_and_set_bit(PERF_STS_LNKUP, &peer->sts)) {
 493                         perf_cmd_exec(peer, PERF_CMD_SSIZE);
 494                 } else if (!lnk_up &&
 495                            test_and_clear_bit(PERF_STS_LNKUP, &peer->sts)) {
 496                         perf_cmd_exec(peer, PERF_CMD_CLEAR);
 497                 }
 498         }
 499 }
 500
 501 static void perf_db_event(void *ctx, int vec)
 502 {
 503         struct perf_ctx *perf = ctx;
 504
 505         dev_dbg(&perf->ntb->dev, "DB vec %d mask %#llx bits %#llx\n", vec,
 506                 ntb_db_vector_mask(perf->ntb, vec), ntb_db_read(perf->ntb));
 507
 508         /* Just receive all available commands */
 509         (void)perf_cmd_recv(perf);
 510 }
 511
 512 static void perf_msg_event(void *ctx)
 513 {
 514         struct perf_ctx *perf = ctx;
 515
 516         dev_dbg(&perf->ntb->dev, "Msg status bits %#llx\n",
 517                 ntb_msg_read_sts(perf->ntb));
 518
 519         /* Messages are only sent one-by-one */
 520         (void)perf_cmd_recv(perf);
 521 }
 522
 523 static const struct ntb_ctx_ops perf_ops = {
 524         .link_event = perf_link_event,
 525         .db_event = perf_db_event,
 526         .msg_event = perf_msg_event
 527 };
 528
 529 static void perf_free_outbuf(struct perf_peer *peer)
 530 {
 531         (void)ntb_peer_mw_clear_trans(peer->perf->ntb, peer->pidx, peer->gidx);
 532 }
 533
 534 static int perf_setup_outbuf(struct perf_peer *peer)
 535 {
 536         struct perf_ctx *perf = peer->perf;
 537         int ret;
 538
 539         /* Outbuf size can be unaligned due to custom max_mw_size */
 540         ret = ntb_peer_mw_set_trans(perf->ntb, peer->pidx, peer->gidx,
 541                                     peer->outbuf_xlat, peer->outbuf_size);
 542         if (ret) {
 543                 dev_err(&perf->ntb->dev, "Failed to set outbuf translation\n");
 544                 return ret;
 545         }
 546
 547         /* Initialization is finally done */
 548         set_bit(PERF_STS_DONE, &peer->sts);
 549
 550         return 0;
 551 }
 552
 553 static void perf_free_inbuf(struct perf_peer *peer)
 554 {
 555         if (!peer->inbuf)
 556                 return;
 557
 558         (void)ntb_mw_clear_trans(peer->perf->ntb, peer->pidx, peer->gidx);
 559         dma_free_coherent(&peer->perf->ntb->dev, peer->inbuf_size,
 560                           peer->inbuf, peer->inbuf_xlat);
 561         peer->inbuf = NULL;
 562 }
 563
 564 static int perf_setup_inbuf(struct perf_peer *peer)
 565 {
 566         resource_size_t xlat_align, size_align, size_max;
 567         struct perf_ctx *perf = peer->perf;
 568         int ret;
 569
 570         /* Get inbound MW parameters */
 571         ret = ntb_mw_get_align(perf->ntb, peer->pidx, perf->gidx,
 572                                &xlat_align, &size_align, &size_max);
 573         if (ret) {
 574                 dev_err(&perf->ntb->dev, "Couldn't get inbuf restrictions\n");
 575                 return ret;
 576         }
 577
 578         if (peer->inbuf_size > size_max) {
 579                 dev_err(&perf->ntb->dev, "Too big inbuf size %pa > %pa\n",
 580                         &peer->inbuf_size, &size_max);
 581                 return -EINVAL;
 582         }
 583
 584         peer->inbuf_size = round_up(peer->inbuf_size, size_align);
 585
 586         perf_free_inbuf(peer);
 587
 588         peer->inbuf = dma_alloc_coherent(&perf->ntb->dev, peer->inbuf_size,
 589                                          &peer->inbuf_xlat, GFP_KERNEL);
 590         if (!peer->inbuf) {
 591                 dev_err(&perf->ntb->dev, "Failed to alloc inbuf of %pa\n",
 592                         &peer->inbuf_size);
 593                 return -ENOMEM;
 594         }
 595         if (!IS_ALIGNED(peer->inbuf_xlat, xlat_align)) {
 596                 dev_err(&perf->ntb->dev, "Unaligned inbuf allocated\n");
 597                 goto err_free_inbuf;
 598         }
 599
 600         ret = ntb_mw_set_trans(perf->ntb, peer->pidx, peer->gidx,
 601                                peer->inbuf_xlat, peer->inbuf_size);
 602         if (ret) {
 603                 dev_err(&perf->ntb->dev, "Failed to set inbuf translation\n");
 604                 goto err_free_inbuf;
 605         }
 606
 607         /*
 608          * We submit inbuf xlat transmission cmd for execution here to follow
 609          * the code architecture, even though this method is called from service
 610          * work itself so the command will be executed right after it returns.
 611          */
 612         (void)perf_cmd_exec(peer, PERF_CMD_SXLAT);
 613
 614         return 0;
 615
 616 err_free_inbuf:
 617         perf_free_inbuf(peer);
 618
 619         return ret;
 620 }
 621
 622 static void perf_service_work(struct work_struct *work)
 623 {
 624         struct perf_peer *peer = to_peer_service(work);
 625
 626         if (test_and_clear_bit(PERF_CMD_SSIZE, &peer->sts))
 627                 perf_cmd_send(peer, PERF_CMD_SSIZE, peer->outbuf_size);
 628
 629         if (test_and_clear_bit(PERF_CMD_RSIZE, &peer->sts))
 630                 perf_setup_inbuf(peer);
 631
 632         if (test_and_clear_bit(PERF_CMD_SXLAT, &peer->sts))
 633                 perf_cmd_send(peer, PERF_CMD_SXLAT, peer->inbuf_xlat);
 634
 635         if (test_and_clear_bit(PERF_CMD_RXLAT, &peer->sts))
 636                 perf_setup_outbuf(peer);
 637
 638         if (test_and_clear_bit(PERF_CMD_CLEAR, &peer->sts)) {
 639                 clear_bit(PERF_STS_DONE, &peer->sts);
 640                 if (test_bit(0, &peer->perf->busy_flag) &&
 641                     peer == peer->perf->test_peer) {
 642                         dev_warn(&peer->perf->ntb->dev,
 643                                 "Freeing while test on-fly\n");
 644                         perf_terminate_test(peer->perf);
 645                 }
 646                 perf_free_outbuf(peer);
 647                 perf_free_inbuf(peer);
 648         }
 649 }
 650
 651 static int perf_init_service(struct perf_ctx *perf)
 652 {
 653         u64 mask;
 654
 655         if (ntb_peer_mw_count(perf->ntb) < perf->pcnt + 1) {
 656                 dev_err(&perf->ntb->dev, "Not enough memory windows\n");
 657                 return -EINVAL;
 658         }
 659
 660         if (ntb_msg_count(perf->ntb) >= PERF_MSG_CNT) {
 661                 perf->cmd_send = perf_msg_cmd_send;
 662                 perf->cmd_recv = perf_msg_cmd_recv;
 663
 664                 dev_dbg(&perf->ntb->dev, "Message service initialized\n");
 665
 666                 return 0;
 667         }
 668
 669         dev_dbg(&perf->ntb->dev, "Message service unsupported\n");
 670
 671         mask = GENMASK_ULL(perf->pcnt, 0);
 672         if (ntb_spad_count(perf->ntb) >= PERF_SPAD_CNT(perf->pcnt) &&
 673             (ntb_db_valid_mask(perf->ntb) & mask) == mask) {
 674                 perf->cmd_send = perf_spad_cmd_send;
 675                 perf->cmd_recv = perf_spad_cmd_recv;
 676
 677                 dev_dbg(&perf->ntb->dev, "Scratchpad service initialized\n");
 678
 679                 return 0;
 680         }
 681
 682         dev_dbg(&perf->ntb->dev, "Scratchpad service unsupported\n");
 683
 684         dev_err(&perf->ntb->dev, "Command services unsupported\n");
 685
 686         return -EINVAL;
 687 }
 688
 689 static int perf_enable_service(struct perf_ctx *perf)
 690 {
 691         u64 mask, incmd_bit;
 692         int ret, sidx, scnt;
 693
 694         mask = ntb_db_valid_mask(perf->ntb);
 695         (void)ntb_db_set_mask(perf->ntb, mask);
 696
 697         ret = ntb_set_ctx(perf->ntb, perf, &perf_ops);
 698         if (ret)
 699                 return ret;
 700
 701         if (perf->cmd_send == perf_msg_cmd_send) {
 702                 u64 inbits, outbits;
 703
 704                 inbits = ntb_msg_inbits(perf->ntb);
 705                 outbits = ntb_msg_outbits(perf->ntb);
 706                 (void)ntb_msg_set_mask(perf->ntb, inbits | outbits);
 707
 708                 incmd_bit = BIT_ULL(__ffs64(inbits));
 709                 ret = ntb_msg_clear_mask(perf->ntb, incmd_bit);
 710
 711                 dev_dbg(&perf->ntb->dev, "MSG sts unmasked %#llx\n", incmd_bit);
 712         } else {
 713                 scnt = ntb_spad_count(perf->ntb);
 714                 for (sidx = 0; sidx < scnt; sidx++)
 715                         ntb_spad_write(perf->ntb, sidx, PERF_CMD_INVAL);
 716                 incmd_bit = PERF_SPAD_NOTIFY(perf->gidx);
 717                 ret = ntb_db_clear_mask(perf->ntb, incmd_bit);
 718
 719                 dev_dbg(&perf->ntb->dev, "DB bits unmasked %#llx\n", incmd_bit);
 720         }
 721         if (ret) {
 722                 ntb_clear_ctx(perf->ntb);
 723                 return ret;
 724         }
 725
 726         ntb_link_enable(perf->ntb, NTB_SPEED_AUTO, NTB_WIDTH_AUTO);
 727         /* Might be not necessary */
 728         ntb_link_event(perf->ntb);
 729
 730         return 0;
 731 }
 732
 733 static void perf_disable_service(struct perf_ctx *perf)
 734 {
 735         int pidx;
 736
 737         if (perf->cmd_send == perf_msg_cmd_send) {
 738                 u64 inbits;
 739
 740                 inbits = ntb_msg_inbits(perf->ntb);
 741                 (void)ntb_msg_set_mask(perf->ntb, inbits);
 742         } else {
 743                 (void)ntb_db_set_mask(perf->ntb, PERF_SPAD_NOTIFY(perf->gidx));
 744         }
 745
 746         ntb_clear_ctx(perf->ntb);
 747
 748         for (pidx = 0; pidx < perf->pcnt; pidx++)
 749                 perf_cmd_exec(&perf->peers[pidx], PERF_CMD_CLEAR);
 750
 751         for (pidx = 0; pidx < perf->pcnt; pidx++)
 752                 flush_work(&perf->peers[pidx].service);
 753
 754         for (pidx = 0; pidx < perf->pcnt; pidx++) {
 755                 struct perf_peer *peer = &perf->peers[pidx];
 756
 757                 ntb_spad_write(perf->ntb, PERF_SPAD_CMD(peer->gidx), 0);
 758         }
 759
 760         ntb_db_clear(perf->ntb, PERF_SPAD_NOTIFY(perf->gidx));
 761
 762         ntb_link_disable(perf->ntb);
 763 }
 764
 765 /*==============================================================================
 766  *                      Performance measuring work-thread
 767  *==============================================================================
 768  */
 769
 770 static void perf_dma_copy_callback(void *data)
 771 {
 772         struct perf_thread *pthr = data;
 773
 774         atomic_dec(&pthr->dma_sync);
 775         wake_up(&pthr->dma_wait);
 776 }
 777
 778 static int perf_copy_chunk(struct perf_thread *pthr,
 779                            void __iomem *dst, void *src, size_t len)
 780 {
 781         struct dma_async_tx_descriptor *tx;
 782         struct dmaengine_unmap_data *unmap;
 783         struct device *dma_dev;
 784         int try = 0, ret = 0;
 785
 786         if (!use_dma) {
 787                 memcpy_toio(dst, src, len);
 788                 goto ret_check_tsync;
 789         }
 790
 791         dma_dev = pthr->dma_chan->device->dev;
 792
 793         if (!is_dma_copy_aligned(pthr->dma_chan->device, offset_in_page(src),
 794                                  offset_in_page(dst), len))
 795                 return -EIO;
 796
 797         unmap = dmaengine_get_unmap_data(dma_dev, 2, GFP_NOWAIT);
 798         if (!unmap)
 799                 return -ENOMEM;
 800
 801         unmap->len = len;
 802         unmap->addr[0] = dma_map_page(dma_dev, virt_to_page(src),
 803                 offset_in_page(src), len, DMA_TO_DEVICE);
 804         if (dma_mapping_error(dma_dev, unmap->addr[0])) {
 805                 ret = -EIO;
 806                 goto err_free_resource;
 807         }
 808         unmap->to_cnt = 1;
 809
 810         unmap->addr[1] = dma_map_page(dma_dev, virt_to_page(dst),
 811                 offset_in_page(dst), len, DMA_FROM_DEVICE);
 812         if (dma_mapping_error(dma_dev, unmap->addr[1])) {
 813                 ret = -EIO;
 814                 goto err_free_resource;
 815         }
 816         unmap->from_cnt = 1;
 817
 818         do {
 819                 tx = dmaengine_prep_dma_memcpy(pthr->dma_chan, unmap->addr[1],
 820                         unmap->addr[0], len, DMA_PREP_INTERRUPT | DMA_CTRL_ACK);
 821                 if (!tx)
 822                         msleep(DMA_MDELAY);
 823         } while (!tx && (try++ < DMA_TRIES));
 824
 825         if (!tx) {
 826                 ret = -EIO;
 827                 goto err_free_resource;
 828         }
 829
 830         tx->callback = perf_dma_copy_callback;
 831         tx->callback_param = pthr;
 832         dma_set_unmap(tx, unmap);
 833
 834         ret = dma_submit_error(dmaengine_submit(tx));
 835         if (ret) {
 836                 dmaengine_unmap_put(unmap);
 837                 goto err_free_resource;
 838         }
 839
 840         dmaengine_unmap_put(unmap);
 841
 842         atomic_inc(&pthr->dma_sync);
 843         dma_async_issue_pending(pthr->dma_chan);
 844
 845 ret_check_tsync:
 846         return likely(atomic_read(&pthr->perf->tsync) > 0) ? 0 : -EINTR;
 847
 848 err_free_resource:
 849         dmaengine_unmap_put(unmap);
 850
 851         return ret;
 852 }
 853
 854 static bool perf_dma_filter(struct dma_chan *chan, void *data)
 855 {
 856         struct perf_ctx *perf = data;
 857         int node;
 858
 859         node = dev_to_node(&perf->ntb->dev);
 860
 861         return node == NUMA_NO_NODE || node == dev_to_node(chan->device->dev);
 862 }
 863
 864 static int perf_init_test(struct perf_thread *pthr)
 865 {
 866         struct perf_ctx *perf = pthr->perf;
 867         dma_cap_mask_t dma_mask;
 868
 869         pthr->src = kmalloc_node(perf->test_peer->outbuf_size, GFP_KERNEL,
 870                                  dev_to_node(&perf->ntb->dev));
 871         if (!pthr->src)
 872                 return -ENOMEM;
 873
 874         get_random_bytes(pthr->src, perf->test_peer->outbuf_size);
 875
 876         if (!use_dma)
 877                 return 0;
 878
 879         dma_cap_zero(dma_mask);
 880         dma_cap_set(DMA_MEMCPY, dma_mask);
 881         pthr->dma_chan = dma_request_channel(dma_mask, perf_dma_filter, perf);
 882         if (!pthr->dma_chan) {
 883                 dev_err(&perf->ntb->dev, "%d: Failed to get DMA channel\n",
 884                         pthr->tidx);
 885                 atomic_dec(&perf->tsync);
 886                 wake_up(&perf->twait);
 887                 kfree(pthr->src);
 888                 return -ENODEV;
 889         }
 890
 891         atomic_set(&pthr->dma_sync, 0);
 892
 893         return 0;
 894 }
 895
 896 static int perf_run_test(struct perf_thread *pthr)
 897 {
 898         struct perf_peer *peer = pthr->perf->test_peer;
 899         struct perf_ctx *perf = pthr->perf;
 900         void __iomem *flt_dst, *bnd_dst;
 901         u64 total_size, chunk_size;
 902         void *flt_src;
 903         int ret = 0;
 904
 905         total_size = 1ULL << total_order;
 906         chunk_size = 1ULL << chunk_order;
 907         chunk_size = min_t(u64, peer->outbuf_size, chunk_size);
 908
 909         flt_src = pthr->src;
 910         bnd_dst = peer->outbuf + peer->outbuf_size;
 911         flt_dst = peer->outbuf;
 912
 913         pthr->duration = ktime_get();
 914
 915         /* Copied field is cleared on test launch stage */
 916         while (pthr->copied < total_size) {
 917                 ret = perf_copy_chunk(pthr, flt_dst, flt_src, chunk_size);
 918                 if (ret) {
 919                         dev_err(&perf->ntb->dev, "%d: Got error %d on test\n",
 920                                 pthr->tidx, ret);
 921                         return ret;
 922                 }
 923
 924                 pthr->copied += chunk_size;
 925
 926                 flt_dst += chunk_size;
 927                 flt_src += chunk_size;
 928                 if (flt_dst >= bnd_dst || flt_dst < peer->outbuf) {
 929                         flt_dst = peer->outbuf;
 930                         flt_src = pthr->src;
 931                 }
 932
 933                 /* Give up CPU to give a chance for other threads to use it */
 934                 schedule();
 935         }
 936
 937         return 0;
 938 }
 939
 940 static int perf_sync_test(struct perf_thread *pthr)
 941 {
 942         struct perf_ctx *perf = pthr->perf;
 943
 944         if (!use_dma)
 945                 goto no_dma_ret;
 946
 947         wait_event(pthr->dma_wait,
 948                    (atomic_read(&pthr->dma_sync) == 0 ||
 949                     atomic_read(&perf->tsync) < 0));
 950
 951         if (atomic_read(&perf->tsync) < 0)
 952                 return -EINTR;
 953
 954 no_dma_ret:
 955         pthr->duration = ktime_sub(ktime_get(), pthr->duration);
 956
 957         dev_dbg(&perf->ntb->dev, "%d: copied %llu bytes\n",
 958                 pthr->tidx, pthr->copied);
 959
 960         dev_dbg(&perf->ntb->dev, "%d: lasted %llu usecs\n",
 961                 pthr->tidx, ktime_to_us(pthr->duration));
 962
 963         dev_dbg(&perf->ntb->dev, "%d: %llu MBytes/s\n", pthr->tidx,
 964                 div64_u64(pthr->copied, ktime_to_us(pthr->duration)));
 965
 966         return 0;
 967 }
 968
 969 static void perf_clear_test(struct perf_thread *pthr)
 970 {
 971         struct perf_ctx *perf = pthr->perf;
 972
 973         if (!use_dma)
 974                 goto no_dma_notify;
 975
 976         /*
 977          * If test finished without errors, termination isn't needed.
 978          * We call it anyway just to be sure of the transfers completion.
 979          */
 980         (void)dmaengine_terminate_sync(pthr->dma_chan);
 981
 982         dma_release_channel(pthr->dma_chan);
 983
 984 no_dma_notify:
 985         atomic_dec(&perf->tsync);
 986         wake_up(&perf->twait);
 987         kfree(pthr->src);
 988 }
 989
 990 static void perf_thread_work(struct work_struct *work)
 991 {
 992         struct perf_thread *pthr = to_thread_work(work);
 993         int ret;
 994
 995         /*
 996          * Perform stages in compliance with use_dma flag value.
 997          * Test status is changed only if error happened, otherwise
 998          * status -ENODATA is kept while test is on-fly. Results
 999          * synchronization is performed only if test fininshed
1000          * without an error or interruption.
1001          */
1002         ret = perf_init_test(pthr);
1003         if (ret) {
1004                 pthr->status = ret;
1005                 return;
1006         }
1007
1008         ret = perf_run_test(pthr);
1009         if (ret) {
1010                 pthr->status = ret;
1011                 goto err_clear_test;
1012         }
1013
1014         pthr->status = perf_sync_test(pthr);
1015
1016 err_clear_test:
1017         perf_clear_test(pthr);
1018 }
1019
1020 static int perf_set_tcnt(struct perf_ctx *perf, u8 tcnt)
1021 {
1022         if (tcnt == 0 || tcnt > MAX_THREADS_CNT)
1023                 return -EINVAL;
1024
1025         if (test_and_set_bit_lock(0, &perf->busy_flag))
1026                 return -EBUSY;
1027
1028         perf->tcnt = tcnt;
1029
1030         clear_bit_unlock(0, &perf->busy_flag);
1031
1032         return 0;
1033 }
1034
1035 static void perf_terminate_test(struct perf_ctx *perf)
1036 {
1037         int tidx;
1038
1039         atomic_set(&perf->tsync, -1);
1040         wake_up(&perf->twait);
1041
1042         for (tidx = 0; tidx < MAX_THREADS_CNT; tidx++) {
1043                 wake_up(&perf->threads[tidx].dma_wait);
1044                 cancel_work_sync(&perf->threads[tidx].work);
1045         }
1046 }
1047
1048 static int perf_submit_test(struct perf_peer *peer)
1049 {
1050         struct perf_ctx *perf = peer->perf;
1051         struct perf_thread *pthr;
1052         int tidx, ret;
1053
1054         if (!test_bit(PERF_STS_DONE, &peer->sts))
1055                 return -ENOLINK;
1056
1057         if (test_and_set_bit_lock(0, &perf->busy_flag))
1058                 return -EBUSY;
1059
1060         perf->test_peer = peer;
1061         atomic_set(&perf->tsync, perf->tcnt);
1062
1063         for (tidx = 0; tidx < MAX_THREADS_CNT; tidx++) {
1064                 pthr = &perf->threads[tidx];
1065
1066                 pthr->status = -ENODATA;
1067                 pthr->copied = 0;
1068                 pthr->duration = ktime_set(0, 0);
1069                 if (tidx < perf->tcnt)
1070                         (void)queue_work(perf_wq, &pthr->work);
1071         }
1072
1073         ret = wait_event_interruptible(perf->twait,
1074                                        atomic_read(&perf->tsync) <= 0);
1075         if (ret == -ERESTARTSYS) {
1076                 perf_terminate_test(perf);
1077                 ret = -EINTR;
1078         }
1079
1080         clear_bit_unlock(0, &perf->busy_flag);
1081
1082         return ret;
1083 }
1084
1085 static int perf_read_stats(struct perf_ctx *perf, char *buf,
1086                            size_t size, ssize_t *pos)
1087 {
1088         struct perf_thread *pthr;
1089         int tidx;
1090
1091         if (test_and_set_bit_lock(0, &perf->busy_flag))
1092                 return -EBUSY;
1093
1094         (*pos) += scnprintf(buf + *pos, size - *pos,
1095                 "    Peer %d test statistics:\n", perf->test_peer->pidx);
1096
1097         for (tidx = 0; tidx < MAX_THREADS_CNT; tidx++) {
1098                 pthr = &perf->threads[tidx];
1099
1100                 if (pthr->status == -ENODATA)
1101                         continue;
1102
1103                 if (pthr->status) {
1104                         (*pos) += scnprintf(buf + *pos, size - *pos,
1105                                 "%d: error status %d\n", tidx, pthr->status);
1106                         continue;
1107                 }
1108
1109                 (*pos) += scnprintf(buf + *pos, size - *pos,
1110                         "%d: copied %llu bytes in %llu usecs, %llu MBytes/s\n",
1111                         tidx, pthr->copied, ktime_to_us(pthr->duration),
1112                         div64_u64(pthr->copied, ktime_to_us(pthr->duration)));
1113         }
1114
1115         clear_bit_unlock(0, &perf->busy_flag);
1116
1117         return 0;
1118 }
1119
1120 static void perf_init_threads(struct perf_ctx *perf)
1121 {
1122         struct perf_thread *pthr;
1123         int tidx;
1124
1125         perf->tcnt = DEF_THREADS_CNT;
1126         perf->test_peer = &perf->peers[0];
1127         init_waitqueue_head(&perf->twait);
1128
1129         for (tidx = 0; tidx < MAX_THREADS_CNT; tidx++) {
1130                 pthr = &perf->threads[tidx];
1131
1132                 pthr->perf = perf;
1133                 pthr->tidx = tidx;
1134                 pthr->status = -ENODATA;
1135                 init_waitqueue_head(&pthr->dma_wait);
1136                 INIT_WORK(&pthr->work, perf_thread_work);
1137         }
1138 }
1139
1140 static void perf_clear_threads(struct perf_ctx *perf)
1141 {
1142         perf_terminate_test(perf);
1143 }
1144
1145 /*==============================================================================
1146  *                               DebugFS nodes
1147  *==============================================================================
1148  */
1149
1150 static ssize_t perf_dbgfs_read_info(struct file *filep, char __user *ubuf,
1151                                     size_t size, loff_t *offp)
1152 {
1153         struct perf_ctx *perf = filep->private_data;
1154         struct perf_peer *peer;
1155         size_t buf_size;
1156         ssize_t pos = 0;
1157         int ret, pidx;
1158         char *buf;
1159
1160         buf_size = min_t(size_t, size, 0x1000U);
1161
1162         buf = kmalloc(buf_size, GFP_KERNEL);
1163         if (!buf)
1164                 return -ENOMEM;
1165
1166         pos += scnprintf(buf + pos, buf_size - pos,
1167                 "    Performance measuring tool info:\n\n");
1168
1169         pos += scnprintf(buf + pos, buf_size - pos,
1170                 "Local port %d, Global index %d\n", ntb_port_number(perf->ntb),
1171                 perf->gidx);
1172         pos += scnprintf(buf + pos, buf_size - pos, "Test status: ");
1173         if (test_bit(0, &perf->busy_flag)) {
1174                 pos += scnprintf(buf + pos, buf_size - pos,
1175                         "on-fly with port %d (%d)\n",
1176                         ntb_peer_port_number(perf->ntb, perf->test_peer->pidx),
1177                         perf->test_peer->pidx);
1178         } else {
1179                 pos += scnprintf(buf + pos, buf_size - pos, "idle\n");
1180         }
1181
1182         for (pidx = 0; pidx < perf->pcnt; pidx++) {
1183                 peer = &perf->peers[pidx];
1184
1185                 pos += scnprintf(buf + pos, buf_size - pos,
1186                         "Port %d (%d), Global index %d:\n",
1187                         ntb_peer_port_number(perf->ntb, peer->pidx), peer->pidx,
1188                         peer->gidx);
1189
1190                 pos += scnprintf(buf + pos, buf_size - pos,
1191                         "\tLink status: %s\n",
1192                         test_bit(PERF_STS_LNKUP, &peer->sts) ? "up" : "down");
1193
1194                 pos += scnprintf(buf + pos, buf_size - pos,
1195                         "\tOut buffer addr 0x%pK\n", peer->outbuf);
1196
1197                 pos += scnprintf(buf + pos, buf_size - pos,
1198                         "\tOut buffer size %pa\n", &peer->outbuf_size);
1199
1200                 pos += scnprintf(buf + pos, buf_size - pos,
1201                         "\tOut buffer xlat 0x%016llx[p]\n", peer->outbuf_xlat);
1202
1203                 if (!peer->inbuf) {
1204                         pos += scnprintf(buf + pos, buf_size - pos,
1205                                 "\tIn buffer addr: unallocated\n");
1206                         continue;
1207                 }
1208
1209                 pos += scnprintf(buf + pos, buf_size - pos,
1210                         "\tIn buffer addr 0x%pK\n", peer->inbuf);
1211
1212                 pos += scnprintf(buf + pos, buf_size - pos,
1213                         "\tIn buffer size %pa\n", &peer->inbuf_size);
1214
1215                 pos += scnprintf(buf + pos, buf_size - pos,
1216                         "\tIn buffer xlat %pad[p]\n", &peer->inbuf_xlat);
1217         }
1218
1219         ret = simple_read_from_buffer(ubuf, size, offp, buf, pos);
1220         kfree(buf);
1221
1222         return ret;
1223 }
1224
1225 static const struct file_operations perf_dbgfs_info = {
1226         .open = simple_open,
1227         .read = perf_dbgfs_read_info
1228 };
1229
1230 static ssize_t perf_dbgfs_read_run(struct file *filep, char __user *ubuf,
1231                                    size_t size, loff_t *offp)
1232 {
1233         struct perf_ctx *perf = filep->private_data;
1234         ssize_t ret, pos = 0;
1235         char *buf;
1236
1237         buf = kmalloc(PERF_BUF_LEN, GFP_KERNEL);
1238         if (!buf)
1239                 return -ENOMEM;
1240
1241         ret = perf_read_stats(perf, buf, PERF_BUF_LEN, &pos);
1242         if (ret)
1243                 goto err_free;
1244
1245         ret = simple_read_from_buffer(ubuf, size, offp, buf, pos);
1246 err_free:
1247         kfree(buf);
1248
1249         return ret;
1250 }
1251
1252 static ssize_t perf_dbgfs_write_run(struct file *filep, const char __user *ubuf,
1253                                     size_t size, loff_t *offp)
1254 {
1255         struct perf_ctx *perf = filep->private_data;
1256         struct perf_peer *peer;
1257         int pidx, ret;
1258
1259         ret = kstrtoint_from_user(ubuf, size, 0, &pidx);
1260         if (ret)
1261                 return ret;
1262
1263         if (pidx < 0 || pidx >= perf->pcnt)
1264                 return -EINVAL;
1265
1266         peer = &perf->peers[pidx];
1267
1268         ret = perf_submit_test(peer);
1269         if (ret)
1270                 return ret;
1271
1272         return size;
1273 }
1274
1275 static const struct file_operations perf_dbgfs_run = {
1276         .open = simple_open,
1277         .read = perf_dbgfs_read_run,
1278         .write = perf_dbgfs_write_run
1279 };
1280
1281 static ssize_t perf_dbgfs_read_tcnt(struct file *filep, char __user *ubuf,
1282                                     size_t size, loff_t *offp)
1283 {
1284         struct perf_ctx *perf = filep->private_data;
1285         char buf[8];
1286         ssize_t pos;
1287
1288         pos = scnprintf(buf, sizeof(buf), "%hhu\n", perf->tcnt);
1289
1290         return simple_read_from_buffer(ubuf, size, offp, buf, pos);
1291 }
1292
1293 static ssize_t perf_dbgfs_write_tcnt(struct file *filep,
1294                                      const char __user *ubuf,
1295                                      size_t size, loff_t *offp)
1296 {
1297         struct perf_ctx *perf = filep->private_data;
1298         int ret;
1299         u8 val;
1300
1301         ret = kstrtou8_from_user(ubuf, size, 0, &val);
1302         if (ret)
1303                 return ret;
1304
1305         ret = perf_set_tcnt(perf, val);
1306         if (ret)
1307                 return ret;
1308
1309         return size;
1310 }
1311
1312 static const struct file_operations perf_dbgfs_tcnt = {
1313         .open = simple_open,
1314         .read = perf_dbgfs_read_tcnt,
1315         .write = perf_dbgfs_write_tcnt
1316 };
1317
1318 static void perf_setup_dbgfs(struct perf_ctx *perf)
1319 {
1320         struct pci_dev *pdev = perf->ntb->pdev;
1321
1322         perf->dbgfs_dir = debugfs_create_dir(pci_name(pdev), perf_dbgfs_topdir);
1323         if (!perf->dbgfs_dir) {
1324                 dev_warn(&perf->ntb->dev, "DebugFS unsupported\n");
1325                 return;
1326         }
1327
1328         debugfs_create_file("info", 0600, perf->dbgfs_dir, perf,
1329                             &perf_dbgfs_info);
1330
1331         debugfs_create_file("run", 0600, perf->dbgfs_dir, perf,
1332                             &perf_dbgfs_run);
1333
1334         debugfs_create_file("threads_count", 0600, perf->dbgfs_dir, perf,
1335                             &perf_dbgfs_tcnt);
1336
1337         /* They are made read-only for test exec safety and integrity */
1338         debugfs_create_u8("chunk_order", 0500, perf->dbgfs_dir, &chunk_order);
1339
1340         debugfs_create_u8("total_order", 0500, perf->dbgfs_dir, &total_order);
1341
1342         debugfs_create_bool("use_dma", 0500, perf->dbgfs_dir, &use_dma);
1343 }
1344
1345 static void perf_clear_dbgfs(struct perf_ctx *perf)
1346 {
1347         debugfs_remove_recursive(perf->dbgfs_dir);
1348 }
1349
1350 /*==============================================================================
1351  *                        Basic driver initialization
1352  *==============================================================================
1353  */
1354
1355 static struct perf_ctx *perf_create_data(struct ntb_dev *ntb)
1356 {
1357         struct perf_ctx *perf;
1358
1359         perf = devm_kzalloc(&ntb->dev, sizeof(*perf), GFP_KERNEL);
1360         if (!perf)
1361                 return ERR_PTR(-ENOMEM);
1362
1363         perf->pcnt = ntb_peer_port_count(ntb);
1364         perf->peers = devm_kcalloc(&ntb->dev, perf->pcnt, sizeof(*perf->peers),
1365                                   GFP_KERNEL);
1366         if (!perf->peers)
1367                 return ERR_PTR(-ENOMEM);
1368
1369         perf->ntb = ntb;
1370
1371         return perf;
1372 }
1373
1374 static int perf_setup_peer_mw(struct perf_peer *peer)
1375 {
1376         struct perf_ctx *perf = peer->perf;
1377         phys_addr_t phys_addr;
1378         int ret;
1379
1380         /* Get outbound MW parameters and map it */
1381         ret = ntb_peer_mw_get_addr(perf->ntb, peer->gidx, &phys_addr,
1382                                    &peer->outbuf_size);
1383         if (ret)
1384                 return ret;
1385
1386         peer->outbuf = devm_ioremap_wc(&perf->ntb->dev, phys_addr,
1387                                         peer->outbuf_size);
1388         if (!peer->outbuf)
1389                 return -ENOMEM;
1390
1391         if (max_mw_size && peer->outbuf_size > max_mw_size) {
1392                 peer->outbuf_size = max_mw_size;
1393                 dev_warn(&peer->perf->ntb->dev,
1394                         "Peer %d outbuf reduced to %pa\n", peer->pidx,
1395                         &peer->outbuf_size);
1396         }
1397
1398         return 0;
1399 }
1400
1401 static int perf_init_peers(struct perf_ctx *perf)
1402 {
1403         struct perf_peer *peer;
1404         int pidx, lport, ret;
1405
1406         lport = ntb_port_number(perf->ntb);
1407         perf->gidx = -1;
1408         for (pidx = 0; pidx < perf->pcnt; pidx++) {
1409                 peer = &perf->peers[pidx];
1410
1411                 peer->perf = perf;
1412                 peer->pidx = pidx;
1413                 if (lport < ntb_peer_port_number(perf->ntb, pidx)) {
1414                         if (perf->gidx == -1)
1415                                 perf->gidx = pidx;
1416                         peer->gidx = pidx + 1;
1417                 } else {
1418                         peer->gidx = pidx;
1419                 }
1420                 INIT_WORK(&peer->service, perf_service_work);
1421         }
1422         if (perf->gidx == -1)
1423                 perf->gidx = pidx;
1424
1425         for (pidx = 0; pidx < perf->pcnt; pidx++) {
1426                 ret = perf_setup_peer_mw(&perf->peers[pidx]);
1427                 if (ret)
1428                         return ret;
1429         }
1430
1431         dev_dbg(&perf->ntb->dev, "Global port index %d\n", perf->gidx);
1432
1433         return 0;
1434 }
1435
1436 static int perf_probe(struct ntb_client *client, struct ntb_dev *ntb)
1437 {
1438         struct perf_ctx *perf;
1439         int ret;
1440
1441         perf = perf_create_data(ntb);
1442         if (IS_ERR(perf))
1443                 return PTR_ERR(perf);
1444
1445         ret = perf_init_peers(perf);
1446         if (ret)
1447                 return ret;
1448
1449         perf_init_threads(perf);
1450
1451         ret = perf_init_service(perf);
1452         if (ret)
1453                 return ret;
1454
1455         ret = perf_enable_service(perf);
1456         if (ret)
1457                 return ret;
1458
1459         perf_setup_dbgfs(perf);
1460
1461         return 0;
1462 }
1463
1464 static void perf_remove(struct ntb_client *client, struct ntb_dev *ntb)
1465 {
1466         struct perf_ctx *perf = ntb->ctx;
1467
1468         perf_clear_dbgfs(perf);
1469
1470         perf_disable_service(perf);
1471
1472         perf_clear_threads(perf);
1473 }
1474
1475 static struct ntb_client perf_client = {
1476         .ops = {
1477                 .probe = perf_probe,
1478                 .remove = perf_remove
1479         }
1480 };
1481
1482 static int __init perf_init(void)
1483 {
1484         int ret;
1485
1486         if (chunk_order > MAX_CHUNK_ORDER) {
1487                 chunk_order = MAX_CHUNK_ORDER;
1488                 pr_info("Chunk order reduced to %hhu\n", chunk_order);
1489         }
1490
1491         if (total_order < chunk_order) {
1492                 total_order = chunk_order;
1493                 pr_info("Total data order reduced to %hhu\n", total_order);
1494         }
1495
1496         perf_wq = alloc_workqueue("perf_wq", WQ_UNBOUND | WQ_SYSFS, 0);
1497         if (!perf_wq)
1498                 return -ENOMEM;
1499
1500         if (debugfs_initialized())
1501                 perf_dbgfs_topdir = debugfs_create_dir(KBUILD_MODNAME, NULL);
1502
1503         ret = ntb_register_client(&perf_client);
1504         if (ret) {
1505                 debugfs_remove_recursive(perf_dbgfs_topdir);
1506                 destroy_workqueue(perf_wq);
1507         }
1508
1509         return ret;
1510 }
1511 module_init(perf_init);
1512
1513 static void __exit perf_exit(void)
1514 {
1515         ntb_unregister_client(&perf_client);
1516         debugfs_remove_recursive(perf_dbgfs_topdir);
1517         destroy_workqueue(perf_wq);
1518 }
1519 module_exit(perf_exit);
1520