1184610Salfred/* $FreeBSD$ */ 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; 78184610Salfred uint8_t got_dump_device_desc:1; 79184610Salfred uint8_t got_dump_curr_config:1; 80184610Salfred uint8_t got_dump_all_config:1; 81184610Salfred uint8_t got_dump_info:1; 82188622Sthompsa uint8_t got_show_iface_driver:1; 83184610Salfred uint8_t got_remove_device_quirk:1; 84184610Salfred uint8_t got_add_device_quirk:1; 85201705Sthompsa uint8_t got_remove_quirk:1; 86201705Sthompsa uint8_t got_add_quirk:1; 87185087Salfred uint8_t got_dump_string:1; 88185087Salfred uint8_t got_do_request:1; 89184610Salfred}; 90184610Salfred 91184610Salfredstruct token { 92184610Salfred const char *name; 93184610Salfred uint8_t value; 94184610Salfred uint8_t narg; 95184610Salfred}; 96184610Salfred 97184610Salfredenum { 98184610Salfred T_UNIT, 99184610Salfred T_ADDR, 100201705Sthompsa T_UGEN, 101184610Salfred T_IFACE, 102184610Salfred T_SET_CONFIG, 103184610Salfred T_SET_ALT, 104188987Sthompsa T_SET_TEMPLATE, 105188987Sthompsa T_GET_TEMPLATE, 106184610Salfred T_ADD_DEVICE_QUIRK, 107184610Salfred T_REMOVE_DEVICE_QUIRK, 108201705Sthompsa T_ADD_QUIRK, 109201705Sthompsa T_REMOVE_QUIRK, 110188622Sthompsa T_SHOW_IFACE_DRIVER, 111184610Salfred T_DUMP_QUIRK_NAMES, 112184610Salfred T_DUMP_DEVICE_QUIRKS, 113184610Salfred T_DUMP_DEVICE_DESC, 114184610Salfred T_DUMP_CURR_CONFIG_DESC, 115184610Salfred T_DUMP_ALL_CONFIG_DESC, 116185087Salfred T_DUMP_STRING, 117184610Salfred T_DUMP_INFO, 118184610Salfred T_SUSPEND, 119184610Salfred T_RESUME, 120184610Salfred T_POWER_OFF, 121184610Salfred T_POWER_SAVE, 122184610Salfred T_POWER_ON, 123184610Salfred T_RESET, 124184610Salfred T_LIST, 125185087Salfred T_DO_REQUEST, 126184610Salfred}; 127184610Salfred 128184610Salfredstatic struct options options; 129184610Salfred 130184610Salfredstatic const struct token token[] = { 131184610Salfred {"-u", T_UNIT, 1}, 132184610Salfred {"-a", T_ADDR, 1}, 133201705Sthompsa {"-d", T_UGEN, 1}, 134184610Salfred {"-i", T_IFACE, 1}, 135184610Salfred {"set_config", T_SET_CONFIG, 1}, 136184610Salfred {"set_alt", T_SET_ALT, 1}, 137188987Sthompsa {"set_template", T_SET_TEMPLATE, 1}, 138188987Sthompsa {"get_template", T_GET_TEMPLATE, 0}, 139184610Salfred {"add_dev_quirk_vplh", T_ADD_DEVICE_QUIRK, 5}, 140184610Salfred {"remove_dev_quirk_vplh", T_REMOVE_DEVICE_QUIRK, 5}, 141201705Sthompsa {"add_quirk", T_ADD_QUIRK, 1}, 142201705Sthompsa {"remove_quirk", T_REMOVE_QUIRK, 1}, 143184610Salfred {"dump_quirk_names", T_DUMP_QUIRK_NAMES, 0}, 144184610Salfred {"dump_device_quirks", T_DUMP_DEVICE_QUIRKS, 0}, 145184610Salfred {"dump_device_desc", T_DUMP_DEVICE_DESC, 0}, 146184610Salfred {"dump_curr_config_desc", T_DUMP_CURR_CONFIG_DESC, 0}, 147184610Salfred {"dump_all_config_desc", T_DUMP_ALL_CONFIG_DESC, 0}, 148185087Salfred {"dump_string", T_DUMP_STRING, 1}, 149184610Salfred {"dump_info", T_DUMP_INFO, 0}, 150188622Sthompsa {"show_ifdrv", T_SHOW_IFACE_DRIVER, 0}, 151184610Salfred {"suspend", T_SUSPEND, 0}, 152184610Salfred {"resume", T_RESUME, 0}, 153184610Salfred {"power_off", T_POWER_OFF, 0}, 154184610Salfred {"power_save", T_POWER_SAVE, 0}, 155184610Salfred {"power_on", T_POWER_ON, 0}, 156184610Salfred {"reset", T_RESET, 0}, 157184610Salfred {"list", T_LIST, 0}, 158185087Salfred {"do_request", T_DO_REQUEST, 5}, 159184610Salfred}; 160184610Salfred 161184610Salfredstatic void 162184610Salfredbe_dev_remove_quirk(struct libusb20_backend *pbe, 163184610Salfred uint16_t vid, uint16_t pid, uint16_t lorev, uint16_t hirev, 164184610Salfred const char *str) 165184610Salfred{ 166184610Salfred struct libusb20_quirk q; 167185087Salfred int error; 168184610Salfred 169184610Salfred memset(&q, 0, sizeof(q)); 170184610Salfred 171184610Salfred q.vid = vid; 172184610Salfred q.pid = pid; 173184610Salfred q.bcdDeviceLow = lorev; 174184610Salfred q.bcdDeviceHigh = hirev; 175184610Salfred strlcpy(q.quirkname, str, sizeof(q.quirkname)); 176184610Salfred 177185087Salfred error = libusb20_be_remove_dev_quirk(pbe, &q); 178185087Salfred if (error) { 179202026Sthompsa fprintf(stderr, "Removing quirk '%s' failed, continuing.\n", str); 180184610Salfred } 181184610Salfred return; 182184610Salfred} 183184610Salfred 184184610Salfredstatic void 185184610Salfredbe_dev_add_quirk(struct libusb20_backend *pbe, 186184610Salfred uint16_t vid, uint16_t pid, uint16_t lorev, uint16_t hirev, 187184610Salfred const char *str) 188184610Salfred{ 189184610Salfred struct libusb20_quirk q; 190185087Salfred int error; 191184610Salfred 192184610Salfred memset(&q, 0, sizeof(q)); 193184610Salfred 194184610Salfred q.vid = vid; 195184610Salfred q.pid = pid; 196184610Salfred q.bcdDeviceLow = lorev; 197184610Salfred q.bcdDeviceHigh = hirev; 198184610Salfred strlcpy(q.quirkname, str, sizeof(q.quirkname)); 199184610Salfred 200185087Salfred error = libusb20_be_add_dev_quirk(pbe, &q); 201185087Salfred if (error) { 202202026Sthompsa fprintf(stderr, "Adding quirk '%s' failed, continuing.\n", str); 203184610Salfred } 204184610Salfred return; 205184610Salfred} 206184610Salfred 207184610Salfredstatic uint8_t 208184610Salfredget_token(const char *str, uint8_t narg) 209184610Salfred{ 210184610Salfred uint8_t n; 211184610Salfred 212184610Salfred for (n = 0; n != (sizeof(token) / sizeof(token[0])); n++) { 213184610Salfred if (strcasecmp(str, token[n].name) == 0) { 214184610Salfred if (token[n].narg > narg) { 215185087Salfred /* too few arguments */ 216184610Salfred break; 217184610Salfred } 218184610Salfred return (token[n].value); 219184610Salfred } 220184610Salfred } 221184610Salfred return (0 - 1); 222184610Salfred} 223184610Salfred 224184610Salfredstatic uid_t 225184610Salfrednum_id(const char *name, const char *type) 226184610Salfred{ 227184610Salfred uid_t val; 228184610Salfred char *ep; 229184610Salfred 230184610Salfred errno = 0; 231184610Salfred val = strtoul(name, &ep, 0); 232184610Salfred if (errno) { 233184610Salfred err(1, "%s", name); 234184610Salfred } 235184610Salfred if (*ep != '\0') { 236184610Salfred errx(1, "%s: illegal %s name", name, type); 237184610Salfred } 238184610Salfred return (val); 239184610Salfred} 240184610Salfred 241189110Sthompsastatic int 242189110Sthompsaget_int(const char *s) 243184610Salfred{ 244189110Sthompsa int val; 245184610Salfred char *ep; 246184610Salfred 247184610Salfred errno = 0; 248189110Sthompsa val = strtoul(s, &ep, 0); 249184610Salfred if (errno) { 250184610Salfred err(1, "%s", s); 251184610Salfred } 252184610Salfred if (*ep != '\0') { 253189110Sthompsa errx(1, "illegal number: %s", s); 254184610Salfred } 255184610Salfred return val; 256184610Salfred} 257184610Salfred 258184610Salfredstatic void 259201705Sthompsaduplicate_option(const char *ptr) 260201705Sthompsa{ 261202026Sthompsa fprintf(stderr, "Syntax error: " 262201705Sthompsa "Duplicate option: '%s'\n", ptr); 263201705Sthompsa exit(1); 264201705Sthompsa} 265201705Sthompsa 266201705Sthompsastatic void 267184610Salfredusage(void) 268184610Salfred{ 269202026Sthompsa fprintf(stderr, "" 270184610Salfred "usbconfig - configure the USB subsystem" "\n" 271184610Salfred "usage: usbconfig -u <busnum> -a <devaddr> -i <ifaceindex> [cmds...]" "\n" 272201705Sthompsa "usage: usbconfig -d [ugen]<busnum>.<devaddr> -i <ifaceindex> [cmds...]" "\n" 273184610Salfred "commands:" "\n" 274185087Salfred " set_config <cfg_index>" "\n" 275185087Salfred " set_alt <alt_index>" "\n" 276188987Sthompsa " set_template <template>" "\n" 277188987Sthompsa " get_template" "\n" 278184610Salfred " add_dev_quirk_vplh <vid> <pid> <lo_rev> <hi_rev> <quirk>" "\n" 279184610Salfred " remove_dev_quirk_vplh <vid> <pid> <lo_rev> <hi_rev> <quirk>" "\n" 280201705Sthompsa " add_quirk <quirk>" "\n" 281201705Sthompsa " remove_quirk <quirk>" "\n" 282184610Salfred " dump_quirk_names" "\n" 283184610Salfred " dump_device_quirks" "\n" 284184610Salfred " dump_device_desc" "\n" 285184610Salfred " dump_curr_config_desc" "\n" 286184610Salfred " dump_all_config_desc" "\n" 287185087Salfred " dump_string <index>" "\n" 288184610Salfred " dump_info" "\n" 289188622Sthompsa " show_ifdrv" "\n" 290184610Salfred " suspend" "\n" 291184610Salfred " resume" "\n" 292184610Salfred " power_off" "\n" 293184610Salfred " power_save" "\n" 294184610Salfred " power_on" "\n" 295184610Salfred " reset" "\n" 296184610Salfred " list" "\n" 297185087Salfred " do_request <bmReqTyp> <bReq> <wVal> <wIdx> <wLen> <data...>" "\n" 298184610Salfred ); 299184610Salfred exit(1); 300184610Salfred} 301184610Salfred 302184610Salfredstatic void 303184610Salfredreset_options(struct options *opt) 304184610Salfred{ 305185087Salfred if (opt->buffer) 306185087Salfred free(opt->buffer); 307184610Salfred memset(opt, 0, sizeof(*opt)); 308184610Salfred return; 309184610Salfred} 310184610Salfred 311184610Salfredstatic void 312184610Salfredflush_command(struct libusb20_backend *pbe, struct options *opt) 313184610Salfred{ 314184610Salfred struct libusb20_device *pdev = NULL; 315184610Salfred uint32_t matches = 0; 316184610Salfred uint8_t dump_any; 317184610Salfred 318184610Salfred /* check for invalid option combinations */ 319184610Salfred if ((opt->got_suspend + 320184610Salfred opt->got_resume + 321184610Salfred opt->got_reset + 322184610Salfred opt->got_set_config + 323184610Salfred opt->got_set_alt + 324184610Salfred opt->got_power_save + 325184610Salfred opt->got_power_on + 326184610Salfred opt->got_power_off) > 1) { 327187168Sthompsa err(1, "can only specify one of 'set_config', " 328184610Salfred "'set_alt', 'reset', 'suspend', 'resume', " 329184610Salfred "'power_save', 'power_on' and 'power_off' " 330184610Salfred "at the same time!"); 331184610Salfred } 332184610Salfred if (opt->got_dump_quirk_names) { 333184610Salfred opt->got_any--; 334184610Salfred dump_be_quirk_names(pbe); 335184610Salfred } 336184610Salfred if (opt->got_dump_device_quirks) { 337184610Salfred opt->got_any--; 338184610Salfred dump_be_dev_quirks(pbe); 339184610Salfred } 340184610Salfred if (opt->got_remove_device_quirk) { 341184610Salfred opt->got_any--; 342184610Salfred be_dev_remove_quirk(pbe, 343184610Salfred opt->vid, opt->pid, opt->lo_rev, opt->hi_rev, opt->quirkname); 344184610Salfred } 345184610Salfred if (opt->got_add_device_quirk) { 346184610Salfred opt->got_any--; 347184610Salfred be_dev_add_quirk(pbe, 348184610Salfred opt->vid, opt->pid, opt->lo_rev, opt->hi_rev, opt->quirkname); 349184610Salfred } 350188987Sthompsa if (opt->got_set_template) { 351188987Sthompsa opt->got_any--; 352188987Sthompsa if (libusb20_be_set_template(pbe, opt->template)) { 353202026Sthompsa fprintf(stderr, "Setting USB template %u failed, " 354188987Sthompsa "continuing.\n", opt->template); 355188987Sthompsa } 356188987Sthompsa } 357188987Sthompsa if (opt->got_get_template) { 358188987Sthompsa opt->got_any--; 359188987Sthompsa if (libusb20_be_get_template(pbe, &opt->template)) 360188987Sthompsa printf("USB template: <unknown>\n"); 361188987Sthompsa else 362188987Sthompsa printf("USB template: %u\n", opt->template); 363188987Sthompsa } 364184610Salfred if (opt->got_any == 0) { 365184610Salfred /* 366184610Salfred * do not scan through all the devices if there are no valid 367184610Salfred * options 368184610Salfred */ 369184610Salfred goto done; 370184610Salfred } 371184610Salfred while ((pdev = libusb20_be_device_foreach(pbe, pdev))) { 372184610Salfred 373184610Salfred if (opt->got_bus && 374184610Salfred (libusb20_dev_get_bus_number(pdev) != opt->bus)) { 375184610Salfred continue; 376184610Salfred } 377184610Salfred if (opt->got_addr && 378184610Salfred (libusb20_dev_get_address(pdev) != opt->addr)) { 379184610Salfred continue; 380184610Salfred } 381184610Salfred matches++; 382184610Salfred 383201705Sthompsa if (opt->got_remove_quirk) { 384201705Sthompsa struct LIBUSB20_DEVICE_DESC_DECODED *ddesc; 385201705Sthompsa 386201705Sthompsa ddesc = libusb20_dev_get_device_desc(pdev); 387201705Sthompsa 388201705Sthompsa be_dev_remove_quirk(pbe, 389201705Sthompsa ddesc->idVendor, ddesc->idProduct, 390201705Sthompsa ddesc->bcdDevice, ddesc->bcdDevice, 391201705Sthompsa opt->quirkname); 392201705Sthompsa } 393201705Sthompsa 394201705Sthompsa if (opt->got_add_quirk) { 395201705Sthompsa struct LIBUSB20_DEVICE_DESC_DECODED *ddesc; 396201705Sthompsa 397201705Sthompsa ddesc = libusb20_dev_get_device_desc(pdev); 398201705Sthompsa 399201705Sthompsa be_dev_add_quirk(pbe, 400201705Sthompsa ddesc->idVendor, ddesc->idProduct, 401201705Sthompsa ddesc->bcdDevice, ddesc->bcdDevice, 402201705Sthompsa opt->quirkname); 403201705Sthompsa } 404201705Sthompsa 405184610Salfred if (libusb20_dev_open(pdev, 0)) { 406184610Salfred err(1, "could not open device"); 407184610Salfred } 408185087Salfred if (opt->got_dump_string) { 409201705Sthompsa dump_string_by_index(pdev, opt->string_index); 410185087Salfred } 411185087Salfred if (opt->got_do_request) { 412185087Salfred uint16_t actlen; 413185087Salfred uint16_t t; 414185087Salfred 415185087Salfred if (libusb20_dev_request_sync(pdev, &opt->setup, 416185087Salfred opt->buffer, &actlen, 5000 /* 5 seconds */ , 0)) { 417185087Salfred printf("REQUEST = <ERROR>\n"); 418185087Salfred } else if (!(opt->setup.bmRequestType & 419185087Salfred LIBUSB20_ENDPOINT_IN)) { 420185087Salfred printf("REQUEST = <OK>\n"); 421185087Salfred } else { 422185087Salfred t = actlen; 423185087Salfred printf("REQUEST = <"); 424185087Salfred for (t = 0; t != actlen; t++) { 425185087Salfred printf("0x%02x%s", 426185087Salfred ((uint8_t *)opt->buffer)[t], 427185087Salfred (t == (actlen - 1)) ? "" : " "); 428185087Salfred } 429185087Salfred printf("><"); 430185087Salfred for (t = 0; t != actlen; t++) { 431185087Salfred char c; 432185087Salfred 433185087Salfred c = ((uint8_t *)opt->buffer)[t]; 434185087Salfred if ((c != '<') && 435185087Salfred (c != '>') && isprint(c)) { 436185087Salfred putchar(c); 437185087Salfred } 438185087Salfred } 439185087Salfred printf(">\n"); 440185087Salfred } 441185087Salfred } 442184610Salfred if (opt->got_set_config) { 443184610Salfred if (libusb20_dev_set_config_index(pdev, 444184610Salfred opt->config_index)) { 445184610Salfred err(1, "could not set config index"); 446184610Salfred } 447184610Salfred } 448184610Salfred if (opt->got_set_alt) { 449185087Salfred if (libusb20_dev_set_alt_index(pdev, opt->iface, 450185087Salfred opt->alt_index)) { 451184610Salfred err(1, "could not set alternate setting"); 452184610Salfred } 453184610Salfred } 454184610Salfred if (opt->got_reset) { 455184610Salfred if (libusb20_dev_reset(pdev)) { 456184610Salfred err(1, "could not reset device"); 457184610Salfred } 458184610Salfred } 459184610Salfred if (opt->got_suspend) { 460185087Salfred if (libusb20_dev_set_power_mode(pdev, 461185087Salfred LIBUSB20_POWER_SUSPEND)) { 462184610Salfred err(1, "could not set suspend"); 463184610Salfred } 464184610Salfred } 465184610Salfred if (opt->got_resume) { 466185087Salfred if (libusb20_dev_set_power_mode(pdev, 467185087Salfred LIBUSB20_POWER_RESUME)) { 468184610Salfred err(1, "could not set resume"); 469184610Salfred } 470184610Salfred } 471184610Salfred if (opt->got_power_off) { 472185087Salfred if (libusb20_dev_set_power_mode(pdev, 473185087Salfred LIBUSB20_POWER_OFF)) { 474184610Salfred err(1, "could not set power OFF"); 475184610Salfred } 476184610Salfred } 477184610Salfred if (opt->got_power_save) { 478185087Salfred if (libusb20_dev_set_power_mode(pdev, 479185087Salfred LIBUSB20_POWER_SAVE)) { 480184610Salfred err(1, "could not set power SAVE"); 481184610Salfred } 482184610Salfred } 483184610Salfred if (opt->got_power_on) { 484185087Salfred if (libusb20_dev_set_power_mode(pdev, 485185087Salfred LIBUSB20_POWER_ON)) { 486184610Salfred err(1, "could not set power ON"); 487184610Salfred } 488184610Salfred } 489184610Salfred dump_any = 490184610Salfred (opt->got_dump_device_desc || 491184610Salfred opt->got_dump_curr_config || 492184610Salfred opt->got_dump_all_config || 493189110Sthompsa opt->got_dump_info); 494184610Salfred 495184610Salfred if (opt->got_list || dump_any) { 496188622Sthompsa dump_device_info(pdev, 497188622Sthompsa opt->got_show_iface_driver); 498184610Salfred } 499184610Salfred if (opt->got_dump_device_desc) { 500184610Salfred printf("\n"); 501184610Salfred dump_device_desc(pdev); 502184610Salfred } 503184610Salfred if (opt->got_dump_all_config) { 504184610Salfred printf("\n"); 505184610Salfred dump_config(pdev, 1); 506184610Salfred } else if (opt->got_dump_curr_config) { 507184610Salfred printf("\n"); 508184610Salfred dump_config(pdev, 0); 509184610Salfred } 510184610Salfred if (dump_any) { 511184610Salfred printf("\n"); 512184610Salfred } 513184610Salfred if (libusb20_dev_close(pdev)) { 514184610Salfred err(1, "could not close device"); 515184610Salfred } 516184610Salfred } 517184610Salfred 518184610Salfred if (matches == 0) { 519187162Sthompsa printf("No device match or lack of permissions.\n"); 520184610Salfred } 521184610Salfreddone: 522184610Salfred reset_options(opt); 523184610Salfred 524184610Salfred return; 525184610Salfred} 526184610Salfred 527184610Salfredint 528184610Salfredmain(int argc, char **argv) 529184610Salfred{ 530184610Salfred struct libusb20_backend *pbe; 531184610Salfred struct options *opt = &options; 532201705Sthompsa const char *ptr; 533201705Sthompsa int unit; 534201705Sthompsa int addr; 535184610Salfred int n; 536184610Salfred int t; 537184610Salfred 538184610Salfred if (argc < 1) { 539184610Salfred usage(); 540184610Salfred } 541184610Salfred pbe = libusb20_be_alloc_default(); 542184610Salfred if (pbe == NULL) 543184610Salfred err(1, "could not access USB backend\n"); 544184610Salfred 545184610Salfred for (n = 1; n != argc; n++) { 546184610Salfred 547184610Salfred /* get number of additional options */ 548184610Salfred t = (argc - n - 1); 549184610Salfred if (t > 255) 550184610Salfred t = 255; 551184610Salfred switch (get_token(argv[n], t)) { 552201705Sthompsa case T_ADD_QUIRK: 553201705Sthompsa if (opt->got_add_quirk) { 554201705Sthompsa flush_command(pbe, opt); 555201705Sthompsa } 556201705Sthompsa opt->quirkname = argv[n + 1]; 557201705Sthompsa n++; 558201705Sthompsa 559201705Sthompsa opt->got_add_quirk = 1; 560201705Sthompsa opt->got_any++; 561201705Sthompsa break; 562201705Sthompsa 563201705Sthompsa case T_REMOVE_QUIRK: 564201705Sthompsa if (opt->got_remove_quirk) { 565201705Sthompsa flush_command(pbe, opt); 566201705Sthompsa } 567201705Sthompsa opt->quirkname = argv[n + 1]; 568201705Sthompsa n++; 569201705Sthompsa 570201705Sthompsa opt->got_remove_quirk = 1; 571201705Sthompsa opt->got_any++; 572201705Sthompsa break; 573201705Sthompsa 574184610Salfred case T_ADD_DEVICE_QUIRK: 575184610Salfred if (opt->got_add_device_quirk) { 576184610Salfred flush_command(pbe, opt); 577184610Salfred } 578184610Salfred opt->vid = num_id(argv[n + 1], "Vendor ID"); 579184610Salfred opt->pid = num_id(argv[n + 2], "Product ID"); 580184610Salfred opt->lo_rev = num_id(argv[n + 3], "Low Revision"); 581184610Salfred opt->hi_rev = num_id(argv[n + 4], "High Revision"); 582184610Salfred opt->quirkname = argv[n + 5]; 583184610Salfred n += 5; 584184610Salfred 585184610Salfred opt->got_add_device_quirk = 1; 586184610Salfred opt->got_any++; 587184610Salfred break; 588184610Salfred 589184610Salfred case T_REMOVE_DEVICE_QUIRK: 590184610Salfred if (opt->got_remove_device_quirk) { 591184610Salfred flush_command(pbe, opt); 592184610Salfred } 593184610Salfred opt->vid = num_id(argv[n + 1], "Vendor ID"); 594184610Salfred opt->pid = num_id(argv[n + 2], "Product ID"); 595184610Salfred opt->lo_rev = num_id(argv[n + 3], "Low Revision"); 596184610Salfred opt->hi_rev = num_id(argv[n + 4], "High Revision"); 597184610Salfred opt->quirkname = argv[n + 5]; 598184610Salfred n += 5; 599184610Salfred opt->got_remove_device_quirk = 1; 600184610Salfred opt->got_any++; 601184610Salfred break; 602184610Salfred 603184610Salfred case T_DUMP_QUIRK_NAMES: 604201705Sthompsa if (opt->got_dump_quirk_names) 605201705Sthompsa duplicate_option(argv[n]); 606184610Salfred opt->got_dump_quirk_names = 1; 607184610Salfred opt->got_any++; 608184610Salfred break; 609184610Salfred 610184610Salfred case T_DUMP_DEVICE_QUIRKS: 611201705Sthompsa if (opt->got_dump_device_quirks) 612201705Sthompsa duplicate_option(argv[n]); 613184610Salfred opt->got_dump_device_quirks = 1; 614184610Salfred opt->got_any++; 615184610Salfred break; 616184610Salfred 617188622Sthompsa case T_SHOW_IFACE_DRIVER: 618188622Sthompsa opt->got_show_iface_driver = 1; 619188622Sthompsa break; 620188622Sthompsa 621201705Sthompsa case T_UGEN: 622201705Sthompsa if (opt->got_any) { 623201705Sthompsa /* allow multiple commands on the same line */ 624201705Sthompsa flush_command(pbe, opt); 625201705Sthompsa } 626201705Sthompsa ptr = argv[n + 1]; 627201705Sthompsa 628201705Sthompsa if ((ptr[0] == 'u') && 629201705Sthompsa (ptr[1] == 'g') && 630201705Sthompsa (ptr[2] == 'e') && 631201705Sthompsa (ptr[3] == 'n')) 632201705Sthompsa ptr += 4; 633201705Sthompsa 634201705Sthompsa if ((sscanf(ptr, "%d.%d", 635201705Sthompsa &unit, &addr) != 2) || 636201705Sthompsa (unit < 0) || (unit > 65535) || 637201705Sthompsa (addr < 0) || (addr > 65535)) { 638201705Sthompsa errx(1, "cannot " 639201705Sthompsa "parse '%s'", argv[n + 1]); 640201705Sthompsa } 641201705Sthompsa opt->bus = unit; 642201705Sthompsa opt->addr = addr; 643201705Sthompsa opt->got_bus = 1; 644214271Shselasky opt->got_addr = 1; 645201705Sthompsa n++; 646201705Sthompsa break; 647201705Sthompsa 648184610Salfred case T_UNIT: 649184610Salfred if (opt->got_any) { 650184610Salfred /* allow multiple commands on the same line */ 651184610Salfred flush_command(pbe, opt); 652184610Salfred } 653184610Salfred opt->bus = num_id(argv[n + 1], "busnum"); 654184610Salfred opt->got_bus = 1; 655184610Salfred n++; 656184610Salfred break; 657184610Salfred case T_ADDR: 658184610Salfred opt->addr = num_id(argv[n + 1], "addr"); 659184610Salfred opt->got_addr = 1; 660184610Salfred n++; 661184610Salfred break; 662184610Salfred case T_IFACE: 663184610Salfred opt->iface = num_id(argv[n + 1], "iface"); 664184610Salfred opt->got_iface = 1; 665184610Salfred n++; 666184610Salfred break; 667184610Salfred case T_SET_CONFIG: 668201705Sthompsa if (opt->got_set_config) 669201705Sthompsa duplicate_option(argv[n]); 670185087Salfred opt->config_index = num_id(argv[n + 1], "cfg_index"); 671184610Salfred opt->got_set_config = 1; 672184610Salfred opt->got_any++; 673184610Salfred n++; 674184610Salfred break; 675184610Salfred case T_SET_ALT: 676201705Sthompsa if (opt->got_set_alt) 677201705Sthompsa duplicate_option(argv[n]); 678185087Salfred opt->alt_index = num_id(argv[n + 1], "cfg_index"); 679184610Salfred opt->got_set_alt = 1; 680184610Salfred opt->got_any++; 681184610Salfred n++; 682184610Salfred break; 683188987Sthompsa case T_SET_TEMPLATE: 684201705Sthompsa if (opt->got_set_template) 685201705Sthompsa duplicate_option(argv[n]); 686189110Sthompsa opt->template = get_int(argv[n + 1]); 687188987Sthompsa opt->got_set_template = 1; 688188987Sthompsa opt->got_any++; 689188987Sthompsa n++; 690188987Sthompsa break; 691188987Sthompsa case T_GET_TEMPLATE: 692201705Sthompsa if (opt->got_get_template) 693201705Sthompsa duplicate_option(argv[n]); 694188987Sthompsa opt->got_get_template = 1; 695188987Sthompsa opt->got_any++; 696188987Sthompsa break; 697184610Salfred case T_DUMP_DEVICE_DESC: 698201705Sthompsa if (opt->got_dump_device_desc) 699201705Sthompsa duplicate_option(argv[n]); 700184610Salfred opt->got_dump_device_desc = 1; 701184610Salfred opt->got_any++; 702184610Salfred break; 703184610Salfred case T_DUMP_CURR_CONFIG_DESC: 704201705Sthompsa if (opt->got_dump_curr_config) 705201705Sthompsa duplicate_option(argv[n]); 706184610Salfred opt->got_dump_curr_config = 1; 707184610Salfred opt->got_any++; 708184610Salfred break; 709184610Salfred case T_DUMP_ALL_CONFIG_DESC: 710201705Sthompsa if (opt->got_dump_all_config) 711201705Sthompsa duplicate_option(argv[n]); 712184610Salfred opt->got_dump_all_config = 1; 713184610Salfred opt->got_any++; 714184610Salfred break; 715184610Salfred case T_DUMP_INFO: 716201705Sthompsa if (opt->got_dump_info) 717201705Sthompsa duplicate_option(argv[n]); 718184610Salfred opt->got_dump_info = 1; 719184610Salfred opt->got_any++; 720184610Salfred break; 721185087Salfred case T_DUMP_STRING: 722201705Sthompsa if (opt->got_dump_string) 723201705Sthompsa duplicate_option(argv[n]); 724185087Salfred opt->string_index = num_id(argv[n + 1], "str_index"); 725185087Salfred opt->got_dump_string = 1; 726185087Salfred opt->got_any++; 727185087Salfred n++; 728185087Salfred break; 729184610Salfred case T_SUSPEND: 730201705Sthompsa if (opt->got_suspend) 731201705Sthompsa duplicate_option(argv[n]); 732184610Salfred opt->got_suspend = 1; 733184610Salfred opt->got_any++; 734184610Salfred break; 735184610Salfred case T_RESUME: 736201705Sthompsa if (opt->got_resume) 737201705Sthompsa duplicate_option(argv[n]); 738184610Salfred opt->got_resume = 1; 739184610Salfred opt->got_any++; 740184610Salfred break; 741184610Salfred case T_POWER_OFF: 742201705Sthompsa if (opt->got_power_off) 743201705Sthompsa duplicate_option(argv[n]); 744184610Salfred opt->got_power_off = 1; 745184610Salfred opt->got_any++; 746184610Salfred break; 747184610Salfred case T_POWER_SAVE: 748201705Sthompsa if (opt->got_power_save) 749201705Sthompsa duplicate_option(argv[n]); 750184610Salfred opt->got_power_save = 1; 751184610Salfred opt->got_any++; 752184610Salfred break; 753184610Salfred case T_POWER_ON: 754201705Sthompsa if (opt->got_power_on) 755201705Sthompsa duplicate_option(argv[n]); 756184610Salfred opt->got_power_on = 1; 757184610Salfred opt->got_any++; 758184610Salfred break; 759184610Salfred case T_RESET: 760201705Sthompsa if (opt->got_reset) 761201705Sthompsa duplicate_option(argv[n]); 762184610Salfred opt->got_reset = 1; 763184610Salfred opt->got_any++; 764184610Salfred break; 765184610Salfred case T_LIST: 766201705Sthompsa if (opt->got_list) 767201705Sthompsa duplicate_option(argv[n]); 768184610Salfred opt->got_list = 1; 769184610Salfred opt->got_any++; 770184610Salfred break; 771185087Salfred case T_DO_REQUEST: 772201705Sthompsa if (opt->got_do_request) 773201705Sthompsa duplicate_option(argv[n]); 774185087Salfred LIBUSB20_INIT(LIBUSB20_CONTROL_SETUP, &opt->setup); 775185087Salfred opt->setup.bmRequestType = num_id(argv[n + 1], "bmReqTyp"); 776185087Salfred opt->setup.bRequest = num_id(argv[n + 2], "bReq"); 777185087Salfred opt->setup.wValue = num_id(argv[n + 3], "wVal"); 778185087Salfred opt->setup.wIndex = num_id(argv[n + 4], "wIndex"); 779185087Salfred opt->setup.wLength = num_id(argv[n + 5], "wLen"); 780185087Salfred if (opt->setup.wLength != 0) { 781185087Salfred opt->buffer = malloc(opt->setup.wLength); 782185087Salfred } else { 783185087Salfred opt->buffer = NULL; 784185087Salfred } 785185087Salfred 786185087Salfred n += 5; 787185087Salfred 788185087Salfred if (!(opt->setup.bmRequestType & 789185087Salfred LIBUSB20_ENDPOINT_IN)) { 790185087Salfred /* copy in data */ 791185087Salfred t = (argc - n - 1); 792185087Salfred if (t < opt->setup.wLength) { 793185087Salfred err(1, "request data missing"); 794185087Salfred } 795185087Salfred t = opt->setup.wLength; 796185087Salfred while (t--) { 797185087Salfred ((uint8_t *)opt->buffer)[t] = 798185087Salfred num_id(argv[n + t + 1], "req_data"); 799185087Salfred } 800185087Salfred n += opt->setup.wLength; 801185087Salfred } 802185087Salfred opt->got_do_request = 1; 803185087Salfred opt->got_any++; 804185087Salfred break; 805184610Salfred default: 806184610Salfred usage(); 807184610Salfred break; 808184610Salfred } 809184610Salfred } 810184610Salfred if (opt->got_any) { 811184610Salfred /* flush out last command */ 812184610Salfred flush_command(pbe, opt); 813184610Salfred } else { 814184610Salfred /* list all the devices */ 815184610Salfred opt->got_list = 1; 816184610Salfred opt->got_any++; 817184610Salfred flush_command(pbe, opt); 818184610Salfred } 819184610Salfred /* release data */ 820184610Salfred libusb20_be_free(pbe); 821184610Salfred 822184610Salfred return (0); 823184610Salfred} 824