/**
   * Userspace PCI Endpoint Test Module
   *
   * Copyright (C) 2017 Texas Instruments
   * Author: Kishon Vijay Abraham I <kishon@ti.com>
   *
   * This program is free software: you can redistribute it and/or modify
   * it under the terms of the GNU General Public License version 2 of
   * the License as published by the Free Software Foundation.
   *
   * This program is distributed in the hope that it will be useful,
   * but WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   * GNU General Public License for more details.
   *
   * You should have received a copy of the GNU General Public License
   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
   */
  
  #include <errno.h>
  #include <fcntl.h>
  #include <stdbool.h>
  #include <stdio.h>
  #include <stdlib.h>
  #include <sys/ioctl.h>
  #include <time.h>
  #include <unistd.h>
  
  #include <linux/pcitest.h>
  
  #define BILLION 1E9
  
  static char *result[] = { "NOT OKAY", "OKAY" };
  
  struct pci_test {
  	char		*device;
  	char		barnum;
  	bool		legacyirq;
  	unsigned int	msinum;
  	bool		read;
  	bool		write;
  	bool		copy;
  	unsigned long	size;
  };
  
  static int run_test(struct pci_test *test)
  {
  	long ret;
  	int fd;
  	struct timespec start, end;
  	double time;
  
  	fd = open(test->device, O_RDWR);
  	if (fd < 0) {
  		perror("can't open PCI Endpoint Test device");
  		return fd;
  	}
  
  	if (test->barnum >= 0 && test->barnum <= 5) {
  		ret = ioctl(fd, PCITEST_BAR, test->barnum);
  		fprintf(stdout, "BAR%d:\t\t", test->barnum);
  		if (ret < 0)
  			fprintf(stdout, "TEST FAILED
  ");
  		else
  			fprintf(stdout, "%s
  ", result[ret]);
  	}
  
  	if (test->legacyirq) {
  		ret = ioctl(fd, PCITEST_LEGACY_IRQ, 0);
  		fprintf(stdout, "LEGACY IRQ:\t");
  		if (ret < 0)
  			fprintf(stdout, "TEST FAILED
  ");
  		else
  			fprintf(stdout, "%s
  ", result[ret]);
  	}
  
  	if (test->msinum > 0 && test->msinum <= 32) {
  		ret = ioctl(fd, PCITEST_MSI, test->msinum);
  		fprintf(stdout, "MSI%d:\t\t", test->msinum);
  		if (ret < 0)
  			fprintf(stdout, "TEST FAILED
  ");
  		else
  			fprintf(stdout, "%s
  ", result[ret]);
  	}
  
  	if (test->write) {
  		ret = ioctl(fd, PCITEST_WRITE, test->size);
  		fprintf(stdout, "WRITE (%7ld bytes):\t\t", test->size);
  		if (ret < 0)
  			fprintf(stdout, "TEST FAILED
  ");
  		else
  			fprintf(stdout, "%s
  ", result[ret]);
  	}
  
  	if (test->read) {
  		ret = ioctl(fd, PCITEST_READ, test->size);
  		fprintf(stdout, "READ (%7ld bytes):\t\t", test->size);
  		if (ret < 0)
  			fprintf(stdout, "TEST FAILED
  ");
  		else
  			fprintf(stdout, "%s
  ", result[ret]);
  	}
  
  	if (test->copy) {
  		ret = ioctl(fd, PCITEST_COPY, test->size);
  		fprintf(stdout, "COPY (%7ld bytes):\t\t", test->size);
  		if (ret < 0)
  			fprintf(stdout, "TEST FAILED
  ");
  		else
  			fprintf(stdout, "%s
  ", result[ret]);
  	}
  
  	fflush(stdout);
  }
  
  int main(int argc, char **argv)
  {
  	int c;
  	struct pci_test *test;
  
  	test = calloc(1, sizeof(*test));
  	if (!test) {
  		perror("Fail to allocate memory for pci_test
  ");
  		return -ENOMEM;
  	}
  
  	/* since '0' is a valid BAR number, initialize it to -1 */
  	test->barnum = -1;
  
  	/* set default size as 100KB */
  	test->size = 0x19000;
  
  	/* set default endpoint device */
  	test->device = "/dev/pci-endpoint-test.0";
  
  	while ((c = getopt(argc, argv, "D:b:m:lrwcs:")) != EOF)
  	switch (c) {
  	case 'D':
  		test->device = optarg;
  		continue;
  	case 'b':
  		test->barnum = atoi(optarg);
  		if (test->barnum < 0 || test->barnum > 5)
  			goto usage;
  		continue;
  	case 'l':
  		test->legacyirq = true;
  		continue;
  	case 'm':
  		test->msinum = atoi(optarg);
  		if (test->msinum < 1 || test->msinum > 32)
  			goto usage;
  		continue;
  	case 'r':
  		test->read = true;
  		continue;
  	case 'w':
  		test->write = true;
  		continue;
  	case 'c':
  		test->copy = true;
  		continue;
  	case 's':
  		test->size = strtoul(optarg, NULL, 0);
  		continue;
  	case '?':
  	case 'h':
  	default:
  usage:
  		fprintf(stderr,
  			"usage: %s [options]
  "
  			"Options:
  "
  			"\t-D <dev>		PCI endpoint test device {default: /dev/pci-endpoint-test.0}
  "
  			"\t-b <bar num>		BAR test (bar number between 0..5)
  "
  			"\t-m <msi num>		MSI test (msi number between 1..32)
  "

  			"\t-l			Legacy IRQ test
  "

  			"\t-r			Read buffer test
  "
  			"\t-w			Write buffer test
  "
  			"\t-c			Copy buffer test
  "
  			"\t-s <size>		Size of buffer {default: 100KB}
  ",
  			argv[0]);
  		return -EINVAL;
  	}
  
  	run_test(test);
  	return 0;
  }