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                 libusb_close(hdl);
 218
 219                 if (usb_get_port_path(devlist[i], connection_id, sizeof(connection_id)) < 0)
 220                         continue;
 221
 222                 prof = NULL;
 223                 for (j = 0; supported_device[j].vid; j++) {
 224                         if (des.idVendor == supported_device[j].vid &&
 225                                         des.idProduct == supported_device[j].pid) {
 226                                 prof = &supported_device[j];
 227                                 break;
 228                         }
 229                 }
 230
 231                 if (!prof)
 232                         continue;
 233
 234                 sdi = g_malloc0(sizeof(struct sr_dev_inst));
 235                 sdi->status = SR_ST_INITIALIZING;
 236                 sdi->vendor = g_strdup(prof->vendor);
 237                 sdi->model = g_strdup(prof->model);
 238                 sdi->version = g_strdup(prof->model_version);
 239                 sdi->serial_num = g_strdup(serial_num);
 240                 sdi->connection_id = g_strdup(connection_id);
 241
 242                 /* Logic channels, all in one channel group. */
 243                 cg = g_malloc0(sizeof(struct sr_channel_group));
 244                 cg->name = g_strdup("Logic");
 245                 for (j = 0; j < NUM_CHANNELS; j++) {
 246                         sprintf(channel_name, "%d", j);
 247                         ch = sr_channel_new(sdi, j, SR_CHANNEL_LOGIC,
 248                                                 TRUE, channel_name);
 249                         cg->channels = g_slist_append(cg->channels, ch);
 250                 }
 251                 sdi->channel_groups = g_slist_append(NULL, cg);
 252
 253                 devc = dslogic_dev_new();
 254                 devc->profile = prof;
 255                 sdi->priv = devc;
 256                 devices = g_slist_append(devices, sdi);
 257
 258                 devc->samplerates = samplerates;
 259                 devc->num_samplerates = ARRAY_SIZE(samplerates);
 260                 has_firmware = usb_match_manuf_prod(devlist[i], "DreamSourceLab", "USB-based Instrument");
 261
 262                 if (has_firmware) {
 263                         /* Already has the firmware, so fix the new address. */
 264                         sr_dbg("Found a DSLogic device.");
 265                         sdi->status = SR_ST_INACTIVE;
 266                         sdi->inst_type = SR_INST_USB;
 267                         sdi->conn = sr_usb_dev_inst_new(libusb_get_bus_number(devlist[i]),
 268                                         libusb_get_device_address(devlist[i]), NULL);
 269                 } else {
 270                         if (ezusb_upload_firmware(drvc->sr_ctx, devlist[i],
 271                                         USB_CONFIGURATION, prof->firmware) == SR_OK) {
 272                                 /* Store when this device's FW was updated. */
 273                                 devc->fw_updated = g_get_monotonic_time();
 274                         } else {
 275                                 sr_err("Firmware upload failed for "
 276                                        "device %d.%d (logical), name %s.",
 277                                        libusb_get_bus_number(devlist[i]),
 278                                        libusb_get_device_address(devlist[i]),
 279                                        prof->firmware);
 280                         }
 281                         sdi->inst_type = SR_INST_USB;
 282                         sdi->conn = sr_usb_dev_inst_new(libusb_get_bus_number(devlist[i]),
 283                                         0xff, NULL);
 284                 }
 285         }
 286         libusb_free_device_list(devlist, 1);
 287         g_slist_free_full(conn_devices, (GDestroyNotify)sr_usb_dev_inst_free);
 288
 289         return std_scan_complete(di, devices);
 290 }
 291
 292 static int dev_open(struct sr_dev_inst *sdi)
 293 {
 294         struct sr_dev_driver *di = sdi->driver;
 295         struct sr_usb_dev_inst *usb;
 296         struct dev_context *devc;
 297         int ret;
 298         int64_t timediff_us, timediff_ms;
 299
 300         devc = sdi->priv;
 301         usb = sdi->conn;
 302
 303         /*
 304          * If the firmware was recently uploaded, wait up to MAX_RENUM_DELAY_MS
 305          * milliseconds for the FX2 to renumerate.
 306          */
 307         ret = SR_ERR;
 308         if (devc->fw_updated > 0) {
 309                 sr_info("Waiting for device to reset.");
 310                 /* Takes >= 300ms for the FX2 to be gone from the USB bus. */
 311                 g_usleep(300 * 1000);
 312                 timediff_ms = 0;
 313                 while (timediff_ms < MAX_RENUM_DELAY_MS) {
 314                         if ((ret = dslogic_dev_open(sdi, di)) == SR_OK)
 315                                 break;
 316                         g_usleep(100 * 1000);
 317
 318                         timediff_us = g_get_monotonic_time() - devc->fw_updated;
 319                         timediff_ms = timediff_us / 1000;
 320                         sr_spew("Waited %" PRIi64 "ms.", timediff_ms);
 321                 }
 322                 if (ret != SR_OK) {
 323                         sr_err("Device failed to renumerate.");
 324                         return SR_ERR;
 325                 }
 326                 sr_info("Device came back after %" PRIi64 "ms.", timediff_ms);
 327         } else {
 328                 sr_info("Firmware upload was not needed.");
 329                 ret = dslogic_dev_open(sdi, di);
 330         }
 331
 332         if (ret != SR_OK) {
 333                 sr_err("Unable to open device.");
 334                 return SR_ERR;
 335         }
 336
 337         ret = libusb_claim_interface(usb->devhdl, USB_INTERFACE);
 338         if (ret != 0) {
 339                 switch (ret) {
 340                 case LIBUSB_ERROR_BUSY:
 341                         sr_err("Unable to claim USB interface. Another "
 342                                "program or driver has already claimed it.");
 343                         break;
 344                 case LIBUSB_ERROR_NO_DEVICE:
 345                         sr_err("Device has been disconnected.");
 346                         break;
 347                 default:
 348                         sr_err("Unable to claim interface: %s.",
 349                                libusb_error_name(ret));
 350                         break;
 351                 }
 352
 353                 return SR_ERR;
 354         }
 355
 356
 357         if ((ret = dslogic_fpga_firmware_upload(sdi)) != SR_OK)
 358                 return ret;
 359
 360         if (devc->cur_samplerate == 0) {
 361                 /* Samplerate hasn't been set; default to the slowest one. */
 362                 devc->cur_samplerate = devc->samplerates[0];
 363         }
 364
 365         if (devc->cur_threshold == 0.0) {
 366                 devc->cur_threshold = thresholds[1][0];
 367                 return dslogic_set_voltage_threshold(sdi, devc->cur_threshold);
 368         }
 369
 370         return SR_OK;
 371 }
 372
 373 static int dev_close(struct sr_dev_inst *sdi)
 374 {
 375         struct sr_usb_dev_inst *usb;
 376
 377         usb = sdi->conn;
 378
 379         if (!usb->devhdl)
 380                 return SR_ERR_BUG;
 381
 382         sr_info("Closing device on %d.%d (logical) / %s (physical) interface %d.",
 383                 usb->bus, usb->address, sdi->connection_id, USB_INTERFACE);
 384         libusb_release_interface(usb->devhdl, USB_INTERFACE);
 385         libusb_close(usb->devhdl);
 386         usb->devhdl = NULL;
 387
 388         return SR_OK;
 389 }
 390
 391 static int config_get(uint32_t key, GVariant **data,
 392         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
 393 {
 394         struct dev_context *devc;
 395         struct sr_usb_dev_inst *usb;
 396         int idx;
 397
 398         (void)cg;
 399
 400         if (!sdi)
 401                 return SR_ERR_ARG;
 402
 403         devc = sdi->priv;
 404
 405         switch (key) {
 406         case SR_CONF_CONN:
 407                 if (!sdi->conn)
 408                         return SR_ERR_ARG;
 409                 usb = sdi->conn;
 410                 if (usb->address == 255)
 411                         /* Device still needs to re-enumerate after firmware
 412                          * upload, so we don't know its (future) address. */
 413                         return SR_ERR;
 414                 *data = g_variant_new_printf("%d.%d", usb->bus, usb->address);
 415                 break;
 416         case SR_CONF_VOLTAGE_THRESHOLD:
 417                 if (!strcmp(devc->profile->model, "DSLogic")) {
 418                         if ((idx = std_double_tuple_idx_d0(devc->cur_threshold,
 419                                         ARRAY_AND_SIZE(thresholds))) < 0)
 420                                 return SR_ERR_BUG;
 421                         *data = std_gvar_tuple_double(thresholds[idx][0],
 422                                         thresholds[idx][1]);
 423                 } else {
 424                         *data = std_gvar_tuple_double(devc->cur_threshold, devc->cur_threshold);
 425                 }
 426                 break;
 427         case SR_CONF_LIMIT_SAMPLES:
 428                 *data = g_variant_new_uint64(devc->limit_samples);
 429                 break;
 430         case SR_CONF_SAMPLERATE:
 431                 *data = g_variant_new_uint64(devc->cur_samplerate);
 432                 break;
 433         case SR_CONF_CAPTURE_RATIO:
 434                 *data = g_variant_new_uint64(devc->capture_ratio);
 435                 break;
 436         case SR_CONF_EXTERNAL_CLOCK:
 437                 *data = g_variant_new_boolean(devc->external_clock);
 438                 break;
 439         case SR_CONF_CONTINUOUS:
 440                 *data = g_variant_new_boolean(devc->continuous_mode);
 441                 break;
 442         case SR_CONF_CLOCK_EDGE:
 443                 idx = devc->clock_edge;
 444                 if (idx >= (int)ARRAY_SIZE(signal_edges))
 445                         return SR_ERR_BUG;
 446                 *data = g_variant_new_string(signal_edges[0]);
 447                 break;
 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         int idx;
 460         gdouble low, high;
 461
 462         (void)cg;
 463
 464         if (!sdi)
 465                 return SR_ERR_ARG;
 466
 467         devc = sdi->priv;
 468
 469         switch (key) {
 470         case SR_CONF_SAMPLERATE:
 471                 if ((idx = std_u64_idx(data, devc->samplerates, devc->num_samplerates)) < 0)
 472                         return SR_ERR_ARG;
 473                 devc->cur_samplerate = devc->samplerates[idx];
 474                 break;
 475         case SR_CONF_LIMIT_SAMPLES:
 476                 devc->limit_samples = g_variant_get_uint64(data);
 477                 break;
 478         case SR_CONF_CAPTURE_RATIO:
 479                 devc->capture_ratio = g_variant_get_uint64(data);
 480                 break;
 481         case SR_CONF_VOLTAGE_THRESHOLD:
 482                 if (!strcmp(devc->profile->model, "DSLogic")) {
 483                         if ((idx = std_double_tuple_idx(data, ARRAY_AND_SIZE(thresholds))) < 0)
 484                                 return SR_ERR_ARG;
 485                         devc->cur_threshold = thresholds[idx][0];
 486                         return dslogic_fpga_firmware_upload(sdi);
 487                 } else {
 488                         g_variant_get(data, "(dd)", &low, &high);
 489                         return dslogic_set_voltage_threshold(sdi, (low + high) / 2.0);
 490                 }
 491                 break;
 492         case SR_CONF_EXTERNAL_CLOCK:
 493                 devc->external_clock = g_variant_get_boolean(data);
 494                 break;
 495         case SR_CONF_CONTINUOUS:
 496                 devc->continuous_mode = g_variant_get_boolean(data);
 497                 break;
 498         case SR_CONF_CLOCK_EDGE:
 499                 if ((idx = std_str_idx(data, ARRAY_AND_SIZE(signal_edges))) < 0)
 500                         return SR_ERR_ARG;
 501                 devc->clock_edge = idx;
 502                 break;
 503         default:
 504                 return SR_ERR_NA;
 505         }
 506
 507         return SR_OK;
 508 }
 509
 510 static int config_list(uint32_t key, GVariant **data,
 511         const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
 512 {
 513         struct dev_context *devc;
 514
 515         devc = (sdi) ? sdi->priv : NULL;
 516
 517         switch (key) {
 518         case SR_CONF_SCAN_OPTIONS:
 519         case SR_CONF_DEVICE_OPTIONS:
 520                 return STD_CONFIG_LIST(key, data, sdi, cg, scanopts, drvopts, devopts);
 521         case SR_CONF_VOLTAGE_THRESHOLD:
 522                 if (!devc || !devc->profile)
 523                         return SR_ERR_ARG;
 524                 if (!strcmp(devc->profile->model, "DSLogic"))
 525                         *data = std_gvar_thresholds(ARRAY_AND_SIZE(thresholds));
 526                 else
 527                         *data = std_gvar_min_max_step_thresholds(0.0, 5.0, 0.1);
 528                 break;
 529         case SR_CONF_SAMPLERATE:
 530                 if (!devc)
 531                         return SR_ERR_ARG;
 532                 *data = std_gvar_samplerates(devc->samplerates, devc->num_samplerates);
 533                 break;
 534         case SR_CONF_TRIGGER_MATCH:
 535                 *data = std_gvar_array_i32(ARRAY_AND_SIZE(trigger_matches));
 536                 break;
 537         case SR_CONF_CLOCK_EDGE:
 538                 *data = g_variant_new_strv(ARRAY_AND_SIZE(signal_edges));
 539                 break;
 540         default:
 541                 return SR_ERR_NA;
 542         }
 543
 544         return SR_OK;
 545 }
 546
 547 static struct sr_dev_driver dreamsourcelab_dslogic_driver_info = {
 548         .name = "dreamsourcelab-dslogic",
 549         .longname = "DreamSourceLab DSLogic",
 550         .api_version = 1,
 551         .init = std_init,
 552         .cleanup = std_cleanup,
 553         .scan = scan,
 554         .dev_list = std_dev_list,
 555         .dev_clear = std_dev_clear,
 556         .config_get = config_get,
 557         .config_set = config_set,
 558         .config_list = config_list,
 559         .dev_open = dev_open,
 560         .dev_close = dev_close,
 561         .dev_acquisition_start = dslogic_acquisition_start,
 562         .dev_acquisition_stop = dslogic_acquisition_stop,
 563         .context = NULL,
 564 };
 565 SR_REGISTER_DEV_DRIVER(dreamsourcelab_dslogic_driver_info);