1 files changed, 8 insertions, 9 deletions
diff --git a/Source/WebKit2/WebProcess/WebConnectionToUIProcess.cpp b/Source/WebKit2/WebProcess/WebConnectionToUIProcess.cpp
index aeb602ac8..e1a51f029 100644
--- a/Source/WebKit2/WebProcess/WebConnectionToUIProcess.cpp
+++ b/Source/WebKit2/WebProcess/WebConnectionToUIProcess.cpp
@@ -26,7 +26,6 @@
 #include "config.h"
 #include "WebConnectionToUIProcess.h"
 
-#include "InjectedBundleUserMessageCoders.h"
 #include "WebConnectionMessages.h"
 #include "WebProcess.h"
 
@@ -34,9 +33,9 @@ using namespace WebCore;
 
 namespace WebKit {
 
-PassRefPtr<WebConnectionToUIProcess> WebConnectionToUIProcess::create(WebProcess* process)
+Ref<WebConnectionToUIProcess> WebConnectionToUIProcess::create(WebProcess* process)
 {
-    return adoptRef(new WebConnectionToUIProcess(process));
+    return adoptRef(*new WebConnectionToUIProcess(process));
 }
 
 WebConnectionToUIProcess::WebConnectionToUIProcess(WebProcess* process)
@@ -47,20 +46,20 @@ WebConnectionToUIProcess::WebConnectionToUIProcess(WebProcess* process)
 
 void WebConnectionToUIProcess::invalidate()
 {
-    m_process = 0;
+    m_process->removeMessageReceiver(Messages::WebConnection::messageReceiverName());
+    m_process = nullptr;
 }
 
 // WebConnection
 
-void WebConnectionToUIProcess::encodeMessageBody(IPC::ArgumentEncoder& encoder, API::Object* messageBody)
+RefPtr<API::Object> WebConnectionToUIProcess::transformHandlesToObjects(API::Object* object)
 {
-    encoder << InjectedBundleUserMessageEncoder(messageBody);
+    return m_process->transformHandlesToObjects(object);
 }
 
-bool WebConnectionToUIProcess::decodeMessageBody(IPC::ArgumentDecoder& decoder, RefPtr<API::Object>& messageBody)
+RefPtr<API::Object> WebConnectionToUIProcess::transformObjectsToHandles(API::Object* object)
 {
-    InjectedBundleUserMessageDecoder messageBodyDecoder(messageBody);
-    return decoder.decode(messageBodyDecoder);
+    return m_process->transformObjectsToHandles(object);
 }
 
 bool WebConnectionToUIProcess::hasValidConnection() const