usbconfig.c revision 202026
1184610Salfred/* $FreeBSD: head/usr.sbin/usbconfig/usbconfig.c 202026 2010-01-10 19:21:23Z thompsa $ */
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>
36184610Salfred
37184610Salfred#include <libusb20_desc.h>
38184610Salfred#include <libusb20.h>
39184610Salfred
40184610Salfred#include "dump.h"
41184610Salfred
42184610Salfredstruct options {
43184610Salfred	const char *quirkname;
44185087Salfred	void   *buffer;
45188987Sthompsa	int template;
46184610Salfred	gid_t	gid;
47184610Salfred	uid_t	uid;
48184610Salfred	mode_t	mode;
49184610Salfred	uint32_t got_any;
50185087Salfred	struct LIBUSB20_CONTROL_SETUP_DECODED setup;
51184610Salfred	uint16_t bus;
52184610Salfred	uint16_t addr;
53184610Salfred	uint16_t iface;
54184610Salfred	uint16_t vid;
55184610Salfred	uint16_t pid;
56184610Salfred	uint16_t lo_rev;		/* inclusive */
57184610Salfred	uint16_t hi_rev;		/* inclusive */
58185087Salfred	uint8_t	string_index;
59184610Salfred	uint8_t	config_index;
60184610Salfred	uint8_t	alt_index;
61184610Salfred	uint8_t	got_list:1;
62184610Salfred	uint8_t	got_bus:1;
63184610Salfred	uint8_t	got_addr:1;
64184610Salfred	uint8_t	got_iface:1;
65184610Salfred	uint8_t	got_set_config:1;
66184610Salfred	uint8_t	got_set_alt:1;
67188987Sthompsa	uint8_t	got_set_template:1;
68188987Sthompsa	uint8_t	got_get_template:1;
69184610Salfred	uint8_t	got_suspend:1;
70184610Salfred	uint8_t	got_resume:1;
71184610Salfred	uint8_t	got_reset:1;
72184610Salfred	uint8_t	got_power_off:1;
73184610Salfred	uint8_t	got_power_save:1;
74184610Salfred	uint8_t	got_power_on:1;
75184610Salfred	uint8_t	got_dump_device_quirks:1;
76184610Salfred	uint8_t	got_dump_quirk_names:1;
77184610Salfred	uint8_t	got_dump_device_desc:1;
78184610Salfred	uint8_t	got_dump_curr_config:1;
79184610Salfred	uint8_t	got_dump_all_config:1;
80184610Salfred	uint8_t	got_dump_info:1;
81188622Sthompsa	uint8_t	got_show_iface_driver:1;
82184610Salfred	uint8_t	got_remove_device_quirk:1;
83184610Salfred	uint8_t	got_add_device_quirk:1;
84201705Sthompsa	uint8_t	got_remove_quirk:1;
85201705Sthompsa	uint8_t	got_add_quirk:1;
86185087Salfred	uint8_t	got_dump_string:1;
87185087Salfred	uint8_t	got_do_request:1;
88184610Salfred};
89184610Salfred
90184610Salfredstruct token {
91184610Salfred	const char *name;
92184610Salfred	uint8_t	value;
93184610Salfred	uint8_t	narg;
94184610Salfred};
95184610Salfred
96184610Salfredenum {
97184610Salfred	T_UNIT,
98184610Salfred	T_ADDR,
99201705Sthompsa	T_UGEN,
100184610Salfred	T_IFACE,
101184610Salfred	T_SET_CONFIG,
102184610Salfred	T_SET_ALT,
103188987Sthompsa	T_SET_TEMPLATE,
104188987Sthompsa	T_GET_TEMPLATE,
105184610Salfred	T_ADD_DEVICE_QUIRK,
106184610Salfred	T_REMOVE_DEVICE_QUIRK,
107201705Sthompsa	T_ADD_QUIRK,
108201705Sthompsa	T_REMOVE_QUIRK,
109188622Sthompsa	T_SHOW_IFACE_DRIVER,
110184610Salfred	T_DUMP_QUIRK_NAMES,
111184610Salfred	T_DUMP_DEVICE_QUIRKS,
112184610Salfred	T_DUMP_DEVICE_DESC,
113184610Salfred	T_DUMP_CURR_CONFIG_DESC,
114184610Salfred	T_DUMP_ALL_CONFIG_DESC,
115185087Salfred	T_DUMP_STRING,
116184610Salfred	T_DUMP_INFO,
117184610Salfred	T_SUSPEND,
118184610Salfred	T_RESUME,
119184610Salfred	T_POWER_OFF,
120184610Salfred	T_POWER_SAVE,
121184610Salfred	T_POWER_ON,
122184610Salfred	T_RESET,
123184610Salfred	T_LIST,
124185087Salfred	T_DO_REQUEST,
125184610Salfred};
126184610Salfred
127184610Salfredstatic struct options options;
128184610Salfred
129184610Salfredstatic const struct token token[] = {
130184610Salfred	{"-u", T_UNIT, 1},
131184610Salfred	{"-a", T_ADDR, 1},
132201705Sthompsa	{"-d", T_UGEN, 1},
133184610Salfred	{"-i", T_IFACE, 1},
134184610Salfred	{"set_config", T_SET_CONFIG, 1},
135184610Salfred	{"set_alt", T_SET_ALT, 1},
136188987Sthompsa	{"set_template", T_SET_TEMPLATE, 1},
137188987Sthompsa	{"get_template", T_GET_TEMPLATE, 0},
138184610Salfred	{"add_dev_quirk_vplh", T_ADD_DEVICE_QUIRK, 5},
139184610Salfred	{"remove_dev_quirk_vplh", T_REMOVE_DEVICE_QUIRK, 5},
140201705Sthompsa	{"add_quirk", T_ADD_QUIRK, 1},
141201705Sthompsa	{"remove_quirk", T_REMOVE_QUIRK, 1},
142184610Salfred	{"dump_quirk_names", T_DUMP_QUIRK_NAMES, 0},
143184610Salfred	{"dump_device_quirks", T_DUMP_DEVICE_QUIRKS, 0},
144184610Salfred	{"dump_device_desc", T_DUMP_DEVICE_DESC, 0},
145184610Salfred	{"dump_curr_config_desc", T_DUMP_CURR_CONFIG_DESC, 0},
146184610Salfred	{"dump_all_config_desc", T_DUMP_ALL_CONFIG_DESC, 0},
147185087Salfred	{"dump_string", T_DUMP_STRING, 1},
148184610Salfred	{"dump_info", T_DUMP_INFO, 0},
149188622Sthompsa	{"show_ifdrv", T_SHOW_IFACE_DRIVER, 0},
150184610Salfred	{"suspend", T_SUSPEND, 0},
151184610Salfred	{"resume", T_RESUME, 0},
152184610Salfred	{"power_off", T_POWER_OFF, 0},
153184610Salfred	{"power_save", T_POWER_SAVE, 0},
154184610Salfred	{"power_on", T_POWER_ON, 0},
155184610Salfred	{"reset", T_RESET, 0},
156184610Salfred	{"list", T_LIST, 0},
157185087Salfred	{"do_request", T_DO_REQUEST, 5},
158184610Salfred};
159184610Salfred
160184610Salfredstatic void
161184610Salfredbe_dev_remove_quirk(struct libusb20_backend *pbe,
162184610Salfred    uint16_t vid, uint16_t pid, uint16_t lorev, uint16_t hirev,
163184610Salfred    const char *str)
164184610Salfred{
165184610Salfred	struct libusb20_quirk q;
166185087Salfred	int error;
167184610Salfred
168184610Salfred	memset(&q, 0, sizeof(q));
169184610Salfred
170184610Salfred	q.vid = vid;
171184610Salfred	q.pid = pid;
172184610Salfred	q.bcdDeviceLow = lorev;
173184610Salfred	q.bcdDeviceHigh = hirev;
174184610Salfred	strlcpy(q.quirkname, str, sizeof(q.quirkname));
175184610Salfred
176185087Salfred	error = libusb20_be_remove_dev_quirk(pbe, &q);
177185087Salfred	if (error) {
178202026Sthompsa		fprintf(stderr, "Removing quirk '%s' failed, continuing.\n", str);
179184610Salfred	}
180184610Salfred	return;
181184610Salfred}
182184610Salfred
183184610Salfredstatic void
184184610Salfredbe_dev_add_quirk(struct libusb20_backend *pbe,
185184610Salfred    uint16_t vid, uint16_t pid, uint16_t lorev, uint16_t hirev,
186184610Salfred    const char *str)
187184610Salfred{
188184610Salfred	struct libusb20_quirk q;
189185087Salfred	int error;
190184610Salfred
191184610Salfred	memset(&q, 0, sizeof(q));
192184610Salfred
193184610Salfred	q.vid = vid;
194184610Salfred	q.pid = pid;
195184610Salfred	q.bcdDeviceLow = lorev;
196184610Salfred	q.bcdDeviceHigh = hirev;
197184610Salfred	strlcpy(q.quirkname, str, sizeof(q.quirkname));
198184610Salfred
199185087Salfred	error = libusb20_be_add_dev_quirk(pbe, &q);
200185087Salfred	if (error) {
201202026Sthompsa		fprintf(stderr, "Adding quirk '%s' failed, continuing.\n", str);
202184610Salfred	}
203184610Salfred	return;
204184610Salfred}
205184610Salfred
206184610Salfredstatic uint8_t
207184610Salfredget_token(const char *str, uint8_t narg)
208184610Salfred{
209184610Salfred	uint8_t n;
210184610Salfred
211184610Salfred	for (n = 0; n != (sizeof(token) / sizeof(token[0])); n++) {
212184610Salfred		if (strcasecmp(str, token[n].name) == 0) {
213184610Salfred			if (token[n].narg > narg) {
214185087Salfred				/* too few arguments */
215184610Salfred				break;
216184610Salfred			}
217184610Salfred			return (token[n].value);
218184610Salfred		}
219184610Salfred	}
220184610Salfred	return (0 - 1);
221184610Salfred}
222184610Salfred
223184610Salfredstatic uid_t
224184610Salfrednum_id(const char *name, const char *type)
225184610Salfred{
226184610Salfred	uid_t val;
227184610Salfred	char *ep;
228184610Salfred
229184610Salfred	errno = 0;
230184610Salfred	val = strtoul(name, &ep, 0);
231184610Salfred	if (errno) {
232184610Salfred		err(1, "%s", name);
233184610Salfred	}
234184610Salfred	if (*ep != '\0') {
235184610Salfred		errx(1, "%s: illegal %s name", name, type);
236184610Salfred	}
237184610Salfred	return (val);
238184610Salfred}
239184610Salfred
240189110Sthompsastatic int
241189110Sthompsaget_int(const char *s)
242184610Salfred{
243189110Sthompsa	int val;
244184610Salfred	char *ep;
245184610Salfred
246184610Salfred	errno = 0;
247189110Sthompsa	val = strtoul(s, &ep, 0);
248184610Salfred	if (errno) {
249184610Salfred		err(1, "%s", s);
250184610Salfred	}
251184610Salfred	if (*ep != '\0') {
252189110Sthompsa		errx(1, "illegal number: %s", s);
253184610Salfred	}
254184610Salfred	return val;
255184610Salfred}
256184610Salfred
257184610Salfredstatic void
258201705Sthompsaduplicate_option(const char *ptr)
259201705Sthompsa{
260202026Sthompsa	fprintf(stderr, "Syntax error: "
261201705Sthompsa	    "Duplicate option: '%s'\n", ptr);
262201705Sthompsa	exit(1);
263201705Sthompsa}
264201705Sthompsa
265201705Sthompsastatic void
266184610Salfredusage(void)
267184610Salfred{
268202026Sthompsa	fprintf(stderr, ""
269184610Salfred	    "usbconfig - configure the USB subsystem" "\n"
270184610Salfred	    "usage: usbconfig -u <busnum> -a <devaddr> -i <ifaceindex> [cmds...]" "\n"
271201705Sthompsa	    "usage: usbconfig -d [ugen]<busnum>.<devaddr> -i <ifaceindex> [cmds...]" "\n"
272184610Salfred	    "commands:" "\n"
273185087Salfred	    "  set_config <cfg_index>" "\n"
274185087Salfred	    "  set_alt <alt_index>" "\n"
275188987Sthompsa	    "  set_template <template>" "\n"
276188987Sthompsa	    "  get_template" "\n"
277184610Salfred	    "  add_dev_quirk_vplh <vid> <pid> <lo_rev> <hi_rev> <quirk>" "\n"
278184610Salfred	    "  remove_dev_quirk_vplh <vid> <pid> <lo_rev> <hi_rev> <quirk>" "\n"
279201705Sthompsa	    "  add_quirk <quirk>" "\n"
280201705Sthompsa	    "  remove_quirk <quirk>" "\n"
281184610Salfred	    "  dump_quirk_names" "\n"
282184610Salfred	    "  dump_device_quirks" "\n"
283184610Salfred	    "  dump_device_desc" "\n"
284184610Salfred	    "  dump_curr_config_desc" "\n"
285184610Salfred	    "  dump_all_config_desc" "\n"
286185087Salfred	    "  dump_string <index>" "\n"
287184610Salfred	    "  dump_info" "\n"
288188622Sthompsa	    "  show_ifdrv" "\n"
289184610Salfred	    "  suspend" "\n"
290184610Salfred	    "  resume" "\n"
291184610Salfred	    "  power_off" "\n"
292184610Salfred	    "  power_save" "\n"
293184610Salfred	    "  power_on" "\n"
294184610Salfred	    "  reset" "\n"
295184610Salfred	    "  list" "\n"
296185087Salfred	    "  do_request <bmReqTyp> <bReq> <wVal> <wIdx> <wLen> <data...>" "\n"
297184610Salfred	);
298184610Salfred	exit(1);
299184610Salfred}
300184610Salfred
301184610Salfredstatic void
302184610Salfredreset_options(struct options *opt)
303184610Salfred{
304185087Salfred	if (opt->buffer)
305185087Salfred		free(opt->buffer);
306184610Salfred	memset(opt, 0, sizeof(*opt));
307184610Salfred	return;
308184610Salfred}
309184610Salfred
310184610Salfredstatic void
311184610Salfredflush_command(struct libusb20_backend *pbe, struct options *opt)
312184610Salfred{
313184610Salfred	struct libusb20_device *pdev = NULL;
314184610Salfred	uint32_t matches = 0;
315184610Salfred	uint8_t dump_any;
316184610Salfred
317184610Salfred	/* check for invalid option combinations */
318184610Salfred	if ((opt->got_suspend +
319184610Salfred	    opt->got_resume +
320184610Salfred	    opt->got_reset +
321184610Salfred	    opt->got_set_config +
322184610Salfred	    opt->got_set_alt +
323184610Salfred	    opt->got_power_save +
324184610Salfred	    opt->got_power_on +
325184610Salfred	    opt->got_power_off) > 1) {
326187168Sthompsa		err(1, "can only specify one of 'set_config', "
327184610Salfred		    "'set_alt', 'reset', 'suspend', 'resume', "
328184610Salfred		    "'power_save', 'power_on' and 'power_off' "
329184610Salfred		    "at the same time!");
330184610Salfred	}
331184610Salfred	if (opt->got_dump_quirk_names) {
332184610Salfred		opt->got_any--;
333184610Salfred		dump_be_quirk_names(pbe);
334184610Salfred	}
335184610Salfred	if (opt->got_dump_device_quirks) {
336184610Salfred		opt->got_any--;
337184610Salfred		dump_be_dev_quirks(pbe);
338184610Salfred	}
339184610Salfred	if (opt->got_remove_device_quirk) {
340184610Salfred		opt->got_any--;
341184610Salfred		be_dev_remove_quirk(pbe,
342184610Salfred		    opt->vid, opt->pid, opt->lo_rev, opt->hi_rev, opt->quirkname);
343184610Salfred	}
344184610Salfred	if (opt->got_add_device_quirk) {
345184610Salfred		opt->got_any--;
346184610Salfred		be_dev_add_quirk(pbe,
347184610Salfred		    opt->vid, opt->pid, opt->lo_rev, opt->hi_rev, opt->quirkname);
348184610Salfred	}
349188987Sthompsa	if (opt->got_set_template) {
350188987Sthompsa		opt->got_any--;
351188987Sthompsa		if (libusb20_be_set_template(pbe, opt->template)) {
352202026Sthompsa			fprintf(stderr, "Setting USB template %u failed, "
353188987Sthompsa			    "continuing.\n", opt->template);
354188987Sthompsa		}
355188987Sthompsa	}
356188987Sthompsa	if (opt->got_get_template) {
357188987Sthompsa		opt->got_any--;
358188987Sthompsa		if (libusb20_be_get_template(pbe, &opt->template))
359188987Sthompsa			printf("USB template: <unknown>\n");
360188987Sthompsa		else
361188987Sthompsa			printf("USB template: %u\n", opt->template);
362188987Sthompsa	}
363184610Salfred	if (opt->got_any == 0) {
364184610Salfred		/*
365184610Salfred		 * do not scan through all the devices if there are no valid
366184610Salfred		 * options
367184610Salfred		 */
368184610Salfred		goto done;
369184610Salfred	}
370184610Salfred	while ((pdev = libusb20_be_device_foreach(pbe, pdev))) {
371184610Salfred
372184610Salfred		if (opt->got_bus &&
373184610Salfred		    (libusb20_dev_get_bus_number(pdev) != opt->bus)) {
374184610Salfred			continue;
375184610Salfred		}
376184610Salfred		if (opt->got_addr &&
377184610Salfred		    (libusb20_dev_get_address(pdev) != opt->addr)) {
378184610Salfred			continue;
379184610Salfred		}
380184610Salfred		matches++;
381184610Salfred
382201705Sthompsa		if (opt->got_remove_quirk) {
383201705Sthompsa			struct LIBUSB20_DEVICE_DESC_DECODED *ddesc;
384201705Sthompsa
385201705Sthompsa			ddesc = libusb20_dev_get_device_desc(pdev);
386201705Sthompsa
387201705Sthompsa			be_dev_remove_quirk(pbe,
388201705Sthompsa			    ddesc->idVendor, ddesc->idProduct,
389201705Sthompsa			    ddesc->bcdDevice, ddesc->bcdDevice,
390201705Sthompsa			    opt->quirkname);
391201705Sthompsa		}
392201705Sthompsa
393201705Sthompsa		if (opt->got_add_quirk) {
394201705Sthompsa			struct LIBUSB20_DEVICE_DESC_DECODED *ddesc;
395201705Sthompsa
396201705Sthompsa			ddesc = libusb20_dev_get_device_desc(pdev);
397201705Sthompsa
398201705Sthompsa			be_dev_add_quirk(pbe,
399201705Sthompsa			    ddesc->idVendor, ddesc->idProduct,
400201705Sthompsa			    ddesc->bcdDevice, ddesc->bcdDevice,
401201705Sthompsa			    opt->quirkname);
402201705Sthompsa		}
403201705Sthompsa
404184610Salfred		if (libusb20_dev_open(pdev, 0)) {
405184610Salfred			err(1, "could not open device");
406184610Salfred		}
407185087Salfred		if (opt->got_dump_string) {
408201705Sthompsa			dump_string_by_index(pdev, opt->string_index);
409185087Salfred		}
410185087Salfred		if (opt->got_do_request) {
411185087Salfred			uint16_t actlen;
412185087Salfred			uint16_t t;
413185087Salfred
414185087Salfred			if (libusb20_dev_request_sync(pdev, &opt->setup,
415185087Salfred			    opt->buffer, &actlen, 5000 /* 5 seconds */ , 0)) {
416185087Salfred				printf("REQUEST = <ERROR>\n");
417185087Salfred			} else if (!(opt->setup.bmRequestType &
418185087Salfred			    LIBUSB20_ENDPOINT_IN)) {
419185087Salfred				printf("REQUEST = <OK>\n");
420185087Salfred			} else {
421185087Salfred				t = actlen;
422185087Salfred				printf("REQUEST = <");
423185087Salfred				for (t = 0; t != actlen; t++) {
424185087Salfred					printf("0x%02x%s",
425185087Salfred					    ((uint8_t *)opt->buffer)[t],
426185087Salfred					    (t == (actlen - 1)) ? "" : " ");
427185087Salfred				}
428185087Salfred				printf("><");
429185087Salfred				for (t = 0; t != actlen; t++) {
430185087Salfred					char c;
431185087Salfred
432185087Salfred					c = ((uint8_t *)opt->buffer)[t];
433185087Salfred					if ((c != '<') &&
434185087Salfred					    (c != '>') && isprint(c)) {
435185087Salfred						putchar(c);
436185087Salfred					}
437185087Salfred				}
438185087Salfred				printf(">\n");
439185087Salfred			}
440185087Salfred		}
441184610Salfred		if (opt->got_set_config) {
442184610Salfred			if (libusb20_dev_set_config_index(pdev,
443184610Salfred			    opt->config_index)) {
444184610Salfred				err(1, "could not set config index");
445184610Salfred			}
446184610Salfred		}
447184610Salfred		if (opt->got_set_alt) {
448185087Salfred			if (libusb20_dev_set_alt_index(pdev, opt->iface,
449185087Salfred			    opt->alt_index)) {
450184610Salfred				err(1, "could not set alternate setting");
451184610Salfred			}
452184610Salfred		}
453184610Salfred		if (opt->got_reset) {
454184610Salfred			if (libusb20_dev_reset(pdev)) {
455184610Salfred				err(1, "could not reset device");
456184610Salfred			}
457184610Salfred		}
458184610Salfred		if (opt->got_suspend) {
459185087Salfred			if (libusb20_dev_set_power_mode(pdev,
460185087Salfred			    LIBUSB20_POWER_SUSPEND)) {
461184610Salfred				err(1, "could not set suspend");
462184610Salfred			}
463184610Salfred		}
464184610Salfred		if (opt->got_resume) {
465185087Salfred			if (libusb20_dev_set_power_mode(pdev,
466185087Salfred			    LIBUSB20_POWER_RESUME)) {
467184610Salfred				err(1, "could not set resume");
468184610Salfred			}
469184610Salfred		}
470184610Salfred		if (opt->got_power_off) {
471185087Salfred			if (libusb20_dev_set_power_mode(pdev,
472185087Salfred			    LIBUSB20_POWER_OFF)) {
473184610Salfred				err(1, "could not set power OFF");
474184610Salfred			}
475184610Salfred		}
476184610Salfred		if (opt->got_power_save) {
477185087Salfred			if (libusb20_dev_set_power_mode(pdev,
478185087Salfred			    LIBUSB20_POWER_SAVE)) {
479184610Salfred				err(1, "could not set power SAVE");
480184610Salfred			}
481184610Salfred		}
482184610Salfred		if (opt->got_power_on) {
483185087Salfred			if (libusb20_dev_set_power_mode(pdev,
484185087Salfred			    LIBUSB20_POWER_ON)) {
485184610Salfred				err(1, "could not set power ON");
486184610Salfred			}
487184610Salfred		}
488184610Salfred		dump_any =
489184610Salfred		    (opt->got_dump_device_desc ||
490184610Salfred		    opt->got_dump_curr_config ||
491184610Salfred		    opt->got_dump_all_config ||
492189110Sthompsa		    opt->got_dump_info);
493184610Salfred
494184610Salfred		if (opt->got_list || dump_any) {
495188622Sthompsa			dump_device_info(pdev,
496188622Sthompsa			    opt->got_show_iface_driver);
497184610Salfred		}
498184610Salfred		if (opt->got_dump_device_desc) {
499184610Salfred			printf("\n");
500184610Salfred			dump_device_desc(pdev);
501184610Salfred		}
502184610Salfred		if (opt->got_dump_all_config) {
503184610Salfred			printf("\n");
504184610Salfred			dump_config(pdev, 1);
505184610Salfred		} else if (opt->got_dump_curr_config) {
506184610Salfred			printf("\n");
507184610Salfred			dump_config(pdev, 0);
508184610Salfred		}
509184610Salfred		if (dump_any) {
510184610Salfred			printf("\n");
511184610Salfred		}
512184610Salfred		if (libusb20_dev_close(pdev)) {
513184610Salfred			err(1, "could not close device");
514184610Salfred		}
515184610Salfred	}
516184610Salfred
517184610Salfred	if (matches == 0) {
518187162Sthompsa		printf("No device match or lack of permissions.\n");
519184610Salfred	}
520184610Salfreddone:
521184610Salfred	reset_options(opt);
522184610Salfred
523184610Salfred	return;
524184610Salfred}
525184610Salfred
526184610Salfredint
527184610Salfredmain(int argc, char **argv)
528184610Salfred{
529184610Salfred	struct libusb20_backend *pbe;
530184610Salfred	struct options *opt = &options;
531201705Sthompsa	const char *ptr;
532201705Sthompsa	int unit;
533201705Sthompsa	int addr;
534184610Salfred	int n;
535184610Salfred	int t;
536184610Salfred
537184610Salfred	if (argc < 1) {
538184610Salfred		usage();
539184610Salfred	}
540184610Salfred	pbe = libusb20_be_alloc_default();
541184610Salfred	if (pbe == NULL)
542184610Salfred		err(1, "could not access USB backend\n");
543184610Salfred
544184610Salfred	for (n = 1; n != argc; n++) {
545184610Salfred
546184610Salfred		/* get number of additional options */
547184610Salfred		t = (argc - n - 1);
548184610Salfred		if (t > 255)
549184610Salfred			t = 255;
550184610Salfred		switch (get_token(argv[n], t)) {
551201705Sthompsa		case T_ADD_QUIRK:
552201705Sthompsa			if (opt->got_add_quirk) {
553201705Sthompsa				flush_command(pbe, opt);
554201705Sthompsa			}
555201705Sthompsa			opt->quirkname = argv[n + 1];
556201705Sthompsa			n++;
557201705Sthompsa
558201705Sthompsa			opt->got_add_quirk = 1;
559201705Sthompsa			opt->got_any++;
560201705Sthompsa			break;
561201705Sthompsa
562201705Sthompsa		case T_REMOVE_QUIRK:
563201705Sthompsa			if (opt->got_remove_quirk) {
564201705Sthompsa				flush_command(pbe, opt);
565201705Sthompsa			}
566201705Sthompsa			opt->quirkname = argv[n + 1];
567201705Sthompsa			n++;
568201705Sthompsa
569201705Sthompsa			opt->got_remove_quirk = 1;
570201705Sthompsa			opt->got_any++;
571201705Sthompsa			break;
572201705Sthompsa
573184610Salfred		case T_ADD_DEVICE_QUIRK:
574184610Salfred			if (opt->got_add_device_quirk) {
575184610Salfred				flush_command(pbe, opt);
576184610Salfred			}
577184610Salfred			opt->vid = num_id(argv[n + 1], "Vendor ID");
578184610Salfred			opt->pid = num_id(argv[n + 2], "Product ID");
579184610Salfred			opt->lo_rev = num_id(argv[n + 3], "Low Revision");
580184610Salfred			opt->hi_rev = num_id(argv[n + 4], "High Revision");
581184610Salfred			opt->quirkname = argv[n + 5];
582184610Salfred			n += 5;
583184610Salfred
584184610Salfred			opt->got_add_device_quirk = 1;
585184610Salfred			opt->got_any++;
586184610Salfred			break;
587184610Salfred
588184610Salfred		case T_REMOVE_DEVICE_QUIRK:
589184610Salfred			if (opt->got_remove_device_quirk) {
590184610Salfred				flush_command(pbe, opt);
591184610Salfred			}
592184610Salfred			opt->vid = num_id(argv[n + 1], "Vendor ID");
593184610Salfred			opt->pid = num_id(argv[n + 2], "Product ID");
594184610Salfred			opt->lo_rev = num_id(argv[n + 3], "Low Revision");
595184610Salfred			opt->hi_rev = num_id(argv[n + 4], "High Revision");
596184610Salfred			opt->quirkname = argv[n + 5];
597184610Salfred			n += 5;
598184610Salfred			opt->got_remove_device_quirk = 1;
599184610Salfred			opt->got_any++;
600184610Salfred			break;
601184610Salfred
602184610Salfred		case T_DUMP_QUIRK_NAMES:
603201705Sthompsa			if (opt->got_dump_quirk_names)
604201705Sthompsa				duplicate_option(argv[n]);
605184610Salfred			opt->got_dump_quirk_names = 1;
606184610Salfred			opt->got_any++;
607184610Salfred			break;
608184610Salfred
609184610Salfred		case T_DUMP_DEVICE_QUIRKS:
610201705Sthompsa			if (opt->got_dump_device_quirks)
611201705Sthompsa				duplicate_option(argv[n]);
612184610Salfred			opt->got_dump_device_quirks = 1;
613184610Salfred			opt->got_any++;
614184610Salfred			break;
615184610Salfred
616188622Sthompsa		case T_SHOW_IFACE_DRIVER:
617188622Sthompsa			opt->got_show_iface_driver = 1;
618188622Sthompsa			break;
619188622Sthompsa
620201705Sthompsa		case T_UGEN:
621201705Sthompsa			if (opt->got_any) {
622201705Sthompsa				/* allow multiple commands on the same line */
623201705Sthompsa				flush_command(pbe, opt);
624201705Sthompsa			}
625201705Sthompsa			ptr = argv[n + 1];
626201705Sthompsa
627201705Sthompsa			if ((ptr[0] == 'u') &&
628201705Sthompsa			    (ptr[1] == 'g') &&
629201705Sthompsa			    (ptr[2] == 'e') &&
630201705Sthompsa			    (ptr[3] == 'n'))
631201705Sthompsa				ptr += 4;
632201705Sthompsa
633201705Sthompsa			if ((sscanf(ptr, "%d.%d",
634201705Sthompsa			    &unit, &addr) != 2) ||
635201705Sthompsa			    (unit < 0) || (unit > 65535) ||
636201705Sthompsa			    (addr < 0) || (addr > 65535)) {
637201705Sthompsa				errx(1, "cannot "
638201705Sthompsa				    "parse '%s'", argv[n + 1]);
639201705Sthompsa			}
640201705Sthompsa			opt->bus = unit;
641201705Sthompsa			opt->addr = addr;
642201705Sthompsa			opt->got_bus = 1;
643201705Sthompsa;			opt->got_addr = 1;
644201705Sthompsa			n++;
645201705Sthompsa			break;
646201705Sthompsa
647184610Salfred		case T_UNIT:
648184610Salfred			if (opt->got_any) {
649184610Salfred				/* allow multiple commands on the same line */
650184610Salfred				flush_command(pbe, opt);
651184610Salfred			}
652184610Salfred			opt->bus = num_id(argv[n + 1], "busnum");
653184610Salfred			opt->got_bus = 1;
654184610Salfred			n++;
655184610Salfred			break;
656184610Salfred		case T_ADDR:
657184610Salfred			opt->addr = num_id(argv[n + 1], "addr");
658184610Salfred			opt->got_addr = 1;
659184610Salfred			n++;
660184610Salfred			break;
661184610Salfred		case T_IFACE:
662184610Salfred			opt->iface = num_id(argv[n + 1], "iface");
663184610Salfred			opt->got_iface = 1;
664184610Salfred			n++;
665184610Salfred			break;
666184610Salfred		case T_SET_CONFIG:
667201705Sthompsa			if (opt->got_set_config)
668201705Sthompsa				duplicate_option(argv[n]);
669185087Salfred			opt->config_index = num_id(argv[n + 1], "cfg_index");
670184610Salfred			opt->got_set_config = 1;
671184610Salfred			opt->got_any++;
672184610Salfred			n++;
673184610Salfred			break;
674184610Salfred		case T_SET_ALT:
675201705Sthompsa			if (opt->got_set_alt)
676201705Sthompsa				duplicate_option(argv[n]);
677185087Salfred			opt->alt_index = num_id(argv[n + 1], "cfg_index");
678184610Salfred			opt->got_set_alt = 1;
679184610Salfred			opt->got_any++;
680184610Salfred			n++;
681184610Salfred			break;
682188987Sthompsa		case T_SET_TEMPLATE:
683201705Sthompsa			if (opt->got_set_template)
684201705Sthompsa				duplicate_option(argv[n]);
685189110Sthompsa			opt->template = get_int(argv[n + 1]);
686188987Sthompsa			opt->got_set_template = 1;
687188987Sthompsa			opt->got_any++;
688188987Sthompsa			n++;
689188987Sthompsa			break;
690188987Sthompsa		case T_GET_TEMPLATE:
691201705Sthompsa			if (opt->got_get_template)
692201705Sthompsa				duplicate_option(argv[n]);
693188987Sthompsa			opt->got_get_template = 1;
694188987Sthompsa			opt->got_any++;
695188987Sthompsa			break;
696184610Salfred		case T_DUMP_DEVICE_DESC:
697201705Sthompsa			if (opt->got_dump_device_desc)
698201705Sthompsa				duplicate_option(argv[n]);
699184610Salfred			opt->got_dump_device_desc = 1;
700184610Salfred			opt->got_any++;
701184610Salfred			break;
702184610Salfred		case T_DUMP_CURR_CONFIG_DESC:
703201705Sthompsa			if (opt->got_dump_curr_config)
704201705Sthompsa				duplicate_option(argv[n]);
705184610Salfred			opt->got_dump_curr_config = 1;
706184610Salfred			opt->got_any++;
707184610Salfred			break;
708184610Salfred		case T_DUMP_ALL_CONFIG_DESC:
709201705Sthompsa			if (opt->got_dump_all_config)
710201705Sthompsa				duplicate_option(argv[n]);
711184610Salfred			opt->got_dump_all_config = 1;
712184610Salfred			opt->got_any++;
713184610Salfred			break;
714184610Salfred		case T_DUMP_INFO:
715201705Sthompsa			if (opt->got_dump_info)
716201705Sthompsa				duplicate_option(argv[n]);
717184610Salfred			opt->got_dump_info = 1;
718184610Salfred			opt->got_any++;
719184610Salfred			break;
720185087Salfred		case T_DUMP_STRING:
721201705Sthompsa			if (opt->got_dump_string)
722201705Sthompsa				duplicate_option(argv[n]);
723185087Salfred			opt->string_index = num_id(argv[n + 1], "str_index");
724185087Salfred			opt->got_dump_string = 1;
725185087Salfred			opt->got_any++;
726185087Salfred			n++;
727185087Salfred			break;
728184610Salfred		case T_SUSPEND:
729201705Sthompsa			if (opt->got_suspend)
730201705Sthompsa				duplicate_option(argv[n]);
731184610Salfred			opt->got_suspend = 1;
732184610Salfred			opt->got_any++;
733184610Salfred			break;
734184610Salfred		case T_RESUME:
735201705Sthompsa			if (opt->got_resume)
736201705Sthompsa				duplicate_option(argv[n]);
737184610Salfred			opt->got_resume = 1;
738184610Salfred			opt->got_any++;
739184610Salfred			break;
740184610Salfred		case T_POWER_OFF:
741201705Sthompsa			if (opt->got_power_off)
742201705Sthompsa				duplicate_option(argv[n]);
743184610Salfred			opt->got_power_off = 1;
744184610Salfred			opt->got_any++;
745184610Salfred			break;
746184610Salfred		case T_POWER_SAVE:
747201705Sthompsa			if (opt->got_power_save)
748201705Sthompsa				duplicate_option(argv[n]);
749184610Salfred			opt->got_power_save = 1;
750184610Salfred			opt->got_any++;
751184610Salfred			break;
752184610Salfred		case T_POWER_ON:
753201705Sthompsa			if (opt->got_power_on)
754201705Sthompsa				duplicate_option(argv[n]);
755184610Salfred			opt->got_power_on = 1;
756184610Salfred			opt->got_any++;
757184610Salfred			break;
758184610Salfred		case T_RESET:
759201705Sthompsa			if (opt->got_reset)
760201705Sthompsa				duplicate_option(argv[n]);
761184610Salfred			opt->got_reset = 1;
762184610Salfred			opt->got_any++;
763184610Salfred			break;
764184610Salfred		case T_LIST:
765201705Sthompsa			if (opt->got_list)
766201705Sthompsa				duplicate_option(argv[n]);
767184610Salfred			opt->got_list = 1;
768184610Salfred			opt->got_any++;
769184610Salfred			break;
770185087Salfred		case T_DO_REQUEST:
771201705Sthompsa			if (opt->got_do_request)
772201705Sthompsa				duplicate_option(argv[n]);
773185087Salfred			LIBUSB20_INIT(LIBUSB20_CONTROL_SETUP, &opt->setup);
774185087Salfred			opt->setup.bmRequestType = num_id(argv[n + 1], "bmReqTyp");
775185087Salfred			opt->setup.bRequest = num_id(argv[n + 2], "bReq");
776185087Salfred			opt->setup.wValue = num_id(argv[n + 3], "wVal");
777185087Salfred			opt->setup.wIndex = num_id(argv[n + 4], "wIndex");
778185087Salfred			opt->setup.wLength = num_id(argv[n + 5], "wLen");
779185087Salfred			if (opt->setup.wLength != 0) {
780185087Salfred				opt->buffer = malloc(opt->setup.wLength);
781185087Salfred			} else {
782185087Salfred				opt->buffer = NULL;
783185087Salfred			}
784185087Salfred
785185087Salfred			n += 5;
786185087Salfred
787185087Salfred			if (!(opt->setup.bmRequestType &
788185087Salfred			    LIBUSB20_ENDPOINT_IN)) {
789185087Salfred				/* copy in data */
790185087Salfred				t = (argc - n - 1);
791185087Salfred				if (t < opt->setup.wLength) {
792185087Salfred					err(1, "request data missing");
793185087Salfred				}
794185087Salfred				t = opt->setup.wLength;
795185087Salfred				while (t--) {
796185087Salfred					((uint8_t *)opt->buffer)[t] =
797185087Salfred					    num_id(argv[n + t + 1], "req_data");
798185087Salfred				}
799185087Salfred				n += opt->setup.wLength;
800185087Salfred			}
801185087Salfred			opt->got_do_request = 1;
802185087Salfred			opt->got_any++;
803185087Salfred			break;
804184610Salfred		default:
805184610Salfred			usage();
806184610Salfred			break;
807184610Salfred		}
808184610Salfred	}
809184610Salfred	if (opt->got_any) {
810184610Salfred		/* flush out last command */
811184610Salfred		flush_command(pbe, opt);
812184610Salfred	} else {
813184610Salfred		/* list all the devices */
814184610Salfred		opt->got_list = 1;
815184610Salfred		opt->got_any++;
816184610Salfred		flush_command(pbe, opt);
817184610Salfred	}
818184610Salfred	/* release data */
819184610Salfred	libusb20_be_free(pbe);
820184610Salfred
821184610Salfred	return (0);
822184610Salfred}
823