[mirror_ubuntu-jammy-kernel.git] / tools / spi / spidev_test.c

// SPDX-License-Identifier: GPL-2.0-only
/*
 * SPI testing utility (using spidev driver)
 *
 * Copyright (c) 2007  MontaVista Software, Inc.
 * Copyright (c) 2007  Anton Vorontsov <avorontsov@ru.mvista.com>
 *
 * Cross-compile with cross-gcc -I/path/to/cross-kernel/include
 */

#include <stdint.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <getopt.h>
#include <fcntl.h>
#include <time.h>
#include <sys/ioctl.h>
#include <linux/ioctl.h>
#include <sys/stat.h>
#include <linux/types.h>
#include <linux/spi/spidev.h>

#define ARRAY_SIZE(a) (sizeof(a) / sizeof((a)[0]))

static void pabort(const char *s)
{
	if (errno != 0)
		perror(s);
	else
		printf("%s\n", s);

	abort();
}

static const char *device = "/dev/spidev1.1";
static uint32_t mode;
static uint8_t bits = 8;
static char *input_file;
static char *output_file;
static uint32_t speed = 500000;
static uint16_t delay;
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[] = {
	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
	0x40, 0x00, 0x00, 0x00, 0x00, 0x95,
	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
	0xF0, 0x0D,
};

uint8_t default_rx[ARRAY_SIZE(default_tx)] = {0, };
char *input_tx;

static void hex_dump(const void *src, size_t length, size_t line_size,
		     char *prefix)
{
	int i = 0;
	const unsigned char *address = src;
	const unsigned char *line = address;
	unsigned char c;

	printf("%s | ", prefix);
	while (length-- > 0) {
		printf("%02X ", *address++);
		if (!(++i % line_size) || (length == 0 && i % line_size)) {
			if (length == 0) {
				while (i++ % line_size)
					printf("__ ");
			}
			printf(" |");
			while (line < address) {
				c = *line++;
				printf("%c", (c < 32 || c > 126) ? '.' : c);
			}
			printf("|\n");
			if (length > 0)
				printf("%s | ", prefix);
		}
	}
}

/*
 *  Unescape - process hexadecimal escape character
 *      converts shell input "\x23" -> 0x23
 */
static int unescape(char *_dst, char *_src, size_t len)
{
	int ret = 0;
	int match;
	char *src = _src;
	char *dst = _dst;
	unsigned int ch;

	while (*src) {
		if (*src == '\\' && *(src+1) == 'x') {
			match = sscanf(src + 2, "%2x", &ch);
			if (!match)
				pabort("malformed input string");

			src += 4;
			*dst++ = (unsigned char)ch;
		} else {
			*dst++ = *src++;
		}
		ret++;
	}
	return ret;
}

static void transfer(int fd, uint8_t const *tx, uint8_t const *rx, size_t len)
{
	int ret;
	int out_fd;
	struct spi_ioc_transfer tr = {
		.tx_buf = (unsigned long)tx,
		.rx_buf = (unsigned long)rx,
		.len = len,
		.delay_usecs = delay,
		.speed_hz = speed,
		.bits_per_word = bits,
	};

	if (mode & SPI_TX_QUAD)
		tr.tx_nbits = 4;
	else if (mode & SPI_TX_DUAL)
		tr.tx_nbits = 2;
	if (mode & SPI_RX_QUAD)
		tr.rx_nbits = 4;
	else if (mode & SPI_RX_DUAL)
		tr.rx_nbits = 2;
	if (!(mode & SPI_LOOP)) {
		if (mode & (SPI_TX_QUAD | SPI_TX_DUAL))
			tr.rx_buf = 0;
		else if (mode & (SPI_RX_QUAD | SPI_RX_DUAL))
			tr.tx_buf = 0;
	}

	ret = ioctl(fd, SPI_IOC_MESSAGE(1), &tr);
	if (ret < 1)
		pabort("can't send spi message");

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

	if (output_file) {
		out_fd = open(output_file, O_WRONLY | O_CREAT | O_TRUNC, 0666);
		if (out_fd < 0)
			pabort("could not open output file");

		ret = write(out_fd, rx, len);
		if (ret != len)
			pabort("not all bytes written to output file");

		close(out_fd);
	}

	if (verbose)
		hex_dump(rx, len, 32, "RX");
}

