1// Copyright (c) 2012 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "chrome/browser/image_decoder.h"
6
7#include "base/bind.h"
8#include "chrome/browser/browser_process.h"
9#include "chrome/common/chrome_utility_messages.h"
10#include "content/public/browser/browser_thread.h"
11#include "content/public/browser/utility_process_host.h"
12
13using content::BrowserThread;
14using content::UtilityProcessHost;
15
16ImageDecoder::ImageDecoder(Delegate* delegate,
17                           const std::string& image_data,
18                           ImageCodec image_codec)
19    : delegate_(delegate),
20      image_data_(image_data.begin(), image_data.end()),
21      image_codec_(image_codec),
22      task_runner_(NULL),
23      shrink_to_fit_(false) {
24}
25
26ImageDecoder::~ImageDecoder() {}
27
28void ImageDecoder::Start(scoped_refptr<base::SequencedTaskRunner> task_runner) {
29  task_runner_ = task_runner;
30  BrowserThread::PostTask(
31     BrowserThread::IO, FROM_HERE,
32     base::Bind(&ImageDecoder::DecodeImageInSandbox, this, image_data_));
33}
34
35bool ImageDecoder::OnMessageReceived(const IPC::Message& message) {
36  bool handled = true;
37  IPC_BEGIN_MESSAGE_MAP(ImageDecoder, message)
38    IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_DecodeImage_Succeeded,
39                        OnDecodeImageSucceeded)
40    IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_DecodeImage_Failed,
41                        OnDecodeImageFailed)
42    IPC_MESSAGE_UNHANDLED(handled = false)
43  IPC_END_MESSAGE_MAP()
44  return handled;
45}
46
47void ImageDecoder::OnDecodeImageSucceeded(const SkBitmap& decoded_image) {
48  DCHECK(task_runner_->RunsTasksOnCurrentThread());
49  if (delegate_)
50    delegate_->OnImageDecoded(this, decoded_image);
51}
52
53void ImageDecoder::OnDecodeImageFailed() {
54  DCHECK(task_runner_->RunsTasksOnCurrentThread());
55  if (delegate_)
56    delegate_->OnDecodeImageFailed(this);
57}
58
59void ImageDecoder::DecodeImageInSandbox(
60    const std::vector<unsigned char>& image_data) {
61  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
62  UtilityProcessHost* utility_process_host;
63  utility_process_host = UtilityProcessHost::Create(this, task_runner_.get());
64  if (image_codec_ == ROBUST_JPEG_CODEC) {
65    utility_process_host->Send(
66        new ChromeUtilityMsg_RobustJPEGDecodeImage(image_data));
67  } else {
68    utility_process_host->Send(
69        new ChromeUtilityMsg_DecodeImage(image_data, shrink_to_fit_));
70  }
71}
72