123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515 |
- /*
- * 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 <windows.h>
- #include <winerror.h>
- #include <errno.h>
- #include <usb100.h>
- #include <adb_api.h>
- #include <stdio.h>
- #include "sysdeps.h"
- #define TRACE_TAG TRACE_USB
- #include "adb.h"
- /** Structure usb_handle describes our connection to the usb device via
- AdbWinApi.dll. This structure is returned from usb_open() routine and
- is expected in each subsequent call that is accessing the device.
- */
- struct usb_handle {
- /// Previous entry in the list of opened usb handles
- usb_handle *prev;
- /// Next entry in the list of opened usb handles
- usb_handle *next;
- /// Handle to USB interface
- ADBAPIHANDLE adb_interface;
- /// Handle to USB read pipe (endpoint)
- ADBAPIHANDLE adb_read_pipe;
- /// Handle to USB write pipe (endpoint)
- ADBAPIHANDLE adb_write_pipe;
- /// Interface name
- char* interface_name;
- /// Mask for determining when to use zero length packets
- unsigned zero_mask;
- };
- /// Class ID assigned to the device by androidusb.sys
- static const GUID usb_class_id = ANDROID_USB_CLASS_ID;
- /// List of opened usb handles
- static usb_handle handle_list = {
- .prev = &handle_list,
- .next = &handle_list,
- };
- /// Locker for the list of opened usb handles
- ADB_MUTEX_DEFINE( usb_lock );
- /// Checks if there is opened usb handle in handle_list for this device.
- int known_device(const char* dev_name);
- /// Checks if there is opened usb handle in handle_list for this device.
- /// usb_lock mutex must be held before calling this routine.
- int known_device_locked(const char* dev_name);
- /// Registers opened usb handle (adds it to handle_list).
- int register_new_device(usb_handle* handle);
- /// Checks if interface (device) matches certain criteria
- int recognized_device(usb_handle* handle);
- /// Enumerates present and available interfaces (devices), opens new ones and
- /// registers usb transport for them.
- void find_devices();
- /// Entry point for thread that polls (every second) for new usb interfaces.
- /// This routine calls find_devices in infinite loop.
- void* device_poll_thread(void* unused);
- /// Initializes this module
- void usb_init();
- /// Cleans up this module
- void usb_cleanup();
- /// Opens usb interface (device) by interface (device) name.
- usb_handle* do_usb_open(const wchar_t* interface_name);
- /// Writes data to the opened usb handle
- int usb_write(usb_handle* handle, const void* data, int len);
- /// Reads data using the opened usb handle
- int usb_read(usb_handle *handle, void* data, int len);
- /// Cleans up opened usb handle
- void usb_cleanup_handle(usb_handle* handle);
- /// Cleans up (but don't close) opened usb handle
- void usb_kick(usb_handle* handle);
- /// Closes opened usb handle
- int usb_close(usb_handle* handle);
- /// Gets interface (device) name for an opened usb handle
- const char *usb_name(usb_handle* handle);
- int known_device_locked(const char* dev_name) {
- usb_handle* usb;
- if (NULL != dev_name) {
- // Iterate through the list looking for the name match.
- for(usb = handle_list.next; usb != &handle_list; usb = usb->next) {
- // In Windows names are not case sensetive!
- if((NULL != usb->interface_name) &&
- (0 == stricmp(usb->interface_name, dev_name))) {
- return 1;
- }
- }
- }
- return 0;
- }
- int known_device(const char* dev_name) {
- int ret = 0;
- if (NULL != dev_name) {
- adb_mutex_lock(&usb_lock);
- ret = known_device_locked(dev_name);
- adb_mutex_unlock(&usb_lock);
- }
- return ret;
- }
- int register_new_device(usb_handle* handle) {
- if (NULL == handle)
- return 0;
- adb_mutex_lock(&usb_lock);
- // Check if device is already in the list
- if (known_device_locked(handle->interface_name)) {
- adb_mutex_unlock(&usb_lock);
- return 0;
- }
- // Not in the list. Add this handle to the list.
- handle->next = &handle_list;
- handle->prev = handle_list.prev;
- handle->prev->next = handle;
- handle->next->prev = handle;
- adb_mutex_unlock(&usb_lock);
- return 1;
- }
- void* device_poll_thread(void* unused) {
- D("Created device thread\n");
- while(1) {
- find_devices();
- adb_sleep_ms(1000);
- }
- return NULL;
- }
- void usb_init() {
- adb_thread_t tid;
- if(adb_thread_create(&tid, device_poll_thread, NULL)) {
- fatal_errno("cannot create input thread");
- }
- }
- void usb_cleanup() {
- }
- usb_handle* do_usb_open(const wchar_t* interface_name) {
- // Allocate our handle
- usb_handle* ret = (usb_handle*)malloc(sizeof(usb_handle));
- if (NULL == ret)
- return NULL;
- // Set linkers back to the handle
- ret->next = ret;
- ret->prev = ret;
- // Create interface.
- ret->adb_interface = AdbCreateInterfaceByName(interface_name);
- if (NULL == ret->adb_interface) {
- free(ret);
- errno = GetLastError();
- return NULL;
- }
- // Open read pipe (endpoint)
- ret->adb_read_pipe =
- AdbOpenDefaultBulkReadEndpoint(ret->adb_interface,
- AdbOpenAccessTypeReadWrite,
- AdbOpenSharingModeReadWrite);
- if (NULL != ret->adb_read_pipe) {
- // Open write pipe (endpoint)
- ret->adb_write_pipe =
- AdbOpenDefaultBulkWriteEndpoint(ret->adb_interface,
- AdbOpenAccessTypeReadWrite,
- AdbOpenSharingModeReadWrite);
- if (NULL != ret->adb_write_pipe) {
- // Save interface name
- unsigned long name_len = 0;
- // First get expected name length
- AdbGetInterfaceName(ret->adb_interface,
- NULL,
- &name_len,
- true);
- if (0 != name_len) {
- ret->interface_name = (char*)malloc(name_len);
- if (NULL != ret->interface_name) {
- // Now save the name
- if (AdbGetInterfaceName(ret->adb_interface,
- ret->interface_name,
- &name_len,
- true)) {
- // We're done at this point
- return ret;
- }
- } else {
- SetLastError(ERROR_OUTOFMEMORY);
- }
- }
- }
- }
- // Something went wrong.
- int saved_errno = GetLastError();
- usb_cleanup_handle(ret);
- free(ret);
- SetLastError(saved_errno);
- return NULL;
- }
- int usb_write(usb_handle* handle, const void* data, int len) {
- unsigned long time_out = 5000;
- unsigned long written = 0;
- int ret;
- D("usb_write %d\n", len);
- if (NULL != handle) {
- // Perform write
- ret = AdbWriteEndpointSync(handle->adb_write_pipe,
- (void*)data,
- (unsigned long)len,
- &written,
- time_out);
- int saved_errno = GetLastError();
- if (ret) {
- // Make sure that we've written what we were asked to write
- D("usb_write got: %ld, expected: %d\n", written, len);
- if (written == (unsigned long)len) {
- if(handle->zero_mask && (len & handle->zero_mask) == 0) {
- // Send a zero length packet
- AdbWriteEndpointSync(handle->adb_write_pipe,
- (void*)data,
- 0,
- &written,
- time_out);
- }
- return 0;
- }
- } else {
- // assume ERROR_INVALID_HANDLE indicates we are disconnected
- if (saved_errno == ERROR_INVALID_HANDLE)
- usb_kick(handle);
- }
- errno = saved_errno;
- } else {
- D("usb_write NULL handle\n");
- SetLastError(ERROR_INVALID_HANDLE);
- }
- D("usb_write failed: %d\n", errno);
- return -1;
- }
- int usb_read(usb_handle *handle, void* data, int len) {
- unsigned long time_out = 0;
- unsigned long read = 0;
- int ret;
- D("usb_read %d\n", len);
- if (NULL != handle) {
- while (len > 0) {
- int xfer = (len > 4096) ? 4096 : len;
- ret = AdbReadEndpointSync(handle->adb_read_pipe,
- (void*)data,
- (unsigned long)xfer,
- &read,
- time_out);
- int saved_errno = GetLastError();
- D("usb_write got: %ld, expected: %d, errno: %d\n", read, xfer, saved_errno);
- if (ret) {
- data += read;
- len -= read;
- if (len == 0)
- return 0;
- } else {
- // assume ERROR_INVALID_HANDLE indicates we are disconnected
- if (saved_errno == ERROR_INVALID_HANDLE)
- usb_kick(handle);
- break;
- }
- errno = saved_errno;
- }
- } else {
- D("usb_read NULL handle\n");
- SetLastError(ERROR_INVALID_HANDLE);
- }
- D("usb_read failed: %d\n", errno);
- return -1;
- }
- void usb_cleanup_handle(usb_handle* handle) {
- if (NULL != handle) {
- if (NULL != handle->interface_name)
- free(handle->interface_name);
- if (NULL != handle->adb_write_pipe)
- AdbCloseHandle(handle->adb_write_pipe);
- if (NULL != handle->adb_read_pipe)
- AdbCloseHandle(handle->adb_read_pipe);
- if (NULL != handle->adb_interface)
- AdbCloseHandle(handle->adb_interface);
- handle->interface_name = NULL;
- handle->adb_write_pipe = NULL;
- handle->adb_read_pipe = NULL;
- handle->adb_interface = NULL;
- }
- }
- void usb_kick(usb_handle* handle) {
- if (NULL != handle) {
- adb_mutex_lock(&usb_lock);
- usb_cleanup_handle(handle);
- adb_mutex_unlock(&usb_lock);
- } else {
- SetLastError(ERROR_INVALID_HANDLE);
- errno = ERROR_INVALID_HANDLE;
- }
- }
- int usb_close(usb_handle* handle) {
- D("usb_close\n");
- if (NULL != handle) {
- // Remove handle from the list
- adb_mutex_lock(&usb_lock);
- if ((handle->next != handle) && (handle->prev != handle)) {
- handle->next->prev = handle->prev;
- handle->prev->next = handle->next;
- handle->prev = handle;
- handle->next = handle;
- }
- adb_mutex_unlock(&usb_lock);
- // Cleanup handle
- usb_cleanup_handle(handle);
- free(handle);
- }
- return 0;
- }
- const char *usb_name(usb_handle* handle) {
- if (NULL == handle) {
- SetLastError(ERROR_INVALID_HANDLE);
- errno = ERROR_INVALID_HANDLE;
- return NULL;
- }
- return (const char*)handle->interface_name;
- }
- int recognized_device(usb_handle* handle) {
- if (NULL == handle)
- return 0;
- // Check vendor and product id first
- USB_DEVICE_DESCRIPTOR device_desc;
- if (!AdbGetUsbDeviceDescriptor(handle->adb_interface,
- &device_desc)) {
- return 0;
- }
- // Then check interface properties
- USB_INTERFACE_DESCRIPTOR interf_desc;
- if (!AdbGetUsbInterfaceDescriptor(handle->adb_interface,
- &interf_desc)) {
- return 0;
- }
- // Must have two endpoints
- if (2 != interf_desc.bNumEndpoints) {
- return 0;
- }
- if (is_adb_interface(device_desc.idVendor, device_desc.idProduct,
- interf_desc.bInterfaceClass, interf_desc.bInterfaceSubClass, interf_desc.bInterfaceProtocol)) {
- if(interf_desc.bInterfaceProtocol == 0x01) {
- AdbEndpointInformation endpoint_info;
- // assuming zero is a valid bulk endpoint ID
- if (AdbGetEndpointInformation(handle->adb_interface, 0, &endpoint_info)) {
- handle->zero_mask = endpoint_info.max_packet_size - 1;
- }
- }
- return 1;
- }
- return 0;
- }
- void find_devices() {
- usb_handle* handle = NULL;
- char entry_buffer[2048];
- char interf_name[2048];
- AdbInterfaceInfo* next_interface = (AdbInterfaceInfo*)(&entry_buffer[0]);
- unsigned long entry_buffer_size = sizeof(entry_buffer);
- char* copy_name;
- // Enumerate all present and active interfaces.
- ADBAPIHANDLE enum_handle =
- AdbEnumInterfaces(usb_class_id, true, true, true);
- if (NULL == enum_handle)
- return;
- while (AdbNextInterface(enum_handle, next_interface, &entry_buffer_size)) {
- // TODO: FIXME - temp hack converting wchar_t into char.
- // It would be better to change AdbNextInterface so it will return
- // interface name as single char string.
- const wchar_t* wchar_name = next_interface->device_name;
- for(copy_name = interf_name;
- L'\0' != *wchar_name;
- wchar_name++, copy_name++) {
- *copy_name = (char)(*wchar_name);
- }
- *copy_name = '\0';
- // Lets see if we already have this device in the list
- if (!known_device(interf_name)) {
- // This seems to be a new device. Open it!
- handle = do_usb_open(next_interface->device_name);
- if (NULL != handle) {
- // Lets see if this interface (device) belongs to us
- if (recognized_device(handle)) {
- D("adding a new device %s\n", interf_name);
- char serial_number[512];
- unsigned long serial_number_len = sizeof(serial_number);
- if (AdbGetSerialNumber(handle->adb_interface,
- serial_number,
- &serial_number_len,
- true)) {
- // Lets make sure that we don't duplicate this device
- if (register_new_device(handle)) {
- register_usb_transport(handle, serial_number, NULL, 1);
- } else {
- D("register_new_device failed for %s\n", interf_name);
- usb_cleanup_handle(handle);
- free(handle);
- }
- } else {
- D("cannot get serial number\n");
- usb_cleanup_handle(handle);
- free(handle);
- }
- } else {
- usb_cleanup_handle(handle);
- free(handle);
- }
- }
- }
- entry_buffer_size = sizeof(entry_buffer);
- }
- AdbCloseHandle(enum_handle);
- }
|