tools/usb/usbip/src/usbip.c

   1 /*
   2  * command structure borrowed from udev
   3  * (git://git.kernel.org/pub/scm/linux/hotplug/udev.git)
   4  *
   5  * Copyright (C) 2011 matt mooney <mfm@muteddisk.com>
   6  *               2005-2007 Takahiro Hirofuchi
   7  *
   8  * This program is free software: you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation, either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program. If not, see <http://www.gnu.org/licenses/>.
  20  */
  21
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24
  25 #include <getopt.h>
  26 #include <syslog.h>
  27
  28 #include "usbip_common.h"
  29 #include "usbip_network.h"
  30 #include "usbip.h"
  31
  32 static int usbip_help(int argc, char *argv[]);
  33 static int usbip_version(int argc, char *argv[]);
  34
  35 static const char usbip_version_string[] = PACKAGE_STRING;
  36
  37 static const char usbip_usage_string[] =
  38         "usbip [--debug] [--log] [--tcp-port PORT] [version]\n"
  39         "             [help] <command> <args>\n";
  40
  41 static void usbip_usage(void)
  42 {
  43         printf("usage: %s", usbip_usage_string);
  44 }
  45
  46 struct command {
  47         const char *name;
  48         int (*fn)(int argc, char *argv[]);
  49         const char *help;
  50         void (*usage)(void);
  51 };
  52
  53 static const struct command cmds[] = {
  54         {
  55                 .name  = "help",
  56                 .fn    = usbip_help,
  57                 .help  = NULL,
  58                 .usage = NULL
  59         },
  60         {
  61                 .name  = "version",
  62                 .fn    = usbip_version,
  63                 .help  = NULL,
  64                 .usage = NULL
  65         },
  66         {
  67                 .name  = "attach",
  68                 .fn    = usbip_attach,
  69                 .help  = "Attach a remote USB device",
  70                 .usage = usbip_attach_usage
  71         },
  72         {
  73                 .name  = "detach",
  74                 .fn    = usbip_detach,
  75                 .help  = "Detach a remote USB device",
  76                 .usage = usbip_detach_usage
  77         },
  78         {
  79                 .name  = "list",
  80                 .fn    = usbip_list,
  81                 .help  = "List exportable or local USB devices",
  82                 .usage = usbip_list_usage
  83         },
  84         {
  85                 .name  = "bind",
  86                 .fn    = usbip_bind,
  87                 .help  = "Bind device to " USBIP_HOST_DRV_NAME ".ko",
  88                 .usage = usbip_bind_usage
  89         },
  90         {
  91                 .name  = "unbind",
  92                 .fn    = usbip_unbind,
  93                 .help  = "Unbind device from " USBIP_HOST_DRV_NAME ".ko",
  94                 .usage = usbip_unbind_usage
  95         },
  96         {
  97                 .name  = "port",
  98                 .fn    = usbip_port_show,
  99                 .help  = "Show imported USB devices",
 100                 .usage = NULL
 101         },
 102         { NULL, NULL, NULL, NULL }
 103 };
 104
 105 static int usbip_help(int argc, char *argv[])
 106 {
 107         const struct command *cmd;
 108         int i;
 109         int ret = 0;
 110
 111         if (argc > 1 && argv++) {
 112                 for (i = 0; cmds[i].name != NULL; i++)
 113                         if (!strcmp(cmds[i].name, argv[0]) && cmds[i].usage) {
 114                                 cmds[i].usage();
 115                                 goto done;
 116                         }
 117                 ret = -1;
 118         }
 119
 120         usbip_usage();
 121         printf("\n");
 122         for (cmd = cmds; cmd->name != NULL; cmd++)
 123                 if (cmd->help != NULL)
 124                         printf("  %-10s %s\n", cmd->name, cmd->help);
 125         printf("\n");
 126 done:
 127         return ret;
 128 }
 129
 130 static int usbip_version(int argc, char *argv[])
 131 {
 132         (void) argc;
 133         (void) argv;
 134
 135         printf(PROGNAME " (%s)\n", usbip_version_string);
 136         return 0;
 137 }
 138
 139 static int run_command(const struct command *cmd, int argc, char *argv[])
 140 {
 141         dbg("running command: `%s'", cmd->name);
 142         return cmd->fn(argc, argv);
 143 }
 144
 145 int main(int argc, char *argv[])
 146 {
 147         static const struct option opts[] = {
 148                 { "debug",    no_argument,       NULL, 'd' },
 149                 { "log",      no_argument,       NULL, 'l' },
 150                 { "tcp-port", required_argument, NULL, 't' },
 151                 { NULL,       0,                 NULL,  0  }
 152         };
 153
 154         char *cmd;
 155         int opt;
 156         int i, rc = -1;
 157
 158         usbip_use_stderr = 1;
 159         opterr = 0;
 160         for (;;) {
 161                 opt = getopt_long(argc, argv, "+dlt:", opts, NULL);
 162
 163                 if (opt == -1)
 164                         break;
 165
 166                 switch (opt) {
 167                 case 'd':
 168                         usbip_use_debug = 1;
 169                         break;
 170                 case 'l':
 171                         usbip_use_syslog = 1;
 172                         openlog("", LOG_PID, LOG_USER);
 173                         break;
 174                 case 't':
 175                         usbip_setup_port_number(optarg);
 176                         break;
 177                 case '?':
 178                         printf("usbip: invalid option\n");
 179                 default:
 180                         usbip_usage();
 181                         goto out;
 182                 }
 183         }
 184
 185         cmd = argv[optind];
 186         if (cmd) {
 187                 for (i = 0; cmds[i].name != NULL; i++)
 188                         if (!strcmp(cmds[i].name, cmd)) {
 189                                 argc -= optind;
 190                                 argv += optind;
 191                                 optind = 0;
 192                                 rc = run_command(&cmds[i], argc, argv);
 193                                 goto out;
 194                         }
 195         }
 196
 197         /* invalid command */
 198         usbip_help(0, NULL);
 199 out:
 200         return (rc > -1 ? EXIT_SUCCESS : EXIT_FAILURE);
 201 }