first lines of password UI code

git-svn-id: http://svn.miranda-ng.org/main/trunk@7028 1316c22d-e87f-b044-9b9b-93d7a3e3ba9c

6 files changed, 208 insertions, 18 deletions

diff --git a/plugins/Db3x_mmap/src/commonheaders.h b/plugins/Db3x_mmap/src/commonheaders.h
index bb31a01615..f8978defe4 100644
--- a/plugins/Db3x_mmap/src/commonheaders.h
+++ b/plugins/Db3x_mmap/src/commonheaders.h
@@ -34,6 +34,8 @@ Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 #include <m_system_cpp.h>

 #include <m_database.h>

 #include <m_langpack.h>

+#include <m_clist.h>

+#include <m_icolib.h>

 #include <m_crypto.h>

 

 #include "database.h"

@@ -41,6 +43,7 @@ Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 #include "resource.h"

 #include "version.h"

 

+extern HINSTANCE g_hInst;

 extern LIST<CDb3Mmap> g_Dbs;

 extern DBSignature dbSignature, dbSignatureIM, dbSignatureSA;

 

diff --git a/plugins/Db3x_mmap/src/dbcrypt.cpp b/plugins/Db3x_mmap/src/dbcrypt.cpp
index 79a3744462..4cd24f2ee9 100644
--- a/plugins/Db3x_mmap/src/dbcrypt.cpp
+++ b/plugins/Db3x_mmap/src/dbcrypt.cpp
@@ -149,26 +149,15 @@ LBL_CreateProvider:
 	if ((m_crypto = pProvider->pFactory()) == NULL)

 		return 3;

 

-	size_t iKeyLength = m_crypto->getKeyLength();

-

 	dbv.type = DBVT_BLOB;

 	dbcgs.szSetting = "StoredKey";

 	if (GetContactSetting(NULL, &dbcgs)) {

 LBL_SetNewKey:

 		m_crypto->generateKey(); // unencrypted key

-		

-		BYTE *pKey = (BYTE*)_alloca(iKeyLength);

-		m_crypto->getKey(pKey, iKeyLength);

-

-		DBCONTACTWRITESETTING dbcws = { "CryptoEngine", "StoredKey" };

-		dbcws.value.type = DBVT_BLOB;

-		dbcws.value.cpbVal = (WORD)iKeyLength;

-		dbcws.value.pbVal = pKey;

-		WriteContactSetting(NULL, &dbcws);

-

-		memset(pKey, 0, iKeyLength);

+		StoreKey();		

 	}

 	else {

+		size_t iKeyLength = m_crypto->getKeyLength();

 		if (dbv.cpbVal != (WORD)iKeyLength)

 			goto LBL_SetNewKey;

 

@@ -186,5 +175,21 @@ LBL_SetNewKey:
 		DBWrite(0, &dbSignature, sizeof(dbSignature));

 	}

 

+	InitDialogs();

 	return 0;

 }

+

+void CDb3Mmap::StoreKey()

+{

+	size_t iKeyLength = m_crypto->getKeyLength();

+	BYTE *pKey = (BYTE*)_alloca(iKeyLength);

+	m_crypto->getKey(pKey, iKeyLength);

+

+	DBCONTACTWRITESETTING dbcws = { "CryptoEngine", "StoredKey" };

+	dbcws.value.type = DBVT_BLOB;

+	dbcws.value.cpbVal = (WORD)iKeyLength;

+	dbcws.value.pbVal = pKey;

+	WriteContactSetting(NULL, &dbcws);

+

+	SecureZeroMemory(pKey, iKeyLength);

+}

diff --git a/plugins/Db3x_mmap/src/dbintf.h b/plugins/Db3x_mmap/src/dbintf.h
index 7f0455c6f9..d0919c28c8 100644
--- a/plugins/Db3x_mmap/src/dbintf.h
+++ b/plugins/Db3x_mmap/src/dbintf.h
@@ -152,6 +152,7 @@ struct CDb3Base : public MIDatabase, public MIDatabaseChecker, public MZeroedObj
 	void DatabaseCorruption(TCHAR *text);

 

 	__forceinline HANDLE getFile() const { return m_hDbFile; }

