]> sigrok.org Git - libsigrok.git/blame_incremental - src/hardware/ftdi-la/api.c
ftdi-la: Cleanup ftdi_context handling
[libsigrok.git] / src / hardware / ftdi-la / api.c
... / ...
CommitLineData
1/*
2 * This file is part of the libsigrok project.
3 *
4 * Copyright (C) 2015 Sergey Alirzaev <zl29ah@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 <ftdi.h>
22#include <libusb.h>
23#include <libsigrok/libsigrok.h>
24#include "libsigrok-internal.h"
25#include "protocol.h"
26
27static const uint32_t scanopts[] = {
28 SR_CONF_CONN,
29};
30
31static const uint32_t devopts[] = {
32 SR_CONF_LOGIC_ANALYZER,
33 SR_CONF_CONTINUOUS,
34 SR_CONF_LIMIT_SAMPLES | SR_CONF_SET,
35 SR_CONF_SAMPLERATE | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
36 SR_CONF_CONN | SR_CONF_GET,
37};
38
39static const uint64_t samplerates[] = {
40 SR_HZ(3600),
41 SR_MHZ(10),
42 SR_HZ(1),
43};
44
45static const struct ftdi_chip_desc ft2232h_desc = {
46 .vendor = 0x0403,
47 .product = 0x6010,
48 .samplerate_div = 20,
49 .channel_names = {
50 "ADBUS0",
51 "ADBUS1",
52 "ADBUS2",
53 "ADBUS3",
54 "ADBUS4",
55 "ADBUS5",
56 "ADBUS6",
57 "ADBUS7",
58 /* TODO: BDBUS[0..7] channels. */
59 NULL
60 }
61};
62
63static const struct ftdi_chip_desc ft232r_desc = {
64 .vendor = 0x0403,
65 .product = 0x6001,
66 .samplerate_div = 30,
67 .channel_names = {
68 "TXD",
69 "RXD",
70 "RTS#",
71 "CTS#",
72 "DTR#",
73 "DSR#",
74 "DCD#",
75 "RI#",
76 NULL
77 }
78};
79
80static const struct ftdi_chip_desc *chip_descs[] = {
81 &ft2232h_desc,
82 &ft232r_desc,
83};
84
85static void scan_device(struct sr_dev_driver *di, struct ftdi_context *ftdic,
86 struct libusb_device *dev, GSList **devices)
87{
88 struct libusb_device_descriptor usb_desc;
89 const struct ftdi_chip_desc *desc;
90 struct dev_context *devc;
91 char *vendor, *model, *serial_num;
92 struct sr_dev_inst *sdi;
93 struct drv_context *drvc;
94 int rv;
95
96 drvc = di->context;
97 libusb_get_device_descriptor(dev, &usb_desc);
98
99 desc = NULL;
100 for (unsigned long i = 0; i < ARRAY_SIZE(chip_descs); i++) {
101 desc = chip_descs[i];
102 if (desc->vendor == usb_desc.idVendor &&
103 desc->product == usb_desc.idProduct)
104 break;
105 }
106
107 if (!desc) {
108 sr_spew("Unsupported FTDI device 0x%4x:0x%4x.",
109 usb_desc.idVendor, usb_desc.idProduct);
110 return;
111 }
112
113 /* Allocate memory for our private device context. */
114 devc = g_malloc0(sizeof(struct dev_context));
115
116 /* Allocate memory for the incoming data. */
117 devc->data_buf = g_malloc0(DATA_BUF_SIZE);
118
119 snprintf(devc->address, sizeof(devc->address), "d:%u/%u",
120 libusb_get_bus_number(dev), libusb_get_device_address(dev));
121
122 devc->desc = desc;
123
124 vendor = g_malloc(32);
125 model = g_malloc(32);
126 serial_num = g_malloc(32);
127 rv = ftdi_usb_get_strings(ftdic, dev, vendor, 32,
128 model, 32, serial_num, 32);
129 switch (rv) {
130 case 0:
131 break;
132 case -9:
133 sr_dbg("The device lacks a serial number.");
134 g_free(serial_num);
135 serial_num = NULL;
136 break;
137 default:
138 sr_err("Failed to get the FTDI strings: %d", rv);
139 goto err_free_strings;
140 }
141 sr_dbg("Found an FTDI device: %s.", model);
142
143 /* Register the device with libsigrok. */
144 sdi = g_malloc0(sizeof(struct sr_dev_inst));
145 sdi->status = SR_ST_INACTIVE;
146 sdi->vendor = vendor;
147 sdi->model = model;
148 sdi->serial_num = serial_num;
149 sdi->driver = di;
150 sdi->priv = devc;
151
152 for (char *const *chan = &(desc->channel_names[0]); *chan; chan++)
153 sr_channel_new(sdi, chan - &(desc->channel_names[0]),
154 SR_CHANNEL_LOGIC, TRUE, *chan);
155
156 *devices = g_slist_append(*devices, sdi);
157 drvc->instances = g_slist_append(drvc->instances, sdi);
158 return;
159
160err_free_strings:
161 g_free(vendor);
162 g_free(model);
163 g_free(serial_num);
164 g_free(devc->data_buf);
165 g_free(devc);
166}
167
168static GSList *scan_all(struct ftdi_context *ftdic, struct sr_dev_driver *di,
169 GSList *options)
170{
171 GSList *devices;
172 struct ftdi_device_list *devlist = 0;
173 struct ftdi_device_list *curdev;
174 int ret;
175
176 (void)options;
177
178 devices = NULL;
179
180 ret = ftdi_usb_find_all(ftdic, &devlist, 0, 0);
181 if (ret < 0) {
182 sr_err("Failed to list devices (%d): %s", ret,
183 ftdi_get_error_string(ftdic));
184 return NULL;
185 }
186
187 sr_dbg("Number of FTDI devices found: %d", ret);
188
189 curdev = devlist;
190 while (curdev) {
191 scan_device(di, ftdic, curdev->dev, &devices);
192 curdev = curdev->next;
193 }
194
195 ftdi_list_free(&devlist);
196
197 return devices;
198}
199
200static GSList *scan(struct sr_dev_driver *di, GSList *options)
201{
202 struct ftdi_context *ftdic;
203 struct sr_config *src;
204 struct sr_usb_dev_inst *usb;
205 const char *conn;
206 GSList *l, *conn_devices;
207 GSList *devices;
208 struct drv_context *drvc;
209 libusb_device **devlist;
210 int i;
211
212 drvc = di->context;
213 drvc->instances = NULL;
214 conn = NULL;
215 for (l = options; l; l = l->next) {
216 src = l->data;
217 if (src->key == SR_CONF_CONN) {
218 conn = g_variant_get_string(src->data, NULL);
219 break;
220 }
221 }
222
223 /* Allocate memory for the FTDI context (ftdic) and initialize it. */
224 ftdic = ftdi_new();
225 if (!ftdic) {
226 sr_err("Failed to initialize libftdi.");
227 return NULL;
228 }
229
230 if (conn) {
231 devices = NULL;
232 libusb_get_device_list(drvc->sr_ctx->libusb_ctx, &devlist);
233 for (i = 0; devlist[i]; i++) {
234 conn_devices = sr_usb_find(drvc->sr_ctx->libusb_ctx, conn);
235 for (l = conn_devices; l; l = l->next) {
236 usb = l->data;
237 if (usb->bus == libusb_get_bus_number(devlist[i])
238 && usb->address == libusb_get_device_address(devlist[i])) {
239 scan_device(di, ftdic, devlist[i], &devices);
240 }
241 }
242 }
243 libusb_free_device_list(devlist, 1);
244 } else
245 devices = scan_all(ftdic, di, options);
246
247 ftdi_free(ftdic);
248
249 return devices;
250}
251
252static void clear_helper(void *priv)
253{
254 struct dev_context *devc;
255
256 devc = priv;
257 g_free(devc->data_buf);
258 g_free(devc);
259}
260
261static int dev_clear(const struct sr_dev_driver *di)
262{
263 return std_dev_clear(di, clear_helper);
264}
265
266static int dev_open(struct sr_dev_inst *sdi)
267{
268 struct dev_context *devc;
269 int ret = SR_OK;
270
271 devc = sdi->priv;
272
273 devc->ftdic = ftdi_new();
274 if (!devc->ftdic)
275 return SR_ERR;
276
277 ret = ftdi_usb_open_string(devc->ftdic, devc->address);
278 if (ret < 0) {
279 /* Log errors, except for -3 ("device not found"). */
280 if (ret != -3)
281 sr_err("Failed to open device (%d): %s", ret,
282 ftdi_get_error_string(devc->ftdic));
283 goto err_ftdi_free;
284 }
285
286 /* Purge RX/TX buffers in the FTDI chip. */
287 ret = ftdi_usb_purge_buffers(devc->ftdic);
288 if (ret < 0) {
289 sr_err("Failed to purge FTDI RX/TX buffers (%d): %s.",
290 ret, ftdi_get_error_string(devc->ftdic));
291 goto err_dev_open_close_ftdic;
292 }
293 sr_dbg("FTDI chip buffers purged successfully.");
294
295 /* Reset the FTDI bitmode. */
296 ret = ftdi_set_bitmode(devc->ftdic, 0x00, BITMODE_RESET);
297 if (ret < 0) {
298 sr_err("Failed to reset the FTDI chip bitmode (%d): %s.",
299 ret, ftdi_get_error_string(devc->ftdic));
300 goto err_dev_open_close_ftdic;
301 }
302 sr_dbg("FTDI chip bitmode reset successfully.");
303
304 ret = ftdi_set_bitmode(devc->ftdic, 0x00, BITMODE_BITBANG);
305 if (ret < 0) {
306 sr_err("Failed to put FTDI chip into bitbang mode (%d): %s.",
307 ret, ftdi_get_error_string(devc->ftdic));
308 goto err_dev_open_close_ftdic;
309 }
310 sr_dbg("FTDI chip bitbang mode entered successfully.");
311
312 sdi->status = SR_ST_ACTIVE;
313
314 return SR_OK;
315err_dev_open_close_ftdic:
316 ftdi_usb_close(devc->ftdic);
317err_ftdi_free:
318 ftdi_free(devc->ftdic);
319 return SR_ERR;
320}
321
322static int dev_close(struct sr_dev_inst *sdi)
323{
324 struct dev_context *devc;
325
326 devc = sdi->priv;
327
328 if (devc->ftdic) {
329 ftdi_usb_close(devc->ftdic);
330 ftdi_free(devc->ftdic);
331 devc->ftdic = NULL;
332 }
333
334 sdi->status = SR_ST_INACTIVE;
335
336 return SR_OK;
337}
338
339static int config_get(uint32_t key, GVariant **data,
340 const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
341{
342 int ret;
343 struct dev_context *devc;
344 struct sr_usb_dev_inst *usb;
345 char str[128];
346
347 (void)cg;
348
349 devc = sdi->priv;
350
351 ret = SR_OK;
352 switch (key) {
353 case SR_CONF_SAMPLERATE:
354 *data = g_variant_new_uint64(devc->cur_samplerate);
355 break;
356 case SR_CONF_CONN:
357 if (!sdi || !sdi->conn)
358 return SR_ERR_ARG;
359 usb = sdi->conn;
360 snprintf(str, 128, "%d.%d", usb->bus, usb->address);
361 *data = g_variant_new_string(str);
362 break;
363 default:
364 return SR_ERR_NA;
365 }
366
367 return ret;
368}
369
370static int config_set(uint32_t key, GVariant *data,
371 const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
372{
373 int ret;
374 struct dev_context *devc;
375 uint64_t value;
376
377 (void)cg;
378
379 if (sdi->status != SR_ST_ACTIVE)
380 return SR_ERR_DEV_CLOSED;
381
382 devc = sdi->priv;
383
384 ret = SR_OK;
385 switch (key) {
386 case SR_CONF_LIMIT_MSEC:
387 value = g_variant_get_uint64(data);
388 /* TODO: Implement. */
389 ret = SR_ERR_NA;
390 break;
391 case SR_CONF_LIMIT_SAMPLES:
392 if (g_variant_get_uint64(data) == 0)
393 return SR_ERR_ARG;
394 devc->limit_samples = g_variant_get_uint64(data);
395 break;
396 case SR_CONF_SAMPLERATE:
397 value = g_variant_get_uint64(data);
398 if (value < 3600)
399 return SR_ERR_SAMPLERATE;
400 devc->cur_samplerate = value;
401 return ftdi_la_set_samplerate(devc);
402 default:
403 ret = SR_ERR_NA;
404 }
405
406 return ret;
407}
408
409static int config_list(uint32_t key, GVariant **data,
410 const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
411{
412 int ret;
413 GVariant *gvar;
414 GVariantBuilder gvb;
415
416 (void)sdi;
417 (void)cg;
418
419 ret = SR_OK;
420 switch (key) {
421 case SR_CONF_SCAN_OPTIONS:
422 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
423 scanopts, ARRAY_SIZE(scanopts), sizeof(uint32_t));
424 break;
425 case SR_CONF_DEVICE_OPTIONS:
426 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_UINT32,
427 devopts, ARRAY_SIZE(devopts), sizeof(uint32_t));
428 break;
429 case SR_CONF_SAMPLERATE:
430 g_variant_builder_init(&gvb, G_VARIANT_TYPE("a{sv}"));
431 gvar = g_variant_new_fixed_array(G_VARIANT_TYPE("t"),
432 samplerates, ARRAY_SIZE(samplerates), sizeof(uint64_t));
433 g_variant_builder_add(&gvb, "{sv}", "samplerate-steps", gvar);
434 *data = g_variant_builder_end(&gvb);
435 break;
436 default:
437 return SR_ERR_NA;
438 }
439
440 return ret;
441}
442
443static int dev_acquisition_start(const struct sr_dev_inst *sdi)
444{
445 struct dev_context *devc;
446
447 devc = sdi->priv;
448
449 if (sdi->status != SR_ST_ACTIVE)
450 return SR_ERR_DEV_CLOSED;
451
452 if (!devc->ftdic)
453 return SR_ERR_BUG;
454
455 ftdi_set_bitmode(devc->ftdic, 0, BITMODE_BITBANG);
456
457 /* Properly reset internal variables before every new acquisition. */
458 devc->samples_sent = 0;
459 devc->bytes_received = 0;
460
461 std_session_send_df_header(sdi, LOG_PREFIX);
462
463 /* Hook up a dummy handler to receive data from the device. */
464 sr_session_source_add(sdi->session, -1, G_IO_IN, 0,
465 ftdi_la_receive_data, (void *)sdi);
466
467 return SR_OK;
468}
469
470static int dev_acquisition_stop(struct sr_dev_inst *sdi)
471{
472 if (sdi->status != SR_ST_ACTIVE)
473 return SR_ERR_DEV_CLOSED;
474
475 sr_dbg("Stopping acquisition.");
476 sr_session_source_remove(sdi->session, -1);
477
478 std_session_send_df_end(sdi, LOG_PREFIX);
479
480 return SR_OK;
481}
482
483static struct sr_dev_driver ftdi_la_driver_info = {
484 .name = "ftdi-la",
485 .longname = "FTDI LA",
486 .api_version = 1,
487 .init = std_init,
488 .cleanup = std_cleanup,
489 .scan = scan,
490 .dev_list = std_dev_list,
491 .dev_clear = dev_clear,
492 .config_get = config_get,
493 .config_set = config_set,
494 .config_list = config_list,
495 .dev_open = dev_open,
496 .dev_close = dev_close,
497 .dev_acquisition_start = dev_acquisition_start,
498 .dev_acquisition_stop = dev_acquisition_stop,
499 .context = NULL,
500};
501SR_REGISTER_DEV_DRIVER(ftdi_la_driver_info);