src/tools/android/forwarder2/forwarders_manager.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "tools/android/forwarder2/forwarders_manager.h"
   6
   7 #include <algorithm>
   8
   9 #include "base/basictypes.h"
  10 #include "base/bind.h"
  11 #include "base/location.h"
  12 #include "base/logging.h"
  13 #include "base/message_loop/message_loop_proxy.h"
  14 #include "tools/android/forwarder2/forwarder.h"
  15 #include "tools/android/forwarder2/socket.h"
  16
  17 namespace forwarder2 {
  18
  19 ForwardersManager::ForwardersManager() : delegate_(new Delegate()) {}
  20
  21 ForwardersManager::~ForwardersManager() {
  22   delegate_->Clear();
  23 }
  24
  25 void ForwardersManager::CreateAndStartNewForwarder(scoped_ptr<Socket> socket1,
  26                                                    scoped_ptr<Socket> socket2) {
  27   delegate_->CreateAndStartNewForwarder(socket1.Pass(), socket2.Pass());
  28 }
  29
  30 ForwardersManager::Delegate::Delegate() {}
  31
  32 ForwardersManager::Delegate::~Delegate() {
  33   // The forwarder instances should already have been deleted on their
  34   // construction thread. Deleting them here would be unsafe since we don't know
  35   // which thread this destructor is called on.
  36   DCHECK(forwarders_.empty());
  37 }
  38
  39 void ForwardersManager::Delegate::Clear() {
  40   if (!forwarders_constructor_runner_) {
  41     DCHECK(forwarders_.empty());
  42     return;
  43   }
  44   if (forwarders_constructor_runner_->RunsTasksOnCurrentThread()) {
  45     ClearOnForwarderConstructorThread();
  46     return;
  47   }
  48   forwarders_constructor_runner_->PostTask(
  49       FROM_HERE,
  50       base::Bind(
  51           &ForwardersManager::Delegate::ClearOnForwarderConstructorThread,
  52           this));
  53 }
  54
  55 void ForwardersManager::Delegate::CreateAndStartNewForwarder(
  56     scoped_ptr<Socket> socket1,
  57     scoped_ptr<Socket> socket2) {
  58   const scoped_refptr<base::SingleThreadTaskRunner> current_task_runner(
  59       base::MessageLoopProxy::current());
  60   DCHECK(current_task_runner);
  61   if (forwarders_constructor_runner_) {
  62     DCHECK_EQ(current_task_runner, forwarders_constructor_runner_);
  63   } else {
  64     forwarders_constructor_runner_ = current_task_runner;
  65   }
  66   forwarders_.push_back(
  67       new Forwarder(socket1.Pass(), socket2.Pass(),
  68                     &deletion_notifier_,
  69                     base::Bind(&ForwardersManager::Delegate::OnForwarderError,
  70                                this)));
  71   forwarders_.back()->Start();
  72 }
  73
  74 void ForwardersManager::Delegate::OnForwarderError(
  75     scoped_ptr<Forwarder> forwarder) {
  76   DCHECK(forwarders_constructor_runner_->RunsTasksOnCurrentThread());
  77   const ScopedVector<Forwarder>::iterator it = std::find(
  78       forwarders_.begin(), forwarders_.end(), forwarder.get());
  79   DCHECK(it != forwarders_.end());
  80   std::swap(*it, forwarders_.back());
  81   forwarders_.pop_back();
  82   ignore_result(forwarder.release());  // Deleted by the pop_back() above.
  83 }
  84
  85 void ForwardersManager::Delegate::ClearOnForwarderConstructorThread() {
  86   DCHECK(forwarders_constructor_runner_->RunsTasksOnCurrentThread());
  87   deletion_notifier_.Notify();
  88   forwarders_.clear();
  89 }
  90
  91 }  // namespace forwarder2