static void print_usage(const char *prog)
{
	printf("Usage: %s [-DsbdlHOLC3vpNR24SI]\n", prog);
	puts("  -D --device   device to use (default /dev/spidev1.1)\n"
	     "  -s --speed    max speed (Hz)\n"
	     "  -d --delay    delay (usec)\n"
	     "  -b --bpw      bits per word\n"
	     "  -i --input    input data from a file (e.g. \"test.bin\")\n"
	     "  -o --output   output data to a file (e.g. \"results.bin\")\n"
	     "  -l --loop     loopback\n"
	     "  -H --cpha     clock phase\n"
	     "  -O --cpol     clock polarity\n"
	     "  -L --lsb      least significant bit first\n"
	     "  -C --cs-high  chip select active high\n"
	     "  -3 --3wire    SI/SO signals shared\n"
	     "  -v --verbose  Verbose (show tx buffer)\n"
	     "  -p            Send data (e.g. \"1234\\xde\\xad\")\n"
	     "  -N --no-cs    no chip select\n"
	     "  -R --ready    slave pulls low to pause\n"
	     "  -2 --dual     dual transfer\n"
	     "  -4 --quad     quad transfer\n"
	     "  -S --size     transfer size\n"
	     "  -I --iter     iterations\n");
	exit(1);
}

static void parse_opts(int argc, char *argv[])
{
	while (1) {
		static const struct option lopts[] = {
			{ "device",  1, 0, 'D' },
			{ "speed",   1, 0, 's' },
			{ "delay",   1, 0, 'd' },
			{ "bpw",     1, 0, 'b' },
			{ "input",   1, 0, 'i' },
			{ "output",  1, 0, 'o' },
			{ "loop",    0, 0, 'l' },
			{ "cpha",    0, 0, 'H' },
			{ "cpol",    0, 0, 'O' },
			{ "lsb",     0, 0, 'L' },
			{ "cs-high", 0, 0, 'C' },
			{ "3wire",   0, 0, '3' },
			{ "no-cs",   0, 0, 'N' },
			{ "ready",   0, 0, 'R' },
			{ "dual",    0, 0, '2' },
			{ "verbose", 0, 0, 'v' },
			{ "quad",    0, 0, '4' },
			{ "size",    1, 0, 'S' },
			{ "iter",    1, 0, 'I' },
			{ NULL, 0, 0, 0 },
		};
		int c;

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

		if (c == -1)
			break;

		switch (c) {
		case 'D':
			device = optarg;
			break;
		case 's':
			speed = atoi(optarg);
			break;
		case 'd':
			delay = atoi(optarg);
			break;
		case 'b':
			bits = atoi(optarg);
			break;
		case 'i':
			input_file = optarg;
			break;
		case 'o':
			output_file = optarg;
			break;
		case 'l':
			mode |= SPI_LOOP;
			break;
		case 'H':
			mode |= SPI_CPHA;
			break;
		case 'O':
			mode |= SPI_CPOL;
			break;
		case 'L':
			mode |= SPI_LSB_FIRST;
			break;
		case 'C':
			mode |= SPI_CS_HIGH;
			break;
		case '3':
			mode |= SPI_3WIRE;
			break;
		case 'N':
			mode |= SPI_NO_CS;
			break;
		case 'v':
			verbose = 1;
			break;
		case 'R':
			mode |= SPI_READY;
			break;
		case 'p':
			input_tx = optarg;
			break;
		case '2':
			mode |= SPI_TX_DUAL;
			break;
		case '4':
			mode |= SPI_TX_QUAD;
			break;
		case 'S':
			transfer_size = atoi(optarg);
			break;
		case 'I':
			iterations = atoi(optarg);
			break;
		default:
			print_usage(argv[0]);
		}
	}
	if (mode & SPI_LOOP) {
		if (mode & SPI_TX_DUAL)
			mode |= SPI_RX_DUAL;
		if (mode & SPI_TX_QUAD)
			mode |= SPI_RX_QUAD;
	}
}

static void transfer_escaped_string(int fd, char *str)
{
	size_t size = strlen(str);
	uint8_t *tx;
	uint8_t *rx;

	tx = malloc(size);
	if (!tx)
		pabort("can't allocate tx buffer");

	rx = malloc(size);
	if (!rx)
		pabort("can't allocate rx buffer");

	size = unescape((char *)tx, str, size);
	transfer(fd, tx, rx, size);
	free(rx);
	free(tx);
}

