123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151 |
- #ifndef BOOST_BEAST_ZLIB_DETAIL_WINDOW_HPP
- #define BOOST_BEAST_ZLIB_DETAIL_WINDOW_HPP
- #include <boost/assert.hpp>
- #include <boost/make_unique.hpp>
- #include <cstdint>
- #include <cstring>
- #include <memory>
- namespace boost {
- namespace beast {
- namespace zlib {
- namespace detail {
- class window
- {
- std::unique_ptr<std::uint8_t[]> p_;
- std::uint16_t i_ = 0;
- std::uint16_t size_ = 0;
- std::uint16_t capacity_ = 0;
- std::uint8_t bits_ = 0;
- public:
- int
- bits() const
- {
- return bits_;
- }
- unsigned
- capacity() const
- {
- return capacity_;
- }
- unsigned
- size() const
- {
- return size_;
- }
- void
- reset(int bits)
- {
- if(bits_ != bits)
- {
- p_.reset();
- bits_ = static_cast<std::uint8_t>(bits);
- capacity_ = 1U << bits_;
- }
- i_ = 0;
- size_ = 0;
- }
- void
- read(std::uint8_t* out, std::size_t pos, std::size_t n)
- {
- if(i_ >= size_)
- {
-
- std::memcpy(out, &p_[i_ - pos], n);
- return;
- }
- auto i = ((i_ - pos) + capacity_) % capacity_;
- auto m = capacity_ - i;
- if(n <= m)
- {
- std::memcpy(out, &p_[i], n);
- return;
- }
- std::memcpy(out, &p_[i], m);
- out += m;
- std::memcpy(out, &p_[0], n - m);
- }
- void
- write(std::uint8_t const* in, std::size_t n)
- {
- if(! p_)
- p_ = boost::make_unique<
- std::uint8_t[]>(capacity_);
- if(n >= capacity_)
- {
- i_ = 0;
- size_ = capacity_;
- std::memcpy(&p_[0], in + (n - size_), size_);
- return;
- }
- if(i_ + n <= capacity_)
- {
- std::memcpy(&p_[i_], in, n);
- if(size_ >= capacity_ - n)
- size_ = capacity_;
- else
- size_ = static_cast<std::uint16_t>(size_ + n);
- i_ = static_cast<std::uint16_t>(
- (i_ + n) % capacity_);
- return;
- }
- auto m = capacity_ - i_;
- std::memcpy(&p_[i_], in, m);
- in += m;
- i_ = static_cast<std::uint16_t>(n - m);
- std::memcpy(&p_[0], in, i_);
- size_ = capacity_;
- }
- };
- }
- }
- }
- }
- #endif
|