Post a Message from Java to JS

android_webview/native/aw_message_port_service_impl.cc

 // Copyright 2015 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 "android_webview/native/aw_message_port_service_impl.h"
 
 #include "android_webview/browser/aw_browser_context.h"
 #include "android_webview/browser/aw_message_port_message_filter.h"
 #include "android_webview/native/aw_contents.h"
 #include "base/android/jni_array.h"
 #include "base/android/jni_string.h"
 #include "base/bind.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/message_port_provider.h"
 #include "jni/AwMessagePortService_jni.h"
 
 namespace android_webview {
 
 using base::android::AttachCurrentThread;
+using base::android::ConvertJavaStringToUTF16;
 using base::android::ConvertUTF16ToJavaString;
 using base::android::ScopedJavaGlobalRef;
 using base::android::ScopedJavaLocalRef;
 using base::android::ToJavaIntArray;
 using content::BrowserThread;
 using content::MessagePortProvider;
 
 //static
 AwMessagePortServiceImpl* AwMessagePortServiceImpl::GetInstance() {
   return static_cast<AwMessagePortServiceImpl*>(
@@ skipping 35 matching lines @@
                  filter,
                  portId1,
                  portId2),
       base::Bind(&AwMessagePortServiceImpl::OnMessageChannelCreated,
                  base::Unretained(this),
                  base::Owned(j_callback),
                  base::Owned(portId1),
                  base::Owned(portId2)));
 }
 
-void AwMessagePortServiceImpl::OnConvertedMessage(
+void AwMessagePortServiceImpl::OnConvertedInMessage(
     int message_port_id,
     const base::ListValue& message,
     const std::vector<int>& sent_message_port_ids) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   JNIEnv* env = AttachCurrentThread();
   ScopedJavaLocalRef<jobject> jobj = java_ref_.get(env);
   if (jobj.is_null())
     return;
 
   if (message.GetSize() != 1) {
@@ skipping 21 matching lines @@
     AwMessagePortMessageFilter* filter) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   for (MessagePorts::iterator iter = ports_.begin();
        iter != ports_.end(); iter++) {
     if (iter->second == filter) {
       ports_.erase(iter);
     }
   }
 }
 
+void AwMessagePortServiceImpl::PostOutMessage(JNIEnv* env, jobject obj,
+    int sender_id, jstring message, jintArray sent_ports) {
+  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  base::string16* j_message = new base::string16;
+  ConvertJavaStringToUTF16(env, message, j_message);
+  std::vector<int>* j_sent_ports = new std::vector<int>;
+  if (sent_ports != nullptr)
+    base::android::JavaIntArrayToIntVector(env, sent_ports, j_sent_ports);
+
+  BrowserThread::PostTask(
+      BrowserThread::IO,
+      FROM_HERE,
+      base::Bind(&AwMessagePortServiceImpl::PostOutMessageOnIOThread,
+                 base::Unretained(this),
+                 sender_id,
+                 base::Owned(j_message),
+                 base::Owned(j_sent_ports)));
+}
+
+void AwMessagePortServiceImpl::RemoveSentPorts(
+    const std::vector<int>& sent_ports) {
+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
+  // Remove the filters that are associated with the transferred ports
+  for (std::vector<int>::const_iterator iter = sent_ports.begin();
+       iter != sent_ports.end(); iter++)
+    ports_.erase(*iter);
+}
+
+void AwMessagePortServiceImpl::PostOutMessageOnIOThread(
+    int sender_id,
+    base::string16* message,
+    std::vector<int>* sent_ports) {
+  RemoveSentPorts(*sent_ports);
+  ports_[sender_id]->SendOutMessage(sender_id, *message, *sent_ports);
+}
+
+
 void AwMessagePortServiceImpl::CreateMessageChannelOnIOThread(
     scoped_refptr<AwMessagePortMessageFilter> filter,
     int* portId1,
     int* portId2) {
-  content::MessagePortProvider::CreateMessageChannel(filter.get(), portId1,
-      portId2);
+  MessagePortProvider::CreateMessageChannel(filter.get(), portId1, portId2);
   AddPort(*portId1, filter.get());
   AddPort(*portId2, filter.get());
 }
 
 void AwMessagePortServiceImpl::OnMessageChannelCreated(
     ScopedJavaGlobalRef<jobject>* callback,
     int* port1,
     int* port2) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
   JNIEnv* env = AttachCurrentThread();
@@ skipping 19 matching lines @@
 }
 
 // static
 jlong InitAwMessagePortService(JNIEnv* env, jobject obj) {
   AwMessagePortServiceImpl* service = AwMessagePortServiceImpl::GetInstance();
   service->Init(env, obj);
   return reinterpret_cast<intptr_t>(service);
 }
 
 }  // namespace android_webview