15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/verify_config_window_win.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <atlbase.h> 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <atlwin.h> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <windows.h> 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/base64.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/compiler_specific.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 14868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/utf_string_conversions.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/pin_hash.h" 162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "remoting/host/win/core_resource.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/protocol/authentication_method.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace remoting { 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)VerifyConfigWindowWin::VerifyConfigWindowWin(const std::string& email, 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& host_id, const std::string& host_secret_hash) 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : email_(email), 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host_id_(host_id), 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) host_secret_hash_(host_secret_hash) { 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VerifyConfigWindowWin::OnCancel(UINT code, int id, HWND control) { 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EndDialog(ERROR_CANCELLED); 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VerifyConfigWindowWin::OnClose() { 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EndDialog(ERROR_CANCELLED); 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)LRESULT VerifyConfigWindowWin::OnInitDialog(HWND wparam, LPARAM lparam) { 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Set the small window icon. 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (icon_.LoadIcon(IDD, ::GetSystemMetrics(SM_CXSMICON), 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ::GetSystemMetrics(SM_CYSMICON)) != NULL) { 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetIcon(icon_, FALSE); 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CenterWindow(); 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CWindow email_text(GetDlgItem(IDC_EMAIL)); 465d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) email_text.SetWindowText(base::UTF8ToUTF16(email_).c_str()); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return TRUE; 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VerifyConfigWindowWin::OnOk(UINT code, int id, HWND control) { 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (VerifyHostSecretHash()) { 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EndDialog(ERROR_SUCCESS); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EndDialog(ERROR_LOGON_FAILURE); 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void VerifyConfigWindowWin::CenterWindow() { 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Get the window dimensions. 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RECT rect; 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!GetWindowRect(&rect)) { 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Center against the owner window unless it is minimized or invisible. 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HWND owner = ::GetWindow(m_hWnd, GW_OWNER); 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (owner != NULL) { 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD style = ::GetWindowLong(owner, GWL_STYLE); 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if ((style & WS_MINIMIZE) != 0 || (style & WS_VISIBLE) == 0) { 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) owner = NULL; 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Make sure that the window will not end up split by a monitor's boundary. 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RECT area_rect; 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!::SystemParametersInfo(SPI_GETWORKAREA, NULL, &area_rect, NULL)) { 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // On a multi-monitor system use the monitor where the owner window is shown. 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RECT owner_rect = area_rect; 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (owner != NULL && ::GetWindowRect(owner, &owner_rect)) { 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HMONITOR monitor = ::MonitorFromRect(&owner_rect, MONITOR_DEFAULTTONEAREST); 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (monitor != NULL) { 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MONITORINFO monitor_info = {0}; 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) monitor_info.cbSize = sizeof(monitor_info); 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (::GetMonitorInfo(monitor, &monitor_info)) { 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) area_rect = monitor_info.rcWork; 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LONG width = rect.right - rect.left; 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LONG height = rect.bottom - rect.top; 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LONG x = (owner_rect.left + owner_rect.right - width) / 2; 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LONG y = (owner_rect.top + owner_rect.bottom - height) / 2; 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) x = std::max(x, area_rect.left); 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) x = std::min(x, area_rect.right - width); 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) y = std::max(y, area_rect.top); 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) y = std::min(y, area_rect.bottom - width); 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetWindowPos(NULL, x, y, -1, -1, SWP_NOSIZE | SWP_NOZORDER | SWP_NOACTIVATE); 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool VerifyConfigWindowWin::VerifyHostSecretHash() { 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CWindow pin_edit(GetDlgItem(IDC_PIN)); 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Get the PIN length. 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int pin_length = pin_edit.GetWindowTextLength(); 1115d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) scoped_ptr<base::char16[]> pin(new base::char16[pin_length + 1]); 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Get the PIN making sure it is NULL terminated even if an error occurs. 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int result = pin_edit.GetWindowText(pin.get(), pin_length + 1); 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pin[std::min(result, pin_length)] = 0; 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1175d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) return VerifyHostPinHash(host_secret_hash_, 1185d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) host_id_, base::UTF16ToUTF8(pin.get())); 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace remoting 122