00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039 #include "OneHop.h"
00040 #include "ariba/overlay/BaseOverlay.h"
00041
00042 #include "ariba/overlay/modules/onehop/messages/OneHopMessage.h"
00043 #include "ariba/overlay/modules/onehop/messages/NodeListingRequest.h"
00044 #include "ariba/overlay/modules/onehop/messages/NodeListingReply.h"
00045
00046 namespace ariba {
00047 namespace overlay {
00048
00049 use_logging_cpp( OneHop );
00050
00051 OneHop::OneHop(BaseOverlay& _baseoverlay, const NodeID& _nodeid,
00052 OverlayStructureEvents* _eventsReceiver, const OverlayParameterSet& param)
00053 : OverlayInterface( _baseoverlay, _nodeid, _eventsReceiver, param ),
00054 state( OneHopStateInvalid ) {
00055
00056
00057
00058
00059 overlayNodes.insert( make_pair(_nodeid, LinkID::UNSPECIFIED) );
00060
00061 Timer::setInterval(5000);
00062 Timer::start();
00063 }
00064
00065 OneHop::~OneHop(){
00066 Timer::stop();
00067 deleteOverlay();
00068 }
00069
00070 const EndpointDescriptor& OneHop::resolveNode(const NodeID& node){
00071
00072 OverlayNodeMapping::const_iterator i = overlayNodes.find( node );
00073 if (i == overlayNodes.end()) return EndpointDescriptor::UNSPECIFIED();
00074
00075 const EndpointDescriptor& ep = baseoverlay.getEndpointDescriptor( i->second );
00076
00077 logging_debug( "resolved node " << node.toString() << " to endpoint " << ep.toString() );
00078 return ep;
00079 }
00080
00081 void OneHop::routeMessage(const NodeID& destnode, Message* msg){
00082
00083
00084
00085
00086 logging_debug( "routing message to node " << destnode.toString() );
00087
00088
00089 if(destnode == nodeid)
00090 baseoverlay.incomingRouteMessage( msg, LinkID::UNSPECIFIED, nodeid );
00091
00092
00093 OverlayNodeMapping::const_iterator i = overlayNodes.find( destnode );
00094 if (i == overlayNodes.end()) {
00095 logging_error( "not able to route message to node " << destnode.toString() );
00096 return;
00097 }
00098 OneHopMessage onehopRoute( OneHopMessage::OneHopMessageTypeRoute );
00099 onehopRoute.encapsulate(msg);
00100
00101 baseoverlay.sendMessage( &onehopRoute, i->second );
00102 }
00103
00104 void OneHop::routeMessage(const NodeID& node, const LinkID& link, Message* msg) {
00105 OneHopMessage onehopRoute( OneHopMessage::OneHopMessageTypeRoute );
00106 onehopRoute.encapsulate(msg);
00107 baseoverlay.sendMessage( &onehopRoute, link );
00108 }
00109
00111 const LinkID& OneHop::getNextLinkId( const NodeID& id ) const {
00112 OverlayNodeMapping::const_iterator i = overlayNodes.find( id );
00113 if (i == overlayNodes.end()) return LinkID::UNSPECIFIED;
00114 return i->second;
00115 }
00116
00117 void OneHop::createOverlay() {
00118
00119
00120 logging_info( "creating onehop overlay structure" );
00121 }
00122
00123 void OneHop::deleteOverlay(){
00124
00125 logging_info( "deleting onehop overlay structure" );
00126 state = OneHopStateInvalid;
00127 }
00128
00129 OverlayInterface::NodeList OneHop::getKnownNodes(bool deep) const {
00130
00131 OverlayInterface::NodeList retlist;
00132
00133 OverlayNodeMapping::const_iterator i = overlayNodes.begin();
00134 OverlayNodeMapping::const_iterator iend = overlayNodes.end();
00135
00136 for( ; i != iend; i++ )
00137 retlist.push_back( i->first );
00138
00139 return retlist;
00140 }
00141
00142 void OneHop::joinOverlay(const EndpointDescriptor& bootstrapEp){
00143
00144 logging_info( "joining onehop overlay structure through end-point " <<
00145 (bootstrapEp.isUnspecified() ? "local" : bootstrapEp.toString()) );
00146
00147 if( bootstrapEp.isUnspecified() ){
00148
00149
00150
00151
00152
00153 state = OneHopStateCompleted;
00154 } else {
00155 bootstrapLinks.push_back(
00156 baseoverlay.establishDirectLink( bootstrapEp,
00157 OverlayInterface::OVERLAY_SERVICE_ID )
00158 );
00159 }
00160 }
00161
00162 void OneHop::leaveOverlay(){
00163
00164 logging_info( "leaving onehop overlay structure" );
00165
00166
00167
00168
00169
00170
00171 state = OneHopStateInvalid;
00172
00173
00174
00175
00176
00177
00178 OverlayNodeMapping::iterator i = overlayNodes.begin();
00179 OverlayNodeMapping::iterator iend = overlayNodes.end();
00180
00181 for( ; i != iend; i++){
00182 if( i->first != nodeid && i->second != LinkID::UNSPECIFIED ){
00183
00184 OneHopMessage msg (OneHopMessage::OneHopMessageTypeLeave);
00185 baseoverlay.sendMessage( &msg, i->second );
00186 }
00187 }
00188 }
00189
00190
00191 void OneHop::onLinkDown(const LinkID& lnk, const NodeID& remote){
00192
00193
00194
00195 if( state == OneHopStateInvalid ) return;
00196
00197
00198 logging_debug( "link " << lnk.toString() << " to node " << remote.toString() << " went down, removing node" );
00199
00200 OverlayNodeMapping::iterator i = overlayNodes.begin();
00201 OverlayNodeMapping::iterator iend = overlayNodes.end();
00202
00203 for( ; i != iend; i++ ){
00204 if( i->second == lnk ){
00205 overlayNodes.erase( i );
00206 break;
00207 }
00208 }
00209
00210 vector<LinkID>::iterator it = std::find( bootstrapLinks.begin(), bootstrapLinks.end(), lnk );
00211 if( it != bootstrapLinks.end() ) bootstrapLinks.erase( it );
00212 }
00213
00214 void OneHop::onLinkUp(const LinkID& lnk, const NodeID& remote){
00215
00216 logging_debug( "link is up, sending out node listing request" );
00217
00218 NodeListingRequest requestmsg;
00219 OneHopMessage onemsg( OneHopMessage::OneHopMessageTypeListingRequest );
00220 onemsg.encapsulate( &requestmsg );
00221
00222 baseoverlay.sendMessage( &onemsg, lnk );
00223 }
00224
00225 void OneHop::onMessage(const DataMessage& msg, const NodeID& remote, const LinkID& lnk){
00226
00227 OneHopMessage* onemsg = msg.getMessage()->convert<OneHopMessage>();
00228 if( onemsg == NULL ) return;
00229
00230
00231
00232
00233
00234 if( onemsg->isType( OneHopMessage::OneHopMessageTypeListingRequest ) ){
00235
00236 NodeListingRequest* request = onemsg->decapsulate<NodeListingRequest>();
00237
00238 logging_info( "onehop received node listing request from node " << remote.toString() );
00239
00240
00241
00242
00243
00244 overlayNodes.insert( make_pair(remote, lnk) );
00245
00246
00247
00248
00249
00250
00251 OneHopMessage onehopReply( OneHopMessage::OneHopMessageTypeListingReply );
00252 NodeListingReply listingReply;
00253
00254 OverlayNodeMapping::iterator i = overlayNodes.begin();
00255 OverlayNodeMapping::iterator iend = overlayNodes.end();
00256
00257 logging_debug( "sending out node listing reply with the following items" );
00258
00259 for( ; i != iend; i++ ){
00260
00261 const NodeID node = i->first;
00262 const LinkID link = i->second;
00263 const EndpointDescriptor& endpoint = baseoverlay.getEndpointDescriptor( link );
00264
00265 logging_debug( "node: " + node.toString() + ", endp: " + endpoint.toString());
00266 listingReply.add( node, const_cast<EndpointDescriptor*>(new EndpointDescriptor(endpoint)) );
00267 }
00268
00269 onehopReply.encapsulate( &listingReply );
00270 baseoverlay.sendMessage( &onehopReply, lnk );
00271
00272
00273
00274
00275
00276
00277 eventsReceiver->onNodeJoin( remote );
00278
00279 }
00280
00281
00282
00283
00284
00285 if( onemsg->isType( OneHopMessage::OneHopMessageTypeListingReply) ){
00286
00287 NodeListingReply* reply = onemsg->decapsulate<NodeListingReply>();
00288
00289 logging_debug( "received node listing reply from node " << remote.toString()
00290 << " with all overlay nodes. connecting to all of them" );
00291
00292
00293
00294
00295
00296
00297
00298 const NodeListingReply::NodeEndpointList& endpoints = reply->getList();
00299 logging_debug( "received " << endpoints.size() << " nodes in listing" );
00300
00301 NodeListingReply::NodeEndpointList::const_iterator i = endpoints.begin();
00302 NodeListingReply::NodeEndpointList::const_iterator iend = endpoints.end();
00303
00304 for( ; i != iend; i++ ){
00305
00306
00307
00308
00309
00310
00311 const NodeID& node = (*i).first;
00312 if( overlayNodes.find(node) != overlayNodes.end() ) continue;
00313 if( node == nodeid ) continue;
00314
00315 logging_debug( "building up link to node in overlay " << node.toString() );
00316 const LinkID link = baseoverlay.establishDirectLink( *((*i).second),
00317 OverlayInterface::OVERLAY_SERVICE_ID );
00318
00319 overlayNodes.insert( make_pair(node, link) );
00320
00321 }
00322
00323 }
00324
00325
00326
00327
00328
00329 if( onemsg->isType(OneHopMessage::OneHopMessageTypeLeave) ){
00330
00331 logging_debug("received leave message from " <<
00332 remote.toString() << " on link " << lnk.toString());
00333
00334
00335 baseoverlay.dropLink( lnk );
00336
00337 }
00338
00339
00340
00341
00342
00343 if( onemsg->isType( OneHopMessage::OneHopMessageTypeRoute) ){
00344 logging_debug( "Route message arrived at destination node -> delegate to BaseOverlay" );
00345 baseoverlay.incomingRouteMessage( onemsg, lnk, remote);
00346 }
00347
00348 }
00349
00350 void OneHop::eventFunction(){
00351
00352 logging_debug("<<<<<<<<<<<<<<<<onehop-table<<<<<<<<<<<<<<<<<<<");
00353
00354 OverlayNodeMapping::iterator i = overlayNodes.begin();
00355 OverlayNodeMapping::iterator iend = overlayNodes.end();
00356
00357 for( ; i != iend; i++ ){
00358
00359 const NodeID node = i->first;
00360 const LinkID link = i->second;
00361 const EndpointDescriptor& endpoint = baseoverlay.getEndpointDescriptor( link );
00362
00363 logging_debug( "node: " << node.toString() <<
00364 ", link_: " << link.toString() << ", endp: " << endpoint.toString());
00365 }
00366
00367 logging_debug(">>>>>>>>>>>>>>>>>onehop-table>>>>>>>>>>>>>>>>>>>>>");
00368
00369 }
00370
00371 }}