| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/devtools/device/usb/usb_device_provider.h" | 5 #include "chrome/browser/devtools/device/usb/usb_device_provider.h" |
| 6 | 6 |
| 7 #include "base/strings/stringprintf.h" | 7 #include "base/strings/stringprintf.h" |
| 8 #include "chrome/browser/devtools/device/adb/adb_device_info_query.h" | 8 #include "chrome/browser/devtools/device/adb/adb_device_info_query.h" |
| 9 #include "chrome/browser/devtools/device/usb/android_rsa.h" | 9 #include "chrome/browser/devtools/device/usb/android_rsa.h" |
| 10 #include "chrome/browser/devtools/device/usb/android_usb_device.h" | 10 #include "chrome/browser/devtools/device/usb/android_usb_device.h" |
| 11 #include "crypto/rsa_private_key.h" | 11 #include "crypto/rsa_private_key.h" |
| 12 #include "net/base/net_errors.h" | 12 #include "net/base/net_errors.h" |
| 13 #include "net/socket/stream_socket.h" | 13 #include "net/socket/stream_socket.h" |
| 14 | 14 |
| 15 namespace { | 15 namespace { |
| 16 | 16 |
| 17 const char kLocalAbstractCommand[] = "localabstract:%s"; | 17 const char kLocalAbstractCommand[] = "localabstract:%s"; |
| 18 | 18 |
| 19 const int kBufferSize = 16 * 1024; | 19 const int kBufferSize = 16 * 1024; |
| 20 | 20 |
| 21 void OnOpenSocket(const UsbDeviceProvider::SocketCallback& callback, | 21 void OnOpenSocket(const UsbDeviceProvider::SocketCallback& callback, |
| 22 net::StreamSocket* socket_raw, | 22 net::StreamSocket* socket, |
| 23 int result) { | 23 int result) { |
| 24 scoped_ptr<net::StreamSocket> socket(socket_raw); | 24 callback.Run(result, result == net::OK ? socket : NULL); |
| 25 if (result != net::OK) | |
| 26 socket.reset(); | |
| 27 callback.Run(result, socket.Pass()); | |
| 28 } | 25 } |
| 29 | 26 |
| 30 void OnRead(net::StreamSocket* socket, | 27 void OnRead(net::StreamSocket* socket, |
| 31 scoped_refptr<net::IOBuffer> buffer, | 28 scoped_refptr<net::IOBuffer> buffer, |
| 32 const std::string& data, | 29 const std::string& data, |
| 33 const UsbDeviceProvider::CommandCallback& callback, | 30 const UsbDeviceProvider::CommandCallback& callback, |
| 34 int result) { | 31 int result) { |
| 35 if (result <= 0) { | 32 if (result <= 0) { |
| 36 callback.Run(result, result == 0 ? data : std::string()); | 33 callback.Run(result, result == 0 ? data : std::string()); |
| 37 delete socket; | 34 delete socket; |
| (...skipping 26 matching lines...) Expand all Loading... |
| 64 } | 61 } |
| 65 | 62 |
| 66 void RunCommand(scoped_refptr<AndroidUsbDevice> device, | 63 void RunCommand(scoped_refptr<AndroidUsbDevice> device, |
| 67 const std::string& command, | 64 const std::string& command, |
| 68 const UsbDeviceProvider::CommandCallback& callback) { | 65 const UsbDeviceProvider::CommandCallback& callback) { |
| 69 net::StreamSocket* socket = device->CreateSocket(command); | 66 net::StreamSocket* socket = device->CreateSocket(command); |
| 70 if (!socket) { | 67 if (!socket) { |
| 71 callback.Run(net::ERR_CONNECTION_FAILED, std::string()); | 68 callback.Run(net::ERR_CONNECTION_FAILED, std::string()); |
| 72 return; | 69 return; |
| 73 } | 70 } |
| 74 int result = socket->Connect( | 71 int result = socket->Connect(base::Bind(&OpenedForCommand, callback, socket)); |
| 75 base::Bind(&OpenedForCommand, callback, socket)); | |
| 76 if (result != net::ERR_IO_PENDING) | 72 if (result != net::ERR_IO_PENDING) |
| 77 callback.Run(result, std::string()); | 73 callback.Run(result, std::string()); |
| 78 } | 74 } |
| 79 | 75 |
| 80 } // namespace | 76 } // namespace |
| 81 | 77 |
| 82 // static | 78 // static |
| 83 void UsbDeviceProvider::CountDevices( | 79 void UsbDeviceProvider::CountDevices( |
| 84 const base::Callback<void(int)>& callback) { | 80 const base::Callback<void(int)>& callback) { |
| 85 AndroidUsbDevice::CountDevices(callback); | 81 AndroidUsbDevice::CountDevices(callback); |
| (...skipping 18 matching lines...) Expand all Loading... |
| 104 return; | 100 return; |
| 105 } | 101 } |
| 106 AdbDeviceInfoQuery::Start(base::Bind(&RunCommand, it->second), callback); | 102 AdbDeviceInfoQuery::Start(base::Bind(&RunCommand, it->second), callback); |
| 107 } | 103 } |
| 108 | 104 |
| 109 void UsbDeviceProvider::OpenSocket(const std::string& serial, | 105 void UsbDeviceProvider::OpenSocket(const std::string& serial, |
| 110 const std::string& name, | 106 const std::string& name, |
| 111 const SocketCallback& callback) { | 107 const SocketCallback& callback) { |
| 112 UsbDeviceMap::iterator it = device_map_.find(serial); | 108 UsbDeviceMap::iterator it = device_map_.find(serial); |
| 113 if (it == device_map_.end()) { | 109 if (it == device_map_.end()) { |
| 114 callback.Run(net::ERR_CONNECTION_FAILED, | 110 callback.Run(net::ERR_CONNECTION_FAILED, NULL); |
| 115 make_scoped_ptr<net::StreamSocket>(NULL)); | |
| 116 return; | 111 return; |
| 117 } | 112 } |
| 118 std::string socket_name = | 113 std::string socket_name = |
| 119 base::StringPrintf(kLocalAbstractCommand, name.c_str()); | 114 base::StringPrintf(kLocalAbstractCommand, name.c_str()); |
| 120 net::StreamSocket* socket = it->second->CreateSocket(socket_name); | 115 net::StreamSocket* socket = it->second->CreateSocket(socket_name); |
| 121 if (!socket) { | 116 if (!socket) { |
| 122 callback.Run(net::ERR_CONNECTION_FAILED, | 117 callback.Run(net::ERR_CONNECTION_FAILED, NULL); |
| 123 make_scoped_ptr<net::StreamSocket>(NULL)); | |
| 124 return; | 118 return; |
| 125 } | 119 } |
| 126 int result = socket->Connect(base::Bind(&OnOpenSocket, callback, socket)); | 120 int result = socket->Connect(base::Bind(&OnOpenSocket, callback, socket)); |
| 127 if (result != net::ERR_IO_PENDING) | 121 if (result != net::ERR_IO_PENDING) |
| 128 callback.Run(result, make_scoped_ptr<net::StreamSocket>(NULL)); | 122 callback.Run(result, NULL); |
| 129 } | 123 } |
| 130 | 124 |
| 131 void UsbDeviceProvider::ReleaseDevice(const std::string& serial) { | 125 void UsbDeviceProvider::ReleaseDevice(const std::string& serial) { |
| 132 device_map_.erase(serial); | 126 device_map_.erase(serial); |
| 133 } | 127 } |
| 134 | 128 |
| 135 UsbDeviceProvider::~UsbDeviceProvider() { | 129 UsbDeviceProvider::~UsbDeviceProvider() { |
| 136 } | 130 } |
| 137 | 131 |
| 138 void UsbDeviceProvider::EnumeratedDevices(const SerialsCallback& callback, | 132 void UsbDeviceProvider::EnumeratedDevices(const SerialsCallback& callback, |
| 139 const AndroidUsbDevices& devices) { | 133 const AndroidUsbDevices& devices) { |
| 140 std::vector<std::string> result; | 134 std::vector<std::string> result; |
| 141 device_map_.clear(); | 135 device_map_.clear(); |
| 142 for (AndroidUsbDevices::const_iterator it = devices.begin(); | 136 for (AndroidUsbDevices::const_iterator it = devices.begin(); |
| 143 it != devices.end(); ++it) { | 137 it != devices.end(); ++it) { |
| 144 result.push_back((*it)->serial()); | 138 result.push_back((*it)->serial()); |
| 145 device_map_[(*it)->serial()] = *it; | 139 device_map_[(*it)->serial()] = *it; |
| 146 (*it)->InitOnCallerThread(); | 140 (*it)->InitOnCallerThread(); |
| 147 } | 141 } |
| 148 callback.Run(result); | 142 callback.Run(result); |
| 149 } | 143 } |
| 150 | 144 |
| OLD | NEW |