[libsigrok.git] / src / usb.c

/*
 * This file is part of the libsigrok project.
 *
 * Copyright (C) 2012 Uwe Hermann <uwe@hermann-uwe.de>
 * Copyright (C) 2012 Bert Vermeulen <bert@biot.com>
 *
 * 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 2 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, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
 */

#include <stdlib.h>
#include <memory.h>
#include <glib.h>
#include <libusb.h>
#include <libsigrok/libsigrok.h>
#include "libsigrok-internal.h"

/* SR_CONF_CONN takes one of these: */
#define CONN_USB_VIDPID  "^([0-9a-z]{4})\\.([0-9a-z]{4})$"
#define CONN_USB_BUSADDR "^(\\d+)\\.(\\d+)$"

#define LOG_PREFIX "usb"

/**
 * Find USB devices according to a connection string.
 *
 * @param usb_ctx libusb context to use while scanning.
 * @param conn Connection string specifying the device(s) to match. This
 * can be of the form "<bus>.<address>", or "<vendorid>.<productid>".
 *
 * @return A GSList of struct sr_usb_dev_inst, with bus and address fields
 * matching the device that matched the connection string. The GSList and
 * its contents must be freed by the caller.
 */
SR_PRIV GSList *sr_usb_find(libusb_context *usb_ctx, const char *conn)
{
	struct sr_usb_dev_inst *usb;
	struct libusb_device **devlist;
	struct libusb_device_descriptor des;
	GSList *devices;
	GRegex *reg;
	GMatchInfo *match;
	int vid, pid, bus, addr, b, a, ret, i;
	char *mstr;

	vid = pid = bus = addr = 0;
	reg = g_regex_new(CONN_USB_VIDPID, 0, 0, NULL);
	if (g_regex_match(reg, conn, 0, &match)) {
		if ((mstr = g_match_info_fetch(match, 1)))
			vid = strtoul(mstr, NULL, 16);
		g_free(mstr);

		if ((mstr = g_match_info_fetch(match, 2)))
			pid = strtoul(mstr, NULL, 16);
		g_free(mstr);
		sr_dbg("Trying to find USB device with VID:PID = %04x:%04x.",
		       vid, pid);
	} else {
		g_match_info_unref(match);
		g_regex_unref(reg);
		reg = g_regex_new(CONN_USB_BUSADDR, 0, 0, NULL);
		if (g_regex_match(reg, conn, 0, &match)) {
			if ((mstr = g_match_info_fetch(match, 1)))
				bus = strtoul(mstr, NULL, 10);
			g_free(mstr);

			if ((mstr = g_match_info_fetch(match, 2)))
				addr = strtoul(mstr, NULL, 10);
			g_free(mstr);
			sr_dbg("Trying to find USB device with bus.address = "
			       "%d.%d.", bus, addr);
		}
	}
	g_match_info_unref(match);
	g_regex_unref(reg);

	if (vid + pid + bus + addr == 0) {
		sr_err("Neither VID:PID nor bus.address was specified.");
		return NULL;
	}

	if (bus > 64) {
		sr_err("Invalid bus specified: %d.", bus);
		return NULL;
	}

	if (addr > 127) {
		sr_err("Invalid address specified: %d.", addr);
		return NULL;
	}

	/* Looks like a valid USB device specification, but is it connected? */
	devices = NULL;
	libusb_get_device_list(usb_ctx, &devlist);
	for (i = 0; devlist[i]; i++) {
		if ((ret = libusb_get_device_descriptor(devlist[i], &des))) {
			sr_err("Failed to get device descriptor: %s.",
			       libusb_error_name(ret));
			continue;
		}

		if (vid + pid && (des.idVendor != vid || des.idProduct != pid))
			continue;

		b = libusb_get_bus_number(devlist[i]);
		a = libusb_get_device_address(devlist[i]);
		if (bus + addr && (b != bus || a != addr))
			continue;

		sr_dbg("Found USB device (VID:PID = %04x:%04x, bus.address = "
		       "%d.%d).", des.idVendor, des.idProduct, b, a);

		usb = sr_usb_dev_inst_new(libusb_get_bus_number(devlist[i]),
				libusb_get_device_address(devlist[i]), NULL);
		devices = g_slist_append(devices, usb);
	}
	libusb_free_device_list(devlist, 1);

	sr_dbg("Found %d device(s).", g_slist_length(devices));
	
	return devices;
}

