boost/asio
2
0
Fork 0
mirror of https://github.com/boostorg/asio.git synced 2026-01-27 18:42:07 +00:00
Code Issues Projects Releases Wiki Activity
Files
bc07d2d3c1873b466fbe218a2c8964fa08046fd8
asio/test/read_until.cpp
Christopher Kohlhoff e97ac69ad8 Merge from trunk. 
........
  r57393 | hkaiser | 2009-11-05 11:26:15 +1100 (Thu, 05 Nov 2009) | 1 line
  
  Asio: disabled VC workaround for VC2010 beta2 compiler. Fixes #3796.
........
  r58621 | chris_kohlhoff | 2010-01-02 10:04:35 +1100 (Sat, 02 Jan 2010) | 2 lines
  
  Wrap long line.
........
  r58624 | chris_kohlhoff | 2010-01-02 17:09:02 +1100 (Sat, 02 Jan 2010) | 3 lines
  
  Windows needs the OVERLAPPED structure to be valid until both the initiating
  function call has returned and the completion packet has been delivered.
........
  r58625 | chris_kohlhoff | 2010-01-02 18:16:41 +1100 (Sat, 02 Jan 2010) | 2 lines
  
  Use specific type_traits headers.
........
  r58626 | chris_kohlhoff | 2010-01-02 18:18:09 +1100 (Sat, 02 Jan 2010) | 2 lines
  
  Include specific headers in unit tests rather than the convenience header asio.hpp.
........
  r58627 | chris_kohlhoff | 2010-01-02 19:24:12 +1100 (Sat, 02 Jan 2010) | 3 lines
  
  Use boost::addressof to get the address of handler objects, rather than
  applying operator& directly. Fixes #2977.
........
  r58628 | chris_kohlhoff | 2010-01-02 20:48:01 +1100 (Sat, 02 Jan 2010) | 3 lines
  
  Don't block signals while performing system calls, but instead restart the
  calls if they are interrupted.
........
  r58629 | chris_kohlhoff | 2010-01-02 21:20:12 +1100 (Sat, 02 Jan 2010) | 2 lines
  
  Ensure that kqueue support is enabled for BSD platforms. Fixes #3626.
........
  r58630 | chris_kohlhoff | 2010-01-02 21:30:41 +1100 (Sat, 02 Jan 2010) | 2 lines
  
  Add boost_ prefix to extern "C" thread entry point function. Fixes #3809.
........
  r58647 | chris_kohlhoff | 2010-01-03 07:36:59 +1100 (Sun, 03 Jan 2010) | 2 lines
  
  Use a pool of strand implementations to make copying of strands cheaper.
........
  r58650 | chris_kohlhoff | 2010-01-03 08:35:33 +1100 (Sun, 03 Jan 2010) | 4 lines
  
  In getaddrinfo emulation, only check the socket type (SOCK_STREAM or SOCK_DGRAM)
  if a service name has been specified. This should allow the emulation to work
  with raw sockets.
........
  r58651 | chris_kohlhoff | 2010-01-03 08:37:10 +1100 (Sun, 03 Jan 2010) | 3 lines
  
  Add a workaround for some broken Windows firewalls that make a socket
  appear bound to 0.0.0.0 when it is in fact bound to 127.0.0.1.
........
  r58652 | chris_kohlhoff | 2010-01-03 08:38:44 +1100 (Sun, 03 Jan 2010) | 2 lines
  
  Only include implementation headers required for each platform.
........


[SVN r58669]
2010-01-04 11:55:09 +00:00

851 lines
24 KiB
C++
Raw Blame History

