/*

 *

 * Copyright (c) 2016, Oracle and/or its affiliates.

 *

 * This library is free software; you can redistribute it and/or

 * modify it under the terms of the GNU Lesser General Public

 * License as published by the Free Software Foundation; either

 * version 2.1 of the License, or (at your option) any later version.

 *

 * This library 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

 * Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public

 * License along with this library; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

 */

#include <config.h>

#include <sys/time.h>

#include <sys/types.h>

#include <sys/list.h>

#include <sys/stat.h>

#include <strings.h>

#include <errno.h>

#include <fcntl.h>

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <wait.h>

#include <unistd.h>

#include <aio.h>

#include <libdevinfo.h>

#include <sys/nvpair.h>

#include <sys/devctl.h>

#include <sys/usb/clients/ugen/usb_ugen.h>

#include <errno.h>

#include <sys/usb/usba.h>

#include <sys/pci.h>

#include "libusbi.h"

#include "sunos_usb.h"

#define UPDATEDRV_PATH	"/usr/sbin/update_drv"

#define UPDATEDRV	"update_drv"

typedef list_t string_list_t;

typedef struct string_node {

	char		*string;

	list_node_t	link;

} string_node_t;

/*

 * Backend functions

 */

static int sunos_init(struct libusb_context *);

static void sunos_exit(struct libusb_context *);

static int sunos_get_device_list(struct libusb_context *,

    struct discovered_devs **);

static int sunos_open(struct libusb_device_handle *);

static void sunos_close(struct libusb_device_handle *);

static int sunos_get_device_descriptor(struct libusb_device *,

    uint8_t*, int *);

static int sunos_get_active_config_descriptor(struct libusb_device *,

    uint8_t*, size_t, int *);

static int sunos_get_config_descriptor(struct libusb_device *, uint8_t,

    uint8_t*, size_t, int *);

static int sunos_get_configuration(struct libusb_device_handle *, int *);

static int sunos_set_configuration(struct libusb_device_handle *, int);

static int sunos_claim_interface(struct libusb_device_handle *, int);

static int sunos_release_interface(struct libusb_device_handle *, int);

static int sunos_set_interface_altsetting(struct libusb_device_handle *,

    int, int);

static int sunos_clear_halt(struct libusb_device_handle *, uint8_t);

static int sunos_reset_device(struct libusb_device_handle *);

static void sunos_destroy_device(struct libusb_device *);

static int sunos_submit_transfer(struct usbi_transfer *);

static int sunos_cancel_transfer(struct usbi_transfer *);

static void sunos_clear_transfer_priv(struct usbi_transfer *);

static int sunos_handle_transfer_completion(struct usbi_transfer *);

static int sunos_clock_gettime(int, struct timespec *);

static int sunos_kernel_driver_active(struct libusb_device_handle *, int interface);

static int sunos_detach_kernel_driver (struct libusb_device_handle *dev, int interface_number);

static int sunos_attach_kernel_driver (struct libusb_device_handle *dev, int interface_number);

static int sunos_usb_open_ep0(sunos_dev_handle_priv_t *hpriv, sunos_dev_priv_t *dpriv);

static int sunos_usb_ioctl(struct libusb_device *dev, int cmd);

static struct devctl_iocdata iocdata;

static int sunos_get_link(di_devlink_t devlink, void *arg)

{

	walk_link_t *larg = (walk_link_t *)arg;

	const char *p;

	const char *q;

	if (larg->path) {

		char *content = (char *)di_devlink_content(devlink);

		char *start = strstr(content, "/devices/");

		start += strlen("/devices");

		usbi_dbg("%s", start);

		/* line content must have minor node */

		if (start == NULL ||

		    strncmp(start, larg->path, larg->len) != 0 ||

		    start[larg->len] != ':')

			return (DI_WALK_CONTINUE);

	}

	p = di_devlink_path(devlink);

	q = strrchr(p, '/');

	usbi_dbg("%s", q);

	*(larg->linkpp) = strndup(p, strlen(p) - strlen(q));

	return (DI_WALK_TERMINATE);

}

