source: source/ariba/communication/BaseCommunication.h@ 10700

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

#ifndef BASECOMMUNICATION_H_
#define BASECOMMUNICATION_H_

// boost & std includes
#include <boost/unordered_map.hpp>
#include <boost/unordered_set.hpp>
#include <map>
#include <set>
#include <vector>
#include <iostream>
#include <algorithm>
#include <boost/foreach.hpp>

// utilities
#include "ariba/utility/types.h"
#include "ariba/utility/messages.h"
#include "ariba/utility/logging/Logging.h"
#include "ariba/utility/misc/Demultiplexer.hpp"
#include "ariba/utility/system/SystemEventListener.h"

// new transport and addressing
#include "ariba/utility/addressing/addressing.hpp"
#include "ariba/utility/transport/transport.hpp"
#include "ariba/utility/transport/transport_connection.hpp"

// communication
#include "ariba/communication/CommunicationEvents.h"
#include "ariba/communication/EndpointDescriptor.h"
#include "ariba/communication/messages/AribaBaseMsg.h"

// network changes
#include "ariba/communication/networkinfo/NetworkChangeInterface.h"
#include "ariba/communication/networkinfo/NetworkChangeDetection.h"
#include "ariba/communication/networkinfo/NetworkInformation.h"

// disabled
//#ifndef UNDERLAY_OMNET
//  #include "ariba/communication/modules/transport/tcp/TCPTransport.h"
//  #include "ariba/communication/modules/network/ip/IPv4NetworkProtocol.h"
//  using ariba::communication::IPv4NetworkProtocol;
//  using ariba::communication::TCPTransport;
//#endif

namespace ariba {
  class SideportListener;
}

