]> sigrok.org Git - libsigrok.git/blob - src/hardware/sysclk-lwla/api.c
sysclk-lwla: Fix a segfault in dev_close().
[libsigrok.git] / src / hardware / sysclk-lwla / api.c
1 /*
2  * This file is part of the libsigrok project.
3  *
4  * Copyright (C) 2014 Daniel Elstner <daniel.kitta@gmail.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 <config.h>
21 #include <glib.h>
22 #include <libusb.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <libsigrok/libsigrok.h>
26 #include <libsigrok-internal.h>
27 #include "protocol.h"
28
29 static const uint32_t scanopts[] = {
30         SR_CONF_CONN,
31 };
32
33 static const uint32_t drvopts[] = {
34         SR_CONF_LOGIC_ANALYZER,
35 };
36
37 static const int32_t trigger_matches[] = {
38         SR_TRIGGER_ZERO,
39         SR_TRIGGER_ONE,
40         SR_TRIGGER_RISING,
41         SR_TRIGGER_FALLING,
42 };
43
44 static const char *trigger_sources[] = {
45         [TRIGGER_CHANNELS] = "CH",
46         [TRIGGER_EXT_TRG] = "TRG",
47 };
48
49 static const char *signal_edges[] = {
50         [EDGE_POSITIVE] = "r",
51         [EDGE_NEGATIVE] = "f",
52 };
53
54 static struct sr_dev_inst *dev_inst_new(const struct model_info *model)
55 {
56         struct sr_dev_inst *sdi;
57         struct dev_context *devc;
58         int i;
59         char name[8];
60
61         devc = g_malloc0(sizeof(struct dev_context));
62         devc->model = model;
63         devc->active_fpga_config = FPGA_NOCONF;
64         devc->cfg_rle = TRUE;
65         devc->samplerate = model->samplerates[0];
66         devc->channel_mask = (UINT64_C(1) << model->num_channels) - 1;
67
68         sdi = g_malloc0(sizeof(struct sr_dev_inst));
69         sdi->status = SR_ST_INACTIVE;
70         sdi->vendor = g_strdup("SysClk");
71         sdi->model = g_strdup(model->name);
72         sdi->priv = devc;
73
74         for (i = 0; i < model->num_channels; i++) {
75                 g_snprintf(name, sizeof(name), "CH%d", i + 1);
76                 sr_channel_new(sdi, i, SR_CHANNEL_LOGIC, TRUE, name);
77         }
78
79         return sdi;
80 }
81
82 /* Create a new device instance for a libusb device if it is a SysClk LWLA
83  * device and also matches the connection specification.
84  */
85 static struct sr_dev_inst *dev_inst_new_matching(GSList *conn_matches,
86                                                  libusb_device *dev)
87 {
88         GSList *node;
89         struct sr_usb_dev_inst *usb;
90         const struct model_info *model;
91         struct sr_dev_inst *sdi;
92         struct libusb_device_descriptor des;
93         int bus, address, ret;
94         unsigned int vid, pid;
95
96         bus = libusb_get_bus_number(dev);
97         address = libusb_get_device_address(dev);
98
99         for (node = conn_matches; node != NULL; node = node->next) {
100                 usb = node->data;
101                 if (usb && usb->bus == bus && usb->address == address)
102                         break; /* found */
103         }
104         if (conn_matches && !node)
105                 return NULL; /* no match */
106
107         ret = libusb_get_device_descriptor(dev, &des);
108         if (ret != 0) {
109                 sr_err("Failed to get USB device descriptor: %s.",
110                         libusb_error_name(ret));
111                 return NULL;
112         }
113         vid = des.idVendor;
114         pid = des.idProduct;
115
116         /* Create sigrok device instance. */
117         if (vid == USB_VID_SYSCLK && pid == USB_PID_LWLA1016) {
118                 model = &lwla1016_info;
119         } else if (vid == USB_VID_SYSCLK && pid == USB_PID_LWLA1034) {
120                 model = &lwla1034_info;
121         } else {
122                 if (conn_matches)
123                         sr_warn("USB device %d.%d (%04x:%04x) is not a"
124                                 " SysClk LWLA.", bus, address, vid, pid);
125                 return NULL;
126         }
127         sdi = dev_inst_new(model);
128
129         sdi->inst_type = SR_INST_USB;
130         sdi->conn = sr_usb_dev_inst_new(bus, address, NULL);
131
132         return sdi;
133 }
134
135 static GSList *scan(struct sr_dev_driver *di, GSList *options)
136 {
137         GSList *conn_devices, *devices, *node;
138         struct drv_context *drvc;
139         struct sr_dev_inst *sdi;
140         struct sr_config *src;
141         const char *conn;
142         libusb_device **devlist;
143         ssize_t num_devs, i;
144
145         drvc = di->context;
146         conn = NULL;
147         conn_devices = NULL;
148         devices = NULL;
149
150         for (node = options; node != NULL; node = node->next) {
151                 src = node->data;
152                 if (src->key == SR_CONF_CONN) {
153                         conn = g_variant_get_string(src->data, NULL);
154                         break;
155                 }
156         }
157         if (conn) {
158                 /* Find devices matching the connection specification. */
159                 conn_devices = sr_usb_find(drvc->sr_ctx->libusb_ctx, conn);
160         }
161
162         /* List all libusb devices. */
163         num_devs = libusb_get_device_list(drvc->sr_ctx->libusb_ctx, &devlist);
164         if (num_devs < 0) {
165                 sr_err("Failed to list USB devices: %s.",
166                         libusb_error_name(num_devs));
167                 g_slist_free_full(conn_devices,
168                         (GDestroyNotify)&sr_usb_dev_inst_free);
169                 return NULL;
170         }
171
172         /* Scan the USB device list for matching LWLA devices. */
173         for (i = 0; i < num_devs; i++) {
174                 sdi = dev_inst_new_matching(conn_devices, devlist[i]);
175                 if (!sdi)
176                         continue; /* no match */
177
178                 /* Register device instance with driver. */
179                 devices = g_slist_append(devices, sdi);
180         }
181
182         libusb_free_device_list(devlist, 1);
183         g_slist_free_full(conn_devices, (GDestroyNotify)&sr_usb_dev_inst_free);
184
185         return std_scan_complete(di, devices);
186 }
187
188 /* Drain any pending data from the USB transfer buffers on the device.
189  * This may be necessary e.g. after a crash or generally to clean up after
190  * an abnormal condition.
191  */
192 static int drain_usb(struct sr_usb_dev_inst *usb, unsigned int endpoint)
193 {
194         int drained, xfer_len, ret;
195         unsigned char buf[512];
196         const unsigned int drain_timeout_ms = 10;
197
198         drained = 0;
199         do {
200                 xfer_len = 0;
201                 ret = libusb_bulk_transfer(usb->devhdl, endpoint,
202                                            buf, sizeof(buf), &xfer_len,
203                                            drain_timeout_ms);
204                 drained += xfer_len;
205         } while (ret == LIBUSB_SUCCESS && xfer_len != 0);
206
207         if (ret != LIBUSB_SUCCESS && ret != LIBUSB_ERROR_TIMEOUT) {
208                 sr_err("Failed to drain USB endpoint %u: %s.",
209                        endpoint & (LIBUSB_ENDPOINT_IN - 1),
210                        libusb_error_name(ret));
211                 return SR_ERR;
212         }
213         if (drained > 0) {
214                 sr_warn("Drained %d bytes from USB endpoint %u.",
215                         drained, endpoint & (LIBUSB_ENDPOINT_IN - 1));
216         }
217
218         return SR_OK;
219 }
220
221 static int dev_open(struct sr_dev_inst *sdi)
222 {
223         struct drv_context *drvc;
224         struct dev_context *devc;
225         struct sr_usb_dev_inst *usb;
226         int i, ret;
227
228         drvc = sdi->driver->context;
229         devc = sdi->priv;
230         usb = sdi->conn;
231
232         /* Try the whole shebang three times, fingers crossed. */
233         for (i = 0; i < 3; i++) {
234                 ret = sr_usb_open(drvc->sr_ctx->libusb_ctx, usb);
235                 if (ret != SR_OK)
236                         return ret;
237
238                 ret = libusb_set_configuration(usb->devhdl, USB_CONFIG);
239                 if (ret != LIBUSB_SUCCESS) {
240                         sr_err("Failed to set USB configuration: %s.",
241                                 libusb_error_name(ret));
242                         sr_usb_close(usb);
243                         return SR_ERR;
244                 }
245
246                 ret = libusb_claim_interface(usb->devhdl, USB_INTERFACE);
247                 if (ret != LIBUSB_SUCCESS) {
248                         sr_err("Failed to claim interface: %s.",
249                                 libusb_error_name(ret));
250                         sr_usb_close(usb);
251                         return SR_ERR;
252                 }
253
254                 ret = drain_usb(usb, EP_REPLY);
255                 if (ret != SR_OK) {
256                         sr_usb_close(usb);
257                         return ret;
258                 }
259                 /* This delay appears to be necessary for reliable operation. */
260                 g_usleep(30 * 1000);
261
262                 devc->active_fpga_config = FPGA_NOCONF;
263                 devc->short_transfer_quirk = FALSE;
264                 devc->state = STATE_IDLE;
265
266                 ret = (*devc->model->apply_fpga_config)(sdi);
267
268                 if (ret == SR_OK)
269                         ret = (*devc->model->device_init_check)(sdi);
270                 if (ret == SR_OK)
271                         break;
272
273                 /* Rinse and repeat. */
274                 sr_usb_close(usb);
275         }
276
277         if (ret == SR_OK && devc->short_transfer_quirk)
278                 sr_warn("Short transfer quirk detected! "
279                         "Memory reads will be slow.");
280         return ret;
281 }
282
283 static int dev_close(struct sr_dev_inst *sdi)
284 {
285         struct dev_context *devc;
286         struct sr_usb_dev_inst *usb;
287         int ret;
288
289         devc = sdi->priv;
290         usb = sdi->conn;
291
292         if (devc->acquisition) {
293                 sr_err("Cannot close device during acquisition!");
294                 /* Request stop, leak handle, and prepare for the worst. */
295                 devc->cancel_requested = TRUE;
296                 return SR_ERR_BUG;
297         }
298
299         /* Download of the shutdown bitstream, if any. */
300         ret = (*devc->model->apply_fpga_config)(sdi);
301         if (ret != SR_OK)
302                 sr_warn("Unable to shut down device.");
303
304         if (usb->devhdl)
305                 libusb_release_interface(usb->devhdl, USB_INTERFACE);
306
307         sr_usb_close(usb);
308
309         return SR_OK;
310 }
311
312 /* Check whether the device options contain a specific key.
313  * Also match against get/set/list bits if specified.
314  */
315 static int has_devopt(const struct model_info *model, uint32_t key)
316 {
317         unsigned int i;
318
319         for (i = 0; i < model->num_devopts; i++) {
320                 if ((model->devopts[i] & (SR_CONF_MASK | key)) == key)
321                         return TRUE;
322         }
323
324         return FALSE;
325 }
326
327 static int config_get(uint32_t key, GVariant **data,
328         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
329 {
330         struct dev_context *devc;
331         unsigned int idx;
332
333         (void)cg;
334
335         if (!sdi)
336                 return SR_ERR_ARG;
337
338         devc = sdi->priv;
339
340         if (!has_devopt(devc->model, key | SR_CONF_GET))
341                 return SR_ERR_NA;
342
343         switch (key) {
344         case SR_CONF_SAMPLERATE:
345                 *data = g_variant_new_uint64(devc->samplerate);
346                 break;
347         case SR_CONF_LIMIT_MSEC:
348                 *data = g_variant_new_uint64(devc->limit_msec);
349                 break;
350         case SR_CONF_LIMIT_SAMPLES:
351                 *data = g_variant_new_uint64(devc->limit_samples);
352                 break;
353         case SR_CONF_RLE:
354                 *data = g_variant_new_boolean(devc->cfg_rle);
355                 break;
356         case SR_CONF_EXTERNAL_CLOCK:
357                 *data = g_variant_new_boolean(devc->cfg_clock_source
358                                                 == CLOCK_EXT_CLK);
359                 break;
360         case SR_CONF_CLOCK_EDGE:
361                 idx = devc->cfg_clock_edge;
362                 if (idx >= ARRAY_SIZE(signal_edges))
363                         return SR_ERR_BUG;
364                 *data = g_variant_new_string(signal_edges[idx]);
365                 break;
366         case SR_CONF_TRIGGER_SOURCE:
367                 idx = devc->cfg_trigger_source;
368                 if (idx >= ARRAY_SIZE(trigger_sources))
369                         return SR_ERR_BUG;
370                 *data = g_variant_new_string(trigger_sources[idx]);
371                 break;
372         case SR_CONF_TRIGGER_SLOPE:
373                 idx = devc->cfg_trigger_slope;
374                 if (idx >= ARRAY_SIZE(signal_edges))
375                         return SR_ERR_BUG;
376                 *data = g_variant_new_string(signal_edges[idx]);
377                 break;
378         default:
379                 /* Must not happen for a key listed in devopts. */
380                 return SR_ERR_BUG;
381         }
382
383         return SR_OK;
384 }
385
386 static int config_set(uint32_t key, GVariant *data,
387         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
388 {
389         uint64_t value;
390         struct dev_context *devc;
391         int idx;
392
393         (void)cg;
394
395         if (!sdi)
396                 return SR_ERR_ARG;
397
398         devc = sdi->priv;
399
400         if (!has_devopt(devc->model, key | SR_CONF_SET))
401                 return SR_ERR_NA;
402
403         switch (key) {
404         case SR_CONF_SAMPLERATE:
405                 value = g_variant_get_uint64(data);
406                 if (value < devc->model->samplerates[devc->model->num_samplerates - 1]
407                                 || value > devc->model->samplerates[0])
408                         return SR_ERR_SAMPLERATE;
409                 devc->samplerate = value;
410                 break;
411         case SR_CONF_LIMIT_MSEC:
412                 value = g_variant_get_uint64(data);
413                 if (value > MAX_LIMIT_MSEC)
414                         return SR_ERR_ARG;
415                 devc->limit_msec = value;
416                 break;
417         case SR_CONF_LIMIT_SAMPLES:
418                 value = g_variant_get_uint64(data);
419                 if (value > MAX_LIMIT_SAMPLES)
420                         return SR_ERR_ARG;
421                 devc->limit_samples = value;
422                 break;
423         case SR_CONF_RLE:
424                 devc->cfg_rle = g_variant_get_boolean(data);
425                 break;
426         case SR_CONF_EXTERNAL_CLOCK:
427                 devc->cfg_clock_source = (g_variant_get_boolean(data))
428                         ? CLOCK_EXT_CLK : CLOCK_INTERNAL;
429                 break;
430         case SR_CONF_CLOCK_EDGE:
431                 if ((idx = std_str_idx(data, ARRAY_AND_SIZE(signal_edges))) < 0)
432                         return SR_ERR_ARG;
433                 devc->cfg_clock_edge = idx;
434                 break;
435         case SR_CONF_TRIGGER_SOURCE:
436                 if ((idx = std_str_idx(data, ARRAY_AND_SIZE(trigger_sources))) < 0)
437                         return SR_ERR_ARG;
438                 devc->cfg_trigger_source = idx;
439                 break;
440         case SR_CONF_TRIGGER_SLOPE:
441                 if ((idx = std_str_idx(data, ARRAY_AND_SIZE(signal_edges))) < 0)
442                         return SR_ERR_ARG;
443                 devc->cfg_trigger_slope = idx;
444                 break;
445         default:
446                 /* Must not happen for a key listed in devopts. */
447                 return SR_ERR_BUG;
448         }
449
450         return SR_OK;
451 }
452
453 static int config_channel_set(const struct sr_dev_inst *sdi,
454         struct sr_channel *ch, unsigned int changes)
455 {
456         uint64_t channel_bit;
457         struct dev_context *devc;
458
459         if (!sdi)
460                 return SR_ERR_ARG;
461
462         devc = sdi->priv;
463
464         if (ch->index < 0 || ch->index >= devc->model->num_channels) {
465                 sr_err("Channel index %d out of range.", ch->index);
466                 return SR_ERR_BUG;
467         }
468
469         if ((changes & SR_CHANNEL_SET_ENABLED) != 0) {
470                 channel_bit = UINT64_C(1) << ch->index;
471
472                 /* Enable or disable logic input for this channel. */
473                 if (ch->enabled)
474                         devc->channel_mask |= channel_bit;
475                 else
476                         devc->channel_mask &= ~channel_bit;
477         }
478
479         return SR_OK;
480 }
481
482 /* Derive trigger masks from the session's trigger configuration. */
483 static int prepare_trigger_masks(const struct sr_dev_inst *sdi)
484 {
485         uint64_t trigger_mask, trigger_values, trigger_edge_mask;
486         uint64_t level_bit, type_bit;
487         struct dev_context *devc;
488         struct sr_trigger *trigger;
489         struct sr_trigger_stage *stage;
490         struct sr_trigger_match *match;
491         const GSList *node;
492         int idx;
493         enum sr_trigger_matches trg;
494
495         devc = sdi->priv;
496
497         trigger = sr_session_trigger_get(sdi->session);
498         if (!trigger || !trigger->stages)
499                 return SR_OK;
500
501         if (trigger->stages->next) {
502                 sr_err("This device only supports 1 trigger stage.");
503                 return SR_ERR_ARG;
504         }
505         stage = trigger->stages->data;
506
507         trigger_mask = 0;
508         trigger_values = 0;
509         trigger_edge_mask = 0;
510
511         for (node = stage->matches; node; node = node->next) {
512                 match = node->data;
513
514                 if (!match->channel->enabled)
515                         continue; /* Ignore disabled channel. */
516
517                 idx = match->channel->index;
518                 trg = match->match;
519
520                 if (idx < 0 || idx >= devc->model->num_channels) {
521                         sr_err("Channel index %d out of range.", idx);
522                         return SR_ERR_BUG; /* Should not happen. */
523                 }
524                 if (trg != SR_TRIGGER_ZERO
525                                 && trg != SR_TRIGGER_ONE
526                                 && trg != SR_TRIGGER_RISING
527                                 && trg != SR_TRIGGER_FALLING) {
528                         sr_err("Unsupported trigger match for CH%d.", idx + 1);
529                         return SR_ERR_ARG;
530                 }
531                 level_bit = (trg == SR_TRIGGER_ONE
532                         || trg == SR_TRIGGER_RISING) ? 1 : 0;
533                 type_bit = (trg == SR_TRIGGER_RISING
534                         || trg == SR_TRIGGER_FALLING) ? 1 : 0;
535
536                 trigger_mask |= UINT64_C(1) << idx;
537                 trigger_values |= level_bit << idx;
538                 trigger_edge_mask |= type_bit << idx;
539         }
540         devc->trigger_mask = trigger_mask;
541         devc->trigger_values = trigger_values;
542         devc->trigger_edge_mask = trigger_edge_mask;
543
544         return SR_OK;
545 }
546
547 static int config_commit(const struct sr_dev_inst *sdi)
548 {
549         struct dev_context *devc;
550         int ret;
551
552         devc = sdi->priv;
553
554         if (devc->acquisition) {
555                 sr_err("Acquisition still in progress?");
556                 return SR_ERR;
557         }
558
559         ret = prepare_trigger_masks(sdi);
560         if (ret != SR_OK)
561                 return ret;
562
563         ret = (*devc->model->apply_fpga_config)(sdi);
564         if (ret != SR_OK) {
565                 sr_err("Failed to apply FPGA configuration.");
566                 return ret;
567         }
568
569         return SR_OK;
570 }
571
572 static int config_list(uint32_t key, GVariant **data,
573         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
574 {
575         struct dev_context *devc;
576
577         devc = (sdi) ? sdi->priv : NULL;
578
579         switch (key) {
580         case SR_CONF_SCAN_OPTIONS:
581         case SR_CONF_DEVICE_OPTIONS:
582                 return std_opts_config_list(key, data, sdi, cg,
583                         ARRAY_AND_SIZE(scanopts), ARRAY_AND_SIZE(drvopts),
584                         (devc) ? devc->model->devopts : NULL,
585                         (devc) ? devc->model->num_devopts : 0);
586         }
587
588         if (!has_devopt(devc->model, key | SR_CONF_LIST))
589                 return SR_ERR_NA;
590
591         switch (key) {
592         case SR_CONF_SAMPLERATE:
593                 *data = std_gvar_samplerates(devc->model->samplerates, devc->model->num_samplerates);
594                 break;
595         case SR_CONF_TRIGGER_MATCH:
596                 *data = std_gvar_array_i32(ARRAY_AND_SIZE(trigger_matches));
597                 break;
598         case SR_CONF_TRIGGER_SOURCE:
599                 *data = g_variant_new_strv(ARRAY_AND_SIZE(trigger_sources));
600                 break;
601         case SR_CONF_TRIGGER_SLOPE:
602         case SR_CONF_CLOCK_EDGE:
603                 *data = g_variant_new_strv(ARRAY_AND_SIZE(signal_edges));
604                 break;
605         default:
606                 /* Must not happen for a key listed in devopts. */
607                 return SR_ERR_BUG;
608         }
609
610         return SR_OK;
611 }
612
613 /* Set up the device hardware to begin capturing samples as soon as the
614  * configured trigger conditions are met, or immediately if no triggers
615  * are configured.
616  */
617 static int dev_acquisition_start(const struct sr_dev_inst *sdi)
618 {
619         return lwla_start_acquisition(sdi);
620 }
621
622 static int dev_acquisition_stop(struct sr_dev_inst *sdi)
623 {
624         struct dev_context *devc;
625
626         devc = sdi->priv;
627
628         if (devc->state != STATE_IDLE && !devc->cancel_requested) {
629                 devc->cancel_requested = TRUE;
630                 sr_dbg("Requesting cancel.");
631         }
632
633         return SR_OK;
634 }
635
636 static struct sr_dev_driver sysclk_lwla_driver_info = {
637         .name = "sysclk-lwla",
638         .longname = "SysClk LWLA series",
639         .api_version = 1,
640         .init = std_init,
641         .cleanup = std_cleanup,
642         .scan = scan,
643         .dev_list = std_dev_list,
644         .dev_clear = std_dev_clear,
645         .config_get = config_get,
646         .config_set = config_set,
647         .config_channel_set = config_channel_set,
648         .config_commit = config_commit,
649         .config_list = config_list,
650         .dev_open = dev_open,
651         .dev_close = dev_close,
652         .dev_acquisition_start = dev_acquisition_start,
653         .dev_acquisition_stop = dev_acquisition_stop,
654         .context = NULL,
655 };
656 SR_REGISTER_DEV_DRIVER(sysclk_lwla_driver_info);