| 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, | 22 net::StreamSocket* socket_raw, |
| 23 int result) { | 23 int result) { |
| 24 callback.Run(result, result == net::OK ? socket : NULL); | 24 scoped_ptr<net::StreamSocket> socket(socket_raw); |
| 25 if (result != net::OK) |
| 26 socket.reset(); |
| 27 callback.Run(result, socket.Pass()); |
| 25 } | 28 } |
| 26 | 29 |
| 27 void OnRead(net::StreamSocket* socket, | 30 void OnRead(net::StreamSocket* socket, |
| 28 scoped_refptr<net::IOBuffer> buffer, | 31 scoped_refptr<net::IOBuffer> buffer, |
| 29 const std::string& data, | 32 const std::string& data, |
| 30 const UsbDeviceProvider::CommandCallback& callback, | 33 const UsbDeviceProvider::CommandCallback& callback, |
| 31 int result) { | 34 int result) { |
| 32 if (result <= 0) { | 35 if (result <= 0) { |
| 33 callback.Run(result, result == 0 ? data : std::string()); | 36 callback.Run(result, result == 0 ? data : std::string()); |
| 34 delete socket; | 37 delete socket; |
| (...skipping 26 matching lines...) Expand all Loading... |
| 61 } | 64 } |
| 62 | 65 |
| 63 void RunCommand(scoped_refptr<AndroidUsbDevice> device, | 66 void RunCommand(scoped_refptr<AndroidUsbDevice> device, |
| 64 const std::string& command, | 67 const std::string& command, |
| 65 const UsbDeviceProvider::CommandCallback& callback) { | 68 const UsbDeviceProvider::CommandCallback& callback) { |
| 66 net::StreamSocket* socket = device->CreateSocket(command); | 69 net::StreamSocket* socket = device->CreateSocket(command); |
| 67 if (!socket) { | 70 if (!socket) { |
| 68 callback.Run(net::ERR_CONNECTION_FAILED, std::string()); | 71 callback.Run(net::ERR_CONNECTION_FAILED, std::string()); |
| 69 return; | 72 return; |
| 70 } | 73 } |
| 71 int result = socket->Connect(base::Bind(&OpenedForCommand, callback, socket)); | 74 int result = socket->Connect( |
| 75 base::Bind(&OpenedForCommand, callback, socket)); |
| 72 if (result != net::ERR_IO_PENDING) | 76 if (result != net::ERR_IO_PENDING) |
| 73 callback.Run(result, std::string()); | 77 callback.Run(result, std::string()); |
| 74 } | 78 } |
| 75 | 79 |
| 76 } // namespace | 80 } // namespace |
| 77 | 81 |
| 78 // static | 82 // static |
| 79 void UsbDeviceProvider::CountDevices( | 83 void UsbDeviceProvider::CountDevices( |
| 80 const base::Callback<void(int)>& callback) { | 84 const base::Callback<void(int)>& callback) { |
| 81 AndroidUsbDevice::CountDevices(callback); | 85 AndroidUsbDevice::CountDevices(callback); |
| (...skipping 18 matching lines...) Expand all Loading... |
| 100 return; | 104 return; |
| 101 } | 105 } |
| 102 AdbDeviceInfoQuery::Start(base::Bind(&RunCommand, it->second), callback); | 106 AdbDeviceInfoQuery::Start(base::Bind(&RunCommand, it->second), callback); |
| 103 } | 107 } |
| 104 | 108 |
| 105 void UsbDeviceProvider::OpenSocket(const std::string& serial, | 109 void UsbDeviceProvider::OpenSocket(const std::string& serial, |
| 106 const std::string& name, | 110 const std::string& name, |
| 107 const SocketCallback& callback) { | 111 const SocketCallback& callback) { |
| 108 UsbDeviceMap::iterator it = device_map_.find(serial); | 112 UsbDeviceMap::iterator it = device_map_.find(serial); |
| 109 if (it == device_map_.end()) { | 113 if (it == device_map_.end()) { |
| 110 callback.Run(net::ERR_CONNECTION_FAILED, NULL); | 114 callback.Run(net::ERR_CONNECTION_FAILED, |
| 115 make_scoped_ptr<net::StreamSocket>(NULL)); |
| 111 return; | 116 return; |
| 112 } | 117 } |
| 113 std::string socket_name = | 118 std::string socket_name = |
| 114 base::StringPrintf(kLocalAbstractCommand, name.c_str()); | 119 base::StringPrintf(kLocalAbstractCommand, name.c_str()); |
| 115 net::StreamSocket* socket = it->second->CreateSocket(socket_name); | 120 net::StreamSocket* socket = it->second->CreateSocket(socket_name); |
| 116 if (!socket) { | 121 if (!socket) { |
| 117 callback.Run(net::ERR_CONNECTION_FAILED, NULL); | 122 callback.Run(net::ERR_CONNECTION_FAILED, |
| 123 make_scoped_ptr<net::StreamSocket>(NULL)); |
| 118 return; | 124 return; |
| 119 } | 125 } |
| 120 int result = socket->Connect(base::Bind(&OnOpenSocket, callback, socket)); | 126 int result = socket->Connect(base::Bind(&OnOpenSocket, callback, socket)); |
| 121 if (result != net::ERR_IO_PENDING) | 127 if (result != net::ERR_IO_PENDING) |
| 122 callback.Run(result, NULL); | 128 callback.Run(result, make_scoped_ptr<net::StreamSocket>(NULL)); |
| 123 } | 129 } |
| 124 | 130 |
| 125 void UsbDeviceProvider::ReleaseDevice(const std::string& serial) { | 131 void UsbDeviceProvider::ReleaseDevice(const std::string& serial) { |
| 126 device_map_.erase(serial); | 132 device_map_.erase(serial); |
| 127 } | 133 } |
| 128 | 134 |
| 129 UsbDeviceProvider::~UsbDeviceProvider() { | 135 UsbDeviceProvider::~UsbDeviceProvider() { |
| 130 } | 136 } |
| 131 | 137 |
| 132 void UsbDeviceProvider::EnumeratedDevices(const SerialsCallback& callback, | 138 void UsbDeviceProvider::EnumeratedDevices(const SerialsCallback& callback, |
| 133 const AndroidUsbDevices& devices) { | 139 const AndroidUsbDevices& devices) { |
| 134 std::vector<std::string> result; | 140 std::vector<std::string> result; |
| 135 device_map_.clear(); | 141 device_map_.clear(); |
| 136 for (AndroidUsbDevices::const_iterator it = devices.begin(); | 142 for (AndroidUsbDevices::const_iterator it = devices.begin(); |
| 137 it != devices.end(); ++it) { | 143 it != devices.end(); ++it) { |
| 138 result.push_back((*it)->serial()); | 144 result.push_back((*it)->serial()); |
| 139 device_map_[(*it)->serial()] = *it; | 145 device_map_[(*it)->serial()] = *it; |
| 140 (*it)->InitOnCallerThread(); | 146 (*it)->InitOnCallerThread(); |
| 141 } | 147 } |
| 142 callback.Run(result); | 148 callback.Run(result); |
| 143 } | 149 } |
| 144 | 150 |
| OLD | NEW |