static int sunos_physpath_to_devlink(

	const char *node_path, const char *match, char **link_path)

{

	walk_link_t larg;

	di_devlink_handle_t hdl;

	*link_path = NULL;

	larg.linkpp = link_path;

	if ((hdl = di_devlink_init(NULL, 0)) == NULL) {

		usbi_dbg("di_devlink_init failure");

		return (-1);

	}

	larg.len = strlen(node_path);

	larg.path = (char *)node_path;

	(void) di_devlink_walk(hdl, match, NULL, DI_PRIMARY_LINK,

	    (void *)&larg, sunos_get_link);

	(void) di_devlink_fini(&hdl);

	if (*link_path == NULL) {

		usbi_dbg("there is no devlink for this path");

		return (-1);

	}

	return 0;

}

static int

sunos_usb_ioctl(struct libusb_device *dev, int cmd)

{

	int fd;

	nvlist_t *nvlist;

	char *end;

	char *phypath;

	char *hubpath;

	char path_arg[PATH_MAX];

	sunos_dev_priv_t *dpriv;

	devctl_ap_state_t devctl_ap_state;

	dpriv = (sunos_dev_priv_t *)dev->os_priv;

	phypath = dpriv->phypath;

	end = strrchr(phypath, '/');

	if (end == NULL)

		return (-1);

	hubpath = strndup(phypath, end - phypath);

	if (hubpath == NULL)

		return (-1);

	end = strrchr(hubpath, '@');

	if (end == NULL) {

		free(hubpath);

		return (-1);

	}

	end++;

	usbi_dbg("unitaddr: %s", end);

	nvlist_alloc(&nvlist, NV_UNIQUE_NAME_TYPE, KM_NOSLEEP);

	nvlist_add_int32(nvlist, "port", dev->port_number);

	//find the hub path

	snprintf(path_arg, sizeof(path_arg), "/devices%s:hubd", hubpath);

	usbi_dbg("ioctl hub path: %s", path_arg);

	fd = open(path_arg, O_RDONLY);

	if (fd < 0) {

		usbi_err(DEVICE_CTX(dev), "open failed: %d (%s)", errno, strerror(errno));

		nvlist_free(nvlist);

		free(hubpath);

		return (-1);

	}

	memset(&iocdata, 0, sizeof(iocdata));

	memset(&devctl_ap_state, 0, sizeof(devctl_ap_state));

	nvlist_pack(nvlist, (char **)&iocdata.nvl_user, &iocdata.nvl_usersz, NV_ENCODE_NATIVE, 0);

	iocdata.cmd = DEVCTL_AP_GETSTATE;

	iocdata.flags = 0;

	iocdata.c_nodename = "hub";

	iocdata.c_unitaddr = end;

	iocdata.cpyout_buf = &devctl_ap_state;

	usbi_dbg("%p, %d", iocdata.nvl_user, iocdata.nvl_usersz);

	errno = 0;

	if (ioctl(fd, DEVCTL_AP_GETSTATE, &iocdata) == -1) {

		usbi_err(DEVICE_CTX(dev), "ioctl failed: fd %d, cmd %x, errno %d (%s)",

			 fd, DEVCTL_AP_GETSTATE, errno, strerror(errno));

	} else {

		usbi_dbg("dev rstate: %d", devctl_ap_state.ap_rstate);

		usbi_dbg("dev ostate: %d", devctl_ap_state.ap_ostate);

	}

	errno = 0;

	iocdata.cmd = cmd;

	if (ioctl(fd, (int)cmd, &iocdata) != 0) {

		usbi_err(DEVICE_CTX(dev), "ioctl failed: fd %d, cmd %x, errno %d (%s)",

			 fd, cmd, errno, strerror(errno));

		sleep(2);

	}

	close(fd);

	free(iocdata.nvl_user);

	nvlist_free(nvlist);

	free(hubpath);

	return (-errno);

}

