This repository has been archived on 2020-05-25. You can view files and clone it, but cannot push or open issues or pull requests.
tdlib-fork/td/mtproto/PacketStorer.h
levlam 80c35676a2 Update copyright year.
GitOrigin-RevId: 09afb551b6e637dc69739fa735b0051a38b9e14c
2020-01-01 04:23:48 +03:00

45 lines
1007 B
C++

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2020
//
// 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)
//
#pragma once
#include "td/utils/StorerBase.h"
#include "td/utils/tl_storers.h"
#include <limits>
namespace td {
namespace mtproto {
template <class Impl>
class PacketStorer
: public Storer
, public Impl {
public:
size_t size() const override {
if (size_ != std::numeric_limits<size_t>::max()) {
return size_;
}
TlStorerCalcLength storer;
this->do_store(storer);
return size_ = storer.get_length();
}
size_t store(uint8 *ptr) const override {
TlStorerUnsafe storer(ptr);
this->do_store(storer);
return static_cast<size_t>(storer.get_buf() - ptr);
}
using Impl::Impl;
private:
mutable size_t size_ = std::numeric_limits<size_t>::max();
};
} // namespace mtproto
} // namespace td