1184610Salfred/* $FreeBSD: stable/10/usr.sbin/usbconfig/usbconfig.c 356408 2020-01-06 09:40:59Z hselasky $ */
2184610Salfred/*-
3201705Sthompsa * Copyright (c) 2008-2009 Hans Petter Selasky. All rights reserved.
4184610Salfred *
5184610Salfred * Redistribution and use in source and binary forms, with or without
6184610Salfred * modification, are permitted provided that the following conditions
7184610Salfred * are met:
8184610Salfred * 1. Redistributions of source code must retain the above copyright
9184610Salfred *    notice, this list of conditions and the following disclaimer.
10184610Salfred * 2. Redistributions in binary form must reproduce the above copyright
11184610Salfred *    notice, this list of conditions and the following disclaimer in the
12184610Salfred *    documentation and/or other materials provided with the distribution.
13184610Salfred *
14184610Salfred * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15184610Salfred * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16184610Salfred * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17184610Salfred * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18184610Salfred * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19184610Salfred * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20184610Salfred * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21184610Salfred * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22184610Salfred * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23184610Salfred * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24184610Salfred * SUCH DAMAGE.
25184610Salfred */
26184610Salfred
27184610Salfred#include <stdio.h>
28184610Salfred#include <stdlib.h>
29184610Salfred#include <stdint.h>
30184610Salfred#include <err.h>
31184610Salfred#include <string.h>
32184610Salfred#include <pwd.h>
33184610Salfred#include <grp.h>
34184610Salfred#include <errno.h>
35185087Salfred#include <ctype.h>
36248236Shselasky#include <sys/types.h>
37184610Salfred
38184610Salfred#include <libusb20_desc.h>
39184610Salfred#include <libusb20.h>
40184610Salfred
41184610Salfred#include "dump.h"
42184610Salfred
43184610Salfredstruct options {
44184610Salfred	const char *quirkname;
45185087Salfred	void   *buffer;
46188987Sthompsa	int template;
47184610Salfred	gid_t	gid;
48184610Salfred	uid_t	uid;
49184610Salfred	mode_t	mode;
50184610Salfred	uint32_t got_any;
51185087Salfred	struct LIBUSB20_CONTROL_SETUP_DECODED setup;
52184610Salfred	uint16_t bus;
53184610Salfred	uint16_t addr;
54184610Salfred	uint16_t iface;
55184610Salfred	uint16_t vid;
56184610Salfred	uint16_t pid;
57184610Salfred	uint16_t lo_rev;		/* inclusive */
58184610Salfred	uint16_t hi_rev;		/* inclusive */
59185087Salfred	uint8_t	string_index;
60184610Salfred	uint8_t	config_index;
61184610Salfred	uint8_t	alt_index;
62184610Salfred	uint8_t	got_list:1;
63184610Salfred	uint8_t	got_bus:1;
64184610Salfred	uint8_t	got_addr:1;
65184610Salfred	uint8_t	got_iface:1;
66184610Salfred	uint8_t	got_set_config:1;
67184610Salfred	uint8_t	got_set_alt:1;
68188987Sthompsa	uint8_t	got_set_template:1;
69188987Sthompsa	uint8_t	got_get_template:1;
70184610Salfred	uint8_t	got_suspend:1;
71184610Salfred	uint8_t	got_resume:1;
72184610Salfred	uint8_t	got_reset:1;
73184610Salfred	uint8_t	got_power_off:1;
74184610Salfred	uint8_t	got_power_save:1;
75184610Salfred	uint8_t	got_power_on:1;
76184610Salfred	uint8_t	got_dump_device_quirks:1;
77184610Salfred	uint8_t	got_dump_quirk_names:1;
78356404Shselasky	uint8_t	got_dump_all_desc:1;
79184610Salfred	uint8_t	got_dump_device_desc:1;
80184610Salfred	uint8_t	got_dump_curr_config:1;
81184610Salfred	uint8_t	got_dump_all_config:1;
82184610Salfred	uint8_t	got_dump_info:1;
83356408Shselasky  	uint8_t	got_dump_stats:1;
84188622Sthompsa	uint8_t	got_show_iface_driver:1;
85184610Salfred	uint8_t	got_remove_device_quirk:1;
86184610Salfred	uint8_t	got_add_device_quirk:1;
87201705Sthompsa	uint8_t	got_remove_quirk:1;
88201705Sthompsa	uint8_t	got_add_quirk:1;
89185087Salfred	uint8_t	got_dump_string:1;
90185087Salfred	uint8_t	got_do_request:1;
91356407Shselasky	uint8_t	got_detach_kernel_driver:1;
92184610Salfred};
93184610Salfred
94184610Salfredstruct token {
95184610Salfred	const char *name;
96184610Salfred	uint8_t	value;
97184610Salfred	uint8_t	narg;
98184610Salfred};
99184610Salfred
100184610Salfredenum {
101184610Salfred	T_UNIT,
102184610Salfred	T_ADDR,
103201705Sthompsa	T_UGEN,
104184610Salfred	T_IFACE,
105184610Salfred	T_SET_CONFIG,
106184610Salfred	T_SET_ALT,
107188987Sthompsa	T_SET_TEMPLATE,
108188987Sthompsa	T_GET_TEMPLATE,
109184610Salfred	T_ADD_DEVICE_QUIRK,
110184610Salfred	T_REMOVE_DEVICE_QUIRK,
111201705Sthompsa	T_ADD_QUIRK,
112201705Sthompsa	T_REMOVE_QUIRK,
113188622Sthompsa	T_SHOW_IFACE_DRIVER,
114356407Shselasky	T_DETACH_KERNEL_DRIVER,
115184610Salfred	T_DUMP_QUIRK_NAMES,
116184610Salfred	T_DUMP_DEVICE_QUIRKS,
117356404Shselasky	T_DUMP_ALL_DESC,
118184610Salfred	T_DUMP_DEVICE_DESC,
119184610Salfred	T_DUMP_CURR_CONFIG_DESC,
120184610Salfred	T_DUMP_ALL_CONFIG_DESC,
121185087Salfred	T_DUMP_STRING,
122184610Salfred	T_DUMP_INFO,
123356408Shselasky	T_DUMP_STATS,
124184610Salfred	T_SUSPEND,
125184610Salfred	T_RESUME,
126184610Salfred	T_POWER_OFF,
127184610Salfred	T_POWER_SAVE,
128184610Salfred	T_POWER_ON,
129184610Salfred	T_RESET,
130184610Salfred	T_LIST,
131185087Salfred	T_DO_REQUEST,
132184610Salfred};
133184610Salfred
134184610Salfredstatic struct options options;
135184610Salfred
136184610Salfredstatic const struct token token[] = {
137184610Salfred	{"-u", T_UNIT, 1},
138184610Salfred	{"-a", T_ADDR, 1},
139201705Sthompsa	{"-d", T_UGEN, 1},
140184610Salfred	{"-i", T_IFACE, 1},
141184610Salfred	{"set_config", T_SET_CONFIG, 1},
142184610Salfred	{"set_alt", T_SET_ALT, 1},
143188987Sthompsa	{"set_template", T_SET_TEMPLATE, 1},
144188987Sthompsa	{"get_template", T_GET_TEMPLATE, 0},
145184610Salfred	{"add_dev_quirk_vplh", T_ADD_DEVICE_QUIRK, 5},
146184610Salfred	{"remove_dev_quirk_vplh", T_REMOVE_DEVICE_QUIRK, 5},
147201705Sthompsa	{"add_quirk", T_ADD_QUIRK, 1},
148201705Sthompsa	{"remove_quirk", T_REMOVE_QUIRK, 1},
149356407Shselasky	{"detach_kernel_driver", T_DETACH_KERNEL_DRIVER, 0},
150184610Salfred	{"dump_quirk_names", T_DUMP_QUIRK_NAMES, 0},
151184610Salfred	{"dump_device_quirks", T_DUMP_DEVICE_QUIRKS, 0},
152356404Shselasky	{"dump_all_desc", T_DUMP_ALL_DESC, 0},
153184610Salfred	{"dump_device_desc", T_DUMP_DEVICE_DESC, 0},
154184610Salfred	{"dump_curr_config_desc", T_DUMP_CURR_CONFIG_DESC, 0},
155184610Salfred	{"dump_all_config_desc", T_DUMP_ALL_CONFIG_DESC, 0},
156185087Salfred	{"dump_string", T_DUMP_STRING, 1},
157184610Salfred	{"dump_info", T_DUMP_INFO, 0},
158356408Shselasky	{"dump_stats", T_DUMP_STATS, 0},
159188622Sthompsa	{"show_ifdrv", T_SHOW_IFACE_DRIVER, 0},
160184610Salfred	{"suspend", T_SUSPEND, 0},
161184610Salfred	{"resume", T_RESUME, 0},
162184610Salfred	{"power_off", T_POWER_OFF, 0},
163184610Salfred	{"power_save", T_POWER_SAVE, 0},
164184610Salfred	{"power_on", T_POWER_ON, 0},
165184610Salfred	{"reset", T_RESET, 0},
166184610Salfred	{"list", T_LIST, 0},
167185087Salfred	{"do_request", T_DO_REQUEST, 5},
168184610Salfred};
169184610Salfred
170184610Salfredstatic void
171184610Salfredbe_dev_remove_quirk(struct libusb20_backend *pbe,
172184610Salfred    uint16_t vid, uint16_t pid, uint16_t lorev, uint16_t hirev,
173184610Salfred    const char *str)
174184610Salfred{
175184610Salfred	struct libusb20_quirk q;
176185087Salfred	int error;
177184610Salfred
178184610Salfred	memset(&q, 0, sizeof(q));
179184610Salfred
180184610Salfred	q.vid = vid;
181184610Salfred	q.pid = pid;
182184610Salfred	q.bcdDeviceLow = lorev;
183184610Salfred	q.bcdDeviceHigh = hirev;
184184610Salfred	strlcpy(q.quirkname, str, sizeof(q.quirkname));
185184610Salfred
186185087Salfred	error = libusb20_be_remove_dev_quirk(pbe, &q);
187185087Salfred	if (error) {
188202026Sthompsa		fprintf(stderr, "Removing quirk '%s' failed, continuing.\n", str);
189184610Salfred	}
190184610Salfred	return;
191184610Salfred}
192184610Salfred
193184610Salfredstatic void
194184610Salfredbe_dev_add_quirk(struct libusb20_backend *pbe,
195184610Salfred    uint16_t vid, uint16_t pid, uint16_t lorev, uint16_t hirev,
196184610Salfred    const char *str)
197184610Salfred{
198184610Salfred	struct libusb20_quirk q;
199185087Salfred	int error;
200184610Salfred
201184610Salfred	memset(&q, 0, sizeof(q));
202184610Salfred
203184610Salfred	q.vid = vid;
204184610Salfred	q.pid = pid;
205184610Salfred	q.bcdDeviceLow = lorev;
206184610Salfred	q.bcdDeviceHigh = hirev;
207184610Salfred	strlcpy(q.quirkname, str, sizeof(q.quirkname));
208184610Salfred
209185087Salfred	error = libusb20_be_add_dev_quirk(pbe, &q);
210185087Salfred	if (error) {
211202026Sthompsa		fprintf(stderr, "Adding quirk '%s' failed, continuing.\n", str);
212184610Salfred	}
213184610Salfred	return;
214184610Salfred}
215184610Salfred
216184610Salfredstatic uint8_t
217184610Salfredget_token(const char *str, uint8_t narg)
218184610Salfred{
219184610Salfred	uint8_t n;
220184610Salfred
221184610Salfred	for (n = 0; n != (sizeof(token) / sizeof(token[0])); n++) {
222184610Salfred		if (strcasecmp(str, token[n].name) == 0) {
223184610Salfred			if (token[n].narg > narg) {
224185087Salfred				/* too few arguments */
225184610Salfred				break;
226184610Salfred			}
227184610Salfred			return (token[n].value);
228184610Salfred		}
229184610Salfred	}
230184610Salfred	return (0 - 1);
231184610Salfred}
232184610Salfred
233184610Salfredstatic uid_t
234184610Salfrednum_id(const char *name, const char *type)
235184610Salfred{
236184610Salfred	uid_t val;
237184610Salfred	char *ep;
238184610Salfred
239184610Salfred	errno = 0;
240184610Salfred	val = strtoul(name, &ep, 0);
241184610Salfred	if (errno) {
242184610Salfred		err(1, "%s", name);
243184610Salfred	}
244184610Salfred	if (*ep != '\0') {
245184610Salfred		errx(1, "%s: illegal %s name", name, type);
246184610Salfred	}
247184610Salfred	return (val);
248184610Salfred}
249184610Salfred
250189110Sthompsastatic int
251189110Sthompsaget_int(const char *s)
252184610Salfred{
253189110Sthompsa	int val;
254184610Salfred	char *ep;
255184610Salfred
256184610Salfred	errno = 0;
257189110Sthompsa	val = strtoul(s, &ep, 0);
258184610Salfred	if (errno) {
259184610Salfred		err(1, "%s", s);
260184610Salfred	}
261184610Salfred	if (*ep != '\0') {
262189110Sthompsa		errx(1, "illegal number: %s", s);
263184610Salfred	}
264184610Salfred	return val;
265184610Salfred}
266184610Salfred
267184610Salfredstatic void
268201705Sthompsaduplicate_option(const char *ptr)
269201705Sthompsa{
270202026Sthompsa	fprintf(stderr, "Syntax error: "
271201705Sthompsa	    "Duplicate option: '%s'\n", ptr);
272201705Sthompsa	exit(1);
273201705Sthompsa}
274201705Sthompsa
275201705Sthompsastatic void
276184610Salfredusage(void)
277184610Salfred{
278202026Sthompsa	fprintf(stderr, ""
279184610Salfred	    "usbconfig - configure the USB subsystem" "\n"
280184610Salfred	    "usage: usbconfig -u <busnum> -a <devaddr> -i <ifaceindex> [cmds...]" "\n"
281201705Sthompsa	    "usage: usbconfig -d [ugen]<busnum>.<devaddr> -i <ifaceindex> [cmds...]" "\n"
282184610Salfred	    "commands:" "\n"
283185087Salfred	    "  set_config <cfg_index>" "\n"
284185087Salfred	    "  set_alt <alt_index>" "\n"
285188987Sthompsa	    "  set_template <template>" "\n"
286188987Sthompsa	    "  get_template" "\n"
287184610Salfred	    "  add_dev_quirk_vplh <vid> <pid> <lo_rev> <hi_rev> <quirk>" "\n"
288184610Salfred	    "  remove_dev_quirk_vplh <vid> <pid> <lo_rev> <hi_rev> <quirk>" "\n"
289201705Sthompsa	    "  add_quirk <quirk>" "\n"
290201705Sthompsa	    "  remove_quirk <quirk>" "\n"
291356407Shselasky	    "  detach_kernel_driver" "\n"
292184610Salfred	    "  dump_quirk_names" "\n"
293184610Salfred	    "  dump_device_quirks" "\n"
294356404Shselasky	    "  dump_all_desc" "\n"
295184610Salfred	    "  dump_device_desc" "\n"
296184610Salfred	    "  dump_curr_config_desc" "\n"
297184610Salfred	    "  dump_all_config_desc" "\n"
298185087Salfred	    "  dump_string <index>" "\n"
299184610Salfred	    "  dump_info" "\n"
300356408Shselasky	    "  dump_stats" "\n"
301188622Sthompsa	    "  show_ifdrv" "\n"
302184610Salfred	    "  suspend" "\n"
303184610Salfred	    "  resume" "\n"
304184610Salfred	    "  power_off" "\n"
305184610Salfred	    "  power_save" "\n"
306184610Salfred	    "  power_on" "\n"
307184610Salfred	    "  reset" "\n"
308184610Salfred	    "  list" "\n"
309185087Salfred	    "  do_request <bmReqTyp> <bReq> <wVal> <wIdx> <wLen> <data...>" "\n"
310184610Salfred	);
311184610Salfred	exit(1);
312184610Salfred}
313184610Salfred
314184610Salfredstatic void
315184610Salfredreset_options(struct options *opt)
316184610Salfred{
317185087Salfred	if (opt->buffer)
318185087Salfred		free(opt->buffer);
319184610Salfred	memset(opt, 0, sizeof(*opt));
320184610Salfred	return;
321184610Salfred}
322184610Salfred
323184610Salfredstatic void
324184610Salfredflush_command(struct libusb20_backend *pbe, struct options *opt)
325184610Salfred{
326184610Salfred	struct libusb20_device *pdev = NULL;
327184610Salfred	uint32_t matches = 0;
328184610Salfred	uint8_t dump_any;
329184610Salfred
330184610Salfred	/* check for invalid option combinations */
331184610Salfred	if ((opt->got_suspend +
332184610Salfred	    opt->got_resume +
333184610Salfred	    opt->got_reset +
334184610Salfred	    opt->got_set_config +
335184610Salfred	    opt->got_set_alt +
336184610Salfred	    opt->got_power_save +
337184610Salfred	    opt->got_power_on +
338184610Salfred	    opt->got_power_off) > 1) {
339187168Sthompsa		err(1, "can only specify one of 'set_config', "
340184610Salfred		    "'set_alt', 'reset', 'suspend', 'resume', "
341184610Salfred		    "'power_save', 'power_on' and 'power_off' "
342184610Salfred		    "at the same time!");
343184610Salfred	}
344184610Salfred	if (opt->got_dump_quirk_names) {
345184610Salfred		opt->got_any--;
346184610Salfred		dump_be_quirk_names(pbe);
347184610Salfred	}
348184610Salfred	if (opt->got_dump_device_quirks) {
349184610Salfred		opt->got_any--;
350184610Salfred		dump_be_dev_quirks(pbe);
351184610Salfred	}
352184610Salfred	if (opt->got_remove_device_quirk) {
353184610Salfred		opt->got_any--;
354184610Salfred		be_dev_remove_quirk(pbe,
355184610Salfred		    opt->vid, opt->pid, opt->lo_rev, opt->hi_rev, opt->quirkname);
356184610Salfred	}
357184610Salfred	if (opt->got_add_device_quirk) {
358184610Salfred		opt->got_any--;
359184610Salfred		be_dev_add_quirk(pbe,
360184610Salfred		    opt->vid, opt->pid, opt->lo_rev, opt->hi_rev, opt->quirkname);
361184610Salfred	}
362188987Sthompsa	if (opt->got_set_template) {
363188987Sthompsa		opt->got_any--;
364188987Sthompsa		if (libusb20_be_set_template(pbe, opt->template)) {
365202026Sthompsa			fprintf(stderr, "Setting USB template %u failed, "
366188987Sthompsa			    "continuing.\n", opt->template);
367188987Sthompsa		}
368188987Sthompsa	}
369188987Sthompsa	if (opt->got_get_template) {
370188987Sthompsa		opt->got_any--;
371188987Sthompsa		if (libusb20_be_get_template(pbe, &opt->template))
372188987Sthompsa			printf("USB template: <unknown>\n");
373188987Sthompsa		else
374188987Sthompsa			printf("USB template: %u\n", opt->template);
375188987Sthompsa	}
376184610Salfred	if (opt->got_any == 0) {
377184610Salfred		/*
378184610Salfred		 * do not scan through all the devices if there are no valid
379184610Salfred		 * options
380184610Salfred		 */
381184610Salfred		goto done;
382184610Salfred	}
383184610Salfred	while ((pdev = libusb20_be_device_foreach(pbe, pdev))) {
384184610Salfred
385184610Salfred		if (opt->got_bus &&
386184610Salfred		    (libusb20_dev_get_bus_number(pdev) != opt->bus)) {
387184610Salfred			continue;
388184610Salfred		}
389184610Salfred		if (opt->got_addr &&
390184610Salfred		    (libusb20_dev_get_address(pdev) != opt->addr)) {
391184610Salfred			continue;
392184610Salfred		}
393184610Salfred		matches++;
394184610Salfred
395201705Sthompsa		if (opt->got_remove_quirk) {
396201705Sthompsa			struct LIBUSB20_DEVICE_DESC_DECODED *ddesc;
397201705Sthompsa
398201705Sthompsa			ddesc = libusb20_dev_get_device_desc(pdev);
399201705Sthompsa
400201705Sthompsa			be_dev_remove_quirk(pbe,
401201705Sthompsa			    ddesc->idVendor, ddesc->idProduct,
402201705Sthompsa			    ddesc->bcdDevice, ddesc->bcdDevice,
403201705Sthompsa			    opt->quirkname);
404201705Sthompsa		}
405201705Sthompsa
406201705Sthompsa		if (opt->got_add_quirk) {
407201705Sthompsa			struct LIBUSB20_DEVICE_DESC_DECODED *ddesc;
408201705Sthompsa
409201705Sthompsa			ddesc = libusb20_dev_get_device_desc(pdev);
410201705Sthompsa
411201705Sthompsa			be_dev_add_quirk(pbe,
412201705Sthompsa			    ddesc->idVendor, ddesc->idProduct,
413201705Sthompsa			    ddesc->bcdDevice, ddesc->bcdDevice,
414201705Sthompsa			    opt->quirkname);
415201705Sthompsa		}
416201705Sthompsa
417184610Salfred		if (libusb20_dev_open(pdev, 0)) {
418184610Salfred			err(1, "could not open device");
419184610Salfred		}
420185087Salfred		if (opt->got_dump_string) {
421201705Sthompsa			dump_string_by_index(pdev, opt->string_index);
422185087Salfred		}
423185087Salfred		if (opt->got_do_request) {
424185087Salfred			uint16_t actlen;
425185087Salfred			uint16_t t;
426185087Salfred
427185087Salfred			if (libusb20_dev_request_sync(pdev, &opt->setup,
428185087Salfred			    opt->buffer, &actlen, 5000 /* 5 seconds */ , 0)) {
429185087Salfred				printf("REQUEST = <ERROR>\n");
430185087Salfred			} else if (!(opt->setup.bmRequestType &
431185087Salfred			    LIBUSB20_ENDPOINT_IN)) {
432185087Salfred				printf("REQUEST = <OK>\n");
433185087Salfred			} else {
434185087Salfred				t = actlen;
435185087Salfred				printf("REQUEST = <");
436185087Salfred				for (t = 0; t != actlen; t++) {
437185087Salfred					printf("0x%02x%s",
438185087Salfred					    ((uint8_t *)opt->buffer)[t],
439185087Salfred					    (t == (actlen - 1)) ? "" : " ");
440185087Salfred				}
441185087Salfred				printf("><");
442185087Salfred				for (t = 0; t != actlen; t++) {
443185087Salfred					char c;
444185087Salfred
445185087Salfred					c = ((uint8_t *)opt->buffer)[t];
446185087Salfred					if ((c != '<') &&
447185087Salfred					    (c != '>') && isprint(c)) {
448185087Salfred						putchar(c);
449185087Salfred					}
450185087Salfred				}
451185087Salfred				printf(">\n");
452185087Salfred			}
453185087Salfred		}
454184610Salfred		if (opt->got_set_config) {
455184610Salfred			if (libusb20_dev_set_config_index(pdev,
456184610Salfred			    opt->config_index)) {
457184610Salfred				err(1, "could not set config index");
458184610Salfred			}
459184610Salfred		}
460184610Salfred		if (opt->got_set_alt) {
461185087Salfred			if (libusb20_dev_set_alt_index(pdev, opt->iface,
462185087Salfred			    opt->alt_index)) {
463184610Salfred				err(1, "could not set alternate setting");
464184610Salfred			}
465184610Salfred		}
466184610Salfred		if (opt->got_reset) {
467184610Salfred			if (libusb20_dev_reset(pdev)) {
468184610Salfred				err(1, "could not reset device");
469184610Salfred			}
470184610Salfred		}
471184610Salfred		if (opt->got_suspend) {
472185087Salfred			if (libusb20_dev_set_power_mode(pdev,
473185087Salfred			    LIBUSB20_POWER_SUSPEND)) {
474184610Salfred				err(1, "could not set suspend");
475184610Salfred			}
476184610Salfred		}
477184610Salfred		if (opt->got_resume) {
478185087Salfred			if (libusb20_dev_set_power_mode(pdev,
479185087Salfred			    LIBUSB20_POWER_RESUME)) {
480184610Salfred				err(1, "could not set resume");
481184610Salfred			}
482184610Salfred		}
483184610Salfred		if (opt->got_power_off) {
484185087Salfred			if (libusb20_dev_set_power_mode(pdev,
485185087Salfred			    LIBUSB20_POWER_OFF)) {
486184610Salfred				err(1, "could not set power OFF");
487184610Salfred			}
488184610Salfred		}
489184610Salfred		if (opt->got_power_save) {
490185087Salfred			if (libusb20_dev_set_power_mode(pdev,
491185087Salfred			    LIBUSB20_POWER_SAVE)) {
492184610Salfred				err(1, "could not set power SAVE");
493184610Salfred			}
494184610Salfred		}
495184610Salfred		if (opt->got_power_on) {
496185087Salfred			if (libusb20_dev_set_power_mode(pdev,
497185087Salfred			    LIBUSB20_POWER_ON)) {
498184610Salfred				err(1, "could not set power ON");
499184610Salfred			}
500184610Salfred		}
501356407Shselasky		if (opt->got_detach_kernel_driver) {
502356407Shselasky			if (libusb20_dev_detach_kernel_driver(pdev, opt->iface)) {
503356407Shselasky				err(1, "could not detach kernel driver");
504356407Shselasky			}
505356407Shselasky		}
506184610Salfred		dump_any =
507356404Shselasky		    (opt->got_dump_all_desc ||
508356404Shselasky		    opt->got_dump_device_desc ||
509184610Salfred		    opt->got_dump_curr_config ||
510184610Salfred		    opt->got_dump_all_config ||
511356408Shselasky		    opt->got_dump_info ||
512356408Shselasky		    opt->got_dump_stats);
513184610Salfred
514184610Salfred		if (opt->got_list || dump_any) {
515188622Sthompsa			dump_device_info(pdev,
516188622Sthompsa			    opt->got_show_iface_driver);
517184610Salfred		}
518184610Salfred		if (opt->got_dump_device_desc) {
519184610Salfred			printf("\n");
520184610Salfred			dump_device_desc(pdev);
521184610Salfred		}
522184610Salfred		if (opt->got_dump_all_config) {
523184610Salfred			printf("\n");
524184610Salfred			dump_config(pdev, 1);
525184610Salfred		} else if (opt->got_dump_curr_config) {
526184610Salfred			printf("\n");
527184610Salfred			dump_config(pdev, 0);
528356404Shselasky		} else if (opt->got_dump_all_desc) {
529356404Shselasky			printf("\n");
530356404Shselasky			dump_device_desc(pdev);
531356404Shselasky			dump_config(pdev, 1);
532184610Salfred		}
533356408Shselasky		if (opt->got_dump_stats) {
534356408Shselasky			printf("\n");
535356408Shselasky			dump_device_stats(pdev);
536356408Shselasky		}
537184610Salfred		if (dump_any) {
538184610Salfred			printf("\n");
539184610Salfred		}
540184610Salfred		if (libusb20_dev_close(pdev)) {
541184610Salfred			err(1, "could not close device");
542184610Salfred		}
543184610Salfred	}
544184610Salfred
545184610Salfred	if (matches == 0) {
546187162Sthompsa		printf("No device match or lack of permissions.\n");
547184610Salfred	}
548184610Salfreddone:
549184610Salfred	reset_options(opt);
550184610Salfred
551184610Salfred	return;
552184610Salfred}
553184610Salfred
554184610Salfredint
555184610Salfredmain(int argc, char **argv)
556184610Salfred{
557184610Salfred	struct libusb20_backend *pbe;
558184610Salfred	struct options *opt = &options;
559201705Sthompsa	const char *ptr;
560201705Sthompsa	int unit;
561201705Sthompsa	int addr;
562184610Salfred	int n;
563184610Salfred	int t;
564184610Salfred
565184610Salfred	if (argc < 1) {
566184610Salfred		usage();
567184610Salfred	}
568184610Salfred	pbe = libusb20_be_alloc_default();
569184610Salfred	if (pbe == NULL)
570184610Salfred		err(1, "could not access USB backend\n");
571184610Salfred
572184610Salfred	for (n = 1; n != argc; n++) {
573184610Salfred
574184610Salfred		/* get number of additional options */
575184610Salfred		t = (argc - n - 1);
576184610Salfred		if (t > 255)
577184610Salfred			t = 255;
578184610Salfred		switch (get_token(argv[n], t)) {
579201705Sthompsa		case T_ADD_QUIRK:
580201705Sthompsa			if (opt->got_add_quirk) {
581201705Sthompsa				flush_command(pbe, opt);
582201705Sthompsa			}
583201705Sthompsa			opt->quirkname = argv[n + 1];
584201705Sthompsa			n++;
585201705Sthompsa
586201705Sthompsa			opt->got_add_quirk = 1;
587201705Sthompsa			opt->got_any++;
588201705Sthompsa			break;
589201705Sthompsa
590201705Sthompsa		case T_REMOVE_QUIRK:
591201705Sthompsa			if (opt->got_remove_quirk) {
592201705Sthompsa				flush_command(pbe, opt);
593201705Sthompsa			}
594201705Sthompsa			opt->quirkname = argv[n + 1];
595201705Sthompsa			n++;
596201705Sthompsa
597201705Sthompsa			opt->got_remove_quirk = 1;
598201705Sthompsa			opt->got_any++;
599201705Sthompsa			break;
600201705Sthompsa
601184610Salfred		case T_ADD_DEVICE_QUIRK:
602184610Salfred			if (opt->got_add_device_quirk) {
603184610Salfred				flush_command(pbe, opt);
604184610Salfred			}
605184610Salfred			opt->vid = num_id(argv[n + 1], "Vendor ID");
606184610Salfred			opt->pid = num_id(argv[n + 2], "Product ID");
607184610Salfred			opt->lo_rev = num_id(argv[n + 3], "Low Revision");
608184610Salfred			opt->hi_rev = num_id(argv[n + 4], "High Revision");
609184610Salfred			opt->quirkname = argv[n + 5];
610184610Salfred			n += 5;
611184610Salfred
612184610Salfred			opt->got_add_device_quirk = 1;
613184610Salfred			opt->got_any++;
614184610Salfred			break;
615184610Salfred
616184610Salfred		case T_REMOVE_DEVICE_QUIRK:
617184610Salfred			if (opt->got_remove_device_quirk) {
618184610Salfred				flush_command(pbe, opt);
619184610Salfred			}
620184610Salfred			opt->vid = num_id(argv[n + 1], "Vendor ID");
621184610Salfred			opt->pid = num_id(argv[n + 2], "Product ID");
622184610Salfred			opt->lo_rev = num_id(argv[n + 3], "Low Revision");
623184610Salfred			opt->hi_rev = num_id(argv[n + 4], "High Revision");
624184610Salfred			opt->quirkname = argv[n + 5];
625184610Salfred			n += 5;
626184610Salfred			opt->got_remove_device_quirk = 1;
627184610Salfred			opt->got_any++;
628184610Salfred			break;
629184610Salfred
630356407Shselasky		case T_DETACH_KERNEL_DRIVER:
631356407Shselasky			if (opt->got_detach_kernel_driver)
632356407Shselasky				duplicate_option(argv[n]);
633356407Shselasky			opt->got_detach_kernel_driver = 1;
634356407Shselasky			opt->got_any++;
635356407Shselasky			break;
636356407Shselasky
637184610Salfred		case T_DUMP_QUIRK_NAMES:
638201705Sthompsa			if (opt->got_dump_quirk_names)
639201705Sthompsa				duplicate_option(argv[n]);
640184610Salfred			opt->got_dump_quirk_names = 1;
641184610Salfred			opt->got_any++;
642184610Salfred			break;
643184610Salfred
644184610Salfred		case T_DUMP_DEVICE_QUIRKS:
645201705Sthompsa			if (opt->got_dump_device_quirks)
646201705Sthompsa				duplicate_option(argv[n]);
647184610Salfred			opt->got_dump_device_quirks = 1;
648184610Salfred			opt->got_any++;
649184610Salfred			break;
650184610Salfred
651188622Sthompsa		case T_SHOW_IFACE_DRIVER:
652188622Sthompsa			opt->got_show_iface_driver = 1;
653188622Sthompsa			break;
654188622Sthompsa
655201705Sthompsa		case T_UGEN:
656201705Sthompsa			if (opt->got_any) {
657201705Sthompsa				/* allow multiple commands on the same line */
658201705Sthompsa				flush_command(pbe, opt);
659201705Sthompsa			}
660201705Sthompsa			ptr = argv[n + 1];
661201705Sthompsa
662201705Sthompsa			if ((ptr[0] == 'u') &&
663201705Sthompsa			    (ptr[1] == 'g') &&
664201705Sthompsa			    (ptr[2] == 'e') &&
665201705Sthompsa			    (ptr[3] == 'n'))
666201705Sthompsa				ptr += 4;
667201705Sthompsa
668201705Sthompsa			if ((sscanf(ptr, "%d.%d",
669201705Sthompsa			    &unit, &addr) != 2) ||
670201705Sthompsa			    (unit < 0) || (unit > 65535) ||
671201705Sthompsa			    (addr < 0) || (addr > 65535)) {
672201705Sthompsa				errx(1, "cannot "
673201705Sthompsa				    "parse '%s'", argv[n + 1]);
674201705Sthompsa			}
675201705Sthompsa			opt->bus = unit;
676201705Sthompsa			opt->addr = addr;
677201705Sthompsa			opt->got_bus = 1;
678214271Shselasky			opt->got_addr = 1;
679201705Sthompsa			n++;
680201705Sthompsa			break;
681201705Sthompsa
682184610Salfred		case T_UNIT:
683184610Salfred			if (opt->got_any) {
684184610Salfred				/* allow multiple commands on the same line */
685184610Salfred				flush_command(pbe, opt);
686184610Salfred			}
687184610Salfred			opt->bus = num_id(argv[n + 1], "busnum");
688184610Salfred			opt->got_bus = 1;
689184610Salfred			n++;
690184610Salfred			break;
691184610Salfred		case T_ADDR:
692184610Salfred			opt->addr = num_id(argv[n + 1], "addr");
693184610Salfred			opt->got_addr = 1;
694184610Salfred			n++;
695184610Salfred			break;
696184610Salfred		case T_IFACE:
697184610Salfred			opt->iface = num_id(argv[n + 1], "iface");
698184610Salfred			opt->got_iface = 1;
699184610Salfred			n++;
700184610Salfred			break;
701184610Salfred		case T_SET_CONFIG:
702201705Sthompsa			if (opt->got_set_config)
703201705Sthompsa				duplicate_option(argv[n]);
704185087Salfred			opt->config_index = num_id(argv[n + 1], "cfg_index");
705184610Salfred			opt->got_set_config = 1;
706184610Salfred			opt->got_any++;
707184610Salfred			n++;
708184610Salfred			break;
709184610Salfred		case T_SET_ALT:
710201705Sthompsa			if (opt->got_set_alt)
711201705Sthompsa				duplicate_option(argv[n]);
712185087Salfred			opt->alt_index = num_id(argv[n + 1], "cfg_index");
713184610Salfred			opt->got_set_alt = 1;
714184610Salfred			opt->got_any++;
715184610Salfred			n++;
716184610Salfred			break;
717188987Sthompsa		case T_SET_TEMPLATE:
718201705Sthompsa			if (opt->got_set_template)
719201705Sthompsa				duplicate_option(argv[n]);
720189110Sthompsa			opt->template = get_int(argv[n + 1]);
721188987Sthompsa			opt->got_set_template = 1;
722188987Sthompsa			opt->got_any++;
723188987Sthompsa			n++;
724188987Sthompsa			break;
725188987Sthompsa		case T_GET_TEMPLATE:
726201705Sthompsa			if (opt->got_get_template)
727201705Sthompsa				duplicate_option(argv[n]);
728188987Sthompsa			opt->got_get_template = 1;
729188987Sthompsa			opt->got_any++;
730188987Sthompsa			break;
731356404Shselasky		case T_DUMP_ALL_DESC:
732356404Shselasky			if (opt->got_dump_all_desc)
733356404Shselasky				duplicate_option(argv[n]);
734356404Shselasky			opt->got_dump_all_desc = 1;
735356404Shselasky			opt->got_any++;
736356404Shselasky			break;
737184610Salfred		case T_DUMP_DEVICE_DESC:
738201705Sthompsa			if (opt->got_dump_device_desc)
739201705Sthompsa				duplicate_option(argv[n]);
740184610Salfred			opt->got_dump_device_desc = 1;
741184610Salfred			opt->got_any++;
742184610Salfred			break;
743184610Salfred		case T_DUMP_CURR_CONFIG_DESC:
744201705Sthompsa			if (opt->got_dump_curr_config)
745201705Sthompsa				duplicate_option(argv[n]);
746184610Salfred			opt->got_dump_curr_config = 1;
747184610Salfred			opt->got_any++;
748184610Salfred			break;
749184610Salfred		case T_DUMP_ALL_CONFIG_DESC:
750201705Sthompsa			if (opt->got_dump_all_config)
751201705Sthompsa				duplicate_option(argv[n]);
752184610Salfred			opt->got_dump_all_config = 1;
753184610Salfred			opt->got_any++;
754184610Salfred			break;
755184610Salfred		case T_DUMP_INFO:
756201705Sthompsa			if (opt->got_dump_info)
757201705Sthompsa				duplicate_option(argv[n]);
758184610Salfred			opt->got_dump_info = 1;
759184610Salfred			opt->got_any++;
760184610Salfred			break;
761356408Shselasky		case T_DUMP_STATS:
762356408Shselasky			if (opt->got_dump_stats)
763356408Shselasky				duplicate_option(argv[n]);
764356408Shselasky			opt->got_dump_stats = 1;
765356408Shselasky			opt->got_any++;
766356408Shselasky			break;
767185087Salfred		case T_DUMP_STRING:
768201705Sthompsa			if (opt->got_dump_string)
769201705Sthompsa				duplicate_option(argv[n]);
770185087Salfred			opt->string_index = num_id(argv[n + 1], "str_index");
771185087Salfred			opt->got_dump_string = 1;
772185087Salfred			opt->got_any++;
773185087Salfred			n++;
774185087Salfred			break;
775184610Salfred		case T_SUSPEND:
776201705Sthompsa			if (opt->got_suspend)
777201705Sthompsa				duplicate_option(argv[n]);
778184610Salfred			opt->got_suspend = 1;
779184610Salfred			opt->got_any++;
780184610Salfred			break;
781184610Salfred		case T_RESUME:
782201705Sthompsa			if (opt->got_resume)
783201705Sthompsa				duplicate_option(argv[n]);
784184610Salfred			opt->got_resume = 1;
785184610Salfred			opt->got_any++;
786184610Salfred			break;
787184610Salfred		case T_POWER_OFF:
788201705Sthompsa			if (opt->got_power_off)
789201705Sthompsa				duplicate_option(argv[n]);
790184610Salfred			opt->got_power_off = 1;
791184610Salfred			opt->got_any++;
792184610Salfred			break;
793184610Salfred		case T_POWER_SAVE:
794201705Sthompsa			if (opt->got_power_save)
795201705Sthompsa				duplicate_option(argv[n]);
796184610Salfred			opt->got_power_save = 1;
797184610Salfred			opt->got_any++;
798184610Salfred			break;
799184610Salfred		case T_POWER_ON:
800201705Sthompsa			if (opt->got_power_on)
801201705Sthompsa				duplicate_option(argv[n]);
802184610Salfred			opt->got_power_on = 1;
803184610Salfred			opt->got_any++;
804184610Salfred			break;
805184610Salfred		case T_RESET:
806201705Sthompsa			if (opt->got_reset)
807201705Sthompsa				duplicate_option(argv[n]);
808184610Salfred			opt->got_reset = 1;
809184610Salfred			opt->got_any++;
810184610Salfred			break;
811184610Salfred		case T_LIST:
812201705Sthompsa			if (opt->got_list)
813201705Sthompsa				duplicate_option(argv[n]);
814184610Salfred			opt->got_list = 1;
815184610Salfred			opt->got_any++;
816184610Salfred			break;
817185087Salfred		case T_DO_REQUEST:
818201705Sthompsa			if (opt->got_do_request)
819201705Sthompsa				duplicate_option(argv[n]);
820185087Salfred			LIBUSB20_INIT(LIBUSB20_CONTROL_SETUP, &opt->setup);
821185087Salfred			opt->setup.bmRequestType = num_id(argv[n + 1], "bmReqTyp");
822185087Salfred			opt->setup.bRequest = num_id(argv[n + 2], "bReq");
823185087Salfred			opt->setup.wValue = num_id(argv[n + 3], "wVal");
824185087Salfred			opt->setup.wIndex = num_id(argv[n + 4], "wIndex");
825185087Salfred			opt->setup.wLength = num_id(argv[n + 5], "wLen");
826185087Salfred			if (opt->setup.wLength != 0) {
827185087Salfred				opt->buffer = malloc(opt->setup.wLength);
828185087Salfred			} else {
829185087Salfred				opt->buffer = NULL;
830185087Salfred			}
831185087Salfred
832185087Salfred			n += 5;
833185087Salfred
834185087Salfred			if (!(opt->setup.bmRequestType &
835185087Salfred			    LIBUSB20_ENDPOINT_IN)) {
836185087Salfred				/* copy in data */
837185087Salfred				t = (argc - n - 1);
838185087Salfred				if (t < opt->setup.wLength) {
839185087Salfred					err(1, "request data missing");
840185087Salfred				}
841185087Salfred				t = opt->setup.wLength;
842185087Salfred				while (t--) {
843185087Salfred					((uint8_t *)opt->buffer)[t] =
844185087Salfred					    num_id(argv[n + t + 1], "req_data");
845185087Salfred				}
846185087Salfred				n += opt->setup.wLength;
847185087Salfred			}
848185087Salfred			opt->got_do_request = 1;
849185087Salfred			opt->got_any++;
850185087Salfred			break;
851184610Salfred		default:
852356405Shselasky			if (n == 1) {
853356405Shselasky				ptr = argv[n];
854356405Shselasky
855356405Shselasky				if ((ptr[0] == 'u') &&
856356405Shselasky				    (ptr[1] == 'g') &&
857356405Shselasky				    (ptr[2] == 'e') &&
858356405Shselasky				    (ptr[3] == 'n'))
859356405Shselasky					ptr += 4;
860356405Shselasky
861356405Shselasky				if ((sscanf(ptr, "%d.%d",
862356405Shselasky				    &unit, &addr) != 2) ||
863356405Shselasky				    (unit < 0) || (unit > 65535) ||
864356405Shselasky				    (addr < 0) || (addr > 65535)) {
865356405Shselasky					usage();
866356405Shselasky					break;
867356405Shselasky				}
868356405Shselasky
869356405Shselasky				opt->bus = unit;
870356405Shselasky				opt->addr = addr;
871356405Shselasky				opt->got_bus = 1;
872356405Shselasky				opt->got_addr = 1;
873356405Shselasky				break;
874356405Shselasky			}
875184610Salfred			usage();
876184610Salfred			break;
877184610Salfred		}
878184610Salfred	}
879184610Salfred	if (opt->got_any) {
880184610Salfred		/* flush out last command */
881184610Salfred		flush_command(pbe, opt);
882184610Salfred	} else {
883184610Salfred		/* list all the devices */
884184610Salfred		opt->got_list = 1;
885184610Salfred		opt->got_any++;
886184610Salfred		flush_command(pbe, opt);
887184610Salfred	}
888184610Salfred	/* release data */
889184610Salfred	libusb20_be_free(pbe);
890184610Salfred
891184610Salfred	return (0);
892184610Salfred}
893