static int

sunos_kernel_driver_active(struct libusb_device_handle *dev, int interface)

{

	sunos_dev_priv_t *dpriv;

	dpriv = (sunos_dev_priv_t *)dev->dev->os_priv;

	usbi_dbg("%s", dpriv->ugenpath);

	return (dpriv->ugenpath == NULL);

}

/*

 * Private functions

 */

static int _errno_to_libusb(int);

static int sunos_usb_get_status(int fd);

static int sunos_init(struct libusb_context *ctx)

{

	return (LIBUSB_SUCCESS);

}

static void sunos_exit(struct libusb_context *ctx)

{

	usbi_dbg("");

}

static string_list_t *

sunos_new_string_list(void)

{

	string_list_t *list;

	list = calloc(1, sizeof(*list));

	if (list != NULL)

		list_create(list, sizeof(string_node_t),

			    offsetof(string_node_t, link));

	return (list);

}

static int

sunos_append_to_string_list(string_list_t *list, const char *arg)

{

	string_node_t *np;

	np = calloc(1, sizeof(*np));

	if (!np)

		return (-1);

	np->string = strdup(arg);

	if (!np->string) {

		free(np);

		return (-1);

	}

	list_insert_tail(list, np);

	return (0);

}

static void

sunos_free_string_list(string_list_t *list)

{

	string_node_t *np;

	while ((np = list_remove_head(list)) != NULL) {

		free(np->string);

		free(np);

	}

	free(list);

}

static char **

sunos_build_argv_list(string_list_t *list)

{

	char **argv_list;

	string_node_t *np;

	int n;

	n = 1; /* Start at 1 for NULL terminator */

	for (np = list_head(list); np != NULL; np = list_next(list, np))

		n++;

	argv_list = calloc(n, sizeof(char *));

	if (argv_list == NULL)

		return NULL;

	n = 0;

	for (np = list_head(list); np != NULL; np = list_next(list, np))

		argv_list[n++] = np->string;

	return (argv_list);

}

static int

sunos_exec_command(struct libusb_context *ctx, const char *path,

	string_list_t *list)

{

	pid_t pid;

	int status;

	int waitstat;

	int exit_status;

	char **argv_list;

	argv_list = sunos_build_argv_list(list);

	if (argv_list == NULL)

		return (-1);

	pid = fork();

	if (pid == 0) {

		/* child */

		execv(path, argv_list);

		_exit(127);

	} else if (pid > 0) {

		/* parent */

		do {

			waitstat = waitpid(pid, &status, 0);

		} while ((waitstat == -1 && errno == EINTR) ||

			 (waitstat == 0 && !WIFEXITED(status) && !WIFSIGNALED(status)));

		if (waitstat == 0) {

			if (WIFEXITED(status))

				exit_status = WEXITSTATUS(status);

			else

				exit_status = WTERMSIG(status);

		} else {

			usbi_err(ctx, "waitpid failed: errno %d (%s)", errno, strerror(errno));

			exit_status = -1;

		}

	} else {

		/* fork failed */

		usbi_err(ctx, "fork failed: errno %d (%s)", errno, strerror(errno));

		exit_status = -1;

	}

	free(argv_list);

	return (exit_status);

}

static int

sunos_detach_kernel_driver(struct libusb_device_handle *dev_handle,

	int interface_number)

