1 files changed, 128 insertions, 0 deletions
diff --git a/protocols/Tox/src/tox_connection.cpp b/protocols/Tox/src/tox_connection.cpp
new file mode 100644
index 0000000000..2d78fd86fe
--- /dev/null
+++ b/protocols/Tox/src/tox_connection.cpp
@@ -0,0 +1,128 @@
+#include "stdafx.h"
+
+bool CToxProto::IsOnline()
+{
+	return toxThread && m_iStatus >= ID_STATUS_ONLINE;
+}
+
+void CToxProto::TryConnect(Tox *tox)
+{
+	TOX_CONNECTION connectionStatus = tox_self_get_connection_status(tox);
+	if (connectionStatus != TOX_CONNECTION_NONE)
+	{
+		debugLogA(__FUNCTION__": successfuly connected to DHT");
+
+		m_iStatus = m_iDesiredStatus;
+		ProtoBroadcastAck(NULL, ACKTYPE_STATUS, ACKRESULT_SUCCESS, (HANDLE)ID_STATUS_CONNECTING, m_iStatus);
+		tox_self_set_status(tox, MirandaToToxStatus(m_iStatus));
+
+		debugLogA(__FUNCTION__": changing status from %i to %i", ID_STATUS_CONNECTING, m_iDesiredStatus);
+
+		LoadFriendList(tox);
+		return;
+	}
+
+	if (m_iStatus++ > TOX_MAX_CONNECT_RETRIES)
+	{
+		SetStatus(ID_STATUS_OFFLINE);
+		ProtoBroadcastAck(NULL, ACKTYPE_LOGIN, ACKRESULT_FAILED, NULL, LOGINERR_NONETWORK);
+		debugLogA(__FUNCTION__": failed to connect to DHT");
+		return;
+	}
+}
+
+void CToxProto::CheckConnection(Tox *tox, int &retriesCount)
+{
+	TOX_CONNECTION connectionStatus = tox_self_get_connection_status(tox);
+	if (connectionStatus != TOX_CONNECTION_NONE)
+	{
+		if (retriesCount < TOX_MAX_DISCONNECT_RETRIES)
+		{
+			debugLogA(__FUNCTION__": restored connection with DHT");
+			retriesCount = TOX_MAX_DISCONNECT_RETRIES;
+		}
+	}
+	else
+	{
+		if (retriesCount == TOX_MAX_DISCONNECT_RETRIES)
+		{
+			retriesCount--;
+			debugLogA(__FUNCTION__": lost connection with DHT");
+		}
+		else if (!(--retriesCount))
+		{
+			debugLogA(__FUNCTION__": disconnected from DHT");
+			SetStatus(ID_STATUS_OFFLINE);
+			return;
+		}
+	}
+}
+
+void CToxProto::CheckingThread(void *arg)
+{
+	Thread_SetName("TOX: CheckingThread");
+
+	debugLogA(__FUNCTION__": entering");
+
+	Tox *tox = (Tox*)arg;
+	int retriesCount = TOX_MAX_DISCONNECT_RETRIES;
+	while (!isTerminated)
+	{
+		if (m_iStatus < ID_STATUS_ONLINE)
+			TryConnect(tox);
+		else
+			CheckConnection(tox, retriesCount);
+
+		WaitForSingleObject(hTerminateEvent, 1000);
+	}
+
+	debugLogA(__FUNCTION__": leaving");
+}
+
+void CToxProto::PollingThread(void*)
+{
+	Thread_SetName("TOX: PollingThread");
+
+	debugLogA(__FUNCTION__": entering");
+
+	Tox_Options *options = GetToxOptions();
+	if (!options)
+	{
+		SetStatus(ID_STATUS_OFFLINE);
+		ProtoBroadcastAck(NULL, ACKTYPE_LOGIN, ACKRESULT_FAILED, NULL);
+		debugLogA(__FUNCTION__": leaving");
+		return;
+	}
+
+	TOX_ERR_NEW error;
+	CToxThread toxThread(options, &error);
+	if (error != TOX_ERR_NEW_OK)
+	{
+		SetStatus(ID_STATUS_OFFLINE);
+		debugLogA(__FUNCTION__": failed to initialize tox core (%d)", error);
+		ShowNotification(TranslateT("Unable to initialize Tox core"), ToxErrorToString(error), MB_ICONERROR);
+		tox_options_free(options);
+		debugLogA(__FUNCTION__": leaving");
+		return;
+	}
+	tox_options_free(options);
+
+	this->toxThread = &toxThread;
+	InitToxCore(toxThread.Tox());
+	BootstrapNodes(toxThread.Tox());
+	ForkThread(&CToxProto::CheckingThread, toxThread.Tox());
+
+	while (!isTerminated)
+	{
+		tox_iterate(toxThread.Tox(), this);
+		uint32_t interval = tox_iteration_interval(toxThread.Tox());
+		interval = interval ? interval : 50;
+
+		WaitForSingleObject(hTerminateEvent, interval);
+	}
+
+	UninitToxCore(toxThread.Tox());
+	this->toxThread = NULL;
+
+	debugLogA(__FUNCTION__": leaving");
+}
+\ No newline at end of file