/* This file is part of Telegram Desktop, the official desktop version of Telegram messaging app, see https://telegram.org Telegram Desktop is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. It is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. In addition, as a special exception, the copyright holders give permission to link the code of portions of this program with the OpenSSL library. Full license: https://github.com/telegramdesktop/tdesktop/blob/master/LICENSE Copyright (c) 2014-2016 John Preston, https://desktop.telegram.org */ #pragma once #ifndef OS_MAC_OLD #include <memory> #endif // OS_MAC_OLD namespace base { namespace internal { template <typename Return, typename ...Args> struct lambda_wrap_helper_base { using construct_copy_other_type = void(*)(void *, const void *); // dst, src using construct_move_other_type = void(*)(void *, void *); // dst, src using call_type = Return(*)(const void *, Args...); using destruct_type = void(*)(const void *); lambda_wrap_helper_base() = delete; lambda_wrap_helper_base(const lambda_wrap_helper_base &other) = delete; lambda_wrap_helper_base &operator=(const lambda_wrap_helper_base &other) = delete; lambda_wrap_helper_base( construct_copy_other_type construct_copy_other, construct_move_other_type construct_move_other, call_type call, destruct_type destruct) : construct_copy_other(construct_copy_other) , construct_move_other(construct_move_other) , call(call) , destruct(destruct) { } const construct_copy_other_type construct_copy_other; const construct_move_other_type construct_move_other; const call_type call; const destruct_type destruct; static constexpr size_t kFullStorageSize = sizeof(void*) + 24U; static constexpr size_t kStorageSize = kFullStorageSize - sizeof(void*); template <typename Lambda> using IsLarge = std_::integral_constant<bool, !(sizeof(std_::decay_simple_t<Lambda>) <= kStorageSize)>; protected: static void bad_construct_copy(void *lambda, const void *source) { throw std::exception(); } }; template <typename Return, typename ...Args> struct lambda_wrap_empty : public lambda_wrap_helper_base<Return, Args...> { static void construct_copy_other_method(void *lambda, const void *source) { } static void construct_move_other_method(void *lambda, void *source) { } static Return call_method(const void *lambda, Args... args) { throw std::exception(); } static void destruct_method(const void *lambda) { } lambda_wrap_empty() : lambda_wrap_helper_base<Return, Args...>( &lambda_wrap_empty::construct_copy_other_method, &lambda_wrap_empty::construct_move_other_method, &lambda_wrap_empty::call_method, &lambda_wrap_empty::destruct_method) { } static const lambda_wrap_empty<Return, Args...> instance; }; template <typename Return, typename ...Args> const lambda_wrap_empty<Return, Args...> lambda_wrap_empty<Return, Args...>::instance; template <typename Lambda, typename IsLarge, typename Return, typename ...Args> struct lambda_wrap_helper_move_impl; // // Disable large lambda support. // If you really need it, just store data in some std_::unique_ptr<struct>. // //template <typename Lambda, typename Return, typename ...Args> //struct lambda_wrap_helper_move_impl<Lambda, std_::true_type, Return, Args...> : public lambda_wrap_helper_base<Return, Args...> { // using JustLambda = std_::decay_simple_t<Lambda>; // using LambdaPtr = std_::unique_ptr<JustLambda>; // using Parent = lambda_wrap_helper_base<Return, Args...>; // static void construct_move_other_method(void *lambda, void *source) { // auto source_lambda = static_cast<LambdaPtr*>(source); // new (lambda) LambdaPtr(std_::move(*source_lambda)); // } // static void construct_move_lambda_method(void *lambda, void *source) { // auto source_lambda = static_cast<JustLambda*>(source); // new (lambda) LambdaPtr(std_::make_unique<JustLambda>(static_cast<JustLambda&&>(*source_lambda))); // } // static Return call_method(const void *lambda, Args... args) { // return (**static_cast<const LambdaPtr*>(lambda))(std_::forward<Args>(args)...); // } // static void destruct_method(const void *lambda) { // static_cast<const LambdaPtr*>(lambda)->~LambdaPtr(); // } // lambda_wrap_helper_move_impl() : Parent( // &Parent::bad_construct_copy, // &lambda_wrap_helper_move_impl::construct_move_other_method, // &lambda_wrap_helper_move_impl::call_method, // &lambda_wrap_helper_move_impl::destruct_method) { // } // //protected: // lambda_wrap_helper_move_impl( // typename Parent::construct_copy_other_type construct_copy_other // ) : Parent( // construct_copy_other, // &lambda_wrap_helper_move_impl::construct_move_other_method, // &lambda_wrap_helper_move_impl::call_method, // &lambda_wrap_helper_move_impl::destruct_method) { // } // //}; template <typename Lambda, typename Return, typename ...Args> struct lambda_wrap_helper_move_impl<Lambda, std_::false_type, Return, Args...> : public lambda_wrap_helper_base<Return, Args...> { using JustLambda = std_::decay_simple_t<Lambda>; using Parent = lambda_wrap_helper_base<Return, Args...>; static void construct_move_other_method(void *lambda, void *source) { auto source_lambda = static_cast<JustLambda*>(source); new (lambda) JustLambda(static_cast<JustLambda&&>(*source_lambda)); } static void construct_move_lambda_method(void *lambda, void *source) { static_assert(alignof(JustLambda) <= alignof(void*), "Bad lambda alignment."); auto space = sizeof(JustLambda); auto aligned = std_::align(alignof(JustLambda), space, lambda, space); t_assert(aligned == lambda); auto source_lambda = static_cast<JustLambda*>(source); new (lambda) JustLambda(static_cast<JustLambda&&>(*source_lambda)); } static Return call_method(const void *lambda, Args... args) { return (*static_cast<const JustLambda*>(lambda))(std_::forward<Args>(args)...); } static void destruct_method(const void *lambda) { static_cast<const JustLambda*>(lambda)->~JustLambda(); } lambda_wrap_helper_move_impl() : Parent( &Parent::bad_construct_copy, &lambda_wrap_helper_move_impl::construct_move_other_method, &lambda_wrap_helper_move_impl::call_method, &lambda_wrap_helper_move_impl::destruct_method) { } protected: lambda_wrap_helper_move_impl( typename Parent::construct_copy_other_type construct_copy_other ) : Parent( construct_copy_other, &lambda_wrap_helper_move_impl::construct_move_other_method, &lambda_wrap_helper_move_impl::call_method, &lambda_wrap_helper_move_impl::destruct_method) { } }; template <typename Lambda, typename Return, typename ...Args> struct lambda_wrap_helper_move : public lambda_wrap_helper_move_impl<Lambda , typename lambda_wrap_helper_base<Return, Args...>::template IsLarge<Lambda> , Return, Args...> { static const lambda_wrap_helper_move instance; }; template <typename Lambda, typename Return, typename ...Args> const lambda_wrap_helper_move<Lambda, Return, Args...> lambda_wrap_helper_move<Lambda, Return, Args...>::instance = {}; template <typename Lambda, typename IsLarge, typename Return, typename ...Args> struct lambda_wrap_helper_copy_impl; // // Disable large lambda support. // If you really need it, just store data in some QSharedPointer<struct>. // //template <typename Lambda, typename Return, typename ...Args> //struct lambda_wrap_helper_copy_impl<Lambda, std_::true_type, Return, Args...> : public lambda_wrap_helper_move_impl<Lambda, std_::true_type, Return, Args...> { // using JustLambda = std_::decay_simple_t<Lambda>; // using LambdaPtr = std_::unique_ptr<JustLambda>; // using Parent = lambda_wrap_helper_move_impl<Lambda, std_::true_type, Return, Args...>; // static void construct_copy_other_method(void *lambda, const void *source) { // auto source_lambda = static_cast<const LambdaPtr*>(source); // new (lambda) LambdaPtr(std_::make_unique<JustLambda>(*source_lambda->get())); // } // static void construct_copy_lambda_method(void *lambda, const void *source) { // auto source_lambda = static_cast<const JustLambda*>(source); // new (lambda) LambdaPtr(std_::make_unique<JustLambda>(static_cast<const JustLambda &>(*source_lambda))); // } // lambda_wrap_helper_copy_impl() : Parent(&lambda_wrap_helper_copy_impl::construct_copy_other_method) { // } // //}; template <typename Lambda, typename Return, typename ...Args> struct lambda_wrap_helper_copy_impl<Lambda, std_::false_type, Return, Args...> : public lambda_wrap_helper_move_impl<Lambda, std_::false_type, Return, Args...> { using JustLambda = std_::decay_simple_t<Lambda>; using Parent = lambda_wrap_helper_move_impl<Lambda, std_::false_type, Return, Args...>; static void construct_copy_other_method(void *lambda, const void *source) { auto source_lambda = static_cast<const JustLambda*>(source); new (lambda) JustLambda(static_cast<const JustLambda &>(*source_lambda)); } static void construct_copy_lambda_method(void *lambda, const void *source) { static_assert(alignof(JustLambda) <= alignof(void*), "Bad lambda alignment."); auto space = sizeof(JustLambda); auto aligned = std_::align(alignof(JustLambda), space, lambda, space); t_assert(aligned == lambda); auto source_lambda = static_cast<const JustLambda*>(source); new (lambda) JustLambda(static_cast<const JustLambda &>(*source_lambda)); } lambda_wrap_helper_copy_impl() : Parent(&lambda_wrap_helper_copy_impl::construct_copy_other_method) { } }; template <typename Lambda, typename Return, typename ...Args> struct lambda_wrap_helper_copy : public lambda_wrap_helper_copy_impl<Lambda , typename lambda_wrap_helper_base<Return, Args...>::template IsLarge<Lambda> , Return, Args...> { static const lambda_wrap_helper_copy instance; }; template <typename Lambda, typename Return, typename ...Args> const lambda_wrap_helper_copy<Lambda, Return, Args...> lambda_wrap_helper_copy<Lambda, Return, Args...>::instance = {}; } // namespace internal template <typename Function> class lambda_unique; template <typename Function> class lambda_wrap; template <typename Return, typename ...Args> class lambda_unique<Return(Args...)> { using BaseHelper = internal::lambda_wrap_helper_base<Return, Args...>; using EmptyHelper = internal::lambda_wrap_empty<Return, Args...>; template <typename Lambda> using IsUnique = std_::is_same<lambda_unique, std_::decay_simple_t<Lambda>>; template <typename Lambda> using IsWrap = std_::is_same<lambda_wrap<Return(Args...)>, std_::decay_simple_t<Lambda>>; template <typename Lambda> using IsOther = std_::enable_if_t<!IsUnique<Lambda>::value && !IsWrap<Lambda>::value>; template <typename Lambda> using IsRvalue = std_::enable_if_t<std_::is_rvalue_reference<Lambda&&>::value>; public: lambda_unique() : helper_(&EmptyHelper::instance) { } lambda_unique(const lambda_unique &other) = delete; lambda_unique &operator=(const lambda_unique &other) = delete; lambda_unique(lambda_unique &&other) : helper_(other.helper_) { helper_->construct_move_other(storage_, other.storage_); } lambda_unique &operator=(lambda_unique &&other) { auto temp = std_::move(other); helper_->destruct(storage_); helper_ = temp.helper_; helper_->construct_move_other(storage_, temp.storage_); return *this; } void swap(lambda_unique &other) { if (this != &other) { lambda_unique temp = std_::move(other); other = std_::move(*this); *this = std_::move(other); } } template <typename Lambda, typename = IsOther<Lambda>, typename = IsRvalue<Lambda>> lambda_unique(Lambda &&other) : helper_(&internal::lambda_wrap_helper_move<Lambda, Return, Args...>::instance) { internal::lambda_wrap_helper_move<Lambda, Return, Args...>::construct_move_lambda_method(storage_, &other); } template <typename Lambda, typename = IsOther<Lambda>, typename = IsRvalue<Lambda>> lambda_unique &operator=(Lambda &&other) { auto temp = std_::move(other); helper_->destruct(storage_); helper_ = &internal::lambda_wrap_helper_move<Lambda, Return, Args...>::instance; internal::lambda_wrap_helper_move<Lambda, Return, Args...>::construct_move_lambda_method(storage_, &temp); return *this; } inline Return operator()(Args... args) const { return helper_->call(storage_, std_::forward<Args>(args)...); } explicit operator bool() const { return (helper_ != &EmptyHelper::instance); } ~lambda_unique() { helper_->destruct(storage_); } protected: struct Private { }; lambda_unique(const BaseHelper *helper, const Private &) : helper_(helper) { } const BaseHelper *helper_; static_assert(BaseHelper::kStorageSize % sizeof(void*) == 0, "Bad pointer size."); void *(storage_[BaseHelper::kStorageSize / sizeof(void*)]); }; template <typename Return, typename ...Args> class lambda_wrap<Return(Args...)> : public lambda_unique<Return(Args...)> { using BaseHelper = internal::lambda_wrap_helper_base<Return, Args...>; using Parent = lambda_unique<Return(Args...)>; template <typename Lambda> using IsOther = std_::enable_if_t<!std_::is_same<lambda_wrap, std_::decay_simple_t<Lambda>>::value>; template <typename Lambda> using IsRvalue = std_::enable_if_t<std_::is_rvalue_reference<Lambda&&>::value>; template <typename Lambda> using IsNotRvalue = std_::enable_if_t<!std_::is_rvalue_reference<Lambda&&>::value>; public: lambda_wrap() = default; lambda_wrap(const lambda_wrap &other) : Parent(other.helper_, typename Parent::Private()) { this->helper_->construct_copy_other(this->storage_, other.storage_); } lambda_wrap &operator=(const lambda_wrap &other) { auto temp = other; temp.swap(*this); return *this; } lambda_wrap(lambda_wrap &&other) = default; lambda_wrap &operator=(lambda_wrap &&other) = default; void swap(lambda_wrap &other) { if (this != &other) { lambda_wrap temp = std_::move(other); other = std_::move(*this); *this = std_::move(other); } } template <typename Lambda, typename = IsOther<Lambda>> lambda_wrap(const Lambda &other) : Parent(&internal::lambda_wrap_helper_copy<Lambda, Return, Args...>::instance, typename Parent::Private()) { internal::lambda_wrap_helper_copy<Lambda, Return, Args...>::construct_copy_lambda_method(this->storage_, &other); } template <typename Lambda, typename = IsOther<Lambda>, typename = IsRvalue<Lambda>> lambda_wrap(Lambda &&other) : Parent(&internal::lambda_wrap_helper_copy<Lambda, Return, Args...>::instance, typename Parent::Private()) { internal::lambda_wrap_helper_copy<Lambda, Return, Args...>::construct_move_lambda_method(this->storage_, &other); } template <typename Lambda, typename = IsOther<Lambda>> lambda_wrap &operator=(const Lambda &other) { auto temp = other; this->helper_->destruct(this->storage_); this->helper_ = &internal::lambda_wrap_helper_copy<Lambda, Return, Args...>::instance; internal::lambda_wrap_helper_copy<Lambda, Return, Args...>::construct_copy_lambda_method(this->storage_, &other); return *this; } template <typename Lambda, typename = IsOther<Lambda>, typename = IsRvalue<Lambda>> lambda_wrap &operator=(Lambda &&other) { auto temp = std_::move(other); this->helper_->destruct(this->storage_); this->helper_ = &internal::lambda_wrap_helper_copy<Lambda, Return, Args...>::instance; internal::lambda_wrap_helper_copy<Lambda, Return, Args...>::construct_move_lambda_method(this->storage_, &other); return *this; } }; } // namespace base // While we still use rpcDone() and rpcFail() #include "mtproto/rpc_sender.h" template <typename FunctionType> struct LambdaUniqueHelper; template <typename Lambda, typename R, typename ...Args> struct LambdaUniqueHelper<R(Lambda::*)(Args...) const> { using UniqueType = base::lambda_unique<R(Args...)>; }; template <typename FunctionType> using LambdaGetUnique = typename LambdaUniqueHelper<FunctionType>::UniqueType; template <typename Base, typename FunctionType> class RPCHandlerImplementation : public Base { protected: using Lambda = base::lambda_unique<FunctionType>; using Parent = RPCHandlerImplementation<Base, FunctionType>; public: RPCHandlerImplementation(Lambda &&handler) : _handler(std_::move(handler)) { } protected: Lambda _handler; }; template <typename FunctionType> using RPCDoneHandlerImplementation = RPCHandlerImplementation<RPCAbstractDoneHandler, FunctionType>; template <typename R> class RPCDoneHandlerImplementationBare : public RPCDoneHandlerImplementation<R(const mtpPrime*, const mtpPrime*)> { // done(from, end) public: using RPCDoneHandlerImplementation<R(const mtpPrime*, const mtpPrime*)>::Parent::Parent; void operator()(mtpRequestId requestId, const mtpPrime *from, const mtpPrime *end) const override { return this->_handler ? this->_handler(from, end) : void(0); } }; template <typename R> class RPCDoneHandlerImplementationBareReq : public RPCDoneHandlerImplementation<R(const mtpPrime*, const mtpPrime*, mtpRequestId)> { // done(from, end, req_id) public: using RPCDoneHandlerImplementation<R(const mtpPrime*, const mtpPrime*, mtpRequestId)>::Parent::Parent; void operator()(mtpRequestId requestId, const mtpPrime *from, const mtpPrime *end) const override { return this->_handler ? this->_handler(from, end, requestId) : void(0); } }; template <typename R, typename T> class RPCDoneHandlerImplementationPlain : public RPCDoneHandlerImplementation<R(const T&)> { // done(result) public: using RPCDoneHandlerImplementation<R(const T&)>::Parent::Parent; void operator()(mtpRequestId requestId, const mtpPrime *from, const mtpPrime *end) const override { return this->_handler ? this->_handler(T(from, end)) : void(0); } }; template <typename R, typename T> class RPCDoneHandlerImplementationReq : public RPCDoneHandlerImplementation<R(const T&, mtpRequestId)> { // done(result, req_id) public: using RPCDoneHandlerImplementation<R(const T&, mtpRequestId)>::Parent::Parent; void operator()(mtpRequestId requestId, const mtpPrime *from, const mtpPrime *end) const override { return this->_handler ? this->_handler(T(from, end), requestId) : void(0); } }; template <typename R> class RPCDoneHandlerImplementationNo : public RPCDoneHandlerImplementation<R()> { // done() public: using RPCDoneHandlerImplementation<R()>::Parent::Parent; void operator()(mtpRequestId requestId, const mtpPrime *from, const mtpPrime *end) const override { return this->_handler ? this->_handler() : void(0); } }; template <typename R> class RPCDoneHandlerImplementationNoReq : public RPCDoneHandlerImplementation<R(mtpRequestId)> { // done(req_id) public: using RPCDoneHandlerImplementation<R(mtpRequestId)>::Parent::Parent; void operator()(mtpRequestId requestId, const mtpPrime *from, const mtpPrime *end) const override { return this->_handler ? this->_handler(requestId) : void(0); } }; template <typename R> inline RPCDoneHandlerPtr rpcDone_lambda_wrap_helper(base::lambda_unique<R(const mtpPrime*, const mtpPrime*)> &&lambda) { return RPCDoneHandlerPtr(new RPCDoneHandlerImplementationBare<R>(std_::move(lambda))); } template <typename R> inline RPCDoneHandlerPtr rpcDone_lambda_wrap_helper(base::lambda_unique<R(const mtpPrime*, const mtpPrime*, mtpRequestId)> &&lambda) { return RPCDoneHandlerPtr(new RPCDoneHandlerImplementationBareReq<R>(std_::move(lambda))); } template <typename R, typename T> inline RPCDoneHandlerPtr rpcDone_lambda_wrap_helper(base::lambda_unique<R(const T&)> &&lambda) { return RPCDoneHandlerPtr(new RPCDoneHandlerImplementationPlain<R, T>(std_::move(lambda))); } template <typename R, typename T> inline RPCDoneHandlerPtr rpcDone_lambda_wrap_helper(base::lambda_unique<R(const T&, mtpRequestId)> &&lambda) { return RPCDoneHandlerPtr(new RPCDoneHandlerImplementationReq<R, T>(std_::move(lambda))); } template <typename R> inline RPCDoneHandlerPtr rpcDone_lambda_wrap_helper(base::lambda_unique<R()> &&lambda) { return RPCDoneHandlerPtr(new RPCDoneHandlerImplementationNo<R>(std_::move(lambda))); } template <typename R> inline RPCDoneHandlerPtr rpcDone_lambda_wrap_helper(base::lambda_unique<R(mtpRequestId)> &&lambda) { return RPCDoneHandlerPtr(new RPCDoneHandlerImplementationNoReq<R>(std_::move(lambda))); } template <typename Lambda, typename = std_::enable_if_t<std_::is_rvalue_reference<Lambda&&>::value>> RPCDoneHandlerPtr rpcDone(Lambda &&lambda) { return rpcDone_lambda_wrap_helper(LambdaGetUnique<decltype(&Lambda::operator())>(std_::move(lambda))); } template <typename FunctionType> using RPCFailHandlerImplementation = RPCHandlerImplementation<RPCAbstractFailHandler, FunctionType>; class RPCFailHandlerImplementationPlain : public RPCFailHandlerImplementation<bool(const RPCError&)> { // fail(error) public: using Parent::Parent; bool operator()(mtpRequestId requestId, const RPCError &error) const override { return _handler ? _handler(error) : true; } }; class RPCFailHandlerImplementationReq : public RPCFailHandlerImplementation<bool(const RPCError&, mtpRequestId)> { // fail(error, req_id) public: using Parent::Parent; bool operator()(mtpRequestId requestId, const RPCError &error) const override { return this->_handler ? this->_handler(error, requestId) : true; } }; class RPCFailHandlerImplementationNo : public RPCFailHandlerImplementation<bool()> { // fail() public: using Parent::Parent; bool operator()(mtpRequestId requestId, const RPCError &error) const override { return this->_handler ? this->_handler() : true; } }; class RPCFailHandlerImplementationNoReq : public RPCFailHandlerImplementation<bool(mtpRequestId)> { // fail(req_id) public: using Parent::Parent; bool operator()(mtpRequestId requestId, const RPCError &error) const override { return this->_handler ? this->_handler(requestId) : true; } }; inline RPCFailHandlerPtr rpcFail_lambda_wrap_helper(base::lambda_unique<bool(const RPCError&)> &&lambda) { return RPCFailHandlerPtr(new RPCFailHandlerImplementationPlain(std_::move(lambda))); } inline RPCFailHandlerPtr rpcFail_lambda_wrap_helper(base::lambda_unique<bool(const RPCError&, mtpRequestId)> &&lambda) { return RPCFailHandlerPtr(new RPCFailHandlerImplementationReq(std_::move(lambda))); } inline RPCFailHandlerPtr rpcFail_lambda_wrap_helper(base::lambda_unique<bool()> &&lambda) { return RPCFailHandlerPtr(new RPCFailHandlerImplementationNo(std_::move(lambda))); } inline RPCFailHandlerPtr rpcFail_lambda_wrap_helper(base::lambda_unique<bool(mtpRequestId)> &&lambda) { return RPCFailHandlerPtr(new RPCFailHandlerImplementationNoReq(std_::move(lambda))); } template <typename Lambda, typename = std_::enable_if_t<std_::is_rvalue_reference<Lambda&&>::value>> RPCFailHandlerPtr rpcFail(Lambda &&lambda) { return rpcFail_lambda_wrap_helper(LambdaGetUnique<decltype(&Lambda::operator())>(std_::move(lambda))); }