drivers/misc/pci_endpoint_test.c

   1 /**
   2  * Host side test driver to test endpoint functionality
   3  *
   4  * Copyright (C) 2017 Texas Instruments
   5  * Author: Kishon Vijay Abraham I <kishon@ti.com>
   6  *
   7  * This program is free software: you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License version 2 of
   9  * the License as published by the Free Software Foundation.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 #include <linux/crc32.h>
  21 #include <linux/delay.h>
  22 #include <linux/fs.h>
  23 #include <linux/io.h>
  24 #include <linux/interrupt.h>
  25 #include <linux/irq.h>
  26 #include <linux/miscdevice.h>
  27 #include <linux/module.h>
  28 #include <linux/mutex.h>
  29 #include <linux/random.h>
  30 #include <linux/slab.h>
  31 #include <linux/pci.h>
  32 #include <linux/pci_ids.h>
  33
  34 #include <linux/pci_regs.h>
  35
  36 #include <uapi/linux/pcitest.h>
  37
  38 #define DRV_MODULE_NAME                         "pci-endpoint-test"
  39
  40 #define IRQ_TYPE_UNDEFINED                      -1
  41 #define IRQ_TYPE_LEGACY                         0
  42 #define IRQ_TYPE_MSI                            1
  43 #define IRQ_TYPE_MSIX                           2
  44
  45 #define PCI_ENDPOINT_TEST_MAGIC                 0x0
  46
  47 #define PCI_ENDPOINT_TEST_COMMAND               0x4
  48 #define COMMAND_RAISE_LEGACY_IRQ                BIT(0)
  49 #define COMMAND_RAISE_MSI_IRQ                   BIT(1)
  50 #define COMMAND_RAISE_MSIX_IRQ                  BIT(2)
  51 #define COMMAND_READ                            BIT(3)
  52 #define COMMAND_WRITE                           BIT(4)
  53 #define COMMAND_COPY                            BIT(5)
  54
  55 #define PCI_ENDPOINT_TEST_STATUS                0x8
  56 #define STATUS_READ_SUCCESS                     BIT(0)
  57 #define STATUS_READ_FAIL                        BIT(1)
  58 #define STATUS_WRITE_SUCCESS                    BIT(2)
  59 #define STATUS_WRITE_FAIL                       BIT(3)
  60 #define STATUS_COPY_SUCCESS                     BIT(4)
  61 #define STATUS_COPY_FAIL                        BIT(5)
  62 #define STATUS_IRQ_RAISED                       BIT(6)
  63 #define STATUS_SRC_ADDR_INVALID                 BIT(7)
  64 #define STATUS_DST_ADDR_INVALID                 BIT(8)
  65
  66 #define PCI_ENDPOINT_TEST_LOWER_SRC_ADDR        0x0c
  67 #define PCI_ENDPOINT_TEST_UPPER_SRC_ADDR        0x10
  68
  69 #define PCI_ENDPOINT_TEST_LOWER_DST_ADDR        0x14
  70 #define PCI_ENDPOINT_TEST_UPPER_DST_ADDR        0x18
  71
  72 #define PCI_ENDPOINT_TEST_SIZE                  0x1c
  73 #define PCI_ENDPOINT_TEST_CHECKSUM              0x20
  74
  75 #define PCI_ENDPOINT_TEST_IRQ_TYPE              0x24
  76 #define PCI_ENDPOINT_TEST_IRQ_NUMBER            0x28
  77
  78 static DEFINE_IDA(pci_endpoint_test_ida);
  79
  80 #define to_endpoint_test(priv) container_of((priv), struct pci_endpoint_test, \
  81                                             miscdev)
  82
  83 static bool no_msi;
  84 module_param(no_msi, bool, 0444);
  85 MODULE_PARM_DESC(no_msi, "Disable MSI interrupt in pci_endpoint_test");
  86
  87 static int irq_type = IRQ_TYPE_MSI;
  88 module_param(irq_type, int, 0444);
  89 MODULE_PARM_DESC(irq_type, "IRQ mode selection in pci_endpoint_test (0 - Legacy, 1 - MSI, 2 - MSI-X)");
  90
  91 enum pci_barno {
  92         BAR_0,
  93         BAR_1,
  94         BAR_2,
  95         BAR_3,
  96         BAR_4,
  97         BAR_5,
  98 };
  99
 100 struct pci_endpoint_test {
 101         struct pci_dev  *pdev;
 102         void __iomem    *base;
 103         void __iomem    *bar[6];
 104         struct completion irq_raised;
 105         int             last_irq;
 106         int             num_irqs;
 107         /* mutex to protect the ioctls */
 108         struct mutex    mutex;
 109         struct miscdevice miscdev;
 110         enum pci_barno test_reg_bar;
 111         size_t alignment;
 112 };
 113
 114 struct pci_endpoint_test_data {
 115         enum pci_barno test_reg_bar;
 116         size_t alignment;
 117         int irq_type;
 118 };
 119
 120 static inline u32 pci_endpoint_test_readl(struct pci_endpoint_test *test,
 121                                           u32 offset)
 122 {
 123         return readl(test->base + offset);
 124 }
 125
 126 static inline void pci_endpoint_test_writel(struct pci_endpoint_test *test,
 127                                             u32 offset, u32 value)
 128 {
 129         writel(value, test->base + offset);
 130 }
 131
 132 static inline u32 pci_endpoint_test_bar_readl(struct pci_endpoint_test *test,
 133                                               int bar, int offset)
 134 {
 135         return readl(test->bar[bar] + offset);
 136 }
 137
 138 static inline void pci_endpoint_test_bar_writel(struct pci_endpoint_test *test,
 139                                                 int bar, u32 offset, u32 value)
 140 {
 141         writel(value, test->bar[bar] + offset);
 142 }
 143
 144 static irqreturn_t pci_endpoint_test_irqhandler(int irq, void *dev_id)
 145 {
 146         struct pci_endpoint_test *test = dev_id;
 147         u32 reg;
 148
 149         reg = pci_endpoint_test_readl(test, PCI_ENDPOINT_TEST_STATUS);
 150         if (reg & STATUS_IRQ_RAISED) {
 151                 test->last_irq = irq;
 152                 complete(&test->irq_raised);
 153                 reg &= ~STATUS_IRQ_RAISED;
 154         }
 155         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_STATUS,
 156                                  reg);
 157
 158         return IRQ_HANDLED;
 159 }
 160
 161 static void pci_endpoint_test_free_irq_vectors(struct pci_endpoint_test *test)
 162 {
 163         struct pci_dev *pdev = test->pdev;
 164
 165         pci_free_irq_vectors(pdev);
 166 }
 167
 168 static bool pci_endpoint_test_alloc_irq_vectors(struct pci_endpoint_test *test,
 169                                                 int type)
 170 {
 171         int irq = -1;
 172         struct pci_dev *pdev = test->pdev;
 173         struct device *dev = &pdev->dev;
 174         bool res = true;
 175
 176         switch (type) {
 177         case IRQ_TYPE_LEGACY:
 178                 irq = pci_alloc_irq_vectors(pdev, 1, 1, PCI_IRQ_LEGACY);
 179                 if (irq < 0)
 180                         dev_err(dev, "Failed to get Legacy interrupt\n");
 181                 break;
 182         case IRQ_TYPE_MSI:
 183                 irq = pci_alloc_irq_vectors(pdev, 1, 32, PCI_IRQ_MSI);
 184                 if (irq < 0)
 185                         dev_err(dev, "Failed to get MSI interrupts\n");
 186                 break;
 187         case IRQ_TYPE_MSIX:
 188                 irq = pci_alloc_irq_vectors(pdev, 1, 2048, PCI_IRQ_MSIX);
 189                 if (irq < 0)
 190                         dev_err(dev, "Failed to get MSI-X interrupts\n");
 191                 break;
 192         default:
 193                 dev_err(dev, "Invalid IRQ type selected\n");
 194         }
 195
 196         if (irq < 0) {
 197                 irq = 0;
 198                 res = false;
 199         }
 200         test->num_irqs = irq;
 201
 202         return res;
 203 }
 204
 205 static void pci_endpoint_test_release_irq(struct pci_endpoint_test *test)
 206 {
 207         int i;
 208         struct pci_dev *pdev = test->pdev;
 209         struct device *dev = &pdev->dev;
 210
 211         for (i = 0; i < test->num_irqs; i++)
 212                 devm_free_irq(dev, pci_irq_vector(pdev, i), test);
 213
 214         test->num_irqs = 0;
 215 }
 216
 217 static bool pci_endpoint_test_request_irq(struct pci_endpoint_test *test)
 218 {
 219         int i;
 220         int err;
 221         struct pci_dev *pdev = test->pdev;
 222         struct device *dev = &pdev->dev;
 223
 224         for (i = 0; i < test->num_irqs; i++) {
 225                 err = devm_request_irq(dev, pci_irq_vector(pdev, i),
 226                                        pci_endpoint_test_irqhandler,
 227                                        IRQF_SHARED, DRV_MODULE_NAME, test);
 228                 if (err)
 229                         goto fail;
 230         }
 231
 232         return true;
 233
 234 fail:
 235         switch (irq_type) {
 236         case IRQ_TYPE_LEGACY:
 237                 dev_err(dev, "Failed to request IRQ %d for Legacy\n",
 238                         pci_irq_vector(pdev, i));
 239                 break;
 240         case IRQ_TYPE_MSI:
 241                 dev_err(dev, "Failed to request IRQ %d for MSI %d\n",
 242                         pci_irq_vector(pdev, i),
 243                         i + 1);
 244                 break;
 245         case IRQ_TYPE_MSIX:
 246                 dev_err(dev, "Failed to request IRQ %d for MSI-X %d\n",
 247                         pci_irq_vector(pdev, i),
 248                         i + 1);
 249                 break;
 250         }
 251
 252         return false;
 253 }
 254
 255 static bool pci_endpoint_test_bar(struct pci_endpoint_test *test,
 256                                   enum pci_barno barno)
 257 {
 258         int j;
 259         u32 val;
 260         int size;
 261         struct pci_dev *pdev = test->pdev;
 262
 263         if (!test->bar[barno])
 264                 return false;
 265
 266         size = pci_resource_len(pdev, barno);
 267
 268         if (barno == test->test_reg_bar)
 269                 size = 0x4;
 270
 271         for (j = 0; j < size; j += 4)
 272                 pci_endpoint_test_bar_writel(test, barno, j, 0xA0A0A0A0);
 273
 274         for (j = 0; j < size; j += 4) {
 275                 val = pci_endpoint_test_bar_readl(test, barno, j);
 276                 if (val != 0xA0A0A0A0)
 277                         return false;
 278         }
 279
 280         return true;
 281 }
 282
 283 static bool pci_endpoint_test_legacy_irq(struct pci_endpoint_test *test)
 284 {
 285         u32 val;
 286
 287         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_IRQ_TYPE,
 288                                  IRQ_TYPE_LEGACY);
 289         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_IRQ_NUMBER, 0);
 290         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_COMMAND,
 291                                  COMMAND_RAISE_LEGACY_IRQ);
 292         val = wait_for_completion_timeout(&test->irq_raised,
 293                                           msecs_to_jiffies(1000));
 294         if (!val)
 295                 return false;
 296
 297         return true;
 298 }
 299
 300 static bool pci_endpoint_test_msi_irq(struct pci_endpoint_test *test,
 301                                        u16 msi_num, bool msix)
 302 {
 303         u32 val;
 304         struct pci_dev *pdev = test->pdev;
 305
 306         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_IRQ_TYPE,
 307                                  msix == false ? IRQ_TYPE_MSI :
 308                                  IRQ_TYPE_MSIX);
 309         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_IRQ_NUMBER, msi_num);
 310         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_COMMAND,
 311                                  msix == false ? COMMAND_RAISE_MSI_IRQ :
 312                                  COMMAND_RAISE_MSIX_IRQ);
 313         val = wait_for_completion_timeout(&test->irq_raised,
 314                                           msecs_to_jiffies(1000));
 315         if (!val)
 316                 return false;
 317
 318         if (pci_irq_vector(pdev, msi_num - 1) == test->last_irq)
 319                 return true;
 320
 321         return false;
 322 }
 323
 324 static bool pci_endpoint_test_copy(struct pci_endpoint_test *test, size_t size)
 325 {
 326         bool ret = false;
 327         void *src_addr;
 328         void *dst_addr;
 329         dma_addr_t src_phys_addr;
 330         dma_addr_t dst_phys_addr;
 331         struct pci_dev *pdev = test->pdev;
 332         struct device *dev = &pdev->dev;
 333         void *orig_src_addr;
 334         dma_addr_t orig_src_phys_addr;
 335         void *orig_dst_addr;
 336         dma_addr_t orig_dst_phys_addr;
 337         size_t offset;
 338         size_t alignment = test->alignment;
 339         u32 src_crc32;
 340         u32 dst_crc32;
 341
 342         if (size > SIZE_MAX - alignment)
 343                 goto err;
 344
 345         if (irq_type < IRQ_TYPE_LEGACY || irq_type > IRQ_TYPE_MSIX) {
 346                 dev_err(dev, "Invalid IRQ type option\n");
 347                 goto err;
 348         }
 349
 350         orig_src_addr = dma_alloc_coherent(dev, size + alignment,
 351                                            &orig_src_phys_addr, GFP_KERNEL);
 352         if (!orig_src_addr) {
 353                 dev_err(dev, "Failed to allocate source buffer\n");
 354                 ret = false;
 355                 goto err;
 356         }
 357
 358         if (alignment && !IS_ALIGNED(orig_src_phys_addr, alignment)) {
 359                 src_phys_addr = PTR_ALIGN(orig_src_phys_addr, alignment);
 360                 offset = src_phys_addr - orig_src_phys_addr;
 361                 src_addr = orig_src_addr + offset;
 362         } else {
 363                 src_phys_addr = orig_src_phys_addr;
 364                 src_addr = orig_src_addr;
 365         }
 366
 367         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_LOWER_SRC_ADDR,
 368                                  lower_32_bits(src_phys_addr));
 369
 370         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_UPPER_SRC_ADDR,
 371                                  upper_32_bits(src_phys_addr));
 372
 373         get_random_bytes(src_addr, size);
 374         src_crc32 = crc32_le(~0, src_addr, size);
 375
 376         orig_dst_addr = dma_alloc_coherent(dev, size + alignment,
 377                                            &orig_dst_phys_addr, GFP_KERNEL);
 378         if (!orig_dst_addr) {
 379                 dev_err(dev, "Failed to allocate destination address\n");
 380                 ret = false;
 381                 goto err_orig_src_addr;
 382         }
 383
 384         if (alignment && !IS_ALIGNED(orig_dst_phys_addr, alignment)) {
 385                 dst_phys_addr = PTR_ALIGN(orig_dst_phys_addr, alignment);
 386                 offset = dst_phys_addr - orig_dst_phys_addr;
 387                 dst_addr = orig_dst_addr + offset;
 388         } else {
 389                 dst_phys_addr = orig_dst_phys_addr;
 390                 dst_addr = orig_dst_addr;
 391         }
 392
 393         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_LOWER_DST_ADDR,
 394                                  lower_32_bits(dst_phys_addr));
 395         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_UPPER_DST_ADDR,
 396                                  upper_32_bits(dst_phys_addr));
 397
 398         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_SIZE,
 399                                  size);
 400
 401         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_IRQ_TYPE, irq_type);
 402         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_IRQ_NUMBER, 1);
 403         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_COMMAND,
 404                                  COMMAND_COPY);
 405
 406         wait_for_completion(&test->irq_raised);
 407
 408         dst_crc32 = crc32_le(~0, dst_addr, size);
 409         if (dst_crc32 == src_crc32)
 410                 ret = true;
 411
 412         dma_free_coherent(dev, size + alignment, orig_dst_addr,
 413                           orig_dst_phys_addr);
 414
 415 err_orig_src_addr:
 416         dma_free_coherent(dev, size + alignment, orig_src_addr,
 417                           orig_src_phys_addr);
 418
 419 err:
 420         return ret;
 421 }
 422
 423 static bool pci_endpoint_test_write(struct pci_endpoint_test *test, size_t size)
 424 {
 425         bool ret = false;
 426         u32 reg;
 427         void *addr;
 428         dma_addr_t phys_addr;
 429         struct pci_dev *pdev = test->pdev;
 430         struct device *dev = &pdev->dev;
 431         void *orig_addr;
 432         dma_addr_t orig_phys_addr;
 433         size_t offset;
 434         size_t alignment = test->alignment;
 435         u32 crc32;
 436
 437         if (size > SIZE_MAX - alignment)
 438                 goto err;
 439
 440         if (irq_type < IRQ_TYPE_LEGACY || irq_type > IRQ_TYPE_MSIX) {
 441                 dev_err(dev, "Invalid IRQ type option\n");
 442                 goto err;
 443         }
 444
 445         orig_addr = dma_alloc_coherent(dev, size + alignment, &orig_phys_addr,
 446                                        GFP_KERNEL);
 447         if (!orig_addr) {
 448                 dev_err(dev, "Failed to allocate address\n");
 449                 ret = false;
 450                 goto err;
 451         }
 452
 453         if (alignment && !IS_ALIGNED(orig_phys_addr, alignment)) {
 454                 phys_addr =  PTR_ALIGN(orig_phys_addr, alignment);
 455                 offset = phys_addr - orig_phys_addr;
 456                 addr = orig_addr + offset;
 457         } else {
 458                 phys_addr = orig_phys_addr;
 459                 addr = orig_addr;
 460         }
 461
 462         get_random_bytes(addr, size);
 463
 464         crc32 = crc32_le(~0, addr, size);
 465         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_CHECKSUM,
 466                                  crc32);
 467
 468         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_LOWER_SRC_ADDR,
 469                                  lower_32_bits(phys_addr));
 470         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_UPPER_SRC_ADDR,
 471                                  upper_32_bits(phys_addr));
 472
 473         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_SIZE, size);
 474
 475         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_IRQ_TYPE, irq_type);
 476         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_IRQ_NUMBER, 1);
 477         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_COMMAND,
 478                                  COMMAND_READ);
 479
 480         wait_for_completion(&test->irq_raised);
 481
 482         reg = pci_endpoint_test_readl(test, PCI_ENDPOINT_TEST_STATUS);
 483         if (reg & STATUS_READ_SUCCESS)
 484                 ret = true;
 485
 486         dma_free_coherent(dev, size + alignment, orig_addr, orig_phys_addr);
 487
 488 err:
 489         return ret;
 490 }
 491
 492 static bool pci_endpoint_test_read(struct pci_endpoint_test *test, size_t size)
 493 {
 494         bool ret = false;
 495         void *addr;
 496         dma_addr_t phys_addr;
 497         struct pci_dev *pdev = test->pdev;
 498         struct device *dev = &pdev->dev;
 499         void *orig_addr;
 500         dma_addr_t orig_phys_addr;
 501         size_t offset;
 502         size_t alignment = test->alignment;
 503         u32 crc32;
 504
 505         if (size > SIZE_MAX - alignment)
 506                 goto err;
 507
 508         if (irq_type < IRQ_TYPE_LEGACY || irq_type > IRQ_TYPE_MSIX) {
 509                 dev_err(dev, "Invalid IRQ type option\n");
 510                 goto err;
 511         }
 512
 513         orig_addr = dma_alloc_coherent(dev, size + alignment, &orig_phys_addr,
 514                                        GFP_KERNEL);
 515         if (!orig_addr) {
 516                 dev_err(dev, "Failed to allocate destination address\n");
 517                 ret = false;
 518                 goto err;
 519         }
 520
 521         if (alignment && !IS_ALIGNED(orig_phys_addr, alignment)) {
 522                 phys_addr = PTR_ALIGN(orig_phys_addr, alignment);
 523                 offset = phys_addr - orig_phys_addr;
 524                 addr = orig_addr + offset;
 525         } else {
 526                 phys_addr = orig_phys_addr;
 527                 addr = orig_addr;
 528         }
 529
 530         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_LOWER_DST_ADDR,
 531                                  lower_32_bits(phys_addr));
 532         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_UPPER_DST_ADDR,
 533                                  upper_32_bits(phys_addr));
 534
 535         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_SIZE, size);
 536
 537         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_IRQ_TYPE, irq_type);
 538         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_IRQ_NUMBER, 1);
 539         pci_endpoint_test_writel(test, PCI_ENDPOINT_TEST_COMMAND,
 540                                  COMMAND_WRITE);
 541
 542         wait_for_completion(&test->irq_raised);
 543
 544         crc32 = crc32_le(~0, addr, size);
 545         if (crc32 == pci_endpoint_test_readl(test, PCI_ENDPOINT_TEST_CHECKSUM))
 546                 ret = true;
 547
 548         dma_free_coherent(dev, size + alignment, orig_addr, orig_phys_addr);
 549 err:
 550         return ret;
 551 }
 552
 553 static bool pci_endpoint_test_set_irq(struct pci_endpoint_test *test,
 554                                       int req_irq_type)
 555 {
 556         struct pci_dev *pdev = test->pdev;
 557         struct device *dev = &pdev->dev;
 558
 559         if (req_irq_type < IRQ_TYPE_LEGACY || req_irq_type > IRQ_TYPE_MSIX) {
 560                 dev_err(dev, "Invalid IRQ type option\n");
 561                 return false;
 562         }
 563
 564         if (irq_type == req_irq_type)
 565                 return true;
 566
 567         pci_endpoint_test_release_irq(test);
 568         pci_endpoint_test_free_irq_vectors(test);
 569
 570         if (!pci_endpoint_test_alloc_irq_vectors(test, req_irq_type))
 571                 goto err;
 572
 573         if (!pci_endpoint_test_request_irq(test))
 574                 goto err;
 575
 576         irq_type = req_irq_type;
 577         return true;
 578
 579 err:
 580         pci_endpoint_test_free_irq_vectors(test);
 581         irq_type = IRQ_TYPE_UNDEFINED;
 582         return false;
 583 }
 584
 585 static long pci_endpoint_test_ioctl(struct file *file, unsigned int cmd,
 586                                     unsigned long arg)
 587 {
 588         int ret = -EINVAL;
 589         enum pci_barno bar;
 590         struct pci_endpoint_test *test = to_endpoint_test(file->private_data);
 591
 592         mutex_lock(&test->mutex);
 593         switch (cmd) {
 594         case PCITEST_BAR:
 595                 bar = arg;
 596                 if (bar < 0 || bar > 5)
 597                         goto ret;
 598                 ret = pci_endpoint_test_bar(test, bar);
 599                 break;
 600         case PCITEST_LEGACY_IRQ:
 601                 ret = pci_endpoint_test_legacy_irq(test);
 602                 break;
 603         case PCITEST_MSI:
 604         case PCITEST_MSIX:
 605                 ret = pci_endpoint_test_msi_irq(test, arg, cmd == PCITEST_MSIX);
 606                 break;
 607         case PCITEST_WRITE:
 608                 ret = pci_endpoint_test_write(test, arg);
 609                 break;
 610         case PCITEST_READ:
 611                 ret = pci_endpoint_test_read(test, arg);
 612                 break;
 613         case PCITEST_COPY:
 614                 ret = pci_endpoint_test_copy(test, arg);
 615                 break;
 616         case PCITEST_SET_IRQTYPE:
 617                 ret = pci_endpoint_test_set_irq(test, arg);
 618                 break;
 619         case PCITEST_GET_IRQTYPE:
 620                 ret = irq_type;
 621                 break;
 622         }
 623
 624 ret:
 625         mutex_unlock(&test->mutex);
 626         return ret;
 627 }
 628
 629 static const struct file_operations pci_endpoint_test_fops = {
 630         .owner = THIS_MODULE,
 631         .unlocked_ioctl = pci_endpoint_test_ioctl,
 632 };
 633
 634 static int pci_endpoint_test_probe(struct pci_dev *pdev,
 635                                    const struct pci_device_id *ent)
 636 {
 637         int err;
 638         int id;
 639         char name[20];
 640         enum pci_barno bar;
 641         void __iomem *base;
 642         struct device *dev = &pdev->dev;
 643         struct pci_endpoint_test *test;
 644         struct pci_endpoint_test_data *data;
 645         enum pci_barno test_reg_bar = BAR_0;
 646         struct miscdevice *misc_device;
 647
 648         if (pci_is_bridge(pdev))
 649                 return -ENODEV;
 650
 651         test = devm_kzalloc(dev, sizeof(*test), GFP_KERNEL);
 652         if (!test)
 653                 return -ENOMEM;
 654
 655         test->test_reg_bar = 0;
 656         test->alignment = 0;
 657         test->pdev = pdev;
 658
 659         if (no_msi)
 660                 irq_type = IRQ_TYPE_LEGACY;
 661
 662         data = (struct pci_endpoint_test_data *)ent->driver_data;
 663         if (data) {
 664                 test_reg_bar = data->test_reg_bar;
 665                 test->alignment = data->alignment;
 666                 irq_type = data->irq_type;
 667         }
 668
 669         init_completion(&test->irq_raised);
 670         mutex_init(&test->mutex);
 671
 672         err = pci_enable_device(pdev);
 673         if (err) {
 674                 dev_err(dev, "Cannot enable PCI device\n");
 675                 return err;
 676         }
 677
 678         err = pci_request_regions(pdev, DRV_MODULE_NAME);
 679         if (err) {
 680                 dev_err(dev, "Cannot obtain PCI resources\n");
 681                 goto err_disable_pdev;
 682         }
 683
 684         pci_set_master(pdev);
 685
 686         if (!pci_endpoint_test_alloc_irq_vectors(test, irq_type))
 687                 goto err_disable_irq;
 688
 689         if (!pci_endpoint_test_request_irq(test))
 690                 goto err_disable_irq;
 691
 692         for (bar = BAR_0; bar <= BAR_5; bar++) {
 693                 if (pci_resource_flags(pdev, bar) & IORESOURCE_MEM) {
 694                         base = pci_ioremap_bar(pdev, bar);
 695                         if (!base) {
 696                                 dev_err(dev, "Failed to read BAR%d\n", bar);
 697                                 WARN_ON(bar == test_reg_bar);
 698                         }
 699                         test->bar[bar] = base;
 700                 }
 701         }
 702
 703         test->base = test->bar[test_reg_bar];
 704         if (!test->base) {
 705                 err = -ENOMEM;
 706                 dev_err(dev, "Cannot perform PCI test without BAR%d\n",
 707                         test_reg_bar);
 708                 goto err_iounmap;
 709         }
 710
 711         pci_set_drvdata(pdev, test);
 712
 713         id = ida_simple_get(&pci_endpoint_test_ida, 0, 0, GFP_KERNEL);
 714         if (id < 0) {
 715                 err = id;
 716                 dev_err(dev, "Unable to get id\n");
 717                 goto err_iounmap;
 718         }
 719
 720         snprintf(name, sizeof(name), DRV_MODULE_NAME ".%d", id);
 721         misc_device = &test->miscdev;
 722         misc_device->minor = MISC_DYNAMIC_MINOR;
 723         misc_device->name = kstrdup(name, GFP_KERNEL);
 724         if (!misc_device->name) {
 725                 err = -ENOMEM;
 726                 goto err_ida_remove;
 727         }
 728         misc_device->fops = &pci_endpoint_test_fops,
 729
 730         err = misc_register(misc_device);
 731         if (err) {
 732                 dev_err(dev, "Failed to register device\n");
 733                 goto err_kfree_name;
 734         }
 735
 736         return 0;
 737
 738 err_kfree_name:
 739         kfree(misc_device->name);
 740
 741 err_ida_remove:
 742         ida_simple_remove(&pci_endpoint_test_ida, id);
 743
 744 err_iounmap:
 745         for (bar = BAR_0; bar <= BAR_5; bar++) {
 746                 if (test->bar[bar])
 747                         pci_iounmap(pdev, test->bar[bar]);
 748         }
 749         pci_endpoint_test_release_irq(test);
 750
 751 err_disable_irq:
 752         pci_endpoint_test_free_irq_vectors(test);
 753         pci_release_regions(pdev);
 754
 755 err_disable_pdev:
 756         pci_disable_device(pdev);
 757
 758         return err;
 759 }
 760
 761 static void pci_endpoint_test_remove(struct pci_dev *pdev)
 762 {
 763         int id;
 764         enum pci_barno bar;
 765         struct pci_endpoint_test *test = pci_get_drvdata(pdev);
 766         struct miscdevice *misc_device = &test->miscdev;
 767
 768         if (sscanf(misc_device->name, DRV_MODULE_NAME ".%d", &id) != 1)
 769                 return;
 770         if (id < 0)
 771                 return;
 772
 773         misc_deregister(&test->miscdev);
 774         kfree(misc_device->name);
 775         ida_simple_remove(&pci_endpoint_test_ida, id);
 776         for (bar = BAR_0; bar <= BAR_5; bar++) {
 777                 if (test->bar[bar])
 778                         pci_iounmap(pdev, test->bar[bar]);
 779         }
 780
 781         pci_endpoint_test_release_irq(test);
 782         pci_endpoint_test_free_irq_vectors(test);
 783
 784         pci_release_regions(pdev);
 785         pci_disable_device(pdev);
 786 }
 787
 788 static const struct pci_device_id pci_endpoint_test_tbl[] = {
 789         { PCI_DEVICE(PCI_VENDOR_ID_TI, PCI_DEVICE_ID_TI_DRA74x) },
 790         { PCI_DEVICE(PCI_VENDOR_ID_TI, PCI_DEVICE_ID_TI_DRA72x) },
 791         { PCI_DEVICE(PCI_VENDOR_ID_FREESCALE, 0x81c0) },
 792         { PCI_DEVICE(PCI_VENDOR_ID_SYNOPSYS, 0xedda) },
 793         { }
 794 };
 795 MODULE_DEVICE_TABLE(pci, pci_endpoint_test_tbl);
 796
 797 static struct pci_driver pci_endpoint_test_driver = {
 798         .name           = DRV_MODULE_NAME,
 799         .id_table       = pci_endpoint_test_tbl,
 800         .probe          = pci_endpoint_test_probe,
 801         .remove         = pci_endpoint_test_remove,
 802 };
 803 module_pci_driver(pci_endpoint_test_driver);
 804
 805 MODULE_DESCRIPTION("PCI ENDPOINT TEST HOST DRIVER");
 806 MODULE_AUTHOR("Kishon Vijay Abraham I <kishon@ti.com>");
 807 MODULE_LICENSE("GPL v2");