daemon/src/routing/EpidemicRoutingExtension.h

Go to the documentation of this file.

/*
 * EpidemicRoutingExtension.h
 *
 *  Created on: 18.02.2010
 *      Author: morgenro
 */

#ifndef EPIDEMICROUTINGEXTENSION_H_
#define EPIDEMICROUTINGEXTENSION_H_

#include "core/Node.h"

#include "net/DiscoveryServiceProvider.h"
#include "routing/SummaryVector.h"
#include "routing/BaseRouter.h"
#include "routing/NeighborDatabase.h"

#include <ibrdtn/data/Block.h>
#include <ibrdtn/data/SDNV.h>
#include <ibrdtn/data/BundleString.h>
#include <ibrdtn/data/BundleList.h>
#include <ibrdtn/data/ExtensionBlockFactory.h>

#include <ibrcommon/thread/Queue.h>

#include <list>
#include <queue>

namespace dtn
{
        namespace routing
        {
                class EpidemicRoutingExtension : public BaseRouter::ThreadedExtension, public dtn::net::DiscoveryServiceProvider
                {
                private:
                        static const dtn::data::EID EPIDEMIC_ROUTING_ADDRESS;

                public:
                        EpidemicRoutingExtension();
                        ~EpidemicRoutingExtension();

                        void notify(const dtn::core::Event *evt);

                        void update(std::string &name, std::string &data);

                        virtual void stopExtension();

                        class EpidemicExtensionBlock : public dtn::data::Block
                        {
                        public:
                                class Factory : public dtn::data::ExtensionBlockFactory
                                {
                                public:
                                        Factory() : dtn::data::ExtensionBlockFactory(EpidemicExtensionBlock::BLOCK_TYPE) {};
                                        virtual ~Factory() {};
                                        virtual dtn::data::Block* create();
                                };

                                static const char BLOCK_TYPE = 201;

                                EpidemicExtensionBlock();
                                ~EpidemicExtensionBlock();

                                void set(dtn::data::SDNV value);
                                dtn::data::SDNV get() const;

                                void setPurgeVector(const SummaryVector &vector);
                                void setSummaryVector(const SummaryVector &vector);
                                const SummaryVector& getSummaryVector() const;
                                const SummaryVector& getPurgeVector() const;

                                virtual size_t getLength() const;
                                virtual std::ostream &serialize(std::ostream &stream) const;
                                virtual std::istream &deserialize(std::istream &stream);

                        private:
                                dtn::data::SDNV _counter;
                                dtn::data::BundleString _data;
                                SummaryVector _vector;
                                SummaryVector _purge;
                        };

                protected:
                        void run();
                        bool __cancellation();

                private:
                        class Task
                        {
                        public:
                                virtual ~Task() {};
                                virtual std::string toString() = 0;
                        };

                        class ExpireTask : public Task
                        {
                        public:
                                ExpireTask(const size_t timestamp);
                                virtual ~ExpireTask();

                                virtual std::string toString();

                                const size_t timestamp;
                        };

                        class SearchNextBundleTask : public Task
                        {
                        public:
                                SearchNextBundleTask(const dtn::data::EID &eid);
                                virtual ~SearchNextBundleTask();

                                virtual std::string toString();

                                const dtn::data::EID eid;
                        };

                        class TransferSummaryVectorTask : public Task
                        {
                        public:
                                TransferSummaryVectorTask(const dtn::data::EID &eid);
                                virtual ~TransferSummaryVectorTask();

                                virtual std::string toString();

                                const dtn::data::EID eid;
                        };

                        class ProcessBundleTask : public Task
                        {
                        public:
                                ProcessBundleTask(const dtn::data::MetaBundle &meta, const dtn::data::EID &origin);
                                virtual ~ProcessBundleTask();

                                virtual std::string toString();

                                const dtn::data::MetaBundle bundle;
                                const dtn::data::EID origin;
                        };

                        void prepareEpidemicInfo(dtn::data::Bundle &b);

                        ibrcommon::Mutex _list_mutex;

                        dtn::routing::BundleSummary _purge_vector;

                        dtn::routing::NeighborDatabase _neighbors;

                        ibrcommon::Queue<EpidemicRoutingExtension::Task* > _taskqueue;

                        dtn::data::Bundle _epidemic_bundle;
                };

                static EpidemicRoutingExtension::EpidemicExtensionBlock::Factory __EpidemicExtensionFactory__;
        }
}

#endif /* EPIDEMICROUTINGEXTENSION_H_ */