]>
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 <[email protected]> | |
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> |
20 | #include <fcntl.h> | |
21 | #include <sys/ioctl.h> | |
8736f802 | 22 | #include <linux/ioctl.h> |
7af475a5 | 23 | #include <sys/stat.h> |
22b238bd AV |
24 | #include <linux/types.h> |
25 | #include <linux/spi/spidev.h> | |
26 | ||
27 | #define ARRAY_SIZE(a) (sizeof(a) / sizeof((a)[0])) | |
28 | ||
29 | static void pabort(const char *s) | |
30 | { | |
31 | perror(s); | |
32 | abort(); | |
33 | } | |
34 | ||
cd58310d | 35 | static const char *device = "/dev/spidev1.1"; |
c2e78c34 | 36 | static uint32_t mode; |
22b238bd | 37 | static uint8_t bits = 8; |
7af475a5 | 38 | static char *input_file; |
983b2788 | 39 | static char *output_file; |
22b238bd AV |
40 | static uint32_t speed = 500000; |
41 | static uint16_t delay; | |
31a5c5a7 | 42 | static int verbose; |
22b238bd | 43 | |
30061915 AR |
44 | uint8_t default_tx[] = { |
45 | 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, | |
46 | 0x40, 0x00, 0x00, 0x00, 0x00, 0x95, | |
47 | 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, | |
48 | 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, | |
49 | 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, | |
50 | 0xF0, 0x0D, | |
51 | }; | |
52 | ||
53 | uint8_t default_rx[ARRAY_SIZE(default_tx)] = {0, }; | |
54 | char *input_tx; | |
55 | ||
b2cfc904 JC |
56 | static void hex_dump(const void *src, size_t length, size_t line_size, |
57 | char *prefix) | |
b78ce7ed AR |
58 | { |
59 | int i = 0; | |
60 | const unsigned char *address = src; | |
61 | const unsigned char *line = address; | |
62 | unsigned char c; | |
63 | ||
64 | printf("%s | ", prefix); | |
65 | while (length-- > 0) { | |
66 | printf("%02X ", *address++); | |
67 | if (!(++i % line_size) || (length == 0 && i % line_size)) { | |
68 | if (length == 0) { | |
69 | while (i++ % line_size) | |
70 | printf("__ "); | |
71 | } | |
72 | printf(" | "); /* right close */ | |
73 | while (line < address) { | |
74 | c = *line++; | |
75 | printf("%c", (c < 33 || c == 255) ? 0x2E : c); | |
76 | } | |
77 | printf("\n"); | |
78 | if (length > 0) | |
79 | printf("%s | ", prefix); | |
80 | } | |
81 | } | |
82 | } | |
83 | ||
30061915 AR |
84 | /* |
85 | * Unescape - process hexadecimal escape character | |
86 | * converts shell input "\x23" -> 0x23 | |
87 | */ | |
07eec628 | 88 | static int unescape(char *_dst, char *_src, size_t len) |
30061915 AR |
89 | { |
90 | int ret = 0; | |
a20874f7 | 91 | int match; |
30061915 AR |
92 | char *src = _src; |
93 | char *dst = _dst; | |
94 | unsigned int ch; | |
95 | ||
96 | while (*src) { | |
97 | if (*src == '\\' && *(src+1) == 'x') { | |
a20874f7 JC |
98 | match = sscanf(src + 2, "%2x", &ch); |
99 | if (!match) | |
100 | pabort("malformed input string"); | |
101 | ||
30061915 AR |
102 | src += 4; |
103 | *dst++ = (unsigned char)ch; | |
104 | } else { | |
105 | *dst++ = *src++; | |
106 | } | |
107 | ret++; | |
108 | } | |
109 | return ret; | |
110 | } | |
111 | ||
112 | static void transfer(int fd, uint8_t const *tx, uint8_t const *rx, size_t len) | |
22b238bd AV |
113 | { |
114 | int ret; | |
983b2788 | 115 | int out_fd; |
22b238bd AV |
116 | struct spi_ioc_transfer tr = { |
117 | .tx_buf = (unsigned long)tx, | |
118 | .rx_buf = (unsigned long)rx, | |
30061915 | 119 | .len = len, |
22b238bd AV |
120 | .delay_usecs = delay, |
121 | .speed_hz = speed, | |
122 | .bits_per_word = bits, | |
123 | }; | |
124 | ||
c2e78c34 GU |
125 | if (mode & SPI_TX_QUAD) |
126 | tr.tx_nbits = 4; | |
127 | else if (mode & SPI_TX_DUAL) | |
128 | tr.tx_nbits = 2; | |
129 | if (mode & SPI_RX_QUAD) | |
130 | tr.rx_nbits = 4; | |
131 | else if (mode & SPI_RX_DUAL) | |
132 | tr.rx_nbits = 2; | |
133 | if (!(mode & SPI_LOOP)) { | |
134 | if (mode & (SPI_TX_QUAD | SPI_TX_DUAL)) | |
135 | tr.rx_buf = 0; | |
136 | else if (mode & (SPI_RX_QUAD | SPI_RX_DUAL)) | |
137 | tr.tx_buf = 0; | |
138 | } | |
139 | ||
22b238bd | 140 | ret = ioctl(fd, SPI_IOC_MESSAGE(1), &tr); |
95b1ed2a | 141 | if (ret < 1) |
22b238bd AV |
142 | pabort("can't send spi message"); |
143 | ||
31a5c5a7 | 144 | if (verbose) |
30061915 | 145 | hex_dump(tx, len, 32, "TX"); |
983b2788 JC |
146 | |
147 | if (output_file) { | |
148 | out_fd = open(output_file, O_WRONLY | O_CREAT | O_TRUNC, 0666); | |
149 | if (out_fd < 0) | |
150 | pabort("could not open output file"); | |
151 | ||
152 | ret = write(out_fd, rx, len); | |
153 | if (ret != len) | |
edd3899c | 154 | pabort("not all bytes written to output file"); |
983b2788 JC |
155 | |
156 | close(out_fd); | |
157 | } | |
158 | ||
159 | if (verbose || !output_file) | |
160 | hex_dump(rx, len, 32, "RX"); | |
22b238bd AV |
161 | } |
162 | ||
b7ed698c | 163 | static void print_usage(const char *prog) |
22b238bd AV |
164 | { |
165 | printf("Usage: %s [-DsbdlHOLC3]\n", prog); | |
166 | puts(" -D --device device to use (default /dev/spidev1.1)\n" | |
167 | " -s --speed max speed (Hz)\n" | |
168 | " -d --delay delay (usec)\n" | |
b2cfc904 | 169 | " -b --bpw bits per word\n" |
7af475a5 | 170 | " -i --input input data from a file (e.g. \"test.bin\")\n" |
983b2788 | 171 | " -o --output output data to a file (e.g. \"results.bin\")\n" |
22b238bd AV |
172 | " -l --loop loopback\n" |
173 | " -H --cpha clock phase\n" | |
174 | " -O --cpol clock polarity\n" | |
175 | " -L --lsb least significant bit first\n" | |
176 | " -C --cs-high chip select active high\n" | |
925d16a2 | 177 | " -3 --3wire SI/SO signals shared\n" |
31a5c5a7 | 178 | " -v --verbose Verbose (show tx buffer)\n" |
30061915 | 179 | " -p Send data (e.g. \"1234\\xde\\xad\")\n" |
925d16a2 | 180 | " -N --no-cs no chip select\n" |
c2e78c34 GU |
181 | " -R --ready slave pulls low to pause\n" |
182 | " -2 --dual dual transfer\n" | |
183 | " -4 --quad quad transfer\n"); | |
22b238bd AV |
184 | exit(1); |
185 | } | |
186 | ||
b7ed698c | 187 | static void parse_opts(int argc, char *argv[]) |
22b238bd AV |
188 | { |
189 | while (1) { | |
cd58310d | 190 | static const struct option lopts[] = { |
22b238bd AV |
191 | { "device", 1, 0, 'D' }, |
192 | { "speed", 1, 0, 's' }, | |
193 | { "delay", 1, 0, 'd' }, | |
194 | { "bpw", 1, 0, 'b' }, | |
7af475a5 | 195 | { "input", 1, 0, 'i' }, |
983b2788 | 196 | { "output", 1, 0, 'o' }, |
22b238bd AV |
197 | { "loop", 0, 0, 'l' }, |
198 | { "cpha", 0, 0, 'H' }, | |
199 | { "cpol", 0, 0, 'O' }, | |
200 | { "lsb", 0, 0, 'L' }, | |
201 | { "cs-high", 0, 0, 'C' }, | |
202 | { "3wire", 0, 0, '3' }, | |
b55f627f DB |
203 | { "no-cs", 0, 0, 'N' }, |
204 | { "ready", 0, 0, 'R' }, | |
c2e78c34 | 205 | { "dual", 0, 0, '2' }, |
31a5c5a7 | 206 | { "verbose", 0, 0, 'v' }, |
c2e78c34 | 207 | { "quad", 0, 0, '4' }, |
22b238bd AV |
208 | { NULL, 0, 0, 0 }, |
209 | }; | |
210 | int c; | |
211 | ||
983b2788 | 212 | c = getopt_long(argc, argv, "D:s:d:b:i:o:lHOLC3NR24p:v", |
7af475a5 | 213 | lopts, NULL); |
22b238bd AV |
214 | |
215 | if (c == -1) | |
216 | break; | |
217 | ||
218 | switch (c) { | |
219 | case 'D': | |
220 | device = optarg; | |
221 | break; | |
222 | case 's': | |
223 | speed = atoi(optarg); | |
224 | break; | |
225 | case 'd': | |
226 | delay = atoi(optarg); | |
227 | break; | |
228 | case 'b': | |
229 | bits = atoi(optarg); | |
230 | break; | |
7af475a5 JC |
231 | case 'i': |
232 | input_file = optarg; | |
233 | break; | |
983b2788 JC |
234 | case 'o': |
235 | output_file = optarg; | |
236 | break; | |
22b238bd AV |
237 | case 'l': |
238 | mode |= SPI_LOOP; | |
239 | break; | |
240 | case 'H': | |
241 | mode |= SPI_CPHA; | |
242 | break; | |
243 | case 'O': | |
244 | mode |= SPI_CPOL; | |
245 | break; | |
246 | case 'L': | |
247 | mode |= SPI_LSB_FIRST; | |
248 | break; | |
249 | case 'C': | |
250 | mode |= SPI_CS_HIGH; | |
251 | break; | |
252 | case '3': | |
253 | mode |= SPI_3WIRE; | |
254 | break; | |
b55f627f DB |
255 | case 'N': |
256 | mode |= SPI_NO_CS; | |
257 | break; | |
31a5c5a7 AR |
258 | case 'v': |
259 | verbose = 1; | |
260 | break; | |
b55f627f DB |
261 | case 'R': |
262 | mode |= SPI_READY; | |
263 | break; | |
30061915 AR |
264 | case 'p': |
265 | input_tx = optarg; | |
266 | break; | |
c2e78c34 GU |
267 | case '2': |
268 | mode |= SPI_TX_DUAL; | |
269 | break; | |
270 | case '4': | |
271 | mode |= SPI_TX_QUAD; | |
272 | break; | |
22b238bd AV |
273 | default: |
274 | print_usage(argv[0]); | |
275 | break; | |
276 | } | |
277 | } | |
c2e78c34 GU |
278 | if (mode & SPI_LOOP) { |
279 | if (mode & SPI_TX_DUAL) | |
280 | mode |= SPI_RX_DUAL; | |
281 | if (mode & SPI_TX_QUAD) | |
282 | mode |= SPI_RX_QUAD; | |
283 | } | |
22b238bd AV |
284 | } |
285 | ||
5c437a40 JC |
286 | static void transfer_escaped_string(int fd, char *str) |
287 | { | |
0278b34b | 288 | size_t size = strlen(str); |
5c437a40 JC |
289 | uint8_t *tx; |
290 | uint8_t *rx; | |
291 | ||
292 | tx = malloc(size); | |
293 | if (!tx) | |
294 | pabort("can't allocate tx buffer"); | |
295 | ||
296 | rx = malloc(size); | |
297 | if (!rx) | |
298 | pabort("can't allocate rx buffer"); | |
299 | ||
300 | size = unescape((char *)tx, str, size); | |
301 | transfer(fd, tx, rx, size); | |
302 | free(rx); | |
303 | free(tx); | |
304 | } | |
305 | ||
7af475a5 JC |
306 | static void transfer_file(int fd, char *filename) |
307 | { | |
308 | ssize_t bytes; | |
309 | struct stat sb; | |
310 | int tx_fd; | |
311 | uint8_t *tx; | |
312 | uint8_t *rx; | |
313 | ||
314 | if (stat(filename, &sb) == -1) | |
315 | pabort("can't stat input file"); | |
316 | ||
317 | tx_fd = open(filename, O_RDONLY); | |
e634b76c | 318 | if (tx_fd < 0) |
7af475a5 JC |
319 | pabort("can't open input file"); |
320 | ||
321 | tx = malloc(sb.st_size); | |
322 | if (!tx) | |
323 | pabort("can't allocate tx buffer"); | |
324 | ||
325 | rx = malloc(sb.st_size); | |
326 | if (!rx) | |
327 | pabort("can't allocate rx buffer"); | |
328 | ||
329 | bytes = read(tx_fd, tx, sb.st_size); | |
330 | if (bytes != sb.st_size) | |
331 | pabort("failed to read input file"); | |
332 | ||
333 | transfer(fd, tx, rx, sb.st_size); | |
334 | free(rx); | |
335 | free(tx); | |
336 | close(tx_fd); | |
337 | } | |
338 | ||
22b238bd AV |
339 | int main(int argc, char *argv[]) |
340 | { | |
341 | int ret = 0; | |
342 | int fd; | |
343 | ||
344 | parse_opts(argc, argv); | |
345 | ||
346 | fd = open(device, O_RDWR); | |
347 | if (fd < 0) | |
348 | pabort("can't open device"); | |
349 | ||
350 | /* | |
351 | * spi mode | |
352 | */ | |
c2e78c34 | 353 | ret = ioctl(fd, SPI_IOC_WR_MODE32, &mode); |
22b238bd AV |
354 | if (ret == -1) |
355 | pabort("can't set spi mode"); | |
356 | ||
c2e78c34 | 357 | ret = ioctl(fd, SPI_IOC_RD_MODE32, &mode); |
22b238bd AV |
358 | if (ret == -1) |
359 | pabort("can't get spi mode"); | |
360 | ||
361 | /* | |
362 | * bits per word | |
363 | */ | |
364 | ret = ioctl(fd, SPI_IOC_WR_BITS_PER_WORD, &bits); | |
365 | if (ret == -1) | |
366 | pabort("can't set bits per word"); | |
367 | ||
368 | ret = ioctl(fd, SPI_IOC_RD_BITS_PER_WORD, &bits); | |
369 | if (ret == -1) | |
370 | pabort("can't get bits per word"); | |
371 | ||
372 | /* | |
373 | * max speed hz | |
374 | */ | |
375 | ret = ioctl(fd, SPI_IOC_WR_MAX_SPEED_HZ, &speed); | |
376 | if (ret == -1) | |
377 | pabort("can't set max speed hz"); | |
378 | ||
379 | ret = ioctl(fd, SPI_IOC_RD_MAX_SPEED_HZ, &speed); | |
380 | if (ret == -1) | |
381 | pabort("can't get max speed hz"); | |
382 | ||
c2e78c34 | 383 | printf("spi mode: 0x%x\n", mode); |
22b238bd AV |
384 | printf("bits per word: %d\n", bits); |
385 | printf("max speed: %d Hz (%d KHz)\n", speed, speed/1000); | |
386 | ||
7af475a5 JC |
387 | if (input_tx && input_file) |
388 | pabort("only one of -p and --input may be selected"); | |
389 | ||
5c437a40 JC |
390 | if (input_tx) |
391 | transfer_escaped_string(fd, input_tx); | |
7af475a5 JC |
392 | else if (input_file) |
393 | transfer_file(fd, input_file); | |
5c437a40 | 394 | else |
30061915 | 395 | transfer(fd, default_tx, default_rx, sizeof(default_tx)); |
22b238bd AV |
396 | |
397 | close(fd); | |
398 | ||
399 | return ret; | |
400 | } |