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> 18 #include <string.h> 19 #include <getopt.h> 20 #include <fcntl.h> 21 #include <time.h> 22 #include <sys/ioctl.h> 23 #include <linux/ioctl.h> 24 #include <sys/stat.h> 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 36 static const char *device = "/dev/spidev1.1"; 37 static uint32_t mode; 38 static uint8_t bits = 8; 39 static char *input_file; 40 static char *output_file; 41 static uint32_t speed = 500000; 42 static uint16_t delay; 43 static int verbose; 44 static int transfer_size; 45 static int iterations; 46 static int interval = 5; /* interval in seconds for showing transfer rate */ 47 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 60 static void hex_dump(const void *src, size_t length, size_t line_size, 61 char *prefix) 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(" |"); 77 while (line < address) { 78 c = *line++; 79 printf("%c", (c < 32 || c > 126) ? '.' : c); 80 } 81 printf("|\n"); 82 if (length > 0) 83 printf("%s | ", prefix); 84 } 85 } 86 } 87 88 /* 89 * Unescape - process hexadecimal escape character 90 * converts shell input "\x23" -> 0x23 91 */ 92 static int unescape(char *_dst, char *_src, size_t len) 93 { 94 int ret = 0; 95 int match; 96 char *src = _src; 97 char *dst = _dst; 98 unsigned int ch; 99 100 while (*src) { 101 if (*src == '\\' && *(src+1) == 'x') { 102 match = sscanf(src + 2, "%2x", &ch); 103 if (!match) 104 pabort("malformed input string"); 105 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) 117 { 118 int ret; 119 int out_fd; 120 struct spi_ioc_transfer tr = { 121 .tx_buf = (unsigned long)tx, 122 .rx_buf = (unsigned long)rx, 123 .len = len, 124 .delay_usecs = delay, 125 .speed_hz = speed, 126 .bits_per_word = bits, 127 }; 128 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 144 ret = ioctl(fd, SPI_IOC_MESSAGE(1), &tr); 145 if (ret < 1) 146 pabort("can't send spi message"); 147 148 if (verbose) 149 hex_dump(tx, len, 32, "TX"); 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) 158 pabort("not all bytes written to output file"); 159 160 close(out_fd); 161 } 162 163 if (verbose) 164 hex_dump(rx, len, 32, "RX"); 165 } 166 167 static void print_usage(const char *prog) 168 { 169 printf("Usage: %s [-DsbdlHOLC3vpNR24SI]\n", prog); 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" 173 " -b --bpw bits per word\n" 174 " -i --input input data from a file (e.g. \"test.bin\")\n" 175 " -o --output output data to a file (e.g. \"results.bin\")\n" 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" 181 " -3 --3wire SI/SO signals shared\n" 182 " -v --verbose Verbose (show tx buffer)\n" 183 " -p Send data (e.g. \"1234\\xde\\xad\")\n" 184 " -N --no-cs no chip select\n" 185 " -R --ready slave pulls low to pause\n" 186 " -2 --dual dual transfer\n" 187 " -4 --quad quad transfer\n" 188 " -S --size transfer size\n" 189 " -I --iter iterations\n"); 190 exit(1); 191 } 192 193 static void parse_opts(int argc, char *argv[]) 194 { 195 while (1) { 196 static const struct option lopts[] = { 197 { "device", 1, 0, 'D' }, 198 { "speed", 1, 0, 's' }, 199 { "delay", 1, 0, 'd' }, 200 { "bpw", 1, 0, 'b' }, 201 { "input", 1, 0, 'i' }, 202 { "output", 1, 0, 'o' }, 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' }, 209 { "no-cs", 0, 0, 'N' }, 210 { "ready", 0, 0, 'R' }, 211 { "dual", 0, 0, '2' }, 212 { "verbose", 0, 0, 'v' }, 213 { "quad", 0, 0, '4' }, 214 { "size", 1, 0, 'S' }, 215 { "iter", 1, 0, 'I' }, 216 { NULL, 0, 0, 0 }, 217 }; 218 int c; 219 220 c = getopt_long(argc, argv, "D:s:d:b:i:o:lHOLC3NR24p:vS:I:", 221 lopts, NULL); 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; 239 case 'i': 240 input_file = optarg; 241 break; 242 case 'o': 243 output_file = optarg; 244 break; 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; 263 case 'N': 264 mode |= SPI_NO_CS; 265 break; 266 case 'v': 267 verbose = 1; 268 break; 269 case 'R': 270 mode |= SPI_READY; 271 break; 272 case 'p': 273 input_tx = optarg; 274 break; 275 case '2': 276 mode |= SPI_TX_DUAL; 277 break; 278 case '4': 279 mode |= SPI_TX_QUAD; 280 break; 281 case 'S': 282 transfer_size = atoi(optarg); 283 break; 284 case 'I': 285 iterations = atoi(optarg); 286 break; 287 default: 288 print_usage(argv[0]); 289 break; 290 } 291 } 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 } 298 } 299 300 static void transfer_escaped_string(int fd, char *str) 301 { 302 size_t size = strlen(str); 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 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); 332 if (tx_fd < 0) 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 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 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 */ 418 ret = ioctl(fd, SPI_IOC_WR_MODE32, &mode); 419 if (ret == -1) 420 pabort("can't set spi mode"); 421 422 ret = ioctl(fd, SPI_IOC_RD_MODE32, &mode); 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 448 printf("spi mode: 0x%x\n", mode); 449 printf("bits per word: %d\n", bits); 450 printf("max speed: %d Hz (%d KHz)\n", speed, speed/1000); 451 452 if (input_tx && input_file) 453 pabort("only one of -p and --input may be selected"); 454 455 if (input_tx) 456 transfer_escaped_string(fd, input_tx); 457 else if (input_file) 458 transfer_file(fd, input_file); 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, ¤t); 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 478 transfer(fd, default_tx, default_rx, sizeof(default_tx)); 479 480 close(fd); 481 482 return ret; 483 } 484