SR_PRIV int sr_usb_open(libusb_context *usb_ctx, struct sr_usb_dev_inst *usb)
{
	struct libusb_device **devlist;
	struct libusb_device_descriptor des;
	int ret, r, cnt, i, a, b;

	sr_dbg("Trying to open USB device %d.%d.", usb->bus, usb->address);

	if ((cnt = libusb_get_device_list(usb_ctx, &devlist)) < 0) {
		sr_err("Failed to retrieve device list: %s.",
		       libusb_error_name(cnt));
		return SR_ERR;
	}

	ret = SR_ERR;
	for (i = 0; i < cnt; i++) {
		if ((r = libusb_get_device_descriptor(devlist[i], &des)) < 0) {
			sr_err("Failed to get device descriptor: %s.",
			       libusb_error_name(r));
			continue;
		}

		b = libusb_get_bus_number(devlist[i]);
		a = libusb_get_device_address(devlist[i]);
		if (b != usb->bus || a != usb->address)
			continue;

		if ((r = libusb_open(devlist[i], &usb->devhdl)) < 0) {
			sr_err("Failed to open device: %s.",
			       libusb_error_name(r));
			break;
		}

		sr_dbg("Opened USB device (VID:PID = %04x:%04x, bus.address = "
		       "%d.%d).", des.idVendor, des.idProduct, b, a);

		ret = SR_OK;
		break;
	}

	libusb_free_device_list(devlist, 1);

	return ret;
}

SR_PRIV void sr_usb_close(struct sr_usb_dev_inst *usb)
{
	libusb_close(usb->devhdl);
	usb->devhdl = NULL;
	sr_dbg("Closed USB device %d.%d.", usb->bus, usb->address);
}

#ifdef _WIN32
/* Thread used to run libusb_wait_for_event() and set a pollable event. */
static gpointer usb_thread(gpointer data)
{
	struct sr_context *ctx = data;

	while (ctx->usb_thread_running) {
		/* Acquire event waiters lock, needed for libusb_wait_for_event(). */
		libusb_lock_event_waiters(ctx->libusb_ctx);
		/* Wait for any libusb event. The main thread can interrupt this wait
		 * by calling libusb_unlock_events(). */
		libusb_wait_for_event(ctx->libusb_ctx, NULL);
		/* Release event waiters lock. */
		libusb_unlock_event_waiters(ctx->libusb_ctx);
		/* Set event that the main loop will be polling on. */
		SetEvent(ctx->usb_wait_complete_event);
		/* Wait for the main thread to signal us to run again. */
		WaitForSingleObject(ctx->usb_wait_request_event, INFINITE);
		ResetEvent(ctx->usb_wait_request_event);
	}

	return NULL;
}

/* Callback wrapper run when main g_poll() gets a USB event or timeout. */
static int usb_callback(int fd, int revents, void *cb_data)
{
	struct sr_context *ctx = cb_data;
	int ret;

	/* Run registered callback to handle libusb events. */
	ret = ctx->usb_cb(fd, revents, ctx->usb_cb_data);

	/* Were we triggered by an event from the wait thread, rather than by a
	 * timeout? */
	int triggered_by_event = (revents & G_IO_IN);

	/* If so, and if the USB event source has not been removed from the
	 * session, reset the event that woke us and tell the wait thread to start
	 * waiting for events again. */
	if (triggered_by_event && ctx->usb_thread_running) {
		ResetEvent(ctx->usb_wait_complete_event);
		SetEvent(ctx->usb_wait_request_event);
	}

	return ret;
}
#endif

