123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656 |
- /*
- * Copyright (C) 2009 bsdroid project
- * Alexey Tarasov <tarasov@dodologics.com>
- *
- * Copyright (C) 2007 The Android Open Source Project
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- #include <sys/endian.h>
- #include <sys/ioctl.h>
- #include <sys/types.h>
- #include <sys/uio.h>
- #include <err.h>
- #include <errno.h>
- #include <poll.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <strings.h>
- #include <string.h>
- #include <sysexits.h>
- #include <unistd.h>
- #include <libusb.h>
- #include "sysdeps.h"
- #define TRACE_TAG TRACE_USB
- #include "adb.h"
- static adb_mutex_t usb_lock = ADB_MUTEX_INITIALIZER;
- static libusb_context *ctx = NULL;
- struct usb_handle
- {
- usb_handle *prev;
- usb_handle *next;
- libusb_device *dev;
- libusb_device_handle *devh;
- int interface;
- uint8_t dev_bus;
- uint8_t dev_addr;
- int zero_mask;
- unsigned char end_point_address[2];
- char serial[128];
- adb_cond_t notify;
- adb_mutex_t lock;
- };
- static struct usb_handle handle_list = {
- .prev = &handle_list,
- .next = &handle_list,
- };
- void
- usb_cleanup()
- {
- libusb_exit(ctx);
- }
- void
- report_bulk_libusb_error(int r)
- {
- switch (r) {
- case LIBUSB_ERROR_TIMEOUT:
- D("Transfer timeout\n");
- break;
- case LIBUSB_ERROR_PIPE:
- D("Control request is not supported\n");
- break;
- case LIBUSB_ERROR_OVERFLOW:
- D("Device offered more data\n");
- break;
- case LIBUSB_ERROR_NO_DEVICE :
- D("Device was disconnected\n");
- break;
- default:
- D("Error %d during transfer\n", r);
- break;
- };
- }
- static int
- usb_bulk_write(usb_handle *uh, const void *data, int len)
- {
- int r = 0;
- int transferred = 0;
- r = libusb_bulk_transfer(uh->devh, uh->end_point_address[1], (void *)data, len,
- &transferred, 0);
- if (r != 0) {
- D("usb_bulk_write(): ");
- report_bulk_libusb_error(r);
- return r;
- }
- return (transferred);
- }
- static int
- usb_bulk_read(usb_handle *uh, void *data, int len)
- {
- int r = 0;
- int transferred = 0;
- r = libusb_bulk_transfer(uh->devh, uh->end_point_address[0], data, len,
- &transferred, 0);
- if (r != 0) {
- D("usb_bulk_read(): ");
- report_bulk_libusb_error(r);
- return r;
- }
- return (transferred);
- }
- int
- usb_write(struct usb_handle *uh, const void *_data, int len)
- {
- unsigned char *data = (unsigned char*) _data;
- int n;
- int need_zero = 0;
- if (uh->zero_mask == 1) {
- if (!(len & uh->zero_mask)) {
- need_zero = 1;
- }
- }
- D("usb_write(): %p:%d -> transport %p\n", _data, len, uh);
- while (len > 0) {
- int xfer = (len > 4096) ? 4096 : len;
- n = usb_bulk_write(uh, data, xfer);
- if (n != xfer) {
- D("usb_write(): failed for transport %p (%d bytes left)\n", uh, len);
- return -1;
- }
- len -= xfer;
- data += xfer;
- }
- if (need_zero){
- n = usb_bulk_write(uh, _data, 0);
- if (n < 0) {
- D("usb_write(): failed to finish operation for transport %p\n", uh);
- }
- return n;
- }
- return 0;
- }
- int
- usb_read(struct usb_handle *uh, void *_data, int len)
- {
- unsigned char *data = (unsigned char*) _data;
- int n;
- D("usb_read(): %p:%d <- transport %p\n", _data, len, uh);
- while (len > 0) {
- int xfer = (len > 4096) ? 4096 : len;
- n = usb_bulk_read(uh, data, xfer);
- if (n != xfer) {
- if (n > 0) {
- data += n;
- len -= n;
- continue;
- }
- D("usb_read(): failed for transport %p (%d bytes left)\n", uh, len);
- return -1;
- }
- len -= xfer;
- data += xfer;
- }
- return 0;
- }
- int
- usb_close(struct usb_handle *h)
- {
- D("usb_close(): closing transport %p\n", h);
- adb_mutex_lock(&usb_lock);
- h->next->prev = h->prev;
- h->prev->next = h->next;
- h->prev = NULL;
- h->next = NULL;
- libusb_release_interface(h->devh, h->interface);
- libusb_close(h->devh);
- libusb_unref_device(h->dev);
- adb_mutex_unlock(&usb_lock);
- free(h);
- return (0);
- }
- void usb_kick(struct usb_handle *h)
- {
- D("usb_cick(): kicking transport %p\n", h);
- adb_mutex_lock(&h->lock);
- unregister_usb_transport(h);
- adb_mutex_unlock(&h->lock);
- h->next->prev = h->prev;
- h->prev->next = h->next;
- h->prev = NULL;
- h->next = NULL;
- libusb_release_interface(h->devh, h->interface);
- libusb_close(h->devh);
- libusb_unref_device(h->dev);
- free(h);
- }
- int
- check_usb_interface(libusb_interface *interface,
- libusb_device_descriptor *desc,
- struct usb_handle *uh)
- {
- int e;
- if (interface->num_altsetting == 0) {
- D("check_usb_interface(): No interface settings\n");
- return -1;
- }
- libusb_interface_descriptor *idesc = &interface->altsetting[0];
- if (idesc->bNumEndpoints != 2) {
- D("check_usb_interface(): Interface have not 2 endpoints, ignoring\n");
- return -1;
- }
- for (e = 0; e < idesc->bNumEndpoints; e++) {
- libusb_endpoint_descriptor *edesc = &idesc->endpoint[e];
- if (edesc->bmAttributes != LIBUSB_TRANSFER_TYPE_BULK) {
- D("check_usb_interface(): Endpoint (%u) is not bulk (%u), ignoring\n",
- edesc->bmAttributes, LIBUSB_TRANSFER_TYPE_BULK);
- return -1;
- }
- if (edesc->bEndpointAddress & LIBUSB_ENDPOINT_IN)
- uh->end_point_address[0] = edesc->bEndpointAddress;
- else
- uh->end_point_address[1] = edesc->bEndpointAddress;
- /* aproto 01 needs 0 termination */
- if (idesc->bInterfaceProtocol == 0x01) {
- uh->zero_mask = edesc->wMaxPacketSize - 1;
- D("check_usb_interface(): Forced Android interface protocol v.1\n");
- }
- }
- D("check_usb_interface(): Device: %04x:%04x "
- "iclass: %x, isclass: %x, iproto: %x ep: %x/%x-> ",
- desc->idVendor, desc->idProduct, idesc->bInterfaceClass,
- idesc->bInterfaceSubClass, idesc->bInterfaceProtocol,
- uh->end_point_address[0], uh->end_point_address[1]);
- if (!is_adb_interface(desc->idVendor, desc->idProduct,
- idesc->bInterfaceClass, idesc->bInterfaceSubClass,
- idesc->bInterfaceProtocol))
- {
- D("not matches\n");
- return -1;
- }
- D("matches\n");
- return 1;
- }
- int
- check_usb_interfaces(libusb_config_descriptor *config,
- libusb_device_descriptor *desc, struct usb_handle *uh)
- {
- int i;
- for (i = 0; i < config->bNumInterfaces; ++i) {
- if (check_usb_interface(&config->interface[i], desc, uh) != -1) {
- /* found some interface and saved information about it */
- D("check_usb_interfaces(): Interface %d of %04x:%04x "
- "matches Android device\n", i, desc->idVendor,
- desc->idProduct);
- return i;
- }
- }
- return -1;
- }
- int
- register_device(struct usb_handle *uh, const char *serial)
- {
- D("register_device(): Registering %p [%s] as USB transport\n",
- uh, serial);
- struct usb_handle *usb= NULL;
- usb = calloc(1, sizeof(struct usb_handle));
- memcpy(usb, uh, sizeof(struct usb_handle));
- strcpy(usb->serial, uh->serial);
- adb_cond_init(&usb->notify, 0);
- adb_mutex_init(&usb->lock, 0);
- adb_mutex_lock(&usb_lock);
- usb->next = &handle_list;
- usb->prev = handle_list.prev;
- usb->prev->next = usb;
- usb->next->prev = usb;
- adb_mutex_unlock(&usb_lock);
- register_usb_transport(usb, serial, NULL, 1);
- return (1);
- }
- int
- already_registered(usb_handle *uh)
- {
- struct usb_handle *usb= NULL;
- int exists = 0;
- adb_mutex_lock(&usb_lock);
- for (usb = handle_list.next; usb != &handle_list; usb = usb->next) {
- if ((usb->dev_bus == uh->dev_bus) &&
- (usb->dev_addr == uh->dev_addr))
- {
- exists = 1;
- break;
- }
- }
- adb_mutex_unlock(&usb_lock);
- return exists;
- }
- void
- check_device(libusb_device *dev)
- {
- struct usb_handle uh;
- int i = 0;
- int found = -1;
- char serial[256] = {0};
- libusb_device_descriptor desc;
- libusb_config_descriptor *config = NULL;
- int r = libusb_get_device_descriptor(dev, &desc);
- if (r != LIBUSB_SUCCESS) {
- D("check_device(): Failed to get device descriptor\n");
- return;
- }
- if ((desc.idVendor == 0) && (desc.idProduct == 0))
- return;
- D("check_device(): Probing usb device %04x:%04x\n",
- desc.idVendor, desc.idProduct);
- if (!is_adb_interface (desc.idVendor, desc.idProduct,
- ADB_CLASS, ADB_SUBCLASS, ADB_PROTOCOL))
- {
- D("check_device(): Ignored due unknown vendor id\n");
- return;
- }
- uh.dev_bus = libusb_get_bus_number(dev);
- uh.dev_addr = libusb_get_device_address(dev);
- if (already_registered(&uh)) {
- D("check_device(): Device (bus: %d, address: %d) "
- "is already registered\n", uh.dev_bus, uh.dev_addr);
- return;
- }
- D("check_device(): Device bus: %d, address: %d\n",
- uh.dev_bus, uh.dev_addr);
- r = libusb_get_active_config_descriptor(dev, &config);
- if (r != 0) {
- if (r == LIBUSB_ERROR_NOT_FOUND) {
- D("check_device(): Device %4x:%4x is unconfigured\n",
- desc.idVendor, desc.idProduct);
- return;
- }
- D("check_device(): Failed to get configuration for %4x:%4x\n",
- desc.idVendor, desc.idProduct);
- return;
- }
- if (config == NULL) {
- D("check_device(): Sanity check failed after "
- "getting active config\n");
- return;
- }
- if (config->interface != NULL) {
- found = check_usb_interfaces(config, &desc, &uh);
- }
- /* not needed anymore */
- libusb_free_config_descriptor(config);
- r = libusb_open(dev, &uh.devh);
- uh.dev = dev;
- if (r != 0) {
- switch (r) {
- case LIBUSB_ERROR_NO_MEM:
- D("check_device(): Memory allocation problem\n");
- break;
- case LIBUSB_ERROR_ACCESS:
- D("check_device(): Permissions problem, "
- "current user priveleges are messed up?\n");
- break;
- case LIBUSB_ERROR_NO_DEVICE:
- D("check_device(): Device disconected, bad cable?\n");
- break;
- default:
- D("check_device(): libusb triggered error %d\n", r);
- }
- // skip rest
- found = -1;
- }
- if (found >= 0) {
- D("check_device(): Device matches Android interface\n");
- // read the device's serial number
- memset(serial, 0, sizeof(serial));
- uh.interface = found;
- r = libusb_claim_interface(uh.devh, uh.interface);
- if (r < 0) {
- D("check_device(): Failed to claim interface %d\n",
- uh.interface);
- goto fail;
- }
- if (desc.iSerialNumber) {
- // reading serial
- uint16_t buffer[128] = {0};
- uint16_t languages[128] = {0};
- int languageCount = 0;
- memset(languages, 0, sizeof(languages));
- r = libusb_control_transfer(uh.devh,
- LIBUSB_ENDPOINT_IN | LIBUSB_REQUEST_TYPE_STANDARD | LIBUSB_RECIPIENT_DEVICE,
- LIBUSB_REQUEST_GET_DESCRIPTOR, LIBUSB_DT_STRING << 8,
- 0, (uint8_t *)languages, sizeof(languages), 0);
- if (r <= 0) {
- D("check_device(): Failed to get languages count\n");
- goto fail;
- }
- languageCount = (r - 2) / 2;
- for (i = 1; i <= languageCount; ++i) {
- memset(buffer, 0, sizeof(buffer));
- r = libusb_control_transfer(uh.devh,
- LIBUSB_ENDPOINT_IN | LIBUSB_REQUEST_TYPE_STANDARD | LIBUSB_RECIPIENT_DEVICE,
- LIBUSB_REQUEST_GET_DESCRIPTOR, (LIBUSB_DT_STRING << 8) | desc.iSerialNumber,
- languages[i], (uint8_t *)buffer, sizeof(buffer), 0);
- if (r > 0) { /* converting serial */
- int j = 0;
- r /= 2;
- for (j = 1; j < r; ++j)
- serial[j - 1] = buffer[j];
- serial[j - 1] = '\0';
- break; /* languagesCount cycle */
- }
- }
- if (register_device(&uh, serial) == 0) {
- D("check_device(): Failed to register device\n");
- goto fail_interface;
- }
- libusb_ref_device(dev);
- }
- }
- return;
- fail_interface:
- libusb_release_interface(uh.devh, uh.interface);
- fail:
- libusb_close(uh.devh);
- uh.devh = NULL;
- }
- int
- check_device_connected(struct usb_handle *uh)
- {
- int r = libusb_kernel_driver_active(uh->devh, uh->interface);
- if (r == LIBUSB_ERROR_NO_DEVICE)
- return 0;
- if (r < 0)
- return -1;
- return 1;
- }
- void
- kick_disconnected()
- {
- struct usb_handle *usb= NULL;
- adb_mutex_lock(&usb_lock);
- for (usb = handle_list.next; usb != &handle_list; usb = usb->next) {
- if (check_device_connected(usb) == 0) {
- D("kick_disconnected(): Transport %p is not online anymore\n",
- usb);
- usb_kick(usb);
- }
- }
- adb_mutex_unlock(&usb_lock);
- }
- void
- scan_usb_devices()
- {
- D("scan_usb_devices(): started\n");
- libusb_device **devs= NULL;
- libusb_device *dev= NULL;
- ssize_t cnt = libusb_get_device_list(ctx, &devs);
- if (cnt < 0) {
- D("scan_usb_devices(): Failed to get device list (error: %d)\n",
- cnt);
- return;
- }
- int i = 0;
- while ((dev = devs[i++]) != NULL) {
- check_device(dev);
- }
- libusb_free_device_list(devs, 1);
- }
- void *
- device_poll_thread(void* unused)
- {
- D("device_poll_thread(): Created USB scan thread\n");
- for (;;) {
- sleep(5);
- kick_disconnected();
- scan_usb_devices();
- }
- /* never reaching this point */
- return (NULL);
- }
- static void
- sigalrm_handler(int signo)
- {
- /* nothing */
- }
- void
- usb_init()
- {
- D("usb_init(): started\n");
- adb_thread_t tid;
- struct sigaction actions;
- int r = libusb_init(&ctx);
- if (r != LIBUSB_SUCCESS) {
- err(EX_IOERR, "Failed to init libusb\n");
- }
- memset(&actions, 0, sizeof(actions));
- sigemptyset(&actions.sa_mask);
- actions.sa_flags = 0;
- actions.sa_handler = sigalrm_handler;
- sigaction(SIGALRM, &actions, NULL);
- /* initial device scan */
- scan_usb_devices();
- /* starting USB event polling thread */
- if (adb_thread_create(&tid, device_poll_thread, NULL)) {
- err(EX_IOERR, "cannot create USB scan thread\n");
- }
- D("usb_init(): finished\n");
- }
|