]> sigrok.org Git - libsigrok.git/blob - hardware/serial-dmm/api.c
987fafbfadbd0f5fb5ff0ce44b0b604103b28f1c
[libsigrok.git] / hardware / serial-dmm / api.c
1 /*
2  * This file is part of the sigrok project.
3  *
4  * Copyright (C) 2012 Bert Vermeulen <bert@biot.com>
5  * Copyright (C) 2012 Alexandru Gagniuc <mr.nuke.me@gmail.com>
6  * Copyright (C) 2012 Uwe Hermann <uwe@hermann-uwe.de>
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 <sys/types.h>
23 #include <sys/stat.h>
24 #include <fcntl.h>
25 #include <string.h>
26 #include <errno.h>
27 #include <glib.h>
28 #include "libsigrok.h"
29 #include "libsigrok-internal.h"
30 #include "protocol.h"
31
32 static const int hwopts[] = {
33         SR_CONF_CONN,
34         SR_CONF_SERIALCOMM,
35         0,
36 };
37
38 static const int hwcaps[] = {
39         SR_CONF_MULTIMETER,
40         SR_CONF_LIMIT_SAMPLES,
41         SR_CONF_LIMIT_MSEC,
42         SR_CONF_CONTINUOUS,
43         0,
44 };
45
46 SR_PRIV struct sr_dev_driver digitek_dt4000zc_driver_info;
47 SR_PRIV struct sr_dev_driver tekpower_tp4000zc_driver_info;
48 SR_PRIV struct sr_dev_driver metex_me31_driver_info;
49 SR_PRIV struct sr_dev_driver peaktech_3410_driver_info;
50 SR_PRIV struct sr_dev_driver mastech_mas345_driver_info;
51 SR_PRIV struct sr_dev_driver va_va18b_driver_info;
52 SR_PRIV struct sr_dev_driver metex_m3640d_driver_info;
53 SR_PRIV struct sr_dev_driver peaktech_4370_driver_info;
54 SR_PRIV struct sr_dev_driver pce_pce_dm32_driver_info;
55 SR_PRIV struct sr_dev_driver radioshack_22_168_driver_info;
56 SR_PRIV struct sr_dev_driver radioshack_22_812_driver_info;
57 SR_PRIV struct sr_dev_driver voltcraft_vc820_ser_driver_info;
58 SR_PRIV struct sr_dev_driver voltcraft_vc840_ser_driver_info;
59 SR_PRIV struct sr_dev_driver uni_t_ut61e_ser_driver_info;
60
61 SR_PRIV struct dmm_info dmms[] = {
62         {
63                 "Digitek", "DT4000ZC", "2400/8n1", 2400,
64                 FS9721_PACKET_SIZE, NULL,
65                 sr_fs9721_packet_valid, sr_fs9721_parse,
66                 dmm_details_dt4000zc,
67                 &digitek_dt4000zc_driver_info, receive_data_DIGITEK_DT4000ZC,
68         },
69         {
70                 "TekPower", "TP4000ZC", "2400/8n1", 2400,
71                 FS9721_PACKET_SIZE, NULL,
72                 sr_fs9721_packet_valid, sr_fs9721_parse,
73                 dmm_details_tp4000zc,
74                 &tekpower_tp4000zc_driver_info, receive_data_TEKPOWER_TP4000ZC,
75         },
76         {
77                 "Metex", "ME-31", "600/7n2/rts=0/dtr=1", 600,
78                 METEX14_PACKET_SIZE, sr_metex14_packet_request,
79                 sr_metex14_packet_valid, sr_metex14_parse,
80                 NULL,
81                 &metex_me31_driver_info, receive_data_METEX_ME31,
82         },
83         {
84                 "Peaktech", "3410", "600/7n2/rts=0/dtr=1", 600,
85                 METEX14_PACKET_SIZE, sr_metex14_packet_request,
86                 sr_metex14_packet_valid, sr_metex14_parse,
87                 NULL,
88                 &peaktech_3410_driver_info, receive_data_PEAKTECH_3410,
89         },
90         {
91                 "MASTECH", "MAS345", "600/7n2/rts=0/dtr=1", 600,
92                 METEX14_PACKET_SIZE, sr_metex14_packet_request,
93                 sr_metex14_packet_valid, sr_metex14_parse,
94                 NULL,
95                 &mastech_mas345_driver_info, receive_data_MASTECH_MAS345,
96         },
97         {
98                 "V&A", "VA18B", "2400/8n1", 2400,
99                 FS9721_PACKET_SIZE, NULL,
100                 sr_fs9721_packet_valid, sr_fs9721_parse,
101                 dmm_details_va18b,
102                 &va_va18b_driver_info, receive_data_VA_VA18B,
103         },
104         {
105                 "Metex", "M-3640D", "1200/7n2/rts=0/dtr=1", 1200,
106                 METEX14_PACKET_SIZE, sr_metex14_packet_request,
107                 sr_metex14_packet_valid, sr_metex14_parse,
108                 NULL,
109                 &metex_m3640d_driver_info, receive_data_METEX_M3640D,
110         },
111         {
112                 "PeakTech", "4370", "1200/7n2/rts=0/dtr=1", 1200,
113                 METEX14_PACKET_SIZE, sr_metex14_packet_request,
114                 sr_metex14_packet_valid, sr_metex14_parse,
115                 NULL,
116                 &peaktech_4370_driver_info, receive_data_PEAKTECH_4370,
117         },
118         {
119                 "PCE", "PCE-DM32", "2400/8n1", 2400,
120                 FS9721_PACKET_SIZE, NULL,
121                 sr_fs9721_packet_valid, sr_fs9721_parse,
122                 dmm_details_pce_dm32,
123                 &pce_pce_dm32_driver_info, receive_data_PCE_PCE_DM32,
124         },
125         {
126                 "RadioShack", "22-168", "1200/7n2/rts=0/dtr=1", 1200,
127                 METEX14_PACKET_SIZE, sr_metex14_packet_request,
128                 sr_metex14_packet_valid, sr_metex14_parse,
129                 NULL,
130                 &radioshack_22_168_driver_info, receive_data_RADIOSHACK_22_168,
131         },
132         {
133                 "RadioShack", "22-812", "4800/8n1/rts=0/dtr=1", 4800,
134                 RS9LCD_PACKET_SIZE, NULL,
135                 sr_rs9lcd_packet_valid, sr_rs9lcd_parse,
136                 NULL,
137                 &radioshack_22_812_driver_info, receive_data_RADIOSHACK_22_812,
138         },
139         {
140                 "Voltcraft", "VC-820 (UT-D02 cable)", "2400/8n1/rts=0/dtr=1",
141                 2400, FS9721_PACKET_SIZE, NULL,
142                 sr_fs9721_packet_valid, sr_fs9721_parse,
143                 NULL,
144                 &voltcraft_vc820_ser_driver_info,
145                 receive_data_VOLTCRAFT_VC820_SER,
146         },
147         {
148                 "Voltcraft", "VC-840 (UT-D02 cable)", "2400/8n1/rts=0/dtr=1",
149                 2400, FS9721_PACKET_SIZE, NULL,
150                 sr_fs9721_packet_valid, sr_fs9721_parse,
151                 NULL,
152                 &voltcraft_vc840_ser_driver_info,
153                 receive_data_VOLTCRAFT_VC840_SER,
154         },
155         {
156                 /* Note: ES51922 baudrate is actually 19230! */
157                 "UNI-T", "UT61E (UT-D02 cable)", "19200/7o1/rts=0/dtr=1",
158                 19200, ES51922_PACKET_SIZE, NULL,
159                 sr_es51922_packet_valid, sr_es51922_parse, NULL,
160                 &uni_t_ut61e_ser_driver_info, receive_data_UNI_T_UT61E_SER,
161         },
162 };
163
164 /* Properly close and free all devices. */
165 static int clear_instances(int dmm)
166 {
167         struct sr_dev_inst *sdi;
168         struct drv_context *drvc;
169         struct dev_context *devc;
170         GSList *l;
171         struct sr_dev_driver *di;
172
173         di = dmms[dmm].di;
174
175         if (!(drvc = di->priv))
176                 return SR_OK;
177
178         drvc = di->priv;
179         for (l = drvc->instances; l; l = l->next) {
180                 if (!(sdi = l->data))
181                         continue;
182                 if (!(devc = sdi->priv))
183                         continue;
184                 sr_serial_dev_inst_free(devc->serial);
185                 sr_dev_inst_free(sdi);
186         }
187         g_slist_free(drvc->instances);
188         drvc->instances = NULL;
189
190         return SR_OK;
191 }
192
193 static int hw_init(struct sr_context *sr_ctx, int dmm)
194 {
195         sr_dbg("Selected '%s' subdriver.", dmms[dmm].di->name);
196
197         return std_hw_init(sr_ctx, dmms[dmm].di, DRIVER_LOG_DOMAIN);
198 }
199
200 static GSList *scan(const char *conn, const char *serialcomm, int dmm)
201 {
202         struct sr_dev_inst *sdi;
203         struct drv_context *drvc;
204         struct dev_context *devc;
205         struct sr_probe *probe;
206         struct sr_serial_dev_inst *serial;
207         GSList *devices;
208         int dropped, ret;
209         size_t len;
210         uint8_t buf[128];
211
212         if (!(serial = sr_serial_dev_inst_new(conn, serialcomm)))
213                 return NULL;
214
215         if (serial_open(serial, SERIAL_RDWR | SERIAL_NONBLOCK) != SR_OK)
216                 return NULL;
217
218         sr_info("Probing serial port %s.", conn);
219
220         drvc = dmms[dmm].di->priv;
221         devices = NULL;
222         serial_flush(serial);
223
224         /* Request a packet if the DMM requires this. */
225         if (dmms[dmm].packet_request) {
226                 if ((ret = dmms[dmm].packet_request(serial)) < 0) {
227                         sr_err("Failed to request packet: %d.", ret);
228                         return FALSE;
229                 }
230         }
231
232         /*
233          * There's no way to get an ID from the multimeter. It just sends data
234          * periodically (or upon request), so the best we can do is check if
235          * the packets match the expected format.
236          */
237
238         /* Let's get a bit of data and see if we can find a packet. */
239         len = sizeof(buf);
240         ret = serial_stream_detect(serial, buf, &len, dmms[dmm].packet_size,
241                                    dmms[dmm].packet_valid, 1000,
242                                    dmms[dmm].baudrate);
243         if (ret != SR_OK)
244                 goto scan_cleanup;
245
246         /*
247          * If we dropped more than two packets worth of data, something is
248          * wrong. We shouldn't quit however, since the dropped bytes might be
249          * just zeroes at the beginning of the stream. Those can occur as a
250          * combination of the nonstandard cable that ships with some devices
251          * and the serial port or USB to serial adapter.
252          */
253         dropped = len - dmms[dmm].packet_size;
254         if (dropped > 2 * dmms[dmm].packet_size)
255                 sr_warn("Had to drop too much data.");
256
257         sr_info("Found device on port %s.", conn);
258
259         if (!(sdi = sr_dev_inst_new(0, SR_ST_INACTIVE, dmms[dmm].vendor,
260                                     dmms[dmm].device, "")))
261                 goto scan_cleanup;
262
263         if (!(devc = g_try_malloc0(sizeof(struct dev_context)))) {
264                 sr_err("Device context malloc failed.");
265                 goto scan_cleanup;
266         }
267
268         devc->serial = serial;
269
270         sdi->priv = devc;
271         sdi->driver = dmms[dmm].di;
272         if (!(probe = sr_probe_new(0, SR_PROBE_ANALOG, TRUE, "P1")))
273                 goto scan_cleanup;
274         sdi->probes = g_slist_append(sdi->probes, probe);
275         drvc->instances = g_slist_append(drvc->instances, sdi);
276         devices = g_slist_append(devices, sdi);
277
278 scan_cleanup:
279         serial_close(serial);
280
281         return devices;
282 }
283
284 static GSList *hw_scan(GSList *options, int dmm)
285 {
286         struct sr_config *src;
287         GSList *l, *devices;
288         const char *conn, *serialcomm;
289
290         conn = serialcomm = NULL;
291         for (l = options; l; l = l->next) {
292                 src = l->data;
293                 switch (src->key) {
294                 case SR_CONF_CONN:
295                         conn = src->value;
296                         break;
297                 case SR_CONF_SERIALCOMM:
298                         serialcomm = src->value;
299                         break;
300                 }
301         }
302         if (!conn)
303                 return NULL;
304
305         if (serialcomm) {
306                 /* Use the provided comm specs. */
307                 devices = scan(conn, serialcomm, dmm);
308         } else {
309                 /* Try the default. */
310                 devices = scan(conn, dmms[dmm].conn, dmm);
311         }
312
313         return devices;
314 }
315
316 static GSList *hw_dev_list(int dmm)
317 {
318         struct drv_context *drvc;
319
320         drvc = dmms[dmm].di->priv;
321
322         return drvc->instances;
323 }
324
325 static int hw_dev_open(struct sr_dev_inst *sdi)
326 {
327         struct dev_context *devc;
328
329         if (!(devc = sdi->priv)) {
330                 sr_err("sdi->priv was NULL.");
331                 return SR_ERR_BUG;
332         }
333
334         if (serial_open(devc->serial, SERIAL_RDWR | SERIAL_NONBLOCK) != SR_OK)
335                 return SR_ERR;
336
337         sdi->status = SR_ST_ACTIVE;
338
339         return SR_OK;
340 }
341
342 static int hw_dev_close(struct sr_dev_inst *sdi)
343 {
344         struct dev_context *devc;
345
346         if (!(devc = sdi->priv)) {
347                 sr_err("sdi->priv was NULL.");
348                 return SR_ERR_BUG;
349         }
350
351         if (devc->serial && devc->serial->fd != -1) {
352                 serial_close(devc->serial);
353                 sdi->status = SR_ST_INACTIVE;
354         }
355
356         return SR_OK;
357 }
358
359 static int hw_cleanup(int dmm)
360 {
361         clear_instances(dmm);
362
363         return SR_OK;
364 }
365
366 static int config_set(int id, const void *value, const struct sr_dev_inst *sdi)
367 {
368         struct dev_context *devc;
369
370         if (sdi->status != SR_ST_ACTIVE)
371                 return SR_ERR;
372
373         if (!(devc = sdi->priv)) {
374                 sr_err("sdi->priv was NULL.");
375                 return SR_ERR_BUG;
376         }
377
378         switch (id) {
379         case SR_CONF_LIMIT_SAMPLES:
380                 devc->limit_samples = *(const uint64_t *)value;
381                 sr_dbg("Setting sample limit to %" PRIu64 ".",
382                        devc->limit_samples);
383                 break;
384         case SR_CONF_LIMIT_MSEC:
385                 devc->limit_msec = *(const uint64_t *)value;
386                 sr_dbg("Setting time limit to %" PRIu64 "ms.",
387                        devc->limit_msec);
388                 break;
389         default:
390                 sr_err("Unknown capability: %d.", id);
391                 return SR_ERR;
392                 break;
393         }
394
395         return SR_OK;
396 }
397
398 static int config_list(int key, const void **data, const struct sr_dev_inst *sdi)
399 {
400
401         (void)sdi;
402
403         switch (key) {
404         case SR_CONF_SCAN_OPTIONS:
405                 *data = hwopts;
406                 break;
407         case SR_CONF_DEVICE_OPTIONS:
408                 *data = hwcaps;
409                 break;
410         default:
411                 return SR_ERR_ARG;
412         }
413
414         return SR_OK;
415 }
416
417 static int hw_dev_acquisition_start(const struct sr_dev_inst *sdi,
418                                     void *cb_data, int dmm)
419 {
420         struct sr_datafeed_packet packet;
421         struct sr_datafeed_header header;
422         struct dev_context *devc;
423
424         if (!(devc = sdi->priv)) {
425                 sr_err("sdi->priv was NULL.");
426                 return SR_ERR_BUG;
427         }
428
429         sr_dbg("Starting acquisition.");
430
431         devc->cb_data = cb_data;
432
433         /*
434          * Reset the number of samples to take. If we've already collected our
435          * quota, but we start a new session, and don't reset this, we'll just
436          * quit without acquiring any new samples.
437          */
438         devc->num_samples = 0;
439         devc->starttime = g_get_monotonic_time();
440
441         /* Send header packet to the session bus. */
442         sr_dbg("Sending SR_DF_HEADER.");
443         packet.type = SR_DF_HEADER;
444         packet.payload = (uint8_t *)&header;
445         header.feed_version = 1;
446         gettimeofday(&header.starttime, NULL);
447         sr_session_send(devc->cb_data, &packet);
448
449         /* Poll every 50ms, or whenever some data comes in. */
450         sr_source_add(devc->serial->fd, G_IO_IN, 50,
451                       dmms[dmm].receive_data, (void *)sdi);
452
453         return SR_OK;
454 }
455
456 static int hw_dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data)
457 {
458         struct sr_datafeed_packet packet;
459         struct dev_context *devc;
460
461         if (sdi->status != SR_ST_ACTIVE)
462                 return SR_ERR;
463
464         if (!(devc = sdi->priv)) {
465                 sr_err("sdi->priv was NULL.");
466                 return SR_ERR_BUG;
467         }
468
469         sr_dbg("Stopping acquisition.");
470
471         sr_source_remove(devc->serial->fd);
472         hw_dev_close((struct sr_dev_inst *)sdi);
473
474         /* Send end packet to the session bus. */
475         sr_dbg("Sending SR_DF_END.");
476         packet.type = SR_DF_END;
477         sr_session_send(cb_data, &packet);
478
479         return SR_OK;
480 }
481
482 /* Driver-specific API function wrappers */
483 #define HW_INIT(X) \
484 static int hw_init_##X(struct sr_context *sr_ctx) { return hw_init(sr_ctx, X); }
485 #define HW_CLEANUP(X) \
486 static int hw_cleanup_##X(void) { return hw_cleanup(X); }
487 #define HW_SCAN(X) \
488 static GSList *hw_scan_##X(GSList *options) { return hw_scan(options, X); }
489 #define HW_DEV_LIST(X) \
490 static GSList *hw_dev_list_##X(void) { return hw_dev_list(X); }
491 #define CLEAR_INSTANCES(X) \
492 static int clear_instances_##X(void) { return clear_instances(X); }
493 #define HW_DEV_ACQUISITION_START(X) \
494 static int hw_dev_acquisition_start_##X(const struct sr_dev_inst *sdi, \
495 void *cb_data) { return hw_dev_acquisition_start(sdi, cb_data, X); }
496
497 /* Driver structs and API function wrappers */
498 #define DRV(ID, ID_UPPER, NAME, LONGNAME) \
499 HW_INIT(ID_UPPER) \
500 HW_CLEANUP(ID_UPPER) \
501 HW_SCAN(ID_UPPER) \
502 HW_DEV_LIST(ID_UPPER) \
503 CLEAR_INSTANCES(ID_UPPER) \
504 HW_DEV_ACQUISITION_START(ID_UPPER) \
505 SR_PRIV struct sr_dev_driver ID##_driver_info = { \
506         .name = NAME, \
507         .longname = LONGNAME, \
508         .api_version = 1, \
509         .init = hw_init_##ID_UPPER, \
510         .cleanup = hw_cleanup_##ID_UPPER, \
511         .scan = hw_scan_##ID_UPPER, \
512         .dev_list = hw_dev_list_##ID_UPPER, \
513         .dev_clear = clear_instances_##ID_UPPER, \
514         .config_set = config_set, \
515         .config_list = config_list, \
516         .dev_open = hw_dev_open, \
517         .dev_close = hw_dev_close, \
518         .dev_acquisition_start = hw_dev_acquisition_start_##ID_UPPER, \
519         .dev_acquisition_stop = hw_dev_acquisition_stop, \
520         .priv = NULL, \
521 };
522
523 DRV(digitek_dt4000zc, DIGITEK_DT4000ZC, "digitek-dt4000zc", "Digitek DT4000ZC")
524 DRV(tekpower_tp4000zc, TEKPOWER_TP4000ZC, "tekpower-tp4000zc", "TekPower TP4000ZC")
525 DRV(metex_me31, METEX_ME31, "metex-me31", "Metex ME-31")
526 DRV(peaktech_3410, PEAKTECH_3410, "peaktech-3410", "PeakTech 3410")
527 DRV(mastech_mas345, MASTECH_MAS345, "mastech-mas345", "MASTECH MAS345")
528 DRV(va_va18b, VA_VA18B, "va-va18b", "V&A VA18B")
529 DRV(metex_m3640d, METEX_M3640D, "metex-m3640d", "Metex M-3640D")
530 DRV(peaktech_4370, PEAKTECH_4370, "peaktech-4370", "PeakTech 4370")
531 DRV(pce_pce_dm32, PCE_PCE_DM32, "pce-pce-dm32", "PCE PCE-DM32")
532 DRV(radioshack_22_168, RADIOSHACK_22_168, "radioshack-22-168", "RadioShack 22-168")
533 DRV(radioshack_22_812, RADIOSHACK_22_812, "radioshack-22-812", "RadioShack 22-812")
534 DRV(voltcraft_vc820_ser, VOLTCRAFT_VC820_SER, "voltcraft-vc820-ser", "Voltcraft VC-820 (UT-D02 cable)")
535 DRV(voltcraft_vc840_ser, VOLTCRAFT_VC840_SER, "voltcraft-vc840-ser", "Voltcraft VC-840 (UT-D02 cable)")
536 DRV(uni_t_ut61e_ser, UNI_T_UT61E_SER, "uni-t-ut61e-ser", "UNI-T UT61E (UT-D02 cable)")