{

	struct libusb_context *ctx = HANDLE_CTX(dev_handle);

	string_list_t *list;

	char path_arg[PATH_MAX];

	sunos_dev_priv_t *dpriv;

	int r;

	dpriv = (sunos_dev_priv_t *)dev_handle->dev->os_priv;

	snprintf(path_arg, sizeof(path_arg), "\'\"%s\"\'", dpriv->phypath);

	usbi_dbg("%s", path_arg);

	list = sunos_new_string_list();

	if (list == NULL)

		return (LIBUSB_ERROR_NO_MEM);

	/* attach ugen driver */

	r = 0;

	r |= sunos_append_to_string_list(list, UPDATEDRV);

	r |= sunos_append_to_string_list(list, "-a"); /* add rule */

	r |= sunos_append_to_string_list(list, "-i"); /* specific device */

	r |= sunos_append_to_string_list(list, path_arg); /* physical path */

	r |= sunos_append_to_string_list(list, "ugen");

	if (r) {

		sunos_free_string_list(list);

		return (LIBUSB_ERROR_NO_MEM);

	}

	r = sunos_exec_command(ctx, UPDATEDRV_PATH, list);

	sunos_free_string_list(list);

	if (r < 0)

		return (LIBUSB_ERROR_OTHER);

	/* reconfigure the driver node */

	r = 0;

	r |= sunos_usb_ioctl(dev_handle->dev, DEVCTL_AP_DISCONNECT);

	r |= sunos_usb_ioctl(dev_handle->dev, DEVCTL_AP_CONFIGURE);

	if (r)

		usbi_warn(HANDLE_CTX(dev_handle), "one or more ioctls failed");

	snprintf(path_arg, sizeof(path_arg), "^usb/%x.%x", dpriv->dev_descr.idVendor,

	    dpriv->dev_descr.idProduct);

	sunos_physpath_to_devlink(dpriv->phypath, path_arg, &dpriv->ugenpath);

	if (access(dpriv->ugenpath, F_OK) == -1) {

		usbi_err(HANDLE_CTX(dev_handle), "fail to detach kernel driver");

		return (LIBUSB_ERROR_IO);

	}

	return sunos_usb_open_ep0((sunos_dev_handle_priv_t *)dev_handle->os_priv, dpriv);

}

static int

sunos_attach_kernel_driver(struct libusb_device_handle *dev_handle,

	int interface_number)

{

	struct libusb_context *ctx = HANDLE_CTX(dev_handle);

	string_list_t *list;

	char path_arg[PATH_MAX];

	sunos_dev_priv_t *dpriv;

	int r;

	/* we open the dev in detach driver, so we need close it first. */

	sunos_close(dev_handle);

	dpriv = (sunos_dev_priv_t *)dev_handle->dev->os_priv;

	snprintf(path_arg, sizeof(path_arg), "\'\"%s\"\'", dpriv->phypath);

	usbi_dbg("%s", path_arg);

	list = sunos_new_string_list();

	if (list == NULL)

		return (LIBUSB_ERROR_NO_MEM);

	/* detach ugen driver */

	r = 0;

	r |= sunos_append_to_string_list(list, UPDATEDRV);

	r |= sunos_append_to_string_list(list, "-d"); /* add rule */

	r |= sunos_append_to_string_list(list, "-i"); /* specific device */

	r |= sunos_append_to_string_list(list, path_arg); /* physical path */

	r |= sunos_append_to_string_list(list, "ugen");

	if (r) {

		sunos_free_string_list(list);

		return (LIBUSB_ERROR_NO_MEM);

	}

	r = sunos_exec_command(ctx, UPDATEDRV_PATH, list);

	sunos_free_string_list(list);

	if (r < 0)

		return (LIBUSB_ERROR_OTHER);

	/* reconfigure the driver node */

	r = 0;

	r |= sunos_usb_ioctl(dev_handle->dev, DEVCTL_AP_CONFIGURE);

	r |= sunos_usb_ioctl(dev_handle->dev, DEVCTL_AP_DISCONNECT);

	r |= sunos_usb_ioctl(dev_handle->dev, DEVCTL_AP_CONFIGURE);

	if (r)

		usbi_warn(HANDLE_CTX(dev_handle), "one or more ioctls failed");

	return 0;

}

static int

sunos_fill_in_dev_info(di_node_t node, struct libusb_device *dev)

