1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
/*
Copyright © 2016-19 Miranda NG team
This program 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 2 of the License, or
(at your option) any later version.
This program 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.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "stdafx.h"
void CDiscordProto::Push(AsyncHttpRequest *pReq, int iTimeout)
{
pReq->timeout = iTimeout;
{
mir_cslock lck(m_csHttpQueue);
m_arHttpQueue.insert(pReq);
}
SetEvent(m_evRequestsQueue);
}
/////////////////////////////////////////////////////////////////////////////////////////
static LONG g_reqNum = 0;
AsyncHttpRequest::AsyncHttpRequest(CDiscordProto *ppro, int iRequestType, LPCSTR _url, MTHttpRequestHandler pFunc, JSONNode *pRoot)
{
if (*_url == '/') { // relative url leads to a site
m_szUrl = "https://discordapp.com/api/v6";
m_szUrl += _url;
m_bMainSite = true;
}
else {
m_szUrl = _url;
m_bMainSite = false;
}
flags = NLHRF_HTTP11 | NLHRF_REDIRECT | NLHRF_SSL;
if (ppro->m_szAccessToken != nullptr) {
AddHeader("Authorization", ppro->m_szAccessToken);
flags |= NLHRF_DUMPASTEXT | NLHRF_NODUMPHEADERS;
}
else flags |= NLHRF_NODUMPSEND;
if (pRoot != nullptr) {
ptrW text(json_write(pRoot));
pData = mir_utf8encodeW(text);
dataLength = (int)mir_strlen(pData);
}
AddHeader("Content-Type", "application/json");
m_pFunc = pFunc;
requestType = iRequestType;
m_iErrorCode = 0;
m_iReqNum = ::InterlockedIncrement(&g_reqNum);
}
/////////////////////////////////////////////////////////////////////////////////////////
void CDiscordProto::ServerThread(void*)
{
m_szAccessToken = getStringA("AccessToken");
m_hAPIConnection = nullptr;
m_bTerminated = false;
debugLogA("CDiscordProto::WorkerThread: %s", "entering");
if (m_szAccessToken != nullptr)
// try to receive a response from server
RetrieveMyInfo();
else {
if (mir_wstrlen(m_wszEmail) == 0) {
ConnectionFailed(LOGINERR_BADUSERID);
return;
}
ptrW wszPassword(getWStringA(DB_KEY_PASSWORD));
if (wszPassword == nullptr) {
ConnectionFailed(LOGINERR_WRONGPASSWORD);
return;
}
JSONNode root; root << WCHAR_PARAM("email", m_wszEmail) << WCHAR_PARAM("password", wszPassword);
Push(new AsyncHttpRequest(this, REQUEST_POST, "/auth/login", &CDiscordProto::OnReceiveToken, &root));
}
while (true) {
WaitForSingleObject(m_evRequestsQueue, 1000);
if (m_bTerminated)
break;
AsyncHttpRequest *pReq;
bool need_sleep = false;
while (true) {
{
mir_cslock lck(m_csHttpQueue);
if (m_arHttpQueue.getCount() == 0)
break;
pReq = m_arHttpQueue[0];
m_arHttpQueue.remove(0);
need_sleep = (m_arHttpQueue.getCount() > 1);
}
if (m_bTerminated)
break;
ExecuteRequest(pReq);
if (need_sleep) {
Sleep(330);
debugLogA("CDiscordProto::WorkerThread: %s", "need to sleep");
}
}
}
m_hWorkerThread = nullptr;
if (m_hAPIConnection) {
Netlib_CloseHandle(m_hAPIConnection);
m_hAPIConnection = nullptr;
}
debugLogA("CDiscordProto::WorkerThread: %s", "leaving");
}
|
