remoting/host/verify_config_window_win.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "remoting/host/verify_config_window_win.h"
   6
   7 #include <atlbase.h>
   8 #include <atlwin.h>
   9 #include <windows.h>
  10
  11 #include "base/base64.h"
  12 #include "base/compiler_specific.h"
  13 #include "base/logging.h"
  14 #include "base/strings/utf_string_conversions.h"
  15 #include "remoting/host/pin_hash.h"
  16 #include "remoting/host/win/core_resource.h"
  17 #include "remoting/protocol/authentication_method.h"
  18
  19 namespace remoting {
  20
  21 VerifyConfigWindowWin::VerifyConfigWindowWin(const std::string& email,
  22     const std::string& host_id, const std::string& host_secret_hash)
  23   : email_(email),
  24     host_id_(host_id),
  25     host_secret_hash_(host_secret_hash) {
  26 }
  27
  28 void VerifyConfigWindowWin::OnCancel(UINT code, int id, HWND control) {
  29   EndDialog(ERROR_CANCELLED);
  30 }
  31
  32 void VerifyConfigWindowWin::OnClose() {
  33   EndDialog(ERROR_CANCELLED);
  34 }
  35
  36 LRESULT VerifyConfigWindowWin::OnInitDialog(HWND wparam, LPARAM lparam) {
  37   // Set the small window icon.
  38   if (icon_.LoadIcon(IDD, ::GetSystemMetrics(SM_CXSMICON),
  39                      ::GetSystemMetrics(SM_CYSMICON)) != nullptr) {
  40     SetIcon(icon_, FALSE);
  41   }
  42
  43   CenterWindow();
  44
  45   CWindow email_text(GetDlgItem(IDC_EMAIL));
  46   email_text.SetWindowText(base::UTF8ToUTF16(email_).c_str());
  47   return TRUE;
  48 }
  49
  50 void VerifyConfigWindowWin::OnOk(UINT code, int id, HWND control) {
  51   if (VerifyHostSecretHash()) {
  52     EndDialog(ERROR_SUCCESS);
  53   } else {
  54     EndDialog(ERROR_LOGON_FAILURE);
  55   }
  56 }
  57
  58 void VerifyConfigWindowWin::CenterWindow() {
  59   // Get the window dimensions.
  60   RECT rect;
  61   if (!GetWindowRect(&rect)) {
  62     return;
  63   }
  64
  65   // Center against the owner window unless it is minimized or invisible.
  66   HWND owner = ::GetWindow(m_hWnd, GW_OWNER);
  67   if (owner != nullptr) {
  68     DWORD style = ::GetWindowLong(owner, GWL_STYLE);
  69     if ((style & WS_MINIMIZE) != 0 || (style & WS_VISIBLE) == 0) {
  70       owner = nullptr;
  71     }
  72   }
  73
  74   // Make sure that the window will not end up split by a monitor's boundary.
  75   RECT area_rect;
  76   if (!::SystemParametersInfo(SPI_GETWORKAREA, 0, &area_rect, 0)) {
  77     return;
  78   }
  79
  80   // On a multi-monitor system use the monitor where the owner window is shown.
  81   RECT owner_rect = area_rect;
  82   if (owner != nullptr && ::GetWindowRect(owner, &owner_rect)) {
  83     HMONITOR monitor = ::MonitorFromRect(&owner_rect, MONITOR_DEFAULTTONEAREST);
  84     if (monitor != nullptr) {
  85       MONITORINFO monitor_info = {0};
  86       monitor_info.cbSize = sizeof(monitor_info);
  87       if (::GetMonitorInfo(monitor, &monitor_info)) {
  88         area_rect = monitor_info.rcWork;
  89       }
  90     }
  91   }
  92
  93   LONG width = rect.right - rect.left;
  94   LONG height = rect.bottom - rect.top;
  95   LONG x  = (owner_rect.left + owner_rect.right - width) / 2;
  96   LONG y = (owner_rect.top + owner_rect.bottom - height) / 2;
  97
  98   x = std::max(x, area_rect.left);
  99   x = std::min(x, area_rect.right - width);
 100   y = std::max(y, area_rect.top);
 101   y = std::min(y, area_rect.bottom - width);
 102
 103   SetWindowPos(nullptr, x, y, -1, -1,
 104                SWP_NOSIZE | SWP_NOZORDER | SWP_NOACTIVATE);
 105 }
 106
 107 bool VerifyConfigWindowWin::VerifyHostSecretHash() {
 108   CWindow pin_edit(GetDlgItem(IDC_PIN));
 109
 110   // Get the PIN length.
 111   int pin_length = pin_edit.GetWindowTextLength();
 112   scoped_ptr<base::char16[]> pin(new base::char16[pin_length + 1]);
 113
 114   // Get the PIN making sure it is nullptr terminated even if an error occurs.
 115   int result = pin_edit.GetWindowText(pin.get(), pin_length + 1);
 116   pin[std::min(result, pin_length)] = 0;
 117
 118   return VerifyHostPinHash(host_secret_hash_,
 119                            host_id_, base::UTF16ToUTF8(pin.get()));
 120 }
 121
 122 }  // namespace remoting