{

	int	proplen;

	int	*i, n, *addr, *port_prop;

	char	*phypath;

	uint8_t	*rdata;

	struct libusb_device_descriptor	*descr;

	sunos_dev_priv_t	*dpriv = (sunos_dev_priv_t *)dev->os_priv;

	char	match_str[PATH_MAX];

	/* Device descriptors */

	proplen = di_prop_lookup_bytes(DDI_DEV_T_ANY, node,

	    "usb-dev-descriptor", &rdata);

	if (proplen <= 0) {

		return (LIBUSB_ERROR_IO);

	}

	descr = (struct libusb_device_descriptor *)rdata;

	bcopy(descr, &dpriv->dev_descr, LIBUSB_DT_DEVICE_SIZE);

	dpriv->dev_descr.bcdUSB = libusb_cpu_to_le16(descr->bcdUSB);

	dpriv->dev_descr.idVendor = libusb_cpu_to_le16(descr->idVendor);

	dpriv->dev_descr.idProduct = libusb_cpu_to_le16(descr->idProduct);

	dpriv->dev_descr.bcdDevice = libusb_cpu_to_le16(descr->bcdDevice);

	/* Raw configuration descriptors */

	proplen = di_prop_lookup_bytes(DDI_DEV_T_ANY, node,

	    "usb-raw-cfg-descriptors", &rdata);

	if (proplen <= 0) {

		usbi_dbg("can't find raw config descriptors");

		return (LIBUSB_ERROR_IO);

	}

	dpriv->raw_cfgdescr = calloc(1, proplen);

	if (dpriv->raw_cfgdescr == NULL) {

		return (LIBUSB_ERROR_NO_MEM);

	} else {

		bcopy(rdata, dpriv->raw_cfgdescr, proplen);

		dpriv->cfgvalue = ((struct libusb_config_descriptor *)

		    rdata)->bConfigurationValue;

	}

	n = di_prop_lookup_ints(DDI_DEV_T_ANY, node, "reg", &port_prop);

	if ((n != 1) || (*port_prop <= 0)) {

		return (LIBUSB_ERROR_IO);

	}

	dev->port_number = *port_prop;

	/* device physical path */

	phypath = di_devfs_path(node);

	if (phypath) {

		dpriv->phypath = strdup(phypath);

		snprintf(match_str, sizeof(match_str), "^usb/%x.%x", dpriv->dev_descr.idVendor, dpriv->dev_descr.idProduct);

		usbi_dbg("match is %s", match_str);

		sunos_physpath_to_devlink(dpriv->phypath, match_str,  &dpriv->ugenpath);

		di_devfs_path_free(phypath);

	} else {

		free(dpriv->raw_cfgdescr);

		return (LIBUSB_ERROR_IO);

	}

	/* address */

	n = di_prop_lookup_ints(DDI_DEV_T_ANY, node, "assigned-address", &addr);

	if (n != 1 || *addr == 0) {

		usbi_dbg("can't get address");

	} else {

		dev->device_address = *addr;

	}

	/* speed */

	if (di_prop_lookup_ints(DDI_DEV_T_ANY, node, "low-speed", &i) >= 0) {

		dev->speed = LIBUSB_SPEED_LOW;

	} else if (di_prop_lookup_ints(DDI_DEV_T_ANY, node, "high-speed", &i) >= 0) {

		dev->speed = LIBUSB_SPEED_HIGH;

	} else if (di_prop_lookup_ints(DDI_DEV_T_ANY, node, "full-speed", &i) >= 0) {

		dev->speed = LIBUSB_SPEED_FULL;

	} else if (di_prop_lookup_ints(DDI_DEV_T_ANY, node, "super-speed", &i) >= 0) {

		dev->speed = LIBUSB_SPEED_SUPER;

	}

	usbi_dbg("vid=%x pid=%x, path=%s, bus_nmber=0x%x, port_number=%d, "

	    "speed=%d", dpriv->dev_descr.idVendor, dpriv->dev_descr.idProduct,

	    dpriv->phypath, dev->bus_number, dev->port_number, dev->speed);

	return (LIBUSB_SUCCESS);

}

