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