spi-test.c

/*
 * SPI testing utility (using spidev driver)
 *
 * Copyright (c) 2007  MontaVista Software, Inc.
 * Copyright (c) 2007  Anton Vorontsov <avorontsov@ru.mvista.com>
 * Copyright (c) 2013  ORSoC AB
 *
 * 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
 *
 * To test PCI-to-tinySPI implementation on ORSoC ARMSoC borad:
 *  ./spi-test -CHO -D /dev/spidev0.0  0x80,0,0,0,0,0,0,0
 * Expected output is:
 * 80 00 00 00 00 00 00 00
 * FF FF FF FE 00 08 A0 01
 */

#include <stdint.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <getopt.h>
#include <fcntl.h>
#include <string.h>
#include <sys/ioctl.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 uint8_t mode = 0;
static uint8_t bits = 8;
static uint32_t speed = 500000;
static uint16_t delay = 0;
static struct spi_ioc_transfer *spi_xfrs = NULL;
static int ntransfers = 0;
static int toggle_cs = 0;
static int block_length = 0;
static int quiet = 0;

static void add_transfer(char * data, int len)
{
	struct spi_ioc_transfer *xfr;
	char *txbuf = malloc(len);
	char *rxbuf = malloc(len);

	memcpy(txbuf, data, len);
	memset(rxbuf, 0xff, len);
	
	ntransfers += 1;
	spi_xfrs = realloc(spi_xfrs, sizeof(*spi_xfrs) * ntransfers);
	xfr = &spi_xfrs[ntransfers - 1];

	xfr->tx_buf = (unsigned long)txbuf;
	xfr->rx_buf = (unsigned long)rxbuf;
	xfr->len = len;
	xfr->speed_hz = speed;
	xfr->delay_usecs = delay;
	xfr->bits_per_word = bits;
	xfr->cs_change =  toggle_cs;
	xfr->pad = 0;
}

static void show_spi_xfrs(void)
{
	int i, j;
	struct spi_ioc_transfer *xfr;

	for (i = 0; i < ntransfers; ++i) {
		xfr = &spi_xfrs[i];
		for (j = 0; j < xfr->len; ++j) {
			printf(" %02X",
			       ((unsigned char *)(intptr_t)xfr->tx_buf)[j]);
		}
		printf("\n");
		for (j = 0; j < xfr->len; ++j) {
			printf(" %02X",
			       ((unsigned char *)(intptr_t)xfr->rx_buf)[j]);
		}
		printf("\n\n");
	}
}

static void transfer(int fd)
{
	int ret;
	ret = ioctl(fd, SPI_IOC_MESSAGE(ntransfers), spi_xfrs);
	if (ret < 1)
		pabort("can't send spi message");
}

static void print_usage(const char *prog)
{
	printf("Usage: %s [-DsbdlnHOLC3] [data,..]\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"
	     "  -n --len      length\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"
	     "  -q --quiet    Don't output data send/received\n"
	     "  data is a series of octets to send, separated by comma, or b# to switch to another bits per word\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' },
			{ "len",     1, 0, 'n' },
			{ "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' },
			{ "quiet",   0, 0, 'q' },
			{ NULL, 0, 0, 0 },
		};
		int c;

		c = getopt_long(argc, argv, "D:s:d:b:n:lHOLC3NRq", lopts, NULL);

		if (c == -1)
			break;

		switch (c) {
		case 'D':
			device = optarg;
			break;
		case 's':
			speed = strtol(optarg, NULL, 0);
			break;
		case 'd':
			delay = strtol(optarg, NULL, 0);
			break;
		case 'b':
			bits = strtol(optarg, NULL, 0);
			break;
		case 'n':
			block_length = strtol(optarg, NULL, 0);
			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 'R':
			mode |= SPI_READY;
			break;
		case 'q':
			quiet = 1;
			break;
		default:
			print_usage(argv[0]);
			break;
		}
	}
}

void parse_transfer(char *arg)
{
	int len = 0;
	char *value;
	char buf[8192];
	for (value = strtok(arg, ","); value; value = strtok(NULL, ",")) {
		if (*value == 'b') {
			value++;
			bits = strtol(value, NULL, 0);
		} else if (*value == '@') {
			value++;
			FILE *in = fopen(value, "rb");
			if (!in) {
				perror("Failed to open input file");
				exit(1);
			}
			while ((len = fread(buf, 1, 2048, in)) > 0) {
				add_transfer(buf, len);
			}
			if (ferror(in)) {
				perror("Failed to read input file");
				exit(1);
			}
			fclose(in);
		} else {
			buf[len++] = strtol(value, NULL, 0);
		}
	}
	if (len > 0)
		add_transfer(buf, len);
}

void make_bulk_transfer(int len)
{
	uint8_t *buf = malloc(len);
	if (NULL == buf)
		pabort("malloc failed");
	memset(buf, 0xCB, len);
	add_transfer(buf, len);
}

int main(int argc, char *argv[])
{
	int ret = 0;
	int fd;
	unsigned i;

	parse_opts(argc, argv);

	if (block_length > 0) {
		make_bulk_transfer(block_length);
	} else {
		while (optind < argc)
			parse_transfer(argv[optind++]);
	}

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

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

	ret = ioctl(fd, SPI_IOC_RD_MODE, &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");

	if (block_length > 0) {
		i = 0;
		while (1) {
			transfer(fd);
			printf("%u\n", i);
			++i;
		}
	} else {
		transfer(fd);
		if (!quiet)
			show_spi_xfrs();
	}

	close(fd);

	return ret;
}