summaryrefslogtreecommitdiffstats
path: root/content/public/browser/browser_child_process_host_iterator.cc
blob: 534a4315d05163f42693e28950cd91fa1c188908 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "content/public/browser/browser_child_process_host_iterator.h"

#include "base/logging.h"
#include "content/browser/browser_child_process_host.h"
#include "content/public/browser/browser_thread.h"

namespace content {

BrowserChildProcessHostIterator::BrowserChildProcessHostIterator()
    : all_(true), type_(content::PROCESS_TYPE_UNKNOWN) {
  CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)) <<
        "BrowserChildProcessHostIterator must be used on the IO thread.";
  iterator_ = ::BrowserChildProcessHost::GetIterator()->begin();
}

BrowserChildProcessHostIterator::BrowserChildProcessHostIterator(
    content::ProcessType type)
    : all_(false), type_(type) {
  CHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)) <<
        "BrowserChildProcessHostIterator must be used on the IO thread.";
  iterator_ = ::BrowserChildProcessHost::GetIterator()->begin();
  if (!Done() && (*iterator_)->GetData().type != type_)
    ++(*this);
}

bool BrowserChildProcessHostIterator::operator++() {
  CHECK(!Done());
  do {
    ++iterator_;
    if (Done())
      break;

    if (!all_ && (*iterator_)->GetData().type != type_)
      continue;

    return true;
  } while (true);

  return false;
}

bool BrowserChildProcessHostIterator::Done() {
  return iterator_ == ::BrowserChildProcessHost::GetIterator()->end();
}

const ChildProcessData& BrowserChildProcessHostIterator::GetData() {
  CHECK(!Done());
  return (*iterator_)->GetData();
}

bool BrowserChildProcessHostIterator::Send(IPC::Message* message) {
  CHECK(!Done());
  return (*iterator_)->Send(message);
}

BrowserChildProcessHostDelegate*
    BrowserChildProcessHostIterator::GetDelegate() {
  return (*iterator_)->delegate();
}

}  // namespace content