Initial community commit

1 files changed, 405 insertions, 0 deletions

diff --git a/Src/auth/Loginbox/popupPasscode.cpp b/Src/auth/Loginbox/popupPasscode.cpp
new file mode 100644
index 00000000..583a817d
--- /dev/null
+++ b/Src/auth/Loginbox/popupPasscode.cpp
@@ -0,0 +1,405 @@
+#include "./popupPasscode.h"
+#include "./common.h"
+#include "./loginNotifier.h"
+#include "./dataCredentials.h"
+#include "./editboxExtender.h"
+
+#include "../resource.h"
+#include "../api.h"
+#include "../api_auth.h"
+
+#include <windows.h>
+#include <commctrl.h>
+#include <strsafe.h>
+
+#define MAX_PASSCODE_LENGTH			6
+#define MIN_PASSCODE_LENGTH			6
+
+#define TIMER_UPDATETITLE_ID		14
+#define TIMER_UPDATETITLE_DELAY		50
+
+static HRESULT CALLBACK LoginPopupPasscode_CreateInstance(HWND hwnd, LPARAM param, LoginPopup **instance)
+{
+	if (NULL == instance) return E_POINTER;
+	if (NULL == hwnd) return E_INVALIDARG;
+
+	*instance = new LoginPopupPasscode(hwnd);
+	if (NULL == instance) return E_OUTOFMEMORY;
+
+	return S_OK;
+}
+
+LoginPopupPasscode::LoginPopupPasscode(HWND hwnd)
+	: LoginPopup(hwnd, -1, MAKEINTRESOURCE(IDS_POPUP_PASSCODE_TITLE)), 
+	loginData(NULL), message(NULL), messageType(-1)
+{
+}
+
+LoginPopupPasscode::~LoginPopupPasscode()
+{
+	if (NULL != loginData)
+		loginData->Release();
+
+	LoginBox_FreeString(message);
+}
+
+HWND LoginPopupPasscode::CreatePopup(HWND hParent, LoginData *loginData)
+{
+	LoginDataCredentials *credentials;
+	if (NULL == loginData || 
+		FAILED(loginData->QueryInterface(IID_LoginDataCredentials, (void**)&credentials)))
+	{
+		return NULL;
+	}
+
+	HWND hwnd = LoginPopup::CreatePopup(MAKEINTRESOURCE(IDD_POPUP_PASSCODE), hParent, (LPARAM)credentials, LoginPopupPasscode_CreateInstance);
+	credentials->Release();
+	return hwnd;
+}
+
+void LoginPopupPasscode::UpdateLayout(BOOL fRedraw)
+{	
+	LoginPopup::UpdateLayout(fRedraw);
+
+	RECT rect;
+	if (FALSE == GetInfoRect(&rect)) return;
+
+	const INT szButtons[] = { IDOK, };
+	const INT szControls[] = { IDC_PASSCODE_TITLE, IDC_PASSCODE_EDIT, IDC_PASSCODE_HINT, };
+
+	HDWP hdwp = BeginDeferWindowPos(ARRAYSIZE(szControls) + ARRAYSIZE(szButtons));
+	if (NULL == hdwp) return;
+
+	hdwp = LayoutButtons(hdwp, szButtons, ARRAYSIZE(szButtons), fRedraw, NULL);
+
+	LONG top = rect.top;
+	RECT controlRect;
+
+	for(INT i = 0; i < ARRAYSIZE(szControls); i++)
+	{
+		HWND hControl  = GetDlgItem(hwnd, szControls[i]);
+		if (NULL == hControl || FALSE == GetWindowRect(hControl, &controlRect))
+			continue;
+
+		MapWindowPoints(HWND_DESKTOP, hwnd, (POINT*)&controlRect, 2);
+
+		hdwp = DeferWindowPos(hdwp, hControl, NULL, 
+					rect.left, top, rect.right - rect.left, controlRect.bottom - controlRect.top, 
+					SWP_NOACTIVATE | SWP_NOZORDER);
+
+		if (NULL == hdwp)
+			return;
+
+		top += controlRect.bottom - controlRect.top;
+
+		switch(szControls[i])
+		{
+			case IDC_PASSCODE_TITLE:	top += 2; break;
+		}
+		
+	}
+
+
+
+	EndDeferWindowPos(hdwp);
+}
+
+
+void LoginPopupPasscode::EndDialog(INT_PTR code)
+{
+	if (IDOK == code)
+	{
+		code = -1;
+		if (NULL != loginData)
+		{
+			HWND hEdit = GetDlgItem(hwnd, IDC_PASSCODE_EDIT);
+			if (NULL != hEdit)
+			{
+				WCHAR szBuffer[64] = {0};
+				GetWindowText(hEdit, szBuffer, ARRAYSIZE(szBuffer));
+				if(SUCCEEDED(loginData->SetPasscode(szBuffer)))
+					code = IDOK;
+			}
+		}
+	}
+
+	NPPNRESULT result;
+	result.exitCode = code;
+	result.loginData = loginData;
+	if (NULL != result.loginData)
+		result.loginData->AddRef();
+
+	SendNotification(NPPN_RESULT, (NMHDR*)&result);
+
+	if (NULL != result.loginData)
+		result.loginData->Release();
+
+	LoginPopup::EndDialog(code);
+}
+
+BOOL LoginPopupPasscode::Validate()
+{
+	BOOL validatedOk = TRUE;
+
+	HWND hEdit = GetDlgItem(hwnd, IDC_PASSCODE_EDIT);
+	if (NULL == hEdit)
+	{
+		SetAlert(NLNTYPE_ERROR, MAKEINTRESOURCE(IDS_ERR_UNEXPECTED));
+		validatedOk = FALSE;
+	}
+	else
+	{
+		INT cchText = (INT)SendMessage(hEdit, WM_GETTEXTLENGTH, 0, 0L);
+		if (cchText < MIN_PASSCODE_LENGTH || cchText > MAX_PASSCODE_LENGTH)
+		{
+			WCHAR szBuffer[256] = {0}, szTemplate[256] = {0};
+			WASABI_API_LNGSTRINGW_BUF(IDS_ERR_PASSCODE_BADLENGTH_TEMPLATE, szTemplate, ARRAYSIZE(szTemplate));
+			StringCchPrintf(szBuffer, ARRAYSIZE(szBuffer), szTemplate, MAX_PASSCODE_LENGTH);
+			SetAlert(NLNTYPE_ERROR, szBuffer);
+			validatedOk = FALSE;
+		}
+		else
+		{
+			WCHAR szPasscode[MAX_PASSCODE_LENGTH + 1] = {0};
+			WORD szType[MAX_PASSCODE_LENGTH] = {0};
+			SendMessage(hEdit, WM_GETTEXT, (WPARAM)ARRAYSIZE(szPasscode), (LPARAM)szPasscode);
+			if (FALSE != GetStringTypeEx(LOCALE_USER_DEFAULT, CT_CTYPE1, szPasscode, cchText, szType))
+			{
+				for(INT i = 0; i < cchText; i++)
+				{
+					if (0 == (C1_DIGIT & szType[i]))
+					{
+						SetAlert(NLNTYPE_ERROR, MAKEINTRESOURCE(IDS_ERR_PASSCODE_ONLYDIGITS));
+						validatedOk = FALSE;
+						break;
+					}
+				}
+			}
+			else
+			{
+				SetAlert(NLNTYPE_ERROR, MAKEINTRESOURCE(IDS_ERR_UNEXPECTED));
+				validatedOk = FALSE;
+			}
+		}
+	}
+	
+	UpdateTitle(TRUE);
+	return validatedOk;
+}
+
+BOOL LoginPopupPasscode::OnInitDialog(HWND hFocus, LPARAM param)
+{
+	loginData = (LoginDataCredentials*)param;
+	if (NULL != loginData)
+		loginData->AddRef();
+
+	if (NULL != loginData)
+	{
+		LPCWSTR pcode = loginData->GetPasscode();
+		if (NULL != pcode && L'\0' != *pcode)
+		{
+			SetAlert(NLNTYPE_ERROR, MAKEINTRESOURCE(IDS_ERR_PASSCODE_INVALID));
+			UpdateTitle(FALSE);
+		}
+	}
+
+	HWND hEdit = GetDlgItem(hwnd, IDC_PASSCODE_EDIT);
+	if (NULL != hEdit)
+	{
+		EditboxExtender_AttachWindow(hEdit);
+	}
+		
+	HWND hHint = GetDlgItem(hwnd, IDC_PASSCODE_HINT);
+	if (NULL != hHint)
+	{
+		WCHAR szBuffer[128], szTemplate[128] = {0};
+		WASABI_API_LNGSTRINGW_BUF(IDS_PASSCODE_EDIT_HINT, szTemplate, ARRAYSIZE(szTemplate));
+		StringCchPrintf(szBuffer, ARRAYSIZE(szBuffer), szTemplate, MAX_PASSCODE_LENGTH);
+		SendMessage(hHint, WM_SETTEXT, 0, (LPARAM)szBuffer);
+	}
+	
+	return LoginPopup::OnInitDialog(hFocus, param);
+}
+
+void LoginPopupPasscode::OnCommand(UINT commandId, UINT eventType, HWND hControl)
+{
+	if (IDOK == commandId && FALSE == Validate())
+		return;
+
+	LoginPopup::OnCommand(commandId, eventType, hControl);
+}
+
+LRESULT LoginPopupPasscode::OnNotify(UINT controlId, const NMHDR *pnmh)
+{
+	switch(controlId)
+	{
+		case IDC_PASSCODE_EDIT:
+			switch(pnmh->code)
+			{
+				case NM_CHAR:	
+					return OnEditboxChar(pnmh->hwndFrom, ((NMCHAR*)pnmh)->ch);
+				case NM_KEYDOWN:	
+					return OnEditboxKey(pnmh->hwndFrom, ((NMKEY*)pnmh)->nVKey, ((NMKEY*)pnmh)->uFlags);
+				case EENM_PASTE:
+					return OnEditboxPaste(pnmh->hwndFrom, ((EENMPASTE*)pnmh)->text);
+			}
+			return 0;
+	}
+	return LoginPopup::OnNotify(controlId, pnmh);
+}
+
+HBRUSH LoginPopupPasscode::OnGetStaticColor(HDC hdc, HWND hControl)
+{
+	HBRUSH hb = LoginPopup::OnGetStaticColor(hdc, hControl);
+
+	INT controlId = (NULL != hControl) ? (INT)GetWindowLongPtr(hControl, GWLP_ID) : 0;
+	switch(controlId)
+	{
+		case IDC_PASSCODE_HINT:
+			SetTextColor(hdc, GetSysColor(COLOR_GRAYTEXT));
+			break;
+	}
+
+	return hb;
+}
+
+
+LRESULT	LoginPopupPasscode::OnEditboxKey(HWND hEdit, UINT vKey, UINT flags)
+{
+	if (-1 != alertType)
+	{
+		RemoveAlert();
+		SetTimer(hwnd, TIMER_UPDATETITLE_ID, TIMER_UPDATETITLE_DELAY, NULL);
+	}
+	return 0L;
+}
+
+LRESULT LoginPopupPasscode::OnEditboxChar(HWND hEdit, UINT ch)
+{	
+	if (VK_BACK == ch)
+		return FALSE;
+	
+	BOOL filtered = FALSE;
+
+	INT selBegin, selEnd;
+	SendMessage(hEdit, EM_GETSEL, (WPARAM)&selBegin, (LPARAM)&selEnd);
+
+	INT cchText = (INT)SendMessage(hEdit, WM_GETTEXTLENGTH, 0, 0L);
+	if (selBegin == selEnd && cchText >= MAX_PASSCODE_LENGTH)
+	{
+		WCHAR szBuffer[256] = {0}, szTemplate[256] = {0};
+		WASABI_API_LNGSTRINGW_BUF(IDS_ERR_PASSCODE_BADLENGTH_TEMPLATE, szTemplate, ARRAYSIZE(szTemplate));
+		StringCchPrintf(szBuffer, ARRAYSIZE(szBuffer), szTemplate, MAX_PASSCODE_LENGTH);
+		SetAlert(NLNTYPE_WARNING, szBuffer);
+		KillTimer(hwnd, TIMER_UPDATETITLE_ID);
+		UpdateTitle(TRUE);
+		filtered = TRUE;
+	}
+	else
+	{
+		WORD charType;
+		if (FALSE != GetStringTypeEx(LOCALE_USER_DEFAULT, CT_CTYPE1, (WCHAR*)&ch, 1, &charType))
+		{
+			if (0 == (C1_DIGIT & charType))
+			{		
+				SetAlert(NLNTYPE_WARNING, MAKEINTRESOURCE(IDS_ERR_PASSCODE_ONLYDIGITS));
+				KillTimer(hwnd, TIMER_UPDATETITLE_ID);
+				UpdateTitle(TRUE);
+				filtered = TRUE;
+			}
+		}
+	}
+
+	return filtered;
+}
+
+LRESULT LoginPopupPasscode::OnEditboxPaste(HWND hEdit, LPCWSTR pszText)
+{
+	if (NULL == pszText)
+		return TRUE;
+
+	INT cchText = lstrlen(pszText);
+	if (0 == cchText) return TRUE;
+
+	WCHAR szBuffer[MAX_PASSCODE_LENGTH + 1] = {0};
+	INT  iBuffer = 0;
+	
+	WORD charType;
+
+	BOOL validatedOk = TRUE;
+
+	
+	INT maxSize  = MAX_PASSCODE_LENGTH - (INT)SendMessage(hEdit, WM_GETTEXTLENGTH, 0, 0L);
+	if (maxSize < MAX_PASSCODE_LENGTH)
+	{
+		INT selBegin, selEnd;
+		SendMessage(hEdit, EM_GETSEL, (WPARAM)&selBegin, (LPARAM)&selEnd);
+		selEnd -= selBegin;
+		if (selEnd < 0) selEnd = -selBegin;
+		maxSize += selEnd;
+	}
+	for (INT i =0; i <cchText; i++)
+	{
+		if (FALSE == GetStringTypeEx(LOCALE_USER_DEFAULT, CT_CTYPE1, &pszText[i], 1, &charType))
+		{
+			SetAlert(NLNTYPE_WARNING, MAKEINTRESOURCE(IDS_ERR_UNEXPECTED));
+			validatedOk = FALSE;
+			break;
+		}
+		else if (0 == (C1_DIGIT & charType))
+		{		
+			if (0 == ((C1_SPACE | C1_CNTRL | C1_BLANK) & charType))
+			{
+				SetAlert(NLNTYPE_WARNING, MAKEINTRESOURCE(IDS_ERR_PASSCODE_ONLYDIGITS));
+				validatedOk = FALSE;
+				break;
+			}
+		}
+		else
+		{
+			if (iBuffer > maxSize)
+			{
+				WCHAR szBuffer[256], szTemplate[256] = {0};
+				WASABI_API_LNGSTRINGW_BUF(IDS_ERR_PASSCODE_BADLENGTH_TEMPLATE, szTemplate, ARRAYSIZE(szTemplate));
+				StringCchPrintf(szBuffer, ARRAYSIZE(szBuffer), szTemplate, MAX_PASSCODE_LENGTH);
+				SetAlert(NLNTYPE_WARNING, szBuffer);
+				validatedOk = FALSE;
+				break;
+			}
+			else
+			{
+				szBuffer[iBuffer] = pszText[i];
+				iBuffer++;
+			}
+		}
+	}
+
+	if(FALSE != validatedOk)
+	{
+		szBuffer[iBuffer] = L'\0';
+		SendMessage(hEdit, EM_REPLACESEL, (WPARAM)TRUE, (LPARAM)szBuffer);
+	}
+	else
+	{
+		UpdateTitle(TRUE);
+	}
+
+	return TRUE;
+}
+
+INT_PTR LoginPopupPasscode::DialogProc(UINT uMsg, WPARAM wParam, LPARAM lParam)
+{
+	switch (uMsg)
+	{
+		case WM_TIMER:
+			switch(wParam)
+			{
+				case TIMER_UPDATETITLE_ID:
+					KillTimer(hwnd, wParam);
+					UpdateTitle((-1 != alertType));
+					return TRUE;
+			}
+			break;
+	}
+	return LoginPopup::DialogProc(uMsg, wParam, lParam);
+}