| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "tools/android/forwarder2/device_controller.h" | 5 #include "tools/android/forwarder2/device_controller.h" |
| 6 | 6 |
| 7 #include <errno.h> | 7 #include <errno.h> |
| 8 #include <stdlib.h> | 8 #include <stdlib.h> |
| 9 | 9 |
| 10 #include "base/logging.h" | 10 #include "base/logging.h" |
| 11 #include "base/memory/scoped_ptr.h" | 11 #include "base/memory/scoped_ptr.h" |
| 12 #include "base/safe_strerror_posix.h" | 12 #include "base/safe_strerror_posix.h" |
| 13 #include "tools/android/forwarder2/command.h" | 13 #include "tools/android/forwarder2/command.h" |
| 14 #include "tools/android/forwarder2/device_listener.h" | 14 #include "tools/android/forwarder2/device_listener.h" |
| 15 #include "tools/android/forwarder2/socket.h" | 15 #include "tools/android/forwarder2/socket.h" |
| 16 | 16 |
| 17 namespace forwarder2 { | 17 namespace forwarder2 { |
| 18 | 18 |
| 19 DeviceController::DeviceController(int exit_notifier_fd) | 19 DeviceController::DeviceController(int exit_notifier_fd) |
| 20 : exit_notifier_fd_(exit_notifier_fd) { | 20 : exit_notifier_fd_(exit_notifier_fd) { |
| 21 kickstart_adb_socket_.set_exit_notifier_fd(exit_notifier_fd); | 21 kickstart_adb_socket_.AddEventFd(exit_notifier_fd); |
| 22 } | 22 } |
| 23 | 23 |
| 24 DeviceController::~DeviceController() { | 24 DeviceController::~DeviceController() { |
| 25 KillAllListeners(); | 25 KillAllListeners(); |
| 26 CleanUpDeadListeners(); | 26 CleanUpDeadListeners(); |
| 27 CHECK_EQ(0, listeners_.size()); | 27 CHECK_EQ(0, listeners_.size()); |
| 28 } | 28 } |
| 29 | 29 |
| 30 void DeviceController::CleanUpDeadListeners() { | 30 void DeviceController::CleanUpDeadListeners() { |
| 31 // Clean up dead listeners. | 31 // Clean up dead listeners. |
| (...skipping 21 matching lines...) Expand all Loading... |
| 53 } | 53 } |
| 54 LOG(INFO) << "Listening on Unix Domain Socket " << adb_unix_socket; | 54 LOG(INFO) << "Listening on Unix Domain Socket " << adb_unix_socket; |
| 55 return true; | 55 return true; |
| 56 } | 56 } |
| 57 | 57 |
| 58 void DeviceController::Start() { | 58 void DeviceController::Start() { |
| 59 while (true) { | 59 while (true) { |
| 60 CleanUpDeadListeners(); | 60 CleanUpDeadListeners(); |
| 61 scoped_ptr<Socket> socket(new Socket); | 61 scoped_ptr<Socket> socket(new Socket); |
| 62 if (!kickstart_adb_socket_.Accept(socket.get())) { | 62 if (!kickstart_adb_socket_.Accept(socket.get())) { |
| 63 if (!kickstart_adb_socket_.exited()) { | 63 if (!kickstart_adb_socket_.DidReceiveEvent()) { |
| 64 LOG(ERROR) << "Could not Accept DeviceController socket: " | 64 LOG(ERROR) << "Could not Accept DeviceController socket: " |
| 65 << safe_strerror(errno); | 65 << safe_strerror(errno); |
| 66 } else { | 66 } else { |
| 67 LOG(INFO) << "Received exit notification"; | 67 LOG(INFO) << "Received exit notification"; |
| 68 } | 68 } |
| 69 break; | 69 break; |
| 70 } | 70 } |
| 71 // So that |socket| doesn't block on read if it has notifications. | 71 // So that |socket| doesn't block on read if it has notifications. |
| 72 socket->set_exit_notifier_fd(exit_notifier_fd_); | 72 socket->AddEventFd(exit_notifier_fd_); |
| 73 int port; | 73 int port; |
| 74 command::Type command; | 74 command::Type command; |
| 75 if (!ReadCommand(socket.get(), &port, &command)) { | 75 if (!ReadCommand(socket.get(), &port, &command)) { |
| 76 LOG(ERROR) << "Invalid command received."; | 76 LOG(ERROR) << "Invalid command received."; |
| 77 continue; | 77 continue; |
| 78 } | 78 } |
| 79 DeviceListener* listener = listeners_.Lookup(port); | 79 DeviceListener* listener = listeners_.Lookup(port); |
| 80 switch (command) { | 80 switch (command) { |
| 81 case command::LISTEN: { | 81 case command::LISTEN: { |
| 82 if (listener != NULL) { | 82 if (listener != NULL) { |
| (...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 122 // TODO(felipeg): add a KillAllListeners command. | 122 // TODO(felipeg): add a KillAllListeners command. |
| 123 LOG(ERROR) << "Invalid command received. Port: " << port | 123 LOG(ERROR) << "Invalid command received. Port: " << port |
| 124 << " Command: " << command; | 124 << " Command: " << command; |
| 125 } | 125 } |
| 126 } | 126 } |
| 127 KillAllListeners(); | 127 KillAllListeners(); |
| 128 CleanUpDeadListeners(); | 128 CleanUpDeadListeners(); |
| 129 } | 129 } |
| 130 | 130 |
| 131 } // namespace forwarder | 131 } // namespace forwarder |
| OLD | NEW |