static void transfer_file(int fd, char *filename)
{
	ssize_t bytes;
	struct stat sb;
	int tx_fd;
	uint8_t *tx;
	uint8_t *rx;

	if (stat(filename, &sb) == -1)
		pabort("can't stat input file");

	tx_fd = open(filename, O_RDONLY);
	if (tx_fd < 0)
		pabort("can't open input file");

	tx = malloc(sb.st_size);
	if (!tx)
		pabort("can't allocate tx buffer");

	rx = malloc(sb.st_size);
	if (!rx)
		pabort("can't allocate rx buffer");

	bytes = read(tx_fd, tx, sb.st_size);
	if (bytes != sb.st_size)
		pabort("failed to read input file");

	transfer(fd, tx, rx, sb.st_size);
	free(rx);
	free(tx);
	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 ret = 0;
	int fd;

	parse_opts(argc, argv);

	if (input_tx && input_file)
		pabort("only one of -p and --input may be selected");

	fd = open(device, O_RDWR);
	if (fd < 0)
		pabort("can't open device");

	/*
	 * spi mode
	 */
	ret = ioctl(fd, SPI_IOC_WR_MODE32, &mode);
	if (ret == -1)
		pabort("can't set spi mode");

	ret = ioctl(fd, SPI_IOC_RD_MODE32, &mode);
	if (ret == -1)
		pabort("can't get spi mode");

	/*
	 * bits per word
	 */
	ret = ioctl(fd, SPI_IOC_WR_BITS_PER_WORD, &bits);
	if (ret == -1)
		pabort("can't set bits per word");

	ret = ioctl(fd, SPI_IOC_RD_BITS_PER_WORD, &bits);
	if (ret == -1)
		pabort("can't get bits per word");

	/*
	 * max speed hz
	 */
	ret = ioctl(fd, SPI_IOC_WR_MAX_SPEED_HZ, &speed);
	if (ret == -1)
		pabort("can't set max speed hz");

	ret = ioctl(fd, SPI_IOC_RD_MAX_SPEED_HZ, &speed);
	if (ret == -1)
		pabort("can't get max speed hz");

	printf("spi mode: 0x%x\n", mode);
	printf("bits per word: %d\n", bits);
	printf("max speed: %d Hz (%d KHz)\n", speed, speed/1000);

	if (input_tx)
		transfer_escaped_string(fd, input_tx);
	else if (input_file)
		transfer_file(fd, input_file);
	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));

	close(fd);

	return ret;
}
Commit	Line	Data
84a14ae8	1	// SPDX-License-Identifier: GPL-2.0-only
22b238bd AV	2	/*
	3	* SPI testing utility (using spidev driver)
	4	*
	5	* Copyright (c) 2007 MontaVista Software, Inc.
	6	* Copyright (c) 2007 Anton Vorontsov <avorontsov@ru.mvista.com>
	7	*
22b238bd AV	8	* Cross-compile with cross-gcc -I/path/to/cross-kernel/include
	9	*/
	10
	11	#include <stdint.h>
	12	#include <unistd.h>
	13	#include <stdio.h>
	14	#include <stdlib.h>
b78ce7ed	15	#include <string.h>
470a072e	16	#include <errno.h>
22b238bd AV	17	#include <getopt.h>
22b238bd AV	18	#include <fcntl.h>
9006a7b3	19	#include <time.h>
22b238bd	20	#include <sys/ioctl.h>
8736f802	21	#include <linux/ioctl.h>
7af475a5	22	#include <sys/stat.h>
22b238bd AV	23	#include <linux/types.h>
	24	#include <linux/spi/spidev.h>
	25
	26	#define ARRAY_SIZE(a) (sizeof(a) / sizeof((a)[0]))
	27
	28	static void pabort(const char *s)
	29	{
470a072e TY	30	if (errno != 0)
	31	perror(s);
	32	else
	33	printf("%s\n", s);
	34
22b238bd AV	35	abort();
	36	}
	37