SR_PRIV int usb_source_add(struct sr_session *session, struct sr_context *ctx,
		int timeout, sr_receive_data_callback cb, void *cb_data)
{
	if (ctx->usb_source_present) {
		sr_err("A USB event source is already present.");
		return SR_ERR;
	}

#ifdef _WIN32
	/* Create events used to signal between main and USB wait threads. */
	ctx->usb_wait_request_event = CreateEvent(NULL, TRUE, FALSE, NULL);
	ctx->usb_wait_complete_event = CreateEvent(NULL, TRUE, FALSE, NULL);
	/* Start USB wait thread. */
	ctx->usb_thread_running = TRUE;
	ctx->usb_thread = g_thread_new("usb", usb_thread, ctx);
	/* Add event, set by USB wait thread, to session poll set. */
	ctx->usb_pollfd.fd = ctx->usb_wait_complete_event;
	ctx->usb_pollfd.events = G_IO_IN;
	ctx->usb_pollfd.revents = 0;
	ctx->usb_cb = cb;
	ctx->usb_cb_data = cb_data;
	sr_session_source_add_internal(session, &ctx->usb_pollfd, timeout,
			usb_callback, ctx, (gintptr)&ctx->usb_pollfd, TRUE);
#else
	const struct libusb_pollfd **lupfd;
	unsigned int i;

	lupfd = libusb_get_pollfds(ctx->libusb_ctx);
	for (i = 0; lupfd[i]; i++) {
		GPollFD p;

		p.fd = lupfd[i]->fd;
		p.events = lupfd[i]->events;
		p.revents = 0;

		sr_session_source_add_internal(session, &p, timeout,
				cb, cb_data, p.fd, TRUE);
	}
	free(lupfd);
#endif
	ctx->usb_source_present = TRUE;

	return SR_OK;
}

SR_PRIV int usb_source_remove(struct sr_session *session, struct sr_context *ctx)
{
	if (!ctx->usb_source_present)
		return SR_OK;

#ifdef _WIN32
	/* Signal the USB wait thread to stop, then unblock it so it does. */
	ctx->usb_thread_running = FALSE;
	SetEvent(ctx->usb_wait_request_event);
	libusb_unlock_events(ctx->libusb_ctx);
	/* Wait for USB wait thread to terminate. */
	g_thread_join(ctx->usb_thread);
	/* Remove USB event from session poll set. */
	sr_session_source_remove_pollfd(session, &ctx->usb_pollfd);
	/* Close event handles that were used between threads. */
	CloseHandle(ctx->usb_wait_request_event);
	CloseHandle(ctx->usb_wait_complete_event);
#else
	const struct libusb_pollfd **lupfd;
	unsigned int i;

	lupfd = libusb_get_pollfds(ctx->libusb_ctx);
	for (i = 0; lupfd[i]; i++)
		sr_session_source_remove(session, lupfd[i]->fd);
	free(lupfd);
#endif
	ctx->usb_source_present = FALSE;

	return SR_OK;
}

SR_PRIV int usb_get_port_path(libusb_device *dev, char *path, int path_len)
{
	uint8_t port_numbers[8];
	int i, n, len;

/*
 * FreeBSD requires that devices prior to calling libusb_get_port_numbers()
 * have been opened with libusb_open().
 */
#ifdef __FreeBSD__
	struct libusb_device_handle *devh;
	if (libusb_open(dev, &devh) != 0)
		return SR_ERR;
#endif
	n = libusb_get_port_numbers(dev, port_numbers, sizeof(port_numbers));
#ifdef __FreeBSD__
	libusb_close(devh);
#endif

/* Workaround FreeBSD libusb_get_port_numbers() returning 0. */
#ifdef __FreeBSD__
	if (n == 0) {
		port_numbers[0] = libusb_get_device_address(dev);
		n = 1;
	}
#endif
	if (n < 1)
		return SR_ERR;

	len = snprintf(path, path_len, "usb/%d-%d",
	               libusb_get_bus_number(dev), port_numbers[0]);

	for (i = 1; i < n; i++)
		len += snprintf(path+len, path_len-len, ".%d", port_numbers[i]);

	return SR_OK;
}
Commit	Line	Data
0c632d36	1	/*
50985c20	2	* This file is part of the libsigrok project.
0c632d36 UH	3	*
	4	* Copyright (C) 2012 Uwe Hermann <uwe@hermann-uwe.de>
	5	* Copyright (C) 2012 Bert Vermeulen <bert@biot.com>
	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 2 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, write to the Free Software
	19	* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
	20	*/
	21
	22	#include <stdlib.h>
