2007-09-29 18:14:03 +02:00
|
|
|
/*
|
|
|
|
|
2015-06-03 07:18:48 +02:00
|
|
|
Copyright (c) 2007-2015, Arvid Norberg
|
2007-09-29 18:14:03 +02:00
|
|
|
All rights reserved.
|
|
|
|
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
|
|
modification, are permitted provided that the following conditions
|
|
|
|
are met:
|
|
|
|
|
|
|
|
* Redistributions of source code must retain the above copyright
|
|
|
|
notice, this list of conditions and the following disclaimer.
|
|
|
|
* Redistributions in binary form must reproduce the above copyright
|
|
|
|
notice, this list of conditions and the following disclaimer in
|
|
|
|
the documentation and/or other materials provided with the distribution.
|
|
|
|
* Neither the name of the author nor the names of its
|
|
|
|
contributors may be used to endorse or promote products derived
|
|
|
|
from this software without specific prior written permission.
|
|
|
|
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
|
|
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
|
|
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
|
|
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
|
|
|
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
|
|
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
|
|
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
|
|
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
|
|
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
|
|
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
|
|
POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
#ifndef TORRENT_CHAINED_BUFFER_HPP_INCLUDED
|
|
|
|
#define TORRENT_CHAINED_BUFFER_HPP_INCLUDED
|
|
|
|
|
2011-11-02 10:31:32 +01:00
|
|
|
#include "libtorrent/config.hpp"
|
2015-04-24 07:37:17 +02:00
|
|
|
#include "libtorrent/disk_io_job.hpp" // for block_cache_reference
|
|
|
|
#include "libtorrent/debug.hpp"
|
|
|
|
|
|
|
|
#include "libtorrent/aux_/disable_warnings_push.hpp"
|
2011-11-02 10:31:32 +01:00
|
|
|
|
2008-05-28 09:52:25 +02:00
|
|
|
#include <boost/version.hpp>
|
2008-05-20 08:03:46 +02:00
|
|
|
#if BOOST_VERSION < 103500
|
|
|
|
#include <asio/buffer.hpp>
|
|
|
|
#else
|
2008-05-03 18:05:42 +02:00
|
|
|
#include <boost/asio/buffer.hpp>
|
2008-05-20 08:03:46 +02:00
|
|
|
#endif
|
2014-07-06 21:18:00 +02:00
|
|
|
#include <deque>
|
|
|
|
#include <vector>
|
2011-01-16 21:46:17 +01:00
|
|
|
#include <string.h> // for memcpy
|
2007-09-29 18:14:03 +02:00
|
|
|
|
2015-04-24 07:37:17 +02:00
|
|
|
#include "libtorrent/aux_/disable_warnings_pop.hpp"
|
2014-07-06 21:18:00 +02:00
|
|
|
|
2007-09-29 18:14:03 +02:00
|
|
|
namespace libtorrent
|
|
|
|
{
|
2008-05-20 08:03:46 +02:00
|
|
|
#if BOOST_VERSION >= 103500
|
2008-05-03 18:05:42 +02:00
|
|
|
namespace asio = boost::asio;
|
2008-05-20 08:03:46 +02:00
|
|
|
#endif
|
2014-07-06 21:18:00 +02:00
|
|
|
struct TORRENT_EXTRA_EXPORT chained_buffer : private single_threaded
|
2007-09-29 18:14:03 +02:00
|
|
|
{
|
2011-08-14 01:01:38 +02:00
|
|
|
chained_buffer(): m_bytes(0), m_capacity(0)
|
|
|
|
{
|
2014-07-06 21:18:00 +02:00
|
|
|
thread_started();
|
2014-01-19 20:45:50 +01:00
|
|
|
#if TORRENT_USE_ASSERTS
|
2011-08-14 01:01:38 +02:00
|
|
|
m_destructed = false;
|
|
|
|
#endif
|
|
|
|
}
|
2007-09-29 18:14:03 +02:00
|
|
|
|
2014-07-06 21:18:00 +02:00
|
|
|
// destructs/frees the buffer (1st arg) with
|
|
|
|
// 2nd argument as userdata
|
|
|
|
typedef void (*free_buffer_fun)(char*, void*, block_cache_reference ref);
|
|
|
|
|
2007-09-29 18:14:03 +02:00
|
|
|
struct buffer_t
|
|
|
|
{
|
2014-07-06 21:18:00 +02:00
|
|
|
free_buffer_fun free_fun;
|
|
|
|
void* userdata;
|
2007-09-29 18:14:03 +02:00
|
|
|
char* buf; // the first byte of the buffer
|
|
|
|
char* start; // the first byte to send/receive in the buffer
|
2013-10-05 06:49:00 +02:00
|
|
|
int size; // the total size of the buffer
|
2007-09-29 18:14:03 +02:00
|
|
|
int used_size; // this is the number of bytes to send/receive
|
2014-07-06 21:18:00 +02:00
|
|
|
block_cache_reference ref;
|
2007-09-29 18:14:03 +02:00
|
|
|
};
|
|
|
|
|
|
|
|
bool empty() const { return m_bytes == 0; }
|
|
|
|
int size() const { return m_bytes; }
|
|
|
|
int capacity() const { return m_capacity; }
|
|
|
|
|
2011-08-14 01:01:38 +02:00
|
|
|
void pop_front(int bytes_to_pop);
|
2007-09-29 18:14:03 +02:00
|
|
|
|
2011-08-14 01:01:38 +02:00
|
|
|
void append_buffer(char* buffer, int s, int used_size
|
2014-07-06 21:18:00 +02:00
|
|
|
, free_buffer_fun destructor, void* userdata
|
|
|
|
, block_cache_reference ref = block_cache_reference());
|
2007-09-29 18:14:03 +02:00
|
|
|
|
2014-11-23 07:14:47 +01:00
|
|
|
void prepend_buffer(char* buffer, int s, int used_size
|
|
|
|
, free_buffer_fun destructor, void* userdata
|
|
|
|
, block_cache_reference ref = block_cache_reference());
|
|
|
|
|
2007-09-29 18:14:03 +02:00
|
|
|
// returns the number of bytes available at the
|
|
|
|
// end of the last chained buffer.
|
2011-08-14 01:01:38 +02:00
|
|
|
int space_in_last_buffer();
|
2007-09-29 18:14:03 +02:00
|
|
|
|
|
|
|
// tries to copy the given buffer to the end of the
|
|
|
|
// last chained buffer. If there's not enough room
|
|
|
|
// it returns false
|
2011-08-14 01:01:38 +02:00
|
|
|
char* append(char const* buf, int s);
|
2007-09-29 18:14:03 +02:00
|
|
|
|
|
|
|
// tries to allocate memory from the end
|
|
|
|
// of the last buffer. If there isn't
|
|
|
|
// enough room, returns 0
|
2011-08-14 01:01:38 +02:00
|
|
|
char* allocate_appendix(int s);
|
2007-09-29 18:14:03 +02:00
|
|
|
|
2014-07-06 21:18:00 +02:00
|
|
|
std::vector<asio::const_buffer> const& build_iovec(int to_send);
|
|
|
|
|
|
|
|
void clear();
|
2007-09-29 18:14:03 +02:00
|
|
|
|
2014-11-23 07:14:47 +01:00
|
|
|
void build_mutable_iovec(int bytes, std::vector<asio::mutable_buffer>& vec);
|
|
|
|
|
2011-08-14 01:01:38 +02:00
|
|
|
~chained_buffer();
|
2007-09-29 18:14:03 +02:00
|
|
|
|
|
|
|
private:
|
2014-11-23 07:14:47 +01:00
|
|
|
template <typename Buffer>
|
|
|
|
void build_vec(int bytes, std::vector<Buffer>& vec);
|
2007-09-29 18:14:03 +02:00
|
|
|
|
|
|
|
// this is the list of all the buffers we want to
|
|
|
|
// send
|
2014-07-06 21:18:00 +02:00
|
|
|
std::deque<buffer_t> m_vec;
|
2013-11-26 05:39:33 +01:00
|
|
|
|
2007-09-29 18:14:03 +02:00
|
|
|
// this is the number of bytes in the send buf.
|
|
|
|
// this will always be equal to the sum of the
|
|
|
|
// size of all buffers in vec
|
|
|
|
int m_bytes;
|
|
|
|
|
|
|
|
// the total size of all buffers in the chain
|
|
|
|
// including unused space
|
|
|
|
int m_capacity;
|
|
|
|
|
2014-07-06 21:18:00 +02:00
|
|
|
// this is the vector of buffers used when
|
|
|
|
// invoking the async write call
|
|
|
|
std::vector<asio::const_buffer> m_tmp_vec;
|
|
|
|
|
2014-01-19 20:45:50 +01:00
|
|
|
#if TORRENT_USE_ASSERTS
|
2011-08-14 01:01:38 +02:00
|
|
|
bool m_destructed;
|
|
|
|
#endif
|
2007-09-29 18:14:03 +02:00
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|