cd58310d	38	static const char *device = "/dev/spidev1.1";
c2e78c34	39	static uint32_t mode;
22b238bd	40	static uint8_t bits = 8;
7af475a5	41	static char *input_file;
983b2788	42	static char *output_file;
22b238bd AV	43	static uint32_t speed = 500000;
22b238bd AV	44	static uint16_t delay;
31a5c5a7	45	static int verbose;
9006a7b3 FI	46	static int transfer_size;
	47	static int iterations;
	48	static int interval = 5; /* interval in seconds for showing transfer rate */
22b238bd	49
30061915 AR	50	uint8_t default_tx[] = {
	51	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
	52	0x40, 0x00, 0x00, 0x00, 0x00, 0x95,
	53	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
	54	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
	55	0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
	56	0xF0, 0x0D,
	57	};
	58
	59	uint8_t default_rx[ARRAY_SIZE(default_tx)] = {0, };
	60	char *input_tx;
	61
b2cfc904 JC	62	static void hex_dump(const void *src, size_t length, size_t line_size,
b2cfc904 JC	63	char *prefix)
b78ce7ed AR	64	{
	65	int i = 0;
	66	const unsigned char *address = src;
	67	const unsigned char *line = address;
	68	unsigned char c;
	69
	70	printf("%s \| ", prefix);
	71	while (length-- > 0) {
	72	printf("%02X ", *address++);
	73	if (!(++i % line_size) \|\| (length == 0 && i % line_size)) {
	74	if (length == 0) {
	75	while (i++ % line_size)
	76	printf("__ ");
	77	}
35386dfd	78	printf(" \|");
b78ce7ed AR	79	while (line < address) {
b78ce7ed AR	80	c = *line++;
35386dfd	81	printf("%c", (c < 32 \|\| c > 126) ? '.' : c);
b78ce7ed	82	}
35386dfd	83	printf("\|\n");
b78ce7ed AR	84	if (length > 0)
	85	printf("%s \| ", prefix);
	86	}
	87	}
	88	}
	89
30061915 AR	90	/*
	91	* Unescape - process hexadecimal escape character
	92	* converts shell input "\x23" -> 0x23
	93	*/
07eec628	94	static int unescape(char _dst, char _src, size_t len)
30061915 AR	95	{
30061915 AR	96	int ret = 0;
a20874f7	97	int match;
30061915 AR	98	char *src = _src;
	99	char *dst = _dst;
	100	unsigned int ch;
	101
	102	while (*src) {
	103	if (src == '\\' && (src+1) == 'x') {
a20874f7 JC	104	match = sscanf(src + 2, "%2x", &ch);
	105	if (!match)
	106	pabort("malformed input string");
	107
30061915 AR	108	src += 4;
	109	*dst++ = (unsigned char)ch;
	110	} else {
	111	dst++ = src++;
	112	}
	113	ret++;
	114	}
	115	return ret;
	116	}
	117
	118	static void transfer(int fd, uint8_t const tx, uint8_t const rx, size_t len)
