source: source/ariba/utility/transport/messages/shared_buffer.hpp@ 12060

//-----------------------------------------------------------------------------
// Part of reboost (http://reboost.org).  Released under the
// BSD 2-clause license (http://www.opensource.org/licenses/bsd-license.php).
// Copyright 2012, Sebastian Mies <mies@reboost.org> --- All rights reserved.
//-----------------------------------------------------------------------------

#ifndef REBOOST_SHARED_BUFFER_HPP_
#define REBOOST_SHARED_BUFFER_HPP_

#include <cstring>
#include <string>
#include <boost/shared_ptr.hpp>

#include "ariba/config.h"
#ifdef DEBUG
#include <boost/thread/mutex.hpp>
#endif

#include "buffer.hpp"

#include <stdexcept> 

namespace reboost {

class illegal_sub_buffer: public std::runtime_error
{
public:
    /** Takes a character string describing the error.  */
    explicit illegal_sub_buffer(const std::string& __arg)  :
        std::runtime_error(__arg)
    {
    }
    
    virtual ~illegal_sub_buffer() throw() {}
};

/**
 * A simple shared buffer.
 *
 * Important: if not shared, a buffer is writable. After the buffer is shared,
 * the buffer is immutable. It uses shared_ptr/default allocators and prints
 * error messages to <tt>cerr</tt> if buffers leaked at the end of the
 * program.
 *
 * @author Sebastian Mies <mies@reboost.org>
 */
class shared_buffer_t: public buffer_t {
        typedef shared_buffer_t self;
private:
        static void onexit();
        static size_t init();
        
#ifdef DEBUG
        static size_t allocated_buffers;
        static boost::mutex allocated_buffers_mutex;
#endif

        class deleteable_buffer: public buffer_t {
        public:
                deleteable_buffer(bsize_t size) :
                        buffer_t(size)
                {
                        
#ifdef DEBUG
                        boost::mutex::scoped_lock lock(allocated_buffers_mutex);
                        allocated_buffers++;
#endif
                        
                }
                
                deleteable_buffer(boctet_t* data, bsize_t size) :
                        buffer_t(size, data)
                {
                        
#ifdef DEBUG
                        boost::mutex::scoped_lock lock(allocated_buffers_mutex);
                        allocated_buffers++;
#endif
                        
                }
                
                ~deleteable_buffer() {
                        if (!is_null()) delete[] data_;
                        
#ifdef DEBUG
                        boost::mutex::scoped_lock lock(allocated_buffers_mutex);
                        allocated_buffers--;
#endif
                        
                }
        };
        boost::shared_ptr<deleteable_buffer> parent;

public:
        /// create null-buffer
        inline shared_buffer_t() :
                buffer_t(), parent() {
        }

        /// create a shared buffer of a specific size
        inline shared_buffer_t(bsize_t size) :
                buffer_t(), parent(new deleteable_buffer(size)) {
                buffer_t::operator=(*parent);
        }

        /// create shared buffer from string
        inline shared_buffer_t(const char* string) :
                buffer_t(), parent() {
                parent.reset(new deleteable_buffer(strlen(string)));
                memcpy(parent->mutable_data(),string,parent->size());
                data(parent->mutable_data()); size(parent->size());
        }

        /// create shared buffer from buffer
        /// ----> ALTERED BEHAVIOUR:
        /// ----> This constructor transfers ownership rater than copying the data.
        inline shared_buffer_t(boctet_t* buffer, bsize_t size) :
                buffer_t(size, buffer),
                parent(new deleteable_buffer(buffer, size))
        {
        }

//    /// XXX debug... copy!
//      /// create shared buffer from buffer
//      inline shared_buffer_t(const char* buffer, bsize_t size) :
//              buffer_t(), parent(new deleteable_buffer(size)) {
//              memcpy(parent->mutable_data(), buffer, parent->size());
//              data(parent->mutable_data()); this->size(parent->size());
//      }

        /// clone data from a normal buffer
        inline shared_buffer_t(const buffer_t& rhs) :
                buffer_t(), parent(new deleteable_buffer(rhs.size())) {
                buffer_t::operator=(*parent);
                rhs.copy_to(*parent, 0);
        }

        /// copy constructor
        inline shared_buffer_t(const self& rhs) :
                buffer_t(rhs), parent(rhs.parent) {
        }

        /// copy a buffer pointer and size.
        inline self& operator=(const self& rhs) {
                parent = rhs.parent;
                buffer_t::operator=(rhs);
                return (*this);
        }

        /// return sub-buffer.
        inline self operator()(bsize_t index, bsize_t size = 0) const
        {
            // special cases
            if ( index + size > size_ )
            {
                // empty sub-buffer
            if ( index == size_ )
            {
                self n;
                return n;
            }
         
            // ERROR: index out of bounds
            throw illegal_sub_buffer("Index or size out of bounds in shared_buffer"); 
            }

            // regular case
                self n(*this);
                n.data_ += index;
                if (size == 0) n.size_ -= index;
                else n.size_ = size;
                return (n);
        }

        /// returns the number of references
        inline size_t use_count() const {
                return (parent.use_count());
        }

        /// returns true, if this is root buffer
        inline bool is_root() const {
                return ((parent.get() != NULL) && (this->data() == parent->data()) && (
                                this->size() == parent->size()));
        }

        /// returns true, if the buffer is not shared
        inline bool is_unique() const {
                return parent.unique();
        }

        /// makes the buffer unique / not shared by copying the contents
        inline void make_unique() {
                if (is_unique()) return;
                boost::shared_ptr<deleteable_buffer> old = parent;
                parent.reset(new deleteable_buffer(old->size()));
                buffer_t::data(parent->mutable_data());
                old->copy_to(*parent,0);
        }

        /// resize the shared buffer (and make unique)
        inline void resize( bsize_t new_size ) {
                boost::shared_ptr<deleteable_buffer> old = parent;
                parent.reset(new deleteable_buffer(new_size));
                buffer_t::data(parent->mutable_data());
                (*old)(0,new_size).copy_to(*parent,0);
                size_ = new_size;
        }

        /// returns a pointer to mutable data, if shared_buffer is not shared yet.
        inline boctet_t * mutable_data() {
                assert(parent.unique());
                return buffer_t::mutable_data();
        }

        /// drop reference to buffer
        inline void reset() {
                parent.reset();
                data(NULL);
                size(0);
        }

        /// concatenate buffers
        self operator+( const buffer_t& rhs ) const {
                self nb(size()+rhs.size());
                this->copy_to(nb,0);
                rhs.copy_to(nb,size());
                return nb;
        }
};

} /* namespace reboost */

#endif /* REBOOST_SHARED_BUFFER_HPP_ */