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