22b238bd AV	119	{
22b238bd AV	120	int ret;
983b2788	121	int out_fd;
22b238bd AV	122	struct spi_ioc_transfer tr = {
	123	.tx_buf = (unsigned long)tx,
	124	.rx_buf = (unsigned long)rx,
30061915	125	.len = len,
22b238bd AV	126	.delay_usecs = delay,
	127	.speed_hz = speed,
	128	.bits_per_word = bits,
	129	};
	130
c2e78c34 GU	131	if (mode & SPI_TX_QUAD)
	132	tr.tx_nbits = 4;
	133	else if (mode & SPI_TX_DUAL)
	134	tr.tx_nbits = 2;
	135	if (mode & SPI_RX_QUAD)
	136	tr.rx_nbits = 4;
	137	else if (mode & SPI_RX_DUAL)
	138	tr.rx_nbits = 2;
	139	if (!(mode & SPI_LOOP)) {
	140	if (mode & (SPI_TX_QUAD \| SPI_TX_DUAL))
	141	tr.rx_buf = 0;
	142	else if (mode & (SPI_RX_QUAD \| SPI_RX_DUAL))
	143	tr.tx_buf = 0;
	144	}
	145
22b238bd	146	ret = ioctl(fd, SPI_IOC_MESSAGE(1), &tr);
95b1ed2a	147	if (ret < 1)
22b238bd AV	148	pabort("can't send spi message");
22b238bd AV	149
31a5c5a7	150	if (verbose)
30061915	151	hex_dump(tx, len, 32, "TX");
983b2788 JC	152
	153	if (output_file) {
	154	out_fd = open(output_file, O_WRONLY \| O_CREAT \| O_TRUNC, 0666);
	155	if (out_fd < 0)
	156	pabort("could not open output file");
	157
	158	ret = write(out_fd, rx, len);
	159	if (ret != len)
edd3899c	160	pabort("not all bytes written to output file");
983b2788 JC	161
	162	close(out_fd);
	163	}
	164
9006a7b3	165	if (verbose)
983b2788	166	hex_dump(rx, len, 32, "RX");
22b238bd AV	167	}
22b238bd AV	168
b7ed698c	169	static void print_usage(const char *prog)
22b238bd	170	{
9006a7b3	171	printf("Usage: %s [-DsbdlHOLC3vpNR24SI]\n", prog);
22b238bd AV	172	puts(" -D --device device to use (default /dev/spidev1.1)\n"
	173	" -s --speed max speed (Hz)\n"
	174	" -d --delay delay (usec)\n"
b2cfc904	175	" -b --bpw bits per word\n"
7af475a5	176	" -i --input input data from a file (e.g. \"test.bin\")\n"
983b2788	177	" -o --output output data to a file (e.g. \"results.bin\")\n"
22b238bd AV	178	" -l --loop loopback\n"
	179	" -H --cpha clock phase\n"
	180	" -O --cpol clock polarity\n"
	181	" -L --lsb least significant bit first\n"
	182	" -C --cs-high chip select active high\n"
925d16a2	183	" -3 --3wire SI/SO signals shared\n"
31a5c5a7	184	" -v --verbose Verbose (show tx buffer)\n"
30061915	185	" -p Send data (e.g. \"1234\\xde\\xad\")\n"
925d16a2	186	" -N --no-cs no chip select\n"
c2e78c34 GU	187	" -R --ready slave pulls low to pause\n"
c2e78c34 GU	188	" -2 --dual dual transfer\n"
9006a7b3 FI	189	" -4 --quad quad transfer\n"
	190	" -S --size transfer size\n"
	191	" -I --iter iterations\n");
22b238bd AV	192	exit(1);
	193	}
	194
