[libsigrok.git] / src / hardware / dcttech-usbrelay / api.c

/*
 * This file is part of the libsigrok project.
 *
 * Copyright (C) 2021 Gerhard Sittig <gerhard.sittig@gmx.net>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <config.h>

#include <ctype.h>
#include <hidapi.h>
#include <string.h>

#include "protocol.h"

static const uint32_t scanopts[] = {
	SR_CONF_CONN,
};

static const uint32_t drvopts[] = {
	SR_CONF_MULTIPLEXER,
};

static const uint32_t devopts[] = {
	SR_CONF_CONN | SR_CONF_GET,
	SR_CONF_ENABLED | SR_CONF_SET, /* Enable/disable all relays at once. */
};

static const uint32_t devopts_cg[] = {
	SR_CONF_ENABLED | SR_CONF_GET | SR_CONF_SET,
};

static struct sr_dev_driver dcttech_usbrelay_driver_info;

static struct sr_dev_inst *probe_device_common(const char *path,
	uint16_t vid, uint16_t pid,
	const wchar_t *vendor, const wchar_t *product)
{
	char nonws[16], *s, *endp;
	unsigned long relay_count;
	hid_device *hid;
	int ret;
	char serno[SERNO_LENGTH + 1];
	uint8_t curr_state;
	uint8_t report[1 + REPORT_BYTECOUNT];
	GString *txt;
	size_t snr_pos;
	char c;
	struct sr_dev_inst *sdi;
	struct dev_context *devc;
	struct channel_group_context *cgc;
	size_t idx, nr;
	struct sr_channel_group *cg;

	/*
	 * Get relay count from product string. Weak condition,
	 * accept any trailing number regardless of preceeding text.
	 */
	snprintf(nonws, sizeof(nonws), "%ls", product);
	s = nonws;
	s += strlen(s);
	while (s > nonws && isdigit((int)s[-1]))
		s--;
	ret = sr_atoul_base(s, &relay_count, &endp, 10);
	if (ret != SR_OK || !endp || *endp)
		return NULL;
	if (!relay_count)
		return NULL;
	sr_info("Relay count %lu from product string %s.", relay_count, nonws);

	/* Open device, need to communicate to identify. */
	if (vid && pid)
		hid = hid_open(vid, pid, NULL);
	else
		hid = hid_open_path(path);
	if (!hid) {
		sr_err("Cannot open %s.", path);
		return NULL;
	}

	/* Get an HID report. */
	hid_set_nonblocking(hid, 0);
	memset(&report, 0, sizeof(report));
	report[0] = REPORT_NUMBER;
	ret = hid_get_feature_report(hid, report, sizeof(report));
	hid_close(hid);
	if (sr_log_loglevel_get() >= SR_LOG_SPEW) {
		txt = sr_hexdump_new(report, sizeof(report));
		sr_spew("Got report bytes: %s, rc %d.", txt->str, ret);
		sr_hexdump_free(txt);
	}
	if (ret < 0) {
		sr_err("Cannot read %s: %ls.", path, hid_error(NULL));
		return NULL;
	}
	if (ret != sizeof(report)) {
		sr_err("Unexpected HID report length %d from %s.", ret, path);
		return NULL;
	}

	/*
	 * Serial number must be all printable characters. Relay state
	 * is for information only, gets re-retrieved before configure
	 * API calls (get/set).
	 */
	memset(serno, 0, sizeof(serno));
	for (snr_pos = 0; snr_pos < SERNO_LENGTH; snr_pos++) {
		c = report[1 + snr_pos];
		serno[snr_pos] = c;
		if (c < 0x20 || c > 0x7e) {
			sr_warn("Skipping %s, non-printable serial.", path);
			return NULL;
		}
	}
	curr_state = report[1 + STATE_INDEX];
	sr_info("HID report data: serial number %s, relay state 0x%02x.",
		serno, curr_state);

	/* Create a device instance. */
	sdi = g_malloc0(sizeof(*sdi));
	sdi->vendor = g_strdup_printf("%ls", vendor);
	sdi->model = g_strdup_printf("%ls", product);
	sdi->serial_num = g_strdup(serno);
	sdi->connection_id = g_strdup(path);
	sdi->driver = &dcttech_usbrelay_driver_info;
	sdi->inst_type = SR_INST_USB;

	/* Create channels (groups). */
	devc = g_malloc0(sizeof(*devc));
	sdi->priv = devc;
	devc->hid_path = g_strdup(path);
	devc->usb_vid = vid;
	devc->usb_pid = pid;
	devc->relay_count = relay_count;
	devc->relay_mask = (1U << relay_count) - 1;
	for (idx = 0; idx < devc->relay_count; idx++) {
		nr = idx + 1;
		cg = g_malloc0(sizeof(*cg));
		cg->name = g_strdup_printf("R%zu", nr);
		cgc = g_malloc0(sizeof(*cgc));
		cg->priv = cgc;
		cgc->number = nr;
		sdi->channel_groups = g_slist_append(sdi->channel_groups, cg);
	}

	return sdi;
}

