| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <memory> | 5 #include <memory> |
| 6 #include <vector> | 6 #include <vector> |
| 7 | 7 |
| 8 #include "base/bind.h" | 8 #include "base/bind.h" |
| 9 #include "base/lazy_instance.h" | 9 #include "base/lazy_instance.h" |
| 10 #include "base/message_loop/message_loop.h" | 10 #include "base/message_loop/message_loop.h" |
| 11 #include "gin/array_buffer.h" | 11 #include "gin/array_buffer.h" |
| 12 #include "gin/public/isolate_holder.h" | 12 #include "gin/public/isolate_holder.h" |
| 13 #include "services/image_decoder/image_decoder_impl.h" | 13 #include "services/data_decoder/image_decoder_impl.h" |
| 14 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
| 15 #include "third_party/WebKit/public/platform/scheduler/utility/webthread_impl_fo
r_utility_thread.h" | 15 #include "third_party/WebKit/public/platform/scheduler/utility/webthread_impl_fo
r_utility_thread.h" |
| 16 #include "third_party/WebKit/public/web/WebKit.h" | 16 #include "third_party/WebKit/public/web/WebKit.h" |
| 17 #include "third_party/skia/include/core/SkBitmap.h" | 17 #include "third_party/skia/include/core/SkBitmap.h" |
| 18 #include "ui/gfx/codec/jpeg_codec.h" | 18 #include "ui/gfx/codec/jpeg_codec.h" |
| 19 | 19 |
| 20 #if defined(V8_USE_EXTERNAL_STARTUP_DATA) | 20 #if defined(V8_USE_EXTERNAL_STARTUP_DATA) |
| 21 #include "gin/v8_initializer.h" | 21 #include "gin/v8_initializer.h" |
| 22 #endif | 22 #endif |
| 23 | 23 |
| 24 namespace image_decoder { | 24 namespace data_decoder { |
| 25 | 25 |
| 26 namespace { | 26 namespace { |
| 27 | 27 |
| 28 const int64_t kTestMaxImageSize = 128 * 1024; | 28 const int64_t kTestMaxImageSize = 128 * 1024; |
| 29 | 29 |
| 30 bool CreateJPEGImage(int width, | 30 bool CreateJPEGImage(int width, |
| 31 int height, | 31 int height, |
| 32 SkColor color, | 32 SkColor color, |
| 33 std::vector<unsigned char>* output) { | 33 std::vector<unsigned char>* output) { |
| 34 SkBitmap bitmap; | 34 SkBitmap bitmap; |
| (...skipping 76 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 111 } // namespace | 111 } // namespace |
| 112 | 112 |
| 113 // Test that DecodeImage() doesn't return image message > (max message size) | 113 // Test that DecodeImage() doesn't return image message > (max message size) |
| 114 TEST_F(ImageDecoderImplTest, DecodeImageSizeLimit) { | 114 TEST_F(ImageDecoderImplTest, DecodeImageSizeLimit) { |
| 115 // Approx max height for 3:2 image that will fit in the allotted space. | 115 // Approx max height for 3:2 image that will fit in the allotted space. |
| 116 // 1.5 for width/height ratio, 4 for bytes/pixel. | 116 // 1.5 for width/height ratio, 4 for bytes/pixel. |
| 117 int max_height_for_msg = sqrt(kTestMaxImageSize / (1.5 * 4)); | 117 int max_height_for_msg = sqrt(kTestMaxImageSize / (1.5 * 4)); |
| 118 int base_msg_size = sizeof(skia::mojom::Bitmap::Data_); | 118 int base_msg_size = sizeof(skia::mojom::Bitmap::Data_); |
| 119 | 119 |
| 120 // Sizes which should trigger dimension-halving 0, 1 and 2 times | 120 // Sizes which should trigger dimension-halving 0, 1 and 2 times |
| 121 int heights[] = {max_height_for_msg - 10, | 121 int heights[] = {max_height_for_msg - 10, max_height_for_msg + 10, |
| 122 max_height_for_msg + 10, | |
| 123 2 * max_height_for_msg + 10}; | 122 2 * max_height_for_msg + 10}; |
| 124 int widths[] = {heights[0] * 3 / 2, heights[1] * 3 / 2, heights[2] * 3 / 2}; | 123 int widths[] = {heights[0] * 3 / 2, heights[1] * 3 / 2, heights[2] * 3 / 2}; |
| 125 for (size_t i = 0; i < arraysize(heights); i++) { | 124 for (size_t i = 0; i < arraysize(heights); i++) { |
| 126 std::vector<unsigned char> jpg; | 125 std::vector<unsigned char> jpg; |
| 127 ASSERT_TRUE(CreateJPEGImage(widths[i], heights[i], SK_ColorRED, &jpg)); | 126 ASSERT_TRUE(CreateJPEGImage(widths[i], heights[i], SK_ColorRED, &jpg)); |
| 128 | 127 |
| 129 Request request(decoder()); | 128 Request request(decoder()); |
| 130 request.DecodeImage(jpg, true); | 129 request.DecodeImage(jpg, true); |
| 131 ASSERT_FALSE(request.bitmap().isNull()); | 130 ASSERT_FALSE(request.bitmap().isNull()); |
| 132 | 131 |
| (...skipping 21 matching lines...) Expand all Loading... |
| 154 // The "jpeg" is just some "random" data; | 153 // The "jpeg" is just some "random" data; |
| 155 const char kRandomData[] = "u gycfy7xdjkhfgui bdui "; | 154 const char kRandomData[] = "u gycfy7xdjkhfgui bdui "; |
| 156 std::vector<unsigned char> jpg(kRandomData, | 155 std::vector<unsigned char> jpg(kRandomData, |
| 157 kRandomData + sizeof(kRandomData)); | 156 kRandomData + sizeof(kRandomData)); |
| 158 | 157 |
| 159 Request request(decoder()); | 158 Request request(decoder()); |
| 160 request.DecodeImage(jpg, false); | 159 request.DecodeImage(jpg, false); |
| 161 EXPECT_TRUE(request.bitmap().isNull()); | 160 EXPECT_TRUE(request.bitmap().isNull()); |
| 162 } | 161 } |
| 163 | 162 |
| 164 } // namespace image_decoder | 163 } // namespace data_decoder |
| OLD | NEW |