+	__forceinline bool isEncrypted() const { return m_bEncrypted; }

 

 public:

 	STDMETHODIMP_(void)   SetCacheSafetyMode(BOOL);

@@ -297,6 +298,8 @@ struct CDb3Mmap : public CDb3Base
 	CDb3Mmap(const TCHAR* ptszFileName);

 	~CDb3Mmap();

 

+	void     StoreKey(void);

+

 protected:

 	virtual	DWORD GetSettingsGroupOfsByModuleNameOfs(DBContact *dbc,DWORD ofsContact,DWORD ofsModuleName);

 	virtual	void  DBMoveChunk(DWORD ofsDest, DWORD ofsSource, int bytes);

@@ -312,6 +315,8 @@ protected:
 

 	void  Map();

 	void  ReMap(DWORD needed);

+

+	void  InitDialogs();

 };

 

 typedef int (CDb3Base::*CheckWorker)(int);

diff --git a/plugins/Db3x_mmap/src/dbsettings.cpp b/plugins/Db3x_mmap/src/dbsettings.cpp
index cd5a246264..bd5915cbb1 100644
--- a/plugins/Db3x_mmap/src/dbsettings.cpp
+++ b/plugins/Db3x_mmap/src/dbsettings.cpp
@@ -83,7 +83,7 @@ int CDb3Base::GetContactSettingWorker(HANDLE hContact, DBCONTACTGETSETTING *dbcg
 	DBVARIANT *pCachedValue = m_cache->GetCachedValuePtr(hContact, szCachedSettingName, 0);

 	if (pCachedValue != NULL) {

 		if (pCachedValue->type == DBVT_ASCIIZ || pCachedValue->type == DBVT_UTF8) {

-			bool bIsEncrypted = m_bEncrypted || isEncrypted(dbcgs->szModule, dbcgs->szSetting);

+			bool bIsEncrypted = m_bEncrypted || ::isEncrypted(dbcgs->szModule, dbcgs->szSetting);

 			int cbOrigLen = dbcgs->pValue->cchVal;

 			char *cbOrigPtr = dbcgs->pValue->pszVal;

 			memcpy(dbcgs->pValue, pCachedValue, sizeof(DBVARIANT));

@@ -412,7 +412,7 @@ STDMETHODIMP_(BOOL) CDb3Base::WriteContactSetting(HANDLE hContact, DBCONTACTWRIT
 	case DBVT_ASCIIZ: case DBVT_UTF8:

 		if (tmp.value.pszVal == NULL) return 1;

 		tmp.value.cchVal = (WORD)strlen(tmp.value.pszVal);

-		bIsEncrypted = m_bEncrypted || isEncrypted(dbcws->szModule, dbcws->szSetting);

+		bIsEncrypted = m_bEncrypted || ::isEncrypted(dbcws->szModule, dbcws->szSetting);

 		if (bIsEncrypted) {

 			size_t len;

 			BYTE *pResult = m_crypto->encodeString(tmp.value.pszVal, &len);

diff --git a/plugins/Db3x_mmap/src/resource.h b/plugins/Db3x_mmap/src/resource.h
index f321454cd5..071a0af2e6 100644
--- a/plugins/Db3x_mmap/src/resource.h
+++ b/plugins/Db3x_mmap/src/resource.h
@@ -2,17 +2,24 @@
 // Microsoft Visual C++ generated include file.

 // Used by D:\Myranda\plugins\Db3x_mmap\res\db3x_mmap.rc

 //

+

+#define IDREMOVE                        3

+

 #define IDI_ICONPASS                    100

-#define IDD_LOGIN                       101

-#define IDD_CHANGEPASS                  102

+#define IDI_LOGO                        101

+#define IDD_LOGIN                       102

+#define IDD_NEWPASS                     103

+#define IDD_CHANGEPASS                  104

+

 #define IDC_HEADERBAR                   1001

 #define IDC_LANG                        1002

 #define IDC_USERPASS                    1003

 #define IDC_USERPASS1                   1004

 #define IDC_USERPASS2                   1005

+#define IDC_OLDPASS                     1006

 

 // Next default values for new objects

-// 

+//

 #ifdef APSTUDIO_INVOKED

 #ifndef APSTUDIO_READONLY_SYMBOLS

 #define _APS_NEXT_RESOURCE_VALUE        103

diff --git a/plugins/Db3x_mmap/src/ui.cpp b/plugins/Db3x_mmap/src/ui.cpp
new file mode 100644
index 0000000000..2ae486219a
--- /dev/null
+++ b/plugins/Db3x_mmap/src/ui.cpp
@@ -0,0 +1,170 @@
+/*

+

+Miranda NG: the free IM client for Microsoft* Windows*

+

+Copyright 2012-13 Miranda NG project,

+all portions of this codebase are copyrighted to the people

+listed in contributors.txt.

+

+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 "commonheaders.h"

+

+#define MS_DB_CHANGEPASSWORD "DB/ChangePassword"

+

+static IconItem iconList[] =

+{

+	{ LPGEN("Logo"),     "logo",     IDI_LOGO     },

+	{ LPGEN("Password"), "password", IDI_ICONPASS }

+};

+

+static HGENMENU hSetPwdMenu;

+

+static int oldLangID;

+void LanguageChanged(HWND hDlg)

+{

+	UINT LangID = (UINT)GetKeyboardLayout(0);

+	char Lang[3] = { 0 };

+	if (LangID != oldLangID) {

+		oldLangID = LangID;

+		GetLocaleInfoA(MAKELCID((LangID & 0xffffffff), SORT_DEFAULT), LOCALE_SABBREVLANGNAME, Lang, 2);

+		Lang[0] = toupper(Lang[0]);

+		Lang[1] = tolower(Lang[1]);

+		SetDlgItemTextA(hDlg, IDC_LANG, Lang);

+	}

+}

+

+/////////////////////////////////////////////////////////////////////////////////////////

+

+struct DlgChangePassParam

+{

+	CDb3Mmap *db;

+	TCHAR newPass[100];

+};

+

+static INT_PTR CALLBACK sttChangePassword(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)

+{

+	DlgChangePassParam *param = (DlgChangePassParam*)GetWindowLongPtr(hwndDlg, GWLP_USERDATA);

+

+	switch (uMsg) {

+	case WM_INITDIALOG:

+		TranslateDialogDefault(hwndDlg);

+		SendMessage(GetDlgItem(hwndDlg, IDC_HEADERBAR), WM_SETICON, ICON_SMALL, (LPARAM)Skin_GetIconByHandle(iconList[0].hIcolib));

+

+		param = (DlgChangePassParam*)lParam;

+		SetWindowLongPtr(hwndDlg, GWLP_USERDATA, lParam);

+

+		oldLangID = 0;

+		SetTimer(hwndDlg, 1, 200, NULL);

+		LanguageChanged(hwndDlg);

+		return TRUE;

+

+	case WM_CTLCOLORSTATIC:

+		if ((HWND)lParam == GetDlgItem(hwndDlg, IDC_LANG)) {

+			SetTextColor((HDC)wParam, GetSysColor(COLOR_HIGHLIGHTTEXT));

+			SetBkMode((HDC)wParam, TRANSPARENT);

+			return (BOOL)GetSysColorBrush(COLOR_HIGHLIGHT);

+		}

+		return FALSE;

+

+	case WM_COMMAND:

+		switch (LOWORD(wParam)) {

+		case IDCANCEL:

+			EndDialog(hwndDlg, IDCANCEL);

+			break;

+

+		case IDOK:

+			GetDlgItemText(hwndDlg, IDC_USERPASS1, param->newPass, SIZEOF(param->newPass));

+			if (_tcslen(param->newPass) < 3) {

+				SetDlgItemText(hwndDlg, IDC_HEADERBAR, TranslateT("Password is too short!"));

+LBL_Error:

+				SendDlgItemMessage(hwndDlg, IDC_HEADERBAR, WM_NCPAINT, 0, 0);

+				SetDlgItemTextA(hwndDlg, IDC_USERPASS1, "");

+				SetDlgItemTextA(hwndDlg, IDC_USERPASS2, "");

+				break;

+			}

+

+			TCHAR buf2[100];

+			GetDlgItemText(hwndDlg, IDC_USERPASS2, buf2, SIZEOF(buf2));

+			if (_tcscmp(param->newPass, buf2)) {

+				SetWindowText(GetDlgItem(hwndDlg, IDC_HEADERBAR), TranslateT("Passwords do not match!"));

+				goto LBL_Error;

+			}

+

+			if (param->db->isEncrypted()) {

+				GetDlgItemText(hwndDlg, IDC_OLDPASS, buf2, SIZEOF(buf2));

+				ptrA oldPass(mir_utf8encodeT(buf2));

+				if (!param->db->m_crypto->checkPassword(oldPass)) {

+					SetWindowText(GetDlgItem(hwndDlg, IDC_HEADERBAR), TranslateT("Wrong old password entered!"));

+					goto LBL_Error;

+				}

+			}

+			EndDialog(hwndDlg, IDOK);

+		}

+		break;

+

+	case WM_TIMER:

+		LanguageChanged(hwndDlg);

+		return FALSE;

+

+	case WM_DESTROY:

+		KillTimer(hwndDlg, 1);

+		Skin_ReleaseIcon((HICON)SendMessage(hwndDlg, WM_GETICON, ICON_SMALL, 0));

+	}

+

+	return FALSE;

+}

+

+static INT_PTR ChangePassword(void* obj, LPARAM, LPARAM)

+{

+	CDb3Mmap *db = (CDb3Mmap*)obj;

+	DlgChangePassParam param = { db };

+	if (IDOK == DialogBoxParam(g_hInst, MAKEINTRESOURCE(db->isEncrypted() ? IDD_CHANGEPASS : IDD_NEWPASS), 0, sttChangePassword, (LPARAM)&param)) {

+		ptrA newPass(mir_utf8encodeT(param.newPass));

+		db->m_crypto->setPassword(newPass);

+		// db->StoreKey();

+	}

+		

+	return 0;

+}

+

+/////////////////////////////////////////////////////////////////////////////////////////

+

+static int OnModulesLoaded(PVOID obj, WPARAM, LPARAM)

+{

+	CDb3Mmap *db = (CDb3Mmap*)obj;

+

+	Icon_Register(g_hInst, LPGEN("Database"), iconList, SIZEOF(iconList), "mmap");

+

+	// main menu item

+	CLISTMENUITEM mi = { sizeof(mi) };

+	mi.flags = CMIM_ALL | CMIF_TCHAR;

+	mi.icolibItem = iconList[1].hIcolib;

+	mi.ptszName = (db->isEncrypted()) ? LPGENT("Change password") : LPGENT("Set password");

+	mi.ptszPopupName = LPGENT("Database");

+	mi.pszService = MS_DB_CHANGEPASSWORD;

+	mi.position = 500000000;

+	hSetPwdMenu = Menu_AddMainMenuItem(&mi);

+	return 0;

+}

+

+/////////////////////////////////////////////////////////////////////////////////////////

+

+void CDb3Mmap::InitDialogs()

+{

+	CreateServiceFunctionObj(MS_DB_CHANGEPASSWORD, ChangePassword, this);

+	HookEventObj(ME_SYSTEM_MODULESLOADED, OnModulesLoaded, this);

+}