daemon/src/routing/epidemic/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 "core/AbstractWorker.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/ExtensionBlock.h>

#include <ibrcommon/thread/Queue.h>

#include <list>
#include <queue>

namespace dtn
{
        namespace routing
        {
                class EpidemicRoutingExtension : public BaseRouter::ThreadedExtension
                {
                public:
                        EpidemicRoutingExtension();
                        virtual ~EpidemicRoutingExtension();

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

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

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

                        class ExecutableTask : public Task
                        {
                        public:
                                virtual ~ExecutableTask() {};
                                virtual std::string toString() = 0;
                                virtual void execute() const = 0;
                        };

                        class EpidemicEndpoint : public dtn::core::AbstractWorker
                        {
                        public:
                                EpidemicEndpoint(ibrcommon::Queue<EpidemicRoutingExtension::Task* > &queue, dtn::routing::BundleSummary &purge);
                                virtual ~EpidemicEndpoint();

                                void callbackBundleReceived(const Bundle &b);
                                void query(const dtn::data::EID &eid);

                                void send(const dtn::data::Bundle &b);

                        private:
                                ibrcommon::Queue<EpidemicRoutingExtension::Task* > &_taskqueue;
                                dtn::routing::BundleSummary &_purge_vector;
                        };

                        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 ProcessEpidemicBundleTask : public Task
                        {
                        public:
                                ProcessEpidemicBundleTask(const dtn::data::Bundle &b);
                                virtual ~ProcessEpidemicBundleTask();

                                virtual std::string toString();

                                const dtn::data::Bundle bundle;
                        };

                        class QuerySummaryVectorTask : public ExecutableTask
                        {
                        public:
                                QuerySummaryVectorTask(NeighborDatabase &ndb, const dtn::data::EID &origin, EpidemicEndpoint &endpoint);
                                virtual ~QuerySummaryVectorTask();

                                virtual std::string toString();
                                virtual void execute() const;

                                NeighborDatabase &ndb;
                                const dtn::data::EID origin;
                                EpidemicEndpoint &endpoint;
                        };

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

                                virtual std::string toString();

                                const dtn::data::EID peer;
                                const dtn::data::MetaBundle meta;
                        };

                        void processECM(const dtn::data::Bundle &b);

                        dtn::routing::BundleSummary _purge_vector;

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

                        EpidemicEndpoint _endpoint;
                };
        }
}

#endif /* EPIDEMICROUTINGEXTENSION_H_ */