source: source/ariba/AribaModule.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 "AribaModule.h"

// boost/std includes
#include <assert.h>
#include <boost/regex.hpp>
#include <boost/foreach.hpp>
#include <boost/algorithm/string.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);
const string AribaModule::BootstrapMechanismNames[5] = {"invalid", "static", "broadcast", "mdns", "sdp"};

AribaModule::AribaModule()
        : started(false), base_comm(NULL), sideport_sniffer(NULL) {
        endpoints = "tcp{41322};rfcomm{10};";
}

AribaModule::~AribaModule() {
}

string AribaModule::getBootstrapHints(const Name& spoVNetName) const {
        std::ostringstream o;
        int i=0;
        BOOST_FOREACH( const BootstrapInfo& info, bootstrapNodes ) {
                o << info.spovnetName.toString() << "{";
                if (i!=0) o << ",";

                BOOST_FOREACH( const BootstrapNode& node, info.nodes ) {
                        if( node.desc != NULL )
                                o << node.desc->toString();
                        else if(node.mechanism == BootstrapMechanismBroadcast
                                        || node.mechanism == BootstrapMechanismMulticastDNS
                                        || node.mechanism == BootstrapMechanismSDP){
                                o << BootstrapMechanismNames[node.mechanism];
                                if( !node.info.empty() ) o << "{" << node.info << "}";
                                o << ";";
                        }
                }
                o << "}";
                i++;
        }
        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;

        boost::erase_all(boot_info, " ");
        boost::erase_all(boot_info, "\t");
        boost::erase_all(boot_info, "\n");
        boost::erase_all(boot_info, "\r");

        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);

                        // find static bootstrap info --> BootstrapMechanismStatic
                        EndpointDescriptor* desc = EndpointDescriptor::fromString(data);
                        addBootstrapNode(name, desc, "", BootstrapMechanismStatic);

                        // find automatic bootstrap info --> {BootstrapMechanismBroadcast,
                        //                                      BootstrapMechanismMulticastDNS,BootstrapMechanismSDP}
                        typedef vector< string > split_vector_type;
                        split_vector_type splitvec;
                        boost::split( splitvec, data, boost::is_any_of(";") );
                        for(unsigned int i=0; i<splitvec.size(); i++){
                                string x = splitvec[i];
                                split_vector_type innervec;

                                boost::split( innervec, x, boost::is_any_of("{}") );
                                BootstrapNode node;
                                if(innervec.size() < 1) continue;

                                if(innervec[0] == BootstrapMechanismNames[BootstrapMechanismBroadcast])
                                        node.mechanism = BootstrapMechanismBroadcast;
                                else if(innervec[0] == BootstrapMechanismNames[BootstrapMechanismMulticastDNS])
                                        node.mechanism = BootstrapMechanismMulticastDNS;
                                else if(innervec[0] == BootstrapMechanismNames[BootstrapMechanismSDP])
                                        node.mechanism = BootstrapMechanismSDP;
                                else
                                        continue;

                                if(innervec.size() > 1)
                                        node.info = innervec[1];

                                this->addBootstrapNode(name, node);
                        }
                }
        }

        logging_info( "Added bootstrap hints: " << getBootstrapHints() );
}

void AribaModule::addBootstrapNode(const Name& spovnet,
                communication::EndpointDescriptor* desc, const string& info,
                const BootstrapMechanism& mechanism) {
        BootstrapNode node(0, desc, mechanism, info);
        addBootstrapNode(spovnet, node);
}

void AribaModule::addBootstrapNode(const Name& spovnet, const BootstrapNode& node){
        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);
        }
}

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

string AribaModule::getBootstrapInfo(
                const Name& spovnet, const BootstrapMechanism mechanism) const {
        BOOST_FOREACH( const BootstrapInfo& info, bootstrapNodes ) {
                if( info.spovnetName == spovnet ) {
                        BOOST_FOREACH( const BootstrapNode& node, info.nodes ) {
                                if( node.mechanism == mechanism && node.desc != NULL )
                                        return node.info;
                        }
                }
        }

        return string();
}

vector<AribaModule::BootstrapMechanism> AribaModule::getBootstrapMechanisms(
                const Name& spovnet) const {
        vector<AribaModule::BootstrapMechanism> ret;
        BOOST_FOREACH( const BootstrapInfo& info, bootstrapNodes ) {
                if( info.spovnetName == spovnet ) {
                        BOOST_FOREACH( const BootstrapNode& node, info.nodes ) {
                                if(std::find(ret.begin(), ret.end(), node.mechanism) == ret.end())
                                        ret.push_back(node.mechanism);
                        }
                }
        }
        return ret;
}

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();
        return "";
}

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

const string AribaModule::getLocalEndpoints() {
        return base_comm->getEndpointDescriptor().toString();
}

} // namespace ariba