Adding URLResponse Disk Cache to mojo.

shell/android/android_handler.cc

 // Copyright 2014 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 "shell/android/android_handler.h"
 
 #include "base/android/jni_android.h"
 #include "base/android/jni_string.h"
 #include "base/files/file_path.h"
 #include "base/logging.h"
+#include "base/message_loop/message_loop.h"
+#include "base/run_loop.h"
 #include "base/scoped_native_library.h"
 #include "base/trace_event/trace_event.h"
 #include "jni/AndroidHandler_jni.h"
 #include "mojo/common/data_pipe_utils.h"
 #include "mojo/public/c/system/main.h"
 #include "mojo/public/cpp/application/application_impl.h"
 #include "shell/android/run_android_application_function.h"
 #include "shell/native_application_support.h"
 
 using base::android::AttachCurrentThread;
@@ skipping 15 matching lines @@
                            uintptr_t tracing_id,
                            const base::FilePath& app_path,
                            jint j_handle) {
   mojo::InterfaceRequest<mojo::Application> application_request =
       mojo::MakeRequest<mojo::Application>(
           mojo::MakeScopedHandle(mojo::MessagePipeHandle(j_handle)));
 
   // Load the library, so that we can set the application context there if
   // needed.
   // TODO(vtl): We'd use a ScopedNativeLibrary, but it doesn't have .get()!
-  base::NativeLibrary app_library =
-      LoadNativeApplication(app_path, NativeApplicationCleanup::DELETE);
+  base::NativeLibrary app_library = LoadNativeApplication(app_path);
   if (!app_library)
     return;
 
   // Set the application context if needed. Most applications will need to
   // access the Android ApplicationContext in which they are run. If the
   // application library exports the InitApplicationContext function, we will
   // set it there.
   const char* init_application_context_name = "InitApplicationContext";
   typedef void (*InitApplicationContextFn)(
       const base::android::JavaRef<jobject>&);
@@ skipping 25 matching lines @@
 AndroidHandler::~AndroidHandler() {
 }
 
 void AndroidHandler::RunApplication(
     mojo::InterfaceRequest<mojo::Application> application_request,
     mojo::URLResponsePtr response) {
   JNIEnv* env = AttachCurrentThread();
   uintptr_t tracing_id = reinterpret_cast<uintptr_t>(this);
   TRACE_EVENT_ASYNC_BEGIN1("android_handler", "AndroidHandler::RunApplication",
                            tracing_id, "url", std::string(response->url));
-  ScopedJavaLocalRef<jstring> j_archive_path =
-      Java_AndroidHandler_getNewTempArchivePath(env, GetApplicationContext());
-  base::FilePath archive_path(
-      ConvertJavaStringToUTF8(env, j_archive_path.obj()));
+  base::FilePath extracted_dir;
+  base::FilePath cache_dir;
+  {
+    base::MessageLoop loop;

[blundell, 2015/05/07 11:29:22]

Any reason not to put the stuff from line 110 on in a separate method and use
PostTaskAndReply()?

[qsr, 2015/05/07 12:49:02]

Because the runloop must die after line 108. We do not want an active runloop
after this, as the thread must be given to the android application.

+    handler_task_runner_->PostTask(
+        FROM_HERE,
+        base::Bind(&AndroidHandler::ExtractApplication, base::Unretained(this),
+                   base::Unretained(&extracted_dir),
+                   base::Unretained(&cache_dir), base::Passed(response.Pass()),
+                   base::Bind(base::IgnoreResult(
+                                  &base::SingleThreadTaskRunner::PostTask),
+                              loop.task_runner(), FROM_HERE,
+                              base::MessageLoop::QuitWhenIdleClosure())));
+    base::RunLoop().Run();
+  }
 
-  mojo::common::BlockingCopyToFile(response->body.Pass(), archive_path);
+  ScopedJavaLocalRef<jstring> j_extracted_dir =
+      ConvertUTF8ToJavaString(env, extracted_dir.value());
+  ScopedJavaLocalRef<jstring> j_cache_dir =
+      ConvertUTF8ToJavaString(env, cache_dir.value());
   RunAndroidApplicationFn run_android_application_fn = &RunAndroidApplication;
   Java_AndroidHandler_bootstrap(
-      env, GetApplicationContext(), tracing_id, j_archive_path.obj(),
+      env, GetApplicationContext(), tracing_id, j_extracted_dir.obj(),
+      j_cache_dir.obj(),
       application_request.PassMessagePipe().release().value(),
       reinterpret_cast<jlong>(run_android_application_fn));
 }
 
 void AndroidHandler::Initialize(mojo::ApplicationImpl* app) {
+  handler_task_runner_ = base::MessageLoop::current()->task_runner();
+  app->ConnectToService("mojo:service_cache", &service_cache_);
 }
 
 bool AndroidHandler::ConfigureIncomingConnection(
     mojo::ApplicationConnection* connection) {
   connection->AddService(&content_handler_factory_);
   connection->AddService(&intent_receiver_manager_factory_);
   return true;
 }
 
+void AndroidHandler::ExtractApplication(base::FilePath* extracted_dir,
+                                        base::FilePath* cache_dir,
+                                        mojo::URLResponsePtr response,
+                                        const base::Closure& callback) {
+  service_cache_->GetExtractedContent(
+      response.Pass(),
+      [extracted_dir, cache_dir, callback](mojo::Array<uint8_t> extracted_path,
+                                           mojo::Array<uint8_t> cache_path) {
+        if (extracted_path.is_null()) {
+          *extracted_dir = base::FilePath();
+          *cache_dir = base::FilePath();
+        } else {
+          *extracted_dir = base::FilePath(
+              std::string(reinterpret_cast<char*>(&extracted_path.front()),
+                          extracted_path.size()));
+          *cache_dir = base::FilePath(std::string(
+              reinterpret_cast<char*>(&cache_path.front()), cache_path.size()));
+        }
+        callback.Run();
+      });
+}
+
 bool RegisterAndroidHandlerJni(JNIEnv* env) {
   return RegisterNativesImpl(env);
 }
 
 }  // namespace shell