/* Copyright © 2016-18 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 . */ #include "stdafx.h" static int compareMessages(const SnowFlake *p1, const SnowFlake *p2) { return compareInt64(*p1, *p2); } static int compareRequests(const AsyncHttpRequest *p1, const AsyncHttpRequest *p2) { return p1->m_iReqNum - p2->m_iReqNum; } int compareUsers(const CDiscordUser *p1, const CDiscordUser *p2) { return compareInt64(p1->id, p2->id); } static int compareGuilds(const CDiscordGuild *p1, const CDiscordGuild *p2) { return compareInt64(p1->id, p2->id); } CDiscordProto::CDiscordProto(const char *proto_name, const wchar_t *username) : PROTO(proto_name, username), m_arHttpQueue(10, compareRequests), m_evRequestsQueue(CreateEvent(nullptr, FALSE, FALSE, nullptr)), m_wszDefaultGroup(this, DB_KEY_GROUP, DB_KEYVAL_GROUP), m_wszEmail(this, DB_KEY_EMAIL, L""), m_bHideGroupchats(this, DB_KEY_HIDECHATS, true), arGuilds(1, compareGuilds), arMarkReadQueue(1, compareUsers), arOwnMessages(1, compareMessages), arUsers(10, compareUsers) { // Services CreateProtoService(PS_CREATEACCMGRUI, &CDiscordProto::SvcCreateAccMgrUI); CreateProtoService(PS_GETAVATARINFO, &CDiscordProto::GetAvatarInfo); CreateProtoService(PS_GETAVATARCAPS, &CDiscordProto::GetAvatarCaps); CreateProtoService(PS_GETMYAVATAR, &CDiscordProto::GetMyAvatar); CreateProtoService(PS_SETMYAVATAR, &CDiscordProto::SetMyAvatar); // Events HookProtoEvent(ME_OPT_INITIALISE, &CDiscordProto::OnOptionsInit); HookProtoEvent(ME_DB_EVENT_MARKED_READ, &CDiscordProto::OnDbEventRead); // database db_set_resident(m_szModuleName, "XStatusMsg"); // Network initialization CMStringW descr; NETLIBUSER nlu = {}; nlu.szSettingsModule = m_szModuleName; nlu.flags = NUF_OUTGOING | NUF_HTTPCONNS | NUF_UNICODE; descr.Format(TranslateT("%s server connection"), m_tszUserName); nlu.szDescriptiveName.w = descr.GetBuffer(); m_hNetlibUser = Netlib_RegisterUser(&nlu); CMStringA module(FORMAT, "%s.Gateway", m_szModuleName); nlu.szSettingsModule = module.GetBuffer(); nlu.flags = NUF_OUTGOING | NUF_UNICODE; descr.Format(TranslateT("%s gateway connection"), m_tszUserName); nlu.szDescriptiveName.w = descr.GetBuffer(); m_hGatewayNetlibUser = Netlib_RegisterUser(&nlu); } CDiscordProto::~CDiscordProto() { debugLogA("CDiscordProto::~CDiscordProto"); Netlib_CloseHandle(m_hNetlibUser); m_hNetlibUser = nullptr; m_arHttpQueue.destroy(); ::CloseHandle(m_evRequestsQueue); } ///////////////////////////////////////////////////////////////////////////////////////// void CDiscordProto::OnModulesLoaded() { // Fill users list for (auto &hContact : AccContacts()) { CDiscordUser *pNew = new CDiscordUser(getId(hContact, DB_KEY_ID)); pNew->hContact = hContact; pNew->channelId = getId(hContact, DB_KEY_CHANNELID); pNew->lastMsg.id = getId(hContact, DB_KEY_LASTMSGID); pNew->wszUsername = ptrW(getWStringA(hContact, DB_KEY_NICK)); pNew->iDiscriminator = getDword(hContact, DB_KEY_DISCR); arUsers.insert(pNew); } GCREGISTER gcr = {}; gcr.dwFlags = GC_TYPNOTIF | GC_CHANMGR; gcr.ptszDispName = m_tszUserName; gcr.pszModule = m_szModuleName; Chat_Register(&gcr); // Clist Clist_GroupCreate(0, m_wszDefaultGroup); HookProtoEvent(ME_GC_EVENT, &CDiscordProto::GroupchatEventHook); HookProtoEvent(ME_GC_BUILDMENU, &CDiscordProto::GroupchatMenuHook); InitMenus(); } void CDiscordProto::OnShutdown() { debugLogA("CDiscordProto::OnPreShutdown"); m_bTerminated = true; SetEvent(m_evRequestsQueue); if (m_hGatewayConnection) Netlib_Shutdown(m_hGatewayConnection); } ///////////////////////////////////////////////////////////////////////////////////////// INT_PTR CDiscordProto::GetCaps(int type, MCONTACT) { switch (type) { case PFLAGNUM_1: return PF1_IM | PF1_MODEMSGRECV | PF1_SERVERCLIST | PF1_BASICSEARCH | PF1_EXTSEARCH | PF1_ADDSEARCHRES | PF1_FILESEND; case PFLAGNUM_2: return PF2_ONLINE | PF2_SHORTAWAY | PF2_LONGAWAY | PF2_HEAVYDND | PF2_INVISIBLE; case PFLAGNUM_3: return PF2_ONLINE | PF2_LONGAWAY | PF2_HEAVYDND | PF2_INVISIBLE; case PFLAGNUM_4: return PF4_FORCEADDED | PF4_FORCEAUTH | PF4_NOCUSTOMAUTH | PF4_NOAUTHDENYREASON | PF4_SUPPORTTYPING | PF4_SUPPORTIDLE | PF4_AVATARS | PF4_IMSENDOFFLINE; case PFLAG_UNIQUEIDTEXT: return (INT_PTR)Translate("User ID"); } return 0; } int CDiscordProto::SetStatus(int iNewStatus) { debugLogA("CDiscordProto::SetStatus iNewStatus = %d, m_iStatus = %d, m_iDesiredStatus = %d m_hWorkerThread = %p", iNewStatus, m_iStatus, m_iDesiredStatus, m_hWorkerThread); if (iNewStatus == m_iStatus) return 0; m_iDesiredStatus = iNewStatus; int iOldStatus = m_iStatus; // go offline if (iNewStatus == ID_STATUS_OFFLINE) { if (m_bOnline) { SetServerStatus(ID_STATUS_OFFLINE); ShutdownSession(); } m_iStatus = m_iDesiredStatus; setAllContactStatuses(ID_STATUS_OFFLINE, true); ProtoBroadcastAck(0, ACKTYPE_STATUS, ACKRESULT_SUCCESS, (HANDLE)iOldStatus, m_iStatus); } // not logged in? come on else if (m_hWorkerThread == nullptr && !IsStatusConnecting(m_iStatus)) { m_iStatus = ID_STATUS_CONNECTING; ProtoBroadcastAck(0, ACKTYPE_STATUS, ACKRESULT_SUCCESS, (HANDLE)iOldStatus, m_iStatus); m_hWorkerThread = ForkThreadEx(&CDiscordProto::ServerThread, nullptr, nullptr); } else if (m_bOnline) { debugLogA("setting server online status to %d", iNewStatus); SetServerStatus(iNewStatus); } return 0; } ///////////////////////////////////////////////////////////////////////////////////////// static INT_PTR CALLBACK AdvancedSearchDlgProc(HWND hwndDlg, UINT msg, WPARAM wParam, LPARAM) { switch (msg) { case WM_INITDIALOG: TranslateDialogDefault(hwndDlg); SetFocus(GetDlgItem(hwndDlg, IDC_NICK)); return TRUE; case WM_COMMAND: if (HIWORD(wParam) == EN_SETFOCUS) PostMessage(GetParent(hwndDlg), WM_COMMAND, MAKEWPARAM(0, EN_SETFOCUS), (LPARAM)hwndDlg); } return FALSE; } HWND CDiscordProto::CreateExtendedSearchUI(HWND hwndParent) { if (hwndParent) return CreateDialogParam(g_plugin.getInst(), MAKEINTRESOURCE(IDD_EXTSEARCH), hwndParent, AdvancedSearchDlgProc, 0); return nullptr; } ///////////////////////////////////////////////////////////////////////////////////////// void CDiscordProto::SearchThread(void *param) { Sleep(100); PROTOSEARCHRESULT psr = { 0 }; psr.cbSize = sizeof(psr); psr.flags = PSR_UNICODE; psr.nick.w = (wchar_t*)param; psr.firstName.w = L""; psr.lastName.w = L""; psr.id.w = L""; ProtoBroadcastAck(0, ACKTYPE_SEARCH, ACKRESULT_DATA, (HANDLE)param, (LPARAM)&psr); ProtoBroadcastAck(0, ACKTYPE_SEARCH, ACKRESULT_SUCCESS, (HANDLE)param, 0); mir_free(param); } HWND CDiscordProto::SearchAdvanced(HWND hwndDlg) { if (!m_bOnline || !IsWindow(hwndDlg)) return nullptr; wchar_t wszNick[200]; GetDlgItemTextW(hwndDlg, IDC_NICK, wszNick, _countof(wszNick)); if (wszNick[0] == 0) // empty string? reject return nullptr; wchar_t *p = wcschr(wszNick, '#'); if (p == nullptr) // wrong user id return nullptr; p = mir_wstrdup(wszNick); ForkThread(&CDiscordProto::SearchThread, p); return (HWND)p; } HANDLE CDiscordProto::SearchBasic(const wchar_t *wszId) { if (!m_bOnline) return nullptr; CMStringA szUrl = "/users/"; szUrl.AppendFormat(ptrA(mir_utf8encodeW(wszId))); AsyncHttpRequest *pReq = new AsyncHttpRequest(this, REQUEST_GET, szUrl, &CDiscordProto::OnReceiveMyInfo); pReq->pUserInfo = (void*)-1; Push(pReq); return (HANDLE)1; // Success } //////////////////////////////////////////////////////////////////////////////////////// // Authorization int CDiscordProto::AuthRequest(MCONTACT hContact, const wchar_t*) { ptrW wszUsername(getWStringA(hContact, DB_KEY_NICK)); int iDiscriminator(getDword(hContact, DB_KEY_DISCR, -1)); if (wszUsername == nullptr || iDiscriminator == -1) return 1; // error JSONNode root; root << WCHAR_PARAM("username", wszUsername) << INT_PARAM("discriminator", iDiscriminator); AsyncHttpRequest *pReq = new AsyncHttpRequest(this, REQUEST_POST, "/users/@me/relationships", nullptr, &root); pReq->pUserInfo = (void*)hContact; Push(pReq); return 0; } int CDiscordProto::AuthRecv(MCONTACT, PROTORECVEVENT *pre) { return Proto_AuthRecv(m_szModuleName, pre); } int CDiscordProto::Authorize(MEVENT hDbEvent) { DBEVENTINFO dbei = {}; if ((dbei.cbBlob = db_event_getBlobSize(hDbEvent)) == (DWORD)(-1)) return 1; if ((dbei.pBlob = (PBYTE)alloca(dbei.cbBlob)) == nullptr) return 1; if (db_event_get(hDbEvent, &dbei)) return 1; if (dbei.eventType != EVENTTYPE_AUTHREQUEST) return 1; if (mir_strcmp(dbei.szModule, m_szModuleName)) return 1; MCONTACT hContact = DbGetAuthEventContact(&dbei); CMStringA szUrl(FORMAT, "/users/@me/relationships/%lld", getId(hContact, DB_KEY_ID)); Push(new AsyncHttpRequest(this, REQUEST_PUT, szUrl, nullptr)); return 0; } int CDiscordProto::AuthDeny(MEVENT hDbEvent, const wchar_t*) { DBEVENTINFO dbei = {}; if ((dbei.cbBlob = db_event_getBlobSize(hDbEvent)) == (DWORD)(-1)) return 1; if ((dbei.pBlob = (PBYTE)alloca(dbei.cbBlob)) == nullptr) return 1; if (db_event_get(hDbEvent, &dbei)) return 1; if (dbei.eventType != EVENTTYPE_AUTHREQUEST) return 1; if (mir_strcmp(dbei.szModule, m_szModuleName)) return 1; MCONTACT hContact = DbGetAuthEventContact(&dbei); RemoveFriend(getId(hContact, DB_KEY_ID)); return 0; } //////////////////////////////////////////////////////////////////////////////////////// MCONTACT CDiscordProto::AddToList(int flags, PROTOSEARCHRESULT *psr) { if (mir_wstrlen(psr->nick.w) == 0) return 0; wchar_t *p = wcschr(psr->nick.w, '#'); if (p == nullptr) return 0; MCONTACT hContact = db_add_contact(); Proto_AddToContact(hContact, m_szModuleName); if (flags & PALF_TEMPORARY) db_set_b(hContact, "CList", "NotOnList", 1); *p = 0; CDiscordUser *pUser = new CDiscordUser(0); pUser->hContact = hContact; pUser->wszUsername = psr->nick.w; pUser->iDiscriminator = _wtoi(p + 1); *p = '#'; if (mir_wstrlen(psr->id.w)) { pUser->id = _wtoi64(psr->id.w); setId(hContact, DB_KEY_ID, pUser->id); } db_set_ws(hContact, "CList", "Group", m_wszDefaultGroup); setWString(hContact, DB_KEY_NICK, pUser->wszUsername); setDword(hContact, DB_KEY_DISCR, pUser->iDiscriminator); arUsers.insert(pUser); return hContact; } //////////////////////////////////////////////////////////////////////////////////////// // RecvMsg int CDiscordProto::RecvMsg(MCONTACT hContact, PROTORECVEVENT *evt) { T2Utf szResUtf((const wchar_t*)evt->lParam); evt->pCustomData = (char*)szResUtf; evt->cbCustomDataSize = (DWORD)mir_strlen(szResUtf); return CSuper::RecvMsg(hContact, evt); } //////////////////////////////////////////////////////////////////////////////////////// // SendMsg void __cdecl CDiscordProto::SendMessageAckThread(void *param) { Sleep(100); ProtoBroadcastAck((UINT_PTR)param, ACKTYPE_MESSAGE, ACKRESULT_FAILED, (HANDLE)1, (LPARAM)Translate("Protocol is offline or user isn't authorized yet")); } int CDiscordProto::SendMsg(MCONTACT hContact, int /*flags*/, const char *pszSrc) { if (!m_bOnline) { ForkThread(&CDiscordProto::SendMessageAckThread, (void*)hContact); return 1; } ptrW wszText(mir_utf8decodeW(pszSrc)); if (wszText == nullptr) return 0; CDiscordUser *pUser = FindUser(getId(hContact, DB_KEY_ID)); if (pUser == nullptr || pUser->id == 0) return 0; // no channel - we need to create one if (pUser->channelId == 0) { JSONNode list(JSON_ARRAY); list.set_name("recipients"); list << INT64_PARAM("", pUser->id); JSONNode body; body << list; CMStringA szUrl(FORMAT, "/users/%lld/channels", m_ownId); // theoretically we get the same data from the gateway thread, but there could be a delay // so we bind data analysis to the http packet reply mir_cslock lck(m_csHttpQueue); ExecuteRequest(new AsyncHttpRequest(this, REQUEST_POST, szUrl, &CDiscordProto::OnReceiveCreateChannel, &body)); if (pUser->channelId == 0) return 0; } // we generate a random 64-bit integer and pass it to the server // to distinguish our own messages from these generated by another clients SnowFlake nonce; Utils_GetRandom(&nonce, sizeof(nonce)); JSONNode body; body << WCHAR_PARAM("content", wszText) << INT64_PARAM("nonce", nonce); arOwnMessages.insert(new SnowFlake(nonce)); CMStringA szUrl(FORMAT, "/channels/%lld/messages", pUser->channelId); AsyncHttpRequest *pReq = new AsyncHttpRequest(this, REQUEST_POST, szUrl, &CDiscordProto::OnReceiveMessage, &body); pReq->pUserInfo = (void*)hContact; Push(pReq); return pReq->m_iReqNum; } ///////////////////////////////////////////////////////////////////////////////////////// int CDiscordProto::UserIsTyping(MCONTACT hContact, int type) { if (type == PROTOTYPE_SELFTYPING_ON) { CMStringA szUrl(FORMAT, "/channels/%lld/typing", getId(hContact, DB_KEY_CHANNELID)); Push(new AsyncHttpRequest(this, REQUEST_POST, szUrl, nullptr)); } return 0; } ///////////////////////////////////////////////////////////////////////////////////////// void CDiscordProto::MarkReadTimerProc(HWND hwnd, UINT, UINT_PTR id, DWORD) { CDiscordProto *ppro = (CDiscordProto*)(id - 1); mir_cslock lck(ppro->csMarkReadQueue); while (ppro->arMarkReadQueue.getCount()) { CDiscordUser *pUser = ppro->arMarkReadQueue[0]; CMStringA szUrl(FORMAT, "/channels/%lld/messages/%lld/ack", pUser->channelId, pUser->lastMsg.id); ppro->Push(new AsyncHttpRequest(ppro, REQUEST_POST, szUrl, nullptr)); ppro->arMarkReadQueue.remove(0); } KillTimer(hwnd, id); } int CDiscordProto::OnDbEventRead(WPARAM, LPARAM hDbEvent) { MCONTACT hContact = db_event_getContact(hDbEvent); if (!hContact) return 0; // filter out only events of my protocol const char *szProto = GetContactProto(hContact); if (mir_strcmp(szProto, m_szModuleName)) return 0; if (m_bOnline) { SetTimer(g_hwndHeartbeat, UINT_PTR(this) + 1, 200, &CDiscordProto::MarkReadTimerProc); CDiscordUser *pUser = FindUser(getId(hContact, DB_KEY_ID)); if (pUser != nullptr) { mir_cslock lck(csMarkReadQueue); if (arMarkReadQueue.indexOf(pUser) == -1) arMarkReadQueue.insert(pUser); } } return 0; } ///////////////////////////////////////////////////////////////////////////////////////// void CDiscordProto::OnContactDeleted(MCONTACT hContact) { CDiscordUser *pUser = FindUser(getId(hContact, DB_KEY_ID)); if (pUser == nullptr || !m_bOnline) return; if (pUser->channelId) Push(new AsyncHttpRequest(this, REQUEST_DELETE, CMStringA(FORMAT, "/channels/%lld", pUser->channelId), nullptr)); if (pUser->id) RemoveFriend(pUser->id); } ///////////////////////////////////////////////////////////////////////////////////////// struct SendFileThreadParam { MCONTACT hContact; CMStringW wszDescr, wszFileName; SendFileThreadParam(MCONTACT _p1, LPCWSTR _p2, LPCWSTR _p3) : hContact(_p1), wszFileName(_p2), wszDescr(_p3) {} }; void CDiscordProto::SendFileThread(void *param) { SendFileThreadParam *p = (SendFileThreadParam*)param; FILE *in = _wfopen(p->wszFileName, L"rb"); if (in == nullptr) { debugLogA("cannot open file %S for reading", p->wszFileName.c_str()); LBL_Error: ProtoBroadcastAck(p->hContact, ACKTYPE_FILE, ACKRESULT_FAILED, param); delete p; return; } ProtoBroadcastAck(p->hContact, ACKTYPE_FILE, ACKRESULT_INITIALISING, param); char szRandom[16], szRandomText[33]; Utils_GetRandom(szRandom, _countof(szRandom)); bin2hex(szRandom, _countof(szRandom), szRandomText); CMStringA szBoundary(FORMAT, "----Boundary%s", szRandomText); CMStringA szUrl(FORMAT, "/channels/%lld/messages", getId(p->hContact, DB_KEY_CHANNELID)); AsyncHttpRequest *pReq = new AsyncHttpRequest(this, REQUEST_POST, szUrl, &CDiscordProto::OnReceiveFile); pReq->AddHeader("Content-Type", CMStringA("multipart/form-data; boundary=" + szBoundary)); pReq->AddHeader("Accept", "*/*"); szBoundary.Insert(0, "--"); CMStringA szBody; szBody.Append(szBoundary + "\r\n"); szBody.Append("Content-Disposition: form-data; name=\"content\"\r\n\r\n"); szBody.Append(ptrA(mir_utf8encodeW(p->wszDescr))); szBody.Append("\r\n"); szBody.Append(szBoundary + "\r\n"); szBody.Append("Content-Disposition: form-data; name=\"tts\"\r\n\r\nfalse\r\n"); wchar_t *pFileName = wcsrchr(p->wszFileName.GetBuffer(), '\\'); if (pFileName != nullptr) pFileName++; else pFileName = p->wszFileName.GetBuffer(); szBody.Append(szBoundary + "\r\n"); szBody.AppendFormat("Content-Disposition: form-data; name=\"file\"; filename=\"%s\"\r\n", ptrA(mir_utf8encodeW(pFileName))); szBody.AppendFormat("Content-Type: %S\r\n", ProtoGetAvatarMimeType(ProtoGetAvatarFileFormat(p->wszFileName))); szBody.Append("\r\n"); size_t cbBytes = filelength(fileno(in)); szBoundary.Insert(0, "\r\n"); szBoundary.Append("--\r\n"); pReq->dataLength = int(szBody.GetLength() + szBoundary.GetLength() + cbBytes); pReq->pData = (char*)mir_alloc(pReq->dataLength+1); memcpy(pReq->pData, szBody.c_str(), szBody.GetLength()); size_t cbRead = fread(pReq->pData + szBody.GetLength(), 1, cbBytes, in); fclose(in); if (cbBytes != cbRead) { debugLogA("cannot read file %S: %d bytes read instead of %d", p->wszFileName.c_str(), cbRead, cbBytes); delete pReq; goto LBL_Error; } memcpy(pReq->pData + szBody.GetLength() + cbBytes, szBoundary, szBoundary.GetLength()); pReq->pUserInfo = p; Push(pReq); ProtoBroadcastAck(p->hContact, ACKTYPE_FILE, ACKRESULT_CONNECTED, param); } void CDiscordProto::OnReceiveFile(NETLIBHTTPREQUEST *pReply, AsyncHttpRequest *pReq) { SendFileThreadParam *p = (SendFileThreadParam*)pReq->pUserInfo; if (pReply->resultCode != 200) { ProtoBroadcastAck(p->hContact, ACKTYPE_FILE, ACKRESULT_FAILED, p); debugLogA("CDiscordProto::SendFile failed: %d", pReply->resultCode); } else { ProtoBroadcastAck(p->hContact, ACKTYPE_FILE, ACKRESULT_SUCCESS, p); debugLogA("CDiscordProto::SendFile succeeded"); } delete p; } HANDLE CDiscordProto::SendFile(MCONTACT hContact, const wchar_t *szDescription, wchar_t **ppszFiles) { SnowFlake id = getId(hContact, DB_KEY_CHANNELID); if (id == 0) return nullptr; // we don't wanna block the main thread, right? SendFileThreadParam *param = new SendFileThreadParam(hContact, ppszFiles[0], szDescription); ForkThread(&CDiscordProto::SendFileThread, param); return param; }