Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 9006a7b3 authored by Frode Isaksen's avatar Frode Isaksen Committed by Mark Brown
Browse files

spi: spidev_test: add option to continuously transfer data 



Add option to send+recv <size> bytes with <iter> iterations
and show the transfer rate every 5 seconds.
Example:
rate: tx 4235.2kbps, rx 4235.2kbps
total: tx 4882.8KB, rx 4882.8KB

Signed-off-by: default avatarFrode Isaksen <fisaksen@baylibre.com>
Signed-off-by: default avatarMark Brown <broonie@kernel.org>
parent fafd6794

tools/spi/spidev_test.c

+88 −5
Original line number Original line Diff line number Diff line
@@ -18,6 +18,7 @@ 
#include <string.h>

#include <string.h>

#include <getopt.h>

#include <getopt.h>

#include <fcntl.h>

#include <fcntl.h>

#include <time.h>

#include <sys/ioctl.h>

#include <sys/ioctl.h>

#include <linux/ioctl.h>

#include <linux/ioctl.h>

#include <sys/stat.h>

#include <sys/stat.h>
@@ -40,6 +41,9 @@ static char *output_file; 
static uint32_t speed = 500000;

static uint32_t speed = 500000;

static uint16_t delay;

static uint16_t delay;

static int verbose;

static int verbose;

static int transfer_size;

static int iterations;

static int interval = 5; /* interval in seconds for showing transfer rate */





uint8_t default_tx[] = {

uint8_t default_tx[] = {

	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,

	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
@@ -156,13 +160,13 @@ static void transfer(int fd, uint8_t const *tx, uint8_t const *rx, size_t len) 
		close(out_fd);

		close(out_fd);

	}

	}





	if (verbose || !output_file)

	if (verbose)

		hex_dump(rx, len, 32, "RX");

		hex_dump(rx, len, 32, "RX");

}

}





static void print_usage(const char *prog)

static void print_usage(const char *prog)

{

{

	printf("Usage: %s [-DsbdlHOLC3]\n", prog);

	printf("Usage: %s [-DsbdlHOLC3vpNR24SI]\n", prog);

	puts("  -D --device   device to use (default /dev/spidev1.1)\n"

	puts("  -D --device   device to use (default /dev/spidev1.1)\n"

	     "  -s --speed    max speed (Hz)\n"

	     "  -s --speed    max speed (Hz)\n"

	     "  -d --delay    delay (usec)\n"

	     "  -d --delay    delay (usec)\n"
@@ -180,7 +184,9 @@ static void print_usage(const char *prog) 
	     "  -N --no-cs    no chip select\n"

	     "  -N --no-cs    no chip select\n"

	     "  -R --ready    slave pulls low to pause\n"

	     "  -R --ready    slave pulls low to pause\n"

	     "  -2 --dual     dual transfer\n"

	     "  -2 --dual     dual transfer\n"

	     "  -4 --quad     quad transfer\n");

	     "  -4 --quad     quad transfer\n"

	     "  -S --size     transfer size\n"

	     "  -I --iter     iterations\n");

	exit(1);

	exit(1);

}

}
@@ -205,11 +211,13 @@ static void parse_opts(int argc, char *argv[]) 
			{ "dual",    0, 0, '2' },

			{ "dual",    0, 0, '2' },

			{ "verbose", 0, 0, 'v' },

			{ "verbose", 0, 0, 'v' },

			{ "quad",    0, 0, '4' },

			{ "quad",    0, 0, '4' },

			{ "size",    1, 0, 'S' },

			{ "iter",    1, 0, 'I' },

			{ NULL, 0, 0, 0 },

			{ NULL, 0, 0, 0 },

		};

		};

		int c;

		int c;





		c = getopt_long(argc, argv, "D:s:d:b:i:o:lHOLC3NR24p:v",

		c = getopt_long(argc, argv, "D:s:d:b:i:o:lHOLC3NR24p:vS:I:",

				lopts, NULL);

				lopts, NULL);





		if (c == -1)

		if (c == -1)
@@ -270,6 +278,12 @@ static void parse_opts(int argc, char *argv[]) 
		case '4':

		case '4':

			mode |= SPI_TX_QUAD;

			mode |= SPI_TX_QUAD;

			break;

			break;

		case 'S':

			transfer_size = atoi(optarg);

			break;

		case 'I':

			iterations = atoi(optarg);

			break;

		default:

		default:

			print_usage(argv[0]);

			print_usage(argv[0]);

			break;

			break;
@@ -336,6 +350,57 @@ static void transfer_file(int fd, char *filename) 
	close(tx_fd);

	close(tx_fd);

}

}





static uint64_t _read_count;

static uint64_t _write_count;



static void show_transfer_rate(void)

{

	static uint64_t prev_read_count, prev_write_count;

	double rx_rate, tx_rate;



	rx_rate = ((_read_count - prev_read_count) * 8) / (interval*1000.0);

	tx_rate = ((_write_count - prev_write_count) * 8) / (interval*1000.0);



	printf("rate: tx %.1fkbps, rx %.1fkbps\n", rx_rate, tx_rate);



	prev_read_count = _read_count;

	prev_write_count = _write_count;

}



static void transfer_buf(int fd, int len)

{

	uint8_t *tx;

	uint8_t *rx;

	int i;



	tx = malloc(len);

	if (!tx)

		pabort("can't allocate tx buffer");

	for (i = 0; i < len; i++)

		tx[i] = random();



	rx = malloc(len);

	if (!rx)

		pabort("can't allocate rx buffer");



	transfer(fd, tx, rx, len);



	_write_count += len;

	_read_count += len;



	if (mode & SPI_LOOP) {

		if (memcmp(tx, rx, len)) {

			fprintf(stderr, "transfer error !\n");

			hex_dump(tx, len, 32, "TX");

			hex_dump(rx, len, 32, "RX");

			exit(1);

		}

	}



	free(rx);

	free(tx);

}



int main(int argc, char *argv[])

int main(int argc, char *argv[])

{

{

	int ret = 0;

	int ret = 0;
@@ -391,7 +456,25 @@ int main(int argc, char *argv[]) 
		transfer_escaped_string(fd, input_tx);

		transfer_escaped_string(fd, input_tx);

	else if (input_file)

	else if (input_file)

		transfer_file(fd, input_file);

		transfer_file(fd, input_file);

	else

	else if (transfer_size) {

		struct timespec last_stat;



		clock_gettime(CLOCK_MONOTONIC, &last_stat);



		while (iterations-- > 0) {

			struct timespec current;



			transfer_buf(fd, transfer_size);



			clock_gettime(CLOCK_MONOTONIC, &current);

			if (current.tv_sec - last_stat.tv_sec > interval) {

				show_transfer_rate();

				last_stat = current;

			}

		}

		printf("total: tx %.1fKB, rx %.1fKB\n",

		       _write_count/1024.0, _read_count/1024.0);

	} else

		transfer(fd, default_tx, default_rx, sizeof(default_tx));

		transfer(fd, default_tx, default_rx, sizeof(default_tx));





	close(fd);

	close(fd);