1
2
3
4
5
6
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
21static char *result[] = { "NOT OKAY", "OKAY" };
22static char *irq[] = { "LEGACY", "MSI", "MSI-X" };
23
24struct 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
41static 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;
155}
156
157int 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
169 test->barnum = -1;
170
171
172 test->size = 0x19000;
173
174
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:
229usage:
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
