src/hardware/kingst-la2016/api.c

   1 /*
   2  * This file is part of the libsigrok project.
   3  *
   4  * Copyright (C) 2020 Florian Schmidt <schmidt_florian@gmx.de>
   5  * Copyright (C) 2013 Marcus Comstedt <marcus@mc.pp.se>
   6  * Copyright (C) 2013 Bert Vermeulen <bert@biot.com>
   7  * Copyright (C) 2012 Joel Holdsworth <joel@airwebreathe.org.uk>
   8  *
   9  * This program is free software: you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation, either version 3 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21  */
  22
  23 /* mostly stolen from src/hardware/saleae-logic16/ */
  24
  25 #include <config.h>
  26 #include <glib.h>
  27 #include <libusb.h>
  28 #include <stdlib.h>
  29 #include <string.h>
  30 #include <math.h>
  31 #include <libsigrok/libsigrok.h>
  32 #include "libsigrok-internal.h"
  33 #include "protocol.h"
  34
  35 static const uint32_t scanopts[] = {
  36         SR_CONF_CONN,
  37 };
  38
  39 static const uint32_t drvopts[] = {
  40         SR_CONF_LOGIC_ANALYZER,
  41 };
  42
  43 static const uint32_t devopts[] = {
  44         /* TODO: SR_CONF_CONTINUOUS, */
  45         SR_CONF_CONN | SR_CONF_GET,
  46         SR_CONF_SAMPLERATE | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
  47         SR_CONF_LIMIT_SAMPLES | SR_CONF_SET | SR_CONF_GET | SR_CONF_LIST,
  48         SR_CONF_VOLTAGE_THRESHOLD | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
  49         SR_CONF_LOGIC_THRESHOLD | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
  50         SR_CONF_LOGIC_THRESHOLD_CUSTOM | SR_CONF_GET | SR_CONF_SET,
  51         SR_CONF_TRIGGER_MATCH | SR_CONF_LIST,
  52         SR_CONF_CAPTURE_RATIO | SR_CONF_GET | SR_CONF_SET,
  53 };
  54
  55 static const int32_t trigger_matches[] = {
  56         SR_TRIGGER_ZERO,
  57         SR_TRIGGER_ONE,
  58         SR_TRIGGER_RISING,
  59         SR_TRIGGER_FALLING,
  60 };
  61
  62 static const char *channel_names[] = {
  63         "0", "1", "2", "3", "4", "5", "6", "7",
  64         "8", "9", "10", "11", "12", "13", "14", "15",
  65 };
  66
  67 static const uint64_t samplerates_la2016[] = {
  68         SR_KHZ(20),
  69         SR_KHZ(50),
  70         SR_KHZ(100),
  71         SR_KHZ(200),
  72         SR_KHZ(500),
  73         SR_MHZ(1),
  74         SR_MHZ(2),
  75         SR_MHZ(4),
  76         SR_MHZ(5),
  77         SR_MHZ(8),
  78         SR_MHZ(10),
  79         SR_MHZ(20),
  80         SR_MHZ(50),
  81         SR_MHZ(100),
  82         SR_MHZ(200),
  83 };
  84
  85 static const uint64_t samplerates_la1016[] = {
  86         SR_KHZ(20),
  87         SR_KHZ(50),
  88         SR_KHZ(100),
  89         SR_KHZ(200),
  90         SR_KHZ(500),
  91         SR_MHZ(1),
  92         SR_MHZ(2),
  93         SR_MHZ(4),
  94         SR_MHZ(5),
  95         SR_MHZ(8),
  96         SR_MHZ(10),
  97         SR_MHZ(20),
  98         SR_MHZ(50),
  99         SR_MHZ(100),
 100 };
 101
 102 static const float logic_threshold_value[] = {
 103         1.58,
 104         2.5,
 105         1.165,
 106         1.5,
 107         1.25,
 108         0.9,
 109         0.75,
 110         0.60,
 111         0.45,
 112 };
 113
 114 static const char *logic_threshold[] = {
 115         "TTL 5V",
 116         "CMOS 5V",
 117         "CMOS 3.3V",
 118         "CMOS 3.0V",
 119         "CMOS 2.5V",
 120         "CMOS 1.8V",
 121         "CMOS 1.5V",
 122         "CMOS 1.2V",
 123         "CMOS 0.9V",
 124         "USER",
 125 };
 126
 127 #define MAX_NUM_LOGIC_THRESHOLD_ENTRIES ARRAY_SIZE(logic_threshold)
 128
 129 static GSList *scan(struct sr_dev_driver *di, GSList *options)
 130 {
 131         struct drv_context *drvc;
 132         struct dev_context *devc;
 133         struct sr_dev_inst *sdi;
 134         struct sr_usb_dev_inst *usb;
 135         struct sr_config *src;
 136         GSList *l;
 137         GSList *devices;
 138         GSList *conn_devices;
 139         struct libusb_device_descriptor des;
 140         libusb_device **devlist;
 141         unsigned int i, j;
 142         const char *conn;
 143         char connection_id[64];
 144         int64_t fw_updated;
 145         unsigned int dev_addr;
 146
 147         drvc = di->context;
 148
 149         conn = NULL;
 150         for (l = options; l; l = l->next) {
 151                 src = l->data;
 152                 switch (src->key) {
 153                 case SR_CONF_CONN:
 154                         conn = g_variant_get_string(src->data, NULL);
 155                         break;
 156                 }
 157         }
 158         if (conn)
 159                 conn_devices = sr_usb_find(drvc->sr_ctx->libusb_ctx, conn);
 160         else
 161                 conn_devices = NULL;
 162
 163         /* Find all LA2016 devices and upload firmware to them. */
 164         devices = NULL;
 165         libusb_get_device_list(drvc->sr_ctx->libusb_ctx, &devlist);
 166         for (i = 0; devlist[i]; i++) {
 167                 if (conn) {
 168                         usb = NULL;
 169                         for (l = conn_devices; l; l = l->next) {
 170                                 usb = l->data;
 171                                 if (usb->bus == libusb_get_bus_number(devlist[i]) &&
 172                                     usb->address == libusb_get_device_address(devlist[i]))
 173                                         break;
 174                         }
 175                         if (!l) {
 176                                 /* This device matched none of the ones that
 177                                  * matched the conn specification. */
 178                                 continue;
 179                         }
 180                 }
 181
 182                 libusb_get_device_descriptor(devlist[i], &des);
 183
 184                 if (usb_get_port_path(devlist[i], connection_id, sizeof(connection_id)) < 0)
 185                         continue;
 186
 187                 if (des.idVendor != LA2016_VID || des.idProduct != LA2016_PID)
 188                         continue;
 189
 190                 /* Already has the firmware */
 191                 sr_dbg("Found a LA2016 device.");
 192                 sdi = g_malloc0(sizeof(struct sr_dev_inst));
 193                 sdi->status = SR_ST_INITIALIZING;
 194                 sdi->connection_id = g_strdup(connection_id);
 195
 196                 fw_updated = 0;
 197                 dev_addr = libusb_get_device_address(devlist[i]);
 198                 if (des.iProduct != 2) {
 199                         sr_info("device at '%s' has no firmware loaded!", connection_id);
 200
 201                         if (la2016_upload_firmware(drvc->sr_ctx, devlist[i], des.idProduct) != SR_OK) {
 202                                 sr_err("uC firmware upload failed!");
 203                                 g_free(sdi->connection_id);
 204                                 g_free(sdi);
 205                                 continue;
 206                         }
 207                         fw_updated = g_get_monotonic_time();
 208                         dev_addr = 0xff; /* to mark that we don't know address yet... ugly */
 209                 }
 210
 211                 sdi->vendor = g_strdup("Kingst");
 212                 sdi->model = g_strdup("LA2016");
 213
 214                 for (j = 0; j < ARRAY_SIZE(channel_names); j++)
 215                         sr_channel_new(sdi, j, SR_CHANNEL_LOGIC, TRUE, channel_names[j]);
 216
 217                 devices = g_slist_append(devices, sdi);
 218
 219                 devc = g_malloc0(sizeof(struct dev_context));
 220                 sdi->priv = devc;
 221                 devc->fw_updated = fw_updated;
 222                 devc->threshold_voltage_idx = 0;
 223                 devc->threshold_voltage = logic_threshold_value[devc->threshold_voltage_idx];
 224
 225                 sdi->status = SR_ST_INACTIVE;
 226                 sdi->inst_type = SR_INST_USB;
 227
 228                 sdi->conn = sr_usb_dev_inst_new(
 229                         libusb_get_bus_number(devlist[i]),
 230                         dev_addr, NULL);
 231         }
 232         libusb_free_device_list(devlist, 1);
 233         g_slist_free_full(conn_devices, (GDestroyNotify)sr_usb_dev_inst_free);
 234
 235         return std_scan_complete(di, devices);
 236 }
 237
 238 static int la2016_dev_open(struct sr_dev_inst *sdi)
 239 {
 240         struct sr_dev_driver *di;
 241         libusb_device **devlist;
 242         struct sr_usb_dev_inst *usb;
 243         struct libusb_device_descriptor des;
 244         struct drv_context *drvc;
 245         int ret, i, device_count;
 246         char connection_id[64];
 247
 248         di = sdi->driver;
 249         drvc = di->context;
 250         usb = sdi->conn;
 251         ret = SR_ERR;
 252
 253         device_count = libusb_get_device_list(drvc->sr_ctx->libusb_ctx, &devlist);
 254         if (device_count < 0) {
 255                 sr_err("Failed to get device list: %s.", libusb_error_name(device_count));
 256                 return SR_ERR;
 257         }
 258
 259         for (i = 0; i < device_count; i++) {
 260                 libusb_get_device_descriptor(devlist[i], &des);
 261
 262                 if (des.idVendor != LA2016_VID || des.idProduct != LA2016_PID || des.iProduct != 2)
 263                         continue;
 264
 265                 if ((sdi->status == SR_ST_INITIALIZING) || (sdi->status == SR_ST_INACTIVE)) {
 266                         /*
 267                          * Check device by its physical USB bus/port address.
 268                          */
 269                         if (usb_get_port_path(devlist[i], connection_id, sizeof(connection_id)) < 0)
 270                                 continue;
 271
 272                         if (strcmp(sdi->connection_id, connection_id))
 273                                 /* This is not the one. */
 274                                 continue;
 275                 }
 276
 277                 if (!(ret = libusb_open(devlist[i], &usb->devhdl))) {
 278                         if (usb->address == 0xff)
 279                                 /*
 280                                  * First time we touch this device after FW
 281                                  * upload, so we don't know the address yet.
 282                                  */
 283                                 usb->address = libusb_get_device_address(devlist[i]);
 284                 } else {
 285                         sr_err("Failed to open device: %s.", libusb_error_name(ret));
 286                         ret = SR_ERR;
 287                         break;
 288                 }
 289
 290                 ret = libusb_claim_interface(usb->devhdl, USB_INTERFACE);
 291                 if (ret == LIBUSB_ERROR_BUSY) {
 292                         sr_err("Unable to claim USB interface. Another "
 293                                "program or driver has already claimed it.");
 294                         ret = SR_ERR;
 295                         break;
 296                 } else if (ret == LIBUSB_ERROR_NO_DEVICE) {
 297                         sr_err("Device has been disconnected.");
 298                         ret = SR_ERR;
 299                         break;
 300                 } else if (ret != 0) {
 301                         sr_err("Unable to claim interface: %s.", libusb_error_name(ret));
 302                         ret = SR_ERR;
 303                         break;
 304                 }
 305
 306                 if ((ret = la2016_init_device(sdi)) != SR_OK) {
 307                         sr_err("Failed to init device.");
 308                         break;
 309                 }
 310
 311                 sr_info("Opened device on %d.%d (logical) / %s (physical), interface %d.",
 312                         usb->bus, usb->address, sdi->connection_id, USB_INTERFACE);
 313
 314                 ret = SR_OK;
 315
 316                 break;
 317         }
 318
 319         libusb_free_device_list(devlist, 1);
 320
 321         if (ret != SR_OK) {
 322                 if (usb->devhdl) {
 323                         libusb_release_interface(usb->devhdl, USB_INTERFACE);
 324                         libusb_close(usb->devhdl);
 325                         usb->devhdl = NULL;
 326                 }
 327                 return SR_ERR;
 328         }
 329
 330         return SR_OK;
 331 }
 332
 333 static int dev_open(struct sr_dev_inst *sdi)
 334 {
 335         struct dev_context *devc;
 336         int64_t timediff_us, timediff_ms;
 337         uint64_t reset_done;
 338         uint64_t now;
 339         int ret;
 340
 341         devc = sdi->priv;
 342
 343         /*
 344          * If the firmware was recently uploaded, wait up to MAX_RENUM_DELAY_MS
 345          * milliseconds for the FX2 to renumerate.
 346          */
 347         ret = SR_ERR;
 348         if (devc->fw_updated > 0) {
 349                 sr_info("Waiting for device to reset after firmware upload.");
 350                 /* Takes >= 2000ms for the uC to be gone from the USB bus. */
 351                 reset_done = devc->fw_updated + 18 * (uint64_t)1e5; /* 1.8 seconds */
 352                 now = g_get_monotonic_time();
 353                 if (reset_done > now)
 354                         g_usleep(reset_done - now);
 355                 timediff_ms = 0;
 356                 while (timediff_ms < MAX_RENUM_DELAY_MS) {
 357                         g_usleep(200 * 1000);
 358
 359                         timediff_us = g_get_monotonic_time() - devc->fw_updated;
 360                         timediff_ms = timediff_us / 1000;
 361
 362                         if ((ret = la2016_dev_open(sdi)) == SR_OK)
 363                                 break;
 364                         sr_spew("Waited %" PRIi64 "ms.", timediff_ms);
 365                 }
 366                 if (ret != SR_OK) {
 367                         sr_err("Device failed to re-enumerate.");
 368                         return SR_ERR;
 369                 }
 370                 sr_info("Device came back after %" PRIi64 "ms.", timediff_ms);
 371         } else {
 372                 ret = la2016_dev_open(sdi);
 373         }
 374
 375         if (ret != SR_OK) {
 376                 sr_err("Unable to open device.");
 377                 return SR_ERR;
 378         }
 379
 380         return SR_OK;
 381 }
 382
 383 static int dev_close(struct sr_dev_inst *sdi)
 384 {
 385         struct sr_usb_dev_inst *usb;
 386
 387         usb = sdi->conn;
 388
 389         if (!usb->devhdl)
 390                 return SR_ERR_BUG;
 391
 392         la2016_deinit_device(sdi);
 393
 394         sr_info("Closing device on %d.%d (logical) / %s (physical) interface %d.",
 395                 usb->bus, usb->address, sdi->connection_id, USB_INTERFACE);
 396         libusb_release_interface(usb->devhdl, USB_INTERFACE);
 397         libusb_close(usb->devhdl);
 398         usb->devhdl = NULL;
 399
 400         return SR_OK;
 401 }
 402
 403 static int config_get(uint32_t key, GVariant **data,
 404         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
 405 {
 406         struct dev_context *devc;
 407         struct sr_usb_dev_inst *usb;
 408         double rounded;
 409
 410         (void)cg;
 411
 412         if (!sdi)
 413                 return SR_ERR_ARG;
 414         devc = sdi->priv;
 415
 416         switch (key) {
 417         case SR_CONF_CONN:
 418                 if (!sdi->conn)
 419                         return SR_ERR_ARG;
 420                 usb = sdi->conn;
 421                 if (usb->address == 255) {
 422                         /* Device still needs to re-enumerate after firmware
 423                          * upload, so we don't know its (future) address. */
 424                         return SR_ERR;
 425                 }
 426                 *data = g_variant_new_printf("%d.%d", usb->bus, usb->address);
 427                 break;
 428         case SR_CONF_SAMPLERATE:
 429                 *data = g_variant_new_uint64(devc->cur_samplerate);
 430                 break;
 431         case SR_CONF_LIMIT_SAMPLES:
 432                 *data = g_variant_new_uint64(devc->limit_samples);
 433                 break;
 434         case SR_CONF_CAPTURE_RATIO:
 435                 *data = g_variant_new_uint64(devc->capture_ratio);
 436                 break;
 437         case SR_CONF_VOLTAGE_THRESHOLD:
 438                 rounded = (int)(devc->threshold_voltage / 0.1) * 0.1;
 439                 *data = std_gvar_tuple_double(rounded, rounded + 0.1);
 440                 return SR_OK;
 441         case SR_CONF_LOGIC_THRESHOLD:
 442                 *data = g_variant_new_string(logic_threshold[devc->threshold_voltage_idx]);
 443                 break;
 444         case SR_CONF_LOGIC_THRESHOLD_CUSTOM:
 445                 *data = g_variant_new_double(devc->threshold_voltage);
 446                 break;
 447
 448         default:
 449                 return SR_ERR_NA;
 450         }
 451
 452         return SR_OK;
 453 }
 454
 455 static int config_set(uint32_t key, GVariant *data,
 456         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
 457 {
 458         struct dev_context *devc;
 459         double low, high;
 460         int idx;
 461
 462         (void)cg;
 463
 464         devc = sdi->priv;
 465
 466         switch (key) {
 467         case SR_CONF_SAMPLERATE:
 468                 devc->cur_samplerate = g_variant_get_uint64(data);
 469                 break;
 470         case SR_CONF_LIMIT_SAMPLES:
 471                 devc->limit_samples = g_variant_get_uint64(data);
 472                 break;
 473         case SR_CONF_CAPTURE_RATIO:
 474                 devc->capture_ratio = g_variant_get_uint64(data);
 475                 break;
 476         case SR_CONF_VOLTAGE_THRESHOLD:
 477                 g_variant_get(data, "(dd)", &low, &high);
 478                 devc->threshold_voltage = (low + high) / 2.0;
 479                 devc->threshold_voltage_idx = MAX_NUM_LOGIC_THRESHOLD_ENTRIES - 1; /* USER */
 480                 break;
 481         case SR_CONF_LOGIC_THRESHOLD: {
 482                 if ((idx = std_str_idx(data, logic_threshold, MAX_NUM_LOGIC_THRESHOLD_ENTRIES)) < 0)
 483                         return SR_ERR_ARG;
 484                 if (idx == MAX_NUM_LOGIC_THRESHOLD_ENTRIES - 1) {
 485                         /* user threshold */
 486                 } else {
 487                         devc->threshold_voltage = logic_threshold_value[idx];
 488                 }
 489                 devc->threshold_voltage_idx = idx;
 490                 break;
 491         }
 492         case SR_CONF_LOGIC_THRESHOLD_CUSTOM:
 493                 devc->threshold_voltage = g_variant_get_double(data);
 494                 break;
 495         default:
 496                 return SR_ERR_NA;
 497         }
 498
 499         return SR_OK;
 500 }
 501
 502 static int config_list(uint32_t key, GVariant **data,
 503         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
 504 {
 505         struct dev_context *devc;
 506
 507         switch (key) {
 508         case SR_CONF_SCAN_OPTIONS:
 509         case SR_CONF_DEVICE_OPTIONS:
 510                 return STD_CONFIG_LIST(key, data, sdi, cg, scanopts, drvopts, devopts);
 511         case SR_CONF_SAMPLERATE:
 512                 if (!sdi)
 513                         return SR_ERR_ARG;
 514                 devc = sdi->priv;
 515                 if (devc->max_samplerate == SR_MHZ(200)) {
 516                         *data = std_gvar_samplerates(ARRAY_AND_SIZE(samplerates_la2016));
 517                 }
 518                 else {
 519                         *data = std_gvar_samplerates(ARRAY_AND_SIZE(samplerates_la1016));
 520                 }
 521                 break;
 522         case SR_CONF_LIMIT_SAMPLES:
 523                 *data = std_gvar_tuple_u64(LA2016_NUM_SAMPLES_MIN, LA2016_NUM_SAMPLES_MAX);
 524                 break;
 525         case SR_CONF_VOLTAGE_THRESHOLD:
 526                 *data = std_gvar_min_max_step_thresholds(
 527                         LA2016_THR_VOLTAGE_MIN,
 528                         LA2016_THR_VOLTAGE_MAX, 0.1);
 529                 break;
 530         case SR_CONF_TRIGGER_MATCH:
 531                 *data = std_gvar_array_i32(ARRAY_AND_SIZE(trigger_matches));
 532                 break;
 533         case SR_CONF_LOGIC_THRESHOLD:
 534                 *data = g_variant_new_strv(logic_threshold, MAX_NUM_LOGIC_THRESHOLD_ENTRIES);
 535                 break;
 536         default:
 537                 return SR_ERR_NA;
 538         }
 539
 540         return SR_OK;
 541 }
 542
 543 static void send_chunk(struct sr_dev_inst *sdi,
 544         const uint8_t *packets, unsigned int num_tfers)
 545 {
 546         struct dev_context *devc;
 547         struct sr_datafeed_logic logic;
 548         struct sr_datafeed_packet sr_packet;
 549         unsigned int max_samples, n_samples, total_samples, free_n_samples;
 550         unsigned int i, j, k;
 551         int do_signal_trigger;
 552         uint16_t *wp;
 553         const uint8_t *rp;
 554         uint16_t state;
 555         uint8_t repetitions;
 556
 557         devc = sdi->priv;
 558
 559         logic.unitsize = 2;
 560         logic.data = devc->convbuffer;
 561
 562         sr_packet.type = SR_DF_LOGIC;
 563         sr_packet.payload = &logic;
 564
 565         max_samples = devc->convbuffer_size / 2;
 566         n_samples = 0;
 567         wp = (uint16_t *)devc->convbuffer;
 568         total_samples = 0;
 569         do_signal_trigger = 0;
 570
 571         if (devc->had_triggers_configured && devc->reading_behind_trigger == 0 && devc->info.n_rep_packets_before_trigger == 0) {
 572                 std_session_send_df_trigger(sdi);
 573                 devc->reading_behind_trigger = 1;
 574         }
 575
 576         rp = packets;
 577         for (i = 0; i < num_tfers; i++) {
 578                 for (k = 0; k < NUM_PACKETS_IN_CHUNK; k++) {
 579                         free_n_samples = max_samples - n_samples;
 580                         if (free_n_samples < 256 || do_signal_trigger) {
 581                                 logic.length = n_samples * 2;
 582                                 sr_session_send(sdi, &sr_packet);
 583                                 n_samples = 0;
 584                                 wp = (uint16_t *)devc->convbuffer;
 585                                 if (do_signal_trigger) {
 586                                         std_session_send_df_trigger(sdi);
 587                                         do_signal_trigger = 0;
 588                                 }
 589                         }
 590
 591                         state = read_u16le_inc(&rp);
 592                         repetitions = read_u8_inc(&rp);
 593                         for (j = 0; j < repetitions; j++)
 594                                 *wp++ = state;
 595
 596                         n_samples += repetitions;
 597                         total_samples += repetitions;
 598                         devc->total_samples += repetitions;
 599                         if (!devc->reading_behind_trigger) {
 600                                 devc->n_reps_until_trigger--;
 601                                 if (devc->n_reps_until_trigger == 0) {
 602                                         devc->reading_behind_trigger = 1;
 603                                         do_signal_trigger = 1;
 604                                         sr_dbg("  here is trigger position after %" PRIu64 " samples, %.6fms",
 605                                                devc->total_samples,
 606                                                (double)devc->total_samples / devc->cur_samplerate * 1e3);
 607                                 }
 608                         }
 609                 }
 610                 (void)read_u8_inc(&rp); /* Skip sequence number. */
 611         }
 612         if (n_samples) {
 613                 logic.length = n_samples * 2;
 614                 sr_session_send(sdi, &sr_packet);
 615                 if (do_signal_trigger) {
 616                         std_session_send_df_trigger(sdi);
 617                 }
 618         }
 619         sr_dbg("send_chunk done after %d samples", total_samples);
 620 }
 621
 622 static void LIBUSB_CALL receive_transfer(struct libusb_transfer *transfer)
 623 {
 624         struct sr_dev_inst *sdi;
 625         struct dev_context *devc;
 626         struct sr_usb_dev_inst *usb;
 627         int ret;
 628
 629         sdi = transfer->user_data;
 630         devc = sdi->priv;
 631         usb = sdi->conn;
 632
 633         sr_dbg("receive_transfer(): status %s received %d bytes.",
 634                libusb_error_name(transfer->status), transfer->actual_length);
 635
 636         if (transfer->status == LIBUSB_TRANSFER_TIMED_OUT) {
 637                 sr_err("bulk transfer timeout!");
 638                 devc->transfer_finished = 1;
 639         }
 640         send_chunk(sdi, transfer->buffer, transfer->actual_length / TRANSFER_PACKET_LENGTH);
 641
 642         devc->n_bytes_to_read -= transfer->actual_length;
 643         if (devc->n_bytes_to_read) {
 644                 uint32_t to_read = devc->n_bytes_to_read;
 645                 /* determine read size for the next usb transfer */
 646                 if (to_read >= LA2016_USB_BUFSZ)
 647                         to_read = LA2016_USB_BUFSZ;
 648                 else /* last transfer, make read size some multiple of LA2016_EP6_PKTSZ */
 649                         to_read = (to_read + (LA2016_EP6_PKTSZ-1)) & ~(LA2016_EP6_PKTSZ-1);
 650                 libusb_fill_bulk_transfer(
 651                         transfer, usb->devhdl,
 652                         0x86, transfer->buffer, to_read,
 653                         receive_transfer, (void *)sdi, DEFAULT_TIMEOUT_MS);
 654
 655                 if ((ret = libusb_submit_transfer(transfer)) == 0)
 656                         return;
 657                 sr_err("Failed to submit further transfer: %s.", libusb_error_name(ret));
 658         }
 659
 660         g_free(transfer->buffer);
 661         libusb_free_transfer(transfer);
 662         devc->transfer_finished = 1;
 663 }
 664
 665 static int handle_event(int fd, int revents, void *cb_data)
 666 {
 667         const struct sr_dev_inst *sdi;
 668         struct dev_context *devc;
 669         struct drv_context *drvc;
 670         struct timeval tv;
 671
 672         (void)fd;
 673         (void)revents;
 674
 675         sdi = cb_data;
 676         devc = sdi->priv;
 677         drvc = sdi->driver->context;
 678
 679         if (devc->have_trigger == 0) {
 680                 if (la2016_has_triggered(sdi) == 0) {
 681                         /* not yet ready for download */
 682                         return TRUE;
 683                 }
 684                 devc->have_trigger = 1;
 685                 devc->transfer_finished = 0;
 686                 devc->reading_behind_trigger = 0;
 687                 devc->total_samples = 0;
 688                 /* we can start retrieving data! */
 689                 if (la2016_start_retrieval(sdi, receive_transfer) != SR_OK) {
 690                         sr_err("failed to start retrieval!");
 691                         return FALSE;
 692                 }
 693                 sr_dbg("retrieval is started...");
 694                 std_session_send_df_frame_begin(sdi);
 695
 696                 return TRUE;
 697         }
 698
 699         tv.tv_sec = tv.tv_usec = 0;
 700         libusb_handle_events_timeout(drvc->sr_ctx->libusb_ctx, &tv);
 701
 702         if (devc->transfer_finished) {
 703                 sr_dbg("transfer is finished!");
 704                 std_session_send_df_frame_end(sdi);
 705
 706                 usb_source_remove(sdi->session, drvc->sr_ctx);
 707                 std_session_send_df_end(sdi);
 708
 709                 la2016_stop_acquisition(sdi);
 710
 711                 g_free(devc->convbuffer);
 712                 devc->convbuffer = NULL;
 713
 714                 devc->transfer = NULL;
 715
 716                 sr_dbg("transfer is now finished");
 717         }
 718
 719         return TRUE;
 720 }
 721
 722 static void abort_acquisition(struct dev_context *devc)
 723 {
 724         if (devc->transfer)
 725                 libusb_cancel_transfer(devc->transfer);
 726 }
 727
 728 static int configure_channels(const struct sr_dev_inst *sdi)
 729 {
 730         struct dev_context *devc;
 731
 732         devc = sdi->priv;
 733         devc->cur_channels = 0;
 734         devc->num_channels = 0;
 735
 736         for (GSList *l = sdi->channels; l; l = l->next) {
 737                 struct sr_channel *ch = (struct sr_channel*)l->data;
 738                 if (ch->enabled == FALSE)
 739                         continue;
 740                 devc->cur_channels |= 1 << ch->index;
 741                 devc->num_channels++;
 742         }
 743
 744         return SR_OK;
 745 }
 746
 747 static int dev_acquisition_start(const struct sr_dev_inst *sdi)
 748 {
 749         struct sr_dev_driver *di;
 750         struct drv_context *drvc;
 751         struct dev_context *devc;
 752         int ret;
 753
 754         di = sdi->driver;
 755         drvc = di->context;
 756         devc = sdi->priv;
 757
 758         if (configure_channels(sdi) != SR_OK) {
 759                 sr_err("Failed to configure channels.");
 760                 return SR_ERR;
 761         }
 762
 763         devc->convbuffer_size = 4 * 1024 * 1024;
 764         if (!(devc->convbuffer = g_try_malloc(devc->convbuffer_size))) {
 765                 sr_err("Conversion buffer malloc failed.");
 766                 return SR_ERR_MALLOC;
 767         }
 768
 769         if ((ret = la2016_setup_acquisition(sdi)) != SR_OK) {
 770                 g_free(devc->convbuffer);
 771                 devc->convbuffer = NULL;
 772                 return ret;
 773         }
 774
 775         devc->ctx = drvc->sr_ctx;
 776
 777         if ((ret = la2016_start_acquisition(sdi)) != SR_OK) {
 778                 abort_acquisition(devc);
 779                 return ret;
 780         }
 781
 782         devc->have_trigger = 0;
 783         usb_source_add(sdi->session, drvc->sr_ctx, 50, handle_event, (void *)sdi);
 784
 785         std_session_send_df_header(sdi);
 786
 787         return SR_OK;
 788 }
 789
 790 static int dev_acquisition_stop(struct sr_dev_inst *sdi)
 791 {
 792         int ret;
 793
 794         ret = la2016_abort_acquisition(sdi);
 795         abort_acquisition(sdi->priv);
 796
 797         return ret;
 798 }
 799
 800 static struct sr_dev_driver kingst_la2016_driver_info = {
 801         .name = "kingst-la2016",
 802         .longname = "Kingst LA2016",
 803         .api_version = 1,
 804         .init = std_init,
 805         .cleanup = std_cleanup,
 806         .scan = scan,
 807         .dev_list = std_dev_list,
 808         .dev_clear = std_dev_clear,
 809         .config_get = config_get,
 810         .config_set = config_set,
 811         .config_list = config_list,
 812         .dev_open = dev_open,
 813         .dev_close = dev_close,
 814         .dev_acquisition_start = dev_acquisition_start,
 815         .dev_acquisition_stop = dev_acquisition_stop,
 816         .context = NULL,
 817 };
 818 SR_REGISTER_DEV_DRIVER(kingst_la2016_driver_info);