OLD | NEW |
1 // Copyright 2016 The PDFium Authors. All rights reserved. | 1 // Copyright 2016 The PDFium 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 <cstdint> | 5 #include <cstdint> |
6 #include <memory> | 6 #include <memory> |
7 #include <vector> | 7 #include <vector> |
8 | 8 |
9 #include "core/fxcodec/codec/codec_int.h" | 9 #include "core/fxcodec/codec/codec_int.h" |
10 #include "core/include/fxge/fx_dib.h" | 10 #include "core/include/fxge/fx_dib.h" |
11 | 11 |
12 CCodec_JpxModule g_module; | 12 CCodec_JpxModule g_module; |
13 | 13 |
14 struct DecoderDeleter { | 14 struct DecoderDeleter { |
15 void operator()(CJPX_Decoder* decoder) { g_module.DestroyDecoder(decoder); } | 15 void operator()(CJPX_Decoder* decoder) { g_module.DestroyDecoder(decoder); } |
16 }; | 16 }; |
17 | 17 |
18 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) { | 18 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) { |
19 std::unique_ptr<CJPX_Decoder, DecoderDeleter> decoder( | 19 std::unique_ptr<CJPX_Decoder, DecoderDeleter> decoder( |
20 g_module.CreateDecoder(data, size, nullptr)); | 20 g_module.CreateDecoder(data, size, nullptr)); |
21 if (!decoder) | 21 if (!decoder) |
22 return 0; | 22 return 0; |
23 | 23 |
24 FX_DWORD width; | 24 uint32_t width; |
25 FX_DWORD height; | 25 uint32_t height; |
26 FX_DWORD components; | 26 uint32_t components; |
27 g_module.GetImageInfo(decoder.get(), &width, &height, &components); | 27 g_module.GetImageInfo(decoder.get(), &width, &height, &components); |
28 | 28 |
29 FXDIB_Format format; | 29 FXDIB_Format format; |
30 if (components == 1) { | 30 if (components == 1) { |
31 format = FXDIB_8bppRgb; | 31 format = FXDIB_8bppRgb; |
32 } else if (components <= 3) { | 32 } else if (components <= 3) { |
33 format = FXDIB_Rgb; | 33 format = FXDIB_Rgb; |
34 } else if (components == 4) { | 34 } else if (components == 4) { |
35 format = FXDIB_Rgb32; | 35 format = FXDIB_Rgb32; |
36 } else { | 36 } else { |
37 width = (width * components + 2) / 3; | 37 width = (width * components + 2) / 3; |
38 format = FXDIB_Rgb; | 38 format = FXDIB_Rgb; |
39 } | 39 } |
40 | 40 |
41 std::unique_ptr<CFX_DIBitmap> bitmap(new CFX_DIBitmap); | 41 std::unique_ptr<CFX_DIBitmap> bitmap(new CFX_DIBitmap); |
42 if (!bitmap->Create(width, height, format)) | 42 if (!bitmap->Create(width, height, format)) |
43 return 0; | 43 return 0; |
44 | 44 |
45 std::vector<uint8_t> output_offsets(components); | 45 std::vector<uint8_t> output_offsets(components); |
46 for (FX_DWORD i = 0; i < components; ++i) | 46 for (uint32_t i = 0; i < components; ++i) |
47 output_offsets[i] = i; | 47 output_offsets[i] = i; |
48 | 48 |
49 g_module.Decode(decoder.get(), bitmap->GetBuffer(), bitmap->GetPitch(), | 49 g_module.Decode(decoder.get(), bitmap->GetBuffer(), bitmap->GetPitch(), |
50 output_offsets); | 50 output_offsets); |
51 return 0; | 51 return 0; |
52 } | 52 } |
OLD | NEW |