From f9d22579b0d94894770361b871141e6c0f76842b Mon Sep 17 00:00:00 2001
From: George Hazan <george.hazan@gmail.com>
Date: Sun, 6 Oct 2013 20:27:53 +0000
Subject: VK: first version that logs in

git-svn-id: http://svn.miranda-ng.org/main/trunk@6379 1316c22d-e87f-b044-9b9b-93d7a3e3ba9c
---
 protocols/VKontakte/src/vk_thread.cpp | 98 ++++++++++++++++++++++++++++++++++-
 1 file changed, 97 insertions(+), 1 deletion(-)

(limited to 'protocols/VKontakte/src/vk_thread.cpp')

diff --git a/protocols/VKontakte/src/vk_thread.cpp b/protocols/VKontakte/src/vk_thread.cpp
index 71ee02c2a3..23ef63165a 100644
--- a/protocols/VKontakte/src/vk_thread.cpp
+++ b/protocols/VKontakte/src/vk_thread.cpp
@@ -27,19 +27,115 @@ void CVkProto::ShutdownSession()
 	OnLoggedOut();
 }
 
+void CVkProto::ConnectionFailed(int iReason)
+{
+	delSetting("AccessToken");
+
+	ProtoBroadcastAck(NULL, ACKTYPE_LOGIN, ACKRESULT_FAILED, NULL, iReason);
+
+	ShutdownSession();
+}
+
+void CVkProto::OnLoggedIn()
+{
+	m_bOnline = true;
+
+	ProtoBroadcastAck(NULL, ACKTYPE_STATUS, ACKRESULT_SUCCESS, (HANDLE)m_iStatus, m_iDesiredStatus);
+	m_iStatus = m_iDesiredStatus;
+
+	RequestMyInfo();
+}
+
 void CVkProto::OnLoggedOut()
 {
 	m_hWorkerThread = 0;
+	m_bOnline = false;
 
 	ProtoBroadcastAck(NULL, ACKTYPE_STATUS, ACKRESULT_SUCCESS, (HANDLE)m_iStatus, ID_STATUS_OFFLINE);
 	m_iStatus = m_iDesiredStatus = ID_STATUS_OFFLINE;
 }
 
+/////////////////////////////////////////////////////////////////////////////////////////
+
+extern CMStringA loginCookie;
+
+static char VK_TOKEN_BEG[] = "access_token=";
+
 void CVkProto::OnOAuthAuthorize(NETLIBHTTPREQUEST *reply)
 {
+	if (reply->resultCode == 302) { // manual redirect
+		LPCSTR pszLocation = findHeader(reply, "Location");
+		if (pszLocation) {
+			if ( !_strnicmp(pszLocation, VK_REDIRECT_URL, sizeof(VK_REDIRECT_URL)-1)) {
+				m_szAccessToken = NULL;
+				LPCSTR p = strstr(pszLocation, VK_TOKEN_BEG);
+				if (p) {
+					p += sizeof(VK_TOKEN_BEG)-1;
+					for (LPCSTR q = p+1; *q; q++) {
+						if (*q == '&' || *q == '=' || *q == '\"') {
+							m_szAccessToken = mir_strndup(p, q-p);
+							break;
+						}
+					}
+					if (m_szAccessToken == NULL)
+						m_szAccessToken = mir_strdup(p);
+					setString("AccessToken", m_szAccessToken);
+				}
+				else delSetting("AccessToken");
+
+				OnLoggedIn();
+			}
+			else {
+				AsyncHttpRequest *pReq = new AsyncHttpRequest();
+				pReq->requestType = REQUEST_GET;
+				pReq->flags = NLHRF_PERSISTENT | NLHRF_HTTP11 | NLHRF_REMOVEHOST | NLHRF_SMARTREMOVEHOST;
+				pReq->m_pFunc = &CVkProto::OnOAuthAuthorize;
+				pReq->AddHeader("Referer", m_prevUrl);
+				pReq->Redirect(reply);
+				if (pReq->szUrl) {
+					if ( strstr(pReq->szUrl, "login.vk.com"))
+						pReq->AddHeader("Cookie", loginCookie);
+					m_prevUrl = pReq->szUrl;
+				}
+				PushAsyncHttpRequest(pReq);
+			}
+		}
+		else ConnectionFailed(LOGINERR_NOSERVER);
+		return;
+	}
+
+	if (reply->resultCode != 200) { // something went wrong
+LBL_NoForm:
+		ConnectionFailed(LOGINERR_NOSERVER);
+		return;
+	}
+
+	if ( strstr(reply->pData, ptrA( mir_utf8encodecp("неверный логин или пароль", 1251)))) {
+		ConnectionFailed(LOGINERR_WRONGPASSWORD);
+		return;
+	}
+
+	// Application requests access to user's account
+	if ( !strstr(reply->pData, "form method=\"post\""))
+		goto LBL_NoForm;
+
+	CMStringA szAction;
+	CMStringA szBody = AutoFillForm(reply->pData, szAction);
+	if (szAction.IsEmpty() || szBody.IsEmpty())
+		goto LBL_NoForm;
+
+	AsyncHttpRequest *pReq = new AsyncHttpRequest();
+	pReq->requestType = REQUEST_POST;
+	pReq->flags = NLHRF_PERSISTENT | NLHRF_HTTP11 | NLHRF_REMOVEHOST | NLHRF_SMARTREMOVEHOST;
+	pReq->pData = mir_strdup(szBody);
+	pReq->dataLength = szBody.GetLength();
+	pReq->szUrl = mir_strdup(szAction); m_prevUrl = pReq->szUrl;
+	pReq->m_pFunc = &CVkProto::OnOAuthAuthorize;
+	pReq->AddHeader("Content-Type", "application/x-www-form-urlencoded");
+	pReq->Redirect(reply);
+	PushAsyncHttpRequest(pReq);
 }
 
 void CVkProto::RequestMyInfo()
 {
 }
-
-- 
cgit v1.2.3