daemon/src/routing/NeighborDatabase.cpp

Go to the documentation of this file.

/*
 * NeighborDatabase.cpp
 *
 *  Created on: 23.07.2010
 *      Author: morgenro
 */

#include "routing/NeighborDatabase.h"
#include <ibrdtn/utils/Clock.h>
#include <ibrcommon/Logger.h>
#include <limits>

namespace dtn
{
        namespace routing
        {
                NeighborDatabase::NeighborEntry::NeighborEntry()
                 : eid(), _transit_max(5), _filter(), _filter_expire(0), _filter_state(FILTER_EXPIRED)
                {};

                NeighborDatabase::NeighborEntry::NeighborEntry(const dtn::data::EID &e)
                 : eid(e), _transit_max(5), _filter(), _filter_expire(0), _filter_state(FILTER_EXPIRED)
                { }

                NeighborDatabase::NeighborEntry::~NeighborEntry()
                { }

                void NeighborDatabase::NeighborEntry::update(const ibrcommon::BloomFilter &bf, const size_t lifetime)
                {
                        _filter = bf;

                        if (lifetime == 0)
                        {
                                _filter_expire = std::numeric_limits<std::size_t>::max();
                        }
                        else
                        {
                                _filter_expire = dtn::utils::Clock::getExpireTime(lifetime);
                        }

                        _filter_state = FILTER_AVAILABLE;
                }

                void NeighborDatabase::NeighborEntry::reset()
                {
                        _filter_state = FILTER_EXPIRED;
                }

                void NeighborDatabase::NeighborEntry::add(const dtn::data::MetaBundle &bundle)
                {
                        _summary.add(bundle);
                }

                bool NeighborDatabase::NeighborEntry::has(const dtn::data::BundleID &id, const bool require_bloomfilter) const
                {
                        if (require_bloomfilter && (_filter_state != FILTER_AVAILABLE))
                                throw BloomfilterNotAvailableException(eid);

                        if (_filter_state == FILTER_AVAILABLE)
                        {
                                if (_filter.contains(id.toString()))
                                        return true;
                        }

                        if (_summary.contains(id))
                                return true;

                        return false;
                }

                void NeighborDatabase::NeighborEntry::expire(const size_t timestamp)
                {
                        if ((_filter_expire > 0) && (_filter_expire < timestamp))
                        {
                                IBRCOMMON_LOGGER_DEBUG(15) << "summary vector of " << eid.getString() << " is expired" << IBRCOMMON_LOGGER_ENDL;

                                // set the filter state to expired once
                                _filter_state = FILTER_EXPIRED;
                        }

                        _summary.expire(timestamp);
                }

                void NeighborDatabase::expire(const size_t timestamp)
                {
                        for (std::map<dtn::data::EID, NeighborDatabase::NeighborEntry* >::const_iterator iter = _entries.begin(); iter != _entries.end(); iter++)
                        {
                                (*iter).second->expire(timestamp);
                        }
                }

                void NeighborDatabase::NeighborEntry::acquireFilterRequest() throw (NoMoreTransfersAvailable)
                {
                        if (_filter_state != FILTER_EXPIRED)
                                throw NoMoreTransfersAvailable();

                        _filter_state = FILTER_AWAITING;
                }

                void NeighborDatabase::NeighborEntry::acquireTransfer(const dtn::data::BundleID &id) throw (NoMoreTransfersAvailable, AlreadyInTransitException)
                {
                        ibrcommon::MutexLock l(_transit_lock);

                        // check if the bundle is already in transit
                        if (_transit_bundles.find(id) != _transit_bundles.end()) throw AlreadyInTransitException();

                        // check if enough resources available to transfer the bundle
                        if (_transit_bundles.size() >= _transit_max) throw NoMoreTransfersAvailable();

                        // insert the bundle into the transit list
                        _transit_bundles.insert(id);

                        IBRCOMMON_LOGGER_DEBUG(20) << "acquire transfer of " << id.toString() << " (" << _transit_bundles.size() << " bundles in transit)" << IBRCOMMON_LOGGER_ENDL;
                }

                void NeighborDatabase::NeighborEntry::releaseTransfer(const dtn::data::BundleID &id)
                {
                        ibrcommon::MutexLock l(_transit_lock);
                        _transit_bundles.erase(id);

                        IBRCOMMON_LOGGER_DEBUG(20) << "release transfer of " << id.toString() << " (" << _transit_bundles.size() << " bundles in transit)" << IBRCOMMON_LOGGER_ENDL;
                }

                NeighborDatabase::NeighborDatabase()
                {
                }

                NeighborDatabase::~NeighborDatabase()
                {
                        std::set<dtn::data::EID> ret;

                        for (std::map<dtn::data::EID, NeighborDatabase::NeighborEntry* >::const_iterator iter = _entries.begin(); iter != _entries.end(); iter++)
                        {
                                delete (*iter).second;
                        }
                }

                NeighborDatabase::NeighborEntry& NeighborDatabase::create(const dtn::data::EID &eid)
                {
                        if (_entries.find(eid) == _entries.end())
                        {
                                NeighborEntry *entry = new NeighborEntry(eid);
                                _entries[eid] = entry;
                        }

                        return (*_entries[eid]);
                }

                NeighborDatabase::NeighborEntry& NeighborDatabase::get(const dtn::data::EID &eid) throw (NeighborNotAvailableException)
                {
                        if (_entries.find(eid) == _entries.end())
                        {
                                throw NeighborNotAvailableException();
                        }

                        return (*_entries[eid]);
                }

                NeighborDatabase::NeighborEntry& NeighborDatabase::reset(const dtn::data::EID &eid)
                {
                        try {
                                NeighborDatabase::NeighborEntry &e = get(eid);
                                e.reset();
                        } catch (const NeighborNotAvailableException&) { };
                }

                void NeighborDatabase::remove(const dtn::data::EID &eid)
                {
                        _entries.erase(eid);
                }

                void NeighborDatabase::addBundle(const dtn::data::EID &neighbor, const dtn::data::MetaBundle &b)
                {
                        try {
                                NeighborDatabase::NeighborEntry &entry = get(neighbor);
                                entry.add(b);
                        } catch (const NeighborDatabase::NeighborNotAvailableException&) { };
                }
        }
}