1 | // [License]
|
---|
2 | // The Ariba-Underlay Copyright
|
---|
3 | //
|
---|
4 | // Copyright (c) 2008-2009, Institute of Telematics, UniversitÀt Karlsruhe (TH)
|
---|
5 | //
|
---|
6 | // Institute of Telematics
|
---|
7 | // UniversitÀt Karlsruhe (TH)
|
---|
8 | // Zirkel 2, 76128 Karlsruhe
|
---|
9 | // Germany
|
---|
10 | //
|
---|
11 | // Redistribution and use in source and binary forms, with or without
|
---|
12 | // modification, are permitted provided that the following conditions are
|
---|
13 | // met:
|
---|
14 | //
|
---|
15 | // 1. Redistributions of source code must retain the above copyright
|
---|
16 | // notice, this list of conditions and the following disclaimer.
|
---|
17 | // 2. Redistributions in binary form must reproduce the above copyright
|
---|
18 | // notice, this list of conditions and the following disclaimer in the
|
---|
19 | // documentation and/or other materials provided with the distribution.
|
---|
20 | //
|
---|
21 | // THIS SOFTWARE IS PROVIDED BY THE INSTITUTE OF TELEMATICS ``AS IS'' AND
|
---|
22 | // ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
---|
23 | // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
---|
24 | // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE ARIBA PROJECT OR
|
---|
25 | // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
---|
26 | // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
---|
27 | // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
---|
28 | // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
---|
29 | // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
---|
30 | // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
---|
31 | // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
---|
32 | //
|
---|
33 | // The views and conclusions contained in the software and documentation
|
---|
34 | // are those of the authors and should not be interpreted as representing
|
---|
35 | // official policies, either expressed or implied, of the Institute of
|
---|
36 | // Telematics.
|
---|
37 | // [License]
|
---|
38 |
|
---|
39 | #include "ariba/overlay/BaseOverlay.h"
|
---|
40 | #include "ariba/overlay/messages/OverlayMsg.h"
|
---|
41 |
|
---|
42 | #include "Chord.h"
|
---|
43 | #include "detail/chord_routing_table.hpp"
|
---|
44 |
|
---|
45 | #include "messages/Discovery.h"
|
---|
46 |
|
---|
47 | namespace ariba {
|
---|
48 | namespace overlay {
|
---|
49 |
|
---|
50 | enum signalMessageTypes {
|
---|
51 | typeDiscovery = OverlayMsg::typeSignalingStart + 0x01,
|
---|
52 | typeLeave = OverlayMsg::typeSignalingStart + 0x02,
|
---|
53 | };
|
---|
54 |
|
---|
55 | typedef chord_routing_table::item route_item;
|
---|
56 |
|
---|
57 | use_logging_cpp( Chord );
|
---|
58 |
|
---|
59 | Chord::Chord(BaseOverlay& _baseoverlay, const NodeID& _nodeid,
|
---|
60 | OverlayStructureEvents* _eventsReceiver, const OverlayParameterSet& param) :
|
---|
61 | OverlayInterface(_baseoverlay, _nodeid, _eventsReceiver, param) {
|
---|
62 |
|
---|
63 | // create routing table
|
---|
64 | this->table = new chord_routing_table(_nodeid, 4);
|
---|
65 | orphan_removal_counter = 0;
|
---|
66 | discovery_count = 0;
|
---|
67 | stabilize_counter = 0;
|
---|
68 | stabilize_finger = 0;
|
---|
69 | }
|
---|
70 |
|
---|
71 | Chord::~Chord() {
|
---|
72 |
|
---|
73 | // delete routing table
|
---|
74 | delete table;
|
---|
75 | }
|
---|
76 |
|
---|
77 | /// helper: sets up a link using the base overlay
|
---|
78 | LinkID Chord::setup(const EndpointDescriptor& endpoint, const NodeID& remote ) {
|
---|
79 |
|
---|
80 | // check if we already have a connection
|
---|
81 | for (int i=0; i<table->size(); i++)
|
---|
82 | if ((*table)[i]->ref_count > 0 && (*table)[i]->id == remote && !((*table)[i]->info.isUnspecified()))
|
---|
83 | return LinkID::UNSPECIFIED;
|
---|
84 |
|
---|
85 | // check if we are already trying to establish a link
|
---|
86 | for (size_t i=0; i<pending.size(); i++)
|
---|
87 | if ( pending[i] == remote ) {
|
---|
88 | logging_debug("Already trying to establish a link to node "
|
---|
89 | << remote.toString() );
|
---|
90 | return LinkID::UNSPECIFIED;
|
---|
91 | }
|
---|
92 |
|
---|
93 | // adding node to list of pending connections
|
---|
94 | pending.push_back( remote );
|
---|
95 |
|
---|
96 | logging_info("Request to setup link to " << endpoint.toString() );
|
---|
97 |
|
---|
98 | // establish link via base overlay
|
---|
99 | return baseoverlay.establishLink( endpoint, remote,
|
---|
100 | OverlayInterface::OVERLAY_SERVICE_ID );
|
---|
101 | }
|
---|
102 |
|
---|
103 | /// helper: sends a message using the "base overlay"
|
---|
104 | seqnum_t Chord::send( OverlayMsg* msg, const LinkID& link ) {
|
---|
105 | if (link.isUnspecified()) return 0;
|
---|
106 | return baseoverlay.send_link( msg, link );
|
---|
107 | }
|
---|
108 |
|
---|
109 | /// sends a discovery message
|
---|
110 | void Chord::send_discovery_to(const NodeID& remote, int ttl) {
|
---|
111 | LinkID link = getNextLinkId(remote);
|
---|
112 | if ( remote == nodeid || link.isUnspecified()) return;
|
---|
113 | if ( table->size() == 0 ) return;
|
---|
114 | ttl = 2;
|
---|
115 |
|
---|
116 | OverlayMsg msg( typeDiscovery );
|
---|
117 | msg.setRegisterRelay(true);
|
---|
118 | Discovery dmsg( Discovery::normal, (uint8_t)ttl, baseoverlay.getEndpointDescriptor() );
|
---|
119 | msg.encapsulate(&dmsg);
|
---|
120 |
|
---|
121 | // send to node
|
---|
122 | baseoverlay.send_node( &msg, remote );
|
---|
123 | }
|
---|
124 |
|
---|
125 | void Chord::discover_neighbors( const LinkID& link ) {
|
---|
126 | uint8_t ttl = 1;
|
---|
127 | {
|
---|
128 | // send predecessor discovery
|
---|
129 | OverlayMsg msg( typeDiscovery );
|
---|
130 | msg.setRegisterRelay(true);
|
---|
131 | Discovery dmsg( Discovery::predecessor, ttl,
|
---|
132 | baseoverlay.getEndpointDescriptor() );
|
---|
133 | msg.encapsulate(&dmsg);
|
---|
134 | send(&msg, link);
|
---|
135 | }
|
---|
136 | {
|
---|
137 | // send successor discovery
|
---|
138 | OverlayMsg msg( typeDiscovery );
|
---|
139 | msg.setSourceEndpoint( baseoverlay.getEndpointDescriptor() );
|
---|
140 | msg.setRegisterRelay(true);
|
---|
141 | Discovery dmsg( Discovery::successor, ttl,
|
---|
142 | baseoverlay.getEndpointDescriptor() );
|
---|
143 | msg.encapsulate(&dmsg);
|
---|
144 | send(&msg, link);
|
---|
145 | }
|
---|
146 | }
|
---|
147 |
|
---|
148 |
|
---|
149 | void Chord::createOverlay() {
|
---|
150 | }
|
---|
151 |
|
---|
152 | void Chord::deleteOverlay() {
|
---|
153 |
|
---|
154 | }
|
---|
155 |
|
---|
156 | void Chord::joinOverlay(const EndpointDescriptor& boot) {
|
---|
157 | logging_info( "joining Chord overlay structure through end-point " <<
|
---|
158 | (boot.isUnspecified() ? "local" : boot.toString()) );
|
---|
159 |
|
---|
160 | // initiator? no->setup first link
|
---|
161 | if (!boot.isUnspecified())
|
---|
162 | bootstrapLinks.push_back( setup(boot) );
|
---|
163 |
|
---|
164 | // timer for stabilization management
|
---|
165 | Timer::setInterval(1000);
|
---|
166 | Timer::start();
|
---|
167 | }
|
---|
168 |
|
---|
169 | void Chord::leaveOverlay() {
|
---|
170 | Timer::stop();
|
---|
171 | for (size_t i = 0; i < table->size(); i++) {
|
---|
172 | route_item* it = (*table)[i];
|
---|
173 | OverlayMsg msg( typeLeave );
|
---|
174 | send( &msg, it->info );
|
---|
175 | }
|
---|
176 | }
|
---|
177 |
|
---|
178 | /// @see OverlayInterface.h
|
---|
179 | const EndpointDescriptor& Chord::resolveNode(const NodeID& node) {
|
---|
180 | const route_item* item = table->get(node);
|
---|
181 | if (item == NULL || item->info.isUnspecified()) return EndpointDescriptor::UNSPECIFIED();
|
---|
182 | return baseoverlay.getEndpointDescriptor(item->info);
|
---|
183 | }
|
---|
184 |
|
---|
185 | /// @see OverlayInterface.h
|
---|
186 | const LinkID& Chord::getNextLinkId( const NodeID& id ) const {
|
---|
187 | // get next hop
|
---|
188 | const route_item* item = table->get_next_hop(id);
|
---|
189 |
|
---|
190 | // returns a unspecified id when this is itself
|
---|
191 | if (item == NULL || item->id == nodeid)
|
---|
192 | return LinkID::UNSPECIFIED;
|
---|
193 |
|
---|
194 | /// return routing info
|
---|
195 | return item->info;
|
---|
196 | }
|
---|
197 |
|
---|
198 | OverlayInterface::NodeList Chord::getKnownNodes(bool deep) const {
|
---|
199 | OverlayInterface::NodeList nodelist;
|
---|
200 |
|
---|
201 | if( deep ){
|
---|
202 | // all nodes that I know, fingers, succ/pred
|
---|
203 | for (size_t i = 0; i < table->size(); i++){
|
---|
204 | if ((*table)[i]->ref_count != 0
|
---|
205 | && !(*table)[i]->info.isUnspecified())
|
---|
206 | nodelist.push_back((*table)[i]->id);
|
---|
207 | }
|
---|
208 | } else {
|
---|
209 | // only succ and pred
|
---|
210 | if( table->get_predesessor() != NULL ){
|
---|
211 | nodelist.push_back( *(table->get_predesessor()) );
|
---|
212 | }
|
---|
213 | if( table->get_successor() != NULL ){
|
---|
214 | OverlayInterface::NodeList::iterator i =
|
---|
215 | std::find( nodelist.begin(), nodelist.end(), *(table->get_successor()) );
|
---|
216 | if( i == nodelist.end() )
|
---|
217 | nodelist.push_back( *(table->get_successor()) );
|
---|
218 | }
|
---|
219 | }
|
---|
220 |
|
---|
221 | return nodelist;
|
---|
222 | }
|
---|
223 |
|
---|
224 | /// @see CommunicationListener.h
|
---|
225 | /// @see OverlayInterface.h
|
---|
226 | void Chord::onLinkUp(const LinkID& lnk, const NodeID& remote) {
|
---|
227 | logging_info("link_up: link=" << lnk.toString() << " remote=" <<
|
---|
228 | remote.toString() );
|
---|
229 | for (vector<NodeID>::iterator i=pending.begin(); i!=pending.end(); i++)
|
---|
230 | if (*i == remote) {
|
---|
231 | pending.erase(i);
|
---|
232 | break;
|
---|
233 | }
|
---|
234 |
|
---|
235 | if (remote==nodeid) {
|
---|
236 | logging_warn("dropping link that has been established to myself (nodes have same nodeid?)");
|
---|
237 | baseoverlay.dropLink(lnk);
|
---|
238 | return;
|
---|
239 | }
|
---|
240 |
|
---|
241 | route_item* item = table->insert(remote);
|
---|
242 |
|
---|
243 | // item added to routing table?
|
---|
244 | if (item != NULL) { // yes-> add to routing table
|
---|
245 | logging_info("new routing neighbor: " << remote.toString()
|
---|
246 | << " with link " << lnk.toString());
|
---|
247 | // replace with new link
|
---|
248 | if (!item->info.isUnspecified() || item->info!=lnk)
|
---|
249 | baseoverlay.dropLink(item->info);
|
---|
250 | item->info = lnk;
|
---|
251 |
|
---|
252 | // discover neighbors of new overlay neighbor
|
---|
253 | showLinks();
|
---|
254 | } else { // no-> add orphan entry to routing table
|
---|
255 | logging_info("new orphan: " << remote.toString()
|
---|
256 | << " with link " << lnk.toString());
|
---|
257 | table->insert_orphan(remote)->info = lnk;
|
---|
258 | }
|
---|
259 |
|
---|
260 | // erase bootstrap link
|
---|
261 | vector<LinkID>::iterator it = std::find(bootstrapLinks.begin(), bootstrapLinks.end(), lnk);
|
---|
262 | if( it != bootstrapLinks.end() ) bootstrapLinks.erase( it );
|
---|
263 | }
|
---|
264 |
|
---|
265 | /// @see CommunicationListener.h or @see OverlayInterface.h
|
---|
266 | void Chord::onLinkDown(const LinkID& lnk, const NodeID& remote) {
|
---|
267 | logging_debug("link_down: link=" << lnk.toString() << " remote=" <<
|
---|
268 | remote.toString() );
|
---|
269 |
|
---|
270 | // remove link from routing table
|
---|
271 | route_item* item = table->get(remote);
|
---|
272 | if (item!=NULL && item->info==lnk) {
|
---|
273 | item->info = LinkID::UNSPECIFIED;
|
---|
274 | table->remove(remote);
|
---|
275 | }
|
---|
276 | }
|
---|
277 |
|
---|
278 | /// @see CommunicationListener.h
|
---|
279 | /// @see OverlayInterface.h
|
---|
280 | void Chord::onMessage(const DataMessage& msg, const NodeID& remote,
|
---|
281 | const LinkID& link) {
|
---|
282 |
|
---|
283 | // decode message
|
---|
284 | OverlayMsg* m = dynamic_cast<OverlayMsg*>(msg.getMessage());
|
---|
285 | if (m == NULL) return;
|
---|
286 |
|
---|
287 | // handle messages
|
---|
288 | switch (m->getType()) {
|
---|
289 |
|
---|
290 | // discovery request
|
---|
291 | case typeDiscovery: {
|
---|
292 | // decapsulate message
|
---|
293 | Discovery* dmsg = m->decapsulate<Discovery> ();
|
---|
294 | logging_debug("Received discovery message with"
|
---|
295 | << " src=" << m->getSourceNode().toString()
|
---|
296 | << " dst=" << m->getDestinationNode().toString()
|
---|
297 | << " ttl=" << (int)dmsg->getTTL()
|
---|
298 | << " type=" << (int)dmsg->getType()
|
---|
299 | );
|
---|
300 |
|
---|
301 | // add discovery node id
|
---|
302 | bool found = false;
|
---|
303 | BOOST_FOREACH( NodeID& value, discovery )
|
---|
304 | if (value == m->getSourceNode()) {
|
---|
305 | found = true;
|
---|
306 | break;
|
---|
307 | }
|
---|
308 | if (!found) discovery.push_back(m->getSourceNode());
|
---|
309 |
|
---|
310 | // check if source node can be added to routing table and setup link
|
---|
311 | if (m->getSourceNode() != nodeid)
|
---|
312 | setup( dmsg->getEndpoint(), m->getSourceNode() );
|
---|
313 |
|
---|
314 | // delegate discovery message
|
---|
315 | switch (dmsg->getType()) {
|
---|
316 |
|
---|
317 | // normal: route discovery message like every other message
|
---|
318 | case Discovery::normal: {
|
---|
319 | // closest node? yes-> split to follow successor and predecessor
|
---|
320 | if ( table->is_closest_to(m->getDestinationNode()) ) {
|
---|
321 | logging_debug("Discovery split:");
|
---|
322 | if (!table->get_successor()->isUnspecified()) {
|
---|
323 | OverlayMsg omsg(*m);
|
---|
324 | dmsg->setType(Discovery::successor);
|
---|
325 | omsg.encapsulate(dmsg);
|
---|
326 | logging_debug("* Routing to successor "
|
---|
327 | << table->get_successor()->toString() );
|
---|
328 | baseoverlay.send( &omsg, *table->get_successor() );
|
---|
329 | }
|
---|
330 |
|
---|
331 | // send predecessor message
|
---|
332 | if (!table->get_predesessor()->isUnspecified()) {
|
---|
333 | OverlayMsg omsg(*m);
|
---|
334 | dmsg->setType(Discovery::predecessor);
|
---|
335 | omsg.encapsulate(dmsg);
|
---|
336 | logging_debug("* Routing to predecessor "
|
---|
337 | << table->get_predesessor()->toString() );
|
---|
338 | baseoverlay.send( &omsg, *table->get_predesessor() );
|
---|
339 | }
|
---|
340 | }
|
---|
341 | // no-> route message
|
---|
342 | else {
|
---|
343 | baseoverlay.route( m );
|
---|
344 | }
|
---|
345 | break;
|
---|
346 | }
|
---|
347 |
|
---|
348 | // successor mode: follow the successor until TTL is zero
|
---|
349 | case Discovery::successor:
|
---|
350 | case Discovery::predecessor: {
|
---|
351 | // reached destination? no->forward!
|
---|
352 | if (m->getDestinationNode() != nodeid) {
|
---|
353 | OverlayMsg omsg(*m);
|
---|
354 | omsg.encapsulate(dmsg);
|
---|
355 | omsg.setService(OverlayInterface::OVERLAY_SERVICE_ID);
|
---|
356 | baseoverlay.route( &omsg );
|
---|
357 | break;
|
---|
358 | }
|
---|
359 |
|
---|
360 | // time to live ended? yes-> stop routing
|
---|
361 | if (dmsg->getTTL() == 0 || dmsg->getTTL() > 10) break;
|
---|
362 |
|
---|
363 | // decrease time-to-live
|
---|
364 | dmsg->setTTL(dmsg->getTTL() - 1);
|
---|
365 |
|
---|
366 | const route_item* item = NULL;
|
---|
367 | if (dmsg->getType() == Discovery::successor &&
|
---|
368 | table->get_successor() != NULL) {
|
---|
369 | item = table->get(*table->get_successor());
|
---|
370 | } else {
|
---|
371 | if (table->get_predesessor()!=NULL)
|
---|
372 | item = table->get(*table->get_predesessor());
|
---|
373 | }
|
---|
374 | if (item == NULL)
|
---|
375 | break;
|
---|
376 |
|
---|
377 | logging_debug("Routing discovery message to succ/pred "
|
---|
378 | << item->id.toString() );
|
---|
379 | OverlayMsg omsg(*m);
|
---|
380 | omsg.encapsulate(dmsg);
|
---|
381 | omsg.setDestinationNode(item->id);
|
---|
382 | omsg.setService(OverlayInterface::OVERLAY_SERVICE_ID);
|
---|
383 | baseoverlay.send(&omsg, omsg.getDestinationNode());
|
---|
384 | break;
|
---|
385 | }}
|
---|
386 | delete dmsg;
|
---|
387 | break;
|
---|
388 | }
|
---|
389 |
|
---|
390 | // leave
|
---|
391 | case typeLeave: {
|
---|
392 | if (link!=LinkID::UNSPECIFIED) {
|
---|
393 | route_item* item = table->get(remote);
|
---|
394 | if (item!=NULL) item->info = LinkID::UNSPECIFIED;
|
---|
395 | table->remove(remote);
|
---|
396 | baseoverlay.dropLink(link);
|
---|
397 | }
|
---|
398 | break;
|
---|
399 | }}
|
---|
400 | }
|
---|
401 |
|
---|
402 | void Chord::eventFunction() {
|
---|
403 | stabilize_counter++;
|
---|
404 | if (stabilize_counter < 0 || stabilize_counter == 2) {
|
---|
405 |
|
---|
406 | // reset counter
|
---|
407 | stabilize_counter = 0;
|
---|
408 |
|
---|
409 | // clear pending connections
|
---|
410 | pending.clear();
|
---|
411 |
|
---|
412 | // get number of real neighbors
|
---|
413 | size_t numNeighbors = 0;
|
---|
414 | for (size_t i = 0; i < table->size(); i++) {
|
---|
415 | route_item* it = (*table)[i];
|
---|
416 | if (it->ref_count != 0 && !it->info.isUnspecified()) numNeighbors++;
|
---|
417 | }
|
---|
418 | logging_info("Running stabilization: #links="
|
---|
419 | << table->size() << " #neighbors=" << numNeighbors );
|
---|
420 |
|
---|
421 | // updating neighbors
|
---|
422 | logging_debug("Discover new ring neighbors");
|
---|
423 | for (int i=0; i<table->size(); i++) {
|
---|
424 | LinkID id = (*table)[i]->info;
|
---|
425 | if (!id.isUnspecified()) discover_neighbors(id);
|
---|
426 | }
|
---|
427 |
|
---|
428 | // sending discovery
|
---|
429 | logging_debug("Sending discovery message to my neighbors and fingers");
|
---|
430 | stabilize_finger = ((stabilize_finger+1) % table->get_finger_table_size() );
|
---|
431 | const NodeID disc = table->get_finger_table(stabilize_finger).get_compare().get_center();
|
---|
432 | if (disc != nodeid)
|
---|
433 | send_discovery_to(disc);
|
---|
434 |
|
---|
435 | // remove orphan links
|
---|
436 | orphan_removal_counter++;
|
---|
437 | if (orphan_removal_counter <0 || orphan_removal_counter >= 2) {
|
---|
438 | logging_info("Discovered nodes: ");
|
---|
439 | BOOST_FOREACH( NodeID& id, discovery )
|
---|
440 | logging_info("* " << id.toString());
|
---|
441 | discovery.clear();
|
---|
442 | logging_info("Running orphan removal");
|
---|
443 | orphan_removal_counter = 0;
|
---|
444 | for (size_t i = 0; i < table->size(); i++) {
|
---|
445 | route_item* it = (*table)[i];
|
---|
446 | if (it->ref_count == 0 && !it->info.isUnspecified()) {
|
---|
447 | logging_info("Dropping orphaned link " << it->info.toString() << " to " << it->id.toString());
|
---|
448 | table->insert(it->id);
|
---|
449 | if (it->ref_count==0) {
|
---|
450 | LinkID id = it->info;
|
---|
451 | it->info = LinkID::UNSPECIFIED;
|
---|
452 | baseoverlay.dropLink(id);
|
---|
453 | }
|
---|
454 | }
|
---|
455 | }
|
---|
456 | }
|
---|
457 | }
|
---|
458 | }
|
---|
459 |
|
---|
460 | void Chord::showLinks() {
|
---|
461 | logging_info("--- chord routing information ----------------------------------");
|
---|
462 | logging_info("predecessor: " << (table->get_predesessor()==NULL? "<none>" :
|
---|
463 | table->get_predesessor()->toString()) );
|
---|
464 | logging_info("node_id : " << nodeid.toString() );
|
---|
465 | logging_info("successor : " << (table->get_successor()==NULL? "<none>" :
|
---|
466 | table->get_successor()->toString()));
|
---|
467 | logging_info("----------------------------------------------------------------");
|
---|
468 | }
|
---|
469 |
|
---|
470 | }} // namespace ariba, overlay
|
---|