+ libusb_free_device_list(devlist, 1);
+
+ if (sdi->status != SR_ST_ACTIVE)
+ return SR_ERR;
+
+ return SR_OK;
+}
+
+SR_PRIV void hantek_6xxx_close(struct sr_dev_inst *sdi)
+{
+ struct sr_usb_dev_inst *usb;
+
+ usb = sdi->conn;
+
+ if (!usb->devhdl)
+ return;
+
+ sr_info("Closing device on %d.%d (logical) / %s (physical) interface %d.",
+ usb->bus, usb->address, sdi->connection_id, USB_INTERFACE);
+ libusb_release_interface(usb->devhdl, USB_INTERFACE);
+ libusb_close(usb->devhdl);
+ usb->devhdl = NULL;
+ sdi->status = SR_ST_INACTIVE;
+}
+
+SR_PRIV int hantek_6xxx_get_channeldata(const struct sr_dev_inst *sdi,
+ libusb_transfer_cb_fn cb, uint32_t data_amount)
+{
+ struct sr_usb_dev_inst *usb;
+ struct libusb_transfer *transfer;
+ int ret;
+ unsigned char *buf;
+
+ sr_dbg("Request channel data.");
+
+ usb = sdi->conn;
+
+ if (!(buf = g_try_malloc(data_amount))) {
+ sr_err("Failed to malloc USB endpoint buffer.");
+ return SR_ERR_MALLOC;
+ }
+ transfer = libusb_alloc_transfer(0);
+ libusb_fill_bulk_transfer(transfer, usb->devhdl, HANTEK_EP_IN, buf,
+ data_amount, cb, (void *)sdi, 4000);
+ if ((ret = libusb_submit_transfer(transfer)) != 0) {
+ sr_err("Failed to submit transfer: %s.",
+ libusb_error_name(ret));
+ /* TODO: Free them all. */
+ libusb_free_transfer(transfer);
+ g_free(buf);
+ return SR_ERR;
+ }
+
+ return SR_OK;
+}
+
+static uint8_t samplerate_to_reg(uint64_t samplerate)
+{
+ const uint64_t samplerate_values[] = {SAMPLERATE_VALUES};
+ const uint8_t samplerate_regs[] = {SAMPLERATE_REGS};
+ uint32_t i;
+
+ for (i = 0; i < ARRAY_SIZE(samplerate_values); i++) {
+ if (samplerate_values[i] == samplerate)
+ return samplerate_regs[i];
+ }
+
+ sr_err("Failed to convert samplerate: %" PRIu64 ".", samplerate);
+
+ return samplerate_regs[ARRAY_SIZE(samplerate_values) - 1];
+}
+
+static uint8_t voltage_to_reg(uint8_t state)
+{
+ const uint8_t vdiv_reg[] = {VDIV_REG};
+
+ if (state < ARRAY_SIZE(vdiv_reg)) {
+ return vdiv_reg[state];
+ } else {
+ sr_err("Failed to convert vdiv: %d.", state);
+ return vdiv_reg[ARRAY_SIZE(vdiv_reg) - 1];
+ }
+}
+
+static int write_control(const struct sr_dev_inst *sdi,
+ enum control_requests reg, uint8_t value)
+{
+ struct sr_usb_dev_inst *usb = sdi->conn;
+ int ret;
+
+ sr_spew("hantek_6xxx_write_control: 0x%x 0x%x", reg, value);
+
+ if ((ret = libusb_control_transfer(usb->devhdl,
+ LIBUSB_REQUEST_TYPE_VENDOR, (uint8_t)reg,
+ 0, 0, &value, 1, 100)) <= 0) {
+ sr_err("Failed to control transfer: 0x%x: %s.", reg,
+ libusb_error_name(ret));
+ return ret;
+ }
+
+ return 0;
+}
+
+SR_PRIV int hantek_6xxx_start_data_collecting(const struct sr_dev_inst *sdi)
+{
+ sr_dbg("trigger");
+ return write_control(sdi, TRIGGER_REG, 1);
+}
+
+SR_PRIV int hantek_6xxx_stop_data_collecting(const struct sr_dev_inst *sdi)
+{
+ return write_control(sdi, TRIGGER_REG, 0);
+}
+
+SR_PRIV int hantek_6xxx_update_samplerate(const struct sr_dev_inst *sdi)
+{
+ struct dev_context *devc = sdi->priv;
+ sr_dbg("update samplerate %d", samplerate_to_reg(devc->samplerate));
+
+ return write_control(sdi, SAMPLERATE_REG, samplerate_to_reg(devc->samplerate));
+}
+
+SR_PRIV int hantek_6xxx_update_vdiv(const struct sr_dev_inst *sdi)
+{
+ struct dev_context *devc = sdi->priv;
+ int ret1, ret2;
+
+ sr_dbg("update vdiv %d %d", voltage_to_reg(devc->voltage[0]),
+ voltage_to_reg(devc->voltage[1]));
+
+ ret1 = write_control(sdi, VDIV_CH1_REG, voltage_to_reg(devc->voltage[0]));
+ ret2 = write_control(sdi, VDIV_CH2_REG, voltage_to_reg(devc->voltage[1]));
+
+ return MIN(ret1, ret2);
+}
+
+SR_PRIV int hantek_6xxx_update_channels(const struct sr_dev_inst *sdi)
+{
+ struct dev_context *devc = sdi->priv;
+ uint8_t chan = devc->ch_enabled[1] ? 2 : 1;
+ sr_dbg("update channels amount %d", chan);
+
+ return write_control(sdi, CHANNELS_REG, chan);
+}
+
+SR_PRIV int hantek_6xxx_init(const struct sr_dev_inst *sdi)
+{
+ sr_dbg("Initializing");
+
+ hantek_6xxx_update_samplerate(sdi);
+ hantek_6xxx_update_vdiv(sdi);
+ // hantek_6xxx_update_channels(sdi); /* Only 2 channel mode supported. */
+
+ return SR_OK;