]>
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 | int sp_flags = 0; | |
55 | ||
56 | if (!serial) { | |
57 | sr_dbg("Invalid serial port."); | |
58 | return SR_ERR; | |
59 | } | |
60 | ||
61 | sr_spew("Opening serial port '%s' (flags %d).", serial->port, flags); | |
62 | ||
63 | sp_get_port_by_name(serial->port, &serial->data); | |
64 | ||
65 | if (flags & SERIAL_RDWR) | |
66 | sp_flags = (SP_MODE_READ | SP_MODE_WRITE); | |
67 | else if (flags & SERIAL_RDONLY) | |
68 | sp_flags = SP_MODE_READ; | |
69 | if (flags & SERIAL_NONBLOCK) | |
70 | sp_flags |= SP_MODE_NONBLOCK; | |
71 | ||
72 | ret = sp_open(serial->data, sp_flags); | |
73 | ||
74 | switch (ret) { | |
75 | case SP_ERR_ARG: | |
76 | sr_err("Attempt to open serial port with invalid parameters."); | |
77 | return SR_ERR_ARG; | |
78 | case SP_ERR_FAIL: | |
79 | error = sp_last_error_message(); | |
80 | sr_err("Error opening port: %s.", error); | |
81 | sp_free_error_message(error); | |
82 | return SR_ERR; | |
83 | } | |
84 | ||
85 | #ifndef _WIN32 | |
86 | serial->fd = serial->data->fd; | |
87 | #endif | |
88 | ||
89 | if (serial->serialcomm) | |
90 | return serial_set_paramstr(serial, serial->serialcomm); | |
91 | else | |
92 | return SR_OK; | |
93 | } | |
94 | ||
95 | /** | |
96 | * Close the specified serial port. | |
97 | * | |
98 | * @param serial Previously initialized serial port structure. | |
99 | * | |
100 | * @return SR_OK on success, SR_ERR on failure. | |
101 | */ | |
102 | SR_PRIV int serial_close(struct sr_serial_dev_inst *serial) | |
103 | { | |
104 | int ret; | |
105 | char *error; | |
106 | ||
107 | if (!serial) { | |
108 | sr_dbg("Invalid serial port."); | |
109 | return SR_ERR; | |
110 | } | |
111 | ||
112 | if (serial->fd == -1) { | |
113 | sr_dbg("Cannot close unopened serial port %s (fd %d).", | |
114 | serial->port, serial->fd); | |
115 | return SR_ERR; | |
116 | } | |
117 | ||
118 | sr_spew("Closing serial port %s (fd %d).", serial->port, serial->fd); | |
119 | ||
120 | ret = sp_close(serial->data); | |
121 | sp_free_port(serial->data); | |
122 | ||
123 | switch (ret) { | |
124 | case SP_ERR_ARG: | |
125 | sr_err("Attempt to close an invalid serial port."); | |
126 | return SR_ERR_ARG; | |
127 | case SP_ERR_FAIL: | |
128 | error = sp_last_error_message(); | |
129 | sr_err("Error closing port: %s.", error); | |
130 | sp_free_error_message(error); | |
131 | return SR_ERR; | |
132 | } | |
133 | ||
134 | serial->fd = -1; | |
135 | ||
136 | return SR_OK; | |
137 | } | |
138 | ||
139 | /** | |
140 | * Flush serial port buffers. | |
141 | * | |
142 | * @param serial Previously initialized serial port structure. | |
143 | * | |
144 | * @return SR_OK on success, SR_ERR on failure. | |
145 | */ | |
146 | SR_PRIV int serial_flush(struct sr_serial_dev_inst *serial) | |
147 | { | |
148 | int ret; | |
149 | char *error; | |
150 | ||
151 | if (!serial) { | |
152 | sr_dbg("Invalid serial port."); | |
153 | return SR_ERR; | |
154 | } | |
155 | ||
156 | if (serial->fd == -1) { | |
157 | sr_dbg("Cannot flush unopened serial port %s (fd %d).", | |
158 | serial->port, serial->fd); | |
159 | return SR_ERR; | |
160 | } | |
161 | ||
162 | sr_spew("Flushing serial port %s (fd %d).", serial->port, serial->fd); | |
163 | ||
164 | ret = sp_flush(serial->data, SP_BUF_BOTH); | |
165 | ||
166 | switch (ret) { | |
167 | case SP_ERR_ARG: | |
168 | sr_err("Attempt to flush an invalid serial port."); | |
169 | return SR_ERR_ARG; | |
170 | case SP_ERR_FAIL: | |
171 | error = sp_last_error_message(); | |
172 | sr_err("Error flushing port: %s.", error); | |
173 | sp_free_error_message(error); | |
174 | return SR_ERR; | |
175 | } | |
176 | ||
177 | return SR_OK; | |
178 | } | |
179 | ||
180 | /** | |
181 | * Write a number of bytes to the specified serial port. | |
182 | * | |
183 | * @param serial Previously initialized serial port structure. | |
184 | * @param buf Buffer containing the bytes to write. | |
185 | * @param count Number of bytes to write. | |
186 | * | |
187 | * @return The number of bytes written, or a negative error code upon failure. | |
188 | */ | |
189 | SR_PRIV int serial_write(struct sr_serial_dev_inst *serial, | |
190 | const void *buf, size_t count) | |
191 | { | |
192 | ssize_t ret; | |
193 | char *error; | |
194 | ||
195 | if (!serial) { | |
196 | sr_dbg("Invalid serial port."); | |
197 | return SR_ERR; | |
198 | } | |
199 | ||
200 | if (serial->fd == -1) { | |
201 | sr_dbg("Cannot use unopened serial port %s (fd %d).", | |
202 | serial->port, serial->fd); | |
203 | return SR_ERR; | |
204 | } | |
205 | ||
206 | ret = sp_write(serial->data, buf, count); | |
207 | ||
208 | switch (ret) { | |
209 | case SP_ERR_ARG: | |
210 | sr_err("Attempted serial port write with invalid arguments."); | |
211 | return SR_ERR_ARG; | |
212 | case SP_ERR_FAIL: | |
213 | error = sp_last_error_message(); | |
214 | sr_err("Write error: %s.", error); | |
215 | sp_free_error_message(error); | |
216 | return SR_ERR; | |
217 | } | |
218 | ||
219 | sr_spew("Wrote %d/%d bytes (fd %d).", ret, count, serial->fd); | |
220 | ||
221 | return ret; | |
222 | } | |
223 | ||
224 | /** | |
225 | * Read a number of bytes from the specified serial port. | |
226 | * | |
227 | * @param serial Previously initialized serial port structure. | |
228 | * @param buf Buffer where to store the bytes that are read. | |
229 | * @param count The number of bytes to read. | |
230 | * | |
231 | * @return The number of bytes read, or a negative error code upon failure. | |
232 | */ | |
233 | SR_PRIV int serial_read(struct sr_serial_dev_inst *serial, void *buf, | |
234 | size_t count) | |
235 | { | |
236 | ssize_t ret; | |
237 | //char *error; | |
238 | ||
239 | if (!serial) { | |
240 | sr_dbg("Invalid serial port."); | |
241 | return SR_ERR; | |
242 | } | |
243 | ||
244 | if (serial->fd == -1) { | |
245 | sr_dbg("Cannot use unopened serial port %s (fd %d).", | |
246 | serial->port, serial->fd); | |
247 | return SR_ERR; | |
248 | } | |
249 | ||
250 | ret = sp_read(serial->data, buf, count); | |
251 | ||
252 | switch (ret) { | |
253 | case SP_ERR_ARG: | |
254 | sr_err("Attempted serial port read with invalid arguments."); | |
255 | return SR_ERR_ARG; | |
256 | // Temporarily disabled, will come back later. | |
257 | // case SP_ERR_FAIL: | |
258 | // error = sp_last_error_message(); | |
259 | // sr_err("Read error: %s.", error); | |
260 | // sp_free_error_message(error); | |
261 | // return SR_ERR; | |
262 | } | |
263 | ||
264 | if (ret >= 0) | |
265 | sr_spew("Read %d/%d bytes (fd %d).", ret, count, serial->fd); | |
266 | ||
267 | return ret; | |
268 | } | |
269 | ||
270 | /** | |
271 | * Set serial parameters for the specified serial port. | |
272 | * | |
273 | * @param serial Previously initialized serial port structure. | |
274 | * @param baudrate The baudrate to set. | |
275 | * @param bits The number of data bits to use. | |
276 | * @param parity The parity setting to use (0 = none, 1 = even, 2 = odd). | |
277 | * @param stopbits The number of stop bits to use (1 or 2). | |
278 | * @param flowcontrol The flow control settings to use (0 = none, 1 = RTS/CTS, | |
279 | * 2 = XON/XOFF). | |
280 | * | |
281 | * @return SR_OK upon success, SR_ERR upon failure. | |
282 | */ | |
283 | SR_PRIV int serial_set_params(struct sr_serial_dev_inst *serial, int baudrate, | |
284 | int bits, int parity, int stopbits, | |
285 | int flowcontrol, int rts, int dtr) | |
286 | { | |
287 | int ret; | |
288 | char *error; | |
289 | struct sp_port_config config; | |
290 | ||
291 | if (!serial) { | |
292 | sr_dbg("Invalid serial port."); | |
293 | return SR_ERR; | |
294 | } | |
295 | ||
296 | if (serial->fd == -1) { | |
297 | sr_dbg("Cannot configure unopened serial port %s (fd %d).", | |
298 | serial->port, serial->fd); | |
299 | return SR_ERR; | |
300 | } | |
301 | ||
302 | sr_spew("Setting serial parameters on port %s (fd %d).", serial->port, | |
303 | serial->fd); | |
304 | ||
305 | config.baudrate = baudrate; | |
306 | config.bits = bits; | |
307 | switch (parity) { | |
308 | case 0: | |
309 | config.parity = SP_PARITY_NONE; | |
310 | break; | |
311 | case 1: | |
312 | config.parity = SP_PARITY_EVEN; | |
313 | break; | |
314 | case 2: | |
315 | config.parity = SP_PARITY_ODD; | |
316 | break; | |
317 | default: | |
318 | return SR_ERR_ARG; | |
319 | } | |
320 | config.stopbits = stopbits; | |
321 | config.rts = flowcontrol == 1 ? SP_RTS_FLOW_CONTROL : rts; | |
322 | config.cts = flowcontrol == 1 ? SP_CTS_FLOW_CONTROL : SP_CTS_IGNORE; | |
323 | config.dtr = dtr; | |
324 | config.dsr = SP_DSR_IGNORE; | |
325 | config.xon_xoff = flowcontrol == 2 ? SP_XONXOFF_INOUT : SP_XONXOFF_DISABLED; | |
326 | ||
327 | ret = sp_set_config(serial->data, &config); | |
328 | ||
329 | switch (ret) { | |
330 | case SP_ERR_ARG: | |
331 | sr_err("Invalid arguments for setting serial port parameters."); | |
332 | return SR_ERR_ARG; | |
333 | case SP_ERR_FAIL: | |
334 | error = sp_last_error_message(); | |
335 | sr_err("Error setting serial port parameters: %s.", error); | |
336 | sp_free_error_message(error); | |
337 | return SR_ERR; | |
338 | } | |
339 | ||
340 | return SR_OK; | |
341 | } | |
342 | ||
343 | /** | |
344 | * Set serial parameters for the specified serial port. | |
345 | * | |
346 | * @param serial Previously initialized serial port structure. | |
347 | * @param paramstr A serial communication parameters string, in the form | |
348 | * of <speed>/<data bits><parity><stopbits><flow>, for example "9600/8n1" or | |
349 | * "600/7o2" or "460800/8n1/flow=2" where flow is 0 for none, 1 for rts/cts and 2 for xon/xoff. | |
350 | * | |
351 | * @return SR_OK upon success, SR_ERR upon failure. | |
352 | */ | |
353 | #define SERIAL_COMM_SPEC "^(\\d+)/([78])([neo])([12])(.*)$" | |
354 | SR_PRIV int serial_set_paramstr(struct sr_serial_dev_inst *serial, | |
355 | const char *paramstr) | |
356 | { | |
357 | GRegex *reg; | |
358 | GMatchInfo *match; | |
359 | int speed, databits, parity, stopbits, flow, rts, dtr, i; | |
360 | char *mstr, **opts, **kv; | |
361 | ||
362 | speed = databits = parity = stopbits = flow = 0; | |
363 | rts = dtr = -1; | |
364 | sr_spew("Parsing parameters from \"%s\".", paramstr); | |
365 | reg = g_regex_new(SERIAL_COMM_SPEC, 0, 0, NULL); | |
366 | if (g_regex_match(reg, paramstr, 0, &match)) { | |
367 | if ((mstr = g_match_info_fetch(match, 1))) | |
368 | speed = strtoul(mstr, NULL, 10); | |
369 | g_free(mstr); | |
370 | if ((mstr = g_match_info_fetch(match, 2))) | |
371 | databits = strtoul(mstr, NULL, 10); | |
372 | g_free(mstr); | |
373 | if ((mstr = g_match_info_fetch(match, 3))) { | |
374 | switch (mstr[0]) { | |
375 | case 'n': | |
376 | parity = SERIAL_PARITY_NONE; | |
377 | break; | |
378 | case 'e': | |
379 | parity = SERIAL_PARITY_EVEN; | |
380 | break; | |
381 | case 'o': | |
382 | parity = SERIAL_PARITY_ODD; | |
383 | break; | |
384 | } | |
385 | } | |
386 | g_free(mstr); | |
387 | if ((mstr = g_match_info_fetch(match, 4))) | |
388 | stopbits = strtoul(mstr, NULL, 10); | |
389 | g_free(mstr); | |
390 | if ((mstr = g_match_info_fetch(match, 5)) && mstr[0] != '\0') { | |
391 | if (mstr[0] != '/') { | |
392 | sr_dbg("missing separator before extra options"); | |
393 | speed = 0; | |
394 | } else { | |
395 | /* A set of "key=value" options separated by / */ | |
396 | opts = g_strsplit(mstr + 1, "/", 0); | |
397 | for (i = 0; opts[i]; i++) { | |
398 | kv = g_strsplit(opts[i], "=", 2); | |
399 | if (!strncmp(kv[0], "rts", 3)) { | |
400 | if (kv[1][0] == '1') | |
401 | rts = 1; | |
402 | else if (kv[1][0] == '0') | |
403 | rts = 0; | |
404 | else { | |
405 | sr_dbg("invalid value for rts: %c", kv[1][0]); | |
406 | speed = 0; | |
407 | } | |
408 | } else if (!strncmp(kv[0], "dtr", 3)) { | |
409 | if (kv[1][0] == '1') | |
410 | dtr = 1; | |
411 | else if (kv[1][0] == '0') | |
412 | dtr = 0; | |
413 | else { | |
414 | sr_dbg("invalid value for dtr: %c", kv[1][0]); | |
415 | speed = 0; | |
416 | } | |
417 | } else if (!strncmp(kv[0], "flow", 4)) { | |
418 | if (kv[1][0] == '0') | |
419 | flow = 0; | |
420 | else if (kv[1][0] == '1') | |
421 | flow = 1; | |
422 | else if (kv[1][0] == '2') | |
423 | flow = 2; | |
424 | else { | |
425 | sr_dbg("invalid value for flow: %c", kv[1][0]); | |
426 | speed = 0; | |
427 | } | |
428 | } | |
429 | g_strfreev(kv); | |
430 | } | |
431 | g_strfreev(opts); | |
432 | } | |
433 | } | |
434 | g_free(mstr); | |
435 | } | |
436 | g_match_info_unref(match); | |
437 | g_regex_unref(reg); | |
438 | ||
439 | if (speed) { | |
440 | return serial_set_params(serial, speed, databits, parity, | |
441 | stopbits, flow, rts, dtr); | |
442 | } else { | |
443 | sr_dbg("Could not infer speed from parameter string."); | |
444 | return SR_ERR_ARG; | |
445 | } | |
446 | } | |
447 | ||
448 | /** | |
449 | * Read a line from the specified serial port. | |
450 | * | |
451 | * @param serial Previously initialized serial port structure. | |
452 | * @param buf Buffer where to store the bytes that are read. | |
453 | * @param buflen Size of the buffer. | |
454 | * @param timeout_ms How long to wait for a line to come in. | |
455 | * | |
456 | * Reading stops when CR of LR is found, which is stripped from the buffer. | |
457 | * | |
458 | * @return SR_OK on success, SR_ERR on failure. | |
459 | */ | |
460 | SR_PRIV int serial_readline(struct sr_serial_dev_inst *serial, char **buf, | |
461 | int *buflen, gint64 timeout_ms) | |
462 | { | |
463 | gint64 start; | |
464 | int maxlen, len; | |
465 | ||
466 | if (!serial || serial->fd == -1) { | |
467 | sr_dbg("Invalid serial port."); | |
468 | return SR_ERR; | |
469 | } | |
470 | ||
471 | if (serial->fd == -1) { | |
472 | sr_dbg("Cannot use unopened serial port %s (fd %d).", | |
473 | serial->port, serial->fd); | |
474 | return -1; | |
475 | } | |
476 | ||
477 | timeout_ms *= 1000; | |
478 | start = g_get_monotonic_time(); | |
479 | ||
480 | maxlen = *buflen; | |
481 | *buflen = len = 0; | |
482 | while(1) { | |
483 | len = maxlen - *buflen - 1; | |
484 | if (len < 1) | |
485 | break; | |
486 | len = serial_read(serial, *buf + *buflen, 1); | |
487 | if (len > 0) { | |
488 | *buflen += len; | |
489 | *(*buf + *buflen) = '\0'; | |
490 | if (*buflen > 0 && (*(*buf + *buflen - 1) == '\r' | |
491 | || *(*buf + *buflen - 1) == '\n')) { | |
492 | /* Strip CR/LF and terminate. */ | |
493 | *(*buf + --*buflen) = '\0'; | |
494 | break; | |
495 | } | |
496 | } | |
497 | if (g_get_monotonic_time() - start > timeout_ms) | |
498 | /* Timeout */ | |
499 | break; | |
500 | if (len < 1) | |
501 | g_usleep(2000); | |
502 | } | |
503 | if (*buflen) | |
504 | sr_dbg("Received %d: '%s'.", *buflen, *buf); | |
505 | ||
506 | return SR_OK; | |
507 | } | |
508 | ||
509 | /** | |
510 | * Try to find a valid packet in a serial data stream. | |
511 | * | |
512 | * @param serial Previously initialized serial port structure. | |
513 | * @param buf Buffer containing the bytes to write. | |
514 | * @param count Size of the buffer. | |
515 | * @param packet_size Size, in bytes, of a valid packet. | |
516 | * @param is_valid Callback that assesses whether the packet is valid or not. | |
517 | * @param timeout_ms The timeout after which, if no packet is detected, to | |
518 | * abort scanning. | |
519 | * @param baudrate The baudrate of the serial port. This parameter is not | |
520 | * critical, but it helps fine tune the serial port polling | |
521 | * delay. | |
522 | * | |
523 | * @return SR_OK if a valid packet is found within the given timeout, | |
524 | * SR_ERR upon failure. | |
525 | */ | |
526 | SR_PRIV int serial_stream_detect(struct sr_serial_dev_inst *serial, | |
527 | uint8_t *buf, size_t *buflen, | |
528 | size_t packet_size, packet_valid_t is_valid, | |
529 | uint64_t timeout_ms, int baudrate) | |
530 | { | |
531 | uint64_t start, time, byte_delay_us; | |
532 | size_t ibuf, i, maxlen; | |
533 | int len; | |
534 | ||
535 | maxlen = *buflen; | |
536 | ||
537 | sr_dbg("Detecting packets on FD %d (timeout = %" PRIu64 | |
538 | "ms, baudrate = %d).", serial->fd, timeout_ms, baudrate); | |
539 | ||
540 | if (maxlen < (packet_size / 2) ) { | |
541 | sr_err("Buffer size must be at least twice the packet size."); | |
542 | return SR_ERR; | |
543 | } | |
544 | ||
545 | /* Assume 8n1 transmission. That is 10 bits for every byte. */ | |
546 | byte_delay_us = 10 * (1000000 / baudrate); | |
547 | start = g_get_monotonic_time(); | |
548 | ||
549 | i = ibuf = len = 0; | |
550 | while (ibuf < maxlen) { | |
551 | len = serial_read(serial, &buf[ibuf], 1); | |
552 | if (len > 0) { | |
553 | ibuf += len; | |
554 | } else if (len == 0) { | |
555 | /* No logging, already done in serial_read(). */ | |
556 | } else { | |
557 | /* Error reading byte, but continuing anyway. */ | |
558 | } | |
559 | ||
560 | time = g_get_monotonic_time() - start; | |
561 | time /= 1000; | |
562 | ||
563 | if ((ibuf - i) >= packet_size) { | |
564 | /* We have at least a packet's worth of data. */ | |
565 | if (is_valid(&buf[i])) { | |
566 | sr_spew("Found valid %d-byte packet after " | |
567 | "%" PRIu64 "ms.", (ibuf - i), time); | |
568 | *buflen = ibuf; | |
569 | return SR_OK; | |
570 | } else { | |
571 | sr_spew("Got %d bytes, but not a valid " | |
572 | "packet.", (ibuf - i)); | |
573 | } | |
574 | /* Not a valid packet. Continue searching. */ | |
575 | i++; | |
576 | } | |
577 | if (time >= timeout_ms) { | |
578 | /* Timeout */ | |
579 | sr_dbg("Detection timed out after %dms.", time); | |
580 | break; | |
581 | } | |
582 | if (len < 1) | |
583 | g_usleep(byte_delay_us); | |
584 | } | |
585 | ||
586 | *buflen = ibuf; | |
587 | ||
588 | sr_err("Didn't find a valid packet (read %d bytes).", *buflen); | |
589 | ||
590 | return SR_ERR; | |
591 | } |