OLD | NEW |
| (Empty) |
1 // Copyright 2013 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/win/message_window.h" | |
6 | |
7 #include "base/logging.h" | |
8 #include "base/process_util.h" | |
9 #include "base/strings/stringprintf.h" | |
10 #include "base/strings/utf_string_conversions.h" | |
11 #include "base/win/wrapped_window_proc.h" | |
12 | |
13 const char kClassNameFormat[] = "Chromoting_MessageWindow_%p"; | |
14 | |
15 namespace remoting { | |
16 namespace win { | |
17 | |
18 MessageWindow::MessageWindow() | |
19 : atom_(0), | |
20 instance_(NULL), | |
21 window_(NULL) { | |
22 class_name_ = base::StringPrintf(kClassNameFormat, this); | |
23 instance_ = base::GetModuleFromAddress(static_cast<WNDPROC>( | |
24 &base::win::WrappedWindowProc<WindowProc>)); | |
25 } | |
26 | |
27 MessageWindow::MessageWindow(const std::string& class_name, HINSTANCE instance) | |
28 : atom_(0), | |
29 class_name_(class_name), | |
30 instance_(instance), | |
31 window_(NULL) { | |
32 } | |
33 | |
34 MessageWindow::~MessageWindow() { | |
35 DCHECK(CalledOnValidThread()); | |
36 | |
37 if (window_ != NULL) { | |
38 DestroyWindow(window_); | |
39 window_ = NULL; | |
40 } | |
41 | |
42 if (atom_ != 0) { | |
43 UnregisterClass(MAKEINTATOM(atom_), instance_); | |
44 atom_ = 0; | |
45 } | |
46 } | |
47 | |
48 bool MessageWindow::Create(Delegate* delegate) { | |
49 DCHECK(CalledOnValidThread()); | |
50 DCHECK(!atom_); | |
51 DCHECK(!window_); | |
52 | |
53 // Register a separate window class for each instance of |MessageWindow|. | |
54 string16 class_name = UTF8ToUTF16(class_name_); | |
55 WNDCLASSEX window_class; | |
56 window_class.cbSize = sizeof(window_class); | |
57 window_class.style = 0; | |
58 window_class.lpfnWndProc = &base::win::WrappedWindowProc<WindowProc>; | |
59 window_class.cbClsExtra = 0; | |
60 window_class.cbWndExtra = 0; | |
61 window_class.hInstance = instance_; | |
62 window_class.hIcon = NULL; | |
63 window_class.hCursor = NULL; | |
64 window_class.hbrBackground = NULL; | |
65 window_class.lpszMenuName = NULL; | |
66 window_class.lpszClassName = class_name.c_str(); | |
67 window_class.hIconSm = NULL; | |
68 atom_ = RegisterClassEx(&window_class); | |
69 if (atom_ == 0) { | |
70 LOG_GETLASTERROR(ERROR) | |
71 << "Failed to register the window class '" << class_name_ << "'"; | |
72 return false; | |
73 } | |
74 | |
75 window_ = CreateWindow(MAKEINTATOM(atom_), 0, 0, 0, 0, 0, 0, HWND_MESSAGE, 0, | |
76 instance_, delegate); | |
77 if (!window_) { | |
78 LOG_GETLASTERROR(ERROR) << "Failed to create a message-only window"; | |
79 return false; | |
80 } | |
81 | |
82 return true; | |
83 } | |
84 | |
85 // static | |
86 LRESULT CALLBACK MessageWindow::WindowProc(HWND hwnd, | |
87 UINT message, | |
88 WPARAM wparam, | |
89 LPARAM lparam) { | |
90 Delegate* delegate = NULL; | |
91 | |
92 // Set up the delegate before handling WM_CREATE. | |
93 if (message == WM_CREATE) { | |
94 CREATESTRUCT* cs = reinterpret_cast<CREATESTRUCT*>(lparam); | |
95 delegate = reinterpret_cast<Delegate*>(cs->lpCreateParams); | |
96 | |
97 // Store pointer to the delegate to the window's user data. | |
98 SetLastError(ERROR_SUCCESS); | |
99 LONG_PTR result = SetWindowLongPtr( | |
100 hwnd, GWLP_USERDATA, reinterpret_cast<LONG_PTR>(delegate)); | |
101 CHECK(result != 0 || GetLastError() == ERROR_SUCCESS); | |
102 } else { | |
103 delegate = reinterpret_cast<Delegate*>(GetWindowLongPtr(hwnd, | |
104 GWLP_USERDATA)); | |
105 } | |
106 | |
107 // Handle the message. | |
108 if (delegate) { | |
109 LRESULT message_result; | |
110 if (delegate->HandleMessage(hwnd, message, wparam, lparam, &message_result)) | |
111 return message_result; | |
112 } | |
113 | |
114 return DefWindowProc(hwnd, message, wparam, lparam); | |
115 } | |
116 | |
117 } // namespace win | |
118 } // namespace remoting | |
OLD | NEW |