tools: PCI: Add a userspace tool to test PCI endpoint

/**

 * 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\n");

		else

			fprintf(stdout, "%s\n", result[ret]);

	}

	if (test->legacyirq) {

		ret = ioctl(fd, PCITEST_LEGACY_IRQ, 0);

		fprintf(stdout, "LEGACY IRQ:\t");

		if (ret < 0)

			fprintf(stdout, "TEST FAILED\n");

		else

			fprintf(stdout, "%s\n", 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\n");

		else

			fprintf(stdout, "%s\n", 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\n");

		else

			fprintf(stdout, "%s\n", 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\n");

		else

			fprintf(stdout, "%s\n", 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\n");

		else

			fprintf(stdout, "%s\n", 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\n");

		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]\n"

			"Options:\n"

			"\t-D <dev>		PCI endpoint test device {default: /dev/pci-endpoint-test.0}\n"

			"\t-b <bar num>		BAR test (bar number between 0..5)\n"

			"\t-m <msi num>		MSI test (msi number between 1..32)\n"

			"\t-r			Read buffer test\n"

			"\t-w			Write buffer test\n"

			"\t-c			Copy buffer test\n"

			"\t-s <size>		Size of buffer {default: 100KB}\n",

			argv[0]);

		return -EINVAL;

	}

	run_test(test);

	return 0;

}