source: source/ariba/overlay/modules/onehop/OneHop.cpp@ 3603

// [Licence]

// The Ariba-Underlay Copyright

//

// Copyright (c) 2008-2009, Institute of Telematics, UniversitÀt Karlsruhe (TH)

//

// Institute of Telematics

// UniversitÀt Karlsruhe (TH)

// Zirkel 2, 76128 Karlsruhe

// Germany

//

// Redistribution and use in source and binary forms, with or without

// modification, are permitted provided that the following conditions are

// met:

//

// 1. Redistributions of source code must retain the above copyright

// notice, this list of conditions and the following disclaimer.

// 2. Redistributions in binary form must reproduce the above copyright

// notice, this list of conditions and the following disclaimer in the

// documentation and/or other materials provided with the distribution.

//

// THIS SOFTWARE IS PROVIDED BY THE INSTITUTE OF TELEMATICS ``AS IS'' AND

// ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE ARIBA PROJECT OR

// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,

// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR

// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF

// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING

// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS

// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

//

// The views and conclusions contained in the software and documentation

// are those of the authors and should not be interpreted as representing

// official policies, either expressed or implied, of the Institute of

// Telematics.

// [Licence]



#include "OneHop.h"

#include "ariba/overlay/BaseOverlay.h"



namespace ariba {

namespace overlay {



use_logging_cpp( OneHop );



OneHop::OneHop(BaseOverlay& _baseoverlay, const NodeID& _nodeid, OverlayStructureEvents* _eventsReceiver)

        :       OverlayInterface( _baseoverlay, _nodeid, _eventsReceiver ),

                state           ( OneHopStateInvalid ),

                bootstrapLink   ( LinkID::UNSPECIFIED ),

