// Copyright © 2010-2012 sss // // 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, write to the Free Software // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. #include "stdafx.h" #pragma comment(lib, "shlwapi.lib") extern HFONT bold_font; extern bool bAutoExchange; void ShowFirstRunDialog(); HWND hwndFirstRun = nullptr, hwndSetDirs = nullptr, hwndNewKey = nullptr, hwndKeyGen = nullptr, hwndSelectExistingKey = nullptr; //int itemnum = 0; HWND hwndList_g = nullptr; BOOL CheckStateStoreDB(HWND hwndDlg, int idCtrl, const char* szSetting) { BOOL state = IsDlgButtonChecked(hwndDlg, idCtrl); db_set_b(NULL, szGPGModuleName, szSetting, (BYTE)state); return state; } wchar_t key_id_global[17] = { 0 }; extern HINSTANCE hInst; bool gpg_validate_paths(wchar_t *gpg_bin_path, wchar_t *gpg_home_path) { wstring tmp = gpg_bin_path; if (!tmp.empty()) { wchar_t mir_path[MAX_PATH]; PathToAbsoluteW(L"\\", mir_path); SetCurrentDirectoryW(mir_path); if (!boost::filesystem::exists(tmp)) { MessageBox(nullptr, TranslateT("GPG binary does not exist.\nPlease choose another location"), TranslateT("Warning"), MB_OK); return false; } } else { MessageBox(nullptr, TranslateT("Please choose GPG binary location"), TranslateT("Warning"), MB_OK); return false; } { bool bad_version = false; db_set_ws(NULL, szGPGModuleName, "szGpgBinPath", tmp.c_str()); string out; DWORD code; std::vector cmd; cmd.push_back(L"--version"); gpg_execution_params params(cmd); pxResult result; params.out = &out; params.code = &code; params.result = &result; gpg_valid = true; gpg_launcher(params); gpg_valid = false; db_unset(NULL, szGPGModuleName, "szGpgBinPath"); string::size_type p1 = out.find("(GnuPG) "); if (p1 != string::npos) { p1 += mir_strlen("(GnuPG) "); if (out[p1] != '1') bad_version = true; } else { bad_version = false; MessageBox(nullptr, TranslateT("This is not GnuPG binary!\nIt is recommended that you use GnuPG v1.x.x with this plugin."), TranslateT("Warning"), MB_OK); return false; } if (bad_version) MessageBox(nullptr, TranslateT("Unsupported GnuPG version found, use at you own risk!\nIt is recommended that you use GnuPG v1.x.x with this plugin."), TranslateT("Warning"), MB_OK); } tmp = gpg_home_path; if (tmp[tmp.length()] == '\\') tmp.erase(tmp.length()); if (tmp.empty()) { MessageBox(nullptr, TranslateT("Please set keyring's home directory"), TranslateT("Warning"), MB_OK); return false; } { wchar_t *path = UniGetContactSettingUtf(NULL, szGPGModuleName, "szHomePath", L""); DWORD dwFileAttr = GetFileAttributes(path); if (dwFileAttr != INVALID_FILE_ATTRIBUTES) { dwFileAttr &= ~FILE_ATTRIBUTE_READONLY; SetFileAttributes(path, dwFileAttr); } mir_free(path); } return true; } void gpg_save_paths(wchar_t *gpg_bin_path, wchar_t *gpg_home_path) { db_set_ws(NULL, szGPGModuleName, "szGpgBinPath", gpg_bin_path); db_set_ws(NULL, szGPGModuleName, "szHomePath", gpg_home_path); } bool gpg_use_new_random_key(char *account_name = Translate("Default"), wchar_t *gpg_bin_path = nullptr, wchar_t *gpg_home_dir = nullptr) { if (gpg_bin_path && gpg_home_dir) gpg_save_paths(gpg_bin_path, gpg_home_dir); { wstring path; { // generating key file wchar_t *tmp = nullptr; if (gpg_home_dir) tmp = gpg_home_dir; else tmp = UniGetContactSettingUtf(NULL, szGPGModuleName, "szHomePath", L""); path = tmp; if(!gpg_home_dir) mir_free(tmp); path.append(L"\\new_key"); wfstream f(path.c_str(), std::ios::out); if (!f.is_open()) { MessageBox(nullptr, TranslateT("Failed to open file"), TranslateT("Error"), MB_OK); return false; } f << "Key-Type: RSA"; f << "\n"; f << "Key-Length: 4096"; f << "\n"; f << "Subkey-Type: RSA"; f << "\n"; f << "Name-Real: "; f << get_random(6).c_str(); f << "\n"; f << "Name-Email: "; f << get_random(5).c_str(); f << "@"; f << get_random(5).c_str(); f << "."; f << get_random(3).c_str(); f << "\n"; f.close(); } { // gpg execution DWORD code; string out; std::vector cmd; cmd.push_back(L"--batch"); cmd.push_back(L"--yes"); cmd.push_back(L"--gen-key"); cmd.push_back(path); gpg_execution_params params(cmd); pxResult result; params.out = &out; params.code = &code; params.result = &result; if (!gpg_launcher(params, boost::posix_time::minutes(10))) return false; if (result == pxNotFound) return false; boost::filesystem::remove(path); string::size_type p1 = 0; if ((p1 = out.find("key ")) != string::npos) path = toUTF16(out.substr(p1 + 4, 8)); else path.clear(); } if (!path.empty()) { string out; DWORD code; std::vector cmd; cmd.push_back(L"--batch"); cmd.push_back(L"-a"); cmd.push_back(L"--export"); cmd.push_back(path); gpg_execution_params params(cmd); pxResult result; params.out = &out; params.code = &code; params.result = &result; if (!gpg_launcher(params)) { return false; } if (result == pxNotFound) return false; string::size_type s = 0; while ((s = out.find("\r", s)) != string::npos) { out.erase(s, 1); } { if (!mir_strcmp(account_name, Translate("Default"))) { db_set_s(NULL, szGPGModuleName, "GPGPubKey", out.c_str()); db_set_ws(NULL, szGPGModuleName, "KeyID", path.c_str()); } else { std::string acc_str = account_name; acc_str += "_GPGPubKey"; db_set_s(NULL, szGPGModuleName, acc_str.c_str(), out.c_str()); acc_str = account_name; acc_str += "_KeyID"; db_set_ws(NULL, szGPGModuleName, acc_str.c_str(), path.c_str()); } } } } return true; } class CDlgFirstRun : public CDlgBase { public: CDlgFirstRun() : CDlgBase(hInst, IDD_FIRST_RUN), list_KEY_LIST(this, IDC_KEY_LIST), btn_COPY_PUBKEY(this, IDC_COPY_PUBKEY), btn_EXPORT_PRIVATE(this, IDC_EXPORT_PRIVATE), btn_CHANGE_PASSWD(this, IDC_CHANGE_PASSWD), btn_GENERATE_RANDOM(this, IDC_GENERATE_RANDOM), btn_GENERATE_KEY(this, IDC_GENERATE_KEY), btn_OTHER(this, IDC_OTHER), btn_DELETE_KEY(this, IDC_DELETE_KEY), btn_OK(this, ID_OK), edit_KEY_PASSWORD(this, IDC_KEY_PASSWORD), combo_ACCOUNT(this, IDC_ACCOUNT), lbl_KEY_ID(this, IDC_KEY_ID), lbl_GENERATING_KEY(this, IDC_GENERATING_KEY) { fp[0] = 0; hwndList_g = list_KEY_LIST.GetHwnd(); btn_COPY_PUBKEY.OnClick = Callback(this, &CDlgFirstRun::onClick_COPY_PUBKEY); btn_EXPORT_PRIVATE.OnClick = Callback(this, &CDlgFirstRun::onClick_EXPORT_PRIVATE); btn_CHANGE_PASSWD.OnClick = Callback(this, &CDlgFirstRun::onClick_CHANGE_PASSWD); btn_GENERATE_RANDOM.OnClick = Callback(this, &CDlgFirstRun::onClick_GENERATE_RANDOM); btn_GENERATE_KEY.OnClick = Callback(this, &CDlgFirstRun::onClick_GENERATE_KEY); btn_OTHER.OnClick = Callback(this, &CDlgFirstRun::onClick_OTHER); btn_DELETE_KEY.OnClick = Callback(this, &CDlgFirstRun::onClick_DELETE_KEY); btn_OK.OnClick = Callback(this, &CDlgFirstRun::onClick_OK); } virtual void OnInitDialog() override { SetWindowPos(m_hwnd, nullptr, firstrun_rect.left, firstrun_rect.top, 0, 0, SWP_NOSIZE | SWP_SHOWWINDOW); SetCaption(TranslateT("Set own key")); btn_COPY_PUBKEY.Disable(); btn_EXPORT_PRIVATE.Disable(); btn_CHANGE_PASSWD.Disable(); list_KEY_LIST.AddColumn(0, TranslateT("Key ID"), 50); list_KEY_LIST.AddColumn(1, TranslateT("Email"), 30); list_KEY_LIST.AddColumn(2, TranslateT("Name"), 250); list_KEY_LIST.AddColumn(3, TranslateT("Creation date"), 30); list_KEY_LIST.AddColumn(4, TranslateT("Expire date"), 30); list_KEY_LIST.AddColumn(5, TranslateT("Key length"), 30); list_KEY_LIST.AddColumn(6, TranslateT("Accounts"), 30); list_KEY_LIST.SetExtendedListViewStyle(LVS_EX_FULLROWSELECT | LVS_EX_SINGLEROW); refresh_key_list(); { combo_ACCOUNT.AddString(TranslateT("Default")); int count = 0; PROTOACCOUNT **accounts; Proto_EnumAccounts(&count, &accounts); for (int n = 0; n < count; n++) { if (StriStr(accounts[n]->szModuleName, "metacontacts")) continue; if (StriStr(accounts[n]->szModuleName, "weather")) continue; std::string acc = toUTF8(accounts[n]->tszAccountName); acc += "("; acc += accounts[n]->szModuleName; acc += ")"; //acc += "_KeyID"; combo_ACCOUNT.AddStringA(acc.c_str()); } combo_ACCOUNT.SelectString(TranslateT("Default")); string keyinfo = Translate("key ID"); keyinfo += ": "; char *keyid = UniGetContactSettingUtf(NULL, szGPGModuleName, "KeyID", ""); keyinfo += (mir_strlen(keyid) > 0) ? keyid : Translate("not set"); mir_free(keyid); lbl_KEY_ID.SetTextA(keyinfo.c_str()); } combo_ACCOUNT.OnChange = Callback(this, &CDlgFirstRun::onChange_ACCOUNT); list_KEY_LIST.OnItemChanged = Callback(this, &CDlgFirstRun::onChange_KEY_LIST); } void onClick_COPY_PUBKEY(CCtrlButton*) { int i = list_KEY_LIST.GetSelectionMark(); if (i == -1) return; if (OpenClipboard(m_hwnd)) { list_KEY_LIST.GetItemText(i, 0, fp, _countof(fp)); string out; DWORD code; std::vector cmd; cmd.push_back(L"--batch"); cmd.push_back(L"-a"); cmd.push_back(L"--export"); cmd.push_back(fp); gpg_execution_params params(cmd); pxResult result; params.out = &out; params.code = &code; params.result = &result; if (!gpg_launcher(params)) return; if (result == pxNotFound) return; boost::algorithm::erase_all(out, "\r"); HGLOBAL hMem = GlobalAlloc(GMEM_MOVEABLE, out.size() + 1); if (!hMem) { MessageBox(nullptr, TranslateT("Failed to allocate memory"), TranslateT("Error"), MB_OK); return; } char *szKey = (char*)GlobalLock(hMem); if (!szKey) { wchar_t msg[64]; mir_snwprintf(msg, TranslateT("Failed to lock memory with error %d"), GetLastError()); MessageBox(nullptr, msg, TranslateT("Error"), MB_OK); GlobalFree(hMem); } memcpy(szKey, out.c_str(), out.size()); szKey[out.size()] = '\0'; EmptyClipboard(); GlobalUnlock(hMem); if (!SetClipboardData(CF_OEMTEXT, hMem)) { GlobalFree(hMem); wchar_t msg[64]; mir_snwprintf(msg, TranslateT("Failed write to clipboard with error %d"), GetLastError()); MessageBox(nullptr, msg, TranslateT("Error"), MB_OK); } CloseClipboard(); } } void onClick_EXPORT_PRIVATE(CCtrlButton*) { { int i = list_KEY_LIST.GetSelectionMark(); if (i == -1) return; wchar_t *p = GetFilePath(L"Choose file to export key", L"*", L"Any file", true); if (!p || !p[0]) { delete[] p; //TODO: handle error return; } char *path = mir_u2a(p); delete[] p; std::ofstream file; file.open(path, std::ios::trunc | std::ios::out); mir_free(path); if (!file.is_open()) return; //TODO: handle error list_KEY_LIST.GetItemText(i, 0, fp, _countof(fp)); string out; DWORD code; std::vector cmd; cmd.push_back(L"--batch"); cmd.push_back(L"-a"); cmd.push_back(L"--export-secret-keys"); cmd.push_back(fp); gpg_execution_params params(cmd); pxResult result; params.out = &out; params.code = &code; params.result = &result; if (!gpg_launcher(params)) { return; } if (result == pxNotFound) return; boost::algorithm::erase_all(out, "\r"); file << out; if (file.is_open()) file.close(); } } void onClick_CHANGE_PASSWD(CCtrlButton*) { int i = list_KEY_LIST.GetSelectionMark(); if (i == -1) return; list_KEY_LIST.GetItemText(i, 0, key_id_global, _countof(key_id_global)); //temporary code follows std::vector cmd; std::string old_pass, new_pass; string output; DWORD exitcode; cmd.push_back(L"--edit-key"); cmd.push_back(key_id_global); cmd.push_back(L"passwd"); gpg_execution_params_pass params(cmd, old_pass, new_pass); pxResult result; params.out = &output; params.code = &exitcode; params.result = &result; boost::thread gpg_thread(boost::bind(&pxEexcute_passwd_change_thread, ¶ms)); if (!gpg_thread.timed_join(boost::posix_time::minutes(10))) { gpg_thread.~thread(); if (params.child) boost::process::terminate(*(params.child)); if (bDebugLog) debuglog << std::string(time_str() + ": GPG execution timed out, aborted"); this->Close(); } } void onClick_GENERATE_RANDOM(CCtrlButton*) { lbl_GENERATING_KEY.SendMsg(WM_SETFONT, (WPARAM)bold_font, TRUE); lbl_GENERATING_KEY.SetText(TranslateT("Generating new random key, please wait")); btn_GENERATE_KEY.Disable(); btn_OTHER.Disable(); btn_DELETE_KEY.Disable(); list_KEY_LIST.Disable(); btn_GENERATE_RANDOM.Disable(); gpg_use_new_random_key(combo_ACCOUNT.GetTextA()); this->Close(); } void onClick_GENERATE_KEY(CCtrlButton*) { void ShowKeyGenDialog(); ShowKeyGenDialog(); refresh_key_list(); } void onClick_OTHER(CCtrlButton*) { void ShowLoadPublicKeyDialog(bool = false); item_num = 0; //black magic here user_data[1] = 0; ShowLoadPublicKeyDialog(true); refresh_key_list(); } void onClick_DELETE_KEY(CCtrlButton*) { int i = list_KEY_LIST.GetSelectionMark(); if (i == -1) return; list_KEY_LIST.GetItemText(i, 0, fp, _countof(fp)); { string out; DWORD code; std::vector cmd; cmd.push_back(L"--batch"); cmd.push_back(L"--fingerprint"); cmd.push_back(fp); gpg_execution_params params(cmd); pxResult result; params.out = &out; params.code = &code; params.result = &result; if (!gpg_launcher(params)) return; if (result == pxNotFound) return; string::size_type s = out.find("Key fingerprint = "); s += mir_strlen("Key fingerprint = "); string::size_type s2 = out.find("\n", s); wchar_t *str = nullptr; { string tmp = out.substr(s, s2 - s - 1).c_str(); string::size_type p = 0; while ((p = tmp.find(" ", p)) != string::npos) tmp.erase(p, 1); str = mir_a2u(tmp.c_str()); } cmd.clear(); out.clear(); cmd.push_back(L"--batch"); cmd.push_back(L"--delete-secret-and-public-key"); cmd.push_back(L"--fingerprint"); cmd.push_back(str); mir_free(str); if (!gpg_launcher(params)) return; if (result == pxNotFound) return; } { char *buf = mir_strdup(combo_ACCOUNT.GetTextA()); if (!mir_strcmp(buf, Translate("Default"))) { db_unset(NULL, szGPGModuleName, "GPGPubKey"); db_unset(NULL, szGPGModuleName, "KeyID"); db_unset(NULL, szGPGModuleName, "KeyComment"); db_unset(NULL, szGPGModuleName, "KeyMainName"); db_unset(NULL, szGPGModuleName, "KeyMainEmail"); db_unset(NULL, szGPGModuleName, "KeyType"); } else { std::string acc_str = buf; acc_str += "_GPGPubKey"; db_unset(NULL, szGPGModuleName, acc_str.c_str()); acc_str = buf; acc_str += "_KeyMainName"; db_unset(NULL, szGPGModuleName, acc_str.c_str()); acc_str = buf; acc_str += "_KeyID"; db_unset(NULL, szGPGModuleName, acc_str.c_str()); acc_str = buf; acc_str += "_KeyComment"; db_unset(NULL, szGPGModuleName, acc_str.c_str()); acc_str = buf; acc_str += "_KeyMainEmail"; db_unset(NULL, szGPGModuleName, acc_str.c_str()); acc_str = buf; acc_str += "_KeyType"; db_unset(NULL, szGPGModuleName, acc_str.c_str()); } if (buf) mir_free(buf); } list_KEY_LIST.DeleteItem(i); } void onClick_OK(CCtrlButton*) { { int i = list_KEY_LIST.GetSelectionMark(); if (i == -1) return; list_KEY_LIST.GetItemText(i, 0, fp, _countof(fp)); wchar_t *name = new wchar_t[65]; list_KEY_LIST.GetItemText(i, 2, name, 64); { if (wcschr(name, '(')) { wstring str = name; wstring::size_type p = str.find(L"(") - 1; mir_wstrcpy(name, str.substr(0, p).c_str()); } } string out; DWORD code; std::vector cmd; cmd.push_back(L"--batch"); cmd.push_back(L"-a"); cmd.push_back(L"--export"); cmd.push_back(fp); gpg_execution_params params(cmd); pxResult result; params.out = &out; params.code = &code; params.result = &result; if (!gpg_launcher(params)) { delete [] name; return; } if (result == pxNotFound) { delete[] name; return; } boost::algorithm::erase_all(out, "\r"); { char *buf = mir_strdup(combo_ACCOUNT.GetTextA()); if (!mir_strcmp(buf, Translate("Default"))) { db_set_s(NULL, szGPGModuleName, "GPGPubKey", out.c_str()); db_set_ws(NULL, szGPGModuleName, "KeyMainName", name); db_set_ws(NULL, szGPGModuleName, "KeyID", fp); } else { std::string acc_str = buf; acc_str += "_GPGPubKey"; db_set_s(NULL, szGPGModuleName, acc_str.c_str(), out.c_str()); acc_str = buf; acc_str += "_KeyMainName"; db_set_ws(NULL, szGPGModuleName, acc_str.c_str(), name); acc_str = buf; acc_str += "_KeyID"; db_set_ws(NULL, szGPGModuleName, acc_str.c_str(), fp); } if (!mir_strcmp(buf, Translate("Default"))) { wstring keyinfo = TranslateT("Default private key ID"); keyinfo += L": "; keyinfo += (fp[0]) ? fp : L"not set"; extern HWND hwndCurKey_p; SetWindowText(hwndCurKey_p, keyinfo.c_str()); } if (buf) mir_free(buf); } wchar_t *passwd = mir_wstrdup(edit_KEY_PASSWORD.GetText()); if (passwd && passwd[0]) { string dbsetting = "szKey_"; char *keyid = mir_u2a(fp); dbsetting += keyid; mir_free(keyid); dbsetting += "_Password"; db_set_ws(NULL, szGPGModuleName, dbsetting.c_str(), passwd); } mir_free(passwd); delete[] name; } //bAutoExchange = CheckStateStoreDB(hwndDlg, IDC_AUTO_EXCHANGE, "bAutoExchange") != 0; //TODO: check is it just typo, or doing something gpg_valid = isGPGValid(); gpg_keyexist = isGPGKeyExist(); DestroyWindow(m_hwnd); } void onChange_ACCOUNT(CCtrlCombo*) { char *buf = mir_strdup(combo_ACCOUNT.GetTextA()); if (!mir_strcmp(buf, Translate("Default"))) { string keyinfo = Translate("key ID"); keyinfo += ": "; char *keyid = UniGetContactSettingUtf(NULL, szGPGModuleName, "KeyID", ""); keyinfo += (mir_strlen(keyid) > 0) ? keyid : Translate("not set"); mir_free(keyid); lbl_KEY_ID.SetTextA(keyinfo.c_str()); } else { string keyinfo = Translate("key ID"); keyinfo += ": "; std::string acc_str = buf; acc_str += "_KeyID"; char *keyid = UniGetContactSettingUtf(NULL, szGPGModuleName, acc_str.c_str(), ""); keyinfo += (mir_strlen(keyid) > 0) ? keyid : Translate("not set"); mir_free(keyid); lbl_KEY_ID.SetTextA(keyinfo.c_str()); } if (buf) mir_free(buf); } void onChange_KEY_LIST(CCtrlListView::TEventInfo *ev) //TODO: check if this work { if (ev->nmlv) { NMLISTVIEW *hdr = ev->nmlv; if (hdr->hdr.code == NM_CLICK) { btn_OK.Enable(); btn_COPY_PUBKEY.Enable(); btn_EXPORT_PRIVATE.Enable(); btn_CHANGE_PASSWD.Enable(); } } } virtual void OnDestroy() override { GetWindowRect(m_hwnd, &firstrun_rect); db_set_dw(NULL, szGPGModuleName, "FirstrunWindowX", firstrun_rect.left); db_set_dw(NULL, szGPGModuleName, "FirstrunWindowY", firstrun_rect.top); hwndFirstRun = nullptr; delete this; } private: void refresh_key_list() { list_KEY_LIST.DeleteAllItems(); int i = 1; { { //parse gpg output string out; DWORD code; pxResult result; wstring::size_type p = 0, p2 = 0, stop = 0; { std::vector cmd; cmd.push_back(L"--batch"); cmd.push_back(L"--list-secret-keys"); gpg_execution_params params(cmd); params.out = &out; params.code = &code; params.result = &result; if (!gpg_launcher(params)) return; if (result == pxNotFound) return; } while (p != string::npos) { if ((p = out.find("sec ", p)) == string::npos) break; p += 5; if (p < stop) break; stop = p; p2 = out.find("/", p) - 1; wchar_t *key_len = mir_wstrdup(toUTF16(out.substr(p, p2 - p)).c_str()), *creation_date = nullptr, *expire_date = nullptr; p2 += 2; p = out.find(" ", p2); std::wstring key_id = toUTF16(out.substr(p2, p - p2)); p += 1; p2 = out.find(" ", p); std::string::size_type p3 = out.find("\n", p); if ((p2 != std::string::npos) && (p3 < p2)) { p2 = p3; creation_date = mir_wstrdup(toUTF16(out.substr(p, p2 - p - 1)).c_str()); } else { creation_date = mir_wstrdup(toUTF16(out.substr(p, p2 - p)).c_str()); p2 = out.find("[", p2); p2 = out.find("expires:", p2); p2 += mir_strlen("expires:"); if (p2 != std::string::npos) { p2++; p = p2; p2 = out.find("]", p); expire_date = mir_wstrdup(toUTF16(out.substr(p, p2 - p)).c_str()); //check expiration bool expired = false; { boost::posix_time::ptime now = boost::posix_time::second_clock::local_time(); wchar_t buf[5]; wcsncpy_s(buf, expire_date, _TRUNCATE); int year = _wtoi(buf); if (year < now.date().year()) expired = true; else if (year == now.date().year()) { wcsncpy_s(buf, (expire_date + 5), _TRUNCATE); int month = _wtoi(buf); if (month < now.date().month()) expired = true; else if (month == now.date().month()) { wcsncpy_s(buf, (expire_date + 8), _TRUNCATE); unsigned day = _wtoi(buf); if (day <= now.date().day_number()) expired = true; } } } if (expired) { mir_free(key_len); mir_free(creation_date); mir_free(expire_date); //mimic normal behaviour p = out.find("uid ", p); p2 = out.find_first_not_of(" ", p + 5); p = out.find("<", p2); p++; //p2 = out.find(">", p); // continue; //does not add to key list } } } int row = list_KEY_LIST.AddItem(L"", 0); list_KEY_LIST.SetItemText(row, 3, creation_date); mir_free(creation_date); if (expire_date) { list_KEY_LIST.SetItemText(row, 4, expire_date); mir_free(expire_date); } list_KEY_LIST.SetItemText(row, 5, key_len); mir_free(key_len); list_KEY_LIST.SetItemText(row, 0, (wchar_t*)key_id.c_str()); p = out.find("uid ", p); p2 = out.find_first_not_of(" ", p + 5); p = out.find("<", p2); wstring tmp = toUTF16(out.substr(p2, p - p2)); list_KEY_LIST.SetItemText(row, 2, (wchar_t*)tmp.c_str()); p++; p2 = out.find(">", p); tmp = toUTF16(out.substr(p, p2 - p)); list_KEY_LIST.SetItemText(row, 1, (wchar_t*)tmp.c_str()); // get accounts int count = 0; PROTOACCOUNT **accounts; Proto_EnumAccounts(&count, &accounts); std::wstring accs; for (int n = 0; n < count; n++) { std::string setting = toUTF8(accounts[n]->tszAccountName); setting += "("; setting += accounts[n]->szModuleName; setting += ")"; setting += "_KeyID"; wchar_t *str = UniGetContactSettingUtf(NULL, szGPGModuleName, setting.c_str(), L""); if (key_id == str) { if (!accs.empty()) accs += L","; accs += accounts[n]->tszAccountName; } mir_free(str); } list_KEY_LIST.SetItemText(row, 6, (wchar_t*)accs.c_str()); } i++; list_KEY_LIST.SetColumnWidth(0, LVSCW_AUTOSIZE); list_KEY_LIST.SetColumnWidth(1, LVSCW_AUTOSIZE); list_KEY_LIST.SetColumnWidth(2, LVSCW_AUTOSIZE); list_KEY_LIST.SetColumnWidth(3, LVSCW_AUTOSIZE); list_KEY_LIST.SetColumnWidth(4, LVSCW_AUTOSIZE); list_KEY_LIST.SetColumnWidth(5, LVSCW_AUTOSIZE); list_KEY_LIST.SetColumnWidth(6, LVSCW_AUTOSIZE); } } } CCtrlListView list_KEY_LIST; CCtrlButton btn_COPY_PUBKEY, btn_EXPORT_PRIVATE, btn_CHANGE_PASSWD, btn_GENERATE_RANDOM, btn_GENERATE_KEY, btn_OTHER, btn_DELETE_KEY, btn_OK; CCtrlEdit edit_KEY_PASSWORD; CCtrlCombo combo_ACCOUNT; CCtrlData lbl_KEY_ID, lbl_GENERATING_KEY; wchar_t fp[16]; }; class CDlgGpgBinOpts : public CDlgBase { public: CDlgGpgBinOpts() : CDlgBase(hInst, IDD_BIN_PATH), btn_SET_BIN_PATH(this, IDC_SET_BIN_PATH), btn_SET_HOME_DIR(this, IDC_SET_HOME_DIR), btn_OK(this, ID_OK), btn_GENERATE_RANDOM(this, IDC_GENERATE_RANDOM), edit_BIN_PATH(this, IDC_BIN_PATH), edit_HOME_DIR(this, IDC_HOME_DIR), chk_AUTO_EXCHANGE(this, IDC_AUTO_EXCHANGE) { btn_SET_BIN_PATH.OnClick = Callback(this, &CDlgGpgBinOpts::onClick_SET_BIN_PATH); btn_SET_HOME_DIR.OnClick = Callback(this, &CDlgGpgBinOpts::onClick_SET_HOME_DIR); btn_OK.OnClick = Callback(this, &CDlgGpgBinOpts::onClick_OK); btn_GENERATE_RANDOM.OnClick = Callback(this, &CDlgGpgBinOpts::onClick_GENERATE_RANDOM); } virtual void OnInitDialog() override { CMStringW path; bool gpg_exists = false, lang_exists = false; { wchar_t mir_path[MAX_PATH]; PathToAbsoluteW(L"\\", mir_path); SetCurrentDirectoryW(mir_path); CMStringW gpg_path(mir_path); gpg_path.Append(L"\\GnuPG\\gpg.exe"); CMStringW gpg_lang_path(mir_path); gpg_lang_path.Append(L"\\GnuPG\\gnupg.nls\\en@quot.mo"); if (boost::filesystem::exists(gpg_path.c_str())) { gpg_exists = true; path = L"GnuPG\\gpg.exe"; } else path = gpg_path; if (boost::filesystem::exists(gpg_lang_path.c_str())) lang_exists = true; if (gpg_exists && !lang_exists) MessageBox(nullptr, TranslateT("GPG binary found in Miranda folder, but English locale does not exist.\nIt's highly recommended that you place \\gnupg.nls\\en@quot.mo in GnuPG folder under Miranda root.\nWithout this file you may experience many problems with GPG output on non-English systems\nand plugin may completely not work.\nYou have been warned."), TranslateT("Warning"), MB_OK); } DWORD len = MAX_PATH; bool bad_version = false; { ptrW tmp; if (!gpg_exists) { tmp = UniGetContactSettingUtf(NULL, szGPGModuleName, "szGpgBinPath", (SHGetValueW(HKEY_CURRENT_USER, L"Software\\GNU\\GnuPG", L"gpgProgram", 0, (void*)path.c_str(), &len) == ERROR_SUCCESS) ? path.c_str() : L""); if (tmp[0]) if (!boost::filesystem::exists((wchar_t*)tmp)) MessageBox(nullptr, TranslateT("Wrong GPG binary location found in system.\nPlease choose another location"), TranslateT("Warning"), MB_OK); } else tmp = mir_wstrdup(path.c_str()); edit_BIN_PATH.SetText(tmp); if (gpg_exists/* && lang_exists*/) { db_set_ws(NULL, szGPGModuleName, "szGpgBinPath", tmp); string out; DWORD code; std::vector cmd; cmd.push_back(L"--version"); gpg_execution_params params(cmd); pxResult result; params.out = &out; params.code = &code; params.result = &result; gpg_valid = true; gpg_launcher(params); gpg_valid = false; db_unset(NULL, szGPGModuleName, "szGpgBinPath"); string::size_type p1 = out.find("(GnuPG) "); if (p1 != string::npos) { p1 += mir_strlen("(GnuPG) "); if (out[p1] != '1') bad_version = true; } else { bad_version = false; MessageBox(nullptr, TranslateT("This is not GnuPG binary!\nIt is recommended that you use GnuPG v1.x.x with this plugin."), TranslateT("Error"), MB_OK); } if (bad_version) MessageBox(nullptr, TranslateT("Unsupported GnuPG version found, use at you own risk!\nIt is recommended that you use GnuPG v1.x.x with this plugin."), TranslateT("Warning"), MB_OK); } } { ptrW tmp(UniGetContactSettingUtf(NULL, szGPGModuleName, "szHomePath", L"")); if (!tmp[0]) { wchar_t mir_path[MAX_PATH]; PathToAbsoluteW(L"\\", mir_path); mir_wstrcat(mir_path, L"\\gpg"); if (_waccess(mir_path, 0) != -1) { tmp = mir_wstrdup(mir_path); MessageBox(nullptr, TranslateT("\"GPG\" directory found in Miranda root.\nAssuming it's GPG home directory.\nGPG home directory set."), TranslateT("Info"), MB_OK); } else { wstring path_ = _wgetenv(L"APPDATA"); path_ += L"\\GnuPG"; tmp = mir_wstrdup(path_.c_str()); } } edit_HOME_DIR.SetText(!gpg_exists ? tmp : L"gpg"); } //TODO: additional check for write access if (gpg_exists && lang_exists && !bad_version) MessageBox(nullptr, TranslateT("Your GPG version is supported. The language file was found.\nGPG plugin should work fine.\nPress OK to continue."), TranslateT("Info"), MB_OK); chk_AUTO_EXCHANGE.Enable(); } void onClick_SET_BIN_PATH(CCtrlButton*) { GetFilePath(L"Choose gpg.exe", "szGpgBinPath", L"*.exe", L"EXE Executables"); CMStringW tmp(ptrW(UniGetContactSettingUtf(NULL, szGPGModuleName, "szGpgBinPath", L"gpg.exe"))); edit_BIN_PATH.SetText(tmp); wchar_t mir_path[MAX_PATH]; PathToAbsoluteW(L"\\", mir_path); if (tmp.Find(mir_path, 0) == 0) { CMStringW path = tmp.Mid(mir_wstrlen(mir_path)); edit_BIN_PATH.SetText(path); } } void onClick_SET_HOME_DIR(CCtrlButton*) { GetFolderPath(L"Set home directory", "szHomePath"); CMStringW tmp(ptrW(UniGetContactSettingUtf(NULL, szGPGModuleName, "szHomePath", L""))); edit_HOME_DIR.SetText(tmp); wchar_t mir_path[MAX_PATH]; PathToAbsoluteW(L"\\", mir_path); PathToAbsoluteW(L"\\", mir_path); if (tmp.Find(mir_path, 0) == 0) { CMStringW path = tmp.Mid(mir_wstrlen(mir_path)); edit_HOME_DIR.SetText(path); } } void onClick_OK(CCtrlButton*) { if (gpg_validate_paths(edit_BIN_PATH.GetText(), edit_HOME_DIR.GetText())) { gpg_save_paths(edit_BIN_PATH.GetText(), edit_HOME_DIR.GetText()); gpg_valid = true; db_set_b(NULL, szGPGModuleName, "FirstRun", 0); this->Hide(); ShowFirstRunDialog(); this->Close(); } } void onClick_GENERATE_RANDOM(CCtrlButton*) { if (gpg_validate_paths(edit_BIN_PATH.GetText(), edit_HOME_DIR.GetText())) { gpg_save_paths(edit_BIN_PATH.GetText(), edit_HOME_DIR.GetText()); gpg_valid = true; if (gpg_use_new_random_key()) { db_set_b(NULL, szGPGModuleName, "bAutoExchange", bAutoExchange = chk_AUTO_EXCHANGE.GetState()); gpg_valid = true; db_set_b(NULL, szGPGModuleName, "FirstRun", 0); this->Close(); } } } virtual void OnDestroy() override { hwndSetDirs = nullptr; void InitCheck(); InitCheck(); delete this; } private: CCtrlButton btn_SET_BIN_PATH, btn_SET_HOME_DIR, btn_OK, btn_GENERATE_RANDOM; CCtrlEdit edit_BIN_PATH, edit_HOME_DIR; CCtrlCheck chk_AUTO_EXCHANGE; }; class CDlgNewKey : public CDlgBase { public: CDlgNewKey() : CDlgBase(hInst, IDD_NEW_KEY), lbl_KEY_FROM(this, IDC_KEY_FROM), lbl_MESSAGE(this, IDC_MESSAGE), btn_IMPORT(this, ID_IMPORT), btn_IMPORT_AND_USE(this, IDC_IMPORT_AND_USE), btn_IGNORE_KEY(this, IDC_IGNORE_KEY) { hContact = INVALID_CONTACT_ID; btn_IMPORT.OnClick = Callback(this, &CDlgNewKey::onClick_IMPORT); btn_IMPORT_AND_USE.OnClick = Callback(this, &CDlgNewKey::onClick_IMPORT_AND_USE); btn_IGNORE_KEY.OnClick = Callback(this, &CDlgNewKey::onClick_IGNORE_KEY); } virtual void OnInitDialog() override { hContact = new_key_hcnt; //new_key_hcnt_mutex.unlock(); SetWindowPos(m_hwnd, nullptr, new_key_rect.left, new_key_rect.top, 0, 0, SWP_NOSIZE | SWP_SHOWWINDOW); wchar_t *tmp = UniGetContactSettingUtf(hContact, szGPGModuleName, "GPGPubKey", L""); lbl_MESSAGE.SetText(tmp[0] ? TranslateT("There is existing key for contact, would you like to replace it with new key?") : TranslateT("New public key was received, do you want to import it?")); btn_IMPORT_AND_USE.Enable(db_get_b(hContact, szGPGModuleName, "GPGEncryption", 0)); btn_IMPORT.SetText(tmp[0] ? TranslateT("Replace") : TranslateT("Accept")); mir_free(tmp); tmp = new wchar_t[256]; mir_snwprintf(tmp, 255 * sizeof(wchar_t), TranslateT("Received key from %s"), pcli->pfnGetContactDisplayName(hContact, 0)); lbl_KEY_FROM.SetText(tmp); mir_free(tmp); } virtual void OnDestroy() override { GetWindowRect(m_hwnd, &new_key_rect); db_set_dw(NULL, szGPGModuleName, "NewKeyWindowX", new_key_rect.left); db_set_dw(NULL, szGPGModuleName, "NewKeyWindowY", new_key_rect.top); delete this; } void onClick_IMPORT(CCtrlButton*) { void ImportKey(); ImportKey(); this->Close(); } void onClick_IMPORT_AND_USE(CCtrlButton*) { void ImportKey(); ImportKey(); db_set_b(hContact, szGPGModuleName, "GPGEncryption", 1); void setSrmmIcon(MCONTACT hContact); void setClistIcon(MCONTACT hContact); setSrmmIcon(hContact); setClistIcon(hContact); this->Close(); } void onClick_IGNORE_KEY(CCtrlButton*) { this->Close(); } private: MCONTACT hContact; CCtrlData lbl_KEY_FROM, lbl_MESSAGE; CCtrlButton btn_IMPORT, btn_IMPORT_AND_USE, btn_IGNORE_KEY; }; class CDlgKeyGen : public CDlgBase //TODO: in modal mode window destroying on any button press even without direct "Close" call { public: CDlgKeyGen() : CDlgBase(hInst, IDD_KEY_GEN), combo_KEY_TYPE(this, IDC_KEY_TYPE), edit_KEY_LENGTH(this, IDC_KEY_LENGTH), edit_KEY_PASSWD(this, IDC_KEY_PASSWD), edit_KEY_REAL_NAME(this, IDC_KEY_REAL_NAME), edit_KEY_EMAIL(this, IDC_KEY_EMAIL), edit_KEY_COMMENT(this, IDC_KEY_COMMENT), edit_KEY_EXPIRE_DATE(this, IDC_KEY_EXPIRE_DATE), lbl_GENERATING_TEXT(this, IDC_GENERATING_TEXT), btn_OK(this, IDOK), btn_CANCEL(this, IDCANCEL) { btn_OK.OnClick = Callback(this, &CDlgKeyGen::onClick_OK); btn_CANCEL.OnClick = Callback(this, &CDlgKeyGen::onClick_CANCEL); } virtual void OnInitDialog() override { SetWindowPos(m_hwnd, nullptr, key_gen_rect.left, key_gen_rect.top, 0, 0, SWP_NOSIZE | SWP_SHOWWINDOW); SetCaption(TranslateT("Key Generation dialog")); combo_KEY_TYPE.AddString(L"RSA"); combo_KEY_TYPE.AddString(L"DSA"); combo_KEY_TYPE.SelectString(L"RSA"); edit_KEY_EXPIRE_DATE.SetText(L"0"); edit_KEY_LENGTH.SetText(L"4096"); } void onClick_OK(CCtrlButton*) { { wstring path; { //data sanity checks wchar_t *tmp = mir_wstrdup(combo_KEY_TYPE.GetText()); if (!tmp) { MessageBox(nullptr, TranslateT("You must set encryption algorithm first"), TranslateT("Error"), MB_OK); return; } if (mir_wstrlen(tmp) < 3) { mir_free(tmp); tmp = nullptr; MessageBox(nullptr, TranslateT("You must set encryption algorithm first"), TranslateT("Error"), MB_OK); return; } mir_free(tmp); tmp = mir_wstrdup(edit_KEY_LENGTH.GetText()); if (!tmp) { MessageBox(nullptr, TranslateT("Key length must be of length from 1024 to 4096 bits"), TranslateT("Error"), MB_OK); return; } int length = _wtoi(tmp); mir_free(tmp); if (length < 1024 || length > 4096) { MessageBox(nullptr, TranslateT("Key length must be of length from 1024 to 4096 bits"), TranslateT("Error"), MB_OK); return; } tmp = mir_wstrdup(edit_KEY_EXPIRE_DATE.GetText()); if (!tmp) { MessageBox(nullptr, TranslateT("Invalid date"), TranslateT("Error"), MB_OK); return; } if (mir_wstrlen(tmp) != 10 && tmp[0] != '0') { MessageBox(nullptr, TranslateT("Invalid date"), TranslateT("Error"), MB_OK); mir_free(tmp); return; } mir_free(tmp); tmp = mir_wstrdup(edit_KEY_REAL_NAME.GetText()); if (!tmp) { MessageBox(nullptr, TranslateT("Name must contain at least 5 characters"), TranslateT("Error"), MB_OK); return; } if (mir_wstrlen(tmp) < 5) { MessageBox(nullptr, TranslateT("Name must contain at least 5 characters"), TranslateT("Error"), MB_OK); mir_free(tmp); return; } else if (wcschr(tmp, '(') || wcschr(tmp, ')')) { MessageBox(nullptr, TranslateT("Name cannot contain '(' or ')'"), TranslateT("Error"), MB_OK); mir_free(tmp); return; } mir_free(tmp); tmp = mir_wstrdup(edit_KEY_EMAIL.GetText()); if (!tmp) { MessageBox(nullptr, TranslateT("Invalid Email"), TranslateT("Error"), MB_OK); return; } if ((mir_wstrlen(tmp)) < 5 || (!wcschr(tmp, '@')) || (!wcschr(tmp, '.'))) { MessageBox(nullptr, TranslateT("Invalid Email"), TranslateT("Error"), MB_OK); mir_free(tmp); return; } mir_free(tmp); } { //generating key file wchar_t *tmp = UniGetContactSettingUtf(NULL, szGPGModuleName, "szHomePath", L""); char *tmp2;// = mir_u2a(tmp); path = tmp; mir_free(tmp); // mir_free(tmp2); path.append(L"\\new_key"); wfstream f(path.c_str(), std::ios::out); if (!f.is_open()) { MessageBox(nullptr, TranslateT("Failed to open file"), TranslateT("Error"), MB_OK); return; } f << "Key-Type: "; tmp2 = mir_u2a(combo_KEY_TYPE.GetText()); char *subkeytype = (char*)mir_alloc(6); if (strstr(tmp2, "RSA")) mir_strcpy(subkeytype, "RSA"); else if (strstr(tmp2, "DSA")) //this is useless check for now, but it will be required if someone add another key types support mir_strcpy(subkeytype, "ELG-E"); f << tmp2; mir_free(tmp2); f << "\n"; f << "Key-Length: "; f << _wtoi(edit_KEY_LENGTH.GetText()); f << "\n"; f << "Subkey-Length: "; f << _wtoi(edit_KEY_LENGTH.GetText()); f << "\n"; f << "Subkey-Type: "; f << subkeytype; mir_free(subkeytype); f << "\n"; if (edit_KEY_PASSWD.GetText()[0]) { f << "Passphrase: "; f << toUTF8(edit_KEY_PASSWD.GetText()).c_str(); f << "\n"; } f << "Name-Real: "; f << toUTF8(edit_KEY_REAL_NAME.GetText()).c_str(); f << "\n"; if (edit_KEY_COMMENT.GetText()[0]) { f << "Name-Comment: "; f << toUTF8(edit_KEY_COMMENT.GetText()).c_str(); f << "\n"; } f << "Name-Email: "; f << toUTF8(edit_KEY_EMAIL.GetText()).c_str(); f << "\n"; f << "Expire-Date: "; f << toUTF8(edit_KEY_EXPIRE_DATE.GetText()).c_str(); f << "\n"; f.close(); } { //gpg execution DWORD code; string out; std::vector cmd; cmd.push_back(L"--batch"); cmd.push_back(L"--yes"); cmd.push_back(L"--gen-key"); cmd.push_back(path); gpg_execution_params params(cmd); pxResult result; params.out = &out; params.code = &code; params.result = &result; lbl_GENERATING_TEXT.SendMsg(WM_SETFONT, (WPARAM)bold_font, TRUE); lbl_GENERATING_TEXT.SetText(TranslateT("Generating new key, please wait...")); btn_CANCEL.Disable(); btn_OK.Disable(); combo_KEY_TYPE.Disable(); edit_KEY_LENGTH.Disable(); edit_KEY_PASSWD.Disable(); edit_KEY_REAL_NAME.Disable(); edit_KEY_EMAIL.Disable(); edit_KEY_COMMENT.Disable(); edit_KEY_EXPIRE_DATE.Disable(); if (!gpg_launcher(params, boost::posix_time::minutes(10))) return; if (result == pxNotFound) return; } boost::filesystem::remove(path); } this->Close(); } void onClick_CANCEL(CCtrlButton*) { this->Close(); } virtual void OnDestroy() override { GetWindowRect(m_hwnd, &key_gen_rect); db_set_dw(NULL, szGPGModuleName, "KeyGenWindowX", key_gen_rect.left); db_set_dw(NULL, szGPGModuleName, "KeyGenWindowY", key_gen_rect.top); delete this; } private: CCtrlCombo combo_KEY_TYPE; CCtrlEdit edit_KEY_LENGTH, edit_KEY_PASSWD, edit_KEY_REAL_NAME, edit_KEY_EMAIL, edit_KEY_COMMENT, edit_KEY_EXPIRE_DATE; CCtrlData lbl_GENERATING_TEXT; CCtrlButton btn_OK, btn_CANCEL; }; int itemnum2 = 0; class CDlgLoadExistingKey : public CDlgBase { public: CDlgLoadExistingKey() : CDlgBase(hInst, IDD_LOAD_EXISTING_KEY), btn_OK(this, IDOK), btn_CANCEL(this, IDCANCEL), list_EXISTING_KEY_LIST(this, IDC_EXISTING_KEY_LIST) { id[0] = 0; btn_OK.OnClick = Callback(this, &CDlgLoadExistingKey::onClick_OK); btn_CANCEL.OnClick = Callback(this, &CDlgLoadExistingKey::onClick_CANCEL); } virtual void OnInitDialog() override { SetWindowPos(m_hwnd, nullptr, load_existing_key_rect.left, load_existing_key_rect.top, 0, 0, SWP_NOSIZE | SWP_SHOWWINDOW); list_EXISTING_KEY_LIST.AddColumn(0, TranslateT("Key ID"), 50); list_EXISTING_KEY_LIST.AddColumn(1, TranslateT("Email"), 30); list_EXISTING_KEY_LIST.AddColumn(2, TranslateT("Name"), 250); list_EXISTING_KEY_LIST.AddColumn(3, TranslateT("Creation date"), 30); list_EXISTING_KEY_LIST.AddColumn(4, TranslateT("Expiration date"), 30); list_EXISTING_KEY_LIST.AddColumn(5, TranslateT("Key length"), 30); list_EXISTING_KEY_LIST.SetExtendedListViewStyle(LVS_EX_FULLROWSELECT | LVS_EX_SINGLEROW); int i = 1; { {//parse gpg output string out; DWORD code; string::size_type p = 0, p2 = 0, stop = 0; { std::vector cmd; cmd.push_back(L"--batch"); cmd.push_back(L"--list-keys"); gpg_execution_params params(cmd); pxResult result; params.out = &out; params.code = &code; params.result = &result; if (!gpg_launcher(params)) return; if (result == pxNotFound) return; } while (p != string::npos) { if ((p = out.find("pub ", p)) == string::npos) break; p += 5; if (p < stop) break; stop = p; p2 = out.find("/", p) - 1; wchar_t *tmp = mir_wstrdup(toUTF16(out.substr(p, p2 - p)).c_str()); int row = list_EXISTING_KEY_LIST.AddItem(L"", 0); list_EXISTING_KEY_LIST.SetItemText(row, 5, tmp); mir_free(tmp); p2 += 2; p = out.find(" ", p2); tmp = mir_wstrdup(toUTF16(out.substr(p2, p - p2)).c_str()); list_EXISTING_KEY_LIST.SetItemText(row, 0, tmp); mir_free(tmp); p++; p2 = out.find("\n", p); string::size_type p3 = out.substr(p, p2 - p).find("["); if (p3 != string::npos) { p3 += p; p2 = p3; p2--; p3++; p3 += mir_strlen("expires: "); string::size_type p4 = out.find("]", p3); tmp = mir_wstrdup(toUTF16(out.substr(p3, p4 - p3)).c_str()); list_EXISTING_KEY_LIST.SetItemText(row, 4, tmp); mir_free(tmp); } else p2--; tmp = mir_wstrdup(toUTF16(out.substr(p, p2 - p)).c_str()); list_EXISTING_KEY_LIST.SetItemText(row, 3, tmp); mir_free(tmp); p = out.find("uid ", p); p += mir_strlen("uid "); p2 = out.find("\n", p); p3 = out.substr(p, p2 - p).find("<"); if (p3 != string::npos) { p3 += p; p2 = p3; p2--; p3++; string::size_type p4 = out.find(">", p3); tmp = mir_wstrdup(toUTF16(out.substr(p3, p4 - p3)).c_str()); list_EXISTING_KEY_LIST.SetItemText(row, 1, tmp); mir_free(tmp); } else p2--; p = out.find_first_not_of(" ", p); tmp = mir_wstrdup(toUTF16(out.substr(p, p2 - p)).c_str()); list_EXISTING_KEY_LIST.SetItemText(row, 2, tmp); mir_free(tmp); // p = out.find("sub ", p2) + 6; // p = out.find(" ", p) + 1; // p2 = out.find("\n", p); // tmp = mir_wstrdup(toUTF16(out.substr(p,p2-p-1)).c_str()); // ListView_SetItemText(hwndList, iRow, 3, tmp); // mir_free(tmp); list_EXISTING_KEY_LIST.SetColumnWidth(0, LVSCW_AUTOSIZE);// not sure about this list_EXISTING_KEY_LIST.SetColumnWidth(1, LVSCW_AUTOSIZE); list_EXISTING_KEY_LIST.SetColumnWidth(2, LVSCW_AUTOSIZE); list_EXISTING_KEY_LIST.SetColumnWidth(3, LVSCW_AUTOSIZE); list_EXISTING_KEY_LIST.SetColumnWidth(4, LVSCW_AUTOSIZE); list_EXISTING_KEY_LIST.SetColumnWidth(5, LVSCW_AUTOSIZE); i++; } } } list_EXISTING_KEY_LIST.OnItemChanged = Callback(this, &CDlgLoadExistingKey::onChange_EXISTING_KEY_LIST); } virtual void OnDestroy() override { GetWindowRect(m_hwnd, &load_existing_key_rect); db_set_dw(NULL, szGPGModuleName, "LoadExistingKeyWindowX", load_existing_key_rect.left); db_set_dw(NULL, szGPGModuleName, "LoadExistingKeyWindowY", load_existing_key_rect.top); delete this; } void onClick_OK(CCtrlButton*) { int i = list_EXISTING_KEY_LIST.GetSelectionMark(); if (i == -1) return; //TODO: error message list_EXISTING_KEY_LIST.GetItemText(i, 0, id, _countof(id)); extern CCtrlEdit *edit_p_PubKeyEdit; string out; DWORD code; std::vector cmd; cmd.push_back(L"--batch"); cmd.push_back(L"-a"); cmd.push_back(L"--export"); cmd.push_back(id); gpg_execution_params params(cmd); pxResult result; params.out = &out; params.code = &code; params.result = &result; if (!gpg_launcher(params)) return; if (result == pxNotFound) return; string::size_type s = 0; while ((s = out.find("\r", s)) != string::npos) { out.erase(s, 1); } std::string::size_type p1 = 0, p2 = 0; p1 = out.find("-----BEGIN PGP PUBLIC KEY BLOCK-----"); if (p1 != std::string::npos) { p2 = out.find("-----END PGP PUBLIC KEY BLOCK-----", p1); if (p2 != std::string::npos) { p2 += mir_strlen("-----END PGP PUBLIC KEY BLOCK-----"); out = out.substr(p1, p2 - p1); wchar_t *tmp = mir_a2u(out.c_str()); if (edit_p_PubKeyEdit) edit_p_PubKeyEdit->SetText(tmp); mir_free(tmp); } else MessageBox(nullptr, TranslateT("Failed to export public key."), TranslateT("Error"), MB_OK); } else MessageBox(nullptr, TranslateT("Failed to export public key."), TranslateT("Error"), MB_OK); // SetDlgItemText(hPubKeyEdit, IDC_PUBLIC_KEY_EDIT, tmp); this->Close(); } void onClick_CANCEL(CCtrlButton*) { this->Close(); } void onChange_EXISTING_KEY_LIST(CCtrlListView::TEventInfo * /*ev*/) //TODO: check if this work { if (list_EXISTING_KEY_LIST.GetSelectionMark() != -1) btn_OK.Enable(); } private: wchar_t id[16]; CCtrlButton btn_OK, btn_CANCEL; CCtrlListView list_EXISTING_KEY_LIST; }; class CDlgImportKey : public CDlgBase { public: CDlgImportKey() : CDlgBase(hInst, IDD_IMPORT_KEY), combo_KEYSERVER(this, IDC_KEYSERVER), btn_IMPORT(this, IDC_IMPORT) { hContact = INVALID_CONTACT_ID; btn_IMPORT.OnClick = Callback(this, &CDlgImportKey::onClick_IMPORT); } virtual void OnInitDialog() override { hContact = new_key_hcnt; new_key_hcnt_mutex.unlock(); SetWindowPos(m_hwnd, nullptr, import_key_rect.left, import_key_rect.top, 0, 0, SWP_NOSIZE | SWP_SHOWWINDOW); combo_KEYSERVER.AddString(L"subkeys.pgp.net"); combo_KEYSERVER.AddString(L"keys.gnupg.net"); } virtual void OnDestroy() override { GetWindowRect(m_hwnd, &import_key_rect); db_set_dw(NULL, szGPGModuleName, "ImportKeyWindowX", import_key_rect.left); db_set_dw(NULL, szGPGModuleName, "ImportKeyWindowY", import_key_rect.top); delete this; } void onClick_IMPORT(CCtrlButton*) { string out; DWORD code; std::vector cmd; cmd.push_back(L"--keyserver"); cmd.push_back(combo_KEYSERVER.GetText()); cmd.push_back(L"--recv-keys"); cmd.push_back(toUTF16(hcontact_data[hContact].key_in_prescense)); gpg_execution_params params(cmd); pxResult result; params.out = &out; params.code = &code; params.result = &result; gpg_launcher(params); MessageBoxA(nullptr, out.c_str(), "GPG output", MB_OK); } private: MCONTACT hContact; CCtrlCombo combo_KEYSERVER; CCtrlButton btn_IMPORT; }; void ShowFirstRunDialog() { CDlgFirstRun *d = new CDlgFirstRun; d->Show(); } void ShowSetDirsDialog() { CDlgGpgBinOpts *d = new CDlgGpgBinOpts; d->Show(); } void ShowNewKeyDialog() { CDlgNewKey *d = new CDlgNewKey; d->Show(); } void ShowKeyGenDialog() { CDlgKeyGen *d = new CDlgKeyGen; d->DoModal(); } void ShowSelectExistingKeyDialog() { CDlgLoadExistingKey *d = new CDlgLoadExistingKey; d->DoModal(); } void ShowImportKeyDialog() { CDlgImportKey *d = new CDlgImportKey; //TODO: shoud it be modal ? d->Show(); } void FirstRun() { if (!db_get_b(NULL, szGPGModuleName, "FirstRun", 1)) return; ShowSetDirsDialog(); } void InitCheck() { { // parse gpg output wchar_t *current_home = UniGetContactSettingUtf(NULL, szGPGModuleName, "szHomePath", L""); db_set_ws(NULL, szGPGModuleName, "szHomePath", L""); //we do not need home for gpg binary validation gpg_valid = isGPGValid(); db_set_ws(NULL, szGPGModuleName, "szHomePath", current_home); //return current home dir back mir_free(current_home); bool home_dir_access = false, temp_access = false; wchar_t *home_dir = UniGetContactSettingUtf(NULL, szGPGModuleName, "szHomePath", L""); std::wstring test_path = home_dir; mir_free(home_dir); test_path += L"/"; test_path += toUTF16(get_random(13)); wfstream test_file; test_file.open(test_path, std::ios::trunc | std::ios::out); if (test_file.is_open() && test_file.good()) { test_file << L"access_test\n"; if (test_file.good()) home_dir_access = true; test_file.close(); boost::filesystem::remove(test_path); } home_dir = _tgetenv(L"TEMP"); test_path = home_dir; test_path += L"/"; test_path += toUTF16(get_random(13)); test_file.open(test_path, std::ios::trunc | std::ios::out); if (test_file.is_open() && test_file.good()) { test_file << L"access_test\n"; if (test_file.good()) temp_access = true; test_file.close(); boost::filesystem::remove(test_path); } if (!home_dir_access || !temp_access || !gpg_valid) { wchar_t buf[4096]; wcsncpy(buf, gpg_valid ? TranslateT("GPG binary is set and valid (this is good).\n") : TranslateT("GPG binary unset or invalid (plugin will not work).\n"), _countof(buf)); mir_wstrncat(buf, home_dir_access ? TranslateT("Home dir write access granted (this is good).\n") : TranslateT("Home dir has no write access (plugin most probably will not work).\n"), _countof(buf) - mir_wstrlen(buf)); mir_wstrncat(buf, temp_access ? TranslateT("Temp dir write access granted (this is good).\n") : TranslateT("Temp dir has no write access (plugin should work, but may have some problems, file transfers will not work)."), _countof(buf) - mir_wstrlen(buf)); if (!gpg_valid) mir_wstrncat(buf, TranslateT("\nGPG will be disabled until you solve these problems"), _countof(buf) - mir_wstrlen(buf)); MessageBox(nullptr, buf, TranslateT("GPG plugin problems"), MB_OK); } if (!gpg_valid) return; gpg_keyexist = isGPGKeyExist(); string out; DWORD code; pxResult result; wstring::size_type p = 0, p2 = 0; { std::vector cmd; cmd.push_back(L"--batch"); cmd.push_back(L"--list-secret-keys"); gpg_execution_params params(cmd); params.out = &out; params.code = &code; params.result = &result; if (!gpg_launcher(params)) return; if (result == pxNotFound) return; } home_dir = UniGetContactSettingUtf(NULL, szGPGModuleName, "szHomePath", L""); wstring tmp_dir = home_dir; mir_free(home_dir); tmp_dir += L"\\tmp"; _wmkdir(tmp_dir.c_str()); int count = 0; PROTOACCOUNT **accounts; Proto_EnumAccounts(&count, &accounts); string question; char *keyid = nullptr; for (int i = 0; i < count; i++) { if (StriStr(accounts[i]->szModuleName, "metacontacts")) continue; if (StriStr(accounts[i]->szModuleName, "weather")) continue; std::string acc = toUTF8(accounts[i]->tszAccountName); acc += "("; acc += accounts[i]->szModuleName; acc += ")"; acc += "_KeyID"; keyid = UniGetContactSettingUtf(NULL, szGPGModuleName, acc.c_str(), ""); if (keyid[0]) { question = Translate("Your secret key with ID: "); mir_free(keyid); keyid = UniGetContactSettingUtf(NULL, szGPGModuleName, "KeyID", ""); if ((p = out.find(keyid)) == string::npos) { question += keyid; question += Translate(" for account "); question += toUTF8(accounts[i]->tszAccountName); question += Translate(" deleted from GPG secret keyring.\nDo you want to set another key?"); if (MessageBoxA(nullptr, question.c_str(), Translate("Own secret key warning"), MB_YESNO) == IDYES) ShowFirstRunDialog(); } p2 = p; p = out.find("[", p); p2 = out.find("\n", p2); if ((p != std::string::npos) && (p < p2)) { p = out.find("expires:", p); p += mir_strlen("expires:"); p++; p2 = out.find("]", p); wchar_t *expire_date = mir_wstrdup(toUTF16(out.substr(p, p2 - p)).c_str()); bool expired = false; { boost::posix_time::ptime now = boost::posix_time::second_clock::local_time(); wchar_t buf[5]; wcsncpy_s(buf, expire_date, _TRUNCATE); int year = _wtoi(buf); if (year < now.date().year()) expired = true; else if (year == now.date().year()) { wcsncpy_s(buf, (expire_date + 5), _TRUNCATE); int month = _wtoi(buf); if (month < now.date().month()) expired = true; else if (month == now.date().month()) { wcsncpy_s(buf, (expire_date + 8), _TRUNCATE); unsigned day = _wtoi(buf); if (day <= now.date().day_number()) expired = true; } } } if (expired) { question += keyid; question += Translate(" for account "); question += toUTF8(accounts[i]->tszAccountName); question += Translate(" expired and will not work.\nDo you want to set another key?"); if (MessageBoxA(nullptr, question.c_str(), Translate("Own secret key warning"), MB_YESNO) == IDYES) ShowFirstRunDialog(); } mir_free(expire_date); } } if (keyid) { mir_free(keyid); keyid = nullptr; } } question = Translate("Your secret key with ID: "); keyid = UniGetContactSettingUtf(NULL, szGPGModuleName, "KeyID", ""); char *key = UniGetContactSettingUtf(NULL, szGPGModuleName, "GPGPubKey", ""); if (!db_get_b(NULL, szGPGModuleName, "FirstRun", 1) && (!keyid[0] || !key[0])) { question = Translate("You didn't set a private key.\nWould you like to set it now?"); if (MessageBoxA(nullptr, question.c_str(), Translate("Own private key warning"), MB_YESNO) == IDYES) ShowFirstRunDialog(); } if ((p = out.find(keyid)) == string::npos) { question += keyid; question += Translate(" deleted from GPG secret keyring.\nDo you want to set another key?"); if (MessageBoxA(nullptr, question.c_str(), Translate("Own secret key warning"), MB_YESNO) == IDYES) ShowFirstRunDialog(); } p2 = p; p = out.find("[", p); p2 = out.find("\n", p2); if ((p != std::string::npos) && (p < p2)) { p = out.find("expires:", p); p += mir_strlen("expires:"); p++; p2 = out.find("]", p); wchar_t *expire_date = mir_wstrdup(toUTF16(out.substr(p, p2 - p)).c_str()); bool expired = false; { boost::posix_time::ptime now = boost::posix_time::second_clock::local_time(); wchar_t buf[5]; wcsncpy_s(buf, expire_date, _TRUNCATE); int year = _wtoi(buf); if (year < now.date().year()) expired = true; else if (year == now.date().year()) { wcsncpy_s(buf, (expire_date + 5), _TRUNCATE); int month = _wtoi(buf); if (month < now.date().month()) expired = true; else if (month == now.date().month()) { wcsncpy_s(buf, (expire_date + 8), _TRUNCATE); unsigned day = _wtoi(buf); if (day <= now.date().day_number()) expired = true; } } } if (expired) { question += keyid; question += Translate(" expired and will not work.\nDo you want to set another key?"); if (MessageBoxA(nullptr, question.c_str(), Translate("Own secret key warning"), MB_YESNO) == IDYES) ShowFirstRunDialog(); } mir_free(expire_date); } // TODO: check for expired key mir_free(keyid); mir_free(key); } { wchar_t *path = UniGetContactSettingUtf(NULL, szGPGModuleName, "szHomePath", L""); DWORD dwFileAttr = GetFileAttributes(path); if (dwFileAttr != INVALID_FILE_ATTRIBUTES) { dwFileAttr &= ~FILE_ATTRIBUTE_READONLY; SetFileAttributes(path, dwFileAttr); } mir_free(path); } if (bAutoExchange) { int count = 0; PROTOACCOUNT **accounts; Proto_EnumAccounts(&count, &accounts); ICQ_CUSTOMCAP cap; cap.cbSize = sizeof(ICQ_CUSTOMCAP); cap.hIcon = nullptr; strncpy(cap.name, "GPG Key AutoExchange", MAX_CAPNAME); strncpy(cap.caps, "GPGAutoExchange", sizeof(cap.caps)); for (int i = 0; i < count; i++) if (ProtoServiceExists(accounts[i]->szProtoName, PS_ICQ_ADDCAPABILITY)) CallProtoService(accounts[i]->szProtoName, PS_ICQ_ADDCAPABILITY, 0, (LPARAM)&cap); } if (bFileTransfers) { int count = 0; PROTOACCOUNT **accounts; Proto_EnumAccounts(&count, &accounts); ICQ_CUSTOMCAP cap; cap.cbSize = sizeof(ICQ_CUSTOMCAP); cap.hIcon = nullptr; strncpy(cap.name, "GPG Encrypted FileTransfers", MAX_CAPNAME); strncpy(cap.caps, "GPGFileTransfer", sizeof(cap.caps)); for (int i = 0; i < count; i++) if (ProtoServiceExists(accounts[i]->szProtoName, PS_ICQ_ADDCAPABILITY)) CallProtoService(accounts[i]->szProtoName, PS_ICQ_ADDCAPABILITY, 0, (LPARAM)&cap); } } void ImportKey() { MCONTACT hContact = new_key_hcnt; new_key_hcnt_mutex.unlock(); bool for_all_sub = false; if (db_mc_isMeta(hContact)) { if (MessageBox(nullptr, TranslateT("Do you want to load key for all subcontacts?"), TranslateT("Metacontact detected"), MB_YESNO) == IDYES) for_all_sub = true; if (for_all_sub) { int count = db_mc_getSubCount(hContact); for (int i = 0; i < count; i++) { MCONTACT hcnt = db_mc_getSub(hContact, i); if (hcnt) db_set_ws(hcnt, szGPGModuleName, "GPGPubKey", new_key.c_str()); } } else db_set_ws(metaGetMostOnline(hContact), szGPGModuleName, "GPGPubKey", new_key.c_str()); } else db_set_ws(hContact, szGPGModuleName, "GPGPubKey", new_key.c_str()); new_key.clear(); // gpg execute block std::vector cmd; wchar_t tmp2[MAX_PATH] = { 0 }; { wcsncpy(tmp2, ptrW(UniGetContactSettingUtf(NULL, szGPGModuleName, "szHomePath", L"")), MAX_PATH - 1); mir_wstrncat(tmp2, L"\\", _countof(tmp2) - mir_wstrlen(tmp2)); mir_wstrncat(tmp2, L"temporary_exported.asc", _countof(tmp2) - mir_wstrlen(tmp2)); boost::filesystem::remove(tmp2); ptrW ptmp; if (db_mc_isMeta(hContact)) ptmp = UniGetContactSettingUtf(metaGetMostOnline(hContact), szGPGModuleName, "GPGPubKey", L""); else ptmp = UniGetContactSettingUtf(hContact, szGPGModuleName, "GPGPubKey", L""); wfstream f(tmp2, std::ios::out); f << ptmp.get(); f.close(); cmd.push_back(L"--batch"); cmd.push_back(L"--import"); cmd.push_back(tmp2); } gpg_execution_params params(cmd); string output; DWORD exitcode; pxResult result; params.out = &output; params.code = &exitcode; params.result = &result; if (!gpg_launcher(params)) return; if (result == pxNotFound) return; if (db_mc_isMeta(hContact)) { if (for_all_sub) { int count = db_mc_getSubCount(hContact); for (int i = 0; i < count; i++) { MCONTACT hcnt = db_mc_getSub(hContact, i); if (hcnt) { char *tmp = nullptr; string::size_type s = output.find("gpg: key ") + mir_strlen("gpg: key "); string::size_type s2 = output.find(":", s); db_set_s(hcnt, szGPGModuleName, "KeyID", output.substr(s, s2 - s).c_str()); s = output.find("“", s2); if (s == string::npos) { s = output.find("\"", s2); s += 1; } else s += 3; bool uncommon = false; if ((s2 = output.find("(", s)) == string::npos) { if ((s2 = output.find("<", s)) == string::npos) { s2 = output.find("”", s); uncommon = true; } } else if (s2 > output.find("<", s)) s2 = output.find("<", s); if (s != string::npos && s2 != string::npos) { tmp = (char*)mir_alloc(sizeof(char)*(output.substr(s, s2 - s - (uncommon ? 1 : 0)).length() + 1)); mir_strcpy(tmp, output.substr(s, s2 - s - (uncommon ? 1 : 0)).c_str()); mir_utf8decode(tmp, nullptr); db_set_s(hcnt, szGPGModuleName, "KeyMainName", tmp); mir_free(tmp); } if ((s = output.find(")", s2)) == string::npos) s = output.find(">", s2); else if (s > output.find(">", s2)) s = output.find(">", s2); s2++; if (s != string::npos && s2 != string::npos) { if (output[s] == ')') { tmp = (char*)mir_alloc(sizeof(char)* (output.substr(s2, s - s2).length() + 1)); mir_strcpy(tmp, output.substr(s2, s - s2).c_str()); mir_utf8decode(tmp, nullptr); db_set_s(hcnt, szGPGModuleName, "KeyComment", tmp); mir_free(tmp); s += 3; s2 = output.find(">", s); if (s != string::npos && s2 != string::npos) { tmp = (char*)mir_alloc(sizeof(char)*(output.substr(s, s2 - s).length() + 1)); mir_strcpy(tmp, output.substr(s, s2 - s).c_str()); mir_utf8decode(tmp, nullptr); db_set_s(hcnt, szGPGModuleName, "KeyMainEmail", tmp); mir_free(tmp); } } else { tmp = (char*)mir_alloc(sizeof(char)* (output.substr(s2, s - s2).length() + 1)); mir_strcpy(tmp, output.substr(s2, s - s2).c_str()); mir_utf8decode(tmp, nullptr); db_set_s(hcnt, szGPGModuleName, "KeyMainEmail", output.substr(s2, s - s2).c_str()); mir_free(tmp); } } db_unset(hcnt, szGPGModuleName, "bAlwatsTrust"); } } } else { char *tmp = nullptr; string::size_type s = output.find("gpg: key ") + mir_strlen("gpg: key "); string::size_type s2 = output.find(":", s); db_set_s(metaGetMostOnline(hContact), szGPGModuleName, "KeyID", output.substr(s, s2 - s).c_str()); s = output.find("“", s2); if (s == string::npos) { s = output.find("\"", s2); s += 1; } else s += 3; bool uncommon = false; if ((s2 = output.find("(", s)) == string::npos) { if ((s2 = output.find("<", s)) == string::npos) { s2 = output.find("”", s); uncommon = true; } } else if (s2 > output.find("<", s)) s2 = output.find("<", s); if (s != string::npos && s2 != string::npos) { tmp = (char*)mir_alloc(sizeof(char)*(output.substr(s, s2 - s - (uncommon ? 1 : 0)).length() + 1)); mir_strcpy(tmp, output.substr(s, s2 - s - (uncommon ? 1 : 0)).c_str()); mir_utf8decode(tmp, nullptr); db_set_s(metaGetMostOnline(hContact), szGPGModuleName, "KeyMainName", tmp); mir_free(tmp); } if ((s = output.find(")", s2)) == string::npos) s = output.find(">", s2); else if (s > output.find(">", s2)) s = output.find(">", s2); s2++; if (s != string::npos && s2 != string::npos) { if (output[s] == ')') { tmp = (char*)mir_alloc(sizeof(char)* (output.substr(s2, s - s2).length() + 1)); mir_strcpy(tmp, output.substr(s2, s - s2).c_str()); mir_utf8decode(tmp, nullptr); db_set_s(metaGetMostOnline(hContact), szGPGModuleName, "KeyComment", tmp); mir_free(tmp); s += 3; s2 = output.find(">", s); if (s != string::npos && s2 != string::npos) { tmp = (char*)mir_alloc(sizeof(char)*(output.substr(s, s2 - s).length() + 1)); mir_strcpy(tmp, output.substr(s, s2 - s).c_str()); mir_utf8decode(tmp, nullptr); db_set_s(metaGetMostOnline(hContact), szGPGModuleName, "KeyMainEmail", tmp); mir_free(tmp); } } else { tmp = (char*)mir_alloc(sizeof(char)* (output.substr(s2, s - s2).length() + 1)); mir_strcpy(tmp, output.substr(s2, s - s2).c_str()); mir_utf8decode(tmp, nullptr); db_set_s(metaGetMostOnline(hContact), szGPGModuleName, "KeyMainEmail", output.substr(s2, s - s2).c_str()); mir_free(tmp); } } db_unset(metaGetMostOnline(hContact), szGPGModuleName, "bAlwatsTrust"); } } else { char *tmp = nullptr; string::size_type s = output.find("gpg: key ") + mir_strlen("gpg: key "); string::size_type s2 = output.find(":", s); db_set_s(hContact, szGPGModuleName, "KeyID", output.substr(s, s2 - s).c_str()); s = output.find("“", s2); if (s == string::npos) { s = output.find("\"", s2); s += 1; } else s += 3; bool uncommon = false; if ((s2 = output.find("(", s)) == string::npos) { if ((s2 = output.find("<", s)) == string::npos) { s2 = output.find("”", s); uncommon = true; } } else if (s2 > output.find("<", s)) s2 = output.find("<", s); if (s != string::npos && s2 != string::npos) { tmp = (char*)mir_alloc(sizeof(char)*(output.substr(s, s2 - s - (uncommon ? 1 : 0)).length() + 1)); mir_strcpy(tmp, output.substr(s, s2 - s - (uncommon ? 1 : 0)).c_str()); mir_utf8decode(tmp, nullptr); db_set_s(hContact, szGPGModuleName, "KeyMainName", tmp); mir_free(tmp); } if ((s = output.find(")", s2)) == string::npos) s = output.find(">", s2); else if (s > output.find(">", s2)) s = output.find(">", s2); s2++; if (s != string::npos && s2 != string::npos) { if (output[s] == ')') { tmp = (char*)mir_alloc(sizeof(char)* (output.substr(s2, s - s2).length() + 1)); mir_strcpy(tmp, output.substr(s2, s - s2).c_str()); mir_utf8decode(tmp, nullptr); db_set_s(hContact, szGPGModuleName, "KeyComment", tmp); mir_free(tmp); s += 3; s2 = output.find(">", s); if (s != string::npos && s2 != string::npos) { tmp = (char*)mir_alloc(sizeof(char)*(output.substr(s, s2 - s).length() + 1)); mir_strcpy(tmp, output.substr(s, s2 - s).c_str()); mir_utf8decode(tmp, nullptr); db_set_s(hContact, szGPGModuleName, "KeyMainEmail", tmp); mir_free(tmp); } } else { tmp = (char*)mir_alloc(sizeof(char)* (output.substr(s2, s - s2).length() + 1)); mir_strcpy(tmp, output.substr(s2, s - s2).c_str()); mir_utf8decode(tmp, nullptr); db_set_s(hContact, szGPGModuleName, "KeyMainEmail", output.substr(s2, s - s2).c_str()); mir_free(tmp); } } db_unset(hContact, szGPGModuleName, "bAlwatsTrust"); } MessageBox(nullptr, toUTF16(output).c_str(), L"", MB_OK); boost::filesystem::remove(tmp2); }