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/renderer/extensions/set_icon_natives.h" 6 7 #include <limits> 8 9 #include "base/memory/scoped_ptr.h" 10 #include "chrome/common/render_messages.h" 11 #include "chrome/renderer/extensions/request_sender.h" 12 #include "third_party/skia/include/core/SkBitmap.h" 13 #include "third_party/skia/include/core/SkColor.h" 14 15 namespace { 16 17 const char* kImageSizeKeys[] = { "19", "38" }; 18 const char kInvalidDimensions[] = "ImageData has invalid dimensions."; 19 const char kInvalidData[] = "ImageData data length does not match dimensions."; 20 const char kNoMemory[] = "Chrome was unable to initialize icon."; 21 22 } // namespace 23 24 namespace extensions { 25 26 SetIconNatives::SetIconNatives(Dispatcher* dispatcher, 27 RequestSender* request_sender, 28 ChromeV8Context* context) 29 : ChromeV8Extension(dispatcher, context), 30 request_sender_(request_sender) { 31 RouteFunction( 32 "SetIconCommon", 33 base::Bind(&SetIconNatives::SetIconCommon, base::Unretained(this))); 34 } 35 36 bool SetIconNatives::ConvertImageDataToBitmapValue( 37 const v8::Local<v8::Object> image_data, 38 Value** bitmap_value) { 39 v8::Local<v8::Object> data = 40 image_data->Get(v8::String::New("data"))->ToObject(); 41 int width = image_data->Get(v8::String::New("width"))->Int32Value(); 42 int height = image_data->Get(v8::String::New("height"))->Int32Value(); 43 44 if (width <= 0 || height <= 0) { 45 v8::ThrowException( 46 v8::Exception::Error(v8::String::New(kInvalidDimensions))); 47 return false; 48 } 49 50 // We need to be able to safely check |data_length| == 4 * width * height 51 // without overflowing below. 52 int max_width = (std::numeric_limits<int>::max() / 4) / height; 53 if (width > max_width) { 54 v8::ThrowException( 55 v8::Exception::Error(v8::String::New(kInvalidDimensions))); 56 return false; 57 } 58 59 int data_length = data->Get(v8::String::New("length"))->Int32Value(); 60 if (data_length != 4 * width * height) { 61 v8::ThrowException(v8::Exception::Error(v8::String::New(kInvalidData))); 62 return false; 63 } 64 65 SkBitmap bitmap; 66 bitmap.setConfig(SkBitmap::kARGB_8888_Config, width, height); 67 if (!bitmap.allocPixels()) { 68 v8::ThrowException(v8::Exception::Error(v8::String::New(kNoMemory))); 69 return false; 70 } 71 bitmap.eraseARGB(0, 0, 0, 0); 72 73 uint32_t* pixels = bitmap.getAddr32(0, 0); 74 for (int t = 0; t < width*height; t++) { 75 // |data| is RGBA, pixels is ARGB. 76 pixels[t] = SkPreMultiplyColor( 77 ((data->Get(v8::Integer::New(4*t + 3))->Int32Value() & 0xFF) << 24) | 78 ((data->Get(v8::Integer::New(4*t + 0))->Int32Value() & 0xFF) << 16) | 79 ((data->Get(v8::Integer::New(4*t + 1))->Int32Value() & 0xFF) << 8) | 80 ((data->Get(v8::Integer::New(4*t + 2))->Int32Value() & 0xFF) << 0)); 81 } 82 83 // Construct the Value object. 84 IPC::Message bitmap_pickle; 85 IPC::WriteParam(&bitmap_pickle, bitmap); 86 *bitmap_value = base::BinaryValue::CreateWithCopiedBuffer( 87 static_cast<const char*>(bitmap_pickle.data()), bitmap_pickle.size()); 88 89 return true; 90 } 91 92 bool SetIconNatives::ConvertImageDataSetToBitmapValueSet( 93 const v8::FunctionCallbackInfo<v8::Value>& args, 94 base::DictionaryValue* bitmap_set_value) { 95 v8::Local<v8::Object> extension_args = args[1]->ToObject(); 96 v8::Local<v8::Object> details = 97 extension_args->Get(v8::String::New("0"))->ToObject(); 98 v8::Local<v8::Object> image_data_set = 99 details->Get(v8::String::New("imageData"))->ToObject(); 100 101 DCHECK(bitmap_set_value); 102 for (size_t i = 0; i < arraysize(kImageSizeKeys); i++) { 103 if (!image_data_set->Has(v8::String::New(kImageSizeKeys[i]))) 104 continue; 105 v8::Local<v8::Object> image_data = 106 image_data_set->Get(v8::String::New(kImageSizeKeys[i]))->ToObject(); 107 Value* image_data_bitmap = NULL; 108 if (!ConvertImageDataToBitmapValue(image_data, &image_data_bitmap)) 109 return false; 110 bitmap_set_value->Set(kImageSizeKeys[i], image_data_bitmap); 111 } 112 return true; 113 } 114 115 void SetIconNatives::SetIconCommon( 116 const v8::FunctionCallbackInfo<v8::Value>& args) { 117 scoped_ptr<base::DictionaryValue> bitmap_set_value( 118 new base::DictionaryValue()); 119 if (!ConvertImageDataSetToBitmapValueSet(args, bitmap_set_value.get())) 120 return; 121 122 v8::Local<v8::Object> extension_args = args[1]->ToObject(); 123 v8::Local<v8::Object> details = 124 extension_args->Get(v8::String::New("0"))->ToObject(); 125 126 base::DictionaryValue* dict = new base::DictionaryValue(); 127 dict->Set("imageData", bitmap_set_value.release()); 128 129 if (details->Has(v8::String::New("tabId"))) { 130 dict->SetInteger("tabId", 131 details->Get(v8::String::New("tabId"))->Int32Value()); 132 } 133 134 ListValue list_value; 135 list_value.Append(dict); 136 137 std::string name = *v8::String::AsciiValue(args[0]); 138 int request_id = args[2]->Int32Value(); 139 bool has_callback = args[3]->BooleanValue(); 140 bool for_io_thread = args[4]->BooleanValue(); 141 142 request_sender_->StartRequest(context(), 143 name, 144 request_id, 145 has_callback, 146 for_io_thread, 147 &list_value); 148 } 149 150 } // namespace extensions 151