| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898 |
- //
- // Copyright (c) 2016-2017 Vinnie Falco (vinnie dot falco at gmail 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)
- //
- // Official repository: https://github.com/boostorg/beast
- //
- #ifndef BOOST_BEAST_WEBSOCKET_IMPL_WRITE_IPP
- #define BOOST_BEAST_WEBSOCKET_IMPL_WRITE_IPP
- #include <boost/beast/core/bind_handler.hpp>
- #include <boost/beast/core/buffers_cat.hpp>
- #include <boost/beast/core/buffers_prefix.hpp>
- #include <boost/beast/core/buffers_suffix.hpp>
- #include <boost/beast/core/flat_static_buffer.hpp>
- #include <boost/beast/core/type_traits.hpp>
- #include <boost/beast/core/detail/clamp.hpp>
- #include <boost/beast/core/detail/config.hpp>
- #include <boost/beast/websocket/detail/frame.hpp>
- #include <boost/asio/associated_allocator.hpp>
- #include <boost/asio/associated_executor.hpp>
- #include <boost/asio/coroutine.hpp>
- #include <boost/asio/executor_work_guard.hpp>
- #include <boost/asio/handler_continuation_hook.hpp>
- #include <boost/asio/handler_invoke_hook.hpp>
- #include <boost/assert.hpp>
- #include <boost/config.hpp>
- #include <boost/throw_exception.hpp>
- #include <algorithm>
- #include <memory>
- namespace boost {
- namespace beast {
- namespace websocket {
- namespace detail {
- // Compress a buffer sequence
- // Returns: `true` if more calls are needed
- //
- template<>
- template<class ConstBufferSequence>
- bool
- stream_base<true>::
- deflate(
- boost::asio::mutable_buffer& out,
- buffers_suffix<ConstBufferSequence>& cb,
- bool fin,
- std::size_t& total_in,
- error_code& ec)
- {
- using boost::asio::buffer;
- BOOST_ASSERT(out.size() >= 6);
- auto& zo = this->pmd_->zo;
- zlib::z_params zs;
- zs.avail_in = 0;
- zs.next_in = nullptr;
- zs.avail_out = out.size();
- zs.next_out = out.data();
- for(auto in : beast::detail::buffers_range(cb))
- {
- zs.avail_in = in.size();
- if(zs.avail_in == 0)
- continue;
- zs.next_in = in.data();
- zo.write(zs, zlib::Flush::none, ec);
- if(ec)
- {
- if(ec != zlib::error::need_buffers)
- return false;
- BOOST_ASSERT(zs.avail_out == 0);
- BOOST_ASSERT(zs.total_out == out.size());
- ec.assign(0, ec.category());
- break;
- }
- if(zs.avail_out == 0)
- {
- BOOST_ASSERT(zs.total_out == out.size());
- break;
- }
- BOOST_ASSERT(zs.avail_in == 0);
- }
- total_in = zs.total_in;
- cb.consume(zs.total_in);
- if(zs.avail_out > 0 && fin)
- {
- auto const remain = boost::asio::buffer_size(cb);
- if(remain == 0)
- {
- // Inspired by Mark Adler
- // https://github.com/madler/zlib/issues/149
- //
- // VFALCO We could do this flush twice depending
- // on how much space is in the output.
- zo.write(zs, zlib::Flush::block, ec);
- BOOST_ASSERT(! ec || ec == zlib::error::need_buffers);
- if(ec == zlib::error::need_buffers)
- ec.assign(0, ec.category());
- if(ec)
- return false;
- if(zs.avail_out >= 6)
- {
- zo.write(zs, zlib::Flush::full, ec);
- BOOST_ASSERT(! ec);
- // remove flush marker
- zs.total_out -= 4;
- out = buffer(out.data(), zs.total_out);
- return false;
- }
- }
- }
- ec.assign(0, ec.category());
- out = buffer(out.data(), zs.total_out);
- return true;
- }
- template<>
- inline
- void
- stream_base<true>::
- do_context_takeover_write(role_type role)
- {
- if((role == role_type::client &&
- this->pmd_config_.client_no_context_takeover) ||
- (role == role_type::server &&
- this->pmd_config_.server_no_context_takeover))
- {
- this->pmd_->zo.reset();
- }
- }
- } // detail
- //------------------------------------------------------------------------------
- template<class NextLayer, bool deflateSupported>
- template<class Buffers, class Handler>
- class stream<NextLayer, deflateSupported>::write_some_op
- : public boost::asio::coroutine
- {
- Handler h_;
- stream<NextLayer, deflateSupported>& ws_;
- boost::asio::executor_work_guard<decltype(std::declval<
- stream<NextLayer, deflateSupported>&>().get_executor())> wg_;
- buffers_suffix<Buffers> cb_;
- detail::frame_header fh_;
- detail::prepared_key key_;
- std::size_t bytes_transferred_ = 0;
- std::size_t remain_;
- std::size_t in_;
- int how_;
- bool fin_;
- bool more_;
- bool cont_ = false;
- public:
- static constexpr int id = 2; // for soft_mutex
- write_some_op(write_some_op&&) = default;
- write_some_op(write_some_op const&) = delete;
- template<class DeducedHandler>
- write_some_op(
- DeducedHandler&& h,
- stream<NextLayer, deflateSupported>& ws,
- bool fin,
- Buffers const& bs)
- : h_(std::forward<DeducedHandler>(h))
- , ws_(ws)
- , wg_(ws_.get_executor())
- , cb_(bs)
- , fin_(fin)
- {
- }
- using allocator_type =
- boost::asio::associated_allocator_t<Handler>;
- allocator_type
- get_allocator() const noexcept
- {
- return (boost::asio::get_associated_allocator)(h_);
- }
- using executor_type = boost::asio::associated_executor_t<
- Handler, decltype(std::declval<stream<NextLayer, deflateSupported>&>().get_executor())>;
- executor_type
- get_executor() const noexcept
- {
- return (boost::asio::get_associated_executor)(
- h_, ws_.get_executor());
- }
- Handler&
- handler()
- {
- return h_;
- }
- void operator()(
- error_code ec = {},
- std::size_t bytes_transferred = 0,
- bool cont = true);
- friend
- bool asio_handler_is_continuation(write_some_op* op)
- {
- using boost::asio::asio_handler_is_continuation;
- return op->cont_ || asio_handler_is_continuation(
- std::addressof(op->h_));
- }
- template<class Function>
- friend
- void asio_handler_invoke(Function&& f, write_some_op* op)
- {
- using boost::asio::asio_handler_invoke;
- asio_handler_invoke(
- f, std::addressof(op->h_));
- }
- };
- template<class NextLayer, bool deflateSupported>
- template<class Buffers, class Handler>
- void
- stream<NextLayer, deflateSupported>::
- write_some_op<Buffers, Handler>::
- operator()(
- error_code ec,
- std::size_t bytes_transferred,
- bool cont)
- {
- using beast::detail::clamp;
- using boost::asio::buffer;
- using boost::asio::buffer_copy;
- using boost::asio::buffer_size;
- using boost::asio::mutable_buffer;
- enum
- {
- do_nomask_nofrag,
- do_nomask_frag,
- do_mask_nofrag,
- do_mask_frag,
- do_deflate
- };
- std::size_t n;
- boost::asio::mutable_buffer b;
- cont_ = cont;
- BOOST_ASIO_CORO_REENTER(*this)
- {
- // Set up the outgoing frame header
- if(! ws_.wr_cont_)
- {
- ws_.begin_msg();
- fh_.rsv1 = ws_.wr_compress_;
- }
- else
- {
- fh_.rsv1 = false;
- }
- fh_.rsv2 = false;
- fh_.rsv3 = false;
- fh_.op = ws_.wr_cont_ ?
- detail::opcode::cont : ws_.wr_opcode_;
- fh_.mask =
- ws_.role_ == role_type::client;
- // Choose a write algorithm
- if(ws_.wr_compress_)
- {
- how_ = do_deflate;
- }
- else if(! fh_.mask)
- {
- if(! ws_.wr_frag_)
- {
- how_ = do_nomask_nofrag;
- }
- else
- {
- BOOST_ASSERT(ws_.wr_buf_size_ != 0);
- remain_ = buffer_size(cb_);
- if(remain_ > ws_.wr_buf_size_)
- how_ = do_nomask_frag;
- else
- how_ = do_nomask_nofrag;
- }
- }
- else
- {
- if(! ws_.wr_frag_)
- {
- how_ = do_mask_nofrag;
- }
- else
- {
- BOOST_ASSERT(ws_.wr_buf_size_ != 0);
- remain_ = buffer_size(cb_);
- if(remain_ > ws_.wr_buf_size_)
- how_ = do_mask_frag;
- else
- how_ = do_mask_nofrag;
- }
- }
- // Maybe suspend
- if(ws_.wr_block_.try_lock(this))
- {
- // Make sure the stream is open
- if(! ws_.check_open(ec))
- goto upcall;
- }
- else
- {
- do_suspend:
- // Suspend
- BOOST_ASIO_CORO_YIELD
- ws_.paused_wr_.emplace(std::move(*this));
- // Acquire the write block
- ws_.wr_block_.lock(this);
- // Resume
- BOOST_ASIO_CORO_YIELD
- boost::asio::post(
- ws_.get_executor(), std::move(*this));
- BOOST_ASSERT(ws_.wr_block_.is_locked(this));
- // Make sure the stream is open
- if(! ws_.check_open(ec))
- goto upcall;
- }
- //------------------------------------------------------------------
- if(how_ == do_nomask_nofrag)
- {
- fh_.fin = fin_;
- fh_.len = buffer_size(cb_);
- ws_.wr_fb_.reset();
- detail::write<flat_static_buffer_base>(
- ws_.wr_fb_, fh_);
- ws_.wr_cont_ = ! fin_;
- // Send frame
- BOOST_ASIO_CORO_YIELD
- boost::asio::async_write(ws_.stream_,
- buffers_cat(ws_.wr_fb_.data(), cb_),
- std::move(*this));
- if(! ws_.check_ok(ec))
- goto upcall;
- bytes_transferred_ += clamp(fh_.len);
- goto upcall;
- }
- //------------------------------------------------------------------
- else if(how_ == do_nomask_frag)
- {
- for(;;)
- {
- n = clamp(remain_, ws_.wr_buf_size_);
- fh_.len = n;
- remain_ -= n;
- fh_.fin = fin_ ? remain_ == 0 : false;
- ws_.wr_fb_.reset();
- detail::write<flat_static_buffer_base>(
- ws_.wr_fb_, fh_);
- ws_.wr_cont_ = ! fin_;
- // Send frame
- BOOST_ASIO_CORO_YIELD
- boost::asio::async_write(
- ws_.stream_, buffers_cat(
- ws_.wr_fb_.data(), buffers_prefix(
- clamp(fh_.len), cb_)),
- std::move(*this));
- if(! ws_.check_ok(ec))
- goto upcall;
- n = clamp(fh_.len); // because yield
- bytes_transferred_ += n;
- if(remain_ == 0)
- break;
- cb_.consume(n);
- fh_.op = detail::opcode::cont;
- // Allow outgoing control frames to
- // be sent in between message frames
- ws_.wr_block_.unlock(this);
- if( ws_.paused_close_.maybe_invoke() ||
- ws_.paused_rd_.maybe_invoke() ||
- ws_.paused_ping_.maybe_invoke())
- {
- BOOST_ASSERT(ws_.wr_block_.is_locked());
- goto do_suspend;
- }
- ws_.wr_block_.lock(this);
- }
- goto upcall;
- }
- //------------------------------------------------------------------
- else if(how_ == do_mask_nofrag)
- {
- remain_ = buffer_size(cb_);
- fh_.fin = fin_;
- fh_.len = remain_;
- fh_.key = ws_.create_mask();
- detail::prepare_key(key_, fh_.key);
- ws_.wr_fb_.reset();
- detail::write<flat_static_buffer_base>(
- ws_.wr_fb_, fh_);
- n = clamp(remain_, ws_.wr_buf_size_);
- buffer_copy(buffer(
- ws_.wr_buf_.get(), n), cb_);
- detail::mask_inplace(buffer(
- ws_.wr_buf_.get(), n), key_);
- remain_ -= n;
- ws_.wr_cont_ = ! fin_;
- // Send frame header and partial payload
- BOOST_ASIO_CORO_YIELD
- boost::asio::async_write(
- ws_.stream_, buffers_cat(ws_.wr_fb_.data(),
- buffer(ws_.wr_buf_.get(), n)),
- std::move(*this));
- if(! ws_.check_ok(ec))
- goto upcall;
- bytes_transferred_ +=
- bytes_transferred - ws_.wr_fb_.size();
- while(remain_ > 0)
- {
- cb_.consume(ws_.wr_buf_size_);
- n = clamp(remain_, ws_.wr_buf_size_);
- buffer_copy(buffer(
- ws_.wr_buf_.get(), n), cb_);
- detail::mask_inplace(buffer(
- ws_.wr_buf_.get(), n), key_);
- remain_ -= n;
- // Send partial payload
- BOOST_ASIO_CORO_YIELD
- boost::asio::async_write(ws_.stream_,
- buffer(ws_.wr_buf_.get(), n),
- std::move(*this));
- if(! ws_.check_ok(ec))
- goto upcall;
- bytes_transferred_ += bytes_transferred;
- }
- goto upcall;
- }
- //------------------------------------------------------------------
- else if(how_ == do_mask_frag)
- {
- for(;;)
- {
- n = clamp(remain_, ws_.wr_buf_size_);
- remain_ -= n;
- fh_.len = n;
- fh_.key = ws_.create_mask();
- fh_.fin = fin_ ? remain_ == 0 : false;
- detail::prepare_key(key_, fh_.key);
- buffer_copy(buffer(
- ws_.wr_buf_.get(), n), cb_);
- detail::mask_inplace(buffer(
- ws_.wr_buf_.get(), n), key_);
- ws_.wr_fb_.reset();
- detail::write<flat_static_buffer_base>(
- ws_.wr_fb_, fh_);
- ws_.wr_cont_ = ! fin_;
- // Send frame
- BOOST_ASIO_CORO_YIELD
- boost::asio::async_write(ws_.stream_,
- buffers_cat(ws_.wr_fb_.data(),
- buffer(ws_.wr_buf_.get(), n)),
- std::move(*this));
- if(! ws_.check_ok(ec))
- goto upcall;
- n = bytes_transferred - ws_.wr_fb_.size();
- bytes_transferred_ += n;
- if(remain_ == 0)
- break;
- cb_.consume(n);
- fh_.op = detail::opcode::cont;
- // Allow outgoing control frames to
- // be sent in between message frames:
- ws_.wr_block_.unlock(this);
- if( ws_.paused_close_.maybe_invoke() ||
- ws_.paused_rd_.maybe_invoke() ||
- ws_.paused_ping_.maybe_invoke())
- {
- BOOST_ASSERT(ws_.wr_block_.is_locked());
- goto do_suspend;
- }
- ws_.wr_block_.lock(this);
- }
- goto upcall;
- }
- //------------------------------------------------------------------
- else if(how_ == do_deflate)
- {
- for(;;)
- {
- b = buffer(ws_.wr_buf_.get(),
- ws_.wr_buf_size_);
- more_ = ws_.deflate(b, cb_, fin_, in_, ec);
- if(! ws_.check_ok(ec))
- goto upcall;
- n = buffer_size(b);
- if(n == 0)
- {
- // The input was consumed, but there
- // is no output due to compression
- // latency.
- BOOST_ASSERT(! fin_);
- BOOST_ASSERT(buffer_size(cb_) == 0);
- goto upcall;
- }
- if(fh_.mask)
- {
- fh_.key = ws_.create_mask();
- detail::prepared_key key;
- detail::prepare_key(key, fh_.key);
- detail::mask_inplace(b, key);
- }
- fh_.fin = ! more_;
- fh_.len = n;
- ws_.wr_fb_.reset();
- detail::write<
- flat_static_buffer_base>(ws_.wr_fb_, fh_);
- ws_.wr_cont_ = ! fin_;
- // Send frame
- BOOST_ASIO_CORO_YIELD
- boost::asio::async_write(ws_.stream_,
- buffers_cat(ws_.wr_fb_.data(), b),
- std::move(*this));
- if(! ws_.check_ok(ec))
- goto upcall;
- bytes_transferred_ += in_;
- if(more_)
- {
- fh_.op = detail::opcode::cont;
- fh_.rsv1 = false;
- // Allow outgoing control frames to
- // be sent in between message frames:
- ws_.wr_block_.unlock(this);
- if( ws_.paused_close_.maybe_invoke() ||
- ws_.paused_rd_.maybe_invoke() ||
- ws_.paused_ping_.maybe_invoke())
- {
- BOOST_ASSERT(ws_.wr_block_.is_locked());
- goto do_suspend;
- }
- ws_.wr_block_.lock(this);
- }
- else
- {
- if(fh_.fin)
- ws_.do_context_takeover_write(ws_.role_);
- goto upcall;
- }
- }
- }
- //--------------------------------------------------------------------------
- upcall:
- ws_.wr_block_.unlock(this);
- ws_.paused_close_.maybe_invoke() ||
- ws_.paused_rd_.maybe_invoke() ||
- ws_.paused_ping_.maybe_invoke();
- if(! cont_)
- {
- BOOST_ASIO_CORO_YIELD
- boost::asio::post(
- ws_.get_executor(),
- bind_handler(std::move(*this), ec, bytes_transferred_));
- }
- h_(ec, bytes_transferred_);
- }
- }
- //------------------------------------------------------------------------------
- template<class NextLayer, bool deflateSupported>
- template<class ConstBufferSequence>
- std::size_t
- stream<NextLayer, deflateSupported>::
- write_some(bool fin, ConstBufferSequence const& buffers)
- {
- static_assert(is_sync_stream<next_layer_type>::value,
- "SyncStream requirements not met");
- static_assert(boost::asio::is_const_buffer_sequence<
- ConstBufferSequence>::value,
- "ConstBufferSequence requirements not met");
- error_code ec;
- auto const bytes_transferred =
- write_some(fin, buffers, ec);
- if(ec)
- BOOST_THROW_EXCEPTION(system_error{ec});
- return bytes_transferred;
- }
- template<class NextLayer, bool deflateSupported>
- template<class ConstBufferSequence>
- std::size_t
- stream<NextLayer, deflateSupported>::
- write_some(bool fin,
- ConstBufferSequence const& buffers, error_code& ec)
- {
- static_assert(is_sync_stream<next_layer_type>::value,
- "SyncStream requirements not met");
- static_assert(boost::asio::is_const_buffer_sequence<
- ConstBufferSequence>::value,
- "ConstBufferSequence requirements not met");
- using beast::detail::clamp;
- using boost::asio::buffer;
- using boost::asio::buffer_copy;
- using boost::asio::buffer_size;
- std::size_t bytes_transferred = 0;
- ec.assign(0, ec.category());
- // Make sure the stream is open
- if(! check_open(ec))
- return bytes_transferred;
- detail::frame_header fh;
- if(! wr_cont_)
- {
- begin_msg();
- fh.rsv1 = wr_compress_;
- }
- else
- {
- fh.rsv1 = false;
- }
- fh.rsv2 = false;
- fh.rsv3 = false;
- fh.op = wr_cont_ ?
- detail::opcode::cont : wr_opcode_;
- fh.mask = role_ == role_type::client;
- auto remain = buffer_size(buffers);
- if(wr_compress_)
- {
- buffers_suffix<
- ConstBufferSequence> cb{buffers};
- for(;;)
- {
- auto b = buffer(
- wr_buf_.get(), wr_buf_size_);
- auto const more = this->deflate(
- b, cb, fin, bytes_transferred, ec);
- if(! check_ok(ec))
- return bytes_transferred;
- auto const n = buffer_size(b);
- if(n == 0)
- {
- // The input was consumed, but there
- // is no output due to compression
- // latency.
- BOOST_ASSERT(! fin);
- BOOST_ASSERT(buffer_size(cb) == 0);
- fh.fin = false;
- break;
- }
- if(fh.mask)
- {
- fh.key = this->create_mask();
- detail::prepared_key key;
- detail::prepare_key(key, fh.key);
- detail::mask_inplace(b, key);
- }
- fh.fin = ! more;
- fh.len = n;
- detail::fh_buffer fh_buf;
- detail::write<
- flat_static_buffer_base>(fh_buf, fh);
- wr_cont_ = ! fin;
- boost::asio::write(stream_,
- buffers_cat(fh_buf.data(), b), ec);
- if(! check_ok(ec))
- return bytes_transferred;
- if(! more)
- break;
- fh.op = detail::opcode::cont;
- fh.rsv1 = false;
- }
- if(fh.fin)
- this->do_context_takeover_write(role_);
- }
- else if(! fh.mask)
- {
- if(! wr_frag_)
- {
- // no mask, no autofrag
- fh.fin = fin;
- fh.len = remain;
- detail::fh_buffer fh_buf;
- detail::write<
- flat_static_buffer_base>(fh_buf, fh);
- wr_cont_ = ! fin;
- boost::asio::write(stream_,
- buffers_cat(fh_buf.data(), buffers), ec);
- if(! check_ok(ec))
- return bytes_transferred;
- bytes_transferred += remain;
- }
- else
- {
- // no mask, autofrag
- BOOST_ASSERT(wr_buf_size_ != 0);
- buffers_suffix<
- ConstBufferSequence> cb{buffers};
- for(;;)
- {
- auto const n = clamp(remain, wr_buf_size_);
- remain -= n;
- fh.len = n;
- fh.fin = fin ? remain == 0 : false;
- detail::fh_buffer fh_buf;
- detail::write<
- flat_static_buffer_base>(fh_buf, fh);
- wr_cont_ = ! fin;
- boost::asio::write(stream_,
- buffers_cat(fh_buf.data(),
- buffers_prefix(n, cb)), ec);
- if(! check_ok(ec))
- return bytes_transferred;
- bytes_transferred += n;
- if(remain == 0)
- break;
- fh.op = detail::opcode::cont;
- cb.consume(n);
- }
- }
- }
- else if(! wr_frag_)
- {
- // mask, no autofrag
- fh.fin = fin;
- fh.len = remain;
- fh.key = this->create_mask();
- detail::prepared_key key;
- detail::prepare_key(key, fh.key);
- detail::fh_buffer fh_buf;
- detail::write<
- flat_static_buffer_base>(fh_buf, fh);
- buffers_suffix<
- ConstBufferSequence> cb{buffers};
- {
- auto const n = clamp(remain, wr_buf_size_);
- auto const b = buffer(wr_buf_.get(), n);
- buffer_copy(b, cb);
- cb.consume(n);
- remain -= n;
- detail::mask_inplace(b, key);
- wr_cont_ = ! fin;
- boost::asio::write(stream_,
- buffers_cat(fh_buf.data(), b), ec);
- if(! check_ok(ec))
- return bytes_transferred;
- bytes_transferred += n;
- }
- while(remain > 0)
- {
- auto const n = clamp(remain, wr_buf_size_);
- auto const b = buffer(wr_buf_.get(), n);
- buffer_copy(b, cb);
- cb.consume(n);
- remain -= n;
- detail::mask_inplace(b, key);
- boost::asio::write(stream_, b, ec);
- if(! check_ok(ec))
- return bytes_transferred;
- bytes_transferred += n;
- }
- }
- else
- {
- // mask, autofrag
- BOOST_ASSERT(wr_buf_size_ != 0);
- buffers_suffix<
- ConstBufferSequence> cb{buffers};
- for(;;)
- {
- fh.key = this->create_mask();
- detail::prepared_key key;
- detail::prepare_key(key, fh.key);
- auto const n = clamp(remain, wr_buf_size_);
- auto const b = buffer(wr_buf_.get(), n);
- buffer_copy(b, cb);
- detail::mask_inplace(b, key);
- fh.len = n;
- remain -= n;
- fh.fin = fin ? remain == 0 : false;
- wr_cont_ = ! fh.fin;
- detail::fh_buffer fh_buf;
- detail::write<
- flat_static_buffer_base>(fh_buf, fh);
- boost::asio::write(stream_,
- buffers_cat(fh_buf.data(), b), ec);
- if(! check_ok(ec))
- return bytes_transferred;
- bytes_transferred += n;
- if(remain == 0)
- break;
- fh.op = detail::opcode::cont;
- cb.consume(n);
- }
- }
- return bytes_transferred;
- }
- template<class NextLayer, bool deflateSupported>
- template<class ConstBufferSequence, class WriteHandler>
- BOOST_ASIO_INITFN_RESULT_TYPE(
- WriteHandler, void(error_code, std::size_t))
- stream<NextLayer, deflateSupported>::
- async_write_some(bool fin,
- ConstBufferSequence const& bs, WriteHandler&& handler)
- {
- static_assert(is_async_stream<next_layer_type>::value,
- "AsyncStream requirements not met");
- static_assert(boost::asio::is_const_buffer_sequence<
- ConstBufferSequence>::value,
- "ConstBufferSequence requirements not met");
- BOOST_BEAST_HANDLER_INIT(
- WriteHandler, void(error_code, std::size_t));
- write_some_op<ConstBufferSequence, BOOST_ASIO_HANDLER_TYPE(
- WriteHandler, void(error_code, std::size_t))>{
- std::move(init.completion_handler), *this, fin, bs}(
- {}, 0, false);
- return init.result.get();
- }
- //------------------------------------------------------------------------------
- template<class NextLayer, bool deflateSupported>
- template<class ConstBufferSequence>
- std::size_t
- stream<NextLayer, deflateSupported>::
- write(ConstBufferSequence const& buffers)
- {
- static_assert(is_sync_stream<next_layer_type>::value,
- "SyncStream requirements not met");
- static_assert(boost::asio::is_const_buffer_sequence<
- ConstBufferSequence>::value,
- "ConstBufferSequence requirements not met");
- error_code ec;
- auto const bytes_transferred = write(buffers, ec);
- if(ec)
- BOOST_THROW_EXCEPTION(system_error{ec});
- return bytes_transferred;
- }
- template<class NextLayer, bool deflateSupported>
- template<class ConstBufferSequence>
- std::size_t
- stream<NextLayer, deflateSupported>::
- write(ConstBufferSequence const& buffers, error_code& ec)
- {
- static_assert(is_sync_stream<next_layer_type>::value,
- "SyncStream requirements not met");
- static_assert(boost::asio::is_const_buffer_sequence<
- ConstBufferSequence>::value,
- "ConstBufferSequence requirements not met");
- return write_some(true, buffers, ec);
- }
- template<class NextLayer, bool deflateSupported>
- template<class ConstBufferSequence, class WriteHandler>
- BOOST_ASIO_INITFN_RESULT_TYPE(
- WriteHandler, void(error_code, std::size_t))
- stream<NextLayer, deflateSupported>::
- async_write(
- ConstBufferSequence const& bs, WriteHandler&& handler)
- {
- static_assert(is_async_stream<next_layer_type>::value,
- "AsyncStream requirements not met");
- static_assert(boost::asio::is_const_buffer_sequence<
- ConstBufferSequence>::value,
- "ConstBufferSequence requirements not met");
- BOOST_BEAST_HANDLER_INIT(
- WriteHandler, void(error_code, std::size_t));
- write_some_op<ConstBufferSequence, BOOST_ASIO_HANDLER_TYPE(
- WriteHandler, void(error_code, std::size_t))>{
- std::move(init.completion_handler), *this, true, bs}(
- {}, 0, false);
- return init.result.get();
- }
- } // websocket
- } // beast
- } // boost
- #endif
|