tools/pci/pcitest.c

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