src/hardware/dreamsourcelab-dslogic/api.c

   1 /*
   2  * This file is part of the libsigrok project.
   3  *
   4  * Copyright (C) 2013 Bert Vermeulen <bert@biot.com>
   5  * Copyright (C) 2012 Joel Holdsworth <joel@airwebreathe.org.uk>
   6  *
   7  * This program is free software: you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation, either version 3 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21 #include <config.h>
  22 #include <math.h>
  23 #include "protocol.h"
  24
  25 static const struct dslogic_profile supported_device[] = {
  26         /* DreamSourceLab DSLogic */
  27         { 0x2a0e, 0x0001, "DreamSourceLab", "DSLogic", NULL,
  28                 "dreamsourcelab-dslogic-fx2.fw",
  29                 0, "DreamSourceLab", "DSLogic", 256 * 1024 * 1024},
  30         /* DreamSourceLab DSCope */
  31         { 0x2a0e, 0x0002, "DreamSourceLab", "DSCope", NULL,
  32                 "dreamsourcelab-dscope-fx2.fw",
  33                 0, "DreamSourceLab", "DSCope", 256 * 1024 * 1024},
  34         /* DreamSourceLab DSLogic Pro */
  35         { 0x2a0e, 0x0003, "DreamSourceLab", "DSLogic Pro", NULL,
  36                 "dreamsourcelab-dslogic-pro-fx2.fw",
  37                 0, "DreamSourceLab", "DSLogic", 256 * 1024 * 1024},
  38         /* DreamSourceLab DSLogic Plus */
  39         { 0x2a0e, 0x0020, "DreamSourceLab", "DSLogic Plus", NULL,
  40                 "dreamsourcelab-dslogic-plus-fx2.fw",
  41                 0, "DreamSourceLab", "DSLogic", 256 * 1024 * 1024},
  42         /* DreamSourceLab DSLogic Basic */
  43         { 0x2a0e, 0x0021, "DreamSourceLab", "DSLogic Basic", NULL,
  44                 "dreamsourcelab-dslogic-basic-fx2.fw",
  45                 0, "DreamSourceLab", "DSLogic", 256 * 1024},
  46
  47         ALL_ZERO
  48 };
  49
  50 static const uint32_t scanopts[] = {
  51         SR_CONF_CONN,
  52 };
  53
  54 static const uint32_t drvopts[] = {
  55         SR_CONF_LOGIC_ANALYZER,
  56 };
  57
  58 static const uint32_t devopts[] = {
  59         SR_CONF_CONTINUOUS | SR_CONF_SET | SR_CONF_GET,
  60         SR_CONF_LIMIT_SAMPLES | SR_CONF_GET | SR_CONF_SET,
  61         SR_CONF_VOLTAGE_THRESHOLD | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
  62         SR_CONF_CONN | SR_CONF_GET,
  63         SR_CONF_SAMPLERATE | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
  64         SR_CONF_TRIGGER_MATCH | SR_CONF_LIST,
  65         SR_CONF_CAPTURE_RATIO | SR_CONF_GET | SR_CONF_SET,
  66         SR_CONF_EXTERNAL_CLOCK | SR_CONF_GET | SR_CONF_SET,
  67         SR_CONF_CLOCK_EDGE | SR_CONF_GET | SR_CONF_SET | SR_CONF_LIST,
  68 };
  69
  70 static const int32_t trigger_matches[] = {
  71         SR_TRIGGER_ZERO,
  72         SR_TRIGGER_ONE,
  73         SR_TRIGGER_RISING,
  74         SR_TRIGGER_FALLING,
  75         SR_TRIGGER_EDGE,
  76 };
  77
  78 static const char *signal_edges[] = {
  79         [DS_EDGE_RISING] = "rising",
  80         [DS_EDGE_FALLING] = "falling",
  81 };
  82
  83 static const double thresholds[][2] = {
  84         { 0.7, 1.4 },
  85         { 1.4, 3.6 },
  86 };
  87
  88 static const uint64_t samplerates[] = {
  89         SR_KHZ(10),
  90         SR_KHZ(20),
  91         SR_KHZ(50),
  92         SR_KHZ(100),
  93         SR_KHZ(200),
  94         SR_KHZ(500),
  95         SR_MHZ(1),
  96         SR_MHZ(2),
  97         SR_MHZ(5),
  98         SR_MHZ(10),
  99         SR_MHZ(20),
 100         SR_MHZ(25),
 101         SR_MHZ(50),
 102         SR_MHZ(100),
 103         SR_MHZ(200),
 104         SR_MHZ(400),
 105 };
 106
 107 static gboolean is_plausible(const struct libusb_device_descriptor *des)
 108 {
 109         int i;
 110
 111         for (i = 0; supported_device[i].vid; i++) {
 112                 if (des->idVendor != supported_device[i].vid)
 113                         continue;
 114                 if (des->idProduct == supported_device[i].pid)
 115                         return TRUE;
 116         }
 117
 118         return FALSE;
 119 }
 120
 121 static GSList *scan(struct sr_dev_driver *di, GSList *options)
 122 {
 123         struct drv_context *drvc;
 124         struct dev_context *devc;
 125         struct sr_dev_inst *sdi;
 126         struct sr_usb_dev_inst *usb;
 127         struct sr_channel *ch;
 128         struct sr_channel_group *cg;
 129         struct sr_config *src;
 130         const struct dslogic_profile *prof;
 131         GSList *l, *devices, *conn_devices;
 132         gboolean has_firmware;
 133         struct libusb_device_descriptor des;
 134         libusb_device **devlist;
 135         struct libusb_device_handle *hdl;
 136         int ret, i, j;
 137         const char *conn;
 138         char manufacturer[64], product[64], serial_num[64], connection_id[64];
 139         char channel_name[16];
 140
 141         drvc = di->context;
 142
 143         conn = NULL;
 144         for (l = options; l; l = l->next) {
 145                 src = l->data;
 146                 switch (src->key) {
 147                 case SR_CONF_CONN:
 148                         conn = g_variant_get_string(src->data, NULL);
 149                         break;
 150                 }
 151         }
 152         if (conn)
 153                 conn_devices = sr_usb_find(drvc->sr_ctx->libusb_ctx, conn);
 154         else
 155                 conn_devices = NULL;
 156
 157         /* Find all DSLogic compatible devices and upload firmware to them. */
 158         devices = NULL;
 159         libusb_get_device_list(drvc->sr_ctx->libusb_ctx, &devlist);
 160         for (i = 0; devlist[i]; i++) {
 161                 if (conn) {
 162                         usb = NULL;
 163                         for (l = conn_devices; l; l = l->next) {
 164                                 usb = l->data;
 165                                 if (usb->bus == libusb_get_bus_number(devlist[i])
 166                                         && usb->address == libusb_get_device_address(devlist[i]))
 167                                         break;
 168                         }
 169                         if (!l)
 170                                 /* This device matched none of the ones that
 171                                  * matched the conn specification. */
 172                                 continue;
 173                 }
 174
 175                 libusb_get_device_descriptor(devlist[i], &des);
 176
 177                 if (!is_plausible(&des))
 178                         continue;
 179
 180                 if ((ret = libusb_open(devlist[i], &hdl)) < 0) {
 181                         sr_warn("Failed to open potential device with "
 182                                 "VID:PID %04x:%04x: %s.", des.idVendor,
 183                                 des.idProduct, libusb_error_name(ret));
 184                         continue;
 185                 }
 186
 187                 if (des.iManufacturer == 0) {
 188                         manufacturer[0] = '\0';
 189                 } else if ((ret = libusb_get_string_descriptor_ascii(hdl,
 190                                 des.iManufacturer, (unsigned char *) manufacturer,
 191                                 sizeof(manufacturer))) < 0) {
 192                         sr_warn("Failed to get manufacturer string descriptor: %s.",
 193                                 libusb_error_name(ret));
 194                         continue;
 195                 }
 196
 197                 if (des.iProduct == 0) {
 198                         product[0] = '\0';
 199                 } else if ((ret = libusb_get_string_descriptor_ascii(hdl,
 200                                 des.iProduct, (unsigned char *) product,
 201                                 sizeof(product))) < 0) {
 202                         sr_warn("Failed to get product string descriptor: %s.",
 203                                 libusb_error_name(ret));
 204                         continue;
 205                 }
 206
 207                 if (des.iSerialNumber == 0) {
 208                         serial_num[0] = '\0';
 209                 } else if ((ret = libusb_get_string_descriptor_ascii(hdl,
 210                                 des.iSerialNumber, (unsigned char *) serial_num,
 211                                 sizeof(serial_num))) < 0) {
 212                         sr_warn("Failed to get serial number string descriptor: %s.",
 213                                 libusb_error_name(ret));
 214                         continue;
 215                 }
 216
 217                 usb_get_port_path(devlist[i], connection_id, sizeof(connection_id));
 218
 219                 libusb_close(hdl);
 220
 221                 prof = NULL;
 222                 for (j = 0; supported_device[j].vid; j++) {
 223                         if (des.idVendor == supported_device[j].vid &&
 224                                         des.idProduct == supported_device[j].pid) {
 225                                 prof = &supported_device[j];
 226                                 break;
 227                         }
 228                 }
 229
 230                 if (!prof)
 231                         continue;
 232
 233                 sdi = g_malloc0(sizeof(struct sr_dev_inst));
 234                 sdi->status = SR_ST_INITIALIZING;
 235                 sdi->vendor = g_strdup(prof->vendor);
 236                 sdi->model = g_strdup(prof->model);
 237                 sdi->version = g_strdup(prof->model_version);
 238                 sdi->serial_num = g_strdup(serial_num);
 239                 sdi->connection_id = g_strdup(connection_id);
 240
 241                 /* Logic channels, all in one channel group. */
 242                 cg = g_malloc0(sizeof(struct sr_channel_group));
 243                 cg->name = g_strdup("Logic");
 244                 for (j = 0; j < NUM_CHANNELS; j++) {
 245                         sprintf(channel_name, "%d", j);
 246                         ch = sr_channel_new(sdi, j, SR_CHANNEL_LOGIC,
 247                                                 TRUE, channel_name);
 248                         cg->channels = g_slist_append(cg->channels, ch);
 249                 }
 250                 sdi->channel_groups = g_slist_append(NULL, cg);
 251
 252                 devc = dslogic_dev_new();
 253                 devc->profile = prof;
 254                 sdi->priv = devc;
 255                 devices = g_slist_append(devices, sdi);
 256
 257                 devc->samplerates = samplerates;
 258                 devc->num_samplerates = ARRAY_SIZE(samplerates);
 259                 has_firmware = usb_match_manuf_prod(devlist[i], "DreamSourceLab", "USB-based Instrument");
 260
 261                 if (has_firmware) {
 262                         /* Already has the firmware, so fix the new address. */
 263                         sr_dbg("Found a DSLogic device.");
 264                         sdi->status = SR_ST_INACTIVE;
 265                         sdi->inst_type = SR_INST_USB;
 266                         sdi->conn = sr_usb_dev_inst_new(libusb_get_bus_number(devlist[i]),
 267                                         libusb_get_device_address(devlist[i]), NULL);
 268                 } else {
 269                         if (ezusb_upload_firmware(drvc->sr_ctx, devlist[i],
 270                                         USB_CONFIGURATION, prof->firmware) == SR_OK)
 271                                 /* Store when this device's FW was updated. */
 272                                 devc->fw_updated = g_get_monotonic_time();
 273                         else
 274                                 sr_err("Firmware upload failed for "
 275                                        "device %d.%d (logical).",
 276                                        libusb_get_bus_number(devlist[i]),
 277                                        libusb_get_device_address(devlist[i]));
 278                         sdi->inst_type = SR_INST_USB;
 279                         sdi->conn = sr_usb_dev_inst_new(libusb_get_bus_number(devlist[i]),
 280                                         0xff, NULL);
 281                 }
 282         }
 283         libusb_free_device_list(devlist, 1);
 284         g_slist_free_full(conn_devices, (GDestroyNotify)sr_usb_dev_inst_free);
 285
 286         return std_scan_complete(di, devices);
 287 }
 288
 289 static int dev_open(struct sr_dev_inst *sdi)
 290 {
 291         struct sr_dev_driver *di = sdi->driver;
 292         struct sr_usb_dev_inst *usb;
 293         struct dev_context *devc;
 294         int ret;
 295         int64_t timediff_us, timediff_ms;
 296
 297         devc = sdi->priv;
 298         usb = sdi->conn;
 299
 300         /*
 301          * If the firmware was recently uploaded, wait up to MAX_RENUM_DELAY_MS
 302          * milliseconds for the FX2 to renumerate.
 303          */
 304         ret = SR_ERR;
 305         if (devc->fw_updated > 0) {
 306                 sr_info("Waiting for device to reset.");
 307                 /* Takes >= 300ms for the FX2 to be gone from the USB bus. */
 308                 g_usleep(300 * 1000);
 309                 timediff_ms = 0;
 310                 while (timediff_ms < MAX_RENUM_DELAY_MS) {
 311                         if ((ret = dslogic_dev_open(sdi, di)) == SR_OK)
 312                                 break;
 313                         g_usleep(100 * 1000);
 314
 315                         timediff_us = g_get_monotonic_time() - devc->fw_updated;
 316                         timediff_ms = timediff_us / 1000;
 317                         sr_spew("Waited %" PRIi64 "ms.", timediff_ms);
 318                 }
 319                 if (ret != SR_OK) {
 320                         sr_err("Device failed to renumerate.");
 321                         return SR_ERR;
 322                 }
 323                 sr_info("Device came back after %" PRIi64 "ms.", timediff_ms);
 324         } else {
 325                 sr_info("Firmware upload was not needed.");
 326                 ret = dslogic_dev_open(sdi, di);
 327         }
 328
 329         if (ret != SR_OK) {
 330                 sr_err("Unable to open device.");
 331                 return SR_ERR;
 332         }
 333
 334         ret = libusb_claim_interface(usb->devhdl, USB_INTERFACE);
 335         if (ret != 0) {
 336                 switch (ret) {
 337                 case LIBUSB_ERROR_BUSY:
 338                         sr_err("Unable to claim USB interface. Another "
 339                                "program or driver has already claimed it.");
 340                         break;
 341                 case LIBUSB_ERROR_NO_DEVICE:
 342                         sr_err("Device has been disconnected.");
 343                         break;
 344                 default:
 345                         sr_err("Unable to claim interface: %s.",
 346                                libusb_error_name(ret));
 347                         break;
 348                 }
 349
 350                 return SR_ERR;
 351         }
 352
 353
 354         if ((ret = dslogic_fpga_firmware_upload(sdi)) != SR_OK)
 355                 return ret;
 356
 357         if (devc->cur_samplerate == 0) {
 358                 /* Samplerate hasn't been set; default to the slowest one. */
 359                 devc->cur_samplerate = devc->samplerates[0];
 360         }
 361
 362         if (devc->cur_threshold == 0.0)
 363                 devc->cur_threshold = 1.5;
 364
 365         return SR_OK;
 366 }
 367
 368 static int dev_close(struct sr_dev_inst *sdi)
 369 {
 370         struct sr_usb_dev_inst *usb;
 371
 372         usb = sdi->conn;
 373
 374         if (!usb->devhdl)
 375                 return SR_ERR_BUG;
 376
 377         sr_info("Closing device on %d.%d (logical) / %s (physical) interface %d.",
 378                 usb->bus, usb->address, sdi->connection_id, USB_INTERFACE);
 379         libusb_release_interface(usb->devhdl, USB_INTERFACE);
 380         libusb_close(usb->devhdl);
 381         usb->devhdl = NULL;
 382
 383         return SR_OK;
 384 }
 385
 386 static int config_get(uint32_t key, GVariant **data,
 387         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
 388 {
 389         struct dev_context *devc;
 390         struct sr_usb_dev_inst *usb;
 391         int idx;
 392
 393         (void)cg;
 394
 395         if (!sdi)
 396                 return SR_ERR_ARG;
 397
 398         devc = sdi->priv;
 399
 400         switch (key) {
 401         case SR_CONF_CONN:
 402                 if (!sdi->conn)
 403                         return SR_ERR_ARG;
 404                 usb = sdi->conn;
 405                 if (usb->address == 255)
 406                         /* Device still needs to re-enumerate after firmware
 407                          * upload, so we don't know its (future) address. */
 408                         return SR_ERR;
 409                 *data = g_variant_new_printf("%d.%d", usb->bus, usb->address);
 410                 break;
 411         case SR_CONF_VOLTAGE_THRESHOLD:
 412                 if (!strcmp(devc->profile->model, "DSLogic")) {
 413                         if ((idx = std_double_tuple_idx_d0(devc->cur_threshold,
 414                                         ARRAY_AND_SIZE(thresholds))) < 0)
 415                                 return SR_ERR_BUG;
 416                         *data = std_gvar_tuple_double(thresholds[idx][0],
 417                                         thresholds[idx][1]);
 418                 } else {
 419                         *data = std_gvar_tuple_double(devc->cur_threshold, devc->cur_threshold);
 420                 }
 421                 break;
 422         case SR_CONF_LIMIT_SAMPLES:
 423                 *data = g_variant_new_uint64(devc->limit_samples);
 424                 break;
 425         case SR_CONF_SAMPLERATE:
 426                 *data = g_variant_new_uint64(devc->cur_samplerate);
 427                 break;
 428         case SR_CONF_CAPTURE_RATIO:
 429                 *data = g_variant_new_uint64(devc->capture_ratio);
 430                 break;
 431         case SR_CONF_EXTERNAL_CLOCK:
 432                 *data = g_variant_new_boolean(devc->external_clock);
 433                 break;
 434         case SR_CONF_CONTINUOUS:
 435                 *data = g_variant_new_boolean(devc->continuous_mode);
 436                 break;
 437         case SR_CONF_CLOCK_EDGE:
 438                 idx = devc->clock_edge;
 439                 if (idx >= (int)ARRAY_SIZE(signal_edges))
 440                         return SR_ERR_BUG;
 441                 *data = g_variant_new_string(signal_edges[0]);
 442                 break;
 443         default:
 444                 return SR_ERR_NA;
 445         }
 446
 447         return SR_OK;
 448 }
 449
 450 static int config_set(uint32_t key, GVariant *data,
 451         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
 452 {
 453         struct dev_context *devc;
 454         int idx;
 455         gdouble low, high;
 456
 457         (void)cg;
 458
 459         if (!sdi)
 460                 return SR_ERR_ARG;
 461
 462         devc = sdi->priv;
 463
 464         switch (key) {
 465         case SR_CONF_SAMPLERATE:
 466                 if ((idx = std_u64_idx(data, devc->samplerates, devc->num_samplerates)) < 0)
 467                         return SR_ERR_ARG;
 468                 devc->cur_samplerate = devc->samplerates[idx];
 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                 if (!strcmp(devc->profile->model, "DSLogic")) {
 478                         if ((idx = std_double_tuple_idx(data, ARRAY_AND_SIZE(thresholds))) < 0)
 479                                 return SR_ERR_ARG;
 480                         devc->cur_threshold = thresholds[idx][0];
 481                         return dslogic_fpga_firmware_upload(sdi);
 482                 } else {
 483                         g_variant_get(data, "(dd)", &low, &high);
 484                         return dslogic_set_voltage_threshold(sdi, (low + high) / 2.0);
 485                 }
 486                 break;
 487         case SR_CONF_EXTERNAL_CLOCK:
 488                 devc->external_clock = g_variant_get_boolean(data);
 489                 break;
 490         case SR_CONF_CONTINUOUS:
 491                 devc->continuous_mode = g_variant_get_boolean(data);
 492                 break;
 493         case SR_CONF_CLOCK_EDGE:
 494                 if ((idx = std_str_idx(data, ARRAY_AND_SIZE(signal_edges))) < 0)
 495                         return SR_ERR_ARG;
 496                 devc->clock_edge = idx;
 497                 break;
 498         default:
 499                 return SR_ERR_NA;
 500         }
 501
 502         return SR_OK;
 503 }
 504
 505 static int config_list(uint32_t key, GVariant **data,
 506         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
 507 {
 508         struct dev_context *devc;
 509
 510         devc = (sdi) ? sdi->priv : NULL;
 511
 512         switch (key) {
 513         case SR_CONF_SCAN_OPTIONS:
 514         case SR_CONF_DEVICE_OPTIONS:
 515                 return STD_CONFIG_LIST(key, data, sdi, cg, scanopts, drvopts, devopts);
 516         case SR_CONF_VOLTAGE_THRESHOLD:
 517                 if (!strcmp(devc->profile->model, "DSLogic"))
 518                         *data = std_gvar_thresholds(ARRAY_AND_SIZE(thresholds));
 519                 else
 520                         *data = std_gvar_min_max_step_thresholds(0.0, 5.0, 0.1);
 521                 break;
 522         case SR_CONF_SAMPLERATE:
 523                 *data = std_gvar_samplerates(devc->samplerates, devc->num_samplerates);
 524                 break;
 525         case SR_CONF_TRIGGER_MATCH:
 526                 *data = std_gvar_array_i32(ARRAY_AND_SIZE(trigger_matches));
 527                 break;
 528         case SR_CONF_CLOCK_EDGE:
 529                 *data = g_variant_new_strv(ARRAY_AND_SIZE(signal_edges));
 530                 break;
 531         default:
 532                 return SR_ERR_NA;
 533         }
 534
 535         return SR_OK;
 536 }
 537
 538 static struct sr_dev_driver dreamsourcelab_dslogic_driver_info = {
 539         .name = "dreamsourcelab-dslogic",
 540         .longname = "DreamSourceLab DSLogic",
 541         .api_version = 1,
 542         .init = std_init,
 543         .cleanup = std_cleanup,
 544         .scan = scan,
 545         .dev_list = std_dev_list,
 546         .dev_clear = std_dev_clear,
 547         .config_get = config_get,
 548         .config_set = config_set,
 549         .config_list = config_list,
 550         .dev_open = dev_open,
 551         .dev_close = dev_close,
 552         .dev_acquisition_start = dslogic_acquisition_start,
 553         .dev_acquisition_stop = dslogic_acquisition_stop,
 554         .context = NULL,
 555 };
 556 SR_REGISTER_DEV_DRIVER(dreamsourcelab_dslogic_driver_info);