static struct sr_dev_inst *probe_device_enum(struct hid_device_info *dev)
{
	return probe_device_common(dev->path, 0, 0,
		dev->manufacturer_string, dev->product_string);
}

static struct sr_dev_inst *probe_device_conn(const char *path)
{
	char vid_pid[12];
	uint16_t vid, pid;
	const char *s;
	char *endp;
	unsigned long num;
	hid_device *dev;
	gboolean ok;
	int ret;
	wchar_t vendor[32], product[32];

	/*
	 * The hidapi(3) library's API strives for maximum portability,
	 * thus won't provide ways of getting a path from alternative
	 * presentations like VID:PID pairs, bus.addr specs, etc. The
	 * typical V-USB setup neither provides reliable serial numbers
	 * (that USB enumeration would cover). So this driver's support
	 * for conn= specs beyond Unix style path names is limited, too.
	 * This implementation tries "VID.PID" then assumes "path". The
	 * inability to even get the path for a successfully opened HID
	 * results in redundancy across the places which open devices.
	 */

	/* Check for "<vid>.<pid>" specs. */
	vid = pid = 0;
	s = path;
	ret = sr_atoul_base(s, &num, &endp, 16);
	if (ret == SR_OK && endp && endp == s + 4 && *endp == '.' && num) {
		vid = num;
		s = ++endp;
	}
	ret = sr_atoul_base(s, &num, &endp, 16);
	if (ret == SR_OK && endp && endp == s + 4 && *endp == '\0' && num) {
		pid = num;
		s = ++endp;
	}
	if (vid && pid) {
		snprintf(vid_pid, sizeof(vid_pid), "%04x.%04x", vid, pid);
		path = vid_pid;
		sr_dbg("Using VID.PID %s.", path);
	}

	/* Open the device, get vendor and product strings. */
	if (vid && pid)
		dev = hid_open(vid, pid, NULL);
	else
		dev = hid_open_path(path);
	if (!dev) {
		sr_err("Cannot open %s.", path);
		return NULL;
	}
	ok = TRUE;
	ret = hid_get_manufacturer_string(dev, vendor, ARRAY_SIZE(vendor));
	if (ret != 0)
		ok = FALSE;
	if (!wcslen(vendor))
		ok = FALSE;
	ret = hid_get_product_string(dev, product, ARRAY_SIZE(product));
	if (ret != 0)
		ok = FALSE;
	if (!wcslen(product))
		ok = FALSE;
	hid_close(dev);
	if (!ok)
		return NULL;

	return probe_device_common(path, vid, pid, vendor, product);
}

