source: source/ariba/AribaModule.cpp@ 5317

// [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 "AribaModule.h"

// boost/std includes
#include <assert.h>
#include <boost/regex.hpp>
#include <boost/foreach.hpp>

// ariba includes
#include "ariba/SideportListener.h"
#include "ariba/utility/misc/Helper.h"
#include "ariba/utility/misc/StringFormat.h"
#include "ariba/communication/BaseCommunication.h"
#include "ariba/communication/EndpointDescriptor.h"

using namespace ariba::utility::Helper;
using ariba::communication::BaseCommunication;
using ariba::communication::EndpointDescriptor;

namespace ariba {

use_logging_cpp(AribaModule);

AribaModule::AribaModule()
        : base_comm(NULL), sideport_sniffer(NULL), started(false) {

        endpoints = "tcp{41322}";

        //srand( time(NULL) );
        //uint16_t tcpport = (rand() + 1024) % 50000;
        //uint16_t udpport = (rand() + 1024) % 50000;
}

AribaModule::~AribaModule() {
}

string AribaModule::getBootstrapHints(const Name& spoVNetName) const {
        std::ostringstream o;
        BOOST_FOREACH( const BootstrapInfo& info, bootstrapNodes ) {
                o << info.spovnetName.toString() << "{";
                int i=0;
                BOOST_FOREACH( const BootstrapNode& node, info.nodes ) {
                        if (i!=0) o << ",";
                        if( node.desc != NULL ) o << node.desc->toString();
                        i++;
                }
                o << "}";
        }
        return o.str();
}

void AribaModule::addBootstrapHints(string boot_info) {
        using namespace boost::xpressive;
        using namespace ariba::utility::string_format;
        using namespace ariba::utility::Helper;
        using namespace std;

        smatch match;
        if (regex_search(boot_info, match, robjects)) {
                regex_nav nav = match;
                for (int i = 0; i < nav.size(); i++) {
                        string type = nav[i][robject_id].str();
                        string data = nav[i][robject_data].str();
                        data = data.substr(1, data.size() - 2);
                        Name name(type);
                        EndpointDescriptor* desc = EndpointDescriptor::fromString(data);
                        logging_debug("Added bootstap info for " << type << ": " << desc->toString() );
                        addBootstrapNode(name, desc);
                }
        }
}

void AribaModule::addBootstrapNode(const Name& spovnet,
                communication::EndpointDescriptor* desc) {

        // set bootstrap node
        BootstrapNode node;
        node.timestamp = 0;
        node.desc = desc;
        bool added = false;

        // add node to existing bootstrap list
        BOOST_FOREACH( BootstrapInfo& info, bootstrapNodes ){
                if (info.spovnetName == spovnet) {
                        info.nodes.push_back(node);
                        added = true;
                        break;
                }
        }

        // create new entry
        if (!added) {
                BootstrapInfo info;
                info.spovnetName = spovnet;
                info.nodes.push_back(node);
                bootstrapNodes.push_back(info);
        }

        logging_debug( "Added bootstrap info: " << getBootstrapHints() );
}

const communication::EndpointDescriptor* AribaModule::getBootstrapNode(
                const Name& spovnet) const {
        BOOST_FOREACH( const BootstrapInfo& info, bootstrapNodes ) {
                if( info.spovnetName == spovnet ) {
                        BOOST_FOREACH( const BootstrapNode& node, info.nodes ) {
                                if( node.desc != NULL ) return node.desc;
                        }
                }
        }
        return NULL;
}

void AribaModule::registerSideportListener(SideportListener* sideport){
        sideport_sniffer = sideport;
}

// @see Module.h
void AribaModule::initialize() {

        // preconditions
        assert(!started);

        // init variables
        base_comm = NULL;
}

// @see Module.h
void AribaModule::start() {

        // preconditions
        assert(base_comm == NULL);
        assert(!started);

        // create the base communication component
        started = true;
        base_comm = new BaseCommunication();
        base_comm->setEndpoints(endpoints);
}

// @see Module.h
void AribaModule::stop() {

        // preconditions
        assert(base_comm != NULL);
        assert(started);

        // delete base communication component
        started = false;
        delete base_comm;
}

// @see Module.h
string AribaModule::getName() const {
        return "ariba";
}

// @see Module.h
void AribaModule::setProperty(string key, string value) {
        if (key == "endpoints") endpoints = value;
        else if (key == "bootstrap.hints") addBootstrapHints(value);
}

// @see Module.h
const string AribaModule::getProperty(string key) const {
        if (key == "endpoints") return endpoints; // TODO: return local endpoints
        else if (key == "bootstrap.hints") return getBootstrapHints();
}

// @see Module.h
const vector<string> AribaModule::getProperties() const {
        vector<string> properties;
        properties.push_back("endpoints");
        properties.push_back("bootstrap.hints");
        return properties;
}

} // namespace ariba