freebsd-dev/usr.sbin/usbconfig/usbconfig.c
2009-03-17 21:21:33 +00:00

712 lines
16 KiB
C

/* $FreeBSD$ */
/*-
* Copyright (c) 2008 Hans Petter Selasky. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <err.h>
#include <string.h>
#include <pwd.h>
#include <grp.h>
#include <errno.h>
#include <ctype.h>
#include <libusb20_desc.h>
#include <libusb20.h>
#include "dump.h"
struct options {
const char *quirkname;
void *buffer;
int template;
gid_t gid;
uid_t uid;
mode_t mode;
uint32_t got_any;
struct LIBUSB20_CONTROL_SETUP_DECODED setup;
uint16_t bus;
uint16_t addr;
uint16_t iface;
uint16_t vid;
uint16_t pid;
uint16_t lo_rev; /* inclusive */
uint16_t hi_rev; /* inclusive */
uint8_t string_index;
uint8_t config_index;
uint8_t alt_index;
uint8_t got_list:1;
uint8_t got_bus:1;
uint8_t got_addr:1;
uint8_t got_iface:1;
uint8_t got_set_config:1;
uint8_t got_set_alt:1;
uint8_t got_set_template:1;
uint8_t got_get_template:1;
uint8_t got_suspend:1;
uint8_t got_resume:1;
uint8_t got_reset:1;
uint8_t got_power_off:1;
uint8_t got_power_save:1;
uint8_t got_power_on:1;
uint8_t got_dump_device_quirks:1;
uint8_t got_dump_quirk_names:1;
uint8_t got_dump_device_desc:1;
uint8_t got_dump_curr_config:1;
uint8_t got_dump_all_config:1;
uint8_t got_dump_info:1;
uint8_t got_show_iface_driver:1;
uint8_t got_remove_device_quirk:1;
uint8_t got_add_device_quirk:1;
uint8_t got_dump_string:1;
uint8_t got_do_request:1;
};
struct token {
const char *name;
uint8_t value;
uint8_t narg;
};
enum {
T_UNIT,
T_ADDR,
T_IFACE,
T_SET_CONFIG,
T_SET_ALT,
T_SET_TEMPLATE,
T_GET_TEMPLATE,
T_ADD_DEVICE_QUIRK,
T_REMOVE_DEVICE_QUIRK,
T_SHOW_IFACE_DRIVER,
T_DUMP_QUIRK_NAMES,
T_DUMP_DEVICE_QUIRKS,
T_DUMP_DEVICE_DESC,
T_DUMP_CURR_CONFIG_DESC,
T_DUMP_ALL_CONFIG_DESC,
T_DUMP_STRING,
T_DUMP_INFO,
T_SUSPEND,
T_RESUME,
T_POWER_OFF,
T_POWER_SAVE,
T_POWER_ON,
T_RESET,
T_LIST,
T_DO_REQUEST,
};
static struct options options;
static const struct token token[] = {
{"-u", T_UNIT, 1},
{"-a", T_ADDR, 1},
{"-i", T_IFACE, 1},
{"set_config", T_SET_CONFIG, 1},
{"set_alt", T_SET_ALT, 1},
{"set_template", T_SET_TEMPLATE, 1},
{"get_template", T_GET_TEMPLATE, 0},
{"add_dev_quirk_vplh", T_ADD_DEVICE_QUIRK, 5},
{"remove_dev_quirk_vplh", T_REMOVE_DEVICE_QUIRK, 5},
{"dump_quirk_names", T_DUMP_QUIRK_NAMES, 0},
{"dump_device_quirks", T_DUMP_DEVICE_QUIRKS, 0},
{"dump_device_desc", T_DUMP_DEVICE_DESC, 0},
{"dump_curr_config_desc", T_DUMP_CURR_CONFIG_DESC, 0},
{"dump_all_config_desc", T_DUMP_ALL_CONFIG_DESC, 0},
{"dump_string", T_DUMP_STRING, 1},
{"dump_info", T_DUMP_INFO, 0},
{"show_ifdrv", T_SHOW_IFACE_DRIVER, 0},
{"suspend", T_SUSPEND, 0},
{"resume", T_RESUME, 0},
{"power_off", T_POWER_OFF, 0},
{"power_save", T_POWER_SAVE, 0},
{"power_on", T_POWER_ON, 0},
{"reset", T_RESET, 0},
{"list", T_LIST, 0},
{"do_request", T_DO_REQUEST, 5},
};
static void
be_dev_remove_quirk(struct libusb20_backend *pbe,
uint16_t vid, uint16_t pid, uint16_t lorev, uint16_t hirev,
const char *str)
{
struct libusb20_quirk q;
int error;
memset(&q, 0, sizeof(q));
q.vid = vid;
q.pid = pid;
q.bcdDeviceLow = lorev;
q.bcdDeviceHigh = hirev;
strlcpy(q.quirkname, str, sizeof(q.quirkname));
error = libusb20_be_remove_dev_quirk(pbe, &q);
if (error) {
printf("Removing quirk '%s' failed, continuing.\n", str);
}
return;
}
static void
be_dev_add_quirk(struct libusb20_backend *pbe,
uint16_t vid, uint16_t pid, uint16_t lorev, uint16_t hirev,
const char *str)
{
struct libusb20_quirk q;
int error;
memset(&q, 0, sizeof(q));
q.vid = vid;
q.pid = pid;
q.bcdDeviceLow = lorev;
q.bcdDeviceHigh = hirev;
strlcpy(q.quirkname, str, sizeof(q.quirkname));
error = libusb20_be_add_dev_quirk(pbe, &q);
if (error) {
printf("Adding quirk '%s' failed, continuing.\n", str);
}
return;
}
static uint8_t
get_token(const char *str, uint8_t narg)
{
uint8_t n;
for (n = 0; n != (sizeof(token) / sizeof(token[0])); n++) {
if (strcasecmp(str, token[n].name) == 0) {
if (token[n].narg > narg) {
/* too few arguments */
break;
}
return (token[n].value);
}
}
return (0 - 1);
}
static uid_t
num_id(const char *name, const char *type)
{
uid_t val;
char *ep;
errno = 0;
val = strtoul(name, &ep, 0);
if (errno) {
err(1, "%s", name);
}
if (*ep != '\0') {
errx(1, "%s: illegal %s name", name, type);
}
return (val);
}
static int
get_int(const char *s)
{
int val;
char *ep;
errno = 0;
val = strtoul(s, &ep, 0);
if (errno) {
err(1, "%s", s);
}
if (*ep != '\0') {
errx(1, "illegal number: %s", s);
}
return val;
}
static void
usage(void)
{
printf(""
"usbconfig - configure the USB subsystem" "\n"
"usage: usbconfig -u <busnum> -a <devaddr> -i <ifaceindex> [cmds...]" "\n"
"commands:" "\n"
" set_config <cfg_index>" "\n"
" set_alt <alt_index>" "\n"
" set_template <template>" "\n"
" get_template" "\n"
" add_dev_quirk_vplh <vid> <pid> <lo_rev> <hi_rev> <quirk>" "\n"
" remove_dev_quirk_vplh <vid> <pid> <lo_rev> <hi_rev> <quirk>" "\n"
" dump_quirk_names" "\n"
" dump_device_quirks" "\n"
" dump_device_desc" "\n"
" dump_curr_config_desc" "\n"
" dump_all_config_desc" "\n"
" dump_string <index>" "\n"
" dump_info" "\n"
" show_ifdrv" "\n"
" suspend" "\n"
" resume" "\n"
" power_off" "\n"
" power_save" "\n"
" power_on" "\n"
" reset" "\n"
" list" "\n"
" do_request <bmReqTyp> <bReq> <wVal> <wIdx> <wLen> <data...>" "\n"
);
exit(1);
}
static void
reset_options(struct options *opt)
{
if (opt->buffer)
free(opt->buffer);
memset(opt, 0, sizeof(*opt));
return;
}
static void
flush_command(struct libusb20_backend *pbe, struct options *opt)
{
struct libusb20_device *pdev = NULL;
uint32_t matches = 0;
uint8_t dump_any;
/* check for invalid option combinations */
if ((opt->got_suspend +
opt->got_resume +
opt->got_reset +
opt->got_set_config +
opt->got_set_alt +
opt->got_power_save +
opt->got_power_on +
opt->got_power_off) > 1) {
err(1, "can only specify one of 'set_config', "
"'set_alt', 'reset', 'suspend', 'resume', "
"'power_save', 'power_on' and 'power_off' "
"at the same time!");
}
if (opt->got_dump_quirk_names) {
opt->got_any--;
dump_be_quirk_names(pbe);
}
if (opt->got_dump_device_quirks) {
opt->got_any--;
dump_be_dev_quirks(pbe);
}
if (opt->got_remove_device_quirk) {
opt->got_any--;
be_dev_remove_quirk(pbe,
opt->vid, opt->pid, opt->lo_rev, opt->hi_rev, opt->quirkname);
}
if (opt->got_add_device_quirk) {
opt->got_any--;
be_dev_add_quirk(pbe,
opt->vid, opt->pid, opt->lo_rev, opt->hi_rev, opt->quirkname);
}
if (opt->got_set_template) {
opt->got_any--;
if (libusb20_be_set_template(pbe, opt->template)) {
printf("Setting USB template %u failed, "
"continuing.\n", opt->template);
}
}
if (opt->got_get_template) {
opt->got_any--;
if (libusb20_be_get_template(pbe, &opt->template))
printf("USB template: <unknown>\n");
else
printf("USB template: %u\n", opt->template);
}
if (opt->got_any == 0) {
/*
* do not scan through all the devices if there are no valid
* options
*/
goto done;
}
while ((pdev = libusb20_be_device_foreach(pbe, pdev))) {
if (opt->got_bus &&
(libusb20_dev_get_bus_number(pdev) != opt->bus)) {
continue;
}
if (opt->got_addr &&
(libusb20_dev_get_address(pdev) != opt->addr)) {
continue;
}
matches++;
if (libusb20_dev_open(pdev, 0)) {
err(1, "could not open device");
}
if (opt->got_dump_string) {
char *pbuf;
pbuf = malloc(256);
if (pbuf == NULL) {
err(1, "out of memory");
}
if (libusb20_dev_req_string_simple_sync(pdev,
opt->string_index, pbuf, 256)) {
printf("STRING_0x%02x = <read error>\n",
opt->string_index);
} else {
printf("STRING_0x%02x = <%s>\n",
opt->string_index, pbuf);
}
free(pbuf);
}
if (opt->got_do_request) {
uint16_t actlen;
uint16_t t;
if (libusb20_dev_request_sync(pdev, &opt->setup,
opt->buffer, &actlen, 5000 /* 5 seconds */ , 0)) {
printf("REQUEST = <ERROR>\n");
} else if (!(opt->setup.bmRequestType &
LIBUSB20_ENDPOINT_IN)) {
printf("REQUEST = <OK>\n");
} else {
t = actlen;
printf("REQUEST = <");
for (t = 0; t != actlen; t++) {
printf("0x%02x%s",
((uint8_t *)opt->buffer)[t],
(t == (actlen - 1)) ? "" : " ");
}
printf("><");
for (t = 0; t != actlen; t++) {
char c;
c = ((uint8_t *)opt->buffer)[t];
if ((c != '<') &&
(c != '>') && isprint(c)) {
putchar(c);
}
}
printf(">\n");
}
}
if (opt->got_set_config) {
if (libusb20_dev_set_config_index(pdev,
opt->config_index)) {
err(1, "could not set config index");
}
}
if (opt->got_set_alt) {
if (libusb20_dev_set_alt_index(pdev, opt->iface,
opt->alt_index)) {
err(1, "could not set alternate setting");
}
}
if (opt->got_reset) {
if (libusb20_dev_reset(pdev)) {
err(1, "could not reset device");
}
}
if (opt->got_suspend) {
if (libusb20_dev_set_power_mode(pdev,
LIBUSB20_POWER_SUSPEND)) {
err(1, "could not set suspend");
}
}
if (opt->got_resume) {
if (libusb20_dev_set_power_mode(pdev,
LIBUSB20_POWER_RESUME)) {
err(1, "could not set resume");
}
}
if (opt->got_power_off) {
if (libusb20_dev_set_power_mode(pdev,
LIBUSB20_POWER_OFF)) {
err(1, "could not set power OFF");
}
}
if (opt->got_power_save) {
if (libusb20_dev_set_power_mode(pdev,
LIBUSB20_POWER_SAVE)) {
err(1, "could not set power SAVE");
}
}
if (opt->got_power_on) {
if (libusb20_dev_set_power_mode(pdev,
LIBUSB20_POWER_ON)) {
err(1, "could not set power ON");
}
}
dump_any =
(opt->got_dump_device_desc ||
opt->got_dump_curr_config ||
opt->got_dump_all_config ||
opt->got_dump_info);
if (opt->got_list || dump_any) {
dump_device_info(pdev,
opt->got_show_iface_driver);
}
if (opt->got_dump_device_desc) {
printf("\n");
dump_device_desc(pdev);
}
if (opt->got_dump_all_config) {
printf("\n");
dump_config(pdev, 1);
} else if (opt->got_dump_curr_config) {
printf("\n");
dump_config(pdev, 0);
}
if (dump_any) {
printf("\n");
}
if (libusb20_dev_close(pdev)) {
err(1, "could not close device");
}
}
if (matches == 0) {
printf("No device match or lack of permissions.\n");
}
done:
reset_options(opt);
return;
}
int
main(int argc, char **argv)
{
struct libusb20_backend *pbe;
struct options *opt = &options;
int n;
int t;
if (argc < 1) {
usage();
}
pbe = libusb20_be_alloc_default();
if (pbe == NULL)
err(1, "could not access USB backend\n");
for (n = 1; n != argc; n++) {
/* get number of additional options */
t = (argc - n - 1);
if (t > 255)
t = 255;
switch (get_token(argv[n], t)) {
case T_ADD_DEVICE_QUIRK:
if (opt->got_add_device_quirk) {
flush_command(pbe, opt);
}
opt->vid = num_id(argv[n + 1], "Vendor ID");
opt->pid = num_id(argv[n + 2], "Product ID");
opt->lo_rev = num_id(argv[n + 3], "Low Revision");
opt->hi_rev = num_id(argv[n + 4], "High Revision");
opt->quirkname = argv[n + 5];
n += 5;
opt->got_add_device_quirk = 1;
opt->got_any++;
break;
case T_REMOVE_DEVICE_QUIRK:
if (opt->got_remove_device_quirk) {
flush_command(pbe, opt);
}
opt->vid = num_id(argv[n + 1], "Vendor ID");
opt->pid = num_id(argv[n + 2], "Product ID");
opt->lo_rev = num_id(argv[n + 3], "Low Revision");
opt->hi_rev = num_id(argv[n + 4], "High Revision");
opt->quirkname = argv[n + 5];
n += 5;
opt->got_remove_device_quirk = 1;
opt->got_any++;
break;
case T_DUMP_QUIRK_NAMES:
opt->got_dump_quirk_names = 1;
opt->got_any++;
break;
case T_DUMP_DEVICE_QUIRKS:
opt->got_dump_device_quirks = 1;
opt->got_any++;
break;
case T_SHOW_IFACE_DRIVER:
opt->got_show_iface_driver = 1;
break;
case T_UNIT:
if (opt->got_any) {
/* allow multiple commands on the same line */
flush_command(pbe, opt);
}
opt->bus = num_id(argv[n + 1], "busnum");
opt->got_bus = 1;
n++;
break;
case T_ADDR:
opt->addr = num_id(argv[n + 1], "addr");
opt->got_addr = 1;
n++;
break;
case T_IFACE:
opt->iface = num_id(argv[n + 1], "iface");
opt->got_iface = 1;
n++;
break;
case T_SET_CONFIG:
opt->config_index = num_id(argv[n + 1], "cfg_index");
opt->got_set_config = 1;
opt->got_any++;
n++;
break;
case T_SET_ALT:
opt->alt_index = num_id(argv[n + 1], "cfg_index");
opt->got_set_alt = 1;
opt->got_any++;
n++;
break;
case T_SET_TEMPLATE:
opt->template = get_int(argv[n + 1]);
opt->got_set_template = 1;
opt->got_any++;
n++;
break;
case T_GET_TEMPLATE:
opt->got_get_template = 1;
opt->got_any++;
break;
case T_DUMP_DEVICE_DESC:
opt->got_dump_device_desc = 1;
opt->got_any++;
break;
case T_DUMP_CURR_CONFIG_DESC:
opt->got_dump_curr_config = 1;
opt->got_any++;
break;
case T_DUMP_ALL_CONFIG_DESC:
opt->got_dump_all_config = 1;
opt->got_any++;
break;
case T_DUMP_INFO:
opt->got_dump_info = 1;
opt->got_any++;
break;
case T_DUMP_STRING:
if (opt->got_dump_string) {
flush_command(pbe, opt);
}
opt->string_index = num_id(argv[n + 1], "str_index");
opt->got_dump_string = 1;
opt->got_any++;
n++;
break;
case T_SUSPEND:
opt->got_suspend = 1;
opt->got_any++;
break;
case T_RESUME:
opt->got_resume = 1;
opt->got_any++;
break;
case T_POWER_OFF:
opt->got_power_off = 1;
opt->got_any++;
break;
case T_POWER_SAVE:
opt->got_power_save = 1;
opt->got_any++;
break;
case T_POWER_ON:
opt->got_power_on = 1;
opt->got_any++;
break;
case T_RESET:
opt->got_reset = 1;
opt->got_any++;
break;
case T_LIST:
opt->got_list = 1;
opt->got_any++;
break;
case T_DO_REQUEST:
if (opt->got_do_request) {
flush_command(pbe, opt);
}
LIBUSB20_INIT(LIBUSB20_CONTROL_SETUP, &opt->setup);
opt->setup.bmRequestType = num_id(argv[n + 1], "bmReqTyp");
opt->setup.bRequest = num_id(argv[n + 2], "bReq");
opt->setup.wValue = num_id(argv[n + 3], "wVal");
opt->setup.wIndex = num_id(argv[n + 4], "wIndex");
opt->setup.wLength = num_id(argv[n + 5], "wLen");
if (opt->setup.wLength != 0) {
opt->buffer = malloc(opt->setup.wLength);
} else {
opt->buffer = NULL;
}
n += 5;
if (!(opt->setup.bmRequestType &
LIBUSB20_ENDPOINT_IN)) {
/* copy in data */
t = (argc - n - 1);
if (t < opt->setup.wLength) {
err(1, "request data missing");
}
t = opt->setup.wLength;
while (t--) {
((uint8_t *)opt->buffer)[t] =
num_id(argv[n + t + 1], "req_data");
}
n += opt->setup.wLength;
}
opt->got_do_request = 1;
opt->got_any++;
break;
default:
usage();
break;
}
}
if (opt->got_any) {
/* flush out last command */
flush_command(pbe, opt);
} else {
/* list all the devices */
opt->got_list = 1;
opt->got_any++;
flush_command(pbe, opt);
}
/* release data */
libusb20_be_free(pbe);
return (0);
}