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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
// Copyright 2014 The Chromium OS 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 "peerd/peer_manager_impl.h"
#include <string>
#include <dbus/object_path.h>
#include "peerd/dbus_constants.h"
using chromeos::dbus_utils::AsyncEventSequencer;
using chromeos::dbus_utils::ExportedObjectManager;
using peerd::technologies::tech_t;
using std::string;
using std::to_string;
using std::unique_ptr;
namespace peerd {
PeerManagerImpl::PeerManagerImpl(const scoped_refptr<dbus::Bus> bus,
ExportedObjectManager* object_manager)
: bus_(bus),
object_manager_(object_manager) {
}
void PeerManagerImpl::OnPeerDiscovered(const string& peer_id,
const string& name,
const string& note,
const base::Time& last_seen,
tech_t which_technology) {
auto it = peers_.find(peer_id);
if (it != peers_.end()) {
it->second->UpdateFromAdvertisement(name, note, last_seen,
which_technology);
return;
}
dbus::ObjectPath path{
string(dbus_constants::kPeerPrefix) + to_string(++peers_discovered_)};
unique_ptr<DiscoveredPeer> peer{new DiscoveredPeer{
bus_, object_manager_, path, which_technology}};
scoped_refptr<AsyncEventSequencer> sequencer(new AsyncEventSequencer());
const bool registered = peer->RegisterAsync(
nullptr, peer_id, name, note, last_seen,
sequencer->GetHandler("Failed to expose DiscoveredPeer.", true));
sequencer->OnAllTasksCompletedCall({});
if (registered) {
peers_[peer_id].swap(peer);
} else {
LOG(INFO) << "Discovered corrupted peer advertisement; discarding it.";
}
}
void PeerManagerImpl::OnServiceDiscovered(const string& peer_id,
const string& service_id,
const Service::ServiceInfo& info,
const Service::IpAddresses& addresses,
const base::Time& last_seen,
tech_t which_technology) {
auto it = peers_.find(peer_id);
if (it == peers_.end()) {
// A service was found that corresponds to no particular peer.
// We could just silently add a peer entry without name/note fields here,
// or we can discard the service. Lets discard the service until it is
// known that we need to support this case.
LOG(WARNING) << "Found service=" << service_id << " but had no peer="
<< peer_id;
return;
}
it->second->UpdateService(service_id, addresses, info,
last_seen, which_technology);
}
void PeerManagerImpl::OnPeerRemoved(const string& peer_id,
tech_t which_technology) {
auto it = peers_.find(peer_id);
if (it == peers_.end()) {
LOG(WARNING) << "Tried to remove technology=" << which_technology
<< " from peer=" << peer_id
<< " that was never discovered.";
return;
}
it->second->RemoveTechnology(which_technology);
if (it->second->GetTechnologyCount() == 0) {
peers_.erase(it);
}
}
void PeerManagerImpl::OnServiceRemoved(const string& peer_id,
const string& service_id,
tech_t which_technology) {
auto it = peers_.find(peer_id);
if (it == peers_.end()) {
LOG(WARNING) << "Tried to remove service from peer that was "
<< "never discovered: " << peer_id;
return;
}
it->second->RemoveTechnologyFromService(service_id, which_technology);
}
void PeerManagerImpl::OnTechnologyShutdown(tech_t which_technology) {
auto it = peers_.begin();
while (it != peers_.end()) {
it->second->RemoveTechnology(which_technology);
if (it->second->GetTechnologyCount() == 0) {
it = peers_.erase(it);
} else {
++it;
}
}
}
} // namespace peerd
|
