daemon/src/core/BundleStorage.cpp

Go to the documentation of this file.

/*
 * BundleStorage.cpp
 *
 *  Created on: 24.03.2009
 *      Author: morgenro
 */

#include "core/BundleCore.h"
#include "core/BundleStorage.h"
#include "core/CustodyEvent.h"
#include "core/BundleGeneratedEvent.h"
#include <ibrdtn/data/BundleID.h>

namespace dtn
{
        namespace core
        {
                BundleStorage::BundleStorage()
                {
                }

                BundleStorage::~BundleStorage()
                {
                }

                void BundleStorage::remove(const dtn::data::Bundle &b)
                {
                        remove(dtn::data::BundleID(b));
                };

                const dtn::data::EID BundleStorage::acceptCustody(const dtn::data::MetaBundle &meta)
                {
                        if (!meta.get(Bundle::CUSTODY_REQUESTED))
                                throw ibrcommon::Exception("custody transfer is not requested for this bundle.");

                        if (meta.custodian == EID())
                                throw ibrcommon::Exception("no previous custodian is set.");

                        // create a new bundle
                        Bundle custody_bundle;

                        // set priority to HIGH
                        custody_bundle.set(dtn::data::PrimaryBlock::PRIORITY_BIT1, false);
                        custody_bundle.set(dtn::data::PrimaryBlock::PRIORITY_BIT2, true);

                        // send a custody signal with accept flag
                        CustodySignalBlock &signal = custody_bundle.push_back<CustodySignalBlock>();

                        // set the bundle to match
                        signal.setMatch(meta);

                        // set accepted
                        signal._status |= 1;

                        custody_bundle.set(dtn::data::PrimaryBlock::DESTINATION_IS_SINGLETON, true);
                        custody_bundle._destination = meta.custodian;
                        custody_bundle._source = dtn::core::BundleCore::local;

                        // send the custody accepted bundle
                        dtn::core::BundleGeneratedEvent::raise(custody_bundle);

                        // raise the custody accepted event
                        dtn::core::CustodyEvent::raise(meta, CUSTODY_ACCEPT);

                        return dtn::core::BundleCore::local;
                }

                void BundleStorage::rejectCustody(const dtn::data::MetaBundle &meta)
                {
                        if (!meta.get(Bundle::CUSTODY_REQUESTED))
                                throw ibrcommon::Exception("custody transfer is not requested for this bundle.");

                        if (meta.custodian == EID())
                                throw ibrcommon::Exception("no previous custodian is set.");

                        // create a new bundle
                        Bundle b;

                        // send a custody signal with reject flag
                        CustodySignalBlock &signal = b.push_back<CustodySignalBlock>();

                        // set the bundle to match
                        signal.setMatch(meta);

                        b._destination = meta.custodian;
                        b._source = BundleCore::local;

                        // send the custody rejected bundle
                        dtn::core::BundleGeneratedEvent::raise(b);

                        // raise the custody rejected event
                        dtn::core::CustodyEvent::raise(b, CUSTODY_REJECT);
                }
        }
}