static int

sunos_add_devices(di_devlink_t link, void *arg)

{

	struct devlink_cbarg	*largs = (struct devlink_cbarg *)arg;

	struct node_args	*nargs;

	di_node_t		myself, dn;

	uint64_t		session_id = 0;

	uint64_t		sid = 0;

	uint64_t		bdf = 0;

	struct libusb_device	*dev;

	sunos_dev_priv_t	*devpriv;

	int			n, *j;

	int			i = 0;

	int			*addr_prop;

	uint8_t			bus_number = 0;

	uint32_t * 		regbuf = NULL;

	uint32_t		reg;

	nargs = (struct node_args *)largs->nargs;

	myself = largs->myself;

	/*

	 * Construct session ID.

	 * session ID = dev_addr | hub addr |parent hub addr|...|root hub bdf

	 * 		8 bits       8bits          8 bits               16bits

	 */

	if (myself == DI_NODE_NIL)

		return (DI_WALK_CONTINUE);

	dn = myself;

	/* find the root hub */

	while (di_prop_lookup_ints(DDI_DEV_T_ANY, dn, "root-hub", &j) != 0) {

		usbi_dbg("find_root_hub:%s", di_devfs_path(dn));

		n = di_prop_lookup_ints(DDI_DEV_T_ANY, dn,

				"assigned-address", &addr_prop);

		session_id |= ((addr_prop[0] & 0xff) << i++ * 8);

		dn = di_parent_node(dn);

	}

	/* dn is the root hub node */

	n = di_prop_lookup_ints(DDI_DEV_T_ANY, dn, "reg", (int **)&regbuf);

	reg = regbuf[0];

	bdf = (PCI_REG_BUS_G(reg) << 8) | (PCI_REG_DEV_G(reg) << 3) | PCI_REG_FUNC_G(reg);

	/* bdf must larger than i*8 bits */

	session_id |= (bdf << i * 8);

	bus_number = (PCI_REG_DEV_G(reg) << 3) | PCI_REG_FUNC_G(reg);

	usbi_dbg("device bus address=%s:%x, name:%s",

	    di_bus_addr(myself), bus_number, di_node_name(dn));

	usbi_dbg("session id org:%lx", session_id);

	/* dn is the usb device */

	for (dn = di_child_node(myself); dn != DI_NODE_NIL; dn = di_sibling_node(dn)) {

		usbi_dbg("device path:%s", di_devfs_path(dn));

		/* skip hub devices, because its driver can not been unload */

		if (di_prop_lookup_ints(DDI_DEV_T_ANY, dn, "usb-port-count", &addr_prop) != -1)

			continue;

		/* usb_addr */

		n = di_prop_lookup_ints(DDI_DEV_T_ANY, dn,

		    "assigned-address", &addr_prop);

		if ((n != 1) || (addr_prop[0] == 0)) {

			usbi_dbg("cannot get valid usb_addr");

			continue;

		}

		sid = (session_id << 8) | (addr_prop[0] & 0xff) ;

		usbi_dbg("session id %lx", sid);

		dev = usbi_get_device_by_session_id(nargs->ctx, sid);

		if (dev == NULL) {

			dev = usbi_alloc_device(nargs->ctx, sid);

			if (dev == NULL) {

				usbi_dbg("can't alloc device");

				continue;

			}

			devpriv = (sunos_dev_priv_t *)dev->os_priv;

			dev->bus_number = bus_number;

			if (sunos_fill_in_dev_info(dn, dev) != LIBUSB_SUCCESS) {

				libusb_unref_device(dev);

				usbi_dbg("get infomation fail");

				continue;

			}

			if (usbi_sanitize_device(dev) < 0) {

				libusb_unref_device(dev);

				usbi_dbg("sanatize failed: ");

				return (DI_WALK_TERMINATE);

			}

		} else {

			devpriv = (sunos_dev_priv_t *)dev->os_priv;

			usbi_dbg("Dev %s exists", devpriv->ugenpath);

		}

		if (discovered_devs_append(*(nargs->discdevs), dev) == NULL) {

			usbi_dbg("cannot append device");

		}

		/*

		 * we alloc and hence ref this dev. We don't need to ref it

		 * hereafter. Front end or app should take care of their ref.

		 */

		libusb_unref_device(dev);

		usbi_dbg("Device %s %s id=0x%llx, devcount:%d, bdf=%x",

		    devpriv->ugenpath, di_devfs_path(dn), (uint64_t)sid,

		    (*nargs->discdevs)->len, bdf);

	}

	return (DI_WALK_CONTINUE);

}

