init

Telegram/SourceFiles/mtproto/connection_tcp.h

@@ -0,0 +1,102 @@
+/*
+This file is part of Telegram Desktop,
+the official desktop application for the Telegram messaging service.
+
+For license and copyright information please follow this link:
+https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
+*/
+#pragma once
+
+#include "mtproto/connection_abstract.h"
+#include "mtproto/mtproto_auth_key.h"
+
+namespace MTP {
+namespace details {
+
+class AbstractSocket;
+
+class TcpConnection : public AbstractConnection {
+public:
+	TcpConnection(
+		not_null<Instance*> instance,
+		QThread *thread,
+		const ProxyData &proxy);
+
+	ConnectionPointer clone(const ProxyData &proxy) override;
+
+	crl::time pingTime() const override;
+	crl::time fullConnectTimeout() const override;
+	void sendData(mtpBuffer &&buffer) override;
+	void disconnectFromServer() override;
+	void connectToServer(
+		const QString &address,
+		int port,
+		const bytes::vector &protocolSecret,
+		int16 protocolDcId,
+		bool protocolForFiles) override;
+	void timedOut() override;
+	bool isConnected() const override;
+
+	int32 debugState() const override;
+
+	QString transport() const override;
+	QString tag() const override;
+
+	~TcpConnection();
+
+private:
+	enum class Status {
+		Waiting = 0,
+		Ready,
+		Finished,
+	};
+
+	void socketRead();
+	bytes::const_span prepareConnectionStartPrefix(bytes::span buffer);
+
+	void socketPacket(bytes::const_span bytes);
+
+	void socketConnected();
+	void socketDisconnected();
+	void socketError();
+
+	mtpBuffer parsePacket(bytes::const_span bytes);
+	void ensureAvailableInBuffer(int amount);
+	static uint32 fourCharsToUInt(char ch1, char ch2, char ch3, char ch4) {
+		char ch[4] = { ch1, ch2, ch3, ch4 };
+		return *reinterpret_cast<uint32*>(ch);
+	}
+
+	const not_null<Instance*> _instance;
+	std::unique_ptr<AbstractSocket> _socket;
+	bool _connectionStarted = false;
+
+	int _offsetBytes = 0;
+	int _readBytes = 0;
+	int _leftBytes = 0;
+	bytes::vector _smallBuffer;
+	bytes::vector _largeBuffer;
+	bool _usingLargeBuffer = false;
+
+	uchar _sendKey[CTRState::KeySize];
+	CTRState _sendState;
+	uchar _receiveKey[CTRState::KeySize];
+	CTRState _receiveState;
+	class Protocol;
+	std::unique_ptr<Protocol> _protocol;
+	int16 _protocolDcId = 0;
+
+	Status _status = Status::Waiting;
+	MTPint128 _checkNonce;
+
+	QString _address;
+	int32 _port = 0;
+	crl::time _pingTime = 0;
+
+	rpl::lifetime _connectedLifetime;
+	rpl::lifetime _lifetime;
+
+};
+
+} // namespace details
+} // namespace MTP