source: source/ariba/overlay/OverlayBootstrap.cpp@ 7532

// [License]
// 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.
// [License]

#include "OverlayBootstrap.h"

#include "BaseOverlay.h"
#include "ariba/utility/bootstrap/modules/bluetoothsdp/BluetoothSdp.h"

using ariba::utility::BluetoothSdp;

namespace ariba {
namespace overlay {

use_logging_cpp(OverlayBootstrap);
SystemEventType OverlayBootstrapMethodType("OverlayBootstrapMethodType");

OverlayBootstrap::OverlayBootstrap()
        :       manager( BootstrapManager::instance() ),
                spovnetid( SpoVNetID::UNSPECIFIED ),
                nodeid( NodeID::UNSPECIFIED ),
                overlay( NULL ),
                watchtimer(this),
                haveOverlayConnection(false){

        srand(time(NULL));

        BluetoothSdp::CONNECTION_CHECKER = this;
}

OverlayBootstrap::~OverlayBootstrap(){
}

void OverlayBootstrap::start(BaseOverlay* _overlay,
                const SpoVNetID& _spovnetid, const NodeID& _nodeid,
                vector<pair<BootstrapManager::BootstrapType,string> > modules){
        overlay = _overlay;
        spovnetid = _spovnetid;
        nodeid = _nodeid;

        logging_info("starting overlay bootstrap");

        manager.registerCallback( this );

        typedef pair<BootstrapManager::BootstrapType,string> X;
        BOOST_FOREACH( X i, modules){
                manager.registerModule( i.first, i.second );
        }

        watchtimer.startWatchdog();
}

void OverlayBootstrap::stop(){
        overlay = NULL;
        spovnetid = SpoVNetID::UNSPECIFIED;
        nodeid = NodeID::UNSPECIFIED;

        logging_info("stopping overlay bootstrap");

        manager.unregisterCallback( this );
        manager.unregisterAllModules();

        watchtimer.stopWatchdog();
}

void OverlayBootstrap::handleSystemEvent(const SystemEvent& event){
        JoinData* data = event.getData<JoinData>();

        // announcement for our spovnet
        logging_info( "found bootstrap node for our SpoVNetID " << data->spovnetid.toString()
                        << " on NodeID " << data->nodeid.toString() << " with endpoint " << data->endpoint.toString() );

        // tell the base overlay to join using this endpoint
        assert( overlay != NULL );
        overlay->joinSpoVNet( spovnetid, data->endpoint );

        delete data;
}

void OverlayBootstrap::onBootstrapServiceFound(string name, string info1, string info2, string info3){
        if( overlay == NULL ) return;
        if(name.length() <= 0 || info1.length() <= 0 || info2.length() <= 0 || info3.length() <= 0) return;

        //
        // generate the types
        //

        SpoVNetID sid( info1 );
        NodeID nid( info2 );
        EndpointDescriptor ep( info3 );

        //
        // is this announcement of interest for us?
        //

        // announcement for another spovnet
        if( sid != this->spovnetid ) return;

        // announcement with our nodeid (either our announcement
        // or a node with the same id, any way -> ignore)
        if( nid == this->nodeid ) return;

        //
        // send out the bootstrap information as
        // event to synchronize into the system queue
        //

        JoinData* data = new JoinData();
        data->spovnetid = sid;
        data->nodeid = nid;
        data->endpoint = ep;

        SystemQueue::instance().scheduleEvent(
                        SystemEvent( this, OverlayBootstrapMethodType, data), 0 );
}

void OverlayBootstrap::publish(const EndpointDescriptor& _ep){

        ostringstream r;
        r << std::hex << rand();

        randname = r.str();
        manager.publish( randname, spovnetid.toString(), nodeid.toString(), _ep.toString() );
}

void OverlayBootstrap::revoke(){
        manager.revoke( randname );
}

void OverlayBootstrap::recordJoin(const EndpointDescriptor& _ep){
        boost::mutex::scoped_lock lock(lastJoinesMutex);

        JoinData data;
        data.spovnetid = spovnetid;
        data.nodeid = nodeid;
        data.endpoint = _ep;

        logging_info("recording bootstrap information " << data.endpoint.toString());

        lastJoines.push_front(data);
}

bool OverlayBootstrap::haveOverlayConnections(){
        boost::mutex::scoped_lock lock(haveOverlayConnectionMutex);
        return haveOverlayConnection;
}

void OverlayBootstrap::checkOverlayStatus(){

        // if we have no overlay neighbors, try to bootstrap using
        // bootstrap information that we already used

        {       //limit history to 10 endpoints
                boost::mutex::scoped_lock lock(lastJoinesMutex);
                while(lastJoines.size() > 10)
                        lastJoines.pop_back();
        }

        {
                boost::mutex::scoped_lock lock(haveOverlayConnectionMutex);
                haveOverlayConnection = overlay->getOverlayNeighbors().size() > 0;

                // we have overlay neighbors -> ok nothing to do
                if(haveOverlayConnection > 0) return;
        }

        // no overlay neighbors, see if we can join using old information
        logging_info("overlay not joined, checking for earlier used bootstrap information");
        EndpointDescriptor joinep = EndpointDescriptor::UNSPECIFIED();

        // no overlay neighbors -> try out already
        // successfully used bootstrap nodes
        JoinData data;
        {
                boost::mutex::scoped_lock lock(lastJoinesMutex);
                JoinStack::iterator i = lastJoines.begin();
                if(i == lastJoines.end()) return;

                // use last used element and then put it into back
                joinep = (*i).endpoint;

                if(lastJoines.size() >= 2)
                        swap( *lastJoines.begin(), *(--(lastJoines.end())) );
        }

        logging_info("no overlay conenctivity detected, " <<
                                        "trying to join using old bootstrap information: " <<
                                        joinep.toString());

        // try to join using this node, if the join is successfull
        // the endpoint will again be inserted using recordJoin
        overlay->joinSpoVNet( spovnetid, joinep );
}

OverlayBootstrap::WatchdogTimer::WatchdogTimer(OverlayBootstrap* _obj) : obj(_obj) {
}

void OverlayBootstrap::WatchdogTimer::startWatchdog(){
        Timer::setInterval(5000);
        Timer::start();
}

void OverlayBootstrap::WatchdogTimer::stopWatchdog(){
        Timer::stop();
}

void OverlayBootstrap::WatchdogTimer::eventFunction(){
        if(obj == NULL) return;
        obj->checkOverlayStatus();
}

}} // namespace ariba, overlay