db156e54	23	#include <memory.h>
0c632d36 UH	24	#include <glib.h>
0c632d36 UH	25	#include <libusb.h>
c1aae900	26	#include <libsigrok/libsigrok.h>
0c632d36 UH	27	#include "libsigrok-internal.h"
0c632d36 UH	28
1953564a	29	/* SR_CONF_CONN takes one of these: */
1eb0a0df	30	#define CONN_USB_VIDPID "^([0-9a-z]{4})\\.([0-9a-z]{4})$"
0c632d36 UH	31	#define CONN_USB_BUSADDR "^(\\d+)\\.(\\d+)$"
0c632d36 UH	32
3544f848	33	#define LOG_PREFIX "usb"
0c632d36	34
7ae6a758 BV	35	/**
	36	* Find USB devices according to a connection string.
	37	*
	38	* @param usb_ctx libusb context to use while scanning.
	39	* @param conn Connection string specifying the device(s) to match. This
	40	* can be of the form "<bus>.<address>", or "<vendorid>.<productid>".
	41	*
	42	* @return A GSList of struct sr_usb_dev_inst, with bus and address fields
	43	* matching the device that matched the connection string. The GSList and
	44	* its contents must be freed by the caller.
	45	*/
	46	SR_PRIV GSList sr_usb_find(libusb_context usb_ctx, const char *conn)