                pendingLinks    ( 0 ) {



        //

        // insert us as the first node in the overlay

        //

        overlayNodes.insert( make_pair(_nodeid, LinkID::UNSPECIFIED) );

}



OneHop::~OneHop(){

        deleteOverlay();

}



const EndpointDescriptor& OneHop::resolveNode(const NodeID& node){



        OverlayNodeMapping::const_iterator i = overlayNodes.find( node );

        if (i == overlayNodes.end()) return EndpointDescriptor::UNSPECIFIED;



        const EndpointDescriptor& ep = baseoverlay.getEndpointDescriptor( i->second );



        logging_debug( "resolved node " << node.toString() << " to endpoint " << ep.toString() );

        return ep;

}



void OneHop::routeMessage(const NodeID& destnode, Message* msg){



        // in the fullmesh overlay we know every other node

        // so we also have a link to each other node



        logging_debug( "routing message to node " << destnode.toString() );



        OverlayNodeMapping::const_iterator i = overlayNodes.find( destnode );

        if (i == overlayNodes.end()) {

                logging_error( "not able to route message to node " << destnode.toString() );

                return;

        }



        baseoverlay.sendMessage( msg, i->second );

}



void OneHop::createOverlay(){



        // don't need to bootstrap against ourselfs.

        // the create and join process is completed now.



        logging_info( "creating onehop overlay structure" );

        state = OneHopStateCompleted;

}



void OneHop::deleteOverlay(){



        logging_info( "deleting onehop overlay structure" );

        state = OneHopStateInvalid;

        pendingLinks = 0;

}



OverlayInterface::NodeList OneHop::getKnownNodes() const {



        OverlayInterface::NodeList retlist;



        OverlayNodeMapping::const_iterator i = overlayNodes.begin();

        OverlayNodeMapping::const_iterator iend = overlayNodes.end();



        for( ; i != iend; i++ )

                retlist.push_back( i->first );



        return retlist;

}



void OneHop::joinOverlay(const EndpointDescriptor& bootstrapEp){



        logging_info( "joining onehop overlay structure through endpoint " <<

                        (bootstrapEp == EndpointDescriptor::UNSPECIFIED ?

                                        "local" : bootstrapEp.toString()) );



        state = OneHopStateJoinInitiated;

        pendingLinks = 0;



        if( bootstrapEp == EndpointDescriptor::UNSPECIFIED ){



                // we are the initiator and we are to bootstrap against

                // ourselfs. in the onehop overlay this is not an issue

                // and we can just ignore this call.



                state = OneHopStateCompleted;





        } else {



                bootstrapLink = baseoverlay.establishLink( bootstrapEp,

                                        OverlayInterface::OVERLAY_SERVICE_ID );

        }

}



void OneHop::leaveOverlay(){



        logging_info( "leaving onehop overlay structure" );



        // set the state to invalid, this will prevent from

        // handling onLinkDown events, as we are traversing the

        // overlayNodes map and the onLinkDown function is called

        // from the BaseOverlay and OneHop::onLinkDown will also

        // try to access the overlayNodes structure.

        state = OneHopStateInvalid;



        //

        // close all links, this will indicate the other nodes that we left

        //



        OverlayNodeMapping::iterator i = overlayNodes.begin();

        OverlayNodeMapping::iterator iend = overlayNodes.end();



        for( ; i != iend; i++){

                if( i->first != nodeid && i->second != LinkID::UNSPECIFIED )

                        baseoverlay.dropLink( i->second );

        }



        pendingLinks = 0;

}





void OneHop::onLinkDown(const LinkID& lnk, const NodeID& remote){



        // don't handle when we are in state-invalid,

        // see comment in OneHop::leaveOverlay

        if( state == OneHopStateInvalid ) return;



        //

        // node went down, remove from overlay mapping

        //



        logging_debug( "node " << remote.toString() << " left overlay structure" );



        OverlayNodeMapping::iterator i = overlayNodes.begin();

        OverlayNodeMapping::iterator iend = overlayNodes.end();



        for( ; i != iend; i++ ){

                if( i->second == lnk ){

                        overlayNodes.erase( i );

                        break;

                }

        }

}



void OneHop::onLinkUp(const LinkID& lnk, const NodeID& remote){



        //

        // as soon as a link goes up, we always request the node listing.

        // and try to get connections to as much nodes as possible in a greedy way.

        //



        if( lnk != bootstrapLink ){

                if( pendingLinks > 0 )  pendingLinks--;

                if( pendingLinks == 0 ) state = OneHopStateCompleted;

        }



        logging_debug( "link is up, sending out node listing request" );



        NodeListingRequest requestmsg;

        OneHopMessage onemsg( OneHopMessage::OneHopMessageTypeListingRequest );

        onemsg.encapsulate( &requestmsg );



        state = OneHopStateJoinListingRequested;

        baseoverlay.sendMessage( &onemsg, lnk );

}



void OneHop::onMessage(const DataMessage& msg, const NodeID& remote, const LinkID& lnk){



        OneHopMessage* onemsg = msg.getMessage()->convert<OneHopMessage>();

        if( onemsg == NULL ) return;



        //

        // handle node listing request

        //



        if( onemsg->isType( OneHopMessage::OneHopMessageTypeListingRequest) ){



                NodeListingRequest* request = onemsg->decapsulate<NodeListingRequest>();



                logging_info( "onehop received node listing request from node " << remote.toString() );



                //

                // first, insert the nodes and the link into our mapping

                //



                overlayNodes.insert( make_pair(remote, lnk) );



                //

                // send back a message with all nodes

                // and their current EndpointDescriptor

                //



                OneHopMessage onehopReply( OneHopMessage::OneHopMessageTypeListingReply );

                NodeListingReply listingReply;



                OverlayNodeMapping::iterator i = overlayNodes.begin();

                OverlayNodeMapping::iterator iend = overlayNodes.end();



                logging_debug( "sending out node listing reply with the following items" );



                for( ; i != iend; i++ ){



                        const NodeID node = i->first;

                        const LinkID link = i->second;

                        const EndpointDescriptor& endpoint = baseoverlay.getEndpointDescriptor( link );



                        logging_debug( "node: " + node.toString() + ", endp: " + endpoint.toString());

                        listingReply.add( node, const_cast<EndpointDescriptor*>(new EndpointDescriptor(endpoint)) );

                }



                onehopReply.encapsulate( &listingReply );

                baseoverlay.sendMessage( &onehopReply, lnk );



                //

                // now that we know the node, we can tell the baseoverlay

                // that the node has joined our overlay structure

                //



                eventsReceiver->onNodeJoin( remote );



        } // if( request != NULL )



        //

        // handle node listing reply

        //



        if( onemsg->isType( OneHopMessage::OneHopMessageTypeListingReply) ){



                NodeListingReply* reply = onemsg->decapsulate<NodeListingReply>();



                logging_debug( "received node listing reply from node " << remote.toString()

                                        << " with all overlay nodes. connecting to all of them" );



                //

                // get out all the EndpointDescriptors from the

                // overlay nodes and connect to all nodes where

                // we don't have a link yet

                //



                const NodeListingReply::NodeEndpointList& endpoints = reply->getList();

                logging_debug( "received " << endpoints.size() << " nodes in listing" );

                pendingLinks = 0;



                NodeListingReply::NodeEndpointList::const_iterator i = endpoints.begin();

                NodeListingReply::NodeEndpointList::const_iterator iend = endpoints.end();



                for( ; i != iend; i++ ){



                        //

                        // don't connect to nodes that we already have

                        // a link to and don't connect to ourself

                        //



                        const NodeID& node = (*i).first;

                        if( overlayNodes.find(node) != overlayNodes.end() ) continue;

                        if( node == nodeid ) continue;



                        logging_debug( "building up link to node in overlay " << node.toString() );

                        const LinkID link = baseoverlay.establishLink( *((*i).second),

                                                        OverlayInterface::OVERLAY_SERVICE_ID );



                        overlayNodes.insert( make_pair(node, link) );

                        pendingLinks++;



                } // for( ; i != iend; i++ )



        } // if( reply != NULL )



}



}} // namespace ariba, overlay