15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef TOOLS_ANDROID_FORWARDER2_DEVICE_CONTROLLER_H_ 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define TOOLS_ANDROID_FORWARDER2_DEVICE_CONTROLLER_H_ 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/basictypes.h" 11d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch#include "base/containers/hash_tables.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/linked_ptr.h" 13d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch#include "base/memory/ref_counted.h" 14d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch#include "base/memory/scoped_ptr.h" 15d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch#include "base/memory/weak_ptr.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "tools/android/forwarder2/socket.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 18d3868032626d59662ff73b372b5d584c1d144c53Ben Murdochnamespace base { 19d3868032626d59662ff73b372b5d584c1d144c53Ben Murdochclass SingleThreadTaskRunner; 20d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch} // namespace base 21d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace forwarder2 { 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class DeviceListener; 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 26d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch// There is a single DeviceController per device_forwarder process, and it is in 27d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch// charge of managing all active redirections on the device side (one 28d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch// DeviceListener each). 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class DeviceController { 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 31d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch static scoped_ptr<DeviceController> Create(const std::string& adb_unix_socket, 32d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch int exit_notifier_fd); 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ~DeviceController(); 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Start(); 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 38d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch typedef base::hash_map< 39d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch int /* port */, linked_ptr<DeviceListener> > ListenersMap; 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 41d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch DeviceController(scoped_ptr<Socket> host_socket, int exit_notifier_fd); 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 43d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch void AcceptHostCommandSoon(); 44d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch void AcceptHostCommandInternal(); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 46d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch // Note that this can end up being called after the DeviceController is 47d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch // destroyed which is why a weak pointer is used. 48f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) static void DeleteListenerOnError( 49d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch const base::WeakPtr<DeviceController>& device_controller_ptr, 50f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) scoped_ptr<DeviceListener> device_listener); 51d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch 52d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch const scoped_ptr<Socket> host_socket_; 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Used to notify the controller to exit. 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const int exit_notifier_fd_; 55d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch // Lets ensure DeviceListener instances are deleted on the thread they were 56d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch // created on. 57d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch const scoped_refptr<base::SingleThreadTaskRunner> construction_task_runner_; 58d3868032626d59662ff73b372b5d584c1d144c53Ben Murdoch ListenersMap listeners_; 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 601320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci //WeakPtrFactory's documentation says: 611320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci // Member variables should appear before the WeakPtrFactory, to ensure 621320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci // that any WeakPtrs to Controller are invalidated before its members 631320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci // variable's destructors are executed, rendering them invalid. 641320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci base::WeakPtrFactory<DeviceController> weak_ptr_factory_; 651320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(DeviceController); 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace forwarder 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // TOOLS_ANDROID_FORWARDER2_DEVICE_CONTROLLER_H_ 72