]> sigrok.org Git - libsigrok.git/blob - hardware/lascar-el-usb/api.c
probe_groups: API changes required to implement probe groups.
[libsigrok.git] / hardware / lascar-el-usb / api.c
1 /*
2  * This file is part of the libsigrok project.
3  *
4  * Copyright (C) 2012 Bert Vermeulen <bert@biot.com>
5  *
6  * This program is free software: you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation, either version 3 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
18  */
19
20 #include <glib.h>
21 #include <libusb.h>
22 #include "libsigrok.h"
23 #include "libsigrok-internal.h"
24 #include "protocol.h"
25
26 SR_PRIV struct sr_dev_inst *lascar_scan(int bus, int address);
27 SR_PRIV struct sr_dev_driver lascar_el_usb_driver_info;
28 static struct sr_dev_driver *di = &lascar_el_usb_driver_info;
29
30 static const int32_t hwopts[] = {
31         SR_CONF_CONN,
32 };
33
34 static const int32_t hwcaps[] = {
35         SR_CONF_THERMOMETER,
36         SR_CONF_HYGROMETER,
37         SR_CONF_DATALOG,
38         SR_CONF_LIMIT_SAMPLES,
39 };
40
41 static int dev_clear(void)
42 {
43         return std_dev_clear(di, NULL);
44 }
45
46 static int init(struct sr_context *sr_ctx)
47 {
48         return std_init(sr_ctx, di, LOG_PREFIX);
49 }
50
51 static GSList *scan(GSList *options)
52 {
53         struct drv_context *drvc;
54         struct sr_dev_inst *sdi;
55         struct sr_usb_dev_inst *usb;
56         struct sr_config *src;
57         GSList *usb_devices, *devices, *l;
58         const char *conn;
59
60         drvc = di->priv;
61
62         conn = NULL;
63         for (l = options; l; l = l->next) {
64                 src = l->data;
65                 switch (src->key) {
66                 case SR_CONF_CONN:
67                         conn = g_variant_get_string(src->data, NULL);
68                         break;
69                 }
70         }
71         if (!conn)
72                 return NULL;
73
74         devices = NULL;
75         if ((usb_devices = sr_usb_find(drvc->sr_ctx->libusb_ctx, conn))) {
76                 /* We have a list of sr_usb_dev_inst matching the connection
77                  * string. Wrap them in sr_dev_inst and we're done. */
78                 for (l = usb_devices; l; l = l->next) {
79                         usb = l->data;
80                         if (!(sdi = lascar_scan(usb->bus, usb->address))) {
81                                 /* Not a Lascar EL-USB. */
82                                 g_free(usb);
83                                 continue;
84                         }
85                         sdi->inst_type = SR_INST_USB;
86                         sdi->conn = usb;
87                         drvc->instances = g_slist_append(drvc->instances, sdi);
88                         devices = g_slist_append(devices, sdi);
89                 }
90                 g_slist_free(usb_devices);
91         } else
92                 g_slist_free_full(usb_devices, g_free);
93
94         return devices;
95 }
96
97 static GSList *dev_list(void)
98 {
99         return ((struct drv_context *)(di->priv))->instances;
100 }
101
102 static int dev_open(struct sr_dev_inst *sdi)
103 {
104         struct drv_context *drvc;
105         struct sr_usb_dev_inst *usb;
106         int ret;
107
108         if (!(drvc = di->priv)) {
109                 sr_err("Driver was not initialized.");
110                 return SR_ERR;
111         }
112
113         usb = sdi->conn;
114
115         if (sr_usb_open(drvc->sr_ctx->libusb_ctx, usb) != SR_OK)
116                 return SR_ERR;
117
118         if ((ret = libusb_claim_interface(usb->devhdl, LASCAR_INTERFACE))) {
119                 sr_err("Failed to claim interface: %s.", libusb_error_name(ret));
120                 return SR_ERR;
121         }
122         sdi->status = SR_ST_ACTIVE;
123
124         return ret;
125 }
126
127 static int dev_close(struct sr_dev_inst *sdi)
128 {
129         struct sr_usb_dev_inst *usb;
130
131         if (!di->priv) {
132                 sr_err("Driver was not initialized.");
133                 return SR_ERR;
134         }
135
136         usb = sdi->conn;
137
138         if (!usb->devhdl)
139                 /*  Nothing to do. */
140                 return SR_OK;
141
142         libusb_release_interface(usb->devhdl, LASCAR_INTERFACE);
143         libusb_close(usb->devhdl);
144         usb->devhdl = NULL;
145         sdi->status = SR_ST_INACTIVE;
146
147         return SR_OK;
148 }
149
150 static int cleanup(void)
151 {
152         int ret;
153         struct drv_context *drvc;
154
155         if (!(drvc = di->priv))
156                 /* Can get called on an unused driver, doesn't matter. */
157                 return SR_OK;
158
159         ret = dev_clear();
160         g_free(drvc);
161         di->priv = NULL;
162
163         return ret;
164 }
165
166 static int config_get(int id, GVariant **data, const struct sr_dev_inst *sdi,
167                 const struct sr_probe_group *probe_group)
168 {
169         struct dev_context *devc;
170         struct sr_usb_dev_inst *usb;
171         int ret;
172         char str[128];
173
174         (void)probe_group;
175
176         devc = sdi->priv;
177         switch (id) {
178         case SR_CONF_CONN:
179                 if (!sdi || !sdi->conn)
180                         return SR_ERR_ARG;
181                 usb = sdi->conn;
182                 snprintf(str, 128, "%d.%d", usb->bus, usb->address);
183                 *data = g_variant_new_string(str);
184                 break;
185         case SR_CONF_DATALOG:
186                 if (!sdi)
187                         return SR_ERR_ARG;
188                 if ((ret = lascar_is_logging(sdi)) == -1)
189                         return SR_ERR;
190                 *data = g_variant_new_boolean(ret ? TRUE : FALSE);
191                 break;
192         case SR_CONF_LIMIT_SAMPLES:
193                 *data = g_variant_new_uint64(devc->limit_samples);
194                 break;
195         default:
196                 return SR_ERR_NA;
197         }
198
199         return SR_OK;
200 }
201
202 static int config_set(int id, GVariant *data, const struct sr_dev_inst *sdi,
203                 const struct sr_probe_group *probe_group)
204 {
205         struct dev_context *devc;
206         int ret;
207
208         (void)probe_group;
209
210         if (sdi->status != SR_ST_ACTIVE)
211                 return SR_ERR_DEV_CLOSED;
212
213         if (!di->priv) {
214                 sr_err("Driver was not initialized.");
215                 return SR_ERR;
216         }
217
218         devc = sdi->priv;
219         ret = SR_OK;
220         switch (id) {
221         case SR_CONF_DATALOG:
222                 if (g_variant_get_boolean(data)) {
223                         /* Start logging. */
224                         ret = lascar_start_logging(sdi);
225                 } else {
226                         /* Stop logging. */
227                         ret = lascar_stop_logging(sdi);
228                 }
229                 break;
230         case SR_CONF_LIMIT_SAMPLES:
231                 devc->limit_samples = g_variant_get_uint64(data);
232                 sr_dbg("Setting sample limit to %" PRIu64 ".",
233                        devc->limit_samples);
234                 break;
235         default:
236                 ret = SR_ERR_NA;
237         }
238
239         return ret;
240 }
241
242 static int config_list(int key, GVariant **data, const struct sr_dev_inst *sdi,
243                 const struct sr_probe_group *probe_group)
244 {
245         (void)sdi;
246         (void)probe_group;
247
248         switch (key) {
249         case SR_CONF_SCAN_OPTIONS:
250                 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32,
251                                 hwopts, ARRAY_SIZE(hwopts), sizeof(int32_t));
252                 break;
253         case SR_CONF_DEVICE_OPTIONS:
254                 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32,
255                                 hwcaps, ARRAY_SIZE(hwcaps), sizeof(int32_t));
256                 break;
257         default:
258                 return SR_ERR_NA;
259         }
260
261         return SR_OK;
262 }
263
264 static void mark_xfer(struct libusb_transfer *xfer)
265 {
266
267         if (xfer->status == LIBUSB_TRANSFER_COMPLETED)
268                 xfer->user_data = GINT_TO_POINTER(1);
269         else
270                 xfer->user_data = GINT_TO_POINTER(-1);
271
272 }
273
274 /* The Lascar software, in its infinite ignorance, reads a set of four
275  * bytes from the device config struct and interprets it as a float.
276  * That only works because they only use windows, and only on x86. However
277  * we may be running on any architecture, any operating system. So we have
278  * to convert these four bytes as the Lascar software would on windows/x86,
279  * to the local representation of a float.
280  * The source format is little-endian, with IEEE 754-2008 BINARY32 encoding. */
281 static float binary32_le_to_float(unsigned char *buf)
282 {
283         GFloatIEEE754 f;
284
285         f.v_float = 0;
286         f.mpn.sign = (buf[3] & 0x80) ? 1 : 0;
287         f.mpn.biased_exponent = (buf[3] << 1) | (buf[2] >> 7);
288         f.mpn.mantissa = buf[0] | (buf[1] << 8) | ((buf[2] & 0x7f) << 16);
289
290         return f.v_float;
291 }
292
293 static int lascar_proc_config(const struct sr_dev_inst *sdi)
294 {
295         struct dev_context *devc;
296         struct sr_usb_dev_inst *usb;
297         int dummy, ret;
298
299         devc = sdi->priv;
300         usb = sdi->conn;
301
302         if (lascar_get_config(usb->devhdl, devc->config, &dummy) != SR_OK)
303                 return SR_ERR;
304
305         ret = SR_OK;
306         switch (devc->profile->logformat) {
307         case LOG_TEMP_RH:
308                 devc->sample_size = 2;
309                 devc->temp_unit = devc->config[0x2e] | (devc->config[0x2f] << 8);
310                 if (devc->temp_unit != 0 && devc->temp_unit != 1) {
311                         sr_dbg("invalid temperature unit %d", devc->temp_unit);
312                         /* Default to Celcius, we're all adults here. */
313                         devc->temp_unit = 0;
314                 } else
315                         sr_dbg("temperature unit is %s", devc->temp_unit
316                                         ? "Fahrenheit" : "Celcius");
317                 break;
318         case LOG_CO:
319                 devc->sample_size = 2;
320                 devc->co_high = binary32_le_to_float(devc->config + 0x24);
321                 devc->co_low = binary32_le_to_float(devc->config + 0x28);
322                 sr_dbg("EL-USB-CO calibration high %f low %f", devc->co_high,
323                                 devc->co_low);
324                 break;
325         default:
326                 ret = SR_ERR_ARG;
327         }
328         devc->logged_samples = devc->config[0x1e] | (devc->config[0x1f] << 8);
329         sr_dbg("device log contains %d samples.", devc->logged_samples);
330
331         return ret;
332 }
333
334 static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
335 {
336         struct sr_datafeed_packet packet;
337         struct sr_datafeed_meta meta;
338         struct sr_config *src;
339         struct dev_context *devc;
340         struct drv_context *drvc;
341         struct sr_usb_dev_inst *usb;
342         struct libusb_transfer *xfer_in, *xfer_out;
343         const struct libusb_pollfd **pfd;
344         struct timeval tv;
345         uint64_t interval;
346         int ret, i;
347         unsigned char cmd[3], resp[4], *buf;
348
349         if (sdi->status != SR_ST_ACTIVE)
350                 return SR_ERR_DEV_CLOSED;
351
352         if (!di->priv) {
353                 sr_err("Driver was not initialized.");
354                 return SR_ERR;
355         }
356
357         drvc = di->priv;
358         devc = sdi->priv;
359         usb = sdi->conn;
360         devc->cb_data = cb_data;
361
362         if (lascar_proc_config(sdi) != SR_OK)
363                 return SR_ERR;
364
365         sr_dbg("Starting log retrieval.");
366
367         /* Send header packet to the session bus. */
368         std_session_send_df_header(cb_data, LOG_PREFIX);
369
370         interval = (devc->config[0x1c] | (devc->config[0x1d] << 8)) * 1000;
371         packet.type = SR_DF_META;
372         packet.payload = &meta;
373         src = sr_config_new(SR_CONF_SAMPLE_INTERVAL, g_variant_new_uint64(interval));
374         meta.config = g_slist_append(NULL, src);
375         sr_session_send(devc->cb_data, &packet);
376         g_free(src);
377
378         if (devc->logged_samples == 0) {
379                 /* This ensures the frontend knows the session is done. */
380                 packet.type = SR_DF_END;
381                 sr_session_send(devc->cb_data, &packet);
382                 return SR_OK;
383         }
384
385         if (!(xfer_in = libusb_alloc_transfer(0)) ||
386                         !(xfer_out = libusb_alloc_transfer(0)))
387                 return SR_ERR;
388
389         libusb_control_transfer(usb->devhdl, LIBUSB_REQUEST_TYPE_VENDOR,
390                         0x00, 0xffff, 0x00, NULL, 0, 50);
391         libusb_control_transfer(usb->devhdl, LIBUSB_REQUEST_TYPE_VENDOR,
392                         0x02, 0x0002, 0x00, NULL, 0, 50);
393         libusb_control_transfer(usb->devhdl, LIBUSB_REQUEST_TYPE_VENDOR,
394                         0x02, 0x0001, 0x00, NULL, 0, 50);
395
396
397         /* Flush input. The F321 requires this. */
398         while (libusb_bulk_transfer(usb->devhdl, LASCAR_EP_IN, resp,
399                         256, &ret, 5) == 0 && ret > 0)
400                 ;
401
402         libusb_fill_bulk_transfer(xfer_in, usb->devhdl, LASCAR_EP_IN,
403                         resp, sizeof(resp), mark_xfer, 0, 10000);
404         if (libusb_submit_transfer(xfer_in) != 0) {
405                 libusb_free_transfer(xfer_in);
406                 libusb_free_transfer(xfer_out);
407                 return SR_ERR;
408         }
409
410         cmd[0] = 0x03;
411         cmd[1] = 0xff;
412         cmd[2] = 0xff;
413         libusb_fill_bulk_transfer(xfer_out, usb->devhdl, LASCAR_EP_OUT,
414                         cmd, 3, mark_xfer, 0, 100);
415         if (libusb_submit_transfer(xfer_out) != 0) {
416                 libusb_free_transfer(xfer_in);
417                 libusb_free_transfer(xfer_out);
418                 return SR_ERR;
419         }
420
421         tv.tv_sec = 0;
422         tv.tv_usec = 0;
423         while (!xfer_in->user_data || !xfer_out->user_data) {
424                 g_usleep(5000);
425                 libusb_handle_events_timeout(drvc->sr_ctx->libusb_ctx, &tv);
426         }
427         if (xfer_in->user_data != GINT_TO_POINTER(1) ||
428                         xfer_in->user_data != GINT_TO_POINTER(1)) {
429                 sr_dbg("no response to log transfer request");
430                 libusb_free_transfer(xfer_in);
431                 libusb_free_transfer(xfer_out);
432                 return SR_ERR;
433         }
434         if (xfer_in->actual_length != 3 || xfer_in->buffer[0] != 2) {
435                 sr_dbg("invalid response to log transfer request");
436                 libusb_free_transfer(xfer_in);
437                 libusb_free_transfer(xfer_out);
438                 return SR_ERR;
439         }
440         devc->log_size = xfer_in->buffer[1] + (xfer_in->buffer[2] << 8);
441         libusb_free_transfer(xfer_out);
442
443         pfd = libusb_get_pollfds(drvc->sr_ctx->libusb_ctx);
444         for (i = 0; pfd[i]; i++) {
445                 /* Handle USB events every 100ms, for decent latency. */
446                 sr_source_add(pfd[i]->fd, pfd[i]->events, 100,
447                                 lascar_el_usb_handle_events, (void *)sdi);
448                 /* We'll need to remove this fd later. */
449                 devc->usbfd[i] = pfd[i]->fd;
450         }
451         devc->usbfd[i] = -1;
452
453         buf = g_try_malloc(4096);
454         libusb_fill_bulk_transfer(xfer_in, usb->devhdl, LASCAR_EP_IN,
455                         buf, 4096, lascar_el_usb_receive_transfer, cb_data, 100);
456         if ((ret = libusb_submit_transfer(xfer_in) != 0)) {
457                 sr_err("Unable to submit transfer: %s.", libusb_error_name(ret));
458                 libusb_free_transfer(xfer_in);
459                 g_free(buf);
460                 return SR_ERR;
461         }
462
463         return SR_OK;
464 }
465
466 SR_PRIV int dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data)
467 {
468         (void)cb_data;
469
470         if (!di->priv) {
471                 sr_err("Driver was not initialized.");
472                 return SR_ERR;
473         }
474
475         if (sdi->status != SR_ST_ACTIVE) {
476                 sr_err("Device inactive, can't stop acquisition.");
477                 return SR_ERR;
478         }
479
480         sdi->status = SR_ST_STOPPING;
481         /* TODO: free ongoing transfers? */
482
483         return SR_OK;
484 }
485
486 SR_PRIV struct sr_dev_driver lascar_el_usb_driver_info = {
487         .name = "lascar-el-usb",
488         .longname = "Lascar EL-USB",
489         .api_version = 1,
490         .init = init,
491         .cleanup = cleanup,
492         .scan = scan,
493         .dev_list = dev_list,
494         .dev_clear = dev_clear,
495         .config_get = config_get,
496         .config_set = config_set,
497         .config_list = config_list,
498         .dev_open = dev_open,
499         .dev_close = dev_close,
500         .dev_acquisition_start = dev_acquisition_start,
501         .dev_acquisition_stop = dev_acquisition_stop,
502         .priv = NULL,
503 };