static int

sunos_walk_minor_node_link(di_node_t node, void *args)

{

        di_minor_t minor = DI_MINOR_NIL;

        char *minor_path;

        struct devlink_cbarg arg;

	struct node_args *nargs = (struct node_args *)args;

	di_devlink_handle_t devlink_hdl = nargs->dlink_hdl;

	/* walk each minor to find usb devices */

        while ((minor = di_minor_next(node, minor)) != DI_MINOR_NIL) {

                minor_path = di_devfs_minor_path(minor);

                arg.nargs = args;

		arg.myself = node;

                arg.minor = minor;

                (void) di_devlink_walk(devlink_hdl,

		    "^usb/hub[0-9]+", minor_path,

		    DI_PRIMARY_LINK, (void *)&arg, sunos_add_devices);

                di_devfs_path_free(minor_path);

        }

	/* switch to a different node */

	nargs->last_ugenpath = NULL;

	return (DI_WALK_CONTINUE);

}

int

sunos_get_device_list(struct libusb_context * ctx,

	struct discovered_devs **discdevs)

{

	di_node_t root_node;

	struct node_args args;

	di_devlink_handle_t devlink_hdl;

	args.ctx = ctx;

	args.discdevs = discdevs;

	args.last_ugenpath = NULL;

	if ((root_node = di_init("/", DINFOCPYALL)) == DI_NODE_NIL) {

		usbi_dbg("di_int() failed: %s", strerror(errno));

		return (LIBUSB_ERROR_IO);

	}

	if ((devlink_hdl = di_devlink_init(NULL, 0)) == NULL) {

		di_fini(root_node);

		usbi_dbg("di_devlink_init() failed: %s", strerror(errno));

		return (LIBUSB_ERROR_IO);

	}

	args.dlink_hdl = devlink_hdl;

	/* walk each node to find USB devices */

	if (di_walk_node(root_node, DI_WALK_SIBFIRST, &args,

	    sunos_walk_minor_node_link) == -1) {

		usbi_dbg("di_walk_node() failed: %s", strerror(errno));

		di_fini(root_node);

		return (LIBUSB_ERROR_IO);

	}

	di_fini(root_node);

	di_devlink_fini(&devlink_hdl);

	usbi_dbg("%d devices", (*discdevs)->len);

	return ((*discdevs)->len);

}

static int

sunos_usb_open_ep0(sunos_dev_handle_priv_t *hpriv, sunos_dev_priv_t *dpriv)

{

	char filename[PATH_MAX + 1];

	if (hpriv->eps[0].datafd > 0) {

		return (LIBUSB_SUCCESS);

	}

	snprintf(filename, PATH_MAX, "%s/cntrl0", dpriv->ugenpath);

	usbi_dbg("opening %s", filename);

	hpriv->eps[0].datafd = open(filename, O_RDWR);

	if (hpriv->eps[0].datafd < 0) {

		return(_errno_to_libusb(errno));

	}

	snprintf(filename, PATH_MAX, "%s/cntrl0stat", dpriv->ugenpath);

	hpriv->eps[0].statfd = open(filename, O_RDONLY);

	if (hpriv->eps[0].statfd < 0) {

		close(hpriv->eps[0].datafd);

		hpriv->eps[0].datafd = -1;

		return(_errno_to_libusb(errno));

	}

	return (LIBUSB_SUCCESS);

}

