Cleanup in ProcessSingleton.

chrome/browser/process_singleton_win.cc

 // Copyright (c) 2011 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.
 
 #include "chrome/browser/process_singleton.h"
 
 #include "base/base_paths.h"
 #include "base/command_line.h"
 #include "base/file_path.h"
 #include "base/path_service.h"
@@ skipping 84 matching lines @@
     DCHECK(result == WAIT_OBJECT_0) << "Result = " << result <<
         "GetLastError = " << GetLastError();
 
     // We now own the mutex so we are the only process that can create the
     // window at this time, but we must still check if someone created it
     // between the time where we looked for it above and the time the mutex
     // was given to us.
     remote_window_ = FindWindowEx(HWND_MESSAGE, NULL,
                                   chrome::kMessageWindowClass,
                                   user_data_dir.value().c_str());
-    if (!remote_window_)
-      Create();
+    if (!remote_window_) {
+      HINSTANCE hinst =
+          base::GetModuleFromAddress(ProcessSingleton::WndProcStatic);
+
+      WNDCLASSEX wc = {0};
+      wc.cbSize = sizeof(wc);
+      wc.lpfnWndProc =
+          base::win::WrappedWindowProc<ProcessSingleton::WndProcStatic>;
+      wc.hInstance = hinst;
+      wc.lpszClassName = chrome::kMessageWindowClass;
+      ATOM clazz = RegisterClassEx(&wc);
+      DCHECK(clazz);
+
+      FilePath user_data_dir;
+      PathService::Get(chrome::DIR_USER_DATA, &user_data_dir);
+
+      // Set the window's title to the path of our user data directory so other
+      // Chrome instances can decide if they should forward to us or not.
+      window_ = CreateWindow(chrome::kMessageWindowClass,
+                             user_data_dir.value().c_str(),
+                             0, 0, 0, 0, 0, HWND_MESSAGE, 0, hinst, 0);
+      ui::CheckWindowCreated(window_);
+      ui::SetWindowUserData(window_, this);
+    }
     BOOL success = ReleaseMutex(only_me);
     DCHECK(success) << "GetLastError = " << GetLastError();
   }
 }
 
 ProcessSingleton::~ProcessSingleton() {
-  if (window_) {
-    DestroyWindow(window_);
-    UnregisterClass(chrome::kMessageWindowClass, GetModuleHandle(NULL));
-  }
+  Cleanup();
 }
 
 ProcessSingleton::NotifyResult ProcessSingleton::NotifyOtherProcess() {
   if (is_virtualized_)
     return PROCESS_NOTIFIED;  // We already spawned the process in this case.
   else if (!remote_window_)
     return PROCESS_NONE;
 
   // Found another window, send our command line to it
   // format is "START\0<<<current directory>>>\0<<<commandline>>>".
@@ skipping 63 matching lines @@
   // Time to take action. Kill the browser process.
   base::KillProcessById(process_id, content::RESULT_CODE_HUNG, true);
   remote_window_ = NULL;
   return PROCESS_NONE;
 }
 
 ProcessSingleton::NotifyResult ProcessSingleton::NotifyOtherProcessOrCreate() {
   NotifyResult result = NotifyOtherProcess();
   if (result != PROCESS_NONE)
     return result;
-  return Create() ? PROCESS_NONE : PROFILE_IN_USE;
+  return window_ ? PROCESS_NONE : PROFILE_IN_USE;
 }
 
-// For windows, there is no need to call Create() since the call is made in
+// On Windows, there is no need to call Create() since the call is made in
 // the constructor but to avoid having more platform specific code in
 // browser_main.cc we tolerate a second call which will do nothing.
 bool ProcessSingleton::Create() {
   DCHECK(!remote_window_);
-  if (window_)
-    return true;
-
-  HINSTANCE hinst = GetModuleHandle(NULL);
-
-  WNDCLASSEX wc = {0};
-  wc.cbSize = sizeof(wc);
-  wc.lpfnWndProc =
-      base::win::WrappedWindowProc<ProcessSingleton::WndProcStatic>;
-  wc.hInstance = hinst;
-  wc.lpszClassName = chrome::kMessageWindowClass;
-  ATOM clazz = RegisterClassEx(&wc);
-  DCHECK(clazz);
-
-  FilePath user_data_dir;
-  PathService::Get(chrome::DIR_USER_DATA, &user_data_dir);
-
-  // Set the window's title to the path of our user data directory so other
-  // Chrome instances can decide if they should forward to us or not.
-  window_ = CreateWindow(chrome::kMessageWindowClass,
-                         user_data_dir.value().c_str(),
-                         0, 0, 0, 0, 0, HWND_MESSAGE, 0, hinst, 0);
-  ui::CheckWindowCreated(window_);
-  ui::SetWindowUserData(window_, this);
-  return true;
+  return window_ != NULL;
 }
 
 void ProcessSingleton::Cleanup() {
+  // Window classes registered by DLLs are not cleaned up automatically on
+  // process exit, so we must unregister at the earliest chance possible.
+  // During the fast shutdown sequence, ProcessSingleton::Cleanup() is
+  // called if our process was the first to start.  Therefore we try cleaning
+  // up here, and again in the destructor if needed to catch as many cases
+  // as possible.
+  if (window_) {
+    DestroyWindow(window_);
+    UnregisterClass(chrome::kMessageWindowClass, GetModuleHandle(NULL));
+    window_ = NULL;
+  }
 }
 
 LRESULT ProcessSingleton::OnCopyData(HWND hwnd, const COPYDATASTRUCT* cds) {
   // If locked, it means we are not ready to process this message because
   // we are probably in a first run critical phase.
   if (locked_) {
 #if defined(USE_AURA)
     NOTIMPLEMENTED();
 #else
     // Attempt to place ourselves in the foreground / flash the task bar.
@@ skipping 99 matching lines @@
   switch (message) {
     case WM_COPYDATA:
       return OnCopyData(reinterpret_cast<HWND>(wparam),
                         reinterpret_cast<COPYDATASTRUCT*>(lparam));
     default:
       break;
   }
 
   return ::DefWindowProc(hwnd, message, wparam, lparam);
 }