static GSList *scan(struct sr_dev_driver *di, GSList *options)
{
	const char *conn;
	GSList *devices;
	struct drv_context *drvc;
	struct hid_device_info *devs, *curdev;
	wchar_t *ws;
	char nonws[32];
	struct sr_dev_inst *sdi;

	/* Get optional conn= spec when provided. */
	conn = NULL;
	(void)sr_serial_extract_options(options, &conn, NULL);
	if (conn && !*conn)
		conn = NULL;

	devices = NULL;
	drvc = di->context;
	drvc->instances = NULL;

	/*
	 * The firmware is V-USB based. The USB VID:PID identification
	 * is shared across several projects. Need to inspect the vendor
	 * and product _strings_ to actually identify the device. The
	 * USB serial number need not be present nor reliable. The HID
	 * report content will carry the board's serial number.
	 *
	 * When conn= was specified, then have HIDAPI open _this_ device
	 * and skip the enumeration. Which allows users to specify paths
	 * that need not match the enumeration's details.
	 */
	if (conn) {
		sr_info("Checking HID path %s.", conn);
		sdi = probe_device_conn(conn);
		if (!sdi)
			sr_warn("Failed to communicate to %s.", conn);
		else
			devices = g_slist_append(devices, sdi);
	}
	devs = hid_enumerate(VENDOR_ID, PRODUCT_ID);
	for (curdev = devs; curdev; curdev = curdev->next) {
		if (conn)
			break;
		if (!curdev->vendor_id || !curdev->product_id)
			continue;
		if (!curdev->manufacturer_string || !curdev->product_string)
			continue;
		if (!*curdev->manufacturer_string || !*curdev->product_string)
			continue;
		sr_dbg("Checking %04hx:%04hx, vendor %ls, product %ls.",
			curdev->vendor_id, curdev->product_id,
			curdev->manufacturer_string, curdev->product_string);

		/* Check USB details retrieved by enumeration. */
		ws = curdev->manufacturer_string;
		if (!ws || !wcslen(ws))
			continue;
		snprintf(nonws, sizeof(nonws), "%ls", ws);
		if (strcmp(nonws, VENDOR_STRING) != 0)
			continue;
		ws = curdev->product_string;
		if (!ws || !wcslen(ws))
			continue;
		snprintf(nonws, sizeof(nonws), "%ls", ws);
		if (!g_str_has_prefix(nonws, PRODUCT_STRING_PREFIX))
			continue;

		/* Identify device by communicating to it. */
		sr_info("Checking HID path %s.", curdev->path);
		sdi = probe_device_enum(curdev);
		if (!sdi) {
			sr_warn("Failed to communicate to %s.", curdev->path);
			continue;
		}
		devices = g_slist_append(devices, sdi);
	}
	hid_free_enumeration(devs);

	return devices;
}

static int dev_open(struct sr_dev_inst *sdi)
{
	struct dev_context *devc;

	devc = sdi->priv;

	if (devc->hid_dev) {
		hid_close(devc->hid_dev);
		devc->hid_dev = NULL;
	}

	if (devc->usb_vid && devc->usb_pid)
		devc->hid_dev = hid_open(devc->usb_vid, devc->usb_pid, NULL);
	else
		devc->hid_dev = hid_open_path(devc->hid_path);
	if (!devc->hid_dev)
		return SR_ERR_IO;

	(void)dcttech_usbrelay_update_state(sdi);

	return SR_OK;
}

static int dev_close(struct sr_dev_inst *sdi)
{
	struct dev_context *devc;

	devc = sdi->priv;

	if (devc->hid_dev) {
		hid_close(devc->hid_dev);
		devc->hid_dev = NULL;
	}

	return SR_OK;
}

