1 // SPDX-License-Identifier: GPL-2.0-only 2 /** 3 * Userspace PCI Endpoint Test Module 4 * 5 * Copyright (C) 2017 Texas Instruments 6 * Author: Kishon Vijay Abraham I <kishon@ti.com> 7 */ 8 9 #include <errno.h> 10 #include <fcntl.h> 11 #include <stdbool.h> 12 #include <stdio.h> 13 #include <stdlib.h> 14 #include <sys/ioctl.h> 15 #include <unistd.h> 16 17 #include <linux/pcitest.h> 18 19 #define BILLION 1E9 20 21 static char *result[] = { "NOT OKAY", "OKAY" }; 22 static char *irq[] = { "LEGACY", "MSI", "MSI-X" }; 23 24 struct pci_test { 25 char *device; 26 char barnum; 27 bool legacyirq; 28 unsigned int msinum; 29 unsigned int msixnum; 30 int irqtype; 31 bool set_irqtype; 32 bool get_irqtype; 33 bool clear_irq; 34 bool read; 35 bool write; 36 bool copy; 37 unsigned long size; 38 bool use_dma; 39 }; 40 41 static int run_test(struct pci_test *test) 42 { 43 struct pci_endpoint_test_xfer_param param; 44 int ret = -EINVAL; 45 int fd; 46 47 fd = open(test->device, O_RDWR); 48 if (fd < 0) { 49 perror("can't open PCI Endpoint Test device"); 50 return -ENODEV; 51 } 52 53 if (test->barnum >= 0 && test->barnum <= 5) { 54 ret = ioctl(fd, PCITEST_BAR, test->barnum); 55 fprintf(stdout, "BAR%d:\t\t", test->barnum); 56 if (ret < 0) 57 fprintf(stdout, "TEST FAILED\n"); 58 else 59 fprintf(stdout, "%s\n", result[ret]); 60 } 61 62 if (test->set_irqtype) { 63 ret = ioctl(fd, PCITEST_SET_IRQTYPE, test->irqtype); 64 fprintf(stdout, "SET IRQ TYPE TO %s:\t\t", irq[test->irqtype]); 65 if (ret < 0) 66 fprintf(stdout, "FAILED\n"); 67 else 68 fprintf(stdout, "%s\n", result[ret]); 69 } 70 71 if (test->get_irqtype) { 72 ret = ioctl(fd, PCITEST_GET_IRQTYPE); 73 fprintf(stdout, "GET IRQ TYPE:\t\t"); 74 if (ret < 0) 75 fprintf(stdout, "FAILED\n"); 76 else 77 fprintf(stdout, "%s\n", irq[ret]); 78 } 79 80 if (test->clear_irq) { 81 ret = ioctl(fd, PCITEST_CLEAR_IRQ); 82 fprintf(stdout, "CLEAR IRQ:\t\t"); 83 if (ret < 0) 84 fprintf(stdout, "FAILED\n"); 85 else 86 fprintf(stdout, "%s\n", result[ret]); 87 } 88 89 if (test->legacyirq) { 90 ret = ioctl(fd, PCITEST_LEGACY_IRQ, 0); 91 fprintf(stdout, "LEGACY IRQ:\t"); 92 if (ret < 0) 93 fprintf(stdout, "TEST FAILED\n"); 94 else 95 fprintf(stdout, "%s\n", result[ret]); 96 } 97 98 if (test->msinum > 0 && test->msinum <= 32) { 99 ret = ioctl(fd, PCITEST_MSI, test->msinum); 100 fprintf(stdout, "MSI%d:\t\t", test->msinum); 101 if (ret < 0) 102 fprintf(stdout, "TEST FAILED\n"); 103 else 104 fprintf(stdout, "%s\n", result[ret]); 105 } 106 107 if (test->msixnum > 0 && test->msixnum <= 2048) { 108 ret = ioctl(fd, PCITEST_MSIX, test->msixnum); 109 fprintf(stdout, "MSI-X%d:\t\t", test->msixnum); 110 if (ret < 0) 111 fprintf(stdout, "TEST FAILED\n"); 112 else 113 fprintf(stdout, "%s\n", result[ret]); 114 } 115 116 if (test->write) { 117 param.size = test->size; 118 if (test->use_dma) 119 param.flags = PCITEST_FLAGS_USE_DMA; 120 ret = ioctl(fd, PCITEST_WRITE, ¶m); 121 fprintf(stdout, "WRITE (%7ld bytes):\t\t", test->size); 122 if (ret < 0) 123 fprintf(stdout, "TEST FAILED\n"); 124 else 125 fprintf(stdout, "%s\n", result[ret]); 126 } 127 128 if (test->read) { 129 param.size = test->size; 130 if (test->use_dma) 131 param.flags = PCITEST_FLAGS_USE_DMA; 132 ret = ioctl(fd, PCITEST_READ, ¶m); 133 fprintf(stdout, "READ (%7ld bytes):\t\t", test->size); 134 if (ret < 0) 135 fprintf(stdout, "TEST FAILED\n"); 136 else 137 fprintf(stdout, "%s\n", result[ret]); 138 } 139 140 if (test->copy) { 141 param.size = test->size; 142 if (test->use_dma) 143 param.flags = PCITEST_FLAGS_USE_DMA; 144 ret = ioctl(fd, PCITEST_COPY, ¶m); 145 fprintf(stdout, "COPY (%7ld bytes):\t\t", test->size); 146 if (ret < 0) 147 fprintf(stdout, "TEST FAILED\n"); 148 else 149 fprintf(stdout, "%s\n", result[ret]); 150 } 151 152 fflush(stdout); 153 close(fd); 154 return (ret < 0) ? ret : 1 - ret; /* return 0 if test succeeded */ 155 } 156 157 int main(int argc, char **argv) 158 { 159 int c; 160 struct pci_test *test; 161 162 test = calloc(1, sizeof(*test)); 163 if (!test) { 164 perror("Fail to allocate memory for pci_test\n"); 165 return -ENOMEM; 166 } 167 168 /* since '0' is a valid BAR number, initialize it to -1 */ 169 test->barnum = -1; 170 171 /* set default size as 100KB */ 172 test->size = 0x19000; 173 174 /* set default endpoint device */ 175 test->device = "/dev/pci-endpoint-test.0"; 176 177 while ((c = getopt(argc, argv, "D:b:m:x:i:deIlhrwcs:")) != EOF) 178 switch (c) { 179 case 'D': 180 test->device = optarg; 181 continue; 182 case 'b': 183 test->barnum = atoi(optarg); 184 if (test->barnum < 0 || test->barnum > 5) 185 goto usage; 186 continue; 187 case 'l': 188 test->legacyirq = true; 189 continue; 190 case 'm': 191 test->msinum = atoi(optarg); 192 if (test->msinum < 1 || test->msinum > 32) 193 goto usage; 194 continue; 195 case 'x': 196 test->msixnum = atoi(optarg); 197 if (test->msixnum < 1 || test->msixnum > 2048) 198 goto usage; 199 continue; 200 case 'i': 201 test->irqtype = atoi(optarg); 202 if (test->irqtype < 0 || test->irqtype > 2) 203 goto usage; 204 test->set_irqtype = true; 205 continue; 206 case 'I': 207 test->get_irqtype = true; 208 continue; 209 case 'r': 210 test->read = true; 211 continue; 212 case 'w': 213 test->write = true; 214 continue; 215 case 'c': 216 test->copy = true; 217 continue; 218 case 'e': 219 test->clear_irq = true; 220 continue; 221 case 's': 222 test->size = strtoul(optarg, NULL, 0); 223 continue; 224 case 'd': 225 test->use_dma = true; 226 continue; 227 case 'h': 228 default: 229 usage: 230 fprintf(stderr, 231 "usage: %s [options]\n" 232 "Options:\n" 233 "\t-D <dev> PCI endpoint test device {default: /dev/pci-endpoint-test.0}\n" 234 "\t-b <bar num> BAR test (bar number between 0..5)\n" 235 "\t-m <msi num> MSI test (msi number between 1..32)\n" 236 "\t-x <msix num> \tMSI-X test (msix number between 1..2048)\n" 237 "\t-i <irq type> \tSet IRQ type (0 - Legacy, 1 - MSI, 2 - MSI-X)\n" 238 "\t-e Clear IRQ\n" 239 "\t-I Get current IRQ type configured\n" 240 "\t-d Use DMA\n" 241 "\t-l Legacy IRQ test\n" 242 "\t-r Read buffer test\n" 243 "\t-w Write buffer test\n" 244 "\t-c Copy buffer test\n" 245 "\t-s <size> Size of buffer {default: 100KB}\n" 246 "\t-h Print this help message\n", 247 argv[0]); 248 return -EINVAL; 249 } 250 251 return run_test(test); 252 } 253