1
2
3
4
5
6
7
8
9
10
11#include <stdint.h>
12#include <unistd.h>
13#include <stdio.h>
14#include <stdlib.h>
15#include <string.h>
16#include <errno.h>
17#include <getopt.h>
18#include <fcntl.h>
19#include <time.h>
20#include <sys/ioctl.h>
21#include <linux/ioctl.h>
22#include <sys/stat.h>
23#include <linux/types.h>
24#include <linux/spi/spidev.h>
25
26#define ARRAY_SIZE(a) (sizeof(a) / sizeof((a)[0]))
27
28static void pabort(const char *s)
29{
30 if (errno != 0)
31 perror(s);
32 else
33 printf("%s\n", s);
34
35 abort();
36}
37
38static const char *device = "/dev/spidev1.1";
39static uint32_t mode;
40static uint8_t bits = 8;
41static char *input_file;
42static char *output_file;
43static uint32_t speed = 500000;
44static uint16_t delay;
45static int verbose;
46static int transfer_size;
47static int iterations;
48static int interval = 5;
49
50uint8_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
59uint8_t default_rx[ARRAY_SIZE(default_tx)] = {0, };
60char *input_tx;
61
62static void hex_dump(const void *src, size_t length, size_t line_size,
63 char *prefix)
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 }
78 printf(" |");
79 while (line < address) {
80 c = *line++;
81 printf("%c", (c < 32 || c > 126) ? '.' : c);
82 }
83 printf("|\n");
84 if (length > 0)
85 printf("%s | ", prefix);
86 }
87 }
88}
89
90
91
92
93
94static int unescape(char *_dst, char *_src, size_t len)
95{
96 int ret = 0;
97 int match;
98 char *src = _src;
99 char *dst = _dst;
100 unsigned int ch;
101
102 while (*src) {
103 if (*src == '\\' && *(src+1) == 'x') {
104 match = sscanf(src + 2, "%2x", &ch);
105 if (!match)
106 pabort("malformed input string");
107
108 src += 4;
109 *dst++ = (unsigned char)ch;
110 } else {
111 *dst++ = *src++;
112 }
113 ret++;
114 }
115 return ret;
116}
117
118static void transfer(int fd, uint8_t const *tx, uint8_t const *rx, size_t len)
119{
120 int ret;
121 int out_fd;
122 struct spi_ioc_transfer tr = {
123 .tx_buf = (unsigned long)tx,
124 .rx_buf = (unsigned long)rx,
125 .len = len,
126 .delay_usecs = delay,
127 .speed_hz = speed,
128 .bits_per_word = bits,
129 };
130
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
146 ret = ioctl(fd, SPI_IOC_MESSAGE(1), &tr);
147 if (ret < 1)
148 pabort("can't send spi message");
149
150 if (verbose)
151 hex_dump(tx, len, 32, "TX");
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)
160 pabort("not all bytes written to output file");
161
162 close(out_fd);
163 }
164
165 if (verbose)
166 hex_dump(rx, len, 32, "RX");
167}
168
169static void print_usage(const char *prog)
170{
171 printf("Usage: %s [-DsbdlHOLC3vpNR24SI]\n", prog);
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"
175 " -b --bpw bits per word\n"
176 " -i --input input data from a file (e.g. \"test.bin\")\n"
177 " -o --output output data to a file (e.g. \"results.bin\")\n"
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"
183 " -3 --3wire SI/SO signals shared\n"
184 " -v --verbose Verbose (show tx buffer)\n"
185 " -p Send data (e.g. \"1234\\xde\\xad\")\n"
186 " -N --no-cs no chip select\n"
187 " -R --ready slave pulls low to pause\n"
188 " -2 --dual dual transfer\n"
189 " -4 --quad quad transfer\n"
190 " -S --size transfer size\n"
191 " -I --iter iterations\n");
192 exit(1);
193}
194
195static void parse_opts(int argc, char *argv[])
196{
197 while (1) {
198 static const struct option lopts[] = {
199 { "device", 1, 0, 'D' },
200 { "speed", 1, 0, 's' },
201 { "delay", 1, 0, 'd' },
202 { "bpw", 1, 0, 'b' },
203 { "input", 1, 0, 'i' },
204 { "output", 1, 0, 'o' },
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' },
211 { "no-cs", 0, 0, 'N' },
212 { "ready", 0, 0, 'R' },
213 { "dual", 0, 0, '2' },
214 { "verbose", 0, 0, 'v' },
215 { "quad", 0, 0, '4' },
216 { "size", 1, 0, 'S' },
217 { "iter", 1, 0, 'I' },
218 { NULL, 0, 0, 0 },
219 };
220 int c;
221
222 c = getopt_long(argc, argv, "D:s:d:b:i:o:lHOLC3NR24p:vS:I:",
223 lopts, NULL);
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;
241 case 'i':
242 input_file = optarg;
243 break;
244 case 'o':
245 output_file = optarg;
246 break;
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;
265 case 'N':
266 mode |= SPI_NO_CS;
267 break;
268 case 'v':
269 verbose = 1;
270 break;
271 case 'R':
272 mode |= SPI_READY;
273 break;
274 case 'p':
275 input_tx = optarg;
276 break;
277 case '2':
278 mode |= SPI_TX_DUAL;
279 break;
280 case '4':
281 mode |= SPI_TX_QUAD;
282 break;
283 case 'S':
284 transfer_size = atoi(optarg);
285 break;
286 case 'I':
287 iterations = atoi(optarg);
288 break;
289 default:
290 print_usage(argv[0]);
291 }
292 }
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 }
299}
300
301static void transfer_escaped_string(int fd, char *str)
302{
303 size_t size = strlen(str);
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
321static 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);
333 if (tx_fd < 0)
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
354static uint64_t _read_count;
355static uint64_t _write_count;
356
357static 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
371static 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
405int main(int argc, char *argv[])
406{
407 int ret = 0;
408 int fd;
409
410 parse_opts(argc, argv);
411
412 if (input_tx && input_file)
413 pabort("only one of -p and --input may be selected");
414
415 fd = open(device, O_RDWR);
416 if (fd < 0)
417 pabort("can't open device");
418
419
420
421
422 ret = ioctl(fd, SPI_IOC_WR_MODE32, &mode);
423 if (ret == -1)
424 pabort("can't set spi mode");
425
426 ret = ioctl(fd, SPI_IOC_RD_MODE32, &mode);
427 if (ret == -1)
428 pabort("can't get spi mode");
429
430
431
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
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
452 printf("spi mode: 0x%x\n", mode);
453 printf("bits per word: %d\n", bits);
454 printf("max speed: %d Hz (%d KHz)\n", speed, speed/1000);
455
456 if (input_tx)
457 transfer_escaped_string(fd, input_tx);
458 else if (input_file)
459 transfer_file(fd, input_file);
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, ¤t);
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
479 transfer(fd, default_tx, default_rx, sizeof(default_tx));
480
481 close(fd);
482
483 return ret;
484}
485
