2 files changed, 14 insertions, 0 deletions
diff --git a/chrome_frame/ready_mode/internal/ready_prompt_window.cc b/chrome_frame/ready_mode/internal/ready_prompt_window.cc
index 2c1e3e5..39ccbfa 100644
--- a/chrome_frame/ready_mode/internal/ready_prompt_window.cc
+++ b/chrome_frame/ready_mode/internal/ready_prompt_window.cc
@@ -28,9 +28,12 @@ ReadyPromptWindow::ReadyPromptWindow(
     : frame_(frame),
       ready_mode_state_(ready_mode_state),
       url_launcher_(url_launcher),
+      icon_(NULL),
       weak_ptr_factory_(ALLOW_THIS_IN_INITIALIZER_LIST(this)) {
 }
 ReadyPromptWindow::~ReadyPromptWindow() {
+  if (icon_)
+    ::DestroyIcon(icon_);
 }
 
 base::WeakPtr<ReadyPromptWindow> ReadyPromptWindow::CreateInstance(
@@ -58,6 +61,16 @@ base::WeakPtr<ReadyPromptWindow> ReadyPromptWindow::CreateInstance(
   instance->link_->SetHyperLinkExtendedStyle(HLINK_NOTIFYBUTTON,
                                              HLINK_NOTIFYBUTTON);
 
+  CStatic icon_control(instance->GetDlgItem(IDC_PROMPT_ICON));
+
+  instance->icon_ = static_cast<HICON>(
+      ::LoadImage(_AtlBaseModule.GetResourceInstance(),
+                  MAKEINTRESOURCE(IDI_CHROME_FRAME_ICON),
+                  IMAGE_ICON, 16, 16, 0));
+
+  if (instance->icon_)
+    icon_control.SetIcon(instance->icon_);
+
   return instance;
 }
 
diff --git a/chrome_frame/ready_mode/internal/ready_prompt_window.h b/chrome_frame/ready_mode/internal/ready_prompt_window.h
index 5dc87ae..12676a9 100644
--- a/chrome_frame/ready_mode/internal/ready_prompt_window.h
+++ b/chrome_frame/ready_mode/internal/ready_prompt_window.h
@@ -99,6 +99,7 @@ class ReadyPromptWindow
   scoped_ptr<WTL::CHyperLink> link_;
   scoped_ptr<UrlLauncher> url_launcher_;
 
+  HICON icon_;
   base::WeakPtrFactory<ReadyPromptWindow> weak_ptr_factory_;
   DISALLOW_COPY_AND_ASSIGN(ReadyPromptWindow);
 };  // class ReadyPromptWindow