b7ed698c	195	static void parse_opts(int argc, char *argv[])
22b238bd AV	196	{
22b238bd AV	197	while (1) {
cd58310d	198	static const struct option lopts[] = {
22b238bd AV	199	{ "device", 1, 0, 'D' },
	200	{ "speed", 1, 0, 's' },
	201	{ "delay", 1, 0, 'd' },
	202	{ "bpw", 1, 0, 'b' },
7af475a5	203	{ "input", 1, 0, 'i' },
983b2788	204	{ "output", 1, 0, 'o' },
22b238bd AV	205	{ "loop", 0, 0, 'l' },
	206	{ "cpha", 0, 0, 'H' },
	207	{ "cpol", 0, 0, 'O' },
	208	{ "lsb", 0, 0, 'L' },
	209	{ "cs-high", 0, 0, 'C' },
	210	{ "3wire", 0, 0, '3' },
b55f627f DB	211	{ "no-cs", 0, 0, 'N' },
b55f627f DB	212	{ "ready", 0, 0, 'R' },
c2e78c34	213	{ "dual", 0, 0, '2' },
31a5c5a7	214	{ "verbose", 0, 0, 'v' },
c2e78c34	215	{ "quad", 0, 0, '4' },
9006a7b3 FI	216	{ "size", 1, 0, 'S' },
9006a7b3 FI	217	{ "iter", 1, 0, 'I' },
22b238bd AV	218	{ NULL, 0, 0, 0 },
	219	};
	220	int c;
	221
9006a7b3	222	c = getopt_long(argc, argv, "D:s:d:b:i:o:lHOLC3NR24p:vS:I:",
7af475a5	223	lopts, NULL);
22b238bd AV	224
	225	if (c == -1)
	226	break;
	227
	228	switch (c) {
	229	case 'D':
	230	device = optarg;
	231	break;
	232	case 's':
	233	speed = atoi(optarg);
	234	break;
	235	case 'd':
	236	delay = atoi(optarg);
	237	break;
	238	case 'b':
	239	bits = atoi(optarg);
	240	break;
7af475a5 JC	241	case 'i':
	242	input_file = optarg;
	243	break;
983b2788 JC	244	case 'o':
	245	output_file = optarg;
	246	break;
22b238bd AV	247	case 'l':
	248	mode \|= SPI_LOOP;
	249	break;
	250	case 'H':
	251	mode \|= SPI_CPHA;
	252	break;
	253	case 'O':
	254	mode \|= SPI_CPOL;
	255	break;
	256	case 'L':
	257	mode \|= SPI_LSB_FIRST;
	258	break;
	259	case 'C':
	260	mode \|= SPI_CS_HIGH;
	261	break;
	262	case '3':
	263	mode \|= SPI_3WIRE;
	264	break;
b55f627f DB	265	case 'N':
	266	mode \|= SPI_NO_CS;
	267	break;
31a5c5a7 AR	268	case 'v':
	269	verbose = 1;
	270	break;
b55f627f DB	271	case 'R':
	272	mode \|= SPI_READY;
	273	break;
30061915 AR	274	case 'p':
	275	input_tx = optarg;
	276	break;
c2e78c34 GU	277	case '2':
	278	mode \|= SPI_TX_DUAL;
	279	break;
	280	case '4':
	281	mode \|= SPI_TX_QUAD;
	282	break;
9006a7b3 FI	283	case 'S':
	284	transfer_size = atoi(optarg);
	285	break;
	286	case 'I':
	287	iterations = atoi(optarg);
	288	break;
22b238bd AV	289	default:
22b238bd AV	290	print_usage(argv[0]);
22b238bd AV	291	}
22b238bd AV	292	}
c2e78c34 GU	293	if (mode & SPI_LOOP) {
	294	if (mode & SPI_TX_DUAL)
	295	mode \|= SPI_RX_DUAL;
	296	if (mode & SPI_TX_QUAD)
	297	mode \|= SPI_RX_QUAD;
	298	}
22b238bd AV	299	}
22b238bd AV	300
5c437a40 JC	301	static void transfer_escaped_string(int fd, char *str)
5c437a40 JC	302	{
0278b34b	303	size_t size = strlen(str);
5c437a40 JC	304	uint8_t *tx;
	305	uint8_t *rx;
	306
	307	tx = malloc(size);
	308	if (!tx)
	309	pabort("can't allocate tx buffer");
	310
	311	rx = malloc(size);
	312	if (!rx)
	313	pabort("can't allocate rx buffer");
	314
	315	size = unescape((char *)tx, str, size);
	316	transfer(fd, tx, rx, size);
	317	free(rx);
	318	free(tx);
	319	}
	320
7af475a5 JC	321	static void transfer_file(int fd, char *filename)
	322	{
	323	ssize_t bytes;
	324	struct stat sb;
	325	int tx_fd;
	326	uint8_t *tx;
	327	uint8_t *rx;
	328
	329	if (stat(filename, &sb) == -1)
	330	pabort("can't stat input file");
	331
	332	tx_fd = open(filename, O_RDONLY);
e634b76c	333	if (tx_fd < 0)
7af475a5 JC	334	pabort("can't open input file");
	335
	336	tx = malloc(sb.st_size);
	337	if (!tx)
	338	pabort("can't allocate tx buffer");
	339
	340	rx = malloc(sb.st_size);
	341	if (!rx)
	342	pabort("can't allocate rx buffer");
	343
	344	bytes = read(tx_fd, tx, sb.st_size);
	345	if (bytes != sb.st_size)
	346	pabort("failed to read input file");
	347
	348	transfer(fd, tx, rx, sb.st_size);
	349	free(rx);
	350	free(tx);
	351	close(tx_fd);
	352	}
	353
