#include "common.h" WORD CToxProto::GetContactStatus(MCONTACT hContact) { return getWord(hContact, "Status", ID_STATUS_OFFLINE); } void CToxProto::SetContactStatus(MCONTACT hContact, WORD status) { WORD oldStatus = GetContactStatus(hContact); if (oldStatus != status) { setWord(hContact, "Status", status); } } void CToxProto::SetAllContactsStatus(WORD status) { for (MCONTACT hContact = db_find_first(m_szModuleName); hContact; hContact = db_find_next(hContact, m_szModuleName)) { SetContactStatus(hContact, status); } } MCONTACT CToxProto::GetContactFromAuthEvent(HANDLE hEvent) { DWORD body[3]; DBEVENTINFO dbei = { sizeof(DBEVENTINFO) }; dbei.cbBlob = sizeof(DWORD)* 2; dbei.pBlob = (PBYTE)&body; if (db_event_get(hEvent, &dbei)) return INVALID_CONTACT_ID; if (dbei.eventType != EVENTTYPE_AUTHREQUEST) return INVALID_CONTACT_ID; if (strcmp(dbei.szModule, m_szModuleName) != 0) return INVALID_CONTACT_ID; return DbGetAuthEventContact(&dbei); } bool CToxProto::IsMe(const std::string &id) { std::string ownId = getStringA(NULL, TOX_SETTINGS_ID); return strnicmp(id.c_str(), ownId.c_str(), TOX_CLIENT_ID_SIZE) == 0; } MCONTACT CToxProto::FindContact(const std::string &id) { MCONTACT hContact = NULL; for (hContact = db_find_first(m_szModuleName); hContact; hContact = db_find_next(hContact, m_szModuleName)) { DBVARIANT dbv; if (!db_get(hContact, m_szModuleName, TOX_SETTINGS_ID, &dbv)) { std::string clientId; // temporary code for contact id conversion if (dbv.type == DBVT_BLOB) { std::vector pubKey(dbv.cpbVal); memcpy(&pubKey[0], dbv.pbVal, dbv.cpbVal); clientId = DataToHexString(pubKey); delSetting(hContact, TOX_SETTINGS_ID); setString(hContact, TOX_SETTINGS_ID, clientId.c_str()); } else if (dbv.type == DBVT_ASCIIZ) { clientId = dbv.pszVal; } db_free(&dbv); if (_strnicmp(id.c_str(), clientId.c_str(), TOX_CLIENT_ID_SIZE) == 0) { break; } } } return hContact; } MCONTACT CToxProto::FindContact(const int friendNumber) { std::vector clientId(TOX_CLIENT_ID_SIZE); tox_get_client_id(tox, friendNumber, clientId.data()); std::string id = DataToHexString(clientId); return FindContact(id); } MCONTACT CToxProto::AddContact(const std::string &id, const std::tstring &dnsId, bool isTemporary) { MCONTACT hContact = FindContact(id); if (!hContact) { hContact = (MCONTACT)CallService(MS_DB_CONTACT_ADD, 0, 0); CallService(MS_PROTO_ADDTOCONTACT, hContact, (LPARAM)m_szModuleName); setString(hContact, TOX_SETTINGS_ID, id.c_str()); if (!dnsId.empty()) { setTString(hContact, TOX_SETTINGS_DNS, dnsId.c_str()); } DBVARIANT dbv; if (!getTString(TOX_SETTINGS_GROUP, &dbv)) { db_set_ts(hContact, "CList", "Group", dbv.ptszVal); db_free(&dbv); } setByte(hContact, "Auth", 1); setByte(hContact, "Grant", 1); if (isTemporary) { db_set_b(hContact, "CList", "NotOnList", 1); } } return hContact; } void CToxProto::LoadFriendList() { uint32_t count = tox_count_friendlist(tox); if (count > 0) { int32_t *friends = (int32_t*)mir_alloc(count * sizeof(int32_t)); tox_get_friendlist(tox, friends, count); std::vector id(TOX_CLIENT_ID_SIZE); for (uint32_t i = 0; i < count; ++i) { tox_get_client_id(tox, friends[i], id.data()); MCONTACT hContact = AddContact(DataToHexString(id), _T("")); if (hContact) { delSetting(hContact, "Auth"); delSetting(hContact, "Grant"); int size = tox_get_name_size(tox, friends[i]); std::vector username(size); tox_get_name(tox, friends[i], &username[0]); std::string nick(username.begin(), username.end()); setWString(hContact, "Nick", ptrW(Utf8DecodeW(nick.c_str()))); uint64_t timestamp = tox_get_last_online(tox, friends[i]); if (timestamp) { setDword(hContact, "LastEventDateTS", timestamp); } } } mir_free(friends); } else { debugLogA("CToxProto::LoadContactList: your friend list is empty"); } } int CToxProto::OnContactDeleted(MCONTACT hContact, LPARAM lParam) { std::string id = getStringA(hContact, TOX_SETTINGS_ID); std::vector clientId = HexStringToData(id); uint32_t number = tox_get_friend_number(tox, clientId.data()); if (number == TOX_ERROR || tox_del_friend(tox, number) == TOX_ERROR) { return 1; } SaveToxProfile(); return 0; } void CToxProto::OnFriendRequest(Tox *tox, const uint8_t *address, const uint8_t *message, const uint16_t messageSize, void *arg) { CToxProto *proto = (CToxProto*)arg; // trim tox address to tox id std::vector clientId(address, address + TOX_CLIENT_ID_SIZE); std::string id = proto->DataToHexString(clientId); MCONTACT hContact = proto->AddContact(id, _T("")); if (!hContact) { return; } proto->delSetting(hContact, "Auth"); PROTORECVEVENT pre = { 0 }; pre.flags = PREF_UTF; pre.timestamp = time(NULL); pre.lParam = (DWORD)(sizeof(DWORD)* 2 + id.length() + messageSize + 5); /*blob is: 0(DWORD), hContact(DWORD), nick(ASCIIZ), firstName(ASCIIZ), lastName(ASCIIZ), id(ASCIIZ), reason(ASCIIZ)*/ PBYTE pBlob, pCurBlob; pCurBlob = pBlob = (PBYTE)mir_calloc(pre.lParam); *((PDWORD)pCurBlob) = 0; pCurBlob += sizeof(DWORD); *((PDWORD)pCurBlob) = (DWORD)hContact; pCurBlob += sizeof(DWORD); pCurBlob += 3; strcpy((char *)pCurBlob, id.c_str()); pCurBlob += id.length() + 1; strcpy((char *)pCurBlob, (char*)message); pre.szMessage = (char*)pBlob; ProtoChainRecv(hContact, PSR_AUTH, 0, (LPARAM)&pre); } void CToxProto::OnFriendNameChange(Tox *tox, const int number, const uint8_t *name, const uint16_t nameSize, void *arg) { CToxProto *proto = (CToxProto*)arg; MCONTACT hContact = proto->FindContact(number); if (hContact) { proto->setString(hContact, "Nick", (char*)name); } } void CToxProto::OnStatusMessageChanged(Tox *tox, const int number, const uint8_t* message, const uint16_t messageSize, void *arg) { CToxProto *proto = (CToxProto*)arg; MCONTACT hContact = proto->FindContact(number); if (hContact) { ptrW statusMessage(mir_utf8decodeW((char*)message)); db_set_ws(hContact, "CList", "StatusMsg", statusMessage); } } void CToxProto::OnUserStatusChanged(Tox *tox, int32_t number, uint8_t usertatus, void *arg) { CToxProto *proto = (CToxProto*)arg; MCONTACT hContact = proto->FindContact(number); if (hContact) { TOX_USERSTATUS userstatus = (TOX_USERSTATUS)usertatus; int status = proto->ToxToMirandaStatus(userstatus); proto->SetContactStatus(hContact, status); } } void CToxProto::OnConnectionStatusChanged(Tox *tox, const int number, const uint8_t status, void *arg) { CToxProto *proto = (CToxProto*)arg; MCONTACT hContact = proto->FindContact(number); if (hContact) { int newStatus = status ? ID_STATUS_ONLINE : ID_STATUS_OFFLINE; proto->SetContactStatus(hContact, newStatus); if (status) { proto->delSetting(hContact, "Auth"); } } } int CToxProto::OnUserInfoInit(WPARAM wParam, LPARAM lParam) { if (!CallService(MS_PROTO_ISPROTOCOLLOADED, 0, (LPARAM)m_szModuleName)) { return 0; } MCONTACT hContact = lParam; char *szProto = GetContactProto(hContact); if (szProto != NULL && !strcmp(szProto, m_szModuleName)) { OPTIONSDIALOGPAGE odp = { sizeof(odp) }; odp.flags = ODPF_TCHAR | ODPF_DONTTRANSLATE; odp.hInstance = g_hInstance; odp.dwInitParam = (LPARAM)this; odp.ptszTitle = m_tszUserName; odp.pfnDlgProc = UserInfoProc; odp.position = -2000000000; odp.pszTemplate = MAKEINTRESOURCEA(IDD_USER_INFO); UserInfo_AddPage(wParam, &odp); } return 0; } INT_PTR CToxProto::UserInfoProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam) { CToxProto *proto = (CToxProto*)GetWindowLongPtr(hwnd, GWLP_USERDATA); switch (uMsg) { case WM_INITDIALOG: TranslateDialogDefault(hwnd); { proto = (CToxProto*)lParam; SetWindowLongPtr(hwnd, GWLP_USERDATA, lParam); } break; case WM_NOTIFY: switch (((LPNMHDR)lParam)->idFrom) { case 0: switch (((LPNMHDR)lParam)->code) { case PSN_INFOCHANGED: { MCONTACT hContact = (MCONTACT)((LPPSHNOTIFY)lParam)->lParam; char *szProto = (hContact == NULL) ? szProto = proto->m_szModuleName : (char*)CallService(MS_PROTO_GETCONTACTBASEPROTO, hContact, 0); if (szProto == NULL) { break; } SetDlgItemText(hwnd, IDC_DNS_ID, ptrT(proto->getTStringA(hContact, TOX_SETTINGS_DNS))); } break; case PSN_PARAMCHANGED: SetWindowLongPtr(hwnd, GWLP_USERDATA, ((PSHNOTIFY*)lParam)->lParam); break; case PSN_APPLY: MCONTACT hContact = (MCONTACT)((LPPSHNOTIFY)lParam)->lParam; char *szProto = (hContact == NULL) ? szProto = proto->m_szModuleName : (char*)CallService(MS_PROTO_GETCONTACTBASEPROTO, hContact, 0); if (szProto == NULL) { break; } TCHAR dnsId[MAX_PATH]; GetDlgItemText(hwnd, IDC_DNS_ID, dnsId, MAX_PATH); proto->setTString(hContact, TOX_SETTINGS_DNS, dnsId); break; } break; } break; case WM_COMMAND: if ((HWND)lParam == GetFocus() && HIWORD(wParam) == EN_CHANGE) { SendMessage(GetParent(hwnd), PSM_CHANGED, 0, 0); } break; break; } return FALSE; }