static void

sunos_usb_close_all_eps(sunos_dev_handle_priv_t *hdev)

{

	int i;

	/* not close ep0 */

	for (i = 1; i < USB_MAXENDPOINTS; i++) {

		if (hdev->eps[i].datafd != -1) {

			(void) close(hdev->eps[i].datafd);

			hdev->eps[i].datafd = -1;

		}

		if (hdev->eps[i].statfd != -1) {

			(void) close(hdev->eps[i].statfd);

			hdev->eps[i].statfd = -1;

		}

	}

}

static void

sunos_usb_close_ep0(sunos_dev_handle_priv_t *hdev, sunos_dev_priv_t *dpriv)

{

	if (hdev->eps[0].datafd >= 0) {

		close(hdev->eps[0].datafd);

		close(hdev->eps[0].statfd);

		hdev->eps[0].datafd = -1;

		hdev->eps[0].statfd = -1;

	}

}

static uchar_t

sunos_usb_ep_index(uint8_t ep_addr)

{

	return ((ep_addr & LIBUSB_ENDPOINT_ADDRESS_MASK) +

	    ((ep_addr & LIBUSB_ENDPOINT_DIR_MASK) ? 16 : 0));

}

static int

sunos_find_interface(struct libusb_device_handle *hdev,

    uint8_t endpoint, uint8_t *interface)

{

	struct libusb_config_descriptor *config;

	int r;

	int iface_idx;

	r = libusb_get_active_config_descriptor(hdev->dev, &config);

	if (r < 0) {

		return (LIBUSB_ERROR_INVALID_PARAM);

	}

	for (iface_idx = 0; iface_idx < config->bNumInterfaces; iface_idx++) {

		const struct libusb_interface *iface =

		    &config->interface[iface_idx];

		int altsetting_idx;

		for (altsetting_idx = 0; altsetting_idx < iface->num_altsetting;

		    altsetting_idx++) {

			const struct libusb_interface_descriptor *altsetting =

			    &iface->altsetting[altsetting_idx];

			int ep_idx;

			for (ep_idx = 0; ep_idx < altsetting->bNumEndpoints;

			    ep_idx++) {

				const struct libusb_endpoint_descriptor *ep =

					&altsetting->endpoint[ep_idx];

				if (ep->bEndpointAddress == endpoint) {

					*interface = iface_idx;

					libusb_free_config_descriptor(config);

					return (LIBUSB_SUCCESS);

				}

			}

		}

	}

	libusb_free_config_descriptor(config);

	return (LIBUSB_ERROR_INVALID_PARAM);

}

static int

sunos_check_device_and_status_open(struct libusb_device_handle *hdl,

    uint8_t ep_addr, int ep_type)

{

	char	filename[PATH_MAX + 1], statfilename[PATH_MAX + 1];

	char	cfg_num[16], alt_num[16];

	int	fd, fdstat, mode;

	uint8_t	ifc = 0;

	uint8_t	ep_index;

	sunos_dev_handle_priv_t *hpriv;

	usbi_dbg("open ep 0x%02x", ep_addr);

	hpriv = (sunos_dev_handle_priv_t *)hdl->os_priv;

	ep_index = sunos_usb_ep_index(ep_addr);

	/* ep already opened */

	if ((hpriv->eps[ep_index].datafd > 0) &&

	    (hpriv->eps[ep_index].statfd > 0)) {

		usbi_dbg("ep 0x%02x already opened, return success",

			ep_addr);

		return (0);

	}

	if (sunos_find_interface(hdl, ep_addr, &ifc) < 0) {

		usbi_dbg("can't find interface for endpoint 0x%02x",

		    ep_addr);

		return (EACCES);

	}