Turn the utility process image decoder into a Mojo service.

chrome/browser/image_decoder.cc

 // Copyright (c) 2012 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 "chrome/browser/image_decoder.h"
 
 #include "base/bind.h"
+#include "base/callback.h"
 #include "base/thread_task_runner_handle.h"
 #include "chrome/browser/browser_process.h"
+#include "chrome/common/chrome_switches.h"
 #include "chrome/common/chrome_utility_messages.h"
 #include "chrome/grit/generated_resources.h"
+#include "components/image_decoder/public/interfaces/image_decoder.mojom.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/utility_process_host.h"
+#include "content/public/common/service_registry.h"
+#include "skia/public/type_converters.h"
 #include "ui/base/l10n/l10n_util.h"
 
 using content::BrowserThread;
 using content::UtilityProcessHost;
 
 namespace {
 
 // static, Leaky to allow access from any thread.
 base::LazyInstance<ImageDecoder>::Leaky g_decoder = LAZY_INSTANCE_INITIALIZER;
 
 // How long to wait after the last request has been received before ending
 // batch mode.
 const int kBatchModeTimeoutSeconds = 5;
 
+void OnDecodeImageDone(
+    base::Callback<void(int)> fail_callback,
+    base::Callback<void(const SkBitmap&, int)> success_callback,
+    int request_id, skia::BitmapPtr image) {
+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
+  if (image) {
+    SkBitmap bitmap = image.To<SkBitmap>();
+    if (!bitmap.empty()) {
+      success_callback.Run(bitmap, request_id);
+      return;
+    }
+  }
+  fail_callback.Run(request_id);
+}
+
 }  // namespace
 
 ImageDecoder::ImageDecoder()
     : image_request_id_counter_(0) {
   // A single ImageDecoder instance should live for the life of the program.
   // Explicitly add a reference so the object isn't deleted.
   AddRef();
 }
 
 ImageDecoder::~ImageDecoder() {
@@ skipping 88 matching lines @@
   if (!batch_mode_timer_) {
     // Created here so it will call StopBatchMode() on the right thread.
     batch_mode_timer_.reset(new base::DelayTimer<ImageDecoder>(
         FROM_HERE,
         base::TimeDelta::FromSeconds(kBatchModeTimeoutSeconds),
         this,
         &ImageDecoder::StopBatchMode));
   }
   batch_mode_timer_->Reset();
 
-  switch (image_codec) {
+  if (switches::MojoUtilityServicesEnabled()) {
+    image_decoder::ImageCodec mojo_codec = image_decoder::IMAGE_CODEC_DEFAULT;
 #if defined(OS_CHROMEOS)
-    case ROBUST_JPEG_CODEC:
-      utility_process_host_->Send(new ChromeUtilityMsg_RobustJPEGDecodeImage(
-          image_data, request_id));
-      break;
+    if (image_codec == ROBUST_JPEG_CODEC)
+      mojo_codec = image_decoder::IMAGE_CODEC_ROBUST_JPEG;
 #endif  // defined(OS_CHROMEOS)
-    case DEFAULT_CODEC:
-      utility_process_host_->Send(new ChromeUtilityMsg_DecodeImage(
-          image_data, shrink_to_fit, request_id));
-      break;
+    (*decoder_)->DecodeImage(
+        mojo::Array<uint8_t>::From(image_data),
+        mojo_codec,
+        shrink_to_fit,
+        base::Bind(&OnDecodeImageDone,
+                   base::Bind(&ImageDecoder::OnDecodeImageFailed, this),
+                   base::Bind(&ImageDecoder::OnDecodeImageSucceeded, this),
+                   request_id));
+  } else {
+    switch (image_codec) {
+#if defined(OS_CHROMEOS)
+      case ROBUST_JPEG_CODEC:
+        utility_process_host_->Send(
+            new ChromeUtilityMsg_RobustJPEGDecodeImage(image_data, request_id));
+        break;
+#endif  // defined(OS_CHROMEOS)
+      case DEFAULT_CODEC:
+        utility_process_host_->Send(new ChromeUtilityMsg_DecodeImage(
+            image_data, shrink_to_fit, request_id));
+        break;
+    }
   }
 }
 
 void ImageDecoder::CancelImpl(ImageRequest* image_request) {
   base::AutoLock lock(map_lock_);
   for (auto it = image_request_id_map_.begin();
        it != image_request_id_map_.end();) {
     if (it->second == image_request) {
       image_request_id_map_.erase(it++);
     } else {
       ++it;
     }
   }
 }
 
 void ImageDecoder::StartBatchMode() {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   utility_process_host_ =
       UtilityProcessHost::Create(
           this, base::ThreadTaskRunnerHandle::Get().get())->AsWeakPtr();
   utility_process_host_->SetName(l10n_util::GetStringUTF16(
       IDS_UTILITY_PROCESS_IMAGE_DECODER_NAME));
-  if (!utility_process_host_->StartBatchMode()) {
-     utility_process_host_.reset();
-     return;
+  if (switches::MojoUtilityServicesEnabled()) {
+    decoder_.reset(new image_decoder::ImageDecoderPtr);
+    if (!utility_process_host_->StartMojoMode()) {
+      utility_process_host_.reset();
+      return;
+    }
+    content::ServiceRegistry* service_registry =
+        utility_process_host_->GetServiceRegistry();
+    service_registry->ConnectToRemoteService(mojo::GetProxy(decoder_.get()));
+  } else {
+    if (!utility_process_host_->StartBatchMode()) {
+      utility_process_host_.reset();
+      return;
+    }
   }
 }
 
 void ImageDecoder::StopBatchMode() {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   {
     // Check for outstanding requests and wait for them to finish.
     base::AutoLock lock(map_lock_);
     if (!image_request_id_map_.empty()) {
       batch_mode_timer_->Reset();
       return;
     }
   }
 
   if (utility_process_host_) {
-    utility_process_host_->EndBatchMode();
+    if (switches::MojoUtilityServicesEnabled()) {
+      // In Mojo, the utility process needs to be explicitly shut down by
+      // deleting the host.
+      delete utility_process_host_.get();
+      decoder_.reset();
+    } else {
+      utility_process_host_->EndBatchMode();
+    }
     utility_process_host_.reset();
   }
 }
 
 void ImageDecoder::FailAllRequests() {
   RequestMap requests;
   {
     base::AutoLock lock(map_lock_);
     requests = image_request_id_map_;
   }
@@ skipping 85 matching lines @@
     auto it = image_request_id_map_.find(request_id);
     if (it == image_request_id_map_.end())
       return;
     image_request = it->second;
     image_request_id_map_.erase(it);
   }
 
   DCHECK(image_request->task_runner()->RunsTasksOnCurrentThread());
   image_request->OnDecodeImageFailed();
 }