mirror of
https://github.com/vale981/tdesktop
synced 2025-03-11 21:46:41 -04:00
7d89b54d1c
If we start logging in and we know, that some of the authorization keys were read from the hard drive, not generated, we destroy all the existing authorization keys and start generating new keys.
145 lines
4.6 KiB
C++
145 lines
4.6 KiB
C++
/*
|
|
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-2017 John Preston, https://desktop.telegram.org
|
|
*/
|
|
#pragma once
|
|
|
|
#include "mtproto/dcenter.h"
|
|
#include <map>
|
|
#include <set>
|
|
|
|
namespace MTP {
|
|
|
|
class DcOptions;
|
|
class Session;
|
|
|
|
class Instance : public QObject {
|
|
Q_OBJECT
|
|
|
|
public:
|
|
struct Config {
|
|
static constexpr auto kNoneMainDc = -1;
|
|
static constexpr auto kNotSetMainDc = 0;
|
|
static constexpr auto kDefaultMainDc = 2;
|
|
|
|
DcId mainDcId = kNotSetMainDc;
|
|
AuthKeysList keys;
|
|
};
|
|
enum class Mode {
|
|
Normal,
|
|
KeysDestroyer,
|
|
};
|
|
Instance(DcOptions *options, Mode mode, Config &&config);
|
|
|
|
Instance(const Instance &other) = delete;
|
|
Instance &operator=(const Instance &other) = delete;
|
|
|
|
void suggestMainDcId(DcId mainDcId);
|
|
void setMainDcId(DcId mainDcId);
|
|
DcId mainDcId() const;
|
|
|
|
void setKeyForWrite(DcId dcId, const AuthKeyPtr &key);
|
|
AuthKeysList getKeysForWrite() const;
|
|
void addKeysForDestroy(AuthKeysList &&keys);
|
|
|
|
DcOptions *dcOptions();
|
|
|
|
template <typename TRequest>
|
|
mtpRequestId send(const TRequest &request, RPCResponseHandler callbacks = RPCResponseHandler(), ShiftedDcId dcId = 0, TimeMs msCanWait = 0, mtpRequestId after = 0) {
|
|
if (auto session = getSession(dcId)) {
|
|
return session->send(request, callbacks, msCanWait, true, !dcId, after);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
template <typename TRequest>
|
|
mtpRequestId send(const TRequest &request, RPCDoneHandlerPtr onDone, RPCFailHandlerPtr onFail = RPCFailHandlerPtr(), int32 dc = 0, TimeMs msCanWait = 0, mtpRequestId after = 0) {
|
|
return send(request, RPCResponseHandler(onDone, onFail), dc, msCanWait, after);
|
|
}
|
|
|
|
void sendAnything(ShiftedDcId dcId = 0, TimeMs msCanWait = 0) {
|
|
if (auto session = getSession(dcId)) {
|
|
session->sendAnything(msCanWait);
|
|
}
|
|
}
|
|
|
|
void restart();
|
|
void restart(ShiftedDcId shiftedDcId);
|
|
int32 dcstate(ShiftedDcId shiftedDcId = 0);
|
|
QString dctransport(ShiftedDcId shiftedDcId = 0);
|
|
void ping();
|
|
void cancel(mtpRequestId requestId);
|
|
int32 state(mtpRequestId requestId); // < 0 means waiting for such count of ms
|
|
void killSession(ShiftedDcId shiftedDcId);
|
|
void stopSession(ShiftedDcId shiftedDcId);
|
|
void logout(RPCDoneHandlerPtr onDone, RPCFailHandlerPtr onFail);
|
|
|
|
internal::DcenterPtr getDcById(ShiftedDcId shiftedDcId);
|
|
void unpaused();
|
|
|
|
void queueQuittingConnection(std::unique_ptr<internal::Connection> connection);
|
|
|
|
void setUpdatesHandler(RPCDoneHandlerPtr onDone);
|
|
void setGlobalFailHandler(RPCFailHandlerPtr onFail);
|
|
void setStateChangedHandler(base::lambda<void(ShiftedDcId shiftedDcId, int32 state)> &&handler);
|
|
void setSessionResetHandler(base::lambda<void(ShiftedDcId shiftedDcId)> &&handler);
|
|
void clearGlobalHandlers();
|
|
|
|
void onStateChange(ShiftedDcId dcWithShift, int32 state);
|
|
void onSessionReset(ShiftedDcId dcWithShift);
|
|
|
|
void registerRequest(mtpRequestId requestId, ShiftedDcId dcWithShift);
|
|
mtpRequestId storeRequest(mtpRequest &request, const RPCResponseHandler &parser);
|
|
mtpRequest getRequest(mtpRequestId requestId);
|
|
void clearCallbacksDelayed(const RPCCallbackClears &requestIds);
|
|
|
|
void execCallback(mtpRequestId requestId, const mtpPrime *from, const mtpPrime *end);
|
|
bool hasCallbacks(mtpRequestId requestId);
|
|
void globalCallback(const mtpPrime *from, const mtpPrime *end);
|
|
|
|
// return true if need to clean request data
|
|
bool rpcErrorOccured(mtpRequestId requestId, const RPCFailHandlerPtr &onFail, const RPCError &err);
|
|
|
|
bool isKeysDestroyer() const;
|
|
void scheduleKeyDestroy(ShiftedDcId shiftedDcId);
|
|
|
|
~Instance();
|
|
|
|
public slots:
|
|
void configLoadRequest();
|
|
void connectionFinished(internal::Connection *connection);
|
|
|
|
signals:
|
|
void configLoaded();
|
|
void keyDestroyed(qint32 shiftedDcId);
|
|
void allKeysDestroyed();
|
|
|
|
private slots:
|
|
void onKeyDestroyed(qint32 shiftedDcId);
|
|
void onClearKilledSessions();
|
|
|
|
private:
|
|
internal::Session *getSession(ShiftedDcId shiftedDcId);
|
|
|
|
class Private;
|
|
const std::unique_ptr<Private> _private;
|
|
|
|
};
|
|
|
|
} // namespace MTP
|