1 files changed, 9 insertions, 4 deletions
diff --git a/chrome/browser/chromeos/notifications/balloon_view_host.h b/chrome/browser/chromeos/notifications/balloon_view_host.h
index 5b6a3ca5..7b1a3e5 100644
--- a/chrome/browser/chromeos/notifications/balloon_view_host.h
+++ b/chrome/browser/chromeos/notifications/balloon_view_host.h
@@ -12,6 +12,7 @@
 #include <string>
 
 #include "base/callback.h"
+#include "base/compiler_specific.h"
 #include "ui/gfx/native_widget_types.h"
 
 class ListValue;
@@ -33,15 +34,19 @@ class BalloonViewHost : public ::BalloonViewHost {
   bool AddWebUIMessageCallback(const std::string& message,
                                MessageCallback* callback);
 
-  // Process WebUI message.
-  virtual void ProcessWebUIMessage(
-      const ExtensionHostMsg_DomMessage_Params& params);
-
  private:
+  // RenderViewHostDelegate
+  virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE;
+
   // A map of message name -> message handling callback.
   typedef std::map<std::string, MessageCallback*> MessageCallbackMap;
   MessageCallbackMap message_callbacks_;
 
+  // Message handlers.
+  virtual void OnWebUISend(const GURL& source_url,
+                           const std::string& name,
+                           const ListValue& args);
+
   DISALLOW_COPY_AND_ASSIGN(BalloonViewHost);
 };