namespace ariba {
namespace communication {

using namespace std;
using namespace ariba::addressing;
using namespace ariba::transport;
using namespace ariba::utility;

// use base ariba types (clarifies multiple definitions)
using ariba::utility::Message;
using ariba::utility::seqnum_t;

/**
 * This class implements the Ariba Base Communication<br />
 *
 * Its primary task is to provide an abstraction to existing
 * protocols and addressing schemes.
 *
 * @author Sebastian Mies, Christoph Mayer
 */
class BaseCommunication:
        public NetworkChangeInterface,
        public SystemEventListener,
        public transport_listener {

        use_logging_h(BaseCommunication);
        friend class ariba::SideportListener;

public:
        /// Default ctor that just creates an non-functional base communication
        BaseCommunication();

        /// Default dtor that does nothing
        virtual ~BaseCommunication();

        /// Startup the base communication, start modules etc.
        void start();

        /// Stops the base communication, stop modules etc.
        void stop();

        /// Sets the endpoints
        void setEndpoints( string& endpoints );

        /// Check whether the base communication has been started up
        bool isStarted();

        /// Establishes a link to another end-point.
        const LinkID establishLink(const EndpointDescriptor& descriptor,
                const LinkID& linkid = LinkID::UNSPECIFIED, const QoSParameterSet& qos =
                                QoSParameterSet::DEFAULT, const SecurityParameterSet& sec =
                                SecurityParameterSet::DEFAULT);

        /// Drops a link
        void dropLink(const LinkID link);

        /**
         * Sends a message though an existing link to an end-point.
         *
         * @param lid The link identifier
         * @param message The message to be sent
         * @return A sequence number for this message
         */
        seqnum_t sendMessage(const LinkID lid, const Message* message);

        /**
         * Returns the end-point descriptor
         *
         * @param link the link id of the requested end-point
         * @return The end-point descriptor of the link's end-point
         */
        const EndpointDescriptor& getEndpointDescriptor(const LinkID link =
                        LinkID::UNSPECIFIED) const;

        /**
         * Get local links to the given endpoint of all local link
         * using the default parameter EndpointDescriptor::UNSPECIFIED
         * @param ep The remote endpoint to get all links to.
         * @return List of LinkID
         */
        LinkIDs getLocalLinks(const address_v* addr) const;

        /**
         * Registers a receiver.
         *
         * @param _receiver The receiving side
         */
        void registerMessageReceiver(MessageReceiver* receiver) {
                messageReceiver = receiver;
        }

        /**
         * Unregister a receiver.
         *
         * @param _receiver The receiving side
         */
        void unregisterMessageReceiver(MessageReceiver* receiver) {
                messageReceiver = NULL;
        }

        void registerEventListener(CommunicationEvents* _events);

        void unregisterEventListener(CommunicationEvents* _events);

        /// called when a system event is emitted by system queue
        virtual void handleSystemEvent(const SystemEvent& event);

        /**
         * called within the ASIO thread
         * when a message is received from underlay transport
         */ 
        virtual void receive_message(transport_connection::sptr connection,
                reboost::message_t msg);

protected:

        /**
         * called within the ARIBA thread (System Queue)
         * when a message is received from underlay transport
         */ 
        void receiveMessage(transport_connection::sptr connection,
                reboost::message_t msg);

        /// called when a network interface change happens
        virtual void onNetworkChange(
                const NetworkChangeInterface::NetworkChangeInfo& info);

private:
        /**
         * A link descriptor consisting of the end-point descriptor and currently
         * used underlay address.
         */
        class LinkDescriptor {
        public:

                /// default constructor
                LinkDescriptor() :
                        localLink(LinkID::UNSPECIFIED), localLocator(NULL),
                        remoteLink(LinkID::UNSPECIFIED), remoteLocator(NULL),
                        up(false) {
                }

                ~LinkDescriptor() {
                        if (localLocator!=NULL)  delete localLocator;
                        if (remoteLocator!=NULL) delete remoteLocator;
                }

                bool isUnspecified() const {
                        return (this == &UNSPECIFIED());
                }

                static LinkDescriptor& UNSPECIFIED(){
                        static LinkDescriptor* unspec = NULL;
                        if(unspec == NULL) unspec = new LinkDescriptor();
                        return *unspec;
                }

                bool unspecified;

                /// link identifiers
                LinkID localLink;
                const address_v* localLocator;

                /// used underlay addresses for the link
                LinkID remoteLink;
                const address_v* remoteLocator;

                /// the remote end-point descriptor
                EndpointDescriptor remoteEndpoint;

                /// flag, whether this link is up
                bool up;
                
                /// connection if link is up
                transport_connection::sptr connection;
        };

        /// Link management: list of links
        typedef vector<LinkDescriptor*> LinkSet;

        /// Link management: the set of currently managed links
        LinkSet linkSet;

        /// Link management: add a link
        void addLink( LinkDescriptor* link );

        /// Link management: remove a link
        void removeLink(const LinkID& localLink);

        /// Link management: get link information using the local link
        LinkDescriptor& queryLocalLink(const LinkID& localLink) const;

        /// Link management: get link information using the remote link
        LinkDescriptor& queryRemoteLink(const LinkID& remoteLink) const;

        /// The local end-point descriptor
        EndpointDescriptor localDescriptor;

#ifndef UNDERLAY_OMNET
        /// network change detector
        NetworkChangeDetection networkMonitor;
#endif

        /// list of all remote addresses of links to end-points
        class endpoint_reference {
        public:
                int count; ///< the number of open links to this end-point
                const address_v* endpoint; ///< the end-point itself
        };
        vector<endpoint_reference> remote_endpoints;

        /// adds an end-point to the list
        void add_endpoint( const address_v* endpoint );

        /// removes an end-point from the list
        void remove_endpoint( const address_v* endpoint );

        /// event listener
        typedef set<CommunicationEvents*> EventListenerSet;
        EventListenerSet eventListener;

        /// sequence numbers
        seqnum_t currentSeqnum;

        /// transport peer
        transport_peer* transport;

        /// the base overlay message receiver
        MessageReceiver* messageReceiver;

        /// convenience: send message to peer
        void send( Message* message, const EndpointDescriptor& endpoint );
        void send( Message* message, const LinkDescriptor& descriptor );

        /// state of the base communication
        bool started;

};

}} // namespace ariba, communication

#endif /* BASECOMMUNICATION_H_ */