0c632d36	47	{
7ae6a758	48	struct sr_usb_dev_inst *usb;
0c632d36 UH	49	struct libusb_device **devlist;
	50	struct libusb_device_descriptor des;
	51	GSList *devices;
	52	GRegex *reg;
	53	GMatchInfo *match;
	54	int vid, pid, bus, addr, b, a, ret, i;
	55	char *mstr;
	56
	57	vid = pid = bus = addr = 0;
	58	reg = g_regex_new(CONN_USB_VIDPID, 0, 0, NULL);
	59	if (g_regex_match(reg, conn, 0, &match)) {
e7f378fd	60	if ((mstr = g_match_info_fetch(match, 1)))
0c632d36	61	vid = strtoul(mstr, NULL, 16);
0c632d36 UH	62	g_free(mstr);
0c632d36 UH	63
e7f378fd	64	if ((mstr = g_match_info_fetch(match, 2)))
0c632d36	65	pid = strtoul(mstr, NULL, 16);
0c632d36 UH	66	g_free(mstr);
	67	sr_dbg("Trying to find USB device with VID:PID = %04x:%04x.",
	68	vid, pid);
	69	} else {
	70	g_match_info_unref(match);
	71	g_regex_unref(reg);
	72	reg = g_regex_new(CONN_USB_BUSADDR, 0, 0, NULL);
	73	if (g_regex_match(reg, conn, 0, &match)) {
e7f378fd	74	if ((mstr = g_match_info_fetch(match, 1)))
1eb0a0df	75	bus = strtoul(mstr, NULL, 10);
0c632d36 UH	76	g_free(mstr);
0c632d36 UH	77
e7f378fd	78	if ((mstr = g_match_info_fetch(match, 2)))
1eb0a0df	79	addr = strtoul(mstr, NULL, 10);
0c632d36 UH	80	g_free(mstr);
	81	sr_dbg("Trying to find USB device with bus.address = "
	82	"%d.%d.", bus, addr);
	83	}
	84	}
	85	g_match_info_unref(match);
	86	g_regex_unref(reg);
	87
	88	if (vid + pid + bus + addr == 0) {
e7f378fd	89	sr_err("Neither VID:PID nor bus.address was specified.");
0c632d36 UH	90	return NULL;
	91	}
	92
	93	if (bus > 64) {
	94	sr_err("Invalid bus specified: %d.", bus);
	95	return NULL;
	96	}
	97
	98	if (addr > 127) {
	99	sr_err("Invalid address specified: %d.", addr);
	100	return NULL;
	101	}
	102
	103	/* Looks like a valid USB device specification, but is it connected? */
	104	devices = NULL;
	105	libusb_get_device_list(usb_ctx, &devlist);
	106	for (i = 0; devlist[i]; i++) {
	107	if ((ret = libusb_get_device_descriptor(devlist[i], &des))) {
d4928d71 PS	108	sr_err("Failed to get device descriptor: %s.",
d4928d71 PS	109	libusb_error_name(ret));
0c632d36 UH	110	continue;
	111	}
	112
7ae6a758	113	if (vid + pid && (des.idVendor != vid \|\| des.idProduct != pid))
0c632d36	114	continue;
0c632d36	115
7ae6a758 BV	116	b = libusb_get_bus_number(devlist[i]);
	117	a = libusb_get_device_address(devlist[i]);
	118	if (bus + addr && (b != bus \|\| a != addr))
0c632d36	119	continue;
0c632d36 UH	120
	121	sr_dbg("Found USB device (VID:PID = %04x:%04x, bus.address = "
	122	"%d.%d).", des.idVendor, des.idProduct, b, a);
	123
7ae6a758 BV	124	usb = sr_usb_dev_inst_new(libusb_get_bus_number(devlist[i]),
	125	libusb_get_device_address(devlist[i]), NULL);
	126	devices = g_slist_append(devices, usb);
0c632d36 UH	127	}
	128	libusb_free_device_list(devlist, 1);
	129
	130	sr_dbg("Found %d device(s).", g_slist_length(devices));
	131
	132	return devices;
	133	}
	134
	135	SR_PRIV int sr_usb_open(libusb_context usb_ctx, struct sr_usb_dev_inst usb)
	136	{
	137	struct libusb_device **devlist;
	138	struct libusb_device_descriptor des;
	139	int ret, r, cnt, i, a, b;
	140
c5f1a021	141	sr_dbg("Trying to open USB device %d.%d.", usb->bus, usb->address);
0c632d36 UH	142
	143	if ((cnt = libusb_get_device_list(usb_ctx, &devlist)) < 0) {
	144	sr_err("Failed to retrieve device list: %s.",
	145	libusb_error_name(cnt));
	146	return SR_ERR;
	147	}
	148
	149	ret = SR_ERR;
	150	for (i = 0; i < cnt; i++) {
	151	if ((r = libusb_get_device_descriptor(devlist[i], &des)) < 0) {
	152	sr_err("Failed to get device descriptor: %s.",
	153	libusb_error_name(r));
	154	continue;
	155	}
	156
	157	b = libusb_get_bus_number(devlist[i]);
	158	a = libusb_get_device_address(devlist[i]);
c5f1a021	159	if (b != usb->bus \|\| a != usb->address)
0c632d36	160	continue;
0c632d36 UH	161
	162	if ((r = libusb_open(devlist[i], &usb->devhdl)) < 0) {
	163	sr_err("Failed to open device: %s.",
	164	libusb_error_name(r));
	165	break;
	166	}
	167
	168	sr_dbg("Opened USB device (VID:PID = %04x:%04x, bus.address = "
	169	"%d.%d).", des.idVendor, des.idProduct, b, a);
	170
	171	ret = SR_OK;
	172	break;
	173	}
	174
	175	libusb_free_device_list(devlist, 1);
	176
	177	return ret;
	178	}
6c60facc	179
67e95ed3 BV	180	SR_PRIV void sr_usb_close(struct sr_usb_dev_inst *usb)
	181	{
	182	libusb_close(usb->devhdl);
	183	usb->devhdl = NULL;
	184	sr_dbg("Closed USB device %d.%d.", usb->bus, usb->address);
	185	}
	186
