Fix Up SingletonHwnd Observer Lifetime Issues

ui/gfx/win/singleton_hwnd.h

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef UI_GFX_WIN_SINGLETON_HWND_H_
 #define UI_GFX_WIN_SINGLETON_HWND_H_
 
 #include <windows.h>
 #include <vector>
 
 #include "base/basictypes.h"
+#include "base/callback.h"
 #include "base/callback_forward.h"
 #include "base/observer_list.h"
 #include "ui/gfx/win/window_impl.h"
 
 template<typename T> struct DefaultSingletonTraits;
 
 namespace gfx {
 
 // Singleton message-only HWND that allows interested clients to receive WM_*
 // notifications.
 class GFX_EXPORT SingletonHwnd : public WindowImpl {
  public:
-  static SingletonHwnd* GetInstance();
+  // Singleton lifetime management is tricky. This observer handles the correct
+  // cleanup if either the SingletonHwnd or forwarded object is destroyed first.
+  // To use this class, make sure this object is bound to your object's lifetime
+  // and use SetWndProc to set your WndProc.
+  class GFX_EXPORT Observer {

[sky, 2015/04/24 20:58:23]

Put this into its own header. (From the style guide: "Prefer putting delegate
classes in their own header files").

[robliao, 2015/04/24 21:49:03]

Done.

+   public:
+    typedef base::Callback<void(HWND, UINT, WPARAM, LPARAM)> WndProc;
 
-  // Observer interface for receiving Windows WM_* notifications.
-  class Observer {
-   public:
-    virtual void OnWndProc(HWND hwnd,
-                           UINT message,
-                           WPARAM wparam,
-                           LPARAM lparam) = 0;
+    Observer();
+    ~Observer();
+
+    void SetWndProc(const WndProc& wndProc);

[sky, 2015/04/24 20:58:23]

wnd_proc

[robliao, 2015/04/24 21:49:03]

Done.

+    void ClearWndProc();
+
+   private:
+    friend class SingletonHwnd;
+
+    void OnWndProc(HWND hwnd, UINT message, WPARAM wparam, LPARAM lparam);
+
+    WndProc wndProc_;

[sky, 2015/04/24 20:58:23]

wm_proc_.

[robliao, 2015/04/24 21:49:03]

Done.

+
+    DISALLOW_COPY_AND_ASSIGN(Observer);
   };
 
-  // Add/remove observer to receive WM_* notifications.
-  void AddObserver(Observer* observer);
-  void RemoveObserver(Observer* observer);
+  static SingletonHwnd* GetInstance();
 
   // Windows callback for WM_* notifications.
   BOOL ProcessWindowMessage(HWND window,
                             UINT message,
                             WPARAM wparam,
                             LPARAM lparam,
                             LRESULT& result,
                             DWORD msg_map_id) override;
 
  private:
   friend struct DefaultSingletonTraits<SingletonHwnd>;
 
   SingletonHwnd();
   ~SingletonHwnd() override;
 
+  // Add/remove Observer to forward WM_* notifications.
+  void AddObserver(Observer* observer);
+  void RemoveObserver(Observer* observer);
+
   // List of registered observers.
   ObserverList<Observer> observer_list_;
 
   DISALLOW_COPY_AND_ASSIGN(SingletonHwnd);
 };
 
 }  // namespace gfx
 
 #endif  // UI_GFX_WIN_SINGLETON_HWND_H_