static int config_get(uint32_t key, GVariant **data,
	const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
{
	gboolean on;
	int ret;

	if (!cg) {
		switch (key) {
		case SR_CONF_CONN:
			if (!sdi->connection_id)
				return SR_ERR_NA;
			*data = g_variant_new_string(sdi->connection_id);
			return SR_OK;
		default:
			return SR_ERR_NA;
		}
	}

	switch (key) {
	case SR_CONF_ENABLED:
		ret = dcttech_usbrelay_query_cg(sdi, cg, &on);
		if (ret != SR_OK)
			return ret;
		*data = g_variant_new_boolean(on);
		return SR_OK;
	default:
		return SR_ERR_NA;
	}
}

static int config_set(uint32_t key, GVariant *data,
	const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
{
	gboolean on;

	if (!cg) {
		switch (key) {
		case SR_CONF_ENABLED:
			/* Enable/disable all channels at the same time. */
			on = g_variant_get_boolean(data);
			return dcttech_usbrelay_switch_cg(sdi, cg, on);
		default:
			return SR_ERR_NA;
		}
	} else {
		switch (key) {
		case SR_CONF_ENABLED:
			on = g_variant_get_boolean(data);
			return dcttech_usbrelay_switch_cg(sdi, cg, on);
		default:
			return SR_ERR_NA;
		}
	}

	return SR_OK;
}

static int config_list(uint32_t key, GVariant **data,
	const struct sr_dev_inst *sdi, const struct sr_channel_group *cg)
{

	if (!cg) {
		switch (key) {
		case SR_CONF_SCAN_OPTIONS:
		case SR_CONF_DEVICE_OPTIONS:
			return STD_CONFIG_LIST(key, data, sdi, cg,
				scanopts, drvopts, devopts);
		default:
			return SR_ERR_NA;
		}
	}

	switch (key) {
	case SR_CONF_DEVICE_OPTIONS:
		*data = std_gvar_array_u32(ARRAY_AND_SIZE(devopts_cg));
		return SR_OK;
	default:
		return SR_ERR_NA;
	}
}

static struct sr_dev_driver dcttech_usbrelay_driver_info = {
	.name = "dcttech-usbrelay",
	.longname = "dcttech usbrelay",
	.api_version = 1,
	.init = std_init,
	.cleanup = std_cleanup,
	.scan = scan,
	.dev_list = std_dev_list,
	.dev_clear = std_dev_clear,
	.config_get = config_get,
	.config_set = config_set,
	.config_list = config_list,
	.dev_open = dev_open,
	.dev_close = dev_close,
	.dev_acquisition_start = std_dummy_dev_acquisition_start,
	.dev_acquisition_stop = std_dummy_dev_acquisition_stop,
	.context = NULL,
};
SR_REGISTER_DEV_DRIVER(dcttech_usbrelay_driver_info);
Commit	Line	Data
64d54a71 GS	1	/*
	2	* This file is part of the libsigrok project.
	3	*
	4	* Copyright (C) 2021 Gerhard Sittig <gerhard.sittig@gmx.net>
	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 <config.h>
321f85fb	21
0498ef4e	22	#include <ctype.h>
321f85fb GS	23	#include <hidapi.h>
	24	#include <string.h>
	25
64d54a71 GS	26	#include "protocol.h"
64d54a71 GS	27
321f85fb GS	28	static const uint32_t scanopts[] = {
	29	SR_CONF_CONN,
	30	};
	31
	32	static const uint32_t drvopts[] = {
	33	SR_CONF_MULTIPLEXER,
	34	};
	35
	36	static const uint32_t devopts[] = {
	37	SR_CONF_CONN \| SR_CONF_GET,
	38	SR_CONF_ENABLED \| SR_CONF_SET, /* Enable/disable all relays at once. */
	39	};
	40
	41	static const uint32_t devopts_cg[] = {
	42	SR_CONF_ENABLED \| SR_CONF_GET \| SR_CONF_SET,
	43	};
	44
64d54a71 GS	45	static struct sr_dev_driver dcttech_usbrelay_driver_info;
64d54a71 GS	46
0498ef4e	47	static struct sr_dev_inst probe_device_common(const char path,
8e5dbaf6	48	uint16_t vid, uint16_t pid,
0498ef4e	49	const wchar_t vendor, const wchar_t product)
321f85fb	50	{
0498ef4e GS	51	char nonws[16], s, endp;
0498ef4e GS	52	unsigned long relay_count;
321f85fb GS	53	hid_device *hid;
	54	int ret;
	55	char serno[SERNO_LENGTH + 1];
	56	uint8_t curr_state;
	57	uint8_t report[1 + REPORT_BYTECOUNT];
	58	GString *txt;
	59	size_t snr_pos;
	60	char c;
	61	struct sr_dev_inst *sdi;
	62	struct dev_context *devc;
	63	struct channel_group_context *cgc;
	64	size_t idx, nr;
	65	struct sr_channel_group *cg;
	66
0498ef4e GS	67	/*
	68	* Get relay count from product string. Weak condition,
	69	* accept any trailing number regardless of preceeding text.
	70	*/
	71	snprintf(nonws, sizeof(nonws), "%ls", product);
	72	s = nonws;
	73	s += strlen(s);
	74	while (s > nonws && isdigit((int)s[-1]))
	75	s--;
	76	ret = sr_atoul_base(s, &relay_count, &endp, 10);
	77	if (ret != SR_OK \|\| !endp \|\| *endp)
	78	return NULL;
	79	if (!relay_count)
	80	return NULL;
	81	sr_info("Relay count %lu from product string %s.", relay_count, nonws);
	82
321f85fb	83	/* Open device, need to communicate to identify. */
8e5dbaf6 GS	84	if (vid && pid)
	85	hid = hid_open(vid, pid, NULL);
	86	else
	87	hid = hid_open_path(path);
996331ce	88	if (!hid) {
0498ef4e	89	sr_err("Cannot open %s.", path);
321f85fb	90	return NULL;
996331ce	91	}
321f85fb GS	92
	93	/* Get an HID report. */
	94	hid_set_nonblocking(hid, 0);
	95	memset(&report, 0, sizeof(report));
	96	report[0] = REPORT_NUMBER;
	97	ret = hid_get_feature_report(hid, report, sizeof(report));
	98	hid_close(hid);
	99	if (sr_log_loglevel_get() >= SR_LOG_SPEW) {
	100	txt = sr_hexdump_new(report, sizeof(report));
0498ef4e	101	sr_spew("Got report bytes: %s, rc %d.", txt->str, ret);
321f85fb GS	102	sr_hexdump_free(txt);
321f85fb GS	103	}
996331ce	104	if (ret < 0) {
0498ef4e	105	sr_err("Cannot read %s: %ls.", path, hid_error(NULL));
996331ce GS	106	return NULL;
	107	}
	108	if (ret != sizeof(report)) {
0498ef4e	109	sr_err("Unexpected HID report length %d from %s.", ret, path);
321f85fb	110	return NULL;
996331ce	111	}
321f85fb GS	112
	113	/*
	114	* Serial number must be all printable characters. Relay state
	115	* is for information only, gets re-retrieved before configure
	116	* API calls (get/set).
	117	*/
	118	memset(serno, 0, sizeof(serno));
	119	for (snr_pos = 0; snr_pos < SERNO_LENGTH; snr_pos++) {
	120	c = report[1 + snr_pos];
	121	serno[snr_pos] = c;
996331ce	122	if (c < 0x20 \|\| c > 0x7e) {
0498ef4e	123	sr_warn("Skipping %s, non-printable serial.", path);
321f85fb	124	return NULL;
996331ce	125	}
321f85fb GS	126	}
321f85fb GS	127	curr_state = report[1 + STATE_INDEX];
0498ef4e GS	128	sr_info("HID report data: serial number %s, relay state 0x%02x.",
0498ef4e GS	129	serno, curr_state);
321f85fb	130
996331ce	131	/* Create a device instance. */
321f85fb	132	sdi = g_malloc0(sizeof(*sdi));
0498ef4e GS	133	sdi->vendor = g_strdup_printf("%ls", vendor);
0498ef4e GS	134	sdi->model = g_strdup_printf("%ls", product);
e333a40c	135	sdi->serial_num = g_strdup(serno);
0498ef4e	136	sdi->connection_id = g_strdup(path);
996331ce GS	137	sdi->driver = &dcttech_usbrelay_driver_info;
	138	sdi->inst_type = SR_INST_USB;
	139
	140	/* Create channels (groups). */
321f85fb GS	141	devc = g_malloc0(sizeof(*devc));
321f85fb GS	142	sdi->priv = devc;
0498ef4e	143	devc->hid_path = g_strdup(path);
8e5dbaf6 GS	144	devc->usb_vid = vid;
8e5dbaf6 GS	145	devc->usb_pid = pid;
321f85fb GS	146	devc->relay_count = relay_count;
	147	devc->relay_mask = (1U << relay_count) - 1;
	148	for (idx = 0; idx < devc->relay_count; idx++) {
	149	nr = idx + 1;
	150	cg = g_malloc0(sizeof(*cg));
	151	cg->name = g_strdup_printf("R%zu", nr);
	152	cgc = g_malloc0(sizeof(*cgc));
	153	cg->priv = cgc;
	154	cgc->number = nr;
	155	sdi->channel_groups = g_slist_append(sdi->channel_groups, cg);
	156	}
	157
	158	return sdi;
	159	}
	160
0498ef4e GS	161	static struct sr_dev_inst probe_device_enum(struct hid_device_info dev)
0498ef4e GS	162	{
8e5dbaf6	163	return probe_device_common(dev->path, 0, 0,
0498ef4e GS	164	dev->manufacturer_string, dev->product_string);
	165	}
	166
8e5dbaf6	167	static struct sr_dev_inst probe_device_conn(const char path)
0498ef4e	168	{
8e5dbaf6 GS	169	char vid_pid[12];
	170	uint16_t vid, pid;
	171	const char *s;
	172	char *endp;
	173	unsigned long num;
0498ef4e GS	174	hid_device *dev;
	175	gboolean ok;
	176	int ret;
	177	wchar_t vendor[32], product[32];
	178
	179	/*
8e5dbaf6 GS	180	* The hidapi(3) library's API strives for maximum portability,
	181	* thus won't provide ways of getting a path from alternative
	182	* presentations like VID:PID pairs, bus.addr specs, etc. The
	183	* typical V-USB setup neither provides reliable serial numbers
	184	* (that USB enumeration would cover). So this driver's support
	185	* for conn= specs beyond Unix style path names is limited, too.
	186	* This implementation tries "VID.PID" then assumes "path". The
	187	* inability to even get the path for a successfully opened HID
	188	* results in redundancy across the places which open devices.
0498ef4e GS	189	*/
0498ef4e GS	190
8e5dbaf6 GS	191	/* Check for "<vid>.<pid>" specs. */
	192	vid = pid = 0;
	193	s = path;
	194	ret = sr_atoul_base(s, &num, &endp, 16);
	195	if (ret == SR_OK && endp && endp == s + 4 && *endp == '.' && num) {
	196	vid = num;
	197	s = ++endp;
	198	}
	199	ret = sr_atoul_base(s, &num, &endp, 16);
	200	if (ret == SR_OK && endp && endp == s + 4 && *endp == '\0' && num) {
	201	pid = num;
	202	s = ++endp;
	203	}
	204	if (vid && pid) {
	205	snprintf(vid_pid, sizeof(vid_pid), "%04x.%04x", vid, pid);
	206	path = vid_pid;
	207	sr_dbg("Using VID.PID %s.", path);
	208	}
	209
	210	/* Open the device, get vendor and product strings. */
	211	if (vid && pid)
	212	dev = hid_open(vid, pid, NULL);
	213	else
	214	dev = hid_open_path(path);
0498ef4e GS	215	if (!dev) {
	216	sr_err("Cannot open %s.", path);
	217	return NULL;
	218	}
0498ef4e GS	219	ok = TRUE;
	220	ret = hid_get_manufacturer_string(dev, vendor, ARRAY_SIZE(vendor));
	221	if (ret != 0)
	222	ok = FALSE;
	223	if (!wcslen(vendor))
	224	ok = FALSE;
	225	ret = hid_get_product_string(dev, product, ARRAY_SIZE(product));
	226	if (ret != 0)
	227	ok = FALSE;
	228	if (!wcslen(product))
	229	ok = FALSE;
	230	hid_close(dev);
	231	if (!ok)
	232	return NULL;
	233
8e5dbaf6	234	return probe_device_common(path, vid, pid, vendor, product);
0498ef4e GS	235	}
0498ef4e GS	236
64d54a71 GS	237	static GSList scan(struct sr_dev_driver di, GSList *options)
64d54a71 GS	238	{
321f85fb	239	const char *conn;
64d54a71	240	GSList *devices;
321f85fb GS	241	struct drv_context *drvc;
321f85fb GS	242	struct hid_device_info devs, curdev;
321f85fb GS	243	wchar_t *ws;
321f85fb GS	244	char nonws[32];
321f85fb GS	245	struct sr_dev_inst *sdi;
	246
	247	/* Get optional conn= spec when provided. */
	248	conn = NULL;
	249	(void)sr_serial_extract_options(options, &conn, NULL);
	250	if (conn && !*conn)
	251	conn = NULL;
64d54a71 GS	252
	253	devices = NULL;
	254	drvc = di->context;
	255	drvc->instances = NULL;
	256
321f85fb GS	257	/*
	258	* The firmware is V-USB based. The USB VID:PID identification
	259	* is shared across several projects. Need to inspect the vendor
0498ef4e GS	260	* and product _strings_ to actually identify the device. The
	261	* USB serial number need not be present nor reliable. The HID
	262	* report content will carry the board's serial number.
321f85fb	263	*
0498ef4e GS	264	* When conn= was specified, then have HIDAPI open _this_ device
	265	* and skip the enumeration. Which allows users to specify paths
	266	* that need not match the enumeration's details.
321f85fb	267	*/
0498ef4e GS	268	if (conn) {
0498ef4e GS	269	sr_info("Checking HID path %s.", conn);
8e5dbaf6	270	sdi = probe_device_conn(conn);
0498ef4e GS	271	if (!sdi)
	272	sr_warn("Failed to communicate to %s.", conn);
	273	else
	274	devices = g_slist_append(devices, sdi);
	275	}
321f85fb GS	276	devs = hid_enumerate(VENDOR_ID, PRODUCT_ID);
321f85fb GS	277	for (curdev = devs; curdev; curdev = curdev->next) {
0498ef4e GS	278	if (conn)
0498ef4e GS	279	break;
321f85fb GS	280	if (!curdev->vendor_id \|\| !curdev->product_id)
	281	continue;
	282	if (!curdev->manufacturer_string \|\| !curdev->product_string)
	283	continue;
	284	if (!curdev->manufacturer_string \|\| !curdev->product_string)
	285	continue;
0498ef4e	286	sr_dbg("Checking %04hx:%04hx, vendor %ls, product %ls.",
321f85fb GS	287	curdev->vendor_id, curdev->product_id,
	288	curdev->manufacturer_string, curdev->product_string);
	289
	290	/* Check USB details retrieved by enumeration. */
	291	ws = curdev->manufacturer_string;
	292	if (!ws \|\| !wcslen(ws))
	293	continue;
	294	snprintf(nonws, sizeof(nonws), "%ls", ws);
	295	if (strcmp(nonws, VENDOR_STRING) != 0)
	296	continue;
	297	ws = curdev->product_string;
	298	if (!ws \|\| !wcslen(ws))
	299	continue;
	300	snprintf(nonws, sizeof(nonws), "%ls", ws);
0498ef4e	301	if (!g_str_has_prefix(nonws, PRODUCT_STRING_PREFIX))
321f85fb	302	continue;
321f85fb GS	303
321f85fb GS	304	/* Identify device by communicating to it. */
0498ef4e GS	305	sr_info("Checking HID path %s.", curdev->path);
0498ef4e GS	306	sdi = probe_device_enum(curdev);
321f85fb GS	307	if (!sdi) {
	308	sr_warn("Failed to communicate to %s.", curdev->path);
	309	continue;
	310	}
321f85fb GS	311	devices = g_slist_append(devices, sdi);
	312	}
	313	hid_free_enumeration(devs);
64d54a71 GS	314
	315	return devices;
	316	}
	317
	318	static int dev_open(struct sr_dev_inst *sdi)
	319	{
321f85fb GS	320	struct dev_context *devc;
	321
	322	devc = sdi->priv;
	323
	324	if (devc->hid_dev) {
	325	hid_close(devc->hid_dev);
	326	devc->hid_dev = NULL;
	327	}
64d54a71	328
8e5dbaf6 GS	329	if (devc->usb_vid && devc->usb_pid)
	330	devc->hid_dev = hid_open(devc->usb_vid, devc->usb_pid, NULL);
	331	else
	332	devc->hid_dev = hid_open_path(devc->hid_path);
321f85fb GS	333	if (!devc->hid_dev)
	334	return SR_ERR_IO;
	335
	336	(void)dcttech_usbrelay_update_state(sdi);
64d54a71 GS	337
	338	return SR_OK;
	339	}
	340
	341	static int dev_close(struct sr_dev_inst *sdi)
	342	{
321f85fb GS	343	struct dev_context *devc;
	344
	345	devc = sdi->priv;
64d54a71	346
321f85fb GS	347	if (devc->hid_dev) {
	348	hid_close(devc->hid_dev);
	349	devc->hid_dev = NULL;
	350	}
64d54a71 GS	351
	352	return SR_OK;
	353	}
	354
	355	static int config_get(uint32_t key, GVariant **data,
	356	const struct sr_dev_inst sdi, const struct sr_channel_group cg)
	357	{
321f85fb	358	gboolean on;
64d54a71 GS	359	int ret;
64d54a71 GS	360
321f85fb GS	361	if (!cg) {
	362	switch (key) {
	363	case SR_CONF_CONN:
e333a40c	364	if (!sdi->connection_id)
321f85fb	365	return SR_ERR_NA;
e333a40c	366	*data = g_variant_new_string(sdi->connection_id);
321f85fb GS	367	return SR_OK;
	368	default:
	369	return SR_ERR_NA;
	370	}
	371	}
64d54a71	372
64d54a71	373	switch (key) {
321f85fb GS	374	case SR_CONF_ENABLED:
	375	ret = dcttech_usbrelay_query_cg(sdi, cg, &on);
	376	if (ret != SR_OK)
	377	return ret;
	378	*data = g_variant_new_boolean(on);
	379	return SR_OK;
64d54a71 GS	380	default:
	381	return SR_ERR_NA;
	382	}
64d54a71 GS	383	}
	384
	385	static int config_set(uint32_t key, GVariant *data,
	386	const struct sr_dev_inst sdi, const struct sr_channel_group cg)
	387	{
321f85fb GS	388	gboolean on;
	389
	390	if (!cg) {
	391	switch (key) {
	392	case SR_CONF_ENABLED:
	393	/* Enable/disable all channels at the same time. */
	394	on = g_variant_get_boolean(data);
	395	return dcttech_usbrelay_switch_cg(sdi, cg, on);
	396	default:
	397	return SR_ERR_NA;
	398	}
	399	} else {
	400	switch (key) {
	401	case SR_CONF_ENABLED:
	402	on = g_variant_get_boolean(data);
	403	return dcttech_usbrelay_switch_cg(sdi, cg, on);
	404	default:
	405	return SR_ERR_NA;
	406	}
64d54a71 GS	407	}
64d54a71 GS	408
321f85fb	409	return SR_OK;
64d54a71 GS	410	}
	411
	412	static int config_list(uint32_t key, GVariant **data,
	413	const struct sr_dev_inst sdi, const struct sr_channel_group cg)
	414	{
64d54a71	415
321f85fb GS	416	if (!cg) {
	417	switch (key) {
	418	case SR_CONF_SCAN_OPTIONS:
	419	case SR_CONF_DEVICE_OPTIONS:
	420	return STD_CONFIG_LIST(key, data, sdi, cg,
	421	scanopts, drvopts, devopts);
	422	default:
	423	return SR_ERR_NA;
	424	}
	425	}
64d54a71	426
64d54a71	427	switch (key) {
321f85fb GS	428	case SR_CONF_DEVICE_OPTIONS:
	429	*data = std_gvar_array_u32(ARRAY_AND_SIZE(devopts_cg));
	430	return SR_OK;
64d54a71 GS	431	default:
	432	return SR_ERR_NA;
	433	}
64d54a71 GS	434	}
	435
	436	static struct sr_dev_driver dcttech_usbrelay_driver_info = {
	437	.name = "dcttech-usbrelay",
	438	.longname = "dcttech usbrelay",
	439	.api_version = 1,
	440	.init = std_init,
	441	.cleanup = std_cleanup,
	442	.scan = scan,
	443	.dev_list = std_dev_list,
	444	.dev_clear = std_dev_clear,
	445	.config_get = config_get,
	446	.config_set = config_set,
	447	.config_list = config_list,
	448	.dev_open = dev_open,
	449	.dev_close = dev_close,
321f85fb GS	450	.dev_acquisition_start = std_dummy_dev_acquisition_start,
321f85fb GS	451	.dev_acquisition_stop = std_dummy_dev_acquisition_stop,
64d54a71 GS	452	.context = NULL,
	453	};
	454	SR_REGISTER_DEV_DRIVER(dcttech_usbrelay_driver_info);