tools/usb/usbip/src/usbip_attach.c

   1 // SPDX-License-Identifier: GPL-2.0-or-later
   2 /*
   3  * Copyright (C) 2011 matt mooney <mfm@muteddisk.com>
   4  *               2005-2007 Takahiro Hirofuchi
   5  * Copyright (C) 2015-2016 Samsung Electronics
   6  *               Igor Kotrasinski <i.kotrasinsk@samsung.com>
   7  *               Krzysztof Opasiak <k.opasiak@samsung.com>
   8  */
   9
  10 #include <sys/stat.h>
  11
  12 #include <limits.h>
  13 #include <stdint.h>
  14 #include <stdio.h>
  15 #include <string.h>
  16
  17 #include <fcntl.h>
  18 #include <getopt.h>
  19 #include <unistd.h>
  20 #include <errno.h>
  21
  22 #include "vhci_driver.h"
  23 #include "usbip_common.h"
  24 #include "usbip_network.h"
  25 #include "usbip.h"
  26
  27 static const char usbip_attach_usage_string[] =
  28         "usbip attach <args>\n"
  29         "    -r, --remote=<host>      The machine with exported USB devices\n"
  30         "    -b, --busid=<busid>    Busid of the device on <host>\n"
  31         "    -d, --device=<devid>    Id of the virtual UDC on <host>\n";
  32
  33 void usbip_attach_usage(void)
  34 {
  35         printf("usage: %s", usbip_attach_usage_string);
  36 }
  37
  38 #define MAX_BUFF 100
  39 static int record_connection(char *host, char *port, char *busid, int rhport)
  40 {
  41         int fd;
  42         char path[PATH_MAX+1];
  43         char buff[MAX_BUFF+1];
  44         int ret;
  45
  46         ret = mkdir(VHCI_STATE_PATH, 0700);
  47         if (ret < 0) {
  48                 /* if VHCI_STATE_PATH exists, then it better be a directory */
  49                 if (errno == EEXIST) {
  50                         struct stat s;
  51
  52                         ret = stat(VHCI_STATE_PATH, &s);
  53                         if (ret < 0)
  54                                 return -1;
  55                         if (!(s.st_mode & S_IFDIR))
  56                                 return -1;
  57                 } else
  58                         return -1;
  59         }
  60
  61         snprintf(path, PATH_MAX, VHCI_STATE_PATH"/port%d", rhport);
  62
  63         fd = open(path, O_WRONLY|O_CREAT|O_TRUNC, S_IRWXU);
  64         if (fd < 0)
  65                 return -1;
  66
  67         snprintf(buff, MAX_BUFF, "%s %s %s\n",
  68                         host, port, busid);
  69
  70         ret = write(fd, buff, strlen(buff));
  71         if (ret != (ssize_t) strlen(buff)) {
  72                 close(fd);
  73                 return -1;
  74         }
  75
  76         close(fd);
  77
  78         return 0;
  79 }
  80
  81 static int import_device(int sockfd, struct usbip_usb_device *udev)
  82 {
  83         int rc;
  84         int port;
  85         uint32_t speed = udev->speed;
  86
  87         rc = usbip_vhci_driver_open();
  88         if (rc < 0) {
  89                 err("open vhci_driver");
  90                 goto err_out;
  91         }
  92
  93         do {
  94                 port = usbip_vhci_get_free_port(speed);
  95                 if (port < 0) {
  96                         err("no free port");
  97                         goto err_driver_close;
  98                 }
  99
 100                 dbg("got free port %d", port);
 101
 102                 rc = usbip_vhci_attach_device(port, sockfd, udev->busnum,
 103                                               udev->devnum, udev->speed);
 104                 if (rc < 0 && errno != EBUSY) {
 105                         err("import device");
 106                         goto err_driver_close;
 107                 }
 108         } while (rc < 0);
 109
 110         usbip_vhci_driver_close();
 111
 112         return port;
 113
 114 err_driver_close:
 115         usbip_vhci_driver_close();
 116 err_out:
 117         return -1;
 118 }
 119
 120 static int query_import_device(int sockfd, char *busid)
 121 {
 122         int rc;
 123         struct op_import_request request;
 124         struct op_import_reply   reply;
 125         uint16_t code = OP_REP_IMPORT;
 126         int status;
 127
 128         memset(&request, 0, sizeof(request));
 129         memset(&reply, 0, sizeof(reply));
 130
 131         /* send a request */
 132         rc = usbip_net_send_op_common(sockfd, OP_REQ_IMPORT, 0);
 133         if (rc < 0) {
 134                 err("send op_common");
 135                 return -1;
 136         }
 137
 138         strncpy(request.busid, busid, SYSFS_BUS_ID_SIZE-1);
 139
 140         PACK_OP_IMPORT_REQUEST(0, &request);
 141
 142         rc = usbip_net_send(sockfd, (void *) &request, sizeof(request));
 143         if (rc < 0) {
 144                 err("send op_import_request");
 145                 return -1;
 146         }
 147
 148         /* receive a reply */
 149         rc = usbip_net_recv_op_common(sockfd, &code, &status);
 150         if (rc < 0) {
 151                 err("Attach Request for %s failed - %s\n",
 152                     busid, usbip_op_common_status_string(status));
 153                 return -1;
 154         }
 155
 156         rc = usbip_net_recv(sockfd, (void *) &reply, sizeof(reply));
 157         if (rc < 0) {
 158                 err("recv op_import_reply");
 159                 return -1;
 160         }
 161
 162         PACK_OP_IMPORT_REPLY(0, &reply);
 163
 164         /* check the reply */
 165         if (strncmp(reply.udev.busid, busid, SYSFS_BUS_ID_SIZE)) {
 166                 err("recv different busid %s", reply.udev.busid);
 167                 return -1;
 168         }
 169
 170         /* import a device */
 171         return import_device(sockfd, &reply.udev);
 172 }
 173
 174 static int attach_device(char *host, char *busid)
 175 {
 176         int sockfd;
 177         int rc;
 178         int rhport;
 179
 180         sockfd = usbip_net_tcp_connect(host, usbip_port_string);
 181         if (sockfd < 0) {
 182                 err("tcp connect");
 183                 return -1;
 184         }
 185
 186         rhport = query_import_device(sockfd, busid);
 187         if (rhport < 0)
 188                 return -1;
 189
 190         close(sockfd);
 191
 192         rc = record_connection(host, usbip_port_string, busid, rhport);
 193         if (rc < 0) {
 194                 err("record connection");
 195                 return -1;
 196         }
 197
 198         return 0;
 199 }
 200
 201 int usbip_attach(int argc, char *argv[])
 202 {
 203         static const struct option opts[] = {
 204                 { "remote", required_argument, NULL, 'r' },
 205                 { "busid",  required_argument, NULL, 'b' },
 206                 { "device",  required_argument, NULL, 'd' },
 207                 { NULL, 0,  NULL, 0 }
 208         };
 209         char *host = NULL;
 210         char *busid = NULL;
 211         int opt;
 212         int ret = -1;
 213
 214         for (;;) {
 215                 opt = getopt_long(argc, argv, "d:r:b:", opts, NULL);
 216
 217                 if (opt == -1)
 218                         break;
 219
 220                 switch (opt) {
 221                 case 'r':
 222                         host = optarg;
 223                         break;
 224                 case 'd':
 225                 case 'b':
 226                         busid = optarg;
 227                         break;
 228                 default:
 229                         goto err_out;
 230                 }
 231         }
 232
 233         if (!host || !busid)
 234                 goto err_out;
 235
 236         ret = attach_device(host, busid);
 237         goto out;
 238
 239 err_out:
 240         usbip_attach_usage();
 241 out:
 242         return ret;
 243 }