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 #ifndef TOOLS_ANDROID_FORWARDER2_HOST_CONTROLLER_H_ | 5 #ifndef TOOLS_ANDROID_FORWARDER2_HOST_CONTROLLER_H_ |
6 #define TOOLS_ANDROID_FORWARDER2_HOST_CONTROLLER_H_ | 6 #define TOOLS_ANDROID_FORWARDER2_HOST_CONTROLLER_H_ |
7 | 7 |
8 #include <string> | 8 #include <string> |
9 | 9 |
10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
11 #include "base/callback.h" | 11 #include "base/callback.h" |
12 #include "base/compiler_specific.h" | 12 #include "base/compiler_specific.h" |
13 #include "base/memory/scoped_ptr.h" | 13 #include "base/memory/scoped_ptr.h" |
14 #include "base/threading/thread.h" | 14 #include "base/threading/thread.h" |
15 #include "tools/android/forwarder2/pipe_notifier.h" | 15 #include "tools/android/forwarder2/pipe_notifier.h" |
16 #include "tools/android/forwarder2/socket.h" | 16 #include "tools/android/forwarder2/socket.h" |
17 | 17 |
18 namespace forwarder2 { | 18 namespace forwarder2 { |
19 | 19 |
20 class HostProxy; | 20 class HostProxy; |
21 | 21 |
22 class HostController { | 22 class HostController { |
23 public: | 23 public: |
24 // Callback used for self-deletion that lets the client perform some cleanup | 24 // Callback used for self-deletion that lets the client perform some cleanup |
25 // work before deleting the HostController instance. | 25 // work before deleting the HostController instance. |
26 typedef base::Callback<void (HostController*)> DeleteCallback; | 26 typedef base::Callback<void (scoped_ptr<HostController>)> DeletionCallback; |
27 | 27 |
28 // If |device_port| is zero, it dynamically allocates a port. | 28 // If |device_port| is zero then a dynamic port is allocated (and retrievable |
29 HostController(int device_port, | 29 // through device_port() below). |
30 const std::string& forward_to_host, | 30 static scoped_ptr<HostController> Create( |
31 int forward_to_host_port, | 31 int device_port, |
32 int adb_port, | 32 int host_port, |
33 int exit_notifier_fd, | 33 int adb_port, |
34 const DeleteCallback& delete_callback); | 34 int exit_notifier_fd, |
| 35 const DeletionCallback& deletion_callback); |
35 | 36 |
36 ~HostController(); | 37 ~HostController(); |
37 | 38 |
38 // Connects to the device forwarder app and sets the |device_port_| to the | |
39 // dynamically allocated port (when the provided |device_port| is zero). | |
40 // Returns true on success. Clients must call Connect() before calling | |
41 // Start(). | |
42 bool Connect(); | |
43 | |
44 // Starts the internal controller thread. | 39 // Starts the internal controller thread. |
45 void Start(); | 40 void Start(); |
46 | 41 |
47 int adb_port() const { return adb_port_; } | 42 int adb_port() const { return adb_port_; } |
48 | 43 |
49 // Gets the current device allocated port. Must be called after Connect(). | |
50 int device_port() const { return device_port_; } | 44 int device_port() const { return device_port_; } |
51 | 45 |
52 private: | 46 private: |
53 void ThreadHandler(); | 47 HostController(int device_port, |
| 48 int host_port, |
| 49 int adb_port, |
| 50 int exit_notifier_fd, |
| 51 const DeletionCallback& deletion_callback, |
| 52 scoped_ptr<Socket> adb_control_socket, |
| 53 scoped_ptr<PipeNotifier> delete_controller_notifier); |
54 | 54 |
55 void StartForwarder(scoped_ptr<Socket> host_server_data_socket_ptr); | 55 void ReadNextCommandSoon(); |
| 56 void ReadCommandOnInternalThread(); |
| 57 |
| 58 void StartForwarder(scoped_ptr<Socket> host_server_data_socket); |
56 | 59 |
57 // Helper method that creates a socket and adds the appropriate event file | 60 // Helper method that creates a socket and adds the appropriate event file |
58 // descriptors. | 61 // descriptors. |
59 scoped_ptr<Socket> CreateSocket(); | 62 scoped_ptr<Socket> CreateSocket(); |
60 | 63 |
61 void SelfDelete(); | 64 void SelfDelete(); |
62 | 65 |
63 Socket adb_control_socket_; | 66 static void SelfDeleteOnDeletionTaskRunner( |
64 int device_port_; | 67 const DeletionCallback& deletion_callback, |
65 const std::string forward_to_host_; | 68 scoped_ptr<HostController> controller); |
66 const int forward_to_host_port_; | 69 |
| 70 const int device_port_; |
| 71 const int host_port_; |
67 const int adb_port_; | 72 const int adb_port_; |
68 // Used to notify the controller when the process is killed. | 73 // Used to notify the controller when the process is killed. |
69 const int global_exit_notifier_fd_; | 74 const int global_exit_notifier_fd_; |
70 // Used to let the client delete the instance in case an error happened. | 75 // Used to let the client delete the instance in case an error happened. |
71 const DeleteCallback delete_callback_; | 76 const DeletionCallback deletion_callback_; |
| 77 scoped_ptr<Socket> adb_control_socket_; |
| 78 scoped_ptr<PipeNotifier> delete_controller_notifier_; |
72 // Used to cancel the pending blocking IO operations when the host controller | 79 // Used to cancel the pending blocking IO operations when the host controller |
73 // instance is deleted. | 80 // instance is deleted. |
74 PipeNotifier delete_controller_notifier_; | 81 // Task runner used for deletion set at construction time (i.e. the object is |
75 bool ready_; | 82 // deleted on the same thread it is created on). |
| 83 const scoped_refptr<base::SingleThreadTaskRunner> deletion_task_runner_; |
76 base::Thread thread_; | 84 base::Thread thread_; |
77 | 85 |
78 DISALLOW_COPY_AND_ASSIGN(HostController); | 86 DISALLOW_COPY_AND_ASSIGN(HostController); |
79 }; | 87 }; |
80 | 88 |
81 } // namespace forwarder2 | 89 } // namespace forwarder2 |
82 | 90 |
83 #endif // TOOLS_ANDROID_FORWARDER2_HOST_CONTROLLER_H_ | 91 #endif // TOOLS_ANDROID_FORWARDER2_HOST_CONTROLLER_H_ |
OLD | NEW |