//
// read_until.cpp
// ~~~~~~~~~~~~~~
//
// Copyright (c) 2003-2010 Christopher M. Kohlhoff (chris at kohlhoff dot com)
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
// Disable autolinking for unit tests.
#if !defined(BOOST_ALL_NO_LIB)
#define BOOST_ALL_NO_LIB 1
#endif // !defined(BOOST_ALL_NO_LIB)
// Test that header file is self-contained.
#include <boost/asio/read_until.hpp>
#include <boost/bind.hpp>
#include <boost/noncopyable.hpp>
#include <cstring>
#include <boost/asio/io_service.hpp>
#include <boost/asio/placeholders.hpp>
#include <boost/asio/streambuf.hpp>
#include "unit_test.hpp"
class test_stream
: private boost::noncopyable
{
public:
typedef boost::asio::io_service io_service_type;
test_stream(boost::asio::io_service& io_service)
: io_service_(io_service),
length_(0),
position_(0),
next_read_length_(0)
{
}
io_service_type& get_io_service()
{
return io_service_;
}
void reset(const void* data, size_t length)
{
using namespace std; // For memcpy.
BOOST_CHECK(length <= max_length);
memcpy(data_, data, length);
length_ = length;
position_ = 0;
next_read_length_ = length;
}
void next_read_length(size_t length)
{
next_read_length_ = length;
}
template <typename Mutable_Buffers>
size_t read_some(const Mutable_Buffers& buffers)
{
using namespace std; // For memcpy.
size_t total_length = 0;
typename Mutable_Buffers::const_iterator iter = buffers.begin();
typename Mutable_Buffers::const_iterator end = buffers.end();
for (; iter != end && total_length < next_read_length_; ++iter)
{
size_t length = boost::asio::buffer_size(*iter);
if (length > length_ - position_)
length = length_ - position_;
if (length > next_read_length_ - total_length)
length = next_read_length_ - total_length;
memcpy(boost::asio::buffer_cast<void*>(*iter), data_ + position_, length);
position_ += length;
total_length += length;
}
return total_length;
}
template <typename Mutable_Buffers>
size_t read_some(const Mutable_Buffers& buffers,
boost::system::error_code& ec)
{
ec = boost::system::error_code();
return read_some(buffers);
}
template <typename Mutable_Buffers, typename Handler>
void async_read_some(const Mutable_Buffers& buffers, Handler handler)
{
size_t bytes_transferred = read_some(buffers);
io_service_.post(boost::asio::detail::bind_handler(
handler, boost::system::error_code(), bytes_transferred));
}
private:
io_service_type& io_service_;
enum { max_length = 8192 };
char data_[max_length];
size_t length_;
size_t position_;
size_t next_read_length_;
};
static const char read_data[]
= "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
void test_char_read_until()
{
boost::asio::io_service ios;
test_stream s(ios);
boost::asio::streambuf sb1;
boost::asio::streambuf sb2(25);
boost::system::error_code ec;
s.reset(read_data, sizeof(read_data));
sb1.consume(sb1.size());
std::size_t length = boost::asio::read_until(s, sb1, 'Z');
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
sb1.consume(sb1.size());
length = boost::asio::read_until(s, sb1, 'Z');
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
sb1.consume(sb1.size());
length = boost::asio::read_until(s, sb1, 'Z');
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
sb1.consume(sb1.size());
length = boost::asio::read_until(s, sb1, 'Z', ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
sb1.consume(sb1.size());
length = boost::asio::read_until(s, sb1, 'Z', ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
sb1.consume(sb1.size());
length = boost::asio::read_until(s, sb1, 'Z', ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, 'Z', ec);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, 'Z', ec);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, 'Z', ec);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, 'Y', ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, 'Y', ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, 'Y', ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
}
void test_string_read_until()
{
boost::asio::io_service ios;
test_stream s(ios);
boost::asio::streambuf sb1;
boost::asio::streambuf sb2(25);
boost::system::error_code ec;
s.reset(read_data, sizeof(read_data));
sb1.consume(sb1.size());
std::size_t length = boost::asio::read_until(s, sb1, "XYZ");
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
sb1.consume(sb1.size());
length = boost::asio::read_until(s, sb1, "XYZ");
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
sb1.consume(sb1.size());
length = boost::asio::read_until(s, sb1, "XYZ");
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
sb1.consume(sb1.size());
length = boost::asio::read_until(s, sb1, "XYZ", ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
sb1.consume(sb1.size());
length = boost::asio::read_until(s, sb1, "XYZ", ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
sb1.consume(sb1.size());
length = boost::asio::read_until(s, sb1, "XYZ", ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, "XYZ", ec);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, "XYZ", ec);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, "XYZ", ec);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, "WXY", ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, "WXY", ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, "WXY", ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
}
class match_char
{
public:
explicit match_char(char c) : c_(c) {}
template <typename Iterator>
std::pair<Iterator, bool> operator()(
Iterator begin, Iterator end) const
{
Iterator i = begin;
while (i != end)
if (c_ == *i++)
return std::make_pair(i, true);
return std::make_pair(i, false);
}
private:
char c_;
};
namespace boost {
namespace asio {
template <> struct is_match_condition<match_char>
: public boost::true_type {};
} // namespace asio
} // namespace boost
void test_match_condition_read_until()
{
boost::asio::io_service ios;
test_stream s(ios);
boost::asio::streambuf sb1;
boost::asio::streambuf sb2(25);
boost::system::error_code ec;
s.reset(read_data, sizeof(read_data));
sb1.consume(sb1.size());
std::size_t length = boost::asio::read_until(s, sb1, match_char('Z'));
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
sb1.consume(sb1.size());
length = boost::asio::read_until(s, sb1, match_char('Z'));
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
sb1.consume(sb1.size());
length = boost::asio::read_until(s, sb1, match_char('Z'));
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
sb1.consume(sb1.size());
length = boost::asio::read_until(s, sb1, match_char('Z'), ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
sb1.consume(sb1.size());
length = boost::asio::read_until(s, sb1, match_char('Z'), ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
sb1.consume(sb1.size());
length = boost::asio::read_until(s, sb1, match_char('Z'), ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, match_char('Z'), ec);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, match_char('Z'), ec);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, match_char('Z'), ec);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, match_char('Y'), ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, match_char('Y'), ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
sb2.consume(sb2.size());
length = boost::asio::read_until(s, sb2, match_char('Y'), ec);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
}
void async_read_handler(
const boost::system::error_code& err, boost::system::error_code* err_out,
std::size_t bytes_transferred, std::size_t* bytes_out, bool* called)
{
*err_out = err;
*bytes_out = bytes_transferred;
*called = true;
}
void test_char_async_read_until()
{
boost::asio::io_service ios;
test_stream s(ios);
boost::asio::streambuf sb1;
boost::asio::streambuf sb2(25);
boost::system::error_code ec;
std::size_t length;
bool called;
s.reset(read_data, sizeof(read_data));
ec = boost::system::error_code();
length = 0;
called = false;
sb1.consume(sb1.size());
boost::asio::async_read_until(s, sb1, 'Z',
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
ec = boost::system::error_code();
length = 0;
called = false;
sb1.consume(sb1.size());
boost::asio::async_read_until(s, sb1, 'Z',
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
ec = boost::system::error_code();
length = 0;
called = false;
sb1.consume(sb1.size());
boost::asio::async_read_until(s, sb1, 'Z',
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, 'Z',
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, 'Z',
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, 'Z',
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, 'Y',
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, 'Y',
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, 'Y',
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
}
void test_string_async_read_until()
{
boost::asio::io_service ios;
test_stream s(ios);
boost::asio::streambuf sb1;
boost::asio::streambuf sb2(25);
boost::system::error_code ec;
std::size_t length;
bool called;
s.reset(read_data, sizeof(read_data));
ec = boost::system::error_code();
length = 0;
called = false;
sb1.consume(sb1.size());
boost::asio::async_read_until(s, sb1, "XYZ",
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
ec = boost::system::error_code();
length = 0;
called = false;
sb1.consume(sb1.size());
boost::asio::async_read_until(s, sb1, "XYZ",
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
ec = boost::system::error_code();
length = 0;
called = false;
sb1.consume(sb1.size());
boost::asio::async_read_until(s, sb1, "XYZ",
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, "XYZ",
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, "XYZ",
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, "XYZ",
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, "WXY",
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, "WXY",
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, "WXY",
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
}
void test_match_condition_async_read_until()
{
boost::asio::io_service ios;
test_stream s(ios);
boost::asio::streambuf sb1;
boost::asio::streambuf sb2(25);
boost::system::error_code ec;
std::size_t length;
bool called;
s.reset(read_data, sizeof(read_data));
ec = boost::system::error_code();
length = 0;
called = false;
sb1.consume(sb1.size());
boost::asio::async_read_until(s, sb1, match_char('Z'),
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
ec = boost::system::error_code();
length = 0;
called = false;
sb1.consume(sb1.size());
boost::asio::async_read_until(s, sb1, match_char('Z'),
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
ec = boost::system::error_code();
length = 0;
called = false;
sb1.consume(sb1.size());
boost::asio::async_read_until(s, sb1, match_char('Z'),
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 26);
s.reset(read_data, sizeof(read_data));
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, match_char('Z'),
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, match_char('Z'),
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, match_char('Z'),
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(ec == boost::asio::error::not_found);
BOOST_CHECK(length == 0);
s.reset(read_data, sizeof(read_data));
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, match_char('Y'),
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
s.reset(read_data, sizeof(read_data));
s.next_read_length(1);
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, match_char('Y'),
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
s.reset(read_data, sizeof(read_data));
s.next_read_length(10);
ec = boost::system::error_code();
length = 0;
called = false;
sb2.consume(sb2.size());
boost::asio::async_read_until(s, sb2, match_char('Y'),
boost::bind(async_read_handler, boost::asio::placeholders::error, &ec,
boost::asio::placeholders::bytes_transferred, &length, &called));
ios.reset();
ios.run();
BOOST_CHECK(called);
BOOST_CHECK(!ec);
BOOST_CHECK(length == 25);
}
test_suite* init_unit_test_suite(int, char*[])
{
test_suite* test = BOOST_TEST_SUITE("read_until");
test->add(BOOST_TEST_CASE(&test_char_read_until));
test->add(BOOST_TEST_CASE(&test_string_read_until));
test->add(BOOST_TEST_CASE(&test_match_condition_read_until));
test->add(BOOST_TEST_CASE(&test_char_async_read_until));
test->add(BOOST_TEST_CASE(&test_string_async_read_until));
test->add(BOOST_TEST_CASE(&test_match_condition_async_read_until));
return test;
}
Reference in New Issue View Git Blame Copy Permalink