]>
Commit | Line | Data |
---|---|---|
1 | /* | |
2 | * This file is part of the libsigrok project. | |
3 | * | |
4 | * Copyright (C) 2010-2012 Bert Vermeulen <bert@biot.com> | |
5 | * Copyright (C) 2010-2012 Uwe Hermann <uwe@hermann-uwe.de> | |
6 | * Copyright (C) 2012 Alexandru Gagniuc <mr.nuke.me@gmail.com> | |
7 | * | |
8 | * This program is free software: you can redistribute it and/or modify | |
9 | * it under the terms of the GNU General Public License as published by | |
10 | * the Free Software Foundation, either version 3 of the License, or | |
11 | * (at your option) any later version. | |
12 | * | |
13 | * This program is distributed in the hope that it will be useful, | |
14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
16 | * GNU General Public License for more details. | |
17 | * | |
18 | * You should have received a copy of the GNU General Public License | |
19 | * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
20 | */ | |
21 | ||
22 | #include <string.h> | |
23 | #include <stdlib.h> | |
24 | #include <glib.h> | |
25 | #include <libserialport.h> | |
26 | #include "libsigrok.h" | |
27 | #include "libsigrok-internal.h" | |
28 | ||
29 | /* Message logging helpers with subsystem-specific prefix string. */ | |
30 | #define LOG_PREFIX "serial: " | |
31 | #define sr_log(l, s, args...) sr_log(l, LOG_PREFIX s, ## args) | |
32 | #define sr_spew(s, args...) sr_spew(LOG_PREFIX s, ## args) | |
33 | #define sr_dbg(s, args...) sr_dbg(LOG_PREFIX s, ## args) | |
34 | #define sr_info(s, args...) sr_info(LOG_PREFIX s, ## args) | |
35 | #define sr_warn(s, args...) sr_warn(LOG_PREFIX s, ## args) | |
36 | #define sr_err(s, args...) sr_err(LOG_PREFIX s, ## args) | |
37 | ||
38 | /** | |
39 | * Open the specified serial port. | |
40 | * | |
41 | * @param serial Previously initialized serial port structure. | |
42 | * @param flags Flags to use when opening the serial port. Possible flags | |
43 | * include SERIAL_RDWR, SERIAL_RDONLY, SERIAL_NONBLOCK. | |
44 | * | |
45 | * If the serial structure contains a serialcomm string, it will be | |
46 | * passed to serial_set_paramstr() after the port is opened. | |
47 | * | |
48 | * @return SR_OK on success, SR_ERR on failure. | |
49 | */ | |
50 | SR_PRIV int serial_open(struct sr_serial_dev_inst *serial, int flags) | |
51 | { | |
52 | int ret; | |
53 | char *error; | |
54 | ||
55 | if (!serial) { | |
56 | sr_dbg("Invalid serial port."); | |
57 | return SR_ERR; | |
58 | } | |
59 | ||
60 | sr_spew("Opening serial port '%s' (flags %d).", serial->port, flags); | |
61 | ||
62 | sp_get_port_by_name(serial->port, &serial->data); | |
63 | ||
64 | ret = sp_open(serial->data, flags); | |
65 | ||
66 | switch (ret) { | |
67 | case SP_ERR_ARG: | |
68 | sr_err("Attempt to open serial port with invalid parameters."); | |
69 | return SR_ERR_ARG; | |
70 | case SP_ERR_FAIL: | |
71 | error = sp_last_error_message(); | |
72 | sr_err("Error opening port: %s.", error); | |
73 | sp_free_error_message(error); | |
74 | return SR_ERR; | |
75 | } | |
76 | ||
77 | #ifndef _WIN32 | |
78 | serial->fd = serial->data->fd; | |
79 | #endif | |
80 | ||
81 | if (serial->serialcomm) | |
82 | return serial_set_paramstr(serial, serial->serialcomm); | |
83 | else | |
84 | return SR_OK; | |
85 | } | |
86 | ||
87 | /** | |
88 | * Close the specified serial port. | |
89 | * | |
90 | * @param serial Previously initialized serial port structure. | |
91 | * | |
92 | * @return SR_OK on success, SR_ERR on failure. | |
93 | */ | |
94 | SR_PRIV int serial_close(struct sr_serial_dev_inst *serial) | |
95 | { | |
96 | int ret; | |
97 | char *error; | |
98 | ||
99 | if (!serial) { | |
100 | sr_dbg("Invalid serial port."); | |
101 | return SR_ERR; | |
102 | } | |
103 | ||
104 | if (serial->fd == -1) { | |
105 | sr_dbg("Cannot close unopened serial port %s (fd %d).", | |
106 | serial->port, serial->fd); | |
107 | return SR_ERR; | |
108 | } | |
109 | ||
110 | sr_spew("Closing serial port %s (fd %d).", serial->port, serial->fd); | |
111 | ret = SR_OK; | |
112 | ||
113 | ret = sp_close(serial->data); | |
114 | ||
115 | switch (ret) { | |
116 | case SP_ERR_ARG: | |
117 | sr_err("Attempt to close an invalid serial port."); | |
118 | return SR_ERR_ARG; | |
119 | case SP_ERR_FAIL: | |
120 | error = sp_last_error_message(); | |
121 | sr_err("Error closing port: %s.", error); | |
122 | sp_free_error_message(error); | |
123 | return SR_ERR; | |
124 | } | |
125 | ||
126 | serial->fd = -1; | |
127 | ||
128 | return ret; | |
129 | } | |
130 | ||
131 | /** | |
132 | * Flush serial port buffers. | |
133 | * | |
134 | * @param serial Previously initialized serial port structure. | |
135 | * | |
136 | * @return SR_OK on success, SR_ERR on failure. | |
137 | */ | |
138 | SR_PRIV int serial_flush(struct sr_serial_dev_inst *serial) | |
139 | { | |
140 | int ret; | |
141 | char *error; | |
142 | ||
143 | if (!serial) { | |
144 | sr_dbg("Invalid serial port."); | |
145 | return SR_ERR; | |
146 | } | |
147 | ||
148 | if (serial->fd == -1) { | |
149 | sr_dbg("Cannot flush unopened serial port %s (fd %d).", | |
150 | serial->port, serial->fd); | |
151 | return SR_ERR; | |
152 | } | |
153 | ||
154 | sr_spew("Flushing serial port %s (fd %d).", serial->port, serial->fd); | |
155 | ||
156 | ret = sp_flush(serial->data); | |
157 | ||
158 | switch (ret) { | |
159 | case SP_ERR_ARG: | |
160 | sr_err("Attempt to flush an invalid serial port."); | |
161 | return SR_ERR_ARG; | |
162 | case SP_ERR_FAIL: | |
163 | error = sp_last_error_message(); | |
164 | sr_err("Error flushing port: %s.", error); | |
165 | sp_free_error_message(error); | |
166 | return SR_ERR; | |
167 | } | |
168 | ||
169 | return ret; | |
170 | } | |
171 | ||
172 | /** | |
173 | * Write a number of bytes to the specified serial port. | |
174 | * | |
175 | * @param serial Previously initialized serial port structure. | |
176 | * @param buf Buffer containing the bytes to write. | |
177 | * @param count Number of bytes to write. | |
178 | * | |
179 | * @return The number of bytes written, or -1 upon failure. | |
180 | */ | |
181 | SR_PRIV int serial_write(struct sr_serial_dev_inst *serial, | |
182 | const void *buf, size_t count) | |
183 | { | |
184 | ssize_t ret; | |
185 | char *error; | |
186 | ||
187 | if (!serial) { | |
188 | sr_dbg("Invalid serial port."); | |
189 | return -1; | |
190 | } | |
191 | ||
192 | if (serial->fd == -1) { | |
193 | sr_dbg("Cannot use unopened serial port %s (fd %d).", | |
194 | serial->port, serial->fd); | |
195 | return -1; | |
196 | } | |
197 | ||
198 | ret = sp_write(serial->data, buf, count); | |
199 | ||
200 | switch (ret) { | |
201 | case SP_ERR_ARG: | |
202 | sr_err("Attempted serial port write with invalid arguments."); | |
203 | return SR_ERR_ARG; | |
204 | case SP_ERR_FAIL: | |
205 | error = sp_last_error_message(); | |
206 | sr_err("Write error: %s.", error); | |
207 | sp_free_error_message(error); | |
208 | return SR_ERR; | |
209 | } | |
210 | ||
211 | sr_spew("Wrote %d/%d bytes (fd %d).", ret, count, serial->fd); | |
212 | ||
213 | return ret; | |
214 | } | |
215 | ||
216 | /** | |
217 | * Read a number of bytes from the specified serial port. | |
218 | * | |
219 | * @param serial Previously initialized serial port structure. | |
220 | * @param buf Buffer where to store the bytes that are read. | |
221 | * @param count The number of bytes to read. | |
222 | * | |
223 | * @return The number of bytes read, or -1 upon failure. | |
224 | */ | |
225 | SR_PRIV int serial_read(struct sr_serial_dev_inst *serial, void *buf, | |
226 | size_t count) | |
227 | { | |
228 | ssize_t ret; | |
229 | char *error; | |
230 | ||
231 | if (!serial) { | |
232 | sr_dbg("Invalid serial port."); | |
233 | return -1; | |
234 | } | |
235 | ||
236 | if (serial->fd == -1) { | |
237 | sr_dbg("Cannot use unopened serial port %s (fd %d).", | |
238 | serial->port, serial->fd); | |
239 | return -1; | |
240 | } | |
241 | ||
242 | ret = sp_read(serial->data, buf, count); | |
243 | ||
244 | switch (ret) { | |
245 | case SP_ERR_ARG: | |
246 | sr_err("Attempted serial port read with invalid arguments."); | |
247 | return SR_ERR_ARG; | |
248 | case SP_ERR_FAIL: | |
249 | error = sp_last_error_message(); | |
250 | sr_err("Read error: %s.", error); | |
251 | sp_free_error_message(error); | |
252 | return SR_ERR; | |
253 | } | |
254 | ||
255 | sr_spew("Read %d/%d bytes (fd %d).", ret, count, serial->fd); | |
256 | ||
257 | return ret; | |
258 | } | |
259 | ||
260 | /** | |
261 | * Set serial parameters for the specified serial port. | |
262 | * | |
263 | * @param serial Previously initialized serial port structure. | |
264 | * @param baudrate The baudrate to set. | |
265 | * @param bits The number of data bits to use. | |
266 | * @param parity The parity setting to use (0 = none, 1 = even, 2 = odd). | |
267 | * @param stopbits The number of stop bits to use (1 or 2). | |
268 | * @param flowcontrol The flow control settings to use (0 = none, 1 = RTS/CTS, | |
269 | * 2 = XON/XOFF). | |
270 | * | |
271 | * @return SR_OK upon success, SR_ERR upon failure. | |
272 | */ | |
273 | SR_PRIV int serial_set_params(struct sr_serial_dev_inst *serial, int baudrate, | |
274 | int bits, int parity, int stopbits, | |
275 | int flowcontrol, int rts, int dtr) | |
276 | { | |
277 | int ret; | |
278 | char *error; | |
279 | ||
280 | if (!serial) { | |
281 | sr_dbg("Invalid serial port."); | |
282 | return SR_ERR; | |
283 | } | |
284 | ||
285 | if (serial->fd == -1) { | |
286 | sr_dbg("Cannot configure unopened serial port %s (fd %d).", | |
287 | serial->port, serial->fd); | |
288 | return SR_ERR; | |
289 | } | |
290 | ||
291 | sr_spew("Setting serial parameters on port %s (fd %d).", serial->port, | |
292 | serial->fd); | |
293 | ||
294 | ret = sp_set_params(serial->data, baudrate, bits, parity, stopbits, | |
295 | flowcontrol, rts, dtr); | |
296 | ||
297 | switch (ret) { | |
298 | case SP_ERR_ARG: | |
299 | sr_err("Invalid arguments for setting serial port parameters."); | |
300 | return SR_ERR_ARG; | |
301 | case SP_ERR_FAIL: | |
302 | error = sp_last_error_message(); | |
303 | sr_err("Error setting serial port parameters: %s.", error); | |
304 | sp_free_error_message(error); | |
305 | return SR_ERR; | |
306 | } | |
307 | ||
308 | return SR_OK; | |
309 | } | |
310 | ||
311 | /** | |
312 | * Set serial parameters for the specified serial port. | |
313 | * | |
314 | * @param serial Previously initialized serial port structure. | |
315 | * @param paramstr A serial communication parameters string, in the form | |
316 | * of <speed>/<data bits><parity><stopbits><flow>, for example "9600/8n1" or | |
317 | * "600/7o2" or "460800/8n1/flow=2" where flow is 0 for none, 1 for rts/cts and 2 for xon/xoff. | |
318 | * | |
319 | * @return SR_OK upon success, SR_ERR upon failure. | |
320 | */ | |
321 | #define SERIAL_COMM_SPEC "^(\\d+)/([78])([neo])([12])(.*)$" | |
322 | SR_PRIV int serial_set_paramstr(struct sr_serial_dev_inst *serial, | |
323 | const char *paramstr) | |
324 | { | |
325 | GRegex *reg; | |
326 | GMatchInfo *match; | |
327 | int speed, databits, parity, stopbits, flow, rts, dtr, i; | |
328 | char *mstr, **opts, **kv; | |
329 | ||
330 | speed = databits = parity = stopbits = flow = 0; | |
331 | rts = dtr = -1; | |
332 | sr_spew("Parsing parameters from \"%s\".", paramstr); | |
333 | reg = g_regex_new(SERIAL_COMM_SPEC, 0, 0, NULL); | |
334 | if (g_regex_match(reg, paramstr, 0, &match)) { | |
335 | if ((mstr = g_match_info_fetch(match, 1))) | |
336 | speed = strtoul(mstr, NULL, 10); | |
337 | g_free(mstr); | |
338 | if ((mstr = g_match_info_fetch(match, 2))) | |
339 | databits = strtoul(mstr, NULL, 10); | |
340 | g_free(mstr); | |
341 | if ((mstr = g_match_info_fetch(match, 3))) { | |
342 | switch (mstr[0]) { | |
343 | case 'n': | |
344 | parity = SERIAL_PARITY_NONE; | |
345 | break; | |
346 | case 'e': | |
347 | parity = SERIAL_PARITY_EVEN; | |
348 | break; | |
349 | case 'o': | |
350 | parity = SERIAL_PARITY_ODD; | |
351 | break; | |
352 | } | |
353 | } | |
354 | g_free(mstr); | |
355 | if ((mstr = g_match_info_fetch(match, 4))) | |
356 | stopbits = strtoul(mstr, NULL, 10); | |
357 | g_free(mstr); | |
358 | if ((mstr = g_match_info_fetch(match, 5)) && mstr[0] != '\0') { | |
359 | if (mstr[0] != '/') { | |
360 | sr_dbg("missing separator before extra options"); | |
361 | speed = 0; | |
362 | } else { | |
363 | /* A set of "key=value" options separated by / */ | |
364 | opts = g_strsplit(mstr + 1, "/", 0); | |
365 | for (i = 0; opts[i]; i++) { | |
366 | kv = g_strsplit(opts[i], "=", 2); | |
367 | if (!strncmp(kv[0], "rts", 3)) { | |
368 | if (kv[1][0] == '1') | |
369 | rts = 1; | |
370 | else if (kv[1][0] == '0') | |
371 | rts = 0; | |
372 | else { | |
373 | sr_dbg("invalid value for rts: %c", kv[1][0]); | |
374 | speed = 0; | |
375 | } | |
376 | } else if (!strncmp(kv[0], "dtr", 3)) { | |
377 | if (kv[1][0] == '1') | |
378 | dtr = 1; | |
379 | else if (kv[1][0] == '0') | |
380 | dtr = 0; | |
381 | else { | |
382 | sr_dbg("invalid value for dtr: %c", kv[1][0]); | |
383 | speed = 0; | |
384 | } | |
385 | } else if (!strncmp(kv[0], "flow", 4)) { | |
386 | if (kv[1][0] == '0') | |
387 | flow = 0; | |
388 | else if (kv[1][0] == '1') | |
389 | flow = 1; | |
390 | else if (kv[1][0] == '2') | |
391 | flow = 2; | |
392 | else { | |
393 | sr_dbg("invalid value for flow: %c", kv[1][0]); | |
394 | speed = 0; | |
395 | } | |
396 | } | |
397 | g_strfreev(kv); | |
398 | } | |
399 | g_strfreev(opts); | |
400 | } | |
401 | } | |
402 | g_free(mstr); | |
403 | } | |
404 | g_match_info_unref(match); | |
405 | g_regex_unref(reg); | |
406 | ||
407 | if (speed) { | |
408 | return serial_set_params(serial, speed, databits, parity, | |
409 | stopbits, flow, rts, dtr); | |
410 | } else { | |
411 | sr_dbg("Could not infer speed from parameter string."); | |
412 | return SR_ERR_ARG; | |
413 | } | |
414 | } | |
415 | ||
416 | /** | |
417 | * Read a line from the specified serial port. | |
418 | * | |
419 | * @param serial Previously initialized serial port structure. | |
420 | * @param buf Buffer where to store the bytes that are read. | |
421 | * @param buflen Size of the buffer. | |
422 | * @param timeout_ms How long to wait for a line to come in. | |
423 | * | |
424 | * Reading stops when CR of LR is found, which is stripped from the buffer. | |
425 | * | |
426 | * @return SR_OK on success, SR_ERR on failure. | |
427 | */ | |
428 | SR_PRIV int serial_readline(struct sr_serial_dev_inst *serial, char **buf, | |
429 | int *buflen, gint64 timeout_ms) | |
430 | { | |
431 | gint64 start; | |
432 | int maxlen, len; | |
433 | ||
434 | if (!serial || serial->fd == -1) { | |
435 | sr_dbg("Invalid serial port."); | |
436 | return SR_ERR; | |
437 | } | |
438 | ||
439 | if (serial->fd == -1) { | |
440 | sr_dbg("Cannot use unopened serial port %s (fd %d).", | |
441 | serial->port, serial->fd); | |
442 | return -1; | |
443 | } | |
444 | ||
445 | timeout_ms *= 1000; | |
446 | start = g_get_monotonic_time(); | |
447 | ||
448 | maxlen = *buflen; | |
449 | *buflen = len = 0; | |
450 | while(1) { | |
451 | len = maxlen - *buflen - 1; | |
452 | if (len < 1) | |
453 | break; | |
454 | len = serial_read(serial, *buf + *buflen, 1); | |
455 | if (len > 0) { | |
456 | *buflen += len; | |
457 | *(*buf + *buflen) = '\0'; | |
458 | if (*buflen > 0 && (*(*buf + *buflen - 1) == '\r' | |
459 | || *(*buf + *buflen - 1) == '\n')) { | |
460 | /* Strip CR/LF and terminate. */ | |
461 | *(*buf + --*buflen) = '\0'; | |
462 | break; | |
463 | } | |
464 | } | |
465 | if (g_get_monotonic_time() - start > timeout_ms) | |
466 | /* Timeout */ | |
467 | break; | |
468 | if (len < 1) | |
469 | g_usleep(2000); | |
470 | } | |
471 | if (*buflen) | |
472 | sr_dbg("Received %d: '%s'.", *buflen, *buf); | |
473 | ||
474 | return SR_OK; | |
475 | } | |
476 | ||
477 | /** | |
478 | * Try to find a valid packet in a serial data stream. | |
479 | * | |
480 | * @param serial Previously initialized serial port structure. | |
481 | * @param buf Buffer containing the bytes to write. | |
482 | * @param count Size of the buffer. | |
483 | * @param packet_size Size, in bytes, of a valid packet. | |
484 | * @param is_valid Callback that assesses whether the packet is valid or not. | |
485 | * @param timeout_ms The timeout after which, if no packet is detected, to | |
486 | * abort scanning. | |
487 | * @param baudrate The baudrate of the serial port. This parameter is not | |
488 | * critical, but it helps fine tune the serial port polling | |
489 | * delay. | |
490 | * | |
491 | * @return SR_OK if a valid packet is found within the given timeout, | |
492 | * SR_ERR upon failure. | |
493 | */ | |
494 | SR_PRIV int serial_stream_detect(struct sr_serial_dev_inst *serial, | |
495 | uint8_t *buf, size_t *buflen, | |
496 | size_t packet_size, packet_valid_t is_valid, | |
497 | uint64_t timeout_ms, int baudrate) | |
498 | { | |
499 | uint64_t start, time, byte_delay_us; | |
500 | size_t ibuf, i, maxlen; | |
501 | int len; | |
502 | ||
503 | maxlen = *buflen; | |
504 | ||
505 | sr_dbg("Detecting packets on FD %d (timeout = %" PRIu64 | |
506 | "ms, baudrate = %d).", serial->fd, timeout_ms, baudrate); | |
507 | ||
508 | if (maxlen < (packet_size / 2) ) { | |
509 | sr_err("Buffer size must be at least twice the packet size."); | |
510 | return SR_ERR; | |
511 | } | |
512 | ||
513 | /* Assume 8n1 transmission. That is 10 bits for every byte. */ | |
514 | byte_delay_us = 10 * (1000000 / baudrate); | |
515 | start = g_get_monotonic_time(); | |
516 | ||
517 | i = ibuf = len = 0; | |
518 | while (ibuf < maxlen) { | |
519 | len = serial_read(serial, &buf[ibuf], 1); | |
520 | if (len > 0) { | |
521 | ibuf += len; | |
522 | } else if (len == 0) { | |
523 | /* No logging, already done in serial_read(). */ | |
524 | } else { | |
525 | /* Error reading byte, but continuing anyway. */ | |
526 | } | |
527 | ||
528 | time = g_get_monotonic_time() - start; | |
529 | time /= 1000; | |
530 | ||
531 | if ((ibuf - i) >= packet_size) { | |
532 | /* We have at least a packet's worth of data. */ | |
533 | if (is_valid(&buf[i])) { | |
534 | sr_spew("Found valid %d-byte packet after " | |
535 | "%" PRIu64 "ms.", (ibuf - i), time); | |
536 | *buflen = ibuf; | |
537 | return SR_OK; | |
538 | } else { | |
539 | sr_spew("Got %d bytes, but not a valid " | |
540 | "packet.", (ibuf - i)); | |
541 | } | |
542 | /* Not a valid packet. Continue searching. */ | |
543 | i++; | |
544 | } | |
545 | if (time >= timeout_ms) { | |
546 | /* Timeout */ | |
547 | sr_dbg("Detection timed out after %dms.", time); | |
548 | break; | |
549 | } | |
550 | if (len < 1) | |
551 | g_usleep(byte_delay_us); | |
552 | } | |
553 | ||
554 | *buflen = ibuf; | |
555 | ||
556 | sr_err("Didn't find a valid packet (read %d bytes).", *buflen); | |
557 | ||
558 | return SR_ERR; | |
559 | } |