PayloadIntegrityBlock.h

Go to the documentation of this file.

/*
 * PayloadIntegrityBlock.h
 *
 * Copyright (C) 2011 IBR, TU Braunschweig
 *
 * Written-by: Johannes Morgenroth <morgenroth@ibr.cs.tu-bs.de>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

#ifndef _PAYLOAD_INTEGRITY_BLOCK_H_
#define _PAYLOAD_INTEGRITY_BLOCK_H_

#include "ibrdtn/security/SecurityBlock.h"
#include "ibrdtn/security/SecurityKey.h"
#include "ibrdtn/data/ExtensionBlock.h"
#include "ibrdtn/data/Bundle.h"
#include <openssl/evp.h>

namespace dtn
{
        namespace security
        {
                class PayloadIntegrityBlock : public SecurityBlock
                {
                        friend class dtn::data::Bundle;
                        public:
                                class Factory : public dtn::data::ExtensionBlock::Factory
                                {
                                public:
                                        Factory() : dtn::data::ExtensionBlock::Factory(PayloadIntegrityBlock::BLOCK_TYPE) {};
                                        virtual ~Factory() {};
                                        virtual dtn::data::Block* create();
                                };

                                static const dtn::data::block_t BLOCK_TYPE;
                                
                                virtual ~PayloadIntegrityBlock();

                                static void sign(dtn::data::Bundle &bundle, const SecurityKey &key, const dtn::data::EID& destination);

                                static void verify(const dtn::data::Bundle &bundle, const SecurityKey &key);

                                static void strip(dtn::data::Bundle& bundle, const SecurityKey &key, const bool all = false);

                                static void strip(dtn::data::Bundle& bundle);

                                virtual std::istream &deserialize(std::istream &stream, const dtn::data::Length &length);

                        protected:
                                PayloadIntegrityBlock();
                                
                                virtual dtn::data::Length getSecurityResultSize() const;

                        private:
                                dtn::data::Length result_size;

                                static const std::string calcHash(const dtn::data::Bundle &bundle, const SecurityKey &key, PayloadIntegrityBlock& ignore);

                                static void verify(const dtn::data::Bundle& bundle, const SecurityKey &key, const PayloadIntegrityBlock &sb, const bool use_eid = true);

                                void setResultSize(const SecurityKey &key);
                };

                static PayloadIntegrityBlock::Factory __PayloadIntegrityBlockFactory__;
        }
}
#endif /* _PAYLOAD_INTEGRITY_BLOCK_H_ */