87 lines
3.8 KiB
C++
87 lines
3.8 KiB
C++
/*
|
|
This file is part of Telegram Desktop,
|
|
an unofficial desktop 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.
|
|
|
|
Full license: https://github.com/telegramdesktop/tdesktop/blob/master/LICENSE
|
|
Copyright (c) 2014 John Preston, https://tdesktop.com
|
|
*/
|
|
#pragma once
|
|
|
|
template <typename TRequest>
|
|
mtpRequestId MTProtoSession::send(const TRequest &request, RPCResponseHandler callbacks, uint64 msCanWait, uint32 layer, bool toMainDC) {
|
|
mtpRequestId requestId = 0;
|
|
if (layer && dc->needConnectionInit()) {
|
|
MTPInitConnection<TRequest> requestWrap(MTPinitConnection<TRequest>(MTP_int(ApiId), MTP_string(cApiDeviceModel()), MTP_string(cApiSystemVersion()), MTP_string(cApiAppVersion()), MTP_string(ApiLang), request));
|
|
return sendFirst(requestWrap, callbacks, msCanWait, layer, toMainDC);
|
|
}
|
|
try {
|
|
uint32 requestSize = request.size() >> 2;
|
|
if (dc->connectionInited()) layer = 0;
|
|
mtpRequest reqSerialized(mtpRequestData::prepare(requestSize + (layer ? 1 : 0)));
|
|
if (layer) reqSerialized->push_back(mtpLayers[layer]);
|
|
request.write(*reqSerialized);
|
|
|
|
DEBUG_LOG(("MTP Info: adding request to toSendMap, msCanWait %1").arg(msCanWait));
|
|
|
|
reqSerialized->msDate = getms(); // > 0 - can send without container
|
|
requestId = _mtp_internal::storeRequest(reqSerialized, callbacks);
|
|
|
|
sendPrepared(reqSerialized, msCanWait);
|
|
} catch (Exception &e) {
|
|
requestId = 0;
|
|
_mtp_internal::rpcErrorOccured(requestId, callbacks, rpcClientError("NO_REQUEST_ID", QString("send() failed to queue request, exception: %1").arg(e.what())));
|
|
}
|
|
if (requestId) _mtp_internal::registerRequest(requestId, toMainDC ? -getDC() : getDC());
|
|
return requestId;
|
|
}
|
|
|
|
class RPCWrappedDcDoneHandler : public RPCAbstractDoneHandler {
|
|
public:
|
|
RPCWrappedDcDoneHandler(const MTProtoDCPtr &dc, const RPCDoneHandlerPtr &ondone) : _dc(dc), _ondone(ondone) {
|
|
}
|
|
|
|
void operator()(mtpRequestId requestId, const mtpPrime *from, const mtpPrime *end) const {
|
|
_dc->setConnectionInited();
|
|
return (*_ondone)(requestId, from, end);
|
|
}
|
|
|
|
private:
|
|
MTProtoDCPtr _dc;
|
|
RPCDoneHandlerPtr _ondone;
|
|
};
|
|
|
|
template <typename TRequest>
|
|
mtpRequestId MTProtoSession::sendFirst(const MTPInitConnection<TRequest> &request, RPCResponseHandler callbacks, uint64 msCanWait, uint32 layer, bool toMainDC) {
|
|
mtpRequestId requestId = 0;
|
|
try {
|
|
uint32 requestSize = request.size() >> 2;
|
|
mtpRequest reqSerialized(mtpRequestData::prepare(requestSize + (layer ? 1 : 0)));
|
|
if (layer) reqSerialized->push_back(mtpLayers[layer]);
|
|
request.write(*reqSerialized);
|
|
|
|
DEBUG_LOG(("MTP Info: adding wrapped to init connection request to toSendMap, msCanWait %1").arg(msCanWait));
|
|
callbacks.onDone = RPCDoneHandlerPtr(new RPCWrappedDcDoneHandler(dc, callbacks.onDone));
|
|
reqSerialized->msDate = getms(); // > 0 - can send without container
|
|
requestId = _mtp_internal::storeRequest(reqSerialized, callbacks);
|
|
|
|
sendPrepared(reqSerialized, msCanWait);
|
|
} catch (Exception &e) {
|
|
requestId = 0;
|
|
_mtp_internal::rpcErrorOccured(requestId, callbacks, rpcClientError("NO_REQUEST_ID", QString("sendFirst() failed to queue request, exception: %1").arg(e.what())));
|
|
}
|
|
if (requestId) {
|
|
_mtp_internal::registerRequest(requestId, toMainDC ? -getDC() : getDC());
|
|
}
|
|
return requestId;
|
|
}
|