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

/*
 * SPI testing utility (using spidev driver)
 *
 * Copyright (c) 2007  MontaVista Software, Inc.
 * Copyright (c) 2007  Anton Vorontsov <avorontsov@ru.mvista.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License.
 *
 * 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 <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)
{
	perror(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(" | ");  /* right close */
			while (line < address) {
				c = *line++;
				printf("%c", (c < 33 || c == 255) ? 0x2E : 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]);
			break;
		}
	}
	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);

	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 && input_file)
		pabort("only one of -p and --input may be selected");

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