5321ac6b	187	#ifdef _WIN32
f6c30de4	188	/* Thread used to run libusb_wait_for_event() and set a pollable event. */
dcc94340	189	static gpointer usb_thread(gpointer data)
5321ac6b ML	190	{
	191	struct sr_context *ctx = data;
	192
	193	while (ctx->usb_thread_running) {
f6c30de4 ML	194	/* Acquire event waiters lock, needed for libusb_wait_for_event(). */
	195	libusb_lock_event_waiters(ctx->libusb_ctx);
	196	/* Wait for any libusb event. The main thread can interrupt this wait
	197	* by calling libusb_unlock_events(). */
5321ac6b	198	libusb_wait_for_event(ctx->libusb_ctx, NULL);
f6c30de4 ML	199	/* Release event waiters lock. */
	200	libusb_unlock_event_waiters(ctx->libusb_ctx);
	201	/* Set event that the main loop will be polling on. */
	202	SetEvent(ctx->usb_wait_complete_event);
	203	/* Wait for the main thread to signal us to run again. */
	204	WaitForSingleObject(ctx->usb_wait_request_event, INFINITE);
	205	ResetEvent(ctx->usb_wait_request_event);
5321ac6b ML	206	}
	207
	208	return NULL;
	209	}
	210
f6c30de4	211	/* Callback wrapper run when main g_poll() gets a USB event or timeout. */
dcc94340	212	static int usb_callback(int fd, int revents, void *cb_data)
5321ac6b ML	213	{
	214	struct sr_context *ctx = cb_data;
	215	int ret;
	216
f6c30de4	217	/* Run registered callback to handle libusb events. */
5321ac6b	218	ret = ctx->usb_cb(fd, revents, ctx->usb_cb_data);
34ea7f96	219
f6c30de4 ML	220	/* Were we triggered by an event from the wait thread, rather than by a
	221	* timeout? */
	222	int triggered_by_event = (revents & G_IO_IN);
	223
	224	/* If so, and if the USB event source has not been removed from the
	225	* session, reset the event that woke us and tell the wait thread to start
	226	* waiting for events again. */
	227	if (triggered_by_event && ctx->usb_thread_running) {
	228	ResetEvent(ctx->usb_wait_complete_event);
	229	SetEvent(ctx->usb_wait_request_event);
34ea7f96	230	}
5321ac6b ML	231
	232	return ret;
	233	}
	234	#endif
	235
