chrome/browser/chromeos/extensions/wallpaper_function_base.cc

   1 // Copyright 2013 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/chromeos/extensions/wallpaper_function_base.h"
   6
   7 #include "base/synchronization/cancellation_flag.h"
   8 #include "chrome/browser/image_decoder.h"
   9 #include "chrome/grit/generated_resources.h"
  10 #include "chromeos/login/login_state.h"
  11 #include "components/wallpaper/wallpaper_layout.h"
  12 #include "ui/base/l10n/l10n_util.h"
  13
  14 using content::BrowserThread;
  15
  16 namespace wallpaper_api_util {
  17 namespace {
  18
  19 // Keeps in sync (same order) with WallpaperLayout enum in header file.
  20 const char* const kWallpaperLayoutArrays[] = {
  21   "CENTER",
  22   "CENTER_CROPPED",
  23   "STRETCH",
  24   "TILE"
  25 };
  26
  27 const int kWallpaperLayoutCount = arraysize(kWallpaperLayoutArrays);
  28
  29 }  // namespace
  30
  31 const char kCancelWallpaperMessage[] = "Set wallpaper was canceled.";
  32
  33 wallpaper::WallpaperLayout GetLayoutEnum(const std::string& layout) {
  34   for (int i = 0; i < kWallpaperLayoutCount; i++) {
  35     if (layout.compare(kWallpaperLayoutArrays[i]) == 0)
  36       return static_cast<wallpaper::WallpaperLayout>(i);
  37   }
  38   // Default to use CENTER layout.
  39   return wallpaper::WALLPAPER_LAYOUT_CENTER;
  40 }
  41
  42 }  // namespace wallpaper_api_util
  43
  44 class WallpaperFunctionBase::UnsafeWallpaperDecoder
  45     : public ImageDecoder::ImageRequest {
  46  public:
  47   explicit UnsafeWallpaperDecoder(scoped_refptr<WallpaperFunctionBase> function)
  48       : function_(function) {}
  49
  50   void Start(const std::vector<char>& image_data) {
  51     DCHECK_CURRENTLY_ON(BrowserThread::UI);
  52
  53     // This function can only be called after user login. It is fine to use
  54     // unsafe image decoder here. Before user login, a robust jpeg decoder will
  55     // be used.
  56     CHECK(chromeos::LoginState::Get()->IsUserLoggedIn());
  57     std::string image_data_str(image_data.begin(), image_data.end());
  58     ImageDecoder::StartWithOptions(this, image_data_str,
  59                                    ImageDecoder::DEFAULT_CODEC, true);
  60   }
  61
  62   void Cancel() {
  63     cancel_flag_.Set();
  64   }
  65
  66   void OnImageDecoded(const SkBitmap& decoded_image) override {
  67     // Make the SkBitmap immutable as we won't modify it. This is important
  68     // because otherwise it gets duplicated during painting, wasting memory.
  69     SkBitmap immutable(decoded_image);
  70     immutable.setImmutable();
  71     gfx::ImageSkia final_image = gfx::ImageSkia::CreateFrom1xBitmap(immutable);
  72     final_image.MakeThreadSafe();
  73     if (cancel_flag_.IsSet()) {
  74       function_->OnCancel();
  75       delete this;
  76       return;
  77     }
  78     function_->OnWallpaperDecoded(final_image);
  79     delete this;
  80   }
  81
  82   void OnDecodeImageFailed() override {
  83     function_->OnFailure(
  84         l10n_util::GetStringUTF8(IDS_WALLPAPER_MANAGER_INVALID_WALLPAPER));
  85     delete this;
  86   }
  87
  88  private:
  89   scoped_refptr<WallpaperFunctionBase> function_;
  90   base::CancellationFlag cancel_flag_;
  91
  92   DISALLOW_COPY_AND_ASSIGN(UnsafeWallpaperDecoder);
  93 };
  94
  95 WallpaperFunctionBase::UnsafeWallpaperDecoder*
  96     WallpaperFunctionBase::unsafe_wallpaper_decoder_;
  97
  98 WallpaperFunctionBase::WallpaperFunctionBase() {
  99 }
 100
 101 WallpaperFunctionBase::~WallpaperFunctionBase() {
 102 }
 103
 104 void WallpaperFunctionBase::StartDecode(const std::vector<char>& data) {
 105   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 106   if (unsafe_wallpaper_decoder_)
 107     unsafe_wallpaper_decoder_->Cancel();
 108   unsafe_wallpaper_decoder_ = new UnsafeWallpaperDecoder(this);
 109   unsafe_wallpaper_decoder_->Start(data);
 110 }
 111
 112 void WallpaperFunctionBase::OnCancel() {
 113   unsafe_wallpaper_decoder_ = NULL;
 114   SetError(wallpaper_api_util::kCancelWallpaperMessage);
 115   SendResponse(false);
 116 }
 117
 118 void WallpaperFunctionBase::OnFailure(const std::string& error) {
 119   unsafe_wallpaper_decoder_ = NULL;
 120   SetError(error);
 121   SendResponse(false);
 122 }