]> sigrok.org Git - libsigrok.git/blame_incremental - hardware/victor-dmm/api.c
Use consistent API callback function names.
[libsigrok.git] / hardware / victor-dmm / api.c
... / ...
CommitLineData
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 <stdlib.h>
23#include <string.h>
24#include "libsigrok.h"
25#include "libsigrok-internal.h"
26#include "protocol.h"
27
28#define VICTOR_VID 0x1244
29#define VICTOR_PID 0xd237
30#define VICTOR_VENDOR "Victor"
31#define VICTOR_INTERFACE 0
32#define VICTOR_ENDPOINT LIBUSB_ENDPOINT_IN | 1
33
34SR_PRIV struct sr_dev_driver victor_dmm_driver_info;
35static struct sr_dev_driver *di = &victor_dmm_driver_info;
36static int dev_close(struct sr_dev_inst *sdi);
37static int dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data);
38
39static const int32_t hwopts[] = {
40 SR_CONF_CONN,
41};
42
43static const int32_t hwcaps[] = {
44 SR_CONF_MULTIMETER,
45 SR_CONF_LIMIT_MSEC,
46 SR_CONF_LIMIT_SAMPLES,
47 SR_CONF_CONTINUOUS,
48};
49
50/* Properly close and free all devices. */
51static int clear_instances(void)
52{
53 struct sr_dev_inst *sdi;
54 struct drv_context *drvc;
55 struct dev_context *devc;
56 GSList *l;
57
58 if (!(drvc = di->priv))
59 /* Can get called on an unused driver, doesn't matter. */
60 return SR_OK;
61
62 for (l = drvc->instances; l; l = l->next) {
63 if (!(sdi = l->data))
64 continue;
65 if (!(devc = sdi->priv))
66 continue;
67 dev_close(sdi);
68 sr_usb_dev_inst_free(sdi->conn);
69 sr_dev_inst_free(sdi);
70 }
71
72 g_slist_free(drvc->instances);
73 drvc->instances = NULL;
74
75 return SR_OK;
76}
77
78static int init(struct sr_context *sr_ctx)
79{
80 return std_hw_init(sr_ctx, di, LOG_PREFIX);
81}
82
83static GSList *scan(GSList *options)
84{
85 struct drv_context *drvc;
86 struct dev_context *devc;
87 struct sr_dev_inst *sdi;
88 struct sr_probe *probe;
89 struct libusb_device_descriptor des;
90 libusb_device **devlist;
91 GSList *devices;
92 int ret, devcnt, i;
93
94 (void)options;
95
96 drvc = di->priv;
97
98 devices = NULL;
99 libusb_get_device_list(drvc->sr_ctx->libusb_ctx, &devlist);
100 for (i = 0; devlist[i]; i++) {
101 if ((ret = libusb_get_device_descriptor(devlist[i], &des)) != 0) {
102 sr_warn("Failed to get device descriptor: %s",
103 libusb_error_name(ret));
104 continue;
105 }
106
107 if (des.idVendor != VICTOR_VID || des.idProduct != VICTOR_PID)
108 continue;
109
110 devcnt = g_slist_length(drvc->instances);
111 if (!(sdi = sr_dev_inst_new(devcnt, SR_ST_INACTIVE,
112 VICTOR_VENDOR, NULL, NULL)))
113 return NULL;
114 sdi->driver = di;
115
116 if (!(devc = g_try_malloc0(sizeof(struct dev_context))))
117 return NULL;
118 sdi->priv = devc;
119
120 if (!(probe = sr_probe_new(0, SR_PROBE_ANALOG, TRUE, "P1")))
121 return NULL;
122 sdi->probes = g_slist_append(NULL, probe);
123
124 if (!(sdi->conn = sr_usb_dev_inst_new(libusb_get_bus_number(devlist[i]),
125 libusb_get_device_address(devlist[i]), NULL)))
126 return NULL;
127 sdi->inst_type = SR_INST_USB;
128
129 drvc->instances = g_slist_append(drvc->instances, sdi);
130 devices = g_slist_append(devices, sdi);
131 }
132 libusb_free_device_list(devlist, 1);
133
134 return devices;
135}
136
137static GSList *dev_list(void)
138{
139 return ((struct drv_context *)(di->priv))->instances;
140}
141
142static int dev_open(struct sr_dev_inst *sdi)
143{
144 struct drv_context *drvc = di->priv;
145 struct sr_usb_dev_inst *usb;
146 libusb_device **devlist;
147 int ret, i;
148
149 if (!di->priv) {
150 sr_err("Driver was not initialized.");
151 return SR_ERR;
152 }
153
154 usb = sdi->conn;
155
156 libusb_get_device_list(drvc->sr_ctx->libusb_ctx, &devlist);
157 for (i = 0; devlist[i]; i++) {
158 if (libusb_get_bus_number(devlist[i]) != usb->bus
159 || libusb_get_device_address(devlist[i]) != usb->address)
160 continue;
161 if ((ret = libusb_open(devlist[i], &usb->devhdl))) {
162 sr_err("Failed to open device: %s.", libusb_error_name(ret));
163 return SR_ERR;
164 }
165 break;
166 }
167 libusb_free_device_list(devlist, 1);
168 if (!devlist[i]) {
169 sr_err("Device not found.");
170 return SR_ERR;
171 }
172
173 /* The device reports as HID class, so the kernel would have
174 * claimed it. */
175 if (libusb_kernel_driver_active(usb->devhdl, 0) == 1) {
176 if ((ret = libusb_detach_kernel_driver(usb->devhdl, 0)) < 0) {
177 sr_err("Failed to detach kernel driver: %s.",
178 libusb_error_name(ret));
179 return SR_ERR;
180 }
181 }
182
183 if ((ret = libusb_claim_interface(usb->devhdl,
184 VICTOR_INTERFACE))) {
185 sr_err("Failed to claim interface: %s.", libusb_error_name(ret));
186 return SR_ERR;
187 }
188 sdi->status = SR_ST_ACTIVE;
189
190 return SR_OK;
191}
192
193static int dev_close(struct sr_dev_inst *sdi)
194{
195 struct sr_usb_dev_inst *usb;
196
197 if (!di->priv) {
198 sr_err("Driver was not initialized.");
199 return SR_ERR;
200 }
201
202 usb = sdi->conn;
203
204 if (!usb->devhdl)
205 /* Nothing to do. */
206 return SR_OK;
207
208 libusb_release_interface(usb->devhdl, VICTOR_INTERFACE);
209 libusb_close(usb->devhdl);
210 usb->devhdl = NULL;
211 sdi->status = SR_ST_INACTIVE;
212
213 return SR_OK;
214}
215
216static int cleanup(void)
217{
218 struct drv_context *drvc;
219
220 if (!(drvc = di->priv))
221 /* Can get called on an unused driver, doesn't matter. */
222 return SR_OK;
223
224 clear_instances();
225 g_free(drvc);
226 di->priv = NULL;
227
228 return SR_OK;
229}
230
231static int config_get(int id, GVariant **data, const struct sr_dev_inst *sdi)
232{
233 struct sr_usb_dev_inst *usb;
234 char str[128];
235
236 switch (id) {
237 case SR_CONF_CONN:
238 if (!sdi || !sdi->conn)
239 return SR_ERR_ARG;
240 usb = sdi->conn;
241 snprintf(str, 128, "%d.%d", usb->bus, usb->address);
242 *data = g_variant_new_string(str);
243 break;
244 default:
245 return SR_ERR_NA;
246 }
247
248 return SR_OK;
249}
250
251static int config_set(int id, GVariant *data, const struct sr_dev_inst *sdi)
252{
253 struct dev_context *devc;
254 gint64 now;
255 int ret;
256
257 if (sdi->status != SR_ST_ACTIVE)
258 return SR_ERR_DEV_CLOSED;
259
260 if (!di->priv) {
261 sr_err("Driver was not initialized.");
262 return SR_ERR;
263 }
264
265 devc = sdi->priv;
266 ret = SR_OK;
267 switch (id) {
268 case SR_CONF_LIMIT_MSEC:
269 devc->limit_msec = g_variant_get_uint64(data);
270 now = g_get_monotonic_time() / 1000;
271 devc->end_time = now + devc->limit_msec;
272 sr_dbg("Setting time limit to %" PRIu64 "ms.",
273 devc->limit_msec);
274 break;
275 case SR_CONF_LIMIT_SAMPLES:
276 devc->limit_samples = g_variant_get_uint64(data);
277 sr_dbg("Setting sample limit to %" PRIu64 ".",
278 devc->limit_samples);
279 break;
280 default:
281 ret = SR_ERR_NA;
282 }
283
284 return ret;
285}
286
287static int config_list(int key, GVariant **data, const struct sr_dev_inst *sdi)
288{
289
290 (void)sdi;
291
292 switch (key) {
293 case SR_CONF_SCAN_OPTIONS:
294 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32,
295 hwopts, ARRAY_SIZE(hwopts), sizeof(int32_t));
296 break;
297 case SR_CONF_DEVICE_OPTIONS:
298 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32,
299 hwcaps, ARRAY_SIZE(hwcaps), sizeof(int32_t));
300 break;
301 default:
302 return SR_ERR_NA;
303 }
304
305 return SR_OK;
306}
307
308static void receive_transfer(struct libusb_transfer *transfer)
309{
310 struct dev_context *devc;
311 struct sr_dev_inst *sdi;
312 int ret;
313
314 sdi = transfer->user_data;
315 devc = sdi->priv;
316 if (transfer->status == LIBUSB_TRANSFER_NO_DEVICE) {
317 /* USB device was unplugged. */
318 dev_acquisition_stop(sdi, sdi);
319 } else if (transfer->status == LIBUSB_TRANSFER_COMPLETED) {
320 sr_dbg("Got %d-byte packet.", transfer->actual_length);
321 if (transfer->actual_length == DMM_DATA_SIZE) {
322 victor_dmm_receive_data(sdi, transfer->buffer);
323 if (devc->limit_samples) {
324 if (devc->num_samples >= devc->limit_samples)
325 dev_acquisition_stop(sdi, sdi);
326 }
327 }
328 }
329 /* Anything else is either an error or a timeout, which is fine:
330 * we were just going to send another transfer request anyway. */
331
332 if (sdi->status == SR_ST_ACTIVE) {
333 /* Send the same request again. */
334 if ((ret = libusb_submit_transfer(transfer) != 0)) {
335 sr_err("Unable to resubmit transfer: %s.",
336 libusb_error_name(ret));
337 g_free(transfer->buffer);
338 libusb_free_transfer(transfer);
339 dev_acquisition_stop(sdi, sdi);
340 }
341 } else {
342 /* This was the last transfer we're going to receive, so
343 * clean up now. */
344 g_free(transfer->buffer);
345 libusb_free_transfer(transfer);
346 }
347}
348
349static int handle_events(int fd, int revents, void *cb_data)
350{
351 struct dev_context *devc;
352 struct drv_context *drvc = di->priv;
353 struct sr_datafeed_packet packet;
354 struct sr_dev_inst *sdi;
355 struct timeval tv;
356 gint64 now;
357 int i;
358
359 (void)fd;
360 (void)revents;
361
362 sdi = cb_data;
363 devc = sdi->priv;
364
365 if (devc->limit_msec) {
366 now = g_get_monotonic_time() / 1000;
367 if (now > devc->end_time)
368 dev_acquisition_stop(sdi, sdi);
369 }
370
371 if (sdi->status == SR_ST_STOPPING) {
372 for (i = 0; devc->usbfd[i] != -1; i++)
373 sr_source_remove(devc->usbfd[i]);
374
375 dev_close(sdi);
376
377 packet.type = SR_DF_END;
378 sr_session_send(cb_data, &packet);
379 }
380
381 memset(&tv, 0, sizeof(struct timeval));
382 libusb_handle_events_timeout_completed(drvc->sr_ctx->libusb_ctx, &tv,
383 NULL);
384
385 return TRUE;
386}
387
388static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
389{
390 struct dev_context *devc;
391 struct drv_context *drvc = di->priv;
392 const struct libusb_pollfd **pfd;
393 struct sr_usb_dev_inst *usb;
394 struct libusb_transfer *transfer;
395 int ret, i;
396 unsigned char *buf;
397
398 if (sdi->status != SR_ST_ACTIVE)
399 return SR_ERR_DEV_CLOSED;
400
401 if (!di->priv) {
402 sr_err("Driver was not initialized.");
403 return SR_ERR;
404 }
405
406 devc = sdi->priv;
407 usb = sdi->conn;
408 devc->cb_data = cb_data;
409
410 /* Send header packet to the session bus. */
411 std_session_send_df_header(cb_data, LOG_PREFIX);
412
413 pfd = libusb_get_pollfds(drvc->sr_ctx->libusb_ctx);
414 for (i = 0; pfd[i]; i++) {
415 /* Handle USB events every 100ms, for decent latency. */
416 sr_source_add(pfd[i]->fd, pfd[i]->events, 100,
417 handle_events, (void *)sdi);
418 /* We'll need to remove this fd later. */
419 devc->usbfd[i] = pfd[i]->fd;
420 }
421 devc->usbfd[i] = -1;
422
423 buf = g_try_malloc(DMM_DATA_SIZE);
424 transfer = libusb_alloc_transfer(0);
425 /* Each transfer request gets 100ms to arrive before it's restarted.
426 * The device only sends 1 transfer/second no matter how many
427 * times you ask, but we want to keep step with the USB events
428 * handling above. */
429 libusb_fill_interrupt_transfer(transfer, usb->devhdl,
430 VICTOR_ENDPOINT, buf, DMM_DATA_SIZE, receive_transfer,
431 cb_data, 100);
432 if ((ret = libusb_submit_transfer(transfer) != 0)) {
433 sr_err("Unable to submit transfer: %s.", libusb_error_name(ret));
434 libusb_free_transfer(transfer);
435 g_free(buf);
436 return SR_ERR;
437 }
438
439 return SR_OK;
440}
441
442static int dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data)
443{
444 (void)cb_data;
445
446 if (!di->priv) {
447 sr_err("Driver was not initialized.");
448 return SR_ERR;
449 }
450
451 if (sdi->status != SR_ST_ACTIVE) {
452 sr_err("Device not active, can't stop acquisition.");
453 return SR_ERR;
454 }
455
456 sdi->status = SR_ST_STOPPING;
457
458 return SR_OK;
459}
460
461SR_PRIV struct sr_dev_driver victor_dmm_driver_info = {
462 .name = "victor-dmm",
463 .longname = "Victor DMMs",
464 .api_version = 1,
465 .init = init,
466 .cleanup = cleanup,
467 .scan = scan,
468 .dev_list = dev_list,
469 .dev_clear = clear_instances,
470 .config_get = config_get,
471 .config_set = config_set,
472 .config_list = config_list,
473 .dev_open = dev_open,
474 .dev_close = dev_close,
475 .dev_acquisition_start = dev_acquisition_start,
476 .dev_acquisition_stop = dev_acquisition_stop,
477 .priv = NULL,
478};