102f1239 BV	236	SR_PRIV int usb_source_add(struct sr_session session, struct sr_context ctx,
102f1239 BV	237	int timeout, sr_receive_data_callback cb, void *cb_data)
6c60facc	238	{
6640324f ML	239	if (ctx->usb_source_present) {
	240	sr_err("A USB event source is already present.");
	241	return SR_ERR;
	242	}
	243
6c60facc	244	#ifdef _WIN32
f6c30de4 ML	245	/* Create events used to signal between main and USB wait threads. */
	246	ctx->usb_wait_request_event = CreateEvent(NULL, TRUE, FALSE, NULL);
	247	ctx->usb_wait_complete_event = CreateEvent(NULL, TRUE, FALSE, NULL);
	248	/* Start USB wait thread. */
5321ac6b ML	249	ctx->usb_thread_running = TRUE;
5321ac6b ML	250	ctx->usb_thread = g_thread_new("usb", usb_thread, ctx);
f6c30de4 ML	251	/* Add event, set by USB wait thread, to session poll set. */
f6c30de4 ML	252	ctx->usb_pollfd.fd = ctx->usb_wait_complete_event;
5321ac6b	253	ctx->usb_pollfd.events = G_IO_IN;
89efe064	254	ctx->usb_pollfd.revents = 0;
5321ac6b ML	255	ctx->usb_cb = cb;
5321ac6b ML	256	ctx->usb_cb_data = cb_data;
62d7945f DE	257	sr_session_source_add_internal(session, &ctx->usb_pollfd, timeout,
62d7945f DE	258	usb_callback, ctx, (gintptr)&ctx->usb_pollfd, TRUE);
6c60facc ML	259	#else
	260	const struct libusb_pollfd **lupfd;
	261	unsigned int i;
	262
	263	lupfd = libusb_get_pollfds(ctx->libusb_ctx);
62d7945f DE	264	for (i = 0; lupfd[i]; i++) {
	265	GPollFD p;
	266
	267	p.fd = lupfd[i]->fd;
	268	p.events = lupfd[i]->events;
	269	p.revents = 0;
	270
	271	sr_session_source_add_internal(session, &p, timeout,
	272	cb, cb_data, p.fd, TRUE);
	273	}
6c60facc	274	free(lupfd);
5321ac6b	275	#endif
6640324f	276	ctx->usb_source_present = TRUE;
6c60facc ML	277
6c60facc ML	278	return SR_OK;
6c60facc ML	279	}
6c60facc ML	280
102f1239	281	SR_PRIV int usb_source_remove(struct sr_session session, struct sr_context ctx)
6c60facc	282	{
6640324f ML	283	if (!ctx->usb_source_present)
	284	return SR_OK;
	285
6c60facc	286	#ifdef _WIN32
f6c30de4	287	/* Signal the USB wait thread to stop, then unblock it so it does. */
5321ac6b	288	ctx->usb_thread_running = FALSE;
f6c30de4	289	SetEvent(ctx->usb_wait_request_event);
5321ac6b	290	libusb_unlock_events(ctx->libusb_ctx);
f6c30de4	291	/* Wait for USB wait thread to terminate. */
5321ac6b	292	g_thread_join(ctx->usb_thread);
f6c30de4	293	/* Remove USB event from session poll set. */
102f1239	294	sr_session_source_remove_pollfd(session, &ctx->usb_pollfd);
f6c30de4 ML	295	/* Close event handles that were used between threads. */
	296	CloseHandle(ctx->usb_wait_request_event);
	297	CloseHandle(ctx->usb_wait_complete_event);
6c60facc ML	298	#else
	299	const struct libusb_pollfd **lupfd;
	300	unsigned int i;
	301
	302	lupfd = libusb_get_pollfds(ctx->libusb_ctx);
	303	for (i = 0; lupfd[i]; i++)
102f1239	304	sr_session_source_remove(session, lupfd[i]->fd);
6c60facc	305	free(lupfd);
5321ac6b	306	#endif
6640324f	307	ctx->usb_source_present = FALSE;
6c60facc ML	308
6c60facc ML	309	return SR_OK;
6c60facc	310	}
db156e54	311
76bc5f63	312	SR_PRIV int usb_get_port_path(libusb_device dev, char path, int path_len)
db156e54	313	{
76bc5f63 AJ	314	uint8_t port_numbers[8];
	315	int i, n, len;
	316
19643b96 UJ	317	/*
	318	* FreeBSD requires that devices prior to calling libusb_get_port_numbers()
	319	* have been opened with libusb_open().
	320	*/
	321	#ifdef __FreeBSD__
	322	struct libusb_device_handle *devh;
	323	if (libusb_open(dev, &devh) != 0)
	324	return SR_ERR;
	325	#endif
76bc5f63	326	n = libusb_get_port_numbers(dev, port_numbers, sizeof(port_numbers));
19643b96 UJ	327	#ifdef __FreeBSD__
	328	libusb_close(devh);
	329	#endif
76bc5f63	330
19643b96 UJ	331	/* Workaround FreeBSD libusb_get_port_numbers() returning 0. */
	332	#ifdef __FreeBSD__
	333	if (n == 0) {
	334	port_numbers[0] = libusb_get_device_address(dev);
	335	n = 1;
	336	}
	337	#endif
2f004b4b VP	338	if (n < 1)
	339	return SR_ERR;
	340
76bc5f63 AJ	341	len = snprintf(path, path_len, "usb/%d-%d",
76bc5f63 AJ	342	libusb_get_bus_number(dev), port_numbers[0]);
db156e54	343
76bc5f63 AJ	344	for (i = 1; i < n; i++)
76bc5f63 AJ	345	len += snprintf(path+len, path_len-len, ".%d", port_numbers[i]);
db156e54 SA	346
	347	return SR_OK;
	348	}