9006a7b3 FI	354	static uint64_t _read_count;
	355	static uint64_t _write_count;
	356
	357	static void show_transfer_rate(void)
	358	{
	359	static uint64_t prev_read_count, prev_write_count;
	360	double rx_rate, tx_rate;
	361
	362	rx_rate = ((_read_count - prev_read_count) * 8) / (interval*1000.0);
	363	tx_rate = ((_write_count - prev_write_count) * 8) / (interval*1000.0);
	364
	365	printf("rate: tx %.1fkbps, rx %.1fkbps\n", rx_rate, tx_rate);
	366
	367	prev_read_count = _read_count;
	368	prev_write_count = _write_count;
	369	}
	370
	371	static void transfer_buf(int fd, int len)
	372	{
	373	uint8_t *tx;
	374	uint8_t *rx;
	375	int i;
	376
	377	tx = malloc(len);
	378	if (!tx)
	379	pabort("can't allocate tx buffer");
	380	for (i = 0; i < len; i++)
	381	tx[i] = random();
	382
	383	rx = malloc(len);
	384	if (!rx)
	385	pabort("can't allocate rx buffer");
	386
	387	transfer(fd, tx, rx, len);
	388
	389	_write_count += len;
	390	_read_count += len;
	391
	392	if (mode & SPI_LOOP) {
	393	if (memcmp(tx, rx, len)) {
	394	fprintf(stderr, "transfer error !\n");
	395	hex_dump(tx, len, 32, "TX");
	396	hex_dump(rx, len, 32, "RX");
	397	exit(1);
	398	}
	399	}
	400
	401	free(rx);
	402	free(tx);
	403	}
	404
22b238bd AV	405	int main(int argc, char *argv[])
	406	{
	407	int ret = 0;
	408	int fd;
	409
	410	parse_opts(argc, argv);
	411
1f3c3632 TY	412	if (input_tx && input_file)
	413	pabort("only one of -p and --input may be selected");
	414
22b238bd AV	415	fd = open(device, O_RDWR);
	416	if (fd < 0)
	417	pabort("can't open device");
	418
	419	/*
	420	* spi mode
	421	*/
c2e78c34	422	ret = ioctl(fd, SPI_IOC_WR_MODE32, &mode);
22b238bd AV	423	if (ret == -1)
	424	pabort("can't set spi mode");
	425
c2e78c34	426	ret = ioctl(fd, SPI_IOC_RD_MODE32, &mode);
22b238bd AV	427	if (ret == -1)
	428	pabort("can't get spi mode");
	429
	430	/*
	431	* bits per word
	432	*/
	433	ret = ioctl(fd, SPI_IOC_WR_BITS_PER_WORD, &bits);
	434	if (ret == -1)
	435	pabort("can't set bits per word");
	436
	437	ret = ioctl(fd, SPI_IOC_RD_BITS_PER_WORD, &bits);
	438	if (ret == -1)
	439	pabort("can't get bits per word");
	440
	441	/*
	442	* max speed hz
	443	*/
	444	ret = ioctl(fd, SPI_IOC_WR_MAX_SPEED_HZ, &speed);
	445	if (ret == -1)
	446	pabort("can't set max speed hz");
	447
	448	ret = ioctl(fd, SPI_IOC_RD_MAX_SPEED_HZ, &speed);
	449	if (ret == -1)
	450	pabort("can't get max speed hz");
	451
c2e78c34	452	printf("spi mode: 0x%x\n", mode);
22b238bd AV	453	printf("bits per word: %d\n", bits);
	454	printf("max speed: %d Hz (%d KHz)\n", speed, speed/1000);
	455
5c437a40 JC	456	if (input_tx)
5c437a40 JC	457	transfer_escaped_string(fd, input_tx);
7af475a5 JC	458	else if (input_file)
7af475a5 JC	459	transfer_file(fd, input_file);
9006a7b3 FI	460	else if (transfer_size) {
	461	struct timespec last_stat;
	462
	463	clock_gettime(CLOCK_MONOTONIC, &last_stat);
	464
	465	while (iterations-- > 0) {
	466	struct timespec current;
	467
	468	transfer_buf(fd, transfer_size);
	469
	470	clock_gettime(CLOCK_MONOTONIC, &current);
	471	if (current.tv_sec - last_stat.tv_sec > interval) {
	472	show_transfer_rate();
	473	last_stat = current;
	474	}
	475	}
	476	printf("total: tx %.1fKB, rx %.1fKB\n",
	477	_write_count/1024.0, _read_count/1024.0);
	478	} else
30061915	479	transfer(fd, default_tx, default_rx, sizeof(default_tx));
22b238bd AV	480
	481	close(fd);
	482
	483	return ret;
	484	}